1 /*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <unistd.h>
18
19 #include "perfetto/ext/base/file_utils.h"
20 #include "perfetto/ext/base/string_utils.h"
21 #include "perfetto/ext/base/temp_file.h"
22 #include "src/base/test/test_task_runner.h"
23 #include "src/traced/probes/common/cpu_freq_info_for_testing.h"
24 #include "src/traced/probes/sys_stats/sys_stats_data_source.h"
25 #include "src/tracing/core/trace_writer_for_testing.h"
26 #include "test/gtest_and_gmock.h"
27
28 #include "protos/perfetto/common/sys_stats_counters.gen.h"
29 #include "protos/perfetto/config/data_source_config.gen.h"
30 #include "protos/perfetto/config/sys_stats/sys_stats_config.gen.h"
31 #include "protos/perfetto/trace/sys_stats/sys_stats.gen.h"
32
33 using ::testing::_;
34 using ::testing::Invoke;
35 using ::testing::Return;
36 using ::testing::UnorderedElementsAre;
37
38 namespace perfetto {
39 namespace {
40
41 const char kMockMeminfo[] = R"(
42 MemTotal: 3744240 kB
43 MemFree: 73328 kB
44 MemAvailable: 629896 kB
45 Buffers: 19296 kB
46 Cached: 731032 kB
47 SwapCached: 4936 kB
48 Active: 1616348 kB
49 Inactive: 745492 kB
50 Active(anon): 1322636 kB
51 Inactive(anon): 449172 kB
52 Active(file): 293712 kB
53 Inactive(file): 296320 kB
54 Unevictable: 142152 kB
55 Mlocked: 142152 kB
56 SwapTotal: 524284 kB
57 SwapFree: 128 kB
58 Dirty: 0 kB
59 Writeback: 0 kB
60 AnonPages: 1751140 kB
61 Mapped: 508372 kB
62 Shmem: 18604 kB
63 Slab: 240352 kB
64 SReclaimable: 64684 kB
65 SUnreclaim: 175668 kB
66 KernelStack: 62672 kB
67 PageTables: 70108 kB
68 NFS_Unstable: 0 kB
69 Bounce: 0 kB
70 WritebackTmp: 0 kB
71 CommitLimit: 2396404 kB
72 Committed_AS: 81911488 kB
73 VmallocTotal: 258867136 kB
74 VmallocUsed: 0 kB
75 VmallocChunk: 0 kB
76 CmaTotal: 196608 kB
77 CmaFree: 60 kB)";
78
79 const char kMockVmstat[] = R"(
80 nr_free_pages 16449
81 nr_alloc_batch 79
82 nr_inactive_anon 112545
83 nr_active_anon 322027
84 nr_inactive_file 75904
85 nr_active_file 87939
86 nr_unevictable 35538
87 nr_mlock 35538
88 nr_anon_pages 429005
89 nr_mapped 125844
90 nr_file_pages 205523
91 nr_dirty 23
92 nr_writeback 0
93 nr_slab_reclaimable 15840
94 nr_slab_unreclaimable 43912
95 nr_page_table_pages 17158
96 nr_kernel_stack 3822
97 nr_overhead 0
98 nr_unstable 0
99 nr_bounce 0
100 nr_vmscan_write 558690
101 nr_vmscan_immediate_reclaim 14853
102 nr_writeback_temp 0
103 nr_isolated_anon 0
104 nr_isolated_file 0
105 nr_shmem 5027
106 nr_dirtied 6732417
107 nr_written 6945513
108 nr_pages_scanned 0
109 workingset_refault 32784684
110 workingset_activate 8200928
111 workingset_nodereclaim 0
112 nr_anon_transparent_hugepages 0
113 nr_free_cma 0
114 nr_swapcache 1254
115 nr_dirty_threshold 33922
116 nr_dirty_background_threshold 8449
117 pgpgin 161257156
118 pgpgout 35973852
119 pgpgoutclean 37181384
120 pswpin 185308
121 pswpout 557662
122 pgalloc_dma 79259070
123 pgalloc_normal 88265512
124 pgalloc_movable 0
125 pgfree 175051592
126 pgactivate 11897892
127 pgdeactivate 20412230
128 pgfault 181696234
129 pgmajfault 1060871
130 pgrefill_dma 12970047
131 pgrefill_normal 14391564
132 pgrefill_movable 0
133 pgsteal_kswapd_dma 19471476
134 pgsteal_kswapd_normal 21138380
135 pgsteal_kswapd_movable 0
136 pgsteal_direct 91537
137 pgsteal_direct_dma 40625
138 pgsteal_direct_normal 50912
139 pgsteal_direct_movable 0
140 pgscan_kswapd_dma 23544417
141 pgscan_kswapd_normal 25623715
142 pgscan_kswapd_movable 0
143 pgscan_direct_dma 50369
144 pgscan_direct_normal 66284
145 pgscan_direct_movable 0
146 pgscan_direct_throttle 0
147 pginodesteal 0
148 slabs_scanned 39582828
149 kswapd_inodesteal 110199
150 kswapd_low_wmark_hit_quickly 21321
151 kswapd_high_wmark_hit_quickly 4112
152 pageoutrun 37666
153 allocstall 1587
154 pgrotated 12086
155 drop_pagecache 0
156 drop_slab 0
157 pgmigrate_success 5923482
158 pgmigrate_fail 3439
159 compact_migrate_scanned 92906456
160 compact_free_scanned 467077168
161 compact_isolated 13456528
162 compact_stall 197
163 compact_fail 42
164 compact_success 155
165 compact_daemon_wake 2131
166 unevictable_pgs_culled 50170
167 unevictable_pgs_scanned 0
168 unevictable_pgs_rescued 14640
169 unevictable_pgs_mlocked 52520
170 unevictable_pgs_munlocked 14640
171 unevictable_pgs_cleared 2342
172 unevictable_pgs_stranded 2342)";
173
174 const char kMockStat[] = R"(
175 cpu 2655987 822682 2352153 8801203 41917 322733 175055 0 0 0
176 cpu0 762178 198125 902284 8678856 41716 152974 68262 0 0 0
177 cpu1 613833 243394 504323 15194 96 60625 28785 0 0 0
178 cpu2 207349 95060 248856 17351 42 32148 26108 0 0 0
179 cpu3 138474 92158 174852 17537 48 25076 25035 0 0 0
180 cpu4 278720 34689 141048 18117 1 20782 5873 0 0 0
181 cpu5 235376 33907 85098 18278 2 10049 3774 0 0 0
182 cpu6 239568 67149 155814 17890 5 11518 3807 0 0 0
183 cpu7 180484 58196 139874 17975 3 9556 13407 0 0 0
184 intr 238128517 0 0 0 63500984 0 6253792 6 4 5 0 0 0 0 0 0 0 160331 0 0 14 0 0 0 0 0 0 0 0 0 0 0 20430 2279 11 11 83272 0 0 0 0 0 0 0 5754 220829 0 154753 908545 1824602 7314228 0 0 0 6898259 0 0 10 0 0 2 0 0 0 0 0 0 0 42 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6 575816 1447531 134022 0 0 0 0 0 435008 319921 2755476 0 0 0 0 91 310212 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6 4 0 0 545 901 554 9 3377 4184 12 10 588851 0 2 1109045 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6 8 0 0 0 0 0 0 0 0 0 0 0 0 497 0 0 0 0 0 26172 0 0 0 0 0 0 0 1362 0 0 0 0 0 0 0 424 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 23427 0 0 0 0 1 1298 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 108 0 0 0 0 86 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1784935 407979 2140 10562241 52374 74699 6976 84926 222 169088 0 0 0 0 174 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 2789 51543 0 83 0 0 0 0 0 0 0 0 0 0 0 0 0 0 8 8 0 13 11 17 1393 0 0 0 0 0 0 0 0 0 0 26 0 0 2 106 0 0 0 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 11150 0 13 0 1 390 6 0 6 4 0 0 0 0 352 284743 2 0 0 24 3 0 3 0 0 0 12 0 668788 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 680 0 0
185 ctxt 373122860
186 btime 1536912218
187 processes 243320
188 procs_running 1
189 procs_blocked 0
190 softirq 84611084 10220177 28299167 155083 3035679 6390543 66234 4396819 15604187 0 16443195)";
191
192 const char kMockBuddy[] = R"(
193 Node 0, zone DMA 2743 1659 2063 685 27 4 0 0 0 0 0
194 Node 0, zone Normal 143 744 89 1080 105 1 0 2 0 2 2
195 Node 0, zone HighMem 345 90 156 3 5 2 0 0 0 0 0
196 Node 1, zone Normal 233 123 453 10 5 1 0 2 0 0 3)";
197
198 const char kDevfreq1[] = "1000000";
199 const char kDevfreq2[] = "20000000";
200
201 const char kMockDiskStat[] = R"(
202 253 0 zram0 13886 0 111088 128 57298 0 458384 48 0 15248 176 0 0 0 0 0 0
203 8 0 sda 54133 5368 8221736 75929 30333 1157434 9599744 143190 0 63672 249858 9595 0 2160072 19411 6649 11327
204 8 1 sda1 18 6 632 7 39 49 704 92 0 156 100 0 0 0 0 0 0)";
205
206 class TestSysStatsDataSource : public SysStatsDataSource {
207 public:
TestSysStatsDataSource(base::TaskRunner * task_runner,TracingSessionID id,std::unique_ptr<TraceWriter> writer,const DataSourceConfig & config,std::unique_ptr<CpuFreqInfo> cpu_freq_info,OpenFunction open_fn)208 TestSysStatsDataSource(base::TaskRunner* task_runner,
209 TracingSessionID id,
210 std::unique_ptr<TraceWriter> writer,
211 const DataSourceConfig& config,
212 std::unique_ptr<CpuFreqInfo> cpu_freq_info,
213 OpenFunction open_fn)
214 : SysStatsDataSource(task_runner,
215 id,
216 std::move(writer),
217 config,
218 std::move(cpu_freq_info),
219 open_fn) {}
220
221 MOCK_METHOD(base::ScopedDir, OpenDevfreqDir, (), (override));
222 MOCK_METHOD(const char*,
223 ReadDevfreqCurFreq,
224 (const std::string& deviceName),
225 (override));
226 };
227
MockOpenReadOnly(const char * path)228 base::ScopedFile MockOpenReadOnly(const char* path) {
229 base::TempFile tmp_ = base::TempFile::CreateUnlinked();
230 if (!strcmp(path, "/proc/meminfo")) {
231 EXPECT_GT(pwrite(tmp_.fd(), kMockMeminfo, strlen(kMockMeminfo), 0), 0);
232 } else if (!strcmp(path, "/proc/vmstat")) {
233 EXPECT_GT(pwrite(tmp_.fd(), kMockVmstat, strlen(kMockVmstat), 0), 0);
234 } else if (!strcmp(path, "/proc/stat")) {
235 EXPECT_GT(pwrite(tmp_.fd(), kMockStat, strlen(kMockStat), 0), 0);
236 } else if (!strcmp(path, "/proc/buddyinfo")) {
237 EXPECT_GT(pwrite(tmp_.fd(), kMockBuddy, strlen(kMockBuddy), 0), 0);
238 } else if (!strcmp(path, "/proc/diskstats")) {
239 EXPECT_GT(pwrite(tmp_.fd(), kMockDiskStat, strlen(kMockDiskStat), 0), 0);
240 } else {
241 PERFETTO_FATAL("Unexpected file opened %s", path);
242 }
243 return tmp_.ReleaseFD();
244 }
245
246 class SysStatsDataSourceTest : public ::testing::Test {
247 protected:
GetSysStatsDataSource(const DataSourceConfig & cfg)248 std::unique_ptr<TestSysStatsDataSource> GetSysStatsDataSource(
249 const DataSourceConfig& cfg) {
250 auto writer =
251 std::unique_ptr<TraceWriterForTesting>(new TraceWriterForTesting());
252 writer_raw_ = writer.get();
253 auto instance =
254 std::unique_ptr<TestSysStatsDataSource>(new TestSysStatsDataSource(
255 &task_runner_, 0, std::move(writer), cfg,
256 cpu_freq_info_for_testing_.GetInstance(), MockOpenReadOnly));
257 instance->set_ns_per_user_hz_for_testing(1000000000ull / 100); // 100 Hz.
258 instance->Start();
259 return instance;
260 }
261
Poller(SysStatsDataSource * ds,std::function<void ()> checkpoint)262 void Poller(SysStatsDataSource* ds, std::function<void()> checkpoint) {
263 if (ds->tick_for_testing())
264 checkpoint();
265 else
266 task_runner_.PostDelayedTask(
267 [ds, checkpoint, this] { Poller(ds, checkpoint); }, 1);
268 }
269
WaitTick(SysStatsDataSource * data_source)270 void WaitTick(SysStatsDataSource* data_source) {
271 auto checkpoint = task_runner_.CreateCheckpoint("on_tick");
272 Poller(data_source, checkpoint);
273 task_runner_.RunUntilCheckpoint("on_tick");
274 }
275
276 TraceWriterForTesting* writer_raw_ = nullptr;
277 base::TestTaskRunner task_runner_;
278 CpuFreqInfoForTesting cpu_freq_info_for_testing_;
279 };
280
TEST_F(SysStatsDataSourceTest,Meminfo)281 TEST_F(SysStatsDataSourceTest, Meminfo) {
282 using C = protos::gen::MeminfoCounters;
283 DataSourceConfig config;
284 protos::gen::SysStatsConfig sys_cfg;
285 sys_cfg.set_meminfo_period_ms(10);
286 sys_cfg.add_meminfo_counters(C::MEMINFO_MEM_TOTAL);
287 sys_cfg.add_meminfo_counters(C::MEMINFO_MEM_FREE);
288 sys_cfg.add_meminfo_counters(C::MEMINFO_ACTIVE_ANON);
289 sys_cfg.add_meminfo_counters(C::MEMINFO_INACTIVE_FILE);
290 sys_cfg.add_meminfo_counters(C::MEMINFO_CMA_FREE);
291 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
292 auto data_source = GetSysStatsDataSource(config);
293
294 WaitTick(data_source.get());
295
296 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
297 ASSERT_TRUE(packet.has_sys_stats());
298 const auto& sys_stats = packet.sys_stats();
299 EXPECT_EQ(sys_stats.vmstat_size(), 0);
300 EXPECT_EQ(sys_stats.buddy_info_size(), 0);
301 EXPECT_EQ(sys_stats.cpu_stat_size(), 0);
302 EXPECT_EQ(sys_stats.devfreq_size(), 0);
303
304 using KV = std::pair<int, uint64_t>;
305 std::vector<KV> kvs;
306 for (const auto& kv : sys_stats.meminfo())
307 kvs.push_back({kv.key(), kv.value()});
308
309 EXPECT_THAT(kvs,
310 UnorderedElementsAre(KV{C::MEMINFO_MEM_TOTAL, 3744240}, //
311 KV{C::MEMINFO_MEM_FREE, 73328}, //
312 KV{C::MEMINFO_ACTIVE_ANON, 1322636}, //
313 KV{C::MEMINFO_INACTIVE_FILE, 296320}, //
314 KV{C::MEMINFO_CMA_FREE, 60}));
315 }
316
TEST_F(SysStatsDataSourceTest,MeminfoAll)317 TEST_F(SysStatsDataSourceTest, MeminfoAll) {
318 DataSourceConfig config;
319 protos::gen::SysStatsConfig sys_cfg;
320 sys_cfg.set_meminfo_period_ms(10);
321 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
322 auto data_source = GetSysStatsDataSource(config);
323
324 WaitTick(data_source.get());
325
326 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
327 ASSERT_TRUE(packet.has_sys_stats());
328 const auto& sys_stats = packet.sys_stats();
329 EXPECT_EQ(sys_stats.vmstat_size(), 0);
330 EXPECT_EQ(sys_stats.buddy_info_size(), 0);
331 EXPECT_EQ(sys_stats.cpu_stat_size(), 0);
332 EXPECT_EQ(sys_stats.devfreq_size(), 0);
333 EXPECT_GE(sys_stats.meminfo_size(), 10);
334 }
335
TEST_F(SysStatsDataSourceTest,Vmstat)336 TEST_F(SysStatsDataSourceTest, Vmstat) {
337 using C = protos::gen::VmstatCounters;
338 DataSourceConfig config;
339 protos::gen::SysStatsConfig sys_cfg;
340 sys_cfg.set_vmstat_period_ms(10);
341 sys_cfg.add_vmstat_counters(C::VMSTAT_NR_FREE_PAGES);
342 sys_cfg.add_vmstat_counters(C::VMSTAT_PGACTIVATE);
343 sys_cfg.add_vmstat_counters(C::VMSTAT_PGMIGRATE_FAIL);
344 sys_cfg.add_vmstat_counters(C::VMSTAT_PGSTEAL_DIRECT);
345 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
346 auto data_source = GetSysStatsDataSource(config);
347
348 WaitTick(data_source.get());
349
350 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
351 ASSERT_TRUE(packet.has_sys_stats());
352 const auto& sys_stats = packet.sys_stats();
353 EXPECT_EQ(sys_stats.meminfo_size(), 0);
354 EXPECT_EQ(sys_stats.cpu_stat_size(), 0);
355 EXPECT_EQ(sys_stats.devfreq_size(), 0);
356
357 using KV = std::pair<int, uint64_t>;
358 std::vector<KV> kvs;
359 for (const auto& kv : sys_stats.vmstat())
360 kvs.push_back({kv.key(), kv.value()});
361
362 EXPECT_THAT(kvs, UnorderedElementsAre(KV{C::VMSTAT_NR_FREE_PAGES, 16449}, //
363 KV{C::VMSTAT_PGACTIVATE, 11897892}, //
364 KV{C::VMSTAT_PGMIGRATE_FAIL, 3439}, //
365 KV{C::VMSTAT_PGSTEAL_DIRECT, 91537} //
366 ));
367 }
368
TEST_F(SysStatsDataSourceTest,VmstatAll)369 TEST_F(SysStatsDataSourceTest, VmstatAll) {
370 DataSourceConfig config;
371 protos::gen::SysStatsConfig sys_cfg;
372 sys_cfg.set_vmstat_period_ms(10);
373 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
374 auto data_source = GetSysStatsDataSource(config);
375
376 WaitTick(data_source.get());
377
378 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
379 ASSERT_TRUE(packet.has_sys_stats());
380 const auto& sys_stats = packet.sys_stats();
381 EXPECT_EQ(sys_stats.meminfo_size(), 0);
382 EXPECT_EQ(sys_stats.cpu_stat_size(), 0);
383 EXPECT_EQ(sys_stats.devfreq_size(), 0);
384 EXPECT_EQ(sys_stats.buddy_info_size(), 0);
385 EXPECT_GE(sys_stats.vmstat_size(), 10);
386 }
387
TEST_F(SysStatsDataSourceTest,BuddyinfoAll)388 TEST_F(SysStatsDataSourceTest, BuddyinfoAll) {
389 DataSourceConfig config;
390 protos::gen::SysStatsConfig sys_cfg;
391 sys_cfg.set_buddyinfo_period_ms(10);
392 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
393 auto data_source = GetSysStatsDataSource(config);
394
395 WaitTick(data_source.get());
396
397 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
398 ASSERT_TRUE(packet.has_sys_stats());
399 const auto& sys_stats = packet.sys_stats();
400 EXPECT_EQ(sys_stats.meminfo_size(), 0);
401 EXPECT_EQ(sys_stats.cpu_stat_size(), 0);
402 EXPECT_EQ(sys_stats.devfreq_size(), 0);
403 EXPECT_GE(sys_stats.vmstat_size(), 0);
404 EXPECT_EQ(sys_stats.buddy_info_size(), 4);
405
406 EXPECT_EQ(sys_stats.buddy_info()[0].node(), "0");
407 EXPECT_EQ(sys_stats.buddy_info()[0].zone(), "DMA");
408 EXPECT_EQ(sys_stats.buddy_info()[0].order_pages()[0], 2743u);
409 EXPECT_EQ(sys_stats.buddy_info()[0].order_pages()[5], 4u);
410 EXPECT_EQ(sys_stats.buddy_info()[0].order_pages()[10], 0u);
411
412 EXPECT_EQ(sys_stats.buddy_info()[1].node(), "0");
413 EXPECT_EQ(sys_stats.buddy_info()[1].zone(), "Normal");
414 EXPECT_EQ(sys_stats.buddy_info()[1].order_pages()[0], 143u);
415 EXPECT_EQ(sys_stats.buddy_info()[1].order_pages()[5], 1u);
416 EXPECT_EQ(sys_stats.buddy_info()[1].order_pages()[10], 2u);
417
418 EXPECT_EQ(sys_stats.buddy_info()[2].node(), "0");
419 EXPECT_EQ(sys_stats.buddy_info()[2].zone(), "HighMem");
420 EXPECT_EQ(sys_stats.buddy_info()[2].order_pages()[0], 345u);
421 EXPECT_EQ(sys_stats.buddy_info()[2].order_pages()[5], 2u);
422 EXPECT_EQ(sys_stats.buddy_info()[2].order_pages()[10], 0u);
423
424 EXPECT_EQ(sys_stats.buddy_info()[3].node(), "1");
425 EXPECT_EQ(sys_stats.buddy_info()[3].zone(), "Normal");
426 EXPECT_EQ(sys_stats.buddy_info()[3].order_pages()[0], 233u);
427 EXPECT_EQ(sys_stats.buddy_info()[3].order_pages()[5], 1u);
428 EXPECT_EQ(sys_stats.buddy_info()[3].order_pages()[10], 3u);
429 }
430
TEST_F(SysStatsDataSourceTest,DevfreqAll)431 TEST_F(SysStatsDataSourceTest, DevfreqAll) {
432 DataSourceConfig config;
433 protos::gen::SysStatsConfig sys_cfg;
434 sys_cfg.set_devfreq_period_ms(10);
435 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
436 auto data_source = GetSysStatsDataSource(config);
437
438 // Create dirs and symlinks, but only read the symlinks.
439 std::vector<std::string> dirs_to_delete;
440 std::vector<std::string> symlinks_to_delete;
441 auto make_devfreq_paths = [&symlinks_to_delete, &dirs_to_delete](
442 base::TempDir& temp_dir, base::TempDir& sym_dir,
443 const char* name) {
444 base::StackString<256> path("%s/%s", temp_dir.path().c_str(), name);
445 dirs_to_delete.push_back(path.ToStdString());
446 mkdir(path.c_str(), 0755);
447 base::StackString<256> sym_path("%s/%s", sym_dir.path().c_str(), name);
448 symlinks_to_delete.push_back(sym_path.ToStdString());
449 symlink(path.c_str(), sym_path.c_str());
450 };
451 auto fake_devfreq = base::TempDir::Create();
452 auto fake_devfreq_symdir = base::TempDir::Create();
453 static const char* const devfreq_names[] = {"10010.devfreq_device_a",
454 "10020.devfreq_device_b"};
455 for (auto dev : devfreq_names) {
456 make_devfreq_paths(fake_devfreq, fake_devfreq_symdir, dev);
457 }
458
459 EXPECT_CALL(*data_source, OpenDevfreqDir())
460 .WillRepeatedly(Invoke([&fake_devfreq_symdir] {
461 return base::ScopedDir(opendir(fake_devfreq_symdir.path().c_str()));
462 }));
463 EXPECT_CALL(*data_source, ReadDevfreqCurFreq("10010.devfreq_device_a"))
464 .WillRepeatedly(Return(kDevfreq1));
465 EXPECT_CALL(*data_source, ReadDevfreqCurFreq("10020.devfreq_device_b"))
466 .WillRepeatedly(Return(kDevfreq2));
467
468 WaitTick(data_source.get());
469
470 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
471 ASSERT_TRUE(packet.has_sys_stats());
472 const auto& sys_stats = packet.sys_stats();
473 EXPECT_EQ(sys_stats.meminfo_size(), 0);
474 EXPECT_EQ(sys_stats.cpu_stat_size(), 0);
475
476 using KV = std::pair<std::string, uint64_t>;
477 std::vector<KV> kvs;
478 for (const auto& kv : sys_stats.devfreq())
479 kvs.push_back({kv.key(), kv.value()});
480 EXPECT_THAT(kvs,
481 UnorderedElementsAre(KV{"10010.devfreq_device_a", 1000000},
482 KV{"10020.devfreq_device_b", 20000000}));
483 for (const std::string& path : dirs_to_delete)
484 base::Rmdir(path);
485 for (const std::string& path : symlinks_to_delete)
486 remove(path.c_str());
487 }
488
TEST_F(SysStatsDataSourceTest,StatAll)489 TEST_F(SysStatsDataSourceTest, StatAll) {
490 DataSourceConfig config;
491 protos::gen::SysStatsConfig sys_cfg;
492 sys_cfg.set_stat_period_ms(10);
493 config.set_sys_stats_config_raw(sys_cfg.SerializeAsString());
494 auto data_source = GetSysStatsDataSource(config);
495
496 WaitTick(data_source.get());
497
498 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
499 ASSERT_TRUE(packet.has_sys_stats());
500 const auto& sys_stats = packet.sys_stats();
501 EXPECT_EQ(sys_stats.meminfo_size(), 0);
502 EXPECT_EQ(sys_stats.vmstat_size(), 0);
503 EXPECT_EQ(sys_stats.buddy_info_size(), 0);
504
505 ASSERT_EQ(sys_stats.cpu_stat_size(), 8);
506 EXPECT_EQ(sys_stats.cpu_stat()[0].user_ns(), 762178 * 10000000ull);
507 EXPECT_EQ(sys_stats.cpu_stat()[0].system_mode_ns(), 902284 * 10000000ull);
508 EXPECT_EQ(sys_stats.cpu_stat()[0].softirq_ns(), 68262 * 10000000ull);
509 EXPECT_EQ(sys_stats.cpu_stat()[7].user_ns(), 180484 * 10000000ull);
510 EXPECT_EQ(sys_stats.cpu_stat()[7].system_mode_ns(), 139874 * 10000000ull);
511 EXPECT_EQ(sys_stats.cpu_stat()[7].softirq_ns(), 13407 * 10000000ull);
512
513 EXPECT_EQ(sys_stats.num_forks(), 243320u);
514
515 EXPECT_EQ(sys_stats.num_irq_total(), 238128517u);
516 ASSERT_EQ(sys_stats.num_irq_size(), 102);
517 EXPECT_EQ(sys_stats.num_irq()[0].count(), 63500984u);
518 EXPECT_EQ(sys_stats.num_irq()[0].irq(), 3);
519 EXPECT_EQ(sys_stats.num_irq()[1].count(), 6253792u);
520 EXPECT_EQ(sys_stats.num_irq()[1].irq(), 5);
521 EXPECT_EQ(sys_stats.num_irq()[101].count(), 680u);
522
523 EXPECT_EQ(sys_stats.num_softirq_total(), 84611084u);
524 ASSERT_EQ(sys_stats.num_softirq_size(), 10);
525 EXPECT_EQ(sys_stats.num_softirq()[0].count(), 10220177u);
526 EXPECT_EQ(sys_stats.num_softirq()[9].count(), 16443195u);
527
528 EXPECT_EQ(sys_stats.num_softirq_total(), 84611084u);
529 }
530
TEST_F(SysStatsDataSourceTest,StatForksOnly)531 TEST_F(SysStatsDataSourceTest, StatForksOnly) {
532 protos::gen::SysStatsConfig cfg;
533 cfg.set_stat_period_ms(10);
534 cfg.add_stat_counters(protos::gen::SysStatsConfig::STAT_FORK_COUNT);
535 DataSourceConfig config_obj;
536 config_obj.set_sys_stats_config_raw(cfg.SerializeAsString());
537 auto data_source = GetSysStatsDataSource(config_obj);
538
539 WaitTick(data_source.get());
540
541 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
542 ASSERT_TRUE(packet.has_sys_stats());
543 const auto& sys_stats = packet.sys_stats();
544 EXPECT_EQ(sys_stats.meminfo_size(), 0);
545 EXPECT_EQ(sys_stats.vmstat_size(), 0);
546 EXPECT_EQ(sys_stats.buddy_info_size(), 0);
547 ASSERT_EQ(sys_stats.cpu_stat_size(), 0);
548 EXPECT_EQ(sys_stats.num_forks(), 243320u);
549 EXPECT_EQ(sys_stats.num_irq_total(), 0u);
550 ASSERT_EQ(sys_stats.num_irq_size(), 0);
551 EXPECT_EQ(sys_stats.num_softirq_total(), 0u);
552 ASSERT_EQ(sys_stats.num_softirq_size(), 0);
553 }
554
TEST_F(SysStatsDataSourceTest,Cpufreq)555 TEST_F(SysStatsDataSourceTest, Cpufreq) {
556 protos::gen::SysStatsConfig cfg;
557 cfg.set_cpufreq_period_ms(10);
558 DataSourceConfig config_obj;
559 config_obj.set_sys_stats_config_raw(cfg.SerializeAsString());
560 auto data_source = GetSysStatsDataSource(config_obj);
561
562 WaitTick(data_source.get());
563
564 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
565 ASSERT_TRUE(packet.has_sys_stats());
566 const auto& sys_stats = packet.sys_stats();
567 EXPECT_GT(sys_stats.cpufreq_khz_size(), 0);
568 EXPECT_EQ(sys_stats.cpufreq_khz()[0], 2650000u);
569 if (sys_stats.cpufreq_khz_size() > 1) {
570 // We emulated 2 CPUs but it is possible the test system is single core.
571 EXPECT_EQ(sys_stats.cpufreq_khz()[1], 3698200u);
572 }
573 for (unsigned int i = 2;
574 i < static_cast<unsigned int>(sys_stats.cpufreq_khz_size()); i++) {
575 // For cpux which scaling_cur_freq was not emulated in unittest, cpufreq
576 // should be recorded as 0
577 EXPECT_EQ(sys_stats.cpufreq_khz()[i], 0u);
578 }
579 }
580
TEST_F(SysStatsDataSourceTest,DiskStat)581 TEST_F(SysStatsDataSourceTest, DiskStat) {
582 protos::gen::SysStatsConfig cfg;
583 cfg.set_diskstat_period_ms(10);
584 DataSourceConfig config_obj;
585 config_obj.set_sys_stats_config_raw(cfg.SerializeAsString());
586 auto data_source = GetSysStatsDataSource(config_obj);
587
588 WaitTick(data_source.get());
589
590 protos::gen::TracePacket packet = writer_raw_->GetOnlyTracePacket();
591 ASSERT_TRUE(packet.has_sys_stats());
592 const auto& sys_stats = packet.sys_stats();
593 EXPECT_EQ(sys_stats.disk_stat_size(), 3);
594
595 EXPECT_EQ(sys_stats.disk_stat()[0].device_name(), "zram0");
596 EXPECT_EQ(sys_stats.disk_stat()[0].read_sectors(), 111088u);
597 EXPECT_EQ(sys_stats.disk_stat()[0].write_sectors(), 458384u);
598 EXPECT_EQ(sys_stats.disk_stat()[0].discard_sectors(), 0u);
599 EXPECT_EQ(sys_stats.disk_stat()[0].flush_count(), 0u);
600 EXPECT_EQ(sys_stats.disk_stat()[0].read_time_ms(), 128u);
601 EXPECT_EQ(sys_stats.disk_stat()[0].write_time_ms(), 48u);
602 EXPECT_EQ(sys_stats.disk_stat()[0].discard_time_ms(), 0u);
603 EXPECT_EQ(sys_stats.disk_stat()[0].flush_time_ms(), 0u);
604 EXPECT_EQ(sys_stats.disk_stat()[1].device_name(), "sda");
605 EXPECT_EQ(sys_stats.disk_stat()[1].read_sectors(), 8221736u);
606 EXPECT_EQ(sys_stats.disk_stat()[1].write_sectors(), 9599744u);
607 EXPECT_EQ(sys_stats.disk_stat()[1].discard_sectors(), 2160072u);
608 EXPECT_EQ(sys_stats.disk_stat()[1].flush_count(), 6649u);
609 EXPECT_EQ(sys_stats.disk_stat()[1].read_time_ms(), 75929u);
610 EXPECT_EQ(sys_stats.disk_stat()[1].write_time_ms(), 143190u);
611 EXPECT_EQ(sys_stats.disk_stat()[1].discard_time_ms(), 19411u);
612 EXPECT_EQ(sys_stats.disk_stat()[1].flush_time_ms(), 11327u);
613 EXPECT_EQ(sys_stats.disk_stat()[2].device_name(), "sda1");
614 EXPECT_EQ(sys_stats.disk_stat()[2].read_sectors(), 632u);
615 EXPECT_EQ(sys_stats.disk_stat()[2].write_sectors(), 704u);
616 EXPECT_EQ(sys_stats.disk_stat()[2].discard_sectors(), 0u);
617 EXPECT_EQ(sys_stats.disk_stat()[2].flush_count(), 0u);
618 EXPECT_EQ(sys_stats.disk_stat()[2].read_time_ms(), 7u);
619 EXPECT_EQ(sys_stats.disk_stat()[2].write_time_ms(), 92u);
620 EXPECT_EQ(sys_stats.disk_stat()[2].discard_time_ms(), 0u);
621 EXPECT_EQ(sys_stats.disk_stat()[2].flush_time_ms(), 0u);
622 }
623
624 } // namespace
625 } // namespace perfetto
626