-
Notifications
You must be signed in to change notification settings - Fork 1
/
sha.py
47 lines (43 loc) · 1.58 KB
/
sha.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from SHA_Utils.help_function import initializer, chunker, preprocessMessage, b2Tob16
from SHA_Utils.simple_function import *
from SHA_Utils.const import *
def sha256(message):
k = initializer(K)
h0, h1, h2, h3, h4, h5, h6, h7 = initializer(H)
chunks = preprocessMessage(message)
for chunk in chunks:
w = chunker(chunk, 32)
for _ in range(48):
w.append(32 * [0])
for i in range(16, 64):
s0 = XORXOR(rotr(w[i-15], 7), rotr(w[i-15], 18), shr(w[i-15], 3) )
s1 = XORXOR(rotr(w[i-2], 17), rotr(w[i-2], 19), shr(w[i-2], 10))
w[i] = add(add(add(w[i-16], s0), w[i-7]), s1)
a, b, c, d, e, f, g, h = h0, h1, h2, h3, h4, h5, h6, h7
for j in range(64):
S1 = XORXOR(rotr(e, 6), rotr(e, 11), rotr(e, 25) )
ch = XOR(AND(e, f), AND(NOT(e), g))
temp1 = add(add(add(add(h, S1), ch), k[j]), w[j])
S0 = XORXOR(rotr(a, 2), rotr(a, 13), rotr(a, 22))
m = XORXOR(AND(a, b), AND(a, c), AND(b, c))
temp2 = add(S0, m)
h = g
g = f
f = e
e = add(d, temp1)
d = c
c = b
b = a
a = add(temp1, temp2)
h0 = add(h0, a)
h1 = add(h1, b)
h2 = add(h2, c)
h3 = add(h3, d)
h4 = add(h4, e)
h5 = add(h5, f)
h6 = add(h6, g)
h7 = add(h7, h)
digest = ''
for val in [h0, h1, h2, h3, h4, h5, h6, h7]:
digest += b2Tob16(val)
return digest