workbench/c/StackSnoop.c

   1 /*
   2     Copyright © 1995-2011, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Desc:
   6     Lang: English
   7  */
   8
   9 #include <exec/exec.h>
  10 #include <exec/execbase.h>
  11 #include <exec/tasks.h>
  12 #include <exec/rawfmt.h>
  13 #include <dos/dos.h>
  14 #include <proto/exec.h>
  15 #include <proto/dos.h>
  16 #include <proto/debug.h>
  17 #include <proto/alib.h>
  18
  19 #include <string.h>
  20 #include <setjmp.h>
  21
  22 #define ARG_TEMPLATE "TASK/K,DEBUG/S"
  23
  24 #define ARG_TASK    0
  25 #define ARG_DEBUG   1
  26 #define NUM_ARGS    2
  27
  28 static struct RDArgs *MyArgs;
  29 static IPTR Args[NUM_ARGS];
  30 static char s[256];
  31 static jmp_buf exit_buf;
  32
  33 static UBYTE outbuffer[20000];
  34 static LONG outbuffer_size;
  35
  36 static void Cleanup(char *msg)
  37 {
  38     WORD rc;
  39
  40     if (msg)
  41     {
  42         Printf("stacksnoop: %s\n",msg);
  43         rc = RETURN_WARN;
  44     } else {
  45         rc = RETURN_OK;
  46     }
  47
  48     if (MyArgs) FreeArgs(MyArgs);
  49
  50     if (rc != RETURN_OK)
  51         longjmp(exit_buf, rc);
  52 }
  53
  54 static void GetArguments(void)
  55 {
  56     if (!(MyArgs = ReadArgs(ARG_TEMPLATE,Args,0)))
  57     {
  58         Fault(IoErr(),0,s,255);
  59         Cleanup(s);
  60     }
  61 }
  62
  63 static int out (const UBYTE * fmt, ...)
  64 {
  65     va_list      ap;
  66     int          result;
  67
  68     va_start (ap, fmt);
  69
  70     VNewRawDoFmt(fmt, RAWFMTFUNC_STRING, &outbuffer[outbuffer_size], ap);
  71     result = strlen(&outbuffer[outbuffer_size]);
  72
  73     if (Args[ARG_DEBUG])
  74         KPutStr(&outbuffer[outbuffer_size]);
  75
  76     outbuffer_size += result;
  77
  78     va_end (ap);
  79
  80     return result;
  81 }
  82
  83
  84 static void CheckTaskStack(struct Task *task)
  85 {
  86     UBYTE *startcheck, *endcheck;
  87     LONG stacksize, stackinc, unusedstack = 0;
  88
  89     stacksize = (LONG)( ((UBYTE *)task->tc_SPUpper) - ((UBYTE *)task->tc_SPLower) );
  90
  91 #if AROS_STACK_GROWS_DOWNWARDS
  92     startcheck = (UBYTE *)task->tc_SPLower;
  93     endcheck = ((UBYTE *)task->tc_SPUpper) - 1;
  94     stackinc = 1;
  95 #else
  96     startcheck = ((UBYTE *)task->tc_SPUpper) - 1;
  97     endcheck = (UBYTE *)task->tc_SPLower;
  98     stackinc = -1;
  99 #endif
 100
 101     for(; startcheck != endcheck; startcheck += stackinc)
 102     {
 103         if (*startcheck != 0xE1) break;
 104         unusedstack++;
 105     }
 106
 107     out("Task %x (%25s) Stack Size =%6d, Left =%6d, Used %s%6d%s\n",
 108                 task,
 109                 task->tc_Node.ln_Name ? task->tc_Node.ln_Name : "<NONAME>",
 110                 stacksize,
 111                 unusedstack,
 112                 ((stacksize - unusedstack) < stacksize) ? "=" : ">",
 113                 stacksize - unusedstack,
 114                 (unusedstack < 512) ? ": Needs more stack!!!!!!!!!" : "");
 115 }
 116
 117 static void Action(void)
 118 {
 119     struct Task *task;
 120     WORD i;
 121
 122     Disable();
 123
 124     out("\n------------------------------------------------------------------------------\n\n");
 125
 126     task = (struct Task *)SysBase->TaskReady.lh_Head;
 127     for(i = 0; i < 2;i++)
 128     {
 129         while(task->tc_Node.ln_Succ)
 130         {
 131             CheckTaskStack(task);
 132
 133             task = (struct Task *)task->tc_Node.ln_Succ;
 134         } /* while(task->tc_Node.ln_Succ) */
 135
 136         task = (struct Task *)SysBase->TaskWait.lh_Head;
 137
 138     } /* for(i = 0; i < 2;i++) */
 139     out("\n");
 140     CheckTaskStack(FindTask(NULL));
 141     out("\n------------------------------------------------------------------------------\n\n");
 142
 143     Enable();
 144
 145     PutStr(outbuffer);
 146 }
 147
 148 ULONG __nocommandline = 1;
 149
 150 int main(void)
 151 {
 152     int rc;
 153
 154     if ((rc = setjmp(exit_buf)) != 0)
 155         return rc;
 156
 157     GetArguments();
 158     Action();
 159     Cleanup(0);
 160     return 0;
 161 }