From 101fbe31005c07cde0fdbdb38a4f6c3499e07edc Mon Sep 17 00:00:00 2001 From: dota17 Date: Thu, 8 Aug 2019 21:28:20 +0800 Subject: [PATCH] sample/signal-test: add NULL checks Fixes: #865 --- sample/signal-test.c | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/sample/signal-test.c b/sample/signal-test.c index 18668350..a8564840 100644 --- a/sample/signal-test.c +++ b/sample/signal-test.c @@ -46,6 +46,7 @@ main(int argc, char **argv) { struct event *signal_int; struct event_base* base; + int ret = 0; #ifdef _WIN32 WORD wVersionRequested; WSADATA wsaData; @@ -57,16 +58,26 @@ main(int argc, char **argv) /* Initalize the event library */ base = event_base_new(); + if (!base) { + ret = 1; + goto out; + } /* Initalize one event */ signal_int = evsignal_new(base, SIGINT, signal_cb, event_self_cbarg()); - + if (!signal_int) { + ret = 2; + goto out; + } event_add(signal_int, NULL); event_base_dispatch(base); - event_free(signal_int); - event_base_free(base); - return (0); +out: + if (signal_int) + event_free(signal_int); + if (base) + event_base_free(base); + return ret; } -- 2.50.1