1 {-# OPTIONS -fno-warn-incomplete-patterns -optc-DNON_POSIX_SOURCE #-}
3 -----------------------------------------------------------------------------
4 -- $Id: Main.hs,v 1.146 2005/02/02 13:27:07 simonpj Exp $
8 -- (c) The University of Glasgow 2002
10 -----------------------------------------------------------------------------
12 module Main (main) where
14 #include "HsVersions.h"
17 import InteractiveUI ( ghciWelcomeMsg, interactiveUI )
18 import DriverState ( isInteractiveMode )
22 import CompManager ( cmInit, cmLoadModules, cmDepAnal )
23 import HscTypes ( GhciMode(..) )
24 import Config ( cBooterVersion, cGhcUnregisterised, cProjectVersion )
25 import SysTools ( initSysTools, cleanTempFiles, normalisePath )
26 import Packages ( dumpPackages, initPackages, haskell98PackageId, PackageIdH(..) )
27 import DriverPipeline ( staticLink, doMkDLL, compileFile )
28 import DriverState ( isLinkMode,
29 isCompManagerMode, isInterpretiveMode,
30 buildStgToDo, findBuildTag, unregFlags,
31 v_GhcMode, GhcMode(..),
32 v_Keep_tmp_files, v_Ld_inputs, v_Ways,
33 v_Output_file, v_Output_hi, v_GhcLink,
34 verifyOutputFiles, GhcLink(..)
38 import DriverMkDepend ( doMkDependHS )
39 import DriverPhases ( Phase, isStopLn, isSourceFilename )
41 import DriverUtil ( add, handle, handleDyn, later, unknownFlagsErr )
42 import CmdLineOpts ( DynFlags(..), HscTarget(..), v_Static_hsc_opts,
44 import BasicTypes ( failed )
47 import Panic ( GhcException(..), panic, installSignalHandlers )
49 import DATA_IOREF ( readIORef, writeIORef )
50 import EXCEPTION ( throwDyn, Exception(..),
51 AsyncException(StackOverflow) )
53 -- Standard Haskell libraries
55 import System ( getArgs, exitWith, ExitCode(..) )
60 -----------------------------------------------------------------------------
63 -- new mkdependHS doesn't support all the options that the old one did (-X et al.)
64 -- time commands when run with -v
68 -- Win32 support: proper signal handling
69 -- make sure OPTIONS in .hs file propogate to .hc file if -C or -keep-hc-file-too
70 -- reading the package configuration file is too slow
73 -----------------------------------------------------------------------------
74 -- Differences vs. old driver:
76 -- No more "Enter your Haskell program, end with ^D (on a line of its own):"
77 -- consistency checking removed (may do this properly later)
80 -----------------------------------------------------------------------------
84 -- top-level exception handler: any unrecognised exception is a compiler bug.
85 handle (\exception -> do
88 -- an IO exception probably isn't our fault, so don't panic
89 IOException _ -> hPutStrLn stderr (show exception)
90 AsyncException StackOverflow ->
91 hPutStrLn stderr "stack overflow: use +RTS -K<size> to increase it"
92 _other -> hPutStr stderr (show (Panic (show exception)))
93 exitWith (ExitFailure 1)
96 -- all error messages are propagated as exceptions
100 PhaseFailed _ code -> exitWith code
101 Interrupted -> exitWith (ExitFailure 1)
102 _ -> do hPutStrLn stderr (show (dyn :: GhcException))
103 exitWith (ExitFailure 1)
106 installSignalHandlers
109 let (minusB_args, argv') = partition (prefixMatch "-B") argv
110 top_dir <- initSysTools minusB_args
112 -- Process all the other arguments, and get the source files
113 non_static <- processStaticFlags argv'
114 mode <- readIORef v_GhcMode
116 -- -O and --interactive are not a good combination
117 -- ditto with any kind of way selection
118 orig_ways <- readIORef v_Ways
119 when (notNull orig_ways && isInterpretiveMode mode) $
120 do throwDyn (UsageError
121 "--interactive can't be used with -prof, -ticky, -unreg or -smp.")
123 -- Find the build tag, and re-process the build-specific options.
124 -- Also add in flags for unregisterised compilation, if
125 -- GhcUnregisterised=YES.
126 way_opts <- findBuildTag
127 let unreg_opts | cGhcUnregisterised == "YES" = unregFlags
129 extra_non_static <- processStaticFlags (unreg_opts ++ way_opts)
131 -- Give the static flags to hsc
132 static_opts <- buildStaticHscOpts
133 writeIORef v_Static_hsc_opts static_opts
135 -- build the default DynFlags (these may be adjusted on a per
136 -- module basis by OPTIONS pragmas and settings in the interpreter).
138 stg_todo <- buildStgToDo
140 -- set the "global" HscTarget. The HscTarget can be further adjusted on a module
141 -- by module basis, using only the -fvia-C and -fasm flags. If the global
142 -- HscTarget is not HscC or HscAsm, -fvia-C and -fasm have no effect.
143 let dflags0 = defaultDynFlags
144 let lang = case mode of
145 DoInteractive -> HscInterpreted
146 DoEval _ -> HscInterpreted
147 _other -> hscTarget dflags0
149 let dflags1 = dflags0{ stgToDo = stg_todo,
151 -- leave out hscOutName for now
152 hscOutName = panic "Main.main:hscOutName not set",
153 verbosity = case mode of
158 -- The rest of the arguments are "dynamic"
159 -- Leftover ones are presumably files
160 (dflags2, fileish_args) <- processDynamicFlags
161 (extra_non_static ++ non_static) dflags1
163 -- make sure we clean up after ourselves
164 later (do forget_it <- readIORef v_Keep_tmp_files
165 unless forget_it $ do
166 cleanTempFiles dflags2
168 -- exceptions will be blocked while we clean the temporary files,
169 -- so there shouldn't be any difficulty if we receive further
172 -- Read the package config(s), and process the package-related
173 -- command-line flags
174 dflags <- initPackages dflags2
178 We split out the object files (.o, .dll) and add them
179 to v_Ld_inputs for use by the linker.
181 The following things should be considered compilation manager inputs:
183 - haskell source files (strings ending in .hs, .lhs or other
184 haskellish extension),
186 - module names (not forgetting hierarchical module names),
188 - and finally we consider everything not containing a '.' to be
189 a comp manager input, as shorthand for a .hs or .lhs filename.
191 Everything else is considered to be a linker object, and passed
192 straight through to the linker.
194 looks_like_an_input m = isSourceFilename m
195 || looksLikeModuleName m
198 -- To simplify the handling of filepaths, we normalise all filepaths right
199 -- away - e.g., for win32 platforms, backslashes are converted
200 -- into forward slashes.
201 normal_fileish_paths = map normalisePath fileish_args
202 (srcs, objs) = partition looks_like_an_input normal_fileish_paths
204 -- Note: have v_Ld_inputs maintain the order in which 'objs' occurred on
206 mapM_ (add v_Ld_inputs) (reverse objs)
208 ---------------- Display banners and configuration -----------
209 showBanners mode dflags static_opts
211 ---------------- Final sanity checking -----------
212 checkOptions mode srcs objs
214 ---------------- Do the business -----------
217 DoMake -> doMake dflags srcs
218 DoMkDependHS -> doMkDependHS dflags srcs
219 StopBefore p -> do { o_files <- compileFiles mode dflags srcs
220 ; doLink dflags p o_files }
222 DoInteractive -> noInteractiveError
223 DoEval _ -> noInteractiveError
225 noInteractiveError = throwDyn (CmdLineError "not built for interactive use")
227 DoInteractive -> interactiveUI dflags srcs Nothing
228 DoEval expr -> interactiveUI dflags srcs (Just expr)
231 -- -----------------------------------------------------------------------------
232 -- Option sanity checks
234 checkOptions :: GhcMode -> [String] -> [String] -> IO ()
235 -- Final sanity checking before kicking off a compilation (pipeline).
236 checkOptions mode srcs objs = do
237 -- Complain about any unknown flags
238 let unknown_opts = [ f | f@('-':_) <- srcs ]
239 when (notNull unknown_opts) (unknownFlagsErr unknown_opts)
242 ohi <- readIORef v_Output_hi
244 (isCompManagerMode mode || srcs `lengthExceeds` 1))
245 then throwDyn (UsageError "-ohi can only be used when compiling a single source file")
248 -- -o sanity checking
249 o_file <- readIORef v_Output_file
250 if (srcs `lengthExceeds` 1 && isJust o_file && not (isLinkMode mode))
251 then throwDyn (UsageError "can't apply -o to multiple source files")
254 -- Check that there are some input files
255 -- (except in the interactive case)
256 if null srcs && null objs && not (isInterpretiveMode mode)
257 then throwDyn (UsageError "no input files")
260 -- Verify that output files point somewhere sensible.
263 -- -----------------------------------------------------------------------------
264 -- Compile files in one-shot mode.
266 compileFiles :: GhcMode
268 -> [String] -- Source files
269 -> IO [String] -- Object files
270 compileFiles mode dflags srcs = mapM (compileFile mode dflags) srcs
273 doLink :: DynFlags -> Phase -> [FilePath] -> IO ()
274 doLink dflags stop_phase o_files
275 | not (isStopLn stop_phase)
276 = return () -- We stopped before the linking phase
279 = do { ghc_link <- readIORef v_GhcLink
282 StaticLink -> staticLink dflags o_files link_pkgs
283 MkDLL -> doMkDLL dflags o_files link_pkgs
286 -- Always link in the haskell98 package for static linking. Other
287 -- packages have to be specified via the -package flag.
289 | ExtPackage h98_id <- haskell98PackageId (pkgState dflags) = [h98_id]
293 -- ----------------------------------------------------------------------------
296 doMake :: DynFlags -> [String] -> IO ()
297 doMake dflags [] = throwDyn (UsageError "no input files")
298 doMake dflags srcs = do
299 state <- cmInit Batch dflags
300 graph <- cmDepAnal state srcs
301 (_, ok_flag, _) <- cmLoadModules state graph
302 when (failed ok_flag) (exitWith (ExitFailure 1))
305 -- ---------------------------------------------------------------------------
306 -- Various banners and verbosity output.
308 showBanners :: GhcMode -> DynFlags -> [String] -> IO ()
309 showBanners mode dflags static_opts = do
310 let verb = verbosity dflags
312 -- Show the GHCi banner
314 when (isInteractiveMode mode && verb >= 1) $
315 hPutStrLn stdout ghciWelcomeMsg
318 -- Display details of the configuration in verbose mode
320 do hPutStr stderr "Glasgow Haskell Compiler, Version "
321 hPutStr stderr cProjectVersion
322 hPutStr stderr ", for Haskell 98, compiled by GHC version "
323 hPutStrLn stderr cBooterVersion
329 hPutStrLn stderr ("Hsc static flags: " ++ unwords static_opts)