namespace llvm {
namespace coverage {
-/// \brief Writer of the filenames section for the instrumentation
-/// based code coverage.
-class CoverageFilenamesSectionWriter {
- ArrayRef<StringRef> Filenames;
-
-public:
- CoverageFilenamesSectionWriter(ArrayRef<StringRef> Filenames)
- : Filenames(Filenames) {}
-
- /// \brief Write encoded filenames to the given output stream.
- void write(raw_ostream &OS);
-};
-
/// \brief Writer for instrumentation based coverage mapping data.
class CoverageMappingWriter {
ArrayRef<unsigned> VirtualFileMapping;
void write(raw_ostream &OS);
};
+/// \brief Encode a list of filenames and raw coverage mapping data using the
+/// latest coverage data format.
+///
+/// Set \p FilenamesSize to the size of the filenames section.
+///
+/// Set \p CoverageMappingsSize to the size of the coverage mapping section
+/// (including any necessary padding bytes).
+Expected<std::string> encodeFilenamesAndRawMappings(
+ ArrayRef<std::string> Filenames, ArrayRef<std::string> CoverageMappings,
+ size_t &FilenamesSize, size_t &CoverageMappingsSize);
+
} // end namespace coverage
} // end namespace llvm
using namespace llvm;
using namespace coverage;
-void CoverageFilenamesSectionWriter::write(raw_ostream &OS) {
- encodeULEB128(Filenames.size(), OS);
- for (const auto &Filename : Filenames) {
- encodeULEB128(Filename.size(), OS);
- OS << Filename;
- }
-}
-
namespace {
/// \brief Gather only the expressions that are used by the mapping
/// regions in this function.
// Ensure that all file ids have at least one mapping region.
assert(CurrentFileID == (VirtualFileMapping.size() - 1));
}
+
+/// \brief Encode coverage data into \p OS.
+static void encodeCoverageData(ArrayRef<std::string> Filenames,
+ ArrayRef<std::string> CoverageMappings,
+ size_t &FilenamesSize,
+ size_t &CoverageMappingsSize, raw_ostream &OS) {
+ size_t OSOffset = OS.GetNumBytesInBuffer();
+
+ // Encode the filenames.
+ encodeULEB128(Filenames.size(), OS);
+ for (const auto &Filename : Filenames) {
+ encodeULEB128(Filename.size(), OS);
+ OS << Filename;
+ }
+
+ FilenamesSize = OS.GetNumBytesInBuffer() - OSOffset;
+
+ // Encode the coverage mappings.
+ for (const auto &RawMapping : CoverageMappings)
+ OS << RawMapping;
+
+ // Pad the output stream to an 8-byte boundary. Account for the padding bytes
+ // in \p CoverageMappingsSize.
+ if (size_t Rem = OS.GetNumBytesInBuffer() % 8) {
+ CoverageMappingsSize += 8 - Rem;
+ for (size_t I = 0, S = 8 - Rem; I < S; ++I)
+ OS << '\0';
+ }
+
+ CoverageMappingsSize = OS.GetNumBytesInBuffer() - FilenamesSize - OSOffset;
+}
+
+namespace llvm {
+namespace coverage {
+
+Expected<std::string> encodeFilenamesAndRawMappings(
+ ArrayRef<std::string> Filenames, ArrayRef<std::string> CoverageMappings,
+ size_t &FilenamesSize, size_t &CoverageMappingsSize) {
+ std::string CoverageData;
+ {
+ raw_string_ostream OS{CoverageData};
+ encodeCoverageData(Filenames, CoverageMappings, FilenamesSize,
+ CoverageMappingsSize, OS);
+ }
+ return std::move(CoverageData);
+}
+
+} // end namespace coverage
+} // end namespace llvm