1// SPDX-License-Identifier: GPL-2.0
2
3#include <test_progs.h>
4
5#include "cgroup_helpers.h"
6
7#define PING_CMD	"ping -q -c1 -w1 127.0.0.1 > /dev/null"
8
9static char bpf_log_buf[BPF_LOG_BUF_SIZE];
10
11static int prog_load(void)
12{
13	struct bpf_insn prog[] = {
14		BPF_MOV64_IMM(BPF_REG_0, 1), /* r0 = 1 */
15		BPF_EXIT_INSN(),
16	};
17	size_t insns_cnt = ARRAY_SIZE(prog);
18
19	return bpf_test_load_program(BPF_PROG_TYPE_CGROUP_SKB,
20			       prog, insns_cnt, "GPL", 0,
21			       bpf_log_buf, BPF_LOG_BUF_SIZE);
22}
23
24void serial_test_cgroup_attach_autodetach(void)
25{
26	__u32 duration = 0, prog_cnt = 4, attach_flags;
27	int allow_prog[2] = {-1};
28	__u32 prog_ids[2] = {0};
29	void *ptr = NULL;
30	int cg = 0, i;
31	int attempts;
32
33	for (i = 0; i < ARRAY_SIZE(allow_prog); i++) {
34		allow_prog[i] = prog_load();
35		if (CHECK(allow_prog[i] < 0, "prog_load",
36			  "verifier output:\n%s\n-------\n", bpf_log_buf))
37			goto err;
38	}
39
40	if (CHECK_FAIL(setup_cgroup_environment()))
41		goto err;
42
43	/* create a cgroup, attach two programs and remember their ids */
44	cg = create_and_get_cgroup("/cg_autodetach");
45	if (CHECK_FAIL(cg < 0))
46		goto err;
47
48	if (CHECK_FAIL(join_cgroup("/cg_autodetach")))
49		goto err;
50
51	for (i = 0; i < ARRAY_SIZE(allow_prog); i++)
52		if (CHECK(bpf_prog_attach(allow_prog[i], cg,
53					  BPF_CGROUP_INET_EGRESS,
54					  BPF_F_ALLOW_MULTI),
55			  "prog_attach", "prog[%d], errno=%d\n", i, errno))
56			goto err;
57
58	/* make sure that programs are attached and run some traffic */
59	if (CHECK(bpf_prog_query(cg, BPF_CGROUP_INET_EGRESS, 0, &attach_flags,
60				 prog_ids, &prog_cnt),
61		  "prog_query", "errno=%d\n", errno))
62		goto err;
63	if (CHECK_FAIL(system(PING_CMD)))
64		goto err;
65
66	/* allocate some memory (4Mb) to pin the original cgroup */
67	ptr = malloc(4 * (1 << 20));
68	if (CHECK_FAIL(!ptr))
69		goto err;
70
71	/* close programs and cgroup fd */
72	for (i = 0; i < ARRAY_SIZE(allow_prog); i++) {
73		close(allow_prog[i]);
74		allow_prog[i] = -1;
75	}
76
77	close(cg);
78	cg = 0;
79
80	/* leave the cgroup and remove it. don't detach programs */
81	cleanup_cgroup_environment();
82
83	/* wait for the asynchronous auto-detachment.
84	 * wait for no more than 5 sec and give up.
85	 */
86	for (i = 0; i < ARRAY_SIZE(prog_ids); i++) {
87		for (attempts = 5; attempts >= 0; attempts--) {
88			int fd = bpf_prog_get_fd_by_id(prog_ids[i]);
89
90			if (fd < 0)
91				break;
92
93			/* don't leave the fd open */
94			close(fd);
95
96			if (CHECK_FAIL(!attempts))
97				goto err;
98
99			sleep(1);
100		}
101	}
102
103err:
104	for (i = 0; i < ARRAY_SIZE(allow_prog); i++)
105		if (allow_prog[i] >= 0)
106			close(allow_prog[i]);
107	if (cg)
108		close(cg);
109	free(ptr);
110	cleanup_cgroup_environment();
111}
112