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
|
/* SPDX-License-Identifier: GPL-2.0 */
#include <pthread.h>
#include <stdlib.h>
#include <signal.h>
#include <unistd.h>
#include <linux/compiler.h>
#include "../tests.h"
static volatile sig_atomic_t done;
/* We want to check this symbol in perf report */
noinline void test_loop(void);
static void sighandler(int sig __maybe_unused)
{
done = 1;
}
noinline void test_loop(void)
{
while (!done);
}
static void *thfunc(void *arg)
{
void (*loop_fn)(void) = arg;
loop_fn();
return NULL;
}
static int thloop(int argc, const char **argv)
{
int nt = 2, sec = 1, err = 1;
pthread_t *thread_list = NULL;
if (argc > 0)
sec = atoi(argv[0]);
if (sec <= 0) {
fprintf(stderr, "Error: seconds (%d) must be >= 1\n", sec);
return 1;
}
if (argc > 1)
nt = atoi(argv[1]);
if (nt <= 0) {
fprintf(stderr, "Error: thread count (%d) must be >= 1\n", nt);
return 1;
}
signal(SIGINT, sighandler);
signal(SIGALRM, sighandler);
thread_list = calloc(nt, sizeof(pthread_t));
if (thread_list == NULL) {
fprintf(stderr, "Error: malloc failed for %d threads\n", nt);
goto out;
}
for (int i = 1; i < nt; i++) {
int ret = pthread_create(&thread_list[i], NULL, thfunc, test_loop);
if (ret) {
fprintf(stderr, "Error: failed to create thread %d\n", i);
done = 1; // Ensure started threads terminate.
goto out;
}
}
alarm(sec);
test_loop();
err = 0;
out:
for (int i = 1; i < nt; i++) {
if (thread_list && thread_list[i])
pthread_join(thread_list[i], /*retval=*/NULL);
}
free(thread_list);
return err;
}
DEFINE_WORKLOAD(thloop);
|
