mirror of
https://github.com/superseriousbusiness/gotosocial
synced 2024-12-23 19:23:12 +00:00
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
|
package link
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
|
|||
|
|
"github.com/cilium/ebpf"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// XDPAttachFlags represents how XDP program will be attached to interface.
|
|||
|
|
type XDPAttachFlags uint32
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
// XDPGenericMode (SKB) links XDP BPF program for drivers which do
|
|||
|
|
// not yet support native XDP.
|
|||
|
|
XDPGenericMode XDPAttachFlags = 1 << (iota + 1)
|
|||
|
|
// XDPDriverMode links XDP BPF program into the driver’s receive path.
|
|||
|
|
XDPDriverMode
|
|||
|
|
// XDPOffloadMode offloads the entire XDP BPF program into hardware.
|
|||
|
|
XDPOffloadMode
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type XDPOptions struct {
|
|||
|
|
// Program must be an XDP BPF program.
|
|||
|
|
Program *ebpf.Program
|
|||
|
|
|
|||
|
|
// Interface is the interface index to attach program to.
|
|||
|
|
Interface int
|
|||
|
|
|
|||
|
|
// Flags is one of XDPAttachFlags (optional).
|
|||
|
|
//
|
|||
|
|
// Only one XDP mode should be set, without flag defaults
|
|||
|
|
// to driver/generic mode (best effort).
|
|||
|
|
Flags XDPAttachFlags
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AttachXDP links an XDP BPF program to an XDP hook.
|
|||
|
|
func AttachXDP(opts XDPOptions) (Link, error) {
|
|||
|
|
if t := opts.Program.Type(); t != ebpf.XDP {
|
|||
|
|
return nil, fmt.Errorf("invalid program type %s, expected XDP", t)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if opts.Interface < 1 {
|
|||
|
|
return nil, fmt.Errorf("invalid interface index: %d", opts.Interface)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
rawLink, err := AttachRawLink(RawLinkOptions{
|
|||
|
|
Program: opts.Program,
|
|||
|
|
Attach: ebpf.AttachXDP,
|
|||
|
|
Target: opts.Interface,
|
|||
|
|
Flags: uint32(opts.Flags),
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return rawLink, err
|
|||
|
|
}
|