1 /* 2 * Copyright (C) 2017-2018 Netronome Systems, Inc. 3 * 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. 8 * 9 * The BSD 2-Clause License: 10 * 11 * Redistribution and use in source and binary forms, with or 12 * without modification, are permitted provided that the following 13 * conditions are met: 14 * 15 * 1. Redistributions of source code must retain the above 16 * copyright notice, this list of conditions and the following 17 * disclaimer. 18 * 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. 23 * 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. 32 */ 33 34 #ifndef __BPF_TOOL_H 35 #define __BPF_TOOL_H 36 37 /* BFD and kernel.h both define GCC_VERSION, differently */ 38 #undef GCC_VERSION 39 #include <stdbool.h> 40 #include <stdio.h> 41 #include <linux/bpf.h> 42 #include <linux/compiler.h> 43 #include <linux/kernel.h> 44 #include <linux/hashtable.h> 45 46 #include "json_writer.h" 47 48 #define ptr_to_u64(ptr) ((__u64)(unsigned long)(ptr)) 49 50 #define NEXT_ARG() ({ argc--; argv++; if (argc < 0) usage(); }) 51 #define NEXT_ARGP() ({ (*argc)--; (*argv)++; if (*argc < 0) usage(); }) 52 #define BAD_ARG() ({ p_err("what is '%s'?", *argv); -1; }) 53 #define GET_ARG() ({ argc--; *argv++; }) 54 #define REQ_ARGS(cnt) \ 55 ({ \ 56 int _cnt = (cnt); \ 57 bool _res; \ 58 \ 59 if (argc < _cnt) { \ 60 p_err("'%s' needs at least %d arguments, %d found", \ 61 argv[-1], _cnt, argc); \ 62 _res = false; \ 63 } else { \ 64 _res = true; \ 65 } \ 66 _res; \ 67 }) 68 69 #define ERR_MAX_LEN 1024 70 71 #define BPF_TAG_FMT "%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx" 72 73 #define HELP_SPEC_PROGRAM \ 74 "PROG := { id PROG_ID | pinned FILE | tag PROG_TAG }" 75 #define HELP_SPEC_OPTIONS \ 76 "OPTIONS := { {-j|--json} [{-p|--pretty}] | {-f|--bpffs} }" 77 78 enum bpf_obj_type { 79 BPF_OBJ_UNKNOWN, 80 BPF_OBJ_PROG, 81 BPF_OBJ_MAP, 82 }; 83 84 extern const char *bin_name; 85 86 extern json_writer_t *json_wtr; 87 extern bool json_output; 88 extern bool show_pinned; 89 extern struct pinned_obj_table prog_table; 90 extern struct pinned_obj_table map_table; 91 92 void p_err(const char *fmt, ...); 93 void p_info(const char *fmt, ...); 94 95 bool is_prefix(const char *pfx, const char *str); 96 void fprint_hex(FILE *f, void *arg, unsigned int n, const char *sep); 97 void usage(void) __noreturn; 98 99 struct pinned_obj_table { 100 DECLARE_HASHTABLE(table, 16); 101 }; 102 103 struct pinned_obj { 104 __u32 id; 105 char *path; 106 struct hlist_node hash; 107 }; 108 109 int build_pinned_obj_table(struct pinned_obj_table *table, 110 enum bpf_obj_type type); 111 void delete_pinned_obj_table(struct pinned_obj_table *tab); 112 void print_dev_plain(__u32 ifindex, __u64 ns_dev, __u64 ns_inode); 113 void print_dev_json(__u32 ifindex, __u64 ns_dev, __u64 ns_inode); 114 115 struct cmd { 116 const char *cmd; 117 int (*func)(int argc, char **argv); 118 }; 119 120 int cmd_select(const struct cmd *cmds, int argc, char **argv, 121 int (*help)(int argc, char **argv)); 122 123 int get_fd_type(int fd); 124 const char *get_fd_type_name(enum bpf_obj_type type); 125 char *get_fdinfo(int fd, const char *key); 126 int open_obj_pinned(char *path); 127 int open_obj_pinned_any(char *path, enum bpf_obj_type exp_type); 128 int do_pin_any(int argc, char **argv, int (*get_fd_by_id)(__u32)); 129 int do_pin_fd(int fd, const char *name); 130 131 int do_prog(int argc, char **arg); 132 int do_map(int argc, char **arg); 133 int do_event_pipe(int argc, char **argv); 134 int do_cgroup(int argc, char **arg); 135 int do_perf(int argc, char **arg); 136 137 int prog_parse_fd(int *argc, char ***argv); 138 int map_parse_fd_and_info(int *argc, char ***argv, void *info, __u32 *info_len); 139 140 void disasm_print_insn(unsigned char *image, ssize_t len, int opcodes, 141 const char *arch); 142 void print_data_json(uint8_t *data, size_t len); 143 void print_hex_data_json(uint8_t *data, size_t len); 144 145 unsigned int get_page_size(void); 146 unsigned int get_possible_cpus(void); 147 const char *ifindex_to_bfd_name_ns(__u32 ifindex, __u64 ns_dev, __u64 ns_ino); 148 149 #endif 150