tools/testing/selftests/vDSO/vdso_test_getcpu.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * vdso_test_getcpu.c: Sample code to test parse_vdso.c and vDSO getcpu()
   4  *
   5  * Copyright (c) 2020 Arm Ltd
   6  */
   7
   8 #include <stdint.h>
   9 #include <elf.h>
  10 #include <stdio.h>
  11 #include <sys/auxv.h>
  12 #include <sys/time.h>
  13
  14 #include "../kselftest.h"
  15 #include "parse_vdso.h"
  16
  17 #if defined(__riscv)
  18 const char *version = "LINUX_4.15";
  19 #else
  20 const char *version = "LINUX_2.6";
  21 #endif
  22 const char *name = "__vdso_getcpu";
  23
  24 struct getcpu_cache;
  25 typedef long (*getcpu_t)(unsigned int *, unsigned int *,
  26                          struct getcpu_cache *);
  27
  28 int main(int argc, char **argv)
  29 {
  30         unsigned long sysinfo_ehdr;
  31         unsigned int cpu, node;
  32         getcpu_t get_cpu;
  33         long ret;
  34
  35         sysinfo_ehdr = getauxval(AT_SYSINFO_EHDR);
  36         if (!sysinfo_ehdr) {
  37                 printf("AT_SYSINFO_EHDR is not present!\n");
  38                 return KSFT_SKIP;
  39         }
  40
  41         vdso_init_from_sysinfo_ehdr(getauxval(AT_SYSINFO_EHDR));
  42
  43         get_cpu = (getcpu_t)vdso_sym(version, name);
  44         if (!get_cpu) {
  45                 printf("Could not find %s\n", name);
  46                 return KSFT_SKIP;
  47         }
  48
  49         ret = get_cpu(&cpu, &node, 0);
  50         if (ret == 0) {
  51                 printf("Running on CPU %u node %u\n", cpu, node);
  52         } else {
  53                 printf("%s failed\n", name);
  54                 return KSFT_FAIL;
  55         }
  56
  57         return 0;
  58 }