79 lines · cpp
1//===-- Unittests for pread and pwrite ------------------------------------===//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/fcntl/open.h"10#include "src/unistd/close.h"11#include "src/unistd/fsync.h"12#include "src/unistd/pread.h"13#include "src/unistd/pwrite.h"14#include "src/unistd/unlink.h"15#include "src/unistd/write.h"16#include "test/UnitTest/ErrnoCheckingTest.h"17#include "test/UnitTest/ErrnoSetterMatcher.h"18#include "test/UnitTest/Test.h"19 20#include <sys/stat.h>21 22using LlvmLibcUniStd = LIBC_NAMESPACE::testing::ErrnoCheckingTest;23 24TEST_F(LlvmLibcUniStd, PWriteAndPReadBackTest) {25 // The strategy here is that we first create a file and write to it. Next,26 // we open that file again and write at an offset. Finally, we open the27 // file again and pread at an offset and make sure that only expected data28 // is being read back. This also confirms that pwrite happened successfully.29 constexpr const char HELLO[] = "hello";30 constexpr ssize_t HELLO_SIZE = sizeof(HELLO);31 constexpr off_t OFFSET = 3;32 constexpr const char OFFSET_TEXT[] = "helhello";33 constexpr ssize_t OFFSET_TEXT_SIZE = sizeof(OFFSET_TEXT);34 35 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds;36 37 constexpr const char *FILENAME = "pread_pwrite.test";38 auto TEST_FILE = libc_make_test_file_path(FILENAME);39 int fd = LIBC_NAMESPACE::open(TEST_FILE, O_WRONLY | O_CREAT, S_IRWXU);40 ASSERT_ERRNO_SUCCESS();41 ASSERT_GT(fd, 0);42 ASSERT_THAT(LIBC_NAMESPACE::write(fd, HELLO, HELLO_SIZE),43 Succeeds(HELLO_SIZE));44 ASSERT_THAT(LIBC_NAMESPACE::fsync(fd), Succeeds(0));45 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));46 47 fd = LIBC_NAMESPACE::open(TEST_FILE, O_WRONLY);48 ASSERT_ERRNO_SUCCESS();49 ASSERT_GT(fd, 0);50 ASSERT_THAT(LIBC_NAMESPACE::pwrite(fd, HELLO, HELLO_SIZE, OFFSET),51 Succeeds(HELLO_SIZE));52 ASSERT_THAT(LIBC_NAMESPACE::fsync(fd), Succeeds(0));53 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));54 55 fd = LIBC_NAMESPACE::open(TEST_FILE, O_RDONLY);56 ASSERT_ERRNO_SUCCESS();57 ASSERT_GT(fd, 0);58 char read_buf[OFFSET_TEXT_SIZE];59 ASSERT_THAT(LIBC_NAMESPACE::pread(fd, read_buf, HELLO_SIZE, OFFSET),60 Succeeds(HELLO_SIZE));61 EXPECT_STREQ(read_buf, HELLO);62 ASSERT_THAT(LIBC_NAMESPACE::pread(fd, read_buf, OFFSET_TEXT_SIZE, 0),63 Succeeds(OFFSET_TEXT_SIZE));64 EXPECT_STREQ(read_buf, OFFSET_TEXT);65 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));66 67 ASSERT_THAT(LIBC_NAMESPACE::unlink(TEST_FILE), Succeeds(0));68}69 70TEST_F(LlvmLibcUniStd, PWriteFails) {71 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;72 EXPECT_THAT(LIBC_NAMESPACE::pwrite(-1, "", 1, 0), Fails<ssize_t>(EBADF));73}74 75TEST_F(LlvmLibcUniStd, PReadFails) {76 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;77 EXPECT_THAT(LIBC_NAMESPACE::pread(-1, nullptr, 1, 0), Fails<ssize_t>(EBADF));78}79