1 //===- DIADataStream.cpp - DIA implementation of IPDBDataStream -*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/DebugInfo/PDB/DIA/DIADataStream.h"
11 #include "llvm/ADT/ArrayRef.h"
12 #include "llvm/Support/ConvertUTF.h"
13 
14 using namespace llvm;
15 
16 DIADataStream::DIADataStream(CComPtr<IDiaEnumDebugStreamData> DiaStreamData)
17     : StreamData(DiaStreamData) {}
18 
19 uint32_t DIADataStream::getRecordCount() const {
20   LONG Count = 0;
21   return (S_OK == StreamData->get_Count(&Count)) ? Count : 0;
22 }
23 
24 std::string DIADataStream::getName() const {
25   CComBSTR Name16;
26   if (S_OK != StreamData->get_name(&Name16))
27     return std::string();
28 
29   std::string Name8;
30   llvm::ArrayRef<char> Name16Bytes(reinterpret_cast<char *>(Name16.m_str),
31                                    Name16.ByteLength());
32   if (!llvm::convertUTF16ToUTF8String(Name16Bytes, Name8))
33     return std::string();
34   return Name8;
35 }
36 
37 llvm::Optional<DIADataStream::RecordType>
38 DIADataStream::getItemAtIndex(uint32_t Index) const {
39   RecordType Record;
40   DWORD RecordSize = 0;
41   StreamData->Item(Index, 0, &RecordSize, nullptr);
42   if (RecordSize == 0)
43     return llvm::Optional<RecordType>();
44 
45   Record.resize(RecordSize);
46   if (S_OK != StreamData->Item(Index, RecordSize, &RecordSize, &Record[0]))
47     return llvm::Optional<RecordType>();
48   return Record;
49 }
50 
51 bool DIADataStream::getNext(RecordType &Record) {
52   Record.clear();
53   DWORD RecordSize = 0;
54   ULONG CountFetched = 0;
55   StreamData->Next(1, 0, &RecordSize, nullptr, &CountFetched);
56   if (RecordSize == 0)
57     return false;
58 
59   Record.resize(RecordSize);
60   if (S_OK ==
61       StreamData->Next(1, RecordSize, &RecordSize, &Record[0], &CountFetched))
62     return false;
63   return true;
64 }
65 
66 void DIADataStream::reset() { StreamData->Reset(); }
67 
68 DIADataStream *DIADataStream::clone() const {
69   CComPtr<IDiaEnumDebugStreamData> EnumeratorClone;
70   if (S_OK != StreamData->Clone(&EnumeratorClone))
71     return nullptr;
72 
73   return new DIADataStream(EnumeratorClone);
74 }
75