ghc/compiler/main/Main.hs

   1 {-# OPTIONS -fno-warn-incomplete-patterns #-}
   2 -----------------------------------------------------------------------------
   3 -- $Id: Main.hs,v 1.63 2001/04/26 13:52:57 simonmar Exp $
   4 --
   5 -- GHC Driver program
   6 --
   7 -- (c) Simon Marlow 2000
   8 --
   9 -----------------------------------------------------------------------------
  10
  11 -- with path so that ghc -M can find config.h
  12 #include "../includes/config.h"
  13
  14 module Main (main) where
  15
  16 #include "HsVersions.h"
  17
  18
  19 #ifdef GHCI
  20 import InteractiveUI
  21 #endif
  22
  23 #ifndef mingw32_TARGET_OS
  24 import Dynamic
  25 import Posix
  26 #endif
  27
  28 import CompManager
  29 import ParsePkgConf
  30 import DriverPipeline
  31 import DriverState
  32 import DriverFlags
  33 import DriverMkDepend
  34 import DriverUtil
  35 import Panic
  36 import DriverPhases     ( Phase(..), haskellish_file, objish_file )
  37 import CmdLineOpts
  38 import TmpFiles
  39 import Finder           ( initFinder )
  40 import CmStaticInfo
  41 import Config
  42 import Outputable
  43 import Util
  44
  45 import Concurrent
  46 import Directory
  47 import IOExts
  48 import Exception
  49
  50 import IO
  51 import Monad
  52 import List
  53 import System
  54 import Maybe
  55
  56
  57 -----------------------------------------------------------------------------
  58 -- Changes:
  59
  60 -- * -fglasgow-exts NO LONGER IMPLIES -package lang!!!  (-fglasgow-exts is a
  61 --   dynamic flag whereas -package is a static flag.)
  62
  63 -----------------------------------------------------------------------------
  64 -- ToDo:
  65
  66 -- -nohi doesn't work
  67 -- new mkdependHS doesn't support all the options that the old one did (-X et al.)
  68 -- time commands when run with -v
  69 -- split marker
  70 -- java generation
  71 -- user ways
  72 -- Win32 support: proper signal handling
  73 -- make sure OPTIONS in .hs file propogate to .hc file if -C or -keep-hc-file-too
  74 -- reading the package configuration file is too slow
  75 -- -K<size>
  76
  77 -----------------------------------------------------------------------------
  78 -- Differences vs. old driver:
  79
  80 -- No more "Enter your Haskell program, end with ^D (on a line of its own):"
  81 -- consistency checking removed (may do this properly later)
  82 -- removed -noC
  83 -- no -Ofile
  84
  85 -----------------------------------------------------------------------------
  86 -- Main loop
  87
  88 main =
  89   -- top-level exception handler: any unrecognised exception is a compiler bug.
  90   handle (\exception -> do hPutStr stderr (show (Panic (show exception)))
  91                            exitWith (ExitFailure 1)
  92          ) $ do
  93
  94   -- all error messages are propagated as exceptions
  95   handleDyn (\dyn -> case dyn of
  96                           PhaseFailed _phase code -> exitWith code
  97                           Interrupted -> exitWith (ExitFailure 1)
  98                           _ -> do hPutStrLn stderr (show (dyn :: GhcException))
  99                                   exitWith (ExitFailure 1)
 100             ) $ do
 101
 102    -- make sure we clean up after ourselves
 103    later (do  forget_it <- readIORef v_Keep_tmp_files
 104               unless forget_it $ do
 105               verb <- dynFlag verbosity
 106               cleanTempFiles verb
 107      ) $ do
 108         -- exceptions will be blocked while we clean the temporary files,
 109         -- so there shouldn't be any difficulty if we receive further
 110         -- signals.
 111
 112         -- install signal handlers
 113    main_thread <- myThreadId
 114 #ifndef mingw32_TARGET_OS
 115    let sig_handler = Catch (throwTo main_thread
 116                                 (DynException (toDyn Interrupted)))
 117    installHandler sigQUIT sig_handler Nothing
 118    installHandler sigINT  sig_handler Nothing
 119 #endif
 120
 121    argv   <- getArgs
 122
 123         -- grab any -B options from the command line first
 124    argv'  <- setTopDir argv
 125    top_dir <- readIORef v_TopDir
 126
 127    let installed s = top_dir ++ '/':s
 128        inplace s   = top_dir ++ '/':cCURRENT_DIR ++ '/':s
 129
 130        installed_pkgconfig = installed ("package.conf")
 131        inplace_pkgconfig   = inplace (cGHC_DRIVER_DIR ++ "/package.conf.inplace")
 132
 133         -- discover whether we're running in a build tree or in an installation,
 134         -- by looking for the package configuration file.
 135    am_installed <- doesFileExist installed_pkgconfig
 136
 137    if am_installed
 138         then writeIORef v_Path_package_config installed_pkgconfig
 139         else do am_inplace <- doesFileExist inplace_pkgconfig
 140                 if am_inplace
 141                     then writeIORef v_Path_package_config inplace_pkgconfig
 142                     else throwDyn (InstallationError
 143                                      ("Can't find package.conf in " ++
 144                                       inplace_pkgconfig))
 145
 146         -- set the location of our various files
 147    if am_installed
 148         then do writeIORef v_Path_usage (installed "ghc-usage.txt")
 149                 writeIORef v_Pgm_L (installed "unlit")
 150                 writeIORef v_Pgm_m (installed "ghc-asm")
 151                 writeIORef v_Pgm_s (installed "ghc-split")
 152
 153         else do writeIORef v_Path_usage (inplace (cGHC_DRIVER_DIR ++ "/ghc-usage.txt"))
 154                 writeIORef v_Pgm_L (inplace cGHC_UNLIT)
 155                 writeIORef v_Pgm_m (inplace cGHC_MANGLER)
 156                 writeIORef v_Pgm_s (inplace cGHC_SPLIT)
 157
 158         -- read the package configuration
 159    conf_file <- readIORef v_Path_package_config
 160    r <- parsePkgConf conf_file
 161    case r of {
 162         Left err -> throwDyn (InstallationError (showSDoc err));
 163         Right pkg_details -> do
 164
 165    writeIORef v_Package_details pkg_details
 166
 167         -- find the phase to stop after (i.e. -E, -C, -c, -S flags)
 168    (flags2, mode, stop_flag) <- getGhcMode argv'
 169    writeIORef v_GhcMode mode
 170
 171         -- Show the GHCi banner?
 172 #  ifdef GHCI
 173    when (mode == DoInteractive) $
 174       hPutStrLn stdout ghciWelcomeMsg
 175 #  endif
 176
 177         -- process all the other arguments, and get the source files
 178    non_static <- processArgs static_flags flags2 []
 179
 180         -- -O and --interactive are not a good combination
 181         -- ditto with any kind of way selection
 182    orig_opt_level <- readIORef v_OptLevel
 183    when (orig_opt_level > 0 && mode == DoInteractive) $
 184       do putStr "warning: -O conflicts with --interactive; -O turned off.\n"
 185          writeIORef v_OptLevel 0
 186    orig_ways <- readIORef v_Ways
 187    when (not (null orig_ways) && mode == DoInteractive) $
 188       do throwDyn (UsageError
 189                    "--interactive can't be used with -prof, -ticky, -unreg or -smp.")
 190
 191         -- Find the build tag, and re-process the build-specific options.
 192         -- Also add in flags for unregisterised compilation, if
 193         -- GhcUnregisterised=YES.
 194    way_opts <- findBuildTag
 195    let unreg_opts | cGhcUnregisterised == "YES" = unregFlags
 196                   | otherwise = []
 197    way_non_static <- processArgs static_flags (unreg_opts ++ way_opts) []
 198
 199         -- give the static flags to hsc
 200    static_opts <- buildStaticHscOpts
 201    writeIORef v_Static_hsc_opts static_opts
 202
 203    -- build the default DynFlags (these may be adjusted on a per
 204    -- module basis by OPTIONS pragmas and settings in the interpreter).
 205
 206    core_todo <- buildCoreToDo
 207    stg_todo  <- buildStgToDo
 208
 209    -- set the "global" HscLang.  The HscLang can be further adjusted on a module
 210    -- by module basis, using only the -fvia-C and -fasm flags.  If the global
 211    -- HscLang is not HscC or HscAsm, -fvia-C and -fasm have no effect.
 212    opt_level  <- readIORef v_OptLevel
 213
 214
 215    let lang = case mode of
 216                  StopBefore HCc -> HscC
 217                  DoInteractive  -> HscInterpreted
 218                  _other        | opt_level >= 1  -> HscC  -- -O implies -fvia-C
 219                                | otherwise       -> defaultHscLang
 220
 221    writeIORef v_DynFlags
 222         defaultDynFlags{ coreToDo = core_todo,
 223                          stgToDo  = stg_todo,
 224                          hscLang  = lang,
 225                          -- leave out hscOutName for now
 226                          hscOutName = panic "Main.main:hscOutName not set",
 227
 228                          verbosity = case mode of
 229                                         DoInteractive -> 1
 230                                         DoMake        -> 1
 231                                         _other        -> 0,
 232                         }
 233
 234         -- the rest of the arguments are "dynamic"
 235    srcs <- processArgs dynamic_flags (way_non_static ++ non_static) []
 236         -- save the "initial DynFlags" away
 237    init_dyn_flags <- readIORef v_DynFlags
 238    writeIORef v_InitDynFlags init_dyn_flags
 239
 240         -- complain about any unknown flags
 241    mapM unknownFlagErr [ f | f@('-':_) <- srcs ]
 242
 243    verb <- dynFlag verbosity
 244
 245    when (verb >= 2)
 246         (do hPutStr stderr "Glasgow Haskell Compiler, Version "
 247             hPutStr stderr cProjectVersion
 248             hPutStr stderr ", for Haskell 98, compiled by GHC version "
 249             hPutStrLn stderr cBooterVersion)
 250
 251    when (verb >= 2)
 252         (hPutStrLn stderr ("Using package config file: " ++ conf_file))
 253
 254    when (verb >= 3)
 255         (hPutStrLn stderr ("Hsc static flags: " ++ unwords static_opts))
 256
 257         -- initialise the finder
 258    pkg_avails <- getPackageInfo
 259    initFinder pkg_avails
 260
 261         -- mkdependHS is special
 262    when (mode == DoMkDependHS) beginMkDependHS
 263
 264         -- -ohi sanity checking
 265    ohi    <- readIORef v_Output_hi
 266    if (isJust ohi &&
 267         (mode == DoMake || mode == DoInteractive || length srcs > 1))
 268         then throwDyn (UsageError "-ohi can only be used when compiling a single source file")
 269         else do
 270
 271         -- make/interactive require invoking the compilation manager
 272    if (mode == DoMake)        then beginMake srcs        else do
 273    if (mode == DoInteractive) then beginInteractive srcs else do
 274
 275         -- -o sanity checking
 276    o_file <- readIORef v_Output_file
 277    if (length srcs > 1 && isJust o_file && mode /= DoLink && mode /= DoMkDLL)
 278         then throwDyn (UsageError "can't apply -o to multiple source files")
 279         else do
 280
 281    if null srcs then throwDyn (UsageError "no input files") else do
 282
 283    let compileFile src = do
 284           writeIORef v_DynFlags init_dyn_flags
 285
 286           exists <- doesFileExist src
 287           when (not exists) $
 288                 throwDyn (CmdLineError ("file `" ++ src ++ "' does not exist"))
 289
 290           -- We compile in two stages, because the file may have an
 291           -- OPTIONS pragma that affects the compilation pipeline (eg. -fvia-C)
 292           let (basename, suffix) = splitFilename src
 293
 294           -- just preprocess
 295           pp <- if not (haskellish_file src) || mode == StopBefore Hsc
 296                         then return src else do
 297                 phases <- genPipeline (StopBefore Hsc) stop_flag
 298                             False{-not persistent-} defaultHscLang src
 299                 pipeLoop phases src False{-no linking-} False{-no -o flag-}
 300                         basename suffix
 301
 302           -- rest of compilation
 303           dyn_flags <- readIORef v_DynFlags
 304           phases <- genPipeline mode stop_flag True (hscLang dyn_flags) pp
 305           r <- pipeLoop phases pp (mode==DoLink || mode==DoMkDLL) True{-use -o flag-}
 306                         basename suffix
 307           return r
 308
 309    o_files <- mapM compileFile srcs
 310
 311    when (mode == DoMkDependHS) endMkDependHS
 312    when (mode == DoLink) (doLink o_files)
 313    when (mode == DoMkDLL) (doMkDLL o_files)
 314   }
 315         -- grab the last -B option on the command line, and
 316         -- set topDir to its value.
 317 setTopDir :: [String] -> IO [String]
 318 setTopDir args = do
 319   let (minusbs, others) = partition (prefixMatch "-B") args
 320   (case minusbs of
 321     []   -> throwDyn (InstallationError ("missing -B<dir> option"))
 322     some -> writeIORef v_TopDir (drop 2 (last some)))
 323   return others
 324
 325 beginMake :: [String] -> IO ()
 326 beginMake fileish_args
 327   = do let (objs, mods) = partition objish_file fileish_args
 328        mapM (add v_Ld_inputs) objs
 329
 330        case mods of
 331          []    -> throwDyn (UsageError "no input files")
 332          [mod] -> do state <- cmInit Batch
 333                      cmLoadModule state mod
 334                      return ()
 335          _     -> throwDyn (UsageError "only one module allowed with --make")
 336
 337
 338 beginInteractive :: [String] -> IO ()
 339 #ifndef GHCI
 340 beginInteractive = throwDyn (CmdLineError "not built for interactive use")
 341 #else
 342 beginInteractive fileish_args
 343   = do minus_ls <- readIORef v_Cmdline_libraries
 344
 345        let (objs, mods) = partition objish_file fileish_args
 346            libs = map Left objs ++ map Right minus_ls
 347
 348        state <- cmInit Interactive
 349        case mods of
 350           []    -> interactiveUI state Nothing    libs
 351           [mod] -> interactiveUI state (Just mod) libs
 352           _     -> throwDyn (UsageError
 353                              "only one module allowed with --interactive")
 354 #endif