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

Address incorrect typing in pycryptodome dependency

Browse Source
This commit is contained in:
Alex Stokes
2019-09-04 10:19:27 -07:00
parent a099b9c65d
commit 451ec2664a
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/crypto/rsa.py
View File

@ -24,7 +24,8 @@ class RSAPublicKey(PublicKey):
def verify(self, data: bytes, signature: bytes) -> bool: def verify(self, data: bytes, signature: bytes) -> bool:
h = SHA256.new(data) h = SHA256.new(data)
try: try:
pkcs1_15.new(self.impl).verify(h, signature) # NOTE: the typing in ``pycryptodome`` is wrong on the arguments to ``verify``.
pkcs1_15.new(self.impl).verify(h, signature) # type: ignore
except (ValueError, TypeError): except (ValueError, TypeError):
return False return False
return True return True
@ -47,7 +48,8 @@ class RSAPrivateKey(PrivateKey):
def sign(self, data: bytes) -> bytes: def sign(self, data: bytes) -> bytes:
h = SHA256.new(data) h = SHA256.new(data)
return pkcs1_15.new(self.impl).sign(h) # NOTE: the typing in ``pycryptodome`` is wrong on the arguments to ``sign``.
return pkcs1_15.new(self.impl).sign(h) # type: ignore
def get_public_key(self) -> PublicKey: def get_public_key(self) -> PublicKey:
return RSAPublicKey(self.impl.publickey()) return RSAPublicKey(self.impl.publickey())