Issue #532631: Add paranoid check to avoid potential buffer overflow
[python.git] / Modules / hashlib.h
blobb563729ff21631bd9f815dd4c8f2711daf781f0d
1 /* Common code for use by all hashlib related modules. */
3 /*
4 * Given a PyObject* obj, fill in the Py_buffer* viewp with the result
5 * of PyObject_GetBuffer. Sets and exception and issues a returns
6 * on any errors.
7 */
8 #define GET_BUFFER_VIEW_OR_ERROUT(obj, viewp, error_return) do { \
9 if (PyUnicode_Check((obj))) { \
10 PyErr_SetString(PyExc_TypeError, \
11 "Unicode-objects must be encoded before hashing");\
12 return error_return; \
13 } \
14 if (!PyObject_CheckBuffer((obj))) { \
15 PyErr_SetString(PyExc_TypeError, \
16 "object supporting the buffer API required"); \
17 return error_return; \
18 } \
19 if (PyObject_GetBuffer((obj), (viewp), PyBUF_SIMPLE) == -1) { \
20 return error_return; \
21 } \
22 if ((viewp)->ndim > 1) { \
23 PyErr_SetString(PyExc_BufferError, \
24 "Buffer must be single dimension"); \
25 PyBuffer_Release((viewp)); \
26 return error_return; \
27 } \
28 } while(0);