From 80452d958907cc3a26ffee1df186de119578cd28 Mon Sep 17 00:00:00 2001 From: mhchia Date: Wed, 21 Aug 2019 13:59:42 +0800 Subject: [PATCH] Fix `make_exchange_message` to use the new API --- libp2p/security/insecure/transport.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/libp2p/security/insecure/transport.py b/libp2p/security/insecure/transport.py index 1048791f..850bdf36 100644 --- a/libp2p/security/insecure/transport.py +++ b/libp2p/security/insecure/transport.py @@ -1,4 +1,5 @@ from libp2p.crypto.keys import PublicKey +from libp2p.crypto.pb import crypto_pb2 from libp2p.crypto.utils import pubkey_from_protobuf from libp2p.network.connection.raw_connection_interface import IRawConnection from libp2p.peer.id import ID @@ -86,6 +87,8 @@ class InsecureTransport(BaseSecureTransport): def make_exchange_message(pubkey: PublicKey) -> plaintext_pb2.Exchange: - pubkey_pb = pubkey.serialize_to_protobuf() + pubkey_pb = crypto_pb2.PublicKey( + key_type=pubkey.get_type().value, data=pubkey.to_bytes() + ) id_bytes = ID.from_pubkey(pubkey).to_bytes() return plaintext_pb2.Exchange(id=id_bytes, pubkey=pubkey_pb)