/ adafruit_hashlib / _sha512.py
_sha512.py
1 # The MIT License (MIT) 2 # 3 # Copyright (c) 2017 Paul Sokolovsky 4 # Modified by Brent Rubell for Adafruit Industries, 2019 5 # 6 # Permission is hereby granted, free of charge, to any person obtaining a copy 7 # of this software and associated documentation files (the "Software"), to deal 8 # in the Software without restriction, including without limitation the rights 9 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 10 # copies of the Software, and to permit persons to whom the Software is 11 # furnished to do so, subject to the following conditions: 12 # 13 # The above copyright notice and this permission notice shall be included in 14 # all copies or substantial portions of the Software. 15 # 16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 19 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 22 # THE SOFTWARE. 23 """ 24 `_sha512.py` 25 ====================================================== 26 SHA-512 Hash Algorithm, this code was ported from 27 CPython's sha512module.c. 28 * Author(s): Paul Sokolovsky, Brent Rubell 29 """ 30 # pylint: disable=invalid-name, unnecessary-lambda, missing-docstring, line-too-long 31 32 # SHA Block size and message digest sizes, in bytes. 33 SHA_BLOCKSIZE = 128 34 SHA_DIGESTSIZE = 64 35 36 37 def new_shaobject(): 38 """Struct. for storing SHA information.""" 39 return { 40 "digest": [0] * 8, 41 "count_lo": 0, 42 "count_hi": 0, 43 "data": [0] * SHA_BLOCKSIZE, 44 "local": 0, 45 "digestsize": 0, 46 } 47 48 49 # Various logical functions 50 ROR64 = ( 51 lambda x, y: (((x & 0xFFFFFFFFFFFFFFFF) >> (y & 63)) | (x << (64 - (y & 63)))) 52 & 0xFFFFFFFFFFFFFFFF 53 ) 54 Ch = lambda x, y, z: (z ^ (x & (y ^ z))) 55 Maj = lambda x, y, z: (((x | y) & z) | (x & y)) 56 S = lambda x, n: ROR64(x, n) 57 R = lambda x, n: (x & 0xFFFFFFFFFFFFFFFF) >> n 58 Sigma0 = lambda x: (S(x, 28) ^ S(x, 34) ^ S(x, 39)) 59 Sigma1 = lambda x: (S(x, 14) ^ S(x, 18) ^ S(x, 41)) 60 Gamma0 = lambda x: (S(x, 1) ^ S(x, 8) ^ R(x, 7)) 61 Gamma1 = lambda x: (S(x, 19) ^ S(x, 61) ^ R(x, 6)) 62 63 # pylint: disable=protected-access, too-many-statements 64 def sha_transform(sha_info): 65 W = [] 66 67 d = sha_info["data"] 68 for i in range(0, 16): 69 W.append( 70 (d[8 * i] << 56) 71 + (d[8 * i + 1] << 48) 72 + (d[8 * i + 2] << 40) 73 + (d[8 * i + 3] << 32) 74 + (d[8 * i + 4] << 24) 75 + (d[8 * i + 5] << 16) 76 + (d[8 * i + 6] << 8) 77 + d[8 * i + 7] 78 ) 79 80 for i in range(16, 80): 81 W.append( 82 (Gamma1(W[i - 2]) + W[i - 7] + Gamma0(W[i - 15]) + W[i - 16]) 83 & 0xFFFFFFFFFFFFFFFF 84 ) 85 86 ss = sha_info["digest"][:] 87 88 # pylint: disable=line-too-long, too-many-arguments 89 def RND(a, b, c, d, e, f, g, h, i, ki): 90 t0 = (h + Sigma1(e) + Ch(e, f, g) + ki + W[i]) & 0xFFFFFFFFFFFFFFFF 91 t1 = (Sigma0(a) + Maj(a, b, c)) & 0xFFFFFFFFFFFFFFFF 92 d = (d + t0) & 0xFFFFFFFFFFFFFFFF 93 h = (t0 + t1) & 0xFFFFFFFFFFFFFFFF 94 return d & 0xFFFFFFFFFFFFFFFF, h & 0xFFFFFFFFFFFFFFFF 95 96 ss[3], ss[7] = RND( 97 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 0, 0x428A2F98D728AE22 98 ) 99 ss[2], ss[6] = RND( 100 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 1, 0x7137449123EF65CD 101 ) 102 ss[1], ss[5] = RND( 103 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 2, 0xB5C0FBCFEC4D3B2F 104 ) 105 ss[0], ss[4] = RND( 106 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 3, 0xE9B5DBA58189DBBC 107 ) 108 ss[7], ss[3] = RND( 109 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 4, 0x3956C25BF348B538 110 ) 111 ss[6], ss[2] = RND( 112 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 5, 0x59F111F1B605D019 113 ) 114 ss[5], ss[1] = RND( 115 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 6, 0x923F82A4AF194F9B 116 ) 117 ss[4], ss[0] = RND( 118 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 7, 0xAB1C5ED5DA6D8118 119 ) 120 ss[3], ss[7] = RND( 121 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 8, 0xD807AA98A3030242 122 ) 123 ss[2], ss[6] = RND( 124 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 9, 0x12835B0145706FBE 125 ) 126 ss[1], ss[5] = RND( 127 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 10, 0x243185BE4EE4B28C 128 ) 129 ss[0], ss[4] = RND( 130 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 11, 0x550C7DC3D5FFB4E2 131 ) 132 ss[7], ss[3] = RND( 133 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 12, 0x72BE5D74F27B896F 134 ) 135 ss[6], ss[2] = RND( 136 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 13, 0x80DEB1FE3B1696B1 137 ) 138 ss[5], ss[1] = RND( 139 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 14, 0x9BDC06A725C71235 140 ) 141 ss[4], ss[0] = RND( 142 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 15, 0xC19BF174CF692694 143 ) 144 ss[3], ss[7] = RND( 145 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 16, 0xE49B69C19EF14AD2 146 ) 147 ss[2], ss[6] = RND( 148 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 17, 0xEFBE4786384F25E3 149 ) 150 ss[1], ss[5] = RND( 151 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 18, 0x0FC19DC68B8CD5B5 152 ) 153 ss[0], ss[4] = RND( 154 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 19, 0x240CA1CC77AC9C65 155 ) 156 ss[7], ss[3] = RND( 157 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 20, 0x2DE92C6F592B0275 158 ) 159 ss[6], ss[2] = RND( 160 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 21, 0x4A7484AA6EA6E483 161 ) 162 ss[5], ss[1] = RND( 163 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 22, 0x5CB0A9DCBD41FBD4 164 ) 165 ss[4], ss[0] = RND( 166 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 23, 0x76F988DA831153B5 167 ) 168 ss[3], ss[7] = RND( 169 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 24, 0x983E5152EE66DFAB 170 ) 171 ss[2], ss[6] = RND( 172 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 25, 0xA831C66D2DB43210 173 ) 174 ss[1], ss[5] = RND( 175 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 26, 0xB00327C898FB213F 176 ) 177 ss[0], ss[4] = RND( 178 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 27, 0xBF597FC7BEEF0EE4 179 ) 180 ss[7], ss[3] = RND( 181 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 28, 0xC6E00BF33DA88FC2 182 ) 183 ss[6], ss[2] = RND( 184 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 29, 0xD5A79147930AA725 185 ) 186 ss[5], ss[1] = RND( 187 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 30, 0x06CA6351E003826F 188 ) 189 ss[4], ss[0] = RND( 190 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 31, 0x142929670A0E6E70 191 ) 192 ss[3], ss[7] = RND( 193 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 32, 0x27B70A8546D22FFC 194 ) 195 ss[2], ss[6] = RND( 196 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 33, 0x2E1B21385C26C926 197 ) 198 ss[1], ss[5] = RND( 199 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 34, 0x4D2C6DFC5AC42AED 200 ) 201 ss[0], ss[4] = RND( 202 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 35, 0x53380D139D95B3DF 203 ) 204 ss[7], ss[3] = RND( 205 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 36, 0x650A73548BAF63DE 206 ) 207 ss[6], ss[2] = RND( 208 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 37, 0x766A0ABB3C77B2A8 209 ) 210 ss[5], ss[1] = RND( 211 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 38, 0x81C2C92E47EDAEE6 212 ) 213 ss[4], ss[0] = RND( 214 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 39, 0x92722C851482353B 215 ) 216 ss[3], ss[7] = RND( 217 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 40, 0xA2BFE8A14CF10364 218 ) 219 ss[2], ss[6] = RND( 220 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 41, 0xA81A664BBC423001 221 ) 222 ss[1], ss[5] = RND( 223 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 42, 0xC24B8B70D0F89791 224 ) 225 ss[0], ss[4] = RND( 226 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 43, 0xC76C51A30654BE30 227 ) 228 ss[7], ss[3] = RND( 229 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 44, 0xD192E819D6EF5218 230 ) 231 ss[6], ss[2] = RND( 232 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 45, 0xD69906245565A910 233 ) 234 ss[5], ss[1] = RND( 235 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 46, 0xF40E35855771202A 236 ) 237 ss[4], ss[0] = RND( 238 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 47, 0x106AA07032BBD1B8 239 ) 240 ss[3], ss[7] = RND( 241 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 48, 0x19A4C116B8D2D0C8 242 ) 243 ss[2], ss[6] = RND( 244 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 49, 0x1E376C085141AB53 245 ) 246 ss[1], ss[5] = RND( 247 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 50, 0x2748774CDF8EEB99 248 ) 249 ss[0], ss[4] = RND( 250 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 51, 0x34B0BCB5E19B48A8 251 ) 252 ss[7], ss[3] = RND( 253 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 52, 0x391C0CB3C5C95A63 254 ) 255 ss[6], ss[2] = RND( 256 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 53, 0x4ED8AA4AE3418ACB 257 ) 258 ss[5], ss[1] = RND( 259 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 54, 0x5B9CCA4F7763E373 260 ) 261 ss[4], ss[0] = RND( 262 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 55, 0x682E6FF3D6B2B8A3 263 ) 264 ss[3], ss[7] = RND( 265 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 56, 0x748F82EE5DEFB2FC 266 ) 267 ss[2], ss[6] = RND( 268 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 57, 0x78A5636F43172F60 269 ) 270 ss[1], ss[5] = RND( 271 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 58, 0x84C87814A1F0AB72 272 ) 273 ss[0], ss[4] = RND( 274 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 59, 0x8CC702081A6439EC 275 ) 276 ss[7], ss[3] = RND( 277 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 60, 0x90BEFFFA23631E28 278 ) 279 ss[6], ss[2] = RND( 280 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 61, 0xA4506CEBDE82BDE9 281 ) 282 ss[5], ss[1] = RND( 283 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 62, 0xBEF9A3F7B2C67915 284 ) 285 ss[4], ss[0] = RND( 286 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 63, 0xC67178F2E372532B 287 ) 288 ss[3], ss[7] = RND( 289 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 64, 0xCA273ECEEA26619C 290 ) 291 ss[2], ss[6] = RND( 292 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 65, 0xD186B8C721C0C207 293 ) 294 ss[1], ss[5] = RND( 295 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 66, 0xEADA7DD6CDE0EB1E 296 ) 297 ss[0], ss[4] = RND( 298 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 67, 0xF57D4F7FEE6ED178 299 ) 300 ss[7], ss[3] = RND( 301 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 68, 0x06F067AA72176FBA 302 ) 303 ss[6], ss[2] = RND( 304 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 69, 0x0A637DC5A2C898A6 305 ) 306 ss[5], ss[1] = RND( 307 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 70, 0x113F9804BEF90DAE 308 ) 309 ss[4], ss[0] = RND( 310 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 71, 0x1B710B35131C471B 311 ) 312 ss[3], ss[7] = RND( 313 ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], 72, 0x28DB77F523047D84 314 ) 315 ss[2], ss[6] = RND( 316 ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], 73, 0x32CAAB7B40C72493 317 ) 318 ss[1], ss[5] = RND( 319 ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], ss[5], 74, 0x3C9EBE0A15C9BEBC 320 ) 321 ss[0], ss[4] = RND( 322 ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], ss[4], 75, 0x431D67C49C100D4C 323 ) 324 ss[7], ss[3] = RND( 325 ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], ss[3], 76, 0x4CC5D4BECB3E42B6 326 ) 327 ss[6], ss[2] = RND( 328 ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], ss[2], 77, 0x597F299CFC657E2A 329 ) 330 ss[5], ss[1] = RND( 331 ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], ss[1], 78, 0x5FCB6FAB3AD6FAEC 332 ) 333 ss[4], ss[0] = RND( 334 ss[1], ss[2], ss[3], ss[4], ss[5], ss[6], ss[7], ss[0], 79, 0x6C44198C4A475817 335 ) 336 337 # Feedback 338 dig = [] 339 for i, x in enumerate(sha_info["digest"]): 340 dig.append((x + ss[i]) & 0xFFFFFFFFFFFFFFFF) 341 sha_info["digest"] = dig 342 343 344 def sha_init(): 345 """Initialize the SHA digest.""" 346 sha_info = new_shaobject() 347 sha_info["digest"] = [ 348 0x6A09E667F3BCC908, 349 0xBB67AE8584CAA73B, 350 0x3C6EF372FE94F82B, 351 0xA54FF53A5F1D36F1, 352 0x510E527FADE682D1, 353 0x9B05688C2B3E6C1F, 354 0x1F83D9ABFB41BD6B, 355 0x5BE0CD19137E2179, 356 ] 357 sha_info["count_lo"] = 0 358 sha_info["count_hi"] = 0 359 sha_info["local"] = 0 360 sha_info["digestsize"] = 64 361 return sha_info 362 363 364 def sha384_init(): 365 """Initialize a SHA384 digest.""" 366 sha_info = new_shaobject() 367 sha_info["digest"] = [ 368 0xCBBB9D5DC1059ED8, 369 0x629A292A367CD507, 370 0x9159015A3070DD17, 371 0x152FECD8F70E5939, 372 0x67332667FFC00B31, 373 0x8EB44A8768581511, 374 0xDB0C2E0D64F98FA7, 375 0x47B5481DBEFA4FA4, 376 ] 377 sha_info["count_lo"] = 0 378 sha_info["count_hi"] = 0 379 sha_info["local"] = 0 380 sha_info["digestsize"] = 48 381 return sha_info 382 383 384 def getbuf(s): 385 if isinstance(s, str): 386 return s.encode("ascii") 387 return bytes(s) 388 389 390 def sha_update(sha_info, buffer): 391 """Update the SHA digest. 392 :param dict sha_info: SHA Digest. 393 :param str buffer: SHA buffer size. 394 """ 395 if isinstance(buffer, str): 396 raise TypeError("Unicode strings must be encoded before hashing") 397 count = len(buffer) 398 buffer_idx = 0 399 clo = (sha_info["count_lo"] + (count << 3)) & 0xFFFFFFFF 400 if clo < sha_info["count_lo"]: 401 sha_info["count_hi"] += 1 402 sha_info["count_lo"] = clo 403 404 sha_info["count_hi"] += count >> 29 405 406 if sha_info["local"]: 407 i = SHA_BLOCKSIZE - sha_info["local"] 408 if i > count: 409 i = count 410 411 # copy buffer 412 for x in enumerate(buffer[buffer_idx : buffer_idx + i]): 413 sha_info["data"][sha_info["local"] + x[0]] = x[1] 414 415 count -= i 416 buffer_idx += i 417 418 sha_info["local"] += i 419 if sha_info["local"] == SHA_BLOCKSIZE: 420 sha_transform(sha_info) 421 sha_info["local"] = 0 422 else: 423 return 424 425 while count >= SHA_BLOCKSIZE: 426 # copy buffer 427 sha_info["data"] = list(buffer[buffer_idx : buffer_idx + SHA_BLOCKSIZE]) 428 count -= SHA_BLOCKSIZE 429 buffer_idx += SHA_BLOCKSIZE 430 sha_transform(sha_info) 431 432 # copy buffer 433 pos = sha_info["local"] 434 sha_info["data"][pos : pos + count] = list(buffer[buffer_idx : buffer_idx + count]) 435 sha_info["local"] = count 436 437 438 def sha_final(sha_info): 439 """Finish computing the SHA Digest.""" 440 lo_bit_count = sha_info["count_lo"] 441 hi_bit_count = sha_info["count_hi"] 442 count = (lo_bit_count >> 3) & 0x7F 443 sha_info["data"][count] = 0x80 444 count += 1 445 if count > SHA_BLOCKSIZE - 16: 446 # zero the bytes in data after the count 447 sha_info["data"] = sha_info["data"][:count] + ([0] * (SHA_BLOCKSIZE - count)) 448 sha_transform(sha_info) 449 # zero bytes in data 450 sha_info["data"] = [0] * SHA_BLOCKSIZE 451 else: 452 sha_info["data"] = sha_info["data"][:count] + ([0] * (SHA_BLOCKSIZE - count)) 453 454 sha_info["data"][112] = 0 455 sha_info["data"][113] = 0 456 sha_info["data"][114] = 0 457 sha_info["data"][115] = 0 458 sha_info["data"][116] = 0 459 sha_info["data"][117] = 0 460 sha_info["data"][118] = 0 461 sha_info["data"][119] = 0 462 463 sha_info["data"][120] = (hi_bit_count >> 24) & 0xFF 464 sha_info["data"][121] = (hi_bit_count >> 16) & 0xFF 465 sha_info["data"][122] = (hi_bit_count >> 8) & 0xFF 466 sha_info["data"][123] = (hi_bit_count >> 0) & 0xFF 467 sha_info["data"][124] = (lo_bit_count >> 24) & 0xFF 468 sha_info["data"][125] = (lo_bit_count >> 16) & 0xFF 469 sha_info["data"][126] = (lo_bit_count >> 8) & 0xFF 470 sha_info["data"][127] = (lo_bit_count >> 0) & 0xFF 471 472 sha_transform(sha_info) 473 474 dig = [] 475 for i in sha_info["digest"]: 476 dig.extend( 477 [ 478 ((i >> 56) & 0xFF), 479 ((i >> 48) & 0xFF), 480 ((i >> 40) & 0xFF), 481 ((i >> 32) & 0xFF), 482 ((i >> 24) & 0xFF), 483 ((i >> 16) & 0xFF), 484 ((i >> 8) & 0xFF), 485 (i & 0xFF), 486 ] 487 ) 488 return bytes(dig) 489 490 491 # pylint: disable=protected-access 492 class sha512: 493 digest_size = digestsize = SHA_DIGESTSIZE 494 block_size = SHA_BLOCKSIZE 495 name = "sha512" 496 497 def __init__(self, s=None): 498 """Constructs a SHA512 hash object. 499 """ 500 self._sha = sha_init() 501 if s: 502 sha_update(self._sha, getbuf(s)) 503 504 def update(self, s): 505 """Updates the hash object with a bytes-like object, s.""" 506 sha_update(self._sha, getbuf(s)) 507 508 def digest(self): 509 """Returns the digest of the data passed to the update() 510 method so far.""" 511 return sha_final(self._sha.copy())[: self._sha["digestsize"]] 512 513 def hexdigest(self): 514 """Like digest() except the digest is returned as a string object of 515 double length, containing only hexadecimal digits. 516 """ 517 return "".join(["%.2x" % i for i in self.digest()]) 518 519 def copy(self): 520 """Return a copy (“clone”) of the hash object. 521 """ 522 new = sha512() 523 new._sha = self._sha.copy() 524 return new 525 526 527 # pylint: disable=protected-access, super-init-not-called 528 class sha384(sha512): 529 digest_size = digestsize = 48 530 name = "sha384" 531 532 def __init__(self, s=None): 533 """Constructs a SHA224 hash object. 534 """ 535 self._sha = sha384_init() 536 if s: 537 sha_update(self._sha, getbuf(s)) 538 539 def copy(self): 540 """Return a copy (“clone”) of the hash object. 541 """ 542 new = sha384() 543 new._sha = self._sha.copy() 544 return new