1 //===- DIASourceFile.cpp - DIA implementation of IPDBSourceFile -*- 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/DIASourceFile.h" 10 #include "llvm/DebugInfo/PDB/ConcreteSymbolEnumerator.h" 11 #include "llvm/DebugInfo/PDB/DIA/DIAEnumSymbols.h" 12 #include "llvm/DebugInfo/PDB/DIA/DIASession.h" 13 #include "llvm/DebugInfo/PDB/DIA/DIAUtils.h" 14 #include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h" 15 16 using namespace llvm; 17 using namespace llvm::pdb; 18 19 DIASourceFile::DIASourceFile(const DIASession &PDBSession, 20 CComPtr<IDiaSourceFile> DiaSourceFile) 21 : Session(PDBSession), SourceFile(DiaSourceFile) {} 22 23 std::string DIASourceFile::getFileName() const { 24 return invokeBstrMethod(*SourceFile, &IDiaSourceFile::get_fileName); 25 } 26 27 uint32_t DIASourceFile::getUniqueId() const { 28 DWORD Id; 29 return (S_OK == SourceFile->get_uniqueId(&Id)) ? Id : 0; 30 } 31 32 std::string DIASourceFile::getChecksum() const { 33 DWORD ByteSize = 0; 34 HRESULT Result = SourceFile->get_checksum(0, &ByteSize, nullptr); 35 if (ByteSize == 0) 36 return std::string(); 37 std::vector<BYTE> ChecksumBytes(ByteSize); 38 Result = SourceFile->get_checksum(ByteSize, &ByteSize, &ChecksumBytes[0]); 39 if (S_OK != Result) 40 return std::string(); 41 return std::string(ChecksumBytes.begin(), ChecksumBytes.end()); 42 } 43 44 PDB_Checksum DIASourceFile::getChecksumType() const { 45 DWORD Type; 46 HRESULT Result = SourceFile->get_checksumType(&Type); 47 if (S_OK != Result) 48 return PDB_Checksum::None; 49 return static_cast<PDB_Checksum>(Type); 50 } 51 52 std::unique_ptr<IPDBEnumChildren<PDBSymbolCompiland>> 53 DIASourceFile::getCompilands() const { 54 CComPtr<IDiaEnumSymbols> DiaEnumerator; 55 HRESULT Result = SourceFile->get_compilands(&DiaEnumerator); 56 if (S_OK != Result) 57 return nullptr; 58 59 auto Enumerator = std::unique_ptr<IPDBEnumSymbols>( 60 new DIAEnumSymbols(Session, DiaEnumerator)); 61 return std::unique_ptr<IPDBEnumChildren<PDBSymbolCompiland>>( 62 new ConcreteSymbolEnumerator<PDBSymbolCompiland>(std::move(Enumerator))); 63 } 64