1*3cab2bb3Spatrick //===-- stats_test.cpp ------------------------------------------*- C++ -*-===//
2*3cab2bb3Spatrick //
3*3cab2bb3Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*3cab2bb3Spatrick // See https://llvm.org/LICENSE.txt for license information.
5*3cab2bb3Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*3cab2bb3Spatrick //
7*3cab2bb3Spatrick //===----------------------------------------------------------------------===//
8*3cab2bb3Spatrick
9*3cab2bb3Spatrick #include "tests/scudo_unit_test.h"
10*3cab2bb3Spatrick
11*3cab2bb3Spatrick #include "stats.h"
12*3cab2bb3Spatrick
TEST(ScudoStatsTest,LocalStats)13*3cab2bb3Spatrick TEST(ScudoStatsTest, LocalStats) {
14*3cab2bb3Spatrick scudo::LocalStats LStats;
15*3cab2bb3Spatrick LStats.init();
16*3cab2bb3Spatrick for (scudo::uptr I = 0; I < scudo::StatCount; I++)
17*3cab2bb3Spatrick EXPECT_EQ(LStats.get(static_cast<scudo::StatType>(I)), 0U);
18*3cab2bb3Spatrick LStats.add(scudo::StatAllocated, 4096U);
19*3cab2bb3Spatrick EXPECT_EQ(LStats.get(scudo::StatAllocated), 4096U);
20*3cab2bb3Spatrick LStats.sub(scudo::StatAllocated, 4096U);
21*3cab2bb3Spatrick EXPECT_EQ(LStats.get(scudo::StatAllocated), 0U);
22*3cab2bb3Spatrick LStats.set(scudo::StatAllocated, 4096U);
23*3cab2bb3Spatrick EXPECT_EQ(LStats.get(scudo::StatAllocated), 4096U);
24*3cab2bb3Spatrick }
25*3cab2bb3Spatrick
TEST(ScudoStatsTest,GlobalStats)26*3cab2bb3Spatrick TEST(ScudoStatsTest, GlobalStats) {
27*3cab2bb3Spatrick scudo::GlobalStats GStats;
28*3cab2bb3Spatrick GStats.init();
29*3cab2bb3Spatrick scudo::uptr Counters[scudo::StatCount] = {};
30*3cab2bb3Spatrick GStats.get(Counters);
31*3cab2bb3Spatrick for (scudo::uptr I = 0; I < scudo::StatCount; I++)
32*3cab2bb3Spatrick EXPECT_EQ(Counters[I], 0U);
33*3cab2bb3Spatrick scudo::LocalStats LStats;
34*3cab2bb3Spatrick LStats.init();
35*3cab2bb3Spatrick GStats.link(&LStats);
36*3cab2bb3Spatrick for (scudo::uptr I = 0; I < scudo::StatCount; I++)
37*3cab2bb3Spatrick LStats.add(static_cast<scudo::StatType>(I), 4096U);
38*3cab2bb3Spatrick GStats.get(Counters);
39*3cab2bb3Spatrick for (scudo::uptr I = 0; I < scudo::StatCount; I++)
40*3cab2bb3Spatrick EXPECT_EQ(Counters[I], 4096U);
41*3cab2bb3Spatrick // Unlinking the local stats move numbers to the global stats.
42*3cab2bb3Spatrick GStats.unlink(&LStats);
43*3cab2bb3Spatrick GStats.get(Counters);
44*3cab2bb3Spatrick for (scudo::uptr I = 0; I < scudo::StatCount; I++)
45*3cab2bb3Spatrick EXPECT_EQ(Counters[I], 4096U);
46*3cab2bb3Spatrick }
47