]>
Commit | Line | Data |
---|---|---|
86af8b41 BB |
1 | /* Copyright (c) 2016 PLUMgrid |
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 | #include <linux/bpf.h> | |
3993f2cb | 8 | #include <linux/if_link.h> |
86af8b41 BB |
9 | #include <assert.h> |
10 | #include <errno.h> | |
11 | #include <signal.h> | |
12 | #include <stdio.h> | |
13 | #include <stdlib.h> | |
14 | #include <string.h> | |
86af8b41 | 15 | #include <unistd.h> |
3993f2cb | 16 | #include <libgen.h> |
6dfca831 | 17 | #include <sys/resource.h> |
e00c7b21 | 18 | |
86af8b41 | 19 | #include "bpf_load.h" |
e00c7b21 | 20 | #include "bpf_util.h" |
86af8b41 BB |
21 | #include "libbpf.h" |
22 | ||
86af8b41 | 23 | static int ifindex; |
6387d011 | 24 | static __u32 xdp_flags; |
86af8b41 BB |
25 | |
26 | static void int_exit(int sig) | |
27 | { | |
6387d011 | 28 | set_link_xdp_fd(ifindex, -1, xdp_flags); |
86af8b41 BB |
29 | exit(0); |
30 | } | |
31 | ||
32 | /* simple per-protocol drop counter | |
33 | */ | |
34 | static void poll_stats(int interval) | |
35 | { | |
e00c7b21 | 36 | unsigned int nr_cpus = bpf_num_possible_cpus(); |
86af8b41 BB |
37 | const unsigned int nr_keys = 256; |
38 | __u64 values[nr_cpus], prev[nr_keys][nr_cpus]; | |
39 | __u32 key; | |
40 | int i; | |
41 | ||
42 | memset(prev, 0, sizeof(prev)); | |
43 | ||
44 | while (1) { | |
45 | sleep(interval); | |
46 | ||
47 | for (key = 0; key < nr_keys; key++) { | |
48 | __u64 sum = 0; | |
49 | ||
d40fc181 | 50 | assert(bpf_map_lookup_elem(map_fd[0], &key, values) == 0); |
86af8b41 BB |
51 | for (i = 0; i < nr_cpus; i++) |
52 | sum += (values[i] - prev[key][i]); | |
53 | if (sum) | |
54 | printf("proto %u: %10llu pkt/s\n", | |
55 | key, sum / interval); | |
56 | memcpy(prev[key], values, sizeof(values)); | |
57 | } | |
58 | } | |
59 | } | |
60 | ||
3993f2cb | 61 | static void usage(const char *prog) |
86af8b41 | 62 | { |
3993f2cb DA |
63 | fprintf(stderr, |
64 | "usage: %s [OPTS] IFINDEX\n\n" | |
65 | "OPTS:\n" | |
0489df9a DB |
66 | " -S use skb-mode\n" |
67 | " -N enforce native mode\n", | |
3993f2cb DA |
68 | prog); |
69 | } | |
86af8b41 | 70 | |
3993f2cb DA |
71 | int main(int argc, char **argv) |
72 | { | |
6dfca831 | 73 | struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY}; |
0489df9a | 74 | const char *optstr = "SN"; |
3993f2cb DA |
75 | char filename[256]; |
76 | int opt; | |
77 | ||
78 | while ((opt = getopt(argc, argv, optstr)) != -1) { | |
79 | switch (opt) { | |
80 | case 'S': | |
6387d011 | 81 | xdp_flags |= XDP_FLAGS_SKB_MODE; |
3993f2cb | 82 | break; |
0489df9a DB |
83 | case 'N': |
84 | xdp_flags |= XDP_FLAGS_DRV_MODE; | |
85 | break; | |
3993f2cb DA |
86 | default: |
87 | usage(basename(argv[0])); | |
88 | return 1; | |
89 | } | |
90 | } | |
86af8b41 | 91 | |
3993f2cb DA |
92 | if (optind == argc) { |
93 | usage(basename(argv[0])); | |
86af8b41 BB |
94 | return 1; |
95 | } | |
6dfca831 TD |
96 | |
97 | if (setrlimit(RLIMIT_MEMLOCK, &r)) { | |
98 | perror("setrlimit(RLIMIT_MEMLOCK)"); | |
99 | return 1; | |
100 | } | |
101 | ||
3993f2cb | 102 | ifindex = strtoul(argv[optind], NULL, 0); |
86af8b41 | 103 | |
3993f2cb | 104 | snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]); |
86af8b41 BB |
105 | |
106 | if (load_bpf_file(filename)) { | |
107 | printf("%s", bpf_log_buf); | |
108 | return 1; | |
109 | } | |
110 | ||
111 | if (!prog_fd[0]) { | |
112 | printf("load_bpf_file: %s\n", strerror(errno)); | |
113 | return 1; | |
114 | } | |
115 | ||
116 | signal(SIGINT, int_exit); | |
ad990dbe | 117 | signal(SIGTERM, int_exit); |
86af8b41 | 118 | |
6387d011 | 119 | if (set_link_xdp_fd(ifindex, prog_fd[0], xdp_flags) < 0) { |
86af8b41 BB |
120 | printf("link set xdp fd failed\n"); |
121 | return 1; | |
122 | } | |
123 | ||
124 | poll_stats(2); | |
125 | ||
126 | return 0; | |
127 | } |