1 // SPDX-License-Identifier: GPL-2.0
2 #define _GNU_SOURCE
3 #include <linux/membarrier.h>
4 #include <syscall.h>
5 #include <stdio.h>
6 #include <errno.h>
7 #include <string.h>
8 #include <pthread.h>
9 
10 #include "membarrier_test_impl.h"
11 
12 static int thread_ready, thread_quit;
13 static pthread_mutex_t test_membarrier_thread_mutex =
14 	PTHREAD_MUTEX_INITIALIZER;
15 static pthread_cond_t test_membarrier_thread_cond =
16 	PTHREAD_COND_INITIALIZER;
17 
18 void *test_membarrier_thread(void *arg)
19 {
20 	pthread_mutex_lock(&test_membarrier_thread_mutex);
21 	thread_ready = 1;
22 	pthread_cond_broadcast(&test_membarrier_thread_cond);
23 	pthread_mutex_unlock(&test_membarrier_thread_mutex);
24 
25 	pthread_mutex_lock(&test_membarrier_thread_mutex);
26 	while (!thread_quit)
27 		pthread_cond_wait(&test_membarrier_thread_cond,
28 				  &test_membarrier_thread_mutex);
29 	pthread_mutex_unlock(&test_membarrier_thread_mutex);
30 
31 	return NULL;
32 }
33 
34 static int test_mt_membarrier(void)
35 {
36 	int i;
37 	pthread_t test_thread;
38 
39 	pthread_create(&test_thread, NULL,
40 		       test_membarrier_thread, NULL);
41 
42 	pthread_mutex_lock(&test_membarrier_thread_mutex);
43 	while (!thread_ready)
44 		pthread_cond_wait(&test_membarrier_thread_cond,
45 				  &test_membarrier_thread_mutex);
46 	pthread_mutex_unlock(&test_membarrier_thread_mutex);
47 
48 	test_membarrier_fail();
49 
50 	test_membarrier_success();
51 
52 	pthread_mutex_lock(&test_membarrier_thread_mutex);
53 	thread_quit = 1;
54 	pthread_cond_broadcast(&test_membarrier_thread_cond);
55 	pthread_mutex_unlock(&test_membarrier_thread_mutex);
56 
57 	pthread_join(test_thread, NULL);
58 
59 	return 0;
60 }
61 
62 int main(int argc, char **argv)
63 {
64 	ksft_print_header();
65 	ksft_set_plan(16);
66 
67 	test_membarrier_query();
68 
69 	/* Multi-threaded */
70 	test_mt_membarrier();
71 
72 	return ksft_exit_pass();
73 }
74