sockex2_user.c 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  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/libbpf.h>
  7. #include "sock_example.h"
  8. #include <unistd.h>
  9. #include <arpa/inet.h>
  10. #include <sys/resource.h>
  11. struct pair {
  12. __u64 packets;
  13. __u64 bytes;
  14. };
  15. int main(int ac, char **argv)
  16. {
  17. struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
  18. struct bpf_object *obj;
  19. int map_fd, prog_fd;
  20. char filename[256];
  21. int i, sock;
  22. FILE *f;
  23. snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
  24. setrlimit(RLIMIT_MEMLOCK, &r);
  25. if (bpf_prog_load(filename, BPF_PROG_TYPE_SOCKET_FILTER,
  26. &obj, &prog_fd))
  27. return 1;
  28. map_fd = bpf_object__find_map_fd_by_name(obj, "hash_map");
  29. sock = open_raw_sock("lo");
  30. assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, &prog_fd,
  31. sizeof(prog_fd)) == 0);
  32. f = popen("ping -4 -c5 localhost", "r");
  33. (void) f;
  34. for (i = 0; i < 5; i++) {
  35. int key = 0, next_key;
  36. struct pair value;
  37. while (bpf_map_get_next_key(map_fd, &key, &next_key) == 0) {
  38. bpf_map_lookup_elem(map_fd, &next_key, &value);
  39. printf("ip %s bytes %lld packets %lld\n",
  40. inet_ntoa((struct in_addr){htonl(next_key)}),
  41. value.bytes, value.packets);
  42. key = next_key;
  43. }
  44. sleep(1);
  45. }
  46. return 0;
  47. }