1 files changed, 20 insertions, 9 deletions

diff --git a/tools/lib/traceevent/plugin_function.c b/tools/lib/traceevent/plugin_function.c
index 87acf9c77948..aad92ad5e96f 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++)