goadb/adb/dir_entries.go

120 lines
2.4 KiB
Go
Raw Normal View History

package adb
import (
"fmt"
"os"
"time"
2020-07-13 06:18:10 +00:00
"goadb/wire"
)
// DirEntry holds information about a directory entry on a device.
type DirEntry struct {
Name string
Mode os.FileMode
Size int32
ModifiedAt time.Time
}
// DirEntries iterates over directory entries.
type DirEntries struct {
scanner wire.SyncScanner
currentEntry *DirEntry
err error
}
// ReadAllDirEntries reads all the remaining directory entries into a slice,
// closes self, and returns any error.
// If err is non-nil, result will contain any entries read until the error occurred.
func (entries *DirEntries) ReadAll() (result []*DirEntry, err error) {
defer entries.Close()
for entries.Next() {
result = append(result, entries.Entry())
}
err = entries.Err()
return
}
func (entries *DirEntries) Next() bool {
if entries.err != nil {
return false
}
entry, done, err := readNextDirListEntry(entries.scanner)
if err != nil {
entries.err = err
2015-04-29 17:23:51 +00:00
entries.Close()
return false
}
entries.currentEntry = entry
if done {
2015-04-29 17:23:51 +00:00
entries.Close()
return false
}
return true
}
func (entries *DirEntries) Entry() *DirEntry {
return entries.currentEntry
}
func (entries *DirEntries) Err() error {
return entries.err
}
2015-04-29 17:23:51 +00:00
// Close closes the connection to the adb.
// Next() will call Close() before returning false.
func (entries *DirEntries) Close() error {
return entries.scanner.Close()
}
func readNextDirListEntry(s wire.SyncScanner) (entry *DirEntry, done bool, err error) {
status, err := s.ReadStatus("dir-entry")
if err != nil {
return
}
if status == "DONE" {
done = true
return
} else if status != "DENT" {
err = fmt.Errorf("error reading dir entries: expected dir entry ID 'DENT', but got '%s'", status)
return
}
mode, err := s.ReadFileMode()
if err != nil {
2015-04-29 17:23:51 +00:00
err = fmt.Errorf("error reading dir entries: error reading file mode: %v", err)
return
}
size, err := s.ReadInt32()
if err != nil {
2015-04-29 17:23:51 +00:00
err = fmt.Errorf("error reading dir entries: error reading file size: %v", err)
return
}
mtime, err := s.ReadTime()
if err != nil {
2015-04-29 17:23:51 +00:00
err = fmt.Errorf("error reading dir entries: error reading file time: %v", err)
return
}
name, err := s.ReadString()
if err != nil {
2015-04-29 17:23:51 +00:00
err = fmt.Errorf("error reading dir entries: error reading file name: %v", err)
return
}
done = false
entry = &DirEntry{
Name: name,
Mode: mode,
Size: size,
ModifiedAt: mtime,
}
return
}