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 |
21 | prepare (void) |
22 | { |
23 | size_t test_dir_len = strlen (test_dir); |
24 | static const char dir_name[] = "/tst-openat.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 | |
52 | static int |
53 | do_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 | |
99 | /* Before closing the file, try using this file descriptor to open |
100 | another file. This must fail. */ |
101 | int fd2 = openat (fd: fd, file: "should-not-work" , O_CREAT|O_RDWR, 0666); |
102 | if (fd2 != -1) |
103 | { |
104 | puts (s: "openat using descriptor for normal file worked" ); |
105 | return 1; |
106 | } |
107 | if (errno != ENOTDIR) |
108 | { |
109 | puts (s: "error for openat using descriptor for normal file not ENOTDIR " ); |
110 | return 1; |
111 | } |
112 | |
113 | close (fd: fd); |
114 | puts (s: "file created" ); |
115 | |
116 | /* fdopendir takes over the descriptor, make a copy. */ |
117 | dupfd = dup (fd: dir_fd); |
118 | if (dupfd == -1) |
119 | { |
120 | puts (s: "dup failed" ); |
121 | return 1; |
122 | } |
123 | if (lseek (fd: dupfd, offset: 0, SEEK_SET) != 0) |
124 | { |
125 | puts (s: "2nd lseek failed" ); |
126 | return 1; |
127 | } |
128 | |
129 | /* The directory should be empty safe the . and .. files. */ |
130 | dir = fdopendir (fd: dupfd); |
131 | if (dir == NULL) |
132 | { |
133 | puts (s: "fdopendir failed" ); |
134 | return 1; |
135 | } |
136 | bool seen_file = false; |
137 | while ((d = readdir64 (dirp: dir)) != NULL) |
138 | if (strcmp (d->d_name, "." ) != 0 && strcmp (d->d_name, ".." ) != 0) |
139 | { |
140 | if (strcmp (d->d_name, "some-file" ) != 0) |
141 | { |
142 | printf (format: "temp directory contains file \"%s\"\n" , d->d_name); |
143 | return 1; |
144 | } |
145 | |
146 | seen_file = true; |
147 | } |
148 | closedir (dirp: dir); |
149 | |
150 | if (!seen_file) |
151 | { |
152 | puts (s: "file not created in correct directory" ); |
153 | return 1; |
154 | } |
155 | |
156 | int cwdfd = open (file: "." , O_RDONLY | O_DIRECTORY); |
157 | if (cwdfd == -1) |
158 | { |
159 | puts (s: "cannot get descriptor for cwd" ); |
160 | return 1; |
161 | } |
162 | |
163 | if (fchdir (fd: dir_fd) != 0) |
164 | { |
165 | puts (s: "1st fchdir failed" ); |
166 | return 1; |
167 | } |
168 | |
169 | if (unlink (name: "some-file" ) != 0) |
170 | { |
171 | puts (s: "unlink failed" ); |
172 | return 1; |
173 | } |
174 | |
175 | if (fchdir (fd: cwdfd) != 0) |
176 | { |
177 | puts (s: "2nd fchdir failed" ); |
178 | return 1; |
179 | } |
180 | |
181 | close (fd: dir_fd); |
182 | close (fd: cwdfd); |
183 | |
184 | /* With the file descriptor closed the next call must fail. */ |
185 | fd = openat (fd: dir_fd, file: "some-file" , O_CREAT|O_RDWR|O_EXCL, 0666); |
186 | if (fd != -1) |
187 | { |
188 | puts (s: "openat using closed descriptor succeeded" ); |
189 | return 1; |
190 | } |
191 | if (errno != EBADF) |
192 | { |
193 | puts (s: "openat using closed descriptor did not set EBADF" ); |
194 | return 1; |
195 | } |
196 | |
197 | fd = openat (fd: -1, file: "some-file" , O_CREAT|O_RDWR|O_EXCL, 0666); |
198 | if (fd != -1) |
199 | { |
200 | puts (s: "openat using -1 descriptor succeeded" ); |
201 | return 1; |
202 | } |
203 | if (errno != EBADF) |
204 | { |
205 | puts (s: "openat using -1 descriptor did not set EBADF" ); |
206 | return 1; |
207 | } |
208 | |
209 | return 0; |
210 | } |
211 | |