From 95349b79a5a646736c706fe19645181146ee9486 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期二, 20 十月 2020 16:29:59 +0800 Subject: [PATCH] update --- src/socket/net_mod_socket.h | 27 ++++++++++++--------------- 1 files changed, 12 insertions(+), 15 deletions(-) diff --git a/src/socket/net_mod_socket.h b/src/socket/net_mod_socket.h index c133302..4e4394c 100644 --- a/src/socket/net_mod_socket.h +++ b/src/socket/net_mod_socket.h @@ -3,7 +3,9 @@ #include "usg_common.h" #include "shm_mod_socket.h" #include "socket_io.h" +#include <poll.h> +#define OPEN_MAX 1024 #define GET(p) (*(uint32_t *)(p)) #define PUT(p, val) (*(uint32_t *)(p) = (val)) @@ -20,20 +22,24 @@ int key; }; -#define NET_MODE_REQUEST_HEAD_LENGTH 16 +#define NET_MODE_REQUEST_HEAD_LENGTH (NI_MAXHOST + 5 * sizeof(uint32_t)) struct net_mod_request_head_t { uint32_t mod; + char host[NI_MAXHOST]; + uint32_t port; uint32_t key; uint32_t content_length; uint32_t topic_length; }; -#define NET_MODE_RESPONSE_HEAD_LENGTH 8 +#define NET_MODE_RESPONSE_HEAD_LENGTH (NI_MAXHOST + 4 * sizeof(uint32_t)) struct net_mod_response_head_t { // socket_mod_t mod; - // int key; + char host[NI_MAXHOST]; + uint32_t port; + uint32_t key; uint32_t code; uint32_t content_length; }; @@ -41,7 +47,7 @@ struct net_mod_recv_msg_t { - char host[128]; + char host[NI_MAXHOST]; int port; int key; void *content; @@ -51,22 +57,13 @@ class NetModSocket { struct pool{ /* Represents a pool of connected descriptors */ //line:conc:echoservers:beginpool - int maxfd; /* Largest descriptor in read_set */ - fd_set read_set; /* Set of all active descriptors */ - fd_set ready_read_set; /* Subset of descriptors ready for reading */ - - fd_set write_set; - fd_set ready_write_set; - - fd_set except_set; - fd_set ready_except_set; int nready; /* Number of ready descriptors from select */ int maxi; /* Highwater index into client array */ - int connfd[FD_SETSIZE]; /* Set of active descriptors */ + struct pollfd conns[OPEN_MAX]; // net_node_t *nodes[FD_SETSIZE]; // rio_t clientrio[FD_SETSIZE]; /* Set of active read buffers */ - std::map<int, net_node_t*> connfdNodeMap; + // std::map<int, net_node_t*> connfdNodeMap; std::map<std::string, int> connectionMap; } ; -- Gitblit v1.8.0