src/chrome/browser/extensions/chrome_extension_web_contents_observer.cc

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