// headersEqual compares a response and request AoE header for equality, but
// does not factor in their Arg fields.
func headersEqual(res, req aoe.Header) bool {
if !res.FlagResponse {
panic("response AoE header did not have FlagResponse set, please verify parameter order")
}
// Set request's response flag for comparison
req.FlagResponse = true
// Clear arguments of both request and response so we can compare the
// headers, and because the arguments are compared separately
req.Arg = nil
res.Arg = nil
return reflect.DeepEqual(res, req)
}
// testBPFProgram builds a BPF program for the input server and compares the
// output of the program against the input AoE header, returning whether or
// not the header was accepted by the filter.
func testBPFProgram(t *testing.T, s *Server, h *aoe.Header) bool {
filter, ok := bpf.Disassemble(s.mustAssembleBPF(testMTU))
if !ok {
t.Fatal("failed to decode all BPF instructions")
}
vm, err := bpf.NewVM(filter)
if !ok {
t.Fatalf("failed to load BPF program: %v", err)
}
// Fill in empty AoE header fields not relevant to this test
h.Version = aoe.Version
h.Command = aoe.CommandQueryConfigInformation
h.Arg = &aoe.ConfigArg{
Command: aoe.ConfigCommandRead,
}
hb, err := h.MarshalBinary()
if err != nil {
t.Fatalf("failed to marshal AoE header to binary: %v", err)
}
f := ðernet.Frame{
EtherType: aoe.EtherType,
Payload: hb,
}
fb, err := f.MarshalBinary()
if err != nil {
t.Fatalf("failed to marshal Ethernet frame to binary: %v", err)
}
out, err := vm.Run(fb)
if err != nil {
t.Fatalf("failed to run BPF program: %v", err)
}
return out > 0
}
