diff options
| author | bors <bors@rust-lang.org> | 2020-07-19 07:25:18 +0000 |
|---|---|---|
| committer | bors <bors@rust-lang.org> | 2020-07-19 07:25:18 +0000 |
| commit | 47ea6d90b073ab977cf072e2f5f46d63de532cc6 (patch) | |
| tree | d408537bbf5d8d3063d19914f6e30efe02c74681 /src/rustllvm/CoverageMappingWrapper.cpp | |
| parent | 0701419e96d94e5493c7ebfcecb66511ab0aa778 (diff) | |
| parent | a6f8b8a2116f0ea7e31d572d3120508678ed8079 (diff) | |
| download | rust-47ea6d90b073ab977cf072e2f5f46d63de532cc6.tar.gz rust-47ea6d90b073ab977cf072e2f5f46d63de532cc6.zip | |
Auto merge of #74091 - richkadel:llvm-coverage-map-gen-4, r=tmandry
Generating the coverage map @tmandry @wesleywiser rustc now generates the coverage map and can support (limited) coverage report generation, at the function level. Example commands to generate a coverage report: ```shell $ BUILD=$HOME/rust/build/x86_64-unknown-linux-gnu $ $BUILD/stage1/bin/rustc -Zinstrument-coverage \ $HOME/rust/src/test/run-make-fulldeps/instrument-coverage/main.rs $ LLVM_PROFILE_FILE="main.profraw" ./main called $ $BUILD/llvm/bin/llvm-profdata merge -sparse main.profraw -o main.profdata $ $BUILD/llvm/bin/llvm-cov show --instr-profile=main.profdata main ```  r? @wesleywiser Rust compiler MCP rust-lang/compiler-team#278 Relevant issue: #34701 - Implement support for LLVMs code coverage instrumentation
Diffstat (limited to 'src/rustllvm/CoverageMappingWrapper.cpp')
| -rw-r--r-- | src/rustllvm/CoverageMappingWrapper.cpp | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/src/rustllvm/CoverageMappingWrapper.cpp b/src/rustllvm/CoverageMappingWrapper.cpp new file mode 100644 index 00000000000..c6c4cdb5562 --- /dev/null +++ b/src/rustllvm/CoverageMappingWrapper.cpp @@ -0,0 +1,115 @@ +#include "rustllvm.h" +#include "llvm/ProfileData/Coverage/CoverageMapping.h" +#include "llvm/ProfileData/Coverage/CoverageMappingWriter.h" +#include "llvm/ProfileData/InstrProf.h" +#include "llvm/ADT/ArrayRef.h" + +#include <iostream> + +using namespace llvm; + +extern "C" SmallVectorTemplateBase<coverage::CounterExpression> + *LLVMRustCoverageSmallVectorCounterExpressionCreate() { + return new SmallVector<coverage::CounterExpression, 32>(); +} + +extern "C" void LLVMRustCoverageSmallVectorCounterExpressionDispose( + SmallVectorTemplateBase<coverage::CounterExpression> *Vector) { + delete Vector; +} + +extern "C" void LLVMRustCoverageSmallVectorCounterExpressionAdd( + SmallVectorTemplateBase<coverage::CounterExpression> *Expressions, + coverage::CounterExpression::ExprKind Kind, + unsigned LeftIndex, + unsigned RightIndex) { + auto LHS = coverage::Counter::getCounter(LeftIndex); + auto RHS = coverage::Counter::getCounter(RightIndex); + Expressions->push_back(coverage::CounterExpression { Kind, LHS, RHS }); +} + +extern "C" SmallVectorTemplateBase<coverage::CounterMappingRegion> + *LLVMRustCoverageSmallVectorCounterMappingRegionCreate() { + return new SmallVector<coverage::CounterMappingRegion, 32>(); +} + +extern "C" void LLVMRustCoverageSmallVectorCounterMappingRegionDispose( + SmallVectorTemplateBase<coverage::CounterMappingRegion> *Vector) { + delete Vector; +} + +extern "C" void LLVMRustCoverageSmallVectorCounterMappingRegionAdd( + SmallVectorTemplateBase<coverage::CounterMappingRegion> *MappingRegions, + unsigned Index, + unsigned FileID, + unsigned LineStart, + unsigned ColumnStart, + unsigned LineEnd, + unsigned ColumnEnd) { + auto Counter = coverage::Counter::getCounter(Index); + MappingRegions->push_back(coverage::CounterMappingRegion::makeRegion( + Counter, FileID, LineStart, + ColumnStart, LineEnd, ColumnEnd)); + + // FIXME(richkadel): As applicable, implement additional CounterMappingRegion types using the + // static method alternatives to `coverage::CounterMappingRegion::makeRegion`: + // + // makeExpansion(unsigned FileID, unsigned ExpandedFileID, unsigned LineStart, + // unsigned ColumnStart, unsigned LineEnd, unsigned ColumnEnd) { + // makeSkipped(unsigned FileID, unsigned LineStart, unsigned ColumnStart, + // unsigned LineEnd, unsigned ColumnEnd) { + // makeGapRegion(Counter Count, unsigned FileID, unsigned LineStart, + // unsigned ColumnStart, unsigned LineEnd, unsigned ColumnEnd) { +} + +extern "C" void LLVMRustCoverageWriteFilenamesSectionToBuffer( + const char* const Filenames[], + size_t FilenamesLen, + RustStringRef BufferOut) { + SmallVector<StringRef,32> FilenameRefs; + for (size_t i = 0; i < FilenamesLen; i++) { + FilenameRefs.push_back(StringRef(Filenames[i])); + } + auto FilenamesWriter = coverage::CoverageFilenamesSectionWriter( + makeArrayRef(FilenameRefs)); + RawRustStringOstream OS(BufferOut); + FilenamesWriter.write(OS); +} + +extern "C" void LLVMRustCoverageWriteMappingToBuffer( + const unsigned *VirtualFileMappingIDs, + unsigned NumVirtualFileMappingIDs, + const SmallVectorImpl<coverage::CounterExpression> *Expressions, + SmallVectorImpl<coverage::CounterMappingRegion> *MappingRegions, + RustStringRef BufferOut) { + auto CoverageMappingWriter = coverage::CoverageMappingWriter( + makeArrayRef(VirtualFileMappingIDs, NumVirtualFileMappingIDs), + makeArrayRef(*Expressions), + MutableArrayRef<coverage::CounterMappingRegion> { *MappingRegions }); + RawRustStringOstream OS(BufferOut); + CoverageMappingWriter.write(OS); +} + +extern "C" uint64_t LLVMRustCoverageComputeHash(const char *Name) { + StringRef NameRef(Name); + return IndexedInstrProf::ComputeHash(NameRef); +} + +extern "C" void LLVMRustCoverageWriteSectionNameToString(LLVMModuleRef M, + RustStringRef Str) { + Triple TargetTriple(unwrap(M)->getTargetTriple()); + auto name = getInstrProfSectionName(IPSK_covmap, + TargetTriple.getObjectFormat()); + RawRustStringOstream OS(Str); + OS << name; +} + +extern "C" void LLVMRustCoverageWriteMappingVarNameToString(RustStringRef Str) { + auto name = getCoverageMappingVarName(); + RawRustStringOstream OS(Str); + OS << name; +} + +extern "C" uint32_t LLVMRustCoverageMappingVersion() { + return coverage::CovMapVersion::CurrentVersion; +} |
