tools/llvm-cov/TestingSupport.cpp

   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
  20 using namespace llvm;
  21 using namespace object;
  22
  23 int 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 }