123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199 |
- // 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_message_filter.h"
- #include "base/bind.h"
- #include "base/check.h"
- #include "base/location.h"
- #include "base/memory/ptr_util.h"
- #include "base/memory/ref_counted.h"
- #include "base/synchronization/waitable_event.h"
- #include "base/threading/thread_task_runner_handle.h"
- #include "base/trace_event/trace_event.h"
- #include "ipc/ipc_channel.h"
- #include "ipc/ipc_sync_message.h"
- #include "mojo/public/cpp/bindings/associated_receiver.h"
- #include "mojo/public/cpp/bindings/sync_handle_registry.h"
- namespace IPC {
- namespace {
- // A generic callback used when watching handles synchronously. Sets |*signal|
- // to true.
- void OnEventReady(bool* signal) {
- *signal = true;
- }
- } // namespace
- bool SyncMessageFilter::Send(Message* message) {
- if (!message->is_sync()) {
- {
- base::AutoLock auto_lock(lock_);
- if (!io_task_runner_.get()) {
- pending_messages_.emplace_back(base::WrapUnique(message));
- return true;
- }
- }
- io_task_runner_->PostTask(
- FROM_HERE,
- base::BindOnce(&SyncMessageFilter::SendOnIOThread, this, message));
- return true;
- }
- base::WaitableEvent done_event(
- base::WaitableEvent::ResetPolicy::MANUAL,
- base::WaitableEvent::InitialState::NOT_SIGNALED);
- PendingSyncMsg pending_message(
- SyncMessage::GetMessageId(*message),
- static_cast<SyncMessage*>(message)->GetReplyDeserializer(),
- &done_event);
- {
- base::AutoLock auto_lock(lock_);
- // Can't use this class on the main thread or else it can lead to deadlocks.
- // Also by definition, can't use this on IO thread since we're blocking it.
- if (base::ThreadTaskRunnerHandle::IsSet()) {
- DCHECK(base::ThreadTaskRunnerHandle::Get() != listener_task_runner_);
- DCHECK(base::ThreadTaskRunnerHandle::Get() != io_task_runner_);
- }
- pending_sync_messages_.insert(&pending_message);
- if (io_task_runner_.get()) {
- io_task_runner_->PostTask(
- FROM_HERE,
- base::BindOnce(&SyncMessageFilter::SendOnIOThread, this, message));
- } else {
- pending_messages_.emplace_back(base::WrapUnique(message));
- }
- }
- {
- bool done = false;
- bool shutdown = false;
- scoped_refptr<mojo::SyncHandleRegistry> registry =
- mojo::SyncHandleRegistry::current();
- mojo::SyncHandleRegistry::EventCallbackSubscription shutdown_subscription =
- registry->RegisterEvent(shutdown_event_,
- base::BindRepeating(&OnEventReady, &shutdown));
- mojo::SyncHandleRegistry::EventCallbackSubscription done_subscription =
- registry->RegisterEvent(&done_event,
- base::BindRepeating(&OnEventReady, &done));
- const bool* stop_flags[] = {&done, &shutdown};
- registry->Wait(stop_flags, 2);
- if (done) {
- TRACE_EVENT_WITH_FLOW0("toplevel.flow", "SyncMessageFilter::Send",
- &done_event, TRACE_EVENT_FLAG_FLOW_IN);
- }
- }
- {
- base::AutoLock auto_lock(lock_);
- delete pending_message.deserializer;
- pending_sync_messages_.erase(&pending_message);
- }
- return pending_message.send_result;
- }
- void SyncMessageFilter::OnFilterAdded(Channel* channel) {
- std::vector<std::unique_ptr<Message>> pending_messages;
- {
- base::AutoLock auto_lock(lock_);
- channel_ = channel;
- io_task_runner_ = base::ThreadTaskRunnerHandle::Get();
- std::swap(pending_messages_, pending_messages);
- }
- for (auto& msg : pending_messages)
- SendOnIOThread(msg.release());
- }
- void SyncMessageFilter::OnChannelError() {
- base::AutoLock auto_lock(lock_);
- channel_ = nullptr;
- SignalAllEvents();
- }
- void SyncMessageFilter::OnChannelClosing() {
- base::AutoLock auto_lock(lock_);
- channel_ = nullptr;
- SignalAllEvents();
- }
- bool SyncMessageFilter::OnMessageReceived(const Message& message) {
- base::AutoLock auto_lock(lock_);
- for (PendingSyncMessages::iterator iter = pending_sync_messages_.begin();
- iter != pending_sync_messages_.end(); ++iter) {
- if (SyncMessage::IsMessageReplyTo(message, (*iter)->id)) {
- if (!message.is_reply_error()) {
- (*iter)->send_result =
- (*iter)->deserializer->SerializeOutputParameters(message);
- }
- TRACE_EVENT_WITH_FLOW0("toplevel.flow",
- "SyncMessageFilter::OnMessageReceived",
- (*iter)->done_event, TRACE_EVENT_FLAG_FLOW_OUT);
- (*iter)->done_event->Signal();
- return true;
- }
- }
- return false;
- }
- SyncMessageFilter::SyncMessageFilter(base::WaitableEvent* shutdown_event)
- : channel_(nullptr),
- listener_task_runner_(base::ThreadTaskRunnerHandle::Get()),
- shutdown_event_(shutdown_event) {}
- SyncMessageFilter::~SyncMessageFilter() = default;
- void SyncMessageFilter::SendOnIOThread(Message* message) {
- if (channel_) {
- channel_->Send(message);
- return;
- }
- if (message->is_sync()) {
- // We don't know which thread sent it, but it doesn't matter, just signal
- // them all.
- base::AutoLock auto_lock(lock_);
- SignalAllEvents();
- }
- delete message;
- }
- void SyncMessageFilter::SignalAllEvents() {
- lock_.AssertAcquired();
- for (PendingSyncMessages::iterator iter = pending_sync_messages_.begin();
- iter != pending_sync_messages_.end(); ++iter) {
- TRACE_EVENT_WITH_FLOW0("toplevel.flow",
- "SyncMessageFilter::SignalAllEvents",
- (*iter)->done_event, TRACE_EVENT_FLAG_FLOW_OUT);
- (*iter)->done_event->Signal();
- }
- }
- void SyncMessageFilter::GetRemoteAssociatedInterface(
- mojo::GenericPendingAssociatedReceiver receiver) {
- base::AutoLock auto_lock(lock_);
- DCHECK(io_task_runner_ && io_task_runner_->BelongsToCurrentThread());
- if (!channel_) {
- // Attach the associated interface to a disconnected pipe, so that the
- // associated interface pointer can be used to make calls (which are
- // dropped).
- mojo::AssociateWithDisconnectedPipe(receiver.PassHandle());
- return;
- }
- Channel::AssociatedInterfaceSupport* support =
- channel_->GetAssociatedInterfaceSupport();
- support->GetRemoteAssociatedInterface(std::move(receiver));
- }
- } // namespace IPC
|