Home | History | Annotate | Download | only in base 
      1 // Copyright (c) 2011 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 "net/base/io_buffer.h"
      6 
      7 #include "base/logging.h"
      8 
      9 namespace net {
     10 
     11 IOBuffer::IOBuffer()
     12     : data_(NULL) {
     13 }
     14 
     15 IOBuffer::IOBuffer(int buffer_size) {
     16   CHECK_GE(buffer_size, 0);
     17   data_ = new char[buffer_size];
     18 }
     19 
     20 IOBuffer::IOBuffer(char* data)
     21     : data_(data) {
     22 }
     23 
     24 IOBuffer::~IOBuffer() {
     25   delete[] data_;
     26   data_ = NULL;
     27 }
     28 
     29 IOBufferWithSize::IOBufferWithSize(int size)
     30     : IOBuffer(size),
     31       size_(size) {
     32 }
     33 
     34 IOBufferWithSize::IOBufferWithSize(char* data, int size)
     35     : IOBuffer(data),
     36       size_(size) {
     37 }
     38 
     39 IOBufferWithSize::~IOBufferWithSize() {
     40 }
     41 
     42 StringIOBuffer::StringIOBuffer(const std::string& s)
     43     : IOBuffer(static_cast<char*>(NULL)),
     44       string_data_(s) {
     45   CHECK_LT(s.size(), static_cast<size_t>(INT_MAX));
     46   data_ = const_cast<char*>(string_data_.data());
     47 }
     48 
     49 StringIOBuffer::StringIOBuffer(scoped_ptr<std::string> s)
     50     : IOBuffer(static_cast<char*>(NULL)) {
     51   CHECK_LT(s->size(), static_cast<size_t>(INT_MAX));
     52   string_data_.swap(*s.get());
     53   data_ = const_cast<char*>(string_data_.data());
     54 }
     55 
     56 StringIOBuffer::~StringIOBuffer() {
     57   // We haven't allocated the buffer, so remove it before the base class
     58   // destructor tries to delete[] it.
     59   data_ = NULL;
     60 }
     61 
     62 DrainableIOBuffer::DrainableIOBuffer(IOBuffer* base, int size)
     63     : IOBuffer(base->data()),
     64       base_(base),
     65       size_(size),
     66       used_(0) {
     67 }
     68 
     69 void DrainableIOBuffer::DidConsume(int bytes) {
     70   SetOffset(used_ + bytes);
     71 }
     72 
     73 int DrainableIOBuffer::BytesRemaining() const {
     74   return size_ - used_;
     75 }
     76 
     77 // Returns the number of consumed bytes.
     78 int DrainableIOBuffer::BytesConsumed() const {
     79   return used_;
     80 }
     81 
     82 void DrainableIOBuffer::SetOffset(int bytes) {
     83   DCHECK_GE(bytes, 0);
     84   DCHECK_LE(bytes, size_);
     85   used_ = bytes;
     86   data_ = base_->data() + used_;
     87 }
     88 
     89 DrainableIOBuffer::~DrainableIOBuffer() {
     90   // The buffer is owned by the |base_| instance.
     91   data_ = NULL;
     92 }
     93 
     94 GrowableIOBuffer::GrowableIOBuffer()
     95     : IOBuffer(),
     96       capacity_(0),
     97       offset_(0) {
     98 }
     99 
    100 void GrowableIOBuffer::SetCapacity(int capacity) {
    101   DCHECK_GE(capacity, 0);
    102   // realloc will crash if it fails.
    103   real_data_.reset(static_cast<char*>(realloc(real_data_.release(), capacity)));
    104   capacity_ = capacity;
    105   if (offset_ > capacity)
    106     set_offset(capacity);
    107   else
    108     set_offset(offset_);  // The pointer may have changed.
    109 }
    110 
    111 void GrowableIOBuffer::set_offset(int offset) {
    112   DCHECK_GE(offset, 0);
    113   DCHECK_LE(offset, capacity_);
    114   offset_ = offset;
    115   data_ = real_data_.get() + offset;
    116 }
    117 
    118 int GrowableIOBuffer::RemainingCapacity() {
    119   return capacity_ - offset_;
    120 }
    121 
    122 char* GrowableIOBuffer::StartOfBuffer() {
    123   return real_data_.get();
    124 }
    125 
    126 GrowableIOBuffer::~GrowableIOBuffer() {
    127   data_ = NULL;
    128 }
    129 
    130 PickledIOBuffer::PickledIOBuffer() : IOBuffer() {}
    131 
    132 void PickledIOBuffer::Done() {
    133   data_ = const_cast<char*>(static_cast<const char*>(pickle_.data()));
    134 }
    135 
    136 PickledIOBuffer::~PickledIOBuffer() { data_ = NULL; }
    137 
    138 WrappedIOBuffer::WrappedIOBuffer(const char* data)
    139     : IOBuffer(const_cast<char*>(data)) {
    140 }
    141 
    142 WrappedIOBuffer::~WrappedIOBuffer() {
    143   data_ = NULL;
    144 }
    145 
    146 }  // namespace net
    147