remoting/host/gnubby_socket.h

// Copyright 2014 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.

#ifndef REMOTING_HOST_GNUBBY_SOCKET_H_
#define REMOTING_HOST_GNUBBY_SOCKET_H_

#include <string>
#include <vector>

#include "base/callback.h"
#include "base/memory/scoped_ptr.h"
#include "base/threading/non_thread_safe.h"

namespace base {
class Timer;
}  // namespace base

namespace net {
class StreamListenSocket;
}  // namespace net

namespace remoting {

// Class that manages a socket used for gnubby requests.
class GnubbySocket : public base::NonThreadSafe {
 public:
  GnubbySocket(scoped_ptr<net::StreamListenSocket> socket,
               const base::Closure& timeout_callback);
  ~GnubbySocket();

  // Adds data to the current request.
  void AddRequestData(const char* data, int data_len);

  // Gets the current request data and clears it.
  void GetAndClearRequestData(std::string* data_out);

  // Returns true if the current request is complete.
  bool IsRequestComplete() const;

  // Returns true if the stated request size is larger than the allowed maximum.
  bool IsRequestTooLarge() const;

  // Sends response data to the socket.
  void SendResponse(const std::string& data);

  // Sends an SSH error code to the socket.
  void SendSshError();

  // Returns true if |socket| is the same one owned by this object.
  bool IsSocket(net::StreamListenSocket* socket) const;

  // Sets a timer for testing.
  void SetTimerForTesting(scoped_ptr<base::Timer> timer);

 private:
  // Returns the stated request length.
  size_t GetRequestLength() const;

  // Returns the response length bytes.
  std::string GetResponseLengthAsBytes(const std::string& response) const;

  // Resets the socket activity timer.
  void ResetTimer();

  // The socket.
  scoped_ptr<net::StreamListenSocket> socket_;

  // Request data.
  std::vector<char> request_data_;

  // The activity timer.
  scoped_ptr<base::Timer> timer_;

  DISALLOW_COPY_AND_ASSIGN(GnubbySocket);
};

}  // namespace remoting

#endif  // REMOTING_HOST_GNUBBY_SOCKET_H_