diff --git a/Modules/md5module.c b/Modules/md5module.c index 269a298c05a..82bd415d764 100644 --- a/Modules/md5module.c +++ b/Modules/md5module.c @@ -161,6 +161,10 @@ static PyMethodDef md5_methods[] = { static PyObject * md5_getattr(md5object *self, char *name) { + if (strcmp(name, "digest_size") == 0) { + return PyInt_FromLong(16); + } + return Py_FindMethod(md5_methods, (PyObject *)self, name); } @@ -264,11 +268,13 @@ static PyMethodDef md5_functions[] = { DL_EXPORT(void) initmd5(void) { - PyObject *m, *d; + PyObject *m, *d, *i; MD5type.ob_type = &PyType_Type; m = Py_InitModule3("md5", md5_functions, module_doc); d = PyModule_GetDict(m); PyDict_SetItemString(d, "MD5Type", (PyObject *)&MD5type); + if ( (i = PyInt_FromLong(16)) != NULL) + PyDict_SetItemString(d, "digest_size", i); /* No need to check the error here, the caller will do that */ } diff --git a/Modules/shamodule.c b/Modules/shamodule.c index 8bfcec7b82b..724f6d61787 100644 --- a/Modules/shamodule.c +++ b/Modules/shamodule.c @@ -5,7 +5,7 @@ /* See below for information about the original code this module was based upon. Additional work performed by: - Andrew Kuchling (amk1@bigfoot.com) + Andrew Kuchling (akuchlin@mems-exchange.org) Greg Stein (gstein@lyra.org) */ @@ -458,8 +458,8 @@ SHA_getattr(PyObject *self, char *name) { if (strcmp(name, "blocksize")==0) return PyInt_FromLong(1); - if (strcmp(name, "digestsize")==0) - return PyInt_FromLong(20); + if (strcmp(name, "digest_size")==0 || strcmp(name, "digestsize")==0) + return PyInt_FromLong(20); return Py_FindMethod(SHA_methods, self, name); } @@ -542,4 +542,5 @@ initsha(void) functions require an integral number of blocks */ insint("digestsize", 20); + insint("digest_size", 20); }