]> granicus.if.org Git - python/commitdiff
Issue #12764: Fix a crash in ctypes when the name of a Structure field is not
authorAmaury Forgeot d'Arc <amauryfa@gmail.com>
Fri, 2 Sep 2011 18:32:23 +0000 (20:32 +0200)
committerAmaury Forgeot d'Arc <amauryfa@gmail.com>
Fri, 2 Sep 2011 18:32:23 +0000 (20:32 +0200)
a string.

Lib/ctypes/test/test_structures.py
Misc/NEWS
Modules/_ctypes/stgdict.c

index 77cfb26711913f098ca92e71340c0d23e2cd4b41..a84bae0640b0cd21f63d0bf919cd9f4261d70617 100644 (file)
@@ -239,6 +239,14 @@ class StructureTestCase(unittest.TestCase):
             pass
         self.assertRaises(TypeError, setattr, POINT, "_fields_", [("x", 1), ("y", 2)])
 
+    def test_invalid_name(self):
+        # field name must be string
+        def declare_with_name(name):
+            class S(Structure):
+                _fields_ = [(name, c_int)]
+
+        self.assertRaises(TypeError, declare_with_name, u"x\xe9")
+
     def test_intarray_fields(self):
         class SomeInts(Structure):
             _fields_ = [("a", c_int * 4)]
index 8c77c9f91fb88a4aeb2f80472fadd2be200f0ce7..6d57532578ac0e7a089590ad155c59eb05153d78 100644 (file)
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -180,6 +180,9 @@ Library
 Extension Modules
 -----------------
 
+- Issue #12764: Fix a crash in ctypes when the name of a Structure field is not
+  a string.
+
 - Issue #9651: Fix a crash when ctypes.create_string_buffer(0) was passed to
   some functions like file.write().
 
index 4e7ea12fb34d8545b8e3ca127cb197d59071645d..4d4ecc47174649cf411954a2c76b8a6860650384 100644 (file)
@@ -494,8 +494,21 @@ PyCStructUnionType_update_stgdict(PyObject *type, PyObject *fields, int isStruct
             char *fieldfmt = dict->format ? dict->format : "B";
             char *fieldname = PyString_AsString(name);
             char *ptr;
-            Py_ssize_t len = strlen(fieldname) + strlen(fieldfmt);
-            char *buf = alloca(len + 2 + 1);
+            Py_ssize_t len; 
+            char *buf;
+
+            if (fieldname == NULL)
+            {
+                PyErr_Format(PyExc_TypeError,
+                             "structure field name must be string not %s",
+                             name->ob_type->tp_name);
+                                
+                Py_DECREF(pair);
+                return -1;
+            }
+
+            len = strlen(fieldname) + strlen(fieldfmt);
+            buf = alloca(len + 2 + 1);
 
             sprintf(buf, "%s:%s:", fieldfmt, fieldname);