使用aio写socket通信程序

realtang

以前我写了一篇用SIGIO完成socket编程的帖子。
http://linuxsir.cn/bbs/showthread.php?t=214611
这篇文章采用一种更好的方法来完成。
发送端代码

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. }

复制代码

发送端同上篇帖子相比，没啥区别。
主要不同的在于接收端，因为是采用了aio，所以很有些不同
接收端代码

2. #include <aio.h>
3. #include <stdio.h>
4. #include <stdlib.h>
5. #include <unistd.h>
6. #include <fcntl.h>
7. #include <errno.h>
8. #include <netinet/in.h>                 /*socket address struct*/
9. #include <arpa/inet.h>                        /*host to network convertion*/
10. #include <sys/socket.h>
11. #include <sys/types.h>
12. #include <signal.h>
13. //#include <sys/ioctl.h>
14. #define MAX_TRANSPORT_LENTH 512
15. #define MAX_LIST                5
17. void sig_handler(int signo, siginfo_t *info, void *context )
18. {
19.         int ret;
20.         struct aiocb *req;
21.         /* Ensure it's our signal */
22.         if (info->si_signo == SIGIO) {
24.                 req = (struct aiocb *)info->si_value.sival_ptr;
25.                 if (aio_error( req ) == 0) {
26.                         ret = aio_return( req );
27.                         (char*)(req->aio_buf+ret) == '\0';//通常是拷贝一下数据，交给线程池去处理，以空间换取时间，从而加快主循环
28.                         puts(req->aio_buf);
29.                 }
31.         }
33.         return;       
34. }
36. int main()
37. {
38.         int skt_fd,ret;
39.         struct aiocb my_aiocb;
40.         struct sigaction sig_act;
41.        
42.         struct sockaddr_in addr;
43.         memset(&addr,0,sizeof(addr));
44.         addr.sin_family =  AF_INET;
45.         addr.sin_addr.s_addr = INADDR_ANY;
46.         addr.sin_port = htons(50001);
47.        
48.        
49.         skt_fd = socket(AF_INET, SOCK_DGRAM, 0);
50.         if(skt_fd == -1)
51.         {
52.                 perror("Create socket failed");
53.                 exit(-1);
54.         }
55.         ret = bind(skt_fd,(struct sockaddr *)&addr,sizeof(addr));
56.         if(-1 == ret)
57.         {
58.                 perror("Bind socket failed");
59.                 exit(-1);
60.         }
62.         memset(&my_aiocb,0,sizeof(my_aiocb));
63.         my_aiocb.aio_buf = malloc(MAX_TRANSPORT_LENTH+1);       
64.         if (!my_aiocb.aio_buf)
65.                 perror("malloc");
66.        
67.         my_aiocb.aio_fildes = skt_fd;
68.         my_aiocb.aio_nbytes = MAX_TRANSPORT_LENTH;
69.         my_aiocb.aio_offset = 0;
70.        
71.         my_aiocb.aio_sigevent.sigev_notify = SIGEV_SIGNAL;
72.         my_aiocb.aio_sigevent.sigev_signo = SIGIO;
73.         my_aiocb.aio_sigevent.sigev_value.sival_ptr = &my_aiocb;
75.        
76.         sigemptyset(&sig_act.sa_mask);
77.         sig_act.sa_flags = SA_SIGINFO;
78.         sig_act.sa_sigaction = sig_handler;       
79.        
80.         const struct aiocb *cblist[MAX_LIST] ={&my_aiocb,NULL,NULL,NULL,NULL};
81.         sigaction( SIGIO, &sig_act, NULL );
82.        
83.         ret = aio_read( &my_aiocb );       
84.         //程序挂起直到等到某个信号
85.         while ( aio_suspend( cblist, MAX_LIST, NULL ) == 0)
86.         {
87.             ret = aio_read( &my_aiocb );
88.         };
89.         puts("Interrupted finish.");
90.         close(skt_fd);       
91. }

复制代码

在这里，如果用my_aiocb.aio_sigevent.sigev_notify = SIGEV_THREAD;
则可以指定一个回调函数来完成读取数据的操作，这种方法还是希望对此文感兴趣的伙伴加以完善吧。

haohao_h

linux下据说还是epoll的效率更高吧

信天游

编译接受程序出错：

[aaron@localhost tcp]$ gcc -o aiorecv aiorecv.c
aiorecv.c: 在函数 ‘sig_handler’ 中：
aiorecv.c:27: 警告：传递实参 1(属于 ‘puts’)丢弃了指针目标类型的限定
aiorecv.c: 在函数 ‘main’ 中：
aiorecv.c:42: 警告：隐式声明与内建函数 ‘memset’ 不兼容
/tmp/ccIg4qNh.o: In function `sig_handler':aiorecv.c.text+0x20)：对‘aio_error’未定义的引 用
:aiorecv.c.text+0x2f)：对‘aio_return’未定义的引用
/tmp/ccIg4qNh.o: In function `main':aiorecv.c.text+0x21d)：对‘aio_read’未定义的引用
:aiorecv.c.text+0x230)：对‘aio_read’未定义的引用
:aiorecv.c.text+0x251)：对‘aio_suspend’未定义的引用
collect2: ld 返回 1

realtang

编译时加上-lr选项

Illidan

aio是啥东东？

realtang

异步输入输出

tsuibin

写的真不错！

watalou

Post by realtang;1658105
编译时加上-lr选项

在Ubuntu 9.10下这个编译选项已经改为 -lrt　了