登录
首页 >  Golang >  Go问答

bpf调用加载eBPF程序时参数列表太长怎么解决

来源:Golang技术栈

时间:2023-03-06 18:40:29 141浏览 收藏

从现在开始,努力学习吧!本文《bpf调用加载eBPF程序时参数列表太长怎么解决》主要讲解了golang等等相关知识点,我会在golang学习网中持续更新相关的系列文章,欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧,希望能帮到你!

问题内容

我正在尝试通过bpfGo 中的系统调用加载 eBPF 程序,但我看到系统调用返回错误。为了限制问题,我使用了以下最小的 eBPF 程序,它什么都不做:

struct task_group {};

Go程序的重要部分如下:

b, err := ioutil.ReadFile("bpf/bbf_tty.o")
if err != nil {
    fmt.Print(err)
}

progType := BPF_PROG_TYPE_KPROBE
insns := unsafe.Pointer(&b)
insnCnt := len(b)

lba := struct {
    progType    uint32
    pad0        [4]byte
    insnCnt     uint32
    pad1        [4]byte
    insns       uint64
    license     uint64
    logLevel    uint32
    pad2        [4]byte
    logSize     uint32
    pad3        [4]byte
    logBuf      uint64
    kernVersion uint32
    pad4        [4]byte
}{
    progType:    uint32(progType),
    insns:       uint64(uintptr(insns)),
    insnCnt:     uint32(insnCnt),
    license:     uint64(uintptr(0)),
    logBuf:      uint64(uintptr(0)),
    logSize:     uint32(0),
    logLevel:    uint32(0),
    kernVersion: uint32(4),
}

ret, _, err := unix.Syscall(
    unix.SYS_BPF,
    bpf.BPF_PROG_LOAD,
    uintptr(unsafe.Pointer(&lba)),
    unsafe.Sizeof(lba),
)

if ret != 0 || err != 0 {
    return fmt.Errorf("Unable to load program: %s", err)
}

但是,返回的错误是Unable to load program: argument list too long. 为什么是这样?或者更好的是,如何获得更详细的输出来找出问题的根本原因?

从[这里开始](http://elixir.free- electrons.com/linux/latest/source/kernel/bpf/syscall.c),只有三个地方E2BIG(参数列表太长)从bpf系统调用返回,但似乎没有一个适合。

如果需要,我可以提供更完整的代码版本,为了简洁起见,我只是试图去掉不相关的部分。

为了帮助重现这个问题,我在下面包含了我的完整 BPF 程序。完整的回购在这里:

#include 
#include 
#include 

#include 

#include 
#include 

#include 
#include 

#include "lib/utils.h"
#include "lib/common.h"
#include "lib/maps.h"
#include "lib/xdp.h"
#include "lib/eps.h"
#include "lib/events.h"

// define structures
enum pid_type
{
    PIDTYPE_PID,
    PIDTYPE_PGID,
    PIDTYPE_SID,
    PIDTYPE_MAX,
    // only valid to __task_pid_nr_ns() 
    __PIDTYPE_TGID
};
struct upid {
  int nr;
};
struct pid
{
  struct upid numbers[1];
};
struct pid_link
{
  struct pid *pid;
};
struct task_group {
};
struct task_struct {
  struct task_struct *group_leader;
  struct pid_link           pids[PIDTYPE_MAX];
};
struct sid_t {
    int sid;
};

#define BUFSIZE 256
struct tty_write_t {
    int count;
    char buf[BUFSIZE];
    unsigned int sessionid;
};

// define maps
struct bpf_elf_map __section_maps active_sids = {
    .type       = BPF_MAP_TYPE_HASH,
    .size_key   = sizeof(struct sid_t),
    .size_value = sizeof(uint64_t),
};

struct bpf_elf_map __section_maps tty_writes = {
    .type       = BPF_MAP_TYPE_PERF_EVENT_ARRAY,
};

// save_sid saves a sessionid generated from a call
// to setsid to the active_sids map
int save_sid(struct pt_regs *ctx) {

    struct sid_t sid_struct = {};
    int sid = PT_REGS_RC(ctx);
    uint64_t time_ns = bpf_ktime_get_ns();

    sid_struct.sid = sid;

    bpf_map_update(&sid_struct, &time_ns);

    return 0;

}

//int kprobe__tty_write(struct pt_regs *ctx, struct file *file, const char __user *buf, size_t count)
int kprobe__tty_write(struct pt_regs *ctx, struct file *file, const char *buf, size_t count)
{
    struct task_struct *task;
    struct pid_link pid_link;
    struct pid pid;
    int sessionid;

    // get current sessionid
    task = (struct task_struct *)bpf_get_current_task();
    bpf_probe_read(&pid_link, sizeof(pid_link), (void *)&task->group_leader->pids[PIDTYPE_SID]);
    bpf_probe_read(&pid, sizeof(pid), (void *)pid_link.pid);
    sessionid = pid.numbers[0].nr;

    // build session struct key
    struct sid_t sid_key;
    sid_key.sid = sessionid;

    // if sid does not exist in our map then return
    //u64 *time_ns = active_sids.lookup(&sid_key);
    //if (!time_ns) {
    //    return 0;
    //}

    // bpf_probe_read() can only use a fixed size, so truncate to count
    // in user space:
    struct tty_write_t tty_write = {};
    bpf_probe_read(&tty_write.buf, BUFSIZE, (void *)buf);
    if (count > BUFSIZE) {
        tty_write.count = BUFSIZE;
    } else {
        tty_write.count = count;
    }

    // add sessionid to tty_write structure and submit
    tty_write.sessionid = sessionid;
    bpf_perf_event_output(ctx, &tty_write, sizeof(tty_write));

    return 0;
}

正确答案

您的问题是您尝试加载 BPF 字节码的方式。

b, err := ioutil.ReadFile("bpf/bbf_tty.o")

我从未使用过 Go,但据我了解,这会从 ELF 目标文件中读取所有字节,无需任何特定处理,然后在您的代码中将它们提供给bpf()系统调用。

问题是,事情不是这样工作的:当它编译成 eBPF 时,clang 将你的程序放入 一个特定的部分(默认情况下,.text,但你可以指定另一个名称)。此外,如果您使用 eBPF 映射,则会发生一些神奇的事情(“ap 重定位”),以便您的 ELF 文件可以嵌入映射信息,并且调用的用户空间程序bpf()可以检索它并将其发送到内核。

因此,当您加载整个文件以将其发送到 时bpf(),您会加载实际的字节码,以及所有 ELF 部分和标题。内核可能不太喜欢它。我不知道如何在 Go 中修复它,但这里有一些可能有用的提示:

  • libbpf,一个可以从 ELF 文件加载 eBPF 程序的 C 库:位于内核树中。

  • Gobpf,一些使用 Go 的 eBPF 程序的框架(链接)。我从未使用过它,但他们肯定会有一些代码来从目标文件加载程序吗?

今天带大家了解了golang的相关知识,希望对你有所帮助;关于Golang的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>