0

0

如何在Golang中编写端口转发工具Port Forwarder Go语言io.Copy流量代理

P粉602998670

P粉602998670

发布时间:2026-03-09 12:27:14

|

818人浏览过

|

来源于php中文网

原创

io.copy 单向阻塞导致双向 tcp 连接卡死或丢数据,必须用 errgroup.group 并发启动两个 io.copy 并统一管控生命周期,配合 setdeadline 和局部缓冲区确保安全高效转发。

如何在golang中编写端口转发工具port forwarder go语言io.copy流量代理

为什么 io.Copy 直接用在端口转发里会卡住或丢数据

因为 io.Copy 是单向阻塞复制,而 TCP 连接是双向的。只调一个 io.Copy(比如从 client 到 server),另一头没人在读,对端发来的数据就堆积在内核缓冲区,最终触发 TCP 窗口关闭、连接假死,甚至被中间设备断连。

常见现象:curl 能发请求但收不到响应;HTTP 返回 200 但 body 空;SSH 登录后立刻断开。

  • 必须同时启动两个 io.Copy goroutine,分别处理 client→remote 和 remote→client
  • 两个 io.Copy 要共用同一个 sync.WaitGrouperrgroup.Group 控制生命周期
  • 任一方向出错(如 EOF、timeout、read: connection reset)不能直接 return,得通知另一方也退出,否则 goroutine 泄漏

如何安全地关闭双向 io.Copy 连接

很多人用 defer conn.Close() 就以为万事大吉,但实际中 client 或 remote 可能已提前断开,goroutine 卡在 io.CopyRead 上,永远等不到 EOF。

关键不是“怎么关”,而是“谁来触发关”和“关得是否干净”:

立即学习go语言免费学习笔记(深入)”;

WordAi
WordAi

WordAI是一个AI驱动的内容重写平台

下载
  • errgroup.Group 启动两个 io.Copy,它天然支持任意一个出错就 cancel 其余任务
  • 不要依赖 conn.Close() 触发读端退出 —— io.Copy 在已关闭的 conn 上会立即返回 error,但你要确保 close 发生在 copy 结束之后,否则可能 panic
  • 给底层连接加 SetDeadline(比如 30 秒),避免因网络抖动无限 hang 住
  • 示例核心逻辑:
    g, _ := errgroup.WithContext(ctx)
    g.Go(func() error { return io.Copy(remote, client) })
    g.Go(func() error { return io.Copy(client, remote) })
    _ = g.Wait() // 任一 copy 出错,另一个也会被 cancel

net.Conn 复用时为什么会出现 “use of closed network connection”

错误信息:write tcp 127.0.0.1:8080->127.0.0.1:54322: use of closed network connection。这不是 bug,是典型的竞态:一个 goroutine 已 close conn,另一个还在往它写。

根本原因在于没做连接状态同步:

  • 别在多个 goroutine 里裸调 conn.Write —— 即使用了 errgroup,也要确保所有写操作都在 copy goroutine 内部完成
  • 如果要加日志或统计,用 io.TeeReader/io.TeeWriter 包一层,别自己另起 goroutine 读写原 conn
  • 连接池?别在这类代理场景里用 http.Transport 那套复用逻辑 —— 端口转发是 1:1 生命周期绑定,复用只会放大状态混乱
  • 最简方案:每个连接独占一对 goroutine + 一个 errgroup.Group,结束后统一 close

要不要加 buffer 或换用 io.CopyBuffer

默认 io.Copy 用 32KB 缓冲区,在大多数内网转发场景够用。但如果你代理的是高频小包(比如 WebSocket ping/pong、Redis 请求),默认 buffer 会导致 syscall 过多,CPU 毛刺明显。

这时可以换 io.CopyBuffer,但注意两点:

  • buffer 大小不是越大越好 —— 超过 64KB 容易触发 Linux 的 copy_to_user 分页开销,实测 16KB~32KB 平衡点最佳
  • buffer 必须是局部变量或 sync.Pool 获取,绝不能全局复用 —— 否则并发时内容被覆盖,出现乱码或协议解析失败
  • 示例:
    buf := make([]byte, 32*1024)
    _, err := io.CopyBuffer(dst, src, buf)

真正影响吞吐的往往不是 buffer,而是连接建立延迟和 TLS 握手 —— 如果你转发 HTTPS 流量,别优化 io.Copy,先确认是否真需要透传(而不是用 httputil.NewSingleHostReverseProxy

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

210

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

247

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

355

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

214

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

407

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

490

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

200

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

1397

2025.06.17

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

59

2026.03.06

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 6万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号