Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly.

Patch by Serhiy Storchaka.
This commit is contained in:
Antoine Pitrou 2012-08-15 23:18:25 +02:00
parent dd7c55250d
commit 6f430e4963
11 changed files with 87 additions and 39 deletions

View File

@ -10,6 +10,9 @@ What's New in Python 3.2.4
Core and Builtins Core and Builtins
----------------- -----------------
- Issue #15604: Update uses of PyObject_IsTrue() to check for and handle
errors correctly. Patch by Serhiy Storchaka.
- Issue #13119: sys.stdout and sys.stderr are now using "\r\n" newline on - Issue #13119: sys.stdout and sys.stderr are now using "\r\n" newline on
Windows, as Python 2. Windows, as Python 2.

View File

@ -166,8 +166,12 @@ _set_bool(const char *name, int *target, PyObject *src, int dflt)
{ {
if (src == NULL) if (src == NULL)
*target = dflt; *target = dflt;
else else {
*target = PyObject_IsTrue(src); int b = PyObject_IsTrue(src);
if (b < 0)
return -1;
*target = b;
}
return 0; return 0;
} }

View File

@ -1046,8 +1046,11 @@ textiowrapper_init(textio *self, PyObject *args, PyObject *kwds)
res = PyObject_CallMethod(buffer, "seekable", NULL); res = PyObject_CallMethod(buffer, "seekable", NULL);
if (res == NULL) if (res == NULL)
goto error; goto error;
self->seekable = self->telling = PyObject_IsTrue(res); r = PyObject_IsTrue(res);
Py_DECREF(res); Py_DECREF(res);
if (r < 0)
goto error;
self->seekable = self->telling = r;
self->has_read1 = PyObject_HasAttrString(buffer, "read1"); self->has_read1 = PyObject_HasAttrString(buffer, "read1");

View File

@ -525,6 +525,8 @@ subprocess_fork_exec(PyObject* self, PyObject *args)
return NULL; return NULL;
close_fds = PyObject_IsTrue(py_close_fds); close_fds = PyObject_IsTrue(py_close_fds);
if (close_fds < 0)
return NULL;
if (close_fds && errpipe_write < 3) { /* precondition */ if (close_fds && errpipe_write < 3) { /* precondition */
PyErr_SetString(PyExc_ValueError, "errpipe_write must be >= 3"); PyErr_SetString(PyExc_ValueError, "errpipe_write must be >= 3");
return NULL; return NULL;

View File

@ -883,6 +883,7 @@ PySSL_peercert(PySSLSocket *self, PyObject *args)
int len; int len;
int verification; int verification;
PyObject *binary_mode = Py_None; PyObject *binary_mode = Py_None;
int b;
if (!PyArg_ParseTuple(args, "|O:peer_certificate", &binary_mode)) if (!PyArg_ParseTuple(args, "|O:peer_certificate", &binary_mode))
return NULL; return NULL;
@ -890,7 +891,10 @@ PySSL_peercert(PySSLSocket *self, PyObject *args)
if (!self->peer_cert) if (!self->peer_cert)
Py_RETURN_NONE; Py_RETURN_NONE;
if (PyObject_IsTrue(binary_mode)) { b = PyObject_IsTrue(binary_mode);
if (b < 0)
return NULL;
if (b) {
/* return cert in DER-encoded format */ /* return cert in DER-encoded format */
unsigned char *bytes_buf = NULL; unsigned char *bytes_buf = NULL;

View File

@ -903,11 +903,13 @@ dropwhile_next(dropwhileobject *lz)
} }
ok = PyObject_IsTrue(good); ok = PyObject_IsTrue(good);
Py_DECREF(good); Py_DECREF(good);
if (!ok) { if (ok == 0) {
lz->start = 1; lz->start = 1;
return item; return item;
} }
Py_DECREF(item); Py_DECREF(item);
if (ok < 0)
return NULL;
} }
} }
@ -1043,9 +1045,10 @@ takewhile_next(takewhileobject *lz)
} }
ok = PyObject_IsTrue(good); ok = PyObject_IsTrue(good);
Py_DECREF(good); Py_DECREF(good);
if (ok) if (ok > 0)
return item; return item;
Py_DECREF(item); Py_DECREF(item);
if (ok == 0)
lz->stop = 1; lz->stop = 1;
return NULL; return NULL;
} }
@ -2959,9 +2962,11 @@ filterfalse_next(filterfalseobject *lz)
ok = PyObject_IsTrue(good); ok = PyObject_IsTrue(good);
Py_DECREF(good); Py_DECREF(good);
} }
if (!ok) if (ok == 0)
return item; return item;
Py_DECREF(item); Py_DECREF(item);
if (ok < 0)
return NULL;
} }
} }

View File

@ -401,10 +401,14 @@ parser_st2tuple(PyST_Object *self, PyObject *args, PyObject *kw)
int lineno = 0; int lineno = 0;
int col_offset = 0; int col_offset = 0;
if (line_option != NULL) { if (line_option != NULL) {
lineno = (PyObject_IsTrue(line_option) != 0) ? 1 : 0; lineno = PyObject_IsTrue(line_option);
if (lineno < 0)
return NULL;
} }
if (col_option != NULL) { if (col_option != NULL) {
col_offset = (PyObject_IsTrue(col_option) != 0) ? 1 : 0; col_offset = PyObject_IsTrue(col_option);
if (col_offset < 0)
return NULL;
} }
/* /*
* Convert ST into a tuple representation. Use Guido's function, * Convert ST into a tuple representation. Use Guido's function,
@ -444,10 +448,14 @@ parser_st2list(PyST_Object *self, PyObject *args, PyObject *kw)
int lineno = 0; int lineno = 0;
int col_offset = 0; int col_offset = 0;
if (line_option != 0) { if (line_option != 0) {
lineno = PyObject_IsTrue(line_option) ? 1 : 0; lineno = PyObject_IsTrue(line_option);
if (lineno < 0)
return NULL;
} }
if (col_option != NULL) { if (col_option != 0) {
col_offset = (PyObject_IsTrue(col_option) != 0) ? 1 : 0; col_offset = PyObject_IsTrue(col_option);
if (col_offset < 0)
return NULL;
} }
/* /*
* Convert ST into a tuple representation. Use Guido's function, * Convert ST into a tuple representation. Use Guido's function,

View File

@ -1033,13 +1033,16 @@ static PyObject *
xmlparse_UseForeignDTD(xmlparseobject *self, PyObject *args) xmlparse_UseForeignDTD(xmlparseobject *self, PyObject *args)
{ {
PyObject *flagobj = NULL; PyObject *flagobj = NULL;
XML_Bool flag = XML_TRUE; int flag = 1;
enum XML_Error rc; enum XML_Error rc;
if (!PyArg_UnpackTuple(args, "UseForeignDTD", 0, 1, &flagobj)) if (!PyArg_ParseTuple(args, "O:UseForeignDTD", &flagobj))
return NULL; return NULL;
if (flagobj != NULL) if (flagobj != NULL) {
flag = PyObject_IsTrue(flagobj) ? XML_TRUE : XML_FALSE; flag = PyObject_IsTrue(flagobj);
rc = XML_UseForeignDTD(self->itself, flag); if (flag < 0)
return NULL;
}
rc = XML_UseForeignDTD(self->itself, flag ? XML_TRUE : XML_FALSE);
if (rc != XML_ERROR_NONE) { if (rc != XML_ERROR_NONE) {
return set_error(self, rc); return set_error(self, rc);
} }
@ -1397,7 +1400,10 @@ xmlparse_setattro(xmlparseobject *self, PyObject *name, PyObject *v)
} }
assert(PyUnicode_Check(name)); assert(PyUnicode_Check(name));
if (PyUnicode_CompareWithASCIIString(name, "buffer_text") == 0) { if (PyUnicode_CompareWithASCIIString(name, "buffer_text") == 0) {
if (PyObject_IsTrue(v)) { int b = PyObject_IsTrue(v);
if (b < 0)
return -1;
if (b) {
if (self->buffer == NULL) { if (self->buffer == NULL) {
self->buffer = malloc(self->buffer_size); self->buffer = malloc(self->buffer_size);
if (self->buffer == NULL) { if (self->buffer == NULL) {
@ -1416,25 +1422,25 @@ xmlparse_setattro(xmlparseobject *self, PyObject *name, PyObject *v)
return 0; return 0;
} }
if (PyUnicode_CompareWithASCIIString(name, "namespace_prefixes") == 0) { if (PyUnicode_CompareWithASCIIString(name, "namespace_prefixes") == 0) {
if (PyObject_IsTrue(v)) int b = PyObject_IsTrue(v);
self->ns_prefixes = 1; if (b < 0)
else return -1;
self->ns_prefixes = 0; self->ns_prefixes = b;
XML_SetReturnNSTriplet(self->itself, self->ns_prefixes); XML_SetReturnNSTriplet(self->itself, self->ns_prefixes);
return 0; return 0;
} }
if (PyUnicode_CompareWithASCIIString(name, "ordered_attributes") == 0) { if (PyUnicode_CompareWithASCIIString(name, "ordered_attributes") == 0) {
if (PyObject_IsTrue(v)) int b = PyObject_IsTrue(v);
self->ordered_attributes = 1; if (b < 0)
else return -1;
self->ordered_attributes = 0; self->ordered_attributes = b;
return 0; return 0;
} }
if (PyUnicode_CompareWithASCIIString(name, "specified_attributes") == 0) { if (PyUnicode_CompareWithASCIIString(name, "specified_attributes") == 0) {
if (PyObject_IsTrue(v)) int b = PyObject_IsTrue(v);
self->specified_attributes = 1; if (b < 0)
else return -1;
self->specified_attributes = 0; self->specified_attributes = b;
return 0; return 0;
} }

View File

@ -340,11 +340,15 @@ type_set_abstractmethods(PyTypeObject *type, PyObject *value, void *context)
abc.ABCMeta.__new__, so this function doesn't do anything abc.ABCMeta.__new__, so this function doesn't do anything
special to update subclasses. special to update subclasses.
*/ */
int res; int abstract, res;
if (value != NULL) { if (value != NULL) {
abstract = PyObject_IsTrue(value);
if (abstract < 0)
return -1;
res = PyDict_SetItemString(type->tp_dict, "__abstractmethods__", value); res = PyDict_SetItemString(type->tp_dict, "__abstractmethods__", value);
} }
else { else {
abstract = 0;
res = PyDict_DelItemString(type->tp_dict, "__abstractmethods__"); res = PyDict_DelItemString(type->tp_dict, "__abstractmethods__");
if (res && PyErr_ExceptionMatches(PyExc_KeyError)) { if (res && PyErr_ExceptionMatches(PyExc_KeyError)) {
PyErr_SetString(PyExc_AttributeError, "__abstractmethods__"); PyErr_SetString(PyExc_AttributeError, "__abstractmethods__");
@ -353,13 +357,11 @@ type_set_abstractmethods(PyTypeObject *type, PyObject *value, void *context)
} }
if (res == 0) { if (res == 0) {
PyType_Modified(type); PyType_Modified(type);
if (value && PyObject_IsTrue(value)) { if (abstract)
type->tp_flags |= Py_TPFLAGS_IS_ABSTRACT; type->tp_flags |= Py_TPFLAGS_IS_ABSTRACT;
} else
else {
type->tp_flags &= ~Py_TPFLAGS_IS_ABSTRACT; type->tp_flags &= ~Py_TPFLAGS_IS_ABSTRACT;
} }
}
return res; return res;
} }

View File

@ -428,9 +428,11 @@ filter_next(filterobject *lz)
ok = PyObject_IsTrue(good); ok = PyObject_IsTrue(good);
Py_DECREF(good); Py_DECREF(good);
} }
if (ok) if (ok > 0)
return item; return item;
Py_DECREF(item); Py_DECREF(item);
if (ok < 0)
return NULL;
} }
} }

View File

@ -1338,7 +1338,10 @@ load_source_module(char *name, char *pathname, FILE *fp)
name, pathname); name, pathname);
if (cpathname) { if (cpathname) {
PyObject *ro = PySys_GetObject("dont_write_bytecode"); PyObject *ro = PySys_GetObject("dont_write_bytecode");
if (ro == NULL || !PyObject_IsTrue(ro)) int b = (ro == NULL) ? 0 : PyObject_IsTrue(ro);
if (b < 0)
goto error_exit;
if (!b)
write_compiled_module(co, cpathname, &st); write_compiled_module(co, cpathname, &st);
} }
} }
@ -2504,7 +2507,13 @@ import_module_level(char *name, PyObject *globals, PyObject *locals,
} }
if (fromlist != NULL) { if (fromlist != NULL) {
if (fromlist == Py_None || !PyObject_IsTrue(fromlist)) int b = (fromlist == Py_None) ? 0 : PyObject_IsTrue(fromlist);
if (b < 0) {
Py_DECREF(tail);
Py_DECREF(head);
goto error_exit;
}
if (!b)
fromlist = NULL; fromlist = NULL;
} }