[project @ 2002-01-23 23:53:54 by sof]
[ghc-hetmet.git] / ghc / compiler / compMan / CompManager.lhs
index 8c7cf64..691d499 100644 (file)
 %
-% (c) The University of Glasgow, 2000
+% (c) The University of Glasgow, 2002
+%
+% The Compilation Manager
 %
-\section[CompManager]{The Compilation Manager}
-
 \begin{code}
+{-# OPTIONS -fvia-C #-}
 module CompManager ( 
-    cmInit,      -- :: GhciMode -> IO CmState
+    ModuleGraph, 
+
+    CmState, emptyCmState,  -- abstract
 
-    cmLoadModule, -- :: CmState -> FilePath -> IO (CmState, [String])
+    cmInit,       -- :: GhciMode -> IO CmState
 
-    cmUnload,    -- :: CmState -> DynFlags -> IO CmState
+    cmDepAnal,    -- :: CmState -> DynFlags -> [FilePath] -> IO ModuleGraph
 
-    cmSetContext, -- :: CmState -> String -> IO CmState
+    cmLoadModules, -- :: CmState -> DynFlags -> ModuleGraph
+                  --    -> IO (CmState, [String])
 
-    cmGetContext, -- :: CmState -> IO String
+    cmUnload,     -- :: CmState -> DynFlags -> IO CmState
 
 #ifdef GHCI
-    cmRunStmt,   --  :: CmState -> DynFlags -> String -> IO (CmState, [Name])
+    cmModuleIsInterpreted, -- :: CmState -> String -> IO Bool
+
+    cmSetContext,  -- :: CmState -> DynFlags -> [String] -> [String] -> IO CmState
+    cmGetContext,  -- :: CmState -> IO ([String],[String])
+
+    cmInfoThing,   -- :: CmState -> DynFlags -> String
+                  --   -> IO (CmState, [(TyThing,Fixity)])
 
-    cmTypeOfExpr, --  :: CmState -> DynFlags -> String
-                 --  -> IO (CmState, Maybe String)
+    CmRunResult(..),
+    cmRunStmt,    -- :: CmState -> DynFlags -> String
+                  --    -> IO (CmState, CmRunResult)
 
-    cmTypeOfName, -- :: CmState -> Name -> IO (Maybe String)
+    cmTypeOfExpr,  -- :: CmState -> DynFlags -> String
+                  --   -> IO (CmState, Maybe String)
 
-    cmCompileExpr,-- :: CmState -> DynFlags -> String 
-                 -- -> IO (CmState, Maybe HValue)#endif
+    cmTypeOfName,  -- :: CmState -> Name -> IO (Maybe String)
+
+    HValue,
+    cmCompileExpr, -- :: CmState -> DynFlags -> String 
+                  --   -> IO (CmState, Maybe HValue)
+
+    cmGetModuleGraph,          -- :: CmState -> ModuleGraph
+    cmGetLinkables,            -- :: CmState -> [Linkable]
+
+    cmGetBindings,     -- :: CmState -> [TyThing]
+    cmGetPrintUnqual,  -- :: CmState -> PrintUnqualified
 #endif
-    CmState, emptyCmState  -- abstract
+
+    -- utils
+    showModMsg,                -- 
   )
 where
 
 #include "HsVersions.h"
 
+import MkIface --tmp
+import HsSyn   -- tmp
+
 import CmLink
 import CmTypes
 import DriverPipeline
-import DriverFlags     ( getDynFlags )
+import DriverState     ( v_Output_file )
 import DriverPhases
 import DriverUtil
 import Finder
+#ifdef GHCI
+import HscMain         ( initPersistentCompilerState, hscThing )
+#else
 import HscMain         ( initPersistentCompilerState )
+#endif
 import HscTypes
-import RnEnv           ( unQualInScope )
-import Id              ( idType, idName )
-import Name            ( Name, NamedThing(..), nameRdrName )
+import Name            ( Name, NamedThing(..), nameRdrName, nameModule,
+                         isHomePackageName, isGlobalName )
 import NameEnv
-import RdrName         ( lookupRdrEnv, emptyRdrEnv )
+import Rename          ( mkGlobalContext )
+import RdrName         ( emptyRdrEnv )
 import Module
 import GetImports
-import Type            ( tidyType )
-import VarEnv          ( emptyTidyEnv )
 import UniqFM
 import Unique          ( Uniquable )
-import Digraph         ( SCC(..), stronglyConnComp, flattenSCC )
+import Digraph         ( SCC(..), stronglyConnComp, flattenSCC, flattenSCCs )
 import ErrUtils                ( showPass )
 import SysTools                ( cleanTempFilesExcept )
 import Util
 import Outputable
 import Panic
-import CmdLineOpts     ( DynFlags(..) )
+import CmdLineOpts     ( DynFlags(..), getDynFlags )
+
 import IOExts
 
 #ifdef GHCI
+import RdrName         ( lookupRdrEnv )
+import Id              ( idType, idName )
+import Type            ( tidyType )
+import VarEnv          ( emptyTidyEnv )
+import BasicTypes      ( Fixity, defaultFixity )
 import Interpreter     ( HValue )
 import HscMain         ( hscStmt )
 import PrelGHC         ( unsafeCoerce# )
+
+import Foreign
+import CForeign
+import Exception       ( Exception, try )
 #endif
 
 -- lang
@@ -94,8 +132,8 @@ data CmState
         pls    :: PersistentLinkerState    -- link's persistent state
      }
 
-emptyCmState :: GhciMode -> Module -> IO CmState
-emptyCmState gmode mod
+emptyCmState :: GhciMode -> IO CmState
+emptyCmState gmode
     = do pcs     <- initPersistentCompilerState
          pls     <- emptyPLS
          return (CmState { hst    = emptySymbolTable,
@@ -103,18 +141,18 @@ emptyCmState gmode mod
                            ui     = emptyUI,
                            mg     = emptyMG, 
                            gmode  = gmode,
-                          ic     = emptyInteractiveContext mod,
+                          ic     = emptyInteractiveContext,
                            pcs    = pcs,
                            pls    = pls })
 
-emptyInteractiveContext mod
-  = InteractiveContext { ic_module = mod, 
-                        ic_rn_env = emptyRdrEnv,
+emptyInteractiveContext
+  = InteractiveContext { ic_toplev_scope = [],
+                        ic_exports = [],
+                        ic_rn_gbl_env = emptyRdrEnv,
+                        ic_print_unqual = alwaysQualify,
+                        ic_rn_local_env = emptyRdrEnv,
                         ic_type_env = emptyTypeEnv }
 
-defaultCurrentModuleName = mkModuleName "Prelude"
-GLOBAL_VAR(defaultCurrentModule, error "no defaultCurrentModule", Module)
-
 -- CM internal types
 type UnlinkedImage = [Linkable]        -- the unlinked images (should be a set, really)
 emptyUI :: UnlinkedImage
@@ -128,63 +166,123 @@ emptyMG = []
 -- Produce an initial CmState.
 
 cmInit :: GhciMode -> IO CmState
-cmInit mode = do
-   prel <- moduleNameToModule defaultCurrentModuleName
-   writeIORef defaultCurrentModule prel
-   emptyCmState mode prel
+cmInit mode = emptyCmState mode
+
+-----------------------------------------------------------------------------
+-- Grab information from the CmState
+
+cmGetModuleGraph        = mg
+cmGetLinkables          = ui
+
+cmGetBindings cmstate = nameEnvElts (ic_type_env (ic cmstate))
+cmGetPrintUnqual cmstate = ic_print_unqual (ic cmstate)
 
 -----------------------------------------------------------------------------
 -- Setting the context doesn't throw away any bindings; the bindings
 -- we've built up in the InteractiveContext simply move to the new
 -- module.  They always shadow anything in scope in the current context.
 
-cmSetContext :: CmState -> String -> IO CmState
-cmSetContext cmstate str
-   = do let mn = mkModuleName str
-           modules_loaded = [ (name_of_summary s, ms_mod s)  | s <- mg cmstate ]
-
-        m <- case lookup mn modules_loaded of
-               Just m  -> return m
-               Nothing -> do
-                  mod <- moduleNameToModule mn
-                  if isHomeModule mod 
-                       then throwDyn (CmdLineError (showSDoc 
-                               (quotes (ppr (moduleName mod))
-                                 <+> text "is not currently loaded")))
-                       else return mod
-
-       return cmstate{ ic = (ic cmstate){ic_module=m} }
-               
-cmGetContext :: CmState -> IO String
-cmGetContext cmstate = return (moduleUserString (ic_module (ic cmstate)))
-
-moduleNameToModule :: ModuleName -> IO Module
-moduleNameToModule mn
- = do maybe_stuff <- findModule mn
-      case maybe_stuff of
-       Nothing -> throwDyn (CmdLineError ("can't find module `"
+cmSetContext
+       :: CmState -> DynFlags
+       -> [String]             -- take the top-level scopes of these modules
+       -> [String]             -- and the just the exports from these
+       -> IO CmState
+cmSetContext cmstate dflags toplevs exports = do 
+  let CmState{ hit=hit, hst=hst, pcs=pcs, ic=old_ic } = cmstate
+
+  toplev_mods <- mapM (getTopLevModule hit)    (map mkModuleName toplevs)
+  export_mods <- mapM (moduleNameToModule hit) (map mkModuleName exports)
+
+  (new_pcs, print_unqual, maybe_env)
+      <- mkGlobalContext dflags hit hst pcs toplev_mods export_mods
+
+  case maybe_env of 
+    Nothing -> return cmstate
+    Just env -> return cmstate{ pcs = new_pcs,
+                               ic = old_ic{ ic_toplev_scope = toplev_mods,
+                                            ic_exports = export_mods,
+                                            ic_rn_gbl_env = env,
+                                            ic_print_unqual = print_unqual } }
+
+getTopLevModule hit mn =
+  case lookupModuleEnvByName hit mn of
+    Just iface
+      | Just _ <- mi_globals iface -> return (mi_module iface)
+    _other -> throwDyn (CmdLineError (
+         "cannot enter the top-level scope of a compiled module (module `" ++
+          moduleNameUserString mn ++ "')"))
+
+moduleNameToModule :: HomeIfaceTable -> ModuleName -> IO Module
+moduleNameToModule hit mn = do
+  case lookupModuleEnvByName hit mn of
+    Just iface -> return (mi_module iface)
+    _not_a_home_module -> do
+       maybe_stuff <- findModule mn
+        case maybe_stuff of
+         Nothing -> throwDyn (CmdLineError ("can't find module `"
                                    ++ moduleNameUserString mn ++ "'"))
-       Just (m,_) -> return m
+         Just (m,_) -> return m
+
+cmGetContext :: CmState -> IO ([String],[String])
+cmGetContext CmState{ic=ic} = 
+  return (map moduleUserString (ic_toplev_scope ic), 
+         map moduleUserString (ic_exports ic))
+
+cmModuleIsInterpreted :: CmState -> String -> IO Bool
+cmModuleIsInterpreted cmstate str 
+ = case lookupModuleEnvByName (hit cmstate) (mkModuleName str) of
+      Just iface         -> return (not (isNothing (mi_globals iface)))
+      _not_a_home_module -> return False
+
+-----------------------------------------------------------------------------
+-- cmInfoThing: convert a String to a TyThing
+
+-- A string may refer to more than one TyThing (eg. a constructor,
+-- and type constructor), so we return a list of all the possible TyThings.
+
+#ifdef GHCI
+cmInfoThing :: CmState -> DynFlags -> String -> IO (CmState, [(TyThing,Fixity)])
+cmInfoThing cmstate dflags id
+   = do (new_pcs, things) <- hscThing dflags hst hit pcs icontext id
+       let pairs = map (\x -> (x, getFixity new_pcs (getName x))) things
+       return (cmstate{ pcs=new_pcs }, pairs)
+   where
+     CmState{ hst=hst, hit=hit, pcs=pcs, pls=pls, ic=icontext } = cmstate
+
+     getFixity :: PersistentCompilerState -> Name -> Fixity
+     getFixity pcs name
+       | isGlobalName name,
+         Just iface  <- lookupModuleEnv iface_table (nameModule name),
+         Just fixity <- lookupNameEnv (mi_fixities iface) name
+         = fixity
+       | otherwise
+         = defaultFixity
+       where iface_table | isHomePackageName name = hit
+                         | otherwise              = pcs_PIT pcs
+#endif
 
 -----------------------------------------------------------------------------
 -- cmRunStmt:  Run a statement/expr.
 
 #ifdef GHCI
-cmRunStmt :: CmState -> DynFlags -> String
-       -> IO (CmState,                 -- new state
-              [Name])                  -- names bound by this evaluation
-cmRunStmt cmstate dflags expr
+data CmRunResult
+  = CmRunOk [Name]             -- names bound by this evaluation
+  | CmRunFailed 
+  | CmRunException Exception   -- statement raised an exception
+
+cmRunStmt :: CmState -> DynFlags -> String -> IO (CmState, CmRunResult)                
+cmRunStmt cmstate@CmState{ hst=hst, hit=hit, pcs=pcs, pls=pls, ic=icontext }
+          dflags expr
    = do 
        let InteractiveContext { 
-               ic_rn_env = rn_env, 
-               ic_type_env = type_env,
-               ic_module   = this_mod } = icontext
+               ic_rn_local_env = rn_env, 
+               ic_type_env     = type_env } = icontext
 
         (new_pcs, maybe_stuff) 
            <- hscStmt dflags hst hit pcs icontext expr False{-stmt-}
 
         case maybe_stuff of
-          Nothing -> return (cmstate{ pcs=new_pcs }, [])
+          Nothing -> return (cmstate{ pcs=new_pcs }, CmRunFailed)
           Just (ids, _, bcos) -> do
 
                -- update the interactive context
@@ -203,25 +301,70 @@ cmRunStmt cmstate dflags expr
                    new_type_env = extendNameEnvList filtered_type_env  
                                        [ (getName id, AnId id) | id <- ids]
 
-                   new_ic = icontext { ic_rn_env   = new_rn_env, 
-                                       ic_type_env = new_type_env }
+                   new_ic = icontext { ic_rn_local_env = new_rn_env, 
+                                       ic_type_env     = new_type_env }
 
                -- link it
                hval <- linkExpr pls bcos
 
                -- run it!
                let thing_to_run = unsafeCoerce# hval :: IO [HValue]
-               hvals <- thing_to_run
-
-               -- Get the newly bound things, and bind them.  Don't forget
-               -- to delete any shadowed bindings from the closure_env, lest
-               -- we end up with a space leak.
-               pls <- delListFromClosureEnv pls shadowed
-               new_pls <- addListToClosureEnv pls (zip names hvals)
-
-               return (cmstate{ pcs=new_pcs, pls=new_pls, ic=new_ic }, names)
-   where
-       CmState{ hst=hst, hit=hit, pcs=pcs, pls=pls, ic=icontext } = cmstate
+               either_hvals <- sandboxIO thing_to_run
+               case either_hvals of
+                  Left err
+                       -> do hPutStrLn stderr ("unknown failure, code " ++ show err)
+                             return ( cmstate{ pcs=new_pcs, ic=new_ic }, CmRunFailed )
+
+                  Right maybe_hvals ->
+                    case maybe_hvals of
+                       Left e -> 
+                           return ( cmstate{ pcs=new_pcs, ic=new_ic }, 
+                                    CmRunException e )
+                       Right hvals -> do
+                            -- Get the newly bound things, and bind them.  
+                            -- Don't forget to delete any shadowed bindings from the
+                            -- closure_env, lest we end up with a space leak.
+                            pls <- delListFromClosureEnv pls shadowed
+                            new_pls <- addListToClosureEnv pls (zip names hvals)
+            
+                            return (cmstate{ pcs=new_pcs, pls=new_pls, ic=new_ic }, 
+                                    CmRunOk names)
+
+
+-- We run the statement in a "sandbox" to protect the rest of the
+-- system from anything the expression might do.  For now, this
+-- consists of just wrapping it in an exception handler, but see below
+-- for another version.
+
+sandboxIO :: IO a -> IO (Either Int (Either Exception a))
+sandboxIO thing = do
+  r <- Exception.try thing
+  return (Right r)
+
+{-
+-- This version of sandboxIO runs the expression in a completely new
+-- RTS main thread.  It is disabled for now because ^C exceptions
+-- won't be delivered to the new thread, instead they'll be delivered
+-- to the (blocked) GHCi main thread.
+
+sandboxIO :: IO a -> IO (Either Int (Either Exception a))
+sandboxIO thing = do
+  st_thing <- newStablePtr (Exception.try thing)
+  alloca $ \ p_st_result -> do
+    stat <- rts_evalStableIO st_thing p_st_result
+    freeStablePtr st_thing
+    if stat == 1
+       then do st_result <- peek p_st_result
+               result <- deRefStablePtr st_result
+               freeStablePtr st_result
+               return (Right result)
+       else do
+               return (Left (fromIntegral stat))
+
+foreign import "rts_evalStableIO"  {- safe -}
+  rts_evalStableIO :: StablePtr (IO a) -> Ptr (StablePtr a) -> IO CInt
+  -- more informative than the C type!
+-}
 #endif
 
 -----------------------------------------------------------------------------
@@ -237,15 +380,11 @@ cmTypeOfExpr cmstate dflags expr
 
        case maybe_stuff of
           Nothing -> return (new_cmstate, Nothing)
-          Just (_, ty, _) ->
-            let pit = pcs_PIT pcs
-                modname = moduleName (ic_module ic)
-                tidy_ty = tidyType emptyTidyEnv ty
-                str = case lookupIfaceByModName hit pit modname of
-                         Nothing    -> showSDoc (ppr tidy_ty)
-                         Just iface -> showSDocForUser unqual (ppr tidy_ty)
-                            where unqual = unQualInScope (mi_globals iface)
-            in return (new_cmstate, Just str)
+          Just (_, ty, _) -> return (new_cmstate, Just str)
+            where 
+               str = showSDocForUser unqual (ppr tidy_ty)
+               unqual  = ic_print_unqual ic
+               tidy_ty = tidyType emptyTidyEnv ty
    where
        CmState{ hst=hst, hit=hit, pcs=pcs, ic=ic } = cmstate
 #endif
@@ -258,15 +397,11 @@ cmTypeOfName :: CmState -> Name -> IO (Maybe String)
 cmTypeOfName CmState{ hit=hit, pcs=pcs, ic=ic } name
  = case lookupNameEnv (ic_type_env ic) name of
        Nothing -> return Nothing
-       Just (AnId id) -> 
-          let pit = pcs_PIT pcs
-              modname = moduleName (ic_module ic)
-              ty = tidyType emptyTidyEnv (idType id)
-              str = case lookupIfaceByModName hit pit modname of
-                       Nothing    -> showSDoc (ppr ty)
-                       Just iface -> showSDocForUser unqual (ppr ty)
-                          where unqual = unQualInScope (mi_globals iface)
-          in return (Just str)
+       Just (AnId id) -> return (Just str)
+          where
+            unqual = ic_print_unqual ic
+            ty = tidyType emptyTidyEnv (idType id)
+            str = showSDocForUser unqual (ppr ty)
 
        _ -> panic "cmTypeOfName"
 #endif
@@ -279,9 +414,8 @@ cmCompileExpr :: CmState -> DynFlags -> String -> IO (CmState, Maybe HValue)
 cmCompileExpr cmstate dflags expr
    = do 
        let InteractiveContext { 
-               ic_rn_env = rn_env, 
-               ic_type_env = type_env,
-               ic_module   = this_mod } = icontext
+               ic_rn_local_env = rn_env, 
+               ic_type_env     = type_env } = icontext
 
         (new_pcs, maybe_stuff) 
            <- hscStmt dflags hst hit pcs icontext 
@@ -307,21 +441,6 @@ cmCompileExpr cmstate dflags expr
 #endif
 
 -----------------------------------------------------------------------------
--- cmInfo: return "info" about an expression.  The info might be:
---
---     * its type, for an expression,
---     * the class definition, for a class
---     * the datatype definition, for a tycon (or synonym)
---     * the export list, for a module
---
--- Can be used to find the type of the last expression compiled, by looking
--- for "it".
-
-cmInfo :: CmState -> String -> IO (Maybe String)
-cmInfo cmstate str 
- = do error "cmInfo not implemented yet"
-
------------------------------------------------------------------------------
 -- Unload the compilation manager's state: everything it knows about the
 -- current collection of modules in the Home package.
 
@@ -337,18 +456,35 @@ cmUnload state@CmState{ gmode=mode, pls=pls, pcs=pcs } dflags
       new_state <- cmInit mode
       return new_state{ pcs=pcs, pls=new_pls }
 
+
+-----------------------------------------------------------------------------
+-- Trace dependency graph
+
+-- This is a seperate pass so that the caller can back off and keep
+-- the current state if the downsweep fails.
+
+cmDepAnal :: CmState -> DynFlags -> [FilePath] -> IO ModuleGraph
+cmDepAnal cmstate dflags rootnames
+  = do showPass dflags "Chasing dependencies"
+       when (verbosity dflags >= 1 && gmode cmstate == Batch) $
+           hPutStrLn stderr (showSDoc (hcat [
+            text progName, text ": chasing modules from: ",
+            hcat (punctuate comma (map text rootnames))]))
+       downsweep rootnames (mg cmstate)
+
 -----------------------------------------------------------------------------
 -- The real business of the compilation manager: given a system state and
 -- a module name, try and bring the module up to date, probably changing
 -- the system state at the same time.
 
-cmLoadModule :: CmState 
-             -> FilePath
+cmLoadModules :: CmState 
+            -> DynFlags
+             -> ModuleGraph
              -> IO (CmState,           -- new state
                    Bool,               -- was successful
                    [String])           -- list of modules loaded
 
-cmLoadModule cmstate1 rootname
+cmLoadModules cmstate1 dflags mg2unsorted
    = do -- version 1's are the original, before downsweep
         let pls1      = pls    cmstate1
         let pcs1      = pcs    cmstate1
@@ -357,20 +493,17 @@ cmLoadModule cmstate1 rootname
        -- similarly, ui1 is the (complete) set of linkables from
        -- the previous pass, if any.
         let ui1       = ui     cmstate1
-       let mg1       = mg     cmstate1
-       let ic1       = ic     cmstate1
 
         let ghci_mode = gmode cmstate1 -- this never changes
 
         -- Do the downsweep to reestablish the module graph
-       dflags <- getDynFlags
         let verb = verbosity dflags
 
-       showPass dflags "Chasing dependencies"
-        when (verb >= 1 && ghci_mode == Batch) $
-           hPutStrLn stderr (progName ++ ": chasing modules from: " ++ rootname)
+       -- Find out if we have a Main module
+        let a_root_is_Main 
+               = any ((=="Main").moduleNameUserString.name_of_summary) 
+                     mg2unsorted
 
-        (mg2unsorted, a_root_is_Main) <- downsweep [rootname] mg1
         let mg2unsorted_names = map name_of_summary mg2unsorted
 
         -- reachable_from follows source as well as normal imports
@@ -419,7 +552,7 @@ cmLoadModule cmstate1 rootname
 
        -- unload any modules which aren't going to be re-linked this
        -- time around.
-       pls2 <- unload ghci_mode dflags stable_linkables pls1
+       pls2 <- CmLink.unload ghci_mode dflags stable_linkables pls1
 
         -- We could at this point detect cycles which aren't broken by
         -- a source-import, and complain immediately, but it seems better
@@ -443,9 +576,13 @@ cmLoadModule cmstate1 rootname
 
         let threaded2 = CmThreaded pcs1 hst1 hit1
 
+       -- clean up between compilations
+       let cleanup = cleanTempFilesExcept verb 
+                         (ppFilesFromSummaries (flattenSCCs upsweep_these))
+
         (upsweep_complete_success, threaded3, modsUpswept, newLis)
            <- upsweep_mods ghci_mode dflags valid_linkables reachable_from 
-                           threaded2 upsweep_these
+                           threaded2 cleanup upsweep_these
 
         let ui3 = add_to_ui valid_linkables newLis
         let (CmThreaded pcs3 hst3 hit3) = threaded3
@@ -475,6 +612,13 @@ cmLoadModule cmstate1 rootname
              -- clean up after ourselves
              cleanTempFilesExcept verb (ppFilesFromSummaries modsDone)
 
+             -- issue a warning for the confusing case where the user said '-o foo'
+             -- but we're not going to do any linking.
+             ofile <- readIORef v_Output_file
+             when (ghci_mode == Batch && isJust ofile && not a_root_is_Main
+                    && verb > 0) $
+                hPutStrLn stderr "Warning: output was redirected with -o, but no output will be generated\nbecause there is no Main module."
+
              -- link everything together
               linkresult <- link ghci_mode dflags a_root_is_Main ui3 pls2
 
@@ -512,33 +656,36 @@ cmLoadModule cmstate1 rootname
 
 
 -- Finish up after a cmLoad.
---
+
+-- If the link failed, unload everything and return.
+cmLoadFinish ok (LinkFailed pls) hst hit ui mods ghci_mode pcs = do
+  dflags <- getDynFlags
+  new_pls <- CmLink.unload ghci_mode dflags [] pls 
+  new_state <- cmInit ghci_mode
+  return (new_state{ pcs=pcs, pls=new_pls }, False, [])
+
 -- Empty the interactive context and set the module context to the topmost
 -- newly loaded module, or the Prelude if none were loaded.
-cmLoadFinish ok linkresult hst hit ui mods ghci_mode pcs
-  = do case linkresult of {
-          LinkErrs _ _ -> panic "cmLoadModule: link failed (2)";
-          LinkOK pls   -> do
-
-       def_mod <- readIORef defaultCurrentModule
-       let current_mod = case mods of 
-                               []    -> def_mod
-                               (x:_) -> ms_mod x
-
-                  new_ic = emptyInteractiveContext current_mod
-
-           new_cmstate = CmState{ hst=hst, hit=hit, 
-                                  ui=ui, mg=mods,
-                                  gmode=ghci_mode, pcs=pcs, 
-                                 pls=pls,
-                                 ic = new_ic }
+cmLoadFinish ok (LinkOK pls) hst hit ui mods ghci_mode pcs
+  = do let new_cmstate = CmState{ hst=hst, hit=hit, ui=ui, mg=mods,
+                                  gmode=ghci_mode, pcs=pcs, pls=pls,
+                                 ic = emptyInteractiveContext }
            mods_loaded = map (moduleNameUserString.name_of_summary) mods
 
        return (new_cmstate, ok, mods_loaded)
-    }
 
+-- used to fish out the preprocess output files for the purposes
+-- of cleaning up.
 ppFilesFromSummaries summaries
-  = [ fn | Just fn <- map (ml_hspp_file . ms_location) summaries ]
+  = [ fn | Just fn <- map toPpFile summaries ]
+  where
+   toPpFile sum
+     | hspp /= ml_hs_file loc = hspp
+     | otherwise              = Nothing
+    where
+      loc  = ms_location sum
+      hspp = ml_hspp_file loc
+
 
 -----------------------------------------------------------------------------
 -- getValidLinkables
@@ -638,7 +785,13 @@ getValidLinkable old_linkables objects_allowed new_linkables summary
            src_date = ms_hs_date summary
 
           valid_linkable
-             =  filter (\l -> linkableTime l > src_date) linkable
+             =  filter (\l -> linkableTime l >= src_date) linkable
+               -- why '>=' rather than '>' above?  If the filesystem stores
+               -- times to the nearset second, we may occasionally find that
+               -- the object & source have the same modification time, 
+               -- especially if the source was automatically generated
+               -- and compiled.  Using >= is slightly unsafe, but it matches
+               -- make's behaviour.
 
        return (valid_linkable ++ new_linkables)
 
@@ -761,6 +914,7 @@ upsweep_mods :: GhciMode
              -> UnlinkedImage         -- valid linkables
              -> (ModuleName -> [ModuleName])  -- to construct downward closures
              -> CmThreaded            -- PCS & HST & HIT
+            -> IO ()                 -- how to clean up unwanted tmp files
              -> [SCC ModSummary]      -- mods to do (the worklist)
                                       -- ...... RETURNING ......
              -> IO (Bool{-complete success?-},
@@ -768,17 +922,17 @@ upsweep_mods :: GhciMode
                     [ModSummary],     -- mods which succeeded
                     [Linkable])       -- new linkables
 
-upsweep_mods ghci_mode dflags oldUI reachable_from threaded 
+upsweep_mods ghci_mode dflags oldUI reachable_from threaded cleanup
      []
    = return (True, threaded, [], [])
 
-upsweep_mods ghci_mode dflags oldUI reachable_from threaded 
+upsweep_mods ghci_mode dflags oldUI reachable_from threaded cleanup
      ((CyclicSCC ms):_)
    = do hPutStrLn stderr ("Module imports form a cycle for modules:\n\t" ++
                           unwords (map (moduleNameUserString.name_of_summary) ms))
         return (False, threaded, [], [])
 
-upsweep_mods ghci_mode dflags oldUI reachable_from threaded 
+upsweep_mods ghci_mode dflags oldUI reachable_from threaded cleanup
      ((AcyclicSCC mod):mods)
    = do --case threaded of
         --   CmThreaded pcsz hstz hitz
@@ -787,12 +941,16 @@ upsweep_mods ghci_mode dflags oldUI reachable_from threaded
         (threaded1, maybe_linkable) 
            <- upsweep_mod ghci_mode dflags oldUI threaded mod 
                           (reachable_from (name_of_summary mod))
+
+       -- remove unwanted tmp files between compilations
+       cleanup
+
         case maybe_linkable of
            Just linkable 
               -> -- No errors; do the rest
                  do (restOK, threaded2, modOKs, linkables) 
                        <- upsweep_mods ghci_mode dflags oldUI reachable_from 
-                                       threaded1 mods
+                                       threaded1 cleanup mods
                     return (restOK, threaded2, mod:modOKs, linkable:linkables)
            Nothing -- we got a compilation error; give up now
               -> return (False, threaded1, [], [])
@@ -811,7 +969,6 @@ upsweep_mod :: GhciMode
 upsweep_mod ghci_mode dflags oldUI threaded1 summary1 reachable_inc_me
    = do 
         let mod_name = name_of_summary summary1
-       let verb = verbosity dflags
 
         let (CmThreaded pcs1 hst1 hit1) = threaded1
         let old_iface = lookupUFM hit1 mod_name
@@ -888,10 +1045,9 @@ downwards_closure_of_module summaries root
 
          res = simple_transitive_closure (map toEdge summaries) [root]
      in
-         --trace (showSDoc (text "DC of mod" <+> ppr root
-         --                 <+> text "=" <+> ppr res)) (
+--         trace (showSDoc (text "DC of mod" <+> ppr root
+--                          <+> text "=" <+> ppr res)) $
          res
-         --)
 
 -- Calculate transitive closures from a set of roots given an adjacency list
 simple_transitive_closure :: Eq a => [(a,[a])] -> [a] -> [a]
@@ -931,22 +1087,26 @@ topological_sort include_source_imports summaries
          sccs
 
 
+-----------------------------------------------------------------------------
+-- Downsweep (dependency analysis)
+
 -- Chase downwards from the specified root set, returning summaries
 -- for all home modules encountered.  Only follow source-import
--- links.  Also returns a Bool to indicate whether any of the roots
--- are module Main.
-downsweep :: [FilePath] -> [ModSummary] -> IO ([ModSummary], Bool)
-downsweep rootNm old_summaries
-   = do rootSummaries <- mapM getRootSummary rootNm
-        let a_root_is_Main 
-               = any ((=="Main").moduleNameUserString.name_of_summary) 
-                     rootSummaries
+-- links.
+
+-- We pass in the previous collection of summaries, which is used as a
+-- cache to avoid recalculating a module summary if the source is
+-- unchanged.
+
+downsweep :: [FilePath] -> [ModSummary] -> IO [ModSummary]
+downsweep roots old_summaries
+   = do rootSummaries <- mapM getRootSummary roots
         all_summaries
            <- loop (concat (map ms_imps rootSummaries))
                (mkModuleEnv [ (mod, s) | s <- rootSummaries, 
                                          let mod = ms_mod s, isHomeModule mod 
                             ])
-        return (all_summaries, a_root_is_Main)
+        return all_summaries
      where
        getRootSummary :: FilePath -> IO ModSummary
        getRootSummary file
@@ -1003,8 +1163,8 @@ downsweep rootNm old_summaries
 
 -- We have two types of summarisation:
 --
---    * Summarise a file.  This is used for the root module passed to
---     cmLoadModule.  The file is read, and used to determine the root
+--    * Summarise a file.  This is used for the root module(s) passed to
+--     cmLoadModules.  The file is read, and used to determine the root
 --     module name.  The module name may differ from the filename.
 --
 --    * Summarise a module.  We are given a module name, and must provide
@@ -1016,10 +1176,10 @@ summariseFile file
    = do hspp_fn <- preprocess file
         (srcimps,imps,mod_name) <- getImportsFromFile hspp_fn
 
-        let (path, basename, ext) = splitFilename3 file
+        let (path, basename, _ext) = splitFilename3 file
 
-       Just (mod, location)
-          <- mkHomeModuleLocn mod_name (path ++ '/':basename) (Just file)
+       (mod, location)
+          <- mkHomeModuleLocn mod_name (path ++ '/':basename) file
 
         src_timestamp
            <- case ml_hs_file location of 
@@ -1039,14 +1199,7 @@ summarise mod location old_summary
    = do let hs_fn = unJust "summarise" (ml_hs_file location)
 
         case ml_hs_file location of {
-           Nothing -> do {
-               dflags <- getDynFlags;
-               when (verbosity dflags >= 1) $
-                   hPutStrLn stderr ("WARNING: module `" ++ 
-                       moduleUserString mod ++ "' has no source file.");
-               return Nothing;
-            };
-
+           Nothing -> noHsFileErr mod;
            Just src_fn -> do
 
         src_timestamp <- getModificationTime src_fn