2015-03-20 02:32:25 +03:00
|
|
|
// Copyright 2015 Google Inc. All Rights Reserved.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
|
|
|
package flushfs
|
|
|
|
|
2015-03-20 03:21:38 +03:00
|
|
|
import (
|
2015-03-20 03:49:45 +03:00
|
|
|
"fmt"
|
2015-03-24 08:17:10 +03:00
|
|
|
"io"
|
2015-03-20 03:34:16 +03:00
|
|
|
"os"
|
2015-03-20 03:38:40 +03:00
|
|
|
"sync"
|
2015-03-20 03:34:16 +03:00
|
|
|
|
2015-03-20 03:21:38 +03:00
|
|
|
"github.com/jacobsa/fuse"
|
2015-03-24 08:14:42 +03:00
|
|
|
"github.com/jacobsa/fuse/fuseops"
|
2015-03-20 03:21:38 +03:00
|
|
|
)
|
2015-03-20 02:32:25 +03:00
|
|
|
|
2015-03-24 04:02:48 +03:00
|
|
|
// Create a file system whose sole contents are a file named "foo" and a
|
|
|
|
// directory named "bar".
|
2015-03-20 02:32:25 +03:00
|
|
|
//
|
|
|
|
// The file may be opened for reading and/or writing. Its initial contents are
|
|
|
|
// empty. Whenever a flush or fsync is received, the supplied function will be
|
2015-03-20 03:52:54 +03:00
|
|
|
// called with the current contents of the file and its status returned.
|
2015-03-24 04:02:48 +03:00
|
|
|
//
|
|
|
|
// The directory cannot be modified.
|
2015-03-20 02:32:25 +03:00
|
|
|
func NewFileSystem(
|
2015-03-20 03:52:54 +03:00
|
|
|
reportFlush func(string) error,
|
2015-03-24 08:14:42 +03:00
|
|
|
reportFsync func(string) error) (server fuse.Server, err error) {
|
|
|
|
server = &flushFS{
|
2015-03-20 05:27:27 +03:00
|
|
|
reportFlush: reportFlush,
|
|
|
|
reportFsync: reportFsync,
|
|
|
|
}
|
|
|
|
|
2015-03-20 03:21:38 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-03-24 04:02:48 +03:00
|
|
|
const (
|
2015-03-24 08:14:42 +03:00
|
|
|
fooID = fuseops.RootInodeID + 1 + iota
|
2015-03-24 04:02:48 +03:00
|
|
|
barID
|
|
|
|
)
|
2015-03-20 03:38:40 +03:00
|
|
|
|
2015-03-20 03:21:38 +03:00
|
|
|
type flushFS struct {
|
2015-03-20 05:27:27 +03:00
|
|
|
reportFlush func(string) error
|
|
|
|
reportFsync func(string) error
|
2015-03-20 03:34:16 +03:00
|
|
|
|
2015-03-20 03:49:45 +03:00
|
|
|
mu sync.Mutex
|
|
|
|
fooContents []byte // GUARDED_BY(mu)
|
2015-03-20 03:21:38 +03:00
|
|
|
}
|
2015-03-20 03:22:16 +03:00
|
|
|
|
2015-03-20 03:38:40 +03:00
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
|
|
// Helpers
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
|
|
|
2015-03-20 03:40:14 +03:00
|
|
|
// LOCKS_REQUIRED(fs.mu)
|
2015-03-24 08:14:42 +03:00
|
|
|
func (fs *flushFS) rootAttributes() fuseops.InodeAttributes {
|
|
|
|
return fuseops.InodeAttributes{
|
2015-03-20 03:40:14 +03:00
|
|
|
Nlink: 1,
|
|
|
|
Mode: 0777 | os.ModeDir,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-03-20 03:38:40 +03:00
|
|
|
// LOCKS_REQUIRED(fs.mu)
|
2015-03-24 08:14:42 +03:00
|
|
|
func (fs *flushFS) fooAttributes() fuseops.InodeAttributes {
|
|
|
|
return fuseops.InodeAttributes{
|
2015-03-20 03:38:40 +03:00
|
|
|
Nlink: 1,
|
|
|
|
Mode: 0777,
|
2015-03-20 05:28:51 +03:00
|
|
|
Size: uint64(len(fs.fooContents)),
|
2015-03-20 03:38:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-03-24 04:02:48 +03:00
|
|
|
// LOCKS_REQUIRED(fs.mu)
|
2015-03-24 08:14:42 +03:00
|
|
|
func (fs *flushFS) barAttributes() fuseops.InodeAttributes {
|
|
|
|
return fuseops.InodeAttributes{
|
2015-03-24 04:02:48 +03:00
|
|
|
Nlink: 1,
|
|
|
|
Mode: 0777 | os.ModeDir,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-03-24 08:14:42 +03:00
|
|
|
// LOCKS_REQUIRED(fs.mu)
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) ServeOps(c *fuse.Connection) {
|
|
|
|
for {
|
|
|
|
op, err := c.ReadOp()
|
|
|
|
if err == io.EOF {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
switch typed := op.(type) {
|
|
|
|
case *fuseops.InitOp:
|
|
|
|
fs.init(typed)
|
|
|
|
|
|
|
|
case *fuseops.LookUpInodeOp:
|
|
|
|
fs.lookUpInode(typed)
|
|
|
|
|
|
|
|
case *fuseops.GetInodeAttributesOp:
|
|
|
|
fs.getInodeAttributes(typed)
|
|
|
|
|
|
|
|
case *fuseops.OpenFileOp:
|
|
|
|
fs.openFile(typed)
|
|
|
|
|
|
|
|
case *fuseops.ReadFileOp:
|
|
|
|
fs.readFile(typed)
|
2015-03-24 08:19:09 +03:00
|
|
|
|
|
|
|
case *fuseops.WriteFileOp:
|
|
|
|
fs.writeFile(typed)
|
|
|
|
|
|
|
|
case *fuseops.SyncFileOp:
|
|
|
|
fs.syncFile(typed)
|
|
|
|
|
|
|
|
case *fuseops.FlushFileOp:
|
|
|
|
fs.flushFile(typed)
|
|
|
|
|
|
|
|
case *fuseops.OpenDirOp:
|
|
|
|
fs.openDir(typed)
|
2015-03-24 08:17:10 +03:00
|
|
|
|
|
|
|
default:
|
|
|
|
typed.Respond(fuse.ENOSYS)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-03-24 08:14:42 +03:00
|
|
|
|
2015-03-20 03:38:40 +03:00
|
|
|
////////////////////////////////////////////////////////////////////////
|
2015-03-24 08:14:42 +03:00
|
|
|
// Op methods
|
2015-03-20 03:38:40 +03:00
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) init(op *fuseops.InitOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
|
|
|
|
2015-03-20 03:22:16 +03:00
|
|
|
return
|
|
|
|
}
|
2015-03-20 03:38:40 +03:00
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) lookUpInode(op *fuseops.LookUpInodeOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 03:38:40 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
// Sanity check.
|
2015-03-24 08:17:10 +03:00
|
|
|
if op.Parent != fuseops.RootInodeID {
|
2015-03-20 03:38:40 +03:00
|
|
|
err = fuse.ENOENT
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-03-24 04:02:48 +03:00
|
|
|
// Set up the entry.
|
2015-03-24 08:17:10 +03:00
|
|
|
switch op.Name {
|
2015-03-24 04:02:48 +03:00
|
|
|
case "foo":
|
2015-03-24 08:17:10 +03:00
|
|
|
op.Entry = fuseops.ChildInodeEntry{
|
2015-03-24 04:02:48 +03:00
|
|
|
Child: fooID,
|
|
|
|
Attributes: fs.fooAttributes(),
|
|
|
|
}
|
|
|
|
|
|
|
|
case "bar":
|
2015-03-24 08:17:10 +03:00
|
|
|
op.Entry = fuseops.ChildInodeEntry{
|
2015-03-24 04:02:48 +03:00
|
|
|
Child: barID,
|
|
|
|
Attributes: fs.barAttributes(),
|
|
|
|
}
|
|
|
|
|
|
|
|
default:
|
|
|
|
err = fuse.ENOENT
|
|
|
|
return
|
2015-03-20 03:38:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
2015-03-20 03:40:14 +03:00
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) getInodeAttributes(op *fuseops.GetInodeAttributesOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 03:40:14 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
switch op.Inode {
|
2015-03-24 08:14:42 +03:00
|
|
|
case fuseops.RootInodeID:
|
2015-03-24 08:17:10 +03:00
|
|
|
op.Attributes = fs.rootAttributes()
|
2015-03-20 03:40:14 +03:00
|
|
|
return
|
|
|
|
|
|
|
|
case fooID:
|
2015-03-24 08:17:10 +03:00
|
|
|
op.Attributes = fs.fooAttributes()
|
2015-03-20 03:40:14 +03:00
|
|
|
return
|
|
|
|
|
2015-03-24 04:02:48 +03:00
|
|
|
case barID:
|
2015-03-24 08:17:10 +03:00
|
|
|
op.Attributes = fs.barAttributes()
|
2015-03-24 04:02:48 +03:00
|
|
|
return
|
|
|
|
|
2015-03-20 03:40:14 +03:00
|
|
|
default:
|
|
|
|
err = fuse.ENOENT
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
2015-03-20 03:40:53 +03:00
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) openFile(op *fuseops.OpenFileOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 03:40:53 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
// Sanity check.
|
2015-03-24 08:17:10 +03:00
|
|
|
if op.Inode != fooID {
|
2015-03-20 03:40:53 +03:00
|
|
|
err = fuse.ENOSYS
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
2015-03-20 03:49:45 +03:00
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) readFile(op *fuseops.ReadFileOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 05:51:04 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
// Ensure the offset is in range.
|
2015-03-24 08:17:10 +03:00
|
|
|
if op.Offset > int64(len(fs.fooContents)) {
|
2015-03-20 05:51:04 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Read what we can.
|
2015-03-24 08:17:10 +03:00
|
|
|
op.Data = make([]byte, op.Size)
|
|
|
|
copy(op.Data, fs.fooContents[op.Offset:])
|
2015-03-20 05:51:04 +03:00
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) writeFile(op *fuseops.WriteFileOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 03:49:45 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
// Ensure that the contents slice is long enough.
|
2015-03-24 08:17:10 +03:00
|
|
|
newLen := int(op.Offset) + len(op.Data)
|
2015-03-20 03:49:45 +03:00
|
|
|
if len(fs.fooContents) < newLen {
|
|
|
|
padding := make([]byte, newLen-len(fs.fooContents))
|
|
|
|
fs.fooContents = append(fs.fooContents, padding...)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Copy in the data.
|
2015-03-24 08:17:10 +03:00
|
|
|
n := copy(fs.fooContents[op.Offset:], op.Data)
|
2015-03-20 03:49:45 +03:00
|
|
|
|
|
|
|
// Sanity check.
|
2015-03-24 08:17:10 +03:00
|
|
|
if n != len(op.Data) {
|
2015-03-20 03:49:45 +03:00
|
|
|
panic(fmt.Sprintf("Unexpected short copy: %v", n))
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
2015-03-20 05:27:27 +03:00
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) syncFile(op *fuseops.SyncFileOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 05:40:53 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
err = fs.reportFsync(string(fs.fooContents))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) flushFile(op *fuseops.FlushFileOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-20 05:27:27 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
err = fs.reportFlush(string(fs.fooContents))
|
|
|
|
return
|
|
|
|
}
|
2015-03-24 04:02:48 +03:00
|
|
|
|
2015-03-24 08:17:10 +03:00
|
|
|
func (fs *flushFS) openDir(op *fuseops.OpenDirOp) {
|
|
|
|
var err error
|
|
|
|
defer func() { op.Respond(err) }()
|
2015-03-24 04:02:48 +03:00
|
|
|
|
|
|
|
fs.mu.Lock()
|
|
|
|
defer fs.mu.Unlock()
|
|
|
|
|
|
|
|
// Sanity check.
|
2015-03-24 08:17:10 +03:00
|
|
|
if op.Inode != barID {
|
2015-03-24 04:02:48 +03:00
|
|
|
err = fuse.ENOSYS
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|