如何通过eBPF的BCC接口为用python编写的eBPF程序编写单元测试?

t3irkdon  于 2022-12-17  发布在  Python
关注(0)|答案(1)|浏览(176)

我知道这是一个奇怪的一个,我可能正在寻找一个凄凉的主题的帮助,因为经过几天的冲刷网无济于事,我决定接触到这个社区...
所以我需要用BCC为用Python编写的eBPF程序编写单元测试
任何建议都将受到极大的赞赏,甚至指出我的大方向,我可以进一步寻找解决我目前面临的问题的见解
下面是一个简单的例子:

from bcc import BPF

# define BPF program
prog = """
int hello(void *ctx) {
    bpf_trace_printk("Hello, World!\\n");
    return 0;
}
"""

# load BPF program
b = BPF(text=prog)
b.attach_kprobe(event=b.get_syscall_fnname("clone"), fn_name="hello")

# header
print("%-18s %-16s %-6s %s" % ("TIME(s)", "COMM", "PID", "MESSAGE"))

# format output
while 1:
    try:
        (task, pid, cpu, flags, ts, msg) = b.trace_fields()
    except ValueError:
        continue
    print("%-18.9f %-16s %-6d %s" % (ts, task, pid, msg))

我需要分别对C代码和python代码进行单元测试,以查看它们的行为是否符合预期

okxuctiv

okxuctiv1#

你可以用Python常用的方法对Python逻辑进行单元测试。对于BPF部分,有两种方法:

  • 使用另一个Python程序来加载和触发您编写的BPF程序。要触发您的程序,您可以使用BPF_PROG_TEST_RUN,或者您可以按计划附加它并触发附加点(例如,如果程序附加到clone(2)syscall,则调用clone)。
  • 你可以在头文件中将BPF程序拆分成C函数。你可以编写一个C单元测试,其中包含该头文件,并使用要检查的输入来调用这些函数。要使其工作,你需要确保C函数不调用任何可能被编译到非BPF后端的BPF特定函数。给定Map和帮助器,这可能有点困难。

相关问题