1 // SPDX-License-Identifier: GPL-2.0
2 #include <stdio.h>
3 #include <assert.h>
4 #include <linux/bpf.h>
5 #include <bpf/bpf.h>
6 #include "bpf_load.h"
7 #include "sock_example.h"
8 #include <unistd.h>
9 #include <arpa/inet.h>
10
main(int ac,char ** argv)11 int main(int ac, char **argv)
12 {
13 char filename[256];
14 FILE *f;
15 int i, sock;
16
17 snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
18
19 if (load_bpf_file(filename)) {
20 printf("%s", bpf_log_buf);
21 return 1;
22 }
23
24 sock = open_raw_sock("lo");
25
26 assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, prog_fd,
27 sizeof(prog_fd[0])) == 0);
28
29 f = popen("ping -c5 localhost", "r");
30 (void) f;
31
32 for (i = 0; i < 5; i++) {
33 long long tcp_cnt, udp_cnt, icmp_cnt;
34 int key;
35
36 key = IPPROTO_TCP;
37 assert(bpf_map_lookup_elem(map_fd[0], &key, &tcp_cnt) == 0);
38
39 key = IPPROTO_UDP;
40 assert(bpf_map_lookup_elem(map_fd[0], &key, &udp_cnt) == 0);
41
42 key = IPPROTO_ICMP;
43 assert(bpf_map_lookup_elem(map_fd[0], &key, &icmp_cnt) == 0);
44
45 printf("TCP %lld UDP %lld ICMP %lld bytes\n",
46 tcp_cnt, udp_cnt, icmp_cnt);
47 sleep(1);
48 }
49
50 return 0;
51 }
52
