summaryrefslogtreecommitdiff
path: root/tools/testing/radix-tree/iteration_check_2.c
blob: aac5c50a3674cbd51fe2bf08745a8d7c8ce32ca6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * iteration_check_2.c: Check that deleting a tagged entry doesn't cause
 * an RCU walker to finish early.
 * Copyright (c) 2020 Oracle
 * Author: Matthew Wilcox <willy@infradead.org>
 */
#include <pthread.h>
#include "test.h"

static volatile bool test_complete;

static void *iterator(void *arg)
{
	XA_STATE(xas, arg, 0);
	void *entry;

	rcu_register_thread();

	while (!test_complete) {
		xas_set(&xas, 0);
		rcu_read_lock();
		xas_for_each_marked(&xas, entry, ULONG_MAX, XA_MARK_0)
			;
		rcu_read_unlock();
		assert(xas.xa_index >= 100);
	}

	rcu_unregister_thread();
	return NULL;
}

static void *throbber(void *arg)
{
	struct xarray *xa = arg;

	rcu_register_thread();

	while (!test_complete) {
		int i;

		for (i = 0; i < 100; i++) {
			xa_store(xa, i, xa_mk_value(i), GFP_KERNEL);
			xa_set_mark(xa, i, XA_MARK_0);
		}
		for (i = 0; i < 100; i++)
			xa_erase(xa, i);
	}

	rcu_unregister_thread();
	return NULL;
}

void iteration_test2(unsigned test_duration)
{
	pthread_t threads[2];
	DEFINE_XARRAY(array);
	int i;

	printv(1, "Running iteration test 2 for %d seconds\n", test_duration);

	test_complete = false;

	xa_store(&array, 100, xa_mk_value(100), GFP_KERNEL);
	xa_set_mark(&array, 100, XA_MARK_0);

	if (pthread_create(&threads[0], NULL, iterator, &array)) {
		perror("create iterator thread");
		exit(1);
	}
	if (pthread_create(&threads[1], NULL, throbber, &array)) {
		perror("create throbber thread");
		exit(1);
	}

	sleep(test_duration);
	test_complete = true;

	for (i = 0; i < 2; i++) {
		if (pthread_join(threads[i], NULL)) {
			perror("pthread_join");
			exit(1);
		}
	}

	xa_destroy(&array);
}