实时SIGIO队列与sigwaitinfo的结合使用（提高socket服务能力）

realtang

这个是接收端也就是服务端的代码。

2. #define _GNU_SOURCE
4. #include <stdio.h>
5. #include <stdlib.h>
6. #include <unistd.h>
7. #include <fcntl.h>
8. #include <errno.h>
9. #include <string.h>
10. #include <signal.h>
11. #include <netinet/in.h>
12. #include <arpa/inet.h>
13. #include <sys/socket.h>
14. #include <sys/types.h>
15. #include <sys/ioctl.h>
17. #define MAX_LENTH 1500
19. #define SIG_FIFO_IO SIGRTMIN+1
20. //这时SIGIO表示实时信号队列已满
22. static char recv_buf[MAX_LENTH] = {0};
24. int main(int argc, char *argv[])
25. {
26.         int sockfd, on = 1;
27.         struct sigaction action;
28.         sigset_t newmask, oldmask;
29.         struct sockaddr_in addr;
31.         memset(&addr, 0, sizeof(addr));
32.         addr.sin_family =  AF_INET;
33.         addr.sin_addr.s_addr = INADDR_ANY;
34.         addr.sin_port = htons(50001);
35.         if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
36.                 perror("Create socket failed");
37.                 exit(-1);
38.         }
39.         if (bind(sockfd, (struct sockaddr *)&addr, sizeof(addr)) == -1) {
40.                 perror("Bind socket failed");
41.                 exit(-1);
42.         }
44. /*         memset(&action, 0, sizeof(action));
45.         action.sa_handler = sigio_handler;
46.         action.sa_flags = 0;
47.         sigaction(SIGIO, &action, NULL); */
48.        
49.         if (fcntl(sockfd, F_SETOWN, getpid()) == -1) {
50.                 perror("Fcntl F_SETOWN failed");
51.                 exit(-1);               
52.         }
53.         if (fcntl(sockfd, F_SETSIG, SIG_FIFO_IO) == -1) {
54.                 perror("Fcntl F_SETSIG failed");
55.                 exit(-1);               
56.         }       
57.         if (ioctl(sockfd, FIOASYNC, &on) == -1) {
58.                 perror("Ioctl FIOASYNC failed");
59.                 exit(-1);               
60.         }
61.         
62.         int rcvd_sig, len;
63.         siginfo_t info;
64.         sigemptyset(&newmask);
65.         sigaddset(&newmask, SIG_FIFO_IO);
66.         sigprocmask(SIG_BLOCK, &newmask, &oldmask);
68.         while (1) {
69.                 rcvd_sig = sigwaitinfo(&newmask, &info);//siginfo中的_sigpool{si_band, si_fd},其中si_fd为socket的文件描述符值
70.                
71.                 if (rcvd_sig == -1) {
72.                    perror("sigio: sigwaitinfo");
73.                    exit(-1);
74.                 }
75.                 else {
76.                         printf("Signal %d, socket fd %d  received from socket\n",
77.                                  rcvd_sig, info.si_fd);
78.                         len = recv(sockfd, recv_buf, MAX_LENTH, MSG_DONTWAIT);
79.                         if( len>0)
80.                                 puts( recv_buf);
81.                         exit(-1);
82.                 }
83.         }
85. }

复制代码

效率应该比较高，通过siginfo_t的si_fd可以找到相应的socket文件描述符，所以对多个客户端的连接也能处理，如果还配以使用si_band还能知道该信号具体的含义如下：
POLL_IN     Data input available.  
POLL_OUT     Output buffers available.  
POLL_MSG     Input message available.  
POLL_ERR     I/O error.   
POLL_PRI     High priority input available.  
POLL_HUP     Device disconnected.
这些东西很像poll系统调用中struct pollfd中的revents。   

发送端的代码跟我以前写的这个系列的一样，抄在下方。

2. #include <stdio.h>
3. #include <stdlib.h>
4. #include <unistd.h>
5. #include <fcntl.h>
6. #include <errno.h>
7. #include <netinet/in.h>                 /*socket address struct*/
8. #include <arpa/inet.h>                        /*host to network convertion*/
9. #include <sys/socket.h>
10. #include <signal.h>
11. #define MAX_TRANSPORT_LENTH 512
13. int main()
14. {
15.         struct sockaddr_in addr;
16.         memset(&addr,0,sizeof(addr));
17.         addr.sin_family =  AF_INET;
18.         addr.sin_addr.s_addr = inet_addr("127.0.0.1");
19.         addr.sin_port = htons(50001);
20.        
21.         int sock;
22.         sock = socket(AF_INET,SOCK_DGRAM,0);
23.         if(sock == -1)
24.         {
25.                 perror("Create socket failed");
26.                 exit(-1);
27.         }
28.        
29.         int ret;
30.         ret = connect(sock,(struct sockaddr *)&addr,sizeof(addr));
31.         if(ret == -1)
32.         {
33.                 perror("Connect socket failed");
34.                 exit(-1);               
35.         }               
36.         while(1)
37.         {
38.                 printf("Will send messge to server\n");
39.                 write(sock,"Some unknown infomation\n",MAX_TRANSPORT_LENTH);
40.                 sleep(1);
41.         }
42.        
43. }

复制代码

以上代码版权属于realtang，您可以在GPLv3授权下使用。

maqianli

请教老大：sigaction(SIGIO, &action, NULL);  中的信号有没有自定义的说法？
windows下不是可以自定义消息，linux支持这种做法吗？我一直困惑这个问题。

如果不能自定义，那怎么知道发出的信号是哪个信号？谢谢！

maqianli

还问老大：
怎么不直接把对客户端的处理直接放才信号处理好处里边？