fixup! crypto: fix DiffieHellman argument validation

nodejs · Mar 20, 2021 · 1812e6d · 1812e6d
1 parent 5aa7b2c
commit 1812e6d
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/internal/crypto/diffiehellman.js b/lib/internal/crypto/diffiehellman.js
@@ -118,18 +118,19 @@ function DiffieHellman(sizeOrKey, keyEncoding, generator, genEncoding) {
   if (typeof sizeOrKey !== 'number')
     sizeOrKey = toBuf(sizeOrKey, keyEncoding);
 
-  if (!generator)
+  if (!generator) {
     generator = DH_GENERATOR;
-  else if (typeof generator === 'number')
+  } else if (typeof generator === 'number') {
     validateInt32(generator, 'generator');
-  else if (generator !== true)
+  } else if (generator !== true) {
     generator = toBuf(generator, genEncoding);
-  else
+  } else {
     throw new ERR_INVALID_ARG_TYPE(
       'generator',
       ['number', 'string', 'ArrayBuffer', 'Buffer', 'TypedArray', 'DataView'],
       generator
     );
+  }
 
 
   this[kHandle] = new _DiffieHellman(sizeOrKey, generator);