test_cgrp2_sock2.c 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. // SPDX-License-Identifier: GPL-2.0
  2. /* eBPF example program:
  3. *
  4. * - Loads eBPF program
  5. *
  6. * The eBPF program loads a filter from file and attaches the
  7. * program to a cgroup using BPF_PROG_ATTACH
  8. */
  9. #define _GNU_SOURCE
  10. #include <stdio.h>
  11. #include <stdlib.h>
  12. #include <stddef.h>
  13. #include <string.h>
  14. #include <unistd.h>
  15. #include <assert.h>
  16. #include <errno.h>
  17. #include <fcntl.h>
  18. #include <net/if.h>
  19. #include <linux/bpf.h>
  20. #include <bpf/bpf.h>
  21. #include "bpf_insn.h"
  22. #include "bpf_load.h"
  23. static int usage(const char *argv0)
  24. {
  25. printf("Usage: %s cg-path filter-path [filter-id]\n", argv0);
  26. return EXIT_FAILURE;
  27. }
  28. int main(int argc, char **argv)
  29. {
  30. int cg_fd, ret, filter_id = 0;
  31. if (argc < 3)
  32. return usage(argv[0]);
  33. cg_fd = open(argv[1], O_DIRECTORY | O_RDONLY);
  34. if (cg_fd < 0) {
  35. printf("Failed to open cgroup path: '%s'\n", strerror(errno));
  36. return EXIT_FAILURE;
  37. }
  38. if (load_bpf_file(argv[2]))
  39. return EXIT_FAILURE;
  40. printf("Output from kernel verifier:\n%s\n-------\n", bpf_log_buf);
  41. if (argc > 3)
  42. filter_id = atoi(argv[3]);
  43. if (filter_id >= prog_cnt) {
  44. printf("Invalid program id; program not found in file\n");
  45. return EXIT_FAILURE;
  46. }
  47. ret = bpf_prog_attach(prog_fd[filter_id], cg_fd,
  48. BPF_CGROUP_INET_SOCK_CREATE, 0);
  49. if (ret < 0) {
  50. printf("Failed to attach prog to cgroup: '%s'\n",
  51. strerror(errno));
  52. return EXIT_FAILURE;
  53. }
  54. return EXIT_SUCCESS;
  55. }