18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
28c2ecf20Sopenharmony_ci#define _GNU_SOURCE
38c2ecf20Sopenharmony_ci#include <sched.h>
48c2ecf20Sopenharmony_ci
58c2ecf20Sopenharmony_ci#include <linux/unistd.h>
68c2ecf20Sopenharmony_ci#include <linux/futex.h>
78c2ecf20Sopenharmony_ci#include <stdio.h>
88c2ecf20Sopenharmony_ci#include <string.h>
98c2ecf20Sopenharmony_ci#include <sys/syscall.h>
108c2ecf20Sopenharmony_ci#include <sys/types.h>
118c2ecf20Sopenharmony_ci#include <sys/wait.h>
128c2ecf20Sopenharmony_ci#include <time.h>
138c2ecf20Sopenharmony_ci#include <unistd.h>
148c2ecf20Sopenharmony_ci
158c2ecf20Sopenharmony_ci#include "log.h"
168c2ecf20Sopenharmony_ci#include "timens.h"
178c2ecf20Sopenharmony_ci
188c2ecf20Sopenharmony_ci#define NSEC_PER_SEC 1000000000ULL
198c2ecf20Sopenharmony_ci
208c2ecf20Sopenharmony_cistatic int run_test(int clockid)
218c2ecf20Sopenharmony_ci{
228c2ecf20Sopenharmony_ci	int futex_op = FUTEX_WAIT_BITSET;
238c2ecf20Sopenharmony_ci	struct timespec timeout, end;
248c2ecf20Sopenharmony_ci	int val = 0;
258c2ecf20Sopenharmony_ci
268c2ecf20Sopenharmony_ci	if (clockid == CLOCK_REALTIME)
278c2ecf20Sopenharmony_ci		futex_op |= FUTEX_CLOCK_REALTIME;
288c2ecf20Sopenharmony_ci
298c2ecf20Sopenharmony_ci	clock_gettime(clockid, &timeout);
308c2ecf20Sopenharmony_ci	timeout.tv_nsec += NSEC_PER_SEC / 10; // 100ms
318c2ecf20Sopenharmony_ci	if (timeout.tv_nsec > NSEC_PER_SEC) {
328c2ecf20Sopenharmony_ci		timeout.tv_sec++;
338c2ecf20Sopenharmony_ci		timeout.tv_nsec -= NSEC_PER_SEC;
348c2ecf20Sopenharmony_ci	}
358c2ecf20Sopenharmony_ci
368c2ecf20Sopenharmony_ci	if (syscall(__NR_futex, &val, futex_op, 0,
378c2ecf20Sopenharmony_ci		    &timeout, 0, FUTEX_BITSET_MATCH_ANY) >= 0) {
388c2ecf20Sopenharmony_ci		ksft_test_result_fail("futex didn't return ETIMEDOUT\n");
398c2ecf20Sopenharmony_ci		return 1;
408c2ecf20Sopenharmony_ci	}
418c2ecf20Sopenharmony_ci
428c2ecf20Sopenharmony_ci	if (errno != ETIMEDOUT) {
438c2ecf20Sopenharmony_ci		ksft_test_result_fail("futex didn't return ETIMEDOUT: %s\n",
448c2ecf20Sopenharmony_ci							strerror(errno));
458c2ecf20Sopenharmony_ci		return 1;
468c2ecf20Sopenharmony_ci	}
478c2ecf20Sopenharmony_ci
488c2ecf20Sopenharmony_ci	clock_gettime(clockid, &end);
498c2ecf20Sopenharmony_ci
508c2ecf20Sopenharmony_ci	if (end.tv_sec < timeout.tv_sec ||
518c2ecf20Sopenharmony_ci	    (end.tv_sec == timeout.tv_sec && end.tv_nsec < timeout.tv_nsec)) {
528c2ecf20Sopenharmony_ci		ksft_test_result_fail("futex slept less than 100ms\n");
538c2ecf20Sopenharmony_ci		return 1;
548c2ecf20Sopenharmony_ci	}
558c2ecf20Sopenharmony_ci
568c2ecf20Sopenharmony_ci
578c2ecf20Sopenharmony_ci	ksft_test_result_pass("futex with the %d clockid\n", clockid);
588c2ecf20Sopenharmony_ci
598c2ecf20Sopenharmony_ci	return 0;
608c2ecf20Sopenharmony_ci}
618c2ecf20Sopenharmony_ci
628c2ecf20Sopenharmony_ciint main(int argc, char *argv[])
638c2ecf20Sopenharmony_ci{
648c2ecf20Sopenharmony_ci	int status, len, fd;
658c2ecf20Sopenharmony_ci	char buf[4096];
668c2ecf20Sopenharmony_ci	pid_t pid;
678c2ecf20Sopenharmony_ci	struct timespec mtime_now;
688c2ecf20Sopenharmony_ci
698c2ecf20Sopenharmony_ci	nscheck();
708c2ecf20Sopenharmony_ci
718c2ecf20Sopenharmony_ci	ksft_set_plan(2);
728c2ecf20Sopenharmony_ci
738c2ecf20Sopenharmony_ci	clock_gettime(CLOCK_MONOTONIC, &mtime_now);
748c2ecf20Sopenharmony_ci
758c2ecf20Sopenharmony_ci	if (unshare_timens())
768c2ecf20Sopenharmony_ci		return 1;
778c2ecf20Sopenharmony_ci
788c2ecf20Sopenharmony_ci	len = snprintf(buf, sizeof(buf), "%d %d 0",
798c2ecf20Sopenharmony_ci			CLOCK_MONOTONIC, 70 * 24 * 3600);
808c2ecf20Sopenharmony_ci	fd = open("/proc/self/timens_offsets", O_WRONLY);
818c2ecf20Sopenharmony_ci	if (fd < 0)
828c2ecf20Sopenharmony_ci		return pr_perror("/proc/self/timens_offsets");
838c2ecf20Sopenharmony_ci
848c2ecf20Sopenharmony_ci	if (write(fd, buf, len) != len)
858c2ecf20Sopenharmony_ci		return pr_perror("/proc/self/timens_offsets");
868c2ecf20Sopenharmony_ci
878c2ecf20Sopenharmony_ci	close(fd);
888c2ecf20Sopenharmony_ci
898c2ecf20Sopenharmony_ci	pid = fork();
908c2ecf20Sopenharmony_ci	if (pid < 0)
918c2ecf20Sopenharmony_ci		return pr_perror("Unable to fork");
928c2ecf20Sopenharmony_ci	if (pid == 0) {
938c2ecf20Sopenharmony_ci		int ret = 0;
948c2ecf20Sopenharmony_ci
958c2ecf20Sopenharmony_ci		ret |= run_test(CLOCK_REALTIME);
968c2ecf20Sopenharmony_ci		ret |= run_test(CLOCK_MONOTONIC);
978c2ecf20Sopenharmony_ci		if (ret)
988c2ecf20Sopenharmony_ci			ksft_exit_fail();
998c2ecf20Sopenharmony_ci		ksft_exit_pass();
1008c2ecf20Sopenharmony_ci		return 0;
1018c2ecf20Sopenharmony_ci	}
1028c2ecf20Sopenharmony_ci
1038c2ecf20Sopenharmony_ci	if (waitpid(pid, &status, 0) != pid)
1048c2ecf20Sopenharmony_ci		return pr_perror("Unable to wait the child process");
1058c2ecf20Sopenharmony_ci
1068c2ecf20Sopenharmony_ci	if (WIFEXITED(status))
1078c2ecf20Sopenharmony_ci		return WEXITSTATUS(status);
1088c2ecf20Sopenharmony_ci
1098c2ecf20Sopenharmony_ci	return 1;
1108c2ecf20Sopenharmony_ci}
111