Golang 二进制数据打包与SHA256哈希校验实践

type packet struct {
    a uint8  // 一个无符号8位整数
    b []byte // 一个字节切片，用于表示固定长度的字符串
}

package main

import (
    "bytes"
    "crypto/sha256"
    "encoding/binary"
    "fmt"
)

type packet struct {
    a uint8
    b []byte
}

func main() {
    var p = packet{}
    p.a = 1
    // 将字符串转换为字节切片，并确保其长度符合预期（这里是5字节，包含填充）
    p.b = []byte("foo\x00\x00") 

    buf := bytes.Buffer{} // 初始化一个字节缓冲区

    // 写入第一个字段 (uint8)
    // 注意：binary.Write的第二个参数是字节序，第三个参数必须是可寻址的（指针）或可直接写入的值
    err := binary.Write(&buf, binary.BigEndian, p.a) 
    if err != nil {
        fmt.Println("写入p.a失败:", err)
        return
    }

    // 写入第二个字段 ([]byte)
    // 对于字节切片，可以直接使用Buffer的Write方法
    _, err = buf.Write(p.b) 
    if err != nil {
        fmt.Println("写入p.b失败:", err)
        return
    }

    // 至此，buf中包含了打包好的二进制数据
    fmt.Printf("打包后的二进制数据: %x\n", buf.Bytes())

    // ... 接下来是哈希计算
}

    // ... (接上文代码)

    h := sha256.New() // 创建一个新的SHA256哈希器
    h.Write(buf.Bytes()) // 将缓冲区中的所有二进制数据写入哈希器

    // 计算哈希值。Sum方法接受一个字节切片作为参数，
    // 如果传入非nil切片，会将哈希值追加到该切片末尾。
    // 传入nil或空切片表示返回一个新的切片。
    hash := h.Sum(nil) 

    fmt.Printf("SHA256哈希校验和: %x\n", hash)

    // 如果需要将哈希值附加到数据包末尾，可以这样做：
    // buf.Write(hash)
    // fmt.Printf("包含哈希的完整数据包: %x\n", buf.Bytes())
}

package main

import (
    "bytes"
    "crypto/sha256"
    "encoding/binary"
    "fmt"
)

// packet 定义了要打包的数据结构
type packet struct {
    a uint8  // 一个无符号8位整数
    b []byte // 一个字节切片，用于表示固定长度的字符串
}

func main() {
    var p = packet{}
    p.a = 1
    // 示例：一个5字节的字符串，其中包含空字节填充
    p.b = []byte("foo\x00\x00") 

    // 初始化一个字节缓冲区，用于存储打包后的二进制数据
    buf := bytes.Buffer{}

    // 1. 写入第一个字段 'a' (uint8)
    // 使用binary.BigEndian指定大端字节序。
    // 对于单个字节的uint8，字节序没有实际影响，但保持一致性是好习惯。
    err := binary.Write(&buf, binary.BigEndian, p.a)
    if err != nil {
        fmt.Println("Error writing p.a:", err)
        return
    }

    // 2. 写入第二个字段 'b' ([]byte)
    // 对于字节切片，直接使用bytes.Buffer的Write方法即可。
    _, err = buf.Write(p.b)
    if err != nil {
        fmt.Println("Error writing p.b:", err)
        return
    }

    // 打印打包后的原始二进制数据，方便调试
    fmt.Printf("Packed binary data: %x\n", buf.Bytes())

    // 3. 计算SHA256哈希校验和
    h := sha256.New()         // 创建一个新的SHA256哈希器
    h.Write(buf.Bytes())      // 将打包后的所有二进制数据写入哈希器
    hash := h.Sum(nil)        // 计算哈希值，返回一个新的字节切片

    // 打印计算出的SHA256哈希值
    fmt.Printf("SHA256 checksum: %x\n", hash)

    // 如果需要将哈希值附加到数据包末尾，可以执行以下操作：
    // _, err = buf.Write(hash)
    // if err != nil {
    //  fmt.Println("Error appending hash:", err)
    //  return
    // }
    // fmt.Printf("Full packet with hash: %x\n", buf.Bytes())
}