From: Ziye Yang <optimist...@gmail.com>

This patch is used to fix the memory leak issue of logid.
We use the ASAN test in SPDK when intergrating DPDK and
find this memory leak issue.

Signed-off-by: Ziye Yang <ziye.y...@intel.com>
---
 lib/librte_eal/linuxapp/eal/eal.c     | 21 ++++++++++++++++++++-
 lib/librte_eal/linuxapp/eal/eal_log.c | 11 ++++++++++-
 2 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/lib/librte_eal/linuxapp/eal/eal.c 
b/lib/librte_eal/linuxapp/eal/eal.c
index e59ac65..e150200 100644
--- a/lib/librte_eal/linuxapp/eal/eal.c
+++ b/lib/librte_eal/linuxapp/eal/eal.c
@@ -793,7 +793,7 @@ static void rte_eal_init_alert(const char *msg)
        int i, fctret, ret;
        pthread_t thread_id;
        static rte_atomic32_t run_once = RTE_ATOMIC32_INIT(0);
-       const char *logid;
+       char *logid;
        char cpuset[RTE_CPU_AFFINITY_STR_LEN];
        char thread_name[RTE_MAX_THREAD_NAME_LEN];
 
@@ -812,6 +812,12 @@ static void rte_eal_init_alert(const char *msg)
 
        logid = strrchr(argv[0], '/');
        logid = strdup(logid ? logid + 1: argv[0]);
+       if (!logid) {
+               rte_eal_init_alert("Cannot allocate memory for logid\n");
+               rte_errno = ENOMEM;
+               rte_atomic32_clear(&run_once);
+               return -1;
+       }
 
        thread_id = pthread_self();
 
@@ -823,6 +829,8 @@ static void rte_eal_init_alert(const char *msg)
        if (rte_eal_cpu_init() < 0) {
                rte_eal_init_alert("Cannot detect lcores.");
                rte_errno = ENOTSUP;
+               rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
 
@@ -831,6 +839,7 @@ static void rte_eal_init_alert(const char *msg)
                rte_eal_init_alert("Invalid 'command line' arguments.");
                rte_errno = EINVAL;
                rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
 
@@ -838,12 +847,14 @@ static void rte_eal_init_alert(const char *msg)
                rte_eal_init_alert("Cannot init plugins\n");
                rte_errno = EINVAL;
                rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
 
        if (eal_option_device_parse()) {
                rte_errno = ENODEV;
                rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
 
@@ -851,6 +862,8 @@ static void rte_eal_init_alert(const char *msg)
 
        if (rte_eal_intr_init() < 0) {
                rte_eal_init_alert("Cannot init interrupt-handling thread\n");
+               rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
 
@@ -861,6 +874,8 @@ static void rte_eal_init_alert(const char *msg)
                rte_eal_init_alert("failed to init mp channel\n");
                if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
                        rte_errno = EFAULT;
+                       rte_atomic32_clear(&run_once);
+                       free(logid);
                        return -1;
                }
        }
@@ -869,6 +884,7 @@ static void rte_eal_init_alert(const char *msg)
                rte_eal_init_alert("Cannot scan the buses for devices\n");
                rte_errno = ENODEV;
                rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
 
@@ -893,6 +909,7 @@ static void rte_eal_init_alert(const char *msg)
                        rte_eal_init_alert("Cannot get hugepage information.");
                        rte_errno = EACCES;
                        rte_atomic32_clear(&run_once);
+                       free(logid);
                        return -1;
                }
        }
@@ -919,8 +936,10 @@ static void rte_eal_init_alert(const char *msg)
                rte_eal_init_alert("Cannot init logging.");
                rte_errno = ENOMEM;
                rte_atomic32_clear(&run_once);
+               free(logid);
                return -1;
        }
+       free(logid);
 
 #ifdef VFIO_PRESENT
        if (rte_eal_vfio_setup() < 0) {
diff --git a/lib/librte_eal/linuxapp/eal/eal_log.c 
b/lib/librte_eal/linuxapp/eal/eal_log.c
index 9d02ddd..dcc1583 100644
--- a/lib/librte_eal/linuxapp/eal/eal_log.c
+++ b/lib/librte_eal/linuxapp/eal/eal_log.c
@@ -19,6 +19,9 @@
 
 #include "eal_private.h"
 
+#define DEFAULT_LOG_ID_LEN 128
+char g_logid[DEFAULT_LOG_ID_LEN];
+
 /*
  * default log function
  */
@@ -49,12 +52,18 @@
 rte_eal_log_init(const char *id, int facility)
 {
        FILE *log_stream;
+       int str_len;
 
        log_stream = fopencookie(NULL, "w+", console_log_func);
        if (log_stream == NULL)
                return -1;
 
-       openlog(id, LOG_NDELAY | LOG_PID, facility);
+       str_len = strlen(id);
+       if(str_len >= DEFAULT_LOG_ID_LEN)
+               return -1;
+
+       strncpy(g_logid, id, str_len);
+       openlog(g_logid, LOG_NDELAY | LOG_PID, facility);
 
        eal_log_set_default(log_stream);
 
-- 
1.9.3

Reply via email to