diff Modules/socketmodule.c @ 68660:2af7a6d765fd 3.2

Issue #5421: merge fix
author Antoine Pitrou <solipsis@pitrou.net>
date Thu, 17 Mar 2011 22:40:18 +0100
parents dc245d681901 c61b72b0650d
children 713006ecee0c 90cdc371a3b8
line wrap: on
line diff
--- a/Modules/socketmodule.c	Thu Mar 17 13:47:12 2011 -0400
+++ b/Modules/socketmodule.c	Thu Mar 17 22:40:18 2011 +0100
@@ -2747,17 +2747,28 @@
     Py_buffer pbuf;
     PyObject *addro;
     char *buf;
-    Py_ssize_t len;
+    Py_ssize_t len, arglen;
     sock_addr_t addrbuf;
     int addrlen, n = -1, flags, timeout;
 
     flags = 0;
-    if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
-        PyErr_Clear();
-        if (!PyArg_ParseTuple(args, "y*iO:sendto",
-                              &pbuf, &flags, &addro))
-            return NULL;
+    arglen = PyTuple_Size(args);
+    switch (arglen) {
+        case 2:
+            PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro);
+            break;
+        case 3:
+            PyArg_ParseTuple(args, "y*iO:sendto",
+                             &pbuf, &flags, &addro);
+            break;
+        default:
+            PyErr_Format(PyExc_TypeError,
+                         "sendto() takes 2 or 3 arguments (%d given)",
+                         arglen);
     }
+    if (PyErr_Occurred())
+        return NULL;
+
     buf = pbuf.buf;
     len = pbuf.len;