| Index: remoting/host/security_key/gnubby_socket.h
|
| diff --git a/remoting/host/security_key/gnubby_socket.h b/remoting/host/security_key/gnubby_socket.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..c0e1b4d0aa002a760283a653a17a9bcc50b0697c
|
| --- /dev/null
|
| +++ b/remoting/host/security_key/gnubby_socket.h
|
| @@ -0,0 +1,113 @@
|
| +// 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.
|
| +
|
| +#ifndef REMOTING_HOST_SECURITY_KEY_GNUBBY_SOCKET_H_
|
| +#define REMOTING_HOST_SECURITY_KEY_GNUBBY_SOCKET_H_
|
| +
|
| +#include <stddef.h>
|
| +
|
| +#include <memory>
|
| +#include <string>
|
| +#include <vector>
|
| +
|
| +#include "base/callback.h"
|
| +#include "base/macros.h"
|
| +#include "base/memory/ref_counted.h"
|
| +#include "base/threading/thread_checker.h"
|
| +
|
| +namespace base {
|
| +class Timer;
|
| +} // namespace base
|
| +
|
| +namespace net {
|
| +class DrainableIOBuffer;
|
| +class IOBufferWithSize;
|
| +class StreamSocket;
|
| +} // namespace net
|
| +
|
| +namespace remoting {
|
| +
|
| +// Class that manages reading requests and sending responses. The socket can
|
| +// only handle receiving one request at a time. It expects to receive no extra
|
| +// bytes over the wire, which is checked by IsRequestTooLarge method.
|
| +class GnubbySocket {
|
| + public:
|
| + GnubbySocket(std::unique_ptr<net::StreamSocket> socket,
|
| + base::TimeDelta timeout,
|
| + const base::Closure& timeout_callback);
|
| + ~GnubbySocket();
|
| +
|
| + // Returns false if the request has not yet completed, or is too large to be
|
| + // processed. Otherwise, the cached request data is copied into |data_out| and
|
| + // the internal buffer resets and is ready for the next request.
|
| + bool GetAndClearRequestData(std::string* data_out);
|
| +
|
| + // Sends response data to the socket.
|
| + void SendResponse(const std::string& data);
|
| +
|
| + // Sends an SSH error code to the socket.
|
| + void SendSshError();
|
| +
|
| + // |request_received_callback| is used to notify the caller that request data
|
| + // has been fully read, and caller is to use GetAndClearRequestData method to
|
| + // get the request data.
|
| + void StartReadingRequest(const base::Closure& request_received_callback);
|
| +
|
| + private:
|
| + // Called when bytes are written to |socket_|.
|
| + void OnDataWritten(int result);
|
| +
|
| + // Continues writing to |socket_| if needed.
|
| + void DoWrite();
|
| +
|
| + // Called when bytes are read from |socket_|.
|
| + void OnDataRead(int bytes_read);
|
| +
|
| + // Continues to read.
|
| + void DoRead();
|
| +
|
| + // 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;
|
| +
|
| + // 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();
|
| +
|
| + // Ensures GnubbySocket methods are called on the same thread.
|
| + base::ThreadChecker thread_checker_;
|
| +
|
| + // The socket.
|
| + std::unique_ptr<net::StreamSocket> socket_;
|
| +
|
| + // Invoked when request data has been read.
|
| + base::Closure request_received_callback_;
|
| +
|
| + // Indicates whether read has completed and |request_received_callback_| is
|
| + // about to be run.
|
| + bool read_completed_;
|
| +
|
| + // Request data.
|
| + std::vector<char> request_data_;
|
| +
|
| + scoped_refptr<net::DrainableIOBuffer> write_buffer_;
|
| +
|
| + scoped_refptr<net::IOBufferWithSize> read_buffer_;
|
| +
|
| + // The activity timer.
|
| + std::unique_ptr<base::Timer> timer_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(GnubbySocket);
|
| +};
|
| +
|
| +} // namespace remoting
|
| +
|
| +#endif // REMOTING_HOST_SECURITY_KEY_GNUBBY_SOCKET_H_
|
|
|