1 // SPDX-License-Identifier: GPL-2.0
2 #include <test_progs.h>
3 #include "test_probe_read_user_str.skel.h"
4
5 static const char str1[] = "mestring";
6 static const char str2[] = "mestringalittlebigger";
7 static const char str3[] = "mestringblubblubblubblubblub";
8
test_one_str(struct test_probe_read_user_str * skel,const char * str,size_t len)9 static int test_one_str(struct test_probe_read_user_str *skel, const char *str,
10 size_t len)
11 {
12 int err, duration = 0;
13 char buf[256];
14
15 /* Ensure bytes after string are ones */
16 memset(buf, 1, sizeof(buf));
17 memcpy(buf, str, len);
18
19 /* Give prog our userspace pointer */
20 skel->bss->user_ptr = buf;
21
22 /* Trigger tracepoint */
23 usleep(1);
24
25 /* Did helper fail? */
26 if (CHECK(skel->bss->ret < 0, "prog_ret", "prog returned: %ld\n",
27 skel->bss->ret))
28 return 1;
29
30 /* Check that string was copied correctly */
31 err = memcmp(skel->bss->buf, str, len);
32 if (CHECK(err, "memcmp", "prog copied wrong string"))
33 return 1;
34
35 /* Now check that no extra trailing bytes were copied */
36 memset(buf, 0, sizeof(buf));
37 err = memcmp(skel->bss->buf + len, buf, sizeof(buf) - len);
38 if (CHECK(err, "memcmp", "trailing bytes were not stripped"))
39 return 1;
40
41 return 0;
42 }
43
test_probe_read_user_str(void)44 void test_probe_read_user_str(void)
45 {
46 struct test_probe_read_user_str *skel;
47 int err, duration = 0;
48
49 skel = test_probe_read_user_str__open_and_load();
50 if (CHECK(!skel, "test_probe_read_user_str__open_and_load",
51 "skeleton open and load failed\n"))
52 return;
53
54 /* Give pid to bpf prog so it doesn't read from anyone else */
55 skel->bss->pid = getpid();
56
57 err = test_probe_read_user_str__attach(skel);
58 if (CHECK(err, "test_probe_read_user_str__attach",
59 "skeleton attach failed: %d\n", err))
60 goto out;
61
62 if (test_one_str(skel, str1, sizeof(str1)))
63 goto out;
64 if (test_one_str(skel, str2, sizeof(str2)))
65 goto out;
66 if (test_one_str(skel, str3, sizeof(str3)))
67 goto out;
68
69 out:
70 test_probe_read_user_str__destroy(skel);
71 }
72