Skip to content

Commit

Permalink
[NFC][scudo] Add releasePagesToOS test
Browse files Browse the repository at this point in the history
  • Loading branch information
vitalybuka committed May 23, 2021
1 parent 6994bf7 commit 0bccdf8
Show file tree
Hide file tree
Showing 2 changed files with 59 additions and 0 deletions.
1 change: 1 addition & 0 deletions compiler-rt/lib/scudo/standalone/tests/CMakeLists.txt
Expand Up @@ -81,6 +81,7 @@ set(SCUDO_UNIT_TEST_SOURCES
checksum_test.cpp
chunk_test.cpp
combined_test.cpp
common_test.cpp
flags_test.cpp
list_test.cpp
map_test.cpp
Expand Down
58 changes: 58 additions & 0 deletions compiler-rt/lib/scudo/standalone/tests/common_test.cpp
@@ -0,0 +1,58 @@
//===-- common_test.cpp ---------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "internal_defs.h"
#include "tests/scudo_unit_test.h"

#include "common.h"
#include <algorithm>
#include <fstream>

namespace scudo {

static uptr getResidentMemorySize() {
uptr Size;
uptr Resident;
std::ifstream IFS("/proc/self/statm");
IFS >> Size;
IFS >> Resident;
return Resident * getPageSizeCached();
}

// Fuchsia needs getResidentMemorySize implementation.
TEST(ScudoCommonTest, SKIP_ON_FUCHSIA(ResidentMemorySize)) {
uptr OnStart = getResidentMemorySize();
EXPECT_GT(OnStart, 0);

const uptr Size = 1ull << 30;
const uptr Threshold = Size >> 3;

MapPlatformData Data;
uptr *P = reinterpret_cast<uptr *>(
map(nullptr, Size, "ResidentMemorySize", 0, &Data));
const uptr N = Size / sizeof(*P);
ASSERT_NE(nullptr, P);
EXPECT_EQ(std::count(P, P + N, 0), N);
EXPECT_LT(getResidentMemorySize() - OnStart, Threshold);

memset(P, 1, Size);
EXPECT_EQ(std::count(P, P + N, 0), 0);
EXPECT_LT(getResidentMemorySize() - Size, Threshold);

releasePagesToOS((uptr)P, 0, Size, &Data);
EXPECT_EQ(std::count(P, P + N, 0), N);
// FIXME: does not work with QEMU-user.
// EXPECT_LT(getResidentMemorySize() - OnStart, Threshold) << OnStart << " "
// << getResidentMemorySize();

memset(P, 1, Size);
EXPECT_EQ(std::count(P, P + N, 0), 0);
EXPECT_LT(getResidentMemorySize() - Size, Threshold);
}

} // namespace scudo

0 comments on commit 0bccdf8

Please sign in to comment.