Loading...
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 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 | // 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(); int proto_fd = -1; skel = test_ringbuf_multi__open(); if (CHECK(!skel, "skel_open", "skeleton open failed\n")) return; /* validate ringbuf size adjustment logic */ ASSERT_EQ(bpf_map__max_entries(skel->maps.ringbuf1), page_size, "rb1_size_before"); ASSERT_OK(bpf_map__set_max_entries(skel->maps.ringbuf1, page_size + 1), "rb1_resize"); ASSERT_EQ(bpf_map__max_entries(skel->maps.ringbuf1), 2 * page_size, "rb1_size_after"); ASSERT_OK(bpf_map__set_max_entries(skel->maps.ringbuf1, page_size), "rb1_reset"); ASSERT_EQ(bpf_map__max_entries(skel->maps.ringbuf1), page_size, "rb1_size_final"); proto_fd = bpf_map_create(BPF_MAP_TYPE_RINGBUF, NULL, 0, 0, page_size, NULL); if (CHECK(proto_fd < 0, "bpf_map_create", "bpf_map_create failed\n")) goto cleanup; err = bpf_map__set_inner_map_fd(skel->maps.ringbuf_hash, proto_fd); if (CHECK(err != 0, "bpf_map__set_inner_map_fd", "bpf_map__set_inner_map_fd failed\n")) goto cleanup; err = test_ringbuf_multi__load(skel); if (CHECK(err != 0, "skel_load", "skeleton load failed\n")) goto cleanup; close(proto_fd); proto_fd = -1; /* make sure we can't resize ringbuf after object load */ if (!ASSERT_ERR(bpf_map__set_max_entries(skel->maps.ringbuf1, 3 * page_size), "rb1_resize_after_load")) 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: if (proto_fd >= 0) close(proto_fd); ring_buffer__free(ringbuf); test_ringbuf_multi__destroy(skel); } |