webrunner/renderer/on_load_script_injector.cc

// Copyright 2018 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 "webrunner/renderer/on_load_script_injector.h"

#include <lib/zx/vmo.h>
#include <utility>
#include <vector>

#include "base/auto_reset.h"
#include "base/memory/shared_memory_handle.h"
#include "base/strings/utf_string_conversions.h"
#include "content/public/renderer/render_frame.h"
#include "third_party/blink/public/common/associated_interfaces/associated_interface_registry.h"

namespace webrunner {

OnLoadScriptInjector::OnLoadScriptInjector(content::RenderFrame* frame)
    : RenderFrameObserver(frame), weak_ptr_factory_(this) {
  render_frame()->GetAssociatedInterfaceRegistry()->AddInterface(
      base::BindRepeating(&OnLoadScriptInjector::BindToRequest,
                          weak_ptr_factory_.GetWeakPtr()));
}

OnLoadScriptInjector::~OnLoadScriptInjector() {}

void OnLoadScriptInjector::BindToRequest(
    mojom::OnLoadScriptInjectorAssociatedRequest request) {
  bindings_.AddBinding(this, std::move(request));
}

void OnLoadScriptInjector::DidClearWindowObject() {
  // The script may cause the window to be cleared (e.g. by producing a page
  // load event), so the guard is used to prevent reentrancy and potential
  // infinite loops.
  if (is_handling_clear_window_object_)
    return;

  base::AutoReset<bool> clear_window_reset(&is_handling_clear_window_object_,
                                           true);

  for (mojo::ScopedSharedBufferHandle& script : on_load_scripts_) {
    DCHECK_EQ(script->GetSize() % 2, 0u);  // Crude check to see this is UTF-16.

    auto mapping = script->Map(script->GetSize());
    base::string16 script_converted(static_cast<base::char16*>(mapping.get()),
                                    script->GetSize() / sizeof(base::char16));
    render_frame()->ExecuteJavaScript(script_converted);
  }
}

void OnLoadScriptInjector::AddOnLoadScript(
    mojo::ScopedSharedBufferHandle script) {
  on_load_scripts_.push_back(std::move(script));
}

void OnLoadScriptInjector::ClearOnLoadScripts() {
  on_load_scripts_.clear();
}

void OnLoadScriptInjector::OnDestruct() {
  delete this;
}

}  // namespace webrunner