984 lines
21 KiB
Go
984 lines
21 KiB
Go
// Copyright 2017 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 fs
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/user"
|
|
"path/filepath"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var OsFs FileSystem = osFs{}
|
|
|
|
func NewMockFs(files map[string][]byte) *MockFs {
|
|
workDir := "/cwd"
|
|
fs := &MockFs{
|
|
Clock: NewClock(time.Unix(2, 2)),
|
|
workDir: workDir,
|
|
}
|
|
fs.root = *fs.newDir()
|
|
fs.MkDirs(workDir)
|
|
|
|
for path, bytes := range files {
|
|
dir := filepath.Dir(path)
|
|
fs.MkDirs(dir)
|
|
fs.WriteFile(path, bytes, 0777)
|
|
}
|
|
|
|
return fs
|
|
}
|
|
|
|
type FileSystem interface {
|
|
// getting information about files
|
|
Open(name string) (file io.ReadCloser, err error)
|
|
Lstat(path string) (stats os.FileInfo, err error)
|
|
Stat(path string) (stats os.FileInfo, err error)
|
|
ReadDir(path string) (contents []DirEntryInfo, err error)
|
|
|
|
InodeNumber(info os.FileInfo) (number uint64, err error)
|
|
DeviceNumber(info os.FileInfo) (number uint64, err error)
|
|
PermTime(info os.FileInfo) (time time.Time, err error)
|
|
|
|
// changing contents of the filesystem
|
|
Rename(oldPath string, newPath string) (err error)
|
|
WriteFile(path string, data []byte, perm os.FileMode) (err error)
|
|
Remove(path string) (err error)
|
|
RemoveAll(path string) (err error)
|
|
|
|
// metadata about the filesystem
|
|
ViewId() (id string) // Some unique id of the user accessing the filesystem
|
|
}
|
|
|
|
// DentryInfo is a subset of the functionality available through os.FileInfo that might be able
|
|
// to be gleaned through only a syscall.Getdents without requiring a syscall.Lstat of every file.
|
|
type DirEntryInfo interface {
|
|
Name() string
|
|
Mode() os.FileMode // the file type encoded as an os.FileMode
|
|
IsDir() bool
|
|
}
|
|
|
|
type dirEntryInfo struct {
|
|
name string
|
|
mode os.FileMode
|
|
modeExists bool
|
|
}
|
|
|
|
var _ DirEntryInfo = os.FileInfo(nil)
|
|
|
|
func (d *dirEntryInfo) Name() string { return d.name }
|
|
func (d *dirEntryInfo) Mode() os.FileMode { return d.mode }
|
|
func (d *dirEntryInfo) IsDir() bool { return d.mode.IsDir() }
|
|
func (d *dirEntryInfo) String() string { return d.name + ": " + d.mode.String() }
|
|
|
|
// osFs implements FileSystem using the local disk.
|
|
type osFs struct{}
|
|
|
|
var _ FileSystem = (*osFs)(nil)
|
|
|
|
func (osFs) Open(name string) (io.ReadCloser, error) { return os.Open(name) }
|
|
|
|
func (osFs) Lstat(path string) (stats os.FileInfo, err error) {
|
|
return os.Lstat(path)
|
|
}
|
|
|
|
func (osFs) Stat(path string) (stats os.FileInfo, err error) {
|
|
return os.Stat(path)
|
|
}
|
|
|
|
func (osFs) ReadDir(path string) (contents []DirEntryInfo, err error) {
|
|
entries, err := readdir(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, entry := range entries {
|
|
contents = append(contents, entry)
|
|
}
|
|
|
|
return contents, nil
|
|
}
|
|
|
|
func (osFs) Rename(oldPath string, newPath string) error {
|
|
return os.Rename(oldPath, newPath)
|
|
}
|
|
|
|
func (osFs) WriteFile(path string, data []byte, perm os.FileMode) error {
|
|
return ioutil.WriteFile(path, data, perm)
|
|
}
|
|
|
|
func (osFs) Remove(path string) error {
|
|
return os.Remove(path)
|
|
}
|
|
|
|
func (osFs) RemoveAll(path string) error {
|
|
return os.RemoveAll(path)
|
|
}
|
|
|
|
func (osFs) ViewId() (id string) {
|
|
user, err := user.Current()
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
username := user.Username
|
|
|
|
hostname, err := os.Hostname()
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return username + "@" + hostname
|
|
}
|
|
|
|
type Clock struct {
|
|
time time.Time
|
|
}
|
|
|
|
func NewClock(startTime time.Time) *Clock {
|
|
return &Clock{time: startTime}
|
|
|
|
}
|
|
|
|
func (c *Clock) Tick() {
|
|
c.time = c.time.Add(time.Microsecond)
|
|
}
|
|
|
|
func (c *Clock) Time() time.Time {
|
|
return c.time
|
|
}
|
|
|
|
// given "/a/b/c/d", pathSplit returns ("/a/b/c", "d")
|
|
func pathSplit(path string) (dir string, leaf string) {
|
|
dir, leaf = filepath.Split(path)
|
|
if dir != "/" && len(dir) > 0 {
|
|
dir = dir[:len(dir)-1]
|
|
}
|
|
return dir, leaf
|
|
}
|
|
|
|
// MockFs supports singlethreaded writes and multithreaded reads
|
|
type MockFs struct {
|
|
// configuration
|
|
viewId string //
|
|
deviceNumber uint64
|
|
|
|
// implementation
|
|
root mockDir
|
|
Clock *Clock
|
|
workDir string
|
|
nextInodeNumber uint64
|
|
|
|
// history of requests, for tests to check
|
|
StatCalls []string
|
|
ReadDirCalls []string
|
|
aggregatesLock sync.Mutex
|
|
}
|
|
|
|
var _ FileSystem = (*MockFs)(nil)
|
|
|
|
type mockInode struct {
|
|
modTime time.Time
|
|
permTime time.Time
|
|
sys interface{}
|
|
inodeNumber uint64
|
|
readErr error
|
|
}
|
|
|
|
func (m mockInode) ModTime() time.Time {
|
|
return m.modTime
|
|
}
|
|
|
|
func (m mockInode) Sys() interface{} {
|
|
return m.sys
|
|
}
|
|
|
|
type mockFile struct {
|
|
bytes []byte
|
|
|
|
mockInode
|
|
}
|
|
|
|
type mockLink struct {
|
|
target string
|
|
|
|
mockInode
|
|
}
|
|
|
|
type mockDir struct {
|
|
mockInode
|
|
|
|
subdirs map[string]*mockDir
|
|
files map[string]*mockFile
|
|
symlinks map[string]*mockLink
|
|
}
|
|
|
|
func (m *MockFs) resolve(path string, followLastLink bool) (result string, err error) {
|
|
if !filepath.IsAbs(path) {
|
|
path = filepath.Join(m.workDir, path)
|
|
}
|
|
path = filepath.Clean(path)
|
|
|
|
return m.followLinks(path, followLastLink, 10)
|
|
}
|
|
|
|
// note that followLinks can return a file path that doesn't exist
|
|
func (m *MockFs) followLinks(path string, followLastLink bool, count int) (canonicalPath string, err error) {
|
|
if path == "/" {
|
|
return path, nil
|
|
}
|
|
|
|
parentPath, leaf := pathSplit(path)
|
|
if parentPath == path {
|
|
err = fmt.Errorf("Internal error: %v yields itself as a parent", path)
|
|
panic(err.Error())
|
|
}
|
|
|
|
parentPath, err = m.followLinks(parentPath, true, count)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
parentNode, err := m.getDir(parentPath, false)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if parentNode.readErr != nil {
|
|
return "", &os.PathError{
|
|
Op: "read",
|
|
Path: path,
|
|
Err: parentNode.readErr,
|
|
}
|
|
}
|
|
|
|
link, isLink := parentNode.symlinks[leaf]
|
|
if isLink && followLastLink {
|
|
if count <= 0 {
|
|
// probably a loop
|
|
return "", &os.PathError{
|
|
Op: "read",
|
|
Path: path,
|
|
Err: fmt.Errorf("too many levels of symbolic links"),
|
|
}
|
|
}
|
|
|
|
if link.readErr != nil {
|
|
return "", &os.PathError{
|
|
Op: "read",
|
|
Path: path,
|
|
Err: link.readErr,
|
|
}
|
|
}
|
|
|
|
target := m.followLink(link, parentPath)
|
|
return m.followLinks(target, followLastLink, count-1)
|
|
}
|
|
return path, nil
|
|
}
|
|
|
|
func (m *MockFs) followLink(link *mockLink, parentPath string) (result string) {
|
|
return filepath.Clean(filepath.Join(parentPath, link.target))
|
|
}
|
|
|
|
func (m *MockFs) getFile(parentDir *mockDir, fileName string) (file *mockFile, err error) {
|
|
file, isFile := parentDir.files[fileName]
|
|
if !isFile {
|
|
_, isDir := parentDir.subdirs[fileName]
|
|
_, isLink := parentDir.symlinks[fileName]
|
|
if isDir || isLink {
|
|
return nil, &os.PathError{
|
|
Op: "open",
|
|
Path: fileName,
|
|
Err: os.ErrInvalid,
|
|
}
|
|
}
|
|
|
|
return nil, &os.PathError{
|
|
Op: "open",
|
|
Path: fileName,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
if file.readErr != nil {
|
|
return nil, &os.PathError{
|
|
Op: "open",
|
|
Path: fileName,
|
|
Err: file.readErr,
|
|
}
|
|
}
|
|
return file, nil
|
|
}
|
|
|
|
func (m *MockFs) getInode(parentDir *mockDir, name string) (inode *mockInode, err error) {
|
|
file, isFile := parentDir.files[name]
|
|
if isFile {
|
|
return &file.mockInode, nil
|
|
}
|
|
link, isLink := parentDir.symlinks[name]
|
|
if isLink {
|
|
return &link.mockInode, nil
|
|
}
|
|
dir, isDir := parentDir.subdirs[name]
|
|
if isDir {
|
|
return &dir.mockInode, nil
|
|
}
|
|
return nil, &os.PathError{
|
|
Op: "stat",
|
|
Path: name,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
|
|
}
|
|
|
|
func (m *MockFs) Open(path string) (io.ReadCloser, error) {
|
|
path, err := m.resolve(path, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
parentPath, base := pathSplit(path)
|
|
parentDir, err := m.getDir(parentPath, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
file, err := m.getFile(parentDir, base)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return struct {
|
|
io.Closer
|
|
*bytes.Reader
|
|
}{
|
|
ioutil.NopCloser(nil),
|
|
bytes.NewReader(file.bytes),
|
|
}, nil
|
|
|
|
}
|
|
|
|
// a mockFileInfo is for exporting file stats in a way that satisfies the FileInfo interface
|
|
type mockFileInfo struct {
|
|
path string
|
|
size int64
|
|
modTime time.Time // time at which the inode's contents were modified
|
|
permTime time.Time // time at which the inode's permissions were modified
|
|
mode os.FileMode
|
|
inodeNumber uint64
|
|
deviceNumber uint64
|
|
}
|
|
|
|
func (m *mockFileInfo) Name() string {
|
|
return m.path
|
|
}
|
|
|
|
func (m *mockFileInfo) Size() int64 {
|
|
return m.size
|
|
}
|
|
|
|
func (m *mockFileInfo) Mode() os.FileMode {
|
|
return m.mode
|
|
}
|
|
|
|
func (m *mockFileInfo) ModTime() time.Time {
|
|
return m.modTime
|
|
}
|
|
|
|
func (m *mockFileInfo) IsDir() bool {
|
|
return m.mode&os.ModeDir != 0
|
|
}
|
|
|
|
func (m *mockFileInfo) Sys() interface{} {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) dirToFileInfo(d *mockDir, path string) (info *mockFileInfo) {
|
|
return &mockFileInfo{
|
|
path: filepath.Base(path),
|
|
size: 1,
|
|
modTime: d.modTime,
|
|
permTime: d.permTime,
|
|
mode: os.ModeDir,
|
|
inodeNumber: d.inodeNumber,
|
|
deviceNumber: m.deviceNumber,
|
|
}
|
|
|
|
}
|
|
|
|
func (m *MockFs) fileToFileInfo(f *mockFile, path string) (info *mockFileInfo) {
|
|
return &mockFileInfo{
|
|
path: filepath.Base(path),
|
|
size: 1,
|
|
modTime: f.modTime,
|
|
permTime: f.permTime,
|
|
mode: 0,
|
|
inodeNumber: f.inodeNumber,
|
|
deviceNumber: m.deviceNumber,
|
|
}
|
|
}
|
|
|
|
func (m *MockFs) linkToFileInfo(l *mockLink, path string) (info *mockFileInfo) {
|
|
return &mockFileInfo{
|
|
path: filepath.Base(path),
|
|
size: 1,
|
|
modTime: l.modTime,
|
|
permTime: l.permTime,
|
|
mode: os.ModeSymlink,
|
|
inodeNumber: l.inodeNumber,
|
|
deviceNumber: m.deviceNumber,
|
|
}
|
|
}
|
|
|
|
func (m *MockFs) Lstat(path string) (stats os.FileInfo, err error) {
|
|
// update aggregates
|
|
m.aggregatesLock.Lock()
|
|
m.StatCalls = append(m.StatCalls, path)
|
|
m.aggregatesLock.Unlock()
|
|
|
|
// resolve symlinks
|
|
path, err = m.resolve(path, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// special case for root dir
|
|
if path == "/" {
|
|
return m.dirToFileInfo(&m.root, "/"), nil
|
|
}
|
|
|
|
// determine type and handle appropriately
|
|
parentPath, baseName := pathSplit(path)
|
|
dir, err := m.getDir(parentPath, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
subdir, subdirExists := dir.subdirs[baseName]
|
|
if subdirExists {
|
|
return m.dirToFileInfo(subdir, path), nil
|
|
}
|
|
file, fileExists := dir.files[baseName]
|
|
if fileExists {
|
|
return m.fileToFileInfo(file, path), nil
|
|
}
|
|
link, linkExists := dir.symlinks[baseName]
|
|
if linkExists {
|
|
return m.linkToFileInfo(link, path), nil
|
|
}
|
|
// not found
|
|
return nil, &os.PathError{
|
|
Op: "stat",
|
|
Path: path,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
|
|
func (m *MockFs) Stat(path string) (stats os.FileInfo, err error) {
|
|
// resolve symlinks
|
|
path, err = m.resolve(path, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return m.Lstat(path)
|
|
}
|
|
|
|
func (m *MockFs) InodeNumber(info os.FileInfo) (number uint64, err error) {
|
|
mockInfo, ok := info.(*mockFileInfo)
|
|
if ok {
|
|
return mockInfo.inodeNumber, nil
|
|
}
|
|
return 0, fmt.Errorf("%v is not a mockFileInfo", info)
|
|
}
|
|
func (m *MockFs) DeviceNumber(info os.FileInfo) (number uint64, err error) {
|
|
mockInfo, ok := info.(*mockFileInfo)
|
|
if ok {
|
|
return mockInfo.deviceNumber, nil
|
|
}
|
|
return 0, fmt.Errorf("%v is not a mockFileInfo", info)
|
|
}
|
|
func (m *MockFs) PermTime(info os.FileInfo) (when time.Time, err error) {
|
|
mockInfo, ok := info.(*mockFileInfo)
|
|
if ok {
|
|
return mockInfo.permTime, nil
|
|
}
|
|
return time.Date(0, 0, 0, 0, 0, 0, 0, nil),
|
|
fmt.Errorf("%v is not a mockFileInfo", info)
|
|
}
|
|
|
|
func (m *MockFs) ReadDir(path string) (contents []DirEntryInfo, err error) {
|
|
// update aggregates
|
|
m.aggregatesLock.Lock()
|
|
m.ReadDirCalls = append(m.ReadDirCalls, path)
|
|
m.aggregatesLock.Unlock()
|
|
|
|
// locate directory
|
|
path, err = m.resolve(path, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
results := []DirEntryInfo{}
|
|
dir, err := m.getDir(path, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if dir.readErr != nil {
|
|
return nil, &os.PathError{
|
|
Op: "read",
|
|
Path: path,
|
|
Err: dir.readErr,
|
|
}
|
|
}
|
|
// describe its contents
|
|
for name, subdir := range dir.subdirs {
|
|
dirInfo := m.dirToFileInfo(subdir, name)
|
|
results = append(results, dirInfo)
|
|
}
|
|
for name, file := range dir.files {
|
|
info := m.fileToFileInfo(file, name)
|
|
results = append(results, info)
|
|
}
|
|
for name, link := range dir.symlinks {
|
|
info := m.linkToFileInfo(link, name)
|
|
results = append(results, info)
|
|
}
|
|
return results, nil
|
|
}
|
|
|
|
func (m *MockFs) Rename(sourcePath string, destPath string) error {
|
|
// validate source parent exists
|
|
sourcePath, err := m.resolve(sourcePath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
sourceParentPath := filepath.Dir(sourcePath)
|
|
sourceParentDir, err := m.getDir(sourceParentPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if sourceParentDir == nil {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: sourcePath,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
if sourceParentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: sourcePath,
|
|
Err: sourceParentDir.readErr,
|
|
}
|
|
}
|
|
|
|
// validate dest parent exists
|
|
destPath, err = m.resolve(destPath, false)
|
|
destParentPath := filepath.Dir(destPath)
|
|
destParentDir, err := m.getDir(destParentPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if destParentDir == nil {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: destParentPath,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
if destParentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: destParentPath,
|
|
Err: destParentDir.readErr,
|
|
}
|
|
}
|
|
// check the source and dest themselves
|
|
sourceBase := filepath.Base(sourcePath)
|
|
destBase := filepath.Base(destPath)
|
|
|
|
file, sourceIsFile := sourceParentDir.files[sourceBase]
|
|
dir, sourceIsDir := sourceParentDir.subdirs[sourceBase]
|
|
link, sourceIsLink := sourceParentDir.symlinks[sourceBase]
|
|
|
|
// validate that the source exists
|
|
if !sourceIsFile && !sourceIsDir && !sourceIsLink {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: sourcePath,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
|
|
}
|
|
|
|
// validate the destination doesn't already exist as an incompatible type
|
|
_, destWasFile := destParentDir.files[destBase]
|
|
_, destWasDir := destParentDir.subdirs[destBase]
|
|
_, destWasLink := destParentDir.symlinks[destBase]
|
|
|
|
if destWasDir {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: destPath,
|
|
Err: errors.New("destination exists as a directory"),
|
|
}
|
|
}
|
|
|
|
if sourceIsDir && (destWasFile || destWasLink) {
|
|
return &os.PathError{
|
|
Op: "move",
|
|
Path: destPath,
|
|
Err: errors.New("destination exists as a file"),
|
|
}
|
|
}
|
|
|
|
if destWasFile {
|
|
delete(destParentDir.files, destBase)
|
|
}
|
|
if destWasDir {
|
|
delete(destParentDir.subdirs, destBase)
|
|
}
|
|
if destWasLink {
|
|
delete(destParentDir.symlinks, destBase)
|
|
}
|
|
|
|
if sourceIsFile {
|
|
destParentDir.files[destBase] = file
|
|
delete(sourceParentDir.files, sourceBase)
|
|
}
|
|
if sourceIsDir {
|
|
destParentDir.subdirs[destBase] = dir
|
|
delete(sourceParentDir.subdirs, sourceBase)
|
|
}
|
|
if sourceIsLink {
|
|
destParentDir.symlinks[destBase] = link
|
|
delete(destParentDir.symlinks, sourceBase)
|
|
}
|
|
|
|
destParentDir.modTime = m.Clock.Time()
|
|
sourceParentDir.modTime = m.Clock.Time()
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) newInodeNumber() uint64 {
|
|
result := m.nextInodeNumber
|
|
m.nextInodeNumber++
|
|
return result
|
|
}
|
|
|
|
func (m *MockFs) WriteFile(filePath string, data []byte, perm os.FileMode) error {
|
|
filePath, err := m.resolve(filePath, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
parentPath := filepath.Dir(filePath)
|
|
parentDir, err := m.getDir(parentPath, false)
|
|
if err != nil || parentDir == nil {
|
|
return &os.PathError{
|
|
Op: "write",
|
|
Path: parentPath,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
if parentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "write",
|
|
Path: parentPath,
|
|
Err: parentDir.readErr,
|
|
}
|
|
}
|
|
|
|
baseName := filepath.Base(filePath)
|
|
_, exists := parentDir.files[baseName]
|
|
if !exists {
|
|
parentDir.modTime = m.Clock.Time()
|
|
parentDir.files[baseName] = m.newFile()
|
|
} else {
|
|
readErr := parentDir.files[baseName].readErr
|
|
if readErr != nil {
|
|
return &os.PathError{
|
|
Op: "write",
|
|
Path: filePath,
|
|
Err: readErr,
|
|
}
|
|
}
|
|
}
|
|
file := parentDir.files[baseName]
|
|
file.bytes = data
|
|
file.modTime = m.Clock.Time()
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) newFile() *mockFile {
|
|
newFile := &mockFile{}
|
|
newFile.inodeNumber = m.newInodeNumber()
|
|
newFile.modTime = m.Clock.Time()
|
|
newFile.permTime = newFile.modTime
|
|
return newFile
|
|
}
|
|
|
|
func (m *MockFs) newDir() *mockDir {
|
|
newDir := &mockDir{
|
|
subdirs: make(map[string]*mockDir, 0),
|
|
files: make(map[string]*mockFile, 0),
|
|
symlinks: make(map[string]*mockLink, 0),
|
|
}
|
|
newDir.inodeNumber = m.newInodeNumber()
|
|
newDir.modTime = m.Clock.Time()
|
|
newDir.permTime = newDir.modTime
|
|
return newDir
|
|
}
|
|
|
|
func (m *MockFs) newLink(target string) *mockLink {
|
|
newLink := &mockLink{
|
|
target: target,
|
|
}
|
|
newLink.inodeNumber = m.newInodeNumber()
|
|
newLink.modTime = m.Clock.Time()
|
|
newLink.permTime = newLink.modTime
|
|
|
|
return newLink
|
|
}
|
|
func (m *MockFs) MkDirs(path string) error {
|
|
_, err := m.getDir(path, true)
|
|
return err
|
|
}
|
|
|
|
// getDir doesn't support symlinks
|
|
func (m *MockFs) getDir(path string, createIfMissing bool) (dir *mockDir, err error) {
|
|
cleanedPath := filepath.Clean(path)
|
|
if cleanedPath == "/" {
|
|
return &m.root, nil
|
|
}
|
|
|
|
parentPath, leaf := pathSplit(cleanedPath)
|
|
if len(parentPath) >= len(path) {
|
|
return &m.root, nil
|
|
}
|
|
parent, err := m.getDir(parentPath, createIfMissing)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if parent.readErr != nil {
|
|
return nil, &os.PathError{
|
|
Op: "stat",
|
|
Path: path,
|
|
Err: parent.readErr,
|
|
}
|
|
}
|
|
childDir, dirExists := parent.subdirs[leaf]
|
|
if !dirExists {
|
|
if createIfMissing {
|
|
// confirm that a file with the same name doesn't already exist
|
|
_, fileExists := parent.files[leaf]
|
|
if fileExists {
|
|
return nil, &os.PathError{
|
|
Op: "mkdir",
|
|
Path: path,
|
|
Err: os.ErrExist,
|
|
}
|
|
}
|
|
// create this directory
|
|
childDir = m.newDir()
|
|
parent.subdirs[leaf] = childDir
|
|
parent.modTime = m.Clock.Time()
|
|
} else {
|
|
return nil, &os.PathError{
|
|
Op: "stat",
|
|
Path: path,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
}
|
|
return childDir, nil
|
|
|
|
}
|
|
|
|
func (m *MockFs) Remove(path string) (err error) {
|
|
path, err = m.resolve(path, false)
|
|
parentPath, leaf := pathSplit(path)
|
|
if len(leaf) == 0 {
|
|
return fmt.Errorf("Cannot remove %v\n", path)
|
|
}
|
|
parentDir, err := m.getDir(parentPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if parentDir == nil {
|
|
return &os.PathError{
|
|
Op: "remove",
|
|
Path: path,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
if parentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "remove",
|
|
Path: path,
|
|
Err: parentDir.readErr,
|
|
}
|
|
}
|
|
_, isDir := parentDir.subdirs[leaf]
|
|
if isDir {
|
|
return &os.PathError{
|
|
Op: "remove",
|
|
Path: path,
|
|
Err: os.ErrInvalid,
|
|
}
|
|
}
|
|
_, isLink := parentDir.symlinks[leaf]
|
|
if isLink {
|
|
delete(parentDir.symlinks, leaf)
|
|
} else {
|
|
_, isFile := parentDir.files[leaf]
|
|
if !isFile {
|
|
return &os.PathError{
|
|
Op: "remove",
|
|
Path: path,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
delete(parentDir.files, leaf)
|
|
}
|
|
parentDir.modTime = m.Clock.Time()
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) Symlink(oldPath string, newPath string) (err error) {
|
|
newPath, err = m.resolve(newPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
newParentPath, leaf := pathSplit(newPath)
|
|
newParentDir, err := m.getDir(newParentPath, false)
|
|
if newParentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "link",
|
|
Path: newPath,
|
|
Err: newParentDir.readErr,
|
|
}
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
newParentDir.symlinks[leaf] = m.newLink(oldPath)
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) RemoveAll(path string) (err error) {
|
|
path, err = m.resolve(path, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
parentPath, leaf := pathSplit(path)
|
|
if len(leaf) == 0 {
|
|
return fmt.Errorf("Cannot remove %v\n", path)
|
|
}
|
|
parentDir, err := m.getDir(parentPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if parentDir == nil {
|
|
return &os.PathError{
|
|
Op: "removeAll",
|
|
Path: path,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
if parentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "removeAll",
|
|
Path: path,
|
|
Err: parentDir.readErr,
|
|
}
|
|
|
|
}
|
|
_, isFile := parentDir.files[leaf]
|
|
_, isLink := parentDir.symlinks[leaf]
|
|
if isFile || isLink {
|
|
return m.Remove(path)
|
|
}
|
|
_, isDir := parentDir.subdirs[leaf]
|
|
if !isDir {
|
|
if !isDir {
|
|
return &os.PathError{
|
|
Op: "removeAll",
|
|
Path: path,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
}
|
|
}
|
|
|
|
delete(parentDir.subdirs, leaf)
|
|
parentDir.modTime = m.Clock.Time()
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) SetReadable(path string, readable bool) error {
|
|
var readErr error
|
|
if !readable {
|
|
readErr = os.ErrPermission
|
|
}
|
|
return m.SetReadErr(path, readErr)
|
|
}
|
|
|
|
func (m *MockFs) SetReadErr(path string, readErr error) error {
|
|
path, err := m.resolve(path, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
parentPath, leaf := filepath.Split(path)
|
|
parentDir, err := m.getDir(parentPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if parentDir.readErr != nil {
|
|
return &os.PathError{
|
|
Op: "chmod",
|
|
Path: parentPath,
|
|
Err: parentDir.readErr,
|
|
}
|
|
}
|
|
|
|
inode, err := m.getInode(parentDir, leaf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
inode.readErr = readErr
|
|
inode.permTime = m.Clock.Time()
|
|
return nil
|
|
}
|
|
|
|
func (m *MockFs) ClearMetrics() {
|
|
m.ReadDirCalls = []string{}
|
|
m.StatCalls = []string{}
|
|
}
|
|
|
|
func (m *MockFs) ViewId() (id string) {
|
|
return m.viewId
|
|
}
|
|
|
|
func (m *MockFs) SetViewId(id string) {
|
|
m.viewId = id
|
|
}
|
|
func (m *MockFs) SetDeviceNumber(deviceNumber uint64) {
|
|
m.deviceNumber = deviceNumber
|
|
}
|