Navigation Menu

Skip to content

Commit

Permalink
bpo-38132: Check EVP_DigestUpdate for error (GH-16041)
Browse files Browse the repository at this point in the history
(cherry picked from commit 8c74574)

Co-authored-by: Christian Heimes <christian@python.org>
  • Loading branch information
miss-islington and tiran committed Sep 12, 2019
1 parent 2f01cf6 commit 0d7cb5b
Showing 1 changed file with 16 additions and 6 deletions.
22 changes: 16 additions & 6 deletions Modules/_hashopenssl.c
Expand Up @@ -96,7 +96,7 @@ newEVPobject(void)
return retval; return retval;
} }


static void static int
EVP_hash(EVPobject *self, const void *vp, Py_ssize_t len) EVP_hash(EVPobject *self, const void *vp, Py_ssize_t len)
{ {
unsigned int process; unsigned int process;
Expand All @@ -108,11 +108,12 @@ EVP_hash(EVPobject *self, const void *vp, Py_ssize_t len)
process = Py_SAFE_DOWNCAST(len, Py_ssize_t, unsigned int); process = Py_SAFE_DOWNCAST(len, Py_ssize_t, unsigned int);
if (!EVP_DigestUpdate(self->ctx, (const void*)cp, process)) { if (!EVP_DigestUpdate(self->ctx, (const void*)cp, process)) {
_setException(PyExc_ValueError); _setException(PyExc_ValueError);
break; return -1;
} }
len -= process; len -= process;
cp += process; cp += process;
} }
return 0;
} }


/* Internal methods for a hash object */ /* Internal methods for a hash object */
Expand Down Expand Up @@ -243,6 +244,7 @@ static PyObject *
EVP_update(EVPobject *self, PyObject *obj) EVP_update(EVPobject *self, PyObject *obj)
/*[clinic end generated code: output=ec1d55ed2432e966 input=9b30ec848f015501]*/ /*[clinic end generated code: output=ec1d55ed2432e966 input=9b30ec848f015501]*/
{ {
int result;
Py_buffer view; Py_buffer view;


GET_BUFFER_VIEW_OR_ERROUT(obj, &view); GET_BUFFER_VIEW_OR_ERROUT(obj, &view);
Expand All @@ -255,14 +257,17 @@ EVP_update(EVPobject *self, PyObject *obj)
if (self->lock != NULL) { if (self->lock != NULL) {
Py_BEGIN_ALLOW_THREADS Py_BEGIN_ALLOW_THREADS
PyThread_acquire_lock(self->lock, 1); PyThread_acquire_lock(self->lock, 1);
EVP_hash(self, view.buf, view.len); result = EVP_hash(self, view.buf, view.len);
PyThread_release_lock(self->lock); PyThread_release_lock(self->lock);
Py_END_ALLOW_THREADS Py_END_ALLOW_THREADS
} else { } else {
EVP_hash(self, view.buf, view.len); result = EVP_hash(self, view.buf, view.len);
} }


PyBuffer_Release(&view); PyBuffer_Release(&view);

if (result == -1)
return NULL;
Py_RETURN_NONE; Py_RETURN_NONE;
} }


Expand Down Expand Up @@ -396,6 +401,7 @@ static PyObject *
EVPnew(const EVP_MD *digest, EVPnew(const EVP_MD *digest,
const unsigned char *cp, Py_ssize_t len) const unsigned char *cp, Py_ssize_t len)
{ {
int result = 0;
EVPobject *self; EVPobject *self;


if (!digest) { if (!digest) {
Expand All @@ -415,10 +421,14 @@ EVPnew(const EVP_MD *digest,
if (cp && len) { if (cp && len) {
if (len >= HASHLIB_GIL_MINSIZE) { if (len >= HASHLIB_GIL_MINSIZE) {
Py_BEGIN_ALLOW_THREADS Py_BEGIN_ALLOW_THREADS
EVP_hash(self, cp, len); result = EVP_hash(self, cp, len);
Py_END_ALLOW_THREADS Py_END_ALLOW_THREADS
} else { } else {
EVP_hash(self, cp, len); result = EVP_hash(self, cp, len);
}
if (result == -1) {
Py_DECREF(self);
return NULL;
} }
} }


Expand Down

0 comments on commit 0d7cb5b

Please sign in to comment.