77 lines
1.8 KiB
Python
77 lines
1.8 KiB
Python
import hashlib
|
|
import base58
|
|
import multihash
|
|
|
|
# MaxInlineKeyLength is the maximum length a key can be for it to be inlined in
|
|
# the peer ID.
|
|
# * When `len(pubKey.Bytes()) <= MaxInlineKeyLength`, the peer ID is the
|
|
# identity multihash hash of the public key.
|
|
# * When `len(pubKey.Bytes()) > MaxInlineKeyLength`, the peer ID is the
|
|
# sha2-256 multihash of the public key.
|
|
MAX_INLINE_KEY_LENGTH = 42
|
|
|
|
|
|
class ID:
|
|
|
|
def __init__(self, id_str):
|
|
self._id_str = id_str
|
|
|
|
def get_raw_id(self):
|
|
return self._id_str
|
|
|
|
def pretty(self):
|
|
return base58.b58encode(self._id_str).decode()
|
|
|
|
def get_xor_id(self):
|
|
return int(digest(self.get_raw_id()).hex(), 16)
|
|
|
|
def __str__(self):
|
|
pid = self.pretty()
|
|
return pid
|
|
|
|
__repr__ = __str__
|
|
|
|
def __eq__(self, other):
|
|
#pylint: disable=protected-access
|
|
return self._id_str == other._id_str
|
|
|
|
def __hash__(self):
|
|
return hash(self._id_str)
|
|
|
|
|
|
def id_b58_encode(peer_id):
|
|
"""
|
|
return a b58-encoded string
|
|
"""
|
|
#pylint: disable=protected-access
|
|
return base58.b58encode(peer_id._id_str).decode()
|
|
|
|
|
|
def id_b58_decode(peer_id_str):
|
|
"""
|
|
return a base58-decoded peer ID
|
|
"""
|
|
return ID(base58.b58decode(peer_id_str))
|
|
|
|
|
|
def id_from_public_key(key):
|
|
# export into binary format
|
|
key_bin = key.exportKey("DER")
|
|
|
|
algo = multihash.Func.sha2_256
|
|
# TODO: seems identity is not yet supported in pymultihash
|
|
# if len(b) <= MAX_INLINE_KEY_LENGTH:
|
|
# algo multihash.func.identity
|
|
|
|
mh_digest = multihash.digest(key_bin, algo)
|
|
return ID(mh_digest.encode())
|
|
|
|
|
|
def id_from_private_key(key):
|
|
return id_from_public_key(key.publickey())
|
|
|
|
def digest(string):
|
|
if not isinstance(string, bytes):
|
|
string = str(string).encode('utf8')
|
|
return hashlib.sha1(string).digest()
|