1 //===-- HostInfoWindows.cpp -----------------------------------------------===//
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 "lldb/Host/windows/windows.h"
10 
11 #include <objbase.h>
12 
13 #include <mutex>
14 
15 #include "lldb/Host/windows/HostInfoWindows.h"
16 #include "lldb/Host/windows/PosixApi.h"
17 #include "lldb/Utility/UserIDResolver.h"
18 #include "llvm/ADT/SmallString.h"
19 #include "llvm/Support/ConvertUTF.h"
20 #include "llvm/Support/FileSystem.h"
21 #include "llvm/Support/ManagedStatic.h"
22 #include "llvm/Support/Path.h"
23 #include "llvm/Support/Threading.h"
24 #include "llvm/Support/raw_ostream.h"
25 
26 using namespace lldb_private;
27 
28 namespace {
29 class WindowsUserIDResolver : public UserIDResolver {
30 protected:
31   llvm::Optional<std::string> DoGetUserName(id_t uid) override {
32     return llvm::None;
33   }
34   llvm::Optional<std::string> DoGetGroupName(id_t gid) override {
35     return llvm::None;
36   }
37 };
38 } // namespace
39 
40 FileSpec HostInfoWindows::m_program_filespec;
41 
42 void HostInfoWindows::Initialize(SharedLibraryDirectoryHelper *helper) {
43   ::CoInitializeEx(nullptr, COINIT_MULTITHREADED);
44   HostInfoBase::Initialize(helper);
45 }
46 
47 void HostInfoWindows::Terminate() {
48   HostInfoBase::Terminate();
49   ::CoUninitialize();
50 }
51 
52 size_t HostInfoWindows::GetPageSize() {
53   SYSTEM_INFO systemInfo;
54   GetNativeSystemInfo(&systemInfo);
55   return systemInfo.dwPageSize;
56 }
57 
58 llvm::VersionTuple HostInfoWindows::GetOSVersion() {
59   OSVERSIONINFOEX info;
60 
61   ZeroMemory(&info, sizeof(OSVERSIONINFOEX));
62   info.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
63 #pragma warning(push)
64 #pragma warning(disable : 4996)
65   // Starting with Microsoft SDK for Windows 8.1, this function is deprecated
66   // in favor of the new Windows Version Helper APIs.  Since we don't specify a
67   // minimum SDK version, it's easier to simply disable the warning rather than
68   // try to support both APIs.
69   if (GetVersionEx((LPOSVERSIONINFO)&info) == 0)
70     return llvm::VersionTuple();
71 #pragma warning(pop)
72 
73   return llvm::VersionTuple(info.dwMajorVersion, info.dwMinorVersion,
74                             info.wServicePackMajor);
75 }
76 
77 llvm::Optional<std::string> HostInfoWindows::GetOSBuildString() {
78   llvm::VersionTuple version = GetOSVersion();
79   if (version.empty())
80     return llvm::None;
81 
82   return "Windows NT " + version.getAsString();
83 }
84 
85 bool HostInfoWindows::GetOSKernelDescription(std::string &s) {
86   llvm::Optional<std::string> build = GetOSBuildString();
87   s = build.getValueOr("");
88   return build.hasValue();
89 }
90 
91 bool HostInfoWindows::GetHostname(std::string &s) {
92   wchar_t buffer[MAX_COMPUTERNAME_LENGTH + 1];
93   DWORD dwSize = MAX_COMPUTERNAME_LENGTH + 1;
94   if (!::GetComputerNameW(buffer, &dwSize))
95     return false;
96 
97   // The conversion requires an empty string.
98   s.clear();
99   return llvm::convertWideToUTF8(buffer, s);
100 }
101 
102 FileSpec HostInfoWindows::GetProgramFileSpec() {
103   static llvm::once_flag g_once_flag;
104   llvm::call_once(g_once_flag, []() {
105     std::vector<wchar_t> buffer(PATH_MAX);
106     ::GetModuleFileNameW(NULL, buffer.data(), buffer.size());
107     std::string path;
108     llvm::convertWideToUTF8(buffer.data(), path);
109     m_program_filespec.SetFile(path, FileSpec::Style::native);
110   });
111   return m_program_filespec;
112 }
113 
114 FileSpec HostInfoWindows::GetDefaultShell() {
115   // Try to retrieve ComSpec from the environment. On the rare occasion
116   // that it fails, try a well-known path for ComSpec instead.
117 
118   std::string shell;
119   if (GetEnvironmentVar("ComSpec", shell))
120     return FileSpec(shell);
121 
122   return FileSpec("C:\\Windows\\system32\\cmd.exe");
123 }
124 
125 bool HostInfoWindows::GetEnvironmentVar(const std::string &var_name,
126                                         std::string &var) {
127   std::wstring wvar_name;
128   if (!llvm::ConvertUTF8toWide(var_name, wvar_name))
129     return false;
130 
131   if (const wchar_t *wvar = _wgetenv(wvar_name.c_str()))
132     return llvm::convertWideToUTF8(wvar, var);
133   return false;
134 }
135 
136 static llvm::ManagedStatic<WindowsUserIDResolver> g_user_id_resolver;
137 
138 UserIDResolver &HostInfoWindows::GetUserIDResolver() {
139   return *g_user_id_resolver;
140 }
141