fujuntang
2021-09-11 dc01e4cbb01e96d19b470a366bbe648d426ed171
src/bh_api.cpp
@@ -139,6 +139,12 @@
    gNetmod_socket = net_mod_socket_open();
    hashtable_t *hashtable = mm_get_hashtable();
    key = hashtable_alloc_key(hashtable);
    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);
    net_mod_socket_bind(gNetmod_socket, key);
  
    rv = net_mod_socket_reg(gNetmod_socket, &pData, sizeof(ProcInfo), NULL, 0, timeout_ms, PROC_REG);
@@ -932,8 +938,8 @@
  ::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 
@@ -1042,8 +1048,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 +1098,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 +1142,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 +1178,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 
@@ -1165,6 +1213,7 @@
  int val;
  int len;
  int min;
  int data;
  int sec, nsec;
  std::string MsgID;
  int timeout_ms = 3000;
@@ -1267,20 +1316,21 @@
        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);
          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);
          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);
          rv = net_mod_socket_sendto(gNetmod_socket, topics_buf, len, val, SVR_STR, data);
        } 
        free(topics_buf);
@@ -1335,6 +1385,7 @@
  int size;
  int val;
  int min, len;
  int data;
  net_node_t node;
  int node_size;  
  int recv_arr_size;
@@ -1427,6 +1478,7 @@
        len += strlen(_input1.data);
#endif
        data = net_mod_socket_svr_get(gNetmod_socket);
        topics_buf = (char *)malloc(len);
        if (topics_buf == NULL) {
          
@@ -1512,7 +1564,7 @@
        memcpy(*proc_id, rr.proc_id.c_str(), *proc_id_len);
      }
      topics_buf = (char *)malloc(size);
      topics_buf = (char *)malloc(size + 10);
      if (topics_buf == NULL) {
        
        rv = EBUS_NO_MEM;
@@ -1525,7 +1577,7 @@
        return false;
      }
      memset(topics_buf, 0x00, size);
      memset(topics_buf, 0x00, size + 10);
      
      memcpy(topics_buf, buf, size);
      rr.data = topics_buf;
@@ -1578,6 +1630,7 @@
  int key;
  int size;
  int len;
  int data;
  int sec, nsec;
  char buf_temp[MAX_STR_LEN] = { 0x00 };
  char *topics_buf = NULL;
@@ -1600,20 +1653,21 @@
      return false;
  }
  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);
    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);
    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);
    rv = net_mod_socket_recvfrom(gNetmod_socket, &buf, &size, &key, SVR_STR, data);
  }
  if (rv == 0) {
@@ -1693,6 +1747,7 @@
int BHSendReply(void *src, const void *reply, const int reply_len)
{
  int rv;
  int data;
  const char *_input;
  
#if defined(PRO_DE_SERIALIZE)
@@ -1735,7 +1790,8 @@
  rv = pthread_mutex_trylock(&mutex);
  if (rv == 0) {
    rv = net_mod_socket_sendto(gNetmod_socket, _input, strlen(_input), *(int *)src);
    data = net_mod_socket_svr_get(gNetmod_socket);
    rv = net_mod_socket_sendto(gNetmod_socket, _input, strlen(_input), *(int *)src, SVR_STR, data);
    memset(errString, 0x00, sizeof(errString));
    strncpy(errString, bus_strerror(rv), sizeof(errString));