Untitled diff

Erstellt Diff läuft nie ab
1 Entfernung
290 Zeilen
2 Hinzufügungen
290 Zeilen
//===- DIASession.cpp - DIA implementation of IPDBSession -------*- C++ -*-===//
//===- DIASession.cpp - DIA implementation of IPDBSession -------*- C++ -*-===//
//
//
// The LLVM Compiler Infrastructure
// The LLVM Compiler Infrastructure
//
//
// This file is distributed under the University of Illinois Open Source
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
// License. See LICENSE.TXT for details.
//
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
#include "llvm/DebugInfo/PDB/DIA/DIASession.h"
#include "llvm/DebugInfo/PDB/DIA/DIASession.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/DebugInfo/PDB/DIA/DIAEnumDebugStreams.h"
#include "llvm/DebugInfo/PDB/DIA/DIAEnumDebugStreams.h"
#include "llvm/DebugInfo/PDB/DIA/DIAEnumLineNumbers.h"
#include "llvm/DebugInfo/PDB/DIA/DIAEnumLineNumbers.h"
#include "llvm/DebugInfo/PDB/DIA/DIAEnumSourceFiles.h"
#include "llvm/DebugInfo/PDB/DIA/DIAEnumSourceFiles.h"
#include "llvm/DebugInfo/PDB/DIA/DIAError.h"
#include "llvm/DebugInfo/PDB/DIA/DIAError.h"
#include "llvm/DebugInfo/PDB/DIA/DIARawSymbol.h"
#include "llvm/DebugInfo/PDB/DIA/DIARawSymbol.h"
#include "llvm/DebugInfo/PDB/DIA/DIASourceFile.h"
#include "llvm/DebugInfo/PDB/DIA/DIASourceFile.h"
#include "llvm/DebugInfo/PDB/DIA/DIASupport.h"
#include "llvm/DebugInfo/PDB/DIA/DIASupport.h"
#include "llvm/DebugInfo/PDB/GenericError.h"
#include "llvm/DebugInfo/PDB/GenericError.h"
#include "llvm/DebugInfo/PDB/PDB.h"
#include "llvm/DebugInfo/PDB/PDB.h"
#include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
#include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
#include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
#include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
#include "llvm/Support/ConvertUTF.h"
#include "llvm/Support/ConvertUTF.h"


using namespace llvm;
using namespace llvm;
using namespace llvm::pdb;
using namespace llvm::pdb;


namespace {
namespace {


Error ErrorFromHResult(HRESULT Result) {
Error ErrorFromHResult(HRESULT Result) {
switch (Result) {
switch (Result) {
case E_PDB_NOT_FOUND:
case E_PDB_NOT_FOUND:
return make_error<GenericError>(generic_error_code::invalid_path);
return make_error<GenericError>(generic_error_code::invalid_path);
case E_PDB_FORMAT:
case E_PDB_FORMAT:
return make_error<DIAError>(dia_error_code::invalid_file_format);
return make_error<DIAError>(dia_error_code::invalid_file_format);
case E_INVALIDARG:
case E_INVALIDARG:
return make_error<DIAError>(dia_error_code::invalid_parameter);
return make_error<DIAError>(dia_error_code::invalid_parameter);
case E_UNEXPECTED:
case E_UNEXPECTED:
return make_error<DIAError>(dia_error_code::already_loaded);
return make_error<DIAError>(dia_error_code::already_loaded);
case E_PDB_INVALID_SIG:
case E_PDB_INVALID_SIG:
case E_PDB_INVALID_AGE:
case E_PDB_INVALID_AGE:
return make_error<DIAError>(dia_error_code::debug_info_mismatch);
return make_error<DIAError>(dia_error_code::debug_info_mismatch);
default:
default:
return make_error<DIAError>(dia_error_code::unspecified);
return make_error<DIAError>(dia_error_code::unspecified);
}
}
}
}


Error LoadDIA(CComPtr<IDiaDataSource> &DiaDataSource) {
Error LoadDIA(CComPtr<IDiaDataSource> &DiaDataSource) {
if (SUCCEEDED(CoCreateInstance(CLSID_DiaSource, nullptr, CLSCTX_INPROC_SERVER,
if (SUCCEEDED(CoCreateInstance(CLSID_DiaSource, nullptr, CLSCTX_INPROC_SERVER,
IID_IDiaDataSource,
IID_IDiaDataSource,
reinterpret_cast<LPVOID *>(&DiaDataSource))))
reinterpret_cast<LPVOID *>(&DiaDataSource))))
return Error::success();
return Error::success();


// If the CoCreateInstance call above failed, msdia*.dll is not registered.
// If the CoCreateInstance call above failed, msdia*.dll is not registered.
// Try loading the DLL corresponding to the #included DIA SDK.
// Try loading the DLL corresponding to the #included DIA SDK.
#if !defined(_MSC_VER)
#if !defined(_MSC_VER)
return llvm::make_error<GenericError>(
return llvm::make_error<GenericError>(
"DIA is only supported when using MSVC.");
"DIA is only supported when using MSVC.");
#endif
#endif


const wchar_t *msdia_dll = nullptr;
const wchar_t *msdia_dll = nullptr;
#if _MSC_VER == 1900
#if _MSC_VER >= 1900 && _MSC_VER < 2000
msdia_dll = L"msdia140.dll"; // VS2015
msdia_dll = L"msdia140.dll"; // VS2015
#elif _MSC_VER == 1800
#elif _MSC_VER == 1800
msdia_dll = L"msdia120.dll"; // VS2013
msdia_dll = L"msdia120.dll"; // VS2013
#else
#else
#error "Unknown Visual Studio version."
#error "Unknown Visual Studio version."
#endif
#endif


HRESULT HR;
HRESULT HR;
if (FAILED(HR = NoRegCoCreate(msdia_dll, CLSID_DiaSource, IID_IDiaDataSource,
if (FAILED(HR = NoRegCoCreate(msdia_dll, CLSID_DiaSource, IID_IDiaDataSource,
reinterpret_cast<LPVOID *>(&DiaDataSource))))
reinterpret_cast<LPVOID *>(&DiaDataSource))))
return ErrorFromHResult(HR);
return ErrorFromHResult(HR);
return Error::success();
return Error::success();
}
}


}
}


DIASession::DIASession(CComPtr<IDiaSession> DiaSession) : Session(DiaSession) {}
DIASession::DIASession(CComPtr<IDiaSession> DiaSession) : Session(DiaSession) {}


Error DIASession::createFromPdb(StringRef Path,
Error DIASession::createFromPdb(StringRef Path,
std::unique_ptr<IPDBSession> &Session) {
std::unique_ptr<IPDBSession> &Session) {
CComPtr<IDiaDataSource> DiaDataSource;
CComPtr<IDiaDataSource> DiaDataSource;
CComPtr<IDiaSession> DiaSession;
CComPtr<IDiaSession> DiaSession;


// We assume that CoInitializeEx has already been called by the executable.
// We assume that CoInitializeEx has already been called by the executable.
if (auto E = LoadDIA(DiaDataSource))
if (auto E = LoadDIA(DiaDataSource))
return E;
return E;


llvm::SmallVector<UTF16, 128> Path16;
llvm::SmallVector<UTF16, 128> Path16;
if (!llvm::convertUTF8ToUTF16String(Path, Path16))
if (!llvm::convertUTF8ToUTF16String(Path, Path16))
return make_error<GenericError>(generic_error_code::invalid_path);
return make_error<GenericError>(generic_error_code::invalid_path);


const wchar_t *Path16Str = reinterpret_cast<const wchar_t*>(Path16.data());
const wchar_t *Path16Str = reinterpret_cast<const wchar_t*>(Path16.data());
HRESULT HR;
HRESULT HR;
if (FAILED(HR = DiaDataSource->loadDataFromPdb(Path16Str)))
if (FAILED(HR = DiaDataSource->loadDataFromPdb(Path16Str)))
return ErrorFromHResult(HR);
return ErrorFromHResult(HR);


if (FAILED(HR = DiaDataSource->openSession(&DiaSession)))
if (FAILED(HR = DiaDataSource->openSession(&DiaSession)))
return ErrorFromHResult(HR);
return ErrorFromHResult(HR);