12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214 |
- // Copyright (C) 2019 Nicola Murino
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published
- // by the Free Software Foundation, version 3.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- // Package vfs provides local and remote filesystems support
- package vfs
- import (
- "errors"
- "fmt"
- "io"
- "net/url"
- "os"
- "path"
- "path/filepath"
- "runtime"
- "strconv"
- "strings"
- "sync"
- "time"
- "github.com/eikenb/pipeat"
- "github.com/pkg/sftp"
- "github.com/sftpgo/sdk"
- "github.com/drakkan/sftpgo/v2/internal/kms"
- "github.com/drakkan/sftpgo/v2/internal/logger"
- "github.com/drakkan/sftpgo/v2/internal/util"
- )
- const (
- dirMimeType = "inode/directory"
- s3fsName = "S3Fs"
- gcsfsName = "GCSFs"
- azBlobFsName = "AzureBlobFs"
- lastModifiedField = "sftpgo_last_modified"
- preResumeTimeout = 90 * time.Second
- // ListerBatchSize defines the default limit for DirLister implementations
- ListerBatchSize = 1000
- )
- // Additional checks for files
- const (
- CheckParentDir = 1
- CheckResume = 2
- )
- var (
- validAzAccessTier = []string{"", "Archive", "Hot", "Cool"}
- // ErrStorageSizeUnavailable is returned if the storage backend does not support getting the size
- ErrStorageSizeUnavailable = errors.New("unable to get available size for this storage backend")
- // ErrVfsUnsupported defines the error for an unsupported VFS operation
- ErrVfsUnsupported = errors.New("not supported")
- errInvalidDirListerLimit = errors.New("dir lister: invalid limit, must be > 0")
- tempPath string
- sftpFingerprints []string
- allowSelfConnections int
- renameMode int
- readMetadata int
- resumeMaxSize int64
- uploadMode int
- )
- // SetAllowSelfConnections sets the desired behaviour for self connections
- func SetAllowSelfConnections(value int) {
- allowSelfConnections = value
- }
- // SetTempPath sets the path for temporary files
- func SetTempPath(fsPath string) {
- tempPath = fsPath
- }
- // GetTempPath returns the path for temporary files
- func GetTempPath() string {
- return tempPath
- }
- // SetSFTPFingerprints sets the SFTP host key fingerprints
- func SetSFTPFingerprints(fp []string) {
- sftpFingerprints = fp
- }
- // SetRenameMode sets the rename mode
- func SetRenameMode(val int) {
- renameMode = val
- }
- // SetReadMetadataMode sets the read metadata mode
- func SetReadMetadataMode(val int) {
- readMetadata = val
- }
- // SetResumeMaxSize sets the max size allowed for resuming uploads for backends
- // with immutable objects
- func SetResumeMaxSize(val int64) {
- resumeMaxSize = val
- }
- // SetUploadMode sets the upload mode
- func SetUploadMode(val int) {
- uploadMode = val
- }
- // Fs defines the interface for filesystem backends
- type Fs interface {
- Name() string
- ConnectionID() string
- Stat(name string) (os.FileInfo, error)
- Lstat(name string) (os.FileInfo, error)
- Open(name string, offset int64) (File, PipeReader, func(), error)
- Create(name string, flag, checks int) (File, PipeWriter, func(), error)
- Rename(source, target string) (int, int64, error)
- Remove(name string, isDir bool) error
- Mkdir(name string) error
- Symlink(source, target string) error
- Chown(name string, uid int, gid int) error
- Chmod(name string, mode os.FileMode) error
- Chtimes(name string, atime, mtime time.Time, isUploading bool) error
- Truncate(name string, size int64) error
- ReadDir(dirname string) (DirLister, error)
- Readlink(name string) (string, error)
- IsUploadResumeSupported() bool
- IsConditionalUploadResumeSupported(size int64) bool
- IsAtomicUploadSupported() bool
- CheckRootPath(username string, uid int, gid int) bool
- ResolvePath(virtualPath string) (string, error)
- IsNotExist(err error) bool
- IsPermission(err error) bool
- IsNotSupported(err error) bool
- ScanRootDirContents() (int, int64, error)
- GetDirSize(dirname string) (int, int64, error)
- GetAtomicUploadPath(name string) string
- GetRelativePath(name string) string
- Walk(root string, walkFn filepath.WalkFunc) error
- Join(elem ...string) string
- HasVirtualFolders() bool
- GetMimeType(name string) (string, error)
- GetAvailableDiskSize(dirName string) (*sftp.StatVFS, error)
- Close() error
- }
- // FsRealPather is a Fs that implements the RealPath method.
- type FsRealPather interface {
- Fs
- RealPath(p string) (string, error)
- }
- // FsFileCopier is a Fs that implements the CopyFile method.
- type FsFileCopier interface {
- Fs
- CopyFile(source, target string, srcSize int64) (int, int64, error)
- }
- // File defines an interface representing a SFTPGo file
- type File interface {
- io.Reader
- io.Writer
- io.Closer
- io.ReaderAt
- io.WriterAt
- io.Seeker
- Stat() (os.FileInfo, error)
- Name() string
- Truncate(size int64) error
- }
- // PipeWriter defines an interface representing a SFTPGo pipe writer
- type PipeWriter interface {
- io.Writer
- io.WriterAt
- io.Closer
- Done(err error)
- GetWrittenBytes() int64
- }
- // PipeReader defines an interface representing a SFTPGo pipe reader
- type PipeReader interface {
- io.Reader
- io.ReaderAt
- io.Closer
- setMetadata(value map[string]string)
- setMetadataFromPointerVal(value map[string]*string)
- Metadata() map[string]string
- }
- // DirLister defines an interface for a directory lister
- type DirLister interface {
- Next(limit int) ([]os.FileInfo, error)
- Close() error
- }
- // Metadater defines an interface to implement to return metadata for a file
- type Metadater interface {
- Metadata() map[string]string
- }
- type baseDirLister struct {
- cache []os.FileInfo
- }
- func (l *baseDirLister) Next(limit int) ([]os.FileInfo, error) {
- if limit <= 0 {
- return nil, errInvalidDirListerLimit
- }
- if len(l.cache) >= limit {
- return l.returnFromCache(limit), nil
- }
- return l.returnFromCache(limit), io.EOF
- }
- func (l *baseDirLister) returnFromCache(limit int) []os.FileInfo {
- if len(l.cache) >= limit {
- result := l.cache[:limit]
- l.cache = l.cache[limit:]
- return result
- }
- result := l.cache
- l.cache = nil
- return result
- }
- func (l *baseDirLister) Close() error {
- l.cache = nil
- return nil
- }
- // QuotaCheckResult defines the result for a quota check
- type QuotaCheckResult struct {
- HasSpace bool
- AllowedSize int64
- AllowedFiles int
- UsedSize int64
- UsedFiles int
- QuotaSize int64
- QuotaFiles int
- }
- // GetRemainingSize returns the remaining allowed size
- func (q *QuotaCheckResult) GetRemainingSize() int64 {
- if q.QuotaSize > 0 {
- return q.QuotaSize - q.UsedSize
- }
- return 0
- }
- // GetRemainingFiles returns the remaining allowed files
- func (q *QuotaCheckResult) GetRemainingFiles() int {
- if q.QuotaFiles > 0 {
- return q.QuotaFiles - q.UsedFiles
- }
- return 0
- }
- // S3FsConfig defines the configuration for S3 based filesystem
- type S3FsConfig struct {
- sdk.BaseS3FsConfig
- AccessSecret *kms.Secret `json:"access_secret,omitempty"`
- }
- // HideConfidentialData hides confidential data
- func (c *S3FsConfig) HideConfidentialData() {
- if c.AccessSecret != nil {
- c.AccessSecret.Hide()
- }
- }
- func (c *S3FsConfig) isEqual(other S3FsConfig) bool {
- if c.Bucket != other.Bucket {
- return false
- }
- if c.KeyPrefix != other.KeyPrefix {
- return false
- }
- if c.Region != other.Region {
- return false
- }
- if c.AccessKey != other.AccessKey {
- return false
- }
- if c.RoleARN != other.RoleARN {
- return false
- }
- if c.Endpoint != other.Endpoint {
- return false
- }
- if c.StorageClass != other.StorageClass {
- return false
- }
- if c.ACL != other.ACL {
- return false
- }
- if !c.areMultipartFieldsEqual(other) {
- return false
- }
- if c.ForcePathStyle != other.ForcePathStyle {
- return false
- }
- if c.SkipTLSVerify != other.SkipTLSVerify {
- return false
- }
- return c.isSecretEqual(other)
- }
- func (c *S3FsConfig) areMultipartFieldsEqual(other S3FsConfig) bool {
- if c.UploadPartSize != other.UploadPartSize {
- return false
- }
- if c.UploadConcurrency != other.UploadConcurrency {
- return false
- }
- if c.DownloadConcurrency != other.DownloadConcurrency {
- return false
- }
- if c.DownloadPartSize != other.DownloadPartSize {
- return false
- }
- if c.DownloadPartMaxTime != other.DownloadPartMaxTime {
- return false
- }
- if c.UploadPartMaxTime != other.UploadPartMaxTime {
- return false
- }
- return true
- }
- func (c *S3FsConfig) isSecretEqual(other S3FsConfig) bool {
- if c.AccessSecret == nil {
- c.AccessSecret = kms.NewEmptySecret()
- }
- if other.AccessSecret == nil {
- other.AccessSecret = kms.NewEmptySecret()
- }
- return c.AccessSecret.IsEqual(other.AccessSecret)
- }
- func (c *S3FsConfig) checkCredentials() error {
- if c.AccessKey == "" && !c.AccessSecret.IsEmpty() {
- return util.NewI18nError(
- errors.New("access_key cannot be empty with access_secret not empty"),
- util.I18nErrorAccessKeyRequired,
- )
- }
- if c.AccessSecret.IsEmpty() && c.AccessKey != "" {
- return util.NewI18nError(
- errors.New("access_secret cannot be empty with access_key not empty"),
- util.I18nErrorAccessSecretRequired,
- )
- }
- if c.AccessSecret.IsEncrypted() && !c.AccessSecret.IsValid() {
- return errors.New("invalid encrypted access_secret")
- }
- if !c.AccessSecret.IsEmpty() && !c.AccessSecret.IsValidInput() {
- return errors.New("invalid access_secret")
- }
- return nil
- }
- // ValidateAndEncryptCredentials validates the configuration and encrypts access secret if it is in plain text
- func (c *S3FsConfig) ValidateAndEncryptCredentials(additionalData string) error {
- if err := c.validate(); err != nil {
- var errI18n *util.I18nError
- errValidation := util.NewValidationError(fmt.Sprintf("could not validate s3config: %v", err))
- if errors.As(err, &errI18n) {
- return util.NewI18nError(errValidation, errI18n.Message)
- }
- return util.NewI18nError(errValidation, util.I18nErrorFsValidation)
- }
- if c.AccessSecret.IsPlain() {
- c.AccessSecret.SetAdditionalData(additionalData)
- err := c.AccessSecret.Encrypt()
- if err != nil {
- return util.NewI18nError(
- util.NewValidationError(fmt.Sprintf("could not encrypt s3 access secret: %v", err)),
- util.I18nErrorFsValidation,
- )
- }
- }
- return nil
- }
- func (c *S3FsConfig) checkPartSizeAndConcurrency() error {
- if c.UploadPartSize != 0 && (c.UploadPartSize < 5 || c.UploadPartSize > 5000) {
- return util.NewI18nError(
- errors.New("upload_part_size cannot be != 0, lower than 5 (MB) or greater than 5000 (MB)"),
- util.I18nErrorULPartSizeInvalid,
- )
- }
- if c.UploadConcurrency < 0 || c.UploadConcurrency > 64 {
- return util.NewI18nError(
- fmt.Errorf("invalid upload concurrency: %v", c.UploadConcurrency),
- util.I18nErrorULConcurrencyInvalid,
- )
- }
- if c.DownloadPartSize != 0 && (c.DownloadPartSize < 5 || c.DownloadPartSize > 5000) {
- return util.NewI18nError(
- errors.New("download_part_size cannot be != 0, lower than 5 (MB) or greater than 5000 (MB)"),
- util.I18nErrorDLPartSizeInvalid,
- )
- }
- if c.DownloadConcurrency < 0 || c.DownloadConcurrency > 64 {
- return util.NewI18nError(
- fmt.Errorf("invalid download concurrency: %v", c.DownloadConcurrency),
- util.I18nErrorDLConcurrencyInvalid,
- )
- }
- return nil
- }
- func (c *S3FsConfig) isSameResource(other S3FsConfig) bool {
- if c.Bucket != other.Bucket {
- return false
- }
- if c.Endpoint != other.Endpoint {
- return false
- }
- return c.Region == other.Region
- }
- // validate returns an error if the configuration is not valid
- func (c *S3FsConfig) validate() error {
- if c.AccessSecret == nil {
- c.AccessSecret = kms.NewEmptySecret()
- }
- if c.Bucket == "" {
- return util.NewI18nError(errors.New("bucket cannot be empty"), util.I18nErrorBucketRequired)
- }
- // the region may be embedded within the endpoint for some S3 compatible
- // object storage, for example B2
- if c.Endpoint == "" && c.Region == "" {
- return util.NewI18nError(errors.New("region cannot be empty"), util.I18nErrorRegionRequired)
- }
- if err := c.checkCredentials(); err != nil {
- return err
- }
- if c.KeyPrefix != "" {
- if strings.HasPrefix(c.KeyPrefix, "/") {
- return util.NewI18nError(errors.New("key_prefix cannot start with /"), util.I18nErrorKeyPrefixInvalid)
- }
- c.KeyPrefix = path.Clean(c.KeyPrefix)
- if !strings.HasSuffix(c.KeyPrefix, "/") {
- c.KeyPrefix += "/"
- }
- }
- c.StorageClass = strings.TrimSpace(c.StorageClass)
- c.ACL = strings.TrimSpace(c.ACL)
- return c.checkPartSizeAndConcurrency()
- }
- // GCSFsConfig defines the configuration for Google Cloud Storage based filesystem
- type GCSFsConfig struct {
- sdk.BaseGCSFsConfig
- Credentials *kms.Secret `json:"credentials,omitempty"`
- }
- // HideConfidentialData hides confidential data
- func (c *GCSFsConfig) HideConfidentialData() {
- if c.Credentials != nil {
- c.Credentials.Hide()
- }
- }
- // ValidateAndEncryptCredentials validates the configuration and encrypts credentials if they are in plain text
- func (c *GCSFsConfig) ValidateAndEncryptCredentials(additionalData string) error {
- if err := c.validate(); err != nil {
- var errI18n *util.I18nError
- errValidation := util.NewValidationError(fmt.Sprintf("could not validate GCS config: %v", err))
- if errors.As(err, &errI18n) {
- return util.NewI18nError(errValidation, errI18n.Message)
- }
- return util.NewI18nError(errValidation, util.I18nErrorFsValidation)
- }
- if c.Credentials.IsPlain() {
- c.Credentials.SetAdditionalData(additionalData)
- err := c.Credentials.Encrypt()
- if err != nil {
- return util.NewI18nError(
- util.NewValidationError(fmt.Sprintf("could not encrypt GCS credentials: %v", err)),
- util.I18nErrorFsValidation,
- )
- }
- }
- return nil
- }
- func (c *GCSFsConfig) isEqual(other GCSFsConfig) bool {
- if c.Bucket != other.Bucket {
- return false
- }
- if c.KeyPrefix != other.KeyPrefix {
- return false
- }
- if c.AutomaticCredentials != other.AutomaticCredentials {
- return false
- }
- if c.StorageClass != other.StorageClass {
- return false
- }
- if c.ACL != other.ACL {
- return false
- }
- if c.UploadPartSize != other.UploadPartSize {
- return false
- }
- if c.UploadPartMaxTime != other.UploadPartMaxTime {
- return false
- }
- if c.Credentials == nil {
- c.Credentials = kms.NewEmptySecret()
- }
- if other.Credentials == nil {
- other.Credentials = kms.NewEmptySecret()
- }
- return c.Credentials.IsEqual(other.Credentials)
- }
- func (c *GCSFsConfig) isSameResource(other GCSFsConfig) bool {
- return c.Bucket == other.Bucket
- }
- // validate returns an error if the configuration is not valid
- func (c *GCSFsConfig) validate() error {
- if c.Credentials == nil || c.AutomaticCredentials == 1 {
- c.Credentials = kms.NewEmptySecret()
- }
- if c.Bucket == "" {
- return util.NewI18nError(errors.New("bucket cannot be empty"), util.I18nErrorBucketRequired)
- }
- if c.KeyPrefix != "" {
- if strings.HasPrefix(c.KeyPrefix, "/") {
- return util.NewI18nError(errors.New("key_prefix cannot start with /"), util.I18nErrorKeyPrefixInvalid)
- }
- c.KeyPrefix = path.Clean(c.KeyPrefix)
- if !strings.HasSuffix(c.KeyPrefix, "/") {
- c.KeyPrefix += "/"
- }
- }
- if c.Credentials.IsEncrypted() && !c.Credentials.IsValid() {
- return errors.New("invalid encrypted credentials")
- }
- if c.AutomaticCredentials == 0 && !c.Credentials.IsValidInput() {
- return util.NewI18nError(errors.New("invalid credentials"), util.I18nErrorFsCredentialsRequired)
- }
- c.StorageClass = strings.TrimSpace(c.StorageClass)
- c.ACL = strings.TrimSpace(c.ACL)
- if c.UploadPartSize < 0 {
- c.UploadPartSize = 0
- }
- if c.UploadPartMaxTime < 0 {
- c.UploadPartMaxTime = 0
- }
- return nil
- }
- // AzBlobFsConfig defines the configuration for Azure Blob Storage based filesystem
- type AzBlobFsConfig struct {
- sdk.BaseAzBlobFsConfig
- // Storage Account Key leave blank to use SAS URL.
- // The access key is stored encrypted based on the kms configuration
- AccountKey *kms.Secret `json:"account_key,omitempty"`
- // Shared access signature URL, leave blank if using account/key
- SASURL *kms.Secret `json:"sas_url,omitempty"`
- }
- // HideConfidentialData hides confidential data
- func (c *AzBlobFsConfig) HideConfidentialData() {
- if c.AccountKey != nil {
- c.AccountKey.Hide()
- }
- if c.SASURL != nil {
- c.SASURL.Hide()
- }
- }
- func (c *AzBlobFsConfig) isEqual(other AzBlobFsConfig) bool {
- if c.Container != other.Container {
- return false
- }
- if c.AccountName != other.AccountName {
- return false
- }
- if c.Endpoint != other.Endpoint {
- return false
- }
- if c.SASURL.IsEmpty() {
- c.SASURL = kms.NewEmptySecret()
- }
- if other.SASURL.IsEmpty() {
- other.SASURL = kms.NewEmptySecret()
- }
- if !c.SASURL.IsEqual(other.SASURL) {
- return false
- }
- if c.KeyPrefix != other.KeyPrefix {
- return false
- }
- if c.UploadPartSize != other.UploadPartSize {
- return false
- }
- if c.UploadConcurrency != other.UploadConcurrency {
- return false
- }
- if c.DownloadPartSize != other.DownloadPartSize {
- return false
- }
- if c.DownloadConcurrency != other.DownloadConcurrency {
- return false
- }
- if c.UseEmulator != other.UseEmulator {
- return false
- }
- if c.AccessTier != other.AccessTier {
- return false
- }
- return c.isSecretEqual(other)
- }
- func (c *AzBlobFsConfig) isSecretEqual(other AzBlobFsConfig) bool {
- if c.AccountKey == nil {
- c.AccountKey = kms.NewEmptySecret()
- }
- if other.AccountKey == nil {
- other.AccountKey = kms.NewEmptySecret()
- }
- return c.AccountKey.IsEqual(other.AccountKey)
- }
- // ValidateAndEncryptCredentials validates the configuration and encrypts access secret if it is in plain text
- func (c *AzBlobFsConfig) ValidateAndEncryptCredentials(additionalData string) error {
- if err := c.validate(); err != nil {
- var errI18n *util.I18nError
- errValidation := util.NewValidationError(fmt.Sprintf("could not validate Azure Blob config: %v", err))
- if errors.As(err, &errI18n) {
- return util.NewI18nError(errValidation, errI18n.Message)
- }
- return util.NewI18nError(errValidation, util.I18nErrorFsValidation)
- }
- if c.AccountKey.IsPlain() {
- c.AccountKey.SetAdditionalData(additionalData)
- if err := c.AccountKey.Encrypt(); err != nil {
- return util.NewI18nError(
- util.NewValidationError(fmt.Sprintf("could not encrypt Azure blob account key: %v", err)),
- util.I18nErrorFsValidation,
- )
- }
- }
- if c.SASURL.IsPlain() {
- c.SASURL.SetAdditionalData(additionalData)
- if err := c.SASURL.Encrypt(); err != nil {
- return util.NewI18nError(
- util.NewValidationError(fmt.Sprintf("could not encrypt Azure blob SAS URL: %v", err)),
- util.I18nErrorFsValidation,
- )
- }
- }
- return nil
- }
- func (c *AzBlobFsConfig) checkCredentials() error {
- if c.SASURL.IsPlain() {
- _, err := url.Parse(c.SASURL.GetPayload())
- if err != nil {
- return util.NewI18nError(err, util.I18nErrorSASURLInvalid)
- }
- return nil
- }
- if c.SASURL.IsEncrypted() && !c.SASURL.IsValid() {
- return errors.New("invalid encrypted sas_url")
- }
- if !c.SASURL.IsEmpty() {
- return nil
- }
- if c.AccountName == "" || !c.AccountKey.IsValidInput() {
- return util.NewI18nError(errors.New("credentials cannot be empty or invalid"), util.I18nErrorAccountNameRequired)
- }
- if c.AccountKey.IsEncrypted() && !c.AccountKey.IsValid() {
- return errors.New("invalid encrypted account_key")
- }
- return nil
- }
- func (c *AzBlobFsConfig) checkPartSizeAndConcurrency() error {
- if c.UploadPartSize < 0 || c.UploadPartSize > 100 {
- return util.NewI18nError(
- fmt.Errorf("invalid upload part size: %v", c.UploadPartSize),
- util.I18nErrorULPartSizeInvalid,
- )
- }
- if c.UploadConcurrency < 0 || c.UploadConcurrency > 64 {
- return util.NewI18nError(
- fmt.Errorf("invalid upload concurrency: %v", c.UploadConcurrency),
- util.I18nErrorULConcurrencyInvalid,
- )
- }
- if c.DownloadPartSize < 0 || c.DownloadPartSize > 100 {
- return util.NewI18nError(
- fmt.Errorf("invalid download part size: %v", c.DownloadPartSize),
- util.I18nErrorDLPartSizeInvalid,
- )
- }
- if c.DownloadConcurrency < 0 || c.DownloadConcurrency > 64 {
- return util.NewI18nError(
- fmt.Errorf("invalid upload concurrency: %v", c.DownloadConcurrency),
- util.I18nErrorDLConcurrencyInvalid,
- )
- }
- return nil
- }
- func (c *AzBlobFsConfig) tryDecrypt() error {
- if err := c.AccountKey.TryDecrypt(); err != nil {
- return fmt.Errorf("unable to decrypt account key: %w", err)
- }
- if err := c.SASURL.TryDecrypt(); err != nil {
- return fmt.Errorf("unable to decrypt SAS URL: %w", err)
- }
- return nil
- }
- func (c *AzBlobFsConfig) isSameResource(other AzBlobFsConfig) bool {
- if c.AccountName != other.AccountName {
- return false
- }
- if c.Endpoint != other.Endpoint {
- return false
- }
- return c.SASURL.GetPayload() == other.SASURL.GetPayload()
- }
- // validate returns an error if the configuration is not valid
- func (c *AzBlobFsConfig) validate() error {
- if c.AccountKey == nil {
- c.AccountKey = kms.NewEmptySecret()
- }
- if c.SASURL == nil {
- c.SASURL = kms.NewEmptySecret()
- }
- // container could be embedded within SAS URL we check this at runtime
- if c.SASURL.IsEmpty() && c.Container == "" {
- return util.NewI18nError(errors.New("container cannot be empty"), util.I18nErrorContainerRequired)
- }
- if err := c.checkCredentials(); err != nil {
- return err
- }
- if c.KeyPrefix != "" {
- if strings.HasPrefix(c.KeyPrefix, "/") {
- return util.NewI18nError(errors.New("key_prefix cannot start with /"), util.I18nErrorKeyPrefixInvalid)
- }
- c.KeyPrefix = path.Clean(c.KeyPrefix)
- if !strings.HasSuffix(c.KeyPrefix, "/") {
- c.KeyPrefix += "/"
- }
- }
- if err := c.checkPartSizeAndConcurrency(); err != nil {
- return err
- }
- if !util.Contains(validAzAccessTier, c.AccessTier) {
- return fmt.Errorf("invalid access tier %q, valid values: \"''%v\"", c.AccessTier, strings.Join(validAzAccessTier, ", "))
- }
- return nil
- }
- // CryptFsConfig defines the configuration to store local files as encrypted
- type CryptFsConfig struct {
- sdk.OSFsConfig
- Passphrase *kms.Secret `json:"passphrase,omitempty"`
- }
- // HideConfidentialData hides confidential data
- func (c *CryptFsConfig) HideConfidentialData() {
- if c.Passphrase != nil {
- c.Passphrase.Hide()
- }
- }
- func (c *CryptFsConfig) isEqual(other CryptFsConfig) bool {
- if c.Passphrase == nil {
- c.Passphrase = kms.NewEmptySecret()
- }
- if other.Passphrase == nil {
- other.Passphrase = kms.NewEmptySecret()
- }
- return c.Passphrase.IsEqual(other.Passphrase)
- }
- // ValidateAndEncryptCredentials validates the configuration and encrypts the passphrase if it is in plain text
- func (c *CryptFsConfig) ValidateAndEncryptCredentials(additionalData string) error {
- if err := c.validate(); err != nil {
- var errI18n *util.I18nError
- errValidation := util.NewValidationError(fmt.Sprintf("could not validate crypt fs config: %v", err))
- if errors.As(err, &errI18n) {
- return util.NewI18nError(errValidation, errI18n.Message)
- }
- return util.NewI18nError(errValidation, util.I18nErrorFsValidation)
- }
- if c.Passphrase.IsPlain() {
- c.Passphrase.SetAdditionalData(additionalData)
- if err := c.Passphrase.Encrypt(); err != nil {
- return util.NewI18nError(
- util.NewValidationError(fmt.Sprintf("could not encrypt Crypt fs passphrase: %v", err)),
- util.I18nErrorFsValidation,
- )
- }
- }
- return nil
- }
- func (c *CryptFsConfig) isSameResource(other CryptFsConfig) bool {
- return c.Passphrase.GetPayload() == other.Passphrase.GetPayload()
- }
- // validate returns an error if the configuration is not valid
- func (c *CryptFsConfig) validate() error {
- if c.Passphrase == nil || c.Passphrase.IsEmpty() {
- return util.NewI18nError(errors.New("invalid passphrase"), util.I18nErrorPassphraseRequired)
- }
- if !c.Passphrase.IsValidInput() {
- return util.NewI18nError(errors.New("passphrase cannot be empty or invalid"), util.I18nErrorPassphraseRequired)
- }
- if c.Passphrase.IsEncrypted() && !c.Passphrase.IsValid() {
- return errors.New("invalid encrypted passphrase")
- }
- return nil
- }
- // pipeWriter defines a wrapper for pipeat.PipeWriterAt.
- type pipeWriter struct {
- *pipeat.PipeWriterAt
- err error
- done chan bool
- }
- // NewPipeWriter initializes a new PipeWriter
- func NewPipeWriter(w *pipeat.PipeWriterAt) PipeWriter {
- return &pipeWriter{
- PipeWriterAt: w,
- err: nil,
- done: make(chan bool),
- }
- }
- // Close waits for the upload to end, closes the pipeat.PipeWriterAt and returns an error if any.
- func (p *pipeWriter) Close() error {
- p.PipeWriterAt.Close() //nolint:errcheck // the returned error is always null
- <-p.done
- return p.err
- }
- // Done unlocks other goroutines waiting on Close().
- // It must be called when the upload ends
- func (p *pipeWriter) Done(err error) {
- p.err = err
- p.done <- true
- }
- func newPipeWriterAtOffset(w *pipeat.PipeWriterAt, offset int64) PipeWriter {
- return &pipeWriterAtOffset{
- pipeWriter: &pipeWriter{
- PipeWriterAt: w,
- err: nil,
- done: make(chan bool),
- },
- offset: offset,
- writeOffset: offset,
- }
- }
- type pipeWriterAtOffset struct {
- *pipeWriter
- offset int64
- writeOffset int64
- }
- func (p *pipeWriterAtOffset) WriteAt(buf []byte, off int64) (int, error) {
- if off < p.offset {
- return 0, fmt.Errorf("invalid offset %d, minimum accepted %d", off, p.offset)
- }
- return p.pipeWriter.WriteAt(buf, off-p.offset)
- }
- func (p *pipeWriterAtOffset) Write(buf []byte) (int, error) {
- n, err := p.WriteAt(buf, p.writeOffset)
- p.writeOffset += int64(n)
- return n, err
- }
- // NewPipeReader initializes a new PipeReader
- func NewPipeReader(r *pipeat.PipeReaderAt) PipeReader {
- return &pipeReader{
- PipeReaderAt: r,
- }
- }
- // pipeReader defines a wrapper for pipeat.PipeReaderAt.
- type pipeReader struct {
- *pipeat.PipeReaderAt
- mu sync.RWMutex
- metadata map[string]string
- }
- func (p *pipeReader) setMetadata(value map[string]string) {
- p.mu.Lock()
- defer p.mu.Unlock()
- p.metadata = value
- }
- func (p *pipeReader) setMetadataFromPointerVal(value map[string]*string) {
- p.mu.Lock()
- defer p.mu.Unlock()
- if len(value) == 0 {
- p.metadata = nil
- return
- }
- p.metadata = map[string]string{}
- for k, v := range value {
- val := util.GetStringFromPointer(v)
- if val != "" {
- p.metadata[k] = val
- }
- }
- }
- // Metadata implements the Metadater interface
- func (p *pipeReader) Metadata() map[string]string {
- p.mu.RLock()
- defer p.mu.RUnlock()
- if len(p.metadata) == 0 {
- return nil
- }
- result := make(map[string]string)
- for k, v := range p.metadata {
- result[k] = v
- }
- return result
- }
- func isEqualityCheckModeValid(mode int) bool {
- return mode >= 0 || mode <= 1
- }
- // isDirectory checks if a path exists and is a directory
- func isDirectory(fs Fs, path string) (bool, error) {
- fileInfo, err := fs.Stat(path)
- if err != nil {
- return false, err
- }
- return fileInfo.IsDir(), err
- }
- // IsLocalOsFs returns true if fs is a local filesystem implementation
- func IsLocalOsFs(fs Fs) bool {
- return fs.Name() == osFsName
- }
- // IsCryptOsFs returns true if fs is an encrypted local filesystem implementation
- func IsCryptOsFs(fs Fs) bool {
- return fs.Name() == cryptFsName
- }
- // IsSFTPFs returns true if fs is an SFTP filesystem
- func IsSFTPFs(fs Fs) bool {
- return strings.HasPrefix(fs.Name(), sftpFsName)
- }
- // IsHTTPFs returns true if fs is an HTTP filesystem
- func IsHTTPFs(fs Fs) bool {
- return strings.HasPrefix(fs.Name(), httpFsName)
- }
- // IsBufferedLocalOrSFTPFs returns true if this is a buffered SFTP or local filesystem
- func IsBufferedLocalOrSFTPFs(fs Fs) bool {
- if osFs, ok := fs.(*OsFs); ok {
- return osFs.writeBufferSize > 0
- }
- if !IsSFTPFs(fs) {
- return false
- }
- return !fs.IsUploadResumeSupported()
- }
- // FsOpenReturnsFile returns true if fs.Open returns a *os.File handle
- func FsOpenReturnsFile(fs Fs) bool {
- if osFs, ok := fs.(*OsFs); ok {
- return osFs.readBufferSize == 0
- }
- if sftpFs, ok := fs.(*SFTPFs); ok {
- return sftpFs.config.BufferSize == 0
- }
- return false
- }
- // IsLocalOrSFTPFs returns true if fs is local or SFTP
- func IsLocalOrSFTPFs(fs Fs) bool {
- return IsLocalOsFs(fs) || IsSFTPFs(fs)
- }
- // HasTruncateSupport returns true if the fs supports truncate files
- func HasTruncateSupport(fs Fs) bool {
- return IsLocalOsFs(fs) || IsSFTPFs(fs) || IsHTTPFs(fs)
- }
- // IsRenameAtomic returns true if renaming a directory is supposed to be atomic
- func IsRenameAtomic(fs Fs) bool {
- if strings.HasPrefix(fs.Name(), s3fsName) {
- return false
- }
- if strings.HasPrefix(fs.Name(), gcsfsName) {
- return false
- }
- if strings.HasPrefix(fs.Name(), azBlobFsName) {
- return false
- }
- return true
- }
- // HasImplicitAtomicUploads returns true if the fs don't persists partial files on error
- func HasImplicitAtomicUploads(fs Fs) bool {
- if strings.HasPrefix(fs.Name(), s3fsName) {
- return uploadMode&4 == 0
- }
- if strings.HasPrefix(fs.Name(), gcsfsName) {
- return uploadMode&8 == 0
- }
- if strings.HasPrefix(fs.Name(), azBlobFsName) {
- return uploadMode&16 == 0
- }
- return false
- }
- // HasOpenRWSupport returns true if the fs can open a file
- // for reading and writing at the same time
- func HasOpenRWSupport(fs Fs) bool {
- if IsLocalOsFs(fs) {
- return true
- }
- if IsSFTPFs(fs) && fs.IsUploadResumeSupported() {
- return true
- }
- return false
- }
- // IsLocalOrCryptoFs returns true if fs is local or local encrypted
- func IsLocalOrCryptoFs(fs Fs) bool {
- return IsLocalOsFs(fs) || IsCryptOsFs(fs)
- }
- // SetPathPermissions calls fs.Chown.
- // It does nothing for local filesystem on windows
- func SetPathPermissions(fs Fs, path string, uid int, gid int) {
- if uid == -1 && gid == -1 {
- return
- }
- if IsLocalOsFs(fs) {
- if runtime.GOOS == "windows" {
- return
- }
- }
- if err := fs.Chown(path, uid, gid); err != nil {
- fsLog(fs, logger.LevelWarn, "error chowning path %v: %v", path, err)
- }
- }
- // IsUploadResumeSupported returns true if resuming uploads is supported
- func IsUploadResumeSupported(fs Fs, size int64) bool {
- if fs.IsUploadResumeSupported() {
- return true
- }
- return fs.IsConditionalUploadResumeSupported(size)
- }
- func getLastModified(metadata map[string]string) int64 {
- if val, ok := metadata[lastModifiedField]; ok {
- lastModified, err := strconv.ParseInt(val, 10, 64)
- if err == nil {
- return lastModified
- }
- }
- return 0
- }
- func getAzureLastModified(metadata map[string]*string) int64 {
- for k, v := range metadata {
- if strings.ToLower(k) == lastModifiedField {
- if val := util.GetStringFromPointer(v); val != "" {
- lastModified, err := strconv.ParseInt(val, 10, 64)
- if err == nil {
- return lastModified
- }
- }
- return 0
- }
- }
- return 0
- }
- func validateOSFsConfig(config *sdk.OSFsConfig) error {
- if config.ReadBufferSize < 0 || config.ReadBufferSize > 10 {
- return fmt.Errorf("invalid read buffer size must be between 0 and 10 MB")
- }
- if config.WriteBufferSize < 0 || config.WriteBufferSize > 10 {
- return fmt.Errorf("invalid write buffer size must be between 0 and 10 MB")
- }
- return nil
- }
- func doCopy(dst io.Writer, src io.Reader, buf []byte) (written int64, err error) {
- if buf == nil {
- buf = make([]byte, 32768)
- }
- for {
- nr, er := src.Read(buf)
- if nr > 0 {
- nw, ew := dst.Write(buf[0:nr])
- if nw < 0 || nr < nw {
- nw = 0
- if ew == nil {
- ew = errors.New("invalid write")
- }
- }
- written += int64(nw)
- if ew != nil {
- err = ew
- break
- }
- if nr != nw {
- err = io.ErrShortWrite
- break
- }
- }
- if er != nil {
- if er != io.EOF {
- err = er
- }
- break
- }
- }
- return written, err
- }
- func getMountPath(mountPath string) string {
- if mountPath == "/" {
- return ""
- }
- return mountPath
- }
- func getLocalTempDir() string {
- if tempPath != "" {
- return tempPath
- }
- return filepath.Clean(os.TempDir())
- }
- func doRecursiveRename(fs Fs, source, target string,
- renameFn func(string, string, os.FileInfo, int) (int, int64, error),
- recursion int,
- ) (int, int64, error) {
- var numFiles int
- var filesSize int64
- if recursion > util.MaxRecursion {
- return numFiles, filesSize, util.ErrRecursionTooDeep
- }
- recursion++
- lister, err := fs.ReadDir(source)
- if err != nil {
- return numFiles, filesSize, err
- }
- defer lister.Close()
- for {
- entries, err := lister.Next(ListerBatchSize)
- finished := errors.Is(err, io.EOF)
- if err != nil && !finished {
- return numFiles, filesSize, err
- }
- for _, info := range entries {
- sourceEntry := fs.Join(source, info.Name())
- targetEntry := fs.Join(target, info.Name())
- files, size, err := renameFn(sourceEntry, targetEntry, info, recursion)
- if err != nil {
- if fs.IsNotExist(err) {
- fsLog(fs, logger.LevelInfo, "skipping rename for %q: %v", sourceEntry, err)
- continue
- }
- return numFiles, filesSize, err
- }
- numFiles += files
- filesSize += size
- }
- if finished {
- return numFiles, filesSize, nil
- }
- }
- }
- func fsLog(fs Fs, level logger.LogLevel, format string, v ...any) {
- logger.Log(level, fs.Name(), fs.ConnectionID(), format, v...)
- }
|