1 //===- TestingSupport.cpp - Convert objects files into test files --------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/Object/ObjectFile.h"
11 #include "llvm/Support/raw_ostream.h"
12 #include "llvm/Support/LEB128.h"
13 #include "llvm/Support/CommandLine.h"
14 #include "llvm/Support/ManagedStatic.h"
15 #include "llvm/Support/MemoryObject.h"
16 #include "llvm/Support/Signals.h"
17 #include "llvm/Support/PrettyStackTrace.h"
18 #include <system_error>
22 using namespace object;
24 int convert_for_testing_main(int argc, const char **argv) {
25 sys::PrintStackTraceOnErrorSignal();
26 PrettyStackTraceProgram X(argc, argv);
27 llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
29 cl::opt<std::string> InputSourceFile(cl::Positional, cl::Required,
30 cl::desc("<Source file>"));
32 cl::opt<std::string> OutputFilename(
35 "File with the profile data obtained after an instrumented run"));
37 cl::ParseCommandLineOptions(argc, argv, "LLVM code coverage tool\n");
39 auto ObjErr = llvm::object::ObjectFile::createObjectFile(InputSourceFile);
40 if (auto Err = ObjErr.getError()) {
41 errs() << "error: " << Err.message() << "\n";
44 ObjectFile *OF = ObjErr.get().getBinary().get();
45 auto BytesInAddress = OF->getBytesInAddress();
46 if (BytesInAddress != 8) {
47 errs() << "error: 64 bit binary expected\n";
51 // Look for the sections that we are interested in.
52 int FoundSectionCount = 0;
53 SectionRef ProfileNames, CoverageMapping;
54 for (const auto &Section : OF->sections()) {
56 if (Section.getName(Name))
58 if (Name == "__llvm_prf_names") {
59 ProfileNames = Section;
60 } else if (Name == "__llvm_covmap") {
61 CoverageMapping = Section;
66 if (FoundSectionCount != 2)
69 // Get the contents of the given sections.
70 uint64_t ProfileNamesAddress = ProfileNames.getAddress();
71 StringRef CoverageMappingData;
72 StringRef ProfileNamesData;
73 if (CoverageMapping.getContents(CoverageMappingData) ||
74 ProfileNames.getContents(ProfileNamesData))
79 sys::fs::openFileForWrite(OutputFilename, FD, sys::fs::F_None)) {
80 errs() << "error: " << Err.message() << "\n";
84 raw_fd_ostream OS(FD, true);
85 OS << "llvmcovmtestdata";
86 encodeULEB128(ProfileNamesData.size(), OS);
87 encodeULEB128(ProfileNamesAddress, OS);
88 OS << ProfileNamesData << CoverageMappingData;