sandbox/win/src/win2k_threadpool.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "sandbox/win/src/win2k_threadpool.h"
   6
   7 #include "sandbox/win/src/win_utils.h"
   8
   9 namespace sandbox {
  10
  11 bool Win2kThreadPool::RegisterWait(const void* cookie, HANDLE waitable_object,
  12                                    CrossCallIPCCallback callback,
  13                                    void* context) {
  14   if (0 == cookie) {
  15     return false;
  16   }
  17   HANDLE pool_object = NULL;
  18   // create a wait for a kernel object, with no timeout
  19   if (!::RegisterWaitForSingleObject(&pool_object, waitable_object, callback,
  20                                      context, INFINITE, WT_EXECUTEDEFAULT)) {
  21     return false;
  22   }
  23   PoolObject pool_obj = {cookie, pool_object};
  24   AutoLock lock(&lock_);
  25   pool_objects_.push_back(pool_obj);
  26   return true;
  27 }
  28
  29 bool Win2kThreadPool::UnRegisterWaits(void* cookie) {
  30   if (0 == cookie) {
  31     return false;
  32   }
  33   AutoLock lock(&lock_);
  34   bool success = true;
  35   PoolObjects::iterator it = pool_objects_.begin();
  36   while (it != pool_objects_.end()) {
  37     if (it->cookie == cookie) {
  38       HANDLE wait = it->wait;
  39       it = pool_objects_.erase(it);
  40       success &= (::UnregisterWaitEx(wait, INVALID_HANDLE_VALUE) != 0);
  41     } else {
  42       ++it;
  43     }
  44   }
  45   return success;
  46 }
  47
  48 size_t Win2kThreadPool::OutstandingWaits() {
  49   AutoLock lock(&lock_);
  50   return pool_objects_.size();
  51 }
  52
  53 Win2kThreadPool::~Win2kThreadPool() {
  54   // Here we used to unregister all the pool wait handles. Now, following the
  55   // rest of the code we avoid lengthy or blocking calls given that the process
  56   // is being torn down.
  57   ::DeleteCriticalSection(&lock_);
  58 }
  59
  60 }  // namespace sandbox