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 | |
NAKAMURA Takumi | fd28949 | 2011-02-09 04:18:30 +0000 | [diff] [blame^] | 19 | // mingw-w64 tends to define it as 0x0502 in its headers. |
| 20 | #undef _WIN32_WINNT |
| 21 | |
Jeff Cohen | 6d23522 | 2005-07-13 02:15:18 +0000 | [diff] [blame] | 22 | // Require at least Windows 2000 API. |
| 23 | #define _WIN32_WINNT 0x0500 |
Michael J. Spencer | eebe970 | 2010-11-09 15:11:19 +0000 | [diff] [blame] | 24 | #define _WIN32_IE 0x0500 // MinGW at it again. |
Michael J. Spencer | 966023b | 2010-11-09 15:11:07 +0000 | [diff] [blame] | 25 | #define WIN32_LEAN_AND_MEAN |
Jeff Cohen | 6d23522 | 2005-07-13 02:15:18 +0000 | [diff] [blame] | 26 | |
Michael J. Spencer | 64cf752 | 2010-11-09 15:10:29 +0000 | [diff] [blame] | 27 | #include "llvm/Config/config.h" // Get build system configuration settings |
NAKAMURA Takumi | 6d600da | 2011-02-04 12:53:04 +0000 | [diff] [blame] | 28 | #include <windows.h> |
| 29 | #include <shlobj.h> |
Reid Spencer | 0b22ba4 | 2004-09-15 05:48:11 +0000 | [diff] [blame] | 30 | #include <cassert> |
| 31 | #include <string> |
| 32 | |
Reid Spencer | 5a06077 | 2006-08-23 07:30:48 +0000 | [diff] [blame] | 33 | inline bool MakeErrMsg(std::string* ErrMsg, const std::string& prefix) { |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 34 | if (!ErrMsg) |
Reid Spencer | 5a06077 | 2006-08-23 07:30:48 +0000 | [diff] [blame] | 35 | return true; |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 36 | char *buffer = NULL; |
| 37 | FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM, |
| 38 | NULL, GetLastError(), 0, (LPSTR)&buffer, 1, NULL); |
Anton Korobeynikov | 7d51544 | 2006-09-01 20:35:17 +0000 | [diff] [blame] | 39 | *ErrMsg = prefix + buffer; |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 40 | LocalFree(buffer); |
Reid Spencer | 5a06077 | 2006-08-23 07:30:48 +0000 | [diff] [blame] | 41 | return true; |
Reid Spencer | 4ce5dc6 | 2006-08-21 06:02:44 +0000 | [diff] [blame] | 42 | } |
Jeff Cohen | 0a18267 | 2007-03-05 05:22:08 +0000 | [diff] [blame] | 43 | |
| 44 | class AutoHandle { |
| 45 | HANDLE handle; |
| 46 | |
| 47 | public: |
| 48 | AutoHandle(HANDLE h) : handle(h) {} |
| 49 | |
| 50 | ~AutoHandle() { |
| 51 | if (handle) |
| 52 | CloseHandle(handle); |
| 53 | } |
| 54 | |
| 55 | operator HANDLE() { |
| 56 | return handle; |
| 57 | } |
| 58 | |
| 59 | AutoHandle &operator=(HANDLE h) { |
| 60 | handle = h; |
| 61 | return *this; |
| 62 | } |
| 63 | }; |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 64 | |
Michael J. Spencer | 476d2f7 | 2010-12-06 06:02:07 +0000 | [diff] [blame] | 65 | template <class HandleType, uintptr_t InvalidHandle, |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 66 | class DeleterType, DeleterType D> |
| 67 | class ScopedHandle { |
| 68 | HandleType Handle; |
| 69 | |
| 70 | public: |
| 71 | ScopedHandle() : Handle(InvalidHandle) {} |
| 72 | ScopedHandle(HandleType handle) : Handle(handle) {} |
| 73 | |
| 74 | ~ScopedHandle() { |
Michael J. Spencer | 476d2f7 | 2010-12-06 06:02:07 +0000 | [diff] [blame] | 75 | if (Handle != HandleType(InvalidHandle)) |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 76 | D(Handle); |
| 77 | } |
| 78 | |
| 79 | HandleType take() { |
| 80 | HandleType temp = Handle; |
Michael J. Spencer | 476d2f7 | 2010-12-06 06:02:07 +0000 | [diff] [blame] | 81 | Handle = HandleType(InvalidHandle); |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 82 | return temp; |
| 83 | } |
| 84 | |
| 85 | operator HandleType() const { return Handle; } |
| 86 | |
| 87 | ScopedHandle &operator=(HandleType handle) { |
| 88 | Handle = handle; |
| 89 | return *this; |
| 90 | } |
| 91 | |
| 92 | typedef void (*unspecified_bool_type)(); |
| 93 | static void unspecified_bool_true() {} |
| 94 | |
| 95 | // True if Handle is valid. |
| 96 | operator unspecified_bool_type() const { |
Michael J. Spencer | 476d2f7 | 2010-12-06 06:02:07 +0000 | [diff] [blame] | 97 | return Handle == HandleType(InvalidHandle) ? 0 : unspecified_bool_true; |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 98 | } |
| 99 | |
Michael J. Spencer | 753cbbb | 2010-12-06 04:28:42 +0000 | [diff] [blame] | 100 | bool operator!() const { |
Michael J. Spencer | 476d2f7 | 2010-12-06 06:02:07 +0000 | [diff] [blame] | 101 | return Handle == HandleType(InvalidHandle); |
Michael J. Spencer | 753cbbb | 2010-12-06 04:28:42 +0000 | [diff] [blame] | 102 | } |
Michael J. Spencer | 7eec50b | 2010-12-06 04:28:13 +0000 | [diff] [blame] | 103 | }; |
Michael J. Spencer | 753cbbb | 2010-12-06 04:28:42 +0000 | [diff] [blame] | 104 | |
Michael J. Spencer | 476d2f7 | 2010-12-06 06:02:07 +0000 | [diff] [blame] | 105 | typedef ScopedHandle<HANDLE, uintptr_t(-1), |
Michael J. Spencer | 753cbbb | 2010-12-06 04:28:42 +0000 | [diff] [blame] | 106 | BOOL (WINAPI*)(HANDLE), ::FindClose> |
| 107 | ScopedFindHandle; |
Michael J. Spencer | 58fe86d | 2010-12-09 17:37:18 +0000 | [diff] [blame] | 108 | |
| 109 | namespace llvm { |
| 110 | template <class T> |
| 111 | class SmallVectorImpl; |
| 112 | |
| 113 | template <class T> |
| 114 | typename SmallVectorImpl<T>::const_pointer |
| 115 | c_str(SmallVectorImpl<T> &str) { |
| 116 | str.push_back(0); |
| 117 | str.pop_back(); |
| 118 | return str.data(); |
| 119 | } |
| 120 | } // end namespace llvm. |