Better error handling when allocating and freeing permdb objectpermdb-alloc-fix-2

1 files changed, 10 insertions, 1 deletions

diff --git a/c_src/permdbpy.c b/c_src/permdbpy.c
index beea36a..bfad068 100644
--- a/c_src/permdbpy.c
+++ b/c_src/permdbpy.c
@@ -45,8 +45,17 @@ PyTypeObject permdb_type = {
 permdb_object_py *
 permdb_alloc_py(const char *dbpath)
 {
+        struct permdb_object *permdb;
+
+        permdb = permdb_alloc(dbpath);
+
+        if (permdb == NULL) {
+                PyErr_SetString(PyExc_RuntimeError, "Cannot allocate permdb object");
+                return NULL;
+        }
+
         permdb_object_py *state = PyObject_New(permdb_object_py, &permdb_type);
-        state->permdb = permdb_alloc(dbpath);
+        state->permdb = permdb;
         return state;
 }