1 // SPDX-License-Identifier: GPL-2.0
2 #include <test_progs.h>
3
4 #define IFINDEX_LO 1
5 #define XDP_FLAGS_REPLACE (1U << 4)
6
test_xdp_attach(void)7 void test_xdp_attach(void)
8 {
9 __u32 duration = 0, id1, id2, id0 = 0, len;
10 struct bpf_object *obj1, *obj2, *obj3;
11 const char *file = "./test_xdp.o";
12 struct bpf_prog_info info = {};
13 int err, fd1, fd2, fd3;
14 DECLARE_LIBBPF_OPTS(bpf_xdp_set_link_opts, opts,
15 .old_fd = -1);
16
17 len = sizeof(info);
18
19 err = bpf_prog_load(file, BPF_PROG_TYPE_XDP, &obj1, &fd1);
20 if (CHECK_FAIL(err))
21 return;
22 err = bpf_obj_get_info_by_fd(fd1, &info, &len);
23 if (CHECK_FAIL(err))
24 goto out_1;
25 id1 = info.id;
26
27 err = bpf_prog_load(file, BPF_PROG_TYPE_XDP, &obj2, &fd2);
28 if (CHECK_FAIL(err))
29 goto out_1;
30
31 memset(&info, 0, sizeof(info));
32 err = bpf_obj_get_info_by_fd(fd2, &info, &len);
33 if (CHECK_FAIL(err))
34 goto out_2;
35 id2 = info.id;
36
37 err = bpf_prog_load(file, BPF_PROG_TYPE_XDP, &obj3, &fd3);
38 if (CHECK_FAIL(err))
39 goto out_2;
40
41 err = bpf_set_link_xdp_fd_opts(IFINDEX_LO, fd1, XDP_FLAGS_REPLACE,
42 &opts);
43 if (CHECK(err, "load_ok", "initial load failed"))
44 goto out_close;
45
46 err = bpf_get_link_xdp_id(IFINDEX_LO, &id0, 0);
47 if (CHECK(err || id0 != id1, "id1_check",
48 "loaded prog id %u != id1 %u, err %d", id0, id1, err))
49 goto out_close;
50
51 err = bpf_set_link_xdp_fd_opts(IFINDEX_LO, fd2, XDP_FLAGS_REPLACE,
52 &opts);
53 if (CHECK(!err, "load_fail", "load with expected id didn't fail"))
54 goto out;
55
56 opts.old_fd = fd1;
57 err = bpf_set_link_xdp_fd_opts(IFINDEX_LO, fd2, 0, &opts);
58 if (CHECK(err, "replace_ok", "replace valid old_fd failed"))
59 goto out;
60 err = bpf_get_link_xdp_id(IFINDEX_LO, &id0, 0);
61 if (CHECK(err || id0 != id2, "id2_check",
62 "loaded prog id %u != id2 %u, err %d", id0, id2, err))
63 goto out_close;
64
65 err = bpf_set_link_xdp_fd_opts(IFINDEX_LO, fd3, 0, &opts);
66 if (CHECK(!err, "replace_fail", "replace invalid old_fd didn't fail"))
67 goto out;
68
69 err = bpf_set_link_xdp_fd_opts(IFINDEX_LO, -1, 0, &opts);
70 if (CHECK(!err, "remove_fail", "remove invalid old_fd didn't fail"))
71 goto out;
72
73 opts.old_fd = fd2;
74 err = bpf_set_link_xdp_fd_opts(IFINDEX_LO, -1, 0, &opts);
75 if (CHECK(err, "remove_ok", "remove valid old_fd failed"))
76 goto out;
77
78 err = bpf_get_link_xdp_id(IFINDEX_LO, &id0, 0);
79 if (CHECK(err || id0 != 0, "unload_check",
80 "loaded prog id %u != 0, err %d", id0, err))
81 goto out_close;
82 out:
83 bpf_set_link_xdp_fd(IFINDEX_LO, -1, 0);
84 out_close:
85 bpf_object__close(obj3);
86 out_2:
87 bpf_object__close(obj2);
88 out_1:
89 bpf_object__close(obj1);
90 }
91
