1babf8a97SMark Brown // SPDX-License-Identifier: GPL-2.0-only
2babf8a97SMark Brown /*
3babf8a97SMark Brown * vdso_test_gettimeofday.c: Sample code to test parse_vdso.c and
4babf8a97SMark Brown * vDSO gettimeofday()
5babf8a97SMark Brown * Copyright (c) 2014 Andy Lutomirski
6babf8a97SMark Brown *
7babf8a97SMark Brown * Compile with:
8babf8a97SMark Brown * gcc -std=gnu99 vdso_test_gettimeofday.c parse_vdso_gettimeofday.c
9babf8a97SMark Brown *
10babf8a97SMark Brown * Tested on x86, 32-bit and 64-bit. It may work on other architectures, too.
11babf8a97SMark Brown */
12babf8a97SMark Brown
13babf8a97SMark Brown #include <stdint.h>
14babf8a97SMark Brown #include <elf.h>
15babf8a97SMark Brown #include <stdio.h>
16babf8a97SMark Brown #include <sys/auxv.h>
17babf8a97SMark Brown #include <sys/time.h>
18babf8a97SMark Brown
19babf8a97SMark Brown #include "../kselftest.h"
20cd76ca4dSMark Brown #include "parse_vdso.h"
21babf8a97SMark Brown
22babf8a97SMark Brown /*
23babf8a97SMark Brown * ARM64's vDSO exports its gettimeofday() implementation with a different
24babf8a97SMark Brown * name and version from other architectures, so we need to handle it as
25babf8a97SMark Brown * a special case.
26babf8a97SMark Brown */
27babf8a97SMark Brown #if defined(__aarch64__)
28babf8a97SMark Brown const char *version = "LINUX_2.6.39";
29babf8a97SMark Brown const char *name = "__kernel_gettimeofday";
30*d942f231SGuo Ren #elif defined(__riscv)
31*d942f231SGuo Ren const char *version = "LINUX_4.15";
32*d942f231SGuo Ren const char *name = "__vdso_gettimeofday";
33babf8a97SMark Brown #else
34babf8a97SMark Brown const char *version = "LINUX_2.6";
35babf8a97SMark Brown const char *name = "__vdso_gettimeofday";
36babf8a97SMark Brown #endif
37babf8a97SMark Brown
main(int argc,char ** argv)38babf8a97SMark Brown int main(int argc, char **argv)
39babf8a97SMark Brown {
40babf8a97SMark Brown unsigned long sysinfo_ehdr = getauxval(AT_SYSINFO_EHDR);
41babf8a97SMark Brown if (!sysinfo_ehdr) {
42babf8a97SMark Brown printf("AT_SYSINFO_EHDR is not present!\n");
43babf8a97SMark Brown return KSFT_SKIP;
44babf8a97SMark Brown }
45babf8a97SMark Brown
46babf8a97SMark Brown vdso_init_from_sysinfo_ehdr(getauxval(AT_SYSINFO_EHDR));
47babf8a97SMark Brown
48babf8a97SMark Brown /* Find gettimeofday. */
49babf8a97SMark Brown typedef long (*gtod_t)(struct timeval *tv, struct timezone *tz);
50babf8a97SMark Brown gtod_t gtod = (gtod_t)vdso_sym(version, name);
51babf8a97SMark Brown
52babf8a97SMark Brown if (!gtod) {
53babf8a97SMark Brown printf("Could not find %s\n", name);
54babf8a97SMark Brown return KSFT_SKIP;
55babf8a97SMark Brown }
56babf8a97SMark Brown
57babf8a97SMark Brown struct timeval tv;
58babf8a97SMark Brown long ret = gtod(&tv, 0);
59babf8a97SMark Brown
60babf8a97SMark Brown if (ret == 0) {
61babf8a97SMark Brown printf("The time is %lld.%06lld\n",
62babf8a97SMark Brown (long long)tv.tv_sec, (long long)tv.tv_usec);
63babf8a97SMark Brown } else {
64babf8a97SMark Brown printf("%s failed\n", name);
65babf8a97SMark Brown return KSFT_FAIL;
66babf8a97SMark Brown }
67babf8a97SMark Brown
68babf8a97SMark Brown return 0;
69babf8a97SMark Brown }
70