如何在 bytes.Buffer 中指定偏移位置写入数据(如协议包头长度字段)

夏枫大大_4123

夏枫大大_4123

2026-09-24

562人浏览

原创

如何在 bytes.Buffer 中指定偏移位置写入数据(如协议包头长度字段)

go 中 bytes.buffer 本身不支持随机写入或 seek 操作,但可通过预分配缓冲区 + 直接操作底层字节切片,在指定偏移处安全写入长度、id 等协议头字段。本文详解实现原理、正确写法及常见误区。

go 中 bytes.buffer 本身不支持随机写入或 seek 操作,但可通过预分配缓冲区 + 直接操作底层字节切片,在指定偏移处安全写入长度、id 等协议头字段。本文详解实现原理、正确写法及常见误区。

在构建网络协议包(如游戏服务器响应包)时,一个典型需求是:先写入变长的包体数据,再回填固定位置的包头字段(例如前 2 字节为消息 ID,紧随其后的 2 字节为总长度)。由于 bytes.Buffer 是顺序写入的抽象,它没有 Seek()WriteAt() 方法,因此无法像文件或 *os.File 那样直接跳转到 offset=0 再覆盖写入。

你最初尝试的写法存在两个关键问题:

  1. binary.Write(b, ..., 0) 先写入一个占位长度(如 0),但后续 b.Len() 包含了这个初始 4 字节(假设是 uint32),导致最终长度值比实际有效载荷多出头长度;
  2. b.Bytes() 返回的是只读视图(底层底层数组可修改,但需确保未被扩容),而你在 packet[2:] 上调用 PutUint16 前,若 b 在写入过程中触发了内存扩容(如初始 buffer 太小),packet 切片将指向已失效的旧底层数组 —— 这会导致静默错误:修改无效或 panic

✅ 正确做法是:预分配足够空间的底层数组,用 bytes.NewBuffer() 封装,确保所有写入都在同一底层数组内完成;之后通过 b.Bytes() 获取稳定切片,再用 binary.LittleEndian.PutUintXX() 直接修改指定偏移处的字节。

以下是推荐实现(以 2 字节 ID + 2 字节总长度的包头为例):

func CreatePacket(data []byte) []byte {
    // 预分配 4 字节头空间(ID:2 + Len:2),避免后续扩容
    header := make([]byte, 4)
    b := bytes.NewBuffer(header)

    // 写入完整 payload(自动追加到 header 后)
    binary.Write(b, binary.LittleEndian, data)

    // 获取完整字节切片(此时底层数组确定且未扩容)
    packet := b.Bytes()

    // 安全写入头字段:offset 0 开始写 uint16 ID
    binary.LittleEndian.PutUint16(packet, 0xF3)
    // offset 2 开始写 uint16 总长度(含 header!)
    binary.LittleEndian.PutUint16(packet[2:], uint16(len(packet)))

    return packet
}

⚠️ 关键注意事项:

  • 必须预分配头空间:使用 make([]byte, 4) 而非 bytes.Buffer{} 零值,否则 b.Bytes() 初始返回空切片,后续 binary.Write 可能触发扩容,导致 packet 指向过期内存;
  • 长度字段应包含整个包len(packet) 是 header + payload 总字节数,符合协议约定;
  • 字段大小需跨平台一致:明确使用 uint16/uint32 并配对 PutUint16/PutUint32,避免 int 在不同架构下宽度不一;
  • 避免重复创建 Bufferbytes.Buffer 是轻量结构体,每次新建无性能负担;若追求极致,可复用(b.Reset()),但需注意底层数组可能残留旧数据(此处因预分配+重写头,影响可控)。

此模式简洁、零分配(除预分配外)、无反射、完全内存安全,是 Go 网络编程中构造二进制协议包的标准实践。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Go中Type关键字的用法
Go中Type关键字的用法

Go中Type关键字的用法有定义新的类型别名或者创建新的结构体类型。本专题为大家提供Go相关的文章、下载、课程内容,供大家免费下载体验。

2023.09.06

2289

5

go怎么实现链表
go怎么实现链表

go通过定义一个节点结构体、定义一个链表结构体、定义一些方法来操作链表、实现一个方法来删除链表中的一个节点和实现一个方法来打印链表中的所有节点的方法实现链表。

2023.09.25

4487

6

go语言编程软件有哪些
go语言编程软件有哪些

go语言编程软件有Go编译器、Go开发环境、Go包管理器、Go测试框架、Go文档生成器、Go代码质量工具和Go性能分析工具等。本专题为大家提供go语言相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.13

602

5

0基础如何学go语言
0基础如何学go语言

0基础学习Go语言需要分阶段进行,从基础知识到实践项目,逐步深入。php中文网给大家带来了go语言相关的教程以及文章,欢迎大家前来学习。

2023.10.26

6225

20

Go语言实现运算符重载有哪些方法
Go语言实现运算符重载有哪些方法

Go语言不支持运算符重载,但可以通过一些方法来模拟运算符重载的效果。使用函数重载来模拟运算符重载,可以为不同的类型定义不同的函数,以实现类似运算符重载的效果,通过函数重载,可以为不同的类型实现不同的操作。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.02.23

2116

5

Go语言中的运算符有哪些
Go语言中的运算符有哪些

Go语言中的运算符有:1、加法运算符;2、减法运算符;3、乘法运算符;4、除法运算符;5、取余运算符;6、比较运算符;7、位运算符;8、按位与运算符;9、按位或运算符;10、按位异或运算符等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.02.23

2344

5

go语言开发工具大全
go语言开发工具大全

本专题整合了go语言开发工具大全,想了解更多相关详细内容,请阅读下面的文章。

2025.06.11

5139

12

go语言引用传递
go语言引用传递

本专题整合了go语言引用传递机制,想了解更多相关内容,请阅读专题下面的文章。

2025.06.26

3317

12

go语言版本升级
go语言版本升级

本专题整合了go语言版本升级方法,想了解更多相关内容,请阅读专题下面的文章。

2025.06.27

6010

16

热门下载

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

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.1万人学习