linux c socket之异步IO

权声明：本文为博主原创文章，未经博主允许不得转载。

[cpp]  view plain copy

1. /*  
2.  * File:   UDPEchoClient-TimeOut.c 
3.  * Author: 云守护 
4.  */  
5. #include <stdio.h>  
6. #include <stdlib.h>  
7. #include <string.h>  
8. #include <unistd.h>  
9. #include <errno.h>  
10. #include <signal.h>  
11. #include <sys/socket.h>  
12. #include <netinet/in.h>  
13. #include <netdb.h>  
14. #include "DieWithMessage.h"  
15. //超时设置  
16. static const unsigned int TIMEOUT_SEC=3;  
17. //重连次数  
18. static const unsigned int MAXTRIES=5;  
19. //已经尝试次数  
20. unsigned int tries=0;  
21. //闹铃到来处理函数，超时时间到  
22. void CatchAlarm(int ignored);  
23. int main(int argc, char** argv) {  
24.     if(argc<3||argc>4)  
25.         DieWithUserMessage("param","<Server Address/Name> <Echo Word> [<Server port/service>]\n");  
26.     char *server=argv[1];  
27.     char* echoString=argv[2];  
28.     size_t echoStringLen=strlen(echoString);  
29.     if(echoStringLen>1024*8)  
30.         DieWithUserMessage(echoString,"too long");  
31.     char * service=(argc==4)?argv[3]:"echo";  
32.     //配置地址信息  
33.     struct addrinfo addr_criteria;  
34.     memset(&addr_criteria,0,sizeof(addr_criteria));  
35.     addr_criteria.ai_family=AF_UNSPEC;  
36.     addr_criteria.ai_socktype=SOCK_DGRAM;  
37.     addr_criteria.ai_protocol=IPPROTO_UDP;  
38.       
39.     struct addrinfo *server_addr;  
40.     //自动获取地址信息  
41.     int ret=getaddrinfo(server,service,&addr_criteria,&server_addr);  
42.     if(ret!=0)  
43.         DieWithUserMessage("getaddrinfo() failed!",gai_strerror(ret));  
44.     //建立socket  
45.     int sock=socket(server_addr->ai_family,server_addr->ai_socktype,server_addr->ai_protocol);  
46.     if(sock<0)  
47.         DieWithSystemMessage("socket failed!");  
48.     //信号处理者  
49.     struct sigaction handler;  
50.     //设置SIGALRM信号到来处理函数  
51.     handler.sa_handler=CatchAlarm;  
52.     //填充信号掩码  
53.     if(sigfillset(&handler.sa_mask)<0)  
54.         DieWithSystemMessage("sigfillset() failed!");  
55.     //信号标志归零  
56.     handler.sa_flags=0;  
57.     //注册信号量  
58.     if(sigaction(SIGALRM,&handler,0)<0)  
59.         DieWithSystemMessage("sigaction() failed for SIGALRM");  
60.     //开始发送数据  
61.     ssize_t numBytes=sendto(sock,echoString,echoStringLen,0,server_addr->ai_addr,server_addr->ai_addrlen);  
62.     if(numBytes<0)  
63.         DieWithSystemMessage("send() failed!");  
64.     else if(numBytes!=echoStringLen)  
65.     {  
66.         DieWithUserMessage("sendto() error","sent unexpected number of bytes");          
67.     }  
68.     //然后接收数据，都是循环接收的  
69.     struct sockaddr_storage fromaddr;  
70.     socklen_t fromaddrLen=sizeof(fromaddr);  
71.     //设置定时器，超时设置  
72.     alarm(TIMEOUT_SEC);  
73.     char buffer[BUFSIZ+1];  
74.     //接收数据  
75.     while((numBytes=recvfrom(sock,buffer,BUFSIZ,0,(struct sockaddr*)&fromaddr,&fromaddrLen))<0)  
76.     {  
77.         if(errno=EINTR)//闹铃响了，即超时了。执行了CatchAlarm  
78.         {  
79.             if(tries<MAXTRIES)  
80.             {   //小于重试次数，继续重新发送数据  
81.                 numBytes=sendto(sock,echoString,echoStringLen,0,(struct sockaddr*)server_addr->ai_addr,server_addr->ai_addrlen);  
82.                 if(numBytes<0)  
83.                     DieWithSystemMessage("sendto() failed!");  
84.                 else if(numBytes!=echoStringLen)  
85.                 {  
86.                     DieWithUserMessage("sendto() error","sent unexpected number of bytes");  
87.                 }  
88.             }else{  
89.                 DieWithUserMessage("No Response","Unable to communitcate with server");  
90.             }  
91.         }else{  
92.             DieWithSystemMessage("recvfrom() failed!");  
93.               
94.         }  
95.     }  
96.     //闹铃归零  
97.     alarm(0);  
98.     buffer[echoStringLen]='\0';  
99.     printf("Received: %s \n",buffer);  
100.     close(sock);  
101.     return (EXIT_SUCCESS);  
102. }  
103. //闹铃响了，执行该函数  
104. void CatchAlarm(int ignored){  
105.     tries+=1;  
106. }  

[cpp]  view plain copy

1. /*  
2.  * File:   UDPEchoServer-SIGIO.c 
3.  * Author: 云守护 
4.  */  
5. #include <stdio.h>  
6. #include <stdlib.h>  
7. #include <string.h>  
8. #include <unistd.h>  
9. #include <fcntl.h>  
10. #include <sys/file.h>  
11. #include <signal.h>  
12. #include <errno.h>  
13. #include <sys/types.h>  
14. #include <sys/socket.h>  
15. #include <netdb.h>  
16. #include "DieWithMessage.h"  
17. //用户空闲时任务  
18. void UseIdleTime();  
19. //异步IO处理者  
20. void SIGIOHandler(int signalType);  
21. int server_sock;  
22. #define MAXSTRINGLENGTH 1024*8  
23.   
24. //打印客户端地址消息  
25. void PrintSocketAddress(const struct sockaddr *address, FILE *stream) {  
26.   // Test for address and stream  
27.   if (address == NULL || stream == NULL)  
28.     return;  
29.   
30.   void *numericAddress; // Pointer to binary address  
31.   // Buffer to contain result (IPv6 sufficient to hold IPv4)  
32.   char addrBuffer[INET6_ADDRSTRLEN];  
33.   in_port_t port; // Port to print  
34.   // Set pointer to address based on address family  
35.   switch (address->sa_family) {  
36.   case AF_INET:  
37.     numericAddress = &((struct sockaddr_in *) address)->sin_addr;  
38.     port = ntohs(((struct sockaddr_in *) address)->sin_port);  
39.     break;  
40.   case AF_INET6:  
41.     numericAddress = &((struct sockaddr_in6 *) address)->sin6_addr;  
42.     port = ntohs(((struct sockaddr_in6 *) address)->sin6_port);  
43.     break;  
44.   default:  
45.     fputs("[unknown type]", stream);    // Unhandled type  
46.     return;  
47.   }  
48.   // Convert binary to printable address  
49.   if (inet_ntop(address->sa_family, numericAddress, addrBuffer,  
50.       sizeof(addrBuffer)) == NULL)  
51.     fputs("[invalid address]", stream); // Unable to convert  
52.   else {  
53.     fprintf(stream, "%s", addrBuffer);  
54.     if (port != 0)                // Zero not valid in any socket addr  
55.       fprintf(stream, "-%u", port);  
56.   }  
57. }  
58.   
59. int main(int argc, char** argv) {  
60.     if(argc!=2)  
61.         DieWithUserMessage("param","<server port/service>");  
62.     char *service=argv[1];  
63.     //配置地址信息  
64.     struct addrinfo addr_criteria;  
65.     memset(&addr_criteria,0,sizeof(addr_criteria));  
66.     addr_criteria.ai_family=AF_UNSPEC; //地址族  
67.     addr_criteria.ai_flags=AI_PASSIVE;  
68.     addr_criteria.ai_socktype=SOCK_DGRAM; //流  
69.     addr_criteria.ai_protocol=IPPROTO_UDP;//UDP  
70.       
71.     struct addrinfo *server_addr;  
72.     //自动获取，消除IP4和IP6的依赖  getaddrinfo() is reentrant and allows programs to eliminate IPv4-versus-IPv6 depen‐ dencies.  
73.     int retVal=getaddrinfo(NULL,service,&addr_criteria,&server_addr);  
74.     if(retVal!=0)  
75.         DieWithUserMessage("getaddrinfo() failed!",gai_strerror(retVal));  
76.     //建立socket  
77.     server_sock=socket(server_addr->ai_family,server_addr->ai_socktype,server_addr->ai_protocol);  
78.     if(server_sock<0)  
79.         DieWithSystemMessage("socket() failed!");  
80.     //绑定端口  
81.     if(bind(server_sock,server_addr->ai_addr,server_addr->ai_addrlen)<0)  
82.     {  
83.         DieWithSystemMessage("bind() failed!");  
84.     }  
85.     freeaddrinfo(server_addr);  
86.       
87.     printf("server success %s\n",service);  
88.     //操作信号，设置异步IO  
89.     struct sigaction handler;  
90.     //设置异步到来时的处理函数  
91.     handler.sa_handler=SIGIOHandler;  
92.     //填充信号掩码  
93.     if(sigfillset(&handler.sa_mask)<0)  
94.         DieWithSystemMessage("sigfillset() failed!");  
95.     //标志设置0，即无标志  
96.     handler.sa_flags=0;  
97.     //注册SIGIO信号  
98.     if(sigaction(SIGIO,&handler,0)<0)  
99.         DieWithSystemMessage("sigaction() failed for SIGIO");  
100.     //设置server_sock接收SIGIO信号信息  
101.     if(fcntl(server_sock,F_SETOWN,getpid())<0)  
102.         DieWithSystemMessage("fcntl failed!");  
103.     //使用非阻塞IO和SIGIO信号发送  
104.     if(fcntl(server_sock,F_SETFL,O_NONBLOCK|FASYNC)<0)  
105.         DieWithSystemMessage("Unable to put client sock into non-blocking/async mode");  
106.     for(;;)  
107.     {  
108.         //死循环，等待异步IO的到来。就不用使用多线程了。  
109.         UseIdleTime();  
110.     }  
111.     return (EXIT_SUCCESS);  
112. }  
113. void UseIdleTime()  
114. {  
115.     puts(".");  
116.     sleep(3);//3s  
117. }  
118. //异步IO处理函数  
119. void SIGIOHandler(int signalType)  
120. {  
121.     ssize_t numBytesRcvd;  
122.     do{  
123.         //只要有输入  
124.         struct sockaddr_storage client_addr;  
125.         size_t client_Len=sizeof(client_addr);  
126.         char buffer[MAXSTRINGLENGTH];  
127.         //接收数据  
128.         numBytesRcvd=recvfrom(server_sock,buffer,MAXSTRINGLENGTH,0,(struct sockaddr*)&client_addr,&client_Len);  
129.         if(numBytesRcvd<0)  
130.         {  
131.             if(errno!=EWOULDBLOCK)  
132.             {  
133.                 //判断recvfrom函数是否阻塞  
134.                 DieWithSystemMessage("recvfrom failed!");  
135.             }  
136.         }else{  
137.                 fprintf(stdout,"handling client ");  
138.                 //打印地址消息  
139.                 PrintSocketAddress((struct sockaddr*)&client_addr,stdout);  
140.                 fputc('\n',stdout);  
141.                 //把数据发送回去  
142.                 ssize_t numByteSent=sendto(server_sock,buffer,numBytesRcvd,0,(struct sockaddr*)&client_addr,sizeof(client_addr));  
143.                 if(numByteSent<0)  
144.                     DieWithSystemMessage("sendto() failed!");  
145.                 else if(numByteSent!=numBytesRcvd)  
146.                 {  
147.                     DieWithUserMessage("sendto()","sent unexpected number of bytes");  
148.                       
149.                 }  
150.         }  
151.         //循环接收数据  
152.     }while(numBytesRcvd>=0);  
153.       
154. }  

[cpp]  view plain copy

1. /*  
2.  * File:   DieWithMessage.h 
3.  * Author: root 
4.  * 
5.  * Created on 2013年11月13日, 下午3:52 
6.  */  
7.   
8. #ifndef DIEWITHMESSAGE_H  
9. #define DIEWITHMESSAGE_H  
10.   
11. #ifdef  __cplusplus  
12. extern "C" {  
13. #endif  
14. #include <stdio.h>  
15. #include <stdlib.h>  
16.  void  DieWithUserMessage(const char *msg,const char * detail)  
17.  {  
18.      fputs(msg,stderr);  
19.      fputs(":",stderr);  
20.      fputs(detail,stderr);  
21.      fputs("\n",stderr);  
22.      exit(1);  
23.  }  
24.  void DieWithSystemMessage(const char* msg)  
25.  {  
26.      perror(msg);  
27.      exit(1);  
28.  }  
29.   
30.   
31.   
32. #ifdef  __cplusplus  
33. }  
34. #endif  
35.   
36. #endif  /* DIEWITHMESSAGE_H */  

from： http://blog.csdn.net/earbao/article/details/16337639