如何设置模糊测试的输入约束？

Question

假设我有以下结构

type Hdr struct{
  Src      uint16
  Dst      uint16
  Priotity byte
  Pktcnt   byte
  Opcode   byte
  Ver      byte
}

我有两个函数Marshal和Unmarshal ，它们将Hdr编码为二进制格式：

 0                   1          
 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|              Src              |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|              Dst              |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|  Prio |  Cnt  | Opcode|  Ver  |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+

我想使用Marshal Fuzz 生成随机、有效的Hdr实例，然后编组为二进制、 Unmarshal二进制并确保 output 与原始输入匹配。

我遇到的主要问题是我无法弄清楚如何告诉Priotity Fuzz 像优先级这样的字段不能大于 15，否则它们在编组时会被截断（只有 4 位）。 如何设置此约束？

更新

这只是一个玩具箱。 很多时候，像上面这样的协议，像opcode这样的东西会触发二次更复杂的解析/审查。 模糊测试仍然可以在约束内找到非常有用的问题（即：如果 Prio 0x00和 Cnt 0x2F辅助解析器将出错，因为分隔符是\ ）。

Answer 1

为了跳过无趣的结果，在你的模糊测试中调用t.Skip function。像这样：

f.Fuzz(func(t *testing.T, b []byte) {
    a, err := Unmarshal(b)
    if err != nil {
        t.Skip()
        return
    }
    c, err := Marshal(a)
    if err != nil || !bytes.Equal(b, c) {
        t.Errorf("Eek!")
    }
})

Answer 2

编辑

我不确定模糊测试是否适合这里。 模糊测试旨在发现意外输入：多字节 UTF8 输入（有效和无效）； 负值； 巨大的价值，很长的长度等。这些将试图捕捉“边缘”情况。

就您而言，您知道：

• Unmarshal输入负载必须为 6 个字节（否则会出错）
• 你确切地知道你的内部“边缘”

所以 vanilla testing.T测试可能更适合这里。

---

把事情简单化。

如果你不想“浪费”一个 Fuzz 输入并且你知道你的代码的输入限制，你可以尝试这样的事情：

func coerce(h *Hdr) (skip bool) {

    h.Priotity &= 0x0f // ensure priority is 0-15
    h.OpCode %= 20     // ensure opcode is 0-19 

    return false       // optionally skip this test
}

在您的测试中-可以测试强制值-或跳过（如@jch所示）：

import "github.com/google/go-cmp/cmp"

f.Fuzz(func(t *testing.T, src, dst uint16, pri, count, op, ver byte) {
    h := Hdr{src, dst, pri, count, op, ver}

    if coerce(&h) {
        t.Skip()
        return
    }

    bs, err := Marshal(h)     // check err

    h2, err := Unmarhsal(bs)  // check err

    if !cmp.Equal(h, h2) {
        t.Errorf("Marshal/Unmarshal validation failed for: %+v", h)
    }
}