| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "blimp/net/stream_packet_reader.h" | 5 #include "blimp/net/stream_packet_reader.h" |
| 6 | 6 |
| 7 #include <iostream> | 7 #include <iostream> |
| 8 | 8 |
| 9 #include "base/callback_helpers.h" | 9 #include "base/callback_helpers.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "base/memory/weak_ptr.h" | 11 #include "base/memory/weak_ptr.h" |
| 12 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
| 13 #include "base/sys_byteorder.h" | 13 #include "base/sys_byteorder.h" |
| 14 #include "blimp/net/blimp_connection_statistics.h" | |
| 15 #include "blimp/net/common.h" | 14 #include "blimp/net/common.h" |
| 16 #include "net/base/io_buffer.h" | 15 #include "net/base/io_buffer.h" |
| 17 #include "net/base/net_errors.h" | 16 #include "net/base/net_errors.h" |
| 18 #include "net/socket/stream_socket.h" | 17 #include "net/socket/stream_socket.h" |
| 19 | 18 |
| 20 namespace blimp { | 19 namespace blimp { |
| 21 | 20 |
| 22 std::ostream& operator<<(std::ostream& out, | 21 std::ostream& operator<<(std::ostream& out, |
| 23 const StreamPacketReader::ReadState state) { | 22 const StreamPacketReader::ReadState state) { |
| 24 switch (state) { | 23 switch (state) { |
| 25 case StreamPacketReader::ReadState::HEADER: | 24 case StreamPacketReader::ReadState::HEADER: |
| 26 out << "HEADER"; | 25 out << "HEADER"; |
| 27 break; | 26 break; |
| 28 case StreamPacketReader::ReadState::PAYLOAD: | 27 case StreamPacketReader::ReadState::PAYLOAD: |
| 29 out << "PAYLOAD"; | 28 out << "PAYLOAD"; |
| 30 break; | 29 break; |
| 31 case StreamPacketReader::ReadState::IDLE: | 30 case StreamPacketReader::ReadState::IDLE: |
| 32 out << "IDLE"; | 31 out << "IDLE"; |
| 33 break; | 32 break; |
| 34 } | 33 } |
| 35 return out; | 34 return out; |
| 36 } | 35 } |
| 37 | 36 |
| 38 StreamPacketReader::StreamPacketReader(net::StreamSocket* socket, | 37 StreamPacketReader::StreamPacketReader(net::StreamSocket* socket) |
| 39 BlimpConnectionStatistics* statistics) | 38 : read_state_(ReadState::IDLE), socket_(socket), weak_factory_(this) { |
| 40 : read_state_(ReadState::IDLE), | |
| 41 socket_(socket), | |
| 42 statistics_(statistics), | |
| 43 weak_factory_(this) { | |
| 44 DCHECK(socket_); | 39 DCHECK(socket_); |
| 45 header_buffer_ = new net::GrowableIOBuffer; | 40 header_buffer_ = new net::GrowableIOBuffer; |
| 46 header_buffer_->SetCapacity(kPacketHeaderSizeBytes); | 41 header_buffer_->SetCapacity(kPacketHeaderSizeBytes); |
| 47 } | 42 } |
| 48 | 43 |
| 49 StreamPacketReader::~StreamPacketReader() {} | 44 StreamPacketReader::~StreamPacketReader() {} |
| 50 | 45 |
| 51 void StreamPacketReader::ReadPacket( | 46 void StreamPacketReader::ReadPacket( |
| 52 const scoped_refptr<net::GrowableIOBuffer>& buf, | 47 const scoped_refptr<net::GrowableIOBuffer>& buf, |
| 53 const net::CompletionCallback& callback) { | 48 const net::CompletionCallback& callback) { |
| (...skipping 82 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 136 DCHECK_EQ(ReadState::PAYLOAD, read_state_); | 131 DCHECK_EQ(ReadState::PAYLOAD, read_state_); |
| 137 DCHECK_GE(result, 0); | 132 DCHECK_GE(result, 0); |
| 138 | 133 |
| 139 payload_buffer_->set_offset(payload_buffer_->offset() + result); | 134 payload_buffer_->set_offset(payload_buffer_->offset() + result); |
| 140 if (static_cast<size_t>(payload_buffer_->offset()) < payload_size_) { | 135 if (static_cast<size_t>(payload_buffer_->offset()) < payload_size_) { |
| 141 return socket_->Read(payload_buffer_.get(), | 136 return socket_->Read(payload_buffer_.get(), |
| 142 payload_size_ - payload_buffer_->offset(), | 137 payload_size_ - payload_buffer_->offset(), |
| 143 base::Bind(&StreamPacketReader::OnReadComplete, | 138 base::Bind(&StreamPacketReader::OnReadComplete, |
| 144 weak_factory_.GetWeakPtr())); | 139 weak_factory_.GetWeakPtr())); |
| 145 } | 140 } |
| 146 statistics_->Add(BlimpConnectionStatistics::BYTES_RECEIVED, payload_size_); | |
| 147 | 141 |
| 148 // Finished reading the payload. | 142 // Finished reading the payload. |
| 149 read_state_ = ReadState::IDLE; | 143 read_state_ = ReadState::IDLE; |
| 150 payload_buffer_->set_offset(0); | 144 payload_buffer_->set_offset(0); |
| 151 return payload_size_; | 145 return payload_size_; |
| 152 } | 146 } |
| 153 | 147 |
| 154 void StreamPacketReader::OnReadComplete(int result) { | 148 void StreamPacketReader::OnReadComplete(int result) { |
| 155 DCHECK_NE(net::ERR_IO_PENDING, result); | 149 DCHECK_NE(net::ERR_IO_PENDING, result); |
| 156 | 150 |
| (...skipping 11 matching lines...) Expand all Loading... |
| 168 // If all reading completed, either successfully or by error, inform the | 162 // If all reading completed, either successfully or by error, inform the |
| 169 // caller. | 163 // caller. |
| 170 if (result != net::ERR_IO_PENDING) { | 164 if (result != net::ERR_IO_PENDING) { |
| 171 payload_buffer_ = nullptr; | 165 payload_buffer_ = nullptr; |
| 172 base::ResetAndReturn(&callback_) | 166 base::ResetAndReturn(&callback_) |
| 173 .Run(result == net::OK ? payload_size_ : result); | 167 .Run(result == net::OK ? payload_size_ : result); |
| 174 } | 168 } |
| 175 } | 169 } |
| 176 | 170 |
| 177 } // namespace blimp | 171 } // namespace blimp |
| OLD | NEW |