1#include <dirent.h>
2#include <fcntl.h>
3#include <stdbool.h>
4#include <stdio.h>
5#include <stdlib.h>
6#include <string.h>
7#include <unistd.h>
8
9
10static void prepare (void);
11#define PREPARE(argc, argv) prepare ()
12
13static int do_test (void);
14#define TEST_FUNCTION do_test ()
15
16#include "../test-skeleton.c"
17
18static int dir_fd;
19
20static void
21prepare (void)
22{
23 size_t test_dir_len = strlen (test_dir);
24 static const char dir_name[] = "/tst-unlinkat.XXXXXX";
25
26 size_t dirbuflen = test_dir_len + sizeof (dir_name);
27 char *dirbuf = malloc (size: dirbuflen);
28 if (dirbuf == NULL)
29 {
30 puts (s: "out of memory");
31 exit (1);
32 }
33
34 snprintf (s: dirbuf, maxlen: dirbuflen, format: "%s%s", test_dir, dir_name);
35 if (mkdtemp (template: dirbuf) == NULL)
36 {
37 puts (s: "cannot create temporary directory");
38 exit (1);
39 }
40
41 add_temp_file (name: dirbuf);
42
43 dir_fd = open (file: dirbuf, O_RDONLY | O_DIRECTORY);
44 if (dir_fd == -1)
45 {
46 puts (s: "cannot open directory");
47 exit (1);
48 }
49}
50
51
52static int
53do_test (void)
54{
55 /* fdopendir takes over the descriptor, make a copy. */
56 int dupfd = dup (fd: dir_fd);
57 if (dupfd == -1)
58 {
59 puts (s: "dup failed");
60 return 1;
61 }
62 if (lseek (fd: dupfd, offset: 0, SEEK_SET) != 0)
63 {
64 puts (s: "1st lseek failed");
65 return 1;
66 }
67
68 /* The directory should be empty safe the . and .. files. */
69 DIR *dir = fdopendir (fd: dupfd);
70 if (dir == NULL)
71 {
72 puts (s: "fdopendir failed");
73 return 1;
74 }
75 struct dirent64 *d;
76 while ((d = readdir64 (dirp: dir)) != NULL)
77 if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
78 {
79 printf (format: "temp directory contains file \"%s\"\n", d->d_name);
80 return 1;
81 }
82 closedir (dirp: dir);
83
84 /* Try to create a file. */
85 int fd = openat (fd: dir_fd, file: "some-file", O_CREAT|O_RDWR|O_EXCL, 0666);
86 if (fd == -1)
87 {
88 if (errno == ENOSYS)
89 {
90 puts (s: "*at functions not supported");
91 return 0;
92 }
93
94 puts (s: "file creation failed");
95 return 1;
96 }
97 write (fd, "hello", 5);
98 close (fd: fd);
99 puts (s: "file created");
100
101 /* fdopendir takes over the descriptor, make a copy. */
102 dupfd = dup (fd: dir_fd);
103 if (dupfd == -1)
104 {
105 puts (s: "2nd dup failed");
106 return 1;
107 }
108 if (lseek (fd: dupfd, offset: 0, SEEK_SET) != 0)
109 {
110 puts (s: "2nd lseek failed");
111 return 1;
112 }
113
114 /* The directory should be empty safe the . and .. files. */
115 dir = fdopendir (fd: dupfd);
116 if (dir == NULL)
117 {
118 puts (s: "2nd fdopendir failed");
119 return 1;
120 }
121 bool seen_file = false;
122 while ((d = readdir64 (dirp: dir)) != NULL)
123 if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
124 {
125 if (strcmp (d->d_name, "some-file") != 0)
126 {
127 printf (format: "temp directory contains file \"%s\"\n", d->d_name);
128 return 1;
129 }
130
131 seen_file = true;
132 }
133 closedir (dirp: dir);
134
135 if (!seen_file)
136 {
137 puts (s: "file not created in correct directory");
138 return 1;
139 }
140
141 /* Remove the file now. */
142 if (unlinkat (fd: dir_fd, name: "some-file", flag: 0) != 0)
143 {
144 puts (s: "unlinkat failed");
145 return 1;
146 }
147
148 /* We won't need dir_fd anymore after this, so use it. */
149 if (lseek (fd: dir_fd, offset: 0, SEEK_SET) != 0)
150 {
151 puts (s: "3rd lseek failed");
152 return 1;
153 }
154
155 /* The directory should be empty safe the . and .. files. */
156 dir = fdopendir (fd: dir_fd);
157 if (dir == NULL)
158 {
159 puts (s: "3rd fdopendir failed");
160 return 1;
161 }
162 while ((d = readdir64 (dirp: dir)) != NULL)
163 if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
164 {
165 if (strcmp (d->d_name, "some-file") == 0)
166 {
167 puts (s: "some-file not removed");
168 return 1;
169 }
170 else
171 {
172 printf (format: "temp directory contains file \"%s\"\n", d->d_name);
173 return 1;
174 }
175 }
176 closedir (dirp: dir);
177
178 return 0;
179}
180

source code of glibc/io/tst-unlinkat.c