72 lines · cpp
1//===-- Unittests for msync -----------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "src/sys/mman/mlock.h"10#include "src/sys/mman/mmap.h"11#include "src/sys/mman/msync.h"12#include "src/sys/mman/munlock.h"13#include "src/sys/mman/munmap.h"14#include "src/unistd/sysconf.h"15#include "test/UnitTest/ErrnoCheckingTest.h"16#include "test/UnitTest/ErrnoSetterMatcher.h"17#include "test/UnitTest/Test.h"18 19const size_t PAGE_SIZE = LIBC_NAMESPACE::sysconf(_SC_PAGESIZE);20 21using namespace LIBC_NAMESPACE::testing::ErrnoSetterMatcher;22using LlvmLibcMsyncTest = LIBC_NAMESPACE::testing::ErrnoCheckingTest;23 24struct PageHolder {25 size_t size;26 void *addr;27 28 PageHolder()29 : size(PAGE_SIZE),30 addr(LIBC_NAMESPACE::mmap(nullptr, size, PROT_READ | PROT_WRITE,31 MAP_ANONYMOUS | MAP_PRIVATE, -1, 0)) {}32 ~PageHolder() {33 if (addr != MAP_FAILED)34 LIBC_NAMESPACE::munmap(addr, size);35 }36 37 char &operator[](size_t i) { return reinterpret_cast<char *>(addr)[i]; }38 39 bool is_valid() { return addr != MAP_FAILED; }40};41 42TEST_F(LlvmLibcMsyncTest, UnMappedMemory) {43 EXPECT_THAT(LIBC_NAMESPACE::msync(nullptr, 1024, MS_SYNC), Fails(ENOMEM));44 EXPECT_THAT(LIBC_NAMESPACE::msync(nullptr, 1024, MS_ASYNC), Fails(ENOMEM));45}46 47TEST_F(LlvmLibcMsyncTest, LockedPage) {48 PageHolder page;49 ASSERT_TRUE(page.is_valid());50 ASSERT_THAT(LIBC_NAMESPACE::mlock(page.addr, page.size), Succeeds());51 EXPECT_THAT(52 LIBC_NAMESPACE::msync(page.addr, page.size, MS_SYNC | MS_INVALIDATE),53 Fails(EBUSY));54 ASSERT_THAT(LIBC_NAMESPACE::munlock(page.addr, page.size), Succeeds());55 EXPECT_THAT(LIBC_NAMESPACE::msync(page.addr, page.size, MS_SYNC), Succeeds());56}57 58TEST_F(LlvmLibcMsyncTest, UnalignedAddress) {59 PageHolder page;60 ASSERT_TRUE(page.is_valid());61 EXPECT_THAT(LIBC_NAMESPACE::msync(&page[1], page.size - 1, MS_SYNC),62 Fails(EINVAL));63}64 65TEST_F(LlvmLibcMsyncTest, InvalidFlag) {66 PageHolder page;67 ASSERT_TRUE(page.is_valid());68 EXPECT_THAT(LIBC_NAMESPACE::msync(page.addr, page.size, MS_SYNC | MS_ASYNC),69 Fails(EINVAL));70 EXPECT_THAT(LIBC_NAMESPACE::msync(page.addr, page.size, -1), Fails(EINVAL));71}72