找不到XDP源码中的头文件进行Go程序编译

在 Go 程序中编译 XDP 模块时，遇到了找不到 bpf/bpf_helpers.h 头文件的问题。此错误是由 XDP 模块造成的，因为它使用了该头文件中的宏和函数。该头文件不在 github.com/iovisor/gobpf/tree/master/elf/include 中。建议使用 Cilium 的 eBPF 库，它提供了加载 .o 文件到接口并与 bpf 映射交互的示例程序。确保 XDP 函数以 sec("xdp") 开头，并使用 ebpf.LoadCollectionSpec 加载程序。

我的 xdp 模块自行编译，但在 go 程序上下文中编译失败，因为它找不到 bpf/bpf_helpers.h 文件。这是导致问题的代码：

package main

import (
    "fmt"
    bpf "github.com/iovisor/gobpf/bcc"
    log "github.com/sirupsen/logrus"
    "io/ioutil"
    "os"
)

/*
#cgo cflags: -i/usr/include/bcc/compat
#cgo ldflags: -lbcc
#include <bcc/bcc_common.h>
#include <bcc/libbpf.h>
void perf_reader_free(void *ptr);
*/
import "c"

func main() {
    // get the source code from disk
    source, err := ioutil.readfile("xdp/collect_ips.c")

    if err != nil {
        log.fatalln("cannot read collect_ips.c")
    }

    // compile module
    module := bpf.newmodule(string(source), []string{
        "-wall",
        "-o2",
    })
    defer module.close()

    // load module
    fn, err := module.load("collect_ips", c.bpf_prog_type_xdp, 1, 65536) // problem happens here

    // ...

go 程序编译得很好，但是当我运行该程序时，我得到了这个：

/virtual/main.c:2:10: fatal error: 'bpf/bpf_helpers.h' file not found
#include <bpf/bpf_helpers.h>
         ^~~~~~~~~~~~~~~~~~~
9 warnings and 1 error generated.
panic: runtime error: invalid memory address or nil pointer dereference
        panic: runtime error: invalid memory address or nil pointer dereference
[signal sigsegv: segmentation violation code=0x1 addr=0x0 pc=0x4b23d4]

goroutine 1 [running]:
github.com/iovisor/gobpf/bcc.(*module).close.func1(0x203000)
        /home/me/go/pkg/mod/github.com/iovisor/[email protected]/bcc/module.go:132 +0x14
github.com/iovisor/gobpf/bcc.(*module).close(0x0)
        /home/me/go/pkg/mod/github.com/iovisor/[email protected]/bcc/module.go:132 +0x36
panic({0x4da720, 0x59ec60})
        /usr/lib/go-1.17/src/runtime/panic.go:1038 +0x215
github.com/iovisor/gobpf/bcc.(*module).load(0x0, {0x4ef9f2, 0xb}, 0x2, 0x2, 0x8001)
        /home/me/go/pkg/mod/github.com/iovisor/[email protected]/bcc/module.go:202 +0x36
main.main()
        /home/me/go/src/code.squarespace.net/net/prism-stream/cmd/server/main.go:35 +0x1b7

这个问题是由于我的 xdp 模块而发生的，因为如果我在 c 源代码中注释掉这个头文件，错误就会转移到另一个头文件。

我认为发生这种情况是因为 bpf_helpers.h 在这里不存在 https://github.com/iovisor/gobpf/tree/master/elf/include。如果这是问题，有没有办法使用 /usr/include/bpf 中的头文件？

如果我从 xdp 代码中取出 bpf_helpers.h，我会收到一条错误消息，抱怨在我的代码中使用了 sec：

struct bpf_map_def SEC("maps") addr_map = {
    .type = BPF_MAP_TYPE_LRU_HASH,
    .key_size = sizeof(struct addr_desc_struct),
    .value_size = sizeof(long),
    .max_entries = 4096
};

我将 sec 的宏从 bpf_helpers.h 复制到我的代码中，但随后出现 error: 变量具有不完整类型 'struct bpf_map_def'。我还使用 bpf_map_lookup_elem() 和 bpf_map_update_elem()，它们在 bpf/ 目录中定义。

正确答案

我使用了 eBPF library from Cilium。这是一个示例程序，它将 .o 文件加载到接口，等待 10 秒，然后打印 bpf 映射中的所有内容。

您必须确保您的 xdp 函数前面有 sec("xdp")。而如果您使用 xdp-loader 加载程序，则可以将您想要的任何内容传递给 sec，只要它与函数名称不同即可。

package main

import (
    "bytes"
    "fmt"
    "github.com/cilium/ebpf"
    log "github.com/sirupsen/logrus"
    "github.com/vishvananda/netlink"
    "time"
)


func main() {
    var objects struct {
        XCProg  *ebpf.Program `ebpf:"collect_ips"` // Name of function containing the XDP program
        XCMap   *ebpf.Map     `ebpf:"addr_map"` // Name of the map 
    }

    spec, err := ebpf.LoadCollectionSpec("dist/collect_ips.o")

    if err != nil {
        log.Fatalln("ebpf.LoadCollectionSpec", err)
    }

    if err := spec.LoadAndAssign(&objects, nil); err != nil {
        log.Fatalln("ebpf.LoadAndAssign", err)
    }

    link, err := netlink.LinkByName("enp0s8")

    if err != nil {
        log.Fatalln("netlink.LinkByName", err)
    }

    err = netlink.LinkSetXdpFdWithFlags(link, objects.XCProg.FD(), 2)

    if err != nil {
        log.Fatalln("netlink.LinkSetXdpFdWithFlags")
    }

    time.Sleep(time.Second*10)

    var keyBytes, valueBytes []byte

    iterator := objects.XCMap.Iterate()

    if iterator == nil {
        log.Fatalln("nil iterator")
    }

    for iterator.Next(&keyBytes, &valueBytes) != false {
        fmt.Printf("%x: %x", keyBytes, valueBytes)
    }

    defer objects.XCProg.Close()
    defer objects.XCMap.Close()
}

好了，本文到此结束，带大家了解了《找不到XDP源码中的头文件进行Go程序编译》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！