mirror of
				https://github.com/c64scene-ar/llvm-6502.git
				synced 2025-10-31 08:16:47 +00:00 
			
		
		
		
	script. This is in preparation for changes to lots of include lines. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@229088 91177308-0d34-0410-b5e6-96231b3b80d8
		
			
				
	
	
		
			74 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			74 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| //===- DIADataStream.cpp - DIA implementation of IPDBDataStream -*- C++ -*-===//
 | |
| //
 | |
| //                     The LLVM Compiler Infrastructure
 | |
| //
 | |
| // This file is distributed under the University of Illinois Open Source
 | |
| // License. See LICENSE.TXT for details.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| #include "llvm/DebugInfo/PDB/DIA/DIADataStream.h"
 | |
| #include "llvm/Support/ConvertUTF.h"
 | |
| 
 | |
| using namespace llvm;
 | |
| 
 | |
| DIADataStream::DIADataStream(CComPtr<IDiaEnumDebugStreamData> DiaStreamData)
 | |
|     : StreamData(DiaStreamData) {}
 | |
| 
 | |
| uint32_t DIADataStream::getRecordCount() const {
 | |
|   LONG Count = 0;
 | |
|   return (S_OK == StreamData->get_Count(&Count)) ? Count : 0;
 | |
| }
 | |
| 
 | |
| std::string DIADataStream::getName() const {
 | |
|   CComBSTR Name16;
 | |
|   if (S_OK != StreamData->get_name(&Name16))
 | |
|     return std::string();
 | |
| 
 | |
|   std::string Name8;
 | |
|   llvm::ArrayRef<char> Name16Bytes(reinterpret_cast<char *>(Name16.m_str),
 | |
|                                    Name16.ByteLength());
 | |
|   if (!llvm::convertUTF16ToUTF8String(Name16Bytes, Name8))
 | |
|     return std::string();
 | |
|   return Name8;
 | |
| }
 | |
| 
 | |
| llvm::Optional<DIADataStream::RecordType>
 | |
| DIADataStream::getItemAtIndex(uint32_t Index) const {
 | |
|   RecordType Record;
 | |
|   DWORD RecordSize = 0;
 | |
|   StreamData->Item(Index, 0, &RecordSize, nullptr);
 | |
|   if (RecordSize == 0)
 | |
|     return llvm::Optional<RecordType>();
 | |
| 
 | |
|   Record.resize(RecordSize);
 | |
|   if (S_OK != StreamData->Item(Index, RecordSize, &RecordSize, &Record[0]))
 | |
|     return llvm::Optional<RecordType>();
 | |
|   return Record;
 | |
| }
 | |
| 
 | |
| bool DIADataStream::getNext(RecordType &Record) {
 | |
|   Record.clear();
 | |
|   DWORD RecordSize = 0;
 | |
|   ULONG CountFetched = 0;
 | |
|   StreamData->Next(1, 0, &RecordSize, nullptr, &CountFetched);
 | |
|   if (RecordSize == 0)
 | |
|     return false;
 | |
| 
 | |
|   Record.resize(RecordSize);
 | |
|   if (S_OK ==
 | |
|       StreamData->Next(1, RecordSize, &RecordSize, &Record[0], &CountFetched))
 | |
|     return false;
 | |
|   return true;
 | |
| }
 | |
| 
 | |
| void DIADataStream::reset() { StreamData->Reset(); }
 | |
| 
 | |
| DIADataStream *DIADataStream::clone() const {
 | |
|   CComPtr<IDiaEnumDebugStreamData> EnumeratorClone;
 | |
|   if (S_OK != StreamData->Clone(&EnumeratorClone))
 | |
|     return nullptr;
 | |
| 
 | |
|   return new DIADataStream(EnumeratorClone);
 | |
| }
 |