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