OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/rtc_base/thread_annotations.h" | 11 #include "webrtc/rtc_base/thread_annotations.h" |
12 #include "webrtc/test/gtest.h" | 12 #include "webrtc/test/gtest.h" |
13 | 13 |
14 namespace { | 14 namespace { |
15 | 15 |
16 class RTC_LOCKABLE Lock { | 16 class LOCKABLE Lock { |
17 public: | 17 public: |
18 void EnterWrite() const RTC_EXCLUSIVE_LOCK_FUNCTION() {} | 18 void EnterWrite() const EXCLUSIVE_LOCK_FUNCTION() {} |
19 void EnterRead() const RTC_SHARED_LOCK_FUNCTION() {} | 19 void EnterRead() const SHARED_LOCK_FUNCTION() {} |
20 bool TryEnterWrite() const RTC_EXCLUSIVE_TRYLOCK_FUNCTION(true) { | 20 bool TryEnterWrite() const EXCLUSIVE_TRYLOCK_FUNCTION(true) { return true; } |
21 return true; | 21 bool TryEnterRead() const SHARED_TRYLOCK_FUNCTION(true) { return true; } |
22 } | 22 void Leave() const UNLOCK_FUNCTION() {} |
23 bool TryEnterRead() const RTC_SHARED_TRYLOCK_FUNCTION(true) { return true; } | |
24 void Leave() const RTC_UNLOCK_FUNCTION() {} | |
25 }; | 23 }; |
26 | 24 |
27 class RTC_SCOPED_LOCKABLE ScopeLock { | 25 class SCOPED_LOCKABLE ScopeLock { |
28 public: | 26 public: |
29 explicit ScopeLock(const Lock& lock) RTC_EXCLUSIVE_LOCK_FUNCTION(lock) {} | 27 explicit ScopeLock(const Lock& lock) EXCLUSIVE_LOCK_FUNCTION(lock) {} |
30 ~ScopeLock() RTC_UNLOCK_FUNCTION() {} | 28 ~ScopeLock() UNLOCK_FUNCTION() {} |
31 }; | 29 }; |
32 | 30 |
33 class ThreadSafe { | 31 class ThreadSafe { |
34 public: | 32 public: |
35 ThreadSafe() { | 33 ThreadSafe() { |
36 pt_protected_by_lock_ = new int; | 34 pt_protected_by_lock_ = new int; |
37 pt_protected_by_anything_ = new int; | 35 pt_protected_by_anything_ = new int; |
38 } | 36 } |
39 | 37 |
40 ~ThreadSafe() { | 38 ~ThreadSafe() { |
(...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
116 unprotected_ = protected_by_lock_; | 114 unprotected_ = protected_by_lock_; |
117 unprotected_ = *pt_protected_by_lock_; | 115 unprotected_ = *pt_protected_by_lock_; |
118 } | 116 } |
119 | 117 |
120 void WriteProtectedFunction() EXCLUSIVE_LOCKS_REQUIRED(lock_, pt_lock_) { | 118 void WriteProtectedFunction() EXCLUSIVE_LOCKS_REQUIRED(lock_, pt_lock_) { |
121 int x = protected_by_lock_; | 119 int x = protected_by_lock_; |
122 *pt_protected_by_lock_ = x; | 120 *pt_protected_by_lock_ = x; |
123 protected_by_lock_ = unprotected_; | 121 protected_by_lock_ = unprotected_; |
124 } | 122 } |
125 | 123 |
126 const Lock& GetLock() RTC_LOCK_RETURNED(lock_) { return lock_; } | 124 const Lock& GetLock() LOCK_RETURNED(lock_) { return lock_; } |
127 | 125 |
128 Lock anylock_ ACQUIRED_BEFORE(lock_); | 126 Lock anylock_ ACQUIRED_BEFORE(lock_); |
129 Lock lock_; | 127 Lock lock_; |
130 Lock pt_lock_ ACQUIRED_AFTER(lock_); | 128 Lock pt_lock_ ACQUIRED_AFTER(lock_); |
131 | 129 |
132 int unprotected_ = 0; | 130 int unprotected_ = 0; |
133 | 131 |
134 int protected_by_lock_ GUARDED_BY(lock_) = 0; | 132 int protected_by_lock_ GUARDED_BY(lock_) = 0; |
135 int protected_by_anything_ RTC_GUARDED_VAR = 0; | 133 int protected_by_anything_ GUARDED_VAR = 0; |
136 | 134 |
137 int* pt_protected_by_lock_ RTC_PT_GUARDED_BY(pt_lock_); | 135 int* pt_protected_by_lock_ PT_GUARDED_BY(pt_lock_); |
138 int* pt_protected_by_anything_ RTC_PT_GUARDED_VAR; | 136 int* pt_protected_by_anything_ PT_GUARDED_VAR; |
139 }; | 137 }; |
140 | 138 |
141 } // namespace | 139 } // namespace |
142 | 140 |
143 TEST(ThreadAnnotationsTest, Test) { | 141 TEST(ThreadAnnotationsTest, Test) { |
144 // This test ensure thread annotations doesn't break compilation. | 142 // This test ensure thread annotations doesn't break compilation. |
145 // Thus no run-time expectations. | 143 // Thus no run-time expectations. |
146 ThreadSafe t; | 144 ThreadSafe t; |
147 t.LockInOrder(); | 145 t.LockInOrder(); |
148 t.UnprotectedFunction(); | 146 t.UnprotectedFunction(); |
149 t.ReadProtected(); | 147 t.ReadProtected(); |
150 t.WriteProtected(); | 148 t.WriteProtected(); |
151 t.CallReadProtectedFunction(); | 149 t.CallReadProtectedFunction(); |
152 t.CallWriteProtectedFunction(); | 150 t.CallWriteProtectedFunction(); |
153 } | 151 } |
OLD | NEW |