Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/extensions/chrome_extension_web_contents_observer.h" | 5 #include "chrome/browser/extensions/chrome_extension_web_contents_observer.h" |
| 6 | 6 |
| 7 #include "chrome/browser/extensions/api/messaging/message_service.h" | |
| 8 #include "chrome/browser/extensions/error_console/error_console.h" | 7 #include "chrome/browser/extensions/error_console/error_console.h" |
| 9 #include "chrome/browser/extensions/extension_service.h" | 8 #include "chrome/browser/extensions/extension_service.h" |
| 10 #include "chrome/common/extensions/chrome_extension_messages.h" | 9 #include "chrome/common/extensions/chrome_extension_messages.h" |
| 11 #include "content/public/browser/browser_context.h" | 10 #include "content/public/browser/browser_context.h" |
| 12 #include "content/public/browser/render_process_host.h" | 11 #include "content/public/browser/render_process_host.h" |
| 13 #include "content/public/browser/render_view_host.h" | 12 #include "content/public/browser/render_view_host.h" |
| 14 #include "extensions/browser/extension_registry.h" | 13 #include "extensions/browser/extension_registry.h" |
| 15 #include "extensions/browser/extension_system.h" | 14 #include "extensions/browser/extension_system.h" |
| 16 #include "extensions/common/api/messaging/message.h" | |
| 17 #include "extensions/common/extension_messages.h" | 15 #include "extensions/common/extension_messages.h" |
|
Lei Zhang
2014/07/17 17:56:42
This can be removed too.
not at google - send to devlin
2014/07/17 19:38:42
Done.
| |
| 18 #include "extensions/common/extension_urls.h" | 16 #include "extensions/common/extension_urls.h" |
| 19 | 17 |
| 20 using content::BrowserContext; | 18 using content::BrowserContext; |
| 21 | 19 |
| 22 DEFINE_WEB_CONTENTS_USER_DATA_KEY( | 20 DEFINE_WEB_CONTENTS_USER_DATA_KEY( |
| 23 extensions::ChromeExtensionWebContentsObserver); | 21 extensions::ChromeExtensionWebContentsObserver); |
| 24 | 22 |
| 25 namespace extensions { | 23 namespace extensions { |
| 26 | 24 |
| 27 ChromeExtensionWebContentsObserver::ChromeExtensionWebContentsObserver( | 25 ChromeExtensionWebContentsObserver::ChromeExtensionWebContentsObserver( |
| 28 content::WebContents* web_contents) | 26 content::WebContents* web_contents) |
| 29 : ExtensionWebContentsObserver(web_contents) {} | 27 : ExtensionWebContentsObserver(web_contents) {} |
| 30 | 28 |
| 31 ChromeExtensionWebContentsObserver::~ChromeExtensionWebContentsObserver() {} | 29 ChromeExtensionWebContentsObserver::~ChromeExtensionWebContentsObserver() {} |
| 32 | 30 |
| 33 void ChromeExtensionWebContentsObserver::RenderViewCreated( | 31 void ChromeExtensionWebContentsObserver::RenderViewCreated( |
| 34 content::RenderViewHost* render_view_host) { | 32 content::RenderViewHost* render_view_host) { |
| 35 ReloadIfTerminated(render_view_host); | 33 ReloadIfTerminated(render_view_host); |
| 36 ExtensionWebContentsObserver::RenderViewCreated(render_view_host); | 34 ExtensionWebContentsObserver::RenderViewCreated(render_view_host); |
| 37 } | 35 } |
| 38 | 36 |
| 39 bool ChromeExtensionWebContentsObserver::OnMessageReceived( | 37 bool ChromeExtensionWebContentsObserver::OnMessageReceived( |
| 40 const IPC::Message& message) { | |
| 41 bool handled = true; | |
| 42 IPC_BEGIN_MESSAGE_MAP(ChromeExtensionWebContentsObserver, message) | |
| 43 IPC_MESSAGE_HANDLER(ExtensionHostMsg_PostMessage, OnPostMessage) | |
| 44 IPC_MESSAGE_UNHANDLED(handled = false) | |
| 45 IPC_END_MESSAGE_MAP() | |
| 46 return handled; | |
| 47 } | |
| 48 | |
| 49 bool ChromeExtensionWebContentsObserver::OnMessageReceived( | |
| 50 const IPC::Message& message, | 38 const IPC::Message& message, |
| 51 content::RenderFrameHost* render_frame_host) { | 39 content::RenderFrameHost* render_frame_host) { |
| 52 #if defined(ENABLE_EXTENSIONS) | 40 #if defined(ENABLE_EXTENSIONS) |
| 53 bool handled = true; | 41 bool handled = true; |
| 54 IPC_BEGIN_MESSAGE_MAP(ChromeExtensionWebContentsObserver, message) | 42 IPC_BEGIN_MESSAGE_MAP(ChromeExtensionWebContentsObserver, message) |
| 55 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DetailedConsoleMessageAdded, | 43 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DetailedConsoleMessageAdded, |
| 56 OnDetailedConsoleMessageAdded) | 44 OnDetailedConsoleMessageAdded) |
| 57 IPC_MESSAGE_UNHANDLED(handled = false) | 45 IPC_MESSAGE_UNHANDLED(handled = false) |
| 58 IPC_END_MESSAGE_MAP() | 46 IPC_END_MESSAGE_MAP() |
| 59 #else | 47 #else |
| (...skipping 24 matching lines...) Expand all Loading... | |
| 84 source, | 72 source, |
| 85 message, | 73 message, |
| 86 stack_trace, | 74 stack_trace, |
| 87 web_contents()->GetLastCommittedURL(), | 75 web_contents()->GetLastCommittedURL(), |
| 88 static_cast<logging::LogSeverity>(severity_level), | 76 static_cast<logging::LogSeverity>(severity_level), |
| 89 render_view_host->GetRoutingID(), | 77 render_view_host->GetRoutingID(), |
| 90 render_view_host->GetProcess()->GetID()))); | 78 render_view_host->GetProcess()->GetID()))); |
| 91 #endif | 79 #endif |
| 92 } | 80 } |
| 93 | 81 |
| 94 void ChromeExtensionWebContentsObserver::OnPostMessage(int port_id, | |
| 95 const Message& message) { | |
| 96 MessageService* message_service = MessageService::Get(browser_context()); | |
| 97 if (message_service) { | |
| 98 message_service->PostMessage(port_id, message); | |
| 99 } | |
| 100 } | |
| 101 | |
| 102 void ChromeExtensionWebContentsObserver::ReloadIfTerminated( | 82 void ChromeExtensionWebContentsObserver::ReloadIfTerminated( |
| 103 content::RenderViewHost* render_view_host) { | 83 content::RenderViewHost* render_view_host) { |
| 104 std::string extension_id = GetExtensionId(render_view_host); | 84 std::string extension_id = GetExtensionId(render_view_host); |
| 105 if (extension_id.empty()) | 85 if (extension_id.empty()) |
| 106 return; | 86 return; |
| 107 | 87 |
| 108 ExtensionRegistry* registry = ExtensionRegistry::Get(browser_context()); | 88 ExtensionRegistry* registry = ExtensionRegistry::Get(browser_context()); |
| 109 | 89 |
| 110 // Reload the extension if it has crashed. | 90 // Reload the extension if it has crashed. |
| 111 // TODO(yoz): This reload doesn't happen synchronously for unpacked | 91 // TODO(yoz): This reload doesn't happen synchronously for unpacked |
| 112 // extensions. It seems to be fast enough, but there is a race. | 92 // extensions. It seems to be fast enough, but there is a race. |
| 113 // We should delay loading until the extension has reloaded. | 93 // We should delay loading until the extension has reloaded. |
| 114 if (registry->GetExtensionById(extension_id, ExtensionRegistry::TERMINATED)) { | 94 if (registry->GetExtensionById(extension_id, ExtensionRegistry::TERMINATED)) { |
| 115 ExtensionSystem::Get(browser_context())-> | 95 ExtensionSystem::Get(browser_context())-> |
| 116 extension_service()->ReloadExtension(extension_id); | 96 extension_service()->ReloadExtension(extension_id); |
| 117 } | 97 } |
| 118 } | 98 } |
| 119 | 99 |
| 120 } // namespace extensions | 100 } // namespace extensions |
| OLD | NEW |