Add sandbox support for associating peer processes

sandbox/src/broker_services.h

 // 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_SRC_BROKER_SERVICES_H__
 #define SANDBOX_SRC_BROKER_SERVICES_H__
 
 #include <list>
 #include <set>
 #include "base/basictypes.h"
@@ skipping 27 matching lines @@
 
   virtual TargetPolicy* CreatePolicy();
 
   virtual ResultCode SpawnTarget(const wchar_t* exe_path,
                                  const wchar_t* command_line,
                                  TargetPolicy* policy,
                                  PROCESS_INFORMATION* target);
 
   virtual ResultCode WaitForAllTargets();
 
+  virtual ResultCode AddTargetPeer(HANDLE peer_process);
+
   // Checks if the supplied process ID matches one of the broker's active
   // target processes
   // Returns:
   //   true if there is an active target process for this ID, otherwise false.
   bool IsActiveTarget(DWORD process_id);
 
  private:
   // Helper structure that allows the Broker to associate a job notification
   // with a job object and with a policy.
   struct JobTracker {
     HANDLE job;
     PolicyBase* policy;
     JobTracker(HANDLE cjob, PolicyBase* cpolicy)
         : job(cjob), policy(cpolicy) {
      }
   };
 
   // Releases the Job and notifies the associated Policy object to its
   // resources as well.
   static void FreeResources(JobTracker* tracker);
 
   // The routine that the worker thread executes. It is in charge of
   // notifications and cleanup-related tasks.
   static DWORD WINAPI TargetEventsThread(PVOID param);
 
+  // Removes a target peer from the process list if it expires.
+  static VOID CALLBACK RemovePeer(PVOID process, BOOLEAN);
+
   // The completion port used by the job objects to communicate events to
   // the worker thread.
   HANDLE job_port_;
 
   // Handle to a manual-reset event that is signaled when the total target
   // process count reaches zero.
   HANDLE no_targets_;
 
   // Handle to the worker thread that reacts to job notifications.
   HANDLE job_thread_;
@@ skipping 12 matching lines @@
   // Provides a fast lookup to identify sandboxed processes.
   std::set<DWORD> child_process_ids_;
 
   DISALLOW_COPY_AND_ASSIGN(BrokerServicesBase);
 };
 
 }  // namespace sandbox
 
 
 #endif  // SANDBOX_SRC_BROKER_SERVICES_H__