225 lines
4.9 KiB
Go
225 lines
4.9 KiB
Go
// 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
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
|
|
"github.com/jacobsa/fuse"
|
|
"github.com/jacobsa/fuse/fuseutil"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
// Create a file system containing a single file named "foo".
|
|
//
|
|
// 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
|
|
// called with the current contents of the file and its status returned.
|
|
func NewFileSystem(
|
|
reportFlush func(string) error,
|
|
reportFsync func(string) error) (fs fuse.FileSystem, err error) {
|
|
fs = &flushFS{
|
|
reportFlush: reportFlush,
|
|
reportFsync: reportFsync,
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
const fooID = fuse.RootInodeID + 1
|
|
|
|
type flushFS struct {
|
|
fuseutil.NotImplementedFileSystem
|
|
reportFlush func(string) error
|
|
reportFsync func(string) error
|
|
|
|
mu sync.Mutex
|
|
fooContents []byte // GUARDED_BY(mu)
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
// Helpers
|
|
////////////////////////////////////////////////////////////////////////
|
|
|
|
// LOCKS_REQUIRED(fs.mu)
|
|
func (fs *flushFS) rootAttributes() fuse.InodeAttributes {
|
|
return fuse.InodeAttributes{
|
|
Nlink: 1,
|
|
Mode: 0777 | os.ModeDir,
|
|
}
|
|
}
|
|
|
|
// LOCKS_REQUIRED(fs.mu)
|
|
func (fs *flushFS) fooAttributes() fuse.InodeAttributes {
|
|
return fuse.InodeAttributes{
|
|
Nlink: 1,
|
|
Mode: 0777,
|
|
Size: uint64(len(fs.fooContents)),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
// File system methods
|
|
////////////////////////////////////////////////////////////////////////
|
|
|
|
func (fs *flushFS) Init(
|
|
ctx context.Context,
|
|
req *fuse.InitRequest) (
|
|
resp *fuse.InitResponse, err error) {
|
|
resp = &fuse.InitResponse{}
|
|
return
|
|
}
|
|
|
|
func (fs *flushFS) LookUpInode(
|
|
ctx context.Context,
|
|
req *fuse.LookUpInodeRequest) (
|
|
resp *fuse.LookUpInodeResponse, err error) {
|
|
resp = &fuse.LookUpInodeResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
// Sanity check.
|
|
if req.Parent != fuse.RootInodeID || req.Name != "foo" {
|
|
err = fuse.ENOENT
|
|
return
|
|
}
|
|
|
|
resp.Entry = fuse.ChildInodeEntry{
|
|
Child: fooID,
|
|
Attributes: fs.fooAttributes(),
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (fs *flushFS) GetInodeAttributes(
|
|
ctx context.Context,
|
|
req *fuse.GetInodeAttributesRequest) (
|
|
resp *fuse.GetInodeAttributesResponse, err error) {
|
|
resp = &fuse.GetInodeAttributesResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
switch req.Inode {
|
|
case fuse.RootInodeID:
|
|
resp.Attributes = fs.rootAttributes()
|
|
return
|
|
|
|
case fooID:
|
|
resp.Attributes = fs.fooAttributes()
|
|
return
|
|
|
|
default:
|
|
err = fuse.ENOENT
|
|
return
|
|
}
|
|
}
|
|
|
|
func (fs *flushFS) OpenFile(
|
|
ctx context.Context,
|
|
req *fuse.OpenFileRequest) (
|
|
resp *fuse.OpenFileResponse, err error) {
|
|
resp = &fuse.OpenFileResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
// Sanity check.
|
|
if req.Inode != fooID {
|
|
err = fuse.ENOSYS
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (fs *flushFS) ReadFile(
|
|
ctx context.Context,
|
|
req *fuse.ReadFileRequest) (
|
|
resp *fuse.ReadFileResponse, err error) {
|
|
resp = &fuse.ReadFileResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
// Ensure the offset is in range.
|
|
if req.Offset > int64(len(fs.fooContents)) {
|
|
return
|
|
}
|
|
|
|
// Read what we can.
|
|
resp.Data = make([]byte, req.Size)
|
|
copy(resp.Data, fs.fooContents[req.Offset:])
|
|
|
|
return
|
|
}
|
|
|
|
func (fs *flushFS) WriteFile(
|
|
ctx context.Context,
|
|
req *fuse.WriteFileRequest) (
|
|
resp *fuse.WriteFileResponse, err error) {
|
|
resp = &fuse.WriteFileResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
// Ensure that the contents slice is long enough.
|
|
newLen := int(req.Offset) + len(req.Data)
|
|
if len(fs.fooContents) < newLen {
|
|
padding := make([]byte, newLen-len(fs.fooContents))
|
|
fs.fooContents = append(fs.fooContents, padding...)
|
|
}
|
|
|
|
// Copy in the data.
|
|
n := copy(fs.fooContents[req.Offset:], req.Data)
|
|
|
|
// Sanity check.
|
|
if n != len(req.Data) {
|
|
panic(fmt.Sprintf("Unexpected short copy: %v", n))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (fs *flushFS) SyncFile(
|
|
ctx context.Context,
|
|
req *fuse.SyncFileRequest) (
|
|
resp *fuse.SyncFileResponse, err error) {
|
|
resp = &fuse.SyncFileResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
err = fs.reportFsync(string(fs.fooContents))
|
|
return
|
|
}
|
|
|
|
func (fs *flushFS) FlushFile(
|
|
ctx context.Context,
|
|
req *fuse.FlushFileRequest) (
|
|
resp *fuse.FlushFileResponse, err error) {
|
|
resp = &fuse.FlushFileResponse{}
|
|
|
|
fs.mu.Lock()
|
|
defer fs.mu.Unlock()
|
|
|
|
err = fs.reportFlush(string(fs.fooContents))
|
|
return
|
|
}
|