Home | History | Annotate | Line # | Download | only in Native 
      1 //===- TpiStream.cpp - PDB Type Info (TPI) Stream 2 Access ------*- C++ -*-===//
      2 //
      3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
      4 // See https://llvm.org/LICENSE.txt for license information.
      5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
      6 //
      7 //===----------------------------------------------------------------------===//
      8 
      9 #ifndef LLVM_DEBUGINFO_PDB_NATIVE_TPISTREAM_H
     10 #define LLVM_DEBUGINFO_PDB_NATIVE_TPISTREAM_H
     11 
     12 #include "llvm/DebugInfo/CodeView/CVRecord.h"
     13 #include "llvm/DebugInfo/PDB/Native/HashTable.h"
     14 #include "llvm/DebugInfo/PDB/Native/RawConstants.h"
     15 #include "llvm/DebugInfo/PDB/Native/RawTypes.h"
     16 #include "llvm/DebugInfo/PDB/PDBTypes.h"
     17 #include "llvm/Support/BinaryStreamArray.h"
     18 #include "llvm/Support/BinaryStreamRef.h"
     19 #include "llvm/Support/raw_ostream.h"
     20 
     21 #include "llvm/Support/Error.h"
     22 
     23 namespace llvm {
     24 namespace codeview {
     25 class LazyRandomTypeCollection;
     26 }
     27 namespace msf {
     28 class MappedBlockStream;
     29 }
     30 namespace pdb {
     31 class PDBFile;
     32 
     33 class TpiStream {
     34   friend class TpiStreamBuilder;
     35 
     36 public:
     37   TpiStream(PDBFile &File, std::unique_ptr<msf::MappedBlockStream> Stream);
     38   ~TpiStream();
     39   Error reload();
     40 
     41   PdbRaw_TpiVer getTpiVersion() const;
     42 
     43   uint32_t TypeIndexBegin() const;
     44   uint32_t TypeIndexEnd() const;
     45   uint32_t getNumTypeRecords() const;
     46   uint16_t getTypeHashStreamIndex() const;
     47   uint16_t getTypeHashStreamAuxIndex() const;
     48 
     49   uint32_t getHashKeySize() const;
     50   uint32_t getNumHashBuckets() const;
     51   FixedStreamArray<support::ulittle32_t> getHashValues() const;
     52   FixedStreamArray<codeview::TypeIndexOffset> getTypeIndexOffsets() const;
     53   HashTable<support::ulittle32_t> &getHashAdjusters();
     54 
     55   codeview::CVTypeRange types(bool *HadError) const;
     56   const codeview::CVTypeArray &typeArray() const { return TypeRecords; }
     57 
     58   codeview::LazyRandomTypeCollection &typeCollection() { return *Types; }
     59 
     60   Expected<codeview::TypeIndex>
     61   findFullDeclForForwardRef(codeview::TypeIndex ForwardRefTI) const;
     62 
     63   std::vector<codeview::TypeIndex> findRecordsByName(StringRef Name) const;
     64 
     65   codeview::CVType getType(codeview::TypeIndex Index);
     66 
     67   BinarySubstreamRef getTypeRecordsSubstream() const;
     68 
     69   Error commit();
     70 
     71   void buildHashMap();
     72 
     73   bool supportsTypeLookup() const;
     74 
     75 private:
     76   PDBFile &Pdb;
     77   std::unique_ptr<msf::MappedBlockStream> Stream;
     78 
     79   std::unique_ptr<codeview::LazyRandomTypeCollection> Types;
     80 
     81   BinarySubstreamRef TypeRecordsSubstream;
     82 
     83   codeview::CVTypeArray TypeRecords;
     84 
     85   std::unique_ptr<BinaryStream> HashStream;
     86   FixedStreamArray<support::ulittle32_t> HashValues;
     87   FixedStreamArray<codeview::TypeIndexOffset> TypeIndexOffsets;
     88   HashTable<support::ulittle32_t> HashAdjusters;
     89 
     90   std::vector<std::vector<codeview::TypeIndex>> HashMap;
     91 
     92   const TpiStreamHeader *Header;
     93 };
     94 }
     95 }
     96 
     97 #endif
     98