Added Buffer conversion for entropy in Thirdparty API fromKryptoKit(), some Buffer type assertions

Author: holgerd77

src/index.ts

@@ -255,7 +255,7 @@ export default class Wallet {
     if (icapDirect) {
       const max = new ethUtil.BN('088f924eeceeda7fe92e1f5b0fffffffffffffff', 16)
       while (true) {
-        const privateKey = randomBytes(32)
+        const privateKey = randomBytes(32) as Buffer
         if (new ethUtil.BN(ethUtil.privateToAddress(privateKey)).lte(max)) {
           return new Wallet(privateKey)
         }
@@ -274,7 +274,7 @@ export default class Wallet {
     }
 
     while (true) {
-      const privateKey = randomBytes(32)
+      const privateKey = randomBytes(32) as Buffer
       const address = ethUtil.privateToAddress(privateKey)
 
       if (pattern.test(address.toString('hex'))) {
@@ -352,7 +352,7 @@ export default class Wallet {
       kdfparams.R,
       kdfparams.P,
       kdfparams.DkLen,
-    )
+    ) as Buffer
 
     const ciphertext = Buffer.from(json.Crypto.CipherText, 'hex')
     const mac = ethUtil.keccak256(Buffer.concat([derivedKey.slice(16, 32), ciphertext]))
@@ -399,7 +399,7 @@ export default class Wallet {
         kdfparams.r,
         kdfparams.p,
         kdfparams.dklen,
-      )
+      ) as Buffer
     } else if (json.crypto.kdf === 'pbkdf2') {
       kdfparams = json.crypto.kdfparams
 
@@ -572,7 +572,7 @@ export default class Wallet {
           kdfParams.r,
           kdfParams.p,
           kdfParams.dklen,
-        )
+        ) as Buffer
         break
       default:
         throw new Error('Unsupported kdf')

src/thirdparty.ts

@@ -1,5 +1,5 @@
 import * as crypto from 'crypto'
-import * as ethUtil from 'ethereumjs-util'
+import { keccak256, sha256, toBuffer } from 'ethereumjs-util'
 
 import Wallet from './index'
 
@@ -167,7 +167,7 @@ export function fromEtherWallet(input: string | EtherWalletOptions, password: st
  * Third Party API: Import a brain wallet used by Ether.Camp
  */
 export function fromEtherCamp(passphrase: string): Wallet {
-  return new Wallet(ethUtil.keccak256(Buffer.from(passphrase)))
+  return new Wallet(keccak256(Buffer.from(passphrase)))
 }
 
 /**
@@ -210,13 +210,13 @@ export function fromKryptoKit(entropy: string, password: string): Wallet {
 
   let privateKey: Buffer
   if (type === 'd') {
-    privateKey = ethUtil.sha256(entropy)
+    privateKey = sha256(toBuffer(entropy))
   } else if (type === 'q') {
     if (typeof password !== 'string') {
       throw new Error('Password required')
     }
 
-    const encryptedSeed = ethUtil.sha256(Buffer.from(entropy.slice(0, 30)))
+    const encryptedSeed = sha256(Buffer.from(entropy.slice(0, 30)))
     const checksum = entropy.slice(30, 46)
 
     const salt = kryptoKitBrokenScryptSeed(encryptedSeed)
@@ -244,8 +244,7 @@ export function fromKryptoKit(entropy: string, password: string): Wallet {
     if (checksum.length > 0) {
       if (
         checksum !==
-        ethUtil
-          .sha256(ethUtil.sha256(privateKey))
+        sha256(sha256(privateKey))
           .slice(0, 8)
           .toString('hex')
       ) {