Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(494)

Unified Diff: remoting/host/security_key/gnubby_socket.h

Issue 2167473003: Revert of Renaming Gnubby and RemoteSecurityKey files/classes/members (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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_
« no previous file with comments | « remoting/host/security_key/gnubby_extension_session_unittest.cc ('k') | remoting/host/security_key/gnubby_socket.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698