From: Georg Brandl Date: Sat, 19 Sep 2009 07:35:07 +0000 (+0000) Subject: #6944: the argument to PyArg_ParseTuple should be a tuple, otherwise a SystemError... X-Git-Tag: v2.7a1~495 X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=ab849891ef1d96c4f770fbd256b2436359146f03;p=python #6944: the argument to PyArg_ParseTuple should be a tuple, otherwise a SystemError is set. Also clean up another usage of PyArg_ParseTuple. --- diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index 4b26824f7f..66a402b4e3 100644 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -281,7 +281,7 @@ class GeneralModuleTests(unittest.TestCase): # On some versions, this loses a reference orig = sys.getrefcount(__name__) socket.getnameinfo(__name__,0) - except SystemError: + except TypeError: if sys.getrefcount(__name__) <> orig: self.fail("socket.getnameinfo loses a reference") diff --git a/Misc/NEWS b/Misc/NEWS index 8b0a67853a..cac78e1c94 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -1333,6 +1333,9 @@ C-API Extension Modules ----------------- +- Issue #6944: Fix a SystemError when socket.getnameinfo() was called + with something other than a tuple as first argument. + - Issue #6865: Fix reference counting issue in the initialization of the pwd module. diff --git a/Modules/socketmodule.c b/Modules/socketmodule.c index 5575855ef7..18cce690a8 100644 --- a/Modules/socketmodule.c +++ b/Modules/socketmodule.c @@ -4100,8 +4100,13 @@ socket_getnameinfo(PyObject *self, PyObject *args) flags = flowinfo = scope_id = 0; if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags)) return NULL; - if (!PyArg_ParseTuple(sa, "si|ii", - &hostp, &port, &flowinfo, &scope_id)) + if (!PyTuple_Check(sa)) { + PyErr_SetString(PyExc_TypeError, + "getnameinfo() argument 1 must be a tuple"); + return NULL; + } + if (!PyArg_ParseTuple(sa, "si|ii", + &hostp, &port, &flowinfo, &scope_id)) return NULL; PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port); memset(&hints, 0, sizeof(hints)); @@ -4124,9 +4129,7 @@ socket_getnameinfo(PyObject *self, PyObject *args) switch (res->ai_family) { case AF_INET: { - char *t1; - int t2; - if (PyArg_ParseTuple(sa, "si", &t1, &t2) == 0) { + if (PyTuple_GET_SIZE(sa) != 2) { PyErr_SetString(socket_error, "IPv4 sockaddr must be 2 tuple"); goto fail;