mmentovai@google.com | aa13be6 | 2008-09-03 03:20:34 +0900 | [diff] [blame] | 1 | // Copyright (c) 2006-2008 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 "base/at_exit.h" |
| 6 | #include "base/logging.h" |
| 7 | |
| 8 | namespace base { |
| 9 | |
| 10 | // Keep a stack of registered AtExitManagers. We always operate on the most |
| 11 | // recent, and we should never have more than one outside of testing, when we |
| 12 | // use the shadow version of the constructor. We don't protect this for |
| 13 | // thread-safe access, since it will only be modified in testing. |
| 14 | static AtExitManager* g_top_manager = NULL; |
| 15 | |
| 16 | AtExitManager::AtExitManager() : next_manager_(NULL) { |
| 17 | DCHECK(!g_top_manager); |
| 18 | g_top_manager = this; |
| 19 | } |
| 20 | |
| 21 | AtExitManager::AtExitManager(bool shadow) : next_manager_(g_top_manager) { |
| 22 | DCHECK(shadow || !g_top_manager); |
| 23 | g_top_manager = this; |
| 24 | } |
| 25 | |
| 26 | AtExitManager::~AtExitManager() { |
| 27 | if (!g_top_manager) { |
| 28 | NOTREACHED() << "Tried to ~AtExitManager without an AtExitManager"; |
| 29 | return; |
| 30 | } |
| 31 | DCHECK(g_top_manager == this); |
| 32 | |
| 33 | ProcessCallbacksNow(); |
| 34 | g_top_manager = next_manager_; |
| 35 | } |
| 36 | |
| 37 | // static |
deanm@google.com | f6299e7 | 2008-09-08 18:06:51 +0900 | [diff] [blame] | 38 | void AtExitManager::RegisterCallback(AtExitCallbackType func, void* param) { |
mmentovai@google.com | aa13be6 | 2008-09-03 03:20:34 +0900 | [diff] [blame] | 39 | if (!g_top_manager) { |
| 40 | NOTREACHED() << "Tried to RegisterCallback without an AtExitManager"; |
| 41 | return; |
| 42 | } |
| 43 | |
deanm@google.com | f6299e7 | 2008-09-08 18:06:51 +0900 | [diff] [blame] | 44 | DCHECK(func); |
| 45 | |
mmentovai@google.com | aa13be6 | 2008-09-03 03:20:34 +0900 | [diff] [blame] | 46 | AutoLock lock(g_top_manager->lock_); |
deanm@google.com | f6299e7 | 2008-09-08 18:06:51 +0900 | [diff] [blame] | 47 | g_top_manager->stack_.push(CallbackAndParam(func, param)); |
mmentovai@google.com | aa13be6 | 2008-09-03 03:20:34 +0900 | [diff] [blame] | 48 | } |
| 49 | |
| 50 | // static |
| 51 | void AtExitManager::ProcessCallbacksNow() { |
| 52 | if (!g_top_manager) { |
| 53 | NOTREACHED() << "Tried to ProcessCallbacksNow without an AtExitManager"; |
| 54 | return; |
| 55 | } |
| 56 | |
| 57 | AutoLock lock(g_top_manager->lock_); |
| 58 | |
| 59 | while (!g_top_manager->stack_.empty()) { |
deanm@google.com | f6299e7 | 2008-09-08 18:06:51 +0900 | [diff] [blame] | 60 | CallbackAndParam callback_and_param = g_top_manager->stack_.top(); |
mmentovai@google.com | aa13be6 | 2008-09-03 03:20:34 +0900 | [diff] [blame] | 61 | g_top_manager->stack_.pop(); |
deanm@google.com | f6299e7 | 2008-09-08 18:06:51 +0900 | [diff] [blame] | 62 | |
| 63 | callback_and_param.func_(callback_and_param.param_); |
mmentovai@google.com | aa13be6 | 2008-09-03 03:20:34 +0900 | [diff] [blame] | 64 | } |
| 65 | } |
| 66 | |
| 67 | } // namespace base |