httpdtest.go 64 KB

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