extensions/renderer/programmatic_script_injector.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 "extensions/renderer/programmatic_script_injector.h"
   6
   7 #include <vector>
   8
   9 #include "base/values.h"
  10 #include "content/public/renderer/render_view.h"
  11 #include "extensions/common/error_utils.h"
  12 #include "extensions/common/extension_messages.h"
  13 #include "extensions/common/manifest_constants.h"
  14 #include "extensions/common/permissions/permissions_data.h"
  15 #include "extensions/renderer/injection_host.h"
  16 #include "extensions/renderer/script_context.h"
  17 #include "third_party/WebKit/public/platform/WebString.h"
  18 #include "third_party/WebKit/public/web/WebDocument.h"
  19 #include "third_party/WebKit/public/web/WebFrame.h"
  20 #include "third_party/WebKit/public/web/WebScriptSource.h"
  21
  22 namespace extensions {
  23
  24 ProgrammaticScriptInjector::ProgrammaticScriptInjector(
  25     const ExtensionMsg_ExecuteCode_Params& params,
  26     blink::WebFrame* web_frame)
  27     : params_(new ExtensionMsg_ExecuteCode_Params(params)),
  28       url_(ScriptContext::GetDataSourceURLForFrame(web_frame)),
  29       render_view_(content::RenderView::FromWebView(web_frame->view())),
  30       results_(new base::ListValue()),
  31       finished_(false) {
  32 }
  33
  34 ProgrammaticScriptInjector::~ProgrammaticScriptInjector() {
  35 }
  36
  37 UserScript::InjectionType ProgrammaticScriptInjector::script_type()
  38     const {
  39   return UserScript::PROGRAMMATIC_SCRIPT;
  40 }
  41
  42 bool ProgrammaticScriptInjector::ShouldExecuteInChildFrames() const {
  43   return params_->all_frames;
  44 }
  45
  46 bool ProgrammaticScriptInjector::ShouldExecuteInMainWorld() const {
  47   return params_->in_main_world;
  48 }
  49
  50 bool ProgrammaticScriptInjector::IsUserGesture() const {
  51   return params_->user_gesture;
  52 }
  53
  54 bool ProgrammaticScriptInjector::ExpectsResults() const {
  55   return params_->wants_result;
  56 }
  57
  58 bool ProgrammaticScriptInjector::ShouldInjectJs(
  59     UserScript::RunLocation run_location) const {
  60   return GetRunLocation() == run_location && params_->is_javascript;
  61 }
  62
  63 bool ProgrammaticScriptInjector::ShouldInjectCss(
  64     UserScript::RunLocation run_location) const {
  65   return GetRunLocation() == run_location && !params_->is_javascript;
  66 }
  67
  68 PermissionsData::AccessType ProgrammaticScriptInjector::CanExecuteOnFrame(
  69     const InjectionHost* injection_host,
  70     blink::WebFrame* frame,
  71     int tab_id,
  72     const GURL& top_url) const {
  73   GURL effective_document_url = ScriptContext::GetEffectiveDocumentURL(
  74       frame, frame->document().url(), params_->match_about_blank);
  75   if (params_->is_web_view) {
  76     if (frame->parent()) {
  77       // This is a subframe inside <webview>, so allow it.
  78       return PermissionsData::ACCESS_ALLOWED;
  79     }
  80
  81     return effective_document_url == params_->webview_src
  82                ? PermissionsData::ACCESS_ALLOWED
  83                : PermissionsData::ACCESS_DENIED;
  84   }
  85   DCHECK_EQ(injection_host->id().type(), HostID::EXTENSIONS);
  86
  87   return injection_host->CanExecuteOnFrame(
  88       effective_document_url, top_url, tab_id, true /* is_declarative */);
  89 }
  90
  91 std::vector<blink::WebScriptSource> ProgrammaticScriptInjector::GetJsSources(
  92     UserScript::RunLocation run_location) const {
  93   DCHECK_EQ(GetRunLocation(), run_location);
  94   DCHECK(params_->is_javascript);
  95
  96   return std::vector<blink::WebScriptSource>(
  97       1,
  98       blink::WebScriptSource(
  99           blink::WebString::fromUTF8(params_->code), params_->file_url));
 100 }
 101
 102 std::vector<std::string> ProgrammaticScriptInjector::GetCssSources(
 103     UserScript::RunLocation run_location) const {
 104   DCHECK_EQ(GetRunLocation(), run_location);
 105   DCHECK(!params_->is_javascript);
 106
 107   return std::vector<std::string>(1, params_->code);
 108 }
 109
 110 void ProgrammaticScriptInjector::GetRunInfo(
 111     ScriptsRunInfo* scripts_run_info,
 112     UserScript::RunLocation run_location) const {
 113 }
 114
 115 void ProgrammaticScriptInjector::OnInjectionComplete(
 116     scoped_ptr<base::ListValue> execution_results,
 117     UserScript::RunLocation run_location) {
 118   results_ = execution_results.Pass();
 119   Finish(std::string());
 120 }
 121
 122 void ProgrammaticScriptInjector::OnWillNotInject(InjectFailureReason reason) {
 123   std::string error;
 124   switch (reason) {
 125     case NOT_ALLOWED:
 126       error = ErrorUtils::FormatErrorMessage(manifest_errors::kCannotAccessPage,
 127                                              url_.spec());
 128       break;
 129     case EXTENSION_REMOVED:  // no special error here.
 130     case WONT_INJECT:
 131       break;
 132   }
 133   Finish(error);
 134 }
 135
 136 UserScript::RunLocation ProgrammaticScriptInjector::GetRunLocation() const {
 137   return static_cast<UserScript::RunLocation>(params_->run_at);
 138 }
 139
 140 void ProgrammaticScriptInjector::Finish(const std::string& error) {
 141   DCHECK(!finished_);
 142   finished_ = true;
 143
 144   render_view_->Send(new ExtensionHostMsg_ExecuteCodeFinished(
 145       render_view_->GetRoutingID(),
 146       params_->request_id,
 147       error,
 148       url_,
 149       *results_));
 150 }
 151
 152 }  // namespace extensions