From 91e8618d06ea60b7140e3a3398b7ce95278da74d Mon Sep 17 00:00:00 2001 From: Brian Overstreet Date: Tue, 25 Feb 2020 11:03:12 -0800 Subject: [PATCH] Raise import error on module import failure to avoid segfault --- confluent_kafka/src/confluent_kafka.c | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/confluent_kafka/src/confluent_kafka.c b/confluent_kafka/src/confluent_kafka.c index 279d8ee27..84c777398 100644 --- a/confluent_kafka/src/confluent_kafka.c +++ b/confluent_kafka/src/confluent_kafka.c @@ -1952,10 +1952,10 @@ PyObject *cfl_PyObject_lookup (const char *modulename, const char *typename) { PyObject *module = PyImport_ImportModule(modulename); PyObject *obj; - if (!modulename) { - PyErr_Format(PyExc_TypeError, - "Module %s not found when looking up %s.%s", - modulename, modulename, typename); + if (!module) { + PyErr_Format(PyExc_ImportError, + "Module not found when looking up %s.%s", + modulename, typename); return NULL; }