Index: base/win/windows_version.cc |
=================================================================== |
--- base/win/windows_version.cc (revision 0) |
+++ base/win/windows_version.cc (working copy) |
@@ -2,48 +2,18 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
-#include "base/win_util.h" |
+#include "base/win/windows_version.h" |
-#include <aclapi.h> |
-#include <propvarutil.h> |
-#include <sddl.h> |
-#include <shlobj.h> |
+#include <windows.h> |
#include "base/logging.h" |
-#include "base/registry.h" |
-#include "base/scoped_handle.h" |
-#include "base/scoped_ptr.h" |
-#include "base/string_util.h" |
-#include "base/stringprintf.h" |
-namespace win_util { |
+namespace base { |
+namespace win { |
-#define SIZEOF_STRUCT_WITH_SPECIFIED_LAST_MEMBER(struct_name, member) \ |
- offsetof(struct_name, member) + \ |
- (sizeof static_cast<struct_name*>(NULL)->member) |
-#define NONCLIENTMETRICS_SIZE_PRE_VISTA \ |
- SIZEOF_STRUCT_WITH_SPECIFIED_LAST_MEMBER(NONCLIENTMETRICS, lfMessageFont) |
- |
-const PROPERTYKEY kPKEYAppUserModelID = |
- { { 0x9F4C2855, 0x9F79, 0x4B39, |
- { 0xA8, 0xD0, 0xE1, 0xD4, 0x2D, 0xE1, 0xD5, 0xF3, } }, 5 }; |
- |
-void GetNonClientMetrics(NONCLIENTMETRICS* metrics) { |
- DCHECK(metrics); |
- |
- static const UINT SIZEOF_NONCLIENTMETRICS = |
- (GetWinVersion() >= WINVERSION_VISTA) ? |
- sizeof(NONCLIENTMETRICS) : NONCLIENTMETRICS_SIZE_PRE_VISTA; |
- metrics->cbSize = SIZEOF_NONCLIENTMETRICS; |
- const bool success = !!SystemParametersInfo(SPI_GETNONCLIENTMETRICS, |
- SIZEOF_NONCLIENTMETRICS, metrics, |
- 0); |
- DCHECK(success); |
-} |
- |
-WinVersion GetWinVersion() { |
+Version GetWinVersion() { |
static bool checked_version = false; |
- static WinVersion win_version = WINVERSION_PRE_2000; |
+ static Version win_version = VERSION_PRE_2000; |
if (!checked_version) { |
OSVERSIONINFOEX version_info; |
version_info.dwOSVersionInfoSize = sizeof version_info; |
@@ -51,29 +21,29 @@ |
if (version_info.dwMajorVersion == 5) { |
switch (version_info.dwMinorVersion) { |
case 0: |
- win_version = WINVERSION_2000; |
+ win_version = VERSION_2000; |
break; |
case 1: |
- win_version = WINVERSION_XP; |
+ win_version = VERSION_XP; |
break; |
case 2: |
default: |
- win_version = WINVERSION_SERVER_2003; |
+ win_version = VERSION_SERVER_2003; |
break; |
} |
} else if (version_info.dwMajorVersion == 6) { |
if (version_info.wProductType != VER_NT_WORKSTATION) { |
// 2008 is 6.0, and 2008 R2 is 6.1. |
- win_version = WINVERSION_2008; |
+ win_version = VERSION_2008; |
} else { |
if (version_info.dwMinorVersion == 0) { |
- win_version = WINVERSION_VISTA; |
+ win_version = VERSION_VISTA; |
} else { |
- win_version = WINVERSION_WIN7; |
+ win_version = VERSION_WIN7; |
} |
} |
} else if (version_info.dwMajorVersion > 6) { |
- win_version = WINVERSION_WIN7; |
+ win_version = VERSION_WIN7; |
} |
checked_version = true; |
} |
@@ -98,349 +68,5 @@ |
*minor = service_pack_minor; |
} |
-bool AddAccessToKernelObject(HANDLE handle, WELL_KNOWN_SID_TYPE known_sid, |
- ACCESS_MASK access) { |
- PSECURITY_DESCRIPTOR descriptor = NULL; |
- PACL old_dacl = NULL; |
- PACL new_dacl = NULL; |
- |
- if (ERROR_SUCCESS != GetSecurityInfo(handle, SE_KERNEL_OBJECT, |
- DACL_SECURITY_INFORMATION, NULL, NULL, &old_dacl, NULL, |
- &descriptor)) |
- return false; |
- |
- BYTE sid[SECURITY_MAX_SID_SIZE] = {0}; |
- DWORD size_sid = SECURITY_MAX_SID_SIZE; |
- |
- if (known_sid == WinSelfSid) { |
- // We hijack WinSelfSid when we want to add the current user instead of |
- // a known sid. |
- HANDLE token = NULL; |
- if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &token)) { |
- LocalFree(descriptor); |
- return false; |
- } |
- |
- DWORD size = sizeof(TOKEN_USER) + size_sid; |
- scoped_array<BYTE> token_user_bytes(new BYTE[size]); |
- TOKEN_USER* token_user = |
- reinterpret_cast<TOKEN_USER*>(token_user_bytes.get()); |
- BOOL ret = GetTokenInformation(token, TokenUser, token_user, size, &size); |
- |
- CloseHandle(token); |
- |
- if (!ret) { |
- LocalFree(descriptor); |
- return false; |
- } |
- memcpy(sid, token_user->User.Sid, size_sid); |
- } else { |
- if (!CreateWellKnownSid(known_sid , NULL, sid, &size_sid)) { |
- LocalFree(descriptor); |
- return false; |
- } |
- } |
- |
- EXPLICIT_ACCESS new_access = {0}; |
- new_access.grfAccessMode = GRANT_ACCESS; |
- new_access.grfAccessPermissions = access; |
- new_access.grfInheritance = NO_INHERITANCE; |
- |
- new_access.Trustee.pMultipleTrustee = NULL; |
- new_access.Trustee.MultipleTrusteeOperation = NO_MULTIPLE_TRUSTEE; |
- new_access.Trustee.TrusteeForm = TRUSTEE_IS_SID; |
- new_access.Trustee.ptstrName = reinterpret_cast<LPWSTR>(&sid); |
- |
- if (ERROR_SUCCESS != SetEntriesInAcl(1, &new_access, old_dacl, &new_dacl)) { |
- LocalFree(descriptor); |
- return false; |
- } |
- |
- DWORD result = SetSecurityInfo(handle, SE_KERNEL_OBJECT, |
- DACL_SECURITY_INFORMATION, NULL, NULL, |
- new_dacl, NULL); |
- |
- LocalFree(new_dacl); |
- LocalFree(descriptor); |
- |
- if (ERROR_SUCCESS != result) |
- return false; |
- |
- return true; |
-} |
- |
-bool GetUserSidString(std::wstring* user_sid) { |
- // Get the current token. |
- HANDLE token = NULL; |
- if (!::OpenProcessToken(::GetCurrentProcess(), TOKEN_QUERY, &token)) |
- return false; |
- ScopedHandle token_scoped(token); |
- |
- DWORD size = sizeof(TOKEN_USER) + SECURITY_MAX_SID_SIZE; |
- scoped_array<BYTE> user_bytes(new BYTE[size]); |
- TOKEN_USER* user = reinterpret_cast<TOKEN_USER*>(user_bytes.get()); |
- |
- if (!::GetTokenInformation(token, TokenUser, user, size, &size)) |
- return false; |
- |
- if (!user->User.Sid) |
- return false; |
- |
- // Convert the data to a string. |
- wchar_t* sid_string; |
- if (!::ConvertSidToStringSid(user->User.Sid, &sid_string)) |
- return false; |
- |
- *user_sid = sid_string; |
- |
- ::LocalFree(sid_string); |
- |
- return true; |
-} |
- |
-bool GetLogonSessionOnlyDACL(SECURITY_DESCRIPTOR** security_descriptor) { |
- // Get the current token. |
- HANDLE token = NULL; |
- if (!OpenProcessToken(::GetCurrentProcess(), TOKEN_QUERY, &token)) |
- return false; |
- ScopedHandle token_scoped(token); |
- |
- // Get the size of the TokenGroups structure. |
- DWORD size = 0; |
- BOOL result = GetTokenInformation(token, TokenGroups, NULL, 0, &size); |
- if (result != FALSE && GetLastError() != ERROR_INSUFFICIENT_BUFFER) |
- return false; |
- |
- // Get the data. |
- scoped_array<char> token_groups_chars(new char[size]); |
- TOKEN_GROUPS* token_groups = |
- reinterpret_cast<TOKEN_GROUPS*>(token_groups_chars.get()); |
- |
- if (!GetTokenInformation(token, TokenGroups, token_groups, size, &size)) |
- return false; |
- |
- // Look for the logon sid. |
- SID* logon_sid = NULL; |
- for (unsigned int i = 0; i < token_groups->GroupCount ; ++i) { |
- if ((token_groups->Groups[i].Attributes & SE_GROUP_LOGON_ID) != 0) { |
- logon_sid = static_cast<SID*>(token_groups->Groups[i].Sid); |
- break; |
- } |
- } |
- |
- if (!logon_sid) |
- return false; |
- |
- // Convert the data to a string. |
- wchar_t* sid_string; |
- if (!ConvertSidToStringSid(logon_sid, &sid_string)) |
- return false; |
- |
- static const wchar_t dacl_format[] = L"D:(A;OICI;GA;;;%ls)"; |
- wchar_t dacl[SECURITY_MAX_SID_SIZE + arraysize(dacl_format) + 1] = {0}; |
- wsprintf(dacl, dacl_format, sid_string); |
- |
- LocalFree(sid_string); |
- |
- // Convert the string to a security descriptor |
- if (!ConvertStringSecurityDescriptorToSecurityDescriptor( |
- dacl, |
- SDDL_REVISION_1, |
- reinterpret_cast<PSECURITY_DESCRIPTOR*>(security_descriptor), |
- NULL)) { |
- return false; |
- } |
- |
- return true; |
-} |
- |
-#pragma warning(push) |
-#pragma warning(disable:4312 4244) |
-WNDPROC SetWindowProc(HWND hwnd, WNDPROC proc) { |
- // The reason we don't return the SetwindowLongPtr() value is that it returns |
- // the orignal window procedure and not the current one. I don't know if it is |
- // a bug or an intended feature. |
- WNDPROC oldwindow_proc = |
- reinterpret_cast<WNDPROC>(GetWindowLongPtr(hwnd, GWLP_WNDPROC)); |
- SetWindowLongPtr(hwnd, GWLP_WNDPROC, reinterpret_cast<LONG_PTR>(proc)); |
- return oldwindow_proc; |
-} |
- |
-void* SetWindowUserData(HWND hwnd, void* user_data) { |
- return |
- reinterpret_cast<void*>(SetWindowLongPtr(hwnd, GWLP_USERDATA, |
- reinterpret_cast<LONG_PTR>(user_data))); |
-} |
- |
-void* GetWindowUserData(HWND hwnd) { |
- return reinterpret_cast<void*>(GetWindowLongPtr(hwnd, GWLP_USERDATA)); |
-} |
- |
-// Maps to the WNDPROC for a window that was active before the subclass was |
-// installed. |
-static const wchar_t* const kHandlerKey = L"__ORIGINAL_MESSAGE_HANDLER__"; |
- |
-bool IsSubclassed(HWND window, WNDPROC subclass_proc) { |
- WNDPROC original_handler = |
- reinterpret_cast<WNDPROC>(GetWindowLongPtr(window, GWLP_WNDPROC)); |
- return original_handler == subclass_proc; |
-} |
- |
-bool Subclass(HWND window, WNDPROC subclass_proc) { |
- WNDPROC original_handler = |
- reinterpret_cast<WNDPROC>(GetWindowLongPtr(window, GWLP_WNDPROC)); |
- if (original_handler != subclass_proc) { |
- win_util::SetWindowProc(window, subclass_proc); |
- SetProp(window, kHandlerKey, original_handler); |
- return true; |
- } |
- return false; |
-} |
- |
-bool Unsubclass(HWND window, WNDPROC subclass_proc) { |
- WNDPROC current_handler = |
- reinterpret_cast<WNDPROC>(GetWindowLongPtr(window, GWLP_WNDPROC)); |
- if (current_handler == subclass_proc) { |
- HANDLE original_handler = GetProp(window, kHandlerKey); |
- if (original_handler) { |
- RemoveProp(window, kHandlerKey); |
- win_util::SetWindowProc(window, |
- reinterpret_cast<WNDPROC>(original_handler)); |
- return true; |
- } |
- } |
- return false; |
-} |
- |
-WNDPROC GetSuperclassWNDPROC(HWND window) { |
- return reinterpret_cast<WNDPROC>(GetProp(window, kHandlerKey)); |
-} |
- |
-#pragma warning(pop) |
- |
-bool IsShiftPressed() { |
- return (::GetKeyState(VK_SHIFT) & 0x8000) == 0x8000; |
-} |
- |
-bool IsCtrlPressed() { |
- return (::GetKeyState(VK_CONTROL) & 0x8000) == 0x8000; |
-} |
- |
-bool IsAltPressed() { |
- return (::GetKeyState(VK_MENU) & 0x8000) == 0x8000; |
-} |
- |
-std::wstring GetClassName(HWND window) { |
- // GetClassNameW will return a truncated result (properly null terminated) if |
- // the given buffer is not large enough. So, it is not possible to determine |
- // that we got the entire class name if the result is exactly equal to the |
- // size of the buffer minus one. |
- DWORD buffer_size = MAX_PATH; |
- while (true) { |
- std::wstring output; |
- DWORD size_ret = |
- GetClassNameW(window, WriteInto(&output, buffer_size), buffer_size); |
- if (size_ret == 0) |
- break; |
- if (size_ret < (buffer_size - 1)) { |
- output.resize(size_ret); |
- return output; |
- } |
- buffer_size *= 2; |
- } |
- return std::wstring(); // error |
-} |
- |
-bool UserAccountControlIsEnabled() { |
- RegKey key(HKEY_LOCAL_MACHINE, |
- L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Policies\\System", |
- KEY_READ); |
- DWORD uac_enabled; |
- if (!key.ReadValueDW(L"EnableLUA", &uac_enabled)) |
- return true; |
- // Users can set the EnableLUA value to something arbitrary, like 2, which |
- // Vista will treat as UAC enabled, so we make sure it is not set to 0. |
- return (uac_enabled != 0); |
-} |
- |
-std::wstring FormatMessage(unsigned messageid) { |
- wchar_t* string_buffer = NULL; |
- unsigned string_length = ::FormatMessage( |
- FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | |
- FORMAT_MESSAGE_IGNORE_INSERTS, NULL, messageid, 0, |
- reinterpret_cast<wchar_t *>(&string_buffer), 0, NULL); |
- |
- std::wstring formatted_string; |
- if (string_buffer) { |
- formatted_string = string_buffer; |
- LocalFree(reinterpret_cast<HLOCAL>(string_buffer)); |
- } else { |
- // The formating failed. simply convert the message value into a string. |
- base::SStringPrintf(&formatted_string, L"message number %d", messageid); |
- } |
- return formatted_string; |
-} |
- |
-std::wstring FormatLastWin32Error() { |
- return FormatMessage(GetLastError()); |
-} |
- |
-bool SetAppIdForPropertyStore(IPropertyStore* property_store, |
- const wchar_t* app_id) { |
- DCHECK(property_store); |
- |
- // App id should be less than 128 chars and contain no space. And recommended |
- // format is CompanyName.ProductName[.SubProduct.ProductNumber]. |
- // See http://msdn.microsoft.com/en-us/library/dd378459%28VS.85%29.aspx |
- DCHECK(lstrlen(app_id) < 128 && wcschr(app_id, L' ') == NULL); |
- |
- PROPVARIANT property_value; |
- if (FAILED(InitPropVariantFromString(app_id, &property_value))) |
- return false; |
- |
- HRESULT result = property_store->SetValue(kPKEYAppUserModelID, |
- property_value); |
- if (S_OK == result) |
- result = property_store->Commit(); |
- |
- PropVariantClear(&property_value); |
- return SUCCEEDED(result); |
-} |
- |
-static const char16 kAutoRunKeyPath[] = |
- L"Software\\Microsoft\\Windows\\CurrentVersion\\Run"; |
- |
-bool AddCommandToAutoRun(HKEY root_key, const string16& name, |
- const string16& command) { |
- RegKey autorun_key(root_key, kAutoRunKeyPath, KEY_SET_VALUE); |
- return autorun_key.WriteValue(name.c_str(), command.c_str()); |
-} |
- |
-bool RemoveCommandFromAutoRun(HKEY root_key, const string16& name) { |
- RegKey autorun_key(root_key, kAutoRunKeyPath, KEY_SET_VALUE); |
- return autorun_key.DeleteValue(name.c_str()); |
-} |
- |
-} // namespace win_util |
- |
-#ifdef _MSC_VER |
-// |
-// If the ASSERT below fails, please install Visual Studio 2005 Service Pack 1. |
-// |
-extern char VisualStudio2005ServicePack1Detection[10]; |
-COMPILE_ASSERT(sizeof(&VisualStudio2005ServicePack1Detection) == sizeof(void*), |
- VS2005SP1Detect); |
-// |
-// Chrome requires at least Service Pack 1 for Visual Studio 2005. |
-// |
-#endif // _MSC_VER |
- |
-#ifndef COPY_FILE_COPY_SYMLINK |
-#error You must install the Windows 2008 or Vista Software Development Kit and \ |
-set it as your default include path to build this library. You can grab it by \ |
-searching for "download windows sdk 2008" in your favorite web search engine. \ |
-Also make sure you register the SDK with Visual Studio, by selecting \ |
-"Integrate Windows SDK with Visual Studio 2005" from the Windows SDK \ |
-menu (see Start - All Programs - Microsoft Windows SDK - \ |
-Visual Studio Registration). |
-#endif |
+} // namespace win |
+} // namespace base |