bitcoinrb 0.0.1

data/lib/bitcoin/message/error.rb

@@ -0,0 +1,10 @@
+module Bitcoin
+  module Message
+
+    # P2P message layer Error
+    class Error < StandardError
+
+    end
+
+  end
+end

data/lib/bitcoin/message/fee_filter.rb

@@ -0,0 +1,27 @@
+module Bitcoin
+  module Message
+
+    # feefilter message
+    # https://bitcoin.org/en/developer-reference#feefilter
+    class FeeFilter < Base
+
+      COMMAND = 'feefilter'
+
+      # The fee rate (in satoshis per kilobyte)
+      attr_accessor :fee_rate
+
+      def initialize(fee_rate)
+        @fee_rate = fee_rate
+      end
+
+      def self.parse_from_payload(payload)
+        new(payload.unpack('Q').first)
+      end
+
+      def to_payload
+        [fee_rate].pack('Q')
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/filter_add.rb

@@ -0,0 +1,28 @@
+module Bitcoin
+  module Message
+
+    # filteradd message
+    # https://bitcoin.org/en/developer-reference#filteradd
+    class FilterAdd < Base
+
+      COMMAND = 'filteradd'
+
+      # element must be sent in the byte order they would use when appearing in a raw transaction;
+      attr_accessor :element
+
+      def initialize(element)
+        @element = element
+      end
+
+      def self.parse_from_payload(payload)
+        new(Bitcoin.unpack_var_string(payload).first.bth)
+      end
+
+      def to_payload
+        Bitcoin.pack_var_string(element.htb)
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/filter_clear.rb

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

data/lib/bitcoin/message/filter_load.rb

@@ -0,0 +1,43 @@
+module Bitcoin
+  module Message
+
+    # filterload message
+    # https://bitcoin.org/en/developer-reference#filterload
+    class FilterLoad < Base
+
+      COMMAND = 'filterload'
+
+      BLOOM_UPDATE_NONE = 0
+      BLOOM_UPDATE_ALL = 1
+      BLOOM_UPDATE_P2PUBKEY_ONLY = 2
+
+      attr_accessor :filter # bin format
+      attr_accessor :func_count
+      attr_accessor :tweak
+      attr_accessor :flag
+
+      def initialize(filter, func_count, tweak = 0, flag = BLOOM_UPDATE_ALL)
+        @filter = filter
+        @func_count = func_count
+        @tweak = tweak
+        @flag = flag
+      end
+
+      def self.parse_from_payload(payload)
+        buf = StringIO.new(payload)
+        filter_count = Bitcoin.unpack_var_int_from_io(buf)
+        filter = buf.read(filter_count).unpack('C*')
+        func_count = buf.read(4).unpack('V').first
+        tweak = buf.read(4).unpack('V').first
+        flag = buf.read(1).unpack('C').first
+        new(filter, func_count, tweak, flag)
+      end
+
+      def to_payload
+        Bitcoin.pack_var_int(filter.size) << filter.pack('C*') << [func_count, tweak, flag].pack('VVC')
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/get_addr.rb

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

data/lib/bitcoin/message/get_blocks.rb

@@ -0,0 +1,29 @@
+module Bitcoin
+  module Message
+
+    # block message
+    # https://bitcoin.org/en/developer-reference#getblocks
+    class GetBlocks < Base
+      include HeadersParser
+      extend HeadersParser
+
+      COMMAND = 'getblocks'
+
+      # protocol version
+      attr_accessor :version
+
+      # block header hashes
+      attr_accessor :hashes
+
+      attr_accessor :stop_hash
+
+      def initialize(version, hashes, stop_hash = DEFAULT_STOP_HASH)
+        @version = version
+        @hashes = hashes
+        @stop_hash = stop_hash
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/get_data.rb

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

data/lib/bitcoin/message/get_headers.rb

@@ -0,0 +1,28 @@
+module Bitcoin
+  module Message
+
+    # getheaders message
+    # https://bitcoin.org/en/developer-reference#getheaders
+    class GetHeaders < Base
+      include HeadersParser
+      extend HeadersParser
+
+      COMMAND = 'getheaders'
+
+      # protocol version
+      attr_accessor :version
+
+      # block header hashes
+      attr_accessor :hashes
+
+      attr_accessor :stop_hash
+
+      def initialize(version, hashes, stop_hash = DEFAULT_STOP_HASH)
+        @version = version
+        @hashes = hashes
+        @stop_hash = stop_hash
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/handler.rb

@@ -0,0 +1,170 @@
+module Bitcoin
+  module Message
+
+    # Default P2P message handler.
+    class Handler
+
+      attr_reader :logger
+      attr_reader :conn
+
+      def initialize(conn, logger = Bitcoin::Logger.create(:parser))
+        @conn = conn
+        @logger = logger
+        @message = ""
+      end
+
+      # handle p2p message.
+      def handle(message)
+        logger.info "handle message #{message.bth}"
+        begin
+          parse(message)
+        rescue Error => e
+          logger.error("invalid header magic. #{e.message}")
+          conn.close
+        end
+      end
+
+      private
+
+      def parse(message)
+        @message += message
+        command, payload, rest = parse_header
+        return unless command
+
+        handle_command(command, payload)
+        @message = ""
+        parse(rest) if rest && rest.bytesize > 0
+      end
+
+      def parse_header
+        head_magic = Bitcoin.chain_params.magic_head
+        return if @message.nil? || @message.size < HEADER_SIZE
+
+        magic, command, length, checksum = @message.unpack('a4A12Va4')
+        raise Error, "invalid header magic. #{magic.bth}" unless magic.bth == head_magic
+
+        payload = @message[HEADER_SIZE...(HEADER_SIZE + length)]
+        return if payload.size < length
+        raise Error, "header checksum mismatch. #{checksum.bth}" unless Bitcoin.double_sha256(payload)[0...4] == checksum
+
+        rest = @message[(HEADER_SIZE + length)..-1]
+        [command, payload, rest]
+      end
+
+      def handle_command(command, payload)
+        logger.info("process command #{command}. payload = #{payload.bth}")
+        case command
+        when Version::COMMAND
+          on_version(Version.parse_from_payload(payload))
+        when VerAck::COMMAND
+          on_ver_ack
+        when Addr::COMMAND
+          on_addr(Addr.parse_from_payload(payload))
+        when SendHeaders::COMMAND
+          on_send_headers
+        when FeeFilter::COMMAND
+          on_fee_filter(FeeFilter.parse_from_payload(payload))
+        when Ping::COMMAND
+          on_ping(Ping.parse_from_payload(payload))
+        when Pong::COMMAND
+          on_pong(Pong.parse_from_payload(payload))
+        when GetHeaders::COMMAND
+          on_get_headers(GetHeaders.parse_from_payload(payload))
+        when Headers::COMMAND
+          on_headers(Headers.parse_from_payload(payload))
+        when Block::COMMAND
+          on_block(Block.parse_from_payload(payload))
+        when Tx::COMMAND
+          on_tx(Tx.parse_from_payload(payload))
+        when NotFound::COMMAND
+          on_not_found(NotFound.parse_from_payload(payload))
+        when MemPool::COMMAND
+          on_mem_pool
+        when Reject::COMMAND
+          on_reject(Reject.parse_from_payload(payload))
+        when SendCmpct::COMMAND
+          on_send_cmpct(SendCmpct.parse_from_payload(payload))
+        else
+          logger.warn("unsupported command received. #{command}")
+          conn.close("with command #{command}")
+        end
+      end
+
+      def on_version(version)
+        logger.info("receive version message. #{version.to_json}")
+        conn.send_data(VerAck.new.to_pkt)
+      end
+
+      def on_ver_ack
+        logger.info('receive verack message.')
+        conn.handshake_done
+      end
+
+      def on_addr(addr)
+        logger.info("receive addr message. #{addr.to_json}")
+        # TODO
+      end
+
+      def on_send_headers
+        logger.info('receive sendheaders message.')
+        conn.sendheaders = true
+      end
+
+      def on_fee_filter(fee_filter)
+        logger.info("receive feefilter message. #{fee_filter.to_json}")
+        conn.fee_rate = fee_filter.fee_rate
+      end
+
+      def on_ping(ping)
+        logger.info("receive ping message. #{ping.to_json}")
+        conn.send_data(ping.to_response)
+      end
+
+      def on_pong(pong)
+        logger.info("receive pong message. #{pong.to_json}")
+        # TODO calculate response
+      end
+
+      def on_get_headers(headers)
+        logger.info("receive getheaders message. #{headers.to_json}")
+        # TODO
+      end
+
+      def on_headers(headers)
+        logger.info("receive headers message. #{headers.to_json}")
+        # TODO
+      end
+
+      def on_block(block)
+        logger.info("receive block message. #{block.to_json}")
+        # TODO
+      end
+
+      def on_tx(tx)
+        logger.info("receive tx message. #{tx.to_json}")
+        # TODO
+      end
+
+      def on_not_found(not_found)
+        logger.info("receive notfound message. #{not_found.to_json}")
+        # TODO
+      end
+
+      def on_mem_pool
+        logger.info('receive mempool message.')
+        # TODO return mempool tx
+      end
+
+      def on_reject(reject)
+        logger.warn("receive reject message. #{reject.to_json}")
+        # TODO
+      end
+
+      def on_send_cmpct(cmpct)
+        logger.info("receive sendcmpct message. #{cmpct.to_json}")
+        # TODO if mode is high and version is 1, relay block with cmpctblock message
+      end
+
+    end
+  end
+end

data/lib/bitcoin/message/headers.rb

@@ -0,0 +1,34 @@
+module Bitcoin
+  module Message
+
+    # headers message
+    # https://bitcoin.org/en/developer-reference#headers
+    class Headers < Base
+
+      COMMAND = 'headers'
+
+      attr_accessor :headers
+
+      def initialize(headers = [])
+        @headers = headers
+      end
+
+      def self.parse_from_payload(payload)
+        buf = StringIO.new(payload)
+        header_count = Bitcoin.unpack_var_int_from_io(buf)
+        h = new
+        header_count.times do
+          h.headers << Bitcoin::BlockHeader.parse_from_payload(buf.read(80))
+          buf.read(1) # read tx count 0x00 (headers message doesn't include any tx.)
+        end
+        h
+      end
+
+      def to_payload
+        Bitcoin.pack_var_int(headers.size) << headers.map { |h| h.to_payload << 0x00 }.join
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/message/headers_parser.rb

@@ -0,0 +1,24 @@
+module Bitcoin
+  module Message
+
+    # Common message parser which handle multiple block headers as a payload.
+    module HeadersParser
+
+      def parse_from_payload(payload)
+        ver, payload = payload.unpack('Va*')
+        size, payload = Bitcoin.unpack_var_int(payload)
+        hashes = []
+        buf = StringIO.new(payload)
+        size.times do
+          hashes << buf.read(32).reverse.bth
+        end
+        new(ver, hashes, buf.read(32).reverse.bth)
+      end
+
+      def to_payload
+        [version].pack('V') << Bitcoin.pack_var_int(hashes.length) << hashes.map{|h|h.htb.reverse}.join << stop_hash.htb.reverse
+      end
+
+    end
+  end
+end