Add valid_pubkey?

Band1to · Nov 11, 2014 · b65e748 · b65e748
1 parent 0e58a0b
commit b65e748
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/bitcoin.rb b/lib/bitcoin.rb
@@ -93,6 +93,14 @@ def valid_address?(address)
       address_checksum?(address)
     end
 
+    # check if given +pubkey+ is valid.
+    def valid_pubkey?(pubkey)
+      ::OpenSSL::PKey::EC::Point.from_hex(bitcoin_elliptic_curve.group, pubkey)
+      true
+    rescue OpenSSL::PKey::EC::Point::Error
+      false
+    end
+
     # get hash160 for given +address+. returns nil if address is invalid.
     def hash160_from_address(address)
       return nil  unless valid_address?(address)

diff --git a/spec/bitcoin/bitcoin_spec.rb b/spec/bitcoin/bitcoin_spec.rb
@@ -163,6 +163,14 @@
     success.should == true
   end
 
+  it 'validate bitcoin public key' do
+    key = Bitcoin::Key.generate
+    Bitcoin.valid_pubkey?(key.pub_compressed).should == true
+    Bitcoin.valid_pubkey?(key.pub_uncompressed).should == true
+    Bitcoin.valid_pubkey?(key.addr).should == false
+    Bitcoin.valid_pubkey?(key.priv).should == false
+  end
+
   it 'validate p2sh address' do
     Bitcoin.network = :testnet
     Bitcoin.valid_address?("2MyLngQnhzjzatKsB7XfHYoP9e2XUXSiBMM").should == true