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