src/mojo/public/cpp/environment/lib/buffer_tls.cc

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