diff options
Diffstat (limited to 'psycopg')
| -rw-r--r-- | psycopg/connection_int.c | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/psycopg/connection_int.c b/psycopg/connection_int.c index 7f6b755..17509cf 100644 --- a/psycopg/connection_int.c +++ b/psycopg/connection_int.c @@ -1081,7 +1081,16 @@ conn_poll(connectionObject *self) /* An async query has just finished: parse the tuple in the * target cursor. */ cursorObject *curs; - PyObject *py_curs = PyWeakref_GetObject(self->async_cursor); + PyObject *py_curs; + if (!(py_curs = PyWeakref_GetObject(self->async_cursor))) { + /* It shouldn't happen but consider it to avoid dereferencing + * a null pointer below. */ + pq_clear_async(self); + PyErr_SetString(PyExc_SystemError, + "got null dereferencing cursor weakref"); + res = PSYCO_POLL_ERROR; + break; + } if (Py_None == py_curs) { pq_clear_async(self); PyErr_SetString(InterfaceError, |