test/exec/stackswap.c

   1 /*
   2     Copyright © 1995-2014, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <exec/tasks.h>
   7 #include <proto/dos.h>
   8 #include <proto/exec.h>
   9
  10 #define print Printf
  11 /*
  12 #include <aros/debug.h>
  13 #define print bug
  14 */
  15 #define STACK_SIZE 32768
  16
  17 #ifdef __MORPHOS__
  18 #define StackSwapArgs PPCStackSwapArgs
  19 #define NewStackSwap NewPPCStackSwap
  20 #endif
  21
  22 struct StackSwapStruct sss;
  23
  24 int __nocommandline = 1;
  25
  26 void PrintSSS(struct StackSwapStruct *ss)
  27 {
  28     print("StackSwapStruct contents:\n");
  29     print("Lower: 0x%P, Upper: 0x%P, SP: 0x%P\n", ss->stk_Lower, ss->stk_Upper, ss->stk_Pointer);
  30 }
  31
  32 void PrintTaskStack(void)
  33 {
  34     struct Task *t = FindTask(NULL);
  35
  36     print("Current program stack:\n");
  37     print("Lower: 0x%P, Upper: 0x%P, SP: 0x%P\n", t->tc_SPLower, t->tc_SPUpper, AROS_GET_SP);
  38 }
  39
  40 void Sub(IPTR a1, IPTR a2)
  41 {
  42     print("Stack swap done\n");
  43     PrintSSS(&sss);
  44     PrintTaskStack();
  45     print("Arguments: %08lX, %08lX\n", a1, a2);
  46     print("Coming back to original stack...\n");
  47 }
  48
  49 /* We keep quiet unless an error occurs to avoid context switches */
  50 BOOL SilentSub(IPTR a1, IPTR a2)
  51 {
  52     BOOL success = TRUE;
  53     APTR sp, spreg;
  54     struct Task *t = FindTask(NULL);
  55
  56     if(a2 != (IPTR)t)
  57     {
  58         print("ERROR: second parameter has incorrect value!\n");
  59         success = FALSE;
  60     }
  61
  62     sp = AROS_GET_SP;
  63     if(t->tc_SPReg < t->tc_SPLower || t->tc_SPReg > t->tc_SPUpper
  64         || sp < t->tc_SPLower || sp > t->tc_SPUpper)
  65     {
  66         spreg = t->tc_SPReg;
  67         print("ERROR: invalid stack values in user function:\n");
  68         print("Lower: 0x%P\tUpper: 0x%P\nSP(struct): 0x%P\tSP(reg): 0x%P\n",
  69             t->tc_SPLower, t->tc_SPUpper, spreg, sp);
  70         success = FALSE;
  71     }
  72
  73     return success;
  74 }
  75
  76 int main(void)
  77 {
  78     struct StackSwapArgs args;
  79     APTR sp, spreg;
  80     struct Task *t = FindTask(NULL);
  81
  82     sss.stk_Lower = AllocMem(STACK_SIZE, MEMF_ANY);
  83     if (!sss.stk_Lower)
  84     {
  85         print("Failed to allocate new stack!\n");
  86
  87         return 1;
  88     }
  89
  90     sss.stk_Upper = sss.stk_Lower + STACK_SIZE;
  91     sss.stk_Pointer = sss.stk_Upper - sizeof(IPTR);
  92     PrintSSS(&sss);
  93     PrintTaskStack();
  94
  95     print("Checking StackSwap()...\n");
  96     StackSwap(&sss);
  97
  98     Sub(0x12345678, 0xC001C0DE);
  99
 100     StackSwap(&sss);
 101     print("Came back from StackSwap()\n");
 102     PrintSSS(&sss);
 103     PrintTaskStack();
 104
 105     print("Checking NewStackSwap()...\n");
 106     args.Args[0] = 0x1234ABCD;
 107     args.Args[1] = 0xC0DEC001;
 108
 109     NewStackSwap(&sss, Sub, &args);
 110     print("Came back from NewStackSwap()\n");
 111     PrintSSS(&sss);
 112     PrintTaskStack();
 113
 114     /* Check that stack fields in task structure make sense */
 115
 116     print("Checking NewStackSwap() again...\n");
 117     args.Args[0] = 0x1234ABCD;
 118     args.Args[1] = (IPTR)FindTask(NULL);
 119
 120     Forbid();
 121     NewStackSwap(&sss, SilentSub, &args);
 122
 123     sp = AROS_GET_SP;
 124     if(t->tc_SPReg < t->tc_SPLower || t->tc_SPReg > t->tc_SPUpper
 125         || sp < t->tc_SPLower || sp > t->tc_SPUpper)
 126     {
 127         spreg = t->tc_SPReg;
 128         print("ERROR: invalid stack values after return from user function:\n");
 129         print("Lower: 0x%P\tUpper: 0x%P\nSP(struct): 0x%P\tSP(reg): 0x%P\n",
 130             t->tc_SPLower, t->tc_SPUpper, spreg, sp);
 131     }
 132     Permit();
 133
 134     FreeMem(sss.stk_Lower, STACK_SIZE);
 135
 136     print("Done!\n");
 137     return 0;
 138 }