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 #include <tools/libc_compat.h> 46 47 #include "json_writer.h" 48 49 #define ptr_to_u64(ptr) ((__u64)(unsigned long)(ptr)) 50 51 #define NEXT_ARG() ({ argc--; argv++; if (argc < 0) usage(); }) 52 #define NEXT_ARGP() ({ (*argc)--; (*argv)++; if (*argc < 0) usage(); }) 53 #define BAD_ARG() ({ p_err("what is '%s'?", *argv); -1; }) 54 #define GET_ARG() ({ argc--; *argv++; }) 55 #define REQ_ARGS(cnt) \ 56 ({ \ 57 int _cnt = (cnt); \ 58 bool _res; \ 59 \ 60 if (argc < _cnt) { \ 61 p_err("'%s' needs at least %d arguments, %d found", \ 62 argv[-1], _cnt, argc); \ 63 _res = false; \ 64 } else { \ 65 _res = true; \ 66 } \ 67 _res; \ 68 }) 69 70 #define ERR_MAX_LEN 1024 71 72 #define BPF_TAG_FMT "%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx" 73 74 #define HELP_SPEC_PROGRAM \ 75 "PROG := { id PROG_ID | pinned FILE | tag PROG_TAG }" 76 #define HELP_SPEC_OPTIONS \ 77 "OPTIONS := { {-j|--json} [{-p|--pretty}] | {-f|--bpffs} | {-m|--mapcompat}" 78 #define HELP_SPEC_MAP \ 79 "MAP := { id MAP_ID | pinned FILE }" 80 81 enum bpf_obj_type { 82 BPF_OBJ_UNKNOWN, 83 BPF_OBJ_PROG, 84 BPF_OBJ_MAP, 85 }; 86 87 extern const char *bin_name; 88 89 extern json_writer_t *json_wtr; 90 extern bool json_output; 91 extern bool show_pinned; 92 extern int bpf_flags; 93 extern struct pinned_obj_table prog_table; 94 extern struct pinned_obj_table map_table; 95 96 void p_err(const char *fmt, ...); 97 void p_info(const char *fmt, ...); 98 99 bool is_prefix(const char *pfx, const char *str); 100 void fprint_hex(FILE *f, void *arg, unsigned int n, const char *sep); 101 void usage(void) __noreturn; 102 103 void set_max_rlimit(void); 104 105 struct pinned_obj_table { 106 DECLARE_HASHTABLE(table, 16); 107 }; 108 109 struct pinned_obj { 110 __u32 id; 111 char *path; 112 struct hlist_node hash; 113 }; 114 115 int build_pinned_obj_table(struct pinned_obj_table *table, 116 enum bpf_obj_type type); 117 void delete_pinned_obj_table(struct pinned_obj_table *tab); 118 void print_dev_plain(__u32 ifindex, __u64 ns_dev, __u64 ns_inode); 119 void print_dev_json(__u32 ifindex, __u64 ns_dev, __u64 ns_inode); 120 121 struct cmd { 122 const char *cmd; 123 int (*func)(int argc, char **argv); 124 }; 125 126 int cmd_select(const struct cmd *cmds, int argc, char **argv, 127 int (*help)(int argc, char **argv)); 128 129 int get_fd_type(int fd); 130 const char *get_fd_type_name(enum bpf_obj_type type); 131 char *get_fdinfo(int fd, const char *key); 132 int open_obj_pinned(char *path, bool quiet); 133 int open_obj_pinned_any(char *path, enum bpf_obj_type exp_type); 134 int mount_bpffs_for_pin(const char *name); 135 int do_pin_any(int argc, char **argv, int (*get_fd_by_id)(__u32)); 136 int do_pin_fd(int fd, const char *name); 137 138 int do_prog(int argc, char **arg); 139 int do_map(int argc, char **arg); 140 int do_event_pipe(int argc, char **argv); 141 int do_cgroup(int argc, char **arg); 142 int do_perf(int argc, char **arg); 143 int do_net(int argc, char **arg); 144 145 int parse_u32_arg(int *argc, char ***argv, __u32 *val, const char *what); 146 int prog_parse_fd(int *argc, char ***argv); 147 int map_parse_fd(int *argc, char ***argv); 148 int map_parse_fd_and_info(int *argc, char ***argv, void *info, __u32 *info_len); 149 150 #ifdef HAVE_LIBBFD_SUPPORT 151 void disasm_print_insn(unsigned char *image, ssize_t len, int opcodes, 152 const char *arch, const char *disassembler_options); 153 int disasm_init(void); 154 #else 155 static inline 156 void disasm_print_insn(unsigned char *image, ssize_t len, int opcodes, 157 const char *arch, const char *disassembler_options) 158 { 159 } 160 static inline int disasm_init(void) 161 { 162 p_err("No libbfd support"); 163 return -1; 164 } 165 #endif 166 void print_data_json(uint8_t *data, size_t len); 167 void print_hex_data_json(uint8_t *data, size_t len); 168 169 unsigned int get_page_size(void); 170 unsigned int get_possible_cpus(void); 171 const char * 172 ifindex_to_bfd_params(__u32 ifindex, __u64 ns_dev, __u64 ns_ino, 173 const char **opt); 174 175 struct btf_dumper { 176 const struct btf *btf; 177 json_writer_t *jw; 178 bool is_plain_text; 179 }; 180 181 /* btf_dumper_type - print data along with type information 182 * @d: an instance containing context for dumping types 183 * @type_id: index in btf->types array. this points to the type to be dumped 184 * @data: pointer the actual data, i.e. the values to be printed 185 * 186 * Returns zero on success and negative error code otherwise 187 */ 188 int btf_dumper_type(const struct btf_dumper *d, __u32 type_id, 189 const void *data); 190 void btf_dumper_type_only(const struct btf *btf, __u32 func_type_id, 191 char *func_only, int size); 192 193 struct nlattr; 194 struct ifinfomsg; 195 struct tcmsg; 196 int do_xdp_dump(struct ifinfomsg *ifinfo, struct nlattr **tb); 197 int do_filter_dump(struct tcmsg *ifinfo, struct nlattr **tb, const char *kind, 198 const char *devname, int ifindex); 199 #endif 200