1 //===-- Unittests for fchmodat --------------------------------------------===//
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/errno/libc_errno.h"
10 #include "src/fcntl/open.h"
11 #include "src/sys/stat/fchmodat.h"
12 #include "src/unistd/close.h"
13 #include "src/unistd/write.h"
14 #include "test/UnitTest/ErrnoSetterMatcher.h"
15 #include "test/UnitTest/Test.h"
16
17 #include "hdr/fcntl_macros.h"
18 #include <sys/stat.h>
19
TEST(LlvmLibcFchmodatTest,ChangeAndOpen)20 TEST(LlvmLibcFchmodatTest, ChangeAndOpen) {
21 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;
22 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds;
23
24 // The test file is initially writable. We open it for writing and ensure
25 // that it indeed can be opened for writing. Next, we close the file and
26 // make it readonly using chmod. We test that chmod actually succeeded by
27 // trying to open the file for writing and failing.
28 constexpr const char *TEST_FILE = "testdata/fchmodat.test";
29 constexpr const char *TEST_DIR = "testdata";
30 constexpr const char *TEST_FILE_BASENAME = "fchmodat.test";
31 const char WRITE_DATA[] = "fchmodat test";
32 constexpr ssize_t WRITE_SIZE = ssize_t(sizeof(WRITE_DATA));
33 LIBC_NAMESPACE::libc_errno = 0;
34
35 int fd = LIBC_NAMESPACE::open(TEST_FILE, O_CREAT | O_WRONLY, S_IRWXU);
36 ASSERT_GT(fd, 0);
37 ASSERT_ERRNO_SUCCESS();
38 ASSERT_EQ(LIBC_NAMESPACE::write(fd, WRITE_DATA, sizeof(WRITE_DATA)),
39 WRITE_SIZE);
40 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));
41
42 int dirfd = LIBC_NAMESPACE::open(TEST_DIR, O_DIRECTORY);
43 ASSERT_GT(dirfd, 0);
44 ASSERT_ERRNO_SUCCESS();
45
46 EXPECT_THAT(LIBC_NAMESPACE::fchmodat(dirfd, TEST_FILE_BASENAME, S_IRUSR, 0),
47 Succeeds(0));
48
49 // Opening for writing should fail.
50 EXPECT_EQ(LIBC_NAMESPACE::open(TEST_FILE, O_APPEND | O_WRONLY), -1);
51 ASSERT_ERRNO_FAILURE();
52 LIBC_NAMESPACE::libc_errno = 0;
53 // But opening for reading should succeed.
54 fd = LIBC_NAMESPACE::open(TEST_FILE, O_APPEND | O_RDONLY);
55 EXPECT_GT(fd, 0);
56 ASSERT_ERRNO_SUCCESS();
57
58 EXPECT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));
59 EXPECT_THAT(LIBC_NAMESPACE::fchmodat(dirfd, TEST_FILE_BASENAME, S_IRWXU, 0),
60 Succeeds(0));
61
62 EXPECT_THAT(LIBC_NAMESPACE::close(dirfd), Succeeds(0));
63 }
64
TEST(LlvmLibcFchmodatTest,NonExistentFile)65 TEST(LlvmLibcFchmodatTest, NonExistentFile) {
66 LIBC_NAMESPACE::libc_errno = 0;
67 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;
68 ASSERT_THAT(
69 LIBC_NAMESPACE::fchmodat(AT_FDCWD, "non-existent-file", S_IRUSR, 0),
70 Fails(ENOENT));
71 LIBC_NAMESPACE::libc_errno = 0;
72 }
73