Katyusha's blog

1
// Code generated by bpf2go; DO NOT EDIT.
2
//go:build 386 || amd64 || arm || arm64 || loong64 || mips64le || mipsle || ppc64le || riscv64
3

4
package collector
5

6
import (
7
  "bytes"
8
  _ "embed"
9
  "fmt"
10
  "io"
11

12
  "github.com/cilium/ebpf"
13
)
14

15
type cpuTaskStat struct {
16
  RunNs  uint64
17
  RunqNs uint64
18
  Ctx    uint64
19
  Comm   [16]int8
20
}
21

22
// loadCpu returns the embedded CollectionSpec for cpu.
23
func loadCpu() (*ebpf.CollectionSpec, error) {
24
  reader := bytes.NewReader(_CpuBytes)
25
  spec, err := ebpf.LoadCollectionSpecFromReader(reader)
26
  if err != nil {
27
    return nil, fmt.Errorf("can't load cpu: %w", err)
28
  }
29

30
  return spec, err
31
}
32

33
// loadCpuObjects loads cpu and converts it into a struct.
34
//
35
// The following types are suitable as obj argument:
36
//
37
//  *cpuObjects
38
//  *cpuPrograms
39
//  *cpuMaps
40
//
41
// See ebpf.CollectionSpec.LoadAndAssign documentation for details.
42
func loadCpuObjects(obj interface{}, opts *ebpf.CollectionOptions) error {
43
  spec, err := loadCpu()
44
  if err != nil {
45
    return err
46
  }
47

48
  return spec.LoadAndAssign(obj, opts)
49
}
50

51
// cpuSpecs contains maps and programs before they are loaded into the kernel.
52
//
53
// It can be passed ebpf.CollectionSpec.Assign.
54
type cpuSpecs struct {
55
  cpuProgramSpecs
56
  cpuMapSpecs
57
}
58

59
// cpuSpecs contains programs before they are loaded into the kernel.
60
//
61
// It can be passed ebpf.CollectionSpec.Assign.
62
type cpuProgramSpecs struct {
63
  HandleSwitch *ebpf.ProgramSpec `ebpf:"handle_switch"`
64
  HandleWakeup *ebpf.ProgramSpec `ebpf:"handle_wakeup"`
65
}
66

67
// cpuMapSpecs contains maps before they are loaded into the kernel.
68
//
69
// It can be passed ebpf.CollectionSpec.Assign.
70
type cpuMapSpecs struct {
71
  OncpuStart *ebpf.MapSpec `ebpf:"oncpu_start"`
72
  Stats      *ebpf.MapSpec `ebpf:"stats"`
73
  WakeupTs   *ebpf.MapSpec `ebpf:"wakeup_ts"`
74
}
75

76
// cpuObjects contains all objects after they have been loaded into the kernel.
77
//
78
// It can be passed to loadCpuObjects or ebpf.CollectionSpec.LoadAndAssign.
79
type cpuObjects struct {
80
  cpuPrograms
81
  cpuMaps
82
}
83

84
func (o *cpuObjects) Close() error {
85
  return _CpuClose(
86
    &o.cpuPrograms,
87
    &o.cpuMaps,
88
  )
89
}
90

91
// cpuMaps contains all maps after they have been loaded into the kernel.
92
//
93
// It can be passed to loadCpuObjects or ebpf.CollectionSpec.LoadAndAssign.
94
type cpuMaps struct {
95
  OncpuStart *ebpf.Map `ebpf:"oncpu_start"`
96
  Stats      *ebpf.Map `ebpf:"stats"`
97
  WakeupTs   *ebpf.Map `ebpf:"wakeup_ts"`
98
}
99

100
func (m *cpuMaps) Close() error {
101
  return _CpuClose(
102
    m.OncpuStart,
103
    m.Stats,
104
    m.WakeupTs,
105
  )
106
}
107

108
// cpuPrograms contains all programs after they have been loaded into the kernel.
109
//
110
// It can be passed to loadCpuObjects or ebpf.CollectionSpec.LoadAndAssign.
111
type cpuPrograms struct {
112
  HandleSwitch *ebpf.Program `ebpf:"handle_switch"`
113
  HandleWakeup *ebpf.Program `ebpf:"handle_wakeup"`
114
}
115

116
func (p *cpuPrograms) Close() error {
117
  return _CpuClose(
118
    p.HandleSwitch,
119
    p.HandleWakeup,
120
  )
121
}
122

123
func _CpuClose(closers ...io.Closer) error {
124
  for _, closer := range closers {
125
    if err := closer.Close(); err != nil {
126
      return err
127
    }
128
  }
129
  return nil
130
}
131

132
// Do not access this directly.
133
//
134
//go:embed cpu_bpfel.o
135
var _CpuBytes []byte