Skip .text section in bpf_program__next, so it iterates
only throught the probes code. We are about to add .text
code support, so the .text section needs to get separated
from probes.

Link: http://lkml.kernel.org/n/tip-lcftc9k0fby5b5g2o0lhu...@git.kernel.org
Signed-off-by: Jiri Olsa <jo...@kernel.org>
---
 tools/lib/bpf/libbpf.c | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/tools/lib/bpf/libbpf.c b/tools/lib/bpf/libbpf.c
index 6b9df10470e8..07a6d8f5e5ab 100644
--- a/tools/lib/bpf/libbpf.c
+++ b/tools/lib/bpf/libbpf.c
@@ -1705,22 +1705,34 @@ void *bpf_object__priv(struct bpf_object *obj)
 struct bpf_program *
 bpf_program__next(struct bpf_program *prev, struct bpf_object *obj)
 {
+       struct bpf_program *prog;
        size_t idx;
 
        if (!obj->programs)
                return NULL;
+
        /* First handler */
        if (prev == NULL)
-               return &obj->programs[0];
+               idx = 0;
+       else
+               idx = (prev - obj->programs) + 1;
 
-       if (prev->obj != obj) {
+       if (prev && (prev->obj != obj)) {
                pr_warning("error: program handler doesn't match object\n");
                return NULL;
        }
 
-       idx = (prev - obj->programs) + 1;
        if (idx >= obj->nr_programs)
                return NULL;
+
+       prog = &obj->programs[idx];
+
+       if (prog->idx == obj->efile.text_shndx)
+               idx++;
+
+       if (idx >= obj->nr_programs)
+               return NULL;
+
        return &obj->programs[idx];
 }
 
-- 
2.13.6

Reply via email to