blob: 01c61b11205f7dff92dc799e2ba494d501a2dd11 [file] [log] [blame]
/*
* Copyright (C) 2024 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 <cstdint>
#include <string>
#include "perfetto/base/status.h"
#include "src/base/test/status_matchers.h"
#include "src/trace_redaction/build_timeline.h"
#include "src/trace_redaction/optimize_timeline.h"
#include "src/trace_redaction/scrub_process_stats.h"
#include "src/trace_redaction/trace_redaction_framework.h"
#include "src/trace_redaction/trace_redaction_integration_fixture.h"
#include "src/trace_redaction/trace_redactor.h"
#include "test/gtest_and_gmock.h"
#include "protos/perfetto/trace/ps/process_stats.pbzero.h"
#include "protos/perfetto/trace/trace.pbzero.h"
#include "protos/perfetto/trace/trace_packet.pbzero.h"
namespace perfetto::trace_redaction {
class ScrubProcessStatsTest : public testing::Test,
protected TraceRedactionIntegrationFixure {
protected:
void SetUp() override {
trace_redactor()->emplace_collect<BuildTimeline>();
trace_redactor()->emplace_build<OptimizeTimeline>();
trace_redactor()->emplace_transform<ScrubProcessStats>();
// Package "com.Unity.com.unity.multiplayer.samples.coop";
context()->package_uid = 10252;
}
// Gets pids from all process_stats messages in the trace (bytes).
base::FlatSet<int32_t> GetAllPids(const std::string& bytes) const {
base::FlatSet<int32_t> pids;
protos::pbzero::Trace::Decoder decoder(bytes);
for (auto packet = decoder.packet(); packet; ++packet) {
protos::pbzero::TracePacket::Decoder trace_packet(packet->as_bytes());
if (!trace_packet.has_process_stats()) {
continue;
}
protos::pbzero::ProcessStats::Decoder process_stats(
trace_packet.process_stats());
for (auto process = process_stats.processes(); process; ++process) {
protos::pbzero::ProcessStats::Process::Decoder p(process->as_bytes());
PERFETTO_DCHECK(p.has_pid());
pids.insert(p.pid());
}
}
return pids;
}
};
// This test is a canary for changes to the test data. If the test data was to
// change, every test in this file would fail.
//
// SELECT DISTINCT pid
// FROM process
// WHERE upid IN (
// SELECT DISTINCT upid
// FROM counter
// JOIN process_counter_track ON counter.track_id=process_counter_track.id
// WHERE name!='oom_score_adj'
// )
// ORDER BY pid
//
// NOTE: WHERE name!='oom_score_adj' is used because there are two sources for
// oom_score_adj values and we only want process stats here.
TEST_F(ScrubProcessStatsTest, VerifyTraceStats) {
base::FlatSet<int32_t> expected = {
1, 578, 581, 696, 697, 698, 699, 700, 701, 704,
709, 710, 718, 728, 749, 750, 751, 752, 756, 760,
761, 762, 873, 874, 892, 1046, 1047, 1073, 1074, 1091,
1092, 1093, 1101, 1103, 1104, 1105, 1106, 1107, 1110, 1111,
1112, 1113, 1115, 1116, 1118, 1119, 1120, 1121, 1123, 1124,
1125, 1126, 1127, 1129, 1130, 1131, 1133, 1140, 1145, 1146,
1147, 1151, 1159, 1163, 1164, 1165, 1166, 1167, 1168, 1175,
1177, 1205, 1206, 1235, 1237, 1238, 1248, 1251, 1254, 1255,
1295, 1296, 1298, 1300, 1301, 1303, 1304, 1312, 1317, 1325,
1339, 1340, 1363, 1374, 1379, 1383, 1388, 1392, 1408, 1409,
1410, 1413, 1422, 1426, 1427, 1428, 1429, 1433, 1436, 1448,
1450, 1451, 1744, 1774, 1781, 1814, 2262, 2268, 2286, 2392,
2456, 2502, 2510, 2518, 2528, 2569, 3171, 3195, 3262, 3286,
3310, 3338, 3442, 3955, 4386, 4759, 5935, 6034, 6062, 6167,
6547, 6573, 6720, 6721, 6725, 6944, 6984, 7105, 7207, 7557,
7636, 7786, 7874, 7958, 7960, 7967, 15449, 15685, 15697, 16453,
19683, 21124, 21839, 23150, 23307, 23876, 24317, 25017, 25126, 25450,
25474, 27271, 30604, 32289,
};
auto original = LoadOriginal();
ASSERT_OK(original) << original.status().c_message();
auto actual = GetAllPids(*original);
for (auto pid : expected) {
ASSERT_TRUE(actual.count(pid))
<< "pid " << pid << " was not found in the trace";
}
for (auto pid : actual) {
ASSERT_TRUE(expected.count(pid))
<< "pid " << pid << " was found in the trace";
}
}
// Package name: "com.Unity.com.unity.multiplayer.samples.coop"
// Package pid: 7105
TEST_F(ScrubProcessStatsTest, OnlyKeepsStatsForPackage) {
auto result = Redact();
ASSERT_OK(result) << result.c_message();
auto redacted = LoadRedacted();
ASSERT_OK(redacted) << redacted.status().c_message();
auto actual = GetAllPids(*redacted);
ASSERT_EQ(actual.size(), 1u);
ASSERT_TRUE(actual.count(7105));
}
} // namespace perfetto::trace_redaction