The root inode is now only initialized once.
Privated File.Reader because it really shouldn't be public.
This commit is contained in:
@@ -28,14 +28,18 @@ var (
|
|||||||
//File can be either a file or folder. When reading from a squashfs, it reads from the datablocks.
|
//File can be either a file or folder. When reading from a squashfs, it reads from the datablocks.
|
||||||
//When writing, this holds the information on WHERE the file will be placed inside the archive.
|
//When writing, this holds the information on WHERE the file will be placed inside the archive.
|
||||||
//
|
//
|
||||||
//Implements os.FileInfo and io.ReadCloser
|
//If copying data from a squashfs, the returned reader from io.Sys() implements io.WriterTo which
|
||||||
|
//will be significantly faster then calling Read directly.
|
||||||
|
//Ex: use io.Sys().(io.Reader) for io.Copy instead of using the File directly.
|
||||||
|
//
|
||||||
|
//Implements os.FileInfo and io.Reader
|
||||||
type File struct {
|
type File struct {
|
||||||
Reader io.Reader
|
reader io.Reader
|
||||||
Parent *File
|
Parent *File
|
||||||
r *Reader //Underlying reader. When writing, will probably be an os.File. When reading this is kept nil UNTIL reading to save memory.
|
r *Reader //Underlying reader. When writing, will probably be an os.File. When reading this is kept nil UNTIL reading to save memory.
|
||||||
in *inode.Inode
|
in *inode.Inode
|
||||||
name string
|
name string
|
||||||
path string
|
dir string
|
||||||
filType int //The file's type, using inode types.
|
filType int //The file's type, using inode types.
|
||||||
|
|
||||||
}
|
}
|
||||||
@@ -75,20 +79,20 @@ func (f *File) ModTime() time.Time {
|
|||||||
return time.Unix(int64(f.in.Header.ModifiedTime), 0)
|
return time.Unix(int64(f.in.Header.ModifiedTime), 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
//Sys returns the underlying reader, file.Reader. If the reader isn't initialized, it will initialize it.
|
//Sys returns the underlying reader. If the reader isn't initialized, it will initialize it.
|
||||||
//If called on something other then a file, returns nil.
|
//If called on something other then a file, returns nil.
|
||||||
func (f *File) Sys() interface{} {
|
func (f *File) Sys() interface{} {
|
||||||
if f.IsFile() {
|
if !f.IsFile() {
|
||||||
if f.Reader == nil && f.r != nil {
|
return nil
|
||||||
|
}
|
||||||
|
if f.reader == nil && f.r != nil {
|
||||||
var err error
|
var err error
|
||||||
f.Reader, err = f.r.newFileReader(f.in)
|
f.reader, err = f.r.newFileReader(f.in)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return f.Reader
|
return f.reader
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
//GetChildren returns a *squashfs.File slice of every direct child of the directory. If the File is not a directory, will return ErrNotDirectory
|
//GetChildren returns a *squashfs.File slice of every direct child of the directory. If the File is not a directory, will return ErrNotDirectory
|
||||||
@@ -112,7 +116,7 @@ func (f *File) GetChildren() (children []*File, err error) {
|
|||||||
}
|
}
|
||||||
fil.Parent = f
|
fil.Parent = f
|
||||||
if f.name != "" {
|
if f.name != "" {
|
||||||
fil.path = f.Path()
|
fil.dir = f.Path()
|
||||||
}
|
}
|
||||||
children = append(children, fil)
|
children = append(children, fil)
|
||||||
}
|
}
|
||||||
@@ -160,26 +164,23 @@ func (f *File) GetChildrenRecursively() (children []*File, err error) {
|
|||||||
//Path returns the path of the file within the archive.
|
//Path returns the path of the file within the archive.
|
||||||
func (f *File) Path() string {
|
func (f *File) Path() string {
|
||||||
if f.name == "" {
|
if f.name == "" {
|
||||||
return f.path
|
return f.dir
|
||||||
}
|
}
|
||||||
return f.path + "/" + f.name
|
return f.dir + "/" + f.name
|
||||||
}
|
}
|
||||||
|
|
||||||
//GetFileAtPath tries to return the File at the given path, relative to the file.
|
//GetFileAtPath tries to return the File at the given path, relative to the file.
|
||||||
//Returns nil if called on something other then a folder, OR if the path goes oustide the archive.
|
//Returns nil if called on something other then a folder, OR if the path goes oustide the archive.
|
||||||
//Allows wildcards supported by path.Match (namely * and ?).
|
//Allows wildcards supported by path.Match (namely * and ?) and will return the FIRST file that matches.
|
||||||
func (f *File) GetFileAtPath(dirPath string) *File {
|
func (f *File) GetFileAtPath(dirPath string) *File {
|
||||||
if dirPath == "" {
|
if dirPath == "" {
|
||||||
return f
|
return f
|
||||||
}
|
}
|
||||||
dirPath = strings.TrimSuffix(strings.TrimPrefix(dirPath, "/"), "/")
|
dirPath = path.Clean(dirPath)
|
||||||
if dirPath != "" && !f.IsDir() {
|
dirPath = strings.TrimPrefix(dirPath, "/")
|
||||||
|
if dirPath != "." && !f.IsDir() {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
for strings.HasSuffix(dirPath, "./") {
|
|
||||||
//since you can TECHNICALLY have an infinite amount of ./ and it would still be valid.
|
|
||||||
dirPath = strings.TrimPrefix(dirPath, "./")
|
|
||||||
}
|
|
||||||
split := strings.Split(dirPath, "/")
|
split := strings.Split(dirPath, "/")
|
||||||
if split[0] == ".." && f.name == "" {
|
if split[0] == ".." && f.name == "" {
|
||||||
return nil
|
return nil
|
||||||
@@ -398,18 +399,7 @@ func (f *File) ExtractWithOptions(path string, dereferenceSymlink, unbreakSymlin
|
|||||||
errs = append(errs, err)
|
errs = append(errs, err)
|
||||||
return
|
return
|
||||||
} //Since we will be reading from the file
|
} //Since we will be reading from the file
|
||||||
if f.Reader == nil && f.r != nil {
|
_, err = io.Copy(fil, f.Sys().(io.Reader))
|
||||||
f.Reader, err = f.r.newFileReader(f.in)
|
|
||||||
if err != nil {
|
|
||||||
if verbose {
|
|
||||||
fmt.Println("Error while Copying data to:", path+"/"+f.name)
|
|
||||||
fmt.Println(err)
|
|
||||||
}
|
|
||||||
errs = append(errs, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_, err = io.Copy(fil, f.Reader)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if verbose {
|
if verbose {
|
||||||
fmt.Println("Error while Copying data to:", path+"/"+f.name)
|
fmt.Println("Error while Copying data to:", path+"/"+f.name)
|
||||||
@@ -495,13 +485,13 @@ func (f *File) Read(p []byte) (int, error) {
|
|||||||
return 0, io.EOF
|
return 0, io.EOF
|
||||||
}
|
}
|
||||||
var err error
|
var err error
|
||||||
if f.Reader == nil && f.r != nil {
|
if f.reader == nil && f.r != nil {
|
||||||
f.Reader, err = f.r.newFileReader(f.in)
|
f.reader, err = f.r.newFileReader(f.in)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return f.Reader.Read(p)
|
return f.reader.Read(p)
|
||||||
}
|
}
|
||||||
|
|
||||||
//ReadDirFromInode returns a fully populated Directory from a given Inode.
|
//ReadDirFromInode returns a fully populated Directory from a given Inode.
|
||||||
|
|||||||
+1
-1
@@ -12,7 +12,7 @@ import (
|
|||||||
type fragmentEntry struct {
|
type fragmentEntry struct {
|
||||||
Start uint64
|
Start uint64
|
||||||
Size uint32
|
Size uint32
|
||||||
Unused uint32
|
// Unused uint32
|
||||||
}
|
}
|
||||||
|
|
||||||
//GetFragmentDataFromInode returns the fragment data for a given inode.
|
//GetFragmentDataFromInode returns the fragment data for a given inode.
|
||||||
|
|||||||
@@ -30,6 +30,7 @@ var (
|
|||||||
type Reader struct {
|
type Reader struct {
|
||||||
r io.ReaderAt
|
r io.ReaderAt
|
||||||
decompressor compression.Decompressor
|
decompressor compression.Decompressor
|
||||||
|
root *File
|
||||||
fragOffsets []uint64
|
fragOffsets []uint64
|
||||||
idTable []uint32
|
idTable []uint32
|
||||||
super superblock
|
super superblock
|
||||||
@@ -153,15 +154,21 @@ func (r *Reader) ModTime() time.Time {
|
|||||||
|
|
||||||
//ExtractTo tries to extract ALL files to the given path. This is the same as getting the root folder and extracting that.
|
//ExtractTo tries to extract ALL files to the given path. This is the same as getting the root folder and extracting that.
|
||||||
func (r *Reader) ExtractTo(path string) []error {
|
func (r *Reader) ExtractTo(path string) []error {
|
||||||
|
if r.root == nil {
|
||||||
root, err := r.GetRootFolder()
|
root, err := r.GetRootFolder()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return []error{err}
|
return []error{err}
|
||||||
}
|
}
|
||||||
return root.ExtractTo(path)
|
r.root = root
|
||||||
|
}
|
||||||
|
return r.root.ExtractTo(path)
|
||||||
}
|
}
|
||||||
|
|
||||||
//GetRootFolder returns a squashfs.File that references the root directory of the squashfs archive.
|
//GetRootFolder returns a squashfs.File that references the root directory of the squashfs archive.
|
||||||
func (r *Reader) GetRootFolder() (root *File, err error) {
|
func (r *Reader) GetRootFolder() (root *File, err error) {
|
||||||
|
if r.root != nil {
|
||||||
|
return r.root, nil
|
||||||
|
}
|
||||||
root = new(File)
|
root = new(File)
|
||||||
mr, err := r.newMetadataReaderFromInodeRef(r.super.RootInodeRef)
|
mr, err := r.newMetadataReaderFromInodeRef(r.super.RootInodeRef)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -171,7 +178,7 @@ func (r *Reader) GetRootFolder() (root *File, err error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
root.path = "/"
|
root.dir = "/"
|
||||||
root.filType = root.in.Type
|
root.filType = root.in.Type
|
||||||
root.r = r
|
root.r = r
|
||||||
return root, nil
|
return root, nil
|
||||||
@@ -179,20 +186,24 @@ func (r *Reader) GetRootFolder() (root *File, err error) {
|
|||||||
|
|
||||||
//GetAllFiles returns a slice of ALL files and folders contained in the squashfs.
|
//GetAllFiles returns a slice of ALL files and folders contained in the squashfs.
|
||||||
func (r *Reader) GetAllFiles() (fils []*File, err error) {
|
func (r *Reader) GetAllFiles() (fils []*File, err error) {
|
||||||
root, err := r.GetRootFolder()
|
if r.root == nil {
|
||||||
|
_, err := r.GetRootFolder()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return root.GetChildrenRecursively()
|
}
|
||||||
|
return r.root.GetChildrenRecursively()
|
||||||
}
|
}
|
||||||
|
|
||||||
//FindFile returns the first file (in the same order as Reader.GetAllFiles) that the given function returns true for. Returns nil if nothing is found.
|
//FindFile returns the first file (in the same order as Reader.GetAllFiles) that the given function returns true for. Returns nil if nothing is found.
|
||||||
func (r *Reader) FindFile(query func(*File) bool) *File {
|
func (r *Reader) FindFile(query func(*File) bool) *File {
|
||||||
root, err := r.GetRootFolder()
|
if r.root == nil {
|
||||||
|
_, err := r.GetRootFolder()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
fils, err := root.GetChildren()
|
}
|
||||||
|
fils, err := r.root.GetChildren()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user