143 lines
3.3 KiB
Go
143 lines
3.3 KiB
Go
package encfs
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
// copied from afero/os.go
|
|
|
|
type EncryptionMasterKey struct {
|
|
key []byte
|
|
}
|
|
|
|
type EncFs struct {
|
|
key *EncryptionMasterKey
|
|
}
|
|
|
|
func NewEncFs(key *EncryptionMasterKey) afero.Fs {
|
|
return &EncFs{
|
|
key: key,
|
|
}
|
|
}
|
|
|
|
func (*EncFs) Name() string { return "EncFs" }
|
|
|
|
func (encFs *EncFs) Create(name string) (afero.File, error) {
|
|
if err := checkFileExt(name); err != nil {
|
|
return nil, err
|
|
}
|
|
f, e := os.Create(name)
|
|
if f == nil {
|
|
// while this looks strange, we need to return a bare nil (of type nil) not
|
|
// a nil value of type *os.File or nil won't be nil
|
|
return nil, e
|
|
}
|
|
return convertOsFileToEncFile(name, f, e, encFs, true)
|
|
}
|
|
|
|
func (*EncFs) Mkdir(name string, perm os.FileMode) error {
|
|
return os.Mkdir(name, perm)
|
|
}
|
|
|
|
func (*EncFs) MkdirAll(path string, perm os.FileMode) error {
|
|
return os.MkdirAll(path, perm)
|
|
}
|
|
|
|
func (encFs *EncFs) Open(name string) (afero.File, error) {
|
|
if err := checkFileExt(name); err != nil {
|
|
return nil, err
|
|
}
|
|
f, e := os.Open(name)
|
|
if f == nil {
|
|
// while this looks strange, we need to return a bare nil (of type nil) not
|
|
// a nil value of type *os.File or nil won't be nil
|
|
return nil, e
|
|
}
|
|
return convertOsFileToEncFile(name, f, e, encFs, false)
|
|
}
|
|
|
|
func (encFs *EncFs) OpenFile(name string, flag int, perm os.FileMode) (afero.File, error) {
|
|
if err := checkFileExt(name); err != nil {
|
|
return nil, err
|
|
}
|
|
f, e := os.OpenFile(name, flag, perm)
|
|
if f == nil {
|
|
// while this looks strange, we need to return a bare nil (of type nil) not
|
|
// a nil value of type *os.File or nil won't be nil
|
|
return nil, e
|
|
}
|
|
return convertOsFileToEncFile(name, f, e, encFs, false)
|
|
}
|
|
|
|
func (*EncFs) Remove(name string) error {
|
|
encFileMetaName := name + EncFileExt
|
|
_ = os.Remove(encFileMetaName)
|
|
return os.Remove(name)
|
|
}
|
|
|
|
func (encFs *EncFs) RemoveAll(path string) error {
|
|
fileInfo, err := os.Stat(path)
|
|
if err == nil && !fileInfo.IsDir() {
|
|
return encFs.Remove(path)
|
|
}
|
|
return os.RemoveAll(path)
|
|
}
|
|
|
|
func (*EncFs) Rename(oldname, newname string) error {
|
|
oldEncFileMetaName := oldname + EncFileExt
|
|
newEncFileMetaName := newname + EncFileExt
|
|
_ = os.Rename(oldEncFileMetaName, newEncFileMetaName)
|
|
return os.Rename(oldname, newname)
|
|
}
|
|
|
|
func (*EncFs) Stat(name string) (os.FileInfo, error) {
|
|
return os.Stat(name)
|
|
}
|
|
|
|
func (*EncFs) Chmod(name string, mode os.FileMode) error {
|
|
return os.Chmod(name, mode)
|
|
}
|
|
|
|
func (*EncFs) Chown(name string, uid, gid int) error {
|
|
return os.Chown(name, uid, gid)
|
|
}
|
|
|
|
func (*EncFs) Chtimes(name string, atime time.Time, mtime time.Time) error {
|
|
return os.Chtimes(name, atime, mtime)
|
|
}
|
|
|
|
func (*EncFs) LstatIfPossible(name string) (os.FileInfo, bool, error) {
|
|
fi, err := os.Lstat(name)
|
|
return fi, true, err
|
|
}
|
|
|
|
func (*EncFs) SymlinkIfPossible(oldname, newname string) error {
|
|
return os.Symlink(oldname, newname)
|
|
}
|
|
|
|
func (*EncFs) ReadlinkIfPossible(name string) (string, error) {
|
|
return os.Readlink(name)
|
|
}
|
|
|
|
func checkFileExt(name string) error {
|
|
if strings.HasSuffix(name, EncFileExt) {
|
|
return ErrFileForbiddenFileExt
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func convertOsFileToEncFile(name string, file *os.File, e error, encFs *EncFs, isCreate bool) (afero.File, error) {
|
|
if e != nil {
|
|
return nil, e
|
|
}
|
|
encFile, err := NewEncFile(name, file, encFs, isCreate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return encFile, nil
|
|
}
|