18c2ecf20Sopenharmony_ci#define _GNU_SOURCE
28c2ecf20Sopenharmony_ci
38c2ecf20Sopenharmony_ci#include <stdio.h>
48c2ecf20Sopenharmony_ci#include <unistd.h>
58c2ecf20Sopenharmony_ci#include <bpf/libbpf.h>
68c2ecf20Sopenharmony_ci
78c2ecf20Sopenharmony_ciint main(int argc, char **argv)
88c2ecf20Sopenharmony_ci{
98c2ecf20Sopenharmony_ci	struct bpf_link *link = NULL;
108c2ecf20Sopenharmony_ci	struct bpf_program *prog;
118c2ecf20Sopenharmony_ci	struct bpf_object *obj;
128c2ecf20Sopenharmony_ci	char filename[256];
138c2ecf20Sopenharmony_ci	char command[256];
148c2ecf20Sopenharmony_ci	int ret = 0;
158c2ecf20Sopenharmony_ci	FILE *f;
168c2ecf20Sopenharmony_ci
178c2ecf20Sopenharmony_ci	if (!argv[1]) {
188c2ecf20Sopenharmony_ci		fprintf(stderr, "ERROR: Run with the btrfs device argument!\n");
198c2ecf20Sopenharmony_ci		return 0;
208c2ecf20Sopenharmony_ci	}
218c2ecf20Sopenharmony_ci
228c2ecf20Sopenharmony_ci	snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
238c2ecf20Sopenharmony_ci	obj = bpf_object__open_file(filename, NULL);
248c2ecf20Sopenharmony_ci	if (libbpf_get_error(obj)) {
258c2ecf20Sopenharmony_ci		fprintf(stderr, "ERROR: opening BPF object file failed\n");
268c2ecf20Sopenharmony_ci		return 0;
278c2ecf20Sopenharmony_ci	}
288c2ecf20Sopenharmony_ci
298c2ecf20Sopenharmony_ci	prog = bpf_object__find_program_by_name(obj, "bpf_prog1");
308c2ecf20Sopenharmony_ci	if (!prog) {
318c2ecf20Sopenharmony_ci		fprintf(stderr, "ERROR: finding a prog in obj file failed\n");
328c2ecf20Sopenharmony_ci		goto cleanup;
338c2ecf20Sopenharmony_ci	}
348c2ecf20Sopenharmony_ci
358c2ecf20Sopenharmony_ci	/* load BPF program */
368c2ecf20Sopenharmony_ci	if (bpf_object__load(obj)) {
378c2ecf20Sopenharmony_ci		fprintf(stderr, "ERROR: loading BPF object file failed\n");
388c2ecf20Sopenharmony_ci		goto cleanup;
398c2ecf20Sopenharmony_ci	}
408c2ecf20Sopenharmony_ci
418c2ecf20Sopenharmony_ci	link = bpf_program__attach(prog);
428c2ecf20Sopenharmony_ci	if (libbpf_get_error(link)) {
438c2ecf20Sopenharmony_ci		fprintf(stderr, "ERROR: bpf_program__attach failed\n");
448c2ecf20Sopenharmony_ci		link = NULL;
458c2ecf20Sopenharmony_ci		goto cleanup;
468c2ecf20Sopenharmony_ci	}
478c2ecf20Sopenharmony_ci
488c2ecf20Sopenharmony_ci	snprintf(command, 256, "mount %s tmpmnt/", argv[1]);
498c2ecf20Sopenharmony_ci	f = popen(command, "r");
508c2ecf20Sopenharmony_ci	ret = pclose(f);
518c2ecf20Sopenharmony_ci
528c2ecf20Sopenharmony_cicleanup:
538c2ecf20Sopenharmony_ci	bpf_link__destroy(link);
548c2ecf20Sopenharmony_ci	bpf_object__close(obj);
558c2ecf20Sopenharmony_ci	return ret ? 0 : 1;
568c2ecf20Sopenharmony_ci}
57