1 #include <stdio.h> 2 #include <assert.h> 3 #include <linux/bpf.h> 4 #include "libbpf.h" 5 #include "bpf_load.h" 6 #include <unistd.h> 7 #include <arpa/inet.h> 8 9 struct pair { 10 __u64 packets; 11 __u64 bytes; 12 }; 13 14 int main(int ac, char **argv) 15 { 16 char filename[256]; 17 FILE *f; 18 int i, sock; 19 20 snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]); 21 22 if (load_bpf_file(filename)) { 23 printf("%s", bpf_log_buf); 24 return 1; 25 } 26 27 sock = open_raw_sock("lo"); 28 29 assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, prog_fd, 30 sizeof(prog_fd[0])) == 0); 31 32 f = popen("ping -c5 localhost", "r"); 33 (void) f; 34 35 for (i = 0; i < 5; i++) { 36 int key = 0, next_key; 37 struct pair value; 38 39 while (bpf_get_next_key(map_fd[0], &key, &next_key) == 0) { 40 bpf_lookup_elem(map_fd[0], &next_key, &value); 41 printf("ip %s bytes %lld packets %lld\n", 42 inet_ntoa((struct in_addr){htonl(next_key)}), 43 value.bytes, value.packets); 44 key = next_key; 45 } 46 sleep(1); 47 } 48 return 0; 49 } 50