blob: c380b81bf5694382e5ae2f0d29362106fe713c23 [file] [log] [blame]
// 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 <memory>
#include <string>
#include "base/macros.h"
#include "base/scoped_observer.h"
#include "extensions/common/user_script.h"
#include "extensions/renderer/script_injection.h"
#include "extensions/renderer/user_script_set.h"
class InjectionHost;
namespace blink {
class WebLocalFrame;
namespace extensions {
// A ScriptInjector for UserScripts.
class UserScriptInjector : public ScriptInjector,
public UserScriptSet::Observer {
UserScriptInjector(const UserScript* user_script,
UserScriptSet* user_script_set,
bool is_declarative);
~UserScriptInjector() override;
// UserScriptSet::Observer implementation.
void OnUserScriptsUpdated(const std::set<HostID>& changed_hosts,
const UserScriptList& scripts) override;
// ScriptInjector implementation.
UserScript::InjectionType script_type() const override;
bool ShouldExecuteInMainWorld() const override;
bool IsUserGesture() const override;
bool ExpectsResults() const override;
bool ShouldInjectJs(
UserScript::RunLocation run_location,
const std::set<std::string>& executing_scripts) const override;
bool ShouldInjectCss(
UserScript::RunLocation run_location,
const std::set<std::string>& injected_stylesheets) const override;
PermissionsData::AccessType CanExecuteOnFrame(
const InjectionHost* injection_host,
blink::WebLocalFrame* web_frame,
int tab_id) override;
std::vector<blink::WebScriptSource> GetJsSources(
UserScript::RunLocation run_location,
std::set<std::string>* executing_scripts,
size_t* num_injected_js_scripts) const override;
std::vector<blink::WebString> GetCssSources(
UserScript::RunLocation run_location,
std::set<std::string>* injected_stylesheets,
size_t* num_injected_stylesheets) const override;
void OnInjectionComplete(std::unique_ptr<base::Value> execution_result,
UserScript::RunLocation run_location,
content::RenderFrame* render_frame) override;
void OnWillNotInject(InjectFailureReason reason,
content::RenderFrame* render_frame) override;
// The associated user script. Owned by the UserScriptInjector that created
// this object.
const UserScript* script_;
// The UserScriptSet that eventually owns the UserScript this
// UserScriptInjector points to.
// Outlives |this|.
UserScriptSet* const user_script_set_;
// The id of the associated user script. We cache this because when we update
// the |script_| associated with this injection, the old referance may be
// deleted.
int script_id_;
// The associated host id, preserved for the same reason as |script_id|.
HostID host_id_;
// Indicates whether or not this script is declarative. This influences which
// script permissions are checked before injection.
bool is_declarative_;
ScopedObserver<UserScriptSet, UserScriptSet::Observer>
} // namespace extensions