From 2b08f88f19b0270c9db993c940181535c33a71b3 Mon Sep 17 00:00:00 2001 From: Paul LeoNerd Evans Date: Thu, 9 Oct 2008 23:19:10 +0100 Subject: Better free() tracking in constructor failure cases --- driver-ti.c | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'driver-ti.c') diff --git a/driver-ti.c b/driver-ti.c index 66cdbe7..00090bd 100644 --- a/driver-ti.c +++ b/driver-ti.c @@ -33,6 +33,8 @@ static void *new_driver(termkey_t *tk, const char *term) return NULL; termkey_ti *ti = malloc(sizeof *ti); + if(!ti) + return NULL; ti->tk = tk; @@ -40,6 +42,8 @@ static void *new_driver(termkey_t *tk, const char *term) ti->nseqs = 0; ti->seqs = malloc(ti->alloced_seqs * sizeof(ti->seqs[0])); + if(!ti->seqs) + goto abort_free_ti; int i; for(i = 0; strfnames[i]; i++) { @@ -64,6 +68,11 @@ static void *new_driver(termkey_t *tk, const char *term) } return ti; + +abort_free_ti: + free(ti); + + return NULL; } static void free_driver(void *private) -- cgit v1.2.3