io/bug-ftw4.c

   1 /* Test if ftw function doesn't leak fds.
   2    Copyright (C) 2003 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, write to the Free
  18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  19    02111-1307 USA.  */
  20
  21 #include <fcntl.h>
  22 #include <ftw.h>
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <string.h>
  26 #include <sys/stat.h>
  27 #include <unistd.h>
  28
  29 static int cb_called;
  30
  31 static int
  32 cb (const char *name, const struct stat64 *st, int type)
  33 {
  34   return cb_called++ & 1;
  35 }
  36
  37 int
  38 main (void)
  39 {
  40   char name[32] = "/tmp/ftwXXXXXX", *p;
  41   int ret, i, result = 0, fd, fd1, fd2;
  42
  43   if (mkdtemp (name) == NULL)
  44     {
  45       printf ("Couldn't make temporary directory: %m\n");
  46       exit (EXIT_FAILURE);
  47     }
  48   p = strchr (name, '\0');
  49   strcpy (p, "/1");
  50   if (mkdir (name, 0755) < 0)
  51     {
  52       printf ("Couldn't make temporary subdirectory: %m\n");
  53       exit (EXIT_FAILURE);
  54     }
  55   *p = '\0';
  56
  57   ret = ftw64 (name, cb, 20);
  58   if (ret != 1)
  59     {
  60       printf ("ftw64 returned %d instead of 1", ret);
  61       result = 1;
  62     }
  63
  64   fd = open (name, O_RDONLY);
  65   if (fd < 0)
  66     {
  67       printf ("open failed: %m\n");
  68       result = 1;
  69     }
  70   fd1 = open (name, O_RDONLY);
  71   if (fd1 < 0)
  72     {
  73       printf ("open failed: %m\n");
  74       result = 1;
  75     }
  76   else
  77     close (fd1);
  78   if (fd >= 0)
  79     close (fd);
  80
  81   for (i = 0; i < 128; ++i)
  82     {
  83       ret = ftw64 (name, cb, 20);
  84       if (ret != 1)
  85         {
  86           printf ("ftw64 returned %d instead of 1", ret);
  87           result = 1;
  88         }
  89     }
  90
  91   fd = open (name, O_RDONLY);
  92   if (fd < 0)
  93     {
  94       printf ("open failed: %m\n");
  95       result = 1;
  96     }
  97   fd2 = open (name, O_RDONLY);
  98   if (fd2 < 0)
  99     {
 100       printf ("open failed: %m\n");
 101       result = 1;
 102     }
 103   else
 104     close (fd2);
 105   if (fd >= 0)
 106     close (fd);
 107
 108   if (fd2 >= fd1 + 128)
 109     {
 110       printf ("ftw64 leaking fds: %d -> %d\n", fd1, fd2);
 111       result = 1;
 112     }
 113
 114   if (cb_called != 129 * 2)
 115     {
 116       printf ("callback called %d times\n", cb_called);
 117       result = 1;
 118     }
 119
 120   strcpy (p, "/1");
 121   rmdir (name);
 122   *p = '\0';
 123   rmdir (name);
 124   return result;
 125 }