1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (C) 2023. Huawei Technologies Co., Ltd */
5 #include <sys/syscall.h>
6 #include <test_progs.h>
8 #include "access_map_in_map.skel.h"
11 pthread_barrier_t barrier;
17 static int wait_for_start_or_abort(struct thread_ctx *ctx)
19 while (!ctx->start && !ctx->abort)
21 return ctx->abort ? -1 : 0;
24 static void *update_map_fn(void *data)
26 struct thread_ctx *ctx = data;
27 int loop = ctx->loop, err = 0;
29 if (wait_for_start_or_abort(ctx) < 0)
31 pthread_barrier_wait(&ctx->barrier);
36 fd = bpf_map_create(BPF_MAP_TYPE_ARRAY, NULL, 4, 4, 1, NULL);
39 pthread_barrier_wait(&ctx->barrier);
43 /* Remove the old inner map */
44 if (bpf_map_update_elem(ctx->outer_map_fd, &zero, &fd, 0) < 0)
47 pthread_barrier_wait(&ctx->barrier);
55 static void *access_map_fn(void *data)
57 struct thread_ctx *ctx = data;
60 if (wait_for_start_or_abort(ctx) < 0)
62 pthread_barrier_wait(&ctx->barrier);
65 /* Access the old inner map */
67 pthread_barrier_wait(&ctx->barrier);
73 static void test_map_in_map_access(const char *prog_name, const char *map_name)
75 struct access_map_in_map *skel;
76 struct bpf_map *outer_map;
77 struct bpf_program *prog;
78 struct thread_ctx ctx;
82 skel = access_map_in_map__open();
83 if (!ASSERT_OK_PTR(skel, "access_map_in_map open"))
86 prog = bpf_object__find_program_by_name(skel->obj, prog_name);
87 if (!ASSERT_OK_PTR(prog, "find program"))
89 bpf_program__set_autoload(prog, true);
91 outer_map = bpf_object__find_map_by_name(skel->obj, map_name);
92 if (!ASSERT_OK_PTR(outer_map, "find map"))
95 err = access_map_in_map__load(skel);
96 if (!ASSERT_OK(err, "access_map_in_map load"))
99 err = access_map_in_map__attach(skel);
100 if (!ASSERT_OK(err, "access_map_in_map attach"))
103 skel->bss->tgid = getpid();
105 memset(&ctx, 0, sizeof(ctx));
106 pthread_barrier_init(&ctx.barrier, NULL, 2);
107 ctx.outer_map_fd = bpf_map__fd(outer_map);
110 err = pthread_create(&tid[0], NULL, update_map_fn, &ctx);
111 if (!ASSERT_OK(err, "close_thread"))
114 err = pthread_create(&tid[1], NULL, access_map_fn, &ctx);
115 if (!ASSERT_OK(err, "read_thread")) {
117 pthread_join(tid[0], NULL);
122 pthread_join(tid[0], NULL);
123 pthread_join(tid[1], NULL);
125 ASSERT_OK(ctx.err, "err");
127 access_map_in_map__destroy(skel);
130 void test_map_in_map(void)
132 if (test__start_subtest("acc_map_in_array"))
133 test_map_in_map_access("access_map_in_array", "outer_array_map");
134 if (test__start_subtest("sleepable_acc_map_in_array"))
135 test_map_in_map_access("sleepable_access_map_in_array", "outer_array_map");
136 if (test__start_subtest("acc_map_in_htab"))
137 test_map_in_map_access("access_map_in_htab", "outer_htab_map");
138 if (test__start_subtest("sleepable_acc_map_in_htab"))
139 test_map_in_map_access("sleepable_access_map_in_htab", "outer_htab_map");