tcp_write_xmit
一、讲解
这个函数 tcp_write_xmit 是Linux内核TCP协议栈中的一部分,其基本作用是发送数据包到网络。这个函数会根据不同情况推进发送队列的头部,确保只要远程窗口有空间,就可以发送数据。
下面是对该函数的一些主要逻辑的中文解释:
1. 初始化:函数开头有一些初始化操作,比如用 tcp_mstamp_refresh(tp) 刷新时间戳,`tcp_mtu_probe(sk)` 尝试路径MTU发现过程(如果适用),以及 max_segs 的计算。
2. 循环发送:接下来,函数进入一个while循环,尝试发送所有已经排队的SKB(socket buffer)。循环的每一次迭代都会尝试发送一个SKB,直到没有更多可以发送的,或者遇到问题而中断。
3. 发送条件检查:在每次尝试发送前,会进行一系列条件检查:
- tcp_pacing_check(sk) 检查是否应该基于网络拥塞避免算法暂时停止发送。
- tcp_cwnd_test(tp, skb) 和 tcp_snd_wnd_test(tp, skb, mss_now) 分别检查拥塞窗口(cwnd)和发送窗口(snd_wnd),以确保我们没有发送超出对方TCP流控制和拥塞控制的数据。
4. 确定发送大小:对于要发送的SKB,它会计算出可以一次发送多少数据(TSO分段,即TCP段上合并发送),以及是否应该延迟发送,从而进行网络流量整形。
5. 发送和处理:如果所有条件均符合,就会通过 tcp_transmit_skb(sk, skb, 1, gfp) 将SKB发送到网络。发送后,会进行一些更新,比如更新拥塞窗口相关数据。
6. 结束条件:如果在发送过程中遇到资源限制(如拥塞窗口满了,或者接收窗口满了),就会跳出发送循环。
7. 后处理:函数的最后部分会基于发送情况更新一些计时器,比如记录流控制限制的时间,决定是否触发进一步的丢包探测等。
整个函数的设计关注于什么时候发送数据,以及如何基于当前网络条件(例如拥塞控制、窗口大小等)做出正确的发送决策。这是一个TCP协议中用于管理数据发送的核心路径,确保数据以有效和合理的方式在网络中传输。
二、中文注释
/* 这个函数用于将数据包写入网络,并推进发送队列头部。这个操作发生在
* 接收到的确认(ACK)扩展了远程窗口时。
*
* LARGESEND注释:!tcp_urg_mode是过度的限制,实际上,只有从snd_up-64k-mss到snd_up
* 这段范围内的帧不能是大型帧。考虑到紧急(URG)数据的使用比较少,这不是一个严重的问题。
*
* 当push_one > 0时,最多发送一个数据包。当push_one == 2时,暂时忽略拥塞窗口(cwnd)的限制,
* 强制至多发送一个数据包。
*
* 如果没有在传输中的分段(即所有分段都已确认),且我们有排队的分段,
* 但现在由于轻微的发送窗口(SWS)问题或其他问题而无法发送任何分段,
* 则返回true。
*/
static bool tcp_write_xmit(struct sock *sk, unsigned int mss_now, int nonagle,
int push_one, gfp_t gfp)
{
struct tcp_sock *tp = tcp_sk(sk); // 获取TCP套接字结构体
struct sk_buff *skb; // 定义一个套接字缓冲区指针
unsigned int tso_segs, sent_pkts; // 定义传输段的数量和已发送的数据包数
int cwnd_quota; // 定义拥塞窗口配额
int result; // 定义结果变量
bool is_cwnd_limited = false, is_rwnd_limited = false; // 定义拥塞窗口和接收窗口限制标记
u32 max_segs; // 定义最大段数
sent_pkts = 0; // 初始化已发送的数据包数量
tcp_mstamp_refresh(tp); // 刷新时间戳
if (!push_one) {
/* 执行MTU探测 */
result = tcp_mtu_probe(sk);
if (!result) {
return false;
} else if (result > 0) {
sent_pkts = 1;
}
}
max_segs = tcp_tso_segs(sk, mss_now); // 计算最大可发送的段数
while ((skb = tcp_send_head(sk))) { // 遍历发送队列
unsigned int limit; // 定义发送限制
if (tcp_pacing_check(sk)) // 执行流量整形检查
break;
tso_segs = tcp_init_tso_segs(skb, mss_now); // 初始化TSO段
BUG_ON(!tso_segs); // 如果tso_segs为0,触发BUG
if (unlikely(tp->repair) && tp->repair_queue == TCP_SEND_QUEUE) {
/* 将"skb_mstamp"用作重传计时器的起始点 */
tcp_update_skb_after_send(tp, skb);
goto repair; // 跳过网络传输
}
cwnd_quota = tcp_cwnd_test(tp, skb); // 检查拥塞窗口
if (!cwnd_quota) {
if (push_one == 2)
/* 发送一个丢失探针包 */
cwnd_quota = 1;
else
break;
}
if (unlikely(!tcp_snd_wnd_test(tp, skb, mss_now))) {
is_rwnd_limited = true; // 判断接收窗口是否限制了发送
break;
}
if (tso_segs == 1) {
if (unlikely(!tcp_nagle_test(tp, skb, mss_now,
(tcp_skb_is_last(sk, skb) ?
nonagle : TCP_NAGLE_PUSH))))
break;
} else {
if (!push_one &&
tcp_tso_should_defer(sk, skb, &is_cwnd_limited,
&is_rwnd_limited, max_segs))
break;
}
limit = mss_now;
if (tso_segs > 1 && !tcp_urg_mode(tp))
limit = tcp_mss_split_point(sk, skb, mss_now,
min_t(unsigned int,
cwnd_quota,
max_segs),
nonagle);
if (skb->len > limit &&
unlikely(tso_fragment(sk, TCP_FRAG_IN_WRITE_QUEUE,
skb, limit, mss_now, gfp)))
break;
if (tcp_small_queue_check(sk, skb, 0)) // 小队列检查,防止发送队列过长
break;
if (unlikely(tcp_transmit_skb(sk, skb, 1, gfp))) // 尝试传输skb
break;
repair:
/* 推进发送队列头部,这个skb已经被发送出去。
* 这个调用将增加packets_out的计数。
*/
tcp_event_new_data_sent(sk, skb);
tcp_minshall_update(tp, mss_now, skb); // 更新发送窗口大小
sent_pkts += tcp_skb_pcount(skb); // 累加已发送的数据包计数
if (push_one) // 如果设置了push_one标志,只发送一个包,然后停止处理
break;
}
if (is_rwnd_limited) // 如果接收窗口限制了速度,则启动相应的计时器
tcp_chrono_start(sk, TCP_CHRONO_RWND_LIMITED);
else // 否则,停止该计时器
tcp_chrono_stop(sk, TCP_CHRONO_RWND_LIMITED);
if (likely(sent_pkts)) { // 如果发送出去了数据包
if (tcp_in_cwnd_reduction(sk)) // 如果TCP处于拥塞窗口减少状态
tp->prr_out += sent_pkts; // 更新Proportional Rate Reduction的计数
/* 每次尾部丢包事件只发送一个丢包探测。 */
if (push_one != 2)
tcp_schedule_loss_probe(sk, false); // 计划发送丢包探测
is_cwnd_limited |= (tcp_packets_in_flight(tp) >= tp->snd_cwnd); // 检查是否受到拥塞窗口的限制
tcp_cwnd_validate(sk, is_cwnd_limited); // 验证拥塞窗口限制是否处于激活状态
return false;
}
return !tp->packets_out && !tcp_write_queue_empty(sk); // 如果没有在传输中的数据包,并且写队列不为空,返回true
}
这段代码是Linux内核中用于TCP协议的数据包发送机制的一部分,主要负责在可能的情况下将数据包或数据段发送到网络。在这个过程中,它或许会遇到多种情况,如拥塞窗口(cwnd)限制、接收窗口(rwnd)限制、网络层的流量控制、MTU探测等,并且会相应地更新状态信息(例如发送计时器)和内部计数器。如果确定无法发送数据(例如由于发送窗口关闭等原因),函数可能会返回true,表示有待发送的数据但目前无法发送。在每次有效发送后,它还会安排适当的丢失探测和拥塞窗口更新。
tcp_write_queue_purge
一、讲解
这个函数 tcp_write_queue_purge 是针对 TCP 协议在 Linux 内核网络栈中的一个函数,用于清除指定 socket (sk) 写队列中的所有 sk_buff 结构(即待发送数据包)。
具体功能如下:
1. tcp_chrono_stop(sk, TCP_CHRONO_BUSY);
停止针对 socket sk 的 TCP_CHRONO_BUSY 计时器。该计时器用于衡量 socket 处于忙碌状态的时间。
2. while ((skb = __skb_dequeue(&sk->sk_write_queue)) != NULL) {
这是一个 while 循环,从 socket sk 的写队列(sk_write_queue)中逐个取出 sk_buff(数据包)。
3. tcp_skb_tsorted_anchor_cleanup(skb);
清理与给定 sk_buff 相关的时间排序元数据。
4. sk_wmem_free_skb(sk, skb);
释放刚从队列中取出的 sk_buff 的内存,并调整 socket sk 的写内存计数器。
5. tcp_rtx_queue_purge(sk);
清空重传队列,释放所有在 TCP 重传队列中的 sk_buff 结构。
6. INIT_LIST_HEAD(&tcp_sk(sk)->tsorted_sent_queue);
初始化 socket sk 的 ts_sorted_sent_queue 链表头。该链表管理时间排序的已发送数据包队列。
7. sk_mem_reclaim(sk);
尝试回收 socket sk 的内存,根据当前内存使用情况,可能会释放或重新调整部分内存资源。
8. tcp_clear_all_retrans_hints(tcp_sk(sk));
清除全部 retrans hint 标记,这些标记用于优化数据包的重传处理。
9. tcp_sk(sk)->packets_out = 0;
将 socket sk 的 packets_out 计数器重置为 0,这个计数器记录了在飞数据包的数目(在路上但尚未确认的数据包)。
10. inet_csk(sk)->icsk_backoff = 0;
重置 socket sk 的指数退避计数器 icsk_backoff 到 0。在遇到网络拥堵导致超时重传时,该计数器值会增加。
总体来说,这个函数的作用是清理给定 socket sk 的所有已排队但尚未发送的数据包,确保在某些情况下(例如关闭连接前)发送队列被适当清空,从而释放相关资源。
二、中文注释
void tcp_write_queue_purge(struct sock *sk)
{
struct sk_buff *skb;
// 停止TCP的计时器(例如,忙时计时器)
tcp_chrono_stop(sk, TCP_CHRONO_BUSY);
// 循环,一直到写队列为空
while ((skb = __skb_dequeue(&sk->sk_write_queue)) != NULL) {
// 清理已排序发送队列的anchor(锚)信息
tcp_skb_tsorted_anchor_cleanup(skb);
// 释放skb占用的写缓冲区内存
sk_wmem_free_skb(sk, skb);
}
// 清除重传队列
tcp_rtx_queue_purge(sk);
// 初始化TCP控制块的已排序发送队列头部
INIT_LIST_HEAD(&tcp_sk(sk)->tsorted_sent_queue);
// 试图回收sk_buff结构所占用的内存
sk_mem_reclaim(sk);
// 清除所有用于快速重传的标志
tcp_clear_all_retrans_hints(tcp_sk(sk));
// 将"出站数据包数量"计数器设置为0
tcp_sk(sk)->packets_out = 0;
// 将网络传输层的退避级别设为0
inet_csk(sk)->icsk_backoff = 0;
}
以上是该函数的中文注释。函数的功能是清理TCP套接字的写队列,释放其中的skb(socket缓冲区),清理重传队列,重置相关的计数器和状态,以便套接字可以被安全地关闭或重置。