Add sandbox to the repository.

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@13 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 126 insertions, 0 deletions

diff --git a/sandbox/src/sandbox_utils.cc b/sandbox/src/sandbox_utils.cc
new file mode 100644
index 0000000..748575b
--- /dev/null
+++ b/sandbox/src/sandbox_utils.cc
@@ -0,0 +1,126 @@
+// Copyright 2008, Google Inc.
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are
+// met:
+//
+//    * Redistributions of source code must retain the above copyright
+// notice, this list of conditions and the following disclaimer.
+//    * Redistributions in binary form must reproduce the above
+// copyright notice, this list of conditions and the following disclaimer
+// in the documentation and/or other materials provided with the
+// distribution.
+//    * Neither the name of Google Inc. nor the names of its
+// contributors may be used to endorse or promote products derived from
+// this software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+#include "sandbox/src/sandbox_utils.h"
+
+#include <windows.h>
+
+#include "base/logging.h"
+#include "sandbox/src/internal_types.h"
+#include "sandbox/src/nt_internals.h"
+
+namespace sandbox {
+
+bool GetModuleHandleHelper(DWORD flags, const wchar_t* module_name,
+                           HMODULE* module) {
+  DCHECK(module);
+
+  HMODULE kernel32_base = ::GetModuleHandle(kKerneldllName);
+  if (!kernel32_base) {
+    NOTREACHED();
+    return false;
+  }
+
+  GetModuleHandleExFunction get_module_handle_ex = reinterpret_cast<
+      GetModuleHandleExFunction>(::GetProcAddress(kernel32_base,
+                                                  "GetModuleHandleExW"));
+  if (get_module_handle_ex) {
+    BOOL ret = get_module_handle_ex(flags, module_name, module);
+    return (ret ? true : false);
+  }
+
+  if (!flags) {
+    *module = ::LoadLibrary(module_name);
+  } else if (flags & GET_MODULE_HANDLE_EX_FLAG_PIN) {
+    NOTREACHED();
+    return false;
+  } else if (!(flags & GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS)) {
+    DCHECK((flags & GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT) ==
+           GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT);
+
+    *module = ::GetModuleHandle(module_name);
+  } else {
+    DCHECK((flags & (GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT |
+                    GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS)) ==
+           (GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT |
+            GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS));
+
+    MEMORY_BASIC_INFORMATION info = {0};
+    size_t returned = VirtualQuery(module_name, &info, sizeof(info));
+    if (sizeof(info) != returned)
+      return false;
+    *module = reinterpret_cast<HMODULE>(info.AllocationBase);
+  }
+  return true;
+}
+
+bool IsXPSP2OrLater() {
+  OSVERSIONINFOEX version = {0};
+  version.dwOSVersionInfoSize = sizeof(version);
+  if (!::GetVersionEx(reinterpret_cast<OSVERSIONINFO*>(&version))) {
+    NOTREACHED();
+    return false;
+  }
+
+  // Vista or later
+  if (version.dwMajorVersion > 5)
+    return true;
+
+  // 2k, xp or 2003
+  if (version.dwMajorVersion == 5) {
+    // 2003
+    if (version.dwMinorVersion > 1)
+      return true;
+
+    // 2000
+    if (version.dwMinorVersion == 0)
+      return false;
+
+    // Windows Xp Sp2 or later
+    if (version.wServicePackMajor >= 2)
+      return true;
+  }
+
+  return false;
+}
+
+void InitObjectAttribs(const std::wstring& name, ULONG attributes, HANDLE root,
+                       OBJECT_ATTRIBUTES* obj_attr, UNICODE_STRING* uni_name) {
+  static RtlInitUnicodeStringFunction RtlInitUnicodeString;
+  if (!RtlInitUnicodeString) {
+    HMODULE ntdll = ::GetModuleHandle(kNtdllName);
+    RtlInitUnicodeString = reinterpret_cast<RtlInitUnicodeStringFunction>(
+      GetProcAddress(ntdll, "RtlInitUnicodeString"));
+    DCHECK(RtlInitUnicodeString);
+  }
+  RtlInitUnicodeString(uni_name, name.c_str());
+  InitializeObjectAttributes(obj_attr, uni_name, attributes, root, NULL);
+}
+
+};  // namespace sandbox