1 /* Test for fchmodat function. */
2
3 #include <dirent.h>
4 #include <fcntl.h>
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
8 #include <unistd.h>
9 #include <sys/stat.h>
10
11 #include <support/xunistd.h>
12
13 static void prepare (void);
14 #define PREPARE(argc, argv) prepare ()
15
16 static int do_test (void);
17 #define TEST_FUNCTION do_test ()
18
19 #include "../test-skeleton.c"
20
21 static int dir_fd;
22
23 static void
24 prepare (void)
25 {
26 size_t test_dir_len = strlen (test_dir);
27 static const char dir_name[] = "/tst-fchmodat.XXXXXX";
28
29 size_t dirbuflen = test_dir_len + sizeof (dir_name);
30 char *dirbuf = malloc (dirbuflen);
31 if (dirbuf == NULL)
32 {
33 puts ("out of memory");
34 exit (1);
35 }
36
37 snprintf (dirbuf, dirbuflen, "%s%s", test_dir, dir_name);
38 if (mkdtemp (dirbuf) == NULL)
39 {
40 puts ("cannot create temporary directory");
41 exit (1);
42 }
43
44 add_temp_file (dirbuf);
45
46 dir_fd = open (dirbuf, O_RDONLY | O_DIRECTORY);
47 if (dir_fd == -1)
48 {
49 puts ("cannot open directory");
50 exit (1);
51 }
52 }
53
54
55 static int
56 do_test (void)
57 {
58 /* fdopendir takes over the descriptor, make a copy. */
59 int dupfd = dup (dir_fd);
60 if (dupfd == -1)
61 {
62 puts ("dup failed");
63 return 1;
64 }
65 if (lseek (dupfd, 0, SEEK_SET) != 0)
66 {
67 puts ("1st lseek failed");
68 return 1;
69 }
70
71 /* The directory should be empty save the . and .. files. */
72 DIR *dir = fdopendir (dupfd);
73 if (dir == NULL)
74 {
75 puts ("fdopendir failed");
76 return 1;
77 }
78 struct dirent64 *d;
79 while ((d = readdir64 (dir)) != NULL)
80 if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
81 {
82 printf ("temp directory contains file \"%s\"\n", d->d_name);
83 return 1;
84 }
85 closedir (dir);
86
87 umask (022);
88
89 /* Try to create a file. */
90 int fd = openat (dir_fd, "some-file", O_CREAT|O_RDWR|O_EXCL, 0666);
91 if (fd == -1)
92 {
93 if (errno == ENOSYS)
94 {
95 puts ("*at functions not supported");
96 return 0;
97 }
98
99 puts ("file creation failed");
100 return 1;
101 }
102 xwrite (fd, "hello", 5);
103 puts ("file created");
104
105 struct stat64 st1;
106 if (fstat64 (fd, &st1) != 0)
107 {
108 puts ("fstat64 failed");
109 return 1;
110 }
111
112 /* Before closing the file, try using this file descriptor to open
113 another file. This must fail. */
114 if (fchmodat (fd, "some-file", 0400, 0) != -1)
115 {
116 puts ("fchmodat using descriptor for normal file worked");
117 return 1;
118 }
119 if (errno != ENOTDIR)
120 {
121 puts ("\
122 error for fchmodat using descriptor for normal file not ENOTDIR ");
123 return 1;
124 }
125
126 close (fd);
127
128 if ((st1.st_mode & 0777) != 0644)
129 {
130 printf ("openat created mode %04o, not 0644\n", (st1.st_mode & 0777));
131 return 1;
132 }
133
134 if (fchmodat (dir_fd, "some-file", 0400, 0) != 0)
135 {
136 puts ("fchownat failed");
137 return 1;
138 }
139
140 struct stat64 st2;
141 if (fstatat64 (dir_fd, "some-file", &st2, 0) != 0)
142 {
143 puts ("fstatat64 failed");
144 return 1;
145 }
146
147 if ((st2.st_mode & 0777) != 0400)
148 {
149 puts ("mode change failed");
150 return 1;
151 }
152
153 if (unlinkat (dir_fd, "some-file", 0) != 0)
154 {
155 puts ("unlinkat failed");
156 return 1;
157 }
158
159 /* Create a file descriptor which is closed again right away. */
160 int dir_fd2 = dup (dir_fd);
161 if (dir_fd2 == -1)
162 {
163 puts ("dup failed");
164 return 1;
165 }
166 close (dir_fd2);
167
168 if (fchmodat (dir_fd2, "some-file", 0400, 0) != -1)
169 {
170 puts ("fchmodat using closed descriptor worked");
171 return 1;
172 }
173 if (errno != EBADF)
174 {
175 puts ("error for fchmodat using closed descriptor not EBADF ");
176 return 1;
177 }
178
179 close (dir_fd);
180
181 if (fchmodat (-1, "some-file", 0400, 0) != -1)
182 {
183 puts ("fchmodat using invalid descriptor worked");
184 return 1;
185 }
186 if (errno != EBADF)
187 {
188 puts ("error for fchmodat using invalid descriptor not EBADF ");
189 return 1;
190 }
191
192 return 0;
193 }