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

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

Version: ~ [ linux-5.8-rc3 ] ~ [ linux-5.7.5 ] ~ [ linux-5.6.19 ] ~ [ linux-5.5.19 ] ~ [ linux-5.4.48 ] ~ [ linux-5.3.18 ] ~ [ linux-5.2.21 ] ~ [ linux-5.1.21 ] ~ [ linux-5.0.21 ] ~ [ linux-4.20.17 ] ~ [ linux-4.19.129 ] ~ [ linux-4.18.20 ] ~ [ linux-4.17.19 ] ~ [ linux-4.16.18 ] ~ [ linux-4.15.18 ] ~ [ linux-4.14.185 ] ~ [ linux-4.13.16 ] ~ [ linux-4.12.14 ] ~ [ linux-4.11.12 ] ~ [ linux-4.10.17 ] ~ [ linux-4.9.228 ] ~ [ linux-4.8.17 ] ~ [ linux-4.7.10 ] ~ [ linux-4.6.7 ] ~ [ linux-4.5.7 ] ~ [ linux-4.4.228 ] ~ [ linux-4.3.6 ] ~ [ linux-4.2.8 ] ~ [ linux-4.1.52 ] ~ [ linux-4.0.9 ] ~ [ linux-3.19.8 ] ~ [ linux-3.18.140 ] ~ [ linux-3.17.8 ] ~ [ linux-3.16.85 ] ~ [ linux-3.15.10 ] ~ [ linux-3.14.79 ] ~ [ linux-3.13.11 ] ~ [ linux-3.12.74 ] ~ [ linux-3.11.10 ] ~ [ 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.5 ] ~ [ policy-sample ] ~
Architecture: ~ [ i386 ] ~ [ alpha ] ~ [ m68k ] ~ [ mips ] ~ [ ppc ] ~ [ sparc ] ~ [ sparc64 ] ~

  1 /* Copyright (c) 2017 Covalent IO, Inc. http://covalent.io
  2  *
  3  * This program is free software; you can redistribute it and/or
  4  * modify it under the terms of version 2 of the GNU General Public
  5  * License as published by the Free Software Foundation.
  6  *
  7  * This program is distributed in the hope that it will be useful, but
  8  * WITHOUT ANY WARRANTY; without even the implied warranty of
  9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 10  * General Public License for more details.
 11  */
 12 #include <linux/bpf.h>
 13 #include <linux/if_link.h>
 14 #include <assert.h>
 15 #include <errno.h>
 16 #include <signal.h>
 17 #include <stdio.h>
 18 #include <stdlib.h>
 19 #include <stdbool.h>
 20 #include <string.h>
 21 #include <unistd.h>
 22 #include <libgen.h>
 23 #include <sys/resource.h>
 24 
 25 #include "bpf_load.h"
 26 #include "bpf_util.h"
 27 #include <bpf/bpf.h>
 28 
 29 static int ifindex_in;
 30 static int ifindex_out;
 31 static bool ifindex_out_xdp_dummy_attached = true;
 32 
 33 static __u32 xdp_flags;
 34 
 35 static void int_exit(int sig)
 36 {
 37         bpf_set_link_xdp_fd(ifindex_in, -1, xdp_flags);
 38         if (ifindex_out_xdp_dummy_attached)
 39                 bpf_set_link_xdp_fd(ifindex_out, -1, xdp_flags);
 40         exit(0);
 41 }
 42 
 43 static void poll_stats(int interval, int ifindex)
 44 {
 45         unsigned int nr_cpus = bpf_num_possible_cpus();
 46         __u64 values[nr_cpus], prev[nr_cpus];
 47 
 48         memset(prev, 0, sizeof(prev));
 49 
 50         while (1) {
 51                 __u64 sum = 0;
 52                 __u32 key = 0;
 53                 int i;
 54 
 55                 sleep(interval);
 56                 assert(bpf_map_lookup_elem(map_fd[1], &key, values) == 0);
 57                 for (i = 0; i < nr_cpus; i++)
 58                         sum += (values[i] - prev[i]);
 59                 if (sum)
 60                         printf("ifindex %i: %10llu pkt/s\n",
 61                                ifindex, sum / interval);
 62                 memcpy(prev, values, sizeof(values));
 63         }
 64 }
 65 
 66 static void usage(const char *prog)
 67 {
 68         fprintf(stderr,
 69                 "usage: %s [OPTS] IFINDEX_IN IFINDEX_OUT\n\n"
 70                 "OPTS:\n"
 71                 "    -S    use skb-mode\n"
 72                 "    -N    enforce native mode\n",
 73                 prog);
 74 }
 75 
 76 int main(int argc, char **argv)
 77 {
 78         struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
 79         const char *optstr = "SN";
 80         char filename[256];
 81         int ret, opt, key = 0;
 82 
 83         while ((opt = getopt(argc, argv, optstr)) != -1) {
 84                 switch (opt) {
 85                 case 'S':
 86                         xdp_flags |= XDP_FLAGS_SKB_MODE;
 87                         break;
 88                 case 'N':
 89                         xdp_flags |= XDP_FLAGS_DRV_MODE;
 90                         break;
 91                 default:
 92                         usage(basename(argv[0]));
 93                         return 1;
 94                 }
 95         }
 96 
 97         if (optind == argc) {
 98                 printf("usage: %s IFINDEX_IN IFINDEX_OUT\n", argv[0]);
 99                 return 1;
100         }
101 
102         if (setrlimit(RLIMIT_MEMLOCK, &r)) {
103                 perror("setrlimit(RLIMIT_MEMLOCK)");
104                 return 1;
105         }
106 
107         ifindex_in = strtoul(argv[optind], NULL, 0);
108         ifindex_out = strtoul(argv[optind + 1], NULL, 0);
109         printf("input: %d output: %d\n", ifindex_in, ifindex_out);
110 
111         snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
112 
113         if (load_bpf_file(filename)) {
114                 printf("%s", bpf_log_buf);
115                 return 1;
116         }
117 
118         if (!prog_fd[0]) {
119                 printf("load_bpf_file: %s\n", strerror(errno));
120                 return 1;
121         }
122 
123         if (bpf_set_link_xdp_fd(ifindex_in, prog_fd[0], xdp_flags) < 0) {
124                 printf("ERROR: link set xdp fd failed on %d\n", ifindex_in);
125                 return 1;
126         }
127 
128         /* Loading dummy XDP prog on out-device */
129         if (bpf_set_link_xdp_fd(ifindex_out, prog_fd[1],
130                             (xdp_flags | XDP_FLAGS_UPDATE_IF_NOEXIST)) < 0) {
131                 printf("WARN: link set xdp fd failed on %d\n", ifindex_out);
132                 ifindex_out_xdp_dummy_attached = false;
133         }
134 
135         signal(SIGINT, int_exit);
136         signal(SIGTERM, int_exit);
137 
138         printf("map[0] (vports) = %i, map[1] (map) = %i, map[2] (count) = %i\n",
139                 map_fd[0], map_fd[1], map_fd[2]);
140 
141         /* populate virtual to physical port map */
142         ret = bpf_map_update_elem(map_fd[0], &key, &ifindex_out, 0);
143         if (ret) {
144                 perror("bpf_update_elem");
145                 goto out;
146         }
147 
148         poll_stats(2, ifindex_out);
149 
150 out:
151         return 0;
152 }
153 

~ [ 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