# # =================================================================== # The contents of this file are dedicated to the public domain. To # the extent that dedication to the public domain is not available, # everyone is granted a worldwide, perpetual, royalty-free, # non-exclusive license to exercise all rights associated with the # contents of this file for any purpose whatsoever. # No rights are reserved. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS # BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN # ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE # SOFTWARE. # =================================================================== from __future__ import annotations from typing import Optional, Union from Crypto.Hash.SHA256 import _load_sha_ni_lib from Crypto.Util import _cpu_features from Crypto.Util._raw_api import ( SmartPointer, VoidPointer, c_size_t, c_uint8_ptr_len, create_string_buffer, get_raw_buffer, load_pycryptodome_raw_lib, ) Buffer = Union[bytes, bytearray, memoryview] _sha224_cdecl = """ int SHA224_init(void **shaState); int SHA224_destroy(void *shaState); int SHA224_update(void *hs, const uint8_t *buf, size_t len); int SHA224_digest(const void *shaState, uint8_t *digest, size_t digest_size); int SHA224_copy(const void *src, void *dst); int SHA224_oneshot(const uint8_t *in, size_t len, uint8_t *digest, size_t digest_size); int SHA224_pbkdf2_hmac_assist(const void *inner, const void *outer, const uint8_t *first_digest, uint8_t *final_digest, size_t iterations, size_t digest_size); """ _raw_sha224_portable_lib = load_pycryptodome_raw_lib("Crypto.Hash._SHA224", _sha224_cdecl) _raw_sha224_shani_lib = _load_sha_ni_lib("Crypto.Hash._SHA224_shani", _sha224_cdecl) _raw_sha224_armv8_lib = _cpu_features.load_optional_lib( "Crypto.Hash._SHA224_armv8", _sha224_cdecl, _cpu_features.have_sha256_armv8 ) # The implementation used by SHA224Hash if _raw_sha224_shani_lib is not None: _raw_sha224_lib = _raw_sha224_shani_lib elif _raw_sha224_armv8_lib is not None: _raw_sha224_lib = _raw_sha224_armv8_lib else: _raw_sha224_lib = _raw_sha224_portable_lib class SHA224Hash: """A SHA-224 hash object. Do not instantiate directly. Use the :func:`new` function. :ivar oid: ASN.1 Object ID :vartype oid: string :ivar block_size: the size in bytes of the internal message block, input to the compression function :vartype block_size: integer :ivar digest_size: the size in bytes of the resulting hash :vartype digest_size: integer """ # The size of the resulting hash in bytes. digest_size: int = 28 # The internal block size of the hash algorithm in bytes. block_size: int = 64 # ASN.1 Object ID oid: str = "2.16.840.1.101.3.4.2.4" def __init__(self, data: Optional[Buffer] = None) -> None: state = VoidPointer() result = _raw_sha224_lib.SHA224_init(state.address_of()) if result: raise ValueError("Error %d while instantiating SHA224" % result) self._state = SmartPointer(state.get(), _raw_sha224_lib.SHA224_destroy) if data: self.update(data) def update(self, data: Buffer) -> None: """Continue hashing of a message by consuming the next chunk of data. Args: data (byte string/byte array/memoryview): The next chunk of the message being hashed. """ data_ptr, data_len = c_uint8_ptr_len(data) result = _raw_sha224_lib.SHA224_update(self._state.get(), data_ptr, c_size_t(data_len)) if result: raise ValueError("Error %d while hashing data with SHA224" % result) def digest(self) -> bytes: """Return the **binary** (non-printable) digest of the message that has been hashed so far. :return: The hash digest, computed over the data processed so far. Binary form. :rtype: byte string """ bfr = create_string_buffer(self.digest_size) result = _raw_sha224_lib.SHA224_digest(self._state.get(), bfr, c_size_t(self.digest_size)) if result: raise ValueError("Error %d while making SHA224 digest" % result) return get_raw_buffer(bfr) def hexdigest(self) -> str: """Return the **printable** digest of the message that has been hashed so far. :return: The hash digest, computed over the data processed so far. Hexadecimal encoded. :rtype: string """ return self.digest().hex() def copy(self) -> SHA224Hash: """Return a copy ("clone") of the hash object. The copy will have the same internal state as the original hash object. This can be used to efficiently compute the digests of strings that share a common initial substring. :return: A hash object of the same type """ clone = SHA224Hash() result = _raw_sha224_lib.SHA224_copy(self._state.get(), clone._state.get()) if result: raise ValueError("Error %d while copying SHA224" % result) return clone def new(self, data: Optional[Buffer] = None) -> SHA224Hash: """Create a fresh SHA-224 hash object.""" return SHA224Hash(data) def new(data: Optional[Buffer] = None) -> SHA224Hash: """Create a new hash object. :parameter data: Optional. The very first chunk of the message to hash. It is equivalent to an early call to :meth:`SHA224Hash.update`. :type data: byte string/byte array/memoryview :Return: A :class:`SHA224Hash` hash object """ return SHA224Hash().new(data) def digest(data: Buffer) -> bytes: """Compute the SHA-224 digest of a complete message, in one go. It returns the same bytes as ``new(data).digest()``, but it is faster for short messages. Args: data (bytes/bytearray/memoryview): The whole message to hash. :Return: the 28-byte digest :rtype: byte string """ bfr = create_string_buffer(28) data_ptr, data_len = c_uint8_ptr_len(data) result = _raw_sha224_lib.SHA224_oneshot(data_ptr, c_size_t(data_len), bfr, c_size_t(28)) if result: raise ValueError("Error %d while computing SHA224" % result) return get_raw_buffer(bfr) # The size of the resulting hash in bytes. digest_size: int = SHA224Hash.digest_size # The internal block size of the hash algorithm in bytes. block_size: int = SHA224Hash.block_size def _pbkdf2_hmac_assist(inner, outer, first_digest, iterations): """Compute the expensive inner loop in PBKDF-HMAC.""" # Not an assert: a negative number would become a huge one in C if iterations < 1: raise ValueError("The number of iterations must be positive") bfr = create_string_buffer(len(first_digest)) result = _raw_sha224_lib.SHA224_pbkdf2_hmac_assist( inner._state.get(), outer._state.get(), first_digest, bfr, c_size_t(iterations), c_size_t(len(first_digest)), ) if result: raise ValueError("Error %d with PBKDF2-HMAC assist for SHA224" % result) return get_raw_buffer(bfr)