io/bug-ftw4.c

   1 /* Test if ftw function doesn't leak fds.
   2    Copyright (C) 2003-2017 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Jakub Jelinek <jakub@redhat.com>, 2003.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, see
  18    <http://www.gnu.org/licenses/>.  */
  19
  20 #include <fcntl.h>
  21 #include <ftw.h>
  22 #include <stdlib.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <sys/stat.h>
  26 #include <unistd.h>
  27
  28 static int cb_called;
  29
  30 static int
  31 cb (const char *name, const struct stat64 *st, int type)
  32 {
  33   return cb_called++ & 1;
  34 }
  35
  36 int
  37 main (void)
  38 {
  39   char name[32] = "/tmp/ftwXXXXXX", *p;
  40   int ret, i, result = 0, fd, fd1, fd2;
  41
  42   if (mkdtemp (name) == NULL)
  43     {
  44       printf ("Couldn't make temporary directory: %m\n");
  45       exit (EXIT_FAILURE);
  46     }
  47   p = strchr (name, '\0');
  48   strcpy (p, "/1");
  49   if (mkdir (name, 0755) < 0)
  50     {
  51       printf ("Couldn't make temporary subdirectory: %m\n");
  52       exit (EXIT_FAILURE);
  53     }
  54   *p = '\0';
  55
  56   ret = ftw64 (name, cb, 20);
  57   if (ret != 1)
  58     {
  59       printf ("ftw64 returned %d instead of 1", ret);
  60       result = 1;
  61     }
  62
  63   fd = open (name, O_RDONLY);
  64   if (fd < 0)
  65     {
  66       printf ("open failed: %m\n");
  67       result = 1;
  68     }
  69   fd1 = open (name, O_RDONLY);
  70   if (fd1 < 0)
  71     {
  72       printf ("open failed: %m\n");
  73       result = 1;
  74     }
  75   else
  76     close (fd1);
  77   if (fd >= 0)
  78     close (fd);
  79
  80   for (i = 0; i < 128; ++i)
  81     {
  82       ret = ftw64 (name, cb, 20);
  83       if (ret != 1)
  84         {
  85           printf ("ftw64 returned %d instead of 1", ret);
  86           result = 1;
  87         }
  88     }
  89
  90   fd = open (name, O_RDONLY);
  91   if (fd < 0)
  92     {
  93       printf ("open failed: %m\n");
  94       result = 1;
  95     }
  96   fd2 = open (name, O_RDONLY);
  97   if (fd2 < 0)
  98     {
  99       printf ("open failed: %m\n");
 100       result = 1;
 101     }
 102   else
 103     close (fd2);
 104   if (fd >= 0)
 105     close (fd);
 106
 107   if (fd2 >= fd1 + 128)
 108     {
 109       printf ("ftw64 leaking fds: %d -> %d\n", fd1, fd2);
 110       result = 1;
 111     }
 112
 113   if (cb_called != 129 * 2)
 114     {
 115       printf ("callback called %d times\n", cb_called);
 116       result = 1;
 117     }
 118
 119   strcpy (p, "/1");
 120   rmdir (name);
 121   *p = '\0';
 122   rmdir (name);
 123   return result;
 124 }