varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 16:10:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix signature validator:

Browse Source 
Add prefix and return verify result
This commit is contained in:
NIC619
2019-11-28 18:45:00 +08:00
parent d5d6962dce
commit 064c109b64
2 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
libp2p/pubsub/pubsub.py
View File

@ -49,6 +49,8 @@ SyncValidatorFn = Callable[[ID, rpc_pb2.Message], bool]
AsyncValidatorFn = Callable[[ID, rpc_pb2.Message], Awaitable[bool]]
ValidatorFn = Union[SyncValidatorFn, AsyncValidatorFn]
PUBSUB_SIGNING_PREFIX = "libp2p-pubsub:"
class TopicValidator(NamedTuple):
validator: ValidatorFn
@ -475,7 +477,9 @@ class Pubsub:
if self.strict_signing:
priv_key = self.sign_key
signature = priv_key.sign(msg.SerializeToString())
signature = priv_key.sign(
PUBSUB_SIGNING_PREFIX.encode() + msg.SerializeToString()
)
msg.key = self.host.get_public_key().serialize()
msg.signature = signature
@ -536,7 +540,19 @@ class Pubsub:
logger.debug("Reject because no signature attached for msg: %s", msg)
return
# Validate the signature of the message
if not signature_validator(deserialize_public_key(msg.key), msg):
# First, construct the original payload that's signed by 'msg.key'
msg_without_key_sig = rpc_pb2.Message(
data=msg.data,
topicIDs=msg.topicIDs,
from_id=msg.from_id,
seqno=msg.seqno,
)
payload = (
PUBSUB_SIGNING_PREFIX.encode() + msg_without_key_sig.SerializeToString()
)
if not signature_validator(
deserialize_public_key(msg.key), payload, msg.signature
):
logger.debug("Signature validation failed for msg: %s", msg)
return