psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-27 03:13:43 +03:00
Code Issues Packages Projects Releases Wiki Activity

Check for failed allocation in the notice callback

Browse Source
This commit is contained in:
Daniele Varrazzo 2011-02-24 10:14:06 +00:00
parent 343687ebc8
commit 29ac03ef04
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
psycopg/connection_int.c
View File

@ -69,7 +69,15 @@ conn_notice_callback(void *args, const char *message)
*/ */
notice = (struct connectionObject_notice *) notice = (struct connectionObject_notice *)
malloc(sizeof(struct connectionObject_notice)); malloc(sizeof(struct connectionObject_notice));
if (NULL == notice) {
/* Discard the notice in case of failed allocation. */
return;
}
notice->message = strdup(message); notice->message = strdup(message);
if (NULL == notice->message) {
free(notice);
return;
}
notice->next = self->notice_pending; notice->next = self->notice_pending;
self->notice_pending = notice; self->notice_pending = notice;
} }