-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdrop_x86_bpfel.go
137 lines (119 loc) · 3.33 KB
/
drop_x86_bpfel.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
// Code generated by bpf2go; DO NOT EDIT.
//go:build 386 || amd64
package main
import (
"bytes"
_ "embed"
"fmt"
"io"
"github.com/cilium/ebpf"
)
// loadDrop returns the embedded CollectionSpec for drop.
func loadDrop() (*ebpf.CollectionSpec, error) {
reader := bytes.NewReader(_DropBytes)
spec, err := ebpf.LoadCollectionSpecFromReader(reader)
if err != nil {
return nil, fmt.Errorf("can't load drop: %w", err)
}
return spec, err
}
// loadDropObjects loads drop and converts it into a struct.
//
// The following types are suitable as obj argument:
//
// *dropObjects
// *dropPrograms
// *dropMaps
//
// See ebpf.CollectionSpec.LoadAndAssign documentation for details.
func loadDropObjects(obj interface{}, opts *ebpf.CollectionOptions) error {
spec, err := loadDrop()
if err != nil {
return err
}
return spec.LoadAndAssign(obj, opts)
}
// dropSpecs contains maps and programs before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type dropSpecs struct {
dropProgramSpecs
dropMapSpecs
}
// dropSpecs contains programs before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type dropProgramSpecs struct {
DropPackets *ebpf.ProgramSpec `ebpf:"drop_packets"`
KprobeInetBind *ebpf.ProgramSpec `ebpf:"kprobe_inet_bind"`
}
// dropMapSpecs contains maps before they are loaded into the kernel.
//
// It can be passed ebpf.CollectionSpec.Assign.
type dropMapSpecs struct {
AssignedPortMap *ebpf.MapSpec `ebpf:"assigned_port_map"`
IsReqProcessMap *ebpf.MapSpec `ebpf:"is_req_process_map"`
PktCount *ebpf.MapSpec `ebpf:"pkt_count"`
PortMap *ebpf.MapSpec `ebpf:"port_map"`
ProcNameLenMap *ebpf.MapSpec `ebpf:"proc_name_len_map"`
ProcNameMap *ebpf.MapSpec `ebpf:"proc_name_map"`
}
// dropObjects contains all objects after they have been loaded into the kernel.
//
// It can be passed to loadDropObjects or ebpf.CollectionSpec.LoadAndAssign.
type dropObjects struct {
dropPrograms
dropMaps
}
func (o *dropObjects) Close() error {
return _DropClose(
&o.dropPrograms,
&o.dropMaps,
)
}
// dropMaps contains all maps after they have been loaded into the kernel.
//
// It can be passed to loadDropObjects or ebpf.CollectionSpec.LoadAndAssign.
type dropMaps struct {
AssignedPortMap *ebpf.Map `ebpf:"assigned_port_map"`
IsReqProcessMap *ebpf.Map `ebpf:"is_req_process_map"`
PktCount *ebpf.Map `ebpf:"pkt_count"`
PortMap *ebpf.Map `ebpf:"port_map"`
ProcNameLenMap *ebpf.Map `ebpf:"proc_name_len_map"`
ProcNameMap *ebpf.Map `ebpf:"proc_name_map"`
}
func (m *dropMaps) Close() error {
return _DropClose(
m.AssignedPortMap,
m.IsReqProcessMap,
m.PktCount,
m.PortMap,
m.ProcNameLenMap,
m.ProcNameMap,
)
}
// dropPrograms contains all programs after they have been loaded into the kernel.
//
// It can be passed to loadDropObjects or ebpf.CollectionSpec.LoadAndAssign.
type dropPrograms struct {
DropPackets *ebpf.Program `ebpf:"drop_packets"`
KprobeInetBind *ebpf.Program `ebpf:"kprobe_inet_bind"`
}
func (p *dropPrograms) Close() error {
return _DropClose(
p.DropPackets,
p.KprobeInetBind,
)
}
func _DropClose(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 drop_x86_bpfel.o
var _DropBytes []byte