| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "sandbox/src/sandbox_utils.h" | |
| 6 | |
| 7 #include <windows.h> | |
| 8 | |
| 9 #include "base/logging.h" | |
| 10 #include "base/win/windows_version.h" | |
| 11 #include "sandbox/src/internal_types.h" | |
| 12 #include "sandbox/src/nt_internals.h" | |
| 13 | |
| 14 namespace sandbox { | |
| 15 | |
| 16 bool GetModuleHandleHelper(DWORD flags, const wchar_t* module_name, | |
| 17 HMODULE* module) { | |
| 18 DCHECK(module); | |
| 19 | |
| 20 HMODULE kernel32_base = ::GetModuleHandle(kKerneldllName); | |
| 21 if (!kernel32_base) { | |
| 22 NOTREACHED(); | |
| 23 return false; | |
| 24 } | |
| 25 | |
| 26 GetModuleHandleExFunction get_module_handle_ex = reinterpret_cast< | |
| 27 GetModuleHandleExFunction>(::GetProcAddress(kernel32_base, | |
| 28 "GetModuleHandleExW")); | |
| 29 if (get_module_handle_ex) { | |
| 30 BOOL ret = get_module_handle_ex(flags, module_name, module); | |
| 31 return (ret ? true : false); | |
| 32 } | |
| 33 | |
| 34 if (!flags) { | |
| 35 *module = ::LoadLibrary(module_name); | |
| 36 } else if (flags & GET_MODULE_HANDLE_EX_FLAG_PIN) { | |
| 37 NOTREACHED(); | |
| 38 return false; | |
| 39 } else if (!(flags & GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS)) { | |
| 40 DCHECK((flags & GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT) == | |
| 41 GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT); | |
| 42 | |
| 43 *module = ::GetModuleHandle(module_name); | |
| 44 } else { | |
| 45 DCHECK((flags & (GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT | | |
| 46 GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS)) == | |
| 47 (GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT | | |
| 48 GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS)); | |
| 49 | |
| 50 MEMORY_BASIC_INFORMATION info = {0}; | |
| 51 size_t returned = VirtualQuery(module_name, &info, sizeof(info)); | |
| 52 if (sizeof(info) != returned) | |
| 53 return false; | |
| 54 *module = reinterpret_cast<HMODULE>(info.AllocationBase); | |
| 55 } | |
| 56 return true; | |
| 57 } | |
| 58 | |
| 59 bool IsXPSP2OrLater() { | |
| 60 base::win::Version version = base::win::GetVersion(); | |
| 61 return (version > base::win::VERSION_XP) || | |
| 62 ((version == base::win::VERSION_XP) && | |
| 63 (base::win::OSInfo::GetInstance()->service_pack().major >= 2)); | |
| 64 } | |
| 65 | |
| 66 void InitObjectAttribs(const std::wstring& name, ULONG attributes, HANDLE root, | |
| 67 OBJECT_ATTRIBUTES* obj_attr, UNICODE_STRING* uni_name) { | |
| 68 static RtlInitUnicodeStringFunction RtlInitUnicodeString; | |
| 69 if (!RtlInitUnicodeString) { | |
| 70 HMODULE ntdll = ::GetModuleHandle(kNtdllName); | |
| 71 RtlInitUnicodeString = reinterpret_cast<RtlInitUnicodeStringFunction>( | |
| 72 GetProcAddress(ntdll, "RtlInitUnicodeString")); | |
| 73 DCHECK(RtlInitUnicodeString); | |
| 74 } | |
| 75 RtlInitUnicodeString(uni_name, name.c_str()); | |
| 76 InitializeObjectAttributes(obj_attr, uni_name, attributes, root, NULL); | |
| 77 } | |
| 78 | |
| 79 }; // namespace sandbox | |
| OLD | NEW |