burmatovan/openuds
1
0
Fork 0
forked from shaba/openuds
Code Issues Pull Requests Projects Releases Wiki Activity

Added types to crypto

Browse Source
This commit is contained in:
Adolfo Gómez García 2021-07-08 17:46:46 +02:00
parent 582ba01014
commit 962015c355
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/src/uds/core/managers/crypto.py
View File

@ -159,7 +159,7 @@ class CryptoManager:
return toDecode[4 : 4 + struct.unpack('>i', toDecode[:4])[0]]
def xor(self, s1: typing.Union[str, bytes], s2: typing.Union[str, bytes]) -> bytes:
if len(s2) == 0:
if not s2:
return b'' # Protect against division by cero
if isinstance(s1, str):
@ -207,7 +207,7 @@ class CryptoManager:
except Exception as e:
raise e
def loadCertificate(self, certificate: typing.Union[str, bytes]):
def loadCertificate(self, certificate: typing.Union[str, bytes]) -> x509.Certificate:
if isinstance(certificate, str):
certificate = certificate.encode()