1 // SPDX-License-Identifier: GPL-2.0
2 #define _GNU_SOURCE
3 #include <errno.h>
4 #include <fcntl.h>
5 #include <sched.h>
6 #include <stdio.h>
7 #include <stdbool.h>
8 #include <sys/stat.h>
9 #include <sys/syscall.h>
10 #include <sys/types.h>
11 #include <sys/wait.h>
12 #include <time.h>
13 #include <unistd.h>
14 #include <string.h>
15 #include <pthread.h>
16
17 #include "log.h"
18 #include "timens.h"
19
20 #define OFFSET (36000)
21
22 struct thread_args {
23 char *tst_name;
24 struct timespec *now;
25 };
26
tcheck(void * _args)27 static void *tcheck(void *_args)
28 {
29 struct thread_args *args = _args;
30 struct timespec *now = args->now, tst;
31 int i;
32
33 for (i = 0; i < 2; i++) {
34 _gettime(CLOCK_MONOTONIC, &tst, i);
35 if (abs(tst.tv_sec - now->tv_sec) > 5) {
36 pr_fail("%s: in-thread: unexpected value: %ld (%ld)\n",
37 args->tst_name, tst.tv_sec, now->tv_sec);
38 return (void *)1UL;
39 }
40 }
41 return NULL;
42 }
43
check_in_thread(char * tst_name,struct timespec * now)44 static int check_in_thread(char *tst_name, struct timespec *now)
45 {
46 struct thread_args args = {
47 .tst_name = tst_name,
48 .now = now,
49 };
50 pthread_t th;
51 void *retval;
52
53 if (pthread_create(&th, NULL, tcheck, &args))
54 return pr_perror("thread");
55 if (pthread_join(th, &retval))
56 return pr_perror("pthread_join");
57 return !(retval == NULL);
58 }
59
check(char * tst_name,struct timespec * now)60 static int check(char *tst_name, struct timespec *now)
61 {
62 struct timespec tst;
63 int i;
64
65 for (i = 0; i < 2; i++) {
66 _gettime(CLOCK_MONOTONIC, &tst, i);
67 if (abs(tst.tv_sec - now->tv_sec) > 5)
68 return pr_fail("%s: unexpected value: %ld (%ld)\n",
69 tst_name, tst.tv_sec, now->tv_sec);
70 }
71 if (check_in_thread(tst_name, now))
72 return 1;
73 ksft_test_result_pass("%s\n", tst_name);
74 return 0;
75 }
76
main(int argc,char * argv[])77 int main(int argc, char *argv[])
78 {
79 struct timespec now;
80 int status;
81 pid_t pid;
82
83 if (argc > 1) {
84 char *endptr;
85
86 ksft_cnt.ksft_pass = 1;
87 now.tv_sec = strtoul(argv[1], &endptr, 0);
88 if (*endptr != 0)
89 return pr_perror("strtoul");
90
91 return check("child after exec", &now);
92 }
93
94 nscheck();
95
96 ksft_set_plan(4);
97
98 clock_gettime(CLOCK_MONOTONIC, &now);
99
100 if (unshare_timens())
101 return 1;
102
103 if (_settime(CLOCK_MONOTONIC, OFFSET))
104 return 1;
105
106 if (check("parent before vfork", &now))
107 return 1;
108
109 pid = vfork();
110 if (pid < 0)
111 return pr_perror("fork");
112
113 if (pid == 0) {
114 char now_str[64];
115 char *cargv[] = {"exec", now_str, NULL};
116 char *cenv[] = {NULL};
117
118 /* Check for proper vvar offsets after execve. */
119 snprintf(now_str, sizeof(now_str), "%ld", now.tv_sec + OFFSET);
120 execve("/proc/self/exe", cargv, cenv);
121 pr_perror("execve");
122 _exit(1);
123 }
124
125 if (waitpid(pid, &status, 0) != pid)
126 return pr_perror("waitpid");
127
128 if (status)
129 ksft_exit_fail();
130 ksft_inc_pass_cnt();
131 ksft_test_result_pass("wait for child\n");
132
133 /* Check that we are still in the source timens. */
134 if (check("parent after vfork", &now))
135 return 1;
136
137 ksft_exit_pass();
138 return 0;
139 }
140