jyapayne/swig
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Check Py{Bytes,String}_AsStringAndSize() for failure

Browse source 
PyBytes_AsStringAndSize() and PyString_AsStringAndSize() were not
being checked for failure.

Closes #1349.
This commit is contained in:
Zackery Spytz 2020-01-09 21:14:12 -07:00
commit 6adf19b52f
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
Lib/python/pyhead.swg
View file

@ -43,8 +43,11 @@ SWIG_Python_str_AsChar(PyObject *str)
if (str) {
char *cstr;
Py_ssize_t len;
PyBytes_AsStringAndSize(str, &cstr, &len);
if (PyBytes_AsStringAndSize(str, &cstr, &len) == -1)
return NULL;
newstr = (char *) malloc(len+1);
if (!newstr)
return NULL;
memcpy(newstr, cstr, len+1);
Py_XDECREF(str);
}

6
Lib/python/pystrings.swg
View file

@ -32,9 +32,11 @@ SWIG_AsCharPtrAndSize(PyObject *obj, char** cptr, size_t* psize, int *alloc)
if (alloc)
*alloc = SWIG_NEWOBJ;
%#endif
PyBytes_AsStringAndSize(obj, &cstr, &len);
if (PyBytes_AsStringAndSize(obj, &cstr, &len) == -1)
return SWIG_TypeError;
%#else
PyString_AsStringAndSize(obj, &cstr, &len);
if (PyString_AsStringAndSize(obj, &cstr, &len) == -1)
return SWIG_TypeError;
%#endif
if (cptr) {
if (alloc) {