Home | History | Annotate | Download | only in llvm-cov 
      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 }
     95