// Copyright 2015 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 "services/tracing/service.h"

#include <stddef.h>
#include <stdint.h>

#include <utility>

#include "base/bind.h"
#include "base/logging.h"
#include "base/memory/ptr_util.h"
#include "base/message_loop/message_loop.h"
#include "mojo/public/cpp/system/wait.h"
#include "services/service_manager/public/cpp/service_info.h"

namespace tracing {

Service::Service() : collector_binding_(this), tracing_active_(false) {
  registry_.AddInterface<mojom::Factory>(this);
  registry_.AddInterface<mojom::Collector>(this);
  registry_.AddInterface<mojom::StartupPerformanceDataCollector>(this);
}
Service::~Service() {}

void Service::OnBindInterface(const service_manager::ServiceInfo& source_info,
                              const std::string& interface_name,
                              mojo::ScopedMessagePipeHandle interface_pipe) {
  registry_.BindInterface(source_info.identity, interface_name,
                          std::move(interface_pipe));
}

bool Service::OnServiceManagerConnectionLost() {
  // TODO(beng): This is only required because Service isn't run by
  // ServiceRunner - instead it's launched automatically by the standalone
  // service manager. It shouldn't be.
  base::MessageLoop::current()->QuitWhenIdle();
  return false;
}

void Service::Create(const service_manager::Identity& remote_identity,
                     mojom::FactoryRequest request) {
  bindings_.AddBinding(this, std::move(request));
}

void Service::Create(const service_manager::Identity& remote_identity,
                     mojom::CollectorRequest request) {
  collector_binding_.Bind(std::move(request));
}

void Service::Create(const service_manager::Identity& remote_identity,
                     mojom::StartupPerformanceDataCollectorRequest request) {
  startup_performance_data_collector_bindings_.AddBinding(this,
                                                          std::move(request));
}

void Service::CreateRecorder(mojom::ProviderPtr provider) {
  if (tracing_active_) {
    mojom::RecorderPtr recorder_ptr;
    recorder_impls_.push_back(
        base::MakeUnique<Recorder>(MakeRequest(&recorder_ptr), sink_.get()));
    provider->StartTracing(tracing_categories_, std::move(recorder_ptr));
  }
  provider_ptrs_.AddPtr(std::move(provider));
}

void Service::Start(mojo::ScopedDataPipeProducerHandle stream,
                    const std::string& categories) {
  tracing_categories_ = categories;
  sink_.reset(new DataSink(std::move(stream)));
  provider_ptrs_.ForAllPtrs(
    [categories, this](mojom::Provider* controller) {
      mojom::RecorderPtr ptr;
      recorder_impls_.push_back(
          base::MakeUnique<Recorder>(MakeRequest(&ptr), sink_.get()));
      controller->StartTracing(categories, std::move(ptr));
    });
  tracing_active_ = true;
}

void Service::StopAndFlush() {
  // Remove any collectors that closed their message pipes before we called
  // StopTracing().
  for (int i = static_cast<int>(recorder_impls_.size()) - 1; i >= 0; --i) {
    if (!recorder_impls_[i]->RecorderHandle().is_valid()) {
      recorder_impls_.erase(recorder_impls_.begin() + i);
    }
  }

  tracing_active_ = false;
  provider_ptrs_.ForAllPtrs(
      [](mojom::Provider* controller) { controller->StopTracing(); });

  // Sending the StopTracing message to registered controllers will request that
  // they send trace data back via the collector interface and, when they are
  // done, close the collector pipe. We don't know how long they will take. We
  // want to read all data that any collector might send until all collectors or
  // closed or an (arbitrary) deadline has passed. Since the bindings don't
  // support this directly we do our own WaitMany over the handles and read
  // individual messages until all are closed or our absolute deadline has
  // elapsed.
  static const MojoDeadline kTimeToWaitMicros = 1000 * 1000;
  MojoTimeTicks end = MojoGetTimeTicksNow() + kTimeToWaitMicros;

  while (!recorder_impls_.empty()) {
    MojoTimeTicks now = MojoGetTimeTicksNow();
    if (now >= end)  // Timed out?
      break;

    std::vector<mojo::Handle> handles;
    std::vector<MojoHandleSignals> signals;
    for (const auto& it : recorder_impls_) {
      handles.push_back(it->RecorderHandle());
      signals.push_back(MOJO_HANDLE_SIGNAL_READABLE |
                        MOJO_HANDLE_SIGNAL_PEER_CLOSED);
    }
    std::vector<MojoHandleSignalsState> signals_states(signals.size());
    size_t result_index;

    // TODO(rockot): Use a timed wait here to avoid hanging forever in the case
    // of a misbehaving or unresponsive collector.
    MojoResult rv =
        mojo::WaitMany(handles.data(), signals.data(), handles.size(),
                       &result_index, signals_states.data());
    if (rv == MOJO_RESULT_OK || rv == MOJO_RESULT_FAILED_PRECONDITION) {
      // Iterate backwards so we can remove closed pipes from |recorder_impls_|
      // without invalidating subsequent offsets.
      for (size_t i = signals_states.size(); i != 0; --i) {
        size_t index = i - 1;
        MojoHandleSignals satisfied = signals_states[index].satisfied_signals;
        // To avoid dropping data, don't close unless there's no
        // readable signal.
        if (satisfied & MOJO_HANDLE_SIGNAL_READABLE)
          recorder_impls_[index]->TryRead();
        else if (satisfied & MOJO_HANDLE_SIGNAL_PEER_CLOSED)
          recorder_impls_.erase(recorder_impls_.begin() + index);
      }
    }
  }
  AllDataCollected();
}

void Service::SetServiceManagerProcessCreationTime(int64_t time) {
  if (startup_performance_times_.service_manager_process_creation_time == 0)
    startup_performance_times_.service_manager_process_creation_time = time;
}

void Service::SetServiceManagerMainEntryPointTime(int64_t time) {
  if (startup_performance_times_.service_manager_main_entry_point_time == 0)
    startup_performance_times_.service_manager_main_entry_point_time = time;
}

void Service::SetBrowserMessageLoopStartTicks(int64_t ticks) {
  if (startup_performance_times_.browser_message_loop_start_ticks == 0)
    startup_performance_times_.browser_message_loop_start_ticks = ticks;
}

void Service::SetBrowserWindowDisplayTicks(int64_t ticks) {
  if (startup_performance_times_.browser_window_display_ticks == 0)
    startup_performance_times_.browser_window_display_ticks = ticks;
}

void Service::SetBrowserOpenTabsTimeDelta(int64_t delta) {
  if (startup_performance_times_.browser_open_tabs_time_delta == 0)
    startup_performance_times_.browser_open_tabs_time_delta = delta;
}

void Service::SetFirstWebContentsMainFrameLoadTicks(int64_t ticks) {
  if (startup_performance_times_.first_web_contents_main_frame_load_ticks == 0)
    startup_performance_times_.first_web_contents_main_frame_load_ticks = ticks;
}

void Service::SetFirstVisuallyNonEmptyLayoutTicks(int64_t ticks) {
  if (startup_performance_times_.first_visually_non_empty_layout_ticks == 0)
    startup_performance_times_.first_visually_non_empty_layout_ticks = ticks;
}

void Service::GetStartupPerformanceTimes(
    const GetStartupPerformanceTimesCallback& callback) {
  callback.Run(startup_performance_times_.Clone());
}

void Service::AllDataCollected() {
  recorder_impls_.clear();
  sink_.reset();
}

}  // namespace tracing