1 files changed, 180 insertions, 0 deletions
diff --git a/cluefs_frontend/fe_dir.go b/cluefs_frontend/fe_dir.go
new file mode 100644
index 0000000..65d92f1
--- /dev/null
+++ b/cluefs_frontend/fe_dir.go
@@ -0,0 +1,180 @@
+package cluefs_frontend
+
+// frontend sits between FUSE and ClueFS
+// and uses cryptfs for all crypto operations
+//
+//          cryptfs
+//             ^
+//             |
+//             v
+// FUSE <-> frontend <-> ClueFS
+//
+// This file handles directories
+
+import (
+	"fmt"
+	"github.com/rfjakob/gocryptfs/cryptfs"
+	"github.com/rfjakob/cluefs/lib/cluefs"
+	"bazil.org/fuse"
+	fusefs "bazil.org/fuse/fs"
+	"golang.org/x/net/context"
+)
+
+type Dir struct {
+	*cluefs.Dir
+	crfs *cryptfs.CryptFS
+}
+
+func NewDir(parent string, name string, fs *FS) *Dir {
+	cryptfs.Debug.Printf("NewDir parent=%s name=%s\n", parent, name)
+	return &Dir {
+		Dir: cluefs.NewDir(parent, name, fs.ClueFS),
+		crfs: fs.CryptFS,
+	}
+}
+
+func (d *Dir) Open(ctx context.Context, req *fuse.OpenRequest, resp *fuse.OpenResponse) (fusefs.Handle, error) {
+	cryptfs.Debug.Printf("Open\n")
+	h, err := d.Dir.Open(ctx, req, resp)
+	if err != nil {
+		return nil, err
+	}
+	clueDir := h.(*cluefs.Dir)
+
+	return &Dir {
+		Dir: clueDir,
+		crfs: d.crfs,
+	}, nil
+}
+
+func (d *Dir) Lookup(ctx context.Context, req *fuse.LookupRequest, resp *fuse.LookupResponse) (fusefs.Node, error) {
+	cryptfs.Debug.Printf("Lookup %s\n", req.Name)
+	req.Name = d.crfs.EncryptPath(req.Name)
+	node, err := d.Dir.Lookup(ctx, req, resp)
+	if err != nil {
+		return nil, err
+	}
+	clueDir, ok := node.(*cluefs.Dir)
+	if ok {
+		return &Dir {
+			Dir: clueDir,
+			crfs: d.crfs,
+		}, nil
+	} else {
+		resp.Attr.Size = d.crfs.PlainSize(resp.Attr.Size)
+		clueFile := node.(*cluefs.File)
+		return &File {
+			File: clueFile,
+			crfs: d.crfs,
+		}, nil
+	}
+}
+
+func (d *Dir) ReadDirAll(ctx context.Context) ([]fuse.Dirent, error) {
+	cryptfs.Debug.Printf("ReadDirAll\n")
+	entries, err := d.Dir.ReadDirAll(ctx)
+	if err != nil {
+		return nil, err
+	}
+	var decrypted []fuse.Dirent
+	for _, e := range entries {
+		if e.Name == "." || e.Name == ".." {
+			decrypted = append(decrypted, e)
+			continue
+		}
+		newName, err := d.crfs.DecryptPath(e.Name)
+		if err != nil {
+			fmt.Printf("ReadDirAll: Error decoding \"%s\": %s\n", e.Name, err.Error())
+			continue
+		}
+		e.Name = newName
+		decrypted = append(decrypted, e)
+	}
+	return decrypted, nil
+}
+
+func (d *Dir) Mkdir(ctx context.Context, req *fuse.MkdirRequest) (fusefs.Node, error) {
+	cryptfs.Debug.Printf("Mkdir %s\n", req.Name)
+	req.Name = d.crfs.EncryptPath(req.Name)
+	n, err := d.Dir.Mkdir(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	clueDir := n.(*cluefs.Dir)
+	return &Dir {
+		Dir: clueDir,
+		crfs: d.crfs,
+	}, nil
+}
+
+func (d *Dir) Remove(ctx context.Context, req *fuse.RemoveRequest) error {
+	cryptfs.Debug.Printf("Remove\n")
+	req.Name = d.crfs.EncryptPath(req.Name)
+	return d.Dir.Remove(ctx, req)
+}
+
+func (d *Dir) Create(ctx context.Context, req *fuse.CreateRequest, resp *fuse.CreateResponse) (fusefs.Node, fusefs.Handle, error) {
+	cryptfs.Debug.Printf("Create\n")
+	req.Flags, _ = fixFlags(req.Flags)
+	req.Name = d.crfs.EncryptPath(req.Name)
+	n, _, err := d.Dir.Create(ctx, req, resp)
+	if err != nil {
+		return nil, nil, err
+	}
+	clueFile := n.(*cluefs.File)
+	cryptFile := &File {
+		File: clueFile,
+		crfs: d.crfs,
+	}
+	return cryptFile, cryptFile, nil
+}
+func (d *Dir) Symlink(ctx context.Context, req *fuse.SymlinkRequest) (fusefs.Node, error) {
+	req.Target = d.crfs.EncryptPath(req.Target)
+	req.NewName = d.crfs.EncryptPath(req.NewName)
+	node, err := d.Dir.Symlink(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	clueDir, ok := node.(*cluefs.Dir)
+	if ok {
+		return &Dir {
+			Dir: clueDir,
+			crfs: d.crfs,
+		}, nil
+	} else {
+		clueFile := node.(*cluefs.File)
+		return &File {
+			File: clueFile,
+			crfs: d.crfs,
+		}, nil
+	}
+}
+// We need to overwrite Readlink for both Dir and File. Do both right here to keep them in sync.
+func (d *Dir) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) {
+	dest, err := d.Dir.Readlink(ctx, req)
+	if err != nil {
+		return "", err
+	}
+	return d.crfs.DecryptPath(dest)
+}
+func (d *File) Readlink(ctx context.Context, req *fuse.ReadlinkRequest) (string, error) {
+	dest, err := d.File.Readlink(ctx, req)
+	if err != nil {
+		return "", err
+	}
+	return d.crfs.DecryptPath(dest)
+}
+// We need to overwrite Rename for both Dir and File. Do both right here to keep them in sync.
+func (d *Dir) Rename(ctx context.Context, req *fuse.RenameRequest, newDir fusefs.Node) error {
+	req.OldName = d.crfs.EncryptPath(req.OldName)
+	req.NewName = d.crfs.EncryptPath(req.NewName)
+	destDir := newDir.(*Dir)
+	return d.Dir.Rename(ctx, req, destDir.Dir)
+}
+func (d *File) Rename(ctx context.Context, req *fuse.RenameRequest, newDir fusefs.Node) error {
+	req.OldName = d.crfs.EncryptPath(req.OldName)
+	req.NewName = d.crfs.EncryptPath(req.NewName)
+	destDir := newDir.(*Dir)
+	return d.File.Rename(ctx, req, destDir.Dir)
+}