Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(125)

Unified Diff: base/win/windows_version.cc

Issue 3823002: Move windows version-related stuff out of base/win_util and into base/win/win... (Closed) Base URL: svn://chrome-svn/chrome/trunk/src/
Patch Set: '' Created 10 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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

Powered by Google App Engine
This is Rietveld 408576698