Add a sandbox API for broker handle duplication

sandbox/tests/common/controller.h

-// Copyright (c) 2006-2010 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.
 
 #ifndef SANDBOX_TESTS_COMMON_CONTROLLER_H_
 #define SANDBOX_TESTS_COMMON_CONTROLLER_H__
 
 #include <windows.h>
 #include <string>
 
+#include "base/win/scoped_handle.h"
 #include "sandbox/src/sandbox.h"
 
 namespace sandbox {
 
 // See winerror.h for details.
 #define SEVERITY_INFO_FLAGS   0x40000000
 #define SEVERITY_ERROR_FLAGS  0xC0000000
 #define CUSTOMER_CODE         0x20000000
 #define SBOX_TESTS_FACILITY   0x05B10000
 
@@ skipping 63 matching lines @@
   // succeeds.
   bool AddFsRule(TargetPolicy::Semantics semantics, const wchar_t* pattern);
 
   // Starts a child process in the sandbox and ask it to run |command|. Returns
   // a SboxTestResult. By default, the test runs AFTER_REVERT.
   int RunTest(const wchar_t* command);
 
   // Sets the timeout value for the child to run the command and return.
   void SetTimeout(DWORD timeout_ms);
 
+  // Sets TestRunner to return without waiting for the process to exit.
+  void SetAsynchronous(bool is_async) { is_async_ = is_async; }
+
   // Sets the desired state for the test to run.
   void SetTestState(SboxTestsState desired_state);
 
   // Returns the pointers to the policy object. It can be used to modify
   // the policy manually.
   TargetPolicy* GetPolicy();
 
+  // Return the process handle for an asynchronous test.
+  HANDLE process() { return target_process_; }
+
+  // Return the process ID for an asynchronous test.
+  DWORD process_id() { return target_process_id_; }
+
  private:
   // Initializes the data in the object. Sets is_init_ to tree if the
   // function succeeds. This is meant to be called from the constructor.
   void Init(JobLevel job_level, TokenLevel startup_token,
             TokenLevel main_token);
 
   // The actual runner.
   int InternalRunTest(const wchar_t* command);
 
   BrokerServices* broker_;
   TargetPolicy* policy_;
   DWORD timeout_;
   SboxTestsState state_;
   bool is_init_;
+  bool is_async_;
+  base::win::ScopedHandle target_process_;
+  DWORD target_process_id_;
 };
 
 // Returns the broker services.
 BrokerServices* GetBroker();
 
 // Constructs a full path to a file inside the system32 (or syswow64) folder.
 std::wstring MakePathToSys(const wchar_t* name, bool is_obj_man_path);
 
 // Runs the given test on the target process.
 int DispatchCall(int argc, wchar_t **argv);
 
 }  // namespace sandbox
 
 #endif  // SANDBOX_TESTS_COMMON_CONTROLLER_H_