Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- WindowsSupport.h - Common Windows Include File -----------*- 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 | // This file defines things specific to Windows implementations. In addition to |
||
| 10 | // providing some helpers for working with win32 APIs, this header wraps |
||
| 11 | // <windows.h> with some portability macros. Always include WindowsSupport.h |
||
| 12 | // instead of including <windows.h> directly. |
||
| 13 | // |
||
| 14 | //===----------------------------------------------------------------------===// |
||
| 15 | |||
| 16 | //===----------------------------------------------------------------------===// |
||
| 17 | //=== WARNING: Implementation here must contain only generic Win32 code that |
||
| 18 | //=== is guaranteed to work on *all* Win32 variants. |
||
| 19 | //===----------------------------------------------------------------------===// |
||
| 20 | |||
| 21 | #ifndef LLVM_SUPPORT_WINDOWSSUPPORT_H |
||
| 22 | #define LLVM_SUPPORT_WINDOWSSUPPORT_H |
||
| 23 | |||
| 24 | // mingw-w64 tends to define it as 0x0502 in its headers. |
||
| 25 | #undef _WIN32_WINNT |
||
| 26 | #undef _WIN32_IE |
||
| 27 | |||
| 28 | // Require at least Windows 7 API. |
||
| 29 | #define _WIN32_WINNT 0x0601 |
||
| 30 | #define _WIN32_IE 0x0800 // MinGW at it again. FIXME: verify if still needed. |
||
| 31 | #define WIN32_LEAN_AND_MEAN |
||
| 32 | #ifndef NOMINMAX |
||
| 33 | #define NOMINMAX |
||
| 34 | #endif |
||
| 35 | |||
| 36 | #include "llvm/ADT/SmallVector.h" |
||
| 37 | #include "llvm/ADT/StringExtras.h" |
||
| 38 | #include "llvm/ADT/StringRef.h" |
||
| 39 | #include "llvm/ADT/Twine.h" |
||
| 40 | #include "llvm/Config/llvm-config.h" // Get build system configuration settings |
||
| 41 | #include "llvm/Support/Allocator.h" |
||
| 42 | #include "llvm/Support/Chrono.h" |
||
| 43 | #include "llvm/Support/Compiler.h" |
||
| 44 | #include "llvm/Support/ErrorHandling.h" |
||
| 45 | #include "llvm/Support/VersionTuple.h" |
||
| 46 | #include <cassert> |
||
| 47 | #include <string> |
||
| 48 | #include <system_error> |
||
| 49 | #include <windows.h> |
||
| 50 | |||
| 51 | // Must be included after windows.h |
||
| 52 | #include <wincrypt.h> |
||
| 53 | |||
| 54 | namespace llvm { |
||
| 55 | |||
| 56 | /// Determines if the program is running on Windows 8 or newer. This |
||
| 57 | /// reimplements one of the helpers in the Windows 8.1 SDK, which are intended |
||
| 58 | /// to supercede raw calls to GetVersionEx. Old SDKs, Cygwin, and MinGW don't |
||
| 59 | /// yet have VersionHelpers.h, so we have our own helper. |
||
| 60 | bool RunningWindows8OrGreater(); |
||
| 61 | |||
| 62 | /// Determines if the program is running on Windows 11 or Windows Server 2022. |
||
| 63 | bool RunningWindows11OrGreater(); |
||
| 64 | |||
| 65 | /// Returns the Windows version as Major.Minor.0.BuildNumber. Uses |
||
| 66 | /// RtlGetVersion or GetVersionEx under the hood depending on what is available. |
||
| 67 | /// GetVersionEx is deprecated, but this API exposes the build number which can |
||
| 68 | /// be useful for working around certain kernel bugs. |
||
| 69 | llvm::VersionTuple GetWindowsOSVersion(); |
||
| 70 | |||
| 71 | bool MakeErrMsg(std::string *ErrMsg, const std::string &prefix); |
||
| 72 | |||
| 73 | // Include GetLastError() in a fatal error message. |
||
| 74 | [[noreturn]] inline void ReportLastErrorFatal(const char *Msg) { |
||
| 75 | std::string ErrMsg; |
||
| 76 | MakeErrMsg(&ErrMsg, Msg); |
||
| 77 | llvm::report_fatal_error(Twine(ErrMsg)); |
||
| 78 | } |
||
| 79 | |||
| 80 | template <typename HandleTraits> |
||
| 81 | class ScopedHandle { |
||
| 82 | typedef typename HandleTraits::handle_type handle_type; |
||
| 83 | handle_type Handle; |
||
| 84 | |||
| 85 | ScopedHandle(const ScopedHandle &other) = delete; |
||
| 86 | void operator=(const ScopedHandle &other) = delete; |
||
| 87 | public: |
||
| 88 | ScopedHandle() |
||
| 89 | : Handle(HandleTraits::GetInvalid()) {} |
||
| 90 | |||
| 91 | explicit ScopedHandle(handle_type h) |
||
| 92 | : Handle(h) {} |
||
| 93 | |||
| 94 | ~ScopedHandle() { |
||
| 95 | if (HandleTraits::IsValid(Handle)) |
||
| 96 | HandleTraits::Close(Handle); |
||
| 97 | } |
||
| 98 | |||
| 99 | handle_type take() { |
||
| 100 | handle_type t = Handle; |
||
| 101 | Handle = HandleTraits::GetInvalid(); |
||
| 102 | return t; |
||
| 103 | } |
||
| 104 | |||
| 105 | ScopedHandle &operator=(handle_type h) { |
||
| 106 | if (HandleTraits::IsValid(Handle)) |
||
| 107 | HandleTraits::Close(Handle); |
||
| 108 | Handle = h; |
||
| 109 | return *this; |
||
| 110 | } |
||
| 111 | |||
| 112 | // True if Handle is valid. |
||
| 113 | explicit operator bool() const { |
||
| 114 | return HandleTraits::IsValid(Handle) ? true : false; |
||
| 115 | } |
||
| 116 | |||
| 117 | operator handle_type() const { |
||
| 118 | return Handle; |
||
| 119 | } |
||
| 120 | }; |
||
| 121 | |||
| 122 | struct CommonHandleTraits { |
||
| 123 | typedef HANDLE handle_type; |
||
| 124 | |||
| 125 | static handle_type GetInvalid() { |
||
| 126 | return INVALID_HANDLE_VALUE; |
||
| 127 | } |
||
| 128 | |||
| 129 | static void Close(handle_type h) { |
||
| 130 | ::CloseHandle(h); |
||
| 131 | } |
||
| 132 | |||
| 133 | static bool IsValid(handle_type h) { |
||
| 134 | return h != GetInvalid(); |
||
| 135 | } |
||
| 136 | }; |
||
| 137 | |||
| 138 | struct JobHandleTraits : CommonHandleTraits { |
||
| 139 | static handle_type GetInvalid() { |
||
| 140 | return NULL; |
||
| 141 | } |
||
| 142 | }; |
||
| 143 | |||
| 144 | struct CryptContextTraits : CommonHandleTraits { |
||
| 145 | typedef HCRYPTPROV handle_type; |
||
| 146 | |||
| 147 | static handle_type GetInvalid() { |
||
| 148 | return 0; |
||
| 149 | } |
||
| 150 | |||
| 151 | static void Close(handle_type h) { |
||
| 152 | ::CryptReleaseContext(h, 0); |
||
| 153 | } |
||
| 154 | |||
| 155 | static bool IsValid(handle_type h) { |
||
| 156 | return h != GetInvalid(); |
||
| 157 | } |
||
| 158 | }; |
||
| 159 | |||
| 160 | struct RegTraits : CommonHandleTraits { |
||
| 161 | typedef HKEY handle_type; |
||
| 162 | |||
| 163 | static handle_type GetInvalid() { |
||
| 164 | return NULL; |
||
| 165 | } |
||
| 166 | |||
| 167 | static void Close(handle_type h) { |
||
| 168 | ::RegCloseKey(h); |
||
| 169 | } |
||
| 170 | |||
| 171 | static bool IsValid(handle_type h) { |
||
| 172 | return h != GetInvalid(); |
||
| 173 | } |
||
| 174 | }; |
||
| 175 | |||
| 176 | struct FindHandleTraits : CommonHandleTraits { |
||
| 177 | static void Close(handle_type h) { |
||
| 178 | ::FindClose(h); |
||
| 179 | } |
||
| 180 | }; |
||
| 181 | |||
| 182 | struct FileHandleTraits : CommonHandleTraits {}; |
||
| 183 | |||
| 184 | typedef ScopedHandle<CommonHandleTraits> ScopedCommonHandle; |
||
| 185 | typedef ScopedHandle<FileHandleTraits> ScopedFileHandle; |
||
| 186 | typedef ScopedHandle<CryptContextTraits> ScopedCryptContext; |
||
| 187 | typedef ScopedHandle<RegTraits> ScopedRegHandle; |
||
| 188 | typedef ScopedHandle<FindHandleTraits> ScopedFindHandle; |
||
| 189 | typedef ScopedHandle<JobHandleTraits> ScopedJobHandle; |
||
| 190 | |||
| 191 | template <class T> |
||
| 192 | class SmallVectorImpl; |
||
| 193 | |||
| 194 | template <class T> |
||
| 195 | typename SmallVectorImpl<T>::const_pointer |
||
| 196 | c_str(SmallVectorImpl<T> &str) { |
||
| 197 | str.push_back(0); |
||
| 198 | str.pop_back(); |
||
| 199 | return str.data(); |
||
| 200 | } |
||
| 201 | |||
| 202 | namespace sys { |
||
| 203 | |||
| 204 | inline std::chrono::nanoseconds toDuration(FILETIME Time) { |
||
| 205 | ULARGE_INTEGER TimeInteger; |
||
| 206 | TimeInteger.LowPart = Time.dwLowDateTime; |
||
| 207 | TimeInteger.HighPart = Time.dwHighDateTime; |
||
| 208 | |||
| 209 | // FILETIME's are # of 100 nanosecond ticks (1/10th of a microsecond) |
||
| 210 | return std::chrono::nanoseconds(100 * TimeInteger.QuadPart); |
||
| 211 | } |
||
| 212 | |||
| 213 | inline TimePoint<> toTimePoint(FILETIME Time) { |
||
| 214 | ULARGE_INTEGER TimeInteger; |
||
| 215 | TimeInteger.LowPart = Time.dwLowDateTime; |
||
| 216 | TimeInteger.HighPart = Time.dwHighDateTime; |
||
| 217 | |||
| 218 | // Adjust for different epoch |
||
| 219 | TimeInteger.QuadPart -= 11644473600ll * 10000000; |
||
| 220 | |||
| 221 | // FILETIME's are # of 100 nanosecond ticks (1/10th of a microsecond) |
||
| 222 | return TimePoint<>(std::chrono::nanoseconds(100 * TimeInteger.QuadPart)); |
||
| 223 | } |
||
| 224 | |||
| 225 | inline FILETIME toFILETIME(TimePoint<> TP) { |
||
| 226 | ULARGE_INTEGER TimeInteger; |
||
| 227 | TimeInteger.QuadPart = TP.time_since_epoch().count() / 100; |
||
| 228 | TimeInteger.QuadPart += 11644473600ll * 10000000; |
||
| 229 | |||
| 230 | FILETIME Time; |
||
| 231 | Time.dwLowDateTime = TimeInteger.LowPart; |
||
| 232 | Time.dwHighDateTime = TimeInteger.HighPart; |
||
| 233 | return Time; |
||
| 234 | } |
||
| 235 | |||
| 236 | namespace windows { |
||
| 237 | // Returns command line arguments. Unlike arguments given to main(), |
||
| 238 | // this function guarantees that the returned arguments are encoded in |
||
| 239 | // UTF-8 regardless of the current code page setting. |
||
| 240 | std::error_code GetCommandLineArguments(SmallVectorImpl<const char *> &Args, |
||
| 241 | BumpPtrAllocator &Alloc); |
||
| 242 | |||
| 243 | /// Convert UTF-8 path to a suitable UTF-16 path for use with the Win32 Unicode |
||
| 244 | /// File API. |
||
| 245 | std::error_code widenPath(const Twine &Path8, SmallVectorImpl<wchar_t> &Path16, |
||
| 246 | size_t MaxPathLen = MAX_PATH); |
||
| 247 | |||
| 248 | } // end namespace windows |
||
| 249 | } // end namespace sys |
||
| 250 | } // end namespace llvm. |
||
| 251 | |||
| 252 | #endif |