123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142 |
- // Copyright (c) 2012 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 NET_SERVER_HTTP_CONNECTION_H_
- #define NET_SERVER_HTTP_CONNECTION_H_
- #include <memory>
- #include <string>
- #include "base/containers/queue.h"
- #include "base/memory/ref_counted.h"
- #include "net/base/io_buffer.h"
- namespace net {
- class StreamSocket;
- class WebSocket;
- // A container which has all information of an http connection. It includes
- // id, underlying socket, and pending read/write data.
- class HttpConnection {
- public:
- // IOBuffer for data read. It's a wrapper around GrowableIOBuffer, with more
- // functions for buffer management. It moves unconsumed data to the start of
- // buffer.
- class ReadIOBuffer : public IOBuffer {
- public:
- static const int kInitialBufSize = 1024;
- static const int kMinimumBufSize = 128;
- static const int kCapacityIncreaseFactor = 2;
- static const int kDefaultMaxBufferSize = 1 * 1024 * 1024; // 1 Mbytes.
- ReadIOBuffer();
- ReadIOBuffer(const ReadIOBuffer&) = delete;
- ReadIOBuffer& operator=(const ReadIOBuffer&) = delete;
- // Capacity.
- int GetCapacity() const;
- void SetCapacity(int capacity);
- // Increases capacity and returns true if capacity is not beyond the limit.
- bool IncreaseCapacity();
- // Start of read data.
- char* StartOfBuffer() const;
- // Returns the bytes of read data.
- int GetSize() const;
- // More read data was appended.
- void DidRead(int bytes);
- // Capacity for which more read data can be appended.
- int RemainingCapacity() const;
- // Removes consumed data and moves unconsumed data to the start of buffer.
- void DidConsume(int bytes);
- // Limit of how much internal capacity can increase.
- int max_buffer_size() const { return max_buffer_size_; }
- void set_max_buffer_size(int max_buffer_size) {
- max_buffer_size_ = max_buffer_size;
- }
- private:
- ~ReadIOBuffer() override;
- scoped_refptr<GrowableIOBuffer> base_;
- int max_buffer_size_ = kDefaultMaxBufferSize;
- };
- // IOBuffer of pending data to write which has a queue of pending data. Each
- // pending data is stored in std::string. data() is the data of first
- // std::string stored.
- class QueuedWriteIOBuffer : public IOBuffer {
- public:
- static const int kDefaultMaxBufferSize = 1 * 1024 * 1024; // 1 Mbytes.
- QueuedWriteIOBuffer();
- QueuedWriteIOBuffer(const QueuedWriteIOBuffer&) = delete;
- QueuedWriteIOBuffer& operator=(const QueuedWriteIOBuffer&) = delete;
- // Whether or not pending data exists.
- bool IsEmpty() const;
- // Appends new pending data and returns true if total size doesn't exceed
- // the limit, |total_size_limit_|. It would change data() if new data is
- // the first pending data.
- bool Append(const std::string& data);
- // Consumes data and changes data() accordingly. It cannot be more than
- // GetSizeToWrite().
- void DidConsume(int size);
- // Gets size of data to write this time. It is NOT total data size.
- int GetSizeToWrite() const;
- // Total size of all pending data.
- int total_size() const { return total_size_; }
- // Limit of how much data can be pending.
- int max_buffer_size() const { return max_buffer_size_; }
- void set_max_buffer_size(int max_buffer_size) {
- max_buffer_size_ = max_buffer_size;
- }
- private:
- ~QueuedWriteIOBuffer() override;
- // This needs to indirect since we need pointer stability for the payload
- // chunks, as they may be handed out via net::IOBuffer::data().
- base::queue<std::unique_ptr<std::string>> pending_data_;
- int total_size_ = 0;
- int max_buffer_size_ = kDefaultMaxBufferSize;
- };
- HttpConnection(int id, std::unique_ptr<StreamSocket> socket);
- HttpConnection(const HttpConnection&) = delete;
- HttpConnection& operator=(const HttpConnection&) = delete;
- ~HttpConnection();
- int id() const { return id_; }
- StreamSocket* socket() const { return socket_.get(); }
- ReadIOBuffer* read_buf() const { return read_buf_.get(); }
- QueuedWriteIOBuffer* write_buf() const { return write_buf_.get(); }
- WebSocket* web_socket() const { return web_socket_.get(); }
- void SetWebSocket(std::unique_ptr<WebSocket> web_socket);
- private:
- const int id_;
- const std::unique_ptr<StreamSocket> socket_;
- const scoped_refptr<ReadIOBuffer> read_buf_;
- const scoped_refptr<QueuedWriteIOBuffer> write_buf_;
- std::unique_ptr<WebSocket> web_socket_;
- };
- } // namespace net
- #endif // NET_SERVER_HTTP_CONNECTION_H_
|