OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #include "mojo/public/environment/buffer_tls.h" | 5 #include "mojo/public/environment/buffer_tls.h" |
6 | 6 |
7 #include <assert.h> | 7 #include <assert.h> |
8 | 8 |
| 9 #include "mojo/public/cpp/utility/lib/thread_local.h" |
9 #include "mojo/public/environment/lib/buffer_tls_setup.h" | 10 #include "mojo/public/environment/lib/buffer_tls_setup.h" |
10 #include "mojo/public/utility/lib/thread_local.h" | |
11 | 11 |
12 namespace mojo { | 12 namespace mojo { |
13 namespace internal { | 13 namespace internal { |
14 | 14 |
15 static ThreadLocalPointer<Buffer> current_buffer; | 15 static ThreadLocalPointer<Buffer> current_buffer; |
16 | 16 |
17 void SetUpCurrentBuffer() { | 17 void SetUpCurrentBuffer() { |
18 current_buffer.Allocate(); | 18 current_buffer.Allocate(); |
19 } | 19 } |
20 | 20 |
21 void TearDownCurrentBuffer() { | 21 void TearDownCurrentBuffer() { |
22 assert(!current_buffer.Get()); | 22 assert(!current_buffer.Get()); |
23 current_buffer.Free(); | 23 current_buffer.Free(); |
24 } | 24 } |
25 | 25 |
26 Buffer* GetCurrentBuffer() { | 26 Buffer* GetCurrentBuffer() { |
27 return current_buffer.Get(); | 27 return current_buffer.Get(); |
28 } | 28 } |
29 | 29 |
30 Buffer* SetCurrentBuffer(Buffer* buf) { | 30 Buffer* SetCurrentBuffer(Buffer* buf) { |
31 Buffer* old_buf = current_buffer.Get(); | 31 Buffer* old_buf = current_buffer.Get(); |
32 current_buffer.Set(buf); | 32 current_buffer.Set(buf); |
33 return old_buf; | 33 return old_buf; |
34 } | 34 } |
35 | 35 |
36 } // namespace internal | 36 } // namespace internal |
37 } // namespace mojo | 37 } // namespace mojo |
OLD | NEW |