Compare commits
30 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 87f7533a17 | |||
| 7e1a584e8f | |||
| 942e0f770f | |||
| 7d16990277 | |||
| d2c72f9464 | |||
| 2ba4551fb9 | |||
| 6931075e7e | |||
| 55a25c9d45 | |||
| 94b45c8402 | |||
| 01de43a5ae | |||
| 5b29f4d029 | |||
| 6c7e926649 | |||
| 72d85d7810 | |||
| 67df5f40c6 | |||
| 1ae5593e6c | |||
| 653c4a167b | |||
| 9fe17650b8 | |||
| e9e967f085 | |||
| 187da99dd6 | |||
| 75d2a29319 | |||
| ce2e45ceec | |||
| 089ef53c8c | |||
| 658e5c9e0b | |||
| f2d86aff96 | |||
| f61237a1f0 | |||
| 820e06e792 | |||
| 4f8f5f6928 | |||
| 1b5078c7bd | |||
| 56fdba2f28 | |||
| ffbf4ebc64 |
@@ -1 +1,2 @@
|
||||
testing
|
||||
/go-unsquashfs
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
# squashfs (WIP)
|
||||
# squashfs
|
||||
|
||||
[](https://pkg.go.dev/github.com/CalebQ42/squashfs) [](https://goreportcard.com/report/github.com/CalebQ42/squashfs)
|
||||
|
||||
A PURE Go library to read and write squashfs.
|
||||
A PURE Go library to read squashfs. There is currently no plans to add archive creation support as it will almost always be better to just call `mksquashfs`. I could see some possible use cases, but probably won't spend time on it unless it's requested (open a discussion fi you want this feature).
|
||||
|
||||
Currently has support for reading squashfs files and extracting files and folders.
|
||||
|
||||
@@ -14,9 +14,18 @@ Thanks also to [distri's squashfs library](https://github.com/distr1/distri/tree
|
||||
## Limitations
|
||||
|
||||
* No Xattr parsing. This is simply because I haven't done any research on it and how to apply these in a pure go way.
|
||||
* Socket files are not extracted.
|
||||
* From my research, it seems like a socket file would be useless if it could be created.
|
||||
* Fifo files are ignored on `darwin`
|
||||
|
||||
## Performance
|
||||
## Issues
|
||||
|
||||
This library, decompressing the Firefox AppImage and using go tests, takes about twice as long as `unsquashfs` on my quad core laptop. (~1 second with the library and about half a second with `unsquashfs`).
|
||||
* Significantly slower then `unsquashfs` when extracting folders (about 5 ~ 7 times slower on a ~100MB archive using zstd compression)
|
||||
* This seems to be related to above along with the general optimization of `unsquashfs` and it's compression libraries.
|
||||
* The larger the file's tree, the slower the extraction will be. Arch Linux's Live USB's airootfs.sfs takes ~35x longer for a full extraction.
|
||||
|
||||
**My recents tests have shown the Firefox AppImage might be an outlier and this library might be considerably slower (4x ~ 6x time slower then `unsquashfs`)**
|
||||
## Recommendations on Usage
|
||||
|
||||
Due to the above performance consideration, this library should only be used to access files within the archive without extraction, or to mount it via Fuse.
|
||||
|
||||
* Neither of these use cases are largely effected by the issue above.
|
||||
|
||||
@@ -0,0 +1,37 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/CalebQ42/squashfs"
|
||||
)
|
||||
|
||||
func main() {
|
||||
verbose := flag.Bool("v", false, "Verbose")
|
||||
ignore := flag.Bool("ip", false, "Ignore Permissions and extract all files/folders with 0755")
|
||||
flag.Parse()
|
||||
if len(flag.Args()) < 2 {
|
||||
fmt.Println("Please provide a file name and extraction path")
|
||||
os.Exit(0)
|
||||
}
|
||||
f, err := os.Open(flag.Arg(0))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
r, err := squashfs.NewReader(f)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
op := squashfs.DefaultOptions()
|
||||
op.Verbose = *verbose
|
||||
op.IgnorePerm = *ignore
|
||||
n := time.Now()
|
||||
err = r.ExtractWithOptions(flag.Arg(1), op)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fmt.Println("Took:", time.Since(n))
|
||||
}
|
||||
@@ -0,0 +1,150 @@
|
||||
package squashfs
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
|
||||
"github.com/CalebQ42/squashfs/internal/inode"
|
||||
"github.com/seaweedfs/fuse"
|
||||
"github.com/seaweedfs/fuse/fs"
|
||||
)
|
||||
|
||||
// Mounts the archive to the given mountpoint using fuse2. Non-blocking.
|
||||
// If Unmount does not get called, the mount point must be unmounted using umount before the directory can be used again.
|
||||
func (r *Reader) MountFuse2(mountpoint string) (err error) {
|
||||
if r.con != nil {
|
||||
return errors.New("squashfs archive already mounted")
|
||||
}
|
||||
r.con2, err = fuse.Mount(mountpoint, fuse.ReadOnly())
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
<-r.con2.Ready
|
||||
r.mount2Done = make(chan struct{})
|
||||
go func() {
|
||||
fs.Serve(r.con2, squashFuse2{r: r})
|
||||
close(r.mount2Done)
|
||||
}()
|
||||
return
|
||||
}
|
||||
|
||||
// Blocks until the mount ends.
|
||||
// Fuse2 version.
|
||||
func (r *Reader) MountWaitFuse2() {
|
||||
if r.mount2Done != nil {
|
||||
<-r.mount2Done
|
||||
}
|
||||
}
|
||||
|
||||
// Unmounts the archive.
|
||||
// Fuse2 version.
|
||||
func (r *Reader) UnmountFuse2() error {
|
||||
if r.con != nil {
|
||||
defer func() { r.con = nil }()
|
||||
return r.con.Close()
|
||||
}
|
||||
return errors.New("squashfs archive is not mounted")
|
||||
}
|
||||
|
||||
type squashFuse2 struct {
|
||||
r *Reader
|
||||
}
|
||||
|
||||
func (s squashFuse2) Root() (fs.Node, error) {
|
||||
return fileNode2{File: s.r.FS.File}, nil
|
||||
}
|
||||
|
||||
type fileNode2 struct {
|
||||
*File
|
||||
}
|
||||
|
||||
func (f fileNode2) Attr(ctx context.Context, attr *fuse.Attr) error {
|
||||
attr.Blocks = f.r.s.Size / 512
|
||||
if f.r.s.Size%512 > 0 {
|
||||
attr.Blocks++
|
||||
}
|
||||
attr.Gid = f.r.ids[f.i.GidInd]
|
||||
attr.Inode = uint64(f.i.Num)
|
||||
attr.Mode = f.i.Mode()
|
||||
attr.Nlink = f.i.LinkCount()
|
||||
attr.Size = f.i.Size()
|
||||
attr.Uid = f.r.ids[f.i.UidInd]
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f fileNode2) Id() uint64 {
|
||||
return uint64(f.i.Num)
|
||||
}
|
||||
|
||||
func (f fileNode2) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) {
|
||||
return f.SymlinkPath(), nil
|
||||
}
|
||||
|
||||
func (f fileNode2) Lookup(ctx context.Context, name string) (fs.Node, error) {
|
||||
asFS, err := f.FS()
|
||||
if err != nil {
|
||||
return nil, fuse.ENOTDIR
|
||||
}
|
||||
ret, err := asFS.OpenFile(name)
|
||||
if err != nil {
|
||||
return nil, fuse.ENOENT
|
||||
}
|
||||
return fileNode2{File: ret}, nil
|
||||
}
|
||||
|
||||
func (f fileNode2) ReadAll(ctx context.Context) ([]byte, error) {
|
||||
if f.IsRegular() {
|
||||
var buf bytes.Buffer
|
||||
_, err := f.WriteTo(&buf)
|
||||
return buf.Bytes(), err
|
||||
}
|
||||
return nil, ENODATA
|
||||
}
|
||||
|
||||
func (f fileNode2) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error {
|
||||
if f.IsRegular() {
|
||||
buf := make([]byte, req.Size)
|
||||
n, err := f.File.ReadAt(buf, req.Offset)
|
||||
if err == io.EOF {
|
||||
resp.Data = buf[:n]
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return ENODATA
|
||||
}
|
||||
|
||||
func (f fileNode2) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error) {
|
||||
asFS, err := f.FS()
|
||||
if err != nil {
|
||||
return nil, fuse.ENOTDIR
|
||||
}
|
||||
var t fuse.DirentType
|
||||
for i := range asFS.e {
|
||||
switch asFS.e[i].Type {
|
||||
case inode.Fil:
|
||||
t = fuse.DT_File
|
||||
case inode.Dir:
|
||||
t = fuse.DT_Dir
|
||||
case inode.Block:
|
||||
t = fuse.DT_Block
|
||||
case inode.Sym:
|
||||
t = fuse.DT_Link
|
||||
case inode.Char:
|
||||
t = fuse.DT_Char
|
||||
case inode.Fifo:
|
||||
t = fuse.DT_FIFO
|
||||
case inode.Sock:
|
||||
t = fuse.DT_Socket
|
||||
default:
|
||||
t = fuse.DT_Unknown
|
||||
}
|
||||
out = append(out, fuse.Dirent{
|
||||
Inode: uint64(asFS.e[i].Num),
|
||||
Type: t,
|
||||
Name: asFS.e[i].Name,
|
||||
})
|
||||
}
|
||||
return
|
||||
}
|
||||
@@ -0,0 +1,148 @@
|
||||
package squashfs
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
|
||||
"github.com/CalebQ42/fuse"
|
||||
"github.com/CalebQ42/fuse/fs"
|
||||
"github.com/CalebQ42/squashfs/internal/inode"
|
||||
)
|
||||
|
||||
// Mounts the archive to the given mountpoint using fuse3. Non-blocking.
|
||||
// If Unmount does not get called, the mount point must be unmounted using umount before the directory can be used again.
|
||||
func (r *Reader) Mount(mountpoint string) (err error) {
|
||||
if r.con != nil {
|
||||
return errors.New("squashfs archive already mounted")
|
||||
}
|
||||
r.con, err = fuse.Mount(mountpoint, fuse.ReadOnly())
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
<-r.con.Ready
|
||||
r.mountDone = make(chan struct{})
|
||||
go func() {
|
||||
fs.Serve(r.con, squashFuse{r: r})
|
||||
close(r.mountDone)
|
||||
}()
|
||||
return
|
||||
}
|
||||
|
||||
// Blocks until the mount ends.
|
||||
func (r *Reader) MountWait() {
|
||||
if r.mountDone != nil {
|
||||
<-r.mountDone
|
||||
}
|
||||
}
|
||||
|
||||
// Unmounts the archive.
|
||||
func (r *Reader) Unmount() error {
|
||||
if r.con != nil {
|
||||
defer func() { r.con = nil }()
|
||||
return r.con.Close()
|
||||
}
|
||||
return errors.New("squashfs archive is not mounted")
|
||||
}
|
||||
|
||||
type squashFuse struct {
|
||||
r *Reader
|
||||
}
|
||||
|
||||
func (s squashFuse) Root() (fs.Node, error) {
|
||||
return fileNode{File: s.r.FS.File}, nil
|
||||
}
|
||||
|
||||
type fileNode struct {
|
||||
*File
|
||||
}
|
||||
|
||||
func (f fileNode) Attr(ctx context.Context, attr *fuse.Attr) error {
|
||||
attr.Blocks = f.r.s.Size / 512
|
||||
if f.r.s.Size%512 > 0 {
|
||||
attr.Blocks++
|
||||
}
|
||||
attr.Gid = f.r.ids[f.i.GidInd]
|
||||
attr.Inode = uint64(f.i.Num)
|
||||
attr.Mode = f.i.Mode()
|
||||
attr.Nlink = f.i.LinkCount()
|
||||
attr.Size = f.i.Size()
|
||||
attr.Uid = f.r.ids[f.i.UidInd]
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f fileNode) Id() uint64 {
|
||||
return uint64(f.i.Num)
|
||||
}
|
||||
|
||||
func (f fileNode) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) {
|
||||
return f.SymlinkPath(), nil
|
||||
}
|
||||
|
||||
func (f fileNode) Lookup(ctx context.Context, name string) (fs.Node, error) {
|
||||
asFS, err := f.FS()
|
||||
if err != nil {
|
||||
return nil, fuse.ENOTDIR
|
||||
}
|
||||
ret, err := asFS.OpenFile(name)
|
||||
if err != nil {
|
||||
return nil, fuse.ENOENT
|
||||
}
|
||||
return fileNode{File: ret}, nil
|
||||
}
|
||||
|
||||
func (f fileNode) ReadAll(ctx context.Context) ([]byte, error) {
|
||||
if f.IsRegular() {
|
||||
var buf bytes.Buffer
|
||||
_, err := f.WriteTo(&buf)
|
||||
return buf.Bytes(), err
|
||||
}
|
||||
return nil, ENODATA
|
||||
}
|
||||
|
||||
func (f fileNode) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error {
|
||||
if f.IsRegular() {
|
||||
buf := make([]byte, req.Size)
|
||||
n, err := f.File.ReadAt(buf, req.Offset)
|
||||
if err == io.EOF {
|
||||
resp.Data = buf[:n]
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return ENODATA
|
||||
}
|
||||
|
||||
func (f fileNode) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error) {
|
||||
asFS, err := f.FS()
|
||||
if err != nil {
|
||||
return nil, fuse.ENOTDIR
|
||||
}
|
||||
var t fuse.DirentType
|
||||
for i := range asFS.e {
|
||||
switch asFS.e[i].Type {
|
||||
case inode.Fil:
|
||||
t = fuse.DT_File
|
||||
case inode.Dir:
|
||||
t = fuse.DT_Dir
|
||||
case inode.Block:
|
||||
t = fuse.DT_Block
|
||||
case inode.Sym:
|
||||
t = fuse.DT_Link
|
||||
case inode.Char:
|
||||
t = fuse.DT_Char
|
||||
case inode.Fifo:
|
||||
t = fuse.DT_FIFO
|
||||
case inode.Sock:
|
||||
t = fuse.DT_Socket
|
||||
default:
|
||||
t = fuse.DT_Unknown
|
||||
}
|
||||
out = append(out, fuse.Dirent{
|
||||
Inode: uint64(asFS.e[i].Num),
|
||||
Type: t,
|
||||
Name: asFS.e[i].Name,
|
||||
})
|
||||
}
|
||||
return
|
||||
}
|
||||
@@ -0,0 +1,7 @@
|
||||
package squashfs
|
||||
|
||||
import (
|
||||
"golang.org/x/sys/unix"
|
||||
)
|
||||
|
||||
var ENODATA = unix.Errno(unix.ENODATA)
|
||||
@@ -0,0 +1,5 @@
|
||||
package squashfs
|
||||
|
||||
import "github.com/CalebQ42/fuse"
|
||||
|
||||
var ENODATA = fuse.ENODATA
|
||||
@@ -0,0 +1,3 @@
|
||||
package squashfs
|
||||
|
||||
var ENODATA = windows.Errno(windows.ENODATA)
|
||||
@@ -1,11 +1,14 @@
|
||||
module github.com/CalebQ42/squashfs
|
||||
|
||||
go 1.19
|
||||
go 1.21
|
||||
|
||||
require (
|
||||
github.com/klauspost/compress v1.15.9
|
||||
github.com/pierrec/lz4/v4 v4.1.15
|
||||
github.com/CalebQ42/fuse v0.1.0
|
||||
github.com/klauspost/compress v1.16.7
|
||||
github.com/pierrec/lz4/v4 v4.1.18
|
||||
github.com/rasky/go-lzo v0.0.0-20200203143853-96a758eda86e
|
||||
github.com/seaweedfs/fuse v1.2.2
|
||||
github.com/therootcompany/xz v1.0.1
|
||||
github.com/ulikunitz/xz v0.5.10
|
||||
github.com/ulikunitz/xz v0.5.11
|
||||
golang.org/x/sys v0.11.0
|
||||
)
|
||||
|
||||
@@ -1,10 +1,16 @@
|
||||
github.com/klauspost/compress v1.15.9 h1:wKRjX6JRtDdrE9qwa4b/Cip7ACOshUI4smpCQanqjSY=
|
||||
github.com/klauspost/compress v1.15.9/go.mod h1:PhcZ0MbTNciWF3rruxRgKxI5NkcHHrHUDtV4Yw2GlzU=
|
||||
github.com/pierrec/lz4/v4 v4.1.15 h1:MO0/ucJhngq7299dKLwIMtgTfbkoSPF6AoMYDd8Q4q0=
|
||||
github.com/pierrec/lz4/v4 v4.1.15/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
|
||||
github.com/CalebQ42/fuse v0.1.0 h1:KLCNjun7zcd2kBNVFfH+SWJyhuwJdE0nhw5/q8K8HGQ=
|
||||
github.com/CalebQ42/fuse v0.1.0/go.mod h1:pJpoKG03HJKVhsp8o0YQYqmfbFsr3Eowt90yQGQVO+4=
|
||||
github.com/klauspost/compress v1.16.7 h1:2mk3MPGNzKyxErAw8YaohYh69+pa4sIQSC0fPGCFR9I=
|
||||
github.com/klauspost/compress v1.16.7/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||
github.com/pierrec/lz4/v4 v4.1.18 h1:xaKrnTkyoqfh1YItXl56+6KJNVYWlEEPuAQW9xsplYQ=
|
||||
github.com/pierrec/lz4/v4 v4.1.18/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
|
||||
github.com/rasky/go-lzo v0.0.0-20200203143853-96a758eda86e h1:dCWirM5F3wMY+cmRda/B1BiPsFtmzXqV9b0hLWtVBMs=
|
||||
github.com/rasky/go-lzo v0.0.0-20200203143853-96a758eda86e/go.mod h1:9leZcVcItj6m9/CfHY5Em/iBrCz7js8LcRQGTKEEv2M=
|
||||
github.com/seaweedfs/fuse v1.2.2 h1:01l8OjIdyATRNqVc/gDPgFobuC8ubQF3hRKOPColROw=
|
||||
github.com/seaweedfs/fuse v1.2.2/go.mod h1:iwbDQv5BZACY54r6AO/6xsLNuMaYcBKSkLTZVfmK594=
|
||||
github.com/therootcompany/xz v1.0.1 h1:CmOtsn1CbtmyYiusbfmhmkpAAETj0wBIH6kCYaX+xzw=
|
||||
github.com/therootcompany/xz v1.0.1/go.mod h1:3K3UH1yCKgBneZYhuQUvJ9HPD19UEXEI0BWbMn8qNMY=
|
||||
github.com/ulikunitz/xz v0.5.10 h1:t92gobL9l3HE202wg3rlk19F6X+JOxl9BBrCCMYEYd8=
|
||||
github.com/ulikunitz/xz v0.5.10/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=
|
||||
github.com/ulikunitz/xz v0.5.11 h1:kpFauv27b6ynzBNT/Xy+1k+fK4WswhN/6PN5WhFAGw8=
|
||||
github.com/ulikunitz/xz v0.5.11/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=
|
||||
golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM=
|
||||
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
|
||||
@@ -14,15 +14,17 @@ type FullReader struct {
|
||||
sizes []uint32
|
||||
blockSize uint32
|
||||
start uint64
|
||||
fileSize uint64
|
||||
}
|
||||
|
||||
func NewFullReader(r io.ReaderAt, start uint64, d decompress.Decompressor, blockSizes []uint32, blockSize uint32) *FullReader {
|
||||
func NewFullReader(r io.ReaderAt, start uint64, d decompress.Decompressor, blockSizes []uint32, blockSize uint32, fileSize uint64) *FullReader {
|
||||
return &FullReader{
|
||||
r: r,
|
||||
start: start,
|
||||
blockSize: blockSize,
|
||||
sizes: blockSizes,
|
||||
d: d,
|
||||
fileSize: fileSize,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -43,21 +45,24 @@ func (r FullReader) process(index int, offset int64, out chan outDat) {
|
||||
var rdr io.ReadCloser
|
||||
size := realSize(r.sizes[index])
|
||||
if size == 0 {
|
||||
outSize := r.blockSize
|
||||
if r.fileSize < uint64(r.blockSize) {
|
||||
outSize = uint32(r.fileSize)
|
||||
}
|
||||
out <- outDat{
|
||||
i: index,
|
||||
err: nil,
|
||||
data: make([]byte, r.blockSize),
|
||||
data: make([]byte, outSize),
|
||||
}
|
||||
return
|
||||
}
|
||||
// rdr := io.LimitReader(toreader.NewReader(r.r, offset), int64(size))
|
||||
if size == r.sizes[index] {
|
||||
//Special workaround for zstd for increased performancce.
|
||||
if zstd, ok := r.d.(*decompress.Zstd); ok {
|
||||
if dec, ok := r.d.(decompress.Decoder); ok {
|
||||
dat = make([]byte, size)
|
||||
_, err = r.r.ReadAt(dat, offset)
|
||||
if err == nil {
|
||||
dat, err = zstd.Decode(dat)
|
||||
dat, err = dec.Decode(dat)
|
||||
}
|
||||
} else {
|
||||
rdr, err = r.d.Reader(io.LimitReader(toreader.NewReader(r.r, offset), int64(size)))
|
||||
@@ -79,6 +84,91 @@ func (r FullReader) process(index int, offset int64, out chan outDat) {
|
||||
}
|
||||
}
|
||||
|
||||
func (r FullReader) ReadAt(p []byte, off int64) (n int, err error) {
|
||||
out := make(chan outDat, len(r.sizes))
|
||||
offset := r.start
|
||||
num := len(r.sizes)
|
||||
start := off / int64(r.blockSize)
|
||||
end := len(p) / int(r.blockSize)
|
||||
if end%int(r.blockSize) > 0 {
|
||||
end++
|
||||
}
|
||||
if end > len(r.sizes) {
|
||||
if r.fragRdr != nil {
|
||||
end = len(r.sizes)
|
||||
} else {
|
||||
end = len(r.sizes) + 1
|
||||
}
|
||||
}
|
||||
for i := 0; i < num; i++ {
|
||||
if i < int(start) || i > end {
|
||||
offset += uint64(realSize(r.sizes[i]))
|
||||
continue
|
||||
}
|
||||
if i == num-1 && r.fragRdr != nil {
|
||||
go func() {
|
||||
rdr, e := r.fragRdr()
|
||||
if e != nil {
|
||||
out <- outDat{
|
||||
i: num - 1,
|
||||
err: e,
|
||||
}
|
||||
return
|
||||
}
|
||||
dat, e := io.ReadAll(rdr)
|
||||
out <- outDat{
|
||||
i: num - 1,
|
||||
err: e,
|
||||
data: dat,
|
||||
}
|
||||
if clr, ok := rdr.(io.Closer); ok {
|
||||
clr.Close()
|
||||
}
|
||||
}()
|
||||
continue
|
||||
}
|
||||
go r.process(i, int64(offset), out)
|
||||
offset += uint64(realSize(r.sizes[i]))
|
||||
}
|
||||
cache := make(map[int]outDat)
|
||||
for cur := start; cur < int64(end); {
|
||||
dat := <-out
|
||||
if dat.err != nil {
|
||||
err = dat.err
|
||||
return
|
||||
}
|
||||
if dat.i != int(cur) {
|
||||
cache[dat.i] = dat
|
||||
continue
|
||||
}
|
||||
if cur == start {
|
||||
dat.data = dat.data[off%int64(r.blockSize):]
|
||||
}
|
||||
for i := range dat.data {
|
||||
p[n+i] = dat.data[i]
|
||||
}
|
||||
n += len(dat.data)
|
||||
cur++
|
||||
var ok bool
|
||||
for {
|
||||
dat, ok = cache[int(cur)]
|
||||
if !ok {
|
||||
break
|
||||
}
|
||||
for i := range dat.data {
|
||||
p[n+i] = dat.data[i]
|
||||
}
|
||||
n += len(dat.data)
|
||||
cur++
|
||||
delete(cache, int(cur))
|
||||
}
|
||||
}
|
||||
if n < len(p) {
|
||||
err = io.EOF
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (r FullReader) WriteTo(w io.Writer) (n int64, err error) {
|
||||
out := make(chan outDat, len(r.sizes))
|
||||
offset := r.start
|
||||
|
||||
@@ -53,14 +53,14 @@ func (r *Reader) advance() (err error) {
|
||||
} else {
|
||||
r.cur = io.LimitReader(r.master, int64(size))
|
||||
if size == r.blockSizes[0] {
|
||||
if r.d.Resetable() {
|
||||
if rs, ok := r.d.(decompress.Resetable); ok {
|
||||
if r.comRdr == nil {
|
||||
r.cur, err = r.d.Reader(r.cur)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
err = r.d.Reset(r.comRdr, r.cur)
|
||||
err = rs.Reset(r.comRdr, r.cur)
|
||||
r.cur = r.comRdr
|
||||
}
|
||||
} else {
|
||||
|
||||
@@ -12,8 +12,6 @@ func (g GZip) Reader(src io.Reader) (io.ReadCloser, error) {
|
||||
return zlib.NewReader(src)
|
||||
}
|
||||
|
||||
func (g GZip) Resetable() bool { return true }
|
||||
|
||||
func (g GZip) Reset(old, src io.Reader) error {
|
||||
return old.(zlib.Resetter).Reset(src, nil)
|
||||
}
|
||||
|
||||
@@ -1,19 +1,22 @@
|
||||
package decompress
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
)
|
||||
|
||||
var ErrNotResetable = errors.New("decompressor not resetable")
|
||||
|
||||
type Decompressor interface {
|
||||
//Creates a new decompressor reading from src.
|
||||
Reader(src io.Reader) (io.ReadCloser, error)
|
||||
//Reports whether Reset will work or not.
|
||||
Resetable() bool
|
||||
}
|
||||
|
||||
type Resetable interface {
|
||||
//Reset attempts to re-use an old decompressor with new data.
|
||||
//Will return ErrNotResetable if not Resetable().
|
||||
//Must ALWAYS be provided with a reader created with Reader.
|
||||
Reset(old, src io.Reader) error
|
||||
}
|
||||
|
||||
type Decoder interface {
|
||||
//Decodes a chunk of data all at once.
|
||||
Decode(in []byte) ([]byte, error)
|
||||
}
|
||||
|
||||
@@ -12,8 +12,6 @@ func (l Lz4) Reader(r io.Reader) (io.ReadCloser, error) {
|
||||
return io.NopCloser(lz4.NewReader(r)), nil
|
||||
}
|
||||
|
||||
func (l Lz4) Resetable() bool { return true }
|
||||
|
||||
func (l Lz4) Reset(old, src io.Reader) error {
|
||||
old.(*lz4.Reader).Reset(src)
|
||||
return nil
|
||||
|
||||
@@ -12,7 +12,3 @@ func (l Lzma) Reader(r io.Reader) (io.ReadCloser, error) {
|
||||
rdr, err := lzma.NewReader(r)
|
||||
return io.NopCloser(rdr), err
|
||||
}
|
||||
|
||||
func (l Lzma) Resetable() bool { return false }
|
||||
|
||||
func (l Lzma) Reset(old, src io.Reader) error { return ErrNotResetable }
|
||||
|
||||
@@ -16,7 +16,3 @@ func (l Lzo) Reader(r io.Reader) (io.ReadCloser, error) {
|
||||
}
|
||||
return io.NopCloser(bytes.NewReader(cache)), nil
|
||||
}
|
||||
|
||||
func (l Lzo) Resetable() bool { return false }
|
||||
|
||||
func (l Lzo) Reset(old, src io.Reader) error { return ErrNotResetable }
|
||||
|
||||
@@ -13,8 +13,6 @@ func (x Xz) Reader(r io.Reader) (io.ReadCloser, error) {
|
||||
return io.NopCloser(rdr), err
|
||||
}
|
||||
|
||||
func (x Xz) Resetable() bool { return true }
|
||||
|
||||
func (x Xz) Reset(old, src io.Reader) error {
|
||||
return old.(*xz.Reader).Reset(src)
|
||||
}
|
||||
|
||||
@@ -15,13 +15,11 @@ func (z Zstd) Reader(src io.Reader) (io.ReadCloser, error) {
|
||||
return r.IOReadCloser(), err
|
||||
}
|
||||
|
||||
func (z Zstd) Resetable() bool { return true }
|
||||
|
||||
func (z Zstd) Reset(old, src io.Reader) error {
|
||||
return old.(*zstd.Decoder).Reset(src)
|
||||
}
|
||||
|
||||
func (z *Zstd) Decode(in []byte) (out []byte, err error) {
|
||||
func (z Zstd) Decode(in []byte) ([]byte, error) {
|
||||
if z.writeToReader == nil {
|
||||
z.writeToReader, _ = zstd.NewReader(nil)
|
||||
}
|
||||
|
||||
@@ -29,6 +29,7 @@ type Entry struct {
|
||||
BlockStart uint32
|
||||
Type uint16
|
||||
Offset uint16
|
||||
Num uint32
|
||||
}
|
||||
|
||||
func readEntry(r io.Reader) (e entry, err error) {
|
||||
@@ -72,6 +73,7 @@ func ReadEntries(rdr io.Reader, size uint32) (e []Entry, err error) {
|
||||
BlockStart: h.InodeStart,
|
||||
Type: en.Type,
|
||||
Offset: en.Offset,
|
||||
Num: h.Num + uint32(en.NumOffset),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ import (
|
||||
"encoding/binary"
|
||||
"errors"
|
||||
"io"
|
||||
"io/fs"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
@@ -77,3 +78,66 @@ func Read(r io.Reader, blockSize uint32) (i Inode, err error) {
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (i Inode) Mode() (out fs.FileMode) {
|
||||
out = fs.FileMode(i.Perm)
|
||||
switch i.Data.(type) {
|
||||
case Directory:
|
||||
out |= fs.ModeDir
|
||||
case EDirectory:
|
||||
out |= fs.ModeDir
|
||||
case Symlink:
|
||||
out |= fs.ModeSymlink
|
||||
case ESymlink:
|
||||
out |= fs.ModeSymlink
|
||||
case Device:
|
||||
out |= fs.ModeDevice
|
||||
case EDevice:
|
||||
out |= fs.ModeDevice
|
||||
case IPC:
|
||||
out |= fs.ModeNamedPipe
|
||||
case EIPC:
|
||||
out |= fs.ModeNamedPipe
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (i Inode) LinkCount() uint32 {
|
||||
switch i.Data.(type) {
|
||||
case EFile:
|
||||
return i.Data.(EFile).LinkCount
|
||||
case Directory:
|
||||
return i.Data.(Directory).LinkCount
|
||||
case EDirectory:
|
||||
return i.Data.(EDirectory).LinkCount
|
||||
case Device:
|
||||
return i.Data.(Device).LinkCount
|
||||
case EDevice:
|
||||
return i.Data.(EDevice).LinkCount
|
||||
case IPC:
|
||||
return i.Data.(IPC).LinkCount
|
||||
case EIPC:
|
||||
return i.Data.(EIPC).LinkCount
|
||||
case Symlink:
|
||||
return i.Data.(Symlink).LinkCount
|
||||
case ESymlink:
|
||||
return i.Data.(ESymlink).LinkCount
|
||||
default:
|
||||
return 0
|
||||
}
|
||||
}
|
||||
|
||||
func (i Inode) Size() uint64 {
|
||||
switch i.Data.(type) {
|
||||
case File:
|
||||
return uint64(i.Data.(File).Size)
|
||||
case EFile:
|
||||
return i.Data.(EFile).Size
|
||||
// case Directory:
|
||||
// return uint64(i.Data.(Directory).Size)
|
||||
// case EDirectory:
|
||||
// return uint64(i.Data.(EDirectory).Size)
|
||||
default:
|
||||
return 0
|
||||
}
|
||||
}
|
||||
|
||||
@@ -26,7 +26,7 @@ func realSize(siz uint16) uint16 {
|
||||
}
|
||||
|
||||
func (r *Reader) advance() (err error) {
|
||||
if !r.d.Resetable() {
|
||||
if _, ok := r.d.(decompress.Resetable); !ok {
|
||||
if clr, ok := r.cur.(io.Closer); ok {
|
||||
clr.Close()
|
||||
}
|
||||
@@ -39,14 +39,14 @@ func (r *Reader) advance() (err error) {
|
||||
size := realSize(raw)
|
||||
r.cur = io.LimitReader(r.master, int64(size))
|
||||
if size == raw {
|
||||
if r.d.Resetable() {
|
||||
if rs, ok := r.d.(decompress.Resetable); ok {
|
||||
if r.comRdr == nil {
|
||||
r.cur, err = r.d.Reader(r.cur)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
err = r.d.Reset(r.comRdr, r.cur)
|
||||
err = rs.Reset(r.comRdr, r.cur)
|
||||
r.cur = r.comRdr
|
||||
}
|
||||
} else {
|
||||
|
||||
@@ -0,0 +1,23 @@
|
||||
package threadmanager
|
||||
|
||||
type Manager struct {
|
||||
c chan int
|
||||
}
|
||||
|
||||
func NewManager(maxRoutines int) *Manager {
|
||||
m := &Manager{
|
||||
c: make(chan int, maxRoutines),
|
||||
}
|
||||
for i := 0; i < maxRoutines; i++ {
|
||||
m.c <- i
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
func (m *Manager) Lock() int {
|
||||
return <-m.c
|
||||
}
|
||||
|
||||
func (m *Manager) Unlock(n int) {
|
||||
m.c <- n
|
||||
}
|
||||
@@ -0,0 +1,19 @@
|
||||
package toreader
|
||||
|
||||
import "io"
|
||||
|
||||
type OffsetReader struct {
|
||||
r io.ReaderAt
|
||||
off int64
|
||||
}
|
||||
|
||||
func NewOffsetReader(r io.ReaderAt, off int64) *OffsetReader {
|
||||
return &OffsetReader{
|
||||
r: r,
|
||||
off: off,
|
||||
}
|
||||
}
|
||||
|
||||
func (r OffsetReader) ReadAt(p []byte, off int64) (n int, e error) {
|
||||
return r.r.ReadAt(p, off+r.off)
|
||||
}
|
||||
@@ -6,7 +6,8 @@ type ReaderAt struct {
|
||||
d []byte
|
||||
}
|
||||
|
||||
func NewReaderAt(r io.Reader) (ra ReaderAt, err error) {
|
||||
func NewReaderAt(r io.Reader) (ra *ReaderAt, err error) {
|
||||
ra = new(ReaderAt)
|
||||
ra.d, err = io.ReadAll(r)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -7,15 +7,21 @@ import (
|
||||
"math"
|
||||
"time"
|
||||
|
||||
"github.com/CalebQ42/fuse"
|
||||
"github.com/CalebQ42/squashfs/internal/decompress"
|
||||
"github.com/CalebQ42/squashfs/internal/directory"
|
||||
"github.com/CalebQ42/squashfs/internal/inode"
|
||||
"github.com/CalebQ42/squashfs/internal/metadata"
|
||||
"github.com/CalebQ42/squashfs/internal/toreader"
|
||||
fuse2 "github.com/seaweedfs/fuse"
|
||||
)
|
||||
|
||||
type Reader struct {
|
||||
*FS
|
||||
con *fuse.Conn
|
||||
con2 *fuse2.Conn
|
||||
mountDone chan struct{}
|
||||
mount2Done chan struct{}
|
||||
d decompress.Decompressor
|
||||
r io.ReaderAt
|
||||
fragEntries []fragEntry
|
||||
@@ -40,6 +46,10 @@ const (
|
||||
ZSTDCompression
|
||||
)
|
||||
|
||||
func NewReaderAtOffset(r io.ReaderAt, off int64) (*Reader, error) {
|
||||
return NewReader(toreader.NewOffsetReader(r, off))
|
||||
}
|
||||
|
||||
// Creates a new squashfs.Reader from the given io.Reader. NOTE: All data from the io.Reader will be read and stored in memory.
|
||||
func NewReaderFromReader(r io.Reader) (*Reader, error) {
|
||||
rdr, err := toreader.NewReaderAt(r)
|
||||
|
||||
+242
-75
@@ -6,19 +6,23 @@ import (
|
||||
"io/fs"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/CalebQ42/squashfs/internal/data"
|
||||
"github.com/CalebQ42/squashfs/internal/directory"
|
||||
"github.com/CalebQ42/squashfs/internal/inode"
|
||||
"github.com/CalebQ42/squashfs/internal/threadmanager"
|
||||
)
|
||||
|
||||
// File represents a file inside a squashfs archive.
|
||||
type File struct {
|
||||
i inode.Inode
|
||||
rdr io.Reader
|
||||
fullRdr io.WriterTo
|
||||
fullRdr *data.FullReader
|
||||
r *Reader
|
||||
parent *FS
|
||||
e directory.Entry
|
||||
@@ -35,7 +39,7 @@ func (r Reader) newFile(en directory.Entry, parent *FS) (*File, error) {
|
||||
return nil, err
|
||||
}
|
||||
var rdr io.Reader
|
||||
var full io.WriterTo
|
||||
var full *data.FullReader
|
||||
if i.Type == inode.Fil || i.Type == inode.EFil {
|
||||
full, rdr, err = r.getReaders(i)
|
||||
if err != nil {
|
||||
@@ -57,6 +61,21 @@ func (f File) Stat() (fs.FileInfo, error) {
|
||||
return newFileInfo(f.e, f.i), nil
|
||||
}
|
||||
|
||||
// Mode returns the file's fs.FileMode
|
||||
func (f File) Mode() fs.FileMode {
|
||||
switch f.e.Type {
|
||||
case inode.Dir:
|
||||
return fs.FileMode(f.i.Perm) | fs.ModeDir
|
||||
case inode.Char:
|
||||
return fs.FileMode(f.i.Perm) | fs.ModeCharDevice
|
||||
case inode.Block:
|
||||
return fs.FileMode(f.i.Perm) | fs.ModeDevice
|
||||
case inode.Sym:
|
||||
return fs.FileMode(f.i.Perm) | fs.ModeSymlink
|
||||
}
|
||||
return fs.FileMode(f.i.Perm)
|
||||
}
|
||||
|
||||
// Read reads the data from the file. Only works if file is a normal file.
|
||||
func (f File) Read(p []byte) (int, error) {
|
||||
if f.i.Type != inode.Fil && f.i.Type != inode.EFil {
|
||||
@@ -68,13 +87,23 @@ func (f File) Read(p []byte) (int, error) {
|
||||
return f.rdr.Read(p)
|
||||
}
|
||||
|
||||
func (f File) ReadAt(p []byte, off int64) (int, error) {
|
||||
if f.i.Type != inode.Fil && f.i.Type != inode.EFil {
|
||||
return 0, ErrReadNotFile
|
||||
}
|
||||
return f.fullRdr.ReadAt(p, off)
|
||||
}
|
||||
|
||||
// WriteTo writes all data from the file to the writer. This is multi-threaded.
|
||||
// The underlying reader is seperate from the one used with Read and can be reused.
|
||||
func (f File) WriteTo(w io.Writer) (int64, error) {
|
||||
if f.i.Type != inode.Fil && f.i.Type != inode.EFil {
|
||||
return 0, ErrReadNotFile
|
||||
}
|
||||
return f.fullRdr.WriteTo(w)
|
||||
}
|
||||
|
||||
// Close simply nils the underlying reader. Here mostly to satisfy fs.File
|
||||
// Close simply nils the underlying reader.
|
||||
func (f *File) Close() error {
|
||||
f.rdr = nil
|
||||
return nil
|
||||
@@ -84,7 +113,7 @@ func (f *File) Close() error {
|
||||
// If n <= 0 all fs.DirEntry's are returned.
|
||||
func (f *File) ReadDir(n int) (out []fs.DirEntry, err error) {
|
||||
if !f.IsDir() {
|
||||
return nil, errors.New("File is not a directory")
|
||||
return nil, errors.New("file is not a directory")
|
||||
}
|
||||
ents, err := f.r.readDirectory(f.i)
|
||||
if err != nil {
|
||||
@@ -141,6 +170,20 @@ func (f File) IsSymlink() bool {
|
||||
return f.i.Type == inode.Sym || f.i.Type == inode.ESym
|
||||
}
|
||||
|
||||
func (f File) isDeviceOrFifo() bool {
|
||||
return f.i.Type == inode.Char || f.i.Type == inode.Block || f.i.Type == inode.EChar || f.i.Type == inode.EBlock || f.i.Type == inode.Fifo || f.i.Type == inode.EFifo
|
||||
}
|
||||
|
||||
func (f File) deviceDevices() (maj uint32, min uint32) {
|
||||
var dev uint32
|
||||
if f.i.Type == inode.Char || f.i.Type == inode.Block {
|
||||
dev = f.i.Data.(inode.Device).Dev
|
||||
} else if f.i.Type == inode.EChar || f.i.Type == inode.EBlock {
|
||||
dev = f.i.Data.(inode.EDevice).Dev
|
||||
}
|
||||
return dev >> 8, dev & 0x000FF
|
||||
}
|
||||
|
||||
// SymlinkPath returns the symlink's target path. Is the File isn't a symlink, returns an empty string.
|
||||
func (f File) SymlinkPath() string {
|
||||
switch f.i.Type {
|
||||
@@ -177,134 +220,197 @@ func (f File) GetSymlinkFile() *File {
|
||||
|
||||
// ExtractionOptions are available options on how to extract.
|
||||
type ExtractionOptions struct {
|
||||
LogOutput io.Writer //Where error log should write. If nil, uses os.Stdout. Has no effect if verbose is false.
|
||||
DereferenceSymlink bool //Replace symlinks with the target file
|
||||
UnbreakSymlink bool //Try to make sure symlinks remain unbroken when extracted, without changing the symlink
|
||||
Verbose bool //Prints extra info to log on an error
|
||||
FolderPerm fs.FileMode //The permissions used when creating the extraction folder
|
||||
manager *threadmanager.Manager
|
||||
LogOutput io.Writer //Where error log should write.
|
||||
DereferenceSymlink bool //Replace symlinks with the target file.
|
||||
UnbreakSymlink bool //Try to make sure symlinks remain unbroken when extracted, without changing the symlink.
|
||||
Verbose bool //Prints extra info to log on an error.
|
||||
IgnorePerm bool //Ignore file's permissions and instead use Perm.
|
||||
Perm fs.FileMode //Permission to use when IgnorePerm. Defaults to 0755.
|
||||
notFirst bool
|
||||
}
|
||||
|
||||
// DefaultOptions is the default ExtractionOptions.
|
||||
func DefaultOptions() ExtractionOptions {
|
||||
return ExtractionOptions{
|
||||
FolderPerm: 0755,
|
||||
func DefaultOptions() *ExtractionOptions {
|
||||
return &ExtractionOptions{
|
||||
Perm: 0755,
|
||||
}
|
||||
}
|
||||
|
||||
// ExtractTo extracts the File to the given folder with the default options.
|
||||
// If the File is a directory, it instead extracts the directory's contents to the folder.
|
||||
func (f File) ExtractTo(folder string) error {
|
||||
return f.ExtractWithOptions(folder, DefaultOptions())
|
||||
return f.realExtract(folder, DefaultOptions())
|
||||
}
|
||||
|
||||
// ExtractVerbose extracts the File to the folder with the Verbose option.
|
||||
func (f File) ExtractVerbose(folder string) error {
|
||||
op := DefaultOptions()
|
||||
op.Verbose = true
|
||||
return f.realExtract(folder, op)
|
||||
}
|
||||
|
||||
// ExtractIgnorePermissions extracts the File to the folder with the IgnorePerm option.
|
||||
func (f File) ExtractIgnorePermissions(folder string) error {
|
||||
op := DefaultOptions()
|
||||
op.IgnorePerm = true
|
||||
return f.realExtract(folder, op)
|
||||
}
|
||||
|
||||
// ExtractSymlink extracts the File to the folder with the DereferenceSymlink option.
|
||||
// If the File is a directory, it instead extracts the directory's contents to the folder.
|
||||
func (f File) ExtractSymlink(folder string) error {
|
||||
return f.ExtractWithOptions(folder, ExtractionOptions{
|
||||
DereferenceSymlink: true,
|
||||
FolderPerm: 0755,
|
||||
})
|
||||
op := DefaultOptions()
|
||||
op.DereferenceSymlink = true
|
||||
return f.realExtract(folder, op)
|
||||
}
|
||||
|
||||
// ExtractWithOptions extracts the File to the given folder with the given ExtrationOptions.
|
||||
// If the File is a directory, it instead extracts the directory's contents to the folder.
|
||||
func (f File) ExtractWithOptions(folder string, op ExtractionOptions) error {
|
||||
if op.Verbose {
|
||||
if op.LogOutput == nil {
|
||||
op.LogOutput = os.Stdout
|
||||
}
|
||||
func (f File) ExtractWithOptions(folder string, op *ExtractionOptions) error {
|
||||
if op.Verbose && op.LogOutput != nil {
|
||||
log.SetOutput(op.LogOutput)
|
||||
}
|
||||
return f.realExtract(folder, op)
|
||||
}
|
||||
|
||||
func (f File) realExtract(folder string, op ExtractionOptions) error {
|
||||
err := os.MkdirAll(folder, op.FolderPerm)
|
||||
folder = filepath.Clean(folder)
|
||||
if err != nil && !os.IsExist(err) {
|
||||
if op.Verbose {
|
||||
log.Println("Error while creating extraction folder")
|
||||
}
|
||||
return err
|
||||
func (f File) realExtract(folder string, op *ExtractionOptions) (err error) {
|
||||
if op.manager == nil {
|
||||
op.manager = threadmanager.NewManager(runtime.NumCPU())
|
||||
}
|
||||
extDir := folder + "/" + f.e.Name
|
||||
if !op.notFirst {
|
||||
op.notFirst = true
|
||||
if f.IsDir() {
|
||||
filFS, _ := f.FS()
|
||||
var ents []directory.Entry
|
||||
ents, err = f.r.readDirectory(f.i)
|
||||
extDir = folder
|
||||
_, err = os.Open(folder)
|
||||
if err != nil && os.IsNotExist(err) {
|
||||
err = os.Mkdir(extDir, op.Perm)
|
||||
}
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while reading children of", f.path())
|
||||
log.Println("Error while making", folder)
|
||||
}
|
||||
return
|
||||
}
|
||||
if !op.IgnorePerm {
|
||||
defer os.Chmod(extDir, f.Mode())
|
||||
defer os.Chown(extDir, int(f.r.ids[f.i.UidInd]), int(f.r.ids[f.i.GidInd]))
|
||||
}
|
||||
}
|
||||
}
|
||||
switch {
|
||||
case f.IsDir():
|
||||
if folder != extDir && f.e.Name != "" {
|
||||
//First extract it with a permisive permission.
|
||||
err = os.Mkdir(extDir, op.Perm)
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while making directory", extDir)
|
||||
}
|
||||
return
|
||||
}
|
||||
//Then set it to it's actual permissions once we're done with it
|
||||
if !op.IgnorePerm {
|
||||
defer os.Chmod(extDir, f.Mode())
|
||||
defer os.Chown(extDir, int(f.r.ids[f.i.UidInd]), int(f.r.ids[f.i.GidInd]))
|
||||
}
|
||||
}
|
||||
var filFS *FS
|
||||
filFS, err = f.FS()
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while converting", f.path(), "to FS")
|
||||
}
|
||||
return err
|
||||
}
|
||||
errChan := make(chan error)
|
||||
for i := 0; i < len(ents); i++ {
|
||||
go func(ent directory.Entry) {
|
||||
fil, goErr := f.r.newFile(ent, filFS)
|
||||
errChan := make(chan error, len(filFS.e))
|
||||
files := make([]directory.Entry, 0)
|
||||
//Focus on making the folder tree first...
|
||||
var i int
|
||||
for i = 0; i < len(filFS.e); i++ {
|
||||
if filFS.e[i].Type == inode.Fil {
|
||||
files = append(files, filFS.e[i])
|
||||
} else {
|
||||
go func(index int) {
|
||||
subF, goErr := f.r.newFile(filFS.e[index], filFS)
|
||||
if goErr != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while reading info for", filepath.Join(f.path(), ent.Name))
|
||||
log.Println("Error while resolving", extDir)
|
||||
}
|
||||
errChan <- goErr
|
||||
return
|
||||
}
|
||||
if fil.IsDir() {
|
||||
info, _ := fil.Stat()
|
||||
err = os.Mkdir(filepath.Join(folder, fil.e.Name), info.Mode())
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while creating", filepath.Join(folder, fil.e.Name))
|
||||
errChan <- subF.ExtractWithOptions(extDir, op)
|
||||
}(i)
|
||||
}
|
||||
errChan <- err
|
||||
return
|
||||
}
|
||||
errChan <- fil.realExtract(filepath.Join(folder, fil.e.Name), op)
|
||||
} else {
|
||||
errChan <- fil.realExtract(folder, op)
|
||||
}
|
||||
fil.Close()
|
||||
}(ents[i])
|
||||
}
|
||||
for i := 0; i < len(ents); i++ {
|
||||
for i = 0; i < len(filFS.e)-len(files); i++ {
|
||||
err = <-errChan
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
} else if f.IsRegular() {
|
||||
//Then we extract the files.
|
||||
for i = 0; i < len(files); i++ {
|
||||
go func(index int) {
|
||||
n := op.manager.Lock()
|
||||
defer op.manager.Unlock(n)
|
||||
subF, goErr := f.r.newFile(files[index], filFS)
|
||||
if goErr != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while resolving", extDir)
|
||||
}
|
||||
errChan <- goErr
|
||||
return
|
||||
}
|
||||
errChan <- subF.ExtractWithOptions(extDir, op)
|
||||
}(i)
|
||||
}
|
||||
for i = 0; i < len(files); i++ {
|
||||
err = <-errChan
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
case f.IsRegular():
|
||||
var fil *os.File
|
||||
fil, err = os.Create(folder + "/" + f.e.Name)
|
||||
fil, err = os.Create(extDir)
|
||||
if os.IsExist(err) {
|
||||
os.Remove(folder + "/" + f.e.Name)
|
||||
fil, err = os.Create(folder + "/" + f.e.Name)
|
||||
os.Remove(extDir)
|
||||
fil, err = os.Create(extDir)
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while creating", folder+"/"+f.e.Name)
|
||||
log.Println("Error while creating", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
} else if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while creating", folder+"/"+f.e.Name)
|
||||
log.Println("Error while creating", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
defer fil.Close()
|
||||
_, err = io.Copy(fil, f)
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while copying data to", folder+"/"+f.e.Name)
|
||||
log.Println("Error while copying data to", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
} else if f.IsSymlink() {
|
||||
if op.IgnorePerm {
|
||||
os.Chmod(extDir, op.Perm|(f.Mode()&fs.ModeType))
|
||||
} else {
|
||||
os.Chmod(extDir, f.Mode())
|
||||
os.Chown(extDir, int(f.r.ids[f.i.UidInd]), int(f.r.ids[f.i.GidInd]))
|
||||
}
|
||||
case f.IsSymlink():
|
||||
symPath := f.SymlinkPath()
|
||||
if op.DereferenceSymlink {
|
||||
fil := f.GetSymlinkFile()
|
||||
if fil == nil {
|
||||
if op.Verbose {
|
||||
log.Println("Symlink path(", symPath, ") is unobtainable:", folder+"/"+f.e.Name)
|
||||
log.Println("Symlink path(", symPath, ") is unobtainable:", extDir)
|
||||
}
|
||||
return errors.New("cannot get symlink target")
|
||||
}
|
||||
@@ -312,7 +418,7 @@ func (f File) realExtract(folder string, op ExtractionOptions) error {
|
||||
err = fil.realExtract(folder, op)
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while extracting the symlink's file:", folder+"/"+f.e.Name)
|
||||
log.Println("Error while extracting the symlink's file:", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
@@ -321,31 +427,92 @@ func (f File) realExtract(folder string, op ExtractionOptions) error {
|
||||
fil := f.GetSymlinkFile()
|
||||
if fil == nil {
|
||||
if op.Verbose {
|
||||
log.Println("Symlink path(", symPath, ") is unobtainable:", folder+"/"+f.e.Name)
|
||||
log.Println("Symlink path(", symPath, ") is unobtainable:", extDir)
|
||||
}
|
||||
return errors.New("cannot get symlink target")
|
||||
}
|
||||
extractLoc := filepath.Clean(folder + "/" + filepath.Dir(symPath))
|
||||
extractLoc := filepath.Join(folder, filepath.Dir(symPath))
|
||||
err = fil.realExtract(extractLoc, op)
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while extracting ", folder+"/"+f.e.Name)
|
||||
log.Println("Error while extracting ", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
}
|
||||
err = os.Symlink(f.SymlinkPath(), folder+"/"+f.e.Name)
|
||||
err = os.Symlink(f.SymlinkPath(), extDir)
|
||||
if os.IsExist(err) {
|
||||
os.Remove(folder + "/" + f.e.Name)
|
||||
err = os.Symlink(f.SymlinkPath(), folder+"/"+f.e.Name)
|
||||
os.Remove(extDir)
|
||||
err = os.Symlink(f.SymlinkPath(), extDir)
|
||||
}
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while making symlink:", folder+"/"+f.e.Name)
|
||||
log.Println("Error while making symlink:", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
if op.IgnorePerm {
|
||||
os.Chmod(extDir, op.Perm|(f.Mode()&fs.ModeType))
|
||||
} else {
|
||||
os.Chmod(extDir, f.Mode())
|
||||
os.Chown(extDir, int(f.r.ids[f.i.UidInd]), int(f.r.ids[f.i.GidInd]))
|
||||
}
|
||||
case f.isDeviceOrFifo():
|
||||
if runtime.GOOS == "windows" {
|
||||
if op.Verbose {
|
||||
log.Println(extDir, "ignored since it's a device link and can't be created on Windows.")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
_, err = exec.LookPath("mknod")
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Extracting Fifo IPC or Device and mknod is not in PATH")
|
||||
}
|
||||
return err
|
||||
}
|
||||
var typ string
|
||||
if f.i.Type == inode.Char || f.i.Type == inode.EChar {
|
||||
typ = "c"
|
||||
} else if f.i.Type == inode.Block || f.i.Type == inode.EBlock {
|
||||
typ = "b"
|
||||
} else { //Fifo IPC
|
||||
if runtime.GOOS == "darwin" {
|
||||
if op.Verbose {
|
||||
log.Println(extDir, "ignored since it's a Fifo file and can't be created on Darwin.")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
typ = "p"
|
||||
}
|
||||
cmd := exec.Command("mknod", extDir, typ)
|
||||
if typ != "p" {
|
||||
maj, min := f.deviceDevices()
|
||||
cmd.Args = append(cmd.Args, strconv.Itoa(int(maj)), strconv.Itoa(int(min)))
|
||||
}
|
||||
if op.Verbose {
|
||||
cmd.Stdout = op.LogOutput
|
||||
cmd.Stderr = op.LogOutput
|
||||
}
|
||||
err = cmd.Run()
|
||||
if err != nil {
|
||||
if op.Verbose {
|
||||
log.Println("Error while running mknod for", extDir)
|
||||
}
|
||||
return err
|
||||
}
|
||||
if op.IgnorePerm {
|
||||
os.Chmod(extDir, op.Perm|(f.Mode()&fs.ModeType))
|
||||
} else {
|
||||
os.Chmod(extDir, f.Mode())
|
||||
os.Chown(extDir, int(f.r.ids[f.i.UidInd]), int(f.r.ids[f.i.GidInd]))
|
||||
}
|
||||
case f.e.Type == inode.Sock:
|
||||
if op.Verbose {
|
||||
log.Println(extDir, "ignored since it's a socket file.")
|
||||
}
|
||||
default:
|
||||
return errors.New("Unsupported file type. Inode type: " + strconv.Itoa(int(f.i.Type)))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
+8
-3
@@ -39,8 +39,8 @@ func (r Reader) newFS(e directory.Entry, parent *FS) (*FS, error) {
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Open opens the file at name. Returns a squashfs.File.
|
||||
func (f FS) Open(name string) (fs.File, error) {
|
||||
// Opens the file at name. Returns a squashfs.File.
|
||||
func (f FS) OpenFile(name string) (*File, error) {
|
||||
name = filepath.Clean(name)
|
||||
if !fs.ValidPath(name) {
|
||||
return nil, &fs.PathError{
|
||||
@@ -73,7 +73,7 @@ func (f FS) Open(name string) (fs.File, error) {
|
||||
Err: err,
|
||||
}
|
||||
}
|
||||
out, err := newFS.Open(strings.Join(split[1:], "/"))
|
||||
out, err := newFS.OpenFile(strings.Join(split[1:], "/"))
|
||||
if err != nil {
|
||||
err.(*fs.PathError).Path = name
|
||||
}
|
||||
@@ -96,6 +96,11 @@ func (f FS) Open(name string) (fs.File, error) {
|
||||
}
|
||||
}
|
||||
|
||||
// Opens the file at name. Returns a io/fs.File.
|
||||
func (f FS) Open(name string) (fs.File, error) {
|
||||
return f.OpenFile(name)
|
||||
}
|
||||
|
||||
// Glob returns the name of the files at the given pattern.
|
||||
// All paths are relative to the FS.
|
||||
// Uses filepath.Match to compare names.
|
||||
|
||||
+4
-1
@@ -36,23 +36,26 @@ func (r Reader) getReaders(i inode.Inode) (full *data.FullReader, rdr *data.Read
|
||||
var blockSizes []uint32
|
||||
var fragInd uint32
|
||||
var fragSize uint32
|
||||
var fileSize uint64
|
||||
if i.Type == inode.Fil {
|
||||
fragOffset = uint64(i.Data.(inode.File).FragOffset)
|
||||
blockOffset = uint64(i.Data.(inode.File).BlockStart)
|
||||
blockSizes = i.Data.(inode.File).BlockSizes
|
||||
fragInd = i.Data.(inode.File).FragInd
|
||||
fragSize = i.Data.(inode.File).Size % r.s.BlockSize
|
||||
fileSize = uint64(i.Data.(inode.File).Size)
|
||||
} else if i.Type == inode.EFil {
|
||||
fragOffset = uint64(i.Data.(inode.EFile).FragOffset)
|
||||
blockOffset = i.Data.(inode.EFile).BlockStart
|
||||
blockSizes = i.Data.(inode.EFile).BlockSizes
|
||||
fragInd = i.Data.(inode.EFile).FragInd
|
||||
fragSize = uint32(i.Data.(inode.EFile).Size % uint64(r.s.BlockSize))
|
||||
fileSize = i.Data.(inode.EFile).Size
|
||||
} else {
|
||||
return nil, nil, errors.New("getReaders called on non-file type")
|
||||
}
|
||||
rdr = data.NewReader(toreader.NewReader(r.r, int64(blockOffset)), r.d, blockSizes, r.s.BlockSize)
|
||||
full = data.NewFullReader(r.r, uint64(blockOffset), r.d, blockSizes, r.s.BlockSize)
|
||||
full = data.NewFullReader(r.r, uint64(blockOffset), r.d, blockSizes, r.s.BlockSize, fileSize)
|
||||
if fragInd != 0xFFFFFFFF {
|
||||
full.AddFragment(func() (io.Reader, error) {
|
||||
var fragRdr io.Reader
|
||||
|
||||
+27
-7
@@ -18,8 +18,8 @@ import (
|
||||
)
|
||||
|
||||
const (
|
||||
squashfsURL = "https://darkstorm.tech/LinuxPATest.sfs"
|
||||
squashfsName = "LinuxPATest.sfs"
|
||||
squashfsURL = "https://darkstorm.tech/files/LinuxPATest.sfs"
|
||||
squashfsName = "bug.sqfs"
|
||||
|
||||
filePath = "PortableApps/Notepad++Portable/App/DefaultData/Config/contextMenu.xml"
|
||||
)
|
||||
@@ -73,7 +73,6 @@ func TestMisc(t *testing.T) {
|
||||
}
|
||||
|
||||
func BenchmarkRace(b *testing.B) {
|
||||
// tmpDir := b.TempDir()
|
||||
tmpDir := "testing"
|
||||
fil, err := preTest(tmpDir)
|
||||
if err != nil {
|
||||
@@ -107,7 +106,6 @@ func BenchmarkRace(b *testing.B) {
|
||||
}
|
||||
|
||||
func TestExtractQuick(t *testing.T) {
|
||||
|
||||
//First, setup everything and extract the archive using the library and unsquashfs
|
||||
|
||||
// tmpDir := b.TempDir()
|
||||
@@ -124,8 +122,12 @@ func TestExtractQuick(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
os.RemoveAll(filepath.Join(tmpDir, "testLog.txt"))
|
||||
logFil, _ := os.Create(filepath.Join(tmpDir, "testLog.txt"))
|
||||
op := squashfs.DefaultOptions()
|
||||
op.Verbose = true
|
||||
op.IgnorePerm = true
|
||||
op.LogOutput = logFil
|
||||
err = rdr.ExtractWithOptions(libPath, op)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
@@ -177,11 +179,29 @@ func TestSingleFile(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
op := squashfs.DefaultOptions()
|
||||
op.Verbose = true
|
||||
err = f.(*squashfs.File).ExtractWithOptions("testing", op)
|
||||
err = f.(*squashfs.File).ExtractWithOptions("testing", &squashfs.ExtractionOptions{Verbose: true})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Fatal("HI")
|
||||
}
|
||||
|
||||
func TestFuse(t *testing.T) {
|
||||
tmpDir := "testing"
|
||||
fil, err := preTest(tmpDir)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
os.Remove(filepath.Base(filePath))
|
||||
rdr, err := squashfs.NewReader(fil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
err = rdr.Mount("testing/fuseTest")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer rdr.Unmount()
|
||||
rdr.MountWait()
|
||||
t.Fatal("testing")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user