bpf: libbpf: bpftool: Print bpf_line_info during prog dump

This patch adds print bpf_line_info function in 'prog dump jitted'
and 'prog dump xlated':

[root@arch-fb-vm1 bpf]# ~/devshare/fb-kernel/linux/tools/bpf/bpftool/bpftool prog dump jited pinned /sys/fs/bpf/test_btf_haskv
[...]
int test_long_fname_2(struct dummy_tracepoint_args * arg):
bpf_prog_44a040bf25481309_test_long_fname_2:
; static int test_long_fname_2(struct dummy_tracepoint_args *arg)
   0:	push   %rbp
   1:	mov    %rsp,%rbp
   4:	sub    $0x30,%rsp
   b:	sub    $0x28,%rbp
   f:	mov    %rbx,0x0(%rbp)
  13:	mov    %r13,0x8(%rbp)
  17:	mov    %r14,0x10(%rbp)
  1b:	mov    %r15,0x18(%rbp)
  1f:	xor    %eax,%eax
  21:	mov    %rax,0x20(%rbp)
  25:	xor    %esi,%esi
; int key = 0;
  27:	mov    %esi,-0x4(%rbp)
; if (!arg->sock)
  2a:	mov    0x8(%rdi),%rdi
; if (!arg->sock)
  2e:	cmp    $0x0,%rdi
  32:	je     0x0000000000000070
  34:	mov    %rbp,%rsi
; counts = bpf_map_lookup_elem(&btf_map, &key);
  37:	add    $0xfffffffffffffffc,%rsi
  3b:	movabs $0xffff8881139d7480,%rdi
  45:	add    $0x110,%rdi
  4c:	mov    0x0(%rsi),%eax
  4f:	cmp    $0x4,%rax
  53:	jae    0x000000000000005e
  55:	shl    $0x3,%rax
  59:	add    %rdi,%rax
  5c:	jmp    0x0000000000000060
  5e:	xor    %eax,%eax
; if (!counts)
  60:	cmp    $0x0,%rax
  64:	je     0x0000000000000070
; counts->v6++;
  66:	mov    0x4(%rax),%edi
  69:	add    $0x1,%rdi
  6d:	mov    %edi,0x4(%rax)
  70:	mov    0x0(%rbp),%rbx
  74:	mov    0x8(%rbp),%r13
  78:	mov    0x10(%rbp),%r14
  7c:	mov    0x18(%rbp),%r15
  80:	add    $0x28,%rbp
  84:	leaveq
  85:	retq
[...]

With linum:
[root@arch-fb-vm1 bpf]# ~/devshare/fb-kernel/linux/tools/bpf/bpftool/bpftool prog dump jited pinned /sys/fs/bpf/test_btf_haskv linum
int _dummy_tracepoint(struct dummy_tracepoint_args * arg):
bpf_prog_b07ccb89267cf242__dummy_tracepoint:
; return test_long_fname_1(arg); [file:/data/users/kafai/fb-kernel/linux/tools/testing/selftests/bpf/test_btf_haskv.c line_num:54 line_col:9]
   0:	push   %rbp
   1:	mov    %rsp,%rbp
   4:	sub    $0x28,%rsp
   b:	sub    $0x28,%rbp
   f:	mov    %rbx,0x0(%rbp)
  13:	mov    %r13,0x8(%rbp)
  17:	mov    %r14,0x10(%rbp)
  1b:	mov    %r15,0x18(%rbp)
  1f:	xor    %eax,%eax
  21:	mov    %rax,0x20(%rbp)
  25:	callq  0x000000000000851e
; return test_long_fname_1(arg); [file:/data/users/kafai/fb-kernel/linux/tools/testing/selftests/bpf/test_btf_haskv.c line_num:54 line_col:2]
  2a:	xor    %eax,%eax
  2c:	mov    0x0(%rbp),%rbx
  30:	mov    0x8(%rbp),%r13
  34:	mov    0x10(%rbp),%r14
  38:	mov    0x18(%rbp),%r15
  3c:	add    $0x28,%rbp
  40:	leaveq
  41:	retq
[...]

Signed-off-by: Martin KaFai Lau <kafai@fb.com>
Acked-by: Yonghong Song <yhs@fb.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>

1 files changed, 90 insertions, 10 deletions

diff --git a/tools/bpf/bpftool/prog.c b/tools/bpf/bpftool/prog.c
index a9a51123454c..65b921ffd10a 100644
--- a/tools/bpf/bpftool/prog.c
+++ b/tools/bpf/bpftool/prog.c
@@ -423,24 +423,26 @@ static int do_show(int argc, char **argv)
 
 static int do_dump(int argc, char **argv)
 {
+        unsigned int finfo_rec_size, linfo_rec_size, jited_linfo_rec_size;
+        void *func_info = NULL, *linfo = NULL, *jited_linfo = NULL;
+        unsigned int finfo_cnt, linfo_cnt = 0, jited_linfo_cnt = 0;
+        struct bpf_prog_linfo *prog_linfo = NULL;
         unsigned long *func_ksyms = NULL;
         struct bpf_prog_info info = {};
         unsigned int *func_lens = NULL;
         const char *disasm_opt = NULL;
-        unsigned int finfo_rec_size;
         unsigned int nr_func_ksyms;
         unsigned int nr_func_lens;
         struct dump_data dd = {};
         __u32 len = sizeof(info);
         struct btf *btf = NULL;
-        void *func_info = NULL;
-        unsigned int finfo_cnt;
         unsigned int buf_size;
         char *filepath = NULL;
         bool opcodes = false;
         bool visual = false;
         char func_sig[1024];
         unsigned char *buf;
+        bool linum = false;
         __u32 *member_len;
         __u64 *member_ptr;
         ssize_t n;
@@ -484,6 +486,9 @@ static int do_dump(int argc, char **argv)
         } else if (is_prefix(*argv, "visual")) {
                 visual = true;
                 NEXT_ARG();
+        } else if (is_prefix(*argv, "linum")) {
+                linum = true;
+                NEXT_ARG();
         }
 
         if (argc) {
@@ -543,6 +548,32 @@ static int do_dump(int argc, char **argv)
                 }
         }
 
+        linfo_rec_size = info.line_info_rec_size;
+        if (info.line_info_cnt && linfo_rec_size && info.btf_id) {
+                linfo_cnt = info.line_info_cnt;
+                linfo = malloc(linfo_cnt * linfo_rec_size);
+                if (!linfo) {
+                        p_err("mem alloc failed");
+                        close(fd);
+                        goto err_free;
+                }
+        }
+
+        jited_linfo_rec_size = info.jited_line_info_rec_size;
+        if (info.jited_line_info_cnt &&
+            jited_linfo_rec_size &&
+            info.nr_jited_ksyms &&
+            info.nr_jited_func_lens &&
+            info.btf_id) {
+                jited_linfo_cnt = info.jited_line_info_cnt;
+                jited_linfo = malloc(jited_linfo_cnt * jited_linfo_rec_size);
+                if (!jited_linfo) {
+                        p_err("mem alloc failed");
+                        close(fd);
+                        goto err_free;
+                }
+        }
+
         memset(&info, 0, sizeof(info));
 
         *member_ptr = ptr_to_u64(buf);
@@ -554,6 +585,13 @@ static int do_dump(int argc, char **argv)
         info.func_info_cnt = finfo_cnt;
         info.func_info_rec_size = finfo_rec_size;
         info.func_info = ptr_to_u64(func_info);
+        info.line_info_cnt = linfo_cnt;
+        info.line_info_rec_size = linfo_rec_size;
+        info.line_info = ptr_to_u64(linfo);
+        info.jited_line_info_cnt = jited_linfo_cnt;
+        info.jited_line_info_rec_size = jited_linfo_rec_size;
+        info.jited_line_info = ptr_to_u64(jited_linfo);
+
 
         err = bpf_obj_get_info_by_fd(fd, &info, &len);
         close(fd);
@@ -596,6 +634,30 @@ static int do_dump(int argc, char **argv)
                 finfo_cnt = 0;
         }
 
+        if (linfo && info.line_info_cnt != linfo_cnt) {
+                p_err("incorrect line_info_cnt %u vs. expected %u",
+                      info.line_info_cnt, linfo_cnt);
+                goto err_free;
+        }
+
+        if (info.line_info_rec_size != linfo_rec_size) {
+                p_err("incorrect line_info_rec_size %u vs. expected %u",
+                      info.line_info_rec_size, linfo_rec_size);
+                goto err_free;
+        }
+
+        if (jited_linfo && info.jited_line_info_cnt != jited_linfo_cnt) {
+                p_err("incorrect jited_line_info_cnt %u vs. expected %u",
+                      info.jited_line_info_cnt, jited_linfo_cnt);
+                goto err_free;
+        }
+
+        if (info.jited_line_info_rec_size != jited_linfo_rec_size) {
+                p_err("incorrect jited_line_info_rec_size %u vs. expected %u",
+                      info.jited_line_info_rec_size, jited_linfo_rec_size);
+                goto err_free;
+        }
+
         if ((member_len == &info.jited_prog_len &&
              info.jited_prog_insns == 0) ||
             (member_len == &info.xlated_prog_len &&
@@ -609,6 +671,12 @@ static int do_dump(int argc, char **argv)
                 goto err_free;
         }
 
+        if (linfo_cnt) {
+                prog_linfo = bpf_prog_linfo__new(&info);
+                if (!prog_linfo)
+                        p_err("error in processing bpf_line_info.  continue without it.");
+        }
+
         if (filepath) {
                 fd = open(filepath, O_WRONLY | O_CREAT | O_TRUNC, 0600);
                 if (fd < 0) {
@@ -690,8 +758,11 @@ static int do_dump(int argc, char **argv)
                                         printf("%s:\n", sym_name);
                                 }
 
-                                disasm_print_insn(img, lens[i], opcodes, name,
-                                                  disasm_opt);
+                                disasm_print_insn(img, lens[i], opcodes,
+                                                  name, disasm_opt, btf,
+                                                  prog_linfo, ksyms[i], i,
+                                                  linum);
+
                                 img += lens[i];
 
                                 if (json_output)
@@ -704,7 +775,7 @@ static int do_dump(int argc, char **argv)
                                 jsonw_end_array(json_wtr);
                 } else {
                         disasm_print_insn(buf, *member_len, opcodes, name,
-                                          disasm_opt);
+                                          disasm_opt, btf, NULL, 0, 0, false);
                 }
         } else if (visual) {
                 if (json_output)
@@ -718,11 +789,14 @@ static int do_dump(int argc, char **argv)
                 dd.btf = btf;
                 dd.func_info = func_info;
                 dd.finfo_rec_size = finfo_rec_size;
+                dd.prog_linfo = prog_linfo;
 
                 if (json_output)
-                        dump_xlated_json(&dd, buf, *member_len, opcodes);
+                        dump_xlated_json(&dd, buf, *member_len, opcodes,
+                                         linum);
                 else
-                        dump_xlated_plain(&dd, buf, *member_len, opcodes);
+                        dump_xlated_plain(&dd, buf, *member_len, opcodes,
+                                          linum);
                 kernel_syms_destroy(&dd);
         }
 
@@ -730,6 +804,9 @@ static int do_dump(int argc, char **argv)
         free(func_ksyms);
         free(func_lens);
         free(func_info);
+        free(linfo);
+        free(jited_linfo);
+        bpf_prog_linfo__free(prog_linfo);
         return 0;
 
 err_free:
@@ -737,6 +814,9 @@ err_free:
         free(func_ksyms);
         free(func_lens);
         free(func_info);
+        free(linfo);
+        free(jited_linfo);
+        bpf_prog_linfo__free(prog_linfo);
         return -1;
 }
 
@@ -1138,8 +1218,8 @@ static int do_help(int argc, char **argv)
 
         fprintf(stderr,
                 "Usage: %s %s { show | list } [PROG]\n"
-                "       %s %s dump xlated PROG [{ file FILE | opcodes | visual }]\n"
-                "       %s %s dump jited  PROG [{ file FILE | opcodes }]\n"
+                "       %s %s dump xlated PROG [{ file FILE | opcodes | visual | linum }]\n"
+                "       %s %s dump jited  PROG [{ file FILE | opcodes | linum }]\n"
                 "       %s %s pin   PROG FILE\n"
                 "       %s %s { load | loadall } OBJ  PATH \\\n"
                 "                         [type TYPE] [dev NAME] \\\n"