TCP的核心系列 — SACK和DSACK的实现(一)

时间:2023-03-08 16:15:20

TCP的实现中,SACK和DSACK是比较重要的一部分。

SACK和DSACK的处理部分由Ilpo Järvinen (ilpo.jarvinen@helsinki.fi) 维护。

tcp_ack()处理接收到的带有ACK标志的数据段时,如果此ACK处于慢速路径,且此ACK的记分牌不为空,则调用

tcp_sacktag_write_queue()来根据SACK选项标记发送队列中skb的记分牌状态。

笔者主要分析18和37这两个版本的实现。

相对而言,18版本的逻辑清晰,但效率较低;37版本的逻辑复杂,但效率较高。

本文主要内容:18版tcp_sacktag_write_queue()的实现,也即18版SACK和DSACK的实现。

Author:zhangskd @ ****

18版数据结构

/* 这就是一个SACK块 */
struct tcp_sack_block {
u32 start_seq; /* 起始序号 */
u32 end_seq; /* 结束序号 */
};
struct tcp_sock {
...
/* Options received (usually on last packet, some only on SYN packets). */
struct tcp_options_received rx_opt;
...
struct tcp_sack_block recv_sack_cache[4]; /* 保存收到的SACK块,用于提高效率*/
...
/* 快速路径中使用,上次第一个SACK块的结束处,现在直接从这里开始处理 */
struct sk_buff *fastpath_skb_hint;
int fastpath_cnt_hint; /* 快速路径中使用,上次记录的fack_count,现在继续累加 */
... };
struct tcp_options_received {
...
u16 saw_tstamp : 1, /* Saw TIMESTAMP on last packet */
tstamp_ok : 1, /* TIMESTAMP seen on SYN packet */
dsack : 1, /* D-SACK is scheduled, 下一个发送段是否存在D-SACK */
sack_ok : 4, /* SACK seen on SYN packet, 接收方是否支持SACK */
...
u8 num_sacks; /* Number of SACK blocks, 下一个发送段中SACK块数 */
...
};

18版本实现

18版本的逻辑较清晰,我们先来看看。

static int tcp_sacktag_write_queue(struct sock *sk, struct sk_buff *ack_skb, u32 prior_snd_una)
{
const struct inet_connection_sock *icsk = inet_csk(sk);
struct tcp_sock *tp = tcp_sk(sk); /* SACK选项的起始地址,sacked为SACK选项在TCP首部的偏移 */
unsigned char *ptr = ack_skb->h.raw + TCP_SKB_CB(ack_skb)->sacked; struct tcp_sack_block *sp = (struct tcp_sack_block *) (ptr + 2); /* 指向第一个sack块 */
int num_sacks = (ptr[1] - TCPOLEN_SACK_BASE) >> 3; /* sack的块数 */ int reord = tp->packets_out; /* 乱序的起始包位置,一开始设为最大 */
int prior_fackets; /* 上次的fackets_out */
u32 lost_retrans = 0; /* 重传包可能丢失时SACK块结束序号,表示需要遍历到的最高序号 */
int flag = 0; /* 有两种用途:先表示是否为快速路径,后用于返回标志 */
int dup_sack = 0; /* 有没有DSACK */
int i; /* 如果之前没有SACKed的数据 */
if (! tp->sacked_out)
tp->fackets_out = 0; /* FACK是根据最新的SACK来计算的,所以也要为0 */
prior_fackets = tp->fackets_out; /* 处理前先保存上次的fackets_out */ /* SACK fastpath:
* if the only SACK change is the increase of the end_seq of the first block then only
* apply that SACK block and use retrans queue hinting otherwise slowpath.
* 什么是快速路径:就是只有第一个SACK块的结束序号发生变化,其它的都不变。
*/
flag = 1; /* 为1的话为快速路径,0为慢速路径 */ for (i = 0; i < num_sacks; i++) {
__u32 start_seq = ntohl(sp[i].start_seq); /* 块的起始序号 */
__u32 end_seq = ntohl(sp[i].end_seq); /* 块的结束序号 */ /* 判断是否进入快速路径。
* 对第一个块:只要求起始序号相同
* 对于非第一个块:要求起始序号和结束序号都相同
* 也就是说,快速路径指的是只有第一个块的结束序号增加的情况
*/
if (i == 0) {
if (tp->recv_sack_cache[i].start_seq != start_seq)
flag = 0; } else {
if ((tp->recv_sack_cache[i].start_seq != start_seq) ||
(tp->recv_sack_cache[i].end_seq != end_seq))
flag = 0;
} /* 更新,保存这次收到的SACK块 */
tp->recv_sack_cache[i].start_seq = start_seq;
tp->recv_sack_cache[i].end_seq = end_seq; /* Check for D-SACK.
* 检测是否有DSACK ,DSACK块如果有,只能在第一个块
*/
if (i == 0) {
u32 ack = TCP_SKB_CB(ack_skb)->ack_seq; /* 如果第一个SACK块的起始序号小于它的确认序号,说明此SACK块包含了确认过的数据 */
if (before(start_seq, ack)) {
dup_sack = 1;
tp->rx_opt.sack_ok |= 4;
NET_INC_STATS_BH(LINUX_MIB_TCPDSACKRECV); /* 如果第一个SACK块包含在第二个SACK块中,也说明第一个SACK块是重复的,即DSACK */
} else if (num_sacks > 1 &&
!after(end_seq, ntohl(sp[1].end_seq)) &&
!before(start_seq, ntohl(sp[1].start_seq))) {
dup_sack = 1;
tp->rx_opt.sack_ok |= 4;
NET_INC_STATS_BH(LINUX_MIB_TCPDSACKOFORECV);
}
} /* D-SACK for already forgotten data...
* Do dumb counting.
* undo_retrans记录重传数据包的个数,如果undo_retrans降到0,
* 就说明之前的重传都是不必要的,进行拥塞调整撤销。
* 条件:DSACK、undo_marker < end_seq <= prior_snd_una
*/
if (dup_sack && !after(end_seq, prior_snd_una) &&
after(end_seq, tp->undo_marker))
tp->undo_retrans--; /* Eliminate too old ACKs, but take into account more or less fresh ones,
* they can contain valid SACK info.
* tp->max_window为接收方通告过的最大接收窗口。
* 如果SACK信息是很早以前的,直接丢弃。
*/
if (before(ack, prior_snd_una - tp->max_window))
return 0;
} if (flag)
num_sacks = 1; /* 快速路径时只有第一个块有变化,处理第一个块即可 */
else {
int j;
/* 上次第一个SACK块的结束处,也是这次快速路径的开始点,慢速路径中重置了 */
tp->fastpath_skb_hint = NULL; /* order SACK blocks to allow in order walk of the retrans queue.
* 对SACK块按起始序号,从小到大冒泡排序,以便与接下来的顺序遍历。
*/
for (i = num_sacks - 1; i > 0; i--) {
for (j = 0; j < i; j++) {
if (after(ntohl(sp[j].start_seq), ntohl(sp[j+1].start_seq))) {
sp[j].start_seq = htonl(tp->recv_sack_cache[j+1].start_seq);
sp[j].end_seq = htonl(tp->recv_sack_cache[j+1].end_seq);
sp[j+1].start_seq = htonl(tp->recv_sack_cache[j].start_seq);
sp[j+1].end_seq = htonl(tp->recv_sack_cache[j].end_seq);
}
}
}
} /* clear flag as used for different purpose in following code */
flag = 0; /* 用于返回一些标志 */ /* 逐个处理SACK块,可能只有一个,也可能多个 */
for (i = 0; i < num_sacks; i++, sp++) {
struct sk_buff *skb;
__u32 start_seq = ntohl(sp->start_seq); /* SACK块起始序号 */
__u32 end_seq = ntohl(sp->end_seq); /* SACK块结束序号 */
int fack_count; /* 用于更新fackets_out */ /* Use SACK fastpath hint if valid.
* 如果处于快速路径,那么可以不用从头遍历发送队列。
*/
if (tp->fastpath_skb_hint) {
skb = tp->fastpath_skb_hint; /* 从这个段开始处理 */
fack_count = tp->fastpath_cnt_hint; /* 已有的fackets_out */ } else { /* 否则慢速路径,从头开始处理 */
skb = sk->sk_write_queue.next; /* 发送队列头 */
fack_count = 0;
} /* Event B in the comment above.
* high_seq是进入Recovery或Loss时的snd_nxt,如果high_seq被SACK了,那么很可能有数据包
* 丢失了,不然就可以ACK掉high_seq返回Open态了。
*/
if (after(end_seq, tp->high_seq))
flag |= FLAG_DATA_LOST; /* 从skb开始遍历发送队列 */
sk_stream_for_retrans_queue_from(skb, sk) {
int in_sack, pcount;
u8 sacked; /* 记录最后一个正在处理的段,下次进入快速路径时,可以直接从这里
* 开始处理,而不用从头遍历发送队列。
*/
tp->fastpath_skb_hint = skb;
tp->fastpath_cnt_hint = fack_count; /* The retransmission queue is always in order, so we can short-circuit
* the walk early.
* 当前skb段的序号超过SACK块的右端时,说明这个SACK块已经处理好了。
*/
if (! before(TCP_SKB_CB(skb)->seq, end_seq))
break; /* 这个段是否完全包含在SACK块中 */
in_sack = ! after(start_seq, TCP_SKB_CB(skb)->seq) &&
! before(end_seq, TCP_SKB_CB(skb)->end_seq);
pcount = tcp_skb_pcount(skb); /* 这个段分为多少个包 */ /* 如果当前的段是TSO段,且它的一部份包含在SACK块中。
* 那么那些已经被SACK的部分就不用再重传了,所以需要重新分割TSO段。
*/
if (pcount > 1 && ! in_sack &&
after(TCP_SKB_CB(skb)->end_seq, start_seq)) {
unsigned int pkt_len; /* 表示TSO段的后半部在SACK块之外 */
in_sack = ! after(start_seq, TCP_SKB_CB(skb)->seq); if (! in_sack) /* 如果TSO段的前半部在SACK块之外 */
pkt_len = (start_seq - TCP_SKB_CB(skb)->seq); /* SACK块之外段的长度 */
else
pkt_len = (end_seq - TCP_SKB_CB(skb)->seq); /* SACK块之内段的长度 */ /* 把TSO段分为两部分 */
if (tcp_fragment(sk, skb, pkt_len, skb_shinfo(skb)->gso_size))
break; pcount += tcp_skb_pcount(skb); /* skb缩减了,需要重新计算 */
} fack_count += pcount; /* 累加fackets_out */ sacked = TCP_SKB_CB(skb)->sacked; /* 这就是记分板scoreboard */ /* Account D-SACK for retransmitted packet.
* 如果此skb属于DSACK块,且skb被重传过。
* 这里in_sack指的是:全部包含在SACK块中,还有前半部包含也算,因为分割了:)
*/
if ((dup_sack && in_sack) && (sacked & TCPCB_RETRANS) &&
after(TCP_SKB_CB(skb)->end_seq, tp->undo_marker))
tp->undo_retrans--; /* 如果减为0,那么说明之前重传都是不必要的,进行拥塞控制调整撤销 */ /* The frame is ACKed. 当这个skb被确认了*/
if (! after(TCP_SKB_CB(skb)->end_seq, tp->snd_una)) {
/* 乱序情况1:R|S标志,收到DSACK */
if (sacked & TCPCB_RETRANS) {
if ((dup_sack && in_sack) && (sacked & TCPCB_SACKED_ACKED))
reord = min(fack_count, reord); /* 更新乱序的起始位置 */ } else {
/* 乱序情况2:一个包落在highest_sack之前,它既没被SACK过,也不是重传的,
* 现在才到达了,那么它就是乱序了。就是前面的洞自动填满了:)
*/
if (fack_count < prior_fackets && ! (sacked & TCPCB_SACKED_ACKED))
reord = min(fack_count, reord);
} /* Nothing to do; acked frame is about to be dropped.
* 这个skb已经被正常确认了,不用再处理了,它即将被丢弃。
*/
continue;
} /* 如果这个包是重传包,并且它的snd_nxt小于此块的结束序号,
* 那么这个重传包可能是丢失了,我们记录这个块的结束序号,
* 作为接下来遍历的最高序号。
*/
if ((sacked & TCPCB_SACKED_RETRANS) &&
after(end_seq, TCP_SKB_CB(skb)->ack_seq) &&
(! lost_retrans || after(end_seq, lost_retrans)))
lost_retrans = end_seq; /* 如果这个包不包含在SACK块中,即在SACK块之外,则不用继续处理 */
if (! in_sack)
continue; /* 如果skb还没有被标志为SACK,那么进行处理 */
if (! (sacked & TCPCB_SACKED_ACKED)) {
/* 有R标志,表示被重传过 */
if (sacked & TCPCB_SACKED_RETRANS) {
/* If the segment is not tagged as lost, we do not clear RETRANS, believing
* that retransmission is still in flight.
* 如果之前的标志是:R | L,那么好,现在收到包了,可以清除R和L。
* 如果之前的标志是:R,那么认为现在收到的是orig,重传包还在路上,所以不用干活:)
*/
if (sacked & TCPCB_LOST) {
TCP_SKB_CB(skb)->sacked &= ~(TCPCB_LOST | TCPCB_SACKED_RETRANS); /* 取消L和R标志 */
tp->lost_out -= tcp_skb_pcount(skb); /* 更新LOST包个数 */
tp->retrans_out -= tcp_skb_pcount(skb); /* 更新RETRANS包个数 */
/* clear lost hint */
tp->retransmit_skb_hint = NULL;
} } else {
/* New sack for not retransmitted frame, which was in hole. It is reordering.
* 如果一个包落在highest_sack之前,它即没被SACK过,也不是重传的,那么
* 它肯定是乱序了,到现在才被SACK。
*/
if (! (sacked & TCPCB_RETRANS) && fack_count < prior_fackets)
reord = min(fack_count, reord); /* 记录乱序的起始 */ /* 如果有L标志 */
if (sacked & TCPCB_LOST) {
TCP_SKB_CB(skb)->sacked &= ~TCPCB_LOST; /* 清除L标志 */
tp->lost_out -= tcp_skb_pcount(skb); /* 更新lost_out */
/* clear lost hint */
tp->retransmit_skb_hint = NULL;
}
} TCP_SKB_CB(skb)->sacked |= TCPCB_SACKED_ACKED; /* 打上S标志 */
flag |= FLAG_DATA_SACKED; /* New SACK */
tp->sacked_out += tcp_skb_pcount(skb); /* 更新sacked_out */ if (fack_count > tp->fackets_out)
tp->fackets_out = fack_count; /* 更新fackets_out */ } else { /* 已经有S标志 */
/* 如果之前是R|S标志,且这个包被DSACK了,说明是乱序 */
if (dup_sack && (sacked & TCPCB_RETRANS))
reord = min(fack_count, reord);
} /* D-SACK. We can detect redundant retransmission in S|R and plain R frames
* and clear it.
* undo_retrans is decreased above, L|R frames are accounted above as well.
* 如果skb被D-SACK,并且它的重传标志还未被清除,那么现在清除。
*/
if (dup_sack && (TCP_SKB_CB(skb)->sacked & TCPCB_SACKED_RETRANS)) {
TCP_SKB_CB(skb)->sacked &= ~TCPCB_SACKED_RETRANS;
tp->retrans_out -= tcp_skb_pcount(skb);
tp->retransmit_skb_hint = NULL;
}
}
} /* Check for lost retransmit. This superb idea is borrowed from "ratehalving." Event C.
* 如果lost_retrans不为0,且处于Recovery状态,说明有重传包丢失,进行处理。
*/
if (lost_retrans && icsk->icsk_ca_state == TCP_CA_Recovery) {
struct sk_buff *skb; /* 从头开始遍历发送队列 */
sk_stream_for_retrans_queue(skb, sk) {
/* lost_retrans记录的是SACK块结束序号,并且只在小于lost_retrans内有发现重传包丢失 */
if (after(TCP_SKB_CB(skb)->seq, lost_retrans))
break; /* 不关心成功确认过的包 */
if (! after(TCP_SKB_CB(skb)->end_seq, tp->snd_una)
continue; /* 现在判断这个重传包是否丢失。
* 这个包要是重传包,并且它的snd_nxt小于lost_retrans
*/
if ((TCP_SKB_CB(skb)->sacked & TCPCB_SACKED_RETRANS) &&
after(lost_retrans, TCP_SKB_CB(skb)->ack_seq) && (IsFack(tp) ||
!before(lost_retrans, TCP_SKB_CB(skb)->ack_seq + tp->reordering * tp->mss_cache))) {
TCP_SKB_CB(skb)->sacked &= ~TCPCB_SACKED_RETRANS; /* 清除R标志 */
tp->retrans_out -= tcp_skb_pcount(skb); /* 更新retrans_out */
/* clear lost hint */
tp->retransmit_skb_hint = NULL; /* 给这个包重新打上L标志 */
if (! (TCP_SKB_CB(skb)->sacked & (TCPCB_LOST | TCPCB_SACKED_ACKED))) {
tp->lost_out += tcp_skb_pcount(skb); /* 更新lost_out */
TCP_SKB_CB(skb)->sacked |= TCPCB_LOST; /* 打上L标志 */
/* 这个弄错了吧?应该是FLAG_DATA_LOST才对 */
flag |= FLAG_DATA_SACKED;
NET_INC_STATS_BH(LINUX_MIB_TCPLOSTRETRANSMIT);
}
}
}
} tp->left_out = tp->sacked_out + tp->lost_out;
/* 更新乱序队列长度。
* 乱序队列的长度 = fackets_out - reord + 1,reord记录从第几个包开始乱序
*/
if ((reord < tp->fackets_out) && icsk->icsk_ca_state != TCP_CA_Loss)
tcp_update_reordering(sk, ((tp->fackets_out + 1) - reord), 0); #if FASTRETRANS_DEBUG > 0
BUG_TRAP((int) tp->sacked_out >= 0);
BUG_TRAP((int) tp->lost_out >= 0);
BUG_TRAP((int) tp->retrans_out >= 0);
BUG_TRAP((int) tcp_packets_in_flight(tp) >= 0);
#endif return flag;
}

Q: 为什么说18版的实现效率不高呢?

A: 我们收到num_sacks个SACK块,如果符合快速路径,那么遍历一次发送队列就可以了;

但是如果不符合快速路径,那么对于每个SACK块,都要遍历一次发送队列,而且都是从头开始遍历,

这样就做了很多重复工作,复杂度为O(num_sacks * cwnd)。如果cwnd很大的话,CPU消耗会较高。

37版本在这一方面做了一些优化。

对于18版本中的一些细节,接下来会对照37版本的实现进行详细分析,比如:

SACK选项的地址在接收时是如何保存起来的,这是在tcp_rcv_established中处理的。

DSACK的原理和实现,这部分在37中独立出来。

检测重传包是否丢失的原理和实现,这部分在37中独立出来。

乱序是如何检测的,它的原理和实现。

Reference

RFC 2018

RFC 2883