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
|
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "mojo/edk/system/request_context.h"
#include "base/lazy_instance.h"
#include "base/logging.h"
#include "base/threading/thread_local.h"
namespace mojo {
namespace edk {
namespace {
base::LazyInstance<base::ThreadLocalPointer<RequestContext>>::Leaky
g_current_context = LAZY_INSTANCE_INITIALIZER;
} // namespace
RequestContext::RequestContext() : RequestContext(Source::LOCAL_API_CALL) {}
RequestContext::RequestContext(Source source)
: source_(source), tls_context_(g_current_context.Pointer()) {
// We allow nested RequestContexts to exist as long as they aren't actually
// used for anything.
if (!tls_context_->Get())
tls_context_->Set(this);
}
RequestContext::~RequestContext() {
if (IsCurrent()) {
// NOTE: Callbacks invoked by this destructor are allowed to initiate new
// EDK requests on this thread, so we need to reset the thread-local context
// pointer before calling them. We persist the original notification source
// since we're starting over at the bottom of the stack.
tls_context_->Set(nullptr);
MojoWatchNotificationFlags flags = MOJO_WATCH_NOTIFICATION_FLAG_NONE;
if (source_ == Source::SYSTEM)
flags |= MOJO_WATCH_NOTIFICATION_FLAG_FROM_SYSTEM;
for (const WatchNotifyFinalizer& watch :
watch_notify_finalizers_.container()) {
// Establish a new request context for the extent of each callback to
// ensure that they don't themselves invoke callbacks while holding a
// watcher lock.
RequestContext request_context(source_);
watch.watcher->MaybeInvokeCallback(watch.result, watch.state, flags);
}
for (const scoped_refptr<Watcher>& watcher :
watch_cancel_finalizers_.container())
watcher->Cancel();
} else {
// It should be impossible for nested contexts to have finalizers.
DCHECK(watch_notify_finalizers_.container().empty());
DCHECK(watch_cancel_finalizers_.container().empty());
}
}
// static
RequestContext* RequestContext::current() {
DCHECK(g_current_context.Pointer()->Get());
return g_current_context.Pointer()->Get();
}
void RequestContext::AddWatchNotifyFinalizer(
scoped_refptr<Watcher> watcher,
MojoResult result,
const HandleSignalsState& state) {
DCHECK(IsCurrent());
watch_notify_finalizers_->push_back(
WatchNotifyFinalizer(watcher, result, state));
}
void RequestContext::AddWatchCancelFinalizer(scoped_refptr<Watcher> watcher) {
DCHECK(IsCurrent());
watch_cancel_finalizers_->push_back(watcher);
}
bool RequestContext::IsCurrent() const {
return tls_context_->Get() == this;
}
RequestContext::WatchNotifyFinalizer::WatchNotifyFinalizer(
scoped_refptr<Watcher> watcher,
MojoResult result,
const HandleSignalsState& state)
: watcher(watcher), result(result), state(state) {
}
RequestContext::WatchNotifyFinalizer::WatchNotifyFinalizer(
const WatchNotifyFinalizer& other) = default;
RequestContext::WatchNotifyFinalizer::~WatchNotifyFinalizer() {}
} // namespace edk
} // namespace mojo
|
