Add check for NULL after malloc()

This commit is contained in:
Luflosi 2019-02-23 09:06:43 +01:00
parent e44b331cc3
commit 174a45b4bd
No known key found for this signature in database
GPG Key ID: 14140F703B7D8362

View File

@ -204,6 +204,7 @@ match(PyObject *self, PyObject *args) {
opts.delimiter_sz = copy_unicode_object(delimiter, opts.delimiter, arraysz(opts.delimiter)); opts.delimiter_sz = copy_unicode_object(delimiter, opts.delimiter, arraysz(opts.delimiter));
size_t num_lines = PyList_GET_SIZE(lines); size_t num_lines = PyList_GET_SIZE(lines);
char **clines = malloc(sizeof(char*) * num_lines); char **clines = malloc(sizeof(char*) * num_lines);
if (!clines) { return PyErr_NoMemory(); }
size_t *sizes = malloc(sizeof(size_t) * num_lines); size_t *sizes = malloc(sizeof(size_t) * num_lines);
if (!sizes) { return PyErr_NoMemory(); } if (!sizes) { return PyErr_NoMemory(); }
for (size_t i = 0; i < num_lines; i++) { for (size_t i = 0; i < num_lines; i++) {