1  /* Test changing the protections of a file relative to an open directory.
       2     Copyright (C) 2011-2023 Free Software Foundation, Inc.
       3  
       4     This program is free software: you can redistribute it and/or modify
       5     it under the terms of the GNU General Public License as published by
       6     the Free Software Foundation, either version 3 of the License, or
       7     (at your option) any later version.
       8  
       9     This program is distributed in the hope that it will be useful,
      10     but WITHOUT ANY WARRANTY; without even the implied warranty of
      11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      12     GNU General Public License for more details.
      13  
      14     You should have received a copy of the GNU General Public License
      15     along with this program.  If not, see <https://www.gnu.org/licenses/>.  */
      16  
      17  #include <config.h>
      18  
      19  #include <sys/stat.h>
      20  
      21  #include "signature.h"
      22  SIGNATURE_CHECK (fchmodat, int, (int, const char *, mode_t, int));
      23  
      24  #include <errno.h>
      25  #include <fcntl.h>
      26  #include <sys/stat.h>
      27  #include <unistd.h>
      28  
      29  #include "macros.h"
      30  
      31  #define BASE "test-fchmodat."
      32  
      33  int
      34  main (void)
      35  {
      36    /* Test behaviour for invalid file descriptors.  */
      37    {
      38      errno = 0;
      39      ASSERT (fchmodat (-1, "foo", 0600, 0) == -1);
      40      ASSERT (errno == EBADF);
      41    }
      42    {
      43      close (99);
      44      errno = 0;
      45      ASSERT (fchmodat (99, "foo", 0600, 0) == -1);
      46      ASSERT (errno == EBADF);
      47    }
      48  
      49    /* Test that fchmodat works on regular files.  */
      50    {
      51      struct stat statbuf;
      52  
      53      unlink (BASE "file");
      54      ASSERT (close (creat (BASE "file", 0600)) == 0);
      55      ASSERT (fchmodat (AT_FDCWD, BASE "file", 0400, 0) == 0);
      56      ASSERT (stat (BASE "file", &statbuf) >= 0);
      57      ASSERT ((statbuf.st_mode & 0700) == 0400);
      58  
      59      errno = 0;
      60      ASSERT (fchmodat (AT_FDCWD, BASE "file/", 0600, 0) == -1);
      61      ASSERT (errno == ENOTDIR);
      62  
      63      /* Clean up.  */
      64      ASSERT (chmod (BASE "file", 0600) == 0);
      65      ASSERT (unlink (BASE "file") == 0);
      66    }
      67  
      68    /* Test that fchmodat works on directories.  */
      69    {
      70      struct stat statbuf;
      71      rmdir (BASE "dir");
      72      ASSERT (mkdir (BASE "dir", 0700) == 0);
      73      ASSERT (fchmodat (AT_FDCWD, BASE "dir", 0500, 0) == 0);
      74      ASSERT (stat (BASE "dir", &statbuf) >= 0);
      75      ASSERT ((statbuf.st_mode & 0700) == 0500);
      76      ASSERT (fchmodat (AT_FDCWD, BASE "dir/", 0700, 0) == 0);
      77  
      78      /* Clean up.  */
      79      ASSERT (rmdir (BASE "dir") == 0);
      80    }
      81  
      82    /* Test that fchmodat works on non-symlinks, when given
      83       the AT_SYMLINK_NOFOLLOW flag.  */
      84    {
      85      struct stat statbuf;
      86      unlink (BASE "file");
      87      ASSERT (close (creat (BASE "file", 0600)) == 0);
      88      ASSERT (fchmodat (AT_FDCWD, BASE "file", 0400, AT_SYMLINK_NOFOLLOW) == 0);
      89      ASSERT (stat (BASE "file", &statbuf) >= 0);
      90      ASSERT ((statbuf.st_mode & 0700) == 0400);
      91      /* Clean up.  */
      92      ASSERT (chmod (BASE "file", 0600) == 0);
      93      ASSERT (unlink (BASE "file") == 0);
      94    }
      95  
      96    return 0;
      97  }