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 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
|
package link
import (
"io"
"testing"
"github.com/cilium/ebpf"
"github.com/cilium/ebpf/internal/testutils"
)
func TestIter(t *testing.T) {
testutils.SkipOnOldKernel(t, "5.9", "bpf_map iter")
prog := mustLoadProgram(t, ebpf.Tracing, ebpf.AttachTraceIter, "bpf_map")
it, err := AttachIter(IterOptions{
Program: prog,
})
if err != nil {
t.Fatal("Can't create iter:", err)
}
file, err := it.Open()
if err != nil {
t.Fatal("Can't open iter instance:", err)
}
defer file.Close()
contents, err := io.ReadAll(file)
if err != nil {
t.Fatal(err)
}
if len(contents) != 0 {
t.Error("Non-empty output from no-op iterator:", string(contents))
}
testLink(t, it, prog)
}
func TestIterMapElements(t *testing.T) {
testutils.SkipOnOldKernel(t, "5.9", "bpf_map_elem iter")
prog := mustLoadProgram(t, ebpf.Tracing, ebpf.AttachTraceIter, "bpf_map_elem")
arr, err := ebpf.NewMap(&ebpf.MapSpec{
Type: ebpf.Array,
KeySize: 4,
ValueSize: 4,
MaxEntries: 3,
})
if err != nil {
t.Fatal(err)
}
defer arr.Close()
it, err := AttachIter(IterOptions{
Program: prog,
Map: arr,
})
if err != nil {
t.Fatal("Can't create iter:", err)
}
defer it.Close()
file, err := it.Open()
if err != nil {
t.Fatal("Can't open iter instance:", err)
}
defer file.Close()
contents, err := io.ReadAll(file)
if err != nil {
t.Fatal(err)
}
if len(contents) != 0 {
t.Error("Non-empty output from no-op iterator:", string(contents))
}
}
func TestUDPIter(t *testing.T) {
// Introduced by 5788b3a07fc5 ("net: bpf: Implement bpf iterator for udp")
testutils.SkipOnOldKernel(t, "5.9", "udp iter")
prog := mustLoadProgram(t, ebpf.Tracing, ebpf.AttachTraceIter, "udp")
it, err := AttachIter(IterOptions{
Program: prog,
})
if err != nil {
t.Fatal("Can't create iter:", err)
}
file, err := it.Open()
if err != nil {
t.Fatal("Can't open iter instance:", err)
}
defer file.Close()
contents, err := io.ReadAll(file)
if err != nil {
t.Fatal(err)
}
if len(contents) != 0 {
t.Error("Non-empty output from no-op iterator:", string(contents))
}
testLink(t, it, prog)
}
|