15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/src/win2k_threadpool.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/src/win_utils.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace sandbox { 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Win2kThreadPool::RegisterWait(const void* cookie, HANDLE waitable_object, 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CrossCallIPCCallback callback, 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void* context) { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (0 == cookie) { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE pool_object = NULL; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // create a wait for a kernel object, with no timeout 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!::RegisterWaitForSingleObject(&pool_object, waitable_object, callback, 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context, INFINITE, WT_EXECUTEDEFAULT)) { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PoolObject pool_obj = {cookie, pool_object}; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AutoLock lock(&lock_); 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pool_objects_.push_back(pool_obj); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Win2kThreadPool::UnRegisterWaits(void* cookie) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (0 == cookie) { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AutoLock lock(&lock_); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool success = true; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PoolObjects::iterator it = pool_objects_.begin(); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (it != pool_objects_.end()) { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it->cookie == cookie) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE wait = it->wait; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it = pool_objects_.erase(it); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) success &= (::UnregisterWaitEx(wait, INVALID_HANDLE_VALUE) != 0); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++it; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return success; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t Win2kThreadPool::OutstandingWaits() { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AutoLock lock(&lock_); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return pool_objects_.size(); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Win2kThreadPool::~Win2kThreadPool() { 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Here we used to unregister all the pool wait handles. Now, following the 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // rest of the code we avoid lengthy or blocking calls given that the process 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // is being torn down. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteCriticalSection(&lock_); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace sandbox 61