Made it work with new library

This commit is contained in:
Caleb Gardner
2023-12-27 23:59:32 -06:00
parent 95378b9c42
commit 18b4139373
7 changed files with 208 additions and 84 deletions
+82 -42
View File
@@ -1,4 +1,4 @@
package squashfs package squashfuse
import ( import (
"bytes" "bytes"
@@ -6,98 +6,133 @@ import (
"errors" "errors"
"io" "io"
"github.com/CalebQ42/squashfs/internal/inode" "github.com/CalebQ42/squashfs"
squashfslow "github.com/CalebQ42/squashfs/low"
"github.com/CalebQ42/squashfs/low/inode"
"github.com/seaweedfs/fuse" "github.com/seaweedfs/fuse"
"github.com/seaweedfs/fuse/fs" "github.com/seaweedfs/fuse/fs"
) )
// Mounts the archive to the given mountpoint using fuse2. Non-blocking. type Fuse2Mount struct {
r *squashfslow.Reader
con *fuse.Conn
mountDone chan struct{}
}
func NewFuse2Mount(r *squashfs.Reader) *Mount {
return NewMountFromLow(r.Low)
}
func NewFuse2MountFromLow(r *squashfslow.Reader) *Mount {
return &Mount{r: r}
}
// 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. // 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) { func (m *Fuse2Mount) Mount(mountpoint string) (err error) {
if r.con != nil { if m.con != nil {
return errors.New("squashfs archive already mounted") return errors.New("squashfs archive already mounted")
} }
r.con2, err = fuse.Mount(mountpoint, fuse.ReadOnly()) m.con, err = fuse.Mount(mountpoint, fuse.ReadOnly())
if err != nil { if err != nil {
return return
} }
<-r.con2.Ready <-m.con.Ready
r.mount2Done = make(chan struct{}) m.mountDone = make(chan struct{})
go func() { go func() {
fs.Serve(r.con2, squashFuse2{r: r}) fs.Serve(m.con, squashFuse2{r: m.r})
close(r.mount2Done) close(m.mountDone)
}() }()
return return
} }
// Blocks until the mount ends. // Blocks until the mount ends.
// Fuse2 version. func (m *Fuse2Mount) MountWait() {
func (r *Reader) MountWaitFuse2() { if m.mountDone != nil {
if r.mount2Done != nil { <-m.mountDone
<-r.mount2Done
} }
} }
// Unmounts the archive. // Unmounts the archive.
// Fuse2 version. func (m *Fuse2Mount) Unmount() error {
func (r *Reader) UnmountFuse2() error { if m.con != nil {
if r.con != nil { defer func() { m.con = nil }()
defer func() { r.con = nil }() return m.con.Close()
return r.con.Close()
} }
return errors.New("squashfs archive is not mounted") return errors.New("squashfs archive is not mounted")
} }
type squashFuse2 struct { type squashFuse2 struct {
r *Reader r *squashfslow.Reader
} }
func (s squashFuse2) Root() (fs.Node, error) { func (s squashFuse2) Root() (fs.Node, error) {
return fileNode2{File: s.r.FS.File}, nil return fileNode2{
FileBase: &s.r.Root.FileBase,
r: s.r,
}, nil
} }
type fileNode2 struct { type fileNode2 struct {
*File *squashfslow.FileBase
r *squashfslow.Reader
} }
func (f fileNode2) Attr(ctx context.Context, attr *fuse.Attr) error { func (f fileNode2) Attr(ctx context.Context, attr *fuse.Attr) error {
attr.Blocks = f.r.s.Size / 512 var err error
if f.r.s.Size%512 > 0 { attr.Gid, err = f.r.Id(f.Inode.GidInd)
if err != nil {
return err
}
attr.Uid, err = f.r.Id(f.Inode.UidInd)
if err != nil {
return err
}
attr.Size = f.Inode.Size()
attr.Blocks = f.Inode.Size() / 512
if f.Inode.Size()%512 > 0 {
attr.Blocks++ attr.Blocks++
} }
attr.Gid = f.r.ids[f.i.GidInd] attr.Inode = uint64(f.Inode.Num)
attr.Inode = uint64(f.i.Num) attr.Mode = f.Inode.Mode()
attr.Mode = f.i.Mode() attr.Nlink = f.Inode.LinkCount()
attr.Nlink = f.i.LinkCount()
attr.Size = f.i.Size()
attr.Uid = f.r.ids[f.i.UidInd]
return nil return nil
} }
func (f fileNode2) Id() uint64 { func (f fileNode2) Id() uint64 {
return uint64(f.i.Num) return uint64(f.Inode.Num)
} }
func (f fileNode2) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) { func (f fileNode2) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) {
return f.SymlinkPath(), nil switch f.Inode.Type {
case inode.Sym:
return string(f.Inode.Data.(inode.Symlink).Target), nil
case inode.ESym:
return string(f.Inode.Data.(inode.ESymlink).Target), nil
}
return "", nil
} }
func (f fileNode2) Lookup(ctx context.Context, name string) (fs.Node, error) { func (f fileNode2) Lookup(ctx context.Context, name string) (fs.Node, error) {
asFS, err := f.FS() asFS, err := f.ToDir(f.r)
if err != nil { if err != nil {
return nil, fuse.ENOTDIR return nil, fuse.ENOTDIR
} }
ret, err := asFS.OpenFile(name) ret, err := asFS.Open(f.r, name)
if err != nil { if err != nil {
return nil, fuse.ENOENT return nil, fuse.ENOENT
} }
return fileNode2{File: ret}, nil return fileNode2{FileBase: ret}, nil
} }
func (f fileNode2) ReadAll(ctx context.Context) ([]byte, error) { func (f fileNode2) ReadAll(ctx context.Context) ([]byte, error) {
if f.IsRegular() { if f.IsRegular() {
rdr, err := f.GetFullReader(f.r)
if err != nil {
return nil, err
}
var buf bytes.Buffer var buf bytes.Buffer
_, err := f.WriteTo(&buf) _, err = rdr.WriteTo(&buf)
return buf.Bytes(), err return buf.Bytes(), err
} }
return nil, ENODATA return nil, ENODATA
@@ -105,8 +140,13 @@ func (f fileNode2) ReadAll(ctx context.Context) ([]byte, error) {
func (f fileNode2) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error { func (f fileNode2) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error {
if f.IsRegular() { if f.IsRegular() {
rdr, err := f.GetReader(f.r)
if err != nil {
return err
}
buf := make([]byte, req.Size) buf := make([]byte, req.Size)
n, err := f.File.ReadAt(buf, req.Offset) rdr.Read(make([]byte, req.Offset))
n, err := rdr.Read(buf)
if err == io.EOF { if err == io.EOF {
resp.Data = buf[:n] resp.Data = buf[:n]
} }
@@ -116,13 +156,13 @@ func (f fileNode2) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.R
} }
func (f fileNode2) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error) { func (f fileNode2) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error) {
asFS, err := f.FS() asFS, err := f.ToDir(f.r)
if err != nil { if err != nil {
return nil, fuse.ENOTDIR return nil, fuse.ENOTDIR
} }
var t fuse.DirentType var t fuse.DirentType
for i := range asFS.e { for i := range asFS.Entries {
switch asFS.e[i].Type { switch asFS.Entries[i].InodeType {
case inode.Fil: case inode.Fil:
t = fuse.DT_File t = fuse.DT_File
case inode.Dir: case inode.Dir:
@@ -141,9 +181,9 @@ func (f fileNode2) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error
t = fuse.DT_Unknown t = fuse.DT_Unknown
} }
out = append(out, fuse.Dirent{ out = append(out, fuse.Dirent{
Inode: uint64(asFS.e[i].Num), Inode: uint64(asFS.Entries[i].Num),
Type: t, Type: t,
Name: asFS.e[i].Name, Name: asFS.Entries[i].Name,
}) })
} }
return return
+81 -39
View File
@@ -1,4 +1,4 @@
package squashfs package squashfuse
import ( import (
"bytes" "bytes"
@@ -8,94 +8,131 @@ import (
"github.com/CalebQ42/fuse" "github.com/CalebQ42/fuse"
"github.com/CalebQ42/fuse/fs" "github.com/CalebQ42/fuse/fs"
"github.com/CalebQ42/squashfs/internal/inode" "github.com/CalebQ42/squashfs"
squashfslow "github.com/CalebQ42/squashfs/low"
"github.com/CalebQ42/squashfs/low/inode"
) )
type Mount struct {
r *squashfslow.Reader
con *fuse.Conn
mountDone chan struct{}
}
func NewMount(r *squashfs.Reader) *Mount {
return NewMountFromLow(r.Low)
}
func NewMountFromLow(r *squashfslow.Reader) *Mount {
return &Mount{r: r}
}
// Mounts the archive to the given mountpoint using fuse3. Non-blocking. // 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. // 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) { func (m *Mount) Mount(mountpoint string) (err error) {
if r.con != nil { if m.con != nil {
return errors.New("squashfs archive already mounted") return errors.New("squashfs archive already mounted")
} }
r.con, err = fuse.Mount(mountpoint, fuse.ReadOnly()) m.con, err = fuse.Mount(mountpoint, fuse.ReadOnly())
if err != nil { if err != nil {
return return
} }
<-r.con.Ready <-m.con.Ready
r.mountDone = make(chan struct{}) m.mountDone = make(chan struct{})
go func() { go func() {
fs.Serve(r.con, squashFuse{r: r}) fs.Serve(m.con, squashFuse{r: m.r})
close(r.mountDone) close(m.mountDone)
}() }()
return return
} }
// Blocks until the mount ends. // Blocks until the mount ends.
func (r *Reader) MountWait() { func (m *Mount) MountWait() {
if r.mountDone != nil { if m.mountDone != nil {
<-r.mountDone <-m.mountDone
} }
} }
// Unmounts the archive. // Unmounts the archive.
func (r *Reader) Unmount() error { func (m *Mount) Unmount() error {
if r.con != nil { if m.con != nil {
defer func() { r.con = nil }() defer func() { m.con = nil }()
return r.con.Close() return m.con.Close()
} }
return errors.New("squashfs archive is not mounted") return errors.New("squashfs archive is not mounted")
} }
type squashFuse struct { type squashFuse struct {
r *Reader r *squashfslow.Reader
} }
func (s squashFuse) Root() (fs.Node, error) { func (s squashFuse) Root() (fs.Node, error) {
return fileNode{File: s.r.FS.File}, nil return fileNode{
FileBase: &s.r.Root.FileBase,
r: s.r,
}, nil
} }
type fileNode struct { type fileNode struct {
*File *squashfslow.FileBase
r *squashfslow.Reader
} }
func (f fileNode) Attr(ctx context.Context, attr *fuse.Attr) error { func (f fileNode) Attr(ctx context.Context, attr *fuse.Attr) error {
attr.Blocks = f.r.s.Size / 512 var err error
if f.r.s.Size%512 > 0 { attr.Gid, err = f.r.Id(f.Inode.GidInd)
if err != nil {
return err
}
attr.Uid, err = f.r.Id(f.Inode.UidInd)
if err != nil {
return err
}
attr.Size = f.Inode.Size()
attr.Blocks = f.Inode.Size() / 512
if f.Inode.Size()%512 > 0 {
attr.Blocks++ attr.Blocks++
} }
attr.Gid = f.r.ids[f.i.GidInd] attr.Inode = uint64(f.Inode.Num)
attr.Inode = uint64(f.i.Num) attr.Mode = f.Inode.Mode()
attr.Mode = f.i.Mode() attr.Nlink = f.Inode.LinkCount()
attr.Nlink = f.i.LinkCount()
attr.Size = f.i.Size()
attr.Uid = f.r.ids[f.i.UidInd]
return nil return nil
} }
func (f fileNode) Id() uint64 { func (f fileNode) Id() uint64 {
return uint64(f.i.Num) return uint64(f.Inode.Num)
} }
func (f fileNode) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) { func (f fileNode) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) {
return f.SymlinkPath(), nil switch f.Inode.Type {
case inode.Sym:
return string(f.Inode.Data.(inode.Symlink).Target), nil
case inode.ESym:
return string(f.Inode.Data.(inode.ESymlink).Target), nil
}
return "", nil
} }
func (f fileNode) Lookup(ctx context.Context, name string) (fs.Node, error) { func (f fileNode) Lookup(ctx context.Context, name string) (fs.Node, error) {
asFS, err := f.FS() asFS, err := f.ToDir(f.r)
if err != nil { if err != nil {
return nil, fuse.ENOTDIR return nil, fuse.ENOTDIR
} }
ret, err := asFS.OpenFile(name) ret, err := asFS.Open(f.r, name)
if err != nil { if err != nil {
return nil, fuse.ENOENT return nil, fuse.ENOENT
} }
return fileNode{File: ret}, nil return fileNode{FileBase: ret}, nil
} }
func (f fileNode) ReadAll(ctx context.Context) ([]byte, error) { func (f fileNode) ReadAll(ctx context.Context) ([]byte, error) {
if f.IsRegular() { if f.IsRegular() {
rdr, err := f.GetFullReader(f.r)
if err != nil {
return nil, err
}
var buf bytes.Buffer var buf bytes.Buffer
_, err := f.WriteTo(&buf) _, err = rdr.WriteTo(&buf)
return buf.Bytes(), err return buf.Bytes(), err
} }
return nil, ENODATA return nil, ENODATA
@@ -103,8 +140,13 @@ func (f fileNode) ReadAll(ctx context.Context) ([]byte, error) {
func (f fileNode) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error { func (f fileNode) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error {
if f.IsRegular() { if f.IsRegular() {
rdr, err := f.GetReader(f.r)
if err != nil {
return err
}
buf := make([]byte, req.Size) buf := make([]byte, req.Size)
n, err := f.File.ReadAt(buf, req.Offset) rdr.Read(make([]byte, req.Offset))
n, err := rdr.Read(buf)
if err == io.EOF { if err == io.EOF {
resp.Data = buf[:n] resp.Data = buf[:n]
} }
@@ -114,13 +156,13 @@ func (f fileNode) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.Re
} }
func (f fileNode) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error) { func (f fileNode) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error) {
asFS, err := f.FS() asFS, err := f.ToDir(f.r)
if err != nil { if err != nil {
return nil, fuse.ENOTDIR return nil, fuse.ENOTDIR
} }
var t fuse.DirentType var t fuse.DirentType
for i := range asFS.e { for i := range asFS.Entries {
switch asFS.e[i].Type { switch asFS.Entries[i].InodeType {
case inode.Fil: case inode.Fil:
t = fuse.DT_File t = fuse.DT_File
case inode.Dir: case inode.Dir:
@@ -139,9 +181,9 @@ func (f fileNode) ReadDirAll(ctx context.Context) (out []fuse.Dirent, err error)
t = fuse.DT_Unknown t = fuse.DT_Unknown
} }
out = append(out, fuse.Dirent{ out = append(out, fuse.Dirent{
Inode: uint64(asFS.e[i].Num), Inode: uint64(asFS.Entries[i].Num),
Type: t, Type: t,
Name: asFS.e[i].Name, Name: asFS.Entries[i].Name,
}) })
} }
return return
+1 -1
View File
@@ -1,4 +1,4 @@
package squashfs package squashfuse
import ( import (
"golang.org/x/sys/unix" "golang.org/x/sys/unix"
+1 -1
View File
@@ -1,4 +1,4 @@
package squashfs package squashfuse
import "github.com/CalebQ42/fuse" import "github.com/CalebQ42/fuse"
+1 -1
View File
@@ -1,3 +1,3 @@
package squashfs package squashfuse
var ENODATA = windows.Errno(windows.ENODATA) var ENODATA = windows.Errno(windows.ENODATA)
+20
View File
@@ -0,0 +1,20 @@
module github.com/CalebQ42/squashfuse
go 1.21.5
require (
github.com/CalebQ42/fuse v0.1.0
github.com/CalebQ42/squashfs v0.8.4
github.com/seaweedfs/fuse v1.2.3
golang.org/x/sys v0.15.0
)
require (
github.com/klauspost/compress v1.17.4 // indirect
github.com/pierrec/lz4/v4 v4.1.19 // indirect
github.com/rasky/go-lzo v0.0.0-20200203143853-96a758eda86e // indirect
github.com/therootcompany/xz v1.0.1 // indirect
github.com/ulikunitz/xz v0.5.11 // indirect
)
replace github.com/CalebQ42/squashfs => ../squashfs
+22
View File
@@ -0,0 +1,22 @@
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/CalebQ42/squashfs v0.8.4 h1:HnthgRKuLliiMwYsPTSE/ln2zECt7UelYcbsUc5p+PA=
github.com/CalebQ42/squashfs v0.8.4/go.mod h1:CmGHRknB7BlYJ49qSTGpW8wnFcGFdZW0l6+qHOvFr5c=
github.com/klauspost/compress v1.16.7 h1:2mk3MPGNzKyxErAw8YaohYh69+pa4sIQSC0fPGCFR9I=
github.com/klauspost/compress v1.16.7/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
github.com/klauspost/compress v1.17.4 h1:Ej5ixsIri7BrIjBkRZLTo6ghwrEtHFk7ijlczPW4fZ4=
github.com/klauspost/compress v1.17.4/go.mod h1:/dCuZOvVtNoHsyb+cuJD3itjs3NbnF6KH9zAO4BDxPM=
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/pierrec/lz4/v4 v4.1.19 h1:tYLzDnjDXh9qIxSTKHwXwOYmm9d887Y7Y1ZkyXYHAN4=
github.com/pierrec/lz4/v4 v4.1.19/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.3 h1:VH4VF9D3yvuQBILqDbNttz7Whjgo3JBLfpZeecmYfm0=
github.com/seaweedfs/fuse v1.2.3/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.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.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc=
golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=