mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2026-01-12 09:32:12 +00:00
3528647874
Allow the number of threads for the thloop workload to be increased beyond the normal 2. Add error checking to the parsed time and thread count values. Signed-off-by: Ian Rogers <irogers@google.com> Acked-by: Arnaldo Carvalho de Melo <acme@redhat.com> Signed-off-by: Namhyung Kim <namhyung@kernel.org>
83 lines
1.5 KiB
C
83 lines
1.5 KiB
C
/* 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);
|