From d054eae6c954baa857170bb60072c8a2ecea0d6b Mon Sep 17 00:00:00 2001 From: Evgeny Vereshchagin Date: Mon, 9 Jan 2017 22:45:41 +0000 Subject: shared: check strdup != NULL This is a follow-up for dc7dd61de610e9330 --- src/shared/install.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/shared/install.c b/src/shared/install.c index b7e8f14531..478abac8ab 100644 --- a/src/shared/install.c +++ b/src/shared/install.c @@ -1893,7 +1893,11 @@ int unit_file_unmask( if (!GREEDY_REALLOC0(todo, n_allocated, n_todo + 2)) return -ENOMEM; - todo[n_todo++] = strdup(*i); + todo[n_todo] = strdup(*i); + if (!todo[n_todo]) + return -ENOMEM; + + n_todo++; } strv_uniq(todo); -- cgit v1.2.3-54-g00ecf