Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(287)

Side by Side Diff: base/win/scoped_handle.h

Issue 10828160: [Chromoting] Move CreateSessionToken() next to launch process utilities. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: CR feedback. Created 8 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « no previous file | remoting/host/daemon_process_win.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 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 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 #ifndef BASE_WIN_SCOPED_HANDLE_H_ 5 #ifndef BASE_WIN_SCOPED_HANDLE_H_
6 #define BASE_WIN_SCOPED_HANDLE_H_ 6 #define BASE_WIN_SCOPED_HANDLE_H_
7 7
8 #include <windows.h> 8 #include <windows.h>
9 9
10 #include "base/base_export.h" 10 #include "base/base_export.h"
11 #include "base/basictypes.h" 11 #include "base/basictypes.h"
12 #include "base/location.h" 12 #include "base/location.h"
13 #include "base/logging.h" 13 #include "base/logging.h"
14 #include "base/move.h"
14 15
15 namespace base { 16 namespace base {
16 namespace win { 17 namespace win {
17 18
18 // TODO(rvargas): remove this with the rest of the verifier. 19 // TODO(rvargas): remove this with the rest of the verifier.
19 #if defined(COMPILER_MSVC) 20 #if defined(COMPILER_MSVC)
20 // MSDN says to #include <intrin.h>, but that breaks the VS2005 build. 21 // MSDN says to #include <intrin.h>, but that breaks the VS2005 build.
21 extern "C" { 22 extern "C" {
22 void* _ReturnAddress(); 23 void* _ReturnAddress();
23 } 24 }
24 #define BASE_WIN_GET_CALLER _ReturnAddress() 25 #define BASE_WIN_GET_CALLER _ReturnAddress()
25 #elif defined(COMPILER_GCC) 26 #elif defined(COMPILER_GCC)
26 #define BASE_WIN_GET_CALLER __builtin_extract_return_addr(\\ 27 #define BASE_WIN_GET_CALLER __builtin_extract_return_addr(\\
27 __builtin_return_address(0)) 28 __builtin_return_address(0))
28 #endif 29 #endif
29 30
30 // Generic wrapper for raw handles that takes care of closing handles 31 // Generic wrapper for raw handles that takes care of closing handles
31 // automatically. The class interface follows the style of 32 // automatically. The class interface follows the style of
32 // the ScopedStdioHandle class with a few additions: 33 // the ScopedStdioHandle class with a few additions:
33 // - IsValid() method can tolerate multiple invalid handle values such as NULL 34 // - IsValid() method can tolerate multiple invalid handle values such as NULL
34 // and INVALID_HANDLE_VALUE (-1) for Win32 handles. 35 // and INVALID_HANDLE_VALUE (-1) for Win32 handles.
35 // - Receive() method allows to receive a handle value from a function that 36 // - Receive() method allows to receive a handle value from a function that
36 // takes a raw handle pointer only. 37 // takes a raw handle pointer only.
37 template <class Traits, class Verifier> 38 template <class Traits, class Verifier>
38 class GenericScopedHandle { 39 class GenericScopedHandle {
40 MOVE_ONLY_TYPE_FOR_CPP_03(GenericScopedHandle, RValue)
41
39 public: 42 public:
40 typedef typename Traits::Handle Handle; 43 typedef typename Traits::Handle Handle;
41 44
42 GenericScopedHandle() : handle_(Traits::NullHandle()) {} 45 GenericScopedHandle() : handle_(Traits::NullHandle()) {}
43 46
44 explicit GenericScopedHandle(Handle handle) : handle_(Traits::NullHandle()) { 47 explicit GenericScopedHandle(Handle handle) : handle_(Traits::NullHandle()) {
45 Set(handle); 48 Set(handle);
46 } 49 }
47 50
51 // Move constructor for C++03 move emulation of this type.
52 GenericScopedHandle(RValue& other) : handle_(other.Take()) {
53 }
54
48 ~GenericScopedHandle() { 55 ~GenericScopedHandle() {
49 Close(); 56 Close();
50 } 57 }
51 58
52 bool IsValid() const { 59 bool IsValid() const {
53 return Traits::IsHandleValid(handle_); 60 return Traits::IsHandleValid(handle_);
54 } 61 }
55 62
63 // Move operator= for C++03 move emulation of this type.
64 GenericScopedHandle& operator=(RValue& other) {
65 // Swapping the handles helps to avoid problems while assigning a handle
jar (doing other things) 2012/08/08 01:42:52 The usual "trick" with dealing with self assignmen
alexeypa (please no reviews) 2012/08/08 01:59:30 src/base/move.h has a long explanation what it is
66 // to itself. It is also cheap and matches base::scoped_ptr behavior.
67 Swap(other);
68 return *this;
69 }
70
56 void Set(Handle handle) { 71 void Set(Handle handle) {
57 if (handle_ != handle) { 72 if (handle_ != handle) {
58 Close(); 73 Close();
59 74
60 if (Traits::IsHandleValid(handle)) { 75 if (Traits::IsHandleValid(handle)) {
61 handle_ = handle; 76 handle_ = handle;
62 Verifier::StartTracking(handle, this, BASE_WIN_GET_CALLER, 77 Verifier::StartTracking(handle, this, BASE_WIN_GET_CALLER,
63 tracked_objects::GetProgramCounter()); 78 tracked_objects::GetProgramCounter());
64 } 79 }
65 } 80 }
66 } 81 }
67 82
68 Handle Get() const { 83 Handle Get() const {
69 return handle_; 84 return handle_;
70 } 85 }
71 86
72 operator Handle() const { 87 operator Handle() const {
73 return handle_; 88 return handle_;
74 } 89 }
75 90
76 Handle* Receive() { 91 Handle* Receive() {
77 DCHECK(!Traits::IsHandleValid(handle_)) << "Handle must be NULL"; 92 DCHECK(!Traits::IsHandleValid(handle_)) << "Handle must be NULL";
78 93
79 // We cannot track this case :(. Just tell the verifier about it. 94 // We cannot track this case :(. Just tell the verifier about it.
80 Verifier::StartTracking(INVALID_HANDLE_VALUE, this, BASE_WIN_GET_CALLER, 95 Verifier::StartTracking(INVALID_HANDLE_VALUE, this, BASE_WIN_GET_CALLER,
81 tracked_objects::GetProgramCounter()); 96 tracked_objects::GetProgramCounter());
82 return &handle_; 97 return &handle_;
83 } 98 }
84 99
100 void Swap(GenericScopedHandle& other) {
101 Handle tmp = handle_;
102 handle_ = other.handle_;
103 other.handle_ = tmp;
104 }
105
85 // Transfers ownership away from this object. 106 // Transfers ownership away from this object.
86 Handle Take() { 107 Handle Take() {
87 Handle temp = handle_; 108 Handle temp = handle_;
88 handle_ = Traits::NullHandle(); 109 handle_ = Traits::NullHandle();
89 Verifier::StopTracking(temp, this, BASE_WIN_GET_CALLER, 110 Verifier::StopTracking(temp, this, BASE_WIN_GET_CALLER,
90 tracked_objects::GetProgramCounter()); 111 tracked_objects::GetProgramCounter());
91 return temp; 112 return temp;
92 } 113 }
93 114
94 // Explicitly closes the owned handle. 115 // Explicitly closes the owned handle.
95 void Close() { 116 void Close() {
96 if (Traits::IsHandleValid(handle_)) { 117 if (Traits::IsHandleValid(handle_)) {
97 Verifier::StopTracking(handle_, this, BASE_WIN_GET_CALLER, 118 Verifier::StopTracking(handle_, this, BASE_WIN_GET_CALLER,
98 tracked_objects::GetProgramCounter()); 119 tracked_objects::GetProgramCounter());
99 120
100 if (!Traits::CloseHandle(handle_)) 121 if (!Traits::CloseHandle(handle_))
101 CHECK(false); 122 CHECK(false);
102 123
103 handle_ = Traits::NullHandle(); 124 handle_ = Traits::NullHandle();
104 } 125 }
105 } 126 }
106 127
107 private: 128 private:
108 Handle handle_; 129 Handle handle_;
109
110 DISALLOW_COPY_AND_ASSIGN(GenericScopedHandle);
111 }; 130 };
112 131
113 #undef BASE_WIN_GET_CALLER 132 #undef BASE_WIN_GET_CALLER
114 133
115 // The traits class for Win32 handles that can be closed via CloseHandle() API. 134 // The traits class for Win32 handles that can be closed via CloseHandle() API.
116 class HandleTraits { 135 class HandleTraits {
117 public: 136 public:
118 typedef HANDLE Handle; 137 typedef HANDLE Handle;
119 138
120 // Closes the handle. 139 // Closes the handle.
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after
163 private: 182 private:
164 DISALLOW_IMPLICIT_CONSTRUCTORS(VerifierTraits); 183 DISALLOW_IMPLICIT_CONSTRUCTORS(VerifierTraits);
165 }; 184 };
166 185
167 typedef GenericScopedHandle<HandleTraits, VerifierTraits> ScopedHandle; 186 typedef GenericScopedHandle<HandleTraits, VerifierTraits> ScopedHandle;
168 187
169 } // namespace win 188 } // namespace win
170 } // namespace base 189 } // namespace base
171 190
172 #endif // BASE_SCOPED_HANDLE_WIN_H_ 191 #endif // BASE_SCOPED_HANDLE_WIN_H_
OLDNEW
« no previous file with comments | « no previous file | remoting/host/daemon_process_win.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698