def test_pack():
sender, recipient = random_address(), random_address()
version = rlp.sedes.big_endian_int.serialize(discovery.PROTO_VERSION)
payload = (version, sender.to_endpoint(), recipient.to_endpoint())
privkey = keys.PrivateKey(keccak(b"seed"))
message = discovery._pack_v4(discovery.CMD_PING.id, payload, privkey)
pubkey, cmd_id, payload, _ = discovery._unpack_v4(message)
assert pubkey == privkey.public_key
assert cmd_id == discovery.CMD_PING.id
assert len(payload) == discovery.CMD_PING.elem_count
def test_v5_handlers(monkeypatch):
# Ensure we dispatch v5 messages to the appropriate handlers.
# These are hex-encoded messages sent by geth over the wire, obtained via wireshark using the
# ethereum dissectors (https://github.com/ConsenSys/ethereum-dissectors).
v5_handlers = dict(
recv_topic_register=
'74656d706f7261727920646973636f766572792076354b1f9f999e3cf283e42270742eb3b52b1c0b63ede116ea58632517ffccc7516c794f6fc20f199b3a3a887f3e961b453bb9213eebad4581aaf9b2a60da7e5ba7b0106e2da954c455332406434653536373430663837366165663883666f6f8204d283666f6f', # noqa: E501
recv_pong_v5=
'74656d706f7261727920646973636f766572792076356cb538da9382c022cc6d9b860b7cc7660718c731e1e6331de86a7cabfba8f01c061b5d6c09dd84de1d6c653536bb8ad31fb140d55d317dfaea5225a6907f49450102f855cb84a5a587f2827660827660a0b484b40a3712766dbbe8498f110bab968a543f9d0d57ec4cfa3b38b32ef2f237831f5ca3a0bd1ea218eaf7cddc6b068a414ae08876315f92bd134e214d5f3961d1a3e8e35b4ec13c', # noqa: E501
recv_ping_v5=
'74656d706f7261727920646973636f76657279207635c26911f1d2a320f17fa683397028e6cc7ebe114623e28e933d870634aa6050f45960679170192ec0f133dbc3f4e627eef468dbdb48bf4a7848412938da21be8a0001f84104d79000000000000000000000000000000000827660827660cb84b0201d9282848b82848b845b7e5593d6954c4553324064346535363734306638373661656638', # noqa: E501
recv_find_nodehash=
'74656d706f7261727920646973636f7665727920763558dc895847c5d2cc9ab6f722f25b9ff1b9c188af6ff7ed7c716a3cde2e93f3ec1e2b897acb1a33aa1d345c72ea34912287b21480c80ef99241d4bd9fd53711ee0105e6a038f7bb96e94bcd39866baa56038367ad6145de1ee8f4a8b0993ebdf8883a0ad8845b7e5593', # noqa: E501
recv_topic_query=
'74656d706f7261727920646973636f76657279207635308621f1ed59a67613da52da3f84bac3d927d0dc1650b27552b10a0a823fb2133cebf3680f68bf88457bb0c07787031357430985d03afa1af0574e8ef0eab7fc0007d7954c455332406434653536373430663837366165663880', # noqa: E501
recv_topic_nodes=
'74656d706f7261727920646973636f76657279207635e33166b59d20f206f0df2502be59186cb971c76ee91dba94ea9b1bbeb1308a5f4efbdf945ead9ba89d7c2c55d5d841dacdef69a455c98e1a5415e489508afa450008f87ea0cff0456ecbf2e6b0a40bc6541bd209c60ced192509470b405c256a17443674b3f85bf8599000000000000000000000ffff904c1f9c82765f82765fb840e7d624c642b86d3d48cea3e395305345c3a0226fc4c2dfdfbeb94cb6891e5e72b4467c69684ac14b072d2e4fa9c7a731cc1fdf0283abe41186d00b4c879f80ed', # noqa: E501
recv_neighbours_v5=
'74656d706f7261727920646973636f766572792076358f6671ae9611c82c9cb04538aeed13a8b4e8eb8ad0d0dbba4b161ded52b195846c6086a0d42eef44cfcc0b793a0b9420613727958a8956139c127810b94d4e830004f90174f9016cf8599000000000000000000000ffff59401a22826597826597b840d723e264da67820fb0cedb0d03d5d975cc82bffdadd2879f3e5fa58b5525de5fdd0b90002bba44ac9232247dfbccb2a730e5ea98201bab1f1fe72422aa58143ff8599000000000000000000000ffffae6c601a82765f82765fb840779f19056e0a0486c3f6838896a931bf920cd8f551f664022a50690d4cca4730b50a97058aac11a5aa0cc55db6f9207e12a9cd389269f414a98e5b6a2f6c9f89f8599000000000000000000000ffff287603df827663827663b84085c85d7143ae8bb96924f2b54f1b3e70d8c4d367af305325d30a61385a432f247d2c75c45c6b4a60335060d072d7f5b35dd1d4c45f76941f62a4f83b6e75daaff8599000000000000000000000ffff0d4231b0820419820419b8407b46cc366b6cbaec088a7d15688a2c12bb8ba4cf7ee8e01b22ab534829f9ff13f7cc4130f10a4021f7d77e9b9c80a9777f5ddc035efb130fe3b6786434367973845b7e5569', # noqa: E501
)
proto = get_discovery_protocol()
addr = random_address()
for handler, msg in v5_handlers.items():
mock_handler = MockHandler()
monkeypatch.setattr(proto, handler, mock_handler)
proto.datagram_received(decode_hex(msg), (addr.ip, addr.udp_port))
assert mock_handler.called
def __init__(self, bootnodes):
privkey = keys.PrivateKey(keccak(b"seed"))
super().__init__(privkey, random_address(), bootnodes, CancelToken("discovery-test"))