162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci#define _GNU_SOURCE
362306a36Sopenharmony_ci#include <sched.h>
462306a36Sopenharmony_ci
562306a36Sopenharmony_ci#include <linux/unistd.h>
662306a36Sopenharmony_ci#include <linux/futex.h>
762306a36Sopenharmony_ci#include <stdio.h>
862306a36Sopenharmony_ci#include <string.h>
962306a36Sopenharmony_ci#include <sys/syscall.h>
1062306a36Sopenharmony_ci#include <sys/types.h>
1162306a36Sopenharmony_ci#include <sys/wait.h>
1262306a36Sopenharmony_ci#include <time.h>
1362306a36Sopenharmony_ci#include <unistd.h>
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#include "log.h"
1662306a36Sopenharmony_ci#include "timens.h"
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci#define NSEC_PER_SEC 1000000000ULL
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_cistatic int run_test(int clockid)
2162306a36Sopenharmony_ci{
2262306a36Sopenharmony_ci	int futex_op = FUTEX_WAIT_BITSET;
2362306a36Sopenharmony_ci	struct timespec timeout, end;
2462306a36Sopenharmony_ci	int val = 0;
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci	if (clockid == CLOCK_REALTIME)
2762306a36Sopenharmony_ci		futex_op |= FUTEX_CLOCK_REALTIME;
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_ci	clock_gettime(clockid, &timeout);
3062306a36Sopenharmony_ci	timeout.tv_nsec += NSEC_PER_SEC / 10; // 100ms
3162306a36Sopenharmony_ci	if (timeout.tv_nsec > NSEC_PER_SEC) {
3262306a36Sopenharmony_ci		timeout.tv_sec++;
3362306a36Sopenharmony_ci		timeout.tv_nsec -= NSEC_PER_SEC;
3462306a36Sopenharmony_ci	}
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	if (syscall(__NR_futex, &val, futex_op, 0,
3762306a36Sopenharmony_ci		    &timeout, 0, FUTEX_BITSET_MATCH_ANY) >= 0) {
3862306a36Sopenharmony_ci		ksft_test_result_fail("futex didn't return ETIMEDOUT\n");
3962306a36Sopenharmony_ci		return 1;
4062306a36Sopenharmony_ci	}
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_ci	if (errno != ETIMEDOUT) {
4362306a36Sopenharmony_ci		ksft_test_result_fail("futex didn't return ETIMEDOUT: %s\n",
4462306a36Sopenharmony_ci							strerror(errno));
4562306a36Sopenharmony_ci		return 1;
4662306a36Sopenharmony_ci	}
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	clock_gettime(clockid, &end);
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	if (end.tv_sec < timeout.tv_sec ||
5162306a36Sopenharmony_ci	    (end.tv_sec == timeout.tv_sec && end.tv_nsec < timeout.tv_nsec)) {
5262306a36Sopenharmony_ci		ksft_test_result_fail("futex slept less than 100ms\n");
5362306a36Sopenharmony_ci		return 1;
5462306a36Sopenharmony_ci	}
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	ksft_test_result_pass("futex with the %d clockid\n", clockid);
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	return 0;
6062306a36Sopenharmony_ci}
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ciint main(int argc, char *argv[])
6362306a36Sopenharmony_ci{
6462306a36Sopenharmony_ci	int status, len, fd;
6562306a36Sopenharmony_ci	char buf[4096];
6662306a36Sopenharmony_ci	pid_t pid;
6762306a36Sopenharmony_ci	struct timespec mtime_now;
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	nscheck();
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_ci	ksft_set_plan(2);
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	clock_gettime(CLOCK_MONOTONIC, &mtime_now);
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	if (unshare_timens())
7662306a36Sopenharmony_ci		return 1;
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	len = snprintf(buf, sizeof(buf), "%d %d 0",
7962306a36Sopenharmony_ci			CLOCK_MONOTONIC, 70 * 24 * 3600);
8062306a36Sopenharmony_ci	fd = open("/proc/self/timens_offsets", O_WRONLY);
8162306a36Sopenharmony_ci	if (fd < 0)
8262306a36Sopenharmony_ci		return pr_perror("/proc/self/timens_offsets");
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci	if (write(fd, buf, len) != len)
8562306a36Sopenharmony_ci		return pr_perror("/proc/self/timens_offsets");
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	close(fd);
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci	pid = fork();
9062306a36Sopenharmony_ci	if (pid < 0)
9162306a36Sopenharmony_ci		return pr_perror("Unable to fork");
9262306a36Sopenharmony_ci	if (pid == 0) {
9362306a36Sopenharmony_ci		int ret = 0;
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci		ret |= run_test(CLOCK_REALTIME);
9662306a36Sopenharmony_ci		ret |= run_test(CLOCK_MONOTONIC);
9762306a36Sopenharmony_ci		if (ret)
9862306a36Sopenharmony_ci			ksft_exit_fail();
9962306a36Sopenharmony_ci		ksft_exit_pass();
10062306a36Sopenharmony_ci		return 0;
10162306a36Sopenharmony_ci	}
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci	if (waitpid(pid, &status, 0) != pid)
10462306a36Sopenharmony_ci		return pr_perror("Unable to wait the child process");
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	if (WIFEXITED(status))
10762306a36Sopenharmony_ci		return WEXITSTATUS(status);
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	return 1;
11062306a36Sopenharmony_ci}
111