190fadcdb2
Win32 API calls that are Unicode aware require wide character strings, but LLDB uses UTF8 everywhere. This patch does conversions wherever necessary when passing strings into and out of Win32 API calls. Patch by Cameron Differential Revision: http://reviews.llvm.org/D17107 Reviewed By: zturner, amccarth llvm-svn: 264074
151 lines
3.8 KiB
C++
151 lines
3.8 KiB
C++
//===-- HostInfoWindows.cpp -------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lldb/Host/windows/windows.h"
|
|
|
|
#include <objbase.h>
|
|
|
|
#include <mutex> // std::once
|
|
|
|
#include "lldb/Host/windows/HostInfoWindows.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/Support/ConvertUTF.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace lldb_private;
|
|
|
|
FileSpec HostInfoWindows::m_program_filespec;
|
|
|
|
void
|
|
HostInfoWindows::Initialize()
|
|
{
|
|
::CoInitializeEx(nullptr, COINIT_MULTITHREADED);
|
|
HostInfoBase::Initialize();
|
|
}
|
|
|
|
void
|
|
HostInfoWindows::Terminate()
|
|
{
|
|
HostInfoBase::Terminate();
|
|
::CoUninitialize();
|
|
}
|
|
|
|
size_t
|
|
HostInfoWindows::GetPageSize()
|
|
{
|
|
SYSTEM_INFO systemInfo;
|
|
GetNativeSystemInfo(&systemInfo);
|
|
return systemInfo.dwPageSize;
|
|
}
|
|
|
|
bool
|
|
HostInfoWindows::GetOSVersion(uint32_t &major, uint32_t &minor, uint32_t &update)
|
|
{
|
|
OSVERSIONINFOEX info;
|
|
|
|
ZeroMemory(&info, sizeof(OSVERSIONINFOEX));
|
|
info.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4996)
|
|
// Starting with Microsoft SDK for Windows 8.1, this function is deprecated in favor of the
|
|
// new Windows Version Helper APIs. Since we don't specify a minimum SDK version, it's easier
|
|
// to simply disable the warning rather than try to support both APIs.
|
|
if (GetVersionEx((LPOSVERSIONINFO)&info) == 0)
|
|
{
|
|
return false;
|
|
}
|
|
#pragma warning(pop)
|
|
|
|
major = info.dwMajorVersion;
|
|
minor = info.dwMinorVersion;
|
|
update = info.wServicePackMajor;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
HostInfoWindows::GetOSBuildString(std::string &s)
|
|
{
|
|
s.clear();
|
|
uint32_t major, minor, update;
|
|
if (!GetOSVersion(major, minor, update))
|
|
return false;
|
|
|
|
llvm::raw_string_ostream stream(s);
|
|
stream << "Windows NT " << major << "." << minor << "." << update;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
HostInfoWindows::GetOSKernelDescription(std::string &s)
|
|
{
|
|
return GetOSBuildString(s);
|
|
}
|
|
|
|
bool
|
|
HostInfoWindows::GetHostname(std::string &s)
|
|
{
|
|
wchar_t buffer[MAX_COMPUTERNAME_LENGTH + 1];
|
|
DWORD dwSize = MAX_COMPUTERNAME_LENGTH + 1;
|
|
if (!::GetComputerNameW(buffer, &dwSize))
|
|
return false;
|
|
|
|
return llvm::convertWideToUTF8(buffer, s);
|
|
}
|
|
|
|
FileSpec
|
|
HostInfoWindows::GetProgramFileSpec()
|
|
{
|
|
static std::once_flag g_once_flag;
|
|
std::call_once(g_once_flag, []() {
|
|
std::vector<wchar_t> buffer(PATH_MAX);
|
|
::GetModuleFileNameW(NULL, buffer.data(), buffer.size());
|
|
std::string path;
|
|
llvm::convertWideToUTF8(buffer.data(), path);
|
|
m_program_filespec.SetFile(path, false);
|
|
});
|
|
return m_program_filespec;
|
|
}
|
|
|
|
FileSpec
|
|
HostInfoWindows::GetDefaultShell()
|
|
{
|
|
std::string shell;
|
|
GetEnvironmentVar("ComSpec", shell);
|
|
return FileSpec(shell, false);
|
|
}
|
|
|
|
bool
|
|
HostInfoWindows::ComputePythonDirectory(FileSpec &file_spec)
|
|
{
|
|
FileSpec lldb_file_spec;
|
|
if (!GetLLDBPath(lldb::ePathTypeLLDBShlibDir, lldb_file_spec))
|
|
return false;
|
|
llvm::SmallString<64> path(lldb_file_spec.GetDirectory().AsCString());
|
|
llvm::sys::path::remove_filename(path);
|
|
llvm::sys::path::append(path, "lib", "site-packages");
|
|
std::replace(path.begin(), path.end(), '\\', '/');
|
|
file_spec.GetDirectory().SetString(path.c_str());
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
HostInfoWindows::GetEnvironmentVar(const std::string &var_name, std::string &var)
|
|
{
|
|
std::wstring wvar_name;
|
|
if (!llvm::ConvertUTF8toWide(var_name, wvar_name))
|
|
return false;
|
|
|
|
if (const wchar_t *wvar = _wgetenv(wvar_name.c_str()))
|
|
return llvm::convertWideToUTF8(wvar, var);
|
|
return false;
|
|
}
|