如何在fentry/htab_map_update_elem钩子中判断指定BPF Map？

我希望使用fentry/htab_map_update_elem钩子，在特定Map插入值后执行操作。请问如何判断传入的struct bpf_map* map是否为我在BPF代码中声明的Map？示例代码如下：

内核BPF程序 #

struct {
    __uint(type, BPF_MAP_TYPE_HASH);
    __uint(max_entries, 128);
    __type(key, __u32);
    __type(value, __u32);
} mymap SEC("\.maps");

SEC("fentry/htab_map_update_elem")
int mymap_prog(struct bpf_map* map, void* key, void* value, u64 map_flags) {
    // 此处如何判断map == mymap？
}

用户态程序 #

int main() {
    struct mymap_bpf* skel;
    skel = ...

    // 更新mymap并触发Map更新钩子
    bpf_map__update_elem(skel->maps.mymap, ...);
}

（Skeleton头文件由bpftool生成）

方法一：对比Map ID（推荐，兼容性好） #

内核中每个BPF Map都有唯一ID，我们可以在用户态获取目标Map的ID，传递给内核BPF程序，再在钩子中通过bpf_map_get_id()获取传入Map的ID进行对比。

修改后的内核BPF程序：

struct {
    __uint(type, BPF_MAP_TYPE_HASH);
    __uint(max_entries, 128);
    __type(key, __u32);
    __type(value, __u32);
} mymap SEC("\.maps");

// 存储目标Map的ID，由用户态初始化
struct {
    __uint(type, BPF_MAP_TYPE_ARRAY);
    __uint(max_entries, 1);
    __type(key, __u32);
    __type(value, __u32);
} target_map_id SEC("\.maps");

SEC("fentry/htab_map_update_elem")
int mymap_prog(struct bpf_map* map, void* key, void* value, u64 map_flags) {
    __u32 id_key = 0;
    __u32 *target_id = bpf_map_lookup_elem(&target_map_id, &id_key);
    __u32 current_id;

    if (!target_id)
        return 0;

    // 获取当前传入Map的ID
    if (bpf_map_get_id(map, &current_id) != 0)
        return 0;

    // ID一致则执行后续操作
    if (current_id == *target_id) {
        // 这里编写你的自定义操作逻辑
    }

    return 0;
}

用户态补充初始化逻辑：

int main() {
    struct mymap_bpf* skel;
    __u32 id_key = 0;
    __u32 map_id;

    skel = mymap_bpf__open_and_load();
    if (!skel) {
        // 错误处理
        return -1;
    }

    // 获取mymap的ID并写入target_map_id
    map_id = bpf_map__id(skel->maps.mymap);
    bpf_map__update_elem(skel->maps.target_map_id, &id_key, &map_id, BPF_ANY);

    // 挂载BPF程序
    if (mymap_bpf__attach(skel) != 0) {
        mymap_bpf__destroy(skel);
        return -1;
    }

    // 更新mymap触发钩子
    bpf_map__update_elem(skel->maps.mymap, ...);

    // 后续逻辑
    mymap_bpf__destroy(skel);
    return 0;
}

方法二：直接对比Map指针（简单但有局限性） #

BPF程序编译时，mymap的内存地址是确定的，可以直接在钩子中对比指针：

SEC("fentry/htab_map_update_elem")
int mymap_prog(struct bpf_map* map, void* key, void* value, u64 map_flags) {
    if (map == &mymap) {
        // 执行你的自定义操作
    }
    return 0;
}

注意：这种方法依赖内核不改动Map的内存地址，大部分场景可行，但如果内核运行时重新分配Map内存（部分特殊Map类型或内核版本），判断会失效，因此更推荐使用Map ID的方式。

内容的提问来源于stack exchange，提问作者Sanito Gonzalez