From 14c345b38d57fd814f217eb8465963a08ca79f7e Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期五, 05 二月 2021 17:41:09 +0800 Subject: [PATCH] update --- src/socket/shm_socket.cpp | 1045 +++++++++++++++++++++++++-------------------------------- 1 files changed, 461 insertions(+), 584 deletions(-) diff --git a/src/socket/shm_socket.cpp b/src/socket/shm_socket.cpp index 25c3930..14274ca 100644 --- a/src/socket/shm_socket.cpp +++ b/src/socket/shm_socket.cpp @@ -2,60 +2,128 @@ #include "hashtable.h" #include "logger_factory.h" #include <map> +#include <cassert> #include "bus_error.h" +#include "sole.h" static Logger *logger = LoggerFactory::getLogger(); -static void print_msg(char *head, shm_msg_t &msg) { +static void print_msg(char *head, shm_packet_t &msg) { // err_msg(0, "%s: key=%d, type=%d\n", head, msg.key, msg.type); } static pthread_once_t _once_ = PTHREAD_ONCE_INIT; -static pthread_key_t _tmp_recv_socket_key_; +static pthread_key_t _perthread_socket_key_; -static void *_server_run_msg_rev(void *_socket); +static void _destrory_socket_perthread(void *tmp_socket); +static void _create_socket_key_perthread(void); -static void *_client_run_msg_rev(void *_socket); +static int shm_recvpakfrom(shm_socket_t *sockt, shm_packet_t *recvpak , const struct timespec *timeout, int flag); -static int _shm_close_dgram_socket(shm_socket_t *socket); + +static int shm_sendpakto(shm_socket_t *sockt, const shm_packet_t *sendpak, + const int key, const struct timespec *timeout, const int flag); -static int _shm_close_stream_socket(shm_socket_t *socket, bool notifyRemote); +// 妫�鏌ey鏄惁宸茬粡琚娇鐢紝 鏈浣跨敤鍒欑粦瀹歬ey +static LockFreeQueue<shm_packet_t> * shm_socket_bind_queue(int key, bool force) { + hashtable_t *hashtable = mm_get_hashtable(); + LockFreeQueue<shm_packet_t> *queue; + hashtable_lock(hashtable); + void *tmp_ptr = hashtable_get(hashtable, key); -static void _destrory_tmp_recv_socket_(void *tmp_socket); -static void _create_tmp_recv_socket_key(void); -// 妫�鏌ey鏄惁宸茬粡琚娇鐢紝鏄繑鍥�0, 鍚﹁繑鍥�1 -static inline int _shm_socket_check_key(shm_socket_t *socket) { - void *tmp_ptr = mm_get_by_key(socket->key); - if (tmp_ptr!= NULL && tmp_ptr != (void *)1 && !socket->force_bind ) { - bus_errno = EBUS_KEY_INUSED; - logger->error("%s. key = %d ", bus_strerror(EBUS_KEY_INUSED), socket->key); - return 0; - } - return 1; + if (tmp_ptr == NULL || tmp_ptr == (void *)1 ) { + queue = new LockFreeQueue<shm_packet_t>(16); + hashtable_put(hashtable, key, (void *)queue); + hashtable_unlock(hashtable); + return queue; + } else if(force) { + hashtable_unlock(hashtable); + return (LockFreeQueue<shm_packet_t> *) queue; + } + + hashtable_unlock(hashtable); + return NULL; } -SHMQueue<shm_msg_t> *_attach_remote_queue(int key); +/** + * 缁戝畾key鍒伴槦鍒楋紝浣嗘槸骞朵笉浼氬垱寤洪槦鍒椼�� + */ +static LockFreeQueue<shm_packet_t> * shm_socket_attach_queue(int key) { + LockFreeQueue<shm_packet_t> * queue; + hashtable_t *hashtable = mm_get_hashtable(); + void *tmp_ptr = hashtable_get(hashtable, key); + if (tmp_ptr == NULL || tmp_ptr == (void *)1) { + //logger->error("shm_socket._remote_queue_attach锛歝onnet at key %d failed!", key); + return NULL; + } + queue = ( LockFreeQueue<shm_packet_t> *)tmp_ptr; + // hashtable_unlock(hashtable); + return queue; +} - +//鍒犻櫎鍖呭惈鍦╧eys鍐呯殑queue size_t shm_socket_remove_keys(int keys[], size_t length) { - return SHMQueue<shm_msg_t>::remove_queues(keys, length); + hashtable_t *hashtable = mm_get_hashtable(); + LockFreeQueue<shm_packet_t> *mqueue; + size_t count = 0; + for(int i = 0; i< length; i++) { + // 閿�姣佸叡浜唴瀛樼殑queue + mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, keys[i]); + delete mqueue; + hashtable_remove(hashtable, keys[i]); + count++; + } + return count; } + + +// 鍒犻櫎涓嶅湪keys鍐呯殑queue +size_t shm_socket_remove_keys_exclude(int keys[], size_t length) { + hashtable_t *hashtable = mm_get_hashtable(); + std::set<int> *keyset = hashtable_keyset(hashtable); + std::set<int>::iterator keyItr; + LockFreeQueue<shm_packet_t> *mqueue; + bool found; + size_t count = 0; + for (keyItr = keyset->begin(); keyItr != keyset->end(); keyItr++) { + found = false; + for (size_t i = 0; i < length; i++) { + if (*keyItr == keys[i]) { + found = true; + break; + } + } + // 100鍐呯殑鏄痓us鍐呴儴鑷繁鐢ㄧ殑 + if (!found && *keyItr > 100) { + // 閿�姣佸叡浜唴瀛樼殑queue + mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, *keyItr); + delete mqueue; + hashtable_remove(hashtable, *keyItr); + count++; + } + } + delete keyset; + return count; +} + + shm_socket_t *shm_open_socket(shm_socket_type_t socket_type) { int s, type; 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->dispatch_thread = 0; - socket->status = SHM_CONN_CLOSED; + // 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); if (s != 0) @@ -63,7 +131,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"); @@ -71,387 +139,250 @@ 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 ret, s; - + int rv; logger->debug("shm_close_socket\n"); - switch (socket->socket_type) { - case SHM_SOCKET_STREAM: - ret = _shm_close_stream_socket(socket, true); - break; - case SHM_SOCKET_DGRAM: - ret = _shm_close_dgram_socket(socket); - break; - default: - break; + if(sockt->queue != NULL) { + delete sockt->queue; + sockt->queue = NULL; } - s = pthread_mutex_destroy(&(socket->mutex) ); - if(s != 0) { - err_exit(s, "shm_close_socket"); + rv = pthread_mutex_destroy(&(sockt->mutex) ); + if(rv != 0) { + err_exit(rv, "shm_close_socket"); } - free(socket); - return ret; -} - -// int shm_close_socket(shm_socket_t *socket) { - -// // _destrory_tmp_recv_socket_((shm_socket_t *)pthread_getspecific(_tmp_recv_socket_key_)); - -// return shm_close_socket(socket);; -// } - -int shm_socket_bind(shm_socket_t *socket, int key) { - socket->key = key; + free(sockt); return 0; } -int shm_socket_force_bind(shm_socket_t *socket, int key) { - socket->force_bind = true; - socket->key = key; + +int shm_socket_stop(shm_socket_t *sockt) { + struct timespec timeout = {5, 0}; + shm_packet_t sendpak = {0}; + sendpak.key = sockt->key; + sendpak.action = BUS_ACTION_STOP; + sendpak.size = 0; + return shm_sendpakto(sockt, &sendpak, sockt->key, &timeout, BUS_TIMEOUT_FLAG); +} + +int shm_socket_bind(shm_socket_t *sockt, int key) { + sockt->key = key; return 0; } -int shm_listen(shm_socket_t *socket) { - - if (socket->socket_type != SHM_SOCKET_STREAM) { - logger->error("can not invoke shm_listen method with a socket which is not a " - "SHM_SOCKET_STREAM socket"); - exit(1); - } - - int key; - hashtable_t *hashtable = mm_get_hashtable(); - if (socket->key == 0) { - key = hashtable_alloc_key(hashtable); - socket->key = key; - } else { - - if(!_shm_socket_check_key(socket)) { - bus_errno = EBUS_KEY_INUSED; - return EBUS_KEY_INUSED; - } - } - - socket->queue = new SHMQueue<shm_msg_t>(socket->key, 16); - socket->acceptQueue = new LockFreeQueue<shm_msg_t, DM_Allocator>(16); - socket->clientSocketMap = new std::map<int, shm_socket_t *>; - socket->status = SHM_CONN_LISTEN; - pthread_create(&(socket->dispatch_thread), NULL, _server_run_msg_rev, - (void *)socket); - +int shm_socket_force_bind(shm_socket_t *sockt, int key) { + sockt->force_bind = true; + sockt->key = key; return 0; } -/** - * 鎺ュ彈瀹㈡埛绔缓绔嬫柊杩炴帴鐨勮姹� - * -*/ -shm_socket_t *shm_accept(shm_socket_t *socket) { - if (socket->socket_type != SHM_SOCKET_STREAM) { - logger->error("can not invoke shm_accept method with a socket which is not a " - "SHM_SOCKET_STREAM socket"); - exit(1); - } - hashtable_t *hashtable = mm_get_hashtable(); - int client_key; - shm_socket_t *client_socket; - shm_msg_t src; - - if (socket->acceptQueue->pop(src) == 0) { - - // print_msg("===accept:", src); - client_key = src.key; - // client_socket = (shm_socket_t *)malloc(sizeof(shm_socket_t)); - client_socket = shm_open_socket(socket->socket_type); - client_socket->key = socket->key; - // client_socket->queue= socket->queue; - //鍒濆鍖栨秷鎭痲ueue - client_socket->messageQueue = - new LockFreeQueue<shm_msg_t, DM_Allocator>(16); - //杩炴帴鍒板鏂筿ueue - client_socket->remoteQueue = _attach_remote_queue(client_key); - - socket->clientSocketMap->insert({client_key, client_socket}); - - /* -* shm_accept 鐢ㄦ埛鎵ц鐨勬柟娉� -* 涓巁server_run_msg_rev鍦ㄤ袱涓笉鍚岀殑闄愬埗宸ヤ綔,accept瑕佷繚璇佸湪瀹㈡埛鐨勫彂閫佹秷鎭箣鍓嶅畬鎴愯祫婧愮殑鍑嗗宸ヤ綔锛屼互閬垮厤鍑虹幇绔炴�侀棶棰� - */ - //鍙戦�乷pen_reply,鍥炲簲瀹㈡埛绔殑connect璇锋眰 - struct timespec timeout = {1, 0}; - shm_msg_t msg; - msg.key = socket->key; - msg.size = 0; - msg.type = SHM_SOCKET_OPEN_REPLY; - - if (client_socket->remoteQueue->push(msg, &timeout, BUS_TIMEOUT_FLAG) == 0) { - client_socket->status = SHM_CONN_ESTABLISHED; - return client_socket; - } else { - logger->error( "shm_accept: 鍙戦�乷pen_reply澶辫触"); - return NULL; - } - - } else { - err_exit(errno, "shm_accept"); - } - return NULL; +int shm_socket_get_key(shm_socket_t *sockt){ + return sockt->key; } - -/** - * @return 0鎴愬姛. 鍏朵粬鍊煎け璐� - */ -int shm_connect(shm_socket_t *socket, int key) { - if (socket->socket_type != SHM_SOCKET_STREAM) { - logger->error( "can not invoke shm_connect method with a socket which is not " - "a SHM_SOCKET_STREAM socket"); - exit(1); - } - hashtable_t *hashtable = mm_get_hashtable(); - if (hashtable_get(hashtable, key) == NULL) { - logger->error("shm_connect锛歝onnect at key %d failed!", key); - return -1; - } - - if (socket->key == 0) { - socket->key = hashtable_alloc_key(hashtable); - } else { - if(!_shm_socket_check_key(socket)) { - bus_errno = EBUS_KEY_INUSED; - return EBUS_KEY_INUSED; - } - } - - socket->queue = new SHMQueue<shm_msg_t>(socket->key, 16); - - if ((socket->remoteQueue = _attach_remote_queue(key)) == NULL) { - logger->error("connect to %d failted", key); - return -1; - } - socket->messageQueue = new LockFreeQueue<shm_msg_t, DM_Allocator>(16); - - //鍙戦�乷pen璇锋眰 - struct timespec timeout = {1, 0}; - shm_msg_t msg; - msg.key = socket->key; - msg.size = 0; - msg.type = SHM_SOCKET_OPEN; - socket->remoteQueue->push(msg, &timeout, BUS_TIMEOUT_FLAG); - - //鎺ュ彈open reply - if (socket->queue->pop(msg) == 0) { - // 鍦ㄨ繖閲宻erver绔凡缁忓噯澶囧ソ鎺ュ彈瀹㈡埛绔彂閫佽姹備簡,瀹屾垚涓庢湇鍔$鐨勮繛鎺� - if (msg.type == SHM_SOCKET_OPEN_REPLY) { - socket->status = SHM_CONN_ESTABLISHED; - pthread_create(&(socket->dispatch_thread), NULL, _client_run_msg_rev, - (void *)socket); - } else { - logger->error( "shm_connect: 涓嶅尮閰嶇殑搴旂瓟淇℃伅!"); - exit(1); - } - - } else { - logger->error( "connect failted!"); - return -1; - } - - return 0; -} - -int shm_send(shm_socket_t *socket, const void *buf, const int size) { - if (socket->socket_type != SHM_SOCKET_STREAM) { - logger->error("shm_socket.shm_send: can not invoke shm_send method with a socket which is not a " - "SHM_SOCKET_STREAM socket"); - exit(1); - } - hashtable_t *hashtable = mm_get_hashtable(); - if(socket->remoteQueue == NULL) { - err_msg(errno, "褰撳墠瀹㈡埛绔棤杩炴帴!"); - return -1; - } - shm_msg_t dest; - dest.type = SHM_COMMON_MSG; - dest.key = socket->key; - dest.size = size; - dest.buf = mm_malloc(size); - memcpy(dest.buf, buf, size); - - if (socket->remoteQueue->push(dest) == 0) { - return 0; - } else { - logger->error(errno, "connection has been closed!"); - return -1; - } -} - -int shm_recv(shm_socket_t *socket, void **buf, int *size) { - if (socket->socket_type != SHM_SOCKET_STREAM) { - logger->error( "shm_socket.shm_recv: can not invoke shm_recv method in a %d type socket which is " - "not a SHM_SOCKET_STREAM socket ", - socket->socket_type); - exit(1); - } - shm_msg_t src; - - if (socket->messageQueue->pop(src) == 0) { - void *_buf = malloc(src.size); - memcpy(_buf, src.buf, src.size); - *buf = _buf; - *size = src.size; - mm_free(src.buf); - return 0; - } else { - return -1; - } -} // 鐭繛鎺ユ柟寮忓彂閫� -int shm_sendto(shm_socket_t *socket, const void *buf, const int size, +int shm_sendto(shm_socket_t *sockt, const void *buf, const int size, const int key, const struct timespec *timeout, const int flag) { - int s; int rv; - - if (socket->socket_type != SHM_SOCKET_DGRAM) { - logger->error( "shm_socket.shm_sendto: Can't invoke shm_sendto method in a %d type socket which is " - "not a SHM_SOCKET_DGRAM socket ", - socket->socket_type); - exit(0); - } - hashtable_t *hashtable = mm_get_hashtable(); - - if ((s = pthread_mutex_lock(&(socket->mutex))) != 0) - err_exit(s, "shm_sendto : pthread_mutex_lock"); - - if (socket->queue == NULL) { - if (socket->key == 0) { - socket->key = hashtable_alloc_key(hashtable); - } else { - - if(!_shm_socket_check_key(socket)) { - bus_errno = EBUS_KEY_INUSED; - return EBUS_KEY_INUSED; - } - - } - - socket->queue = new SHMQueue<shm_msg_t>(socket->key, 16); + shm_packet_t sendpak = {0}; + sendpak.key = sockt->key; + sendpak.size = size; + if(buf != NULL) { + sendpak.buf = mm_malloc(size); + memcpy(sendpak.buf, buf, size); } - - if ((s = pthread_mutex_unlock(&(socket->mutex))) != 0) - err_exit(s, "shm_sendto : pthread_mutex_unlock"); - - // There is some case where a socket need to send to himeself, for example when bus server need to stop, he need to send himself - // a top message. - - // if (key == socket->key) { - // logger->error( "can not send to your self!"); - // return -1; - // } - - SHMQueue<shm_msg_t> *remoteQueue; - if ((remoteQueue = _attach_remote_queue(key)) == NULL) { - bus_errno = EBUS_CLOSED; - logger->error("sendto key %d failed, %s", key, bus_strerror(bus_errno)); - return EBUS_CLOSED; - } - - shm_msg_t dest; - dest.type = SHM_COMMON_MSG; - dest.key = socket->key; - dest.size = size; - dest.buf = mm_malloc(size); - memcpy(dest.buf, buf, size); - - rv = remoteQueue->push(dest, timeout, flag); - - if (rv == 0) { - // printf("shm_sendto push after\n"); - return 0; - } else { - mm_free(dest.buf); - logger->debug("sendto key %d failed %s", key, bus_strerror(rv)); - return rv; - } + + rv = shm_sendpakto(sockt, &sendpak, key, timeout, flag); + return rv; } -int shm_recvfrom2(shm_socket_t *socket, void **buf, int *size, int *key, const struct timespec *timeout, int flag) { - return 501; + + +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; + if(send_buf != NULL) { + 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(tryn > 0) { + 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); + goto LABLE_SUC; + } + + 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(strlen(recvpak.uuid) == 0) { + continue; + } else if (strncmp(uuid.c_str(), recvpak.uuid, sizeof recvpak.uuid) == 0) { + // 鍙戦�佷笌鎺ュ彈鐨刄UID鍖归厤鎴愬姛 + goto LABLE_SUC; + } else { + // 绛旈潪鎵�闂紝鏀惧埌缂撳瓨閲� + sockt->recvbuf.insert({recvpak.uuid, recvpak}); + continue; + } + } + +LABLE_FAIL: + return EBUS_RECVFROM_WRONG_END; + // return rv; + +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; + + mm_free(recvpak.buf); + + return 0; + + + +} + +/** + * @callback void (*recvandsend_callback_fn)(void *recvbuf, int recvsize, int key, void **sendbuf, int *sendsize) + * sendbuf 鍜� sendsize鏄痗allbak_fn鍥炶皟鍑芥暟鐨勮繑鍥炲��, 鍒嗗埆琛ㄧず鍙戦�佹暟鎹紝鍜屽彂閫佹暟鎹殑澶у皬銆� + * + */ +int shm_recvandsend(shm_socket_t *sockt, recvandsend_callback_fn callback, + const struct timespec *timeout, int flag, void *user_data) { + + int rv; + + void *sendbuf, *recvbuf = NULL; + int sendsize = 0; + shm_packet_t recvpak = {0}; + + rv = shm_recvpakfrom(sockt , &recvpak, timeout, flag); + + if (rv != 0) { + if(rv == ETIMEDOUT){ + logger->debug("%d shm_recvandsend failed %s", shm_socket_get_key(sockt), bus_strerror(EBUS_TIMEOUT)); + return EBUS_TIMEOUT; + } + + logger->error("%d shm_recvandsend failed %s", shm_socket_get_key(sockt), bus_strerror(rv)); + return rv; + } + + if(recvpak.buf != NULL) { + recvbuf = malloc(recvpak.size); + memcpy(recvbuf, recvpak.buf, recvpak.size); + mm_free(recvpak.buf); + } + + callback(recvbuf, recvpak.size, recvpak.key, &sendbuf, &sendsize, user_data); + + shm_packet_t sendpak = {0}; + 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); + } else { + + logger->warn("%d shm_recvandsend : sendbuf is null", shm_socket_get_key(sockt)); + // return -1; + } + + rv = shm_sendpakto(sockt, &sendpak, recvpak.key, timeout, flag); + + return 0; + } // 鐭繛鎺ユ柟寮忔帴鍙� -int shm_recvfrom(shm_socket_t *sokt, void **buf, int *size, int *key, const struct timespec *timeout, int flag) { - int s; +int shm_recvfrom(shm_socket_t *sockt, void **buf, int *size, int *key, const struct timespec *timeout, int flag) { int rv; - - if (sokt->socket_type != SHM_SOCKET_DGRAM) { - logger->error("shm_socket.shm_recvfrom: Can't invoke shm_recvfrom method in a %d type socket which " - "is not a SHM_SOCKET_DGRAM socket ", - sokt->socket_type); - exit(1); - } - hashtable_t *hashtable = mm_get_hashtable(); - - if ((s = pthread_mutex_lock(&(sokt->mutex))) != 0) - err_exit(s, "shm_recvfrom : pthread_mutex_lock"); - - if (sokt->queue == NULL) { - if (sokt->key == 0) { - sokt->key = hashtable_alloc_key(hashtable); - } else { - - if(!_shm_socket_check_key(sokt)) { - bus_errno = EBUS_KEY_INUSED; - return EBUS_KEY_INUSED; - } - } - - sokt->queue = new SHMQueue<shm_msg_t>(sokt->key, 16); - } - if ((s = pthread_mutex_unlock(&(sokt->mutex))) != 0) - err_exit(s, "shm_recvfrom : pthread_mutex_unlock"); + shm_packet_t recvpak; + rv = shm_recvpakfrom(sockt , &recvpak, timeout, flag); - shm_msg_t src; - -printf ("====== before ======\n"); - rv = sokt->queue->pop(src, timeout, flag); -printf ("====== after ======\n %d", rv); + if (rv != 0) { - if (rv == 0) { - if(buf != NULL) { - void *_buf = malloc(src.size); - memcpy(_buf, src.buf, src.size); - *buf = _buf; + if(rv == ETIMEDOUT) + return EBUS_TIMEOUT; + else { + logger->debug("%d shm_recvfrom failed %s", shm_socket_get_key(sockt), bus_strerror(rv)); + return rv; } - if(size != NULL) - *size = src.size; + } - if(key != NULL) - *key = src.key; - mm_free(src.buf); - return 0; - } else { - logger->debug("shm_recvfrom failed %s", bus_strerror(rv)); - return rv; - + if(buf != NULL && recvpak.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_tmp_recv_socket_(void *tmp_socket) +static void _destrory_socket_perthread(void *tmp_socket) { int rv; if(tmp_socket == NULL) @@ -459,7 +390,7 @@ logger->debug("%d destroy tmp socket\n", pthread_self()); shm_close_socket((shm_socket_t *)tmp_socket); - rv = pthread_setspecific(_tmp_recv_socket_key_, NULL); + rv = pthread_setspecific(_perthread_socket_key_, NULL); if ( rv != 0) { logger->error(rv, "shm_sendandrecv : pthread_setspecific"); exit(1); @@ -467,54 +398,48 @@ } /* One-time key creation function */ -static void _create_tmp_recv_socket_key(void) +static void _create_socket_key_perthread(void) { int s; /* Allocate a unique thread-specific data key and save the address of the destructor for thread-specific data buffers */ - s = pthread_key_create(&_tmp_recv_socket_key_, _destrory_tmp_recv_socket_); - //s = pthread_key_create(&_tmp_recv_socket_key_, NULL); + s = pthread_key_create(&_perthread_socket_key_, _destrory_socket_perthread); + //s = pthread_key_create(&_perthread_socket_key_, NULL); if (s != 0) { logger->error(s, "pthread_key_create"); - abort(); /* dump core and terminate */ exit(1); } } // use thread local -int _shm_sendandrecv_thread_local(shm_socket_t *socket, const void *send_buf, +int _shm_sendandrecv_thread_local(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) { int recv_key; int rv; + int tryn = 0; // 鐢╰hread local 淇濊瘉姣忎釜绾跨▼鐢ㄤ竴涓嫭鍗犵殑socket鎺ュ彈瀵规柟杩斿洖鐨勪俊鎭� shm_socket_t *tmp_socket; - - if (socket->socket_type != SHM_SOCKET_DGRAM) { - logger->error( "shm_socket.shm_sendandrecv: Can't invoke shm_sendandrecv method in a %d type socket " - "which is not a SHM_SOCKET_DGRAM socket ", - socket->socket_type); - exit(1); - } - - rv = pthread_once(&_once_, _create_tmp_recv_socket_key); + /* If first call from this thread, allocate buffer for thread, and save its location */ + // logger->debug("%d create tmp socket\n", pthread_self() ); + rv = pthread_once(&_once_, _create_socket_key_perthread); if (rv != 0) { logger->error(rv, "shm_sendandrecv pthread_once"); exit(1); } - tmp_socket = (shm_socket_t *)pthread_getspecific(_tmp_recv_socket_key_); + tmp_socket = (shm_socket_t *)pthread_getspecific(_perthread_socket_key_); if (tmp_socket == NULL) { /* If first call from this thread, allocate buffer for thread, and save its location */ logger->debug("%ld create tmp socket\n", (long)pthread_self() ); tmp_socket = shm_open_socket(SHM_SOCKET_DGRAM); - rv = pthread_setspecific(_tmp_recv_socket_key_, tmp_socket); + rv = pthread_setspecific(_perthread_socket_key_, tmp_socket); if ( rv != 0) { logger->error(rv, "shm_sendandrecv : pthread_setspecific"); exit(1); @@ -522,232 +447,184 @@ } if ((rv = shm_sendto(tmp_socket, send_buf, send_size, send_key, timeout, flags)) == 0) { - rv = shm_recvfrom(tmp_socket, recv_buf, recv_size, &recv_key, timeout, flags); - return rv; - } else { - return rv; - } + + while(tryn < 3) { + tryn++; + rv = shm_recvfrom(tmp_socket, recv_buf, recv_size, &recv_key, timeout, flags); + if(rv != 0) { + logger->error("_shm_sendandrecv_thread_local : %s\n", bus_strerror(rv)); + return rv; + } + + // 瓒呮椂瀵艰嚧鎺ュ彂閫佸璞★紝涓庤繑鍥炲璞′笉瀵瑰簲鐨勬儏鍐� + if(send_key != recv_key) { + logger->debug("======%d use tmp_socket %d, send to %d, receive from %d\n", shm_socket_get_key(sockt), shm_socket_get_key(tmp_socket), send_key, recv_key); + // logger->error( "_shm_sendandrecv_alloc_new: send key expect to equal to recv key! send key =%d , recv key=%d", send_key, recv_key); + // exit(1); + continue; + // return EBUS_RECVFROM_WRONG_END; + } + + return 0; + } + + return EBUS_RECVFROM_WRONG_END; + } + + return rv; } -int _shm_sendandrecv_alloc_new(shm_socket_t *socket, const void *send_buf, +int _shm_sendandrecv_alloc_new(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) { int recv_key; int rv; - // 鐢╰hread local 淇濊瘉姣忎釜绾跨▼鐢ㄤ竴涓嫭鍗犵殑socket鎺ュ彈瀵规柟杩斿洖鐨勪俊鎭� + int tryn = 0; shm_socket_t *tmp_socket; - - if (socket->socket_type != SHM_SOCKET_DGRAM) { - logger->error( "shm_socket.shm_sendandrecv: Can't invoke shm_sendandrecv method in a %d type socket " - "which is not a SHM_SOCKET_DGRAM socket ", - socket->socket_type); - exit(1); - } - /* If first call from this thread, allocate buffer for thread, and save its location */ - // logger->debug("%d create tmp socket\n", pthread_self() ); + tmp_socket = shm_open_socket(SHM_SOCKET_DGRAM); if ((rv = shm_sendto(tmp_socket, send_buf, send_size, send_key, timeout, flags)) == 0) { rv = shm_recvfrom(tmp_socket, recv_buf, recv_size, &recv_key, timeout, flags); + + while(tryn < 3) { + tryn++; + rv = shm_recvfrom(tmp_socket, recv_buf, recv_size, &recv_key, timeout, flags); + if(rv != 0) { + logger->error("_shm_sendandrecv_thread_local : %s\n", bus_strerror(rv)); + return rv; + } + + // 瓒呮椂瀵艰嚧鎺ュ彂閫佸璞★紝涓庤繑鍥炲璞′笉瀵瑰簲鐨勬儏鍐� + if(send_key != recv_key) { + // logger->debug("======%d use tmp_socket %d, send to %d, receive from %d\n", shm_socket_get_key(sockt), shm_socket_get_key(tmp_socket), send_key, recv_key); + // logger->error( "_shm_sendandrecv_alloc_new: send key expect to equal to recv key! send key =%d , recv key=%d", send_key, recv_key); + + continue; + } + return 0; + } + + return EBUS_RECVFROM_WRONG_END; } shm_close_socket(tmp_socket); return rv; } + -int shm_sendandrecv_unsafe(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) { - if (socket->socket_type != SHM_SOCKET_DGRAM) { - logger->error( "shm_socket.shm_sendandrecv_unsafe : Can't invoke shm_sendandrecv method in a %d type socket " - "which is not a SHM_SOCKET_DGRAM socket ", - socket->socket_type); - exit(1); - } - int recv_key; + +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; - - - if ((rv = shm_sendto(socket, send_buf, send_size, send_key, timeout, flags)) == 0) { - rv = shm_recvfrom(socket, recv_buf, recv_size, &recv_key, timeout, flags); - return rv; - } else { - return rv; - } - return -1; -} - -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) { - return _shm_sendandrecv_thread_local(socket, send_buf, send_size, send_key,recv_buf, recv_size, timeout, flags); -} - - -// ============================================================================================================ - -/** - * 缁戝畾key鍒伴槦鍒楋紝浣嗘槸骞朵笉浼氬垱寤洪槦鍒椼�傚鏋滄病鏈夊搴旀寚瀹歬ey鐨勯槦鍒楁彁绀洪敊璇苟閫�鍑� - */ -SHMQueue<shm_msg_t> *_attach_remote_queue(int key) { - hashtable_t *hashtable = mm_get_hashtable(); - if (hashtable_get(hashtable, key) == NULL) { - //logger->error("shm_socket._remote_queue_attach锛歝onnet at key %d failed!", key); - return NULL; + + if( sockt->queue != NULL) + goto LABEL_PUSH; + + if(hashtable_get_queue_count(hashtable) > QUEUE_COUNT_LIMIT) { + return EBUS_EXCEED_LIMIT; } - - SHMQueue<shm_msg_t> *queue = new SHMQueue<shm_msg_t>(key, 0); - return queue; -} - -void _server_close_conn_to_client(shm_socket_t *socket, int key) { - shm_socket_t *client_socket; - std::map<int, shm_socket_t *>::iterator iter = - socket->clientSocketMap->find(key); - if (iter != socket->clientSocketMap->end()) { - client_socket = iter->second; - free((void *)client_socket); - socket->clientSocketMap->erase(iter); - } -} - -/** - * server绔悇绉嶇被鍨嬫秷鎭紙锛夊湪杩欓噷杩涚▼鍒嗘嫞 - */ -void *_server_run_msg_rev(void *_socket) { - pthread_detach(pthread_self()); - shm_socket_t *socket = (shm_socket_t *)_socket; - struct timespec timeout = {1, 0}; - shm_msg_t src; - shm_socket_t *client_socket; - std::map<int, shm_socket_t *>::iterator iter; - - while (socket->queue->pop(src) == 0) { - - switch (src.type) { - case SHM_SOCKET_OPEN: - socket->acceptQueue->push(src, &timeout, BUS_TIMEOUT_FLAG); - break; - case SHM_SOCKET_CLOSE: - _server_close_conn_to_client(socket, src.key); - break; - case SHM_COMMON_MSG: - - iter = socket->clientSocketMap->find(src.key); - if (iter != socket->clientSocketMap->end()) { - client_socket = iter->second; - // print_msg("_server_run_msg_rev push before", src); - client_socket->messageQueue->push(src, &timeout, BUS_TIMEOUT_FLAG); - // print_msg("_server_run_msg_rev push after", src); - } - - break; - - default: - logger->error("shm_socket._server_run_msg_rev: undefined message type."); - } - } - - return NULL; -} - -void _client_close_conn_to_server(shm_socket_t *socket) { - - _shm_close_stream_socket(socket, false); -} - -/** - * client绔殑鍚勭绫诲瀷娑堟伅锛堬級鍦ㄨ繖閲岃繘绋嬪垎鎷� - */ -void *_client_run_msg_rev(void *_socket) { - pthread_detach(pthread_self()); - shm_socket_t *socket = (shm_socket_t *)_socket; - struct timespec timeout = {1, 0}; - shm_msg_t src; - - while (socket->queue->pop(src) == 0) { - switch (src.type) { - - case SHM_SOCKET_CLOSE: - _client_close_conn_to_server(socket); - break; - case SHM_COMMON_MSG: - socket->messageQueue->push(src, &timeout, BUS_TIMEOUT_FLAG); - break; - default: - logger->error( "shm_socket._client_run_msg_rev: undefined message type."); - } - } - - return NULL; -} - -int _shm_close_stream_socket(shm_socket_t *socket, bool notifyRemote) { - socket->status = SHM_CONN_CLOSED; - //缁欏鏂瑰彂閫佷竴涓叧闂繛鎺ョ殑娑堟伅 - struct timespec timeout = {1, 0}; - shm_msg_t close_msg; - - close_msg.key = socket->key; - close_msg.size = 0; - close_msg.type = SHM_SOCKET_CLOSE; - if (notifyRemote && socket->remoteQueue != NULL) { - socket->remoteQueue->push(close_msg, &timeout, BUS_TIMEOUT_FLAG); - } - - if (socket->queue != NULL) { - delete socket->queue; - socket->queue = NULL; - } - - - - if (socket->messageQueue != NULL) { - delete socket->messageQueue; - socket->messageQueue = NULL; - } - - if (socket->acceptQueue != NULL) { - delete socket->acceptQueue; - socket->acceptQueue = NULL; - } - - if (socket->clientSocketMap != NULL) { - shm_socket_t *client_socket; - for (auto iter = socket->clientSocketMap->begin(); - iter != socket->clientSocketMap->end(); iter++) { - client_socket = iter->second; - - client_socket->remoteQueue->push(close_msg, &timeout, BUS_TIMEOUT_FLAG); - client_socket->remoteQueue = NULL; - - delete client_socket->messageQueue; - client_socket->messageQueue = NULL; - free((void *)client_socket); - } - delete socket->clientSocketMap; - } - - if (socket->dispatch_thread != 0) - pthread_cancel(socket->dispatch_thread); - - - return 0; - -} - -int _shm_close_dgram_socket(shm_socket_t *socket){ - if(socket->queue != NULL) { - delete socket->queue; - socket->queue = NULL; - } - - return 0; -} - + { + 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 (sendpak->action != BUS_ACTION_STOP && 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(); + shm_packet_t recvpak; + + if( sockt->queue != NULL) + goto LABEL_POP; + + if(hashtable_get_queue_count(hashtable) > QUEUE_COUNT_LIMIT) { + return EBUS_EXCEED_LIMIT; + } + + { + 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: + + // + // printf("%p start recv.....\n", sockt); + + rv = sockt->queue->pop(recvpak, timeout, flag); + if(rv != 0) + return rv; + + + if(recvpak.action == BUS_ACTION_STOP) { + return EBUS_STOPED; + } + *_recvpak = recvpak; + 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