mirror of
https://github.com/RPCS3/llvm.git
synced 2024-12-26 22:26:16 +00:00
8175be535a
There are two methods in SectionRef that can fail: * getName: The index into the string table can be invalid. * getContents: The section might point to invalid contents. Every other method will always succeed and returning and std::error_code just complicates the code. For example, a section can have an invalid alignment, but if we are able to get to the section structure at all and create a SectionRef, we will always be able to read that invalid alignment. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@219314 91177308-0d34-0410-b5e6-96231b3b80d8
92 lines
2.8 KiB
C++
92 lines
2.8 KiB
C++
//===- TestingSupport.cpp - Convert objects files into test files --------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Object/ObjectFile.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/LEB128.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/MemoryObject.h"
|
|
#include "llvm/Support/Signals.h"
|
|
#include "llvm/Support/PrettyStackTrace.h"
|
|
#include <system_error>
|
|
#include <functional>
|
|
|
|
using namespace llvm;
|
|
using namespace object;
|
|
|
|
int convert_for_testing_main(int argc, const char **argv) {
|
|
sys::PrintStackTraceOnErrorSignal();
|
|
PrettyStackTraceProgram X(argc, argv);
|
|
llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
|
|
|
|
cl::opt<std::string> InputSourceFile(cl::Positional, cl::Required,
|
|
cl::desc("<Source file>"));
|
|
|
|
cl::opt<std::string> OutputFilename(
|
|
"o", cl::Required,
|
|
cl::desc(
|
|
"File with the profile data obtained after an instrumented run"));
|
|
|
|
cl::ParseCommandLineOptions(argc, argv, "LLVM code coverage tool\n");
|
|
|
|
auto ObjErr = llvm::object::ObjectFile::createObjectFile(InputSourceFile);
|
|
if (auto Err = ObjErr.getError()) {
|
|
errs() << "error: " << Err.message() << "\n";
|
|
return 1;
|
|
}
|
|
ObjectFile *OF = ObjErr.get().getBinary().get();
|
|
auto BytesInAddress = OF->getBytesInAddress();
|
|
if (BytesInAddress != 8) {
|
|
errs() << "error: 64 bit binary expected\n";
|
|
return 1;
|
|
}
|
|
|
|
// Look for the sections that we are interested in.
|
|
int FoundSectionCount = 0;
|
|
SectionRef ProfileNames, CoverageMapping;
|
|
for (const auto &Section : OF->sections()) {
|
|
StringRef Name;
|
|
if (Section.getName(Name))
|
|
return 1;
|
|
if (Name == "__llvm_prf_names") {
|
|
ProfileNames = Section;
|
|
} else if (Name == "__llvm_covmap") {
|
|
CoverageMapping = Section;
|
|
} else
|
|
continue;
|
|
++FoundSectionCount;
|
|
}
|
|
if (FoundSectionCount != 2)
|
|
return 1;
|
|
|
|
// Get the contents of the given sections.
|
|
uint64_t ProfileNamesAddress = ProfileNames.getAddress();
|
|
StringRef CoverageMappingData;
|
|
StringRef ProfileNamesData;
|
|
if (CoverageMapping.getContents(CoverageMappingData) ||
|
|
ProfileNames.getContents(ProfileNamesData))
|
|
return 1;
|
|
|
|
int FD;
|
|
if (auto Err =
|
|
sys::fs::openFileForWrite(OutputFilename, FD, sys::fs::F_None)) {
|
|
errs() << "error: " << Err.message() << "\n";
|
|
return 1;
|
|
}
|
|
|
|
raw_fd_ostream OS(FD, true);
|
|
OS << "llvmcovmtestdata";
|
|
encodeULEB128(ProfileNamesData.size(), OS);
|
|
encodeULEB128(ProfileNamesAddress, OS);
|
|
OS << ProfileNamesData << CoverageMappingData;
|
|
|
|
return 0;
|
|
}
|