diff --git a/src/lib/adafruit_hashlib/_sha384.mpy b/src/lib/adafruit_hashlib/_sha384.mpy new file mode 100644 index 0000000..10d0718 Binary files /dev/null and b/src/lib/adafruit_hashlib/_sha384.mpy differ diff --git a/src/lib/adafruit_hashlib/_sha512.py b/src/lib/adafruit_hashlib/_sha512.py new file mode 100644 index 0000000..23e23ec --- /dev/null +++ b/src/lib/adafruit_hashlib/_sha512.py @@ -0,0 +1,507 @@ +SHA_BLOCKSIZE = 128 +SHA_DIGESTSIZE = 64 + + +def new_shaobject(): + """Struct. for storing SHA information.""" + return { + "digest": [0] * 8, + "count_lo": 0, + "count_hi": 0, + "data": [0] * SHA_BLOCKSIZE, + "local": 0, + "digestsize": 0, + } + + +# Various logical functions +ROR64 = ( + lambda x, y: (((x & 0xFFFFFFFFFFFFFFFF) >> (y & 63)) | (x << (64 - (y & 63)))) + & 0xFFFFFFFFFFFFFFFF +) +Ch = lambda x, y, z: (z ^ (x & (y ^ z))) +Maj = lambda x, y, z: (((x | y) & z) | (x & y)) +S = lambda x, n: ROR64(x, n) +R = lambda x, n: (x & 0xFFFFFFFFFFFFFFFF) >> n +Sigma0 = lambda x: (S(x, 28) ^ S(x, 34) ^ S(x, 39)) +Sigma1 = lambda x: (S(x, 14) ^ S(x, 18) ^ S(x, 41)) +Gamma0 = lambda x: (S(x, 1) ^ S(x, 8) ^ R(x, 7)) +Gamma1 = lambda x: (S(x, 19) ^ S(x, 61) ^ R(x, 6)) + +# pylint: disable=protected-access, too-many-statements +def sha_transform(sha_info): + W = [] + + d = sha_info["data"] + for i in range(0, 16): + W.append( + (d[8 * i] << 56) + + (d[8 * i + 1] << 48) + + (d[8 * i + 2] << 40) + + (d[8 * i + 3] << 32) + + (d[8 * i + 4] << 24) + + (d[8 * i + 5] << 16) + + (d[8 * i + 6] << 8) + + d[8 * i + 7] + ) + + for i in range(16, 80): + W.append( + (Gamma1(W[i - 2]) + W[i - 7] + Gamma0(W[i - 15]) + W[i - 16]) + & 0xFFFFFFFFFFFFFFFF + ) + + ss = sha_info["digest"][:] + + # pylint: disable=line-too-long, too-many-arguments + def RND(a, b, c, d, e, f, g, h, i, ki): + t0 = (h + Sigma1(e) + Ch(e, f, g) + ki + W[i]) & 0xFFFFFFFFFFFFFFFF + t1 = (Sigma0(a) + Maj(a, b, c)) & 0xFFFFFFFFFFFFFFFF + d = (d + t0) & 0xFFFFFFFFFFFFFFFF + h = (t0 + t1) & 0xFFFFFFFFFFFFFFFF + return d & 0xFFFFFFFFFFFFFFFF, h & 0xFFFFFFFFFFFFFFFF + + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 0, 0x428A2F98D728AE22 + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 1, 0x7137449123EF65CD + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 2, 0xB5C0FBCFEC4D3B2F + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 3, 0xE9B5DBA58189DBBC + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 4, 0x3956C25BF348B538 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 5, 0x59F111F1B605D019 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 6, 0x923F82A4AF194F9B + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 7, 0xAB1C5ED5DA6D8118 + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 8, 0xD807AA98A3030242 + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 9, 0x12835B0145706FBE + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 10, 0x243185BE4EE4B28C + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 11, 0x550C7DC3D5FFB4E2 + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 12, 0x72BE5D74F27B896F + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 13, 0x80DEB1FE3B1696B1 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 14, 0x9BDC06A725C71235 + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 15, 0xC19BF174CF692694 + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 16, 0xE49B69C19EF14AD2 + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 17, 0xEFBE4786384F25E3 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 18, 0x0FC19DC68B8CD5B5 + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 19, 0x240CA1CC77AC9C65 + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 20, 0x2DE92C6F592B0275 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 21, 0x4A7484AA6EA6E483 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 22, 0x5CB0A9DCBD41FBD4 + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 23, 0x76F988DA831153B5 + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 24, 0x983E5152EE66DFAB + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 25, 0xA831C66D2DB43210 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 26, 0xB00327C898FB213F + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 27, 0xBF597FC7BEEF0EE4 + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 28, 0xC6E00BF33DA88FC2 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 29, 0xD5A79147930AA725 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 30, 0x06CA6351E003826F + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 31, 0x142929670A0E6E70 + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 32, 0x27B70A8546D22FFC + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 33, 0x2E1B21385C26C926 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 34, 0x4D2C6DFC5AC42AED + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 35, 0x53380D139D95B3DF + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 36, 0x650A73548BAF63DE + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 37, 0x766A0ABB3C77B2A8 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 38, 0x81C2C92E47EDAEE6 + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 39, 0x92722C851482353B + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 40, 0xA2BFE8A14CF10364 + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 41, 0xA81A664BBC423001 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 42, 0xC24B8B70D0F89791 + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 43, 0xC76C51A30654BE30 + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 44, 0xD192E819D6EF5218 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 45, 0xD69906245565A910 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 46, 0xF40E35855771202A + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 47, 0x106AA07032BBD1B8 + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 48, 0x19A4C116B8D2D0C8 + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 49, 0x1E376C085141AB53 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 50, 0x2748774CDF8EEB99 + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 51, 0x34B0BCB5E19B48A8 + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 52, 0x391C0CB3C5C95A63 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 53, 0x4ED8AA4AE3418ACB + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 54, 0x5B9CCA4F7763E373 + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 55, 0x682E6FF3D6B2B8A3 + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 56, 0x748F82EE5DEFB2FC + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 57, 0x78A5636F43172F60 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 58, 0x84C87814A1F0AB72 + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 59, 0x8CC702081A6439EC + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 60, 0x90BEFFFA23631E28 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 61, 0xA4506CEBDE82BDE9 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 62, 0xBEF9A3F7B2C67915 + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 63, 0xC67178F2E372532B + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 64, 0xCA273ECEEA26619C + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 65, 0xD186B8C721C0C207 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 66, 0xEADA7DD6CDE0EB1E + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 67, 0xF57D4F7FEE6ED178 + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 68, 0x06F067AA72176FBA + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 69, 0x0A637DC5A2C898A6 + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 70, 0x113F9804BEF90DAE + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 71, 0x1B710B35131C471B + ) + ss[3], ss[7] = RND( + ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 72, 0x28DB77F523047D84 + ) + ss[2], ss[6] = RND( + ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 73, 0x32CAAB7B40C72493 + ) + ss[1], ss[5] = RND( + ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 74, 0x3C9EBE0A15C9BEBC + ) + ss[0], ss[4] = RND( + ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 75, 0x431D67C49C100D4C + ) + ss[7], ss[3] = RND( + ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 76, 0x4CC5D4BECB3E42B6 + ) + ss[6], ss[2] = RND( + ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 77, 0x597F299CFC657E2A + ) + ss[5], ss[1] = RND( + ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 78, 0x5FCB6FAB3AD6FAEC + ) + ss[4], ss[0] = RND( + ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 79, 0x6C44198C4A475817 + ) + + # Feedback + dig = [] + for i, x in enumerate(sha_info["digest"]): + dig.append((x + ss[i]) & 0xFFFFFFFFFFFFFFFF) + sha_info["digest"] = dig + + +def sha_init(): + """Initialize the SHA digest.""" + sha_info = new_shaobject() + sha_info["digest"] = [ + 0x6A09E667F3BCC908, + 0xBB67AE8584CAA73B, + 0x3C6EF372FE94F82B, + 0xA54FF53A5F1D36F1, + 0x510E527FADE682D1, + 0x9B05688C2B3E6C1F, + 0x1F83D9ABFB41BD6B, + 0x5BE0CD19137E2179, + ] + sha_info["count_lo"] = 0 + sha_info["count_hi"] = 0 + sha_info["local"] = 0 + sha_info["digestsize"] = 64 + return sha_info + + +def sha384_init(): + """Initialize a SHA384 digest.""" + sha_info = new_shaobject() + sha_info["digest"] = [ + 0xCBBB9D5DC1059ED8, + 0x629A292A367CD507, + 0x9159015A3070DD17, + 0x152FECD8F70E5939, + 0x67332667FFC00B31, + 0x8EB44A8768581511, + 0xDB0C2E0D64F98FA7, + 0x47B5481DBEFA4FA4, + ] + sha_info["count_lo"] = 0 + sha_info["count_hi"] = 0 + sha_info["local"] = 0 + sha_info["digestsize"] = 48 + return sha_info + + +def getbuf(s): + if isinstance(s, str): + return s.encode("ascii") + return bytes(s) + + +def sha_update(sha_info, buffer): + """Update the SHA digest. + :param dict sha_info: SHA Digest. + :param str buffer: SHA buffer size. + """ + if isinstance(buffer, str): + raise TypeError("Unicode strings must be encoded before hashing") + count = len(buffer) + buffer_idx = 0 + clo = (sha_info["count_lo"] + (count << 3)) & 0xFFFFFFFF + if clo < sha_info["count_lo"]: + sha_info["count_hi"] += 1 + sha_info["count_lo"] = clo + + sha_info["count_hi"] += count >> 29 + + if sha_info["local"]: + i = SHA_BLOCKSIZE - sha_info["local"] + if i > count: + i = count + + # copy buffer + for x in enumerate(buffer[buffer_idx : buffer_idx + i]): + sha_info["data"][sha_info["local"] + x[0]] = x[1] + + count -= i + buffer_idx += i + + sha_info["local"] += i + if sha_info["local"] == SHA_BLOCKSIZE: + sha_transform(sha_info) + sha_info["local"] = 0 + else: + return + + while count >= SHA_BLOCKSIZE: + # copy buffer + sha_info["data"] = list(buffer[buffer_idx : buffer_idx + SHA_BLOCKSIZE]) + count -= SHA_BLOCKSIZE + buffer_idx += SHA_BLOCKSIZE + sha_transform(sha_info) + + # copy buffer + pos = sha_info["local"] + sha_info["data"][pos : pos + count] = list(buffer[buffer_idx : buffer_idx + count]) + sha_info["local"] = count + + +def sha_final(sha_info): + """Finish computing the SHA Digest.""" + lo_bit_count = sha_info["count_lo"] + hi_bit_count = sha_info["count_hi"] + count = (lo_bit_count >> 3) & 0x7F + sha_info["data"][count] = 0x80 + count += 1 + if count > SHA_BLOCKSIZE - 16: + # zero the bytes in data after the count + sha_info["data"] = sha_info["data"][:count] + ([0] * (SHA_BLOCKSIZE - count)) + sha_transform(sha_info) + # zero bytes in data + sha_info["data"] = [0] * SHA_BLOCKSIZE + else: + sha_info["data"] = sha_info["data"][:count] + ([0] * (SHA_BLOCKSIZE - count)) + + sha_info["data"][112] = 0 + sha_info["data"][113] = 0 + sha_info["data"][114] = 0 + sha_info["data"][115] = 0 + sha_info["data"][116] = 0 + sha_info["data"][117] = 0 + sha_info["data"][118] = 0 + sha_info["data"][119] = 0 + + sha_info["data"][120] = (hi_bit_count >> 24) & 0xFF + sha_info["data"][121] = (hi_bit_count >> 16) & 0xFF + sha_info["data"][122] = (hi_bit_count >> 8) & 0xFF + sha_info["data"][123] = (hi_bit_count >> 0) & 0xFF + sha_info["data"][124] = (lo_bit_count >> 24) & 0xFF + sha_info["data"][125] = (lo_bit_count >> 16) & 0xFF + sha_info["data"][126] = (lo_bit_count >> 8) & 0xFF + sha_info["data"][127] = (lo_bit_count >> 0) & 0xFF + + sha_transform(sha_info) + + dig = [] + for i in sha_info["digest"]: + dig.extend( + [ + ((i >> 56) & 0xFF), + ((i >> 48) & 0xFF), + ((i >> 40) & 0xFF), + ((i >> 32) & 0xFF), + ((i >> 24) & 0xFF), + ((i >> 16) & 0xFF), + ((i >> 8) & 0xFF), + (i & 0xFF), + ] + ) + return bytes(dig) + + +# pylint: disable=protected-access +class sha512: + digest_size = digestsize = SHA_DIGESTSIZE + block_size = SHA_BLOCKSIZE + name = "sha512" + + def __init__(self, s=None): + """Constructs a SHA512 hash object.""" + self._sha = sha_init() + if s: + sha_update(self._sha, getbuf(s)) + + def update(self, s): + """Updates the hash object with a bytes-like object, s.""" + sha_update(self._sha, getbuf(s)) + + def digest(self): + """Returns the digest of the data passed to the update() + method so far.""" + return sha_final(self._sha.copy())[: self._sha["digestsize"]] + + def hexdigest(self): + """Like digest() except the digest is returned as a string object of + double length, containing only hexadecimal digits. + """ + return "".join(["%.2x" % i for i in self.digest()]) + + def copy(self): + """Return a copy (“clone”) of the hash object.""" + new = sha512() + new._sha = self._sha.copy() + return new + + +# pylint: disable=protected-access, super-init-not-called +class sha384(sha512): + digest_size = digestsize = 48 + name = "sha384" + + def __init__(self, s=None): + """Constructs a SHA224 hash object.""" + self._sha = sha384_init() + if s: + sha_update(self._sha, getbuf(s)) + + def copy(self): + """Return a copy (“clone”) of the hash object.""" + new = sha384() + new._sha = self._sha.copy() \ No newline at end of file