| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2009 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "sandbox/src/broker_services.h" | 5 #include "sandbox/src/broker_services.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/threading/platform_thread.h" | 8 #include "base/threading/platform_thread.h" |
| 9 #include "sandbox/src/sandbox_policy_base.h" | 9 #include "sandbox/src/sandbox_policy_base.h" |
| 10 #include "sandbox/src/sandbox.h" | 10 #include "sandbox/src/sandbox.h" |
| 11 #include "sandbox/src/target_process.h" | 11 #include "sandbox/src/target_process.h" |
| (...skipping 165 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 177 case JOB_OBJECT_MSG_NEW_PROCESS: { | 177 case JOB_OBJECT_MSG_NEW_PROCESS: { |
| 178 ++target_counter; | 178 ++target_counter; |
| 179 if (1 == target_counter) { | 179 if (1 == target_counter) { |
| 180 ::ResetEvent(no_targets); | 180 ::ResetEvent(no_targets); |
| 181 } | 181 } |
| 182 break; | 182 break; |
| 183 } | 183 } |
| 184 | 184 |
| 185 case JOB_OBJECT_MSG_EXIT_PROCESS: | 185 case JOB_OBJECT_MSG_EXIT_PROCESS: |
| 186 case JOB_OBJECT_MSG_ABNORMAL_EXIT_PROCESS: { | 186 case JOB_OBJECT_MSG_ABNORMAL_EXIT_PROCESS: { |
| 187 { | |
| 188 AutoLock lock(&broker->lock_); | |
| 189 broker->child_process_ids_.erase(reinterpret_cast<DWORD>(ovl)); | |
| 190 } | |
| 191 --target_counter; | 187 --target_counter; |
| 192 if (0 == target_counter) | 188 if (0 == target_counter) |
| 193 ::SetEvent(no_targets); | 189 ::SetEvent(no_targets); |
| 194 | 190 |
| 195 DCHECK(target_counter >= 0); | 191 DCHECK(target_counter >= 0); |
| 196 break; | 192 break; |
| 197 } | 193 } |
| 198 | 194 |
| 199 case JOB_OBJECT_MSG_ACTIVE_PROCESS_LIMIT: { | 195 case JOB_OBJECT_MSG_ACTIVE_PROCESS_LIMIT: { |
| 200 break; | 196 break; |
| (...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 289 | 285 |
| 290 // We are going to keep a pointer to the policy because we'll call it when | 286 // We are going to keep a pointer to the policy because we'll call it when |
| 291 // the job object generates notifications using the completion port. | 287 // the job object generates notifications using the completion port. |
| 292 policy_base->AddRef(); | 288 policy_base->AddRef(); |
| 293 JobTracker* tracker = new JobTracker(job, policy_base); | 289 JobTracker* tracker = new JobTracker(job, policy_base); |
| 294 if (!AssociateCompletionPort(job, job_port_, tracker)) | 290 if (!AssociateCompletionPort(job, job_port_, tracker)) |
| 295 return SpawnCleanup(target, 0); | 291 return SpawnCleanup(target, 0); |
| 296 // Save the tracker because in cleanup we might need to force closing | 292 // Save the tracker because in cleanup we might need to force closing |
| 297 // the Jobs. | 293 // the Jobs. |
| 298 tracker_list_.push_back(tracker); | 294 tracker_list_.push_back(tracker); |
| 299 child_process_ids_.insert(process_info.dwProcessId); | |
| 300 | 295 |
| 301 // We return the caller a duplicate of the process handle so they | 296 // We return the caller a duplicate of the process handle so they |
| 302 // can close it at will. | 297 // can close it at will. |
| 303 HANDLE dup_process_handle = NULL; | 298 HANDLE dup_process_handle = NULL; |
| 304 if (!::DuplicateHandle(::GetCurrentProcess(), process_info.hProcess, | 299 if (!::DuplicateHandle(::GetCurrentProcess(), process_info.hProcess, |
| 305 ::GetCurrentProcess(), &dup_process_handle, | 300 ::GetCurrentProcess(), &dup_process_handle, |
| 306 0, FALSE, DUPLICATE_SAME_ACCESS)) | 301 0, FALSE, DUPLICATE_SAME_ACCESS)) |
| 307 return SpawnCleanup(target, 0); | 302 return SpawnCleanup(target, 0); |
| 308 | 303 |
| 309 *target_info = process_info; | 304 *target_info = process_info; |
| 310 target_info->hProcess = dup_process_handle; | 305 target_info->hProcess = dup_process_handle; |
| 311 return SBOX_ALL_OK; | 306 return SBOX_ALL_OK; |
| 312 } | 307 } |
| 313 | 308 |
| 314 | 309 |
| 315 ResultCode BrokerServicesBase::WaitForAllTargets() { | 310 ResultCode BrokerServicesBase::WaitForAllTargets() { |
| 316 ::WaitForSingleObject(no_targets_, INFINITE); | 311 ::WaitForSingleObject(no_targets_, INFINITE); |
| 317 return SBOX_ALL_OK; | 312 return SBOX_ALL_OK; |
| 318 } | 313 } |
| 319 | 314 |
| 320 bool BrokerServicesBase::IsActiveTarget(DWORD process_id) { | |
| 321 AutoLock lock(&lock_); | |
| 322 return child_process_ids_.find(process_id) != child_process_ids_.end(); | |
| 323 } | |
| 324 | |
| 325 } // namespace sandbox | 315 } // namespace sandbox |
| OLD | NEW |