xref: /freebsd/contrib/llvm-project/clang/lib/Frontend/CompilerInvocation.cpp (revision e64bea71c21eb42e97aa615188ba91f6cce0d36d)
1 //===- CompilerInvocation.cpp ---------------------------------------------===//
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 #include "clang/Frontend/CompilerInvocation.h"
10 #include "TestModuleFileExtension.h"
11 #include "clang/Basic/Builtins.h"
12 #include "clang/Basic/CharInfo.h"
13 #include "clang/Basic/CodeGenOptions.h"
14 #include "clang/Basic/CommentOptions.h"
15 #include "clang/Basic/Diagnostic.h"
16 #include "clang/Basic/DiagnosticDriver.h"
17 #include "clang/Basic/DiagnosticOptions.h"
18 #include "clang/Basic/FileSystemOptions.h"
19 #include "clang/Basic/LLVM.h"
20 #include "clang/Basic/LangOptions.h"
21 #include "clang/Basic/LangStandard.h"
22 #include "clang/Basic/ObjCRuntime.h"
23 #include "clang/Basic/Sanitizers.h"
24 #include "clang/Basic/SourceLocation.h"
25 #include "clang/Basic/TargetOptions.h"
26 #include "clang/Basic/Version.h"
27 #include "clang/Basic/XRayInstr.h"
28 #include "clang/Config/config.h"
29 #include "clang/Driver/Driver.h"
30 #include "clang/Driver/Options.h"
31 #include "clang/Frontend/CommandLineSourceLoc.h"
32 #include "clang/Frontend/DependencyOutputOptions.h"
33 #include "clang/Frontend/FrontendDiagnostic.h"
34 #include "clang/Frontend/FrontendOptions.h"
35 #include "clang/Frontend/MigratorOptions.h"
36 #include "clang/Frontend/PreprocessorOutputOptions.h"
37 #include "clang/Frontend/TextDiagnosticBuffer.h"
38 #include "clang/Frontend/Utils.h"
39 #include "clang/Lex/HeaderSearchOptions.h"
40 #include "clang/Lex/PreprocessorOptions.h"
41 #include "clang/Serialization/ASTBitCodes.h"
42 #include "clang/Serialization/ModuleFileExtension.h"
43 #include "clang/StaticAnalyzer/Core/AnalyzerOptions.h"
44 #include "llvm/ADT/APInt.h"
45 #include "llvm/ADT/ArrayRef.h"
46 #include "llvm/ADT/CachedHashString.h"
47 #include "llvm/ADT/FloatingPointMode.h"
48 #include "llvm/ADT/STLExtras.h"
49 #include "llvm/ADT/SmallVector.h"
50 #include "llvm/ADT/StringRef.h"
51 #include "llvm/ADT/StringSwitch.h"
52 #include "llvm/ADT/Twine.h"
53 #include "llvm/Config/llvm-config.h"
54 #include "llvm/Frontend/Debug/Options.h"
55 #include "llvm/IR/DebugInfoMetadata.h"
56 #include "llvm/Linker/Linker.h"
57 #include "llvm/MC/MCTargetOptions.h"
58 #include "llvm/Option/Arg.h"
59 #include "llvm/Option/ArgList.h"
60 #include "llvm/Option/OptSpecifier.h"
61 #include "llvm/Option/OptTable.h"
62 #include "llvm/Option/Option.h"
63 #include "llvm/ProfileData/InstrProfReader.h"
64 #include "llvm/Remarks/HotnessThresholdParser.h"
65 #include "llvm/Support/CodeGen.h"
66 #include "llvm/Support/Compiler.h"
67 #include "llvm/Support/Error.h"
68 #include "llvm/Support/ErrorHandling.h"
69 #include "llvm/Support/ErrorOr.h"
70 #include "llvm/Support/FileSystem.h"
71 #include "llvm/Support/HashBuilder.h"
72 #include "llvm/Support/MathExtras.h"
73 #include "llvm/Support/MemoryBuffer.h"
74 #include "llvm/Support/Path.h"
75 #include "llvm/Support/Process.h"
76 #include "llvm/Support/Regex.h"
77 #include "llvm/Support/VersionTuple.h"
78 #include "llvm/Support/VirtualFileSystem.h"
79 #include "llvm/Support/raw_ostream.h"
80 #include "llvm/Target/TargetOptions.h"
81 #include "llvm/TargetParser/Host.h"
82 #include "llvm/TargetParser/Triple.h"
83 #include <algorithm>
84 #include <cassert>
85 #include <cstddef>
86 #include <cstring>
87 #include <ctime>
88 #include <fstream>
89 #include <limits>
90 #include <memory>
91 #include <optional>
92 #include <string>
93 #include <tuple>
94 #include <type_traits>
95 #include <utility>
96 #include <vector>
97 
98 using namespace clang;
99 using namespace driver;
100 using namespace options;
101 using namespace llvm::opt;
102 
103 //===----------------------------------------------------------------------===//
104 // Helpers.
105 //===----------------------------------------------------------------------===//
106 
107 // Parse misexpect tolerance argument value.
108 // Valid option values are integers in the range [0, 100)
parseToleranceOption(StringRef Arg)109 static Expected<std::optional<uint32_t>> parseToleranceOption(StringRef Arg) {
110   uint32_t Val;
111   if (Arg.getAsInteger(10, Val))
112     return llvm::createStringError(llvm::inconvertibleErrorCode(),
113                                    "Not an integer: %s", Arg.data());
114   return Val;
115 }
116 
117 //===----------------------------------------------------------------------===//
118 // Initialization.
119 //===----------------------------------------------------------------------===//
120 
make_shared_copy(const T & X)121 template <class T> std::shared_ptr<T> make_shared_copy(const T &X) {
122   return std::make_shared<T>(X);
123 }
124 
CompilerInvocationBase()125 CompilerInvocationBase::CompilerInvocationBase()
126     : LangOpts(std::make_shared<LangOptions>()),
127       TargetOpts(std::make_shared<TargetOptions>()),
128       DiagnosticOpts(std::make_shared<DiagnosticOptions>()),
129       HSOpts(std::make_shared<HeaderSearchOptions>()),
130       PPOpts(std::make_shared<PreprocessorOptions>()),
131       AnalyzerOpts(std::make_shared<AnalyzerOptions>()),
132       MigratorOpts(std::make_shared<MigratorOptions>()),
133       APINotesOpts(std::make_shared<APINotesOptions>()),
134       CodeGenOpts(std::make_shared<CodeGenOptions>()),
135       FSOpts(std::make_shared<FileSystemOptions>()),
136       FrontendOpts(std::make_shared<FrontendOptions>()),
137       DependencyOutputOpts(std::make_shared<DependencyOutputOptions>()),
138       PreprocessorOutputOpts(std::make_shared<PreprocessorOutputOptions>()) {}
139 
140 CompilerInvocationBase &
deep_copy_assign(const CompilerInvocationBase & X)141 CompilerInvocationBase::deep_copy_assign(const CompilerInvocationBase &X) {
142   if (this != &X) {
143     LangOpts = make_shared_copy(X.getLangOpts());
144     TargetOpts = make_shared_copy(X.getTargetOpts());
145     DiagnosticOpts = make_shared_copy(X.getDiagnosticOpts());
146     HSOpts = make_shared_copy(X.getHeaderSearchOpts());
147     PPOpts = make_shared_copy(X.getPreprocessorOpts());
148     AnalyzerOpts = make_shared_copy(X.getAnalyzerOpts());
149     MigratorOpts = make_shared_copy(X.getMigratorOpts());
150     APINotesOpts = make_shared_copy(X.getAPINotesOpts());
151     CodeGenOpts = make_shared_copy(X.getCodeGenOpts());
152     FSOpts = make_shared_copy(X.getFileSystemOpts());
153     FrontendOpts = make_shared_copy(X.getFrontendOpts());
154     DependencyOutputOpts = make_shared_copy(X.getDependencyOutputOpts());
155     PreprocessorOutputOpts = make_shared_copy(X.getPreprocessorOutputOpts());
156   }
157   return *this;
158 }
159 
160 CompilerInvocationBase &
shallow_copy_assign(const CompilerInvocationBase & X)161 CompilerInvocationBase::shallow_copy_assign(const CompilerInvocationBase &X) {
162   if (this != &X) {
163     LangOpts = X.LangOpts;
164     TargetOpts = X.TargetOpts;
165     DiagnosticOpts = X.DiagnosticOpts;
166     HSOpts = X.HSOpts;
167     PPOpts = X.PPOpts;
168     AnalyzerOpts = X.AnalyzerOpts;
169     MigratorOpts = X.MigratorOpts;
170     APINotesOpts = X.APINotesOpts;
171     CodeGenOpts = X.CodeGenOpts;
172     FSOpts = X.FSOpts;
173     FrontendOpts = X.FrontendOpts;
174     DependencyOutputOpts = X.DependencyOutputOpts;
175     PreprocessorOutputOpts = X.PreprocessorOutputOpts;
176   }
177   return *this;
178 }
179 
CompilerInvocation(const CowCompilerInvocation & X)180 CompilerInvocation::CompilerInvocation(const CowCompilerInvocation &X)
181     : CompilerInvocationBase(EmptyConstructor{}) {
182   CompilerInvocationBase::deep_copy_assign(X);
183 }
184 
185 CompilerInvocation &
operator =(const CowCompilerInvocation & X)186 CompilerInvocation::operator=(const CowCompilerInvocation &X) {
187   CompilerInvocationBase::deep_copy_assign(X);
188   return *this;
189 }
190 
191 template <typename T>
ensureOwned(std::shared_ptr<T> & Storage)192 T &ensureOwned(std::shared_ptr<T> &Storage) {
193   if (Storage.use_count() > 1)
194     Storage = std::make_shared<T>(*Storage);
195   return *Storage;
196 }
197 
getMutLangOpts()198 LangOptions &CowCompilerInvocation::getMutLangOpts() {
199   return ensureOwned(LangOpts);
200 }
201 
getMutTargetOpts()202 TargetOptions &CowCompilerInvocation::getMutTargetOpts() {
203   return ensureOwned(TargetOpts);
204 }
205 
getMutDiagnosticOpts()206 DiagnosticOptions &CowCompilerInvocation::getMutDiagnosticOpts() {
207   return ensureOwned(DiagnosticOpts);
208 }
209 
getMutHeaderSearchOpts()210 HeaderSearchOptions &CowCompilerInvocation::getMutHeaderSearchOpts() {
211   return ensureOwned(HSOpts);
212 }
213 
getMutPreprocessorOpts()214 PreprocessorOptions &CowCompilerInvocation::getMutPreprocessorOpts() {
215   return ensureOwned(PPOpts);
216 }
217 
getMutAnalyzerOpts()218 AnalyzerOptions &CowCompilerInvocation::getMutAnalyzerOpts() {
219   return ensureOwned(AnalyzerOpts);
220 }
221 
getMutMigratorOpts()222 MigratorOptions &CowCompilerInvocation::getMutMigratorOpts() {
223   return ensureOwned(MigratorOpts);
224 }
225 
getMutAPINotesOpts()226 APINotesOptions &CowCompilerInvocation::getMutAPINotesOpts() {
227   return ensureOwned(APINotesOpts);
228 }
229 
getMutCodeGenOpts()230 CodeGenOptions &CowCompilerInvocation::getMutCodeGenOpts() {
231   return ensureOwned(CodeGenOpts);
232 }
233 
getMutFileSystemOpts()234 FileSystemOptions &CowCompilerInvocation::getMutFileSystemOpts() {
235   return ensureOwned(FSOpts);
236 }
237 
getMutFrontendOpts()238 FrontendOptions &CowCompilerInvocation::getMutFrontendOpts() {
239   return ensureOwned(FrontendOpts);
240 }
241 
getMutDependencyOutputOpts()242 DependencyOutputOptions &CowCompilerInvocation::getMutDependencyOutputOpts() {
243   return ensureOwned(DependencyOutputOpts);
244 }
245 
246 PreprocessorOutputOptions &
getMutPreprocessorOutputOpts()247 CowCompilerInvocation::getMutPreprocessorOutputOpts() {
248   return ensureOwned(PreprocessorOutputOpts);
249 }
250 
251 //===----------------------------------------------------------------------===//
252 // Normalizers
253 //===----------------------------------------------------------------------===//
254 
255 using ArgumentConsumer = CompilerInvocation::ArgumentConsumer;
256 
257 #define OPTTABLE_STR_TABLE_CODE
258 #include "clang/Driver/Options.inc"
259 #undef OPTTABLE_STR_TABLE_CODE
260 
lookupStrInTable(unsigned Offset)261 static llvm::StringRef lookupStrInTable(unsigned Offset) {
262   return OptionStrTable[Offset];
263 }
264 
265 #define SIMPLE_ENUM_VALUE_TABLE
266 #include "clang/Driver/Options.inc"
267 #undef SIMPLE_ENUM_VALUE_TABLE
268 
normalizeSimpleFlag(OptSpecifier Opt,unsigned TableIndex,const ArgList & Args,DiagnosticsEngine & Diags)269 static std::optional<bool> normalizeSimpleFlag(OptSpecifier Opt,
270                                                unsigned TableIndex,
271                                                const ArgList &Args,
272                                                DiagnosticsEngine &Diags) {
273   if (Args.hasArg(Opt))
274     return true;
275   return std::nullopt;
276 }
277 
normalizeSimpleNegativeFlag(OptSpecifier Opt,unsigned,const ArgList & Args,DiagnosticsEngine &)278 static std::optional<bool> normalizeSimpleNegativeFlag(OptSpecifier Opt,
279                                                        unsigned,
280                                                        const ArgList &Args,
281                                                        DiagnosticsEngine &) {
282   if (Args.hasArg(Opt))
283     return false;
284   return std::nullopt;
285 }
286 
287 /// The tblgen-erated code passes in a fifth parameter of an arbitrary type, but
288 /// denormalizeSimpleFlags never looks at it. Avoid bloating compile-time with
289 /// unnecessary template instantiations and just ignore it with a variadic
290 /// argument.
denormalizeSimpleFlag(ArgumentConsumer Consumer,unsigned SpellingOffset,Option::OptionClass,unsigned,...)291 static void denormalizeSimpleFlag(ArgumentConsumer Consumer,
292                                   unsigned SpellingOffset, Option::OptionClass,
293                                   unsigned, /*T*/...) {
294   Consumer(lookupStrInTable(SpellingOffset));
295 }
denormalizeSimpleFlag(ArgumentConsumer Consumer,const Twine & Spelling,Option::OptionClass,unsigned,...)296 static void denormalizeSimpleFlag(ArgumentConsumer Consumer,
297                                   const Twine &Spelling, Option::OptionClass,
298                                   unsigned, /*T*/...) {
299   Consumer(Spelling);
300 }
301 
is_uint64_t_convertible()302 template <typename T> static constexpr bool is_uint64_t_convertible() {
303   return !std::is_same_v<T, uint64_t> && llvm::is_integral_or_enum<T>::value;
304 }
305 
306 template <typename T,
307           std::enable_if_t<!is_uint64_t_convertible<T>(), bool> = false>
makeFlagToValueNormalizer(T Value)308 static auto makeFlagToValueNormalizer(T Value) {
309   return [Value](OptSpecifier Opt, unsigned, const ArgList &Args,
310                  DiagnosticsEngine &) -> std::optional<T> {
311     if (Args.hasArg(Opt))
312       return Value;
313     return std::nullopt;
314   };
315 }
316 
317 template <typename T,
318           std::enable_if_t<is_uint64_t_convertible<T>(), bool> = false>
makeFlagToValueNormalizer(T Value)319 static auto makeFlagToValueNormalizer(T Value) {
320   return makeFlagToValueNormalizer(uint64_t(Value));
321 }
322 
makeBooleanOptionNormalizer(bool Value,bool OtherValue,OptSpecifier OtherOpt)323 static auto makeBooleanOptionNormalizer(bool Value, bool OtherValue,
324                                         OptSpecifier OtherOpt) {
325   return [Value, OtherValue,
326           OtherOpt](OptSpecifier Opt, unsigned, const ArgList &Args,
327                     DiagnosticsEngine &) -> std::optional<bool> {
328     if (const Arg *A = Args.getLastArg(Opt, OtherOpt)) {
329       return A->getOption().matches(Opt) ? Value : OtherValue;
330     }
331     return std::nullopt;
332   };
333 }
334 
makeBooleanOptionDenormalizer(bool Value)335 static auto makeBooleanOptionDenormalizer(bool Value) {
336   return [Value](ArgumentConsumer Consumer, unsigned SpellingOffset,
337                  Option::OptionClass, unsigned, bool KeyPath) {
338     if (KeyPath == Value)
339       Consumer(lookupStrInTable(SpellingOffset));
340   };
341 }
342 
denormalizeStringImpl(ArgumentConsumer Consumer,const Twine & Spelling,Option::OptionClass OptClass,unsigned,const Twine & Value)343 static void denormalizeStringImpl(ArgumentConsumer Consumer,
344                                   const Twine &Spelling,
345                                   Option::OptionClass OptClass, unsigned,
346                                   const Twine &Value) {
347   switch (OptClass) {
348   case Option::SeparateClass:
349   case Option::JoinedOrSeparateClass:
350   case Option::JoinedAndSeparateClass:
351     Consumer(Spelling);
352     Consumer(Value);
353     break;
354   case Option::JoinedClass:
355   case Option::CommaJoinedClass:
356     Consumer(Spelling + Value);
357     break;
358   default:
359     llvm_unreachable("Cannot denormalize an option with option class "
360                      "incompatible with string denormalization.");
361   }
362 }
363 
364 template <typename T>
365 static void
denormalizeString(ArgumentConsumer Consumer,unsigned SpellingOffset,Option::OptionClass OptClass,unsigned TableIndex,T Value)366 denormalizeString(ArgumentConsumer Consumer, unsigned SpellingOffset,
367                   Option::OptionClass OptClass, unsigned TableIndex, T Value) {
368   denormalizeStringImpl(Consumer, lookupStrInTable(SpellingOffset), OptClass,
369                         TableIndex, Twine(Value));
370 }
371 
372 template <typename T>
denormalizeString(ArgumentConsumer Consumer,const Twine & Spelling,Option::OptionClass OptClass,unsigned TableIndex,T Value)373 static void denormalizeString(ArgumentConsumer Consumer, const Twine &Spelling,
374                               Option::OptionClass OptClass, unsigned TableIndex,
375                               T Value) {
376   denormalizeStringImpl(Consumer, Spelling, OptClass, TableIndex, Twine(Value));
377 }
378 
379 static std::optional<SimpleEnumValue>
findValueTableByName(const SimpleEnumValueTable & Table,StringRef Name)380 findValueTableByName(const SimpleEnumValueTable &Table, StringRef Name) {
381   for (int I = 0, E = Table.Size; I != E; ++I)
382     if (Name == Table.Table[I].Name)
383       return Table.Table[I];
384 
385   return std::nullopt;
386 }
387 
388 static std::optional<SimpleEnumValue>
findValueTableByValue(const SimpleEnumValueTable & Table,unsigned Value)389 findValueTableByValue(const SimpleEnumValueTable &Table, unsigned Value) {
390   for (int I = 0, E = Table.Size; I != E; ++I)
391     if (Value == Table.Table[I].Value)
392       return Table.Table[I];
393 
394   return std::nullopt;
395 }
396 
normalizeSimpleEnum(OptSpecifier Opt,unsigned TableIndex,const ArgList & Args,DiagnosticsEngine & Diags)397 static std::optional<unsigned> normalizeSimpleEnum(OptSpecifier Opt,
398                                                    unsigned TableIndex,
399                                                    const ArgList &Args,
400                                                    DiagnosticsEngine &Diags) {
401   assert(TableIndex < SimpleEnumValueTablesSize);
402   const SimpleEnumValueTable &Table = SimpleEnumValueTables[TableIndex];
403 
404   auto *Arg = Args.getLastArg(Opt);
405   if (!Arg)
406     return std::nullopt;
407 
408   StringRef ArgValue = Arg->getValue();
409   if (auto MaybeEnumVal = findValueTableByName(Table, ArgValue))
410     return MaybeEnumVal->Value;
411 
412   Diags.Report(diag::err_drv_invalid_value)
413       << Arg->getAsString(Args) << ArgValue;
414   return std::nullopt;
415 }
416 
denormalizeSimpleEnumImpl(ArgumentConsumer Consumer,unsigned SpellingOffset,Option::OptionClass OptClass,unsigned TableIndex,unsigned Value)417 static void denormalizeSimpleEnumImpl(ArgumentConsumer Consumer,
418                                       unsigned SpellingOffset,
419                                       Option::OptionClass OptClass,
420                                       unsigned TableIndex, unsigned Value) {
421   assert(TableIndex < SimpleEnumValueTablesSize);
422   const SimpleEnumValueTable &Table = SimpleEnumValueTables[TableIndex];
423   if (auto MaybeEnumVal = findValueTableByValue(Table, Value)) {
424     denormalizeString(Consumer, lookupStrInTable(SpellingOffset), OptClass,
425                       TableIndex, MaybeEnumVal->Name);
426   } else {
427     llvm_unreachable("The simple enum value was not correctly defined in "
428                      "the tablegen option description");
429   }
430 }
431 
432 template <typename T>
denormalizeSimpleEnum(ArgumentConsumer Consumer,unsigned SpellingOffset,Option::OptionClass OptClass,unsigned TableIndex,T Value)433 static void denormalizeSimpleEnum(ArgumentConsumer Consumer,
434                                   unsigned SpellingOffset,
435                                   Option::OptionClass OptClass,
436                                   unsigned TableIndex, T Value) {
437   return denormalizeSimpleEnumImpl(Consumer, SpellingOffset, OptClass,
438                                    TableIndex, static_cast<unsigned>(Value));
439 }
440 
normalizeString(OptSpecifier Opt,int TableIndex,const ArgList & Args,DiagnosticsEngine & Diags)441 static std::optional<std::string> normalizeString(OptSpecifier Opt,
442                                                   int TableIndex,
443                                                   const ArgList &Args,
444                                                   DiagnosticsEngine &Diags) {
445   auto *Arg = Args.getLastArg(Opt);
446   if (!Arg)
447     return std::nullopt;
448   return std::string(Arg->getValue());
449 }
450 
451 template <typename IntTy>
normalizeStringIntegral(OptSpecifier Opt,int,const ArgList & Args,DiagnosticsEngine & Diags)452 static std::optional<IntTy> normalizeStringIntegral(OptSpecifier Opt, int,
453                                                     const ArgList &Args,
454                                                     DiagnosticsEngine &Diags) {
455   auto *Arg = Args.getLastArg(Opt);
456   if (!Arg)
457     return std::nullopt;
458   IntTy Res;
459   if (StringRef(Arg->getValue()).getAsInteger(0, Res)) {
460     Diags.Report(diag::err_drv_invalid_int_value)
461         << Arg->getAsString(Args) << Arg->getValue();
462     return std::nullopt;
463   }
464   return Res;
465 }
466 
467 static std::optional<std::vector<std::string>>
normalizeStringVector(OptSpecifier Opt,int,const ArgList & Args,DiagnosticsEngine &)468 normalizeStringVector(OptSpecifier Opt, int, const ArgList &Args,
469                       DiagnosticsEngine &) {
470   return Args.getAllArgValues(Opt);
471 }
472 
denormalizeStringVector(ArgumentConsumer Consumer,unsigned SpellingOffset,Option::OptionClass OptClass,unsigned TableIndex,const std::vector<std::string> & Values)473 static void denormalizeStringVector(ArgumentConsumer Consumer,
474                                     unsigned SpellingOffset,
475                                     Option::OptionClass OptClass,
476                                     unsigned TableIndex,
477                                     const std::vector<std::string> &Values) {
478   switch (OptClass) {
479   case Option::CommaJoinedClass: {
480     std::string CommaJoinedValue;
481     if (!Values.empty()) {
482       CommaJoinedValue.append(Values.front());
483       for (const std::string &Value : llvm::drop_begin(Values, 1)) {
484         CommaJoinedValue.append(",");
485         CommaJoinedValue.append(Value);
486       }
487     }
488     denormalizeString(Consumer, SpellingOffset,
489                       Option::OptionClass::JoinedClass, TableIndex,
490                       CommaJoinedValue);
491     break;
492   }
493   case Option::JoinedClass:
494   case Option::SeparateClass:
495   case Option::JoinedOrSeparateClass:
496     for (const std::string &Value : Values)
497       denormalizeString(Consumer, SpellingOffset, OptClass, TableIndex, Value);
498     break;
499   default:
500     llvm_unreachable("Cannot denormalize an option with option class "
501                      "incompatible with string vector denormalization.");
502   }
503 }
504 
normalizeTriple(OptSpecifier Opt,int TableIndex,const ArgList & Args,DiagnosticsEngine & Diags)505 static std::optional<std::string> normalizeTriple(OptSpecifier Opt,
506                                                   int TableIndex,
507                                                   const ArgList &Args,
508                                                   DiagnosticsEngine &Diags) {
509   auto *Arg = Args.getLastArg(Opt);
510   if (!Arg)
511     return std::nullopt;
512   return llvm::Triple::normalize(Arg->getValue());
513 }
514 
515 template <typename T, typename U>
mergeForwardValue(T KeyPath,U Value)516 static T mergeForwardValue(T KeyPath, U Value) {
517   return static_cast<T>(Value);
518 }
519 
mergeMaskValue(T KeyPath,U Value)520 template <typename T, typename U> static T mergeMaskValue(T KeyPath, U Value) {
521   return KeyPath | Value;
522 }
523 
extractForwardValue(T KeyPath)524 template <typename T> static T extractForwardValue(T KeyPath) {
525   return KeyPath;
526 }
527 
528 template <typename T, typename U, U Value>
extractMaskValue(T KeyPath)529 static T extractMaskValue(T KeyPath) {
530   return ((KeyPath & Value) == Value) ? static_cast<T>(Value) : T();
531 }
532 
533 #define PARSE_OPTION_WITH_MARSHALLING(                                         \
534     ARGS, DIAGS, PREFIX_TYPE, SPELLING_OFFSET, ID, KIND, GROUP, ALIAS,         \
535     ALIASARGS, FLAGS, VISIBILITY, PARAM, HELPTEXT, HELPTEXTSFORVARIANTS,       \
536     METAVAR, VALUES, SHOULD_PARSE, ALWAYS_EMIT, KEYPATH, DEFAULT_VALUE,        \
537     IMPLIED_CHECK, IMPLIED_VALUE, NORMALIZER, DENORMALIZER, MERGER, EXTRACTOR, \
538     TABLE_INDEX)                                                               \
539   if ((VISIBILITY) & options::CC1Option) {                                     \
540     KEYPATH = MERGER(KEYPATH, DEFAULT_VALUE);                                  \
541     if (IMPLIED_CHECK)                                                         \
542       KEYPATH = MERGER(KEYPATH, IMPLIED_VALUE);                                \
543     if (SHOULD_PARSE)                                                          \
544       if (auto MaybeValue = NORMALIZER(OPT_##ID, TABLE_INDEX, ARGS, DIAGS))    \
545         KEYPATH =                                                              \
546             MERGER(KEYPATH, static_cast<decltype(KEYPATH)>(*MaybeValue));      \
547   }
548 
549 // Capture the extracted value as a lambda argument to avoid potential issues
550 // with lifetime extension of the reference.
551 #define GENERATE_OPTION_WITH_MARSHALLING(                                      \
552     CONSUMER, PREFIX_TYPE, SPELLING_OFFSET, ID, KIND, GROUP, ALIAS, ALIASARGS, \
553     FLAGS, VISIBILITY, PARAM, HELPTEXT, HELPTEXTSFORVARIANTS, METAVAR, VALUES, \
554     SHOULD_PARSE, ALWAYS_EMIT, KEYPATH, DEFAULT_VALUE, IMPLIED_CHECK,          \
555     IMPLIED_VALUE, NORMALIZER, DENORMALIZER, MERGER, EXTRACTOR, TABLE_INDEX)   \
556   if ((VISIBILITY) & options::CC1Option) {                                     \
557     [&](const auto &Extracted) {                                               \
558       if (ALWAYS_EMIT ||                                                       \
559           (Extracted !=                                                        \
560            static_cast<decltype(KEYPATH)>((IMPLIED_CHECK) ? (IMPLIED_VALUE)    \
561                                                           : (DEFAULT_VALUE)))) \
562         DENORMALIZER(CONSUMER, SPELLING_OFFSET, Option::KIND##Class,           \
563                      TABLE_INDEX, Extracted);                                  \
564     }(EXTRACTOR(KEYPATH));                                                     \
565   }
566 
567 static StringRef GetInputKindName(InputKind IK);
568 
FixupInvocation(CompilerInvocation & Invocation,DiagnosticsEngine & Diags,const ArgList & Args,InputKind IK)569 static bool FixupInvocation(CompilerInvocation &Invocation,
570                             DiagnosticsEngine &Diags, const ArgList &Args,
571                             InputKind IK) {
572   unsigned NumErrorsBefore = Diags.getNumErrors();
573 
574   LangOptions &LangOpts = Invocation.getLangOpts();
575   CodeGenOptions &CodeGenOpts = Invocation.getCodeGenOpts();
576   TargetOptions &TargetOpts = Invocation.getTargetOpts();
577   FrontendOptions &FrontendOpts = Invocation.getFrontendOpts();
578   CodeGenOpts.XRayInstrumentFunctions = LangOpts.XRayInstrument;
579   CodeGenOpts.XRayAlwaysEmitCustomEvents = LangOpts.XRayAlwaysEmitCustomEvents;
580   CodeGenOpts.XRayAlwaysEmitTypedEvents = LangOpts.XRayAlwaysEmitTypedEvents;
581   CodeGenOpts.DisableFree = FrontendOpts.DisableFree;
582   FrontendOpts.GenerateGlobalModuleIndex = FrontendOpts.UseGlobalModuleIndex;
583   if (FrontendOpts.ShowStats)
584     CodeGenOpts.ClearASTBeforeBackend = false;
585   LangOpts.SanitizeCoverage = CodeGenOpts.hasSanitizeCoverage();
586   LangOpts.ForceEmitVTables = CodeGenOpts.ForceEmitVTables;
587   LangOpts.SpeculativeLoadHardening = CodeGenOpts.SpeculativeLoadHardening;
588   LangOpts.CurrentModule = LangOpts.ModuleName;
589 
590   llvm::Triple T(TargetOpts.Triple);
591   llvm::Triple::ArchType Arch = T.getArch();
592 
593   CodeGenOpts.CodeModel = TargetOpts.CodeModel;
594   CodeGenOpts.LargeDataThreshold = TargetOpts.LargeDataThreshold;
595 
596   if (LangOpts.getExceptionHandling() !=
597           LangOptions::ExceptionHandlingKind::None &&
598       T.isWindowsMSVCEnvironment())
599     Diags.Report(diag::err_fe_invalid_exception_model)
600         << static_cast<unsigned>(LangOpts.getExceptionHandling()) << T.str();
601 
602   if (LangOpts.AppleKext && !LangOpts.CPlusPlus)
603     Diags.Report(diag::warn_c_kext);
604 
605   if (LangOpts.NewAlignOverride &&
606       !llvm::isPowerOf2_32(LangOpts.NewAlignOverride)) {
607     Arg *A = Args.getLastArg(OPT_fnew_alignment_EQ);
608     Diags.Report(diag::err_fe_invalid_alignment)
609         << A->getAsString(Args) << A->getValue();
610     LangOpts.NewAlignOverride = 0;
611   }
612 
613   // The -f[no-]raw-string-literals option is only valid in C and in C++
614   // standards before C++11.
615   if (LangOpts.CPlusPlus11) {
616     if (Args.hasArg(OPT_fraw_string_literals, OPT_fno_raw_string_literals)) {
617       Args.claimAllArgs(OPT_fraw_string_literals, OPT_fno_raw_string_literals);
618       Diags.Report(diag::warn_drv_fraw_string_literals_in_cxx11)
619           << bool(LangOpts.RawStringLiterals);
620     }
621 
622     // Do not allow disabling raw string literals in C++11 or later.
623     LangOpts.RawStringLiterals = true;
624   }
625 
626   // Prevent the user from specifying both -fsycl-is-device and -fsycl-is-host.
627   if (LangOpts.SYCLIsDevice && LangOpts.SYCLIsHost)
628     Diags.Report(diag::err_drv_argument_not_allowed_with) << "-fsycl-is-device"
629                                                           << "-fsycl-is-host";
630 
631   if (Args.hasArg(OPT_fgnu89_inline) && LangOpts.CPlusPlus)
632     Diags.Report(diag::err_drv_argument_not_allowed_with)
633         << "-fgnu89-inline" << GetInputKindName(IK);
634 
635   if (Args.hasArg(OPT_hlsl_entrypoint) && !LangOpts.HLSL)
636     Diags.Report(diag::err_drv_argument_not_allowed_with)
637         << "-hlsl-entry" << GetInputKindName(IK);
638 
639   if (Args.hasArg(OPT_fdx_rootsignature_version) && !LangOpts.HLSL)
640     Diags.Report(diag::err_drv_argument_not_allowed_with)
641         << "-fdx-rootsignature-version" << GetInputKindName(IK);
642 
643   if (Args.hasArg(OPT_fgpu_allow_device_init) && !LangOpts.HIP)
644     Diags.Report(diag::warn_ignored_hip_only_option)
645         << Args.getLastArg(OPT_fgpu_allow_device_init)->getAsString(Args);
646 
647   if (Args.hasArg(OPT_gpu_max_threads_per_block_EQ) && !LangOpts.HIP)
648     Diags.Report(diag::warn_ignored_hip_only_option)
649         << Args.getLastArg(OPT_gpu_max_threads_per_block_EQ)->getAsString(Args);
650 
651   // When these options are used, the compiler is allowed to apply
652   // optimizations that may affect the final result. For example
653   // (x+y)+z is transformed to x+(y+z) but may not give the same
654   // final result; it's not value safe.
655   // Another example can be to simplify x/x to 1.0 but x could be 0.0, INF
656   // or NaN. Final result may then differ. An error is issued when the eval
657   // method is set with one of these options.
658   if (Args.hasArg(OPT_ffp_eval_method_EQ)) {
659     if (LangOpts.ApproxFunc)
660       Diags.Report(diag::err_incompatible_fp_eval_method_options) << 0;
661     if (LangOpts.AllowFPReassoc)
662       Diags.Report(diag::err_incompatible_fp_eval_method_options) << 1;
663     if (LangOpts.AllowRecip)
664       Diags.Report(diag::err_incompatible_fp_eval_method_options) << 2;
665   }
666 
667   // -cl-strict-aliasing needs to emit diagnostic in the case where CL > 1.0.
668   // This option should be deprecated for CL > 1.0 because
669   // this option was added for compatibility with OpenCL 1.0.
670   if (Args.getLastArg(OPT_cl_strict_aliasing) &&
671       (LangOpts.getOpenCLCompatibleVersion() > 100))
672     Diags.Report(diag::warn_option_invalid_ocl_version)
673         << LangOpts.getOpenCLVersionString()
674         << Args.getLastArg(OPT_cl_strict_aliasing)->getAsString(Args);
675 
676   if (Arg *A = Args.getLastArg(OPT_fdefault_calling_conv_EQ)) {
677     auto DefaultCC = LangOpts.getDefaultCallingConv();
678 
679     bool emitError = (DefaultCC == LangOptions::DCC_FastCall ||
680                       DefaultCC == LangOptions::DCC_StdCall) &&
681                      Arch != llvm::Triple::x86;
682     emitError |= (DefaultCC == LangOptions::DCC_VectorCall ||
683                   DefaultCC == LangOptions::DCC_RegCall) &&
684                  !T.isX86();
685     emitError |= DefaultCC == LangOptions::DCC_RtdCall && Arch != llvm::Triple::m68k;
686     if (emitError)
687       Diags.Report(diag::err_drv_argument_not_allowed_with)
688           << A->getSpelling() << T.getTriple();
689   }
690 
691   return Diags.getNumErrors() == NumErrorsBefore;
692 }
693 
694 //===----------------------------------------------------------------------===//
695 // Deserialization (from args)
696 //===----------------------------------------------------------------------===//
697 
getOptimizationLevel(ArgList & Args,InputKind IK,DiagnosticsEngine & Diags)698 static unsigned getOptimizationLevel(ArgList &Args, InputKind IK,
699                                      DiagnosticsEngine &Diags) {
700   unsigned DefaultOpt = 0;
701   if ((IK.getLanguage() == Language::OpenCL ||
702        IK.getLanguage() == Language::OpenCLCXX) &&
703       !Args.hasArg(OPT_cl_opt_disable))
704     DefaultOpt = 2;
705 
706   if (Arg *A = Args.getLastArg(options::OPT_O_Group)) {
707     if (A->getOption().matches(options::OPT_O0))
708       return 0;
709 
710     if (A->getOption().matches(options::OPT_Ofast))
711       return 3;
712 
713     assert(A->getOption().matches(options::OPT_O));
714 
715     StringRef S(A->getValue());
716     if (S == "s" || S == "z")
717       return 2;
718 
719     if (S == "g")
720       return 1;
721 
722     return getLastArgIntValue(Args, OPT_O, DefaultOpt, Diags);
723   }
724 
725   return DefaultOpt;
726 }
727 
getOptimizationLevelSize(ArgList & Args)728 static unsigned getOptimizationLevelSize(ArgList &Args) {
729   if (Arg *A = Args.getLastArg(options::OPT_O_Group)) {
730     if (A->getOption().matches(options::OPT_O)) {
731       switch (A->getValue()[0]) {
732       default:
733         return 0;
734       case 's':
735         return 1;
736       case 'z':
737         return 2;
738       }
739     }
740   }
741   return 0;
742 }
743 
GenerateArg(ArgumentConsumer Consumer,llvm::opt::OptSpecifier OptSpecifier)744 static void GenerateArg(ArgumentConsumer Consumer,
745                         llvm::opt::OptSpecifier OptSpecifier) {
746   Option Opt = getDriverOptTable().getOption(OptSpecifier);
747   denormalizeSimpleFlag(Consumer, Opt.getPrefixedName(),
748                         Option::OptionClass::FlagClass, 0);
749 }
750 
GenerateArg(ArgumentConsumer Consumer,llvm::opt::OptSpecifier OptSpecifier,const Twine & Value)751 static void GenerateArg(ArgumentConsumer Consumer,
752                         llvm::opt::OptSpecifier OptSpecifier,
753                         const Twine &Value) {
754   Option Opt = getDriverOptTable().getOption(OptSpecifier);
755   denormalizeString(Consumer, Opt.getPrefixedName(), Opt.getKind(), 0, Value);
756 }
757 
758 // Parse command line arguments into CompilerInvocation.
759 using ParseFn =
760     llvm::function_ref<bool(CompilerInvocation &, ArrayRef<const char *>,
761                             DiagnosticsEngine &, const char *)>;
762 
763 // Generate command line arguments from CompilerInvocation.
764 using GenerateFn = llvm::function_ref<void(
765     CompilerInvocation &, SmallVectorImpl<const char *> &,
766     CompilerInvocation::StringAllocator)>;
767 
768 /// May perform round-trip of command line arguments. By default, the round-trip
769 /// is enabled in assert builds. This can be overwritten at run-time via the
770 /// "-round-trip-args" and "-no-round-trip-args" command line flags, or via the
771 /// ForceRoundTrip parameter.
772 ///
773 /// During round-trip, the command line arguments are parsed into a dummy
774 /// CompilerInvocation, which is used to generate the command line arguments
775 /// again. The real CompilerInvocation is then created by parsing the generated
776 /// arguments, not the original ones. This (in combination with tests covering
777 /// argument behavior) ensures the generated command line is complete (doesn't
778 /// drop/mangle any arguments).
779 ///
780 /// Finally, we check the command line that was used to create the real
781 /// CompilerInvocation instance. By default, we compare it to the command line
782 /// the real CompilerInvocation generates. This checks whether the generator is
783 /// deterministic. If \p CheckAgainstOriginalInvocation is enabled, we instead
784 /// compare it to the original command line to verify the original command-line
785 /// was canonical and can round-trip exactly.
RoundTrip(ParseFn Parse,GenerateFn Generate,CompilerInvocation & RealInvocation,CompilerInvocation & DummyInvocation,ArrayRef<const char * > CommandLineArgs,DiagnosticsEngine & Diags,const char * Argv0,bool CheckAgainstOriginalInvocation=false,bool ForceRoundTrip=false)786 static bool RoundTrip(ParseFn Parse, GenerateFn Generate,
787                       CompilerInvocation &RealInvocation,
788                       CompilerInvocation &DummyInvocation,
789                       ArrayRef<const char *> CommandLineArgs,
790                       DiagnosticsEngine &Diags, const char *Argv0,
791                       bool CheckAgainstOriginalInvocation = false,
792                       bool ForceRoundTrip = false) {
793 #ifndef NDEBUG
794   bool DoRoundTripDefault = true;
795 #else
796   bool DoRoundTripDefault = false;
797 #endif
798 
799   bool DoRoundTrip = DoRoundTripDefault;
800   if (ForceRoundTrip) {
801     DoRoundTrip = true;
802   } else {
803     for (const auto *Arg : CommandLineArgs) {
804       if (Arg == StringRef("-round-trip-args"))
805         DoRoundTrip = true;
806       if (Arg == StringRef("-no-round-trip-args"))
807         DoRoundTrip = false;
808     }
809   }
810 
811   // If round-trip was not requested, simply run the parser with the real
812   // invocation diagnostics.
813   if (!DoRoundTrip)
814     return Parse(RealInvocation, CommandLineArgs, Diags, Argv0);
815 
816   // Serializes quoted (and potentially escaped) arguments.
817   auto SerializeArgs = [](ArrayRef<const char *> Args) {
818     std::string Buffer;
819     llvm::raw_string_ostream OS(Buffer);
820     for (const char *Arg : Args) {
821       llvm::sys::printArg(OS, Arg, /*Quote=*/true);
822       OS << ' ';
823     }
824     return Buffer;
825   };
826 
827   // Setup a dummy DiagnosticsEngine.
828   DiagnosticOptions DummyDiagOpts;
829   DiagnosticsEngine DummyDiags(new DiagnosticIDs(), DummyDiagOpts);
830   DummyDiags.setClient(new TextDiagnosticBuffer());
831 
832   // Run the first parse on the original arguments with the dummy invocation and
833   // diagnostics.
834   if (!Parse(DummyInvocation, CommandLineArgs, DummyDiags, Argv0) ||
835       DummyDiags.getNumWarnings() != 0) {
836     // If the first parse did not succeed, it must be user mistake (invalid
837     // command line arguments). We won't be able to generate arguments that
838     // would reproduce the same result. Let's fail again with the real
839     // invocation and diagnostics, so all side-effects of parsing are visible.
840     unsigned NumWarningsBefore = Diags.getNumWarnings();
841     auto Success = Parse(RealInvocation, CommandLineArgs, Diags, Argv0);
842     if (!Success || Diags.getNumWarnings() != NumWarningsBefore)
843       return Success;
844 
845     // Parse with original options and diagnostics succeeded even though it
846     // shouldn't have. Something is off.
847     Diags.Report(diag::err_cc1_round_trip_fail_then_ok);
848     Diags.Report(diag::note_cc1_round_trip_original)
849         << SerializeArgs(CommandLineArgs);
850     return false;
851   }
852 
853   // Setup string allocator.
854   llvm::BumpPtrAllocator Alloc;
855   llvm::StringSaver StringPool(Alloc);
856   auto SA = [&StringPool](const Twine &Arg) {
857     return StringPool.save(Arg).data();
858   };
859 
860   // Generate arguments from the dummy invocation. If Generate is the
861   // inverse of Parse, the newly generated arguments must have the same
862   // semantics as the original.
863   SmallVector<const char *> GeneratedArgs;
864   Generate(DummyInvocation, GeneratedArgs, SA);
865 
866   // Run the second parse, now on the generated arguments, and with the real
867   // invocation and diagnostics. The result is what we will end up using for the
868   // rest of compilation, so if Generate is not inverse of Parse, something down
869   // the line will break.
870   bool Success2 = Parse(RealInvocation, GeneratedArgs, Diags, Argv0);
871 
872   // The first parse on original arguments succeeded, but second parse of
873   // generated arguments failed. Something must be wrong with the generator.
874   if (!Success2) {
875     Diags.Report(diag::err_cc1_round_trip_ok_then_fail);
876     Diags.Report(diag::note_cc1_round_trip_generated)
877         << 1 << SerializeArgs(GeneratedArgs);
878     return false;
879   }
880 
881   SmallVector<const char *> ComparisonArgs;
882   if (CheckAgainstOriginalInvocation)
883     // Compare against original arguments.
884     ComparisonArgs.assign(CommandLineArgs.begin(), CommandLineArgs.end());
885   else
886     // Generate arguments again, this time from the options we will end up using
887     // for the rest of the compilation.
888     Generate(RealInvocation, ComparisonArgs, SA);
889 
890   // Compares two lists of arguments.
891   auto Equal = [](const ArrayRef<const char *> A,
892                   const ArrayRef<const char *> B) {
893     return std::equal(A.begin(), A.end(), B.begin(), B.end(),
894                       [](const char *AElem, const char *BElem) {
895                         return StringRef(AElem) == StringRef(BElem);
896                       });
897   };
898 
899   // If we generated different arguments from what we assume are two
900   // semantically equivalent CompilerInvocations, the Generate function may
901   // be non-deterministic.
902   if (!Equal(GeneratedArgs, ComparisonArgs)) {
903     Diags.Report(diag::err_cc1_round_trip_mismatch);
904     Diags.Report(diag::note_cc1_round_trip_generated)
905         << 1 << SerializeArgs(GeneratedArgs);
906     Diags.Report(diag::note_cc1_round_trip_generated)
907         << 2 << SerializeArgs(ComparisonArgs);
908     return false;
909   }
910 
911   Diags.Report(diag::remark_cc1_round_trip_generated)
912       << 1 << SerializeArgs(GeneratedArgs);
913   Diags.Report(diag::remark_cc1_round_trip_generated)
914       << 2 << SerializeArgs(ComparisonArgs);
915 
916   return Success2;
917 }
918 
checkCC1RoundTrip(ArrayRef<const char * > Args,DiagnosticsEngine & Diags,const char * Argv0)919 bool CompilerInvocation::checkCC1RoundTrip(ArrayRef<const char *> Args,
920                                            DiagnosticsEngine &Diags,
921                                            const char *Argv0) {
922   CompilerInvocation DummyInvocation1, DummyInvocation2;
923   return RoundTrip(
924       [](CompilerInvocation &Invocation, ArrayRef<const char *> CommandLineArgs,
925          DiagnosticsEngine &Diags, const char *Argv0) {
926         return CreateFromArgsImpl(Invocation, CommandLineArgs, Diags, Argv0);
927       },
928       [](CompilerInvocation &Invocation, SmallVectorImpl<const char *> &Args,
929          StringAllocator SA) {
930         Args.push_back("-cc1");
931         Invocation.generateCC1CommandLine(Args, SA);
932       },
933       DummyInvocation1, DummyInvocation2, Args, Diags, Argv0,
934       /*CheckAgainstOriginalInvocation=*/true, /*ForceRoundTrip=*/true);
935 }
936 
addDiagnosticArgs(ArgList & Args,OptSpecifier Group,OptSpecifier GroupWithValue,std::vector<std::string> & Diagnostics)937 static void addDiagnosticArgs(ArgList &Args, OptSpecifier Group,
938                               OptSpecifier GroupWithValue,
939                               std::vector<std::string> &Diagnostics) {
940   for (auto *A : Args.filtered(Group)) {
941     if (A->getOption().getKind() == Option::FlagClass) {
942       // The argument is a pure flag (such as OPT_Wall or OPT_Wdeprecated). Add
943       // its name (minus the "W" or "R" at the beginning) to the diagnostics.
944       Diagnostics.push_back(
945           std::string(A->getOption().getName().drop_front(1)));
946     } else if (A->getOption().matches(GroupWithValue)) {
947       // This is -Wfoo= or -Rfoo=, where foo is the name of the diagnostic
948       // group. Add only the group name to the diagnostics.
949       Diagnostics.push_back(
950           std::string(A->getOption().getName().drop_front(1).rtrim("=-")));
951     } else {
952       // Otherwise, add its value (for OPT_W_Joined and similar).
953       Diagnostics.push_back(A->getValue());
954     }
955   }
956 }
957 
958 // Parse the Static Analyzer configuration. If \p Diags is set to nullptr,
959 // it won't verify the input.
960 static void parseAnalyzerConfigs(AnalyzerOptions &AnOpts,
961                                  DiagnosticsEngine *Diags);
962 
getAllNoBuiltinFuncValues(ArgList & Args,std::vector<std::string> & Funcs)963 static void getAllNoBuiltinFuncValues(ArgList &Args,
964                                       std::vector<std::string> &Funcs) {
965   std::vector<std::string> Values = Args.getAllArgValues(OPT_fno_builtin_);
966   auto BuiltinEnd = llvm::partition(Values, Builtin::Context::isBuiltinFunc);
967   Funcs.insert(Funcs.end(), Values.begin(), BuiltinEnd);
968 }
969 
GenerateAnalyzerArgs(const AnalyzerOptions & Opts,ArgumentConsumer Consumer)970 static void GenerateAnalyzerArgs(const AnalyzerOptions &Opts,
971                                  ArgumentConsumer Consumer) {
972   const AnalyzerOptions *AnalyzerOpts = &Opts;
973 
974 #define ANALYZER_OPTION_WITH_MARSHALLING(...)                                  \
975   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
976 #include "clang/Driver/Options.inc"
977 #undef ANALYZER_OPTION_WITH_MARSHALLING
978 
979   if (Opts.AnalysisConstraintsOpt != RangeConstraintsModel) {
980     switch (Opts.AnalysisConstraintsOpt) {
981 #define ANALYSIS_CONSTRAINTS(NAME, CMDFLAG, DESC, CREATFN)                     \
982   case NAME##Model:                                                            \
983     GenerateArg(Consumer, OPT_analyzer_constraints, CMDFLAG);                  \
984     break;
985 #include "clang/StaticAnalyzer/Core/Analyses.def"
986     default:
987       llvm_unreachable("Tried to generate unknown analysis constraint.");
988     }
989   }
990 
991   if (Opts.AnalysisDiagOpt != PD_HTML) {
992     switch (Opts.AnalysisDiagOpt) {
993 #define ANALYSIS_DIAGNOSTICS(NAME, CMDFLAG, DESC, CREATFN)                     \
994   case PD_##NAME:                                                              \
995     GenerateArg(Consumer, OPT_analyzer_output, CMDFLAG);                       \
996     break;
997 #include "clang/StaticAnalyzer/Core/Analyses.def"
998     default:
999       llvm_unreachable("Tried to generate unknown analysis diagnostic client.");
1000     }
1001   }
1002 
1003   if (Opts.AnalysisPurgeOpt != PurgeStmt) {
1004     switch (Opts.AnalysisPurgeOpt) {
1005 #define ANALYSIS_PURGE(NAME, CMDFLAG, DESC)                                    \
1006   case NAME:                                                                   \
1007     GenerateArg(Consumer, OPT_analyzer_purge, CMDFLAG);                        \
1008     break;
1009 #include "clang/StaticAnalyzer/Core/Analyses.def"
1010     default:
1011       llvm_unreachable("Tried to generate unknown analysis purge mode.");
1012     }
1013   }
1014 
1015   if (Opts.InliningMode != NoRedundancy) {
1016     switch (Opts.InliningMode) {
1017 #define ANALYSIS_INLINING_MODE(NAME, CMDFLAG, DESC)                            \
1018   case NAME:                                                                   \
1019     GenerateArg(Consumer, OPT_analyzer_inlining_mode, CMDFLAG);                \
1020     break;
1021 #include "clang/StaticAnalyzer/Core/Analyses.def"
1022     default:
1023       llvm_unreachable("Tried to generate unknown analysis inlining mode.");
1024     }
1025   }
1026 
1027   for (const auto &CP : Opts.CheckersAndPackages) {
1028     OptSpecifier Opt =
1029         CP.second ? OPT_analyzer_checker : OPT_analyzer_disable_checker;
1030     GenerateArg(Consumer, Opt, CP.first);
1031   }
1032 
1033   AnalyzerOptions ConfigOpts;
1034   parseAnalyzerConfigs(ConfigOpts, nullptr);
1035 
1036   // Sort options by key to avoid relying on StringMap iteration order.
1037   SmallVector<std::pair<StringRef, StringRef>, 4> SortedConfigOpts;
1038   for (const auto &C : Opts.Config)
1039     SortedConfigOpts.emplace_back(C.getKey(), C.getValue());
1040   llvm::sort(SortedConfigOpts, llvm::less_first());
1041 
1042   for (const auto &[Key, Value] : SortedConfigOpts) {
1043     // Don't generate anything that came from parseAnalyzerConfigs. It would be
1044     // redundant and may not be valid on the command line.
1045     auto Entry = ConfigOpts.Config.find(Key);
1046     if (Entry != ConfigOpts.Config.end() && Entry->getValue() == Value)
1047       continue;
1048 
1049     GenerateArg(Consumer, OPT_analyzer_config, Key + "=" + Value);
1050   }
1051 
1052   // Nothing to generate for FullCompilerInvocation.
1053 }
1054 
ParseAnalyzerArgs(AnalyzerOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags)1055 static bool ParseAnalyzerArgs(AnalyzerOptions &Opts, ArgList &Args,
1056                               DiagnosticsEngine &Diags) {
1057   unsigned NumErrorsBefore = Diags.getNumErrors();
1058 
1059   AnalyzerOptions *AnalyzerOpts = &Opts;
1060 
1061 #define ANALYZER_OPTION_WITH_MARSHALLING(...)                                  \
1062   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
1063 #include "clang/Driver/Options.inc"
1064 #undef ANALYZER_OPTION_WITH_MARSHALLING
1065 
1066   if (Arg *A = Args.getLastArg(OPT_analyzer_constraints)) {
1067     StringRef Name = A->getValue();
1068     AnalysisConstraints Value = llvm::StringSwitch<AnalysisConstraints>(Name)
1069 #define ANALYSIS_CONSTRAINTS(NAME, CMDFLAG, DESC, CREATFN) \
1070       .Case(CMDFLAG, NAME##Model)
1071 #include "clang/StaticAnalyzer/Core/Analyses.def"
1072       .Default(NumConstraints);
1073     if (Value == NumConstraints) {
1074       Diags.Report(diag::err_drv_invalid_value)
1075         << A->getAsString(Args) << Name;
1076     } else {
1077 #ifndef LLVM_WITH_Z3
1078       if (Value == AnalysisConstraints::Z3ConstraintsModel) {
1079         Diags.Report(diag::err_analyzer_not_built_with_z3);
1080       }
1081 #endif // LLVM_WITH_Z3
1082       Opts.AnalysisConstraintsOpt = Value;
1083     }
1084   }
1085 
1086   if (Arg *A = Args.getLastArg(OPT_analyzer_output)) {
1087     StringRef Name = A->getValue();
1088     AnalysisDiagClients Value = llvm::StringSwitch<AnalysisDiagClients>(Name)
1089 #define ANALYSIS_DIAGNOSTICS(NAME, CMDFLAG, DESC, CREATFN) \
1090       .Case(CMDFLAG, PD_##NAME)
1091 #include "clang/StaticAnalyzer/Core/Analyses.def"
1092       .Default(NUM_ANALYSIS_DIAG_CLIENTS);
1093     if (Value == NUM_ANALYSIS_DIAG_CLIENTS) {
1094       Diags.Report(diag::err_drv_invalid_value)
1095         << A->getAsString(Args) << Name;
1096     } else {
1097       Opts.AnalysisDiagOpt = Value;
1098     }
1099   }
1100 
1101   if (Arg *A = Args.getLastArg(OPT_analyzer_purge)) {
1102     StringRef Name = A->getValue();
1103     AnalysisPurgeMode Value = llvm::StringSwitch<AnalysisPurgeMode>(Name)
1104 #define ANALYSIS_PURGE(NAME, CMDFLAG, DESC) \
1105       .Case(CMDFLAG, NAME)
1106 #include "clang/StaticAnalyzer/Core/Analyses.def"
1107       .Default(NumPurgeModes);
1108     if (Value == NumPurgeModes) {
1109       Diags.Report(diag::err_drv_invalid_value)
1110         << A->getAsString(Args) << Name;
1111     } else {
1112       Opts.AnalysisPurgeOpt = Value;
1113     }
1114   }
1115 
1116   if (Arg *A = Args.getLastArg(OPT_analyzer_inlining_mode)) {
1117     StringRef Name = A->getValue();
1118     AnalysisInliningMode Value = llvm::StringSwitch<AnalysisInliningMode>(Name)
1119 #define ANALYSIS_INLINING_MODE(NAME, CMDFLAG, DESC) \
1120       .Case(CMDFLAG, NAME)
1121 #include "clang/StaticAnalyzer/Core/Analyses.def"
1122       .Default(NumInliningModes);
1123     if (Value == NumInliningModes) {
1124       Diags.Report(diag::err_drv_invalid_value)
1125         << A->getAsString(Args) << Name;
1126     } else {
1127       Opts.InliningMode = Value;
1128     }
1129   }
1130 
1131   Opts.CheckersAndPackages.clear();
1132   for (const Arg *A :
1133        Args.filtered(OPT_analyzer_checker, OPT_analyzer_disable_checker)) {
1134     A->claim();
1135     bool IsEnabled = A->getOption().getID() == OPT_analyzer_checker;
1136     // We can have a list of comma separated checker names, e.g:
1137     // '-analyzer-checker=cocoa,unix'
1138     StringRef CheckerAndPackageList = A->getValue();
1139     SmallVector<StringRef, 16> CheckersAndPackages;
1140     CheckerAndPackageList.split(CheckersAndPackages, ",");
1141     for (const StringRef &CheckerOrPackage : CheckersAndPackages)
1142       Opts.CheckersAndPackages.emplace_back(std::string(CheckerOrPackage),
1143                                             IsEnabled);
1144   }
1145 
1146   // Go through the analyzer configuration options.
1147   for (const auto *A : Args.filtered(OPT_analyzer_config)) {
1148 
1149     // We can have a list of comma separated config names, e.g:
1150     // '-analyzer-config key1=val1,key2=val2'
1151     StringRef configList = A->getValue();
1152     SmallVector<StringRef, 4> configVals;
1153     configList.split(configVals, ",");
1154     for (const auto &configVal : configVals) {
1155       StringRef key, val;
1156       std::tie(key, val) = configVal.split("=");
1157       if (val.empty()) {
1158         Diags.Report(SourceLocation(),
1159                      diag::err_analyzer_config_no_value) << configVal;
1160         break;
1161       }
1162       if (val.contains('=')) {
1163         Diags.Report(SourceLocation(),
1164                      diag::err_analyzer_config_multiple_values)
1165           << configVal;
1166         break;
1167       }
1168 
1169       // TODO: Check checker options too, possibly in CheckerRegistry.
1170       // Leave unknown non-checker configs unclaimed.
1171       if (!key.contains(":") && Opts.isUnknownAnalyzerConfig(key)) {
1172         if (Opts.ShouldEmitErrorsOnInvalidConfigValue)
1173           Diags.Report(diag::err_analyzer_config_unknown) << key;
1174         continue;
1175       }
1176 
1177       A->claim();
1178       Opts.Config[key] = std::string(val);
1179     }
1180   }
1181 
1182   if (Opts.ShouldEmitErrorsOnInvalidConfigValue)
1183     parseAnalyzerConfigs(Opts, &Diags);
1184   else
1185     parseAnalyzerConfigs(Opts, nullptr);
1186 
1187   llvm::raw_string_ostream os(Opts.FullCompilerInvocation);
1188   for (unsigned i = 0; i < Args.getNumInputArgStrings(); ++i) {
1189     if (i != 0)
1190       os << " ";
1191     os << Args.getArgString(i);
1192   }
1193 
1194   return Diags.getNumErrors() == NumErrorsBefore;
1195 }
1196 
getStringOption(AnalyzerOptions::ConfigTable & Config,StringRef OptionName,StringRef DefaultVal)1197 static StringRef getStringOption(AnalyzerOptions::ConfigTable &Config,
1198                                  StringRef OptionName, StringRef DefaultVal) {
1199   return Config.insert({OptionName, std::string(DefaultVal)}).first->second;
1200 }
1201 
initOption(AnalyzerOptions::ConfigTable & Config,DiagnosticsEngine * Diags,StringRef & OptionField,StringRef Name,StringRef DefaultVal)1202 static void initOption(AnalyzerOptions::ConfigTable &Config,
1203                        DiagnosticsEngine *Diags,
1204                        StringRef &OptionField, StringRef Name,
1205                        StringRef DefaultVal) {
1206   // String options may be known to invalid (e.g. if the expected string is a
1207   // file name, but the file does not exist), those will have to be checked in
1208   // parseConfigs.
1209   OptionField = getStringOption(Config, Name, DefaultVal);
1210 }
1211 
initOption(AnalyzerOptions::ConfigTable & Config,DiagnosticsEngine * Diags,bool & OptionField,StringRef Name,bool DefaultVal)1212 static void initOption(AnalyzerOptions::ConfigTable &Config,
1213                        DiagnosticsEngine *Diags,
1214                        bool &OptionField, StringRef Name, bool DefaultVal) {
1215   auto PossiblyInvalidVal =
1216       llvm::StringSwitch<std::optional<bool>>(
1217           getStringOption(Config, Name, (DefaultVal ? "true" : "false")))
1218           .Case("true", true)
1219           .Case("false", false)
1220           .Default(std::nullopt);
1221 
1222   if (!PossiblyInvalidVal) {
1223     if (Diags)
1224       Diags->Report(diag::err_analyzer_config_invalid_input)
1225         << Name << "a boolean";
1226     else
1227       OptionField = DefaultVal;
1228   } else
1229     OptionField = *PossiblyInvalidVal;
1230 }
1231 
initOption(AnalyzerOptions::ConfigTable & Config,DiagnosticsEngine * Diags,unsigned & OptionField,StringRef Name,unsigned DefaultVal)1232 static void initOption(AnalyzerOptions::ConfigTable &Config,
1233                        DiagnosticsEngine *Diags,
1234                        unsigned &OptionField, StringRef Name,
1235                        unsigned DefaultVal) {
1236 
1237   OptionField = DefaultVal;
1238   bool HasFailed = getStringOption(Config, Name, std::to_string(DefaultVal))
1239                      .getAsInteger(0, OptionField);
1240   if (Diags && HasFailed)
1241     Diags->Report(diag::err_analyzer_config_invalid_input)
1242       << Name << "an unsigned";
1243 }
1244 
initOption(AnalyzerOptions::ConfigTable & Config,DiagnosticsEngine * Diags,PositiveAnalyzerOption & OptionField,StringRef Name,unsigned DefaultVal)1245 static void initOption(AnalyzerOptions::ConfigTable &Config,
1246                        DiagnosticsEngine *Diags,
1247                        PositiveAnalyzerOption &OptionField, StringRef Name,
1248                        unsigned DefaultVal) {
1249   auto Parsed = PositiveAnalyzerOption::create(
1250       getStringOption(Config, Name, std::to_string(DefaultVal)));
1251   if (Parsed.has_value()) {
1252     OptionField = Parsed.value();
1253     return;
1254   }
1255   if (Diags && !Parsed.has_value())
1256     Diags->Report(diag::err_analyzer_config_invalid_input)
1257         << Name << "a positive";
1258 
1259   OptionField = DefaultVal;
1260 }
1261 
parseAnalyzerConfigs(AnalyzerOptions & AnOpts,DiagnosticsEngine * Diags)1262 static void parseAnalyzerConfigs(AnalyzerOptions &AnOpts,
1263                                  DiagnosticsEngine *Diags) {
1264   // TODO: There's no need to store the entire configtable, it'd be plenty
1265   // enough to store checker options.
1266 
1267 #define ANALYZER_OPTION(TYPE, NAME, CMDFLAG, DESC, DEFAULT_VAL)                \
1268   initOption(AnOpts.Config, Diags, AnOpts.NAME, CMDFLAG, DEFAULT_VAL);
1269 #define ANALYZER_OPTION_DEPENDS_ON_USER_MODE(...)
1270 #include "clang/StaticAnalyzer/Core/AnalyzerOptions.def"
1271 
1272   assert(AnOpts.UserMode == "shallow" || AnOpts.UserMode == "deep");
1273   const bool InShallowMode = AnOpts.UserMode == "shallow";
1274 
1275 #define ANALYZER_OPTION(...)
1276 #define ANALYZER_OPTION_DEPENDS_ON_USER_MODE(TYPE, NAME, CMDFLAG, DESC,        \
1277                                              SHALLOW_VAL, DEEP_VAL)            \
1278   initOption(AnOpts.Config, Diags, AnOpts.NAME, CMDFLAG,                       \
1279              InShallowMode ? SHALLOW_VAL : DEEP_VAL);
1280 #include "clang/StaticAnalyzer/Core/AnalyzerOptions.def"
1281 
1282   // At this point, AnalyzerOptions is configured. Let's validate some options.
1283 
1284   // FIXME: Here we try to validate the silenced checkers or packages are valid.
1285   // The current approach only validates the registered checkers which does not
1286   // contain the runtime enabled checkers and optimally we would validate both.
1287   if (!AnOpts.RawSilencedCheckersAndPackages.empty()) {
1288     std::vector<StringRef> Checkers =
1289         AnOpts.getRegisteredCheckers(/*IncludeExperimental=*/true);
1290     std::vector<StringRef> Packages =
1291         AnOpts.getRegisteredPackages(/*IncludeExperimental=*/true);
1292 
1293     SmallVector<StringRef, 16> CheckersAndPackages;
1294     AnOpts.RawSilencedCheckersAndPackages.split(CheckersAndPackages, ";");
1295 
1296     for (const StringRef &CheckerOrPackage : CheckersAndPackages) {
1297       if (Diags) {
1298         bool IsChecker = CheckerOrPackage.contains('.');
1299         bool IsValidName = IsChecker
1300                                ? llvm::is_contained(Checkers, CheckerOrPackage)
1301                                : llvm::is_contained(Packages, CheckerOrPackage);
1302 
1303         if (!IsValidName)
1304           Diags->Report(diag::err_unknown_analyzer_checker_or_package)
1305               << CheckerOrPackage;
1306       }
1307 
1308       AnOpts.SilencedCheckersAndPackages.emplace_back(CheckerOrPackage);
1309     }
1310   }
1311 
1312   if (!Diags)
1313     return;
1314 
1315   if (AnOpts.ShouldTrackConditionsDebug && !AnOpts.ShouldTrackConditions)
1316     Diags->Report(diag::err_analyzer_config_invalid_input)
1317         << "track-conditions-debug" << "'track-conditions' to also be enabled";
1318 
1319   if (!AnOpts.CTUDir.empty() && !llvm::sys::fs::is_directory(AnOpts.CTUDir))
1320     Diags->Report(diag::err_analyzer_config_invalid_input) << "ctu-dir"
1321                                                            << "a filename";
1322 
1323   if (!AnOpts.ModelPath.empty() &&
1324       !llvm::sys::fs::is_directory(AnOpts.ModelPath))
1325     Diags->Report(diag::err_analyzer_config_invalid_input) << "model-path"
1326                                                            << "a filename";
1327 }
1328 
1329 /// Generate a remark argument. This is an inverse of `ParseOptimizationRemark`.
1330 static void
GenerateOptimizationRemark(ArgumentConsumer Consumer,OptSpecifier OptEQ,StringRef Name,const CodeGenOptions::OptRemark & Remark)1331 GenerateOptimizationRemark(ArgumentConsumer Consumer, OptSpecifier OptEQ,
1332                            StringRef Name,
1333                            const CodeGenOptions::OptRemark &Remark) {
1334   if (Remark.hasValidPattern()) {
1335     GenerateArg(Consumer, OptEQ, Remark.Pattern);
1336   } else if (Remark.Kind == CodeGenOptions::RK_Enabled) {
1337     GenerateArg(Consumer, OPT_R_Joined, Name);
1338   } else if (Remark.Kind == CodeGenOptions::RK_Disabled) {
1339     GenerateArg(Consumer, OPT_R_Joined, StringRef("no-") + Name);
1340   }
1341 }
1342 
1343 /// Parse a remark command line argument. It may be missing, disabled/enabled by
1344 /// '-R[no-]group' or specified with a regular expression by '-Rgroup=regexp'.
1345 /// On top of that, it can be disabled/enabled globally by '-R[no-]everything'.
1346 static CodeGenOptions::OptRemark
ParseOptimizationRemark(DiagnosticsEngine & Diags,ArgList & Args,OptSpecifier OptEQ,StringRef Name)1347 ParseOptimizationRemark(DiagnosticsEngine &Diags, ArgList &Args,
1348                         OptSpecifier OptEQ, StringRef Name) {
1349   CodeGenOptions::OptRemark Result;
1350 
1351   auto InitializeResultPattern = [&Diags, &Args, &Result](const Arg *A,
1352                                                           StringRef Pattern) {
1353     Result.Pattern = Pattern.str();
1354 
1355     std::string RegexError;
1356     Result.Regex = std::make_shared<llvm::Regex>(Result.Pattern);
1357     if (!Result.Regex->isValid(RegexError)) {
1358       Diags.Report(diag::err_drv_optimization_remark_pattern)
1359           << RegexError << A->getAsString(Args);
1360       return false;
1361     }
1362 
1363     return true;
1364   };
1365 
1366   for (Arg *A : Args) {
1367     if (A->getOption().matches(OPT_R_Joined)) {
1368       StringRef Value = A->getValue();
1369 
1370       if (Value == Name)
1371         Result.Kind = CodeGenOptions::RK_Enabled;
1372       else if (Value == "everything")
1373         Result.Kind = CodeGenOptions::RK_EnabledEverything;
1374       else if (Value.split('-') == std::make_pair(StringRef("no"), Name))
1375         Result.Kind = CodeGenOptions::RK_Disabled;
1376       else if (Value == "no-everything")
1377         Result.Kind = CodeGenOptions::RK_DisabledEverything;
1378       else
1379         continue;
1380 
1381       if (Result.Kind == CodeGenOptions::RK_Disabled ||
1382           Result.Kind == CodeGenOptions::RK_DisabledEverything) {
1383         Result.Pattern = "";
1384         Result.Regex = nullptr;
1385       } else {
1386         InitializeResultPattern(A, ".*");
1387       }
1388     } else if (A->getOption().matches(OptEQ)) {
1389       Result.Kind = CodeGenOptions::RK_WithPattern;
1390       if (!InitializeResultPattern(A, A->getValue()))
1391         return CodeGenOptions::OptRemark();
1392     }
1393   }
1394 
1395   return Result;
1396 }
1397 
parseDiagnosticLevelMask(StringRef FlagName,const std::vector<std::string> & Levels,DiagnosticsEngine & Diags,DiagnosticLevelMask & M)1398 static bool parseDiagnosticLevelMask(StringRef FlagName,
1399                                      const std::vector<std::string> &Levels,
1400                                      DiagnosticsEngine &Diags,
1401                                      DiagnosticLevelMask &M) {
1402   bool Success = true;
1403   for (const auto &Level : Levels) {
1404     DiagnosticLevelMask const PM =
1405       llvm::StringSwitch<DiagnosticLevelMask>(Level)
1406         .Case("note",    DiagnosticLevelMask::Note)
1407         .Case("remark",  DiagnosticLevelMask::Remark)
1408         .Case("warning", DiagnosticLevelMask::Warning)
1409         .Case("error",   DiagnosticLevelMask::Error)
1410         .Default(DiagnosticLevelMask::None);
1411     if (PM == DiagnosticLevelMask::None) {
1412       Success = false;
1413       Diags.Report(diag::err_drv_invalid_value) << FlagName << Level;
1414     }
1415     M = M | PM;
1416   }
1417   return Success;
1418 }
1419 
parseSanitizerKinds(StringRef FlagName,const std::vector<std::string> & Sanitizers,DiagnosticsEngine & Diags,SanitizerSet & S)1420 static void parseSanitizerKinds(StringRef FlagName,
1421                                 const std::vector<std::string> &Sanitizers,
1422                                 DiagnosticsEngine &Diags, SanitizerSet &S) {
1423   for (const auto &Sanitizer : Sanitizers) {
1424     SanitizerMask K = parseSanitizerValue(Sanitizer, /*AllowGroups=*/false);
1425     if (K == SanitizerMask())
1426       Diags.Report(diag::err_drv_invalid_value) << FlagName << Sanitizer;
1427     else
1428       S.set(K, true);
1429   }
1430 }
1431 
serializeSanitizerKinds(SanitizerSet S)1432 static SmallVector<StringRef, 4> serializeSanitizerKinds(SanitizerSet S) {
1433   SmallVector<StringRef, 4> Values;
1434   serializeSanitizerSet(S, Values);
1435   return Values;
1436 }
1437 
1438 static SanitizerMaskCutoffs
parseSanitizerWeightedKinds(StringRef FlagName,const std::vector<std::string> & Sanitizers,DiagnosticsEngine & Diags)1439 parseSanitizerWeightedKinds(StringRef FlagName,
1440                             const std::vector<std::string> &Sanitizers,
1441                             DiagnosticsEngine &Diags) {
1442   SanitizerMaskCutoffs Cutoffs;
1443   for (const auto &Sanitizer : Sanitizers) {
1444     if (!parseSanitizerWeightedValue(Sanitizer, /*AllowGroups=*/false, Cutoffs))
1445       Diags.Report(diag::err_drv_invalid_value) << FlagName << Sanitizer;
1446   }
1447   return Cutoffs;
1448 }
1449 
parseXRayInstrumentationBundle(StringRef FlagName,StringRef Bundle,ArgList & Args,DiagnosticsEngine & D,XRayInstrSet & S)1450 static void parseXRayInstrumentationBundle(StringRef FlagName, StringRef Bundle,
1451                                            ArgList &Args, DiagnosticsEngine &D,
1452                                            XRayInstrSet &S) {
1453   llvm::SmallVector<StringRef, 2> BundleParts;
1454   llvm::SplitString(Bundle, BundleParts, ",");
1455   for (const auto &B : BundleParts) {
1456     auto Mask = parseXRayInstrValue(B);
1457     if (Mask == XRayInstrKind::None)
1458       if (B != "none")
1459         D.Report(diag::err_drv_invalid_value) << FlagName << Bundle;
1460       else
1461         S.Mask = Mask;
1462     else if (Mask == XRayInstrKind::All)
1463       S.Mask = Mask;
1464     else
1465       S.set(Mask, true);
1466   }
1467 }
1468 
serializeXRayInstrumentationBundle(const XRayInstrSet & S)1469 static std::string serializeXRayInstrumentationBundle(const XRayInstrSet &S) {
1470   llvm::SmallVector<StringRef, 2> BundleParts;
1471   serializeXRayInstrValue(S, BundleParts);
1472   std::string Buffer;
1473   llvm::raw_string_ostream OS(Buffer);
1474   llvm::interleave(BundleParts, OS, [&OS](StringRef Part) { OS << Part; }, ",");
1475   return Buffer;
1476 }
1477 
1478 // Set the profile kind using fprofile-instrument-use-path.
setPGOUseInstrumentor(CodeGenOptions & Opts,const Twine & ProfileName,llvm::vfs::FileSystem & FS,DiagnosticsEngine & Diags)1479 static void setPGOUseInstrumentor(CodeGenOptions &Opts,
1480                                   const Twine &ProfileName,
1481                                   llvm::vfs::FileSystem &FS,
1482                                   DiagnosticsEngine &Diags) {
1483   auto ReaderOrErr = llvm::IndexedInstrProfReader::create(ProfileName, FS);
1484   if (auto E = ReaderOrErr.takeError()) {
1485     unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
1486                                             "Error in reading profile %0: %1");
1487     llvm::handleAllErrors(std::move(E), [&](const llvm::ErrorInfoBase &EI) {
1488       Diags.Report(DiagID) << ProfileName.str() << EI.message();
1489     });
1490     return;
1491   }
1492   std::unique_ptr<llvm::IndexedInstrProfReader> PGOReader =
1493     std::move(ReaderOrErr.get());
1494   // Currently memprof profiles are only added at the IR level. Mark the profile
1495   // type as IR in that case as well and the subsequent matching needs to detect
1496   // which is available (might be one or both).
1497   if (PGOReader->isIRLevelProfile() || PGOReader->hasMemoryProfile()) {
1498     if (PGOReader->hasCSIRLevelProfile())
1499       Opts.setProfileUse(llvm::driver::ProfileInstrKind::ProfileCSIRInstr);
1500     else
1501       Opts.setProfileUse(llvm::driver::ProfileInstrKind::ProfileIRInstr);
1502   } else
1503     Opts.setProfileUse(llvm::driver::ProfileInstrKind::ProfileClangInstr);
1504 }
1505 
setDefaultPointerAuthOptions(PointerAuthOptions & Opts,const LangOptions & LangOpts,const llvm::Triple & Triple)1506 void CompilerInvocation::setDefaultPointerAuthOptions(
1507     PointerAuthOptions &Opts, const LangOptions &LangOpts,
1508     const llvm::Triple &Triple) {
1509   assert(Triple.getArch() == llvm::Triple::aarch64);
1510   if (LangOpts.PointerAuthCalls) {
1511     using Key = PointerAuthSchema::ARM8_3Key;
1512     using Discrimination = PointerAuthSchema::Discrimination;
1513     // If you change anything here, be sure to update <ptrauth.h>.
1514     Opts.FunctionPointers = PointerAuthSchema(
1515         Key::ASIA, false,
1516         LangOpts.PointerAuthFunctionTypeDiscrimination ? Discrimination::Type
1517                                                        : Discrimination::None);
1518 
1519     Opts.CXXVTablePointers = PointerAuthSchema(
1520         Key::ASDA, LangOpts.PointerAuthVTPtrAddressDiscrimination,
1521         LangOpts.PointerAuthVTPtrTypeDiscrimination ? Discrimination::Type
1522                                                     : Discrimination::None);
1523 
1524     if (LangOpts.PointerAuthTypeInfoVTPtrDiscrimination)
1525       Opts.CXXTypeInfoVTablePointer =
1526           PointerAuthSchema(Key::ASDA, true, Discrimination::Constant,
1527                             StdTypeInfoVTablePointerConstantDiscrimination);
1528     else
1529       Opts.CXXTypeInfoVTablePointer =
1530           PointerAuthSchema(Key::ASDA, false, Discrimination::None);
1531 
1532     Opts.CXXVTTVTablePointers =
1533         PointerAuthSchema(Key::ASDA, false, Discrimination::None);
1534     Opts.CXXVirtualFunctionPointers = Opts.CXXVirtualVariadicFunctionPointers =
1535         PointerAuthSchema(Key::ASIA, true, Discrimination::Decl);
1536     Opts.CXXMemberFunctionPointers =
1537         PointerAuthSchema(Key::ASIA, false, Discrimination::Type);
1538 
1539     if (LangOpts.PointerAuthInitFini) {
1540       Opts.InitFiniPointers = PointerAuthSchema(
1541           Key::ASIA, LangOpts.PointerAuthInitFiniAddressDiscrimination,
1542           Discrimination::Constant, InitFiniPointerConstantDiscriminator);
1543     }
1544 
1545     Opts.BlockInvocationFunctionPointers =
1546         PointerAuthSchema(Key::ASIA, true, Discrimination::None);
1547     Opts.BlockHelperFunctionPointers =
1548         PointerAuthSchema(Key::ASIA, true, Discrimination::None);
1549     Opts.BlockByrefHelperFunctionPointers =
1550         PointerAuthSchema(Key::ASIA, true, Discrimination::None);
1551     if (LangOpts.PointerAuthBlockDescriptorPointers)
1552       Opts.BlockDescriptorPointers =
1553           PointerAuthSchema(Key::ASDA, true, Discrimination::Constant,
1554                             BlockDescriptorConstantDiscriminator);
1555 
1556     Opts.ObjCMethodListFunctionPointers =
1557         PointerAuthSchema(Key::ASIA, true, Discrimination::None);
1558     Opts.ObjCMethodListPointer =
1559         PointerAuthSchema(Key::ASDA, true, Discrimination::Constant,
1560                           MethodListPointerConstantDiscriminator);
1561     if (LangOpts.PointerAuthObjcIsa) {
1562       Opts.ObjCIsaPointers =
1563           PointerAuthSchema(Key::ASDA, true, Discrimination::Constant,
1564                             IsaPointerConstantDiscriminator);
1565       Opts.ObjCSuperPointers =
1566           PointerAuthSchema(Key::ASDA, true, Discrimination::Constant,
1567                             SuperPointerConstantDiscriminator);
1568     }
1569 
1570     if (LangOpts.PointerAuthObjcClassROPointers)
1571       Opts.ObjCClassROPointers =
1572           PointerAuthSchema(Key::ASDA, true, Discrimination::Constant,
1573                             ClassROConstantDiscriminator);
1574   }
1575   Opts.ReturnAddresses = LangOpts.PointerAuthReturns;
1576   Opts.AuthTraps = LangOpts.PointerAuthAuthTraps;
1577   Opts.IndirectGotos = LangOpts.PointerAuthIndirectGotos;
1578   Opts.AArch64JumpTableHardening = LangOpts.AArch64JumpTableHardening;
1579 }
1580 
parsePointerAuthOptions(PointerAuthOptions & Opts,const LangOptions & LangOpts,const llvm::Triple & Triple,DiagnosticsEngine & Diags)1581 static void parsePointerAuthOptions(PointerAuthOptions &Opts,
1582                                     const LangOptions &LangOpts,
1583                                     const llvm::Triple &Triple,
1584                                     DiagnosticsEngine &Diags) {
1585   if (!LangOpts.PointerAuthCalls && !LangOpts.PointerAuthReturns &&
1586       !LangOpts.PointerAuthAuthTraps && !LangOpts.PointerAuthIndirectGotos &&
1587       !LangOpts.AArch64JumpTableHardening)
1588     return;
1589 
1590   CompilerInvocation::setDefaultPointerAuthOptions(Opts, LangOpts, Triple);
1591 }
1592 
GenerateCodeGenArgs(const CodeGenOptions & Opts,ArgumentConsumer Consumer,const llvm::Triple & T,const std::string & OutputFile,const LangOptions * LangOpts)1593 void CompilerInvocationBase::GenerateCodeGenArgs(const CodeGenOptions &Opts,
1594                                                  ArgumentConsumer Consumer,
1595                                                  const llvm::Triple &T,
1596                                                  const std::string &OutputFile,
1597                                                  const LangOptions *LangOpts) {
1598   const CodeGenOptions &CodeGenOpts = Opts;
1599 
1600   if (Opts.OptimizationLevel == 0)
1601     GenerateArg(Consumer, OPT_O0);
1602   else
1603     GenerateArg(Consumer, OPT_O, Twine(Opts.OptimizationLevel));
1604 
1605 #define CODEGEN_OPTION_WITH_MARSHALLING(...)                                   \
1606   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
1607 #include "clang/Driver/Options.inc"
1608 #undef CODEGEN_OPTION_WITH_MARSHALLING
1609 
1610   if (Opts.OptimizationLevel > 0) {
1611     if (Opts.Inlining == CodeGenOptions::NormalInlining)
1612       GenerateArg(Consumer, OPT_finline_functions);
1613     else if (Opts.Inlining == CodeGenOptions::OnlyHintInlining)
1614       GenerateArg(Consumer, OPT_finline_hint_functions);
1615     else if (Opts.Inlining == CodeGenOptions::OnlyAlwaysInlining)
1616       GenerateArg(Consumer, OPT_fno_inline);
1617   }
1618 
1619   if (Opts.DirectAccessExternalData && LangOpts->PICLevel != 0)
1620     GenerateArg(Consumer, OPT_fdirect_access_external_data);
1621   else if (!Opts.DirectAccessExternalData && LangOpts->PICLevel == 0)
1622     GenerateArg(Consumer, OPT_fno_direct_access_external_data);
1623 
1624   std::optional<StringRef> DebugInfoVal;
1625   switch (Opts.DebugInfo) {
1626   case llvm::codegenoptions::DebugLineTablesOnly:
1627     DebugInfoVal = "line-tables-only";
1628     break;
1629   case llvm::codegenoptions::DebugDirectivesOnly:
1630     DebugInfoVal = "line-directives-only";
1631     break;
1632   case llvm::codegenoptions::DebugInfoConstructor:
1633     DebugInfoVal = "constructor";
1634     break;
1635   case llvm::codegenoptions::LimitedDebugInfo:
1636     DebugInfoVal = "limited";
1637     break;
1638   case llvm::codegenoptions::FullDebugInfo:
1639     DebugInfoVal = "standalone";
1640     break;
1641   case llvm::codegenoptions::UnusedTypeInfo:
1642     DebugInfoVal = "unused-types";
1643     break;
1644   case llvm::codegenoptions::NoDebugInfo: // default value
1645     DebugInfoVal = std::nullopt;
1646     break;
1647   case llvm::codegenoptions::LocTrackingOnly: // implied value
1648     DebugInfoVal = std::nullopt;
1649     break;
1650   }
1651   if (DebugInfoVal)
1652     GenerateArg(Consumer, OPT_debug_info_kind_EQ, *DebugInfoVal);
1653 
1654   for (const auto &Prefix : Opts.DebugPrefixMap)
1655     GenerateArg(Consumer, OPT_fdebug_prefix_map_EQ,
1656                 Prefix.first + "=" + Prefix.second);
1657 
1658   for (const auto &Prefix : Opts.CoveragePrefixMap)
1659     GenerateArg(Consumer, OPT_fcoverage_prefix_map_EQ,
1660                 Prefix.first + "=" + Prefix.second);
1661 
1662   if (Opts.NewStructPathTBAA)
1663     GenerateArg(Consumer, OPT_new_struct_path_tbaa);
1664 
1665   if (Opts.OptimizeSize == 1)
1666     GenerateArg(Consumer, OPT_O, "s");
1667   else if (Opts.OptimizeSize == 2)
1668     GenerateArg(Consumer, OPT_O, "z");
1669 
1670   // SimplifyLibCalls is set only in the absence of -fno-builtin and
1671   // -ffreestanding. We'll consider that when generating them.
1672 
1673   // NoBuiltinFuncs are generated by LangOptions.
1674 
1675   if (Opts.UnrollLoops && Opts.OptimizationLevel <= 1)
1676     GenerateArg(Consumer, OPT_funroll_loops);
1677   else if (!Opts.UnrollLoops && Opts.OptimizationLevel > 1)
1678     GenerateArg(Consumer, OPT_fno_unroll_loops);
1679 
1680   if (Opts.InterchangeLoops)
1681     GenerateArg(Consumer, OPT_floop_interchange);
1682   else
1683     GenerateArg(Consumer, OPT_fno_loop_interchange);
1684 
1685   if (!Opts.BinutilsVersion.empty())
1686     GenerateArg(Consumer, OPT_fbinutils_version_EQ, Opts.BinutilsVersion);
1687 
1688   if (Opts.DebugNameTable ==
1689       static_cast<unsigned>(llvm::DICompileUnit::DebugNameTableKind::GNU))
1690     GenerateArg(Consumer, OPT_ggnu_pubnames);
1691   else if (Opts.DebugNameTable ==
1692            static_cast<unsigned>(
1693                llvm::DICompileUnit::DebugNameTableKind::Default))
1694     GenerateArg(Consumer, OPT_gpubnames);
1695 
1696   if (Opts.DebugTemplateAlias)
1697     GenerateArg(Consumer, OPT_gtemplate_alias);
1698 
1699   auto TNK = Opts.getDebugSimpleTemplateNames();
1700   if (TNK != llvm::codegenoptions::DebugTemplateNamesKind::Full) {
1701     if (TNK == llvm::codegenoptions::DebugTemplateNamesKind::Simple)
1702       GenerateArg(Consumer, OPT_gsimple_template_names_EQ, "simple");
1703     else if (TNK == llvm::codegenoptions::DebugTemplateNamesKind::Mangled)
1704       GenerateArg(Consumer, OPT_gsimple_template_names_EQ, "mangled");
1705   }
1706   // ProfileInstrumentUsePath is marshalled automatically, no need to generate
1707   // it or PGOUseInstrumentor.
1708 
1709   if (Opts.TimePasses) {
1710     if (Opts.TimePassesPerRun)
1711       GenerateArg(Consumer, OPT_ftime_report_EQ, "per-pass-run");
1712     else
1713       GenerateArg(Consumer, OPT_ftime_report);
1714 
1715     if (Opts.TimePassesJson)
1716       GenerateArg(Consumer, OPT_ftime_report_json);
1717   }
1718 
1719   if (Opts.PrepareForLTO && !Opts.PrepareForThinLTO)
1720     GenerateArg(Consumer, OPT_flto_EQ, "full");
1721 
1722   if (Opts.PrepareForThinLTO)
1723     GenerateArg(Consumer, OPT_flto_EQ, "thin");
1724 
1725   if (!Opts.ThinLTOIndexFile.empty())
1726     GenerateArg(Consumer, OPT_fthinlto_index_EQ, Opts.ThinLTOIndexFile);
1727 
1728   if (Opts.SaveTempsFilePrefix == OutputFile)
1729     GenerateArg(Consumer, OPT_save_temps_EQ, "obj");
1730 
1731   StringRef MemProfileBasename("memprof.profraw");
1732   if (!Opts.MemoryProfileOutput.empty()) {
1733     if (Opts.MemoryProfileOutput == MemProfileBasename) {
1734       GenerateArg(Consumer, OPT_fmemory_profile);
1735     } else {
1736       size_t ArgLength =
1737           Opts.MemoryProfileOutput.size() - MemProfileBasename.size();
1738       GenerateArg(Consumer, OPT_fmemory_profile_EQ,
1739                   Opts.MemoryProfileOutput.substr(0, ArgLength));
1740     }
1741   }
1742 
1743   if (memcmp(Opts.CoverageVersion, "0000", 4))
1744     GenerateArg(Consumer, OPT_coverage_version_EQ,
1745                 StringRef(Opts.CoverageVersion, 4));
1746 
1747   // TODO: Check if we need to generate arguments stored in CmdArgs. (Namely
1748   //  '-fembed_bitcode', which does not map to any CompilerInvocation field and
1749   //  won't be generated.)
1750 
1751   if (Opts.XRayInstrumentationBundle.Mask != XRayInstrKind::All) {
1752     std::string InstrBundle =
1753         serializeXRayInstrumentationBundle(Opts.XRayInstrumentationBundle);
1754     if (!InstrBundle.empty())
1755       GenerateArg(Consumer, OPT_fxray_instrumentation_bundle, InstrBundle);
1756   }
1757 
1758   if (Opts.CFProtectionReturn && Opts.CFProtectionBranch)
1759     GenerateArg(Consumer, OPT_fcf_protection_EQ, "full");
1760   else if (Opts.CFProtectionReturn)
1761     GenerateArg(Consumer, OPT_fcf_protection_EQ, "return");
1762   else if (Opts.CFProtectionBranch)
1763     GenerateArg(Consumer, OPT_fcf_protection_EQ, "branch");
1764 
1765   if (Opts.CFProtectionBranch) {
1766     switch (Opts.getCFBranchLabelScheme()) {
1767     case CFBranchLabelSchemeKind::Default:
1768       break;
1769 #define CF_BRANCH_LABEL_SCHEME(Kind, FlagVal)                                  \
1770   case CFBranchLabelSchemeKind::Kind:                                          \
1771     GenerateArg(Consumer, OPT_mcf_branch_label_scheme_EQ, #FlagVal);           \
1772     break;
1773 #include "clang/Basic/CFProtectionOptions.def"
1774     }
1775   }
1776 
1777   if (Opts.FunctionReturnThunks)
1778     GenerateArg(Consumer, OPT_mfunction_return_EQ, "thunk-extern");
1779 
1780   for (const auto &F : Opts.LinkBitcodeFiles) {
1781     bool Builtint = F.LinkFlags == llvm::Linker::Flags::LinkOnlyNeeded &&
1782                     F.PropagateAttrs && F.Internalize;
1783     GenerateArg(Consumer,
1784                 Builtint ? OPT_mlink_builtin_bitcode : OPT_mlink_bitcode_file,
1785                 F.Filename);
1786   }
1787 
1788   if (Opts.EmulatedTLS)
1789     GenerateArg(Consumer, OPT_femulated_tls);
1790 
1791   if (Opts.FPDenormalMode != llvm::DenormalMode::getIEEE())
1792     GenerateArg(Consumer, OPT_fdenormal_fp_math_EQ, Opts.FPDenormalMode.str());
1793 
1794   if ((Opts.FPDenormalMode != Opts.FP32DenormalMode) ||
1795       (Opts.FP32DenormalMode != llvm::DenormalMode::getIEEE()))
1796     GenerateArg(Consumer, OPT_fdenormal_fp_math_f32_EQ,
1797                 Opts.FP32DenormalMode.str());
1798 
1799   if (Opts.StructReturnConvention == CodeGenOptions::SRCK_OnStack) {
1800     OptSpecifier Opt =
1801         T.isPPC32() ? OPT_maix_struct_return : OPT_fpcc_struct_return;
1802     GenerateArg(Consumer, Opt);
1803   } else if (Opts.StructReturnConvention == CodeGenOptions::SRCK_InRegs) {
1804     OptSpecifier Opt =
1805         T.isPPC32() ? OPT_msvr4_struct_return : OPT_freg_struct_return;
1806     GenerateArg(Consumer, Opt);
1807   }
1808 
1809   if (Opts.EnableAIXExtendedAltivecABI)
1810     GenerateArg(Consumer, OPT_mabi_EQ_vec_extabi);
1811 
1812   if (Opts.XCOFFReadOnlyPointers)
1813     GenerateArg(Consumer, OPT_mxcoff_roptr);
1814 
1815   if (!Opts.OptRecordPasses.empty())
1816     GenerateArg(Consumer, OPT_opt_record_passes, Opts.OptRecordPasses);
1817 
1818   if (!Opts.OptRecordFormat.empty())
1819     GenerateArg(Consumer, OPT_opt_record_format, Opts.OptRecordFormat);
1820 
1821   GenerateOptimizationRemark(Consumer, OPT_Rpass_EQ, "pass",
1822                              Opts.OptimizationRemark);
1823 
1824   GenerateOptimizationRemark(Consumer, OPT_Rpass_missed_EQ, "pass-missed",
1825                              Opts.OptimizationRemarkMissed);
1826 
1827   GenerateOptimizationRemark(Consumer, OPT_Rpass_analysis_EQ, "pass-analysis",
1828                              Opts.OptimizationRemarkAnalysis);
1829 
1830   GenerateArg(Consumer, OPT_fdiagnostics_hotness_threshold_EQ,
1831               Opts.DiagnosticsHotnessThreshold
1832                   ? Twine(*Opts.DiagnosticsHotnessThreshold)
1833                   : "auto");
1834 
1835   GenerateArg(Consumer, OPT_fdiagnostics_misexpect_tolerance_EQ,
1836               Twine(*Opts.DiagnosticsMisExpectTolerance));
1837 
1838   for (StringRef Sanitizer : serializeSanitizerKinds(Opts.SanitizeRecover))
1839     GenerateArg(Consumer, OPT_fsanitize_recover_EQ, Sanitizer);
1840 
1841   for (StringRef Sanitizer : serializeSanitizerKinds(Opts.SanitizeTrap))
1842     GenerateArg(Consumer, OPT_fsanitize_trap_EQ, Sanitizer);
1843 
1844   for (StringRef Sanitizer :
1845        serializeSanitizerKinds(Opts.SanitizeMergeHandlers))
1846     GenerateArg(Consumer, OPT_fsanitize_merge_handlers_EQ, Sanitizer);
1847 
1848   SmallVector<std::string, 4> Values;
1849   serializeSanitizerMaskCutoffs(Opts.SanitizeSkipHotCutoffs, Values);
1850   for (std::string Sanitizer : Values)
1851     GenerateArg(Consumer, OPT_fsanitize_skip_hot_cutoff_EQ, Sanitizer);
1852 
1853   if (Opts.AllowRuntimeCheckSkipHotCutoff) {
1854     GenerateArg(Consumer, OPT_fallow_runtime_check_skip_hot_cutoff_EQ,
1855                 std::to_string(*Opts.AllowRuntimeCheckSkipHotCutoff));
1856   }
1857 
1858   for (StringRef Sanitizer :
1859        serializeSanitizerKinds(Opts.SanitizeAnnotateDebugInfo))
1860     GenerateArg(Consumer, OPT_fsanitize_annotate_debug_info_EQ, Sanitizer);
1861 
1862   if (!Opts.EmitVersionIdentMetadata)
1863     GenerateArg(Consumer, OPT_Qn);
1864 
1865   switch (Opts.FiniteLoops) {
1866   case CodeGenOptions::FiniteLoopsKind::Language:
1867     break;
1868   case CodeGenOptions::FiniteLoopsKind::Always:
1869     GenerateArg(Consumer, OPT_ffinite_loops);
1870     break;
1871   case CodeGenOptions::FiniteLoopsKind::Never:
1872     GenerateArg(Consumer, OPT_fno_finite_loops);
1873     break;
1874   }
1875 
1876   if (Opts.StaticClosure)
1877     GenerateArg(Consumer, OPT_static_libclosure);
1878 }
1879 
ParseCodeGenArgs(CodeGenOptions & Opts,ArgList & Args,InputKind IK,DiagnosticsEngine & Diags,const llvm::Triple & T,const std::string & OutputFile,const LangOptions & LangOptsRef)1880 bool CompilerInvocation::ParseCodeGenArgs(CodeGenOptions &Opts, ArgList &Args,
1881                                           InputKind IK,
1882                                           DiagnosticsEngine &Diags,
1883                                           const llvm::Triple &T,
1884                                           const std::string &OutputFile,
1885                                           const LangOptions &LangOptsRef) {
1886   unsigned NumErrorsBefore = Diags.getNumErrors();
1887 
1888   unsigned OptimizationLevel = getOptimizationLevel(Args, IK, Diags);
1889   // TODO: This could be done in Driver
1890   unsigned MaxOptLevel = 3;
1891   if (OptimizationLevel > MaxOptLevel) {
1892     // If the optimization level is not supported, fall back on the default
1893     // optimization
1894     Diags.Report(diag::warn_drv_optimization_value)
1895         << Args.getLastArg(OPT_O)->getAsString(Args) << "-O" << MaxOptLevel;
1896     OptimizationLevel = MaxOptLevel;
1897   }
1898   Opts.OptimizationLevel = OptimizationLevel;
1899 
1900   // The key paths of codegen options defined in Options.td start with
1901   // "CodeGenOpts.". Let's provide the expected variable name and type.
1902   CodeGenOptions &CodeGenOpts = Opts;
1903   // Some codegen options depend on language options. Let's provide the expected
1904   // variable name and type.
1905   const LangOptions *LangOpts = &LangOptsRef;
1906 
1907 #define CODEGEN_OPTION_WITH_MARSHALLING(...)                                   \
1908   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
1909 #include "clang/Driver/Options.inc"
1910 #undef CODEGEN_OPTION_WITH_MARSHALLING
1911 
1912   // At O0 we want to fully disable inlining outside of cases marked with
1913   // 'alwaysinline' that are required for correctness.
1914   if (Opts.OptimizationLevel == 0) {
1915     Opts.setInlining(CodeGenOptions::OnlyAlwaysInlining);
1916   } else if (const Arg *A = Args.getLastArg(options::OPT_finline_functions,
1917                                             options::OPT_finline_hint_functions,
1918                                             options::OPT_fno_inline_functions,
1919                                             options::OPT_fno_inline)) {
1920     // Explicit inlining flags can disable some or all inlining even at
1921     // optimization levels above zero.
1922     if (A->getOption().matches(options::OPT_finline_functions))
1923       Opts.setInlining(CodeGenOptions::NormalInlining);
1924     else if (A->getOption().matches(options::OPT_finline_hint_functions))
1925       Opts.setInlining(CodeGenOptions::OnlyHintInlining);
1926     else
1927       Opts.setInlining(CodeGenOptions::OnlyAlwaysInlining);
1928   } else {
1929     Opts.setInlining(CodeGenOptions::NormalInlining);
1930   }
1931 
1932   // PIC defaults to -fno-direct-access-external-data while non-PIC defaults to
1933   // -fdirect-access-external-data.
1934   Opts.DirectAccessExternalData =
1935       Args.hasArg(OPT_fdirect_access_external_data) ||
1936       (!Args.hasArg(OPT_fno_direct_access_external_data) &&
1937        LangOpts->PICLevel == 0);
1938 
1939   if (Arg *A = Args.getLastArg(OPT_debug_info_kind_EQ)) {
1940     unsigned Val =
1941         llvm::StringSwitch<unsigned>(A->getValue())
1942             .Case("line-tables-only", llvm::codegenoptions::DebugLineTablesOnly)
1943             .Case("line-directives-only",
1944                   llvm::codegenoptions::DebugDirectivesOnly)
1945             .Case("constructor", llvm::codegenoptions::DebugInfoConstructor)
1946             .Case("limited", llvm::codegenoptions::LimitedDebugInfo)
1947             .Case("standalone", llvm::codegenoptions::FullDebugInfo)
1948             .Case("unused-types", llvm::codegenoptions::UnusedTypeInfo)
1949             .Default(~0U);
1950     if (Val == ~0U)
1951       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args)
1952                                                 << A->getValue();
1953     else
1954       Opts.setDebugInfo(static_cast<llvm::codegenoptions::DebugInfoKind>(Val));
1955   }
1956 
1957   // If -fuse-ctor-homing is set and limited debug info is already on, then use
1958   // constructor homing, and vice versa for -fno-use-ctor-homing.
1959   if (const Arg *A =
1960           Args.getLastArg(OPT_fuse_ctor_homing, OPT_fno_use_ctor_homing)) {
1961     if (A->getOption().matches(OPT_fuse_ctor_homing) &&
1962         Opts.getDebugInfo() == llvm::codegenoptions::LimitedDebugInfo)
1963       Opts.setDebugInfo(llvm::codegenoptions::DebugInfoConstructor);
1964     if (A->getOption().matches(OPT_fno_use_ctor_homing) &&
1965         Opts.getDebugInfo() == llvm::codegenoptions::DebugInfoConstructor)
1966       Opts.setDebugInfo(llvm::codegenoptions::LimitedDebugInfo);
1967   }
1968 
1969   for (const auto &Arg : Args.getAllArgValues(OPT_fdebug_prefix_map_EQ)) {
1970     auto Split = StringRef(Arg).split('=');
1971     Opts.DebugPrefixMap.emplace_back(Split.first, Split.second);
1972   }
1973 
1974   for (const auto &Arg : Args.getAllArgValues(OPT_fcoverage_prefix_map_EQ)) {
1975     auto Split = StringRef(Arg).split('=');
1976     Opts.CoveragePrefixMap.emplace_back(Split.first, Split.second);
1977   }
1978 
1979   const llvm::Triple::ArchType DebugEntryValueArchs[] = {
1980       llvm::Triple::x86, llvm::Triple::x86_64, llvm::Triple::aarch64,
1981       llvm::Triple::arm, llvm::Triple::armeb, llvm::Triple::mips,
1982       llvm::Triple::mipsel, llvm::Triple::mips64, llvm::Triple::mips64el};
1983 
1984   if (Opts.OptimizationLevel > 0 && Opts.hasReducedDebugInfo() &&
1985       llvm::is_contained(DebugEntryValueArchs, T.getArch()))
1986     Opts.EmitCallSiteInfo = true;
1987 
1988   if (!Opts.EnableDIPreservationVerify && Opts.DIBugsReportFilePath.size()) {
1989     Diags.Report(diag::warn_ignoring_verify_debuginfo_preserve_export)
1990         << Opts.DIBugsReportFilePath;
1991     Opts.DIBugsReportFilePath = "";
1992   }
1993 
1994   Opts.NewStructPathTBAA = !Args.hasArg(OPT_no_struct_path_tbaa) &&
1995                            Args.hasArg(OPT_new_struct_path_tbaa);
1996   Opts.OptimizeSize = getOptimizationLevelSize(Args);
1997   Opts.SimplifyLibCalls = !LangOpts->NoBuiltin;
1998   if (Opts.SimplifyLibCalls)
1999     Opts.NoBuiltinFuncs = LangOpts->NoBuiltinFuncs;
2000   Opts.UnrollLoops =
2001       Args.hasFlag(OPT_funroll_loops, OPT_fno_unroll_loops,
2002                    (Opts.OptimizationLevel > 1));
2003   Opts.InterchangeLoops =
2004       Args.hasFlag(OPT_floop_interchange, OPT_fno_loop_interchange, false);
2005   Opts.BinutilsVersion =
2006       std::string(Args.getLastArgValue(OPT_fbinutils_version_EQ));
2007 
2008   Opts.DebugTemplateAlias = Args.hasArg(OPT_gtemplate_alias);
2009 
2010   Opts.DebugNameTable = static_cast<unsigned>(
2011       Args.hasArg(OPT_ggnu_pubnames)
2012           ? llvm::DICompileUnit::DebugNameTableKind::GNU
2013           : Args.hasArg(OPT_gpubnames)
2014                 ? llvm::DICompileUnit::DebugNameTableKind::Default
2015                 : llvm::DICompileUnit::DebugNameTableKind::None);
2016   if (const Arg *A = Args.getLastArg(OPT_gsimple_template_names_EQ)) {
2017     StringRef Value = A->getValue();
2018     if (Value != "simple" && Value != "mangled")
2019       Diags.Report(diag::err_drv_unsupported_option_argument)
2020           << A->getSpelling() << A->getValue();
2021     Opts.setDebugSimpleTemplateNames(
2022         StringRef(A->getValue()) == "simple"
2023             ? llvm::codegenoptions::DebugTemplateNamesKind::Simple
2024             : llvm::codegenoptions::DebugTemplateNamesKind::Mangled);
2025   }
2026 
2027   if (const Arg *A = Args.getLastArg(OPT_ftime_report, OPT_ftime_report_EQ,
2028                                      OPT_ftime_report_json)) {
2029     Opts.TimePasses = true;
2030 
2031     // -ftime-report= is only for new pass manager.
2032     if (const Arg *EQ = Args.getLastArg(OPT_ftime_report_EQ)) {
2033       StringRef Val = EQ->getValue();
2034       if (Val == "per-pass")
2035         Opts.TimePassesPerRun = false;
2036       else if (Val == "per-pass-run")
2037         Opts.TimePassesPerRun = true;
2038       else
2039         Diags.Report(diag::err_drv_invalid_value)
2040             << A->getAsString(Args) << A->getValue();
2041     }
2042 
2043     if (Args.getLastArg(OPT_ftime_report_json))
2044       Opts.TimePassesJson = true;
2045   }
2046 
2047   Opts.PrepareForLTO = false;
2048   Opts.PrepareForThinLTO = false;
2049   if (Arg *A = Args.getLastArg(OPT_flto_EQ)) {
2050     Opts.PrepareForLTO = true;
2051     StringRef S = A->getValue();
2052     if (S == "thin")
2053       Opts.PrepareForThinLTO = true;
2054     else if (S != "full")
2055       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << S;
2056     if (Args.hasArg(OPT_funified_lto))
2057       Opts.PrepareForThinLTO = true;
2058   }
2059   if (Arg *A = Args.getLastArg(OPT_fthinlto_index_EQ)) {
2060     if (IK.getLanguage() != Language::LLVM_IR)
2061       Diags.Report(diag::err_drv_argument_only_allowed_with)
2062           << A->getAsString(Args) << "-x ir";
2063     Opts.ThinLTOIndexFile =
2064         std::string(Args.getLastArgValue(OPT_fthinlto_index_EQ));
2065   }
2066   if (Arg *A = Args.getLastArg(OPT_save_temps_EQ))
2067     Opts.SaveTempsFilePrefix =
2068         llvm::StringSwitch<std::string>(A->getValue())
2069             .Case("obj", OutputFile)
2070             .Default(llvm::sys::path::filename(OutputFile).str());
2071 
2072   // The memory profile runtime appends the pid to make this name more unique.
2073   const char *MemProfileBasename = "memprof.profraw";
2074   if (Args.hasArg(OPT_fmemory_profile_EQ)) {
2075     SmallString<128> Path(Args.getLastArgValue(OPT_fmemory_profile_EQ));
2076     llvm::sys::path::append(Path, MemProfileBasename);
2077     Opts.MemoryProfileOutput = std::string(Path);
2078   } else if (Args.hasArg(OPT_fmemory_profile))
2079     Opts.MemoryProfileOutput = MemProfileBasename;
2080 
2081   if (Opts.CoverageNotesFile.size() || Opts.CoverageDataFile.size()) {
2082     if (Args.hasArg(OPT_coverage_version_EQ)) {
2083       StringRef CoverageVersion = Args.getLastArgValue(OPT_coverage_version_EQ);
2084       if (CoverageVersion.size() != 4) {
2085         Diags.Report(diag::err_drv_invalid_value)
2086             << Args.getLastArg(OPT_coverage_version_EQ)->getAsString(Args)
2087             << CoverageVersion;
2088       } else {
2089         memcpy(Opts.CoverageVersion, CoverageVersion.data(), 4);
2090       }
2091     }
2092   }
2093   // FIXME: For backend options that are not yet recorded as function
2094   // attributes in the IR, keep track of them so we can embed them in a
2095   // separate data section and use them when building the bitcode.
2096   for (const auto &A : Args) {
2097     // Do not encode output and input.
2098     if (A->getOption().getID() == options::OPT_o ||
2099         A->getOption().getID() == options::OPT_INPUT ||
2100         A->getOption().getID() == options::OPT_x ||
2101         A->getOption().getID() == options::OPT_fembed_bitcode ||
2102         A->getOption().matches(options::OPT_W_Group))
2103       continue;
2104     ArgStringList ASL;
2105     A->render(Args, ASL);
2106     for (const auto &arg : ASL) {
2107       StringRef ArgStr(arg);
2108       llvm::append_range(Opts.CmdArgs, ArgStr);
2109       // using \00 to separate each commandline options.
2110       Opts.CmdArgs.push_back('\0');
2111     }
2112   }
2113 
2114   auto XRayInstrBundles =
2115       Args.getAllArgValues(OPT_fxray_instrumentation_bundle);
2116   if (XRayInstrBundles.empty())
2117     Opts.XRayInstrumentationBundle.Mask = XRayInstrKind::All;
2118   else
2119     for (const auto &A : XRayInstrBundles)
2120       parseXRayInstrumentationBundle("-fxray-instrumentation-bundle=", A, Args,
2121                                      Diags, Opts.XRayInstrumentationBundle);
2122 
2123   if (const Arg *A = Args.getLastArg(OPT_fcf_protection_EQ)) {
2124     StringRef Name = A->getValue();
2125     if (Name == "full") {
2126       Opts.CFProtectionReturn = 1;
2127       Opts.CFProtectionBranch = 1;
2128     } else if (Name == "return")
2129       Opts.CFProtectionReturn = 1;
2130     else if (Name == "branch")
2131       Opts.CFProtectionBranch = 1;
2132     else if (Name != "none")
2133       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Name;
2134   }
2135 
2136   if (Opts.CFProtectionBranch && T.isRISCV()) {
2137     if (const Arg *A = Args.getLastArg(OPT_mcf_branch_label_scheme_EQ)) {
2138       const auto Scheme =
2139           llvm::StringSwitch<CFBranchLabelSchemeKind>(A->getValue())
2140 #define CF_BRANCH_LABEL_SCHEME(Kind, FlagVal)                                  \
2141   .Case(#FlagVal, CFBranchLabelSchemeKind::Kind)
2142 #include "clang/Basic/CFProtectionOptions.def"
2143               .Default(CFBranchLabelSchemeKind::Default);
2144       if (Scheme != CFBranchLabelSchemeKind::Default)
2145         Opts.setCFBranchLabelScheme(Scheme);
2146       else
2147         Diags.Report(diag::err_drv_invalid_value)
2148             << A->getAsString(Args) << A->getValue();
2149     }
2150   }
2151 
2152   if (const Arg *A = Args.getLastArg(OPT_mfunction_return_EQ)) {
2153     auto Val = llvm::StringSwitch<llvm::FunctionReturnThunksKind>(A->getValue())
2154                    .Case("keep", llvm::FunctionReturnThunksKind::Keep)
2155                    .Case("thunk-extern", llvm::FunctionReturnThunksKind::Extern)
2156                    .Default(llvm::FunctionReturnThunksKind::Invalid);
2157     // SystemZ might want to add support for "expolines."
2158     if (!T.isX86())
2159       Diags.Report(diag::err_drv_argument_not_allowed_with)
2160           << A->getSpelling() << T.getTriple();
2161     else if (Val == llvm::FunctionReturnThunksKind::Invalid)
2162       Diags.Report(diag::err_drv_invalid_value)
2163           << A->getAsString(Args) << A->getValue();
2164     else if (Val == llvm::FunctionReturnThunksKind::Extern &&
2165              Args.getLastArgValue(OPT_mcmodel_EQ) == "large")
2166       Diags.Report(diag::err_drv_argument_not_allowed_with)
2167           << A->getAsString(Args)
2168           << Args.getLastArg(OPT_mcmodel_EQ)->getAsString(Args);
2169     else
2170       Opts.FunctionReturnThunks = static_cast<unsigned>(Val);
2171   }
2172 
2173   for (auto *A :
2174        Args.filtered(OPT_mlink_bitcode_file, OPT_mlink_builtin_bitcode)) {
2175     CodeGenOptions::BitcodeFileToLink F;
2176     F.Filename = A->getValue();
2177     if (A->getOption().matches(OPT_mlink_builtin_bitcode)) {
2178       F.LinkFlags = llvm::Linker::Flags::LinkOnlyNeeded;
2179       // When linking CUDA bitcode, propagate function attributes so that
2180       // e.g. libdevice gets fast-math attrs if we're building with fast-math.
2181       F.PropagateAttrs = true;
2182       F.Internalize = true;
2183     }
2184     Opts.LinkBitcodeFiles.push_back(F);
2185   }
2186 
2187   if (Arg *A = Args.getLastArg(OPT_fdenormal_fp_math_EQ)) {
2188     StringRef Val = A->getValue();
2189     Opts.FPDenormalMode = llvm::parseDenormalFPAttribute(Val);
2190     Opts.FP32DenormalMode = Opts.FPDenormalMode;
2191     if (!Opts.FPDenormalMode.isValid())
2192       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Val;
2193   }
2194 
2195   if (Arg *A = Args.getLastArg(OPT_fdenormal_fp_math_f32_EQ)) {
2196     StringRef Val = A->getValue();
2197     Opts.FP32DenormalMode = llvm::parseDenormalFPAttribute(Val);
2198     if (!Opts.FP32DenormalMode.isValid())
2199       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Val;
2200   }
2201 
2202   // X86_32 has -fppc-struct-return and -freg-struct-return.
2203   // PPC32 has -maix-struct-return and -msvr4-struct-return.
2204   if (Arg *A =
2205           Args.getLastArg(OPT_fpcc_struct_return, OPT_freg_struct_return,
2206                           OPT_maix_struct_return, OPT_msvr4_struct_return)) {
2207     // TODO: We might want to consider enabling these options on AIX in the
2208     // future.
2209     if (T.isOSAIX())
2210       Diags.Report(diag::err_drv_unsupported_opt_for_target)
2211           << A->getSpelling() << T.str();
2212 
2213     const Option &O = A->getOption();
2214     if (O.matches(OPT_fpcc_struct_return) ||
2215         O.matches(OPT_maix_struct_return)) {
2216       Opts.setStructReturnConvention(CodeGenOptions::SRCK_OnStack);
2217     } else {
2218       assert(O.matches(OPT_freg_struct_return) ||
2219              O.matches(OPT_msvr4_struct_return));
2220       Opts.setStructReturnConvention(CodeGenOptions::SRCK_InRegs);
2221     }
2222   }
2223 
2224   if (Arg *A = Args.getLastArg(OPT_mxcoff_roptr)) {
2225     if (!T.isOSAIX())
2226       Diags.Report(diag::err_drv_unsupported_opt_for_target)
2227           << A->getSpelling() << T.str();
2228 
2229     // Since the storage mapping class is specified per csect,
2230     // without using data sections, it is less effective to use read-only
2231     // pointers. Using read-only pointers may cause other RO variables in the
2232     // same csect to become RW when the linker acts upon `-bforceimprw`;
2233     // therefore, we require that separate data sections
2234     // are used when `-mxcoff-roptr` is in effect. We respect the setting of
2235     // data-sections since we have not found reasons to do otherwise that
2236     // overcome the user surprise of not respecting the setting.
2237     if (!Args.hasFlag(OPT_fdata_sections, OPT_fno_data_sections, false))
2238       Diags.Report(diag::err_roptr_requires_data_sections);
2239 
2240     Opts.XCOFFReadOnlyPointers = true;
2241   }
2242 
2243   if (Arg *A = Args.getLastArg(OPT_mabi_EQ_quadword_atomics)) {
2244     if (!T.isOSAIX() || T.isPPC32())
2245       Diags.Report(diag::err_drv_unsupported_opt_for_target)
2246         << A->getSpelling() << T.str();
2247   }
2248 
2249   bool NeedLocTracking = false;
2250 
2251   if (!Opts.OptRecordFile.empty())
2252     NeedLocTracking = true;
2253 
2254   if (Arg *A = Args.getLastArg(OPT_opt_record_passes)) {
2255     Opts.OptRecordPasses = A->getValue();
2256     NeedLocTracking = true;
2257   }
2258 
2259   if (Arg *A = Args.getLastArg(OPT_opt_record_format)) {
2260     Opts.OptRecordFormat = A->getValue();
2261     NeedLocTracking = true;
2262   }
2263 
2264   Opts.OptimizationRemark =
2265       ParseOptimizationRemark(Diags, Args, OPT_Rpass_EQ, "pass");
2266 
2267   Opts.OptimizationRemarkMissed =
2268       ParseOptimizationRemark(Diags, Args, OPT_Rpass_missed_EQ, "pass-missed");
2269 
2270   Opts.OptimizationRemarkAnalysis = ParseOptimizationRemark(
2271       Diags, Args, OPT_Rpass_analysis_EQ, "pass-analysis");
2272 
2273   NeedLocTracking |= Opts.OptimizationRemark.hasValidPattern() ||
2274                      Opts.OptimizationRemarkMissed.hasValidPattern() ||
2275                      Opts.OptimizationRemarkAnalysis.hasValidPattern();
2276 
2277   bool UsingSampleProfile = !Opts.SampleProfileFile.empty();
2278   bool UsingProfile =
2279       UsingSampleProfile || !Opts.ProfileInstrumentUsePath.empty();
2280 
2281   if (Opts.DiagnosticsWithHotness && !UsingProfile &&
2282       // An IR file will contain PGO as metadata
2283       IK.getLanguage() != Language::LLVM_IR)
2284     Diags.Report(diag::warn_drv_diagnostics_hotness_requires_pgo)
2285         << "-fdiagnostics-show-hotness";
2286 
2287   // Parse remarks hotness threshold. Valid value is either integer or 'auto'.
2288   if (auto *arg =
2289           Args.getLastArg(options::OPT_fdiagnostics_hotness_threshold_EQ)) {
2290     auto ResultOrErr =
2291         llvm::remarks::parseHotnessThresholdOption(arg->getValue());
2292 
2293     if (!ResultOrErr) {
2294       Diags.Report(diag::err_drv_invalid_diagnotics_hotness_threshold)
2295           << "-fdiagnostics-hotness-threshold=";
2296     } else {
2297       Opts.DiagnosticsHotnessThreshold = *ResultOrErr;
2298       if ((!Opts.DiagnosticsHotnessThreshold ||
2299            *Opts.DiagnosticsHotnessThreshold > 0) &&
2300           !UsingProfile)
2301         Diags.Report(diag::warn_drv_diagnostics_hotness_requires_pgo)
2302             << "-fdiagnostics-hotness-threshold=";
2303     }
2304   }
2305 
2306   if (auto *arg =
2307           Args.getLastArg(options::OPT_fdiagnostics_misexpect_tolerance_EQ)) {
2308     auto ResultOrErr = parseToleranceOption(arg->getValue());
2309 
2310     if (!ResultOrErr) {
2311       Diags.Report(diag::err_drv_invalid_diagnotics_misexpect_tolerance)
2312           << "-fdiagnostics-misexpect-tolerance=";
2313     } else {
2314       Opts.DiagnosticsMisExpectTolerance = *ResultOrErr;
2315       if ((!Opts.DiagnosticsMisExpectTolerance ||
2316            *Opts.DiagnosticsMisExpectTolerance > 0) &&
2317           !UsingProfile)
2318         Diags.Report(diag::warn_drv_diagnostics_misexpect_requires_pgo)
2319             << "-fdiagnostics-misexpect-tolerance=";
2320     }
2321   }
2322 
2323   // If the user requested to use a sample profile for PGO, then the
2324   // backend will need to track source location information so the profile
2325   // can be incorporated into the IR.
2326   if (UsingSampleProfile)
2327     NeedLocTracking = true;
2328 
2329   if (!Opts.StackUsageOutput.empty())
2330     NeedLocTracking = true;
2331 
2332   // If the user requested a flag that requires source locations available in
2333   // the backend, make sure that the backend tracks source location information.
2334   if (NeedLocTracking &&
2335       Opts.getDebugInfo() == llvm::codegenoptions::NoDebugInfo)
2336     Opts.setDebugInfo(llvm::codegenoptions::LocTrackingOnly);
2337 
2338   // Parse -fsanitize-recover= arguments.
2339   // FIXME: Report unrecoverable sanitizers incorrectly specified here.
2340   parseSanitizerKinds("-fsanitize-recover=",
2341                       Args.getAllArgValues(OPT_fsanitize_recover_EQ), Diags,
2342                       Opts.SanitizeRecover);
2343   parseSanitizerKinds("-fsanitize-trap=",
2344                       Args.getAllArgValues(OPT_fsanitize_trap_EQ), Diags,
2345                       Opts.SanitizeTrap);
2346   parseSanitizerKinds("-fsanitize-merge=",
2347                       Args.getAllArgValues(OPT_fsanitize_merge_handlers_EQ),
2348                       Diags, Opts.SanitizeMergeHandlers);
2349 
2350   // Parse -fsanitize-skip-hot-cutoff= arguments.
2351   Opts.SanitizeSkipHotCutoffs = parseSanitizerWeightedKinds(
2352       "-fsanitize-skip-hot-cutoff=",
2353       Args.getAllArgValues(OPT_fsanitize_skip_hot_cutoff_EQ), Diags);
2354 
2355   parseSanitizerKinds(
2356       "-fsanitize-annotate-debug-info=",
2357       Args.getAllArgValues(OPT_fsanitize_annotate_debug_info_EQ), Diags,
2358       Opts.SanitizeAnnotateDebugInfo);
2359 
2360   if (StringRef V =
2361           Args.getLastArgValue(OPT_fallow_runtime_check_skip_hot_cutoff_EQ);
2362       !V.empty()) {
2363     double A;
2364     if (V.getAsDouble(A) || A < 0.0 || A > 1.0) {
2365       Diags.Report(diag::err_drv_invalid_value)
2366           << "-fallow-runtime-check-skip-hot-cutoff=" << V;
2367     } else {
2368       Opts.AllowRuntimeCheckSkipHotCutoff = A;
2369     }
2370   }
2371 
2372   Opts.EmitVersionIdentMetadata = Args.hasFlag(OPT_Qy, OPT_Qn, true);
2373 
2374   if (!LangOpts->CUDAIsDevice)
2375     parsePointerAuthOptions(Opts.PointerAuth, *LangOpts, T, Diags);
2376 
2377   if (Args.hasArg(options::OPT_ffinite_loops))
2378     Opts.FiniteLoops = CodeGenOptions::FiniteLoopsKind::Always;
2379   else if (Args.hasArg(options::OPT_fno_finite_loops))
2380     Opts.FiniteLoops = CodeGenOptions::FiniteLoopsKind::Never;
2381 
2382   Opts.EmitIEEENaNCompliantInsts = Args.hasFlag(
2383       options::OPT_mamdgpu_ieee, options::OPT_mno_amdgpu_ieee, true);
2384   if (!Opts.EmitIEEENaNCompliantInsts && !LangOptsRef.NoHonorNaNs)
2385     Diags.Report(diag::err_drv_amdgpu_ieee_without_no_honor_nans);
2386 
2387   Opts.StaticClosure = Args.hasArg(options::OPT_static_libclosure);
2388 
2389   return Diags.getNumErrors() == NumErrorsBefore;
2390 }
2391 
GenerateDependencyOutputArgs(const DependencyOutputOptions & Opts,ArgumentConsumer Consumer)2392 static void GenerateDependencyOutputArgs(const DependencyOutputOptions &Opts,
2393                                          ArgumentConsumer Consumer) {
2394   const DependencyOutputOptions &DependencyOutputOpts = Opts;
2395 #define DEPENDENCY_OUTPUT_OPTION_WITH_MARSHALLING(...)                         \
2396   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
2397 #include "clang/Driver/Options.inc"
2398 #undef DEPENDENCY_OUTPUT_OPTION_WITH_MARSHALLING
2399 
2400   if (Opts.ShowIncludesDest != ShowIncludesDestination::None)
2401     GenerateArg(Consumer, OPT_show_includes);
2402 
2403   for (const auto &Dep : Opts.ExtraDeps) {
2404     switch (Dep.second) {
2405     case EDK_SanitizeIgnorelist:
2406       // Sanitizer ignorelist arguments are generated from LanguageOptions.
2407       continue;
2408     case EDK_ModuleFile:
2409       // Module file arguments are generated from FrontendOptions and
2410       // HeaderSearchOptions.
2411       continue;
2412     case EDK_ProfileList:
2413       // Profile list arguments are generated from LanguageOptions via the
2414       // marshalling infrastructure.
2415       continue;
2416     case EDK_DepFileEntry:
2417       GenerateArg(Consumer, OPT_fdepfile_entry, Dep.first);
2418       break;
2419     }
2420   }
2421 }
2422 
ParseDependencyOutputArgs(DependencyOutputOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags,frontend::ActionKind Action,bool ShowLineMarkers)2423 static bool ParseDependencyOutputArgs(DependencyOutputOptions &Opts,
2424                                       ArgList &Args, DiagnosticsEngine &Diags,
2425                                       frontend::ActionKind Action,
2426                                       bool ShowLineMarkers) {
2427   unsigned NumErrorsBefore = Diags.getNumErrors();
2428 
2429   DependencyOutputOptions &DependencyOutputOpts = Opts;
2430 #define DEPENDENCY_OUTPUT_OPTION_WITH_MARSHALLING(...)                         \
2431   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
2432 #include "clang/Driver/Options.inc"
2433 #undef DEPENDENCY_OUTPUT_OPTION_WITH_MARSHALLING
2434 
2435   if (Args.hasArg(OPT_show_includes)) {
2436     // Writing both /showIncludes and preprocessor output to stdout
2437     // would produce interleaved output, so use stderr for /showIncludes.
2438     // This behaves the same as cl.exe, when /E, /EP or /P are passed.
2439     if (Action == frontend::PrintPreprocessedInput || !ShowLineMarkers)
2440       Opts.ShowIncludesDest = ShowIncludesDestination::Stderr;
2441     else
2442       Opts.ShowIncludesDest = ShowIncludesDestination::Stdout;
2443   } else {
2444     Opts.ShowIncludesDest = ShowIncludesDestination::None;
2445   }
2446 
2447   // Add sanitizer ignorelists as extra dependencies.
2448   // They won't be discovered by the regular preprocessor, so
2449   // we let make / ninja to know about this implicit dependency.
2450   if (!Args.hasArg(OPT_fno_sanitize_ignorelist)) {
2451     for (const auto *A : Args.filtered(OPT_fsanitize_ignorelist_EQ)) {
2452       StringRef Val = A->getValue();
2453       if (!Val.contains('='))
2454         Opts.ExtraDeps.emplace_back(std::string(Val), EDK_SanitizeIgnorelist);
2455     }
2456     if (Opts.IncludeSystemHeaders) {
2457       for (const auto *A : Args.filtered(OPT_fsanitize_system_ignorelist_EQ)) {
2458         StringRef Val = A->getValue();
2459         if (!Val.contains('='))
2460           Opts.ExtraDeps.emplace_back(std::string(Val), EDK_SanitizeIgnorelist);
2461       }
2462     }
2463   }
2464 
2465   // -fprofile-list= dependencies.
2466   for (const auto &Filename : Args.getAllArgValues(OPT_fprofile_list_EQ))
2467     Opts.ExtraDeps.emplace_back(Filename, EDK_ProfileList);
2468 
2469   // Propagate the extra dependencies.
2470   for (const auto *A : Args.filtered(OPT_fdepfile_entry))
2471     Opts.ExtraDeps.emplace_back(A->getValue(), EDK_DepFileEntry);
2472 
2473   // Only the -fmodule-file=<file> form.
2474   for (const auto *A : Args.filtered(OPT_fmodule_file)) {
2475     StringRef Val = A->getValue();
2476     if (!Val.contains('='))
2477       Opts.ExtraDeps.emplace_back(std::string(Val), EDK_ModuleFile);
2478   }
2479 
2480   // Check for invalid combinations of header-include-format
2481   // and header-include-filtering.
2482   if (Opts.HeaderIncludeFormat == HIFMT_Textual &&
2483       Opts.HeaderIncludeFiltering != HIFIL_None) {
2484     if (Args.hasArg(OPT_header_include_format_EQ))
2485       Diags.Report(diag::err_drv_print_header_cc1_invalid_combination)
2486           << headerIncludeFormatKindToString(Opts.HeaderIncludeFormat)
2487           << headerIncludeFilteringKindToString(Opts.HeaderIncludeFiltering);
2488     else
2489       Diags.Report(diag::err_drv_print_header_cc1_invalid_filtering)
2490           << headerIncludeFilteringKindToString(Opts.HeaderIncludeFiltering);
2491   } else if (Opts.HeaderIncludeFormat == HIFMT_JSON &&
2492              Opts.HeaderIncludeFiltering == HIFIL_None) {
2493     if (Args.hasArg(OPT_header_include_filtering_EQ))
2494       Diags.Report(diag::err_drv_print_header_cc1_invalid_combination)
2495           << headerIncludeFormatKindToString(Opts.HeaderIncludeFormat)
2496           << headerIncludeFilteringKindToString(Opts.HeaderIncludeFiltering);
2497     else
2498       Diags.Report(diag::err_drv_print_header_cc1_invalid_format)
2499           << headerIncludeFormatKindToString(Opts.HeaderIncludeFormat);
2500   }
2501 
2502   return Diags.getNumErrors() == NumErrorsBefore;
2503 }
2504 
parseShowColorsArgs(const ArgList & Args,bool DefaultColor)2505 static bool parseShowColorsArgs(const ArgList &Args, bool DefaultColor) {
2506   // Color diagnostics default to auto ("on" if terminal supports) in the driver
2507   // but default to off in cc1, needing an explicit OPT_fdiagnostics_color.
2508   // Support both clang's -f[no-]color-diagnostics and gcc's
2509   // -f[no-]diagnostics-colors[=never|always|auto].
2510   enum {
2511     Colors_On,
2512     Colors_Off,
2513     Colors_Auto
2514   } ShowColors = DefaultColor ? Colors_Auto : Colors_Off;
2515   for (auto *A : Args) {
2516     const Option &O = A->getOption();
2517     if (O.matches(options::OPT_fcolor_diagnostics)) {
2518       ShowColors = Colors_On;
2519     } else if (O.matches(options::OPT_fno_color_diagnostics)) {
2520       ShowColors = Colors_Off;
2521     } else if (O.matches(options::OPT_fdiagnostics_color_EQ)) {
2522       StringRef Value(A->getValue());
2523       if (Value == "always")
2524         ShowColors = Colors_On;
2525       else if (Value == "never")
2526         ShowColors = Colors_Off;
2527       else if (Value == "auto")
2528         ShowColors = Colors_Auto;
2529     }
2530   }
2531   return ShowColors == Colors_On ||
2532          (ShowColors == Colors_Auto &&
2533           llvm::sys::Process::StandardErrHasColors());
2534 }
2535 
checkVerifyPrefixes(const std::vector<std::string> & VerifyPrefixes,DiagnosticsEngine & Diags)2536 static bool checkVerifyPrefixes(const std::vector<std::string> &VerifyPrefixes,
2537                                 DiagnosticsEngine &Diags) {
2538   bool Success = true;
2539   for (const auto &Prefix : VerifyPrefixes) {
2540     // Every prefix must start with a letter and contain only alphanumeric
2541     // characters, hyphens, and underscores.
2542     auto BadChar = llvm::find_if(Prefix, [](char C) {
2543       return !isAlphanumeric(C) && C != '-' && C != '_';
2544     });
2545     if (BadChar != Prefix.end() || !isLetter(Prefix[0])) {
2546       Success = false;
2547       Diags.Report(diag::err_drv_invalid_value) << "-verify=" << Prefix;
2548       Diags.Report(diag::note_drv_verify_prefix_spelling);
2549     }
2550   }
2551   return Success;
2552 }
2553 
GenerateFileSystemArgs(const FileSystemOptions & Opts,ArgumentConsumer Consumer)2554 static void GenerateFileSystemArgs(const FileSystemOptions &Opts,
2555                                    ArgumentConsumer Consumer) {
2556   const FileSystemOptions &FileSystemOpts = Opts;
2557 
2558 #define FILE_SYSTEM_OPTION_WITH_MARSHALLING(...)                               \
2559   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
2560 #include "clang/Driver/Options.inc"
2561 #undef FILE_SYSTEM_OPTION_WITH_MARSHALLING
2562 }
2563 
ParseFileSystemArgs(FileSystemOptions & Opts,const ArgList & Args,DiagnosticsEngine & Diags)2564 static bool ParseFileSystemArgs(FileSystemOptions &Opts, const ArgList &Args,
2565                                 DiagnosticsEngine &Diags) {
2566   unsigned NumErrorsBefore = Diags.getNumErrors();
2567 
2568   FileSystemOptions &FileSystemOpts = Opts;
2569 
2570 #define FILE_SYSTEM_OPTION_WITH_MARSHALLING(...)                               \
2571   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
2572 #include "clang/Driver/Options.inc"
2573 #undef FILE_SYSTEM_OPTION_WITH_MARSHALLING
2574 
2575   return Diags.getNumErrors() == NumErrorsBefore;
2576 }
2577 
GenerateMigratorArgs(const MigratorOptions & Opts,ArgumentConsumer Consumer)2578 static void GenerateMigratorArgs(const MigratorOptions &Opts,
2579                                  ArgumentConsumer Consumer) {
2580   const MigratorOptions &MigratorOpts = Opts;
2581 #define MIGRATOR_OPTION_WITH_MARSHALLING(...)                                  \
2582   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
2583 #include "clang/Driver/Options.inc"
2584 #undef MIGRATOR_OPTION_WITH_MARSHALLING
2585 }
2586 
ParseMigratorArgs(MigratorOptions & Opts,const ArgList & Args,DiagnosticsEngine & Diags)2587 static bool ParseMigratorArgs(MigratorOptions &Opts, const ArgList &Args,
2588                               DiagnosticsEngine &Diags) {
2589   unsigned NumErrorsBefore = Diags.getNumErrors();
2590 
2591   MigratorOptions &MigratorOpts = Opts;
2592 
2593 #define MIGRATOR_OPTION_WITH_MARSHALLING(...)                                  \
2594   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
2595 #include "clang/Driver/Options.inc"
2596 #undef MIGRATOR_OPTION_WITH_MARSHALLING
2597 
2598   return Diags.getNumErrors() == NumErrorsBefore;
2599 }
2600 
GenerateDiagnosticArgs(const DiagnosticOptions & Opts,ArgumentConsumer Consumer,bool DefaultDiagColor)2601 void CompilerInvocationBase::GenerateDiagnosticArgs(
2602     const DiagnosticOptions &Opts, ArgumentConsumer Consumer,
2603     bool DefaultDiagColor) {
2604   const DiagnosticOptions *DiagnosticOpts = &Opts;
2605 #define DIAG_OPTION_WITH_MARSHALLING(...)                                      \
2606   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
2607 #include "clang/Driver/Options.inc"
2608 #undef DIAG_OPTION_WITH_MARSHALLING
2609 
2610   if (!Opts.DiagnosticSerializationFile.empty())
2611     GenerateArg(Consumer, OPT_diagnostic_serialized_file,
2612                 Opts.DiagnosticSerializationFile);
2613 
2614   if (Opts.ShowColors)
2615     GenerateArg(Consumer, OPT_fcolor_diagnostics);
2616 
2617   if (Opts.VerifyDiagnostics &&
2618       llvm::is_contained(Opts.VerifyPrefixes, "expected"))
2619     GenerateArg(Consumer, OPT_verify);
2620 
2621   for (const auto &Prefix : Opts.VerifyPrefixes)
2622     if (Prefix != "expected")
2623       GenerateArg(Consumer, OPT_verify_EQ, Prefix);
2624 
2625   DiagnosticLevelMask VIU = Opts.getVerifyIgnoreUnexpected();
2626   if (VIU == DiagnosticLevelMask::None) {
2627     // This is the default, don't generate anything.
2628   } else if (VIU == DiagnosticLevelMask::All) {
2629     GenerateArg(Consumer, OPT_verify_ignore_unexpected);
2630   } else {
2631     if (static_cast<unsigned>(VIU & DiagnosticLevelMask::Note) != 0)
2632       GenerateArg(Consumer, OPT_verify_ignore_unexpected_EQ, "note");
2633     if (static_cast<unsigned>(VIU & DiagnosticLevelMask::Remark) != 0)
2634       GenerateArg(Consumer, OPT_verify_ignore_unexpected_EQ, "remark");
2635     if (static_cast<unsigned>(VIU & DiagnosticLevelMask::Warning) != 0)
2636       GenerateArg(Consumer, OPT_verify_ignore_unexpected_EQ, "warning");
2637     if (static_cast<unsigned>(VIU & DiagnosticLevelMask::Error) != 0)
2638       GenerateArg(Consumer, OPT_verify_ignore_unexpected_EQ, "error");
2639   }
2640 
2641   for (const auto &Warning : Opts.Warnings) {
2642     // This option is automatically generated from UndefPrefixes.
2643     if (Warning == "undef-prefix")
2644       continue;
2645     // This option is automatically generated from CheckConstexprFunctionBodies.
2646     if (Warning == "invalid-constexpr" || Warning == "no-invalid-constexpr")
2647       continue;
2648     Consumer(StringRef("-W") + Warning);
2649   }
2650 
2651   for (const auto &Remark : Opts.Remarks) {
2652     // These arguments are generated from OptimizationRemark fields of
2653     // CodeGenOptions.
2654     StringRef IgnoredRemarks[] = {"pass",          "no-pass",
2655                                   "pass-analysis", "no-pass-analysis",
2656                                   "pass-missed",   "no-pass-missed"};
2657     if (llvm::is_contained(IgnoredRemarks, Remark))
2658       continue;
2659 
2660     Consumer(StringRef("-R") + Remark);
2661   }
2662 
2663   if (!Opts.DiagnosticSuppressionMappingsFile.empty()) {
2664     GenerateArg(Consumer, OPT_warning_suppression_mappings_EQ,
2665                 Opts.DiagnosticSuppressionMappingsFile);
2666   }
2667 }
2668 
2669 std::unique_ptr<DiagnosticOptions>
CreateAndPopulateDiagOpts(ArrayRef<const char * > Argv)2670 clang::CreateAndPopulateDiagOpts(ArrayRef<const char *> Argv) {
2671   auto DiagOpts = std::make_unique<DiagnosticOptions>();
2672   unsigned MissingArgIndex, MissingArgCount;
2673   InputArgList Args = getDriverOptTable().ParseArgs(
2674       Argv.slice(1), MissingArgIndex, MissingArgCount);
2675 
2676   bool ShowColors = true;
2677   if (std::optional<std::string> NoColor =
2678           llvm::sys::Process::GetEnv("NO_COLOR");
2679       NoColor && !NoColor->empty()) {
2680     // If the user set the NO_COLOR environment variable, we'll honor that
2681     // unless the command line overrides it.
2682     ShowColors = false;
2683   }
2684 
2685   // We ignore MissingArgCount and the return value of ParseDiagnosticArgs.
2686   // Any errors that would be diagnosed here will also be diagnosed later,
2687   // when the DiagnosticsEngine actually exists.
2688   (void)ParseDiagnosticArgs(*DiagOpts, Args, /*Diags=*/nullptr, ShowColors);
2689   return DiagOpts;
2690 }
2691 
ParseDiagnosticArgs(DiagnosticOptions & Opts,ArgList & Args,DiagnosticsEngine * Diags,bool DefaultDiagColor)2692 bool clang::ParseDiagnosticArgs(DiagnosticOptions &Opts, ArgList &Args,
2693                                 DiagnosticsEngine *Diags,
2694                                 bool DefaultDiagColor) {
2695   std::optional<DiagnosticOptions> IgnoringDiagOpts;
2696   std::optional<DiagnosticsEngine> IgnoringDiags;
2697   if (!Diags) {
2698     IgnoringDiagOpts.emplace();
2699     IgnoringDiags.emplace(new DiagnosticIDs(), *IgnoringDiagOpts,
2700                           new IgnoringDiagConsumer());
2701     Diags = &*IgnoringDiags;
2702   }
2703 
2704   unsigned NumErrorsBefore = Diags->getNumErrors();
2705 
2706   // The key paths of diagnostic options defined in Options.td start with
2707   // "DiagnosticOpts->". Let's provide the expected variable name and type.
2708   DiagnosticOptions *DiagnosticOpts = &Opts;
2709 
2710 #define DIAG_OPTION_WITH_MARSHALLING(...)                                      \
2711   PARSE_OPTION_WITH_MARSHALLING(Args, *Diags, __VA_ARGS__)
2712 #include "clang/Driver/Options.inc"
2713 #undef DIAG_OPTION_WITH_MARSHALLING
2714 
2715   llvm::sys::Process::UseANSIEscapeCodes(Opts.UseANSIEscapeCodes);
2716 
2717   if (Arg *A =
2718           Args.getLastArg(OPT_diagnostic_serialized_file, OPT__serialize_diags))
2719     Opts.DiagnosticSerializationFile = A->getValue();
2720   Opts.ShowColors = parseShowColorsArgs(Args, DefaultDiagColor);
2721 
2722   Opts.VerifyDiagnostics = Args.hasArg(OPT_verify) || Args.hasArg(OPT_verify_EQ);
2723   Opts.VerifyPrefixes = Args.getAllArgValues(OPT_verify_EQ);
2724   if (Args.hasArg(OPT_verify))
2725     Opts.VerifyPrefixes.push_back("expected");
2726   // Keep VerifyPrefixes in its original order for the sake of diagnostics, and
2727   // then sort it to prepare for fast lookup using std::binary_search.
2728   if (!checkVerifyPrefixes(Opts.VerifyPrefixes, *Diags))
2729     Opts.VerifyDiagnostics = false;
2730   else
2731     llvm::sort(Opts.VerifyPrefixes);
2732   DiagnosticLevelMask DiagMask = DiagnosticLevelMask::None;
2733   parseDiagnosticLevelMask(
2734       "-verify-ignore-unexpected=",
2735       Args.getAllArgValues(OPT_verify_ignore_unexpected_EQ), *Diags, DiagMask);
2736   if (Args.hasArg(OPT_verify_ignore_unexpected))
2737     DiagMask = DiagnosticLevelMask::All;
2738   Opts.setVerifyIgnoreUnexpected(DiagMask);
2739   if (Opts.TabStop == 0 || Opts.TabStop > DiagnosticOptions::MaxTabStop) {
2740     Diags->Report(diag::warn_ignoring_ftabstop_value)
2741         << Opts.TabStop << DiagnosticOptions::DefaultTabStop;
2742     Opts.TabStop = DiagnosticOptions::DefaultTabStop;
2743   }
2744 
2745   if (const Arg *A = Args.getLastArg(OPT_warning_suppression_mappings_EQ))
2746     Opts.DiagnosticSuppressionMappingsFile = A->getValue();
2747 
2748   addDiagnosticArgs(Args, OPT_W_Group, OPT_W_value_Group, Opts.Warnings);
2749   addDiagnosticArgs(Args, OPT_R_Group, OPT_R_value_Group, Opts.Remarks);
2750 
2751   return Diags->getNumErrors() == NumErrorsBefore;
2752 }
2753 
2754 /// Parse the argument to the -ftest-module-file-extension
2755 /// command-line argument.
2756 ///
2757 /// \returns true on error, false on success.
parseTestModuleFileExtensionArg(StringRef Arg,std::string & BlockName,unsigned & MajorVersion,unsigned & MinorVersion,bool & Hashed,std::string & UserInfo)2758 static bool parseTestModuleFileExtensionArg(StringRef Arg,
2759                                             std::string &BlockName,
2760                                             unsigned &MajorVersion,
2761                                             unsigned &MinorVersion,
2762                                             bool &Hashed,
2763                                             std::string &UserInfo) {
2764   SmallVector<StringRef, 5> Args;
2765   Arg.split(Args, ':', 5);
2766   if (Args.size() < 5)
2767     return true;
2768 
2769   BlockName = std::string(Args[0]);
2770   if (Args[1].getAsInteger(10, MajorVersion)) return true;
2771   if (Args[2].getAsInteger(10, MinorVersion)) return true;
2772   if (Args[3].getAsInteger(2, Hashed)) return true;
2773   if (Args.size() > 4)
2774     UserInfo = std::string(Args[4]);
2775   return false;
2776 }
2777 
2778 /// Return a table that associates command line option specifiers with the
2779 /// frontend action. Note: The pair {frontend::PluginAction, OPT_plugin} is
2780 /// intentionally missing, as this case is handled separately from other
2781 /// frontend options.
getFrontendActionTable()2782 static const auto &getFrontendActionTable() {
2783   static const std::pair<frontend::ActionKind, unsigned> Table[] = {
2784       {frontend::ASTDeclList, OPT_ast_list},
2785 
2786       {frontend::ASTDump, OPT_ast_dump_all_EQ},
2787       {frontend::ASTDump, OPT_ast_dump_all},
2788       {frontend::ASTDump, OPT_ast_dump_EQ},
2789       {frontend::ASTDump, OPT_ast_dump},
2790       {frontend::ASTDump, OPT_ast_dump_lookups},
2791       {frontend::ASTDump, OPT_ast_dump_decl_types},
2792 
2793       {frontend::ASTPrint, OPT_ast_print},
2794       {frontend::ASTView, OPT_ast_view},
2795       {frontend::DumpCompilerOptions, OPT_compiler_options_dump},
2796       {frontend::DumpRawTokens, OPT_dump_raw_tokens},
2797       {frontend::DumpTokens, OPT_dump_tokens},
2798       {frontend::EmitAssembly, OPT_S},
2799       {frontend::EmitBC, OPT_emit_llvm_bc},
2800       {frontend::EmitCIR, OPT_emit_cir},
2801       {frontend::EmitHTML, OPT_emit_html},
2802       {frontend::EmitLLVM, OPT_emit_llvm},
2803       {frontend::EmitLLVMOnly, OPT_emit_llvm_only},
2804       {frontend::EmitCodeGenOnly, OPT_emit_codegen_only},
2805       {frontend::EmitObj, OPT_emit_obj},
2806       {frontend::ExtractAPI, OPT_extract_api},
2807 
2808       {frontend::FixIt, OPT_fixit_EQ},
2809       {frontend::FixIt, OPT_fixit},
2810 
2811       {frontend::GenerateModule, OPT_emit_module},
2812       {frontend::GenerateModuleInterface, OPT_emit_module_interface},
2813       {frontend::GenerateReducedModuleInterface,
2814        OPT_emit_reduced_module_interface},
2815       {frontend::GenerateHeaderUnit, OPT_emit_header_unit},
2816       {frontend::GeneratePCH, OPT_emit_pch},
2817       {frontend::GenerateInterfaceStubs, OPT_emit_interface_stubs},
2818       {frontend::InitOnly, OPT_init_only},
2819       {frontend::ParseSyntaxOnly, OPT_fsyntax_only},
2820       {frontend::ModuleFileInfo, OPT_module_file_info},
2821       {frontend::VerifyPCH, OPT_verify_pch},
2822       {frontend::PrintPreamble, OPT_print_preamble},
2823       {frontend::PrintPreprocessedInput, OPT_E},
2824       {frontend::TemplightDump, OPT_templight_dump},
2825       {frontend::RewriteMacros, OPT_rewrite_macros},
2826       {frontend::RewriteObjC, OPT_rewrite_objc},
2827       {frontend::RewriteTest, OPT_rewrite_test},
2828       {frontend::RunAnalysis, OPT_analyze},
2829       {frontend::RunPreprocessorOnly, OPT_Eonly},
2830       {frontend::PrintDependencyDirectivesSourceMinimizerOutput,
2831        OPT_print_dependency_directives_minimized_source},
2832   };
2833 
2834   return Table;
2835 }
2836 
2837 /// Maps command line option to frontend action.
2838 static std::optional<frontend::ActionKind>
getFrontendAction(OptSpecifier & Opt)2839 getFrontendAction(OptSpecifier &Opt) {
2840   for (const auto &ActionOpt : getFrontendActionTable())
2841     if (ActionOpt.second == Opt.getID())
2842       return ActionOpt.first;
2843 
2844   return std::nullopt;
2845 }
2846 
2847 /// Maps frontend action to command line option.
2848 static std::optional<OptSpecifier>
getProgramActionOpt(frontend::ActionKind ProgramAction)2849 getProgramActionOpt(frontend::ActionKind ProgramAction) {
2850   for (const auto &ActionOpt : getFrontendActionTable())
2851     if (ActionOpt.first == ProgramAction)
2852       return OptSpecifier(ActionOpt.second);
2853 
2854   return std::nullopt;
2855 }
2856 
GenerateFrontendArgs(const FrontendOptions & Opts,ArgumentConsumer Consumer,bool IsHeader)2857 static void GenerateFrontendArgs(const FrontendOptions &Opts,
2858                                  ArgumentConsumer Consumer, bool IsHeader) {
2859   const FrontendOptions &FrontendOpts = Opts;
2860 #define FRONTEND_OPTION_WITH_MARSHALLING(...)                                  \
2861   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
2862 #include "clang/Driver/Options.inc"
2863 #undef FRONTEND_OPTION_WITH_MARSHALLING
2864 
2865   std::optional<OptSpecifier> ProgramActionOpt =
2866       getProgramActionOpt(Opts.ProgramAction);
2867 
2868   // Generating a simple flag covers most frontend actions.
2869   std::function<void()> GenerateProgramAction = [&]() {
2870     GenerateArg(Consumer, *ProgramActionOpt);
2871   };
2872 
2873   if (!ProgramActionOpt) {
2874     // PluginAction is the only program action handled separately.
2875     assert(Opts.ProgramAction == frontend::PluginAction &&
2876            "Frontend action without option.");
2877     GenerateProgramAction = [&]() {
2878       GenerateArg(Consumer, OPT_plugin, Opts.ActionName);
2879     };
2880   }
2881 
2882   // FIXME: Simplify the complex 'AST dump' command line.
2883   if (Opts.ProgramAction == frontend::ASTDump) {
2884     GenerateProgramAction = [&]() {
2885       // ASTDumpLookups, ASTDumpDeclTypes and ASTDumpFilter are generated via
2886       // marshalling infrastructure.
2887 
2888       if (Opts.ASTDumpFormat != ADOF_Default) {
2889         StringRef Format;
2890         switch (Opts.ASTDumpFormat) {
2891         case ADOF_Default:
2892           llvm_unreachable("Default AST dump format.");
2893         case ADOF_JSON:
2894           Format = "json";
2895           break;
2896         }
2897 
2898         if (Opts.ASTDumpAll)
2899           GenerateArg(Consumer, OPT_ast_dump_all_EQ, Format);
2900         if (Opts.ASTDumpDecls)
2901           GenerateArg(Consumer, OPT_ast_dump_EQ, Format);
2902       } else {
2903         if (Opts.ASTDumpAll)
2904           GenerateArg(Consumer, OPT_ast_dump_all);
2905         if (Opts.ASTDumpDecls)
2906           GenerateArg(Consumer, OPT_ast_dump);
2907       }
2908     };
2909   }
2910 
2911   if (Opts.ProgramAction == frontend::FixIt && !Opts.FixItSuffix.empty()) {
2912     GenerateProgramAction = [&]() {
2913       GenerateArg(Consumer, OPT_fixit_EQ, Opts.FixItSuffix);
2914     };
2915   }
2916 
2917   GenerateProgramAction();
2918 
2919   for (const auto &PluginArgs : Opts.PluginArgs) {
2920     Option Opt = getDriverOptTable().getOption(OPT_plugin_arg);
2921     for (const auto &PluginArg : PluginArgs.second)
2922       denormalizeString(Consumer,
2923                         Opt.getPrefix() + Opt.getName() + PluginArgs.first,
2924                         Opt.getKind(), 0, PluginArg);
2925   }
2926 
2927   for (const auto &Ext : Opts.ModuleFileExtensions)
2928     if (auto *TestExt = dyn_cast_or_null<TestModuleFileExtension>(Ext.get()))
2929       GenerateArg(Consumer, OPT_ftest_module_file_extension_EQ, TestExt->str());
2930 
2931   if (!Opts.CodeCompletionAt.FileName.empty())
2932     GenerateArg(Consumer, OPT_code_completion_at,
2933                 Opts.CodeCompletionAt.ToString());
2934 
2935   for (const auto &Plugin : Opts.Plugins)
2936     GenerateArg(Consumer, OPT_load, Plugin);
2937 
2938   // ASTDumpDecls and ASTDumpAll already handled with ProgramAction.
2939 
2940   for (const auto &ModuleFile : Opts.ModuleFiles)
2941     GenerateArg(Consumer, OPT_fmodule_file, ModuleFile);
2942 
2943   if (Opts.AuxTargetCPU)
2944     GenerateArg(Consumer, OPT_aux_target_cpu, *Opts.AuxTargetCPU);
2945 
2946   if (Opts.AuxTargetFeatures)
2947     for (const auto &Feature : *Opts.AuxTargetFeatures)
2948       GenerateArg(Consumer, OPT_aux_target_feature, Feature);
2949 
2950   {
2951     StringRef Preprocessed = Opts.DashX.isPreprocessed() ? "-cpp-output" : "";
2952     StringRef ModuleMap =
2953         Opts.DashX.getFormat() == InputKind::ModuleMap ? "-module-map" : "";
2954     StringRef HeaderUnit = "";
2955     switch (Opts.DashX.getHeaderUnitKind()) {
2956     case InputKind::HeaderUnit_None:
2957       break;
2958     case InputKind::HeaderUnit_User:
2959       HeaderUnit = "-user";
2960       break;
2961     case InputKind::HeaderUnit_System:
2962       HeaderUnit = "-system";
2963       break;
2964     case InputKind::HeaderUnit_Abs:
2965       HeaderUnit = "-header-unit";
2966       break;
2967     }
2968     StringRef Header = IsHeader ? "-header" : "";
2969 
2970     StringRef Lang;
2971     switch (Opts.DashX.getLanguage()) {
2972     case Language::C:
2973       Lang = "c";
2974       break;
2975     case Language::OpenCL:
2976       Lang = "cl";
2977       break;
2978     case Language::OpenCLCXX:
2979       Lang = "clcpp";
2980       break;
2981     case Language::CUDA:
2982       Lang = "cuda";
2983       break;
2984     case Language::HIP:
2985       Lang = "hip";
2986       break;
2987     case Language::CXX:
2988       Lang = "c++";
2989       break;
2990     case Language::ObjC:
2991       Lang = "objective-c";
2992       break;
2993     case Language::ObjCXX:
2994       Lang = "objective-c++";
2995       break;
2996     case Language::Asm:
2997       Lang = "assembler-with-cpp";
2998       break;
2999     case Language::Unknown:
3000       assert(Opts.DashX.getFormat() == InputKind::Precompiled &&
3001              "Generating -x argument for unknown language (not precompiled).");
3002       Lang = "ast";
3003       break;
3004     case Language::LLVM_IR:
3005       Lang = "ir";
3006       break;
3007     case Language::HLSL:
3008       Lang = "hlsl";
3009       break;
3010     case Language::CIR:
3011       Lang = "cir";
3012       break;
3013     }
3014 
3015     GenerateArg(Consumer, OPT_x,
3016                 Lang + HeaderUnit + Header + ModuleMap + Preprocessed);
3017   }
3018 
3019   // OPT_INPUT has a unique class, generate it directly.
3020   for (const auto &Input : Opts.Inputs)
3021     Consumer(Input.getFile());
3022 }
3023 
ParseFrontendArgs(FrontendOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags,bool & IsHeaderFile)3024 static bool ParseFrontendArgs(FrontendOptions &Opts, ArgList &Args,
3025                               DiagnosticsEngine &Diags, bool &IsHeaderFile) {
3026   unsigned NumErrorsBefore = Diags.getNumErrors();
3027 
3028   FrontendOptions &FrontendOpts = Opts;
3029 
3030 #define FRONTEND_OPTION_WITH_MARSHALLING(...)                                  \
3031   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
3032 #include "clang/Driver/Options.inc"
3033 #undef FRONTEND_OPTION_WITH_MARSHALLING
3034 
3035   Opts.ProgramAction = frontend::ParseSyntaxOnly;
3036   if (const Arg *A = Args.getLastArg(OPT_Action_Group)) {
3037     OptSpecifier Opt = OptSpecifier(A->getOption().getID());
3038     std::optional<frontend::ActionKind> ProgramAction = getFrontendAction(Opt);
3039     assert(ProgramAction && "Option specifier not in Action_Group.");
3040 
3041     if (ProgramAction == frontend::ASTDump &&
3042         (Opt == OPT_ast_dump_all_EQ || Opt == OPT_ast_dump_EQ)) {
3043       unsigned Val = llvm::StringSwitch<unsigned>(A->getValue())
3044                          .CaseLower("default", ADOF_Default)
3045                          .CaseLower("json", ADOF_JSON)
3046                          .Default(std::numeric_limits<unsigned>::max());
3047 
3048       if (Val != std::numeric_limits<unsigned>::max())
3049         Opts.ASTDumpFormat = static_cast<ASTDumpOutputFormat>(Val);
3050       else {
3051         Diags.Report(diag::err_drv_invalid_value)
3052             << A->getAsString(Args) << A->getValue();
3053         Opts.ASTDumpFormat = ADOF_Default;
3054       }
3055     }
3056 
3057     if (ProgramAction == frontend::FixIt && Opt == OPT_fixit_EQ)
3058       Opts.FixItSuffix = A->getValue();
3059 
3060     if (ProgramAction == frontend::GenerateInterfaceStubs) {
3061       StringRef ArgStr =
3062           Args.hasArg(OPT_interface_stub_version_EQ)
3063               ? Args.getLastArgValue(OPT_interface_stub_version_EQ)
3064               : "ifs-v1";
3065       if (ArgStr == "experimental-yaml-elf-v1" ||
3066           ArgStr == "experimental-ifs-v1" || ArgStr == "experimental-ifs-v2" ||
3067           ArgStr == "experimental-tapi-elf-v1") {
3068         std::string ErrorMessage =
3069             "Invalid interface stub format: " + ArgStr.str() +
3070             " is deprecated.";
3071         Diags.Report(diag::err_drv_invalid_value)
3072             << "Must specify a valid interface stub format type, ie: "
3073                "-interface-stub-version=ifs-v1"
3074             << ErrorMessage;
3075         ProgramAction = frontend::ParseSyntaxOnly;
3076       } else if (!ArgStr.starts_with("ifs-")) {
3077         std::string ErrorMessage =
3078             "Invalid interface stub format: " + ArgStr.str() + ".";
3079         Diags.Report(diag::err_drv_invalid_value)
3080             << "Must specify a valid interface stub format type, ie: "
3081                "-interface-stub-version=ifs-v1"
3082             << ErrorMessage;
3083         ProgramAction = frontend::ParseSyntaxOnly;
3084       }
3085     }
3086 
3087     Opts.ProgramAction = *ProgramAction;
3088 
3089     // Catch common mistakes when multiple actions are specified for cc1 (e.g.
3090     // -S -emit-llvm means -emit-llvm while -emit-llvm -S means -S). However, to
3091     // support driver `-c -Xclang ACTION` (-cc1 -emit-llvm file -main-file-name
3092     // X ACTION), we suppress the error when the two actions are separated by
3093     // -main-file-name.
3094     //
3095     // As an exception, accept composable -ast-dump*.
3096     if (!A->getSpelling().starts_with("-ast-dump")) {
3097       const Arg *SavedAction = nullptr;
3098       for (const Arg *AA :
3099            Args.filtered(OPT_Action_Group, OPT_main_file_name)) {
3100         if (AA->getOption().matches(OPT_main_file_name)) {
3101           SavedAction = nullptr;
3102         } else if (!SavedAction) {
3103           SavedAction = AA;
3104         } else {
3105           if (!A->getOption().matches(OPT_ast_dump_EQ))
3106             Diags.Report(diag::err_fe_invalid_multiple_actions)
3107                 << SavedAction->getSpelling() << A->getSpelling();
3108           break;
3109         }
3110       }
3111     }
3112   }
3113 
3114   if (const Arg* A = Args.getLastArg(OPT_plugin)) {
3115     Opts.Plugins.emplace_back(A->getValue(0));
3116     Opts.ProgramAction = frontend::PluginAction;
3117     Opts.ActionName = A->getValue();
3118   }
3119   for (const auto *AA : Args.filtered(OPT_plugin_arg))
3120     Opts.PluginArgs[AA->getValue(0)].emplace_back(AA->getValue(1));
3121 
3122   for (const std::string &Arg :
3123          Args.getAllArgValues(OPT_ftest_module_file_extension_EQ)) {
3124     std::string BlockName;
3125     unsigned MajorVersion;
3126     unsigned MinorVersion;
3127     bool Hashed;
3128     std::string UserInfo;
3129     if (parseTestModuleFileExtensionArg(Arg, BlockName, MajorVersion,
3130                                         MinorVersion, Hashed, UserInfo)) {
3131       Diags.Report(diag::err_test_module_file_extension_format) << Arg;
3132 
3133       continue;
3134     }
3135 
3136     // Add the testing module file extension.
3137     Opts.ModuleFileExtensions.push_back(
3138         std::make_shared<TestModuleFileExtension>(
3139             BlockName, MajorVersion, MinorVersion, Hashed, UserInfo));
3140   }
3141 
3142   if (const Arg *A = Args.getLastArg(OPT_code_completion_at)) {
3143     Opts.CodeCompletionAt =
3144       ParsedSourceLocation::FromString(A->getValue());
3145     if (Opts.CodeCompletionAt.FileName.empty()) {
3146       Diags.Report(diag::err_drv_invalid_value)
3147           << A->getAsString(Args) << A->getValue();
3148       Diags.Report(diag::note_command_line_code_loc_requirement);
3149     }
3150   }
3151 
3152   Opts.Plugins = Args.getAllArgValues(OPT_load);
3153   Opts.ASTDumpDecls = Args.hasArg(OPT_ast_dump, OPT_ast_dump_EQ);
3154   Opts.ASTDumpAll = Args.hasArg(OPT_ast_dump_all, OPT_ast_dump_all_EQ);
3155   // Only the -fmodule-file=<file> form.
3156   for (const auto *A : Args.filtered(OPT_fmodule_file)) {
3157     StringRef Val = A->getValue();
3158     if (!Val.contains('='))
3159       Opts.ModuleFiles.push_back(std::string(Val));
3160   }
3161 
3162   if (Opts.ProgramAction != frontend::GenerateModule && Opts.IsSystemModule)
3163     Diags.Report(diag::err_drv_argument_only_allowed_with) << "-fsystem-module"
3164                                                            << "-emit-module";
3165   if (Args.hasArg(OPT_fclangir) || Args.hasArg(OPT_emit_cir))
3166     Opts.UseClangIRPipeline = true;
3167 
3168 #if CLANG_ENABLE_CIR
3169   if (Args.hasArg(OPT_clangir_disable_passes))
3170     Opts.ClangIRDisablePasses = true;
3171 
3172   if (Args.hasArg(OPT_clangir_disable_verifier))
3173     Opts.ClangIRDisableCIRVerifier = true;
3174 #endif // CLANG_ENABLE_CIR
3175 
3176   if (Args.hasArg(OPT_aux_target_cpu))
3177     Opts.AuxTargetCPU = std::string(Args.getLastArgValue(OPT_aux_target_cpu));
3178   if (Args.hasArg(OPT_aux_target_feature))
3179     Opts.AuxTargetFeatures = Args.getAllArgValues(OPT_aux_target_feature);
3180 
3181   InputKind DashX(Language::Unknown);
3182   if (const Arg *A = Args.getLastArg(OPT_x)) {
3183     StringRef XValue = A->getValue();
3184 
3185     // Parse suffixes:
3186     // '<lang>(-[{header-unit,user,system}-]header|[-module-map][-cpp-output])'.
3187     // FIXME: Supporting '<lang>-header-cpp-output' would be useful.
3188     bool Preprocessed = XValue.consume_back("-cpp-output");
3189     bool ModuleMap = XValue.consume_back("-module-map");
3190     // Detect and consume the header indicator.
3191     bool IsHeader =
3192         XValue != "precompiled-header" && XValue.consume_back("-header");
3193 
3194     // If we have c++-{user,system}-header, that indicates a header unit input
3195     // likewise, if the user put -fmodule-header together with a header with an
3196     // absolute path (header-unit-header).
3197     InputKind::HeaderUnitKind HUK = InputKind::HeaderUnit_None;
3198     if (IsHeader || Preprocessed) {
3199       if (XValue.consume_back("-header-unit"))
3200         HUK = InputKind::HeaderUnit_Abs;
3201       else if (XValue.consume_back("-system"))
3202         HUK = InputKind::HeaderUnit_System;
3203       else if (XValue.consume_back("-user"))
3204         HUK = InputKind::HeaderUnit_User;
3205     }
3206 
3207     // The value set by this processing is an un-preprocessed source which is
3208     // not intended to be a module map or header unit.
3209     IsHeaderFile = IsHeader && !Preprocessed && !ModuleMap &&
3210                    HUK == InputKind::HeaderUnit_None;
3211 
3212     // Principal languages.
3213     DashX = llvm::StringSwitch<InputKind>(XValue)
3214                 .Case("c", Language::C)
3215                 .Case("cl", Language::OpenCL)
3216                 .Case("clcpp", Language::OpenCLCXX)
3217                 .Case("cuda", Language::CUDA)
3218                 .Case("hip", Language::HIP)
3219                 .Case("c++", Language::CXX)
3220                 .Case("objective-c", Language::ObjC)
3221                 .Case("objective-c++", Language::ObjCXX)
3222                 .Case("hlsl", Language::HLSL)
3223                 .Default(Language::Unknown);
3224 
3225     // "objc[++]-cpp-output" is an acceptable synonym for
3226     // "objective-c[++]-cpp-output".
3227     if (DashX.isUnknown() && Preprocessed && !IsHeaderFile && !ModuleMap &&
3228         HUK == InputKind::HeaderUnit_None)
3229       DashX = llvm::StringSwitch<InputKind>(XValue)
3230                   .Case("objc", Language::ObjC)
3231                   .Case("objc++", Language::ObjCXX)
3232                   .Default(Language::Unknown);
3233 
3234     // Some special cases cannot be combined with suffixes.
3235     if (DashX.isUnknown() && !Preprocessed && !IsHeaderFile && !ModuleMap &&
3236         HUK == InputKind::HeaderUnit_None)
3237       DashX = llvm::StringSwitch<InputKind>(XValue)
3238                   .Case("cpp-output", InputKind(Language::C).getPreprocessed())
3239                   .Case("assembler-with-cpp", Language::Asm)
3240                   .Cases("ast", "pcm", "precompiled-header",
3241                          InputKind(Language::Unknown, InputKind::Precompiled))
3242                   .Case("ir", Language::LLVM_IR)
3243                   .Case("cir", Language::CIR)
3244                   .Default(Language::Unknown);
3245 
3246     if (DashX.isUnknown())
3247       Diags.Report(diag::err_drv_invalid_value)
3248         << A->getAsString(Args) << A->getValue();
3249 
3250     if (Preprocessed)
3251       DashX = DashX.getPreprocessed();
3252     // A regular header is considered mutually exclusive with a header unit.
3253     if (HUK != InputKind::HeaderUnit_None) {
3254       DashX = DashX.withHeaderUnit(HUK);
3255       IsHeaderFile = true;
3256     } else if (IsHeaderFile)
3257       DashX = DashX.getHeader();
3258     if (ModuleMap)
3259       DashX = DashX.withFormat(InputKind::ModuleMap);
3260   }
3261 
3262   // '-' is the default input if none is given.
3263   std::vector<std::string> Inputs = Args.getAllArgValues(OPT_INPUT);
3264   Opts.Inputs.clear();
3265   if (Inputs.empty())
3266     Inputs.push_back("-");
3267 
3268   if (DashX.getHeaderUnitKind() != InputKind::HeaderUnit_None &&
3269       Inputs.size() > 1)
3270     Diags.Report(diag::err_drv_header_unit_extra_inputs) << Inputs[1];
3271 
3272   for (unsigned i = 0, e = Inputs.size(); i != e; ++i) {
3273     InputKind IK = DashX;
3274     if (IK.isUnknown()) {
3275       IK = FrontendOptions::getInputKindForExtension(
3276         StringRef(Inputs[i]).rsplit('.').second);
3277       // FIXME: Warn on this?
3278       if (IK.isUnknown())
3279         IK = Language::C;
3280       // FIXME: Remove this hack.
3281       if (i == 0)
3282         DashX = IK;
3283     }
3284 
3285     bool IsSystem = false;
3286 
3287     // The -emit-module action implicitly takes a module map.
3288     if (Opts.ProgramAction == frontend::GenerateModule &&
3289         IK.getFormat() == InputKind::Source) {
3290       IK = IK.withFormat(InputKind::ModuleMap);
3291       IsSystem = Opts.IsSystemModule;
3292     }
3293 
3294     Opts.Inputs.emplace_back(std::move(Inputs[i]), IK, IsSystem);
3295   }
3296 
3297   Opts.DashX = DashX;
3298 
3299   return Diags.getNumErrors() == NumErrorsBefore;
3300 }
3301 
GetResourcesPath(const char * Argv0,void * MainAddr)3302 std::string CompilerInvocation::GetResourcesPath(const char *Argv0,
3303                                                  void *MainAddr) {
3304   std::string ClangExecutable =
3305       llvm::sys::fs::getMainExecutable(Argv0, MainAddr);
3306   return Driver::GetResourcesPath(ClangExecutable);
3307 }
3308 
GenerateHeaderSearchArgs(const HeaderSearchOptions & Opts,ArgumentConsumer Consumer)3309 static void GenerateHeaderSearchArgs(const HeaderSearchOptions &Opts,
3310                                      ArgumentConsumer Consumer) {
3311   const HeaderSearchOptions *HeaderSearchOpts = &Opts;
3312 #define HEADER_SEARCH_OPTION_WITH_MARSHALLING(...)                             \
3313   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
3314 #include "clang/Driver/Options.inc"
3315 #undef HEADER_SEARCH_OPTION_WITH_MARSHALLING
3316 
3317   if (Opts.UseLibcxx)
3318     GenerateArg(Consumer, OPT_stdlib_EQ, "libc++");
3319 
3320   if (!Opts.ModuleCachePath.empty())
3321     GenerateArg(Consumer, OPT_fmodules_cache_path, Opts.ModuleCachePath);
3322 
3323   for (const auto &File : Opts.PrebuiltModuleFiles)
3324     GenerateArg(Consumer, OPT_fmodule_file, File.first + "=" + File.second);
3325 
3326   for (const auto &Path : Opts.PrebuiltModulePaths)
3327     GenerateArg(Consumer, OPT_fprebuilt_module_path, Path);
3328 
3329   for (const auto &Macro : Opts.ModulesIgnoreMacros)
3330     GenerateArg(Consumer, OPT_fmodules_ignore_macro, Macro.val());
3331 
3332   auto Matches = [](const HeaderSearchOptions::Entry &Entry,
3333                     llvm::ArrayRef<frontend::IncludeDirGroup> Groups,
3334                     std::optional<bool> IsFramework,
3335                     std::optional<bool> IgnoreSysRoot) {
3336     return llvm::is_contained(Groups, Entry.Group) &&
3337            (!IsFramework || (Entry.IsFramework == *IsFramework)) &&
3338            (!IgnoreSysRoot || (Entry.IgnoreSysRoot == *IgnoreSysRoot));
3339   };
3340 
3341   auto It = Opts.UserEntries.begin();
3342   auto End = Opts.UserEntries.end();
3343 
3344   // Add -I... and -F... options in order.
3345   for (; It < End && Matches(*It, {frontend::Angled}, std::nullopt, true);
3346        ++It) {
3347     OptSpecifier Opt = [It, Matches]() {
3348       if (Matches(*It, frontend::Angled, true, true))
3349         return OPT_F;
3350       if (Matches(*It, frontend::Angled, false, true))
3351         return OPT_I;
3352       llvm_unreachable("Unexpected HeaderSearchOptions::Entry.");
3353     }();
3354 
3355     GenerateArg(Consumer, Opt, It->Path);
3356   }
3357 
3358   // Note: some paths that came from "[-iprefix=xx] -iwithprefixbefore=yy" may
3359   // have already been generated as "-I[xx]yy". If that's the case, their
3360   // position on command line was such that this has no semantic impact on
3361   // include paths.
3362   for (; It < End &&
3363          Matches(*It, {frontend::After, frontend::Angled}, false, true);
3364        ++It) {
3365     OptSpecifier Opt =
3366         It->Group == frontend::After ? OPT_iwithprefix : OPT_iwithprefixbefore;
3367     GenerateArg(Consumer, Opt, It->Path);
3368   }
3369 
3370   // Note: Some paths that came from "-idirafter=xxyy" may have already been
3371   // generated as "-iwithprefix=xxyy". If that's the case, their position on
3372   // command line was such that this has no semantic impact on include paths.
3373   for (; It < End && Matches(*It, {frontend::After}, false, true); ++It)
3374     GenerateArg(Consumer, OPT_idirafter, It->Path);
3375   for (; It < End && Matches(*It, {frontend::Quoted}, false, true); ++It)
3376     GenerateArg(Consumer, OPT_iquote, It->Path);
3377   for (; It < End && Matches(*It, {frontend::System}, false, std::nullopt);
3378        ++It)
3379     GenerateArg(Consumer, It->IgnoreSysRoot ? OPT_isystem : OPT_iwithsysroot,
3380                 It->Path);
3381   for (; It < End && Matches(*It, {frontend::System}, true, true); ++It)
3382     GenerateArg(Consumer, OPT_iframework, It->Path);
3383   for (; It < End && Matches(*It, {frontend::System}, true, false); ++It)
3384     GenerateArg(Consumer, OPT_iframeworkwithsysroot, It->Path);
3385 
3386   // Add the paths for the various language specific isystem flags.
3387   for (; It < End && Matches(*It, {frontend::CSystem}, false, true); ++It)
3388     GenerateArg(Consumer, OPT_c_isystem, It->Path);
3389   for (; It < End && Matches(*It, {frontend::CXXSystem}, false, true); ++It)
3390     GenerateArg(Consumer, OPT_cxx_isystem, It->Path);
3391   for (; It < End && Matches(*It, {frontend::ObjCSystem}, false, true); ++It)
3392     GenerateArg(Consumer, OPT_objc_isystem, It->Path);
3393   for (; It < End && Matches(*It, {frontend::ObjCXXSystem}, false, true); ++It)
3394     GenerateArg(Consumer, OPT_objcxx_isystem, It->Path);
3395 
3396   // Add the internal paths from a driver that detects standard include paths.
3397   // Note: Some paths that came from "-internal-isystem" arguments may have
3398   // already been generated as "-isystem". If that's the case, their position on
3399   // command line was such that this has no semantic impact on include paths.
3400   for (; It < End &&
3401          Matches(*It, {frontend::System, frontend::ExternCSystem}, false, true);
3402        ++It) {
3403     OptSpecifier Opt = It->Group == frontend::System
3404                            ? OPT_internal_isystem
3405                            : OPT_internal_externc_isystem;
3406     GenerateArg(Consumer, Opt, It->Path);
3407   }
3408   for (; It < End && Matches(*It, {frontend::System}, true, true); ++It)
3409     GenerateArg(Consumer, OPT_internal_iframework, It->Path);
3410 
3411   assert(It == End && "Unhandled HeaderSearchOption::Entry.");
3412 
3413   // Add the path prefixes which are implicitly treated as being system headers.
3414   for (const auto &P : Opts.SystemHeaderPrefixes) {
3415     OptSpecifier Opt = P.IsSystemHeader ? OPT_system_header_prefix
3416                                         : OPT_no_system_header_prefix;
3417     GenerateArg(Consumer, Opt, P.Prefix);
3418   }
3419 
3420   for (const std::string &F : Opts.VFSOverlayFiles)
3421     GenerateArg(Consumer, OPT_ivfsoverlay, F);
3422 }
3423 
ParseHeaderSearchArgs(HeaderSearchOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags,const std::string & WorkingDir)3424 static bool ParseHeaderSearchArgs(HeaderSearchOptions &Opts, ArgList &Args,
3425                                   DiagnosticsEngine &Diags,
3426                                   const std::string &WorkingDir) {
3427   unsigned NumErrorsBefore = Diags.getNumErrors();
3428 
3429   HeaderSearchOptions *HeaderSearchOpts = &Opts;
3430 
3431 #define HEADER_SEARCH_OPTION_WITH_MARSHALLING(...)                             \
3432   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
3433 #include "clang/Driver/Options.inc"
3434 #undef HEADER_SEARCH_OPTION_WITH_MARSHALLING
3435 
3436   if (const Arg *A = Args.getLastArg(OPT_stdlib_EQ))
3437     Opts.UseLibcxx = (strcmp(A->getValue(), "libc++") == 0);
3438 
3439   // Canonicalize -fmodules-cache-path before storing it.
3440   SmallString<128> P(Args.getLastArgValue(OPT_fmodules_cache_path));
3441   if (!(P.empty() || llvm::sys::path::is_absolute(P))) {
3442     if (WorkingDir.empty())
3443       llvm::sys::fs::make_absolute(P);
3444     else
3445       llvm::sys::fs::make_absolute(WorkingDir, P);
3446   }
3447   llvm::sys::path::remove_dots(P);
3448   Opts.ModuleCachePath = std::string(P);
3449 
3450   // Only the -fmodule-file=<name>=<file> form.
3451   for (const auto *A : Args.filtered(OPT_fmodule_file)) {
3452     StringRef Val = A->getValue();
3453     if (Val.contains('=')) {
3454       auto Split = Val.split('=');
3455       Opts.PrebuiltModuleFiles.insert_or_assign(
3456           std::string(Split.first), std::string(Split.second));
3457     }
3458   }
3459   for (const auto *A : Args.filtered(OPT_fprebuilt_module_path))
3460     Opts.AddPrebuiltModulePath(A->getValue());
3461 
3462   for (const auto *A : Args.filtered(OPT_fmodules_ignore_macro)) {
3463     StringRef MacroDef = A->getValue();
3464     Opts.ModulesIgnoreMacros.insert(
3465         llvm::CachedHashString(MacroDef.split('=').first));
3466   }
3467 
3468   // Add -I... and -F... options in order.
3469   bool IsSysrootSpecified =
3470       Args.hasArg(OPT__sysroot_EQ) || Args.hasArg(OPT_isysroot);
3471 
3472   // Expand a leading `=` to the sysroot if one was passed (and it's not a
3473   // framework flag).
3474   auto PrefixHeaderPath = [IsSysrootSpecified,
3475                            &Opts](const llvm::opt::Arg *A,
3476                                   bool IsFramework = false) -> std::string {
3477     assert(A->getNumValues() && "Unexpected empty search path flag!");
3478     if (IsSysrootSpecified && !IsFramework && A->getValue()[0] == '=') {
3479       SmallString<32> Buffer;
3480       llvm::sys::path::append(Buffer, Opts.Sysroot,
3481                               llvm::StringRef(A->getValue()).substr(1));
3482       return std::string(Buffer);
3483     }
3484     return A->getValue();
3485   };
3486 
3487   for (const auto *A : Args.filtered(OPT_I, OPT_F)) {
3488     bool IsFramework = A->getOption().matches(OPT_F);
3489     Opts.AddPath(PrefixHeaderPath(A, IsFramework), frontend::Angled,
3490                  IsFramework, /*IgnoreSysroot=*/true);
3491   }
3492 
3493   // Add -iprefix/-iwithprefix/-iwithprefixbefore options.
3494   StringRef Prefix = ""; // FIXME: This isn't the correct default prefix.
3495   for (const auto *A :
3496        Args.filtered(OPT_iprefix, OPT_iwithprefix, OPT_iwithprefixbefore)) {
3497     if (A->getOption().matches(OPT_iprefix))
3498       Prefix = A->getValue();
3499     else if (A->getOption().matches(OPT_iwithprefix))
3500       Opts.AddPath(Prefix.str() + A->getValue(), frontend::After, false, true);
3501     else
3502       Opts.AddPath(Prefix.str() + A->getValue(), frontend::Angled, false, true);
3503   }
3504 
3505   for (const auto *A : Args.filtered(OPT_idirafter))
3506     Opts.AddPath(PrefixHeaderPath(A), frontend::After, false, true);
3507   for (const auto *A : Args.filtered(OPT_iquote))
3508     Opts.AddPath(PrefixHeaderPath(A), frontend::Quoted, false, true);
3509 
3510   for (const auto *A : Args.filtered(OPT_isystem, OPT_iwithsysroot)) {
3511     if (A->getOption().matches(OPT_iwithsysroot)) {
3512       Opts.AddPath(A->getValue(), frontend::System, false,
3513                    /*IgnoreSysRoot=*/false);
3514       continue;
3515     }
3516     Opts.AddPath(PrefixHeaderPath(A), frontend::System, false, true);
3517   }
3518   for (const auto *A : Args.filtered(OPT_iframework))
3519     Opts.AddPath(A->getValue(), frontend::System, true, true);
3520   for (const auto *A : Args.filtered(OPT_iframeworkwithsysroot))
3521     Opts.AddPath(A->getValue(), frontend::System, /*IsFramework=*/true,
3522                  /*IgnoreSysRoot=*/false);
3523 
3524   // Add the paths for the various language specific isystem flags.
3525   for (const auto *A : Args.filtered(OPT_c_isystem))
3526     Opts.AddPath(A->getValue(), frontend::CSystem, false, true);
3527   for (const auto *A : Args.filtered(OPT_cxx_isystem))
3528     Opts.AddPath(A->getValue(), frontend::CXXSystem, false, true);
3529   for (const auto *A : Args.filtered(OPT_objc_isystem))
3530     Opts.AddPath(A->getValue(), frontend::ObjCSystem, false,true);
3531   for (const auto *A : Args.filtered(OPT_objcxx_isystem))
3532     Opts.AddPath(A->getValue(), frontend::ObjCXXSystem, false, true);
3533 
3534   // Add the internal paths from a driver that detects standard include paths.
3535   for (const auto *A :
3536        Args.filtered(OPT_internal_isystem, OPT_internal_externc_isystem)) {
3537     frontend::IncludeDirGroup Group = frontend::System;
3538     if (A->getOption().matches(OPT_internal_externc_isystem))
3539       Group = frontend::ExternCSystem;
3540     Opts.AddPath(A->getValue(), Group, false, true);
3541   }
3542   for (const auto *A : Args.filtered(OPT_internal_iframework))
3543     Opts.AddPath(A->getValue(), frontend::System, true, true);
3544 
3545   // Add the path prefixes which are implicitly treated as being system headers.
3546   for (const auto *A :
3547        Args.filtered(OPT_system_header_prefix, OPT_no_system_header_prefix))
3548     Opts.AddSystemHeaderPrefix(
3549         A->getValue(), A->getOption().matches(OPT_system_header_prefix));
3550 
3551   for (const auto *A : Args.filtered(OPT_ivfsoverlay, OPT_vfsoverlay))
3552     Opts.AddVFSOverlayFile(A->getValue());
3553 
3554   return Diags.getNumErrors() == NumErrorsBefore;
3555 }
3556 
GenerateAPINotesArgs(const APINotesOptions & Opts,ArgumentConsumer Consumer)3557 static void GenerateAPINotesArgs(const APINotesOptions &Opts,
3558                                  ArgumentConsumer Consumer) {
3559   if (!Opts.SwiftVersion.empty())
3560     GenerateArg(Consumer, OPT_fapinotes_swift_version,
3561                 Opts.SwiftVersion.getAsString());
3562 
3563   for (const auto &Path : Opts.ModuleSearchPaths)
3564     GenerateArg(Consumer, OPT_iapinotes_modules, Path);
3565 }
3566 
ParseAPINotesArgs(APINotesOptions & Opts,ArgList & Args,DiagnosticsEngine & diags)3567 static void ParseAPINotesArgs(APINotesOptions &Opts, ArgList &Args,
3568                               DiagnosticsEngine &diags) {
3569   if (const Arg *A = Args.getLastArg(OPT_fapinotes_swift_version)) {
3570     if (Opts.SwiftVersion.tryParse(A->getValue()))
3571       diags.Report(diag::err_drv_invalid_value)
3572           << A->getAsString(Args) << A->getValue();
3573   }
3574   for (const Arg *A : Args.filtered(OPT_iapinotes_modules))
3575     Opts.ModuleSearchPaths.push_back(A->getValue());
3576 }
3577 
GeneratePointerAuthArgs(const LangOptions & Opts,ArgumentConsumer Consumer)3578 static void GeneratePointerAuthArgs(const LangOptions &Opts,
3579                                     ArgumentConsumer Consumer) {
3580   if (Opts.PointerAuthIntrinsics)
3581     GenerateArg(Consumer, OPT_fptrauth_intrinsics);
3582   if (Opts.PointerAuthCalls)
3583     GenerateArg(Consumer, OPT_fptrauth_calls);
3584   if (Opts.PointerAuthReturns)
3585     GenerateArg(Consumer, OPT_fptrauth_returns);
3586   if (Opts.PointerAuthIndirectGotos)
3587     GenerateArg(Consumer, OPT_fptrauth_indirect_gotos);
3588   if (Opts.PointerAuthAuthTraps)
3589     GenerateArg(Consumer, OPT_fptrauth_auth_traps);
3590   if (Opts.PointerAuthVTPtrAddressDiscrimination)
3591     GenerateArg(Consumer, OPT_fptrauth_vtable_pointer_address_discrimination);
3592   if (Opts.PointerAuthVTPtrTypeDiscrimination)
3593     GenerateArg(Consumer, OPT_fptrauth_vtable_pointer_type_discrimination);
3594   if (Opts.PointerAuthTypeInfoVTPtrDiscrimination)
3595     GenerateArg(Consumer, OPT_fptrauth_type_info_vtable_pointer_discrimination);
3596   if (Opts.PointerAuthFunctionTypeDiscrimination)
3597     GenerateArg(Consumer, OPT_fptrauth_function_pointer_type_discrimination);
3598   if (Opts.PointerAuthInitFini)
3599     GenerateArg(Consumer, OPT_fptrauth_init_fini);
3600   if (Opts.PointerAuthInitFiniAddressDiscrimination)
3601     GenerateArg(Consumer, OPT_fptrauth_init_fini_address_discrimination);
3602   if (Opts.PointerAuthELFGOT)
3603     GenerateArg(Consumer, OPT_fptrauth_elf_got);
3604   if (Opts.AArch64JumpTableHardening)
3605     GenerateArg(Consumer, OPT_faarch64_jump_table_hardening);
3606   if (Opts.PointerAuthObjcIsa)
3607     GenerateArg(Consumer, OPT_fptrauth_objc_isa);
3608   if (Opts.PointerAuthObjcInterfaceSel)
3609     GenerateArg(Consumer, OPT_fptrauth_objc_interface_sel);
3610   if (Opts.PointerAuthObjcClassROPointers)
3611     GenerateArg(Consumer, OPT_fptrauth_objc_class_ro);
3612   if (Opts.PointerAuthBlockDescriptorPointers)
3613     GenerateArg(Consumer, OPT_fptrauth_block_descriptor_pointers);
3614 }
3615 
ParsePointerAuthArgs(LangOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags)3616 static void ParsePointerAuthArgs(LangOptions &Opts, ArgList &Args,
3617                                  DiagnosticsEngine &Diags) {
3618   Opts.PointerAuthIntrinsics = Args.hasArg(OPT_fptrauth_intrinsics);
3619   Opts.PointerAuthCalls = Args.hasArg(OPT_fptrauth_calls);
3620   Opts.PointerAuthReturns = Args.hasArg(OPT_fptrauth_returns);
3621   Opts.PointerAuthIndirectGotos = Args.hasArg(OPT_fptrauth_indirect_gotos);
3622   Opts.PointerAuthAuthTraps = Args.hasArg(OPT_fptrauth_auth_traps);
3623   Opts.PointerAuthVTPtrAddressDiscrimination =
3624       Args.hasArg(OPT_fptrauth_vtable_pointer_address_discrimination);
3625   Opts.PointerAuthVTPtrTypeDiscrimination =
3626       Args.hasArg(OPT_fptrauth_vtable_pointer_type_discrimination);
3627   Opts.PointerAuthTypeInfoVTPtrDiscrimination =
3628       Args.hasArg(OPT_fptrauth_type_info_vtable_pointer_discrimination);
3629   Opts.PointerAuthFunctionTypeDiscrimination =
3630       Args.hasArg(OPT_fptrauth_function_pointer_type_discrimination);
3631   Opts.PointerAuthInitFini = Args.hasArg(OPT_fptrauth_init_fini);
3632   Opts.PointerAuthInitFiniAddressDiscrimination =
3633       Args.hasArg(OPT_fptrauth_init_fini_address_discrimination);
3634   Opts.PointerAuthELFGOT = Args.hasArg(OPT_fptrauth_elf_got);
3635   Opts.AArch64JumpTableHardening =
3636       Args.hasArg(OPT_faarch64_jump_table_hardening);
3637   Opts.PointerAuthBlockDescriptorPointers =
3638       Args.hasArg(OPT_fptrauth_block_descriptor_pointers);
3639   Opts.PointerAuthObjcIsa = Args.hasArg(OPT_fptrauth_objc_isa);
3640   Opts.PointerAuthObjcClassROPointers = Args.hasArg(OPT_fptrauth_objc_class_ro);
3641   Opts.PointerAuthObjcInterfaceSel =
3642       Args.hasArg(OPT_fptrauth_objc_interface_sel);
3643 
3644   if (Opts.PointerAuthObjcInterfaceSel)
3645     Opts.PointerAuthObjcInterfaceSelKey =
3646         static_cast<unsigned>(PointerAuthSchema::ARM8_3Key::ASDB);
3647 }
3648 
3649 /// Check if input file kind and language standard are compatible.
IsInputCompatibleWithStandard(InputKind IK,const LangStandard & S)3650 static bool IsInputCompatibleWithStandard(InputKind IK,
3651                                           const LangStandard &S) {
3652   switch (IK.getLanguage()) {
3653   case Language::Unknown:
3654   case Language::LLVM_IR:
3655   case Language::CIR:
3656     llvm_unreachable("should not parse language flags for this input");
3657 
3658   case Language::C:
3659   case Language::ObjC:
3660     return S.getLanguage() == Language::C;
3661 
3662   case Language::OpenCL:
3663     return S.getLanguage() == Language::OpenCL ||
3664            S.getLanguage() == Language::OpenCLCXX;
3665 
3666   case Language::OpenCLCXX:
3667     return S.getLanguage() == Language::OpenCLCXX;
3668 
3669   case Language::CXX:
3670   case Language::ObjCXX:
3671     return S.getLanguage() == Language::CXX;
3672 
3673   case Language::CUDA:
3674     // FIXME: What -std= values should be permitted for CUDA compilations?
3675     return S.getLanguage() == Language::CUDA ||
3676            S.getLanguage() == Language::CXX;
3677 
3678   case Language::HIP:
3679     return S.getLanguage() == Language::CXX || S.getLanguage() == Language::HIP;
3680 
3681   case Language::Asm:
3682     // Accept (and ignore) all -std= values.
3683     // FIXME: The -std= value is not ignored; it affects the tokenization
3684     // and preprocessing rules if we're preprocessing this asm input.
3685     return true;
3686 
3687   case Language::HLSL:
3688     return S.getLanguage() == Language::HLSL;
3689   }
3690 
3691   llvm_unreachable("unexpected input language");
3692 }
3693 
3694 /// Get language name for given input kind.
GetInputKindName(InputKind IK)3695 static StringRef GetInputKindName(InputKind IK) {
3696   switch (IK.getLanguage()) {
3697   case Language::C:
3698     return "C";
3699   case Language::ObjC:
3700     return "Objective-C";
3701   case Language::CXX:
3702     return "C++";
3703   case Language::ObjCXX:
3704     return "Objective-C++";
3705   case Language::OpenCL:
3706     return "OpenCL";
3707   case Language::OpenCLCXX:
3708     return "C++ for OpenCL";
3709   case Language::CUDA:
3710     return "CUDA";
3711   case Language::HIP:
3712     return "HIP";
3713 
3714   case Language::Asm:
3715     return "Asm";
3716   case Language::LLVM_IR:
3717     return "LLVM IR";
3718   case Language::CIR:
3719     return "Clang IR";
3720 
3721   case Language::HLSL:
3722     return "HLSL";
3723 
3724   case Language::Unknown:
3725     break;
3726   }
3727   llvm_unreachable("unknown input language");
3728 }
3729 
getExceptionHandlingName(unsigned EHK)3730 static StringRef getExceptionHandlingName(unsigned EHK) {
3731   switch (static_cast<LangOptions::ExceptionHandlingKind>(EHK)) {
3732   case LangOptions::ExceptionHandlingKind::None:
3733     return "none";
3734   case LangOptions::ExceptionHandlingKind::DwarfCFI:
3735     return "dwarf";
3736   case LangOptions::ExceptionHandlingKind::SjLj:
3737     return "sjlj";
3738   case LangOptions::ExceptionHandlingKind::WinEH:
3739     return "seh";
3740   case LangOptions::ExceptionHandlingKind::Wasm:
3741     return "wasm";
3742   }
3743 
3744   llvm_unreachable("covered switch");
3745 }
3746 
GenerateLangArgs(const LangOptions & Opts,ArgumentConsumer Consumer,const llvm::Triple & T,InputKind IK)3747 void CompilerInvocationBase::GenerateLangArgs(const LangOptions &Opts,
3748                                               ArgumentConsumer Consumer,
3749                                               const llvm::Triple &T,
3750                                               InputKind IK) {
3751   if (IK.getFormat() == InputKind::Precompiled ||
3752       IK.getLanguage() == Language::LLVM_IR ||
3753       IK.getLanguage() == Language::CIR) {
3754     if (Opts.ObjCAutoRefCount)
3755       GenerateArg(Consumer, OPT_fobjc_arc);
3756     if (Opts.PICLevel != 0)
3757       GenerateArg(Consumer, OPT_pic_level, Twine(Opts.PICLevel));
3758     if (Opts.PIE)
3759       GenerateArg(Consumer, OPT_pic_is_pie);
3760     for (StringRef Sanitizer : serializeSanitizerKinds(Opts.Sanitize))
3761       GenerateArg(Consumer, OPT_fsanitize_EQ, Sanitizer);
3762     if (Opts.ExceptionHandling) {
3763       GenerateArg(Consumer, OPT_exception_model,
3764                   getExceptionHandlingName(Opts.ExceptionHandling));
3765     }
3766 
3767     return;
3768   }
3769 
3770   OptSpecifier StdOpt;
3771   switch (Opts.LangStd) {
3772   case LangStandard::lang_opencl10:
3773   case LangStandard::lang_opencl11:
3774   case LangStandard::lang_opencl12:
3775   case LangStandard::lang_opencl20:
3776   case LangStandard::lang_opencl30:
3777   case LangStandard::lang_openclcpp10:
3778   case LangStandard::lang_openclcpp2021:
3779     StdOpt = OPT_cl_std_EQ;
3780     break;
3781   default:
3782     StdOpt = OPT_std_EQ;
3783     break;
3784   }
3785 
3786   auto LangStandard = LangStandard::getLangStandardForKind(Opts.LangStd);
3787   GenerateArg(Consumer, StdOpt, LangStandard.getName());
3788 
3789   if (Opts.IncludeDefaultHeader)
3790     GenerateArg(Consumer, OPT_finclude_default_header);
3791   if (Opts.DeclareOpenCLBuiltins)
3792     GenerateArg(Consumer, OPT_fdeclare_opencl_builtins);
3793 
3794   const LangOptions *LangOpts = &Opts;
3795 
3796 #define LANG_OPTION_WITH_MARSHALLING(...)                                      \
3797   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
3798 #include "clang/Driver/Options.inc"
3799 #undef LANG_OPTION_WITH_MARSHALLING
3800 
3801   // The '-fcf-protection=' option is generated by CodeGenOpts generator.
3802 
3803   if (Opts.ObjC) {
3804     GenerateArg(Consumer, OPT_fobjc_runtime_EQ, Opts.ObjCRuntime.getAsString());
3805 
3806     if (Opts.GC == LangOptions::GCOnly)
3807       GenerateArg(Consumer, OPT_fobjc_gc_only);
3808     else if (Opts.GC == LangOptions::HybridGC)
3809       GenerateArg(Consumer, OPT_fobjc_gc);
3810     else if (Opts.ObjCAutoRefCount == 1)
3811       GenerateArg(Consumer, OPT_fobjc_arc);
3812 
3813     if (Opts.ObjCWeakRuntime)
3814       GenerateArg(Consumer, OPT_fobjc_runtime_has_weak);
3815 
3816     if (Opts.ObjCWeak)
3817       GenerateArg(Consumer, OPT_fobjc_weak);
3818 
3819     if (Opts.ObjCSubscriptingLegacyRuntime)
3820       GenerateArg(Consumer, OPT_fobjc_subscripting_legacy_runtime);
3821   }
3822 
3823   if (Opts.GNUCVersion != 0) {
3824     unsigned Major = Opts.GNUCVersion / 100 / 100;
3825     unsigned Minor = (Opts.GNUCVersion / 100) % 100;
3826     unsigned Patch = Opts.GNUCVersion % 100;
3827     GenerateArg(Consumer, OPT_fgnuc_version_EQ,
3828                 Twine(Major) + "." + Twine(Minor) + "." + Twine(Patch));
3829   }
3830 
3831   if (Opts.IgnoreXCOFFVisibility)
3832     GenerateArg(Consumer, OPT_mignore_xcoff_visibility);
3833 
3834   if (Opts.SignedOverflowBehavior == LangOptions::SOB_Trapping) {
3835     GenerateArg(Consumer, OPT_ftrapv);
3836     GenerateArg(Consumer, OPT_ftrapv_handler, Opts.OverflowHandler);
3837   } else if (Opts.SignedOverflowBehavior == LangOptions::SOB_Defined) {
3838     GenerateArg(Consumer, OPT_fwrapv);
3839   }
3840   if (Opts.PointerOverflowDefined)
3841     GenerateArg(Consumer, OPT_fwrapv_pointer);
3842 
3843   if (Opts.MSCompatibilityVersion != 0) {
3844     unsigned Major = Opts.MSCompatibilityVersion / 10000000;
3845     unsigned Minor = (Opts.MSCompatibilityVersion / 100000) % 100;
3846     unsigned Subminor = Opts.MSCompatibilityVersion % 100000;
3847     GenerateArg(Consumer, OPT_fms_compatibility_version,
3848                 Twine(Major) + "." + Twine(Minor) + "." + Twine(Subminor));
3849   }
3850 
3851   if ((!Opts.GNUMode && !Opts.MSVCCompat && !Opts.CPlusPlus17 && !Opts.C23) ||
3852       T.isOSzOS()) {
3853     if (!Opts.Trigraphs)
3854       GenerateArg(Consumer, OPT_fno_trigraphs);
3855   } else {
3856     if (Opts.Trigraphs)
3857       GenerateArg(Consumer, OPT_ftrigraphs);
3858   }
3859 
3860   if (T.isOSzOS() && !Opts.ZOSExt)
3861     GenerateArg(Consumer, OPT_fno_zos_extensions);
3862   else if (Opts.ZOSExt)
3863     GenerateArg(Consumer, OPT_fzos_extensions);
3864 
3865   if (Opts.Blocks && !(Opts.OpenCL && Opts.OpenCLVersion == 200))
3866     GenerateArg(Consumer, OPT_fblocks);
3867 
3868   if (Opts.ConvergentFunctions)
3869     GenerateArg(Consumer, OPT_fconvergent_functions);
3870   else
3871     GenerateArg(Consumer, OPT_fno_convergent_functions);
3872 
3873   if (Opts.NoBuiltin && !Opts.Freestanding)
3874     GenerateArg(Consumer, OPT_fno_builtin);
3875 
3876   if (!Opts.NoBuiltin)
3877     for (const auto &Func : Opts.NoBuiltinFuncs)
3878       GenerateArg(Consumer, OPT_fno_builtin_, Func);
3879 
3880   if (Opts.LongDoubleSize == 128)
3881     GenerateArg(Consumer, OPT_mlong_double_128);
3882   else if (Opts.LongDoubleSize == 64)
3883     GenerateArg(Consumer, OPT_mlong_double_64);
3884   else if (Opts.LongDoubleSize == 80)
3885     GenerateArg(Consumer, OPT_mlong_double_80);
3886 
3887   // Not generating '-mrtd', it's just an alias for '-fdefault-calling-conv='.
3888 
3889   // OpenMP was requested via '-fopenmp', not implied by '-fopenmp-simd' or
3890   // '-fopenmp-targets='.
3891   if (Opts.OpenMP && !Opts.OpenMPSimd) {
3892     GenerateArg(Consumer, OPT_fopenmp);
3893 
3894     if (Opts.OpenMP != 51)
3895       GenerateArg(Consumer, OPT_fopenmp_version_EQ, Twine(Opts.OpenMP));
3896 
3897     if (!Opts.OpenMPUseTLS)
3898       GenerateArg(Consumer, OPT_fnoopenmp_use_tls);
3899 
3900     if (Opts.OpenMPIsTargetDevice)
3901       GenerateArg(Consumer, OPT_fopenmp_is_target_device);
3902 
3903     if (Opts.OpenMPIRBuilder)
3904       GenerateArg(Consumer, OPT_fopenmp_enable_irbuilder);
3905   }
3906 
3907   if (Opts.OpenMPSimd) {
3908     GenerateArg(Consumer, OPT_fopenmp_simd);
3909 
3910     if (Opts.OpenMP != 51)
3911       GenerateArg(Consumer, OPT_fopenmp_version_EQ, Twine(Opts.OpenMP));
3912   }
3913 
3914   if (Opts.OpenMPThreadSubscription)
3915     GenerateArg(Consumer, OPT_fopenmp_assume_threads_oversubscription);
3916 
3917   if (Opts.OpenMPTeamSubscription)
3918     GenerateArg(Consumer, OPT_fopenmp_assume_teams_oversubscription);
3919 
3920   if (Opts.OpenMPTargetDebug != 0)
3921     GenerateArg(Consumer, OPT_fopenmp_target_debug_EQ,
3922                 Twine(Opts.OpenMPTargetDebug));
3923 
3924   if (Opts.OpenMPCUDANumSMs != 0)
3925     GenerateArg(Consumer, OPT_fopenmp_cuda_number_of_sm_EQ,
3926                 Twine(Opts.OpenMPCUDANumSMs));
3927 
3928   if (Opts.OpenMPCUDABlocksPerSM != 0)
3929     GenerateArg(Consumer, OPT_fopenmp_cuda_blocks_per_sm_EQ,
3930                 Twine(Opts.OpenMPCUDABlocksPerSM));
3931 
3932   if (Opts.OpenMPCUDAReductionBufNum != 1024)
3933     GenerateArg(Consumer, OPT_fopenmp_cuda_teams_reduction_recs_num_EQ,
3934                 Twine(Opts.OpenMPCUDAReductionBufNum));
3935 
3936   if (!Opts.OMPTargetTriples.empty()) {
3937     std::string Targets;
3938     llvm::raw_string_ostream OS(Targets);
3939     llvm::interleave(
3940         Opts.OMPTargetTriples, OS,
3941         [&OS](const llvm::Triple &T) { OS << T.str(); }, ",");
3942     GenerateArg(Consumer, OPT_offload_targets_EQ, Targets);
3943   }
3944 
3945   if (!Opts.OMPHostIRFile.empty())
3946     GenerateArg(Consumer, OPT_fopenmp_host_ir_file_path, Opts.OMPHostIRFile);
3947 
3948   if (Opts.OpenMPCUDAMode)
3949     GenerateArg(Consumer, OPT_fopenmp_cuda_mode);
3950 
3951   if (Opts.OpenACC) {
3952     GenerateArg(Consumer, OPT_fopenacc);
3953     if (!Opts.OpenACCMacroOverride.empty())
3954       GenerateArg(Consumer, OPT_openacc_macro_override,
3955                   Opts.OpenACCMacroOverride);
3956   }
3957 
3958   // The arguments used to set Optimize, OptimizeSize and NoInlineDefine are
3959   // generated from CodeGenOptions.
3960 
3961   if (Opts.DefaultFPContractMode == LangOptions::FPM_Fast)
3962     GenerateArg(Consumer, OPT_ffp_contract, "fast");
3963   else if (Opts.DefaultFPContractMode == LangOptions::FPM_On)
3964     GenerateArg(Consumer, OPT_ffp_contract, "on");
3965   else if (Opts.DefaultFPContractMode == LangOptions::FPM_Off)
3966     GenerateArg(Consumer, OPT_ffp_contract, "off");
3967   else if (Opts.DefaultFPContractMode == LangOptions::FPM_FastHonorPragmas)
3968     GenerateArg(Consumer, OPT_ffp_contract, "fast-honor-pragmas");
3969 
3970   for (StringRef Sanitizer : serializeSanitizerKinds(Opts.Sanitize))
3971     GenerateArg(Consumer, OPT_fsanitize_EQ, Sanitizer);
3972 
3973   // Conflating '-fsanitize-system-ignorelist' and '-fsanitize-ignorelist'.
3974   for (const std::string &F : Opts.NoSanitizeFiles)
3975     GenerateArg(Consumer, OPT_fsanitize_ignorelist_EQ, F);
3976 
3977   switch (Opts.getClangABICompat()) {
3978   case LangOptions::ClangABI::Ver3_8:
3979     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "3.8");
3980     break;
3981   case LangOptions::ClangABI::Ver4:
3982     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "4.0");
3983     break;
3984   case LangOptions::ClangABI::Ver6:
3985     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "6.0");
3986     break;
3987   case LangOptions::ClangABI::Ver7:
3988     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "7.0");
3989     break;
3990   case LangOptions::ClangABI::Ver9:
3991     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "9.0");
3992     break;
3993   case LangOptions::ClangABI::Ver11:
3994     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "11.0");
3995     break;
3996   case LangOptions::ClangABI::Ver12:
3997     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "12.0");
3998     break;
3999   case LangOptions::ClangABI::Ver14:
4000     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "14.0");
4001     break;
4002   case LangOptions::ClangABI::Ver15:
4003     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "15.0");
4004     break;
4005   case LangOptions::ClangABI::Ver17:
4006     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "17.0");
4007     break;
4008   case LangOptions::ClangABI::Ver18:
4009     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "18.0");
4010     break;
4011   case LangOptions::ClangABI::Ver19:
4012     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "19.0");
4013     break;
4014   case LangOptions::ClangABI::Ver20:
4015     GenerateArg(Consumer, OPT_fclang_abi_compat_EQ, "20.0");
4016     break;
4017   case LangOptions::ClangABI::Latest:
4018     break;
4019   }
4020 
4021   if (Opts.getSignReturnAddressScope() ==
4022       LangOptions::SignReturnAddressScopeKind::All)
4023     GenerateArg(Consumer, OPT_msign_return_address_EQ, "all");
4024   else if (Opts.getSignReturnAddressScope() ==
4025            LangOptions::SignReturnAddressScopeKind::NonLeaf)
4026     GenerateArg(Consumer, OPT_msign_return_address_EQ, "non-leaf");
4027 
4028   if (Opts.getSignReturnAddressKey() ==
4029       LangOptions::SignReturnAddressKeyKind::BKey)
4030     GenerateArg(Consumer, OPT_msign_return_address_key_EQ, "b_key");
4031 
4032   if (Opts.CXXABI)
4033     GenerateArg(Consumer, OPT_fcxx_abi_EQ,
4034                 TargetCXXABI::getSpelling(*Opts.CXXABI));
4035 
4036   if (Opts.RelativeCXXABIVTables)
4037     GenerateArg(Consumer, OPT_fexperimental_relative_cxx_abi_vtables);
4038   else
4039     GenerateArg(Consumer, OPT_fno_experimental_relative_cxx_abi_vtables);
4040 
4041   if (Opts.UseTargetPathSeparator)
4042     GenerateArg(Consumer, OPT_ffile_reproducible);
4043   else
4044     GenerateArg(Consumer, OPT_fno_file_reproducible);
4045 
4046   for (const auto &MP : Opts.MacroPrefixMap)
4047     GenerateArg(Consumer, OPT_fmacro_prefix_map_EQ, MP.first + "=" + MP.second);
4048 
4049   if (!Opts.RandstructSeed.empty())
4050     GenerateArg(Consumer, OPT_frandomize_layout_seed_EQ, Opts.RandstructSeed);
4051 }
4052 
ParseLangArgs(LangOptions & Opts,ArgList & Args,InputKind IK,const llvm::Triple & T,std::vector<std::string> & Includes,DiagnosticsEngine & Diags)4053 bool CompilerInvocation::ParseLangArgs(LangOptions &Opts, ArgList &Args,
4054                                        InputKind IK, const llvm::Triple &T,
4055                                        std::vector<std::string> &Includes,
4056                                        DiagnosticsEngine &Diags) {
4057   unsigned NumErrorsBefore = Diags.getNumErrors();
4058 
4059   if (IK.getFormat() == InputKind::Precompiled ||
4060       IK.getLanguage() == Language::LLVM_IR ||
4061       IK.getLanguage() == Language::CIR) {
4062     // ObjCAAutoRefCount and Sanitize LangOpts are used to setup the
4063     // PassManager in BackendUtil.cpp. They need to be initialized no matter
4064     // what the input type is.
4065     if (Args.hasArg(OPT_fobjc_arc))
4066       Opts.ObjCAutoRefCount = 1;
4067     // PICLevel and PIELevel are needed during code generation and this should
4068     // be set regardless of the input type.
4069     Opts.PICLevel = getLastArgIntValue(Args, OPT_pic_level, 0, Diags);
4070     Opts.PIE = Args.hasArg(OPT_pic_is_pie);
4071     parseSanitizerKinds("-fsanitize=", Args.getAllArgValues(OPT_fsanitize_EQ),
4072                         Diags, Opts.Sanitize);
4073 
4074     if (const Arg *A = Args.getLastArg(options::OPT_exception_model)) {
4075       std::optional<LangOptions::ExceptionHandlingKind> EMValue =
4076           llvm::StringSwitch<std::optional<LangOptions::ExceptionHandlingKind>>(
4077               A->getValue())
4078               .Case("dwarf", LangOptions::ExceptionHandlingKind::DwarfCFI)
4079               .Case("sjlj", LangOptions::ExceptionHandlingKind::SjLj)
4080               .Case("seh", LangOptions::ExceptionHandlingKind::WinEH)
4081               .Case("wasm", LangOptions::ExceptionHandlingKind::Wasm)
4082               .Case("none", LangOptions::ExceptionHandlingKind::None)
4083               .Default(std::nullopt);
4084       if (EMValue) {
4085         Opts.ExceptionHandling = static_cast<unsigned>(*EMValue);
4086       } else {
4087         Diags.Report(diag::err_drv_invalid_value)
4088             << A->getAsString(Args) << A->getValue();
4089       }
4090     }
4091 
4092     return Diags.getNumErrors() == NumErrorsBefore;
4093   }
4094 
4095   // Other LangOpts are only initialized when the input is not AST or LLVM IR.
4096   // FIXME: Should we really be parsing this for an Language::Asm input?
4097 
4098   // FIXME: Cleanup per-file based stuff.
4099   LangStandard::Kind LangStd = LangStandard::lang_unspecified;
4100   if (const Arg *A = Args.getLastArg(OPT_std_EQ)) {
4101     LangStd = LangStandard::getLangKind(A->getValue());
4102     if (LangStd == LangStandard::lang_unspecified) {
4103       Diags.Report(diag::err_drv_invalid_value)
4104         << A->getAsString(Args) << A->getValue();
4105       // Report supported standards with short description.
4106       for (unsigned KindValue = 0;
4107            KindValue != LangStandard::lang_unspecified;
4108            ++KindValue) {
4109         const LangStandard &Std = LangStandard::getLangStandardForKind(
4110           static_cast<LangStandard::Kind>(KindValue));
4111         if (IsInputCompatibleWithStandard(IK, Std)) {
4112           auto Diag = Diags.Report(diag::note_drv_use_standard);
4113           Diag << Std.getName() << Std.getDescription();
4114           unsigned NumAliases = 0;
4115 #define LANGSTANDARD(id, name, lang, desc, features)
4116 #define LANGSTANDARD_ALIAS(id, alias) \
4117           if (KindValue == LangStandard::lang_##id) ++NumAliases;
4118 #define LANGSTANDARD_ALIAS_DEPR(id, alias)
4119 #include "clang/Basic/LangStandards.def"
4120           Diag << NumAliases;
4121 #define LANGSTANDARD(id, name, lang, desc, features)
4122 #define LANGSTANDARD_ALIAS(id, alias) \
4123           if (KindValue == LangStandard::lang_##id) Diag << alias;
4124 #define LANGSTANDARD_ALIAS_DEPR(id, alias)
4125 #include "clang/Basic/LangStandards.def"
4126         }
4127       }
4128     } else {
4129       // Valid standard, check to make sure language and standard are
4130       // compatible.
4131       const LangStandard &Std = LangStandard::getLangStandardForKind(LangStd);
4132       if (!IsInputCompatibleWithStandard(IK, Std)) {
4133         Diags.Report(diag::err_drv_argument_not_allowed_with)
4134           << A->getAsString(Args) << GetInputKindName(IK);
4135       }
4136     }
4137   }
4138 
4139   // -cl-std only applies for OpenCL language standards.
4140   // Override the -std option in this case.
4141   if (const Arg *A = Args.getLastArg(OPT_cl_std_EQ)) {
4142     LangStandard::Kind OpenCLLangStd
4143       = llvm::StringSwitch<LangStandard::Kind>(A->getValue())
4144         .Cases("cl", "CL", LangStandard::lang_opencl10)
4145         .Cases("cl1.0", "CL1.0", LangStandard::lang_opencl10)
4146         .Cases("cl1.1", "CL1.1", LangStandard::lang_opencl11)
4147         .Cases("cl1.2", "CL1.2", LangStandard::lang_opencl12)
4148         .Cases("cl2.0", "CL2.0", LangStandard::lang_opencl20)
4149         .Cases("cl3.0", "CL3.0", LangStandard::lang_opencl30)
4150         .Cases("clc++", "CLC++", LangStandard::lang_openclcpp10)
4151         .Cases("clc++1.0", "CLC++1.0", LangStandard::lang_openclcpp10)
4152         .Cases("clc++2021", "CLC++2021", LangStandard::lang_openclcpp2021)
4153         .Default(LangStandard::lang_unspecified);
4154 
4155     if (OpenCLLangStd == LangStandard::lang_unspecified) {
4156       Diags.Report(diag::err_drv_invalid_value)
4157         << A->getAsString(Args) << A->getValue();
4158     }
4159     else
4160       LangStd = OpenCLLangStd;
4161   }
4162 
4163   // These need to be parsed now. They are used to set OpenCL defaults.
4164   Opts.IncludeDefaultHeader = Args.hasArg(OPT_finclude_default_header);
4165   Opts.DeclareOpenCLBuiltins = Args.hasArg(OPT_fdeclare_opencl_builtins);
4166 
4167   LangOptions::setLangDefaults(Opts, IK.getLanguage(), T, Includes, LangStd);
4168 
4169   // The key paths of codegen options defined in Options.td start with
4170   // "LangOpts->". Let's provide the expected variable name and type.
4171   LangOptions *LangOpts = &Opts;
4172 
4173 #define LANG_OPTION_WITH_MARSHALLING(...)                                      \
4174   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
4175 #include "clang/Driver/Options.inc"
4176 #undef LANG_OPTION_WITH_MARSHALLING
4177 
4178   if (const Arg *A = Args.getLastArg(OPT_fcf_protection_EQ)) {
4179     StringRef Name = A->getValue();
4180     if (Name == "full") {
4181       Opts.CFProtectionBranch = 1;
4182       Opts.CFProtectionReturn = 1;
4183     } else if (Name == "branch") {
4184       Opts.CFProtectionBranch = 1;
4185     } else if (Name == "return") {
4186       Opts.CFProtectionReturn = 1;
4187     }
4188   }
4189 
4190   if (Opts.CFProtectionBranch) {
4191     if (const Arg *A = Args.getLastArg(OPT_mcf_branch_label_scheme_EQ)) {
4192       const auto Scheme =
4193           llvm::StringSwitch<CFBranchLabelSchemeKind>(A->getValue())
4194 #define CF_BRANCH_LABEL_SCHEME(Kind, FlagVal)                                  \
4195   .Case(#FlagVal, CFBranchLabelSchemeKind::Kind)
4196 #include "clang/Basic/CFProtectionOptions.def"
4197               .Default(CFBranchLabelSchemeKind::Default);
4198       Opts.setCFBranchLabelScheme(Scheme);
4199     }
4200   }
4201 
4202   if ((Args.hasArg(OPT_fsycl_is_device) || Args.hasArg(OPT_fsycl_is_host)) &&
4203       !Args.hasArg(OPT_sycl_std_EQ)) {
4204     // If the user supplied -fsycl-is-device or -fsycl-is-host, but failed to
4205     // provide -sycl-std=, we want to default it to whatever the default SYCL
4206     // version is. I could not find a way to express this with the options
4207     // tablegen because we still want this value to be SYCL_None when the user
4208     // is not in device or host mode.
4209     Opts.setSYCLVersion(LangOptions::SYCL_Default);
4210   }
4211 
4212   if (Opts.ObjC) {
4213     if (Arg *arg = Args.getLastArg(OPT_fobjc_runtime_EQ)) {
4214       StringRef value = arg->getValue();
4215       if (Opts.ObjCRuntime.tryParse(value))
4216         Diags.Report(diag::err_drv_unknown_objc_runtime) << value;
4217     }
4218 
4219     if (Args.hasArg(OPT_fobjc_gc_only))
4220       Opts.setGC(LangOptions::GCOnly);
4221     else if (Args.hasArg(OPT_fobjc_gc))
4222       Opts.setGC(LangOptions::HybridGC);
4223     else if (Args.hasArg(OPT_fobjc_arc)) {
4224       Opts.ObjCAutoRefCount = 1;
4225       if (!Opts.ObjCRuntime.allowsARC())
4226         Diags.Report(diag::err_arc_unsupported_on_runtime);
4227     }
4228 
4229     // ObjCWeakRuntime tracks whether the runtime supports __weak, not
4230     // whether the feature is actually enabled.  This is predominantly
4231     // determined by -fobjc-runtime, but we allow it to be overridden
4232     // from the command line for testing purposes.
4233     if (Args.hasArg(OPT_fobjc_runtime_has_weak))
4234       Opts.ObjCWeakRuntime = 1;
4235     else
4236       Opts.ObjCWeakRuntime = Opts.ObjCRuntime.allowsWeak();
4237 
4238     // ObjCWeak determines whether __weak is actually enabled.
4239     // Note that we allow -fno-objc-weak to disable this even in ARC mode.
4240     if (auto weakArg = Args.getLastArg(OPT_fobjc_weak, OPT_fno_objc_weak)) {
4241       if (!weakArg->getOption().matches(OPT_fobjc_weak)) {
4242         assert(!Opts.ObjCWeak);
4243       } else if (Opts.getGC() != LangOptions::NonGC) {
4244         Diags.Report(diag::err_objc_weak_with_gc);
4245       } else if (!Opts.ObjCWeakRuntime) {
4246         Diags.Report(diag::err_objc_weak_unsupported);
4247       } else {
4248         Opts.ObjCWeak = 1;
4249       }
4250     } else if (Opts.ObjCAutoRefCount) {
4251       Opts.ObjCWeak = Opts.ObjCWeakRuntime;
4252     }
4253 
4254     if (Args.hasArg(OPT_fobjc_subscripting_legacy_runtime))
4255       Opts.ObjCSubscriptingLegacyRuntime =
4256         (Opts.ObjCRuntime.getKind() == ObjCRuntime::FragileMacOSX);
4257   }
4258 
4259   if (Arg *A = Args.getLastArg(options::OPT_fgnuc_version_EQ)) {
4260     // Check that the version has 1 to 3 components and the minor and patch
4261     // versions fit in two decimal digits.
4262     VersionTuple GNUCVer;
4263     bool Invalid = GNUCVer.tryParse(A->getValue());
4264     unsigned Major = GNUCVer.getMajor();
4265     unsigned Minor = GNUCVer.getMinor().value_or(0);
4266     unsigned Patch = GNUCVer.getSubminor().value_or(0);
4267     if (Invalid || GNUCVer.getBuild() || Minor >= 100 || Patch >= 100) {
4268       Diags.Report(diag::err_drv_invalid_value)
4269           << A->getAsString(Args) << A->getValue();
4270     }
4271     Opts.GNUCVersion = Major * 100 * 100 + Minor * 100 + Patch;
4272   }
4273 
4274   if (T.isOSAIX() && (Args.hasArg(OPT_mignore_xcoff_visibility)))
4275     Opts.IgnoreXCOFFVisibility = 1;
4276 
4277   if (Args.hasArg(OPT_ftrapv)) {
4278     Opts.setSignedOverflowBehavior(LangOptions::SOB_Trapping);
4279     // Set the handler, if one is specified.
4280     Opts.OverflowHandler =
4281         std::string(Args.getLastArgValue(OPT_ftrapv_handler));
4282   }
4283   else if (Args.hasArg(OPT_fwrapv))
4284     Opts.setSignedOverflowBehavior(LangOptions::SOB_Defined);
4285   if (Args.hasArg(OPT_fwrapv_pointer))
4286     Opts.PointerOverflowDefined = true;
4287 
4288   Opts.MSCompatibilityVersion = 0;
4289   if (const Arg *A = Args.getLastArg(OPT_fms_compatibility_version)) {
4290     VersionTuple VT;
4291     if (VT.tryParse(A->getValue()))
4292       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args)
4293                                                 << A->getValue();
4294     Opts.MSCompatibilityVersion = VT.getMajor() * 10000000 +
4295                                   VT.getMinor().value_or(0) * 100000 +
4296                                   VT.getSubminor().value_or(0);
4297   }
4298 
4299   // Mimicking gcc's behavior, trigraphs are only enabled if -trigraphs
4300   // is specified, or -std is set to a conforming mode.
4301   // Trigraphs are disabled by default in C++17 and C23 onwards.
4302   // For z/OS, trigraphs are enabled by default (without regard to the above).
4303   Opts.Trigraphs =
4304       (!Opts.GNUMode && !Opts.MSVCCompat && !Opts.CPlusPlus17 && !Opts.C23) ||
4305       T.isOSzOS();
4306   Opts.Trigraphs =
4307       Args.hasFlag(OPT_ftrigraphs, OPT_fno_trigraphs, Opts.Trigraphs);
4308 
4309   Opts.ZOSExt =
4310       Args.hasFlag(OPT_fzos_extensions, OPT_fno_zos_extensions, T.isOSzOS());
4311 
4312   Opts.Blocks = Args.hasArg(OPT_fblocks) || (Opts.OpenCL
4313     && Opts.OpenCLVersion == 200);
4314 
4315   bool HasConvergentOperations = Opts.isTargetDevice() || Opts.OpenCL ||
4316                                  Opts.HLSL || T.isAMDGPU() || T.isNVPTX();
4317   Opts.ConvergentFunctions =
4318       Args.hasFlag(OPT_fconvergent_functions, OPT_fno_convergent_functions,
4319                    HasConvergentOperations);
4320 
4321   Opts.NoBuiltin = Args.hasArg(OPT_fno_builtin) || Opts.Freestanding;
4322   if (!Opts.NoBuiltin)
4323     getAllNoBuiltinFuncValues(Args, Opts.NoBuiltinFuncs);
4324   if (Arg *A = Args.getLastArg(options::OPT_LongDouble_Group)) {
4325     if (A->getOption().matches(options::OPT_mlong_double_64))
4326       Opts.LongDoubleSize = 64;
4327     else if (A->getOption().matches(options::OPT_mlong_double_80))
4328       Opts.LongDoubleSize = 80;
4329     else if (A->getOption().matches(options::OPT_mlong_double_128))
4330       Opts.LongDoubleSize = 128;
4331     else
4332       Opts.LongDoubleSize = 0;
4333   }
4334   if (Opts.FastRelaxedMath || Opts.CLUnsafeMath)
4335     Opts.setDefaultFPContractMode(LangOptions::FPM_Fast);
4336 
4337   llvm::sort(Opts.ModuleFeatures);
4338 
4339   // -mrtd option
4340   if (Arg *A = Args.getLastArg(OPT_mrtd)) {
4341     if (Opts.getDefaultCallingConv() != LangOptions::DCC_None)
4342       Diags.Report(diag::err_drv_argument_not_allowed_with)
4343           << A->getSpelling() << "-fdefault-calling-conv";
4344     else {
4345       switch (T.getArch()) {
4346       case llvm::Triple::x86:
4347         Opts.setDefaultCallingConv(LangOptions::DCC_StdCall);
4348         break;
4349       case llvm::Triple::m68k:
4350         Opts.setDefaultCallingConv(LangOptions::DCC_RtdCall);
4351         break;
4352       default:
4353         Diags.Report(diag::err_drv_argument_not_allowed_with)
4354             << A->getSpelling() << T.getTriple();
4355       }
4356     }
4357   }
4358 
4359   // Check if -fopenmp is specified and set default version to 5.1.
4360   Opts.OpenMP = Args.hasArg(OPT_fopenmp) ? 51 : 0;
4361   // Check if -fopenmp-simd is specified.
4362   bool IsSimdSpecified =
4363       Args.hasFlag(options::OPT_fopenmp_simd, options::OPT_fno_openmp_simd,
4364                    /*Default=*/false);
4365   Opts.OpenMPSimd = !Opts.OpenMP && IsSimdSpecified;
4366   Opts.OpenMPUseTLS =
4367       Opts.OpenMP && !Args.hasArg(options::OPT_fnoopenmp_use_tls);
4368   Opts.OpenMPIsTargetDevice =
4369       Opts.OpenMP && Args.hasArg(options::OPT_fopenmp_is_target_device);
4370   Opts.OpenMPIRBuilder =
4371       Opts.OpenMP && Args.hasArg(options::OPT_fopenmp_enable_irbuilder);
4372   bool IsTargetSpecified =
4373       Opts.OpenMPIsTargetDevice || Args.hasArg(options::OPT_offload_targets_EQ);
4374 
4375   if (Opts.OpenMP || Opts.OpenMPSimd) {
4376     if (int Version = getLastArgIntValue(
4377             Args, OPT_fopenmp_version_EQ,
4378             (IsSimdSpecified || IsTargetSpecified) ? 51 : Opts.OpenMP, Diags))
4379       Opts.OpenMP = Version;
4380     // Provide diagnostic when a given target is not expected to be an OpenMP
4381     // device or host.
4382     if (!Opts.OpenMPIsTargetDevice) {
4383       switch (T.getArch()) {
4384       default:
4385         break;
4386       // Add unsupported host targets here:
4387       case llvm::Triple::nvptx:
4388       case llvm::Triple::nvptx64:
4389         Diags.Report(diag::err_drv_omp_host_target_not_supported) << T.str();
4390         break;
4391       }
4392     }
4393   }
4394 
4395   // Set the flag to prevent the implementation from emitting device exception
4396   // handling code for those requiring so.
4397   if ((Opts.OpenMPIsTargetDevice && (T.isNVPTX() || T.isAMDGCN())) ||
4398       Opts.OpenCLCPlusPlus) {
4399 
4400     Opts.Exceptions = 0;
4401     Opts.CXXExceptions = 0;
4402   }
4403   if (Opts.OpenMPIsTargetDevice && T.isNVPTX()) {
4404     Opts.OpenMPCUDANumSMs =
4405         getLastArgIntValue(Args, options::OPT_fopenmp_cuda_number_of_sm_EQ,
4406                            Opts.OpenMPCUDANumSMs, Diags);
4407     Opts.OpenMPCUDABlocksPerSM =
4408         getLastArgIntValue(Args, options::OPT_fopenmp_cuda_blocks_per_sm_EQ,
4409                            Opts.OpenMPCUDABlocksPerSM, Diags);
4410     Opts.OpenMPCUDAReductionBufNum = getLastArgIntValue(
4411         Args, options::OPT_fopenmp_cuda_teams_reduction_recs_num_EQ,
4412         Opts.OpenMPCUDAReductionBufNum, Diags);
4413   }
4414 
4415   // Set the value of the debugging flag used in the new offloading device RTL.
4416   // Set either by a specific value or to a default if not specified.
4417   if (Opts.OpenMPIsTargetDevice && (Args.hasArg(OPT_fopenmp_target_debug) ||
4418                                     Args.hasArg(OPT_fopenmp_target_debug_EQ))) {
4419     Opts.OpenMPTargetDebug = getLastArgIntValue(
4420         Args, OPT_fopenmp_target_debug_EQ, Opts.OpenMPTargetDebug, Diags);
4421     if (!Opts.OpenMPTargetDebug && Args.hasArg(OPT_fopenmp_target_debug))
4422       Opts.OpenMPTargetDebug = 1;
4423   }
4424 
4425   if (Opts.OpenMPIsTargetDevice) {
4426     if (Args.hasArg(OPT_fopenmp_assume_teams_oversubscription))
4427       Opts.OpenMPTeamSubscription = true;
4428     if (Args.hasArg(OPT_fopenmp_assume_threads_oversubscription))
4429       Opts.OpenMPThreadSubscription = true;
4430   }
4431 
4432   // Get the OpenMP target triples if any.
4433   if (Arg *A = Args.getLastArg(options::OPT_offload_targets_EQ)) {
4434     enum ArchPtrSize { Arch16Bit, Arch32Bit, Arch64Bit };
4435     auto getArchPtrSize = [](const llvm::Triple &T) {
4436       if (T.isArch16Bit())
4437         return Arch16Bit;
4438       if (T.isArch32Bit())
4439         return Arch32Bit;
4440       assert(T.isArch64Bit() && "Expected 64-bit architecture");
4441       return Arch64Bit;
4442     };
4443 
4444     for (unsigned i = 0; i < A->getNumValues(); ++i) {
4445       llvm::Triple TT(A->getValue(i));
4446 
4447       if (TT.getArch() == llvm::Triple::UnknownArch ||
4448           !(TT.getArch() == llvm::Triple::aarch64 || TT.isPPC() ||
4449             TT.getArch() == llvm::Triple::spirv64 ||
4450             TT.getArch() == llvm::Triple::systemz ||
4451             TT.getArch() == llvm::Triple::loongarch64 ||
4452             TT.getArch() == llvm::Triple::nvptx ||
4453             TT.getArch() == llvm::Triple::nvptx64 || TT.isAMDGCN() ||
4454             TT.getArch() == llvm::Triple::x86 ||
4455             TT.getArch() == llvm::Triple::x86_64))
4456         Diags.Report(diag::err_drv_invalid_omp_target) << A->getValue(i);
4457       else if (getArchPtrSize(T) != getArchPtrSize(TT))
4458         Diags.Report(diag::err_drv_incompatible_omp_arch)
4459             << A->getValue(i) << T.str();
4460       else
4461         Opts.OMPTargetTriples.push_back(TT);
4462     }
4463   }
4464 
4465   // Get OpenMP host file path if any and report if a non existent file is
4466   // found
4467   if (Arg *A = Args.getLastArg(options::OPT_fopenmp_host_ir_file_path)) {
4468     Opts.OMPHostIRFile = A->getValue();
4469     if (!llvm::sys::fs::exists(Opts.OMPHostIRFile))
4470       Diags.Report(diag::err_drv_omp_host_ir_file_not_found)
4471           << Opts.OMPHostIRFile;
4472   }
4473 
4474   // Set CUDA mode for OpenMP target NVPTX/AMDGCN if specified in options
4475   Opts.OpenMPCUDAMode = Opts.OpenMPIsTargetDevice &&
4476                         (T.isNVPTX() || T.isAMDGCN()) &&
4477                         Args.hasArg(options::OPT_fopenmp_cuda_mode);
4478 
4479   // OpenACC Configuration.
4480   if (Args.hasArg(options::OPT_fopenacc)) {
4481     Opts.OpenACC = true;
4482 
4483     if (Arg *A = Args.getLastArg(options::OPT_openacc_macro_override))
4484       Opts.OpenACCMacroOverride = A->getValue();
4485   }
4486 
4487   // FIXME: Eliminate this dependency.
4488   unsigned Opt = getOptimizationLevel(Args, IK, Diags),
4489        OptSize = getOptimizationLevelSize(Args);
4490   Opts.Optimize = Opt != 0;
4491   Opts.OptimizeSize = OptSize != 0;
4492 
4493   // This is the __NO_INLINE__ define, which just depends on things like the
4494   // optimization level and -fno-inline, not actually whether the backend has
4495   // inlining enabled.
4496   Opts.NoInlineDefine = !Opts.Optimize;
4497   if (Arg *InlineArg = Args.getLastArg(
4498           options::OPT_finline_functions, options::OPT_finline_hint_functions,
4499           options::OPT_fno_inline_functions, options::OPT_fno_inline))
4500     if (InlineArg->getOption().matches(options::OPT_fno_inline))
4501       Opts.NoInlineDefine = true;
4502 
4503   if (Arg *A = Args.getLastArg(OPT_ffp_contract)) {
4504     StringRef Val = A->getValue();
4505     if (Val == "fast")
4506       Opts.setDefaultFPContractMode(LangOptions::FPM_Fast);
4507     else if (Val == "on")
4508       Opts.setDefaultFPContractMode(LangOptions::FPM_On);
4509     else if (Val == "off")
4510       Opts.setDefaultFPContractMode(LangOptions::FPM_Off);
4511     else if (Val == "fast-honor-pragmas")
4512       Opts.setDefaultFPContractMode(LangOptions::FPM_FastHonorPragmas);
4513     else
4514       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Val;
4515   }
4516 
4517   if (auto *A =
4518           Args.getLastArg(OPT_fsanitize_undefined_ignore_overflow_pattern_EQ)) {
4519     for (int i = 0, n = A->getNumValues(); i != n; ++i) {
4520       Opts.OverflowPatternExclusionMask |=
4521           llvm::StringSwitch<unsigned>(A->getValue(i))
4522               .Case("none", LangOptionsBase::None)
4523               .Case("all", LangOptionsBase::All)
4524               .Case("add-unsigned-overflow-test",
4525                     LangOptionsBase::AddUnsignedOverflowTest)
4526               .Case("add-signed-overflow-test",
4527                     LangOptionsBase::AddSignedOverflowTest)
4528               .Case("negated-unsigned-const", LangOptionsBase::NegUnsignedConst)
4529               .Case("unsigned-post-decr-while",
4530                     LangOptionsBase::PostDecrInWhile)
4531               .Default(0);
4532     }
4533   }
4534 
4535   // Parse -fsanitize= arguments.
4536   parseSanitizerKinds("-fsanitize=", Args.getAllArgValues(OPT_fsanitize_EQ),
4537                       Diags, Opts.Sanitize);
4538   Opts.NoSanitizeFiles = Args.getAllArgValues(OPT_fsanitize_ignorelist_EQ);
4539   std::vector<std::string> systemIgnorelists =
4540       Args.getAllArgValues(OPT_fsanitize_system_ignorelist_EQ);
4541   Opts.NoSanitizeFiles.insert(Opts.NoSanitizeFiles.end(),
4542                               systemIgnorelists.begin(),
4543                               systemIgnorelists.end());
4544 
4545   if (Arg *A = Args.getLastArg(OPT_fclang_abi_compat_EQ)) {
4546     Opts.setClangABICompat(LangOptions::ClangABI::Latest);
4547 
4548     StringRef Ver = A->getValue();
4549     std::pair<StringRef, StringRef> VerParts = Ver.split('.');
4550     unsigned Major, Minor = 0;
4551 
4552     // Check the version number is valid: either 3.x (0 <= x <= 9) or
4553     // y or y.0 (4 <= y <= current version).
4554     if (!VerParts.first.starts_with("0") &&
4555         !VerParts.first.getAsInteger(10, Major) && 3 <= Major &&
4556         Major <= MAX_CLANG_ABI_COMPAT_VERSION &&
4557         (Major == 3
4558              ? VerParts.second.size() == 1 &&
4559                    !VerParts.second.getAsInteger(10, Minor)
4560              : VerParts.first.size() == Ver.size() || VerParts.second == "0")) {
4561       // Got a valid version number.
4562       if (Major == 3 && Minor <= 8)
4563         Opts.setClangABICompat(LangOptions::ClangABI::Ver3_8);
4564       else if (Major <= 4)
4565         Opts.setClangABICompat(LangOptions::ClangABI::Ver4);
4566       else if (Major <= 6)
4567         Opts.setClangABICompat(LangOptions::ClangABI::Ver6);
4568       else if (Major <= 7)
4569         Opts.setClangABICompat(LangOptions::ClangABI::Ver7);
4570       else if (Major <= 9)
4571         Opts.setClangABICompat(LangOptions::ClangABI::Ver9);
4572       else if (Major <= 11)
4573         Opts.setClangABICompat(LangOptions::ClangABI::Ver11);
4574       else if (Major <= 12)
4575         Opts.setClangABICompat(LangOptions::ClangABI::Ver12);
4576       else if (Major <= 14)
4577         Opts.setClangABICompat(LangOptions::ClangABI::Ver14);
4578       else if (Major <= 15)
4579         Opts.setClangABICompat(LangOptions::ClangABI::Ver15);
4580       else if (Major <= 17)
4581         Opts.setClangABICompat(LangOptions::ClangABI::Ver17);
4582       else if (Major <= 18)
4583         Opts.setClangABICompat(LangOptions::ClangABI::Ver18);
4584       else if (Major <= 19)
4585         Opts.setClangABICompat(LangOptions::ClangABI::Ver19);
4586       else if (Major <= 20)
4587         Opts.setClangABICompat(LangOptions::ClangABI::Ver20);
4588     } else if (Ver != "latest") {
4589       Diags.Report(diag::err_drv_invalid_value)
4590           << A->getAsString(Args) << A->getValue();
4591     }
4592   }
4593 
4594   if (Arg *A = Args.getLastArg(OPT_msign_return_address_EQ)) {
4595     StringRef SignScope = A->getValue();
4596 
4597     if (SignScope.equals_insensitive("none"))
4598       Opts.setSignReturnAddressScope(
4599           LangOptions::SignReturnAddressScopeKind::None);
4600     else if (SignScope.equals_insensitive("all"))
4601       Opts.setSignReturnAddressScope(
4602           LangOptions::SignReturnAddressScopeKind::All);
4603     else if (SignScope.equals_insensitive("non-leaf"))
4604       Opts.setSignReturnAddressScope(
4605           LangOptions::SignReturnAddressScopeKind::NonLeaf);
4606     else
4607       Diags.Report(diag::err_drv_invalid_value)
4608           << A->getAsString(Args) << SignScope;
4609 
4610     if (Arg *A = Args.getLastArg(OPT_msign_return_address_key_EQ)) {
4611       StringRef SignKey = A->getValue();
4612       if (!SignScope.empty() && !SignKey.empty()) {
4613         if (SignKey == "a_key")
4614           Opts.setSignReturnAddressKey(
4615               LangOptions::SignReturnAddressKeyKind::AKey);
4616         else if (SignKey == "b_key")
4617           Opts.setSignReturnAddressKey(
4618               LangOptions::SignReturnAddressKeyKind::BKey);
4619         else
4620           Diags.Report(diag::err_drv_invalid_value)
4621               << A->getAsString(Args) << SignKey;
4622       }
4623     }
4624   }
4625 
4626   // The value can be empty, which indicates the system default should be used.
4627   StringRef CXXABI = Args.getLastArgValue(OPT_fcxx_abi_EQ);
4628   if (!CXXABI.empty()) {
4629     if (!TargetCXXABI::isABI(CXXABI)) {
4630       Diags.Report(diag::err_invalid_cxx_abi) << CXXABI;
4631     } else {
4632       auto Kind = TargetCXXABI::getKind(CXXABI);
4633       if (!TargetCXXABI::isSupportedCXXABI(T, Kind))
4634         Diags.Report(diag::err_unsupported_cxx_abi) << CXXABI << T.str();
4635       else
4636         Opts.CXXABI = Kind;
4637     }
4638   }
4639 
4640   Opts.RelativeCXXABIVTables =
4641       Args.hasFlag(options::OPT_fexperimental_relative_cxx_abi_vtables,
4642                    options::OPT_fno_experimental_relative_cxx_abi_vtables,
4643                    TargetCXXABI::usesRelativeVTables(T));
4644 
4645   // RTTI is on by default.
4646   bool HasRTTI = !Args.hasArg(options::OPT_fno_rtti);
4647   Opts.OmitVTableRTTI =
4648       Args.hasFlag(options::OPT_fexperimental_omit_vtable_rtti,
4649                    options::OPT_fno_experimental_omit_vtable_rtti, false);
4650   if (Opts.OmitVTableRTTI && HasRTTI)
4651     Diags.Report(diag::err_drv_using_omit_rtti_component_without_no_rtti);
4652 
4653   for (const auto &A : Args.getAllArgValues(OPT_fmacro_prefix_map_EQ)) {
4654     auto Split = StringRef(A).split('=');
4655     Opts.MacroPrefixMap.insert(
4656         {std::string(Split.first), std::string(Split.second)});
4657   }
4658 
4659   Opts.UseTargetPathSeparator =
4660       !Args.getLastArg(OPT_fno_file_reproducible) &&
4661       (Args.getLastArg(OPT_ffile_compilation_dir_EQ) ||
4662        Args.getLastArg(OPT_fmacro_prefix_map_EQ) ||
4663        Args.getLastArg(OPT_ffile_reproducible));
4664 
4665   // Error if -mvscale-min is unbounded.
4666   if (Arg *A = Args.getLastArg(options::OPT_mvscale_min_EQ)) {
4667     unsigned VScaleMin;
4668     if (StringRef(A->getValue()).getAsInteger(10, VScaleMin) || VScaleMin == 0)
4669       Diags.Report(diag::err_cc1_unbounded_vscale_min);
4670   }
4671   if (Arg *A = Args.getLastArg(options::OPT_mvscale_streaming_min_EQ)) {
4672     unsigned VScaleMin;
4673     if (StringRef(A->getValue()).getAsInteger(10, VScaleMin) || VScaleMin == 0)
4674       Diags.Report(diag::err_cc1_unbounded_vscale_min);
4675   }
4676 
4677   if (const Arg *A = Args.getLastArg(OPT_frandomize_layout_seed_file_EQ)) {
4678     std::ifstream SeedFile(A->getValue(0));
4679 
4680     if (!SeedFile.is_open())
4681       Diags.Report(diag::err_drv_cannot_open_randomize_layout_seed_file)
4682           << A->getValue(0);
4683 
4684     std::getline(SeedFile, Opts.RandstructSeed);
4685   }
4686 
4687   if (const Arg *A = Args.getLastArg(OPT_frandomize_layout_seed_EQ))
4688     Opts.RandstructSeed = A->getValue(0);
4689 
4690   // Validate options for HLSL
4691   if (Opts.HLSL) {
4692     // TODO: Revisit restricting SPIR-V to logical once we've figured out how to
4693     // handle PhysicalStorageBuffer64 memory model
4694     if (T.isDXIL() || T.isSPIRVLogical()) {
4695       enum { ShaderModel, VulkanEnv, ShaderStage };
4696       enum { OS, Environment };
4697 
4698       int ExpectedOS = T.isSPIRVLogical() ? VulkanEnv : ShaderModel;
4699 
4700       if (T.getOSName().empty()) {
4701         Diags.Report(diag::err_drv_hlsl_bad_shader_required_in_target)
4702             << ExpectedOS << OS << T.str();
4703       } else if (T.getEnvironmentName().empty()) {
4704         Diags.Report(diag::err_drv_hlsl_bad_shader_required_in_target)
4705             << ShaderStage << Environment << T.str();
4706       } else if (!T.isShaderStageEnvironment()) {
4707         Diags.Report(diag::err_drv_hlsl_bad_shader_unsupported)
4708             << ShaderStage << T.getEnvironmentName() << T.str();
4709       }
4710 
4711       if (T.isDXIL()) {
4712         if (!T.isShaderModelOS() || T.getOSVersion() == VersionTuple(0)) {
4713           Diags.Report(diag::err_drv_hlsl_bad_shader_unsupported)
4714               << ShaderModel << T.getOSName() << T.str();
4715         }
4716         // Validate that if fnative-half-type is given, that
4717         // the language standard is at least hlsl2018, and that
4718         // the target shader model is at least 6.2.
4719         if (Args.getLastArg(OPT_fnative_half_type)) {
4720           const LangStandard &Std =
4721               LangStandard::getLangStandardForKind(Opts.LangStd);
4722           if (!(Opts.LangStd >= LangStandard::lang_hlsl2018 &&
4723                 T.getOSVersion() >= VersionTuple(6, 2)))
4724             Diags.Report(diag::err_drv_hlsl_16bit_types_unsupported)
4725                 << "-enable-16bit-types" << true << Std.getName()
4726                 << T.getOSVersion().getAsString();
4727         }
4728       } else if (T.isSPIRVLogical()) {
4729         if (!T.isVulkanOS() || T.getVulkanVersion() == VersionTuple(0)) {
4730           Diags.Report(diag::err_drv_hlsl_bad_shader_unsupported)
4731               << VulkanEnv << T.getOSName() << T.str();
4732         }
4733         if (Args.getLastArg(OPT_fnative_half_type)) {
4734           const LangStandard &Std =
4735               LangStandard::getLangStandardForKind(Opts.LangStd);
4736           if (!(Opts.LangStd >= LangStandard::lang_hlsl2018))
4737             Diags.Report(diag::err_drv_hlsl_16bit_types_unsupported)
4738                 << "-fnative-half-type" << false << Std.getName();
4739         }
4740       } else {
4741         llvm_unreachable("expected DXIL or SPIR-V target");
4742       }
4743     } else
4744       Diags.Report(diag::err_drv_hlsl_unsupported_target) << T.str();
4745 
4746     if (Opts.LangStd < LangStandard::lang_hlsl202x) {
4747       const LangStandard &Requested =
4748           LangStandard::getLangStandardForKind(Opts.LangStd);
4749       const LangStandard &Recommended =
4750           LangStandard::getLangStandardForKind(LangStandard::lang_hlsl202x);
4751       Diags.Report(diag::warn_hlsl_langstd_minimal)
4752           << Requested.getName() << Recommended.getName();
4753     }
4754   }
4755 
4756   return Diags.getNumErrors() == NumErrorsBefore;
4757 }
4758 
isStrictlyPreprocessorAction(frontend::ActionKind Action)4759 static bool isStrictlyPreprocessorAction(frontend::ActionKind Action) {
4760   switch (Action) {
4761   case frontend::ASTDeclList:
4762   case frontend::ASTDump:
4763   case frontend::ASTPrint:
4764   case frontend::ASTView:
4765   case frontend::EmitAssembly:
4766   case frontend::EmitBC:
4767   case frontend::EmitCIR:
4768   case frontend::EmitHTML:
4769   case frontend::EmitLLVM:
4770   case frontend::EmitLLVMOnly:
4771   case frontend::EmitCodeGenOnly:
4772   case frontend::EmitObj:
4773   case frontend::ExtractAPI:
4774   case frontend::FixIt:
4775   case frontend::GenerateModule:
4776   case frontend::GenerateModuleInterface:
4777   case frontend::GenerateReducedModuleInterface:
4778   case frontend::GenerateHeaderUnit:
4779   case frontend::GeneratePCH:
4780   case frontend::GenerateInterfaceStubs:
4781   case frontend::ParseSyntaxOnly:
4782   case frontend::ModuleFileInfo:
4783   case frontend::VerifyPCH:
4784   case frontend::PluginAction:
4785   case frontend::RewriteObjC:
4786   case frontend::RewriteTest:
4787   case frontend::RunAnalysis:
4788   case frontend::TemplightDump:
4789     return false;
4790 
4791   case frontend::DumpCompilerOptions:
4792   case frontend::DumpRawTokens:
4793   case frontend::DumpTokens:
4794   case frontend::InitOnly:
4795   case frontend::PrintPreamble:
4796   case frontend::PrintPreprocessedInput:
4797   case frontend::RewriteMacros:
4798   case frontend::RunPreprocessorOnly:
4799   case frontend::PrintDependencyDirectivesSourceMinimizerOutput:
4800     return true;
4801   }
4802   llvm_unreachable("invalid frontend action");
4803 }
4804 
isCodeGenAction(frontend::ActionKind Action)4805 static bool isCodeGenAction(frontend::ActionKind Action) {
4806   switch (Action) {
4807   case frontend::EmitAssembly:
4808   case frontend::EmitBC:
4809   case frontend::EmitCIR:
4810   case frontend::EmitHTML:
4811   case frontend::EmitLLVM:
4812   case frontend::EmitLLVMOnly:
4813   case frontend::EmitCodeGenOnly:
4814   case frontend::EmitObj:
4815   case frontend::GenerateModule:
4816   case frontend::GenerateModuleInterface:
4817   case frontend::GenerateReducedModuleInterface:
4818   case frontend::GenerateHeaderUnit:
4819   case frontend::GeneratePCH:
4820   case frontend::GenerateInterfaceStubs:
4821     return true;
4822   case frontend::ASTDeclList:
4823   case frontend::ASTDump:
4824   case frontend::ASTPrint:
4825   case frontend::ASTView:
4826   case frontend::ExtractAPI:
4827   case frontend::FixIt:
4828   case frontend::ParseSyntaxOnly:
4829   case frontend::ModuleFileInfo:
4830   case frontend::VerifyPCH:
4831   case frontend::PluginAction:
4832   case frontend::RewriteObjC:
4833   case frontend::RewriteTest:
4834   case frontend::RunAnalysis:
4835   case frontend::TemplightDump:
4836   case frontend::DumpCompilerOptions:
4837   case frontend::DumpRawTokens:
4838   case frontend::DumpTokens:
4839   case frontend::InitOnly:
4840   case frontend::PrintPreamble:
4841   case frontend::PrintPreprocessedInput:
4842   case frontend::RewriteMacros:
4843   case frontend::RunPreprocessorOnly:
4844   case frontend::PrintDependencyDirectivesSourceMinimizerOutput:
4845     return false;
4846   }
4847   llvm_unreachable("invalid frontend action");
4848 }
4849 
GeneratePreprocessorArgs(const PreprocessorOptions & Opts,ArgumentConsumer Consumer,const LangOptions & LangOpts,const FrontendOptions & FrontendOpts,const CodeGenOptions & CodeGenOpts)4850 static void GeneratePreprocessorArgs(const PreprocessorOptions &Opts,
4851                                      ArgumentConsumer Consumer,
4852                                      const LangOptions &LangOpts,
4853                                      const FrontendOptions &FrontendOpts,
4854                                      const CodeGenOptions &CodeGenOpts) {
4855   const PreprocessorOptions *PreprocessorOpts = &Opts;
4856 
4857 #define PREPROCESSOR_OPTION_WITH_MARSHALLING(...)                              \
4858   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
4859 #include "clang/Driver/Options.inc"
4860 #undef PREPROCESSOR_OPTION_WITH_MARSHALLING
4861 
4862   if (Opts.PCHWithHdrStop && !Opts.PCHWithHdrStopCreate)
4863     GenerateArg(Consumer, OPT_pch_through_hdrstop_use);
4864 
4865   for (const auto &D : Opts.DeserializedPCHDeclsToErrorOn)
4866     GenerateArg(Consumer, OPT_error_on_deserialized_pch_decl, D);
4867 
4868   if (Opts.PrecompiledPreambleBytes != std::make_pair(0u, false))
4869     GenerateArg(Consumer, OPT_preamble_bytes_EQ,
4870                 Twine(Opts.PrecompiledPreambleBytes.first) + "," +
4871                     (Opts.PrecompiledPreambleBytes.second ? "1" : "0"));
4872 
4873   for (const auto &M : Opts.Macros) {
4874     // Don't generate __CET__ macro definitions. They are implied by the
4875     // -fcf-protection option that is generated elsewhere.
4876     if (M.first == "__CET__=1" && !M.second &&
4877         !CodeGenOpts.CFProtectionReturn && CodeGenOpts.CFProtectionBranch)
4878       continue;
4879     if (M.first == "__CET__=2" && !M.second && CodeGenOpts.CFProtectionReturn &&
4880         !CodeGenOpts.CFProtectionBranch)
4881       continue;
4882     if (M.first == "__CET__=3" && !M.second && CodeGenOpts.CFProtectionReturn &&
4883         CodeGenOpts.CFProtectionBranch)
4884       continue;
4885 
4886     GenerateArg(Consumer, M.second ? OPT_U : OPT_D, M.first);
4887   }
4888 
4889   for (const auto &I : Opts.Includes) {
4890     // Don't generate OpenCL includes. They are implied by other flags that are
4891     // generated elsewhere.
4892     if (LangOpts.OpenCL && LangOpts.IncludeDefaultHeader &&
4893         ((LangOpts.DeclareOpenCLBuiltins && I == "opencl-c-base.h") ||
4894          I == "opencl-c.h"))
4895       continue;
4896     // Don't generate HLSL includes. They are implied by other flags that are
4897     // generated elsewhere.
4898     if (LangOpts.HLSL && I == "hlsl.h")
4899       continue;
4900 
4901     GenerateArg(Consumer, OPT_include, I);
4902   }
4903 
4904   for (const auto &CI : Opts.ChainedIncludes)
4905     GenerateArg(Consumer, OPT_chain_include, CI);
4906 
4907   for (const auto &RF : Opts.RemappedFiles)
4908     GenerateArg(Consumer, OPT_remap_file, RF.first + ";" + RF.second);
4909 
4910   if (Opts.SourceDateEpoch)
4911     GenerateArg(Consumer, OPT_source_date_epoch, Twine(*Opts.SourceDateEpoch));
4912 
4913   if (Opts.DefineTargetOSMacros)
4914     GenerateArg(Consumer, OPT_fdefine_target_os_macros);
4915 
4916   for (const auto &EmbedEntry : Opts.EmbedEntries)
4917     GenerateArg(Consumer, OPT_embed_dir_EQ, EmbedEntry);
4918 
4919   // Don't handle LexEditorPlaceholders. It is implied by the action that is
4920   // generated elsewhere.
4921 }
4922 
ParsePreprocessorArgs(PreprocessorOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags,frontend::ActionKind Action,const FrontendOptions & FrontendOpts)4923 static bool ParsePreprocessorArgs(PreprocessorOptions &Opts, ArgList &Args,
4924                                   DiagnosticsEngine &Diags,
4925                                   frontend::ActionKind Action,
4926                                   const FrontendOptions &FrontendOpts) {
4927   unsigned NumErrorsBefore = Diags.getNumErrors();
4928 
4929   PreprocessorOptions *PreprocessorOpts = &Opts;
4930 
4931 #define PREPROCESSOR_OPTION_WITH_MARSHALLING(...)                              \
4932   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
4933 #include "clang/Driver/Options.inc"
4934 #undef PREPROCESSOR_OPTION_WITH_MARSHALLING
4935 
4936   Opts.PCHWithHdrStop = Args.hasArg(OPT_pch_through_hdrstop_create) ||
4937                         Args.hasArg(OPT_pch_through_hdrstop_use);
4938 
4939   for (const auto *A : Args.filtered(OPT_error_on_deserialized_pch_decl))
4940     Opts.DeserializedPCHDeclsToErrorOn.insert(A->getValue());
4941 
4942   if (const Arg *A = Args.getLastArg(OPT_preamble_bytes_EQ)) {
4943     StringRef Value(A->getValue());
4944     size_t Comma = Value.find(',');
4945     unsigned Bytes = 0;
4946     unsigned EndOfLine = 0;
4947 
4948     if (Comma == StringRef::npos ||
4949         Value.substr(0, Comma).getAsInteger(10, Bytes) ||
4950         Value.substr(Comma + 1).getAsInteger(10, EndOfLine))
4951       Diags.Report(diag::err_drv_preamble_format);
4952     else {
4953       Opts.PrecompiledPreambleBytes.first = Bytes;
4954       Opts.PrecompiledPreambleBytes.second = (EndOfLine != 0);
4955     }
4956   }
4957 
4958   // Add macros from the command line.
4959   for (const auto *A : Args.filtered(OPT_D, OPT_U)) {
4960     if (A->getOption().matches(OPT_D))
4961       Opts.addMacroDef(A->getValue());
4962     else
4963       Opts.addMacroUndef(A->getValue());
4964   }
4965 
4966   // Add the ordered list of -includes.
4967   for (const auto *A : Args.filtered(OPT_include))
4968     Opts.Includes.emplace_back(A->getValue());
4969 
4970   for (const auto *A : Args.filtered(OPT_chain_include))
4971     Opts.ChainedIncludes.emplace_back(A->getValue());
4972 
4973   for (const auto *A : Args.filtered(OPT_remap_file)) {
4974     std::pair<StringRef, StringRef> Split = StringRef(A->getValue()).split(';');
4975 
4976     if (Split.second.empty()) {
4977       Diags.Report(diag::err_drv_invalid_remap_file) << A->getAsString(Args);
4978       continue;
4979     }
4980 
4981     Opts.addRemappedFile(Split.first, Split.second);
4982   }
4983 
4984   if (const Arg *A = Args.getLastArg(OPT_source_date_epoch)) {
4985     StringRef Epoch = A->getValue();
4986     // SOURCE_DATE_EPOCH, if specified, must be a non-negative decimal integer.
4987     // On time64 systems, pick 253402300799 (the UNIX timestamp of
4988     // 9999-12-31T23:59:59Z) as the upper bound.
4989     const uint64_t MaxTimestamp =
4990         std::min<uint64_t>(std::numeric_limits<time_t>::max(), 253402300799);
4991     uint64_t V;
4992     if (Epoch.getAsInteger(10, V) || V > MaxTimestamp) {
4993       Diags.Report(diag::err_fe_invalid_source_date_epoch)
4994           << Epoch << MaxTimestamp;
4995     } else {
4996       Opts.SourceDateEpoch = V;
4997     }
4998   }
4999 
5000   for (const auto *A : Args.filtered(OPT_embed_dir_EQ)) {
5001     StringRef Val = A->getValue();
5002     Opts.EmbedEntries.push_back(std::string(Val));
5003   }
5004 
5005   // Always avoid lexing editor placeholders when we're just running the
5006   // preprocessor as we never want to emit the
5007   // "editor placeholder in source file" error in PP only mode.
5008   if (isStrictlyPreprocessorAction(Action))
5009     Opts.LexEditorPlaceholders = false;
5010 
5011   Opts.DefineTargetOSMacros =
5012       Args.hasFlag(OPT_fdefine_target_os_macros,
5013                    OPT_fno_define_target_os_macros, Opts.DefineTargetOSMacros);
5014 
5015   return Diags.getNumErrors() == NumErrorsBefore;
5016 }
5017 
5018 static void
GeneratePreprocessorOutputArgs(const PreprocessorOutputOptions & Opts,ArgumentConsumer Consumer,frontend::ActionKind Action)5019 GeneratePreprocessorOutputArgs(const PreprocessorOutputOptions &Opts,
5020                                ArgumentConsumer Consumer,
5021                                frontend::ActionKind Action) {
5022   const PreprocessorOutputOptions &PreprocessorOutputOpts = Opts;
5023 
5024 #define PREPROCESSOR_OUTPUT_OPTION_WITH_MARSHALLING(...)                       \
5025   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
5026 #include "clang/Driver/Options.inc"
5027 #undef PREPROCESSOR_OUTPUT_OPTION_WITH_MARSHALLING
5028 
5029   bool Generate_dM = isStrictlyPreprocessorAction(Action) && !Opts.ShowCPP;
5030   if (Generate_dM)
5031     GenerateArg(Consumer, OPT_dM);
5032   if (!Generate_dM && Opts.ShowMacros)
5033     GenerateArg(Consumer, OPT_dD);
5034   if (Opts.DirectivesOnly)
5035     GenerateArg(Consumer, OPT_fdirectives_only);
5036 }
5037 
ParsePreprocessorOutputArgs(PreprocessorOutputOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags,frontend::ActionKind Action)5038 static bool ParsePreprocessorOutputArgs(PreprocessorOutputOptions &Opts,
5039                                         ArgList &Args, DiagnosticsEngine &Diags,
5040                                         frontend::ActionKind Action) {
5041   unsigned NumErrorsBefore = Diags.getNumErrors();
5042 
5043   PreprocessorOutputOptions &PreprocessorOutputOpts = Opts;
5044 
5045 #define PREPROCESSOR_OUTPUT_OPTION_WITH_MARSHALLING(...)                       \
5046   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
5047 #include "clang/Driver/Options.inc"
5048 #undef PREPROCESSOR_OUTPUT_OPTION_WITH_MARSHALLING
5049 
5050   Opts.ShowCPP = isStrictlyPreprocessorAction(Action) && !Args.hasArg(OPT_dM);
5051   Opts.ShowMacros = Args.hasArg(OPT_dM) || Args.hasArg(OPT_dD);
5052   Opts.DirectivesOnly = Args.hasArg(OPT_fdirectives_only);
5053 
5054   return Diags.getNumErrors() == NumErrorsBefore;
5055 }
5056 
GenerateTargetArgs(const TargetOptions & Opts,ArgumentConsumer Consumer)5057 static void GenerateTargetArgs(const TargetOptions &Opts,
5058                                ArgumentConsumer Consumer) {
5059   const TargetOptions *TargetOpts = &Opts;
5060 #define TARGET_OPTION_WITH_MARSHALLING(...)                                    \
5061   GENERATE_OPTION_WITH_MARSHALLING(Consumer, __VA_ARGS__)
5062 #include "clang/Driver/Options.inc"
5063 #undef TARGET_OPTION_WITH_MARSHALLING
5064 
5065   if (!Opts.SDKVersion.empty())
5066     GenerateArg(Consumer, OPT_target_sdk_version_EQ,
5067                 Opts.SDKVersion.getAsString());
5068   if (!Opts.DarwinTargetVariantSDKVersion.empty())
5069     GenerateArg(Consumer, OPT_darwin_target_variant_sdk_version_EQ,
5070                 Opts.DarwinTargetVariantSDKVersion.getAsString());
5071 }
5072 
ParseTargetArgs(TargetOptions & Opts,ArgList & Args,DiagnosticsEngine & Diags)5073 static bool ParseTargetArgs(TargetOptions &Opts, ArgList &Args,
5074                             DiagnosticsEngine &Diags) {
5075   unsigned NumErrorsBefore = Diags.getNumErrors();
5076 
5077   TargetOptions *TargetOpts = &Opts;
5078 
5079 #define TARGET_OPTION_WITH_MARSHALLING(...)                                    \
5080   PARSE_OPTION_WITH_MARSHALLING(Args, Diags, __VA_ARGS__)
5081 #include "clang/Driver/Options.inc"
5082 #undef TARGET_OPTION_WITH_MARSHALLING
5083 
5084   if (Arg *A = Args.getLastArg(options::OPT_target_sdk_version_EQ)) {
5085     llvm::VersionTuple Version;
5086     if (Version.tryParse(A->getValue()))
5087       Diags.Report(diag::err_drv_invalid_value)
5088           << A->getAsString(Args) << A->getValue();
5089     else
5090       Opts.SDKVersion = Version;
5091   }
5092   if (Arg *A =
5093           Args.getLastArg(options::OPT_darwin_target_variant_sdk_version_EQ)) {
5094     llvm::VersionTuple Version;
5095     if (Version.tryParse(A->getValue()))
5096       Diags.Report(diag::err_drv_invalid_value)
5097           << A->getAsString(Args) << A->getValue();
5098     else
5099       Opts.DarwinTargetVariantSDKVersion = Version;
5100   }
5101 
5102   return Diags.getNumErrors() == NumErrorsBefore;
5103 }
5104 
CreateFromArgsImpl(CompilerInvocation & Res,ArrayRef<const char * > CommandLineArgs,DiagnosticsEngine & Diags,const char * Argv0)5105 bool CompilerInvocation::CreateFromArgsImpl(
5106     CompilerInvocation &Res, ArrayRef<const char *> CommandLineArgs,
5107     DiagnosticsEngine &Diags, const char *Argv0) {
5108   unsigned NumErrorsBefore = Diags.getNumErrors();
5109 
5110   // Parse the arguments.
5111   const OptTable &Opts = getDriverOptTable();
5112   llvm::opt::Visibility VisibilityMask(options::CC1Option);
5113   unsigned MissingArgIndex, MissingArgCount;
5114   InputArgList Args = Opts.ParseArgs(CommandLineArgs, MissingArgIndex,
5115                                      MissingArgCount, VisibilityMask);
5116   LangOptions &LangOpts = Res.getLangOpts();
5117 
5118   // Check for missing argument error.
5119   if (MissingArgCount)
5120     Diags.Report(diag::err_drv_missing_argument)
5121         << Args.getArgString(MissingArgIndex) << MissingArgCount;
5122 
5123   // Issue errors on unknown arguments.
5124   for (const auto *A : Args.filtered(OPT_UNKNOWN)) {
5125     auto ArgString = A->getAsString(Args);
5126     std::string Nearest;
5127     if (Opts.findNearest(ArgString, Nearest, VisibilityMask) > 1)
5128       Diags.Report(diag::err_drv_unknown_argument) << ArgString;
5129     else
5130       Diags.Report(diag::err_drv_unknown_argument_with_suggestion)
5131           << ArgString << Nearest;
5132   }
5133 
5134   ParseFileSystemArgs(Res.getFileSystemOpts(), Args, Diags);
5135   ParseMigratorArgs(Res.getMigratorOpts(), Args, Diags);
5136   ParseAnalyzerArgs(Res.getAnalyzerOpts(), Args, Diags);
5137   ParseDiagnosticArgs(Res.getDiagnosticOpts(), Args, &Diags,
5138                       /*DefaultDiagColor=*/false);
5139   ParseFrontendArgs(Res.getFrontendOpts(), Args, Diags, LangOpts.IsHeaderFile);
5140   // FIXME: We shouldn't have to pass the DashX option around here
5141   InputKind DashX = Res.getFrontendOpts().DashX;
5142   ParseTargetArgs(Res.getTargetOpts(), Args, Diags);
5143   llvm::Triple T(Res.getTargetOpts().Triple);
5144   ParseHeaderSearchArgs(Res.getHeaderSearchOpts(), Args, Diags,
5145                         Res.getFileSystemOpts().WorkingDir);
5146   if (Res.getFrontendOpts().GenReducedBMI ||
5147       Res.getFrontendOpts().ProgramAction ==
5148           frontend::GenerateReducedModuleInterface ||
5149       Res.getFrontendOpts().ProgramAction ==
5150           frontend::GenerateModuleInterface) {
5151     Res.getHeaderSearchOpts().ModulesSkipDiagnosticOptions = true;
5152     Res.getHeaderSearchOpts().ModulesSkipHeaderSearchPaths = true;
5153   }
5154   ParseAPINotesArgs(Res.getAPINotesOpts(), Args, Diags);
5155 
5156   ParsePointerAuthArgs(LangOpts, Args, Diags);
5157 
5158   ParseLangArgs(LangOpts, Args, DashX, T, Res.getPreprocessorOpts().Includes,
5159                 Diags);
5160   if (Res.getFrontendOpts().ProgramAction == frontend::RewriteObjC)
5161     LangOpts.ObjCExceptions = 1;
5162 
5163   for (auto Warning : Res.getDiagnosticOpts().Warnings) {
5164     if (Warning == "misexpect" &&
5165         !Diags.isIgnored(diag::warn_profile_data_misexpect, SourceLocation())) {
5166       Res.getCodeGenOpts().MisExpect = true;
5167     }
5168   }
5169 
5170   if (LangOpts.CUDA) {
5171     // During CUDA device-side compilation, the aux triple is the
5172     // triple used for host compilation.
5173     if (LangOpts.CUDAIsDevice)
5174       Res.getTargetOpts().HostTriple = Res.getFrontendOpts().AuxTriple;
5175   }
5176 
5177   if (LangOpts.OpenACC && !Res.getFrontendOpts().UseClangIRPipeline &&
5178       isCodeGenAction(Res.getFrontendOpts().ProgramAction))
5179     Diags.Report(diag::warn_drv_openacc_without_cir);
5180 
5181   // Set the triple of the host for OpenMP device compile.
5182   if (LangOpts.OpenMPIsTargetDevice)
5183     Res.getTargetOpts().HostTriple = Res.getFrontendOpts().AuxTriple;
5184 
5185   ParseCodeGenArgs(Res.getCodeGenOpts(), Args, DashX, Diags, T,
5186                    Res.getFrontendOpts().OutputFile, LangOpts);
5187 
5188   // FIXME: Override value name discarding when asan or msan is used because the
5189   // backend passes depend on the name of the alloca in order to print out
5190   // names.
5191   Res.getCodeGenOpts().DiscardValueNames &=
5192       !LangOpts.Sanitize.has(SanitizerKind::Address) &&
5193       !LangOpts.Sanitize.has(SanitizerKind::KernelAddress) &&
5194       !LangOpts.Sanitize.has(SanitizerKind::Memory) &&
5195       !LangOpts.Sanitize.has(SanitizerKind::KernelMemory);
5196 
5197   ParsePreprocessorArgs(Res.getPreprocessorOpts(), Args, Diags,
5198                         Res.getFrontendOpts().ProgramAction,
5199                         Res.getFrontendOpts());
5200   ParsePreprocessorOutputArgs(Res.getPreprocessorOutputOpts(), Args, Diags,
5201                               Res.getFrontendOpts().ProgramAction);
5202 
5203   ParseDependencyOutputArgs(Res.getDependencyOutputOpts(), Args, Diags,
5204                             Res.getFrontendOpts().ProgramAction,
5205                             Res.getPreprocessorOutputOpts().ShowLineMarkers);
5206   if (!Res.getDependencyOutputOpts().OutputFile.empty() &&
5207       Res.getDependencyOutputOpts().Targets.empty())
5208     Diags.Report(diag::err_fe_dependency_file_requires_MT);
5209 
5210   // If sanitizer is enabled, disable OPT_ffine_grained_bitfield_accesses.
5211   if (Res.getCodeGenOpts().FineGrainedBitfieldAccesses &&
5212       !Res.getLangOpts().Sanitize.empty()) {
5213     Res.getCodeGenOpts().FineGrainedBitfieldAccesses = false;
5214     Diags.Report(diag::warn_drv_fine_grained_bitfield_accesses_ignored);
5215   }
5216 
5217   // Store the command-line for using in the CodeView backend.
5218   if (Res.getCodeGenOpts().CodeViewCommandLine) {
5219     Res.getCodeGenOpts().Argv0 = Argv0;
5220     append_range(Res.getCodeGenOpts().CommandLineArgs, CommandLineArgs);
5221   }
5222 
5223   // Set PGOOptions. Need to create a temporary VFS to read the profile
5224   // to determine the PGO type.
5225   if (!Res.getCodeGenOpts().ProfileInstrumentUsePath.empty()) {
5226     auto FS =
5227         createVFSFromOverlayFiles(Res.getHeaderSearchOpts().VFSOverlayFiles,
5228                                   Diags, llvm::vfs::getRealFileSystem());
5229     setPGOUseInstrumentor(Res.getCodeGenOpts(),
5230                           Res.getCodeGenOpts().ProfileInstrumentUsePath, *FS,
5231                           Diags);
5232   }
5233 
5234   FixupInvocation(Res, Diags, Args, DashX);
5235 
5236   return Diags.getNumErrors() == NumErrorsBefore;
5237 }
5238 
CreateFromArgs(CompilerInvocation & Invocation,ArrayRef<const char * > CommandLineArgs,DiagnosticsEngine & Diags,const char * Argv0)5239 bool CompilerInvocation::CreateFromArgs(CompilerInvocation &Invocation,
5240                                         ArrayRef<const char *> CommandLineArgs,
5241                                         DiagnosticsEngine &Diags,
5242                                         const char *Argv0) {
5243   CompilerInvocation DummyInvocation;
5244 
5245   return RoundTrip(
5246       [](CompilerInvocation &Invocation, ArrayRef<const char *> CommandLineArgs,
5247          DiagnosticsEngine &Diags, const char *Argv0) {
5248         return CreateFromArgsImpl(Invocation, CommandLineArgs, Diags, Argv0);
5249       },
5250       [](CompilerInvocation &Invocation, SmallVectorImpl<const char *> &Args,
5251          StringAllocator SA) {
5252         Args.push_back("-cc1");
5253         Invocation.generateCC1CommandLine(Args, SA);
5254       },
5255       Invocation, DummyInvocation, CommandLineArgs, Diags, Argv0);
5256 }
5257 
getModuleHash() const5258 std::string CompilerInvocation::getModuleHash() const {
5259   // FIXME: Consider using SHA1 instead of MD5.
5260   llvm::HashBuilder<llvm::MD5, llvm::endianness::native> HBuilder;
5261 
5262   // Note: For QoI reasons, the things we use as a hash here should all be
5263   // dumped via the -module-info flag.
5264 
5265   // Start the signature with the compiler version.
5266   HBuilder.add(getClangFullRepositoryVersion());
5267 
5268   // Also include the serialization version, in case LLVM_APPEND_VC_REV is off
5269   // and getClangFullRepositoryVersion() doesn't include git revision.
5270   HBuilder.add(serialization::VERSION_MAJOR, serialization::VERSION_MINOR);
5271 
5272   // Extend the signature with the language options
5273   // FIXME: Replace with C++20 `using enum LangOptions::CompatibilityKind`.
5274   using CK = LangOptions::CompatibilityKind;
5275 #define LANGOPT(Name, Bits, Default, Compatibility, Description)               \
5276   if constexpr (CK::Compatibility != CK::Benign)                               \
5277     HBuilder.add(LangOpts->Name);
5278 #define ENUM_LANGOPT(Name, Type, Bits, Default, Compatibility, Description)    \
5279   if constexpr (CK::Compatibility != CK::Benign)                               \
5280     HBuilder.add(static_cast<unsigned>(LangOpts->get##Name()));
5281 #include "clang/Basic/LangOptions.def"
5282 
5283   HBuilder.addRange(getLangOpts().ModuleFeatures);
5284 
5285   HBuilder.add(getLangOpts().ObjCRuntime);
5286   HBuilder.addRange(getLangOpts().CommentOpts.BlockCommandNames);
5287 
5288   // Extend the signature with the target options.
5289   HBuilder.add(getTargetOpts().Triple, getTargetOpts().CPU,
5290                getTargetOpts().TuneCPU, getTargetOpts().ABI);
5291   HBuilder.addRange(getTargetOpts().FeaturesAsWritten);
5292 
5293   // Extend the signature with preprocessor options.
5294   const PreprocessorOptions &ppOpts = getPreprocessorOpts();
5295   HBuilder.add(ppOpts.UsePredefines, ppOpts.DetailedRecord);
5296 
5297   const HeaderSearchOptions &hsOpts = getHeaderSearchOpts();
5298   for (const auto &Macro : getPreprocessorOpts().Macros) {
5299     // If we're supposed to ignore this macro for the purposes of modules,
5300     // don't put it into the hash.
5301     if (!hsOpts.ModulesIgnoreMacros.empty()) {
5302       // Check whether we're ignoring this macro.
5303       StringRef MacroDef = Macro.first;
5304       if (hsOpts.ModulesIgnoreMacros.count(
5305               llvm::CachedHashString(MacroDef.split('=').first)))
5306         continue;
5307     }
5308 
5309     HBuilder.add(Macro);
5310   }
5311 
5312   // Extend the signature with the sysroot and other header search options.
5313   HBuilder.add(hsOpts.Sysroot, hsOpts.ModuleFormat, hsOpts.UseDebugInfo,
5314                hsOpts.UseBuiltinIncludes, hsOpts.UseStandardSystemIncludes,
5315                hsOpts.UseStandardCXXIncludes, hsOpts.UseLibcxx,
5316                hsOpts.ModulesValidateDiagnosticOptions);
5317   HBuilder.add(hsOpts.ResourceDir);
5318 
5319   if (hsOpts.ModulesStrictContextHash) {
5320     HBuilder.addRange(hsOpts.SystemHeaderPrefixes);
5321     HBuilder.addRange(hsOpts.UserEntries);
5322     HBuilder.addRange(hsOpts.VFSOverlayFiles);
5323 
5324     const DiagnosticOptions &diagOpts = getDiagnosticOpts();
5325 #define DIAGOPT(Name, Bits, Default) HBuilder.add(diagOpts.Name);
5326 #define ENUM_DIAGOPT(Name, Type, Bits, Default)                                \
5327   HBuilder.add(diagOpts.get##Name());
5328 #include "clang/Basic/DiagnosticOptions.def"
5329 #undef DIAGOPT
5330 #undef ENUM_DIAGOPT
5331   }
5332 
5333   // Extend the signature with the user build path.
5334   HBuilder.add(hsOpts.ModuleUserBuildPath);
5335 
5336   // Extend the signature with the module file extensions.
5337   for (const auto &ext : getFrontendOpts().ModuleFileExtensions)
5338     ext->hashExtension(HBuilder);
5339 
5340   // Extend the signature with the Swift version for API notes.
5341   const APINotesOptions &APINotesOpts = getAPINotesOpts();
5342   if (!APINotesOpts.SwiftVersion.empty()) {
5343     HBuilder.add(APINotesOpts.SwiftVersion.getMajor());
5344     if (auto Minor = APINotesOpts.SwiftVersion.getMinor())
5345       HBuilder.add(*Minor);
5346     if (auto Subminor = APINotesOpts.SwiftVersion.getSubminor())
5347       HBuilder.add(*Subminor);
5348     if (auto Build = APINotesOpts.SwiftVersion.getBuild())
5349       HBuilder.add(*Build);
5350   }
5351 
5352   // When compiling with -gmodules, also hash -fdebug-prefix-map as it
5353   // affects the debug info in the PCM.
5354   if (getCodeGenOpts().DebugTypeExtRefs)
5355     HBuilder.addRange(getCodeGenOpts().DebugPrefixMap);
5356 
5357   // Extend the signature with the affecting debug options.
5358   if (getHeaderSearchOpts().ModuleFormat == "obj") {
5359     // FIXME: Replace with C++20 `using enum CodeGenOptions::CompatibilityKind`.
5360     using CK = CodeGenOptions::CompatibilityKind;
5361 #define DEBUGOPT(Name, Bits, Default, Compatibility)                           \
5362   if constexpr (CK::Compatibility == CK::Affecting)                            \
5363     HBuilder.add(CodeGenOpts->Name);
5364 #define VALUE_DEBUGOPT(Name, Bits, Default, Compatibility)                     \
5365   if constexpr (CK::Compatibility == CK::Affecting)                            \
5366     HBuilder.add(CodeGenOpts->Name);
5367 #define ENUM_DEBUGOPT(Name, Type, Bits, Default, Compatibility)                \
5368   if constexpr (CK::Compatibility == CK::Affecting)                            \
5369     HBuilder.add(static_cast<unsigned>(CodeGenOpts->get##Name()));
5370 #include "clang/Basic/DebugOptions.def"
5371   }
5372 
5373   // Extend the signature with the enabled sanitizers, if at least one is
5374   // enabled. Sanitizers which cannot affect AST generation aren't hashed.
5375   SanitizerSet SanHash = getLangOpts().Sanitize;
5376   SanHash.clear(getPPTransparentSanitizers());
5377   if (!SanHash.empty())
5378     HBuilder.add(SanHash.Mask);
5379 
5380   llvm::MD5::MD5Result Result;
5381   HBuilder.getHasher().final(Result);
5382   uint64_t Hash = Result.high() ^ Result.low();
5383   return toString(llvm::APInt(64, Hash), 36, /*Signed=*/false);
5384 }
5385 
generateCC1CommandLine(ArgumentConsumer Consumer) const5386 void CompilerInvocationBase::generateCC1CommandLine(
5387     ArgumentConsumer Consumer) const {
5388   llvm::Triple T(getTargetOpts().Triple);
5389 
5390   GenerateFileSystemArgs(getFileSystemOpts(), Consumer);
5391   GenerateMigratorArgs(getMigratorOpts(), Consumer);
5392   GenerateAnalyzerArgs(getAnalyzerOpts(), Consumer);
5393   GenerateDiagnosticArgs(getDiagnosticOpts(), Consumer,
5394                          /*DefaultDiagColor=*/false);
5395   GenerateFrontendArgs(getFrontendOpts(), Consumer, getLangOpts().IsHeaderFile);
5396   GenerateTargetArgs(getTargetOpts(), Consumer);
5397   GenerateHeaderSearchArgs(getHeaderSearchOpts(), Consumer);
5398   GenerateAPINotesArgs(getAPINotesOpts(), Consumer);
5399   GeneratePointerAuthArgs(getLangOpts(), Consumer);
5400   GenerateLangArgs(getLangOpts(), Consumer, T, getFrontendOpts().DashX);
5401   GenerateCodeGenArgs(getCodeGenOpts(), Consumer, T,
5402                       getFrontendOpts().OutputFile, &getLangOpts());
5403   GeneratePreprocessorArgs(getPreprocessorOpts(), Consumer, getLangOpts(),
5404                            getFrontendOpts(), getCodeGenOpts());
5405   GeneratePreprocessorOutputArgs(getPreprocessorOutputOpts(), Consumer,
5406                                  getFrontendOpts().ProgramAction);
5407   GenerateDependencyOutputArgs(getDependencyOutputOpts(), Consumer);
5408 }
5409 
getCC1CommandLine() const5410 std::vector<std::string> CompilerInvocationBase::getCC1CommandLine() const {
5411   std::vector<std::string> Args{"-cc1"};
5412   generateCC1CommandLine(
5413       [&Args](const Twine &Arg) { Args.push_back(Arg.str()); });
5414   return Args;
5415 }
5416 
resetNonModularOptions()5417 void CompilerInvocation::resetNonModularOptions() {
5418   getLangOpts().resetNonModularOptions();
5419   getPreprocessorOpts().resetNonModularOptions();
5420   getCodeGenOpts().resetNonModularOptions(getHeaderSearchOpts().ModuleFormat);
5421 }
5422 
clearImplicitModuleBuildOptions()5423 void CompilerInvocation::clearImplicitModuleBuildOptions() {
5424   getLangOpts().ImplicitModules = false;
5425   getHeaderSearchOpts().ImplicitModuleMaps = false;
5426   getHeaderSearchOpts().ModuleCachePath.clear();
5427   getHeaderSearchOpts().ModulesValidateOncePerBuildSession = false;
5428   getHeaderSearchOpts().BuildSessionTimestamp = 0;
5429   // The specific values we canonicalize to for pruning don't affect behaviour,
5430   /// so use the default values so they may be dropped from the command-line.
5431   getHeaderSearchOpts().ModuleCachePruneInterval = 7 * 24 * 60 * 60;
5432   getHeaderSearchOpts().ModuleCachePruneAfter = 31 * 24 * 60 * 60;
5433 }
5434 
5435 IntrusiveRefCntPtr<llvm::vfs::FileSystem>
createVFSFromCompilerInvocation(const CompilerInvocation & CI,DiagnosticsEngine & Diags)5436 clang::createVFSFromCompilerInvocation(const CompilerInvocation &CI,
5437                                        DiagnosticsEngine &Diags) {
5438   return createVFSFromCompilerInvocation(CI, Diags,
5439                                          llvm::vfs::getRealFileSystem());
5440 }
5441 
5442 IntrusiveRefCntPtr<llvm::vfs::FileSystem>
createVFSFromCompilerInvocation(const CompilerInvocation & CI,DiagnosticsEngine & Diags,IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS)5443 clang::createVFSFromCompilerInvocation(
5444     const CompilerInvocation &CI, DiagnosticsEngine &Diags,
5445     IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS) {
5446   return createVFSFromOverlayFiles(CI.getHeaderSearchOpts().VFSOverlayFiles,
5447                                    Diags, std::move(BaseFS));
5448 }
5449 
createVFSFromOverlayFiles(ArrayRef<std::string> VFSOverlayFiles,DiagnosticsEngine & Diags,IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS)5450 IntrusiveRefCntPtr<llvm::vfs::FileSystem> clang::createVFSFromOverlayFiles(
5451     ArrayRef<std::string> VFSOverlayFiles, DiagnosticsEngine &Diags,
5452     IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS) {
5453   if (VFSOverlayFiles.empty())
5454     return BaseFS;
5455 
5456   IntrusiveRefCntPtr<llvm::vfs::FileSystem> Result = BaseFS;
5457   // earlier vfs files are on the bottom
5458   for (const auto &File : VFSOverlayFiles) {
5459     llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> Buffer =
5460         Result->getBufferForFile(File);
5461     if (!Buffer) {
5462       Diags.Report(diag::err_missing_vfs_overlay_file) << File;
5463       continue;
5464     }
5465 
5466     IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS = llvm::vfs::getVFSFromYAML(
5467         std::move(Buffer.get()), /*DiagHandler*/ nullptr, File,
5468         /*DiagContext*/ nullptr, Result);
5469     if (!FS) {
5470       Diags.Report(diag::err_invalid_vfs_overlay) << File;
5471       continue;
5472     }
5473 
5474     Result = FS;
5475   }
5476   return Result;
5477 }
5478