feat(NODE-6695): enable KMS retry protocol (#60)

mongodb-js · Jan 30, 2025 · d4b7ef7 · d4b7ef7
1 parent faea384
commit d4b7ef7
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/addon/mongocrypt.cc b/addon/mongocrypt.cc
@@ -575,6 +575,8 @@ MongoCrypt::MongoCrypt(const CallbackInfo& info) : ObjectWrap(info) {
 
     mongocrypt_setopt_use_need_kms_credentials_state(mongo_crypt());
 
+    mongocrypt_setopt_retry_kms(mongo_crypt(), true);
+
     // Initialize after all options are set.
     if (!mongocrypt_init(mongo_crypt())) {
         throw TypeError::New(Env(), errorStringFromStatus(mongo_crypt()));
@@ -947,6 +949,8 @@ Function MongoCryptKMSRequest::Init(Napi::Env env) {
         {InstanceMethod("addResponse", &MongoCryptKMSRequest::AddResponse),
          InstanceAccessor("status", &MongoCryptKMSRequest::Status, nullptr),
          InstanceAccessor("bytesNeeded", &MongoCryptKMSRequest::BytesNeeded, nullptr),
+         InstanceAccessor("uSleep", &MongoCryptKMSRequest::USleep, nullptr),
+         InstanceAccessor("fail", &MongoCryptKMSRequest::Fail, nullptr),
          InstanceAccessor("kmsProvider", &MongoCryptKMSRequest::KMSProvider, nullptr),
          InstanceAccessor("endpoint", &MongoCryptKMSRequest::Endpoint, nullptr),
          InstanceAccessor("message", &MongoCryptKMSRequest::Message, nullptr)});
@@ -977,6 +981,14 @@ Value MongoCryptKMSRequest::BytesNeeded(const CallbackInfo& info) {
     return Number::New(Env(), mongocrypt_kms_ctx_bytes_needed(_kms_context));
 }
 
+Value MongoCryptKMSRequest::USleep(const CallbackInfo& info) {
+    return Number::New(Env(), mongocrypt_kms_ctx_usleep(_kms_context));
+}
+
+Value MongoCryptKMSRequest::Fail(const CallbackInfo& info) {
+    return Boolean::New(Env(), mongocrypt_kms_ctx_fail(_kms_context));
+}
+
 Value MongoCryptKMSRequest::KMSProvider(const CallbackInfo& info) {
     return String::New(Env(), mongocrypt_kms_ctx_get_kms_provider(_kms_context, nullptr));
 }

diff --git a/addon/mongocrypt.h b/addon/mongocrypt.h
@@ -151,6 +151,8 @@ class MongoCryptKMSRequest : public Napi::ObjectWrap<MongoCryptKMSRequest> {
     Napi::Value Status(const Napi::CallbackInfo& info);
     Napi::Value Message(const Napi::CallbackInfo& info);
     Napi::Value BytesNeeded(const Napi::CallbackInfo& info);
+    Napi::Value Fail(const Napi::CallbackInfo& info);
+    Napi::Value USleep(const Napi::CallbackInfo& info);
     Napi::Value KMSProvider(const Napi::CallbackInfo& info);
     Napi::Value Endpoint(const Napi::CallbackInfo& info);
 

diff --git a/package.json b/package.json
@@ -35,7 +35,7 @@
   },
   "license": "Apache-2.0",
   "gypfile": true,
-  "mongodb:libmongocrypt": "1.11.0",
+  "mongodb:libmongocrypt": "1.12.0",
   "dependencies": {
     "node-addon-api": "^4.3.0",
     "prebuild-install": "^7.1.2"

diff --git a/src/index.ts b/src/index.ts
@@ -23,8 +23,10 @@ type MongoCryptBindings = {
 
 export interface MongoCryptKMSRequest {
   addResponse(response: Uint8Array): void;
+  fail(): boolean;
   readonly status: MongoCryptStatus;
   readonly bytesNeeded: number;
+  readonly uSleep: number;
   readonly kmsProvider: string;
   readonly endpoint: string;
   readonly message: Buffer;