1 // SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause) 2 /* Copyright (C) 2017-2018 Netronome Systems, Inc. */ 3 4 #define _GNU_SOURCE 5 #include <errno.h> 6 #include <fcntl.h> 7 #include <signal.h> 8 #include <stdarg.h> 9 #include <stdio.h> 10 #include <stdlib.h> 11 #include <string.h> 12 #include <time.h> 13 #include <unistd.h> 14 #include <net/if.h> 15 #include <sys/ioctl.h> 16 #include <sys/types.h> 17 #include <sys/stat.h> 18 #include <sys/syscall.h> 19 #include <dirent.h> 20 21 #include <linux/err.h> 22 #include <linux/perf_event.h> 23 #include <linux/sizes.h> 24 25 #include <bpf/bpf.h> 26 #include <bpf/btf.h> 27 #include <bpf/hashmap.h> 28 #include <bpf/libbpf.h> 29 #include <bpf/skel_internal.h> 30 31 #include "cfg.h" 32 #include "main.h" 33 #include "xlated_dumper.h" 34 35 #define BPF_METADATA_PREFIX "bpf_metadata_" 36 #define BPF_METADATA_PREFIX_LEN (sizeof(BPF_METADATA_PREFIX) - 1) 37 38 const char * const prog_type_name[] = { 39 [BPF_PROG_TYPE_UNSPEC] = "unspec", 40 [BPF_PROG_TYPE_SOCKET_FILTER] = "socket_filter", 41 [BPF_PROG_TYPE_KPROBE] = "kprobe", 42 [BPF_PROG_TYPE_SCHED_CLS] = "sched_cls", 43 [BPF_PROG_TYPE_SCHED_ACT] = "sched_act", 44 [BPF_PROG_TYPE_TRACEPOINT] = "tracepoint", 45 [BPF_PROG_TYPE_XDP] = "xdp", 46 [BPF_PROG_TYPE_PERF_EVENT] = "perf_event", 47 [BPF_PROG_TYPE_CGROUP_SKB] = "cgroup_skb", 48 [BPF_PROG_TYPE_CGROUP_SOCK] = "cgroup_sock", 49 [BPF_PROG_TYPE_LWT_IN] = "lwt_in", 50 [BPF_PROG_TYPE_LWT_OUT] = "lwt_out", 51 [BPF_PROG_TYPE_LWT_XMIT] = "lwt_xmit", 52 [BPF_PROG_TYPE_SOCK_OPS] = "sock_ops", 53 [BPF_PROG_TYPE_SK_SKB] = "sk_skb", 54 [BPF_PROG_TYPE_CGROUP_DEVICE] = "cgroup_device", 55 [BPF_PROG_TYPE_SK_MSG] = "sk_msg", 56 [BPF_PROG_TYPE_RAW_TRACEPOINT] = "raw_tracepoint", 57 [BPF_PROG_TYPE_CGROUP_SOCK_ADDR] = "cgroup_sock_addr", 58 [BPF_PROG_TYPE_LWT_SEG6LOCAL] = "lwt_seg6local", 59 [BPF_PROG_TYPE_LIRC_MODE2] = "lirc_mode2", 60 [BPF_PROG_TYPE_SK_REUSEPORT] = "sk_reuseport", 61 [BPF_PROG_TYPE_FLOW_DISSECTOR] = "flow_dissector", 62 [BPF_PROG_TYPE_CGROUP_SYSCTL] = "cgroup_sysctl", 63 [BPF_PROG_TYPE_RAW_TRACEPOINT_WRITABLE] = "raw_tracepoint_writable", 64 [BPF_PROG_TYPE_CGROUP_SOCKOPT] = "cgroup_sockopt", 65 [BPF_PROG_TYPE_TRACING] = "tracing", 66 [BPF_PROG_TYPE_STRUCT_OPS] = "struct_ops", 67 [BPF_PROG_TYPE_EXT] = "ext", 68 [BPF_PROG_TYPE_LSM] = "lsm", 69 [BPF_PROG_TYPE_SK_LOOKUP] = "sk_lookup", 70 }; 71 72 const size_t prog_type_name_size = ARRAY_SIZE(prog_type_name); 73 74 enum dump_mode { 75 DUMP_JITED, 76 DUMP_XLATED, 77 }; 78 79 static const char * const attach_type_strings[] = { 80 [BPF_SK_SKB_STREAM_PARSER] = "stream_parser", 81 [BPF_SK_SKB_STREAM_VERDICT] = "stream_verdict", 82 [BPF_SK_SKB_VERDICT] = "skb_verdict", 83 [BPF_SK_MSG_VERDICT] = "msg_verdict", 84 [BPF_FLOW_DISSECTOR] = "flow_dissector", 85 [__MAX_BPF_ATTACH_TYPE] = NULL, 86 }; 87 88 static struct hashmap *prog_table; 89 90 static enum bpf_attach_type parse_attach_type(const char *str) 91 { 92 enum bpf_attach_type type; 93 94 for (type = 0; type < __MAX_BPF_ATTACH_TYPE; type++) { 95 if (attach_type_strings[type] && 96 is_prefix(str, attach_type_strings[type])) 97 return type; 98 } 99 100 return __MAX_BPF_ATTACH_TYPE; 101 } 102 103 static int prep_prog_info(struct bpf_prog_info *const info, enum dump_mode mode, 104 void **info_data, size_t *const info_data_sz) 105 { 106 struct bpf_prog_info holder = {}; 107 size_t needed = 0; 108 void *ptr; 109 110 if (mode == DUMP_JITED) { 111 holder.jited_prog_len = info->jited_prog_len; 112 needed += info->jited_prog_len; 113 } else { 114 holder.xlated_prog_len = info->xlated_prog_len; 115 needed += info->xlated_prog_len; 116 } 117 118 holder.nr_jited_ksyms = info->nr_jited_ksyms; 119 needed += info->nr_jited_ksyms * sizeof(__u64); 120 121 holder.nr_jited_func_lens = info->nr_jited_func_lens; 122 needed += info->nr_jited_func_lens * sizeof(__u32); 123 124 holder.nr_func_info = info->nr_func_info; 125 holder.func_info_rec_size = info->func_info_rec_size; 126 needed += info->nr_func_info * info->func_info_rec_size; 127 128 holder.nr_line_info = info->nr_line_info; 129 holder.line_info_rec_size = info->line_info_rec_size; 130 needed += info->nr_line_info * info->line_info_rec_size; 131 132 holder.nr_jited_line_info = info->nr_jited_line_info; 133 holder.jited_line_info_rec_size = info->jited_line_info_rec_size; 134 needed += info->nr_jited_line_info * info->jited_line_info_rec_size; 135 136 if (needed > *info_data_sz) { 137 ptr = realloc(*info_data, needed); 138 if (!ptr) 139 return -1; 140 141 *info_data = ptr; 142 *info_data_sz = needed; 143 } 144 ptr = *info_data; 145 146 if (mode == DUMP_JITED) { 147 holder.jited_prog_insns = ptr_to_u64(ptr); 148 ptr += holder.jited_prog_len; 149 } else { 150 holder.xlated_prog_insns = ptr_to_u64(ptr); 151 ptr += holder.xlated_prog_len; 152 } 153 154 holder.jited_ksyms = ptr_to_u64(ptr); 155 ptr += holder.nr_jited_ksyms * sizeof(__u64); 156 157 holder.jited_func_lens = ptr_to_u64(ptr); 158 ptr += holder.nr_jited_func_lens * sizeof(__u32); 159 160 holder.func_info = ptr_to_u64(ptr); 161 ptr += holder.nr_func_info * holder.func_info_rec_size; 162 163 holder.line_info = ptr_to_u64(ptr); 164 ptr += holder.nr_line_info * holder.line_info_rec_size; 165 166 holder.jited_line_info = ptr_to_u64(ptr); 167 ptr += holder.nr_jited_line_info * holder.jited_line_info_rec_size; 168 169 *info = holder; 170 return 0; 171 } 172 173 static void print_boot_time(__u64 nsecs, char *buf, unsigned int size) 174 { 175 struct timespec real_time_ts, boot_time_ts; 176 time_t wallclock_secs; 177 struct tm load_tm; 178 179 buf[--size] = '\0'; 180 181 if (clock_gettime(CLOCK_REALTIME, &real_time_ts) || 182 clock_gettime(CLOCK_BOOTTIME, &boot_time_ts)) { 183 perror("Can't read clocks"); 184 snprintf(buf, size, "%llu", nsecs / 1000000000); 185 return; 186 } 187 188 wallclock_secs = (real_time_ts.tv_sec - boot_time_ts.tv_sec) + 189 (real_time_ts.tv_nsec - boot_time_ts.tv_nsec + nsecs) / 190 1000000000; 191 192 193 if (!localtime_r(&wallclock_secs, &load_tm)) { 194 snprintf(buf, size, "%llu", nsecs / 1000000000); 195 return; 196 } 197 198 if (json_output) 199 strftime(buf, size, "%s", &load_tm); 200 else 201 strftime(buf, size, "%FT%T%z", &load_tm); 202 } 203 204 static void show_prog_maps(int fd, __u32 num_maps) 205 { 206 struct bpf_prog_info info = {}; 207 __u32 len = sizeof(info); 208 __u32 map_ids[num_maps]; 209 unsigned int i; 210 int err; 211 212 info.nr_map_ids = num_maps; 213 info.map_ids = ptr_to_u64(map_ids); 214 215 err = bpf_obj_get_info_by_fd(fd, &info, &len); 216 if (err || !info.nr_map_ids) 217 return; 218 219 if (json_output) { 220 jsonw_name(json_wtr, "map_ids"); 221 jsonw_start_array(json_wtr); 222 for (i = 0; i < info.nr_map_ids; i++) 223 jsonw_uint(json_wtr, map_ids[i]); 224 jsonw_end_array(json_wtr); 225 } else { 226 printf(" map_ids "); 227 for (i = 0; i < info.nr_map_ids; i++) 228 printf("%u%s", map_ids[i], 229 i == info.nr_map_ids - 1 ? "" : ","); 230 } 231 } 232 233 static void *find_metadata(int prog_fd, struct bpf_map_info *map_info) 234 { 235 struct bpf_prog_info prog_info; 236 __u32 prog_info_len; 237 __u32 map_info_len; 238 void *value = NULL; 239 __u32 *map_ids; 240 int nr_maps; 241 int key = 0; 242 int map_fd; 243 int ret; 244 __u32 i; 245 246 memset(&prog_info, 0, sizeof(prog_info)); 247 prog_info_len = sizeof(prog_info); 248 ret = bpf_obj_get_info_by_fd(prog_fd, &prog_info, &prog_info_len); 249 if (ret) 250 return NULL; 251 252 if (!prog_info.nr_map_ids) 253 return NULL; 254 255 map_ids = calloc(prog_info.nr_map_ids, sizeof(__u32)); 256 if (!map_ids) 257 return NULL; 258 259 nr_maps = prog_info.nr_map_ids; 260 memset(&prog_info, 0, sizeof(prog_info)); 261 prog_info.nr_map_ids = nr_maps; 262 prog_info.map_ids = ptr_to_u64(map_ids); 263 prog_info_len = sizeof(prog_info); 264 265 ret = bpf_obj_get_info_by_fd(prog_fd, &prog_info, &prog_info_len); 266 if (ret) 267 goto free_map_ids; 268 269 for (i = 0; i < prog_info.nr_map_ids; i++) { 270 map_fd = bpf_map_get_fd_by_id(map_ids[i]); 271 if (map_fd < 0) 272 goto free_map_ids; 273 274 memset(map_info, 0, sizeof(*map_info)); 275 map_info_len = sizeof(*map_info); 276 ret = bpf_obj_get_info_by_fd(map_fd, map_info, &map_info_len); 277 if (ret < 0) { 278 close(map_fd); 279 goto free_map_ids; 280 } 281 282 if (map_info->type != BPF_MAP_TYPE_ARRAY || 283 map_info->key_size != sizeof(int) || 284 map_info->max_entries != 1 || 285 !map_info->btf_value_type_id || 286 !strstr(map_info->name, ".rodata")) { 287 close(map_fd); 288 continue; 289 } 290 291 value = malloc(map_info->value_size); 292 if (!value) { 293 close(map_fd); 294 goto free_map_ids; 295 } 296 297 if (bpf_map_lookup_elem(map_fd, &key, value)) { 298 close(map_fd); 299 free(value); 300 value = NULL; 301 goto free_map_ids; 302 } 303 304 close(map_fd); 305 break; 306 } 307 308 free_map_ids: 309 free(map_ids); 310 return value; 311 } 312 313 static bool has_metadata_prefix(const char *s) 314 { 315 return strncmp(s, BPF_METADATA_PREFIX, BPF_METADATA_PREFIX_LEN) == 0; 316 } 317 318 static void show_prog_metadata(int fd, __u32 num_maps) 319 { 320 const struct btf_type *t_datasec, *t_var; 321 struct bpf_map_info map_info; 322 struct btf_var_secinfo *vsi; 323 bool printed_header = false; 324 unsigned int i, vlen; 325 void *value = NULL; 326 const char *name; 327 struct btf *btf; 328 int err; 329 330 if (!num_maps) 331 return; 332 333 memset(&map_info, 0, sizeof(map_info)); 334 value = find_metadata(fd, &map_info); 335 if (!value) 336 return; 337 338 btf = btf__load_from_kernel_by_id(map_info.btf_id); 339 if (libbpf_get_error(btf)) 340 goto out_free; 341 342 t_datasec = btf__type_by_id(btf, map_info.btf_value_type_id); 343 if (!btf_is_datasec(t_datasec)) 344 goto out_free; 345 346 vlen = btf_vlen(t_datasec); 347 vsi = btf_var_secinfos(t_datasec); 348 349 /* We don't proceed to check the kinds of the elements of the DATASEC. 350 * The verifier enforces them to be BTF_KIND_VAR. 351 */ 352 353 if (json_output) { 354 struct btf_dumper d = { 355 .btf = btf, 356 .jw = json_wtr, 357 .is_plain_text = false, 358 }; 359 360 for (i = 0; i < vlen; i++, vsi++) { 361 t_var = btf__type_by_id(btf, vsi->type); 362 name = btf__name_by_offset(btf, t_var->name_off); 363 364 if (!has_metadata_prefix(name)) 365 continue; 366 367 if (!printed_header) { 368 jsonw_name(json_wtr, "metadata"); 369 jsonw_start_object(json_wtr); 370 printed_header = true; 371 } 372 373 jsonw_name(json_wtr, name + BPF_METADATA_PREFIX_LEN); 374 err = btf_dumper_type(&d, t_var->type, value + vsi->offset); 375 if (err) { 376 p_err("btf dump failed: %d", err); 377 break; 378 } 379 } 380 if (printed_header) 381 jsonw_end_object(json_wtr); 382 } else { 383 json_writer_t *btf_wtr; 384 struct btf_dumper d = { 385 .btf = btf, 386 .is_plain_text = true, 387 }; 388 389 for (i = 0; i < vlen; i++, vsi++) { 390 t_var = btf__type_by_id(btf, vsi->type); 391 name = btf__name_by_offset(btf, t_var->name_off); 392 393 if (!has_metadata_prefix(name)) 394 continue; 395 396 if (!printed_header) { 397 printf("\tmetadata:"); 398 399 btf_wtr = jsonw_new(stdout); 400 if (!btf_wtr) { 401 p_err("jsonw alloc failed"); 402 goto out_free; 403 } 404 d.jw = btf_wtr, 405 406 printed_header = true; 407 } 408 409 printf("\n\t\t%s = ", name + BPF_METADATA_PREFIX_LEN); 410 411 jsonw_reset(btf_wtr); 412 err = btf_dumper_type(&d, t_var->type, value + vsi->offset); 413 if (err) { 414 p_err("btf dump failed: %d", err); 415 break; 416 } 417 } 418 if (printed_header) 419 jsonw_destroy(&btf_wtr); 420 } 421 422 out_free: 423 btf__free(btf); 424 free(value); 425 } 426 427 static void print_prog_header_json(struct bpf_prog_info *info) 428 { 429 jsonw_uint_field(json_wtr, "id", info->id); 430 if (info->type < ARRAY_SIZE(prog_type_name)) 431 jsonw_string_field(json_wtr, "type", 432 prog_type_name[info->type]); 433 else 434 jsonw_uint_field(json_wtr, "type", info->type); 435 436 if (*info->name) 437 jsonw_string_field(json_wtr, "name", info->name); 438 439 jsonw_name(json_wtr, "tag"); 440 jsonw_printf(json_wtr, "\"" BPF_TAG_FMT "\"", 441 info->tag[0], info->tag[1], info->tag[2], info->tag[3], 442 info->tag[4], info->tag[5], info->tag[6], info->tag[7]); 443 444 jsonw_bool_field(json_wtr, "gpl_compatible", info->gpl_compatible); 445 if (info->run_time_ns) { 446 jsonw_uint_field(json_wtr, "run_time_ns", info->run_time_ns); 447 jsonw_uint_field(json_wtr, "run_cnt", info->run_cnt); 448 } 449 if (info->recursion_misses) 450 jsonw_uint_field(json_wtr, "recursion_misses", info->recursion_misses); 451 } 452 453 static void print_prog_json(struct bpf_prog_info *info, int fd) 454 { 455 char *memlock; 456 457 jsonw_start_object(json_wtr); 458 print_prog_header_json(info); 459 print_dev_json(info->ifindex, info->netns_dev, info->netns_ino); 460 461 if (info->load_time) { 462 char buf[32]; 463 464 print_boot_time(info->load_time, buf, sizeof(buf)); 465 466 /* Piggy back on load_time, since 0 uid is a valid one */ 467 jsonw_name(json_wtr, "loaded_at"); 468 jsonw_printf(json_wtr, "%s", buf); 469 jsonw_uint_field(json_wtr, "uid", info->created_by_uid); 470 } 471 472 jsonw_uint_field(json_wtr, "bytes_xlated", info->xlated_prog_len); 473 474 if (info->jited_prog_len) { 475 jsonw_bool_field(json_wtr, "jited", true); 476 jsonw_uint_field(json_wtr, "bytes_jited", info->jited_prog_len); 477 } else { 478 jsonw_bool_field(json_wtr, "jited", false); 479 } 480 481 memlock = get_fdinfo(fd, "memlock"); 482 if (memlock) 483 jsonw_int_field(json_wtr, "bytes_memlock", atoi(memlock)); 484 free(memlock); 485 486 if (info->nr_map_ids) 487 show_prog_maps(fd, info->nr_map_ids); 488 489 if (info->btf_id) 490 jsonw_int_field(json_wtr, "btf_id", info->btf_id); 491 492 if (!hashmap__empty(prog_table)) { 493 struct hashmap_entry *entry; 494 495 jsonw_name(json_wtr, "pinned"); 496 jsonw_start_array(json_wtr); 497 hashmap__for_each_key_entry(prog_table, entry, 498 u32_as_hash_field(info->id)) 499 jsonw_string(json_wtr, entry->value); 500 jsonw_end_array(json_wtr); 501 } 502 503 emit_obj_refs_json(refs_table, info->id, json_wtr); 504 505 show_prog_metadata(fd, info->nr_map_ids); 506 507 jsonw_end_object(json_wtr); 508 } 509 510 static void print_prog_header_plain(struct bpf_prog_info *info) 511 { 512 printf("%u: ", info->id); 513 if (info->type < ARRAY_SIZE(prog_type_name)) 514 printf("%s ", prog_type_name[info->type]); 515 else 516 printf("type %u ", info->type); 517 518 if (*info->name) 519 printf("name %s ", info->name); 520 521 printf("tag "); 522 fprint_hex(stdout, info->tag, BPF_TAG_SIZE, ""); 523 print_dev_plain(info->ifindex, info->netns_dev, info->netns_ino); 524 printf("%s", info->gpl_compatible ? " gpl" : ""); 525 if (info->run_time_ns) 526 printf(" run_time_ns %lld run_cnt %lld", 527 info->run_time_ns, info->run_cnt); 528 if (info->recursion_misses) 529 printf(" recursion_misses %lld", info->recursion_misses); 530 printf("\n"); 531 } 532 533 static void print_prog_plain(struct bpf_prog_info *info, int fd) 534 { 535 char *memlock; 536 537 print_prog_header_plain(info); 538 539 if (info->load_time) { 540 char buf[32]; 541 542 print_boot_time(info->load_time, buf, sizeof(buf)); 543 544 /* Piggy back on load_time, since 0 uid is a valid one */ 545 printf("\tloaded_at %s uid %u\n", buf, info->created_by_uid); 546 } 547 548 printf("\txlated %uB", info->xlated_prog_len); 549 550 if (info->jited_prog_len) 551 printf(" jited %uB", info->jited_prog_len); 552 else 553 printf(" not jited"); 554 555 memlock = get_fdinfo(fd, "memlock"); 556 if (memlock) 557 printf(" memlock %sB", memlock); 558 free(memlock); 559 560 if (info->nr_map_ids) 561 show_prog_maps(fd, info->nr_map_ids); 562 563 if (!hashmap__empty(prog_table)) { 564 struct hashmap_entry *entry; 565 566 hashmap__for_each_key_entry(prog_table, entry, 567 u32_as_hash_field(info->id)) 568 printf("\n\tpinned %s", (char *)entry->value); 569 } 570 571 if (info->btf_id) 572 printf("\n\tbtf_id %d", info->btf_id); 573 574 emit_obj_refs_plain(refs_table, info->id, "\n\tpids "); 575 576 printf("\n"); 577 578 show_prog_metadata(fd, info->nr_map_ids); 579 } 580 581 static int show_prog(int fd) 582 { 583 struct bpf_prog_info info = {}; 584 __u32 len = sizeof(info); 585 int err; 586 587 err = bpf_obj_get_info_by_fd(fd, &info, &len); 588 if (err) { 589 p_err("can't get prog info: %s", strerror(errno)); 590 return -1; 591 } 592 593 if (json_output) 594 print_prog_json(&info, fd); 595 else 596 print_prog_plain(&info, fd); 597 598 return 0; 599 } 600 601 static int do_show_subset(int argc, char **argv) 602 { 603 int *fds = NULL; 604 int nb_fds, i; 605 int err = -1; 606 607 fds = malloc(sizeof(int)); 608 if (!fds) { 609 p_err("mem alloc failed"); 610 return -1; 611 } 612 nb_fds = prog_parse_fds(&argc, &argv, &fds); 613 if (nb_fds < 1) 614 goto exit_free; 615 616 if (json_output && nb_fds > 1) 617 jsonw_start_array(json_wtr); /* root array */ 618 for (i = 0; i < nb_fds; i++) { 619 err = show_prog(fds[i]); 620 if (err) { 621 for (; i < nb_fds; i++) 622 close(fds[i]); 623 break; 624 } 625 close(fds[i]); 626 } 627 if (json_output && nb_fds > 1) 628 jsonw_end_array(json_wtr); /* root array */ 629 630 exit_free: 631 free(fds); 632 return err; 633 } 634 635 static int do_show(int argc, char **argv) 636 { 637 __u32 id = 0; 638 int err; 639 int fd; 640 641 if (show_pinned) { 642 prog_table = hashmap__new(hash_fn_for_key_as_id, 643 equal_fn_for_key_as_id, NULL); 644 if (!prog_table) { 645 p_err("failed to create hashmap for pinned paths"); 646 return -1; 647 } 648 build_pinned_obj_table(prog_table, BPF_OBJ_PROG); 649 } 650 build_obj_refs_table(&refs_table, BPF_OBJ_PROG); 651 652 if (argc == 2) 653 return do_show_subset(argc, argv); 654 655 if (argc) 656 return BAD_ARG(); 657 658 if (json_output) 659 jsonw_start_array(json_wtr); 660 while (true) { 661 err = bpf_prog_get_next_id(id, &id); 662 if (err) { 663 if (errno == ENOENT) { 664 err = 0; 665 break; 666 } 667 p_err("can't get next program: %s%s", strerror(errno), 668 errno == EINVAL ? " -- kernel too old?" : ""); 669 err = -1; 670 break; 671 } 672 673 fd = bpf_prog_get_fd_by_id(id); 674 if (fd < 0) { 675 if (errno == ENOENT) 676 continue; 677 p_err("can't get prog by id (%u): %s", 678 id, strerror(errno)); 679 err = -1; 680 break; 681 } 682 683 err = show_prog(fd); 684 close(fd); 685 if (err) 686 break; 687 } 688 689 if (json_output) 690 jsonw_end_array(json_wtr); 691 692 delete_obj_refs_table(refs_table); 693 694 if (show_pinned) 695 delete_pinned_obj_table(prog_table); 696 697 return err; 698 } 699 700 static int 701 prog_dump(struct bpf_prog_info *info, enum dump_mode mode, 702 char *filepath, bool opcodes, bool visual, bool linum) 703 { 704 struct bpf_prog_linfo *prog_linfo = NULL; 705 const char *disasm_opt = NULL; 706 struct dump_data dd = {}; 707 void *func_info = NULL; 708 struct btf *btf = NULL; 709 char func_sig[1024]; 710 unsigned char *buf; 711 __u32 member_len; 712 ssize_t n; 713 int fd; 714 715 if (mode == DUMP_JITED) { 716 if (info->jited_prog_len == 0 || !info->jited_prog_insns) { 717 p_info("no instructions returned"); 718 return -1; 719 } 720 buf = u64_to_ptr(info->jited_prog_insns); 721 member_len = info->jited_prog_len; 722 } else { /* DUMP_XLATED */ 723 if (info->xlated_prog_len == 0 || !info->xlated_prog_insns) { 724 p_err("error retrieving insn dump: kernel.kptr_restrict set?"); 725 return -1; 726 } 727 buf = u64_to_ptr(info->xlated_prog_insns); 728 member_len = info->xlated_prog_len; 729 } 730 731 if (info->btf_id) { 732 btf = btf__load_from_kernel_by_id(info->btf_id); 733 if (libbpf_get_error(btf)) { 734 p_err("failed to get btf"); 735 return -1; 736 } 737 } 738 739 func_info = u64_to_ptr(info->func_info); 740 741 if (info->nr_line_info) { 742 prog_linfo = bpf_prog_linfo__new(info); 743 if (!prog_linfo) 744 p_info("error in processing bpf_line_info. continue without it."); 745 } 746 747 if (filepath) { 748 fd = open(filepath, O_WRONLY | O_CREAT | O_TRUNC, 0600); 749 if (fd < 0) { 750 p_err("can't open file %s: %s", filepath, 751 strerror(errno)); 752 return -1; 753 } 754 755 n = write(fd, buf, member_len); 756 close(fd); 757 if (n != (ssize_t)member_len) { 758 p_err("error writing output file: %s", 759 n < 0 ? strerror(errno) : "short write"); 760 return -1; 761 } 762 763 if (json_output) 764 jsonw_null(json_wtr); 765 } else if (mode == DUMP_JITED) { 766 const char *name = NULL; 767 768 if (info->ifindex) { 769 name = ifindex_to_bfd_params(info->ifindex, 770 info->netns_dev, 771 info->netns_ino, 772 &disasm_opt); 773 if (!name) 774 return -1; 775 } 776 777 if (info->nr_jited_func_lens && info->jited_func_lens) { 778 struct kernel_sym *sym = NULL; 779 struct bpf_func_info *record; 780 char sym_name[SYM_MAX_NAME]; 781 unsigned char *img = buf; 782 __u64 *ksyms = NULL; 783 __u32 *lens; 784 __u32 i; 785 if (info->nr_jited_ksyms) { 786 kernel_syms_load(&dd); 787 ksyms = u64_to_ptr(info->jited_ksyms); 788 } 789 790 if (json_output) 791 jsonw_start_array(json_wtr); 792 793 lens = u64_to_ptr(info->jited_func_lens); 794 for (i = 0; i < info->nr_jited_func_lens; i++) { 795 if (ksyms) { 796 sym = kernel_syms_search(&dd, ksyms[i]); 797 if (sym) 798 sprintf(sym_name, "%s", sym->name); 799 else 800 sprintf(sym_name, "0x%016llx", ksyms[i]); 801 } else { 802 strcpy(sym_name, "unknown"); 803 } 804 805 if (func_info) { 806 record = func_info + i * info->func_info_rec_size; 807 btf_dumper_type_only(btf, record->type_id, 808 func_sig, 809 sizeof(func_sig)); 810 } 811 812 if (json_output) { 813 jsonw_start_object(json_wtr); 814 if (func_info && func_sig[0] != '\0') { 815 jsonw_name(json_wtr, "proto"); 816 jsonw_string(json_wtr, func_sig); 817 } 818 jsonw_name(json_wtr, "name"); 819 jsonw_string(json_wtr, sym_name); 820 jsonw_name(json_wtr, "insns"); 821 } else { 822 if (func_info && func_sig[0] != '\0') 823 printf("%s:\n", func_sig); 824 printf("%s:\n", sym_name); 825 } 826 827 disasm_print_insn(img, lens[i], opcodes, 828 name, disasm_opt, btf, 829 prog_linfo, ksyms[i], i, 830 linum); 831 832 img += lens[i]; 833 834 if (json_output) 835 jsonw_end_object(json_wtr); 836 else 837 printf("\n"); 838 } 839 840 if (json_output) 841 jsonw_end_array(json_wtr); 842 } else { 843 disasm_print_insn(buf, member_len, opcodes, name, 844 disasm_opt, btf, NULL, 0, 0, false); 845 } 846 } else if (visual) { 847 if (json_output) 848 jsonw_null(json_wtr); 849 else 850 dump_xlated_cfg(buf, member_len); 851 } else { 852 kernel_syms_load(&dd); 853 dd.nr_jited_ksyms = info->nr_jited_ksyms; 854 dd.jited_ksyms = u64_to_ptr(info->jited_ksyms); 855 dd.btf = btf; 856 dd.func_info = func_info; 857 dd.finfo_rec_size = info->func_info_rec_size; 858 dd.prog_linfo = prog_linfo; 859 860 if (json_output) 861 dump_xlated_json(&dd, buf, member_len, opcodes, 862 linum); 863 else 864 dump_xlated_plain(&dd, buf, member_len, opcodes, 865 linum); 866 kernel_syms_destroy(&dd); 867 } 868 869 btf__free(btf); 870 871 return 0; 872 } 873 874 static int do_dump(int argc, char **argv) 875 { 876 struct bpf_prog_info info; 877 __u32 info_len = sizeof(info); 878 size_t info_data_sz = 0; 879 void *info_data = NULL; 880 char *filepath = NULL; 881 bool opcodes = false; 882 bool visual = false; 883 enum dump_mode mode; 884 bool linum = false; 885 int nb_fds, i = 0; 886 int *fds = NULL; 887 int err = -1; 888 889 if (is_prefix(*argv, "jited")) { 890 if (disasm_init()) 891 return -1; 892 mode = DUMP_JITED; 893 } else if (is_prefix(*argv, "xlated")) { 894 mode = DUMP_XLATED; 895 } else { 896 p_err("expected 'xlated' or 'jited', got: %s", *argv); 897 return -1; 898 } 899 NEXT_ARG(); 900 901 if (argc < 2) 902 usage(); 903 904 fds = malloc(sizeof(int)); 905 if (!fds) { 906 p_err("mem alloc failed"); 907 return -1; 908 } 909 nb_fds = prog_parse_fds(&argc, &argv, &fds); 910 if (nb_fds < 1) 911 goto exit_free; 912 913 if (is_prefix(*argv, "file")) { 914 NEXT_ARG(); 915 if (!argc) { 916 p_err("expected file path"); 917 goto exit_close; 918 } 919 if (nb_fds > 1) { 920 p_err("several programs matched"); 921 goto exit_close; 922 } 923 924 filepath = *argv; 925 NEXT_ARG(); 926 } else if (is_prefix(*argv, "opcodes")) { 927 opcodes = true; 928 NEXT_ARG(); 929 } else if (is_prefix(*argv, "visual")) { 930 if (nb_fds > 1) { 931 p_err("several programs matched"); 932 goto exit_close; 933 } 934 935 visual = true; 936 NEXT_ARG(); 937 } else if (is_prefix(*argv, "linum")) { 938 linum = true; 939 NEXT_ARG(); 940 } 941 942 if (argc) { 943 usage(); 944 goto exit_close; 945 } 946 947 if (json_output && nb_fds > 1) 948 jsonw_start_array(json_wtr); /* root array */ 949 for (i = 0; i < nb_fds; i++) { 950 memset(&info, 0, sizeof(info)); 951 952 err = bpf_obj_get_info_by_fd(fds[i], &info, &info_len); 953 if (err) { 954 p_err("can't get prog info: %s", strerror(errno)); 955 break; 956 } 957 958 err = prep_prog_info(&info, mode, &info_data, &info_data_sz); 959 if (err) { 960 p_err("can't grow prog info_data"); 961 break; 962 } 963 964 err = bpf_obj_get_info_by_fd(fds[i], &info, &info_len); 965 if (err) { 966 p_err("can't get prog info: %s", strerror(errno)); 967 break; 968 } 969 970 if (json_output && nb_fds > 1) { 971 jsonw_start_object(json_wtr); /* prog object */ 972 print_prog_header_json(&info); 973 jsonw_name(json_wtr, "insns"); 974 } else if (nb_fds > 1) { 975 print_prog_header_plain(&info); 976 } 977 978 err = prog_dump(&info, mode, filepath, opcodes, visual, linum); 979 980 if (json_output && nb_fds > 1) 981 jsonw_end_object(json_wtr); /* prog object */ 982 else if (i != nb_fds - 1 && nb_fds > 1) 983 printf("\n"); 984 985 if (err) 986 break; 987 close(fds[i]); 988 } 989 if (json_output && nb_fds > 1) 990 jsonw_end_array(json_wtr); /* root array */ 991 992 exit_close: 993 for (; i < nb_fds; i++) 994 close(fds[i]); 995 exit_free: 996 free(info_data); 997 free(fds); 998 return err; 999 } 1000 1001 static int do_pin(int argc, char **argv) 1002 { 1003 int err; 1004 1005 err = do_pin_any(argc, argv, prog_parse_fd); 1006 if (!err && json_output) 1007 jsonw_null(json_wtr); 1008 return err; 1009 } 1010 1011 struct map_replace { 1012 int idx; 1013 int fd; 1014 char *name; 1015 }; 1016 1017 static int map_replace_compar(const void *p1, const void *p2) 1018 { 1019 const struct map_replace *a = p1, *b = p2; 1020 1021 return a->idx - b->idx; 1022 } 1023 1024 static int parse_attach_detach_args(int argc, char **argv, int *progfd, 1025 enum bpf_attach_type *attach_type, 1026 int *mapfd) 1027 { 1028 if (!REQ_ARGS(3)) 1029 return -EINVAL; 1030 1031 *progfd = prog_parse_fd(&argc, &argv); 1032 if (*progfd < 0) 1033 return *progfd; 1034 1035 *attach_type = parse_attach_type(*argv); 1036 if (*attach_type == __MAX_BPF_ATTACH_TYPE) { 1037 p_err("invalid attach/detach type"); 1038 return -EINVAL; 1039 } 1040 1041 if (*attach_type == BPF_FLOW_DISSECTOR) { 1042 *mapfd = 0; 1043 return 0; 1044 } 1045 1046 NEXT_ARG(); 1047 if (!REQ_ARGS(2)) 1048 return -EINVAL; 1049 1050 *mapfd = map_parse_fd(&argc, &argv); 1051 if (*mapfd < 0) 1052 return *mapfd; 1053 1054 return 0; 1055 } 1056 1057 static int do_attach(int argc, char **argv) 1058 { 1059 enum bpf_attach_type attach_type; 1060 int err, progfd; 1061 int mapfd; 1062 1063 err = parse_attach_detach_args(argc, argv, 1064 &progfd, &attach_type, &mapfd); 1065 if (err) 1066 return err; 1067 1068 err = bpf_prog_attach(progfd, mapfd, attach_type, 0); 1069 if (err) { 1070 p_err("failed prog attach to map"); 1071 return -EINVAL; 1072 } 1073 1074 if (json_output) 1075 jsonw_null(json_wtr); 1076 return 0; 1077 } 1078 1079 static int do_detach(int argc, char **argv) 1080 { 1081 enum bpf_attach_type attach_type; 1082 int err, progfd; 1083 int mapfd; 1084 1085 err = parse_attach_detach_args(argc, argv, 1086 &progfd, &attach_type, &mapfd); 1087 if (err) 1088 return err; 1089 1090 err = bpf_prog_detach2(progfd, mapfd, attach_type); 1091 if (err) { 1092 p_err("failed prog detach from map"); 1093 return -EINVAL; 1094 } 1095 1096 if (json_output) 1097 jsonw_null(json_wtr); 1098 return 0; 1099 } 1100 1101 static int check_single_stdin(char *file_data_in, char *file_ctx_in) 1102 { 1103 if (file_data_in && file_ctx_in && 1104 !strcmp(file_data_in, "-") && !strcmp(file_ctx_in, "-")) { 1105 p_err("cannot use standard input for both data_in and ctx_in"); 1106 return -1; 1107 } 1108 1109 return 0; 1110 } 1111 1112 static int get_run_data(const char *fname, void **data_ptr, unsigned int *size) 1113 { 1114 size_t block_size = 256; 1115 size_t buf_size = block_size; 1116 size_t nb_read = 0; 1117 void *tmp; 1118 FILE *f; 1119 1120 if (!fname) { 1121 *data_ptr = NULL; 1122 *size = 0; 1123 return 0; 1124 } 1125 1126 if (!strcmp(fname, "-")) 1127 f = stdin; 1128 else 1129 f = fopen(fname, "r"); 1130 if (!f) { 1131 p_err("failed to open %s: %s", fname, strerror(errno)); 1132 return -1; 1133 } 1134 1135 *data_ptr = malloc(block_size); 1136 if (!*data_ptr) { 1137 p_err("failed to allocate memory for data_in/ctx_in: %s", 1138 strerror(errno)); 1139 goto err_fclose; 1140 } 1141 1142 while ((nb_read += fread(*data_ptr + nb_read, 1, block_size, f))) { 1143 if (feof(f)) 1144 break; 1145 if (ferror(f)) { 1146 p_err("failed to read data_in/ctx_in from %s: %s", 1147 fname, strerror(errno)); 1148 goto err_free; 1149 } 1150 if (nb_read > buf_size - block_size) { 1151 if (buf_size == UINT32_MAX) { 1152 p_err("data_in/ctx_in is too long (max: %d)", 1153 UINT32_MAX); 1154 goto err_free; 1155 } 1156 /* No space for fread()-ing next chunk; realloc() */ 1157 buf_size *= 2; 1158 tmp = realloc(*data_ptr, buf_size); 1159 if (!tmp) { 1160 p_err("failed to reallocate data_in/ctx_in: %s", 1161 strerror(errno)); 1162 goto err_free; 1163 } 1164 *data_ptr = tmp; 1165 } 1166 } 1167 if (f != stdin) 1168 fclose(f); 1169 1170 *size = nb_read; 1171 return 0; 1172 1173 err_free: 1174 free(*data_ptr); 1175 *data_ptr = NULL; 1176 err_fclose: 1177 if (f != stdin) 1178 fclose(f); 1179 return -1; 1180 } 1181 1182 static void hex_print(void *data, unsigned int size, FILE *f) 1183 { 1184 size_t i, j; 1185 char c; 1186 1187 for (i = 0; i < size; i += 16) { 1188 /* Row offset */ 1189 fprintf(f, "%07zx\t", i); 1190 1191 /* Hexadecimal values */ 1192 for (j = i; j < i + 16 && j < size; j++) 1193 fprintf(f, "%02x%s", *(uint8_t *)(data + j), 1194 j % 2 ? " " : ""); 1195 for (; j < i + 16; j++) 1196 fprintf(f, " %s", j % 2 ? " " : ""); 1197 1198 /* ASCII values (if relevant), '.' otherwise */ 1199 fprintf(f, "| "); 1200 for (j = i; j < i + 16 && j < size; j++) { 1201 c = *(char *)(data + j); 1202 if (c < ' ' || c > '~') 1203 c = '.'; 1204 fprintf(f, "%c%s", c, j == i + 7 ? " " : ""); 1205 } 1206 1207 fprintf(f, "\n"); 1208 } 1209 } 1210 1211 static int 1212 print_run_output(void *data, unsigned int size, const char *fname, 1213 const char *json_key) 1214 { 1215 size_t nb_written; 1216 FILE *f; 1217 1218 if (!fname) 1219 return 0; 1220 1221 if (!strcmp(fname, "-")) { 1222 f = stdout; 1223 if (json_output) { 1224 jsonw_name(json_wtr, json_key); 1225 print_data_json(data, size); 1226 } else { 1227 hex_print(data, size, f); 1228 } 1229 return 0; 1230 } 1231 1232 f = fopen(fname, "w"); 1233 if (!f) { 1234 p_err("failed to open %s: %s", fname, strerror(errno)); 1235 return -1; 1236 } 1237 1238 nb_written = fwrite(data, 1, size, f); 1239 fclose(f); 1240 if (nb_written != size) { 1241 p_err("failed to write output data/ctx: %s", strerror(errno)); 1242 return -1; 1243 } 1244 1245 return 0; 1246 } 1247 1248 static int alloc_run_data(void **data_ptr, unsigned int size_out) 1249 { 1250 *data_ptr = calloc(size_out, 1); 1251 if (!*data_ptr) { 1252 p_err("failed to allocate memory for output data/ctx: %s", 1253 strerror(errno)); 1254 return -1; 1255 } 1256 1257 return 0; 1258 } 1259 1260 static int do_run(int argc, char **argv) 1261 { 1262 char *data_fname_in = NULL, *data_fname_out = NULL; 1263 char *ctx_fname_in = NULL, *ctx_fname_out = NULL; 1264 struct bpf_prog_test_run_attr test_attr = {0}; 1265 const unsigned int default_size = SZ_32K; 1266 void *data_in = NULL, *data_out = NULL; 1267 void *ctx_in = NULL, *ctx_out = NULL; 1268 unsigned int repeat = 1; 1269 int fd, err; 1270 1271 if (!REQ_ARGS(4)) 1272 return -1; 1273 1274 fd = prog_parse_fd(&argc, &argv); 1275 if (fd < 0) 1276 return -1; 1277 1278 while (argc) { 1279 if (detect_common_prefix(*argv, "data_in", "data_out", 1280 "data_size_out", NULL)) 1281 return -1; 1282 if (detect_common_prefix(*argv, "ctx_in", "ctx_out", 1283 "ctx_size_out", NULL)) 1284 return -1; 1285 1286 if (is_prefix(*argv, "data_in")) { 1287 NEXT_ARG(); 1288 if (!REQ_ARGS(1)) 1289 return -1; 1290 1291 data_fname_in = GET_ARG(); 1292 if (check_single_stdin(data_fname_in, ctx_fname_in)) 1293 return -1; 1294 } else if (is_prefix(*argv, "data_out")) { 1295 NEXT_ARG(); 1296 if (!REQ_ARGS(1)) 1297 return -1; 1298 1299 data_fname_out = GET_ARG(); 1300 } else if (is_prefix(*argv, "data_size_out")) { 1301 char *endptr; 1302 1303 NEXT_ARG(); 1304 if (!REQ_ARGS(1)) 1305 return -1; 1306 1307 test_attr.data_size_out = strtoul(*argv, &endptr, 0); 1308 if (*endptr) { 1309 p_err("can't parse %s as output data size", 1310 *argv); 1311 return -1; 1312 } 1313 NEXT_ARG(); 1314 } else if (is_prefix(*argv, "ctx_in")) { 1315 NEXT_ARG(); 1316 if (!REQ_ARGS(1)) 1317 return -1; 1318 1319 ctx_fname_in = GET_ARG(); 1320 if (check_single_stdin(data_fname_in, ctx_fname_in)) 1321 return -1; 1322 } else if (is_prefix(*argv, "ctx_out")) { 1323 NEXT_ARG(); 1324 if (!REQ_ARGS(1)) 1325 return -1; 1326 1327 ctx_fname_out = GET_ARG(); 1328 } else if (is_prefix(*argv, "ctx_size_out")) { 1329 char *endptr; 1330 1331 NEXT_ARG(); 1332 if (!REQ_ARGS(1)) 1333 return -1; 1334 1335 test_attr.ctx_size_out = strtoul(*argv, &endptr, 0); 1336 if (*endptr) { 1337 p_err("can't parse %s as output context size", 1338 *argv); 1339 return -1; 1340 } 1341 NEXT_ARG(); 1342 } else if (is_prefix(*argv, "repeat")) { 1343 char *endptr; 1344 1345 NEXT_ARG(); 1346 if (!REQ_ARGS(1)) 1347 return -1; 1348 1349 repeat = strtoul(*argv, &endptr, 0); 1350 if (*endptr) { 1351 p_err("can't parse %s as repeat number", 1352 *argv); 1353 return -1; 1354 } 1355 NEXT_ARG(); 1356 } else { 1357 p_err("expected no more arguments, 'data_in', 'data_out', 'data_size_out', 'ctx_in', 'ctx_out', 'ctx_size_out' or 'repeat', got: '%s'?", 1358 *argv); 1359 return -1; 1360 } 1361 } 1362 1363 err = get_run_data(data_fname_in, &data_in, &test_attr.data_size_in); 1364 if (err) 1365 return -1; 1366 1367 if (data_in) { 1368 if (!test_attr.data_size_out) 1369 test_attr.data_size_out = default_size; 1370 err = alloc_run_data(&data_out, test_attr.data_size_out); 1371 if (err) 1372 goto free_data_in; 1373 } 1374 1375 err = get_run_data(ctx_fname_in, &ctx_in, &test_attr.ctx_size_in); 1376 if (err) 1377 goto free_data_out; 1378 1379 if (ctx_in) { 1380 if (!test_attr.ctx_size_out) 1381 test_attr.ctx_size_out = default_size; 1382 err = alloc_run_data(&ctx_out, test_attr.ctx_size_out); 1383 if (err) 1384 goto free_ctx_in; 1385 } 1386 1387 test_attr.prog_fd = fd; 1388 test_attr.repeat = repeat; 1389 test_attr.data_in = data_in; 1390 test_attr.data_out = data_out; 1391 test_attr.ctx_in = ctx_in; 1392 test_attr.ctx_out = ctx_out; 1393 1394 err = bpf_prog_test_run_xattr(&test_attr); 1395 if (err) { 1396 p_err("failed to run program: %s", strerror(errno)); 1397 goto free_ctx_out; 1398 } 1399 1400 err = 0; 1401 1402 if (json_output) 1403 jsonw_start_object(json_wtr); /* root */ 1404 1405 /* Do not exit on errors occurring when printing output data/context, 1406 * we still want to print return value and duration for program run. 1407 */ 1408 if (test_attr.data_size_out) 1409 err += print_run_output(test_attr.data_out, 1410 test_attr.data_size_out, 1411 data_fname_out, "data_out"); 1412 if (test_attr.ctx_size_out) 1413 err += print_run_output(test_attr.ctx_out, 1414 test_attr.ctx_size_out, 1415 ctx_fname_out, "ctx_out"); 1416 1417 if (json_output) { 1418 jsonw_uint_field(json_wtr, "retval", test_attr.retval); 1419 jsonw_uint_field(json_wtr, "duration", test_attr.duration); 1420 jsonw_end_object(json_wtr); /* root */ 1421 } else { 1422 fprintf(stdout, "Return value: %u, duration%s: %uns\n", 1423 test_attr.retval, 1424 repeat > 1 ? " (average)" : "", test_attr.duration); 1425 } 1426 1427 free_ctx_out: 1428 free(ctx_out); 1429 free_ctx_in: 1430 free(ctx_in); 1431 free_data_out: 1432 free(data_out); 1433 free_data_in: 1434 free(data_in); 1435 1436 return err; 1437 } 1438 1439 static int 1440 get_prog_type_by_name(const char *name, enum bpf_prog_type *prog_type, 1441 enum bpf_attach_type *expected_attach_type) 1442 { 1443 libbpf_print_fn_t print_backup; 1444 int ret; 1445 1446 ret = libbpf_prog_type_by_name(name, prog_type, expected_attach_type); 1447 if (!ret) 1448 return ret; 1449 1450 /* libbpf_prog_type_by_name() failed, let's re-run with debug level */ 1451 print_backup = libbpf_set_print(print_all_levels); 1452 ret = libbpf_prog_type_by_name(name, prog_type, expected_attach_type); 1453 libbpf_set_print(print_backup); 1454 1455 return ret; 1456 } 1457 1458 static int load_with_options(int argc, char **argv, bool first_prog_only) 1459 { 1460 enum bpf_prog_type common_prog_type = BPF_PROG_TYPE_UNSPEC; 1461 DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts, 1462 .relaxed_maps = relaxed_maps, 1463 ); 1464 struct bpf_object_load_attr load_attr = { 0 }; 1465 enum bpf_attach_type expected_attach_type; 1466 struct map_replace *map_replace = NULL; 1467 struct bpf_program *prog = NULL, *pos; 1468 unsigned int old_map_fds = 0; 1469 const char *pinmaps = NULL; 1470 struct bpf_object *obj; 1471 struct bpf_map *map; 1472 const char *pinfile; 1473 unsigned int i, j; 1474 __u32 ifindex = 0; 1475 const char *file; 1476 int idx, err; 1477 1478 1479 if (!REQ_ARGS(2)) 1480 return -1; 1481 file = GET_ARG(); 1482 pinfile = GET_ARG(); 1483 1484 while (argc) { 1485 if (is_prefix(*argv, "type")) { 1486 NEXT_ARG(); 1487 1488 if (common_prog_type != BPF_PROG_TYPE_UNSPEC) { 1489 p_err("program type already specified"); 1490 goto err_free_reuse_maps; 1491 } 1492 if (!REQ_ARGS(1)) 1493 goto err_free_reuse_maps; 1494 1495 err = libbpf_prog_type_by_name(*argv, &common_prog_type, 1496 &expected_attach_type); 1497 if (err < 0) { 1498 /* Put a '/' at the end of type to appease libbpf */ 1499 char *type = malloc(strlen(*argv) + 2); 1500 1501 if (!type) { 1502 p_err("mem alloc failed"); 1503 goto err_free_reuse_maps; 1504 } 1505 *type = 0; 1506 strcat(type, *argv); 1507 strcat(type, "/"); 1508 1509 err = get_prog_type_by_name(type, &common_prog_type, 1510 &expected_attach_type); 1511 free(type); 1512 if (err < 0) 1513 goto err_free_reuse_maps; 1514 } 1515 1516 NEXT_ARG(); 1517 } else if (is_prefix(*argv, "map")) { 1518 void *new_map_replace; 1519 char *endptr, *name; 1520 int fd; 1521 1522 NEXT_ARG(); 1523 1524 if (!REQ_ARGS(4)) 1525 goto err_free_reuse_maps; 1526 1527 if (is_prefix(*argv, "idx")) { 1528 NEXT_ARG(); 1529 1530 idx = strtoul(*argv, &endptr, 0); 1531 if (*endptr) { 1532 p_err("can't parse %s as IDX", *argv); 1533 goto err_free_reuse_maps; 1534 } 1535 name = NULL; 1536 } else if (is_prefix(*argv, "name")) { 1537 NEXT_ARG(); 1538 1539 name = *argv; 1540 idx = -1; 1541 } else { 1542 p_err("expected 'idx' or 'name', got: '%s'?", 1543 *argv); 1544 goto err_free_reuse_maps; 1545 } 1546 NEXT_ARG(); 1547 1548 fd = map_parse_fd(&argc, &argv); 1549 if (fd < 0) 1550 goto err_free_reuse_maps; 1551 1552 new_map_replace = reallocarray(map_replace, 1553 old_map_fds + 1, 1554 sizeof(*map_replace)); 1555 if (!new_map_replace) { 1556 p_err("mem alloc failed"); 1557 goto err_free_reuse_maps; 1558 } 1559 map_replace = new_map_replace; 1560 1561 map_replace[old_map_fds].idx = idx; 1562 map_replace[old_map_fds].name = name; 1563 map_replace[old_map_fds].fd = fd; 1564 old_map_fds++; 1565 } else if (is_prefix(*argv, "dev")) { 1566 NEXT_ARG(); 1567 1568 if (ifindex) { 1569 p_err("offload device already specified"); 1570 goto err_free_reuse_maps; 1571 } 1572 if (!REQ_ARGS(1)) 1573 goto err_free_reuse_maps; 1574 1575 ifindex = if_nametoindex(*argv); 1576 if (!ifindex) { 1577 p_err("unrecognized netdevice '%s': %s", 1578 *argv, strerror(errno)); 1579 goto err_free_reuse_maps; 1580 } 1581 NEXT_ARG(); 1582 } else if (is_prefix(*argv, "pinmaps")) { 1583 NEXT_ARG(); 1584 1585 if (!REQ_ARGS(1)) 1586 goto err_free_reuse_maps; 1587 1588 pinmaps = GET_ARG(); 1589 } else { 1590 p_err("expected no more arguments, 'type', 'map' or 'dev', got: '%s'?", 1591 *argv); 1592 goto err_free_reuse_maps; 1593 } 1594 } 1595 1596 set_max_rlimit(); 1597 1598 obj = bpf_object__open_file(file, &open_opts); 1599 if (libbpf_get_error(obj)) { 1600 p_err("failed to open object file"); 1601 goto err_free_reuse_maps; 1602 } 1603 1604 bpf_object__for_each_program(pos, obj) { 1605 enum bpf_prog_type prog_type = common_prog_type; 1606 1607 if (prog_type == BPF_PROG_TYPE_UNSPEC) { 1608 const char *sec_name = bpf_program__section_name(pos); 1609 1610 err = get_prog_type_by_name(sec_name, &prog_type, 1611 &expected_attach_type); 1612 if (err < 0) 1613 goto err_close_obj; 1614 } 1615 1616 bpf_program__set_ifindex(pos, ifindex); 1617 bpf_program__set_type(pos, prog_type); 1618 bpf_program__set_expected_attach_type(pos, expected_attach_type); 1619 } 1620 1621 qsort(map_replace, old_map_fds, sizeof(*map_replace), 1622 map_replace_compar); 1623 1624 /* After the sort maps by name will be first on the list, because they 1625 * have idx == -1. Resolve them. 1626 */ 1627 j = 0; 1628 while (j < old_map_fds && map_replace[j].name) { 1629 i = 0; 1630 bpf_object__for_each_map(map, obj) { 1631 if (!strcmp(bpf_map__name(map), map_replace[j].name)) { 1632 map_replace[j].idx = i; 1633 break; 1634 } 1635 i++; 1636 } 1637 if (map_replace[j].idx == -1) { 1638 p_err("unable to find map '%s'", map_replace[j].name); 1639 goto err_close_obj; 1640 } 1641 j++; 1642 } 1643 /* Resort if any names were resolved */ 1644 if (j) 1645 qsort(map_replace, old_map_fds, sizeof(*map_replace), 1646 map_replace_compar); 1647 1648 /* Set ifindex and name reuse */ 1649 j = 0; 1650 idx = 0; 1651 bpf_object__for_each_map(map, obj) { 1652 if (!bpf_map__is_offload_neutral(map)) 1653 bpf_map__set_ifindex(map, ifindex); 1654 1655 if (j < old_map_fds && idx == map_replace[j].idx) { 1656 err = bpf_map__reuse_fd(map, map_replace[j++].fd); 1657 if (err) { 1658 p_err("unable to set up map reuse: %d", err); 1659 goto err_close_obj; 1660 } 1661 1662 /* Next reuse wants to apply to the same map */ 1663 if (j < old_map_fds && map_replace[j].idx == idx) { 1664 p_err("replacement for map idx %d specified more than once", 1665 idx); 1666 goto err_close_obj; 1667 } 1668 } 1669 1670 idx++; 1671 } 1672 if (j < old_map_fds) { 1673 p_err("map idx '%d' not used", map_replace[j].idx); 1674 goto err_close_obj; 1675 } 1676 1677 load_attr.obj = obj; 1678 if (verifier_logs) 1679 /* log_level1 + log_level2 + stats, but not stable UAPI */ 1680 load_attr.log_level = 1 + 2 + 4; 1681 1682 err = bpf_object__load_xattr(&load_attr); 1683 if (err) { 1684 p_err("failed to load object file"); 1685 goto err_close_obj; 1686 } 1687 1688 err = mount_bpffs_for_pin(pinfile); 1689 if (err) 1690 goto err_close_obj; 1691 1692 if (first_prog_only) { 1693 prog = bpf_object__next_program(obj, NULL); 1694 if (!prog) { 1695 p_err("object file doesn't contain any bpf program"); 1696 goto err_close_obj; 1697 } 1698 1699 err = bpf_obj_pin(bpf_program__fd(prog), pinfile); 1700 if (err) { 1701 p_err("failed to pin program %s", 1702 bpf_program__section_name(prog)); 1703 goto err_close_obj; 1704 } 1705 } else { 1706 err = bpf_object__pin_programs(obj, pinfile); 1707 if (err) { 1708 p_err("failed to pin all programs"); 1709 goto err_close_obj; 1710 } 1711 } 1712 1713 if (pinmaps) { 1714 err = bpf_object__pin_maps(obj, pinmaps); 1715 if (err) { 1716 p_err("failed to pin all maps"); 1717 goto err_unpin; 1718 } 1719 } 1720 1721 if (json_output) 1722 jsonw_null(json_wtr); 1723 1724 bpf_object__close(obj); 1725 for (i = 0; i < old_map_fds; i++) 1726 close(map_replace[i].fd); 1727 free(map_replace); 1728 1729 return 0; 1730 1731 err_unpin: 1732 if (first_prog_only) 1733 unlink(pinfile); 1734 else 1735 bpf_object__unpin_programs(obj, pinfile); 1736 err_close_obj: 1737 if (!legacy_libbpf) { 1738 p_info("Warning: bpftool is now running in libbpf strict mode and has more stringent requirements about BPF programs.\n" 1739 "If it used to work for this object file but now doesn't, see --legacy option for more details.\n"); 1740 } 1741 1742 bpf_object__close(obj); 1743 err_free_reuse_maps: 1744 for (i = 0; i < old_map_fds; i++) 1745 close(map_replace[i].fd); 1746 free(map_replace); 1747 return -1; 1748 } 1749 1750 static int count_open_fds(void) 1751 { 1752 DIR *dp = opendir("/proc/self/fd"); 1753 struct dirent *de; 1754 int cnt = -3; 1755 1756 if (!dp) 1757 return -1; 1758 1759 while ((de = readdir(dp))) 1760 cnt++; 1761 1762 closedir(dp); 1763 return cnt; 1764 } 1765 1766 static int try_loader(struct gen_loader_opts *gen) 1767 { 1768 struct bpf_load_and_run_opts opts = {}; 1769 struct bpf_loader_ctx *ctx; 1770 int ctx_sz = sizeof(*ctx) + 64 * max(sizeof(struct bpf_map_desc), 1771 sizeof(struct bpf_prog_desc)); 1772 int log_buf_sz = (1u << 24) - 1; 1773 int err, fds_before, fd_delta; 1774 char *log_buf; 1775 1776 ctx = alloca(ctx_sz); 1777 memset(ctx, 0, ctx_sz); 1778 ctx->sz = ctx_sz; 1779 ctx->log_level = 1; 1780 ctx->log_size = log_buf_sz; 1781 log_buf = malloc(log_buf_sz); 1782 if (!log_buf) 1783 return -ENOMEM; 1784 ctx->log_buf = (long) log_buf; 1785 opts.ctx = ctx; 1786 opts.data = gen->data; 1787 opts.data_sz = gen->data_sz; 1788 opts.insns = gen->insns; 1789 opts.insns_sz = gen->insns_sz; 1790 fds_before = count_open_fds(); 1791 err = bpf_load_and_run(&opts); 1792 fd_delta = count_open_fds() - fds_before; 1793 if (err < 0) { 1794 fprintf(stderr, "err %d\n%s\n%s", err, opts.errstr, log_buf); 1795 if (fd_delta) 1796 fprintf(stderr, "loader prog leaked %d FDs\n", 1797 fd_delta); 1798 } 1799 free(log_buf); 1800 return err; 1801 } 1802 1803 static int do_loader(int argc, char **argv) 1804 { 1805 DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts); 1806 DECLARE_LIBBPF_OPTS(gen_loader_opts, gen); 1807 struct bpf_object_load_attr load_attr = {}; 1808 struct bpf_object *obj; 1809 const char *file; 1810 int err = 0; 1811 1812 if (!REQ_ARGS(1)) 1813 return -1; 1814 file = GET_ARG(); 1815 1816 obj = bpf_object__open_file(file, &open_opts); 1817 if (libbpf_get_error(obj)) { 1818 p_err("failed to open object file"); 1819 goto err_close_obj; 1820 } 1821 1822 err = bpf_object__gen_loader(obj, &gen); 1823 if (err) 1824 goto err_close_obj; 1825 1826 load_attr.obj = obj; 1827 if (verifier_logs) 1828 /* log_level1 + log_level2 + stats, but not stable UAPI */ 1829 load_attr.log_level = 1 + 2 + 4; 1830 1831 err = bpf_object__load_xattr(&load_attr); 1832 if (err) { 1833 p_err("failed to load object file"); 1834 goto err_close_obj; 1835 } 1836 1837 if (verifier_logs) { 1838 struct dump_data dd = {}; 1839 1840 kernel_syms_load(&dd); 1841 dump_xlated_plain(&dd, (void *)gen.insns, gen.insns_sz, false, false); 1842 kernel_syms_destroy(&dd); 1843 } 1844 err = try_loader(&gen); 1845 err_close_obj: 1846 bpf_object__close(obj); 1847 return err; 1848 } 1849 1850 static int do_load(int argc, char **argv) 1851 { 1852 if (use_loader) 1853 return do_loader(argc, argv); 1854 return load_with_options(argc, argv, true); 1855 } 1856 1857 static int do_loadall(int argc, char **argv) 1858 { 1859 return load_with_options(argc, argv, false); 1860 } 1861 1862 #ifdef BPFTOOL_WITHOUT_SKELETONS 1863 1864 static int do_profile(int argc, char **argv) 1865 { 1866 p_err("bpftool prog profile command is not supported. Please build bpftool with clang >= 10.0.0"); 1867 return 0; 1868 } 1869 1870 #else /* BPFTOOL_WITHOUT_SKELETONS */ 1871 1872 #include "profiler.skel.h" 1873 1874 struct profile_metric { 1875 const char *name; 1876 struct bpf_perf_event_value val; 1877 struct perf_event_attr attr; 1878 bool selected; 1879 1880 /* calculate ratios like instructions per cycle */ 1881 const int ratio_metric; /* 0 for N/A, 1 for index 0 (cycles) */ 1882 const char *ratio_desc; 1883 const float ratio_mul; 1884 } metrics[] = { 1885 { 1886 .name = "cycles", 1887 .attr = { 1888 .type = PERF_TYPE_HARDWARE, 1889 .config = PERF_COUNT_HW_CPU_CYCLES, 1890 .exclude_user = 1, 1891 }, 1892 }, 1893 { 1894 .name = "instructions", 1895 .attr = { 1896 .type = PERF_TYPE_HARDWARE, 1897 .config = PERF_COUNT_HW_INSTRUCTIONS, 1898 .exclude_user = 1, 1899 }, 1900 .ratio_metric = 1, 1901 .ratio_desc = "insns per cycle", 1902 .ratio_mul = 1.0, 1903 }, 1904 { 1905 .name = "l1d_loads", 1906 .attr = { 1907 .type = PERF_TYPE_HW_CACHE, 1908 .config = 1909 PERF_COUNT_HW_CACHE_L1D | 1910 (PERF_COUNT_HW_CACHE_OP_READ << 8) | 1911 (PERF_COUNT_HW_CACHE_RESULT_ACCESS << 16), 1912 .exclude_user = 1, 1913 }, 1914 }, 1915 { 1916 .name = "llc_misses", 1917 .attr = { 1918 .type = PERF_TYPE_HW_CACHE, 1919 .config = 1920 PERF_COUNT_HW_CACHE_LL | 1921 (PERF_COUNT_HW_CACHE_OP_READ << 8) | 1922 (PERF_COUNT_HW_CACHE_RESULT_MISS << 16), 1923 .exclude_user = 1 1924 }, 1925 .ratio_metric = 2, 1926 .ratio_desc = "LLC misses per million insns", 1927 .ratio_mul = 1e6, 1928 }, 1929 { 1930 .name = "itlb_misses", 1931 .attr = { 1932 .type = PERF_TYPE_HW_CACHE, 1933 .config = 1934 PERF_COUNT_HW_CACHE_ITLB | 1935 (PERF_COUNT_HW_CACHE_OP_READ << 8) | 1936 (PERF_COUNT_HW_CACHE_RESULT_MISS << 16), 1937 .exclude_user = 1 1938 }, 1939 .ratio_metric = 2, 1940 .ratio_desc = "itlb misses per million insns", 1941 .ratio_mul = 1e6, 1942 }, 1943 { 1944 .name = "dtlb_misses", 1945 .attr = { 1946 .type = PERF_TYPE_HW_CACHE, 1947 .config = 1948 PERF_COUNT_HW_CACHE_DTLB | 1949 (PERF_COUNT_HW_CACHE_OP_READ << 8) | 1950 (PERF_COUNT_HW_CACHE_RESULT_MISS << 16), 1951 .exclude_user = 1 1952 }, 1953 .ratio_metric = 2, 1954 .ratio_desc = "dtlb misses per million insns", 1955 .ratio_mul = 1e6, 1956 }, 1957 }; 1958 1959 static __u64 profile_total_count; 1960 1961 #define MAX_NUM_PROFILE_METRICS 4 1962 1963 static int profile_parse_metrics(int argc, char **argv) 1964 { 1965 unsigned int metric_cnt; 1966 int selected_cnt = 0; 1967 unsigned int i; 1968 1969 metric_cnt = sizeof(metrics) / sizeof(struct profile_metric); 1970 1971 while (argc > 0) { 1972 for (i = 0; i < metric_cnt; i++) { 1973 if (is_prefix(argv[0], metrics[i].name)) { 1974 if (!metrics[i].selected) 1975 selected_cnt++; 1976 metrics[i].selected = true; 1977 break; 1978 } 1979 } 1980 if (i == metric_cnt) { 1981 p_err("unknown metric %s", argv[0]); 1982 return -1; 1983 } 1984 NEXT_ARG(); 1985 } 1986 if (selected_cnt > MAX_NUM_PROFILE_METRICS) { 1987 p_err("too many (%d) metrics, please specify no more than %d metrics at at time", 1988 selected_cnt, MAX_NUM_PROFILE_METRICS); 1989 return -1; 1990 } 1991 return selected_cnt; 1992 } 1993 1994 static void profile_read_values(struct profiler_bpf *obj) 1995 { 1996 __u32 m, cpu, num_cpu = obj->rodata->num_cpu; 1997 int reading_map_fd, count_map_fd; 1998 __u64 counts[num_cpu]; 1999 __u32 key = 0; 2000 int err; 2001 2002 reading_map_fd = bpf_map__fd(obj->maps.accum_readings); 2003 count_map_fd = bpf_map__fd(obj->maps.counts); 2004 if (reading_map_fd < 0 || count_map_fd < 0) { 2005 p_err("failed to get fd for map"); 2006 return; 2007 } 2008 2009 err = bpf_map_lookup_elem(count_map_fd, &key, counts); 2010 if (err) { 2011 p_err("failed to read count_map: %s", strerror(errno)); 2012 return; 2013 } 2014 2015 profile_total_count = 0; 2016 for (cpu = 0; cpu < num_cpu; cpu++) 2017 profile_total_count += counts[cpu]; 2018 2019 for (m = 0; m < ARRAY_SIZE(metrics); m++) { 2020 struct bpf_perf_event_value values[num_cpu]; 2021 2022 if (!metrics[m].selected) 2023 continue; 2024 2025 err = bpf_map_lookup_elem(reading_map_fd, &key, values); 2026 if (err) { 2027 p_err("failed to read reading_map: %s", 2028 strerror(errno)); 2029 return; 2030 } 2031 for (cpu = 0; cpu < num_cpu; cpu++) { 2032 metrics[m].val.counter += values[cpu].counter; 2033 metrics[m].val.enabled += values[cpu].enabled; 2034 metrics[m].val.running += values[cpu].running; 2035 } 2036 key++; 2037 } 2038 } 2039 2040 static void profile_print_readings_json(void) 2041 { 2042 __u32 m; 2043 2044 jsonw_start_array(json_wtr); 2045 for (m = 0; m < ARRAY_SIZE(metrics); m++) { 2046 if (!metrics[m].selected) 2047 continue; 2048 jsonw_start_object(json_wtr); 2049 jsonw_string_field(json_wtr, "metric", metrics[m].name); 2050 jsonw_lluint_field(json_wtr, "run_cnt", profile_total_count); 2051 jsonw_lluint_field(json_wtr, "value", metrics[m].val.counter); 2052 jsonw_lluint_field(json_wtr, "enabled", metrics[m].val.enabled); 2053 jsonw_lluint_field(json_wtr, "running", metrics[m].val.running); 2054 2055 jsonw_end_object(json_wtr); 2056 } 2057 jsonw_end_array(json_wtr); 2058 } 2059 2060 static void profile_print_readings_plain(void) 2061 { 2062 __u32 m; 2063 2064 printf("\n%18llu %-20s\n", profile_total_count, "run_cnt"); 2065 for (m = 0; m < ARRAY_SIZE(metrics); m++) { 2066 struct bpf_perf_event_value *val = &metrics[m].val; 2067 int r; 2068 2069 if (!metrics[m].selected) 2070 continue; 2071 printf("%18llu %-20s", val->counter, metrics[m].name); 2072 2073 r = metrics[m].ratio_metric - 1; 2074 if (r >= 0 && metrics[r].selected && 2075 metrics[r].val.counter > 0) { 2076 printf("# %8.2f %-30s", 2077 val->counter * metrics[m].ratio_mul / 2078 metrics[r].val.counter, 2079 metrics[m].ratio_desc); 2080 } else { 2081 printf("%-41s", ""); 2082 } 2083 2084 if (val->enabled > val->running) 2085 printf("(%4.2f%%)", 2086 val->running * 100.0 / val->enabled); 2087 printf("\n"); 2088 } 2089 } 2090 2091 static void profile_print_readings(void) 2092 { 2093 if (json_output) 2094 profile_print_readings_json(); 2095 else 2096 profile_print_readings_plain(); 2097 } 2098 2099 static char *profile_target_name(int tgt_fd) 2100 { 2101 struct bpf_func_info func_info; 2102 struct bpf_prog_info info = {}; 2103 __u32 info_len = sizeof(info); 2104 const struct btf_type *t; 2105 __u32 func_info_rec_size; 2106 struct btf *btf = NULL; 2107 char *name = NULL; 2108 int err; 2109 2110 err = bpf_obj_get_info_by_fd(tgt_fd, &info, &info_len); 2111 if (err) { 2112 p_err("failed to bpf_obj_get_info_by_fd for prog FD %d", tgt_fd); 2113 goto out; 2114 } 2115 2116 if (info.btf_id == 0) { 2117 p_err("prog FD %d doesn't have valid btf", tgt_fd); 2118 goto out; 2119 } 2120 2121 func_info_rec_size = info.func_info_rec_size; 2122 if (info.nr_func_info == 0) { 2123 p_err("bpf_obj_get_info_by_fd for prog FD %d found 0 func_info", tgt_fd); 2124 goto out; 2125 } 2126 2127 memset(&info, 0, sizeof(info)); 2128 info.nr_func_info = 1; 2129 info.func_info_rec_size = func_info_rec_size; 2130 info.func_info = ptr_to_u64(&func_info); 2131 2132 err = bpf_obj_get_info_by_fd(tgt_fd, &info, &info_len); 2133 if (err) { 2134 p_err("failed to get func_info for prog FD %d", tgt_fd); 2135 goto out; 2136 } 2137 2138 btf = btf__load_from_kernel_by_id(info.btf_id); 2139 if (libbpf_get_error(btf)) { 2140 p_err("failed to load btf for prog FD %d", tgt_fd); 2141 goto out; 2142 } 2143 2144 t = btf__type_by_id(btf, func_info.type_id); 2145 if (!t) { 2146 p_err("btf %d doesn't have type %d", 2147 info.btf_id, func_info.type_id); 2148 goto out; 2149 } 2150 name = strdup(btf__name_by_offset(btf, t->name_off)); 2151 out: 2152 btf__free(btf); 2153 return name; 2154 } 2155 2156 static struct profiler_bpf *profile_obj; 2157 static int profile_tgt_fd = -1; 2158 static char *profile_tgt_name; 2159 static int *profile_perf_events; 2160 static int profile_perf_event_cnt; 2161 2162 static void profile_close_perf_events(struct profiler_bpf *obj) 2163 { 2164 int i; 2165 2166 for (i = profile_perf_event_cnt - 1; i >= 0; i--) 2167 close(profile_perf_events[i]); 2168 2169 free(profile_perf_events); 2170 profile_perf_event_cnt = 0; 2171 } 2172 2173 static int profile_open_perf_events(struct profiler_bpf *obj) 2174 { 2175 unsigned int cpu, m; 2176 int map_fd, pmu_fd; 2177 2178 profile_perf_events = calloc( 2179 sizeof(int), obj->rodata->num_cpu * obj->rodata->num_metric); 2180 if (!profile_perf_events) { 2181 p_err("failed to allocate memory for perf_event array: %s", 2182 strerror(errno)); 2183 return -1; 2184 } 2185 map_fd = bpf_map__fd(obj->maps.events); 2186 if (map_fd < 0) { 2187 p_err("failed to get fd for events map"); 2188 return -1; 2189 } 2190 2191 for (m = 0; m < ARRAY_SIZE(metrics); m++) { 2192 if (!metrics[m].selected) 2193 continue; 2194 for (cpu = 0; cpu < obj->rodata->num_cpu; cpu++) { 2195 pmu_fd = syscall(__NR_perf_event_open, &metrics[m].attr, 2196 -1/*pid*/, cpu, -1/*group_fd*/, 0); 2197 if (pmu_fd < 0 || 2198 bpf_map_update_elem(map_fd, &profile_perf_event_cnt, 2199 &pmu_fd, BPF_ANY) || 2200 ioctl(pmu_fd, PERF_EVENT_IOC_ENABLE, 0)) { 2201 p_err("failed to create event %s on cpu %d", 2202 metrics[m].name, cpu); 2203 return -1; 2204 } 2205 profile_perf_events[profile_perf_event_cnt++] = pmu_fd; 2206 } 2207 } 2208 return 0; 2209 } 2210 2211 static void profile_print_and_cleanup(void) 2212 { 2213 profile_close_perf_events(profile_obj); 2214 profile_read_values(profile_obj); 2215 profile_print_readings(); 2216 profiler_bpf__destroy(profile_obj); 2217 2218 close(profile_tgt_fd); 2219 free(profile_tgt_name); 2220 } 2221 2222 static void int_exit(int signo) 2223 { 2224 profile_print_and_cleanup(); 2225 exit(0); 2226 } 2227 2228 static int do_profile(int argc, char **argv) 2229 { 2230 int num_metric, num_cpu, err = -1; 2231 struct bpf_program *prog; 2232 unsigned long duration; 2233 char *endptr; 2234 2235 /* we at least need two args for the prog and one metric */ 2236 if (!REQ_ARGS(3)) 2237 return -EINVAL; 2238 2239 /* parse target fd */ 2240 profile_tgt_fd = prog_parse_fd(&argc, &argv); 2241 if (profile_tgt_fd < 0) { 2242 p_err("failed to parse fd"); 2243 return -1; 2244 } 2245 2246 /* parse profiling optional duration */ 2247 if (argc > 2 && is_prefix(argv[0], "duration")) { 2248 NEXT_ARG(); 2249 duration = strtoul(*argv, &endptr, 0); 2250 if (*endptr) 2251 usage(); 2252 NEXT_ARG(); 2253 } else { 2254 duration = UINT_MAX; 2255 } 2256 2257 num_metric = profile_parse_metrics(argc, argv); 2258 if (num_metric <= 0) 2259 goto out; 2260 2261 num_cpu = libbpf_num_possible_cpus(); 2262 if (num_cpu <= 0) { 2263 p_err("failed to identify number of CPUs"); 2264 goto out; 2265 } 2266 2267 profile_obj = profiler_bpf__open(); 2268 if (!profile_obj) { 2269 p_err("failed to open and/or load BPF object"); 2270 goto out; 2271 } 2272 2273 profile_obj->rodata->num_cpu = num_cpu; 2274 profile_obj->rodata->num_metric = num_metric; 2275 2276 /* adjust map sizes */ 2277 bpf_map__resize(profile_obj->maps.events, num_metric * num_cpu); 2278 bpf_map__resize(profile_obj->maps.fentry_readings, num_metric); 2279 bpf_map__resize(profile_obj->maps.accum_readings, num_metric); 2280 bpf_map__resize(profile_obj->maps.counts, 1); 2281 2282 /* change target name */ 2283 profile_tgt_name = profile_target_name(profile_tgt_fd); 2284 if (!profile_tgt_name) 2285 goto out; 2286 2287 bpf_object__for_each_program(prog, profile_obj->obj) { 2288 err = bpf_program__set_attach_target(prog, profile_tgt_fd, 2289 profile_tgt_name); 2290 if (err) { 2291 p_err("failed to set attach target\n"); 2292 goto out; 2293 } 2294 } 2295 2296 set_max_rlimit(); 2297 err = profiler_bpf__load(profile_obj); 2298 if (err) { 2299 p_err("failed to load profile_obj"); 2300 goto out; 2301 } 2302 2303 err = profile_open_perf_events(profile_obj); 2304 if (err) 2305 goto out; 2306 2307 err = profiler_bpf__attach(profile_obj); 2308 if (err) { 2309 p_err("failed to attach profile_obj"); 2310 goto out; 2311 } 2312 signal(SIGINT, int_exit); 2313 2314 sleep(duration); 2315 profile_print_and_cleanup(); 2316 return 0; 2317 2318 out: 2319 profile_close_perf_events(profile_obj); 2320 if (profile_obj) 2321 profiler_bpf__destroy(profile_obj); 2322 close(profile_tgt_fd); 2323 free(profile_tgt_name); 2324 return err; 2325 } 2326 2327 #endif /* BPFTOOL_WITHOUT_SKELETONS */ 2328 2329 static int do_help(int argc, char **argv) 2330 { 2331 if (json_output) { 2332 jsonw_null(json_wtr); 2333 return 0; 2334 } 2335 2336 fprintf(stderr, 2337 "Usage: %1$s %2$s { show | list } [PROG]\n" 2338 " %1$s %2$s dump xlated PROG [{ file FILE | opcodes | visual | linum }]\n" 2339 " %1$s %2$s dump jited PROG [{ file FILE | opcodes | linum }]\n" 2340 " %1$s %2$s pin PROG FILE\n" 2341 " %1$s %2$s { load | loadall } OBJ PATH \\\n" 2342 " [type TYPE] [dev NAME] \\\n" 2343 " [map { idx IDX | name NAME } MAP]\\\n" 2344 " [pinmaps MAP_DIR]\n" 2345 " %1$s %2$s attach PROG ATTACH_TYPE [MAP]\n" 2346 " %1$s %2$s detach PROG ATTACH_TYPE [MAP]\n" 2347 " %1$s %2$s run PROG \\\n" 2348 " data_in FILE \\\n" 2349 " [data_out FILE [data_size_out L]] \\\n" 2350 " [ctx_in FILE [ctx_out FILE [ctx_size_out M]]] \\\n" 2351 " [repeat N]\n" 2352 " %1$s %2$s profile PROG [duration DURATION] METRICs\n" 2353 " %1$s %2$s tracelog\n" 2354 " %1$s %2$s help\n" 2355 "\n" 2356 " " HELP_SPEC_MAP "\n" 2357 " " HELP_SPEC_PROGRAM "\n" 2358 " TYPE := { socket | kprobe | kretprobe | classifier | action |\n" 2359 " tracepoint | raw_tracepoint | xdp | perf_event | cgroup/skb |\n" 2360 " cgroup/sock | cgroup/dev | lwt_in | lwt_out | lwt_xmit |\n" 2361 " lwt_seg6local | sockops | sk_skb | sk_msg | lirc_mode2 |\n" 2362 " sk_reuseport | flow_dissector | cgroup/sysctl |\n" 2363 " cgroup/bind4 | cgroup/bind6 | cgroup/post_bind4 |\n" 2364 " cgroup/post_bind6 | cgroup/connect4 | cgroup/connect6 |\n" 2365 " cgroup/getpeername4 | cgroup/getpeername6 |\n" 2366 " cgroup/getsockname4 | cgroup/getsockname6 | cgroup/sendmsg4 |\n" 2367 " cgroup/sendmsg6 | cgroup/recvmsg4 | cgroup/recvmsg6 |\n" 2368 " cgroup/getsockopt | cgroup/setsockopt | cgroup/sock_release |\n" 2369 " struct_ops | fentry | fexit | freplace | sk_lookup }\n" 2370 " ATTACH_TYPE := { msg_verdict | skb_verdict | stream_verdict |\n" 2371 " stream_parser | flow_dissector }\n" 2372 " METRIC := { cycles | instructions | l1d_loads | llc_misses | itlb_misses | dtlb_misses }\n" 2373 " " HELP_SPEC_OPTIONS " |\n" 2374 " {-f|--bpffs} | {-m|--mapcompat} | {-n|--nomount} |\n" 2375 " {-L|--use-loader} }\n" 2376 "", 2377 bin_name, argv[-2]); 2378 2379 return 0; 2380 } 2381 2382 static const struct cmd cmds[] = { 2383 { "show", do_show }, 2384 { "list", do_show }, 2385 { "help", do_help }, 2386 { "dump", do_dump }, 2387 { "pin", do_pin }, 2388 { "load", do_load }, 2389 { "loadall", do_loadall }, 2390 { "attach", do_attach }, 2391 { "detach", do_detach }, 2392 { "tracelog", do_tracelog }, 2393 { "run", do_run }, 2394 { "profile", do_profile }, 2395 { 0 } 2396 }; 2397 2398 int do_prog(int argc, char **argv) 2399 { 2400 return cmd_select(cmds, argc, argv, do_help); 2401 } 2402