1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun #include <stdio.h>
3*4882a593Smuzhiyun #include <assert.h>
4*4882a593Smuzhiyun #include <bpf/bpf.h>
5*4882a593Smuzhiyun #include <bpf/libbpf.h>
6*4882a593Smuzhiyun #include "sock_example.h"
7*4882a593Smuzhiyun #include <unistd.h>
8*4882a593Smuzhiyun #include <arpa/inet.h>
9*4882a593Smuzhiyun #include <sys/resource.h>
10*4882a593Smuzhiyun
11*4882a593Smuzhiyun struct flow_key_record {
12*4882a593Smuzhiyun __be32 src;
13*4882a593Smuzhiyun __be32 dst;
14*4882a593Smuzhiyun union {
15*4882a593Smuzhiyun __be32 ports;
16*4882a593Smuzhiyun __be16 port16[2];
17*4882a593Smuzhiyun };
18*4882a593Smuzhiyun __u32 ip_proto;
19*4882a593Smuzhiyun };
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun struct pair {
22*4882a593Smuzhiyun __u64 packets;
23*4882a593Smuzhiyun __u64 bytes;
24*4882a593Smuzhiyun };
25*4882a593Smuzhiyun
main(int argc,char ** argv)26*4882a593Smuzhiyun int main(int argc, char **argv)
27*4882a593Smuzhiyun {
28*4882a593Smuzhiyun int i, sock, key, fd, main_prog_fd, jmp_table_fd, hash_map_fd;
29*4882a593Smuzhiyun struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
30*4882a593Smuzhiyun struct bpf_program *prog;
31*4882a593Smuzhiyun struct bpf_object *obj;
32*4882a593Smuzhiyun const char *section;
33*4882a593Smuzhiyun char filename[256];
34*4882a593Smuzhiyun FILE *f;
35*4882a593Smuzhiyun
36*4882a593Smuzhiyun snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
37*4882a593Smuzhiyun setrlimit(RLIMIT_MEMLOCK, &r);
38*4882a593Smuzhiyun
39*4882a593Smuzhiyun obj = bpf_object__open_file(filename, NULL);
40*4882a593Smuzhiyun if (libbpf_get_error(obj)) {
41*4882a593Smuzhiyun fprintf(stderr, "ERROR: opening BPF object file failed\n");
42*4882a593Smuzhiyun return 0;
43*4882a593Smuzhiyun }
44*4882a593Smuzhiyun
45*4882a593Smuzhiyun /* load BPF program */
46*4882a593Smuzhiyun if (bpf_object__load(obj)) {
47*4882a593Smuzhiyun fprintf(stderr, "ERROR: loading BPF object file failed\n");
48*4882a593Smuzhiyun goto cleanup;
49*4882a593Smuzhiyun }
50*4882a593Smuzhiyun
51*4882a593Smuzhiyun jmp_table_fd = bpf_object__find_map_fd_by_name(obj, "jmp_table");
52*4882a593Smuzhiyun hash_map_fd = bpf_object__find_map_fd_by_name(obj, "hash_map");
53*4882a593Smuzhiyun if (jmp_table_fd < 0 || hash_map_fd < 0) {
54*4882a593Smuzhiyun fprintf(stderr, "ERROR: finding a map in obj file failed\n");
55*4882a593Smuzhiyun goto cleanup;
56*4882a593Smuzhiyun }
57*4882a593Smuzhiyun
58*4882a593Smuzhiyun bpf_object__for_each_program(prog, obj) {
59*4882a593Smuzhiyun fd = bpf_program__fd(prog);
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun section = bpf_program__section_name(prog);
62*4882a593Smuzhiyun if (sscanf(section, "socket/%d", &key) != 1) {
63*4882a593Smuzhiyun fprintf(stderr, "ERROR: finding prog failed\n");
64*4882a593Smuzhiyun goto cleanup;
65*4882a593Smuzhiyun }
66*4882a593Smuzhiyun
67*4882a593Smuzhiyun if (key == 0)
68*4882a593Smuzhiyun main_prog_fd = fd;
69*4882a593Smuzhiyun else
70*4882a593Smuzhiyun bpf_map_update_elem(jmp_table_fd, &key, &fd, BPF_ANY);
71*4882a593Smuzhiyun }
72*4882a593Smuzhiyun
73*4882a593Smuzhiyun sock = open_raw_sock("lo");
74*4882a593Smuzhiyun
75*4882a593Smuzhiyun /* attach BPF program to socket */
76*4882a593Smuzhiyun assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, &main_prog_fd,
77*4882a593Smuzhiyun sizeof(__u32)) == 0);
78*4882a593Smuzhiyun
79*4882a593Smuzhiyun if (argc > 1)
80*4882a593Smuzhiyun f = popen("ping -4 -c5 localhost", "r");
81*4882a593Smuzhiyun else
82*4882a593Smuzhiyun f = popen("netperf -l 4 localhost", "r");
83*4882a593Smuzhiyun (void) f;
84*4882a593Smuzhiyun
85*4882a593Smuzhiyun for (i = 0; i < 5; i++) {
86*4882a593Smuzhiyun struct flow_key_record key = {}, next_key;
87*4882a593Smuzhiyun struct pair value;
88*4882a593Smuzhiyun
89*4882a593Smuzhiyun sleep(1);
90*4882a593Smuzhiyun printf("IP src.port -> dst.port bytes packets\n");
91*4882a593Smuzhiyun while (bpf_map_get_next_key(hash_map_fd, &key, &next_key) == 0) {
92*4882a593Smuzhiyun bpf_map_lookup_elem(hash_map_fd, &next_key, &value);
93*4882a593Smuzhiyun printf("%s.%05d -> %s.%05d %12lld %12lld\n",
94*4882a593Smuzhiyun inet_ntoa((struct in_addr){htonl(next_key.src)}),
95*4882a593Smuzhiyun next_key.port16[0],
96*4882a593Smuzhiyun inet_ntoa((struct in_addr){htonl(next_key.dst)}),
97*4882a593Smuzhiyun next_key.port16[1],
98*4882a593Smuzhiyun value.bytes, value.packets);
99*4882a593Smuzhiyun key = next_key;
100*4882a593Smuzhiyun }
101*4882a593Smuzhiyun }
102*4882a593Smuzhiyun
103*4882a593Smuzhiyun cleanup:
104*4882a593Smuzhiyun bpf_object__close(obj);
105*4882a593Smuzhiyun return 0;
106*4882a593Smuzhiyun }
107