bitcoinrb 0.0.1

data/lib/bitcoin/block_header.rb

@@ -0,0 +1,41 @@
+module Bitcoin
+
+  # Block Header
+  class BlockHeader
+
+    attr_accessor :hash
+    attr_accessor :version
+    attr_accessor :prev_hash
+    attr_accessor :merkle_root
+    attr_accessor :time
+    attr_accessor :bits
+    attr_accessor :nonce
+
+    def initialize(version, prev_hash, merkle_root, time, bits, nonce)
+      @version = version
+      @prev_hash = prev_hash
+      @merkle_root = merkle_root
+      @time = time
+      @bits = bits
+      @nonce = nonce
+      @hash = calc_hash
+    end
+
+    def self.parse_from_payload(payload)
+      version, prev_hash, merkle_root, time, bits, nonce = payload.unpack('Va32a32VVV')
+      new(version, prev_hash.reverse.bth, merkle_root.reverse.bth, time, bits, nonce)
+    end
+
+    def to_payload
+      [version, prev_hash.htb.reverse, merkle_root.htb.reverse, time, bits, nonce].pack('Va32a32VVV')
+    end
+
+    private
+
+    def calc_hash
+      Bitcoin.double_sha256(to_payload).reverse.bth
+    end
+
+  end
+
+end

data/lib/bitcoin/chain_params.rb

@@ -0,0 +1,57 @@
+require 'yaml'
+
+module Bitcoin
+
+  # Network parameter class
+  class ChainParams
+
+    attr_reader :network
+    attr_reader :magic_head
+    attr_reader :message_magic
+    attr_reader :address_version
+    attr_reader :p2sh_version
+    attr_reader :bech32_hrp
+    attr_reader :privkey_version
+    attr_reader :extended_privkey_version
+    attr_reader :extended_pubkey_version
+    attr_reader :default_port
+    attr_reader :protocol_version
+    attr_reader :retarget_interval
+    attr_reader :retarget_time
+    attr_reader :target_spacing
+    attr_reader :max_money
+    attr_reader :bip34_height
+    attr_reader :genesis_hash
+    attr_reader :proof_of_work_limit
+    attr_reader :dns_seeds
+
+    # mainnet params
+    def self.mainnet
+      YAML.load(File.open("#{__dir__}/chainparams/mainnet.yml"))
+    end
+
+    # testnet params
+    def self.testnet
+      YAML.load(File.open("#{__dir__}/chainparams/testnet.yml"))
+    end
+
+    # regtest params
+    def self.regtest
+      YAML.load(File.open("#{__dir__}/chainparams/regtest.yml"))
+    end
+
+    def mainnet?
+      network == 'mainnet'
+    end
+
+    def testnet?
+      network == 'testnet'
+    end
+
+    def regtest?
+      network == 'regtest'
+    end
+
+  end
+
+end

data/lib/bitcoin/chainparams/mainnet.yml

@@ -0,0 +1,25 @@
+--- !ruby/object:Bitcoin::ChainParams
+network: "mainnet"
+magic_head: "f9beb4d9"
+message_magic: "Bitcoin Signed Message:\n"
+address_version: "00"
+p2sh_version: "05"
+bech32_hrp: 'bc'
+privkey_version: "80"
+extended_privkey_version: "0488ade4"
+extended_pubkey_version: "0488b21e"
+default_port: 8333
+protocol_version: 70013
+retarget_interval: 2016
+retarget_time: 1209600 # 2 weeks
+target_spacing: 600 # block interval
+max_money: 21000000
+bip34_height: 227931
+genesis_hash: "000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f"
+proof_of_work_limit: 0x1d00ffff
+dns_seeds:
+  - "seed.bitcoin.sipa.be"
+  - "dnsseed.bluematt.me"
+  - "dnsseed.bitcoin.dashjr.org"
+  - "seed.bitcoinstats.com"
+  - "seed.bitcoin.jonasschnelli.ch"

data/lib/bitcoin/chainparams/regtest.yml

@@ -0,0 +1,20 @@
+--- !ruby/object:Bitcoin::ChainParams
+network: "regtest"
+magic_head: "fabfb5da"
+message_magic: "Bitcoin Signed Message:\n"
+address_version: "6f"
+p2sh_version: "c4"
+bech32_hrp: 'tb'
+privkey_version: "ef"
+extended_privkey_version: "04358394"
+extended_pubkey_version: "043587cf"
+default_port: 18444
+protocol_version: 70013
+retarget_interval: 2016
+retarget_time: 1209600 # 2 weeks
+target_spacing: 600 # block interval
+max_money: 21000000
+bip34_height: 0
+genesis_hash: "0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206"
+proof_of_work_limit: 0x207fffff
+dns_seeds:

data/lib/bitcoin/chainparams/testnet.yml

@@ -0,0 +1,24 @@
+--- !ruby/object:Bitcoin::ChainParams
+network: "testnet"
+magic_head: "0b110907"
+message_magic: "Bitcoin Signed Message:\n"
+address_version: "6f"
+p2sh_version: "c4"
+bech32_hrp: 'tb'
+privkey_version: "ef"
+extended_privkey_version: "04358394"
+extended_pubkey_version: "043587cf"
+default_port: 18333
+protocol_version: 70013
+retarget_interval: 2016
+retarget_time: 1209600 # 2 weeks
+target_spacing: 600 # block interval
+max_money: 21000000
+bip34_height: 227931
+genesis_hash: "000000000933ea01ad0ee984209779baaec3ced90fa3f408719526f8d77f4943"
+proof_of_work_limit: 0x1d00ffff
+dns_seeds:
+  - "testnet-seed.bitcoin.jonasschnelli.ch"
+  - "seed.tbtc.petertodd.org"
+  - "testnet-seed.bluematt.me"
+  - "testnet-seed.bitcoin.schildbach.de"

data/lib/bitcoin/connection.rb

@@ -0,0 +1,66 @@
+module Bitcoin
+
+  # Basic Bitcoin P2P connection class
+  class Connection < EM::Connection
+
+    attr_reader :host, :port, :handler, :logger
+    attr_accessor :connected
+
+    # if true, this peer send new block announcements using a headers message rather than an inv message.
+    attr_accessor :sendheaders
+
+    # minimum fee(in satoshis per kilobyte) for relay tx
+    attr_accessor :fee_rate
+
+    def initialize(host, port)
+      @host = host
+      @port = port
+      @logger = Bitcoin::Logger.create(:connection)
+      @handler = Message::Handler.new(self, @logger)
+      @connected = false
+      @sendheaders = false
+      @attr_accessor = 0
+    end
+
+    def post_init
+      logger.info "connected. #{remote_node}"
+      begin_handshake
+    end
+
+    # handle receiving data from remote node.
+    def receive_data(data)
+      logger.info "receive data from #{remote_node}, data : #{data}"
+      handler.handle(data)
+    end
+
+    # close network connection.
+    def close(msg = '')
+      logger.info "close connection with #{remote_node}. #{msg}"
+      close_connection_after_writing
+      EM.stop
+    end
+
+    def handshake_done
+      logger.info 'handshake finished.'
+      @connected = true
+      # send_data(Message::SendCmpct.new(0, 1))
+
+    end
+
+    private
+
+    def remote_node
+      "#{host}:#{port}"
+    end
+
+    # start handshake
+    def begin_handshake
+      logger.info "begin handshake with #{remote_node}"
+      ver = Bitcoin::Message::Version.new(remote_addr: remote_node, start_height: 1150660)
+      logger.info "send version message. #{ver.to_json}"
+      send_data(ver.to_pkt)
+    end
+
+  end
+
+end

data/lib/bitcoin/ext_key.rb

@@ -0,0 +1,205 @@
+module Bitcoin
+
+  def self.hmac_sha512(key, data)
+    OpenSSL::HMAC.digest(OpenSSL::Digest.new('SHA512'), key, data)
+  end
+
+  # Integers modulo the order of the curve(secp256k1)
+  CURVE_ORDER = 115792089237316195423570985008687907852837564279074904382605163141518161494337
+
+  # BIP32 Extended private key
+  class ExtKey
+
+    attr_accessor :depth
+    attr_accessor :number
+    attr_accessor :chain_code
+    attr_accessor :key
+    attr_accessor :parent_fingerprint
+
+    # generate master key from seed.
+    def self.generate_master(seed)
+      ext_key = ExtKey.new
+      ext_key.depth = ext_key.number = 0
+      ext_key.parent_fingerprint = '00000000'
+      l = Bitcoin.hmac_sha512('Bitcoin seed', seed)
+      left = l[0..31].bth.to_i(16)
+      raise 'invalid key' if left >= CURVE_ORDER || left == 0
+      ext_key.key = Bitcoin::Key.new(priv_key: l[0..31].bth)
+      ext_key.chain_code = l[32..-1]
+      ext_key
+    end
+
+    # get ExtPubkey from priv_key
+    def ext_pubkey
+      k = ExtPubkey.new
+      k.depth = depth
+      k.number = number
+      k.parent_fingerprint = parent_fingerprint
+      k.chain_code = chain_code
+      k.pubkey = key.pubkey
+      k
+    end
+
+    # serialize extended private key
+    def to_payload
+      Bitcoin.chain_params.extended_privkey_version.htb << [depth].pack('C') <<
+          parent_fingerprint.htb << [number].pack('N') << chain_code << [0x00].pack('C') << key.priv_key.htb
+    end
+
+    # Base58 encoded extended private key
+    def to_base58
+      h = to_payload.bth
+      hex = h + Bitcoin.calc_checksum(h)
+      Base58.encode(hex)
+    end
+
+    # get private key(hex)
+    def priv
+      key.priv_key
+    end
+
+    # get public key(hex)
+    def pub
+      key.pubkey
+    end
+
+    # get address
+    def addr
+      key.to_p2pkh
+    end
+
+    # get segwit p2wpkh address
+    def segwit_addr
+      ext_pubkey.segwit_addr
+    end
+
+    # get key identifier
+    def identifier
+      Bitcoin.hash160(key.pubkey)
+    end
+
+    # get fingerprint
+    def fingerprint
+      identifier.slice(0..7)
+    end
+
+    # derive new key
+    def derive(number)
+      new_key = ExtKey.new
+      new_key.depth = depth + 1
+      new_key.number = number
+      new_key.parent_fingerprint = fingerprint
+      if number > (2**31 -1)
+        data = [0x00].pack('C') << key.priv_key.htb << [number].pack('N')
+      else
+        data = key.pubkey.htb << [number].pack('N')
+      end
+      l = Bitcoin.hmac_sha512(chain_code, data)
+      left = l[0..31].bth.to_i(16)
+      raise 'invalid key' if left >= CURVE_ORDER
+      child_priv = (left + key.priv_key.to_i(16)) % CURVE_ORDER
+      raise 'invalid key ' if child_priv >= CURVE_ORDER
+      new_key.key = Bitcoin::Key.new(priv_key: child_priv.to_s(16).rjust(64, '0'))
+      new_key.chain_code = l[32..-1]
+      new_key
+    end
+
+    # import private key from Base58 private key address
+    def self.from_base58(address)
+      data = StringIO.new(Base58.decode(address).htb)
+      ext_key = ExtKey.new
+      data.read(4).bth # version
+      ext_key.depth = data.read(1).unpack('C').first
+      ext_key.parent_fingerprint = data.read(4).bth
+      ext_key.number = data.read(4).unpack('N').first
+      ext_key.chain_code = data.read(32)
+      data.read(1) # 0x00
+      ext_key.key = Bitcoin::Key.new(priv_key: data.read(32).bth)
+      ext_key
+    end
+
+  end
+
+  # BIP-32 Extended public key
+  class ExtPubkey
+    attr_accessor :depth
+    attr_accessor :number
+    attr_accessor :chain_code
+    attr_accessor :pubkey # hex format
+    attr_accessor :parent_fingerprint
+
+    # serialize extended pubkey
+    def to_payload
+      Bitcoin.chain_params.extended_pubkey_version.htb << [depth].pack('C') <<
+          parent_fingerprint.htb << [number].pack('N') << chain_code << pub.htb
+    end
+
+    def pub
+      pubkey
+    end
+
+    # get address
+    def addr
+      Bitcoin::Key.new(pubkey: pubkey).to_p2pkh
+    end
+
+    # get segwit p2wpkh address
+    def segwit_addr
+      hash160 = Bitcoin.hash160(pub)
+      p2wpkh = [ ["00", "14", hash160].join ].pack("H*").bth
+      segwit_addr = Bech32::SegwitAddr.new
+      segwit_addr.hrp = Bitcoin.chain_params.address_version == '00' ? 'bc' : 'tb'
+      segwit_addr.script_pubkey = p2wpkh
+      segwit_addr.addr
+    end
+
+    # get key identifier
+    def identifier
+      Bitcoin.hash160(pub)
+    end
+
+    # get fingerprint
+    def fingerprint
+      identifier.slice(0..7)
+    end
+
+    # Base58 encoded extended pubkey
+    def to_base58
+      h = to_payload.bth
+      hex = h + Bitcoin.calc_checksum(h)
+      Base58.encode(hex)
+    end
+
+    # derive child key
+    def derive(number)
+      new_key = ExtPubkey.new
+      new_key.depth = depth + 1
+      new_key.number = number
+      new_key.parent_fingerprint = fingerprint
+      raise 'hardened key is not support' if number > (2**31 -1)
+      data = pub.htb << [number].pack('N')
+      l = Bitcoin.hmac_sha512(chain_code, data)
+      left = l[0..31].bth.to_i(16)
+      raise 'invalid key' if left >= CURVE_ORDER
+      p1 = Bitcoin::Secp256k1::GROUP.generator.multiply_by_scalar(left)
+      p2 = Bitcoin::Key.new(pubkey: pubkey).to_point
+      new_key.pubkey = ECDSA::Format::PointOctetString.encode(p1 + p2, compression: true).bth
+      new_key.chain_code = l[32..-1]
+      new_key
+    end
+
+    # import pub key from Base58 private key address
+    def self.from_base58(address)
+      data = StringIO.new(Base58.decode(address).htb)
+      ext_pubkey = ExtPubkey.new
+      data.read(4).bth # version
+      ext_pubkey.depth = data.read(1).unpack('C').first
+      ext_pubkey.parent_fingerprint = data.read(4).bth
+      ext_pubkey.number = data.read(4).unpack('N').first
+      ext_pubkey.chain_code = data.read(32)
+      ext_pubkey.pubkey = data.read(33).bth
+      ext_pubkey
+    end
+  end
+
+end