Merge pull request ethereum#1543 from jannikluhn/beacon-chain-commands

Define beacon chain network commands

Author: jannikluhn

trinity/protocol/bcc/__init__.py

@@ -0,0 +1,36 @@
+from rlp import sedes
+
+from p2p.protocol import (
+    Command,
+)
+
+from trinity.rlp.sedes import HashOrNumber
+from eth.beacon.types.blocks import BaseBeaconBlock
+from eth.beacon.types.attestation_records import AttestationRecord
+
+
+class Status(Command):
+    _cmd_id = 0
+    structure = [
+        ('protocol_version', sedes.big_endian_int),
+        ('network_id', sedes.big_endian_int),
+        ('best_hash', sedes.binary),
+    ]
+
+
+class GetBeaconBlocks(Command):
+    _cmd_id = 1
+    structure = [
+        ('block_slot_or_hash', HashOrNumber()),
+        ('max_blocks', sedes.big_endian_int),
+    ]
+
+
+class BeaconBlocks(Command):
+    _cmd_id = 2
+    structure = sedes.CountableList(BaseBeaconBlock)
+
+
+class AttestationRecords(Command):
+    _cmd_id = 3
+    structure = sedes.CountableList(AttestationRecord)

trinity/protocol/bcc/commands.py

@@ -0,0 +1,18 @@
+import logging
+
+from p2p.protocol import Protocol
+
+from trinity.protocol.bcc.commands import (
+    Status,
+    GetBeaconBlocks,
+    BeaconBlocks,
+    AttestationRecords,
+)
+
+
+class BCCProtocol(Protocol):
+    name = "bcc"
+    version = 0
+    _commands = [Status, GetBeaconBlocks, BeaconBlocks, AttestationRecords]
+    cmd_length = 4
+    logger = logging.getLogger("p2p.bcc.BCCProtocol")