optimize: adapt to bpf legacy

This commit is contained in:
mzz2017
2023-01-30 23:21:00 +08:00
parent 52836fa6a8
commit 3602e2e26b
6 changed files with 421 additions and 294 deletions

View File

@ -0,0 +1,31 @@
// Copied from https://github.com/cilium/ebpf/blob/v0.10.0/example_sock_elf_test.go
package internal
import (
"encoding/binary"
"syscall"
"unsafe"
)
func OpenRawSock(index int) (int, error) {
sock, err := syscall.Socket(syscall.AF_PACKET, syscall.SOCK_RAW|syscall.SOCK_NONBLOCK|syscall.SOCK_CLOEXEC, int(Htons(syscall.ETH_P_ALL)))
if err != nil {
return 0, err
}
sll := syscall.SockaddrLinklayer{
Ifindex: index,
Protocol: Htons(syscall.ETH_P_ALL),
}
if err := syscall.Bind(sock, &sll); err != nil {
return 0, err
}
return sock, nil
}
// Htons converts the unsigned short integer hostshort from host byte order to network byte order.
func Htons(i uint16) uint16 {
b := make([]byte, 2)
binary.BigEndian.PutUint16(b, i)
return *(*uint16)(unsafe.Pointer(&b[0]))
}