c0a3c5c81f
It appears that qemu does not actually cause mmap to fail when calling setrlimit to limit the address space size. In the two tests that use setrlimit, detect if mmap still works and skip the tests in that case. Since all Android targets should support setrlimit, compile out the mmap check code for them.
88 lines
2.4 KiB
C++
88 lines
2.4 KiB
C++
//===-- vector_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 "tests/scudo_unit_test.h"
|
|
|
|
#include "vector.h"
|
|
|
|
TEST(ScudoVectorTest, Basic) {
|
|
scudo::Vector<int> V;
|
|
EXPECT_EQ(V.size(), 0U);
|
|
V.push_back(42);
|
|
EXPECT_EQ(V.size(), 1U);
|
|
EXPECT_EQ(V[0], 42);
|
|
V.push_back(43);
|
|
EXPECT_EQ(V.size(), 2U);
|
|
EXPECT_EQ(V[0], 42);
|
|
EXPECT_EQ(V[1], 43);
|
|
}
|
|
|
|
TEST(ScudoVectorTest, Stride) {
|
|
scudo::Vector<scudo::uptr> V;
|
|
for (scudo::uptr I = 0; I < 1000; I++) {
|
|
V.push_back(I);
|
|
EXPECT_EQ(V.size(), I + 1U);
|
|
EXPECT_EQ(V[I], I);
|
|
}
|
|
for (scudo::uptr I = 0; I < 1000; I++)
|
|
EXPECT_EQ(V[I], I);
|
|
}
|
|
|
|
TEST(ScudoVectorTest, ResizeReduction) {
|
|
scudo::Vector<int> V;
|
|
V.push_back(0);
|
|
V.push_back(0);
|
|
EXPECT_EQ(V.size(), 2U);
|
|
V.resize(1);
|
|
EXPECT_EQ(V.size(), 1U);
|
|
}
|
|
|
|
#if defined(__linux__)
|
|
|
|
#include <sys/resource.h>
|
|
|
|
// Verify that if the reallocate fails, nothing new is added.
|
|
TEST(ScudoVectorTest, ReallocateFails) {
|
|
scudo::Vector<char> V;
|
|
scudo::uptr capacity = V.capacity();
|
|
|
|
// Get the current address space size.
|
|
rlimit Limit = {};
|
|
EXPECT_EQ(0, getrlimit(RLIMIT_AS, &Limit));
|
|
|
|
rlimit EmptyLimit = {.rlim_cur = 0, .rlim_max = Limit.rlim_max};
|
|
EXPECT_EQ(0, setrlimit(RLIMIT_AS, &EmptyLimit));
|
|
|
|
// qemu does not honor the setrlimit, so verify before proceeding.
|
|
scudo::MemMapT MemMap;
|
|
if (MemMap.map(/*Addr=*/0U, scudo::getPageSizeCached(), "scudo:test",
|
|
MAP_ALLOWNOMEM)) {
|
|
MemMap.unmap(MemMap.getBase(), MemMap.getCapacity());
|
|
setrlimit(RLIMIT_AS, &Limit);
|
|
GTEST_SKIP() << "Limiting address space does not prevent mmap.";
|
|
}
|
|
|
|
V.resize(capacity);
|
|
// Set the last element so we can check it later.
|
|
V.back() = '\0';
|
|
|
|
// The reallocate should fail, so the capacity should not change.
|
|
V.reserve(capacity + 1000);
|
|
EXPECT_EQ(capacity, V.capacity());
|
|
|
|
// Now try to do a push back and verify that the size does not change.
|
|
scudo::uptr Size = V.size();
|
|
V.push_back('2');
|
|
EXPECT_EQ(Size, V.size());
|
|
// Verify that the last element in the vector did not change.
|
|
EXPECT_EQ('\0', V.back());
|
|
|
|
EXPECT_EQ(0, setrlimit(RLIMIT_AS, &Limit));
|
|
}
|
|
#endif
|