Skip to content

Commit 6c0c7f8

Browse files
pfalconpfalcon
committed
hashlib: Import pure-Python sha256/512 algos from pypy3-2.4.0-src.tar.bz2 .
1 parent 24d3e81 commit 6c0c7f8

File tree

2 files changed

+540
-0
lines changed

2 files changed

+540
-0
lines changed

‎hashlib/hashlib/sha256.py

Lines changed: 256 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,256 @@
1+
SHA_BLOCKSIZE = 64
2+
SHA_DIGESTSIZE = 32
3+
4+
5+
def new_shaobject():
6+
return {
7+
'digest': [0]*8,
8+
'count_lo': 0,
9+
'count_hi': 0,
10+
'data': [0]* SHA_BLOCKSIZE,
11+
'local': 0,
12+
'digestsize': 0
13+
}
14+
15+
ROR = lambda x, y: (((x & 0xffffffff) >> (y & 31)) | (x << (32 - (y & 31)))) & 0xffffffff
16+
Ch = lambda x, y, z: (z ^ (x & (y ^ z)))
17+
Maj = lambda x, y, z: (((x | y) & z) | (x & y))
18+
S = lambda x, n: ROR(x, n)
19+
R = lambda x, n: (x & 0xffffffff) >> n
20+
Sigma0 = lambda x: (S(x, 2) ^ S(x, 13) ^ S(x, 22))
21+
Sigma1 = lambda x: (S(x, 6) ^ S(x, 11) ^ S(x, 25))
22+
Gamma0 = lambda x: (S(x, 7) ^ S(x, 18) ^ R(x, 3))
23+
Gamma1 = lambda x: (S(x, 17) ^ S(x, 19) ^ R(x, 10))
24+
25+
def sha_transform(sha_info):
26+
W = []
27+
28+
d = sha_info['data']
29+
for i in range(0,16):
30+
W.append( (d[4*i]<<24) + (d[4*i+1]<<16) + (d[4*i+2]<<8) + d[4*i+3])
31+
32+
for i in range(16,64):
33+
W.append( (Gamma1(W[i - 2]) + W[i - 7] + Gamma0(W[i - 15]) + W[i - 16]) & 0xffffffff )
34+
35+
ss = sha_info['digest'][:]
36+
37+
def RND(a,b,c,d,e,f,g,h,i,ki):
38+
t0 = h + Sigma1(e) + Ch(e, f, g) + ki + W[i];
39+
t1 = Sigma0(a) + Maj(a, b, c);
40+
d += t0;
41+
h = t0 + t1;
42+
return d & 0xffffffff, h & 0xffffffff
43+
44+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],0,0x428a2f98);
45+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],1,0x71374491);
46+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],2,0xb5c0fbcf);
47+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],3,0xe9b5dba5);
48+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],4,0x3956c25b);
49+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],5,0x59f111f1);
50+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],6,0x923f82a4);
51+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],7,0xab1c5ed5);
52+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],8,0xd807aa98);
53+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],9,0x12835b01);
54+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],10,0x243185be);
55+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],11,0x550c7dc3);
56+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],12,0x72be5d74);
57+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],13,0x80deb1fe);
58+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],14,0x9bdc06a7);
59+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],15,0xc19bf174);
60+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],16,0xe49b69c1);
61+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],17,0xefbe4786);
62+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],18,0x0fc19dc6);
63+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],19,0x240ca1cc);
64+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],20,0x2de92c6f);
65+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],21,0x4a7484aa);
66+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],22,0x5cb0a9dc);
67+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],23,0x76f988da);
68+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],24,0x983e5152);
69+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],25,0xa831c66d);
70+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],26,0xb00327c8);
71+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],27,0xbf597fc7);
72+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],28,0xc6e00bf3);
73+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],29,0xd5a79147);
74+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],30,0x06ca6351);
75+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],31,0x14292967);
76+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],32,0x27b70a85);
77+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],33,0x2e1b2138);
78+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],34,0x4d2c6dfc);
79+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],35,0x53380d13);
80+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],36,0x650a7354);
81+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],37,0x766a0abb);
82+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],38,0x81c2c92e);
83+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],39,0x92722c85);
84+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],40,0xa2bfe8a1);
85+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],41,0xa81a664b);
86+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],42,0xc24b8b70);
87+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],43,0xc76c51a3);
88+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],44,0xd192e819);
89+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],45,0xd6990624);
90+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],46,0xf40e3585);
91+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],47,0x106aa070);
92+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],48,0x19a4c116);
93+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],49,0x1e376c08);
94+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],50,0x2748774c);
95+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],51,0x34b0bcb5);
96+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],52,0x391c0cb3);
97+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],53,0x4ed8aa4a);
98+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],54,0x5b9cca4f);
99+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],55,0x682e6ff3);
100+
ss[3], ss[7] = RND(ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],56,0x748f82ee);
101+
ss[2], ss[6] = RND(ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],57,0x78a5636f);
102+
ss[1], ss[5] = RND(ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],ss[5],58,0x84c87814);
103+
ss[0], ss[4] = RND(ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],ss[4],59,0x8cc70208);
104+
ss[7], ss[3] = RND(ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],ss[3],60,0x90befffa);
105+
ss[6], ss[2] = RND(ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],ss[2],61,0xa4506ceb);
106+
ss[5], ss[1] = RND(ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],ss[1],62,0xbef9a3f7);
107+
ss[4], ss[0] = RND(ss[1],ss[2],ss[3],ss[4],ss[5],ss[6],ss[7],ss[0],63,0xc67178f2);
108+
109+
dig = []
110+
for i, x in enumerate(sha_info['digest']):
111+
dig.append( (x + ss[i]) & 0xffffffff )
112+
sha_info['digest'] = dig
113+
114+
def sha_init():
115+
sha_info = new_shaobject()
116+
sha_info['digest'] = [0x6A09E667, 0xBB67AE85, 0x3C6EF372, 0xA54FF53A, 0x510E527F, 0x9B05688C, 0x1F83D9AB, 0x5BE0CD19]
117+
sha_info['count_lo'] = 0
118+
sha_info['count_hi'] = 0
119+
sha_info['local'] = 0
120+
sha_info['digestsize'] = 32
121+
return sha_info
122+
123+
def sha224_init():
124+
sha_info = new_shaobject()
125+
sha_info['digest'] = [0xc1059ed8, 0x367cd507, 0x3070dd17, 0xf70e5939, 0xffc00b31, 0x68581511, 0x64f98fa7, 0xbefa4fa4]
126+
sha_info['count_lo'] = 0
127+
sha_info['count_hi'] = 0
128+
sha_info['local'] = 0
129+
sha_info['digestsize'] = 28
130+
return sha_info
131+
132+
def sha_update(sha_info, buffer):
133+
if isinstance(buffer, str):
134+
raise TypeError("Unicode strings must be encoded before hashing")
135+
count = len(buffer)
136+
buffer_idx = 0
137+
clo = (sha_info['count_lo'] + (count << 3)) & 0xffffffff
138+
if clo < sha_info['count_lo']:
139+
sha_info['count_hi'] += 1
140+
sha_info['count_lo'] = clo
141+
142+
sha_info['count_hi'] += (count >> 29)
143+
144+
if sha_info['local']:
145+
i = SHA_BLOCKSIZE - sha_info['local']
146+
if i > count:
147+
i = count
148+
149+
# copy buffer
150+
sha_info['data'][sha_info['local']:sha_info['local']+i] = buffer[buffer_idx:buffer_idx+i]
151+
152+
count -= i
153+
buffer_idx += i
154+
155+
sha_info['local'] += i
156+
if sha_info['local'] == SHA_BLOCKSIZE:
157+
sha_transform(sha_info)
158+
sha_info['local'] = 0
159+
else:
160+
return
161+
162+
while count >= SHA_BLOCKSIZE:
163+
# copy buffer
164+
sha_info['data'] = list(buffer[buffer_idx:buffer_idx + SHA_BLOCKSIZE])
165+
count -= SHA_BLOCKSIZE
166+
buffer_idx += SHA_BLOCKSIZE
167+
sha_transform(sha_info)
168+
169+
170+
# copy buffer
171+
pos = sha_info['local']
172+
sha_info['data'][pos:pos+count] = buffer[buffer_idx:buffer_idx + count]
173+
sha_info['local'] = count
174+
175+
def sha_final(sha_info):
176+
lo_bit_count = sha_info['count_lo']
177+
hi_bit_count = sha_info['count_hi']
178+
count = (lo_bit_count >> 3) & 0x3f
179+
sha_info['data'][count] = 0x80;
180+
count += 1
181+
if count > SHA_BLOCKSIZE - 8:
182+
# zero the bytes in data after the count
183+
sha_info['data'] = sha_info['data'][:count] + ([0] * (SHA_BLOCKSIZE - count))
184+
sha_transform(sha_info)
185+
# zero bytes in data
186+
sha_info['data'] = [0] * SHA_BLOCKSIZE
187+
else:
188+
sha_info['data'] = sha_info['data'][:count] + ([0] * (SHA_BLOCKSIZE - count))
189+
190+
sha_info['data'][56] = (hi_bit_count >> 24) & 0xff
191+
sha_info['data'][57] = (hi_bit_count >> 16) & 0xff
192+
sha_info['data'][58] = (hi_bit_count >> 8) & 0xff
193+
sha_info['data'][59] = (hi_bit_count >> 0) & 0xff
194+
sha_info['data'][60] = (lo_bit_count >> 24) & 0xff
195+
sha_info['data'][61] = (lo_bit_count >> 16) & 0xff
196+
sha_info['data'][62] = (lo_bit_count >> 8) & 0xff
197+
sha_info['data'][63] = (lo_bit_count >> 0) & 0xff
198+
199+
sha_transform(sha_info)
200+
201+
dig = []
202+
for i in sha_info['digest']:
203+
dig.extend([ ((i>>24) & 0xff), ((i>>16) & 0xff), ((i>>8) & 0xff), (i & 0xff) ])
204+
return ''.join([chr(i) for i in dig])
205+
206+
class sha256(object):
207+
digest_size = digestsize = SHA_DIGESTSIZE
208+
block_size = SHA_BLOCKSIZE
209+
210+
def __init__(self, s=None):
211+
self._sha = sha_init()
212+
if s:
213+
sha_update(self._sha, s)
214+
215+
def update(self, s):
216+
sha_update(self._sha, s)
217+
218+
def digest(self):
219+
return sha_final(self._sha.copy())[:self._sha['digestsize']]
220+
221+
def hexdigest(self):
222+
return ''.join(['%.2x' % ord(i) for i in self.digest()])
223+
224+
def copy(self):
225+
new = sha256.__new__(sha256)
226+
new._sha = self._sha.copy()
227+
return new
228+
229+
class sha224(sha256):
230+
digest_size = digestsize = 28
231+
232+
def __init__(self, s=None):
233+
self._sha = sha224_init()
234+
if s:
235+
sha_update(self._sha, s)
236+
237+
def copy(self):
238+
new = sha224.__new__(sha224)
239+
new._sha = self._sha.copy()
240+
return new
241+
242+
def test():
243+
a_str = "just a test string"
244+
245+
assert 'e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855' == sha256().hexdigest()
246+
assert 'd7b553c6f09ac85d142415f857c5310f3bbbe7cdd787cce4b985acedd585266f' == sha256(a_str).hexdigest()
247+
assert '8113ebf33c97daa9998762aacafe750c7cefc2b2f173c90c59663a57fe626f21' == sha256(a_str*7).hexdigest()
248+
249+
s = sha256(a_str)
250+
s.update(a_str)
251+
assert '03d9963e05a094593190b6fc794cb1a3e1ac7d7883f0b5855268afeccc70d461' == s.hexdigest()
252+
253+
if __name__ == "__main__":
254+
test()
255+
256+

0 commit comments

Comments
 (0)