Add CopyOnWriteBuffer class

webrtc/base/buffer.cc

 /*
  *  Copyright 2015 The WebRTC Project Authors. All rights reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #include "webrtc/base/buffer.h"
 
 #include <cassert>
 #include <utility>
 
 namespace rtc {
 
-Buffer::Buffer() : size_(0), capacity_(0), data_(nullptr) {
+BufferData::BufferData()
+    : ref_count_(1),
+      size_(0),
+      capacity_(0),
+      data_(nullptr) {
   assert(IsConsistent());
 }
 
-Buffer::Buffer(const Buffer& buf) : Buffer(buf.data(), buf.size()) {
+BufferData::BufferData(const BufferData& data)
+    : ref_count_(1),
+      size_(data.size_),
+      capacity_(data.capacity_),
+      data_(new uint8_t[capacity_]) {
+  assert(IsConsistent());
+  if (data.size_) {
+    std::memcpy(data_.get(), data.data_.get(), data.size_);
+  }
 }
 
-Buffer::Buffer(Buffer&& buf)
-    : size_(buf.size()),
-      capacity_(buf.capacity()),
-      data_(std::move(buf.data_)) {
+BufferData::BufferData(size_t size) : BufferData(size, size) {
+}
+
+BufferData::BufferData(size_t size, size_t capacity)
+    : ref_count_(1),
+      size_(size),
+      capacity_(std::max(size, capacity)),
+      data_(new uint8_t[capacity_]) {
   assert(IsConsistent());
+}
+
+BufferData::BufferData(const void* data, size_t size)
+    : BufferData(data, size, size) {
+}
+
+BufferData::BufferData(const void* data, size_t size, size_t capacity)
+    : BufferData(size, capacity) {
+  std::memcpy(data_.get(), data, size);
+}
+
+BufferData::~BufferData() {
+}
+
+
+Buffer::Buffer() : data_(new BufferData()) {
+}
+
+Buffer::Buffer(const Buffer& buf) : data_(buf.data_) {
+  data_->AddRef();
+}
+
+Buffer::Buffer(Buffer&& buf) : data_(std::move(buf.data_)) {
   buf.OnMovedFrom();
 }
 
 Buffer::Buffer(size_t size) : Buffer(size, size) {
 }
 
 Buffer::Buffer(size_t size, size_t capacity)
-    : size_(size),
-      capacity_(std::max(size, capacity)),
-      data_(new uint8_t[capacity_]) {
+    : data_(new BufferData(size, capacity)) {
   assert(IsConsistent());
 }
 
-// Note: The destructor works even if the buffer has been moved from.
-Buffer::~Buffer() = default;
+Buffer::~Buffer() {
+  data_->Release();
+}
 
 };  // namespace rtc