Port: Remove _GNU_SOURCE, defined in config.h

[lttng-tools.git] / src / bin / lttng-sessiond / modprobe.c

diff --git a/src/bin/lttng-sessiond/modprobe.c b/src/bin/lttng-sessiond/modprobe.c
index 5fed97bf992061ee4efd78c148c41ce83fc4646d..4c1ea67be18c50c4800c5ccf76374ee27e98f6b1 100644 (file)
--- a/src/bin/lttng-sessiond/modprobe.c
+++ b/src/bin/lttng-sessiond/modprobe.c
@@ -16,7 +16,6 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#define _GNU_SOURCE
 #define _LGPL_SOURCE
 #include <assert.h>
 #include <stdio.h>
@@ -141,23 +140,31 @@ void modprobe_remove_lttng_control(void)
                              LTTNG_MOD_REQUIRED);
 }
 
+static void free_probes(void)
+{
+       int i;
+
+       if (!probes) {
+               return;
+       }
+       for (i = 0; i < nr_probes; ++i) {
+               free(probes[i].name);
+       }
+       free(probes);
+       probes = NULL;
+       nr_probes = 0;
+}
+
 /*
  * Remove data kernel modules in reverse load order.
  */
 void modprobe_remove_lttng_data(void)
 {
-       int i;
-
-       if (probes) {
-               modprobe_remove_lttng(probes, nr_probes, LTTNG_MOD_OPTIONAL);
-
-               for (i = 0; i < nr_probes; ++i) {
-                       free(probes[i].name);
-               }
-
-               free(probes);
-               probes = NULL;
+       if (!probes) {
+               return;
        }
+       modprobe_remove_lttng(probes, nr_probes, LTTNG_MOD_OPTIONAL);
+       free_probes();
 }
 
 /*
@@ -351,11 +358,12 @@ static int grow_probes(void)
 static int append_list_to_probes(const char *list)
 {
        char *next;
-       int index = nr_probes, ret;
+       int ret;
+       char *tmp_list, *cur_list;
 
        assert(list);
 
-       char *tmp_list = strdup(list);
+       cur_list = tmp_list = strdup(list);
        if (!tmp_list) {
                PERROR("strdup temp list");
                return -ENOMEM;
@@ -365,11 +373,11 @@ static int append_list_to_probes(const char *list)
                size_t name_len;
                struct kern_modules_param *cur_mod;
 
-               next = strtok(tmp_list, ",");
+               next = strtok(cur_list, ",");
                if (!next) {
                        break;
                }
-               tmp_list = NULL;
+               cur_list = NULL;
 
                /* filter leading spaces */
                while (*next == ' ') {
@@ -379,33 +387,38 @@ static int append_list_to_probes(const char *list)
                if (probes_capacity <= nr_probes) {
                        ret = grow_probes();
                        if (ret) {
-                               return ret;
+                               goto error;
                        }
                }
 
                /* Length 13 is "lttng-probe-" + \0 */
                name_len = strlen(next) + 13;
 
-               cur_mod = &probes[index];
+               cur_mod = &probes[nr_probes];
                cur_mod->name = zmalloc(name_len);
                if (!cur_mod->name) {
                        PERROR("malloc probe list");
-                       return -ENOMEM;
+                       ret = -ENOMEM;
+                       goto error;
                }
 
                ret = snprintf(cur_mod->name, name_len, "lttng-probe-%s", next);
                if (ret < 0) {
                        PERROR("snprintf modprobe name");
-                       return -ENOMEM;
+                       ret = -ENOMEM;
+                       goto error;
                }
 
-               cur_mod++;
                nr_probes++;
        }
 
        free(tmp_list);
-
        return 0;
+
+error:
+       free(tmp_list);
+       free_probes();
+       return ret;
 }
 
 /*
@@ -429,15 +442,14 @@ int modprobe_lttng_data(void)
        if (list) {
                /* User-specified probes. */
                ret = append_list_to_probes(list);
-
                if (ret) {
                        return ret;
                }
        } else {
                /* Default probes. */
                int def_len = ARRAY_SIZE(kern_modules_probes_default);
-               probes = zmalloc(sizeof(*probes) * def_len);
 
+               probes = zmalloc(sizeof(*probes) * def_len);
                if (!probes) {
                        PERROR("malloc probe list");
                        return -ENOMEM;
@@ -450,7 +462,8 @@ int modprobe_lttng_data(void)
 
                        if (!name) {
                                PERROR("strdup probe item");
-                               return -ENOMEM;
+                               ret = -ENOMEM;
+                               goto error;
                        }
 
                        probes[i].name = name;
@@ -469,12 +482,20 @@ int modprobe_lttng_data(void)
        if (list) {
                ret = append_list_to_probes(list);
                if (ret) {
-                       return ret;
+                       goto error;
                }
        }
 
        /*
         * Load probes modules now.
         */
-       return modprobe_lttng(probes, nr_probes, LTTNG_MOD_OPTIONAL);
+       ret = modprobe_lttng(probes, nr_probes, LTTNG_MOD_OPTIONAL);
+       if (ret) {
+               goto error;
+       }
+       return ret;
+
+error:
+       free_probes();
+       return ret;
 }