cryptocoin 0.0.1b

data/lib/cryptocoin/protocol/message/getheaders.rb

@@ -0,0 +1,44 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Getheaders
+        attr_reader :digest_count
+        
+        def parse_from_raw(payload)
+          block_locator_digests, i = [], 0
+          version = payload[0..3]
+          digest_count = Cryptocoin::Protocol::VarLenInt.parse_from_raw(payload[4..-1])
+          digest_count.times do
+            c = digest_count.raw.bytesize+i*32
+            block_locator_digests.push(payload[c..c+31])
+            i+=1
+          end
+          digest_stop = payload[5+i*32..-1]
+        end
+        
+        def initialize(version, digest_count, block_locator_digests, digest_stop)
+          @version_raw = version
+          @digest_count = digest_count
+          @block_locator_digests_raw = block_locator_digests
+          @digest_stop_raw = digest_stop
+        end
+        
+        def version
+          @version ||= @version_raw.unpack('L')[0]
+        end
+        
+        def block_locator_digests
+          @block_locator_digests ||= @block_locator_digests_raw.map{|e| e.unpack('H*')[0] }
+        end
+        
+        def digest_stop
+          @digest_stop ||= @digest_stop_raw.unpack('H*')[0]
+        end
+        
+        def raw
+          @version_raw + @digest_count.raw + @block_locator_digests_raw.join + @digest_stop_raw
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/headers.rb

@@ -0,0 +1,30 @@
+require 'cryptocoin/protocol/block_header'
+
+module Cryptocoin
+  module Protocol
+    class Message
+      class Headers
+        attr_reader :count, :headers
+        def self.parse_from_raw(payload)
+          headers, i = [], 0
+          count = Cryptocoin::Protocol::VarLenInt.parse_from_raw(payload)
+          count.times do
+            c = count.raw.bytesize + i*81
+            headers.push(Cryptocoin::Protocol::BlockHeader.parse_from_raw(payload[c..c+80]))
+            i+=1
+          end
+          self.new(count, headers)
+        end
+        
+        def initialize(count, headers)
+          @count = count
+          @headers = headers
+        end
+        
+        def raw
+          @count.raw + @headers.map{|e| e.raw}.join
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/inv.rb

@@ -0,0 +1,31 @@
+require 'cryptocoin/protocol/inventory_vector'
+require 'cryptocoin/protocol/var_len_int'
+
+module Cryptocoin
+  module Protocol
+    class Message
+      class Inv
+        def self.parse_from_raw(payload)
+          inventory = []
+          i = 0
+          count = Cryptocoin::Protocol::VarLenInt.parse_from_raw(payload)
+          count.times do
+            c = count.raw.bytesize+i*36
+            inventory.push(Cryptocoin::Protocol::InventoryVector.parse_from_raw(payload[c..c+35]))
+            i+=1
+          end
+          self.new(count, inventory)
+        end
+        
+        def initialize(count, inventory)
+          @count_raw = count
+          @inventory = inventory
+        end
+        
+        def raw
+          @count_raw + @inventory.map{|e| e.raw}.join
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/mempool.rb

@@ -0,0 +1,16 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Mempool
+        def self.parse_from_raw(payload)
+          # https://en.bitcoin.it/wiki/Protocol_specification#mempool
+          # No data is sent with this message
+          self.new
+        end        
+        def raw
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/notfound.rb

@@ -0,0 +1,31 @@
+require 'cryptocoin/protocol/inventory_vector'
+require 'cryptocoin/protocol/var_len_int'
+
+module Cryptocoin
+  module Protocol
+    class Message
+      class Notfound
+        def self.parse_from_raw(payload)
+          inventory = []
+          i = 0
+          count = Cryptocoin::Protocol::VarLenInt.parse_from_raw(payload)
+          count.times do
+            c = count.raw.bytesize+i*36
+            inventory.push(Cryptocoin::Protocol::InventoryVector.parse_from_raw(payload[c..c+35]))
+            i+=1
+          end
+          self.new(count, inventory)
+        end
+        
+        def initialize(count, inventory)
+          @count_raw = count
+          @inventory = inventory
+        end
+        
+        def raw
+          @count_raw + @inventory.map{|e| e.raw}.join
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/ping.rb

@@ -0,0 +1,24 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Ping
+        def self.parse_from_raw(payload)
+          nonce = payload
+          self.new(nonce)
+        end
+        
+        def initialize(nonce)
+          @nonce_raw = nonce
+        end
+        
+        def nonce
+          @nonce ||= @nonce_raw.unpack('Q')[0]
+        end
+        
+        def raw
+          @nonce_raw
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/pong.rb

@@ -0,0 +1,24 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Pong
+        def self.parse_from_raw(payload)
+          nonce = payload
+          self.new(nonce)
+        end
+        
+        def initialize(nonce)
+          @nonce_raw = nonce
+        end
+        
+        def nonce
+          @nonce ||= @nonce_raw.unpack('Q')[0]
+        end
+        
+        def raw
+          @nonce_raw
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/reject.rb

@@ -0,0 +1,54 @@
+require 'cryptocoin/protocol/var_len_str'
+
+module Cryptocoin
+  module Protocol
+    class Message
+      class Reject
+        attr_reader :message, :reason
+        
+        def self.parse_from_raw(payload)
+          message = Cryptocoin::Protocol::VarLenStr.parse_from_raw(payload)
+          c = message.raw.bytesize
+          ccode = payload[c..c]
+          reason = Cryptocoin::Protocol::VarLenStr.parse_from_raw(payload[c+1..-1])
+          self.new(message, ccode, reason)
+        end
+        
+        def initialize(message, ccode, reason)
+          @message = message
+          @ccode_raw = ccode
+          @reason = reason
+        end
+        
+        def ccode
+          @ccode ||= @ccode_raw.unpack('H*')[0]
+        end
+        
+        def ccode_name
+          case ccode
+          when '1'
+            'REJECT_MALFORMED'
+          when '10'
+            'REJECT_INVALID'
+          when '11'
+            'REJECT_OBSOLETE'
+          when '12'
+            'REJECT_DUPLICATE'
+          when '40'
+            'REJECT_NONSTANDARD'
+          when '41'
+            'REJECT_DUST'
+          when '42'
+            'REJECT_INSUFFICIENTFEE'
+          when '43'
+            'REJECT_CHECKPOINT'
+          end
+        end
+        
+        def raw
+          @message.raw + @ccode_raw + @reason.raw
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/tx.rb

@@ -0,0 +1,11 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Tx
+        def self.parse_from_raw(payload)
+          Cryptocoin::Structure::Transaction.parse_from_raw(payload)
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/verack.rb

@@ -0,0 +1,16 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Verack
+        def self.parse_from_raw(payload)
+          # https://en.bitcoin.it/wiki/Protocol_specification#verack
+          # Has no payload
+          self.new
+        end        
+        def raw
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message/version.rb

@@ -0,0 +1,59 @@
+module Cryptocoin
+  module Protocol
+    class Message
+      class Version
+        def self.parse_from_raw(payload)
+          version = payload[0..3]
+          services = payload[4..11]
+          timestamp = payload[12..19]
+          address_received = Cryptocoin::Protocol::NetAddr.parse_from_raw(payload[20..45])
+          address_from = Cryptocoin::Protocol::NetAddr.parse_from_raw(payload[46..71])
+          nonce = payload[71..79]
+          user_agent = Cryptocoin::Protocol::VarLenStr.parse_from_raw(payload[80..-1])
+          start_height = payload[-5..-2]
+          relay = payload[-1]
+        end
+        
+        def initialize(version, services, timestamp, address_received, address_from, nonce, user_agent, start_height, relay)
+          @version_raw = version
+          @services_raw = services
+          @timestamp_raw = timestamp
+          @address_received = address_received
+          @address_from = address_from
+          @nonce_raw = nonce
+          @user_agent = user_agent
+          @start_height_raw = start_height
+          @relay_raw = relay
+        end
+        
+        def version
+          @version ||= @version_raw.unpack('l')[0]
+        end
+        
+        def services
+          @services ||= @services_raw.unpack('Q')[0]
+        end
+        
+        def timestamp
+          @timestamp ||= @timestamp_raw.unpack('q')[0]
+        end
+        
+        def nonce
+          @nonce ||= @nonce_raw.unpack('Q')[0]
+        end
+        
+        def start_height
+          @start_height ||= @start_height_raw.unpack('l')[0]
+        end
+        
+        def relay
+          @relay ||= @relay_raw.unpack('c')[0]
+        end 
+        
+        def raw
+          @version_raw + @services_raw + @timestamp_raw + @address_received.raw + @address_from.raw + @nonce_raw + @user_agent.raw + @start_height_raw + @relay_raw
+        end
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/message.rb

@@ -0,0 +1,27 @@
+require 'cryptocoin/protocol/message/addr'
+require 'cryptocoin/protocol/message/alert'
+require 'cryptocoin/protocol/message/block'
+require 'cryptocoin/protocol/message/getaddr'
+require 'cryptocoin/protocol/message/getblocks'
+require 'cryptocoin/protocol/message/getdata'
+require 'cryptocoin/protocol/message/getheaders'
+require 'cryptocoin/protocol/message/headers'
+require 'cryptocoin/protocol/message/inv'
+require 'cryptocoin/protocol/message/mempool'
+require 'cryptocoin/protocol/message/notfound'
+require 'cryptocoin/protocol/message/ping'
+require 'cryptocoin/protocol/message/pong'
+require 'cryptocoin/protocol/message/reject'
+require 'cryptocoin/protocol/message/tx'
+require 'cryptocoin/protocol/message/verack'
+require 'cryptocoin/protocol/message/version'
+
+module Cryptocoin
+  module Protocol
+    class Message
+      def self.parse(command, payload)
+        Cryptocoin::Protocol::Message.const_get(command.capitalize).parse_from_raw(payload)
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/net_addr.rb

@@ -0,0 +1,55 @@
+module Cryptocoin
+  module Protocol
+    class NetAddr
+      def self.parse_from_raw(raw)
+        timestamp = raw[0..3]
+        services = raw[4..11]
+        address = raw[12..27]
+        port = raw[28..29]
+        self.new(timestamp, services, address, port)
+      end
+      
+      def initialize(timestamp, services, address, port)
+        @timestamp_raw = timestamp
+        @services_raw = services
+        @address_raw = address
+        @port_raw = port
+      end
+      
+      def timestamp
+        @timestamp ||= @timestamp_raw.unpack('L')[0]
+      end
+      
+      def services
+        @services ||= @services_raw.unpack('Q')[0]
+      end
+      
+      def address
+        return @address if @address
+        address = @address_raw.unpack('H*')[0]
+        if address[0..11] == "000000000000" # IPv4 address
+          address = address[12..-1]
+          @address = address.each_char.each_slice(2).map{|e| e.join.to_i(16)}.join('.')
+          @address_version = 4
+        else
+          @address = address.each_char.each_slice(4).map{|e| e.join}.join(':')
+          @address_version = 6
+        end
+        @address
+      end
+      
+      def port
+        @port ||= @port_raw.unpack('S')[0]
+      end
+      
+      def address_with_port
+        return "#{@address}:#{@port}" if @address_version == 4
+        "[#{@address}]:#{@port}"
+      end
+      
+      def raw
+        @timestamp_raw + @services_raw + @address_raw + @port_raw
+      end
+	  end
+  end
+end

data/lib/cryptocoin/protocol/packet.rb

@@ -0,0 +1,74 @@
+require 'digest/sha2'
+require 'cryptocoin/protocol/message'
+
+module Cryptocoin
+  module Protocol
+    class Packet
+      attr_reader :message, :network
+      
+      def self.parse_from_raw(raw)
+        magic = raw[0..4]
+        command = raw[5..16]
+        payload_length = raw[17..20]
+        payload_checksum = raw[21..24]
+        payload = raw[25..-1]
+        message = Cryptocoin::Protocol::Message.parse(command, payload)
+      end
+      
+      def self.parse_from_io
+        magic = io.read(4)
+        command = io.read(12)
+        payload_length = io.read(4)
+        payload_checksum = io.read(4)
+        payload = io.read
+        message = Cryptocoin::Protocol::Message.parse(command, payload)
+      end
+      
+      def initialize(magic, command, payload_length, payload_checksum, message, network)
+        @magic_raw = magic
+        @command_raw = command
+        @payload_length_raw = payload_length
+        @payload_checksum_raw = payload_checksum
+        @message = message
+        @network = network
+      end
+      
+      def magic
+        @magic_raw.unpack('L')[0]
+      end
+      
+      def command
+        @command_raw.unpack('a*')[0]
+      end
+      
+      def payload_length
+        @payload_length_raw.unpack('L')[0]
+      end
+      
+      def payload_checksum
+        @payload_checksum_raw.unpack('L')[0]
+      end
+
+      def valid?
+        valid_checksum and valid_magic and valid_message
+      end
+      
+      def raw
+        @magic_raw + @command_raw + @payload_length_raw + @payload_checksum_raw + @message.raw
+      end
+
+      private
+      def valid_checksum
+        Cryptocoin::Digest.new(@message.raw, :binary).sha256[0..4] == @payload_checksum_raw
+      end
+
+      def valid_magic
+        magic == network.magic
+      end
+
+      def valid_message
+        @message.valid?
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/var_len_int.rb

@@ -0,0 +1,85 @@
+module Cryptocoin
+  module Protocol
+    class VarLenInt
+      # https://github.com/andrew12/bitcoin-ruby/blob/master/lib/bitcoin.rb#L132
+      # TODO: test this implementation
+      def self.from_int(i)
+        if i < -0xffffffffffffffff
+          return ArgumentError, "Unrepresentable"
+        elsif i < 0
+          top_32 = ((-i) & 0xffffffff00000000) >> 32
+          btm_32 = (-i) & 0x00000000ffffffff
+          return self.new([0xff, top_32, btm_32].pack("CVV"))
+        elsif i <= 0xfc
+          return self.new([i].pack('C'))
+        elsif i <= 0xffff
+          return self.new([0xfd, i].pack("Cv"))
+        elsif i <= 0xffffffff
+          return self.new([0xfe, i].pack("CV"))
+        else
+          return ArgumentError, "Unrepresentable"
+        end
+      end
+
+      def self.parse_from_io(io)
+        i = io.read(1)
+        j = i.unpack('C')[0]
+        case j
+        when 0xfd
+          self.new(i + io.read(2))
+        when 0xfe
+          self.new(i + io.read(4))
+        when 0xff
+          self.new(i + io.read(8))
+        else
+          puts "Something: #{i}"
+          self.new(i)
+        end
+      end
+
+      def initialize(raw)
+        @head_raw = raw[0]
+        @head = @head_raw.unpack('C')[0]
+        @body = case @head
+        when 0xfd
+          @raw_i = raw[1..3]
+          @raw_i.unpack('v')[0]
+        when 0xfe
+          @raw_i = raw[1..5]
+          @raw_i.unpack('V')[0]
+        when 0xff
+          @raw_i = raw[1..9]
+          @raw_i.unpack('Q')[0]
+        else
+          @raw_i = @head_raw
+          @head
+        end
+      end
+
+      def to_i
+        @body
+      end
+
+      def head
+        @head
+      end
+
+      def body_raw
+        @raw_i
+      end
+
+      def body
+        @body
+      end
+
+      def raw
+        @head_raw + (@raw_i == @head_raw ? '' : @raw_i)
+      end
+
+      def method_missing(name, *args, &block)
+        ret = body.send(name, *args, &block)
+        ret
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol/var_len_str.rb

@@ -0,0 +1,18 @@
+require 'cryptocoin/protocol/var_len_int'
+
+module Cryptocoin
+  module Protocol
+    class VarLenStr < String
+      def initialize(raw)
+        @size = Cryptocoin::Protocol::VarLenInt.new(raw)
+        @str = raw.read(@size)
+      end
+      def size
+        @size
+      end
+      def to_s
+        @str
+      end
+    end
+  end
+end

data/lib/cryptocoin/protocol.rb

@@ -0,0 +1,8 @@
+require 'cryptocoin/protocol/message'
+require 'cryptocoin/protocol/packet'
+require 'cryptocoin/protocol/var_len_int'
+require 'cryptocoin/protocol/var_len_str'
+
+module Cryptocoin
+  module Protocol; end
+end