一个并发10k的简易群聊天室（使用epoll）

local.h

点击(此处)折叠或打开

1. #include <sys/socket.h>
   
2. #include <sys/epoll.h>
   
3. #include <netinet/in.h>
   
4. #include <arpa/inet.h>
   
5. #include <fcntl.h>
   
6. #include <unistd.h>
   
7. #include <stdio.h>
   
8. #include <stdlib.h>
   
9. #include <errno.h>
   
10. #include <time.h>
    
11. #include <list>
    
12. #include <string.h>
    
13. 
    
14. #define BUF_SIZE 1024 //默认缓冲区
    
15. #define SERVER_PORT 8888 //监听端口
    
16. //#define SERVER_HOST "192.168.34.15" //服务器IP地址
    
17. #define SERVER_HOST "127.0.0.1" //服务器IP地址
    
18. #define EPOLL_RUN_TIMEOUT -1 //epoll的超时时间
    
19. #define EPOLL_SIZE 10000 //epoll监听的客户端的最大数目
    
20. 
    
21. #define STR_WELCOME "Welcome to seChat! You ID is: Client #%d"
    
22. #define STR_MESSAGE "Client #%d>> %s"
    
23. #define STR_NOONE_CONNECTED "Noone connected to server except you!"
    
24. #define CMD_EXIT "EXIT"
    
25. 
    
26. //两个有用的宏定义：检查和赋值并且检测
    
27. #define CHK(eval) if(eval < 0){perror("eval"); exit(-1);}
    
28. #define CHK2(res, eval) if((res = eval) < 0){perror("eval"); exit(-1);}
    
29. 
    
30. //================================================================================================
    
31. //函数名： setnonblocking
    
32. //函数描述： 设置socket为不阻塞
    
33. //输入： [in] sockfd socket标示符
    
34. //输出： 无
    
35. //返回： 0
    
36. //================================================================================================
    
37. int setnonblocking(int sockfd);
    
38. 
    
39. //================================================================================================
    
40. //函数名： handle_message
    
41. //函数描述： 处理每个客户端socket
    
42. //输入： [in] new_fd socket标示符
    
43. //输出： 无
    
44. //返回： 返回从客户端接受的数据的长度
    
45. //================================================================================================
    
46. int handle_message(int new_fd);

utils.h

点击(此处)折叠或打开

1. #include "local.h"
   
2. 
   
3. 
   
4. int setnonblocking(int sockfd)
   
5. {
   
6.     CHK(fcntl(sockfd, F_SETFL, fcntl(sockfd, F_GETFD, 0)|O_NONBLOCK));
   
7.     return 0;
   
8. }

server.cpp

点击(此处)折叠或打开

1. #include "local.h"
   
2. #include "utils.h"
   
3. 
   
4. using namespace std;
   
5. 
   
6. // 存放客户端socket描述符的list
   
7. list<int> clients_list;
   
8. 
   
9. int main(int argc, char *argv[])
   
10. {
    
11.     int listener; //监听socket
    
12.     struct sockaddr_in addr, their_addr; 
    
13.     addr.sin_family = PF_INET;
    
14.     addr.sin_port = htons(SERVER_PORT);
    
15.     addr.sin_addr.s_addr = inet_addr(SERVER_HOST);
    
16.     socklen_t socklen;
    
17.     socklen = sizeof(struct sockaddr_in);
    
18. 
    
19.     static struct epoll_event ev, events[EPOLL_SIZE];
    
20.     ev.events = EPOLLIN | EPOLLET; //对读感兴趣，边沿触发
    
21. 
    
22.     char message[BUF_SIZE];
    
23. 
    
24.     int epfd; //epoll描述符
    
25.     clock_t tStart; //计算程序运行时间
    
26. 
    
27.     int client, res, epoll_events_count;
    
28. 
    
29.     CHK2(listener, socket(PF_INET, SOCK_STREAM, 0)); //初始化监听socket
    
30.     setnonblocking(listener); //设置监听socket为不阻塞
    
31.     CHK(bind(listener, (struct sockaddr *)&addr, sizeof(addr))); //绑定监听socket
    
32.     CHK(listen(listener, 1)); //设置监听
    
33. 
    
34.     CHK2(epfd,epoll_create(EPOLL_SIZE)); //创建一个epoll描述符，并将监听socket加入epoll
    
35.     ev.data.fd = listener;
    
36.     CHK(epoll_ctl(epfd, EPOLL_CTL_ADD, listener, &ev));
    
37. 
    
38.     while(1)
    
39.     {
    
40.         CHK2(epoll_events_count,epoll_wait(epfd, events, EPOLL_SIZE, EPOLL_RUN_TIMEOUT));
    
41.         tStart = clock();
    
42.         for(int i = 0; i < epoll_events_count ; i++)
    
43.         {
    
44.             if(events[i].data.fd == listener) //新的连接到来，将连接添加到epoll中，并发送欢迎消息
    
45.             {
    
46.                 CHK2(client,accept(listener, (struct sockaddr *) &their_addr, &socklen));
    
47.                 setnonblocking(client);
    
48.                 ev.data.fd = client;
    
49.                 CHK(epoll_ctl(epfd, EPOLL_CTL_ADD, client, &ev));
    
50. 
    
51.                 clients_list.push_back(client); // 添加新的客户端到list
    
52.                 bzero(message, BUF_SIZE);
    
53.                 res = sprintf(message, STR_WELCOME, client);
    
54.                 CHK2(res, send(client, message, BUF_SIZE, 0));
    
55. 
    
56.             }else 
    
57.             {
    
58.                 CHK2(res,handle_message(events[i].data.fd)); //注意：这里并没有调用epoll_ctl重新设置socket的事件类型，但还是可以继续收到客户端发送过来的信息
    
59.             }
    
60.         }
    
61.         printf("Statistics: %d events handled at: %.2f second(s)\n", epoll_events_count, (double)(clock() - tStart)/CLOCKS_PER_SEC);
    
62.     }
    
63. 
    
64.     close(listener);
    
65.     close(epfd);
    
66. 
    
67.     return 0;
    
68. }
    
69. 
    
70. int handle_message(int client) 
    
71. {
    
72.     char buf[BUF_SIZE], message[BUF_SIZE];
    
73.     bzero(buf, BUF_SIZE);
    
74.     bzero(message, BUF_SIZE);
    
75. 
    
76.     int len;
    
77. 
    
78.     CHK2(len,recv(client, buf, BUF_SIZE, 0)); //接受客户端信息
    
79. 
    
80.     if(len == 0) //客户端关闭或出错，关闭socket，并从list移除socket
    
81.     {
    
82.         CHK(close(client));
    
83.         clients_list.remove(client);
    
84.     }
    
85.     else //向客户端发送信息
    
86.     { 
    
87.         if(clients_list.size() == 1) 
    
88.         { 
    
89.             CHK(send(client, STR_NOONE_CONNECTED, strlen(STR_NOONE_CONNECTED), 0));
    
90.                 return len;
    
91.         }
    
92.         
    
93.         sprintf(message, STR_MESSAGE, client, buf);
    
94.         list<int>::iterator it;
    
95.         for(it = clients_list.begin(); it != clients_list.end(); it++)
    
96.         {
    
97.            if(*it != client)
    
98.            { 
    
99.                 CHK(send(*it, message, BUF_SIZE, 0));
    
100.            }
     
101.         }
     
102.     }
     
103. 
     
104.     return len;
     
105. }

10kclient.cpp

点击(此处)折叠或打开

1. #include "local.h"
   
2. #include "utils.h"
   
3. 
   
4. using namespace std;
   
5. 
   
6. char message[BUF_SIZE]; //接受服务器信息
   
7. list<int> list_of_clients; //存放所有客户端
   
8. int res;
   
9. clock_t tStart;
   
10. 
    
11. int main(int argc, char *argv[])
    
12. {
    
13.     int sock; 
    
14.     struct sockaddr_in addr;
    
15.     addr.sin_family = PF_INET;
    
16.     addr.sin_port = htons(SERVER_PORT);
    
17.     addr.sin_addr.s_addr = inet_addr(SERVER_HOST);
    
18.     
    
19.     tStart = clock();
    
20. 
    
21.     for(int i=0 ; i<EPOLL_SIZE; i++) //生成EPOLL_SIZE个客户端，这里是10000个，模拟高并发
    
22.     {
    
23.        CHK2(sock,socket(PF_INET, SOCK_STREAM, 0));
    
24.        CHK(connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0);
    
25.        list_of_clients.push_back(sock);
    
26. 
    
27.        bzero(&message, BUF_SIZE);
    
28.        CHK2(res,recv(sock, message, BUF_SIZE, 0));
    
29.        printf("%s\n", message);
    
30.     }
    
31.    
    
32.     list<int>::iterator it; //移除所有客户端
    
33.     for(it = list_of_clients.begin(); it != list_of_clients.end() ; it++)
    
34.        close(*it);
    
35. 
    
36.     printf("Test passed at: %.2f second(s)\n", (double)(clock() - tStart)/CLOCKS_PER_SEC); 
    
37.     printf("Total server connections was: %d\n", EPOLL_SIZE);
    
38.     
    
39.     return 0;
    
40. }

参考资料：
google code  svn路径：http://sechat.googlecode.com/svn/trunk/
http://www.cnblogs.com/venow/archive/2012/11/30/2790031.html