internal_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634
  1. package httpd
  2. import (
  3. "context"
  4. "fmt"
  5. "html/template"
  6. "io/ioutil"
  7. "net/http"
  8. "net/http/httptest"
  9. "net/url"
  10. "os"
  11. "path/filepath"
  12. "runtime"
  13. "strings"
  14. "testing"
  15. "github.com/go-chi/chi"
  16. "github.com/stretchr/testify/assert"
  17. "github.com/drakkan/sftpgo/common"
  18. "github.com/drakkan/sftpgo/dataprovider"
  19. "github.com/drakkan/sftpgo/utils"
  20. "github.com/drakkan/sftpgo/vfs"
  21. )
  22. const (
  23. invalidURL = "http://foo\x7f.com/"
  24. inactiveURL = "http://127.0.0.1:12345"
  25. )
  26. func TestGetRespStatus(t *testing.T) {
  27. var err error
  28. err = &dataprovider.MethodDisabledError{}
  29. respStatus := getRespStatus(err)
  30. assert.Equal(t, http.StatusForbidden, respStatus)
  31. err = fmt.Errorf("generic error")
  32. respStatus = getRespStatus(err)
  33. assert.Equal(t, http.StatusInternalServerError, respStatus)
  34. }
  35. func TestCheckResponse(t *testing.T) {
  36. err := checkResponse(http.StatusOK, http.StatusCreated)
  37. assert.Error(t, err)
  38. err = checkResponse(http.StatusBadRequest, http.StatusBadRequest)
  39. assert.NoError(t, err)
  40. }
  41. func TestCheckFolder(t *testing.T) {
  42. expected := &vfs.BaseVirtualFolder{}
  43. actual := &vfs.BaseVirtualFolder{}
  44. err := checkFolder(expected, actual)
  45. assert.Error(t, err)
  46. expected.ID = 1
  47. actual.ID = 2
  48. err = checkFolder(expected, actual)
  49. assert.Error(t, err)
  50. expected.ID = 2
  51. actual.ID = 2
  52. expected.MappedPath = "path"
  53. err = checkFolder(expected, actual)
  54. assert.Error(t, err)
  55. expected.MappedPath = ""
  56. expected.LastQuotaUpdate = 1
  57. err = checkFolder(expected, actual)
  58. assert.Error(t, err)
  59. expected.LastQuotaUpdate = 0
  60. expected.UsedQuotaFiles = 1
  61. err = checkFolder(expected, actual)
  62. assert.Error(t, err)
  63. expected.UsedQuotaFiles = 0
  64. expected.UsedQuotaSize = 1
  65. err = checkFolder(expected, actual)
  66. assert.Error(t, err)
  67. expected.UsedQuotaSize = 0
  68. expected.Users = append(expected.Users, "user1")
  69. err = checkFolder(expected, actual)
  70. assert.Error(t, err)
  71. actual.Users = append(actual.Users, "user2")
  72. err = checkFolder(expected, actual)
  73. assert.Error(t, err)
  74. expected.Users = nil
  75. actual.Users = nil
  76. }
  77. func TestCheckUser(t *testing.T) {
  78. expected := &dataprovider.User{}
  79. actual := &dataprovider.User{}
  80. actual.Password = "password"
  81. err := checkUser(expected, actual)
  82. assert.Error(t, err)
  83. actual.Password = ""
  84. err = checkUser(expected, actual)
  85. assert.Error(t, err)
  86. expected.ID = 1
  87. actual.ID = 2
  88. err = checkUser(expected, actual)
  89. assert.Error(t, err)
  90. expected.ID = 2
  91. actual.ID = 2
  92. expected.Permissions = make(map[string][]string)
  93. expected.Permissions["/"] = []string{dataprovider.PermCreateDirs, dataprovider.PermDelete, dataprovider.PermDownload}
  94. actual.Permissions = make(map[string][]string)
  95. err = checkUser(expected, actual)
  96. assert.Error(t, err)
  97. actual.Permissions["/"] = []string{dataprovider.PermCreateDirs, dataprovider.PermCreateSymlinks}
  98. err = checkUser(expected, actual)
  99. assert.Error(t, err)
  100. expected.Permissions["/"] = append(expected.Permissions["/"], dataprovider.PermRename)
  101. err = checkUser(expected, actual)
  102. assert.Error(t, err)
  103. expected.Permissions = make(map[string][]string)
  104. expected.Permissions["/somedir"] = []string{dataprovider.PermAny}
  105. actual.Permissions = make(map[string][]string)
  106. actual.Permissions["/otherdir"] = []string{dataprovider.PermCreateDirs, dataprovider.PermCreateSymlinks}
  107. err = checkUser(expected, actual)
  108. assert.Error(t, err)
  109. expected.Permissions = make(map[string][]string)
  110. actual.Permissions = make(map[string][]string)
  111. actual.FsConfig.Provider = dataprovider.S3FilesystemProvider
  112. err = checkUser(expected, actual)
  113. assert.Error(t, err)
  114. actual.FsConfig.Provider = dataprovider.LocalFilesystemProvider
  115. expected.VirtualFolders = append(expected.VirtualFolders, vfs.VirtualFolder{
  116. BaseVirtualFolder: vfs.BaseVirtualFolder{
  117. MappedPath: os.TempDir(),
  118. },
  119. VirtualPath: "/vdir",
  120. })
  121. err = checkUser(expected, actual)
  122. assert.Error(t, err)
  123. actual.VirtualFolders = append(actual.VirtualFolders, vfs.VirtualFolder{
  124. BaseVirtualFolder: vfs.BaseVirtualFolder{
  125. MappedPath: os.TempDir(),
  126. },
  127. VirtualPath: "/vdir1",
  128. })
  129. err = checkUser(expected, actual)
  130. assert.Error(t, err)
  131. }
  132. func TestCompareUserFilters(t *testing.T) {
  133. expected := &dataprovider.User{}
  134. actual := &dataprovider.User{}
  135. actual.ID = 1
  136. expected.ID = 1
  137. expected.Filters.AllowedIP = []string{}
  138. actual.Filters.AllowedIP = []string{"192.168.1.2/32"}
  139. err := checkUser(expected, actual)
  140. assert.Error(t, err)
  141. expected.Filters.AllowedIP = []string{"192.168.1.3/32"}
  142. err = checkUser(expected, actual)
  143. assert.Error(t, err)
  144. expected.Filters.AllowedIP = []string{}
  145. actual.Filters.AllowedIP = []string{}
  146. expected.Filters.DeniedIP = []string{}
  147. actual.Filters.DeniedIP = []string{"192.168.1.2/32"}
  148. err = checkUser(expected, actual)
  149. assert.Error(t, err)
  150. expected.Filters.DeniedIP = []string{"192.168.1.3/32"}
  151. err = checkUser(expected, actual)
  152. assert.Error(t, err)
  153. expected.Filters.DeniedIP = []string{}
  154. actual.Filters.DeniedIP = []string{}
  155. expected.Filters.DeniedLoginMethods = []string{}
  156. actual.Filters.DeniedLoginMethods = []string{dataprovider.SSHLoginMethodPublicKey}
  157. err = checkUser(expected, actual)
  158. assert.Error(t, err)
  159. expected.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
  160. err = checkUser(expected, actual)
  161. assert.Error(t, err)
  162. expected.Filters.DeniedLoginMethods = []string{}
  163. actual.Filters.DeniedLoginMethods = []string{}
  164. actual.Filters.DeniedProtocols = []string{common.ProtocolFTP}
  165. err = checkUser(expected, actual)
  166. assert.Error(t, err)
  167. expected.Filters.DeniedProtocols = []string{common.ProtocolWebDAV}
  168. err = checkUser(expected, actual)
  169. assert.Error(t, err)
  170. expected.Filters.DeniedProtocols = []string{}
  171. actual.Filters.DeniedProtocols = []string{}
  172. expected.Filters.MaxUploadFileSize = 0
  173. actual.Filters.MaxUploadFileSize = 100
  174. err = checkUser(expected, actual)
  175. assert.Error(t, err)
  176. actual.Filters.MaxUploadFileSize = 0
  177. expected.Filters.FileExtensions = append(expected.Filters.FileExtensions, dataprovider.ExtensionsFilter{
  178. Path: "/",
  179. AllowedExtensions: []string{".jpg", ".png"},
  180. DeniedExtensions: []string{".zip", ".rar"},
  181. })
  182. err = checkUser(expected, actual)
  183. assert.Error(t, err)
  184. actual.Filters.FileExtensions = append(actual.Filters.FileExtensions, dataprovider.ExtensionsFilter{
  185. Path: "/sub",
  186. AllowedExtensions: []string{".jpg", ".png"},
  187. DeniedExtensions: []string{".zip", ".rar"},
  188. })
  189. err = checkUser(expected, actual)
  190. assert.Error(t, err)
  191. actual.Filters.FileExtensions[0] = dataprovider.ExtensionsFilter{
  192. Path: "/",
  193. AllowedExtensions: []string{".jpg"},
  194. DeniedExtensions: []string{".zip", ".rar"},
  195. }
  196. err = checkUser(expected, actual)
  197. assert.Error(t, err)
  198. actual.Filters.FileExtensions[0] = dataprovider.ExtensionsFilter{
  199. Path: "/",
  200. AllowedExtensions: []string{".tiff", ".png"},
  201. DeniedExtensions: []string{".zip", ".rar"},
  202. }
  203. err = checkUser(expected, actual)
  204. assert.Error(t, err)
  205. actual.Filters.FileExtensions[0] = dataprovider.ExtensionsFilter{
  206. Path: "/",
  207. AllowedExtensions: []string{".jpg", ".png"},
  208. DeniedExtensions: []string{".tar.gz", ".rar"},
  209. }
  210. err = checkUser(expected, actual)
  211. assert.Error(t, err)
  212. actual.Filters.FileExtensions = nil
  213. actual.Filters.FilePatterns = nil
  214. expected.Filters.FileExtensions = nil
  215. expected.Filters.FilePatterns = nil
  216. expected.Filters.FilePatterns = append(expected.Filters.FilePatterns, dataprovider.PatternsFilter{
  217. Path: "/",
  218. AllowedPatterns: []string{"*.jpg", "*.png"},
  219. DeniedPatterns: []string{"*.zip", "*.rar"},
  220. })
  221. err = checkUser(expected, actual)
  222. assert.Error(t, err)
  223. actual.Filters.FilePatterns = append(actual.Filters.FilePatterns, dataprovider.PatternsFilter{
  224. Path: "/sub",
  225. AllowedPatterns: []string{"*.jpg", "*.png"},
  226. DeniedPatterns: []string{"*.zip", "*.rar"},
  227. })
  228. err = checkUser(expected, actual)
  229. assert.Error(t, err)
  230. actual.Filters.FilePatterns[0] = dataprovider.PatternsFilter{
  231. Path: "/",
  232. AllowedPatterns: []string{"*.jpg"},
  233. DeniedPatterns: []string{"*.zip", "*.rar"},
  234. }
  235. err = checkUser(expected, actual)
  236. assert.Error(t, err)
  237. actual.Filters.FilePatterns[0] = dataprovider.PatternsFilter{
  238. Path: "/",
  239. AllowedPatterns: []string{"*.tiff", "*.png"},
  240. DeniedPatterns: []string{"*.zip", "*.rar"},
  241. }
  242. err = checkUser(expected, actual)
  243. assert.Error(t, err)
  244. actual.Filters.FilePatterns[0] = dataprovider.PatternsFilter{
  245. Path: "/",
  246. AllowedPatterns: []string{"*.jpg", "*.png"},
  247. DeniedPatterns: []string{"*.tar.gz", "*.rar"},
  248. }
  249. err = checkUser(expected, actual)
  250. assert.Error(t, err)
  251. }
  252. func TestCompareUserFields(t *testing.T) {
  253. expected := &dataprovider.User{}
  254. actual := &dataprovider.User{}
  255. expected.Permissions = make(map[string][]string)
  256. actual.Permissions = make(map[string][]string)
  257. expected.Username = "test"
  258. err := compareEqualsUserFields(expected, actual)
  259. assert.Error(t, err)
  260. expected.Username = ""
  261. expected.HomeDir = "homedir"
  262. err = compareEqualsUserFields(expected, actual)
  263. assert.Error(t, err)
  264. expected.HomeDir = ""
  265. expected.UID = 1
  266. err = compareEqualsUserFields(expected, actual)
  267. assert.Error(t, err)
  268. expected.UID = 0
  269. expected.GID = 1
  270. err = compareEqualsUserFields(expected, actual)
  271. assert.Error(t, err)
  272. expected.GID = 0
  273. expected.MaxSessions = 2
  274. err = compareEqualsUserFields(expected, actual)
  275. assert.Error(t, err)
  276. expected.MaxSessions = 0
  277. expected.QuotaSize = 4096
  278. err = compareEqualsUserFields(expected, actual)
  279. assert.Error(t, err)
  280. expected.QuotaSize = 0
  281. expected.QuotaFiles = 2
  282. err = compareEqualsUserFields(expected, actual)
  283. assert.Error(t, err)
  284. expected.QuotaFiles = 0
  285. expected.Permissions["/"] = []string{dataprovider.PermCreateDirs}
  286. err = compareEqualsUserFields(expected, actual)
  287. assert.Error(t, err)
  288. expected.Permissions = nil
  289. expected.UploadBandwidth = 64
  290. err = compareEqualsUserFields(expected, actual)
  291. assert.Error(t, err)
  292. expected.UploadBandwidth = 0
  293. expected.DownloadBandwidth = 128
  294. err = compareEqualsUserFields(expected, actual)
  295. assert.Error(t, err)
  296. expected.DownloadBandwidth = 0
  297. expected.Status = 1
  298. err = compareEqualsUserFields(expected, actual)
  299. assert.Error(t, err)
  300. expected.Status = 0
  301. expected.ExpirationDate = 123
  302. err = compareEqualsUserFields(expected, actual)
  303. assert.Error(t, err)
  304. }
  305. func TestCompareUserFsConfig(t *testing.T) {
  306. expected := &dataprovider.User{}
  307. actual := &dataprovider.User{}
  308. expected.FsConfig.Provider = dataprovider.S3FilesystemProvider
  309. err := compareUserFsConfig(expected, actual)
  310. assert.Error(t, err)
  311. expected.FsConfig.Provider = dataprovider.LocalFilesystemProvider
  312. expected.FsConfig.S3Config.Bucket = "bucket"
  313. err = compareUserFsConfig(expected, actual)
  314. assert.Error(t, err)
  315. expected.FsConfig.S3Config.Bucket = ""
  316. expected.FsConfig.S3Config.Region = "region"
  317. err = compareUserFsConfig(expected, actual)
  318. assert.Error(t, err)
  319. expected.FsConfig.S3Config.Region = ""
  320. expected.FsConfig.S3Config.AccessKey = "access key"
  321. err = compareUserFsConfig(expected, actual)
  322. assert.Error(t, err)
  323. expected.FsConfig.S3Config.AccessKey = ""
  324. actual.FsConfig.S3Config.AccessSecret = "access secret"
  325. err = compareUserFsConfig(expected, actual)
  326. assert.Error(t, err)
  327. secret, _ := utils.EncryptData("access secret")
  328. actual.FsConfig.S3Config.AccessSecret = ""
  329. expected.FsConfig.S3Config.AccessSecret = secret
  330. err = compareUserFsConfig(expected, actual)
  331. assert.Error(t, err)
  332. expected.FsConfig.S3Config.AccessSecret = utils.RemoveDecryptionKey(secret)
  333. actual.FsConfig.S3Config.AccessSecret = utils.RemoveDecryptionKey(secret) + "a"
  334. err = compareUserFsConfig(expected, actual)
  335. assert.Error(t, err)
  336. expected.FsConfig.S3Config.AccessSecret = "test"
  337. actual.FsConfig.S3Config.AccessSecret = ""
  338. err = compareUserFsConfig(expected, actual)
  339. assert.Error(t, err)
  340. expected.FsConfig.S3Config.AccessSecret = ""
  341. actual.FsConfig.S3Config.AccessSecret = ""
  342. expected.FsConfig.S3Config.Endpoint = "http://127.0.0.1:9000/"
  343. err = compareUserFsConfig(expected, actual)
  344. assert.Error(t, err)
  345. expected.FsConfig.S3Config.Endpoint = ""
  346. expected.FsConfig.S3Config.StorageClass = "Standard"
  347. err = compareUserFsConfig(expected, actual)
  348. assert.Error(t, err)
  349. expected.FsConfig.S3Config.StorageClass = ""
  350. expected.FsConfig.S3Config.KeyPrefix = "somedir/subdir"
  351. err = compareUserFsConfig(expected, actual)
  352. assert.Error(t, err)
  353. expected.FsConfig.S3Config.KeyPrefix = ""
  354. expected.FsConfig.S3Config.UploadPartSize = 10
  355. err = compareUserFsConfig(expected, actual)
  356. assert.Error(t, err)
  357. expected.FsConfig.S3Config.UploadPartSize = 0
  358. expected.FsConfig.S3Config.UploadConcurrency = 3
  359. err = compareUserFsConfig(expected, actual)
  360. assert.Error(t, err)
  361. }
  362. func TestCompareUserGCSConfig(t *testing.T) {
  363. expected := &dataprovider.User{}
  364. actual := &dataprovider.User{}
  365. expected.FsConfig.GCSConfig.KeyPrefix = "somedir/subdir"
  366. err := compareUserFsConfig(expected, actual)
  367. assert.Error(t, err)
  368. expected.FsConfig.GCSConfig.KeyPrefix = ""
  369. expected.FsConfig.GCSConfig.Bucket = "bucket"
  370. err = compareUserFsConfig(expected, actual)
  371. assert.Error(t, err)
  372. expected.FsConfig.GCSConfig.Bucket = ""
  373. expected.FsConfig.GCSConfig.StorageClass = "Standard"
  374. err = compareUserFsConfig(expected, actual)
  375. assert.Error(t, err)
  376. expected.FsConfig.GCSConfig.StorageClass = ""
  377. expected.FsConfig.GCSConfig.AutomaticCredentials = 1
  378. err = compareUserFsConfig(expected, actual)
  379. assert.Error(t, err)
  380. expected.FsConfig.GCSConfig.AutomaticCredentials = 0
  381. }
  382. func TestCompareUserAzureConfig(t *testing.T) {
  383. expected := &dataprovider.User{}
  384. actual := &dataprovider.User{}
  385. expected.FsConfig.AzBlobConfig.Container = "a"
  386. err := compareUserFsConfig(expected, actual)
  387. assert.Error(t, err)
  388. expected.FsConfig.AzBlobConfig.Container = ""
  389. expected.FsConfig.AzBlobConfig.AccountName = "aname"
  390. err = compareUserFsConfig(expected, actual)
  391. assert.Error(t, err)
  392. expected.FsConfig.AzBlobConfig.AccountName = ""
  393. expected.FsConfig.AzBlobConfig.AccountKey = "akey"
  394. err = compareUserFsConfig(expected, actual)
  395. assert.Error(t, err)
  396. expected.FsConfig.AzBlobConfig.AccountKey = ""
  397. expected.FsConfig.AzBlobConfig.Endpoint = "endpt"
  398. err = compareUserFsConfig(expected, actual)
  399. assert.Error(t, err)
  400. expected.FsConfig.AzBlobConfig.Endpoint = ""
  401. expected.FsConfig.AzBlobConfig.SASURL = "url"
  402. err = compareUserFsConfig(expected, actual)
  403. assert.Error(t, err)
  404. expected.FsConfig.AzBlobConfig.SASURL = ""
  405. expected.FsConfig.AzBlobConfig.UploadPartSize = 1
  406. err = compareUserFsConfig(expected, actual)
  407. assert.Error(t, err)
  408. expected.FsConfig.AzBlobConfig.UploadPartSize = 0
  409. expected.FsConfig.AzBlobConfig.UploadConcurrency = 1
  410. err = compareUserFsConfig(expected, actual)
  411. assert.Error(t, err)
  412. expected.FsConfig.AzBlobConfig.UploadConcurrency = 0
  413. expected.FsConfig.AzBlobConfig.KeyPrefix = "prefix/"
  414. err = compareUserFsConfig(expected, actual)
  415. assert.Error(t, err)
  416. expected.FsConfig.AzBlobConfig.KeyPrefix = ""
  417. expected.FsConfig.AzBlobConfig.UseEmulator = true
  418. err = compareUserFsConfig(expected, actual)
  419. assert.Error(t, err)
  420. expected.FsConfig.AzBlobConfig.UseEmulator = false
  421. }
  422. func TestGCSWebInvalidFormFile(t *testing.T) {
  423. form := make(url.Values)
  424. form.Set("username", "test_username")
  425. form.Set("fs_provider", "2")
  426. req, _ := http.NewRequest(http.MethodPost, webUserPath, strings.NewReader(form.Encode()))
  427. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  428. err := req.ParseForm()
  429. assert.NoError(t, err)
  430. _, err = getFsConfigFromUserPostFields(req)
  431. assert.EqualError(t, err, http.ErrNotMultipart.Error())
  432. }
  433. func TestApiCallsWithBadURL(t *testing.T) {
  434. oldBaseURL := httpBaseURL
  435. oldAuthUsername := authUsername
  436. oldAuthPassword := authPassword
  437. SetBaseURLAndCredentials(invalidURL, oldAuthUsername, oldAuthPassword)
  438. folder := vfs.BaseVirtualFolder{
  439. MappedPath: os.TempDir(),
  440. }
  441. u := dataprovider.User{}
  442. _, _, err := UpdateUser(u, http.StatusBadRequest, "")
  443. assert.Error(t, err)
  444. _, err = RemoveUser(u, http.StatusNotFound)
  445. assert.Error(t, err)
  446. _, err = RemoveFolder(folder, http.StatusNotFound)
  447. assert.Error(t, err)
  448. _, _, err = GetUsers(1, 0, "", http.StatusBadRequest)
  449. assert.Error(t, err)
  450. _, _, err = GetFolders(1, 0, "", http.StatusBadRequest)
  451. assert.Error(t, err)
  452. _, err = UpdateQuotaUsage(u, "", http.StatusNotFound)
  453. assert.Error(t, err)
  454. _, err = UpdateFolderQuotaUsage(folder, "", http.StatusNotFound)
  455. assert.Error(t, err)
  456. _, err = CloseConnection("non_existent_id", http.StatusNotFound)
  457. assert.Error(t, err)
  458. _, _, err = Dumpdata("backup.json", "", http.StatusBadRequest)
  459. assert.Error(t, err)
  460. _, _, err = Loaddata("/tmp/backup.json", "", "", http.StatusBadRequest)
  461. assert.Error(t, err)
  462. SetBaseURLAndCredentials(oldBaseURL, oldAuthUsername, oldAuthPassword)
  463. }
  464. func TestApiCallToNotListeningServer(t *testing.T) {
  465. oldBaseURL := httpBaseURL
  466. oldAuthUsername := authUsername
  467. oldAuthPassword := authPassword
  468. SetBaseURLAndCredentials(inactiveURL, oldAuthUsername, oldAuthPassword)
  469. u := dataprovider.User{}
  470. _, _, err := AddUser(u, http.StatusBadRequest)
  471. assert.Error(t, err)
  472. _, _, err = UpdateUser(u, http.StatusNotFound, "")
  473. assert.Error(t, err)
  474. _, err = RemoveUser(u, http.StatusNotFound)
  475. assert.Error(t, err)
  476. _, _, err = GetUserByID(-1, http.StatusNotFound)
  477. assert.Error(t, err)
  478. _, _, err = GetUsers(100, 0, "", http.StatusOK)
  479. assert.Error(t, err)
  480. _, err = UpdateQuotaUsage(u, "", http.StatusNotFound)
  481. assert.Error(t, err)
  482. _, _, err = GetQuotaScans(http.StatusOK)
  483. assert.Error(t, err)
  484. _, err = StartQuotaScan(u, http.StatusNotFound)
  485. assert.Error(t, err)
  486. folder := vfs.BaseVirtualFolder{
  487. MappedPath: os.TempDir(),
  488. }
  489. _, err = StartFolderQuotaScan(folder, http.StatusNotFound)
  490. assert.Error(t, err)
  491. _, _, err = AddFolder(folder, http.StatusOK)
  492. assert.Error(t, err)
  493. _, err = RemoveFolder(folder, http.StatusOK)
  494. assert.Error(t, err)
  495. _, _, err = GetFolders(0, 0, "", http.StatusOK)
  496. assert.Error(t, err)
  497. _, err = UpdateFolderQuotaUsage(folder, "", http.StatusNotFound)
  498. assert.Error(t, err)
  499. _, _, err = GetFoldersQuotaScans(http.StatusOK)
  500. assert.Error(t, err)
  501. _, _, err = GetConnections(http.StatusOK)
  502. assert.Error(t, err)
  503. _, err = CloseConnection("non_existent_id", http.StatusNotFound)
  504. assert.Error(t, err)
  505. _, _, err = GetVersion(http.StatusOK)
  506. assert.Error(t, err)
  507. _, _, err = GetProviderStatus(http.StatusOK)
  508. assert.Error(t, err)
  509. _, _, err = Dumpdata("backup.json", "0", http.StatusOK)
  510. assert.Error(t, err)
  511. _, _, err = Loaddata("/tmp/backup.json", "", "", http.StatusOK)
  512. assert.Error(t, err)
  513. SetBaseURLAndCredentials(oldBaseURL, oldAuthUsername, oldAuthPassword)
  514. }
  515. func TestBasicAuth(t *testing.T) {
  516. oldAuthUsername := authUsername
  517. oldAuthPassword := authPassword
  518. authUserFile := filepath.Join(os.TempDir(), "http_users.txt")
  519. authUserData := []byte("test1:$2y$05$bcHSED7aO1cfLto6ZdDBOOKzlwftslVhtpIkRhAtSa4GuLmk5mola\n")
  520. err := ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  521. assert.NoError(t, err)
  522. httpAuth, _ = newBasicAuthProvider(authUserFile)
  523. _, _, err = GetVersion(http.StatusUnauthorized)
  524. assert.NoError(t, err)
  525. SetBaseURLAndCredentials(httpBaseURL, "test1", "password1")
  526. _, _, err = GetVersion(http.StatusOK)
  527. assert.NoError(t, err)
  528. SetBaseURLAndCredentials(httpBaseURL, "test1", "wrong_password")
  529. resp, _ := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(metricsPath), nil, "")
  530. defer resp.Body.Close()
  531. assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
  532. authUserData = append(authUserData, []byte("test2:$1$OtSSTL8b$bmaCqEksI1e7rnZSjsIDR1\n")...)
  533. err = ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  534. assert.NoError(t, err)
  535. SetBaseURLAndCredentials(httpBaseURL, "test2", "password2")
  536. _, _, err = GetVersion(http.StatusOK)
  537. assert.NoError(t, err)
  538. SetBaseURLAndCredentials(httpBaseURL, "test2", "wrong_password")
  539. _, _, err = GetVersion(http.StatusOK)
  540. assert.Error(t, err)
  541. authUserData = append(authUserData, []byte("test2:$apr1$gLnIkRIf$Xr/6aJfmIrihP4b2N2tcs/\n")...)
  542. err = ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  543. assert.NoError(t, err)
  544. SetBaseURLAndCredentials(httpBaseURL, "test2", "password2")
  545. _, _, err = GetVersion(http.StatusOK)
  546. assert.NoError(t, err)
  547. SetBaseURLAndCredentials(httpBaseURL, "test2", "wrong_password")
  548. _, _, err = GetVersion(http.StatusOK)
  549. assert.Error(t, err)
  550. authUserData = append(authUserData, []byte("test3:$apr1$gLnIkRIf$Xr/6$aJfmIr$ihP4b2N2tcs/\n")...)
  551. err = ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  552. assert.NoError(t, err)
  553. SetBaseURLAndCredentials(httpBaseURL, "test3", "wrong_password")
  554. _, _, err = GetVersion(http.StatusUnauthorized)
  555. assert.NoError(t, err)
  556. authUserData = append(authUserData, []byte("test4:$invalid$gLnIkRIf$Xr/6$aJfmIr$ihP4b2N2tcs/\n")...)
  557. err = ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  558. assert.NoError(t, err)
  559. SetBaseURLAndCredentials(httpBaseURL, "test3", "password2")
  560. _, _, err = GetVersion(http.StatusUnauthorized)
  561. assert.NoError(t, err)
  562. if runtime.GOOS != "windows" {
  563. authUserData = append(authUserData, []byte("test5:$apr1$gLnIkRIf$Xr/6aJfmIrihP4b2N2tcs/\n")...)
  564. err = ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  565. assert.NoError(t, err)
  566. err = os.Chmod(authUserFile, 0001)
  567. assert.NoError(t, err)
  568. SetBaseURLAndCredentials(httpBaseURL, "test5", "password2")
  569. _, _, err = GetVersion(http.StatusUnauthorized)
  570. assert.NoError(t, err)
  571. err = os.Chmod(authUserFile, os.ModePerm)
  572. assert.NoError(t, err)
  573. }
  574. authUserData = append(authUserData, []byte("\"foo\"bar\"\r\n")...)
  575. err = ioutil.WriteFile(authUserFile, authUserData, os.ModePerm)
  576. assert.NoError(t, err)
  577. SetBaseURLAndCredentials(httpBaseURL, "test2", "password2")
  578. _, _, err = GetVersion(http.StatusUnauthorized)
  579. assert.NoError(t, err)
  580. err = os.Remove(authUserFile)
  581. assert.NoError(t, err)
  582. SetBaseURLAndCredentials(httpBaseURL, oldAuthUsername, oldAuthPassword)
  583. httpAuth, _ = newBasicAuthProvider("")
  584. }
  585. func TestCloseConnectionHandler(t *testing.T) {
  586. req, _ := http.NewRequest(http.MethodDelete, activeConnectionsPath+"/connectionID", nil)
  587. rctx := chi.NewRouteContext()
  588. rctx.URLParams.Add("connectionID", "")
  589. req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  590. rr := httptest.NewRecorder()
  591. handleCloseConnection(rr, req)
  592. assert.Equal(t, http.StatusBadRequest, rr.Code)
  593. }
  594. func TestRenderInvalidTemplate(t *testing.T) {
  595. tmpl, err := template.New("test").Parse("{{.Count}}")
  596. if assert.NoError(t, err) {
  597. templates["no_match"] = tmpl
  598. rw := httptest.NewRecorder()
  599. renderTemplate(rw, "no_match", map[string]string{})
  600. assert.Equal(t, http.StatusInternalServerError, rw.Code)
  601. }
  602. }
  603. func TestQuotaScanInvalidFs(t *testing.T) {
  604. user := dataprovider.User{
  605. Username: "test",
  606. HomeDir: os.TempDir(),
  607. FsConfig: dataprovider.Filesystem{
  608. Provider: dataprovider.S3FilesystemProvider,
  609. },
  610. }
  611. common.QuotaScans.AddUserQuotaScan(user.Username)
  612. err := doQuotaScan(user)
  613. assert.Error(t, err)
  614. }