WolfSSL
/
wolfcrypt-py
mirror of https://github.com/wolfSSL/wolfcrypt-py.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Removed magic numbers

pull/11/head
Daniele Lacamera 2020-01-07 20:20:04 +01:00
parent 0b4cbfcb4d
commit 19cf0db33e
1 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/wolfcrypt/hashes.py
View File

@ -204,15 +204,19 @@ if _lib.SHA3_ENABLED:
""" """
_native_type = "wc_Sha3 *" _native_type = "wc_Sha3 *"
_native_size = _ffi.sizeof("wc_Sha3") _native_size = _ffi.sizeof("wc_Sha3")
SHA3_224_DIGEST_SIZE = 28
SHA3_256_DIGEST_SIZE = 32
SHA3_384_DIGEST_SIZE = 48
SHA3_512_DIGEST_SIZE = 64
def __init__(self): # pylint: disable=W0231 def __init__(self): # pylint: disable=W0231
self._native_object = _ffi.new(self._native_type) self._native_object = _ffi.new(self._native_type)
self.digest_size = 48 self.digest_size = SHA3_384_DIGEST_SIZE
ret = self._init() ret = self._init()
if ret < 0: # pragma: no cover if ret < 0: # pragma: no cover
raise WolfCryptError("Sha3 init error (%d)" % ret) raise WolfCryptError("Sha3 init error (%d)" % ret)
def __init__(self, string, size=48): # pylint: disable=W0231 def __init__(self, string, size=SHA3_384_DIGEST_SIZE): # pylint: disable=W0231
self._native_object = _ffi.new(self._native_type) self._native_object = _ffi.new(self._native_type)
self.digest_size = size self.digest_size = size
ret = self._init() ret = self._init()
@ -222,36 +226,36 @@ if _lib.SHA3_ENABLED:
self.update(string) self.update(string)
def _init(self): def _init(self):
if (self.digest_size != 28 and if (self.digest_size != Sha3.SHA3_224_DIGEST_SIZE and
self.digest_size != 32 and self.digest_size != Sha3.SHA3_256_DIGEST_SIZE and
self.digest_size != 48 and self.digest_size != Sha3.SHA3_384_DIGEST_SIZE and
self.digest_size != 64): self.digest_size != Sha3.SHA3_512_DIGEST_SIZE):
return -1 return -1
if self.digest_size == 28: if self.digest_size == Sha3.SHA3_224_DIGEST_SIZE:
return _lib.wc_InitSha3_224(self._native_object, _ffi.NULL, 0) return _lib.wc_InitSha3_224(self._native_object, _ffi.NULL, 0)
if self.digest_size == 32: if self.digest_size == Sha3.SHA3_256_DIGEST_SIZE:
return _lib.wc_InitSha3_256(self._native_object, _ffi.NULL, 0) return _lib.wc_InitSha3_256(self._native_object, _ffi.NULL, 0)
if self.digest_size == 48: if self.digest_size == Sha3.SHA3_384_DIGEST_SIZE:
return _lib.wc_InitSha3_384(self._native_object, _ffi.NULL, 0) return _lib.wc_InitSha3_384(self._native_object, _ffi.NULL, 0)
if self.digest_size == 64: if self.digest_size == Sha3.SHA3_512_DIGEST_SIZE:
return _lib.wc_InitSha3_512(self._native_object, _ffi.NULL, 0) return _lib.wc_InitSha3_512(self._native_object, _ffi.NULL, 0)
def _update(self, data): def _update(self, data):
if self.digest_size == 28: if self.digest_size == Sha3.SHA3_224_DIGEST_SIZE:
return _lib.wc_Sha3_224_Update(self._native_object, data, len(data)) return _lib.wc_Sha3_224_Update(self._native_object, data, len(data))
if self.digest_size == 32: if self.digest_size == Sha3.SHA3_256_DIGEST_SIZE:
return _lib.wc_Sha3_256_Update(self._native_object, data, len(data)) return _lib.wc_Sha3_256_Update(self._native_object, data, len(data))
if self.digest_size == 48: if self.digest_size == Sha3.SHA3_384_DIGEST_SIZE:
return _lib.wc_Sha3_384_Update(self._native_object, data, len(data)) return _lib.wc_Sha3_384_Update(self._native_object, data, len(data))
if self.digest_size == 64: if self.digest_size == Sha3.SHA3_512_DIGEST_SIZE:
return _lib.wc_Sha3_512_Update(self._native_object, data, len(data)) return _lib.wc_Sha3_512_Update(self._native_object, data, len(data))
def _final(self, obj, ret): def _final(self, obj, ret):
if self.digest_size == 28: if self.digest_size == Sha3.SHA3_224_DIGEST_SIZE:
return _lib.wc_Sha3_224_Final(obj, ret) return _lib.wc_Sha3_224_Final(obj, ret)
if self.digest_size == 32: if self.digest_size == Sha3.SHA3_256_DIGEST_SIZE:
return _lib.wc_Sha3_256_Final(obj, ret) return _lib.wc_Sha3_256_Final(obj, ret)
if self.digest_size == 48: if self.digest_size == Sha3.SHA3_384_DIGEST_SIZE:
return _lib.wc_Sha3_384_Final(obj, ret) return _lib.wc_Sha3_384_Final(obj, ret)
if self.digest_size == 64: if self.digest_size == Sha3.SHA3_512_DIGEST_SIZE:
return _lib.wc_Sha3_512_Final(obj, ret) return _lib.wc_Sha3_512_Final(obj, ret)
# Hmac types # Hmac types