UNIX网络编程——客户/服务器心搏函数

下面是关于回送客户和服务器程序开发一些简单的心搏函数。这些函数可以发现对端主机或到对端的通信路径的过早失效。
         在给出这些函数之前我们必须提出一些警告。首先，有人会想到使用TCP的保持存活特性（SO_KEEPALIVE套接字选项）来提供这种功能，然而TCP得在连接已经闲置2小时之后才发送一个保持存活探测段。意识到这一点以后，他们的下一个问题是如何把保持存活参数改为一个小得多的值（往往是在秒钟的量级），以便更快的检测到失效。尽管缩短TCP的保持存活定时器参数在许多系统上确实可行，但是这些参数通常是按照内核而不是按照每个套接字维护的，因此改动他们将影响所有开启该选项的套接字。另外保持存活选项的用意绝不是这个目的（搞频率的轮询）。
        其次，两个端系统之间短暂的连接性丢失并非总是坏事。TCP一开始就设计成能够对付临时断连，而源自Berkeley的TCP实现将重传8-10分钟才放弃某个连接。较新的IP路由协议能够发现链接的失效，并且有可能在短时间内（譬如在秒钟量级上）启用候选的路径。因此应用程序开发人员必须审查想要引入心搏机制的具体应用，确实在没有听到对端应答的持续时间超过5-10S之后终止相应连接是件好事还是坏事。有些应用系统需要这种功能，不过大多数却并不需要。
       我们将使用TCP的紧急模式周期地轮询对端；在下面的讲解中我们假设每1S轮询一次，若持续5S没有听到对端应答则认为对端已不再存活，不过这些值可以由应用程序改动。

                     

       在这个例子中，客户每隔1S向服务器发送一个带外字节，服务器取该字节将导致它向客户发送回一个带外字节。每端都需要知道对端是否不复存在或者不再可达。客户和服务器每1S递增他们的cnt变量一次，每收到一个带外字节又把该变量重置为0。如果该计数器达到5（也就是说本进程已有5S没有收到来自对端的带外字节），那就认定连接失效。当有带外字节到达时，客户和服务器都是用SIGURG信号得以通知。我们在该图中间指出：数据，回送数据和带外字节都通过单个TCP连接交换。

 

       如下是我们的heatbeat_cli函数设置客户的心搏特性，其中第二个参数是以秒为单位的轮询频率，第三个参数是放弃当前连接之前应该经历的持续无响应轮询次数。

[cpp] view plaincopyprint?

1. #include "unp.h"  
2. /* 给heartbeat_cli的参数的拷贝: 套接口描述字(信号处理程序需用它来发送和接收带外数据)，SIGALRM的频率，在客户认为服务器或连接死掉之前没有来子服务器的响应的SIGALRM的总数，总量nprobes记录从最近一次服务器应答以来的SIGALRM的数目 */  
3. static int servfd;   
4. static int nsec;  /* #seconds between each alarm */  
5. static int maxnprobes;  /* #probes w/no response before quit */  
6. static int nprobes;  /* #probes since last server response */  
7. static void sig_urg(int), sig_alrm(int);  
8. void heartbeat_cli(int servfd_arg, int nsec_arg, int maxnprobes_arg)  
9. {/* heartbeat_cli函数检查并且保存参数，给SIGURG和SIGALRM建立信号处理函数，将套接口的属主设为进程ID，alarm调度一个SIGALRM */  
10.   servfd = servfd_arg;  /* set globals for signal handlers */  
11.   if( (nsec = nsec_arg) < 1)  
12.     nsec = 1;  
13.   if( (maxnprobes = maxnprobes_arg) < nsec)  
14.     maxnprobes = nsec;  
15.   nprobes = 0;  
16.   Signal(SIGURG, sig_urg);  
17.   Fcntl(servfd, F_SETOWN, getpid() );  
18.   Signal(SIGALRM, sig_alrm);  
19.   alarm(nesc);  
20. }  
21. static void sig_urg(int signo)  
22. {/* 当一个带外通知到来时，就会产生这个信号。我们试图去读带外字节，但如果还没有到(EWOULDBLOCK)也没有关系。由于系统不是在线接收带外数据，因此不会干扰客户读取它的普通数据。既然服务器仍然存活，nprobes就重置为0 */  
23.   int n;  
24.   char c;  
25.   if( ( n = recv(servfd, &c, 1, MSG_OOB) ) < 0 )  
26.   {  
27.     if(errno != EWOULDBLOCK)  
28.       err_sys("recv error");  
29.   }  
30.   nprobes = 0;  /* reset counter */  
31.   return;  /* may interrupt client code */  
32. }  
33. static void sig_alrm(int signo)  
34. {/* 这个信号以有规律间隔产生。计数器nprobes增1， 如果达到了maxnprobes，我们认为服务器或者崩溃或者不可达。在这个例子中，我们结束客户进程，尽管其他的设计也可以使用：可以发送给主循环一个信号，或者作为另外一个参数给heartbeat_cli提供一个客户函数，当服务器看来死掉时调用它 */  
35.   if( ++nprobes > maxnprobes)  
36.   {  
37.     fprintf(stderr, "server is unreachable \n");  
38.     exit(0);  
39.   }  
40.   Send(servfd, "1", 1, MSG_OOB);  
41.   alarm(nsec);  
42.   return; /* may interrupt client code */  
43. }  

全局变量
3-6     前3个变量是heartbeat_cli函数参数的副本：套接字描述符（信号处理函数用它来发送和接收带外数据），SIGALRM的频率，在客户认为服务器或连接不复存活之前处理的无服务器响应的SIGALRM总数。变量nprobes计量从收到来自服务器的最后一个应答以来处理的SIGALRM数目。

heartbeat_cli函数
8-20  heartbeat_cli函数检查并保存参数，给SIGURG和SIGALRM建立信号处理函数，并把套接字的属主设置为本进程ID。执行alarm以调度第一个SIGALRM.

SIGURG处理函数
21-32   本信号在某个带外通知到达时产生。我们尝试读入相应的带外字节，不过如果它还没有到达（EWOULDBLOCK），那也没有关系。注意，我们不采用在线接收带外数据方式，因为这种方式会干扰客户读取它的正常数据。既然服务器仍然存活着，我们把nprobes重置为0.

SIGALRM处理函数
33-43   本信号以恒定的间隔产生。递增计数器nprobes，如果达到maxnprobes，我们就认定服务器主机或者已经崩溃，或者不再可达。我们这里是直接结束客户进程。作为带外数据发送一个含有字符1的字节（该值没有任何隐含意义），再执行alarm调度下一个SIGALRM。

 

      下面是服务器程序的心搏函数。

[cpp] view plaincopyprint?

1. #include "unp.h"  
2. static int servfd;  
3. static int nsec;  /* #seconds between each alarm */  
4. static int maxnalarms; /* #alarms w/no client probe before quit */  
5. static int nprobes;  /* #alarms since last client probe */  
6. static void sig_urg(int), sig_alrm(int);  
7. void heartbeat_serv(int servfd_arg, int nsec_arg, int maxnalarms_arg)  
8. {  
9.   servfd = servfd_arg;  /* set globals for signal handlers */  
10.   if( (nsec = nsec_arg) < 1 )  
11.     nsec = 1;  
12.   if( (maxnalarms = maxnalarms_arg) < nsec)  
13.     maxnalarms = nsec;  
14.   Signal(SIGURG, sig_urg);  
15.   Fcntl(servfd, F_SETOWN, getpid());  
16.   Signal(SIGALRM, sig_alrm);  
17.   alarm(nsec);  
18. }  
19. static void sig_urg(int signo)  
20. { /* 当一个带外通知收到时， 服务器试图读入它。就像客户一样，如果带外字节没有到达没有什么关系。带外字节被作为带外数据返回给客户。注意，如果recv返回EWOULDBLOCK错误，那么自动变量c碰巧是什么就送给客户什么。由于我们不用带外字节的值，所以这没有关系。重要的是发送1字节的带外数据，而不管该字节是什么。由于刚收到通知，客户仍存活，所以重置nprobes为0 */  
21.   int n;  
22.   char c;  
23.   if( (n = recv(servfd, &c, 1, MSG_OOB)) < 0)  
24.   {  
25.     if(errno != EWOULDBLOCK)  
26.       err_sys("recv error");  
27.   }  
28.   Send(servfd, &c, 1, MSG_OOB);  /* echo back out-of-hand byte */  
29.   nprobes = 0;  /* reset counter */  
30.   return;  /* may interrupt server code */  
31. }  
32. static void sig_alrm(int signo)  
33. { /* nprobes增1， 如果它到达了调用者指定的值maxnalarms，服务器进程将被终止。否则调度一下SIGALRM */  
34.   if( ++nprobes > maxnalarms)  
35.   {  
36.     printf("no probes from client\n");  
37.     exit(0);  
38.   }  
39.   alarm(nsec);  
40.   return;   /* may interrupt server code */  
41. }  

heartbeat_serv函数
7-18    声明变量，函数heartbeat_serv几乎与客户的心搏初始化函数一样。

SIGURG处理函数
19-31   服务器收到一个带外通知后就尝试读入相应的带外字节。就像客户一样，如果该带外字节还没有到达，那也没有声明关系。服务器把读入的带外字节作为带外数据回送给客户。注意，如果recv返回EWOULDBLOCK错误，那么自动变量C碰巧是什么就回送什么。既然我们不把带外字节的值用于任何目的，这么处置就不会有问题。重要的是发送1字节的带外数据本身，而不是该字节到底是什么。既然刚收到客户仍然存活着的通知，我们把nprobes重置为0.

SIGALRM处理函数
32-41    递增nprobes，如果达到由调用者指定的maxnalarms值，那就终止服务器进程，否则调度下一个SIGALRM。