tests/test-openat.c

   1 /* Test that openat works.
   2    Copyright (C) 2009-2014 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 "progname.h"
  33 #include "macros.h"
  34
  35 #define BASE "test-openat.t"
  36
  37 #include "test-open.h"
  38
  39 static int dfd = AT_FDCWD;
  40
  41 /* Wrapper around openat to test open behavior.  */
  42 static int
  43 do_open (char const *name, int flags, ...)
  44 {
  45   if (flags & O_CREAT)
  46     {
  47       mode_t mode = 0;
  48       va_list arg;
  49       va_start (arg, flags);
  50
  51       /* We have to use PROMOTED_MODE_T instead of mode_t, otherwise GCC 4
  52          creates crashing code when 'mode_t' is smaller than 'int'.  */
  53       mode = va_arg (arg, PROMOTED_MODE_T);
  54
  55       va_end (arg);
  56       return openat (dfd, name, flags, mode);
  57     }
  58   return openat (dfd, name, flags);
  59 }
  60
  61 int
  62 main (int argc _GL_UNUSED, char *argv[])
  63 {
  64   int result;
  65
  66   set_program_name (argv[0]);
  67
  68   /* Test behaviour for invalid file descriptors.  */
  69   {
  70     errno = 0;
  71     ASSERT (openat (-1, "foo", O_RDONLY) == -1);
  72     ASSERT (errno == EBADF);
  73   }
  74   {
  75     close (99);
  76     errno = 0;
  77     ASSERT (openat (99, "foo", O_RDONLY) == -1);
  78     ASSERT (errno == EBADF);
  79   }
  80
  81   /* Basic checks.  */
  82   result = test_open (do_open, false);
  83   dfd = open (".", O_RDONLY);
  84   ASSERT (0 <= dfd);
  85   ASSERT (test_open (do_open, false) == result);
  86   ASSERT (close (dfd) == 0);
  87
  88   /* Check that even when *-safer modules are in use, plain openat can
  89      land in fd 0.  Do this test last, since it is destructive to
  90      stdin.  */
  91   ASSERT (close (STDIN_FILENO) == 0);
  92   ASSERT (openat (AT_FDCWD, ".", O_RDONLY) == STDIN_FILENO);
  93   {
  94     dfd = open (".", O_RDONLY);
  95     ASSERT (STDIN_FILENO < dfd);
  96     ASSERT (chdir ("..") == 0);
  97     ASSERT (close (STDIN_FILENO) == 0);
  98     ASSERT (openat (dfd, ".", O_RDONLY) == STDIN_FILENO);
  99     ASSERT (close (dfd) == 0);
 100   }
 101   return result;
 102 }