1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72

content / browser / tracing / trace_report / trace_report_internals_ui.cc [blame]

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/342213636): Remove this and spanify to fix the errors.
#pragma allow_unsafe_buffers
#endif

#include "content/browser/tracing/trace_report/trace_report_internals_ui.h"

#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "content/browser/tracing/trace_report/trace_report.mojom.h"
#include "content/browser/tracing/trace_report/trace_report_handler.h"
#include "content/browser/webui/web_ui_impl.h"
#include "content/grit/traces_internals_resources.h"
#include "content/grit/traces_internals_resources_map.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "content/public/browser/web_ui_data_source.h"
#include "content/public/common/bindings_policy.h"
#include "content/public/common/url_constants.h"

namespace content {

TraceReportInternalsUI::TraceReportInternalsUI(WebUI* web_ui)
    : WebUIController(web_ui) {
  WebUIDataSource* source = WebUIDataSource::CreateAndAdd(
      web_ui->GetWebContents()->GetBrowserContext(),
      kChromeUITracesInternalsHost);

  source->AddResourcePaths(base::make_span(kTracesInternalsResources,
                                           kTracesInternalsResourcesSize));
  source->AddResourcePath("", IDR_TRACES_INTERNALS_TRACE_REPORT_INTERNALS_HTML);

  // Add TrustedTypes policies necessary for using Polymer.
  source->OverrideContentSecurityPolicy(
      network::mojom::CSPDirectiveName::TrustedTypes,
      "trusted-types static-types polymer-html-literal "
      "polymer-template-event-attribute-policy "
      // Add TrustedTypes policies necessary for using Lit.
      "lit-html-desktop;");
}

TraceReportInternalsUI::~TraceReportInternalsUI() = default;

void TraceReportInternalsUI::WebUIRenderFrameCreated(
    RenderFrameHost* render_frame_host) {
  // Enable the JavaScript Mojo bindings in the renderer process, so the JS
  // code can call the Mojo APIs exposed by this WebUI.
  render_frame_host->EnableMojoJsBindings(nullptr);
}

void TraceReportInternalsUI::BindInterface(
    mojo::PendingReceiver<trace_report::mojom::TraceReportHandlerFactory>
        receiver) {
  page_factory_receiver_.reset();
  page_factory_receiver_.Bind(std::move(receiver));
}

void TraceReportInternalsUI::CreatePageHandler(
    mojo::PendingRemote<trace_report::mojom::Page> page,
    mojo::PendingReceiver<trace_report::mojom::PageHandler> receiver) {
  DCHECK(page);
  ui_handler_ = std::make_unique<TraceReportHandler>(std::move(receiver),
                                                     std::move(page));
}

WEB_UI_CONTROLLER_TYPE_IMPL(TraceReportInternalsUI)

}  // namespace content