fix: Prevent creating files in place of symlinks to directories
This commit is contained in:
@@ -78,9 +78,6 @@ func (f *STFS) Name() string {
|
|||||||
"name": config.FileSystemNameSTFS,
|
"name": config.FileSystemNameSTFS,
|
||||||
})
|
})
|
||||||
|
|
||||||
f.ioLock.Lock()
|
|
||||||
defer f.ioLock.Unlock()
|
|
||||||
|
|
||||||
return config.FileSystemNameSTFS
|
return config.FileSystemNameSTFS
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -442,6 +439,19 @@ func (f *STFS) OpenFile(name string, flag int, perm os.FileMode) (afero.File, er
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if target, err := inventory.Stat(
|
||||||
|
f.metadata,
|
||||||
|
|
||||||
|
name,
|
||||||
|
true,
|
||||||
|
|
||||||
|
f.onHeader,
|
||||||
|
); err == nil {
|
||||||
|
if target.Typeflag == tar.TypeDir {
|
||||||
|
return nil, config.ErrIsDirectory
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if err := f.mknodeWithoutLocking(false, name, perm, false, "", false); err != nil {
|
if err := f.mknodeWithoutLocking(false, name, perm, false, "", false); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -656,36 +656,103 @@ var createTests = []struct {
|
|||||||
name string
|
name string
|
||||||
args createArgs
|
args createArgs
|
||||||
wantErr bool
|
wantErr bool
|
||||||
|
prepare func(symFs) error
|
||||||
}{
|
}{
|
||||||
{
|
{
|
||||||
"Can create file /test.txt",
|
"Can create file /test.txt",
|
||||||
createArgs{"/test.txt"},
|
createArgs{"/test.txt"},
|
||||||
false,
|
false,
|
||||||
|
func(sf symFs) error { return nil },
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"Can create file /test.txt/",
|
"Can create file /test.txt/",
|
||||||
createArgs{"/test.txt/"},
|
createArgs{"/test.txt/"},
|
||||||
false,
|
false,
|
||||||
|
func(sf symFs) error { return nil },
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"Can not create existing file/directory /",
|
"Can not create existing file /",
|
||||||
createArgs{"/"},
|
createArgs{"/"},
|
||||||
true,
|
true,
|
||||||
|
func(sf symFs) error { return nil },
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"Can create file ' '",
|
"Can create file ' '",
|
||||||
createArgs{" "},
|
createArgs{" "},
|
||||||
false,
|
false,
|
||||||
|
func(sf symFs) error { return nil },
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"Can create file ''",
|
"Can create file ''",
|
||||||
createArgs{""},
|
createArgs{""},
|
||||||
true,
|
true,
|
||||||
|
func(sf symFs) error { return nil },
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"Can not create /nonexistent/test.txt",
|
"Can not create /nonexistent/test.txt",
|
||||||
createArgs{"/nonexistent/test.txt"},
|
createArgs{"/nonexistent/test.txt"},
|
||||||
true,
|
true,
|
||||||
|
func(sf symFs) error { return nil },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Can not create file in place of symlink to root",
|
||||||
|
createArgs{"/existingsymlink"},
|
||||||
|
true,
|
||||||
|
func(sf symFs) error {
|
||||||
|
if err := sf.SymlinkIfPossible("/", "/existingsymlink"); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Can create file in place of broken symlink /brokensymlink",
|
||||||
|
createArgs{"/brokensymlink"},
|
||||||
|
false,
|
||||||
|
func(sf symFs) error {
|
||||||
|
if err := sf.SymlinkIfPossible("/test.txt", "/brokensymlink"); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Can create file in place of existing symlink /existingsymlink to file",
|
||||||
|
createArgs{"/existingsymlink"},
|
||||||
|
false,
|
||||||
|
func(sf symFs) error {
|
||||||
|
file, err := sf.Create("/test.txt")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := file.Close(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := sf.SymlinkIfPossible("/test.txt", "/existingsymlink"); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Can not create file in place of existing symlink /existingsymlink to directory",
|
||||||
|
createArgs{"/existingsymlink"},
|
||||||
|
true,
|
||||||
|
func(sf symFs) error {
|
||||||
|
if err := sf.Mkdir("/mydir", os.ModePerm); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := sf.SymlinkIfPossible("/mydir", "/existingsymlink"); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -694,36 +761,47 @@ func TestSTFS_Create(t *testing.T) {
|
|||||||
tt := tt
|
tt := tt
|
||||||
|
|
||||||
runTestForAllFss(t, tt.name, true, true, true, func(t *testing.T, fs fsConfig) {
|
runTestForAllFss(t, tt.name, true, true, true, func(t *testing.T, fs fsConfig) {
|
||||||
file, err := fs.fs.Create(tt.args.name)
|
symFs, ok := fs.fs.(symFs)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := tt.prepare(symFs); err != nil {
|
||||||
|
t.Errorf("%v prepare() error = %v", symFs.Name(), err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := symFs.Create(tt.args.name)
|
||||||
if (err != nil) != tt.wantErr {
|
if (err != nil) != tt.wantErr {
|
||||||
t.Errorf("%v.Create() error = %v, wantErr %v", fs.fs.Name(), err, tt.wantErr)
|
t.Errorf("%v.Create() error = %v, wantErr %v", symFs.Name(), err, tt.wantErr)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if !tt.wantErr {
|
if !tt.wantErr {
|
||||||
want, err := fs.fs.Stat(tt.args.name)
|
want, err := symFs.Stat(tt.args.name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("%v.Stat() error = %v, wantErr %v", fs.fs.Name(), err, tt.wantErr)
|
t.Errorf("%v.Stat() error = %v, wantErr %v", symFs.Name(), err, tt.wantErr)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if file == nil {
|
if file == nil {
|
||||||
t.Errorf("%v.Create() error = %v, wantErr %v", fs.fs.Name(), err, tt.wantErr)
|
t.Errorf("%v.Create() error = %v, wantErr %v", symFs.Name(), err, tt.wantErr)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
got, err := fs.fs.Stat(file.Name())
|
got, err := symFs.Stat(file.Name())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("%v.Stat() error = %v, wantErr %v", fs.fs.Name(), err, tt.wantErr)
|
t.Errorf("%v.Stat() error = %v, wantErr %v", symFs.Name(), err, tt.wantErr)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if !reflect.DeepEqual(got, want) {
|
if !reflect.DeepEqual(got, want) {
|
||||||
t.Errorf("%v.Create().Name() = %v, want %v", fs.fs.Name(), got, want)
|
t.Errorf("%v.Create().Name() = %v, want %v", symFs.Name(), got, want)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user