sysdeps/unix/sysv/linux/tst-getdents64.c

   1 /* Test for reading directories with getdents64.
   2    Copyright (C) 2019 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, see
  17    <https://www.gnu.org/licenses/>.  */
  18
  19 #include <dirent.h>
  20 #include <errno.h>
  21 #include <fcntl.h>
  22 #include <limits.h>
  23 #include <stdbool.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <string.h>
  27 #include <support/check.h>
  28 #include <support/support.h>
  29 #include <support/xunistd.h>
  30 #include <sys/mman.h>
  31 #include <unistd.h>
  32
  33 /* Called by large_buffer_checks below.  */
  34 static void
  35 large_buffer_check (int fd, char *large_buffer, size_t large_buffer_size)
  36 {
  37   xlseek (fd, 0, SEEK_SET);
  38   ssize_t ret = getdents64 (fd, large_buffer, large_buffer_size);
  39   if (ret < 0)
  40     FAIL_EXIT1 ("getdents64 for buffer of %zu bytes failed: %m",
  41                 large_buffer_size);
  42   if (ret < offsetof (struct dirent64, d_name))
  43     FAIL_EXIT1 ("getdents64 for buffer of %zu returned small value %zd",
  44                 large_buffer_size, ret);
  45 }
  46
  47 /* Bug 24740: Make sure that the system call argument is adjusted
  48    properly for the int type.  A large value should stay a large
  49    value, and not wrap around to something small, causing the system
  50    call to fail with EINVAL.  */
  51 static void
  52 large_buffer_checks (int fd)
  53 {
  54   size_t large_buffer_size;
  55   if (!__builtin_add_overflow (UINT_MAX, 2, &large_buffer_size))
  56     {
  57       int flags = MAP_ANONYMOUS | MAP_PRIVATE;
  58 #ifdef MAP_NORESERVE
  59       flags |= MAP_NORESERVE;
  60 #endif
  61       void *large_buffer = mmap (NULL, large_buffer_size,
  62                                  PROT_READ | PROT_WRITE, flags, -1, 0);
  63       if (large_buffer == MAP_FAILED)
  64         printf ("warning: could not allocate %zu bytes of memory,"
  65                 " subtests skipped\n", large_buffer_size);
  66       else
  67         {
  68           large_buffer_check (fd, large_buffer, INT_MAX);
  69           large_buffer_check (fd, large_buffer, (size_t) INT_MAX + 1);
  70           large_buffer_check (fd, large_buffer, (size_t) INT_MAX + 2);
  71           large_buffer_check (fd, large_buffer, UINT_MAX);
  72           large_buffer_check (fd, large_buffer, (size_t) UINT_MAX + 1);
  73           large_buffer_check (fd, large_buffer, (size_t) UINT_MAX + 2);
  74           xmunmap (large_buffer, large_buffer_size);
  75         }
  76     }
  77 }
  78
  79 static int
  80 do_test (void)
  81 {
  82   /* The test compares the iteration order with readdir64.  */
  83   DIR *reference = opendir (".");
  84   TEST_VERIFY_EXIT (reference != NULL);
  85
  86   int fd = xopen (".", O_RDONLY | O_DIRECTORY, 0);
  87   TEST_VERIFY (fd >= 0);
  88
  89   /* Perform two passes, with a rewind operating between passes.  */
  90   for (int pass = 0; pass < 2; ++pass)
  91     {
  92       /* Check that we need to fill the buffer multiple times.  */
  93       int read_count = 0;
  94
  95       while (true)
  96         {
  97           /* Simple way to make sure that the memcpy below does not read
  98              non-existing data.  */
  99           struct
 100           {
 101             char buffer[1024];
 102             struct dirent64 pad;
 103           } data;
 104
 105           ssize_t ret = getdents64 (fd, &data.buffer, sizeof (data.buffer));
 106           if (ret < 0)
 107             FAIL_EXIT1 ("getdents64: %m");
 108           if (ret == 0)
 109             break;
 110           ++read_count;
 111
 112           char *current = data.buffer;
 113           char *end = data.buffer + ret;
 114           while (current != end)
 115             {
 116               struct dirent64 entry;
 117               memcpy (&entry, current, sizeof (entry));
 118               /* Truncate overlong strings.  */
 119               entry.d_name[sizeof (entry.d_name) - 1] = '\0';
 120               TEST_VERIFY (strlen (entry.d_name) < sizeof (entry.d_name) - 1);
 121
 122               errno = 0;
 123               struct dirent64 *refentry = readdir64 (reference);
 124               if (refentry == NULL && errno == 0)
 125                 FAIL_EXIT1 ("readdir64 failed too early, at: %s",
 126                             entry.d_name);
 127               else if (refentry == NULL)
 128                 FAIL_EXIT1 ("readdir64: %m");
 129
 130               TEST_COMPARE_STRING (entry.d_name, refentry->d_name);
 131               TEST_COMPARE (entry.d_ino, refentry->d_ino);
 132               TEST_COMPARE (entry.d_off, refentry->d_off);
 133               TEST_COMPARE (entry.d_type, refentry->d_type);
 134
 135               /* Offset zero is reserved for the first entry.  */
 136               TEST_VERIFY (entry.d_off != 0);
 137
 138               TEST_VERIFY_EXIT (entry.d_reclen <= end - current);
 139               current += entry.d_reclen;
 140             }
 141         }
 142
 143       /* We expect to have reached the end of the stream.  */
 144       errno = 0;
 145       TEST_VERIFY (readdir64 (reference) == NULL);
 146       TEST_COMPARE (errno, 0);
 147
 148       /* direntries_read has been called more than once.  */
 149       TEST_VERIFY (read_count > 0);
 150
 151       /* Rewind both directory streams.  */
 152       xlseek (fd, 0, SEEK_SET);
 153       rewinddir (reference);
 154     }
 155
 156   large_buffer_checks (fd);
 157
 158   xclose (fd);
 159   closedir (reference);
 160   return 0;
 161 }
 162
 163 #include <support/test-driver.c>