| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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 #ifndef NET_BASE_IO_BUFFER_H_ | |
| 6 #define NET_BASE_IO_BUFFER_H_ | |
| 7 | |
| 8 #include <string> | |
| 9 | |
| 10 #include "base/memory/ref_counted.h" | |
| 11 #include "base/memory/scoped_ptr.h" | |
| 12 #include "base/pickle.h" | |
| 13 #include "net/base/net_export.h" | |
| 14 | |
| 15 namespace net { | |
| 16 | |
| 17 // IOBuffers are reference counted data buffers used for easier asynchronous | |
| 18 // IO handling. | |
| 19 // | |
| 20 // They are often used as the destination buffers for Read() operations, or as | |
| 21 // the source buffers for Write() operations. | |
| 22 // | |
| 23 // IMPORTANT: Never re-use an IOBuffer after cancelling the IO operation that | |
| 24 // was using it, since this may lead to memory corruption! | |
| 25 // | |
| 26 // ----------------------- | |
| 27 // Ownership of IOBuffers: | |
| 28 // ----------------------- | |
| 29 // | |
| 30 // Although IOBuffers are RefCountedThreadSafe, they are not intended to be | |
| 31 // used as a shared buffer, nor should they be used simultaneously across | |
| 32 // threads. The fact that they are reference counted is an implementation | |
| 33 // detail for allowing them to outlive cancellation of asynchronous | |
| 34 // operations. | |
| 35 // | |
| 36 // Instead, think of the underlying |char*| buffer contained by the IOBuffer | |
| 37 // as having exactly one owner at a time. | |
| 38 // | |
| 39 // Whenever you call an asynchronous operation that takes an IOBuffer, | |
| 40 // ownership is implicitly transferred to the called function, until the | |
| 41 // operation has completed (at which point it transfers back to the caller). | |
| 42 // | |
| 43 // ==> The IOBuffer's data should NOT be manipulated, destroyed, or read | |
| 44 // until the operation has completed. | |
| 45 // | |
| 46 // ==> Cancellation does NOT count as completion. If an operation using | |
| 47 // an IOBuffer is cancelled, the caller should release their | |
| 48 // reference to this IOBuffer at the time of cancellation since | |
| 49 // they can no longer use it. | |
| 50 // | |
| 51 // For instance, if you were to call a Read() operation on some class which | |
| 52 // takes an IOBuffer, and then delete that class (which generally will | |
| 53 // trigger cancellation), the IOBuffer which had been passed to Read() should | |
| 54 // never be re-used. | |
| 55 // | |
| 56 // This usage contract is assumed by any API which takes an IOBuffer, even | |
| 57 // though it may not be explicitly mentioned in the function's comments. | |
| 58 // | |
| 59 // ----------------------- | |
| 60 // Motivation | |
| 61 // ----------------------- | |
| 62 // | |
| 63 // The motivation for transferring ownership during cancellation is | |
| 64 // to make it easier to work with un-cancellable operations. | |
| 65 // | |
| 66 // For instance, let's say under the hood your API called out to the | |
| 67 // operating system's synchronous ReadFile() function on a worker thread. | |
| 68 // When cancelling through our asynchronous interface, we have no way of | |
| 69 // actually aborting the in progress ReadFile(). We must let it keep running, | |
| 70 // and hence the buffer it was reading into must remain alive. Using | |
| 71 // reference counting we can add a reference to the IOBuffer and make sure | |
| 72 // it is not destroyed until after the synchronous operation has completed. | |
| 73 class NET_EXPORT IOBuffer : public base::RefCountedThreadSafe<IOBuffer> { | |
| 74 public: | |
| 75 IOBuffer(); | |
| 76 explicit IOBuffer(int buffer_size); | |
| 77 | |
| 78 char* data() { return data_; } | |
| 79 | |
| 80 protected: | |
| 81 friend class base::RefCountedThreadSafe<IOBuffer>; | |
| 82 | |
| 83 // Only allow derived classes to specify data_. | |
| 84 // In all other cases, we own data_, and must delete it at destruction time. | |
| 85 explicit IOBuffer(char* data); | |
| 86 | |
| 87 virtual ~IOBuffer(); | |
| 88 | |
| 89 char* data_; | |
| 90 }; | |
| 91 | |
| 92 // This version stores the size of the buffer so that the creator of the object | |
| 93 // doesn't have to keep track of that value. | |
| 94 // NOTE: This doesn't mean that we want to stop sending the size as an explicit | |
| 95 // argument to IO functions. Please keep using IOBuffer* for API declarations. | |
| 96 class NET_EXPORT IOBufferWithSize : public IOBuffer { | |
| 97 public: | |
| 98 explicit IOBufferWithSize(int size); | |
| 99 | |
| 100 int size() const { return size_; } | |
| 101 | |
| 102 protected: | |
| 103 // Purpose of this constructor is to give a subclass access to the base class | |
| 104 // constructor IOBuffer(char*) thus allowing subclass to use underlying | |
| 105 // memory it does not own. | |
| 106 IOBufferWithSize(char* data, int size); | |
| 107 ~IOBufferWithSize() override; | |
| 108 | |
| 109 int size_; | |
| 110 }; | |
| 111 | |
| 112 // This is a read only IOBuffer. The data is stored in a string and | |
| 113 // the IOBuffer interface does not provide a proper way to modify it. | |
| 114 class NET_EXPORT StringIOBuffer : public IOBuffer { | |
| 115 public: | |
| 116 explicit StringIOBuffer(const std::string& s); | |
| 117 explicit StringIOBuffer(scoped_ptr<std::string> s); | |
| 118 | |
| 119 int size() const { return static_cast<int>(string_data_.size()); } | |
| 120 | |
| 121 private: | |
| 122 ~StringIOBuffer() override; | |
| 123 | |
| 124 std::string string_data_; | |
| 125 }; | |
| 126 | |
| 127 // This version wraps an existing IOBuffer and provides convenient functions | |
| 128 // to progressively read all the data. | |
| 129 // | |
| 130 // DrainableIOBuffer is useful when you have an IOBuffer that contains data | |
| 131 // to be written progressively, and Write() function takes an IOBuffer rather | |
| 132 // than char*. DrainableIOBuffer can be used as follows: | |
| 133 // | |
| 134 // // payload is the IOBuffer containing the data to be written. | |
| 135 // buf = new DrainableIOBuffer(payload, payload_size); | |
| 136 // | |
| 137 // while (buf->BytesRemaining() > 0) { | |
| 138 // // Write() takes an IOBuffer. If it takes char*, we could | |
| 139 // // simply use the regular IOBuffer like payload->data() + offset. | |
| 140 // int bytes_written = Write(buf, buf->BytesRemaining()); | |
| 141 // buf->DidConsume(bytes_written); | |
| 142 // } | |
| 143 // | |
| 144 class NET_EXPORT DrainableIOBuffer : public IOBuffer { | |
| 145 public: | |
| 146 DrainableIOBuffer(IOBuffer* base, int size); | |
| 147 | |
| 148 // DidConsume() changes the |data_| pointer so that |data_| always points | |
| 149 // to the first unconsumed byte. | |
| 150 void DidConsume(int bytes); | |
| 151 | |
| 152 // Returns the number of unconsumed bytes. | |
| 153 int BytesRemaining() const; | |
| 154 | |
| 155 // Returns the number of consumed bytes. | |
| 156 int BytesConsumed() const; | |
| 157 | |
| 158 // Seeks to an arbitrary point in the buffer. The notion of bytes consumed | |
| 159 // and remaining are updated appropriately. | |
| 160 void SetOffset(int bytes); | |
| 161 | |
| 162 int size() const { return size_; } | |
| 163 | |
| 164 private: | |
| 165 ~DrainableIOBuffer() override; | |
| 166 | |
| 167 scoped_refptr<IOBuffer> base_; | |
| 168 int size_; | |
| 169 int used_; | |
| 170 }; | |
| 171 | |
| 172 // This version provides a resizable buffer and a changeable offset. | |
| 173 // | |
| 174 // GrowableIOBuffer is useful when you read data progressively without | |
| 175 // knowing the total size in advance. GrowableIOBuffer can be used as | |
| 176 // follows: | |
| 177 // | |
| 178 // buf = new GrowableIOBuffer; | |
| 179 // buf->SetCapacity(1024); // Initial capacity. | |
| 180 // | |
| 181 // while (!some_stream->IsEOF()) { | |
| 182 // // Double the capacity if the remaining capacity is empty. | |
| 183 // if (buf->RemainingCapacity() == 0) | |
| 184 // buf->SetCapacity(buf->capacity() * 2); | |
| 185 // int bytes_read = some_stream->Read(buf, buf->RemainingCapacity()); | |
| 186 // buf->set_offset(buf->offset() + bytes_read); | |
| 187 // } | |
| 188 // | |
| 189 class NET_EXPORT GrowableIOBuffer : public IOBuffer { | |
| 190 public: | |
| 191 GrowableIOBuffer(); | |
| 192 | |
| 193 // realloc memory to the specified capacity. | |
| 194 void SetCapacity(int capacity); | |
| 195 int capacity() { return capacity_; } | |
| 196 | |
| 197 // |offset| moves the |data_| pointer, allowing "seeking" in the data. | |
| 198 void set_offset(int offset); | |
| 199 int offset() { return offset_; } | |
| 200 | |
| 201 int RemainingCapacity(); | |
| 202 char* StartOfBuffer(); | |
| 203 | |
| 204 private: | |
| 205 ~GrowableIOBuffer() override; | |
| 206 | |
| 207 scoped_ptr<char, base::FreeDeleter> real_data_; | |
| 208 int capacity_; | |
| 209 int offset_; | |
| 210 }; | |
| 211 | |
| 212 // This versions allows a pickle to be used as the storage for a write-style | |
| 213 // operation, avoiding an extra data copy. | |
| 214 class NET_EXPORT PickledIOBuffer : public IOBuffer { | |
| 215 public: | |
| 216 PickledIOBuffer(); | |
| 217 | |
| 218 Pickle* pickle() { return &pickle_; } | |
| 219 | |
| 220 // Signals that we are done writing to the pickle and we can use it for a | |
| 221 // write-style IO operation. | |
| 222 void Done(); | |
| 223 | |
| 224 private: | |
| 225 ~PickledIOBuffer() override; | |
| 226 | |
| 227 Pickle pickle_; | |
| 228 }; | |
| 229 | |
| 230 // This class allows the creation of a temporary IOBuffer that doesn't really | |
| 231 // own the underlying buffer. Please use this class only as a last resort. | |
| 232 // A good example is the buffer for a synchronous operation, where we can be | |
| 233 // sure that nobody is keeping an extra reference to this object so the lifetime | |
| 234 // of the buffer can be completely managed by its intended owner. | |
| 235 class NET_EXPORT WrappedIOBuffer : public IOBuffer { | |
| 236 public: | |
| 237 explicit WrappedIOBuffer(const char* data); | |
| 238 | |
| 239 protected: | |
| 240 ~WrappedIOBuffer() override; | |
| 241 }; | |
| 242 | |
| 243 } // namespace net | |
| 244 | |
| 245 #endif // NET_BASE_IO_BUFFER_H_ | |
| OLD | NEW |