-
Notifications
You must be signed in to change notification settings - Fork 7
Expand file tree
/
Copy pathkprobe_x86_bpfel.go
More file actions
197 lines (174 loc) · 4.73 KB
/
kprobe_x86_bpfel.go
File metadata and controls
197 lines (174 loc) · 4.73 KB
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
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
// Code generated by bpf2go; DO NOT EDIT.
//go:build 386 || amd64
package main
import (
"bytes"
_ "embed"
"fmt"
"io"
"structs"
"github.com/cilium/ebpf"
)
type kprobeCounterCfgValue struct {
_ structs.HostLayout
CgrpfsMagic uint64
}
type kprobeStatkey struct {
_ structs.HostLayout
Srcip struct {
_ structs.HostLayout
In6U struct {
_ structs.HostLayout
U6Addr8 [16]uint8
}
}
Dstip struct {
_ structs.HostLayout
In6U struct {
_ structs.HostLayout
U6Addr8 [16]uint8
}
}
Cgroupid uint64
Comm [16]int8
Pid int32
SrcPort uint16
DstPort uint16
Proto uint8
_ [7]byte
}
type kprobeStatvalue struct {
_ structs.HostLayout
Packets uint64
Bytes uint64
}
// loadKprobe returns the embedded CollectionSpec for kprobe.
func loadKprobe() (*ebpf.CollectionSpec, error) {
reader := bytes.NewReader(_KprobeBytes)
spec, err := ebpf.LoadCollectionSpecFromReader(reader)
if err != nil {
return nil, fmt.Errorf("can't load kprobe: %w", err)
}
return spec, err
}
// loadKprobeObjects loads kprobe and converts it into a struct.
//
// The following types are suitable as obj argument:
//
// *kprobeObjects
// *kprobePrograms
// *kprobeMaps
//
// See ebpf.CollectionSpec.LoadAndAssign documentation for details.
func loadKprobeObjects(obj interface{}, opts *ebpf.CollectionOptions) error {
spec, err := loadKprobe()
if err != nil {
return err
}
return spec.LoadAndAssign(obj, opts)
}
// kprobeSpecs contains maps and programs before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type kprobeSpecs struct {
kprobeProgramSpecs
kprobeMapSpecs
kprobeVariableSpecs
}
// kprobeProgramSpecs contains programs before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type kprobeProgramSpecs struct {
IcmpSend *ebpf.ProgramSpec `ebpf:"__icmp_send"`
Icmp6Send *ebpf.ProgramSpec `ebpf:"icmp6_send"`
IcmpRcv *ebpf.ProgramSpec `ebpf:"icmp_rcv"`
Icmpv6Rcv *ebpf.ProgramSpec `ebpf:"icmpv6_rcv"`
Ip6SendSkb *ebpf.ProgramSpec `ebpf:"ip6_send_skb"`
IpSendSkb *ebpf.ProgramSpec `ebpf:"ip_send_skb"`
SkbConsumeUdp *ebpf.ProgramSpec `ebpf:"skb_consume_udp"`
TcpCleanupRbuf *ebpf.ProgramSpec `ebpf:"tcp_cleanup_rbuf"`
TcpSendmsg *ebpf.ProgramSpec `ebpf:"tcp_sendmsg"`
}
// kprobeMapSpecs contains maps before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type kprobeMapSpecs struct {
CounterCfg *ebpf.MapSpec `ebpf:"counter_cfg"`
PktCount *ebpf.MapSpec `ebpf:"pkt_count"`
}
// kprobeVariableSpecs contains global variables before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type kprobeVariableSpecs struct {
}
// kprobeObjects contains all objects after they have been loaded into the kernel.
//
// It can be passed to loadKprobeObjects or ebpf.CollectionSpec.LoadAndAssign.
type kprobeObjects struct {
kprobePrograms
kprobeMaps
kprobeVariables
}
func (o *kprobeObjects) Close() error {
return _KprobeClose(
&o.kprobePrograms,
&o.kprobeMaps,
)
}
// kprobeMaps contains all maps after they have been loaded into the kernel.
//
// It can be passed to loadKprobeObjects or ebpf.CollectionSpec.LoadAndAssign.
type kprobeMaps struct {
CounterCfg *ebpf.Map `ebpf:"counter_cfg"`
PktCount *ebpf.Map `ebpf:"pkt_count"`
}
func (m *kprobeMaps) Close() error {
return _KprobeClose(
m.CounterCfg,
m.PktCount,
)
}
// kprobeVariables contains all global variables after they have been loaded into the kernel.
//
// It can be passed to loadKprobeObjects or ebpf.CollectionSpec.LoadAndAssign.
type kprobeVariables struct {
}
// kprobePrograms contains all programs after they have been loaded into the kernel.
//
// It can be passed to loadKprobeObjects or ebpf.CollectionSpec.LoadAndAssign.
type kprobePrograms struct {
IcmpSend *ebpf.Program `ebpf:"__icmp_send"`
Icmp6Send *ebpf.Program `ebpf:"icmp6_send"`
IcmpRcv *ebpf.Program `ebpf:"icmp_rcv"`
Icmpv6Rcv *ebpf.Program `ebpf:"icmpv6_rcv"`
Ip6SendSkb *ebpf.Program `ebpf:"ip6_send_skb"`
IpSendSkb *ebpf.Program `ebpf:"ip_send_skb"`
SkbConsumeUdp *ebpf.Program `ebpf:"skb_consume_udp"`
TcpCleanupRbuf *ebpf.Program `ebpf:"tcp_cleanup_rbuf"`
TcpSendmsg *ebpf.Program `ebpf:"tcp_sendmsg"`
}
func (p *kprobePrograms) Close() error {
return _KprobeClose(
p.IcmpSend,
p.Icmp6Send,
p.IcmpRcv,
p.Icmpv6Rcv,
p.Ip6SendSkb,
p.IpSendSkb,
p.SkbConsumeUdp,
p.TcpCleanupRbuf,
p.TcpSendmsg,
)
}
func _KprobeClose(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 kprobe_x86_bpfel.o
var _KprobeBytes []byte