httpdtest.go 53 KB

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