tests/test-symlinkat.c

   1 /* Tests of symlinkat and readlinkat.
   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 <unistd.h>
  22
  23 #include "signature.h"
  24 SIGNATURE_CHECK (symlinkat, int, (char const *, int, char const *));
  25 SIGNATURE_CHECK (readlinkat, ssize_t, (int, char const *, char *, size_t));
  26
  27 #include <fcntl.h>
  28 #include <errno.h>
  29 #include <stdbool.h>
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <sys/stat.h>
  34
  35 #ifndef HAVE_SYMLINK
  36 # define HAVE_SYMLINK 0
  37 #endif
  38
  39 #define ASSERT(expr) \
  40   do                                                                         \
  41     {                                                                        \
  42       if (!(expr))                                                           \
  43         {                                                                    \
  44           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__);  \
  45           fflush (stderr);                                                   \
  46           abort ();                                                          \
  47         }                                                                    \
  48     }                                                                        \
  49   while (0)
  50
  51 #define BASE "test-symlinkat.t"
  52
  53 #include "test-readlink.h"
  54 #include "test-symlink.h"
  55
  56 static int dfd = AT_FDCWD;
  57
  58 static int
  59 do_symlink (char const *contents, char const *name)
  60 {
  61   return symlinkat (contents, dfd, name);
  62 }
  63
  64 static ssize_t
  65 do_readlink (char const *name, char *buf, size_t len)
  66 {
  67   return readlinkat (dfd, name, buf, len);
  68 }
  69
  70 int
  71 main (void)
  72 {
  73   char buf[80];
  74   int result;
  75
  76   /* Remove any leftovers from a previous partial run.  */
  77   ASSERT (system ("rm -rf " BASE "*") == 0);
  78
  79   /* Perform same checks as counterpart functions.  */
  80   result = test_readlink (do_readlink, false);
  81   ASSERT (test_symlink (do_symlink, false) == result);
  82   dfd = openat (AT_FDCWD, ".", O_RDONLY);
  83   ASSERT (0 <= dfd);
  84   ASSERT (test_readlink (do_readlink, false) == result);
  85   ASSERT (test_symlink (do_symlink, false) == result);
  86
  87   /* Now perform some cross-directory checks.  Skip everything else on
  88      mingw.  */
  89   if (HAVE_SYMLINK)
  90     {
  91       const char *contents = "don't matter!";
  92       ssize_t exp = strlen (contents);
  93
  94       /* Create link while cwd is '.', then read it in '..'.  */
  95       ASSERT (symlinkat (contents, AT_FDCWD, BASE "link") == 0);
  96       errno = 0;
  97       ASSERT (symlinkat (contents, dfd, BASE "link") == -1);
  98       ASSERT (errno == EEXIST);
  99       ASSERT (chdir ("..") == 0);
 100       errno = 0;
 101       ASSERT (readlinkat (AT_FDCWD, BASE "link", buf, sizeof buf) == -1);
 102       ASSERT (errno == ENOENT);
 103       ASSERT (readlinkat (dfd, BASE "link", buf, sizeof buf) == exp);
 104       ASSERT (strncmp (contents, buf, exp) == 0);
 105       ASSERT (unlinkat (dfd, BASE "link", 0) == 0);
 106
 107       /* Create link while cwd is '..', then read it in '.'.  */
 108       ASSERT (symlinkat (contents, dfd, BASE "link") == 0);
 109       ASSERT (fchdir (dfd) == 0);
 110       errno = 0;
 111       ASSERT (symlinkat (contents, AT_FDCWD, BASE "link") == -1);
 112       ASSERT (errno == EEXIST);
 113       buf[0] = '\0';
 114       ASSERT (readlinkat (AT_FDCWD, BASE "link", buf, sizeof buf) == exp);
 115       ASSERT (strncmp (contents, buf, exp) == 0);
 116       buf[0] = '\0';
 117       ASSERT (readlinkat (dfd, BASE "link", buf, sizeof buf) == exp);
 118       ASSERT (strncmp (contents, buf, exp) == 0);
 119       ASSERT (unlink (BASE "link") == 0);
 120     }
 121
 122   ASSERT (close (dfd) == 0);
 123   if (result == 77)
 124     fputs ("skipping test: symlinks not supported on this file system\n",
 125            stderr);
 126   return result;
 127 }