1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
// Copyright 2016 The Chromium Authors
// 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_SECURITY_KEY_EXTENSION_SESSION_H_
#define REMOTING_HOST_SECURITY_KEY_SECURITY_KEY_EXTENSION_SESSION_H_
#include <memory>
#include <string>
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/threading/thread_checker.h"
#include "base/values.h"
#include "build/build_config.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "remoting/host/host_extension_session.h"
#include "remoting/host/mojom/remote_security_key.mojom.h"
namespace base {
class SingleThreadTaskRunner;
} // namespace base
namespace remoting {
class ClientSessionDetails;
class SecurityKeyAuthHandler;
namespace protocol {
class ClientStub;
}
// A HostExtensionSession implementation that enables Security Key support.
class SecurityKeyExtensionSession : public HostExtensionSession {
public:
SecurityKeyExtensionSession(
ClientSessionDetails* client_session_details,
protocol::ClientStub* client_stub,
scoped_refptr<base::SingleThreadTaskRunner> file_task_runner);
SecurityKeyExtensionSession(const SecurityKeyExtensionSession&) = delete;
SecurityKeyExtensionSession& operator=(const SecurityKeyExtensionSession&) =
delete;
~SecurityKeyExtensionSession() override;
// HostExtensionSession interface.
bool OnExtensionMessage(ClientSessionDetails* client_session_details,
protocol::ClientStub* client_stub,
const protocol::ExtensionMessage& message) override;
#if BUILDFLAG(IS_WIN)
void BindSecurityKeyForwarder(
mojo::PendingReceiver<mojom::SecurityKeyForwarder> receiver);
#endif // BUILDFLAG(IS_WIN)
// Allows overriding SecurityKeyAuthHandler for unit testing.
void SetSecurityKeyAuthHandlerForTesting(
std::unique_ptr<SecurityKeyAuthHandler> security_key_auth_handler);
private:
// These methods process specific security key extension message types.
void ProcessControlMessage(const base::Value::Dict& message_data) const;
void ProcessDataMessage(const base::Value::Dict& message_data) const;
void ProcessErrorMessage(const base::Value::Dict& message_data) const;
void SendMessageToClient(int connection_id, const std::string& data) const;
// Ensures SecurityKeyExtensionSession methods are called on the same thread.
base::ThreadChecker thread_checker_;
// Interface through which messages can be sent to the client.
raw_ptr<protocol::ClientStub> client_stub_ = nullptr;
// Handles platform specific security key operations.
std::unique_ptr<SecurityKeyAuthHandler> security_key_auth_handler_;
};
} // namespace remoting
#endif // REMOTING_HOST_SECURITY_KEY_SECURITY_KEY_EXTENSION_SESSION_H_
|
