xref: /llvm-project/clang/include/clang/Driver/Driver.h (revision 3a9380f21d05eb8ced03349c8c503dc911f22621)
1 //===--- Driver.h - Clang GCC Compatible Driver -----------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef LLVM_CLANG_DRIVER_DRIVER_H
10 #define LLVM_CLANG_DRIVER_DRIVER_H
11 
12 #include "clang/Basic/Diagnostic.h"
13 #include "clang/Basic/HeaderInclude.h"
14 #include "clang/Basic/LLVM.h"
15 #include "clang/Driver/Action.h"
16 #include "clang/Driver/DriverDiagnostic.h"
17 #include "clang/Driver/InputInfo.h"
18 #include "clang/Driver/Options.h"
19 #include "clang/Driver/Phases.h"
20 #include "clang/Driver/ToolChain.h"
21 #include "clang/Driver/Types.h"
22 #include "clang/Driver/Util.h"
23 #include "llvm/ADT/ArrayRef.h"
24 #include "llvm/ADT/STLFunctionalExtras.h"
25 #include "llvm/ADT/StringMap.h"
26 #include "llvm/ADT/StringRef.h"
27 #include "llvm/Option/Arg.h"
28 #include "llvm/Option/ArgList.h"
29 #include "llvm/Support/StringSaver.h"
30 
31 #include <map>
32 #include <set>
33 #include <string>
34 #include <vector>
35 
36 namespace llvm {
37 class Triple;
38 namespace vfs {
39 class FileSystem;
40 }
41 namespace cl {
42 class ExpansionContext;
43 }
44 } // namespace llvm
45 
46 namespace clang {
47 
48 namespace driver {
49 
50 typedef SmallVector<InputInfo, 4> InputInfoList;
51 
52 class Command;
53 class Compilation;
54 class JobAction;
55 class ToolChain;
56 
57 /// Describes the kind of LTO mode selected via -f(no-)?lto(=.*)? options.
58 enum LTOKind {
59   LTOK_None,
60   LTOK_Full,
61   LTOK_Thin,
62   LTOK_Unknown
63 };
64 
65 /// Whether headers used to construct C++20 module units should be looked
66 /// up by the path supplied on the command line, or in the user or system
67 /// search paths.
68 enum ModuleHeaderMode {
69   HeaderMode_None,
70   HeaderMode_Default,
71   HeaderMode_User,
72   HeaderMode_System
73 };
74 
75 /// Options for specifying CUID used by CUDA/HIP for uniquely identifying
76 /// compilation units.
77 class CUIDOptions {
78 public:
79   enum class Kind { Hash, Random, Fixed, None, Invalid };
80 
81   CUIDOptions() = default;
82   CUIDOptions(llvm::opt::DerivedArgList &Args, const Driver &D);
83 
84   // Get the CUID for an input string
85   std::string getCUID(StringRef InputFile,
86                       llvm::opt::DerivedArgList &Args) const;
87 
88   bool isEnabled() const {
89     return UseCUID != Kind::None && UseCUID != Kind::Invalid;
90   }
91 
92 private:
93   Kind UseCUID = Kind::None;
94   StringRef FixedCUID;
95 };
96 
97 /// Driver - Encapsulate logic for constructing compilation processes
98 /// from a set of gcc-driver-like command line arguments.
99 class Driver {
100   DiagnosticsEngine &Diags;
101 
102   IntrusiveRefCntPtr<llvm::vfs::FileSystem> VFS;
103 
104   enum DriverMode {
105     GCCMode,
106     GXXMode,
107     CPPMode,
108     CLMode,
109     FlangMode,
110     DXCMode
111   } Mode;
112 
113   enum SaveTempsMode {
114     SaveTempsNone,
115     SaveTempsCwd,
116     SaveTempsObj
117   } SaveTemps;
118 
119   enum BitcodeEmbedMode {
120     EmbedNone,
121     EmbedMarker,
122     EmbedBitcode
123   } BitcodeEmbed;
124 
125   enum OffloadMode {
126     OffloadHostDevice,
127     OffloadHost,
128     OffloadDevice,
129   } Offload;
130 
131   /// Header unit mode set by -fmodule-header={user,system}.
132   ModuleHeaderMode CXX20HeaderType;
133 
134   /// Set if we should process inputs and jobs with C++20 module
135   /// interpretation.
136   bool ModulesModeCXX20;
137 
138   /// LTO mode selected via -f(no-)?lto(=.*)? options.
139   LTOKind LTOMode;
140 
141   /// LTO mode selected via -f(no-offload-)?lto(=.*)? options.
142   LTOKind OffloadLTOMode;
143 
144   /// Options for CUID
145   CUIDOptions CUIDOpts;
146 
147 public:
148   enum OpenMPRuntimeKind {
149     /// An unknown OpenMP runtime. We can't generate effective OpenMP code
150     /// without knowing what runtime to target.
151     OMPRT_Unknown,
152 
153     /// The LLVM OpenMP runtime. When completed and integrated, this will become
154     /// the default for Clang.
155     OMPRT_OMP,
156 
157     /// The GNU OpenMP runtime. Clang doesn't support generating OpenMP code for
158     /// this runtime but can swallow the pragmas, and find and link against the
159     /// runtime library itself.
160     OMPRT_GOMP,
161 
162     /// The legacy name for the LLVM OpenMP runtime from when it was the Intel
163     /// OpenMP runtime. We support this mode for users with existing
164     /// dependencies on this runtime library name.
165     OMPRT_IOMP5
166   };
167 
168   // Diag - Forwarding function for diagnostics.
169   DiagnosticBuilder Diag(unsigned DiagID) const {
170     return Diags.Report(DiagID);
171   }
172 
173   // FIXME: Privatize once interface is stable.
174 public:
175   /// The name the driver was invoked as.
176   std::string Name;
177 
178   /// The path the driver executable was in, as invoked from the
179   /// command line.
180   std::string Dir;
181 
182   /// The original path to the clang executable.
183   std::string ClangExecutable;
184 
185   /// Target and driver mode components extracted from clang executable name.
186   ParsedClangName ClangNameParts;
187 
188   /// The path to the compiler resource directory.
189   std::string ResourceDir;
190 
191   /// System directory for config files.
192   std::string SystemConfigDir;
193 
194   /// User directory for config files.
195   std::string UserConfigDir;
196 
197   /// A prefix directory used to emulate a limited subset of GCC's '-Bprefix'
198   /// functionality.
199   /// FIXME: This type of customization should be removed in favor of the
200   /// universal driver when it is ready.
201   typedef SmallVector<std::string, 4> prefix_list;
202   prefix_list PrefixDirs;
203 
204   /// sysroot, if present
205   std::string SysRoot;
206 
207   /// Dynamic loader prefix, if present
208   std::string DyldPrefix;
209 
210   /// Driver title to use with help.
211   std::string DriverTitle;
212 
213   /// Information about the host which can be overridden by the user.
214   std::string HostBits, HostMachine, HostSystem, HostRelease;
215 
216   /// The file to log CC_PRINT_PROC_STAT_FILE output to, if enabled.
217   std::string CCPrintStatReportFilename;
218 
219   /// The file to log CC_PRINT_INTERNAL_STAT_FILE output to, if enabled.
220   std::string CCPrintInternalStatReportFilename;
221 
222   /// The file to log CC_PRINT_OPTIONS output to, if enabled.
223   std::string CCPrintOptionsFilename;
224 
225   /// The file to log CC_PRINT_HEADERS output to, if enabled.
226   std::string CCPrintHeadersFilename;
227 
228   /// The file to log CC_LOG_DIAGNOSTICS output to, if enabled.
229   std::string CCLogDiagnosticsFilename;
230 
231   /// An input type and its arguments.
232   using InputTy = std::pair<types::ID, const llvm::opt::Arg *>;
233 
234   /// A list of inputs and their types for the given arguments.
235   using InputList = SmallVector<InputTy, 16>;
236 
237   /// Whether the driver should follow g++ like behavior.
238   bool CCCIsCXX() const { return Mode == GXXMode; }
239 
240   /// Whether the driver is just the preprocessor.
241   bool CCCIsCPP() const { return Mode == CPPMode; }
242 
243   /// Whether the driver should follow gcc like behavior.
244   bool CCCIsCC() const { return Mode == GCCMode; }
245 
246   /// Whether the driver should follow cl.exe like behavior.
247   bool IsCLMode() const { return Mode == CLMode; }
248 
249   /// Whether the driver should invoke flang for fortran inputs.
250   /// Other modes fall back to calling gcc which in turn calls gfortran.
251   bool IsFlangMode() const { return Mode == FlangMode; }
252 
253   /// Whether the driver should follow dxc.exe like behavior.
254   bool IsDXCMode() const { return Mode == DXCMode; }
255 
256   /// Only print tool bindings, don't build any jobs.
257   LLVM_PREFERRED_TYPE(bool)
258   unsigned CCCPrintBindings : 1;
259 
260   /// Set CC_PRINT_OPTIONS mode, which is like -v but logs the commands to
261   /// CCPrintOptionsFilename or to stderr.
262   LLVM_PREFERRED_TYPE(bool)
263   unsigned CCPrintOptions : 1;
264 
265   /// The format of the header information that is emitted. If CC_PRINT_HEADERS
266   /// is set, the format is textual. Otherwise, the format is determined by the
267   /// enviroment variable CC_PRINT_HEADERS_FORMAT.
268   HeaderIncludeFormatKind CCPrintHeadersFormat = HIFMT_None;
269 
270   /// This flag determines whether clang should filter the header information
271   /// that is emitted. If enviroment variable CC_PRINT_HEADERS_FILTERING is set
272   /// to "only-direct-system", only system headers that are directly included
273   /// from non-system headers are emitted.
274   HeaderIncludeFilteringKind CCPrintHeadersFiltering = HIFIL_None;
275 
276   /// Name of the library that provides implementations of
277   /// IEEE-754 128-bit float math functions used by Fortran F128
278   /// runtime library. It should be linked as needed by the linker job.
279   std::string FlangF128MathLibrary;
280 
281   /// Set CC_LOG_DIAGNOSTICS mode, which causes the frontend to log diagnostics
282   /// to CCLogDiagnosticsFilename or to stderr, in a stable machine readable
283   /// format.
284   LLVM_PREFERRED_TYPE(bool)
285   unsigned CCLogDiagnostics : 1;
286 
287   /// Whether the driver is generating diagnostics for debugging purposes.
288   LLVM_PREFERRED_TYPE(bool)
289   unsigned CCGenDiagnostics : 1;
290 
291   /// Set CC_PRINT_PROC_STAT mode, which causes the driver to dump
292   /// performance report to CC_PRINT_PROC_STAT_FILE or to stdout.
293   LLVM_PREFERRED_TYPE(bool)
294   unsigned CCPrintProcessStats : 1;
295 
296   /// Set CC_PRINT_INTERNAL_STAT mode, which causes the driver to dump internal
297   /// performance report to CC_PRINT_INTERNAL_STAT_FILE or to stdout.
298   LLVM_PREFERRED_TYPE(bool)
299   unsigned CCPrintInternalStats : 1;
300 
301   /// Pointer to the ExecuteCC1Tool function, if available.
302   /// When the clangDriver lib is used through clang.exe, this provides a
303   /// shortcut for executing the -cc1 command-line directly, in the same
304   /// process.
305   using CC1ToolFunc =
306       llvm::function_ref<int(SmallVectorImpl<const char *> &ArgV)>;
307   CC1ToolFunc CC1Main = nullptr;
308 
309 private:
310   /// Raw target triple.
311   std::string TargetTriple;
312 
313   /// Name to use when invoking gcc/g++.
314   std::string CCCGenericGCCName;
315 
316   /// Paths to configuration files used.
317   std::vector<std::string> ConfigFiles;
318 
319   /// Allocator for string saver.
320   llvm::BumpPtrAllocator Alloc;
321 
322   /// Object that stores strings read from configuration file.
323   llvm::StringSaver Saver;
324 
325   /// Arguments originated from configuration file (head part).
326   std::unique_ptr<llvm::opt::InputArgList> CfgOptionsHead;
327 
328   /// Arguments originated from configuration file (tail part).
329   std::unique_ptr<llvm::opt::InputArgList> CfgOptionsTail;
330 
331   /// Arguments originated from command line.
332   std::unique_ptr<llvm::opt::InputArgList> CLOptions;
333 
334   /// If this is non-null, the driver will prepend this argument before
335   /// reinvoking clang. This is useful for the llvm-driver where clang's
336   /// realpath will be to the llvm binary and not clang, so it must pass
337   /// "clang" as it's first argument.
338   const char *PrependArg;
339 
340   /// Whether to check that input files exist when constructing compilation
341   /// jobs.
342   LLVM_PREFERRED_TYPE(bool)
343   unsigned CheckInputsExist : 1;
344   /// Whether to probe for PCH files on disk, in order to upgrade
345   /// -include foo.h to -include-pch foo.h.pch.
346   LLVM_PREFERRED_TYPE(bool)
347   unsigned ProbePrecompiled : 1;
348 
349 public:
350   // getFinalPhase - Determine which compilation mode we are in and record
351   // which option we used to determine the final phase.
352   // TODO: Much of what getFinalPhase returns are not actually true compiler
353   //       modes. Fold this functionality into Types::getCompilationPhases and
354   //       handleArguments.
355   phases::ID getFinalPhase(const llvm::opt::DerivedArgList &DAL,
356                            llvm::opt::Arg **FinalPhaseArg = nullptr) const;
357 
358 private:
359   /// Certain options suppress the 'no input files' warning.
360   LLVM_PREFERRED_TYPE(bool)
361   unsigned SuppressMissingInputWarning : 1;
362 
363   /// Cache of all the ToolChains in use by the driver.
364   ///
365   /// This maps from the string representation of a triple to a ToolChain
366   /// created targeting that triple. The driver owns all the ToolChain objects
367   /// stored in it, and will clean them up when torn down.
368   mutable llvm::StringMap<std::unique_ptr<ToolChain>> ToolChains;
369 
370   /// Cache of known offloading architectures for the ToolChain already derived.
371   /// This should only be modified when we first initialize the offloading
372   /// toolchains.
373   llvm::DenseMap<const ToolChain *, llvm::DenseSet<llvm::StringRef>> KnownArchs;
374 
375 private:
376   /// TranslateInputArgs - Create a new derived argument list from the input
377   /// arguments, after applying the standard argument translations.
378   llvm::opt::DerivedArgList *
379   TranslateInputArgs(const llvm::opt::InputArgList &Args) const;
380 
381   // handleArguments - All code related to claiming and printing diagnostics
382   // related to arguments to the driver are done here.
383   void handleArguments(Compilation &C, llvm::opt::DerivedArgList &Args,
384                        const InputList &Inputs, ActionList &Actions) const;
385 
386   // Before executing jobs, sets up response files for commands that need them.
387   void setUpResponseFiles(Compilation &C, Command &Cmd);
388 
389   void generatePrefixedToolNames(StringRef Tool, const ToolChain &TC,
390                                  SmallVectorImpl<std::string> &Names) const;
391 
392   /// Find the appropriate .crash diagonostic file for the child crash
393   /// under this driver and copy it out to a temporary destination with the
394   /// other reproducer related files (.sh, .cache, etc). If not found, suggest a
395   /// directory for the user to look at.
396   ///
397   /// \param ReproCrashFilename The file path to copy the .crash to.
398   /// \param CrashDiagDir       The suggested directory for the user to look at
399   ///                           in case the search or copy fails.
400   ///
401   /// \returns If the .crash is found and successfully copied return true,
402   /// otherwise false and return the suggested directory in \p CrashDiagDir.
403   bool getCrashDiagnosticFile(StringRef ReproCrashFilename,
404                               SmallString<128> &CrashDiagDir);
405 
406 public:
407 
408   /// Takes the path to a binary that's either in bin/ or lib/ and returns
409   /// the path to clang's resource directory.
410   static std::string GetResourcesPath(StringRef BinaryPath);
411 
412   Driver(StringRef ClangExecutable, StringRef TargetTriple,
413          DiagnosticsEngine &Diags, std::string Title = "clang LLVM compiler",
414          IntrusiveRefCntPtr<llvm::vfs::FileSystem> VFS = nullptr);
415 
416   /// @name Accessors
417   /// @{
418 
419   /// Name to use when invoking gcc/g++.
420   const std::string &getCCCGenericGCCName() const { return CCCGenericGCCName; }
421 
422   llvm::ArrayRef<std::string> getConfigFiles() const {
423     return ConfigFiles;
424   }
425 
426   const llvm::opt::OptTable &getOpts() const { return getDriverOptTable(); }
427 
428   DiagnosticsEngine &getDiags() const { return Diags; }
429 
430   llvm::vfs::FileSystem &getVFS() const { return *VFS; }
431 
432   bool getCheckInputsExist() const { return CheckInputsExist; }
433 
434   void setCheckInputsExist(bool Value) { CheckInputsExist = Value; }
435 
436   bool getProbePrecompiled() const { return ProbePrecompiled; }
437   void setProbePrecompiled(bool Value) { ProbePrecompiled = Value; }
438 
439   const char *getPrependArg() const { return PrependArg; }
440   void setPrependArg(const char *Value) { PrependArg = Value; }
441 
442   void setTargetAndMode(const ParsedClangName &TM) { ClangNameParts = TM; }
443 
444   const std::string &getTitle() { return DriverTitle; }
445   void setTitle(std::string Value) { DriverTitle = std::move(Value); }
446 
447   std::string getTargetTriple() const { return TargetTriple; }
448 
449   /// Get the path to the main clang executable.
450   const char *getClangProgramPath() const {
451     return ClangExecutable.c_str();
452   }
453 
454   bool isSaveTempsEnabled() const { return SaveTemps != SaveTempsNone; }
455   bool isSaveTempsObj() const { return SaveTemps == SaveTempsObj; }
456 
457   bool embedBitcodeEnabled() const { return BitcodeEmbed != EmbedNone; }
458   bool embedBitcodeInObject() const { return (BitcodeEmbed == EmbedBitcode); }
459   bool embedBitcodeMarkerOnly() const { return (BitcodeEmbed == EmbedMarker); }
460 
461   bool offloadHostOnly() const { return Offload == OffloadHost; }
462   bool offloadDeviceOnly() const { return Offload == OffloadDevice; }
463 
464   void setFlangF128MathLibrary(std::string name) {
465     FlangF128MathLibrary = std::move(name);
466   }
467   StringRef getFlangF128MathLibrary() const { return FlangF128MathLibrary; }
468 
469   /// Compute the desired OpenMP runtime from the flags provided.
470   OpenMPRuntimeKind getOpenMPRuntime(const llvm::opt::ArgList &Args) const;
471 
472   /// @}
473   /// @name Primary Functionality
474   /// @{
475 
476   /// CreateOffloadingDeviceToolChains - create all the toolchains required to
477   /// support offloading devices given the programming models specified in the
478   /// current compilation. Also, update the host tool chain kind accordingly.
479   void CreateOffloadingDeviceToolChains(Compilation &C, InputList &Inputs);
480 
481   /// BuildCompilation - Construct a compilation object for a command
482   /// line argument vector.
483   ///
484   /// \return A compilation, or 0 if none was built for the given
485   /// argument vector. A null return value does not necessarily
486   /// indicate an error condition, the diagnostics should be queried
487   /// to determine if an error occurred.
488   Compilation *BuildCompilation(ArrayRef<const char *> Args);
489 
490   /// ParseArgStrings - Parse the given list of strings into an
491   /// ArgList.
492   llvm::opt::InputArgList ParseArgStrings(ArrayRef<const char *> Args,
493                                           bool UseDriverMode,
494                                           bool &ContainsError) const;
495 
496   /// BuildInputs - Construct the list of inputs and their types from
497   /// the given arguments.
498   ///
499   /// \param TC - The default host tool chain.
500   /// \param Args - The input arguments.
501   /// \param Inputs - The list to store the resulting compilation
502   /// inputs onto.
503   void BuildInputs(const ToolChain &TC, llvm::opt::DerivedArgList &Args,
504                    InputList &Inputs) const;
505 
506   /// BuildActions - Construct the list of actions to perform for the
507   /// given arguments, which are only done for a single architecture.
508   ///
509   /// \param C - The compilation that is being built.
510   /// \param Args - The input arguments.
511   /// \param Actions - The list to store the resulting actions onto.
512   void BuildActions(Compilation &C, llvm::opt::DerivedArgList &Args,
513                     const InputList &Inputs, ActionList &Actions) const;
514 
515   /// BuildUniversalActions - Construct the list of actions to perform
516   /// for the given arguments, which may require a universal build.
517   ///
518   /// \param C - The compilation that is being built.
519   /// \param TC - The default host tool chain.
520   void BuildUniversalActions(Compilation &C, const ToolChain &TC,
521                              const InputList &BAInputs) const;
522 
523   /// BuildOffloadingActions - Construct the list of actions to perform for the
524   /// offloading toolchain that will be embedded in the host.
525   ///
526   /// \param C - The compilation that is being built.
527   /// \param Args - The input arguments.
528   /// \param Input - The input type and arguments
529   /// \param CUID - The CUID for \p Input
530   /// \param HostAction - The host action used in the offloading toolchain.
531   Action *BuildOffloadingActions(Compilation &C,
532                                  llvm::opt::DerivedArgList &Args,
533                                  const InputTy &Input, StringRef CUID,
534                                  Action *HostAction) const;
535 
536   /// Returns the set of bound architectures active for this offload kind.
537   /// If there are no bound architctures we return a set containing only the
538   /// empty string. The \p SuppressError option is used to suppress errors.
539   llvm::DenseSet<StringRef>
540   getOffloadArchs(Compilation &C, const llvm::opt::DerivedArgList &Args,
541                   Action::OffloadKind Kind, const ToolChain *TC,
542                   bool SuppressError = false) const;
543 
544   /// Check that the file referenced by Value exists. If it doesn't,
545   /// issue a diagnostic and return false.
546   /// If TypoCorrect is true and the file does not exist, see if it looks
547   /// like a likely typo for a flag and if so print a "did you mean" blurb.
548   bool DiagnoseInputExistence(const llvm::opt::DerivedArgList &Args,
549                               StringRef Value, types::ID Ty,
550                               bool TypoCorrect) const;
551 
552   /// BuildJobs - Bind actions to concrete tools and translate
553   /// arguments to form the list of jobs to run.
554   ///
555   /// \param C - The compilation that is being built.
556   void BuildJobs(Compilation &C) const;
557 
558   /// ExecuteCompilation - Execute the compilation according to the command line
559   /// arguments and return an appropriate exit code.
560   ///
561   /// This routine handles additional processing that must be done in addition
562   /// to just running the subprocesses, for example reporting errors, setting
563   /// up response files, removing temporary files, etc.
564   int ExecuteCompilation(Compilation &C,
565      SmallVectorImpl< std::pair<int, const Command *> > &FailingCommands);
566 
567   /// Contains the files in the compilation diagnostic report generated by
568   /// generateCompilationDiagnostics.
569   struct CompilationDiagnosticReport {
570     llvm::SmallVector<std::string, 4> TemporaryFiles;
571   };
572 
573   /// generateCompilationDiagnostics - Generate diagnostics information
574   /// including preprocessed source file(s).
575   ///
576   void generateCompilationDiagnostics(
577       Compilation &C, const Command &FailingCommand,
578       StringRef AdditionalInformation = "",
579       CompilationDiagnosticReport *GeneratedReport = nullptr);
580 
581   enum class CommandStatus {
582     Crash = 1,
583     Error,
584     Ok,
585   };
586 
587   enum class ReproLevel {
588     Off = 0,
589     OnCrash = static_cast<int>(CommandStatus::Crash),
590     OnError = static_cast<int>(CommandStatus::Error),
591     Always = static_cast<int>(CommandStatus::Ok),
592   };
593 
594   bool maybeGenerateCompilationDiagnostics(
595       CommandStatus CS, ReproLevel Level, Compilation &C,
596       const Command &FailingCommand, StringRef AdditionalInformation = "",
597       CompilationDiagnosticReport *GeneratedReport = nullptr) {
598     if (static_cast<int>(CS) > static_cast<int>(Level))
599       return false;
600     if (CS != CommandStatus::Crash)
601       Diags.Report(diag::err_drv_force_crash)
602           << !::getenv("FORCE_CLANG_DIAGNOSTICS_CRASH");
603     // Hack to ensure that diagnostic notes get emitted.
604     Diags.setLastDiagnosticIgnored(false);
605     generateCompilationDiagnostics(C, FailingCommand, AdditionalInformation,
606                                    GeneratedReport);
607     return true;
608   }
609 
610   /// @}
611   /// @name Helper Methods
612   /// @{
613 
614   /// PrintActions - Print the list of actions.
615   void PrintActions(const Compilation &C) const;
616 
617   /// PrintHelp - Print the help text.
618   ///
619   /// \param ShowHidden - Show hidden options.
620   void PrintHelp(bool ShowHidden) const;
621 
622   /// PrintVersion - Print the driver version.
623   void PrintVersion(const Compilation &C, raw_ostream &OS) const;
624 
625   /// GetFilePath - Lookup \p Name in the list of file search paths.
626   ///
627   /// \param TC - The tool chain for additional information on
628   /// directories to search.
629   //
630   // FIXME: This should be in CompilationInfo.
631   std::string GetFilePath(StringRef Name, const ToolChain &TC) const;
632 
633   /// GetProgramPath - Lookup \p Name in the list of program search paths.
634   ///
635   /// \param TC - The provided tool chain for additional information on
636   /// directories to search.
637   //
638   // FIXME: This should be in CompilationInfo.
639   std::string GetProgramPath(StringRef Name, const ToolChain &TC) const;
640 
641   /// Lookup the path to the Standard library module manifest.
642   ///
643   /// \param C - The compilation.
644   /// \param TC - The tool chain for additional information on
645   /// directories to search.
646   //
647   // FIXME: This should be in CompilationInfo.
648   std::string GetStdModuleManifestPath(const Compilation &C,
649                                        const ToolChain &TC) const;
650 
651   /// HandleAutocompletions - Handle --autocomplete by searching and printing
652   /// possible flags, descriptions, and its arguments.
653   void HandleAutocompletions(StringRef PassedFlags) const;
654 
655   /// HandleImmediateArgs - Handle any arguments which should be
656   /// treated before building actions or binding tools.
657   ///
658   /// \return Whether any compilation should be built for this
659   /// invocation. The compilation can only be modified when
660   /// this function returns false.
661   bool HandleImmediateArgs(Compilation &C);
662 
663   /// ConstructAction - Construct the appropriate action to do for
664   /// \p Phase on the \p Input, taking in to account arguments
665   /// like -fsyntax-only or --analyze.
666   Action *ConstructPhaseAction(
667       Compilation &C, const llvm::opt::ArgList &Args, phases::ID Phase,
668       Action *Input,
669       Action::OffloadKind TargetDeviceOffloadKind = Action::OFK_None) const;
670 
671   /// BuildJobsForAction - Construct the jobs to perform for the action \p A and
672   /// return an InputInfo for the result of running \p A.  Will only construct
673   /// jobs for a given (Action, ToolChain, BoundArch, DeviceKind) tuple once.
674   InputInfoList BuildJobsForAction(
675       Compilation &C, const Action *A, const ToolChain *TC, StringRef BoundArch,
676       bool AtTopLevel, bool MultipleArchs, const char *LinkingOutput,
677       std::map<std::pair<const Action *, std::string>, InputInfoList>
678           &CachedResults,
679       Action::OffloadKind TargetDeviceOffloadKind) const;
680 
681   /// Returns the default name for linked images (e.g., "a.out").
682   const char *getDefaultImageName() const;
683 
684   /// Creates a temp file.
685   /// 1. If \p MultipleArch is false or \p BoundArch is empty, the temp file is
686   ///    in the temporary directory with name $Prefix-%%%%%%.$Suffix.
687   /// 2. If \p MultipleArch is true and \p BoundArch is not empty,
688   ///    2a. If \p NeedUniqueDirectory is false, the temp file is in the
689   ///        temporary directory with name $Prefix-$BoundArch-%%%%%.$Suffix.
690   ///    2b. If \p NeedUniqueDirectory is true, the temp file is in a unique
691   ///        subdiretory with random name under the temporary directory, and
692   ///        the temp file itself has name $Prefix-$BoundArch.$Suffix.
693   const char *CreateTempFile(Compilation &C, StringRef Prefix, StringRef Suffix,
694                              bool MultipleArchs = false,
695                              StringRef BoundArch = {},
696                              bool NeedUniqueDirectory = false) const;
697 
698   /// GetNamedOutputPath - Return the name to use for the output of
699   /// the action \p JA. The result is appended to the compilation's
700   /// list of temporary or result files, as appropriate.
701   ///
702   /// \param C - The compilation.
703   /// \param JA - The action of interest.
704   /// \param BaseInput - The original input file that this action was
705   /// triggered by.
706   /// \param BoundArch - The bound architecture.
707   /// \param AtTopLevel - Whether this is a "top-level" action.
708   /// \param MultipleArchs - Whether multiple -arch options were supplied.
709   /// \param NormalizedTriple - The normalized triple of the relevant target.
710   const char *GetNamedOutputPath(Compilation &C, const JobAction &JA,
711                                  const char *BaseInput, StringRef BoundArch,
712                                  bool AtTopLevel, bool MultipleArchs,
713                                  StringRef NormalizedTriple) const;
714 
715   /// GetTemporaryPath - Return the pathname of a temporary file to use
716   /// as part of compilation; the file will have the given prefix and suffix.
717   ///
718   /// GCC goes to extra lengths here to be a bit more robust.
719   std::string GetTemporaryPath(StringRef Prefix, StringRef Suffix) const;
720 
721   /// GetTemporaryDirectory - Return the pathname of a temporary directory to
722   /// use as part of compilation; the directory will have the given prefix.
723   std::string GetTemporaryDirectory(StringRef Prefix) const;
724 
725   /// Return the pathname of the pch file in clang-cl mode.
726   std::string GetClPchPath(Compilation &C, StringRef BaseName) const;
727 
728   /// ShouldUseClangCompiler - Should the clang compiler be used to
729   /// handle this action.
730   bool ShouldUseClangCompiler(const JobAction &JA) const;
731 
732   /// ShouldUseFlangCompiler - Should the flang compiler be used to
733   /// handle this action.
734   bool ShouldUseFlangCompiler(const JobAction &JA) const;
735 
736   /// ShouldEmitStaticLibrary - Should the linker emit a static library.
737   bool ShouldEmitStaticLibrary(const llvm::opt::ArgList &Args) const;
738 
739   /// Returns true if the user has indicated a C++20 header unit mode.
740   bool hasHeaderMode() const { return CXX20HeaderType != HeaderMode_None; }
741 
742   /// Get the mode for handling headers as set by fmodule-header{=}.
743   ModuleHeaderMode getModuleHeaderMode() const { return CXX20HeaderType; }
744 
745   /// Returns true if we are performing any kind of LTO.
746   bool isUsingLTO() const { return getLTOMode() != LTOK_None; }
747 
748   /// Get the specific kind of LTO being performed.
749   LTOKind getLTOMode() const { return LTOMode; }
750 
751   /// Returns true if we are performing any kind of offload LTO.
752   bool isUsingOffloadLTO() const { return getOffloadLTOMode() != LTOK_None; }
753 
754   /// Get the specific kind of offload LTO being performed.
755   LTOKind getOffloadLTOMode() const { return OffloadLTOMode; }
756 
757   /// Get the CUID option.
758   const CUIDOptions &getCUIDOpts() const { return CUIDOpts; }
759 
760 private:
761 
762   /// Tries to load options from configuration files.
763   ///
764   /// \returns true if error occurred.
765   bool loadConfigFiles();
766 
767   /// Tries to load options from default configuration files (deduced from
768   /// executable filename).
769   ///
770   /// \returns true if error occurred.
771   bool loadDefaultConfigFiles(llvm::cl::ExpansionContext &ExpCtx);
772 
773   /// Tries to load options from customization file.
774   ///
775   /// \returns true if error occurred.
776   bool loadZOSCustomizationFile(llvm::cl::ExpansionContext &);
777 
778   /// Read options from the specified file.
779   ///
780   /// \param [in] FileName File to read.
781   /// \param [in] Search and expansion options.
782   /// \returns true, if error occurred while reading.
783   bool readConfigFile(StringRef FileName, llvm::cl::ExpansionContext &ExpCtx);
784 
785   /// Set the driver mode (cl, gcc, etc) from the value of the `--driver-mode`
786   /// option.
787   void setDriverMode(StringRef DriverModeValue);
788 
789   /// Parse the \p Args list for LTO options and record the type of LTO
790   /// compilation based on which -f(no-)?lto(=.*)? option occurs last.
791   void setLTOMode(const llvm::opt::ArgList &Args);
792 
793   /// Retrieves a ToolChain for a particular \p Target triple.
794   ///
795   /// Will cache ToolChains for the life of the driver object, and create them
796   /// on-demand.
797   const ToolChain &getToolChain(const llvm::opt::ArgList &Args,
798                                 const llvm::Triple &Target) const;
799 
800   /// @}
801 
802   /// Retrieves a ToolChain for a particular device \p Target triple
803   ///
804   /// \param[in] HostTC is the host ToolChain paired with the device
805   ///
806   /// \param[in] TargetDeviceOffloadKind (e.g. OFK_Cuda/OFK_OpenMP/OFK_SYCL) is
807   /// an Offloading action that is optionally passed to a ToolChain (used by
808   /// CUDA, to specify if it's used in conjunction with OpenMP)
809   ///
810   /// Will cache ToolChains for the life of the driver object, and create them
811   /// on-demand.
812   const ToolChain &getOffloadingDeviceToolChain(
813       const llvm::opt::ArgList &Args, const llvm::Triple &Target,
814       const ToolChain &HostTC,
815       const Action::OffloadKind &TargetDeviceOffloadKind) const;
816 
817   /// Get bitmasks for which option flags to include and exclude based on
818   /// the driver mode.
819   llvm::opt::Visibility
820   getOptionVisibilityMask(bool UseDriverMode = true) const;
821 
822   /// Helper used in BuildJobsForAction.  Doesn't use the cache when building
823   /// jobs specifically for the given action, but will use the cache when
824   /// building jobs for the Action's inputs.
825   InputInfoList BuildJobsForActionNoCache(
826       Compilation &C, const Action *A, const ToolChain *TC, StringRef BoundArch,
827       bool AtTopLevel, bool MultipleArchs, const char *LinkingOutput,
828       std::map<std::pair<const Action *, std::string>, InputInfoList>
829           &CachedResults,
830       Action::OffloadKind TargetDeviceOffloadKind) const;
831 
832   /// Return the typical executable name for the specified driver \p Mode.
833   static const char *getExecutableForDriverMode(DriverMode Mode);
834 
835 public:
836   /// GetReleaseVersion - Parse (([0-9]+)(.([0-9]+)(.([0-9]+)?))?)? and
837   /// return the grouped values as integers. Numbers which are not
838   /// provided are set to 0.
839   ///
840   /// \return True if the entire string was parsed (9.2), or all
841   /// groups were parsed (10.3.5extrastuff). HadExtra is true if all
842   /// groups were parsed but extra characters remain at the end.
843   static bool GetReleaseVersion(StringRef Str, unsigned &Major, unsigned &Minor,
844                                 unsigned &Micro, bool &HadExtra);
845 
846   /// Parse digits from a string \p Str and fulfill \p Digits with
847   /// the parsed numbers. This method assumes that the max number of
848   /// digits to look for is equal to Digits.size().
849   ///
850   /// \return True if the entire string was parsed and there are
851   /// no extra characters remaining at the end.
852   static bool GetReleaseVersion(StringRef Str,
853                                 MutableArrayRef<unsigned> Digits);
854   /// Compute the default -fmodule-cache-path.
855   /// \return True if the system provides a default cache directory.
856   static bool getDefaultModuleCachePath(SmallVectorImpl<char> &Result);
857 };
858 
859 /// \return True if the last defined optimization level is -Ofast.
860 /// And False otherwise.
861 bool isOptimizationLevelFast(const llvm::opt::ArgList &Args);
862 
863 /// \return True if the argument combination will end up generating remarks.
864 bool willEmitRemarks(const llvm::opt::ArgList &Args);
865 
866 /// Returns the driver mode option's value, i.e. `X` in `--driver-mode=X`. If \p
867 /// Args doesn't mention one explicitly, tries to deduce from `ProgName`.
868 /// Returns empty on failure.
869 /// Common values are "gcc", "g++", "cpp", "cl" and "flang". Returned value need
870 /// not be one of these.
871 llvm::StringRef getDriverMode(StringRef ProgName, ArrayRef<const char *> Args);
872 
873 /// Checks whether the value produced by getDriverMode is for CL mode.
874 bool IsClangCL(StringRef DriverMode);
875 
876 /// Expand response files from a clang driver or cc1 invocation.
877 ///
878 /// \param Args The arguments that will be expanded.
879 /// \param ClangCLMode Whether clang is in CL mode.
880 /// \param Alloc Allocator for new arguments.
881 /// \param FS Filesystem to use when expanding files.
882 llvm::Error expandResponseFiles(SmallVectorImpl<const char *> &Args,
883                                 bool ClangCLMode, llvm::BumpPtrAllocator &Alloc,
884                                 llvm::vfs::FileSystem *FS = nullptr);
885 
886 /// Apply a space separated list of edits to the input argument lists.
887 /// See applyOneOverrideOption.
888 void applyOverrideOptions(SmallVectorImpl<const char *> &Args,
889                           const char *OverrideOpts,
890                           llvm::StringSet<> &SavedStrings,
891                           raw_ostream *OS = nullptr);
892 
893 } // end namespace driver
894 } // end namespace clang
895 
896 #endif
897