Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 1 | //===- Win32/Win32.h - Common Win32 Include File ----------------*- C++ -*-===// |
Misha Brukman | f976c85 | 2005-04-21 22:55:34 +0000 | [diff] [blame] | 2 | // |
Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | f976c85 | 2005-04-21 22:55:34 +0000 | [diff] [blame] | 7 | // |
Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
Jeff Cohen | 6d23522 | 2005-07-13 02:15:18 +0000 | [diff] [blame] | 10 | // This file defines things specific to Win32 implementations. |
Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | //===----------------------------------------------------------------------===// |
Jeff Cohen | 6d23522 | 2005-07-13 02:15:18 +0000 | [diff] [blame] | 15 | //=== WARNING: Implementation here must contain only generic Win32 code that |
| 16 | //=== is guaranteed to work on *all* Win32 variants. |
Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 17 | //===----------------------------------------------------------------------===// |
| 18 | |
Jeff Cohen | 6d23522 | 2005-07-13 02:15:18 +0000 | [diff] [blame] | 19 | // Require at least Windows 2000 API. |
| 20 | #define _WIN32_WINNT 0x0500 |
Michael J. Spencer | eebe970 | 2010-11-09 15:11:19 +0000 | [diff] [blame] | 21 | #define _WIN32_IE 0x0500 // MinGW at it again. |
Michael J. Spencer | 966023b | 2010-11-09 15:11:07 +0000 | [diff] [blame] | 22 | #define WIN32_LEAN_AND_MEAN |
Jeff Cohen | 6d23522 | 2005-07-13 02:15:18 +0000 | [diff] [blame] | 23 | |
Michael J. Spencer | 64cf752 | 2010-11-09 15:10:29 +0000 | [diff] [blame] | 24 | #include "llvm/Config/config.h" // Get build system configuration settings |
Michael J. Spencer | 6d99092 | 2010-11-09 15:10:56 +0000 | [diff] [blame] | 25 | #include <Windows.h> |
Michael J. Spencer | eebe970 | 2010-11-09 15:11:19 +0000 | [diff] [blame] | 26 | #include <ShlObj.h> |
Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 27 | #include <cassert> |
| 28 | #include <string> |
| 29 | |
Reid Spencer | 5a06077 | 2006-08-23 07:30:48 +0000 | [diff] [blame] | 30 | inline bool MakeErrMsg(std::string* ErrMsg, const std::string& prefix) { |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 31 | if (!ErrMsg) |
Reid Spencer | 5a06077 | 2006-08-23 07:30:48 +0000 | [diff] [blame] | 32 | return true; |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 33 | char *buffer = NULL; |
| 34 | FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM, |
| 35 | NULL, GetLastError(), 0, (LPSTR)&buffer, 1, NULL); |
Anton Korobeynikov | 7d51544 | 2006-09-01 20:35:17 +0000 | [diff] [blame] | 36 | *ErrMsg = prefix + buffer; |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 37 | LocalFree(buffer); |
Reid Spencer | 5a06077 | 2006-08-23 07:30:48 +0000 | [diff] [blame] | 38 | return true; |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 39 | } |
Jeff Cohen | 0a18267 | 2007-03-05 05:22:08 +0000 | [diff] [blame] | 40 | |
| 41 | class AutoHandle { |
| 42 | HANDLE handle; |
| 43 | |
| 44 | public: |
| 45 | AutoHandle(HANDLE h) : handle(h) {} |
| 46 | |
| 47 | ~AutoHandle() { |
| 48 | if (handle) |
| 49 | CloseHandle(handle); |
| 50 | } |
| 51 | |
| 52 | operator HANDLE() { |
| 53 | return handle; |
| 54 | } |
| 55 | |
| 56 | AutoHandle &operator=(HANDLE h) { |
| 57 | handle = h; |
| 58 | return *this; |
| 59 | } |
| 60 | }; |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 61 | |
| 62 | template <class HandleType, HandleType InvalidHandle, |
| 63 | class DeleterType, DeleterType D> |
| 64 | class ScopedHandle { |
| 65 | HandleType Handle; |
| 66 | |
| 67 | public: |
| 68 | ScopedHandle() : Handle(InvalidHandle) {} |
| 69 | ScopedHandle(HandleType handle) : Handle(handle) {} |
| 70 | |
| 71 | ~ScopedHandle() { |
| 72 | if (Handle != InvalidHandle) |
| 73 | D(Handle); |
| 74 | } |
| 75 | |
| 76 | HandleType take() { |
| 77 | HandleType temp = Handle; |
| 78 | Handle = InvalidHandle; |
| 79 | return temp; |
| 80 | } |
| 81 | |
| 82 | operator HandleType() const { return Handle; } |
| 83 | |
| 84 | ScopedHandle &operator=(HandleType handle) { |
| 85 | Handle = handle; |
| 86 | return *this; |
| 87 | } |
| 88 | |
| 89 | typedef void (*unspecified_bool_type)(); |
| 90 | static void unspecified_bool_true() {} |
| 91 | |
| 92 | // True if Handle is valid. |
| 93 | operator unspecified_bool_type() const { |
| 94 | return Handle == InvalidHandle ? 0 : unspecified_bool_true; |
| 95 | } |
| 96 | |
Michael J. Spencer | 753cbbb | 2010-12-06 04:28:42 +0000 | [diff] [blame^] | 97 | bool operator!() const { |
| 98 | return Handle == InvalidHandle; |
| 99 | } |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 100 | }; |
Michael J. Spencer | 753cbbb | 2010-12-06 04:28:42 +0000 | [diff] [blame^] | 101 | |
| 102 | typedef ScopedHandle<HANDLE, INVALID_HANDLE_VALUE, |
| 103 | BOOL (WINAPI*)(HANDLE), ::FindClose> |
| 104 | ScopedFindHandle; |