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