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
9struct htab_op_ctx {
10	int fd;
11	int loop;
12	bool stop;
13};
14
15struct htab_val {
16	unsigned int lock;
17	unsigned int data;
18};
19
20static 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
37static 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
62void 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
94reap:
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