forked from Elara6331/itd
Compare commits
27 Commits
2c899b4660
...
fuse
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
008f6b35a9 | ||
|
|
98775600af | ||
|
|
613d33ff4d | ||
|
|
1a1bc30df9 | ||
|
|
c5a6e0d298 | ||
|
|
84c7a33c40 | ||
|
|
dc53ead339 | ||
|
|
9bbdc3bd52 | ||
|
|
b28c386c4e | ||
|
|
4c59561a99 | ||
|
|
955e1323ce | ||
|
|
3b9690103b | ||
|
|
b5328ece92 | ||
|
|
2396623c73 | ||
|
|
a54ca7afdf | ||
|
|
673383f795 | ||
|
|
cc6fc3e1dc | ||
|
|
c05d4fe951 | ||
|
|
87c78566c1 | ||
|
|
8dd8f3d012 | ||
|
|
bf13b96660 | ||
|
|
2440cb954c | ||
|
|
08c8d7e8de | ||
|
|
1799c072ff | ||
|
|
c5ca30cd0a | ||
|
|
c046c67dbd | ||
|
|
e2bd52b5a0 |
560
fuse.go
560
fuse.go
@@ -1,537 +1,33 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"go.arsenm.dev/infinitime"
|
"go.arsenm.dev/itd/internal/fusefs"
|
||||||
"go.arsenm.dev/infinitime/blefs"
|
|
||||||
"go.arsenm.dev/logger/log"
|
|
||||||
"os"
|
"os"
|
||||||
"context"
|
|
||||||
"syscall"
|
|
||||||
"github.com/hanwen/go-fuse/v2/fs"
|
"github.com/hanwen/go-fuse/v2/fs"
|
||||||
"github.com/hanwen/go-fuse/v2/fuse"
|
"github.com/hanwen/go-fuse/v2/fuse"
|
||||||
"strconv"
|
"go.arsenm.dev/logger/log"
|
||||||
"io"
|
"context"
|
||||||
"bytes"
|
"go.arsenm.dev/infinitime"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Device struct {
|
func startFUSE(ctx context.Context, dev *infinitime.Device) error {
|
||||||
dev *infinitime.Device
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i Device) HeartRateBytes() ([]byte, error) {
|
|
||||||
v, err := i.dev.HeartRate()
|
|
||||||
return []byte(strconv.Itoa(int(v)) + "\n"), err
|
|
||||||
}
|
|
||||||
func (i Device) BatteryLevelBytes() ([]byte, error) {
|
|
||||||
v, err := i.dev.BatteryLevel()
|
|
||||||
return []byte(strconv.Itoa(int(v)) + "\n"), err
|
|
||||||
}
|
|
||||||
func (i Device) StepCountBytes() ([]byte, error) {
|
|
||||||
v, err := i.dev.StepCount()
|
|
||||||
return []byte(strconv.Itoa(int(v)) + "\n"), err
|
|
||||||
}
|
|
||||||
func (i Device) MotionBytes() ([]byte, error) {
|
|
||||||
v, err := i.dev.Motion()
|
|
||||||
return []byte(strconv.Itoa(int(v.X)) + " " + strconv.Itoa(int(v.Y)) + " " + strconv.Itoa(int(v.Z)) + "\n"), err
|
|
||||||
}
|
|
||||||
func (i Device) AddressBytes() ([]byte, error) {
|
|
||||||
v := i.dev.Address()
|
|
||||||
return []byte(v + "\n"), nil
|
|
||||||
}
|
|
||||||
func (i Device) VersionBytes() ([]byte, error) {
|
|
||||||
v, err := i.dev.Version()
|
|
||||||
return []byte(v + "\n"), err
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
type ITProperty struct {
|
|
||||||
name string
|
|
||||||
Ino uint64
|
|
||||||
gen func(context.Context) (<-chan uint8, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
type DirEntry struct {
|
|
||||||
isDir bool
|
|
||||||
modtime uint64
|
|
||||||
size uint32
|
|
||||||
path string
|
|
||||||
}
|
|
||||||
|
|
||||||
type ITNode struct {
|
|
||||||
fs.Inode
|
|
||||||
kind int
|
|
||||||
Ino uint64
|
|
||||||
|
|
||||||
lst []DirEntry
|
|
||||||
self DirEntry
|
|
||||||
path string
|
|
||||||
}
|
|
||||||
|
|
||||||
var properties = []ITProperty {
|
|
||||||
ITProperty{"heartrate", 2, nil},
|
|
||||||
ITProperty{"battery", 3, nil},
|
|
||||||
ITProperty{"motion", 4, nil},
|
|
||||||
ITProperty{"stepcount", 5, nil},
|
|
||||||
ITProperty{"version", 6, nil},
|
|
||||||
ITProperty{"address", 7, nil},
|
|
||||||
}
|
|
||||||
var myfs *blefs.FS = nil;
|
|
||||||
var inodemap map[string]uint64 = nil;
|
|
||||||
|
|
||||||
var _ = (fs.NodeReaddirer)((*ITNode)(nil))
|
|
||||||
|
|
||||||
// Readdir is part of the NodeReaddirer interface
|
|
||||||
func (n *ITNode) Readdir(ctx context.Context) (fs.DirStream, syscall.Errno) {
|
|
||||||
switch n.kind {
|
|
||||||
case 0:
|
|
||||||
// root folder
|
|
||||||
r := make([]fuse.DirEntry, 2)
|
|
||||||
r[0] = fuse.DirEntry{
|
|
||||||
Name: "device",
|
|
||||||
Ino: 0,
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
}
|
|
||||||
r[1] = fuse.DirEntry{
|
|
||||||
Name: "fs",
|
|
||||||
Ino: 1,
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
}
|
|
||||||
return fs.NewListDirStream(r), 0
|
|
||||||
|
|
||||||
case 1:
|
|
||||||
// device folder
|
|
||||||
r := make([]fuse.DirEntry, 6)
|
|
||||||
for ind, value := range properties {
|
|
||||||
r[ind] = fuse.DirEntry{
|
|
||||||
Name: value.name,
|
|
||||||
Ino: value.Ino,
|
|
||||||
Mode: fuse.S_IFREG,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return fs.NewListDirStream(r), 0
|
|
||||||
|
|
||||||
case 2:
|
|
||||||
// on device
|
|
||||||
files, _ := myfs.ReadDir(n.path)
|
|
||||||
log.Info("readdir").Str("path", n.path).Int("objects", len(files)).Send()
|
|
||||||
r := make([]fuse.DirEntry, len(files))
|
|
||||||
n.lst = make([]DirEntry, len(files))
|
|
||||||
for ind, entry := range files {
|
|
||||||
info, _ := entry.Info()
|
|
||||||
name := info.Name()
|
|
||||||
|
|
||||||
file := DirEntry{
|
|
||||||
path: n.path + "/" + name,
|
|
||||||
size: uint32(info.Size()),
|
|
||||||
modtime: uint64(info.ModTime().Unix()),
|
|
||||||
isDir: info.IsDir(),
|
|
||||||
}
|
|
||||||
n.lst[ind] = file
|
|
||||||
|
|
||||||
ino := inodemap[file.path]
|
|
||||||
if ino == 0 {
|
|
||||||
ino = uint64(len(inodemap)) + 1
|
|
||||||
inodemap[file.path] = ino
|
|
||||||
}
|
|
||||||
|
|
||||||
if file.isDir {
|
|
||||||
r[ind] = fuse.DirEntry{
|
|
||||||
Name: name,
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
Ino : ino + 10,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
r[ind] = fuse.DirEntry{
|
|
||||||
Name: name,
|
|
||||||
Mode: fuse.S_IFREG,
|
|
||||||
Ino : ino + 10,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return fs.NewListDirStream(r), 0
|
|
||||||
}
|
|
||||||
r := make([]fuse.DirEntry, 0)
|
|
||||||
return fs.NewListDirStream(r), 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeLookuper)((*ITNode)(nil))
|
|
||||||
func (n *ITNode) Lookup(ctx context.Context, name string, out *fuse.EntryOut) (*fs.Inode, syscall.Errno) {
|
|
||||||
switch n.kind {
|
|
||||||
case 0:
|
|
||||||
// root folder
|
|
||||||
if name == "device" {
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
Ino: uint64(0),
|
|
||||||
}
|
|
||||||
operations := &ITNode{kind: 1, Ino: 0}
|
|
||||||
child := n.NewInode(ctx, operations, stable)
|
|
||||||
return child, 0
|
|
||||||
} else if name == "fs" {
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
Ino: uint64(1),
|
|
||||||
}
|
|
||||||
operations := &ITNode{kind: 2, Ino: 1, path : ""}
|
|
||||||
child := n.NewInode(ctx, operations, stable)
|
|
||||||
return child, 0
|
|
||||||
}
|
|
||||||
case 1:
|
|
||||||
// device folder
|
|
||||||
for _, value := range properties {
|
|
||||||
if value.name == name {
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFREG,
|
|
||||||
Ino: uint64(value.Ino),
|
|
||||||
}
|
|
||||||
operations := &ITNode{kind: 3, Ino: value.Ino}
|
|
||||||
child := n.NewInode(ctx, operations, stable)
|
|
||||||
return child, 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil, syscall.ENOENT
|
|
||||||
|
|
||||||
case 2:
|
|
||||||
// FS object
|
|
||||||
if len(n.lst) == 0 {
|
|
||||||
n.Readdir(ctx)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, file := range n.lst {
|
|
||||||
if file.path != n.path + "/" + name {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
log.Info("LookUp successful").Str("path", file.path).Send()
|
|
||||||
|
|
||||||
if file.isDir {
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
Ino: inodemap[file.path],
|
|
||||||
}
|
|
||||||
operations := &ITNode{kind: 2, path: file.path}
|
|
||||||
child := n.NewInode(ctx, operations, stable)
|
|
||||||
return child, 0
|
|
||||||
} else {
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFREG,
|
|
||||||
Ino: inodemap[file.path],
|
|
||||||
}
|
|
||||||
operations := &ITNode{
|
|
||||||
kind: 2, path: file.path,
|
|
||||||
self: file,
|
|
||||||
}
|
|
||||||
child := n.NewInode(ctx, operations, stable)
|
|
||||||
return child, 0
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
log.Warn("LookUp failed").Str("path", n.path + "/" + name).Send()
|
|
||||||
}
|
|
||||||
return nil, syscall.ENOENT
|
|
||||||
}
|
|
||||||
|
|
||||||
type bytesFileReadHandle struct {
|
|
||||||
content []byte
|
|
||||||
}
|
|
||||||
var _ = (fs.FileReader)((*bytesFileReadHandle)(nil))
|
|
||||||
|
|
||||||
func (fh *bytesFileReadHandle) Read(ctx context.Context, dest []byte, off int64) (fuse.ReadResult, syscall.Errno) {
|
|
||||||
log.Info("Executing Read").Int("size", len(fh.content)).Send()
|
|
||||||
end := off + int64(len(dest))
|
|
||||||
if end > int64(len(fh.content)) {
|
|
||||||
end = int64(len(fh.content))
|
|
||||||
}
|
|
||||||
return fuse.ReadResultData(fh.content[off:end]), 0
|
|
||||||
}
|
|
||||||
|
|
||||||
type sensorFileReadHandle struct {
|
|
||||||
ch <-chan uint8
|
|
||||||
}
|
|
||||||
var _ = (fs.FileReader)((*sensorFileReadHandle)(nil))
|
|
||||||
func (fh *sensorFileReadHandle) Read(ctx context.Context, dest []byte, off int64) (fuse.ReadResult, syscall.Errno) {
|
|
||||||
v := <-fh.ch
|
|
||||||
content := []byte(strconv.Itoa(int(v)) + "\n")
|
|
||||||
return fuse.ReadResultData(content), 0
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
type bytesFileWriteHandle struct {
|
|
||||||
content []byte
|
|
||||||
path string
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.FileWriter)((*bytesFileWriteHandle)(nil))
|
|
||||||
func (fh *bytesFileWriteHandle) Write(ctx context.Context, data []byte, off int64) (written uint32, errno syscall.Errno) {
|
|
||||||
log.Info("Executing Write").Str("path", fh.path).Int("prev_size", len(fh.content)).Int("next_size", len(data)).Send()
|
|
||||||
if off != int64(len(fh.content)) {
|
|
||||||
}
|
|
||||||
fh.content = append(fh.content[:], data[:]...)
|
|
||||||
return uint32(len(data)), 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.FileFlusher)((*bytesFileWriteHandle)(nil))
|
|
||||||
func (fh *bytesFileWriteHandle) Flush(ctx context.Context) (errno syscall.Errno) {
|
|
||||||
if len(fh.content) == 0 {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Info("Attempting flush").Str("path", fh.path).Send()
|
|
||||||
fp, err := myfs.Create(fh.path, uint32(len(fh.content)))
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Flush failed: create").Str("path", fh.path).Err(err).Send()
|
|
||||||
return syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
// For every progress event
|
|
||||||
for sent := range fp.Progress() {
|
|
||||||
log.Info("Progress").Int("bytes", int(sent)).Int("of", len(fh.content)).Send();
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
r := bytes.NewReader(fh.content)
|
|
||||||
nread, err := io.Copy(fp, r)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Flush failed: write").Str("path", fh.path).Err(err).Send()
|
|
||||||
fp.Close()
|
|
||||||
return syscall.EROFS
|
|
||||||
}
|
|
||||||
if int(nread) != len(fh.content) {
|
|
||||||
log.Error("Flush failed: write").Str("path", fh.path).Int("expect", len(fh.content)).Int("got", int(nread)).Send()
|
|
||||||
fp.Close()
|
|
||||||
return syscall.EROFS
|
|
||||||
}
|
|
||||||
err = fp.Close()
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Flush failed: close").Str("path", fh.path).Err(err).Send()
|
|
||||||
return syscall.EROFS
|
|
||||||
}
|
|
||||||
log.Info("Flush done").Str("path", fh.path).Int("size", len(fh.content)).Send()
|
|
||||||
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
var _ = (fs.FileFsyncer)((*bytesFileWriteHandle)(nil))
|
|
||||||
func (fh *bytesFileWriteHandle) Fsync(ctx context.Context, flags uint32) (errno syscall.Errno) {
|
|
||||||
return fh.Flush(ctx)
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeGetattrer)((*ITNode)(nil))
|
|
||||||
func (bn *ITNode) Getattr(ctx context.Context, f fs.FileHandle, out *fuse.AttrOut) syscall.Errno {
|
|
||||||
log.Info("getattr").Str("path", bn.path).Send();
|
|
||||||
out.Ino = bn.Ino
|
|
||||||
out.Mtime = bn.self.modtime
|
|
||||||
out.Ctime = bn.self.modtime
|
|
||||||
out.Atime = bn.self.modtime
|
|
||||||
out.Size = uint64(bn.self.size)
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeSetattrer)((*ITNode)(nil))
|
|
||||||
func (bn *ITNode) Setattr(ctx context.Context, fh fs.FileHandle, in *fuse.SetAttrIn, out *fuse.AttrOut) syscall.Errno {
|
|
||||||
log.Info("setattr").Str("path", bn.path).Send()
|
|
||||||
out.Size = 0;
|
|
||||||
out.Mtime = 0;
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeOpener)((*ITNode)(nil))
|
|
||||||
func (f *ITNode) Open(ctx context.Context, openFlags uint32) (fh fs.FileHandle, fuseFlags uint32, errno syscall.Errno) {
|
|
||||||
switch f.kind {
|
|
||||||
case 2:
|
|
||||||
// FS file
|
|
||||||
if openFlags&syscall.O_RDWR != 0 {
|
|
||||||
log.Warn("open: failed RDWR").Str("path", f.path).Send()
|
|
||||||
return nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
if openFlags & syscall.O_WRONLY != 0 {
|
|
||||||
log.Info("Opening file: write").Str("path", f.path).Send()
|
|
||||||
fh = &bytesFileWriteHandle{
|
|
||||||
path : f.path,
|
|
||||||
content : make([]byte, 0),
|
|
||||||
}
|
|
||||||
return fh, fuse.FOPEN_DIRECT_IO, 0
|
|
||||||
} else {
|
|
||||||
log.Info("Opening file: read").Str("path", f.path).Send();
|
|
||||||
fp, err := myfs.Open(f.path)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Opening file failed").Str("path", f.path).Err(err).Send();
|
|
||||||
return nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
defer fp.Close()
|
|
||||||
|
|
||||||
b := &bytes.Buffer{}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
// For every progress event
|
|
||||||
for sent := range fp.Progress() {
|
|
||||||
log.Info("Progress").Int("bytes", int(sent)).Int("of", int(f.self.size)).Send();
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
_, err = io.Copy(b, fp)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Read failed").Str("path", f.path).Err(err).Send()
|
|
||||||
fp.Close()
|
|
||||||
return nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
fh = &bytesFileReadHandle{
|
|
||||||
content: b.Bytes(),
|
|
||||||
}
|
|
||||||
return fh, fuse.FOPEN_DIRECT_IO, 0
|
|
||||||
}
|
|
||||||
|
|
||||||
case 3:
|
|
||||||
// Device file
|
|
||||||
|
|
||||||
// disallow writes
|
|
||||||
if openFlags&(syscall.O_RDWR|syscall.O_WRONLY) != 0 {
|
|
||||||
return nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, value := range properties {
|
|
||||||
if value.Ino == f.Ino {
|
|
||||||
ans, err := value.gen(ctx)
|
|
||||||
if err != nil {
|
|
||||||
return nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
fh = &sensorFileReadHandle{
|
|
||||||
ch: ans,
|
|
||||||
}
|
|
||||||
return fh, fuse.FOPEN_DIRECT_IO, 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeCreater)((*ITNode)(nil))
|
|
||||||
func (f *ITNode) Create(ctx context.Context, name string, flags uint32, mode uint32, out *fuse.EntryOut) (node *fs.Inode, fh fs.FileHandle, fuseFlags uint32, errno syscall.Errno) {
|
|
||||||
if f.kind != 2 {
|
|
||||||
return nil, nil, 0, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
path := f.path + "/" + name
|
|
||||||
ino := uint64(len(inodemap)) + 11
|
|
||||||
inodemap[path] = ino
|
|
||||||
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFREG,
|
|
||||||
Ino: ino,
|
|
||||||
}
|
|
||||||
operations := &ITNode{
|
|
||||||
kind: 2, Ino: ino,
|
|
||||||
path : path,
|
|
||||||
}
|
|
||||||
node = f.NewInode(ctx, operations, stable)
|
|
||||||
|
|
||||||
fh = &bytesFileWriteHandle{
|
|
||||||
path : path,
|
|
||||||
content : make([]byte, 0),
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Info("Creating file").Str("path", path).Send()
|
|
||||||
|
|
||||||
errno = 0
|
|
||||||
return node, fh, fuseFlags, 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeMkdirer)((*ITNode)(nil))
|
|
||||||
func (f *ITNode) Mkdir(ctx context.Context, name string, mode uint32, out *fuse.EntryOut) (*fs.Inode, syscall.Errno) {
|
|
||||||
if f.kind != 2 {
|
|
||||||
return nil, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
path := f.path + "/" + name
|
|
||||||
err := myfs.Mkdir(path)
|
|
||||||
if err != nil {
|
|
||||||
log.Info("Mkdir failed").
|
|
||||||
Str("path", path).
|
|
||||||
Err(err).
|
|
||||||
Send()
|
|
||||||
return nil, syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
ino := uint64(len(inodemap)) + 11
|
|
||||||
inodemap[path] = ino
|
|
||||||
|
|
||||||
stable := fs.StableAttr{
|
|
||||||
Mode: fuse.S_IFDIR,
|
|
||||||
Ino: ino,
|
|
||||||
}
|
|
||||||
operations := &ITNode{
|
|
||||||
kind: 2, Ino: ino,
|
|
||||||
path : path,
|
|
||||||
}
|
|
||||||
node := f.NewInode(ctx, operations, stable)
|
|
||||||
|
|
||||||
log.Info("Mkdir sucess").
|
|
||||||
Str("path", path).
|
|
||||||
Int("ino", int(ino)).
|
|
||||||
Send()
|
|
||||||
return node, 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeRenamer)((*ITNode)(nil))
|
|
||||||
func (f *ITNode) Rename(ctx context.Context, name string, newParent fs.InodeEmbedder, newName string, flags uint32) syscall.Errno {
|
|
||||||
p1 := f.path + "/" + name
|
|
||||||
p2 := newParent.EmbeddedInode().Path(nil)[2:] + "/" + newName
|
|
||||||
|
|
||||||
err := myfs.Rename(p1, p2)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Rename failed").
|
|
||||||
Str("src", p1).
|
|
||||||
Str("dest", p2).
|
|
||||||
Err(err).
|
|
||||||
Send()
|
|
||||||
|
|
||||||
return syscall.EROFS
|
|
||||||
}
|
|
||||||
log.Info("Rename sucess").
|
|
||||||
Str("src", p1).
|
|
||||||
Str("dest", p2).
|
|
||||||
Send()
|
|
||||||
|
|
||||||
ino := inodemap[p1]
|
|
||||||
delete(inodemap, p1)
|
|
||||||
inodemap[p2] = ino
|
|
||||||
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeUnlinker)((*ITNode)(nil))
|
|
||||||
func (f *ITNode) Unlink(ctx context.Context, name string) syscall.Errno {
|
|
||||||
delete(inodemap, f.path + "/" + name)
|
|
||||||
err := myfs.Remove(f.path + "/" + name)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Unlink failed").
|
|
||||||
Str("file", f.path + "/" + name).
|
|
||||||
Err(err).
|
|
||||||
Send()
|
|
||||||
|
|
||||||
return syscall.EROFS
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Info("Unlink success").
|
|
||||||
Str("file", f.path + "/" + name).
|
|
||||||
Send()
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ = (fs.NodeRmdirer)((*ITNode)(nil))
|
|
||||||
func (f *ITNode) Rmdir(ctx context.Context, name string) syscall.Errno {
|
|
||||||
return f.Unlink(ctx, name)
|
|
||||||
}
|
|
||||||
|
|
||||||
func startFuse(ctx context.Context, dev *infinitime.Device) error {
|
|
||||||
// This is where we'll mount the FS
|
// This is where we'll mount the FS
|
||||||
os.Mkdir(k.String("fuse.mountpoint"), 0755)
|
err := os.MkdirAll(k.String("fuse.mountpoint"), 0755)
|
||||||
root := &ITNode{kind: 0}
|
if err != nil && !os.IsExist(err) {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ignore the error because nothing might be mounted on the mountpoint
|
||||||
|
_ = fusefs.Unmount(k.String("fuse.mountpoint"))
|
||||||
|
|
||||||
|
|
||||||
|
root, err := fusefs.BuildRootNode(dev)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Building root node failed").
|
||||||
|
Err(err).
|
||||||
|
Send()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
server, err := fs.Mount(k.String("fuse.mountpoint"), root, &fs.Options{
|
server, err := fs.Mount(k.String("fuse.mountpoint"), root, &fs.Options{
|
||||||
MountOptions: fuse.MountOptions{
|
MountOptions: fuse.MountOptions{
|
||||||
// Set to true to see how the file system works.
|
// Set to true to see how the file system works.
|
||||||
@@ -551,22 +47,12 @@ func startFuse(ctx context.Context, dev *infinitime.Device) error {
|
|||||||
Str("target", k.String("fuse.mountpoint")).
|
Str("target", k.String("fuse.mountpoint")).
|
||||||
Send()
|
Send()
|
||||||
|
|
||||||
mydev := Device{dev : dev};
|
fusefs.BuildProperties(dev)
|
||||||
|
|
||||||
properties[0].gen = mydev.dev.WatchHeartRate;
|
|
||||||
// properties[0].f = mydev.HeartRateBytes;
|
|
||||||
// properties[1].f = mydev.BatteryLevelBytes;
|
|
||||||
// properties[2].f = mydev.MotionBytes;
|
|
||||||
// properties[3].f = mydev.StepCountBytes;
|
|
||||||
// properties[4].f = mydev.VersionBytes;
|
|
||||||
// properties[5].f = mydev.AddressBytes;
|
|
||||||
|
|
||||||
myfs, err = dev.FS()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Warn("Error getting BLE filesystem").Err(err).Send()
|
log.Warn("Error getting BLE filesystem").Err(err).Send()
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
inodemap = make(map[string]uint64)
|
|
||||||
|
|
||||||
// Wait until unmount before exiting
|
// Wait until unmount before exiting
|
||||||
go server.Serve()
|
go server.Serve()
|
||||||
|
|||||||
1
go.sum
1
go.sum
@@ -341,7 +341,6 @@ github.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfn
|
|||||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
||||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
github.com/kylelemons/godebug v0.0.0-20170820004349-d65d576e9348/go.mod h1:B69LEHPfb2qLo0BaaOLcbitczOKLWTsrBG9LczfCD4k=
|
|
||||||
github.com/lucor/goinfo v0.0.0-20210802170112-c078a2b0f08b/go.mod h1:PRq09yoB+Q2OJReAmwzKivcYyremnibWGbK7WfftHzc=
|
github.com/lucor/goinfo v0.0.0-20210802170112-c078a2b0f08b/go.mod h1:PRq09yoB+Q2OJReAmwzKivcYyremnibWGbK7WfftHzc=
|
||||||
github.com/magiconair/properties v1.8.5/go.mod h1:y3VJvCyxH9uVvJTWEGAELF3aiYNyPKd5NZ3oSwXrF60=
|
github.com/magiconair/properties v1.8.5/go.mod h1:y3VJvCyxH9uVvJTWEGAELF3aiYNyPKd5NZ3oSwXrF60=
|
||||||
github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU=
|
github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU=
|
||||||
|
|||||||
572
internal/fusefs/fuse.go
Normal file
572
internal/fusefs/fuse.go
Normal file
@@ -0,0 +1,572 @@
|
|||||||
|
package fusefs
|
||||||
|
|
||||||
|
import (
|
||||||
|
"go.arsenm.dev/infinitime"
|
||||||
|
"go.arsenm.dev/infinitime/blefs"
|
||||||
|
"go.arsenm.dev/logger/log"
|
||||||
|
"context"
|
||||||
|
"syscall"
|
||||||
|
"github.com/hanwen/go-fuse/v2/fs"
|
||||||
|
"github.com/hanwen/go-fuse/v2/fuse"
|
||||||
|
"io"
|
||||||
|
"bytes"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ITProperty struct {
|
||||||
|
name string
|
||||||
|
Ino uint64
|
||||||
|
gen func() ([]byte, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
type DirEntry struct {
|
||||||
|
isDir bool
|
||||||
|
modtime uint64
|
||||||
|
size uint32
|
||||||
|
path string
|
||||||
|
}
|
||||||
|
|
||||||
|
type ITNode struct {
|
||||||
|
fs.Inode
|
||||||
|
kind int
|
||||||
|
Ino uint64
|
||||||
|
|
||||||
|
lst []DirEntry
|
||||||
|
self DirEntry
|
||||||
|
path string
|
||||||
|
}
|
||||||
|
|
||||||
|
var myfs *blefs.FS = nil
|
||||||
|
var inodemap map[string]uint64 = nil
|
||||||
|
|
||||||
|
func BuildRootNode(dev *infinitime.Device) (*ITNode, error) {
|
||||||
|
var err error
|
||||||
|
inodemap = make(map[string]uint64)
|
||||||
|
myfs, err = dev.FS()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Failed to get filesystem").Err(err).Send()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ITNode{kind: 0}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var properties = make([]ITProperty, 6)
|
||||||
|
|
||||||
|
func BuildProperties(dev *infinitime.Device) {
|
||||||
|
properties[0] = ITProperty{"heartrate", 2,
|
||||||
|
func() ([]byte, error) {
|
||||||
|
ans, err := dev.HeartRate()
|
||||||
|
return []byte(strconv.Itoa(int(ans)) + "\n"), err
|
||||||
|
}}
|
||||||
|
properties[1] = ITProperty{"battery", 3,
|
||||||
|
func() ([]byte, error) {
|
||||||
|
ans, err := dev.BatteryLevel()
|
||||||
|
return []byte(strconv.Itoa(int(ans)) + "\n"), err
|
||||||
|
}}
|
||||||
|
properties[2] = ITProperty{"motion", 4,
|
||||||
|
func() ([]byte, error) {
|
||||||
|
ans, err := dev.Motion()
|
||||||
|
return []byte(strconv.Itoa(int(ans.X)) + " " + strconv.Itoa(int(ans.Y)) + " " + strconv.Itoa(int(ans.Z)) + "\n"), err
|
||||||
|
}}
|
||||||
|
properties[3] = ITProperty{"stepcount", 6,
|
||||||
|
func() ([]byte, error) {
|
||||||
|
ans, err := dev.StepCount()
|
||||||
|
return []byte(strconv.Itoa(int(ans)) + "\n"), err
|
||||||
|
}}
|
||||||
|
properties[4] = ITProperty{"version", 7,
|
||||||
|
func() ([]byte, error) {
|
||||||
|
ans, err := dev.Version()
|
||||||
|
return []byte(ans + "\n"), err
|
||||||
|
}}
|
||||||
|
properties[5] = ITProperty{"address", 8,
|
||||||
|
func() ([]byte, error) {
|
||||||
|
ans := dev.Address()
|
||||||
|
return []byte(ans + "\n"), nil
|
||||||
|
}}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
var _ fs.NodeReaddirer = (*ITNode)(nil)
|
||||||
|
|
||||||
|
// Readdir is part of the NodeReaddirer interface
|
||||||
|
func (n *ITNode) Readdir(ctx context.Context) (fs.DirStream, syscall.Errno) {
|
||||||
|
switch n.kind {
|
||||||
|
case 0:
|
||||||
|
// root folder
|
||||||
|
r := make([]fuse.DirEntry, 2)
|
||||||
|
r[0] = fuse.DirEntry{
|
||||||
|
Name: "info",
|
||||||
|
Ino: 0,
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
}
|
||||||
|
r[1] = fuse.DirEntry{
|
||||||
|
Name: "fs",
|
||||||
|
Ino: 1,
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
}
|
||||||
|
return fs.NewListDirStream(r), 0
|
||||||
|
|
||||||
|
case 1:
|
||||||
|
// info folder
|
||||||
|
r := make([]fuse.DirEntry, 6)
|
||||||
|
for ind, value := range properties {
|
||||||
|
r[ind] = fuse.DirEntry{
|
||||||
|
Name: value.name,
|
||||||
|
Ino: value.Ino,
|
||||||
|
Mode: fuse.S_IFREG,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return fs.NewListDirStream(r), 0
|
||||||
|
|
||||||
|
case 2:
|
||||||
|
// on info
|
||||||
|
files, err := myfs.ReadDir(n.path)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE ReadDir failed").Str("path", n.path).Err(err).Send()
|
||||||
|
return nil, syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("FUSE ReadDir succeeded").Str("path", n.path).Int("objects", len(files)).Send()
|
||||||
|
r := make([]fuse.DirEntry, len(files))
|
||||||
|
n.lst = make([]DirEntry, len(files))
|
||||||
|
for ind, entry := range files {
|
||||||
|
info, err := entry.Info()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Info failed").Str("path", n.path).Err(err).Send()
|
||||||
|
return nil, syscallErr(err)
|
||||||
|
}
|
||||||
|
name := info.Name()
|
||||||
|
|
||||||
|
file := DirEntry{
|
||||||
|
path: n.path + "/" + name,
|
||||||
|
size: uint32(info.Size()),
|
||||||
|
modtime: uint64(info.ModTime().Unix()),
|
||||||
|
isDir: info.IsDir(),
|
||||||
|
}
|
||||||
|
n.lst[ind] = file
|
||||||
|
|
||||||
|
ino := inodemap[file.path]
|
||||||
|
if ino == 0 {
|
||||||
|
ino = uint64(len(inodemap)) + 1
|
||||||
|
inodemap[file.path] = ino
|
||||||
|
}
|
||||||
|
|
||||||
|
if file.isDir {
|
||||||
|
r[ind] = fuse.DirEntry{
|
||||||
|
Name: name,
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
Ino : ino + 10,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
r[ind] = fuse.DirEntry{
|
||||||
|
Name: name,
|
||||||
|
Mode: fuse.S_IFREG,
|
||||||
|
Ino : ino + 10,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return fs.NewListDirStream(r), 0
|
||||||
|
}
|
||||||
|
r := make([]fuse.DirEntry, 0)
|
||||||
|
return fs.NewListDirStream(r), 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeLookuper = (*ITNode)(nil)
|
||||||
|
func (n *ITNode) Lookup(ctx context.Context, name string, out *fuse.EntryOut) (*fs.Inode, syscall.Errno) {
|
||||||
|
switch n.kind {
|
||||||
|
case 0:
|
||||||
|
// root folder
|
||||||
|
if name == "info" {
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
Ino: uint64(0),
|
||||||
|
}
|
||||||
|
operations := &ITNode{kind: 1, Ino: 0}
|
||||||
|
child := n.NewInode(ctx, operations, stable)
|
||||||
|
return child, 0
|
||||||
|
} else if name == "fs" {
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
Ino: uint64(1),
|
||||||
|
}
|
||||||
|
operations := &ITNode{kind: 2, Ino: 1, path : ""}
|
||||||
|
child := n.NewInode(ctx, operations, stable)
|
||||||
|
return child, 0
|
||||||
|
}
|
||||||
|
case 1:
|
||||||
|
// info folder
|
||||||
|
for _, value := range properties {
|
||||||
|
if value.name == name {
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFREG,
|
||||||
|
Ino: uint64(value.Ino),
|
||||||
|
}
|
||||||
|
operations := &ITNode{kind: 3, Ino: value.Ino}
|
||||||
|
child := n.NewInode(ctx, operations, stable)
|
||||||
|
return child, 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case 2:
|
||||||
|
// FS object
|
||||||
|
if len(n.lst) == 0 {
|
||||||
|
n.Readdir(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, file := range n.lst {
|
||||||
|
if file.path != n.path + "/" + name {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
log.Debug("FUSE Lookup successful").Str("path", file.path).Send()
|
||||||
|
|
||||||
|
if file.isDir {
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
Ino: inodemap[file.path],
|
||||||
|
}
|
||||||
|
operations := &ITNode{kind: 2, path: file.path}
|
||||||
|
child := n.NewInode(ctx, operations, stable)
|
||||||
|
return child, 0
|
||||||
|
} else {
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFREG,
|
||||||
|
Ino: inodemap[file.path],
|
||||||
|
}
|
||||||
|
operations := &ITNode{
|
||||||
|
kind: 2, path: file.path,
|
||||||
|
self: file,
|
||||||
|
}
|
||||||
|
child := n.NewInode(ctx, operations, stable)
|
||||||
|
return child, 0
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
log.Warn("FUSE Lookup failed").Str("path", n.path + "/" + name).Send()
|
||||||
|
}
|
||||||
|
return nil, syscall.ENOENT
|
||||||
|
}
|
||||||
|
|
||||||
|
type bytesFileReadHandle struct {
|
||||||
|
content []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.FileReader = (*bytesFileReadHandle)(nil)
|
||||||
|
func (fh *bytesFileReadHandle) Read(ctx context.Context, dest []byte, off int64) (fuse.ReadResult, syscall.Errno) {
|
||||||
|
log.Debug("FUSE Executing Read").Int("size", len(fh.content)).Send()
|
||||||
|
end := off + int64(len(dest))
|
||||||
|
if end > int64(len(fh.content)) {
|
||||||
|
end = int64(len(fh.content))
|
||||||
|
}
|
||||||
|
return fuse.ReadResultData(fh.content[off:end]), 0
|
||||||
|
}
|
||||||
|
|
||||||
|
type sensorFileReadHandle struct {
|
||||||
|
content []byte
|
||||||
|
}
|
||||||
|
var _ fs.FileReader = (*sensorFileReadHandle)(nil)
|
||||||
|
func (fh *sensorFileReadHandle) Read(ctx context.Context, dest []byte, off int64) (fuse.ReadResult, syscall.Errno) {
|
||||||
|
log.Info("Executing Read").Int("size", len(fh.content)).Send()
|
||||||
|
end := off + int64(len(dest))
|
||||||
|
if end > int64(len(fh.content)) {
|
||||||
|
end = int64(len(fh.content))
|
||||||
|
}
|
||||||
|
return fuse.ReadResultData(fh.content[off:end]), 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.FileFlusher = (*sensorFileReadHandle)(nil)
|
||||||
|
func (fh *sensorFileReadHandle) Flush(ctx context.Context) (errno syscall.Errno) {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
type bytesFileWriteHandle struct {
|
||||||
|
content []byte
|
||||||
|
path string
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.FileWriter = (*bytesFileWriteHandle)(nil)
|
||||||
|
func (fh *bytesFileWriteHandle) Write(ctx context.Context, data []byte, off int64) (written uint32, errno syscall.Errno) {
|
||||||
|
log.Info("Executing Write").Str("path", fh.path).Int("prev_size", len(fh.content)).Int("next_size", len(data)).Send()
|
||||||
|
if off != int64(len(fh.content)) {
|
||||||
|
log.Error("FUSE Write file size changed unexpectedly").Int("expect", int(off)).Int("received", len(fh.content)).Send()
|
||||||
|
return 0, syscall.ENXIO
|
||||||
|
}
|
||||||
|
fh.content = append(fh.content[:], data[:]...)
|
||||||
|
return uint32(len(data)), 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.FileFlusher = (*bytesFileWriteHandle)(nil)
|
||||||
|
func (fh *bytesFileWriteHandle) Flush(ctx context.Context) (errno syscall.Errno) {
|
||||||
|
|
||||||
|
log.Debug("FUSE Attempting flush").Str("path", fh.path).Send()
|
||||||
|
fp, err := myfs.Create(fh.path, uint32(len(fh.content)))
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Flush failed: create").Str("path", fh.path).Err(err).Send()
|
||||||
|
return syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(fh.content) == 0 {
|
||||||
|
log.Debug("FUSE Flush no data to write").Str("path", fh.path).Send()
|
||||||
|
err = fp.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Flush failed during close").Str("path", fh.path).Err(err).Send()
|
||||||
|
return syscallErr(err)
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
// For every progress event
|
||||||
|
for sent := range fp.Progress() {
|
||||||
|
log.Debug("FUSE Flush progress").Int("bytes", int(sent)).Int("total", len(fh.content)).Send()
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
r := bytes.NewReader(fh.content)
|
||||||
|
nread, err := io.Copy(fp, r)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Flush failed during write").Str("path", fh.path).Err(err).Send()
|
||||||
|
fp.Close()
|
||||||
|
return syscallErr(err)
|
||||||
|
}
|
||||||
|
if int(nread) != len(fh.content) {
|
||||||
|
log.Error("FUSE Flush failed during write").Str("path", fh.path).Int("expect", len(fh.content)).Int("got", int(nread)).Send()
|
||||||
|
fp.Close()
|
||||||
|
return syscall.EIO
|
||||||
|
}
|
||||||
|
err = fp.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Flush failed during close").Str("path", fh.path).Err(err).Send()
|
||||||
|
return syscallErr(err)
|
||||||
|
}
|
||||||
|
log.Debug("FUSE Flush done").Str("path", fh.path).Int("size", len(fh.content)).Send()
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
var _ fs.FileFsyncer = (*bytesFileWriteHandle)(nil)
|
||||||
|
func (fh *bytesFileWriteHandle) Fsync(ctx context.Context, flags uint32) (errno syscall.Errno) {
|
||||||
|
return fh.Flush(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeGetattrer = (*ITNode)(nil)
|
||||||
|
func (bn *ITNode) Getattr(ctx context.Context, f fs.FileHandle, out *fuse.AttrOut) syscall.Errno {
|
||||||
|
log.Debug("FUSE getattr").Str("path", bn.path).Send()
|
||||||
|
out.Ino = bn.Ino
|
||||||
|
out.Mtime = bn.self.modtime
|
||||||
|
out.Ctime = bn.self.modtime
|
||||||
|
out.Atime = bn.self.modtime
|
||||||
|
out.Size = uint64(bn.self.size)
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeSetattrer = (*ITNode)(nil)
|
||||||
|
func (bn *ITNode) Setattr(ctx context.Context, fh fs.FileHandle, in *fuse.SetAttrIn, out *fuse.AttrOut) syscall.Errno {
|
||||||
|
log.Debug("FUSE setattr").Str("path", bn.path).Send()
|
||||||
|
out.Size = 0
|
||||||
|
out.Mtime = 0
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeOpener = (*ITNode)(nil)
|
||||||
|
func (f *ITNode) Open(ctx context.Context, openFlags uint32) (fh fs.FileHandle, fuseFlags uint32, errno syscall.Errno) {
|
||||||
|
switch f.kind {
|
||||||
|
case 2:
|
||||||
|
// FS file
|
||||||
|
if openFlags&syscall.O_RDWR != 0 {
|
||||||
|
log.Error("FUSE Open failed: RDWR").Str("path", f.path).Send()
|
||||||
|
return nil, 0, syscall.EROFS
|
||||||
|
}
|
||||||
|
|
||||||
|
if openFlags & syscall.O_WRONLY != 0 {
|
||||||
|
log.Debug("FUSE Opening for write").Str("path", f.path).Send()
|
||||||
|
fh = &bytesFileWriteHandle{
|
||||||
|
path : f.path,
|
||||||
|
content : make([]byte, 0),
|
||||||
|
}
|
||||||
|
return fh, fuse.FOPEN_DIRECT_IO, 0
|
||||||
|
} else {
|
||||||
|
log.Debug("FUSE Opening for read").Str("path", f.path).Send()
|
||||||
|
fp, err := myfs.Open(f.path)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE: Opening failed").Str("path", f.path).Err(err).Send()
|
||||||
|
return nil, 0, syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer fp.Close()
|
||||||
|
|
||||||
|
b := &bytes.Buffer{}
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
// For every progress event
|
||||||
|
for sent := range fp.Progress() {
|
||||||
|
log.Debug("FUSE Read progress").Int("bytes", int(sent)).Int("total", int(f.self.size)).Send()
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
_, err = io.Copy(b, fp)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Read failed").Str("path", f.path).Err(err).Send()
|
||||||
|
fp.Close()
|
||||||
|
return nil, 0, syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fh = &bytesFileReadHandle{
|
||||||
|
content: b.Bytes(),
|
||||||
|
}
|
||||||
|
return fh, fuse.FOPEN_DIRECT_IO, 0
|
||||||
|
}
|
||||||
|
|
||||||
|
case 3:
|
||||||
|
// Device file
|
||||||
|
|
||||||
|
// disallow writes
|
||||||
|
if openFlags&(syscall.O_RDWR|syscall.O_WRONLY) != 0 {
|
||||||
|
return nil, 0, syscall.EROFS
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, value := range properties {
|
||||||
|
if value.Ino == f.Ino {
|
||||||
|
ans, err := value.gen()
|
||||||
|
if err != nil {
|
||||||
|
return nil, 0, syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fh = &sensorFileReadHandle{
|
||||||
|
content : ans,
|
||||||
|
}
|
||||||
|
return fh, fuse.FOPEN_DIRECT_IO, 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, 0, syscall.EINVAL
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeCreater = (*ITNode)(nil)
|
||||||
|
func (f *ITNode) Create(ctx context.Context, name string, flags uint32, mode uint32, out *fuse.EntryOut) (node *fs.Inode, fh fs.FileHandle, fuseFlags uint32, errno syscall.Errno) {
|
||||||
|
if f.kind != 2 {
|
||||||
|
return nil, nil, 0, syscall.EROFS
|
||||||
|
}
|
||||||
|
|
||||||
|
path := f.path + "/" + name
|
||||||
|
ino := uint64(len(inodemap)) + 11
|
||||||
|
inodemap[path] = ino
|
||||||
|
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFREG,
|
||||||
|
Ino: ino,
|
||||||
|
}
|
||||||
|
operations := &ITNode{
|
||||||
|
kind: 2, Ino: ino,
|
||||||
|
path : path,
|
||||||
|
}
|
||||||
|
node = f.NewInode(ctx, operations, stable)
|
||||||
|
|
||||||
|
fh = &bytesFileWriteHandle{
|
||||||
|
path : path,
|
||||||
|
content : make([]byte, 0),
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("FUSE Creating file").Str("path", path).Send()
|
||||||
|
|
||||||
|
errno = 0
|
||||||
|
return node, fh, fuseFlags, 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeMkdirer = (*ITNode)(nil)
|
||||||
|
func (f *ITNode) Mkdir(ctx context.Context, name string, mode uint32, out *fuse.EntryOut) (*fs.Inode, syscall.Errno) {
|
||||||
|
if f.kind != 2 {
|
||||||
|
return nil, syscall.EROFS
|
||||||
|
}
|
||||||
|
|
||||||
|
path := f.path + "/" + name
|
||||||
|
err := myfs.Mkdir(path)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Mkdir failed").
|
||||||
|
Str("path", path).
|
||||||
|
Err(err).
|
||||||
|
Send()
|
||||||
|
return nil, syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ino := uint64(len(inodemap)) + 11
|
||||||
|
inodemap[path] = ino
|
||||||
|
|
||||||
|
stable := fs.StableAttr{
|
||||||
|
Mode: fuse.S_IFDIR,
|
||||||
|
Ino: ino,
|
||||||
|
}
|
||||||
|
operations := &ITNode{
|
||||||
|
kind: 2, Ino: ino,
|
||||||
|
path : path,
|
||||||
|
}
|
||||||
|
node := f.NewInode(ctx, operations, stable)
|
||||||
|
|
||||||
|
log.Debug("FUSE Mkdir success").
|
||||||
|
Str("path", path).
|
||||||
|
Int("ino", int(ino)).
|
||||||
|
Send()
|
||||||
|
return node, 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeRenamer = (*ITNode)(nil)
|
||||||
|
func (f *ITNode) Rename(ctx context.Context, name string, newParent fs.InodeEmbedder, newName string, flags uint32) syscall.Errno {
|
||||||
|
if f.kind != 2 {
|
||||||
|
return syscall.EROFS
|
||||||
|
}
|
||||||
|
|
||||||
|
p1 := f.path + "/" + name
|
||||||
|
p2 := newParent.EmbeddedInode().Path(nil)[2:] + "/" + newName
|
||||||
|
|
||||||
|
err := myfs.Rename(p1, p2)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Rename failed").
|
||||||
|
Str("src", p1).
|
||||||
|
Str("dest", p2).
|
||||||
|
Err(err).
|
||||||
|
Send()
|
||||||
|
|
||||||
|
return syscallErr(err)
|
||||||
|
}
|
||||||
|
log.Debug("FUSE Rename sucess").
|
||||||
|
Str("src", p1).
|
||||||
|
Str("dest", p2).
|
||||||
|
Send()
|
||||||
|
|
||||||
|
ino := inodemap[p1]
|
||||||
|
delete(inodemap, p1)
|
||||||
|
inodemap[p2] = ino
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeUnlinker = (*ITNode)(nil)
|
||||||
|
func (f *ITNode) Unlink(ctx context.Context, name string) syscall.Errno {
|
||||||
|
if f.kind != 2 {
|
||||||
|
return syscall.EROFS
|
||||||
|
}
|
||||||
|
|
||||||
|
delete(inodemap, f.path + "/" + name)
|
||||||
|
err := myfs.Remove(f.path + "/" + name)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("FUSE Unlink failed").
|
||||||
|
Str("file", f.path + "/" + name).
|
||||||
|
Err(err).
|
||||||
|
Send()
|
||||||
|
|
||||||
|
return syscallErr(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("FUSE Unlink success").
|
||||||
|
Str("file", f.path + "/" + name).
|
||||||
|
Send()
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ fs.NodeRmdirer = (*ITNode)(nil)
|
||||||
|
func (f *ITNode) Rmdir(ctx context.Context, name string) syscall.Errno {
|
||||||
|
return f.Unlink(ctx, name)
|
||||||
|
}
|
||||||
67
internal/fusefs/syscallerr.go
Normal file
67
internal/fusefs/syscallerr.go
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
package fusefs
|
||||||
|
import (
|
||||||
|
"go.arsenm.dev/infinitime/blefs"
|
||||||
|
"syscall"
|
||||||
|
)
|
||||||
|
|
||||||
|
func syscallErr(err error) syscall.Errno {
|
||||||
|
if err == nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
switch err {
|
||||||
|
case blefs.FSError{0x02}: // filesystem error
|
||||||
|
return syscall.EIO // TODO
|
||||||
|
case blefs.FSError{0x05}: // read-only filesystem
|
||||||
|
return syscall.EROFS
|
||||||
|
case blefs.FSError{0x03}: // no such file
|
||||||
|
return syscall.ENOENT
|
||||||
|
case blefs.FSError{0x04}: // protocol error
|
||||||
|
return syscall.EPROTO
|
||||||
|
case blefs.FSError{-5}: // input/output error
|
||||||
|
return syscall.EIO
|
||||||
|
case blefs.FSError{-84}: // filesystem is corrupted
|
||||||
|
return syscall.ENOTRECOVERABLE // TODO
|
||||||
|
case blefs.FSError{-2}: // no such directory entry
|
||||||
|
return syscall.ENOENT
|
||||||
|
case blefs.FSError{-17}: // entry already exists
|
||||||
|
return syscall.EEXIST
|
||||||
|
case blefs.FSError{-20}: // entry is not a directory
|
||||||
|
return syscall.ENOTDIR
|
||||||
|
case blefs.FSError{-39}: // directory is not empty
|
||||||
|
return syscall.ENOTEMPTY
|
||||||
|
case blefs.FSError{-9}: // bad file number
|
||||||
|
return syscall.EBADF
|
||||||
|
case blefs.FSError{-27}: // file is too large
|
||||||
|
return syscall.EFBIG
|
||||||
|
case blefs.FSError{-22}: // invalid parameter
|
||||||
|
return syscall.EINVAL
|
||||||
|
case blefs.FSError{-28}: // no space left on device
|
||||||
|
return syscall.ENOSPC
|
||||||
|
case blefs.FSError{-12}: // no more memory available
|
||||||
|
return syscall.ENOMEM
|
||||||
|
case blefs.FSError{-61}: // no attr available
|
||||||
|
return syscall.ENODATA // TODO
|
||||||
|
case blefs.FSError{-36}: // file name is too long
|
||||||
|
return syscall.ENAMETOOLONG
|
||||||
|
case blefs.ErrFileNotExists: // file does not exist
|
||||||
|
return syscall.ENOENT
|
||||||
|
case blefs.ErrFileReadOnly: // file is read only
|
||||||
|
return syscall.EACCES
|
||||||
|
case blefs.ErrFileWriteOnly: // file is write only
|
||||||
|
return syscall.EACCES
|
||||||
|
case blefs.ErrInvalidOffset: // invalid file offset
|
||||||
|
return syscall.EFAULT // TODO
|
||||||
|
case blefs.ErrOffsetChanged: // offset has already been changed
|
||||||
|
return syscall.ESPIPE
|
||||||
|
case blefs.ErrReadOpen: // only one file can be opened for reading at a time
|
||||||
|
return syscall.ENFILE
|
||||||
|
case blefs.ErrWriteOpen: // only one file can be opened for writing at a time
|
||||||
|
return syscall.ENFILE
|
||||||
|
case blefs.ErrNoRemoveRoot: // refusing to remove root directory
|
||||||
|
return syscall.EPERM
|
||||||
|
}
|
||||||
|
|
||||||
|
return syscall.EIO // TODO
|
||||||
|
|
||||||
|
}
|
||||||
15
internal/fusefs/unmount.go
Normal file
15
internal/fusefs/unmount.go
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
package fusefs
|
||||||
|
|
||||||
|
import (
|
||||||
|
_ "unsafe"
|
||||||
|
"github.com/hanwen/go-fuse/v2/fuse"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Unmount(mountPoint string) error {
|
||||||
|
return unmount(mountPoint, &fuse.MountOptions{DirectMount: false})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unfortunately, the FUSE library does not export its unmount function,
|
||||||
|
// so this is required until that changes
|
||||||
|
//go:linkname unmount github.com/hanwen/go-fuse/v2/fuse.unmount
|
||||||
|
func unmount(mountPoint string, opts *fuse.MountOptions) error
|
||||||
15
main.go
15
main.go
@@ -181,18 +181,19 @@ func main() {
|
|||||||
log.Error("Error intializing puremaps integration").Err(err).Send()
|
log.Error("Error intializing puremaps integration").Err(err).Send()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Start fuse socket
|
||||||
|
if k.Bool("fuse.enabled") {
|
||||||
|
err = startFUSE(ctx, dev)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Error starting fuse socket").Err(err).Send()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Start control socket
|
// Start control socket
|
||||||
err = startSocket(ctx, dev)
|
err = startSocket(ctx, dev)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Error starting socket").Err(err).Send()
|
log.Error("Error starting socket").Err(err).Send()
|
||||||
}
|
}
|
||||||
// Start fuse socket
|
|
||||||
if k.Bool("fuse.enabled") {
|
|
||||||
err = startFuse(ctx, dev)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error starting fuse socket").Err(err).Send()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// Block forever
|
// Block forever
|
||||||
select {}
|
select {}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user