1 files changed, 238 insertions, 0 deletions
diff --git a/weed/mount/weedfs_attr.go b/weed/mount/weedfs_attr.go
new file mode 100644
index 000000000..55a29542a
--- /dev/null
+++ b/weed/mount/weedfs_attr.go
@@ -0,0 +1,238 @@
+package mount
+
+import (
+	"github.com/chrislusf/seaweedfs/weed/filer"
+	"github.com/chrislusf/seaweedfs/weed/glog"
+	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+	"github.com/hanwen/go-fuse/v2/fuse"
+	"os"
+	"syscall"
+	"time"
+)
+
+func (wfs *WFS) GetAttr(cancel <-chan struct{}, input *fuse.GetAttrIn, out *fuse.AttrOut) (code fuse.Status) {
+	if input.NodeId == 1 {
+		wfs.setRootAttr(out)
+		return fuse.OK
+	}
+
+	_, _, entry, status := wfs.maybeReadEntry(input.NodeId)
+	if status == fuse.OK {
+		out.AttrValid = 1
+		wfs.setAttrByPbEntry(&out.Attr, input.NodeId, entry)
+		return status
+	} else {
+		if fh, found := wfs.fhmap.FindFileHandle(input.NodeId); found {
+			out.AttrValid = 1
+			wfs.setAttrByPbEntry(&out.Attr, input.NodeId, fh.entry)
+			out.Nlink = 0
+			return fuse.OK
+		}
+	}
+
+	return status
+}
+
+func (wfs *WFS) SetAttr(cancel <-chan struct{}, input *fuse.SetAttrIn, out *fuse.AttrOut) (code fuse.Status) {
+
+	if wfs.IsOverQuota {
+		return fuse.Status(syscall.ENOSPC)
+	}
+
+	path, fh, entry, status := wfs.maybeReadEntry(input.NodeId)
+	if status != fuse.OK {
+		return status
+	}
+
+	if size, ok := input.GetSize(); ok {
+		glog.V(4).Infof("%v setattr set size=%v chunks=%d", path, size, len(entry.Chunks))
+		if size < filer.FileSize(entry) {
+			// fmt.Printf("truncate %v \n", fullPath)
+			var chunks []*filer_pb.FileChunk
+			var truncatedChunks []*filer_pb.FileChunk
+			for _, chunk := range entry.Chunks {
+				int64Size := int64(chunk.Size)
+				if chunk.Offset+int64Size > int64(size) {
+					// this chunk is truncated
+					int64Size = int64(size) - chunk.Offset
+					if int64Size > 0 {
+						chunks = append(chunks, chunk)
+						glog.V(4).Infof("truncated chunk %+v from %d to %d\n", chunk.GetFileIdString(), chunk.Size, int64Size)
+						chunk.Size = uint64(int64Size)
+					} else {
+						glog.V(4).Infof("truncated whole chunk %+v\n", chunk.GetFileIdString())
+						truncatedChunks = append(truncatedChunks, chunk)
+					}
+				}
+			}
+			// set the new chunks and reset entry cache
+			entry.Chunks = chunks
+			if fh != nil {
+				fh.entryViewCache = nil
+			}
+		}
+		entry.Attributes.Mtime = time.Now().Unix()
+		entry.Attributes.FileSize = size
+
+	}
+
+	if mode, ok := input.GetMode(); ok {
+		// glog.V(4).Infof("setAttr mode %o", mode)
+		entry.Attributes.FileMode = chmod(entry.Attributes.FileMode, mode)
+		if input.NodeId == 1 {
+			wfs.option.MountMode = os.FileMode(chmod(uint32(wfs.option.MountMode), mode))
+		}
+	}
+
+	if uid, ok := input.GetUID(); ok {
+		entry.Attributes.Uid = uid
+		if input.NodeId == 1 {
+			wfs.option.MountUid = uid
+		}
+	}
+
+	if gid, ok := input.GetGID(); ok {
+		entry.Attributes.Gid = gid
+		if input.NodeId == 1 {
+			wfs.option.MountGid = gid
+		}
+	}
+
+	if mtime, ok := input.GetMTime(); ok {
+		entry.Attributes.Mtime = mtime.Unix()
+	}
+
+	if atime, ok := input.GetATime(); ok {
+		entry.Attributes.Mtime = atime.Unix()
+	}
+
+	entry.Attributes.Mtime = time.Now().Unix()
+	out.AttrValid = 1
+	wfs.setAttrByPbEntry(&out.Attr, input.NodeId, entry)
+
+	if fh != nil {
+		fh.dirtyMetadata = true
+		return fuse.OK
+	}
+
+	return wfs.saveEntry(path, entry)
+
+}
+
+func (wfs *WFS) setRootAttr(out *fuse.AttrOut) {
+	now := uint64(time.Now().Unix())
+	out.AttrValid = 119
+	out.Ino = 1
+	setBlksize(&out.Attr, blockSize)
+	out.Uid = wfs.option.MountUid
+	out.Gid = wfs.option.MountGid
+	out.Mtime = now
+	out.Ctime = now
+	out.Atime = now
+	out.Mode = toSyscallType(os.ModeDir) | uint32(wfs.option.MountMode)
+	out.Nlink = 1
+}
+
+func (wfs *WFS) setAttrByPbEntry(out *fuse.Attr, inode uint64, entry *filer_pb.Entry) {
+	out.Ino = inode
+	out.Size = filer.FileSize(entry)
+	out.Blocks = (out.Size + blockSize - 1) / blockSize
+	setBlksize(out, blockSize)
+	out.Mtime = uint64(entry.Attributes.Mtime)
+	out.Ctime = uint64(entry.Attributes.Mtime)
+	out.Atime = uint64(entry.Attributes.Mtime)
+	out.Mode = toSyscallMode(os.FileMode(entry.Attributes.FileMode))
+	if entry.HardLinkCounter > 0 {
+		out.Nlink = uint32(entry.HardLinkCounter)
+	} else {
+		out.Nlink = 1
+	}
+	out.Uid = entry.Attributes.Uid
+	out.Gid = entry.Attributes.Gid
+	out.Rdev = entry.Attributes.Rdev
+}
+
+func (wfs *WFS) setAttrByFilerEntry(out *fuse.Attr, inode uint64, entry *filer.Entry) {
+	out.Ino = inode
+	out.Size = entry.FileSize
+	out.Blocks = (out.Size + blockSize - 1) / blockSize
+	setBlksize(out, blockSize)
+	out.Atime = uint64(entry.Attr.Mtime.Unix())
+	out.Mtime = uint64(entry.Attr.Mtime.Unix())
+	out.Ctime = uint64(entry.Attr.Mtime.Unix())
+	out.Mode = toSyscallMode(entry.Attr.Mode)
+	if entry.HardLinkCounter > 0 {
+		out.Nlink = uint32(entry.HardLinkCounter)
+	} else {
+		out.Nlink = 1
+	}
+	out.Uid = entry.Attr.Uid
+	out.Gid = entry.Attr.Gid
+	out.Rdev = entry.Attr.Rdev
+}
+
+func (wfs *WFS) outputPbEntry(out *fuse.EntryOut, inode uint64, entry *filer_pb.Entry) {
+	out.NodeId = inode
+	out.Generation = 1
+	out.EntryValid = 1
+	out.AttrValid = 1
+	wfs.setAttrByPbEntry(&out.Attr, inode, entry)
+}
+
+func (wfs *WFS) outputFilerEntry(out *fuse.EntryOut, inode uint64, entry *filer.Entry) {
+	out.NodeId = inode
+	out.Generation = 1
+	out.EntryValid = 1
+	out.AttrValid = 1
+	wfs.setAttrByFilerEntry(&out.Attr, inode, entry)
+}
+
+func chmod(existing uint32, mode uint32) uint32 {
+	return existing&^07777 | mode&07777
+}
+
+func toSyscallMode(mode os.FileMode) uint32 {
+	return toSyscallType(mode) | uint32(mode)
+}
+
+func toSyscallType(mode os.FileMode) uint32 {
+	switch mode & os.ModeType {
+	case os.ModeDir:
+		return syscall.S_IFDIR
+	case os.ModeSymlink:
+		return syscall.S_IFLNK
+	case os.ModeNamedPipe:
+		return syscall.S_IFIFO
+	case os.ModeSocket:
+		return syscall.S_IFSOCK
+	case os.ModeDevice:
+		return syscall.S_IFBLK
+	case os.ModeCharDevice:
+		return syscall.S_IFCHR
+	default:
+		return syscall.S_IFREG
+	}
+}
+
+func toOsFileType(mode uint32) os.FileMode {
+	switch mode & (syscall.S_IFMT & 0xffff) {
+	case syscall.S_IFDIR:
+		return os.ModeDir
+	case syscall.S_IFLNK:
+		return os.ModeSymlink
+	case syscall.S_IFIFO:
+		return os.ModeNamedPipe
+	case syscall.S_IFSOCK:
+		return os.ModeSocket
+	case syscall.S_IFBLK:
+		return os.ModeDevice
+	case syscall.S_IFCHR:
+		return os.ModeCharDevice
+	default:
+		return 0
+	}
+}
+
+func toOsFileMode(mode uint32) os.FileMode {
+	return toOsFileType(mode) | os.FileMode(mode&07777)
+}