tracing v2: Add TraceRingBuffer class.

components/tracing/core/trace_ring_buffer.cc

Index: components/tracing/core/trace_ring_buffer.cc
diff --git a/components/tracing/core/trace_ring_buffer.cc b/components/tracing/core/trace_ring_buffer.cc
new file mode 100644
index 0000000000000000000000000000000000000000..6afd7e19ce94ebd5b27217f0aaa62fe25eba1ef8
--- /dev/null
+++ b/components/tracing/core/trace_ring_buffer.cc
@@ -0,0 +1,78 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/tracing/core/trace_ring_buffer.h"
+
+#include "base/threading/platform_thread.h"
+
+namespace tracing {
+namespace v2 {
+
+const size_t TraceRingBuffer::Chunk::kSize = 32 * 1024;
+
+TraceRingBuffer::TraceRingBuffer(uint8_t* begin, size_t size)
+    : num_chunks_(size / Chunk::kSize),
+      current_chunk_idx_(0) {
+  DCHECK_GE(size, Chunk::kSize);
+  DCHECK_EQ(0ul, reinterpret_cast<uintptr_t>(begin) % sizeof(uintptr_t));
+  chunks_.reset(new Chunk[num_chunks_]);
+  uint8_t* chunk_begin = begin;
+  for (size_t i = 0; i < num_chunks_; ++i) {
+    uint8_t* const chunk_end = chunk_begin + Chunk::kSize;
+    chunks_[i].Initialize(chunk_begin, chunk_end);
+    chunk_begin = chunk_end;
+  }
+}
+
+TraceRingBuffer::~TraceRingBuffer() {}
+
+TraceRingBuffer::Chunk* TraceRingBuffer::TakeChunk() {
+  base::AutoLock lock(lock_);
+  DCHECK_GT(num_chunks_, 0ul);
+  DCHECK_LT(current_chunk_idx_, num_chunks_);
+  Chunk* chunk = nullptr;
+  for (size_t i = 0; i < num_chunks_; ++i) {
+    chunk = &chunks_[current_chunk_idx_];
+    current_chunk_idx_ = (current_chunk_idx_ + 1) % num_chunks_;
+    if (chunk->is_returned()) {
+      chunk->Clear();
+      chunk->set_owner(base::PlatformThread::CurrentId());
+      return chunk;
+    }
+  }
+
+  // Bankrupcy: there are more threads than chunks. All chunks were on flight.
+  if (!bankrupcy_chunk_storage_) {
+    bankrupcy_chunk_storage_.reset(new uint8_t[Chunk::kSize]);
+    uint8_t* const begin = &bankrupcy_chunk_storage_.get()[0];
+    bankrupcy_chunk_.Initialize(begin, begin + Chunk::kSize);
+  }
+  bankrupcy_chunk_.Clear();
+  return &bankrupcy_chunk_;
+}
+
+void TraceRingBuffer::ReturnChunk(TraceRingBuffer::Chunk* chunk,
+                                  uint32_t used_size) {
+  base::AutoLock lock(lock_);
+  chunk->set_proto_used_size(used_size);
+  chunk->set_returned();
+}
+
+TraceRingBuffer::Chunk::Chunk()
+    : begin_(nullptr), end_(nullptr), owner_(base::kInvalidThreadId) {}
+
+TraceRingBuffer::Chunk::~Chunk() {}
+
+void TraceRingBuffer::Chunk::Clear() {
+  set_proto_used_size(0);
+}
+
+void TraceRingBuffer::Chunk::Initialize(uint8_t* begin, uint8_t* end) {
+  DCHECK_EQ(0ul, reinterpret_cast<uintptr_t>(begin) % sizeof(uintptr_t));
+  begin_ = begin;
+  end_ = end;
+}
+
+}  // namespace v2
+}  // namespace tracing