Renaming Gnubby and RemoteSecurityKey files/classes/members

remoting/host/security_key/gnubby_socket.cc

-// 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