// 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 }