common/platform/api/quiche_mem_slice.h

// Copyright 2017 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.

#ifndef QUICHE_COMMON_PLATFORM_API_QUICHE_MEM_SLICE_H_
#define QUICHE_COMMON_PLATFORM_API_QUICHE_MEM_SLICE_H_

#include <memory>

#include "absl/strings/string_view.h"
#include "common/platform/api/quiche_export.h"
#include "net/quiche/common/platform/impl/quiche_mem_slice_impl.h"

/* API_DESCRIPTION
 QuicheMemSlice is used to wrap application data and pass to QUIC stream's write
 interface. It refers to a memory block of data which should be around till
 QuicheMemSlice::Reset() is called. It's upto each platform, to implement it as
 reference counted or not.
 API-DESCRIPTION */

namespace quiche {

// QuicheMemSlice is an internally reference counted data buffer used as the
// source buffers for write operations. QuicheMemSlice implicitly maintains a
// reference count and will free the underlying data buffer when the reference
// count reaches zero.
class QUICHE_EXPORT_PRIVATE QuicheMemSlice {
 public:
  // Constructs a empty QuicheMemSlice with no underlying data and 0 reference
  // count.
  QuicheMemSlice() = default;

  // Constructs a QuicheMemSlice that takes ownership of |buffer|.  |length|
  // must not be zero.  To construct an empty QuicheMemSlice, use the
  // zero-argument constructor instead.
  // TODO(vasilvv): switch all users to QuicBuffer version, and make this
  // private.
  QuicheMemSlice(quic::QuicUniqueBufferPtr buffer, size_t length)
      : impl_(std::move(buffer), length) {}

  // Constructs a QuicheMemSlice that takes ownership of |buffer|.  The length
  // of the |buffer| must not be zero.  To construct an empty QuicheMemSlice,
  // use the zero-argument constructor instead.
  explicit QuicheMemSlice(quic::QuicBuffer buffer) : QuicheMemSlice() {
    // Store the size of the buffer *before* calling buffer.Release().
    const size_t size = buffer.size();
    *this = QuicheMemSlice(buffer.Release(), size);
  }

  // Constructs a QuicheMemSlice that takes ownership of |buffer| allocated on
  // heap.  |length| must not be zero.
  QuicheMemSlice(std::unique_ptr<char[]> buffer, size_t length)
      : impl_(std::move(buffer), length) {}

  // Constructs QuicheMemSlice from |impl|. It takes the reference away from
  // |impl|.
  explicit QuicheMemSlice(QuicheMemSliceImpl impl) : impl_(std::move(impl)) {}

  QuicheMemSlice(const QuicheMemSlice& other) = delete;
  QuicheMemSlice& operator=(const QuicheMemSlice& other) = delete;

  // Move constructors. |other| will not hold a reference to the data buffer
  // after this call completes.
  QuicheMemSlice(QuicheMemSlice&& other) = default;
  QuicheMemSlice& operator=(QuicheMemSlice&& other) = default;

  ~QuicheMemSlice() = default;

  // Release the underlying reference. Further access the memory will result in
  // undefined behavior.
  void Reset() { impl_.Reset(); }

  // Returns a const char pointer to underlying data buffer.
  const char* data() const { return impl_.data(); }
  // Returns the length of underlying data buffer.
  size_t length() const { return impl_.length(); }
  // Returns the representation of the underlying data as a string view.
  absl::string_view AsStringView() const {
    return absl::string_view(data(), length());
  }

  bool empty() const { return impl_.empty(); }

  QuicheMemSliceImpl* impl() { return &impl_; }

 private:
  QuicheMemSliceImpl impl_;
};

}  // namespace quiche

#endif  // QUICHE_COMMON_PLATFORM_API_QUICHE_MEM_SLICE_H_