Renamed files to make them more clear
Trying to figure out how to write. Might have to keep tables uncompressed for now.
This commit is contained in:
@@ -0,0 +1,107 @@
|
||||
package squashfs
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"io"
|
||||
|
||||
"github.com/CalebQ42/squashfs/internal/inode"
|
||||
)
|
||||
|
||||
//FileReader provides a io.Reader interface for files within a squashfs archive
|
||||
type fileReader struct {
|
||||
r *Reader
|
||||
data *dataReader
|
||||
in *inode.Inode
|
||||
fragmentData []byte
|
||||
fragged bool
|
||||
fragOnly bool
|
||||
read int
|
||||
FileSize int //FileSize is the total size of the given file
|
||||
}
|
||||
|
||||
var (
|
||||
//ErrPathIsNotFile returns when trying to read from a file, but the given path is NOT a file.
|
||||
errPathIsNotFile = errors.New("the given path is not a file")
|
||||
)
|
||||
|
||||
//ReadFile provides a squashfs.FileReader for the file at the given location.
|
||||
func (r *Reader) newFileReader(in *inode.Inode) (*fileReader, error) {
|
||||
var rdr fileReader
|
||||
rdr.in = in
|
||||
if in.Type != inode.FileType && in.Type != inode.ExtFileType {
|
||||
return nil, errPathIsNotFile
|
||||
}
|
||||
switch in.Type {
|
||||
case inode.FileType:
|
||||
fil := in.Info.(inode.File)
|
||||
rdr.fragged = fil.Fragmented
|
||||
rdr.fragOnly = fil.BlockStart == 0
|
||||
rdr.FileSize = int(fil.Size)
|
||||
case inode.ExtFileType:
|
||||
fil := in.Info.(inode.ExtFile)
|
||||
rdr.fragged = fil.Fragmented
|
||||
rdr.fragOnly = fil.BlockStart == 0
|
||||
rdr.FileSize = int(fil.Size)
|
||||
}
|
||||
var err error
|
||||
if rdr.fragged {
|
||||
rdr.fragmentData, err = r.getFragmentDataFromInode(in)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
if !rdr.fragOnly {
|
||||
rdr.data, err = r.newDataReaderFromInode(in)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return &rdr, nil
|
||||
}
|
||||
|
||||
func (f *fileReader) Read(p []byte) (int, error) {
|
||||
if f.fragOnly {
|
||||
n, err := bytes.NewBuffer(f.fragmentData[f.read:]).Read(p)
|
||||
f.read += n
|
||||
if err != nil {
|
||||
return n, err
|
||||
}
|
||||
return n, nil
|
||||
}
|
||||
var read int
|
||||
n, err := f.data.Read(p)
|
||||
read += n
|
||||
if f.fragged && err == io.EOF {
|
||||
if f.fragmentData == nil {
|
||||
f.fragmentData, err = f.r.getFragmentDataFromInode(f.in)
|
||||
if err != nil {
|
||||
return read, err
|
||||
}
|
||||
}
|
||||
n, err = bytes.NewBuffer(f.fragmentData).Read(p[read:])
|
||||
read += n
|
||||
if err != nil {
|
||||
return read, err
|
||||
}
|
||||
} else if err != nil {
|
||||
return read, err
|
||||
}
|
||||
return read, nil
|
||||
}
|
||||
|
||||
func (f *fileReader) WriteTo(w io.Writer) (int64, error) {
|
||||
if f.fragOnly {
|
||||
n, err := w.Write(f.fragmentData)
|
||||
return int64(n), err
|
||||
}
|
||||
if !f.fragged {
|
||||
return f.data.WriteTo(w)
|
||||
}
|
||||
n, err := f.data.WriteTo(w)
|
||||
if err != nil {
|
||||
return int64(n), err
|
||||
}
|
||||
nn, err := w.Write(f.fragmentData)
|
||||
return int64(nn) + n, err
|
||||
}
|
||||
Reference in New Issue
Block a user