From 2561a007b8d8999a4750046d0cfb3b1ad5af50ac Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期二, 09 四月 2024 15:29:32 +0800 Subject: [PATCH] test for perf --- src/bh_api.cpp | 1615 ++++++++++++++++++++++++++++++++++------------------------- 1 files changed, 932 insertions(+), 683 deletions(-) diff --git a/src/bh_api.cpp b/src/bh_api.cpp index f707e6f..c450bc2 100644 --- a/src/bh_api.cpp +++ b/src/bh_api.cpp @@ -3,21 +3,68 @@ #include "bus_server_socket_wrapper.h" #include "shm_mm_wrapper.h" #include "proc_def.h" +#include "mm.h" #include "usg_common.h" #include "bh_api.h" #include <pthread.h> #include <getopt.h> +#include "msg_mgr.h" +#include "../proto/source/error_msg.pb.h" #include "../proto/source/bhome_msg.pb.h" #include "../proto/source/bhome_msg_api.pb.h" + +#define TIME_WAIT 3 static Logger *logger = LoggerFactory::getLogger(); static int gRun_stat = 0; +static int gRun_flag = true; static void *gNetmod_socket = NULL; static pthread_mutex_t mutex; -static char errString[100] = { 0x00 }; +static pthread_t gTids; + + +static void *client_run_check(void *skptr) { + pthread_detach(pthread_self()); + + int data; + int sec, nsec; + int rv; + int key; + char buf[MAX_STR_LEN] = { 0x00 }; + void *buf_temp = NULL; + int size; + + sec = TIME_WAIT; + nsec = 0; + sprintf(buf, "%s", STR_EXEC); + data = net_mod_socket_int_get(gNetmod_socket); + while(gRun_flag == true) { + + rv = net_mod_socket_recvfrom(gNetmod_socket, &buf_temp, &size, &key, SVR_STR, data); + if (rv == 0) { + + if (strncmp((char *)buf_temp, STR_RET, strlen(STR_RET)) != 0) { + + rv = net_mod_socket_sendto_timeout(gNetmod_socket, buf, strlen(buf), key, sec, nsec, SVR_STR, data); + if (rv != 0) { + logger->error("the process check response failed with error: %s!\n", bus_strerror(rv)); + } + } else { + gRun_flag = false; + } + + BHFree(buf_temp, size); + + } else { + + logger->error("the process check failed with error: %s!\n", bus_strerror(rv)); + + } + } +} int BHRegister(const void *proc_info, const int proc_info_len, void **reply, int *reply_len, const int timeout_ms) { @@ -26,6 +73,7 @@ int count = 0; void *buf = NULL; int min = 0; + char *errString = NULL; ProcInfo pData; #if defined(PRO_DE_SERIALIZE) @@ -38,12 +86,10 @@ }_input; ::bhome_msg::ProcInfo input; - if(!input.ParseFromArray(proc_info, proc_info_len)) { + if ((!input.ParseFromArray(proc_info, proc_info_len)) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - + bus_errorset(rv); + goto exit_entry; } @@ -53,12 +99,10 @@ _input.private_info = input.private_info().c_str(); #else - if ((proc_info == NULL) || (proc_info_len == 0)) { + if ((proc_info == NULL) || (proc_info_len == 0) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x90, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - + bus_errorset(rv); + goto exit_entry; } #endif @@ -66,13 +110,16 @@ memset(&pData, 0x00, sizeof(ProcInfo)); if (gRun_stat == 0) { pthread_mutex_init(&mutex, NULL); + +#if defined(MSG_HANDLER) + msg_init(); +#endif } else { logger->error("the process has already registered!\n"); rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); goto exit_entry; } @@ -84,96 +131,117 @@ shm_mm_wrapper_init(SHM_RES_SIZE); #if defined(PRO_DE_SERIALIZE) - if (_input.proc_id != NULL) { + if (strlen(_input.proc_id) > 0) { count = strlen(_input.proc_id) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : count; strncpy(pData.proc_id, _input.proc_id, min); } - if (_input.name != NULL) { + if (strlen(_input.name) > 0) { count = strlen(_input.name) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1)? (MAX_STR_LEN -1) : count; strncpy(pData.name, _input.name, min); } - if (_input.public_info != NULL) { + if (strlen(_input.public_info) > 0) { count = strlen(_input.public_info) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1)? (MAX_STR_LEN - 1) : count; strncpy(pData.public_info, _input.public_info, min); } - if (_input.private_info != NULL) { + if (strlen(_input.private_info) > 0) { count = strlen(_input.private_info) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1)? (MAX_STR_LEN - 1): count; strncpy(pData.private_info, _input.private_info, min); } #else if (strlen((const char *)(((ProcInfo *)proc_info)->proc_id)) > 0) { count = strlen((const char *)(((ProcInfo *)proc_info)->proc_id)) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1): count; strncpy(pData.proc_id, ((ProcInfo *)proc_info)->proc_id, min); } if (strlen((const char *)(((ProcInfo *)proc_info)->name)) > 0) { count = strlen((const char *)(((ProcInfo *)proc_info)->name)) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : count; strncpy(pData.name, ((ProcInfo *)proc_info)->name, min); } if (strlen((const char *)(((ProcInfo *)proc_info)->public_info)) > 0) { count = strlen((const char *)(((ProcInfo *)proc_info)->public_info)) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : count; strncpy(pData.public_info, ((ProcInfo *)proc_info)->public_info, min); } if (strlen((const char *)(((ProcInfo *)proc_info)->private_info)) > 0) { count = strlen((const char *)(((ProcInfo *)proc_info)->private_info)) + 1; - min = count > MAX_STR_LEN ? MAX_STR_LEN : count; + min = count > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : count; strncpy(pData.private_info, ((ProcInfo *)proc_info)->private_info, min); } #endif + + if (strlen(pData.proc_id) == 0) { + rv = EBUS_INVALID_PARA; + + bus_errorset(rv); + + gRun_stat = 0; + pthread_mutex_unlock(&mutex); + + goto exit_entry; + } gNetmod_socket = net_mod_socket_open(); hashtable_t *hashtable = mm_get_hashtable(); key = hashtable_alloc_key(hashtable); net_mod_socket_bind(gNetmod_socket, key); + count = hashtable_alloc_key(hashtable); + rv = hashtable_alloc_key(hashtable); + net_mod_socket_int_set(gNetmod_socket, count); + net_mod_socket_svr_set(gNetmod_socket, rv); + sprintf(pData.int_info, "%d", count); + sprintf(pData.svr_info, "%d", rv); rv = net_mod_socket_reg(gNetmod_socket, &pData, sizeof(ProcInfo), NULL, 0, timeout_ms, PROC_REG); - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); pthread_mutex_unlock(&mutex); } else { rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); + goto exit_entry; } exit_entry: + errString = bus_strerror(0, 1); #if defined(PRO_DE_SERIALIZE) - ::bhome_msg::MsgCommonReply mcr; - mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); - mcr.mutable_errmsg()->set_errstring(errString); - *reply_len = mcr.ByteSizeLong(); - *reply = malloc(*reply_len); - mcr.SerializePartialToArray(*reply, *reply_len); + ::bhome_msg::MsgCommonReply mcr; + mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); + mcr.mutable_errmsg()->set_errstring(errString); + *reply_len = mcr.ByteSizeLong(); + *reply = malloc(*reply_len); + mcr.SerializePartialToArray(*reply, *reply_len); #else - min = strlen(errString) + 1; - buf = malloc(min) ; - memcpy(buf, errString, strlen(errString)); - *((char *)buf + min - 1) = '\0'; - - *reply = buf; - *reply_len = min; + min = strlen(errString) + 1; + buf = malloc(min) ; + memcpy(buf, errString, strlen(errString)); + *((char *)buf + min - 1) = '\0'; + + *reply = buf; + *reply_len = min; #endif - if (rv == 0) + if (rv == 0) { + gRun_flag = true; + pthread_create(&gTids, NULL, client_run_check, NULL); + return true; + } return false; } @@ -182,7 +250,11 @@ { int rv; int min; + int data; + int diff; void *buf = NULL; + char *errString = NULL; + struct timeval start, end; #if defined(PRO_DE_SERIALIZE) struct _ProcInfo_proto @@ -195,12 +267,10 @@ ::bhome_msg::ProcInfo input; - if(!input.ParseFromArray(proc_info, proc_info_len)) { + if(!input.ParseFromArray(proc_info, proc_info_len) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); goto exit_entry; } @@ -209,6 +279,13 @@ _input.name = input.name().c_str(); _input.public_info = input.public_info().c_str(); _input.private_info = input.private_info().c_str(); +#else + if ((reply == NULL) || (reply_len == NULL)) { + rv = EBUS_INVALID_PARA; + bus_errorset(rv); + + goto exit_entry; + } #endif if (gRun_stat == 0) { @@ -216,8 +293,7 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); goto exit_entry; } @@ -226,7 +302,23 @@ if (rv == 0) { rv = net_mod_socket_reg(gNetmod_socket, NULL, 0, NULL, 0, timeout_ms, PROC_UNREG); if (rv == 0) { - + gettimeofday(&start, NULL); + data = net_mod_socket_int_get(gNetmod_socket); + rv = net_mod_socket_sendto_timeout(gNetmod_socket, STR_RET, strlen(STR_RET), data, 3, 0); + if (rv != 0) { + logger->error("the process check response failed with error: %s!\n", bus_strerror(rv)); + } + + while(gRun_flag == true) { + sleep(1); + + gettimeofday(&end, NULL); + + diff = end.tv_sec - start.tv_sec; + if (diff >= TIME_DUR) + break; + }; + net_mod_socket_close(gNetmod_socket); gNetmod_socket = NULL; @@ -234,20 +326,20 @@ gRun_stat = 0; } - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); pthread_mutex_unlock(&mutex); } else { rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); + + goto exit_entry; } exit_entry: + errString = bus_strerror(0, 1); #if defined(PRO_DE_SERIALIZE) ::bhome_msg::MsgCommonReply mcr; mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); @@ -264,11 +356,13 @@ *reply = buf; *reply_len = min; #endif - - if (rv == 0) + + if (rv == 0) { return true; + } return false; + } int BHRegisterTopics(const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms) @@ -278,6 +372,7 @@ void *buf = NULL; int total = 0; int count = 0; + char *errString = NULL; char *topics_buf = NULL; #if defined(PRO_DE_SERIALIZE) @@ -288,14 +383,12 @@ }_input; ::bhome_msg::MsgTopicList input; - if(!input.ParseFromArray(topics, topics_len)) { + if(!input.ParseFromArray(topics, topics_len) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } _input.amount = input.topic_list_size(); @@ -309,14 +402,12 @@ total += strlen(_input.topics[i]) + 1; } #else - if ((topics == NULL) || (topics_len == 0)) { + if ((topics == NULL) || (topics_len == 0) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } total = topics_len; @@ -326,62 +417,52 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); goto exit_entry; } - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { + topics_buf = (char *)malloc(total); + if (topics_buf == NULL) { + + rv = EBUS_NO_MEM; + bus_errorset(rv); + + logger->error("in BHRegisterTopics: Out of memory!\n"); - topics_buf = (char *)malloc(total); - if (topics_buf == NULL) { - - rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - logger->error("in BHRegisterTopics: Out of memory!\n"); - - goto exit_entry; - } - memset(topics_buf, 0x00, total); + goto exit_entry; + } + memset(topics_buf, 0x00, total); #if defined(PRO_DE_SERIALIZE) - for (i = 0; i < _input.amount; i++) { - - len = strlen(_input.topics[i]); - strncpy(topics_buf + count, _input.topics[i], len); - - count += len; - - if ((_input.amount > 1) && (i < (_input.amount - 1))) { - strncpy(topics_buf + count, STR_MAGIC, strlen(STR_MAGIC)); - - count++; - } - } - -#else - memcpy(topics_buf, topics, topics_len); -#endif - - rv = net_mod_socket_reg(gNetmod_socket, topics_buf, strlen(topics_buf) + 1, NULL, 0, timeout_ms, PROC_REG_TCS); - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + for (i = 0; i < _input.amount; i++) { - free(topics_buf); - pthread_mutex_unlock(&mutex); - - } else { - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + len = strlen(_input.topics[i]); + strncpy(topics_buf + count, _input.topics[i], len); + + count += len; + + if ((_input.amount > 1) && (i < (_input.amount - 1))) { + strncpy(topics_buf + count, STR_MAGIC, strlen(STR_MAGIC)); + + count++; + } } +#else + memcpy(topics_buf, topics, topics_len); + count = topics_len; +#endif + + rv = net_mod_socket_reg(gNetmod_socket, topics_buf, count, NULL, 0, timeout_ms, PROC_REG_TCS); + + free(topics_buf); + + bus_errorset(rv); + exit_entry: + errString = bus_strerror(0, 1); + #if defined(PRO_DE_SERIALIZE) ::bhome_msg::MsgCommonReply mcr; mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); @@ -399,8 +480,9 @@ *reply_len = len; #endif - if (rv == 0) + if (rv == 0) { return true; + } return false; } @@ -411,10 +493,11 @@ int min; void *buf = NULL; int size; + char *errString = NULL; char topics_buf[MAX_STR_LEN] = { 0x00 }; ProcInfo_query *ptr = NULL; ProcInfo *Proc_ptr = NULL; - + #if defined(PRO_DE_SERIALIZE) struct _BHAddress { @@ -425,16 +508,25 @@ }_input0; const char *_input1; + + struct _MsgQueryTopicReply + { + std::string proc_id; + + unsigned long long mq_id; + long long abs_addr; + std::string ip; + int port; + } mtr_list[128]; + int mtr_list_num = 0; ::bhome_msg::BHAddress input0; ::bhome_msg::MsgQueryTopic input1; - if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(topic, topic_len)) { + if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(topic, topic_len) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } _input0.mq_id = input0.mq_id(); @@ -444,13 +536,11 @@ _input1 = input1.topic().c_str(); #else - if ((topic == NULL) || (topic_len == 0)) { + if ((topic == NULL) || (topic_len == 0) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } #endif @@ -458,68 +548,52 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); goto exit_entry; } - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { - #if defined(PRO_DE_SERIALIZE) - min = (strlen(_input1) > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : strlen(_input1)); - strncpy(topics_buf, _input1, min); + min = (strlen(_input1) > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : strlen(_input1)); + strncpy(topics_buf, _input1, min); #else - min = (topic_len > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : topic_len); - buf = const_cast<void *>(topic); - strncpy(topics_buf, (const char *)buf, min); + min = (topic_len > (MAX_STR_LEN - 1) ? (MAX_STR_LEN - 1) : topic_len); + buf = const_cast<void *>(topic); + strncpy(topics_buf, (const char *)buf, min); #endif - rv = net_mod_socket_reg(gNetmod_socket, topics_buf, strlen(topics_buf) + 1, &buf, &size, timeout_ms, PROC_QUE_TCS); - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - pthread_mutex_unlock(&mutex); - - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - } - -exit_entry: + rv = net_mod_socket_reg(gNetmod_socket, topics_buf, min, &buf, &size, timeout_ms, PROC_QUE_TCS); + + bus_errorset(rv); #if defined(PRO_DE_SERIALIZE) - - struct _MsgQueryTopicReply - { - std::string proc_id; - - unsigned long long mq_id; - long long abs_addr; - std::string ip; - int port; - }mtr_list[128]; - int mtr_list_num = 0; if (rv == 0) { - ptr = (ProcInfo_query *)((char *)buf + sizeof(int)); - mtr_list_num = ptr->num; + min = *(int *)buf; + if (min > 0) { + ptr = (ProcInfo_query *)((char *)buf + sizeof(int)); + mtr_list_num = ptr->num; + + if (mtr_list_num > sizeof(mtr_list) / sizeof(mtr_list[0])) { + mtr_list_num = sizeof(mtr_list) / sizeof(mtr_list[0]); + } - if (mtr_list_num > sizeof(mtr_list) / sizeof(mtr_list[0])) { - mtr_list_num = sizeof(mtr_list) / sizeof(mtr_list[0]); + Proc_ptr = &(ptr->procData); + for(int i = 0; i < mtr_list_num; i++) { + mtr_list[i].proc_id = (Proc_ptr + i)->proc_id; + mtr_list[i].mq_id = ID_RSV; + mtr_list[i].abs_addr = ABS_ID_RSV; + mtr_list[i].ip = "127.0.0.1"; + mtr_list[i].port = 5000; + } + } else { + mtr_list_num = 0; } - - for(int i = 0; i < mtr_list_num; i++) { - mtr_list[i].proc_id = ptr->procData.proc_id; - mtr_list[i].mq_id = 0x00; - mtr_list[i].abs_addr = 0x00; - mtr_list[i].ip = "192.168.1.1"; - mtr_list[i].port = 5000; - } + + free(buf); } + +exit_entry: + errString = bus_strerror(0, 1); ::bhome_msg::MsgQueryTopicReply mtr; mtr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); @@ -554,8 +628,9 @@ #endif - if (rv == 0) + if (rv == 0) { return true; + } return false; } @@ -566,6 +641,7 @@ void *buf = NULL; int size; int min; + char *errString = NULL; ProcInfo_sum *Proc_ptr = NULL; char data_buf[MAX_STR_LEN] = { 0x00 }; @@ -582,13 +658,12 @@ ::bhome_msg::BHAddress input0; ::bhome_msg::MsgQueryProc input1; - if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(query, query_len)) { + if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(query, query_len) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } _input0.mq_id = input0.mq_id(); @@ -596,39 +671,31 @@ _input0.ip = input0.ip().c_str(); _input0.port = input0.port(); _input1 = input1.proc_id().c_str(); +#else + if ((reply == NULL) || (reply_len == NULL)) { + rv = EBUS_INVALID_PARA; + bus_errorset(rv); + + goto exit_entry; + } #endif if (gRun_stat == 0) { logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - + bus_errorset(rv); + goto exit_entry; } - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { - - if (query != NULL) { - strncpy(data_buf, (char *)query, (sizeof(data_buf) - 1) > query_len ? query_len : (sizeof(data_buf) - 1)); - } - - rv = net_mod_socket_reg(gNetmod_socket, data_buf, strlen(data_buf), &buf, &size, timeout_ms, PROC_QUE_ATCS); - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - pthread_mutex_unlock(&mutex); - - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + if (query != NULL) { + strncpy(data_buf, (char *)query, (sizeof(data_buf) - 1) > query_len ? query_len : (sizeof(data_buf) - 1)); } + rv = net_mod_socket_reg(gNetmod_socket, data_buf, strlen(data_buf), &buf, &size, timeout_ms, PROC_QUE_ATCS); + bus_errorset(rv); + exit_entry: #if defined(PRO_DE_SERIALIZE) struct _MsgQueryProcReply @@ -653,50 +720,57 @@ if (mpr_list_num > (sizeof(mpr_list) / sizeof(mpr_list[0]))) { mpr_list_num = sizeof(mpr_list) / sizeof(mpr_list[0]); } - - Proc_ptr = (ProcInfo_sum *)((char *)buf + sizeof(int)); - for(int i = 0; i < mpr_list_num; i++) { - mpr_list[i].proc_id = (Proc_ptr + i)->procData.proc_id; - mpr_list[i].name = (Proc_ptr + i)->procData.name; - mpr_list[i].public_info = (Proc_ptr + i)->procData.public_info; - mpr_list[i].private_info = (Proc_ptr + i)->procData.private_info; - mpr_list[i].online = (Proc_ptr + i)->stat; - mpr_list[i].topic_list_num = (Proc_ptr + i)->list_num; - - for(int j = 0; j < mpr_list[i].topic_list_num; j++) - { - if (j == 0) { - mpr_list[i].topic_list[j] = (Proc_ptr + i)->reg_info; - } else if (j == 1) { - mpr_list[i].topic_list[j] = (Proc_ptr + i)->local_info; - } else if (j == 2) { - mpr_list[i].topic_list[j] = (Proc_ptr + i)->net_info; + + if (mpr_list_num > 0) { + Proc_ptr = (ProcInfo_sum *)((char *)buf + sizeof(int)); + for(int i = 0; i < mpr_list_num; i++) { + mpr_list[i].proc_id = (Proc_ptr + i)->procData.proc_id; + mpr_list[i].name = (Proc_ptr + i)->procData.name; + mpr_list[i].public_info = (Proc_ptr + i)->procData.public_info; + mpr_list[i].private_info = (Proc_ptr + i)->procData.private_info; + mpr_list[i].online = (Proc_ptr + i)->stat; + mpr_list[i].topic_list_num = (Proc_ptr + i)->list_num; + + for(int j = 0; j < mpr_list[i].topic_list_num; j++) + { + if (j == 0) { + mpr_list[i].topic_list[j] = (Proc_ptr + i)->reg_info; + } else if (j == 1) { + mpr_list[i].topic_list[j] = (Proc_ptr + i)->local_info; + } else if (j == 2) { + mpr_list[i].topic_list[j] = (Proc_ptr + i)->net_info; + } } } } - ::bhome_msg::MsgQueryProcReply mpr; - mpr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); - mpr.mutable_errmsg()->set_errstring(errString); - - for(int i = 0; i < mpr_list_num; i++) - { - ::bhome_msg::MsgQueryProcReply_Info *mpri = mpr.add_proc_list(); - mpri->mutable_proc()->set_proc_id(mpr_list[i].proc_id); - mpri->mutable_proc()->set_name(mpr_list[i].name); - mpri->mutable_proc()->set_public_info(mpr_list[i].public_info); - mpri->mutable_proc()->set_private_info(mpr_list[i].private_info); - mpri->set_online(mpr_list[i].online); - for(int j = 0; j < mpr_list[i].topic_list_num; j++) - { - mpri->mutable_topics()->add_topic_list(mpr_list[i].topic_list[j]); - } - } - - *reply_len = mpr.ByteSizeLong(); - *reply = malloc(*reply_len); - mpr.SerializePartialToArray(*reply,*reply_len); + free(buf); } + + errString = bus_strerror(0, 1); + + ::bhome_msg::MsgQueryProcReply mpr; + mpr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); + mpr.mutable_errmsg()->set_errstring(errString); + + for(int i = 0; i < mpr_list_num; i++) + { + ::bhome_msg::MsgQueryProcReply_Info *mpri = mpr.add_proc_list(); + mpri->mutable_proc()->set_proc_id(mpr_list[i].proc_id); + mpri->mutable_proc()->set_name(mpr_list[i].name); + mpri->mutable_proc()->set_public_info(mpr_list[i].public_info); + mpri->mutable_proc()->set_private_info(mpr_list[i].private_info); + mpri->set_online(mpr_list[i].online); + for(int j = 0; j < mpr_list[i].topic_list_num; j++) + { + mpri->mutable_topics()->add_topic_list(mpr_list[i].topic_list[j]); + } + } + + *reply_len = mpr.ByteSizeLong(); + *reply = malloc(*reply_len); + mpr.SerializePartialToArray(*reply, *reply_len); + #else if (rv == 0) { *reply = buf; @@ -712,11 +786,11 @@ } #endif - if (rv == 0) + if (rv == 0) { return true; + } return false; - } int BHSubscribeTopics(const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms) @@ -727,6 +801,7 @@ int count = 0; int len, i; void *buf = NULL; + char *errString = NULL; char *topics_buf = NULL; #if defined(PRO_DE_SERIALIZE) @@ -737,13 +812,12 @@ }_input; ::bhome_msg::MsgTopicList input; - if(!input.ParseFromArray(topics, topics_len)) { + if(!input.ParseFromArray(topics, topics_len) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } _input.amount = input.topic_list_size(); @@ -759,12 +833,11 @@ } #else - if ((topics == NULL) || (topics_len == 0)) { + if ((topics == NULL) || (topics_len == 0) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - goto exit_entry; + goto exit_entry; } #endif @@ -772,82 +845,69 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); + + goto exit_entry; + } + + topics_buf = (char *)malloc(total); + if (topics_buf == NULL) { + + rv = EBUS_NO_MEM; + bus_errorset(rv); + + logger->error("in BHSubscribeTopics: Out of memory!\n"); goto exit_entry; } - - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { - - topics_buf = (char *)malloc(total); - if (topics_buf == NULL) { - - rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - logger->error("in BHSubscribeTopics: Out of memory!\n"); - - goto exit_entry; - } - memset(topics_buf, 0x00, total); - + memset(topics_buf, 0x00, total); + #if defined(PRO_DE_SERIALIZE) - for (i = 0; i < _input.amount; i++) { - len = strlen(_input.topics[i]); - strncpy(topics_buf + count, _input.topics[i], min); + for (i = 0; i < _input.amount; i++) { + len = strlen(_input.topics[i]); + strncpy(topics_buf + count, _input.topics[i], len); + + count += len; + + if ((_input.amount > 1) && (i < (_input.amount - 1))) { + strncpy(topics_buf + count, STR_MAGIC, strlen(STR_MAGIC)); - count += len; - - if ((_input.amount > 1) && (i < (_input.amount - 1))) { - strncpy(topics_buf + count, STR_MAGIC, strlen(STR_MAGIC)); - - count++; - } + count++; } - -#else - memcpy(topics_buf, topics, topics_len); -#endif - - if (timeout_ms > 0) { - - sec = timeout_ms / 1000; - nsec = (timeout_ms - sec * 1000) * 1000 * 1000; - rv = net_mod_socket_sub_timeout(gNetmod_socket, topics_buf, strlen(topics_buf) + 1, sec, nsec); - - } else if (timeout_ms == 0) { - - rv = net_mod_socket_sub_nowait(gNetmod_socket, topics_buf, strlen(topics_buf) + 1); - - } else { - - rv = net_mod_socket_sub(gNetmod_socket, topics_buf, strlen(topics_buf) + 1); - - } - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - free(topics_buf); - pthread_mutex_unlock(&mutex); - - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); } -exit_entry: +#else + memcpy(topics_buf, topics, topics_len); + count = topics_len; +#endif + + if (timeout_ms > 0) { + + sec = timeout_ms / 1000; + nsec = (timeout_ms - sec * 1000) * 1000 * 1000; + rv = net_mod_socket_sub_timeout(gNetmod_socket, topics_buf, count, sec, nsec); + + } else if (timeout_ms == 0) { + + rv = net_mod_socket_sub_nowait(gNetmod_socket, topics_buf, count); + + } else { + + rv = net_mod_socket_sub(gNetmod_socket, topics_buf, count); + + } + +exit_entry: + errString = bus_strerror(0, 1); + + free(topics_buf); + #if defined(PRO_DE_SERIALIZE) ::bhome_msg::MsgCommonReply mcr; mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); mcr.mutable_errmsg()->set_errstring(errString); - *reply_len=mcr.ByteSizeLong(); - *reply=malloc(*reply_len); + *reply_len = mcr.ByteSizeLong(); + *reply = malloc(*reply_len); mcr.SerializePartialToArray(*reply,*reply_len); #else len = strlen(errString) + 1; @@ -858,12 +918,12 @@ *reply = buf; *reply_len = len; #endif - - if (rv == 0) + + if (rv == 0) { return true; + } return false; - } int BHSubscribeNetTopics(const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms) @@ -882,6 +942,7 @@ int BHHeartbeat(const void *proc_info, const int proc_info_len, void **reply, int *reply_len, const int timeout_ms) { int rv; + char *errString = NULL; #if defined(PRO_DE_SERIALIZE) struct _ProcInfo_proto @@ -896,11 +957,9 @@ if(!input.ParseFromArray(proc_info,proc_info_len)) { rv = EBUS_INVALID_PARA; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } _input.proc_id = input.proc_id().c_str(); @@ -909,22 +968,28 @@ _input.private_info = input.private_info().c_str(); rv = 0; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); + +exit_entry: + errString = bus_strerror(0, 1); ::bhome_msg::MsgCommonReply mcr; mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); mcr.mutable_errmsg()->set_errstring(errString); - *reply_len=mcr.ByteSizeLong(); - *reply=malloc(*reply_len); + *reply_len = mcr.ByteSizeLong(); + *reply = malloc(*reply_len); mcr.SerializePartialToArray(*reply,*reply_len); #endif - return true; + if (rv == 0) { + return true; + } + + return false; } #if defined(PRO_DE_SERIALIZE) -int BHPublish(const char *msgpub, const int msgpub_len, const int timeout_ms) +int BHPublish(const void *msgpub, const int msgpub_len, const int timeout_ms) #else int BHPublish(const char *topic, const char *content, const int timeout_ms) #endif @@ -946,8 +1011,7 @@ if(!input.ParseFromArray(msgpub, msgpub_len)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); return false; } @@ -958,10 +1022,9 @@ if ((topic == NULL) || (content == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - return false; + bus_errorset(rv); + + return false; } #endif @@ -969,51 +1032,36 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); return false; } - - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { + #if defined(PRO_DE_SERIALIZE) - if (timeout_ms > 0) { - rv = net_mod_socket_pub_timeout(gNetmod_socket, &node_arr, node_arr_len, _input.topic, strlen(_input.topic), _input.data, strlen(_input.data), timeout_ms); - } else if (timeout_ms == 0) { - rv = net_mod_socket_pub_nowait(gNetmod_socket, &node_arr, node_arr_len, _input.topic, strlen(_input.topic), _input.data, strlen(_input.data)); - - } else { - - rv = net_mod_socket_pub(gNetmod_socket, &node_arr, node_arr_len, _input.topic, strlen(_input.topic), _input.data, strlen(_input.data)); - } -#else - if (timeout_ms > 0) { - rv = net_mod_socket_pub_timeout(gNetmod_socket, &node_arr, node_arr_len, topic, strlen(topic), content, strlen(content), timeout_ms); - - } else if (timeout_ms == 0) { - rv = net_mod_socket_pub_nowait(gNetmod_socket, &node_arr, node_arr_len, topic, strlen(topic), content, strlen(content)); + if (timeout_ms > 0) { + rv = net_mod_socket_pub_timeout(gNetmod_socket, &node_arr, node_arr_len, _input.topic, strlen(_input.topic), _input.data, strlen(_input.data), timeout_ms); + } else if (timeout_ms == 0) { + rv = net_mod_socket_pub_nowait(gNetmod_socket, &node_arr, node_arr_len, _input.topic, strlen(_input.topic), _input.data, strlen(_input.data)); - } else { - rv = net_mod_socket_pub(gNetmod_socket, &node_arr, node_arr_len, topic, strlen(topic), content, strlen(content)); - } + } else { + + rv = net_mod_socket_pub(gNetmod_socket, &node_arr, node_arr_len, _input.topic, strlen(_input.topic), _input.data, strlen(_input.data)); + } +#else + if (timeout_ms > 0) { + rv = net_mod_socket_pub_timeout(gNetmod_socket, &node_arr, node_arr_len, topic, strlen(topic), content, strlen(content), timeout_ms); + + } else if (timeout_ms == 0) { + rv = net_mod_socket_pub_nowait(gNetmod_socket, &node_arr, node_arr_len, topic, strlen(topic), content, strlen(content)); + + } else { + rv = net_mod_socket_pub(gNetmod_socket, &node_arr, node_arr_len, topic, strlen(topic), content, strlen(content)); + } #endif - pthread_mutex_unlock(&mutex); - - if (rv > 0) - return true; - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - } - + if (rv > 0) + return true; + return false; } @@ -1022,11 +1070,12 @@ int rv; int len; void *buf; - int key; + int key = 0; 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 { @@ -1039,10 +1088,16 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; + } + + if ((msgpub == NULL) || (msgpub_len == NULL)) { + rv = EBUS_INVALID_PARA; + bus_errorset(rv); + + goto exit_entry; } if (timeout_ms > 0) { @@ -1066,11 +1121,38 @@ 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; + bus_errorset(rv); + + logger->error("in BHReadSub: Out of memory!\n"); + + goto exit_entry; + } + 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; + bus_errorset(rv); + + logger->error("in BHReadSub: Out of memory!\n"); + + free(topics_buf); + goto exit_entry; + } + memset(data_buf, 0x00, size - len + 10); + + len = strlen(topics_buf) + 1; strncpy(data_buf, (char *)buf + len, size - len); } @@ -1078,51 +1160,67 @@ #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); + free(topics_buf); + free(data_buf); + } + +exit_entry: + sprintf(buf_temp, "%d", key); - rsr.proc_id = topics_buf; + if ((proc_id != NULL) && (proc_id_len != NULL)) { + 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); - - ::bhome_msg::MsgPublish Mp; - Mp.set_topic(rsr.topic); - Mp.set_data(rsr.data.data()); - *msgpub_len = Mp.ByteSizeLong(); - *msgpub = malloc(*msgpub_len); - Mp.SerializePartialToArray(*msgpub, *msgpub_len); -#else - void *ptr; - if (len < size) { - ptr = malloc(size - len); - len = size - len; - memcpy(ptr, data_buf, len); - } else { - ptr = malloc(len); - memcpy(ptr, topics_buf, len); - } - *msgpub = ptr; - *msgpub_len = len; - - memset(topics_buf, 0x00, sizeof(topics_buf)); - sprintf(topics_buf, "%d", key); - - *proc_id_len = strlen(topics_buf); - *proc_id = malloc(*proc_id_len); - memcpy(*proc_id, topics_buf, *proc_id_len); - -#endif - - pthread_mutex_unlock(&mutex); - - } else { - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); } + + if (rv != 0) { + rsr.topic = STR_RSV; + if (data_buf != NULL) { + rsr.data = data_buf; + } else { + rsr.data = STR_RSV; + } + } + + ::bhome_msg::MsgPublish Mp; + Mp.set_topic(rsr.topic); + Mp.set_data(rsr.data.data()); + *msgpub_len = Mp.ByteSizeLong(); + *msgpub = malloc(*msgpub_len); + Mp.SerializePartialToArray(*msgpub, *msgpub_len); +#else + void *ptr; + if (len < size) { + ptr = malloc(size - len); + len = size - len; + memcpy(ptr, data_buf, len); + } else { + ptr = malloc(len); + memcpy(ptr, topics_buf, len); + } + *msgpub = ptr; + *msgpub_len = len; + + free(topics_buf); + free(data_buf); + + if ((proc_id != NULL) && (proc_id_len != NULL)) { + memset(buf_temp, 0x00, sizeof(buf_temp)); + sprintf(buf_temp, "%d", key); + + *proc_id_len = strlen(buf_temp); + *proc_id = malloc(*proc_id_len); + memcpy(*proc_id, buf_temp, *proc_id_len); + } + +#endif if (rv == 0) return true; @@ -1133,16 +1231,21 @@ int BHAsyncRequest(const void *remote, const int remote_len, const void *request, const int request_len, void **msg_id, int *msg_id_len) { int rv; - void *buf; + void *buf = NULL; int size; - int val; + int val = 0; int len; int min; + int data; int sec, nsec; + std::string str; std::string MsgID; int timeout_ms = 3000; - char topics_buf[MAX_STR_LEN * MAX_TOPICS_NUN] = { 0x00 }; - + char data_buf[MAX_STR_LEN] = { 0x00 }; + char buf_temp[MAX_STR_LEN] = { 0x00 }; + char *topics_buf = NULL; + hashtable_t *hashtable = mm_get_hashtable(); + #if defined(PRO_DE_SERIALIZE) struct _BHAddress { @@ -1163,10 +1266,9 @@ if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(request, request_len)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } _input0.mq_id = input0.mq_id(); @@ -1180,10 +1282,9 @@ if ((request == NULL) || (request_len == 0)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } #endif @@ -1191,86 +1292,115 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { #if defined(PRO_DE_SERIALIZE) - strncpy(topics_buf, _input1.topic, (sizeof(topics_buf) - 1) > strlen(_input1.topic) ? strlen(_input1.topic) : (sizeof(topics_buf) - 1)); + strncpy(buf_temp, _input1.topic, (sizeof(buf_temp) - 1) > strlen(_input1.topic) ? strlen(_input1.topic) : (sizeof(buf_temp) - 1)); #else - strncpy(topics_buf, (char *)request, (sizeof(topics_buf) - 1) > strlen((char *)request) ? strlen((char *)request) : (sizeof(topics_buf) - 1)); + strncpy(buf_temp, (char *)request, (sizeof(buf_temp) - 1) > strlen((char *)request) ? strlen((char *)request) : (sizeof(buf_temp) - 1)); #endif - - rv = net_mod_socket_reg(gNetmod_socket, topics_buf, strlen(topics_buf), &buf, &size, timeout_ms, PROC_QUE_STCS); - if (rv == 0) { - val = atoi((char *)buf); + str = buf_temp; + val = net_mod_socket_buf_data_get(gNetmod_socket, str); + if ((val > 0) && (hashtable_get(hashtable, val) != NULL)) { - free(buf); + rv = 0; - if (val > 0) { - - len = strlen(topics_buf); -#if defined(PRO_DE_SERIALIZE) - min = (sizeof(topics_buf) - 1 - len ) > strlen(_input1.data) ? strlen(_input1.data) : (sizeof(topics_buf) - 1 - len ); - strncpy(topics_buf + len + 1, _input1.data, min); - len += (min + 1); -#endif - if (timeout_ms > 0) { - - sec = timeout_ms / 1000; - nsec = (timeout_ms - sec * 1000) * 1000 * 1000; - - rv = net_mod_socket_sendto_timeout(gNetmod_socket, topics_buf, len, val, sec, nsec); - - } else if (timeout_ms == 0) { - - rv = net_mod_socket_sendto_nowait(gNetmod_socket, topics_buf, len, val); - - } else { - - rv = net_mod_socket_sendto(gNetmod_socket, topics_buf, len, val); - } - } else { - - rv = EBUS_RES_UNSUPPORT; - - } - } - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - pthread_mutex_unlock(&mutex); - - if((msg_id == NULL) || (msg_id_len == NULL)) { - if (rv == 0) - return true; - - return false; - } } else { - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + if ((val > 0) && (hashtable_get(hashtable, val) == NULL)) { + net_mod_socket_buf_data_del(gNetmod_socket, str); + } + + rv = net_mod_socket_reg(gNetmod_socket, buf_temp, strlen(buf_temp), &buf, &size, timeout_ms, PROC_QUE_STCS); + if (rv == 0) { + + len = size > (sizeof(data_buf) - 1) ? (sizeof(data_buf) - 1) : size; + memcpy(data_buf, (char *)buf, len); + val = atoi((char *)data_buf); + if (val > 0) { + str = buf_temp; + net_mod_socket_buf_data_set(gNetmod_socket, str, val); + } + + free(buf); + } + } + + if (rv == 0) { + if (val > 0) { + + len = strlen(buf_temp) + 1; +#if defined(PRO_DE_SERIALIZE) + len += strlen(_input1.data); +#endif + + topics_buf = (char *)malloc(len); + if (topics_buf == NULL) { + + rv = EBUS_NO_MEM; + bus_errorset(rv); + + logger->error("in BHAsyncRequest: Out of memory!\n"); + + goto exit_entry; + } + memset(topics_buf, 0x00, len); + + strncpy(topics_buf, buf_temp, strlen(buf_temp) + 1); +#if defined(PRO_DE_SERIALIZE) + strncpy(topics_buf + strlen(buf_temp) + 1, _input1.data, strlen(_input1.data)); +#endif + + data = net_mod_socket_svr_get(gNetmod_socket); + if (timeout_ms > 0) { + + sec = timeout_ms / 1000; + nsec = (timeout_ms - sec * 1000) * 1000 * 1000; + + rv = net_mod_socket_sendto_timeout(gNetmod_socket, topics_buf, len, val, sec, nsec, SVR_STR, data); + + } else if (timeout_ms == 0) { + + rv = net_mod_socket_sendto_nowait(gNetmod_socket, topics_buf, len, val, SVR_STR, data); + + } else { + + rv = net_mod_socket_sendto(gNetmod_socket, topics_buf, len, val, SVR_STR, data); + } + + free(topics_buf); + + } else { + + rv = EBUS_RES_UNSUPPORT; + + } + } + + bus_errorset(rv); + +exit_entry: + if((msg_id == NULL) || (msg_id_len == NULL)) { + if (rv == 0) + return true; + + return false; } - if (rv == 0) { - memset(topics_buf, 0x00, sizeof(topics_buf)); - sprintf(topics_buf, "%d", val); - MsgID = topics_buf; + memset(buf_temp, 0x00, sizeof(buf_temp)); + sprintf(buf_temp, "%d", val); + MsgID = buf_temp; - *msg_id_len = MsgID.size(); - *msg_id = malloc(*msg_id_len); - memcpy(*msg_id, MsgID.data(), *msg_id_len); + *msg_id_len = MsgID.size(); + *msg_id = malloc(*msg_id_len); + memcpy(*msg_id, MsgID.data(), *msg_id_len); + if (rv == 0) return true; - } return false; @@ -1280,19 +1410,24 @@ void **reply, int *reply_len, const int timeout_ms) { int rv; - void *buf; + void *buf = NULL; int size; int val; int min, len; + int data; + std::string str; net_node_t node; int node_size; int recv_arr_size; + char data_buf[MAX_STR_LEN] = { 0x00 }; net_mod_recv_msg_t *recv_arr; net_mod_err_t *errarr; int errarr_size = 0; - int sec, nsec; - char topics_buf[MAX_STR_LEN] = { 0x00 }; - + char *errString = NULL; + char buf_temp[MAX_STR_LEN] = { 0x00 }; + char *topics_buf = NULL; + hashtable_t *hashtable = mm_get_hashtable(); + struct _RequestReply { std::string proc_id; @@ -1316,30 +1451,29 @@ ::bhome_msg::BHAddress input0; ::bhome_msg::MsgRequestTopic input1; - if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(request, request_len)) { + if (!input0.ParseFromArray(remote, remote_len) || !input1.ParseFromArray(request, request_len) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } + memset(&node, 0x00, sizeof(node)); _input0.mq_id = input0.mq_id(); - _input0.abs_addr = input0.abs_addr(); - _input0.ip = input0.ip().c_str(); - _input0.port = input0.port(); - _input1.topic = input1.topic().c_str(); - _input1.data = input1.data().c_str(); + _input0.abs_addr = input0.abs_addr(); + _input0.ip = input0.ip().c_str(); + _input0.port = input0.port(); + _input1.topic = input1.topic().c_str(); + _input1.data = input1.data().c_str(); #else - if ((request == NULL) || (request_len == 0)) { + if ((request == NULL) || (request_len == 0) || (reply == NULL) || (reply_len == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } #endif @@ -1347,97 +1481,165 @@ logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; } - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { #if defined(PRO_DE_SERIALIZE) - strncpy(topics_buf, _input1.topic, (sizeof(topics_buf) - 1) > strlen(_input1.topic) ? strlen(_input1.topic) : (sizeof(topics_buf) - 1)); + strncpy(buf_temp, _input1.topic, (sizeof(buf_temp) - 1) > strlen(_input1.topic) ? strlen(_input1.topic) : (sizeof(buf_temp) - 1)); #else - strncpy(topics_buf, (char *)request, (sizeof(topics_buf) - 1) > request_len ? request_len : (sizeof(topics_buf) - 1)); + strncpy(buf_temp, (char *)request, (sizeof(buf_temp) - 1) > request_len ? request_len : (sizeof(buf_temp) - 1)); #endif - - rv = net_mod_socket_reg(gNetmod_socket, topics_buf, strlen(topics_buf), &buf, &size, timeout_ms, PROC_QUE_STCS); + + str = buf_temp; + val = net_mod_socket_buf_data_get(gNetmod_socket, str); + if ((val > 0) && (hashtable_get(hashtable, val) != NULL)) { + rv = 0; + } else { + if ((val > 0) && (hashtable_get(hashtable, val) == NULL)) { + net_mod_socket_buf_data_del(gNetmod_socket, str); + } + + rv = net_mod_socket_reg(gNetmod_socket, buf_temp, strlen(buf_temp), &buf, &size, timeout_ms, PROC_QUE_STCS); if (rv == 0) { - val = atoi((char *)buf); + + len = size > (sizeof(data_buf) - 1) ? (sizeof(data_buf) - 1) : size; + memcpy(data_buf, (char *)buf, len); + val = atoi((char *)data_buf); + if (val > 0) { + str = buf_temp; + net_mod_socket_buf_data_set(gNetmod_socket, str, val); + } free(buf); - - if (val > 0) { - memset(&node, 0x00, sizeof(node)); - - len = strlen(topics_buf); -#if defined(PRO_DE_SERIALIZE) - min = (sizeof(topics_buf) - 1 - len ) > strlen(_input1.data) ? strlen(_input1.data) : (sizeof(topics_buf) - 1 - len ); - strncpy(topics_buf + len + 1, _input1.data, min); - len += (min + 1); -#endif - - node.key = val; - rv = net_mod_socket_sendandrecv(gNetmod_socket, &node, 1, topics_buf, len, &recv_arr, &recv_arr_size, &errarr, &errarr_size); - if (rv > 0) { - if (recv_arr_size > 0) { - - node.key = recv_arr[0].key; - - memset(topics_buf, 0x00, sizeof(topics_buf)); - size = recv_arr[0].content_length; - buf = (char *)malloc(size); - strncpy((char *)buf, (char *)recv_arr[0].content, size); -#if !defined(PRO_DE_SERIALIZE) - *reply = buf; - *reply_len = size; -#endif - } - - net_mod_socket_free_recv_msg_arr(recv_arr, recv_arr_size); - - if(errarr_size > 0) { - free(errarr); - } - - rv = 0; - - } else { - rv = EBUS_TIMEOUT; - } - - } else { - rv = EBUS_RES_UNSUPPORT; - } } - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - if (rv == 0) { - memset(topics_buf, 0x00, sizeof(topics_buf)); - sprintf(topics_buf, "%d", node.key); - - rr.proc_id = topics_buf; - *proc_id_len = rr.proc_id.size(); - *proc_id = malloc(*proc_id_len); - memcpy(*proc_id, rr.proc_id.data(), *proc_id_len); - - memset(topics_buf, 0x00, sizeof(topics_buf)); - memcpy(topics_buf, buf, size); - rr.data = topics_buf; - } - - pthread_mutex_unlock(&mutex); - - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); } -#if defined(PRO_DE_SERIALIZE) + if (rv == 0) { + + if (val > 0) { + memset(&node, 0x00, sizeof(node)); + + len = strlen(buf_temp) + 1; +#if defined(PRO_DE_SERIALIZE) + len += strlen(_input1.data); +#endif + + topics_buf = (char *)malloc(len); + if (topics_buf == NULL) { + + rv = EBUS_NO_MEM; + bus_errorset(rv); + + logger->error("in BHRequest: Out of memory!\n"); + + goto exit_entry; + } + memset(topics_buf, 0x00, len); + + strncpy(topics_buf, buf_temp, strlen(buf_temp) + 1); +#if defined(PRO_DE_SERIALIZE) + strncpy(topics_buf + strlen(buf_temp) + 1, _input1.data, strlen(_input1.data)); +#endif + + node.key = val; + if (timeout_ms > 0) { + + rv = net_mod_socket_sendandrecv_timeout(gNetmod_socket, &node, 1, topics_buf, len, &recv_arr, &recv_arr_size, &errarr, &errarr_size, timeout_ms); + + } else if (timeout_ms == 0) { + + rv = net_mod_socket_sendandrecv_nowait(gNetmod_socket, &node, 1, topics_buf, len, &recv_arr, &recv_arr_size, &errarr, &errarr_size); + + } else { + + rv = net_mod_socket_sendandrecv(gNetmod_socket, &node, 1, topics_buf, len, &recv_arr, &recv_arr_size, &errarr, &errarr_size); + } + if (rv > 0) { + if (recv_arr_size > 0) { + + node.key = recv_arr[0].key; + + size = recv_arr[0].content_length; + buf = (char *)malloc(size); + if (buf == NULL) { + logger->error("in BHRequest: Out of memory!\n"); + + free(topics_buf); + + goto exit_entry; + } + memset((char *)buf, 0x00, size); + + strncpy((char *)buf, (char *)recv_arr[0].content, size); + } + + net_mod_socket_free_recv_msg_arr(recv_arr, recv_arr_size); + + if(errarr_size > 0) { + free(errarr); + } + + rv = 0; + } else { + + rv = EBUS_TIMEOUT; + } + + } else { + rv = EBUS_RES_UNSUPPORT; + } + + free(topics_buf); + } + + bus_errorset(rv); + + if ((proc_id != NULL) && (proc_id_len != NULL)) { + memset(buf_temp, 0x00, sizeof(buf_temp)); + sprintf(buf_temp, "%d", node.key); + + rr.proc_id = buf_temp; + *proc_id_len = rr.proc_id.size(); + *proc_id = malloc(*proc_id_len); + memcpy(*proc_id, rr.proc_id.c_str(), *proc_id_len); + } + + if (rv == 0) { + topics_buf = (char *)malloc(size + 10); + if (topics_buf == NULL) { + + rv = EBUS_NO_MEM; + bus_errorset(rv); + + logger->error("in BHRequest: Out of memory!\n"); + + goto exit_entry; + } + memset(topics_buf, 0x00, size + 10); + + memcpy(topics_buf, buf, size); + rr.data = topics_buf; + + free(buf); + free(topics_buf); + } else { + + rr.data = STR_RSV; + } + +exit_entry: + errString = bus_strerror(0, 1); + + if (rv != 0) { + if ((proc_id != NULL) && (proc_id_len != NULL)) { + *proc_id_len = 0; + *proc_id = NULL; + } + } + +#if defined(PRO_DE_SERIALIZE) ::bhome_msg::MsgRequestTopicReply mrt; mrt.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv)); mrt.mutable_errmsg()->set_errstring(errString); @@ -1446,18 +1648,16 @@ *reply = malloc(*reply_len); mrt.SerializePartialToArray(*reply, *reply_len); #else - if (rv > 0) { - min = strlen(errString) + 1; - buf = malloc(min); - memcpy(buf, errString, strlen(errString)); - *((char *)buf + min - 1) = '\0'; + min = strlen(errString) + 1; + buf = malloc(min); + memcpy(buf, errString, strlen(errString)); + *((char *)buf + min - 1) = '\0'; - *reply = buf; - *reply_len = min; - } + *reply = buf; + *reply_len = min; #endif - if (rv == 0) + if (rv == 0) return true; return false; @@ -1469,87 +1669,124 @@ void *buf; int key; int size; + int len; + int data; int sec, nsec; - char topics_buf[MAX_STR_LEN] = { 0x00 }; - + char buf_temp[MAX_STR_LEN] = { 0x00 }; + char *topics_buf = NULL; + + struct _ReadRequestReply + { + std::string proc_id; + std::string topic; + std::string data; + void *src; + } rrr; + if (gRun_stat == 0) { logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); - return false; + goto exit_entry; + } + + if ((request == NULL) || (request_len == 0) || (src == NULL)) { + rv = EBUS_INVALID_PARA; + bus_errorset(rv); + + goto exit_entry; } - rv = pthread_mutex_trylock(&mutex); + data = net_mod_socket_svr_get(gNetmod_socket); + if (timeout_ms > 0) { + + sec = timeout_ms / 1000; + nsec = (timeout_ms - sec * 1000) * 1000 * 1000; + + rv = net_mod_socket_recvfrom_timeout(gNetmod_socket, &buf, &size, &key, sec, nsec, SVR_STR, data); + + } else if (timeout_ms == 0) { + + rv = net_mod_socket_recvfrom_nowait(gNetmod_socket, &buf, &size, &key, SVR_STR, data); + + } else { + + rv = net_mod_socket_recvfrom(gNetmod_socket, &buf, &size, &key, SVR_STR, data); + } + if (rv == 0) { - if (timeout_ms > 0) { - - sec = timeout_ms / 1000; - nsec = (timeout_ms - sec * 1000) * 1000 * 1000; - - rv = net_mod_socket_recvfrom_timeout(gNetmod_socket, &buf, &size, &key, sec, nsec); - - } else if (timeout_ms == 0) { - - rv = net_mod_socket_recvfrom_nowait(gNetmod_socket, &buf, &size, &key); - - } else { - - rv = net_mod_socket_recvfrom(gNetmod_socket, &buf, &size, &key); - } - - if (rv == 0) { - struct _ReadRequestReply - { - std::string proc_id; - std::string topic; - std::string data; - void *src; - } rrr; - - sprintf(topics_buf, "%d", key); - rrr.proc_id = topics_buf; + + if ((proc_id != NULL) && (proc_id_len != NULL)) { + sprintf(buf_temp, "%d", key); + rrr.proc_id = buf_temp; *proc_id_len = rrr.proc_id.size(); *proc_id = malloc(*proc_id_len); memcpy(*proc_id, rrr.proc_id.data(), *proc_id_len); - - memset(topics_buf, 0x00, sizeof(topics_buf)); - memcpy(topics_buf, buf, size > sizeof(topics_buf) ? sizeof(topics_buf) : size); - rrr.topic = topics_buf; - rrr.data = topics_buf; - -#if defined(PRO_DE_SERIALIZE) - ::bhome_msg::MsgRequestTopic mrt; - mrt.set_topic(rrr.topic); - mrt.set_data(rrr.data.data()); - *request_len = mrt.ByteSizeLong(); - *request = malloc(*request_len); - mrt.SerializePartialToArray(*request,*request_len); -#else - *request = buf; - *request_len = size; -#endif - - buf = malloc(sizeof(int)); - *(int *)buf = key; - *src = buf; } - pthread_mutex_unlock(&mutex); - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + topics_buf = (char *)malloc(size + MIN_STR_LEN); + if (topics_buf == NULL) { + + rv = EBUS_NO_MEM; + bus_errorset(rv); + + logger->error("in BHReadRequest: Out of memory!\n"); + + goto exit_entry; + } + memset(topics_buf, 0x00, size + MIN_STR_LEN); - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + len = strlen((char *)buf); + if (len > size) { + len = size; + } + strncpy(topics_buf, (char *)buf, len); + rrr.topic = topics_buf; + + if (len < size) { + strncpy(topics_buf + len + 1, (char *)buf + len + 1, size - len - 1); + } + + rrr.data = topics_buf + len + 1; + + free(topics_buf); + free(buf); + + bus_errorset(rv); } - + +exit_entry: + + if (rv != 0) { + rrr.topic = STR_RSV; + rrr.data = STR_RSV; + + if ((proc_id != NULL) && (proc_id_len != NULL)) { + + *proc_id_len = 0; + *proc_id = NULL; + } + } + +#if defined(PRO_DE_SERIALIZE) + ::bhome_msg::MsgRequestTopic mrt; + mrt.set_topic(rrr.topic); + mrt.set_data(rrr.data.data()); + *request_len = mrt.ByteSizeLong(); + *request = malloc(*request_len); + mrt.SerializePartialToArray(*request,*request_len); +#else + *request = buf; + *request_len = size; +#endif + + buf = malloc(sizeof(int)); + *(int *)buf = key; + *src = buf; + if (rv == 0) return true; @@ -1559,57 +1796,49 @@ int BHSendReply(void *src, const void *reply, const int reply_len) { int rv; - + int data; + int sec = 3; + int nsec = 0; + const char *_input; + #if defined(PRO_DE_SERIALIZE) ::bhome_msg::MsgRequestTopicReply input; - if (!input.ParseFromArray(reply, reply_len)) { +if (!input.ParseFromArray(reply, reply_len) || (src == NULL)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); return false; } - const char *_input; _input = input.data().data(); - + #else if ((src == NULL) || (reply == NULL) || (reply_len == 0)) { rv = EBUS_INVALID_PARA; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); return false; } + + _input = (char *)reply; + #endif if (gRun_stat == 0) { logger->error("the process has not been registered yet!\n"); rv = EBUS_RES_NO; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); + bus_errorset(rv); return false; } - rv = pthread_mutex_trylock(&mutex); - if (rv == 0) { + data = net_mod_socket_svr_get(gNetmod_socket); + rv = net_mod_socket_sendto_timeout(gNetmod_socket, _input, strlen(_input), *(int *)src, sec, nsec, SVR_STR, data); - rv = net_mod_socket_sendto(gNetmod_socket, reply, reply_len, *(int *)src); - - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - - pthread_mutex_unlock(&mutex); - } else { - - rv = EBUS_RES_BUSY; - memset(errString, 0x00, sizeof(errString)); - strncpy(errString, bus_strerror(rv), sizeof(errString)); - } + bus_errorset(rv); if (rv == 0) return true; @@ -1629,6 +1858,7 @@ int BHGetLastError(void **msg, int *msg_len) { void *buf = NULL; + char *errString = bus_strerror(0, 1); buf = malloc(strlen(errString) + 1); @@ -1645,3 +1875,22 @@ return false; } + +int inter_key_get(void) +{ + if (gNetmod_socket != NULL) + return net_mod_socket_get_key(gNetmod_socket); + + return SHM_BUS_KEY; +} + +void *socket_data_get(void) +{ + return gNetmod_socket; +} + +void inter_key_set(int key) +{ + net_mod_socket_bind(gNetmod_socket, key); +} + -- Gitblit v1.8.0