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