llvm/lib/DebugInfo/PDB/DIA/DIADataStream.cpp

   1 //===- DIADataStream.cpp - DIA implementation of IPDBDataStream -*- 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 #include "llvm/DebugInfo/PDB/DIA/DIADataStream.h"
  10 #include "llvm/DebugInfo/PDB/DIA/DIAUtils.h"
  11
  12 using namespace llvm;
  13 using namespace llvm::pdb;
  14
  15 DIADataStream::DIADataStream(CComPtr<IDiaEnumDebugStreamData> DiaStreamData)
  16     : StreamData(DiaStreamData) {}
  17
  18 uint32_t DIADataStream::getRecordCount() const {
  19   LONG Count = 0;
  20   return (S_OK == StreamData->get_Count(&Count)) ? Count : 0;
  21 }
  22
  23 std::string DIADataStream::getName() const {
  24   return invokeBstrMethod(*StreamData, &IDiaEnumDebugStreamData::get_name);
  25 }
  26
  27 std::optional<DIADataStream::RecordType>
  28 DIADataStream::getItemAtIndex(uint32_t Index) const {
  29   RecordType Record;
  30   DWORD RecordSize = 0;
  31   StreamData->Item(Index, 0, &RecordSize, nullptr);
  32   if (RecordSize == 0)
  33     return std::nullopt;
  34
  35   Record.resize(RecordSize);
  36   if (S_OK != StreamData->Item(Index, RecordSize, &RecordSize, &Record[0]))
  37     return std::nullopt;
  38   return Record;
  39 }
  40
  41 bool DIADataStream::getNext(RecordType &Record) {
  42   Record.clear();
  43   DWORD RecordSize = 0;
  44   ULONG CountFetched = 0;
  45   StreamData->Next(1, 0, &RecordSize, nullptr, &CountFetched);
  46   if (RecordSize == 0)
  47     return false;
  48
  49   Record.resize(RecordSize);
  50   if (S_OK ==
  51       StreamData->Next(1, RecordSize, &RecordSize, &Record[0], &CountFetched))
  52     return false;
  53   return true;
  54 }
  55
  56 void DIADataStream::reset() { StreamData->Reset(); }