162306a36Sopenharmony_ci/* SPDX-License-Identifier: GPL-2.0 */
262306a36Sopenharmony_ci#include <math.h>
362306a36Sopenharmony_ci#include <signal.h>
462306a36Sopenharmony_ci#include <stdlib.h>
562306a36Sopenharmony_ci#include <unistd.h>
662306a36Sopenharmony_ci#include <linux/compiler.h>
762306a36Sopenharmony_ci#include <sys/wait.h>
862306a36Sopenharmony_ci#include "../tests.h"
962306a36Sopenharmony_ci
1062306a36Sopenharmony_cistatic volatile sig_atomic_t done;
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_cistatic void sighandler(int sig __maybe_unused)
1362306a36Sopenharmony_ci{
1462306a36Sopenharmony_ci	done = 1;
1562306a36Sopenharmony_ci}
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_cistatic int __sqrtloop(int sec)
1862306a36Sopenharmony_ci{
1962306a36Sopenharmony_ci	signal(SIGALRM, sighandler);
2062306a36Sopenharmony_ci	alarm(sec);
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ci	while (!done)
2362306a36Sopenharmony_ci		(void)sqrt(rand());
2462306a36Sopenharmony_ci	return 0;
2562306a36Sopenharmony_ci}
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_cistatic int sqrtloop(int argc, const char **argv)
2862306a36Sopenharmony_ci{
2962306a36Sopenharmony_ci	int sec = 1;
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci	if (argc > 0)
3262306a36Sopenharmony_ci		sec = atoi(argv[0]);
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci	switch (fork()) {
3562306a36Sopenharmony_ci	case 0:
3662306a36Sopenharmony_ci		return __sqrtloop(sec);
3762306a36Sopenharmony_ci	case -1:
3862306a36Sopenharmony_ci		return -1;
3962306a36Sopenharmony_ci	default:
4062306a36Sopenharmony_ci		wait(NULL);
4162306a36Sopenharmony_ci	}
4262306a36Sopenharmony_ci	return 0;
4362306a36Sopenharmony_ci}
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ciDEFINE_WORKLOAD(sqrtloop);
46