seaweedfs/weed/command/mount_std.go

// +build linux darwin

package command

import (
	"fmt"
	"os"
	"os/user"
	"path"
	"runtime"
	"strconv"
	"strings"
	"time"

	"github.com/chrislusf/seaweedfs/weed/security"
	"github.com/jacobsa/daemonize"
	"github.com/spf13/viper"

	"github.com/chrislusf/seaweedfs/weed/filesys"
	"github.com/chrislusf/seaweedfs/weed/glog"
	"github.com/chrislusf/seaweedfs/weed/util"
	"github.com/seaweedfs/fuse"
	"github.com/seaweedfs/fuse/fs"
)

func runMount(cmd *Command, args []string) bool {

	util.SetupProfiling(*mountCpuProfile, *mountMemProfile)

	umask, umaskErr := strconv.ParseUint(*mountOptions.umaskString, 8, 64)
	if umaskErr != nil {
		fmt.Printf("can not parse umask %s", *mountOptions.umaskString)
		return false
	}

	return RunMount(
		*mountOptions.filer,
		*mountOptions.filerMountRootPath,
		*mountOptions.dir,
		*mountOptions.collection,
		*mountOptions.replication,
		*mountOptions.dataCenter,
		*mountOptions.chunkSizeLimitMB,
		*mountOptions.allowOthers,
		*mountOptions.ttlSec,
		*mountOptions.dirListingLimit,
		os.FileMode(umask),
	)
}

func RunMount(filer, filerMountRootPath, dir, collection, replication, dataCenter string, chunkSizeLimitMB int,
	allowOthers bool, ttlSec int, dirListingLimit int, umask os.FileMode) bool {

	util.LoadConfiguration("security", false)

	fmt.Printf("This is SeaweedFS version %s %s %s\n", util.VERSION, runtime.GOOS, runtime.GOARCH)
	if dir == "" {
		fmt.Printf("Please specify the mount directory via \"-dir\"")
		return false
	}
	if chunkSizeLimitMB <= 0 {
		fmt.Printf("Please specify a reasonable buffer size.")
		return false
	}

	fuse.Unmount(dir)

	uid, gid := uint32(0), uint32(0)

	// detect mount folder mode
	mountMode := os.ModeDir | 0755
	fileInfo, err := os.Stat(dir)
	if err == nil {
		mountMode = os.ModeDir | fileInfo.Mode()
		uid, gid = util.GetFileUidGid(fileInfo)
		fmt.Printf("mount point owner uid=%d gid=%d mode=%s\n", uid, gid, fileInfo.Mode())
	}

	if uid == 0 {
		if u, err := user.Current(); err == nil {
			if parsedId, pe := strconv.ParseUint(u.Uid, 10, 32); pe == nil {
				uid = uint32(parsedId)
			}
			if parsedId, pe := strconv.ParseUint(u.Gid, 10, 32); pe == nil {
				gid = uint32(parsedId)
			}
			fmt.Printf("current uid=%d gid=%d\n", uid, gid)
		}
	}

	mountName := path.Base(dir)

	options := []fuse.MountOption{
		fuse.VolumeName(mountName),
		fuse.FSName("SeaweedFS"),
		fuse.Subtype("SeaweedFS"),
		fuse.NoAppleDouble(),
		fuse.NoAppleXattr(),
		fuse.NoBrowse(),
		fuse.AutoXattr(),
		fuse.ExclCreate(),
		fuse.DaemonTimeout("3600"),
		fuse.AllowSUID(),
		fuse.DefaultPermissions(),
		fuse.MaxReadahead(1024 * 128),
		fuse.AsyncRead(),
		fuse.WritebackCache(),
		fuse.AllowNonEmptyMount(),
	}
	if allowOthers {
		options = append(options, fuse.AllowOther())
	}

	c, err := fuse.Mount(dir, options...)
	if err != nil {
		glog.Fatal(err)
		daemonize.SignalOutcome(err)
		return false
	}

	util.OnInterrupt(func() {
		fuse.Unmount(dir)
		c.Close()
	})

	filerGrpcAddress, err := parseFilerGrpcAddress(filer)
	if err != nil {
		glog.Fatal(err)
		daemonize.SignalOutcome(err)
		return false
	}

	mountRoot := filerMountRootPath
	if mountRoot != "/" && strings.HasSuffix(mountRoot, "/") {
		mountRoot = mountRoot[0 : len(mountRoot)-1]
	}

	daemonize.SignalOutcome(nil)

	err = fs.Serve(c, filesys.NewSeaweedFileSystem(&filesys.Option{
		FilerGrpcAddress:   filerGrpcAddress,
		GrpcDialOption:     security.LoadClientTLS(viper.Sub("grpc"), "client"),
		FilerMountRootPath: mountRoot,
		Collection:         collection,
		Replication:        replication,
		TtlSec:             int32(ttlSec),
		ChunkSizeLimit:     int64(chunkSizeLimitMB) * 1024 * 1024,
		DataCenter:         dataCenter,
		DirListingLimit:    dirListingLimit,
		EntryCacheTtl:      3 * time.Second,
		MountUid:           uid,
		MountGid:           gid,
		MountMode:          mountMode,
		MountCtime:         fileInfo.ModTime(),
		MountMtime:         time.Now(),
		Umask:              umask,
	}))
	if err != nil {
		fuse.Unmount(dir)
	}

	// check if the mount process has an error to report
	<-c.Ready
	if err := c.MountError; err != nil {
		glog.Fatal(err)
		daemonize.SignalOutcome(err)
	}

	return true
}