tests/text_corruption_helper.c

   1 #include <stdlib.h>
   2 #include <unistd.h>
   3 #include <stdio.h>
   4 #include <sys/sysctl.h>
   5 #include <ptrauth.h>
   6 #include <math.h>
   7 #include <string.h>
   8
   9 __attribute__((noinline))
  10 static void
  11 foo(void)
  12 {
  13         printf("In foo()\n");
  14         fflush(stdout);
  15 }
  16
  17 /*
  18  * volatile to stop the compiler from optimizing away calls to atan()
  19  */
  20 volatile double zero = 0.0;
  21
  22 int
  23 main(int argc, char **argv)
  24 {
  25         void *addr;
  26         size_t s = sizeof(addr);
  27         int err;
  28         int a;
  29
  30         /*
  31          * needs to run as root for sysctl.
  32          */
  33         if (geteuid() != 0) {
  34                 printf("Test not running as root\n");
  35                 exit(-1);
  36         }
  37
  38         if (strcmp(argv[argc - 1], "foo") == 0) {
  39                 foo();
  40         } else if (strcmp(argv[argc - 1], "Xfoo") == 0) {
  41                 printf("Warm up call to foo()\n");
  42                 foo();
  43                 addr = ptrauth_strip(&foo, ptrauth_key_function_pointer);
  44                 err = sysctlbyname("vm.corrupt_text_addr", NULL, NULL, &addr, s);
  45                 foo();
  46         } else if (strcmp(argv[argc - 1], "atan") == 0) {
  47                 printf("atan(0) is %g\n", atan(zero));
  48         } else if (strcmp(argv[argc - 1], "Xatan") == 0) {
  49                 printf("Warmup call to atan(0) is %g\n", atan(zero));
  50                 addr = ptrauth_strip(&atan, ptrauth_key_function_pointer);
  51                 err = sysctlbyname("vm.corrupt_text_addr", NULL, NULL, &addr, s);
  52                 printf("atan(0) is %g\n", atan(zero));
  53         } else {
  54                 exit(-1);
  55         }
  56 }