[project @ 2001-10-25 11:47:03 by simonmar]
[ghc-hetmet.git] / ghc / compiler / compMan / CompManager.lhs
index 1ddddfd..54c4344 100644 (file)
@@ -4,6 +4,7 @@
 \section[CompManager]{The Compilation Manager}
 
 \begin{code}
+{-# OPTIONS -fvia-C #-}
 module CompManager ( 
     cmInit,      -- :: GhciMode -> IO CmState
 
@@ -18,7 +19,8 @@ module CompManager (
 #ifdef GHCI
     cmInfoThing,  -- :: CmState -> DynFlags -> String -> IO (Maybe TyThing)
 
-    cmRunStmt,   -- :: CmState -> DynFlags -> String -> IO (CmState, [Name])
+    CmRunResult(..),
+    cmRunStmt,   -- :: CmState -> DynFlags -> String -> IO (CmState, CmRunResult)
 
     cmTypeOfExpr, -- :: CmState -> DynFlags -> String
                  -- -> IO (CmState, Maybe String)
@@ -38,6 +40,7 @@ import CmLink
 import CmTypes
 import DriverPipeline
 import DriverFlags     ( getDynFlags )
+import DriverState     ( v_Output_file )
 import DriverPhases
 import DriverUtil
 import Finder
@@ -47,16 +50,11 @@ import HscMain              ( initPersistentCompilerState, hscThing )
 import HscMain         ( initPersistentCompilerState )
 #endif
 import HscTypes
-import RnEnv           ( unQualInScope )
-import Id              ( idType, idName )
 import Name            ( Name, NamedThing(..), nameRdrName, nameModule,
                          isHomePackageName )
-import NameEnv
 import RdrName         ( lookupRdrEnv, emptyRdrEnv )
 import Module
 import GetImports
-import Type            ( tidyType )
-import VarEnv          ( emptyTidyEnv )
 import UniqFM
 import Unique          ( Uniquable )
 import Digraph         ( SCC(..), stronglyConnComp, flattenSCC, flattenSCCs )
@@ -64,16 +62,25 @@ import ErrUtils             ( showPass )
 import SysTools                ( cleanTempFilesExcept )
 import Util
 import Outputable
-import BasicTypes      ( Fixity, defaultFixity )
 import Panic
 import CmdLineOpts     ( DynFlags(..) )
 
 import IOExts
 
 #ifdef GHCI
+import Id              ( idType, idName )
+import NameEnv
+import Type            ( tidyType )
+import VarEnv          ( emptyTidyEnv )
+import RnEnv           ( unQualInScope )
+import BasicTypes      ( Fixity, defaultFixity )
 import Interpreter     ( HValue )
 import HscMain         ( hscStmt )
 import PrelGHC         ( unsafeCoerce# )
+
+import Foreign
+import CForeign
+import Exception       ( Exception, try )
 #endif
 
 -- lang
@@ -207,10 +214,15 @@ cmInfoThing cmstate dflags id
 -- 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 
+  | CmRunDeadlocked            -- statement deadlocked
+  | 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, 
@@ -221,7 +233,7 @@ cmRunStmt cmstate dflags expr
            <- 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
@@ -248,17 +260,56 @@ cmRunStmt cmstate dflags expr
 
                -- 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
+                       | err == dEADLOCKED
+                       -> return ( cmstate{ pcs=new_pcs, ic=new_ic }, 
+                                   CmRunDeadlocked )
+                       | otherwise
+                       -> 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", which amounts to calling into
+-- the RTS to request a new main thread.  The main benefit is that we
+-- get to detect a deadlock this way, but also there's no danger that
+-- exceptions raised by the expression can affect the interpreter.
+
+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))
+
+-- ToDo: slurp this in from ghc/includes/RtsAPI.h somehow
+dEADLOCKED = 4 :: Int
+
+foreign import "rts_evalStableIO"  {- safe -}
+  rts_evalStableIO :: StablePtr (IO a) -> Ptr (StablePtr a) -> IO CInt
+  -- more informative than the C type!
 #endif
 
 -----------------------------------------------------------------------------
@@ -281,7 +332,6 @@ cmTypeOfExpr cmstate dflags expr
                tidy_ty = tidyType emptyTidyEnv ty
    where
        CmState{ hst=hst, hit=hit, pcs=pcs, ic=ic } = cmstate
-#endif
 
 getUnqual pcs hit ic
    = case lookupIfaceByModName hit pit modname of
@@ -290,6 +340,7 @@ getUnqual pcs hit ic
  where
     pit = pcs_PIT pcs
     modname = moduleName (ic_module ic)
+#endif
 
 -----------------------------------------------------------------------------
 -- cmTypeOfName: returns a string representing the type of a name.
@@ -344,21 +395,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.
 
@@ -395,7 +431,6 @@ cmLoadModule cmstate1 rootnames
        -- 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
 
@@ -518,6 +553,13 @@ cmLoadModule cmstate1 rootnames
              -- 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
 
@@ -875,7 +917,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
@@ -1079,7 +1120,7 @@ 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
 
        (mod, location)
           <- mkHomeModuleLocn mod_name (path ++ '/':basename) file