1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#include <assert.h>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
static void create_file(const char *path, const char *buffer, int mode) {
int fd = open(path, O_WRONLY | O_CREAT | O_EXCL, mode);
assert(fd >= 0);
int err = write(fd, buffer, sizeof(char) * strlen(buffer));
assert(err == (sizeof(char) * strlen(buffer)));
close(fd);
}
void setup() {
mkdir("nocanread", 0111);
mkdir("foobar", 0777);
create_file("foobar/file.txt", "ride into the danger zone", 0666);
}
void cleanup() {
rmdir("nocanread");
unlink("foobar/file.txt");
rmdir("foobar");
}
void test() {
int err;
int loc;
DIR *dir;
struct dirent *ent;
struct dirent ent_r;
struct dirent *result;
// check bad opendir input
dir = opendir("noexist");
assert(!dir);
assert(errno == ENOENT);
dir = opendir("nocanread");
assert(!dir);
assert(errno == EACCES);
dir = opendir("foobar/file.txt");
assert(!dir);
assert(errno == ENOTDIR);
// check bad readdir input
dir = opendir("foobar");
closedir(dir);
ent = readdir(dir);
assert(!ent);
assert(errno == EBADF);
// check bad readdir_r input
dir = opendir("foobar");
closedir(dir);
err = readdir_r(dir, NULL, &result);
assert(err == EBADF);
//
// do a normal read with readdir
//
dir = opendir("foobar");
assert(dir);
ent = readdir(dir);
assert(!strcmp(ent->d_name, "."));
assert(ent->d_type & DT_DIR);
ent = readdir(dir);
assert(!strcmp(ent->d_name, ".."));
assert(ent->d_type & DT_DIR);
ent = readdir(dir);
assert(!strcmp(ent->d_name, "file.txt"));
assert(ent->d_type & DT_REG);
ent = readdir(dir);
assert(!ent);
// test rewinddir
rewinddir(dir);
ent = readdir(dir);
assert(!strcmp(ent->d_name, "."));
// test seek / tell
rewinddir(dir);
ent = readdir(dir);
assert(!strcmp(ent->d_name, "."));
loc = telldir(dir);
ent = readdir(dir);
assert(!strcmp(ent->d_name, ".."));
ent = readdir(dir);
assert(!strcmp(ent->d_name, "file.txt"));
seekdir(dir, loc);
ent = readdir(dir);
assert(!strcmp(ent->d_name, ".."));
//
// do a normal read with readdir_r
//
rewinddir(dir);
err = readdir_r(dir, &ent_r, &result);
assert(!err);
assert(&ent_r == result);
assert(!strcmp(ent_r.d_name, "."));
err = readdir_r(dir, &ent_r, &result);
assert(!err);
assert(&ent_r == result);
assert(!strcmp(ent_r.d_name, ".."));
err = readdir_r(dir, &ent_r, &result);
assert(!err);
assert(&ent_r == result);
assert(!strcmp(ent_r.d_name, "file.txt"));
err = readdir_r(dir, &ent_r, &result);
assert(!err);
assert(!result);
err = closedir(dir);
assert(!err);
puts("success");
}
int main() {
atexit(cleanup);
signal(SIGABRT, cleanup);
setup();
test();
return EXIT_SUCCESS;
}
|
