Merge pull request #163 from dadepo/add-privkey-pubkey-pair-validation

holgerd77 · web-flow · commit 52ff33f8cc93 · 2022-04-21T09:44:08.000+02:00
Added verifyPublicKey to verify publicKey, privateKey pair.
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/index.ts b/src/index.ts
@@ -646,6 +646,15 @@ export default class Wallet {
   public async toV3String(password: string, opts?: Partial<V3Params>): Promise<string> {
     return JSON.stringify(await this.toV3(password, opts))
   }
+
+  /**
+   * Verify the publicKey, privateKey pair
+   *
+   * @param publicKey the public key to verify against the private key of the wallet
+   */
+  public verifyPublicKey(publicKey: Buffer): boolean {
+    return privateToPublic(this.privateKey as Buffer).equals(publicKey)
+  }
 }
 
 // helpers
diff --git a/test/index.spec.ts b/test/index.spec.ts
@@ -84,6 +84,15 @@ describe('.getChecksumAddressString()', function () {
   })
 })
 
+describe('.verifyPublicKey()', function () {
+  it('should return true if publicKey, privateKey pair is valid', function () {
+    assert.strictEqual(fixtureWallet.verifyPublicKey(fixturePublicKeyBuffer), true)
+  })
+  it('should return false if publicKey, privateKey pair is invalid', function () {
+    assert.strictEqual(fixtureWallet.verifyPublicKey(Buffer.alloc(64, 0)), false)
+  })
+})
+
 describe('public key only wallet', function () {
   const pubKey = Buffer.from(fixturePublicKey, 'hex')
   it('.fromPublicKey() should work', function () {
