// SPDX-License-Identifier: GPL-2.0 #include <test_progs.h> #include "test_probe_read_user_str.skel.h" static const char str1[] = "mestring"; static const char str2[] = "mestringalittlebigger"; static const char str3[] = "mestringblubblubblubblubblub"; static int test_one_str(struct test_probe_read_user_str *skel, const char *str, size_t len) { int err, duration = 0; char buf[256]; /* Ensure bytes after string are ones */ memset(buf, 1, sizeof(buf)); memcpy(buf, str, len); /* Give prog our userspace pointer */ skel->bss->user_ptr = buf; /* Trigger tracepoint */ usleep(1); /* Did helper fail? */ if (CHECK(skel->bss->ret < 0, "prog_ret", "prog returned: %ld\n", skel->bss->ret)) return 1; /* Check that string was copied correctly */ err = memcmp(skel->bss->buf, str, len); if (CHECK(err, "memcmp", "prog copied wrong string")) return 1; /* Now check that no extra trailing bytes were copied */ memset(buf, 0, sizeof(buf)); err = memcmp(skel->bss->buf + len, buf, sizeof(buf) - len); if (CHECK(err, "memcmp", "trailing bytes were not stripped")) return 1; return 0; } void test_probe_read_user_str(void) { struct test_probe_read_user_str *skel; int err, duration = 0; skel = test_probe_read_user_str__open_and_load(); if (CHECK(!skel, "test_probe_read_user_str__open_and_load", "skeleton open and load failed\n")) return; /* Give pid to bpf prog so it doesn't read from anyone else */ skel->bss->pid = getpid(); err = test_probe_read_user_str__attach(skel); if (CHECK(err, "test_probe_read_user_str__attach", "skeleton attach failed: %d\n", err)) goto out; if (test_one_str(skel, str1, sizeof(str1))) goto out; if (test_one_str(skel, str2, sizeof(str2))) goto out; if (test_one_str(skel, str3, sizeof(str3))) goto out; out: test_probe_read_user_str__destroy(skel); }