| OLD | NEW |
| (Empty) |
| 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 #ifndef BASE_WIN_SCOPED_COM_INITIALIZER_H_ | |
| 6 #define BASE_WIN_SCOPED_COM_INITIALIZER_H_ | |
| 7 | |
| 8 #include <objbase.h> | |
| 9 | |
| 10 #include "base/basictypes.h" | |
| 11 #include "base/logging.h" | |
| 12 #include "build/build_config.h" | |
| 13 | |
| 14 namespace base { | |
| 15 namespace win { | |
| 16 | |
| 17 // Initializes COM in the constructor (STA or MTA), and uninitializes COM in the | |
| 18 // destructor. | |
| 19 // | |
| 20 // WARNING: This should only be used once per thread, ideally scoped to a | |
| 21 // similar lifetime as the thread itself. You should not be using this in | |
| 22 // random utility functions that make COM calls -- instead ensure these | |
| 23 // functions are running on a COM-supporting thread! | |
| 24 class ScopedCOMInitializer { | |
| 25 public: | |
| 26 // Enum value provided to initialize the thread as an MTA instead of STA. | |
| 27 enum SelectMTA { kMTA }; | |
| 28 | |
| 29 // Constructor for STA initialization. | |
| 30 ScopedCOMInitializer() { | |
| 31 Initialize(COINIT_APARTMENTTHREADED); | |
| 32 } | |
| 33 | |
| 34 // Constructor for MTA initialization. | |
| 35 explicit ScopedCOMInitializer(SelectMTA mta) { | |
| 36 Initialize(COINIT_MULTITHREADED); | |
| 37 } | |
| 38 | |
| 39 ~ScopedCOMInitializer() { | |
| 40 #ifndef NDEBUG | |
| 41 // Using the windows API directly to avoid dependency on platform_thread. | |
| 42 DCHECK_EQ(GetCurrentThreadId(), thread_id_); | |
| 43 #endif | |
| 44 if (succeeded()) | |
| 45 CoUninitialize(); | |
| 46 } | |
| 47 | |
| 48 bool succeeded() const { return SUCCEEDED(hr_); } | |
| 49 | |
| 50 private: | |
| 51 void Initialize(COINIT init) { | |
| 52 #ifndef NDEBUG | |
| 53 thread_id_ = GetCurrentThreadId(); | |
| 54 #endif | |
| 55 hr_ = CoInitializeEx(NULL, init); | |
| 56 #ifndef NDEBUG | |
| 57 if (hr_ == S_FALSE) | |
| 58 LOG(ERROR) << "Multiple CoInitialize() calls for thread " << thread_id_; | |
| 59 else | |
| 60 DCHECK_NE(RPC_E_CHANGED_MODE, hr_) << "Invalid COM thread model change"; | |
| 61 #endif | |
| 62 } | |
| 63 | |
| 64 HRESULT hr_; | |
| 65 #ifndef NDEBUG | |
| 66 // In debug builds we use this variable to catch a potential bug where a | |
| 67 // ScopedCOMInitializer instance is deleted on a different thread than it | |
| 68 // was initially created on. If that ever happens it can have bad | |
| 69 // consequences and the cause can be tricky to track down. | |
| 70 DWORD thread_id_; | |
| 71 #endif | |
| 72 | |
| 73 DISALLOW_COPY_AND_ASSIGN(ScopedCOMInitializer); | |
| 74 }; | |
| 75 | |
| 76 } // namespace win | |
| 77 } // namespace base | |
| 78 | |
| 79 #endif // BASE_WIN_SCOPED_COM_INITIALIZER_H_ | |
| OLD | NEW |