1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (C) 2023. Huawei Technologies Co., Ltd */
3 #define _GNU_SOURCE
4 #include <sched.h>
5 #include <stdbool.h>
6 #include <test_progs.h>
7 #include "htab_reuse.skel.h"
8
9 struct htab_op_ctx {
10 int fd;
11 int loop;
12 bool stop;
13 };
14
15 struct htab_val {
16 unsigned int lock;
17 unsigned int data;
18 };
19
htab_lookup_fn(void * arg)20 static void *htab_lookup_fn(void *arg)
21 {
22 struct htab_op_ctx *ctx = arg;
23 int i = 0;
24
25 while (i++ < ctx->loop && !ctx->stop) {
26 struct htab_val value;
27 unsigned int key;
28
29 /* Use BPF_F_LOCK to use spin-lock in map value. */
30 key = 7;
31 bpf_map_lookup_elem_flags(ctx->fd, &key, &value, BPF_F_LOCK);
32 }
33
34 return NULL;
35 }
36
htab_update_fn(void * arg)37 static void *htab_update_fn(void *arg)
38 {
39 struct htab_op_ctx *ctx = arg;
40 int i = 0;
41
42 while (i++ < ctx->loop && !ctx->stop) {
43 struct htab_val value;
44 unsigned int key;
45
46 key = 7;
47 value.lock = 0;
48 value.data = key;
49 bpf_map_update_elem(ctx->fd, &key, &value, BPF_F_LOCK);
50 bpf_map_delete_elem(ctx->fd, &key);
51
52 key = 24;
53 value.lock = 0;
54 value.data = key;
55 bpf_map_update_elem(ctx->fd, &key, &value, BPF_F_LOCK);
56 bpf_map_delete_elem(ctx->fd, &key);
57 }
58
59 return NULL;
60 }
61
test_htab_reuse(void)62 void test_htab_reuse(void)
63 {
64 unsigned int i, wr_nr = 1, rd_nr = 4;
65 pthread_t tids[wr_nr + rd_nr];
66 struct htab_reuse *skel;
67 struct htab_op_ctx ctx;
68 int err;
69
70 skel = htab_reuse__open_and_load();
71 if (!ASSERT_OK_PTR(skel, "htab_reuse__open_and_load"))
72 return;
73
74 ctx.fd = bpf_map__fd(skel->maps.htab);
75 ctx.loop = 500;
76 ctx.stop = false;
77
78 memset(tids, 0, sizeof(tids));
79 for (i = 0; i < wr_nr; i++) {
80 err = pthread_create(&tids[i], NULL, htab_update_fn, &ctx);
81 if (!ASSERT_OK(err, "pthread_create")) {
82 ctx.stop = true;
83 goto reap;
84 }
85 }
86 for (i = 0; i < rd_nr; i++) {
87 err = pthread_create(&tids[i + wr_nr], NULL, htab_lookup_fn, &ctx);
88 if (!ASSERT_OK(err, "pthread_create")) {
89 ctx.stop = true;
90 goto reap;
91 }
92 }
93
94 reap:
95 for (i = 0; i < wr_nr + rd_nr; i++) {
96 if (!tids[i])
97 continue;
98 pthread_join(tids[i], NULL);
99 }
100 htab_reuse__destroy(skel);
101 }
102