OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "remoting/protocol/buffered_socket_writer.h" | 5 #include "remoting/protocol/buffered_socket_writer.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/location.h" | 8 #include "base/location.h" |
9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
(...skipping 157 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
168 delete queue_.front(); | 168 delete queue_.front(); |
169 queue_.pop_front(); | 169 queue_.pop_front(); |
170 return result; | 170 return result; |
171 } | 171 } |
172 | 172 |
173 BufferedSocketWriter::BufferedSocketWriter() { | 173 BufferedSocketWriter::BufferedSocketWriter() { |
174 } | 174 } |
175 | 175 |
176 void BufferedSocketWriter::GetNextPacket( | 176 void BufferedSocketWriter::GetNextPacket( |
177 net::IOBuffer** buffer, int* size) { | 177 net::IOBuffer** buffer, int* size) { |
178 if (!current_buf_) { | 178 if (!current_buf_.get()) { |
179 if (queue_.empty()) { | 179 if (queue_.empty()) { |
180 *buffer = NULL; | 180 *buffer = NULL; |
181 return; // Nothing to write. | 181 return; // Nothing to write. |
182 } | 182 } |
183 current_buf_ = new net::DrainableIOBuffer( | 183 current_buf_ = new net::DrainableIOBuffer(queue_.front()->data.get(), |
184 queue_.front()->data, queue_.front()->data->size()); | 184 queue_.front()->data->size()); |
185 } | 185 } |
186 | 186 |
187 *buffer = current_buf_; | 187 *buffer = current_buf_.get(); |
188 *size = current_buf_->BytesRemaining(); | 188 *size = current_buf_->BytesRemaining(); |
189 } | 189 } |
190 | 190 |
191 base::Closure BufferedSocketWriter::AdvanceBufferPosition(int written) { | 191 base::Closure BufferedSocketWriter::AdvanceBufferPosition(int written) { |
192 buffer_size_ -= written; | 192 buffer_size_ -= written; |
193 current_buf_->DidConsume(written); | 193 current_buf_->DidConsume(written); |
194 | 194 |
195 if (current_buf_->BytesRemaining() == 0) { | 195 if (current_buf_->BytesRemaining() == 0) { |
196 current_buf_ = NULL; | 196 current_buf_ = NULL; |
197 return PopQueue(); | 197 return PopQueue(); |
(...skipping 10 matching lines...) Expand all Loading... |
208 | 208 |
209 BufferedDatagramWriter::BufferedDatagramWriter() { | 209 BufferedDatagramWriter::BufferedDatagramWriter() { |
210 } | 210 } |
211 | 211 |
212 void BufferedDatagramWriter::GetNextPacket( | 212 void BufferedDatagramWriter::GetNextPacket( |
213 net::IOBuffer** buffer, int* size) { | 213 net::IOBuffer** buffer, int* size) { |
214 if (queue_.empty()) { | 214 if (queue_.empty()) { |
215 *buffer = NULL; | 215 *buffer = NULL; |
216 return; // Nothing to write. | 216 return; // Nothing to write. |
217 } | 217 } |
218 *buffer = queue_.front()->data; | 218 *buffer = queue_.front()->data.get(); |
219 *size = queue_.front()->data->size(); | 219 *size = queue_.front()->data->size(); |
220 } | 220 } |
221 | 221 |
222 base::Closure BufferedDatagramWriter::AdvanceBufferPosition(int written) { | 222 base::Closure BufferedDatagramWriter::AdvanceBufferPosition(int written) { |
223 DCHECK_EQ(written, queue_.front()->data->size()); | 223 DCHECK_EQ(written, queue_.front()->data->size()); |
224 buffer_size_ -= queue_.front()->data->size(); | 224 buffer_size_ -= queue_.front()->data->size(); |
225 return PopQueue(); | 225 return PopQueue(); |
226 } | 226 } |
227 | 227 |
228 void BufferedDatagramWriter::OnError(int result) { | 228 void BufferedDatagramWriter::OnError(int result) { |
229 // Nothing to do here. | 229 // Nothing to do here. |
230 } | 230 } |
231 | 231 |
232 BufferedDatagramWriter::~BufferedDatagramWriter() { | 232 BufferedDatagramWriter::~BufferedDatagramWriter() { |
233 } | 233 } |
234 | 234 |
235 } // namespace protocol | 235 } // namespace protocol |
236 } // namespace remoting | 236 } // namespace remoting |
OLD | NEW |