httpdtest.go 46 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390
  1. // Package httpdtest provides utilities for testing the exposed REST API.
  2. package httpdtest
  3. import (
  4. "bytes"
  5. "encoding/hex"
  6. "encoding/json"
  7. "errors"
  8. "fmt"
  9. "io"
  10. "net/http"
  11. "net/url"
  12. "path"
  13. "strconv"
  14. "strings"
  15. "github.com/go-chi/render"
  16. "github.com/drakkan/sftpgo/v2/common"
  17. "github.com/drakkan/sftpgo/v2/dataprovider"
  18. "github.com/drakkan/sftpgo/v2/httpclient"
  19. "github.com/drakkan/sftpgo/v2/httpd"
  20. "github.com/drakkan/sftpgo/v2/kms"
  21. "github.com/drakkan/sftpgo/v2/util"
  22. "github.com/drakkan/sftpgo/v2/version"
  23. "github.com/drakkan/sftpgo/v2/vfs"
  24. )
  25. const (
  26. tokenPath = "/api/v2/token"
  27. activeConnectionsPath = "/api/v2/connections"
  28. quotasBasePath = "/api/v2/quotas"
  29. quotaScanPath = "/api/v2/quotas/users/scans"
  30. quotaScanVFolderPath = "/api/v2/quotas/folders/scans"
  31. userPath = "/api/v2/users"
  32. versionPath = "/api/v2/version"
  33. folderPath = "/api/v2/folders"
  34. serverStatusPath = "/api/v2/status"
  35. dumpDataPath = "/api/v2/dumpdata"
  36. loadDataPath = "/api/v2/loaddata"
  37. defenderHosts = "/api/v2/defender/hosts"
  38. defenderBanTime = "/api/v2/defender/bantime"
  39. defenderUnban = "/api/v2/defender/unban"
  40. defenderScore = "/api/v2/defender/score"
  41. adminPath = "/api/v2/admins"
  42. adminPwdPath = "/api/v2/admin/changepwd"
  43. )
  44. const (
  45. defaultTokenAuthUser = "admin"
  46. defaultTokenAuthPass = "password"
  47. )
  48. var (
  49. httpBaseURL = "http://127.0.0.1:8080"
  50. jwtToken = ""
  51. )
  52. // SetBaseURL sets the base url to use for HTTP requests.
  53. // Default URL is "http://127.0.0.1:8080"
  54. func SetBaseURL(url string) {
  55. httpBaseURL = url
  56. }
  57. // SetJWTToken sets the JWT token to use
  58. func SetJWTToken(token string) {
  59. jwtToken = token
  60. }
  61. func sendHTTPRequest(method, url string, body io.Reader, contentType, token string) (*http.Response, error) {
  62. req, err := http.NewRequest(method, url, body)
  63. if err != nil {
  64. return nil, err
  65. }
  66. if contentType != "" {
  67. req.Header.Set("Content-Type", "application/json")
  68. }
  69. if token != "" {
  70. req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", token))
  71. }
  72. return httpclient.GetHTTPClient().Do(req)
  73. }
  74. func buildURLRelativeToBase(paths ...string) string {
  75. // we need to use path.Join and not filepath.Join
  76. // since filepath.Join will use backslash separator on Windows
  77. p := path.Join(paths...)
  78. return fmt.Sprintf("%s/%s", strings.TrimRight(httpBaseURL, "/"), strings.TrimLeft(p, "/"))
  79. }
  80. // GetToken tries to return a JWT token
  81. func GetToken(username, password string) (string, map[string]interface{}, error) {
  82. req, err := http.NewRequest(http.MethodGet, buildURLRelativeToBase(tokenPath), nil)
  83. if err != nil {
  84. return "", nil, err
  85. }
  86. req.SetBasicAuth(username, password)
  87. resp, err := httpclient.GetHTTPClient().Do(req)
  88. if err != nil {
  89. return "", nil, err
  90. }
  91. defer resp.Body.Close()
  92. err = checkResponse(resp.StatusCode, http.StatusOK)
  93. if err != nil {
  94. return "", nil, err
  95. }
  96. responseHolder := make(map[string]interface{})
  97. err = render.DecodeJSON(resp.Body, &responseHolder)
  98. if err != nil {
  99. return "", nil, err
  100. }
  101. return responseHolder["access_token"].(string), responseHolder, nil
  102. }
  103. func getDefaultToken() string {
  104. if jwtToken != "" {
  105. return jwtToken
  106. }
  107. token, _, err := GetToken(defaultTokenAuthUser, defaultTokenAuthPass)
  108. if err != nil {
  109. return ""
  110. }
  111. return token
  112. }
  113. // AddUser adds a new user and checks the received HTTP Status code against expectedStatusCode.
  114. func AddUser(user dataprovider.User, expectedStatusCode int) (dataprovider.User, []byte, error) {
  115. var newUser dataprovider.User
  116. var body []byte
  117. userAsJSON, _ := json.Marshal(user)
  118. resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(userPath), bytes.NewBuffer(userAsJSON),
  119. "application/json", getDefaultToken())
  120. if err != nil {
  121. return newUser, body, err
  122. }
  123. defer resp.Body.Close()
  124. err = checkResponse(resp.StatusCode, expectedStatusCode)
  125. if expectedStatusCode != http.StatusCreated {
  126. body, _ = getResponseBody(resp)
  127. return newUser, body, err
  128. }
  129. if err == nil {
  130. err = render.DecodeJSON(resp.Body, &newUser)
  131. } else {
  132. body, _ = getResponseBody(resp)
  133. }
  134. if err == nil {
  135. err = checkUser(&user, &newUser)
  136. }
  137. return newUser, body, err
  138. }
  139. // UpdateUserWithJSON update a user using the provided JSON as POST body
  140. func UpdateUserWithJSON(user dataprovider.User, expectedStatusCode int, disconnect string, userAsJSON []byte) (dataprovider.User, []byte, error) {
  141. var newUser dataprovider.User
  142. var body []byte
  143. url, err := addDisconnectQueryParam(buildURLRelativeToBase(userPath, url.PathEscape(user.Username)), disconnect)
  144. if err != nil {
  145. return user, body, err
  146. }
  147. resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "application/json",
  148. getDefaultToken())
  149. if err != nil {
  150. return user, body, err
  151. }
  152. defer resp.Body.Close()
  153. body, _ = getResponseBody(resp)
  154. err = checkResponse(resp.StatusCode, expectedStatusCode)
  155. if expectedStatusCode != http.StatusOK {
  156. return newUser, body, err
  157. }
  158. if err == nil {
  159. newUser, body, err = GetUserByUsername(user.Username, expectedStatusCode)
  160. }
  161. if err == nil {
  162. err = checkUser(&user, &newUser)
  163. }
  164. return newUser, body, err
  165. }
  166. // UpdateUser updates an existing user and checks the received HTTP Status code against expectedStatusCode.
  167. func UpdateUser(user dataprovider.User, expectedStatusCode int, disconnect string) (dataprovider.User, []byte, error) {
  168. userAsJSON, _ := json.Marshal(user)
  169. return UpdateUserWithJSON(user, expectedStatusCode, disconnect, userAsJSON)
  170. }
  171. // RemoveUser removes an existing user and checks the received HTTP Status code against expectedStatusCode.
  172. func RemoveUser(user dataprovider.User, expectedStatusCode int) ([]byte, error) {
  173. var body []byte
  174. resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(userPath, url.PathEscape(user.Username)),
  175. nil, "", getDefaultToken())
  176. if err != nil {
  177. return body, err
  178. }
  179. defer resp.Body.Close()
  180. body, _ = getResponseBody(resp)
  181. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  182. }
  183. // GetUserByUsername gets a user by username and checks the received HTTP Status code against expectedStatusCode.
  184. func GetUserByUsername(username string, expectedStatusCode int) (dataprovider.User, []byte, error) {
  185. var user dataprovider.User
  186. var body []byte
  187. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(userPath, url.PathEscape(username)),
  188. nil, "", getDefaultToken())
  189. if err != nil {
  190. return user, body, err
  191. }
  192. defer resp.Body.Close()
  193. err = checkResponse(resp.StatusCode, expectedStatusCode)
  194. if err == nil && expectedStatusCode == http.StatusOK {
  195. err = render.DecodeJSON(resp.Body, &user)
  196. } else {
  197. body, _ = getResponseBody(resp)
  198. }
  199. return user, body, err
  200. }
  201. // GetUsers returns a list of users and checks the received HTTP Status code against expectedStatusCode.
  202. // The number of results can be limited specifying a limit.
  203. // Some results can be skipped specifying an offset.
  204. func GetUsers(limit, offset int64, expectedStatusCode int) ([]dataprovider.User, []byte, error) {
  205. var users []dataprovider.User
  206. var body []byte
  207. url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(userPath), limit, offset)
  208. if err != nil {
  209. return users, body, err
  210. }
  211. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  212. if err != nil {
  213. return users, body, err
  214. }
  215. defer resp.Body.Close()
  216. err = checkResponse(resp.StatusCode, expectedStatusCode)
  217. if err == nil && expectedStatusCode == http.StatusOK {
  218. err = render.DecodeJSON(resp.Body, &users)
  219. } else {
  220. body, _ = getResponseBody(resp)
  221. }
  222. return users, body, err
  223. }
  224. // AddAdmin adds a new user and checks the received HTTP Status code against expectedStatusCode.
  225. func AddAdmin(admin dataprovider.Admin, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
  226. var newAdmin dataprovider.Admin
  227. var body []byte
  228. asJSON, _ := json.Marshal(admin)
  229. resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(adminPath), bytes.NewBuffer(asJSON),
  230. "application/json", getDefaultToken())
  231. if err != nil {
  232. return newAdmin, body, err
  233. }
  234. defer resp.Body.Close()
  235. err = checkResponse(resp.StatusCode, expectedStatusCode)
  236. if expectedStatusCode != http.StatusCreated {
  237. body, _ = getResponseBody(resp)
  238. return newAdmin, body, err
  239. }
  240. if err == nil {
  241. err = render.DecodeJSON(resp.Body, &newAdmin)
  242. } else {
  243. body, _ = getResponseBody(resp)
  244. }
  245. if err == nil {
  246. err = checkAdmin(&admin, &newAdmin)
  247. }
  248. return newAdmin, body, err
  249. }
  250. // UpdateAdmin updates an existing admin and checks the received HTTP Status code against expectedStatusCode
  251. func UpdateAdmin(admin dataprovider.Admin, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
  252. var newAdmin dataprovider.Admin
  253. var body []byte
  254. asJSON, _ := json.Marshal(admin)
  255. resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(adminPath, url.PathEscape(admin.Username)),
  256. bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
  257. if err != nil {
  258. return newAdmin, body, err
  259. }
  260. defer resp.Body.Close()
  261. body, _ = getResponseBody(resp)
  262. err = checkResponse(resp.StatusCode, expectedStatusCode)
  263. if expectedStatusCode != http.StatusOK {
  264. return newAdmin, body, err
  265. }
  266. if err == nil {
  267. newAdmin, body, err = GetAdminByUsername(admin.Username, expectedStatusCode)
  268. }
  269. if err == nil {
  270. err = checkAdmin(&admin, &newAdmin)
  271. }
  272. return newAdmin, body, err
  273. }
  274. // RemoveAdmin removes an existing admin and checks the received HTTP Status code against expectedStatusCode.
  275. func RemoveAdmin(admin dataprovider.Admin, expectedStatusCode int) ([]byte, error) {
  276. var body []byte
  277. resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(adminPath, url.PathEscape(admin.Username)),
  278. nil, "", getDefaultToken())
  279. if err != nil {
  280. return body, err
  281. }
  282. defer resp.Body.Close()
  283. body, _ = getResponseBody(resp)
  284. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  285. }
  286. // GetAdminByUsername gets an admin by username and checks the received HTTP Status code against expectedStatusCode.
  287. func GetAdminByUsername(username string, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
  288. var admin dataprovider.Admin
  289. var body []byte
  290. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(adminPath, url.PathEscape(username)),
  291. nil, "", getDefaultToken())
  292. if err != nil {
  293. return admin, body, err
  294. }
  295. defer resp.Body.Close()
  296. err = checkResponse(resp.StatusCode, expectedStatusCode)
  297. if err == nil && expectedStatusCode == http.StatusOK {
  298. err = render.DecodeJSON(resp.Body, &admin)
  299. } else {
  300. body, _ = getResponseBody(resp)
  301. }
  302. return admin, body, err
  303. }
  304. // GetAdmins returns a list of admins and checks the received HTTP Status code against expectedStatusCode.
  305. // The number of results can be limited specifying a limit.
  306. // Some results can be skipped specifying an offset.
  307. func GetAdmins(limit, offset int64, expectedStatusCode int) ([]dataprovider.Admin, []byte, error) {
  308. var admins []dataprovider.Admin
  309. var body []byte
  310. url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(adminPath), limit, offset)
  311. if err != nil {
  312. return admins, body, err
  313. }
  314. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  315. if err != nil {
  316. return admins, body, err
  317. }
  318. defer resp.Body.Close()
  319. err = checkResponse(resp.StatusCode, expectedStatusCode)
  320. if err == nil && expectedStatusCode == http.StatusOK {
  321. err = render.DecodeJSON(resp.Body, &admins)
  322. } else {
  323. body, _ = getResponseBody(resp)
  324. }
  325. return admins, body, err
  326. }
  327. // ChangeAdminPassword changes the password for an existing admin
  328. func ChangeAdminPassword(currentPassword, newPassword string, expectedStatusCode int) ([]byte, error) {
  329. var body []byte
  330. pwdChange := make(map[string]string)
  331. pwdChange["current_password"] = currentPassword
  332. pwdChange["new_password"] = newPassword
  333. asJSON, _ := json.Marshal(&pwdChange)
  334. resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(adminPwdPath),
  335. bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
  336. if err != nil {
  337. return body, err
  338. }
  339. defer resp.Body.Close()
  340. err = checkResponse(resp.StatusCode, expectedStatusCode)
  341. body, _ = getResponseBody(resp)
  342. return body, err
  343. }
  344. // GetQuotaScans gets active quota scans for users and checks the received HTTP Status code against expectedStatusCode.
  345. func GetQuotaScans(expectedStatusCode int) ([]common.ActiveQuotaScan, []byte, error) {
  346. var quotaScans []common.ActiveQuotaScan
  347. var body []byte
  348. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(quotaScanPath), nil, "", getDefaultToken())
  349. if err != nil {
  350. return quotaScans, body, err
  351. }
  352. defer resp.Body.Close()
  353. err = checkResponse(resp.StatusCode, expectedStatusCode)
  354. if err == nil && expectedStatusCode == http.StatusOK {
  355. err = render.DecodeJSON(resp.Body, &quotaScans)
  356. } else {
  357. body, _ = getResponseBody(resp)
  358. }
  359. return quotaScans, body, err
  360. }
  361. // StartQuotaScan starts a new quota scan for the given user and checks the received HTTP Status code against expectedStatusCode.
  362. func StartQuotaScan(user dataprovider.User, expectedStatusCode int) ([]byte, error) {
  363. var body []byte
  364. resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(quotasBasePath, "users", user.Username, "scan"),
  365. nil, "", getDefaultToken())
  366. if err != nil {
  367. return body, err
  368. }
  369. defer resp.Body.Close()
  370. body, _ = getResponseBody(resp)
  371. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  372. }
  373. // UpdateQuotaUsage updates the user used quota limits and checks the received HTTP Status code against expectedStatusCode.
  374. func UpdateQuotaUsage(user dataprovider.User, mode string, expectedStatusCode int) ([]byte, error) {
  375. var body []byte
  376. userAsJSON, _ := json.Marshal(user)
  377. url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "users", user.Username, "usage"), mode)
  378. if err != nil {
  379. return body, err
  380. }
  381. resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "", getDefaultToken())
  382. if err != nil {
  383. return body, err
  384. }
  385. defer resp.Body.Close()
  386. body, _ = getResponseBody(resp)
  387. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  388. }
  389. // GetConnections returns status and stats for active SFTP/SCP connections
  390. func GetConnections(expectedStatusCode int) ([]common.ConnectionStatus, []byte, error) {
  391. var connections []common.ConnectionStatus
  392. var body []byte
  393. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(activeConnectionsPath), nil, "", getDefaultToken())
  394. if err != nil {
  395. return connections, body, err
  396. }
  397. defer resp.Body.Close()
  398. err = checkResponse(resp.StatusCode, expectedStatusCode)
  399. if err == nil && expectedStatusCode == http.StatusOK {
  400. err = render.DecodeJSON(resp.Body, &connections)
  401. } else {
  402. body, _ = getResponseBody(resp)
  403. }
  404. return connections, body, err
  405. }
  406. // CloseConnection closes an active connection identified by connectionID
  407. func CloseConnection(connectionID string, expectedStatusCode int) ([]byte, error) {
  408. var body []byte
  409. resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(activeConnectionsPath, connectionID),
  410. nil, "", getDefaultToken())
  411. if err != nil {
  412. return body, err
  413. }
  414. defer resp.Body.Close()
  415. err = checkResponse(resp.StatusCode, expectedStatusCode)
  416. body, _ = getResponseBody(resp)
  417. return body, err
  418. }
  419. // AddFolder adds a new folder and checks the received HTTP Status code against expectedStatusCode
  420. func AddFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
  421. var newFolder vfs.BaseVirtualFolder
  422. var body []byte
  423. folderAsJSON, _ := json.Marshal(folder)
  424. resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(folderPath), bytes.NewBuffer(folderAsJSON),
  425. "application/json", getDefaultToken())
  426. if err != nil {
  427. return newFolder, body, err
  428. }
  429. defer resp.Body.Close()
  430. err = checkResponse(resp.StatusCode, expectedStatusCode)
  431. if expectedStatusCode != http.StatusCreated {
  432. body, _ = getResponseBody(resp)
  433. return newFolder, body, err
  434. }
  435. if err == nil {
  436. err = render.DecodeJSON(resp.Body, &newFolder)
  437. } else {
  438. body, _ = getResponseBody(resp)
  439. }
  440. if err == nil {
  441. err = checkFolder(&folder, &newFolder)
  442. }
  443. return newFolder, body, err
  444. }
  445. // UpdateFolder updates an existing folder and checks the received HTTP Status code against expectedStatusCode.
  446. func UpdateFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
  447. var updatedFolder vfs.BaseVirtualFolder
  448. var body []byte
  449. folderAsJSON, _ := json.Marshal(folder)
  450. resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(folderPath, url.PathEscape(folder.Name)),
  451. bytes.NewBuffer(folderAsJSON), "application/json", getDefaultToken())
  452. if err != nil {
  453. return updatedFolder, body, err
  454. }
  455. defer resp.Body.Close()
  456. body, _ = getResponseBody(resp)
  457. err = checkResponse(resp.StatusCode, expectedStatusCode)
  458. if expectedStatusCode != http.StatusOK {
  459. return updatedFolder, body, err
  460. }
  461. if err == nil {
  462. updatedFolder, body, err = GetFolderByName(folder.Name, expectedStatusCode)
  463. }
  464. if err == nil {
  465. err = checkFolder(&folder, &updatedFolder)
  466. }
  467. return updatedFolder, body, err
  468. }
  469. // RemoveFolder removes an existing user and checks the received HTTP Status code against expectedStatusCode.
  470. func RemoveFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) ([]byte, error) {
  471. var body []byte
  472. resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(folderPath, url.PathEscape(folder.Name)),
  473. nil, "", getDefaultToken())
  474. if err != nil {
  475. return body, err
  476. }
  477. defer resp.Body.Close()
  478. body, _ = getResponseBody(resp)
  479. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  480. }
  481. // GetFolderByName gets a folder by name and checks the received HTTP Status code against expectedStatusCode.
  482. func GetFolderByName(name string, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
  483. var folder vfs.BaseVirtualFolder
  484. var body []byte
  485. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(folderPath, url.PathEscape(name)),
  486. nil, "", getDefaultToken())
  487. if err != nil {
  488. return folder, body, err
  489. }
  490. defer resp.Body.Close()
  491. err = checkResponse(resp.StatusCode, expectedStatusCode)
  492. if err == nil && expectedStatusCode == http.StatusOK {
  493. err = render.DecodeJSON(resp.Body, &folder)
  494. } else {
  495. body, _ = getResponseBody(resp)
  496. }
  497. return folder, body, err
  498. }
  499. // GetFolders returns a list of folders and checks the received HTTP Status code against expectedStatusCode.
  500. // The number of results can be limited specifying a limit.
  501. // Some results can be skipped specifying an offset.
  502. // The results can be filtered specifying a folder path, the folder path filter is an exact match
  503. func GetFolders(limit int64, offset int64, expectedStatusCode int) ([]vfs.BaseVirtualFolder, []byte, error) {
  504. var folders []vfs.BaseVirtualFolder
  505. var body []byte
  506. url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(folderPath), limit, offset)
  507. if err != nil {
  508. return folders, body, err
  509. }
  510. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  511. if err != nil {
  512. return folders, body, err
  513. }
  514. defer resp.Body.Close()
  515. err = checkResponse(resp.StatusCode, expectedStatusCode)
  516. if err == nil && expectedStatusCode == http.StatusOK {
  517. err = render.DecodeJSON(resp.Body, &folders)
  518. } else {
  519. body, _ = getResponseBody(resp)
  520. }
  521. return folders, body, err
  522. }
  523. // GetFoldersQuotaScans gets active quota scans for folders and checks the received HTTP Status code against expectedStatusCode.
  524. func GetFoldersQuotaScans(expectedStatusCode int) ([]common.ActiveVirtualFolderQuotaScan, []byte, error) {
  525. var quotaScans []common.ActiveVirtualFolderQuotaScan
  526. var body []byte
  527. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(quotaScanVFolderPath), nil, "", getDefaultToken())
  528. if err != nil {
  529. return quotaScans, body, err
  530. }
  531. defer resp.Body.Close()
  532. err = checkResponse(resp.StatusCode, expectedStatusCode)
  533. if err == nil && expectedStatusCode == http.StatusOK {
  534. err = render.DecodeJSON(resp.Body, &quotaScans)
  535. } else {
  536. body, _ = getResponseBody(resp)
  537. }
  538. return quotaScans, body, err
  539. }
  540. // StartFolderQuotaScan start a new quota scan for the given folder and checks the received HTTP Status code against expectedStatusCode.
  541. func StartFolderQuotaScan(folder vfs.BaseVirtualFolder, expectedStatusCode int) ([]byte, error) {
  542. var body []byte
  543. resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(quotasBasePath, "folders", folder.Name, "scan"),
  544. nil, "", getDefaultToken())
  545. if err != nil {
  546. return body, err
  547. }
  548. defer resp.Body.Close()
  549. body, _ = getResponseBody(resp)
  550. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  551. }
  552. // UpdateFolderQuotaUsage updates the folder used quota limits and checks the received HTTP Status code against expectedStatusCode.
  553. func UpdateFolderQuotaUsage(folder vfs.BaseVirtualFolder, mode string, expectedStatusCode int) ([]byte, error) {
  554. var body []byte
  555. folderAsJSON, _ := json.Marshal(folder)
  556. url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "folders", folder.Name, "usage"), mode)
  557. if err != nil {
  558. return body, err
  559. }
  560. resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(folderAsJSON), "", getDefaultToken())
  561. if err != nil {
  562. return body, err
  563. }
  564. defer resp.Body.Close()
  565. body, _ = getResponseBody(resp)
  566. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  567. }
  568. // GetVersion returns version details
  569. func GetVersion(expectedStatusCode int) (version.Info, []byte, error) {
  570. var appVersion version.Info
  571. var body []byte
  572. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(versionPath), nil, "", getDefaultToken())
  573. if err != nil {
  574. return appVersion, body, err
  575. }
  576. defer resp.Body.Close()
  577. err = checkResponse(resp.StatusCode, expectedStatusCode)
  578. if err == nil && expectedStatusCode == http.StatusOK {
  579. err = render.DecodeJSON(resp.Body, &appVersion)
  580. } else {
  581. body, _ = getResponseBody(resp)
  582. }
  583. return appVersion, body, err
  584. }
  585. // GetStatus returns the server status
  586. func GetStatus(expectedStatusCode int) (httpd.ServicesStatus, []byte, error) {
  587. var response httpd.ServicesStatus
  588. var body []byte
  589. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(serverStatusPath), nil, "", getDefaultToken())
  590. if err != nil {
  591. return response, body, err
  592. }
  593. defer resp.Body.Close()
  594. err = checkResponse(resp.StatusCode, expectedStatusCode)
  595. if err == nil && (expectedStatusCode == http.StatusOK) {
  596. err = render.DecodeJSON(resp.Body, &response)
  597. } else {
  598. body, _ = getResponseBody(resp)
  599. }
  600. return response, body, err
  601. }
  602. // GetDefenderHosts returns hosts that are banned or for which some violations have been detected
  603. func GetDefenderHosts(expectedStatusCode int) ([]common.DefenderEntry, []byte, error) {
  604. var response []common.DefenderEntry
  605. var body []byte
  606. url, err := url.Parse(buildURLRelativeToBase(defenderHosts))
  607. if err != nil {
  608. return response, body, err
  609. }
  610. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  611. if err != nil {
  612. return response, body, err
  613. }
  614. defer resp.Body.Close()
  615. err = checkResponse(resp.StatusCode, expectedStatusCode)
  616. if err == nil && expectedStatusCode == http.StatusOK {
  617. err = render.DecodeJSON(resp.Body, &response)
  618. } else {
  619. body, _ = getResponseBody(resp)
  620. }
  621. return response, body, err
  622. }
  623. // GetDefenderHostByIP returns the host with the given IP, if it exists
  624. func GetDefenderHostByIP(ip string, expectedStatusCode int) (common.DefenderEntry, []byte, error) {
  625. var host common.DefenderEntry
  626. var body []byte
  627. id := hex.EncodeToString([]byte(ip))
  628. resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(defenderHosts, id),
  629. nil, "", getDefaultToken())
  630. if err != nil {
  631. return host, body, err
  632. }
  633. defer resp.Body.Close()
  634. err = checkResponse(resp.StatusCode, expectedStatusCode)
  635. if err == nil && expectedStatusCode == http.StatusOK {
  636. err = render.DecodeJSON(resp.Body, &host)
  637. } else {
  638. body, _ = getResponseBody(resp)
  639. }
  640. return host, body, err
  641. }
  642. // RemoveDefenderHostByIP removes the host with the given IP from the defender list
  643. func RemoveDefenderHostByIP(ip string, expectedStatusCode int) ([]byte, error) {
  644. var body []byte
  645. id := hex.EncodeToString([]byte(ip))
  646. resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(defenderHosts, id), nil, "", getDefaultToken())
  647. if err != nil {
  648. return body, err
  649. }
  650. defer resp.Body.Close()
  651. body, _ = getResponseBody(resp)
  652. return body, checkResponse(resp.StatusCode, expectedStatusCode)
  653. }
  654. // GetBanTime returns the ban time for the given IP address
  655. func GetBanTime(ip string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
  656. var response map[string]interface{}
  657. var body []byte
  658. url, err := url.Parse(buildURLRelativeToBase(defenderBanTime))
  659. if err != nil {
  660. return response, body, err
  661. }
  662. q := url.Query()
  663. q.Add("ip", ip)
  664. url.RawQuery = q.Encode()
  665. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  666. if err != nil {
  667. return response, body, err
  668. }
  669. defer resp.Body.Close()
  670. err = checkResponse(resp.StatusCode, expectedStatusCode)
  671. if err == nil && expectedStatusCode == http.StatusOK {
  672. err = render.DecodeJSON(resp.Body, &response)
  673. } else {
  674. body, _ = getResponseBody(resp)
  675. }
  676. return response, body, err
  677. }
  678. // GetScore returns the score for the given IP address
  679. func GetScore(ip string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
  680. var response map[string]interface{}
  681. var body []byte
  682. url, err := url.Parse(buildURLRelativeToBase(defenderScore))
  683. if err != nil {
  684. return response, body, err
  685. }
  686. q := url.Query()
  687. q.Add("ip", ip)
  688. url.RawQuery = q.Encode()
  689. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  690. if err != nil {
  691. return response, body, err
  692. }
  693. defer resp.Body.Close()
  694. err = checkResponse(resp.StatusCode, expectedStatusCode)
  695. if err == nil && expectedStatusCode == http.StatusOK {
  696. err = render.DecodeJSON(resp.Body, &response)
  697. } else {
  698. body, _ = getResponseBody(resp)
  699. }
  700. return response, body, err
  701. }
  702. // UnbanIP unbans the given IP address
  703. func UnbanIP(ip string, expectedStatusCode int) error {
  704. postBody := make(map[string]string)
  705. postBody["ip"] = ip
  706. asJSON, _ := json.Marshal(postBody)
  707. resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(defenderUnban), bytes.NewBuffer(asJSON),
  708. "", getDefaultToken())
  709. if err != nil {
  710. return err
  711. }
  712. defer resp.Body.Close()
  713. return checkResponse(resp.StatusCode, expectedStatusCode)
  714. }
  715. // Dumpdata requests a backup to outputFile.
  716. // outputFile is relative to the configured backups_path
  717. func Dumpdata(outputFile, outputData, indent string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
  718. var response map[string]interface{}
  719. var body []byte
  720. url, err := url.Parse(buildURLRelativeToBase(dumpDataPath))
  721. if err != nil {
  722. return response, body, err
  723. }
  724. q := url.Query()
  725. if outputData != "" {
  726. q.Add("output-data", outputData)
  727. }
  728. if outputFile != "" {
  729. q.Add("output-file", outputFile)
  730. }
  731. if indent != "" {
  732. q.Add("indent", indent)
  733. }
  734. url.RawQuery = q.Encode()
  735. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  736. if err != nil {
  737. return response, body, err
  738. }
  739. defer resp.Body.Close()
  740. err = checkResponse(resp.StatusCode, expectedStatusCode)
  741. if err == nil && expectedStatusCode == http.StatusOK {
  742. err = render.DecodeJSON(resp.Body, &response)
  743. } else {
  744. body, _ = getResponseBody(resp)
  745. }
  746. return response, body, err
  747. }
  748. // Loaddata restores a backup.
  749. func Loaddata(inputFile, scanQuota, mode string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
  750. var response map[string]interface{}
  751. var body []byte
  752. url, err := url.Parse(buildURLRelativeToBase(loadDataPath))
  753. if err != nil {
  754. return response, body, err
  755. }
  756. q := url.Query()
  757. q.Add("input-file", inputFile)
  758. if scanQuota != "" {
  759. q.Add("scan-quota", scanQuota)
  760. }
  761. if mode != "" {
  762. q.Add("mode", mode)
  763. }
  764. url.RawQuery = q.Encode()
  765. resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
  766. if err != nil {
  767. return response, body, err
  768. }
  769. defer resp.Body.Close()
  770. err = checkResponse(resp.StatusCode, expectedStatusCode)
  771. if err == nil && expectedStatusCode == http.StatusOK {
  772. err = render.DecodeJSON(resp.Body, &response)
  773. } else {
  774. body, _ = getResponseBody(resp)
  775. }
  776. return response, body, err
  777. }
  778. // LoaddataFromPostBody restores a backup
  779. func LoaddataFromPostBody(data []byte, scanQuota, mode string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
  780. var response map[string]interface{}
  781. var body []byte
  782. url, err := url.Parse(buildURLRelativeToBase(loadDataPath))
  783. if err != nil {
  784. return response, body, err
  785. }
  786. q := url.Query()
  787. if scanQuota != "" {
  788. q.Add("scan-quota", scanQuota)
  789. }
  790. if mode != "" {
  791. q.Add("mode", mode)
  792. }
  793. url.RawQuery = q.Encode()
  794. resp, err := sendHTTPRequest(http.MethodPost, url.String(), bytes.NewReader(data), "", getDefaultToken())
  795. if err != nil {
  796. return response, body, err
  797. }
  798. defer resp.Body.Close()
  799. err = checkResponse(resp.StatusCode, expectedStatusCode)
  800. if err == nil && expectedStatusCode == http.StatusOK {
  801. err = render.DecodeJSON(resp.Body, &response)
  802. } else {
  803. body, _ = getResponseBody(resp)
  804. }
  805. return response, body, err
  806. }
  807. func checkResponse(actual int, expected int) error {
  808. if expected != actual {
  809. return fmt.Errorf("wrong status code: got %v want %v", actual, expected)
  810. }
  811. return nil
  812. }
  813. func getResponseBody(resp *http.Response) ([]byte, error) {
  814. return io.ReadAll(resp.Body)
  815. }
  816. func checkFolder(expected *vfs.BaseVirtualFolder, actual *vfs.BaseVirtualFolder) error {
  817. if expected.ID <= 0 {
  818. if actual.ID <= 0 {
  819. return errors.New("actual folder ID must be > 0")
  820. }
  821. } else {
  822. if actual.ID != expected.ID {
  823. return errors.New("folder ID mismatch")
  824. }
  825. }
  826. if expected.Name != actual.Name {
  827. return errors.New("name mismatch")
  828. }
  829. if expected.MappedPath != actual.MappedPath {
  830. return errors.New("mapped path mismatch")
  831. }
  832. if expected.Description != actual.Description {
  833. return errors.New("description mismatch")
  834. }
  835. return compareFsConfig(&expected.FsConfig, &actual.FsConfig)
  836. }
  837. func checkAdmin(expected *dataprovider.Admin, actual *dataprovider.Admin) error {
  838. if actual.Password != "" {
  839. return errors.New("admin password must not be visible")
  840. }
  841. if expected.ID <= 0 {
  842. if actual.ID <= 0 {
  843. return errors.New("actual admin ID must be > 0")
  844. }
  845. } else {
  846. if actual.ID != expected.ID {
  847. return errors.New("admin ID mismatch")
  848. }
  849. }
  850. if err := compareAdminEqualFields(expected, actual); err != nil {
  851. return err
  852. }
  853. if len(expected.Permissions) != len(actual.Permissions) {
  854. return errors.New("permissions mismatch")
  855. }
  856. for _, p := range expected.Permissions {
  857. if !util.IsStringInSlice(p, actual.Permissions) {
  858. return errors.New("permissions content mismatch")
  859. }
  860. }
  861. if len(expected.Filters.AllowList) != len(actual.Filters.AllowList) {
  862. return errors.New("allow list mismatch")
  863. }
  864. for _, v := range expected.Filters.AllowList {
  865. if !util.IsStringInSlice(v, actual.Filters.AllowList) {
  866. return errors.New("allow list content mismatch")
  867. }
  868. }
  869. return nil
  870. }
  871. func compareAdminEqualFields(expected *dataprovider.Admin, actual *dataprovider.Admin) error {
  872. if expected.Username != actual.Username {
  873. return errors.New("sername mismatch")
  874. }
  875. if expected.Email != actual.Email {
  876. return errors.New("email mismatch")
  877. }
  878. if expected.Status != actual.Status {
  879. return errors.New("status mismatch")
  880. }
  881. if expected.Description != actual.Description {
  882. return errors.New("description mismatch")
  883. }
  884. if expected.AdditionalInfo != actual.AdditionalInfo {
  885. return errors.New("additional info mismatch")
  886. }
  887. return nil
  888. }
  889. func checkUser(expected *dataprovider.User, actual *dataprovider.User) error {
  890. if actual.Password != "" {
  891. return errors.New("user password must not be visible")
  892. }
  893. if expected.ID <= 0 {
  894. if actual.ID <= 0 {
  895. return errors.New("actual user ID must be > 0")
  896. }
  897. } else {
  898. if actual.ID != expected.ID {
  899. return errors.New("user ID mismatch")
  900. }
  901. }
  902. if len(expected.Permissions) != len(actual.Permissions) {
  903. return errors.New("permissions mismatch")
  904. }
  905. for dir, perms := range expected.Permissions {
  906. if actualPerms, ok := actual.Permissions[dir]; ok {
  907. for _, v := range actualPerms {
  908. if !util.IsStringInSlice(v, perms) {
  909. return errors.New("permissions contents mismatch")
  910. }
  911. }
  912. } else {
  913. return errors.New("permissions directories mismatch")
  914. }
  915. }
  916. if err := compareUserFilters(expected, actual); err != nil {
  917. return err
  918. }
  919. if err := compareFsConfig(&expected.FsConfig, &actual.FsConfig); err != nil {
  920. return err
  921. }
  922. if err := compareUserVirtualFolders(expected, actual); err != nil {
  923. return err
  924. }
  925. return compareEqualsUserFields(expected, actual)
  926. }
  927. func compareUserVirtualFolders(expected *dataprovider.User, actual *dataprovider.User) error {
  928. if len(actual.VirtualFolders) != len(expected.VirtualFolders) {
  929. return errors.New("virtual folders len mismatch")
  930. }
  931. for _, v := range actual.VirtualFolders {
  932. found := false
  933. for _, v1 := range expected.VirtualFolders {
  934. if path.Clean(v.VirtualPath) == path.Clean(v1.VirtualPath) {
  935. if err := checkFolder(&v1.BaseVirtualFolder, &v.BaseVirtualFolder); err != nil {
  936. return err
  937. }
  938. if v.QuotaSize != v1.QuotaSize {
  939. return errors.New("vfolder quota size mismatch")
  940. }
  941. if (v.QuotaFiles) != (v1.QuotaFiles) {
  942. return errors.New("vfolder quota files mismatch")
  943. }
  944. found = true
  945. break
  946. }
  947. }
  948. if !found {
  949. return errors.New("virtual folders mismatch")
  950. }
  951. }
  952. return nil
  953. }
  954. func compareFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
  955. if expected.Provider != actual.Provider {
  956. return errors.New("fs provider mismatch")
  957. }
  958. if err := compareS3Config(expected, actual); err != nil {
  959. return err
  960. }
  961. if err := compareGCSConfig(expected, actual); err != nil {
  962. return err
  963. }
  964. if err := compareAzBlobConfig(expected, actual); err != nil {
  965. return err
  966. }
  967. if err := checkEncryptedSecret(expected.CryptConfig.Passphrase, actual.CryptConfig.Passphrase); err != nil {
  968. return err
  969. }
  970. return compareSFTPFsConfig(expected, actual)
  971. }
  972. func compareS3Config(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
  973. if expected.S3Config.Bucket != actual.S3Config.Bucket {
  974. return errors.New("fs S3 bucket mismatch")
  975. }
  976. if expected.S3Config.Region != actual.S3Config.Region {
  977. return errors.New("fs S3 region mismatch")
  978. }
  979. if expected.S3Config.AccessKey != actual.S3Config.AccessKey {
  980. return errors.New("fs S3 access key mismatch")
  981. }
  982. if err := checkEncryptedSecret(expected.S3Config.AccessSecret, actual.S3Config.AccessSecret); err != nil {
  983. return fmt.Errorf("fs S3 access secret mismatch: %v", err)
  984. }
  985. if expected.S3Config.Endpoint != actual.S3Config.Endpoint {
  986. return errors.New("fs S3 endpoint mismatch")
  987. }
  988. if expected.S3Config.StorageClass != actual.S3Config.StorageClass {
  989. return errors.New("fs S3 storage class mismatch")
  990. }
  991. if expected.S3Config.UploadPartSize != actual.S3Config.UploadPartSize {
  992. return errors.New("fs S3 upload part size mismatch")
  993. }
  994. if expected.S3Config.DownloadPartMaxTime != actual.S3Config.DownloadPartMaxTime {
  995. return errors.New("fs S3 download part max time mismatch")
  996. }
  997. if expected.S3Config.UploadConcurrency != actual.S3Config.UploadConcurrency {
  998. return errors.New("fs S3 upload concurrency mismatch")
  999. }
  1000. if expected.S3Config.KeyPrefix != actual.S3Config.KeyPrefix &&
  1001. expected.S3Config.KeyPrefix+"/" != actual.S3Config.KeyPrefix {
  1002. return errors.New("fs S3 key prefix mismatch")
  1003. }
  1004. return nil
  1005. }
  1006. func compareGCSConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
  1007. if expected.GCSConfig.Bucket != actual.GCSConfig.Bucket {
  1008. return errors.New("GCS bucket mismatch")
  1009. }
  1010. if expected.GCSConfig.StorageClass != actual.GCSConfig.StorageClass {
  1011. return errors.New("GCS storage class mismatch")
  1012. }
  1013. if expected.GCSConfig.KeyPrefix != actual.GCSConfig.KeyPrefix &&
  1014. expected.GCSConfig.KeyPrefix+"/" != actual.GCSConfig.KeyPrefix {
  1015. return errors.New("GCS key prefix mismatch")
  1016. }
  1017. if expected.GCSConfig.AutomaticCredentials != actual.GCSConfig.AutomaticCredentials {
  1018. return errors.New("GCS automatic credentials mismatch")
  1019. }
  1020. return nil
  1021. }
  1022. func compareSFTPFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
  1023. if expected.SFTPConfig.Endpoint != actual.SFTPConfig.Endpoint {
  1024. return errors.New("SFTPFs endpoint mismatch")
  1025. }
  1026. if expected.SFTPConfig.Username != actual.SFTPConfig.Username {
  1027. return errors.New("SFTPFs username mismatch")
  1028. }
  1029. if expected.SFTPConfig.DisableCouncurrentReads != actual.SFTPConfig.DisableCouncurrentReads {
  1030. return errors.New("SFTPFs disable_concurrent_reads mismatch")
  1031. }
  1032. if expected.SFTPConfig.BufferSize != actual.SFTPConfig.BufferSize {
  1033. return errors.New("SFTPFs buffer_size mismatch")
  1034. }
  1035. if err := checkEncryptedSecret(expected.SFTPConfig.Password, actual.SFTPConfig.Password); err != nil {
  1036. return fmt.Errorf("SFTPFs password mismatch: %v", err)
  1037. }
  1038. if err := checkEncryptedSecret(expected.SFTPConfig.PrivateKey, actual.SFTPConfig.PrivateKey); err != nil {
  1039. return fmt.Errorf("SFTPFs private key mismatch: %v", err)
  1040. }
  1041. if expected.SFTPConfig.Prefix != actual.SFTPConfig.Prefix {
  1042. if expected.SFTPConfig.Prefix != "" && actual.SFTPConfig.Prefix != "/" {
  1043. return errors.New("SFTPFs prefix mismatch")
  1044. }
  1045. }
  1046. if len(expected.SFTPConfig.Fingerprints) != len(actual.SFTPConfig.Fingerprints) {
  1047. return errors.New("SFTPFs fingerprints mismatch")
  1048. }
  1049. for _, value := range actual.SFTPConfig.Fingerprints {
  1050. if !util.IsStringInSlice(value, expected.SFTPConfig.Fingerprints) {
  1051. return errors.New("SFTPFs fingerprints mismatch")
  1052. }
  1053. }
  1054. return nil
  1055. }
  1056. func compareAzBlobConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
  1057. if expected.AzBlobConfig.Container != actual.AzBlobConfig.Container {
  1058. return errors.New("azure Blob container mismatch")
  1059. }
  1060. if expected.AzBlobConfig.AccountName != actual.AzBlobConfig.AccountName {
  1061. return errors.New("azure Blob account name mismatch")
  1062. }
  1063. if err := checkEncryptedSecret(expected.AzBlobConfig.AccountKey, actual.AzBlobConfig.AccountKey); err != nil {
  1064. return fmt.Errorf("azure Blob account key mismatch: %v", err)
  1065. }
  1066. if expected.AzBlobConfig.Endpoint != actual.AzBlobConfig.Endpoint {
  1067. return errors.New("azure Blob endpoint mismatch")
  1068. }
  1069. if err := checkEncryptedSecret(expected.AzBlobConfig.SASURL, actual.AzBlobConfig.SASURL); err != nil {
  1070. return fmt.Errorf("azure Blob SAS URL mismatch: %v", err)
  1071. }
  1072. if expected.AzBlobConfig.UploadPartSize != actual.AzBlobConfig.UploadPartSize {
  1073. return errors.New("azure Blob upload part size mismatch")
  1074. }
  1075. if expected.AzBlobConfig.UploadConcurrency != actual.AzBlobConfig.UploadConcurrency {
  1076. return errors.New("azure Blob upload concurrency mismatch")
  1077. }
  1078. if expected.AzBlobConfig.KeyPrefix != actual.AzBlobConfig.KeyPrefix &&
  1079. expected.AzBlobConfig.KeyPrefix+"/" != actual.AzBlobConfig.KeyPrefix {
  1080. return errors.New("azure Blob key prefix mismatch")
  1081. }
  1082. if expected.AzBlobConfig.UseEmulator != actual.AzBlobConfig.UseEmulator {
  1083. return errors.New("azure Blob use emulator mismatch")
  1084. }
  1085. if expected.AzBlobConfig.AccessTier != actual.AzBlobConfig.AccessTier {
  1086. return errors.New("azure Blob access tier mismatch")
  1087. }
  1088. return nil
  1089. }
  1090. func areSecretEquals(expected, actual *kms.Secret) bool {
  1091. if expected == nil && actual == nil {
  1092. return true
  1093. }
  1094. if expected != nil && expected.IsEmpty() && actual == nil {
  1095. return true
  1096. }
  1097. if actual != nil && actual.IsEmpty() && expected == nil {
  1098. return true
  1099. }
  1100. return false
  1101. }
  1102. func checkEncryptedSecret(expected, actual *kms.Secret) error {
  1103. if areSecretEquals(expected, actual) {
  1104. return nil
  1105. }
  1106. if expected == nil && actual != nil && !actual.IsEmpty() {
  1107. return errors.New("secret mismatch")
  1108. }
  1109. if actual == nil && expected != nil && !expected.IsEmpty() {
  1110. return errors.New("secret mismatch")
  1111. }
  1112. if expected.IsPlain() && actual.IsEncrypted() {
  1113. if actual.GetPayload() == "" {
  1114. return errors.New("invalid secret payload")
  1115. }
  1116. if actual.GetAdditionalData() != "" {
  1117. return errors.New("invalid secret additional data")
  1118. }
  1119. if actual.GetKey() != "" {
  1120. return errors.New("invalid secret key")
  1121. }
  1122. } else {
  1123. if expected.GetStatus() != actual.GetStatus() || expected.GetPayload() != actual.GetPayload() {
  1124. return errors.New("secret mismatch")
  1125. }
  1126. }
  1127. return nil
  1128. }
  1129. func compareUserFilterSubStructs(expected *dataprovider.User, actual *dataprovider.User) error {
  1130. for _, IPMask := range expected.Filters.AllowedIP {
  1131. if !util.IsStringInSlice(IPMask, actual.Filters.AllowedIP) {
  1132. return errors.New("allowed IP contents mismatch")
  1133. }
  1134. }
  1135. for _, IPMask := range expected.Filters.DeniedIP {
  1136. if !util.IsStringInSlice(IPMask, actual.Filters.DeniedIP) {
  1137. return errors.New("denied IP contents mismatch")
  1138. }
  1139. }
  1140. for _, method := range expected.Filters.DeniedLoginMethods {
  1141. if !util.IsStringInSlice(method, actual.Filters.DeniedLoginMethods) {
  1142. return errors.New("denied login methods contents mismatch")
  1143. }
  1144. }
  1145. for _, protocol := range expected.Filters.DeniedProtocols {
  1146. if !util.IsStringInSlice(protocol, actual.Filters.DeniedProtocols) {
  1147. return errors.New("denied protocols contents mismatch")
  1148. }
  1149. }
  1150. for _, options := range expected.Filters.WebClient {
  1151. if !util.IsStringInSlice(options, actual.Filters.WebClient) {
  1152. return errors.New("web client options contents mismatch")
  1153. }
  1154. }
  1155. if expected.Filters.Hooks.ExternalAuthDisabled != actual.Filters.Hooks.ExternalAuthDisabled {
  1156. return errors.New("external_auth_disabled hook mismatch")
  1157. }
  1158. if expected.Filters.Hooks.PreLoginDisabled != actual.Filters.Hooks.PreLoginDisabled {
  1159. return errors.New("pre_login_disabled hook mismatch")
  1160. }
  1161. if expected.Filters.Hooks.CheckPasswordDisabled != actual.Filters.Hooks.CheckPasswordDisabled {
  1162. return errors.New("check_password_disabled hook mismatch")
  1163. }
  1164. if expected.Filters.DisableFsChecks != actual.Filters.DisableFsChecks {
  1165. return errors.New("disable_fs_checks mismatch")
  1166. }
  1167. return nil
  1168. }
  1169. func compareUserFilters(expected *dataprovider.User, actual *dataprovider.User) error {
  1170. if len(expected.Filters.AllowedIP) != len(actual.Filters.AllowedIP) {
  1171. return errors.New("allowed IP mismatch")
  1172. }
  1173. if len(expected.Filters.DeniedIP) != len(actual.Filters.DeniedIP) {
  1174. return errors.New("denied IP mismatch")
  1175. }
  1176. if len(expected.Filters.DeniedLoginMethods) != len(actual.Filters.DeniedLoginMethods) {
  1177. return errors.New("denied login methods mismatch")
  1178. }
  1179. if len(expected.Filters.DeniedProtocols) != len(actual.Filters.DeniedProtocols) {
  1180. return errors.New("denied protocols mismatch")
  1181. }
  1182. if expected.Filters.MaxUploadFileSize != actual.Filters.MaxUploadFileSize {
  1183. return errors.New("max upload file size mismatch")
  1184. }
  1185. if expected.Filters.TLSUsername != actual.Filters.TLSUsername {
  1186. return errors.New("TLSUsername mismatch")
  1187. }
  1188. if len(expected.Filters.WebClient) != len(actual.Filters.WebClient) {
  1189. return errors.New("WebClient filter mismatch")
  1190. }
  1191. if err := compareUserFilterSubStructs(expected, actual); err != nil {
  1192. return err
  1193. }
  1194. return compareUserFilePatternsFilters(expected, actual)
  1195. }
  1196. func checkFilterMatch(expected []string, actual []string) bool {
  1197. if len(expected) != len(actual) {
  1198. return false
  1199. }
  1200. for _, e := range expected {
  1201. if !util.IsStringInSlice(strings.ToLower(e), actual) {
  1202. return false
  1203. }
  1204. }
  1205. return true
  1206. }
  1207. func compareUserFilePatternsFilters(expected *dataprovider.User, actual *dataprovider.User) error {
  1208. if len(expected.Filters.FilePatterns) != len(actual.Filters.FilePatterns) {
  1209. return errors.New("file patterns mismatch")
  1210. }
  1211. for _, f := range expected.Filters.FilePatterns {
  1212. found := false
  1213. for _, f1 := range actual.Filters.FilePatterns {
  1214. if path.Clean(f.Path) == path.Clean(f1.Path) {
  1215. if !checkFilterMatch(f.AllowedPatterns, f1.AllowedPatterns) ||
  1216. !checkFilterMatch(f.DeniedPatterns, f1.DeniedPatterns) {
  1217. return errors.New("file patterns contents mismatch")
  1218. }
  1219. found = true
  1220. }
  1221. }
  1222. if !found {
  1223. return errors.New("file patterns contents mismatch")
  1224. }
  1225. }
  1226. return nil
  1227. }
  1228. func compareEqualsUserFields(expected *dataprovider.User, actual *dataprovider.User) error {
  1229. if expected.Username != actual.Username {
  1230. return errors.New("username mismatch")
  1231. }
  1232. if expected.HomeDir != actual.HomeDir {
  1233. return errors.New("home dir mismatch")
  1234. }
  1235. if expected.UID != actual.UID {
  1236. return errors.New("UID mismatch")
  1237. }
  1238. if expected.GID != actual.GID {
  1239. return errors.New("GID mismatch")
  1240. }
  1241. if expected.MaxSessions != actual.MaxSessions {
  1242. return errors.New("MaxSessions mismatch")
  1243. }
  1244. if expected.QuotaSize != actual.QuotaSize {
  1245. return errors.New("QuotaSize mismatch")
  1246. }
  1247. if expected.QuotaFiles != actual.QuotaFiles {
  1248. return errors.New("QuotaFiles mismatch")
  1249. }
  1250. if len(expected.Permissions) != len(actual.Permissions) {
  1251. return errors.New("permissions mismatch")
  1252. }
  1253. if expected.UploadBandwidth != actual.UploadBandwidth {
  1254. return errors.New("UploadBandwidth mismatch")
  1255. }
  1256. if expected.DownloadBandwidth != actual.DownloadBandwidth {
  1257. return errors.New("DownloadBandwidth mismatch")
  1258. }
  1259. if expected.Status != actual.Status {
  1260. return errors.New("status mismatch")
  1261. }
  1262. if expected.ExpirationDate != actual.ExpirationDate {
  1263. return errors.New("ExpirationDate mismatch")
  1264. }
  1265. if expected.AdditionalInfo != actual.AdditionalInfo {
  1266. return errors.New("AdditionalInfo mismatch")
  1267. }
  1268. if expected.Description != actual.Description {
  1269. return errors.New("description mismatch")
  1270. }
  1271. return nil
  1272. }
  1273. func addLimitAndOffsetQueryParams(rawurl string, limit, offset int64) (*url.URL, error) {
  1274. url, err := url.Parse(rawurl)
  1275. if err != nil {
  1276. return nil, err
  1277. }
  1278. q := url.Query()
  1279. if limit > 0 {
  1280. q.Add("limit", strconv.FormatInt(limit, 10))
  1281. }
  1282. if offset > 0 {
  1283. q.Add("offset", strconv.FormatInt(offset, 10))
  1284. }
  1285. url.RawQuery = q.Encode()
  1286. return url, err
  1287. }
  1288. func addModeQueryParam(rawurl, mode string) (*url.URL, error) {
  1289. url, err := url.Parse(rawurl)
  1290. if err != nil {
  1291. return nil, err
  1292. }
  1293. q := url.Query()
  1294. if len(mode) > 0 {
  1295. q.Add("mode", mode)
  1296. }
  1297. url.RawQuery = q.Encode()
  1298. return url, err
  1299. }
  1300. func addDisconnectQueryParam(rawurl, disconnect string) (*url.URL, error) {
  1301. url, err := url.Parse(rawurl)
  1302. if err != nil {
  1303. return nil, err
  1304. }
  1305. q := url.Query()
  1306. if len(disconnect) > 0 {
  1307. q.Add("disconnect", disconnect)
  1308. }
  1309. url.RawQuery = q.Encode()
  1310. return url, err
  1311. }