xref: /llvm-project/clang/lib/Frontend/FrontendActions.cpp (revision f7d93373969b2b757f5d5ef5e157dabe3bb9b0ae)
1 //===--- FrontendActions.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/FrontendActions.h"
10 #include "clang/AST/ASTConsumer.h"
11 #include "clang/AST/Decl.h"
12 #include "clang/Basic/FileManager.h"
13 #include "clang/Basic/LangStandard.h"
14 #include "clang/Basic/Module.h"
15 #include "clang/Basic/TargetInfo.h"
16 #include "clang/Frontend/ASTConsumers.h"
17 #include "clang/Frontend/CompilerInstance.h"
18 #include "clang/Frontend/FrontendDiagnostic.h"
19 #include "clang/Frontend/MultiplexConsumer.h"
20 #include "clang/Frontend/Utils.h"
21 #include "clang/Lex/DependencyDirectivesScanner.h"
22 #include "clang/Lex/HeaderSearch.h"
23 #include "clang/Lex/Preprocessor.h"
24 #include "clang/Lex/PreprocessorOptions.h"
25 #include "clang/Sema/TemplateInstCallback.h"
26 #include "clang/Serialization/ASTReader.h"
27 #include "clang/Serialization/ASTWriter.h"
28 #include "clang/Serialization/ModuleFile.h"
29 #include "llvm/Support/ErrorHandling.h"
30 #include "llvm/Support/FileSystem.h"
31 #include "llvm/Support/MemoryBuffer.h"
32 #include "llvm/Support/Path.h"
33 #include "llvm/Support/YAMLTraits.h"
34 #include "llvm/Support/raw_ostream.h"
35 #include <memory>
36 #include <optional>
37 #include <system_error>
38 
39 using namespace clang;
40 
41 namespace {
42 CodeCompleteConsumer *GetCodeCompletionConsumer(CompilerInstance &CI) {
43   return CI.hasCodeCompletionConsumer() ? &CI.getCodeCompletionConsumer()
44                                         : nullptr;
45 }
46 
47 void EnsureSemaIsCreated(CompilerInstance &CI, FrontendAction &Action) {
48   if (Action.hasCodeCompletionSupport() &&
49       !CI.getFrontendOpts().CodeCompletionAt.FileName.empty())
50     CI.createCodeCompletionConsumer();
51 
52   if (!CI.hasSema())
53     CI.createSema(Action.getTranslationUnitKind(),
54                   GetCodeCompletionConsumer(CI));
55 }
56 } // namespace
57 
58 //===----------------------------------------------------------------------===//
59 // Custom Actions
60 //===----------------------------------------------------------------------===//
61 
62 std::unique_ptr<ASTConsumer>
63 InitOnlyAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
64   return std::make_unique<ASTConsumer>();
65 }
66 
67 void InitOnlyAction::ExecuteAction() {
68 }
69 
70 // Basically PreprocessOnlyAction::ExecuteAction.
71 void ReadPCHAndPreprocessAction::ExecuteAction() {
72   CompilerInstance &CI = getCompilerInstance();
73   AdjustCI(CI);
74 
75   Preprocessor &PP = CI.getPreprocessor();
76 
77   // Ignore unknown pragmas.
78   PP.IgnorePragmas();
79 
80   Token Tok;
81   // Start parsing the specified input file.
82   PP.EnterMainSourceFile();
83   do {
84     PP.Lex(Tok);
85   } while (Tok.isNot(tok::eof));
86 }
87 
88 std::unique_ptr<ASTConsumer>
89 ReadPCHAndPreprocessAction::CreateASTConsumer(CompilerInstance &CI,
90                                               StringRef InFile) {
91   return std::make_unique<ASTConsumer>();
92 }
93 
94 //===----------------------------------------------------------------------===//
95 // AST Consumer Actions
96 //===----------------------------------------------------------------------===//
97 
98 std::unique_ptr<ASTConsumer>
99 ASTPrintAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
100   if (std::unique_ptr<raw_ostream> OS =
101           CI.createDefaultOutputFile(false, InFile))
102     return CreateASTPrinter(std::move(OS), CI.getFrontendOpts().ASTDumpFilter);
103   return nullptr;
104 }
105 
106 std::unique_ptr<ASTConsumer>
107 ASTDumpAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
108   const FrontendOptions &Opts = CI.getFrontendOpts();
109   return CreateASTDumper(nullptr /*Dump to stdout.*/, Opts.ASTDumpFilter,
110                          Opts.ASTDumpDecls, Opts.ASTDumpAll,
111                          Opts.ASTDumpLookups, Opts.ASTDumpDeclTypes,
112                          Opts.ASTDumpFormat);
113 }
114 
115 std::unique_ptr<ASTConsumer>
116 ASTDeclListAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
117   return CreateASTDeclNodeLister();
118 }
119 
120 std::unique_ptr<ASTConsumer>
121 ASTViewAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
122   return CreateASTViewer();
123 }
124 
125 std::unique_ptr<ASTConsumer>
126 GeneratePCHAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
127   std::string Sysroot;
128   if (!ComputeASTConsumerArguments(CI, /*ref*/ Sysroot))
129     return nullptr;
130 
131   std::string OutputFile;
132   std::unique_ptr<raw_pwrite_stream> OS =
133       CreateOutputFile(CI, InFile, /*ref*/ OutputFile);
134   if (!OS)
135     return nullptr;
136 
137   if (!CI.getFrontendOpts().RelocatablePCH)
138     Sysroot.clear();
139 
140   const auto &FrontendOpts = CI.getFrontendOpts();
141   auto Buffer = std::make_shared<PCHBuffer>();
142   std::vector<std::unique_ptr<ASTConsumer>> Consumers;
143   Consumers.push_back(std::make_unique<PCHGenerator>(
144       CI.getPreprocessor(), CI.getModuleCache(), OutputFile, Sysroot, Buffer,
145       FrontendOpts.ModuleFileExtensions,
146       CI.getPreprocessorOpts().AllowPCHWithCompilerErrors,
147       FrontendOpts.IncludeTimestamps, FrontendOpts.BuildingImplicitModule,
148       +CI.getLangOpts().CacheGeneratedPCH));
149   Consumers.push_back(CI.getPCHContainerWriter().CreatePCHContainerGenerator(
150       CI, std::string(InFile), OutputFile, std::move(OS), Buffer));
151 
152   return std::make_unique<MultiplexConsumer>(std::move(Consumers));
153 }
154 
155 bool GeneratePCHAction::ComputeASTConsumerArguments(CompilerInstance &CI,
156                                                     std::string &Sysroot) {
157   Sysroot = CI.getHeaderSearchOpts().Sysroot;
158   if (CI.getFrontendOpts().RelocatablePCH && Sysroot.empty()) {
159     CI.getDiagnostics().Report(diag::err_relocatable_without_isysroot);
160     return false;
161   }
162 
163   return true;
164 }
165 
166 std::unique_ptr<llvm::raw_pwrite_stream>
167 GeneratePCHAction::CreateOutputFile(CompilerInstance &CI, StringRef InFile,
168                                     std::string &OutputFile) {
169   // Because this is exposed via libclang we must disable RemoveFileOnSignal.
170   std::unique_ptr<raw_pwrite_stream> OS = CI.createDefaultOutputFile(
171       /*Binary=*/true, InFile, /*Extension=*/"", /*RemoveFileOnSignal=*/false);
172   if (!OS)
173     return nullptr;
174 
175   OutputFile = CI.getFrontendOpts().OutputFile;
176   return OS;
177 }
178 
179 bool GeneratePCHAction::shouldEraseOutputFiles() {
180   if (getCompilerInstance().getPreprocessorOpts().AllowPCHWithCompilerErrors)
181     return false;
182   return ASTFrontendAction::shouldEraseOutputFiles();
183 }
184 
185 bool GeneratePCHAction::BeginSourceFileAction(CompilerInstance &CI) {
186   CI.getLangOpts().CompilingPCH = true;
187   return true;
188 }
189 
190 std::vector<std::unique_ptr<ASTConsumer>>
191 GenerateModuleAction::CreateMultiplexConsumer(CompilerInstance &CI,
192                                               StringRef InFile) {
193   std::unique_ptr<raw_pwrite_stream> OS = CreateOutputFile(CI, InFile);
194   if (!OS)
195     return {};
196 
197   std::string OutputFile = CI.getFrontendOpts().OutputFile;
198   std::string Sysroot;
199 
200   auto Buffer = std::make_shared<PCHBuffer>();
201   std::vector<std::unique_ptr<ASTConsumer>> Consumers;
202 
203   Consumers.push_back(std::make_unique<PCHGenerator>(
204       CI.getPreprocessor(), CI.getModuleCache(), OutputFile, Sysroot, Buffer,
205       CI.getFrontendOpts().ModuleFileExtensions,
206       /*AllowASTWithErrors=*/
207       +CI.getFrontendOpts().AllowPCMWithCompilerErrors,
208       /*IncludeTimestamps=*/
209       +CI.getFrontendOpts().BuildingImplicitModule &&
210           +CI.getFrontendOpts().IncludeTimestamps,
211       /*BuildingImplicitModule=*/+CI.getFrontendOpts().BuildingImplicitModule,
212       /*ShouldCacheASTInMemory=*/
213       +CI.getFrontendOpts().BuildingImplicitModule));
214   Consumers.push_back(CI.getPCHContainerWriter().CreatePCHContainerGenerator(
215       CI, std::string(InFile), OutputFile, std::move(OS), Buffer));
216   return Consumers;
217 }
218 
219 std::unique_ptr<ASTConsumer>
220 GenerateModuleAction::CreateASTConsumer(CompilerInstance &CI,
221                                         StringRef InFile) {
222   std::vector<std::unique_ptr<ASTConsumer>> Consumers =
223       CreateMultiplexConsumer(CI, InFile);
224   if (Consumers.empty())
225     return nullptr;
226 
227   return std::make_unique<MultiplexConsumer>(std::move(Consumers));
228 }
229 
230 bool GenerateModuleAction::shouldEraseOutputFiles() {
231   return !getCompilerInstance().getFrontendOpts().AllowPCMWithCompilerErrors &&
232          ASTFrontendAction::shouldEraseOutputFiles();
233 }
234 
235 bool GenerateModuleFromModuleMapAction::BeginSourceFileAction(
236     CompilerInstance &CI) {
237   if (!CI.getLangOpts().Modules) {
238     CI.getDiagnostics().Report(diag::err_module_build_requires_fmodules);
239     return false;
240   }
241 
242   return GenerateModuleAction::BeginSourceFileAction(CI);
243 }
244 
245 std::unique_ptr<raw_pwrite_stream>
246 GenerateModuleFromModuleMapAction::CreateOutputFile(CompilerInstance &CI,
247                                                     StringRef InFile) {
248   // If no output file was provided, figure out where this module would go
249   // in the module cache.
250   if (CI.getFrontendOpts().OutputFile.empty()) {
251     StringRef ModuleMapFile = CI.getFrontendOpts().OriginalModuleMap;
252     if (ModuleMapFile.empty())
253       ModuleMapFile = InFile;
254 
255     HeaderSearch &HS = CI.getPreprocessor().getHeaderSearchInfo();
256     CI.getFrontendOpts().OutputFile =
257         HS.getCachedModuleFileName(CI.getLangOpts().CurrentModule,
258                                    ModuleMapFile);
259   }
260 
261   // Because this is exposed via libclang we must disable RemoveFileOnSignal.
262   return CI.createDefaultOutputFile(/*Binary=*/true, InFile, /*Extension=*/"",
263                                     /*RemoveFileOnSignal=*/false,
264                                     /*CreateMissingDirectories=*/true,
265                                     /*ForceUseTemporary=*/true);
266 }
267 
268 bool GenerateModuleInterfaceAction::BeginSourceFileAction(
269     CompilerInstance &CI) {
270   CI.getLangOpts().setCompilingModule(LangOptions::CMK_ModuleInterface);
271 
272   return GenerateModuleAction::BeginSourceFileAction(CI);
273 }
274 
275 std::unique_ptr<ASTConsumer>
276 GenerateModuleInterfaceAction::CreateASTConsumer(CompilerInstance &CI,
277                                                  StringRef InFile) {
278   CI.getHeaderSearchOpts().ModulesSkipDiagnosticOptions = true;
279   CI.getHeaderSearchOpts().ModulesSkipHeaderSearchPaths = true;
280   CI.getHeaderSearchOpts().ModulesSkipPragmaDiagnosticMappings = true;
281 
282   std::vector<std::unique_ptr<ASTConsumer>> Consumers =
283       CreateMultiplexConsumer(CI, InFile);
284   if (Consumers.empty())
285     return nullptr;
286 
287   return std::make_unique<MultiplexConsumer>(std::move(Consumers));
288 }
289 
290 std::unique_ptr<raw_pwrite_stream>
291 GenerateModuleInterfaceAction::CreateOutputFile(CompilerInstance &CI,
292                                                 StringRef InFile) {
293   return CI.createDefaultOutputFile(/*Binary=*/true, InFile, "pcm");
294 }
295 
296 std::unique_ptr<ASTConsumer>
297 GenerateReducedModuleInterfaceAction::CreateASTConsumer(CompilerInstance &CI,
298                                                         StringRef InFile) {
299   return std::make_unique<ReducedBMIGenerator>(CI.getPreprocessor(),
300                                                CI.getModuleCache(),
301                                                CI.getFrontendOpts().OutputFile);
302 }
303 
304 bool GenerateHeaderUnitAction::BeginSourceFileAction(CompilerInstance &CI) {
305   if (!CI.getLangOpts().CPlusPlusModules) {
306     CI.getDiagnostics().Report(diag::err_module_interface_requires_cpp_modules);
307     return false;
308   }
309   CI.getLangOpts().setCompilingModule(LangOptions::CMK_HeaderUnit);
310   return GenerateModuleAction::BeginSourceFileAction(CI);
311 }
312 
313 std::unique_ptr<raw_pwrite_stream>
314 GenerateHeaderUnitAction::CreateOutputFile(CompilerInstance &CI,
315                                            StringRef InFile) {
316   return CI.createDefaultOutputFile(/*Binary=*/true, InFile, "pcm");
317 }
318 
319 SyntaxOnlyAction::~SyntaxOnlyAction() {
320 }
321 
322 std::unique_ptr<ASTConsumer>
323 SyntaxOnlyAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
324   return std::make_unique<ASTConsumer>();
325 }
326 
327 std::unique_ptr<ASTConsumer>
328 DumpModuleInfoAction::CreateASTConsumer(CompilerInstance &CI,
329                                         StringRef InFile) {
330   return std::make_unique<ASTConsumer>();
331 }
332 
333 std::unique_ptr<ASTConsumer>
334 VerifyPCHAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
335   return std::make_unique<ASTConsumer>();
336 }
337 
338 void VerifyPCHAction::ExecuteAction() {
339   CompilerInstance &CI = getCompilerInstance();
340   bool Preamble = CI.getPreprocessorOpts().PrecompiledPreambleBytes.first != 0;
341   const std::string &Sysroot = CI.getHeaderSearchOpts().Sysroot;
342   std::unique_ptr<ASTReader> Reader(new ASTReader(
343       CI.getPreprocessor(), CI.getModuleCache(), &CI.getASTContext(),
344       CI.getPCHContainerReader(), CI.getFrontendOpts().ModuleFileExtensions,
345       Sysroot.empty() ? "" : Sysroot.c_str(),
346       DisableValidationForModuleKind::None,
347       /*AllowASTWithCompilerErrors*/ false,
348       /*AllowConfigurationMismatch*/ true,
349       /*ValidateSystemInputs*/ true));
350 
351   Reader->ReadAST(getCurrentFile(),
352                   Preamble ? serialization::MK_Preamble
353                            : serialization::MK_PCH,
354                   SourceLocation(),
355                   ASTReader::ARR_ConfigurationMismatch);
356 }
357 
358 namespace {
359 struct TemplightEntry {
360   std::string Name;
361   std::string Kind;
362   std::string Event;
363   std::string DefinitionLocation;
364   std::string PointOfInstantiation;
365 };
366 } // namespace
367 
368 namespace llvm {
369 namespace yaml {
370 template <> struct MappingTraits<TemplightEntry> {
371   static void mapping(IO &io, TemplightEntry &fields) {
372     io.mapRequired("name", fields.Name);
373     io.mapRequired("kind", fields.Kind);
374     io.mapRequired("event", fields.Event);
375     io.mapRequired("orig", fields.DefinitionLocation);
376     io.mapRequired("poi", fields.PointOfInstantiation);
377   }
378 };
379 } // namespace yaml
380 } // namespace llvm
381 
382 namespace {
383 class DefaultTemplateInstCallback : public TemplateInstantiationCallback {
384   using CodeSynthesisContext = Sema::CodeSynthesisContext;
385 
386 public:
387   void initialize(const Sema &) override {}
388 
389   void finalize(const Sema &) override {}
390 
391   void atTemplateBegin(const Sema &TheSema,
392                        const CodeSynthesisContext &Inst) override {
393     displayTemplightEntry<true>(llvm::outs(), TheSema, Inst);
394   }
395 
396   void atTemplateEnd(const Sema &TheSema,
397                      const CodeSynthesisContext &Inst) override {
398     displayTemplightEntry<false>(llvm::outs(), TheSema, Inst);
399   }
400 
401 private:
402   static std::string toString(CodeSynthesisContext::SynthesisKind Kind) {
403     switch (Kind) {
404     case CodeSynthesisContext::TemplateInstantiation:
405       return "TemplateInstantiation";
406     case CodeSynthesisContext::DefaultTemplateArgumentInstantiation:
407       return "DefaultTemplateArgumentInstantiation";
408     case CodeSynthesisContext::DefaultFunctionArgumentInstantiation:
409       return "DefaultFunctionArgumentInstantiation";
410     case CodeSynthesisContext::ExplicitTemplateArgumentSubstitution:
411       return "ExplicitTemplateArgumentSubstitution";
412     case CodeSynthesisContext::DeducedTemplateArgumentSubstitution:
413       return "DeducedTemplateArgumentSubstitution";
414     case CodeSynthesisContext::LambdaExpressionSubstitution:
415       return "LambdaExpressionSubstitution";
416     case CodeSynthesisContext::PriorTemplateArgumentSubstitution:
417       return "PriorTemplateArgumentSubstitution";
418     case CodeSynthesisContext::DefaultTemplateArgumentChecking:
419       return "DefaultTemplateArgumentChecking";
420     case CodeSynthesisContext::ExceptionSpecEvaluation:
421       return "ExceptionSpecEvaluation";
422     case CodeSynthesisContext::ExceptionSpecInstantiation:
423       return "ExceptionSpecInstantiation";
424     case CodeSynthesisContext::DeclaringSpecialMember:
425       return "DeclaringSpecialMember";
426     case CodeSynthesisContext::DeclaringImplicitEqualityComparison:
427       return "DeclaringImplicitEqualityComparison";
428     case CodeSynthesisContext::DefiningSynthesizedFunction:
429       return "DefiningSynthesizedFunction";
430     case CodeSynthesisContext::RewritingOperatorAsSpaceship:
431       return "RewritingOperatorAsSpaceship";
432     case CodeSynthesisContext::Memoization:
433       return "Memoization";
434     case CodeSynthesisContext::ConstraintsCheck:
435       return "ConstraintsCheck";
436     case CodeSynthesisContext::ConstraintSubstitution:
437       return "ConstraintSubstitution";
438     case CodeSynthesisContext::ConstraintNormalization:
439       return "ConstraintNormalization";
440     case CodeSynthesisContext::RequirementParameterInstantiation:
441       return "RequirementParameterInstantiation";
442     case CodeSynthesisContext::ParameterMappingSubstitution:
443       return "ParameterMappingSubstitution";
444     case CodeSynthesisContext::RequirementInstantiation:
445       return "RequirementInstantiation";
446     case CodeSynthesisContext::NestedRequirementConstraintsCheck:
447       return "NestedRequirementConstraintsCheck";
448     case CodeSynthesisContext::InitializingStructuredBinding:
449       return "InitializingStructuredBinding";
450     case CodeSynthesisContext::MarkingClassDllexported:
451       return "MarkingClassDllexported";
452     case CodeSynthesisContext::BuildingBuiltinDumpStructCall:
453       return "BuildingBuiltinDumpStructCall";
454     case CodeSynthesisContext::BuildingDeductionGuides:
455       return "BuildingDeductionGuides";
456     case CodeSynthesisContext::TypeAliasTemplateInstantiation:
457       return "TypeAliasTemplateInstantiation";
458     }
459     return "";
460   }
461 
462   template <bool BeginInstantiation>
463   static void displayTemplightEntry(llvm::raw_ostream &Out, const Sema &TheSema,
464                                     const CodeSynthesisContext &Inst) {
465     std::string YAML;
466     {
467       llvm::raw_string_ostream OS(YAML);
468       llvm::yaml::Output YO(OS);
469       TemplightEntry Entry =
470           getTemplightEntry<BeginInstantiation>(TheSema, Inst);
471       llvm::yaml::EmptyContext Context;
472       llvm::yaml::yamlize(YO, Entry, true, Context);
473     }
474     Out << "---" << YAML << "\n";
475   }
476 
477   static void printEntryName(const Sema &TheSema, const Decl *Entity,
478                              llvm::raw_string_ostream &OS) {
479     auto *NamedTemplate = cast<NamedDecl>(Entity);
480 
481     PrintingPolicy Policy = TheSema.Context.getPrintingPolicy();
482     // FIXME: Also ask for FullyQualifiedNames?
483     Policy.SuppressDefaultTemplateArgs = false;
484     NamedTemplate->getNameForDiagnostic(OS, Policy, true);
485 
486     if (!OS.str().empty())
487       return;
488 
489     Decl *Ctx = Decl::castFromDeclContext(NamedTemplate->getDeclContext());
490     NamedDecl *NamedCtx = dyn_cast_or_null<NamedDecl>(Ctx);
491 
492     if (const auto *Decl = dyn_cast<TagDecl>(NamedTemplate)) {
493       if (const auto *R = dyn_cast<RecordDecl>(Decl)) {
494         if (R->isLambda()) {
495           OS << "lambda at ";
496           Decl->getLocation().print(OS, TheSema.getSourceManager());
497           return;
498         }
499       }
500       OS << "unnamed " << Decl->getKindName();
501       return;
502     }
503 
504     assert(NamedCtx && "NamedCtx cannot be null");
505 
506     if (const auto *Decl = dyn_cast<ParmVarDecl>(NamedTemplate)) {
507       OS << "unnamed function parameter " << Decl->getFunctionScopeIndex()
508          << " ";
509       if (Decl->getFunctionScopeDepth() > 0)
510         OS << "(at depth " << Decl->getFunctionScopeDepth() << ") ";
511       OS << "of ";
512       NamedCtx->getNameForDiagnostic(OS, TheSema.getLangOpts(), true);
513       return;
514     }
515 
516     if (const auto *Decl = dyn_cast<TemplateTypeParmDecl>(NamedTemplate)) {
517       if (const Type *Ty = Decl->getTypeForDecl()) {
518         if (const auto *TTPT = dyn_cast_or_null<TemplateTypeParmType>(Ty)) {
519           OS << "unnamed template type parameter " << TTPT->getIndex() << " ";
520           if (TTPT->getDepth() > 0)
521             OS << "(at depth " << TTPT->getDepth() << ") ";
522           OS << "of ";
523           NamedCtx->getNameForDiagnostic(OS, TheSema.getLangOpts(), true);
524           return;
525         }
526       }
527     }
528 
529     if (const auto *Decl = dyn_cast<NonTypeTemplateParmDecl>(NamedTemplate)) {
530       OS << "unnamed template non-type parameter " << Decl->getIndex() << " ";
531       if (Decl->getDepth() > 0)
532         OS << "(at depth " << Decl->getDepth() << ") ";
533       OS << "of ";
534       NamedCtx->getNameForDiagnostic(OS, TheSema.getLangOpts(), true);
535       return;
536     }
537 
538     if (const auto *Decl = dyn_cast<TemplateTemplateParmDecl>(NamedTemplate)) {
539       OS << "unnamed template template parameter " << Decl->getIndex() << " ";
540       if (Decl->getDepth() > 0)
541         OS << "(at depth " << Decl->getDepth() << ") ";
542       OS << "of ";
543       NamedCtx->getNameForDiagnostic(OS, TheSema.getLangOpts(), true);
544       return;
545     }
546 
547     llvm_unreachable("Failed to retrieve a name for this entry!");
548     OS << "unnamed identifier";
549   }
550 
551   template <bool BeginInstantiation>
552   static TemplightEntry getTemplightEntry(const Sema &TheSema,
553                                           const CodeSynthesisContext &Inst) {
554     TemplightEntry Entry;
555     Entry.Kind = toString(Inst.Kind);
556     Entry.Event = BeginInstantiation ? "Begin" : "End";
557     llvm::raw_string_ostream OS(Entry.Name);
558     printEntryName(TheSema, Inst.Entity, OS);
559     const PresumedLoc DefLoc =
560         TheSema.getSourceManager().getPresumedLoc(Inst.Entity->getLocation());
561     if (!DefLoc.isInvalid())
562       Entry.DefinitionLocation = std::string(DefLoc.getFilename()) + ":" +
563                                  std::to_string(DefLoc.getLine()) + ":" +
564                                  std::to_string(DefLoc.getColumn());
565     const PresumedLoc PoiLoc =
566         TheSema.getSourceManager().getPresumedLoc(Inst.PointOfInstantiation);
567     if (!PoiLoc.isInvalid()) {
568       Entry.PointOfInstantiation = std::string(PoiLoc.getFilename()) + ":" +
569                                    std::to_string(PoiLoc.getLine()) + ":" +
570                                    std::to_string(PoiLoc.getColumn());
571     }
572     return Entry;
573   }
574 };
575 } // namespace
576 
577 std::unique_ptr<ASTConsumer>
578 TemplightDumpAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
579   return std::make_unique<ASTConsumer>();
580 }
581 
582 void TemplightDumpAction::ExecuteAction() {
583   CompilerInstance &CI = getCompilerInstance();
584 
585   // This part is normally done by ASTFrontEndAction, but needs to happen
586   // before Templight observers can be created
587   // FIXME: Move the truncation aspect of this into Sema, we delayed this till
588   // here so the source manager would be initialized.
589   EnsureSemaIsCreated(CI, *this);
590 
591   CI.getSema().TemplateInstCallbacks.push_back(
592       std::make_unique<DefaultTemplateInstCallback>());
593   ASTFrontendAction::ExecuteAction();
594 }
595 
596 namespace {
597   /// AST reader listener that dumps module information for a module
598   /// file.
599   class DumpModuleInfoListener : public ASTReaderListener {
600     llvm::raw_ostream &Out;
601 
602   public:
603     DumpModuleInfoListener(llvm::raw_ostream &Out) : Out(Out) { }
604 
605 #define DUMP_BOOLEAN(Value, Text)                       \
606     Out.indent(4) << Text << ": " << (Value? "Yes" : "No") << "\n"
607 
608     bool ReadFullVersionInformation(StringRef FullVersion) override {
609       Out.indent(2)
610         << "Generated by "
611         << (FullVersion == getClangFullRepositoryVersion()? "this"
612                                                           : "a different")
613         << " Clang: " << FullVersion << "\n";
614       return ASTReaderListener::ReadFullVersionInformation(FullVersion);
615     }
616 
617     void ReadModuleName(StringRef ModuleName) override {
618       Out.indent(2) << "Module name: " << ModuleName << "\n";
619     }
620     void ReadModuleMapFile(StringRef ModuleMapPath) override {
621       Out.indent(2) << "Module map file: " << ModuleMapPath << "\n";
622     }
623 
624     bool ReadLanguageOptions(const LangOptions &LangOpts, bool Complain,
625                              bool AllowCompatibleDifferences) override {
626       Out.indent(2) << "Language options:\n";
627 #define LANGOPT(Name, Bits, Default, Description) \
628       DUMP_BOOLEAN(LangOpts.Name, Description);
629 #define ENUM_LANGOPT(Name, Type, Bits, Default, Description) \
630       Out.indent(4) << Description << ": "                   \
631                     << static_cast<unsigned>(LangOpts.get##Name()) << "\n";
632 #define VALUE_LANGOPT(Name, Bits, Default, Description) \
633       Out.indent(4) << Description << ": " << LangOpts.Name << "\n";
634 #define BENIGN_LANGOPT(Name, Bits, Default, Description)
635 #define BENIGN_ENUM_LANGOPT(Name, Type, Bits, Default, Description)
636 #include "clang/Basic/LangOptions.def"
637 
638       if (!LangOpts.ModuleFeatures.empty()) {
639         Out.indent(4) << "Module features:\n";
640         for (StringRef Feature : LangOpts.ModuleFeatures)
641           Out.indent(6) << Feature << "\n";
642       }
643 
644       return false;
645     }
646 
647     bool ReadTargetOptions(const TargetOptions &TargetOpts, bool Complain,
648                            bool AllowCompatibleDifferences) override {
649       Out.indent(2) << "Target options:\n";
650       Out.indent(4) << "  Triple: " << TargetOpts.Triple << "\n";
651       Out.indent(4) << "  CPU: " << TargetOpts.CPU << "\n";
652       Out.indent(4) << "  TuneCPU: " << TargetOpts.TuneCPU << "\n";
653       Out.indent(4) << "  ABI: " << TargetOpts.ABI << "\n";
654 
655       if (!TargetOpts.FeaturesAsWritten.empty()) {
656         Out.indent(4) << "Target features:\n";
657         for (unsigned I = 0, N = TargetOpts.FeaturesAsWritten.size();
658              I != N; ++I) {
659           Out.indent(6) << TargetOpts.FeaturesAsWritten[I] << "\n";
660         }
661       }
662 
663       return false;
664     }
665 
666     bool ReadDiagnosticOptions(IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts,
667                                bool Complain) override {
668       Out.indent(2) << "Diagnostic options:\n";
669 #define DIAGOPT(Name, Bits, Default) DUMP_BOOLEAN(DiagOpts->Name, #Name);
670 #define ENUM_DIAGOPT(Name, Type, Bits, Default) \
671       Out.indent(4) << #Name << ": " << DiagOpts->get##Name() << "\n";
672 #define VALUE_DIAGOPT(Name, Bits, Default) \
673       Out.indent(4) << #Name << ": " << DiagOpts->Name << "\n";
674 #include "clang/Basic/DiagnosticOptions.def"
675 
676       Out.indent(4) << "Diagnostic flags:\n";
677       for (const std::string &Warning : DiagOpts->Warnings)
678         Out.indent(6) << "-W" << Warning << "\n";
679       for (const std::string &Remark : DiagOpts->Remarks)
680         Out.indent(6) << "-R" << Remark << "\n";
681 
682       return false;
683     }
684 
685     bool ReadHeaderSearchOptions(const HeaderSearchOptions &HSOpts,
686                                  StringRef SpecificModuleCachePath,
687                                  bool Complain) override {
688       Out.indent(2) << "Header search options:\n";
689       Out.indent(4) << "System root [-isysroot=]: '" << HSOpts.Sysroot << "'\n";
690       Out.indent(4) << "Resource dir [ -resource-dir=]: '" << HSOpts.ResourceDir << "'\n";
691       Out.indent(4) << "Module Cache: '" << SpecificModuleCachePath << "'\n";
692       DUMP_BOOLEAN(HSOpts.UseBuiltinIncludes,
693                    "Use builtin include directories [-nobuiltininc]");
694       DUMP_BOOLEAN(HSOpts.UseStandardSystemIncludes,
695                    "Use standard system include directories [-nostdinc]");
696       DUMP_BOOLEAN(HSOpts.UseStandardCXXIncludes,
697                    "Use standard C++ include directories [-nostdinc++]");
698       DUMP_BOOLEAN(HSOpts.UseLibcxx,
699                    "Use libc++ (rather than libstdc++) [-stdlib=]");
700       return false;
701     }
702 
703     bool ReadHeaderSearchPaths(const HeaderSearchOptions &HSOpts,
704                                bool Complain) override {
705       Out.indent(2) << "Header search paths:\n";
706       Out.indent(4) << "User entries:\n";
707       for (const auto &Entry : HSOpts.UserEntries)
708         Out.indent(6) << Entry.Path << "\n";
709       Out.indent(4) << "System header prefixes:\n";
710       for (const auto &Prefix : HSOpts.SystemHeaderPrefixes)
711         Out.indent(6) << Prefix.Prefix << "\n";
712       Out.indent(4) << "VFS overlay files:\n";
713       for (const auto &Overlay : HSOpts.VFSOverlayFiles)
714         Out.indent(6) << Overlay << "\n";
715       return false;
716     }
717 
718     bool ReadPreprocessorOptions(const PreprocessorOptions &PPOpts,
719                                  bool ReadMacros, bool Complain,
720                                  std::string &SuggestedPredefines) override {
721       Out.indent(2) << "Preprocessor options:\n";
722       DUMP_BOOLEAN(PPOpts.UsePredefines,
723                    "Uses compiler/target-specific predefines [-undef]");
724       DUMP_BOOLEAN(PPOpts.DetailedRecord,
725                    "Uses detailed preprocessing record (for indexing)");
726 
727       if (ReadMacros) {
728         Out.indent(4) << "Predefined macros:\n";
729       }
730 
731       for (std::vector<std::pair<std::string, bool/*isUndef*/> >::const_iterator
732              I = PPOpts.Macros.begin(), IEnd = PPOpts.Macros.end();
733            I != IEnd; ++I) {
734         Out.indent(6);
735         if (I->second)
736           Out << "-U";
737         else
738           Out << "-D";
739         Out << I->first << "\n";
740       }
741       return false;
742     }
743 
744     /// Indicates that a particular module file extension has been read.
745     void readModuleFileExtension(
746            const ModuleFileExtensionMetadata &Metadata) override {
747       Out.indent(2) << "Module file extension '"
748                     << Metadata.BlockName << "' " << Metadata.MajorVersion
749                     << "." << Metadata.MinorVersion;
750       if (!Metadata.UserInfo.empty()) {
751         Out << ": ";
752         Out.write_escaped(Metadata.UserInfo);
753       }
754 
755       Out << "\n";
756     }
757 
758     /// Tells the \c ASTReaderListener that we want to receive the
759     /// input files of the AST file via \c visitInputFile.
760     bool needsInputFileVisitation() override { return true; }
761 
762     /// Tells the \c ASTReaderListener that we want to receive the
763     /// input files of the AST file via \c visitInputFile.
764     bool needsSystemInputFileVisitation() override { return true; }
765 
766     /// Indicates that the AST file contains particular input file.
767     ///
768     /// \returns true to continue receiving the next input file, false to stop.
769     bool visitInputFile(StringRef Filename, bool isSystem,
770                         bool isOverridden, bool isExplicitModule) override {
771 
772       Out.indent(2) << "Input file: " << Filename;
773 
774       if (isSystem || isOverridden || isExplicitModule) {
775         Out << " [";
776         if (isSystem) {
777           Out << "System";
778           if (isOverridden || isExplicitModule)
779             Out << ", ";
780         }
781         if (isOverridden) {
782           Out << "Overridden";
783           if (isExplicitModule)
784             Out << ", ";
785         }
786         if (isExplicitModule)
787           Out << "ExplicitModule";
788 
789         Out << "]";
790       }
791 
792       Out << "\n";
793 
794       return true;
795     }
796 
797     /// Returns true if this \c ASTReaderListener wants to receive the
798     /// imports of the AST file via \c visitImport, false otherwise.
799     bool needsImportVisitation() const override { return true; }
800 
801     /// If needsImportVisitation returns \c true, this is called for each
802     /// AST file imported by this AST file.
803     void visitImport(StringRef ModuleName, StringRef Filename) override {
804       Out.indent(2) << "Imports module '" << ModuleName
805                     << "': " << Filename.str() << "\n";
806     }
807 #undef DUMP_BOOLEAN
808   };
809 }
810 
811 bool DumpModuleInfoAction::BeginInvocation(CompilerInstance &CI) {
812   // The Object file reader also supports raw ast files and there is no point in
813   // being strict about the module file format in -module-file-info mode.
814   CI.getHeaderSearchOpts().ModuleFormat = "obj";
815   return true;
816 }
817 
818 static StringRef ModuleKindName(Module::ModuleKind MK) {
819   switch (MK) {
820   case Module::ModuleMapModule:
821     return "Module Map Module";
822   case Module::ModuleInterfaceUnit:
823     return "Interface Unit";
824   case Module::ModuleImplementationUnit:
825     return "Implementation Unit";
826   case Module::ModulePartitionInterface:
827     return "Partition Interface";
828   case Module::ModulePartitionImplementation:
829     return "Partition Implementation";
830   case Module::ModuleHeaderUnit:
831     return "Header Unit";
832   case Module::ExplicitGlobalModuleFragment:
833     return "Global Module Fragment";
834   case Module::ImplicitGlobalModuleFragment:
835     return "Implicit Module Fragment";
836   case Module::PrivateModuleFragment:
837     return "Private Module Fragment";
838   }
839   llvm_unreachable("unknown module kind!");
840 }
841 
842 void DumpModuleInfoAction::ExecuteAction() {
843   assert(isCurrentFileAST() && "dumping non-AST?");
844   // Set up the output file.
845   CompilerInstance &CI = getCompilerInstance();
846   StringRef OutputFileName = CI.getFrontendOpts().OutputFile;
847   if (!OutputFileName.empty() && OutputFileName != "-") {
848     std::error_code EC;
849     OutputStream.reset(new llvm::raw_fd_ostream(
850         OutputFileName.str(), EC, llvm::sys::fs::OF_TextWithCRLF));
851   }
852   llvm::raw_ostream &Out = OutputStream ? *OutputStream : llvm::outs();
853 
854   Out << "Information for module file '" << getCurrentFile() << "':\n";
855   auto &FileMgr = CI.getFileManager();
856   auto Buffer = FileMgr.getBufferForFile(getCurrentFile());
857   StringRef Magic = (*Buffer)->getMemBufferRef().getBuffer();
858   bool IsRaw = Magic.starts_with("CPCH");
859   Out << "  Module format: " << (IsRaw ? "raw" : "obj") << "\n";
860 
861   Preprocessor &PP = CI.getPreprocessor();
862   DumpModuleInfoListener Listener(Out);
863   HeaderSearchOptions &HSOpts = PP.getHeaderSearchInfo().getHeaderSearchOpts();
864 
865   // The FrontendAction::BeginSourceFile () method loads the AST so that much
866   // of the information is already available and modules should have been
867   // loaded.
868 
869   const LangOptions &LO = getCurrentASTUnit().getLangOpts();
870   if (LO.CPlusPlusModules && !LO.CurrentModule.empty()) {
871     ASTReader *R = getCurrentASTUnit().getASTReader().get();
872     unsigned SubModuleCount = R->getTotalNumSubmodules();
873     serialization::ModuleFile &MF = R->getModuleManager().getPrimaryModule();
874     Out << "  ====== C++20 Module structure ======\n";
875 
876     if (MF.ModuleName != LO.CurrentModule)
877       Out << "  Mismatched module names : " << MF.ModuleName << " and "
878           << LO.CurrentModule << "\n";
879 
880     struct SubModInfo {
881       unsigned Idx;
882       Module *Mod;
883       Module::ModuleKind Kind;
884       std::string &Name;
885       bool Seen;
886     };
887     std::map<std::string, SubModInfo> SubModMap;
888     auto PrintSubMapEntry = [&](std::string Name, Module::ModuleKind Kind) {
889       Out << "    " << ModuleKindName(Kind) << " '" << Name << "'";
890       auto I = SubModMap.find(Name);
891       if (I == SubModMap.end())
892         Out << " was not found in the sub modules!\n";
893       else {
894         I->second.Seen = true;
895         Out << " is at index #" << I->second.Idx << "\n";
896       }
897     };
898     Module *Primary = nullptr;
899     for (unsigned Idx = 0; Idx <= SubModuleCount; ++Idx) {
900       Module *M = R->getModule(Idx);
901       if (!M)
902         continue;
903       if (M->Name == LO.CurrentModule) {
904         Primary = M;
905         Out << "  " << ModuleKindName(M->Kind) << " '" << LO.CurrentModule
906             << "' is the Primary Module at index #" << Idx << "\n";
907         SubModMap.insert({M->Name, {Idx, M, M->Kind, M->Name, true}});
908       } else
909         SubModMap.insert({M->Name, {Idx, M, M->Kind, M->Name, false}});
910     }
911     if (Primary) {
912       if (!Primary->submodules().empty())
913         Out << "   Sub Modules:\n";
914       for (auto *MI : Primary->submodules()) {
915         PrintSubMapEntry(MI->Name, MI->Kind);
916       }
917       if (!Primary->Imports.empty())
918         Out << "   Imports:\n";
919       for (auto *IMP : Primary->Imports) {
920         PrintSubMapEntry(IMP->Name, IMP->Kind);
921       }
922       if (!Primary->Exports.empty())
923         Out << "   Exports:\n";
924       for (unsigned MN = 0, N = Primary->Exports.size(); MN != N; ++MN) {
925         if (Module *M = Primary->Exports[MN].getPointer()) {
926           PrintSubMapEntry(M->Name, M->Kind);
927         }
928       }
929     }
930 
931     // Emit the macro definitions in the module file so that we can know how
932     // much definitions in the module file quickly.
933     // TODO: Emit the macro definition bodies completely.
934     if (auto FilteredMacros = llvm::make_filter_range(
935             R->getPreprocessor().macros(),
936             [](const auto &Macro) { return Macro.first->isFromAST(); });
937         !FilteredMacros.empty()) {
938       Out << "   Macro Definitions:\n";
939       for (/*<IdentifierInfo *, MacroState> pair*/ const auto &Macro :
940            FilteredMacros)
941         Out << "     " << Macro.first->getName() << "\n";
942     }
943 
944     // Now let's print out any modules we did not see as part of the Primary.
945     for (const auto &SM : SubModMap) {
946       if (!SM.second.Seen && SM.second.Mod) {
947         Out << "  " << ModuleKindName(SM.second.Kind) << " '" << SM.first
948             << "' at index #" << SM.second.Idx
949             << " has no direct reference in the Primary\n";
950       }
951     }
952     Out << "  ====== ======\n";
953   }
954 
955   // The reminder of the output is produced from the listener as the AST
956   // FileCcontrolBlock is (re-)parsed.
957   ASTReader::readASTFileControlBlock(
958       getCurrentFile(), FileMgr, CI.getModuleCache(),
959       CI.getPCHContainerReader(),
960       /*FindModuleFileExtensions=*/true, Listener,
961       HSOpts.ModulesValidateDiagnosticOptions);
962 }
963 
964 //===----------------------------------------------------------------------===//
965 // Preprocessor Actions
966 //===----------------------------------------------------------------------===//
967 
968 void DumpRawTokensAction::ExecuteAction() {
969   Preprocessor &PP = getCompilerInstance().getPreprocessor();
970   SourceManager &SM = PP.getSourceManager();
971 
972   // Start lexing the specified input file.
973   llvm::MemoryBufferRef FromFile = SM.getBufferOrFake(SM.getMainFileID());
974   Lexer RawLex(SM.getMainFileID(), FromFile, SM, PP.getLangOpts());
975   RawLex.SetKeepWhitespaceMode(true);
976 
977   Token RawTok;
978   RawLex.LexFromRawLexer(RawTok);
979   while (RawTok.isNot(tok::eof)) {
980     PP.DumpToken(RawTok, true);
981     llvm::errs() << "\n";
982     RawLex.LexFromRawLexer(RawTok);
983   }
984 }
985 
986 void DumpTokensAction::ExecuteAction() {
987   Preprocessor &PP = getCompilerInstance().getPreprocessor();
988   // Start preprocessing the specified input file.
989   Token Tok;
990   PP.EnterMainSourceFile();
991   do {
992     PP.Lex(Tok);
993     PP.DumpToken(Tok, true);
994     llvm::errs() << "\n";
995   } while (Tok.isNot(tok::eof));
996 }
997 
998 void PreprocessOnlyAction::ExecuteAction() {
999   Preprocessor &PP = getCompilerInstance().getPreprocessor();
1000 
1001   // Ignore unknown pragmas.
1002   PP.IgnorePragmas();
1003 
1004   Token Tok;
1005   // Start parsing the specified input file.
1006   PP.EnterMainSourceFile();
1007   do {
1008     PP.Lex(Tok);
1009   } while (Tok.isNot(tok::eof));
1010 }
1011 
1012 void PrintPreprocessedAction::ExecuteAction() {
1013   CompilerInstance &CI = getCompilerInstance();
1014   // Output file may need to be set to 'Binary', to avoid converting Unix style
1015   // line feeds (<LF>) to Microsoft style line feeds (<CR><LF>) on Windows.
1016   //
1017   // Look to see what type of line endings the file uses. If there's a
1018   // CRLF, then we won't open the file up in binary mode. If there is
1019   // just an LF or CR, then we will open the file up in binary mode.
1020   // In this fashion, the output format should match the input format, unless
1021   // the input format has inconsistent line endings.
1022   //
1023   // This should be a relatively fast operation since most files won't have
1024   // all of their source code on a single line. However, that is still a
1025   // concern, so if we scan for too long, we'll just assume the file should
1026   // be opened in binary mode.
1027 
1028   bool BinaryMode = false;
1029   if (llvm::Triple(LLVM_HOST_TRIPLE).isOSWindows()) {
1030     BinaryMode = true;
1031     const SourceManager &SM = CI.getSourceManager();
1032     if (std::optional<llvm::MemoryBufferRef> Buffer =
1033             SM.getBufferOrNone(SM.getMainFileID())) {
1034       const char *cur = Buffer->getBufferStart();
1035       const char *end = Buffer->getBufferEnd();
1036       const char *next = (cur != end) ? cur + 1 : end;
1037 
1038       // Limit ourselves to only scanning 256 characters into the source
1039       // file.  This is mostly a check in case the file has no
1040       // newlines whatsoever.
1041       if (end - cur > 256)
1042         end = cur + 256;
1043 
1044       while (next < end) {
1045         if (*cur == 0x0D) {  // CR
1046           if (*next == 0x0A) // CRLF
1047             BinaryMode = false;
1048 
1049           break;
1050         } else if (*cur == 0x0A) // LF
1051           break;
1052 
1053         ++cur;
1054         ++next;
1055       }
1056     }
1057   }
1058 
1059   std::unique_ptr<raw_ostream> OS =
1060       CI.createDefaultOutputFile(BinaryMode, getCurrentFileOrBufferName());
1061   if (!OS) return;
1062 
1063   // If we're preprocessing a module map, start by dumping the contents of the
1064   // module itself before switching to the input buffer.
1065   auto &Input = getCurrentInput();
1066   if (Input.getKind().getFormat() == InputKind::ModuleMap) {
1067     if (Input.isFile()) {
1068       (*OS) << "# 1 \"";
1069       OS->write_escaped(Input.getFile());
1070       (*OS) << "\"\n";
1071     }
1072     getCurrentModule()->print(*OS);
1073     (*OS) << "#pragma clang module contents\n";
1074   }
1075 
1076   DoPrintPreprocessedInput(CI.getPreprocessor(), OS.get(),
1077                            CI.getPreprocessorOutputOpts());
1078 }
1079 
1080 void PrintPreambleAction::ExecuteAction() {
1081   switch (getCurrentFileKind().getLanguage()) {
1082   case Language::C:
1083   case Language::CXX:
1084   case Language::ObjC:
1085   case Language::ObjCXX:
1086   case Language::OpenCL:
1087   case Language::OpenCLCXX:
1088   case Language::CUDA:
1089   case Language::HIP:
1090   case Language::HLSL:
1091   case Language::CIR:
1092     break;
1093 
1094   case Language::Unknown:
1095   case Language::Asm:
1096   case Language::LLVM_IR:
1097   case Language::RenderScript:
1098     // We can't do anything with these.
1099     return;
1100   }
1101 
1102   // We don't expect to find any #include directives in a preprocessed input.
1103   if (getCurrentFileKind().isPreprocessed())
1104     return;
1105 
1106   CompilerInstance &CI = getCompilerInstance();
1107   auto Buffer = CI.getFileManager().getBufferForFile(getCurrentFile());
1108   if (Buffer) {
1109     unsigned Preamble =
1110         Lexer::ComputePreamble((*Buffer)->getBuffer(), CI.getLangOpts()).Size;
1111     llvm::outs().write((*Buffer)->getBufferStart(), Preamble);
1112   }
1113 }
1114 
1115 void DumpCompilerOptionsAction::ExecuteAction() {
1116   CompilerInstance &CI = getCompilerInstance();
1117   std::unique_ptr<raw_ostream> OSP =
1118       CI.createDefaultOutputFile(false, getCurrentFile());
1119   if (!OSP)
1120     return;
1121 
1122   raw_ostream &OS = *OSP;
1123   const Preprocessor &PP = CI.getPreprocessor();
1124   const LangOptions &LangOpts = PP.getLangOpts();
1125 
1126   // FIXME: Rather than manually format the JSON (which is awkward due to
1127   // needing to remove trailing commas), this should make use of a JSON library.
1128   // FIXME: Instead of printing enums as an integral value and specifying the
1129   // type as a separate field, use introspection to print the enumerator.
1130 
1131   OS << "{\n";
1132   OS << "\n\"features\" : [\n";
1133   {
1134     llvm::SmallString<128> Str;
1135 #define FEATURE(Name, Predicate)                                               \
1136   ("\t{\"" #Name "\" : " + llvm::Twine(Predicate ? "true" : "false") + "},\n") \
1137       .toVector(Str);
1138 #include "clang/Basic/Features.def"
1139 #undef FEATURE
1140     // Remove the newline and comma from the last entry to ensure this remains
1141     // valid JSON.
1142     OS << Str.substr(0, Str.size() - 2);
1143   }
1144   OS << "\n],\n";
1145 
1146   OS << "\n\"extensions\" : [\n";
1147   {
1148     llvm::SmallString<128> Str;
1149 #define EXTENSION(Name, Predicate)                                             \
1150   ("\t{\"" #Name "\" : " + llvm::Twine(Predicate ? "true" : "false") + "},\n") \
1151       .toVector(Str);
1152 #include "clang/Basic/Features.def"
1153 #undef EXTENSION
1154     // Remove the newline and comma from the last entry to ensure this remains
1155     // valid JSON.
1156     OS << Str.substr(0, Str.size() - 2);
1157   }
1158   OS << "\n]\n";
1159 
1160   OS << "}";
1161 }
1162 
1163 void PrintDependencyDirectivesSourceMinimizerAction::ExecuteAction() {
1164   CompilerInstance &CI = getCompilerInstance();
1165   SourceManager &SM = CI.getPreprocessor().getSourceManager();
1166   llvm::MemoryBufferRef FromFile = SM.getBufferOrFake(SM.getMainFileID());
1167 
1168   llvm::SmallVector<dependency_directives_scan::Token, 16> Tokens;
1169   llvm::SmallVector<dependency_directives_scan::Directive, 32> Directives;
1170   if (scanSourceForDependencyDirectives(
1171           FromFile.getBuffer(), Tokens, Directives, &CI.getDiagnostics(),
1172           SM.getLocForStartOfFile(SM.getMainFileID()))) {
1173     assert(CI.getDiagnostics().hasErrorOccurred() &&
1174            "no errors reported for failure");
1175 
1176     // Preprocess the source when verifying the diagnostics to capture the
1177     // 'expected' comments.
1178     if (CI.getDiagnosticOpts().VerifyDiagnostics) {
1179       // Make sure we don't emit new diagnostics!
1180       CI.getDiagnostics().setSuppressAllDiagnostics(true);
1181       Preprocessor &PP = getCompilerInstance().getPreprocessor();
1182       PP.EnterMainSourceFile();
1183       Token Tok;
1184       do {
1185         PP.Lex(Tok);
1186       } while (Tok.isNot(tok::eof));
1187     }
1188     return;
1189   }
1190   printDependencyDirectivesAsSource(FromFile.getBuffer(), Directives,
1191                                     llvm::outs());
1192 }
1193 
1194 void GetDependenciesByModuleNameAction::ExecuteAction() {
1195   CompilerInstance &CI = getCompilerInstance();
1196   AdjustCI(CI);
1197 
1198   Preprocessor &PP = CI.getPreprocessor();
1199   SourceManager &SM = PP.getSourceManager();
1200   FileID MainFileID = SM.getMainFileID();
1201   SourceLocation FileStart = SM.getLocForStartOfFile(MainFileID);
1202   SmallVector<std::pair<IdentifierInfo *, SourceLocation>, 2> Path;
1203   IdentifierInfo *ModuleID = PP.getIdentifierInfo(ModuleName);
1204   Path.push_back(std::make_pair(ModuleID, FileStart));
1205   auto ModResult = CI.loadModule(FileStart, Path, Module::Hidden, false);
1206   PPCallbacks *CB = PP.getPPCallbacks();
1207   CB->moduleImport(SourceLocation(), Path, ModResult);
1208 }
1209