src/chrome/renderer/external_extension.cc

   1 // Copyright (c) 2012 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/renderer/external_extension.h"
   6
   7 #include "base/command_line.h"
   8 #include "chrome/common/chrome_switches.h"
   9 #include "chrome/common/render_messages.h"
  10 #include "chrome/common/search_provider.h"
  11 #include "content/public/renderer/render_view.h"
  12 #include "third_party/WebKit/public/web/WebDocument.h"
  13 #include "third_party/WebKit/public/web/WebFrame.h"
  14 #include "third_party/WebKit/public/web/WebView.h"
  15 #include "v8/include/v8.h"
  16
  17 using WebKit::WebFrame;
  18 using WebKit::WebView;
  19 using content::RenderView;
  20
  21 namespace extensions_v8 {
  22
  23 static const char* const kSearchProviderApi =
  24     "var external;"
  25     "if (!external)"
  26     "  external = {};"
  27     "external.AddSearchProvider = function(name) {"
  28     "  native function NativeAddSearchProvider();"
  29     "  NativeAddSearchProvider(name);"
  30     "};"
  31     "external.IsSearchProviderInstalled = function(name) {"
  32     "  native function NativeIsSearchProviderInstalled();"
  33     "  return NativeIsSearchProviderInstalled(name);"
  34     "};";
  35
  36 const char* const kExternalExtensionName = "v8/External";
  37
  38 class ExternalExtensionWrapper : public v8::Extension {
  39  public:
  40   ExternalExtensionWrapper();
  41
  42   // Allows v8's javascript code to call the native functions defined
  43   // in this class for window.external.
  44   virtual v8::Handle<v8::FunctionTemplate> GetNativeFunction(
  45       v8::Handle<v8::String> name) OVERRIDE;
  46
  47   // Helper function to find the RenderView. May return NULL.
  48   static RenderView* GetRenderView();
  49
  50   // Implementation of window.external.AddSearchProvider.
  51   static void AddSearchProvider(
  52       const v8::FunctionCallbackInfo<v8::Value>& args);
  53
  54   // Implementation of window.external.IsSearchProviderInstalled.
  55   static void IsSearchProviderInstalled(
  56       const v8::FunctionCallbackInfo<v8::Value>& args);
  57
  58  private:
  59   DISALLOW_COPY_AND_ASSIGN(ExternalExtensionWrapper);
  60 };
  61
  62 ExternalExtensionWrapper::ExternalExtensionWrapper()
  63     : v8::Extension(
  64         kExternalExtensionName,
  65         kSearchProviderApi) {
  66 }
  67
  68 v8::Handle<v8::FunctionTemplate> ExternalExtensionWrapper::GetNativeFunction(
  69     v8::Handle<v8::String> name) {
  70   if (name->Equals(v8::String::New("NativeAddSearchProvider")))
  71     return v8::FunctionTemplate::New(AddSearchProvider);
  72
  73   if (name->Equals(v8::String::New("NativeIsSearchProviderInstalled")))
  74     return v8::FunctionTemplate::New(IsSearchProviderInstalled);
  75
  76   return v8::Handle<v8::FunctionTemplate>();
  77 }
  78
  79 // static
  80 RenderView* ExternalExtensionWrapper::GetRenderView() {
  81   WebFrame* webframe = WebFrame::frameForCurrentContext();
  82   DCHECK(webframe) << "There should be an active frame since we just got "
  83       "a native function called.";
  84   if (!webframe) return NULL;
  85
  86   WebView* webview = webframe->view();
  87   if (!webview) return NULL;  // can happen during closing
  88
  89   return RenderView::FromWebView(webview);
  90 }
  91
  92 // static
  93 void ExternalExtensionWrapper::AddSearchProvider(
  94     const v8::FunctionCallbackInfo<v8::Value>& args) {
  95   if (!args.Length()) return;
  96
  97   std::string name = std::string(*v8::String::Utf8Value(args[0]));
  98   if (!name.length()) return;
  99
 100   RenderView* render_view = GetRenderView();
 101   if (!render_view) return;
 102
 103   GURL osd_url(name);
 104   if (!osd_url.is_empty() && osd_url.is_valid()) {
 105     render_view->Send(new ChromeViewHostMsg_PageHasOSDD(
 106         render_view->GetRoutingID(), render_view->GetPageId(), osd_url,
 107         search_provider::EXPLICIT_PROVIDER));
 108   }
 109 }
 110
 111 // static
 112 void ExternalExtensionWrapper::IsSearchProviderInstalled(
 113     const v8::FunctionCallbackInfo<v8::Value>& args) {
 114   if (!args.Length()) return;
 115   v8::String::Utf8Value utf8name(args[0]);
 116   if (!utf8name.length()) return;
 117
 118   std::string name = std::string(*utf8name);
 119   RenderView* render_view = GetRenderView();
 120   if (!render_view) return;
 121
 122   WebFrame* webframe = WebFrame::frameForCurrentContext();
 123   if (!webframe) return;
 124
 125   search_provider::InstallState install = search_provider::DENIED;
 126   GURL inquiry_url = GURL(name);
 127   if (!inquiry_url.is_empty()) {
 128       render_view->Send(new ChromeViewHostMsg_GetSearchProviderInstallState(
 129           render_view->GetRoutingID(),
 130           webframe->document().url(),
 131           inquiry_url,
 132           &install));
 133   }
 134
 135   if (install == search_provider::DENIED) {
 136     // FIXME: throw access denied exception.
 137     v8::ThrowException(v8::Exception::Error(v8::String::Empty()));
 138     return;
 139   }
 140   args.GetReturnValue().Set(static_cast<int32_t>(install));
 141 }
 142
 143 v8::Extension* ExternalExtension::Get() {
 144   return new ExternalExtensionWrapper();
 145 }
 146
 147 }  // namespace extensions_v8