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
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "mojo/core/embedder/scoped_ipc_support.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/notreached.h"
#include "base/synchronization/waitable_event.h"
#include "base/task/single_thread_task_runner.h"
#include "base/threading/thread_restrictions.h"
#include "mojo/buildflags.h"
#include "mojo/core/embedder/embedder.h"
#include "mojo/core/ipcz_driver/transport.h"
#if BUILDFLAG(MOJO_SUPPORT_LEGACY_CORE)
#include "mojo/core/core.h"
#endif
namespace mojo::core {
namespace {
#if BUILDFLAG(MOJO_SUPPORT_LEGACY_CORE)
void ShutdownIPCSupport(base::OnceClosure callback) {
Core::Get()->RequestShutdown(std::move(callback));
}
#endif
} // namespace
ScopedIPCSupport::ScopedIPCSupport(
scoped_refptr<base::SingleThreadTaskRunner> io_thread_task_runner,
ShutdownPolicy shutdown_policy)
: shutdown_policy_(shutdown_policy) {
ipcz_driver::Transport::SetIOTaskRunner(io_thread_task_runner);
#if BUILDFLAG(MOJO_SUPPORT_LEGACY_CORE)
if (!IsMojoIpczEnabled()) {
Core::Get()->SetIOTaskRunner(std::move(io_thread_task_runner));
}
#endif
}
ScopedIPCSupport::~ScopedIPCSupport() {
if (IsMojoIpczEnabled()) {
// No extra shutdown required for mojo-ipcz.
// Suppress -Wunused-private-field warning, to not leak the buildflags
// include into the header.
(void)shutdown_policy_;
return;
}
#if BUILDFLAG(MOJO_SUPPORT_LEGACY_CORE)
if (shutdown_policy_ == ShutdownPolicy::FAST) {
ShutdownIPCSupport(base::DoNothing());
return;
}
base::WaitableEvent shutdown_event(
base::WaitableEvent::ResetPolicy::MANUAL,
base::WaitableEvent::InitialState::NOT_SIGNALED);
ShutdownIPCSupport(base::BindOnce(&base::WaitableEvent::Signal,
base::Unretained(&shutdown_event)));
base::ScopedAllowBaseSyncPrimitives allow_io;
shutdown_event.Wait();
#else
NOTREACHED();
#endif
}
} // namespace mojo::core
|
