blob: 86f602ebe5ef9895523909004e67515bae3696a1 [file] [log] [blame]
Gregory P. Smith2f21eb32007-09-09 06:44:34 +00001/* MD5 module */
2
3/* This module provides an interface to the MD5 algorithm */
4
5/* See below for information about the original code this module was
6 based upon. Additional work performed by:
7
8 Andrew Kuchling (amk@amk.ca)
9 Greg Stein (gstein@lyra.org)
10 Trevor Perrin (trevp@trevp.net)
11
12 Copyright (C) 2005-2007 Gregory P. Smith (greg@krypto.org)
13 Licensed to PSF under a Contributor Agreement.
14
15*/
16
17/* MD5 objects */
18
19#include "Python.h"
Gregory P. Smith365a1862009-02-12 07:35:29 +000020#include "hashlib.h"
Gregory P. Smith2f21eb32007-09-09 06:44:34 +000021
22
23/* Some useful types */
24
25#if SIZEOF_INT == 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000026typedef unsigned int MD5_INT32; /* 32-bit integer */
27typedef PY_LONG_LONG MD5_INT64; /* 64-bit integer */
Gregory P. Smith2f21eb32007-09-09 06:44:34 +000028#else
29/* not defined. compilation will die. */
30#endif
31
32/* The MD5 block size and message digest sizes, in bytes */
33
34#define MD5_BLOCKSIZE 64
35#define MD5_DIGESTSIZE 16
36
37/* The structure for storing MD5 info */
38
39struct md5_state {
40 MD5_INT64 length;
41 MD5_INT32 state[4], curlen;
42 unsigned char buf[MD5_BLOCKSIZE];
43};
44
45typedef struct {
46 PyObject_HEAD
47
48 struct md5_state hash_state;
49} MD5object;
50
51
52/* ------------------------------------------------------------------------
53 *
54 * This code for the MD5 algorithm was noted as public domain. The
55 * original headers are pasted below.
56 *
57 * Several changes have been made to make it more compatible with the
58 * Python environment and desired interface.
59 *
60 */
61
62/* LibTomCrypt, modular cryptographic library -- Tom St Denis
63 *
64 * LibTomCrypt is a library that provides various cryptographic
65 * algorithms in a highly modular and flexible manner.
66 *
67 * The library is free for all purposes without any express
68 * guarantee it works.
69 *
70 * Tom St Denis, tomstdenis@gmail.com, http://libtom.org
71 */
72
73/* rotate the hard way (platform optimizations could be done) */
74#define ROLc(x, y) ( (((unsigned long)(x)<<(unsigned long)((y)&31)) | (((unsigned long)(x)&0xFFFFFFFFUL)>>(unsigned long)(32-((y)&31)))) & 0xFFFFFFFFUL)
75
76/* Endian Neutral macros that work on all platforms */
77
78#define STORE32L(x, y) \
79 { (y)[3] = (unsigned char)(((x)>>24)&255); (y)[2] = (unsigned char)(((x)>>16)&255); \
80 (y)[1] = (unsigned char)(((x)>>8)&255); (y)[0] = (unsigned char)((x)&255); }
81
82#define LOAD32L(x, y) \
83 { x = ((unsigned long)((y)[3] & 255)<<24) | \
84 ((unsigned long)((y)[2] & 255)<<16) | \
85 ((unsigned long)((y)[1] & 255)<<8) | \
86 ((unsigned long)((y)[0] & 255)); }
87
88#define STORE64L(x, y) \
89 { (y)[7] = (unsigned char)(((x)>>56)&255); (y)[6] = (unsigned char)(((x)>>48)&255); \
90 (y)[5] = (unsigned char)(((x)>>40)&255); (y)[4] = (unsigned char)(((x)>>32)&255); \
91 (y)[3] = (unsigned char)(((x)>>24)&255); (y)[2] = (unsigned char)(((x)>>16)&255); \
92 (y)[1] = (unsigned char)(((x)>>8)&255); (y)[0] = (unsigned char)((x)&255); }
93
94#ifndef MIN
95 #define MIN(x, y) ( ((x)<(y))?(x):(y) )
96#endif
97
98
99/* MD5 macros */
100
101#define F(x,y,z) (z ^ (x & (y ^ z)))
102#define G(x,y,z) (y ^ (z & (y ^ x)))
103#define H(x,y,z) (x^y^z)
104#define I(x,y,z) (y^(x|(~z)))
105
106#define FF(a,b,c,d,M,s,t) \
107 a = (a + F(b,c,d) + M + t); a = ROLc(a, s) + b;
108
109#define GG(a,b,c,d,M,s,t) \
110 a = (a + G(b,c,d) + M + t); a = ROLc(a, s) + b;
111
112#define HH(a,b,c,d,M,s,t) \
113 a = (a + H(b,c,d) + M + t); a = ROLc(a, s) + b;
114
115#define II(a,b,c,d,M,s,t) \
116 a = (a + I(b,c,d) + M + t); a = ROLc(a, s) + b;
117
118
119static void md5_compress(struct md5_state *md5, unsigned char *buf)
120{
121 MD5_INT32 i, W[16], a, b, c, d;
122
123 assert(md5 != NULL);
124 assert(buf != NULL);
125
126 /* copy the state into 512-bits into W[0..15] */
127 for (i = 0; i < 16; i++) {
128 LOAD32L(W[i], buf + (4*i));
129 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000130
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000131 /* copy state */
132 a = md5->state[0];
133 b = md5->state[1];
134 c = md5->state[2];
135 d = md5->state[3];
136
137 FF(a,b,c,d,W[0],7,0xd76aa478UL)
138 FF(d,a,b,c,W[1],12,0xe8c7b756UL)
139 FF(c,d,a,b,W[2],17,0x242070dbUL)
140 FF(b,c,d,a,W[3],22,0xc1bdceeeUL)
141 FF(a,b,c,d,W[4],7,0xf57c0fafUL)
142 FF(d,a,b,c,W[5],12,0x4787c62aUL)
143 FF(c,d,a,b,W[6],17,0xa8304613UL)
144 FF(b,c,d,a,W[7],22,0xfd469501UL)
145 FF(a,b,c,d,W[8],7,0x698098d8UL)
146 FF(d,a,b,c,W[9],12,0x8b44f7afUL)
147 FF(c,d,a,b,W[10],17,0xffff5bb1UL)
148 FF(b,c,d,a,W[11],22,0x895cd7beUL)
149 FF(a,b,c,d,W[12],7,0x6b901122UL)
150 FF(d,a,b,c,W[13],12,0xfd987193UL)
151 FF(c,d,a,b,W[14],17,0xa679438eUL)
152 FF(b,c,d,a,W[15],22,0x49b40821UL)
153 GG(a,b,c,d,W[1],5,0xf61e2562UL)
154 GG(d,a,b,c,W[6],9,0xc040b340UL)
155 GG(c,d,a,b,W[11],14,0x265e5a51UL)
156 GG(b,c,d,a,W[0],20,0xe9b6c7aaUL)
157 GG(a,b,c,d,W[5],5,0xd62f105dUL)
158 GG(d,a,b,c,W[10],9,0x02441453UL)
159 GG(c,d,a,b,W[15],14,0xd8a1e681UL)
160 GG(b,c,d,a,W[4],20,0xe7d3fbc8UL)
161 GG(a,b,c,d,W[9],5,0x21e1cde6UL)
162 GG(d,a,b,c,W[14],9,0xc33707d6UL)
163 GG(c,d,a,b,W[3],14,0xf4d50d87UL)
164 GG(b,c,d,a,W[8],20,0x455a14edUL)
165 GG(a,b,c,d,W[13],5,0xa9e3e905UL)
166 GG(d,a,b,c,W[2],9,0xfcefa3f8UL)
167 GG(c,d,a,b,W[7],14,0x676f02d9UL)
168 GG(b,c,d,a,W[12],20,0x8d2a4c8aUL)
169 HH(a,b,c,d,W[5],4,0xfffa3942UL)
170 HH(d,a,b,c,W[8],11,0x8771f681UL)
171 HH(c,d,a,b,W[11],16,0x6d9d6122UL)
172 HH(b,c,d,a,W[14],23,0xfde5380cUL)
173 HH(a,b,c,d,W[1],4,0xa4beea44UL)
174 HH(d,a,b,c,W[4],11,0x4bdecfa9UL)
175 HH(c,d,a,b,W[7],16,0xf6bb4b60UL)
176 HH(b,c,d,a,W[10],23,0xbebfbc70UL)
177 HH(a,b,c,d,W[13],4,0x289b7ec6UL)
178 HH(d,a,b,c,W[0],11,0xeaa127faUL)
179 HH(c,d,a,b,W[3],16,0xd4ef3085UL)
180 HH(b,c,d,a,W[6],23,0x04881d05UL)
181 HH(a,b,c,d,W[9],4,0xd9d4d039UL)
182 HH(d,a,b,c,W[12],11,0xe6db99e5UL)
183 HH(c,d,a,b,W[15],16,0x1fa27cf8UL)
184 HH(b,c,d,a,W[2],23,0xc4ac5665UL)
185 II(a,b,c,d,W[0],6,0xf4292244UL)
186 II(d,a,b,c,W[7],10,0x432aff97UL)
187 II(c,d,a,b,W[14],15,0xab9423a7UL)
188 II(b,c,d,a,W[5],21,0xfc93a039UL)
189 II(a,b,c,d,W[12],6,0x655b59c3UL)
190 II(d,a,b,c,W[3],10,0x8f0ccc92UL)
191 II(c,d,a,b,W[10],15,0xffeff47dUL)
192 II(b,c,d,a,W[1],21,0x85845dd1UL)
193 II(a,b,c,d,W[8],6,0x6fa87e4fUL)
194 II(d,a,b,c,W[15],10,0xfe2ce6e0UL)
195 II(c,d,a,b,W[6],15,0xa3014314UL)
196 II(b,c,d,a,W[13],21,0x4e0811a1UL)
197 II(a,b,c,d,W[4],6,0xf7537e82UL)
198 II(d,a,b,c,W[11],10,0xbd3af235UL)
199 II(c,d,a,b,W[2],15,0x2ad7d2bbUL)
200 II(b,c,d,a,W[9],21,0xeb86d391UL)
201
202 md5->state[0] = md5->state[0] + a;
203 md5->state[1] = md5->state[1] + b;
204 md5->state[2] = md5->state[2] + c;
205 md5->state[3] = md5->state[3] + d;
206}
207
208
209/**
210 Initialize the hash state
211 @param sha1 The hash state you wish to initialize
212*/
213void md5_init(struct md5_state *md5)
214{
215 assert(md5 != NULL);
216 md5->state[0] = 0x67452301UL;
217 md5->state[1] = 0xefcdab89UL;
218 md5->state[2] = 0x98badcfeUL;
219 md5->state[3] = 0x10325476UL;
220 md5->curlen = 0;
221 md5->length = 0;
222}
223
224/**
225 Process a block of memory though the hash
226 @param sha1 The hash state
227 @param in The data to hash
228 @param inlen The length of the data (octets)
229*/
230void md5_process(struct md5_state *md5,
Victor Stinner0fcab4a2011-01-04 12:59:15 +0000231 const unsigned char *in, Py_ssize_t inlen)
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000232{
Victor Stinner0fcab4a2011-01-04 12:59:15 +0000233 Py_ssize_t n;
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000234
235 assert(md5 != NULL);
236 assert(in != NULL);
237 assert(md5->curlen <= sizeof(md5->buf));
238
239 while (inlen > 0) {
240 if (md5->curlen == 0 && inlen >= MD5_BLOCKSIZE) {
241 md5_compress(md5, (unsigned char *)in);
242 md5->length += MD5_BLOCKSIZE * 8;
243 in += MD5_BLOCKSIZE;
244 inlen -= MD5_BLOCKSIZE;
245 } else {
Victor Stinner081fe462011-07-08 01:10:28 +0200246 n = MIN(inlen, (Py_ssize_t)(MD5_BLOCKSIZE - md5->curlen));
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000247 memcpy(md5->buf + md5->curlen, in, (size_t)n);
248 md5->curlen += n;
249 in += n;
250 inlen -= n;
251 if (md5->curlen == MD5_BLOCKSIZE) {
252 md5_compress(md5, md5->buf);
253 md5->length += 8*MD5_BLOCKSIZE;
254 md5->curlen = 0;
255 }
256 }
257 }
258}
259
260/**
261 Terminate the hash to get the digest
262 @param sha1 The hash state
263 @param out [out] The destination of the hash (16 bytes)
264*/
265void md5_done(struct md5_state *md5, unsigned char *out)
266{
267 int i;
268
269 assert(md5 != NULL);
270 assert(out != NULL);
271 assert(md5->curlen < sizeof(md5->buf));
272
273 /* increase the length of the message */
274 md5->length += md5->curlen * 8;
275
276 /* append the '1' bit */
277 md5->buf[md5->curlen++] = (unsigned char)0x80;
278
279 /* if the length is currently above 56 bytes we append zeros
280 * then compress. Then we can fall back to padding zeros and length
281 * encoding like normal.
282 */
283 if (md5->curlen > 56) {
284 while (md5->curlen < 64) {
285 md5->buf[md5->curlen++] = (unsigned char)0;
286 }
287 md5_compress(md5, md5->buf);
288 md5->curlen = 0;
289 }
290
291 /* pad upto 56 bytes of zeroes */
292 while (md5->curlen < 56) {
293 md5->buf[md5->curlen++] = (unsigned char)0;
294 }
295
296 /* store length */
297 STORE64L(md5->length, md5->buf+56);
298 md5_compress(md5, md5->buf);
299
300 /* copy output */
301 for (i = 0; i < 4; i++) {
302 STORE32L(md5->state[i], out+(4*i));
303 }
304}
305
306/* .Source: /cvs/libtom/libtomcrypt/src/hashes/md5.c,v $ */
307/* .Revision: 1.10 $ */
308/* .Date: 2007/05/12 14:25:28 $ */
309
310/*
311 * End of copied MD5 code.
312 *
313 * ------------------------------------------------------------------------
314 */
315
316static PyTypeObject MD5type;
317
318
319static MD5object *
320newMD5object(void)
321{
322 return (MD5object *)PyObject_New(MD5object, &MD5type);
323}
324
325
326/* Internal methods for a hash object */
327
328static void
329MD5_dealloc(PyObject *ptr)
330{
331 PyObject_Del(ptr);
332}
333
334
335/* External methods for a hash object */
336
337PyDoc_STRVAR(MD5_copy__doc__, "Return a copy of the hash object.");
338
339static PyObject *
340MD5_copy(MD5object *self, PyObject *unused)
341{
342 MD5object *newobj;
343
Christian Heimes90aa7642007-12-19 02:45:37 +0000344 if (Py_TYPE(self) == &MD5type) {
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000345 if ( (newobj = newMD5object())==NULL)
346 return NULL;
347 } else {
348 if ( (newobj = newMD5object())==NULL)
349 return NULL;
350 }
351
352 newobj->hash_state = self->hash_state;
353 return (PyObject *)newobj;
354}
355
356PyDoc_STRVAR(MD5_digest__doc__,
357"Return the digest value as a string of binary data.");
358
359static PyObject *
360MD5_digest(MD5object *self, PyObject *unused)
361{
362 unsigned char digest[MD5_DIGESTSIZE];
363 struct md5_state temp;
364
365 temp = self->hash_state;
366 md5_done(&temp, digest);
Christian Heimes72b710a2008-05-26 13:28:38 +0000367 return PyBytes_FromStringAndSize((const char *)digest, MD5_DIGESTSIZE);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000368}
369
370PyDoc_STRVAR(MD5_hexdigest__doc__,
371"Return the digest value as a string of hexadecimal digits.");
372
373static PyObject *
374MD5_hexdigest(MD5object *self, PyObject *unused)
375{
376 unsigned char digest[MD5_DIGESTSIZE];
377 struct md5_state temp;
378 PyObject *retval;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200379 Py_UCS1 *hex_digest;
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000380 int i, j;
381
382 /* Get the raw (binary) digest value */
383 temp = self->hash_state;
384 md5_done(&temp, digest);
385
386 /* Create a new string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200387 retval = PyUnicode_New(MD5_DIGESTSIZE * 2, 127);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000388 if (!retval)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000389 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200390 hex_digest = PyUnicode_1BYTE_DATA(retval);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000391
392 /* Make hex version of the digest */
393 for(i=j=0; i<MD5_DIGESTSIZE; i++) {
Victor Stinnerf5cff562011-10-14 02:13:11 +0200394 unsigned char c;
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000395 c = (digest[i] >> 4) & 0xf;
Victor Stinnerf5cff562011-10-14 02:13:11 +0200396 hex_digest[j++] = Py_hexdigits[c];
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000397 c = (digest[i] & 0xf);
Victor Stinnerf5cff562011-10-14 02:13:11 +0200398 hex_digest[j++] = Py_hexdigits[c];
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000399 }
400 return retval;
401}
402
403PyDoc_STRVAR(MD5_update__doc__,
404"Update this hash object's state with the provided string.");
405
406static PyObject *
407MD5_update(MD5object *self, PyObject *args)
408{
Gregory P. Smith365a1862009-02-12 07:35:29 +0000409 PyObject *obj;
Martin v. Löwis7b9cb252008-08-14 15:52:23 +0000410 Py_buffer buf;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000411
Gregory P. Smith365a1862009-02-12 07:35:29 +0000412 if (!PyArg_ParseTuple(args, "O:update", &obj))
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000413 return NULL;
414
Gregory P. Smith365a1862009-02-12 07:35:29 +0000415 GET_BUFFER_VIEW_OR_ERROUT(obj, &buf);
416
Martin v. Löwis7b9cb252008-08-14 15:52:23 +0000417 md5_process(&self->hash_state, buf.buf, buf.len);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000418
Martin v. Löwis7b9cb252008-08-14 15:52:23 +0000419 PyBuffer_Release(&buf);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000420 Py_INCREF(Py_None);
421 return Py_None;
422}
423
424static PyMethodDef MD5_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000425 {"copy", (PyCFunction)MD5_copy, METH_NOARGS, MD5_copy__doc__},
426 {"digest", (PyCFunction)MD5_digest, METH_NOARGS, MD5_digest__doc__},
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000427 {"hexdigest", (PyCFunction)MD5_hexdigest, METH_NOARGS, MD5_hexdigest__doc__},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000428 {"update", (PyCFunction)MD5_update, METH_VARARGS, MD5_update__doc__},
429 {NULL, NULL} /* sentinel */
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000430};
431
432static PyObject *
433MD5_get_block_size(PyObject *self, void *closure)
434{
Christian Heimes217cfd12007-12-02 14:31:20 +0000435 return PyLong_FromLong(MD5_BLOCKSIZE);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000436}
437
438static PyObject *
439MD5_get_name(PyObject *self, void *closure)
440{
441 return PyUnicode_FromStringAndSize("MD5", 3);
442}
443
444static PyObject *
445md5_get_digest_size(PyObject *self, void *closure)
446{
Christian Heimes217cfd12007-12-02 14:31:20 +0000447 return PyLong_FromLong(MD5_DIGESTSIZE);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000448}
449
450
451static PyGetSetDef MD5_getseters[] = {
452 {"block_size",
453 (getter)MD5_get_block_size, NULL,
454 NULL,
455 NULL},
456 {"name",
457 (getter)MD5_get_name, NULL,
458 NULL,
459 NULL},
460 {"digest_size",
461 (getter)md5_get_digest_size, NULL,
462 NULL,
463 NULL},
464 {NULL} /* Sentinel */
465};
466
467static PyTypeObject MD5type = {
468 PyVarObject_HEAD_INIT(NULL, 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000469 "_md5.md5", /*tp_name*/
470 sizeof(MD5object), /*tp_size*/
471 0, /*tp_itemsize*/
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000472 /* methods */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 MD5_dealloc, /*tp_dealloc*/
474 0, /*tp_print*/
475 0, /*tp_getattr*/
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000476 0, /*tp_setattr*/
Mark Dickinsone94c6792009-02-02 20:36:42 +0000477 0, /*tp_reserved*/
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000478 0, /*tp_repr*/
479 0, /*tp_as_number*/
480 0, /*tp_as_sequence*/
481 0, /*tp_as_mapping*/
482 0, /*tp_hash*/
483 0, /*tp_call*/
484 0, /*tp_str*/
485 0, /*tp_getattro*/
486 0, /*tp_setattro*/
487 0, /*tp_as_buffer*/
488 Py_TPFLAGS_DEFAULT, /*tp_flags*/
489 0, /*tp_doc*/
490 0, /*tp_traverse*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000491 0, /*tp_clear*/
492 0, /*tp_richcompare*/
493 0, /*tp_weaklistoffset*/
494 0, /*tp_iter*/
495 0, /*tp_iternext*/
496 MD5_methods, /* tp_methods */
497 NULL, /* tp_members */
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000498 MD5_getseters, /* tp_getset */
499};
500
501
502/* The single module-level function: new() */
503
504PyDoc_STRVAR(MD5_new__doc__,
505"Return a new MD5 hash object; optionally initialized with a string.");
506
507static PyObject *
508MD5_new(PyObject *self, PyObject *args, PyObject *kwdict)
509{
510 static char *kwlist[] = {"string", NULL};
511 MD5object *new;
Gregory P. Smith365a1862009-02-12 07:35:29 +0000512 PyObject *data_obj = NULL;
Martin v. Löwis7b9cb252008-08-14 15:52:23 +0000513 Py_buffer buf;
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000514
Gregory P. Smith365a1862009-02-12 07:35:29 +0000515 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "|O:new", kwlist,
516 &data_obj)) {
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000517 return NULL;
518 }
519
Gregory P. Smith365a1862009-02-12 07:35:29 +0000520 if (data_obj)
521 GET_BUFFER_VIEW_OR_ERROUT(data_obj, &buf);
522
Hirokazu Yamamoto84047492009-03-03 07:49:01 +0000523 if ((new = newMD5object()) == NULL) {
524 if (data_obj)
525 PyBuffer_Release(&buf);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000526 return NULL;
Hirokazu Yamamoto84047492009-03-03 07:49:01 +0000527 }
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000528
529 md5_init(&new->hash_state);
530
531 if (PyErr_Occurred()) {
532 Py_DECREF(new);
Hirokazu Yamamoto84047492009-03-03 07:49:01 +0000533 if (data_obj)
534 PyBuffer_Release(&buf);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000535 return NULL;
536 }
Gregory P. Smith365a1862009-02-12 07:35:29 +0000537 if (data_obj) {
Martin v. Löwis7b9cb252008-08-14 15:52:23 +0000538 md5_process(&new->hash_state, buf.buf, buf.len);
Hirokazu Yamamoto84047492009-03-03 07:49:01 +0000539 PyBuffer_Release(&buf);
Martin v. Löwis7b9cb252008-08-14 15:52:23 +0000540 }
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000541
542 return (PyObject *)new;
543}
544
545
546/* List of functions exported by this module */
547
548static struct PyMethodDef MD5_functions[] = {
549 {"md5", (PyCFunction)MD5_new, METH_VARARGS|METH_KEYWORDS, MD5_new__doc__},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000550 {NULL, NULL} /* Sentinel */
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000551};
552
553
554/* Initialize this module. */
555
556#define insint(n,v) { PyModule_AddIntConstant(m,n,v); }
557
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000558
Martin v. Löwis1a214512008-06-11 05:26:20 +0000559static struct PyModuleDef _md5module = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 PyModuleDef_HEAD_INIT,
561 "_md5",
562 NULL,
563 -1,
564 MD5_functions,
565 NULL,
566 NULL,
567 NULL,
568 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +0000569};
570
571PyMODINIT_FUNC
572PyInit__md5(void)
573{
Christian Heimes90aa7642007-12-19 02:45:37 +0000574 Py_TYPE(&MD5type) = &PyType_Type;
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000575 if (PyType_Ready(&MD5type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +0000576 return NULL;
Martin v. Löwis3447bee2008-06-11 05:37:58 +0000577 return PyModule_Create(&_md5module);
Gregory P. Smith2f21eb32007-09-09 06:44:34 +0000578}