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

source code of glibc/io/bug-ftw4.c