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" | |
18 #include "extensions/common/extension_urls.h" | 15 #include "extensions/common/extension_urls.h" |
19 | 16 |
20 using content::BrowserContext; | 17 using content::BrowserContext; |
21 | 18 |
22 DEFINE_WEB_CONTENTS_USER_DATA_KEY( | 19 DEFINE_WEB_CONTENTS_USER_DATA_KEY( |
23 extensions::ChromeExtensionWebContentsObserver); | 20 extensions::ChromeExtensionWebContentsObserver); |
24 | 21 |
25 namespace extensions { | 22 namespace extensions { |
26 | 23 |
27 ChromeExtensionWebContentsObserver::ChromeExtensionWebContentsObserver( | 24 ChromeExtensionWebContentsObserver::ChromeExtensionWebContentsObserver( |
28 content::WebContents* web_contents) | 25 content::WebContents* web_contents) |
29 : ExtensionWebContentsObserver(web_contents) {} | 26 : ExtensionWebContentsObserver(web_contents) {} |
30 | 27 |
31 ChromeExtensionWebContentsObserver::~ChromeExtensionWebContentsObserver() {} | 28 ChromeExtensionWebContentsObserver::~ChromeExtensionWebContentsObserver() {} |
32 | 29 |
33 void ChromeExtensionWebContentsObserver::RenderViewCreated( | 30 void ChromeExtensionWebContentsObserver::RenderViewCreated( |
34 content::RenderViewHost* render_view_host) { | 31 content::RenderViewHost* render_view_host) { |
35 ReloadIfTerminated(render_view_host); | 32 ReloadIfTerminated(render_view_host); |
36 ExtensionWebContentsObserver::RenderViewCreated(render_view_host); | 33 ExtensionWebContentsObserver::RenderViewCreated(render_view_host); |
37 } | 34 } |
38 | 35 |
39 bool ChromeExtensionWebContentsObserver::OnMessageReceived( | 36 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, | 37 const IPC::Message& message, |
51 content::RenderFrameHost* render_frame_host) { | 38 content::RenderFrameHost* render_frame_host) { |
52 #if defined(ENABLE_EXTENSIONS) | 39 #if defined(ENABLE_EXTENSIONS) |
53 bool handled = true; | 40 bool handled = true; |
54 IPC_BEGIN_MESSAGE_MAP(ChromeExtensionWebContentsObserver, message) | 41 IPC_BEGIN_MESSAGE_MAP(ChromeExtensionWebContentsObserver, message) |
55 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DetailedConsoleMessageAdded, | 42 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DetailedConsoleMessageAdded, |
56 OnDetailedConsoleMessageAdded) | 43 OnDetailedConsoleMessageAdded) |
57 IPC_MESSAGE_UNHANDLED(handled = false) | 44 IPC_MESSAGE_UNHANDLED(handled = false) |
58 IPC_END_MESSAGE_MAP() | 45 IPC_END_MESSAGE_MAP() |
59 #else | 46 #else |
(...skipping 24 matching lines...) Expand all Loading... |
84 source, | 71 source, |
85 message, | 72 message, |
86 stack_trace, | 73 stack_trace, |
87 web_contents()->GetLastCommittedURL(), | 74 web_contents()->GetLastCommittedURL(), |
88 static_cast<logging::LogSeverity>(severity_level), | 75 static_cast<logging::LogSeverity>(severity_level), |
89 render_view_host->GetRoutingID(), | 76 render_view_host->GetRoutingID(), |
90 render_view_host->GetProcess()->GetID()))); | 77 render_view_host->GetProcess()->GetID()))); |
91 #endif | 78 #endif |
92 } | 79 } |
93 | 80 |
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( | 81 void ChromeExtensionWebContentsObserver::ReloadIfTerminated( |
103 content::RenderViewHost* render_view_host) { | 82 content::RenderViewHost* render_view_host) { |
104 std::string extension_id = GetExtensionId(render_view_host); | 83 std::string extension_id = GetExtensionId(render_view_host); |
105 if (extension_id.empty()) | 84 if (extension_id.empty()) |
106 return; | 85 return; |
107 | 86 |
108 ExtensionRegistry* registry = ExtensionRegistry::Get(browser_context()); | 87 ExtensionRegistry* registry = ExtensionRegistry::Get(browser_context()); |
109 | 88 |
110 // Reload the extension if it has crashed. | 89 // Reload the extension if it has crashed. |
111 // TODO(yoz): This reload doesn't happen synchronously for unpacked | 90 // TODO(yoz): This reload doesn't happen synchronously for unpacked |
112 // extensions. It seems to be fast enough, but there is a race. | 91 // extensions. It seems to be fast enough, but there is a race. |
113 // We should delay loading until the extension has reloaded. | 92 // We should delay loading until the extension has reloaded. |
114 if (registry->GetExtensionById(extension_id, ExtensionRegistry::TERMINATED)) { | 93 if (registry->GetExtensionById(extension_id, ExtensionRegistry::TERMINATED)) { |
115 ExtensionSystem::Get(browser_context())-> | 94 ExtensionSystem::Get(browser_context())-> |
116 extension_service()->ReloadExtension(extension_id); | 95 extension_service()->ReloadExtension(extension_id); |
117 } | 96 } |
118 } | 97 } |
119 | 98 |
120 } // namespace extensions | 99 } // namespace extensions |
OLD | NEW |