Commit ec202509 authored by Paul Chaignon's avatar Paul Chaignon Committed by Alexei Starovoitov

bpftool: Match several programs with same tag

When several BPF programs have the same tag, bpftool matches only the
first (in ID order).  This patch changes that behavior such that dump and
show commands return all matched programs.  Commands that require a single
program (e.g., pin and attach) will error out if given a tag that matches
several.  bpftool prog dump will also error out if file or visual are
given and several programs have the given tag.

In the case of the dump command, a program header is added before each
dump only if the tag matches several programs; this patch doesn't change
the output if a single program matches.  The output when several
programs match thus looks as follows.

$ ./bpftool prog dump xlated tag 6deef7357e7b4530
3: cgroup_skb  tag 6deef7357e7b4530  gpl
   0: (bf) r6 = r1
   [...]
   7: (95) exit

4: cgroup_skb  tag 6deef7357e7b4530  gpl
   0: (bf) r6 = r1
   [...]
   7: (95) exit
Signed-off-by: default avatarPaul Chaignon <paul.chaignon@orange.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
Link: https://lore.kernel.org/bpf/fb1fe943202659a69cd21dd5b907c205af1e1e22.1576263640.git.paul.chaignon@gmail.com
parent a06bf42f
...@@ -53,8 +53,10 @@ DESCRIPTION ...@@ -53,8 +53,10 @@ DESCRIPTION
=========== ===========
**bpftool prog { show | list }** [*PROG*] **bpftool prog { show | list }** [*PROG*]
Show information about loaded programs. If *PROG* is Show information about loaded programs. If *PROG* is
specified show information only about given program, otherwise specified show information only about given programs,
list all programs currently loaded on the system. otherwise list all programs currently loaded on the system.
In case of **tag**, *PROG* may match several programs which
will all be shown.
Output will start with program ID followed by program type and Output will start with program ID followed by program type and
zero or more named attributes (depending on kernel version). zero or more named attributes (depending on kernel version).
...@@ -68,11 +70,15 @@ DESCRIPTION ...@@ -68,11 +70,15 @@ DESCRIPTION
performed via the **kernel.bpf_stats_enabled** sysctl knob. performed via the **kernel.bpf_stats_enabled** sysctl knob.
**bpftool prog dump xlated** *PROG* [{ **file** *FILE* | **opcodes** | **visual** | **linum** }] **bpftool prog dump xlated** *PROG* [{ **file** *FILE* | **opcodes** | **visual** | **linum** }]
Dump eBPF instructions of the program from the kernel. By Dump eBPF instructions of the programs from the kernel. By
default, eBPF will be disassembled and printed to standard default, eBPF will be disassembled and printed to standard
output in human-readable format. In this case, **opcodes** output in human-readable format. In this case, **opcodes**
controls if raw opcodes should be printed as well. controls if raw opcodes should be printed as well.
In case of **tag**, *PROG* may match several programs which
will all be dumped. However, if **file** or **visual** is
specified, *PROG* must match a single program.
If **file** is specified, the binary image will instead be If **file** is specified, the binary image will instead be
written to *FILE*. written to *FILE*.
...@@ -80,15 +86,17 @@ DESCRIPTION ...@@ -80,15 +86,17 @@ DESCRIPTION
built instead, and eBPF instructions will be presented with built instead, and eBPF instructions will be presented with
CFG in DOT format, on standard output. CFG in DOT format, on standard output.
If the prog has line_info available, the source line will If the programs have line_info available, the source line will
be displayed by default. If **linum** is specified, be displayed by default. If **linum** is specified,
the filename, line number and line column will also be the filename, line number and line column will also be
displayed on top of the source line. displayed on top of the source line.
**bpftool prog dump jited** *PROG* [{ **file** *FILE* | **opcodes** | **linum** }] **bpftool prog dump jited** *PROG* [{ **file** *FILE* | **opcodes** | **linum** }]
Dump jited image (host machine code) of the program. Dump jited image (host machine code) of the program.
If *FILE* is specified image will be written to a file, If *FILE* is specified image will be written to a file,
otherwise it will be disassembled and printed to stdout. otherwise it will be disassembled and printed to stdout.
*PROG* must match a single program when **file** is specified.
**opcodes** controls if raw opcodes will be printed. **opcodes** controls if raw opcodes will be printed.
......
...@@ -25,6 +25,11 @@ ...@@ -25,6 +25,11 @@
#include "main.h" #include "main.h"
#include "xlated_dumper.h" #include "xlated_dumper.h"
enum dump_mode {
DUMP_JITED,
DUMP_XLATED,
};
static const char * const attach_type_strings[] = { static const char * const attach_type_strings[] = {
[BPF_SK_SKB_STREAM_PARSER] = "stream_parser", [BPF_SK_SKB_STREAM_PARSER] = "stream_parser",
[BPF_SK_SKB_STREAM_VERDICT] = "stream_verdict", [BPF_SK_SKB_STREAM_VERDICT] = "stream_verdict",
...@@ -77,11 +82,12 @@ static void print_boot_time(__u64 nsecs, char *buf, unsigned int size) ...@@ -77,11 +82,12 @@ static void print_boot_time(__u64 nsecs, char *buf, unsigned int size)
strftime(buf, size, "%FT%T%z", &load_tm); strftime(buf, size, "%FT%T%z", &load_tm);
} }
static int prog_fd_by_tag(unsigned char *tag) static int prog_fd_by_tag(unsigned char *tag, int **fds)
{ {
unsigned int id = 0; unsigned int id = 0;
int fd, nb_fds = 0;
void *tmp;
int err; int err;
int fd;
while (true) { while (true) {
struct bpf_prog_info info = {}; struct bpf_prog_info info = {};
...@@ -89,36 +95,53 @@ static int prog_fd_by_tag(unsigned char *tag) ...@@ -89,36 +95,53 @@ static int prog_fd_by_tag(unsigned char *tag)
err = bpf_prog_get_next_id(id, &id); err = bpf_prog_get_next_id(id, &id);
if (err) { if (err) {
p_err("%s", strerror(errno)); if (errno != ENOENT) {
return -1; p_err("%s", strerror(errno));
goto err_close_fds;
}
return nb_fds;
} }
fd = bpf_prog_get_fd_by_id(id); fd = bpf_prog_get_fd_by_id(id);
if (fd < 0) { if (fd < 0) {
p_err("can't get prog by id (%u): %s", p_err("can't get prog by id (%u): %s",
id, strerror(errno)); id, strerror(errno));
return -1; goto err_close_fds;
} }
err = bpf_obj_get_info_by_fd(fd, &info, &len); err = bpf_obj_get_info_by_fd(fd, &info, &len);
if (err) { if (err) {
p_err("can't get prog info (%u): %s", p_err("can't get prog info (%u): %s",
id, strerror(errno)); id, strerror(errno));
close(fd); goto err_close_fd;
return -1;
} }
if (!memcmp(tag, info.tag, BPF_TAG_SIZE)) if (memcmp(tag, info.tag, BPF_TAG_SIZE)) {
return fd; close(fd);
continue;
}
close(fd); if (nb_fds > 0) {
tmp = realloc(*fds, (nb_fds + 1) * sizeof(int));
if (!tmp) {
p_err("failed to realloc");
goto err_close_fd;
}
*fds = tmp;
}
(*fds)[nb_fds++] = fd;
} }
err_close_fd:
close(fd);
err_close_fds:
while (--nb_fds >= 0)
close((*fds)[nb_fds]);
return -1;
} }
int prog_parse_fd(int *argc, char ***argv) static int prog_parse_fds(int *argc, char ***argv, int **fds)
{ {
int fd;
if (is_prefix(**argv, "id")) { if (is_prefix(**argv, "id")) {
unsigned int id; unsigned int id;
char *endptr; char *endptr;
...@@ -132,10 +155,12 @@ int prog_parse_fd(int *argc, char ***argv) ...@@ -132,10 +155,12 @@ int prog_parse_fd(int *argc, char ***argv)
} }
NEXT_ARGP(); NEXT_ARGP();
fd = bpf_prog_get_fd_by_id(id); (*fds)[0] = bpf_prog_get_fd_by_id(id);
if (fd < 0) if ((*fds)[0] < 0) {
p_err("get by id (%u): %s", id, strerror(errno)); p_err("get by id (%u): %s", id, strerror(errno));
return fd; return -1;
}
return 1;
} else if (is_prefix(**argv, "tag")) { } else if (is_prefix(**argv, "tag")) {
unsigned char tag[BPF_TAG_SIZE]; unsigned char tag[BPF_TAG_SIZE];
...@@ -149,7 +174,7 @@ int prog_parse_fd(int *argc, char ***argv) ...@@ -149,7 +174,7 @@ int prog_parse_fd(int *argc, char ***argv)
} }
NEXT_ARGP(); NEXT_ARGP();
return prog_fd_by_tag(tag); return prog_fd_by_tag(tag, fds);
} else if (is_prefix(**argv, "pinned")) { } else if (is_prefix(**argv, "pinned")) {
char *path; char *path;
...@@ -158,13 +183,43 @@ int prog_parse_fd(int *argc, char ***argv) ...@@ -158,13 +183,43 @@ int prog_parse_fd(int *argc, char ***argv)
path = **argv; path = **argv;
NEXT_ARGP(); NEXT_ARGP();
return open_obj_pinned_any(path, BPF_OBJ_PROG); (*fds)[0] = open_obj_pinned_any(path, BPF_OBJ_PROG);
if ((*fds)[0] < 0)
return -1;
return 1;
} }
p_err("expected 'id', 'tag' or 'pinned', got: '%s'?", **argv); p_err("expected 'id', 'tag' or 'pinned', got: '%s'?", **argv);
return -1; return -1;
} }
int prog_parse_fd(int *argc, char ***argv)
{
int *fds = NULL;
int nb_fds, fd;
fds = malloc(sizeof(int));
if (!fds) {
p_err("mem alloc failed");
return -1;
}
nb_fds = prog_parse_fds(argc, argv, &fds);
if (nb_fds != 1) {
if (nb_fds > 1) {
p_err("several programs match this handle");
while (nb_fds--)
close(fds[nb_fds]);
}
fd = -1;
goto exit_free;
}
fd = fds[0];
exit_free:
free(fds);
return fd;
}
static void show_prog_maps(int fd, u32 num_maps) static void show_prog_maps(int fd, u32 num_maps)
{ {
struct bpf_prog_info info = {}; struct bpf_prog_info info = {};
...@@ -194,11 +249,8 @@ static void show_prog_maps(int fd, u32 num_maps) ...@@ -194,11 +249,8 @@ static void show_prog_maps(int fd, u32 num_maps)
} }
} }
static void print_prog_json(struct bpf_prog_info *info, int fd) static void print_prog_header_json(struct bpf_prog_info *info)
{ {
char *memlock;
jsonw_start_object(json_wtr);
jsonw_uint_field(json_wtr, "id", info->id); jsonw_uint_field(json_wtr, "id", info->id);
if (info->type < ARRAY_SIZE(prog_type_name)) if (info->type < ARRAY_SIZE(prog_type_name))
jsonw_string_field(json_wtr, "type", jsonw_string_field(json_wtr, "type",
...@@ -219,7 +271,14 @@ static void print_prog_json(struct bpf_prog_info *info, int fd) ...@@ -219,7 +271,14 @@ static void print_prog_json(struct bpf_prog_info *info, int fd)
jsonw_uint_field(json_wtr, "run_time_ns", info->run_time_ns); jsonw_uint_field(json_wtr, "run_time_ns", info->run_time_ns);
jsonw_uint_field(json_wtr, "run_cnt", info->run_cnt); jsonw_uint_field(json_wtr, "run_cnt", info->run_cnt);
} }
}
static void print_prog_json(struct bpf_prog_info *info, int fd)
{
char *memlock;
jsonw_start_object(json_wtr);
print_prog_header_json(info);
print_dev_json(info->ifindex, info->netns_dev, info->netns_ino); print_dev_json(info->ifindex, info->netns_dev, info->netns_ino);
if (info->load_time) { if (info->load_time) {
...@@ -268,10 +327,8 @@ static void print_prog_json(struct bpf_prog_info *info, int fd) ...@@ -268,10 +327,8 @@ static void print_prog_json(struct bpf_prog_info *info, int fd)
jsonw_end_object(json_wtr); jsonw_end_object(json_wtr);
} }
static void print_prog_plain(struct bpf_prog_info *info, int fd) static void print_prog_header_plain(struct bpf_prog_info *info)
{ {
char *memlock;
printf("%u: ", info->id); printf("%u: ", info->id);
if (info->type < ARRAY_SIZE(prog_type_name)) if (info->type < ARRAY_SIZE(prog_type_name))
printf("%s ", prog_type_name[info->type]); printf("%s ", prog_type_name[info->type]);
...@@ -289,6 +346,13 @@ static void print_prog_plain(struct bpf_prog_info *info, int fd) ...@@ -289,6 +346,13 @@ static void print_prog_plain(struct bpf_prog_info *info, int fd)
printf(" run_time_ns %lld run_cnt %lld", printf(" run_time_ns %lld run_cnt %lld",
info->run_time_ns, info->run_cnt); info->run_time_ns, info->run_cnt);
printf("\n"); printf("\n");
}
static void print_prog_plain(struct bpf_prog_info *info, int fd)
{
char *memlock;
print_prog_header_plain(info);
if (info->load_time) { if (info->load_time) {
char buf[32]; char buf[32];
...@@ -349,6 +413,40 @@ static int show_prog(int fd) ...@@ -349,6 +413,40 @@ static int show_prog(int fd)
return 0; return 0;
} }
static int do_show_subset(int argc, char **argv)
{
int *fds = NULL;
int nb_fds, i;
int err = -1;
fds = malloc(sizeof(int));
if (!fds) {
p_err("mem alloc failed");
return -1;
}
nb_fds = prog_parse_fds(&argc, &argv, &fds);
if (nb_fds < 1)
goto exit_free;
if (json_output && nb_fds > 1)
jsonw_start_array(json_wtr); /* root array */
for (i = 0; i < nb_fds; i++) {
err = show_prog(fds[i]);
if (err) {
for (; i < nb_fds; i++)
close(fds[i]);
break;
}
close(fds[i]);
}
if (json_output && nb_fds > 1)
jsonw_end_array(json_wtr); /* root array */
exit_free:
free(fds);
return err;
}
static int do_show(int argc, char **argv) static int do_show(int argc, char **argv)
{ {
__u32 id = 0; __u32 id = 0;
...@@ -358,15 +456,8 @@ static int do_show(int argc, char **argv) ...@@ -358,15 +456,8 @@ static int do_show(int argc, char **argv)
if (show_pinned) if (show_pinned)
build_pinned_obj_table(&prog_table, BPF_OBJ_PROG); build_pinned_obj_table(&prog_table, BPF_OBJ_PROG);
if (argc == 2) { if (argc == 2)
fd = prog_parse_fd(&argc, &argv); return do_show_subset(argc, argv);
if (fd < 0)
return -1;
err = show_prog(fd);
close(fd);
return err;
}
if (argc) if (argc)
return BAD_ARG(); return BAD_ARG();
...@@ -408,101 +499,32 @@ static int do_show(int argc, char **argv) ...@@ -408,101 +499,32 @@ static int do_show(int argc, char **argv)
return err; return err;
} }
static int do_dump(int argc, char **argv) static int
prog_dump(struct bpf_prog_info *info, enum dump_mode mode,
char *filepath, bool opcodes, bool visual, bool linum)
{ {
struct bpf_prog_info_linear *info_linear;
struct bpf_prog_linfo *prog_linfo = NULL; struct bpf_prog_linfo *prog_linfo = NULL;
enum {DUMP_JITED, DUMP_XLATED} mode;
const char *disasm_opt = NULL; const char *disasm_opt = NULL;
struct bpf_prog_info *info;
struct dump_data dd = {}; struct dump_data dd = {};
void *func_info = NULL; void *func_info = NULL;
struct btf *btf = NULL; struct btf *btf = NULL;
char *filepath = NULL;
bool opcodes = false;
bool visual = false;
char func_sig[1024]; char func_sig[1024];
unsigned char *buf; unsigned char *buf;
bool linum = false;
__u32 member_len; __u32 member_len;
__u64 arrays;
ssize_t n; ssize_t n;
int fd; int fd;
if (is_prefix(*argv, "jited")) {
if (disasm_init())
return -1;
mode = DUMP_JITED;
} else if (is_prefix(*argv, "xlated")) {
mode = DUMP_XLATED;
} else {
p_err("expected 'xlated' or 'jited', got: %s", *argv);
return -1;
}
NEXT_ARG();
if (argc < 2)
usage();
fd = prog_parse_fd(&argc, &argv);
if (fd < 0)
return -1;
if (is_prefix(*argv, "file")) {
NEXT_ARG();
if (!argc) {
p_err("expected file path");
return -1;
}
filepath = *argv;
NEXT_ARG();
} else if (is_prefix(*argv, "opcodes")) {
opcodes = true;
NEXT_ARG();
} else if (is_prefix(*argv, "visual")) {
visual = true;
NEXT_ARG();
} else if (is_prefix(*argv, "linum")) {
linum = true;
NEXT_ARG();
}
if (argc) {
usage();
return -1;
}
if (mode == DUMP_JITED)
arrays = 1UL << BPF_PROG_INFO_JITED_INSNS;
else
arrays = 1UL << BPF_PROG_INFO_XLATED_INSNS;
arrays |= 1UL << BPF_PROG_INFO_JITED_KSYMS;
arrays |= 1UL << BPF_PROG_INFO_JITED_FUNC_LENS;
arrays |= 1UL << BPF_PROG_INFO_FUNC_INFO;
arrays |= 1UL << BPF_PROG_INFO_LINE_INFO;
arrays |= 1UL << BPF_PROG_INFO_JITED_LINE_INFO;
info_linear = bpf_program__get_prog_info_linear(fd, arrays);
close(fd);
if (IS_ERR_OR_NULL(info_linear)) {
p_err("can't get prog info: %s", strerror(errno));
return -1;
}
info = &info_linear->info;
if (mode == DUMP_JITED) { if (mode == DUMP_JITED) {
if (info->jited_prog_len == 0) { if (info->jited_prog_len == 0) {
p_info("no instructions returned"); p_info("no instructions returned");
goto err_free; return -1;
} }
buf = (unsigned char *)(info->jited_prog_insns); buf = (unsigned char *)(info->jited_prog_insns);
member_len = info->jited_prog_len; member_len = info->jited_prog_len;
} else { /* DUMP_XLATED */ } else { /* DUMP_XLATED */
if (info->xlated_prog_len == 0) { if (info->xlated_prog_len == 0) {
p_err("error retrieving insn dump: kernel.kptr_restrict set?"); p_err("error retrieving insn dump: kernel.kptr_restrict set?");
goto err_free; return -1;
} }
buf = (unsigned char *)info->xlated_prog_insns; buf = (unsigned char *)info->xlated_prog_insns;
member_len = info->xlated_prog_len; member_len = info->xlated_prog_len;
...@@ -510,7 +532,7 @@ static int do_dump(int argc, char **argv) ...@@ -510,7 +532,7 @@ static int do_dump(int argc, char **argv)
if (info->btf_id && btf__get_from_id(info->btf_id, &btf)) { if (info->btf_id && btf__get_from_id(info->btf_id, &btf)) {
p_err("failed to get btf"); p_err("failed to get btf");
goto err_free; return -1;
} }
func_info = (void *)info->func_info; func_info = (void *)info->func_info;
...@@ -526,7 +548,7 @@ static int do_dump(int argc, char **argv) ...@@ -526,7 +548,7 @@ static int do_dump(int argc, char **argv)
if (fd < 0) { if (fd < 0) {
p_err("can't open file %s: %s", filepath, p_err("can't open file %s: %s", filepath,
strerror(errno)); strerror(errno));
goto err_free; return -1;
} }
n = write(fd, buf, member_len); n = write(fd, buf, member_len);
...@@ -534,7 +556,7 @@ static int do_dump(int argc, char **argv) ...@@ -534,7 +556,7 @@ static int do_dump(int argc, char **argv)
if (n != member_len) { if (n != member_len) {
p_err("error writing output file: %s", p_err("error writing output file: %s",
n < 0 ? strerror(errno) : "short write"); n < 0 ? strerror(errno) : "short write");
goto err_free; return -1;
} }
if (json_output) if (json_output)
...@@ -548,7 +570,7 @@ static int do_dump(int argc, char **argv) ...@@ -548,7 +570,7 @@ static int do_dump(int argc, char **argv)
info->netns_ino, info->netns_ino,
&disasm_opt); &disasm_opt);
if (!name) if (!name)
goto err_free; return -1;
} }
if (info->nr_jited_func_lens && info->jited_func_lens) { if (info->nr_jited_func_lens && info->jited_func_lens) {
...@@ -643,12 +665,130 @@ static int do_dump(int argc, char **argv) ...@@ -643,12 +665,130 @@ static int do_dump(int argc, char **argv)
kernel_syms_destroy(&dd); kernel_syms_destroy(&dd);
} }
free(info_linear);
return 0; return 0;
}
err_free: static int do_dump(int argc, char **argv)
free(info_linear); {
return -1; struct bpf_prog_info_linear *info_linear;
char *filepath = NULL;
bool opcodes = false;
bool visual = false;
enum dump_mode mode;
bool linum = false;
int *fds = NULL;
int nb_fds, i = 0;
int err = -1;
__u64 arrays;
if (is_prefix(*argv, "jited")) {
if (disasm_init())
return -1;
mode = DUMP_JITED;
} else if (is_prefix(*argv, "xlated")) {
mode = DUMP_XLATED;
} else {
p_err("expected 'xlated' or 'jited', got: %s", *argv);
return -1;
}
NEXT_ARG();
if (argc < 2)
usage();
fds = malloc(sizeof(int));
if (!fds) {
p_err("mem alloc failed");
return -1;
}
nb_fds = prog_parse_fds(&argc, &argv, &fds);
if (nb_fds < 1)
goto exit_free;
if (is_prefix(*argv, "file")) {
NEXT_ARG();
if (!argc) {
p_err("expected file path");
goto exit_close;
}
if (nb_fds > 1) {
p_err("several programs matched");
goto exit_close;
}
filepath = *argv;
NEXT_ARG();
} else if (is_prefix(*argv, "opcodes")) {
opcodes = true;
NEXT_ARG();
} else if (is_prefix(*argv, "visual")) {
if (nb_fds > 1) {
p_err("several programs matched");
goto exit_close;
}
visual = true;
NEXT_ARG();
} else if (is_prefix(*argv, "linum")) {
linum = true;
NEXT_ARG();
}
if (argc) {
usage();
goto exit_close;
}
if (mode == DUMP_JITED)
arrays = 1UL << BPF_PROG_INFO_JITED_INSNS;
else
arrays = 1UL << BPF_PROG_INFO_XLATED_INSNS;
arrays |= 1UL << BPF_PROG_INFO_JITED_KSYMS;
arrays |= 1UL << BPF_PROG_INFO_JITED_FUNC_LENS;
arrays |= 1UL << BPF_PROG_INFO_FUNC_INFO;
arrays |= 1UL << BPF_PROG_INFO_LINE_INFO;
arrays |= 1UL << BPF_PROG_INFO_JITED_LINE_INFO;
if (json_output && nb_fds > 1)
jsonw_start_array(json_wtr); /* root array */
for (i = 0; i < nb_fds; i++) {
info_linear = bpf_program__get_prog_info_linear(fds[i], arrays);
if (IS_ERR_OR_NULL(info_linear)) {
p_err("can't get prog info: %s", strerror(errno));
break;
}
if (json_output && nb_fds > 1) {
jsonw_start_object(json_wtr); /* prog object */
print_prog_header_json(&info_linear->info);
jsonw_name(json_wtr, "insns");
} else if (nb_fds > 1) {
print_prog_header_plain(&info_linear->info);
}
err = prog_dump(&info_linear->info, mode, filepath, opcodes,
visual, linum);
if (json_output && nb_fds > 1)
jsonw_end_object(json_wtr); /* prog object */
else if (i != nb_fds - 1 && nb_fds > 1)
printf("\n");
free(info_linear);
if (err)
break;
close(fds[i]);
}
if (json_output && nb_fds > 1)
jsonw_end_array(json_wtr); /* root array */
exit_close:
for (; i < nb_fds; i++)
close(fds[i]);
exit_free:
free(fds);
return err;
} }
static int do_pin(int argc, char **argv) static int do_pin(int argc, char **argv)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment