~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

TOMOYO Linux Cross Reference
Linux/samples/bpf/xdp_redirect_map_user.c

Version: ~ [ linux-5.19-rc8 ] ~ [ linux-5.18.14 ] ~ [ linux-5.17.15 ] ~ [ linux-5.16.20 ] ~ [ linux-5.15.57 ] ~ [ linux-5.14.21 ] ~ [ linux-5.13.19 ] ~ [ linux-5.12.19 ] ~ [ linux-5.11.22 ] ~ [ linux-5.10.133 ] ~ [ linux-5.9.16 ] ~ [ linux-5.8.18 ] ~ [ linux-5.7.19 ] ~ [ linux-5.6.19 ] ~ [ linux-5.5.19 ] ~ [ linux-5.4.207 ] ~ [ linux-5.3.18 ] ~ [ linux-5.2.21 ] ~ [ linux-5.1.21 ] ~ [ linux-5.0.21 ] ~ [ linux-4.20.17 ] ~ [ linux-4.19.253 ] ~ [ linux-4.18.20 ] ~ [ linux-4.17.19 ] ~ [ linux-4.16.18 ] ~ [ linux-4.15.18 ] ~ [ linux-4.14.289 ] ~ [ linux-4.13.16 ] ~ [ linux-4.12.14 ] ~ [ linux-4.11.12 ] ~ [ linux-4.10.17 ] ~ [ linux-4.9.324 ] ~ [ linux-4.8.17 ] ~ [ linux-4.7.10 ] ~ [ linux-4.6.7 ] ~ [ linux-4.5.7 ] ~ [ linux-4.4.302 ] ~ [ linux-4.3.6 ] ~ [ linux-4.2.8 ] ~ [ linux-4.1.52 ] ~ [ linux-4.0.9 ] ~ [ linux-3.10.108 ] ~ [ linux-2.6.32.71 ] ~ [ linux-2.6.0 ] ~ [ linux-2.4.37.11 ] ~ [ unix-v6-master ] ~ [ ccs-tools-1.8.9 ] ~ [ policy-sample ] ~
Architecture: ~ [ i386 ] ~ [ alpha ] ~ [ m68k ] ~ [ mips ] ~ [ ppc ] ~ [ sparc ] ~ [ sparc64 ] ~

  1 // SPDX-License-Identifier: GPL-2.0-only
  2 /* Copyright (c) 2017 Covalent IO, Inc. http://covalent.io
  3  */
  4 #include <linux/bpf.h>
  5 #include <linux/if_link.h>
  6 #include <assert.h>
  7 #include <errno.h>
  8 #include <signal.h>
  9 #include <stdio.h>
 10 #include <stdlib.h>
 11 #include <stdbool.h>
 12 #include <string.h>
 13 #include <net/if.h>
 14 #include <unistd.h>
 15 #include <libgen.h>
 16 #include <sys/resource.h>
 17 
 18 #include "bpf_util.h"
 19 #include <bpf/bpf.h>
 20 #include <bpf/libbpf.h>
 21 
 22 static int ifindex_in;
 23 static int ifindex_out;
 24 static bool ifindex_out_xdp_dummy_attached = true;
 25 static __u32 prog_id;
 26 static __u32 dummy_prog_id;
 27 
 28 static __u32 xdp_flags = XDP_FLAGS_UPDATE_IF_NOEXIST;
 29 static int rxcnt_map_fd;
 30 
 31 static void int_exit(int sig)
 32 {
 33         __u32 curr_prog_id = 0;
 34 
 35         if (bpf_get_link_xdp_id(ifindex_in, &curr_prog_id, xdp_flags)) {
 36                 printf("bpf_get_link_xdp_id failed\n");
 37                 exit(1);
 38         }
 39         if (prog_id == curr_prog_id)
 40                 bpf_set_link_xdp_fd(ifindex_in, -1, xdp_flags);
 41         else if (!curr_prog_id)
 42                 printf("couldn't find a prog id on iface IN\n");
 43         else
 44                 printf("program on iface IN changed, not removing\n");
 45 
 46         if (ifindex_out_xdp_dummy_attached) {
 47                 curr_prog_id = 0;
 48                 if (bpf_get_link_xdp_id(ifindex_out, &curr_prog_id,
 49                                         xdp_flags)) {
 50                         printf("bpf_get_link_xdp_id failed\n");
 51                         exit(1);
 52                 }
 53                 if (dummy_prog_id == curr_prog_id)
 54                         bpf_set_link_xdp_fd(ifindex_out, -1, xdp_flags);
 55                 else if (!curr_prog_id)
 56                         printf("couldn't find a prog id on iface OUT\n");
 57                 else
 58                         printf("program on iface OUT changed, not removing\n");
 59         }
 60         exit(0);
 61 }
 62 
 63 static void poll_stats(int interval, int ifindex)
 64 {
 65         unsigned int nr_cpus = bpf_num_possible_cpus();
 66         __u64 values[nr_cpus], prev[nr_cpus];
 67 
 68         memset(prev, 0, sizeof(prev));
 69 
 70         while (1) {
 71                 __u64 sum = 0;
 72                 __u32 key = 0;
 73                 int i;
 74 
 75                 sleep(interval);
 76                 assert(bpf_map_lookup_elem(rxcnt_map_fd, &key, values) == 0);
 77                 for (i = 0; i < nr_cpus; i++)
 78                         sum += (values[i] - prev[i]);
 79                 if (sum)
 80                         printf("ifindex %i: %10llu pkt/s\n",
 81                                ifindex, sum / interval);
 82                 memcpy(prev, values, sizeof(values));
 83         }
 84 }
 85 
 86 static void usage(const char *prog)
 87 {
 88         fprintf(stderr,
 89                 "usage: %s [OPTS] <IFNAME|IFINDEX>_IN <IFNAME|IFINDEX>_OUT\n\n"
 90                 "OPTS:\n"
 91                 "    -S    use skb-mode\n"
 92                 "    -N    enforce native mode\n"
 93                 "    -F    force loading prog\n",
 94                 prog);
 95 }
 96 
 97 int main(int argc, char **argv)
 98 {
 99         struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
100         struct bpf_prog_load_attr prog_load_attr = {
101                 .prog_type      = BPF_PROG_TYPE_XDP,
102         };
103         struct bpf_program *prog, *dummy_prog;
104         struct bpf_prog_info info = {};
105         __u32 info_len = sizeof(info);
106         int prog_fd, dummy_prog_fd;
107         const char *optstr = "FSN";
108         struct bpf_object *obj;
109         int ret, opt, key = 0;
110         char filename[256];
111         int tx_port_map_fd;
112 
113         while ((opt = getopt(argc, argv, optstr)) != -1) {
114                 switch (opt) {
115                 case 'S':
116                         xdp_flags |= XDP_FLAGS_SKB_MODE;
117                         break;
118                 case 'N':
119                         /* default, set below */
120                         break;
121                 case 'F':
122                         xdp_flags &= ~XDP_FLAGS_UPDATE_IF_NOEXIST;
123                         break;
124                 default:
125                         usage(basename(argv[0]));
126                         return 1;
127                 }
128         }
129 
130         if (!(xdp_flags & XDP_FLAGS_SKB_MODE))
131                 xdp_flags |= XDP_FLAGS_DRV_MODE;
132 
133         if (optind == argc) {
134                 printf("usage: %s <IFNAME|IFINDEX>_IN <IFNAME|IFINDEX>_OUT\n", argv[0]);
135                 return 1;
136         }
137 
138         if (setrlimit(RLIMIT_MEMLOCK, &r)) {
139                 perror("setrlimit(RLIMIT_MEMLOCK)");
140                 return 1;
141         }
142 
143         ifindex_in = if_nametoindex(argv[optind]);
144         if (!ifindex_in)
145                 ifindex_in = strtoul(argv[optind], NULL, 0);
146 
147         ifindex_out = if_nametoindex(argv[optind + 1]);
148         if (!ifindex_out)
149                 ifindex_out = strtoul(argv[optind + 1], NULL, 0);
150 
151         printf("input: %d output: %d\n", ifindex_in, ifindex_out);
152 
153         snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
154         prog_load_attr.file = filename;
155 
156         if (bpf_prog_load_xattr(&prog_load_attr, &obj, &prog_fd))
157                 return 1;
158 
159         prog = bpf_program__next(NULL, obj);
160         dummy_prog = bpf_program__next(prog, obj);
161         if (!prog || !dummy_prog) {
162                 printf("finding a prog in obj file failed\n");
163                 return 1;
164         }
165         /* bpf_prog_load_xattr gives us the pointer to first prog's fd,
166          * so we're missing only the fd for dummy prog
167          */
168         dummy_prog_fd = bpf_program__fd(dummy_prog);
169         if (prog_fd < 0 || dummy_prog_fd < 0) {
170                 printf("bpf_prog_load_xattr: %s\n", strerror(errno));
171                 return 1;
172         }
173 
174         tx_port_map_fd = bpf_object__find_map_fd_by_name(obj, "tx_port");
175         rxcnt_map_fd = bpf_object__find_map_fd_by_name(obj, "rxcnt");
176         if (tx_port_map_fd < 0 || rxcnt_map_fd < 0) {
177                 printf("bpf_object__find_map_fd_by_name failed\n");
178                 return 1;
179         }
180 
181         if (bpf_set_link_xdp_fd(ifindex_in, prog_fd, xdp_flags) < 0) {
182                 printf("ERROR: link set xdp fd failed on %d\n", ifindex_in);
183                 return 1;
184         }
185 
186         ret = bpf_obj_get_info_by_fd(prog_fd, &info, &info_len);
187         if (ret) {
188                 printf("can't get prog info - %s\n", strerror(errno));
189                 return ret;
190         }
191         prog_id = info.id;
192 
193         /* Loading dummy XDP prog on out-device */
194         if (bpf_set_link_xdp_fd(ifindex_out, dummy_prog_fd,
195                             (xdp_flags | XDP_FLAGS_UPDATE_IF_NOEXIST)) < 0) {
196                 printf("WARN: link set xdp fd failed on %d\n", ifindex_out);
197                 ifindex_out_xdp_dummy_attached = false;
198         }
199 
200         memset(&info, 0, sizeof(info));
201         ret = bpf_obj_get_info_by_fd(dummy_prog_fd, &info, &info_len);
202         if (ret) {
203                 printf("can't get prog info - %s\n", strerror(errno));
204                 return ret;
205         }
206         dummy_prog_id = info.id;
207 
208         signal(SIGINT, int_exit);
209         signal(SIGTERM, int_exit);
210 
211         /* populate virtual to physical port map */
212         ret = bpf_map_update_elem(tx_port_map_fd, &key, &ifindex_out, 0);
213         if (ret) {
214                 perror("bpf_update_elem");
215                 goto out;
216         }
217 
218         poll_stats(2, ifindex_out);
219 
220 out:
221         return 0;
222 }
223 

~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

kernel.org | git.kernel.org | LWN.net | Project Home | Wiki (Japanese) | Wiki (English) | SVN repository | Mail admin

Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.

osdn.jp