StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Generate peer IDs according to the spec

Browse Source
This commit is contained in:
Alex Stokes 2019-08-13 18:19:01 -07:00
parent f9e859a8e9
commit 6506079a36
No known key found for this signature in database
GPG Key ID: 51CE1721B245C086
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libp2p/peer/id.py
View File

@ -4,6 +4,15 @@ from typing import Union
import base58 import base58
import multihash import multihash
from libp2p.crypto.keys import PublicKey
def _serialize_public_key(key: PublicKey) -> bytes:
"""
Serializes ``key`` in the way expected to form valid peer ids.
"""
return key.serialize_to_protobuf().SerializeToString()
class ID: class ID:
@ -53,9 +62,10 @@ class ID:
return pid return pid
@classmethod @classmethod
def from_pubkey(cls, key: bytes) -> "ID": def from_pubkey(cls, key: PublicKey) -> "ID":
serialized_key = _serialize_public_key(key)
algo = multihash.Func.sha2_256 algo = multihash.Func.sha2_256
mh_digest = multihash.digest(key, algo) mh_digest = multihash.digest(serialized_key, algo)
return cls(mh_digest.encode()) return cls(mh_digest.encode())