buildtools/Cmake/test_prog/prog_stacksetup.c

   1 /* Copyright (c) 2010, 2014-2015. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #if defined OSX
   8 #define _XOPEN_SOURCE 700
   9 #endif
  10
  11 #ifdef _XBT_WIN32
  12 #include "xbt/win32_ucontext.h"
  13 #include "win32_ucontext.c"
  14 #endif
  15
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18 #include <string.h>
  19 #if defined(TEST_sigstack) || defined(TEST_sigaltstack)
  20 #include <sys/types.h>
  21 #include <signal.h>
  22 #include <unistd.h>
  23 #endif
  24
  25 #if defined(TEST_makecontext)
  26 #ifdef _XBT_WIN32
  27 #include "xbt/win32_ucontext.h"
  28 #else
  29 #include <ucontext.h>
  30 #endif
  31
  32 #endif
  33 union alltypes {
  34   long l;
  35   double d;
  36   void *vp;
  37   void (*fp) (void);
  38   char *cp;
  39 };
  40 static volatile char *handler_addr = (char *) 0xDEAD;
  41 #if defined(TEST_sigstack) || defined(TEST_sigaltstack)
  42 static volatile int handler_done = 0;
  43 void handler(int sig)
  44 {
  45   char garbage[1024];
  46   int i;
  47   auto int dummy;
  48   for (i = 0; i < 1024; i++)
  49     garbage[i] = 'X';
  50   handler_addr = (char *) &dummy;
  51   handler_done = 1;
  52   return;
  53 }
  54 #endif
  55 #if defined(TEST_makecontext)
  56 static ucontext_t uc_handler;
  57 static ucontext_t uc_main;
  58 void handler(void)
  59 {
  60   char garbage[1024];
  61   int i;
  62   auto int dummy;
  63   for (i = 0; i < 1024; i++)
  64     garbage[i] = 'X';
  65   handler_addr = (char *) &dummy;
  66   swapcontext(&uc_handler, &uc_main);
  67   return;
  68 }
  69 #endif
  70 int main(int argc, char *argv[])
  71 {
  72   FILE *f;
  73   char *skaddr;
  74   char *skbuf;
  75   int sksize;
  76   char result[1024];
  77   int i;
  78   sksize = 32768;
  79   skbuf = (char *) malloc(sksize * 2 + 2 * sizeof(union alltypes));
  80   if (skbuf == NULL)
  81     exit(1);
  82   for (i = 0; i < sksize * 2 + 2 * sizeof(union alltypes); i++)
  83     skbuf[i] = 'A';
  84   skaddr = skbuf + sizeof(union alltypes);
  85 #if defined(TEST_sigstack) || defined(TEST_sigaltstack)
  86   {
  87     struct sigaction sa;
  88 #if defined(TEST_sigstack)
  89     struct sigstack ss;
  90 #elif defined(TEST_sigaltstack) && defined(HAVE_STACK_T)
  91     stack_t ss;
  92 #else
  93     struct sigaltstack ss;
  94 #endif
  95 #if defined(TEST_sigstack)
  96     ss.ss_sp = (void *) (skaddr + sksize);
  97     ss.ss_onstack = 0;
  98     if (sigstack(&ss, NULL) < 0)
  99       exit(1);
 100 #elif defined(TEST_sigaltstack)
 101     ss.ss_sp = (void *) (skaddr + sksize);
 102     ss.ss_size = sksize;
 103     ss.ss_flags = 0;
 104     if (sigaltstack(&ss, NULL) < 0)
 105       exit(1);
 106 #endif
 107     memset((void *) &sa, 0, sizeof(struct sigaction));
 108     sa.sa_handler = handler;
 109     sa.sa_flags = SA_ONSTACK;
 110     sigemptyset(&sa.sa_mask);
 111     sigaction(SIGUSR1, &sa, NULL);
 112     kill(getpid(), SIGUSR1);
 113     while (!handler_done)
 114       /*nop */ ;
 115   }
 116 #endif
 117 #if defined(TEST_makecontext)
 118   {
 119     if (getcontext(&uc_handler) != 0)
 120       exit(1);
 121     uc_handler.uc_link = NULL;
 122     uc_handler.uc_stack.ss_sp = (void *) (skaddr + sksize);
 123     uc_handler.uc_stack.ss_size = sksize;
 124     uc_handler.uc_stack.ss_flags = 0;
 125     makecontext(&uc_handler, handler, 0);
 126     swapcontext(&uc_main, &uc_handler);
 127   }
 128 #endif
 129   if (handler_addr == (char *) 0xDEAD)
 130     exit(1);
 131   if (handler_addr < skaddr + sksize) {
 132     /* stack was placed into lower area */
 133     if (*(skaddr + sksize) != 'A')
 134       sprintf(result, "(skaddr)+(sksize)-%d,(sksize)-%d",
 135               sizeof(union alltypes), sizeof(union alltypes));
 136     else
 137       strcpy(result, "(skaddr)+(sksize),(sksize)");
 138   } else {
 139     /* stack was placed into higher area */
 140     if (*(skaddr + sksize * 2) != 'A')
 141       sprintf(result, "(skaddr),(sksize)-%d", sizeof(union alltypes));
 142     else
 143       strcpy(result, "(skaddr),(sksize)");
 144   }
 145   if ((f = fopen("conftestval", "w")) == NULL)
 146     exit(1);
 147   fprintf(f, "%s\n", result);
 148   fclose(f);
 149   exit(0);
 150   return 1;
 151 }