Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 1 | #define _GNU_SOURCE |
| 2 | |
| 3 | #include <stdio.h> |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 4 | #include <unistd.h> |
Olivier Deprez | 157378f | 2022-04-04 15:47:50 +0200 | [diff] [blame^] | 5 | #include <bpf/libbpf.h> |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 6 | |
| 7 | int main(int argc, char **argv) |
| 8 | { |
Olivier Deprez | 157378f | 2022-04-04 15:47:50 +0200 | [diff] [blame^] | 9 | struct bpf_link *link = NULL; |
| 10 | struct bpf_program *prog; |
| 11 | struct bpf_object *obj; |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 12 | char filename[256]; |
| 13 | char command[256]; |
Olivier Deprez | 157378f | 2022-04-04 15:47:50 +0200 | [diff] [blame^] | 14 | int ret = 0; |
| 15 | FILE *f; |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 16 | |
Olivier Deprez | 0e64123 | 2021-09-23 10:07:05 +0200 | [diff] [blame] | 17 | if (!argv[1]) { |
| 18 | fprintf(stderr, "ERROR: Run with the btrfs device argument!\n"); |
| 19 | return 0; |
| 20 | } |
| 21 | |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 22 | snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]); |
Olivier Deprez | 157378f | 2022-04-04 15:47:50 +0200 | [diff] [blame^] | 23 | obj = bpf_object__open_file(filename, NULL); |
| 24 | if (libbpf_get_error(obj)) { |
| 25 | fprintf(stderr, "ERROR: opening BPF object file failed\n"); |
| 26 | return 0; |
| 27 | } |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 28 | |
Olivier Deprez | 157378f | 2022-04-04 15:47:50 +0200 | [diff] [blame^] | 29 | prog = bpf_object__find_program_by_name(obj, "bpf_prog1"); |
| 30 | if (!prog) { |
| 31 | fprintf(stderr, "ERROR: finding a prog in obj file failed\n"); |
| 32 | goto cleanup; |
| 33 | } |
| 34 | |
| 35 | /* load BPF program */ |
| 36 | if (bpf_object__load(obj)) { |
| 37 | fprintf(stderr, "ERROR: loading BPF object file failed\n"); |
| 38 | goto cleanup; |
| 39 | } |
| 40 | |
| 41 | link = bpf_program__attach(prog); |
| 42 | if (libbpf_get_error(link)) { |
| 43 | fprintf(stderr, "ERROR: bpf_program__attach failed\n"); |
| 44 | link = NULL; |
| 45 | goto cleanup; |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 46 | } |
| 47 | |
| 48 | snprintf(command, 256, "mount %s tmpmnt/", argv[1]); |
| 49 | f = popen(command, "r"); |
| 50 | ret = pclose(f); |
| 51 | |
Olivier Deprez | 157378f | 2022-04-04 15:47:50 +0200 | [diff] [blame^] | 52 | cleanup: |
| 53 | bpf_link__destroy(link); |
| 54 | bpf_object__close(obj); |
Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 55 | return ret ? 0 : 1; |
| 56 | } |