f_bpf_test.go

package tc

import (
	"errors"
	"testing"

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

func TestBpf(t *testing.T) {
	tests := map[string]struct {
		val  Bpf
		err1 error
		err2 error
	}{
		"simple": {val: Bpf{
			Ops:     bytesPtr([]byte{0x6, 0x0, 0x0, 0x0, 0xff, 0xff, 0xff, 0xff}),
			OpsLen:  uint16Ptr(0x1),
			ClassID: uint32Ptr(0x10001),
			Flags:   uint32Ptr(0x1),
		}},
		"da obj /tmp/bpf.o sec foo": {val: Bpf{
			FD: uint32Ptr(8), Name: stringPtr("bpf.o:[foo]"),
			Flags: uint32Ptr(0x1), FlagsGen: uint32Ptr(0x2),
		}},
		"all options": {val: Bpf{
			Ops:     bytesPtr([]byte{0x6, 0x0, 0x0, 0x0, 0xff, 0xff, 0xff, 0xff}),
			OpsLen:  uint16Ptr(0x1),
			ClassID: uint32Ptr(0x10001),
			FD:      uint32Ptr(42),
			Name:    stringPtr("testing"),
			Tag:     bytesPtr([]byte{0xAA, 0x55}),
			ID:      uint32Ptr(42),
		}},
		"filter add dev XXX ingress bpf bytecode '1,6 0 0 4294967295,' flowid 1:1 action drop": {
			val: Bpf{
				Ops:     bytesPtr([]byte{0x6, 0x0, 0x0, 0x0, 0xff, 0xff, 0xff, 0xff}),
				OpsLen:  uint16Ptr(1),
				ClassID: uint32Ptr(0x10001),
				Action: &Action{
					Kind: "gact",
					Gact: &Gact{
						Parms: &GactParms{
							Action: 2, // drop
						},
					},
				},
			},
		},
	}

	for name, testcase := range tests {
		t.Run(name, func(t *testing.T) {
			data, err1 := marshalBpf(&testcase.val)
			if err1 != nil {
				if errors.Is(err1, testcase.err1) {
					return
				}
				t.Fatalf("Unexpected error: %v", err1)
			}
			val := Bpf{}
			err2 := unmarshalBpf(data, &val)
			if err2 != nil {
				if errors.Is(err2, testcase.err2) {
					return
				}
				t.Fatalf("Unexpected error: %v", err2)

			}
			if diff := cmp.Diff(testcase.val, val); diff != "" {
				t.Fatalf("Bpf missmatch (-want +got):\n%s", diff)
			}
		})
	}
	t.Run("nil", func(t *testing.T) {
		_, err := marshalBpf(nil)
		if !errors.Is(err, ErrNoArg) {
			t.Fatalf("unexpected error: %v", err)
		}
	})
}