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
|
# A Linux-only demo, using verify() instead of hard-coding the exact layouts
#
from cffi import FFI
ffi = FFI()
ffi.cdef("""
typedef ... DIR;
struct dirent {
unsigned char d_type; /* type of file; not supported
by all file system types */
char d_name[256]; /* filename */
...;
};
int readdir_r(DIR *dirp, struct dirent *entry, struct dirent **result);
int openat(int dirfd, const char *pathname, int flags);
DIR *fdopendir(int fd);
int closedir(DIR *dirp);
static const int DT_DIR;
""")
ffi.C = ffi.verify("""
#ifndef _ATFILE_SOURCE
# define _ATFILE_SOURCE
#endif
#ifndef _BSD_SOURCE
# define _BSD_SOURCE
#endif
#include <fcntl.h>
#include <sys/types.h>
#include <dirent.h>
""")
def walk(basefd, path):
print '{', path
dirfd = ffi.C.openat(basefd, path, 0)
if dirfd < 0:
# error in openat()
return
dir = ffi.C.fdopendir(dirfd)
dirent = ffi.new("struct dirent")
result = ffi.new("struct dirent *")
while True:
if ffi.C.readdir_r(dir, dirent, result):
# error in readdir_r()
break
if result[0] is None:
break
name = str(dirent.d_name)
print '%3d %s' % (dirent.d_type, name)
if dirent.d_type == ffi.C.DT_DIR and name != '.' and name != '..':
walk(dirfd, name)
ffi.C.closedir(dir)
print '}'
walk(-1, "/tmp")
|
