| // Copyright 2014 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 "content/browser/devtools/embedded_worker_devtools_agent_host.h" |
| |
| #include "content/browser/devtools/devtools_manager_impl.h" |
| #include "content/browser/devtools/devtools_protocol.h" |
| #include "content/browser/devtools/devtools_protocol_constants.h" |
| #include "content/common/devtools_messages.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/render_process_host.h" |
| |
| namespace content { |
| |
| EmbeddedWorkerDevToolsAgentHost::EmbeddedWorkerDevToolsAgentHost( |
| WorkerId worker_id, |
| const SharedWorkerInstance& shared_worker) |
| : shared_worker_(new SharedWorkerInstance(shared_worker)), |
| state_(WORKER_UNINSPECTED), |
| worker_id_(worker_id) { |
| WorkerCreated(); |
| } |
| |
| EmbeddedWorkerDevToolsAgentHost::EmbeddedWorkerDevToolsAgentHost( |
| WorkerId worker_id, |
| const ServiceWorkerIdentifier& service_worker, |
| bool debug_service_worker_on_start) |
| : service_worker_(new ServiceWorkerIdentifier(service_worker)), |
| state_(WORKER_UNINSPECTED), |
| worker_id_(worker_id) { |
| if (debug_service_worker_on_start) |
| state_ = WORKER_PAUSED_FOR_DEBUG_ON_START; |
| WorkerCreated(); |
| } |
| |
| bool EmbeddedWorkerDevToolsAgentHost::IsWorker() const { |
| return true; |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::SendMessageToAgent( |
| IPC::Message* message_raw) { |
| scoped_ptr<IPC::Message> message(message_raw); |
| if (state_ != WORKER_INSPECTED) |
| return; |
| if (RenderProcessHost* host = RenderProcessHost::FromID(worker_id_.first)) { |
| message->set_routing_id(worker_id_.second); |
| host->Send(message.release()); |
| } |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::Attach() { |
| if (state_ != WORKER_INSPECTED) { |
| state_ = WORKER_INSPECTED; |
| AttachToWorker(); |
| } |
| IPCDevToolsAgentHost::Attach(); |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::OnClientDetached() { |
| if (state_ == WORKER_INSPECTED) { |
| state_ = WORKER_UNINSPECTED; |
| DetachFromWorker(); |
| } else if (state_ == WORKER_PAUSED_FOR_REATTACH) { |
| state_ = WORKER_UNINSPECTED; |
| } |
| } |
| |
| bool EmbeddedWorkerDevToolsAgentHost::OnMessageReceived( |
| const IPC::Message& msg) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP(EmbeddedWorkerDevToolsAgentHost, msg) |
| IPC_MESSAGE_HANDLER(DevToolsClientMsg_DispatchOnInspectorFrontend, |
| OnDispatchOnInspectorFrontend) |
| IPC_MESSAGE_HANDLER(DevToolsHostMsg_SaveAgentRuntimeState, |
| OnSaveAgentRuntimeState) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| return handled; |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::WorkerContextStarted() { |
| if (state_ == WORKER_PAUSED_FOR_DEBUG_ON_START) { |
| RenderProcessHost* rph = RenderProcessHost::FromID(worker_id_.first); |
| DevToolsManagerImpl::GetInstance()->Inspect(rph->GetBrowserContext(), this); |
| } else if (state_ == WORKER_PAUSED_FOR_REATTACH) { |
| DCHECK(IsAttached()); |
| state_ = WORKER_INSPECTED; |
| AttachToWorker(); |
| Reattach(saved_agent_state_); |
| } |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::WorkerRestarted(WorkerId worker_id) { |
| DCHECK_EQ(WORKER_TERMINATED, state_); |
| state_ = IsAttached() ? WORKER_PAUSED_FOR_REATTACH : WORKER_UNINSPECTED; |
| worker_id_ = worker_id; |
| WorkerCreated(); |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::WorkerDestroyed() { |
| DCHECK_NE(WORKER_TERMINATED, state_); |
| if (state_ == WORKER_INSPECTED) { |
| DCHECK(IsAttached()); |
| // Client host is debugging this worker agent host. |
| std::string notification = |
| DevToolsProtocol::CreateNotification( |
| devtools::Worker::disconnectedFromWorker::kName, NULL)->Serialize(); |
| DevToolsManagerImpl::GetInstance()->DispatchOnInspectorFrontend( |
| this, notification); |
| DetachFromWorker(); |
| } |
| state_ = WORKER_TERMINATED; |
| Release(); // Balanced in WorkerCreated() |
| } |
| |
| bool EmbeddedWorkerDevToolsAgentHost::Matches( |
| const SharedWorkerInstance& other) { |
| return shared_worker_ && shared_worker_->Matches(other); |
| } |
| |
| bool EmbeddedWorkerDevToolsAgentHost::Matches( |
| const ServiceWorkerIdentifier& other) { |
| return service_worker_ && service_worker_->Matches(other); |
| } |
| |
| EmbeddedWorkerDevToolsAgentHost::~EmbeddedWorkerDevToolsAgentHost() { |
| DCHECK_EQ(WORKER_TERMINATED, state_); |
| EmbeddedWorkerDevToolsManager::GetInstance()->RemoveInspectedWorkerData( |
| worker_id_); |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::AttachToWorker() { |
| if (RenderProcessHost* host = RenderProcessHost::FromID(worker_id_.first)) |
| host->AddRoute(worker_id_.second, this); |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::DetachFromWorker() { |
| if (RenderProcessHost* host = RenderProcessHost::FromID(worker_id_.first)) |
| host->RemoveRoute(worker_id_.second); |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::WorkerCreated() { |
| AddRef(); // Balanced in WorkerDestroyed() |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::OnDispatchOnInspectorFrontend( |
| const std::string& message) { |
| DevToolsManagerImpl::GetInstance()->DispatchOnInspectorFrontend( |
| this, message); |
| } |
| |
| void EmbeddedWorkerDevToolsAgentHost::OnSaveAgentRuntimeState( |
| const std::string& state) { |
| saved_agent_state_ = state; |
| } |
| |
| } // namespace content |