tests/test-fchownat.c

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