【C语言】tcp_sendmsg_locked

一、讲解

tcp_sendmsg_locked 函数是 Linux 内核中实现 TCP 数据发送的一个核心函数。这个函数被调用来将用户空间的数据通过 TCP 发送出去。以下是该函数的基本工作流程的中文解释:
1. 函数初始化和检查:
   - 它首先检查是否使用了 TCP 零拷贝发送(MSG_ZEROCOPY)以及确保发送状态是正确的。
   - 函数通过检查标志位来处理 TCP 快速打开特性。
   - 设置发送超时和评估发送路径是否处于“应用受限”状态。
2. 等待连接完成:
   - 如果 TCP 连接还未建立,它将等待连接完成,除非使用了 TCP 快速打开。
3. 准备发送:
   - 如果存在 TCP 修复模式,它将处理特定的发送队列。
   - 解析传输层控制消息。
   - 清除可能影响发送队列的异步标记。
   - 计算最大段大小(MSS)和发送目标大小。
4. 数据发送循环:
   - 函数进入循环,开始从用户消息(`msg`)中拷贝数据到内核的发送缓冲区。
   - 它处理两种类型的发送缓冲区:线性空间和分散/聚集空间。
   - 可能会处理 socket 的内存分配和等待内存分配。
   - 根据不同情况拷贝数据到 TCP 段(`skb`),并更新 TCP 状态信息(如写序列号)。
5. 错误处理:
   - 如果出现错误或异常,函数会进行错误处理,包括释放必要的资源。
6. 推送数据和结束发送:
   - 完成数据拷贝后,如果已经拷贝了足够的数据,函数将推动网络栈发送这些数据(或者等待发送缓冲区空间可用来发送更多数据)。
   - 根据使用的发送标志,函数可能会标记 PSH 推送位,或使用 Nagle 算法等待发送。
   - 数据发送后,函数执行必要的清理操作,返回拷贝的字节数,或者发送失败时的错误码。
整体上,`tcp_sendmsg_locked` 函数处理了一系列复杂的 TCP 发送逻辑,包括 TCP 发送缓冲区的管理、段的创建及填充、发送拥塞控制、零拷贝优化等。此函数的名称表示它应在相应的 socket 已被锁定的情况下调用,以保证线程安全。这是内核网络栈的核心函数之一,涉及到许多内核编程细节和网络协议的实现。

二、注释

/ tcp_sendmsg_locked函数的实现
int tcp_sendmsg_locked(struct sock *sk, struct msghdr *msg, size_t size)
{// 声明一系列变量struct tcp_sock *tp = tcp_sk(sk); // 获取tcp_sock结构struct ubuf_info *uarg = NULL;struct sk_buff *skb;struct sockcm_cookie sockc;int flags, err, copied = 0;int mss_now = 0, size_goal, copied_syn = 0;bool process_backlog = false;bool zc = false;long timeo;flags = msg->msg_flags; // 获取消息标志// 检查是否启用了零拷贝发送if (flags & MSG_ZEROCOPY && size && sock_flag(sk, SOCK_ZEROCOPY)) {// TCP状态检查if ((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) {err = -EINVAL;goto out_err;}skb = tcp_write_queue_tail(sk);uarg = sock_zerocopy_realloc(sk, size, skb_zcopy(skb));if (!uarg) {err = -ENOBUFS;goto out_err;}zc = sk->sk_route_caps & NETIF_F_SG;if (!zc)uarg->zerocopy = 0;}// 处理快速打开的情况if (unlikely(flags & MSG_FASTOPEN || inet_sk(sk)->defer_connect) && !tp->repair) {err = tcp_sendmsg_fastopen(sk, msg, &copied_syn, size);if (err == -EINPROGRESS && copied_syn > 0)goto out;else if (err)goto out_err;}timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT); // 获取发送超时tcp_rate_check_app_limited(sk); // 检查应用级发送是否受限// 等待连接完成,除非是被动端的TCP快速打开if (((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) && !tcp_passive_fastopen(sk)) {err = sk_stream_wait_connect(sk, &timeo);if (err != 0)goto do_error;}// 如果处于TCP修复状态if (unlikely(tp->repair)) {if (tp->repair_queue == TCP_RECV_QUEUE) {// 修复时发送recv队列中的数据copied = tcp_send_rcvq(sk, msg, size);goto out_nopush;}err = -EINVAL;if (tp->repair_queue == TCP_NO_QUEUE)goto out_err;// 处于发送队列的修复}// 初始化sockcm_cookiesockcm_init(&sockc, sk);if (msg->msg_controllen) {err = sock_cmsg_send(sk, msg, &sockc);if (unlikely(err)) {err = -EINVAL;goto out_err;}}sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk); // 清除异步无空间标志// 开始发送数据copied = 0;
// 重启标签,处理发送过程中需要重启的情况
restart:mss_now = tcp_send_mss(sk, &size_goal, flags); // 获取发送的最大报文段大小err = -EPIPE;if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))goto do_error;// 循环处理要发送的数据while (msg_data_left(msg)) {int copy = 0;skb = tcp_write_queue_tail(sk);if (skb)copy = size_goal - skb->len;if (copy <= 0 || !tcp_skb_can_collapse_to(skb)) {bool first_skb;int linear;// 创建新的数据段
new_segment:if (!sk_stream_memory_free(sk))goto wait_for_sndbuf;if (process_backlog && sk_flush_backlog(sk)) {process_backlog = false;goto restart;}first_skb = tcp_rtx_and_write_queues_empty(sk);linear = select_size(first_skb, zc);skb = sk_stream_alloc_skb(sk, linear, sk->sk_allocation,first_skb);if (!skb)goto wait_for_memory;process_backlog = true;skb->ip_summed = CHECKSUM_PARTIAL;skb_entail(sk, skb);copy = size_goal;// 如果处于修复模式,标记该skb已经“发送”if (tp->repair)TCP_SKB_CB(skb)->sacked |= TCPCB_REPAIRED;}// 尝试附加数据到skb的末尾if (copy > msg_data_left(msg))copy = msg_data_left(msg);// 将数据从用户空间拷贝到skb中if (skb_availroom(skb) > 0 && !zc) {// 有空间进行直接拷贝copy = min_t(int, copy, skb_availroom(skb));err = skb_add_data_nocache(sk, skb, &msg->msg_iter, copy);if (err)goto do_fault;} else if (!zc) {bool merge = true;int i = skb_shinfo(skb)->nr_frags;// 大页内存管理struct page_frag *pfrag = sk_page_frag(sk);// 确保page_frag有足够内存if (!sk_page_frag_refill(sk, pfrag))goto wait_for_memory;// 检查skb是否可以合并到最后的一个fragif (!skb_can_coalesce(skb, i, pfrag->page, pfrag->offset)) {// 如果达到了frag的上限,则新建一个段if (i >= sysctl_max_skb_frags) {tcp_mark_push(tp, skb);goto new_segment;}merge = false;}// 拷贝数据到页内存copy = min_t(int, copy, pfrag->size - pfrag->offset);// 确保套接字有足够的写缓冲区空间if (!sk_wmem_schedule(sk, copy))goto wait_for_memory;// 无拷贝地将数据从用户空间复制到页内存err = skb_copy_to_page_nocache(sk, &msg->msg_iter, skb,pfrag->page,pfrag->offset,copy);if (err)goto do_error;// 更新skb状态if (merge) {// 如果合并成功,增加frag的大小skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy);} else {// 没有合并,就在skb中新增一个page frag描述符skb_fill_page_desc(skb, i, pfrag->page, pfrag->offset, copy);page_ref_inc(pfrag->page); // 增加页引用计数}// 更新page_frag位置pfrag->offset += copy;} else {// 零拷贝的发送方式err = skb_zerocopy_iter_stream(sk, skb, msg, copy, uarg);if (err == -EMSGSIZE || err == -EEXIST) {// 出现错误,需要新段tcp_mark_push(tp, skb);goto new_segment;}if (err < 0)goto do_error;copy = err;}// 更新TCP状态if (!copied)TCP_SKB_CB(skb)->tcp_flags &= ~TCPHDR_PSH;tp->write_seq += copy;TCP_SKB_CB(skb)->end_seq += copy;tcp_skb_pcount_set(skb, 0);copied += copy;if (!msg_data_left(msg)) {// 如果数据已经全部发送完成,设置结束标志if (unlikely(flags & MSG_EOR))TCP_SKB_CB(skb)->eor = 1;goto out;}// 检查skb是否达到了目标大小或者其它特殊情况if (skb->len < size_goal || (flags & MSG_OOB) || unlikely(tp->repair))continue;if (forced_push(tp)) {// 如果需要立即发送数据,则添加PSH标志并推送数据tcp_mark_push(tp, skb);__tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH);} else if (skb == tcp_send_head(sk))// 如果skb是待发送队列的头部,可能需要推送一个分段tcp_push_one(sk, mss_now);continue;// 对于缓冲区溢出,设置标志位并等待可用内存
wait_for_sndbuf:set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
wait_for_memory:if (copied)// 如果已经拷贝了一些数据,则尝试推送tcp_push(sk, flags & ~MSG_MORE, mss_now,TCP_NAGLE_PUSH, size_goal);// 等待足够的发送缓冲区内存err = sk_stream_wait_memory(sk, &timeo);if (err != 0)goto do_error;// 重新计算mss和目标大小mss_now = tcp_send_mss(sk, &size_goal, flags);}
out:// 数据发送完成,调用tcp_push推送所有挂起的数据帧if (copied) {tcp_tx_timestamp(sk, sockc.tsflags);tcp_push(sk, flags, mss_now, tp->nonagle, size_goal);}
out_nopush:// 释放uarg资源sock_zerocopy_put(uarg);return copied + copied_syn;// 处理skb没有复制任何数据的情况
do_fault:if (!skb->len) {tcp_unlink_write_queue(skb, sk);// 这是TCP中除了连接重置以外唯一可能删除send_head的地方tcp_check_send_head(sk, skb);sk_wmem_free_skb(sk, skb);}// 处理错误,如果已经复制了数据,则直接退出
do_error:if (copied + copied_syn)goto out;
out_err:// 处理失败,中止零拷贝操作,记录错误并返回sock_zerocopy_put_abort(uarg);// 根据错误代码设置套接字错误状态,并返回错误err = sk_stream_error(sk, flags, err);// 如果写队列为空,并且返回了EAGAIN错误,则尝试触发epoll的边缘触发事件if (unlikely(skb_queue_len(&sk->sk_write_queue) == 0 &&	
err == -EAGAIN)) {sk->sk_write_space(sk); // 若写入队列为空并且错误为EAGAIN,确保调用sk_write_space来唤醒epoll等待者,唤醒可能在等待发送缓冲区空间的epolltcp_chrono_stop(sk, TCP_CHRONO_SNDBUF_LIMITED); // 停止发送缓冲区限制的计时器}return err; // 返回出错信息
}
EXPORT_SYMBOL_GPL(tcp_sendmsg_locked); // 导出tcp_sendmsg_locked符号,允许其他内核模块调用	

三、tcp_sendmsg

这个函数`tcp_sendmsg`用于处理TCP socket的发送消息操作。让我们逐行地用中文解释这个函数的作用:

int tcp_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
{int ret;lock_sock(sk);  // 对指定的socket加锁,以防止并发访问导致的数据竞争。ret = tcp_sendmsg_locked(sk, msg, size);  // 在锁定后,调用tcp_sendmsg_locked函数发送消息。这个函数实现了消息的发送逻辑,但假设调用它的上下文已经持有了锁。release_sock(sk);  // 消息发送完成后,释放之前获取的锁。return ret;  // 返回tcp_sendmsg_locked函数的返回值,通常是已发送数据的字节数或者一个错误码。
}
EXPORT_SYMBOL(tcp_sendmsg);  // 将tcp_sendmsg函数导出,使它可以被该模块外的代码调用。

总的来说,`tcp_sendmsg`是一个对外暴露的接口,它用于在用户空间调用以发起TCP通信。该函数首先锁定目标socket,然后调用实际发送消息实现的内部函数`tcp_sendmsg_locked`,发送过程完成后释放锁,并返回发送操作的结果(成功发送的字节数或错误码)。通过锁来保证tcp发送操作的线程安全性。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/769406.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【Internet结构和ISP,分组延时、丢失和吞吐量】

文章目录 一、Internet结构和ISP1.互联网络结构&#xff1a;网络的网络2.Internet 结构&#xff1a;network of networks 二、分组延时、丢失和吞吐量1.分组丢失和延时是怎样发生的&#xff1f;2.四种分组延时3.分组丢失4.吞吐量 一、Internet结构和ISP 1.互联网络结构&#x…

(1) 易经与命运_学习笔记

个人笔记&#xff0c;斟酌阅读 占卦的原理 三个铜板&#xff0c;正面是3&#xff0c;反面2&#xff0c;三个一起转&#xff0c;得出6,7,8,9 数字象6老阴7少阳8少阴9老阳 生数和成数 生数和成数应该说出自《河图》。其中一二三四五为生数&#xff0c;六七八九十为成数。 生…

一小时学习redis!

redis 基于内存的数据存储系统 三种使用方式 redis优势 安装redis 最后一种方式只能得到5.0的redis版本 比较老&#xff01; 启动redis redis-server.exe 命令 停止ctrlc或关闭 启动客户端 redis-cli redisinsight安装 字符串 redis区分大小写 默认使用字符串存储 二进制…

iOS开发 - 转源码 - __weak问题解决

iOS开发 - 转源码 - __weak问题解决 在使用clang转换OC为C代码时&#xff0c;可能会遇到以下问题 cannot create __weak reference in file using manual reference 原因 __weak弱引用是需要runtime支持的&#xff0c;如果我们还只是使用静态编译&#xff0c;是无法正常转换的…

Redis持久化【RDB,bgsave的写时复制机制】【AOF,aof重写机制】【Redis混合持久化,以及对应改变aof重写规则】【Redis数据备份策略】

Redis持久化 RDB快照&#xff08;snapshot&#xff09;bgsave的写时复制(COW)机制 AOF&#xff08;append-only file&#xff09;AOF重写 Redis 4.0 混合持久化开启持久化后&#xff0c;AOF重写规则发生了变化 Redis数据备份策略&#xff1a; 转自 图灵课堂 RDB快照&#xff0…

如何学习VBA_3.2.19:利用Shell函数运行可执行程序

我给VBA的定义&#xff1a;VBA是个人小型自动化处理的有效工具。利用好了&#xff0c;可以大大提高自己的劳动效率&#xff0c;而且可以提高数据处理的准确度。我推出的VBA系列教程共九套和一部VBA汉英手册&#xff0c;现在已经全部完成&#xff0c;希望大家利用、学习。 如果…

【暴刷力扣】11. 盛最多水的容器

11. 盛最多水的容器 题目 给定一个长度为 n 的整数数组 height 。有 n 条垂线&#xff0c;第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。 找出其中的两条线&#xff0c;使得它们与 x 轴共同构成的容器可以容纳最多的水。 返回容器可以储存的最大水量。 说明&#xf…

leetcode216组合总和III

本题思考&#xff1a; 对于输入样例k3,n9 输出里面为什么只有 [[1,2,6],[1,3,5],[2,3,4]]而没有下图所示的重复情况出现呢&#xff1f; 当时代码写错了&#xff0c;思考许久不得解&#xff0c;后面经过仔细对比代码之后发现是我的代码出现了逻辑错误&#xff0c;而正是这一关键…

前端Webpack5高级进阶课程

课程介绍 本套视频教程主要内容包含React/Vue最新版本脚手架分析、基于Webpack5编写自己的loader和plugin等&#xff0c;让你开发时选择更多样&#xff0c;最后&#xff0c;用不到一百行的代码实现Webpack打包。通过本套视频教程的学习&#xff0c;可以帮你彻底打通Webpack的任…

Go——指针和内存逃逸

区别于C/C中的指针&#xff0c;Go语言中的指针不能进行偏移和运算&#xff0c;是安全指针。 要搞明白Go语言中的指针概念需要先知道3个概念&#xff1a;指针地址&#xff0c;指针类型和指针取值。 一. Go语言的指针 Go语言中的函数传参都是值拷贝&#xff0c;当我们想修改某个…

新台阶——蓝桥杯单片机省赛第十四届程序设计题目

在做十四届题目之前&#xff0c;常常听学长说&#xff0c;十四届以前拿省一真的是右手就行&#xff0c;并不相信&#xff0c;在经历十四届痛苦的大量修bug和优化之后&#xff0c;或许学长的话真说对了几分。话不多说&#xff0c;我们开始一起完成单片机第十四届程序设计题目。 …

Linux环境基础开发工具yum,vim使用

目录 1.Linux 软件包管理器 yum1.1什么是软件包1.2关于 rzsz1.3注意事项1.4查看软件包1.5如何安装软件1.6如何卸载软件 2.Linux开发工具2.1Linux编辑器-vim使用2.1.1vim的基本概念2.1.2vim的基本操作2.1.3vim正常模式命令集2.1.4vim末行模式命令集2.1.5vim操作总结 2.2简单vim配…

删除数组中的指定元素(了解如何删除数组中的指定元素,并返回一个新的数组,看这一篇就足够了!)

前言&#xff1a;有时候我们会遇到要在数组中删除指定元素&#xff0c;但是不能创建新的数组&#xff0c;那么这个时候应该如何操作呢&#xff1f; ✨✨✨这里是秋刀鱼不做梦的BLOG ✨✨✨想要了解更多内容可以访问我的主页秋刀鱼不做梦-CSDN博客 废话不多讲&#xff0c;让我们…

RuoYi-Vue-Plus(登录流程-验证码生成)

一、登录流程 1- 进入登录页面&#xff0c;调用 com.ruoyi.web.controller.common.CaptchaController 类中的 captchaImage 方法&#xff0c;生成base64的图片 以及 UUID 2- 提交 登录信息 验证码 uuid 比对 错误&#xff1a;返回错误信息&#xff0c;删除缓存的验证码 成功…

「MySQL」数据库约束

&#x1f387;个人主页&#xff1a;Ice_Sugar_7 &#x1f387;所属专栏&#xff1a;数据库 &#x1f387;欢迎点赞收藏加关注哦&#xff01; 数据库约束 &#x1f349;约束类型&#x1f34c;NOT NULL&#x1f34c;UNIQUE&#x1f34c;DEFAULT&#x1f34c;主键&#x1f34c;外键…

layui框架实战案例(25):laydate中mark的数据后端生成

mark&#xff0c;自定义日期标记。该属性是对 calendar 属性的进一步延伸&#xff0c;灵活度更高。属性可批量设置多个日期标记&#xff0c;如&#xff1a; mark: {0-10-14: 生日, //每年每月的某一天0-0-10: 工资, // 每月 10 号2008-8-8: 开幕, // 指定的日期 }其中日期的格式…

【微服务】Spring Boot 版本升级到 2.7.18

前言 目前项目上扫描出一些 Java 依赖的代码漏洞&#xff0c;需要对现有依赖版本升级&#xff0c;记录一下遇到的问题。 <spring-boot.version>2.3.2.RELEASE</spring-boot.version> <spring-cloud.version>Hoxton.SR9</spring-cloud.version> <s…

亚马逊服务器ssh以及scp

ssh awspass.pem为创建服务器时创建的密钥&#xff0c;ubuntu用户 ssh -i "awspass.pem" ubuntuipscp scp -i "awspass.pem" -r dist/* ubuntuip:/home/ubuntu/

菱形继承的问题以及解决方法

大家好&#xff1a; 衷心希望各位点赞。 您的问题请留在评论区&#xff0c;我会及时回答。 菱形继承概念 两个派生类继承同一个基类 又有一个类同时继承这两个派生类 这种继承被称为菱形继承&#xff0c;或者钻石继承。 基类A同时派生出类B和类C&#xff0c;派生类D又同时继…

基于ARM 的Linux系统的交叉编译

返回&#xff1a;OpenCV系列文章目录&#xff08;持续更新中......&#xff09; 上一篇&#xff1a;在 MacOS 中安装 下一篇&#xff1a;MultiArch与Ubuntu/Debian 的交叉编译 警告 本教程可以包含过时的信息。 此步骤已在 Ubuntu Linux 12.04 上进行了测试&#xff0c;但应…