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.
#include "remoting/host/webauthn/remote_webauthn_main.h"
#include "base/at_exit.h"
#include "base/command_line.h"
#include "base/files/file.h"
#include "base/logging.h"
#include "base/memory/scoped_refptr.h"
#include "base/message_loop/message_pump_type.h"
#include "base/run_loop.h"
#include "base/task/single_thread_task_executor.h"
#include "base/task/single_thread_task_runner.h"
#include "base/task/thread_pool/thread_pool_instance.h"
#include "build/build_config.h"
#include "mojo/core/embedder/embedder.h"
#include "mojo/core/embedder/scoped_ipc_support.h"
#include "remoting/base/auto_thread_task_runner.h"
#include "remoting/base/logging.h"
#include "remoting/host/base/host_exit_codes.h"
#include "remoting/host/chromoting_host_services_client.h"
#include "remoting/host/native_messaging/native_messaging_pipe.h"
#include "remoting/host/native_messaging/pipe_messaging_channel.h"
#include "remoting/host/usage_stats_consent.h"
#include "remoting/host/webauthn/remote_webauthn_caller_security_utils.h"
#include "remoting/host/webauthn/remote_webauthn_native_messaging_host.h"
#if BUILDFLAG(IS_LINUX)
#include "remoting/base/crash/crash_reporting_crashpad.h"
#endif // BUILDFLAG(IS_LINUX)
#if BUILDFLAG(IS_WIN)
#include <windows.h>
#include "remoting/base/crash/crash_reporting_breakpad.h"
#endif // BUILDFLAG(IS_WIN)
namespace remoting {
int RemoteWebAuthnMain(int argc, char** argv) {
base::AtExitManager exit_manager;
base::SingleThreadTaskExecutor task_executor(base::MessagePumpType::IO);
base::ThreadPoolInstance::Create("RemoteWebAuthn");
auto task_runner = base::SingleThreadTaskRunner::GetCurrentDefault();
base::CommandLine::Init(argc, argv);
InitHostLogging();
#if defined(REMOTING_ENABLE_CRASH_REPORTING)
if (IsUsageStatsAllowed()) {
#if BUILDFLAG(IS_LINUX)
InitializeCrashpadReporting();
#elif BUILDFLAG(IS_WIN)
InitializeBreakpadReporting();
#endif // BUILDFLAG(IS_LINUX)
}
#endif // defined(REMOTING_ENABLE_CRASH_REPORTING)
if (!IsLaunchedByTrustedProcess()) {
LOG(ERROR) << "Current process is not launched by a trusted process.";
return kNoPermissionExitCode;
}
if (!ChromotingHostServicesClient::Initialize()) {
return kInitializationFailed;
}
mojo::core::Init();
mojo::core::ScopedIPCSupport ipc_support(
task_runner, mojo::core::ScopedIPCSupport::ShutdownPolicy::FAST);
base::File read_file;
base::File write_file;
#if BUILDFLAG(IS_POSIX)
PipeMessagingChannel::OpenAndBlockStdio(read_file, write_file);
#elif BUILDFLAG(IS_WIN)
// GetStdHandle() returns pseudo-handles for stdin and stdout even if
// the hosting executable specifies "Windows" subsystem. However the
// returned handles are invalid in that case unless standard input and
// output are redirected to a pipe or file.
read_file = base::File(GetStdHandle(STD_INPUT_HANDLE));
write_file = base::File(GetStdHandle(STD_OUTPUT_HANDLE));
// After the native messaging channel starts, the native messaging reader
// will keep doing blocking read operations on the input named pipe.
// If any other thread tries to perform any operation on STDIN, it will also
// block because the input named pipe is synchronous (non-overlapped).
// It is pretty common for a DLL to query the device info (GetFileType) of
// the STD* handles at startup. So any LoadLibrary request can potentially
// be blocked. To prevent that from happening we close STDIN and STDOUT
// handles as soon as we retrieve the corresponding file handles.
SetStdHandle(STD_INPUT_HANDLE, nullptr);
SetStdHandle(STD_OUTPUT_HANDLE, nullptr);
#endif
base::RunLoop run_loop;
NativeMessagingPipe native_messaging_pipe;
auto channel = std::make_unique<PipeMessagingChannel>(std::move(read_file),
std::move(write_file));
auto native_messaging_host =
std::make_unique<RemoteWebAuthnNativeMessagingHost>(
base::MakeRefCounted<AutoThreadTaskRunner>(task_runner,
run_loop.QuitClosure()));
native_messaging_host->Start(&native_messaging_pipe);
native_messaging_pipe.Start(std::move(native_messaging_host),
std::move(channel));
run_loop.Run();
// Block until tasks blocking shutdown have completed their execution.
base::ThreadPoolInstance::Get()->Shutdown();
return kSuccessExitCode;
}
} // namespace remoting
|
