Backport 58385 from trunk: fix a double free bug in the _bsddb module
authorGregory P. Smith <greg@mad-scientist.com>
Tue, 9 Oct 2007 07:25:24 +0000 (07:25 +0000)
committerGregory P. Smith <greg@mad-scientist.com>
Tue, 9 Oct 2007 07:25:24 +0000 (07:25 +0000)
on DBCursor.get (and a friends) when passing in a string key.

Lib/bsddb/test/test_misc.py
Modules/_bsddb.c

index 88f700b46979aa218e1c99251694ed5dc78d26d5..e7df4a89073b17af744376a02e6ec6134d1bbfb9 100644 (file)
@@ -52,6 +52,26 @@ class MiscTestCase(unittest.TestCase):
         rp = repr(db)
         self.assertEquals(rp, "{}")
 
+    # http://sourceforge.net/tracker/index.php?func=detail&aid=1708868&group_id=13900&atid=313900
+    #
+    # See the bug report for details.
+    #
+    # The problem was that make_key_dbt() was not allocating a copy of
+    # string keys but FREE_DBT() was always being told to free it when the
+    # database was opened with DB_THREAD.
+    def test04_double_free_make_key_dbt(self):
+        try:
+            db1 = db.DB()
+            db1.open(self.filename, None, db.DB_BTREE,
+                     db.DB_CREATE | db.DB_THREAD)
+
+            curs = db1.cursor()
+            t = curs.get("/foo", db.DB_SET)
+            # double free happened during exit from DBC_get
+        finally:
+            db1.close()
+            os.unlink(self.filename)
+
 
 #----------------------------------------------------------------------
 
index 7b3105496703f45b48ae5ddaa98ad45c50a17f70..195132e3d2542bb82c38c81f796e3153d6297456 100644 (file)
@@ -425,7 +425,19 @@ make_key_dbt(DBObject* self, PyObject* keyobj, DBT* key, int* pflags)
             return 0;
         }
 
-        key->data = PyString_AS_STRING(keyobj);
+        /*
+         * NOTE(gps): I don't like doing a data copy here, it seems
+         * wasteful.  But without a clean way to tell FREE_DBT if it
+         * should free key->data or not we have to.  Other places in
+         * the code check for DB_THREAD and forceably set DBT_MALLOC
+         * when we otherwise would leave flags 0 to indicate that.
+         */
+        key->data = strdup(PyString_AS_STRING(keyobj));
+        if (key->data == NULL) {
+            PyErr_SetString(PyExc_MemoryError, "Key memory allocation failed");
+            return 0;
+        }
+        key->flags = DB_DBT_REALLOC;
         key->size = PyString_GET_SIZE(keyobj);
     }