dir_plan9.go 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. // Copyright 2009 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package os
  5. import (
  6. "io"
  7. "syscall"
  8. )
  9. func (file *File) readdir(n int) ([]FileInfo, error) {
  10. // If this file has no dirinfo, create one.
  11. if file.dirinfo == nil {
  12. file.dirinfo = new(dirInfo)
  13. }
  14. d := file.dirinfo
  15. size := n
  16. if size <= 0 {
  17. size = 100
  18. n = -1
  19. }
  20. fi := make([]FileInfo, 0, size) // Empty with room to grow.
  21. for n != 0 {
  22. // Refill the buffer if necessary.
  23. if d.bufp >= d.nbuf {
  24. nb, err := file.Read(d.buf[:])
  25. // Update the buffer state before checking for errors.
  26. d.bufp, d.nbuf = 0, nb
  27. if err != nil {
  28. if err == io.EOF {
  29. break
  30. }
  31. return fi, &PathError{"readdir", file.name, err}
  32. }
  33. if nb < syscall.STATFIXLEN {
  34. return fi, &PathError{"readdir", file.name, syscall.ErrShortStat}
  35. }
  36. }
  37. // Get a record from the buffer.
  38. b := d.buf[d.bufp:]
  39. m := int(uint16(b[0])|uint16(b[1])<<8) + 2
  40. if m < syscall.STATFIXLEN {
  41. return fi, &PathError{"readdir", file.name, syscall.ErrShortStat}
  42. }
  43. dir, err := syscall.UnmarshalDir(b[:m])
  44. if err != nil {
  45. return fi, &PathError{"readdir", file.name, err}
  46. }
  47. fi = append(fi, fileInfoFromStat(dir))
  48. d.bufp += m
  49. n--
  50. }
  51. if n >= 0 && len(fi) == 0 {
  52. return fi, io.EOF
  53. }
  54. return fi, nil
  55. }
  56. func (file *File) readdirnames(n int) (names []string, err error) {
  57. fi, err := file.Readdir(n)
  58. names = make([]string, len(fi))
  59. for i := range fi {
  60. names[i] = fi[i].Name()
  61. }
  62. return
  63. }