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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// Copyright 2021 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_WEBAUTHN_REMOTE_WEBAUTHN_NATIVE_MESSAGING_HOST_H_
#define REMOTING_HOST_WEBAUTHN_REMOTE_WEBAUTHN_NATIVE_MESSAGING_HOST_H_
#include <memory>
#include "base/containers/flat_map.h"
#include "base/containers/queue.h"
#include "base/functional/callback_forward.h"
#include "base/memory/raw_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "base/values.h"
#include "extensions/browser/api/messaging/native_message_host.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "mojo/public/cpp/bindings/remote_set.h"
#include "remoting/host/chromoting_host_services_provider.h"
#include "remoting/host/mojom/webauthn_proxy.mojom.h"
#include "remoting/host/native_messaging/log_message_handler.h"
namespace remoting {
// Native messaging host for handling remote authentication requests and sending
// them to the remoting host process via mojo.
class RemoteWebAuthnNativeMessagingHost final
: public extensions::NativeMessageHost {
public:
explicit RemoteWebAuthnNativeMessagingHost(
scoped_refptr<base::SingleThreadTaskRunner> task_runner);
RemoteWebAuthnNativeMessagingHost(const RemoteWebAuthnNativeMessagingHost&) =
delete;
RemoteWebAuthnNativeMessagingHost& operator=(
const RemoteWebAuthnNativeMessagingHost&) = delete;
~RemoteWebAuthnNativeMessagingHost() override;
void OnMessage(const std::string& message) override;
void Start(extensions::NativeMessageHost::Client* client) override;
scoped_refptr<base::SingleThreadTaskRunner> task_runner() const override;
private:
friend class RemoteWebAuthnNativeMessagingHostTest;
using IdToRequestMap = base::flat_map<base::Value, mojo::RemoteSetElementId>;
RemoteWebAuthnNativeMessagingHost(
std::unique_ptr<ChromotingHostServicesProvider> host_service_api_client,
scoped_refptr<base::SingleThreadTaskRunner> task_runner);
void ProcessHello(base::Value::Dict response);
void ProcessGetRemoteState(base::Value::Dict response);
void ProcessIsUvpaa(const base::Value::Dict& request,
base::Value::Dict response);
void ProcessCreate(const base::Value::Dict& request,
base::Value::Dict response);
void ProcessGet(const base::Value::Dict& request, base::Value::Dict response);
void ProcessCancel(const base::Value::Dict& request,
base::Value::Dict response);
void OnQueryVersionResult(uint32_t version);
void OnIpcDisconnected();
void OnIsUvpaaResponse(base::Value::Dict response, bool is_available);
void OnCreateResponse(base::Value::Dict response,
mojom::WebAuthnCreateResponsePtr remote_response);
void OnGetResponse(base::Value::Dict response,
mojom::WebAuthnGetResponsePtr remote_response);
void OnCancelResponse(base::Value::Dict response, bool was_canceled);
void QueryNextRemoteState();
void SendNextRemoteState(bool is_remoted);
// Attempts to connect to the IPC server if the connection has not been
// established. Returns a boolean indicating whether there is a valid IPC
// connection to the crd host.
bool EnsureIpcConnection();
void SendMessageToClient(base::Value::Dict message);
// Finds and returns the message ID from |response|. If message ID is not
// found, |response| will be attached with a WebAuthn error dict and sent to
// the NMH client, and `nullptr` will be returned.
const base::Value* FindMessageIdOrSendError(base::Value::Dict& response);
// Finds and returns request[request_data_key]. If request_data_key is not
// found, |response| will be attached with a WebAuthn error dict and sent to
// the NMH client, and `nullptr` will be returned.
const std::string* FindRequestDataOrSendError(
const base::Value::Dict& request,
const std::string& request_data_key,
base::Value::Dict& response);
mojo::PendingReceiver<mojom::WebAuthnRequestCanceller> AddRequestCanceller(
base::Value message_id);
void RemoveRequestCancellerByMessageId(const base::Value& message_id);
void OnRequestCancellerDisconnected(
mojo::RemoteSetElementId disconnecting_canceller);
// Sends a clientDisconnected message to the extension, so it can detach from
// the WebAuthn proxy API and clean up pending requests.
void SendClientDisconnectedMessage();
scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
std::unique_ptr<ChromotingHostServicesProvider> host_service_api_client_;
mojo::Remote<mojom::WebAuthnProxy> remote_;
mojo::RemoteSet<mojom::WebAuthnRequestCanceller> request_cancellers_;
IdToRequestMap id_to_request_canceller_;
// Only available after Start() is called.
raw_ptr<extensions::NativeMessageHost::Client> client_ = nullptr;
std::unique_ptr<LogMessageHandler> log_message_handler_;
base::RepeatingClosure on_request_canceller_disconnected_for_testing_;
// Pending getRemoteStateResponses to be sent.
base::queue<base::Value::Dict> get_remote_state_responses_;
};
} // namespace remoting
#endif // REMOTING_HOST_WEBAUTHN_REMOTE_WEBAUTHN_NATIVE_MESSAGING_HOST_H_
|
