bitcoinrb 0.0.1

data/lib/bitcoin/message/inv.rb

@@ -0,0 +1,21 @@
+module Bitcoin
+  module Message
+
+    # inv message
+    # https://bitcoin.org/en/developer-reference#inv
+    class Inv < Base
+      include InventoriesParser
+      extend InventoriesParser
+
+      COMMAND = 'inv'
+
+      attr_reader :inventories
+
+      def initialize(inventories = [])
+        @inventories = inventories
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/inventories_parser.rb

@@ -0,0 +1,23 @@
+module Bitcoin
+  module Message
+
+    # Common message parser which only handle multiple inventory as payload.
+    module InventoriesParser
+
+      def parse_from_payload(payload)
+        size, payload = Bitcoin.unpack_var_int(payload)
+        buf = StringIO.new(payload)
+        i = new
+        size.times do
+          i.inventories << Inventory.parse_from_payload(buf.read(36))
+        end
+        i
+      end
+
+      def to_payload
+        Bitcoin.pack_var_int(inventories.length) << inventories.map(&:to_payload).join
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/inventory.rb

@@ -0,0 +1,47 @@
+module Bitcoin
+  module Message
+    # inventory class. inventory is a part of message.
+    # https://bitcoin.org/en/developer-reference#term-inventory
+    class Inventory
+
+      SEGWIT_FLAG = 1 << 30
+
+      MSG_TX = 1
+      MSG_BLOCK = 2
+      MSG_FILTERED_BLOCK = 3
+      MSG_CMPCT_BLOCK = 4
+      MSG_WITNESS_TX = SEGWIT_FLAG | MSG_TX
+      MSG_WITNESS_BLOCK = SEGWIT_FLAG | MSG_BLOCK
+      MSG_FILTERED_WITNESS_BLOCK = SEGWIT_FLAG | MSG_FILTERED_BLOCK
+
+      attr_accessor :identifier
+      attr_accessor :hash
+
+      def initialize(identifier, hash)
+        raise Error, "invalid type identifier specified. identifier = #{identifier}" unless valid_identifier?(identifier)
+        @identifier = identifier
+        @hash = hash
+      end
+
+      # parse inventory payload
+      def self.parse_from_payload(payload)
+        raise Error, 'invalid inventory size.' if payload.bytesize != 36
+        identifier = payload[0..4].unpack('V').first
+        hash = payload[4..-1].reverse.bth # internal byte order
+        new(identifier, hash)
+      end
+
+      def to_payload
+        [identifier].pack('V') << hash.htb.reverse
+      end
+
+      private
+
+      def valid_identifier?(identifier)
+        [MSG_TX, MSG_BLOCK, MSG_FILTERED_BLOCK, MSG_CMPCT_BLOCK, MSG_WITNESS_TX,
+         MSG_WITNESS_BLOCK, MSG_FILTERED_WITNESS_BLOCK].include?(identifier)
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/mem_pool.rb

@@ -0,0 +1,17 @@
+module Bitcoin
+  module Message
+
+    # mempool message
+    # https://bitcoin.org/en/developer-reference#mempool
+    class MemPool < Base
+
+      COMMAND = 'mempool'
+
+      def to_payload
+        ''
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/merkle_block.rb

@@ -0,0 +1,42 @@
+module Bitcoin
+  module Message
+
+    # merckleblock message
+    # https://bitcoin.org/en/developer-reference#merkleblock
+    class MerkleBlock < Base
+
+      COMMAND = 'merkleblock'
+
+      attr_accessor :header
+      attr_accessor :tx_count
+      attr_accessor :hashes
+      attr_accessor :flags
+
+      def initialize
+        @hashes = []
+      end
+
+      def self.parse_from_payload(payload)
+        m = new
+        buf = StringIO.new(payload)
+        m.header = Bitcoin::BlockHeader.parse_from_payload(buf.read(80))
+        m.tx_count = buf.read(4).unpack('V').first
+        hash_count = Bitcoin.unpack_var_int_from_io(buf)
+        hash_count.times do
+          m.hashes << buf.read(32).reverse.bth
+        end
+        flag_count = Bitcoin.unpack_var_int_from_io(buf)
+        # A sequence of bits packed eight in a byte with the least significant bit first.
+        m.flags = buf.read(flag_count).bth
+        m
+      end
+
+      def to_payload
+        header.to_payload << [tx_count].pack('V') << Bitcoin.pack_var_int(hashes.size) <<
+            hashes.map{|h|h.htb.reverse}.join << Bitcoin.pack_var_int(flags.htb.bytesize) << flags.htb
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/not_found.rb

@@ -0,0 +1,29 @@
+module Bitcoin
+  module Message
+
+    # notfound message
+    # https://bitcoin.org/en/developer-reference#notfound
+    class NotFound < Base
+
+      attr_accessor :inventory
+
+      COMMAND = 'notfound'
+
+      def initialize(identifier, hash)
+        @inventory = Inventory.new(identifier, hash)
+      end
+
+      def self.parse_from_payload(payload)
+        size, inventory_payload = Bitcoin.unpack_var_int(payload)
+        inventory = Inventory.parse_from_payload(inventory_payload)
+        new(inventory.identifier, inventory.hash)
+      end
+
+      def to_payload
+        Bitcoin.pack_var_int(1) << inventory.to_payload
+      end
+
+    end
+  end
+
+end

data/lib/bitcoin/message/ping.rb

@@ -0,0 +1,30 @@
+module Bitcoin
+  module Message
+
+    # ping message class
+    # https://bitcoin.org/en/developer-reference#ping
+    class Ping < Base
+
+      COMMAND = 'ping'
+
+      attr_accessor :nonce
+
+      def initialize(nonce = SecureRandom.random_number(0xffffffff))
+        @nonce = nonce
+      end
+
+      def self.parse_from_payload(payload)
+        new(payload.unpack('Q').first)
+      end
+
+      def to_payload
+        [nonce].pack('Q')
+      end
+
+      def to_response
+        Pong.new(nonce)
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/pong.rb

@@ -0,0 +1,26 @@
+module Bitcoin
+  module Message
+
+    # pong message
+    # https://bitcoin.org/en/developer-reference#pong
+    class Pong < Base
+
+      COMMAND = 'pong'
+
+      attr_reader :nonce
+
+      def initialize(nonce)
+        @nonce = nonce
+      end
+
+      def self.parse_from_payload(payload)
+        new(payload.unpack('Q').first)
+      end
+
+      def to_payload
+        [nonce].pack('Q')
+      end
+    end
+
+  end
+end

data/lib/bitcoin/message/reject.rb

@@ -0,0 +1,46 @@
+module Bitcoin
+  module Message
+
+    # reject message
+    # https://bitcoin.org/en/developer-reference#reject
+    class Reject < Base
+
+      attr_accessor :message
+      attr_accessor :code
+      attr_accessor :reason
+      attr_accessor :extra
+
+      COMMAND = 'reject'
+
+      CODE_MALFORMED = 0x01
+      CODE_INVALID = 0x10
+      CODE_OBSOLETE = 0x11
+      CODE_DUPLICATE = 0x12
+      CODE_NONSTANDARD = 0x40
+      CODE_DUST = 0x41
+      CODE_INSUFFICIENT_FEE = 0x42
+      CODE_CHECKPOINT = 0x43
+
+      def initialize(message, code, reason = '', extra = '')
+        @message = message
+        @code = code
+        @reason = reason
+        @extra = extra
+      end
+
+      def self.parse_from_payload(payload)
+        message, payload = Bitcoin.unpack_var_string(payload)
+        code, payload = payload.unpack('Ca*')
+        reason, payload = Bitcoin.unpack_var_string(payload)
+        extra = ['tx', 'block'].include?(message) ? payload.reverse.bth : payload
+        new(message, code, reason, extra)
+      end
+
+      def to_payload
+        e = ['tx', 'block'].include?(message) ? extra.htb.reverse : extra
+        Bitcoin.pack_var_string(message) << [code].pack('C') << Bitcoin.pack_var_string(reason) << e
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/send_cmpct.rb

@@ -0,0 +1,43 @@
+module Bitcoin
+  module Message
+
+    # sendcmpct message
+    # https://github.com/bitcoin/bips/blob/master/bip-0152.mediawiki
+    class SendCmpct < Base
+
+      COMMAND = 'sendcmpct'
+
+      MODE_HIGH = 1
+      MODE_LOW = 0
+
+      attr_accessor :mode
+      attr_accessor :version
+      # TODO support version 2
+
+      def initialize(mode, version)
+        @mode = mode
+        @version = version
+      end
+
+      def self.parse_from_payload(payload)
+        buf = StringIO.new(payload)
+        mode = buf.read(1).unpack('c').first
+        version = buf.read(8).unpack('Q').first
+        new(mode, version)
+      end
+
+      def to_payload
+        [mode, version].pack('cQ')
+      end
+
+      def high?
+        mode == 1
+      end
+
+      def low?
+        mode.zero?
+      end
+    end
+
+  end
+end

data/lib/bitcoin/message/send_headers.rb

@@ -0,0 +1,16 @@
+module Bitcoin
+  module Message
+
+    # sendheaders message
+    # https://bitcoin.org/en/developer-reference#sendheaders
+    class SendHeaders < Base
+
+      COMMAND = 'sendheaders'
+
+      def to_payload
+        ''
+      end
+    end
+
+  end
+end

data/lib/bitcoin/message/tx.rb

@@ -0,0 +1,30 @@
+module Bitcoin
+  module Message
+
+    # tx message
+    # https://bitcoin.org/en/developer-reference#tx
+    class Tx < Base
+
+      COMMAND = 'tx'
+
+      attr_accessor :tx
+      attr_accessor :use_segwit
+
+      def initialize(tx, use_segwit = false)
+        @tx = tx
+        @use_segwit = use_segwit
+      end
+
+      def self.parse_from_payload(payload)
+        tx = Bitcoin::Tx.parse_from_payload(payload)
+        new(tx, tx.witness?)
+      end
+
+      def to_payload
+        use_segwit ? tx.to_payload : tx.serialize_old_format
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/ver_ack.rb

@@ -0,0 +1,17 @@
+module Bitcoin
+  module Message
+
+    # verack message
+    # https://bitcoin.org/en/developer-reference#verack
+    class VerAck < Base
+
+      COMMAND = 'verack'
+
+      def to_payload
+        ''
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/version.rb

@@ -0,0 +1,79 @@
+# encoding: ascii-8bit
+module Bitcoin
+  module Message
+
+    # https://bitcoin.org/en/developer-reference#version
+    class Version < Base
+
+      COMMAND = 'version'
+
+      attr_accessor :version
+      attr_accessor :services
+      attr_accessor :timestamp
+      attr_accessor :local_addr
+      attr_accessor :remote_addr
+      attr_accessor :nonce
+      attr_accessor :user_agent
+      attr_accessor :start_height
+      attr_accessor :relay
+
+      def initialize(opts = {})
+        @version = Bitcoin.chain_params.protocol_version
+        @services = Bitcoin::Message::SERVICE_FLAGS[:network]
+        @timestamp = Time.now.to_i
+        @local_addr = "127.0.0.1:#{Bitcoin.chain_params.default_port}"
+        @remote_addr = "127.0.0.1:#{Bitcoin.chain_params.default_port}"
+        @nonce = SecureRandom.random_number(0xffffffffffffffff)
+        @user_agent = Bitcoin::Message::USER_AGENT
+        @start_height = 0
+        @relay = true
+        opts.each { |k, v| send "#{k}=", v }
+      end
+
+      def self.parse_from_payload(payload)
+        version, services, timestamp, remote_addr, local_addr, nonce, rest = payload.unpack('VQQa26a26Qa*')
+        v = new
+        v.version = version
+        v.services = services
+        v.timestamp = timestamp
+        v.remote_addr = v.unpack_addr(remote_addr)
+        v.local_addr = v.unpack_addr(local_addr)
+        v.nonce = nonce
+        user_agent, rest = unpack_var_string(rest)
+        start_height, rest = rest.unpack('Va*')
+        v.user_agent = user_agent
+        v.start_height = start_height
+        v.relay = v.unpack_relay_field(rest).first
+        v
+      end
+
+      def to_payload
+        [
+          [version, services, timestamp].pack('VQQ'),
+          pack_addr(local_addr),
+          pack_addr(remote_addr),
+          [nonce].pack('Q'),
+          pack_var_string(user_agent),
+          [start_height].pack('V'),
+          pack_boolean(relay)
+        ].join
+      end
+
+      def pack_addr(addr)
+        host, port = addr.split(':')
+        sockaddr = Socket.pack_sockaddr_in(port.to_i, host)
+        [[1].pack('Q'), "\x00" * 10, "\xFF\xFF", sockaddr[4...8], sockaddr[2...4]].join
+      end
+
+      def unpack_addr(addr)
+        host, port = addr.unpack('x8x12a4n')
+        "#{host.unpack('C*').join('.')}:#{port}"
+      end
+
+      def unpack_relay_field(payload)
+        ( version >= 70001 && payload ) ? unpack_boolean(payload) : [ true, nil ]
+      end
+
+    end
+  end
+end