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-exception |
6 | // |
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 | |
22 | using LlvmLibcUniStd = LIBC_NAMESPACE::testing::ErrnoCheckingTest; |
23 | |
24 | TEST_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 the |
27 | // file again and pread at an offset and make sure that only expected data |
28 | // 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 | |
70 | TEST_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 | |
75 | TEST_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 | |