讲师中心 微信公众号
AI工具推荐 视频效率加速

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

浅明小哥_5821

浅明小哥_5821

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

|

858人浏览过

|

来源于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.WaitGroup 或 errgroup.Group 控制生命周期
  • 任一方向出错(如 EOF、timeout、read: connection reset)不能直接 return,得通知另一方也退出,否则 goroutine 泄漏

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

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

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

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

Golang Uber Dig
Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等

下载
  • 用 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工具

更多
豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

相关专题

更多
C语言变量命名
C语言变量命名

c语言变量名规则是:1、变量名以英文字母开头;2、变量名中的字母是区分大小写的;3、变量名不能是关键字;4、变量名中不能包含空格、标点符号和类型说明符。php中文网还提供c语言变量的相关下载、相关课程等内容,供大家免费下载使用。

3469

2023.06.20

c语言入门自学零基础
c语言入门自学零基础

C语言是当代人学习及生活中的必备基础知识,应用十分广泛,本专题为大家c语言入门自学零基础的相关文章,以及相关课程,感兴趣的朋友千万不要错过了。

2408

2023.07.25

c语言运算符的优先级顺序
c语言运算符的优先级顺序

c语言运算符的优先级顺序是括号运算符 > 一元运算符 > 算术运算符 > 移位运算符 > 关系运算符 > 位运算符 > 逻辑运算符 > 赋值运算符 > 逗号运算符。本专题为大家提供c语言运算符相关的各种文章、以及下载和课程。

1340

2023.08.02

c语言数据结构
c语言数据结构

数据结构是指将数据按照一定的方式组织和存储的方法。它是计算机科学中的重要概念,用来描述和解决实际问题中的数据组织和处理问题。数据结构可以分为线性结构和非线性结构。线性结构包括数组、链表、堆栈和队列等,而非线性结构包括树和图等。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1278

2023.08.09

c语言random函数用法
c语言random函数用法

c语言random函数用法:1、random.random,随机生成(0,1)之间的浮点数;2、random.randint,随机生成在范围之内的整数,两个参数分别表示上限和下限;3、random.randrange,在指定范围内,按指定基数递增的集合中获得一个随机数;4、random.choice,从序列中随机抽选一个数;5、random.shuffle,随机排序。

1396

2023.09.05

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2298

2023.09.20

c语言get函数的用法
c语言get函数的用法

get函数是一个用于从输入流中获取字符的函数。可以从键盘、文件或其他输入设备中读取字符,并将其存储在指定的变量中。本文介绍了get函数的用法以及一些相关的注意事项。希望这篇文章能够帮助你更好地理解和使用get函数 。

3760

2023.09.20

c数组初始化的方法
c数组初始化的方法

c语言数组初始化的方法有直接赋值法、不完全初始化法、省略数组长度法和二维数组初始化法。详细介绍:1、直接赋值法,这种方法可以直接将数组的值进行初始化;2、不完全初始化法,。这种方法可以在一定程度上节省内存空间;3、省略数组长度法,这种方法可以让编译器自动计算数组的长度;4、二维数组初始化法等等。

17515

2023.09.22

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

20

2026.10.28

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn