Loading...
1// SPDX-License-Identifier: GPL-2.0
2#include <inttypes.h>
3#include <stdio.h>
4#include <stdlib.h>
5#include <unistd.h>
6#include <errno.h>
7#include <string.h>
8#include <linux/bpf.h>
9#include <sys/types.h>
10#include <bpf/bpf.h>
11#include <bpf/libbpf.h>
12
13#include "bpf_rlimit.h"
14#include "bpf_util.h"
15#include "cgroup_helpers.h"
16
17#include "test_tcpbpf.h"
18
19#define EXPECT_EQ(expected, actual, fmt) \
20 do { \
21 if ((expected) != (actual)) { \
22 printf(" Value of: " #actual "\n" \
23 " Actual: %" fmt "\n" \
24 " Expected: %" fmt "\n", \
25 (actual), (expected)); \
26 goto err; \
27 } \
28 } while (0)
29
30int verify_result(const struct tcpbpf_globals *result)
31{
32 __u32 expected_events;
33
34 expected_events = ((1 << BPF_SOCK_OPS_TIMEOUT_INIT) |
35 (1 << BPF_SOCK_OPS_RWND_INIT) |
36 (1 << BPF_SOCK_OPS_TCP_CONNECT_CB) |
37 (1 << BPF_SOCK_OPS_ACTIVE_ESTABLISHED_CB) |
38 (1 << BPF_SOCK_OPS_PASSIVE_ESTABLISHED_CB) |
39 (1 << BPF_SOCK_OPS_NEEDS_ECN) |
40 (1 << BPF_SOCK_OPS_STATE_CB) |
41 (1 << BPF_SOCK_OPS_TCP_LISTEN_CB));
42
43 EXPECT_EQ(expected_events, result->event_map, "#" PRIx32);
44 EXPECT_EQ(501ULL, result->bytes_received, "llu");
45 EXPECT_EQ(1002ULL, result->bytes_acked, "llu");
46 EXPECT_EQ(1, result->data_segs_in, PRIu32);
47 EXPECT_EQ(1, result->data_segs_out, PRIu32);
48 EXPECT_EQ(0x80, result->bad_cb_test_rv, PRIu32);
49 EXPECT_EQ(0, result->good_cb_test_rv, PRIu32);
50 EXPECT_EQ(1, result->num_listen, PRIu32);
51
52 return 0;
53err:
54 return -1;
55}
56
57int verify_sockopt_result(int sock_map_fd)
58{
59 __u32 key = 0;
60 int res;
61 int rv;
62
63 /* check setsockopt for SAVE_SYN */
64 rv = bpf_map_lookup_elem(sock_map_fd, &key, &res);
65 EXPECT_EQ(0, rv, "d");
66 EXPECT_EQ(0, res, "d");
67 key = 1;
68 /* check getsockopt for SAVED_SYN */
69 rv = bpf_map_lookup_elem(sock_map_fd, &key, &res);
70 EXPECT_EQ(0, rv, "d");
71 EXPECT_EQ(1, res, "d");
72 return 0;
73err:
74 return -1;
75}
76
77static int bpf_find_map(const char *test, struct bpf_object *obj,
78 const char *name)
79{
80 struct bpf_map *map;
81
82 map = bpf_object__find_map_by_name(obj, name);
83 if (!map) {
84 printf("%s:FAIL:map '%s' not found\n", test, name);
85 return -1;
86 }
87 return bpf_map__fd(map);
88}
89
90int main(int argc, char **argv)
91{
92 const char *file = "test_tcpbpf_kern.o";
93 int prog_fd, map_fd, sock_map_fd;
94 struct tcpbpf_globals g = {0};
95 const char *cg_path = "/foo";
96 int error = EXIT_FAILURE;
97 struct bpf_object *obj;
98 int cg_fd = -1;
99 __u32 key = 0;
100 int rv;
101
102 if (setup_cgroup_environment())
103 goto err;
104
105 cg_fd = create_and_get_cgroup(cg_path);
106 if (cg_fd < 0)
107 goto err;
108
109 if (join_cgroup(cg_path))
110 goto err;
111
112 if (bpf_prog_load(file, BPF_PROG_TYPE_SOCK_OPS, &obj, &prog_fd)) {
113 printf("FAILED: load_bpf_file failed for: %s\n", file);
114 goto err;
115 }
116
117 rv = bpf_prog_attach(prog_fd, cg_fd, BPF_CGROUP_SOCK_OPS, 0);
118 if (rv) {
119 printf("FAILED: bpf_prog_attach: %d (%s)\n",
120 error, strerror(errno));
121 goto err;
122 }
123
124 if (system("./tcp_server.py")) {
125 printf("FAILED: TCP server\n");
126 goto err;
127 }
128
129 map_fd = bpf_find_map(__func__, obj, "global_map");
130 if (map_fd < 0)
131 goto err;
132
133 sock_map_fd = bpf_find_map(__func__, obj, "sockopt_results");
134 if (sock_map_fd < 0)
135 goto err;
136
137 rv = bpf_map_lookup_elem(map_fd, &key, &g);
138 if (rv != 0) {
139 printf("FAILED: bpf_map_lookup_elem returns %d\n", rv);
140 goto err;
141 }
142
143 if (verify_result(&g)) {
144 printf("FAILED: Wrong stats\n");
145 goto err;
146 }
147
148 if (verify_sockopt_result(sock_map_fd)) {
149 printf("FAILED: Wrong sockopt stats\n");
150 goto err;
151 }
152
153 printf("PASSED!\n");
154 error = 0;
155err:
156 bpf_prog_detach(cg_fd, BPF_CGROUP_SOCK_OPS);
157 close(cg_fd);
158 cleanup_cgroup_environment();
159 return error;
160}
1// SPDX-License-Identifier: GPL-2.0
2#include <stdio.h>
3#include <stdlib.h>
4#include <stdio.h>
5#include <unistd.h>
6#include <errno.h>
7#include <signal.h>
8#include <string.h>
9#include <assert.h>
10#include <linux/perf_event.h>
11#include <linux/ptrace.h>
12#include <linux/bpf.h>
13#include <sys/ioctl.h>
14#include <sys/time.h>
15#include <sys/types.h>
16#include <sys/stat.h>
17#include <fcntl.h>
18#include <bpf/bpf.h>
19#include <bpf/libbpf.h>
20#include "bpf_util.h"
21#include "bpf_rlimit.h"
22#include <linux/perf_event.h>
23#include "test_tcpbpf.h"
24
25static int bpf_find_map(const char *test, struct bpf_object *obj,
26 const char *name)
27{
28 struct bpf_map *map;
29
30 map = bpf_object__find_map_by_name(obj, name);
31 if (!map) {
32 printf("%s:FAIL:map '%s' not found\n", test, name);
33 return -1;
34 }
35 return bpf_map__fd(map);
36}
37
38#define SYSTEM(CMD) \
39 do { \
40 if (system(CMD)) { \
41 printf("system(%s) FAILS!\n", CMD); \
42 } \
43 } while (0)
44
45int main(int argc, char **argv)
46{
47 const char *file = "test_tcpbpf_kern.o";
48 struct tcpbpf_globals g = {0};
49 int cg_fd, prog_fd, map_fd;
50 bool debug_flag = false;
51 int error = EXIT_FAILURE;
52 struct bpf_object *obj;
53 char cmd[100], *dir;
54 struct stat buffer;
55 __u32 key = 0;
56 int pid;
57 int rv;
58
59 if (argc > 1 && strcmp(argv[1], "-d") == 0)
60 debug_flag = true;
61
62 dir = "/tmp/cgroupv2/foo";
63
64 if (stat(dir, &buffer) != 0) {
65 SYSTEM("mkdir -p /tmp/cgroupv2");
66 SYSTEM("mount -t cgroup2 none /tmp/cgroupv2");
67 SYSTEM("mkdir -p /tmp/cgroupv2/foo");
68 }
69 pid = (int) getpid();
70 sprintf(cmd, "echo %d >> /tmp/cgroupv2/foo/cgroup.procs", pid);
71 SYSTEM(cmd);
72
73 cg_fd = open(dir, O_DIRECTORY, O_RDONLY);
74 if (bpf_prog_load(file, BPF_PROG_TYPE_SOCK_OPS, &obj, &prog_fd)) {
75 printf("FAILED: load_bpf_file failed for: %s\n", file);
76 goto err;
77 }
78
79 rv = bpf_prog_attach(prog_fd, cg_fd, BPF_CGROUP_SOCK_OPS, 0);
80 if (rv) {
81 printf("FAILED: bpf_prog_attach: %d (%s)\n",
82 error, strerror(errno));
83 goto err;
84 }
85
86 SYSTEM("./tcp_server.py");
87
88 map_fd = bpf_find_map(__func__, obj, "global_map");
89 if (map_fd < 0)
90 goto err;
91
92 rv = bpf_map_lookup_elem(map_fd, &key, &g);
93 if (rv != 0) {
94 printf("FAILED: bpf_map_lookup_elem returns %d\n", rv);
95 goto err;
96 }
97
98 if (g.bytes_received != 501 || g.bytes_acked != 1002 ||
99 g.data_segs_in != 1 || g.data_segs_out != 1 ||
100 (g.event_map ^ 0x47e) != 0 || g.bad_cb_test_rv != 0x80 ||
101 g.good_cb_test_rv != 0) {
102 printf("FAILED: Wrong stats\n");
103 if (debug_flag) {
104 printf("\n");
105 printf("bytes_received: %d (expecting 501)\n",
106 (int)g.bytes_received);
107 printf("bytes_acked: %d (expecting 1002)\n",
108 (int)g.bytes_acked);
109 printf("data_segs_in: %d (expecting 1)\n",
110 g.data_segs_in);
111 printf("data_segs_out: %d (expecting 1)\n",
112 g.data_segs_out);
113 printf("event_map: 0x%x (at least 0x47e)\n",
114 g.event_map);
115 printf("bad_cb_test_rv: 0x%x (expecting 0x80)\n",
116 g.bad_cb_test_rv);
117 printf("good_cb_test_rv:0x%x (expecting 0)\n",
118 g.good_cb_test_rv);
119 }
120 goto err;
121 }
122 printf("PASSED!\n");
123 error = 0;
124err:
125 bpf_prog_detach(cg_fd, BPF_CGROUP_SOCK_OPS);
126 return error;
127
128}