From: Andy Grover <andy@groveronline.com>
Date: Mon, 19 Mar 2012 04:49:07 +0000 (-0700)
Subject: python: add proper error handling to loaded_modules()
X-Git-Tag: v17~10^2~36
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=705187f17ba8713327b200d8b62f88397e3a0715;p=thirdparty%2Fkmod.git

python: add proper error handling to loaded_modules()

We need to check the result of basically all Py* calls and cleanup
properly if they fail.
---

diff --git a/libkmod/python/libkmod.c b/libkmod/python/libkmod.c
index b3613351..dd82d9e6 100644
--- a/libkmod/python/libkmod.c
+++ b/libkmod/python/libkmod.c
@@ -97,15 +97,32 @@ kmod_obj_loaded_modules(PyObject *self, PyObject *unused_args)
     }
 
     PyObject *pylist = PyList_New(0);
+    if (!pylist) {
+        kmod_module_unref_list(list);
+        return PyErr_NoMemory();
+    }
 
+    /* refcountapallooza. */
     kmod_list_foreach(itr, list) {
         struct kmod_module *mod = kmod_module_get_module(itr);
         const char *name = kmod_module_get_name(mod);
         long size = kmod_module_get_size(mod);
 
 	PyObject *entry = Py_BuildValue("(sl)", name, size);
-
-	PyList_Append(pylist, entry);
+	if (!entry) {
+            Py_DECREF(pylist);
+            kmod_module_unref(mod);
+            kmod_module_unref_list(list);
+            return NULL;
+        }
+
+	if (PyList_Append(pylist, entry) == -1) {
+            Py_DECREF(entry);
+            Py_DECREF(pylist);
+            kmod_module_unref(mod);
+            kmod_module_unref_list(list);
+            return NULL;
+        }
 
 	Py_DECREF(entry);
         kmod_module_unref(mod);