2009-06-04 14:33:57 -06:00
|
|
|
// Copyright 2009 The Go Authors. All rights reserved.
|
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
package os
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os";
|
|
|
|
"syscall";
|
|
|
|
"unsafe";
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
blockSize = 4096 // TODO(r): use statfs
|
|
|
|
)
|
|
|
|
|
|
|
|
// Negative count means read until EOF.
|
2009-06-29 16:13:56 -06:00
|
|
|
func (file *File) Readdirnames(count int) (names []string, err Error) {
|
2009-06-04 14:33:57 -06:00
|
|
|
// If this file has no dirinfo, create one.
|
|
|
|
if file.dirinfo == nil {
|
|
|
|
file.dirinfo = new(dirInfo);
|
|
|
|
// The buffer must be at least a block long.
|
|
|
|
// TODO(r): use fstatfs to find fs block size.
|
|
|
|
file.dirinfo.buf = make([]byte, blockSize);
|
|
|
|
}
|
|
|
|
d := file.dirinfo;
|
|
|
|
size := count;
|
|
|
|
if size < 0 {
|
|
|
|
size = 100
|
|
|
|
}
|
|
|
|
names = make([]string, 0, size); // Empty with room to grow.
|
|
|
|
for count != 0 {
|
|
|
|
// Refill the buffer if necessary
|
|
|
|
if d.bufp >= d.nbuf {
|
|
|
|
var errno int;
|
|
|
|
d.bufp = 0;
|
|
|
|
// Final argument is (basep *uintptr) and the syscall doesn't take nil.
|
|
|
|
d.nbuf, errno = syscall.Getdirentries(file.fd, d.buf, new(uintptr));
|
|
|
|
if errno != 0 {
|
|
|
|
d.nbuf = 0;
|
2009-06-25 21:24:55 -06:00
|
|
|
return names, NewSyscallError("getdirentries", errno);
|
2009-06-04 14:33:57 -06:00
|
|
|
}
|
2009-06-25 21:24:55 -06:00
|
|
|
if d.nbuf <= 0 {
|
2009-06-04 14:33:57 -06:00
|
|
|
break // EOF
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Drain the buffer
|
|
|
|
for count != 0 && d.bufp < d.nbuf {
|
|
|
|
dirent := (*syscall.Dirent)(unsafe.Pointer(&d.buf[d.bufp]));
|
|
|
|
if dirent.Reclen == 0 {
|
|
|
|
d.bufp = d.nbuf;
|
|
|
|
break
|
|
|
|
}
|
|
|
|
d.bufp += int(dirent.Reclen);
|
|
|
|
if dirent.Ino == 0 { // File absent in directory.
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
bytes := (*[len(dirent.Name)]byte)(unsafe.Pointer(&dirent.Name[0]));
|
|
|
|
var name = string(bytes[0:dirent.Namlen]);
|
|
|
|
if name == "." || name == ".." { // Useless names
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
count--;
|
|
|
|
if len(names) == cap(names) {
|
|
|
|
nnames := make([]string, len(names), 2*len(names));
|
|
|
|
for i := 0; i < len(names); i++ {
|
|
|
|
nnames[i] = names[i]
|
|
|
|
}
|
|
|
|
names = nnames;
|
|
|
|
}
|
|
|
|
names = names[0:len(names)+1];
|
|
|
|
names[len(names)-1] = name;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return names, nil
|
|
|
|
}
|