| Index: remoting/host/security_key/gnubby_socket.cc
|
| diff --git a/remoting/host/security_key/gnubby_socket.cc b/remoting/host/security_key/gnubby_socket.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..28a210641ec54d3884ae93b602d4f87761c64765
|
| --- /dev/null
|
| +++ b/remoting/host/security_key/gnubby_socket.cc
|
| @@ -0,0 +1,188 @@
|
| +// Copyright 2016 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "remoting/host/security_key/gnubby_socket.h"
|
| +
|
| +#include <utility>
|
| +
|
| +#include "base/callback_helpers.h"
|
| +#include "base/macros.h"
|
| +#include "base/timer/timer.h"
|
| +#include "net/base/io_buffer.h"
|
| +#include "net/base/net_errors.h"
|
| +#include "net/socket/stream_socket.h"
|
| +
|
| +namespace remoting {
|
| +
|
| +namespace {
|
| +
|
| +const size_t kRequestSizeBytes = 4;
|
| +const size_t kMaxRequestLength = 16384;
|
| +const size_t kRequestReadBufferLength = kRequestSizeBytes + kMaxRequestLength;
|
| +
|
| +// SSH Failure Code
|
| +const char kSshError[] = {0x05};
|
| +
|
| +} // namespace
|
| +
|
| +GnubbySocket::GnubbySocket(std::unique_ptr<net::StreamSocket> socket,
|
| + base::TimeDelta timeout,
|
| + const base::Closure& timeout_callback)
|
| + : socket_(std::move(socket)),
|
| + read_completed_(false),
|
| + read_buffer_(new net::IOBufferWithSize(kRequestReadBufferLength)) {
|
| + timer_.reset(new base::Timer(false, false));
|
| + timer_->Start(FROM_HERE, timeout, timeout_callback);
|
| +}
|
| +
|
| +GnubbySocket::~GnubbySocket() {}
|
| +
|
| +bool GnubbySocket::GetAndClearRequestData(std::string* data_out) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + DCHECK(read_completed_);
|
| +
|
| + if (!read_completed_)
|
| + return false;
|
| + if (!IsRequestComplete() || IsRequestTooLarge())
|
| + return false;
|
| + // The request size is not part of the data; don't send it.
|
| + data_out->assign(request_data_.begin() + kRequestSizeBytes,
|
| + request_data_.end());
|
| + request_data_.clear();
|
| + return true;
|
| +}
|
| +
|
| +void GnubbySocket::SendResponse(const std::string& response_data) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + DCHECK(!write_buffer_);
|
| +
|
| + std::string response_length_string = GetResponseLengthAsBytes(response_data);
|
| + int response_len = response_length_string.size() + response_data.size();
|
| + std::unique_ptr<std::string> response(
|
| + new std::string(response_length_string + response_data));
|
| + write_buffer_ = new net::DrainableIOBuffer(
|
| + new net::StringIOBuffer(std::move(response)), response_len);
|
| + DoWrite();
|
| +}
|
| +
|
| +void GnubbySocket::SendSshError() {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| +
|
| + SendResponse(std::string(kSshError, arraysize(kSshError)));
|
| +}
|
| +
|
| +void GnubbySocket::StartReadingRequest(
|
| + const base::Closure& request_received_callback) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + DCHECK(request_received_callback_.is_null());
|
| +
|
| + request_received_callback_ = request_received_callback;
|
| + DoRead();
|
| +}
|
| +
|
| +void GnubbySocket::OnDataWritten(int result) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + DCHECK(write_buffer_);
|
| +
|
| + if (result < 0) {
|
| + LOG(ERROR) << "Error sending response: " << result;
|
| + return;
|
| + }
|
| + ResetTimer();
|
| + write_buffer_->DidConsume(result);
|
| + DoWrite();
|
| +}
|
| +
|
| +void GnubbySocket::DoWrite() {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + DCHECK(write_buffer_);
|
| +
|
| + if (!write_buffer_->BytesRemaining()) {
|
| + write_buffer_ = nullptr;
|
| + return;
|
| + }
|
| + int result = socket_->Write(
|
| + write_buffer_.get(), write_buffer_->BytesRemaining(),
|
| + base::Bind(&GnubbySocket::OnDataWritten, base::Unretained(this)));
|
| + if (result != net::ERR_IO_PENDING)
|
| + OnDataWritten(result);
|
| +}
|
| +
|
| +void GnubbySocket::OnDataRead(int result) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| +
|
| + if (result <= 0) {
|
| + if (result < 0)
|
| + LOG(ERROR) << "Error reading request: " << result;
|
| + read_completed_ = true;
|
| + base::ResetAndReturn(&request_received_callback_).Run();
|
| + return;
|
| + }
|
| +
|
| + ResetTimer();
|
| + request_data_.insert(request_data_.end(), read_buffer_->data(),
|
| + read_buffer_->data() + result);
|
| + if (IsRequestComplete()) {
|
| + read_completed_ = true;
|
| + base::ResetAndReturn(&request_received_callback_).Run();
|
| + return;
|
| + }
|
| +
|
| + DoRead();
|
| +}
|
| +
|
| +void GnubbySocket::DoRead() {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| +
|
| + int result = socket_->Read(
|
| + read_buffer_.get(), kRequestReadBufferLength,
|
| + base::Bind(&GnubbySocket::OnDataRead, base::Unretained(this)));
|
| + if (result != net::ERR_IO_PENDING)
|
| + OnDataRead(result);
|
| +}
|
| +
|
| +bool GnubbySocket::IsRequestComplete() const {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| +
|
| + if (request_data_.size() < kRequestSizeBytes)
|
| + return false;
|
| + return GetRequestLength() <= request_data_.size();
|
| +}
|
| +
|
| +bool GnubbySocket::IsRequestTooLarge() const {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| +
|
| + if (request_data_.size() < kRequestSizeBytes)
|
| + return false;
|
| + return GetRequestLength() > kMaxRequestLength;
|
| +}
|
| +
|
| +size_t GnubbySocket::GetRequestLength() const {
|
| + DCHECK(request_data_.size() >= kRequestSizeBytes);
|
| +
|
| + return ((request_data_[0] & 255) << 24) + ((request_data_[1] & 255) << 16) +
|
| + ((request_data_[2] & 255) << 8) + (request_data_[3] & 255) +
|
| + kRequestSizeBytes;
|
| +}
|
| +
|
| +std::string GnubbySocket::GetResponseLengthAsBytes(
|
| + const std::string& response) const {
|
| + std::string response_len;
|
| + response_len.reserve(kRequestSizeBytes);
|
| + int len = response.size();
|
| +
|
| + response_len.push_back((len >> 24) & 255);
|
| + response_len.push_back((len >> 16) & 255);
|
| + response_len.push_back((len >> 8) & 255);
|
| + response_len.push_back(len & 255);
|
| +
|
| + return response_len;
|
| +}
|
| +
|
| +void GnubbySocket::ResetTimer() {
|
| + if (timer_->IsRunning())
|
| + timer_->Reset();
|
| +}
|
| +
|
| +} // namespace remoting
|
|
|