2015-03-23 07:52:27 +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.
|
|
|
|
|
|
|
|
// A simple tool for mounting sample file systems, used by the tests in
|
|
|
|
// samples/.
|
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2015-03-24 02:08:17 +03:00
|
|
|
"errors"
|
2015-03-23 07:52:27 +03:00
|
|
|
"flag"
|
2015-03-23 08:01:16 +03:00
|
|
|
"fmt"
|
2015-03-23 08:00:12 +03:00
|
|
|
"log"
|
2015-03-23 08:08:46 +03:00
|
|
|
"os"
|
2015-03-24 03:53:32 +03:00
|
|
|
"runtime"
|
2015-03-23 08:00:12 +03:00
|
|
|
|
2015-03-23 08:08:46 +03:00
|
|
|
"github.com/jacobsa/bazilfuse"
|
2015-03-23 08:00:12 +03:00
|
|
|
"github.com/jacobsa/fuse"
|
2015-03-23 08:08:46 +03:00
|
|
|
"github.com/jacobsa/fuse/samples/flushfs"
|
2015-03-23 08:00:12 +03:00
|
|
|
"golang.org/x/net/context"
|
2015-03-23 07:52:27 +03:00
|
|
|
)
|
|
|
|
|
2015-03-23 08:00:12 +03:00
|
|
|
var fType = flag.String("type", "", "The name of the samples/ sub-dir.")
|
|
|
|
var fMountPoint = flag.String("mount_point", "", "Path to mount point.")
|
2015-03-24 02:08:17 +03:00
|
|
|
var fReadyFile = flag.Uint64("ready_file", 0, "FD to signal when ready.")
|
2015-03-23 07:52:27 +03:00
|
|
|
|
2015-03-24 01:46:05 +03:00
|
|
|
var fFlushesFile = flag.Uint64("flushfs.flushes_file", 0, "")
|
|
|
|
var fFsyncsFile = flag.Uint64("flushfs.fsyncs_file", 0, "")
|
2015-03-23 08:08:46 +03:00
|
|
|
var fFlushError = flag.Int("flushfs.flush_error", 0, "")
|
|
|
|
var fFsyncError = flag.Int("flushfs.fsync_error", 0, "")
|
2015-03-23 07:54:20 +03:00
|
|
|
|
2015-03-24 08:17:56 +03:00
|
|
|
func makeFlushFS() (server fuse.Server, err error) {
|
2015-03-23 08:08:46 +03:00
|
|
|
// Check the flags.
|
2015-03-24 01:46:05 +03:00
|
|
|
if *fFlushesFile == 0 || *fFsyncsFile == 0 {
|
2015-03-23 08:08:46 +03:00
|
|
|
err = fmt.Errorf("You must set the flushfs flags.")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-03-24 01:46:05 +03:00
|
|
|
// Set up the files.
|
|
|
|
flushes := os.NewFile(uintptr(*fFlushesFile), "(flushes file)")
|
|
|
|
fsyncs := os.NewFile(uintptr(*fFsyncsFile), "(fsyncs file)")
|
2015-03-23 08:08:46 +03:00
|
|
|
|
|
|
|
// Set up errors.
|
|
|
|
var flushErr error
|
|
|
|
var fsyncErr error
|
|
|
|
|
|
|
|
if *fFlushError != 0 {
|
|
|
|
flushErr = bazilfuse.Errno(*fFlushError)
|
|
|
|
}
|
|
|
|
|
|
|
|
if *fFsyncError != 0 {
|
|
|
|
fsyncErr = bazilfuse.Errno(*fFsyncError)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Report flushes and fsyncs by writing the contents followed by a newline.
|
|
|
|
report := func(f *os.File, outErr error) func(string) error {
|
|
|
|
return func(s string) (err error) {
|
|
|
|
buf := []byte(s)
|
|
|
|
buf = append(buf, '\n')
|
|
|
|
|
|
|
|
_, err = f.Write(buf)
|
|
|
|
if err != nil {
|
|
|
|
err = fmt.Errorf("Write: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
err = outErr
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
2015-03-23 07:54:20 +03:00
|
|
|
|
2015-03-23 08:08:46 +03:00
|
|
|
reportFlush := report(flushes, flushErr)
|
|
|
|
reportFsync := report(fsyncs, fsyncErr)
|
|
|
|
|
|
|
|
// Create the file system.
|
2015-03-24 08:17:56 +03:00
|
|
|
server, err = flushfs.NewFileSystem(reportFlush, reportFsync)
|
2015-03-23 08:08:46 +03:00
|
|
|
|
|
|
|
return
|
|
|
|
}
|
2015-03-23 08:01:16 +03:00
|
|
|
|
2015-03-24 08:17:56 +03:00
|
|
|
func makeFS() (server fuse.Server, err error) {
|
2015-03-23 08:01:16 +03:00
|
|
|
switch *fType {
|
|
|
|
default:
|
|
|
|
err = fmt.Errorf("Unknown FS type: %v", *fType)
|
|
|
|
|
|
|
|
case "flushfs":
|
2015-03-24 08:17:56 +03:00
|
|
|
server, err = makeFlushFS()
|
2015-03-23 08:01:16 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
2015-03-23 08:00:12 +03:00
|
|
|
|
2015-03-24 02:08:17 +03:00
|
|
|
func getReadyFile() (f *os.File, err error) {
|
|
|
|
if *fReadyFile == 0 {
|
|
|
|
err = errors.New("You must set --ready_file.")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
f = os.NewFile(uintptr(*fReadyFile), "(ready file)")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-03-23 07:52:27 +03:00
|
|
|
func main() {
|
2015-03-23 08:00:12 +03:00
|
|
|
flag.Parse()
|
|
|
|
|
2015-03-24 03:53:32 +03:00
|
|
|
// Allow parallelism in the file system implementation, to help flush out
|
|
|
|
// bugs like https://github.com/jacobsa/fuse/issues/4.
|
|
|
|
runtime.GOMAXPROCS(2)
|
|
|
|
|
2015-03-24 02:08:17 +03:00
|
|
|
// Grab the file to signal when ready.
|
|
|
|
readyFile, err := getReadyFile()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("getReadyFile:", err)
|
|
|
|
}
|
|
|
|
|
2015-03-23 08:00:12 +03:00
|
|
|
// Create an appropriate file system.
|
2015-03-24 08:17:56 +03:00
|
|
|
server, err := makeFS()
|
2015-03-23 08:00:12 +03:00
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("makeFS: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Mount the file system.
|
|
|
|
if *fMountPoint == "" {
|
|
|
|
log.Fatalf("You must set --mount_point.")
|
|
|
|
}
|
|
|
|
|
2015-03-24 08:17:56 +03:00
|
|
|
mfs, err := fuse.Mount(*fMountPoint, server, &fuse.MountConfig{})
|
2015-03-23 08:00:12 +03:00
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Mount: %v", err)
|
|
|
|
}
|
|
|
|
|
2015-03-24 02:08:17 +03:00
|
|
|
// Signal that it is ready.
|
|
|
|
_, err = readyFile.Write([]byte("x"))
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("readyFile.Write: %v", err)
|
|
|
|
}
|
|
|
|
|
2015-03-23 08:00:12 +03:00
|
|
|
// Wait for it to be unmounted.
|
|
|
|
if err = mfs.Join(context.Background()); err != nil {
|
|
|
|
log.Fatalf("Join: %v", err)
|
|
|
|
}
|
2015-03-23 07:52:27 +03:00
|
|
|
}
|