//===- ModuleSubstreamVisitor.h ---------------------------------*- C++ -*-===// // // The LLVM Compiler Infrastructure // // This file is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// #ifndef LLVM_DEBUGINFO_CODEVIEW_MODULESUBSTREAMVISITOR_H #define LLVM_DEBUGINFO_CODEVIEW_MODULESUBSTREAMVISITOR_H #include "llvm/DebugInfo/CodeView/CodeView.h" #include "llvm/DebugInfo/CodeView/CodeViewError.h" #include "llvm/DebugInfo/CodeView/Line.h" #include "llvm/DebugInfo/CodeView/ModuleSubstream.h" #include "llvm/DebugInfo/CodeView/StreamReader.h" #include "llvm/DebugInfo/CodeView/StreamRef.h" namespace llvm { namespace codeview { struct LineColumnEntry { support::ulittle32_t NameIndex; FixedStreamArray LineNumbers; FixedStreamArray Columns; }; template <> class VarStreamArrayExtractor { public: VarStreamArrayExtractor(const LineSubstreamHeader *Header) : Header(Header) {} Error operator()(StreamRef Stream, uint32_t &Len, LineColumnEntry &Item) const { const LineFileBlockHeader *BlockHeader; StreamReader Reader(Stream); if (auto EC = Reader.readObject(BlockHeader)) return EC; bool HasColumn = Header->Flags & LineFlags::HaveColumns; uint32_t LineInfoSize = BlockHeader->NumLines * (sizeof(LineNumberEntry) + (HasColumn ? sizeof(ColumnNumberEntry) : 0)); if (BlockHeader->BlockSize < sizeof(LineFileBlockHeader)) return make_error(cv_error_code::corrupt_record, "Invalid line block record size"); uint32_t Size = BlockHeader->BlockSize - sizeof(LineFileBlockHeader); if (LineInfoSize > Size) return make_error(cv_error_code::corrupt_record, "Invalid line block record size"); // The value recorded in BlockHeader->BlockSize includes the size of // LineFileBlockHeader. Len = BlockHeader->BlockSize; Item.NameIndex = BlockHeader->NameIndex; if (auto EC = Reader.readArray(Item.LineNumbers, BlockHeader->NumLines)) return EC; if (HasColumn) { if (auto EC = Reader.readArray(Item.Columns, BlockHeader->NumLines)) return EC; } return Error::success(); } private: const LineSubstreamHeader *Header; }; struct FileChecksumEntry { uint32_t FileNameOffset; // Byte offset of filename in global stringtable. FileChecksumKind Kind; // The type of checksum. ArrayRef Checksum; // The bytes of the checksum. }; template <> class VarStreamArrayExtractor { public: Error operator()(StreamRef Stream, uint32_t &Len, FileChecksumEntry &Item) const { const FileChecksum *Header; StreamReader Reader(Stream); if (auto EC = Reader.readObject(Header)) return EC; Item.FileNameOffset = Header->FileNameOffset; Item.Kind = static_cast(Header->ChecksumKind); if (auto EC = Reader.readBytes(Item.Checksum, Header->ChecksumSize)) return EC; Len = sizeof(FileChecksum) + Header->ChecksumSize; return Error::success(); } }; typedef VarStreamArray LineInfoArray; typedef VarStreamArray FileChecksumArray; class IModuleSubstreamVisitor { public: virtual ~IModuleSubstreamVisitor() {} virtual Error visitUnknown(ModuleSubstreamKind Kind, StreamRef Data) = 0; virtual Error visitSymbols(StreamRef Data); virtual Error visitLines(StreamRef Data, const LineSubstreamHeader *Header, const LineInfoArray &Lines); virtual Error visitStringTable(StreamRef Data); virtual Error visitFileChecksums(StreamRef Data, const FileChecksumArray &Checksums); virtual Error visitFrameData(StreamRef Data); virtual Error visitInlineeLines(StreamRef Data); virtual Error visitCrossScopeImports(StreamRef Data); virtual Error visitCrossScopeExports(StreamRef Data); virtual Error visitILLines(StreamRef Data); virtual Error visitFuncMDTokenMap(StreamRef Data); virtual Error visitTypeMDTokenMap(StreamRef Data); virtual Error visitMergedAssemblyInput(StreamRef Data); virtual Error visitCoffSymbolRVA(StreamRef Data); }; Error visitModuleSubstream(const ModuleSubstream &R, IModuleSubstreamVisitor &V); } // namespace codeview } // namespace llvm #endif // LLVM_DEBUGINFO_CODEVIEW_MODULESUBSTREAMVISITOR_H