samples/bpf: test both pre-alloc and normal maps

extend test coveraged to include pre-allocated and run-time alloc maps

Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 9 insertions, 1 deletions

diff --git a/samples/bpf/test_maps.c b/samples/bpf/test_maps.c
index 7bd9edd02d9b..47bf0858f9e4 100644
--- a/samples/bpf/test_maps.c
+++ b/samples/bpf/test_maps.c
@@ -468,7 +468,7 @@ static void test_map_parallel(void)
         assert(bpf_get_next_key(map_fd, &key, &key) == -1 && errno == ENOENT);
 }
 
-int main(void)
+static void run_all_tests(void)
 {
         test_hashmap_sanity(0, NULL);
         test_percpu_hashmap_sanity(0, NULL);
@@ -479,6 +479,14 @@ int main(void)
         test_map_large();
         test_map_parallel();
         test_map_stress();
+}
+
+int main(void)
+{
+        map_flags = 0;
+        run_all_tests();
+        map_flags = BPF_F_NO_PREALLOC;
+        run_all_tests();
         printf("test_maps: OK\n");
         return 0;
 }