我的AIO测试程序

tsuibin

看过realtang写的aio程序，确实很棒，代码很精简。

这是我的测试用例，分享一下

2. #include <aio.h>
3. #include <sys/types.h>
4. #include <unistd.h>
5. #include <stdio.h>
6. #include <stdlib.h>
7. #include <netinet/in.h>
8. #include <sys/socket.h>
9. #include <arpa/inet.h>
10. #include <signal.h>
11. #include <sys/wait.h>
12. #include <string.h>
13. #include <errno.h>
15. #define SA struct sockaddr
16. #define LISTENPORT 8888
17. #define BACKLOG 1024
18. /* 编译的时候加 -lrt 参数 real-time 库 */
19. static char buf[BUFSIZ];
20. static struct aiocb aio;
21. static connfd;            /* connection socket descriptor */
22. static global_quit_flag;
24. void signal_handler(void);
25. void sig_chld(int signo);
26. void sig_io(int signo);
27. int create_listenfd(void);
28. void do_loop(int listenfd);
29. void do_request(void);
30. void init_aio(void);
31. void err_sys(const char *errmsg);
33. int main(void)
34. {   
35.         int listenfd;
37.         signal_handler();
39.         listenfd = create_listenfd();
41.         do_loop(listenfd);
43.         exit(0);
44. }
46. void signal_handler(void)
47. {
48.         struct sigaction sigio_act, sigchld_act;
50.         sigio_act.sa_handler = sig_io;   
51.         sigio_act.sa_flags = 0;   
52.         if (sigemptyset(&sigio_act.sa_mask) == -1)
53.                 err_sys("sigemptyset");   
54.         else if (sigaction(SIGIO, &sigio_act, NULL) == -1)
55.                 err_sys("sigaction");
57.         sigchld_act.sa_handler = sig_chld;
58.         sigchld_act.sa_flags = 0;
59.         if (sigemptyset(&sigchld_act.sa_mask) == -1)
60.                 err_sys("sigemptyset");
61.         else if (sigaction(SIGCHLD, &sigchld_act, NULL) == -1)
62.                 err_sys("sigaction");
63. }
65. int create_listenfd(void)
66. {
67.         int listenfd;
68.         struct sockaddr_in servaddr;
70.         if ((listenfd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1)
71.                 err_sys("socket");
73.         bzero(&servaddr, sizeof(servaddr));
74.         servaddr.sin_family = AF_INET;
75.         servaddr.sin_port = htons(LISTENPORT);
76.         servaddr.sin_addr.s_addr = htonl(INADDR_ANY);    /* wildcard address */
78.         if (bind(listenfd, (SA *)&servaddr, sizeof(servaddr)) == -1)
79.                 err_sys("bind");
80.         else if (listen(listenfd, BACKLOG) == -1)
81.                 err_sys("listen");
83.         return listenfd;
84. }
86. void do_loop(int listenfd)
87. {
88.         pid_t pid;
90.         for (;;) {        /* forever process client request */
91.                 if ((connfd = accept(listenfd, NULL, NULL)) == -1) {
92.                         if (errno == EINTR)
93.                                 continue;
94.                         else
95.                                 err_sys("accept");
96.                 }
98.                 if ((pid = fork()) == -1)
99.                         err_sys("fork");
100.                 else if (pid == 0) {        /* child process */
101.                         if (close(listenfd) == -1)
102.                                 err_sys("close");
104.                         do_request();   
106.                         for (;;) {
107.                                 if (global_quit_flag == 1)
108.                                         break;    /* jump out loop */
109.                                 pause();
110.                         }
112.                         exit(0);
113.                 }
115.                 if (close(connfd) == -1)
116.                         err_sys("close");
117.         }
118. }
120. void do_request(void)
121. {
122.         init_aio();        
123.         if (aio_read(&aio) == -1)
124.                 err_sys("aio_read");
125. }
127. void init_aio(void)
128. {
129.         aio.aio_fildes = connfd;
130.         aio.aio_offset = 0;
131.         aio.aio_buf = buf;
132.         aio.aio_nbytes = BUFSIZ;
133.         aio.aio_sigevent.sigev_notify = SIGEV_SIGNAL;
134.         aio.aio_sigevent.sigev_signo = SIGIO;
135.         aio.aio_sigevent.sigev_value.sival_ptr = &aio;
136. }
138. void sig_chld(int signo)
139. {   
140.         while (waitpid(-1, NULL, WNOHANG) > 0)
141.                 /* wait child process */;
142. }
144. void sig_io(int signo)
145. {
146.         int n;
148.         if (aio_error(&aio) == 0) {        /* I/O execute complete */
149.                 n = aio_return(&aio);
150.                 if (n == 0) {
151.                         global_quit_flag = 1;    /* client close connection */
152.                         return ;
153.                 }
154.                 if (send(connfd, (char *)aio.aio_buf, n, 0) != n)
155.                         err_sys("send");
156.                 do_request();
157.         }
158. }
160. void err_sys(const char *errmsg)
161. {
162.         perror(errmsg);
163.         exit(1);
164. }

复制代码