AppPkg/Applications/Python/Python-2.7.10/Modules/md5module.c

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