10b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer//===- Win32/Win32.h - Common Win32 Include File ----------------*- C++ -*-===// 2f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman// 30b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer// The LLVM Compiler Infrastructure 40b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer// 54ee451de366474b9c228b4e5fa573795a715216dChris Lattner// This file is distributed under the University of Illinois Open Source 64ee451de366474b9c228b4e5fa573795a715216dChris Lattner// License. See LICENSE.TXT for details. 7f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman// 80b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer//===----------------------------------------------------------------------===// 90b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer// 106d2352249af8853c8307a7cf679017b32d27958cJeff Cohen// This file defines things specific to Win32 implementations. 110b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer// 120b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer//===----------------------------------------------------------------------===// 130b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer 140b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer//===----------------------------------------------------------------------===// 156d2352249af8853c8307a7cf679017b32d27958cJeff Cohen//=== WARNING: Implementation here must contain only generic Win32 code that 166d2352249af8853c8307a7cf679017b32d27958cJeff Cohen//=== is guaranteed to work on *all* Win32 variants. 170b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer//===----------------------------------------------------------------------===// 180b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer 19fd289497a5e8f83c6d83f1eb7e30fdc52621bf15NAKAMURA Takumi// mingw-w64 tends to define it as 0x0502 in its headers. 20fd289497a5e8f83c6d83f1eb7e30fdc52621bf15NAKAMURA Takumi#undef _WIN32_WINNT 21fd289497a5e8f83c6d83f1eb7e30fdc52621bf15NAKAMURA Takumi 22aa629668db2dc827741d116515276228327b0282NAKAMURA Takumi// Require at least Windows XP(5.1) API. 23aa629668db2dc827741d116515276228327b0282NAKAMURA Takumi#define _WIN32_WINNT 0x0501 2407097ddbc0e07a1c2b3bce2fb64c5e964da9779aNAKAMURA Takumi#define _WIN32_IE 0x0600 // MinGW at it again. 25966023b58ad779cb447a0cfb751de0c823925340Michael J. Spencer#define WIN32_LEAN_AND_MEAN 266d2352249af8853c8307a7cf679017b32d27958cJeff Cohen 2764cf752e9cc6beafbafc387d6a0ccf79bc774e7cMichael J. Spencer#include "llvm/Config/config.h" // Get build system configuration settings 28f1a8d770f27bbce61b102bce5c8461f407843fc2Timur Iskhodzhanov#include "llvm/Support/Compiler.h" 296d600da7346bf898cfbe8f6030051f10d157fcd4NAKAMURA Takumi#include <windows.h> 301dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer#include <wincrypt.h> 316d600da7346bf898cfbe8f6030051f10d157fcd4NAKAMURA Takumi#include <shlobj.h> 320b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer#include <cassert> 330b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer#include <string> 340b22ba4c30519a209bb1286e89e9e512372cbd1bReid Spencer 355a060775da43ec2ef37e839e93a0805d543708a9Reid Spencerinline bool MakeErrMsg(std::string* ErrMsg, const std::string& prefix) { 364ce5dc63778f36f61b510456783f15a224406e68Reid Spencer if (!ErrMsg) 375a060775da43ec2ef37e839e93a0805d543708a9Reid Spencer return true; 384ce5dc63778f36f61b510456783f15a224406e68Reid Spencer char *buffer = NULL; 394ce5dc63778f36f61b510456783f15a224406e68Reid Spencer FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM, 404ce5dc63778f36f61b510456783f15a224406e68Reid Spencer NULL, GetLastError(), 0, (LPSTR)&buffer, 1, NULL); 417d5154490357a0cb12383e15a57640d9c20644a1Anton Korobeynikov *ErrMsg = prefix + buffer; 424ce5dc63778f36f61b510456783f15a224406e68Reid Spencer LocalFree(buffer); 435a060775da43ec2ef37e839e93a0805d543708a9Reid Spencer return true; 444ce5dc63778f36f61b510456783f15a224406e68Reid Spencer} 450a1826724821945653477c867c77ea6f245bab02Jeff Cohen 461dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencertemplate <typename HandleTraits> 471dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencerclass ScopedHandle { 481dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer typedef typename HandleTraits::handle_type handle_type; 491dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer handle_type Handle; 500a1826724821945653477c867c77ea6f245bab02Jeff Cohen 511dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ScopedHandle(const ScopedHandle &other); // = delete; 521dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer void operator=(const ScopedHandle &other); // = delete; 530a1826724821945653477c867c77ea6f245bab02Jeff Cohenpublic: 541dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ScopedHandle() 551dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer : Handle(HandleTraits::GetInvalid()) {} 561dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer 571dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer explicit ScopedHandle(handle_type h) 581dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer : Handle(h) {} 590a1826724821945653477c867c77ea6f245bab02Jeff Cohen 601dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ~ScopedHandle() { 611dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer if (HandleTraits::IsValid(Handle)) 621dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer HandleTraits::Close(Handle); 630a1826724821945653477c867c77ea6f245bab02Jeff Cohen } 640a1826724821945653477c867c77ea6f245bab02Jeff Cohen 651dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer handle_type take() { 661dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer handle_type t = Handle; 671dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer Handle = HandleTraits::GetInvalid(); 681dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return t; 690a1826724821945653477c867c77ea6f245bab02Jeff Cohen } 700a1826724821945653477c867c77ea6f245bab02Jeff Cohen 711dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ScopedHandle &operator=(handle_type h) { 721dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer if (HandleTraits::IsValid(Handle)) 731dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer HandleTraits::Close(Handle); 741dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer Handle = h; 750a1826724821945653477c867c77ea6f245bab02Jeff Cohen return *this; 760a1826724821945653477c867c77ea6f245bab02Jeff Cohen } 771dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer 781dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer // True if Handle is valid. 79453f4f01302f00651aae2fc7658f6e23a2beadb0David Blaikie LLVM_EXPLICIT operator bool() const { 801dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return HandleTraits::IsValid(Handle) ? true : false; 811dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer } 821dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer 831dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer operator handle_type() const { 841dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return Handle; 851dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer } 860a1826724821945653477c867c77ea6f245bab02Jeff Cohen}; 877eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 881dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencerstruct CommonHandleTraits { 891dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer typedef HANDLE handle_type; 907eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 911dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static handle_type GetInvalid() { 921dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return INVALID_HANDLE_VALUE; 931dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer } 947eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 951dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static void Close(handle_type h) { 961dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ::CloseHandle(h); 977eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer } 987eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 991dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static bool IsValid(handle_type h) { 1001dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return h != GetInvalid(); 1017eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer } 1021dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer}; 1037eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 1041dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencerstruct JobHandleTraits : CommonHandleTraits { 1051dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static handle_type GetInvalid() { 1061dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return NULL; 1071dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer } 1081dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer}; 1097eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 1101dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencerstruct CryptContextTraits : CommonHandleTraits { 1111dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer typedef HCRYPTPROV handle_type; 1121dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer 1131dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static handle_type GetInvalid() { 1141dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return 0; 1157eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer } 1167eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 1171dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static void Close(handle_type h) { 1181dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ::CryptReleaseContext(h, 0); 1191dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer } 1207eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 1211dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static bool IsValid(handle_type h) { 1221dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer return h != GetInvalid(); 1237eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer } 1241dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer}; 1257eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer 1261dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencerstruct FindHandleTraits : CommonHandleTraits { 1271dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer static void Close(handle_type h) { 1281dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer ::FindClose(h); 129753cbbbd3ce28253f381caff83ce2a7f6e08785bMichael J. Spencer } 1307eec50bf4f9d050323c0da9ac61ac3b150e4e5b1Michael J. Spencer}; 131753cbbbd3ce28253f381caff83ce2a7f6e08785bMichael J. Spencer 1321dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencerstruct FileHandleTraits : CommonHandleTraits {}; 1331dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencer 1341dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencertypedef ScopedHandle<CommonHandleTraits> ScopedCommonHandle; 1351dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencertypedef ScopedHandle<FileHandleTraits> ScopedFileHandle; 1361dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencertypedef ScopedHandle<CryptContextTraits> ScopedCryptContext; 1371dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencertypedef ScopedHandle<FindHandleTraits> ScopedFindHandle; 1381dd2ee7bf4d848e368829cb01033f297afb674abMichael J. Spencertypedef ScopedHandle<JobHandleTraits> ScopedJobHandle; 13958fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer 14058fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencernamespace llvm { 14158fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencertemplate <class T> 14258fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencerclass SmallVectorImpl; 14358fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer 14458fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencertemplate <class T> 14558fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencertypename SmallVectorImpl<T>::const_pointer 14658fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencerc_str(SmallVectorImpl<T> &str) { 14758fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer str.push_back(0); 14858fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer str.pop_back(); 14958fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer return str.data(); 15058fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer} 15158fe86dc0ecb7efff01abe2b0024a6a53ebb2c81Michael J. Spencer} // end namespace llvm. 152