tests/test-openat.c

   1 /* Test that openat works.
   2    Copyright (C) 2009 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 <http://www.gnu.org/licenses/>.  */
  16
  17 /* Written by Eric Blake <ebb9@byu.net>, 2009.  */
  18
  19 #include <config.h>
  20
  21 #include <fcntl.h>
  22
  23 #include <errno.h>
  24 #include <stdarg.h>
  25 #include <stdbool.h>
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <unistd.h>
  29
  30 #define ASSERT(expr) \
  31   do                                                                         \
  32     {                                                                        \
  33       if (!(expr))                                                           \
  34         {                                                                    \
  35           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
  36           fflush (stderr);                                                   \
  37           abort ();                                                          \
  38         }                                                                    \
  39     }                                                                        \
  40   while (0)
  41
  42 #define BASE "test-openat.t"
  43
  44 #include "test-open.h"
  45
  46 static int dfd = AT_FDCWD;
  47
  48 /* Wrapper around openat to test open behavior.  */
  49 static int
  50 do_open (char const *name, int flags, ...)
  51 {
  52   if (flags & O_CREAT)
  53     {
  54       mode_t mode = 0;
  55       va_list arg;
  56       va_start (arg, flags);
  57
  58       /* We have to use PROMOTED_MODE_T instead of mode_t, otherwise GCC 4
  59          creates crashing code when 'mode_t' is smaller than 'int'.  */
  60       mode = va_arg (arg, PROMOTED_MODE_T);
  61
  62       va_end (arg);
  63       return openat (dfd, name, flags, mode);
  64     }
  65   return openat (dfd, name, flags);
  66 }
  67
  68 int
  69 main (void)
  70 {
  71   int result;
  72
  73   /* Basic checks.  */
  74   result = test_open (do_open, false);
  75   dfd = open (".", O_RDONLY);
  76   ASSERT (0 <= dfd);
  77   ASSERT (test_open (do_open, false) == result);
  78   ASSERT (close (dfd) == 0);
  79
  80   /* Check that even when *-safer modules are in use, plain openat can
  81      land in fd 0.  Do this test last, since it is destructive to
  82      stdin.  */
  83   ASSERT (close (STDIN_FILENO) == 0);
  84   ASSERT (openat (AT_FDCWD, ".", O_RDONLY) == STDIN_FILENO);
  85   {
  86     dfd = open (".", O_RDONLY);
  87     ASSERT (STDIN_FILENO < dfd);
  88     ASSERT (chdir ("..") == 0);
  89     ASSERT (close (STDIN_FILENO) == 0);
  90     ASSERT (openat (dfd, ".", O_RDONLY) == STDIN_FILENO);
  91     ASSERT (close (dfd) == 0);
  92   }
  93   return result;
  94 }