ash_tracing_handler.cc 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. // Copyright 2021 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. #include "ash/hud_display/ash_tracing_handler.h"
  5. #include <sys/mman.h>
  6. #include <algorithm>
  7. #include "ash/hud_display/ash_tracing_request.h"
  8. #include "ash/shell.h"
  9. #include "base/bind.h"
  10. #include "base/files/file.h"
  11. #include "base/files/platform_file.h"
  12. #include "base/logging.h"
  13. #include "base/task/task_traits.h"
  14. #include "base/task/thread_pool.h"
  15. #include "base/threading/sequenced_task_runner_handle.h"
  16. #include "services/tracing/public/cpp/perfetto/perfetto_config.h"
  17. #include "third_party/perfetto/include/perfetto/tracing/core/trace_config.h"
  18. #include "third_party/perfetto/include/perfetto/tracing/tracing.h"
  19. namespace ash {
  20. namespace hud_display {
  21. namespace {
  22. std::unique_ptr<perfetto::TracingSession> (*testing_perfetto_session_creator)(
  23. void) = nullptr;
  24. } // anonymous namespace
  25. AshTracingHandler::AshTracingHandler() {
  26. // Bind sequence checker.
  27. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  28. }
  29. AshTracingHandler::~AshTracingHandler() {
  30. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  31. }
  32. void AshTracingHandler::Start(AshTracingRequest* request) {
  33. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  34. DCHECK(!request_);
  35. DCHECK(!tracing_session_);
  36. request_ = request;
  37. perfetto::TraceConfig perfetto_config = tracing::GetDefaultPerfettoConfig(
  38. base::trace_event::TraceConfig(),
  39. /*privacy_filtering_enabled=*/false,
  40. /*convert_to_legacy_json=*/false,
  41. perfetto::protos::gen::ChromeConfig::USER_INITIATED);
  42. perfetto_config.set_write_into_file(true);
  43. tracing_session_ = testing_perfetto_session_creator
  44. ? testing_perfetto_session_creator()
  45. : perfetto::Tracing::NewTrace();
  46. tracing_session_->Setup(perfetto_config, request->GetPlatformFile());
  47. auto runner = base::SequencedTaskRunnerHandle::Get();
  48. base::WeakPtr<AshTracingHandler> weak_ptr = weak_factory_.GetWeakPtr();
  49. tracing_session_->SetOnStartCallback([runner, weak_ptr]() {
  50. runner->PostTask(
  51. FROM_HERE,
  52. base::BindOnce(&AshTracingHandler::OnTracingStarted, weak_ptr));
  53. });
  54. tracing_session_->SetOnStopCallback([runner, weak_ptr]() {
  55. runner->PostTask(
  56. FROM_HERE,
  57. base::BindOnce(&AshTracingHandler::OnTracingFinished, weak_ptr));
  58. });
  59. tracing_session_->Start();
  60. }
  61. void AshTracingHandler::Stop() {
  62. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  63. DCHECK(tracing_session_);
  64. tracing_session_->Stop();
  65. }
  66. bool AshTracingHandler::IsStarted() const {
  67. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  68. return static_cast<bool>(tracing_session_);
  69. }
  70. // static
  71. void AshTracingHandler::SetPerfettoTracingSessionCreatorForTesting(
  72. std::unique_ptr<perfetto::TracingSession> (*creator)(void)) {
  73. DCHECK(!testing_perfetto_session_creator);
  74. testing_perfetto_session_creator = creator;
  75. }
  76. // static
  77. void AshTracingHandler::ResetPerfettoTracingSessionCreatorForTesting() {
  78. DCHECK(testing_perfetto_session_creator);
  79. testing_perfetto_session_creator = nullptr;
  80. }
  81. void AshTracingHandler::OnTracingStarted() {
  82. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  83. request_->OnTracingStarted();
  84. }
  85. void AshTracingHandler::OnTracingFinished() {
  86. DCHECK_CALLED_ON_VALID_SEQUENCE(my_sequence_checker_);
  87. tracing_session_.reset();
  88. request_->OnTracingFinished();
  89. }
  90. } // namespace hud_display
  91. } // namespace ash