| 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" |
| 11 #include "base/thread_task_runner_handle.h" | 11 #include "base/thread_task_runner_handle.h" |
| 12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
| 13 | 13 |
| 14 namespace remoting { | 14 namespace remoting { |
| 15 namespace protocol { | 15 namespace protocol { |
| 16 | 16 |
| 17 struct BufferedSocketWriterBase::PendingPacket { | 17 struct BufferedSocketWriterBase::PendingPacket { |
| 18 PendingPacket(scoped_refptr<net::IOBufferWithSize> data, | 18 PendingPacket(scoped_refptr<net::IOBufferWithSize> data, |
| 19 const base::Closure& done_task) | 19 const base::Closure& done_task) |
| 20 : data(data), | 20 : data(data), |
| 21 done_task(done_task) { | 21 done_task(done_task) { |
| 22 } | 22 } |
| 23 | 23 |
| 24 scoped_refptr<net::IOBufferWithSize> data; | 24 scoped_refptr<net::IOBufferWithSize> data; |
| 25 base::Closure done_task; | 25 base::Closure done_task; |
| 26 }; | 26 }; |
| 27 | 27 |
| 28 BufferedSocketWriterBase::BufferedSocketWriterBase() | 28 BufferedSocketWriterBase::BufferedSocketWriterBase() |
| 29 : buffer_size_(0), | 29 : buffer_size_(0), |
| 30 socket_(NULL), | 30 socket_(nullptr), |
| 31 write_pending_(false), | 31 write_pending_(false), |
| 32 closed_(false), | 32 closed_(false), |
| 33 destroyed_flag_(NULL) { | 33 destroyed_flag_(nullptr) { |
| 34 } | 34 } |
| 35 | 35 |
| 36 void BufferedSocketWriterBase::Init(net::Socket* socket, | 36 void BufferedSocketWriterBase::Init(net::Socket* socket, |
| 37 const WriteFailedCallback& callback) { | 37 const WriteFailedCallback& callback) { |
| 38 DCHECK(CalledOnValidThread()); | 38 DCHECK(CalledOnValidThread()); |
| 39 DCHECK(socket); | 39 DCHECK(socket); |
| 40 socket_ = socket; | 40 socket_ = socket; |
| 41 write_failed_callback_ = callback; | 41 write_failed_callback_ = callback; |
| 42 } | 42 } |
| 43 | 43 |
| (...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 108 | 108 |
| 109 base::Closure done_task = AdvanceBufferPosition(result); | 109 base::Closure done_task = AdvanceBufferPosition(result); |
| 110 if (!done_task.is_null()) { | 110 if (!done_task.is_null()) { |
| 111 bool destroyed = false; | 111 bool destroyed = false; |
| 112 destroyed_flag_ = &destroyed; | 112 destroyed_flag_ = &destroyed; |
| 113 done_task.Run(); | 113 done_task.Run(); |
| 114 if (destroyed) { | 114 if (destroyed) { |
| 115 // Stop doing anything if we've been destroyed by the callback. | 115 // Stop doing anything if we've been destroyed by the callback. |
| 116 return; | 116 return; |
| 117 } | 117 } |
| 118 destroyed_flag_ = NULL; | 118 destroyed_flag_ = nullptr; |
| 119 } | 119 } |
| 120 | 120 |
| 121 *write_again = true; | 121 *write_again = true; |
| 122 } | 122 } |
| 123 | 123 |
| 124 void BufferedSocketWriterBase::OnWritten(int result) { | 124 void BufferedSocketWriterBase::OnWritten(int result) { |
| 125 DCHECK(CalledOnValidThread()); | 125 DCHECK(CalledOnValidThread()); |
| 126 DCHECK(write_pending_); | 126 DCHECK(write_pending_); |
| 127 write_pending_ = false; | 127 write_pending_ = false; |
| 128 | 128 |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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_.get()) { | 178 if (!current_buf_.get()) { |
| 179 if (queue_.empty()) { | 179 if (queue_.empty()) { |
| 180 *buffer = NULL; | 180 *buffer = nullptr; |
| 181 return; // Nothing to write. | 181 return; // Nothing to write. |
| 182 } | 182 } |
| 183 current_buf_ = new net::DrainableIOBuffer(queue_.front()->data.get(), | 183 current_buf_ = new net::DrainableIOBuffer(queue_.front()->data.get(), |
| 184 queue_.front()->data->size()); | 184 queue_.front()->data->size()); |
| 185 } | 185 } |
| 186 | 186 |
| 187 *buffer = current_buf_.get(); | 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_ = nullptr; |
| 197 return PopQueue(); | 197 return PopQueue(); |
| 198 } | 198 } |
| 199 return base::Closure(); | 199 return base::Closure(); |
| 200 } | 200 } |
| 201 | 201 |
| 202 void BufferedSocketWriter::OnError(int result) { | 202 void BufferedSocketWriter::OnError(int result) { |
| 203 current_buf_ = NULL; | 203 current_buf_ = nullptr; |
| 204 } | 204 } |
| 205 | 205 |
| 206 BufferedSocketWriter::~BufferedSocketWriter() { | 206 BufferedSocketWriter::~BufferedSocketWriter() { |
| 207 } | 207 } |
| 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 = nullptr; |
| 216 return; // Nothing to write. | 216 return; // Nothing to write. |
| 217 } | 217 } |
| 218 *buffer = queue_.front()->data.get(); | 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 |