From 758438289fc45829a8f6cef1b42afed0a1a8cb60 Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期三, 03 二月 2021 15:59:58 +0800
Subject: [PATCH] uuid msg
---
src/socket/shm_socket.cpp | 386 ++++++++++++++++++++++++++++++++++++++----------------
1 files changed, 270 insertions(+), 116 deletions(-)
diff --git a/src/socket/shm_socket.cpp b/src/socket/shm_socket.cpp
index 690124f..bddcec0 100644
--- a/src/socket/shm_socket.cpp
+++ b/src/socket/shm_socket.cpp
@@ -4,6 +4,7 @@
#include <map>
#include <cassert>
#include "bus_error.h"
+#include "sole.h"
static Logger *logger = LoggerFactory::getLogger();
@@ -18,6 +19,12 @@
static void _destrory_socket_perthread(void *tmp_socket);
static void _create_socket_key_perthread(void);
+
+static int shm_recvpakfrom(shm_socket_t *sockt, shm_packet_t *recvpak , const struct timespec *timeout, int flag);
+
+
+static int shm_sendpakto(shm_socket_t *sockt, const shm_packet_t *sendpak,
+ const int key, const struct timespec *timeout, const int flag);
// 妫�鏌ey鏄惁宸茬粡琚娇鐢紝 鏈浣跨敤鍒欑粦瀹歬ey
static LockFreeQueue<shm_packet_t> * shm_socket_bind_queue(int key, bool force) {
@@ -78,11 +85,12 @@
pthread_mutexattr_t mtxAttr;
logger->debug("shm_open_socket\n");
- shm_socket_t *socket = (shm_socket_t *)calloc(1, sizeof(shm_socket_t));
- socket->socket_type = socket_type;
- socket->key = 0;
- socket->force_bind = false;
- socket->queue = NULL;
+ // shm_socket_t *socket = (shm_socket_t *)calloc(1, sizeof(shm_socket_t));
+ shm_socket_t *sockt = new shm_socket_t;
+ sockt->socket_type = socket_type;
+ sockt->key = 0;
+ sockt->force_bind = false;
+ sockt->queue = NULL;
s = pthread_mutexattr_init(&mtxAttr);
@@ -91,7 +99,7 @@
s = pthread_mutexattr_settype(&mtxAttr, PTHREAD_MUTEX_ERRORCHECK);
if (s != 0)
err_exit(s, "pthread_mutexattr_settype");
- s = pthread_mutex_init(&(socket->mutex), &mtxAttr);
+ s = pthread_mutex_init(&(sockt->mutex), &mtxAttr);
if (s != 0)
err_exit(s, "pthread_mutex_init");
@@ -99,45 +107,44 @@
if (s != 0)
err_exit(s, "pthread_mutexattr_destroy");
- return socket;
+ return sockt;
}
-int shm_close_socket(shm_socket_t *socket) {
+int shm_close_socket(shm_socket_t *sockt) {
int s;
logger->debug("shm_close_socket\n");
- if(socket->queue != NULL) {
- delete socket->queue;
- socket->queue = NULL;
+ if(sockt->queue != NULL) {
+ delete sockt->queue;
+ sockt->queue = NULL;
}
- s = pthread_mutex_destroy(&(socket->mutex) );
+ s = pthread_mutex_destroy(&(sockt->mutex) );
if(s != 0) {
err_exit(s, "shm_close_socket");
}
- free(socket);
+ free(sockt);
return 0;
}
-int shm_socket_bind(shm_socket_t *socket, int key) {
- socket->key = key;
+int shm_socket_bind(shm_socket_t *sockt, int key) {
+ sockt->key = key;
return 0;
}
-int shm_socket_force_bind(shm_socket_t *socket, int key) {
- socket->force_bind = true;
- socket->key = key;
+int shm_socket_force_bind(shm_socket_t *sockt, int key) {
+ sockt->force_bind = true;
+ sockt->key = key;
return 0;
}
-int shm_socket_get_key(shm_socket_t *sk){
- return sk->key;
+int shm_socket_get_key(shm_socket_t *sockt){
+ return sockt->key;
}
-
// 鐭繛鎺ユ柟寮忓彂閫�
@@ -145,121 +152,162 @@
const int key, const struct timespec *timeout, const int flag) {
int rv;
-
- hashtable_t *hashtable = mm_get_hashtable();
-
- if( sockt->queue != NULL)
- goto LABEL_PUSH;
- {
- if ((rv = pthread_mutex_lock(&(sockt->mutex))) != 0)
- err_exit(rv, "shm_sendto : pthread_mutex_lock");
+ shm_packet_t sendpak;
+ sendpak.key = sockt->key;
+ sendpak.size = size;
+ sendpak.buf = mm_malloc(size);
+ memcpy(sendpak.buf, buf, size);
+ rv = shm_sendpakto(sockt, &sendpak, key, timeout, flag);
+ return rv;
+}
- if (sockt->queue == NULL) {
- if (sockt->key == 0) {
- sockt->key = hashtable_alloc_key(hashtable);
- }
- sockt->queue = shm_socket_bind_queue( sockt->key, sockt->force_bind);
- if(sockt->queue == NULL ) {
- logger->error("%s. key = %d", bus_strerror(EBUS_KEY_INUSED), sockt->key);
- return EBUS_KEY_INUSED;
- }
+
+
+int shm_sendandrecv(shm_socket_t *sockt, const void *send_buf,
+ const int send_size, const int key, void **recv_buf,
+ int *recv_size, const struct timespec *timeout, int flags) {
+
+ int rv, tryn = 3;
+ shm_packet_t sendpak;
+ shm_packet_t recvpak;
+ std::map<std::string, shm_packet_t>::iterator recvbufIter;
+ std::string uuid = sole::uuid4().str();
+
+ sendpak.key = sockt->key;
+ sendpak.size = send_size;
+ sendpak.buf = mm_malloc(send_size);
+ memcpy(sendpak.buf, send_buf, send_size);
+ memcpy(sendpak.uuid, uuid.c_str(), uuid.length() + 1);
+ // uuid.copy(sendpak.uuid, sizeof sendpak.uuid);
+ rv = shm_sendpakto(sockt, &sendpak, key, timeout, flags);
+
+ if(rv != 0) {
+ return rv;
+ }
+
+ while(true) {
+ tryn--;
+ recvbufIter = sockt->recvbuf.find(uuid);
+ if(recvbufIter != sockt->recvbuf.end()) {
+ // 鍦ㄧ紦瀛橀噷鏌ュ埌浜哢UID鍖归厤鎴愬姛鐨�
+logger->debug("get from recvbuf: %s", uuid.c_str());
+ recvpak = recvbufIter->second;
+ sockt->recvbuf.erase(recvbufIter);
+ break;
}
- if ((rv = pthread_mutex_unlock(&(sockt->mutex))) != 0)
- err_exit(rv, "shm_sendto : pthread_mutex_unlock");
-
+ rv = shm_recvpakfrom(sockt, &recvpak, timeout, flags);
+
+ if (rv != 0) {
+
+ if(rv == ETIMEDOUT)
+ return EBUS_TIMEOUT;
+
+ logger->debug("%d shm_recvfrom failed %s", shm_socket_get_key(sockt), bus_strerror(rv));
+ return rv;
+ }
+
+logger->debug("send uuid:%s, recv uuid: %s", uuid.c_str(), recvpak.uuid);
+ if (strncmp(uuid.c_str(), recvpak.uuid, sizeof recvpak.uuid) == 0) {
+ // 鍙戦�佷笌鎺ュ彈鐨刄UID鍖归厤鎴愬姛
+ break;
+ } else {
+ // 绛旈潪鎵�闂紝鏀惧埌缂撳瓨閲�
+ sockt->recvbuf.insert({recvpak.uuid, recvpak});
+ continue;
+ }
+
+ if(tryn == 0) {
+ // 灏濊瘯浜唗ryn娆￠兘娌℃湁鎴愬姛
+ return EBUS_RECVFROM_WRONG_END;
+ }
+
}
-
- LABEL_PUSH:
- if (key == sockt->key) {
- logger->error( "can not send to your self!");
- return EBUS_SENDTO_SELF;
+LABLE_SUC:
+ if(recv_buf != NULL) {
+ void *_buf = malloc(recvpak.size);
+ memcpy(_buf, recvpak.buf, recvpak.size);
+ *recv_buf = _buf;
}
+
+ if(recv_size != NULL)
+ *recv_size = recvpak.size;
- LockFreeQueue<shm_packet_t> *remoteQueue;
- if ((remoteQueue = shm_socket_attach_queue(key)) == NULL) {
- bus_errno = EBUS_CLOSED;
- logger->error("sendto key %d failed, %s", key, bus_strerror(bus_errno));
- return EBUS_CLOSED;
- }
+ mm_free(recvpak.buf);
- shm_packet_t dest;
- dest.key = sockt->key;
- dest.size = size;
- dest.buf = mm_malloc(size);
- memcpy(dest.buf, buf, size);
+ return 0;
- rv = remoteQueue->push(dest, timeout, flag);
+}
- if (rv == 0) {
- printf("%d sendto %d suc.\n", shm_socket_get_key(sockt), key);
- return 0;
- } else {
- mm_free(dest.buf);
+/**
+ * @callback void (*recv_callback_fn)(void **sendbuf, int *sendsize)
+ * sendbuf 鍜� sendsize鏄痗allbak_fn鍥炶皟鍑芥暟鐨勮繑鍥炲��, 鍒嗗埆琛ㄧず鍙戦�佹暟鎹紝鍜屽彂閫佹暟鎹殑澶у皬銆�
+ *
+ */
+int shm_recvandsend(shm_socket_t *sockt, void **recvbuf, int *recvsize, int *key, recv_callback_fn callback,
+ const struct timespec *timeout, int flag) {
+
+ int rv;
+
+ void *sendbuf = NULL;
+ int sendsize = 0;
+ shm_packet_t recvpak;
+
+ rv = shm_recvpakfrom(sockt , &recvpak, timeout, flag);
+
+
+
+ if (rv != 0) {
if(rv == ETIMEDOUT)
return EBUS_TIMEOUT;
else {
- logger->debug("====%d sendto key %d failed %s", shm_socket_get_key(sockt), key, bus_strerror(rv));
+ logger->debug("%d shm_recvfrom failed %s", shm_socket_get_key(sockt), bus_strerror(rv));
return rv;
}
}
+
+ if(recvbuf != NULL) {
+ void *_buf = malloc(recvpak.size);
+ memcpy(_buf, recvpak.buf, recvpak.size);
+ *recvbuf = _buf;
+ }
+
+ if(recvsize != NULL)
+ *recvsize = recvpak.size;
+
+ if(key != NULL)
+ *key = recvpak.key;
+
+ mm_free(recvpak.buf);
+
+ callback(&sendbuf, &sendsize);
+
+ shm_packet_t sendpak;
+ sendpak.key = sockt->key;
+ sendpak.size = sendsize;
+ memcpy(sendpak.uuid, recvpak.uuid, sizeof sendpak.uuid);
+ if(sendbuf !=NULL && sendsize > 0) {
+ sendpak.buf = mm_malloc(sendsize);
+ memcpy(sendpak.buf, sendbuf, sendsize);
+ }
+
+ rv = shm_sendpakto(sockt, &sendpak, recvpak.key, timeout, flag);
+
+ return 0;
+
}
-
-
// 鐭繛鎺ユ柟寮忔帴鍙�
int shm_recvfrom(shm_socket_t *sockt, void **buf, int *size, int *key, const struct timespec *timeout, int flag) {
int rv;
- hashtable_t *hashtable = mm_get_hashtable();
+ shm_packet_t recvpak;
+ rv = shm_recvpakfrom(sockt , &recvpak, timeout, flag);
- if( sockt->queue != NULL)
- goto LABEL_POP;
+ if (rv != 0) {
- {
- if ((rv = pthread_mutex_lock(&(sockt->mutex))) != 0)
- err_exit(rv, "shm_recvfrom : pthread_mutex_lock");
-
-
- if (sockt->key == 0) {
- sockt->key = hashtable_alloc_key(hashtable);
- }
- sockt->queue = shm_socket_bind_queue( sockt->key, sockt->force_bind);
- if(sockt->queue == NULL ) {
- logger->error("%s. key = %d", bus_strerror(EBUS_KEY_INUSED), sockt->key);
- return EBUS_KEY_INUSED;
- }
-
-
- if ((rv = pthread_mutex_unlock(&(sockt->mutex))) != 0)
- err_exit(rv, "shm_recvfrom : pthread_mutex_unlock");
-
- }
-
-LABEL_POP:
-
- shm_packet_t src;
-
- rv = sockt->queue->pop(src, timeout, flag);
-
- if (rv == 0) {
- if(buf != NULL) {
- void *_buf = malloc(src.size);
- memcpy(_buf, src.buf, src.size);
- *buf = _buf;
- }
-
- if(size != NULL)
- *size = src.size;
-
- if(key != NULL)
- *key = src.key;
-
- mm_free(src.buf);
- return 0;
- } else {
if(rv == ETIMEDOUT)
return EBUS_TIMEOUT;
else {
@@ -267,10 +315,27 @@
return rv;
}
+ }
+
+ if(buf != NULL) {
+ void *_buf = malloc(recvpak.size);
+ memcpy(_buf, recvpak.buf, recvpak.size);
+ *buf = _buf;
}
+
+ if(size != NULL)
+ *size = recvpak.size;
+
+ if(key != NULL)
+ *key = recvpak.key;
+
+ mm_free(recvpak.buf);
+ return 0;
}
+
+// =================================================================================================
/* Free thread-specific data buffer */
static void _destrory_socket_perthread(void *tmp_socket)
@@ -406,10 +471,99 @@
}
-int shm_sendandrecv(shm_socket_t *socket, const void *send_buf,
- const int send_size, const int send_key, void **recv_buf,
- int *recv_size, const struct timespec *timeout, int flags) {
- struct timespec tm = {10, 0};
- return _shm_sendandrecv_thread_local(socket, send_buf, send_size, send_key,recv_buf, recv_size, &tm, flags);
+
+static int shm_sendpakto(shm_socket_t *sockt, const shm_packet_t *sendpak,
+ const int key, const struct timespec *timeout, const int flag) {
+
+ int rv;
+ hashtable_t *hashtable = mm_get_hashtable();
+
+ if( sockt->queue != NULL)
+ goto LABEL_PUSH;
+
+ {
+ if ((rv = pthread_mutex_lock(&(sockt->mutex))) != 0)
+ err_exit(rv, "shm_sendto : pthread_mutex_lock");
+
+ if (sockt->queue == NULL) {
+ if (sockt->key == 0) {
+ sockt->key = hashtable_alloc_key(hashtable);
+ }
+ sockt->queue = shm_socket_bind_queue( sockt->key, sockt->force_bind);
+ if(sockt->queue == NULL ) {
+ logger->error("%s. key = %d", bus_strerror(EBUS_KEY_INUSED), sockt->key);
+ return EBUS_KEY_INUSED;
+ }
+ }
+
+ if ((rv = pthread_mutex_unlock(&(sockt->mutex))) != 0)
+ err_exit(rv, "shm_sendto : pthread_mutex_unlock");
+
+ }
+
+
+ LABEL_PUSH:
+ if (key == sockt->key) {
+ logger->error( "can not send to your self!");
+ return EBUS_SENDTO_SELF;
+ }
+
+ LockFreeQueue<shm_packet_t> *remoteQueue;
+ if ((remoteQueue = shm_socket_attach_queue(key)) == NULL) {
+ bus_errno = EBUS_CLOSED;
+ logger->error("sendto key %d failed, %s", key, bus_strerror(bus_errno));
+ return EBUS_CLOSED;
+ }
+
+
+
+ rv = remoteQueue->push(*sendpak, timeout, flag);
+
+ return rv;
}
+
+// 鐭繛鎺ユ柟寮忔帴鍙�
+static int shm_recvpakfrom(shm_socket_t *sockt, shm_packet_t *recvpak , const struct timespec *timeout, int flag) {
+ int rv;
+
+ hashtable_t *hashtable = mm_get_hashtable();
+
+ if( sockt->queue != NULL)
+ goto LABEL_POP;
+
+ {
+ if ((rv = pthread_mutex_lock(&(sockt->mutex))) != 0)
+ err_exit(rv, "shm_recvfrom : pthread_mutex_lock");
+
+
+ if (sockt->key == 0) {
+ sockt->key = hashtable_alloc_key(hashtable);
+ }
+ sockt->queue = shm_socket_bind_queue( sockt->key, sockt->force_bind);
+ if(sockt->queue == NULL ) {
+ logger->error("%s. key = %d", bus_strerror(EBUS_KEY_INUSED), sockt->key);
+ return EBUS_KEY_INUSED;
+ }
+
+
+ if ((rv = pthread_mutex_unlock(&(sockt->mutex))) != 0)
+ err_exit(rv, "shm_recvfrom : pthread_mutex_unlock");
+
+ }
+
+LABEL_POP:
+
+
+
+ rv = sockt->queue->pop(*recvpak, timeout, flag);
+
+ return rv;
+}
+// int shm_sendandrecv(shm_socket_t *sockt, const void *send_buf,
+// const int send_size, const int send_key, void **recv_buf,
+// int *recv_size, const struct timespec *timeout, int flags) {
+
+// struct timespec tm = {10, 0};
+// return _shm_sendandrecv_thread_local(sockt, send_buf, send_size, send_key,recv_buf, recv_size, &tm, flags);
+// }
--
Gitblit v1.8.0