From: Tom Gundersen <teg@jklm.no>
Date: Mon, 1 Jun 2015 14:28:58 +0000 (+0200)
Subject: sd-device: ensure update_properties_buf() is a noop on failure
X-Git-Tag: v226.4~1^2~332
X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=commitdiff_plain;h=9b3a72550211800eaf34348a76898ed37f931d27

sd-device: ensure update_properties_buf() is a noop on failure

Don't clobber the sd_device struct, and don't leak memory when memory allocation fails.
---

diff --git a/src/libelogind/sd-device/device-private.c b/src/libelogind/sd-device/device-private.c
index deb8efd05..2e6043324 100644
--- a/src/libelogind/sd-device/device-private.c
+++ b/src/libelogind/sd-device/device-private.c
@@ -636,9 +636,10 @@ int device_new_from_nulstr(sd_device **ret, uint8_t *nulstr, size_t len) {
 
 static int device_update_properties_bufs(sd_device *device) {
         const char *val, *prop;
-        uint8_t *buf_nulstr = NULL;
+        _cleanup_free_ char **buf_strv = NULL;
+        _cleanup_free_ uint8_t *buf_nulstr = NULL;
         size_t allocated_nulstr = 0;
-        size_t nulstr_len = 0, num = 0, i;
+        size_t nulstr_len = 0, num = 0, i = 0;
 
         assert(device);
 
@@ -659,20 +660,25 @@ static int device_update_properties_bufs(sd_device *device) {
                 ++num;
         }
 
-        free(device->properties_nulstr);
-        device->properties_nulstr = buf_nulstr;
-        device->properties_nulstr_len = nulstr_len;
+        /* build buf_strv from buf_nulstr */
+        buf_strv = new0(char *, num + 1);
+        if (!buf_strv)
+                return -ENOMEM;
 
-        /* build strv from buf_nulstr */
-        free(device->properties_strv);
-        device->properties_strv = new0(char *, num + 1);
-        i = 0;
         NULSTR_FOREACH(val, (char*) buf_nulstr) {
-                device->properties_strv[i] = (char *) val;
+                buf_strv[i] = (char *) val;
                 assert(i < num);
                 i++;
         }
 
+        free(device->properties_nulstr);
+        device->properties_nulstr = buf_nulstr;
+        buf_nulstr = NULL;
+        device->properties_nulstr_len = nulstr_len;
+        free(device->properties_strv);
+        device->properties_strv = buf_strv;
+        buf_strv = NULL;
+
         device->properties_buf_outdated = false;
 
         return 0;