libc/test/src/sys/stat/fchmod_test.cpp

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