I CAN READ A DIRECTORY NOW

I've only tested it with the root dir, but should work for all directories.

internal/directory/directory.go

@@ -1,10 +1,9 @@
 package directory
 
 import (
+	"bytes"
 	"encoding/binary"
-	"fmt"
 	"io"
-	"math"
 )
 
 //Header is the header for a directory in the directory table
@@ -25,64 +24,69 @@ type EntryInit struct {
 //Entry is an entry in a directory.
 type Entry struct {
 	Init EntryInit
-	Name []byte
+	Name string
 }
 
 //NewEntry creates a new directory entry
-func NewEntry(rdr io.Reader) (*Entry, error) {
+func NewEntry(rdr io.Reader) (Entry, error) {
 	var entry Entry
 	err := binary.Read(rdr, binary.LittleEndian, &entry.Init)
 	if err != nil {
-		return nil, err
+		return Entry{}, err
 	}
-	entry.Name = make([]byte, entry.Init.NameSize+1)
-	err = binary.Read(rdr, binary.LittleEndian, &entry.Name)
+	tmp := make([]byte, entry.Init.NameSize+1)
+	err = binary.Read(rdr, binary.LittleEndian, &tmp)
 	if err != nil {
-		return nil, err
+		return Entry{}, err
 	}
-	return &entry, err
+	entry.Name = string(tmp)
+	return entry, err
 }
 
 //Directory is an entry in the directory table of a squashfs.
 //Will only have multiple headers if there are more then 256 entries
 type Directory struct {
 	Headers []Header
-	Entries []*Entry
+	Entries []Entry
 }
 
 //NewDirectory reads the directory from rdr
-func NewDirectory(rdr io.Reader) (*Directory, error) {
+func NewDirectory(base io.Reader, size uint16) (*Directory, error) {
 	var dir Directory
-	var hdr Header
-	err := binary.Read(rdr, binary.LittleEndian, &hdr)
-	if err != nil {
-		return nil, err
-	}
-	hdr.Count++
-	headers := hdr.Count / 256
-	if hdr.Count%256 > 0 {
-		headers++
-	}
-	fmt.Println("headers", headers)
-	fmt.Println("headers ceil", math.Ceil(float64(hdr.Count)/256))
-	headersRead := 1
-	dir.Headers = make([]Header, headers)
-	dir.Headers[0] = hdr
-	for i := uint32(0); i < hdr.Count; i++ {
-		if i != 0 && i%256 == 0 {
-			var newHdr Header
-			err = binary.Read(rdr, binary.LittleEndian, &newHdr)
-			if err != nil {
-				return &dir, err
+	var err error
+	tmp := make([]byte, size)
+	base.Read(tmp)
+	rdr := bytes.NewBuffer(tmp)
+	for {
+		var hdr Header
+		err = binary.Read(rdr, binary.LittleEndian, &hdr)
+		if err == io.ErrUnexpectedEOF {
+			break
+		} else if err != nil {
+			return nil, err
+		}
+		hdr.Count++
+		headers := hdr.Count / 256
+		if hdr.Count%256 > 0 {
+			headers++
+		}
+		dir.Headers = append(dir.Headers, hdr)
+		for i := uint32(0); i < hdr.Count; i++ {
+			if i != 0 && i%256 == 0 {
+				var newHdr Header
+				err = binary.Read(rdr, binary.LittleEndian, &newHdr)
+				if err != nil {
+					return nil, err
+				}
+				dir.Headers = append(dir.Headers, newHdr)
 			}
-			dir.Headers[headersRead] = newHdr
-			headersRead++
+			var ent Entry
+			ent, err = NewEntry(rdr)
+			if err != nil {
+				return nil, err
+			}
+			dir.Entries = append(dir.Entries, ent)
 		}
-		ent, err := NewEntry(rdr)
-		if err != nil {
-			return &dir, err
-		}
-		dir.Entries = append(dir.Entries, ent)
 	}
 	return &dir, nil
 }