samples/bpf: syscall_tp_user: Fix array out-of-bound access

[linux-2.6-microblaze.git] / samples / bpf / syscall_tp_user.c

diff --git a/samples/bpf/syscall_tp_user.c b/samples/bpf/syscall_tp_user.c
index 18c94c7..7a09ac7 100644 (file)
--- a/samples/bpf/syscall_tp_user.c
+++ b/samples/bpf/syscall_tp_user.c
@@ -48,7 +48,7 @@ static void verify_map(int map_id)
 static int test(char *filename, int nr_tests)
 {
        int map0_fds[nr_tests], map1_fds[nr_tests], fd, i, j = 0;
-       struct bpf_link *links[nr_tests * 4];
+       struct bpf_link **links = NULL;
        struct bpf_object *objs[nr_tests];
        struct bpf_program *prog;
 
@@ -60,6 +60,19 @@ static int test(char *filename, int nr_tests)
                        goto cleanup;
                }
 
+               /* One-time initialization */
+               if (!links) {
+                       int nr_progs = 0;
+
+                       bpf_object__for_each_program(prog, objs[i])
+                               nr_progs += 1;
+
+                       links = calloc(nr_progs * nr_tests, sizeof(struct bpf_link *));
+
+                       if (!links)
+                               goto cleanup;
+               }
+
                /* load BPF program */
                if (bpf_object__load(objs[i])) {
                        fprintf(stderr, "loading BPF object file failed\n");
@@ -107,8 +120,12 @@ static int test(char *filename, int nr_tests)
        }
 
 cleanup:
-       for (j--; j >= 0; j--)
-               bpf_link__destroy(links[j]);
+       if (links) {
+               for (j--; j >= 0; j--)
+                       bpf_link__destroy(links[j]);
+
+               free(links);
+       }
 
        for (i--; i >= 0; i--)
                bpf_object__close(objs[i]);