fix(WalletCrypto): pass bit len of blocks to padding function

blockchain · Feb 5, 2016 · 3cd453e · 3cd453e
1 parent 1b97b5c
commit 3cd453e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/wallet-crypto.js b/src/wallet-crypto.js
@@ -6,7 +6,8 @@ var crypto  = require('crypto')
 
 var SUPPORTED_ENCRYPTION_VERSION = 3
   , SALT_BYTES = 16
-  , KEY_BIT_LEN = 256;
+  , KEY_BIT_LEN = 256
+  , BLOCK_BIT_LEN = 128;
 
 var ALGO = {
   SHA1    : 'sha1',
@@ -101,7 +102,7 @@ var AES = {
     var cipher = crypto.createCipheriv(options.mode || AES.CBC, key, salt || '');
     cipher.setAutoPadding(!options.padding);
 
-    if (options.padding) dataBytes = options.padding.pad(dataBytes, KEY_BIT_LEN / 8);
+    if (options.padding) dataBytes = options.padding.pad(dataBytes, BLOCK_BIT_LEN / 8);
     var encryptedBytes = Buffer.concat([ cipher.update(dataBytes), cipher.final() ]);
 
     return encryptedBytes;