diff --git a/.mise.toml b/.mise.toml index 06b9ea5..91c3d4f 100644 --- a/.mise.toml +++ b/.mise.toml @@ -47,19 +47,11 @@ run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzClientHello ./mtglib/internal/f [tasks."test:fuzz:client-handshake"] description = "Run fuzzy test for ClientHandshake" -run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzClientHandshake ./mtglib/internal/obfuscated2" +run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzClientServerHandshake ./mtglib/internal/obfuscation" -[tasks."test:fuzz:server-generate-handshake-frame"] -description = "Run fuzzy test for ServerGenerateHandshakeFrame" -run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzServerGenerateHandshakeFrame ./mtglib/internal/obfuscated2" - -[tasks."test:fuzz:server-receive"] -description = "Run fuzzy test for ServerReceive" -run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzServerReceive ./mtglib/internal/obfuscated2" - -[tasks."test:fuzz:server-send"] -description = "Run fuzzy test for ServerSend" -run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzServerSend ./mtglib/internal/obfuscated2" +[tasks."test:fuzz:server-handshake-frame"] +description = "Run fuzzy test for GenerateHandshakeFrame" +run = "go test -v {{ vars.fuzzflags }} -fuzz=FuzzGenerateHandshakeFrame ./mtglib/internal/obfuscation" [tasks.static] description = "Build static binary" diff --git a/mtglib/internal/obfuscation/handshake_frame_fuzz_test.go b/mtglib/internal/obfuscation/handshake_frame_fuzz_test.go index 71965ec..3806e15 100644 --- a/mtglib/internal/obfuscation/handshake_frame_fuzz_test.go +++ b/mtglib/internal/obfuscation/handshake_frame_fuzz_test.go @@ -28,6 +28,13 @@ func FuzzGenerateHandshakeFrame(f *testing.F) { frame.data[4]|frame.data[5]|frame.data[6]|frame.data[7]) assert.Equal(t, hfConnectionType[:], frame.connectionType()) + + if arg < 0 { + arg = -arg + } else if arg == 0 { + arg = defaultDC + } + assert.EqualValues(t, arg, frame.dc()) }) }