| 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 "mojo/public/environment/standalone/buffer_tls_setup.h" | 7 #include <assert.h> |
| 8 #include "mojo/public/utility/thread_local.h" | 8 |
| 9 #include "mojo/public/environment/lib/buffer_tls_setup.h" |
| 10 #include "mojo/public/utility/lib/thread_local.h" |
| 9 | 11 |
| 10 namespace mojo { | 12 namespace mojo { |
| 11 namespace internal { | 13 namespace internal { |
| 12 | 14 |
| 13 static ThreadLocalPlatform::SlotType s_slot; | 15 static ThreadLocalPointer<Buffer> current_buffer; |
| 14 | 16 |
| 15 void SetUpCurrentBuffer() { | 17 void SetUpCurrentBuffer() { |
| 16 ThreadLocalPlatform::AllocateSlot(&s_slot); | 18 current_buffer.Allocate(); |
| 17 } | 19 } |
| 18 | 20 |
| 19 void TearDownCurrentBuffer() { | 21 void TearDownCurrentBuffer() { |
| 20 ThreadLocalPlatform::FreeSlot(s_slot); | 22 assert(!current_buffer.Get()); |
| 23 current_buffer.Free(); |
| 21 } | 24 } |
| 22 | 25 |
| 23 Buffer* GetCurrentBuffer() { | 26 Buffer* GetCurrentBuffer() { |
| 24 return static_cast<Buffer*>(ThreadLocalPlatform::GetValueFromSlot(s_slot)); | 27 return current_buffer.Get(); |
| 25 } | 28 } |
| 26 | 29 |
| 27 Buffer* SetCurrentBuffer(Buffer* buf) { | 30 Buffer* SetCurrentBuffer(Buffer* buf) { |
| 28 Buffer* old_buf = GetCurrentBuffer(); | 31 Buffer* old_buf = current_buffer.Get(); |
| 29 ThreadLocalPlatform::SetValueInSlot(s_slot, buf); | 32 current_buffer.Set(buf); |
| 30 return old_buf; | 33 return old_buf; |
| 31 } | 34 } |
| 32 | 35 |
| 33 } // namespace internal | 36 } // namespace internal |
| 34 } // namespace mojo | 37 } // namespace mojo |
| OLD | NEW |