TCP连接的终止----被动关闭

  这篇文章是对 《TCP连接的终止----主动关闭》的补充,重点关注的是被动关闭连接一端的状态迁移及内核中的 处理,被动连接的状态迁移可以参见 《TCP连接的终止----主动关闭》中的状态迁移图,这里就不再画了
被动连接的关闭是从接收到FIN开始的,如果TCP是处于ESTABLISHED状态(我们的讨论假设连接处于此状态),这个FIN包会在tcp_rcv_established()中处理。在tcp_rcv_state_process()中的处理分为快速路径和慢速路径。如果TCP首部中第4个32位字除去保留的bit位和预测标志一致,skb包的序列号和sock结构下一个要接收到序号相等,并且skb包中的确认序列号是有效的,此时skb包会在快速路径中处理。判断时sock实例的预测标志的位图分布通常如下图所示:
其中S对应于TCP首部(struct tcphdr)中的doff成员(tcp首部的长度,以4字节为单位),?通常为0,为1的bit位对应的是ACK标志,snd_wnd则是本端发送窗口的大小。
从上面的预测标志的分布来看,如果设置了FIN标志的话,则检查预测标志时失败,所以会在慢速路径中处理FIN包。
在慢速路径处理中,会首先检查skb包的校验和及包是否是有效的,检查通过后会调用tcp_ack()处理ack的情况,接下来的处理中真正和FIN相关的操作是在调用的tcp_data_queue()中处理的,这个处理是通过调用tcp_fin()函数来处理的,从最初的TCP层接收函数tcp_v4_rcv()到tcp_fin()的处理的代码流程图如下所示:
TCP连接的终止----被动关闭_第1张图片
tcp_fin()中首先调用inet_csk_schedule_ack()设置相关成员,表明需要发送ACK;因为接收了对端发送的FIN包,表明对端已经不会再发送数据包(可以发送ACK),因此关闭接收通道;还要修改sock结构的标志,表示连接将要结束;然后根据sock实例的状态会跳转到不同的分支进行处理,如果是ESTABLISHED状态下,内核会调用tcp_set_state()修改sock实例的状态,并且设置延迟发送ACK的标志,如下所示:
static void tcp_fin(struct sk_buff *skb, struct sock *sk, struct tcphdr *th)
{
    struct tcp_sock *tp = tcp_sk(sk);

    inet_csk_schedule_ack(sk);

    sk->sk_shutdown |= RCV_SHUTDOWN;
    sock_set_flag(sk, SOCK_DONE);

    switch (sk->sk_state) {
    case TCP_SYN_RECV:
    case TCP_ESTABLISHED:
        /* Move to CLOSE_WAIT */
        tcp_set_state(sk, TCP_CLOSE_WAIT);
        inet_csk(sk)->icsk_ack.pingpong = 1;
        break;

    ......
    }
}
tcp_fin()后面的清理乱序队列、状态更改时可能要唤醒相关进程这些操作不是我们关心的,就不作过多说明了。
在tcp_fin()中虽然设置了发送ACK的相关标志,但是要有一个引发ACK发送的操作,或者是给内核发送ACK的一个提示。这个操作是在上层函数tcp_rcv_established()函数中进行的,通过间接调用__tcp_ack_snd_check()中完成。__tcp_ack_snd_check()中会判断当前发送ACK是要立即发送还是延迟发送,如果立即发送则调用tcp_send_ack()来发送ACK,否则调用tcp_send_delayed_ack()延迟发送,代码如下所示:
static void __tcp_ack_snd_check(struct sock *sk, int ofo_possible)
{
    struct tcp_sock *tp = tcp_sk(sk);

        /* More than one full frame received... */
    if (((tp->rcv_nxt - tp->rcv_wup) > inet_csk(sk)->icsk_ack.rcv_mss
         /* ... and right edge of window advances far enough.
          * (tcp_recvmsg() will send ACK otherwise). Or...
          */
         && __tcp_select_window(sk) >= tp->rcv_wnd) ||
        /* We ACK each frame or... */
        tcp_in_quickack_mode(sk) ||
        /* We have out of order data. */
        (ofo_possible && skb_peek(&tp->out_of_order_queue))) {
        /* Then ack it now */
        tcp_send_ack(sk);
    } else {
        /* Else, send delayed ack. */
        tcp_send_delayed_ack(sk);
    }
}
判断的条件是这样的,只要满足以下条件就会立即发送ACK:
1、接收窗口中有多个全尺寸段还未确认
2、当前处于快速确认模式下
3、在启用判断乱序队列的情况下,乱序队列中存在段
这三个条件中,我们可以确定的是第2个条件,我们首先来看判断是否处于快速确认模式的函数tcp_in_quickack_mod()函数的实现,如下所示:
static inline int tcp_in_quickack_mode(const struct sock *sk)
{
    const struct inet_connection_sock *icsk = inet_csk(sk);
    return icsk->icsk_ack.quick && !icsk->icsk_ack.pingpong;
}
在tcp_fin()中如果是ESTABLISHED状态下接收到FIN,会设置pingpong的值为1,所以可以肯定此时不处于快速确认模式下,至于是否要立即发送ACK取决于另外两个判断条件了。
我们接下来的讨论是在从对接收到的FIN确认后开始的。这时TCP连接的关闭已经进行一半了,接下来就是等待本端的上层应用调用close()来执行本端的关闭连接操作,这个操作我们在 《TCP连接的终止----主动关闭》中讲到过,是由tcp_close()来完成的。所以我们还是来看tcp_close(),只是这次sock实例的状态不一样,这时的状态应该为CLOSE_WAIT。
在tcp_close()中我们这次只关注一些和状态相关的一些处理,其他的队列清理、内存回收等就不再介绍了,所以我们只关注下面这部分代码:
void tcp_close(struct sock *sk, long timeout)
{
    struct sk_buff *skb;
    int data_was_unread = 0;
    int state;

    ......

    if (data_was_unread) {
        ......

    } else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
        ......

    } else if (tcp_close_state(sk)) {
        tcp_send_fin(sk);
    }

    ......
}
在tcp_close_state()中sock实例的状态会由CLOSE_WAIT迁移到LAST_ACK状态,返回值为TCP_ACTION_FIN,表示要发送FIN。因此,在第三个if判断中条件为true,所以会调用tcp_send_fin()给对端发送FIN。
当本端接收到TCP连接关闭的最后一个ACK时,由tcp_rcv_state_process()函数来处理,相关的代码如下所示:
int tcp_rcv_state_process(struct sock *sk, struct sk_buff *skb,
              struct tcphdr *th, unsigned len)
{
    struct tcp_sock *tp = tcp_sk(sk);
    struct inet_connection_sock *icsk = inet_csk(sk);
    int queued = 0;
    int res;

    ......

    /* step 5: check the ACK field */
    if (th->ack) {
        int acceptable = tcp_ack(sk, skb, FLAG_SLOWPATH) > 0;

        switch (sk->sk_state) {
        ......

        case TCP_LAST_ACK:
            if (tp->snd_una == tp->write_seq) {
                tcp_update_metrics(sk);
                tcp_done(sk);
                goto discard;
            }
            break;
        }
    } else
        goto discard;

    ......

    switch (sk->sk_state) {
    case TCP_CLOSE_WAIT:
    case TCP_CLOSING:
    case TCP_LAST_ACK:
        if (!before(TCP_SKB_CB(skb)->seq, tp->rcv_nxt))
            break;

    case TCP_FIN_WAIT1:
    case TCP_FIN_WAIT2:
        /* RFC 793 says to queue data in these states,
         * RFC 1122 says we MUST send a reset.
         * BSD 4.4 also does reset.
         */
        if (sk->sk_shutdown & RCV_SHUTDOWN) {
            if (TCP_SKB_CB(skb)->end_seq != TCP_SKB_CB(skb)->seq &&
                after(TCP_SKB_CB(skb)->end_seq - th->fin, tp->rcv_nxt)) {
                NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPABORTONDATA);
                tcp_reset(sk);
                return 1;
            }
        }
        /* Fall through */
    case TCP_ESTABLISHED:
        tcp_data_queue(sk, skb);
        queued = 1;
        break;
    }

    /* tcp_data could move socket to TIME-WAIT */
    if (sk->sk_state != TCP_CLOSE) {
        tcp_data_snd_check(sk);
        tcp_ack_snd_check(sk);
    }

    if (!queued) {
discard:
        __kfree_skb(skb);
    }
    return 0;
}
如果刚好是期望的ACK包,则会在第19-23行代码中处理,调用tcp_done()将套接字状态设置为TCP_CLOSE,并且调用inet_csk_destroy_sock()释放sock实例占用的资源,并且调用sock_put()释放传输控制块(真正的调用sk_free()一般情况下不会是这里,但是这里较少引用计数后,上层再调用sock_put()时就会触发sk_free()操作)。我们知道在主动关闭一端正常情况下会通过定时器来释放描述TIME_WAIT状态的sock结构或者放在twcal_row队列中等待释放,这些释放方式比较明显。还有一个种就是这里看到的通过inet_csk_destroy_sock()来间接完成释放。
有时也可能接收到其他包,如果是包含数据的包,在44-51的处理中会发送RST给对端,如果只是单纯的ACK包,但是确认的序列号不对,则会在tcp_data_queue()中释放掉。
至此,TCP连接被动关闭一方的处理完成了。

你可能感兴趣的:(tcp)