0

我一直在尝试设计一个工具，我可以在其中进行每个进程的跟踪，但这意味着我需要为每个进程提供一个键，以便我可以为每个进程存储键值映射配对。我本能地不喜欢使用结构或字符串作为键，有一段时间我在考虑如何访问 inode 值以将它们用作键。然而，有许多使用结构或字符串作为 hashmap 键的示例，Alexei 建议通常将进程名称用作键. 也就是说，我无法让这种哈希图的基本实现工作。在 BPF 程序中，跟踪点无法找到与 process_name 键相关的值。也许我正在比较内存位置而不是预期的字符串文字？c_types 的幕后是否发生了导致键之间不匹配的事情？

from bcc import BPF
from bcc.utils import printb
from bcc.syscall import syscall_name, syscalls
from ctypes import *

b = BPF(text = """

struct procName {
    char name[16];
};

BPF_HASH(attempt, struct procName, u32);

TRACEPOINT_PROBE(raw_syscalls, sys_exit)
{
    u32 *val;
    struct procName hKey;
    bpf_get_current_comm(hKey.name,16);
    val = attempt.lookup(&hKey);

    if (val)
    {
        bpf_trace_printk("Hello world, I have value %d!\\n", *val);
    }

    return 0;
}
""")

class procName(Structure):
    _fields_ = [("name", (c_char_p*16))]

myFirst = procName(('p','y','t','h','o','n','\0'))
trialUpload[myFirst] = c_int(10)

while 1:
    try:
        (task, pid, cpu, flags, ts, msg) = b.trace_fields()
    except KeyboardInterrupt:
        print("Detaching")
        exit()
    print("%-18.9f %-16s %-6d %s" % (ts, task, pid, msg))
4

1 回答 1

0

原始代码中的错误与 BCC & BPF 无关，位于我对 ctypes 的实现中。对于初学者 - 

class procName(Structure):
    _fields_ = [("name", (c_char_p*16))]

创建一个具有“名称”字段的结构。在上面的定义中，当我想要一个 char[16] 时，名称将是 *char[16] 类型。其次，虽然这

myFirst = procName(('p','y','t','h','o','n','\0'))

可能有效，它不是最佳实践初始化。这是正确的做法——

class procName(Structure):
    _fields_ = [("name", (c_char*16))]

s = "python"

mySecond = procName()
mySecond.name = s

因此，完整的程序结合了基于 process_name 的密钥和实现从 python 传递数据的密钥。

from bcc import BPF
from bcc.utils import printb
from bcc.syscall import syscall_name, syscalls
import ctypes
from ctypes import *

b = BPF(text = """

#include <linux/string.h>

struct procName {
    char name[16];
};

BPF_HASH(attempt, struct procName, u32);

TRACEPOINT_PROBE(raw_syscalls, sys_exit)
{
    u32 *myVal;
    struct procName key;
    bpf_get_current_comm(&(key.name),16);

    myVal = attempt.lookup(&key);

    if (myVal)
    {
        bpf_trace_printk("values: %d\\n", *myVal);
    }

    return 0;

}
""")

class procName(Structure):
    _fields_ = [("name", (c_char*16))]

trialUpload = b["attempt"]

s = "python"

mySecond = procName()
mySecond.name = s
trialUpload[mySecond] = c_int(5)

while 1:
    try:
        (task, pid, cpu, flags, ts, msg) = b.trace_fields()
    except KeyboardInterrupt:
        print("Detaching")
        exit()
    print("%-18.9f %-16s %-6d %s" % (ts, task, pid, msg))
于 2020-04-25T20:27:36.143 回答