主要内容：保活定时器的实现，TCP_USER_TIMEOUT选项的实现。
内核版本：3.15.2
我的博客：http://blog.csdn.net/zhangskd
原理
HTTP有Keepalive功能，TCP也有Keepalive功能，虽然都叫Keepalive，但是它们的目的却是不一样的。
为了说明这一点，先来看下长连接和短连接的定义。
连接的“长短”是什么？
短连接：建立一条连接，传输一个请求，马上关闭连接。
长连接：建立一条连接，传输一个请求，过会儿，又传输若干个请求，最后再关闭连接。
长连接的好处是显而易见的，多个请求可以复用一条连接，省去连接建立和释放的时间开销和系统调用，
但也意味着服务器的一部分资源会被长时间占用着。
HTTP的Keepalive，顾名思义，目的在于延长连接的时间，以便在同一条连接中传输多个HTTP请求。
HTTP服务器一般会提供Keepalive Timeout参数，用来决定连接保持多久，什么时候关闭连接。
当连接使用了Keepalive功能时，对于客户端发送过来的一个请求，服务器端会发送一个响应，然后开始计时，
如果经过Timeout时间后，客户端没有再发送请求过来，服务器端就把连接关了，不再保持连接了。
TCP的Keepalive，是挂羊头卖狗肉的，目的在于看看对方有没有发生异常，如果有异常就及时关闭连接。
当传输双方不主动关闭连接时，就算双方没有交换任何数据，连接也是一直有效的。
如果这个时候对端、中间网络出现异常而导致连接不可用，本端如何得知这一信息呢？
答案就是保活定时器。它每隔一段时间会超时，超时后会检查连接是否空闲太久了，如果空闲的时间超过
了设置时间，就会发送探测报文。然后通过对端是否响应、响应是否符合预期，来判断对端是否正常，
如果不正常，就主动关闭连接，而不用等待HTTP层的关闭了。
当服务器发送探测报文时，客户端可能处于4种不同的情况：仍然正常运行、已经崩溃、已经崩溃并重启了、
由于中间链路问题不可达。在不同的情况下，服务器会得到不一样的反馈。
(1) 客户主机依然正常运行，并且从服务器端可达
客户端的TCP响应正常，从而服务器端知道对方是正常的。保活定时器会在两小时以后继续触发。
(2) 客户主机已经崩溃，并且关闭或者正在重新启动
客户端的TCP没有响应，服务器没有收到对探测包的响应，此后每隔75s发送探测报文，一共发送9次。
socket函数会返回-1，errno设置为ETIMEDOUT，表示连接超时。
(3) 客户主机已经崩溃，并且重新启动了
客户端的TCP发送RST，服务器端收到后关闭此连接。
socket函数会返回-1，errno设置为ECONNRESET，表示连接被对端复位了。
(4) 客户主机依然正常运行，但是从服务器不可达
双方的反应和第二种是一样的，因为服务器不能区分对端异常与中间链路异常。
socket函数会返回-1，errno设置为EHOSTUNREACH，表示对端不可达。
选项
内核默认并不使用TCP Keepalive功能，除非用户设置了SO_KEEPALIVE选项。
有两种方式可以自行调整保活定时器的参数：一种是修改TCP参数，一种是使用TCP层选项。
(1) TCP参数
tcp_keepalive_time
最后一次数据交换到TCP发送第一个保活探测报文的时间，即允许连接空闲的时间，默认为7200s。
tcp_keepalive_intvl
保活探测报文的重传时间，默认为75s。
tcp_keepalive_probes
保活探测报文的发送次数，默认为9次。
Q：一次完整的保活探测需要花费多长时间？
A：tcp_keepalive_time + tcp_keepalive_intvl * tcp_keepalive_probes，默认值为7875s。
如果觉得两个多小时太长了，可以自行调整上述参数。
(2) TCP层选项
TCP_KEEPIDLE：含义同tcp_keepalive_time。
TCP_KEEPINTVL：含义同tcp_keepalive_intvl。
TCP_KEEPCNT：含义同tcp_keepalive_probes。
Q：既然有了TCP参数可供调整，为什么还增加了上述的TCP层选项？
A：TCP参数是面向本机的所有TCP连接，一旦调整了，对所有的连接都有效。
而TCP层选项是面向一条连接的，一旦调整了，只对本条连接有效。
激活
在连接建立后，可以通过设置SO_KEEPALIVE选项，来激活保活定时器。
int keepalive = 1;
setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive));
int sock_setsockopt(struct socket *sock, int level, int optname, char __user *optval,
unsigned int optlen)
{
...
case SO_KEEPALIVE:
#ifdef CONFIG_INET
if (sk->sk_protocol == IPPROTO_TCP && sk->sk_type == SOCK_STREAM)
tcp_set_keepalive(sk, valbool); /* 激活或删除保活定时器 */
#endif
sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool); /* 设置或取消SOCK_KEEPOPEN标志位 */
break;
...
}
static inline void sock_valbool_flag (struct sock *sk, int bit, int valbool)
{
if (valbool)
sock_set_flag(sk, bit);
else
sock_reset_flag(sk, bit);
}
void tcp_set_keepalive(struct sock *sk, int val){ /* 不在以下两个状态设置保活定时器： * TCP_CLOSE：sk_timer用作FIN_WAIT2定时器 * TCP_LISTEN：sk_timer用作SYNACK重传定时器 */ if ((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)) return; /* 如果SO_KEEPALIVE选项值为1，且此前没有设置SOCK_KEEPOPEN标志， * 则激活sk_timer，用作保活定时器。 */ if (val && !sock_flag(sk, SOCK_KEEPOPEN)) inet_csk_reset_keepalive_timer(sk, keepalive_time_when(tcp_sk(sk))); else if (!val) /* 如果SO_KEEPALIVE选项值为0，则删除保活定时器 */ inet_csk_delete_keepalive_timer(sk);} /* 保活定时器的超时时间 */static inline int keepalive_time_when(const struct tcp_sock *tp){ return tp->keepalive_time ? : sysctl_tcp_keepalive_time;}void inet_csk_reset_keepalive_timer (struc sock *sk, unsigned long len){ sk_reset_timer(sk, &sk->sk_timer, jiffies + len);}
可以使用TCP层选项来动态调整保活定时器的参数。
int keepidle = 600;
int keepintvl = 10;
int keepcnt = 6;
setsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &keepidle, sizeof(keepidle));
setsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &keepintvl, sizeof(keepintvl));
setsockopt(fd, SOL_TCP, TCP_KEEPCNT, &keepcnt, sizeof(keepcnt));
struct tcp_sock {
...
/* 最后一次接收到ACK的时间 */
u32 rcv_tstamp; /* timestamp of last received ACK (for keepalives) */
...
/* time before keep alive takes place, 空闲多久后才发送探测报文 */
unsigned int keepalive_time;
/* time iterval between keep alive probes */
unsigned int keepalive_intvl; /* 探测报文之间的时间间隔 */
/* num of allowed keep alive probes */
u8 keepalive_probes; /* 探测报文的发送次数 */
...
struct {
...
/* 最后一次接收到带负荷的报文的时间 */
__u32 lrcvtime; /* timestamp of last received data packet */
...
} icsk_ack;
...
};
#define TCP_KEEPIDLE 4 /* Start Keepalives after this period */
#define TCP_KEEPINTVL 5 /* Interval between keepalives */
#define TCP_KEEPCNT 6 /* Number of keepalives before death */
#define MAX_TCP_KEEPIDLE 32767
#define MAX_TCP_KEEPINTVL 32767
#define MAX_TCP_KEEPCNT 127
static int do_tcp_setsockopt(struct sock *sk, int level, int optname, char __user *optval, unsigned int optlen){ ... case TCP_KEEPIDLE: if (val < 1 || val > MAX_TCP_KEEPIDLE) err = -EINVAL; else { tp->keepalive_time = val * HZ; /* 设置新的空闲时间 */ /* 如果有使用SO_KEEPALIVE选项，连接处于非监听非结束的状态。 * 这个时候保活定时器已经在计时了，这里设置新的超时时间。 */ if (sock_flag(sk, SOCK_KEEPOPEN) && !((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN))) { u32 elapsed = keepalive_time_elapsed(tp); /* 连接已经经历的空闲时间 */ if (tp->keepalive_time > elapsed) elapsed = tp->keepalive_time - elapsed; /* 接着等待的时间，然后超时 */ else elapsed = 0; /* 会导致马上超时 */ inet_csk_reset_keepalive_timer(sk, elapsed); } } break; case TCP_KEEPINTVL: if (val < 1 || val > MAX_TCP_KEEPINTVL) err = -EINVAL; else tp->keepalive_intvl = val * HZ; /* 设置新的探测报文间隔 */ break; case TCP_KEEPCNT: if (val < 1 || val > MAX_TCP_KEEPCNT) err = -EINVAL; else tp->keepalive_probes = val; /* 设置新的探测次数 */ break; ...}
到目前为止，连接已经经历的空闲时间，即最后一次接收到报文至今的时间。
static inline u32 keepalive_time_elapsed (const struct tcp_sock *tp)
{
const struct inet_connection_sock *icsk = &tp->inet_conn;
/* lrcvtime是最后一次接收到数据报的时间
* rcv_tstamp是最后一次接收到ACK的时间
* 返回值就是最后一次接收到报文，到现在的时间，即经历的空闲时间。
*/
return min_t(u32, tcp_time_stamp - icsk->icsk_ack.lrcvtime,
tcp_time_stamp - tp->rcv_tstamp);
}
超时处理函数
我们知道保活定时器、SYNACK重传定时器、FIN_WAIT2定时器是共用一个定时器实例sk->sk_timer，
所以它们的超时处理函数也是一样的，都为tcp_keepalive_timer()。
而在函数内部，可以根据此时连接所处的状态，来判断是哪个定时器触发了超时。
Q：什么时候判断对端为异常并关闭连接？
A：分两种情况。
1. 用户使用了TCP_USER_TIMEOUT选项。当连接的空闲时间超过了用户设置的时间，且有发送过探测报文。
2. 用户没有使用TCP_USER_TIMEOUT选项。当发送保活探测包的次数达到了保活探测的最大次数时。
static void tcp_keepalive_timer (unsigned long data)
{
struct sock *sk = (struct sock *) data;
struct inet_connection_sock *icsk = inet_csk(sk);
struct tcp_sock *tp = tcp_sk(sk);
u32 elapsed;
/* Only process if socket is not in use. */
bh_lock_sock(sk);
/* 加锁以保证在此期间，连接状态不会被用户进程修改。
* 如果用户进程正在使用此sock，那么过50ms再来看看。
*/
if (sock_owned_by_user(sk)) {
/* Try again later. */
inet_csk_reset_keepalive_timer(sk, HZ/20);
goto out;
}
/* 三次握手期间，用作SYNACK定时器 */
if (sk->sk_state == TCP_LISTEN) {
tcp_synack_timer(sk);
goto out;
}
/* 连接释放期间，用作FIN_WAIT2定时器 */
if (sk->sk_state == TCP_FIN_WAIT2 && sock_flag(sk, SOCK_DEAD)) {
...
}
/* 接下来就是用作保活定时器了 */
if (!sock_flag(sk, SOCK_KEEPOPEN) || sk->sk_state == TCP_CLOSE)
goto out;
elapsed = keepalive_time_when(tp); /* 连接的空闲时间超过此值，就发送保活探测报文 */
/* It is alive without keepalive.
* 如果网络中有发送且未确认的数据包，或者发送队列不为空，说明连接不是idle的？
* 既然连接不是idle的，就没有必要探测对端是否正常。
* 保活定时器重新开始计时即可。
*
* 而实际上当网络中有发送且未确认的数据包时，对端也可能会发生异常而没有响应。
* 这个时候会导致数据包的不断重传，只能依靠重传超过了允许的最大时间，来判断连接超时。
* 为了解决这一问题，引入了TCP_USER_TIMEOUT，允许用户指定超时时间，可见下文：）
*/
if (tp->packets_out || tcp_send_head(sk))
goto resched; /* 保活定时器重新开始计时 */
/* 连接经历的空闲时间，即上次收到报文至今的时间 */
elapsed = keepalive_time_elapsed(tp);
/* 如果连接空闲的时间超过了设置的时间值 */
if (elapsed >= keepalive_time_when(tp)) {
/* 什么时候关闭连接？
* 1. 使用了TCP_USER_TIMEOUT选项。当连接空闲时间超过了用户设置的时间，且有发送过探测报文。
* 2. 用户没有使用选项。当发送的保活探测包达到了保活探测的最大次数。
*/
if (icsk->icsk_user_timeout != 0 && elapsed >= icsk->icsk_user_timeout &&
icsk->icsk_probes_out > 0) || (icsk->icsk_user_timeout == 0 &&
icsk->icsk_probes_out >= keepalive_probes(tp))) {
tcp_send_active_reset(sk, GFP_ATOMIC); /* 构造一个RST包并发送 */
tcp_write_err(sk); /* 报告错误，关闭连接 */
goto out;
}
/* 如果还不到关闭连接的时候，就继续发送保活探测包 */
if (tcp_write_wakeup(sk) <= 0) {
icsk->icsk_probes_out++; /* 已发送的保活探测包个数 */
elapsed = keepalive_intvl_when(tp); /* 下次超时的时间，默认为75s */
} else {
/* If keepalive was lost due to local congestion, try harder. */
elapsd = TCP_RESOURCE_PROBE_INTERVAL; /* 默认为500ms，会使超时更加频繁 */
}
} else {
/* 如果连接的空闲时间，还没有超过设定值，则接着等待 */
elapsed = keepalive_time_when(tp) - elapsed;
}
sk_mem_reclaim(sk);
resched: /* 重设保活定时器 */
inet_csk_reset_keepalive_timer(sk, elapsed);
goto out;
out:
bh_unlock_sock(sk);
sock_put(sk);
}
Q：TCP是如何发送Keepalive探测报文的？
A：分两种情况。
1. 有新的数据段可供发送，且对端接收窗口还没被塞满。发送新的数据段，来作为探测包。
2. 没有新的数据段可供发送，或者对端的接收窗口满了。发送序号为snd_una - 1、长度为0的ACK包作为探测包。
/* Initiate keepalive or window probe from timer. */
int tcp_write_wakeup (struct sock *sk)
{
struct tcp_sock *tp = tcp_sk(sk);
struct sk_buff *skb;
if (sk->sk_state == TCP_CLOSE)
return -1;
/* 如果还有未发送过的数据包，并且对端的接收窗口还没有满 */
if ((skb = tcp_send_head(sk)) != NULL && before(TCP_SKB_CB(skb)->seq, tcp_wnd_end(tp))) {
int err;
unsigned int mss = tcp_current_mss(sk); /* 当前的MSS */
/* 对端接收窗口所允许的最大报文长度 */
unsigned int seg_size = tcp_wnd_end(tp) - TCP_SKB_CB(skb)->seq;
/* pushed_seq记录发送出去的最后一个字节的序号 */
if (before(tp->pushed_seq, TCP_SKB_CB(skb)->end_seq))
tp->pushed_seq = TCP_SKB_CB(skb)->end_seq;
/* 如果对端接收窗口小于此数据段的长度，或者此数据段的长度超过了MSS，那么就要进行分段 */
if (seg_size < TCP_SKB_CB(skb)->end_seq - TCP_SKB_CB(skb)->seq || skb->len > mss) {
seg_size = min(seg_size, mss);
TCP_SKB_CB(skb)->tcp_flags |= TCPHDR_PSH; /* 设置PSH标志，让对端马上把数据提交给程序 */
if (tcp_fragment(sk, skb, seg_size, mss)) /* 进行分段 */
return -1;
} else if (! tcp_skb_pcount(skb)) /* 进行TSO分片 */
tcp_set_skb_tso_segs(sk, skb, mss); /* 初始化分片相关变量 */
TCP_SKB_CB(skb)->tcp_flags |= TCPHDR_PSH;
TCP_SKB_CB(skb)->when = tcp_time_stamp;
err = tcp_transmit_skb(sk, skb, 1, GFP_ATOMIC); /* 发送此数据段 */
if (!err)
tcp_event_new_data_sent(sk, skb); /* 发送了新的数据，更新相关参数 */
} else { /* 如果没有新的数据段可用作探测报文发送，或者对端的接收窗口为0 */
/* 处于紧急模式时，额外发送一个序号为snd_una的ACK包，告诉对端紧急指针 */
if (between(tp->snd_up, tp->snd_una + 1, tp->snd_una + 0xFFFF))
tcp_xmit_probe_skb(sk, 1);
/* 发送一个序号为snd_una -1的ACK包，长度为0，这是一个序号过时的报文。
* snd_una: first byte we want an ack for，所以snd_una - 1序号的字节已经被确认过了。
* 对端会响应一个ACK。
*/
return tcp_xmit_probe_skb(sk, 0);
}
}
Q：当没有新的数据可以用作探测包、或者对端的接收窗口为0时，怎么办呢？
A：发送一个序号为snd_una - 1、长度为0的ACK包，对端收到此包后会发送一个ACK响应。
如此一来本端就能够知道对端是否还活着、接收窗口是否打开了。
/* This routine sends a packet with an out of date sequence number.
* It assumes the other end will try to ack it.
*
* Question: what should we make while urgent mode?
* 4.4BSD forces sending single byte of data. We cannot send out of window
* data, because we have SND.NXT == SND.MAX...
*
* Current solution: to send TWO zero-length segments in urgent mode:
* one is with SEG.SEG=SND.UNA to deliver urgent pointer, another is out-of-date with
* SND.UNA - 1 to probe window.
*/
static int tcp_xmit_probe_skb (struct sock *sk, int urgent)
{
struct tcp_sock *tp = tcp_sk(sk);
struct sk_buff *skb;
/* We don't queue it, tcp_transmit_skb() sets ownership. */
skb = alloc_skb(MAX_TCP_HEADER, sk_gfp_atomic(sk, GFP_ATOMIC));
if (skb == NULL)
return -1;
/* Reserve space for headers and set control bits. */
skb_reserve(skb, MAX_TCP_HEADER);
/* Use a previous sequence. This should cause the other end to send an ack.
* Don't queue or clone SKB, just send it.
*/
/* 如果没有设置紧急指针，那么发送的序号为snd_una - 1，否则发送的序号为snd_una */
tcp_init_nondata_skb(skb, tp->snd_una - !urgent, TCPHDR_ACK);
TCP_SKB_CB(skb)->when = tcp_time_stamp;
return tcp_transmit_skb(sk, skb, 0, GFP_ATOMIC); /* 发送探测包 */
}
发送RST包。
/* We get here when a process closes a file descriptor (either due to an explicit close()
* or as a byproduct of exit()'ing) and there was unread data in the receive queue.
* This behavior is recommended by RFC 2525, section 2.17. -DaveM
*/
void tcp_send_active_reset (struct sock *sk, gfp_t priority)
{
struct sk_buff *skb;
/* NOTE: No TCP options attached and we never retransmit this. */
skb = alloc_skb(MAX_TCP_HEADER, priority);
if (!skb) {
NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTFAILED);
return;
}
/* Reserve space for headers and prepare control bits. */
skb_reserve(skb, MAX_TCP_HEADER); /* 为报文头部预留空间 */
/* 初始化不携带数据的skb的一些控制字段 */
tcp_init_nondata_skb(skb, tcp_acceptable_seq(sk), TCPHDR_ACK | TCPHDR_RST);
/* Send if off，发送此RST包*/
TCP_SKB_CB(skb)->when = tcp_time_stamp;
if (tcp_transmit_skb(sk, skb, 0, priority))
NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTFAILED);
TCP_INC_STATS(sock_net(sk), TCP_MIB_OUTRSTS);
}
static inline __u32 tcp_acceptable_seq (const struct sock *sk)
{
const struct tcp_sock *tp = tcp_sk(sk);
/* 如果snd_nxt在对端接收窗口范围内 */
if (! before(tcp_wnd_end(tp), tp->snd_nxt))
return tp->snd_nxt;
else
return tcp_wnd_end(tp);
}
TCP_USER_TIMEOUT选项
从上文可知同时符合以下条件时，保活定时器才会发送探测报文：
1. 网络中没有发送且未确认的数据包。
2. 发送队列为空。
3. 连接的空闲时间超过了设定的时间。
Q：如果网络中有发送且未确认的数据包、或者发送队列不为空时，保活定时器不起作用了，
岂不是不能够检测到对端的异常了？
A：可以使用TCP_USER_TIMEOUT，显式的指定当发送数据多久后还没有得到响应，就判定连接超时，
从而主动关闭连接。
TCP_USER_TIMEOUT选项会影响到超时重传定时器和保活定时器。
(1) 超时重传定时器
判断连接是否超时，分3种情况：
1. SYN包：当SYN包的重传次数达到上限时，判定连接超时。(默认允许重传5次，初始超时时间为1s，总共历时31s)
2. 非SYN包，用户使用TCP_USER_TIMEOUT：当数据包发出去后的等待时间超过用户设置的时间时，判定连接超时。
3. 非SYN包，用户没有使用TCP_USER_TIMEOUT：当数据包发出去后的等待时间超过以TCP_RTO_MIN为初始超时
时间，重传boundary次所花费的时间后，判定连接超时。(boundary的最大值为tcp_retries2，默认值为15)
(2) 保活定时器
判断连接是否异常，分2种情况：
1. 用户使用了TCP_USER_TIMEOUT选项。当连接的空闲时间超过了用户设置的时间，且有发送过探测报文。
2. 用户没有使用TCP_USER_TIMEOUT选项。当发送保活探测包的次数达到了保活探测的最大次数时。