connection_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606
  1. // Copyright (C) 2019-2023 Nicola Murino
  2. //
  3. // This program is free software: you can redistribute it and/or modify
  4. // it under the terms of the GNU Affero General Public License as published
  5. // by the Free Software Foundation, version 3.
  6. //
  7. // This program is distributed in the hope that it will be useful,
  8. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. // GNU Affero General Public License for more details.
  11. //
  12. // You should have received a copy of the GNU Affero General Public License
  13. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  14. package common
  15. import (
  16. "errors"
  17. "os"
  18. "path"
  19. "path/filepath"
  20. "runtime"
  21. "testing"
  22. "time"
  23. "github.com/pkg/sftp"
  24. "github.com/rs/xid"
  25. "github.com/sftpgo/sdk"
  26. "github.com/stretchr/testify/assert"
  27. "github.com/drakkan/sftpgo/v2/internal/dataprovider"
  28. "github.com/drakkan/sftpgo/v2/internal/kms"
  29. "github.com/drakkan/sftpgo/v2/internal/util"
  30. "github.com/drakkan/sftpgo/v2/internal/vfs"
  31. )
  32. var (
  33. errWalkDir = errors.New("err walk dir")
  34. )
  35. // MockOsFs mockable OsFs
  36. type MockOsFs struct {
  37. vfs.Fs
  38. hasVirtualFolders bool
  39. name string
  40. err error
  41. }
  42. // Name returns the name for the Fs implementation
  43. func (fs *MockOsFs) Name() string {
  44. if fs.name != "" {
  45. return fs.name
  46. }
  47. return "mockOsFs"
  48. }
  49. // HasVirtualFolders returns true if folders are emulated
  50. func (fs *MockOsFs) HasVirtualFolders() bool {
  51. return fs.hasVirtualFolders
  52. }
  53. func (fs *MockOsFs) IsUploadResumeSupported() bool {
  54. return !fs.hasVirtualFolders
  55. }
  56. func (fs *MockOsFs) Chtimes(name string, atime, mtime time.Time, isUploading bool) error {
  57. return vfs.ErrVfsUnsupported
  58. }
  59. func (fs *MockOsFs) Lstat(name string) (os.FileInfo, error) {
  60. if fs.err != nil {
  61. return nil, fs.err
  62. }
  63. return fs.Fs.Lstat(name)
  64. }
  65. // Walk returns a duplicate path for testing
  66. func (fs *MockOsFs) Walk(root string, walkFn filepath.WalkFunc) error {
  67. if fs.err == errWalkDir {
  68. walkFn("fsdpath", vfs.NewFileInfo("dpath", true, 0, time.Now(), false), nil) //nolint:errcheck
  69. return walkFn("fsdpath", vfs.NewFileInfo("dpath", true, 0, time.Now(), false), nil) //nolint:errcheck
  70. }
  71. walkFn("fsfpath", vfs.NewFileInfo("fpath", false, 0, time.Now(), false), nil) //nolint:errcheck
  72. return fs.err
  73. }
  74. func newMockOsFs(hasVirtualFolders bool, connectionID, rootDir, name string, err error) vfs.Fs {
  75. return &MockOsFs{
  76. Fs: vfs.NewOsFs(connectionID, rootDir, ""),
  77. name: name,
  78. hasVirtualFolders: hasVirtualFolders,
  79. err: err,
  80. }
  81. }
  82. func TestRemoveErrors(t *testing.T) {
  83. mappedPath := filepath.Join(os.TempDir(), "map")
  84. homePath := filepath.Join(os.TempDir(), "home")
  85. user := dataprovider.User{
  86. BaseUser: sdk.BaseUser{
  87. Username: "remove_errors_user",
  88. HomeDir: homePath,
  89. },
  90. VirtualFolders: []vfs.VirtualFolder{
  91. {
  92. BaseVirtualFolder: vfs.BaseVirtualFolder{
  93. Name: filepath.Base(mappedPath),
  94. MappedPath: mappedPath,
  95. },
  96. VirtualPath: "/virtualpath",
  97. },
  98. },
  99. }
  100. user.Permissions = make(map[string][]string)
  101. user.Permissions["/"] = []string{dataprovider.PermAny}
  102. fs := vfs.NewOsFs("", os.TempDir(), "")
  103. conn := NewBaseConnection("", ProtocolFTP, "", "", user)
  104. err := conn.IsRemoveDirAllowed(fs, mappedPath, "/virtualpath1")
  105. if assert.Error(t, err) {
  106. assert.Contains(t, err.Error(), "permission denied")
  107. }
  108. err = conn.RemoveFile(fs, filepath.Join(homePath, "missing_file"), "/missing_file",
  109. vfs.NewFileInfo("info", false, 100, time.Now(), false))
  110. assert.Error(t, err)
  111. }
  112. func TestSetStatMode(t *testing.T) {
  113. oldSetStatMode := Config.SetstatMode
  114. Config.SetstatMode = 1
  115. fakePath := "fake path"
  116. user := dataprovider.User{
  117. BaseUser: sdk.BaseUser{
  118. HomeDir: os.TempDir(),
  119. },
  120. }
  121. user.Permissions = make(map[string][]string)
  122. user.Permissions["/"] = []string{dataprovider.PermAny}
  123. fs := newMockOsFs(true, "", user.GetHomeDir(), "", nil)
  124. conn := NewBaseConnection("", ProtocolWebDAV, "", "", user)
  125. err := conn.handleChmod(fs, fakePath, fakePath, nil)
  126. assert.NoError(t, err)
  127. err = conn.handleChown(fs, fakePath, fakePath, nil)
  128. assert.NoError(t, err)
  129. err = conn.handleChtimes(fs, fakePath, fakePath, nil)
  130. assert.NoError(t, err)
  131. Config.SetstatMode = 2
  132. err = conn.handleChmod(fs, fakePath, fakePath, nil)
  133. assert.NoError(t, err)
  134. err = conn.handleChtimes(fs, fakePath, fakePath, &StatAttributes{
  135. Atime: time.Now(),
  136. Mtime: time.Now(),
  137. })
  138. assert.NoError(t, err)
  139. Config.SetstatMode = oldSetStatMode
  140. }
  141. func TestRecursiveRenameWalkError(t *testing.T) {
  142. fs := vfs.NewOsFs("", filepath.Clean(os.TempDir()), "")
  143. conn := NewBaseConnection("", ProtocolWebDAV, "", "", dataprovider.User{
  144. BaseUser: sdk.BaseUser{
  145. Permissions: map[string][]string{
  146. "/": {dataprovider.PermListItems, dataprovider.PermUpload,
  147. dataprovider.PermDownload, dataprovider.PermRenameDirs},
  148. },
  149. },
  150. })
  151. err := conn.checkRecursiveRenameDirPermissions(fs, fs, filepath.Join(os.TempDir(), "/source"),
  152. filepath.Join(os.TempDir(), "/target"), "/source", "/target",
  153. vfs.NewFileInfo("source", true, 0, time.Now(), false))
  154. assert.ErrorIs(t, err, os.ErrNotExist)
  155. fs = newMockOsFs(false, "mockID", filepath.Clean(os.TempDir()), "S3Fs", errWalkDir)
  156. err = conn.checkRecursiveRenameDirPermissions(fs, fs, filepath.Join(os.TempDir(), "/source"),
  157. filepath.Join(os.TempDir(), "/target"), "/source", "/target",
  158. vfs.NewFileInfo("source", true, 0, time.Now(), false))
  159. if assert.Error(t, err) {
  160. assert.Equal(t, err.Error(), conn.GetOpUnsupportedError().Error())
  161. }
  162. conn.User.Permissions["/"] = []string{dataprovider.PermListItems, dataprovider.PermUpload,
  163. dataprovider.PermDownload, dataprovider.PermRenameFiles}
  164. // no dir rename permission, the quick check path returns permission error without walking
  165. err = conn.checkRecursiveRenameDirPermissions(fs, fs, filepath.Join(os.TempDir(), "/source"),
  166. filepath.Join(os.TempDir(), "/target"), "/source", "/target",
  167. vfs.NewFileInfo("source", true, 0, time.Now(), false))
  168. if assert.Error(t, err) {
  169. assert.EqualError(t, err, conn.GetPermissionDeniedError().Error())
  170. }
  171. }
  172. func TestCrossRenameFsErrors(t *testing.T) {
  173. fs := vfs.NewOsFs("", os.TempDir(), "")
  174. conn := NewBaseConnection("", ProtocolWebDAV, "", "", dataprovider.User{})
  175. res := conn.hasSpaceForCrossRename(fs, vfs.QuotaCheckResult{}, 1, "missingsource")
  176. assert.False(t, res)
  177. if runtime.GOOS != osWindows {
  178. dirPath := filepath.Join(os.TempDir(), "d")
  179. err := os.Mkdir(dirPath, os.ModePerm)
  180. assert.NoError(t, err)
  181. err = os.Chmod(dirPath, 0001)
  182. assert.NoError(t, err)
  183. res = conn.hasSpaceForCrossRename(fs, vfs.QuotaCheckResult{}, 1, dirPath)
  184. assert.False(t, res)
  185. err = os.Chmod(dirPath, os.ModePerm)
  186. assert.NoError(t, err)
  187. err = os.Remove(dirPath)
  188. assert.NoError(t, err)
  189. }
  190. }
  191. func TestRenameVirtualFolders(t *testing.T) {
  192. vdir := "/avdir"
  193. u := dataprovider.User{}
  194. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  195. BaseVirtualFolder: vfs.BaseVirtualFolder{
  196. Name: "name",
  197. MappedPath: "mappedPath",
  198. },
  199. VirtualPath: vdir,
  200. })
  201. fs := vfs.NewOsFs("", os.TempDir(), "")
  202. conn := NewBaseConnection("", ProtocolFTP, "", "", u)
  203. res := conn.isRenamePermitted(fs, fs, "source", "target", vdir, "vdirtarget", nil)
  204. assert.False(t, res)
  205. }
  206. func TestRenamePerms(t *testing.T) {
  207. src := "source"
  208. target := "target"
  209. sub := "/sub"
  210. subTarget := sub + "/target"
  211. u := dataprovider.User{}
  212. u.Permissions = map[string][]string{}
  213. u.Permissions["/"] = []string{dataprovider.PermCreateDirs, dataprovider.PermUpload, dataprovider.PermCreateSymlinks,
  214. dataprovider.PermDeleteFiles}
  215. conn := NewBaseConnection("", ProtocolSFTP, "", "", u)
  216. assert.False(t, conn.hasRenamePerms(src, target, nil))
  217. u.Permissions["/"] = []string{dataprovider.PermRename}
  218. assert.True(t, conn.hasRenamePerms(src, target, nil))
  219. u.Permissions["/"] = []string{dataprovider.PermCreateDirs, dataprovider.PermUpload, dataprovider.PermDeleteFiles,
  220. dataprovider.PermDeleteDirs}
  221. assert.False(t, conn.hasRenamePerms(src, target, nil))
  222. info := vfs.NewFileInfo(src, true, 0, time.Now(), false)
  223. u.Permissions["/"] = []string{dataprovider.PermRenameFiles}
  224. assert.False(t, conn.hasRenamePerms(src, target, info))
  225. u.Permissions["/"] = []string{dataprovider.PermRenameDirs}
  226. assert.True(t, conn.hasRenamePerms(src, target, info))
  227. u.Permissions["/"] = []string{dataprovider.PermRename}
  228. assert.True(t, conn.hasRenamePerms(src, target, info))
  229. u.Permissions["/"] = []string{dataprovider.PermDownload, dataprovider.PermUpload, dataprovider.PermDeleteDirs}
  230. assert.False(t, conn.hasRenamePerms(src, target, info))
  231. // test with different permissions between source and target
  232. u.Permissions["/"] = []string{dataprovider.PermRename}
  233. u.Permissions[sub] = []string{dataprovider.PermRenameFiles}
  234. assert.False(t, conn.hasRenamePerms(src, subTarget, info))
  235. u.Permissions[sub] = []string{dataprovider.PermRenameDirs}
  236. assert.True(t, conn.hasRenamePerms(src, subTarget, info))
  237. // test files
  238. info = vfs.NewFileInfo(src, false, 0, time.Now(), false)
  239. u.Permissions["/"] = []string{dataprovider.PermRenameDirs}
  240. assert.False(t, conn.hasRenamePerms(src, target, info))
  241. u.Permissions["/"] = []string{dataprovider.PermRenameFiles}
  242. assert.True(t, conn.hasRenamePerms(src, target, info))
  243. u.Permissions["/"] = []string{dataprovider.PermRename}
  244. assert.True(t, conn.hasRenamePerms(src, target, info))
  245. // test with different permissions between source and target
  246. u.Permissions["/"] = []string{dataprovider.PermRename}
  247. u.Permissions[sub] = []string{dataprovider.PermRenameDirs}
  248. assert.False(t, conn.hasRenamePerms(src, subTarget, info))
  249. u.Permissions[sub] = []string{dataprovider.PermRenameFiles}
  250. assert.True(t, conn.hasRenamePerms(src, subTarget, info))
  251. }
  252. func TestRenameNestedFolders(t *testing.T) {
  253. u := dataprovider.User{}
  254. conn := NewBaseConnection("", ProtocolSFTP, "", "", u)
  255. err := conn.checkFolderRename(nil, nil, filepath.Clean(os.TempDir()), filepath.Join(os.TempDir(), "subdir"), "/src", "/dst", nil)
  256. assert.Error(t, err)
  257. err = conn.checkFolderRename(nil, nil, filepath.Join(os.TempDir(), "subdir"), filepath.Clean(os.TempDir()), "/src", "/dst", nil)
  258. assert.Error(t, err)
  259. err = conn.checkFolderRename(nil, nil, "", "", "/src/sub", "/src", nil)
  260. assert.Error(t, err)
  261. }
  262. func TestUpdateQuotaAfterRename(t *testing.T) {
  263. user := dataprovider.User{
  264. BaseUser: sdk.BaseUser{
  265. Username: userTestUsername,
  266. HomeDir: filepath.Join(os.TempDir(), "home"),
  267. },
  268. }
  269. mappedPath := filepath.Join(os.TempDir(), "vdir")
  270. user.Permissions = make(map[string][]string)
  271. user.Permissions["/"] = []string{dataprovider.PermAny}
  272. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  273. BaseVirtualFolder: vfs.BaseVirtualFolder{
  274. MappedPath: mappedPath,
  275. },
  276. VirtualPath: "/vdir",
  277. QuotaFiles: -1,
  278. QuotaSize: -1,
  279. })
  280. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  281. BaseVirtualFolder: vfs.BaseVirtualFolder{
  282. MappedPath: mappedPath,
  283. },
  284. VirtualPath: "/vdir1",
  285. QuotaFiles: -1,
  286. QuotaSize: -1,
  287. })
  288. err := os.MkdirAll(user.GetHomeDir(), os.ModePerm)
  289. assert.NoError(t, err)
  290. err = os.MkdirAll(mappedPath, os.ModePerm)
  291. assert.NoError(t, err)
  292. fs, err := user.GetFilesystem("id")
  293. assert.NoError(t, err)
  294. c := NewBaseConnection("", ProtocolSFTP, "", "", user)
  295. request := sftp.NewRequest("Rename", "/testfile")
  296. if runtime.GOOS != osWindows {
  297. request.Filepath = "/dir"
  298. request.Target = path.Join("/vdir", "dir")
  299. testDirPath := filepath.Join(mappedPath, "dir")
  300. err := os.MkdirAll(testDirPath, os.ModePerm)
  301. assert.NoError(t, err)
  302. err = os.Chmod(testDirPath, 0001)
  303. assert.NoError(t, err)
  304. err = c.updateQuotaAfterRename(fs, request.Filepath, request.Target, testDirPath, 0)
  305. assert.Error(t, err)
  306. err = os.Chmod(testDirPath, os.ModePerm)
  307. assert.NoError(t, err)
  308. }
  309. testFile1 := "/testfile1"
  310. request.Target = testFile1
  311. request.Filepath = path.Join("/vdir", "file")
  312. err = c.updateQuotaAfterRename(fs, request.Filepath, request.Target, filepath.Join(mappedPath, "file"), 0)
  313. assert.Error(t, err)
  314. err = os.WriteFile(filepath.Join(mappedPath, "file"), []byte("test content"), os.ModePerm)
  315. assert.NoError(t, err)
  316. request.Filepath = testFile1
  317. request.Target = path.Join("/vdir", "file")
  318. err = c.updateQuotaAfterRename(fs, request.Filepath, request.Target, filepath.Join(mappedPath, "file"), 12)
  319. assert.NoError(t, err)
  320. err = os.WriteFile(filepath.Join(user.GetHomeDir(), "testfile1"), []byte("test content"), os.ModePerm)
  321. assert.NoError(t, err)
  322. request.Target = testFile1
  323. request.Filepath = path.Join("/vdir", "file")
  324. err = c.updateQuotaAfterRename(fs, request.Filepath, request.Target, filepath.Join(mappedPath, "file"), 12)
  325. assert.NoError(t, err)
  326. request.Target = path.Join("/vdir1", "file")
  327. request.Filepath = path.Join("/vdir", "file")
  328. err = c.updateQuotaAfterRename(fs, request.Filepath, request.Target, filepath.Join(mappedPath, "file"), 12)
  329. assert.NoError(t, err)
  330. err = os.RemoveAll(mappedPath)
  331. assert.NoError(t, err)
  332. err = os.RemoveAll(user.GetHomeDir())
  333. assert.NoError(t, err)
  334. }
  335. func TestErrorsMapping(t *testing.T) {
  336. fs := vfs.NewOsFs("", os.TempDir(), "")
  337. conn := NewBaseConnection("", ProtocolSFTP, "", "", dataprovider.User{BaseUser: sdk.BaseUser{HomeDir: os.TempDir()}})
  338. osErrorsProtocols := []string{ProtocolWebDAV, ProtocolFTP, ProtocolHTTP, ProtocolHTTPShare,
  339. ProtocolDataRetention, ProtocolOIDC, protocolEventAction}
  340. for _, protocol := range supportedProtocols {
  341. conn.SetProtocol(protocol)
  342. err := conn.GetFsError(fs, os.ErrNotExist)
  343. if protocol == ProtocolSFTP {
  344. assert.ErrorIs(t, err, sftp.ErrSSHFxNoSuchFile)
  345. } else if util.Contains(osErrorsProtocols, protocol) {
  346. assert.EqualError(t, err, os.ErrNotExist.Error())
  347. } else {
  348. assert.EqualError(t, err, ErrNotExist.Error())
  349. }
  350. err = conn.GetFsError(fs, os.ErrPermission)
  351. if protocol == ProtocolSFTP {
  352. assert.EqualError(t, err, sftp.ErrSSHFxPermissionDenied.Error())
  353. } else {
  354. assert.EqualError(t, err, ErrPermissionDenied.Error())
  355. }
  356. err = conn.GetFsError(fs, os.ErrClosed)
  357. if protocol == ProtocolSFTP {
  358. assert.ErrorIs(t, err, sftp.ErrSSHFxFailure)
  359. } else {
  360. assert.EqualError(t, err, ErrGenericFailure.Error())
  361. }
  362. err = conn.GetFsError(fs, ErrPermissionDenied)
  363. if protocol == ProtocolSFTP {
  364. assert.ErrorIs(t, err, sftp.ErrSSHFxFailure)
  365. } else {
  366. assert.EqualError(t, err, ErrPermissionDenied.Error())
  367. }
  368. err = conn.GetFsError(fs, vfs.ErrVfsUnsupported)
  369. if protocol == ProtocolSFTP {
  370. assert.EqualError(t, err, sftp.ErrSSHFxOpUnsupported.Error())
  371. } else {
  372. assert.EqualError(t, err, ErrOpUnsupported.Error())
  373. }
  374. err = conn.GetFsError(fs, vfs.ErrStorageSizeUnavailable)
  375. if protocol == ProtocolSFTP {
  376. assert.ErrorIs(t, err, sftp.ErrSSHFxOpUnsupported)
  377. assert.Contains(t, err.Error(), vfs.ErrStorageSizeUnavailable.Error())
  378. } else {
  379. assert.EqualError(t, err, vfs.ErrStorageSizeUnavailable.Error())
  380. }
  381. err = conn.GetQuotaExceededError()
  382. assert.True(t, conn.IsQuotaExceededError(err))
  383. err = conn.GetReadQuotaExceededError()
  384. if protocol == ProtocolSFTP {
  385. assert.ErrorIs(t, err, sftp.ErrSSHFxFailure)
  386. assert.Contains(t, err.Error(), ErrReadQuotaExceeded.Error())
  387. } else {
  388. assert.ErrorIs(t, err, ErrReadQuotaExceeded)
  389. }
  390. err = conn.GetNotExistError()
  391. assert.True(t, conn.IsNotExistError(err))
  392. err = conn.GetFsError(fs, nil)
  393. assert.NoError(t, err)
  394. err = conn.GetOpUnsupportedError()
  395. if protocol == ProtocolSFTP {
  396. assert.EqualError(t, err, sftp.ErrSSHFxOpUnsupported.Error())
  397. } else {
  398. assert.EqualError(t, err, ErrOpUnsupported.Error())
  399. }
  400. err = conn.GetFsError(fs, ErrShuttingDown)
  401. if protocol == ProtocolSFTP {
  402. assert.ErrorIs(t, err, sftp.ErrSSHFxFailure)
  403. assert.Contains(t, err.Error(), ErrShuttingDown.Error())
  404. } else {
  405. assert.EqualError(t, err, ErrShuttingDown.Error())
  406. }
  407. }
  408. }
  409. func TestMaxWriteSize(t *testing.T) {
  410. permissions := make(map[string][]string)
  411. permissions["/"] = []string{dataprovider.PermAny}
  412. user := dataprovider.User{
  413. BaseUser: sdk.BaseUser{
  414. Username: userTestUsername,
  415. Permissions: permissions,
  416. HomeDir: filepath.Clean(os.TempDir()),
  417. },
  418. }
  419. fs, err := user.GetFilesystem("123")
  420. assert.NoError(t, err)
  421. conn := NewBaseConnection("", ProtocolFTP, "", "", user)
  422. quotaResult := vfs.QuotaCheckResult{
  423. HasSpace: true,
  424. }
  425. size, err := conn.GetMaxWriteSize(quotaResult, false, 0, fs.IsUploadResumeSupported())
  426. assert.NoError(t, err)
  427. assert.Equal(t, int64(0), size)
  428. conn.User.Filters.MaxUploadFileSize = 100
  429. size, err = conn.GetMaxWriteSize(quotaResult, false, 0, fs.IsUploadResumeSupported())
  430. assert.NoError(t, err)
  431. assert.Equal(t, int64(100), size)
  432. quotaResult.QuotaSize = 1000
  433. size, err = conn.GetMaxWriteSize(quotaResult, false, 50, fs.IsUploadResumeSupported())
  434. assert.NoError(t, err)
  435. assert.Equal(t, int64(100), size)
  436. quotaResult.QuotaSize = 1000
  437. quotaResult.UsedSize = 990
  438. size, err = conn.GetMaxWriteSize(quotaResult, false, 50, fs.IsUploadResumeSupported())
  439. assert.NoError(t, err)
  440. assert.Equal(t, int64(60), size)
  441. quotaResult.QuotaSize = 0
  442. quotaResult.UsedSize = 0
  443. size, err = conn.GetMaxWriteSize(quotaResult, true, 100, fs.IsUploadResumeSupported())
  444. assert.True(t, conn.IsQuotaExceededError(err))
  445. assert.Equal(t, int64(0), size)
  446. size, err = conn.GetMaxWriteSize(quotaResult, true, 10, fs.IsUploadResumeSupported())
  447. assert.NoError(t, err)
  448. assert.Equal(t, int64(90), size)
  449. fs = newMockOsFs(true, fs.ConnectionID(), user.GetHomeDir(), "", nil)
  450. size, err = conn.GetMaxWriteSize(quotaResult, true, 100, fs.IsUploadResumeSupported())
  451. assert.EqualError(t, err, ErrOpUnsupported.Error())
  452. assert.Equal(t, int64(0), size)
  453. }
  454. func TestCheckParentDirsErrors(t *testing.T) {
  455. permissions := make(map[string][]string)
  456. permissions["/"] = []string{dataprovider.PermAny}
  457. user := dataprovider.User{
  458. BaseUser: sdk.BaseUser{
  459. Username: userTestUsername,
  460. Permissions: permissions,
  461. HomeDir: filepath.Clean(os.TempDir()),
  462. },
  463. FsConfig: vfs.Filesystem{
  464. Provider: sdk.CryptedFilesystemProvider,
  465. },
  466. }
  467. c := NewBaseConnection(xid.New().String(), ProtocolSFTP, "", "", user)
  468. err := c.CheckParentDirs("/a/dir")
  469. assert.Error(t, err)
  470. user.FsConfig.Provider = sdk.LocalFilesystemProvider
  471. user.VirtualFolders = nil
  472. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  473. BaseVirtualFolder: vfs.BaseVirtualFolder{
  474. FsConfig: vfs.Filesystem{
  475. Provider: sdk.CryptedFilesystemProvider,
  476. },
  477. },
  478. VirtualPath: "/vdir",
  479. })
  480. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  481. BaseVirtualFolder: vfs.BaseVirtualFolder{
  482. MappedPath: filepath.Clean(os.TempDir()),
  483. },
  484. VirtualPath: "/vdir/sub",
  485. })
  486. c = NewBaseConnection(xid.New().String(), ProtocolSFTP, "", "", user)
  487. err = c.CheckParentDirs("/vdir/sub/dir")
  488. assert.Error(t, err)
  489. user = dataprovider.User{
  490. BaseUser: sdk.BaseUser{
  491. Username: userTestUsername,
  492. Permissions: permissions,
  493. HomeDir: filepath.Clean(os.TempDir()),
  494. },
  495. FsConfig: vfs.Filesystem{
  496. Provider: sdk.S3FilesystemProvider,
  497. S3Config: vfs.S3FsConfig{
  498. BaseS3FsConfig: sdk.BaseS3FsConfig{
  499. Bucket: "buck",
  500. Region: "us-east-1",
  501. AccessKey: "key",
  502. },
  503. AccessSecret: kms.NewPlainSecret("s3secret"),
  504. },
  505. },
  506. }
  507. c = NewBaseConnection(xid.New().String(), ProtocolSFTP, "", "", user)
  508. err = c.CheckParentDirs("/a/dir")
  509. assert.NoError(t, err)
  510. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  511. BaseVirtualFolder: vfs.BaseVirtualFolder{
  512. MappedPath: filepath.Clean(os.TempDir()),
  513. },
  514. VirtualPath: "/local/dir",
  515. })
  516. c = NewBaseConnection(xid.New().String(), ProtocolSFTP, "", "", user)
  517. err = c.CheckParentDirs("/local/dir/sub-dir")
  518. assert.NoError(t, err)
  519. err = os.RemoveAll(filepath.Join(os.TempDir(), "sub-dir"))
  520. assert.NoError(t, err)
  521. }
  522. func TestErrorResolvePath(t *testing.T) {
  523. u := dataprovider.User{
  524. BaseUser: sdk.BaseUser{
  525. HomeDir: filepath.Join(os.TempDir(), "u"),
  526. Status: 1,
  527. Permissions: map[string][]string{
  528. "/": {dataprovider.PermAny},
  529. },
  530. },
  531. }
  532. u.FsConfig.Provider = sdk.GCSFilesystemProvider
  533. u.FsConfig.GCSConfig.Bucket = "test"
  534. u.FsConfig.GCSConfig.Credentials = kms.NewPlainSecret("invalid JSON for credentials")
  535. u.VirtualFolders = []vfs.VirtualFolder{
  536. {
  537. BaseVirtualFolder: vfs.BaseVirtualFolder{
  538. Name: "f",
  539. MappedPath: filepath.Join(os.TempDir(), "f"),
  540. },
  541. VirtualPath: "/f",
  542. },
  543. }
  544. conn := NewBaseConnection("", ProtocolSFTP, "", "", u)
  545. err := conn.doRecursiveRemoveDirEntry("/vpath", nil)
  546. assert.Error(t, err)
  547. err = conn.checkCopy(vfs.NewFileInfo("name", true, 0, time.Unix(0, 0), false), nil, "/source", "/target")
  548. assert.Error(t, err)
  549. sourceFile := filepath.Join(os.TempDir(), "f", "source")
  550. err = os.MkdirAll(filepath.Dir(sourceFile), os.ModePerm)
  551. assert.NoError(t, err)
  552. err = os.WriteFile(sourceFile, []byte(""), 0666)
  553. assert.NoError(t, err)
  554. err = conn.checkCopy(vfs.NewFileInfo("name", true, 0, time.Unix(0, 0), false), nil, "/f/source", "/target")
  555. assert.Error(t, err)
  556. err = conn.checkCopy(vfs.NewFileInfo("source", false, 0, time.Unix(0, 0), false), vfs.NewFileInfo("target", true, 0, time.Unix(0, 0), false), "/f/source", "/f/target")
  557. assert.Error(t, err)
  558. err = os.RemoveAll(filepath.Dir(sourceFile))
  559. assert.NoError(t, err)
  560. }