Commit-ID:  d8e56c98b7ef96a31a64c69df24ab5d80f90e055
Gitweb:     http://git.kernel.org/tip/d8e56c98b7ef96a31a64c69df24ab5d80f90e055
Author:     Jiri Olsa <[email protected]>
AuthorDate: Tue, 3 Dec 2013 14:09:40 +0100
Committer:  Arnaldo Carvalho de Melo <[email protected]>
CommitDate: Wed, 4 Dec 2013 15:37:58 -0300

tools lib traceevent: Remove malloc_or_die from plugin_function.c

Removing malloc_or_die calls from plugin_function.c, replacing them and
factoring the code with standard realloc and error path.

Suggested-by: Namhyung Kim <[email protected]>
Signed-off-by: Jiri Olsa <[email protected]>
Cc: Corey Ashford <[email protected]>
Cc: David Ahern <[email protected]>
Cc: Frederic Weisbecker <[email protected]>
Cc: Ingo Molnar <[email protected]>
Cc: Namhyung Kim <[email protected]>
Cc: Paul Mackerras <[email protected]>
Cc: Peter Zijlstra <[email protected]>
Cc: Steven Rostedt <[email protected]>
Link: 
http://lkml.kernel.org/r/[email protected]
Signed-off-by: Arnaldo Carvalho de Melo <[email protected]>
---
 tools/lib/traceevent/plugin_function.c | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

diff --git a/tools/lib/traceevent/plugin_function.c 
b/tools/lib/traceevent/plugin_function.c
index 87acf9c..aad92ad 100644
--- a/tools/lib/traceevent/plugin_function.c
+++ b/tools/lib/traceevent/plugin_function.c
@@ -43,11 +43,17 @@ static void add_child(struct func_stack *stack, const char 
*child, int pos)
        if (pos < stack->size)
                free(stack->stack[pos]);
        else {
-               if (!stack->stack)
-                       stack->stack = malloc_or_die(sizeof(char *) * STK_BLK);
-               else
-                       stack->stack = realloc(stack->stack, sizeof(char *) *
-                                              (stack->size + STK_BLK));
+               char **ptr;
+
+               ptr = realloc(stack->stack, sizeof(char *) *
+                             (stack->size + STK_BLK));
+               if (!ptr) {
+                       warning("could not allocate plugin memory\n");
+                       return;
+               }
+
+               stack->stack = ptr;
+
                for (i = stack->size; i < stack->size + STK_BLK; i++)
                        stack->stack[i] = NULL;
                stack->size += STK_BLK;
@@ -64,10 +70,15 @@ static int add_and_get_index(const char *parent, const char 
*child, int cpu)
                return 0;
 
        if (cpu > cpus) {
-               if (fstack)
-                       fstack = realloc(fstack, sizeof(*fstack) * (cpu + 1));
-               else
-                       fstack = malloc_or_die(sizeof(*fstack) * (cpu + 1));
+               struct func_stack *ptr;
+
+               ptr = realloc(fstack, sizeof(*fstack) * (cpu + 1));
+               if (!ptr) {
+                       warning("could not allocate plugin memory\n");
+                       return 0;
+               }
+
+               fstack = ptr;
 
                /* Account for holes in the cpu count */
                for (i = cpus + 1; i <= cpu; i++)
--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to [email protected]
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to