mirror of
https://github.com/k3s-io/k3s.git
synced 2024-06-07 19:41:36 +00:00
77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
|
package link
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/cilium/ebpf"
|
||
|
"github.com/cilium/ebpf/internal"
|
||
|
)
|
||
|
|
||
|
type RawAttachProgramOptions struct {
|
||
|
// File descriptor to attach to. This differs for each attach type.
|
||
|
Target int
|
||
|
// Program to attach.
|
||
|
Program *ebpf.Program
|
||
|
// Program to replace (cgroups).
|
||
|
Replace *ebpf.Program
|
||
|
// Attach must match the attach type of Program (and Replace).
|
||
|
Attach ebpf.AttachType
|
||
|
// Flags control the attach behaviour. This differs for each attach type.
|
||
|
Flags uint32
|
||
|
}
|
||
|
|
||
|
// RawAttachProgram is a low level wrapper around BPF_PROG_ATTACH.
|
||
|
//
|
||
|
// You should use one of the higher level abstractions available in this
|
||
|
// package if possible.
|
||
|
func RawAttachProgram(opts RawAttachProgramOptions) error {
|
||
|
if err := haveProgAttach(); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
var replaceFd uint32
|
||
|
if opts.Replace != nil {
|
||
|
replaceFd = uint32(opts.Replace.FD())
|
||
|
}
|
||
|
|
||
|
attr := internal.BPFProgAttachAttr{
|
||
|
TargetFd: uint32(opts.Target),
|
||
|
AttachBpfFd: uint32(opts.Program.FD()),
|
||
|
ReplaceBpfFd: replaceFd,
|
||
|
AttachType: uint32(opts.Attach),
|
||
|
AttachFlags: uint32(opts.Flags),
|
||
|
}
|
||
|
|
||
|
if err := internal.BPFProgAttach(&attr); err != nil {
|
||
|
return fmt.Errorf("can't attach program: %s", err)
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
type RawDetachProgramOptions struct {
|
||
|
Target int
|
||
|
Program *ebpf.Program
|
||
|
Attach ebpf.AttachType
|
||
|
}
|
||
|
|
||
|
// RawDetachProgram is a low level wrapper around BPF_PROG_DETACH.
|
||
|
//
|
||
|
// You should use one of the higher level abstractions available in this
|
||
|
// package if possible.
|
||
|
func RawDetachProgram(opts RawDetachProgramOptions) error {
|
||
|
if err := haveProgAttach(); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
attr := internal.BPFProgDetachAttr{
|
||
|
TargetFd: uint32(opts.Target),
|
||
|
AttachBpfFd: uint32(opts.Program.FD()),
|
||
|
AttachType: uint32(opts.Attach),
|
||
|
}
|
||
|
if err := internal.BPFProgDetach(&attr); err != nil {
|
||
|
return fmt.Errorf("can't detach program: %s", err)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|