tests/test-openat.c

   1 /* Test that openat works.
   2    Copyright (C) 2009, 2010 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 "signature.h"
  24 SIGNATURE_CHECK (openat, int, (int, char const *, int, ...));
  25
  26 #include <errno.h>
  27 #include <stdarg.h>
  28 #include <stdbool.h>
  29 #include <stdio.h>
  30 #include <unistd.h>
  31
  32 #include "macros.h"
  33
  34 #define BASE "test-openat.t"
  35
  36 #include "test-open.h"
  37
  38 static int dfd = AT_FDCWD;
  39
  40 /* Wrapper around openat to test open behavior.  */
  41 static int
  42 do_open (char const *name, int flags, ...)
  43 {
  44   if (flags & O_CREAT)
  45     {
  46       mode_t mode = 0;
  47       va_list arg;
  48       va_start (arg, flags);
  49
  50       /* We have to use PROMOTED_MODE_T instead of mode_t, otherwise GCC 4
  51          creates crashing code when 'mode_t' is smaller than 'int'.  */
  52       mode = va_arg (arg, PROMOTED_MODE_T);
  53
  54       va_end (arg);
  55       return openat (dfd, name, flags, mode);
  56     }
  57   return openat (dfd, name, flags);
  58 }
  59
  60 int
  61 main (void)
  62 {
  63   int result;
  64
  65   /* Basic checks.  */
  66   result = test_open (do_open, false);
  67   dfd = open (".", O_RDONLY);
  68   ASSERT (0 <= dfd);
  69   ASSERT (test_open (do_open, false) == result);
  70   ASSERT (close (dfd) == 0);
  71
  72   /* Check that even when *-safer modules are in use, plain openat can
  73      land in fd 0.  Do this test last, since it is destructive to
  74      stdin.  */
  75   ASSERT (close (STDIN_FILENO) == 0);
  76   ASSERT (openat (AT_FDCWD, ".", O_RDONLY) == STDIN_FILENO);
  77   {
  78     dfd = open (".", O_RDONLY);
  79     ASSERT (STDIN_FILENO < dfd);
  80     ASSERT (chdir ("..") == 0);
  81     ASSERT (close (STDIN_FILENO) == 0);
  82     ASSERT (openat (dfd, ".", O_RDONLY) == STDIN_FILENO);
  83     ASSERT (close (dfd) == 0);
  84   }
  85   return result;
  86 }