take care of param.nodes

1 files changed, 7 insertions, 3 deletions

diff --git a/lf_fifo_test.c b/lf_fifo_test.c
index 0b41aec..7d6a165 100644
--- a/lf_fifo_test.c
+++ b/lf_fifo_test.c
@@ -94,8 +94,12 @@ void* aggressive_push( void* param )
 
    params = (thread_params_t*)param;
    fifo = params->fifo;
-   nodes = params->nodes = malloc( sizeof(node_t)*params->n);
-   if (nodes==NULL) _failure("nodes malloc failure");
+   nodes = params->nodes;
+   if (nodes==NULL)
+     {
+        nodes = params->nodes = malloc( sizeof(node_t)*params->n);
+        if (nodes==NULL) _failure("nodes malloc failure");
+     }
 
    for(i=0; i<params->n; i++)
      {
@@ -186,7 +190,7 @@ static void run_aggressive_push_pop(int threads_n, int nodes_n)
    for(i=0; i<threads_n; i++)
      {
         j += params[i].n;
-        free(params[i].nodes);
+        if (params[i].nodes) free(params[i].nodes);
      }
    _check((j==(threads_n*nodes_n)),"poped nodes count failure");