1 #include2 #include 3 #include 4 #include 5 #include 6 7 #define PORT 6666 8 #define BUFFER_SIZE 2048 9 #define MAX_ALLOWED_CLIENT 10240 10 11 struct ev_io *libevlist[MAX_ALLOWED_CLIENT] = {NULL}; 12 13 void accept_cb(struct ev_loop *loop, struct ev_io *watcher, int revents); 14 15 void read_cb(struct ev_loop *loop, struct ev_io *watcher, int revents); 16 17 void timer_beat(struct ev_loop *loop, struct ev_timer *watcher, int revents); 18 19 int freelibev(struct ev_loop *loop, int fd); 20 21 22 int main() 23 { 24 struct ev_loop *loop=ev_default_loop(0); 25 int sd; 26 struct sockaddr_in addr; 27 int addr_len = sizeof(addr); 28 29 //创建一个io watcher和一个timer watcher 30 struct ev_io socket_accept; 31 struct ev_timer timeout_w; 32 //创建socket连接 33 sd = socket(PF_INET, SOCK_STREAM, 0); 34 if(sd < 0) 35 { 36 printf("socket error\n"); 37 return -1; 38 } 39 bzero(&addr, sizeof(addr)); 40 addr.sin_family = AF_INET; 41 addr.sin_port = htons(PORT); 42 addr.sin_addr.s_addr = INADDR_ANY; 43 //正常bind 44 if(bind(sd, (struct sockaddr*)&addr, sizeof(addr)) != 0) 45 { 46 printf("bind error\n"); 47 return -1; 48 } 49 //正常listen 50 if(listen(sd, SOMAXCONN) < 0) 51 { 52 printf("listen error\n"); 53 return -1; 54 } 55 //设置fd可重用 56 int bReuseaddr=1; 57 if(setsockopt(sd,SOL_SOCKET ,SO_REUSEADDR,(const char*)&bReuseaddr,sizeof(bReuseaddr)) != 0) 58 { 59 printf("setsockopt error in reuseaddr[%d]\n", sd); 60 return ; 61 } 62 63 //初始化io watcher，用于监听fd 64 ev_io_init(&socket_accept, read_cb, sd, EV_READ); 65 ev_io_start(loop, &socket_accept); 66 67 ev_run(loop, 0); 68 69 return 1; 70 } 71 72 void accept_cb(struct ev_loop *loop, struct ev_io *watcher, int revents) 73 { 74 75 // 如果有链接，则继续监听fd； 76 77 struct sockaddr_in client_addr; 78 socklen_t client_len = sizeof(client_addr); 79 int client_sd; 80 //创建客户端的io watcher 81 struct ev_io *w_client = (struct ev_io*) malloc(sizeof(struct ev_io)); 82 83 if(w_client == NULL) 84 { 85 printf("malloc error in accept_cb\n"); 86 return ; 87 } 88 89 if(EV_ERROR & revents) 90 { 91 printf("error event in accept\n"); 92 return ; 93 } 94 95 //获取与客户端相连的fd 96 client_sd = accept(watcher->fd, (struct sockaddr*)&client_addr, &client_len); 97 if(client_sd < 0) 98 { 99 printf("accept error\n");100 return;101 }102 //如果连接数超出指定范围，则关闭连接103 if( client_sd > MAX_ALLOWED_CLIENT)104 {105 printf("fd too large[%d]\n", client_sd);106 close(client_sd);107 return ;108 }109 110 if(libevlist[client_sd] != NULL)111 {112 printf("client_sd not NULL fd is [%d]\n", client_sd);113 return ;114 }115 116 printf("client connected\n");117 //监听新的fd118 ev_io_init(w_client, read_cb, client_sd, EV_READ);119 ev_io_start(loop, w_client);120 121 libevlist[client_sd] = w_client;122 123 }124 125 void read_cb(struct ev_loop *loop, struct ev_io *watcher, int revents)126 {127 char buffer[BUFFER_SIZE];128 ssize_t read;129 130 if(EV_ERROR & revents)131 {132 printf("error event in read\n");133 return ;134 }135 //正常的recv136 read = recv(watcher->fd, buffer, BUFFER_SIZE, 0);137 if(read < 0)138 {139 printf("read error\n");140 return;141 }142 143 if(read == 0)144 {145 printf("client disconnected.\n");146 //ev_io_stop(loop, watcher);147 //free(watcher);148 //如果客户端断开连接，释放响应的资源，并且关闭监听149 freelibev(loop, watcher->fd);150 return;151 }152 else153 {154 //buffer[read] = '\0';155 printf("receive message:%s\n", buffer);156 }157 //返回给客户端158 send(watcher->fd, buffer, read, 0);159 bzero(buffer, read);160 }161 162 int freelibev(struct ev_loop *loop, int fd)163 {164 //清理相关资源165 166 if(libevlist[fd] == NULL)167 {168 printf("the fd already freed[%d]\n", fd);169 return -1;170 }171 172 close(fd);173 ev_io_stop(loop, libevlist[fd]);174 free(libevlist[fd]);175 libevlist[fd] = NULL;176 return 1;177 }