1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// 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 "chrome/browser/extensions/chrome_extension_web_contents_observer.h"
#include "chrome/browser/extensions/error_console/error_console.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/extensions/window_controller.h"
#include "chrome/common/extensions/chrome_extension_messages.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/render_view_host.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/extension_system.h"
#include "extensions/common/extension_messages.h"
#include "extensions/common/extension_urls.h"
using content::BrowserContext;
DEFINE_WEB_CONTENTS_USER_DATA_KEY(
extensions::ChromeExtensionWebContentsObserver);
namespace extensions {
ChromeExtensionWebContentsObserver::ChromeExtensionWebContentsObserver(
content::WebContents* web_contents)
: ExtensionWebContentsObserver(web_contents) {}
ChromeExtensionWebContentsObserver::~ChromeExtensionWebContentsObserver() {}
void ChromeExtensionWebContentsObserver::RenderViewCreated(
content::RenderViewHost* render_view_host) {
ReloadIfTerminated(render_view_host);
ExtensionWebContentsObserver::RenderViewCreated(render_view_host);
}
bool ChromeExtensionWebContentsObserver::OnMessageReceived(
const IPC::Message& message,
content::RenderFrameHost* render_frame_host) {
if (ExtensionWebContentsObserver::OnMessageReceived(message,
render_frame_host)) {
return true;
}
bool handled = true;
IPC_BEGIN_MESSAGE_MAP_WITH_PARAM(ChromeExtensionWebContentsObserver, message,
render_frame_host)
IPC_MESSAGE_HANDLER(ExtensionHostMsg_DetailedConsoleMessageAdded,
OnDetailedConsoleMessageAdded)
IPC_MESSAGE_UNHANDLED(handled = false)
IPC_END_MESSAGE_MAP()
return handled;
}
void ChromeExtensionWebContentsObserver::OnDetailedConsoleMessageAdded(
content::RenderFrameHost* render_frame_host,
const base::string16& message,
const base::string16& source,
const StackTrace& stack_trace,
int32 severity_level) {
if (!IsSourceFromAnExtension(source))
return;
std::string extension_id = GetExtensionIdFromFrame(render_frame_host);
if (extension_id.empty())
extension_id = GURL(source).host();
ErrorConsole::Get(browser_context())
->ReportError(scoped_ptr<ExtensionError>(new RuntimeError(
extension_id, browser_context()->IsOffTheRecord(), source, message,
stack_trace, web_contents()->GetLastCommittedURL(),
static_cast<logging::LogSeverity>(severity_level),
render_frame_host->GetRoutingID(),
render_frame_host->GetProcess()->GetID())));
}
void ChromeExtensionWebContentsObserver::InitializeRenderFrame(
content::RenderFrameHost* render_frame_host) {
ExtensionWebContentsObserver::InitializeRenderFrame(render_frame_host);
WindowController* controller = dispatcher()->GetExtensionWindowController();
if (controller) {
render_frame_host->Send(new ExtensionMsg_UpdateBrowserWindowId(
render_frame_host->GetRoutingID(), controller->GetWindowId()));
}
}
void ChromeExtensionWebContentsObserver::ReloadIfTerminated(
content::RenderViewHost* render_view_host) {
std::string extension_id = GetExtensionId(render_view_host);
if (extension_id.empty())
return;
ExtensionRegistry* registry = ExtensionRegistry::Get(browser_context());
// Reload the extension if it has crashed.
// TODO(yoz): This reload doesn't happen synchronously for unpacked
// extensions. It seems to be fast enough, but there is a race.
// We should delay loading until the extension has reloaded.
if (registry->GetExtensionById(extension_id, ExtensionRegistry::TERMINATED)) {
ExtensionSystem::Get(browser_context())->
extension_service()->ReloadExtension(extension_id);
}
}
} // namespace extensions
|
