From d054eae6c954baa857170bb60072c8a2ecea0d6b Mon Sep 17 00:00:00 2001
From: Evgeny Vereshchagin <evvers@ya.ru>
Date: Mon, 9 Jan 2017 22:45:41 +0000
Subject: [PATCH] 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 b7e8f14..478abac 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);
-- 
2.7.4