1 | //==- DIAEnumDebugStreams.cpp - DIA Debug Stream Enumerator impl -*- 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/DIAEnumDebugStreams.h" |
10 | #include "llvm/DebugInfo/PDB/DIA/DIADataStream.h" |
11 | #include "llvm/DebugInfo/PDB/PDBSymbol.h" |
12 | |
13 | using namespace llvm; |
14 | using namespace llvm::pdb; |
15 | |
16 | DIAEnumDebugStreams::DIAEnumDebugStreams( |
17 | CComPtr<IDiaEnumDebugStreams> DiaEnumerator) |
18 | : Enumerator(DiaEnumerator) {} |
19 | |
20 | uint32_t DIAEnumDebugStreams::getChildCount() const { |
21 | LONG Count = 0; |
22 | return (S_OK == Enumerator->get_Count(&Count)) ? Count : 0; |
23 | } |
24 | |
25 | std::unique_ptr<IPDBDataStream> |
26 | DIAEnumDebugStreams::getChildAtIndex(uint32_t Index) const { |
27 | CComPtr<IDiaEnumDebugStreamData> Item; |
28 | VARIANT VarIndex; |
29 | VarIndex.vt = VT_I4; |
30 | VarIndex.lVal = Index; |
31 | if (S_OK != Enumerator->Item(VarIndex, &Item)) |
32 | return nullptr; |
33 | |
34 | return std::unique_ptr<IPDBDataStream>(new DIADataStream(Item)); |
35 | } |
36 | |
37 | std::unique_ptr<IPDBDataStream> DIAEnumDebugStreams::getNext() { |
38 | CComPtr<IDiaEnumDebugStreamData> Item; |
39 | ULONG NumFetched = 0; |
40 | if (S_OK != Enumerator->Next(1, &Item, &NumFetched)) |
41 | return nullptr; |
42 | |
43 | return std::unique_ptr<IPDBDataStream>(new DIADataStream(Item)); |
44 | } |
45 | |
46 | void DIAEnumDebugStreams::reset() { Enumerator->Reset(); } |
47 |