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

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