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