11854b97eSAndrei Vagin // SPDX-License-Identifier: GPL-2.0
21854b97eSAndrei Vagin #define _GNU_SOURCE
31854b97eSAndrei Vagin #include <sys/types.h>
41854b97eSAndrei Vagin #include <sys/stat.h>
51854b97eSAndrei Vagin #include <errno.h>
61854b97eSAndrei Vagin #include <fcntl.h>
71854b97eSAndrei Vagin #include <sched.h>
81854b97eSAndrei Vagin #include <time.h>
91854b97eSAndrei Vagin #include <stdio.h>
101854b97eSAndrei Vagin #include <unistd.h>
111854b97eSAndrei Vagin #include <sys/syscall.h>
121854b97eSAndrei Vagin #include <dlfcn.h>
131854b97eSAndrei Vagin
141854b97eSAndrei Vagin #include "log.h"
151854b97eSAndrei Vagin #include "timens.h"
161854b97eSAndrei Vagin
171854b97eSAndrei Vagin typedef int (*vgettime_t)(clockid_t, struct timespec *);
181854b97eSAndrei Vagin
191854b97eSAndrei Vagin vgettime_t vdso_clock_gettime;
201854b97eSAndrei Vagin
fill_function_pointers(void)211854b97eSAndrei Vagin static void fill_function_pointers(void)
221854b97eSAndrei Vagin {
231854b97eSAndrei Vagin void *vdso = dlopen("linux-vdso.so.1",
241854b97eSAndrei Vagin RTLD_LAZY | RTLD_LOCAL | RTLD_NOLOAD);
251854b97eSAndrei Vagin if (!vdso)
261854b97eSAndrei Vagin vdso = dlopen("linux-gate.so.1",
271854b97eSAndrei Vagin RTLD_LAZY | RTLD_LOCAL | RTLD_NOLOAD);
28*f56607e8SChristophe Leroy if (!vdso)
29*f56607e8SChristophe Leroy vdso = dlopen("linux-vdso32.so.1",
30*f56607e8SChristophe Leroy RTLD_LAZY | RTLD_LOCAL | RTLD_NOLOAD);
31*f56607e8SChristophe Leroy if (!vdso)
32*f56607e8SChristophe Leroy vdso = dlopen("linux-vdso64.so.1",
33*f56607e8SChristophe Leroy RTLD_LAZY | RTLD_LOCAL | RTLD_NOLOAD);
341854b97eSAndrei Vagin if (!vdso) {
351854b97eSAndrei Vagin pr_err("[WARN]\tfailed to find vDSO\n");
361854b97eSAndrei Vagin return;
371854b97eSAndrei Vagin }
381854b97eSAndrei Vagin
391854b97eSAndrei Vagin vdso_clock_gettime = (vgettime_t)dlsym(vdso, "__vdso_clock_gettime");
401854b97eSAndrei Vagin if (!vdso_clock_gettime)
41*f56607e8SChristophe Leroy vdso_clock_gettime = (vgettime_t)dlsym(vdso, "__kernel_clock_gettime");
42*f56607e8SChristophe Leroy if (!vdso_clock_gettime)
431854b97eSAndrei Vagin pr_err("Warning: failed to find clock_gettime in vDSO\n");
441854b97eSAndrei Vagin
451854b97eSAndrei Vagin }
461854b97eSAndrei Vagin
test(clock_t clockid,char * clockstr,bool in_ns)471854b97eSAndrei Vagin static void test(clock_t clockid, char *clockstr, bool in_ns)
481854b97eSAndrei Vagin {
491854b97eSAndrei Vagin struct timespec tp, start;
501854b97eSAndrei Vagin long i = 0;
511854b97eSAndrei Vagin const int timeout = 3;
521854b97eSAndrei Vagin
531854b97eSAndrei Vagin vdso_clock_gettime(clockid, &start);
541854b97eSAndrei Vagin tp = start;
551854b97eSAndrei Vagin for (tp = start; start.tv_sec + timeout > tp.tv_sec ||
561854b97eSAndrei Vagin (start.tv_sec + timeout == tp.tv_sec &&
571854b97eSAndrei Vagin start.tv_nsec > tp.tv_nsec); i++) {
581854b97eSAndrei Vagin vdso_clock_gettime(clockid, &tp);
591854b97eSAndrei Vagin }
601854b97eSAndrei Vagin
611854b97eSAndrei Vagin ksft_test_result_pass("%s:\tclock: %10s\tcycles:\t%10ld\n",
621854b97eSAndrei Vagin in_ns ? "ns" : "host", clockstr, i);
631854b97eSAndrei Vagin }
641854b97eSAndrei Vagin
main(int argc,char * argv[])651854b97eSAndrei Vagin int main(int argc, char *argv[])
661854b97eSAndrei Vagin {
671854b97eSAndrei Vagin time_t offset = 10;
681854b97eSAndrei Vagin int nsfd;
691854b97eSAndrei Vagin
701854b97eSAndrei Vagin ksft_set_plan(8);
711854b97eSAndrei Vagin
721854b97eSAndrei Vagin fill_function_pointers();
731854b97eSAndrei Vagin
741854b97eSAndrei Vagin test(CLOCK_MONOTONIC, "monotonic", false);
751854b97eSAndrei Vagin test(CLOCK_MONOTONIC_COARSE, "monotonic-coarse", false);
761854b97eSAndrei Vagin test(CLOCK_MONOTONIC_RAW, "monotonic-raw", false);
771854b97eSAndrei Vagin test(CLOCK_BOOTTIME, "boottime", false);
781854b97eSAndrei Vagin
791854b97eSAndrei Vagin nscheck();
801854b97eSAndrei Vagin
811854b97eSAndrei Vagin if (unshare_timens())
821854b97eSAndrei Vagin return 1;
831854b97eSAndrei Vagin
841854b97eSAndrei Vagin nsfd = open("/proc/self/ns/time_for_children", O_RDONLY);
851854b97eSAndrei Vagin if (nsfd < 0)
861854b97eSAndrei Vagin return pr_perror("Can't open a time namespace");
871854b97eSAndrei Vagin
881854b97eSAndrei Vagin if (_settime(CLOCK_MONOTONIC, offset))
891854b97eSAndrei Vagin return 1;
901854b97eSAndrei Vagin if (_settime(CLOCK_BOOTTIME, offset))
911854b97eSAndrei Vagin return 1;
921854b97eSAndrei Vagin
931854b97eSAndrei Vagin if (setns(nsfd, CLONE_NEWTIME))
941854b97eSAndrei Vagin return pr_perror("setns");
951854b97eSAndrei Vagin
961854b97eSAndrei Vagin test(CLOCK_MONOTONIC, "monotonic", true);
971854b97eSAndrei Vagin test(CLOCK_MONOTONIC_COARSE, "monotonic-coarse", true);
981854b97eSAndrei Vagin test(CLOCK_MONOTONIC_RAW, "monotonic-raw", true);
991854b97eSAndrei Vagin test(CLOCK_BOOTTIME, "boottime", true);
1001854b97eSAndrei Vagin
1011854b97eSAndrei Vagin ksft_exit_pass();
1021854b97eSAndrei Vagin return 0;
1031854b97eSAndrei Vagin }
104