xtensa: implement task_user_regset_view
[linux/fpc-iii.git] / tools / bpf / bpftool / prog.c
blobccee180dfb761248f078ffa4b2154793d0308420
1 /*
2 * Copyright (C) 2017-2018 Netronome Systems, Inc.
4 * This software is dual licensed under the GNU General License Version 2,
5 * June 1991 as shown in the file COPYING in the top-level directory of this
6 * source tree or the BSD 2-Clause License provided below. You have the
7 * option to license this software under the complete terms of either license.
9 * The BSD 2-Clause License:
11 * Redistribution and use in source and binary forms, with or
12 * without modification, are permitted provided that the following
13 * conditions are met:
15 * 1. Redistributions of source code must retain the above
16 * copyright notice, this list of conditions and the following
17 * disclaimer.
19 * 2. Redistributions in binary form must reproduce the above
20 * copyright notice, this list of conditions and the following
21 * disclaimer in the documentation and/or other materials
22 * provided with the distribution.
24 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
25 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
26 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
27 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
28 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
29 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
30 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
31 * SOFTWARE.
34 #define _GNU_SOURCE
35 #include <errno.h>
36 #include <fcntl.h>
37 #include <stdarg.h>
38 #include <stdio.h>
39 #include <stdlib.h>
40 #include <string.h>
41 #include <time.h>
42 #include <unistd.h>
43 #include <net/if.h>
44 #include <sys/types.h>
45 #include <sys/stat.h>
47 #include <linux/err.h>
49 #include <bpf.h>
50 #include <libbpf.h>
52 #include "cfg.h"
53 #include "main.h"
54 #include "xlated_dumper.h"
56 static const char * const prog_type_name[] = {
57 [BPF_PROG_TYPE_UNSPEC] = "unspec",
58 [BPF_PROG_TYPE_SOCKET_FILTER] = "socket_filter",
59 [BPF_PROG_TYPE_KPROBE] = "kprobe",
60 [BPF_PROG_TYPE_SCHED_CLS] = "sched_cls",
61 [BPF_PROG_TYPE_SCHED_ACT] = "sched_act",
62 [BPF_PROG_TYPE_TRACEPOINT] = "tracepoint",
63 [BPF_PROG_TYPE_XDP] = "xdp",
64 [BPF_PROG_TYPE_PERF_EVENT] = "perf_event",
65 [BPF_PROG_TYPE_CGROUP_SKB] = "cgroup_skb",
66 [BPF_PROG_TYPE_CGROUP_SOCK] = "cgroup_sock",
67 [BPF_PROG_TYPE_LWT_IN] = "lwt_in",
68 [BPF_PROG_TYPE_LWT_OUT] = "lwt_out",
69 [BPF_PROG_TYPE_LWT_XMIT] = "lwt_xmit",
70 [BPF_PROG_TYPE_SOCK_OPS] = "sock_ops",
71 [BPF_PROG_TYPE_SK_SKB] = "sk_skb",
72 [BPF_PROG_TYPE_CGROUP_DEVICE] = "cgroup_device",
73 [BPF_PROG_TYPE_SK_MSG] = "sk_msg",
74 [BPF_PROG_TYPE_RAW_TRACEPOINT] = "raw_tracepoint",
75 [BPF_PROG_TYPE_CGROUP_SOCK_ADDR] = "cgroup_sock_addr",
76 [BPF_PROG_TYPE_LIRC_MODE2] = "lirc_mode2",
77 [BPF_PROG_TYPE_FLOW_DISSECTOR] = "flow_dissector",
80 static const char * const attach_type_strings[] = {
81 [BPF_SK_SKB_STREAM_PARSER] = "stream_parser",
82 [BPF_SK_SKB_STREAM_VERDICT] = "stream_verdict",
83 [BPF_SK_MSG_VERDICT] = "msg_verdict",
84 [__MAX_BPF_ATTACH_TYPE] = NULL,
87 enum bpf_attach_type parse_attach_type(const char *str)
89 enum bpf_attach_type type;
91 for (type = 0; type < __MAX_BPF_ATTACH_TYPE; type++) {
92 if (attach_type_strings[type] &&
93 is_prefix(str, attach_type_strings[type]))
94 return type;
97 return __MAX_BPF_ATTACH_TYPE;
100 static void print_boot_time(__u64 nsecs, char *buf, unsigned int size)
102 struct timespec real_time_ts, boot_time_ts;
103 time_t wallclock_secs;
104 struct tm load_tm;
106 buf[--size] = '\0';
108 if (clock_gettime(CLOCK_REALTIME, &real_time_ts) ||
109 clock_gettime(CLOCK_BOOTTIME, &boot_time_ts)) {
110 perror("Can't read clocks");
111 snprintf(buf, size, "%llu", nsecs / 1000000000);
112 return;
115 wallclock_secs = (real_time_ts.tv_sec - boot_time_ts.tv_sec) +
116 (real_time_ts.tv_nsec - boot_time_ts.tv_nsec + nsecs) /
117 1000000000;
120 if (!localtime_r(&wallclock_secs, &load_tm)) {
121 snprintf(buf, size, "%llu", nsecs / 1000000000);
122 return;
125 if (json_output)
126 strftime(buf, size, "%s", &load_tm);
127 else
128 strftime(buf, size, "%FT%T%z", &load_tm);
131 static int prog_fd_by_tag(unsigned char *tag)
133 struct bpf_prog_info info = {};
134 __u32 len = sizeof(info);
135 unsigned int id = 0;
136 int err;
137 int fd;
139 while (true) {
140 err = bpf_prog_get_next_id(id, &id);
141 if (err) {
142 p_err("%s", strerror(errno));
143 return -1;
146 fd = bpf_prog_get_fd_by_id(id);
147 if (fd < 0) {
148 p_err("can't get prog by id (%u): %s",
149 id, strerror(errno));
150 return -1;
153 err = bpf_obj_get_info_by_fd(fd, &info, &len);
154 if (err) {
155 p_err("can't get prog info (%u): %s",
156 id, strerror(errno));
157 close(fd);
158 return -1;
161 if (!memcmp(tag, info.tag, BPF_TAG_SIZE))
162 return fd;
164 close(fd);
168 int prog_parse_fd(int *argc, char ***argv)
170 int fd;
172 if (is_prefix(**argv, "id")) {
173 unsigned int id;
174 char *endptr;
176 NEXT_ARGP();
178 id = strtoul(**argv, &endptr, 0);
179 if (*endptr) {
180 p_err("can't parse %s as ID", **argv);
181 return -1;
183 NEXT_ARGP();
185 fd = bpf_prog_get_fd_by_id(id);
186 if (fd < 0)
187 p_err("get by id (%u): %s", id, strerror(errno));
188 return fd;
189 } else if (is_prefix(**argv, "tag")) {
190 unsigned char tag[BPF_TAG_SIZE];
192 NEXT_ARGP();
194 if (sscanf(**argv, BPF_TAG_FMT, tag, tag + 1, tag + 2,
195 tag + 3, tag + 4, tag + 5, tag + 6, tag + 7)
196 != BPF_TAG_SIZE) {
197 p_err("can't parse tag");
198 return -1;
200 NEXT_ARGP();
202 return prog_fd_by_tag(tag);
203 } else if (is_prefix(**argv, "pinned")) {
204 char *path;
206 NEXT_ARGP();
208 path = **argv;
209 NEXT_ARGP();
211 return open_obj_pinned_any(path, BPF_OBJ_PROG);
214 p_err("expected 'id', 'tag' or 'pinned', got: '%s'?", **argv);
215 return -1;
218 static void show_prog_maps(int fd, u32 num_maps)
220 struct bpf_prog_info info = {};
221 __u32 len = sizeof(info);
222 __u32 map_ids[num_maps];
223 unsigned int i;
224 int err;
226 info.nr_map_ids = num_maps;
227 info.map_ids = ptr_to_u64(map_ids);
229 err = bpf_obj_get_info_by_fd(fd, &info, &len);
230 if (err || !info.nr_map_ids)
231 return;
233 if (json_output) {
234 jsonw_name(json_wtr, "map_ids");
235 jsonw_start_array(json_wtr);
236 for (i = 0; i < info.nr_map_ids; i++)
237 jsonw_uint(json_wtr, map_ids[i]);
238 jsonw_end_array(json_wtr);
239 } else {
240 printf(" map_ids ");
241 for (i = 0; i < info.nr_map_ids; i++)
242 printf("%u%s", map_ids[i],
243 i == info.nr_map_ids - 1 ? "" : ",");
247 static void print_prog_json(struct bpf_prog_info *info, int fd)
249 char *memlock;
251 jsonw_start_object(json_wtr);
252 jsonw_uint_field(json_wtr, "id", info->id);
253 if (info->type < ARRAY_SIZE(prog_type_name))
254 jsonw_string_field(json_wtr, "type",
255 prog_type_name[info->type]);
256 else
257 jsonw_uint_field(json_wtr, "type", info->type);
259 if (*info->name)
260 jsonw_string_field(json_wtr, "name", info->name);
262 jsonw_name(json_wtr, "tag");
263 jsonw_printf(json_wtr, "\"" BPF_TAG_FMT "\"",
264 info->tag[0], info->tag[1], info->tag[2], info->tag[3],
265 info->tag[4], info->tag[5], info->tag[6], info->tag[7]);
267 jsonw_bool_field(json_wtr, "gpl_compatible", info->gpl_compatible);
269 print_dev_json(info->ifindex, info->netns_dev, info->netns_ino);
271 if (info->load_time) {
272 char buf[32];
274 print_boot_time(info->load_time, buf, sizeof(buf));
276 /* Piggy back on load_time, since 0 uid is a valid one */
277 jsonw_name(json_wtr, "loaded_at");
278 jsonw_printf(json_wtr, "%s", buf);
279 jsonw_uint_field(json_wtr, "uid", info->created_by_uid);
282 jsonw_uint_field(json_wtr, "bytes_xlated", info->xlated_prog_len);
284 if (info->jited_prog_len) {
285 jsonw_bool_field(json_wtr, "jited", true);
286 jsonw_uint_field(json_wtr, "bytes_jited", info->jited_prog_len);
287 } else {
288 jsonw_bool_field(json_wtr, "jited", false);
291 memlock = get_fdinfo(fd, "memlock");
292 if (memlock)
293 jsonw_int_field(json_wtr, "bytes_memlock", atoi(memlock));
294 free(memlock);
296 if (info->nr_map_ids)
297 show_prog_maps(fd, info->nr_map_ids);
299 if (!hash_empty(prog_table.table)) {
300 struct pinned_obj *obj;
302 jsonw_name(json_wtr, "pinned");
303 jsonw_start_array(json_wtr);
304 hash_for_each_possible(prog_table.table, obj, hash, info->id) {
305 if (obj->id == info->id)
306 jsonw_string(json_wtr, obj->path);
308 jsonw_end_array(json_wtr);
311 jsonw_end_object(json_wtr);
314 static void print_prog_plain(struct bpf_prog_info *info, int fd)
316 char *memlock;
318 printf("%u: ", info->id);
319 if (info->type < ARRAY_SIZE(prog_type_name))
320 printf("%s ", prog_type_name[info->type]);
321 else
322 printf("type %u ", info->type);
324 if (*info->name)
325 printf("name %s ", info->name);
327 printf("tag ");
328 fprint_hex(stdout, info->tag, BPF_TAG_SIZE, "");
329 print_dev_plain(info->ifindex, info->netns_dev, info->netns_ino);
330 printf("%s", info->gpl_compatible ? " gpl" : "");
331 printf("\n");
333 if (info->load_time) {
334 char buf[32];
336 print_boot_time(info->load_time, buf, sizeof(buf));
338 /* Piggy back on load_time, since 0 uid is a valid one */
339 printf("\tloaded_at %s uid %u\n", buf, info->created_by_uid);
342 printf("\txlated %uB", info->xlated_prog_len);
344 if (info->jited_prog_len)
345 printf(" jited %uB", info->jited_prog_len);
346 else
347 printf(" not jited");
349 memlock = get_fdinfo(fd, "memlock");
350 if (memlock)
351 printf(" memlock %sB", memlock);
352 free(memlock);
354 if (info->nr_map_ids)
355 show_prog_maps(fd, info->nr_map_ids);
357 if (!hash_empty(prog_table.table)) {
358 struct pinned_obj *obj;
360 hash_for_each_possible(prog_table.table, obj, hash, info->id) {
361 if (obj->id == info->id)
362 printf("\n\tpinned %s", obj->path);
366 printf("\n");
369 static int show_prog(int fd)
371 struct bpf_prog_info info = {};
372 __u32 len = sizeof(info);
373 int err;
375 err = bpf_obj_get_info_by_fd(fd, &info, &len);
376 if (err) {
377 p_err("can't get prog info: %s", strerror(errno));
378 return -1;
381 if (json_output)
382 print_prog_json(&info, fd);
383 else
384 print_prog_plain(&info, fd);
386 return 0;
389 static int do_show(int argc, char **argv)
391 __u32 id = 0;
392 int err;
393 int fd;
395 if (show_pinned)
396 build_pinned_obj_table(&prog_table, BPF_OBJ_PROG);
398 if (argc == 2) {
399 fd = prog_parse_fd(&argc, &argv);
400 if (fd < 0)
401 return -1;
403 return show_prog(fd);
406 if (argc)
407 return BAD_ARG();
409 if (json_output)
410 jsonw_start_array(json_wtr);
411 while (true) {
412 err = bpf_prog_get_next_id(id, &id);
413 if (err) {
414 if (errno == ENOENT) {
415 err = 0;
416 break;
418 p_err("can't get next program: %s%s", strerror(errno),
419 errno == EINVAL ? " -- kernel too old?" : "");
420 err = -1;
421 break;
424 fd = bpf_prog_get_fd_by_id(id);
425 if (fd < 0) {
426 if (errno == ENOENT)
427 continue;
428 p_err("can't get prog by id (%u): %s",
429 id, strerror(errno));
430 err = -1;
431 break;
434 err = show_prog(fd);
435 close(fd);
436 if (err)
437 break;
440 if (json_output)
441 jsonw_end_array(json_wtr);
443 return err;
446 static int do_dump(int argc, char **argv)
448 unsigned long *func_ksyms = NULL;
449 struct bpf_prog_info info = {};
450 unsigned int *func_lens = NULL;
451 const char *disasm_opt = NULL;
452 unsigned int nr_func_ksyms;
453 unsigned int nr_func_lens;
454 struct dump_data dd = {};
455 __u32 len = sizeof(info);
456 unsigned int buf_size;
457 char *filepath = NULL;
458 bool opcodes = false;
459 bool visual = false;
460 unsigned char *buf;
461 __u32 *member_len;
462 __u64 *member_ptr;
463 ssize_t n;
464 int err;
465 int fd;
467 if (is_prefix(*argv, "jited")) {
468 member_len = &info.jited_prog_len;
469 member_ptr = &info.jited_prog_insns;
470 } else if (is_prefix(*argv, "xlated")) {
471 member_len = &info.xlated_prog_len;
472 member_ptr = &info.xlated_prog_insns;
473 } else {
474 p_err("expected 'xlated' or 'jited', got: %s", *argv);
475 return -1;
477 NEXT_ARG();
479 if (argc < 2)
480 usage();
482 fd = prog_parse_fd(&argc, &argv);
483 if (fd < 0)
484 return -1;
486 if (is_prefix(*argv, "file")) {
487 NEXT_ARG();
488 if (!argc) {
489 p_err("expected file path");
490 return -1;
493 filepath = *argv;
494 NEXT_ARG();
495 } else if (is_prefix(*argv, "opcodes")) {
496 opcodes = true;
497 NEXT_ARG();
498 } else if (is_prefix(*argv, "visual")) {
499 visual = true;
500 NEXT_ARG();
503 if (argc) {
504 usage();
505 return -1;
508 err = bpf_obj_get_info_by_fd(fd, &info, &len);
509 if (err) {
510 p_err("can't get prog info: %s", strerror(errno));
511 return -1;
514 if (!*member_len) {
515 p_info("no instructions returned");
516 close(fd);
517 return 0;
520 buf_size = *member_len;
522 buf = malloc(buf_size);
523 if (!buf) {
524 p_err("mem alloc failed");
525 close(fd);
526 return -1;
529 nr_func_ksyms = info.nr_jited_ksyms;
530 if (nr_func_ksyms) {
531 func_ksyms = malloc(nr_func_ksyms * sizeof(__u64));
532 if (!func_ksyms) {
533 p_err("mem alloc failed");
534 close(fd);
535 goto err_free;
539 nr_func_lens = info.nr_jited_func_lens;
540 if (nr_func_lens) {
541 func_lens = malloc(nr_func_lens * sizeof(__u32));
542 if (!func_lens) {
543 p_err("mem alloc failed");
544 close(fd);
545 goto err_free;
549 memset(&info, 0, sizeof(info));
551 *member_ptr = ptr_to_u64(buf);
552 *member_len = buf_size;
553 info.jited_ksyms = ptr_to_u64(func_ksyms);
554 info.nr_jited_ksyms = nr_func_ksyms;
555 info.jited_func_lens = ptr_to_u64(func_lens);
556 info.nr_jited_func_lens = nr_func_lens;
558 err = bpf_obj_get_info_by_fd(fd, &info, &len);
559 close(fd);
560 if (err) {
561 p_err("can't get prog info: %s", strerror(errno));
562 goto err_free;
565 if (*member_len > buf_size) {
566 p_err("too many instructions returned");
567 goto err_free;
570 if (info.nr_jited_ksyms > nr_func_ksyms) {
571 p_err("too many addresses returned");
572 goto err_free;
575 if (info.nr_jited_func_lens > nr_func_lens) {
576 p_err("too many values returned");
577 goto err_free;
580 if ((member_len == &info.jited_prog_len &&
581 info.jited_prog_insns == 0) ||
582 (member_len == &info.xlated_prog_len &&
583 info.xlated_prog_insns == 0)) {
584 p_err("error retrieving insn dump: kernel.kptr_restrict set?");
585 goto err_free;
588 if (filepath) {
589 fd = open(filepath, O_WRONLY | O_CREAT | O_TRUNC, 0600);
590 if (fd < 0) {
591 p_err("can't open file %s: %s", filepath,
592 strerror(errno));
593 goto err_free;
596 n = write(fd, buf, *member_len);
597 close(fd);
598 if (n != *member_len) {
599 p_err("error writing output file: %s",
600 n < 0 ? strerror(errno) : "short write");
601 goto err_free;
604 if (json_output)
605 jsonw_null(json_wtr);
606 } else if (member_len == &info.jited_prog_len) {
607 const char *name = NULL;
609 if (info.ifindex) {
610 name = ifindex_to_bfd_params(info.ifindex,
611 info.netns_dev,
612 info.netns_ino,
613 &disasm_opt);
614 if (!name)
615 goto err_free;
618 if (info.nr_jited_func_lens && info.jited_func_lens) {
619 struct kernel_sym *sym = NULL;
620 char sym_name[SYM_MAX_NAME];
621 unsigned char *img = buf;
622 __u64 *ksyms = NULL;
623 __u32 *lens;
624 __u32 i;
626 if (info.nr_jited_ksyms) {
627 kernel_syms_load(&dd);
628 ksyms = (__u64 *) info.jited_ksyms;
631 if (json_output)
632 jsonw_start_array(json_wtr);
634 lens = (__u32 *) info.jited_func_lens;
635 for (i = 0; i < info.nr_jited_func_lens; i++) {
636 if (ksyms) {
637 sym = kernel_syms_search(&dd, ksyms[i]);
638 if (sym)
639 sprintf(sym_name, "%s", sym->name);
640 else
641 sprintf(sym_name, "0x%016llx", ksyms[i]);
642 } else {
643 strcpy(sym_name, "unknown");
646 if (json_output) {
647 jsonw_start_object(json_wtr);
648 jsonw_name(json_wtr, "name");
649 jsonw_string(json_wtr, sym_name);
650 jsonw_name(json_wtr, "insns");
651 } else {
652 printf("%s:\n", sym_name);
655 disasm_print_insn(img, lens[i], opcodes, name,
656 disasm_opt);
657 img += lens[i];
659 if (json_output)
660 jsonw_end_object(json_wtr);
661 else
662 printf("\n");
665 if (json_output)
666 jsonw_end_array(json_wtr);
667 } else {
668 disasm_print_insn(buf, *member_len, opcodes, name,
669 disasm_opt);
671 } else if (visual) {
672 if (json_output)
673 jsonw_null(json_wtr);
674 else
675 dump_xlated_cfg(buf, *member_len);
676 } else {
677 kernel_syms_load(&dd);
678 dd.nr_jited_ksyms = info.nr_jited_ksyms;
679 dd.jited_ksyms = (__u64 *) info.jited_ksyms;
681 if (json_output)
682 dump_xlated_json(&dd, buf, *member_len, opcodes);
683 else
684 dump_xlated_plain(&dd, buf, *member_len, opcodes);
685 kernel_syms_destroy(&dd);
688 free(buf);
689 free(func_ksyms);
690 free(func_lens);
691 return 0;
693 err_free:
694 free(buf);
695 free(func_ksyms);
696 free(func_lens);
697 return -1;
700 static int do_pin(int argc, char **argv)
702 int err;
704 err = do_pin_any(argc, argv, bpf_prog_get_fd_by_id);
705 if (!err && json_output)
706 jsonw_null(json_wtr);
707 return err;
710 struct map_replace {
711 int idx;
712 int fd;
713 char *name;
716 int map_replace_compar(const void *p1, const void *p2)
718 const struct map_replace *a = p1, *b = p2;
720 return a->idx - b->idx;
723 static int do_attach(int argc, char **argv)
725 enum bpf_attach_type attach_type;
726 int err, mapfd, progfd;
728 if (!REQ_ARGS(5)) {
729 p_err("too few parameters for map attach");
730 return -EINVAL;
733 progfd = prog_parse_fd(&argc, &argv);
734 if (progfd < 0)
735 return progfd;
737 attach_type = parse_attach_type(*argv);
738 if (attach_type == __MAX_BPF_ATTACH_TYPE) {
739 p_err("invalid attach type");
740 return -EINVAL;
742 NEXT_ARG();
744 mapfd = map_parse_fd(&argc, &argv);
745 if (mapfd < 0)
746 return mapfd;
748 err = bpf_prog_attach(progfd, mapfd, attach_type, 0);
749 if (err) {
750 p_err("failed prog attach to map");
751 return -EINVAL;
754 if (json_output)
755 jsonw_null(json_wtr);
756 return 0;
759 static int do_detach(int argc, char **argv)
761 enum bpf_attach_type attach_type;
762 int err, mapfd, progfd;
764 if (!REQ_ARGS(5)) {
765 p_err("too few parameters for map detach");
766 return -EINVAL;
769 progfd = prog_parse_fd(&argc, &argv);
770 if (progfd < 0)
771 return progfd;
773 attach_type = parse_attach_type(*argv);
774 if (attach_type == __MAX_BPF_ATTACH_TYPE) {
775 p_err("invalid attach type");
776 return -EINVAL;
778 NEXT_ARG();
780 mapfd = map_parse_fd(&argc, &argv);
781 if (mapfd < 0)
782 return mapfd;
784 err = bpf_prog_detach2(progfd, mapfd, attach_type);
785 if (err) {
786 p_err("failed prog detach from map");
787 return -EINVAL;
790 if (json_output)
791 jsonw_null(json_wtr);
792 return 0;
794 static int do_load(int argc, char **argv)
796 enum bpf_attach_type expected_attach_type;
797 struct bpf_object_open_attr attr = {
798 .prog_type = BPF_PROG_TYPE_UNSPEC,
800 struct map_replace *map_replace = NULL;
801 unsigned int old_map_fds = 0;
802 struct bpf_program *prog;
803 struct bpf_object *obj;
804 struct bpf_map *map;
805 const char *pinfile;
806 unsigned int i, j;
807 __u32 ifindex = 0;
808 int idx, err;
810 if (!REQ_ARGS(2))
811 return -1;
812 attr.file = GET_ARG();
813 pinfile = GET_ARG();
815 while (argc) {
816 if (is_prefix(*argv, "type")) {
817 char *type;
819 NEXT_ARG();
821 if (attr.prog_type != BPF_PROG_TYPE_UNSPEC) {
822 p_err("program type already specified");
823 goto err_free_reuse_maps;
825 if (!REQ_ARGS(1))
826 goto err_free_reuse_maps;
828 /* Put a '/' at the end of type to appease libbpf */
829 type = malloc(strlen(*argv) + 2);
830 if (!type) {
831 p_err("mem alloc failed");
832 goto err_free_reuse_maps;
834 *type = 0;
835 strcat(type, *argv);
836 strcat(type, "/");
838 err = libbpf_prog_type_by_name(type, &attr.prog_type,
839 &expected_attach_type);
840 free(type);
841 if (err < 0) {
842 p_err("unknown program type '%s'", *argv);
843 goto err_free_reuse_maps;
845 NEXT_ARG();
846 } else if (is_prefix(*argv, "map")) {
847 void *new_map_replace;
848 char *endptr, *name;
849 int fd;
851 NEXT_ARG();
853 if (!REQ_ARGS(4))
854 goto err_free_reuse_maps;
856 if (is_prefix(*argv, "idx")) {
857 NEXT_ARG();
859 idx = strtoul(*argv, &endptr, 0);
860 if (*endptr) {
861 p_err("can't parse %s as IDX", *argv);
862 goto err_free_reuse_maps;
864 name = NULL;
865 } else if (is_prefix(*argv, "name")) {
866 NEXT_ARG();
868 name = *argv;
869 idx = -1;
870 } else {
871 p_err("expected 'idx' or 'name', got: '%s'?",
872 *argv);
873 goto err_free_reuse_maps;
875 NEXT_ARG();
877 fd = map_parse_fd(&argc, &argv);
878 if (fd < 0)
879 goto err_free_reuse_maps;
881 new_map_replace = reallocarray(map_replace,
882 old_map_fds + 1,
883 sizeof(*map_replace));
884 if (!new_map_replace) {
885 p_err("mem alloc failed");
886 goto err_free_reuse_maps;
888 map_replace = new_map_replace;
890 map_replace[old_map_fds].idx = idx;
891 map_replace[old_map_fds].name = name;
892 map_replace[old_map_fds].fd = fd;
893 old_map_fds++;
894 } else if (is_prefix(*argv, "dev")) {
895 NEXT_ARG();
897 if (ifindex) {
898 p_err("offload device already specified");
899 goto err_free_reuse_maps;
901 if (!REQ_ARGS(1))
902 goto err_free_reuse_maps;
904 ifindex = if_nametoindex(*argv);
905 if (!ifindex) {
906 p_err("unrecognized netdevice '%s': %s",
907 *argv, strerror(errno));
908 goto err_free_reuse_maps;
910 NEXT_ARG();
911 } else {
912 p_err("expected no more arguments, 'type', 'map' or 'dev', got: '%s'?",
913 *argv);
914 goto err_free_reuse_maps;
918 obj = __bpf_object__open_xattr(&attr, bpf_flags);
919 if (IS_ERR_OR_NULL(obj)) {
920 p_err("failed to open object file");
921 goto err_free_reuse_maps;
924 prog = bpf_program__next(NULL, obj);
925 if (!prog) {
926 p_err("object file doesn't contain any bpf program");
927 goto err_close_obj;
930 bpf_program__set_ifindex(prog, ifindex);
931 if (attr.prog_type == BPF_PROG_TYPE_UNSPEC) {
932 const char *sec_name = bpf_program__title(prog, false);
934 err = libbpf_prog_type_by_name(sec_name, &attr.prog_type,
935 &expected_attach_type);
936 if (err < 0) {
937 p_err("failed to guess program type based on section name %s\n",
938 sec_name);
939 goto err_close_obj;
942 bpf_program__set_type(prog, attr.prog_type);
943 bpf_program__set_expected_attach_type(prog, expected_attach_type);
945 qsort(map_replace, old_map_fds, sizeof(*map_replace),
946 map_replace_compar);
948 /* After the sort maps by name will be first on the list, because they
949 * have idx == -1. Resolve them.
951 j = 0;
952 while (j < old_map_fds && map_replace[j].name) {
953 i = 0;
954 bpf_map__for_each(map, obj) {
955 if (!strcmp(bpf_map__name(map), map_replace[j].name)) {
956 map_replace[j].idx = i;
957 break;
959 i++;
961 if (map_replace[j].idx == -1) {
962 p_err("unable to find map '%s'", map_replace[j].name);
963 goto err_close_obj;
965 j++;
967 /* Resort if any names were resolved */
968 if (j)
969 qsort(map_replace, old_map_fds, sizeof(*map_replace),
970 map_replace_compar);
972 /* Set ifindex and name reuse */
973 j = 0;
974 idx = 0;
975 bpf_map__for_each(map, obj) {
976 if (!bpf_map__is_offload_neutral(map))
977 bpf_map__set_ifindex(map, ifindex);
979 if (j < old_map_fds && idx == map_replace[j].idx) {
980 err = bpf_map__reuse_fd(map, map_replace[j++].fd);
981 if (err) {
982 p_err("unable to set up map reuse: %d", err);
983 goto err_close_obj;
986 /* Next reuse wants to apply to the same map */
987 if (j < old_map_fds && map_replace[j].idx == idx) {
988 p_err("replacement for map idx %d specified more than once",
989 idx);
990 goto err_close_obj;
994 idx++;
996 if (j < old_map_fds) {
997 p_err("map idx '%d' not used", map_replace[j].idx);
998 goto err_close_obj;
1001 err = bpf_object__load(obj);
1002 if (err) {
1003 p_err("failed to load object file");
1004 goto err_close_obj;
1007 if (do_pin_fd(bpf_program__fd(prog), pinfile))
1008 goto err_close_obj;
1010 if (json_output)
1011 jsonw_null(json_wtr);
1013 bpf_object__close(obj);
1014 for (i = 0; i < old_map_fds; i++)
1015 close(map_replace[i].fd);
1016 free(map_replace);
1018 return 0;
1020 err_close_obj:
1021 bpf_object__close(obj);
1022 err_free_reuse_maps:
1023 for (i = 0; i < old_map_fds; i++)
1024 close(map_replace[i].fd);
1025 free(map_replace);
1026 return -1;
1029 static int do_help(int argc, char **argv)
1031 if (json_output) {
1032 jsonw_null(json_wtr);
1033 return 0;
1036 fprintf(stderr,
1037 "Usage: %s %s { show | list } [PROG]\n"
1038 " %s %s dump xlated PROG [{ file FILE | opcodes | visual }]\n"
1039 " %s %s dump jited PROG [{ file FILE | opcodes }]\n"
1040 " %s %s pin PROG FILE\n"
1041 " %s %s load OBJ FILE [type TYPE] [dev NAME] \\\n"
1042 " [map { idx IDX | name NAME } MAP]\n"
1043 " %s %s attach PROG ATTACH_TYPE MAP\n"
1044 " %s %s detach PROG ATTACH_TYPE MAP\n"
1045 " %s %s help\n"
1046 "\n"
1047 " " HELP_SPEC_MAP "\n"
1048 " " HELP_SPEC_PROGRAM "\n"
1049 " TYPE := { socket | kprobe | kretprobe | classifier | action |\n"
1050 " tracepoint | raw_tracepoint | xdp | perf_event | cgroup/skb |\n"
1051 " cgroup/sock | cgroup/dev | lwt_in | lwt_out | lwt_xmit |\n"
1052 " lwt_seg6local | sockops | sk_skb | sk_msg | lirc_mode2 |\n"
1053 " cgroup/bind4 | cgroup/bind6 | cgroup/post_bind4 |\n"
1054 " cgroup/post_bind6 | cgroup/connect4 | cgroup/connect6 |\n"
1055 " cgroup/sendmsg4 | cgroup/sendmsg6 }\n"
1056 " ATTACH_TYPE := { msg_verdict | skb_verdict | skb_parse }\n"
1057 " " HELP_SPEC_OPTIONS "\n"
1059 bin_name, argv[-2], bin_name, argv[-2], bin_name, argv[-2],
1060 bin_name, argv[-2], bin_name, argv[-2], bin_name, argv[-2],
1061 bin_name, argv[-2], bin_name, argv[-2]);
1063 return 0;
1066 static const struct cmd cmds[] = {
1067 { "show", do_show },
1068 { "list", do_show },
1069 { "help", do_help },
1070 { "dump", do_dump },
1071 { "pin", do_pin },
1072 { "load", do_load },
1073 { "attach", do_attach },
1074 { "detach", do_detach },
1075 { 0 }
1078 int do_prog(int argc, char **argv)
1080 return cmd_select(cmds, argc, argv, do_help);