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 <sys/mman.h>
18 #include <sys/types.h>
19 #include <sys/wait.h>
20 #include <unistd.h>
21 
22 #include <gtest/gtest.h>
23 
24 #include <filesystem>
25 #include <string>
26 #include <vector>
27 
28 #include <meminfo/androidprocheaps.h>
29 #include <meminfo/pageacct.h>
30 #include <meminfo/procmeminfo.h>
31 #include <meminfo/sysmeminfo.h>
32 #include <vintf/VintfObject.h>
33 
34 #include <android-base/file.h>
35 #include <android-base/logging.h>
36 #include <android-base/properties.h>
37 #include <android-base/stringprintf.h>
38 
39 using namespace std;
40 using namespace android::meminfo;
41 using android::vintf::KernelVersion;
42 using android::vintf::RuntimeInfo;
43 using android::vintf::VintfObject;
44 
45 namespace fs = std::filesystem;
46 
47 pid_t pid = -1;
48 
TEST(ProcMemInfo,TestWorkingTestReset)49 TEST(ProcMemInfo, TestWorkingTestReset) {
50     // Expect reset to succeed
51     EXPECT_TRUE(ProcMemInfo::ResetWorkingSet(pid));
52 }
53 
TEST(ProcMemInfo,UsageEmpty)54 TEST(ProcMemInfo, UsageEmpty) {
55     // If we created the object for getting working set,
56     // the usage must be empty
57     ProcMemInfo proc_mem(pid, true);
58     const MemUsage& usage = proc_mem.Usage();
59     EXPECT_EQ(usage.rss, 0);
60     EXPECT_EQ(usage.vss, 0);
61     EXPECT_EQ(usage.pss, 0);
62     EXPECT_EQ(usage.uss, 0);
63     EXPECT_EQ(usage.swap, 0);
64 }
65 
TEST(ProcMemInfo,MapsNotEmpty)66 TEST(ProcMemInfo, MapsNotEmpty) {
67     // Make sure the process maps are never empty
68     ProcMemInfo proc_mem(pid);
69     const std::vector<Vma>& maps = proc_mem.Maps();
70     EXPECT_FALSE(maps.empty());
71 }
72 
TEST(ProcMemInfo,MapsUsageNotEmpty)73 TEST(ProcMemInfo, MapsUsageNotEmpty) {
74     ProcMemInfo proc_mem(pid);
75     const std::vector<Vma>& maps = proc_mem.Maps();
76     EXPECT_FALSE(maps.empty());
77     uint64_t total_pss = 0;
78     uint64_t total_rss = 0;
79     uint64_t total_uss = 0;
80     for (auto& map : maps) {
81         ASSERT_NE(0, map.usage.vss);
82         total_rss += map.usage.rss;
83         total_pss += map.usage.pss;
84         total_uss += map.usage.uss;
85     }
86 
87     // Crude check that stats are actually being read.
88     EXPECT_NE(0, total_rss) << "RSS zero for all maps, that is not possible.";
89     EXPECT_NE(0, total_pss) << "PSS zero for all maps, that is not possible.";
90     EXPECT_NE(0, total_uss) << "USS zero for all maps, that is not possible.";
91 }
92 
TEST(ProcMemInfo,MapsUsageEmpty)93 TEST(ProcMemInfo, MapsUsageEmpty) {
94     ProcMemInfo proc_mem(pid);
95     const std::vector<Vma>& maps = proc_mem.MapsWithoutUsageStats();
96     EXPECT_FALSE(maps.empty());
97     // Verify that all usage stats are zero in every map.
98     for (auto& map : maps) {
99         ASSERT_EQ(0, map.usage.vss);
100         ASSERT_EQ(0, map.usage.rss);
101         ASSERT_EQ(0, map.usage.pss);
102         ASSERT_EQ(0, map.usage.uss);
103         ASSERT_EQ(0, map.usage.swap);
104         ASSERT_EQ(0, map.usage.swap_pss);
105         ASSERT_EQ(0, map.usage.private_clean);
106         ASSERT_EQ(0, map.usage.private_dirty);
107         ASSERT_EQ(0, map.usage.shared_clean);
108         ASSERT_EQ(0, map.usage.shared_dirty);
109     }
110 }
111 
TEST(ProcMemInfo,MapsUsageFillInLater)112 TEST(ProcMemInfo, MapsUsageFillInLater) {
113     ProcMemInfo proc_mem(pid);
114     const std::vector<Vma>& maps = proc_mem.MapsWithoutUsageStats();
115     EXPECT_FALSE(maps.empty());
116     for (auto& map : maps) {
117         Vma update_map(map);
118         ASSERT_EQ(map.start, update_map.start);
119         ASSERT_EQ(map.end, update_map.end);
120         ASSERT_EQ(map.offset, update_map.offset);
121         ASSERT_EQ(map.flags, update_map.flags);
122         ASSERT_EQ(map.name, update_map.name);
123         ASSERT_EQ(0, update_map.usage.vss);
124         ASSERT_EQ(0, update_map.usage.rss);
125         ASSERT_EQ(0, update_map.usage.pss);
126         ASSERT_EQ(0, update_map.usage.uss);
127         ASSERT_EQ(0, update_map.usage.swap);
128         ASSERT_EQ(0, update_map.usage.swap_pss);
129         ASSERT_EQ(0, update_map.usage.private_clean);
130         ASSERT_EQ(0, update_map.usage.private_dirty);
131         ASSERT_EQ(0, update_map.usage.shared_clean);
132         ASSERT_EQ(0, update_map.usage.shared_dirty);
133         ASSERT_TRUE(proc_mem.FillInVmaStats(update_map));
134         // Check that at least one usage stat was updated.
135         ASSERT_NE(0, update_map.usage.vss);
136     }
137 }
138 
TEST(ProcMemInfo,MapsUsageFillInAll)139 TEST(ProcMemInfo, MapsUsageFillInAll) {
140     ProcMemInfo proc_mem(pid);
141     const std::vector<Vma>& maps = proc_mem.MapsWithoutUsageStats();
142     EXPECT_FALSE(maps.empty());
143     for (auto& map : maps) {
144         ASSERT_EQ(0, map.usage.vss);
145         ASSERT_EQ(0, map.usage.rss);
146         ASSERT_EQ(0, map.usage.pss);
147         ASSERT_EQ(0, map.usage.uss);
148         ASSERT_EQ(0, map.usage.swap);
149         ASSERT_EQ(0, map.usage.swap_pss);
150         ASSERT_EQ(0, map.usage.private_clean);
151         ASSERT_EQ(0, map.usage.private_dirty);
152         ASSERT_EQ(0, map.usage.shared_clean);
153         ASSERT_EQ(0, map.usage.shared_dirty);
154     }
155     // GetUsageStats' non-default parameter get_wss is false by default in
156     // ProcMemInfo's constructor.
157     ASSERT_TRUE(proc_mem.GetUsageStats(false));
158     for (auto& map : maps) {
159         // Check that at least one usage stat was updated.
160         ASSERT_NE(0, map.usage.vss);
161     }
162 }
163 
TEST(ProcMemInfo,PageMapPresent)164 TEST(ProcMemInfo, PageMapPresent) {
165     static constexpr size_t kNumPages = 20;
166     size_t pagesize = getpagesize();
167     void* ptr = mmap(nullptr, pagesize * (kNumPages + 2), PROT_READ | PROT_WRITE,
168                      MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
169     ASSERT_NE(MAP_FAILED, ptr);
170 
171     // Unmap the first page and the last page so that we guarantee this
172     // map is in a map by itself.
173     ASSERT_EQ(0, munmap(ptr, pagesize));
174     uintptr_t addr = reinterpret_cast<uintptr_t>(ptr) + pagesize;
175     ASSERT_EQ(0, munmap(reinterpret_cast<void*>(addr + kNumPages * pagesize), pagesize));
176 
177     ProcMemInfo proc_mem(getpid());
178     const std::vector<Vma>& maps = proc_mem.MapsWithoutUsageStats();
179     ASSERT_FALSE(maps.empty());
180 
181     // Find the vma associated with our previously created map.
182     const Vma* test_vma = nullptr;
183     for (const Vma& vma : maps) {
184         if (vma.start == addr) {
185             test_vma = &vma;
186             break;
187         }
188     }
189     ASSERT_TRUE(test_vma != nullptr) << "Cannot find test map.";
190 
191     // Verify that none of the pages are listed as present.
192     std::vector<uint64_t> pagemap;
193     ASSERT_TRUE(proc_mem.PageMap(*test_vma, &pagemap));
194     ASSERT_EQ(kNumPages, pagemap.size());
195     for (size_t i = 0; i < pagemap.size(); i++) {
196         EXPECT_FALSE(android::meminfo::page_present(pagemap[i]))
197                 << "Page " << i << " is present and it should not be.";
198     }
199 
200     // Make some of the pages present and verify that we see them
201     // as present.
202     uint8_t* data = reinterpret_cast<uint8_t*>(addr);
203     data[0] = 1;
204     data[pagesize * 5] = 1;
205     data[pagesize * 11] = 1;
206 
207     ASSERT_TRUE(proc_mem.PageMap(*test_vma, &pagemap));
208     ASSERT_EQ(kNumPages, pagemap.size());
209     for (size_t i = 0; i < pagemap.size(); i++) {
210         if (i == 0 || i == 5 || i == 11) {
211             EXPECT_TRUE(android::meminfo::page_present(pagemap[i]))
212                     << "Page " << i << " is not present and it should be.";
213         } else {
214             EXPECT_FALSE(android::meminfo::page_present(pagemap[i]))
215                     << "Page " << i << " is present and it should not be.";
216         }
217     }
218 
219     ASSERT_EQ(0, munmap(reinterpret_cast<void*>(addr), kNumPages * pagesize));
220 }
221 
TEST(ProcMemInfo,WssEmpty)222 TEST(ProcMemInfo, WssEmpty) {
223     // If we created the object for getting usage,
224     // the working set must be empty
225     ProcMemInfo proc_mem(pid, false);
226     const MemUsage& wss = proc_mem.Wss();
227     EXPECT_EQ(wss.rss, 0);
228     EXPECT_EQ(wss.vss, 0);
229     EXPECT_EQ(wss.pss, 0);
230     EXPECT_EQ(wss.uss, 0);
231     EXPECT_EQ(wss.swap, 0);
232 }
233 
TEST(ProcMemInfo,SwapOffsetsEmpty)234 TEST(ProcMemInfo, SwapOffsetsEmpty) {
235     // If we created the object for getting working set,
236     // the swap offsets must be empty
237     ProcMemInfo proc_mem(pid, true);
238     const std::vector<uint64_t>& swap_offsets = proc_mem.SwapOffsets();
239     EXPECT_EQ(swap_offsets.size(), 0);
240 }
241 
TEST(ProcMemInfo,IsSmapsSupportedTest)242 TEST(ProcMemInfo, IsSmapsSupportedTest) {
243     // Check if /proc/self/smaps_rollup exists using the API.
244     bool supported = IsSmapsRollupSupported();
245     EXPECT_EQ(!access("/proc/self/smaps_rollup", F_OK | R_OK), supported);
246 }
247 
TEST(ProcMemInfo,SmapsOrRollupTest)248 TEST(ProcMemInfo, SmapsOrRollupTest) {
249     // Make sure we can parse 'smaps_rollup' correctly
250     std::string rollup =
251             R"rollup(12c00000-7fe859e000 ---p 00000000 00:00 0                                [rollup]
252 Rss:              331908 kB
253 Pss:              202052 kB
254 Shared_Clean:     158492 kB
255 Shared_Dirty:      18928 kB
256 Private_Clean:     90472 kB
257 Private_Dirty:     64016 kB
258 Referenced:       318700 kB
259 Anonymous:         81984 kB
260 AnonHugePages:         0 kB
261 Shared_Hugetlb:        0 kB
262 Private_Hugetlb:       0 kB
263 Swap:               5344 kB
264 SwapPss:             442 kB
265 Locked:          1523537 kB)rollup";
266 
267     TemporaryFile tf;
268     ASSERT_TRUE(tf.fd != -1);
269     ASSERT_TRUE(::android::base::WriteStringToFd(rollup, tf.fd));
270 
271     MemUsage stats;
272     ASSERT_EQ(SmapsOrRollupFromFile(tf.path, &stats), true);
273     EXPECT_EQ(stats.rss, 331908);
274     EXPECT_EQ(stats.pss, 202052);
275     EXPECT_EQ(stats.uss, 154488);
276     EXPECT_EQ(stats.private_clean, 90472);
277     EXPECT_EQ(stats.private_dirty, 64016);
278     EXPECT_EQ(stats.swap_pss, 442);
279 }
280 
TEST(ProcMemInfo,SmapsOrRollupSmapsTest)281 TEST(ProcMemInfo, SmapsOrRollupSmapsTest) {
282     // Make sure /proc/<pid>/smaps is parsed correctly
283     std::string smaps =
284             R"smaps(12c00000-13440000 rw-p 00000000 00:00 0                                  [anon:dalvik-main space (region space)]
285 Name:           [anon:dalvik-main space (region space)]
286 Size:               8448 kB
287 KernelPageSize:        4 kB
288 MMUPageSize:           4 kB
289 Rss:                2652 kB
290 Pss:                2652 kB
291 Shared_Clean:        840 kB
292 Shared_Dirty:         40 kB
293 Private_Clean:        84 kB
294 Private_Dirty:      2652 kB
295 Referenced:         2652 kB
296 Anonymous:          2652 kB
297 AnonHugePages:         0 kB
298 ShmemPmdMapped:        0 kB
299 Shared_Hugetlb:        0 kB
300 Private_Hugetlb:       0 kB
301 Swap:                102 kB
302 SwapPss:              70 kB
303 Locked:             2652 kB
304 VmFlags: rd wr mr mw me ac
305 )smaps";
306 
307     TemporaryFile tf;
308     ASSERT_TRUE(tf.fd != -1);
309     ASSERT_TRUE(::android::base::WriteStringToFd(smaps, tf.fd));
310 
311     MemUsage stats;
312     ASSERT_EQ(SmapsOrRollupFromFile(tf.path, &stats), true);
313     EXPECT_EQ(stats.rss, 2652);
314     EXPECT_EQ(stats.pss, 2652);
315     EXPECT_EQ(stats.uss, 2736);
316     EXPECT_EQ(stats.private_clean, 84);
317     EXPECT_EQ(stats.private_dirty, 2652);
318     EXPECT_EQ(stats.swap_pss, 70);
319 }
320 
TEST(ProcMemInfo,SmapsOrRollupPssRollupTest)321 TEST(ProcMemInfo, SmapsOrRollupPssRollupTest) {
322     // Make sure /proc/<pid>/smaps is parsed correctly
323     // to get the PSS
324     std::string smaps =
325             R"smaps(12c00000-13440000 rw-p 00000000 00:00 0                                  [anon:dalvik-main space (region space)]
326 Name:           [anon:dalvik-main space (region space)]
327 Size:               8448 kB
328 KernelPageSize:        4 kB
329 MMUPageSize:           4 kB
330 Rss:                2652 kB
331 Pss:                2652 kB
332 Shared_Clean:        840 kB
333 Shared_Dirty:         40 kB
334 Private_Clean:        84 kB
335 Private_Dirty:      2652 kB
336 Referenced:         2652 kB
337 Anonymous:          2652 kB
338 AnonHugePages:         0 kB
339 ShmemPmdMapped:        0 kB
340 Shared_Hugetlb:        0 kB
341 Private_Hugetlb:       0 kB
342 Swap:                102 kB
343 SwapPss:              70 kB
344 Locked:             2652 kB
345 VmFlags: rd wr mr mw me ac
346 )smaps";
347 
348     TemporaryFile tf;
349     ASSERT_TRUE(tf.fd != -1);
350     ASSERT_TRUE(::android::base::WriteStringToFd(smaps, tf.fd));
351 
352     uint64_t pss;
353     ASSERT_EQ(SmapsOrRollupPssFromFile(tf.path, &pss), true);
354     EXPECT_EQ(pss, 2652);
355 }
356 
TEST(ProcMemInfo,SmapsOrRollupPssSmapsTest)357 TEST(ProcMemInfo, SmapsOrRollupPssSmapsTest) {
358     // Correctly parse smaps file to gather pss
359     std::string exec_dir = ::android::base::GetExecutableDirectory();
360     std::string path = ::android::base::StringPrintf("%s/testdata1/smaps_short", exec_dir.c_str());
361 
362     uint64_t pss;
363     ASSERT_EQ(SmapsOrRollupPssFromFile(path, &pss), true);
364     EXPECT_EQ(pss, 19119);
365 }
366 
TEST(ProcMemInfo,StatusVmRSSTest)367 TEST(ProcMemInfo, StatusVmRSSTest) {
368     std::string exec_dir = ::android::base::GetExecutableDirectory();
369     std::string path = ::android::base::StringPrintf("%s/testdata1/status", exec_dir.c_str());
370 
371     uint64_t rss;
372     ASSERT_EQ(StatusVmRSSFromFile(path, &rss), true);
373     EXPECT_EQ(rss, 730764);
374 }
375 
TEST(ProcMemInfo,StatusVmRSSBogusFileTest)376 TEST(ProcMemInfo, StatusVmRSSBogusFileTest) {
377     std::string exec_dir = ::android::base::GetExecutableDirectory();
378     std::string path = ::android::base::StringPrintf("%s/testdata1/smaps", exec_dir.c_str());
379 
380     uint64_t rss;
381     ASSERT_EQ(StatusVmRSSFromFile(path, &rss), false);
382 }
383 
TEST(ProcMemInfo,ForEachExistingVmaTest)384 TEST(ProcMemInfo, ForEachExistingVmaTest) {
385     std::string exec_dir = ::android::base::GetExecutableDirectory();
386     std::string path = ::android::base::StringPrintf("%s/testdata1/smaps_short", exec_dir.c_str());
387     ProcMemInfo proc_mem(pid);
388     // Populate maps_.
389     proc_mem.Smaps(path);
390     std::vector<Vma> vmas;
391     auto collect_vmas = [&](const Vma& v) {
392         vmas.push_back(v);
393         return true;
394     };
395     EXPECT_TRUE(proc_mem.ForEachExistingVma(collect_vmas));
396 
397     // The size of vmas is not checked because Smaps() will return 5 vmas on x86
398     // and 6 vmas otherwise, as [vsyscall] is not processed on x86.
399 
400     // Expect values to be equal to what we have in testdata1/smaps_short
401     // Check for names
402     EXPECT_EQ(vmas[0].name, "[anon:dalvik-zygote-jit-code-cache]");
403     EXPECT_EQ(vmas[1].name, "/system/framework/x86_64/boot-framework.art");
404     EXPECT_TRUE(vmas[2].name == "[anon:libc_malloc]" ||
405                 vmas[2].name.starts_with("[anon:scudo:"))
406             << "Unknown map name " << vmas[2].name;
407     EXPECT_EQ(vmas[3].name, "/system/priv-app/SettingsProvider/oat/x86_64/SettingsProvider.odex");
408     EXPECT_EQ(vmas[4].name, "/system/lib64/libhwui.so");
409 
410     // Check start address
411     EXPECT_EQ(vmas[0].start, 0x54c00000);
412     EXPECT_EQ(vmas[1].start, 0x701ea000);
413     EXPECT_EQ(vmas[2].start, 0x70074dd8d000);
414     EXPECT_EQ(vmas[3].start, 0x700755a2d000);
415     EXPECT_EQ(vmas[4].start, 0x7007f85b0000);
416 
417     // Check end address
418     EXPECT_EQ(vmas[0].end, 0x56c00000);
419     EXPECT_EQ(vmas[1].end, 0x70cdb000);
420     EXPECT_EQ(vmas[2].end, 0x70074ee0d000);
421     EXPECT_EQ(vmas[3].end, 0x700755a6e000);
422     EXPECT_EQ(vmas[4].end, 0x7007f8b9b000);
423 
424     // Check Flags
425     EXPECT_EQ(vmas[0].flags, PROT_READ | PROT_EXEC);
426     EXPECT_EQ(vmas[1].flags, PROT_READ | PROT_WRITE);
427     EXPECT_EQ(vmas[2].flags, PROT_READ | PROT_WRITE);
428     EXPECT_EQ(vmas[3].flags, PROT_READ | PROT_EXEC);
429     EXPECT_EQ(vmas[4].flags, PROT_READ | PROT_EXEC);
430 
431     // Check Shared
432     EXPECT_FALSE(vmas[0].is_shared);
433     EXPECT_FALSE(vmas[1].is_shared);
434     EXPECT_FALSE(vmas[2].is_shared);
435     EXPECT_FALSE(vmas[3].is_shared);
436     EXPECT_FALSE(vmas[4].is_shared);
437 
438     // Check Offset
439     EXPECT_EQ(vmas[0].offset, 0x0);
440     EXPECT_EQ(vmas[1].offset, 0x0);
441     EXPECT_EQ(vmas[2].offset, 0x0);
442     EXPECT_EQ(vmas[3].offset, 0x00016000);
443     EXPECT_EQ(vmas[4].offset, 0x001ee000);
444 
445     // Check Inode
446     EXPECT_EQ(vmas[0].inode, 0);
447     EXPECT_EQ(vmas[1].inode, 3165);
448     EXPECT_EQ(vmas[2].inode, 0);
449     EXPECT_EQ(vmas[3].inode, 1947);
450     EXPECT_EQ(vmas[4].inode, 1537);
451 
452     // Check smaps specific fields
453     ASSERT_EQ(vmas[0].usage.vss, 32768);
454     EXPECT_EQ(vmas[1].usage.vss, 11204);
455     EXPECT_EQ(vmas[2].usage.vss, 16896);
456     EXPECT_EQ(vmas[3].usage.vss, 260);
457     EXPECT_EQ(vmas[4].usage.vss, 6060);
458 
459     EXPECT_EQ(vmas[0].usage.rss, 2048);
460     EXPECT_EQ(vmas[1].usage.rss, 11188);
461     EXPECT_EQ(vmas[2].usage.rss, 15272);
462     EXPECT_EQ(vmas[3].usage.rss, 260);
463     EXPECT_EQ(vmas[4].usage.rss, 4132);
464 
465     EXPECT_EQ(vmas[0].usage.pss, 113);
466     EXPECT_EQ(vmas[1].usage.pss, 2200);
467     EXPECT_EQ(vmas[2].usage.pss, 15272);
468     EXPECT_EQ(vmas[3].usage.pss, 260);
469     EXPECT_EQ(vmas[4].usage.pss, 1274);
470 
471     EXPECT_EQ(vmas[0].usage.uss, 0);
472     EXPECT_EQ(vmas[1].usage.uss, 1660);
473     EXPECT_EQ(vmas[2].usage.uss, 15272);
474     EXPECT_EQ(vmas[3].usage.uss, 260);
475     EXPECT_EQ(vmas[4].usage.uss, 0);
476 
477     EXPECT_EQ(vmas[0].usage.private_clean, 0);
478     EXPECT_EQ(vmas[1].usage.private_clean, 0);
479     EXPECT_EQ(vmas[2].usage.private_clean, 0);
480     EXPECT_EQ(vmas[3].usage.private_clean, 260);
481     EXPECT_EQ(vmas[4].usage.private_clean, 0);
482 
483     EXPECT_EQ(vmas[0].usage.private_dirty, 0);
484     EXPECT_EQ(vmas[1].usage.private_dirty, 1660);
485     EXPECT_EQ(vmas[2].usage.private_dirty, 15272);
486     EXPECT_EQ(vmas[3].usage.private_dirty, 0);
487     EXPECT_EQ(vmas[4].usage.private_dirty, 0);
488 
489     EXPECT_EQ(vmas[0].usage.shared_clean, 0);
490     EXPECT_EQ(vmas[1].usage.shared_clean, 80);
491     EXPECT_EQ(vmas[2].usage.shared_clean, 0);
492     EXPECT_EQ(vmas[3].usage.shared_clean, 0);
493     EXPECT_EQ(vmas[4].usage.shared_clean, 4132);
494 
495     EXPECT_EQ(vmas[0].usage.shared_dirty, 2048);
496     EXPECT_EQ(vmas[1].usage.shared_dirty, 9448);
497     EXPECT_EQ(vmas[2].usage.shared_dirty, 0);
498     EXPECT_EQ(vmas[3].usage.shared_dirty, 0);
499     EXPECT_EQ(vmas[4].usage.shared_dirty, 0);
500 
501     EXPECT_EQ(vmas[0].usage.swap, 0);
502     EXPECT_EQ(vmas[1].usage.swap, 0);
503     EXPECT_EQ(vmas[2].usage.swap, 0);
504     EXPECT_EQ(vmas[3].usage.swap, 0);
505     EXPECT_EQ(vmas[4].usage.swap, 0);
506 
507     EXPECT_EQ(vmas[0].usage.swap_pss, 0);
508     EXPECT_EQ(vmas[1].usage.swap_pss, 0);
509     EXPECT_EQ(vmas[2].usage.swap_pss, 0);
510     EXPECT_EQ(vmas[3].usage.swap_pss, 0);
511     EXPECT_EQ(vmas[4].usage.swap_pss, 0);
512 
513 #ifndef __x86_64__
514     // vmas[5] will not exist on x86, as [vsyscall] would not be processed.
515     EXPECT_EQ(vmas[5].name, "[vsyscall]");
516     EXPECT_EQ(vmas[5].start, 0xffffffffff600000);
517     EXPECT_EQ(vmas[5].end, 0xffffffffff601000);
518     EXPECT_EQ(vmas[5].flags, PROT_READ | PROT_EXEC);
519     EXPECT_FALSE(vmas[5].is_shared);
520     EXPECT_EQ(vmas[5].offset, 0x0);
521     EXPECT_EQ(vmas[5].inode, 0);
522     EXPECT_EQ(vmas[5].usage.vss, 4);
523     EXPECT_EQ(vmas[5].usage.rss, 0);
524     EXPECT_EQ(vmas[5].usage.pss, 0);
525     EXPECT_EQ(vmas[5].usage.uss, 0);
526     EXPECT_EQ(vmas[5].usage.private_clean, 0);
527     EXPECT_EQ(vmas[5].usage.private_dirty, 0);
528     EXPECT_EQ(vmas[5].usage.shared_clean, 0);
529     EXPECT_EQ(vmas[5].usage.shared_dirty, 0);
530     EXPECT_EQ(vmas[5].usage.swap, 0);
531     EXPECT_EQ(vmas[5].usage.swap_pss, 0);
532 #endif
533 }
534 
TEST(ProcMemInfo,ForEachVmaFromFile_SmapsTest)535 TEST(ProcMemInfo, ForEachVmaFromFile_SmapsTest) {
536     // Parse smaps file correctly to make callbacks for each virtual memory area (vma)
537     std::string exec_dir = ::android::base::GetExecutableDirectory();
538     std::string path = ::android::base::StringPrintf("%s/testdata1/smaps_short", exec_dir.c_str());
539     ProcMemInfo proc_mem(pid);
540 
541     std::vector<Vma> vmas;
542     auto collect_vmas = [&](const Vma& v) {
543         vmas.push_back(v);
544         return true;
545     };
546     ASSERT_TRUE(ForEachVmaFromFile(path, collect_vmas));
547 
548     // We should get a total of 6 vmas
549     ASSERT_EQ(vmas.size(), 6);
550 
551     // Expect values to be equal to what we have in testdata1/smaps_short
552     // Check for names
553     EXPECT_EQ(vmas[0].name, "[anon:dalvik-zygote-jit-code-cache]");
554     EXPECT_EQ(vmas[1].name, "/system/framework/x86_64/boot-framework.art");
555     EXPECT_TRUE(vmas[2].name == "[anon:libc_malloc]" ||
556                 vmas[2].name.starts_with("[anon:scudo:"))
557             << "Unknown map name " << vmas[2].name;
558     EXPECT_EQ(vmas[3].name, "/system/priv-app/SettingsProvider/oat/x86_64/SettingsProvider.odex");
559     EXPECT_EQ(vmas[4].name, "/system/lib64/libhwui.so");
560     EXPECT_EQ(vmas[5].name, "[vsyscall]");
561 
562     // Check start address
563     EXPECT_EQ(vmas[0].start, 0x54c00000);
564     EXPECT_EQ(vmas[1].start, 0x701ea000);
565     EXPECT_EQ(vmas[2].start, 0x70074dd8d000);
566     EXPECT_EQ(vmas[3].start, 0x700755a2d000);
567     EXPECT_EQ(vmas[4].start, 0x7007f85b0000);
568     EXPECT_EQ(vmas[5].start, 0xffffffffff600000);
569 
570     // Check end address
571     EXPECT_EQ(vmas[0].end, 0x56c00000);
572     EXPECT_EQ(vmas[1].end, 0x70cdb000);
573     EXPECT_EQ(vmas[2].end, 0x70074ee0d000);
574     EXPECT_EQ(vmas[3].end, 0x700755a6e000);
575     EXPECT_EQ(vmas[4].end, 0x7007f8b9b000);
576     EXPECT_EQ(vmas[5].end, 0xffffffffff601000);
577 
578     // Check Flags
579     EXPECT_EQ(vmas[0].flags, PROT_READ | PROT_EXEC);
580     EXPECT_EQ(vmas[1].flags, PROT_READ | PROT_WRITE);
581     EXPECT_EQ(vmas[2].flags, PROT_READ | PROT_WRITE);
582     EXPECT_EQ(vmas[3].flags, PROT_READ | PROT_EXEC);
583     EXPECT_EQ(vmas[4].flags, PROT_READ | PROT_EXEC);
584     EXPECT_EQ(vmas[5].flags, PROT_READ | PROT_EXEC);
585 
586     // Check Shared
587     EXPECT_FALSE(vmas[0].is_shared);
588     EXPECT_FALSE(vmas[1].is_shared);
589     EXPECT_FALSE(vmas[2].is_shared);
590     EXPECT_FALSE(vmas[3].is_shared);
591     EXPECT_FALSE(vmas[4].is_shared);
592     EXPECT_FALSE(vmas[5].is_shared);
593 
594     // Check Offset
595     EXPECT_EQ(vmas[0].offset, 0x0);
596     EXPECT_EQ(vmas[1].offset, 0x0);
597     EXPECT_EQ(vmas[2].offset, 0x0);
598     EXPECT_EQ(vmas[3].offset, 0x00016000);
599     EXPECT_EQ(vmas[4].offset, 0x001ee000);
600     EXPECT_EQ(vmas[5].offset, 0x0);
601 
602     // Check Inode
603     EXPECT_EQ(vmas[0].inode, 0);
604     EXPECT_EQ(vmas[1].inode, 3165);
605     EXPECT_EQ(vmas[2].inode, 0);
606     EXPECT_EQ(vmas[3].inode, 1947);
607     EXPECT_EQ(vmas[4].inode, 1537);
608     EXPECT_EQ(vmas[5].inode, 0);
609 
610     // Check smaps specific fields
611     ASSERT_EQ(vmas[0].usage.vss, 32768);
612     EXPECT_EQ(vmas[1].usage.vss, 11204);
613     EXPECT_EQ(vmas[2].usage.vss, 16896);
614     EXPECT_EQ(vmas[3].usage.vss, 260);
615     EXPECT_EQ(vmas[4].usage.vss, 6060);
616     EXPECT_EQ(vmas[5].usage.vss, 4);
617 
618     EXPECT_EQ(vmas[0].usage.rss, 2048);
619     EXPECT_EQ(vmas[1].usage.rss, 11188);
620     EXPECT_EQ(vmas[2].usage.rss, 15272);
621     EXPECT_EQ(vmas[3].usage.rss, 260);
622     EXPECT_EQ(vmas[4].usage.rss, 4132);
623     EXPECT_EQ(vmas[5].usage.rss, 0);
624 
625     EXPECT_EQ(vmas[0].usage.pss, 113);
626     EXPECT_EQ(vmas[1].usage.pss, 2200);
627     EXPECT_EQ(vmas[2].usage.pss, 15272);
628     EXPECT_EQ(vmas[3].usage.pss, 260);
629     EXPECT_EQ(vmas[4].usage.pss, 1274);
630     EXPECT_EQ(vmas[5].usage.pss, 0);
631 
632     EXPECT_EQ(vmas[0].usage.uss, 0);
633     EXPECT_EQ(vmas[1].usage.uss, 1660);
634     EXPECT_EQ(vmas[2].usage.uss, 15272);
635     EXPECT_EQ(vmas[3].usage.uss, 260);
636     EXPECT_EQ(vmas[4].usage.uss, 0);
637     EXPECT_EQ(vmas[5].usage.uss, 0);
638 
639     EXPECT_EQ(vmas[0].usage.private_clean, 0);
640     EXPECT_EQ(vmas[1].usage.private_clean, 0);
641     EXPECT_EQ(vmas[2].usage.private_clean, 0);
642     EXPECT_EQ(vmas[3].usage.private_clean, 260);
643     EXPECT_EQ(vmas[4].usage.private_clean, 0);
644     EXPECT_EQ(vmas[5].usage.private_clean, 0);
645 
646     EXPECT_EQ(vmas[0].usage.private_dirty, 0);
647     EXPECT_EQ(vmas[1].usage.private_dirty, 1660);
648     EXPECT_EQ(vmas[2].usage.private_dirty, 15272);
649     EXPECT_EQ(vmas[3].usage.private_dirty, 0);
650     EXPECT_EQ(vmas[4].usage.private_dirty, 0);
651     EXPECT_EQ(vmas[5].usage.private_dirty, 0);
652 
653     EXPECT_EQ(vmas[0].usage.shared_clean, 0);
654     EXPECT_EQ(vmas[1].usage.shared_clean, 80);
655     EXPECT_EQ(vmas[2].usage.shared_clean, 0);
656     EXPECT_EQ(vmas[3].usage.shared_clean, 0);
657     EXPECT_EQ(vmas[4].usage.shared_clean, 4132);
658     EXPECT_EQ(vmas[5].usage.shared_clean, 0);
659 
660     EXPECT_EQ(vmas[0].usage.shared_dirty, 2048);
661     EXPECT_EQ(vmas[1].usage.shared_dirty, 9448);
662     EXPECT_EQ(vmas[2].usage.shared_dirty, 0);
663     EXPECT_EQ(vmas[3].usage.shared_dirty, 0);
664     EXPECT_EQ(vmas[4].usage.shared_dirty, 0);
665     EXPECT_EQ(vmas[5].usage.shared_dirty, 0);
666 
667     EXPECT_EQ(vmas[0].usage.swap, 0);
668     EXPECT_EQ(vmas[1].usage.swap, 0);
669     EXPECT_EQ(vmas[2].usage.swap, 0);
670     EXPECT_EQ(vmas[3].usage.swap, 0);
671     EXPECT_EQ(vmas[4].usage.swap, 0);
672     EXPECT_EQ(vmas[5].usage.swap, 0);
673 
674     EXPECT_EQ(vmas[0].usage.swap_pss, 0);
675     EXPECT_EQ(vmas[1].usage.swap_pss, 0);
676     EXPECT_EQ(vmas[2].usage.swap_pss, 0);
677     EXPECT_EQ(vmas[3].usage.swap_pss, 0);
678     EXPECT_EQ(vmas[4].usage.swap_pss, 0);
679     EXPECT_EQ(vmas[5].usage.swap_pss, 0);
680 }
681 
TEST(ProcMemInfo,ForEachVmaFromFile_MapsTest)682 TEST(ProcMemInfo, ForEachVmaFromFile_MapsTest) {
683     // Parse maps file correctly to make callbacks for each virtual memory area (vma)
684     std::string exec_dir = ::android::base::GetExecutableDirectory();
685     std::string path = ::android::base::StringPrintf("%s/testdata1/maps_short", exec_dir.c_str());
686     ProcMemInfo proc_mem(pid);
687 
688     std::vector<Vma> vmas;
689     auto collect_vmas = [&](const Vma& v) {
690         vmas.push_back(v);
691         return true;
692     };
693     ASSERT_TRUE(ForEachVmaFromFile(path, collect_vmas, false));
694 
695     // We should get a total of 6 vmas
696     ASSERT_EQ(vmas.size(), 6);
697 
698     // Expect values to be equal to what we have in testdata1/maps_short
699     // Check for names
700     EXPECT_EQ(vmas[0].name, "[anon:dalvik-zygote-jit-code-cache]");
701     EXPECT_EQ(vmas[1].name, "/system/framework/x86_64/boot-framework.art");
702     EXPECT_TRUE(vmas[2].name == "[anon:libc_malloc]" ||
703                 vmas[2].name.starts_with("[anon:scudo:"))
704             << "Unknown map name " << vmas[2].name;
705     EXPECT_EQ(vmas[3].name, "/system/priv-app/SettingsProvider/oat/x86_64/SettingsProvider.odex");
706     EXPECT_EQ(vmas[4].name, "/system/lib64/libhwui.so");
707     EXPECT_EQ(vmas[5].name, "[vsyscall]");
708 
709     // Check start address
710     EXPECT_EQ(vmas[0].start, 0x54c00000);
711     EXPECT_EQ(vmas[1].start, 0x701ea000);
712     EXPECT_EQ(vmas[2].start, 0x70074dd8d000);
713     EXPECT_EQ(vmas[3].start, 0x700755a2d000);
714     EXPECT_EQ(vmas[4].start, 0x7007f85b0000);
715     EXPECT_EQ(vmas[5].start, 0xffffffffff600000);
716 
717     // Check end address
718     EXPECT_EQ(vmas[0].end, 0x56c00000);
719     EXPECT_EQ(vmas[1].end, 0x70cdb000);
720     EXPECT_EQ(vmas[2].end, 0x70074ee0d000);
721     EXPECT_EQ(vmas[3].end, 0x700755a6e000);
722     EXPECT_EQ(vmas[4].end, 0x7007f8b9b000);
723     EXPECT_EQ(vmas[5].end, 0xffffffffff601000);
724 
725     // Check Flags
726     EXPECT_EQ(vmas[0].flags, PROT_READ | PROT_EXEC);
727     EXPECT_EQ(vmas[1].flags, PROT_READ | PROT_WRITE);
728     EXPECT_EQ(vmas[2].flags, PROT_READ | PROT_WRITE);
729     EXPECT_EQ(vmas[3].flags, PROT_READ | PROT_EXEC);
730     EXPECT_EQ(vmas[4].flags, PROT_READ | PROT_EXEC);
731     EXPECT_EQ(vmas[5].flags, PROT_READ | PROT_EXEC);
732 
733     // Check Shared
734     EXPECT_FALSE(vmas[0].is_shared);
735     EXPECT_FALSE(vmas[1].is_shared);
736     EXPECT_FALSE(vmas[2].is_shared);
737     EXPECT_FALSE(vmas[3].is_shared);
738     EXPECT_FALSE(vmas[4].is_shared);
739     EXPECT_FALSE(vmas[5].is_shared);
740 
741     // Check Offset
742     EXPECT_EQ(vmas[0].offset, 0x0);
743     EXPECT_EQ(vmas[1].offset, 0x0);
744     EXPECT_EQ(vmas[2].offset, 0x0);
745     EXPECT_EQ(vmas[3].offset, 0x00016000);
746     EXPECT_EQ(vmas[4].offset, 0x001ee000);
747     EXPECT_EQ(vmas[5].offset, 0x0);
748 
749     // Check Inode
750     EXPECT_EQ(vmas[0].inode, 0);
751     EXPECT_EQ(vmas[1].inode, 3165);
752     EXPECT_EQ(vmas[2].inode, 0);
753     EXPECT_EQ(vmas[3].inode, 1947);
754     EXPECT_EQ(vmas[4].inode, 1537);
755     EXPECT_EQ(vmas[5].inode, 0);
756 }
757 
TEST(ProcMemInfo,SmapsReturnTest)758 TEST(ProcMemInfo, SmapsReturnTest) {
759     // Make sure Smaps() is never empty for any process
760     ProcMemInfo proc_mem(pid);
761     auto vmas = proc_mem.Smaps();
762     EXPECT_FALSE(vmas.empty());
763 }
764 
TEST(ProcMemInfo,SmapsTest)765 TEST(ProcMemInfo, SmapsTest) {
766     std::string exec_dir = ::android::base::GetExecutableDirectory();
767     std::string path = ::android::base::StringPrintf("%s/testdata1/smaps_short", exec_dir.c_str());
768     ProcMemInfo proc_mem(pid);
769     auto vmas = proc_mem.Smaps(path);
770 
771     ASSERT_FALSE(vmas.empty());
772 #ifndef __x86_64__
773     // We should get a total of 6 vmas
774     ASSERT_EQ(vmas.size(), 6);
775 #else
776     // We should get a total of 5 vmas ([vsyscall] is excluded)
777     ASSERT_EQ(vmas.size(), 5);
778 #endif
779 
780     // Expect values to be equal to what we have in testdata1/smaps_short
781     // Check for sizes first
782     ASSERT_EQ(vmas[0].usage.vss, 32768);
783     EXPECT_EQ(vmas[1].usage.vss, 11204);
784     EXPECT_EQ(vmas[2].usage.vss, 16896);
785     EXPECT_EQ(vmas[3].usage.vss, 260);
786     EXPECT_EQ(vmas[4].usage.vss, 6060);
787 #ifndef __x86_64__
788     EXPECT_EQ(vmas[5].usage.vss, 4);
789 #endif
790 
791     // Check for names
792     EXPECT_EQ(vmas[0].name, "[anon:dalvik-zygote-jit-code-cache]");
793     EXPECT_EQ(vmas[1].name, "/system/framework/x86_64/boot-framework.art");
794     EXPECT_TRUE(vmas[2].name == "[anon:libc_malloc]" ||
795                 vmas[2].name.starts_with("[anon:scudo:"))
796             << "Unknown map name " << vmas[2].name;
797     EXPECT_EQ(vmas[3].name, "/system/priv-app/SettingsProvider/oat/x86_64/SettingsProvider.odex");
798     EXPECT_EQ(vmas[4].name, "/system/lib64/libhwui.so");
799 #ifndef __x86_64__
800     EXPECT_EQ(vmas[5].name, "[vsyscall]");
801 #endif
802 
803     EXPECT_EQ(vmas[0].usage.rss, 2048);
804     EXPECT_EQ(vmas[1].usage.rss, 11188);
805     EXPECT_EQ(vmas[2].usage.rss, 15272);
806     EXPECT_EQ(vmas[3].usage.rss, 260);
807     EXPECT_EQ(vmas[4].usage.rss, 4132);
808 #ifndef __x86_64__
809     EXPECT_EQ(vmas[5].usage.rss, 0);
810 #endif
811 
812     EXPECT_EQ(vmas[0].usage.pss, 113);
813     EXPECT_EQ(vmas[1].usage.pss, 2200);
814     EXPECT_EQ(vmas[2].usage.pss, 15272);
815     EXPECT_EQ(vmas[3].usage.pss, 260);
816     EXPECT_EQ(vmas[4].usage.pss, 1274);
817 #ifndef __x86_64__
818     EXPECT_EQ(vmas[5].usage.pss, 0);
819 #endif
820 
821     EXPECT_EQ(vmas[0].usage.uss, 0);
822     EXPECT_EQ(vmas[1].usage.uss, 1660);
823     EXPECT_EQ(vmas[2].usage.uss, 15272);
824     EXPECT_EQ(vmas[3].usage.uss, 260);
825     EXPECT_EQ(vmas[4].usage.uss, 0);
826 #ifndef __x86_64__
827     EXPECT_EQ(vmas[5].usage.uss, 0);
828 #endif
829 
830     EXPECT_EQ(vmas[0].usage.private_clean, 0);
831     EXPECT_EQ(vmas[1].usage.private_clean, 0);
832     EXPECT_EQ(vmas[2].usage.private_clean, 0);
833     EXPECT_EQ(vmas[3].usage.private_clean, 260);
834     EXPECT_EQ(vmas[4].usage.private_clean, 0);
835 #ifndef __x86_64__
836     EXPECT_EQ(vmas[5].usage.private_clean, 0);
837 #endif
838 
839     EXPECT_EQ(vmas[0].usage.private_dirty, 0);
840     EXPECT_EQ(vmas[1].usage.private_dirty, 1660);
841     EXPECT_EQ(vmas[2].usage.private_dirty, 15272);
842     EXPECT_EQ(vmas[3].usage.private_dirty, 0);
843     EXPECT_EQ(vmas[4].usage.private_dirty, 0);
844 #ifndef __x86_64__
845     EXPECT_EQ(vmas[5].usage.private_dirty, 0);
846 #endif
847 
848     EXPECT_EQ(vmas[0].usage.shared_clean, 0);
849     EXPECT_EQ(vmas[1].usage.shared_clean, 80);
850     EXPECT_EQ(vmas[2].usage.shared_clean, 0);
851     EXPECT_EQ(vmas[3].usage.shared_clean, 0);
852     EXPECT_EQ(vmas[4].usage.shared_clean, 4132);
853 #ifndef __x86_64__
854     EXPECT_EQ(vmas[5].usage.shared_clean, 0);
855 #endif
856 
857     EXPECT_EQ(vmas[0].usage.shared_dirty, 2048);
858     EXPECT_EQ(vmas[1].usage.shared_dirty, 9448);
859     EXPECT_EQ(vmas[2].usage.shared_dirty, 0);
860     EXPECT_EQ(vmas[3].usage.shared_dirty, 0);
861     EXPECT_EQ(vmas[4].usage.shared_dirty, 0);
862 #ifndef __x86_64__
863     EXPECT_EQ(vmas[5].usage.shared_dirty, 0);
864 #endif
865 
866     EXPECT_EQ(vmas[0].usage.swap, 0);
867     EXPECT_EQ(vmas[1].usage.swap, 0);
868     EXPECT_EQ(vmas[2].usage.swap, 0);
869     EXPECT_EQ(vmas[3].usage.swap, 0);
870     EXPECT_EQ(vmas[4].usage.swap, 0);
871 #ifndef __x86_64__
872     EXPECT_EQ(vmas[5].usage.swap, 0);
873 #endif
874 
875     EXPECT_EQ(vmas[0].usage.swap_pss, 0);
876     EXPECT_EQ(vmas[1].usage.swap_pss, 0);
877     EXPECT_EQ(vmas[2].usage.swap_pss, 0);
878     EXPECT_EQ(vmas[3].usage.swap_pss, 0);
879     EXPECT_EQ(vmas[4].usage.swap_pss, 0);
880 #ifndef __x86_64__
881     EXPECT_EQ(vmas[5].usage.swap_pss, 0);
882 #endif
883 }
884 
TEST(ProcMemInfo,SmapsPopulatesUsageTest)885 TEST(ProcMemInfo, SmapsPopulatesUsageTest) {
886     std::string exec_dir = ::android::base::GetExecutableDirectory();
887     std::string path = ::android::base::StringPrintf("%s/testdata1/smaps_short", exec_dir.c_str());
888     ProcMemInfo proc_mem(pid);
889     auto vmas = proc_mem.Smaps(path, true);
890 
891     // Expect values to be equal to sums of usage in testdata1/smaps_short. For
892     // this data, only vss differs on x86.
893 #ifndef __x86_64__
894     EXPECT_EQ(proc_mem.Usage().vss, 67192);
895 #else
896     EXPECT_EQ(proc_mem.Usage().vss, 67188);
897 #endif
898     EXPECT_EQ(proc_mem.Usage().rss, 32900);
899     EXPECT_EQ(proc_mem.Usage().pss, 19119);
900     EXPECT_EQ(proc_mem.Usage().uss, 17192);
901     EXPECT_EQ(proc_mem.Usage().private_clean, 260);
902     EXPECT_EQ(proc_mem.Usage().private_dirty, 16932);
903     EXPECT_EQ(proc_mem.Usage().shared_clean, 4212);
904     EXPECT_EQ(proc_mem.Usage().shared_dirty, 11496);
905     EXPECT_EQ(proc_mem.Usage().swap, 0);
906     EXPECT_EQ(proc_mem.Usage().swap_pss, 0);
907 }
908 
TEST(SysMemInfo,TestSysMemInfoFile)909 TEST(SysMemInfo, TestSysMemInfoFile) {
910     std::string meminfo = R"meminfo(MemTotal:        3019740 kB
911 MemFree:         1809728 kB
912 MemAvailable:    2546560 kB
913 Buffers:           54736 kB
914 Cached:           776052 kB
915 SwapCached:            0 kB
916 Active:           445856 kB
917 Inactive:         459092 kB
918 Active(anon):      78492 kB
919 Inactive(anon):     2240 kB
920 Active(file):     367364 kB
921 Inactive(file):   456852 kB
922 Unevictable:        3096 kB
923 Mlocked:            3096 kB
924 SwapTotal:         32768 kB
925 SwapFree:           4096 kB
926 Dirty:                32 kB
927 Writeback:             0 kB
928 AnonPages:         74988 kB
929 Mapped:            62624 kB
930 Shmem:              4020 kB
931 KReclaimable:      87324 kB
932 Slab:              86464 kB
933 SReclaimable:      44432 kB
934 SUnreclaim:        42032 kB
935 KernelStack:        4880 kB
936 PageTables:         2900 kB
937 NFS_Unstable:          0 kB
938 Bounce:                0 kB
939 WritebackTmp:          0 kB
940 CommitLimit:     1509868 kB
941 Committed_AS:      80296 kB
942 VmallocTotal:   263061440 kB
943 VmallocUsed:       65536 kB
944 VmallocChunk:          0 kB
945 AnonHugePages:      6144 kB
946 ShmemHugePages:        0 kB
947 ShmemPmdMapped:        0 kB
948 CmaTotal:         131072 kB
949 CmaFree:          130380 kB
950 HugePages_Total:       0
951 HugePages_Free:        0
952 HugePages_Rsvd:        0
953 HugePages_Surp:        0
954 Hugepagesize:       2048 kB)meminfo";
955 
956     TemporaryFile tf;
957     ASSERT_TRUE(tf.fd != -1);
958     ASSERT_TRUE(::android::base::WriteStringToFd(meminfo, tf.fd));
959 
960     SysMemInfo mi;
961     ASSERT_TRUE(mi.ReadMemInfo(tf.path));
962     EXPECT_EQ(mi.mem_total_kb(), 3019740);
963     EXPECT_EQ(mi.mem_free_kb(), 1809728);
964     EXPECT_EQ(mi.mem_buffers_kb(), 54736);
965     EXPECT_EQ(mi.mem_cached_kb(), 776052);
966     EXPECT_EQ(mi.mem_shmem_kb(), 4020);
967     EXPECT_EQ(mi.mem_slab_kb(), 86464);
968     EXPECT_EQ(mi.mem_slab_reclaimable_kb(), 44432);
969     EXPECT_EQ(mi.mem_slab_unreclaimable_kb(), 42032);
970     EXPECT_EQ(mi.mem_swap_kb(), 32768);
971     EXPECT_EQ(mi.mem_swap_free_kb(), 4096);
972     EXPECT_EQ(mi.mem_mapped_kb(), 62624);
973     EXPECT_EQ(mi.mem_vmalloc_used_kb(), 65536);
974     EXPECT_EQ(mi.mem_page_tables_kb(), 2900);
975     EXPECT_EQ(mi.mem_kernel_stack_kb(), 4880);
976     EXPECT_EQ(mi.mem_kreclaimable_kb(), 87324);
977     EXPECT_EQ(mi.mem_active_kb(), 445856);
978     EXPECT_EQ(mi.mem_inactive_kb(), 459092);
979     EXPECT_EQ(mi.mem_unevictable_kb(), 3096);
980     EXPECT_EQ(mi.mem_available_kb(), 2546560);
981     EXPECT_EQ(mi.mem_active_anon_kb(), 78492);
982     EXPECT_EQ(mi.mem_inactive_anon_kb(), 2240);
983     EXPECT_EQ(mi.mem_active_file_kb(), 367364);
984     EXPECT_EQ(mi.mem_inactive_file_kb(), 456852);
985     EXPECT_EQ(mi.mem_cma_total_kb(), 131072);
986     EXPECT_EQ(mi.mem_cma_free_kb(), 130380);
987 }
988 
TEST(SysMemInfo,TestEmptyFile)989 TEST(SysMemInfo, TestEmptyFile) {
990     TemporaryFile tf;
991     std::string empty_string = "";
992     ASSERT_TRUE(tf.fd != -1);
993     ASSERT_TRUE(::android::base::WriteStringToFd(empty_string, tf.fd));
994 
995     SysMemInfo mi;
996     EXPECT_TRUE(mi.ReadMemInfo(tf.path));
997     EXPECT_EQ(mi.mem_total_kb(), 0);
998 }
999 
TEST(SysMemInfo,TestZramTotal)1000 TEST(SysMemInfo, TestZramTotal) {
1001     std::string exec_dir = ::android::base::GetExecutableDirectory();
1002 
1003     SysMemInfo mi;
1004     std::string zram_mmstat_dir = exec_dir + "/testdata1/";
1005     EXPECT_EQ(mi.mem_zram_kb(zram_mmstat_dir.c_str()), 30504);
1006 
1007     std::string zram_memused_dir = exec_dir + "/testdata2/";
1008     EXPECT_EQ(mi.mem_zram_kb(zram_memused_dir.c_str()), 30504);
1009 }
1010 
1011 enum {
1012     MEMINFO_TOTAL,
1013     MEMINFO_FREE,
1014     MEMINFO_BUFFERS,
1015     MEMINFO_CACHED,
1016     MEMINFO_SHMEM,
1017     MEMINFO_SLAB,
1018     MEMINFO_SLAB_RECLAIMABLE,
1019     MEMINFO_SLAB_UNRECLAIMABLE,
1020     MEMINFO_SWAP_TOTAL,
1021     MEMINFO_SWAP_FREE,
1022     MEMINFO_ZRAM_TOTAL,
1023     MEMINFO_MAPPED,
1024     MEMINFO_VMALLOC_USED,
1025     MEMINFO_PAGE_TABLES,
1026     MEMINFO_KERNEL_STACK,
1027     MEMINFO_KRECLAIMABLE,
1028     MEMINFO_ACTIVE,
1029     MEMINFO_INACTIVE,
1030     MEMINFO_UNEVICTABLE,
1031     MEMINFO_AVAILABLE,
1032     MEMINFO_ACTIVE_ANON,
1033     MEMINFO_INACTIVE_ANON,
1034     MEMINFO_ACTIVE_FILE,
1035     MEMINFO_INACTIVE_FILE,
1036     MEMINFO_CMA_TOTAL,
1037     MEMINFO_CMA_FREE,
1038     MEMINFO_COUNT
1039 };
1040 
TEST(SysMemInfo,TestZramWithTags)1041 TEST(SysMemInfo, TestZramWithTags) {
1042     std::string meminfo = R"meminfo(MemTotal:        3019740 kB
1043 MemFree:         1809728 kB
1044 MemAvailable:    2546560 kB
1045 Buffers:           54736 kB
1046 Cached:           776052 kB
1047 SwapCached:            0 kB
1048 Active:           445856 kB
1049 Inactive:         459092 kB
1050 Active(anon):      78492 kB
1051 Inactive(anon):     2240 kB
1052 Active(file):     367364 kB
1053 Inactive(file):   456852 kB
1054 Unevictable:        3096 kB
1055 Mlocked:            3096 kB
1056 SwapTotal:         32768 kB
1057 SwapFree:           4096 kB
1058 Dirty:                32 kB
1059 Writeback:             0 kB
1060 AnonPages:         74988 kB
1061 Mapped:            62624 kB
1062 Shmem:              4020 kB
1063 KReclaimable:      87324 kB
1064 Slab:              86464 kB
1065 SReclaimable:      44432 kB
1066 SUnreclaim:        42032 kB
1067 KernelStack:        4880 kB
1068 PageTables:         2900 kB
1069 NFS_Unstable:          0 kB
1070 Bounce:                0 kB
1071 WritebackTmp:          0 kB
1072 CommitLimit:     1509868 kB
1073 Committed_AS:      80296 kB
1074 VmallocTotal:   263061440 kB
1075 VmallocUsed:       65536 kB
1076 VmallocChunk:          0 kB
1077 AnonHugePages:      6144 kB
1078 ShmemHugePages:        0 kB
1079 ShmemPmdMapped:        0 kB
1080 CmaTotal:         131072 kB
1081 CmaFree:          130380 kB
1082 HugePages_Total:       0
1083 HugePages_Free:        0
1084 HugePages_Rsvd:        0
1085 HugePages_Surp:        0
1086 Hugepagesize:       2048 kB)meminfo";
1087 
1088     TemporaryFile tf;
1089     ASSERT_TRUE(tf.fd != -1);
1090     ASSERT_TRUE(::android::base::WriteStringToFd(meminfo, tf.fd));
1091     std::string file = std::string(tf.path);
1092     std::vector<uint64_t> mem;
1093     std::vector<std::string_view> tags(SysMemInfo::kDefaultSysMemInfoTags.begin(),
1094                                        SysMemInfo::kDefaultSysMemInfoTags.end());
1095     auto it = tags.begin();
1096     tags.insert(it + MEMINFO_ZRAM_TOTAL, "Zram:");
1097     SysMemInfo mi;
1098 
1099     // Read system memory info
1100     mem.resize(tags.size());
1101     EXPECT_TRUE(mi.ReadMemInfo(tags.size(), tags.data(), mem.data(), file.c_str()));
1102     EXPECT_EQ(mem[MEMINFO_TOTAL], 3019740);
1103     EXPECT_EQ(mem[MEMINFO_FREE], 1809728);
1104     EXPECT_EQ(mem[MEMINFO_BUFFERS], 54736);
1105     EXPECT_EQ(mem[MEMINFO_CACHED], 776052);
1106     EXPECT_EQ(mem[MEMINFO_SHMEM], 4020);
1107     EXPECT_EQ(mem[MEMINFO_SLAB], 86464);
1108     EXPECT_EQ(mem[MEMINFO_SLAB_RECLAIMABLE], 44432);
1109     EXPECT_EQ(mem[MEMINFO_SLAB_UNRECLAIMABLE], 42032);
1110     EXPECT_EQ(mem[MEMINFO_SWAP_TOTAL], 32768);
1111     EXPECT_EQ(mem[MEMINFO_SWAP_FREE], 4096);
1112     EXPECT_EQ(mem[MEMINFO_MAPPED], 62624);
1113     EXPECT_EQ(mem[MEMINFO_VMALLOC_USED], 65536);
1114     EXPECT_EQ(mem[MEMINFO_PAGE_TABLES], 2900);
1115     EXPECT_EQ(mem[MEMINFO_KERNEL_STACK], 4880);
1116     EXPECT_EQ(mem[MEMINFO_KRECLAIMABLE], 87324);
1117     EXPECT_EQ(mem[MEMINFO_ACTIVE], 445856);
1118     EXPECT_EQ(mem[MEMINFO_INACTIVE], 459092);
1119     EXPECT_EQ(mem[MEMINFO_UNEVICTABLE], 3096);
1120     EXPECT_EQ(mem[MEMINFO_AVAILABLE], 2546560);
1121     EXPECT_EQ(mem[MEMINFO_ACTIVE_ANON], 78492);
1122     EXPECT_EQ(mem[MEMINFO_INACTIVE_ANON], 2240);
1123     EXPECT_EQ(mem[MEMINFO_ACTIVE_FILE], 367364);
1124     EXPECT_EQ(mem[MEMINFO_INACTIVE_FILE], 456852);
1125     EXPECT_EQ(mem[MEMINFO_CMA_TOTAL], 131072);
1126     EXPECT_EQ(mem[MEMINFO_CMA_FREE], 130380);
1127 }
1128 
TEST(SysMemInfo,TestVmallocInfoNoMemory)1129 TEST(SysMemInfo, TestVmallocInfoNoMemory) {
1130     std::string vmallocinfo =
1131             R"vmallocinfo(0x0000000000000000-0x0000000000000000   69632 of_iomap+0x78/0xb0 phys=17a00000 ioremap
1132 0x0000000000000000-0x0000000000000000    8192 of_iomap+0x78/0xb0 phys=b220000 ioremap
1133 0x0000000000000000-0x0000000000000000    8192 of_iomap+0x78/0xb0 phys=17c90000 ioremap
1134 0x0000000000000000-0x0000000000000000    8192 of_iomap+0x78/0xb0 phys=17ca0000 ioremap)vmallocinfo";
1135 
1136     TemporaryFile tf;
1137     ASSERT_TRUE(tf.fd != -1);
1138     ASSERT_TRUE(::android::base::WriteStringToFd(vmallocinfo, tf.fd));
1139     std::string file = std::string(tf.path);
1140 
1141     EXPECT_EQ(ReadVmallocInfo(file.c_str()), 0);
1142 }
1143 
TEST(SysMemInfo,TestVmallocInfoKernel)1144 TEST(SysMemInfo, TestVmallocInfoKernel) {
1145     std::string vmallocinfo =
1146             R"vmallocinfo(0x0000000000000000-0x0000000000000000    8192 drm_property_create_blob+0x44/0xec pages=1 vmalloc)vmallocinfo";
1147 
1148     TemporaryFile tf;
1149     ASSERT_TRUE(tf.fd != -1);
1150     ASSERT_TRUE(::android::base::WriteStringToFd(vmallocinfo, tf.fd));
1151     std::string file = std::string(tf.path);
1152 
1153     EXPECT_EQ(ReadVmallocInfo(file.c_str()), getpagesize());
1154 }
1155 
TEST(SysMemInfo,TestVmallocInfoModule)1156 TEST(SysMemInfo, TestVmallocInfoModule) {
1157     std::string vmallocinfo =
1158             R"vmallocinfo(0x0000000000000000-0x0000000000000000   28672 pktlog_alloc_buf+0xc4/0x15c [wlan] pages=6 vmalloc)vmallocinfo";
1159 
1160     TemporaryFile tf;
1161     ASSERT_TRUE(tf.fd != -1);
1162     ASSERT_TRUE(::android::base::WriteStringToFd(vmallocinfo, tf.fd));
1163     std::string file = std::string(tf.path);
1164 
1165     EXPECT_EQ(ReadVmallocInfo(file.c_str()), 6 * getpagesize());
1166 }
1167 
TEST(SysMemInfo,TestVmallocInfoAll)1168 TEST(SysMemInfo, TestVmallocInfoAll) {
1169     std::string vmallocinfo =
1170             R"vmallocinfo(0x0000000000000000-0x0000000000000000   69632 of_iomap+0x78/0xb0 phys=17a00000 ioremap
1171 0x0000000000000000-0x0000000000000000    8192 of_iomap+0x78/0xb0 phys=b220000 ioremap
1172 0x0000000000000000-0x0000000000000000    8192 of_iomap+0x78/0xb0 phys=17c90000 ioremap
1173 0x0000000000000000-0x0000000000000000    8192 of_iomap+0x78/0xb0 phys=17ca0000 ioremap
1174 0x0000000000000000-0x0000000000000000    8192 drm_property_create_blob+0x44/0xec pages=1 vmalloc
1175 0x0000000000000000-0x0000000000000000   28672 pktlog_alloc_buf+0xc4/0x15c [wlan] pages=6 vmalloc)vmallocinfo";
1176 
1177     TemporaryFile tf;
1178     ASSERT_TRUE(tf.fd != -1);
1179     ASSERT_TRUE(::android::base::WriteStringToFd(vmallocinfo, tf.fd));
1180     std::string file = std::string(tf.path);
1181 
1182     EXPECT_EQ(ReadVmallocInfo(file.c_str()), 7 * getpagesize());
1183 }
1184 
TEST(SysMemInfo,TestReadIonHeapsSizeKb)1185 TEST(SysMemInfo, TestReadIonHeapsSizeKb) {
1186     std::string total_heaps_kb = R"total_heaps_kb(98480)total_heaps_kb";
1187     uint64_t size;
1188 
1189     TemporaryFile tf;
1190     ASSERT_TRUE(tf.fd != -1);
1191     ASSERT_TRUE(::android::base::WriteStringToFd(total_heaps_kb, tf.fd));
1192     std::string file = std::string(tf.path);
1193 
1194     ASSERT_TRUE(ReadIonHeapsSizeKb(&size, file));
1195     EXPECT_EQ(size, 98480);
1196 }
1197 
TEST(SysMemInfo,TestReadIonPoolsSizeKb)1198 TEST(SysMemInfo, TestReadIonPoolsSizeKb) {
1199     std::string total_pools_kb = R"total_pools_kb(416)total_pools_kb";
1200     uint64_t size;
1201 
1202     TemporaryFile tf;
1203     ASSERT_TRUE(tf.fd != -1);
1204     ASSERT_TRUE(::android::base::WriteStringToFd(total_pools_kb, tf.fd));
1205     std::string file = std::string(tf.path);
1206 
1207     ASSERT_TRUE(ReadIonPoolsSizeKb(&size, file));
1208     EXPECT_EQ(size, 416);
1209 }
1210 
TEST(SysMemInfo,TestReadGpuTotalUsageKb)1211 TEST(SysMemInfo, TestReadGpuTotalUsageKb) {
1212     uint64_t size;
1213 
1214     if (android::base::GetIntProperty("ro.vendor.api_level", 0) < __ANDROID_API_S__) {
1215         GTEST_SKIP();
1216     }
1217 
1218     KernelVersion min_kernel_version = KernelVersion(5, 4, 0);
1219     KernelVersion kernel_version = VintfObject::GetInstance()
1220                                            ->getRuntimeInfo(RuntimeInfo::FetchFlag::CPU_VERSION)
1221                                            ->kernelVersion();
1222     if (kernel_version < min_kernel_version) {
1223         GTEST_SKIP();
1224     }
1225 
1226     ASSERT_TRUE(ReadGpuTotalUsageKb(&size));
1227     EXPECT_TRUE(size >= 0);
1228 }
1229 
TEST(AndroidProcHeaps,ExtractAndroidHeapStatsFromFileTest)1230 TEST(AndroidProcHeaps, ExtractAndroidHeapStatsFromFileTest) {
1231     std::string smaps =
1232             R"smaps(12c00000-13440000 rw-p 00000000 00:00 0  [anon:dalvik-main space (region space)]
1233 Name:           [anon:dalvik-main space (region space)]
1234 Size:               8448 kB
1235 KernelPageSize:        4 kB
1236 MMUPageSize:           4 kB
1237 Rss:                2652 kB
1238 Pss:                2652 kB
1239 Shared_Clean:        840 kB
1240 Shared_Dirty:         40 kB
1241 Private_Clean:        84 kB
1242 Private_Dirty:      2652 kB
1243 Referenced:         2652 kB
1244 Anonymous:          2652 kB
1245 AnonHugePages:         0 kB
1246 ShmemPmdMapped:        0 kB
1247 Shared_Hugetlb:        0 kB
1248 Private_Hugetlb:       0 kB
1249 Swap:                102 kB
1250 SwapPss:              70 kB
1251 Locked:             2652 kB
1252 VmFlags: rd wr mr mw me ac
1253 )smaps";
1254 
1255     TemporaryFile tf;
1256     ASSERT_TRUE(tf.fd != -1);
1257     ASSERT_TRUE(::android::base::WriteStringToFd(smaps, tf.fd));
1258 
1259     bool foundSwapPss;
1260     AndroidHeapStats stats[_NUM_HEAP];
1261     ASSERT_TRUE(ExtractAndroidHeapStatsFromFile(tf.path, stats, &foundSwapPss));
1262 
1263     AndroidHeapStats actualStats;
1264     for (int i = 0; i < _NUM_CORE_HEAP; i++) {
1265         actualStats.pss += stats[i].pss;
1266         actualStats.swappablePss += stats[i].swappablePss;
1267         actualStats.rss += stats[i].rss;
1268         actualStats.privateDirty += stats[i].privateDirty;
1269         actualStats.sharedDirty += stats[i].sharedDirty;
1270         actualStats.privateClean += stats[i].privateClean;
1271         actualStats.sharedClean += stats[i].sharedClean;
1272         actualStats.swappedOut += stats[i].swappedOut;
1273         actualStats.swappedOutPss += stats[i].swappedOutPss;
1274     }
1275     EXPECT_EQ(actualStats.pss, 2652);
1276     EXPECT_EQ(actualStats.swappablePss, 0);
1277     EXPECT_EQ(actualStats.rss, 2652);
1278     EXPECT_EQ(actualStats.privateDirty, 2652);
1279     EXPECT_EQ(actualStats.sharedDirty, 40);
1280     EXPECT_EQ(actualStats.privateClean, 84);
1281     EXPECT_EQ(actualStats.sharedClean, 840);
1282     EXPECT_EQ(actualStats.swappedOut, 102);
1283     EXPECT_EQ(actualStats.swappedOutPss, 70);
1284 }
1285 
1286 class DmabufHeapStats : public ::testing::Test {
1287   public:
SetUp()1288     virtual void SetUp() {
1289         fs::current_path(fs::temp_directory_path());
1290         buffer_stats_path = fs::current_path() / "buffers";
1291         ASSERT_TRUE(fs::create_directory(buffer_stats_path));
1292         heap_root_path = fs::current_path() / "dma_heap";
1293         ASSERT_TRUE(fs::create_directory(heap_root_path));
1294     }
TearDown()1295     virtual void TearDown() {
1296         fs::remove_all(buffer_stats_path);
1297         fs::remove_all(heap_root_path);
1298     }
1299 
1300     fs::path buffer_stats_path;
1301     fs::path heap_root_path;
1302 };
1303 
TEST_F(DmabufHeapStats,TestDmabufHeapTotalExportedKb)1304 TEST_F(DmabufHeapStats, TestDmabufHeapTotalExportedKb) {
1305     using android::base::StringPrintf;
1306     uint64_t size;
1307 
1308     auto system_heap_path = heap_root_path / "system";
1309     ASSERT_TRUE(android::base::WriteStringToFile("test", system_heap_path));
1310 
1311     for (unsigned int inode_number = 74831; inode_number < 74841; inode_number++) {
1312         auto buffer_path = buffer_stats_path / StringPrintf("%u", inode_number);
1313         ASSERT_TRUE(fs::create_directories(buffer_path));
1314 
1315         auto buffer_size_path = buffer_path / "size";
1316         const std::string buffer_size = "4096";
1317         ASSERT_TRUE(android::base::WriteStringToFile(buffer_size, buffer_size_path));
1318 
1319         auto exp_name_path = buffer_path / "exporter_name";
1320         const std::string exp_name = inode_number % 2 ? "system" : "other";
1321         ASSERT_TRUE(android::base::WriteStringToFile(exp_name, exp_name_path));
1322     }
1323 
1324     ASSERT_TRUE(ReadDmabufHeapTotalExportedKb(&size, heap_root_path, buffer_stats_path));
1325     ASSERT_EQ(size, 20);
1326 }
1327 
TEST(SysMemInfo,TestReadDmaBufHeapPoolsSizeKb)1328 TEST(SysMemInfo, TestReadDmaBufHeapPoolsSizeKb) {
1329     std::string total_pools_kb = R"total_pools_kb(416)total_pools_kb";
1330     uint64_t size;
1331 
1332     TemporaryFile tf;
1333     ASSERT_TRUE(tf.fd != -1);
1334     ASSERT_TRUE(::android::base::WriteStringToFd(total_pools_kb, tf.fd));
1335     std::string file = std::string(tf.path);
1336 
1337     ASSERT_TRUE(ReadDmabufHeapPoolsSizeKb(&size, file));
1338     EXPECT_EQ(size, 416);
1339 }
1340 
main(int argc,char ** argv)1341 int main(int argc, char** argv) {
1342     ::testing::InitGoogleTest(&argc, argv);
1343     ::android::base::InitLogging(argv, android::base::StderrLogger);
1344     pid = getpid();
1345     return RUN_ALL_TESTS();
1346 }
1347