UML - Monitor stack usage

Index: linux-2.6.21-mm/arch/um/kernel/process.c
===================================================================
@@ -427,3 +427,38 @@ unsigned long arch_align_stack(unsigned 
 	return sp & ~0xf;
 }
 #endif
+
+#ifdef CONFIG_STACK_USAGE
+
+static int lowest_to_date = THREAD_SIZE;
+
+void mark_stack(struct thread_info *stack)
+{
+	if(stack == NULL)
+		return;
+
+	memset(stack, 0x6b, THREAD_SIZE);
+}
+
+void check_stack_usage(struct thread_info *s)
+{
+	unsigned int *stack;
+	unsigned int *p, *end;
+	int left;
+
+	stack = (unsigned int *) (s + 1);
+	end = (unsigned int *) ((unsigned long) stack + THREAD_SIZE);
+	for(p = stack; p < end; p++){
+		if(*p != 0x6b6b6b6b){
+			left = (p - stack);
+			if(left < lowest_to_date){
+				printk("Greatest stack depth - %d bytes left\n",
+				       left * sizeof(*p));
+				lowest_to_date = left;
+			}
+			return;
+		}
+	}
+}
+
+#endif

Index: linux-2.6.21-mm/include/asm-um/thread_info.h
===================================================================
@@ -52,6 +52,20 @@ static inline struct thread_info *curren
 	return ti;
 }
 
+#ifdef CONFIG_STACK_USAGE
+
+extern void mark_stack(struct thread_info *stack);
+extern void check_stack_usage(struct thread_info *stack);
+
+#define alloc_thread_info(tsk) \
+	({ struct thread_info *stack = kmalloc(THREAD_SIZE, GFP_KERNEL); \
+	   mark_stack(stack); \
+	   stack; \
+	 })
+#define free_thread_info(ti) ({ check_stack_usage(ti) ; kfree(ti); })
+
+#else
+
 /* thread information allocation */
 #define alloc_thread_info(tsk) \
 	((struct thread_info *) kmalloc(THREAD_SIZE, GFP_KERNEL))

@@ -59,6 +73,8 @@ static inline struct thread_info *curren
 
 #endif
 
+#endif
+
 #define PREEMPT_ACTIVE		0x10000000
 
 #define TIF_SYSCALL_TRACE	0	/* syscall trace active */