1d2329247SHangbin Liu // SPDX-License-Identifier: GPL-2.0
2d2329247SHangbin Liu #include <linux/bpf.h>
3d2329247SHangbin Liu #include <linux/if_link.h>
4d2329247SHangbin Liu #include <assert.h>
5d2329247SHangbin Liu #include <errno.h>
6d2329247SHangbin Liu #include <signal.h>
7d2329247SHangbin Liu #include <stdio.h>
8d2329247SHangbin Liu #include <stdlib.h>
9d2329247SHangbin Liu #include <string.h>
10d2329247SHangbin Liu #include <net/if.h>
11d2329247SHangbin Liu #include <unistd.h>
12d2329247SHangbin Liu #include <libgen.h>
13d2329247SHangbin Liu #include <sys/ioctl.h>
14d2329247SHangbin Liu #include <sys/types.h>
15d2329247SHangbin Liu #include <sys/socket.h>
16d2329247SHangbin Liu #include <netinet/in.h>
17d2329247SHangbin Liu
18d2329247SHangbin Liu #include "bpf_util.h"
19d2329247SHangbin Liu #include <bpf/bpf.h>
20d2329247SHangbin Liu #include <bpf/libbpf.h>
21d2329247SHangbin Liu
22d2329247SHangbin Liu #define MAX_IFACE_NUM 32
23d2329247SHangbin Liu #define MAX_INDEX_NUM 1024
24d2329247SHangbin Liu
25d2329247SHangbin Liu static __u32 xdp_flags = XDP_FLAGS_UPDATE_IF_NOEXIST;
26d2329247SHangbin Liu static int ifaces[MAX_IFACE_NUM] = {};
27d2329247SHangbin Liu
int_exit(int sig)28d2329247SHangbin Liu static void int_exit(int sig)
29d2329247SHangbin Liu {
30d2329247SHangbin Liu __u32 prog_id = 0;
31d2329247SHangbin Liu int i;
32d2329247SHangbin Liu
33d2329247SHangbin Liu for (i = 0; ifaces[i] > 0; i++) {
34e4e284a8SAndrii Nakryiko if (bpf_xdp_query_id(ifaces[i], xdp_flags, &prog_id)) {
35e4e284a8SAndrii Nakryiko printf("bpf_xdp_query_id failed\n");
36d2329247SHangbin Liu exit(1);
37d2329247SHangbin Liu }
38d2329247SHangbin Liu if (prog_id)
39e4e284a8SAndrii Nakryiko bpf_xdp_detach(ifaces[i], xdp_flags, NULL);
40d2329247SHangbin Liu }
41d2329247SHangbin Liu
42d2329247SHangbin Liu exit(0);
43d2329247SHangbin Liu }
44d2329247SHangbin Liu
get_mac_addr(unsigned int ifindex,void * mac_addr)45d2329247SHangbin Liu static int get_mac_addr(unsigned int ifindex, void *mac_addr)
46d2329247SHangbin Liu {
47d2329247SHangbin Liu char ifname[IF_NAMESIZE];
48d2329247SHangbin Liu struct ifreq ifr;
49d2329247SHangbin Liu int fd, ret = -1;
50d2329247SHangbin Liu
51d2329247SHangbin Liu fd = socket(AF_INET, SOCK_DGRAM, 0);
52d2329247SHangbin Liu if (fd < 0)
53d2329247SHangbin Liu return ret;
54d2329247SHangbin Liu
55d2329247SHangbin Liu if (!if_indextoname(ifindex, ifname))
56d2329247SHangbin Liu goto err_out;
57d2329247SHangbin Liu
58d2329247SHangbin Liu strcpy(ifr.ifr_name, ifname);
59d2329247SHangbin Liu
60d2329247SHangbin Liu if (ioctl(fd, SIOCGIFHWADDR, &ifr) != 0)
61d2329247SHangbin Liu goto err_out;
62d2329247SHangbin Liu
63d2329247SHangbin Liu memcpy(mac_addr, ifr.ifr_hwaddr.sa_data, 6 * sizeof(char));
64d2329247SHangbin Liu ret = 0;
65d2329247SHangbin Liu
66d2329247SHangbin Liu err_out:
67d2329247SHangbin Liu close(fd);
68d2329247SHangbin Liu return ret;
69d2329247SHangbin Liu }
70d2329247SHangbin Liu
usage(const char * prog)71d2329247SHangbin Liu static void usage(const char *prog)
72d2329247SHangbin Liu {
73d2329247SHangbin Liu fprintf(stderr,
74d2329247SHangbin Liu "usage: %s [OPTS] <IFNAME|IFINDEX> <IFNAME|IFINDEX> ...\n"
75d2329247SHangbin Liu "OPTS:\n"
76d2329247SHangbin Liu " -S use skb-mode\n"
77d2329247SHangbin Liu " -N enforce native mode\n"
78d2329247SHangbin Liu " -F force loading prog\n"
79d2329247SHangbin Liu " -X load xdp program on egress\n",
80d2329247SHangbin Liu prog);
81d2329247SHangbin Liu }
82d2329247SHangbin Liu
main(int argc,char ** argv)83d2329247SHangbin Liu int main(int argc, char **argv)
84d2329247SHangbin Liu {
85d2329247SHangbin Liu int prog_fd, group_all, mac_map;
86d2329247SHangbin Liu struct bpf_program *ingress_prog, *egress_prog;
87186d1a86SAndrii Nakryiko int i, err, ret, opt, egress_prog_fd = 0;
88d2329247SHangbin Liu struct bpf_devmap_val devmap_val;
89d2329247SHangbin Liu bool attach_egress_prog = false;
90d2329247SHangbin Liu unsigned char mac_addr[6];
91d2329247SHangbin Liu char ifname[IF_NAMESIZE];
92d2329247SHangbin Liu struct bpf_object *obj;
93d2329247SHangbin Liu unsigned int ifindex;
94d2329247SHangbin Liu char filename[256];
95d2329247SHangbin Liu
96d2329247SHangbin Liu while ((opt = getopt(argc, argv, "SNFX")) != -1) {
97d2329247SHangbin Liu switch (opt) {
98d2329247SHangbin Liu case 'S':
99d2329247SHangbin Liu xdp_flags |= XDP_FLAGS_SKB_MODE;
100d2329247SHangbin Liu break;
101d2329247SHangbin Liu case 'N':
102d2329247SHangbin Liu /* default, set below */
103d2329247SHangbin Liu break;
104d2329247SHangbin Liu case 'F':
105d2329247SHangbin Liu xdp_flags &= ~XDP_FLAGS_UPDATE_IF_NOEXIST;
106d2329247SHangbin Liu break;
107d2329247SHangbin Liu case 'X':
108d2329247SHangbin Liu attach_egress_prog = true;
109d2329247SHangbin Liu break;
110d2329247SHangbin Liu default:
111d2329247SHangbin Liu usage(basename(argv[0]));
112d2329247SHangbin Liu return 1;
113d2329247SHangbin Liu }
114d2329247SHangbin Liu }
115d2329247SHangbin Liu
116d2329247SHangbin Liu if (!(xdp_flags & XDP_FLAGS_SKB_MODE)) {
117d2329247SHangbin Liu xdp_flags |= XDP_FLAGS_DRV_MODE;
118d2329247SHangbin Liu } else if (attach_egress_prog) {
119d2329247SHangbin Liu printf("Load xdp program on egress with SKB mode not supported yet\n");
120d2329247SHangbin Liu goto err_out;
121d2329247SHangbin Liu }
122d2329247SHangbin Liu
123d2329247SHangbin Liu if (optind == argc) {
124d2329247SHangbin Liu printf("usage: %s <IFNAME|IFINDEX> <IFNAME|IFINDEX> ...\n", argv[0]);
125d2329247SHangbin Liu goto err_out;
126d2329247SHangbin Liu }
127d2329247SHangbin Liu
1288955c1a3SHangbin Liu printf("Get interfaces:");
129d2329247SHangbin Liu for (i = 0; i < MAX_IFACE_NUM && argv[optind + i]; i++) {
130d2329247SHangbin Liu ifaces[i] = if_nametoindex(argv[optind + i]);
131d2329247SHangbin Liu if (!ifaces[i])
132d2329247SHangbin Liu ifaces[i] = strtoul(argv[optind + i], NULL, 0);
133d2329247SHangbin Liu if (!if_indextoname(ifaces[i], ifname)) {
134d2329247SHangbin Liu perror("Invalid interface name or i");
135d2329247SHangbin Liu goto err_out;
136d2329247SHangbin Liu }
137d2329247SHangbin Liu if (ifaces[i] > MAX_INDEX_NUM) {
1388955c1a3SHangbin Liu printf(" interface index too large\n");
139d2329247SHangbin Liu goto err_out;
140d2329247SHangbin Liu }
141d2329247SHangbin Liu printf(" %d", ifaces[i]);
142d2329247SHangbin Liu }
143d2329247SHangbin Liu printf("\n");
144d2329247SHangbin Liu
145*afef88e6SDaniel Müller snprintf(filename, sizeof(filename), "%s_kern.bpf.o", argv[0]);
146186d1a86SAndrii Nakryiko obj = bpf_object__open_file(filename, NULL);
147186d1a86SAndrii Nakryiko err = libbpf_get_error(obj);
148186d1a86SAndrii Nakryiko if (err)
149d2329247SHangbin Liu goto err_out;
150186d1a86SAndrii Nakryiko err = bpf_object__load(obj);
151186d1a86SAndrii Nakryiko if (err)
152186d1a86SAndrii Nakryiko goto err_out;
153186d1a86SAndrii Nakryiko prog_fd = bpf_program__fd(bpf_object__next_program(obj, NULL));
154d2329247SHangbin Liu
155d2329247SHangbin Liu if (attach_egress_prog)
156d2329247SHangbin Liu group_all = bpf_object__find_map_fd_by_name(obj, "map_egress");
157d2329247SHangbin Liu else
158d2329247SHangbin Liu group_all = bpf_object__find_map_fd_by_name(obj, "map_all");
159d2329247SHangbin Liu mac_map = bpf_object__find_map_fd_by_name(obj, "mac_map");
160d2329247SHangbin Liu
161d2329247SHangbin Liu if (group_all < 0 || mac_map < 0) {
162d2329247SHangbin Liu printf("bpf_object__find_map_fd_by_name failed\n");
163d2329247SHangbin Liu goto err_out;
164d2329247SHangbin Liu }
165d2329247SHangbin Liu
166d2329247SHangbin Liu if (attach_egress_prog) {
167d2329247SHangbin Liu /* Find ingress/egress prog for 2nd xdp prog */
168d2329247SHangbin Liu ingress_prog = bpf_object__find_program_by_name(obj, "xdp_redirect_map_all_prog");
169d2329247SHangbin Liu egress_prog = bpf_object__find_program_by_name(obj, "xdp_devmap_prog");
170d2329247SHangbin Liu if (!ingress_prog || !egress_prog) {
171d2329247SHangbin Liu printf("finding ingress/egress_prog in obj file failed\n");
172d2329247SHangbin Liu goto err_out;
173d2329247SHangbin Liu }
174d2329247SHangbin Liu prog_fd = bpf_program__fd(ingress_prog);
175d2329247SHangbin Liu egress_prog_fd = bpf_program__fd(egress_prog);
176d2329247SHangbin Liu if (prog_fd < 0 || egress_prog_fd < 0) {
177d2329247SHangbin Liu printf("find egress_prog fd failed\n");
178d2329247SHangbin Liu goto err_out;
179d2329247SHangbin Liu }
180d2329247SHangbin Liu }
181d2329247SHangbin Liu
182d2329247SHangbin Liu signal(SIGINT, int_exit);
183d2329247SHangbin Liu signal(SIGTERM, int_exit);
184d2329247SHangbin Liu
185d2329247SHangbin Liu /* Init forward multicast groups and exclude group */
186d2329247SHangbin Liu for (i = 0; ifaces[i] > 0; i++) {
187d2329247SHangbin Liu ifindex = ifaces[i];
188d2329247SHangbin Liu
189d2329247SHangbin Liu if (attach_egress_prog) {
190d2329247SHangbin Liu ret = get_mac_addr(ifindex, mac_addr);
191d2329247SHangbin Liu if (ret < 0) {
192d2329247SHangbin Liu printf("get interface %d mac failed\n", ifindex);
193d2329247SHangbin Liu goto err_out;
194d2329247SHangbin Liu }
195d2329247SHangbin Liu ret = bpf_map_update_elem(mac_map, &ifindex, mac_addr, 0);
196d2329247SHangbin Liu if (ret) {
197d2329247SHangbin Liu perror("bpf_update_elem mac_map failed\n");
198d2329247SHangbin Liu goto err_out;
199d2329247SHangbin Liu }
200d2329247SHangbin Liu }
201d2329247SHangbin Liu
202d2329247SHangbin Liu /* Add all the interfaces to group all */
203d2329247SHangbin Liu devmap_val.ifindex = ifindex;
204d2329247SHangbin Liu devmap_val.bpf_prog.fd = egress_prog_fd;
205d2329247SHangbin Liu ret = bpf_map_update_elem(group_all, &ifindex, &devmap_val, 0);
206d2329247SHangbin Liu if (ret) {
207d2329247SHangbin Liu perror("bpf_map_update_elem");
208d2329247SHangbin Liu goto err_out;
209d2329247SHangbin Liu }
210d2329247SHangbin Liu
211d2329247SHangbin Liu /* bind prog_fd to each interface */
212e4e284a8SAndrii Nakryiko ret = bpf_xdp_attach(ifindex, prog_fd, xdp_flags, NULL);
213d2329247SHangbin Liu if (ret) {
214d2329247SHangbin Liu printf("Set xdp fd failed on %d\n", ifindex);
215d2329247SHangbin Liu goto err_out;
216d2329247SHangbin Liu }
217d2329247SHangbin Liu }
218d2329247SHangbin Liu
219d2329247SHangbin Liu /* sleep some time for testing */
220d2329247SHangbin Liu sleep(999);
221d2329247SHangbin Liu
222d2329247SHangbin Liu return 0;
223d2329247SHangbin Liu
224d2329247SHangbin Liu err_out:
225d2329247SHangbin Liu return 1;
226d2329247SHangbin Liu }
227