mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
import logging
|
|
|
|
from libp2p.abc import IHost
|
|
from libp2p.peer.envelope import consume_envelope
|
|
from libp2p.peer.id import ID
|
|
from libp2p.pubsub.pb.rpc_pb2 import RPC
|
|
|
|
logger = logging.getLogger("pubsub-example.utils")
|
|
|
|
|
|
def maybe_consume_signed_record(msg: RPC, host: IHost, peer_id: ID) -> bool:
|
|
"""
|
|
Attempt to parse and store a signed-peer-record (Envelope) received during
|
|
PubSub communication. If the record is invalid, the peer-id does not match, or
|
|
updating the peerstore fails, the function logs an error and returns False.
|
|
|
|
Parameters
|
|
----------
|
|
msg : RPC
|
|
The protobuf message received during PubSub communication.
|
|
host : IHost
|
|
The local host instance, providing access to the peerstore for storing
|
|
verified peer records.
|
|
peer_id : ID | None, optional
|
|
The expected peer ID for record validation. If provided, the peer ID
|
|
inside the record must match this value.
|
|
|
|
Returns
|
|
-------
|
|
bool
|
|
True if a valid signed peer record was successfully consumed and stored,
|
|
False otherwise.
|
|
|
|
"""
|
|
if msg.HasField("senderRecord"):
|
|
try:
|
|
# Convert the signed-peer-record(Envelope) from
|
|
# protobuf bytes
|
|
envelope, record = consume_envelope(msg.senderRecord, "libp2p-peer-record")
|
|
if not record.peer_id == peer_id:
|
|
return False
|
|
|
|
# Use the default TTL of 2 hours (7200 seconds)
|
|
if not host.get_peerstore().consume_peer_record(envelope, 7200):
|
|
logger.error("Failed to update the Certified-Addr-Book")
|
|
return False
|
|
except Exception as e:
|
|
logger.error("Failed to update the Certified-Addr-Book: %s", e)
|
|
return False
|
|
return True
|