blob: e2c6cdba9058b4c9de292105801a4d144bffb13d [file] [log] [blame]
// Copyright (c) 2012 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 "ipc/ipc_sync_channel.h"
#include <stddef.h>
#include <stdint.h>
#include <utility>
#include "base/bind.h"
#include "base/lazy_instance.h"
#include "base/location.h"
#include "base/logging.h"
#include "base/macros.h"
#include "base/memory/ptr_util.h"
#include "base/run_loop.h"
#include "base/synchronization/waitable_event.h"
#include "base/threading/thread_local.h"
#include "base/threading/thread_task_runner_handle.h"
#include "base/trace_event/trace_event.h"
#include "ipc/ipc_channel_factory.h"
#include "ipc/ipc_logging.h"
#include "ipc/ipc_message_macros.h"
#include "ipc/ipc_sync_message.h"
#include "ipc/mojo_event.h"
#include "mojo/public/cpp/bindings/sync_handle_registry.h"
#include "mojo/public/cpp/bindings/sync_handle_watcher.h"
using base::WaitableEvent;
namespace IPC {
namespace {
// A generic callback used when watching handles synchronously. Sets |*signal|
// to true. Also sets |*error| to true in case of an error.
void OnSyncHandleReady(bool* signal, bool* error, MojoResult result) {
*signal = true;
*error = result != MOJO_RESULT_OK;
}
// A ReadyCallback for use with mojo::Watcher. Ignores the result (DCHECKs, but
// is only used in cases where failure should be impossible) and runs
// |callback|.
void RunOnHandleReady(const base::Closure& callback, MojoResult result) {
DCHECK(result == MOJO_RESULT_OK || result == MOJO_RESULT_ABORTED);
if (result == MOJO_RESULT_OK)
callback.Run();
}
class PumpMessagesEvent {
public:
PumpMessagesEvent() { event_.Signal(); }
~PumpMessagesEvent() {}
const MojoEvent* event() const { return &event_; }
private:
MojoEvent event_;
DISALLOW_COPY_AND_ASSIGN(PumpMessagesEvent);
};
base::LazyInstance<PumpMessagesEvent>::Leaky g_pump_messages_event =
LAZY_INSTANCE_INITIALIZER;
} // namespace
// When we're blocked in a Send(), we need to process incoming synchronous
// messages right away because it could be blocking our reply (either
// directly from the same object we're calling, or indirectly through one or
// more other channels). That means that in SyncContext's OnMessageReceived,
// we need to process sync message right away if we're blocked. However a
// simple check isn't sufficient, because the listener thread can be in the
// process of calling Send.
// To work around this, when SyncChannel filters a sync message, it sets
// an event that the listener thread waits on during its Send() call. This
// allows us to dispatch incoming sync messages when blocked. The race
// condition is handled because if Send is in the process of being called, it
// will check the event. In case the listener thread isn't sending a message,
// we queue a task on the listener thread to dispatch the received messages.
// The messages are stored in this queue object that's shared among all
// SyncChannel objects on the same thread (since one object can receive a
// sync message while another one is blocked).
class SyncChannel::ReceivedSyncMsgQueue :
public base::RefCountedThreadSafe<ReceivedSyncMsgQueue> {
public:
// Returns the ReceivedSyncMsgQueue instance for this thread, creating one
// if necessary. Call RemoveContext on the same thread when done.
static ReceivedSyncMsgQueue* AddContext() {
// We want one ReceivedSyncMsgQueue per listener thread (i.e. since multiple
// SyncChannel objects can block the same thread).
ReceivedSyncMsgQueue* rv = lazy_tls_ptr_.Pointer()->Get();
if (!rv) {
rv = new ReceivedSyncMsgQueue();
ReceivedSyncMsgQueue::lazy_tls_ptr_.Pointer()->Set(rv);
}
rv->listener_count_++;
return rv;
}
// Prevents messages from being dispatched immediately when the dispatch event
// is signaled. Instead, |*dispatch_flag| will be set.
void BlockDispatch(bool* dispatch_flag) { dispatch_flag_ = dispatch_flag; }
// Allows messages to be dispatched immediately when the dispatch event is
// signaled.
void UnblockDispatch() { dispatch_flag_ = nullptr; }
// Called on IPC thread when a synchronous message or reply arrives.
void QueueMessage(const Message& msg, SyncChannel::SyncContext* context) {
bool was_task_pending;
{
base::AutoLock auto_lock(message_lock_);
was_task_pending = task_pending_;
task_pending_ = true;
// We set the event in case the listener thread is blocked (or is about
// to). In case it's not, the PostTask dispatches the messages.
message_queue_.push_back(QueuedMessage(new Message(msg), context));
message_queue_version_++;
}
dispatch_event_.Signal();
if (!was_task_pending) {
listener_task_runner_->PostTask(
FROM_HERE, base::Bind(&ReceivedSyncMsgQueue::DispatchMessagesTask,
this, base::RetainedRef(context)));
}
}
void QueueReply(const Message &msg, SyncChannel::SyncContext* context) {
received_replies_.push_back(QueuedMessage(new Message(msg), context));
}
// Called on the listener's thread to process any queues synchronous
// messages.
void DispatchMessagesTask(SyncContext* context) {
{
base::AutoLock auto_lock(message_lock_);
task_pending_ = false;
}
context->DispatchMessages();
}
// Dispatches any queued incoming sync messages. If |dispatching_context| is
// not null, messages which target a restricted dispatch channel will only be
// dispatched if |dispatching_context| belongs to the same restricted dispatch
// group as that channel. If |dispatching_context| is null, all queued
// messages are dispatched.
void DispatchMessages(SyncContext* dispatching_context) {
bool first_time = true;
uint32_t expected_version = 0;
SyncMessageQueue::iterator it;
while (true) {
Message* message = nullptr;
scoped_refptr<SyncChannel::SyncContext> context;
{
base::AutoLock auto_lock(message_lock_);
if (first_time || message_queue_version_ != expected_version) {
it = message_queue_.begin();
first_time = false;
}
for (; it != message_queue_.end(); it++) {
int message_group = it->context->restrict_dispatch_group();
if (!dispatching_context ||
message_group == kRestrictDispatchGroup_None ||
message_group == dispatching_context->restrict_dispatch_group()) {
message = it->message;
context = it->context;
it = message_queue_.erase(it);
message_queue_version_++;
expected_version = message_queue_version_;
break;
}
}
}
if (message == nullptr)
break;
context->OnDispatchMessage(*message);
delete message;
}
}
// SyncChannel calls this in its destructor.
void RemoveContext(SyncContext* context) {
base::AutoLock auto_lock(message_lock_);
SyncMessageQueue::iterator iter = message_queue_.begin();
while (iter != message_queue_.end()) {
if (iter->context.get() == context) {
delete iter->message;
iter = message_queue_.erase(iter);
message_queue_version_++;
} else {
iter++;
}
}
if (--listener_count_ == 0) {
DCHECK(lazy_tls_ptr_.Pointer()->Get());
lazy_tls_ptr_.Pointer()->Set(nullptr);
sync_dispatch_watcher_.reset();
}
}
MojoEvent* dispatch_event() { return &dispatch_event_; }
base::SingleThreadTaskRunner* listener_task_runner() {
return listener_task_runner_.get();
}
// Holds a pointer to the per-thread ReceivedSyncMsgQueue object.
static base::LazyInstance<base::ThreadLocalPointer<ReceivedSyncMsgQueue> >
lazy_tls_ptr_;
// Called on the ipc thread to check if we can unblock any current Send()
// calls based on a queued reply.
void DispatchReplies() {
for (size_t i = 0; i < received_replies_.size(); ++i) {
Message* message = received_replies_[i].message;
if (received_replies_[i].context->TryToUnblockListener(message)) {
delete message;
received_replies_.erase(received_replies_.begin() + i);
return;
}
}
}
mojo::Watcher* top_send_done_watcher() {
return top_send_done_watcher_;
}
void set_top_send_done_watcher(mojo::Watcher* watcher) {
top_send_done_watcher_ = watcher;
}
private:
friend class base::RefCountedThreadSafe<ReceivedSyncMsgQueue>;
// See the comment in SyncChannel::SyncChannel for why this event is created
// as manual reset.
ReceivedSyncMsgQueue()
: message_queue_version_(0),
listener_task_runner_(base::ThreadTaskRunnerHandle::Get()),
task_pending_(false),
listener_count_(0),
top_send_done_watcher_(nullptr) {
sync_dispatch_watcher_.reset(new mojo::SyncHandleWatcher(
dispatch_event_.GetHandle(), MOJO_HANDLE_SIGNAL_READABLE,
base::Bind(&ReceivedSyncMsgQueue::OnDispatchHandleReady,
base::Unretained(this))));
sync_dispatch_watcher_->AllowWokenUpBySyncWatchOnSameThread();
}
~ReceivedSyncMsgQueue() {}
void OnDispatchHandleReady(MojoResult result) {
if (result != MOJO_RESULT_OK)
return;
if (dispatch_flag_) {
*dispatch_flag_ = true;
return;
}
// We were woken up during a sync wait, but no specific SyncChannel is
// currently waiting. i.e., some other Mojo interface on this thread is
// waiting for a response. Since we don't support anything analogous to
// restricted dispatch on Mojo interfaces, in this case it's safe to
// dispatch sync messages for any context.
DispatchMessages(nullptr);
}
// Holds information about a queued synchronous message or reply.
struct QueuedMessage {
QueuedMessage(Message* m, SyncContext* c) : message(m), context(c) { }
Message* message;
scoped_refptr<SyncChannel::SyncContext> context;
};
typedef std::list<QueuedMessage> SyncMessageQueue;
SyncMessageQueue message_queue_;
uint32_t message_queue_version_; // Used to signal DispatchMessages to rescan
std::vector<QueuedMessage> received_replies_;
// Signaled when we get a synchronous message that we must respond to, as the
// sender needs its reply before it can reply to our original synchronous
// message.
MojoEvent dispatch_event_;
scoped_refptr<base::SingleThreadTaskRunner> listener_task_runner_;
base::Lock message_lock_;
bool task_pending_;
int listener_count_;
// The current send done handle watcher for this thread. Used to maintain
// a thread-local stack of send done watchers to ensure that nested sync
// message loops complete correctly.
mojo::Watcher* top_send_done_watcher_;
// If not null, the address of a flag to set when the dispatch event signals,
// in lieu of actually dispatching messages. This is used by
// SyncChannel::WaitForReply to restrict the scope of queued messages we're
// allowed to process while it's waiting.
bool* dispatch_flag_ = nullptr;
// Watches |dispatch_event_| during all sync handle watches on this thread.
std::unique_ptr<mojo::SyncHandleWatcher> sync_dispatch_watcher_;
};
base::LazyInstance<base::ThreadLocalPointer<SyncChannel::ReceivedSyncMsgQueue> >
SyncChannel::ReceivedSyncMsgQueue::lazy_tls_ptr_ =
LAZY_INSTANCE_INITIALIZER;
SyncChannel::SyncContext::SyncContext(
Listener* listener,
const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner,
WaitableEvent* shutdown_event)
: ChannelProxy::Context(listener, ipc_task_runner),
received_sync_msgs_(ReceivedSyncMsgQueue::AddContext()),
shutdown_event_(shutdown_event),
restrict_dispatch_group_(kRestrictDispatchGroup_None) {
}
SyncChannel::SyncContext::~SyncContext() {
while (!deserializers_.empty())
Pop();
}
// Adds information about an outgoing sync message to the context so that
// we know how to deserialize the reply. Returns |true| if the message was added
// to the context or |false| if it was rejected (e.g. due to shutdown.)
bool SyncChannel::SyncContext::Push(SyncMessage* sync_msg) {
// Create the tracking information for this message. This object is stored
// by value since all members are pointers that are cheap to copy. These
// pointers are cleaned up in the Pop() function.
//
// The event is created as manual reset because in between Signal and
// OnObjectSignalled, another Send can happen which would stop the watcher
// from being called. The event would get watched later, when the nested
// Send completes, so the event will need to remain set.
base::AutoLock auto_lock(deserializers_lock_);
if (reject_new_deserializers_)
return false;
PendingSyncMsg pending(
SyncMessage::GetMessageId(*sync_msg), sync_msg->GetReplyDeserializer(),
new MojoEvent);
deserializers_.push_back(pending);
return true;
}
bool SyncChannel::SyncContext::Pop() {
bool result;
{
base::AutoLock auto_lock(deserializers_lock_);
PendingSyncMsg msg = deserializers_.back();
delete msg.deserializer;
delete msg.done_event;
msg.done_event = nullptr;
deserializers_.pop_back();
result = msg.send_result;
}
// We got a reply to a synchronous Send() call that's blocking the listener
// thread. However, further down the call stack there could be another
// blocking Send() call, whose reply we received after we made this last
// Send() call. So check if we have any queued replies available that
// can now unblock the listener thread.
ipc_task_runner()->PostTask(
FROM_HERE, base::Bind(&ReceivedSyncMsgQueue::DispatchReplies,
received_sync_msgs_));
return result;
}
MojoEvent* SyncChannel::SyncContext::GetSendDoneEvent() {
base::AutoLock auto_lock(deserializers_lock_);
return deserializers_.back().done_event;
}
MojoEvent* SyncChannel::SyncContext::GetDispatchEvent() {
return received_sync_msgs_->dispatch_event();
}
void SyncChannel::SyncContext::DispatchMessages() {
received_sync_msgs_->DispatchMessages(this);
}
bool SyncChannel::SyncContext::TryToUnblockListener(const Message* msg) {
base::AutoLock auto_lock(deserializers_lock_);
if (deserializers_.empty() ||
!SyncMessage::IsMessageReplyTo(*msg, deserializers_.back().id)) {
return false;
}
if (!msg->is_reply_error()) {
bool send_result = deserializers_.back().deserializer->
SerializeOutputParameters(*msg);
deserializers_.back().send_result = send_result;
DVLOG_IF(1, !send_result) << "Couldn't deserialize reply message";
} else {
DVLOG(1) << "Received error reply";
}
MojoEvent* done_event = deserializers_.back().done_event;
TRACE_EVENT_FLOW_BEGIN0(
TRACE_DISABLED_BY_DEFAULT("ipc.flow"),
"SyncChannel::SyncContext::TryToUnblockListener", done_event);
done_event->Signal();
return true;
}
void SyncChannel::SyncContext::Clear() {
CancelPendingSends();
received_sync_msgs_->RemoveContext(this);
Context::Clear();
}
bool SyncChannel::SyncContext::OnMessageReceived(const Message& msg) {
// Give the filters a chance at processing this message.
if (TryFilters(msg))
return true;
if (TryToUnblockListener(&msg))
return true;
if (msg.is_reply()) {
received_sync_msgs_->QueueReply(msg, this);
return true;
}
if (msg.should_unblock()) {
received_sync_msgs_->QueueMessage(msg, this);
return true;
}
return Context::OnMessageReceivedNoFilter(msg);
}
void SyncChannel::SyncContext::OnChannelError() {
CancelPendingSends();
shutdown_watcher_.StopWatching();
Context::OnChannelError();
}
void SyncChannel::SyncContext::OnChannelOpened() {
shutdown_watcher_.StartWatching(
shutdown_event_,
base::Bind(&SyncChannel::SyncContext::OnShutdownEventSignaled,
base::Unretained(this)));
Context::OnChannelOpened();
}
void SyncChannel::SyncContext::OnChannelClosed() {
CancelPendingSends();
shutdown_watcher_.StopWatching();
Context::OnChannelClosed();
}
void SyncChannel::SyncContext::CancelPendingSends() {
base::AutoLock auto_lock(deserializers_lock_);
reject_new_deserializers_ = true;
PendingSyncMessageQueue::iterator iter;
DVLOG(1) << "Canceling pending sends";
for (iter = deserializers_.begin(); iter != deserializers_.end(); iter++) {
TRACE_EVENT_FLOW_BEGIN0(TRACE_DISABLED_BY_DEFAULT("ipc.flow"),
"SyncChannel::SyncContext::CancelPendingSends",
iter->done_event);
iter->done_event->Signal();
}
}
void SyncChannel::SyncContext::OnShutdownEventSignaled(WaitableEvent* event) {
DCHECK_EQ(event, shutdown_event_);
// Process shut down before we can get a reply to a synchronous message.
// Cancel pending Send calls, which will end up setting the send done event.
CancelPendingSends();
}
// static
std::unique_ptr<SyncChannel> SyncChannel::Create(
const IPC::ChannelHandle& channel_handle,
Channel::Mode mode,
Listener* listener,
const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner,
bool create_pipe_now,
base::WaitableEvent* shutdown_event) {
std::unique_ptr<SyncChannel> channel =
Create(listener, ipc_task_runner, shutdown_event);
channel->Init(channel_handle, mode, create_pipe_now);
return channel;
}
// static
std::unique_ptr<SyncChannel> SyncChannel::Create(
std::unique_ptr<ChannelFactory> factory,
Listener* listener,
const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner,
bool create_pipe_now,
base::WaitableEvent* shutdown_event) {
std::unique_ptr<SyncChannel> channel =
Create(listener, ipc_task_runner, shutdown_event);
channel->Init(std::move(factory), create_pipe_now);
return channel;
}
// static
std::unique_ptr<SyncChannel> SyncChannel::Create(
Listener* listener,
const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner,
WaitableEvent* shutdown_event) {
return base::WrapUnique(
new SyncChannel(listener, ipc_task_runner, shutdown_event));
}
SyncChannel::SyncChannel(
Listener* listener,
const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner,
WaitableEvent* shutdown_event)
: ChannelProxy(new SyncContext(listener, ipc_task_runner, shutdown_event)),
sync_handle_registry_(mojo::SyncHandleRegistry::current()) {
// The current (listener) thread must be distinct from the IPC thread, or else
// sending synchronous messages will deadlock.
DCHECK_NE(ipc_task_runner.get(), base::ThreadTaskRunnerHandle::Get().get());
StartWatching();
}
SyncChannel::~SyncChannel() {
}
void SyncChannel::SetRestrictDispatchChannelGroup(int group) {
sync_context()->set_restrict_dispatch_group(group);
}
scoped_refptr<SyncMessageFilter> SyncChannel::CreateSyncMessageFilter() {
scoped_refptr<SyncMessageFilter> filter = new SyncMessageFilter(
sync_context()->shutdown_event());
AddFilter(filter.get());
if (!did_init())
pre_init_sync_message_filters_.push_back(filter);
return filter;
}
bool SyncChannel::Send(Message* message) {
#ifdef IPC_MESSAGE_LOG_ENABLED
std::string name;
Logging::GetInstance()->GetMessageText(
message->type(), &name, message, nullptr);
TRACE_EVENT1("ipc", "SyncChannel::Send", "name", name);
#else
TRACE_EVENT2("ipc", "SyncChannel::Send",
"class", IPC_MESSAGE_ID_CLASS(message->type()),
"line", IPC_MESSAGE_ID_LINE(message->type()));
#endif
if (!message->is_sync()) {
ChannelProxy::Send(message);
return true;
}
SyncMessage* sync_msg = static_cast<SyncMessage*>(message);
bool pump_messages = sync_msg->ShouldPumpMessages();
// *this* might get deleted in WaitForReply.
scoped_refptr<SyncContext> context(sync_context());
if (!context->Push(sync_msg)) {
DVLOG(1) << "Channel is shutting down. Dropping sync message.";
delete message;
return false;
}
ChannelProxy::Send(message);
// Wait for reply, or for any other incoming synchronous messages.
// |this| might get deleted, so only call static functions at this point.
scoped_refptr<mojo::SyncHandleRegistry> registry = sync_handle_registry_;
WaitForReply(registry.get(), context.get(), pump_messages);
TRACE_EVENT_FLOW_END0(TRACE_DISABLED_BY_DEFAULT("ipc.flow"),
"SyncChannel::Send", context->GetSendDoneEvent());
return context->Pop();
}
void SyncChannel::WaitForReply(mojo::SyncHandleRegistry* registry,
SyncContext* context,
bool pump_messages) {
context->DispatchMessages();
const MojoEvent* pump_messages_event = nullptr;
if (pump_messages)
pump_messages_event = g_pump_messages_event.Get().event();
while (true) {
bool dispatch = false;
bool send_done = false;
bool should_pump_messages = false;
bool error = false;
bool registered = registry->RegisterHandle(
context->GetSendDoneEvent()->GetHandle(),
MOJO_HANDLE_SIGNAL_READABLE,
base::Bind(&OnSyncHandleReady, &send_done, &error));
DCHECK(registered);
if (pump_messages_event) {
registered = registry->RegisterHandle(
pump_messages_event->GetHandle(), MOJO_HANDLE_SIGNAL_READABLE,
base::Bind(&OnSyncHandleReady, &should_pump_messages, &error));
DCHECK(registered);
}
const bool* stop_flags[] = { &dispatch, &send_done, &should_pump_messages };
context->received_sync_msgs()->BlockDispatch(&dispatch);
registry->WatchAllHandles(stop_flags, 3);
context->received_sync_msgs()->UnblockDispatch();
DCHECK(!error);
registry->UnregisterHandle(context->GetSendDoneEvent()->GetHandle());
if (pump_messages_event)
registry->UnregisterHandle(pump_messages_event->GetHandle());
if (dispatch) {
// We're waiting for a reply, but we received a blocking synchronous call.
// We must process it to avoid potential deadlocks.
context->GetDispatchEvent()->Reset();
context->DispatchMessages();
continue;
}
if (should_pump_messages)
WaitForReplyWithNestedMessageLoop(context); // Run a nested message loop.
break;
}
}
void SyncChannel::WaitForReplyWithNestedMessageLoop(SyncContext* context) {
mojo::Watcher send_done_watcher;
ReceivedSyncMsgQueue* sync_msg_queue = context->received_sync_msgs();
DCHECK_NE(sync_msg_queue, nullptr);
mojo::Watcher* old_watcher = sync_msg_queue->top_send_done_watcher();
mojo::Handle old_handle(mojo::kInvalidHandleValue);
mojo::Watcher::ReadyCallback old_callback;
// Maintain a thread-local stack of watchers to ensure nested calls complete
// in the correct sequence, i.e. the outermost call completes first, etc.
if (old_watcher) {
old_callback = old_watcher->ready_callback();
old_handle = old_watcher->handle();
old_watcher->Cancel();
}
sync_msg_queue->set_top_send_done_watcher(&send_done_watcher);
{
base::RunLoop nested_loop;
send_done_watcher.Start(
context->GetSendDoneEvent()->GetHandle(), MOJO_HANDLE_SIGNAL_READABLE,
base::Bind(&RunOnHandleReady, nested_loop.QuitClosure()));
base::MessageLoop::ScopedNestableTaskAllower allow(
base::MessageLoop::current());
nested_loop.Run();
send_done_watcher.Cancel();
}
sync_msg_queue->set_top_send_done_watcher(old_watcher);
if (old_watcher)
old_watcher->Start(old_handle, MOJO_HANDLE_SIGNAL_READABLE, old_callback);
}
void SyncChannel::OnDispatchHandleReady(MojoResult result) {
DCHECK(result == MOJO_RESULT_OK || result == MOJO_RESULT_ABORTED);
if (result == MOJO_RESULT_OK) {
sync_context()->GetDispatchEvent()->Reset();
sync_context()->DispatchMessages();
}
}
void SyncChannel::StartWatching() {
// |dispatch_watcher_| watches the event asynchronously, only dispatching
// messages once the listener thread is unblocked and pumping its task queue.
// The ReceivedSyncMsgQueue also watches this event and may dispatch
// immediately if woken up by a message which it's allowed to dispatch.
dispatch_watcher_.Start(sync_context()->GetDispatchEvent()->GetHandle(),
MOJO_HANDLE_SIGNAL_READABLE,
base::Bind(&SyncChannel::OnDispatchHandleReady,
base::Unretained(this)));
}
void SyncChannel::OnChannelInit() {
pre_init_sync_message_filters_.clear();
}
} // namespace IPC