From 037e52b0d6c18a71c7ead3ca865499a40f890a5f Mon Sep 17 00:00:00 2001
From: Darshan Sen <darshan.sen@postman.com>
Date: Sun, 21 Nov 2021 14:12:54 +0530
Subject: [PATCH] src,crypto: use `ByteSource::ToBuffer()` in `crypto_dh`

Signed-off-by: Darshan Sen <darshan.sen@postman.com>
---
 src/crypto/crypto_dh.cc | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/src/crypto/crypto_dh.cc b/src/crypto/crypto_dh.cc
index 72f6a114ddfbc2..8a2c559d783eda 100644
--- a/src/crypto/crypto_dh.cc
+++ b/src/crypto/crypto_dh.cc
@@ -13,7 +13,6 @@
 
 namespace node {
 
-using v8::ArrayBuffer;
 using v8::ConstructorBehavior;
 using v8::DontDelete;
 using v8::FunctionCallback;
@@ -610,13 +609,8 @@ void DiffieHellman::Stateless(const FunctionCallbackInfo<Value>& args) {
   ManagedEVPPKey our_key = our_key_object->Data()->GetAsymmetricKey();
   ManagedEVPPKey their_key = their_key_object->Data()->GetAsymmetricKey();
 
-  Local<Value> out;
-  {
-    Local<ArrayBuffer> ab = StatelessDiffieHellmanThreadsafe(our_key, their_key)
-        .ToArrayBuffer(env);
-    out = Buffer::New(env, ab, 0, ab->ByteLength())
-        .FromMaybe(Local<Uint8Array>());
-  }
+  Local<Value> out = StatelessDiffieHellmanThreadsafe(our_key, their_key)
+      .ToBuffer(env).FromMaybe(Local<Uint8Array>());
 
   if (Buffer::Length(out) == 0)
     return ThrowCryptoError(env, ERR_get_error(), "diffieHellman failed");