1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
package link
import (
"fmt"
"math"
"net"
"testing"
"github.com/go-quicktest/qt"
"github.com/cilium/ebpf"
"github.com/cilium/ebpf/internal/testutils"
"github.com/cilium/ebpf/internal/unix"
)
func TestAttachTCX(t *testing.T) {
testutils.SkipOnOldKernel(t, "6.6", "TCX link")
prog := mustLoadProgram(t, ebpf.SchedCLS, ebpf.AttachNone, "")
link, _ := mustAttachTCX(t, prog, ebpf.AttachTCXIngress)
testLink(t, link, prog)
}
func TestTCXAnchor(t *testing.T) {
testutils.SkipOnOldKernel(t, "6.6", "TCX link")
a := mustLoadProgram(t, ebpf.SchedCLS, ebpf.AttachNone, "")
b := mustLoadProgram(t, ebpf.SchedCLS, ebpf.AttachNone, "")
linkA, iface := mustAttachTCX(t, a, ebpf.AttachTCXEgress)
programInfo, err := a.Info()
qt.Assert(t, qt.IsNil(err))
programID, _ := programInfo.ID()
linkInfo, err := linkA.Info()
qt.Assert(t, qt.IsNil(err))
linkID := linkInfo.ID
for _, anchor := range []Anchor{
Head(),
Tail(),
BeforeProgram(a),
BeforeProgramByID(programID),
AfterLink(linkA),
AfterLinkByID(linkID),
} {
t.Run(fmt.Sprintf("%T", anchor), func(t *testing.T) {
linkB, err := AttachTCX(TCXOptions{
Program: b,
Attach: ebpf.AttachTCXEgress,
Interface: iface,
Anchor: anchor,
})
qt.Assert(t, qt.IsNil(err))
qt.Assert(t, qt.IsNil(linkB.Close()))
})
}
}
func TestTCXExpectedRevision(t *testing.T) {
testutils.SkipOnOldKernel(t, "6.6", "TCX link")
iface, err := net.InterfaceByName("lo")
qt.Assert(t, qt.IsNil(err))
_, err = AttachTCX(TCXOptions{
Program: mustLoadProgram(t, ebpf.SchedCLS, ebpf.AttachNone, ""),
Attach: ebpf.AttachTCXEgress,
Interface: iface.Index,
ExpectedRevision: math.MaxUint64,
})
qt.Assert(t, qt.ErrorIs(err, unix.ESTALE))
}
func mustAttachTCX(tb testing.TB, prog *ebpf.Program, attachType ebpf.AttachType) (Link, int) {
iface, err := net.InterfaceByName("lo")
qt.Assert(tb, qt.IsNil(err))
link, err := AttachTCX(TCXOptions{
Program: prog,
Attach: attachType,
Interface: iface.Index,
})
qt.Assert(tb, qt.IsNil(err))
tb.Cleanup(func() { qt.Assert(tb, qt.IsNil(link.Close())) })
return link, iface.Index
}
|
