6024e5c395
clang 20.0.0 (based on r547379) from build 12806354. Bug: http://b/379133546 Test: N/A Change-Id: I2eb8938af55d809de674be63cb30cf27e801862b Upstream-Commit: ad834e67b1105d15ef907f6255d4c96e8e733f57
101 lines
3.2 KiB
C++
101 lines
3.2 KiB
C++
//===- CodeGenDataWriter.h --------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains support for writing codegen data.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CGDATA_CODEGENDATAWRITER_H
|
|
#define LLVM_CGDATA_CODEGENDATAWRITER_H
|
|
|
|
#include "llvm/CGData/CodeGenData.h"
|
|
#include "llvm/CGData/OutlinedHashTreeRecord.h"
|
|
#include "llvm/Support/EndianStream.h"
|
|
#include "llvm/Support/Error.h"
|
|
|
|
namespace llvm {
|
|
|
|
/// A struct to define how the data stream should be patched.
|
|
struct CGDataPatchItem {
|
|
uint64_t Pos; // Where to patch.
|
|
uint64_t *D; // Pointer to an array of source data.
|
|
int N; // Number of elements in \c D array.
|
|
};
|
|
|
|
/// A wrapper class to abstract writer stream with support of bytes
|
|
/// back patching.
|
|
class CGDataOStream {
|
|
public:
|
|
CGDataOStream(raw_fd_ostream &FD)
|
|
: IsFDOStream(true), OS(FD), LE(FD, llvm::endianness::little) {}
|
|
CGDataOStream(raw_string_ostream &STR)
|
|
: IsFDOStream(false), OS(STR), LE(STR, llvm::endianness::little) {}
|
|
|
|
uint64_t tell() { return OS.tell(); }
|
|
void write(uint64_t V) { LE.write<uint64_t>(V); }
|
|
void write32(uint32_t V) { LE.write<uint32_t>(V); }
|
|
void write8(uint8_t V) { LE.write<uint8_t>(V); }
|
|
|
|
// \c patch can only be called when all data is written and flushed.
|
|
// For raw_string_ostream, the patch is done on the target string
|
|
// directly and it won't be reflected in the stream's internal buffer.
|
|
void patch(ArrayRef<CGDataPatchItem> P);
|
|
|
|
// If \c OS is an instance of \c raw_fd_ostream, this field will be
|
|
// true. Otherwise, \c OS will be an raw_string_ostream.
|
|
bool IsFDOStream;
|
|
raw_ostream &OS;
|
|
support::endian::Writer LE;
|
|
};
|
|
|
|
class CodeGenDataWriter {
|
|
/// The outlined hash tree to be written.
|
|
OutlinedHashTreeRecord HashTreeRecord;
|
|
|
|
/// A bit mask describing the kind of the codegen data.
|
|
CGDataKind DataKind = CGDataKind::Unknown;
|
|
|
|
public:
|
|
CodeGenDataWriter() = default;
|
|
~CodeGenDataWriter() = default;
|
|
|
|
/// Add the outlined hash tree record. The input Record is released.
|
|
void addRecord(OutlinedHashTreeRecord &Record);
|
|
|
|
/// Write the codegen data to \c OS
|
|
Error write(raw_fd_ostream &OS);
|
|
|
|
/// Write the codegen data in text format to \c OS
|
|
Error writeText(raw_fd_ostream &OS);
|
|
|
|
/// Return the attributes of the current CGData.
|
|
CGDataKind getCGDataKind() const { return DataKind; }
|
|
|
|
/// Return true if the header indicates the data has an outlined hash tree.
|
|
bool hasOutlinedHashTree() const {
|
|
return static_cast<uint32_t>(DataKind) &
|
|
static_cast<uint32_t>(CGDataKind::FunctionOutlinedHashTree);
|
|
}
|
|
|
|
private:
|
|
/// The offset of the outlined hash tree in the file.
|
|
uint64_t OutlinedHashTreeOffset;
|
|
|
|
/// Write the codegen data header to \c COS
|
|
Error writeHeader(CGDataOStream &COS);
|
|
|
|
/// Write the codegen data header in text to \c OS
|
|
Error writeHeaderText(raw_fd_ostream &OS);
|
|
|
|
Error writeImpl(CGDataOStream &COS);
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_CGDATA_CODEGENDATAWRITER_H
|