blob: 05cdca70f704163ce0a4e8bc004b49e62a8145af [file] [log] [blame]
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <memory>
#include <string>
#include <thread>
#include <utility>
#include "base/functional/bind.h"
#include "base/run_loop.h"
#include "base/strings/strcat.h"
#include "base/test/bind.h"
#include "base/threading/thread.h"
#include "base/tracing/perfetto_platform.h"
#include "services/tracing/perfetto/perfetto_service.h"
#include "services/tracing/perfetto/producer_host.h"
#include "services/tracing/perfetto/test_utils.h"
#include "services/tracing/public/cpp/perfetto/producer_client.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/perfetto/include/perfetto/tracing/tracing.h"
#include "third_party/perfetto/protos/perfetto/common/commit_data_request.pb.h"
// TODO(crbug.com/961066): Fix memory leaks in tests and re-enable on LSAN.
#ifdef LEAK_SANITIZER
#define MAYBE_DifferentSharedMemoryBuffersForDifferentAgents \
DISABLED_DifferentSharedMemoryBuffersForDifferentAgents
#else
#define MAYBE_DifferentSharedMemoryBuffersForDifferentAgents \
DifferentSharedMemoryBuffersForDifferentAgents
#endif
namespace tracing {
namespace {
const char kPerfettoTestDataSourceName[] =
"org.chromium.chrome_integration_unittest";
std::string GetPerfettoProducerName() {
return base::StrCat({mojom::kPerfettoProducerNamePrefix, "123"});
}
class PerfettoIntegrationTest : public TracingUnitTest {
public:
void SetUp() override {
TracingUnitTest::SetUp();
data_source_ = TestDataSource::CreateAndRegisterDataSource(
kPerfettoTestDataSourceName, 0);
perfetto_service_ = std::make_unique<PerfettoService>();
RunUntilIdle();
}
void TearDown() override {
perfetto_service_.reset();
TracingUnitTest::TearDown();
}
PerfettoService* perfetto_service() const { return perfetto_service_.get(); }
protected:
std::unique_ptr<TestDataSource> data_source_;
std::unique_ptr<PerfettoService> perfetto_service_;
};
TEST_F(PerfettoIntegrationTest, ProducerDatasourceInitialized) {
std::unique_ptr<MockProducerClient::Handle> dummy_client =
MockProducerClient::Create();
base::RunLoop producer_initialized_runloop;
auto new_producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**dummy_client, producer_initialized_runloop.QuitClosure());
producer_initialized_runloop.Run();
}
TEST_F(PerfettoIntegrationTest, ClientEnabledAndDisabled) {
base::RunLoop on_trace_packets;
MockConsumer consumer({kPerfettoTestDataSourceName},
perfetto_service()->GetService(),
[&on_trace_packets](bool has_more) {
EXPECT_FALSE(has_more);
on_trace_packets.Quit();
});
base::RunLoop client_enabled_callback;
base::RunLoop client_disabled_callback;
std::unique_ptr<MockProducerClient::Handle> client =
MockProducerClient::Create(
/* num_data_sources = */ 1, client_enabled_callback.QuitClosure(),
client_disabled_callback.QuitClosure());
auto producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client);
client_enabled_callback.Run();
RunUntilIdle();
consumer.StopTracing();
client_disabled_callback.Run();
on_trace_packets.Run();
EXPECT_EQ(0u, consumer.received_test_packets());
}
TEST_F(PerfettoIntegrationTest, PacketsEndToEndProducerFirst) {
const size_t kNumPackets = 10;
data_source_->set_send_packet_count(kNumPackets);
base::RunLoop client_enabled_callback;
base::RunLoop client_disabled_callback;
std::unique_ptr<MockProducerClient::Handle> client =
MockProducerClient::Create(
/* num_data_sources = */ 1, client_enabled_callback.QuitClosure(),
client_disabled_callback.QuitClosure());
auto producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client);
base::RunLoop no_more_packets_runloop;
MockConsumer consumer({kPerfettoTestDataSourceName},
perfetto_service()->GetService(),
[&no_more_packets_runloop](bool has_more) {
if (!has_more) {
no_more_packets_runloop.Quit();
}
});
client_enabled_callback.Run();
RunUntilIdle();
consumer.StopTracing();
client_disabled_callback.Run();
no_more_packets_runloop.Run();
EXPECT_EQ(kNumPackets, consumer.received_test_packets());
}
TEST_F(PerfettoIntegrationTest, PacketsEndToEndConsumerFirst) {
const size_t kNumPackets = 10;
data_source_->set_send_packet_count(kNumPackets);
base::RunLoop no_more_packets_runloop;
MockConsumer consumer({kPerfettoTestDataSourceName},
perfetto_service()->GetService(),
[&no_more_packets_runloop](bool has_more) {
if (!has_more) {
no_more_packets_runloop.Quit();
}
});
base::RunLoop client_enabled_callback;
std::unique_ptr<MockProducerClient::Handle> client =
MockProducerClient::Create(
/* num_data_sources = */ 1, client_enabled_callback.QuitClosure());
auto new_producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client);
client_enabled_callback.Run();
RunUntilIdle();
consumer.StopTracing();
no_more_packets_runloop.Run();
EXPECT_EQ(kNumPackets, consumer.received_test_packets());
}
TEST_F(PerfettoIntegrationTest, CommitDataRequestIsMaybeComplete) {
const size_t kNumPackets = 100;
data_source_->set_send_packet_count(kNumPackets);
base::RunLoop no_more_packets_runloop;
MockConsumer consumer({kPerfettoTestDataSourceName},
perfetto_service()->GetService(),
[&no_more_packets_runloop](bool has_more) {
if (!has_more) {
no_more_packets_runloop.Quit();
}
});
base::RunLoop client_enabled_callback;
std::unique_ptr<MockProducerClient::Handle> client =
MockProducerClient::Create(
/* num_data_sources = */ 1, client_enabled_callback.QuitClosure());
auto new_producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client);
client_enabled_callback.Run();
base::RunLoop wait_for_packet_write;
PerfettoTracedProcess::Get()
->GetTaskRunner()
->GetOrCreateTaskRunner()
->PostTaskAndReply(FROM_HERE,
base::BindOnce(&TestDataSource::WritePacketBigly,
base::Unretained(data_source_.get())),
wait_for_packet_write.QuitClosure());
wait_for_packet_write.Run();
RunUntilIdle();
consumer.StopTracing();
no_more_packets_runloop.Run();
// {cli,host}_reqs are a std::vector<std::string>. Each entry of the vector
// is a proto-serialized CommitDataRequest.
const auto& cli_reqs = (*client)->all_client_commit_data_requests();
const auto& host_reqs = new_producer->all_host_commit_data_requests();
ASSERT_EQ(cli_reqs.size(), host_reqs.size());
for (size_t i = 0; i < cli_reqs.size(); i++) {
// Note that the proto-encoded strings are not identical. This is because
// perfetto doesn't emit unset fields. But then when going over mojo these
// unset fields get copied as 0-value (it's fine), so on the host side we
// see extra fields being explicitly zero-initialized. Here we force a
// re-encode using the libprotobuf message for the sake of the comparison.
// libprotobuf will re-normalize messages before serializing.
perfetto::protos::CommitDataRequest cli_req;
ASSERT_TRUE(cli_req.ParseFromString(cli_reqs[i]));
perfetto::protos::CommitDataRequest host_req;
ASSERT_TRUE(host_req.ParseFromString(cli_reqs[i]));
ASSERT_EQ(cli_req.SerializeAsString(), host_req.SerializeAsString());
}
}
TEST_F(PerfettoIntegrationTest, TracingRestarted) {
const size_t kNumPackets = 10;
data_source_->set_send_packet_count(kNumPackets);
base::RunLoop no_more_packets_runloop;
MockConsumer consumer({kPerfettoTestDataSourceName},
perfetto_service()->GetService(),
[&no_more_packets_runloop](bool has_more) {
if (!has_more) {
no_more_packets_runloop.Quit();
}
});
base::RunLoop client_enabled_callback;
base::RunLoop client_disabled_callback;
std::unique_ptr<MockProducerClient::Handle> client =
MockProducerClient::Create(
/* num_data_sources = */ 1, client_enabled_callback.QuitClosure(),
client_disabled_callback.QuitClosure());
auto new_producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client);
client_enabled_callback.Run();
RunUntilIdle();
perfetto::SharedMemory* first_session_shm =
(*client)->shared_memory_for_testing();
consumer.StopTracing();
client_disabled_callback.Run();
no_more_packets_runloop.Run();
EXPECT_EQ(kNumPackets, consumer.received_test_packets());
consumer.FreeBuffers();
base::RunLoop client_reenabled_callback;
(*client)->SetAgentEnabledCallback(client_reenabled_callback.QuitClosure());
consumer.StartTracing();
client_reenabled_callback.Run();
RunUntilIdle();
// We should still be using the same shared memory.
EXPECT_EQ(first_session_shm, (*client)->shared_memory_for_testing());
base::RunLoop client_redisabled_callback;
(*client)->SetAgentDisabledCallback(client_redisabled_callback.QuitClosure());
consumer.StopTracing();
client_redisabled_callback.Run();
EXPECT_EQ(kNumPackets * 2, consumer.received_test_packets());
}
TEST_F(PerfettoIntegrationTest, NoPacketsReceivedOnWrongSourceName) {
const size_t kNumPackets = 10;
data_source_->set_send_packet_count(kNumPackets);
base::RunLoop client_enabled_callback;
base::RunLoop client_disabled_callback;
std::unique_ptr<MockProducerClient::Handle> client =
MockProducerClient::Create(
/* num_data_sources = */ 0, client_enabled_callback.QuitClosure(),
client_disabled_callback.QuitClosure());
base::RunLoop producer_initialized_runloop;
auto new_producer = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), "fake_data_source", perfetto_service(), **client);
base::RunLoop no_more_packets_runloop;
MockConsumer consumer({"fake_data_source"}, perfetto_service()->GetService(),
[&no_more_packets_runloop](bool has_more) {
if (!has_more) {
no_more_packets_runloop.Quit();
}
});
RunUntilIdle();
consumer.StopTracing();
no_more_packets_runloop.Run();
EXPECT_EQ(0u, consumer.received_test_packets());
}
TEST_F(PerfettoIntegrationTest,
MAYBE_DifferentSharedMemoryBuffersForDifferentAgents) {
base::RunLoop client1_enabled_callback;
base::RunLoop client2_enabled_callback;
std::unique_ptr<MockProducerClient::Handle> client1 =
MockProducerClient::Create(
/* num_data_sources = */ 1, client1_enabled_callback.QuitClosure());
auto producer1 = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client1);
// Start the trace here, this is because we need to send the EnableTracing
// call to client1, but constructing client2 will override the
// |producer_client_| pointer in PerfettoTracedProcess::Get() so we wait until
// client1 has been enabled before constructing the second producer client.
MockConsumer consumer({kPerfettoTestDataSourceName},
perfetto_service()->GetService(), nullptr);
client1_enabled_callback.Run();
// client2 will trigger a StartTracing call without shutting down the data
// source first, to prevent this hitting a DCHECK set the previous producer to
// null.
data_source_->ClearProducerForTesting();
std::unique_ptr<MockProducerClient::Handle> client2 =
MockProducerClient::Create(
/* num_data_sources = */ 1, client2_enabled_callback.QuitClosure());
auto producer2 = std::make_unique<MockProducerHost>(
GetPerfettoProducerName(), kPerfettoTestDataSourceName, perfetto_service(),
**client2);
client2_enabled_callback.Run();
EXPECT_TRUE((*client1)->shared_memory_for_testing());
EXPECT_TRUE((*client2)->shared_memory_for_testing());
EXPECT_NE((*client1)->shared_memory_for_testing(),
(*client2)->shared_memory_for_testing());
}
TEST_F(PerfettoIntegrationTest, PerfettoPlatformTest) {
auto* platform =
PerfettoTracedProcess::Get()->perfetto_platform_for_testing();
auto* tls = platform->GetOrCreateThreadLocalObject();
EXPECT_TRUE(tls);
EXPECT_EQ(tls, platform->GetOrCreateThreadLocalObject());
base::Thread thread("TestThread");
thread.Start();
thread.task_runner()->PostTask(
FROM_HERE, base::BindLambdaForTesting([&platform, &tls] {
auto* thread_tls = platform->GetOrCreateThreadLocalObject();
EXPECT_TRUE(thread_tls);
EXPECT_NE(tls, thread_tls);
}));
thread.Stop();
}
TEST_F(PerfettoIntegrationTest, PerfettoClientLibraryTest) {
// Check that PerfettoTracedProcess initialized the client library. Functional
// client library tests are in TracingServiceTest.
EXPECT_TRUE(perfetto::Tracing::IsInitialized());
}
} // namespace
} // namespace tracing