77 lines · cpp
1//===-- Unittests for shm_open/shm_unlink ---------------------------------===//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 "hdr/fcntl_macros.h"10#include "src/fcntl/fcntl.h"11#include "src/sys/mman/mmap.h"12#include "src/sys/mman/munmap.h"13#include "src/sys/mman/shm_open.h"14#include "src/sys/mman/shm_unlink.h"15#include "src/unistd/close.h"16#include "src/unistd/ftruncate.h"17#include "test/UnitTest/ErrnoCheckingTest.h"18#include "test/UnitTest/ErrnoSetterMatcher.h"19#include "test/UnitTest/Test.h"20 21using namespace LIBC_NAMESPACE::testing::ErrnoSetterMatcher;22using LlvmLibcShmTest = LIBC_NAMESPACE::testing::ErrnoCheckingTest;23// since shm_open/shm_unlink are wrappers around open/unlink, we only focus on24// testing basic cases and name conversions.25 26TEST_F(LlvmLibcShmTest, Basic) {27 const char *name = "/test_shm_open";28 int fd;29 ASSERT_THAT(fd = LIBC_NAMESPACE::shm_open(name, O_CREAT | O_RDWR, 0666),30 returns(GE(0)).with_errno(EQ(0)));31 32 // check that FD_CLOEXEC is set by default.33 long flag = LIBC_NAMESPACE::fcntl(fd, F_GETFD);34 ASSERT_GE(static_cast<int>(flag), 0);35 EXPECT_NE(static_cast<int>(flag) & FD_CLOEXEC, 0);36 37 // allocate space using ftruncate38 ASSERT_THAT(LIBC_NAMESPACE::ftruncate(fd, 4096), Succeeds());39 // map the shared memory40 void *addr = LIBC_NAMESPACE::mmap(nullptr, 4096, PROT_READ | PROT_WRITE,41 MAP_SHARED, fd, 0);42 ASSERT_NE(addr, MAP_FAILED);43 // just write random data to the shared memory44 char data[] = "Despite its name, LLVM has little to do with traditional "45 "virtual machines.";46 for (size_t i = 0; i < sizeof(data); ++i)47 static_cast<char *>(addr)[i] = data[i];48 49 // close fd does not affect the mapping50 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds());51 for (size_t i = 0; i < sizeof(data); ++i)52 EXPECT_EQ(static_cast<char *>(addr)[i], data[i]);53 54 // unmap the shared memory55 ASSERT_THAT(LIBC_NAMESPACE::munmap(addr, 4096), Succeeds());56 // remove the shared memory57 ASSERT_THAT(LIBC_NAMESPACE::shm_unlink(name), Succeeds());58}59 60TEST_F(LlvmLibcShmTest, NameConversion) {61 const char *name = "////test_shm_open";62 int fd;63 ASSERT_THAT(fd = LIBC_NAMESPACE::shm_open(name, O_CREAT | O_RDWR, 0666),64 returns(GE(0)).with_errno(EQ(0)));65 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds());66 ASSERT_THAT(LIBC_NAMESPACE::shm_unlink(name), Succeeds());67 68 ASSERT_THAT(LIBC_NAMESPACE::shm_open("/123/123", O_CREAT | O_RDWR, 0666),69 Fails(EINVAL));70 71 ASSERT_THAT(LIBC_NAMESPACE::shm_open("/.", O_CREAT | O_RDWR, 0666),72 Fails(EINVAL));73 74 ASSERT_THAT(LIBC_NAMESPACE::shm_open("/..", O_CREAT | O_RDWR, 0666),75 Fails(EINVAL));76}77