Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 1 | |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 2 | /* MD5 module */ |
| 3 | |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 4 | /* This module provides an interface to the RSA Data Security, |
| 5 | Inc. MD5 Message-Digest Algorithm, described in RFC 1321. |
| 6 | It requires the files md5c.c and md5.h (which are slightly changed |
| 7 | from the versions in the RFC to avoid the "global.h" file.) */ |
| 8 | |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 9 | |
| 10 | /* MD5 objects */ |
| 11 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 12 | #include "Python.h" |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 13 | #include "structmember.h" |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 14 | #include "md5.h" |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 15 | #include "hashlib.h" |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 16 | |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 17 | typedef struct { |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 18 | PyObject_HEAD |
Matthias Klose | 8e39ec7 | 2006-04-03 16:27:50 +0000 | [diff] [blame] | 19 | md5_state_t md5; /* the context holder */ |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 20 | } md5object; |
| 21 | |
Jeremy Hylton | 938ace6 | 2002-07-17 16:30:39 +0000 | [diff] [blame] | 22 | static PyTypeObject MD5type; |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 23 | |
| 24 | #define is_md5object(v) ((v)->ob_type == &MD5type) |
| 25 | |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 26 | static md5object * |
Thomas Wouters | f3f33dc | 2000-07-21 06:00:07 +0000 | [diff] [blame] | 27 | newmd5object(void) |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 28 | { |
| 29 | md5object *md5p; |
| 30 | |
Guido van Rossum | b18618d | 2000-05-03 23:44:39 +0000 | [diff] [blame] | 31 | md5p = PyObject_New(md5object, &MD5type); |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 32 | if (md5p == NULL) |
| 33 | return NULL; |
| 34 | |
Matthias Klose | 8e39ec7 | 2006-04-03 16:27:50 +0000 | [diff] [blame] | 35 | md5_init(&md5p->md5); /* actual initialisation */ |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 36 | return md5p; |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 37 | } |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 38 | |
| 39 | |
| 40 | /* MD5 methods */ |
| 41 | |
| 42 | static void |
Peter Schneider-Kamp | 286da3b | 2000-07-10 12:43:58 +0000 | [diff] [blame] | 43 | md5_dealloc(md5object *md5p) |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 44 | { |
Guido van Rossum | b18618d | 2000-05-03 23:44:39 +0000 | [diff] [blame] | 45 | PyObject_Del(md5p); |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 46 | } |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 47 | |
| 48 | |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 49 | /* MD5 methods-as-attributes */ |
| 50 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 51 | static PyObject * |
Peter Schneider-Kamp | 286da3b | 2000-07-10 12:43:58 +0000 | [diff] [blame] | 52 | md5_update(md5object *self, PyObject *args) |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 53 | { |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 54 | PyObject *data_obj; |
| 55 | Py_buffer view; |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 56 | |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 57 | if (!PyArg_ParseTuple(args, "O:update", &data_obj)) |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 58 | return NULL; |
| 59 | |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 60 | GET_BUFFER_VIEW_OR_ERROUT(data_obj, &view, NULL); |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 61 | |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 62 | md5_append(&self->md5, (unsigned char*)view.buf, |
| 63 | Py_SAFE_DOWNCAST(view.len, Py_ssize_t, unsigned int)); |
| 64 | |
| 65 | PyBuffer_Release(&view); |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 66 | Py_INCREF(Py_None); |
| 67 | return Py_None; |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 68 | } |
| 69 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 70 | PyDoc_STRVAR(update_doc, |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 71 | "update (arg)\n\ |
| 72 | \n\ |
| 73 | Update the md5 object with the string arg. Repeated calls are\n\ |
| 74 | equivalent to a single call with the concatenation of all the\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 75 | arguments."); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 76 | |
| 77 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 78 | static PyObject * |
Neal Norwitz | 3a6f978 | 2002-03-25 20:46:46 +0000 | [diff] [blame] | 79 | md5_digest(md5object *self) |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 80 | { |
Matthias Klose | 8e39ec7 | 2006-04-03 16:27:50 +0000 | [diff] [blame] | 81 | md5_state_t mdContext; |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 82 | unsigned char aDigest[16]; |
| 83 | |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 84 | /* make a temporary copy, and perform the final */ |
| 85 | mdContext = self->md5; |
Matthias Klose | 8e39ec7 | 2006-04-03 16:27:50 +0000 | [diff] [blame] | 86 | md5_finish(&mdContext, aDigest); |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 87 | |
Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 88 | return PyString_FromStringAndSize((char *)aDigest, 16); |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 89 | } |
| 90 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 91 | PyDoc_STRVAR(digest_doc, |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 92 | "digest() -> string\n\ |
| 93 | \n\ |
| 94 | Return the digest of the strings passed to the update() method so\n\ |
Skip Montanaro | bd2de3a | 2003-07-08 21:17:25 +0000 | [diff] [blame] | 95 | far. This is a 16-byte string which may contain non-ASCII characters,\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 96 | including null bytes."); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 97 | |
| 98 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 99 | static PyObject * |
Neal Norwitz | 3a6f978 | 2002-03-25 20:46:46 +0000 | [diff] [blame] | 100 | md5_hexdigest(md5object *self) |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 101 | { |
Matthias Klose | 8e39ec7 | 2006-04-03 16:27:50 +0000 | [diff] [blame] | 102 | md5_state_t mdContext; |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 103 | unsigned char digest[16]; |
| 104 | unsigned char hexdigest[32]; |
| 105 | int i, j; |
| 106 | |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 107 | /* make a temporary copy, and perform the final */ |
| 108 | mdContext = self->md5; |
Matthias Klose | 8e39ec7 | 2006-04-03 16:27:50 +0000 | [diff] [blame] | 109 | md5_finish(&mdContext, digest); |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 110 | |
| 111 | /* Make hex version of the digest */ |
| 112 | for(i=j=0; i<16; i++) { |
| 113 | char c; |
| 114 | c = (digest[i] >> 4) & 0xf; |
| 115 | c = (c>9) ? c+'a'-10 : c + '0'; |
| 116 | hexdigest[j++] = c; |
| 117 | c = (digest[i] & 0xf); |
| 118 | c = (c>9) ? c+'a'-10 : c + '0'; |
| 119 | hexdigest[j++] = c; |
| 120 | } |
Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 121 | return PyString_FromStringAndSize((char*)hexdigest, 32); |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 122 | } |
| 123 | |
| 124 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 125 | PyDoc_STRVAR(hexdigest_doc, |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 126 | "hexdigest() -> string\n\ |
| 127 | \n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 128 | Like digest(), but returns the digest as a string of hexadecimal digits."); |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 129 | |
| 130 | |
| 131 | static PyObject * |
Neal Norwitz | 3a6f978 | 2002-03-25 20:46:46 +0000 | [diff] [blame] | 132 | md5_copy(md5object *self) |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 133 | { |
| 134 | md5object *md5p; |
| 135 | |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 136 | if ((md5p = newmd5object()) == NULL) |
| 137 | return NULL; |
| 138 | |
| 139 | md5p->md5 = self->md5; |
| 140 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 141 | return (PyObject *)md5p; |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 142 | } |
| 143 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 144 | PyDoc_STRVAR(copy_doc, |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 145 | "copy() -> md5 object\n\ |
| 146 | \n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 147 | Return a copy (``clone'') of the md5 object."); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 148 | |
| 149 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 150 | static PyMethodDef md5_methods[] = { |
Neal Norwitz | ba3a16c | 2002-03-31 15:27:00 +0000 | [diff] [blame] | 151 | {"update", (PyCFunction)md5_update, METH_VARARGS, update_doc}, |
Neal Norwitz | 3a6f978 | 2002-03-25 20:46:46 +0000 | [diff] [blame] | 152 | {"digest", (PyCFunction)md5_digest, METH_NOARGS, digest_doc}, |
| 153 | {"hexdigest", (PyCFunction)md5_hexdigest, METH_NOARGS, hexdigest_doc}, |
| 154 | {"copy", (PyCFunction)md5_copy, METH_NOARGS, copy_doc}, |
Barry Warsaw | 3ec0fbb | 2000-08-15 05:59:44 +0000 | [diff] [blame] | 155 | {NULL, NULL} /* sentinel */ |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 156 | }; |
| 157 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 158 | static PyObject * |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 159 | md5_get_block_size(PyObject *self, void *closure) |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 160 | { |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 161 | return PyInt_FromLong(64); |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 162 | } |
| 163 | |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 164 | static PyObject * |
| 165 | md5_get_digest_size(PyObject *self, void *closure) |
| 166 | { |
| 167 | return PyInt_FromLong(16); |
| 168 | } |
| 169 | |
| 170 | static PyObject * |
| 171 | md5_get_name(PyObject *self, void *closure) |
| 172 | { |
Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 173 | return PyString_FromStringAndSize("MD5", 3); |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 174 | } |
| 175 | |
| 176 | static PyGetSetDef md5_getseters[] = { |
| 177 | {"digest_size", |
| 178 | (getter)md5_get_digest_size, NULL, |
| 179 | NULL, |
| 180 | NULL}, |
| 181 | {"block_size", |
| 182 | (getter)md5_get_block_size, NULL, |
| 183 | NULL, |
| 184 | NULL}, |
| 185 | {"name", |
| 186 | (getter)md5_get_name, NULL, |
| 187 | NULL, |
| 188 | NULL}, |
| 189 | /* the old md5 and sha modules support 'digest_size' as in PEP 247. |
| 190 | * the old sha module also supported 'digestsize'. ugh. */ |
| 191 | {"digestsize", |
| 192 | (getter)md5_get_digest_size, NULL, |
| 193 | NULL, |
| 194 | NULL}, |
| 195 | {NULL} /* Sentinel */ |
| 196 | }; |
| 197 | |
| 198 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 199 | PyDoc_STRVAR(module_doc, |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 200 | "This module implements the interface to RSA's MD5 message digest\n\ |
| 201 | algorithm (see also Internet RFC 1321). Its use is quite\n\ |
| 202 | straightforward: use the new() to create an md5 object. You can now\n\ |
| 203 | feed this object with arbitrary strings using the update() method, and\n\ |
| 204 | at any point you can ask it for the digest (a strong kind of 128-bit\n\ |
Thomas Wouters | 7e47402 | 2000-07-16 12:04:32 +0000 | [diff] [blame] | 205 | checksum, a.k.a. ``fingerprint'') of the concatenation of the strings\n\ |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 206 | fed to it so far using the digest() method.\n\ |
| 207 | \n\ |
| 208 | Functions:\n\ |
| 209 | \n\ |
| 210 | new([arg]) -- return a new md5 object, initialized with arg if provided\n\ |
| 211 | md5([arg]) -- DEPRECATED, same as new, but for compatibility\n\ |
| 212 | \n\ |
| 213 | Special Objects:\n\ |
| 214 | \n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 215 | MD5Type -- type object for md5 objects"); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 216 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 217 | PyDoc_STRVAR(md5type_doc, |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 218 | "An md5 represents the object used to calculate the MD5 checksum of a\n\ |
| 219 | string of information.\n\ |
| 220 | \n\ |
| 221 | Methods:\n\ |
| 222 | \n\ |
| 223 | update() -- updates the current digest with an additional string\n\ |
| 224 | digest() -- return the current digest value\n\ |
Neal Norwitz | 98a355d | 2002-10-11 21:53:01 +0000 | [diff] [blame] | 225 | hexdigest() -- return the current digest as a string of hexadecimal digits\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 226 | copy() -- return a copy of the current md5 object"); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 227 | |
Tim Peters | 0c32279 | 2002-07-17 16:49:03 +0000 | [diff] [blame] | 228 | static PyTypeObject MD5type = { |
Martin v. Löwis | 6819210 | 2007-07-21 06:55:02 +0000 | [diff] [blame] | 229 | PyVarObject_HEAD_INIT(NULL, 0) |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 230 | "_md5.md5", /*tp_name*/ |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 231 | sizeof(md5object), /*tp_size*/ |
| 232 | 0, /*tp_itemsize*/ |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 233 | /* methods */ |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 234 | (destructor)md5_dealloc, /*tp_dealloc*/ |
| 235 | 0, /*tp_print*/ |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 236 | 0, /*tp_getattr*/ |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 237 | 0, /*tp_setattr*/ |
| 238 | 0, /*tp_compare*/ |
| 239 | 0, /*tp_repr*/ |
| 240 | 0, /*tp_as_number*/ |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 241 | 0, /*tp_as_sequence*/ |
| 242 | 0, /*tp_as_mapping*/ |
| 243 | 0, /*tp_hash*/ |
| 244 | 0, /*tp_call*/ |
| 245 | 0, /*tp_str*/ |
| 246 | 0, /*tp_getattro*/ |
| 247 | 0, /*tp_setattro*/ |
| 248 | 0, /*tp_as_buffer*/ |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 249 | Py_TPFLAGS_DEFAULT, /*tp_flags*/ |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 250 | md5type_doc, /*tp_doc*/ |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 251 | 0, /*tp_traverse*/ |
| 252 | 0, /*tp_clear*/ |
| 253 | 0, /*tp_richcompare*/ |
| 254 | 0, /*tp_weaklistoffset*/ |
| 255 | 0, /*tp_iter*/ |
| 256 | 0, /*tp_iternext*/ |
| 257 | md5_methods, /*tp_methods*/ |
| 258 | 0, /*tp_members*/ |
| 259 | md5_getseters, /*tp_getset*/ |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 260 | }; |
| 261 | |
| 262 | |
| 263 | /* MD5 functions */ |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 264 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 265 | static PyObject * |
Peter Schneider-Kamp | 286da3b | 2000-07-10 12:43:58 +0000 | [diff] [blame] | 266 | MD5_new(PyObject *self, PyObject *args) |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 267 | { |
| 268 | md5object *md5p; |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 269 | PyObject *data_obj = NULL; |
| 270 | Py_buffer view; |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 271 | |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 272 | if (!PyArg_ParseTuple(args, "|O:new", &data_obj)) |
Guido van Rossum | ef38b78 | 1995-07-26 17:33:10 +0000 | [diff] [blame] | 273 | return NULL; |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 274 | |
Kristján Valur Jónsson | 7705d0a | 2009-03-03 03:20:42 +0000 | [diff] [blame] | 275 | if (data_obj) |
| 276 | GET_BUFFER_VIEW_OR_ERROUT(data_obj, &view, NULL); |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 277 | |
| 278 | if ((md5p = newmd5object()) == NULL) { |
Kristján Valur Jónsson | 7705d0a | 2009-03-03 03:20:42 +0000 | [diff] [blame] | 279 | if (data_obj) |
| 280 | PyBuffer_Release(&view); |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 281 | return NULL; |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 282 | } |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 283 | |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 284 | if (data_obj) { |
| 285 | md5_append(&md5p->md5, (unsigned char*)view.buf, |
| 286 | Py_SAFE_DOWNCAST(view.len, Py_ssize_t, unsigned int)); |
Kristján Valur Jónsson | 7705d0a | 2009-03-03 03:20:42 +0000 | [diff] [blame] | 287 | PyBuffer_Release(&view); |
Gregory P. Smith | ea38826 | 2009-02-13 03:00:00 +0000 | [diff] [blame] | 288 | } |
Kristján Valur Jónsson | 7705d0a | 2009-03-03 03:20:42 +0000 | [diff] [blame] | 289 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 290 | return (PyObject *)md5p; |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 291 | } |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 292 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 293 | PyDoc_STRVAR(new_doc, |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 294 | "new([arg]) -> md5 object\n\ |
| 295 | \n\ |
| 296 | Return a new md5 object. If arg is present, the method call update(arg)\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 297 | is made."); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 298 | |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 299 | |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 300 | /* List of functions exported by this module */ |
| 301 | |
Barry Warsaw | 8b43b19 | 1996-12-09 22:32:36 +0000 | [diff] [blame] | 302 | static PyMethodDef md5_functions[] = { |
Andrew M. Kuchling | e365fb8 | 2000-08-03 02:06:16 +0000 | [diff] [blame] | 303 | {"new", (PyCFunction)MD5_new, METH_VARARGS, new_doc}, |
Guido van Rossum | ef38b78 | 1995-07-26 17:33:10 +0000 | [diff] [blame] | 304 | {NULL, NULL} /* Sentinel */ |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 305 | }; |
| 306 | |
| 307 | |
| 308 | /* Initialize this module. */ |
| 309 | |
Mark Hammond | fe51c6d | 2002-08-02 02:27:13 +0000 | [diff] [blame] | 310 | PyMODINIT_FUNC |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 311 | init_md5(void) |
Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 312 | { |
Fred Drake | 52a42e9 | 2001-11-02 22:05:06 +0000 | [diff] [blame] | 313 | PyObject *m, *d; |
Fred Drake | 0d40ba4 | 2000-02-04 20:33:49 +0000 | [diff] [blame] | 314 | |
Christian Heimes | e93237d | 2007-12-19 02:37:44 +0000 | [diff] [blame] | 315 | Py_TYPE(&MD5type) = &PyType_Type; |
Gregory P. Smith | f21a5f7 | 2005-08-21 18:45:59 +0000 | [diff] [blame] | 316 | if (PyType_Ready(&MD5type) < 0) |
| 317 | return; |
| 318 | m = Py_InitModule3("_md5", md5_functions, module_doc); |
Neal Norwitz | 1ac754f | 2006-01-19 06:09:39 +0000 | [diff] [blame] | 319 | if (m == NULL) |
| 320 | return; |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 321 | d = PyModule_GetDict(m); |
| 322 | PyDict_SetItemString(d, "MD5Type", (PyObject *)&MD5type); |
Fred Drake | 52a42e9 | 2001-11-02 22:05:06 +0000 | [diff] [blame] | 323 | PyModule_AddIntConstant(m, "digest_size", 16); |
Guido van Rossum | d3a6a14 | 1998-10-14 13:46:57 +0000 | [diff] [blame] | 324 | /* No need to check the error here, the caller will do that */ |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 325 | } |