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
3 // found in the LICENSE file.
4
5 #include "chrome/browser/extensions/chrome_extension_web_contents_observer.h"
6
7 #include "chrome/browser/extensions/api/messaging/message_service.h"
8 #include "chrome/browser/extensions/error_console/error_console.h"
9 #include "chrome/browser/extensions/extension_service.h"
10 #include "chrome/common/render_messages.h"
11 #include "content/public/browser/browser_context.h"
12 #include "content/public/browser/render_process_host.h"
13 #include "content/public/browser/render_view_host.h"
14 #include "extensions/browser/extension_registry.h"
15 #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"
19
20 using content::BrowserContext;
21
22 DEFINE_WEB_CONTENTS_USER_DATA_KEY(
23 extensions::ChromeExtensionWebContentsObserver);
24
25 namespace extensions {
26
ChromeExtensionWebContentsObserver(content::WebContents * web_contents)27 ChromeExtensionWebContentsObserver::ChromeExtensionWebContentsObserver(
28 content::WebContents* web_contents)
29 : ExtensionWebContentsObserver(web_contents) {}
30
~ChromeExtensionWebContentsObserver()31 ChromeExtensionWebContentsObserver::~ChromeExtensionWebContentsObserver() {}
32
RenderViewCreated(content::RenderViewHost * render_view_host)33 void ChromeExtensionWebContentsObserver::RenderViewCreated(
34 content::RenderViewHost* render_view_host) {
35 ReloadIfTerminated(render_view_host);
36 ExtensionWebContentsObserver::RenderViewCreated(render_view_host);
37 }
38
OnMessageReceived(const IPC::Message & message)39 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
OnMessageReceived(const IPC::Message & message,content::RenderFrameHost * render_frame_host)49 bool ChromeExtensionWebContentsObserver::OnMessageReceived(
50 const IPC::Message& message,
51 content::RenderFrameHost* render_frame_host) {
52 #if defined(ENABLE_EXTENSIONS)
53 bool handled = true;
54 IPC_BEGIN_MESSAGE_MAP(ChromeExtensionWebContentsObserver, message)
55 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DetailedConsoleMessageAdded,
56 OnDetailedConsoleMessageAdded)
57 IPC_MESSAGE_UNHANDLED(handled = false)
58 IPC_END_MESSAGE_MAP()
59 #else
60 bool handled = false;
61 #endif
62 return handled;
63 }
64
OnDetailedConsoleMessageAdded(const base::string16 & message,const base::string16 & source,const StackTrace & stack_trace,int32 severity_level)65 void ChromeExtensionWebContentsObserver::OnDetailedConsoleMessageAdded(
66 const base::string16& message,
67 const base::string16& source,
68 const StackTrace& stack_trace,
69 int32 severity_level) {
70 #if defined(ENABLE_EXTENSIONS)
71 if (!IsSourceFromAnExtension(source))
72 return;
73
74 content::RenderViewHost* render_view_host =
75 web_contents()->GetRenderViewHost();
76 std::string extension_id = GetExtensionId(render_view_host);
77 if (extension_id.empty())
78 extension_id = GURL(source).host();
79
80 ExtensionSystem::Get(browser_context())->error_console()->ReportError(
81 scoped_ptr<ExtensionError>(
82 new RuntimeError(extension_id,
83 browser_context()->IsOffTheRecord(),
84 source,
85 message,
86 stack_trace,
87 web_contents()->GetLastCommittedURL(),
88 static_cast<logging::LogSeverity>(severity_level),
89 render_view_host->GetRoutingID(),
90 render_view_host->GetProcess()->GetID())));
91 #endif
92 }
93
OnPostMessage(int port_id,const Message & message)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
ReloadIfTerminated(content::RenderViewHost * render_view_host)102 void ChromeExtensionWebContentsObserver::ReloadIfTerminated(
103 content::RenderViewHost* render_view_host) {
104 std::string extension_id = GetExtensionId(render_view_host);
105 if (extension_id.empty())
106 return;
107
108 ExtensionRegistry* registry = ExtensionRegistry::Get(browser_context());
109
110 // Reload the extension if it has crashed.
111 // TODO(yoz): This reload doesn't happen synchronously for unpacked
112 // extensions. It seems to be fast enough, but there is a race.
113 // We should delay loading until the extension has reloaded.
114 if (registry->GetExtensionById(extension_id, ExtensionRegistry::TERMINATED)) {
115 ExtensionSystem::Get(browser_context())->
116 extension_service()->ReloadExtension(extension_id);
117 }
118 }
119
120 } // namespace extensions
121