1  /* Tests of fchownat.
       2     Copyright (C) 2009-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  /* Written by Eric Blake <ebb9@byu.net>, 2009.  */
      18  
      19  #include <config.h>
      20  
      21  #include <unistd.h>
      22  
      23  #include "signature.h"
      24  SIGNATURE_CHECK (fchownat, int, (int, char const *, uid_t, gid_t, int));
      25  
      26  #include <fcntl.h>
      27  #include <errno.h>
      28  #include <stdio.h>
      29  #include <stdlib.h>
      30  #include <sys/stat.h>
      31  
      32  #include "mgetgroups.h"
      33  #include "openat.h"
      34  #include "stat-time.h"
      35  #include "ignore-value.h"
      36  #include "macros.h"
      37  
      38  #ifndef BASE
      39  # define BASE "test-fchownat.t"
      40  #endif
      41  
      42  #include "test-chown.h"
      43  #include "test-lchown.h"
      44  
      45  static int dfd = AT_FDCWD;
      46  
      47  /* Wrapper around fchownat to test chown behavior.  */
      48  static int
      49  do_chown (char const *name, uid_t user, gid_t group)
      50  {
      51  #ifdef TEST_CHOWNAT
      52    return chownat (dfd, name, user, group);
      53  #else
      54    return fchownat (dfd, name, user, group, 0);
      55  #endif
      56  }
      57  
      58  /* Wrapper around fchownat to test lchown behavior.  */
      59  static int
      60  do_lchown (char const *name, uid_t user, gid_t group)
      61  {
      62  #ifdef TEST_CHOWNAT
      63    return lchownat (dfd, name, user, group);
      64  #else
      65    return fchownat (dfd, name, user, group, AT_SYMLINK_NOFOLLOW);
      66  #endif
      67  }
      68  
      69  int
      70  main (_GL_UNUSED int argc, char *argv[])
      71  {
      72    int result1; /* Skip because of no chown/symlink support.  */
      73    int result2; /* Skip because of no lchown support.  */
      74  
      75    /* Clean up any trash from prior testsuite runs.  */
      76    ignore_value (system ("rm -rf " BASE "*"));
      77  
      78    /* Test behaviour for invalid file descriptors.  */
      79    {
      80      errno = 0;
      81      ASSERT (fchownat (-1, "foo", getuid (), getgid (), 0) == -1);
      82      ASSERT (errno == EBADF);
      83    }
      84    {
      85      close (99);
      86      errno = 0;
      87      ASSERT (fchownat (99, "foo", getuid (), getgid (), 0) == -1);
      88      ASSERT (errno == EBADF);
      89    }
      90  
      91    /* Basic tests.  */
      92    result1 = test_chown (do_chown, true);
      93    result2 = test_lchown (do_lchown, result1 == 0);
      94    dfd = open (".", O_RDONLY);
      95    ASSERT (0 <= dfd);
      96    ASSERT (test_chown (do_chown, false) == result1);
      97    ASSERT (test_lchown (do_lchown, false) == result2);
      98    /* We expect 0/0, 0/77, or 77/77, but not 77/0.  */
      99    ASSERT (result1 <= result2);
     100    ASSERT (close (dfd) == 0);
     101  
     102    /* FIXME - add additional tests of dfd not at current directory.  */
     103    return result1 | result2;
     104  }