From 174a45b4bdb1010128ada3f86c602ea05e2bb59b Mon Sep 17 00:00:00 2001 From: Luflosi Date: Sat, 23 Feb 2019 09:06:43 +0100 Subject: [PATCH] Add check for NULL after malloc() --- kittens/choose/main.c | 1 + 1 file changed, 1 insertion(+) diff --git a/kittens/choose/main.c b/kittens/choose/main.c index 1db27dd59..b20fd69a6 100644 --- a/kittens/choose/main.c +++ b/kittens/choose/main.c @@ -204,6 +204,7 @@ match(PyObject *self, PyObject *args) { opts.delimiter_sz = copy_unicode_object(delimiter, opts.delimiter, arraysz(opts.delimiter)); size_t num_lines = PyList_GET_SIZE(lines); char **clines = malloc(sizeof(char*) * num_lines); + if (!clines) { return PyErr_NoMemory(); } size_t *sizes = malloc(sizeof(size_t) * num_lines); if (!sizes) { return PyErr_NoMemory(); } for (size_t i = 0; i < num_lines; i++) {