1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282 |
- //===--- CoverageMappingGen.cpp - Coverage mapping generation ---*- C++ -*-===//
- //
- // The LLVM Compiler Infrastructure
- //
- // This file is distributed under the University of Illinois Open Source
- // License. See LICENSE.TXT for details.
- //
- //===----------------------------------------------------------------------===//
- //
- // Instrumentation-based code coverage mapping generator
- //
- //===----------------------------------------------------------------------===//
- #include "CoverageMappingGen.h"
- #include "CodeGenFunction.h"
- #include "clang/AST/StmtVisitor.h"
- #include "clang/Lex/Lexer.h"
- #include "llvm/ADT/SmallSet.h"
- #include "llvm/ADT/StringExtras.h"
- #include "llvm/ADT/Optional.h"
- #include "llvm/ProfileData/Coverage/CoverageMapping.h"
- #include "llvm/ProfileData/Coverage/CoverageMappingReader.h"
- #include "llvm/ProfileData/Coverage/CoverageMappingWriter.h"
- #include "llvm/ProfileData/InstrProfReader.h"
- #include "llvm/Support/FileSystem.h"
- #include "llvm/Support/Path.h"
- using namespace clang;
- using namespace CodeGen;
- using namespace llvm::coverage;
- void CoverageSourceInfo::SourceRangeSkipped(SourceRange Range) {
- SkippedRanges.push_back(Range);
- }
- namespace {
- /// \brief A region of source code that can be mapped to a counter.
- class SourceMappingRegion {
- Counter Count;
- /// \brief The region's starting location.
- Optional<SourceLocation> LocStart;
- /// \brief The region's ending location.
- Optional<SourceLocation> LocEnd;
- /// Whether this region should be emitted after its parent is emitted.
- bool DeferRegion;
- public:
- SourceMappingRegion(Counter Count, Optional<SourceLocation> LocStart,
- Optional<SourceLocation> LocEnd, bool DeferRegion = false)
- : Count(Count), LocStart(LocStart), LocEnd(LocEnd),
- DeferRegion(DeferRegion) {}
- const Counter &getCounter() const { return Count; }
- void setCounter(Counter C) { Count = C; }
- bool hasStartLoc() const { return LocStart.hasValue(); }
- void setStartLoc(SourceLocation Loc) { LocStart = Loc; }
- SourceLocation getStartLoc() const {
- assert(LocStart && "Region has no start location");
- return *LocStart;
- }
- bool hasEndLoc() const { return LocEnd.hasValue(); }
- void setEndLoc(SourceLocation Loc) { LocEnd = Loc; }
- SourceLocation getEndLoc() const {
- assert(LocEnd && "Region has no end location");
- return *LocEnd;
- }
- bool isDeferred() const { return DeferRegion; }
- void setDeferred(bool Deferred) { DeferRegion = Deferred; }
- };
- /// Spelling locations for the start and end of a source region.
- struct SpellingRegion {
- /// The line where the region starts.
- unsigned LineStart;
- /// The column where the region starts.
- unsigned ColumnStart;
- /// The line where the region ends.
- unsigned LineEnd;
- /// The column where the region ends.
- unsigned ColumnEnd;
- SpellingRegion(SourceManager &SM, SourceLocation LocStart,
- SourceLocation LocEnd) {
- LineStart = SM.getSpellingLineNumber(LocStart);
- ColumnStart = SM.getSpellingColumnNumber(LocStart);
- LineEnd = SM.getSpellingLineNumber(LocEnd);
- ColumnEnd = SM.getSpellingColumnNumber(LocEnd);
- }
- /// Check if the start and end locations appear in source order, i.e
- /// top->bottom, left->right.
- bool isInSourceOrder() const {
- return (LineStart < LineEnd) ||
- (LineStart == LineEnd && ColumnStart <= ColumnEnd);
- }
- };
- /// \brief Provides the common functionality for the different
- /// coverage mapping region builders.
- class CoverageMappingBuilder {
- public:
- CoverageMappingModuleGen &CVM;
- SourceManager &SM;
- const LangOptions &LangOpts;
- private:
- /// \brief Map of clang's FileIDs to IDs used for coverage mapping.
- llvm::SmallDenseMap<FileID, std::pair<unsigned, SourceLocation>, 8>
- FileIDMapping;
- public:
- /// \brief The coverage mapping regions for this function
- llvm::SmallVector<CounterMappingRegion, 32> MappingRegions;
- /// \brief The source mapping regions for this function.
- std::vector<SourceMappingRegion> SourceRegions;
- /// \brief A set of regions which can be used as a filter.
- ///
- /// It is produced by emitExpansionRegions() and is used in
- /// emitSourceRegions() to suppress producing code regions if
- /// the same area is covered by expansion regions.
- typedef llvm::SmallSet<std::pair<SourceLocation, SourceLocation>, 8>
- SourceRegionFilter;
- CoverageMappingBuilder(CoverageMappingModuleGen &CVM, SourceManager &SM,
- const LangOptions &LangOpts)
- : CVM(CVM), SM(SM), LangOpts(LangOpts) {}
- /// \brief Return the precise end location for the given token.
- SourceLocation getPreciseTokenLocEnd(SourceLocation Loc) {
- // We avoid getLocForEndOfToken here, because it doesn't do what we want for
- // macro locations, which we just treat as expanded files.
- unsigned TokLen =
- Lexer::MeasureTokenLength(SM.getSpellingLoc(Loc), SM, LangOpts);
- return Loc.getLocWithOffset(TokLen);
- }
- /// \brief Return the start location of an included file or expanded macro.
- SourceLocation getStartOfFileOrMacro(SourceLocation Loc) {
- if (Loc.isMacroID())
- return Loc.getLocWithOffset(-SM.getFileOffset(Loc));
- return SM.getLocForStartOfFile(SM.getFileID(Loc));
- }
- /// \brief Return the end location of an included file or expanded macro.
- SourceLocation getEndOfFileOrMacro(SourceLocation Loc) {
- if (Loc.isMacroID())
- return Loc.getLocWithOffset(SM.getFileIDSize(SM.getFileID(Loc)) -
- SM.getFileOffset(Loc));
- return SM.getLocForEndOfFile(SM.getFileID(Loc));
- }
- /// \brief Find out where the current file is included or macro is expanded.
- SourceLocation getIncludeOrExpansionLoc(SourceLocation Loc) {
- return Loc.isMacroID() ? SM.getImmediateExpansionRange(Loc).first
- : SM.getIncludeLoc(SM.getFileID(Loc));
- }
- /// \brief Return true if \c Loc is a location in a built-in macro.
- bool isInBuiltin(SourceLocation Loc) {
- return SM.getBufferName(SM.getSpellingLoc(Loc)) == "<built-in>";
- }
- /// \brief Check whether \c Loc is included or expanded from \c Parent.
- bool isNestedIn(SourceLocation Loc, FileID Parent) {
- do {
- Loc = getIncludeOrExpansionLoc(Loc);
- if (Loc.isInvalid())
- return false;
- } while (!SM.isInFileID(Loc, Parent));
- return true;
- }
- /// \brief Get the start of \c S ignoring macro arguments and builtin macros.
- SourceLocation getStart(const Stmt *S) {
- SourceLocation Loc = S->getLocStart();
- while (SM.isMacroArgExpansion(Loc) || isInBuiltin(Loc))
- Loc = SM.getImmediateExpansionRange(Loc).first;
- return Loc;
- }
- /// \brief Get the end of \c S ignoring macro arguments and builtin macros.
- SourceLocation getEnd(const Stmt *S) {
- SourceLocation Loc = S->getLocEnd();
- while (SM.isMacroArgExpansion(Loc) || isInBuiltin(Loc))
- Loc = SM.getImmediateExpansionRange(Loc).first;
- return getPreciseTokenLocEnd(Loc);
- }
- /// \brief Find the set of files we have regions for and assign IDs
- ///
- /// Fills \c Mapping with the virtual file mapping needed to write out
- /// coverage and collects the necessary file information to emit source and
- /// expansion regions.
- void gatherFileIDs(SmallVectorImpl<unsigned> &Mapping) {
- FileIDMapping.clear();
- llvm::SmallSet<FileID, 8> Visited;
- SmallVector<std::pair<SourceLocation, unsigned>, 8> FileLocs;
- for (const auto &Region : SourceRegions) {
- SourceLocation Loc = Region.getStartLoc();
- FileID File = SM.getFileID(Loc);
- if (!Visited.insert(File).second)
- continue;
- // Do not map FileID's associated with system headers.
- if (SM.isInSystemHeader(SM.getSpellingLoc(Loc)))
- continue;
- unsigned Depth = 0;
- for (SourceLocation Parent = getIncludeOrExpansionLoc(Loc);
- Parent.isValid(); Parent = getIncludeOrExpansionLoc(Parent))
- ++Depth;
- FileLocs.push_back(std::make_pair(Loc, Depth));
- }
- std::stable_sort(FileLocs.begin(), FileLocs.end(), llvm::less_second());
- for (const auto &FL : FileLocs) {
- SourceLocation Loc = FL.first;
- FileID SpellingFile = SM.getDecomposedSpellingLoc(Loc).first;
- auto Entry = SM.getFileEntryForID(SpellingFile);
- if (!Entry)
- continue;
- FileIDMapping[SM.getFileID(Loc)] = std::make_pair(Mapping.size(), Loc);
- Mapping.push_back(CVM.getFileID(Entry));
- }
- }
- /// \brief Get the coverage mapping file ID for \c Loc.
- ///
- /// If such file id doesn't exist, return None.
- Optional<unsigned> getCoverageFileID(SourceLocation Loc) {
- auto Mapping = FileIDMapping.find(SM.getFileID(Loc));
- if (Mapping != FileIDMapping.end())
- return Mapping->second.first;
- return None;
- }
- /// \brief Gather all the regions that were skipped by the preprocessor
- /// using the constructs like #if.
- void gatherSkippedRegions() {
- /// An array of the minimum lineStarts and the maximum lineEnds
- /// for mapping regions from the appropriate source files.
- llvm::SmallVector<std::pair<unsigned, unsigned>, 8> FileLineRanges;
- FileLineRanges.resize(
- FileIDMapping.size(),
- std::make_pair(std::numeric_limits<unsigned>::max(), 0));
- for (const auto &R : MappingRegions) {
- FileLineRanges[R.FileID].first =
- std::min(FileLineRanges[R.FileID].first, R.LineStart);
- FileLineRanges[R.FileID].second =
- std::max(FileLineRanges[R.FileID].second, R.LineEnd);
- }
- auto SkippedRanges = CVM.getSourceInfo().getSkippedRanges();
- for (const auto &I : SkippedRanges) {
- auto LocStart = I.getBegin();
- auto LocEnd = I.getEnd();
- assert(SM.isWrittenInSameFile(LocStart, LocEnd) &&
- "region spans multiple files");
- auto CovFileID = getCoverageFileID(LocStart);
- if (!CovFileID)
- continue;
- SpellingRegion SR{SM, LocStart, LocEnd};
- auto Region = CounterMappingRegion::makeSkipped(
- *CovFileID, SR.LineStart, SR.ColumnStart, SR.LineEnd, SR.ColumnEnd);
- // Make sure that we only collect the regions that are inside
- // the souce code of this function.
- if (Region.LineStart >= FileLineRanges[*CovFileID].first &&
- Region.LineEnd <= FileLineRanges[*CovFileID].second)
- MappingRegions.push_back(Region);
- }
- }
- /// \brief Generate the coverage counter mapping regions from collected
- /// source regions.
- void emitSourceRegions(const SourceRegionFilter &Filter) {
- for (const auto &Region : SourceRegions) {
- assert(Region.hasEndLoc() && "incomplete region");
- SourceLocation LocStart = Region.getStartLoc();
- assert(SM.getFileID(LocStart).isValid() && "region in invalid file");
- // Ignore regions from system headers.
- if (SM.isInSystemHeader(SM.getSpellingLoc(LocStart)))
- continue;
- auto CovFileID = getCoverageFileID(LocStart);
- // Ignore regions that don't have a file, such as builtin macros.
- if (!CovFileID)
- continue;
- SourceLocation LocEnd = Region.getEndLoc();
- assert(SM.isWrittenInSameFile(LocStart, LocEnd) &&
- "region spans multiple files");
- // Don't add code regions for the area covered by expansion regions.
- // This not only suppresses redundant regions, but sometimes prevents
- // creating regions with wrong counters if, for example, a statement's
- // body ends at the end of a nested macro.
- if (Filter.count(std::make_pair(LocStart, LocEnd)))
- continue;
- // Find the spelling locations for the mapping region.
- SpellingRegion SR{SM, LocStart, LocEnd};
- assert(SR.isInSourceOrder() && "region start and end out of order");
- MappingRegions.push_back(CounterMappingRegion::makeRegion(
- Region.getCounter(), *CovFileID, SR.LineStart, SR.ColumnStart,
- SR.LineEnd, SR.ColumnEnd));
- }
- }
- /// \brief Generate expansion regions for each virtual file we've seen.
- SourceRegionFilter emitExpansionRegions() {
- SourceRegionFilter Filter;
- for (const auto &FM : FileIDMapping) {
- SourceLocation ExpandedLoc = FM.second.second;
- SourceLocation ParentLoc = getIncludeOrExpansionLoc(ExpandedLoc);
- if (ParentLoc.isInvalid())
- continue;
- auto ParentFileID = getCoverageFileID(ParentLoc);
- if (!ParentFileID)
- continue;
- auto ExpandedFileID = getCoverageFileID(ExpandedLoc);
- assert(ExpandedFileID && "expansion in uncovered file");
- SourceLocation LocEnd = getPreciseTokenLocEnd(ParentLoc);
- assert(SM.isWrittenInSameFile(ParentLoc, LocEnd) &&
- "region spans multiple files");
- Filter.insert(std::make_pair(ParentLoc, LocEnd));
- SpellingRegion SR{SM, ParentLoc, LocEnd};
- assert(SR.isInSourceOrder() && "region start and end out of order");
- MappingRegions.push_back(CounterMappingRegion::makeExpansion(
- *ParentFileID, *ExpandedFileID, SR.LineStart, SR.ColumnStart,
- SR.LineEnd, SR.ColumnEnd));
- }
- return Filter;
- }
- };
- /// \brief Creates unreachable coverage regions for the functions that
- /// are not emitted.
- struct EmptyCoverageMappingBuilder : public CoverageMappingBuilder {
- EmptyCoverageMappingBuilder(CoverageMappingModuleGen &CVM, SourceManager &SM,
- const LangOptions &LangOpts)
- : CoverageMappingBuilder(CVM, SM, LangOpts) {}
- void VisitDecl(const Decl *D) {
- if (!D->hasBody())
- return;
- auto Body = D->getBody();
- SourceLocation Start = getStart(Body);
- SourceLocation End = getEnd(Body);
- if (!SM.isWrittenInSameFile(Start, End)) {
- // Walk up to find the common ancestor.
- // Correct the locations accordingly.
- FileID StartFileID = SM.getFileID(Start);
- FileID EndFileID = SM.getFileID(End);
- while (StartFileID != EndFileID && !isNestedIn(End, StartFileID)) {
- Start = getIncludeOrExpansionLoc(Start);
- assert(Start.isValid() &&
- "Declaration start location not nested within a known region");
- StartFileID = SM.getFileID(Start);
- }
- while (StartFileID != EndFileID) {
- End = getPreciseTokenLocEnd(getIncludeOrExpansionLoc(End));
- assert(End.isValid() &&
- "Declaration end location not nested within a known region");
- EndFileID = SM.getFileID(End);
- }
- }
- SourceRegions.emplace_back(Counter(), Start, End);
- }
- /// \brief Write the mapping data to the output stream
- void write(llvm::raw_ostream &OS) {
- SmallVector<unsigned, 16> FileIDMapping;
- gatherFileIDs(FileIDMapping);
- emitSourceRegions(SourceRegionFilter());
- if (MappingRegions.empty())
- return;
- CoverageMappingWriter Writer(FileIDMapping, None, MappingRegions);
- Writer.write(OS);
- }
- };
- /// \brief A StmtVisitor that creates coverage mapping regions which map
- /// from the source code locations to the PGO counters.
- struct CounterCoverageMappingBuilder
- : public CoverageMappingBuilder,
- public ConstStmtVisitor<CounterCoverageMappingBuilder> {
- /// \brief The map of statements to count values.
- llvm::DenseMap<const Stmt *, unsigned> &CounterMap;
- /// \brief A stack of currently live regions.
- std::vector<SourceMappingRegion> RegionStack;
- /// The currently deferred region: its end location and count can be set once
- /// its parent has been popped from the region stack.
- Optional<SourceMappingRegion> DeferredRegion;
- CounterExpressionBuilder Builder;
- /// \brief A location in the most recently visited file or macro.
- ///
- /// This is used to adjust the active source regions appropriately when
- /// expressions cross file or macro boundaries.
- SourceLocation MostRecentLocation;
- /// \brief Return a counter for the subtraction of \c RHS from \c LHS
- Counter subtractCounters(Counter LHS, Counter RHS) {
- return Builder.subtract(LHS, RHS);
- }
- /// \brief Return a counter for the sum of \c LHS and \c RHS.
- Counter addCounters(Counter LHS, Counter RHS) {
- return Builder.add(LHS, RHS);
- }
- Counter addCounters(Counter C1, Counter C2, Counter C3) {
- return addCounters(addCounters(C1, C2), C3);
- }
- /// \brief Return the region counter for the given statement.
- ///
- /// This should only be called on statements that have a dedicated counter.
- Counter getRegionCounter(const Stmt *S) {
- return Counter::getCounter(CounterMap[S]);
- }
- /// \brief Push a region onto the stack.
- ///
- /// Returns the index on the stack where the region was pushed. This can be
- /// used with popRegions to exit a "scope", ending the region that was pushed.
- size_t pushRegion(Counter Count, Optional<SourceLocation> StartLoc = None,
- Optional<SourceLocation> EndLoc = None) {
- if (StartLoc) {
- MostRecentLocation = *StartLoc;
- completeDeferred(Count, MostRecentLocation);
- }
- RegionStack.emplace_back(Count, StartLoc, EndLoc);
- return RegionStack.size() - 1;
- }
- /// Complete any pending deferred region by setting its end location and
- /// count, and then pushing it onto the region stack.
- size_t completeDeferred(Counter Count, SourceLocation DeferredEndLoc) {
- size_t Index = RegionStack.size();
- if (!DeferredRegion)
- return Index;
- // Consume the pending region.
- SourceMappingRegion DR = DeferredRegion.getValue();
- DeferredRegion = None;
- // If the region ends in an expansion, find the expansion site.
- if (SM.getFileID(DeferredEndLoc) != SM.getMainFileID()) {
- FileID StartFile = SM.getFileID(DR.getStartLoc());
- if (isNestedIn(DeferredEndLoc, StartFile)) {
- do {
- DeferredEndLoc = getIncludeOrExpansionLoc(DeferredEndLoc);
- } while (StartFile != SM.getFileID(DeferredEndLoc));
- }
- }
- // The parent of this deferred region ends where the containing decl ends,
- // so the region isn't useful.
- if (DR.getStartLoc() == DeferredEndLoc)
- return Index;
- // If we're visiting statements in non-source order (e.g switch cases or
- // a loop condition) we can't construct a sensible deferred region.
- if (!SpellingRegion(SM, DR.getStartLoc(), DeferredEndLoc).isInSourceOrder())
- return Index;
- DR.setCounter(Count);
- DR.setEndLoc(DeferredEndLoc);
- handleFileExit(DeferredEndLoc);
- RegionStack.push_back(DR);
- return Index;
- }
- /// \brief Pop regions from the stack into the function's list of regions.
- ///
- /// Adds all regions from \c ParentIndex to the top of the stack to the
- /// function's \c SourceRegions.
- void popRegions(size_t ParentIndex) {
- assert(RegionStack.size() >= ParentIndex && "parent not in stack");
- bool ParentOfDeferredRegion = false;
- while (RegionStack.size() > ParentIndex) {
- SourceMappingRegion &Region = RegionStack.back();
- if (Region.hasStartLoc()) {
- SourceLocation StartLoc = Region.getStartLoc();
- SourceLocation EndLoc = Region.hasEndLoc()
- ? Region.getEndLoc()
- : RegionStack[ParentIndex].getEndLoc();
- while (!SM.isWrittenInSameFile(StartLoc, EndLoc)) {
- // The region ends in a nested file or macro expansion. Create a
- // separate region for each expansion.
- SourceLocation NestedLoc = getStartOfFileOrMacro(EndLoc);
- assert(SM.isWrittenInSameFile(NestedLoc, EndLoc));
- if (!isRegionAlreadyAdded(NestedLoc, EndLoc))
- SourceRegions.emplace_back(Region.getCounter(), NestedLoc, EndLoc);
- EndLoc = getPreciseTokenLocEnd(getIncludeOrExpansionLoc(EndLoc));
- if (EndLoc.isInvalid())
- llvm::report_fatal_error("File exit not handled before popRegions");
- }
- Region.setEndLoc(EndLoc);
- MostRecentLocation = EndLoc;
- // If this region happens to span an entire expansion, we need to make
- // sure we don't overlap the parent region with it.
- if (StartLoc == getStartOfFileOrMacro(StartLoc) &&
- EndLoc == getEndOfFileOrMacro(EndLoc))
- MostRecentLocation = getIncludeOrExpansionLoc(EndLoc);
- assert(SM.isWrittenInSameFile(Region.getStartLoc(), EndLoc));
- SourceRegions.push_back(Region);
- if (ParentOfDeferredRegion) {
- ParentOfDeferredRegion = false;
- // If there's an existing deferred region, keep the old one, because
- // it means there are two consecutive returns (or a similar pattern).
- if (!DeferredRegion.hasValue() &&
- // File IDs aren't gathered within macro expansions, so it isn't
- // useful to try and create a deferred region inside of one.
- (SM.getFileID(EndLoc) == SM.getMainFileID()))
- DeferredRegion =
- SourceMappingRegion(Counter::getZero(), EndLoc, None);
- }
- } else if (Region.isDeferred()) {
- assert(!ParentOfDeferredRegion && "Consecutive deferred regions");
- ParentOfDeferredRegion = true;
- }
- RegionStack.pop_back();
- }
- assert(!ParentOfDeferredRegion && "Deferred region with no parent");
- }
- /// \brief Return the currently active region.
- SourceMappingRegion &getRegion() {
- assert(!RegionStack.empty() && "statement has no region");
- return RegionStack.back();
- }
- /// \brief Propagate counts through the children of \c S.
- Counter propagateCounts(Counter TopCount, const Stmt *S) {
- SourceLocation StartLoc = getStart(S);
- SourceLocation EndLoc = getEnd(S);
- size_t Index = pushRegion(TopCount, StartLoc, EndLoc);
- Visit(S);
- Counter ExitCount = getRegion().getCounter();
- popRegions(Index);
- // The statement may be spanned by an expansion. Make sure we handle a file
- // exit out of this expansion before moving to the next statement.
- if (SM.isBeforeInTranslationUnit(StartLoc, S->getLocStart()))
- MostRecentLocation = EndLoc;
- return ExitCount;
- }
- /// \brief Check whether a region with bounds \c StartLoc and \c EndLoc
- /// is already added to \c SourceRegions.
- bool isRegionAlreadyAdded(SourceLocation StartLoc, SourceLocation EndLoc) {
- return SourceRegions.rend() !=
- std::find_if(SourceRegions.rbegin(), SourceRegions.rend(),
- [&](const SourceMappingRegion &Region) {
- return Region.getStartLoc() == StartLoc &&
- Region.getEndLoc() == EndLoc;
- });
- }
- /// \brief Adjust the most recently visited location to \c EndLoc.
- ///
- /// This should be used after visiting any statements in non-source order.
- void adjustForOutOfOrderTraversal(SourceLocation EndLoc) {
- MostRecentLocation = EndLoc;
- // The code region for a whole macro is created in handleFileExit() when
- // it detects exiting of the virtual file of that macro. If we visited
- // statements in non-source order, we might already have such a region
- // added, for example, if a body of a loop is divided among multiple
- // macros. Avoid adding duplicate regions in such case.
- if (getRegion().hasEndLoc() &&
- MostRecentLocation == getEndOfFileOrMacro(MostRecentLocation) &&
- isRegionAlreadyAdded(getStartOfFileOrMacro(MostRecentLocation),
- MostRecentLocation))
- MostRecentLocation = getIncludeOrExpansionLoc(MostRecentLocation);
- }
- /// \brief Adjust regions and state when \c NewLoc exits a file.
- ///
- /// If moving from our most recently tracked location to \c NewLoc exits any
- /// files, this adjusts our current region stack and creates the file regions
- /// for the exited file.
- void handleFileExit(SourceLocation NewLoc) {
- if (NewLoc.isInvalid() ||
- SM.isWrittenInSameFile(MostRecentLocation, NewLoc))
- return;
- // If NewLoc is not in a file that contains MostRecentLocation, walk up to
- // find the common ancestor.
- SourceLocation LCA = NewLoc;
- FileID ParentFile = SM.getFileID(LCA);
- while (!isNestedIn(MostRecentLocation, ParentFile)) {
- LCA = getIncludeOrExpansionLoc(LCA);
- if (LCA.isInvalid() || SM.isWrittenInSameFile(LCA, MostRecentLocation)) {
- // Since there isn't a common ancestor, no file was exited. We just need
- // to adjust our location to the new file.
- MostRecentLocation = NewLoc;
- return;
- }
- ParentFile = SM.getFileID(LCA);
- }
- llvm::SmallSet<SourceLocation, 8> StartLocs;
- Optional<Counter> ParentCounter;
- for (SourceMappingRegion &I : llvm::reverse(RegionStack)) {
- if (!I.hasStartLoc())
- continue;
- SourceLocation Loc = I.getStartLoc();
- if (!isNestedIn(Loc, ParentFile)) {
- ParentCounter = I.getCounter();
- break;
- }
- while (!SM.isInFileID(Loc, ParentFile)) {
- // The most nested region for each start location is the one with the
- // correct count. We avoid creating redundant regions by stopping once
- // we've seen this region.
- if (StartLocs.insert(Loc).second)
- SourceRegions.emplace_back(I.getCounter(), Loc,
- getEndOfFileOrMacro(Loc));
- Loc = getIncludeOrExpansionLoc(Loc);
- }
- I.setStartLoc(getPreciseTokenLocEnd(Loc));
- }
- if (ParentCounter) {
- // If the file is contained completely by another region and doesn't
- // immediately start its own region, the whole file gets a region
- // corresponding to the parent.
- SourceLocation Loc = MostRecentLocation;
- while (isNestedIn(Loc, ParentFile)) {
- SourceLocation FileStart = getStartOfFileOrMacro(Loc);
- if (StartLocs.insert(FileStart).second)
- SourceRegions.emplace_back(*ParentCounter, FileStart,
- getEndOfFileOrMacro(Loc));
- Loc = getIncludeOrExpansionLoc(Loc);
- }
- }
- MostRecentLocation = NewLoc;
- }
- /// \brief Ensure that \c S is included in the current region.
- void extendRegion(const Stmt *S) {
- SourceMappingRegion &Region = getRegion();
- SourceLocation StartLoc = getStart(S);
- handleFileExit(StartLoc);
- if (!Region.hasStartLoc())
- Region.setStartLoc(StartLoc);
- completeDeferred(Region.getCounter(), StartLoc);
- }
- /// \brief Mark \c S as a terminator, starting a zero region.
- void terminateRegion(const Stmt *S) {
- extendRegion(S);
- SourceMappingRegion &Region = getRegion();
- if (!Region.hasEndLoc())
- Region.setEndLoc(getEnd(S));
- pushRegion(Counter::getZero());
- getRegion().setDeferred(true);
- }
- /// \brief Keep counts of breaks and continues inside loops.
- struct BreakContinue {
- Counter BreakCount;
- Counter ContinueCount;
- };
- SmallVector<BreakContinue, 8> BreakContinueStack;
- CounterCoverageMappingBuilder(
- CoverageMappingModuleGen &CVM,
- llvm::DenseMap<const Stmt *, unsigned> &CounterMap, SourceManager &SM,
- const LangOptions &LangOpts)
- : CoverageMappingBuilder(CVM, SM, LangOpts), CounterMap(CounterMap),
- DeferredRegion(None) {}
- /// \brief Write the mapping data to the output stream
- void write(llvm::raw_ostream &OS) {
- llvm::SmallVector<unsigned, 8> VirtualFileMapping;
- gatherFileIDs(VirtualFileMapping);
- SourceRegionFilter Filter = emitExpansionRegions();
- assert(!DeferredRegion && "Deferred region never completed");
- emitSourceRegions(Filter);
- gatherSkippedRegions();
- if (MappingRegions.empty())
- return;
- CoverageMappingWriter Writer(VirtualFileMapping, Builder.getExpressions(),
- MappingRegions);
- Writer.write(OS);
- }
- void VisitStmt(const Stmt *S) {
- if (S->getLocStart().isValid())
- extendRegion(S);
- for (const Stmt *Child : S->children())
- if (Child)
- this->Visit(Child);
- handleFileExit(getEnd(S));
- }
- void VisitDecl(const Decl *D) {
- assert(!DeferredRegion && "Deferred region never completed");
- Stmt *Body = D->getBody();
- // Do not propagate region counts into system headers.
- if (Body && SM.isInSystemHeader(SM.getSpellingLoc(getStart(Body))))
- return;
- Counter ExitCount = propagateCounts(getRegionCounter(Body), Body);
- assert(RegionStack.empty() && "Regions entered but never exited");
- // Complete any deferred regions introduced by the last statement in a decl.
- popRegions(completeDeferred(ExitCount, getEnd(Body)));
- }
- void VisitReturnStmt(const ReturnStmt *S) {
- extendRegion(S);
- if (S->getRetValue())
- Visit(S->getRetValue());
- terminateRegion(S);
- }
- void VisitCXXThrowExpr(const CXXThrowExpr *E) {
- extendRegion(E);
- if (E->getSubExpr())
- Visit(E->getSubExpr());
- terminateRegion(E);
- }
- void VisitGotoStmt(const GotoStmt *S) { terminateRegion(S); }
- void VisitLabelStmt(const LabelStmt *S) {
- SourceLocation Start = getStart(S);
- // We can't extendRegion here or we risk overlapping with our new region.
- handleFileExit(Start);
- pushRegion(getRegionCounter(S), Start);
- Visit(S->getSubStmt());
- }
- void VisitBreakStmt(const BreakStmt *S) {
- assert(!BreakContinueStack.empty() && "break not in a loop or switch!");
- BreakContinueStack.back().BreakCount = addCounters(
- BreakContinueStack.back().BreakCount, getRegion().getCounter());
- // FIXME: a break in a switch should terminate regions for all preceding
- // case statements, not just the most recent one.
- terminateRegion(S);
- }
- void VisitContinueStmt(const ContinueStmt *S) {
- assert(!BreakContinueStack.empty() && "continue stmt not in a loop!");
- BreakContinueStack.back().ContinueCount = addCounters(
- BreakContinueStack.back().ContinueCount, getRegion().getCounter());
- terminateRegion(S);
- }
- void VisitCallExpr(const CallExpr *E) {
- VisitStmt(E);
- // Terminate the region when we hit a noreturn function.
- // (This is helpful dealing with switch statements.)
- QualType CalleeType = E->getCallee()->getType();
- if (getFunctionExtInfo(*CalleeType).getNoReturn())
- terminateRegion(E);
- }
- void VisitWhileStmt(const WhileStmt *S) {
- extendRegion(S);
- Counter ParentCount = getRegion().getCounter();
- Counter BodyCount = getRegionCounter(S);
- // Handle the body first so that we can get the backedge count.
- BreakContinueStack.push_back(BreakContinue());
- extendRegion(S->getBody());
- Counter BackedgeCount = propagateCounts(BodyCount, S->getBody());
- BreakContinue BC = BreakContinueStack.pop_back_val();
- // Go back to handle the condition.
- Counter CondCount =
- addCounters(ParentCount, BackedgeCount, BC.ContinueCount);
- propagateCounts(CondCount, S->getCond());
- adjustForOutOfOrderTraversal(getEnd(S));
- Counter OutCount =
- addCounters(BC.BreakCount, subtractCounters(CondCount, BodyCount));
- if (OutCount != ParentCount)
- pushRegion(OutCount);
- }
- void VisitDoStmt(const DoStmt *S) {
- extendRegion(S);
- Counter ParentCount = getRegion().getCounter();
- Counter BodyCount = getRegionCounter(S);
- BreakContinueStack.push_back(BreakContinue());
- extendRegion(S->getBody());
- Counter BackedgeCount =
- propagateCounts(addCounters(ParentCount, BodyCount), S->getBody());
- BreakContinue BC = BreakContinueStack.pop_back_val();
- Counter CondCount = addCounters(BackedgeCount, BC.ContinueCount);
- propagateCounts(CondCount, S->getCond());
- Counter OutCount =
- addCounters(BC.BreakCount, subtractCounters(CondCount, BodyCount));
- if (OutCount != ParentCount)
- pushRegion(OutCount);
- }
- void VisitForStmt(const ForStmt *S) {
- extendRegion(S);
- if (S->getInit())
- Visit(S->getInit());
- Counter ParentCount = getRegion().getCounter();
- Counter BodyCount = getRegionCounter(S);
- // Handle the body first so that we can get the backedge count.
- BreakContinueStack.push_back(BreakContinue());
- extendRegion(S->getBody());
- Counter BackedgeCount = propagateCounts(BodyCount, S->getBody());
- BreakContinue BC = BreakContinueStack.pop_back_val();
- // The increment is essentially part of the body but it needs to include
- // the count for all the continue statements.
- if (const Stmt *Inc = S->getInc())
- propagateCounts(addCounters(BackedgeCount, BC.ContinueCount), Inc);
- // Go back to handle the condition.
- Counter CondCount =
- addCounters(ParentCount, BackedgeCount, BC.ContinueCount);
- if (const Expr *Cond = S->getCond()) {
- propagateCounts(CondCount, Cond);
- adjustForOutOfOrderTraversal(getEnd(S));
- }
- Counter OutCount =
- addCounters(BC.BreakCount, subtractCounters(CondCount, BodyCount));
- if (OutCount != ParentCount)
- pushRegion(OutCount);
- }
- void VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
- extendRegion(S);
- Visit(S->getLoopVarStmt());
- Visit(S->getRangeStmt());
- Counter ParentCount = getRegion().getCounter();
- Counter BodyCount = getRegionCounter(S);
- BreakContinueStack.push_back(BreakContinue());
- extendRegion(S->getBody());
- Counter BackedgeCount = propagateCounts(BodyCount, S->getBody());
- BreakContinue BC = BreakContinueStack.pop_back_val();
- Counter LoopCount =
- addCounters(ParentCount, BackedgeCount, BC.ContinueCount);
- Counter OutCount =
- addCounters(BC.BreakCount, subtractCounters(LoopCount, BodyCount));
- if (OutCount != ParentCount)
- pushRegion(OutCount);
- }
- void VisitObjCForCollectionStmt(const ObjCForCollectionStmt *S) {
- extendRegion(S);
- Visit(S->getElement());
- Counter ParentCount = getRegion().getCounter();
- Counter BodyCount = getRegionCounter(S);
- BreakContinueStack.push_back(BreakContinue());
- extendRegion(S->getBody());
- Counter BackedgeCount = propagateCounts(BodyCount, S->getBody());
- BreakContinue BC = BreakContinueStack.pop_back_val();
- Counter LoopCount =
- addCounters(ParentCount, BackedgeCount, BC.ContinueCount);
- Counter OutCount =
- addCounters(BC.BreakCount, subtractCounters(LoopCount, BodyCount));
- if (OutCount != ParentCount)
- pushRegion(OutCount);
- }
- void VisitSwitchStmt(const SwitchStmt *S) {
- extendRegion(S);
- if (S->getInit())
- Visit(S->getInit());
- Visit(S->getCond());
- BreakContinueStack.push_back(BreakContinue());
- const Stmt *Body = S->getBody();
- extendRegion(Body);
- if (const auto *CS = dyn_cast<CompoundStmt>(Body)) {
- if (!CS->body_empty()) {
- // Make a region for the body of the switch. If the body starts with
- // a case, that case will reuse this region; otherwise, this covers
- // the unreachable code at the beginning of the switch body.
- size_t Index =
- pushRegion(Counter::getZero(), getStart(CS->body_front()));
- for (const auto *Child : CS->children())
- Visit(Child);
- // Set the end for the body of the switch, if it isn't already set.
- for (size_t i = RegionStack.size(); i != Index; --i) {
- if (!RegionStack[i - 1].hasEndLoc())
- RegionStack[i - 1].setEndLoc(getEnd(CS->body_back()));
- }
- popRegions(Index);
- }
- } else
- propagateCounts(Counter::getZero(), Body);
- BreakContinue BC = BreakContinueStack.pop_back_val();
- if (!BreakContinueStack.empty())
- BreakContinueStack.back().ContinueCount = addCounters(
- BreakContinueStack.back().ContinueCount, BC.ContinueCount);
- Counter ExitCount = getRegionCounter(S);
- SourceLocation ExitLoc = getEnd(S);
- pushRegion(ExitCount);
- // Ensure that handleFileExit recognizes when the end location is located
- // in a different file.
- MostRecentLocation = getStart(S);
- handleFileExit(ExitLoc);
- }
- void VisitSwitchCase(const SwitchCase *S) {
- extendRegion(S);
- SourceMappingRegion &Parent = getRegion();
- Counter Count = addCounters(Parent.getCounter(), getRegionCounter(S));
- // Reuse the existing region if it starts at our label. This is typical of
- // the first case in a switch.
- if (Parent.hasStartLoc() && Parent.getStartLoc() == getStart(S))
- Parent.setCounter(Count);
- else
- pushRegion(Count, getStart(S));
- if (const auto *CS = dyn_cast<CaseStmt>(S)) {
- Visit(CS->getLHS());
- if (const Expr *RHS = CS->getRHS())
- Visit(RHS);
- }
- Visit(S->getSubStmt());
- }
- void VisitIfStmt(const IfStmt *S) {
- extendRegion(S);
- if (S->getInit())
- Visit(S->getInit());
- // Extend into the condition before we propagate through it below - this is
- // needed to handle macros that generate the "if" but not the condition.
- extendRegion(S->getCond());
- Counter ParentCount = getRegion().getCounter();
- Counter ThenCount = getRegionCounter(S);
- // Emitting a counter for the condition makes it easier to interpret the
- // counter for the body when looking at the coverage.
- propagateCounts(ParentCount, S->getCond());
- extendRegion(S->getThen());
- Counter OutCount = propagateCounts(ThenCount, S->getThen());
- Counter ElseCount = subtractCounters(ParentCount, ThenCount);
- if (const Stmt *Else = S->getElse()) {
- extendRegion(S->getElse());
- OutCount = addCounters(OutCount, propagateCounts(ElseCount, Else));
- } else
- OutCount = addCounters(OutCount, ElseCount);
- if (OutCount != ParentCount)
- pushRegion(OutCount);
- }
- void VisitCXXTryStmt(const CXXTryStmt *S) {
- extendRegion(S);
- // Handle macros that generate the "try" but not the rest.
- extendRegion(S->getTryBlock());
- Counter ParentCount = getRegion().getCounter();
- propagateCounts(ParentCount, S->getTryBlock());
- for (unsigned I = 0, E = S->getNumHandlers(); I < E; ++I)
- Visit(S->getHandler(I));
- Counter ExitCount = getRegionCounter(S);
- pushRegion(ExitCount);
- }
- void VisitCXXCatchStmt(const CXXCatchStmt *S) {
- propagateCounts(getRegionCounter(S), S->getHandlerBlock());
- }
- void VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
- extendRegion(E);
- Counter ParentCount = getRegion().getCounter();
- Counter TrueCount = getRegionCounter(E);
- Visit(E->getCond());
- if (!isa<BinaryConditionalOperator>(E)) {
- extendRegion(E->getTrueExpr());
- propagateCounts(TrueCount, E->getTrueExpr());
- }
- extendRegion(E->getFalseExpr());
- propagateCounts(subtractCounters(ParentCount, TrueCount),
- E->getFalseExpr());
- }
- void VisitBinLAnd(const BinaryOperator *E) {
- extendRegion(E);
- Visit(E->getLHS());
- extendRegion(E->getRHS());
- propagateCounts(getRegionCounter(E), E->getRHS());
- }
- void VisitBinLOr(const BinaryOperator *E) {
- extendRegion(E);
- Visit(E->getLHS());
- extendRegion(E->getRHS());
- propagateCounts(getRegionCounter(E), E->getRHS());
- }
- void VisitLambdaExpr(const LambdaExpr *LE) {
- // Lambdas are treated as their own functions for now, so we shouldn't
- // propagate counts into them.
- }
- };
- std::string getCoverageSection(const CodeGenModule &CGM) {
- return llvm::getInstrProfSectionName(
- llvm::IPSK_covmap,
- CGM.getContext().getTargetInfo().getTriple().getObjectFormat());
- }
- std::string normalizeFilename(StringRef Filename) {
- llvm::SmallString<256> Path(Filename);
- llvm::sys::fs::make_absolute(Path);
- llvm::sys::path::remove_dots(Path, /*remove_dot_dots=*/true);
- return Path.str().str();
- }
- } // end anonymous namespace
- static void dump(llvm::raw_ostream &OS, StringRef FunctionName,
- ArrayRef<CounterExpression> Expressions,
- ArrayRef<CounterMappingRegion> Regions) {
- OS << FunctionName << ":\n";
- CounterMappingContext Ctx(Expressions);
- for (const auto &R : Regions) {
- OS.indent(2);
- switch (R.Kind) {
- case CounterMappingRegion::CodeRegion:
- break;
- case CounterMappingRegion::ExpansionRegion:
- OS << "Expansion,";
- break;
- case CounterMappingRegion::SkippedRegion:
- OS << "Skipped,";
- break;
- }
- OS << "File " << R.FileID << ", " << R.LineStart << ":" << R.ColumnStart
- << " -> " << R.LineEnd << ":" << R.ColumnEnd << " = ";
- Ctx.dump(R.Count, OS);
- if (R.Kind == CounterMappingRegion::ExpansionRegion)
- OS << " (Expanded file = " << R.ExpandedFileID << ")";
- OS << "\n";
- }
- }
- void CoverageMappingModuleGen::addFunctionMappingRecord(
- llvm::GlobalVariable *NamePtr, StringRef NameValue, uint64_t FuncHash,
- const std::string &CoverageMapping, bool IsUsed) {
- llvm::LLVMContext &Ctx = CGM.getLLVMContext();
- if (!FunctionRecordTy) {
- #define COVMAP_FUNC_RECORD(Type, LLVMType, Name, Init) LLVMType,
- llvm::Type *FunctionRecordTypes[] = {
- #include "llvm/ProfileData/InstrProfData.inc"
- };
- FunctionRecordTy =
- llvm::StructType::get(Ctx, makeArrayRef(FunctionRecordTypes),
- /*isPacked=*/true);
- }
- #define COVMAP_FUNC_RECORD(Type, LLVMType, Name, Init) Init,
- llvm::Constant *FunctionRecordVals[] = {
- #include "llvm/ProfileData/InstrProfData.inc"
- };
- FunctionRecords.push_back(llvm::ConstantStruct::get(
- FunctionRecordTy, makeArrayRef(FunctionRecordVals)));
- if (!IsUsed)
- FunctionNames.push_back(
- llvm::ConstantExpr::getBitCast(NamePtr, llvm::Type::getInt8PtrTy(Ctx)));
- CoverageMappings.push_back(CoverageMapping);
- if (CGM.getCodeGenOpts().DumpCoverageMapping) {
- // Dump the coverage mapping data for this function by decoding the
- // encoded data. This allows us to dump the mapping regions which were
- // also processed by the CoverageMappingWriter which performs
- // additional minimization operations such as reducing the number of
- // expressions.
- std::vector<StringRef> Filenames;
- std::vector<CounterExpression> Expressions;
- std::vector<CounterMappingRegion> Regions;
- llvm::SmallVector<std::string, 16> FilenameStrs;
- llvm::SmallVector<StringRef, 16> FilenameRefs;
- FilenameStrs.resize(FileEntries.size());
- FilenameRefs.resize(FileEntries.size());
- for (const auto &Entry : FileEntries) {
- auto I = Entry.second;
- FilenameStrs[I] = normalizeFilename(Entry.first->getName());
- FilenameRefs[I] = FilenameStrs[I];
- }
- RawCoverageMappingReader Reader(CoverageMapping, FilenameRefs, Filenames,
- Expressions, Regions);
- if (Reader.read())
- return;
- dump(llvm::outs(), NameValue, Expressions, Regions);
- }
- }
- void CoverageMappingModuleGen::emit() {
- if (FunctionRecords.empty())
- return;
- llvm::LLVMContext &Ctx = CGM.getLLVMContext();
- auto *Int32Ty = llvm::Type::getInt32Ty(Ctx);
- // Create the filenames and merge them with coverage mappings
- llvm::SmallVector<std::string, 16> FilenameStrs;
- llvm::SmallVector<StringRef, 16> FilenameRefs;
- FilenameStrs.resize(FileEntries.size());
- FilenameRefs.resize(FileEntries.size());
- for (const auto &Entry : FileEntries) {
- auto I = Entry.second;
- FilenameStrs[I] = normalizeFilename(Entry.first->getName());
- FilenameRefs[I] = FilenameStrs[I];
- }
- std::string FilenamesAndCoverageMappings;
- llvm::raw_string_ostream OS(FilenamesAndCoverageMappings);
- CoverageFilenamesSectionWriter(FilenameRefs).write(OS);
- std::string RawCoverageMappings =
- llvm::join(CoverageMappings.begin(), CoverageMappings.end(), "");
- OS << RawCoverageMappings;
- size_t CoverageMappingSize = RawCoverageMappings.size();
- size_t FilenamesSize = OS.str().size() - CoverageMappingSize;
- // Append extra zeroes if necessary to ensure that the size of the filenames
- // and coverage mappings is a multiple of 8.
- if (size_t Rem = OS.str().size() % 8) {
- CoverageMappingSize += 8 - Rem;
- for (size_t I = 0, S = 8 - Rem; I < S; ++I)
- OS << '\0';
- }
- auto *FilenamesAndMappingsVal =
- llvm::ConstantDataArray::getString(Ctx, OS.str(), false);
- // Create the deferred function records array
- auto RecordsTy =
- llvm::ArrayType::get(FunctionRecordTy, FunctionRecords.size());
- auto RecordsVal = llvm::ConstantArray::get(RecordsTy, FunctionRecords);
- llvm::Type *CovDataHeaderTypes[] = {
- #define COVMAP_HEADER(Type, LLVMType, Name, Init) LLVMType,
- #include "llvm/ProfileData/InstrProfData.inc"
- };
- auto CovDataHeaderTy =
- llvm::StructType::get(Ctx, makeArrayRef(CovDataHeaderTypes));
- llvm::Constant *CovDataHeaderVals[] = {
- #define COVMAP_HEADER(Type, LLVMType, Name, Init) Init,
- #include "llvm/ProfileData/InstrProfData.inc"
- };
- auto CovDataHeaderVal = llvm::ConstantStruct::get(
- CovDataHeaderTy, makeArrayRef(CovDataHeaderVals));
- // Create the coverage data record
- llvm::Type *CovDataTypes[] = {CovDataHeaderTy, RecordsTy,
- FilenamesAndMappingsVal->getType()};
- auto CovDataTy = llvm::StructType::get(Ctx, makeArrayRef(CovDataTypes));
- llvm::Constant *TUDataVals[] = {CovDataHeaderVal, RecordsVal,
- FilenamesAndMappingsVal};
- auto CovDataVal =
- llvm::ConstantStruct::get(CovDataTy, makeArrayRef(TUDataVals));
- auto CovData = new llvm::GlobalVariable(
- CGM.getModule(), CovDataTy, true, llvm::GlobalValue::InternalLinkage,
- CovDataVal, llvm::getCoverageMappingVarName());
- CovData->setSection(getCoverageSection(CGM));
- CovData->setAlignment(8);
- // Make sure the data doesn't get deleted.
- CGM.addUsedGlobal(CovData);
- // Create the deferred function records array
- if (!FunctionNames.empty()) {
- auto NamesArrTy = llvm::ArrayType::get(llvm::Type::getInt8PtrTy(Ctx),
- FunctionNames.size());
- auto NamesArrVal = llvm::ConstantArray::get(NamesArrTy, FunctionNames);
- // This variable will *NOT* be emitted to the object file. It is used
- // to pass the list of names referenced to codegen.
- new llvm::GlobalVariable(CGM.getModule(), NamesArrTy, true,
- llvm::GlobalValue::InternalLinkage, NamesArrVal,
- llvm::getCoverageUnusedNamesVarName());
- }
- }
- unsigned CoverageMappingModuleGen::getFileID(const FileEntry *File) {
- auto It = FileEntries.find(File);
- if (It != FileEntries.end())
- return It->second;
- unsigned FileID = FileEntries.size();
- FileEntries.insert(std::make_pair(File, FileID));
- return FileID;
- }
- void CoverageMappingGen::emitCounterMapping(const Decl *D,
- llvm::raw_ostream &OS) {
- assert(CounterMap);
- CounterCoverageMappingBuilder Walker(CVM, *CounterMap, SM, LangOpts);
- Walker.VisitDecl(D);
- Walker.write(OS);
- }
- void CoverageMappingGen::emitEmptyMapping(const Decl *D,
- llvm::raw_ostream &OS) {
- EmptyCoverageMappingBuilder Walker(CVM, SM, LangOpts);
- Walker.VisitDecl(D);
- Walker.write(OS);
- }
|