blob: e9f45a968856e81e44e2e64e6f6f8bb2cb19bb14 [file] [log] [blame]
/*
* Copyright (C) 2019 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <fcntl.h>
#include <chrono>
#include <condition_variable>
#include <functional>
#include <list>
#include <mutex>
#include <regex>
#include <thread>
#include <vector>
// We also want to test legacy trace events.
#define PERFETTO_ENABLE_LEGACY_TRACE_EVENTS 1
#include "perfetto/tracing.h"
#include "test/gtest_and_gmock.h"
#if PERFETTO_BUILDFLAG(PERFETTO_OS_WIN)
#include <Windows.h> // For CreateFile().
#endif
// Deliberately not pulling any non-public perfetto header to spot accidental
// header public -> non-public dependency while building this file.
// These two are the only headers allowed here, see comments in
// api_test_support.h.
#include "src/tracing/test/api_test_support.h"
#include "src/tracing/test/tracing_module.h"
#include "perfetto/tracing/core/data_source_descriptor.h"
#include "perfetto/tracing/core/trace_config.h"
// xxx.pbzero.h includes are for the writing path (the code that pretends to be
// production code).
// yyy.gen.h includes are for the test readback path (the code in the test that
// checks that the results are valid).
#include "protos/perfetto/common/builtin_clock.pbzero.h"
#include "protos/perfetto/common/interceptor_descriptor.gen.h"
#include "protos/perfetto/common/trace_stats.gen.h"
#include "protos/perfetto/common/tracing_service_state.gen.h"
#include "protos/perfetto/common/track_event_descriptor.gen.h"
#include "protos/perfetto/config/interceptor_config.gen.h"
#include "protos/perfetto/config/track_event/track_event_config.gen.h"
#include "protos/perfetto/trace/clock_snapshot.pbzero.h"
#include "protos/perfetto/trace/gpu/gpu_render_stage_event.gen.h"
#include "protos/perfetto/trace/gpu/gpu_render_stage_event.pbzero.h"
#include "protos/perfetto/trace/interned_data/interned_data.gen.h"
#include "protos/perfetto/trace/interned_data/interned_data.pbzero.h"
#include "protos/perfetto/trace/profiling/profile_common.gen.h"
#include "protos/perfetto/trace/test_event.gen.h"
#include "protos/perfetto/trace/test_event.pbzero.h"
#include "protos/perfetto/trace/trace.gen.h"
#include "protos/perfetto/trace/trace.pbzero.h"
#include "protos/perfetto/trace/trace_packet.gen.h"
#include "protos/perfetto/trace/trace_packet.pbzero.h"
#include "protos/perfetto/trace/track_event/chrome_process_descriptor.gen.h"
#include "protos/perfetto/trace/track_event/chrome_process_descriptor.pbzero.h"
#include "protos/perfetto/trace/track_event/debug_annotation.gen.h"
#include "protos/perfetto/trace/track_event/debug_annotation.pbzero.h"
#include "protos/perfetto/trace/track_event/log_message.gen.h"
#include "protos/perfetto/trace/track_event/log_message.pbzero.h"
#include "protos/perfetto/trace/track_event/process_descriptor.gen.h"
#include "protos/perfetto/trace/track_event/process_descriptor.pbzero.h"
#include "protos/perfetto/trace/track_event/source_location.gen.h"
#include "protos/perfetto/trace/track_event/source_location.pbzero.h"
#include "protos/perfetto/trace/track_event/thread_descriptor.gen.h"
#include "protos/perfetto/trace/track_event/thread_descriptor.pbzero.h"
#include "protos/perfetto/trace/track_event/track_descriptor.gen.h"
#include "protos/perfetto/trace/track_event/track_event.gen.h"
// Events in categories starting with "dynamic" will use dynamic category
// lookup.
PERFETTO_DEFINE_TEST_CATEGORY_PREFIXES("dynamic");
// Trace categories used in the tests.
PERFETTO_DEFINE_CATEGORIES(
perfetto::Category("test")
.SetDescription("This is a test category")
.SetTags("tag"),
perfetto::Category("foo"),
perfetto::Category("bar"),
perfetto::Category("cat").SetTags("slow"),
perfetto::Category("cat.verbose").SetTags("debug"),
perfetto::Category("cat-with-dashes"),
perfetto::Category("cat with spaces"),
perfetto::Category::Group("foo,bar"),
perfetto::Category::Group("baz,bar,quux"),
perfetto::Category::Group("red,green,blue,foo"),
perfetto::Category::Group("red,green,blue,yellow"),
perfetto::Category(TRACE_DISABLED_BY_DEFAULT("cat")));
PERFETTO_TRACK_EVENT_STATIC_STORAGE();
// For testing interning of complex objects.
using SourceLocation = std::tuple<const char* /* file_name */,
const char* /* function_name */,
uint32_t /* line_number */>;
namespace std {
template <>
struct hash<SourceLocation> {
size_t operator()(const SourceLocation& value) const {
auto hasher = hash<size_t>();
return hasher(reinterpret_cast<size_t>(get<0>(value))) ^
hasher(reinterpret_cast<size_t>(get<1>(value))) ^
hasher(get<2>(value));
}
};
} // namespace std
// Represents an opaque (from Perfetto's point of view) thread identifier (e.g.,
// base::PlatformThreadId in Chromium).
struct MyThreadId {
MyThreadId(int pid_, int tid_) : pid(pid_), tid(tid_) {}
const int pid = 0;
const int tid = 0;
};
// Represents an opaque timestamp (e.g., base::TimeTicks in Chromium).
class MyTimestamp {
public:
explicit MyTimestamp(uint64_t ts_) : ts(ts_) {}
const uint64_t ts;
};
namespace perfetto {
namespace legacy {
template <>
bool ConvertThreadId(const MyThreadId& thread,
uint64_t* track_uuid_out,
int32_t* pid_override_out,
int32_t* tid_override_out) {
if (!thread.pid && !thread.tid)
return false;
if (!thread.pid) {
// Thread in current process.
*track_uuid_out = perfetto::ThreadTrack::ForThread(
static_cast<base::PlatformThreadId>(thread.tid))
.uuid;
} else {
// Thread in another process.
*pid_override_out = thread.pid;
*tid_override_out = thread.tid;
}
return true;
}
} // namespace legacy
template <>
TraceTimestamp ConvertTimestampToTraceTimeNs(const MyTimestamp& timestamp) {
return {TrackEvent::GetTraceClockId(), timestamp.ts};
}
} // namespace perfetto
namespace {
using ::testing::_;
using ::testing::ContainerEq;
using ::testing::ElementsAre;
using ::testing::HasSubstr;
using ::testing::Invoke;
using ::testing::InvokeWithoutArgs;
using ::testing::NiceMock;
using ::testing::Not;
using ::testing::Property;
using ::testing::StrEq;
// ------------------------------
// Declarations of helper classes
// ------------------------------
class WaitableTestEvent {
public:
bool notified() {
std::unique_lock<std::mutex> lock(mutex_);
return notified_;
}
void Wait() {
std::unique_lock<std::mutex> lock(mutex_);
// TSAN gets confused by wait_for, which we would use here in a perfect
// world.
cv_.wait(lock, [this] { return notified_; });
}
void Notify() {
{
std::lock_guard<std::mutex> lock(mutex_);
notified_ = true;
}
// Do not notify while holding the lock, because then we wake up the other
// end, only for it to fail to acquire the lock.
cv_.notify_one();
}
private:
std::mutex mutex_;
std::condition_variable cv_;
bool notified_ = false;
};
class MockDataSource;
// We can't easily use gmock here because instances of data sources are lazily
// created by the service and are not owned by the test fixture.
struct TestDataSourceHandle {
WaitableTestEvent on_create;
WaitableTestEvent on_setup;
WaitableTestEvent on_start;
WaitableTestEvent on_stop;
MockDataSource* instance;
perfetto::DataSourceConfig config;
bool handle_stop_asynchronously = false;
std::function<void()> on_start_callback;
std::function<void()> on_stop_callback;
std::function<void()> async_stop_closure;
};
class MockDataSource : public perfetto::DataSource<MockDataSource> {
public:
void OnSetup(const SetupArgs&) override;
void OnStart(const StartArgs&) override;
void OnStop(const StopArgs&) override;
TestDataSourceHandle* handle_ = nullptr;
};
constexpr int kTestDataSourceArg = 123;
class MockDataSource2 : public perfetto::DataSource<MockDataSource2> {
public:
MockDataSource2(int arg) { EXPECT_EQ(arg, kTestDataSourceArg); }
void OnSetup(const SetupArgs&) override {}
void OnStart(const StartArgs&) override {}
void OnStop(const StopArgs&) override {}
};
// Used to verify that track event data sources in different namespaces register
// themselves correctly in the muxer.
class MockTracingMuxer : public perfetto::internal::TracingMuxer {
public:
struct DataSource {
const perfetto::DataSourceDescriptor dsd;
perfetto::internal::DataSourceStaticState* static_state;
};
MockTracingMuxer() : TracingMuxer(nullptr), prev_instance_(instance_) {
instance_ = this;
}
~MockTracingMuxer() override { instance_ = prev_instance_; }
bool RegisterDataSource(
const perfetto::DataSourceDescriptor& dsd,
DataSourceFactory,
perfetto::internal::DataSourceStaticState* static_state) override {
data_sources.emplace_back(DataSource{dsd, static_state});
return true;
}
std::unique_ptr<perfetto::TraceWriterBase> CreateTraceWriter(
perfetto::internal::DataSourceStaticState*,
uint32_t,
perfetto::internal::DataSourceState*,
perfetto::BufferExhaustedPolicy) override {
return nullptr;
}
void DestroyStoppedTraceWritersForCurrentThread() override {}
void RegisterInterceptor(
const perfetto::InterceptorDescriptor&,
InterceptorFactory,
perfetto::InterceptorBase::TLSFactory,
perfetto::InterceptorBase::TracePacketCallback) override {}
std::vector<DataSource> data_sources;
private:
TracingMuxer* prev_instance_;
};
struct TestIncrementalState {
TestIncrementalState() { constructed = true; }
// Note: a virtual destructor is not required for incremental state.
~TestIncrementalState() { destroyed = true; }
int count = 100;
static bool constructed;
static bool destroyed;
};
bool TestIncrementalState::constructed;
bool TestIncrementalState::destroyed;
struct TestIncrementalDataSourceTraits
: public perfetto::DefaultDataSourceTraits {
using IncrementalStateType = TestIncrementalState;
};
class TestIncrementalDataSource
: public perfetto::DataSource<TestIncrementalDataSource,
TestIncrementalDataSourceTraits> {
public:
void OnSetup(const SetupArgs&) override {}
void OnStart(const StartArgs&) override {}
void OnStop(const StopArgs&) override {}
};
// A convenience wrapper around TracingSession that allows to do block on
//
struct TestTracingSessionHandle {
perfetto::TracingSession* get() { return session.get(); }
std::unique_ptr<perfetto::TracingSession> session;
WaitableTestEvent on_stop;
};
class MyDebugAnnotation : public perfetto::DebugAnnotation {
public:
~MyDebugAnnotation() override = default;
void Add(
perfetto::protos::pbzero::DebugAnnotation* annotation) const override {
annotation->set_legacy_json_value(R"({"key": 123})");
}
};
// -------------------------
// Declaration of test class
// -------------------------
class PerfettoApiTest : public ::testing::TestWithParam<perfetto::BackendType> {
public:
static PerfettoApiTest* instance;
void SetUp() override {
instance = this;
// Start a fresh system service for this test, tearing down any previous
// service that was running.
uint32_t supported_backends =
perfetto::kInProcessBackend | perfetto::kSystemBackend;
if (!perfetto::test::StartSystemService())
supported_backends &= ~perfetto::kSystemBackend;
// If the system backend wasn't supported, skip all system backend tests.
auto backend = GetParam();
if (!(supported_backends & backend))
GTEST_SKIP();
static bool was_initialized;
if (!was_initialized) {
EXPECT_FALSE(perfetto::Tracing::IsInitialized());
was_initialized = true;
} else {
EXPECT_TRUE(perfetto::Tracing::IsInitialized());
}
// Since the client API can only be initialized once per process, initialize
// both the in-process and system backends for every test here. The actual
// service to be used is chosen by the test parameter.
perfetto::TracingInitArgs args;
args.backends = supported_backends;
perfetto::Tracing::Initialize(args);
RegisterDataSource<MockDataSource>("my_data_source");
perfetto::TrackEvent::Register();
// Make sure our data source always has a valid handle.
data_sources_["my_data_source"];
// If this wasn't the first test to run in this process, any producers
// connected to the old system service will have been disconnected by the
// service restarting above. Wait for all producers to connect again before
// proceeding with the test.
perfetto::test::SyncProducers();
perfetto::test::DisableReconnectLimit();
}
void TearDown() override { instance = nullptr; }
template <typename DataSourceType>
TestDataSourceHandle* RegisterDataSource(std::string name) {
EXPECT_EQ(data_sources_.count(name), 0u);
TestDataSourceHandle* handle = &data_sources_[name];
perfetto::DataSourceDescriptor dsd;
dsd.set_name(name);
DataSourceType::Register(dsd);
return handle;
}
TestTracingSessionHandle* NewTrace(const perfetto::TraceConfig& cfg,
int fd = -1) {
sessions_.emplace_back();
TestTracingSessionHandle* handle = &sessions_.back();
handle->session = perfetto::Tracing::NewTrace(/*BackendType=*/GetParam());
handle->session->SetOnStopCallback([handle] { handle->on_stop.Notify(); });
handle->session->Setup(cfg, fd);
return handle;
}
TestTracingSessionHandle* NewTraceWithCategories(
std::vector<std::string> categories) {
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
for (const auto& category : categories)
te_cfg.add_enabled_categories(category);
ds_cfg->set_track_event_config_raw(te_cfg.SerializeAsString());
return NewTrace(cfg);
}
std::vector<std::string> ReadLogMessagesFromTrace(
perfetto::TracingSession* tracing_session) {
std::vector<char> raw_trace = tracing_session->ReadTraceBlocking();
EXPECT_GE(raw_trace.size(), 0u);
// Read back the trace, maintaining interning tables as we go.
std::vector<std::string> log_messages;
std::map<uint64_t, std::string> log_message_bodies;
std::map<uint64_t, perfetto::protos::gen::SourceLocation> source_locations;
perfetto::protos::gen::Trace parsed_trace;
EXPECT_TRUE(
parsed_trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
for (const auto& packet : parsed_trace.packet()) {
if (!packet.has_track_event())
continue;
if (packet.has_interned_data()) {
const auto& interned_data = packet.interned_data();
for (const auto& it : interned_data.log_message_body()) {
EXPECT_GE(it.iid(), 1u);
EXPECT_EQ(log_message_bodies.find(it.iid()),
log_message_bodies.end());
log_message_bodies[it.iid()] = it.body();
}
for (const auto& it : interned_data.source_locations()) {
EXPECT_GE(it.iid(), 1u);
EXPECT_EQ(source_locations.find(it.iid()), source_locations.end());
source_locations[it.iid()] = it;
}
}
const auto& track_event = packet.track_event();
if (track_event.type() !=
perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN)
continue;
EXPECT_TRUE(track_event.has_log_message());
const auto& log = track_event.log_message();
if (log.source_location_iid()) {
std::stringstream msg;
const auto& source_location =
source_locations[log.source_location_iid()];
msg << source_location.function_name() << "("
<< source_location.file_name() << ":"
<< source_location.line_number()
<< "): " << log_message_bodies[log.body_iid()];
log_messages.emplace_back(msg.str());
} else {
log_messages.emplace_back(log_message_bodies[log.body_iid()]);
}
}
return log_messages;
}
std::vector<std::string> ReadSlicesFromTrace(
perfetto::TracingSession* tracing_session) {
std::vector<char> raw_trace = tracing_session->ReadTraceBlocking();
EXPECT_GE(raw_trace.size(), 0u);
// Read back the trace, maintaining interning tables as we go.
std::vector<std::string> slices;
std::map<uint64_t, std::string> categories;
std::map<uint64_t, std::string> event_names;
std::map<uint64_t, std::string> debug_annotation_names;
perfetto::protos::gen::Trace parsed_trace;
EXPECT_TRUE(
parsed_trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
bool incremental_state_was_cleared = false;
uint32_t sequence_id = 0;
for (const auto& packet : parsed_trace.packet()) {
if (packet.sequence_flags() & perfetto::protos::pbzero::TracePacket::
SEQ_INCREMENTAL_STATE_CLEARED) {
incremental_state_was_cleared = true;
categories.clear();
event_names.clear();
debug_annotation_names.clear();
}
if (!packet.has_track_event())
continue;
// Make sure we only see track events on one sequence.
if (packet.trusted_packet_sequence_id()) {
if (!sequence_id)
sequence_id = packet.trusted_packet_sequence_id();
EXPECT_EQ(sequence_id, packet.trusted_packet_sequence_id());
}
// Update incremental state.
if (packet.has_interned_data()) {
const auto& interned_data = packet.interned_data();
for (const auto& it : interned_data.event_categories()) {
EXPECT_EQ(categories.find(it.iid()), categories.end());
categories[it.iid()] = it.name();
}
for (const auto& it : interned_data.event_names()) {
EXPECT_EQ(event_names.find(it.iid()), event_names.end());
event_names[it.iid()] = it.name();
}
for (const auto& it : interned_data.debug_annotation_names()) {
EXPECT_EQ(debug_annotation_names.find(it.iid()),
debug_annotation_names.end());
debug_annotation_names[it.iid()] = it.name();
}
}
const auto& track_event = packet.track_event();
std::string slice;
switch (track_event.type()) {
case perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN:
slice += "B";
break;
case perfetto::protos::gen::TrackEvent::TYPE_SLICE_END:
slice += "E";
break;
case perfetto::protos::gen::TrackEvent::TYPE_INSTANT:
slice += "I";
break;
case perfetto::protos::gen::TrackEvent::TYPE_UNSPECIFIED: {
EXPECT_TRUE(track_event.has_legacy_event());
EXPECT_FALSE(track_event.type());
auto legacy_event = track_event.legacy_event();
slice += "Legacy_" +
std::string(1, static_cast<char>(legacy_event.phase()));
break;
}
case perfetto::protos::gen::TrackEvent::TYPE_COUNTER:
slice += "C";
break;
default:
ADD_FAILURE();
}
if (track_event.has_legacy_event()) {
auto legacy_event = track_event.legacy_event();
std::stringstream id;
if (legacy_event.has_unscoped_id()) {
id << "(unscoped_id=" << legacy_event.unscoped_id() << ")";
} else if (legacy_event.has_local_id()) {
id << "(local_id=" << legacy_event.local_id() << ")";
} else if (legacy_event.has_global_id()) {
id << "(global_id=" << legacy_event.global_id() << ")";
} else if (legacy_event.has_bind_id()) {
id << "(bind_id=" << legacy_event.bind_id() << ")";
}
if (legacy_event.has_id_scope())
id << "(id_scope=\"" << legacy_event.id_scope() << "\")";
if (legacy_event.use_async_tts())
id << "(use_async_tts)";
if (legacy_event.bind_to_enclosing())
id << "(bind_to_enclosing)";
if (legacy_event.has_flow_direction())
id << "(flow_direction=" << legacy_event.flow_direction() << ")";
if (legacy_event.has_pid_override())
id << "(pid_override=" << legacy_event.pid_override() << ")";
if (legacy_event.has_tid_override())
id << "(tid_override=" << legacy_event.tid_override() << ")";
slice += id.str();
}
size_t category_count = 0;
for (const auto& it : track_event.category_iids())
slice += (category_count++ ? "," : ":") + categories[it];
for (const auto& it : track_event.categories())
slice += (category_count++ ? ",$" : ":$") + it;
if (track_event.has_name_iid())
slice += "." + event_names[track_event.name_iid()];
if (track_event.debug_annotations_size()) {
slice += "(";
bool first_annotation = true;
for (const auto& it : track_event.debug_annotations()) {
if (!first_annotation) {
slice += ",";
}
slice += debug_annotation_names[it.name_iid()] + "=";
std::stringstream value;
if (it.has_bool_value()) {
value << "(bool)" << it.bool_value();
} else if (it.has_uint_value()) {
value << "(uint)" << it.uint_value();
} else if (it.has_int_value()) {
value << "(int)" << it.int_value();
} else if (it.has_double_value()) {
value << "(double)" << it.double_value();
} else if (it.has_string_value()) {
value << "(string)" << it.string_value();
} else if (it.has_pointer_value()) {
value << "(pointer)" << std::hex << it.pointer_value();
} else if (it.has_legacy_json_value()) {
value << "(json)" << it.legacy_json_value();
} else if (it.has_nested_value()) {
value << "(nested)" << it.nested_value().string_value();
}
slice += value.str();
first_annotation = false;
}
slice += ")";
}
slices.push_back(slice);
}
EXPECT_TRUE(incremental_state_was_cleared);
return slices;
}
uint32_t GetMainThreadPacketSequenceId(
const perfetto::protos::gen::Trace& trace) {
for (const auto& packet : trace.packet()) {
if (packet.has_track_descriptor() &&
packet.track_descriptor().thread().tid() ==
static_cast<int32_t>(perfetto::base::GetThreadId())) {
return packet.trusted_packet_sequence_id();
}
}
ADD_FAILURE() << "Main thread not found";
return 0;
}
std::map<std::string, TestDataSourceHandle> data_sources_;
std::list<TestTracingSessionHandle> sessions_; // Needs stable pointers.
};
// ---------------------------------------------
// Definitions for non-inlineable helper methods
// ---------------------------------------------
PerfettoApiTest* PerfettoApiTest::instance;
void MockDataSource::OnSetup(const SetupArgs& args) {
EXPECT_EQ(handle_, nullptr);
auto it = PerfettoApiTest::instance->data_sources_.find(args.config->name());
// We should not see an OnSetup for a data source that we didn't register
// before via PerfettoApiTest::RegisterDataSource().
EXPECT_NE(it, PerfettoApiTest::instance->data_sources_.end());
handle_ = &it->second;
handle_->config = *args.config; // Deliberate copy.
handle_->on_setup.Notify();
}
void MockDataSource::OnStart(const StartArgs&) {
EXPECT_NE(handle_, nullptr);
if (handle_->on_start_callback)
handle_->on_start_callback();
handle_->on_start.Notify();
}
void MockDataSource::OnStop(const StopArgs& args) {
EXPECT_NE(handle_, nullptr);
if (handle_->handle_stop_asynchronously)
handle_->async_stop_closure = args.HandleStopAsynchronously();
if (handle_->on_stop_callback)
handle_->on_stop_callback();
handle_->on_stop.Notify();
}
// -------------
// Test fixtures
// -------------
TEST_P(PerfettoApiTest, StartAndStopWithoutDataSources) {
// Create a new trace session without any data sources configured.
perfetto::TraceConfig cfg;
cfg.add_buffers()->set_size_kb(1024);
auto* tracing_session = NewTrace(cfg);
// This should not timeout.
tracing_session->get()->StartBlocking();
tracing_session->get()->StopBlocking();
}
TEST_P(PerfettoApiTest, TrackEventStartStopAndDestroy) {
// This test used to cause a use after free as the tracing session got
// destroyed. It needed to be run approximately 2000 times to catch it so test
// with --gtest_repeat=3000 (less if running under GDB).
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
// Create five new trace sessions.
std::vector<std::unique_ptr<perfetto::TracingSession>> sessions;
for (size_t i = 0; i < 5; ++i) {
sessions.push_back(perfetto::Tracing::NewTrace(/*BackendType=*/GetParam()));
sessions[i]->Setup(cfg);
sessions[i]->Start();
sessions[i]->Stop();
}
}
TEST_P(PerfettoApiTest, TrackEventStartStopAndStopBlocking) {
// This test used to cause a deadlock (due to StopBlocking() after the session
// already stopped). This usually occurred within 1 or 2 runs of the test so
// use --gtest_repeat=10
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
// Create five new trace sessions.
std::vector<std::unique_ptr<perfetto::TracingSession>> sessions;
for (size_t i = 0; i < 5; ++i) {
sessions.push_back(perfetto::Tracing::NewTrace(/*BackendType=*/GetParam()));
sessions[i]->Setup(cfg);
sessions[i]->Start();
sessions[i]->Stop();
}
for (auto& session : sessions) {
session->StopBlocking();
}
}
TEST_P(PerfettoApiTest, ChangeTraceConfiguration) {
// Setup the trace config.
perfetto::TraceConfig trace_config;
trace_config.set_duration_ms(2000);
trace_config.add_buffers()->set_size_kb(1024);
auto* data_source = trace_config.add_data_sources();
// Configure track events with category "foo".
auto* ds_cfg = data_source->mutable_config();
ds_cfg->set_name("track_event");
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
te_cfg.add_enabled_categories("foo");
ds_cfg->set_track_event_config_raw(te_cfg.SerializeAsString());
// Initially, exclude all producers (the client library's producer is named
// after current process's name, which will not match
// "all_producers_excluded").
data_source->add_producer_name_filter("all_producers_excluded");
auto* tracing_session = NewTrace(trace_config);
tracing_session->get()->StartBlocking();
// Emit a first trace event, this one should be filtered out due
// to the mismatching producer name filter.
TRACE_EVENT_BEGIN("foo", "EventFilteredOut");
TRACE_EVENT_END("foo");
// Remove the producer name filter by changing configs.
data_source->clear_producer_name_filter();
tracing_session->get()->ChangeTraceConfig(trace_config);
// We don't have a blocking version of ChangeTraceConfig, because there is
// currently no response to it from producers or the service. Instead, we sync
// the consumer and producer IPC streams for this test, to ensure that the
// producer_name_filter change has propagated.
tracing_session->get()->GetTraceStatsBlocking(); // sync consumer stream.
perfetto::test::SyncProducers(); // sync producer stream.
// Emit a second trace event, this one should be included because
// the producer name filter was cleared.
TRACE_EVENT_BEGIN("foo", "EventIncluded");
TRACE_EVENT_END("foo");
tracing_session->get()->StopBlocking();
// Verify that only the second event is in the trace data.
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
std::string trace(raw_trace.data(), raw_trace.size());
EXPECT_THAT(trace, Not(HasSubstr("EventFilteredOut")));
EXPECT_THAT(trace, HasSubstr("EventIncluded"));
}
// This is a build-only regression test that checks you can have a track event
// inside a template.
template <typename T>
void TestTrackEventInsideTemplate(T) {
TRACE_EVENT_BEGIN("cat", "Name");
}
// This is a build-only regression test that checks you can specify the tracing
// category as a template argument.
constexpr const char kTestCategory[] = "foo";
template <const char* category>
void TestCategoryAsTemplateParameter() {
TRACE_EVENT_BEGIN(category, "Name");
}
TEST_P(PerfettoApiTest, TrackEvent) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
// Emit one complete track event.
TRACE_EVENT_BEGIN("test", "TestEvent");
TRACE_EVENT_END("test");
perfetto::TrackEvent::Flush();
tracing_session->on_stop.Wait();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
ASSERT_GE(raw_trace.size(), 0u);
// Read back the trace, maintaining interning tables as we go.
perfetto::protos::gen::Trace trace;
std::map<uint64_t, std::string> categories;
std::map<uint64_t, std::string> event_names;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
auto now = perfetto::TrackEvent::GetTraceTimeNs();
#if !PERFETTO_BUILDFLAG(PERFETTO_OS_APPLE) && \
!PERFETTO_BUILDFLAG(PERFETTO_OS_WIN)
auto clock_id = perfetto::protos::pbzero::BUILTIN_CLOCK_BOOTTIME;
#else
auto clock_id = perfetto::protos::pbzero::BUILTIN_CLOCK_MONOTONIC;
#endif
EXPECT_EQ(clock_id, perfetto::TrackEvent::GetTraceClockId());
bool incremental_state_was_cleared = false;
bool begin_found = false;
bool end_found = false;
bool process_descriptor_found = false;
uint32_t sequence_id = 0;
int32_t cur_pid = perfetto::test::GetCurrentProcessId();
for (const auto& packet : trace.packet()) {
if (packet.has_track_descriptor()) {
const auto& desc = packet.track_descriptor();
if (desc.has_process()) {
EXPECT_FALSE(process_descriptor_found);
const auto& pd = desc.process();
EXPECT_EQ(cur_pid, pd.pid());
process_descriptor_found = true;
}
}
if (packet.sequence_flags() &
perfetto::protos::pbzero::TracePacket::SEQ_INCREMENTAL_STATE_CLEARED) {
EXPECT_TRUE(packet.has_trace_packet_defaults());
incremental_state_was_cleared = true;
categories.clear();
event_names.clear();
}
if (!packet.has_track_event())
continue;
EXPECT_TRUE(
packet.sequence_flags() &
(perfetto::protos::pbzero::TracePacket::SEQ_INCREMENTAL_STATE_CLEARED |
perfetto::protos::pbzero::TracePacket::SEQ_NEEDS_INCREMENTAL_STATE));
const auto& track_event = packet.track_event();
// Make sure we only see track events on one sequence.
if (packet.trusted_packet_sequence_id()) {
if (!sequence_id)
sequence_id = packet.trusted_packet_sequence_id();
EXPECT_EQ(sequence_id, packet.trusted_packet_sequence_id());
}
// Update incremental state.
if (packet.has_interned_data()) {
const auto& interned_data = packet.interned_data();
for (const auto& it : interned_data.event_categories()) {
EXPECT_EQ(categories.find(it.iid()), categories.end());
categories[it.iid()] = it.name();
}
for (const auto& it : interned_data.event_names()) {
EXPECT_EQ(event_names.find(it.iid()), event_names.end());
event_names[it.iid()] = it.name();
}
}
EXPECT_GT(packet.timestamp(), 0u);
EXPECT_LE(packet.timestamp(), now);
#if !PERFETTO_BUILDFLAG(PERFETTO_OS_APPLE) && \
!PERFETTO_BUILDFLAG(PERFETTO_OS_WIN)
EXPECT_FALSE(packet.has_timestamp_clock_id());
#else
constexpr auto kClockMonotonic =
perfetto::protos::pbzero::ClockSnapshot::Clock::MONOTONIC;
EXPECT_EQ(packet.timestamp_clock_id(),
static_cast<uint32_t>(kClockMonotonic));
#endif
if (track_event.type() ==
perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN) {
EXPECT_FALSE(begin_found);
EXPECT_EQ(track_event.category_iids().size(), 1u);
EXPECT_GE(track_event.category_iids()[0], 1u);
EXPECT_EQ("test", categories[track_event.category_iids()[0]]);
EXPECT_EQ("TestEvent", event_names[track_event.name_iid()]);
begin_found = true;
} else if (track_event.type() ==
perfetto::protos::gen::TrackEvent::TYPE_SLICE_END) {
EXPECT_FALSE(end_found);
EXPECT_EQ(track_event.category_iids().size(), 0u);
EXPECT_EQ(0u, track_event.name_iid());
end_found = true;
}
}
EXPECT_TRUE(incremental_state_was_cleared);
EXPECT_TRUE(process_descriptor_found);
EXPECT_TRUE(begin_found);
EXPECT_TRUE(end_found);
// Dummy instantiation of test templates.
TestTrackEventInsideTemplate(true);
TestCategoryAsTemplateParameter<kTestCategory>();
}
TEST_P(PerfettoApiTest, TrackEventCategories) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"bar"});
tracing_session->get()->StartBlocking();
// Emit some track events.
TRACE_EVENT_BEGIN("foo", "NotEnabled");
TRACE_EVENT_END("foo");
TRACE_EVENT_BEGIN("bar", "Enabled");
TRACE_EVENT_END("bar");
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
std::string trace(raw_trace.data(), raw_trace.size());
// TODO(skyostil): Come up with a nicer way to verify trace contents.
EXPECT_THAT(trace, HasSubstr("Enabled"));
EXPECT_THAT(trace, Not(HasSubstr("NotEnabled")));
}
TEST_P(PerfettoApiTest, TrackEventRegistrationWithModule) {
MockTracingMuxer muxer;
// Each track event namespace registers its own data source.
perfetto::TrackEvent::Register();
EXPECT_EQ(1u, muxer.data_sources.size());
tracing_module::InitializeCategories();
EXPECT_EQ(2u, muxer.data_sources.size());
// Both data sources have the same name but distinct static data (i.e.,
// individual instance states).
EXPECT_EQ("track_event", muxer.data_sources[0].dsd.name());
EXPECT_EQ("track_event", muxer.data_sources[1].dsd.name());
EXPECT_NE(muxer.data_sources[0].static_state,
muxer.data_sources[1].static_state);
}
TEST_P(PerfettoApiTest, TrackEventDescriptor) {
MockTracingMuxer muxer;
perfetto::TrackEvent::Register();
EXPECT_EQ(1u, muxer.data_sources.size());
EXPECT_EQ("track_event", muxer.data_sources[0].dsd.name());
perfetto::protos::gen::TrackEventDescriptor desc;
auto desc_raw = muxer.data_sources[0].dsd.track_event_descriptor_raw();
EXPECT_TRUE(desc.ParseFromArray(desc_raw.data(), desc_raw.size()));
// Check that the advertised categories match PERFETTO_DEFINE_CATEGORIES (see
// above).
EXPECT_EQ(8, desc.available_categories_size());
EXPECT_EQ("test", desc.available_categories()[0].name());
EXPECT_EQ("This is a test category",
desc.available_categories()[0].description());
EXPECT_EQ("tag", desc.available_categories()[0].tags()[0]);
EXPECT_EQ("foo", desc.available_categories()[1].name());
EXPECT_EQ("bar", desc.available_categories()[2].name());
EXPECT_EQ("cat", desc.available_categories()[3].name());
EXPECT_EQ("slow", desc.available_categories()[3].tags()[0]);
EXPECT_EQ("cat.verbose", desc.available_categories()[4].name());
EXPECT_EQ("debug", desc.available_categories()[4].tags()[0]);
EXPECT_EQ("cat-with-dashes", desc.available_categories()[5].name());
EXPECT_EQ("cat with spaces", desc.available_categories()[6].name());
EXPECT_EQ("disabled-by-default-cat", desc.available_categories()[7].name());
EXPECT_EQ("slow", desc.available_categories()[7].tags()[0]);
}
TEST_P(PerfettoApiTest, TrackEventSharedIncrementalState) {
tracing_module::InitializeCategories();
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
perfetto::internal::TrackEventIncrementalState* main_state = nullptr;
perfetto::TrackEvent::Trace(
[&main_state](perfetto::TrackEvent::TraceContext ctx) {
main_state = ctx.GetIncrementalState();
});
perfetto::internal::TrackEventIncrementalState* module_state =
tracing_module::GetIncrementalState();
// Both track event data sources should use the same incremental state (thanks
// to sharing TLS).
EXPECT_NE(nullptr, main_state);
EXPECT_EQ(main_state, module_state);
tracing_session->get()->StopBlocking();
}
TEST_P(PerfettoApiTest, TrackEventCategoriesWithModule) {
// Check that categories defined in two different category registries are
// enabled and disabled correctly.
tracing_module::InitializeCategories();
// Create a new trace session. Only the "foo" category is enabled. It also
// exists both locally and in the existing module.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
// Emit some track events locally and from the test module.
TRACE_EVENT_BEGIN("foo", "FooEventFromMain");
TRACE_EVENT_END("foo");
tracing_module::EmitTrackEvents();
tracing_module::EmitTrackEvents2();
TRACE_EVENT_BEGIN("bar", "DisabledEventFromMain");
TRACE_EVENT_END("bar");
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
std::string trace(raw_trace.data(), raw_trace.size());
EXPECT_THAT(trace, HasSubstr("FooEventFromMain"));
EXPECT_THAT(trace, Not(HasSubstr("DisabledEventFromMain")));
EXPECT_THAT(trace, HasSubstr("FooEventFromModule"));
EXPECT_THAT(trace, Not(HasSubstr("DisabledEventFromModule")));
EXPECT_THAT(trace, HasSubstr("FooEventFromModule2"));
EXPECT_THAT(trace, Not(HasSubstr("DisabledEventFromModule2")));
perfetto::protos::gen::Trace parsed_trace;
ASSERT_TRUE(parsed_trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
uint32_t sequence_id = 0;
for (const auto& packet : parsed_trace.packet()) {
if (!packet.has_track_event())
continue;
// Make sure we only see track events on one sequence. This means all track
// event modules are sharing the same trace writer (by using the same TLS
// index).
if (packet.trusted_packet_sequence_id()) {
if (!sequence_id)
sequence_id = packet.trusted_packet_sequence_id();
EXPECT_EQ(sequence_id, packet.trusted_packet_sequence_id());
}
}
}
TEST_P(PerfettoApiTest, TrackEventDynamicCategories) {
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
// Session #1 enabled the "dynamic" category.
auto* tracing_session = NewTraceWithCategories({"dynamic"});
tracing_session->get()->StartBlocking();
// Session #2 enables "dynamic_2".
auto* tracing_session2 = NewTraceWithCategories({"dynamic_2"});
tracing_session2->get()->StartBlocking();
// Test naming dynamic categories with std::string.
perfetto::DynamicCategory dynamic{"dynamic"};
TRACE_EVENT_BEGIN(dynamic, "EventInDynamicCategory");
perfetto::DynamicCategory dynamic_disabled{"dynamic_disabled"};
TRACE_EVENT_BEGIN(dynamic_disabled, "EventInDisabledDynamicCategory");
// Test naming dynamic categories statically.
TRACE_EVENT_BEGIN("dynamic", "EventInStaticallyNamedDynamicCategory");
perfetto::DynamicCategory dynamic_2{"dynamic_2"};
TRACE_EVENT_BEGIN(dynamic_2, "EventInSecondDynamicCategory");
TRACE_EVENT_BEGIN("dynamic_2", "EventInSecondStaticallyNamedDynamicCategory");
std::thread thread([] {
// Make sure the category name can actually be computed at runtime.
std::string name = "dyn";
if (perfetto::base::GetThreadId())
name += "amic";
perfetto::DynamicCategory cat{name};
TRACE_EVENT_BEGIN(cat, "DynamicFromOtherThread");
perfetto::DynamicCategory cat2{"dynamic_disabled"};
TRACE_EVENT_BEGIN(cat2, "EventInDisabledDynamicCategory");
});
thread.join();
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
std::string trace(raw_trace.data(), raw_trace.size());
EXPECT_THAT(trace, HasSubstr("EventInDynamicCategory"));
EXPECT_THAT(trace, Not(HasSubstr("EventInDisabledDynamicCategory")));
EXPECT_THAT(trace, HasSubstr("DynamicFromOtherThread"));
EXPECT_THAT(trace, Not(HasSubstr("EventInSecondDynamicCategory")));
EXPECT_THAT(trace, HasSubstr("EventInStaticallyNamedDynamicCategory"));
EXPECT_THAT(trace,
Not(HasSubstr("EventInSecondStaticallyNamedDynamicCategory")));
tracing_session2->get()->StopBlocking();
raw_trace = tracing_session2->get()->ReadTraceBlocking();
trace = std::string(raw_trace.data(), raw_trace.size());
EXPECT_THAT(trace, Not(HasSubstr("EventInDynamicCategory")));
EXPECT_THAT(trace, Not(HasSubstr("EventInDisabledDynamicCategory")));
EXPECT_THAT(trace, Not(HasSubstr("DynamicFromOtherThread")));
EXPECT_THAT(trace, HasSubstr("EventInSecondDynamicCategory"));
EXPECT_THAT(trace, Not(HasSubstr("EventInStaticallyNamedDynamicCategory")));
EXPECT_THAT(trace, HasSubstr("EventInSecondStaticallyNamedDynamicCategory"));
}
TEST_P(PerfettoApiTest, TrackEventConcurrentSessions) {
// Check that categories that are enabled and disabled in two parallel tracing
// sessions don't interfere.
// Session #1 enables the "foo" category.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
// Session #2 enables the "bar" category.
auto* tracing_session2 = NewTraceWithCategories({"bar"});
tracing_session2->get()->StartBlocking();
// Emit some track events under both categories.
TRACE_EVENT_BEGIN("foo", "Session1_First");
TRACE_EVENT_END("foo");
TRACE_EVENT_BEGIN("bar", "Session2_First");
TRACE_EVENT_END("bar");
tracing_session->get()->StopBlocking();
TRACE_EVENT_BEGIN("foo", "Session1_Second");
TRACE_EVENT_END("foo");
TRACE_EVENT_BEGIN("bar", "Session2_Second");
TRACE_EVENT_END("bar");
tracing_session2->get()->StopBlocking();
TRACE_EVENT_BEGIN("foo", "Session1_Third");
TRACE_EVENT_END("foo");
TRACE_EVENT_BEGIN("bar", "Session2_Third");
TRACE_EVENT_END("bar");
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
std::string trace(raw_trace.data(), raw_trace.size());
EXPECT_THAT(trace, HasSubstr("Session1_First"));
EXPECT_THAT(trace, Not(HasSubstr("Session1_Second")));
EXPECT_THAT(trace, Not(HasSubstr("Session1_Third")));
EXPECT_THAT(trace, Not(HasSubstr("Session2_First")));
EXPECT_THAT(trace, Not(HasSubstr("Session2_Second")));
EXPECT_THAT(trace, Not(HasSubstr("Session2_Third")));
std::vector<char> raw_trace2 = tracing_session2->get()->ReadTraceBlocking();
std::string trace2(raw_trace2.data(), raw_trace2.size());
EXPECT_THAT(trace2, Not(HasSubstr("Session1_First")));
EXPECT_THAT(trace2, Not(HasSubstr("Session1_Second")));
EXPECT_THAT(trace2, Not(HasSubstr("Session1_Third")));
EXPECT_THAT(trace2, HasSubstr("Session2_First"));
EXPECT_THAT(trace2, HasSubstr("Session2_Second"));
EXPECT_THAT(trace2, Not(HasSubstr("Session2_Third")));
}
TEST_P(PerfettoApiTest, TrackEventProcessAndThreadDescriptors) {
// Thread and process descriptors can be set before tracing is enabled.
perfetto::TrackEvent::SetProcessDescriptor(
[](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("hello.exe");
desc->set_chrome_process()->set_process_priority(1);
});
// Erased tracks shouldn't show up anywhere.
perfetto::Track erased(1234u);
perfetto::TrackEvent::SetTrackDescriptor(
erased, [](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("ErasedTrack");
});
perfetto::TrackEvent::EraseTrackDescriptor(erased);
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
TRACE_EVENT_INSTANT("test", "MainThreadEvent");
std::thread thread([&] {
perfetto::TrackEvent::SetThreadDescriptor(
[](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("TestThread");
});
TRACE_EVENT_INSTANT("test", "ThreadEvent");
});
thread.join();
// Update the process descriptor while tracing is enabled. It should be
// immediately reflected in the trace.
perfetto::TrackEvent::SetProcessDescriptor(
[](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("goodbye.exe");
});
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
// After tracing ends, setting the descriptor has no immediate effect.
perfetto::TrackEvent::SetProcessDescriptor(
[](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("noop.exe");
});
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
std::vector<perfetto::protos::gen::TrackDescriptor> descs;
std::vector<perfetto::protos::gen::TrackDescriptor> thread_descs;
uint32_t main_thread_sequence = GetMainThreadPacketSequenceId(trace);
for (const auto& packet : trace.packet()) {
if (packet.has_track_descriptor()) {
if (packet.trusted_packet_sequence_id() == main_thread_sequence) {
descs.push_back(packet.track_descriptor());
} else {
thread_descs.push_back(packet.track_descriptor());
}
}
}
// The main thread records the initial process name as well as the one that's
// set during tracing. Additionally it records a thread descriptor for the
// main thread.
EXPECT_EQ(3u, descs.size());
// Default track for the main thread.
EXPECT_EQ(0, descs[0].process().pid());
EXPECT_NE(0, descs[0].thread().pid());
// First process descriptor.
EXPECT_NE(0, descs[1].process().pid());
EXPECT_EQ("hello.exe", descs[1].name());
// Second process descriptor.
EXPECT_NE(0, descs[2].process().pid());
EXPECT_EQ("goodbye.exe", descs[2].name());
// The child thread records only its own thread descriptor (twice, since it
// was mutated).
EXPECT_EQ(2u, thread_descs.size());
EXPECT_EQ("TestThread", thread_descs[0].name());
EXPECT_NE(0, thread_descs[0].thread().pid());
EXPECT_NE(0, thread_descs[0].thread().tid());
EXPECT_EQ("TestThread", thread_descs[1].name());
EXPECT_NE(0, thread_descs[1].thread().pid());
EXPECT_NE(0, thread_descs[1].thread().tid());
}
TEST_P(PerfettoApiTest, CustomTrackDescriptor) {
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
auto track = perfetto::ProcessTrack::Current();
auto desc = track.Serialize();
desc.mutable_process()->set_process_name("testing.exe");
desc.mutable_thread()->set_tid(
static_cast<int32_t>(perfetto::base::GetThreadId()));
desc.mutable_chrome_process()->set_process_priority(123);
perfetto::TrackEvent::SetTrackDescriptor(track, std::move(desc));
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
uint32_t main_thread_sequence = GetMainThreadPacketSequenceId(trace);
bool found_desc = false;
for (const auto& packet : trace.packet()) {
if (packet.trusted_packet_sequence_id() != main_thread_sequence)
continue;
if (packet.has_track_descriptor()) {
auto td = packet.track_descriptor();
EXPECT_TRUE(td.has_process());
EXPECT_NE(0, td.process().pid());
EXPECT_TRUE(td.has_chrome_process());
EXPECT_EQ("testing.exe", td.process().process_name());
EXPECT_EQ(123, td.chrome_process().process_priority());
found_desc = true;
}
}
EXPECT_TRUE(found_desc);
}
TEST_P(PerfettoApiTest, TrackEventCustomTrack) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"bar"});
tracing_session->get()->StartBlocking();
// Declare a custom track and give it a name.
uint64_t async_id = 123;
perfetto::TrackEvent::SetTrackDescriptor(
perfetto::Track(async_id),
[](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("MyCustomTrack");
});
// Start events on one thread and end them on another.
TRACE_EVENT_BEGIN("bar", "AsyncEvent", perfetto::Track(async_id), "debug_arg",
123);
TRACE_EVENT_BEGIN("bar", "SubEvent", perfetto::Track(async_id),
[](perfetto::EventContext) {});
const auto main_thread_track =
perfetto::Track(async_id, perfetto::ThreadTrack::Current());
std::thread thread([&] {
TRACE_EVENT_END("bar", perfetto::Track(async_id));
TRACE_EVENT_END("bar", perfetto::Track(async_id), "arg1", false, "arg2",
true);
const auto thread_track =
perfetto::Track(async_id, perfetto::ThreadTrack::Current());
// Thread-scoped tracks will have different uuids on different thread even
// if the id matches.
ASSERT_NE(main_thread_track.uuid, thread_track.uuid);
});
thread.join();
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
// Check that the track uuids match on the begin and end events.
const auto track = perfetto::Track(async_id);
uint32_t main_thread_sequence = GetMainThreadPacketSequenceId(trace);
int event_count = 0;
bool found_descriptor = false;
for (const auto& packet : trace.packet()) {
if (packet.has_track_descriptor() &&
!packet.track_descriptor().has_process() &&
!packet.track_descriptor().has_thread()) {
auto td = packet.track_descriptor();
EXPECT_EQ("MyCustomTrack", td.name());
EXPECT_EQ(track.uuid, td.uuid());
EXPECT_EQ(perfetto::ProcessTrack::Current().uuid, td.parent_uuid());
found_descriptor = true;
continue;
}
if (!packet.has_track_event())
continue;
auto track_event = packet.track_event();
if (track_event.type() ==
perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN) {
EXPECT_EQ(main_thread_sequence, packet.trusted_packet_sequence_id());
EXPECT_EQ(track.uuid, track_event.track_uuid());
} else {
EXPECT_NE(main_thread_sequence, packet.trusted_packet_sequence_id());
EXPECT_EQ(track.uuid, track_event.track_uuid());
}
event_count++;
}
EXPECT_TRUE(found_descriptor);
EXPECT_EQ(4, event_count);
perfetto::TrackEvent::EraseTrackDescriptor(track);
}
TEST_P(PerfettoApiTest, TrackEventCustomTrackAndTimestamp) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"bar"});
tracing_session->get()->StartBlocking();
// Custom track.
perfetto::Track track(789);
auto empty_lambda = [](perfetto::EventContext) {};
constexpr uint64_t kBeginEventTime = 10;
const MyTimestamp kEndEventTime{15};
TRACE_EVENT_BEGIN("bar", "Event", track, kBeginEventTime, empty_lambda);
TRACE_EVENT_END("bar", track, kEndEventTime, empty_lambda);
constexpr uint64_t kInstantEventTime = 1;
TRACE_EVENT_INSTANT("bar", "InstantEvent", track, kInstantEventTime,
empty_lambda);
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
int event_count = 0;
for (const auto& packet : trace.packet()) {
if (!packet.has_track_event())
continue;
event_count++;
switch (packet.track_event().type()) {
case perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN:
EXPECT_EQ(packet.timestamp(), kBeginEventTime);
break;
case perfetto::protos::gen::TrackEvent::TYPE_SLICE_END:
EXPECT_EQ(packet.timestamp(), kEndEventTime.ts);
break;
case perfetto::protos::gen::TrackEvent::TYPE_INSTANT:
EXPECT_EQ(packet.timestamp(), kInstantEventTime);
break;
case perfetto::protos::gen::TrackEvent::TYPE_COUNTER:
case perfetto::protos::gen::TrackEvent::TYPE_UNSPECIFIED:
ADD_FAILURE();
}
}
EXPECT_EQ(event_count, 3);
perfetto::TrackEvent::EraseTrackDescriptor(track);
}
TEST_P(PerfettoApiTest, TrackEventCustomTrackAndTimestampNoLambda) {
auto* tracing_session = NewTraceWithCategories({"bar"});
tracing_session->get()->StartBlocking();
perfetto::Track track(789);
constexpr uint64_t kBeginEventTime = 10;
constexpr uint64_t kEndEventTime = 15;
TRACE_EVENT_BEGIN("bar", "Event", track, kBeginEventTime);
TRACE_EVENT_END("bar", track, kEndEventTime);
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
int event_count = 0;
for (const auto& packet : trace.packet()) {
if (!packet.has_track_event())
continue;
event_count++;
switch (packet.track_event().type()) {
case perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN:
EXPECT_EQ(packet.timestamp(), kBeginEventTime);
break;
case perfetto::protos::gen::TrackEvent::TYPE_SLICE_END:
EXPECT_EQ(packet.timestamp(), kEndEventTime);
break;
case perfetto::protos::gen::TrackEvent::TYPE_INSTANT:
case perfetto::protos::gen::TrackEvent::TYPE_COUNTER:
case perfetto::protos::gen::TrackEvent::TYPE_UNSPECIFIED:
ADD_FAILURE();
}
}
EXPECT_EQ(event_count, 2);
}
TEST_P(PerfettoApiTest, TrackEventAnonymousCustomTrack) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"bar"});
tracing_session->get()->StartBlocking();
// Emit an async event without giving it an explicit descriptor.
uint64_t async_id = 4004;
auto track = perfetto::Track(async_id, perfetto::ThreadTrack::Current());
TRACE_EVENT_BEGIN("bar", "AsyncEvent", track);
std::thread thread([&] { TRACE_EVENT_END("bar", track); });
thread.join();
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
// Check that a descriptor for the track was emitted.
bool found_descriptor = false;
for (const auto& packet : trace.packet()) {
if (packet.has_track_descriptor() &&
!packet.track_descriptor().has_process() &&
!packet.track_descriptor().has_thread()) {
auto td = packet.track_descriptor();
EXPECT_EQ(track.uuid, td.uuid());
EXPECT_EQ(perfetto::ThreadTrack::Current().uuid, td.parent_uuid());
found_descriptor = true;
}
}
EXPECT_TRUE(found_descriptor);
}
TEST_P(PerfettoApiTest, TrackEventTypedArgs) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
auto random_value = rand();
TRACE_EVENT_BEGIN("foo", "EventWithTypedArg",
[random_value](perfetto::EventContext ctx) {
auto* log = ctx.event()->set_log_message();
log->set_source_location_iid(1);
log->set_body_iid(2);
auto* dbg = ctx.event()->add_debug_annotations();
dbg->set_name("random");
dbg->set_int_value(random_value);
});
TRACE_EVENT_END("foo");
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
std::string trace(raw_trace.data(), raw_trace.size());
perfetto::protos::gen::Trace parsed_trace;
ASSERT_TRUE(parsed_trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
bool found_args = false;
for (const auto& packet : parsed_trace.packet()) {
if (!packet.has_track_event())
continue;
const auto& track_event = packet.track_event();
if (track_event.type() !=
perfetto::protos::gen::TrackEvent::TYPE_SLICE_BEGIN)
continue;
EXPECT_TRUE(track_event.has_log_message());
const auto& log = track_event.log_message();
EXPECT_EQ(1u, log.source_location_iid());
EXPECT_EQ(2u, log.body_iid());
const auto& dbg = track_event.debug_annotations()[0];
EXPECT_EQ("random", dbg.name());
EXPECT_EQ(random_value, dbg.int_value());
found_args = true;
}
EXPECT_TRUE(found_args);
}
struct InternedLogMessageBody
: public perfetto::TrackEventInternedDataIndex<
InternedLogMessageBody,
perfetto::protos::pbzero::InternedData::kLogMessageBodyFieldNumber,
std::string> {
static void Add(perfetto::protos::pbzero::InternedData* interned_data,
size_t iid,
const std::string& value) {
auto l = interned_data->add_log_message_body();
l->set_iid(iid);
l->set_body(value.data(), value.size());
commit_count++;
}
static int commit_count;
};
int InternedLogMessageBody::commit_count = 0;
TEST_P(PerfettoApiTest, TrackEventTypedArgsWithInterning) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
std::stringstream large_message;
for (size_t i = 0; i < 512; i++)
large_message << i << ". Something wicked this way comes. ";
size_t body_iid;
InternedLogMessageBody::commit_count = 0;
TRACE_EVENT_BEGIN("foo", "EventWithState", [&](perfetto::EventContext ctx) {
EXPECT_EQ(0, InternedLogMessageBody::commit_count);
body_iid = InternedLogMessageBody::Get(&ctx, "Alas, poor Yorick!");
auto log = ctx.event()->set_log_message();
log->set_body_iid(body_iid);
EXPECT_EQ(1, InternedLogMessageBody::commit_count);
auto body_iid2 = InternedLogMessageBody::Get(&ctx, "Alas, poor Yorick!");
EXPECT_EQ(body_iid, body_iid2);
EXPECT_EQ(1, InternedLogMessageBody::commit_count);
});
TRACE_EVENT_END("foo");
TRACE_EVENT_BEGIN("foo", "EventWithState", [&](perfetto::EventContext ctx) {
// Check that very large amounts of interned data works.
auto log = ctx.event()->set_log_message();
log->set_body_iid(InternedLogMessageBody::Get(&ctx, large_message.str()));
EXPECT_EQ(2, InternedLogMessageBody::commit_count);
});
TRACE_EVENT_END("foo");
// Make sure interned data persists across trace points.
TRACE_EVENT_BEGIN("foo", "EventWithState", [&](perfetto::EventContext ctx) {
auto body_iid2 = InternedLogMessageBody::Get(&ctx, "Alas, poor Yorick!");
EXPECT_EQ(body_iid, body_iid2);
auto body_iid3 = InternedLogMessageBody::Get(&ctx, "I knew him, Horatio");
EXPECT_NE(body_iid, body_iid3);
auto log = ctx.event()->set_log_message();
log->set_body_iid(body_iid3);
EXPECT_EQ(3, InternedLogMessageBody::commit_count);
});
TRACE_EVENT_END("foo");
tracing_session->get()->StopBlocking();
auto log_messages = ReadLogMessagesFromTrace(tracing_session->get());
EXPECT_THAT(log_messages,
ElementsAre("Alas, poor Yorick!", large_message.str(),
"I knew him, Horatio"));
}
struct InternedLogMessageBodySmall
: public perfetto::TrackEventInternedDataIndex<
InternedLogMessageBodySmall,
perfetto::protos::pbzero::InternedData::kLogMessageBodyFieldNumber,
const char*,
perfetto::SmallInternedDataTraits> {
static void Add(perfetto::protos::pbzero::InternedData* interned_data,
size_t iid,
const char* value) {
auto l = interned_data->add_log_message_body();
l->set_iid(iid);
l->set_body(value);
}
};
TEST_P(PerfettoApiTest, TrackEventTypedArgsWithInterningByValue) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
size_t body_iid;
TRACE_EVENT_BEGIN("foo", "EventWithState", [&](perfetto::EventContext ctx) {
body_iid = InternedLogMessageBodySmall::Get(&ctx, "This above all:");
auto log = ctx.event()->set_log_message();
log->set_body_iid(body_iid);
auto body_iid2 = InternedLogMessageBodySmall::Get(&ctx, "This above all:");
EXPECT_EQ(body_iid, body_iid2);
auto body_iid3 =
InternedLogMessageBodySmall::Get(&ctx, "to thine own self be true");
EXPECT_NE(body_iid, body_iid3);
});
TRACE_EVENT_END("foo");
tracing_session->get()->StopBlocking();
auto log_messages = ReadLogMessagesFromTrace(tracing_session->get());
EXPECT_THAT(log_messages, ElementsAre("This above all:"));
}
struct InternedLogMessageBodyHashed
: public perfetto::TrackEventInternedDataIndex<
InternedLogMessageBodyHashed,
perfetto::protos::pbzero::InternedData::kLogMessageBodyFieldNumber,
std::string,
perfetto::HashedInternedDataTraits> {
static void Add(perfetto::protos::pbzero::InternedData* interned_data,
size_t iid,
const std::string& value) {
auto l = interned_data->add_log_message_body();
l->set_iid(iid);
l->set_body(value.data(), value.size());
}
};
TEST_P(PerfettoApiTest, TrackEventTypedArgsWithInterningByHashing) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
size_t body_iid;
TRACE_EVENT_BEGIN("foo", "EventWithState", [&](perfetto::EventContext ctx) {
// Test using a dynamically created interned value.
body_iid = InternedLogMessageBodyHashed::Get(
&ctx, std::string("Though this ") + "be madness,");
auto log = ctx.event()->set_log_message();
log->set_body_iid(body_iid);
auto body_iid2 =
InternedLogMessageBodyHashed::Get(&ctx, "Though this be madness,");
EXPECT_EQ(body_iid, body_iid2);
auto body_iid3 =
InternedLogMessageBodyHashed::Get(&ctx, "yet there is method in’t");
EXPECT_NE(body_iid, body_iid3);
});
TRACE_EVENT_END("foo");
tracing_session->get()->StopBlocking();
auto log_messages = ReadLogMessagesFromTrace(tracing_session->get());
EXPECT_THAT(log_messages, ElementsAre("Though this be madness,"));
}
struct InternedSourceLocation
: public perfetto::TrackEventInternedDataIndex<
InternedSourceLocation,
perfetto::protos::pbzero::InternedData::kSourceLocationsFieldNumber,
SourceLocation> {
static void Add(perfetto::protos::pbzero::InternedData* interned_data,
size_t iid,
const SourceLocation& value) {
auto l = interned_data->add_source_locations();
auto file_name = std::get<0>(value);
auto function_name = std::get<1>(value);
auto line_number = std::get<2>(value);
l->set_iid(iid);
l->set_file_name(file_name);
l->set_function_name(function_name);
l->set_line_number(line_number);
}
};
TEST_P(PerfettoApiTest, TrackEventTypedArgsWithInterningComplexValue) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
TRACE_EVENT_BEGIN("foo", "EventWithState", [&](perfetto::EventContext ctx) {
const SourceLocation location{"file.cc", "SomeFunction", 123};
auto location_iid = InternedSourceLocation::Get(&ctx, location);
auto body_iid = InternedLogMessageBody::Get(&ctx, "To be, or not to be");
auto log = ctx.event()->set_log_message();
log->set_source_location_iid(location_iid);
log->set_body_iid(body_iid);
auto location_iid2 = InternedSourceLocation::Get(&ctx, location);
EXPECT_EQ(location_iid, location_iid2);
const SourceLocation location2{"file.cc", "SomeFunction", 456};
auto location_iid3 = InternedSourceLocation::Get(&ctx, location2);
EXPECT_NE(location_iid, location_iid3);
});
TRACE_EVENT_END("foo");
tracing_session->get()->StopBlocking();
auto log_messages = ReadLogMessagesFromTrace(tracing_session->get());
EXPECT_THAT(log_messages,
ElementsAre("SomeFunction(file.cc:123): To be, or not to be"));
}
TEST_P(PerfettoApiTest, TrackEventScoped) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
{
uint64_t arg = 123;
TRACE_EVENT("test", "TestEventWithArgs", [&](perfetto::EventContext ctx) {
ctx.event()->set_log_message()->set_body_iid(arg);
});
}
// Ensure a single line if statement counts as a valid scope for the macro.
if (true)
TRACE_EVENT("test", "SingleLineTestEvent");
{
// Make sure you can have multiple scoped events in the same scope.
TRACE_EVENT("test", "TestEvent");
TRACE_EVENT("test", "AnotherEvent");
TRACE_EVENT("foo", "DisabledEvent");
}
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(
slices,
ElementsAre("B:test.TestEventWithArgs", "E", "B:test.SingleLineTestEvent",
"E", "B:test.TestEvent", "B:test.AnotherEvent", "E", "E"));
}
// A class similar to what Protozero generates for extended message.
class TestTrackEvent : public perfetto::protos::pbzero::TrackEvent {
public:
static const int field_number = 9901;
void set_extension_value(int value) {
// 9900-10000 is the range of extension field numbers reserved for testing.
AppendTinyVarInt(field_number, value);
}
};
TEST_P(PerfettoApiTest, ExtensionClass) {
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
{
TRACE_EVENT("test", "TestEventWithExtensionArgs",
[&](perfetto::EventContext ctx) {
ctx.event<TestTrackEvent>()->set_extension_value(42);
});
}
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
EXPECT_GE(raw_trace.size(), 0u);
bool found_extension = false;
perfetto::protos::pbzero::Trace_Decoder trace(
reinterpret_cast<uint8_t*>(raw_trace.data()), raw_trace.size());
for (auto it = trace.packet(); it; ++it) {
perfetto::protos::pbzero::TracePacket_Decoder packet(it->data(),
it->size());
if (!packet.has_track_event())
continue;
auto track_event = packet.track_event();
protozero::ProtoDecoder decoder(track_event.data, track_event.size);
for (protozero::Field f = decoder.ReadField(); f.valid();
f = decoder.ReadField()) {
if (f.id() == TestTrackEvent::field_number) {
found_extension = true;
}
}
}
EXPECT_TRUE(found_extension);
}
TEST_P(PerfettoApiTest, TrackEventInstant) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
TRACE_EVENT_INSTANT("test", "TestEvent");
TRACE_EVENT_INSTANT("test", "AnotherEvent");
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices, ElementsAre("I:test.TestEvent", "I:test.AnotherEvent"));
}
TEST_P(PerfettoApiTest, TrackEventDebugAnnotations) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
enum MyEnum : uint32_t { ENUM_FOO, ENUM_BAR };
enum MySignedEnum : int32_t { SIGNED_ENUM_FOO = -1, SIGNED_ENUM_BAR };
enum class MyClassEnum { VALUE };
TRACE_EVENT_BEGIN("test", "E", "bool_arg", false);
TRACE_EVENT_BEGIN("test", "E", "int_arg", -123);
TRACE_EVENT_BEGIN("test", "E", "uint_arg", 456u);
TRACE_EVENT_BEGIN("test", "E", "float_arg", 3.14159262f);
TRACE_EVENT_BEGIN("test", "E", "double_arg", 6.22);
TRACE_EVENT_BEGIN("test", "E", "str_arg", "hello", "str_arg2",
std::string("tracing"));
TRACE_EVENT_BEGIN("test", "E", "ptr_arg",
reinterpret_cast<void*>(static_cast<intptr_t>(0xbaadf00d)));
TRACE_EVENT_BEGIN("test", "E", "size_t_arg", size_t{42});
TRACE_EVENT_BEGIN("test", "E", "ptrdiff_t_arg", ptrdiff_t{-7});
TRACE_EVENT_BEGIN("test", "E", "enum_arg", ENUM_BAR);
TRACE_EVENT_BEGIN("test", "E", "signed_enum_arg", SIGNED_ENUM_FOO);
TRACE_EVENT_BEGIN("test", "E", "class_enum_arg", MyClassEnum::VALUE);
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(
slices,
ElementsAre(
"B:test.E(bool_arg=(bool)0)", "B:test.E(int_arg=(int)-123)",
"B:test.E(uint_arg=(uint)456)", "B:test.E(float_arg=(double)3.14159)",
"B:test.E(double_arg=(double)6.22)",
"B:test.E(str_arg=(string)hello,str_arg2=(string)tracing)",
"B:test.E(ptr_arg=(pointer)baadf00d)",
"B:test.E(size_t_arg=(uint)42)", "B:test.E(ptrdiff_t_arg=(int)-7)",
"B:test.E(enum_arg=(uint)1)", "B:test.E(signed_enum_arg=(int)-1)",
"B:test.E(class_enum_arg=(int)0)"));
}
TEST_P(PerfettoApiTest, TrackEventCustomDebugAnnotations) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
std::unique_ptr<MyDebugAnnotation> owned_annotation(new MyDebugAnnotation());
TRACE_EVENT_BEGIN("test", "E", "custom_arg", MyDebugAnnotation());
TRACE_EVENT_BEGIN("test", "E", "normal_arg", "x", "custom_arg",
std::move(owned_annotation));
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(
slices,
ElementsAre(
R"(B:test.E(custom_arg=(json){"key": 123}))",
R"(B:test.E(normal_arg=(string)x,custom_arg=(json){"key": 123}))"));
}
TEST_P(PerfettoApiTest, TrackEventCustomRawDebugAnnotations) {
// Note: this class is also testing a non-moveable and non-copiable argument.
class MyRawDebugAnnotation : public perfetto::DebugAnnotation {
public:
MyRawDebugAnnotation() { msg_->set_string_value("nested_value"); }
~MyRawDebugAnnotation() = default;
// |msg_| already deletes these implicitly, but let's be explicit for safety
// against future changes.
MyRawDebugAnnotation(const MyRawDebugAnnotation&) = delete;
MyRawDebugAnnotation(MyRawDebugAnnotation&&) = delete;
void Add(perfetto::protos::pbzero::DebugAnnotation* annotation) const {
auto ranges = msg_.GetRanges();
annotation->AppendScatteredBytes(
perfetto::protos::pbzero::DebugAnnotation::kNestedValueFieldNumber,
&ranges[0], ranges.size());
}
private:
mutable protozero::HeapBuffered<
perfetto::protos::pbzero::DebugAnnotation::NestedValue>
msg_;
};
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"test"});
tracing_session->get()->StartBlocking();
TRACE_EVENT_BEGIN("test", "E", "raw_arg", MyRawDebugAnnotation());
TRACE_EVENT_BEGIN("test", "E", "plain_arg", 42, "raw_arg",
MyRawDebugAnnotation());
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(
slices,
ElementsAre("B:test.E(raw_arg=(nested)nested_value)",
"B:test.E(plain_arg=(int)42,raw_arg=(nested)nested_value)"));
}
TEST_P(PerfettoApiTest, TrackEventComputedName) {
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
// New macros require perfetto::StaticString{} annotation.
for (int i = 0; i < 3; i++)
TRACE_EVENT_BEGIN("test", perfetto::StaticString{i % 2 ? "Odd" : "Even"});
// Legacy macros assume all arguments are static strings.
for (int i = 0; i < 3; i++)
TRACE_EVENT_BEGIN0("test", i % 2 ? "Odd" : "Even");
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices, ElementsAre("B:test.Even", "B:test.Odd", "B:test.Even",
"B:test.Even", "B:test.Odd", "B:test.Even"));
}
TEST_P(PerfettoApiTest, TrackEventArgumentsNotEvaluatedWhenDisabled) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
bool called = false;
auto ArgumentFunction = [&] {
called = true;
return 123;
};
TRACE_EVENT_BEGIN("test", "DisabledEvent", "arg", ArgumentFunction());
{ TRACE_EVENT("test", "DisabledScopedEvent", "arg", ArgumentFunction()); }
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
EXPECT_FALSE(called);
ArgumentFunction();
EXPECT_TRUE(called);
}
TEST_P(PerfettoApiTest, TrackEventConfig) {
auto check_config = [&](perfetto::protos::gen::TrackEventConfig te_cfg) {
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
ds_cfg->set_track_event_config_raw(te_cfg.SerializeAsString());
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
TRACE_EVENT_BEGIN("foo", "FooEvent");
TRACE_EVENT_BEGIN("bar", "BarEvent");
TRACE_EVENT_BEGIN("foo,bar", "MultiFooBar");
TRACE_EVENT_BEGIN("baz,bar,quux", "MultiBar");
TRACE_EVENT_BEGIN("red,green,blue,foo", "MultiFoo");
TRACE_EVENT_BEGIN("red,green,blue,yellow", "MultiNone");
TRACE_EVENT_BEGIN("cat", "SlowEvent");
TRACE_EVENT_BEGIN("cat.verbose", "DebugEvent");
TRACE_EVENT_BEGIN("test", "TagEvent");
TRACE_EVENT_BEGIN(TRACE_DISABLED_BY_DEFAULT("cat"), "SlowDisabledEvent");
TRACE_EVENT_BEGIN("dynamic,foo", "DynamicGroupFooEvent");
perfetto::DynamicCategory dyn{"dynamic,bar"};
TRACE_EVENT_BEGIN(dyn, "DynamicGroupBarEvent");
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
tracing_session->session.reset();
return slices;
};
// Empty config should enable all categories except slow ones.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
auto slices = check_config(te_cfg);
EXPECT_THAT(
slices,
ElementsAre("B:foo.FooEvent", "B:bar.BarEvent", "B:foo,bar.MultiFooBar",
"B:baz,bar,quux.MultiBar", "B:red,green,blue,foo.MultiFoo",
"B:test.TagEvent", "B:$dynamic,$foo.DynamicGroupFooEvent",
"B:$dynamic,$bar.DynamicGroupBarEvent"));
}
// Enable exactly one category.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
te_cfg.add_enabled_categories("foo");
auto slices = check_config(te_cfg);
EXPECT_THAT(slices, ElementsAre("B:foo.FooEvent", "B:foo,bar.MultiFooBar",
"B:red,green,blue,foo.MultiFoo",
"B:$dynamic,$foo.DynamicGroupFooEvent"));
}
// Enable two categories.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
te_cfg.add_enabled_categories("foo");
te_cfg.add_enabled_categories("baz");
te_cfg.add_enabled_categories("bar");
auto slices = check_config(te_cfg);
EXPECT_THAT(
slices,
ElementsAre("B:foo.FooEvent", "B:bar.BarEvent", "B:foo,bar.MultiFooBar",
"B:baz,bar,quux.MultiBar", "B:red,green,blue,foo.MultiFoo",
"B:$dynamic,$foo.DynamicGroupFooEvent",
"B:$dynamic,$bar.DynamicGroupBarEvent"));
}
// Enabling all categories with a pattern doesn't enable slow ones.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_enabled_categories("*");
auto slices = check_config(te_cfg);
EXPECT_THAT(
slices,
ElementsAre("B:foo.FooEvent", "B:bar.BarEvent", "B:foo,bar.MultiFooBar",
"B:baz,bar,quux.MultiBar", "B:red,green,blue,foo.MultiFoo",
"B:test.TagEvent", "B:$dynamic,$foo.DynamicGroupFooEvent",
"B:$dynamic,$bar.DynamicGroupBarEvent"));
}
// Enable with a pattern.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
te_cfg.add_enabled_categories("fo*");
auto slices = check_config(te_cfg);
EXPECT_THAT(slices, ElementsAre("B:foo.FooEvent", "B:foo,bar.MultiFooBar",
"B:red,green,blue,foo.MultiFoo",
"B:$dynamic,$foo.DynamicGroupFooEvent"));
}
// Enable with a tag.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
te_cfg.add_enabled_tags("tag");
auto slices = check_config(te_cfg);
EXPECT_THAT(slices, ElementsAre("B:test.TagEvent"));
}
// Enable just slow categories.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_disabled_categories("*");
te_cfg.add_enabled_tags("slow");
auto slices = check_config(te_cfg);
EXPECT_THAT(slices,
ElementsAre("B:cat.SlowEvent",
"B:disabled-by-default-cat.SlowDisabledEvent"));
}
// Enable everything including slow/debug categories.
{
perfetto::protos::gen::TrackEventConfig te_cfg;
te_cfg.add_enabled_categories("*");
te_cfg.add_enabled_tags("slow");
te_cfg.add_enabled_tags("debug");
auto slices = check_config(te_cfg);
EXPECT_THAT(slices,
ElementsAre("B:foo.FooEvent", "B:bar.BarEvent",
"B:foo,bar.MultiFooBar", "B:baz,bar,quux.MultiBar",
"B:red,green,blue,foo.MultiFoo", "B:cat.SlowEvent",
"B:cat.verbose.DebugEvent", "B:test.TagEvent",
"B:disabled-by-default-cat.SlowDisabledEvent",
"B:$dynamic,$foo.DynamicGroupFooEvent",
"B:$dynamic,$bar.DynamicGroupBarEvent"));
}
}
TEST_P(PerfettoApiTest, OneDataSourceOneEvent) {
auto* data_source = &data_sources_["my_data_source"];
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
ds_cfg->set_legacy_config("test config");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
MockDataSource::Trace([](MockDataSource::TraceContext) {
FAIL() << "Should not be called because the trace was not started";
});
MockDataSource::CallIfEnabled([](uint32_t) {
FAIL() << "Should not be called because the trace was not started";
});
tracing_session->get()->Start();
data_source->on_setup.Wait();
EXPECT_EQ(data_source->config.legacy_config(), "test config");
data_source->on_start.Wait();
// Emit one trace event.
std::atomic<int> trace_lambda_calls{0};
MockDataSource::Trace(
[&trace_lambda_calls](MockDataSource::TraceContext ctx) {
auto packet = ctx.NewTracePacket();
packet->set_timestamp(42);
packet->set_for_testing()->set_str("event 1");
trace_lambda_calls++;
packet->Finalize();
// The SMB scraping logic will skip the last packet because it cannot
// guarantee it's finalized. Create an empty packet so we get the
// previous one and this empty one is ignored.
packet = ctx.NewTracePacket();
});
uint32_t active_instances = 0;
MockDataSource::CallIfEnabled([&active_instances](uint32_t instances) {
active_instances = instances;
});
EXPECT_EQ(1u, active_instances);
data_source->on_stop.Wait();
tracing_session->on_stop.Wait();
EXPECT_EQ(trace_lambda_calls, 1);
MockDataSource::Trace([](MockDataSource::TraceContext) {
FAIL() << "Should not be called because the trace is now stopped";
});
MockDataSource::CallIfEnabled([](uint32_t) {
FAIL() << "Should not be called because the trace is now stopped";
});
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
ASSERT_GE(raw_trace.size(), 0u);
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
bool test_packet_found = false;
for (const auto& packet : trace.packet()) {
if (!packet.has_for_testing())
continue;
EXPECT_FALSE(test_packet_found);
EXPECT_EQ(packet.timestamp(), 42U);
EXPECT_EQ(packet.for_testing().str(), "event 1");
test_packet_found = true;
}
EXPECT_TRUE(test_packet_found);
}
TEST_P(PerfettoApiTest, ConsumerFlush) {
auto* data_source = &data_sources_["my_data_source"];
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
ds_cfg->set_legacy_config("test config");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->Start();
data_source->on_start.Wait();
MockDataSource::Trace([&](MockDataSource::TraceContext ctx) {
auto packet = ctx.NewTracePacket();
packet->set_timestamp(42);
packet->set_for_testing()->set_str("flushed event");
packet->Finalize();
// The SMB scraping logic will skip the last packet because it cannot
// guarantee it's finalized. Create an empty packet so we get the
// previous one and this empty one is ignored.
packet = ctx.NewTracePacket();
});
EXPECT_TRUE(tracing_session->get()->FlushBlocking());
// Deliberately doing ReadTraceBlocking() before StopBlocking() to avoid
// hitting the auto scrape-on-stop behavior of the service.
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
tracing_session->get()->StopBlocking();
ASSERT_GE(raw_trace.size(), 0u);
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
bool test_packet_found = false;
for (const auto& packet : trace.packet()) {
if (!packet.has_for_testing())
continue;
EXPECT_FALSE(test_packet_found);
EXPECT_EQ(packet.timestamp(), 42U);
EXPECT_EQ(packet.for_testing().str(), "flushed event");
test_packet_found = true;
}
EXPECT_TRUE(test_packet_found);
}
TEST_P(PerfettoApiTest, WithBatching) {
auto* data_source = &data_sources_["my_data_source"];
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
ds_cfg->set_legacy_config("test config");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->Start();
data_source->on_setup.Wait();
data_source->on_start.Wait();
std::stringstream first_large_message;
for (size_t i = 0; i < 512; i++)
first_large_message << i << ". Something wicked this way comes. ";
auto first_large_message_str = first_large_message.str();
// Emit one trace event before we begin batching.
MockDataSource::Trace(
[&first_large_message_str](MockDataSource::TraceContext ctx) {
auto packet = ctx.NewTracePacket();
packet->set_timestamp(42);
packet->set_for_testing()->set_str(first_large_message_str);
packet->Finalize();
});
// Simulate the start of a batching cycle by first setting the batching period
// to a very large value and then force-flushing when we are done writing
// data.
ASSERT_TRUE(
perfetto::test::EnableDirectSMBPatching(/*BackendType=*/GetParam()));
perfetto::test::SetBatchCommitsDuration(UINT32_MAX,
/*BackendType=*/GetParam());
std::stringstream second_large_message;
for (size_t i = 0; i < 512; i++)
second_large_message << i << ". Something else wicked this way comes. ";
auto second_large_message_str = second_large_message.str();
// Emit another trace event.
MockDataSource::Trace(
[&second_large_message_str](MockDataSource::TraceContext ctx) {
auto packet = ctx.NewTracePacket();
packet->set_timestamp(43);
packet->set_for_testing()->set_str(second_large_message_str);
packet->Finalize();
// Simulate the end of the batching cycle.
ctx.Flush();
});
data_source->on_stop.Wait();
tracing_session->on_stop.Wait();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
ASSERT_GE(raw_trace.size(), 0u);
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
bool test_packet_1_found = false;
bool test_packet_2_found = false;
for (const auto& packet : trace.packet()) {
if (!packet.has_for_testing())
continue;
EXPECT_TRUE(packet.timestamp() == 42U || packet.timestamp() == 43U);
if (packet.timestamp() == 42U) {
EXPECT_FALSE(test_packet_1_found);
EXPECT_EQ(packet.for_testing().str(), first_large_message_str);
test_packet_1_found = true;
} else {
EXPECT_FALSE(test_packet_2_found);
EXPECT_EQ(packet.for_testing().str(), second_large_message_str);
test_packet_2_found = true;
}
}
EXPECT_TRUE(test_packet_1_found && test_packet_2_found);
}
TEST_P(PerfettoApiTest, BlockingStartAndStop) {
auto* data_source = &data_sources_["my_data_source"];
// Register a second data source to get a bit more coverage.
perfetto::DataSourceDescriptor dsd;
dsd.set_name("my_data_source2");
MockDataSource2::Register(dsd, kTestDataSourceArg);
perfetto::test::SyncProducers();
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source2");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
EXPECT_TRUE(data_source->on_setup.notified());
EXPECT_TRUE(data_source->on_start.notified());
tracing_session->get()->StopBlocking();
EXPECT_TRUE(data_source->on_stop.notified());
EXPECT_TRUE(tracing_session->on_stop.notified());
}
TEST_P(PerfettoApiTest, BlockingStartAndStopOnEmptySession) {
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("non_existent_data_source");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
tracing_session->get()->StopBlocking();
EXPECT_TRUE(tracing_session->on_stop.notified());
}
TEST_P(PerfettoApiTest, WriteEventsAfterDeferredStop) {
auto* data_source = &data_sources_["my_data_source"];
data_source->handle_stop_asynchronously = true;
// Setup the trace config and start the tracing session.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
// Stop and wait for the producer to have seen the stop event.
WaitableTestEvent consumer_stop_signal;
tracing_session->get()->SetOnStopCallback(
[&consumer_stop_signal] { consumer_stop_signal.Notify(); });
tracing_session->get()->Stop();
data_source->on_stop.Wait();
// At this point tracing should be still allowed because of the
// HandleStopAsynchronously() call.
bool lambda_called = false;
// This usleep is here just to prevent that we accidentally pass the test
// just by virtue of hitting some race. We should be able to trace up until
// 5 seconds after seeing the stop when using the deferred stop mechanism.
std::this_thread::sleep_for(std::chrono::milliseconds(250));
MockDataSource::Trace([&lambda_called](MockDataSource::TraceContext ctx) {
auto packet = ctx.NewTracePacket();
packet->set_for_testing()->set_str("event written after OnStop");
packet->Finalize();
ctx.Flush();
lambda_called = true;
});
ASSERT_TRUE(lambda_called);
// Now call the async stop closure. This acks the stop to the service and
// disallows further Trace() calls.
EXPECT_TRUE(data_source->async_stop_closure);
data_source->async_stop_closure();
// Wait that the stop is propagated to the consumer.
consumer_stop_signal.Wait();
MockDataSource::Trace([](MockDataSource::TraceContext) {
FAIL() << "Should not be called after the stop is acked";
});
// Check the contents of the trace.
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
ASSERT_GE(raw_trace.size(), 0u);
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
int test_packet_found = 0;
for (const auto& packet : trace.packet()) {
if (!packet.has_for_testing())
continue;
EXPECT_EQ(packet.for_testing().str(), "event written after OnStop");
test_packet_found++;
}
EXPECT_EQ(test_packet_found, 1);
}
TEST_P(PerfettoApiTest, RepeatedStartAndStop) {
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
for (int i = 0; i < 5; i++) {
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->Start();
std::atomic<bool> stop_called{false};
tracing_session->get()->SetOnStopCallback(
[&stop_called] { stop_called = true; });
tracing_session->get()->StopBlocking();
EXPECT_TRUE(stop_called);
}
}
TEST_P(PerfettoApiTest, SetupWithFile) {
#if PERFETTO_BUILDFLAG(PERFETTO_OS_WIN)
if (GetParam() == perfetto::kSystemBackend)
GTEST_SKIP() << "write_into_file + system mode is not supported on Windows";
#endif
auto temp_file = perfetto::test::CreateTempFile();
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
// Write a trace into |fd|.
auto* tracing_session = NewTrace(cfg, temp_file.fd);
tracing_session->get()->StartBlocking();
tracing_session->get()->StopBlocking();
#if !PERFETTO_BUILDFLAG(PERFETTO_OS_WIN)
// Check that |fd| didn't get closed.
EXPECT_EQ(0, fcntl(temp_file.fd, F_GETFD, 0));
#endif
// Check that the trace got written.
EXPECT_GT(lseek(temp_file.fd, 0, SEEK_END), 0);
EXPECT_EQ(0, close(temp_file.fd));
// Clean up.
EXPECT_EQ(0, remove(temp_file.path.c_str()));
}
TEST_P(PerfettoApiTest, MultipleRegistrations) {
// Attempt to register the same data source again.
perfetto::DataSourceDescriptor dsd;
dsd.set_name("my_data_source");
EXPECT_TRUE(MockDataSource::Register(dsd));
perfetto::test::SyncProducers();
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
// Emit one trace event.
std::atomic<int> trace_lambda_calls{0};
MockDataSource::Trace([&trace_lambda_calls](MockDataSource::TraceContext) {
trace_lambda_calls++;
});
// Make sure the data source got called only once.
tracing_session->get()->StopBlocking();
EXPECT_EQ(trace_lambda_calls, 1);
}
TEST_P(PerfettoApiTest, CustomIncrementalState) {
perfetto::DataSourceDescriptor dsd;
dsd.set_name("incr_data_source");
TestIncrementalDataSource::Register(dsd);
perfetto::test::SyncProducers();
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("incr_data_source");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
// First emit a no-op trace event that initializes the incremental state as a
// side effect.
TestIncrementalDataSource::Trace(
[](TestIncrementalDataSource::TraceContext) {});
EXPECT_TRUE(TestIncrementalState::constructed);
// Check that the incremental state is carried across trace events.
TestIncrementalDataSource::Trace(
[](TestIncrementalDataSource::TraceContext ctx) {
auto* state = ctx.GetIncrementalState();
EXPECT_TRUE(state);
EXPECT_EQ(100, state->count);
state->count++;
});
TestIncrementalDataSource::Trace(
[](TestIncrementalDataSource::TraceContext ctx) {
auto* state = ctx.GetIncrementalState();
EXPECT_EQ(101, state->count);
});
// Make sure the incremental state gets cleaned up between sessions.
tracing_session->get()->StopBlocking();
tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
TestIncrementalDataSource::Trace(
[](TestIncrementalDataSource::TraceContext ctx) {
auto* state = ctx.GetIncrementalState();
EXPECT_TRUE(TestIncrementalState::destroyed);
EXPECT_TRUE(state);
EXPECT_EQ(100, state->count);
});
tracing_session->get()->StopBlocking();
}
// Regression test for b/139110180. Checks that GetDataSourceLocked() can be
// called from OnStart() and OnStop() callbacks without deadlocking.
TEST_P(PerfettoApiTest, GetDataSourceLockedFromCallbacks) {
auto* data_source = &data_sources_["my_data_source"];
// Setup the trace config.
perfetto::TraceConfig cfg;
cfg.set_duration_ms(1);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("my_data_source");
// Create a new trace session.
auto* tracing_session = NewTrace(cfg);
data_source->on_start_callback = [] {
MockDataSource::Trace([](MockDataSource::TraceContext ctx) {
ctx.NewTracePacket()->set_for_testing()->set_str("on-start");
auto ds = ctx.GetDataSourceLocked();
ASSERT_TRUE(!!ds);
ctx.NewTracePacket()->set_for_testing()->set_str("on-start-locked");
});
};
data_source->on_stop_callback = [] {
MockDataSource::Trace([](MockDataSource::TraceContext ctx) {
ctx.NewTracePacket()->set_for_testing()->set_str("on-stop");
auto ds = ctx.GetDataSourceLocked();
ASSERT_TRUE(!!ds);
ctx.NewTracePacket()->set_for_testing()->set_str("on-stop-locked");
ctx.Flush();
});
};
tracing_session->get()->Start();
data_source->on_stop.Wait();
tracing_session->on_stop.Wait();
std::vector<char> raw_trace = tracing_session->get()->ReadTraceBlocking();
ASSERT_GE(raw_trace.size(), 0u);
perfetto::protos::gen::Trace trace;
ASSERT_TRUE(trace.ParseFromArray(raw_trace.data(), raw_trace.size()));
int packets_found = 0;
for (const auto& packet : trace.packet()) {
if (!packet.has_for_testing())
continue;
packets_found |= packet.for_testing().str() == "on-start" ? 1 : 0;
packets_found |= packet.for_testing().str() == "on-start-locked" ? 2 : 0;
packets_found |= packet.for_testing().str() == "on-stop" ? 4 : 0;
packets_found |= packet.for_testing().str() == "on-stop-locked" ? 8 : 0;
}
EXPECT_EQ(packets_found, 1 | 2 | 4 | 8);
}
TEST_P(PerfettoApiTest, OnStartCallback) {
perfetto::TraceConfig cfg;
cfg.set_duration_ms(60000);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
auto* tracing_session = NewTrace(cfg);
WaitableTestEvent got_start;
tracing_session->get()->SetOnStartCallback([&] { got_start.Notify(); });
tracing_session->get()->Start();
got_start.Wait();
tracing_session->get()->StopBlocking();
}
TEST_P(PerfettoApiTest, OnErrorCallback) {
perfetto::TraceConfig cfg;
// Requesting too long |duration_ms| will cause EnableTracing() to fail.
cfg.set_duration_ms(static_cast<uint32_t>(-1));
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
auto* tracing_session = NewTrace(cfg);
WaitableTestEvent got_error;
tracing_session->get()->SetOnErrorCallback([&](perfetto::TracingError error) {
EXPECT_EQ(perfetto::TracingError::kTracingFailed, error.code);
EXPECT_FALSE(error.message.empty());
got_error.Notify();
});
tracing_session->get()->Start();
got_error.Wait();
// Registered error callback will be triggered also by OnDisconnect()
// function. This may happen after exiting this test what would result in
// system crash (|got_error| will not exist at that time). To prevent that
// scenario, error callback has to be cleared.
tracing_session->get()->SetOnErrorCallback(nullptr);
tracing_session->get()->StopBlocking();
}
TEST_P(PerfettoApiTest, GetTraceStats) {
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
// Asynchronous read.
WaitableTestEvent got_stats;
tracing_session->get()->GetTraceStats(
[&got_stats](perfetto::TracingSession::GetTraceStatsCallbackArgs args) {
perfetto::protos::gen::TraceStats trace_stats;
EXPECT_TRUE(args.success);
EXPECT_TRUE(trace_stats.ParseFromArray(args.trace_stats_data.data(),
args.trace_stats_data.size()));
EXPECT_EQ(1, trace_stats.buffer_stats_size());
got_stats.Notify();
});
got_stats.Wait();
// Blocking read.
auto stats = tracing_session->get()->GetTraceStatsBlocking();
perfetto::protos::gen::TraceStats trace_stats;
EXPECT_TRUE(stats.success);
EXPECT_TRUE(trace_stats.ParseFromArray(stats.trace_stats_data.data(),
stats.trace_stats_data.size()));
EXPECT_EQ(1, trace_stats.buffer_stats_size());
tracing_session->get()->StopBlocking();
}
TEST_P(PerfettoApiTest, QueryServiceState) {
class QueryTestDataSource : public perfetto::DataSource<QueryTestDataSource> {
};
RegisterDataSource<QueryTestDataSource>("query_test_data_source");
perfetto::test::SyncProducers();
auto tracing_session =
perfetto::Tracing::NewTrace(/*BackendType=*/GetParam());
// Asynchronous read.
WaitableTestEvent got_state;
tracing_session->QueryServiceState(
[&got_state](
perfetto::TracingSession::QueryServiceStateCallbackArgs result) {
perfetto::protos::gen::TracingServiceState state;
EXPECT_TRUE(result.success);
EXPECT_TRUE(state.ParseFromArray(result.service_state_data.data(),
result.service_state_data.size()));
EXPECT_EQ(1, state.producers_size());
EXPECT_NE(std::string::npos,
state.producers()[0].name().find("integrationtest"));
bool found_ds = false;
for (const auto& ds : state.data_sources())
found_ds |= ds.ds_descriptor().name() == "query_test_data_source";
EXPECT_TRUE(found_ds);
got_state.Notify();
});
got_state.Wait();
// Blocking read.
auto result = tracing_session->QueryServiceStateBlocking();
perfetto::protos::gen::TracingServiceState state;
EXPECT_TRUE(result.success);
EXPECT_TRUE(state.ParseFromArray(result.service_state_data.data(),
result.service_state_data.size()));
EXPECT_EQ(1, state.producers_size());
EXPECT_NE(std::string::npos,
state.producers()[0].name().find("integrationtest"));
bool found_ds = false;
for (const auto& ds : state.data_sources())
found_ds |= ds.ds_descriptor().name() == "query_test_data_source";
EXPECT_TRUE(found_ds);
}
TEST_P(PerfettoApiTest, LegacyTraceEvents) {
// Create a new trace session.
auto* tracing_session =
NewTraceWithCategories({"cat", TRACE_DISABLED_BY_DEFAULT("cat")});
tracing_session->get()->StartBlocking();
// Basic events.
TRACE_EVENT_INSTANT0("cat", "LegacyEvent", TRACE_EVENT_SCOPE_GLOBAL);
TRACE_EVENT_BEGIN1("cat", "LegacyEvent", "arg", 123);
TRACE_EVENT_END2("cat", "LegacyEvent", "arg", "string", "arg2", 0.123f);
// Scoped event.
{ TRACE_EVENT0("cat", "ScopedLegacyEvent"); }
// Event with flow (and disabled category).
TRACE_EVENT_WITH_FLOW0(TRACE_DISABLED_BY_DEFAULT("cat"), "LegacyFlowEvent",
0xdadacafe, TRACE_EVENT_FLAG_FLOW_IN);
// Event with timestamp.
TRACE_EVENT_INSTANT_WITH_TIMESTAMP0("cat", "LegacyInstantEvent",
TRACE_EVENT_SCOPE_GLOBAL,
MyTimestamp{123456789ul});
// Event with id, thread id and timestamp (and dynamic name).
TRACE_EVENT_COPY_BEGIN_WITH_ID_TID_AND_TIMESTAMP0(
"cat", std::string("LegacyWithIdTidAndTimestamp").c_str(), 1,
MyThreadId(123, 456), MyTimestamp{3});
// Event with id.
TRACE_COUNTER1("cat", "LegacyCounter", 1234);
TRACE_COUNTER_ID1("cat", "LegacyCounterWithId", 1234, 9000);
// Metadata event.
TRACE_EVENT_METADATA1("cat", "LegacyMetadata", "obsolete", true);
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(
slices,
ElementsAre(
"I:cat.LegacyEvent", "B:cat.LegacyEvent(arg=(int)123)",
"E.LegacyEvent(arg=(string)string,arg2=(double)0.123)",
"B:cat.ScopedLegacyEvent", "E",
"B(bind_id=3671771902)(flow_direction=1):disabled-by-default-cat."
"LegacyFlowEvent",
"I:cat.LegacyInstantEvent",
"Legacy_S(unscoped_id=1)(pid_override=123)(tid_override=456):cat."
"LegacyWithIdTidAndTimestamp",
"Legacy_C:cat.LegacyCounter(value=(int)1234)",
"Legacy_C(unscoped_id=1234):cat.LegacyCounterWithId(value=(int)9000)",
"Legacy_M:cat.LegacyMetadata"));
}
TEST_P(PerfettoApiTest, LegacyTraceEventsWithCustomAnnotation) {
// Create a new trace session.
auto* tracing_session = NewTraceWithCategories({"cat"});
tracing_session->get()->StartBlocking();
MyDebugAnnotation annotation;
TRACE_EVENT_BEGIN1("cat", "LegacyEvent", "arg", annotation);
std::unique_ptr<MyDebugAnnotation> owned_annotation(new MyDebugAnnotation());
TRACE_EVENT_BEGIN1("cat", "LegacyEvent", "arg", std::move(owned_annotation));
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices,
ElementsAre("B:cat.LegacyEvent(arg=(json){\"key\": 123})",
"B:cat.LegacyEvent(arg=(json){\"key\": 123})"));
}
TEST_P(PerfettoApiTest, LegacyTraceEventsWithConcurrentSessions) {
// Make sure that a uniquely owned debug annotation can be written into
// multiple concurrent tracing sessions.
auto* tracing_session = NewTraceWithCategories({"cat"});
tracing_session->get()->StartBlocking();
auto* tracing_session2 = NewTraceWithCategories({"cat"});
tracing_session2->get()->StartBlocking();
std::unique_ptr<MyDebugAnnotation> owned_annotation(new MyDebugAnnotation());
TRACE_EVENT_BEGIN1("cat", "LegacyEvent", "arg", std::move(owned_annotation));
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices,
ElementsAre("B:cat.LegacyEvent(arg=(json){\"key\": 123})"));
tracing_session2->get()->StopBlocking();
slices = ReadSlicesFromTrace(tracing_session2->get());
EXPECT_THAT(slices,
ElementsAre("B:cat.LegacyEvent(arg=(json){\"key\": 123})"));
}
TEST_P(PerfettoApiTest, LegacyTraceEventsWithId) {
auto* tracing_session = NewTraceWithCategories({"cat"});
tracing_session->get()->StartBlocking();
TRACE_EVENT_ASYNC_BEGIN0("cat", "UnscopedId", 0x1000);
TRACE_EVENT_ASYNC_BEGIN0("cat", "LocalId", TRACE_ID_LOCAL(0x2000));
TRACE_EVENT_ASYNC_BEGIN0("cat", "GlobalId", TRACE_ID_GLOBAL(0x3000));
TRACE_EVENT_ASYNC_BEGIN0(
"cat", "WithScope",
TRACE_ID_WITH_SCOPE("scope string", TRACE_ID_GLOBAL(0x4000)));
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices, ElementsAre("Legacy_S(unscoped_id=4096):cat.UnscopedId",
"Legacy_S(local_id=8192):cat.LocalId",
"Legacy_S(global_id=12288):cat.GlobalId",
"Legacy_S(global_id=16384)(id_scope=\"scope "
"string\"):cat.WithScope"));
}
TEST_P(PerfettoApiTest, LegacyTraceEventsWithFlow) {
auto* tracing_session = NewTraceWithCategories({"cat"});
tracing_session->get()->StartBlocking();
const uint64_t flow_id = 1234;
{
TRACE_EVENT_WITH_FLOW1("cat", "LatencyInfo.Flow", TRACE_ID_GLOBAL(flow_id),
TRACE_EVENT_FLAG_FLOW_OUT, "step", "Begin");
}
{
TRACE_EVENT_WITH_FLOW2("cat", "LatencyInfo.Flow", TRACE_ID_GLOBAL(flow_id),
TRACE_EVENT_FLAG_FLOW_IN | TRACE_EVENT_FLAG_FLOW_OUT,
"step", "Middle", "value", false);
}
{
TRACE_EVENT_WITH_FLOW1("cat", "LatencyInfo.Flow", TRACE_ID_GLOBAL(flow_id),
TRACE_EVENT_FLAG_FLOW_IN, "step", "End");
}
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices,
ElementsAre("B(bind_id=1234)(flow_direction=2):cat.LatencyInfo."
"Flow(step=(string)Begin)",
"E",
"B(bind_id=1234)(flow_direction=3):cat.LatencyInfo."
"Flow(step=(string)Middle,value=(bool)0)",
"E",
"B(bind_id=1234)(flow_direction=1):cat.LatencyInfo."
"Flow(step=(string)End)",
"E"));
}
TEST_P(PerfettoApiTest, LegacyCategoryGroupEnabledState) {
bool foo_status;
bool bar_status;
bool dynamic_status;
TRACE_EVENT_CATEGORY_GROUP_ENABLED("foo", &foo_status);
TRACE_EVENT_CATEGORY_GROUP_ENABLED("bar", &bar_status);
TRACE_EVENT_CATEGORY_GROUP_ENABLED("dynamic", &dynamic_status);
EXPECT_FALSE(foo_status);
EXPECT_FALSE(bar_status);
EXPECT_FALSE(dynamic_status);
const uint8_t* foo_enabled =
TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED("foo");
const uint8_t* bar_enabled =
TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED("bar");
EXPECT_FALSE(*foo_enabled);
EXPECT_FALSE(*bar_enabled);
// The category group enabled pointer can also be retrieved with a
// runtime-computed category name.
std::string computed_cat("cat");
const uint8_t* computed_enabled =
TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED(computed_cat.c_str());
EXPECT_FALSE(*computed_enabled);
// The enabled pointers can be converted back to category names.
EXPECT_EQ("foo", TRACE_EVENT_API_GET_CATEGORY_GROUP_NAME(foo_enabled));
EXPECT_EQ("bar", TRACE_EVENT_API_GET_CATEGORY_GROUP_NAME(bar_enabled));
EXPECT_EQ("cat", TRACE_EVENT_API_GET_CATEGORY_GROUP_NAME(computed_enabled));
auto* tracing_session = NewTraceWithCategories({"foo", "dynamic", "cat"});
tracing_session->get()->StartBlocking();
TRACE_EVENT_CATEGORY_GROUP_ENABLED("foo", &foo_status);
TRACE_EVENT_CATEGORY_GROUP_ENABLED("bar", &bar_status);
TRACE_EVENT_CATEGORY_GROUP_ENABLED("dynamic", &dynamic_status);
EXPECT_TRUE(foo_status);
EXPECT_FALSE(bar_status);
EXPECT_TRUE(dynamic_status);
EXPECT_TRUE(*foo_enabled);
EXPECT_FALSE(*bar_enabled);
EXPECT_TRUE(*computed_enabled);
tracing_session->get()->StopBlocking();
TRACE_EVENT_CATEGORY_GROUP_ENABLED("foo", &foo_status);
TRACE_EVENT_CATEGORY_GROUP_ENABLED("bar", &bar_status);
TRACE_EVENT_CATEGORY_GROUP_ENABLED("dynamic", &dynamic_status);
EXPECT_FALSE(foo_status);
EXPECT_FALSE(bar_status);
EXPECT_FALSE(dynamic_status);
EXPECT_FALSE(*foo_enabled);
EXPECT_FALSE(*bar_enabled);
EXPECT_FALSE(*computed_enabled);
}
TEST_P(PerfettoApiTest, CategoryEnabledState) {
perfetto::DynamicCategory dynamic{"dynamic"};
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("foo"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("bar"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("red,green,blue,foo"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("dynamic"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("dynamic_2"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED(dynamic));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("cat with spaces"));
auto* tracing_session =
NewTraceWithCategories({"foo", "dynamic", "cat with spaces"});
tracing_session->get()->StartBlocking();
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("foo"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("bar"));
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("red,green,blue,foo"));
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("dynamic"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("dynamic_2"));
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED(dynamic));
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("cat with spaces"));
tracing_session->get()->StopBlocking();
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("foo"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("bar"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("red,green,blue,foo"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("dynamic"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("dynamic_2"));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED(dynamic));
EXPECT_FALSE(TRACE_EVENT_CATEGORY_ENABLED("cat with spaces"));
}
class TestInterceptor : public perfetto::Interceptor<TestInterceptor> {
public:
static TestInterceptor* instance;
struct ThreadLocalState : public perfetto::InterceptorBase::ThreadLocalState {
ThreadLocalState(ThreadLocalStateArgs& args) {
// Test accessing instance state from the TLS constructor.
if (auto self = args.GetInterceptorLocked()) {
self->tls_initialized = true;
}
}
std::map<uint64_t, std::string> event_names;
};
TestInterceptor(const std::string& constructor_arg) {
EXPECT_EQ(constructor_arg, "Constructor argument");
// Note: some tests in this suite register multiple track event data
// sources. We only track data for the first in this test.
if (!instance)
instance = this;
}
~TestInterceptor() override {
if (instance != this)
return;
instance = nullptr;
EXPECT_TRUE(setup_called);
EXPECT_TRUE(start_called);
EXPECT_TRUE(stop_called);
EXPECT_TRUE(tls_initialized);
}
void OnSetup(const SetupArgs&) override {
EXPECT_FALSE(setup_called);
EXPECT_FALSE(start_called);
EXPECT_FALSE(stop_called);
setup_called = true;
}
void OnStart(const StartArgs&) override {
EXPECT_TRUE(setup_called);
EXPECT_FALSE(start_called);
EXPECT_FALSE(stop_called);
start_called = true;
}
void OnStop(const StopArgs&) override {
EXPECT_TRUE(setup_called);
EXPECT_TRUE(start_called);
EXPECT_FALSE(stop_called);
stop_called = true;
}
static void OnTracePacket(InterceptorContext context) {
perfetto::protos::pbzero::TracePacket::Decoder packet(
context.packet_data.data, context.packet_data.size);
EXPECT_TRUE(packet.trusted_packet_sequence_id() > 0);
{
auto self = context.GetInterceptorLocked();
ASSERT_TRUE(self);
EXPECT_TRUE(self->setup_called);
EXPECT_TRUE(self->start_called);
EXPECT_FALSE(self->stop_called);
EXPECT_TRUE(self->tls_initialized);
}
auto& tls = context.GetThreadLocalState();
if (packet.sequence_flags() &
perfetto::protos::pbzero::TracePacket::SEQ_INCREMENTAL_STATE_CLEARED) {
tls.event_names.clear();
}
if (packet.has_interned_data()) {
perfetto::protos::pbzero::InternedData::Decoder interned_data(
packet.interned_data());
for (auto it = interned_data.event_names(); it; it++) {
perfetto::protos::pbzero::EventName::Decoder entry(*it);
tls.event_names[entry.iid()] = entry.name().ToStdString();
}
}
if (packet.has_track_event()) {
perfetto::protos::pbzero::TrackEvent::Decoder track_event(
packet.track_event());
uint64_t name_iid = track_event.name_iid();
auto self = context.GetInterceptorLocked();
self->events.push_back(tls.event_names[name_iid].c_str());
}
}
bool setup_called = false;
bool start_called = false;
bool stop_called = false;
bool tls_initialized = false;
std::vector<std::string> events;
};
TestInterceptor* TestInterceptor::instance;
TEST_P(PerfettoApiTest, TracePacketInterception) {
perfetto::InterceptorDescriptor desc;
desc.set_name("test_interceptor");
TestInterceptor::Register(desc, std::string("Constructor argument"));
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
ds_cfg->mutable_interceptor_config()->set_name("test_interceptor");
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
EXPECT_EQ(0u, TestInterceptor::instance->events.size());
// The interceptor should see an event immediately without any explicit
// flushing.
TRACE_EVENT_BEGIN("foo", "Hip");
EXPECT_THAT(TestInterceptor::instance->events, ElementsAre("Hip"));
// Emit another event with the same title to test interning.
TRACE_EVENT_BEGIN("foo", "Hip");
EXPECT_THAT(TestInterceptor::instance->events, ElementsAre("Hip", "Hip"));
// Emit an event from another thread. It should still reach the same
// interceptor instance.
std::thread thread([] { TRACE_EVENT_BEGIN("foo", "Hooray"); });
thread.join();
EXPECT_THAT(TestInterceptor::instance->events,
ElementsAre("Hip", "Hip", "Hooray"));
// Emit a packet that spans multiple segments and must be stitched together.
TestInterceptor::instance->events.clear();
static char long_title[8192];
memset(long_title, 'a', sizeof(long_title) - 1);
long_title[sizeof(long_title) - 1] = 0;
TRACE_EVENT_BEGIN("foo", long_title);
EXPECT_THAT(TestInterceptor::instance->events, ElementsAre(long_title));
tracing_session->get()->StopBlocking();
}
void EmitConsoleEvents() {
TRACE_EVENT_INSTANT("foo", "Instant event");
TRACE_EVENT("foo", "Scoped event");
TRACE_EVENT_BEGIN("foo", "Nested event");
TRACE_EVENT_INSTANT("foo", "Instant event");
TRACE_EVENT_INSTANT("foo", "Annotated event", "foo", 1, "bar", "hello");
TRACE_EVENT_END("foo");
uint64_t async_id = 4004;
auto track = perfetto::Track(async_id, perfetto::ThreadTrack::Current());
perfetto::TrackEvent::SetTrackDescriptor(
track, [](perfetto::protos::pbzero::TrackDescriptor* desc) {
desc->set_name("AsyncTrack");
});
TRACE_EVENT_BEGIN("test", "AsyncEvent", track);
std::thread thread([&] {
TRACE_EVENT("foo", "EventFromAnotherThread");
TRACE_EVENT_INSTANT("foo", "Instant event");
TRACE_EVENT_END("test", track);
});
thread.join();
TRACE_EVENT_INSTANT(
"foo", "More annotations", [](perfetto::EventContext ctx) {
{
auto* dbg = ctx.event()->add_debug_annotations();
dbg->set_name("dict");
auto* nested = dbg->set_nested_value();
nested->set_nested_type(
perfetto::protos::pbzero::DebugAnnotation::NestedValue::DICT);
nested->add_dict_keys("key");
auto* value = nested->add_dict_values();
value->set_int_value(123);
}
{
auto* dbg = ctx.event()->add_debug_annotations();
dbg->set_name("array");
auto* nested = dbg->set_nested_value();
nested->set_nested_type(
perfetto::protos::pbzero::DebugAnnotation::NestedValue::ARRAY);
auto* value = nested->add_array_values();
value->set_string_value("first");
value = nested->add_array_values();
value->set_string_value("second");
}
});
}
TEST_P(PerfettoApiTest, ConsoleInterceptorPrint) {
perfetto::ConsoleInterceptor::Register();
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
ds_cfg->mutable_interceptor_config()->set_name("console");
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
EmitConsoleEvents();
tracing_session->get()->StopBlocking();
}
TEST_P(PerfettoApiTest, ConsoleInterceptorVerify) {
perfetto::ConsoleInterceptor::Register();
auto temp_file = perfetto::test::CreateTempFile();
perfetto::ConsoleInterceptor::SetOutputFdForTesting(temp_file.fd);
perfetto::TraceConfig cfg;
cfg.set_duration_ms(500);
cfg.add_buffers()->set_size_kb(1024);
auto* ds_cfg = cfg.add_data_sources()->mutable_config();
ds_cfg->set_name("track_event");
ds_cfg->mutable_interceptor_config()->set_name("console");
auto* tracing_session = NewTrace(cfg);
tracing_session->get()->StartBlocking();
EmitConsoleEvents();
tracing_session->get()->StopBlocking();
perfetto::ConsoleInterceptor::SetOutputFdForTesting(0);
std::vector<std::string> lines;
FILE* f = fdopen(temp_file.fd, "r");
fseek(f, 0u, SEEK_SET);
std::array<char, 128> line{};
while (fgets(line.data(), line.size(), f)) {
// Ignore timestamps and process/thread ids.
std::string s(line.data() + 28);
// Filter out durations.
s = std::regex_replace(s, std::regex(" [+][0-9]*ms"), "");
lines.push_back(std::move(s));
}
fclose(f);
EXPECT_EQ(0, remove(temp_file.path.c_str()));
// clang-format off
std::vector<std::string> golden_lines = {
"foo Instant event\n",
"foo Scoped event {\n",
"foo - Nested event {\n",
"foo - - Instant event\n",
"foo - - Annotated event(foo:1, bar:hello)\n",
"foo - } Nested event\n",
"test AsyncEvent {\n",
"foo EventFromAnotherThread {\n",
"foo - Instant event\n",
"test } AsyncEvent\n",
"foo } EventFromAnotherThread\n",
"foo - More annotations(dict:{key:123}, array:[first, second])\n",
"foo } Scoped event\n",
};
// clang-format on
EXPECT_THAT(lines, ContainerEq(golden_lines));
}
TEST_P(PerfettoApiTest, TrackEventObserver) {
class Observer : public perfetto::TrackEventSessionObserver {
public:
~Observer() override = default;
void OnSetup(const perfetto::DataSourceBase::SetupArgs&) {
// Since other tests here register multiple track event data sources,
// ignore all but the first notifications.
if (setup_called)
return;
setup_called = true;
if (unsubscribe_at_setup)
perfetto::TrackEvent::RemoveSessionObserver(this);
// This event isn't recorded in the trace because tracing isn't active yet
// when OnSetup is called.
TRACE_EVENT_INSTANT("foo", "OnSetup");
// However the active tracing categories have already been updated at this
// point.
EXPECT_TRUE(perfetto::TrackEvent::IsEnabled());
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("foo"));
}
void OnStart(const perfetto::DataSourceBase::StartArgs&) {
if (start_called)
return;
start_called = true;
EXPECT_TRUE(perfetto::TrackEvent::IsEnabled());
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("foo"));
TRACE_EVENT_INSTANT("foo", "OnStart");
}
void OnStop(const perfetto::DataSourceBase::StopArgs&) {
if (stop_called)
return;
stop_called = true;
EXPECT_TRUE(perfetto::TrackEvent::IsEnabled());
EXPECT_TRUE(TRACE_EVENT_CATEGORY_ENABLED("foo"));
TRACE_EVENT_INSTANT("foo", "OnStop");
perfetto::TrackEvent::Flush();
}
bool setup_called{};
bool start_called{};
bool stop_called{};
bool unsubscribe_at_setup{};
};
EXPECT_FALSE(perfetto::TrackEvent::IsEnabled());
{
Observer observer;
perfetto::TrackEvent::AddSessionObserver(&observer);
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
EXPECT_TRUE(observer.setup_called);
EXPECT_TRUE(observer.start_called);
tracing_session->get()->StopBlocking();
EXPECT_TRUE(observer.stop_called);
perfetto::TrackEvent::RemoveSessionObserver(&observer);
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices, ElementsAre("I:foo.OnStart", "I:foo.OnStop"));
}
// No notifications after removing observer.
{
Observer observer;
perfetto::TrackEvent::AddSessionObserver(&observer);
perfetto::TrackEvent::RemoveSessionObserver(&observer);
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
EXPECT_FALSE(observer.setup_called);
EXPECT_FALSE(observer.start_called);
tracing_session->get()->StopBlocking();
EXPECT_FALSE(observer.stop_called);
}
// Removing observer in a callback.
{
Observer observer;
observer.unsubscribe_at_setup = true;
perfetto::TrackEvent::AddSessionObserver(&observer);
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
EXPECT_TRUE(observer.setup_called);
EXPECT_FALSE(observer.start_called);
tracing_session->get()->StopBlocking();
EXPECT_FALSE(observer.stop_called);
perfetto::TrackEvent::RemoveSessionObserver(&observer);
}
// Multiple observers.
{
Observer observer1;
Observer observer2;
perfetto::TrackEvent::AddSessionObserver(&observer1);
perfetto::TrackEvent::AddSessionObserver(&observer2);
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
tracing_session->get()->StopBlocking();
perfetto::TrackEvent::RemoveSessionObserver(&observer1);
perfetto::TrackEvent::RemoveSessionObserver(&observer2);
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices, ElementsAre("I:foo.OnStart", "I:foo.OnStart",
"I:foo.OnStop", "I:foo.OnStop"));
}
// Multiple observers with one being removed midway.
{
Observer observer1;
Observer observer2;
perfetto::TrackEvent::AddSessionObserver(&observer1);
perfetto::TrackEvent::AddSessionObserver(&observer2);
auto* tracing_session = NewTraceWithCategories({"foo"});
tracing_session->get()->StartBlocking();
perfetto::TrackEvent::RemoveSessionObserver(&observer1);
tracing_session->get()->StopBlocking();
perfetto::TrackEvent::RemoveSessionObserver(&observer2);
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices,
ElementsAre("I:foo.OnStart", "I:foo.OnStart", "I:foo.OnStop"));
}
EXPECT_FALSE(perfetto::TrackEvent::IsEnabled());
}
#if PERFETTO_BUILDFLAG(PERFETTO_COMPILER_CLANG)
struct __attribute__((capability("mutex"))) MockMutex {
void Lock() __attribute__((acquire_capability())) {}
void Unlock() __attribute__((release_capability())) {}
};
struct AnnotatedObject {
MockMutex mutex;
int value __attribute__((guarded_by(mutex))) = {};
};
TEST_P(PerfettoApiTest, ThreadSafetyAnnotation) {
AnnotatedObject obj;
// Access to the locked field is only allowed while holding the mutex.
obj.mutex.Lock();
obj.value = 1;
obj.mutex.Unlock();
auto* tracing_session = NewTraceWithCategories({"cat"});
tracing_session->get()->StartBlocking();
// It should be possible to trace the field while holding the lock.
obj.mutex.Lock();
TRACE_EVENT_INSTANT("cat", "Instant", "value", obj.value);
TRACE_EVENT_INSTANT1("cat", "InstantLegacy", 0, "value", obj.value);
{ TRACE_EVENT("cat", "Scoped", "value", obj.value); }
{ TRACE_EVENT1("cat", "ScopedLegacy", "value", obj.value); }
obj.mutex.Unlock();
perfetto::TrackEvent::Flush();
tracing_session->get()->StopBlocking();
auto slices = ReadSlicesFromTrace(tracing_session->get());
EXPECT_THAT(slices, ElementsAre("I:cat.Instant(value=(int)1)",
"I:cat.InstantLegacy(value=(int)1)",
"B:cat.Scoped(value=(int)1)", "E",
"B:cat.ScopedLegacy(value=(int)1)", "E"));
}
#endif // PERFETTO_BUILDFLAG(PERFETTO_COMPILER_CLANG)
struct BackendTypeAsString {
std::string operator()(
const ::testing::TestParamInfo<perfetto::BackendType>& info) const {
switch (info.param) {
case perfetto::kInProcessBackend:
return "InProc";
case perfetto::kSystemBackend:
return "System";
case perfetto::kCustomBackend:
return "Custom";
case perfetto::kUnspecifiedBackend:
return "Unspec";
}
return nullptr;
}
};
INSTANTIATE_TEST_SUITE_P(PerfettoApiTest,
PerfettoApiTest,
::testing::Values(perfetto::kInProcessBackend,
perfetto::kSystemBackend),
BackendTypeAsString());
} // namespace
PERFETTO_DECLARE_DATA_SOURCE_STATIC_MEMBERS(MockDataSource);
PERFETTO_DECLARE_DATA_SOURCE_STATIC_MEMBERS(MockDataSource2);
PERFETTO_DECLARE_DATA_SOURCE_STATIC_MEMBERS(TestIncrementalDataSource,
TestIncrementalDataSourceTraits);
PERFETTO_DEFINE_DATA_SOURCE_STATIC_MEMBERS(MockDataSource);
PERFETTO_DEFINE_DATA_SOURCE_STATIC_MEMBERS(MockDataSource2);
PERFETTO_DEFINE_DATA_SOURCE_STATIC_MEMBERS(TestIncrementalDataSource,
TestIncrementalDataSourceTraits);