forked from OSchip/llvm-project
304 lines
11 KiB
C++
304 lines
11 KiB
C++
//===- DIASession.cpp - DIA implementation of IPDBSession -------*- 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/DIASession.h"
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIAEnumDebugStreams.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIAEnumLineNumbers.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIAEnumSourceFiles.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIAError.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIARawSymbol.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIASourceFile.h"
|
|
#include "llvm/DebugInfo/PDB/DIA/DIASupport.h"
|
|
#include "llvm/DebugInfo/PDB/GenericError.h"
|
|
#include "llvm/DebugInfo/PDB/PDB.h"
|
|
#include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
|
|
#include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
|
|
#include "llvm/Support/ConvertUTF.h"
|
|
#include "llvm/Support/Format.h"
|
|
#include "llvm/Support/FormatVariadic.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::pdb;
|
|
|
|
template <typename... Ts>
|
|
static Error ErrorFromHResult(HRESULT Result, const char *Str, Ts &&... Args) {
|
|
SmallString<64> MessageStorage;
|
|
StringRef Context;
|
|
if (sizeof...(Args) > 0) {
|
|
MessageStorage = formatv(Str, std::forward<Ts>(Args)...).str();
|
|
Context = MessageStorage;
|
|
} else
|
|
Context = Str;
|
|
|
|
switch (Result) {
|
|
case E_PDB_NOT_FOUND:
|
|
return make_error<GenericError>(generic_error_code::invalid_path, Context);
|
|
case E_PDB_FORMAT:
|
|
return make_error<DIAError>(dia_error_code::invalid_file_format, Context);
|
|
case E_INVALIDARG:
|
|
return make_error<DIAError>(dia_error_code::invalid_parameter, Context);
|
|
case E_UNEXPECTED:
|
|
return make_error<DIAError>(dia_error_code::already_loaded, Context);
|
|
case E_PDB_INVALID_SIG:
|
|
case E_PDB_INVALID_AGE:
|
|
return make_error<DIAError>(dia_error_code::debug_info_mismatch, Context);
|
|
default: {
|
|
std::string S;
|
|
raw_string_ostream OS(S);
|
|
OS << "HRESULT: " << format_hex(static_cast<DWORD>(Result), 10, true)
|
|
<< ": " << Context;
|
|
return make_error<DIAError>(dia_error_code::unspecified, OS.str());
|
|
}
|
|
}
|
|
}
|
|
|
|
static Error LoadDIA(CComPtr<IDiaDataSource> &DiaDataSource) {
|
|
if (SUCCEEDED(CoCreateInstance(CLSID_DiaSource, nullptr, CLSCTX_INPROC_SERVER,
|
|
IID_IDiaDataSource,
|
|
reinterpret_cast<LPVOID *>(&DiaDataSource))))
|
|
return Error::success();
|
|
|
|
// If the CoCreateInstance call above failed, msdia*.dll is not registered.
|
|
// Try loading the DLL corresponding to the #included DIA SDK.
|
|
#if !defined(_MSC_VER)
|
|
return llvm::make_error<GenericError>(
|
|
"DIA is only supported when using MSVC.");
|
|
#else
|
|
const wchar_t *msdia_dll = nullptr;
|
|
#if _MSC_VER >= 1900 && _MSC_VER < 2000
|
|
msdia_dll = L"msdia140.dll"; // VS2015
|
|
#elif _MSC_VER >= 1800
|
|
msdia_dll = L"msdia120.dll"; // VS2013
|
|
#else
|
|
#error "Unknown Visual Studio version."
|
|
#endif
|
|
|
|
HRESULT HR;
|
|
if (FAILED(HR = NoRegCoCreate(msdia_dll, CLSID_DiaSource, IID_IDiaDataSource,
|
|
reinterpret_cast<LPVOID *>(&DiaDataSource))))
|
|
return ErrorFromHResult(HR, "Calling NoRegCoCreate");
|
|
return Error::success();
|
|
#endif
|
|
}
|
|
|
|
DIASession::DIASession(CComPtr<IDiaSession> DiaSession) : Session(DiaSession) {}
|
|
|
|
Error DIASession::createFromPdb(StringRef Path,
|
|
std::unique_ptr<IPDBSession> &Session) {
|
|
CComPtr<IDiaDataSource> DiaDataSource;
|
|
CComPtr<IDiaSession> DiaSession;
|
|
|
|
// We assume that CoInitializeEx has already been called by the executable.
|
|
if (auto E = LoadDIA(DiaDataSource))
|
|
return E;
|
|
|
|
llvm::SmallVector<UTF16, 128> Path16;
|
|
if (!llvm::convertUTF8ToUTF16String(Path, Path16))
|
|
return make_error<GenericError>(generic_error_code::invalid_path);
|
|
|
|
const wchar_t *Path16Str = reinterpret_cast<const wchar_t*>(Path16.data());
|
|
HRESULT HR;
|
|
if (FAILED(HR = DiaDataSource->loadDataFromPdb(Path16Str))) {
|
|
return ErrorFromHResult(HR, "Calling loadDataFromPdb {0}", Path);
|
|
}
|
|
|
|
if (FAILED(HR = DiaDataSource->openSession(&DiaSession)))
|
|
return ErrorFromHResult(HR, "Calling openSession");
|
|
|
|
Session.reset(new DIASession(DiaSession));
|
|
return Error::success();
|
|
}
|
|
|
|
Error DIASession::createFromExe(StringRef Path,
|
|
std::unique_ptr<IPDBSession> &Session) {
|
|
CComPtr<IDiaDataSource> DiaDataSource;
|
|
CComPtr<IDiaSession> DiaSession;
|
|
|
|
// We assume that CoInitializeEx has already been called by the executable.
|
|
if (auto EC = LoadDIA(DiaDataSource))
|
|
return EC;
|
|
|
|
llvm::SmallVector<UTF16, 128> Path16;
|
|
if (!llvm::convertUTF8ToUTF16String(Path, Path16))
|
|
return make_error<GenericError>(generic_error_code::invalid_path, Path);
|
|
|
|
const wchar_t *Path16Str = reinterpret_cast<const wchar_t *>(Path16.data());
|
|
HRESULT HR;
|
|
if (FAILED(HR = DiaDataSource->loadDataForExe(Path16Str, nullptr, nullptr)))
|
|
return ErrorFromHResult(HR, "Calling loadDataForExe");
|
|
|
|
if (FAILED(HR = DiaDataSource->openSession(&DiaSession)))
|
|
return ErrorFromHResult(HR, "Calling openSession");
|
|
|
|
Session.reset(new DIASession(DiaSession));
|
|
return Error::success();
|
|
}
|
|
|
|
uint64_t DIASession::getLoadAddress() const {
|
|
uint64_t LoadAddress;
|
|
bool success = (S_OK == Session->get_loadAddress(&LoadAddress));
|
|
return (success) ? LoadAddress : 0;
|
|
}
|
|
|
|
void DIASession::setLoadAddress(uint64_t Address) {
|
|
Session->put_loadAddress(Address);
|
|
}
|
|
|
|
std::unique_ptr<PDBSymbolExe> DIASession::getGlobalScope() const {
|
|
CComPtr<IDiaSymbol> GlobalScope;
|
|
if (S_OK != Session->get_globalScope(&GlobalScope))
|
|
return nullptr;
|
|
|
|
auto RawSymbol = llvm::make_unique<DIARawSymbol>(*this, GlobalScope);
|
|
auto PdbSymbol(PDBSymbol::create(*this, std::move(RawSymbol)));
|
|
std::unique_ptr<PDBSymbolExe> ExeSymbol(
|
|
static_cast<PDBSymbolExe *>(PdbSymbol.release()));
|
|
return ExeSymbol;
|
|
}
|
|
|
|
std::unique_ptr<PDBSymbol> DIASession::getSymbolById(uint32_t SymbolId) const {
|
|
CComPtr<IDiaSymbol> LocatedSymbol;
|
|
if (S_OK != Session->symbolById(SymbolId, &LocatedSymbol))
|
|
return nullptr;
|
|
|
|
auto RawSymbol = llvm::make_unique<DIARawSymbol>(*this, LocatedSymbol);
|
|
return PDBSymbol::create(*this, std::move(RawSymbol));
|
|
}
|
|
|
|
std::unique_ptr<PDBSymbol>
|
|
DIASession::findSymbolByAddress(uint64_t Address, PDB_SymType Type) const {
|
|
enum SymTagEnum EnumVal = static_cast<enum SymTagEnum>(Type);
|
|
|
|
CComPtr<IDiaSymbol> Symbol;
|
|
if (S_OK != Session->findSymbolByVA(Address, EnumVal, &Symbol)) {
|
|
ULONGLONG LoadAddr = 0;
|
|
if (S_OK != Session->get_loadAddress(&LoadAddr))
|
|
return nullptr;
|
|
DWORD RVA = static_cast<DWORD>(Address - LoadAddr);
|
|
if (S_OK != Session->findSymbolByRVA(RVA, EnumVal, &Symbol))
|
|
return nullptr;
|
|
}
|
|
auto RawSymbol = llvm::make_unique<DIARawSymbol>(*this, Symbol);
|
|
return PDBSymbol::create(*this, std::move(RawSymbol));
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumLineNumbers>
|
|
DIASession::findLineNumbers(const PDBSymbolCompiland &Compiland,
|
|
const IPDBSourceFile &File) const {
|
|
const DIARawSymbol &RawCompiland =
|
|
static_cast<const DIARawSymbol &>(Compiland.getRawSymbol());
|
|
const DIASourceFile &RawFile = static_cast<const DIASourceFile &>(File);
|
|
|
|
CComPtr<IDiaEnumLineNumbers> LineNumbers;
|
|
if (S_OK !=
|
|
Session->findLines(RawCompiland.getDiaSymbol(), RawFile.getDiaFile(),
|
|
&LineNumbers))
|
|
return nullptr;
|
|
|
|
return llvm::make_unique<DIAEnumLineNumbers>(LineNumbers);
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumLineNumbers>
|
|
DIASession::findLineNumbersByAddress(uint64_t Address, uint32_t Length) const {
|
|
CComPtr<IDiaEnumLineNumbers> LineNumbers;
|
|
if (S_OK != Session->findLinesByVA(Address, Length, &LineNumbers))
|
|
return nullptr;
|
|
|
|
return llvm::make_unique<DIAEnumLineNumbers>(LineNumbers);
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumSourceFiles>
|
|
DIASession::findSourceFiles(const PDBSymbolCompiland *Compiland,
|
|
llvm::StringRef Pattern,
|
|
PDB_NameSearchFlags Flags) const {
|
|
IDiaSymbol *DiaCompiland = nullptr;
|
|
CComBSTR Utf16Pattern;
|
|
if (!Pattern.empty())
|
|
Utf16Pattern = CComBSTR(Pattern.data());
|
|
|
|
if (Compiland)
|
|
DiaCompiland = static_cast<const DIARawSymbol &>(Compiland->getRawSymbol())
|
|
.getDiaSymbol();
|
|
|
|
Flags = static_cast<PDB_NameSearchFlags>(
|
|
Flags | PDB_NameSearchFlags::NS_FileNameExtMatch);
|
|
CComPtr<IDiaEnumSourceFiles> SourceFiles;
|
|
if (S_OK !=
|
|
Session->findFile(DiaCompiland, Utf16Pattern.m_str, Flags, &SourceFiles))
|
|
return nullptr;
|
|
return llvm::make_unique<DIAEnumSourceFiles>(*this, SourceFiles);
|
|
}
|
|
|
|
std::unique_ptr<IPDBSourceFile>
|
|
DIASession::findOneSourceFile(const PDBSymbolCompiland *Compiland,
|
|
llvm::StringRef Pattern,
|
|
PDB_NameSearchFlags Flags) const {
|
|
auto SourceFiles = findSourceFiles(Compiland, Pattern, Flags);
|
|
if (!SourceFiles || SourceFiles->getChildCount() == 0)
|
|
return nullptr;
|
|
return SourceFiles->getNext();
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumChildren<PDBSymbolCompiland>>
|
|
DIASession::findCompilandsForSourceFile(llvm::StringRef Pattern,
|
|
PDB_NameSearchFlags Flags) const {
|
|
auto File = findOneSourceFile(nullptr, Pattern, Flags);
|
|
if (!File)
|
|
return nullptr;
|
|
return File->getCompilands();
|
|
}
|
|
|
|
std::unique_ptr<PDBSymbolCompiland>
|
|
DIASession::findOneCompilandForSourceFile(llvm::StringRef Pattern,
|
|
PDB_NameSearchFlags Flags) const {
|
|
auto Compilands = findCompilandsForSourceFile(Pattern, Flags);
|
|
if (!Compilands || Compilands->getChildCount() == 0)
|
|
return nullptr;
|
|
return Compilands->getNext();
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumSourceFiles> DIASession::getAllSourceFiles() const {
|
|
CComPtr<IDiaEnumSourceFiles> Files;
|
|
if (S_OK != Session->findFile(nullptr, nullptr, nsNone, &Files))
|
|
return nullptr;
|
|
|
|
return llvm::make_unique<DIAEnumSourceFiles>(*this, Files);
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumSourceFiles> DIASession::getSourceFilesForCompiland(
|
|
const PDBSymbolCompiland &Compiland) const {
|
|
CComPtr<IDiaEnumSourceFiles> Files;
|
|
|
|
const DIARawSymbol &RawSymbol =
|
|
static_cast<const DIARawSymbol &>(Compiland.getRawSymbol());
|
|
if (S_OK !=
|
|
Session->findFile(RawSymbol.getDiaSymbol(), nullptr, nsNone, &Files))
|
|
return nullptr;
|
|
|
|
return llvm::make_unique<DIAEnumSourceFiles>(*this, Files);
|
|
}
|
|
|
|
std::unique_ptr<IPDBSourceFile>
|
|
DIASession::getSourceFileById(uint32_t FileId) const {
|
|
CComPtr<IDiaSourceFile> LocatedFile;
|
|
if (S_OK != Session->findFileById(FileId, &LocatedFile))
|
|
return nullptr;
|
|
|
|
return llvm::make_unique<DIASourceFile>(*this, LocatedFile);
|
|
}
|
|
|
|
std::unique_ptr<IPDBEnumDataStreams> DIASession::getDebugStreams() const {
|
|
CComPtr<IDiaEnumDebugStreams> DiaEnumerator;
|
|
if (S_OK != Session->getEnumDebugStreams(&DiaEnumerator))
|
|
return nullptr;
|
|
|
|
return llvm::make_unique<DIAEnumDebugStreams>(DiaEnumerator);
|
|
}
|