varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-11 07:30:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

reject dht-msgs upon receiving invalid records

Browse Source
This commit is contained in:
lla-dane
2025-08-15 16:11:27 +05:30
parent efc899e872
commit 57d1c9d807
5 changed files with 89 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libp2p/kad_dht/value_store.py
View File

@ -161,8 +161,11 @@ class ValueStore:
# Check if response is valid
if response.type == Message.MessageType.PUT_VALUE:
# Consume the sender's signed-peer-record if sent
_ = maybe_consume_signed_record(response, self.host)
if not maybe_consume_signed_record(response, self.host):
logger.error(
"Received an invalid-signed-record, ignoring the response"
)
return False
if response.key == key:
result = True
return result
@ -288,7 +291,11 @@ class ValueStore:
and response.record.value
):
# Consume the sender's signed-peer-record
_ = maybe_consume_signed_record(response, self.host)
if not maybe_consume_signed_record(response, self.host):
logger.error(
"Received an invalid-signed-record, ignoring the response"
)
return None
logger.debug(
f"Received value for key {key.hex()} from peer {peer_id}"