1  /* Test changing the protections of a file.
       2     Copyright (C) 2020-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 (chmod, int, (const char *, mode_t));
      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-chmod."
      32  
      33  int
      34  main (void)
      35  {
      36    /* Test that chmod works on regular files.  */
      37    {
      38      struct stat statbuf;
      39      unlink (BASE "file");
      40      ASSERT (close (creat (BASE "file", 0600)) == 0);
      41      ASSERT (chmod (BASE "file", 0400) == 0);
      42      ASSERT (stat (BASE "file", &statbuf) >= 0);
      43      ASSERT ((statbuf.st_mode & 0700) == 0400);
      44  
      45      errno = 0;
      46      ASSERT (chmod (BASE "file/", 0600) == -1);
      47      ASSERT (errno == ENOTDIR);
      48  
      49      /* Clean up.  */
      50      ASSERT (chmod (BASE "file", 0600) == 0);
      51      ASSERT (unlink (BASE "file") == 0);
      52    }
      53  
      54    /* Test that chmod works on directories.  */
      55    {
      56      struct stat statbuf;
      57  
      58      rmdir (BASE "dir");
      59      ASSERT (mkdir (BASE "dir", 0700) == 0);
      60      ASSERT (chmod (BASE "dir", 0500) == 0);
      61      ASSERT (stat (BASE "dir", &statbuf) >= 0);
      62      ASSERT ((statbuf.st_mode & 0700) == 0500);
      63      ASSERT (chmod (BASE "dir/", 0700) == 0);
      64  
      65      /* Clean up.  */
      66      ASSERT (rmdir (BASE "dir") == 0);
      67    }
      68  
      69    /* Test that chmod on symlinks modifies the symlink target.  */
      70    {
      71      unlink (BASE "file");
      72      unlink (BASE "link");
      73      if (symlink (BASE "file", BASE "link") == 0)
      74        {
      75          struct stat statbuf;
      76          ASSERT (close (creat (BASE "file", 0600)) == 0);
      77          chmod (BASE "link", 0400);
      78          ASSERT (stat (BASE "file", &statbuf) >= 0);
      79          ASSERT ((statbuf.st_mode & 0700) == 0400);
      80        }
      81      /* Clean up.  */
      82      unlink (BASE "file");
      83      unlink (BASE "link");
      84    }
      85  
      86    return 0;
      87  }