From 589b4a02bd5d183074994af1a48da6bc1b32e483 Mon Sep 17 00:00:00 2001
From: liuxiaolong <liuxiaolong@aiotlink.com>
Date: 星期二, 07 九月 2021 16:14:28 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.5.22:9091/valib/c_bhomebus

---
 src/bh_api.cpp |   90 ++++++++++++++++++++++++++++++++++++--------
 1 files changed, 73 insertions(+), 17 deletions(-)

diff --git a/src/bh_api.cpp b/src/bh_api.cpp
index 438530e..4f623f8 100644
--- a/src/bh_api.cpp
+++ b/src/bh_api.cpp
@@ -1042,8 +1042,9 @@
   int key;
   int size;
   int sec, nsec;
-  char topics_buf[MAX_STR_LEN] = { 0x00 };
-  char data_buf[MAX_STR_LEN * 3] = { 0x00 }; 
+  char buf_temp[100] = { 0x00 };
+  char *topics_buf = NULL;
+  char *data_buf = NULL; 
 
   struct _ReadSubReply
   {
@@ -1091,11 +1092,43 @@
     if (len > size) {
       len = size;
     }
-    strncpy(topics_buf, (char *)buf, len > (sizeof(topics_buf) - 1) ? (sizeof(topics_buf) - 1) : len);
+    
+    topics_buf = (char *)malloc(len + 10);
+    if (topics_buf == NULL) {
+      
+      rv = EBUS_NO_MEM;
+      memset(errString, 0x00, sizeof(errString));
+      strncpy(errString, bus_strerror(rv), sizeof(errString));
+      
+      logger->error("in BHRequest: Out of memory!\n");
+      
+      pthread_mutex_unlock(&mutex);
+
+      return false;
+    }
+    memset(topics_buf, 0x00, len + 10);
+    
+    strncpy(topics_buf, (char *)buf, len);
     
     if (len < size) {
-      len = strlen(topics_buf) + 1;
+      
+      data_buf = (char *)malloc(size - len + 10);
+      if (data_buf == NULL) {
+        
+        rv = EBUS_NO_MEM;
+        memset(errString, 0x00, sizeof(errString));
+        strncpy(errString, bus_strerror(rv), sizeof(errString));
+        
+        logger->error("in BHRequest: Out of memory!\n");
+        
+        free(topics_buf);
+        pthread_mutex_unlock(&mutex);
 
+        return false;
+      }
+      memset(data_buf, 0x00, size - len + 10);
+
+      len = strlen(topics_buf) + 1;
       strncpy(data_buf, (char *)buf + len, size - len);
     }
 
@@ -1103,17 +1136,23 @@
 
 #if defined(PRO_DE_SERIALIZE)
     rsr.topic = topics_buf;
-    rsr.data = data_buf;
+    if (data_buf != NULL) {
+      rsr.data = data_buf;
+    } else {
+      rsr.data = topics_buf;
+    }
 
-    memset(topics_buf, 0x00, sizeof(topics_buf));
-    sprintf(topics_buf, "%d", key);
+    sprintf(buf_temp, "%d", key);
 
     if ((proc_id != NULL) && (proc_id_len != NULL)) {
-      rsr.proc_id = topics_buf;
+      rsr.proc_id = buf_temp;
       *proc_id_len = rsr.proc_id.size();
       *proc_id = malloc(*proc_id_len);
       memcpy(*proc_id, rsr.proc_id.data(), *proc_id_len);
     }
+    
+    free(topics_buf);
+    free(data_buf);
 
     ::bhome_msg::MsgPublish Mp; 
     Mp.set_topic(rsr.topic);
@@ -1133,14 +1172,17 @@
     }
     *msgpub = ptr;
     *msgpub_len = len;
-    
+   
+    free(topics_buf);
+    free(data_buf);
+
     if ((proc_id != NULL) && (proc_id_len != NULL)) {
-      memset(topics_buf, 0x00, sizeof(topics_buf));
-      sprintf(topics_buf, "%d", key);
+      memset(buf_temp, 0x00, sizeof(buf_temp));
+      sprintf(buf_temp, "%d", key);
       
-      *proc_id_len = strlen(topics_buf);
+      *proc_id_len = strlen(buf_temp);
       *proc_id = malloc(*proc_id_len);
-      memcpy(*proc_id, topics_buf, *proc_id_len);
+      memcpy(*proc_id, buf_temp, *proc_id_len);
     }
     
 #endif 
@@ -1512,12 +1554,26 @@
         memcpy(*proc_id, rr.proc_id.c_str(), *proc_id_len);
       }
 
-      memset(buf_temp, 0x00, sizeof(buf_temp));
-      memcpy(buf_temp, buf, size);
-      rr.data = buf_temp;
+      topics_buf = (char *)malloc(size + 10);
+      if (topics_buf == NULL) {
+        
+        rv = EBUS_NO_MEM;
+        memset(errString, 0x00, sizeof(errString));
+        strncpy(errString, bus_strerror(rv), sizeof(errString));
+        
+        logger->error("in BHRequest: Out of memory!\n");
+        
+        pthread_mutex_unlock(&mutex);
+
+        return false;
+      }
+      memset(topics_buf, 0x00, size + 10);
+      
+      memcpy(topics_buf, buf, size);
+      rr.data = topics_buf;
       
       free(buf);
-      
+      free(topics_buf);
     }
     
     pthread_mutex_unlock(&mutex);

--
Gitblit v1.8.0