mirror of
https://github.com/torvalds/linux.git
synced 2025-12-07 20:06:24 +00:00
f3f4c23e12
Set bpf table sizes dynamically according to the runtime page size value. Do not switch to ASSERT macros, keep CHECK, for consistency with the rest of the test. Can be a separate cleanup patch. Signed-off-by: Yauheni Kaliuta <yauheni.kaliuta@redhat.com> Signed-off-by: Andrii Nakryiko <andrii@kernel.org> Link: https://lore.kernel.org/bpf/20210408061310.95877-8-yauheni.kaliuta@redhat.com
120 lines
3.3 KiB
C
120 lines
3.3 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
#define _GNU_SOURCE
|
|
#include <test_progs.h>
|
|
#include <sys/epoll.h>
|
|
#include "test_ringbuf_multi.skel.h"
|
|
|
|
static int duration = 0;
|
|
|
|
struct sample {
|
|
int pid;
|
|
int seq;
|
|
long value;
|
|
char comm[16];
|
|
};
|
|
|
|
static int process_sample(void *ctx, void *data, size_t len)
|
|
{
|
|
int ring = (unsigned long)ctx;
|
|
struct sample *s = data;
|
|
|
|
switch (s->seq) {
|
|
case 0:
|
|
CHECK(ring != 1, "sample1_ring", "exp %d, got %d\n", 1, ring);
|
|
CHECK(s->value != 333, "sample1_value", "exp %ld, got %ld\n",
|
|
333L, s->value);
|
|
break;
|
|
case 1:
|
|
CHECK(ring != 2, "sample2_ring", "exp %d, got %d\n", 2, ring);
|
|
CHECK(s->value != 777, "sample2_value", "exp %ld, got %ld\n",
|
|
777L, s->value);
|
|
break;
|
|
default:
|
|
CHECK(true, "extra_sample", "unexpected sample seq %d, val %ld\n",
|
|
s->seq, s->value);
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void test_ringbuf_multi(void)
|
|
{
|
|
struct test_ringbuf_multi *skel;
|
|
struct ring_buffer *ringbuf = NULL;
|
|
int err;
|
|
int page_size = getpagesize();
|
|
|
|
skel = test_ringbuf_multi__open();
|
|
if (CHECK(!skel, "skel_open", "skeleton open failed\n"))
|
|
return;
|
|
|
|
err = bpf_map__set_max_entries(skel->maps.ringbuf1, page_size);
|
|
if (CHECK(err != 0, "bpf_map__set_max_entries", "bpf_map__set_max_entries failed\n"))
|
|
goto cleanup;
|
|
|
|
err = bpf_map__set_max_entries(skel->maps.ringbuf2, page_size);
|
|
if (CHECK(err != 0, "bpf_map__set_max_entries", "bpf_map__set_max_entries failed\n"))
|
|
goto cleanup;
|
|
|
|
err = bpf_map__set_max_entries(bpf_map__inner_map(skel->maps.ringbuf_arr), page_size);
|
|
if (CHECK(err != 0, "bpf_map__set_max_entries", "bpf_map__set_max_entries failed\n"))
|
|
goto cleanup;
|
|
|
|
err = test_ringbuf_multi__load(skel);
|
|
if (CHECK(err != 0, "skel_load", "skeleton load failed\n"))
|
|
goto cleanup;
|
|
|
|
/* only trigger BPF program for current process */
|
|
skel->bss->pid = getpid();
|
|
|
|
ringbuf = ring_buffer__new(bpf_map__fd(skel->maps.ringbuf1),
|
|
process_sample, (void *)(long)1, NULL);
|
|
if (CHECK(!ringbuf, "ringbuf_create", "failed to create ringbuf\n"))
|
|
goto cleanup;
|
|
|
|
err = ring_buffer__add(ringbuf, bpf_map__fd(skel->maps.ringbuf2),
|
|
process_sample, (void *)(long)2);
|
|
if (CHECK(err, "ringbuf_add", "failed to add another ring\n"))
|
|
goto cleanup;
|
|
|
|
err = test_ringbuf_multi__attach(skel);
|
|
if (CHECK(err, "skel_attach", "skeleton attachment failed: %d\n", err))
|
|
goto cleanup;
|
|
|
|
/* trigger few samples, some will be skipped */
|
|
skel->bss->target_ring = 0;
|
|
skel->bss->value = 333;
|
|
syscall(__NR_getpgid);
|
|
|
|
/* skipped, no ringbuf in slot 1 */
|
|
skel->bss->target_ring = 1;
|
|
skel->bss->value = 555;
|
|
syscall(__NR_getpgid);
|
|
|
|
skel->bss->target_ring = 2;
|
|
skel->bss->value = 777;
|
|
syscall(__NR_getpgid);
|
|
|
|
/* poll for samples, should get 2 ringbufs back */
|
|
err = ring_buffer__poll(ringbuf, -1);
|
|
if (CHECK(err != 2, "poll_res", "expected 2 records, got %d\n", err))
|
|
goto cleanup;
|
|
|
|
/* expect extra polling to return nothing */
|
|
err = ring_buffer__poll(ringbuf, 0);
|
|
if (CHECK(err < 0, "extra_samples", "poll result: %d\n", err))
|
|
goto cleanup;
|
|
|
|
CHECK(skel->bss->dropped != 0, "err_dropped", "exp %ld, got %ld\n",
|
|
0L, skel->bss->dropped);
|
|
CHECK(skel->bss->skipped != 1, "err_skipped", "exp %ld, got %ld\n",
|
|
1L, skel->bss->skipped);
|
|
CHECK(skel->bss->total != 2, "err_total", "exp %ld, got %ld\n",
|
|
2L, skel->bss->total);
|
|
|
|
cleanup:
|
|
ring_buffer__free(ringbuf);
|
|
test_ringbuf_multi__destroy(skel);
|
|
}
|