edhoc 1.2.3 → 1.3.1

package/include/Utils.h

@@ -2,12 +2,11 @@
 #define UTILS_H
 
 #include <napi.h>
-
 #include <cstdint>
 #include <future>
-#include <iostream>
 #include <string>
 #include <vector>
+
 extern "C" {
 #include "edhoc.h"
 }
@@ -22,49 +21,6 @@ extern "C" {
  */
 class Utils {
  public:
-  using SuccessHandler = std::function<void(Napi::Env, Napi::Value)>;
-  using ErrorHandler = std::function<void(Napi::Env, Napi::Error)>;
-
-  /**
-   * Invokes a JavaScript function with promise handling.
-   * The function is called with the specified arguments, and the result is
-   * passed to a callback function.
-   *
-   * @param env The N-API environment handle.
-   * @param jsObject The N-API object representing the JavaScript class.
-   * @param jsCallback The N-API function object to call.
-   * @param args A vector of N-API values representing the arguments to pass to
-   * the function.
-   * @param callbackLambda A lambda function to handle the result of the
-   * function call.
-   */
-  static void InvokeJSFunctionWithPromiseHandling(Napi::Env env,
-                                                  Napi::Object jsObject,
-                                                  Napi::Function jsCallback,
-                                                  const std::vector<napi_value>& args,
-                                                  SuccessHandler successLambda,
-                                                  ErrorHandler errorLambda);
-
-  /**
-   * Creates a promise error handler for a given promise.
-   *
-   * @tparam T The type of the promise.
-   * @param promise The promise to handle errors for.
-   * @return A lambda function that sets the exception on the promise.
-   */
-  template <typename T>
-  static ErrorHandler CreatePromiseErrorHandler(std::promise<T>& promise, T defaultValue, Napi::Error& lastError) {
-    return [&promise, defaultValue, &lastError](Napi::Env env, Napi::Error error) {
-      if (!error.IsEmpty()) {
-        error.ThrowAsJavaScriptException();
-      }
-      if (env.IsExceptionPending()) {
-        lastError = env.GetAndClearPendingException();
-      }
-      promise.set_value(defaultValue);
-    };
-  }
-
   /**
    * Converts a JavaScript value to an EDHOC connection ID structure.
    * The input can be a number or a buffer.
@@ -95,7 +51,7 @@ class Utils {
   template <typename EnumType>
   static EnumType ConvertToEnum(const Napi::Value& value) {
     if (!value.IsNumber()) {
-      Napi::TypeError::New(value.Env(), "Input value must be a number").ThrowAsJavaScriptException();
+      throw Napi::TypeError::New(value.Env(), "Input value must be a number");
     }
     return static_cast<EnumType>(value.As<Napi::Number>().Int32Value());
   }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "edhoc",
   "description": "A Node.js implementation of EDHOC (Ephemeral Diffie-Hellman Over COSE) protocol for lightweight authenticated key exchange in IoT and other constrained environments.",
-  "version": "1.2.3",
+  "version": "1.3.1",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "author": "Marek Serafin <marek@serafin.email>",
@@ -53,7 +53,6 @@
     "@semantic-release/git": "^10.0.1",
     "@semantic-release/github": "^10.1.1",
     "@types/elliptic": "^6.4.18",
-    "@types/eslint__js": "^8.42.3",
     "@types/jest": "^29.5.14",
     "@types/node": "^22.13.4",
     "eslint": "^8.57.0",

package/src/Binding.cpp

@@ -0,0 +1,434 @@
+#include "Binding.h"
+
+#include <iostream>
+#include <thread>
+
+#include "EdhocComposeAsyncWorker.h"
+#include "EdhocExportOscoreAsyncWorker.h"
+#include "EdhocKeyExporterAsyncWorker.h"
+#include "EdhocKeyUpdateAsyncWorker.h"
+#include "EdhocProcessAsyncWorker.h"
+#include "Suites.h"
+#include "Utils.h"
+
+static constexpr const char* kErrorFailedToInitializeEdhocContext = "Failed to initialize EDHOC context.";
+static constexpr const char* kErrorFailedToSetEdhocConnectionId = "Failed to set EDHOC Connection ID.";
+static constexpr const char* kErrorFailedToSetEdhocMethod = "Failed to set EDHOC Method.";
+static constexpr const char* kErrorArraySuiteIndexesExpected = "Array of suite indexes expected";
+static constexpr const char* kErrorArrayMethodIndexesExpected = "Array of method indexes expected";
+static constexpr const char* kErrorInvalidCipherSuiteIndex = "Invalid cipher suite index";
+static constexpr const char* kErrorFailedToSetCipherSuites = "Failed to set cipher suites";
+static constexpr const char* kErrorExpectedFirstArgumentToBeBuffer = "Expected first argument to be a Buffer";
+static constexpr const char* kErrorExpectedArgumentToBeNumber = "Expected argument to be a number";
+static constexpr const char* kErrorExpectedAFunction = "Expected a function";
+
+Edhoc::Edhoc(const Napi::CallbackInfo& info) : Napi::ObjectWrap<Edhoc>(info) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  // Get the JS object
+  Napi::Object jsEdhoc = info.This().As<Napi::Object>();
+  
+  // Crypto Manager
+  Napi::Object jsCryptoManager = info[4].As<Napi::Object>();
+  this->cryptoManager_ = std::make_unique<EdhocCryptoManager>(jsCryptoManager, jsEdhoc);
+
+  // Credentials
+  Napi::Object jsCredentialManager = info[3].As<Napi::Object>();
+  this->credentialManager_ = std::make_unique<EdhocCredentialManager>(jsCredentialManager, jsEdhoc);
+  
+  // EAD
+  this->eadManager_ = std::make_unique<EdhocEadManager>();
+
+  // Reset the EDHOC context
+  this->Reset(info);
+
+  // Connection ID, Methods, and Suites
+  SetCID(info, info[0]);
+  SetMethods(info, info[1]);
+  SetCipherSuites(info, info[2]);
+}
+
+void Edhoc::Reset(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  bool isInitialized = edhocContext_ != nullptr;
+
+  // Get the Connection ID, Methods, and Suites
+  Napi::Value cid = env.Null();
+  Napi::Value methods = env.Null();
+  Napi::Value suites = env.Null();
+
+  if (isInitialized) {
+    cid = this->GetCID(info);
+    methods = this->GetMethods(info);
+    suites = this->GetCipherSuites(info);
+  }
+
+  // Initialize EDHOC context
+  edhocContext_ = std::make_unique<edhoc_context>();
+
+  if (edhoc_context_init(edhocContext_.get()) != EDHOC_SUCCESS) {
+    Napi::TypeError::New(env, kErrorFailedToInitializeEdhocContext)
+      .ThrowAsJavaScriptException();
+  }
+
+  // Bind all managers
+  if (edhoc_bind_keys(edhocContext_.get(), &this->cryptoManager_.get()->keys) != EDHOC_SUCCESS ||
+      edhoc_bind_crypto(edhocContext_.get(), &this->cryptoManager_.get()->crypto) != EDHOC_SUCCESS ||
+      edhoc_bind_credentials(edhocContext_.get(), &this->credentialManager_.get()->credentials) != EDHOC_SUCCESS ||
+      edhoc_bind_ead(edhocContext_.get(), &this->eadManager_.get()->ead) != EDHOC_SUCCESS) 
+  {
+    Napi::TypeError::New(env, kErrorFailedToInitializeEdhocContext)
+      .ThrowAsJavaScriptException();
+  }
+
+  // If the previous context was initialized, copy the Connection ID, Methods, and Suites
+  if (isInitialized) {
+    this->SetCID(info, cid);
+    this->SetMethods(info, methods);
+    this->SetCipherSuites(info, suites);
+  }
+
+  // Logger
+  edhocContext_->logger = Edhoc::Logger;
+}
+
+Edhoc::~Edhoc() {
+  // Reset the EDHOC context
+}
+
+Napi::Value Edhoc::GetCID(const Napi::CallbackInfo& info) {
+  return Utils::CreateJsValueFromEdhocCid(info.Env(), cid_);
+}
+
+void Edhoc::SetCID(const Napi::CallbackInfo& info, const Napi::Value& value) {
+  cid_ = Utils::ConvertJsValueToEdhocCid(value);
+  int result = edhoc_set_connection_id(edhocContext_.get(), &cid_);
+  if (result != EDHOC_SUCCESS) {
+    Napi::TypeError::New(info.Env(), kErrorFailedToSetEdhocConnectionId).ThrowAsJavaScriptException();
+  }
+}
+
+Napi::Value Edhoc::GetPeerCID(const Napi::CallbackInfo& info) {
+  return Utils::CreateJsValueFromEdhocCid(info.Env(), edhocContext_->EDHOC_PRIVATE(peer_cid));
+}
+
+Napi::Value Edhoc::GetMethods(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::Array result = Napi::Array::New(env, edhocContext_->EDHOC_PRIVATE(method_len));
+  for (size_t i = 0; i < edhocContext_->EDHOC_PRIVATE(method_len); i++) {
+    result.Set(i, Napi::Number::New(env, edhocContext_->EDHOC_PRIVATE(method)[i]));
+  }
+  return result;
+}
+
+void Edhoc::SetMethods(const Napi::CallbackInfo& info, const Napi::Value& value) {
+  Napi::Env env = info.Env();
+
+  if (!value.IsArray()) {
+    Napi::TypeError::New(env, kErrorArrayMethodIndexesExpected)
+      .ThrowAsJavaScriptException();
+  }
+
+  const auto jsArray = value.As<Napi::Array>();
+  std::vector<edhoc_method> methods;
+  methods.reserve(jsArray.Length());
+
+  for (uint32_t i = 0; i < jsArray.Length(); i++) {
+    methods.push_back(static_cast<edhoc_method>(jsArray.Get(i).As<Napi::Number>().Int32Value()));
+  }
+  
+  int result = edhoc_set_methods(edhocContext_.get(), methods.data(), methods.size());
+  if (result != EDHOC_SUCCESS) {
+    Napi::TypeError::New(env, kErrorFailedToSetEdhocMethod)
+      .ThrowAsJavaScriptException();
+  }
+}
+
+Napi::Value Edhoc::GetSelectedMethod(const Napi::CallbackInfo& info) {
+  return Napi::Number::New(info.Env(), edhocContext_->EDHOC_PRIVATE(chosen_method));
+}
+
+void Edhoc::SetCipherSuites(const Napi::CallbackInfo& info, const Napi::Value& value) {
+  Napi::Env env = info.Env();
+
+  if (!value.IsArray()) {
+    Napi::TypeError::New(env, kErrorArraySuiteIndexesExpected)
+      .ThrowAsJavaScriptException();
+  }
+
+  const auto jsArray = value.As<Napi::Array>();
+  std::vector<edhoc_cipher_suite> selected_suites;
+  selected_suites.reserve(jsArray.Length());
+
+  for (uint32_t i = 0; i < jsArray.Length(); i++) {
+    const uint32_t index = jsArray.Get(i).As<Napi::Number>().Uint32Value();
+
+    if (index >= suite_pointers_count || suite_pointers[index] == nullptr) {
+      Napi::RangeError::New(env, kErrorInvalidCipherSuiteIndex)
+        .ThrowAsJavaScriptException();
+    }
+
+    selected_suites.push_back(*suite_pointers[index]);
+  }
+
+  if (edhoc_set_cipher_suites(edhocContext_.get(), selected_suites.data(), selected_suites.size()) != 0) {
+    Napi::TypeError::New(env, kErrorFailedToSetCipherSuites)
+      .ThrowAsJavaScriptException();
+  }
+}
+
+Napi::Value Edhoc::GetCipherSuites(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::Array result = Napi::Array::New(env, edhocContext_->EDHOC_PRIVATE(csuite_len));
+  for (size_t i = 0; i < edhocContext_->EDHOC_PRIVATE(csuite_len); i++) {
+    result.Set(i, Napi::Number::New(env, edhocContext_->EDHOC_PRIVATE(csuite)[i].value));
+  }
+  return result;
+}
+
+Napi::Value Edhoc::GetSelectedCipherSuite(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::Number suite =
+      Napi::Number::New(env, edhocContext_->EDHOC_PRIVATE(csuite)[edhocContext_->EDHOC_PRIVATE(chosen_csuite_idx)].value);
+  return suite;
+}
+
+Napi::Value Edhoc::GetLogger(const Napi::CallbackInfo& info) {
+  return logger_.Value();
+}
+
+void Edhoc::SetLogger(const Napi::CallbackInfo& info, const Napi::Value& value) {
+  if (!info[0].IsFunction()) {
+    Napi::TypeError::New(info.Env(), kErrorExpectedAFunction).ThrowAsJavaScriptException();
+  }
+  Napi::Function jsCallback = info[0].As<Napi::Function>();
+  logger_ = Napi::Persistent(jsCallback);
+}
+
+void Edhoc::Logger(void* user_context, const char* name, const uint8_t* buffer, size_t buffer_length) {
+  RunningContext* context = static_cast<RunningContext*>(const_cast<void*>(user_context));
+  
+  if (!context || !context->GetTsfn() || !context->GetLoggerRef()) {
+    return;
+  }
+
+  const std::vector<uint8_t> bufferCopy(buffer, buffer + buffer_length);
+
+  context->GetTsfn().NonBlockingCall(context, [name = std::string(name), bufferCopy](Napi::Env env, Napi::Function jsCallback, RunningContext* context) {
+    Napi::HandleScope scope(env);
+    try {
+      context->GetLoggerRef().Value().As<Napi::Function>().Call({Napi::String::New(env, name), Napi::Buffer<uint8_t>::Copy(env, bufferCopy.data(), bufferCopy.size())});
+    } catch (const Napi::Error& e) {
+       // This is just a logger, so we don't want to throw an error
+      std::cerr << "Error in Logger: " << e.Get("stack").ToString().Utf8Value() << std::endl;
+    }
+  });
+}
+
+void Edhoc::StartRunningContext(Napi::Env env) {
+  // Initialize the running context
+  this->runningContext_ = std::make_unique<RunningContext>(
+    env, 
+    edhocContext_.get(), 
+    this->cryptoManager_.get(), 
+    this->eadManager_.get(), 
+    this->credentialManager_.get(), 
+    logger_.Value()
+  );
+
+  // Set the user context of the EDHOC context
+  if (edhoc_set_user_context(edhocContext_.get(), static_cast<void*>(this->runningContext_.get())) != EDHOC_SUCCESS) {
+    Napi::TypeError::New(env, kErrorFailedToInitializeEdhocContext)
+      .ThrowAsJavaScriptException();
+  }
+}
+
+Napi::Value Edhoc::ComposeMessage(const Napi::CallbackInfo& info, enum edhoc_message messageNumber) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  // Check if the first argument (EADs) is an array
+  if (info[0].IsArray()) {
+    try {
+      this->eadManager_->StoreEad(messageNumber, info[0].As<Napi::Array>());
+    } catch (const Napi::Error& e) {
+      e.ThrowAsJavaScriptException();
+      return env.Null();
+    }
+  }
+
+  // Start the running context
+  this->StartRunningContext(env);
+
+  // Compose the message
+  EdhocComposeAsyncWorker* worker = new EdhocComposeAsyncWorker(this->runningContext_.get(), messageNumber);
+  worker->Queue();
+
+  // Return the promise of the running context
+  return this->runningContext_->GetPromise();
+}
+
+Napi::Value Edhoc::ProcessMessage(const Napi::CallbackInfo& info, enum edhoc_message messageNumber) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  // Check if the first argument (input buffer) is a buffer
+  if (info.Length() < 1 || !info[0].IsBuffer()) {
+    Napi::TypeError::New(env, kErrorExpectedFirstArgumentToBeBuffer).ThrowAsJavaScriptException();
+    return env.Null();
+  }
+
+  // Get the input buffer
+  Napi::Buffer<uint8_t> inputBuffer = info[0].As<Napi::Buffer<uint8_t>>();
+
+  // Start the running context
+  this->StartRunningContext(env);
+
+  // Process the message
+  EdhocProcessAsyncWorker* worker = new EdhocProcessAsyncWorker(this->runningContext_.get(), messageNumber, inputBuffer);
+  worker->Queue();
+
+  // Return the promise of the running context
+  return this->runningContext_->GetPromise();
+}
+
+Napi::Value Edhoc::ComposeMessage1(const Napi::CallbackInfo& info) {
+  return ComposeMessage(info, EDHOC_MSG_1);
+}
+
+Napi::Value Edhoc::ProcessMessage1(const Napi::CallbackInfo& info) {
+  return ProcessMessage(info, EDHOC_MSG_1);
+}
+
+Napi::Value Edhoc::ComposeMessage2(const Napi::CallbackInfo& info) {
+  return ComposeMessage(info, EDHOC_MSG_2);
+}
+
+Napi::Value Edhoc::ProcessMessage2(const Napi::CallbackInfo& info) {
+  return ProcessMessage(info, EDHOC_MSG_2);
+}
+
+Napi::Value Edhoc::ComposeMessage3(const Napi::CallbackInfo& info) {
+  return ComposeMessage(info, EDHOC_MSG_3);
+}
+
+Napi::Value Edhoc::ProcessMessage3(const Napi::CallbackInfo& info) {
+  return ProcessMessage(info, EDHOC_MSG_3);
+}
+
+Napi::Value Edhoc::ComposeMessage4(const Napi::CallbackInfo& info) {
+  return ComposeMessage(info, EDHOC_MSG_4);
+}
+
+Napi::Value Edhoc::ProcessMessage4(const Napi::CallbackInfo& info) {
+  return ProcessMessage(info, EDHOC_MSG_4);
+}
+
+Napi::Value Edhoc::ExportOSCORE(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  // Start the running context
+  this->StartRunningContext(env);
+
+  // Export the OSCORE
+  EdhocExportOscoreAsyncWorker* worker = new EdhocExportOscoreAsyncWorker(this->runningContext_.get());
+  worker->Queue();
+
+  // Return the promise of the running context
+  return this->runningContext_->GetPromise();
+}
+
+Napi::Value Edhoc::ExportKey(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  // Check if the first argument (label) is a number
+  if (info.Length() < 1 || !info[0].IsNumber()) {
+    Napi::TypeError::New(env, kErrorExpectedArgumentToBeNumber).ThrowAsJavaScriptException();
+    return env.Null();
+  }
+
+  // Check if the second argument (desired length) is a number
+  if (info.Length() < 2 || !info[1].IsNumber()) {
+    Napi::TypeError::New(env, kErrorExpectedArgumentToBeNumber).ThrowAsJavaScriptException();
+    return env.Null();
+  }
+
+  // Get the label and desired length
+  uint16_t label = (uint16_t)info[0].As<Napi::Number>().Uint32Value();
+  uint8_t desiredLength = (uint8_t)info[1].As<Napi::Number>().Uint32Value();
+
+  // Start the running context
+  this->StartRunningContext(env);
+
+  // Export the key
+  EdhocKeyExporterAsyncWorker* worker = new EdhocKeyExporterAsyncWorker(this->runningContext_.get(), label, desiredLength);
+  worker->Queue();
+
+  // Return the promise of the running context
+  return this->runningContext_->GetPromise();
+}
+
+Napi::Value Edhoc::KeyUpdate(const Napi::CallbackInfo& info) {
+  Napi::Env env = info.Env();
+  Napi::HandleScope scope(env);
+
+  // Check if the first argument (context buffer) is a buffer
+  if (info.Length() < 1 || !info[0].IsBuffer()) {
+    Napi::TypeError::New(env, kErrorExpectedFirstArgumentToBeBuffer).ThrowAsJavaScriptException();
+    return env.Null();
+  }
+
+  // Get the context buffer
+  Napi::Buffer<uint8_t> contextBuffer = info[0].As<Napi::Buffer<uint8_t>>();
+  std::vector<uint8_t> contextBufferVector(contextBuffer.Data(), contextBuffer.Data() + contextBuffer.Length());
+
+  // Start the running context
+  this->StartRunningContext(env);
+
+  // Update the key
+  EdhocKeyUpdateAsyncWorker* worker = new EdhocKeyUpdateAsyncWorker(this->runningContext_.get(), contextBufferVector);
+  worker->Queue();
+
+  // Return the promise of the running context
+  return this->runningContext_->GetPromise();
+}
+
+Napi::Object Edhoc::Init(Napi::Env env, Napi::Object exports) {
+  Napi::HandleScope scope(env);
+
+  Napi::Function func = DefineClass(env, "EDHOC", {
+    InstanceAccessor("connectionID", &Edhoc::GetCID, &Edhoc::SetCID),
+    InstanceAccessor<&Edhoc::GetPeerCID>("peerConnectionID"),
+    InstanceAccessor("methods", &Edhoc::GetMethods, &Edhoc::SetMethods),
+    InstanceAccessor<&Edhoc::GetSelectedMethod>("selectedMethod"),
+    InstanceAccessor("cipherSuites", &Edhoc::GetCipherSuites, &Edhoc::SetCipherSuites),
+    InstanceAccessor<&Edhoc::GetSelectedCipherSuite>("selectedSuite"),
+    InstanceAccessor("logger", &Edhoc::GetLogger, &Edhoc::SetLogger),
+    InstanceMethod("reset", &Edhoc::Reset),
+    InstanceMethod("composeMessage1", &Edhoc::ComposeMessage1),
+    InstanceMethod("processMessage1", &Edhoc::ProcessMessage1),
+    InstanceMethod("composeMessage2", &Edhoc::ComposeMessage2),
+    InstanceMethod("processMessage2", &Edhoc::ProcessMessage2),
+    InstanceMethod("composeMessage3", &Edhoc::ComposeMessage3),
+    InstanceMethod("processMessage3", &Edhoc::ProcessMessage3),
+    InstanceMethod("composeMessage4", &Edhoc::ComposeMessage4),
+    InstanceMethod("processMessage4", &Edhoc::ProcessMessage4),
+    InstanceMethod("exportOSCORE", &Edhoc::ExportOSCORE),
+    InstanceMethod("exportKey", &Edhoc::ExportKey),
+    InstanceMethod("keyUpdate", &Edhoc::KeyUpdate),
+  });
+
+  Napi::FunctionReference* constructor = new Napi::FunctionReference();
+  *constructor = Napi::Persistent(func);
+  env.SetInstanceData(constructor);
+
+  exports.Set("EDHOC", func);
+  return exports;
+}
+
+NODE_API_NAMED_ADDON(addon, Edhoc);

package/src/EdhocComposeAsyncWorker.cpp

@@ -5,74 +5,56 @@ static constexpr const char* kErrorInvalidMessageNumber = "Invalid message numbe
 static constexpr const char* kErrorMessageFormat = "Failed to compose EDHOC message %d. Error code: %d";
 static constexpr size_t kErrorBufferSize = 100;
 
-EdhocComposeAsyncWorker::EdhocComposeAsyncWorker(Napi::Env& env,
-                                                 struct edhoc_context& context,
-                                                 int messageNumber,
-                                                 CallbackType callback)
-    : Napi::AsyncWorker(env),
-      deferred(Napi::Promise::Deferred::New(env)),
-      context(context),
-      messageNumber(messageNumber),
-      callback(std::move(callback)) {}
+EdhocComposeAsyncWorker::EdhocComposeAsyncWorker(RunningContext* runningContext, int messageNumber)
+    : Napi::AsyncWorker(runningContext->GetEnv()),
+      runningContext_(runningContext),
+      messageNumber_(messageNumber) {}
 
 void EdhocComposeAsyncWorker::Execute() {
-  composedMessage.resize(kInitialBufferSize);
-  size_t composedMessageLength = 0;
-
-  int ret = EDHOC_ERROR_GENERIC_ERROR;
-  switch (messageNumber) {
-    case EDHOC_MSG_1:
-      ret = edhoc_message_1_compose(&context, composedMessage.data(), composedMessage.size(), &composedMessageLength);
-      break;
-    case EDHOC_MSG_2:
-      ret = edhoc_message_2_compose(&context, composedMessage.data(), composedMessage.size(), &composedMessageLength);
-      break;
-    case EDHOC_MSG_3:
-      ret = edhoc_message_3_compose(&context, composedMessage.data(), composedMessage.size(), &composedMessageLength);
-      break;
-    case EDHOC_MSG_4:
-      ret = edhoc_message_4_compose(&context, composedMessage.data(), composedMessage.size(), &composedMessageLength);
-      break;
-    default:
-      SetError(kErrorInvalidMessageNumber);
-      return;
-  }
-
-  composedMessage.resize(composedMessageLength);
-
-  if (ret != EDHOC_SUCCESS) {
-    char errorMessage[kErrorBufferSize];
-    std::snprintf(errorMessage, kErrorBufferSize, kErrorMessageFormat, messageNumber + 1, ret);
-    SetError(errorMessage);
+  try {
+    composedMessage_.resize(kInitialBufferSize);
+    size_t composedMessageLength = 0;
+
+    int ret = EDHOC_ERROR_GENERIC_ERROR;
+    switch (messageNumber_) {
+      case EDHOC_MSG_1:
+        ret = edhoc_message_1_compose(runningContext_->GetEdhocContext(), composedMessage_.data(), composedMessage_.size(), &composedMessageLength);
+        break;
+      case EDHOC_MSG_2:
+        ret = edhoc_message_2_compose(runningContext_->GetEdhocContext(), composedMessage_.data(), composedMessage_.size(), &composedMessageLength);
+        break;
+      case EDHOC_MSG_3:
+        ret = edhoc_message_3_compose(runningContext_->GetEdhocContext(), composedMessage_.data(), composedMessage_.size(), &composedMessageLength);
+        break;
+      case EDHOC_MSG_4:
+        ret = edhoc_message_4_compose(runningContext_->GetEdhocContext(), composedMessage_.data(), composedMessage_.size(), &composedMessageLength);
+        break;
+      default:
+        SetError(kErrorInvalidMessageNumber);
+        return;
+    }
+
+    composedMessage_.resize(composedMessageLength);
+
+    if (ret != EDHOC_SUCCESS) {
+      char errorMessage[kErrorBufferSize];
+      std::snprintf(errorMessage, kErrorBufferSize, kErrorMessageFormat, messageNumber_ + 1, ret);
+      SetError(errorMessage);
+    }
+  } catch (const std::exception& e) {
+    SetError(e.what());
   }
 }
 
 void EdhocComposeAsyncWorker::OnOK() {
   Napi::Env env = Env();
   Napi::HandleScope scope(env);
-
-  callback(env);
-
-  if (env.IsExceptionPending()) {
-    deferred.Reject(env.GetAndClearPendingException().Value());
-  } else {
-    deferred.Resolve(Napi::Buffer<uint8_t>::Copy(env, composedMessage.data(), composedMessage.size()));
-  }
+  runningContext_->Resolve(Napi::Buffer<uint8_t>::Copy(env, composedMessage_.data(), composedMessage_.size()));
+  runningContext_->GetEadManager()->ClearEadByMessage(static_cast<enum edhoc_message>(messageNumber_));
 }
 
 void EdhocComposeAsyncWorker::OnError(const Napi::Error& error) {
   Napi::Env env = Env();
   Napi::HandleScope scope(env);
-
-  callback(env);
-
-  if (env.IsExceptionPending()) {
-    deferred.Reject(env.GetAndClearPendingException().Value());
-  } else {
-    deferred.Reject(error.Value());
-  }
-}
-
-Napi::Promise EdhocComposeAsyncWorker::GetPromise() {
-  return deferred.Promise();
+  runningContext_->Reject(error.Value());
 }