whatcanGOwrong

go/pkg/mod/github.com/cilium/ebpf@v0.11.0/link/cgroup_test.go

@@ -0,0 +1,71 @@
+package link
+
+import (
+	"testing"
+
+	"github.com/cilium/ebpf"
+	"github.com/cilium/ebpf/internal/testutils"
+)
+
+func TestAttachCgroup(t *testing.T) {
+	cgroup, prog := mustCgroupFixtures(t)
+
+	link, err := AttachCgroup(CgroupOptions{
+		Path:    cgroup.Name(),
+		Attach:  ebpf.AttachCGroupInetEgress,
+		Program: prog,
+	})
+	testutils.SkipIfNotSupported(t, err)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer link.Close()
+
+	if haveBPFLink() == nil {
+		if _, ok := link.(*linkCgroup); !ok {
+			t.Fatalf("Have support for bpf_link, but got %T instead of linkCgroup", link)
+		}
+	} else {
+		if _, ok := link.(*progAttachCgroup); !ok {
+			t.Fatalf("Expected progAttachCgroup, got %T instead", link)
+		}
+	}
+}
+
+func TestProgAttachCgroup(t *testing.T) {
+	cgroup, prog := mustCgroupFixtures(t)
+
+	link, err := newProgAttachCgroup(cgroup, ebpf.AttachCGroupInetEgress, prog, 0)
+	if err != nil {
+		t.Fatal("Can't create link:", err)
+	}
+
+	testLink(t, link, prog)
+}
+
+func TestProgAttachCgroupAllowMulti(t *testing.T) {
+	cgroup, prog := mustCgroupFixtures(t)
+
+	link, err := newProgAttachCgroup(cgroup, ebpf.AttachCGroupInetEgress, prog, flagAllowMulti)
+	testutils.SkipIfNotSupported(t, err)
+	if err != nil {
+		t.Fatal("Can't create link:", err)
+	}
+
+	// It's currently not possible for a program to replace
+	// itself.
+	prog2 := mustLoadProgram(t, ebpf.CGroupSKB, ebpf.AttachCGroupInetEgress, "")
+	testLink(t, link, prog2)
+}
+
+func TestLinkCgroup(t *testing.T) {
+	cgroup, prog := mustCgroupFixtures(t)
+
+	link, err := newLinkCgroup(cgroup, ebpf.AttachCGroupInetEgress, prog)
+	testutils.SkipIfNotSupported(t, err)
+	if err != nil {
+		t.Fatal("Can't create link:", err)
+	}
+
+	testLink(t, link, prog)
+}