1*0b57cec5SDimitry Andric //==- DIAEnumSymbols.cpp - DIA Symbol Enumerator impl ------------*- C++ -*-==//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric
9*0b57cec5SDimitry Andric #include "llvm/DebugInfo/PDB/DIA/DIAEnumSymbols.h"
10*0b57cec5SDimitry Andric #include "llvm/DebugInfo/PDB/DIA/DIARawSymbol.h"
11*0b57cec5SDimitry Andric #include "llvm/DebugInfo/PDB/DIA/DIASession.h"
12*0b57cec5SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbol.h"
13*0b57cec5SDimitry Andric
14*0b57cec5SDimitry Andric using namespace llvm;
15*0b57cec5SDimitry Andric using namespace llvm::pdb;
16*0b57cec5SDimitry Andric
DIAEnumSymbols(const DIASession & PDBSession,CComPtr<IDiaEnumSymbols> DiaEnumerator)17*0b57cec5SDimitry Andric DIAEnumSymbols::DIAEnumSymbols(const DIASession &PDBSession,
18*0b57cec5SDimitry Andric CComPtr<IDiaEnumSymbols> DiaEnumerator)
19*0b57cec5SDimitry Andric : Session(PDBSession), Enumerator(DiaEnumerator) {}
20*0b57cec5SDimitry Andric
getChildCount() const21*0b57cec5SDimitry Andric uint32_t DIAEnumSymbols::getChildCount() const {
22*0b57cec5SDimitry Andric LONG Count = 0;
23*0b57cec5SDimitry Andric return (S_OK == Enumerator->get_Count(&Count)) ? Count : 0;
24*0b57cec5SDimitry Andric }
25*0b57cec5SDimitry Andric
26*0b57cec5SDimitry Andric std::unique_ptr<PDBSymbol>
getChildAtIndex(uint32_t Index) const27*0b57cec5SDimitry Andric DIAEnumSymbols::getChildAtIndex(uint32_t Index) const {
28*0b57cec5SDimitry Andric CComPtr<IDiaSymbol> Item;
29*0b57cec5SDimitry Andric if (S_OK != Enumerator->Item(Index, &Item))
30*0b57cec5SDimitry Andric return nullptr;
31*0b57cec5SDimitry Andric
32*0b57cec5SDimitry Andric std::unique_ptr<DIARawSymbol> RawSymbol(new DIARawSymbol(Session, Item));
33*0b57cec5SDimitry Andric return std::unique_ptr<PDBSymbol>(PDBSymbol::create(Session, std::move(RawSymbol)));
34*0b57cec5SDimitry Andric }
35*0b57cec5SDimitry Andric
getNext()36*0b57cec5SDimitry Andric std::unique_ptr<PDBSymbol> DIAEnumSymbols::getNext() {
37*0b57cec5SDimitry Andric CComPtr<IDiaSymbol> Item;
38*0b57cec5SDimitry Andric ULONG NumFetched = 0;
39*0b57cec5SDimitry Andric if (S_OK != Enumerator->Next(1, &Item, &NumFetched))
40*0b57cec5SDimitry Andric return nullptr;
41*0b57cec5SDimitry Andric
42*0b57cec5SDimitry Andric std::unique_ptr<DIARawSymbol> RawSymbol(new DIARawSymbol(Session, Item));
43*0b57cec5SDimitry Andric return std::unique_ptr<PDBSymbol>(
44*0b57cec5SDimitry Andric PDBSymbol::create(Session, std::move(RawSymbol)));
45*0b57cec5SDimitry Andric }
46*0b57cec5SDimitry Andric
reset()47*0b57cec5SDimitry Andric void DIAEnumSymbols::reset() { Enumerator->Reset(); }
48