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 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 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 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 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