| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "remoting/host/security_key/gnubby_extension_session.h" | 5 #include "remoting/host/security_key/gnubby_extension_session.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/json/json_reader.h" | 8 #include "base/json/json_reader.h" |
| 9 #include "base/json/json_writer.h" | 9 #include "base/json/json_writer.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "base/macros.h" | 11 #include "base/macros.h" |
| 12 #include "base/values.h" | 12 #include "base/values.h" |
| 13 #include "remoting/base/logging.h" | 13 #include "remoting/base/logging.h" |
| 14 #include "remoting/host/client_session_details.h" |
| 14 #include "remoting/host/security_key/gnubby_auth_handler.h" | 15 #include "remoting/host/security_key/gnubby_auth_handler.h" |
| 15 #include "remoting/proto/control.pb.h" | 16 #include "remoting/proto/control.pb.h" |
| 16 #include "remoting/protocol/client_stub.h" | 17 #include "remoting/protocol/client_stub.h" |
| 17 | 18 |
| 18 namespace { | 19 namespace { |
| 19 | 20 |
| 20 // Used as the type attribute of all Security Key protocol::ExtensionMessages. | 21 // Used as the type attribute of all Security Key protocol::ExtensionMessages. |
| 21 const char kExtensionMessageType[] = "gnubby-auth"; | 22 const char kExtensionMessageType[] = "gnubby-auth"; |
| 22 | 23 |
| 23 // Gnubby extension message data members. | 24 // Gnubby extension message data members. |
| (...skipping 29 matching lines...) Expand all Loading... |
| 53 } | 54 } |
| 54 } | 55 } |
| 55 return true; | 56 return true; |
| 56 } | 57 } |
| 57 | 58 |
| 58 } // namespace | 59 } // namespace |
| 59 | 60 |
| 60 namespace remoting { | 61 namespace remoting { |
| 61 | 62 |
| 62 GnubbyExtensionSession::GnubbyExtensionSession( | 63 GnubbyExtensionSession::GnubbyExtensionSession( |
| 64 ClientSessionDetails* client_session_details, |
| 63 protocol::ClientStub* client_stub) | 65 protocol::ClientStub* client_stub) |
| 64 : client_stub_(client_stub) { | 66 : client_stub_(client_stub) { |
| 65 DCHECK(client_stub_); | 67 DCHECK(client_stub_); |
| 66 | 68 |
| 67 gnubby_auth_handler_ = remoting::GnubbyAuthHandler::Create(base::Bind( | 69 gnubby_auth_handler_ = remoting::GnubbyAuthHandler::Create( |
| 68 &GnubbyExtensionSession::SendMessageToClient, base::Unretained(this))); | 70 client_session_details, |
| 71 base::Bind(&GnubbyExtensionSession::SendMessageToClient, |
| 72 base::Unretained(this))); |
| 69 } | 73 } |
| 70 | 74 |
| 71 GnubbyExtensionSession::~GnubbyExtensionSession() {} | 75 GnubbyExtensionSession::~GnubbyExtensionSession() {} |
| 72 | 76 |
| 73 // Returns true if the |message| is a Security Key ExtensionMessage. | 77 // Returns true if the |message| is a Security Key ExtensionMessage. |
| 74 // This is done so the host does not pass |message| to other HostExtensions. | 78 // This is done so the host does not pass |message| to other HostExtensions. |
| 75 // TODO(joedow): Use |client_session_details| to disconnect the session if we | 79 // TODO(joedow): Use |client_session_details| to disconnect the session if we |
| 76 // receive an invalid extension message. | 80 // receive an invalid extension message. |
| 77 bool GnubbyExtensionSession::OnExtensionMessage( | 81 bool GnubbyExtensionSession::OnExtensionMessage( |
| 78 ClientSessionDetails* client_session_details, | 82 ClientSessionDetails* client_session_details, |
| (...skipping 119 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 198 void GnubbyExtensionSession::SetGnubbyAuthHandlerForTesting( | 202 void GnubbyExtensionSession::SetGnubbyAuthHandlerForTesting( |
| 199 std::unique_ptr<GnubbyAuthHandler> gnubby_auth_handler) { | 203 std::unique_ptr<GnubbyAuthHandler> gnubby_auth_handler) { |
| 200 DCHECK(gnubby_auth_handler); | 204 DCHECK(gnubby_auth_handler); |
| 201 | 205 |
| 202 gnubby_auth_handler_ = std::move(gnubby_auth_handler); | 206 gnubby_auth_handler_ = std::move(gnubby_auth_handler); |
| 203 gnubby_auth_handler_->SetSendMessageCallback(base::Bind( | 207 gnubby_auth_handler_->SetSendMessageCallback(base::Bind( |
| 204 &GnubbyExtensionSession::SendMessageToClient, base::Unretained(this))); | 208 &GnubbyExtensionSession::SendMessageToClient, base::Unretained(this))); |
| 205 } | 209 } |
| 206 | 210 |
| 207 } // namespace remoting | 211 } // namespace remoting |
| OLD | NEW |