xref: /freebsd/contrib/llvm-project/llvm/tools/llvm-cov/CodeCoverage.cpp (revision 770cf0a5f02dc8983a89c6568d741fbc25baa999)
1 //===- CodeCoverage.cpp - Coverage tool based on profiling instrumentation-===//
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 // The 'CodeCoverageTool' class implements a command line tool to analyze and
10 // report coverage information using the profiling instrumentation and code
11 // coverage mapping.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "CoverageExporterJson.h"
16 #include "CoverageExporterLcov.h"
17 #include "CoverageFilters.h"
18 #include "CoverageReport.h"
19 #include "CoverageSummaryInfo.h"
20 #include "CoverageViewOptions.h"
21 #include "RenderingSupport.h"
22 #include "SourceCoverageView.h"
23 #include "llvm/ADT/SmallString.h"
24 #include "llvm/ADT/StringRef.h"
25 #include "llvm/Debuginfod/BuildIDFetcher.h"
26 #include "llvm/Debuginfod/Debuginfod.h"
27 #include "llvm/Debuginfod/HTTPClient.h"
28 #include "llvm/Object/BuildID.h"
29 #include "llvm/ProfileData/Coverage/CoverageMapping.h"
30 #include "llvm/ProfileData/InstrProfReader.h"
31 #include "llvm/Support/CommandLine.h"
32 #include "llvm/Support/FileSystem.h"
33 #include "llvm/Support/Format.h"
34 #include "llvm/Support/MemoryBuffer.h"
35 #include "llvm/Support/Path.h"
36 #include "llvm/Support/Process.h"
37 #include "llvm/Support/Program.h"
38 #include "llvm/Support/ScopedPrinter.h"
39 #include "llvm/Support/SpecialCaseList.h"
40 #include "llvm/Support/ThreadPool.h"
41 #include "llvm/Support/Threading.h"
42 #include "llvm/Support/ToolOutputFile.h"
43 #include "llvm/Support/VirtualFileSystem.h"
44 #include "llvm/TargetParser/Triple.h"
45 
46 #include <functional>
47 #include <map>
48 #include <optional>
49 #include <system_error>
50 
51 using namespace llvm;
52 using namespace coverage;
53 
54 void exportCoverageDataToJson(const coverage::CoverageMapping &CoverageMapping,
55                               const CoverageViewOptions &Options,
56                               raw_ostream &OS);
57 
58 namespace {
59 /// The implementation of the coverage tool.
60 class CodeCoverageTool {
61 public:
62   enum Command {
63     /// The show command.
64     Show,
65     /// The report command.
66     Report,
67     /// The export command.
68     Export
69   };
70 
71   int run(Command Cmd, int argc, const char **argv);
72 
73 private:
74   /// Print the error message to the error output stream.
75   void error(const Twine &Message, StringRef Whence = "");
76 
77   /// Print the warning message to the error output stream.
78   void warning(const Twine &Message, StringRef Whence = "");
79 
80   /// Convert \p Path into an absolute path and append it to the list
81   /// of collected paths.
82   void addCollectedPath(const std::string &Path);
83 
84   /// If \p Path is a regular file, collect the path. If it's a
85   /// directory, recursively collect all of the paths within the directory.
86   void collectPaths(const std::string &Path);
87 
88   /// Check if the two given files are the same file.
89   bool isEquivalentFile(StringRef FilePath1, StringRef FilePath2);
90 
91   /// Retrieve a file status with a cache.
92   std::optional<sys::fs::file_status> getFileStatus(StringRef FilePath);
93 
94   /// Return a memory buffer for the given source file.
95   ErrorOr<const MemoryBuffer &> getSourceFile(StringRef SourceFile);
96 
97   /// Create source views for the expansions of the view.
98   void attachExpansionSubViews(SourceCoverageView &View,
99                                ArrayRef<ExpansionRecord> Expansions,
100                                const CoverageMapping &Coverage);
101 
102   /// Create source views for the branches of the view.
103   void attachBranchSubViews(SourceCoverageView &View,
104                             ArrayRef<CountedRegion> Branches);
105 
106   /// Create source views for the MCDC records.
107   void attachMCDCSubViews(SourceCoverageView &View,
108                           ArrayRef<MCDCRecord> MCDCRecords);
109 
110   /// Create the source view of a particular function.
111   std::unique_ptr<SourceCoverageView>
112   createFunctionView(const FunctionRecord &Function,
113                      const CoverageMapping &Coverage);
114 
115   /// Create the main source view of a particular source file.
116   std::unique_ptr<SourceCoverageView>
117   createSourceFileView(StringRef SourceFile, const CoverageMapping &Coverage);
118 
119   /// Load the coverage mapping data. Return nullptr if an error occurred.
120   std::unique_ptr<CoverageMapping> load();
121 
122   /// Create a mapping from files in the Coverage data to local copies
123   /// (path-equivalence).
124   void remapPathNames(const CoverageMapping &Coverage);
125 
126   /// Remove input source files which aren't mapped by \p Coverage.
127   void removeUnmappedInputs(const CoverageMapping &Coverage);
128 
129   /// If a demangler is available, demangle all symbol names.
130   void demangleSymbols(const CoverageMapping &Coverage);
131 
132   /// Write out a source file view to the filesystem.
133   void writeSourceFileView(StringRef SourceFile, CoverageMapping *Coverage,
134                            CoveragePrinter *Printer, bool ShowFilenames);
135 
136   typedef llvm::function_ref<int(int, const char **)> CommandLineParserType;
137 
138   int doShow(int argc, const char **argv,
139              CommandLineParserType commandLineParser);
140 
141   int doReport(int argc, const char **argv,
142                CommandLineParserType commandLineParser);
143 
144   int doExport(int argc, const char **argv,
145                CommandLineParserType commandLineParser);
146 
147   std::vector<StringRef> ObjectFilenames;
148   CoverageViewOptions ViewOpts;
149   CoverageFiltersMatchAll Filters;
150   CoverageFilters IgnoreFilenameFilters;
151 
152   /// True if InputSourceFiles are provided.
153   bool HadSourceFiles = false;
154 
155   /// The path to the indexed profile.
156   std::optional<std::string> PGOFilename;
157 
158   /// A list of input source files.
159   std::vector<std::string> SourceFiles;
160 
161   /// In -path-equivalence mode, this maps the absolute paths from the coverage
162   /// mapping data to the input source files.
163   StringMap<std::string> RemappedFilenames;
164 
165   /// The coverage data path to be remapped from, and the source path to be
166   /// remapped to, when using -path-equivalence.
167   std::optional<std::vector<std::pair<std::string, std::string>>>
168       PathRemappings;
169 
170   /// File status cache used when finding the same file.
171   StringMap<std::optional<sys::fs::file_status>> FileStatusCache;
172 
173   /// The architecture the coverage mapping data targets.
174   std::vector<StringRef> CoverageArches;
175 
176   /// A cache for demangled symbols.
177   DemangleCache DC;
178 
179   /// A lock which guards printing to stderr.
180   std::mutex ErrsLock;
181 
182   /// A container for input source file buffers.
183   std::mutex LoadedSourceFilesLock;
184   std::vector<std::pair<std::string, std::unique_ptr<MemoryBuffer>>>
185       LoadedSourceFiles;
186 
187   /// Allowlist from -name-allowlist to be used for filtering.
188   std::unique_ptr<SpecialCaseList> NameAllowlist;
189 
190   std::unique_ptr<object::BuildIDFetcher> BIDFetcher;
191 
192   bool CheckBinaryIDs;
193 };
194 }
195 
196 static std::string getErrorString(const Twine &Message, StringRef Whence,
197                                   bool Warning) {
198   std::string Str = (Warning ? "warning" : "error");
199   Str += ": ";
200   if (!Whence.empty())
201     Str += Whence.str() + ": ";
202   Str += Message.str() + "\n";
203   return Str;
204 }
205 
206 void CodeCoverageTool::error(const Twine &Message, StringRef Whence) {
207   std::unique_lock<std::mutex> Guard{ErrsLock};
208   ViewOpts.colored_ostream(errs(), raw_ostream::RED)
209       << getErrorString(Message, Whence, false);
210 }
211 
212 void CodeCoverageTool::warning(const Twine &Message, StringRef Whence) {
213   std::unique_lock<std::mutex> Guard{ErrsLock};
214   ViewOpts.colored_ostream(errs(), raw_ostream::RED)
215       << getErrorString(Message, Whence, true);
216 }
217 
218 void CodeCoverageTool::addCollectedPath(const std::string &Path) {
219   SmallString<128> EffectivePath(Path);
220   if (std::error_code EC = sys::fs::make_absolute(EffectivePath)) {
221     error(EC.message(), Path);
222     return;
223   }
224   sys::path::remove_dots(EffectivePath, /*remove_dot_dot=*/true);
225   if (!IgnoreFilenameFilters.matchesFilename(EffectivePath))
226     SourceFiles.emplace_back(EffectivePath.str());
227   HadSourceFiles = !SourceFiles.empty();
228 }
229 
230 void CodeCoverageTool::collectPaths(const std::string &Path) {
231   llvm::sys::fs::file_status Status;
232   llvm::sys::fs::status(Path, Status);
233   if (!llvm::sys::fs::exists(Status)) {
234     if (PathRemappings)
235       addCollectedPath(Path);
236     else
237       warning("Source file doesn't exist, proceeded by ignoring it.", Path);
238     return;
239   }
240 
241   if (llvm::sys::fs::is_regular_file(Status)) {
242     addCollectedPath(Path);
243     return;
244   }
245 
246   if (llvm::sys::fs::is_directory(Status)) {
247     std::error_code EC;
248     for (llvm::sys::fs::recursive_directory_iterator F(Path, EC), E;
249          F != E; F.increment(EC)) {
250 
251       auto Status = F->status();
252       if (!Status) {
253         warning(Status.getError().message(), F->path());
254         continue;
255       }
256 
257       if (Status->type() == llvm::sys::fs::file_type::regular_file)
258         addCollectedPath(F->path());
259     }
260   }
261 }
262 
263 std::optional<sys::fs::file_status>
264 CodeCoverageTool::getFileStatus(StringRef FilePath) {
265   auto It = FileStatusCache.try_emplace(FilePath);
266   auto &CachedStatus = It.first->getValue();
267   if (!It.second)
268     return CachedStatus;
269 
270   sys::fs::file_status Status;
271   if (!sys::fs::status(FilePath, Status))
272     CachedStatus = Status;
273   return CachedStatus;
274 }
275 
276 bool CodeCoverageTool::isEquivalentFile(StringRef FilePath1,
277                                         StringRef FilePath2) {
278   auto Status1 = getFileStatus(FilePath1);
279   auto Status2 = getFileStatus(FilePath2);
280   return Status1 && Status2 && sys::fs::equivalent(*Status1, *Status2);
281 }
282 
283 ErrorOr<const MemoryBuffer &>
284 CodeCoverageTool::getSourceFile(StringRef SourceFile) {
285   // If we've remapped filenames, look up the real location for this file.
286   std::unique_lock<std::mutex> Guard{LoadedSourceFilesLock};
287   if (!RemappedFilenames.empty()) {
288     auto Loc = RemappedFilenames.find(SourceFile);
289     if (Loc != RemappedFilenames.end())
290       SourceFile = Loc->second;
291   }
292   for (const auto &Files : LoadedSourceFiles)
293     if (isEquivalentFile(SourceFile, Files.first))
294       return *Files.second;
295   auto Buffer = MemoryBuffer::getFile(SourceFile);
296   if (auto EC = Buffer.getError()) {
297     error(EC.message(), SourceFile);
298     return EC;
299   }
300   LoadedSourceFiles.emplace_back(std::string(SourceFile),
301                                  std::move(Buffer.get()));
302   return *LoadedSourceFiles.back().second;
303 }
304 
305 void CodeCoverageTool::attachExpansionSubViews(
306     SourceCoverageView &View, ArrayRef<ExpansionRecord> Expansions,
307     const CoverageMapping &Coverage) {
308   if (!ViewOpts.ShowExpandedRegions)
309     return;
310   for (const auto &Expansion : Expansions) {
311     auto ExpansionCoverage = Coverage.getCoverageForExpansion(Expansion);
312     if (ExpansionCoverage.empty())
313       continue;
314     auto SourceBuffer = getSourceFile(ExpansionCoverage.getFilename());
315     if (!SourceBuffer)
316       continue;
317 
318     auto SubViewBranches = ExpansionCoverage.getBranches();
319     auto SubViewExpansions = ExpansionCoverage.getExpansions();
320     auto SubView =
321         SourceCoverageView::create(Expansion.Function.Name, SourceBuffer.get(),
322                                    ViewOpts, std::move(ExpansionCoverage));
323     attachExpansionSubViews(*SubView, SubViewExpansions, Coverage);
324     attachBranchSubViews(*SubView, SubViewBranches);
325     View.addExpansion(Expansion.Region, std::move(SubView));
326   }
327 }
328 
329 void CodeCoverageTool::attachBranchSubViews(SourceCoverageView &View,
330                                             ArrayRef<CountedRegion> Branches) {
331   if (!ViewOpts.ShowBranchCounts && !ViewOpts.ShowBranchPercents)
332     return;
333 
334   const auto *NextBranch = Branches.begin();
335   const auto *EndBranch = Branches.end();
336 
337   // Group branches that have the same line number into the same subview.
338   while (NextBranch != EndBranch) {
339     SmallVector<CountedRegion, 0> ViewBranches;
340     unsigned CurrentLine = NextBranch->LineStart;
341     while (NextBranch != EndBranch && CurrentLine == NextBranch->LineStart)
342       ViewBranches.push_back(*NextBranch++);
343 
344     View.addBranch(CurrentLine, std::move(ViewBranches));
345   }
346 }
347 
348 void CodeCoverageTool::attachMCDCSubViews(SourceCoverageView &View,
349                                           ArrayRef<MCDCRecord> MCDCRecords) {
350   if (!ViewOpts.ShowMCDC)
351     return;
352 
353   const auto *NextRecord = MCDCRecords.begin();
354   const auto *EndRecord = MCDCRecords.end();
355 
356   // Group and process MCDC records that have the same line number into the
357   // same subview.
358   while (NextRecord != EndRecord) {
359     SmallVector<MCDCRecord, 0> ViewMCDCRecords;
360     unsigned CurrentLine = NextRecord->getDecisionRegion().LineEnd;
361     while (NextRecord != EndRecord &&
362            CurrentLine == NextRecord->getDecisionRegion().LineEnd)
363       ViewMCDCRecords.push_back(*NextRecord++);
364 
365     View.addMCDCRecord(CurrentLine, std::move(ViewMCDCRecords));
366   }
367 }
368 
369 std::unique_ptr<SourceCoverageView>
370 CodeCoverageTool::createFunctionView(const FunctionRecord &Function,
371                                      const CoverageMapping &Coverage) {
372   auto FunctionCoverage = Coverage.getCoverageForFunction(Function);
373   if (FunctionCoverage.empty())
374     return nullptr;
375   auto SourceBuffer = getSourceFile(FunctionCoverage.getFilename());
376   if (!SourceBuffer)
377     return nullptr;
378 
379   auto Branches = FunctionCoverage.getBranches();
380   auto Expansions = FunctionCoverage.getExpansions();
381   auto MCDCRecords = FunctionCoverage.getMCDCRecords();
382   auto View = SourceCoverageView::create(DC.demangle(Function.Name),
383                                          SourceBuffer.get(), ViewOpts,
384                                          std::move(FunctionCoverage));
385   attachExpansionSubViews(*View, Expansions, Coverage);
386   attachBranchSubViews(*View, Branches);
387   attachMCDCSubViews(*View, MCDCRecords);
388 
389   return View;
390 }
391 
392 std::unique_ptr<SourceCoverageView>
393 CodeCoverageTool::createSourceFileView(StringRef SourceFile,
394                                        const CoverageMapping &Coverage) {
395   auto SourceBuffer = getSourceFile(SourceFile);
396   if (!SourceBuffer)
397     return nullptr;
398   auto FileCoverage = Coverage.getCoverageForFile(SourceFile);
399   if (FileCoverage.empty())
400     return nullptr;
401 
402   auto Branches = FileCoverage.getBranches();
403   auto Expansions = FileCoverage.getExpansions();
404   auto MCDCRecords = FileCoverage.getMCDCRecords();
405   auto View = SourceCoverageView::create(SourceFile, SourceBuffer.get(),
406                                          ViewOpts, std::move(FileCoverage));
407   attachExpansionSubViews(*View, Expansions, Coverage);
408   attachBranchSubViews(*View, Branches);
409   attachMCDCSubViews(*View, MCDCRecords);
410   if (!ViewOpts.ShowFunctionInstantiations)
411     return View;
412 
413   for (const auto &Group : Coverage.getInstantiationGroups(SourceFile)) {
414     // Skip functions which have a single instantiation.
415     if (Group.size() < 2)
416       continue;
417 
418     for (const FunctionRecord *Function : Group.getInstantiations()) {
419       std::unique_ptr<SourceCoverageView> SubView{nullptr};
420 
421       StringRef Funcname = DC.demangle(Function->Name);
422 
423       if (Function->ExecutionCount > 0) {
424         auto SubViewCoverage = Coverage.getCoverageForFunction(*Function);
425         auto SubViewExpansions = SubViewCoverage.getExpansions();
426         auto SubViewBranches = SubViewCoverage.getBranches();
427         auto SubViewMCDCRecords = SubViewCoverage.getMCDCRecords();
428         SubView = SourceCoverageView::create(
429             Funcname, SourceBuffer.get(), ViewOpts, std::move(SubViewCoverage));
430         attachExpansionSubViews(*SubView, SubViewExpansions, Coverage);
431         attachBranchSubViews(*SubView, SubViewBranches);
432         attachMCDCSubViews(*SubView, SubViewMCDCRecords);
433       }
434 
435       unsigned FileID = Function->CountedRegions.front().FileID;
436       unsigned Line = 0;
437       for (const auto &CR : Function->CountedRegions)
438         if (CR.FileID == FileID)
439           Line = std::max(CR.LineEnd, Line);
440       View->addInstantiation(Funcname, Line, std::move(SubView));
441     }
442   }
443   return View;
444 }
445 
446 static bool modifiedTimeGT(StringRef LHS, StringRef RHS) {
447   sys::fs::file_status Status;
448   if (sys::fs::status(LHS, Status))
449     return false;
450   auto LHSTime = Status.getLastModificationTime();
451   if (sys::fs::status(RHS, Status))
452     return false;
453   auto RHSTime = Status.getLastModificationTime();
454   return LHSTime > RHSTime;
455 }
456 
457 std::unique_ptr<CoverageMapping> CodeCoverageTool::load() {
458   if (PGOFilename) {
459     for (StringRef ObjectFilename : ObjectFilenames)
460       if (modifiedTimeGT(ObjectFilename, PGOFilename.value()))
461         warning("profile data may be out of date - object is newer",
462                 ObjectFilename);
463   }
464   auto FS = vfs::getRealFileSystem();
465   auto CoverageOrErr = CoverageMapping::load(
466       ObjectFilenames, PGOFilename, *FS, CoverageArches,
467       ViewOpts.CompilationDirectory, BIDFetcher.get(), CheckBinaryIDs);
468   if (Error E = CoverageOrErr.takeError()) {
469     error("failed to load coverage: " + toString(std::move(E)));
470     return nullptr;
471   }
472   auto Coverage = std::move(CoverageOrErr.get());
473   unsigned Mismatched = Coverage->getMismatchedCount();
474   if (Mismatched) {
475     warning(Twine(Mismatched) + " functions have mismatched data");
476 
477     if (ViewOpts.Debug) {
478       for (const auto &HashMismatch : Coverage->getHashMismatches())
479         errs() << "hash-mismatch: "
480                << "No profile record found for '" << HashMismatch.first << "'"
481                << " with hash = 0x" << Twine::utohexstr(HashMismatch.second)
482                << '\n';
483     }
484   }
485 
486   remapPathNames(*Coverage);
487 
488   if (!SourceFiles.empty())
489     removeUnmappedInputs(*Coverage);
490 
491   demangleSymbols(*Coverage);
492 
493   return Coverage;
494 }
495 
496 void CodeCoverageTool::remapPathNames(const CoverageMapping &Coverage) {
497   if (!PathRemappings)
498     return;
499 
500   // Convert remapping paths to native paths with trailing separators.
501   auto nativeWithTrailing = [](StringRef Path) -> std::string {
502     if (Path.empty())
503       return "";
504     SmallString<128> NativePath;
505     sys::path::native(Path, NativePath);
506     sys::path::remove_dots(NativePath, true);
507     if (!NativePath.empty() && !sys::path::is_separator(NativePath.back()))
508       NativePath += sys::path::get_separator();
509     return NativePath.c_str();
510   };
511 
512   for (std::pair<std::string, std::string> &PathRemapping : *PathRemappings) {
513     std::string RemapFrom = nativeWithTrailing(PathRemapping.first);
514     std::string RemapTo = nativeWithTrailing(PathRemapping.second);
515 
516     // Create a mapping from coverage data file paths to local paths.
517     for (StringRef Filename : Coverage.getUniqueSourceFiles()) {
518       if (RemappedFilenames.count(Filename) == 1)
519         continue;
520 
521       SmallString<128> NativeFilename;
522       sys::path::native(Filename, NativeFilename);
523       sys::path::remove_dots(NativeFilename, true);
524       if (NativeFilename.starts_with(RemapFrom)) {
525         RemappedFilenames[Filename] =
526             RemapTo + NativeFilename.substr(RemapFrom.size()).str();
527       }
528     }
529   }
530 
531   // Convert input files from local paths to coverage data file paths.
532   StringMap<std::string> InvRemappedFilenames;
533   for (const auto &RemappedFilename : RemappedFilenames)
534     InvRemappedFilenames[RemappedFilename.getValue()] =
535         std::string(RemappedFilename.getKey());
536 
537   for (std::string &Filename : SourceFiles) {
538     SmallString<128> NativeFilename;
539     sys::path::native(Filename, NativeFilename);
540     auto CovFileName = InvRemappedFilenames.find(NativeFilename);
541     if (CovFileName != InvRemappedFilenames.end())
542       Filename = CovFileName->second;
543   }
544 }
545 
546 void CodeCoverageTool::removeUnmappedInputs(const CoverageMapping &Coverage) {
547   std::vector<StringRef> CoveredFiles = Coverage.getUniqueSourceFiles();
548 
549   // The user may have specified source files which aren't in the coverage
550   // mapping. Filter these files away.
551   llvm::erase_if(SourceFiles, [&](const std::string &SF) {
552     return !llvm::binary_search(CoveredFiles, SF);
553   });
554 }
555 
556 void CodeCoverageTool::demangleSymbols(const CoverageMapping &Coverage) {
557   if (!ViewOpts.hasDemangler())
558     return;
559 
560   // Pass function names to the demangler in a temporary file.
561   int InputFD;
562   SmallString<256> InputPath;
563   std::error_code EC =
564       sys::fs::createTemporaryFile("demangle-in", "list", InputFD, InputPath);
565   if (EC) {
566     error(InputPath, EC.message());
567     return;
568   }
569   ToolOutputFile InputTOF{InputPath, InputFD};
570 
571   unsigned NumSymbols = 0;
572   for (const auto &Function : Coverage.getCoveredFunctions()) {
573     InputTOF.os() << Function.Name << '\n';
574     ++NumSymbols;
575   }
576   InputTOF.os().close();
577 
578   // Use another temporary file to store the demangler's output.
579   int OutputFD;
580   SmallString<256> OutputPath;
581   EC = sys::fs::createTemporaryFile("demangle-out", "list", OutputFD,
582                                     OutputPath);
583   if (EC) {
584     error(OutputPath, EC.message());
585     return;
586   }
587   ToolOutputFile OutputTOF{OutputPath, OutputFD};
588   OutputTOF.os().close();
589 
590   // Invoke the demangler.
591   std::vector<StringRef> ArgsV;
592   ArgsV.reserve(ViewOpts.DemanglerOpts.size());
593   llvm::append_range(ArgsV, ViewOpts.DemanglerOpts);
594   std::optional<StringRef> Redirects[] = {
595       InputPath.str(), OutputPath.str(), {""}};
596   std::string ErrMsg;
597   int RC =
598       sys::ExecuteAndWait(ViewOpts.DemanglerOpts[0], ArgsV,
599                           /*env=*/std::nullopt, Redirects, /*secondsToWait=*/0,
600                           /*memoryLimit=*/0, &ErrMsg);
601   if (RC) {
602     error(ErrMsg, ViewOpts.DemanglerOpts[0]);
603     return;
604   }
605 
606   // Parse the demangler's output.
607   auto BufOrError = MemoryBuffer::getFile(OutputPath);
608   if (!BufOrError) {
609     error(OutputPath, BufOrError.getError().message());
610     return;
611   }
612 
613   std::unique_ptr<MemoryBuffer> DemanglerBuf = std::move(*BufOrError);
614 
615   SmallVector<StringRef, 8> Symbols;
616   StringRef DemanglerData = DemanglerBuf->getBuffer();
617   DemanglerData.split(Symbols, '\n', /*MaxSplit=*/NumSymbols,
618                       /*KeepEmpty=*/false);
619   if (Symbols.size() != NumSymbols) {
620     error("demangler did not provide expected number of symbols");
621     return;
622   }
623 
624   // Cache the demangled names.
625   unsigned I = 0;
626   for (const auto &Function : Coverage.getCoveredFunctions())
627     // On Windows, lines in the demangler's output file end with "\r\n".
628     // Splitting by '\n' keeps '\r's, so cut them now.
629     DC.DemangledNames[Function.Name] = std::string(Symbols[I++].rtrim());
630 }
631 
632 void CodeCoverageTool::writeSourceFileView(StringRef SourceFile,
633                                            CoverageMapping *Coverage,
634                                            CoveragePrinter *Printer,
635                                            bool ShowFilenames) {
636   auto View = createSourceFileView(SourceFile, *Coverage);
637   if (!View) {
638     warning("The file '" + SourceFile + "' isn't covered.");
639     return;
640   }
641 
642   auto OSOrErr = Printer->createViewFile(SourceFile, /*InToplevel=*/false);
643   if (Error E = OSOrErr.takeError()) {
644     error("could not create view file!", toString(std::move(E)));
645     return;
646   }
647   auto OS = std::move(OSOrErr.get());
648 
649   View->print(*OS.get(), /*Wholefile=*/true,
650               /*ShowSourceName=*/ShowFilenames,
651               /*ShowTitle=*/ViewOpts.hasOutputDirectory());
652   Printer->closeViewFile(std::move(OS));
653 }
654 
655 int CodeCoverageTool::run(Command Cmd, int argc, const char **argv) {
656   cl::opt<std::string> CovFilename(
657       cl::Positional, cl::desc("Covered executable or object file."));
658 
659   cl::list<std::string> CovFilenames(
660       "object", cl::desc("Coverage executable or object file"));
661 
662   cl::opt<bool> DebugDumpCollectedObjects(
663       "dump-collected-objects", cl::Optional, cl::Hidden,
664       cl::desc("Show the collected coverage object files"));
665 
666   cl::list<std::string> InputSourceFiles("sources", cl::Positional,
667                                          cl::desc("<Source files>"));
668 
669   cl::opt<bool> DebugDumpCollectedPaths(
670       "dump-collected-paths", cl::Optional, cl::Hidden,
671       cl::desc("Show the collected paths to source files"));
672 
673   cl::opt<std::string> PGOFilename(
674       "instr-profile", cl::Optional,
675       cl::desc(
676           "File with the profile data obtained after an instrumented run"));
677 
678   cl::opt<bool> EmptyProfile(
679       "empty-profile", cl::Optional,
680       cl::desc("Use a synthetic profile with no data to generate "
681                "baseline coverage"));
682 
683   cl::list<std::string> Arches(
684       "arch", cl::desc("architectures of the coverage mapping binaries"));
685 
686   cl::opt<bool> DebugDump("dump", cl::Optional,
687                           cl::desc("Show internal debug dump"));
688 
689   cl::list<std::string> DebugFileDirectory(
690       "debug-file-directory",
691       cl::desc("Directories to search for object files by build ID"));
692   cl::opt<bool> Debuginfod(
693       "debuginfod", cl::ZeroOrMore,
694       cl::desc("Use debuginfod to look up object files from profile"),
695       cl::init(canUseDebuginfod()));
696 
697   cl::opt<CoverageViewOptions::OutputFormat> Format(
698       "format", cl::desc("Output format for line-based coverage reports"),
699       cl::values(clEnumValN(CoverageViewOptions::OutputFormat::Text, "text",
700                             "Text output"),
701                  clEnumValN(CoverageViewOptions::OutputFormat::HTML, "html",
702                             "HTML output"),
703                  clEnumValN(CoverageViewOptions::OutputFormat::Lcov, "lcov",
704                             "lcov tracefile output")),
705       cl::init(CoverageViewOptions::OutputFormat::Text));
706 
707   cl::list<std::string> PathRemaps(
708       "path-equivalence", cl::Optional,
709       cl::desc("<from>,<to> Map coverage data paths to local source file "
710                "paths"));
711 
712   cl::OptionCategory FilteringCategory("Function filtering options");
713 
714   cl::list<std::string> NameFilters(
715       "name", cl::Optional,
716       cl::desc("Show code coverage only for functions with the given name"),
717       cl::cat(FilteringCategory));
718 
719   cl::list<std::string> NameFilterFiles(
720       "name-allowlist", cl::Optional,
721       cl::desc("Show code coverage only for functions listed in the given "
722                "file"),
723       cl::cat(FilteringCategory));
724 
725   cl::list<std::string> NameRegexFilters(
726       "name-regex", cl::Optional,
727       cl::desc("Show code coverage only for functions that match the given "
728                "regular expression"),
729       cl::cat(FilteringCategory));
730 
731   cl::list<std::string> IgnoreFilenameRegexFilters(
732       "ignore-filename-regex", cl::Optional,
733       cl::desc("Skip source code files with file paths that match the given "
734                "regular expression"),
735       cl::cat(FilteringCategory));
736 
737   cl::opt<double> RegionCoverageLtFilter(
738       "region-coverage-lt", cl::Optional,
739       cl::desc("Show code coverage only for functions with region coverage "
740                "less than the given threshold"),
741       cl::cat(FilteringCategory));
742 
743   cl::opt<double> RegionCoverageGtFilter(
744       "region-coverage-gt", cl::Optional,
745       cl::desc("Show code coverage only for functions with region coverage "
746                "greater than the given threshold"),
747       cl::cat(FilteringCategory));
748 
749   cl::opt<double> LineCoverageLtFilter(
750       "line-coverage-lt", cl::Optional,
751       cl::desc("Show code coverage only for functions with line coverage less "
752                "than the given threshold"),
753       cl::cat(FilteringCategory));
754 
755   cl::opt<double> LineCoverageGtFilter(
756       "line-coverage-gt", cl::Optional,
757       cl::desc("Show code coverage only for functions with line coverage "
758                "greater than the given threshold"),
759       cl::cat(FilteringCategory));
760 
761   cl::opt<cl::boolOrDefault> UseColor(
762       "use-color", cl::desc("Emit colored output (default=autodetect)"),
763       cl::init(cl::BOU_UNSET));
764 
765   cl::list<std::string> DemanglerOpts(
766       "Xdemangler", cl::desc("<demangler-path>|<demangler-option>"));
767 
768   cl::opt<bool> RegionSummary(
769       "show-region-summary", cl::Optional,
770       cl::desc("Show region statistics in summary table"),
771       cl::init(true));
772 
773   cl::opt<bool> BranchSummary(
774       "show-branch-summary", cl::Optional,
775       cl::desc("Show branch condition statistics in summary table"),
776       cl::init(true));
777 
778   cl::opt<bool> MCDCSummary("show-mcdc-summary", cl::Optional,
779                             cl::desc("Show MCDC statistics in summary table"),
780                             cl::init(false));
781 
782   cl::opt<bool> InstantiationSummary(
783       "show-instantiation-summary", cl::Optional,
784       cl::desc("Show instantiation statistics in summary table"));
785 
786   cl::opt<bool> SummaryOnly(
787       "summary-only", cl::Optional,
788       cl::desc("Export only summary information for each source file"));
789 
790   cl::opt<unsigned> NumThreads(
791       "num-threads", cl::init(0),
792       cl::desc("Number of merge threads to use (default: autodetect)"));
793   cl::alias NumThreadsA("j", cl::desc("Alias for --num-threads"),
794                         cl::aliasopt(NumThreads));
795 
796   cl::opt<std::string> CompilationDirectory(
797       "compilation-dir", cl::init(""),
798       cl::desc("Directory used as a base for relative coverage mapping paths"));
799 
800   cl::opt<bool> CheckBinaryIDs(
801       "check-binary-ids", cl::desc("Fail if an object couldn't be found for a "
802                                    "binary ID in the profile"));
803 
804   auto commandLineParser = [&, this](int argc, const char **argv) -> int {
805     cl::ParseCommandLineOptions(argc, argv, "LLVM code coverage tool\n");
806     ViewOpts.Debug = DebugDump;
807     if (Debuginfod) {
808       HTTPClient::initialize();
809       BIDFetcher = std::make_unique<DebuginfodFetcher>(DebugFileDirectory);
810     } else {
811       BIDFetcher = std::make_unique<object::BuildIDFetcher>(DebugFileDirectory);
812     }
813     this->CheckBinaryIDs = CheckBinaryIDs;
814 
815     if (!PGOFilename.empty() == EmptyProfile) {
816       error(
817           "exactly one of -instr-profile and -empty-profile must be specified");
818       return 1;
819     }
820     if (!PGOFilename.empty()) {
821       this->PGOFilename = std::make_optional(PGOFilename.getValue());
822     }
823 
824     if (!CovFilename.empty())
825       ObjectFilenames.emplace_back(CovFilename);
826     for (const std::string &Filename : CovFilenames)
827       ObjectFilenames.emplace_back(Filename);
828     if (ObjectFilenames.empty() && !Debuginfod && DebugFileDirectory.empty()) {
829       errs() << "No filenames specified!\n";
830       ::exit(1);
831     }
832 
833     if (DebugDumpCollectedObjects) {
834       for (StringRef OF : ObjectFilenames)
835         outs() << OF << '\n';
836       ::exit(0);
837     }
838 
839     ViewOpts.Format = Format;
840     switch (ViewOpts.Format) {
841     case CoverageViewOptions::OutputFormat::Text:
842       ViewOpts.Colors = UseColor == cl::BOU_UNSET
843                             ? sys::Process::StandardOutHasColors()
844                             : UseColor == cl::BOU_TRUE;
845       break;
846     case CoverageViewOptions::OutputFormat::HTML:
847       if (UseColor == cl::BOU_FALSE)
848         errs() << "Color output cannot be disabled when generating html.\n";
849       ViewOpts.Colors = true;
850       break;
851     case CoverageViewOptions::OutputFormat::Lcov:
852       if (UseColor == cl::BOU_TRUE)
853         errs() << "Color output cannot be enabled when generating lcov.\n";
854       ViewOpts.Colors = false;
855       break;
856     }
857 
858     if (!PathRemaps.empty()) {
859       std::vector<std::pair<std::string, std::string>> Remappings;
860 
861       for (const std::string &PathRemap : PathRemaps) {
862         auto EquivPair = StringRef(PathRemap).split(',');
863         if (EquivPair.first.empty() || EquivPair.second.empty()) {
864           error("invalid argument '" + PathRemap +
865                     "', must be in format 'from,to'",
866                 "-path-equivalence");
867           return 1;
868         }
869 
870         Remappings.push_back(
871             {std::string(EquivPair.first), std::string(EquivPair.second)});
872       }
873 
874       PathRemappings = Remappings;
875     }
876 
877     // If a demangler is supplied, check if it exists and register it.
878     if (!DemanglerOpts.empty()) {
879       auto DemanglerPathOrErr = sys::findProgramByName(DemanglerOpts[0]);
880       if (!DemanglerPathOrErr) {
881         error("could not find the demangler!",
882               DemanglerPathOrErr.getError().message());
883         return 1;
884       }
885       DemanglerOpts[0] = *DemanglerPathOrErr;
886       ViewOpts.DemanglerOpts.swap(DemanglerOpts);
887     }
888 
889     // Read in -name-allowlist files.
890     if (!NameFilterFiles.empty()) {
891       std::string SpecialCaseListErr;
892       NameAllowlist = SpecialCaseList::create(
893           NameFilterFiles, *vfs::getRealFileSystem(), SpecialCaseListErr);
894       if (!NameAllowlist)
895         error(SpecialCaseListErr);
896     }
897 
898     // Create the function filters
899     if (!NameFilters.empty() || NameAllowlist || !NameRegexFilters.empty()) {
900       auto NameFilterer = std::make_unique<CoverageFilters>();
901       for (const auto &Name : NameFilters)
902         NameFilterer->push_back(std::make_unique<NameCoverageFilter>(Name));
903       if (NameAllowlist && !NameFilterFiles.empty())
904         NameFilterer->push_back(
905             std::make_unique<NameAllowlistCoverageFilter>(*NameAllowlist));
906       for (const auto &Regex : NameRegexFilters)
907         NameFilterer->push_back(
908             std::make_unique<NameRegexCoverageFilter>(Regex));
909       Filters.push_back(std::move(NameFilterer));
910     }
911 
912     if (RegionCoverageLtFilter.getNumOccurrences() ||
913         RegionCoverageGtFilter.getNumOccurrences() ||
914         LineCoverageLtFilter.getNumOccurrences() ||
915         LineCoverageGtFilter.getNumOccurrences()) {
916       auto StatFilterer = std::make_unique<CoverageFilters>();
917       if (RegionCoverageLtFilter.getNumOccurrences())
918         StatFilterer->push_back(std::make_unique<RegionCoverageFilter>(
919             RegionCoverageFilter::LessThan, RegionCoverageLtFilter));
920       if (RegionCoverageGtFilter.getNumOccurrences())
921         StatFilterer->push_back(std::make_unique<RegionCoverageFilter>(
922             RegionCoverageFilter::GreaterThan, RegionCoverageGtFilter));
923       if (LineCoverageLtFilter.getNumOccurrences())
924         StatFilterer->push_back(std::make_unique<LineCoverageFilter>(
925             LineCoverageFilter::LessThan, LineCoverageLtFilter));
926       if (LineCoverageGtFilter.getNumOccurrences())
927         StatFilterer->push_back(std::make_unique<LineCoverageFilter>(
928             RegionCoverageFilter::GreaterThan, LineCoverageGtFilter));
929       Filters.push_back(std::move(StatFilterer));
930     }
931 
932     // Create the ignore filename filters.
933     for (const auto &RE : IgnoreFilenameRegexFilters)
934       IgnoreFilenameFilters.push_back(
935           std::make_unique<NameRegexCoverageFilter>(RE));
936 
937     if (!Arches.empty()) {
938       for (const std::string &Arch : Arches) {
939         if (Triple(Arch).getArch() == llvm::Triple::ArchType::UnknownArch) {
940           error("unknown architecture: " + Arch);
941           return 1;
942         }
943         CoverageArches.emplace_back(Arch);
944       }
945       if (CoverageArches.size() != 1 &&
946           CoverageArches.size() != ObjectFilenames.size()) {
947         error("number of architectures doesn't match the number of objects");
948         return 1;
949       }
950     }
951 
952     // IgnoreFilenameFilters are applied even when InputSourceFiles specified.
953     for (const std::string &File : InputSourceFiles)
954       collectPaths(File);
955 
956     if (DebugDumpCollectedPaths) {
957       for (const std::string &SF : SourceFiles)
958         outs() << SF << '\n';
959       ::exit(0);
960     }
961 
962     ViewOpts.ShowMCDCSummary = MCDCSummary;
963     ViewOpts.ShowBranchSummary = BranchSummary;
964     ViewOpts.ShowRegionSummary = RegionSummary;
965     ViewOpts.ShowInstantiationSummary = InstantiationSummary;
966     ViewOpts.ExportSummaryOnly = SummaryOnly;
967     ViewOpts.NumThreads = NumThreads;
968     ViewOpts.CompilationDirectory = CompilationDirectory;
969 
970     return 0;
971   };
972 
973   switch (Cmd) {
974   case Show:
975     return doShow(argc, argv, commandLineParser);
976   case Report:
977     return doReport(argc, argv, commandLineParser);
978   case Export:
979     return doExport(argc, argv, commandLineParser);
980   }
981   return 0;
982 }
983 
984 int CodeCoverageTool::doShow(int argc, const char **argv,
985                              CommandLineParserType commandLineParser) {
986 
987   cl::OptionCategory ViewCategory("Viewing options");
988 
989   cl::opt<bool> ShowLineExecutionCounts(
990       "show-line-counts", cl::Optional,
991       cl::desc("Show the execution counts for each line"), cl::init(true),
992       cl::cat(ViewCategory));
993 
994   cl::opt<bool> ShowRegions(
995       "show-regions", cl::Optional,
996       cl::desc("Show the execution counts for each region"),
997       cl::cat(ViewCategory));
998 
999   cl::opt<CoverageViewOptions::BranchOutputType> ShowBranches(
1000       "show-branches", cl::Optional,
1001       cl::desc("Show coverage for branch conditions"), cl::cat(ViewCategory),
1002       cl::values(clEnumValN(CoverageViewOptions::BranchOutputType::Count,
1003                             "count", "Show True/False counts"),
1004                  clEnumValN(CoverageViewOptions::BranchOutputType::Percent,
1005                             "percent", "Show True/False percent")),
1006       cl::init(CoverageViewOptions::BranchOutputType::Off));
1007 
1008   cl::opt<bool> ShowMCDC(
1009       "show-mcdc", cl::Optional,
1010       cl::desc("Show the MCDC Coverage for each applicable boolean expression"),
1011       cl::cat(ViewCategory));
1012 
1013   cl::opt<bool> ShowBestLineRegionsCounts(
1014       "show-line-counts-or-regions", cl::Optional,
1015       cl::desc("Show the execution counts for each line, or the execution "
1016                "counts for each region on lines that have multiple regions"),
1017       cl::cat(ViewCategory));
1018 
1019   cl::opt<bool> ShowExpansions("show-expansions", cl::Optional,
1020                                cl::desc("Show expanded source regions"),
1021                                cl::cat(ViewCategory));
1022 
1023   cl::opt<bool> ShowInstantiations("show-instantiations", cl::Optional,
1024                                    cl::desc("Show function instantiations"),
1025                                    cl::init(true), cl::cat(ViewCategory));
1026 
1027   cl::opt<bool> ShowDirectoryCoverage("show-directory-coverage", cl::Optional,
1028                                       cl::desc("Show directory coverage"),
1029                                       cl::cat(ViewCategory));
1030 
1031   cl::opt<bool> ShowCreatedTime("show-created-time", cl::Optional,
1032                                 cl::desc("Show created time for each page."),
1033                                 cl::init(true), cl::cat(ViewCategory));
1034 
1035   cl::opt<std::string> ShowOutputDirectory(
1036       "output-dir", cl::init(""),
1037       cl::desc("Directory in which coverage information is written out"));
1038   cl::alias ShowOutputDirectoryA("o", cl::desc("Alias for --output-dir"),
1039                                  cl::aliasopt(ShowOutputDirectory));
1040 
1041   cl::opt<bool> BinaryCounters(
1042       "binary-counters", cl::Optional,
1043       cl::desc("Show binary counters (1/0) in lines and branches instead of "
1044                "integer execution counts"),
1045       cl::cat(ViewCategory));
1046 
1047   cl::opt<uint32_t> TabSize(
1048       "tab-size", cl::init(2),
1049       cl::desc(
1050           "Set tab expansion size for html coverage reports (default = 2)"));
1051 
1052   cl::opt<std::string> ProjectTitle(
1053       "project-title", cl::Optional,
1054       cl::desc("Set project title for the coverage report"));
1055 
1056   cl::opt<std::string> CovWatermark(
1057       "coverage-watermark", cl::Optional,
1058       cl::desc("<high>,<low> value indicate thresholds for high and low"
1059                "coverage watermark"));
1060 
1061   auto Err = commandLineParser(argc, argv);
1062   if (Err)
1063     return Err;
1064 
1065   if (ViewOpts.Format == CoverageViewOptions::OutputFormat::Lcov) {
1066     error("lcov format should be used with 'llvm-cov export'.");
1067     return 1;
1068   }
1069 
1070   ViewOpts.HighCovWatermark = 100.0;
1071   ViewOpts.LowCovWatermark = 80.0;
1072   if (!CovWatermark.empty()) {
1073     auto WaterMarkPair = StringRef(CovWatermark).split(',');
1074     if (WaterMarkPair.first.empty() || WaterMarkPair.second.empty()) {
1075       error("invalid argument '" + CovWatermark +
1076                 "', must be in format 'high,low'",
1077             "-coverage-watermark");
1078       return 1;
1079     }
1080 
1081     char *EndPointer = nullptr;
1082     ViewOpts.HighCovWatermark =
1083         strtod(WaterMarkPair.first.begin(), &EndPointer);
1084     if (EndPointer != WaterMarkPair.first.end()) {
1085       error("invalid number '" + WaterMarkPair.first +
1086                 "', invalid value for 'high'",
1087             "-coverage-watermark");
1088       return 1;
1089     }
1090 
1091     ViewOpts.LowCovWatermark =
1092         strtod(WaterMarkPair.second.begin(), &EndPointer);
1093     if (EndPointer != WaterMarkPair.second.end()) {
1094       error("invalid number '" + WaterMarkPair.second +
1095                 "', invalid value for 'low'",
1096             "-coverage-watermark");
1097       return 1;
1098     }
1099 
1100     if (ViewOpts.HighCovWatermark > 100 || ViewOpts.LowCovWatermark < 0 ||
1101         ViewOpts.HighCovWatermark <= ViewOpts.LowCovWatermark) {
1102       error(
1103           "invalid number range '" + CovWatermark +
1104               "', must be both high and low should be between 0-100, and high "
1105               "> low",
1106           "-coverage-watermark");
1107       return 1;
1108     }
1109   }
1110 
1111   ViewOpts.ShowLineNumbers = true;
1112   ViewOpts.ShowLineStats = ShowLineExecutionCounts.getNumOccurrences() != 0 ||
1113                            !ShowRegions || ShowBestLineRegionsCounts;
1114   ViewOpts.ShowRegionMarkers = ShowRegions || ShowBestLineRegionsCounts;
1115   ViewOpts.ShowExpandedRegions = ShowExpansions;
1116   ViewOpts.ShowBranchCounts =
1117       ShowBranches == CoverageViewOptions::BranchOutputType::Count;
1118   ViewOpts.ShowMCDC = ShowMCDC;
1119   ViewOpts.ShowBranchPercents =
1120       ShowBranches == CoverageViewOptions::BranchOutputType::Percent;
1121   ViewOpts.ShowFunctionInstantiations = ShowInstantiations;
1122   ViewOpts.ShowDirectoryCoverage = ShowDirectoryCoverage;
1123   ViewOpts.ShowOutputDirectory = ShowOutputDirectory;
1124   ViewOpts.BinaryCounters = BinaryCounters;
1125   ViewOpts.TabSize = TabSize;
1126   ViewOpts.ProjectTitle = ProjectTitle;
1127 
1128   if (ViewOpts.hasOutputDirectory()) {
1129     if (auto E = sys::fs::create_directories(ViewOpts.ShowOutputDirectory)) {
1130       error("could not create output directory!", E.message());
1131       return 1;
1132     }
1133   }
1134 
1135   if (PGOFilename) {
1136     sys::fs::file_status Status;
1137     if (std::error_code EC = sys::fs::status(PGOFilename.value(), Status)) {
1138       error("could not read profile data!" + EC.message(), PGOFilename.value());
1139       return 1;
1140     }
1141 
1142     if (ShowCreatedTime) {
1143       auto ModifiedTime = Status.getLastModificationTime();
1144       std::string ModifiedTimeStr = to_string(ModifiedTime);
1145       size_t found = ModifiedTimeStr.rfind(':');
1146       ViewOpts.CreatedTimeStr =
1147           (found != std::string::npos)
1148               ? "Created: " + ModifiedTimeStr.substr(0, found)
1149               : "Created: " + ModifiedTimeStr;
1150     }
1151   }
1152 
1153   auto Coverage = load();
1154   if (!Coverage)
1155     return 1;
1156 
1157   auto Printer = CoveragePrinter::create(ViewOpts);
1158 
1159   if (SourceFiles.empty() && !HadSourceFiles)
1160     // Get the source files from the function coverage mapping.
1161     for (StringRef Filename : Coverage->getUniqueSourceFiles()) {
1162       if (!IgnoreFilenameFilters.matchesFilename(Filename))
1163         SourceFiles.push_back(std::string(Filename));
1164     }
1165 
1166   // Create an index out of the source files.
1167   if (ViewOpts.hasOutputDirectory()) {
1168     if (Error E = Printer->createIndexFile(SourceFiles, *Coverage, Filters)) {
1169       error("could not create index file!", toString(std::move(E)));
1170       return 1;
1171     }
1172   }
1173 
1174   if (!Filters.empty()) {
1175     // Build the map of filenames to functions.
1176     std::map<llvm::StringRef, std::vector<const FunctionRecord *>>
1177         FilenameFunctionMap;
1178     for (const auto &SourceFile : SourceFiles)
1179       for (const auto &Function : Coverage->getCoveredFunctions(SourceFile))
1180         if (Filters.matches(*Coverage, Function))
1181           FilenameFunctionMap[SourceFile].push_back(&Function);
1182 
1183     // Only print filter matching functions for each file.
1184     for (const auto &FileFunc : FilenameFunctionMap) {
1185       StringRef File = FileFunc.first;
1186       const auto &Functions = FileFunc.second;
1187 
1188       auto OSOrErr = Printer->createViewFile(File, /*InToplevel=*/false);
1189       if (Error E = OSOrErr.takeError()) {
1190         error("could not create view file!", toString(std::move(E)));
1191         return 1;
1192       }
1193       auto OS = std::move(OSOrErr.get());
1194 
1195       bool ShowTitle = ViewOpts.hasOutputDirectory();
1196       for (const auto *Function : Functions) {
1197         auto FunctionView = createFunctionView(*Function, *Coverage);
1198         if (!FunctionView) {
1199           warning("Could not read coverage for '" + Function->Name + "'.");
1200           continue;
1201         }
1202         FunctionView->print(*OS.get(), /*WholeFile=*/false,
1203                             /*ShowSourceName=*/true, ShowTitle);
1204         ShowTitle = false;
1205       }
1206 
1207       Printer->closeViewFile(std::move(OS));
1208     }
1209     return 0;
1210   }
1211 
1212   // Show files
1213   bool ShowFilenames =
1214       (SourceFiles.size() != 1) || ViewOpts.hasOutputDirectory() ||
1215       (ViewOpts.Format == CoverageViewOptions::OutputFormat::HTML);
1216 
1217   ThreadPoolStrategy S = hardware_concurrency(ViewOpts.NumThreads);
1218   if (ViewOpts.NumThreads == 0) {
1219     // If NumThreads is not specified, create one thread for each input, up to
1220     // the number of hardware cores.
1221     S = heavyweight_hardware_concurrency(SourceFiles.size());
1222     S.Limit = true;
1223   }
1224 
1225   if (!ViewOpts.hasOutputDirectory() || S.ThreadsRequested == 1) {
1226     for (const std::string &SourceFile : SourceFiles)
1227       writeSourceFileView(SourceFile, Coverage.get(), Printer.get(),
1228                           ShowFilenames);
1229   } else {
1230     // In -output-dir mode, it's safe to use multiple threads to print files.
1231     DefaultThreadPool Pool(S);
1232     for (const std::string &SourceFile : SourceFiles)
1233       Pool.async(&CodeCoverageTool::writeSourceFileView, this, SourceFile,
1234                  Coverage.get(), Printer.get(), ShowFilenames);
1235     Pool.wait();
1236   }
1237 
1238   return 0;
1239 }
1240 
1241 int CodeCoverageTool::doReport(int argc, const char **argv,
1242                                CommandLineParserType commandLineParser) {
1243   cl::opt<bool> ShowFunctionSummaries(
1244       "show-functions", cl::Optional, cl::init(false),
1245       cl::desc("Show coverage summaries for each function"));
1246 
1247   auto Err = commandLineParser(argc, argv);
1248   if (Err)
1249     return Err;
1250 
1251   if (ViewOpts.Format == CoverageViewOptions::OutputFormat::HTML) {
1252     error("HTML output for summary reports is not yet supported.");
1253     return 1;
1254   } else if (ViewOpts.Format == CoverageViewOptions::OutputFormat::Lcov) {
1255     error("lcov format should be used with 'llvm-cov export'.");
1256     return 1;
1257   }
1258 
1259   if (PGOFilename) {
1260     sys::fs::file_status Status;
1261     if (std::error_code EC = sys::fs::status(PGOFilename.value(), Status)) {
1262       error("could not read profile data!" + EC.message(), PGOFilename.value());
1263       return 1;
1264     }
1265   }
1266 
1267   auto Coverage = load();
1268   if (!Coverage)
1269     return 1;
1270 
1271   CoverageReport Report(ViewOpts, *Coverage);
1272   if (!ShowFunctionSummaries) {
1273     if (SourceFiles.empty())
1274       Report.renderFileReports(llvm::outs(), IgnoreFilenameFilters);
1275     else
1276       Report.renderFileReports(llvm::outs(), SourceFiles);
1277   } else {
1278     if (SourceFiles.empty()) {
1279       error("source files must be specified when -show-functions=true is "
1280             "specified");
1281       return 1;
1282     }
1283 
1284     Report.renderFunctionReports(SourceFiles, DC, llvm::outs());
1285   }
1286   return 0;
1287 }
1288 
1289 int CodeCoverageTool::doExport(int argc, const char **argv,
1290                                CommandLineParserType commandLineParser) {
1291 
1292   cl::OptionCategory ExportCategory("Exporting options");
1293 
1294   cl::opt<bool> SkipExpansions("skip-expansions", cl::Optional,
1295                                cl::desc("Don't export expanded source regions"),
1296                                cl::cat(ExportCategory));
1297 
1298   cl::opt<bool> SkipFunctions("skip-functions", cl::Optional,
1299                               cl::desc("Don't export per-function data"),
1300                               cl::cat(ExportCategory));
1301 
1302   cl::opt<bool> SkipBranches("skip-branches", cl::Optional,
1303                               cl::desc("Don't export branch data (LCOV)"),
1304                               cl::cat(ExportCategory));
1305 
1306   cl::opt<bool> UnifyInstantiations("unify-instantiations", cl::Optional,
1307                                     cl::desc("Unify function instantiations"),
1308                                     cl::init(true), cl::cat(ExportCategory));
1309 
1310   auto Err = commandLineParser(argc, argv);
1311   if (Err)
1312     return Err;
1313 
1314   ViewOpts.SkipExpansions = SkipExpansions;
1315   ViewOpts.SkipFunctions = SkipFunctions;
1316   ViewOpts.SkipBranches = SkipBranches;
1317   ViewOpts.UnifyFunctionInstantiations = UnifyInstantiations;
1318 
1319   if (ViewOpts.Format != CoverageViewOptions::OutputFormat::Text &&
1320       ViewOpts.Format != CoverageViewOptions::OutputFormat::Lcov) {
1321     error("coverage data can only be exported as textual JSON or an "
1322           "lcov tracefile.");
1323     return 1;
1324   }
1325 
1326   if (PGOFilename) {
1327     sys::fs::file_status Status;
1328     if (std::error_code EC = sys::fs::status(PGOFilename.value(), Status)) {
1329       error("could not read profile data!" + EC.message(), PGOFilename.value());
1330       return 1;
1331     }
1332   }
1333 
1334   auto Coverage = load();
1335   if (!Coverage) {
1336     error("could not load coverage information");
1337     return 1;
1338   }
1339 
1340   std::unique_ptr<CoverageExporter> Exporter;
1341 
1342   switch (ViewOpts.Format) {
1343   case CoverageViewOptions::OutputFormat::Text:
1344     Exporter =
1345         std::make_unique<CoverageExporterJson>(*Coverage, ViewOpts, outs());
1346     break;
1347   case CoverageViewOptions::OutputFormat::HTML:
1348     // Unreachable because we should have gracefully terminated with an error
1349     // above.
1350     llvm_unreachable("Export in HTML is not supported!");
1351   case CoverageViewOptions::OutputFormat::Lcov:
1352     Exporter =
1353         std::make_unique<CoverageExporterLcov>(*Coverage, ViewOpts, outs());
1354     break;
1355   }
1356 
1357   if (SourceFiles.empty())
1358     Exporter->renderRoot(IgnoreFilenameFilters);
1359   else
1360     Exporter->renderRoot(SourceFiles);
1361 
1362   return 0;
1363 }
1364 
1365 int showMain(int argc, const char *argv[]) {
1366   CodeCoverageTool Tool;
1367   return Tool.run(CodeCoverageTool::Show, argc, argv);
1368 }
1369 
1370 int reportMain(int argc, const char *argv[]) {
1371   CodeCoverageTool Tool;
1372   return Tool.run(CodeCoverageTool::Report, argc, argv);
1373 }
1374 
1375 int exportMain(int argc, const char *argv[]) {
1376   CodeCoverageTool Tool;
1377   return Tool.run(CodeCoverageTool::Export, argc, argv);
1378 }
1379