blob: bdcdcb731cfe9a269b96632824dcb56f969f40f5 [file] [log] [blame]
Daniel Dunbar378530c2009-01-05 19:53:30 +00001import os
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +00002import platform
Daniel Dunbar378530c2009-01-05 19:53:30 +00003import sys
4import tempfile
5from pprint import pprint
6
7###
8
9import Arguments
10import Jobs
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +000011import HostInfo
Daniel Dunbar378530c2009-01-05 19:53:30 +000012import Phases
13import Tools
14import Types
15import Util
16
17# FIXME: Clean up naming of options and arguments. Decide whether to
18# rename Option and be consistent about use of Option/Arg.
19
20####
21
Daniel Dunbar378530c2009-01-05 19:53:30 +000022class Driver(object):
Daniel Dunbara0026f22009-01-16 23:12:12 +000023 def __init__(self, driverName, driverDir):
24 self.driverName = driverName
25 self.driverDir = driverDir
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +000026 self.hostInfo = None
Daniel Dunbare9f1a692009-01-06 06:12:13 +000027 self.parser = Arguments.OptionParser()
Daniel Dunbar7c2f91b2009-01-14 01:03:36 +000028 self.cccHostBits = self.cccHostMachine = None
29 self.cccHostSystem = self.cccHostRelease = None
30 self.cccCXX = False
Daniel Dunbar4c751dc2009-01-14 01:32:05 +000031 self.cccEcho = False
Daniel Dunbar7c2f91b2009-01-14 01:03:36 +000032 self.cccFallback = False
Daniel Dunbar72999172009-01-29 23:54:06 +000033 self.cccNoClang = self.cccNoClangCXX = self.cccNoClangPreprocessor = False
34 self.cccClangArchs = None
Daniel Dunbar378530c2009-01-05 19:53:30 +000035
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +000036 # Certain options suppress the 'no input files' warning.
37 self.suppressMissingInputWarning = False
38
Daniel Dunbardbc9ee92009-01-09 22:21:24 +000039 # Host queries which can be forcibly over-riden by the user for
40 # testing purposes.
41 #
42 # FIXME: We should make sure these are drawn from a fixed set so
43 # that nothing downstream ever plays a guessing game.
44
45 def getHostBits(self):
46 if self.cccHostBits:
47 return self.cccHostBits
48
49 return platform.architecture()[0].replace('bit','')
50
51 def getHostMachine(self):
52 if self.cccHostMachine:
53 return self.cccHostMachine
54
55 machine = platform.machine()
56 # Normalize names.
57 if machine == 'Power Macintosh':
58 return 'ppc'
Daniel Dunbar405327e2009-01-27 19:29:51 +000059 if machine == 'x86_64':
60 return 'i386'
Daniel Dunbardbc9ee92009-01-09 22:21:24 +000061 return machine
62
63 def getHostSystemName(self):
64 if self.cccHostSystem:
65 return self.cccHostSystem
66
67 return platform.system().lower()
68
Daniel Dunbarc2148562009-01-12 04:21:12 +000069 def getHostReleaseName(self):
70 if self.cccHostRelease:
71 return self.cccHostRelease
72
73 return platform.release()
74
Daniel Dunbar4c751dc2009-01-14 01:32:05 +000075 def getenvBool(self, name):
76 var = os.getenv(name)
77 if not var:
78 return False
79
80 try:
81 return bool(int(var))
82 except:
83 return False
84
Daniel Dunbardbc9ee92009-01-09 22:21:24 +000085 ###
86
Daniel Dunbarf677a602009-01-21 02:03:52 +000087 def getFilePath(self, name, toolChain=None):
88 tc = toolChain or self.toolChain
89 for p in tc.filePathPrefixes:
90 path = os.path.join(p, name)
91 if os.path.exists(path):
92 return path
93 return name
94
95 def getProgramPath(self, name, toolChain=None):
96 tc = toolChain or self.toolChain
97 for p in tc.programPathPrefixes:
98 path = os.path.join(p, name)
99 if os.path.exists(path):
100 return path
101 return name
102
103 ###
104
Daniel Dunbar74727872009-01-06 01:35:44 +0000105 def run(self, argv):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000106 # FIXME: Things to support from environment: GCC_EXEC_PREFIX,
107 # COMPILER_PATH, LIBRARY_PATH, LPATH, CC_PRINT_OPTIONS,
108 # QA_OVERRIDE_GCC3_OPTIONS, ...?
109
110 # FIXME: -V and -b processing
111
112 # Handle some special -ccc- options used for testing which are
113 # only allowed at the beginning of the command line.
114 cccPrintOptions = False
115 cccPrintPhases = False
Daniel Dunbardbc9ee92009-01-09 22:21:24 +0000116
117 # FIXME: How to handle override of host? ccc specific options?
118 # Abuse -b?
Daniel Dunbar72999172009-01-29 23:54:06 +0000119 arg = os.getenv('CCC_ADD_ARGS')
120 if arg:
121 args = filter(None, map(str.strip, arg.split(',')))
122 argv = args + argv
Daniel Dunbar4c751dc2009-01-14 01:32:05 +0000123
Daniel Dunbar74727872009-01-06 01:35:44 +0000124 while argv and argv[0].startswith('-ccc-'):
Daniel Dunbara0026f22009-01-16 23:12:12 +0000125 fullOpt,argv = argv[0],argv[1:]
126 opt = fullOpt[5:]
Daniel Dunbar378530c2009-01-05 19:53:30 +0000127
128 if opt == 'print-options':
129 cccPrintOptions = True
130 elif opt == 'print-phases':
131 cccPrintPhases = True
Daniel Dunbar7c2f91b2009-01-14 01:03:36 +0000132 elif opt == 'cxx':
133 self.cccCXX = True
Daniel Dunbar4c751dc2009-01-14 01:32:05 +0000134 elif opt == 'echo':
135 self.cccEcho = True
Daniel Dunbar7c2f91b2009-01-14 01:03:36 +0000136 elif opt == 'fallback':
137 self.cccFallback = True
Daniel Dunbar72999172009-01-29 23:54:06 +0000138
139 elif opt == 'no-clang':
140 self.cccNoClang = True
141 elif opt == 'no-clang-cxx':
142 self.cccNoClangCXX = True
143 elif opt == 'no-clang-cpp':
144 self.cccNoClangPreprocessor = True
145 elif opt == 'clang-archs':
146 self.cccClangArchs,argv = argv[0].split(','),argv[1:]
147
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +0000148 elif opt == 'host-bits':
Daniel Dunbardbc9ee92009-01-09 22:21:24 +0000149 self.cccHostBits,argv = argv[0],argv[1:]
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +0000150 elif opt == 'host-machine':
Daniel Dunbardbc9ee92009-01-09 22:21:24 +0000151 self.cccHostMachine,argv = argv[0],argv[1:]
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +0000152 elif opt == 'host-system':
Daniel Dunbardbc9ee92009-01-09 22:21:24 +0000153 self.cccHostSystem,argv = argv[0],argv[1:]
Daniel Dunbarc2148562009-01-12 04:21:12 +0000154 elif opt == 'host-release':
155 self.cccHostRelease,argv = argv[0],argv[1:]
Daniel Dunbar378530c2009-01-05 19:53:30 +0000156 else:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000157 raise Arguments.InvalidArgumentsError("invalid option: %r" % fullOpt)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000158
Daniel Dunbardbc9ee92009-01-09 22:21:24 +0000159 self.hostInfo = HostInfo.getHostInfo(self)
Daniel Dunbar08dea462009-01-10 02:07:54 +0000160 self.toolChain = self.hostInfo.getToolChain()
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +0000161
Daniel Dunbar74727872009-01-06 01:35:44 +0000162 args = self.parser.parseArgs(argv)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000163
164 # FIXME: Ho hum I have just realized -Xarch_ is broken. We really
165 # need to reparse the Arguments after they have been expanded by
166 # -Xarch. How is this going to work?
167 #
168 # Scratch that, we aren't going to do that; it really disrupts the
169 # organization, doesn't consistently work with gcc-dd, and is
170 # confusing. Instead we are going to enforce that -Xarch_ is only
171 # used with options which do not alter the driver behavior. Let's
172 # hope this is ok, because the current architecture is a little
173 # tied to it.
174
175 if cccPrintOptions:
Daniel Dunbar74727872009-01-06 01:35:44 +0000176 self.printOptions(args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000177 sys.exit(0)
178
Daniel Dunbar74727872009-01-06 01:35:44 +0000179 self.handleImmediateOptions(args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000180
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +0000181 if self.hostInfo.useDriverDriver():
Daniel Dunbar74727872009-01-06 01:35:44 +0000182 phases = self.buildPipeline(args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000183 else:
Daniel Dunbar74727872009-01-06 01:35:44 +0000184 phases = self.buildNormalPipeline(args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000185
186 if cccPrintPhases:
Daniel Dunbar74727872009-01-06 01:35:44 +0000187 self.printPhases(phases, args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000188 sys.exit(0)
Daniel Dunbar4ed45ea2009-01-09 01:00:40 +0000189
Daniel Dunbar378530c2009-01-05 19:53:30 +0000190 if 0:
191 print Util.pprint(phases)
192
Daniel Dunbar74727872009-01-06 01:35:44 +0000193 jobs = self.bindPhases(phases, args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000194
195 # FIXME: We should provide some basic sanity checking of the
196 # pipeline as a "verification" sort of stage. For example, the
197 # pipeline should never end up writing to an output file in two
198 # places (I think). The pipeline should also never end up writing
199 # to an output file that is an input.
200 #
201 # This is intended to just be a "verify" step, not a functionality
202 # step. It should catch things like the driver driver not
203 # preventing -save-temps, but it shouldn't change behavior (so we
204 # can turn it off in Release-Asserts builds).
205
206 # Print in -### syntax.
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000207 hasHashHashHash = args.getLastArg(self.parser.hashHashHashOption)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000208 if hasHashHashHash:
209 self.claim(hasHashHashHash)
210 for j in jobs.iterjobs():
211 if isinstance(j, Jobs.Command):
Daniel Dunbard87b6ec2009-01-12 19:36:35 +0000212 print >>sys.stderr, ' "%s"' % '" "'.join(j.getArgv())
Daniel Dunbar378530c2009-01-05 19:53:30 +0000213 elif isinstance(j, Jobs.PipedJob):
214 for c in j.commands:
Daniel Dunbard87b6ec2009-01-12 19:36:35 +0000215 print >>sys.stderr, ' "%s" %c' % ('" "'.join(c.getArgv()),
216 "| "[c is j.commands[-1]])
Daniel Dunbar378530c2009-01-05 19:53:30 +0000217 elif not isinstance(j, JobList):
218 raise ValueError,'Encountered unknown job.'
219 sys.exit(0)
220
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +0000221 vArg = args.getLastArg(self.parser.vOption)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000222 for j in jobs.iterjobs():
223 if isinstance(j, Jobs.Command):
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +0000224 if vArg or self.cccEcho:
225 print >>sys.stderr, ' '.join(map(str,j.getArgv()))
Anders Carlsson76613bf2009-01-18 02:54:17 +0000226 sys.stderr.flush()
Daniel Dunbarb421dba2009-01-07 18:40:45 +0000227 res = os.spawnvp(os.P_WAIT, j.executable, j.getArgv())
Daniel Dunbar378530c2009-01-05 19:53:30 +0000228 if res:
229 sys.exit(res)
230 elif isinstance(j, Jobs.PipedJob):
Daniel Dunbare19ed8e2009-01-17 02:02:35 +0000231 import subprocess
232 procs = []
233 for sj in j.commands:
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +0000234 if vArg or self.cccEcho:
235 print >> sys.stderr, ' '.join(map(str,sj.getArgv()))
Daniel Dunbare19ed8e2009-01-17 02:02:35 +0000236 sys.stdout.flush()
237
238 if not procs:
239 stdin = None
240 else:
241 stdin = procs[-1].stdout
242 if sj is j.commands[-1]:
243 stdout = None
244 else:
245 stdout = subprocess.PIPE
246 procs.append(subprocess.Popen(sj.getArgv(),
247 executable=sj.executable,
248 stdin=stdin,
249 stdout=stdout))
250 for proc in procs:
251 res = proc.wait()
252 if res:
253 sys.exit(res)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000254 else:
255 raise ValueError,'Encountered unknown job.'
256
257 def claim(self, option):
258 # FIXME: Move to OptionList once introduced and implement.
259 pass
260
261 def warning(self, message):
Daniel Dunbara0026f22009-01-16 23:12:12 +0000262 print >>sys.stderr,'%s: %s' % (self.driverName, message)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000263
Daniel Dunbar74727872009-01-06 01:35:44 +0000264 def printOptions(self, args):
265 for i,arg in enumerate(args):
Daniel Dunbar74727872009-01-06 01:35:44 +0000266 if isinstance(arg, Arguments.MultipleValuesArg):
267 values = list(args.getValues(arg))
268 elif isinstance(arg, Arguments.ValueArg):
269 values = [args.getValue(arg)]
270 elif isinstance(arg, Arguments.JoinedAndSeparateValuesArg):
271 values = [args.getJoinedValue(arg), args.getSeparateValue(arg)]
Daniel Dunbar378530c2009-01-05 19:53:30 +0000272 else:
273 values = []
Daniel Dunbar927a5092009-01-06 02:30:10 +0000274 print 'Option %d - Name: "%s", Values: {%s}' % (i, arg.opt.name,
Daniel Dunbar378530c2009-01-05 19:53:30 +0000275 ', '.join(['"%s"' % v
276 for v in values]))
277
Daniel Dunbar74727872009-01-06 01:35:44 +0000278 def printPhases(self, phases, args):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000279 def printPhase(p, f, steps, arch=None):
280 if p in steps:
281 return steps[p]
282 elif isinstance(p, Phases.BindArchAction):
283 for kid in p.inputs:
284 printPhase(kid, f, steps, p.arch)
285 steps[p] = len(steps)
286 return
287
288 if isinstance(p, Phases.InputAction):
289 phaseName = 'input'
Daniel Dunbar74727872009-01-06 01:35:44 +0000290 inputStr = '"%s"' % args.getValue(p.filename)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000291 else:
292 phaseName = p.phase.name
293 inputs = [printPhase(i, f, steps, arch)
294 for i in p.inputs]
295 inputStr = '{%s}' % ', '.join(map(str, inputs))
296 if arch is not None:
Daniel Dunbar74727872009-01-06 01:35:44 +0000297 phaseName += '-' + args.getValue(arch)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000298 steps[p] = index = len(steps)
299 print "%d: %s, %s, %s" % (index,phaseName,inputStr,p.type.name)
300 return index
301 steps = {}
302 for phase in phases:
303 printPhase(phase, sys.stdout, steps)
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +0000304
305 def printVersion(self):
306 # FIXME: Print default target triple.
307 print >>sys.stderr,'ccc version 1.0'
Daniel Dunbar378530c2009-01-05 19:53:30 +0000308
Daniel Dunbar74727872009-01-06 01:35:44 +0000309 def handleImmediateOptions(self, args):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000310 # FIXME: Some driver Arguments are consumed right off the bat,
311 # like -dumpversion. Currently the gcc-dd handles these
312 # poorly, so we should be ok handling them upfront instead of
313 # after driver-driver level dispatching.
314 #
315 # FIXME: The actual order of these options in gcc is all over the
316 # place. The -dump ones seem to be first and in specification
317 # order, but there are other levels of precedence. For example,
318 # -print-search-dirs is evaluated before -print-prog-name=,
319 # regardless of order (and the last instance of -print-prog-name=
320 # wins verse itself).
321 #
322 # FIXME: Do we want to report "argument unused" type errors in the
323 # presence of things like -dumpmachine and -print-search-dirs?
324 # Probably not.
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +0000325 if (args.getLastArg(self.parser.vOption) or
326 args.getLastArg(self.parser.hashHashHashOption)):
327 self.printVersion()
328 self.suppressMissingInputWarning = True
329
Daniel Dunbarf677a602009-01-21 02:03:52 +0000330 arg = (args.getLastArg(self.parser.dumpmachineOption) or
331 args.getLastArg(self.parser.dumpversionOption) or
332 args.getLastArg(self.parser.printSearchDirsOption))
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000333 if arg:
Daniel Dunbarf677a602009-01-21 02:03:52 +0000334 raise NotImplementedError('%s unsupported' % arg.opt.name)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000335
Daniel Dunbarf677a602009-01-21 02:03:52 +0000336 arg = (args.getLastArg(self.parser.dumpspecsOption) or
337 args.getLastArg(self.parser.printMultiDirectoryOption) or
Daniel Dunbar73fb9072009-01-23 02:00:46 +0000338 args.getLastArg(self.parser.printMultiOsDirectoryOption) or
Daniel Dunbarf677a602009-01-21 02:03:52 +0000339 args.getLastArg(self.parser.printMultiLibOption))
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000340 if arg:
Daniel Dunbarf677a602009-01-21 02:03:52 +0000341 raise Arguments.InvalidArgumentsError('%s unsupported by this driver' % arg.opt.name)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000342
343 arg = args.getLastArg(self.parser.printFileNameOption)
344 if arg:
Daniel Dunbarf677a602009-01-21 02:03:52 +0000345 print self.getFilePath(args.getValue(arg))
346 sys.exit(0)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000347
348 arg = args.getLastArg(self.parser.printProgNameOption)
349 if arg:
Daniel Dunbarf677a602009-01-21 02:03:52 +0000350 print self.getProgramPath(args.getValue(arg))
351 sys.exit(0)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000352
Daniel Dunbarf677a602009-01-21 02:03:52 +0000353 arg = args.getLastArg(self.parser.printLibgccFileNameOption)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000354 if arg:
Daniel Dunbarf677a602009-01-21 02:03:52 +0000355 print self.getFilePath('libgcc.a')
356 sys.exit(0)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000357
Daniel Dunbar74727872009-01-06 01:35:44 +0000358 def buildNormalPipeline(self, args):
Daniel Dunbar90c72cd2009-01-21 01:07:49 +0000359 hasAnalyze = args.getLastArg(self.parser.analyzeOption)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000360 hasCombine = args.getLastArg(self.parser.combineOption)
Daniel Dunbar34f60f62009-01-26 17:09:15 +0000361 hasEmitLLVM = args.getLastArg(self.parser.emitLLVMOption)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000362 hasSyntaxOnly = args.getLastArg(self.parser.syntaxOnlyOption)
363 hasDashC = args.getLastArg(self.parser.cOption)
364 hasDashE = args.getLastArg(self.parser.EOption)
365 hasDashS = args.getLastArg(self.parser.SOption)
Daniel Dunbar445c46d2009-01-20 01:53:54 +0000366 hasDashM = args.getLastArg(self.parser.MOption)
367 hasDashMM = args.getLastArg(self.parser.MMOption)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000368
369 inputType = None
370 inputTypeOpt = None
371 inputs = []
372 for a in args:
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000373 if a.opt is self.parser.inputOption:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000374 inputValue = args.getValue(a)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000375 if inputType is None:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000376 base,ext = os.path.splitext(inputValue)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000377 if ext and ext in Types.kTypeSuffixMap:
378 klass = Types.kTypeSuffixMap[ext]
379 else:
380 # FIXME: Its not clear why we shouldn't just
381 # revert to unknown. I think this is more likely a
382 # bug / unintended behavior in gcc. Not very
383 # important though.
384 klass = Types.ObjectType
385 else:
386 assert inputTypeOpt is not None
387 self.claim(inputTypeOpt)
388 klass = inputType
Daniel Dunbara0026f22009-01-16 23:12:12 +0000389
390 # Check that the file exists. It isn't clear this is
391 # worth doing, since the tool presumably does this
392 # anyway, and this just adds an extra stat to the
393 # equation, but this is gcc compatible.
Daniel Dunbar34f60f62009-01-26 17:09:15 +0000394 if inputValue != '-' and not os.path.exists(inputValue):
Daniel Dunbara0026f22009-01-16 23:12:12 +0000395 self.warning("%s: No such file or directory" % inputValue)
396 else:
397 inputs.append((klass, a))
Daniel Dunbarfc2ad022009-01-12 03:33:58 +0000398 elif a.opt.isLinkerInput:
399 # Treat as a linker input.
Daniel Dunbar927a5092009-01-06 02:30:10 +0000400 #
401 # FIXME: This might not be good enough. We may
402 # need to introduce another type for this case, so
403 # that other code which needs to know the inputs
404 # handles this properly. Best not to try and lipo
405 # this, for example.
406 inputs.append((Types.ObjectType, a))
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000407 elif a.opt is self.parser.xOption:
Daniel Dunbar927a5092009-01-06 02:30:10 +0000408 inputTypeOpt = a
409 value = args.getValue(a)
410 if value in Types.kTypeSpecifierMap:
411 inputType = Types.kTypeSpecifierMap[value]
412 else:
413 # FIXME: How are we going to handle diagnostics.
414 self.warning("language %s not recognized" % value)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000415
Daniel Dunbar927a5092009-01-06 02:30:10 +0000416 # FIXME: Its not clear why we shouldn't just
417 # revert to unknown. I think this is more likely a
418 # bug / unintended behavior in gcc. Not very
419 # important though.
Daniel Dunbar6cb42052009-01-13 21:07:43 +0000420 inputType = Types.ObjectType
Daniel Dunbar378530c2009-01-05 19:53:30 +0000421
422 # We claim things here so that options for which we silently allow
423 # override only ever claim the used option.
424 if hasCombine:
425 self.claim(hasCombine)
426
427 finalPhase = Phases.Phase.eOrderPostAssemble
428 finalPhaseOpt = None
429
430 # Determine what compilation mode we are in.
Daniel Dunbar445c46d2009-01-20 01:53:54 +0000431 if hasDashE or hasDashM or hasDashMM:
Daniel Dunbar378530c2009-01-05 19:53:30 +0000432 finalPhase = Phases.Phase.eOrderPreprocess
433 finalPhaseOpt = hasDashE
Daniel Dunbar34f60f62009-01-26 17:09:15 +0000434 elif (hasAnalyze or hasSyntaxOnly or
435 hasEmitLLVM or hasDashS):
Daniel Dunbar90c72cd2009-01-21 01:07:49 +0000436 finalPhase = Phases.Phase.eOrderCompile
Daniel Dunbar34f60f62009-01-26 17:09:15 +0000437 finalPhaseOpt = (hasAnalyze or hasSyntaxOnly or
438 hasEmitLLVM or hasDashS)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000439 elif hasDashC:
440 finalPhase = Phases.Phase.eOrderAssemble
441 finalPhaseOpt = hasDashC
442
443 if finalPhaseOpt:
444 self.claim(finalPhaseOpt)
445
446 # FIXME: Support -combine.
447 if hasCombine:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000448 raise NotImplementedError,"-combine is not yet supported"
449
Daniel Dunbar80e48b72009-01-17 00:53:19 +0000450 # Reject -Z* at the top level for now.
451 arg = args.getLastArg(self.parser.ZOption)
452 if arg:
453 raise Arguments.InvalidArgumentsError("%s: unsupported use of internal gcc option" % ' '.join(args.render(arg)))
454
Daniel Dunbar8fd28cd2009-01-28 19:26:20 +0000455 if not inputs and not self.suppressMissingInputWarning:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000456 raise Arguments.InvalidArgumentsError("no input files")
Daniel Dunbar378530c2009-01-05 19:53:30 +0000457
458 actions = []
459 linkerInputs = []
460 # FIXME: This is gross.
461 linkPhase = Phases.LinkPhase()
462 for klass,input in inputs:
463 # Figure out what step to start at.
464
465 # FIXME: This should be part of the input class probably?
466 # Altough it doesn't quite fit there either, things like
467 # asm-with-preprocess don't easily fit into a linear scheme.
468
469 # FIXME: I think we are going to end up wanting to just build
470 # a simple FSA which we run the inputs down.
471 sequence = []
472 if klass.preprocess:
473 sequence.append(Phases.PreprocessPhase())
474 if klass == Types.ObjectType:
475 sequence.append(linkPhase)
476 elif klass.onlyAssemble:
477 sequence.extend([Phases.AssemblePhase(),
478 linkPhase])
479 elif klass.onlyPrecompile:
480 sequence.append(Phases.PrecompilePhase())
Daniel Dunbar90c72cd2009-01-21 01:07:49 +0000481 elif hasAnalyze:
482 sequence.append(Phases.AnalyzePhase())
483 elif hasSyntaxOnly:
484 sequence.append(Phases.SyntaxOnlyPhase())
Daniel Dunbar34f60f62009-01-26 17:09:15 +0000485 elif hasEmitLLVM:
486 sequence.append(Phases.EmitLLVMPhase())
Daniel Dunbar378530c2009-01-05 19:53:30 +0000487 else:
488 sequence.extend([Phases.CompilePhase(),
489 Phases.AssemblePhase(),
490 linkPhase])
491
492 if sequence[0].order > finalPhase:
493 assert finalPhaseOpt and finalPhaseOpt.opt
494 # FIXME: Explain what type of input file is. Or just match
495 # gcc warning.
Daniel Dunbar74727872009-01-06 01:35:44 +0000496 self.warning("%s: %s input file unused when %s is present" % (args.getValue(input),
Daniel Dunbar378530c2009-01-05 19:53:30 +0000497 sequence[0].name,
498 finalPhaseOpt.opt.name))
499 else:
500 # Build the pipeline for this file.
501
502 current = Phases.InputAction(input, klass)
503 for transition in sequence:
504 # If the current action produces no output, or we are
505 # past what the user requested, we are done.
506 if (current.type is Types.NothingType or
507 transition.order > finalPhase):
508 break
509 else:
510 if isinstance(transition, Phases.PreprocessPhase):
511 assert isinstance(klass.preprocess, Types.InputType)
512 current = Phases.JobAction(transition,
513 [current],
514 klass.preprocess)
515 elif isinstance(transition, Phases.PrecompilePhase):
516 current = Phases.JobAction(transition,
517 [current],
518 Types.PCHType)
Daniel Dunbar90c72cd2009-01-21 01:07:49 +0000519 elif isinstance(transition, Phases.AnalyzePhase):
520 output = Types.PlistType
521 current = Phases.JobAction(transition,
522 [current],
523 output)
524 elif isinstance(transition, Phases.SyntaxOnlyPhase):
525 output = Types.NothingType
526 current = Phases.JobAction(transition,
527 [current],
528 output)
Daniel Dunbar34f60f62009-01-26 17:09:15 +0000529 elif isinstance(transition, Phases.EmitLLVMPhase):
530 if hasDashS:
531 output = Types.LLVMAsmType
532 else:
533 output = Types.LLVMBCType
534 current = Phases.JobAction(transition,
535 [current],
536 output)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000537 elif isinstance(transition, Phases.CompilePhase):
Daniel Dunbar90c72cd2009-01-21 01:07:49 +0000538 output = Types.AsmTypeNoPP
Daniel Dunbar378530c2009-01-05 19:53:30 +0000539 current = Phases.JobAction(transition,
540 [current],
541 output)
542 elif isinstance(transition, Phases.AssemblePhase):
543 current = Phases.JobAction(transition,
544 [current],
545 Types.ObjectType)
546 elif transition is linkPhase:
547 linkerInputs.append(current)
548 current = None
549 break
550 else:
551 raise RuntimeError,'Unrecognized transition: %s.' % transition
552 pass
553
554 if current is not None:
555 assert not isinstance(current, Phases.InputAction)
556 actions.append(current)
557
558 if linkerInputs:
559 actions.append(Phases.JobAction(linkPhase,
560 linkerInputs,
561 Types.ImageType))
562
563 return actions
564
Daniel Dunbar74727872009-01-06 01:35:44 +0000565 def buildPipeline(self, args):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000566 # FIXME: We need to handle canonicalization of the specified arch.
567
Daniel Dunbar31c80812009-01-20 21:29:14 +0000568 archs = {}
Daniel Dunbar445c46d2009-01-20 01:53:54 +0000569 hasDashM = args.getLastArg(self.parser.MGroup)
Daniel Dunbara33176f2009-01-21 18:49:34 +0000570 hasSaveTemps = args.getLastArg(self.parser.saveTempsOption)
Daniel Dunbar74727872009-01-06 01:35:44 +0000571 for arg in args:
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000572 if arg.opt is self.parser.archOption:
Daniel Dunbar31c80812009-01-20 21:29:14 +0000573 # FIXME: Canonicalize this.
574 archName = args.getValue(arg)
575 archs[archName] = arg
576
577 archs = archs.values()
Daniel Dunbar378530c2009-01-05 19:53:30 +0000578 if not archs:
Daniel Dunbarf3d5ca02009-01-13 04:05:40 +0000579 archs.append(args.makeSeparateArg(self.hostInfo.getArchName(args),
Daniel Dunbar1ba90982009-01-07 18:54:26 +0000580 self.parser.archOption))
Daniel Dunbar378530c2009-01-05 19:53:30 +0000581
Daniel Dunbar74727872009-01-06 01:35:44 +0000582 actions = self.buildNormalPipeline(args)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000583
584 # FIXME: Use custom exception for this.
585 #
586 # FIXME: We killed off some others but these aren't yet detected in
587 # a functional manner. If we added information to jobs about which
588 # "auxiliary" files they wrote then we could detect the conflict
589 # these cause downstream.
590 if len(archs) > 1:
591 if hasDashM:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000592 raise Arguments.InvalidArgumentsError("Cannot use -M options with multiple arch flags.")
Daniel Dunbar378530c2009-01-05 19:53:30 +0000593 elif hasSaveTemps:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000594 raise Arguments.InvalidArgumentsError("Cannot use -save-temps with multiple arch flags.")
Daniel Dunbar378530c2009-01-05 19:53:30 +0000595
596 # Execute once per arch.
597 finalActions = []
598 for p in actions:
599 # Make sure we can lipo this kind of output. If not (and it
600 # is an actual output) then we disallow, since we can't
601 # create an output file with the right name without
602 # overwriting it. We could remove this oddity by just
603 # changing the output names to include the arch, which would
604 # also fix -save-temps. Compatibility wins for now.
605 #
606 # FIXME: Is this error substantially less useful than
607 # gcc-dd's? The main problem is that "Cannot use compiler
608 # output with multiple arch flags" won't make sense to most
609 # developers.
610 if (len(archs) > 1 and
611 p.type not in (Types.NothingType,Types.ObjectType,Types.ImageType)):
Daniel Dunbara0026f22009-01-16 23:12:12 +0000612 raise Arguments.InvalidArgumentsError('Cannot use %s output with multiple arch flags.' % p.type.name)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000613
614 inputs = []
615 for arch in archs:
616 inputs.append(Phases.BindArchAction(p, arch))
617
618 # Lipo if necessary. We do it this way because we need to set
619 # the arch flag so that -Xarch_ gets rewritten.
620 if len(inputs) == 1 or p.type == Types.NothingType:
621 finalActions.extend(inputs)
622 else:
623 finalActions.append(Phases.JobAction(Phases.LipoPhase(),
624 inputs,
625 p.type))
626
Daniel Dunbar378530c2009-01-05 19:53:30 +0000627 return finalActions
628
Daniel Dunbar74727872009-01-06 01:35:44 +0000629 def bindPhases(self, phases, args):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000630 jobs = Jobs.JobList()
631
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000632 finalOutput = args.getLastArg(self.parser.oOption)
Daniel Dunbara33176f2009-01-21 18:49:34 +0000633 hasSaveTemps = args.getLastArg(self.parser.saveTempsOption)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000634 hasNoIntegratedCPP = args.getLastArg(self.parser.noIntegratedCPPOption)
Daniel Dunbarf86e98a2009-01-12 09:23:15 +0000635 hasTraditionalCPP = args.getLastArg(self.parser.traditionalCPPOption)
Daniel Dunbare9f1a692009-01-06 06:12:13 +0000636 hasPipe = args.getLastArg(self.parser.pipeOption)
Daniel Dunbarfc2ad022009-01-12 03:33:58 +0000637
Daniel Dunbar378530c2009-01-05 19:53:30 +0000638 # We claim things here so that options for which we silently allow
639 # override only ever claim the used option.
640 if hasPipe:
641 self.claim(hasPipe)
642 # FIXME: Hack, override -pipe till we support it.
Daniel Dunbare19ed8e2009-01-17 02:02:35 +0000643 if hasSaveTemps:
644 self.warning('-pipe ignored because -save-temps specified')
645 hasPipe = None
Daniel Dunbar378530c2009-01-05 19:53:30 +0000646 # Claim these here. Its not completely accurate but any warnings
647 # about these being unused are likely to be noise anyway.
648 if hasSaveTemps:
649 self.claim(hasSaveTemps)
Daniel Dunbarf86e98a2009-01-12 09:23:15 +0000650
651 if hasTraditionalCPP:
652 self.claim(hasTraditionalCPP)
653 elif hasNoIntegratedCPP:
Daniel Dunbar378530c2009-01-05 19:53:30 +0000654 self.claim(hasNoIntegratedCPP)
Daniel Dunbarf86e98a2009-01-12 09:23:15 +0000655
Daniel Dunbare5fb6792009-01-13 06:25:31 +0000656 # FIXME: Move to... somewhere else.
Daniel Dunbar378530c2009-01-05 19:53:30 +0000657 class InputInfo:
658 def __init__(self, source, type, baseInput):
659 self.source = source
660 self.type = type
661 self.baseInput = baseInput
662
663 def __repr__(self):
664 return '%s(%r, %r, %r)' % (self.__class__.__name__,
665 self.source, self.type, self.baseInput)
Daniel Dunbare5fb6792009-01-13 06:25:31 +0000666
667 def isOriginalInput(self):
668 return self.source is self.baseInput
Daniel Dunbar378530c2009-01-05 19:53:30 +0000669
Daniel Dunbarb3492762009-01-13 18:51:26 +0000670 def createJobs(tc, phase,
671 canAcceptPipe=False, atTopLevel=False, arch=None,
Daniel Dunbar31c80812009-01-20 21:29:14 +0000672 tcArgs=None, linkingOutput=None):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000673 if isinstance(phase, Phases.InputAction):
674 return InputInfo(phase.filename, phase.type, phase.filename)
675 elif isinstance(phase, Phases.BindArchAction):
Daniel Dunbar74727872009-01-06 01:35:44 +0000676 archName = args.getValue(phase.arch)
Daniel Dunbar758cf642009-01-11 22:06:22 +0000677 tc = self.hostInfo.getToolChainForArch(archName)
Daniel Dunbarb3492762009-01-13 18:51:26 +0000678 return createJobs(tc, phase.inputs[0],
679 canAcceptPipe, atTopLevel, phase.arch,
Daniel Dunbar31c80812009-01-20 21:29:14 +0000680 None, linkingOutput)
Daniel Dunbarb3492762009-01-13 18:51:26 +0000681
682 if tcArgs is None:
683 tcArgs = tc.translateArgs(args, arch)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000684
685 assert isinstance(phase, Phases.JobAction)
Daniel Dunbar758cf642009-01-11 22:06:22 +0000686 tool = tc.selectTool(phase)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000687
688 # See if we should use an integrated CPP. We only use an
689 # integrated cpp when we have exactly one input, since this is
690 # the only use case we care about.
691 useIntegratedCPP = False
692 inputList = phase.inputs
693 if (not hasNoIntegratedCPP and
Daniel Dunbarf86e98a2009-01-12 09:23:15 +0000694 not hasTraditionalCPP and
Daniel Dunbar378530c2009-01-05 19:53:30 +0000695 not hasSaveTemps and
696 tool.hasIntegratedCPP()):
697 if (len(phase.inputs) == 1 and
Daniel Dunbar7d494092009-01-20 00:47:24 +0000698 isinstance(phase.inputs[0], Phases.JobAction) and
Daniel Dunbar378530c2009-01-05 19:53:30 +0000699 isinstance(phase.inputs[0].phase, Phases.PreprocessPhase)):
700 useIntegratedCPP = True
701 inputList = phase.inputs[0].inputs
702
703 # Only try to use pipes when exactly one input.
Daniel Dunbar1e46f552009-01-22 23:19:32 +0000704 attemptToPipeInput = len(inputList) == 1 and tool.acceptsPipedInput()
705 inputs = [createJobs(tc, p, attemptToPipeInput, False,
Daniel Dunbar31c80812009-01-20 21:29:14 +0000706 arch, tcArgs, linkingOutput)
Daniel Dunbar758cf642009-01-11 22:06:22 +0000707 for p in inputList]
Daniel Dunbar378530c2009-01-05 19:53:30 +0000708
709 # Determine if we should output to a pipe.
710 canOutputToPipe = canAcceptPipe and tool.canPipeOutput()
711 outputToPipe = False
712 if canOutputToPipe:
713 # Some things default to writing to a pipe if the final
714 # phase and there was no user override.
715 #
716 # FIXME: What is the best way to handle this?
Daniel Dunbar1b391272009-01-18 21:35:24 +0000717 if atTopLevel:
718 if (isinstance(phase.phase, Phases.PreprocessPhase) and
719 not finalOutput):
720 outputToPipe = True
Daniel Dunbar378530c2009-01-05 19:53:30 +0000721 elif hasPipe:
722 outputToPipe = True
723
724 # Figure out where to put the job (pipes).
725 jobList = jobs
Daniel Dunbar1e46f552009-01-22 23:19:32 +0000726 if isinstance(inputs[0].source, Jobs.PipedJob):
Daniel Dunbar378530c2009-01-05 19:53:30 +0000727 jobList = inputs[0].source
Daniel Dunbar31c80812009-01-20 21:29:14 +0000728
Daniel Dunbar378530c2009-01-05 19:53:30 +0000729 baseInput = inputs[0].baseInput
Daniel Dunbard9b7a742009-01-21 00:05:15 +0000730 output,jobList = self.getOutputName(phase, outputToPipe, jobs, jobList, baseInput,
731 args, atTopLevel, hasSaveTemps, finalOutput)
Daniel Dunbarb421dba2009-01-07 18:40:45 +0000732 tool.constructJob(phase, arch, jobList, inputs, output, phase.type,
Daniel Dunbar31c80812009-01-20 21:29:14 +0000733 tcArgs, linkingOutput)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000734
735 return InputInfo(output, phase.type, baseInput)
736
737 # It is an error to provide a -o option if we are making multiple
738 # output files.
739 if finalOutput and len([a for a in phases if a.type is not Types.NothingType]) > 1:
Daniel Dunbara0026f22009-01-16 23:12:12 +0000740 raise Arguments.InvalidArgumentsError("cannot specify -o when generating multiple files")
Daniel Dunbar378530c2009-01-05 19:53:30 +0000741
742 for phase in phases:
Daniel Dunbar31c80812009-01-20 21:29:14 +0000743 # If we are linking an image for multiple archs then the
744 # linker wants -arch_multiple and -final_output <final image
745 # name>. Unfortunately this requires some gross contortions.
746 #
747 # FIXME: This is a hack; find a cleaner way to integrate this
748 # into the process.
749 linkingOutput = None
750 if (isinstance(phase, Phases.JobAction) and
751 isinstance(phase.phase, Phases.LipoPhase)):
752 finalOutput = args.getLastArg(self.parser.oOption)
753 if finalOutput:
754 linkingOutput = finalOutput
755 else:
756 linkingOutput = args.makeSeparateArg('a.out',
757 self.parser.oOption)
758
Daniel Dunbarb3492762009-01-13 18:51:26 +0000759 createJobs(self.toolChain, phase,
Daniel Dunbar31c80812009-01-20 21:29:14 +0000760 canAcceptPipe=True, atTopLevel=True,
761 linkingOutput=linkingOutput)
Daniel Dunbar378530c2009-01-05 19:53:30 +0000762
763 return jobs
Daniel Dunbar31c80812009-01-20 21:29:14 +0000764
765 def getOutputName(self, phase, outputToPipe, jobs, jobList, baseInput,
766 args, atTopLevel, hasSaveTemps, finalOutput):
767 # Figure out where to put the output.
768 if phase.type == Types.NothingType:
769 output = None
770 elif outputToPipe:
771 if isinstance(jobList, Jobs.PipedJob):
772 output = jobList
773 else:
774 jobList = output = Jobs.PipedJob([])
775 jobs.addJob(output)
776 else:
777 # Figure out what the derived output location would be.
778 #
779 # FIXME: gcc has some special case in here so that it doesn't
780 # create output files if they would conflict with an input.
781 if phase.type is Types.ImageType:
782 namedOutput = "a.out"
783 else:
784 inputName = args.getValue(baseInput)
785 base,_ = os.path.splitext(inputName)
786 assert phase.type.tempSuffix is not None
787 namedOutput = base + '.' + phase.type.tempSuffix
788
789 # Output to user requested destination?
790 if atTopLevel and finalOutput:
791 output = finalOutput
792 # Contruct a named destination?
793 elif atTopLevel or hasSaveTemps:
794 # As an annoying special case, pch generation
795 # doesn't strip the pathname.
796 if phase.type is Types.PCHType:
797 outputName = namedOutput
798 else:
799 outputName = os.path.basename(namedOutput)
800 output = args.makeSeparateArg(outputName,
801 self.parser.oOption)
802 else:
803 # Output to temp file...
804 fd,filename = tempfile.mkstemp(suffix='.'+phase.type.tempSuffix)
805 output = args.makeSeparateArg(filename,
806 self.parser.oOption)
Daniel Dunbard9b7a742009-01-21 00:05:15 +0000807 return output,jobList