| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/host/gnubby_socket.h" | 5 #include "remoting/host/gnubby_socket.h" |
| 6 | 6 |
| 7 #include "base/callback_helpers.h" |
| 7 #include "base/macros.h" | 8 #include "base/macros.h" |
| 8 #include "base/timer/timer.h" | 9 #include "base/timer/timer.h" |
| 9 #include "net/base/io_buffer.h" | 10 #include "net/base/io_buffer.h" |
| 10 #include "net/base/net_errors.h" | 11 #include "net/base/net_errors.h" |
| 11 #include "net/socket/stream_socket.h" | 12 #include "net/socket/stream_socket.h" |
| 12 | 13 |
| 13 namespace remoting { | 14 namespace remoting { |
| 14 | 15 |
| 15 namespace { | 16 namespace { |
| 16 | 17 |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 65 | 66 |
| 66 void GnubbySocket::SendSshError() { | 67 void GnubbySocket::SendSshError() { |
| 67 DCHECK(CalledOnValidThread()); | 68 DCHECK(CalledOnValidThread()); |
| 68 | 69 |
| 69 SendResponse(std::string(kSshError, arraysize(kSshError))); | 70 SendResponse(std::string(kSshError, arraysize(kSshError))); |
| 70 } | 71 } |
| 71 | 72 |
| 72 void GnubbySocket::StartReadingRequest( | 73 void GnubbySocket::StartReadingRequest( |
| 73 const base::Closure& request_received_callback) { | 74 const base::Closure& request_received_callback) { |
| 74 DCHECK(CalledOnValidThread()); | 75 DCHECK(CalledOnValidThread()); |
| 76 DCHECK(request_received_callback_.is_null()); |
| 75 | 77 |
| 76 request_received_callback_ = request_received_callback; | 78 request_received_callback_ = request_received_callback; |
| 77 DoRead(); | 79 DoRead(); |
| 78 } | 80 } |
| 79 | 81 |
| 80 void GnubbySocket::OnDataWritten(int result) { | 82 void GnubbySocket::OnDataWritten(int result) { |
| 81 DCHECK(CalledOnValidThread()); | 83 DCHECK(CalledOnValidThread()); |
| 82 DCHECK(write_buffer_); | 84 DCHECK(write_buffer_); |
| 83 | 85 |
| 84 if (result < 0) { | 86 if (result < 0) { |
| 85 LOG(ERROR) << "Error in sending response."; | 87 LOG(ERROR) << "Error sending response: " << result; |
| 86 return; | 88 return; |
| 87 } | 89 } |
| 88 ResetTimer(); | 90 ResetTimer(); |
| 89 write_buffer_->DidConsume(result); | 91 write_buffer_->DidConsume(result); |
| 90 DoWrite(); | 92 DoWrite(); |
| 91 } | 93 } |
| 92 | 94 |
| 93 void GnubbySocket::DoWrite() { | 95 void GnubbySocket::DoWrite() { |
| 94 DCHECK(CalledOnValidThread()); | 96 DCHECK(CalledOnValidThread()); |
| 95 DCHECK(write_buffer_); | 97 DCHECK(write_buffer_); |
| 96 | 98 |
| 97 if (!write_buffer_->BytesRemaining()) { | 99 if (!write_buffer_->BytesRemaining()) { |
| 98 write_buffer_ = nullptr; | 100 write_buffer_ = nullptr; |
| 99 return; | 101 return; |
| 100 } | 102 } |
| 101 int result = socket_->Write( | 103 int result = socket_->Write( |
| 102 write_buffer_.get(), write_buffer_->BytesRemaining(), | 104 write_buffer_.get(), write_buffer_->BytesRemaining(), |
| 103 base::Bind(&GnubbySocket::OnDataWritten, base::Unretained(this))); | 105 base::Bind(&GnubbySocket::OnDataWritten, base::Unretained(this))); |
| 104 if (result != net::ERR_IO_PENDING) | 106 if (result != net::ERR_IO_PENDING) |
| 105 OnDataWritten(result); | 107 OnDataWritten(result); |
| 106 } | 108 } |
| 107 | 109 |
| 108 void GnubbySocket::OnDataRead(int bytes_read) { | 110 void GnubbySocket::OnDataRead(int result) { |
| 109 DCHECK(CalledOnValidThread()); | 111 DCHECK(CalledOnValidThread()); |
| 110 | 112 |
| 111 if (bytes_read < 0) { | 113 if (result <= 0) { |
| 112 LOG(ERROR) << "Error in reading request."; | 114 if (result < 0) |
| 115 LOG(ERROR) << "Error reading request: " << result; |
| 113 read_completed_ = true; | 116 read_completed_ = true; |
| 114 request_received_callback_.Run(); | 117 base::ResetAndReturn(&request_received_callback_).Run(); |
| 115 return; | 118 return; |
| 116 } | 119 } |
| 120 |
| 117 ResetTimer(); | 121 ResetTimer(); |
| 118 request_data_.insert(request_data_.end(), read_buffer_->data(), | 122 request_data_.insert(request_data_.end(), read_buffer_->data(), |
| 119 read_buffer_->data() + bytes_read); | 123 read_buffer_->data() + result); |
| 120 if (IsRequestComplete()) { | 124 if (IsRequestComplete()) { |
| 121 read_completed_ = true; | 125 read_completed_ = true; |
| 122 request_received_callback_.Run(); | 126 base::ResetAndReturn(&request_received_callback_).Run(); |
| 123 return; | 127 return; |
| 124 } | 128 } |
| 129 |
| 125 DoRead(); | 130 DoRead(); |
| 126 } | 131 } |
| 127 | 132 |
| 128 void GnubbySocket::DoRead() { | 133 void GnubbySocket::DoRead() { |
| 129 DCHECK(CalledOnValidThread()); | 134 DCHECK(CalledOnValidThread()); |
| 130 | 135 |
| 131 int result = socket_->Read( | 136 int result = socket_->Read( |
| 132 read_buffer_.get(), kRequestReadBufferLength, | 137 read_buffer_.get(), kRequestReadBufferLength, |
| 133 base::Bind(&GnubbySocket::OnDataRead, base::Unretained(this))); | 138 base::Bind(&GnubbySocket::OnDataRead, base::Unretained(this))); |
| 134 if (result != net::ERR_IO_PENDING) | 139 if (result != net::ERR_IO_PENDING) |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 172 | 177 |
| 173 return response_len; | 178 return response_len; |
| 174 } | 179 } |
| 175 | 180 |
| 176 void GnubbySocket::ResetTimer() { | 181 void GnubbySocket::ResetTimer() { |
| 177 if (timer_->IsRunning()) | 182 if (timer_->IsRunning()) |
| 178 timer_->Reset(); | 183 timer_->Reset(); |
| 179 } | 184 } |
| 180 | 185 |
| 181 } // namespace remoting | 186 } // namespace remoting |
| OLD | NEW |