io/tst-fchownat.c

   1 #include <dirent.h>
   2 #include <fcntl.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <unistd.h>
   7
   8
   9 static void prepare (void);
  10 #define PREPARE(argc, argv) prepare ()
  11
  12 static int do_test (void);
  13 #define TEST_FUNCTION do_test ()
  14
  15 #include "../test-skeleton.c"
  16
  17 static int dir_fd;
  18
  19 static void
  20 prepare (void)
  21 {
  22 #if _POSIX_CHOWN_RESTRICTED > 0
  23   uid_t uid = getuid ();
  24   if (uid != 0)
  25     {
  26       puts ("need root privileges");
  27       exit (0);
  28     }
  29 #endif
  30
  31   size_t test_dir_len = strlen (test_dir);
  32   static const char dir_name[] = "/tst-fchownat.XXXXXX";
  33
  34   size_t dirbuflen = test_dir_len + sizeof (dir_name);
  35   char *dirbuf = malloc (dirbuflen);
  36   if (dirbuf == NULL)
  37     {
  38       puts ("out of memory");
  39       exit (1);
  40     }
  41
  42   snprintf (dirbuf, dirbuflen, "%s%s", test_dir, dir_name);
  43   if (mkdtemp (dirbuf) == NULL)
  44     {
  45       puts ("cannot create temporary directory");
  46       exit (1);
  47     }
  48
  49   add_temp_file (dirbuf);
  50
  51   dir_fd = open (dirbuf, O_RDONLY | O_DIRECTORY);
  52   if (dir_fd == -1)
  53     {
  54       puts ("cannot open directory");
  55       exit (1);
  56     }
  57 }
  58
  59
  60 static int
  61 do_test (void)
  62 {
  63   /* fdopendir takes over the descriptor, make a copy.  */
  64   int dupfd = dup (dir_fd);
  65   if (dupfd == -1)
  66     {
  67       puts ("dup failed");
  68       return 1;
  69     }
  70   if (lseek (dupfd, 0, SEEK_SET) != 0)
  71     {
  72       puts ("1st lseek failed");
  73       return 1;
  74     }
  75
  76   /* The directory should be empty safe the . and .. files.  */
  77   DIR *dir = fdopendir (dupfd);
  78   if (dir == NULL)
  79     {
  80       puts ("fdopendir failed");
  81       return 1;
  82     }
  83   struct dirent64 *d;
  84   while ((d = readdir64 (dir)) != NULL)
  85     if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
  86       {
  87         printf ("temp directory contains file \"%s\"\n", d->d_name);
  88         return 1;
  89       }
  90   closedir (dir);
  91
  92   /* Try to create a file.  */
  93   int fd = openat (dir_fd, "some-file", O_CREAT|O_RDWR|O_EXCL, 0666);
  94   if (fd == -1)
  95     {
  96       if (errno == ENOSYS)
  97         {
  98           puts ("*at functions not supported");
  99           return 0;
 100         }
 101
 102       puts ("file creation failed");
 103       return 1;
 104     }
 105   write (fd, "hello", 5);
 106   puts ("file created");
 107
 108   struct stat64 st1;
 109   if (fstat64 (fd, &st1) != 0)
 110     {
 111       puts ("fstat64 failed");
 112       return 1;
 113     }
 114
 115   /* Before closing the file, try using this file descriptor to open
 116      another file.  This must fail.  */
 117   if (fchownat (fd, "some-file", 1, 1, 0) != -1)
 118     {
 119       puts ("fchownat using descriptor for normal file worked");
 120       return 1;
 121     }
 122   if (errno != ENOTDIR)
 123     {
 124       puts ("\
 125 error for fchownat using descriptor for normal file not ENOTDIR ");
 126       return 1;
 127     }
 128
 129   close (fd);
 130
 131   if (fchownat (dir_fd, "some-file", st1.st_uid + 1, st1.st_gid + 1, 0) != 0)
 132     {
 133       puts ("fchownat failed");
 134       return 1;
 135     }
 136
 137   struct stat64 st2;
 138   if (fstatat64 (dir_fd, "some-file", &st2, 0) != 0)
 139     {
 140       puts ("fstatat64 failed");
 141       return 1;
 142     }
 143
 144   if (st1.st_uid + 1 != st2.st_uid || st1.st_gid + 1 != st2.st_gid)
 145     {
 146       puts ("owner change failed");
 147       return 1;
 148     }
 149
 150   if (unlinkat (dir_fd, "some-file", 0) != 0)
 151     {
 152       puts ("unlinkat failed");
 153       return 1;
 154     }
 155
 156   /* Create a file descriptor which is closed again right away.  */
 157   int dir_fd2 = dup (dir_fd);
 158   if (dir_fd2 == -1)
 159     {
 160       puts ("dup failed");
 161       return 1;
 162     }
 163   close (dir_fd2);
 164
 165   if (fchownat (dir_fd2, "some-file", 1, 1, 0) != -1)
 166     {
 167       puts ("fchownat using closed descriptor worked");
 168       return 1;
 169     }
 170   if (errno != EBADF)
 171     {
 172       puts ("error for fchownat using closed descriptor not EBADF ");
 173       return 1;
 174     }
 175
 176   close (dir_fd);
 177
 178   if (fchownat (-1, "some-file", 1, 1, 0) != -1)
 179     {
 180       puts ("fchownat using invalid descriptor worked");
 181       return 1;
 182     }
 183   if (errno != EBADF)
 184     {
 185       puts ("error for fchownat using invalid descriptor not EBADF ");
 186       return 1;
 187     }
 188
 189   return 0;
 190 }