-
Notifications
You must be signed in to change notification settings - Fork 1
/
counter_bpfeb.go
150 lines (128 loc) · 3.55 KB
/
counter_bpfeb.go
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
// Code generated by bpf2go; DO NOT EDIT.
//go:build mips || mips64 || ppc64 || s390x
package main
import (
"bytes"
_ "embed"
"fmt"
"io"
"github.com/cilium/ebpf"
)
type counterStatkey struct {
Srcip struct{ In6U struct{ U6Addr8 [16]uint8 } }
Dstip struct{ In6U struct{ U6Addr8 [16]uint8 } }
SrcPort uint16
DstPort uint16
Proto uint8
_ [3]byte
}
type counterStatvalue struct {
Packets uint64
Bytes uint64
}
// loadCounter returns the embedded CollectionSpec for counter.
func loadCounter() (*ebpf.CollectionSpec, error) {
reader := bytes.NewReader(_CounterBytes)
spec, err := ebpf.LoadCollectionSpecFromReader(reader)
if err != nil {
return nil, fmt.Errorf("can't load counter: %w", err)
}
return spec, err
}
// loadCounterObjects loads counter and converts it into a struct.
//
// The following types are suitable as obj argument:
//
// *counterObjects
// *counterPrograms
// *counterMaps
//
// See ebpf.CollectionSpec.LoadAndAssign documentation for details.
func loadCounterObjects(obj interface{}, opts *ebpf.CollectionOptions) error {
spec, err := loadCounter()
if err != nil {
return err
}
return spec.LoadAndAssign(obj, opts)
}
// counterSpecs contains maps and programs before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type counterSpecs struct {
counterProgramSpecs
counterMapSpecs
counterVariableSpecs
}
// counterProgramSpecs contains programs before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type counterProgramSpecs struct {
TcCountPackets *ebpf.ProgramSpec `ebpf:"tc_count_packets"`
XdpCountPackets *ebpf.ProgramSpec `ebpf:"xdp_count_packets"`
}
// counterMapSpecs contains maps before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type counterMapSpecs struct {
PktCount *ebpf.MapSpec `ebpf:"pkt_count"`
}
// counterVariableSpecs contains global variables before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type counterVariableSpecs struct {
}
// counterObjects contains all objects after they have been loaded into the kernel.
//
// It can be passed to loadCounterObjects or ebpf.CollectionSpec.LoadAndAssign.
type counterObjects struct {
counterPrograms
counterMaps
counterVariables
}
func (o *counterObjects) Close() error {
return _CounterClose(
&o.counterPrograms,
&o.counterMaps,
)
}
// counterMaps contains all maps after they have been loaded into the kernel.
//
// It can be passed to loadCounterObjects or ebpf.CollectionSpec.LoadAndAssign.
type counterMaps struct {
PktCount *ebpf.Map `ebpf:"pkt_count"`
}
func (m *counterMaps) Close() error {
return _CounterClose(
m.PktCount,
)
}
// counterVariables contains all global variables after they have been loaded into the kernel.
//
// It can be passed to loadCounterObjects or ebpf.CollectionSpec.LoadAndAssign.
type counterVariables struct {
}
// counterPrograms contains all programs after they have been loaded into the kernel.
//
// It can be passed to loadCounterObjects or ebpf.CollectionSpec.LoadAndAssign.
type counterPrograms struct {
TcCountPackets *ebpf.Program `ebpf:"tc_count_packets"`
XdpCountPackets *ebpf.Program `ebpf:"xdp_count_packets"`
}
func (p *counterPrograms) Close() error {
return _CounterClose(
p.TcCountPackets,
p.XdpCountPackets,
)
}
func _CounterClose(closers ...io.Closer) error {
for _, closer := range closers {
if err := closer.Close(); err != nil {
return err
}
}
return nil
}
// Do not access this directly.
//
//go:embed counter_bpfeb.o
var _CounterBytes []byte