From d9dfeb2adeec6e538f4bbc603f0d856d22cb6bac Mon Sep 17 00:00:00 2001
From: simonmar <unknown>
Date: Wed, 2 Nov 2005 12:26:21 +0000
Subject: [PATCH] [project @ 2005-11-02 12:26:21 by simonmar]
 scheduleHandleStackOverflow: fix bug in non-threaded RTS -
 if (task->tso != NULL), then (task->tso) is not necessarily
 the TSO we are running (in the threaded RTS it is).

Also fix a few non-threaded warnings while I'm here.
---
 ghc/rts/Schedule.c |   20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/ghc/rts/Schedule.c b/ghc/rts/Schedule.c
index 2bc7472..15d2181 100644
--- a/ghc/rts/Schedule.c
+++ b/ghc/rts/Schedule.c
@@ -210,7 +210,9 @@ static Capability *schedule (Capability *initialCapability, Task *task);
 // scheduler clearer.
 //
 static void schedulePreLoop (void);
+#if defined(SMP)
 static void schedulePushWork(Capability *cap, Task *task);
+#endif
 static void scheduleStartSignalHandlers (void);
 static void scheduleCheckBlockedThreads (Capability *cap);
 static void scheduleCheckBlackHoles (Capability *cap);
@@ -340,7 +342,9 @@ schedule (Capability *initialCapability, Task *task)
 #endif
   nat prev_what_next;
   rtsBool ready_to_gc;
+#if defined(THREADED_RTS)
   rtsBool first = rtsTrue;
+#endif
   
   cap = initialCapability;
 
@@ -722,10 +726,11 @@ schedulePreLoop(void)
  * Push work to other Capabilities if we have some.
  * -------------------------------------------------------------------------- */
 
+#ifdef SMP
 static void
-schedulePushWork(Capability *cap, Task *task)
+schedulePushWork(Capability *cap USED_WHEN_SMP, 
+		 Task *task      USED_WHEN_SMP)
 {
-#ifdef SMP
     Capability *free_caps[n_capabilities], *cap0;
     nat i, n_free_caps;
 
@@ -794,8 +799,8 @@ schedulePushWork(Capability *cap, Task *task)
 	}
     }
     task->cap = cap; // reset to point to our Capability.
-#endif
 }
+#endif
 
 /* ----------------------------------------------------------------------------
  * Start any pending signal handlers
@@ -1554,11 +1559,10 @@ scheduleHandleStackOverflow (Capability *cap, Task *task, StgTSO *t)
 	/* enlarge the stack */
 	StgTSO *new_t = threadStackOverflow(cap, t);
 	
-	/* This TSO has moved, so update any pointers to it from the
-	 * main thread stack.  It better not be on any other queues...
-	 * (it shouldn't be).
+	/* The TSO attached to this Task may have moved, so update the
+	 * pointer to it.
 	 */
-	if (task->tso != NULL) {
+	if (task->tso == t) {
 	    task->tso = new_t;
 	}
 	pushOnRunQueue(cap,new_t);
@@ -2863,7 +2867,7 @@ threadStackOverflow(Capability *cap, StgTSO *tso)
   new_tso_size = round_to_mblocks(new_tso_size);  /* Be MBLOCK-friendly */
   new_stack_size = new_tso_size - TSO_STRUCT_SIZEW;
 
-  IF_DEBUG(scheduler, sched_belch("increasing stack size from %ld words to %d.\n", tso->stack_size, new_stack_size));
+  IF_DEBUG(scheduler, sched_belch("increasing stack size from %ld words to %d.\n", (long)tso->stack_size, new_stack_size));
 
   dest = (StgTSO *)allocate(new_tso_size);
   TICK_ALLOC_TSO(new_stack_size,0);
-- 
1.7.10.4