ciri 0.0.0 → 0.0.1

data/lib/ciri/devp2p/peer.rb

@@ -22,8 +22,9 @@
 # THE SOFTWARE.
 
 
+require 'ciri/actor'
+require 'ciri/utils'
 require_relative 'rlpx'
-require_relative 'actor'
 require_relative 'protocol_io'
 
 module Ciri
@@ -49,6 +50,12 @@ module Ciri
         super()
       end
 
+      def to_s
+        @display_name ||= begin
+          Utils.data_to_hex(node_id.id)[0..8]
+        end
+      end
+
       def node_id
         @node_id ||= RLPX::NodeID.from_raw_id(@handshake.id)
       end
@@ -74,7 +81,7 @@ module Ciri
 
       def start_protocols
         @protocols.each do |protocol|
-          protocol.start.(self, @protocol_io_hash[protocol.name])
+          protocol.start(self, @protocol_io_hash[protocol.name])
         end
       end
 
@@ -94,6 +101,7 @@ module Ciri
       end
 
       private
+
       def find_protocol_io_by_msg_code(code)
         @protocol_io_hash.values.find do |protocol_io|
           offset = protocol_io.offset

data/lib/ciri/devp2p/protocol.rb

@@ -22,8 +22,6 @@
 # THE SOFTWARE.
 
 
-require_relative 'actor'
-
 module Ciri
   module DevP2P
 
@@ -31,12 +29,22 @@ module Ciri
     class Protocol
 
       attr_reader :name, :version, :length
-      attr_accessor :node_info, :peer_info, :start
+      attr_accessor :node_info, :peer_info
 
       def initialize(name:, version:, length:)
         @name = name
         @version = version
         @length = length
+        @start = nil
+      end
+
+      def start=(start_proc)
+        @start = start_proc
+      end
+
+      def start(peer, io)
+        raise NotImplementedError.new('not set protocol start proc') unless @start
+        @start.call(peer, io)
       end
     end
 

data/lib/ciri/devp2p/protocol_io.rb

@@ -21,7 +21,7 @@
 # THE SOFTWARE.
 
 
-require_relative 'actor'
+require 'ciri/actor'
 require_relative 'rlpx/message'
 
 module Ciri
@@ -42,11 +42,14 @@ module Ciri
         @offset = offset
         @frame_io = frame_io
         @msg_queue = Queue.new
+        @mutex = Mutex.new
       end
 
       def send_data(code, data)
-        msg = RLPX::Message.new(code: code, size: data.size, payload: data)
-        write_msg(msg)
+        @mutex.synchronize do
+          msg = RLPX::Message.new(code: code, size: data.size, payload: data)
+          write_msg(msg)
+        end
       end
 
       def write_msg(msg)

data/lib/ciri/devp2p/rlpx.rb

@@ -25,4 +25,5 @@ require_relative 'rlpx/node'
 require_relative 'rlpx/message'
 require_relative 'rlpx/frame_io'
 require_relative 'rlpx/protocol_messages'
+require_relative 'rlpx/protocol_handshake'
 require_relative 'rlpx/encryption_handshake'

data/lib/ciri/devp2p/rlpx/encryption_handshake.rb

@@ -69,7 +69,7 @@ module Ciri
           # xor
           signed = xor(token, nonce)
 
-          signature = random_key.ecdsa_signature(signed)
+          signature = random_key.ecdsa_signature(signed).signature
           initiator_pubkey = private_key.raw_public_key[1..-1]
           AuthMsgV4.new(signature: signature, initiator_pubkey: initiator_pubkey, nonce: nonce, version: 4)
         end

data/lib/ciri/devp2p/rlpx/frame_io.rb

@@ -218,4 +218,4 @@ module Ciri
 
     end
   end
-end
+end

data/lib/ciri/devp2p/rlpx/message.rb

@@ -21,7 +21,7 @@
 # THE SOFTWARE.
 
 
-require 'ciri/rlp/serializable'
+require 'ciri/rlp'
 
 module Ciri
   module DevP2P
@@ -31,13 +31,13 @@ module Ciri
       class Message
         include Ciri::RLP::Serializable
 
+        attr_accessor :received_at
+
         schema [
                  {code: Integer},
                  {size: Integer},
-                 :payload,
-                 :received_at
+                 :payload
                ]
-        default_data(received_at: nil)
       end
 
     end

data/lib/ciri/devp2p/server.rb

@@ -24,10 +24,11 @@
 
 require 'concurrent'
 require 'forwardable'
+require 'ciri/actor'
+require 'ciri/utils/logger'
 require_relative 'rlpx/connection'
 require_relative 'rlpx/protocol_handshake'
 require_relative 'peer'
-require_relative 'actor'
 
 module Ciri
   module DevP2P
@@ -35,6 +36,7 @@ module Ciri
     # DevP2P Server
     # maintain connection, node discovery, rlpx handshake and protocols
     class Server
+      include Utils::Logger
       include RLPX
 
       MAX_ACTIVE_DIAL_TASKS = 16
@@ -47,12 +49,12 @@ module Ciri
       end
 
       attr_reader :handshake, :dial, :scheduler, :protocol_manage, :protocols
-      attr_accessor :logger, :bootstrap_nodes
+      attr_accessor :bootstrap_nodes
 
-      def initialize(private_key:, protocol_manage:, executor:)
+      def initialize(private_key:, protocol_manage:)
         @private_key = private_key
         @name = 'ciri'
-        @scheduler = Scheduler.new(self, executor: executor)
+        @scheduler = Scheduler.new(self)
         @protocol_manage = protocol_manage
         @protocols = protocol_manage.protocols
       end
@@ -66,7 +68,6 @@ module Ciri
         @handshake = ProtocolHandshake.new(version: BASE_PROTOCOL_VERSION, name: @name, id: server_node_id.id, caps: caps)
         # start listen tcp
         @dial = Dial.new(self)
-        @protocol_manage.executor ||= @scheduler.executor
         @protocol_manage.start
         @scheduler.start
       end
@@ -124,19 +125,20 @@ module Ciri
 
       class Scheduler
         include Actor
+        include Utils::Logger
 
         extend Forwardable
 
         attr_reader :server
-        def_delegators :server, :logger
+        def_delegators :server
 
-        def initialize(server, executor:)
+        def initialize(server)
           @server = server
           @queued_tasks = []
           @running_tasks = []
           @peers = {}
           # init actor
-          super(executor: executor)
+          super()
         end
 
         # called by actor loop
@@ -168,11 +170,10 @@ module Ciri
         end
 
         private
+
         def add_peer(connection, handshake)
           server.protocol_handshake_checks(handshake)
           peer = Peer.new(connection, handshake, server.protocols)
-          # set actor executor
-          peer.executor = executor
           @peers[peer.node_id] = peer
           # run peer logic
           # do sub protocol handshake...
@@ -188,16 +189,16 @@ module Ciri
             # remove peer
             self << [:remove_peer, peer, exit_error]
           }
-          logger.debug("add peer: #{peer}")
+          debug("add peer: #{peer}")
         end
 
         def remove_peer(peer, *args)
           error, * = args
-          logger.debug("remove peer: #{peer}, error: #{error}")
+          debug("remove peer: #{peer}, error: #{error}")
         end
 
         def task_done(task, *args)
-          logger.debug("task done: #{task.name}")
+          debug("task done: #{task.name}")
         end
 
       end

data/lib/ciri/eth.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2018, by Jiang Jinyang. <https://justjjy.com>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+
+require_relative 'actor'
+require_relative 'eth/protocol_messages'
+require_relative 'eth/protocol_manage'
+
+module Ciri
+  # implement Ethereum Wire Protocol
+  # https://github.com/ethereum/wiki/wiki/Ethereum-Wire-Protocol
+  module Eth
+  end
+end

data/lib/ciri/eth/peer.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2018, by Jiang Jinyang. <https://justjjy.com>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+
+require 'ciri/chain'
+require 'forwardable'
+require_relative 'protocol_messages'
+
+module Ciri
+  module Eth
+
+    # eth protocol peer
+    class Peer
+      attr_reader :io, :total_difficulty, :status
+
+      extend Forwardable
+
+      def_delegators :@peer, :to_s
+
+      def initialize(protocol_manage:, peer:, io:)
+        @protocol_manage = protocol_manage
+        @io = io
+        @total_difficulty = nil
+        @peer = peer
+      end
+
+      # do eth protocol handshake and return status
+      def handshake(network_id, total_difficulty, head_hash, genesis_hash)
+        status = Status.new(protocol_version: 63, network_id: network_id,
+                            total_difficulty: total_difficulty, current_block: head_hash, genesis_block: genesis_hash)
+        io.send_data(Status::CODE, status.rlp_encode!)
+        msg = io.read_msg
+        @status = Status.rlp_decode!(msg.payload)
+        @total_difficulty = @status.total_difficulty
+        @status
+      end
+
+      def send_msg(msg_class, **data)
+        msg = msg_class.new(data)
+        io.send_data(msg_class::CODE, msg.rlp_encode!)
+      end
+    end
+
+  end
+end

data/lib/ciri/eth/protocol_manage.rb

@@ -0,0 +1,122 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2018, by Jiang Jinyang. <https://justjjy.com>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+
+require 'ciri/actor'
+require 'ciri/chain'
+require_relative 'peer'
+require_relative 'synchronizer'
+
+module Ciri
+  module Eth
+
+    # ProtocolManage
+    class ProtocolManage
+
+      MAX_RESPONSE_HEADERS = 10
+
+      include Ciri::Actor
+
+      attr_reader :protocols, :synchronizer, :chain
+
+      def initialize(protocols:, chain:)
+        @protocols = protocols
+        @peers = {}
+        @chain = chain
+
+        @synchronizer = Synchronizer.new(chain: chain)
+        super()
+      end
+
+      def protocols
+        @protocols.each {|p| p.start = proc {|peer, io| self << [:new_peer, peer, io]}}
+        @protocols
+      end
+
+      def start
+        # start syncing
+        synchronizer.start
+        super
+      end
+
+      # new peer come in
+      def new_peer(peer, io)
+        peer = Peer.new(protocol_manage: self, peer: peer, io: io)
+        peer.handshake(1, chain.total_difficulty, chain.head.get_hash, chain.genesis_hash)
+        @peers[peer] = true
+
+        # register peer to synchronizer
+        synchronizer << [:register_peer, peer]
+        # start handle peer messages
+        executor.post {handle_peer(peer)}
+      end
+
+      def handle_peer(peer)
+        handle_msg(peer, peer.io.read_msg) while true
+      end
+
+      def handle_msg(peer, msg)
+        case msg.code
+        when GetBlockHeaders::CODE
+          get_header_msg = GetBlockHeaders.rlp_decode(msg.payload)
+          hash_or_number = get_header_msg.hash_or_number
+
+          header = if hash_or_number.is_a?(Integer)
+                     chain.get_header_by_number hash_or_number
+                   else
+                     chain.get_header hash_or_number
+                   end
+          headers = []
+
+          if header
+            amount = [MAX_RESPONSE_HEADERS, get_header_msg.amount].min
+            # skip
+            get_header_msg.skip.times do
+              next_header = chain.get_header_by_number header.number + 1
+              break if next_header.nil? || next_header.parent_hash != header.get_hash
+              header = next_header
+            end
+            amount.times do
+              headers << header
+              next_header = chain.get_header_by_number header.number + 1
+              break if next_header.nil? || next_header.parent_hash != header.get_hash
+              header = next_header
+            end
+            header.reverse! if get_header_msg.reverse
+          end
+
+          headers_msg = BlockHeaders.new(headers: headers).rlp_encode!
+          peer.io.send_data(BlockHeaders::CODE, headers_msg)
+        when BlockHeaders::CODE
+          headers = BlockHeaders.rlp_decode!(msg.payload).headers
+          synchronizer << [:receive_headers, peer, headers] unless headers.empty?
+        when BlockBodies::CODE
+          bodies = BlockBodies.rlp_decode!(msg.payload).bodies
+          synchronizer << [:receive_bodies, peer, bodies] unless bodies.empty?
+        else
+          raise StandardError, "unknown code #{msg.code}, #{msg}"
+        end
+      end
+    end
+
+  end
+end