Python code coverage for Modules/sha256module.c

#	count	content
1	n/a	/* SHA256 module */
2	n/a
3	n/a	/* This module provides an interface to NIST's SHA-256 and SHA-224 Algorithms */
4	n/a
5	n/a	/* See below for information about the original code this module was
6	n/a	based upon. Additional work performed by:
7	n/a
8	n/a	Andrew Kuchling (amk@amk.ca)
9	n/a	Greg Stein (gstein@lyra.org)
10	n/a	Trevor Perrin (trevp@trevp.net)
11	n/a
12	n/a	Copyright (C) 2005-2007 Gregory P. Smith (greg@krypto.org)
13	n/a	Licensed to PSF under a Contributor Agreement.
14	n/a
15	n/a	*/
16	n/a
17	n/a	/* SHA objects */
18	n/a
19	n/a	#include "Python.h"
20	n/a	#include "structmember.h"
21	n/a	#include "hashlib.h"
22	n/a	#include "pystrhex.h"
23	n/a
24	n/a	/*[clinic input]
25	n/a	module _sha256
26	n/a	class SHA256Type "SHAobject *" "&PyType_Type"
27	n/a	[clinic start generated code]*/
28	n/a	/[clinic end generated code: output=da39a3ee5e6b4b0d input=71a39174d4f0a744]/
29	n/a
30	n/a	/* Some useful types */
31	n/a
32	n/a	typedef unsigned char SHA_BYTE;
33	n/a
34	n/a	#if SIZEOF_INT == 4
35	n/a	typedef unsigned int SHA_INT32; /* 32-bit integer */
36	n/a	#else
37	n/a	/* not defined. compilation will die. */
38	n/a	#endif
39	n/a
40	n/a	/* The SHA block size and message digest sizes, in bytes */
41	n/a
42	n/a	#define SHA_BLOCKSIZE 64
43	n/a	#define SHA_DIGESTSIZE 32
44	n/a
45	n/a	/* The structure for storing SHA info */
46	n/a
47	n/a	typedef struct {
48	n/a	PyObject_HEAD
49	n/a	SHA_INT32 digest[8]; /* Message digest */
50	n/a	SHA_INT32 count_lo, count_hi; /* 64-bit bit count */
51	n/a	SHA_BYTE data[SHA_BLOCKSIZE]; /* SHA data buffer */
52	n/a	int local; /* unprocessed amount in data */
53	n/a	int digestsize;
54	n/a	} SHAobject;
55	n/a
56	n/a	#include "clinic/sha256module.c.h"
57	n/a
58	n/a	/* When run on a little-endian CPU we need to perform byte reversal on an
59	n/a	array of longwords. */
60	n/a
61	n/a	#if PY_LITTLE_ENDIAN
62	n/a	static void longReverse(SHA_INT32 *buffer, int byteCount)
63	n/a	{
64	n/a	SHA_INT32 value;
65	n/a
66	n/a	byteCount /= sizeof(*buffer);
67	n/a	while (byteCount--) {
68	n/a	value = *buffer;
69	n/a	value = ( ( value & 0xFF00FF00L ) >> 8 ) \| \
70	n/a	( ( value & 0x00FF00FFL ) << 8 );
71	n/a	*buffer++ = ( value << 16 ) \| ( value >> 16 );
72	n/a	}
73	n/a	}
74	n/a	#endif
75	n/a
76	n/a	static void SHAcopy(SHAobject src, SHAobject dest)
77	n/a	{
78	n/a	dest->local = src->local;
79	n/a	dest->digestsize = src->digestsize;
80	n/a	dest->count_lo = src->count_lo;
81	n/a	dest->count_hi = src->count_hi;
82	n/a	memcpy(dest->digest, src->digest, sizeof(src->digest));
83	n/a	memcpy(dest->data, src->data, sizeof(src->data));
84	n/a	}
85	n/a
86	n/a
87	n/a	/* ------------------------------------------------------------------------
88	n/a	*
89	n/a	* This code for the SHA-256 algorithm was noted as public domain. The
90	n/a	* original headers are pasted below.
91	n/a	*
92	n/a	* Several changes have been made to make it more compatible with the
93	n/a	* Python environment and desired interface.
94	n/a	*
95	n/a	*/
96	n/a
97	n/a	/* LibTomCrypt, modular cryptographic library -- Tom St Denis
98	n/a	*
99	n/a	* LibTomCrypt is a library that provides various cryptographic
100	n/a	* algorithms in a highly modular and flexible manner.
101	n/a	*
102	n/a	* The library is free for all purposes without any express
103	n/a	* guarantee it works.
104	n/a	*
105	n/a	* Tom St Denis, tomstdenis@iahu.ca, http://libtom.org
106	n/a	*/
107	n/a
108	n/a
109	n/a	/* SHA256 by Tom St Denis */
110	n/a
111	n/a	/* Various logical functions */
112	n/a	#define ROR(x, y)\
113	n/a	( ((((unsigned long)(x)&0xFFFFFFFFUL)>>(unsigned long)((y)&31)) \| \
114	n/a	((unsigned long)(x)<<(unsigned long)(32-((y)&31)))) & 0xFFFFFFFFUL)
115	n/a	#define Ch(x,y,z) (z ^ (x & (y ^ z)))
116	n/a	#define Maj(x,y,z) (((x \| y) & z) \| (x & y))
117	n/a	#define S(x, n) ROR((x),(n))
118	n/a	#define R(x, n) (((x)&0xFFFFFFFFUL)>>(n))
119	n/a	#define Sigma0(x) (S(x, 2) ^ S(x, 13) ^ S(x, 22))
120	n/a	#define Sigma1(x) (S(x, 6) ^ S(x, 11) ^ S(x, 25))
121	n/a	#define Gamma0(x) (S(x, 7) ^ S(x, 18) ^ R(x, 3))
122	n/a	#define Gamma1(x) (S(x, 17) ^ S(x, 19) ^ R(x, 10))
123	n/a
124	n/a
125	n/a	static void
126	n/a	sha_transform(SHAobject *sha_info)
127	n/a	{
128	n/a	int i;
129	n/a	SHA_INT32 S[8], W[64], t0, t1;
130	n/a
131	n/a	memcpy(W, sha_info->data, sizeof(sha_info->data));
132	n/a	#if PY_LITTLE_ENDIAN
133	n/a	longReverse(W, (int)sizeof(sha_info->data));
134	n/a	#endif
135	n/a
136	n/a	for (i = 16; i < 64; ++i) {
137	n/a	W[i] = Gamma1(W[i - 2]) + W[i - 7] + Gamma0(W[i - 15]) + W[i - 16];
138	n/a	}
139	n/a	for (i = 0; i < 8; ++i) {
140	n/a	S[i] = sha_info->digest[i];
141	n/a	}
142	n/a
143	n/a	/* Compress */
144	n/a	#define RND(a,b,c,d,e,f,g,h,i,ki) \
145	n/a	t0 = h + Sigma1(e) + Ch(e, f, g) + ki + W[i]; \
146	n/a	t1 = Sigma0(a) + Maj(a, b, c); \
147	n/a	d += t0; \
148	n/a	h = t0 + t1;
149	n/a
150	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],0,0x428a2f98);
151	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],1,0x71374491);
152	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],2,0xb5c0fbcf);
153	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],3,0xe9b5dba5);
154	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],4,0x3956c25b);
155	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],5,0x59f111f1);
156	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],6,0x923f82a4);
157	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],7,0xab1c5ed5);
158	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],8,0xd807aa98);
159	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],9,0x12835b01);
160	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],10,0x243185be);
161	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],11,0x550c7dc3);
162	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],12,0x72be5d74);
163	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],13,0x80deb1fe);
164	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],14,0x9bdc06a7);
165	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],15,0xc19bf174);
166	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],16,0xe49b69c1);
167	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],17,0xefbe4786);
168	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],18,0x0fc19dc6);
169	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],19,0x240ca1cc);
170	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],20,0x2de92c6f);
171	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],21,0x4a7484aa);
172	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],22,0x5cb0a9dc);
173	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],23,0x76f988da);
174	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],24,0x983e5152);
175	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],25,0xa831c66d);
176	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],26,0xb00327c8);
177	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],27,0xbf597fc7);
178	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],28,0xc6e00bf3);
179	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],29,0xd5a79147);
180	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],30,0x06ca6351);
181	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],31,0x14292967);
182	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],32,0x27b70a85);
183	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],33,0x2e1b2138);
184	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],34,0x4d2c6dfc);
185	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],35,0x53380d13);
186	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],36,0x650a7354);
187	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],37,0x766a0abb);
188	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],38,0x81c2c92e);
189	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],39,0x92722c85);
190	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],40,0xa2bfe8a1);
191	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],41,0xa81a664b);
192	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],42,0xc24b8b70);
193	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],43,0xc76c51a3);
194	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],44,0xd192e819);
195	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],45,0xd6990624);
196	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],46,0xf40e3585);
197	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],47,0x106aa070);
198	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],48,0x19a4c116);
199	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],49,0x1e376c08);
200	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],50,0x2748774c);
201	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],51,0x34b0bcb5);
202	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],52,0x391c0cb3);
203	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],53,0x4ed8aa4a);
204	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],54,0x5b9cca4f);
205	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],55,0x682e6ff3);
206	n/a	RND(S[0],S[1],S[2],S[3],S[4],S[5],S[6],S[7],56,0x748f82ee);
207	n/a	RND(S[7],S[0],S[1],S[2],S[3],S[4],S[5],S[6],57,0x78a5636f);
208	n/a	RND(S[6],S[7],S[0],S[1],S[2],S[3],S[4],S[5],58,0x84c87814);
209	n/a	RND(S[5],S[6],S[7],S[0],S[1],S[2],S[3],S[4],59,0x8cc70208);
210	n/a	RND(S[4],S[5],S[6],S[7],S[0],S[1],S[2],S[3],60,0x90befffa);
211	n/a	RND(S[3],S[4],S[5],S[6],S[7],S[0],S[1],S[2],61,0xa4506ceb);
212	n/a	RND(S[2],S[3],S[4],S[5],S[6],S[7],S[0],S[1],62,0xbef9a3f7);
213	n/a	RND(S[1],S[2],S[3],S[4],S[5],S[6],S[7],S[0],63,0xc67178f2);
214	n/a
215	n/a	#undef RND
216	n/a
217	n/a	/* feedback */
218	n/a	for (i = 0; i < 8; i++) {
219	n/a	sha_info->digest[i] = sha_info->digest[i] + S[i];
220	n/a	}
221	n/a
222	n/a	}
223	n/a
224	n/a
225	n/a
226	n/a	/* initialize the SHA digest */
227	n/a
228	n/a	static void
229	n/a	sha_init(SHAobject *sha_info)
230	n/a	{
231	n/a	sha_info->digest[0] = 0x6A09E667L;
232	n/a	sha_info->digest[1] = 0xBB67AE85L;
233	n/a	sha_info->digest[2] = 0x3C6EF372L;
234	n/a	sha_info->digest[3] = 0xA54FF53AL;
235	n/a	sha_info->digest[4] = 0x510E527FL;
236	n/a	sha_info->digest[5] = 0x9B05688CL;
237	n/a	sha_info->digest[6] = 0x1F83D9ABL;
238	n/a	sha_info->digest[7] = 0x5BE0CD19L;
239	n/a	sha_info->count_lo = 0L;
240	n/a	sha_info->count_hi = 0L;
241	n/a	sha_info->local = 0;
242	n/a	sha_info->digestsize = 32;
243	n/a	}
244	n/a
245	n/a	static void
246	n/a	sha224_init(SHAobject *sha_info)
247	n/a	{
248	n/a	sha_info->digest[0] = 0xc1059ed8L;
249	n/a	sha_info->digest[1] = 0x367cd507L;
250	n/a	sha_info->digest[2] = 0x3070dd17L;
251	n/a	sha_info->digest[3] = 0xf70e5939L;
252	n/a	sha_info->digest[4] = 0xffc00b31L;
253	n/a	sha_info->digest[5] = 0x68581511L;
254	n/a	sha_info->digest[6] = 0x64f98fa7L;
255	n/a	sha_info->digest[7] = 0xbefa4fa4L;
256	n/a	sha_info->count_lo = 0L;
257	n/a	sha_info->count_hi = 0L;
258	n/a	sha_info->local = 0;
259	n/a	sha_info->digestsize = 28;
260	n/a	}
261	n/a
262	n/a
263	n/a	/* update the SHA digest */
264	n/a
265	n/a	static void
266	n/a	sha_update(SHAobject sha_info, SHA_BYTE buffer, Py_ssize_t count)
267	n/a	{
268	n/a	Py_ssize_t i;
269	n/a	SHA_INT32 clo;
270	n/a
271	n/a	clo = sha_info->count_lo + ((SHA_INT32) count << 3);
272	n/a	if (clo < sha_info->count_lo) {
273	n/a	++sha_info->count_hi;
274	n/a	}
275	n/a	sha_info->count_lo = clo;
276	n/a	sha_info->count_hi += (SHA_INT32) count >> 29;
277	n/a	if (sha_info->local) {
278	n/a	i = SHA_BLOCKSIZE - sha_info->local;
279	n/a	if (i > count) {
280	n/a	i = count;
281	n/a	}
282	n/a	memcpy(((SHA_BYTE *) sha_info->data) + sha_info->local, buffer, i);
283	n/a	count -= i;
284	n/a	buffer += i;
285	n/a	sha_info->local += (int)i;
286	n/a	if (sha_info->local == SHA_BLOCKSIZE) {
287	n/a	sha_transform(sha_info);
288	n/a	}
289	n/a	else {
290	n/a	return;
291	n/a	}
292	n/a	}
293	n/a	while (count >= SHA_BLOCKSIZE) {
294	n/a	memcpy(sha_info->data, buffer, SHA_BLOCKSIZE);
295	n/a	buffer += SHA_BLOCKSIZE;
296	n/a	count -= SHA_BLOCKSIZE;
297	n/a	sha_transform(sha_info);
298	n/a	}
299	n/a	memcpy(sha_info->data, buffer, count);
300	n/a	sha_info->local = (int)count;
301	n/a	}
302	n/a
303	n/a	/* finish computing the SHA digest */
304	n/a
305	n/a	static void
306	n/a	sha_final(unsigned char digest[SHA_DIGESTSIZE], SHAobject *sha_info)
307	n/a	{
308	n/a	int count;
309	n/a	SHA_INT32 lo_bit_count, hi_bit_count;
310	n/a
311	n/a	lo_bit_count = sha_info->count_lo;
312	n/a	hi_bit_count = sha_info->count_hi;
313	n/a	count = (int) ((lo_bit_count >> 3) & 0x3f);
314	n/a	((SHA_BYTE *) sha_info->data)[count++] = 0x80;
315	n/a	if (count > SHA_BLOCKSIZE - 8) {
316	n/a	memset(((SHA_BYTE *) sha_info->data) + count, 0,
317	n/a	SHA_BLOCKSIZE - count);
318	n/a	sha_transform(sha_info);
319	n/a	memset((SHA_BYTE *) sha_info->data, 0, SHA_BLOCKSIZE - 8);
320	n/a	}
321	n/a	else {
322	n/a	memset(((SHA_BYTE *) sha_info->data) + count, 0,
323	n/a	SHA_BLOCKSIZE - 8 - count);
324	n/a	}
325	n/a
326	n/a	/* GJS: note that we add the hi/lo in big-endian. sha_transform will
327	n/a	swap these values into host-order. */
328	n/a	sha_info->data[56] = (hi_bit_count >> 24) & 0xff;
329	n/a	sha_info->data[57] = (hi_bit_count >> 16) & 0xff;
330	n/a	sha_info->data[58] = (hi_bit_count >> 8) & 0xff;
331	n/a	sha_info->data[59] = (hi_bit_count >> 0) & 0xff;
332	n/a	sha_info->data[60] = (lo_bit_count >> 24) & 0xff;
333	n/a	sha_info->data[61] = (lo_bit_count >> 16) & 0xff;
334	n/a	sha_info->data[62] = (lo_bit_count >> 8) & 0xff;
335	n/a	sha_info->data[63] = (lo_bit_count >> 0) & 0xff;
336	n/a	sha_transform(sha_info);
337	n/a	digest[ 0] = (unsigned char) ((sha_info->digest[0] >> 24) & 0xff);
338	n/a	digest[ 1] = (unsigned char) ((sha_info->digest[0] >> 16) & 0xff);
339	n/a	digest[ 2] = (unsigned char) ((sha_info->digest[0] >> 8) & 0xff);
340	n/a	digest[ 3] = (unsigned char) ((sha_info->digest[0] ) & 0xff);
341	n/a	digest[ 4] = (unsigned char) ((sha_info->digest[1] >> 24) & 0xff);
342	n/a	digest[ 5] = (unsigned char) ((sha_info->digest[1] >> 16) & 0xff);
343	n/a	digest[ 6] = (unsigned char) ((sha_info->digest[1] >> 8) & 0xff);
344	n/a	digest[ 7] = (unsigned char) ((sha_info->digest[1] ) & 0xff);
345	n/a	digest[ 8] = (unsigned char) ((sha_info->digest[2] >> 24) & 0xff);
346	n/a	digest[ 9] = (unsigned char) ((sha_info->digest[2] >> 16) & 0xff);
347	n/a	digest[10] = (unsigned char) ((sha_info->digest[2] >> 8) & 0xff);
348	n/a	digest[11] = (unsigned char) ((sha_info->digest[2] ) & 0xff);
349	n/a	digest[12] = (unsigned char) ((sha_info->digest[3] >> 24) & 0xff);
350	n/a	digest[13] = (unsigned char) ((sha_info->digest[3] >> 16) & 0xff);
351	n/a	digest[14] = (unsigned char) ((sha_info->digest[3] >> 8) & 0xff);
352	n/a	digest[15] = (unsigned char) ((sha_info->digest[3] ) & 0xff);
353	n/a	digest[16] = (unsigned char) ((sha_info->digest[4] >> 24) & 0xff);
354	n/a	digest[17] = (unsigned char) ((sha_info->digest[4] >> 16) & 0xff);
355	n/a	digest[18] = (unsigned char) ((sha_info->digest[4] >> 8) & 0xff);
356	n/a	digest[19] = (unsigned char) ((sha_info->digest[4] ) & 0xff);
357	n/a	digest[20] = (unsigned char) ((sha_info->digest[5] >> 24) & 0xff);
358	n/a	digest[21] = (unsigned char) ((sha_info->digest[5] >> 16) & 0xff);
359	n/a	digest[22] = (unsigned char) ((sha_info->digest[5] >> 8) & 0xff);
360	n/a	digest[23] = (unsigned char) ((sha_info->digest[5] ) & 0xff);
361	n/a	digest[24] = (unsigned char) ((sha_info->digest[6] >> 24) & 0xff);
362	n/a	digest[25] = (unsigned char) ((sha_info->digest[6] >> 16) & 0xff);
363	n/a	digest[26] = (unsigned char) ((sha_info->digest[6] >> 8) & 0xff);
364	n/a	digest[27] = (unsigned char) ((sha_info->digest[6] ) & 0xff);
365	n/a	digest[28] = (unsigned char) ((sha_info->digest[7] >> 24) & 0xff);
366	n/a	digest[29] = (unsigned char) ((sha_info->digest[7] >> 16) & 0xff);
367	n/a	digest[30] = (unsigned char) ((sha_info->digest[7] >> 8) & 0xff);
368	n/a	digest[31] = (unsigned char) ((sha_info->digest[7] ) & 0xff);
369	n/a	}
370	n/a
371	n/a	/*
372	n/a	* End of copied SHA code.
373	n/a	*
374	n/a	* ------------------------------------------------------------------------
375	n/a	*/
376	n/a
377	n/a	static PyTypeObject SHA224type;
378	n/a	static PyTypeObject SHA256type;
379	n/a
380	n/a
381	n/a	static SHAobject *
382	n/a	newSHA224object(void)
383	n/a	{
384	n/a	return (SHAobject *)PyObject_New(SHAobject, &SHA224type);
385	n/a	}
386	n/a
387	n/a	static SHAobject *
388	n/a	newSHA256object(void)
389	n/a	{
390	n/a	return (SHAobject *)PyObject_New(SHAobject, &SHA256type);
391	n/a	}
392	n/a
393	n/a	/* Internal methods for a hash object */
394	n/a
395	n/a	static void
396	n/a	SHA_dealloc(PyObject *ptr)
397	n/a	{
398	n/a	PyObject_Del(ptr);
399	n/a	}
400	n/a
401	n/a
402	n/a	/* External methods for a hash object */
403	n/a
404	n/a	/*[clinic input]
405	n/a	SHA256Type.copy
406	n/a
407	n/a	Return a copy of the hash object.
408	n/a	[clinic start generated code]*/
409	n/a
410	n/a	static PyObject *
411	n/a	SHA256Type_copy_impl(SHAobject *self)
412	n/a	/[clinic end generated code: output=1a8bbd66a0c9c168 input=f58840a618d4f2a7]/
413	n/a	{
414	n/a	SHAobject *newobj;
415	n/a
416	n/a	if (Py_TYPE(self) == &SHA256type) {
417	n/a	if ( (newobj = newSHA256object())==NULL)
418	n/a	return NULL;
419	n/a	} else {
420	n/a	if ( (newobj = newSHA224object())==NULL)
421	n/a	return NULL;
422	n/a	}
423	n/a
424	n/a	SHAcopy(self, newobj);
425	n/a	return (PyObject *)newobj;
426	n/a	}
427	n/a
428	n/a	/*[clinic input]
429	n/a	SHA256Type.digest
430	n/a
431	n/a	Return the digest value as a string of binary data.
432	n/a	[clinic start generated code]*/
433	n/a
434	n/a	static PyObject *
435	n/a	SHA256Type_digest_impl(SHAobject *self)
436	n/a	/[clinic end generated code: output=46616a5e909fbc3d input=1fb752e58954157d]/
437	n/a	{
438	n/a	unsigned char digest[SHA_DIGESTSIZE];
439	n/a	SHAobject temp;
440	n/a
441	n/a	SHAcopy(self, &temp);
442	n/a	sha_final(digest, &temp);
443	n/a	return PyBytes_FromStringAndSize((const char *)digest, self->digestsize);
444	n/a	}
445	n/a
446	n/a	/*[clinic input]
447	n/a	SHA256Type.hexdigest
448	n/a
449	n/a	Return the digest value as a string of hexadecimal digits.
450	n/a	[clinic start generated code]*/
451	n/a
452	n/a	static PyObject *
453	n/a	SHA256Type_hexdigest_impl(SHAobject *self)
454	n/a	/[clinic end generated code: output=725f8a7041ae97f3 input=0cc4c714693010d1]/
455	n/a	{
456	n/a	unsigned char digest[SHA_DIGESTSIZE];
457	n/a	SHAobject temp;
458	n/a
459	n/a	/* Get the raw (binary) digest value */
460	n/a	SHAcopy(self, &temp);
461	n/a	sha_final(digest, &temp);
462	n/a
463	n/a	return _Py_strhex((const char *)digest, self->digestsize);
464	n/a	}
465	n/a
466	n/a	/*[clinic input]
467	n/a	SHA256Type.update
468	n/a
469	n/a	obj: object
470	n/a	/
471	n/a
472	n/a	Update this hash object's state with the provided string.
473	n/a	[clinic start generated code]*/
474	n/a
475	n/a	static PyObject *
476	n/a	SHA256Type_update(SHAobject self, PyObject obj)
477	n/a	/[clinic end generated code: output=0967fb2860c66af7 input=b2d449d5b30f0f5a]/
478	n/a	{
479	n/a	Py_buffer buf;
480	n/a
481	n/a	GET_BUFFER_VIEW_OR_ERROUT(obj, &buf);
482	n/a
483	n/a	sha_update(self, buf.buf, buf.len);
484	n/a
485	n/a	PyBuffer_Release(&buf);
486	n/a	Py_RETURN_NONE;
487	n/a	}
488	n/a
489	n/a	static PyMethodDef SHA_methods[] = {
490	n/a	SHA256TYPE_COPY_METHODDEF
491	n/a	SHA256TYPE_DIGEST_METHODDEF
492	n/a	SHA256TYPE_HEXDIGEST_METHODDEF
493	n/a	SHA256TYPE_UPDATE_METHODDEF
494	n/a	{NULL, NULL} /* sentinel */
495	n/a	};
496	n/a
497	n/a	static PyObject *
498	n/a	SHA256_get_block_size(PyObject self, void closure)
499	n/a	{
500	n/a	return PyLong_FromLong(SHA_BLOCKSIZE);
501	n/a	}
502	n/a
503	n/a	static PyObject *
504	n/a	SHA256_get_name(PyObject self, void closure)
505	n/a	{
506	n/a	if (((SHAobject *)self)->digestsize == 32)
507	n/a	return PyUnicode_FromStringAndSize("sha256", 6);
508	n/a	else
509	n/a	return PyUnicode_FromStringAndSize("sha224", 6);
510	n/a	}
511	n/a
512	n/a	static PyGetSetDef SHA_getseters[] = {
513	n/a	{"block_size",
514	n/a	(getter)SHA256_get_block_size, NULL,
515	n/a	NULL,
516	n/a	NULL},
517	n/a	{"name",
518	n/a	(getter)SHA256_get_name, NULL,
519	n/a	NULL,
520	n/a	NULL},
521	n/a	{NULL} /* Sentinel */
522	n/a	};
523	n/a
524	n/a	static PyMemberDef SHA_members[] = {
525	n/a	{"digest_size", T_INT, offsetof(SHAobject, digestsize), READONLY, NULL},
526	n/a	{NULL} /* Sentinel */
527	n/a	};
528	n/a
529	n/a	static PyTypeObject SHA224type = {
530	n/a	PyVarObject_HEAD_INIT(NULL, 0)
531	n/a	"_sha256.sha224", /tp_name/
532	n/a	sizeof(SHAobject), /tp_size/
533	n/a	0, /tp_itemsize/
534	n/a	/* methods */
535	n/a	SHA_dealloc, /tp_dealloc/
536	n/a	0, /tp_print/
537	n/a	0, /tp_getattr/
538	n/a	0, /tp_setattr/
539	n/a	0, /tp_reserved/
540	n/a	0, /tp_repr/
541	n/a	0, /tp_as_number/
542	n/a	0, /tp_as_sequence/
543	n/a	0, /tp_as_mapping/
544	n/a	0, /tp_hash/
545	n/a	0, /tp_call/
546	n/a	0, /tp_str/
547	n/a	0, /tp_getattro/
548	n/a	0, /tp_setattro/
549	n/a	0, /tp_as_buffer/
550	n/a	Py_TPFLAGS_DEFAULT, /tp_flags/
551	n/a	0, /tp_doc/
552	n/a	0, /tp_traverse/
553	n/a	0, /tp_clear/
554	n/a	0, /tp_richcompare/
555	n/a	0, /tp_weaklistoffset/
556	n/a	0, /tp_iter/
557	n/a	0, /tp_iternext/
558	n/a	SHA_methods, /* tp_methods */
559	n/a	SHA_members, /* tp_members */
560	n/a	SHA_getseters, /* tp_getset */
561	n/a	};
562	n/a
563	n/a	static PyTypeObject SHA256type = {
564	n/a	PyVarObject_HEAD_INIT(NULL, 0)
565	n/a	"_sha256.sha256", /tp_name/
566	n/a	sizeof(SHAobject), /tp_size/
567	n/a	0, /tp_itemsize/
568	n/a	/* methods */
569	n/a	SHA_dealloc, /tp_dealloc/
570	n/a	0, /tp_print/
571	n/a	0, /tp_getattr/
572	n/a	0, /tp_setattr/
573	n/a	0, /tp_reserved/
574	n/a	0, /tp_repr/
575	n/a	0, /tp_as_number/
576	n/a	0, /tp_as_sequence/
577	n/a	0, /tp_as_mapping/
578	n/a	0, /tp_hash/
579	n/a	0, /tp_call/
580	n/a	0, /tp_str/
581	n/a	0, /tp_getattro/
582	n/a	0, /tp_setattro/
583	n/a	0, /tp_as_buffer/
584	n/a	Py_TPFLAGS_DEFAULT, /tp_flags/
585	n/a	0, /tp_doc/
586	n/a	0, /tp_traverse/
587	n/a	0, /tp_clear/
588	n/a	0, /tp_richcompare/
589	n/a	0, /tp_weaklistoffset/
590	n/a	0, /tp_iter/
591	n/a	0, /tp_iternext/
592	n/a	SHA_methods, /* tp_methods */
593	n/a	SHA_members, /* tp_members */
594	n/a	SHA_getseters, /* tp_getset */
595	n/a	};
596	n/a
597	n/a
598	n/a	/* The single module-level function: new() */
599	n/a
600	n/a	/*[clinic input]
601	n/a	_sha256.sha256
602	n/a
603	n/a	string: object(c_default="NULL") = b''
604	n/a
605	n/a	Return a new SHA-256 hash object; optionally initialized with a string.
606	n/a	[clinic start generated code]*/
607	n/a
608	n/a	static PyObject *
609	n/a	_sha256_sha256_impl(PyObject module, PyObject string)
610	n/a	/[clinic end generated code: output=fa644436dcea5c31 input=09cce3fb855056b2]/
611	n/a	{
612	n/a	SHAobject *new;
613	n/a	Py_buffer buf;
614	n/a
615	n/a	if (string)
616	n/a	GET_BUFFER_VIEW_OR_ERROUT(string, &buf);
617	n/a
618	n/a	if ((new = newSHA256object()) == NULL) {
619	n/a	if (string)
620	n/a	PyBuffer_Release(&buf);
621	n/a	return NULL;
622	n/a	}
623	n/a
624	n/a	sha_init(new);
625	n/a
626	n/a	if (PyErr_Occurred()) {
627	n/a	Py_DECREF(new);
628	n/a	if (string)
629	n/a	PyBuffer_Release(&buf);
630	n/a	return NULL;
631	n/a	}
632	n/a	if (string) {
633	n/a	sha_update(new, buf.buf, buf.len);
634	n/a	PyBuffer_Release(&buf);
635	n/a	}
636	n/a
637	n/a	return (PyObject *)new;
638	n/a	}
639	n/a
640	n/a	/*[clinic input]
641	n/a	_sha256.sha224
642	n/a
643	n/a	string: object(c_default="NULL") = b''
644	n/a
645	n/a	Return a new SHA-224 hash object; optionally initialized with a string.
646	n/a	[clinic start generated code]*/
647	n/a
648	n/a	static PyObject *
649	n/a	_sha256_sha224_impl(PyObject module, PyObject string)
650	n/a	/[clinic end generated code: output=21e3ba22c3404f93 input=27a04ba24c353a73]/
651	n/a	{
652	n/a	SHAobject *new;
653	n/a	Py_buffer buf;
654	n/a
655	n/a	if (string)
656	n/a	GET_BUFFER_VIEW_OR_ERROUT(string, &buf);
657	n/a
658	n/a	if ((new = newSHA224object()) == NULL) {
659	n/a	if (string)
660	n/a	PyBuffer_Release(&buf);
661	n/a	return NULL;
662	n/a	}
663	n/a
664	n/a	sha224_init(new);
665	n/a
666	n/a	if (PyErr_Occurred()) {
667	n/a	Py_DECREF(new);
668	n/a	if (string)
669	n/a	PyBuffer_Release(&buf);
670	n/a	return NULL;
671	n/a	}
672	n/a	if (string) {
673	n/a	sha_update(new, buf.buf, buf.len);
674	n/a	PyBuffer_Release(&buf);
675	n/a	}
676	n/a
677	n/a	return (PyObject *)new;
678	n/a	}
679	n/a
680	n/a
681	n/a	/* List of functions exported by this module */
682	n/a
683	n/a	static struct PyMethodDef SHA_functions[] = {
684	n/a	_SHA256_SHA256_METHODDEF
685	n/a	_SHA256_SHA224_METHODDEF
686	n/a	{NULL, NULL} /* Sentinel */
687	n/a	};
688	n/a
689	n/a
690	n/a	/* Initialize this module. */
691	n/a
692	n/a	#define insint(n,v) { PyModule_AddIntConstant(m,n,v); }
693	n/a
694	n/a
695	n/a	static struct PyModuleDef _sha256module = {
696	n/a	PyModuleDef_HEAD_INIT,
697	n/a	"_sha256",
698	n/a	NULL,
699	n/a	-1,
700	n/a	SHA_functions,
701	n/a	NULL,
702	n/a	NULL,
703	n/a	NULL,
704	n/a	NULL
705	n/a	};
706	n/a
707	n/a	PyMODINIT_FUNC
708	n/a	PyInit__sha256(void)
709	n/a	{
710	n/a	PyObject *m;
711	n/a
712	n/a	Py_TYPE(&SHA224type) = &PyType_Type;
713	n/a	if (PyType_Ready(&SHA224type) < 0)
714	n/a	return NULL;
715	n/a	Py_TYPE(&SHA256type) = &PyType_Type;
716	n/a	if (PyType_Ready(&SHA256type) < 0)
717	n/a	return NULL;
718	n/a
719	n/a	m = PyModule_Create(&_sha256module);
720	n/a	if (m == NULL)
721	n/a	return NULL;
722	n/a
723	n/a	Py_INCREF((PyObject *)&SHA224type);
724	n/a	PyModule_AddObject(m, "SHA224Type", (PyObject *)&SHA224type);
725	n/a	Py_INCREF((PyObject *)&SHA256type);
726	n/a	PyModule_AddObject(m, "SHA256Type", (PyObject *)&SHA256type);
727	n/a	return m;
728	n/a
729	n/a	}