#include "llvm/ProfileData/SampleProfReader.h"

Inheritance diagram for llvm::sampleprof::SampleProfileReaderBinary:

Public Member Functions
	SampleProfileReaderBinary (std::unique_ptr< MemoryBuffer > B, LLVMContext &C, SampleProfileFormat Format=SPF_None)
std::error_code	readHeader () override
	Read and validate the file header.
std::error_code	readImpl () override
	Read sample profiles from the associated file.
std::vector< FunctionId > *	getNameTable () override
	It includes all the names that have samples either in outline instance or inline instance.
Public Member Functions inherited from llvm::sampleprof::SampleProfileReader
	SampleProfileReader (std::unique_ptr< MemoryBuffer > B, LLVMContext &C, SampleProfileFormat Format=SPF_None)
virtual	~SampleProfileReader ()=default
void	setDiscriminatorMaskedBitFrom (FSDiscriminatorPass P)
	Set the bits for FS discriminators.
uint32_t	getDiscriminatorMask () const
	Get the bitmask the discriminators: For FS profiles, return the bit mask for this pass.
std::error_code	read ()
	The interface to read sample profiles from the associated file.
std::error_code	read (const DenseSet< StringRef > &FuncsToUse)
	Read sample profiles for the given functions.
LLVM_ABI void	dumpFunctionProfile (const FunctionSamples &FS, raw_ostream &OS=dbgs())
	Print the profile for `FunctionSamples` on stream `OS`.
virtual bool	collectFuncsFromModule ()
	Collect functions with definitions in Module M.
LLVM_ABI void	dump (raw_ostream &OS=dbgs())
	Print all the profiles on stream `OS`.
LLVM_ABI void	dumpJson (raw_ostream &OS=dbgs())
	Print all the profiles on stream `OS` in the JSON format.
FunctionSamples *	getSamplesFor (const Function &F)
	Return the samples collected for function `F`.
FunctionSamples *	getSamplesFor (StringRef Fname)
	Return the samples collected for function `F`.
SampleProfileMap &	getProfiles ()
	Return all the profiles.
void	reportError (int64_t LineNumber, const Twine &Msg) const
	Report a parse error message.
ProfileSummary &	getSummary () const
	Return the profile summary.
MemoryBuffer *	getBuffer () const
SampleProfileFormat	getFormat () const
	Return the profile format.
bool	profileIsProbeBased () const
	Whether input profile is based on pseudo probes.
bool	profileIsCS () const
	Whether input profile is fully context-sensitive.
bool	profileIsPreInlined () const
	Whether input profile contains ShouldBeInlined contexts.
bool	profileIsFS () const
	Whether input profile is flow-sensitive.
virtual std::unique_ptr< ProfileSymbolList >	getProfileSymbolList ()
virtual bool	dumpSectionInfo (raw_ostream &OS=dbgs())
bool	useMD5 () const
	Return whether names in the profile are all MD5 numbers.
virtual void	setProfileUseMD5 ()
	Force the profile to use MD5 in Sample contexts, even if function names are present.
void	setSkipFlatProf (bool Skip)
	Don't read profile without context if the flag is set.
virtual bool	hasUniqSuffix ()
	Return whether any name in the profile contains ".__uniq." suffix.
SampleProfileReaderItaniumRemapper *	getRemapper ()
void	setModule (const Module *Mod)
void	setFuncNameToProfNameMap (const HashKeyMap< std::unordered_map, FunctionId, FunctionId > &FPMap)

Protected Member Functions
template<typename T>
ErrorOr< T >	readNumber ()
	Read a numeric value of type T from the profile.
template<typename T>
ErrorOr< T >	readUnencodedNumber ()
	Read a numeric value of type T from the profile.
ErrorOr< StringRef >	readString ()
	Read a string from the profile.
template<typename T>
ErrorOr< size_t >	readStringIndex (T &Table)
	Read the string index and check whether it overflows the table.
std::error_code	readFuncProfile (const uint8_t *Start)
	Read the next function profile instance.
std::error_code	readFuncProfile (const uint8_t *Start, SampleProfileMap &Profiles)
std::error_code	readProfile (FunctionSamples &FProfile)
	Read the contents of the given profile instance.
std::error_code	readMagicIdent ()
	Read the contents of Magic number and Version number.
std::error_code	readSummary ()
	Read profile summary.
std::error_code	readNameTable ()
	Read the whole name table.
ErrorOr< FunctionId >	readStringFromTable (size_t *RetIdx=nullptr)
	Read a string indirectly via the name table. Optionally return the index.
ErrorOr< SampleContextFrames >	readContextFromTable (size_t *RetIdx=nullptr)
	Read a context indirectly via the CSNameTable.
ErrorOr< std::pair< SampleContext, uint64_t > >	readSampleContextFromTable ()
	Read a context indirectly via the CSNameTable if the profile has context, otherwise same as readStringFromTable, also return its hash value.
std::error_code	readCallsiteVTableProf (FunctionSamples &FProfile)
	Read all virtual functions' vtable access counts for `FProfile`.
std::error_code	readVTableTypeCountMap (TypeCountMap &M)
	Read bytes from the input buffer pointed by Data and decode them into `M`.
Protected Member Functions inherited from llvm::sampleprof::SampleProfileReader
LLVM_ABI void	computeSummary ()
	Compute summary for this profile.
virtual std::error_code	read (const DenseSet< StringRef > &FuncsToUse, SampleProfileMap &Profiles)
	Read sample profiles for the given functions and write them to the given profile map.

Protected Attributes
const uint8_t *	Data = nullptr
	Points to the current location in the buffer.
const uint8_t *	End = nullptr
	Points to the end of the buffer.
std::vector< FunctionId >	NameTable
	Function name table.
std::vector< SampleContextFrameVector >	CSNameTable
	CSNameTable is used to save full context vectors.
std::vector< uint64_t >	MD5SampleContextTable
	Table to cache MD5 values of sample contexts corresponding to readSampleContextFromTable(), used to index into Profiles or FuncOffsetTable.
const uint64_t *	MD5SampleContextStart = nullptr
	The starting address of the table of MD5 values of sample contexts.
Protected Attributes inherited from llvm::sampleprof::SampleProfileReader
SampleProfileMap	Profiles
	Map every function to its associated profile.
LLVMContext &	Ctx
	LLVM context used to emit diagnostics.
std::unique_ptr< MemoryBuffer >	Buffer
	Memory buffer holding the profile file.
std::unique_ptr< ProfileSummary >	Summary
	Profile summary information.
std::unique_ptr< SampleProfileReaderItaniumRemapper >	Remapper
const HashKeyMap< std::unordered_map, FunctionId, FunctionId > *	FuncNameToProfNameMap = nullptr
std::unordered_map< uint64_t, std::pair< const uint8_t , const uint8_t > >	FuncMetadataIndex
std::pair< const uint8_t , const uint8_t >	ProfileSecRange
bool	ProfileHasAttribute = false
	Whether the profile has attribute metadata.
bool	ProfileIsProbeBased = false
	Whether samples are collected based on pseudo probes.
bool	ProfileIsCS = false
	Whether function profiles are context-sensitive flat profiles.
bool	ProfileIsPreInlined = false
	Whether function profile contains ShouldBeInlined contexts.
uint32_t	CSProfileCount = 0
	Number of context-sensitive profiles.
bool	ProfileIsFS = false
	Whether the function profiles use FS discriminators.
bool	ReadVTableProf = false
	If true, the profile has vtable profiles and reader should decode them to parse profiles correctly.
SampleProfileFormat	Format = SPF_None
	The format of sample.
const Module *	M = nullptr
	The current module being compiled if SampleProfileReader is used by compiler.
uint32_t	MaskedBitFrom = 31
	Zero out the discriminator bits higher than bit MaskedBitFrom (0 based).
bool	ProfileIsMD5 = false
	Whether the profile uses MD5 for Sample Contexts and function names.
bool	SkipFlatProf = false
	If SkipFlatProf is true, skip functions marked with !Flat in text mode or sections with SecFlagFlat flag in ExtBinary mode.

Additional Inherited Members
Static Public Member Functions inherited from llvm::sampleprof::SampleProfileReader
static LLVM_ABI ErrorOr< std::unique_ptr< SampleProfileReader > >	create (StringRef Filename, LLVMContext &C, vfs::FileSystem &FS, FSDiscriminatorPass P=FSDiscriminatorPass::Base, StringRef RemapFilename="")
	Create a sample profile reader appropriate to the file format.
static LLVM_ABI ErrorOr< std::unique_ptr< SampleProfileReader > >	create (std::unique_ptr< MemoryBuffer > &B, LLVMContext &C, vfs::FileSystem &FS, FSDiscriminatorPass P=FSDiscriminatorPass::Base, StringRef RemapFilename="")
	Create a sample profile reader from the supplied memory buffer.
Static Protected Member Functions inherited from llvm::sampleprof::SampleProfileReader
static std::unique_ptr< ProfileSummary >	takeSummary (SampleProfileReader &Reader)
	Take ownership of the summary of this reader.

Detailed Description

Definition at line 640 of file SampleProfReader.h.

Constructor & Destructor Documentation

◆ SampleProfileReaderBinary()

llvm::sampleprof::SampleProfileReaderBinary::SampleProfileReaderBinary	(	std::unique_ptr< MemoryBuffer >	B,
		LLVMContext &	C,
		SampleProfileFormat	Format = SPF_None )

inline

Definition at line 642 of file SampleProfReader.h.

References B(), llvm::CallingConv::C, llvm::sampleprof::SampleProfileReader::Format, llvm::move(), llvm::sampleprof::SampleProfileReader::SampleProfileReader(), and llvm::sampleprof::SPF_None.

Referenced by llvm::sampleprof::SampleProfileReaderExtBinaryBase::SampleProfileReaderExtBinaryBase(), and llvm::sampleprof::SampleProfileReaderRawBinary::SampleProfileReaderRawBinary().

Member Function Documentation

◆ getNameTable()

std::vector< FunctionId > * llvm::sampleprof::SampleProfileReaderBinary::getNameTable ( )

inlineoverridevirtual

It includes all the names that have samples either in outline instance or inline instance.

Reimplemented from llvm::sampleprof::SampleProfileReader.

Definition at line 654 of file SampleProfReader.h.

References NameTable.

◆ readCallsiteVTableProf()

std::error_code SampleProfileReaderBinary::readCallsiteVTableProf ( FunctionSamples & FProfile )

protected

Read all virtual functions' vtable access counts for FProfile.

Definition at line 681 of file SampleProfReader.cpp.

References assert(), llvm::sampleprof::SampleProfileReader::getDiscriminatorMask(), llvm::sampleprof::FunctionSamples::getTypeSamplesAt(), I, llvm::illegal_line_offset, isOffsetLegal(), readNumber(), llvm::sampleprof::SampleProfileReader::ReadVTableProf, readVTableTypeCountMap(), and llvm::success.

Referenced by readProfile().

◆ readContextFromTable()

ErrorOr< SampleContextFrames > SampleProfileReaderBinary::readContextFromTable ( size_t * RetIdx = nullptr )

protected

Read a context indirectly via the CSNameTable.

Optionally return the index.

Definition at line 611 of file SampleProfReader.cpp.

References CSNameTable, readNumber(), and llvm::truncated_name_table.

Referenced by readSampleContextFromTable().

◆ readFuncProfile() [1/2]

std::error_code SampleProfileReaderBinary::readFuncProfile ( const uint8_t * Start )

protected

Read the next function profile instance.

Definition at line 826 of file SampleProfReader.cpp.

References llvm::sampleprof::SampleProfileReader::Profiles, and readFuncProfile().

Referenced by readFuncProfile(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncProfiles(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncProfiles(), and readImpl().

◆ readFuncProfile() [2/2]

std::error_code SampleProfileReaderBinary::readFuncProfile	(	const uint8_t *	Start,
		SampleProfileMap &	Profiles )

protected

Definition at line 799 of file SampleProfReader.cpp.

References llvm::sampleprof::FunctionSamples::addHeadSamples(), llvm::sampleprof::SampleProfileReader::CSProfileCount, Data, llvm::sampleprof::SampleProfileReader::Profiles, readNumber(), readProfile(), readSampleContextFromTable(), llvm::sampleprof::FunctionSamples::setContext(), and llvm::success.

◆ readHeader()

std::error_code SampleProfileReaderBinary::readHeader ( )

overridevirtual

Read and validate the file header.

Implements llvm::sampleprof::SampleProfileReader.

Reimplemented in llvm::sampleprof::SampleProfileReaderExtBinaryBase.

Definition at line 1639 of file SampleProfReader.cpp.

References llvm::sampleprof::SampleProfileReader::Buffer, Data, End, readMagicIdent(), readNameTable(), readSummary(), and llvm::success.

◆ readImpl()

std::error_code SampleProfileReaderBinary::readImpl ( )

overridevirtual

Read sample profiles from the associated file.

Implements llvm::sampleprof::SampleProfileReader.

Reimplemented in llvm::sampleprof::SampleProfileReaderExtBinaryBase.

Definition at line 830 of file SampleProfReader.cpp.

References Data, End, llvm::sampleprof::FunctionSamples::ProfileIsFS, llvm::sampleprof::SampleProfileReader::ProfileIsFS, ProfileIsFSDisciminator, readFuncProfile(), and llvm::success.

◆ readMagicIdent()

std::error_code SampleProfileReaderBinary::readMagicIdent ( )

protected

Read the contents of Magic number and Version number.

Definition at line 1621 of file SampleProfReader.cpp.

References readNumber(), llvm::sampleprof::SPVersion(), llvm::success, llvm::unsupported_version, and llvm::Version.

Referenced by readHeader(), and llvm::sampleprof::SampleProfileReaderExtBinaryBase::readHeader().

◆ readNameTable()

std::error_code SampleProfileReaderBinary::readNameTable ( )

protected

Read the whole name table.

Definition at line 1233 of file SampleProfReader.cpp.

References llvm::sampleprof::FunctionId::getHashCode(), I, MD5SampleContextStart, MD5SampleContextTable, NameTable, llvm::sampleprof::SampleProfileReader::ProfileIsCS, readNumber(), readString(), Size, llvm::success, and llvm::sampleprof::SampleProfileReader::useMD5().

Referenced by readHeader(), and llvm::sampleprof::SampleProfileReaderExtBinaryBase::readNameTableSec().

◆ readNumber()

template<typename T>

ErrorOr< T > SampleProfileReaderBinary::readNumber ( )

protected

Read a numeric value of type T from the profile.

If an error occurs during decoding, a diagnostic message is emitted and EC is set.

Returns: the read value.

Definition at line 547 of file SampleProfReader.cpp.

References Data, llvm::decodeULEB128(), End, llvm::malformed, llvm::sampleprof::SampleProfileReader::reportError(), T, and llvm::truncated.

Referenced by readCallsiteVTableProf(), readContextFromTable(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readCSNameTableSec(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncMetadata(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncOffsetTable(), readFuncProfile(), readMagicIdent(), readNameTable(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readNameTableSec(), readProfile(), readStringIndex(), readSummary(), and readVTableTypeCountMap().

◆ readProfile()

std::error_code SampleProfileReaderBinary::readProfile ( FunctionSamples & FProfile )

protected

Read the contents of the given profile instance.

Definition at line 713 of file SampleProfReader.cpp.

References llvm::sampleprof::FunctionSamples::addBodySamples(), llvm::sampleprof::FunctionSamples::addCalledTargetSamples(), llvm::sampleprof::FunctionSamples::addTotalSamples(), llvm::sampleprof::FunctionSamples::functionSamplesAt(), llvm::sampleprof::SampleProfileReader::getDiscriminatorMask(), I, llvm::illegal_line_offset, isOffsetLegal(), readCallsiteVTableProf(), readNumber(), readProfile(), readStringFromTable(), llvm::sampleprof::SampleProfileReader::ReadVTableProf, llvm::sampleprof::FunctionSamples::setFunction(), and llvm::success.

Referenced by readFuncProfile(), and readProfile().

◆ readSampleContextFromTable()

ErrorOr< std::pair< SampleContext, uint64_t > > SampleProfileReaderBinary::readSampleContextFromTable ( )

protected

Read a context indirectly via the CSNameTable if the profile has context, otherwise same as readStringFromTable, also return its hash value.

Definition at line 623 of file SampleProfReader.cpp.

References assert(), MD5SampleContextStart, MD5SampleContextTable, llvm::sampleprof::SampleProfileReader::ProfileIsCS, llvm::support::endian::read64le(), readContextFromTable(), readStringFromTable(), and llvm::support::endian::write64le().

Referenced by llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncMetadata(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncMetadata(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readFuncOffsetTable(), and readFuncProfile().

◆ readString()

ErrorOr< StringRef > SampleProfileReaderBinary::readString ( )

protected

Read a string from the profile.

If an error occurs during decoding, a diagnostic message is emitted and EC is set.

Returns: the read value.

Definition at line 565 of file SampleProfReader.cpp.

References Data, End, llvm::sampleprof::SampleProfileReader::reportError(), and llvm::truncated.

Referenced by readNameTable().

◆ readStringFromTable()

ErrorOr< FunctionId > SampleProfileReaderBinary::readStringFromTable ( size_t * RetIdx = nullptr )

protected

Read a string indirectly via the name table. Optionally return the index.

Definition at line 601 of file SampleProfReader.cpp.

References NameTable, and readStringIndex().

Referenced by llvm::sampleprof::SampleProfileReaderExtBinaryBase::readCSNameTableSec(), readProfile(), readSampleContextFromTable(), and readVTableTypeCountMap().

◆ readStringIndex()

template<typename T>

ErrorOr< size_t > SampleProfileReaderBinary::readStringIndex ( T & Table )

inlineprotected

Read the string index and check whether it overflows the table.

Definition at line 591 of file SampleProfReader.cpp.

References readNumber(), T, and llvm::truncated_name_table.

Referenced by readStringFromTable().

◆ readSummary()

std::error_code SampleProfileReaderBinary::readSummary ( )

protected

Read profile summary.

Definition at line 1672 of file SampleProfReader.cpp.

References llvm::ProfileSummary::PSK_Sample, readNumber(), llvm::success, and llvm::sampleprof::SampleProfileReader::Summary.

Referenced by readHeader(), and llvm::sampleprof::SampleProfileReaderExtBinaryBase::readOneSection().

◆ readUnencodedNumber()

template<typename T>

ErrorOr< T > SampleProfileReaderBinary::readUnencodedNumber ( )

protected

Read a numeric value of type T from the profile.

The value is saved without encoded.

Definition at line 578 of file SampleProfReader.cpp.

References Data, End, llvm::support::endian::readNext(), llvm::sampleprof::SampleProfileReader::reportError(), T, and llvm::truncated.

Referenced by llvm::sampleprof::SampleProfileReaderExtBinaryBase::readSecHdrTable(), and llvm::sampleprof::SampleProfileReaderExtBinaryBase::readSecHdrTableEntry().

◆ readVTableTypeCountMap()

std::error_code SampleProfileReaderBinary::readVTableTypeCountMap ( TypeCountMap & M )

protected

Read bytes from the input buffer pointed by Data and decode them into M.

Data will be advanced to the end of the read bytes when this function returns. Returns error if any.

Definition at line 652 of file SampleProfReader.cpp.

References llvm::sampleprof::SampleProfileReader::Buffer, llvm::sampleprof::SampleProfileReader::Ctx, llvm::DS_Warning, I, llvm::sampleprof::SampleProfileReader::M, readNumber(), readStringFromTable(), and llvm::success.

Referenced by readCallsiteVTableProf().

Member Data Documentation

◆ CSNameTable

std::vector<SampleContextFrameVector> llvm::sampleprof::SampleProfileReaderBinary::CSNameTable

protected

CSNameTable is used to save full context vectors.

It is the backing buffer for SampleContextFrames.

Definition at line 729 of file SampleProfReader.h.

Referenced by readContextFromTable(), and llvm::sampleprof::SampleProfileReaderExtBinaryBase::readCSNameTableSec().

◆ Data

const uint8_t* llvm::sampleprof::SampleProfileReaderBinary::Data = nullptr

protected

Points to the current location in the buffer.

Definition at line 719 of file SampleProfReader.h.

◆ End

const uint8_t* llvm::sampleprof::SampleProfileReaderBinary::End = nullptr

protected

Points to the end of the buffer.

Definition at line 722 of file SampleProfReader.h.

◆ MD5SampleContextStart

const uint64_t* llvm::sampleprof::SampleProfileReaderBinary::MD5SampleContextStart = nullptr

protected

The starting address of the table of MD5 values of sample contexts.

For fixed length MD5 non-CS profile it is same as MD5NameMemStart because hashes of non-CS contexts are already in the profile. Otherwise it points to the start of MD5SampleContextTable.

Definition at line 740 of file SampleProfReader.h.

Referenced by llvm::sampleprof::SampleProfileReaderExtBinaryBase::readCSNameTableSec(), readNameTable(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readNameTableSec(), and readSampleContextFromTable().

◆ MD5SampleContextTable

std::vector<uint64_t> llvm::sampleprof::SampleProfileReaderBinary::MD5SampleContextTable

protected

Table to cache MD5 values of sample contexts corresponding to readSampleContextFromTable(), used to index into Profiles or FuncOffsetTable.

Definition at line 734 of file SampleProfReader.h.

Referenced by llvm::sampleprof::SampleProfileReaderExtBinaryBase::readCSNameTableSec(), readNameTable(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readNameTableSec(), and readSampleContextFromTable().

◆ NameTable

std::vector<FunctionId> llvm::sampleprof::SampleProfileReaderBinary::NameTable

protected

Function name table.

Definition at line 725 of file SampleProfReader.h.

Referenced by getNameTable(), readNameTable(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readNameTableSec(), and readStringFromTable().

The documentation for this class was generated from the following files:

include/llvm/ProfileData/SampleProfReader.h
lib/ProfileData/SampleProfReader.cpp

Public Member Functions

Protected Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ SampleProfileReaderBinary()

Member Function Documentation

◆ getNameTable()

◆ readCallsiteVTableProf()

◆ readContextFromTable()

◆ readFuncProfile() [1/2]

◆ readFuncProfile() [2/2]

◆ readHeader()

◆ readImpl()

◆ readMagicIdent()

◆ readNameTable()

◆ readNumber()

◆ readProfile()

◆ readSampleContextFromTable()

◆ readString()

◆ readStringFromTable()

◆ readStringIndex()

◆ readSummary()

◆ readUnencodedNumber()

◆ readVTableTypeCountMap()

Member Data Documentation

◆ CSNameTable

◆ Data

◆ End

◆ MD5SampleContextStart

◆ MD5SampleContextTable

◆ NameTable