Sandbox: Use ScopedProcessInformation.

sandbox/src/interception_unittest.cc

-// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // This file contains unit tests for InterceptionManager.
 // The tests require private information so the whole interception.cc file is
 // included from this file.
 
 #include <windows.h>
 
 #include "base/memory/scoped_ptr.h"
@@ skipping 60 matching lines @@
     (*num_dlls)++;
     dll = reinterpret_cast<DllPatchInfo*>(reinterpret_cast<char*>(dll) +
                                           dll->record_bytes);
   }
 }
 
 TEST(InterceptionManagerTest, BufferLayout1) {
   wchar_t exe_name[MAX_PATH];
   ASSERT_NE(0u, GetModuleFileName(NULL, exe_name, MAX_PATH - 1));
 
-  TargetProcess *target = MakeTestTargetProcess(::GetCurrentProcess(),
+  base::win::ScopedHandle current_process;
+  ASSERT_TRUE(
+      ::DuplicateHandle(::GetCurrentProcess(), ::GetCurrentProcess(),
+                        ::GetCurrentProcess(), current_process.Receive(),
+                        0, FALSE, DUPLICATE_SAME_ACCESS));
+
+  TargetProcess *target = MakeTestTargetProcess(current_process.Take(),

[rvargas (doing something else), 2012/06/26 20:55:07]

reverted changes to this file.

                                                 ::GetModuleHandle(exe_name));
 
   InterceptionManager interceptions(target, true);
 
   // Any pointer will do for a function pointer.
   void* function = &interceptions;
 
   // We don't care about the interceptor id.
   interceptions.AddToPatchedFunctions(L"ntdll.dll", "NtCreateFile",
                                       INTERCEPTION_SERVICE_CALL, function,
@@ skipping 66 matching lines @@
   // interceptor function.
   EXPECT_EQ(6, num_dlls);
   EXPECT_EQ(15, num_functions);
   EXPECT_EQ(4, num_names);
 }
 
 TEST(InterceptionManagerTest, BufferLayout2) {
   wchar_t exe_name[MAX_PATH];
   ASSERT_NE(0u, GetModuleFileName(NULL, exe_name, MAX_PATH - 1));
 
-  TargetProcess *target = MakeTestTargetProcess(::GetCurrentProcess(),
+  base::win::ScopedHandle current_process;
+  ASSERT_TRUE(
+      ::DuplicateHandle(::GetCurrentProcess(), ::GetCurrentProcess(),
+                        ::GetCurrentProcess(), current_process.Receive(),
+                        0, FALSE, DUPLICATE_SAME_ACCESS));
+
+  TargetProcess *target = MakeTestTargetProcess(current_process.Take(),
                                                 ::GetModuleHandle(exe_name));
 
   InterceptionManager interceptions(target, true);
 
   // Any pointer will do for a function pointer.
   void* function = &interceptions;
   interceptions.AddToUnloadModules(L"some01.dll");
   // We don't care about the interceptor id.
   interceptions.AddToPatchedFunctions(L"ntdll.dll", "NtCreateFile",
                                       INTERCEPTION_SERVICE_CALL, function,
@@ skipping 23 matching lines @@
   int num_dlls, num_functions, num_names;
   WalkBuffer(local_buffer.get(), buffer_size, &num_dlls, &num_functions,
              &num_names);
 
   EXPECT_EQ(3, num_dlls);
   EXPECT_EQ(4, num_functions);
   EXPECT_EQ(0, num_names);
 }
 
 }  // namespace sandbox