1 #include <linux/unistd.h>
12 #include <sys/types.h>
13 #include <sys/socket.h>
18 #define BPF_F_PIN (1 << 0)
19 #define BPF_F_GET (1 << 1)
20 #define BPF_F_PIN_GET (BPF_F_PIN | BPF_F_GET)
22 #define BPF_F_KEY (1 << 2)
23 #define BPF_F_VAL (1 << 3)
24 #define BPF_F_KEY_VAL (BPF_F_KEY | BPF_F_VAL)
26 #define BPF_M_UNSPEC 0
30 static void usage(void)
32 printf("Usage: fds_example [...]\n");
33 printf(" -F <file> File to pin/get object\n");
34 printf(" -P |- pin object\n");
35 printf(" -G `- get object\n");
36 printf(" -m eBPF map mode\n");
37 printf(" -k <key> |- map key\n");
38 printf(" -v <value> `- map value\n");
39 printf(" -p eBPF prog mode\n");
40 printf(" -o <object> `- object file\n");
41 printf(" -h Display this help.\n");
44 static int bpf_map_create(void)
46 return bpf_create_map(BPF_MAP_TYPE_ARRAY, sizeof(uint32_t),
47 sizeof(uint32_t), 1024, 0);
50 static int bpf_prog_create(const char *object)
52 static const struct bpf_insn insns[] = {
53 BPF_MOV64_IMM(BPF_REG_0, 1),
58 assert(!load_bpf_file((char *)object));
61 return bpf_load_program(BPF_PROG_TYPE_SOCKET_FILTER,
62 insns, sizeof(insns), "GPL", 0,
63 bpf_log_buf, BPF_LOG_BUF_SIZE);
67 static int bpf_do_map(const char *file, uint32_t flags, uint32_t key,
72 if (flags & BPF_F_PIN) {
73 fd = bpf_map_create();
74 printf("bpf: map fd:%d (%s)\n", fd, strerror(errno));
77 ret = bpf_obj_pin(fd, file);
78 printf("bpf: pin ret:(%d,%s)\n", ret, strerror(errno));
81 fd = bpf_obj_get(file);
82 printf("bpf: get fd:%d (%s)\n", fd, strerror(errno));
86 if ((flags & BPF_F_KEY_VAL) == BPF_F_KEY_VAL) {
87 ret = bpf_map_update_elem(fd, &key, &value, 0);
88 printf("bpf: fd:%d u->(%u:%u) ret:(%d,%s)\n", fd, key, value,
89 ret, strerror(errno));
91 } else if (flags & BPF_F_KEY) {
92 ret = bpf_map_lookup_elem(fd, &key, &value);
93 printf("bpf: fd:%d l->(%u):%u ret:(%d,%s)\n", fd, key, value,
94 ret, strerror(errno));
101 static int bpf_do_prog(const char *file, uint32_t flags, const char *object)
105 if (flags & BPF_F_PIN) {
106 fd = bpf_prog_create(object);
107 printf("bpf: prog fd:%d (%s)\n", fd, strerror(errno));
110 ret = bpf_obj_pin(fd, file);
111 printf("bpf: pin ret:(%d,%s)\n", ret, strerror(errno));
114 fd = bpf_obj_get(file);
115 printf("bpf: get fd:%d (%s)\n", fd, strerror(errno));
119 sock = open_raw_sock("lo");
122 ret = setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, &fd, sizeof(fd));
123 printf("bpf: sock:%d <- fd:%d attached ret:(%d,%s)\n", sock, fd,
124 ret, strerror(errno));
130 int main(int argc, char **argv)
132 const char *file = NULL, *object = NULL;
133 uint32_t key = 0, value = 0, flags = 0;
134 int opt, mode = BPF_M_UNSPEC;
136 while ((opt = getopt(argc, argv, "F:PGmk:v:po:")) != -1) {
148 /* Map-related args */
153 key = strtoul(optarg, NULL, 0);
157 value = strtoul(optarg, NULL, 0);
160 /* Prog-related args */
172 if (!(flags & BPF_F_PIN_GET) || !file)
177 return bpf_do_map(file, flags, key, value);
179 return bpf_do_prog(file, flags, object);