1d53dee3fSAndrii Nakryiko // SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause)
2d53dee3fSAndrii Nakryiko /* Copyright (C) 2020 Facebook */
3d53dee3fSAndrii Nakryiko #include <errno.h>
4622a5b58SMauricio Vásquez #include <linux/err.h>
5d53dee3fSAndrii Nakryiko #include <stdbool.h>
6d53dee3fSAndrii Nakryiko #include <stdio.h>
7d53dee3fSAndrii Nakryiko #include <stdlib.h>
8d53dee3fSAndrii Nakryiko #include <string.h>
9d53dee3fSAndrii Nakryiko #include <unistd.h>
10d6699f8eSQuentin Monnet
11d53dee3fSAndrii Nakryiko #include <bpf/bpf.h>
12d6699f8eSQuentin Monnet #include <bpf/hashmap.h>
13d53dee3fSAndrii Nakryiko
14d53dee3fSAndrii Nakryiko #include "main.h"
15d53dee3fSAndrii Nakryiko #include "skeleton/pid_iter.h"
16d53dee3fSAndrii Nakryiko
17d53dee3fSAndrii Nakryiko #ifdef BPFTOOL_WITHOUT_SKELETONS
18d53dee3fSAndrii Nakryiko
build_obj_refs_table(struct hashmap ** map,enum bpf_obj_type type)19d6699f8eSQuentin Monnet int build_obj_refs_table(struct hashmap **map, enum bpf_obj_type type)
20d53dee3fSAndrii Nakryiko {
21d53dee3fSAndrii Nakryiko return -ENOTSUP;
22d53dee3fSAndrii Nakryiko }
delete_obj_refs_table(struct hashmap * map)23d6699f8eSQuentin Monnet void delete_obj_refs_table(struct hashmap *map) {}
emit_obj_refs_plain(struct hashmap * map,__u32 id,const char * prefix)24d6699f8eSQuentin Monnet void emit_obj_refs_plain(struct hashmap *map, __u32 id, const char *prefix) {}
emit_obj_refs_json(struct hashmap * map,__u32 id,json_writer_t * json_writer)25d6699f8eSQuentin Monnet void emit_obj_refs_json(struct hashmap *map, __u32 id, json_writer_t *json_writer) {}
26d53dee3fSAndrii Nakryiko
27d53dee3fSAndrii Nakryiko #else /* BPFTOOL_WITHOUT_SKELETONS */
28d53dee3fSAndrii Nakryiko
29d53dee3fSAndrii Nakryiko #include "pid_iter.skel.h"
30d53dee3fSAndrii Nakryiko
add_ref(struct hashmap * map,struct pid_iter_entry * e)31d6699f8eSQuentin Monnet static void add_ref(struct hashmap *map, struct pid_iter_entry *e)
32d53dee3fSAndrii Nakryiko {
33d6699f8eSQuentin Monnet struct hashmap_entry *entry;
34d53dee3fSAndrii Nakryiko struct obj_refs *refs;
35d53dee3fSAndrii Nakryiko struct obj_ref *ref;
36d6699f8eSQuentin Monnet int err, i;
37d53dee3fSAndrii Nakryiko void *tmp;
38d53dee3fSAndrii Nakryiko
39c302378bSEduard Zingerman hashmap__for_each_key_entry(map, entry, e->id) {
40c302378bSEduard Zingerman refs = entry->pvalue;
41d53dee3fSAndrii Nakryiko
42d53dee3fSAndrii Nakryiko for (i = 0; i < refs->ref_cnt; i++) {
43d53dee3fSAndrii Nakryiko if (refs->refs[i].pid == e->pid)
44d53dee3fSAndrii Nakryiko return;
45d53dee3fSAndrii Nakryiko }
46d53dee3fSAndrii Nakryiko
47d53dee3fSAndrii Nakryiko tmp = realloc(refs->refs, (refs->ref_cnt + 1) * sizeof(*ref));
48d53dee3fSAndrii Nakryiko if (!tmp) {
49d53dee3fSAndrii Nakryiko p_err("failed to re-alloc memory for ID %u, PID %d, COMM %s...",
50d53dee3fSAndrii Nakryiko e->id, e->pid, e->comm);
51d53dee3fSAndrii Nakryiko return;
52d53dee3fSAndrii Nakryiko }
53d53dee3fSAndrii Nakryiko refs->refs = tmp;
54d53dee3fSAndrii Nakryiko ref = &refs->refs[refs->ref_cnt];
55d53dee3fSAndrii Nakryiko ref->pid = e->pid;
56d53dee3fSAndrii Nakryiko memcpy(ref->comm, e->comm, sizeof(ref->comm));
57d53dee3fSAndrii Nakryiko refs->ref_cnt++;
58d53dee3fSAndrii Nakryiko
59d53dee3fSAndrii Nakryiko return;
60d53dee3fSAndrii Nakryiko }
61d53dee3fSAndrii Nakryiko
62d53dee3fSAndrii Nakryiko /* new ref */
63d53dee3fSAndrii Nakryiko refs = calloc(1, sizeof(*refs));
64d53dee3fSAndrii Nakryiko if (!refs) {
65d53dee3fSAndrii Nakryiko p_err("failed to alloc memory for ID %u, PID %d, COMM %s...",
66d53dee3fSAndrii Nakryiko e->id, e->pid, e->comm);
67d53dee3fSAndrii Nakryiko return;
68d53dee3fSAndrii Nakryiko }
69d53dee3fSAndrii Nakryiko
70d53dee3fSAndrii Nakryiko refs->refs = malloc(sizeof(*refs->refs));
71d53dee3fSAndrii Nakryiko if (!refs->refs) {
72d53dee3fSAndrii Nakryiko free(refs);
73d53dee3fSAndrii Nakryiko p_err("failed to alloc memory for ID %u, PID %d, COMM %s...",
74d53dee3fSAndrii Nakryiko e->id, e->pid, e->comm);
75d53dee3fSAndrii Nakryiko return;
76d53dee3fSAndrii Nakryiko }
77d53dee3fSAndrii Nakryiko ref = &refs->refs[0];
78d53dee3fSAndrii Nakryiko ref->pid = e->pid;
79d53dee3fSAndrii Nakryiko memcpy(ref->comm, e->comm, sizeof(ref->comm));
80d53dee3fSAndrii Nakryiko refs->ref_cnt = 1;
81cbdaf71fSDmitrii Dolgov refs->has_bpf_cookie = e->has_bpf_cookie;
82cbdaf71fSDmitrii Dolgov refs->bpf_cookie = e->bpf_cookie;
83d6699f8eSQuentin Monnet
84c302378bSEduard Zingerman err = hashmap__append(map, e->id, refs);
85d6699f8eSQuentin Monnet if (err)
86d6699f8eSQuentin Monnet p_err("failed to append entry to hashmap for ID %u: %s",
87d6699f8eSQuentin Monnet e->id, strerror(errno));
88d53dee3fSAndrii Nakryiko }
89d53dee3fSAndrii Nakryiko
90d53dee3fSAndrii Nakryiko static int __printf(2, 0)
libbpf_print_none(__maybe_unused enum libbpf_print_level level,__maybe_unused const char * format,__maybe_unused va_list args)91d53dee3fSAndrii Nakryiko libbpf_print_none(__maybe_unused enum libbpf_print_level level,
92d53dee3fSAndrii Nakryiko __maybe_unused const char *format,
93d53dee3fSAndrii Nakryiko __maybe_unused va_list args)
94d53dee3fSAndrii Nakryiko {
95d53dee3fSAndrii Nakryiko return 0;
96d53dee3fSAndrii Nakryiko }
97d53dee3fSAndrii Nakryiko
build_obj_refs_table(struct hashmap ** map,enum bpf_obj_type type)98d6699f8eSQuentin Monnet int build_obj_refs_table(struct hashmap **map, enum bpf_obj_type type)
99d53dee3fSAndrii Nakryiko {
100d53dee3fSAndrii Nakryiko struct pid_iter_entry *e;
101932c6055SAndrii Nakryiko char buf[4096 / sizeof(*e) * sizeof(*e)];
102932c6055SAndrii Nakryiko struct pid_iter_bpf *skel;
103d53dee3fSAndrii Nakryiko int err, ret, fd = -1, i;
104d53dee3fSAndrii Nakryiko
105d6699f8eSQuentin Monnet *map = hashmap__new(hash_fn_for_key_as_id, equal_fn_for_key_as_id, NULL);
106622a5b58SMauricio Vásquez if (IS_ERR(*map)) {
107d6699f8eSQuentin Monnet p_err("failed to create hashmap for PID references");
108d6699f8eSQuentin Monnet return -1;
109d6699f8eSQuentin Monnet }
1106b4384ffSQuentin Monnet set_max_rlimit();
111d53dee3fSAndrii Nakryiko
112d53dee3fSAndrii Nakryiko skel = pid_iter_bpf__open();
113d53dee3fSAndrii Nakryiko if (!skel) {
114d53dee3fSAndrii Nakryiko p_err("failed to open PID iterator skeleton");
115d53dee3fSAndrii Nakryiko return -1;
116d53dee3fSAndrii Nakryiko }
117d53dee3fSAndrii Nakryiko
118d53dee3fSAndrii Nakryiko skel->rodata->obj_type = type;
119d53dee3fSAndrii Nakryiko
120*be24a895SQuentin Monnet if (!verifier_logs) {
121*be24a895SQuentin Monnet libbpf_print_fn_t default_print;
122*be24a895SQuentin Monnet
123*be24a895SQuentin Monnet /* Unless debug information is on, we don't want the output to
124*be24a895SQuentin Monnet * be polluted with libbpf errors if bpf_iter is not supported.
125d53dee3fSAndrii Nakryiko */
126d53dee3fSAndrii Nakryiko default_print = libbpf_set_print(libbpf_print_none);
127d53dee3fSAndrii Nakryiko err = pid_iter_bpf__load(skel);
128d53dee3fSAndrii Nakryiko libbpf_set_print(default_print);
129*be24a895SQuentin Monnet } else {
130*be24a895SQuentin Monnet err = pid_iter_bpf__load(skel);
131*be24a895SQuentin Monnet }
132d53dee3fSAndrii Nakryiko if (err) {
133d53dee3fSAndrii Nakryiko /* too bad, kernel doesn't support BPF iterators yet */
134d53dee3fSAndrii Nakryiko err = 0;
135d53dee3fSAndrii Nakryiko goto out;
136d53dee3fSAndrii Nakryiko }
137d53dee3fSAndrii Nakryiko err = pid_iter_bpf__attach(skel);
138d53dee3fSAndrii Nakryiko if (err) {
139d53dee3fSAndrii Nakryiko /* if we loaded above successfully, attach has to succeed */
140d53dee3fSAndrii Nakryiko p_err("failed to attach PID iterator: %d", err);
141d53dee3fSAndrii Nakryiko goto out;
142d53dee3fSAndrii Nakryiko }
143d53dee3fSAndrii Nakryiko
144d53dee3fSAndrii Nakryiko fd = bpf_iter_create(bpf_link__fd(skel->links.iter));
145d53dee3fSAndrii Nakryiko if (fd < 0) {
146d53dee3fSAndrii Nakryiko err = -errno;
147d53dee3fSAndrii Nakryiko p_err("failed to create PID iterator session: %d", err);
148d53dee3fSAndrii Nakryiko goto out;
149d53dee3fSAndrii Nakryiko }
150d53dee3fSAndrii Nakryiko
151d53dee3fSAndrii Nakryiko while (true) {
152d53dee3fSAndrii Nakryiko ret = read(fd, buf, sizeof(buf));
153d53dee3fSAndrii Nakryiko if (ret < 0) {
15400fa1d83SYonghong Song if (errno == EAGAIN)
15500fa1d83SYonghong Song continue;
156d53dee3fSAndrii Nakryiko err = -errno;
157d53dee3fSAndrii Nakryiko p_err("failed to read PID iterator output: %d", err);
158d53dee3fSAndrii Nakryiko goto out;
159d53dee3fSAndrii Nakryiko }
160d53dee3fSAndrii Nakryiko if (ret == 0)
161d53dee3fSAndrii Nakryiko break;
162d53dee3fSAndrii Nakryiko if (ret % sizeof(*e)) {
163d53dee3fSAndrii Nakryiko err = -EINVAL;
164d53dee3fSAndrii Nakryiko p_err("invalid PID iterator output format");
165d53dee3fSAndrii Nakryiko goto out;
166d53dee3fSAndrii Nakryiko }
167d53dee3fSAndrii Nakryiko ret /= sizeof(*e);
168d53dee3fSAndrii Nakryiko
169d53dee3fSAndrii Nakryiko e = (void *)buf;
170d53dee3fSAndrii Nakryiko for (i = 0; i < ret; i++, e++) {
171d6699f8eSQuentin Monnet add_ref(*map, e);
172d53dee3fSAndrii Nakryiko }
173d53dee3fSAndrii Nakryiko }
174d53dee3fSAndrii Nakryiko err = 0;
175d53dee3fSAndrii Nakryiko out:
176d53dee3fSAndrii Nakryiko if (fd >= 0)
177d53dee3fSAndrii Nakryiko close(fd);
178d53dee3fSAndrii Nakryiko pid_iter_bpf__destroy(skel);
179d53dee3fSAndrii Nakryiko return err;
180d53dee3fSAndrii Nakryiko }
181d53dee3fSAndrii Nakryiko
delete_obj_refs_table(struct hashmap * map)182d6699f8eSQuentin Monnet void delete_obj_refs_table(struct hashmap *map)
183d53dee3fSAndrii Nakryiko {
184d6699f8eSQuentin Monnet struct hashmap_entry *entry;
185d6699f8eSQuentin Monnet size_t bkt;
186d53dee3fSAndrii Nakryiko
187d6699f8eSQuentin Monnet if (!map)
188d6699f8eSQuentin Monnet return;
189d6699f8eSQuentin Monnet
190d6699f8eSQuentin Monnet hashmap__for_each_entry(map, entry, bkt) {
191c302378bSEduard Zingerman struct obj_refs *refs = entry->pvalue;
192d6699f8eSQuentin Monnet
193d53dee3fSAndrii Nakryiko free(refs->refs);
194d53dee3fSAndrii Nakryiko free(refs);
195d53dee3fSAndrii Nakryiko }
196d6699f8eSQuentin Monnet
197d6699f8eSQuentin Monnet hashmap__free(map);
198d53dee3fSAndrii Nakryiko }
199d53dee3fSAndrii Nakryiko
emit_obj_refs_json(struct hashmap * map,__u32 id,json_writer_t * json_writer)200d6699f8eSQuentin Monnet void emit_obj_refs_json(struct hashmap *map, __u32 id,
20154b66c22SQuentin Monnet json_writer_t *json_writer)
202d53dee3fSAndrii Nakryiko {
203d6699f8eSQuentin Monnet struct hashmap_entry *entry;
204d53dee3fSAndrii Nakryiko
205d6699f8eSQuentin Monnet if (hashmap__empty(map))
206d53dee3fSAndrii Nakryiko return;
207d53dee3fSAndrii Nakryiko
208c302378bSEduard Zingerman hashmap__for_each_key_entry(map, entry, id) {
209c302378bSEduard Zingerman struct obj_refs *refs = entry->pvalue;
210d6699f8eSQuentin Monnet int i;
211d6699f8eSQuentin Monnet
212d53dee3fSAndrii Nakryiko if (refs->ref_cnt == 0)
213d53dee3fSAndrii Nakryiko break;
214d53dee3fSAndrii Nakryiko
215cbdaf71fSDmitrii Dolgov if (refs->has_bpf_cookie)
216cbdaf71fSDmitrii Dolgov jsonw_lluint_field(json_writer, "bpf_cookie", refs->bpf_cookie);
217cbdaf71fSDmitrii Dolgov
21854b66c22SQuentin Monnet jsonw_name(json_writer, "pids");
21954b66c22SQuentin Monnet jsonw_start_array(json_writer);
220d53dee3fSAndrii Nakryiko for (i = 0; i < refs->ref_cnt; i++) {
221d6699f8eSQuentin Monnet struct obj_ref *ref = &refs->refs[i];
222d6699f8eSQuentin Monnet
22354b66c22SQuentin Monnet jsonw_start_object(json_writer);
22454b66c22SQuentin Monnet jsonw_int_field(json_writer, "pid", ref->pid);
22554b66c22SQuentin Monnet jsonw_string_field(json_writer, "comm", ref->comm);
22654b66c22SQuentin Monnet jsonw_end_object(json_writer);
227d53dee3fSAndrii Nakryiko }
22854b66c22SQuentin Monnet jsonw_end_array(json_writer);
229d53dee3fSAndrii Nakryiko break;
230d53dee3fSAndrii Nakryiko }
231d53dee3fSAndrii Nakryiko }
232d53dee3fSAndrii Nakryiko
emit_obj_refs_plain(struct hashmap * map,__u32 id,const char * prefix)233d6699f8eSQuentin Monnet void emit_obj_refs_plain(struct hashmap *map, __u32 id, const char *prefix)
234d53dee3fSAndrii Nakryiko {
235d6699f8eSQuentin Monnet struct hashmap_entry *entry;
236d53dee3fSAndrii Nakryiko
237d6699f8eSQuentin Monnet if (hashmap__empty(map))
238d53dee3fSAndrii Nakryiko return;
239d53dee3fSAndrii Nakryiko
240c302378bSEduard Zingerman hashmap__for_each_key_entry(map, entry, id) {
241c302378bSEduard Zingerman struct obj_refs *refs = entry->pvalue;
242d6699f8eSQuentin Monnet int i;
243d6699f8eSQuentin Monnet
244d53dee3fSAndrii Nakryiko if (refs->ref_cnt == 0)
245d53dee3fSAndrii Nakryiko break;
246d53dee3fSAndrii Nakryiko
247cbdaf71fSDmitrii Dolgov if (refs->has_bpf_cookie)
248cbdaf71fSDmitrii Dolgov printf("\n\tbpf_cookie %llu", (unsigned long long) refs->bpf_cookie);
249cbdaf71fSDmitrii Dolgov
250d53dee3fSAndrii Nakryiko printf("%s", prefix);
251d53dee3fSAndrii Nakryiko for (i = 0; i < refs->ref_cnt; i++) {
252d6699f8eSQuentin Monnet struct obj_ref *ref = &refs->refs[i];
253d6699f8eSQuentin Monnet
254d53dee3fSAndrii Nakryiko printf("%s%s(%d)", i == 0 ? "" : ", ", ref->comm, ref->pid);
255d53dee3fSAndrii Nakryiko }
256d53dee3fSAndrii Nakryiko break;
257d53dee3fSAndrii Nakryiko }
258d53dee3fSAndrii Nakryiko }
259d53dee3fSAndrii Nakryiko
260d53dee3fSAndrii Nakryiko
261d53dee3fSAndrii Nakryiko #endif
262