From b8b5059222cc99637d4a1c30e7eab26ca9fd0308 Mon Sep 17 00:00:00 2001 From: fujuntang <fujuntang@smartai.com> Date: 星期二, 07 九月 2021 13:45:12 +0800 Subject: [PATCH] Add the dynamic buf. --- src/bh_api.cpp | 25 ++++++++++++++++--------- 1 files changed, 16 insertions(+), 9 deletions(-) diff --git a/src/bh_api.cpp b/src/bh_api.cpp index d773369..eef4870 100644 --- a/src/bh_api.cpp +++ b/src/bh_api.cpp @@ -1042,6 +1042,7 @@ int key; int size; int sec, nsec; + char buf_temp[100] = { 0x00 }; char *topics_buf = NULL; char *data_buf = NULL; @@ -1135,13 +1136,16 @@ #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); @@ -1168,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 -- Gitblit v1.8.0