src/remoting/host/gnubby_socket.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef REMOTING_HOST_GNUBBY_SOCKET_H_
   6 #define REMOTING_HOST_GNUBBY_SOCKET_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/callback.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/threading/non_thread_safe.h"
  14
  15 namespace base {
  16 class Timer;
  17 }  // namespace base
  18
  19 namespace net {
  20 class StreamListenSocket;
  21 }  // namespace net
  22
  23 namespace remoting {
  24
  25 // Class that manages a socket used for gnubby requests.
  26 class GnubbySocket : public base::NonThreadSafe {
  27  public:
  28   GnubbySocket(scoped_ptr<net::StreamListenSocket> socket,
  29                const base::Closure& timeout_callback);
  30   ~GnubbySocket();
  31
  32   // Adds data to the current request.
  33   void AddRequestData(const char* data, int data_len);
  34
  35   // Gets the current request data and clears it.
  36   void GetAndClearRequestData(std::string* data_out);
  37
  38   // Returns true if the current request is complete.
  39   bool IsRequestComplete() const;
  40
  41   // Returns true if the stated request size is larger than the allowed maximum.
  42   bool IsRequestTooLarge() const;
  43
  44   // Sends response data to the socket.
  45   void SendResponse(const std::string& data);
  46
  47   // Sends an SSH error code to the socket.
  48   void SendSshError();
  49
  50   // Returns true if |socket| is the same one owned by this object.
  51   bool IsSocket(net::StreamListenSocket* socket) const;
  52
  53   // Sets a timer for testing.
  54   void SetTimerForTesting(scoped_ptr<base::Timer> timer);
  55
  56  private:
  57   // Returns the stated request length.
  58   size_t GetRequestLength() const;
  59
  60   // Returns the response length bytes.
  61   std::string GetResponseLengthAsBytes(const std::string& response) const;
  62
  63   // Resets the socket activity timer.
  64   void ResetTimer();
  65
  66   // The socket.
  67   scoped_ptr<net::StreamListenSocket> socket_;
  68
  69   // Request data.
  70   std::vector<char> request_data_;
  71
  72   // The activity timer.
  73   scoped_ptr<base::Timer> timer_;
  74
  75   DISALLOW_COPY_AND_ASSIGN(GnubbySocket);
  76 };
  77
  78 }  // namespace remoting
  79
  80 #endif  // REMOTING_HOST_GNUBBY_SOCKET_H_