142 lines · c
1// SPDX-License-Identifier: GPL-2.02/* Copyright (C) 2023. Huawei Technologies Co., Ltd */3#define _GNU_SOURCE4#include <unistd.h>5#include <sys/syscall.h>6#include <test_progs.h>7#include <bpf/btf.h>8#include "access_map_in_map.skel.h"9 10struct thread_ctx {11 pthread_barrier_t barrier;12 int outer_map_fd;13 int start, abort;14 int loop, err;15};16 17static int wait_for_start_or_abort(struct thread_ctx *ctx)18{19 while (!ctx->start && !ctx->abort)20 usleep(1);21 return ctx->abort ? -1 : 0;22}23 24static void *update_map_fn(void *data)25{26 struct thread_ctx *ctx = data;27 int loop = ctx->loop, err = 0;28 29 if (wait_for_start_or_abort(ctx) < 0)30 return NULL;31 pthread_barrier_wait(&ctx->barrier);32 33 while (loop-- > 0) {34 int fd, zero = 0;35 36 fd = bpf_map_create(BPF_MAP_TYPE_ARRAY, NULL, 4, 4, 1, NULL);37 if (fd < 0) {38 err |= 1;39 pthread_barrier_wait(&ctx->barrier);40 continue;41 }42 43 /* Remove the old inner map */44 if (bpf_map_update_elem(ctx->outer_map_fd, &zero, &fd, 0) < 0)45 err |= 2;46 close(fd);47 pthread_barrier_wait(&ctx->barrier);48 }49 50 ctx->err = err;51 52 return NULL;53}54 55static void *access_map_fn(void *data)56{57 struct thread_ctx *ctx = data;58 int loop = ctx->loop;59 60 if (wait_for_start_or_abort(ctx) < 0)61 return NULL;62 pthread_barrier_wait(&ctx->barrier);63 64 while (loop-- > 0) {65 /* Access the old inner map */66 syscall(SYS_getpgid);67 pthread_barrier_wait(&ctx->barrier);68 }69 70 return NULL;71}72 73static void test_map_in_map_access(const char *prog_name, const char *map_name)74{75 struct access_map_in_map *skel;76 struct bpf_map *outer_map;77 struct bpf_program *prog;78 struct thread_ctx ctx;79 pthread_t tid[2];80 int err;81 82 skel = access_map_in_map__open();83 if (!ASSERT_OK_PTR(skel, "access_map_in_map open"))84 return;85 86 prog = bpf_object__find_program_by_name(skel->obj, prog_name);87 if (!ASSERT_OK_PTR(prog, "find program"))88 goto out;89 bpf_program__set_autoload(prog, true);90 91 outer_map = bpf_object__find_map_by_name(skel->obj, map_name);92 if (!ASSERT_OK_PTR(outer_map, "find map"))93 goto out;94 95 err = access_map_in_map__load(skel);96 if (!ASSERT_OK(err, "access_map_in_map load"))97 goto out;98 99 err = access_map_in_map__attach(skel);100 if (!ASSERT_OK(err, "access_map_in_map attach"))101 goto out;102 103 skel->bss->tgid = getpid();104 105 memset(&ctx, 0, sizeof(ctx));106 pthread_barrier_init(&ctx.barrier, NULL, 2);107 ctx.outer_map_fd = bpf_map__fd(outer_map);108 ctx.loop = 4;109 110 err = pthread_create(&tid[0], NULL, update_map_fn, &ctx);111 if (!ASSERT_OK(err, "close_thread"))112 goto out;113 114 err = pthread_create(&tid[1], NULL, access_map_fn, &ctx);115 if (!ASSERT_OK(err, "read_thread")) {116 ctx.abort = 1;117 pthread_join(tid[0], NULL);118 goto out;119 }120 121 ctx.start = 1;122 pthread_join(tid[0], NULL);123 pthread_join(tid[1], NULL);124 125 ASSERT_OK(ctx.err, "err");126out:127 access_map_in_map__destroy(skel);128}129 130void test_map_in_map(void)131{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");140}141 142