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/ProfileData/InstrProf.h"
  12 #include "llvm/Support/CommandLine.h"
  13 #include "llvm/Support/LEB128.h"
  14 #include "llvm/Support/raw_ostream.h"
  15 #include <functional>
  16 #include <system_error>
  17
  18 using namespace llvm;
  19 using namespace object;
  20
  21 int convertForTestingMain(int argc, const char *argv[]) {
  22   cl::opt<std::string> InputSourceFile(cl::Positional, cl::Required,
  23                                        cl::desc("<Source file>"));
  24
  25   cl::opt<std::string> OutputFilename(
  26       "o", cl::Required,
  27       cl::desc(
  28           "File with the profile data obtained after an instrumented run"));
  29
  30   cl::ParseCommandLineOptions(argc, argv, "LLVM code coverage tool\n");
  31
  32   auto ObjErr = llvm::object::ObjectFile::createObjectFile(InputSourceFile);
  33   if (!ObjErr) {
  34     std::string Buf;
  35     raw_string_ostream OS(Buf);
  36     logAllUnhandledErrors(ObjErr.takeError(), OS);
  37     OS.flush();
  38     errs() << "error: " << Buf;
  39     return 1;
  40   }
  41   ObjectFile *OF = ObjErr.get().getBinary();
  42   auto BytesInAddress = OF->getBytesInAddress();
  43   if (BytesInAddress != 8) {
  44     errs() << "error: 64 bit binary expected\n";
  45     return 1;
  46   }
  47
  48   // Look for the sections that we are interested in.
  49   int FoundSectionCount = 0;
  50   SectionRef ProfileNames, CoverageMapping;
  51   auto ObjFormat = OF->getTripleObjectFormat();
  52   for (const auto &Section : OF->sections()) {
  53     StringRef Name;
  54     if (Section.getName(Name))
  55       return 1;
  56     if (Name == llvm::getInstrProfSectionName(IPSK_name, ObjFormat,
  57                                               /*AddSegmentInfo=*/false)) {
  58       ProfileNames = Section;
  59     } else if (Name == llvm::getInstrProfSectionName(
  60                            IPSK_covmap, ObjFormat, /*AddSegmentInfo=*/false)) {
  61       CoverageMapping = Section;
  62     } else
  63       continue;
  64     ++FoundSectionCount;
  65   }
  66   if (FoundSectionCount != 2)
  67     return 1;
  68
  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))
  75     return 1;
  76
  77   int FD;
  78   if (auto Err = sys::fs::openFileForWrite(OutputFilename, FD)) {
  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;
  88   // Coverage mapping data is expected to have an alignment of 8.
  89   for (unsigned Pad = OffsetToAlignment(OS.tell(), 8); Pad; --Pad)
  90     OS.write(uint8_t(0));
  91   OS << CoverageMappingData;
  92
  93   return 0;
  94 }