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/location.h" | 10 #include "base/location.h" |
(...skipping 20 matching lines...) Expand all Loading... |
31 out << "PAYLOAD"; | 31 out << "PAYLOAD"; |
32 break; | 32 break; |
33 case StreamPacketReader::ReadState::IDLE: | 33 case StreamPacketReader::ReadState::IDLE: |
34 out << "IDLE"; | 34 out << "IDLE"; |
35 break; | 35 break; |
36 } | 36 } |
37 return out; | 37 return out; |
38 } | 38 } |
39 | 39 |
40 StreamPacketReader::StreamPacketReader(net::StreamSocket* socket) | 40 StreamPacketReader::StreamPacketReader(net::StreamSocket* socket) |
41 : read_state_(ReadState::IDLE), socket_(socket), weak_factory_(this) { | 41 : read_state_(ReadState::IDLE), |
| 42 socket_(socket), |
| 43 payload_size_(0), |
| 44 weak_factory_(this) { |
42 DCHECK(socket_); | 45 DCHECK(socket_); |
43 header_buffer_ = new net::GrowableIOBuffer; | 46 header_buffer_ = new net::GrowableIOBuffer; |
44 header_buffer_->SetCapacity(kPacketHeaderSizeBytes); | 47 header_buffer_->SetCapacity(kPacketHeaderSizeBytes); |
45 } | 48 } |
46 | 49 |
47 StreamPacketReader::~StreamPacketReader() {} | 50 StreamPacketReader::~StreamPacketReader() {} |
48 | 51 |
49 void StreamPacketReader::ReadPacket( | 52 void StreamPacketReader::ReadPacket( |
50 const scoped_refptr<net::GrowableIOBuffer>& buf, | 53 const scoped_refptr<net::GrowableIOBuffer>& buf, |
51 const net::CompletionCallback& callback) { | 54 const net::CompletionCallback& callback) { |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
146 | 149 |
147 void StreamPacketReader::OnReadComplete(int result) { | 150 void StreamPacketReader::OnReadComplete(int result) { |
148 DCHECK_NE(net::ERR_IO_PENDING, result); | 151 DCHECK_NE(net::ERR_IO_PENDING, result); |
149 | 152 |
150 if (result == 0 /* EOF */) { | 153 if (result == 0 /* EOF */) { |
151 payload_buffer_ = nullptr; | 154 payload_buffer_ = nullptr; |
152 base::ResetAndReturn(&callback_).Run(net::ERR_CONNECTION_CLOSED); | 155 base::ResetAndReturn(&callback_).Run(net::ERR_CONNECTION_CLOSED); |
153 return; | 156 return; |
154 } | 157 } |
155 | 158 |
156 // If the read was succesful, then process the result. | 159 // If the read was successful, then process the result. |
157 if (result > 0) { | 160 if (result > 0) { |
158 result = DoReadLoop(result); | 161 result = DoReadLoop(result); |
159 } | 162 } |
160 | 163 |
161 // If all reading completed, either successfully or by error, inform the | 164 // If all reading completed, either successfully or by error, inform the |
162 // caller. | 165 // caller. |
163 if (result != net::ERR_IO_PENDING) { | 166 if (result != net::ERR_IO_PENDING) { |
164 payload_buffer_ = nullptr; | 167 payload_buffer_ = nullptr; |
165 base::ResetAndReturn(&callback_).Run(result); | 168 base::ResetAndReturn(&callback_).Run(result); |
166 } | 169 } |
167 } | 170 } |
168 | 171 |
169 int StreamPacketReader::DoRead(net::IOBuffer* buf, int buf_len) { | 172 int StreamPacketReader::DoRead(net::IOBuffer* buf, int buf_len) { |
170 int result = socket_->Read(buf, buf_len, | 173 int result = socket_->Read(buf, buf_len, |
171 base::Bind(&StreamPacketReader::OnReadComplete, | 174 base::Bind(&StreamPacketReader::OnReadComplete, |
172 weak_factory_.GetWeakPtr())); | 175 weak_factory_.GetWeakPtr())); |
173 return (result != 0 ? result : net::ERR_CONNECTION_CLOSED); | 176 return (result != 0 ? result : net::ERR_CONNECTION_CLOSED); |
174 } | 177 } |
175 | 178 |
176 } // namespace blimp | 179 } // namespace blimp |
OLD | NEW |