readdir.py 827 B

1234567891011121314151617181920212223242526272829303132333435
  1. # A Linux-only demo
  2. #
  3. import sys
  4. if not sys.platform.startswith('linux'):
  5. raise Exception("Linux-only demo")
  6. from _readdir import ffi
  7. lib = ffi.dlopen(None)
  8. def walk(basefd, path):
  9. print '{', path
  10. dirfd = lib.openat(basefd, path, 0)
  11. if dirfd < 0:
  12. # error in openat()
  13. return
  14. dir = lib.fdopendir(dirfd)
  15. dirent = ffi.new("struct dirent *")
  16. result = ffi.new("struct dirent **")
  17. while True:
  18. if lib.readdir_r(dir, dirent, result):
  19. # error in readdir_r()
  20. break
  21. if result[0] == ffi.NULL:
  22. break
  23. name = ffi.string(dirent.d_name)
  24. print '%3d %s' % (dirent.d_type, name)
  25. if dirent.d_type == 4 and name != '.' and name != '..':
  26. walk(dirfd, name)
  27. lib.closedir(dir)
  28. print '}'
  29. walk(-1, "/tmp")