2005-05-02 Roland McGrath <roland@redhat.com>
[kopensolaris-gnu/glibc.git] / 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 }