UNP学习笔记（第十四章 高级I/O函数）

本章讨论我们笼统地归为“高级I/O”的各个函数和技术

 

套接字超时

有3种方法在涉及套接字的I/O操作上设置超时

1.调用alarm，它在指定超时时期满时产生SIGALRM信号

2.在select中阻塞等待I/O（select有内置的时间限制），以此代替直接阻塞在read或write调用上

3.使用较新的SO_RCVTIMEO和SO_SNDTIMEO套接字选项。

 

使用SIGALRM为connect设置超时

下面给出我们的connect_timeo函数，它以调用者指定的超时上限调用connect

 1 /* include connect_timeo */
 2 #include    "unp.h"
 3 
 4 static void    connect_alarm(int);
 5 
 6 int
 7 connect_timeo(int sockfd, const SA *saptr, socklen_t salen, int nsec)
 8 {
 9     Sigfunc    *sigfunc;
10     int        n;
11 
12     sigfunc = Signal(SIGALRM, connect_alarm);
13     if (alarm(nsec) != 0)
14         err_msg("connect_timeo: alarm was already set");
15 
16     if ( (n = connect(sockfd, saptr, salen)) < 0) {
17         close(sockfd);
18         if (errno == EINTR)
19             errno = ETIMEDOUT;
20     }
21     alarm(0);                    /* turn off the alarm */
22     Signal(SIGALRM, sigfunc);    /* restore previous signal handler */
23 
24     return(n);
25 }
26 
27 static void
28 connect_alarm(int signo)
29 {
30     return;        /* just interrupt the connect() */
31 }
32 /* end connect_timeo */

View Code

如果connect被中断就会返回EINTR错误。

 

使用SIGALRM为recvfrom设置超时

 1 #include    "unp.h"
 2 
 3 static void    sig_alrm(int);
 4 
 5 void
 6 dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
 7 {
 8     int    n;
 9     char    sendline[MAXLINE], recvline[MAXLINE + 1];
10 
11     Signal(SIGALRM, sig_alrm);
12 
13     while (Fgets(sendline, MAXLINE, fp) != NULL) {
14 
15         Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
16 
17         alarm(5);
18         if ( (n = recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL)) < 0) {
19             if (errno == EINTR)
20                 fprintf(stderr, "socket timeout\n");
21             else
22                 err_sys("recvfrom error");
23         } else {
24             alarm(0);
25             recvline[n] = 0;    /* null terminate */
26             Fputs(recvline, stdout);
27         }
28     }
29 }
30 
31 static void
32 sig_alrm(int signo)
33 {
34     return;            /* just interrupt the recvfrom() */
35 }

View Code

 

使用select为recvfrom设置超时

 1 /* include readable_timeo */
 2 #include    "unp.h"
 3 
 4 int
 5 readable_timeo(int fd, int sec)
 6 {
 7     fd_set            rset;
 8     struct timeval    tv;
 9 
10     FD_ZERO(&rset);
11     FD_SET(fd, &rset);
12 
13     tv.tv_sec = sec;
14     tv.tv_usec = 0;
15 
16     return(select(fd+1, &rset, NULL, NULL, &tv));
17         /* 4> 0 if descriptor is readable */
18 }
19 /* end readable_timeo */

View Code

该函数等待一个描述符变为可读或者发生超时（超时返回0）

我们可以使用该函数来改写上面的dg_cli函数

 1 #include    "unp.h"
 2 
 3 void
 4 dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
 5 {
 6     int    n;
 7     char    sendline[MAXLINE], recvline[MAXLINE + 1];
 8 
 9     while (Fgets(sendline, MAXLINE, fp) != NULL) {
10 
11         Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
12 
13         if (Readable_timeo(sockfd, 5) == 0) {
14             fprintf(stderr, "socket timeout\n");
15         } else {
16             n = Recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);
17             recvline[n] = 0;    /* null terminate */
18             Fputs(recvline, stdout);
19         }
20     }
21 }

View Code

直到readable_timeo告知所关注的描述符已变为可读后我们才调用recvfrom（超时就打印错误）

 

使用SO_RCVTIMEO套接字选项为recvfrom设置超时

下面是使用SO_RCVTIMEO套接字选项的另一个版本的dg_cli函数

 1 #include    "unp.h"
 2 
 3 void
 4 dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
 5 {
 6     int                n;
 7     char            sendline[MAXLINE], recvline[MAXLINE + 1];
 8     struct timeval    tv;
 9 
10     tv.tv_sec = 5;
11     tv.tv_usec = 0;
12     Setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
13 
14     while (Fgets(sendline, MAXLINE, fp) != NULL) {
15 
16         Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
17 
18         n = recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);
19         if (n < 0) {
20             if (errno == EWOULDBLOCK) {
21                 fprintf(stderr, "socket timeout\n");
22                 continue;
23             } else
24                 err_sys("recvfrom error");
25         }
26 
27         recvline[n] = 0;    /* null terminate */
28         Fputs(recvline, stdout);
29     }
30 }

View Code

如果I/O操作超时，recvfrom将返回一个EWOULDBLOCK错误

 

 

 

read和write函数的变体

recv和send允许通过第四个参数从进程到内核传递标志；

readv和writev允许指定往其中输入数据或从其中输出数据的缓冲区向量；

recvmsg和sendmsg结合了其他I/O的所有特性，并具有接收和发送辅助数据的新能力

可以查看之前apue的学习笔记  http://www.cnblogs.com/runnyu/p/4648678.html