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