hc
2024-01-31 f9004dbfff8a3fbbd7e2a88c8a4327c7f2f8e5b2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2020 Facebook */
#include <linux/bpf.h>
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
#include <bpf/bpf_core_read.h>
 
#pragma clang attribute push (__attribute__((preserve_access_index)), apply_to = record)
struct seq_file;
struct bpf_iter_meta {
   struct seq_file *seq;
   __u64 session_id;
   __u64 seq_num;
};
 
struct bpf_map {
   __u32 id;
   char name[16];
   __u32 max_entries;
};
 
struct bpf_iter__bpf_map {
   struct bpf_iter_meta *meta;
   struct bpf_map *map;
};
 
struct btf_type {
   __u32 name_off;
};
 
struct btf_header {
   __u32   str_len;
};
 
struct btf {
   const char *strings;
   struct btf_type **types;
   struct btf_header hdr;
};
 
struct bpf_prog_aux {
   __u32 id;
   char name[16];
   const char *attach_func_name;
   struct bpf_prog *dst_prog;
   struct bpf_func_info *func_info;
   struct btf *btf;
};
 
struct bpf_prog {
   struct bpf_prog_aux *aux;
};
 
struct bpf_iter__bpf_prog {
   struct bpf_iter_meta *meta;
   struct bpf_prog *prog;
};
#pragma clang attribute pop
 
static const char *get_name(struct btf *btf, long btf_id, const char *fallback)
{
   struct btf_type **types, *t;
   unsigned int name_off;
   const char *str;
 
   if (!btf)
       return fallback;
   str = btf->strings;
   types = btf->types;
   bpf_probe_read_kernel(&t, sizeof(t), types + btf_id);
   name_off = BPF_CORE_READ(t, name_off);
   if (name_off >= btf->hdr.str_len)
       return fallback;
   return str + name_off;
}
 
SEC("iter/bpf_map")
int dump_bpf_map(struct bpf_iter__bpf_map *ctx)
{
   struct seq_file *seq = ctx->meta->seq;
   __u64 seq_num = ctx->meta->seq_num;
   struct bpf_map *map = ctx->map;
 
   if (!map)
       return 0;
 
   if (seq_num == 0)
       BPF_SEQ_PRINTF(seq, "  id name             max_entries\n");
 
   BPF_SEQ_PRINTF(seq, "%4u %-16s%6d\n", map->id, map->name, map->max_entries);
   return 0;
}
 
SEC("iter/bpf_prog")
int dump_bpf_prog(struct bpf_iter__bpf_prog *ctx)
{
   struct seq_file *seq = ctx->meta->seq;
   __u64 seq_num = ctx->meta->seq_num;
   struct bpf_prog *prog = ctx->prog;
   struct bpf_prog_aux *aux;
 
   if (!prog)
       return 0;
 
   aux = prog->aux;
   if (seq_num == 0)
       BPF_SEQ_PRINTF(seq, "  id name             attached\n");
 
   BPF_SEQ_PRINTF(seq, "%4u %-16s %s %s\n", aux->id,
              get_name(aux->btf, aux->func_info[0].type_id, aux->name),
              aux->attach_func_name, aux->dst_prog->aux->name);
   return 0;
}
char LICENSE[] SEC("license") = "GPL";