1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390 |
- // Package httpdtest provides utilities for testing the exposed REST API.
- package httpdtest
- import (
- "bytes"
- "encoding/hex"
- "encoding/json"
- "errors"
- "fmt"
- "io"
- "net/http"
- "net/url"
- "path"
- "strconv"
- "strings"
- "github.com/go-chi/render"
- "github.com/drakkan/sftpgo/v2/common"
- "github.com/drakkan/sftpgo/v2/dataprovider"
- "github.com/drakkan/sftpgo/v2/httpclient"
- "github.com/drakkan/sftpgo/v2/httpd"
- "github.com/drakkan/sftpgo/v2/kms"
- "github.com/drakkan/sftpgo/v2/util"
- "github.com/drakkan/sftpgo/v2/version"
- "github.com/drakkan/sftpgo/v2/vfs"
- )
- const (
- tokenPath = "/api/v2/token"
- activeConnectionsPath = "/api/v2/connections"
- quotasBasePath = "/api/v2/quotas"
- quotaScanPath = "/api/v2/quotas/users/scans"
- quotaScanVFolderPath = "/api/v2/quotas/folders/scans"
- userPath = "/api/v2/users"
- versionPath = "/api/v2/version"
- folderPath = "/api/v2/folders"
- serverStatusPath = "/api/v2/status"
- dumpDataPath = "/api/v2/dumpdata"
- loadDataPath = "/api/v2/loaddata"
- defenderHosts = "/api/v2/defender/hosts"
- defenderBanTime = "/api/v2/defender/bantime"
- defenderUnban = "/api/v2/defender/unban"
- defenderScore = "/api/v2/defender/score"
- adminPath = "/api/v2/admins"
- adminPwdPath = "/api/v2/admin/changepwd"
- )
- const (
- defaultTokenAuthUser = "admin"
- defaultTokenAuthPass = "password"
- )
- var (
- httpBaseURL = "http://127.0.0.1:8080"
- jwtToken = ""
- )
- // SetBaseURL sets the base url to use for HTTP requests.
- // Default URL is "http://127.0.0.1:8080"
- func SetBaseURL(url string) {
- httpBaseURL = url
- }
- // SetJWTToken sets the JWT token to use
- func SetJWTToken(token string) {
- jwtToken = token
- }
- func sendHTTPRequest(method, url string, body io.Reader, contentType, token string) (*http.Response, error) {
- req, err := http.NewRequest(method, url, body)
- if err != nil {
- return nil, err
- }
- if contentType != "" {
- req.Header.Set("Content-Type", "application/json")
- }
- if token != "" {
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", token))
- }
- return httpclient.GetHTTPClient().Do(req)
- }
- func buildURLRelativeToBase(paths ...string) string {
- // we need to use path.Join and not filepath.Join
- // since filepath.Join will use backslash separator on Windows
- p := path.Join(paths...)
- return fmt.Sprintf("%s/%s", strings.TrimRight(httpBaseURL, "/"), strings.TrimLeft(p, "/"))
- }
- // GetToken tries to return a JWT token
- func GetToken(username, password string) (string, map[string]interface{}, error) {
- req, err := http.NewRequest(http.MethodGet, buildURLRelativeToBase(tokenPath), nil)
- if err != nil {
- return "", nil, err
- }
- req.SetBasicAuth(username, password)
- resp, err := httpclient.GetHTTPClient().Do(req)
- if err != nil {
- return "", nil, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, http.StatusOK)
- if err != nil {
- return "", nil, err
- }
- responseHolder := make(map[string]interface{})
- err = render.DecodeJSON(resp.Body, &responseHolder)
- if err != nil {
- return "", nil, err
- }
- return responseHolder["access_token"].(string), responseHolder, nil
- }
- func getDefaultToken() string {
- if jwtToken != "" {
- return jwtToken
- }
- token, _, err := GetToken(defaultTokenAuthUser, defaultTokenAuthPass)
- if err != nil {
- return ""
- }
- return token
- }
- // AddUser adds a new user and checks the received HTTP Status code against expectedStatusCode.
- func AddUser(user dataprovider.User, expectedStatusCode int) (dataprovider.User, []byte, error) {
- var newUser dataprovider.User
- var body []byte
- userAsJSON, _ := json.Marshal(user)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(userPath), bytes.NewBuffer(userAsJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newUser, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newUser, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newUser)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkUser(&user, &newUser)
- }
- return newUser, body, err
- }
- // UpdateUserWithJSON update a user using the provided JSON as POST body
- func UpdateUserWithJSON(user dataprovider.User, expectedStatusCode int, disconnect string, userAsJSON []byte) (dataprovider.User, []byte, error) {
- var newUser dataprovider.User
- var body []byte
- url, err := addDisconnectQueryParam(buildURLRelativeToBase(userPath, url.PathEscape(user.Username)), disconnect)
- if err != nil {
- return user, body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "application/json",
- getDefaultToken())
- if err != nil {
- return user, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newUser, body, err
- }
- if err == nil {
- newUser, body, err = GetUserByUsername(user.Username, expectedStatusCode)
- }
- if err == nil {
- err = checkUser(&user, &newUser)
- }
- return newUser, body, err
- }
- // UpdateUser updates an existing user and checks the received HTTP Status code against expectedStatusCode.
- func UpdateUser(user dataprovider.User, expectedStatusCode int, disconnect string) (dataprovider.User, []byte, error) {
- userAsJSON, _ := json.Marshal(user)
- return UpdateUserWithJSON(user, expectedStatusCode, disconnect, userAsJSON)
- }
- // RemoveUser removes an existing user and checks the received HTTP Status code against expectedStatusCode.
- func RemoveUser(user dataprovider.User, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(userPath, url.PathEscape(user.Username)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetUserByUsername gets a user by username and checks the received HTTP Status code against expectedStatusCode.
- func GetUserByUsername(username string, expectedStatusCode int) (dataprovider.User, []byte, error) {
- var user dataprovider.User
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(userPath, url.PathEscape(username)),
- nil, "", getDefaultToken())
- if err != nil {
- return user, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &user)
- } else {
- body, _ = getResponseBody(resp)
- }
- return user, body, err
- }
- // GetUsers returns a list of users and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetUsers(limit, offset int64, expectedStatusCode int) ([]dataprovider.User, []byte, error) {
- var users []dataprovider.User
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(userPath), limit, offset)
- if err != nil {
- return users, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return users, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &users)
- } else {
- body, _ = getResponseBody(resp)
- }
- return users, body, err
- }
- // AddAdmin adds a new user and checks the received HTTP Status code against expectedStatusCode.
- func AddAdmin(admin dataprovider.Admin, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
- var newAdmin dataprovider.Admin
- var body []byte
- asJSON, _ := json.Marshal(admin)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(adminPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newAdmin, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newAdmin, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newAdmin)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkAdmin(&admin, &newAdmin)
- }
- return newAdmin, body, err
- }
- // UpdateAdmin updates an existing admin and checks the received HTTP Status code against expectedStatusCode
- func UpdateAdmin(admin dataprovider.Admin, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
- var newAdmin dataprovider.Admin
- var body []byte
- asJSON, _ := json.Marshal(admin)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(adminPath, url.PathEscape(admin.Username)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newAdmin, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newAdmin, body, err
- }
- if err == nil {
- newAdmin, body, err = GetAdminByUsername(admin.Username, expectedStatusCode)
- }
- if err == nil {
- err = checkAdmin(&admin, &newAdmin)
- }
- return newAdmin, body, err
- }
- // RemoveAdmin removes an existing admin and checks the received HTTP Status code against expectedStatusCode.
- func RemoveAdmin(admin dataprovider.Admin, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(adminPath, url.PathEscape(admin.Username)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetAdminByUsername gets an admin by username and checks the received HTTP Status code against expectedStatusCode.
- func GetAdminByUsername(username string, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
- var admin dataprovider.Admin
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(adminPath, url.PathEscape(username)),
- nil, "", getDefaultToken())
- if err != nil {
- return admin, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &admin)
- } else {
- body, _ = getResponseBody(resp)
- }
- return admin, body, err
- }
- // GetAdmins returns a list of admins and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetAdmins(limit, offset int64, expectedStatusCode int) ([]dataprovider.Admin, []byte, error) {
- var admins []dataprovider.Admin
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(adminPath), limit, offset)
- if err != nil {
- return admins, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return admins, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &admins)
- } else {
- body, _ = getResponseBody(resp)
- }
- return admins, body, err
- }
- // ChangeAdminPassword changes the password for an existing admin
- func ChangeAdminPassword(currentPassword, newPassword string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- pwdChange := make(map[string]string)
- pwdChange["current_password"] = currentPassword
- pwdChange["new_password"] = newPassword
- asJSON, _ := json.Marshal(&pwdChange)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(adminPwdPath),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- body, _ = getResponseBody(resp)
- return body, err
- }
- // GetQuotaScans gets active quota scans for users and checks the received HTTP Status code against expectedStatusCode.
- func GetQuotaScans(expectedStatusCode int) ([]common.ActiveQuotaScan, []byte, error) {
- var quotaScans []common.ActiveQuotaScan
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(quotaScanPath), nil, "", getDefaultToken())
- if err != nil {
- return quotaScans, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, "aScans)
- } else {
- body, _ = getResponseBody(resp)
- }
- return quotaScans, body, err
- }
- // StartQuotaScan starts a new quota scan for the given user and checks the received HTTP Status code against expectedStatusCode.
- func StartQuotaScan(user dataprovider.User, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(quotasBasePath, "users", user.Username, "scan"),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // UpdateQuotaUsage updates the user used quota limits and checks the received HTTP Status code against expectedStatusCode.
- func UpdateQuotaUsage(user dataprovider.User, mode string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- userAsJSON, _ := json.Marshal(user)
- url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "users", user.Username, "usage"), mode)
- if err != nil {
- return body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetConnections returns status and stats for active SFTP/SCP connections
- func GetConnections(expectedStatusCode int) ([]common.ConnectionStatus, []byte, error) {
- var connections []common.ConnectionStatus
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(activeConnectionsPath), nil, "", getDefaultToken())
- if err != nil {
- return connections, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &connections)
- } else {
- body, _ = getResponseBody(resp)
- }
- return connections, body, err
- }
- // CloseConnection closes an active connection identified by connectionID
- func CloseConnection(connectionID string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(activeConnectionsPath, connectionID),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- body, _ = getResponseBody(resp)
- return body, err
- }
- // AddFolder adds a new folder and checks the received HTTP Status code against expectedStatusCode
- func AddFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
- var newFolder vfs.BaseVirtualFolder
- var body []byte
- folderAsJSON, _ := json.Marshal(folder)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(folderPath), bytes.NewBuffer(folderAsJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newFolder, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newFolder, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newFolder)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkFolder(&folder, &newFolder)
- }
- return newFolder, body, err
- }
- // UpdateFolder updates an existing folder and checks the received HTTP Status code against expectedStatusCode.
- func UpdateFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
- var updatedFolder vfs.BaseVirtualFolder
- var body []byte
- folderAsJSON, _ := json.Marshal(folder)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(folderPath, url.PathEscape(folder.Name)),
- bytes.NewBuffer(folderAsJSON), "application/json", getDefaultToken())
- if err != nil {
- return updatedFolder, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return updatedFolder, body, err
- }
- if err == nil {
- updatedFolder, body, err = GetFolderByName(folder.Name, expectedStatusCode)
- }
- if err == nil {
- err = checkFolder(&folder, &updatedFolder)
- }
- return updatedFolder, body, err
- }
- // RemoveFolder removes an existing user and checks the received HTTP Status code against expectedStatusCode.
- func RemoveFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(folderPath, url.PathEscape(folder.Name)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetFolderByName gets a folder by name and checks the received HTTP Status code against expectedStatusCode.
- func GetFolderByName(name string, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
- var folder vfs.BaseVirtualFolder
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(folderPath, url.PathEscape(name)),
- nil, "", getDefaultToken())
- if err != nil {
- return folder, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &folder)
- } else {
- body, _ = getResponseBody(resp)
- }
- return folder, body, err
- }
- // GetFolders returns a list of folders and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- // The results can be filtered specifying a folder path, the folder path filter is an exact match
- func GetFolders(limit int64, offset int64, expectedStatusCode int) ([]vfs.BaseVirtualFolder, []byte, error) {
- var folders []vfs.BaseVirtualFolder
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(folderPath), limit, offset)
- if err != nil {
- return folders, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return folders, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &folders)
- } else {
- body, _ = getResponseBody(resp)
- }
- return folders, body, err
- }
- // GetFoldersQuotaScans gets active quota scans for folders and checks the received HTTP Status code against expectedStatusCode.
- func GetFoldersQuotaScans(expectedStatusCode int) ([]common.ActiveVirtualFolderQuotaScan, []byte, error) {
- var quotaScans []common.ActiveVirtualFolderQuotaScan
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(quotaScanVFolderPath), nil, "", getDefaultToken())
- if err != nil {
- return quotaScans, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, "aScans)
- } else {
- body, _ = getResponseBody(resp)
- }
- return quotaScans, body, err
- }
- // StartFolderQuotaScan start a new quota scan for the given folder and checks the received HTTP Status code against expectedStatusCode.
- func StartFolderQuotaScan(folder vfs.BaseVirtualFolder, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(quotasBasePath, "folders", folder.Name, "scan"),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // UpdateFolderQuotaUsage updates the folder used quota limits and checks the received HTTP Status code against expectedStatusCode.
- func UpdateFolderQuotaUsage(folder vfs.BaseVirtualFolder, mode string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- folderAsJSON, _ := json.Marshal(folder)
- url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "folders", folder.Name, "usage"), mode)
- if err != nil {
- return body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(folderAsJSON), "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetVersion returns version details
- func GetVersion(expectedStatusCode int) (version.Info, []byte, error) {
- var appVersion version.Info
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(versionPath), nil, "", getDefaultToken())
- if err != nil {
- return appVersion, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &appVersion)
- } else {
- body, _ = getResponseBody(resp)
- }
- return appVersion, body, err
- }
- // GetStatus returns the server status
- func GetStatus(expectedStatusCode int) (httpd.ServicesStatus, []byte, error) {
- var response httpd.ServicesStatus
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(serverStatusPath), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && (expectedStatusCode == http.StatusOK) {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // GetDefenderHosts returns hosts that are banned or for which some violations have been detected
- func GetDefenderHosts(expectedStatusCode int) ([]common.DefenderEntry, []byte, error) {
- var response []common.DefenderEntry
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(defenderHosts))
- if err != nil {
- return response, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // GetDefenderHostByIP returns the host with the given IP, if it exists
- func GetDefenderHostByIP(ip string, expectedStatusCode int) (common.DefenderEntry, []byte, error) {
- var host common.DefenderEntry
- var body []byte
- id := hex.EncodeToString([]byte(ip))
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(defenderHosts, id),
- nil, "", getDefaultToken())
- if err != nil {
- return host, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &host)
- } else {
- body, _ = getResponseBody(resp)
- }
- return host, body, err
- }
- // RemoveDefenderHostByIP removes the host with the given IP from the defender list
- func RemoveDefenderHostByIP(ip string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- id := hex.EncodeToString([]byte(ip))
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(defenderHosts, id), nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetBanTime returns the ban time for the given IP address
- func GetBanTime(ip string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
- var response map[string]interface{}
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(defenderBanTime))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- q.Add("ip", ip)
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // GetScore returns the score for the given IP address
- func GetScore(ip string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
- var response map[string]interface{}
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(defenderScore))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- q.Add("ip", ip)
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // UnbanIP unbans the given IP address
- func UnbanIP(ip string, expectedStatusCode int) error {
- postBody := make(map[string]string)
- postBody["ip"] = ip
- asJSON, _ := json.Marshal(postBody)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(defenderUnban), bytes.NewBuffer(asJSON),
- "", getDefaultToken())
- if err != nil {
- return err
- }
- defer resp.Body.Close()
- return checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // Dumpdata requests a backup to outputFile.
- // outputFile is relative to the configured backups_path
- func Dumpdata(outputFile, outputData, indent string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
- var response map[string]interface{}
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(dumpDataPath))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- if outputData != "" {
- q.Add("output-data", outputData)
- }
- if outputFile != "" {
- q.Add("output-file", outputFile)
- }
- if indent != "" {
- q.Add("indent", indent)
- }
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // Loaddata restores a backup.
- func Loaddata(inputFile, scanQuota, mode string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
- var response map[string]interface{}
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(loadDataPath))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- q.Add("input-file", inputFile)
- if scanQuota != "" {
- q.Add("scan-quota", scanQuota)
- }
- if mode != "" {
- q.Add("mode", mode)
- }
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // LoaddataFromPostBody restores a backup
- func LoaddataFromPostBody(data []byte, scanQuota, mode string, expectedStatusCode int) (map[string]interface{}, []byte, error) {
- var response map[string]interface{}
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(loadDataPath))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- if scanQuota != "" {
- q.Add("scan-quota", scanQuota)
- }
- if mode != "" {
- q.Add("mode", mode)
- }
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodPost, url.String(), bytes.NewReader(data), "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- func checkResponse(actual int, expected int) error {
- if expected != actual {
- return fmt.Errorf("wrong status code: got %v want %v", actual, expected)
- }
- return nil
- }
- func getResponseBody(resp *http.Response) ([]byte, error) {
- return io.ReadAll(resp.Body)
- }
- func checkFolder(expected *vfs.BaseVirtualFolder, actual *vfs.BaseVirtualFolder) error {
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual folder ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("folder ID mismatch")
- }
- }
- if expected.Name != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.MappedPath != actual.MappedPath {
- return errors.New("mapped path mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- return compareFsConfig(&expected.FsConfig, &actual.FsConfig)
- }
- func checkAdmin(expected *dataprovider.Admin, actual *dataprovider.Admin) error {
- if actual.Password != "" {
- return errors.New("admin password must not be visible")
- }
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual admin ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("admin ID mismatch")
- }
- }
- if err := compareAdminEqualFields(expected, actual); err != nil {
- return err
- }
- if len(expected.Permissions) != len(actual.Permissions) {
- return errors.New("permissions mismatch")
- }
- for _, p := range expected.Permissions {
- if !util.IsStringInSlice(p, actual.Permissions) {
- return errors.New("permissions content mismatch")
- }
- }
- if len(expected.Filters.AllowList) != len(actual.Filters.AllowList) {
- return errors.New("allow list mismatch")
- }
- for _, v := range expected.Filters.AllowList {
- if !util.IsStringInSlice(v, actual.Filters.AllowList) {
- return errors.New("allow list content mismatch")
- }
- }
- return nil
- }
- func compareAdminEqualFields(expected *dataprovider.Admin, actual *dataprovider.Admin) error {
- if expected.Username != actual.Username {
- return errors.New("sername mismatch")
- }
- if expected.Email != actual.Email {
- return errors.New("email mismatch")
- }
- if expected.Status != actual.Status {
- return errors.New("status mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if expected.AdditionalInfo != actual.AdditionalInfo {
- return errors.New("additional info mismatch")
- }
- return nil
- }
- func checkUser(expected *dataprovider.User, actual *dataprovider.User) error {
- if actual.Password != "" {
- return errors.New("user password must not be visible")
- }
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual user ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("user ID mismatch")
- }
- }
- if len(expected.Permissions) != len(actual.Permissions) {
- return errors.New("permissions mismatch")
- }
- for dir, perms := range expected.Permissions {
- if actualPerms, ok := actual.Permissions[dir]; ok {
- for _, v := range actualPerms {
- if !util.IsStringInSlice(v, perms) {
- return errors.New("permissions contents mismatch")
- }
- }
- } else {
- return errors.New("permissions directories mismatch")
- }
- }
- if err := compareUserFilters(expected, actual); err != nil {
- return err
- }
- if err := compareFsConfig(&expected.FsConfig, &actual.FsConfig); err != nil {
- return err
- }
- if err := compareUserVirtualFolders(expected, actual); err != nil {
- return err
- }
- return compareEqualsUserFields(expected, actual)
- }
- func compareUserVirtualFolders(expected *dataprovider.User, actual *dataprovider.User) error {
- if len(actual.VirtualFolders) != len(expected.VirtualFolders) {
- return errors.New("virtual folders len mismatch")
- }
- for _, v := range actual.VirtualFolders {
- found := false
- for _, v1 := range expected.VirtualFolders {
- if path.Clean(v.VirtualPath) == path.Clean(v1.VirtualPath) {
- if err := checkFolder(&v1.BaseVirtualFolder, &v.BaseVirtualFolder); err != nil {
- return err
- }
- if v.QuotaSize != v1.QuotaSize {
- return errors.New("vfolder quota size mismatch")
- }
- if (v.QuotaFiles) != (v1.QuotaFiles) {
- return errors.New("vfolder quota files mismatch")
- }
- found = true
- break
- }
- }
- if !found {
- return errors.New("virtual folders mismatch")
- }
- }
- return nil
- }
- func compareFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.Provider != actual.Provider {
- return errors.New("fs provider mismatch")
- }
- if err := compareS3Config(expected, actual); err != nil {
- return err
- }
- if err := compareGCSConfig(expected, actual); err != nil {
- return err
- }
- if err := compareAzBlobConfig(expected, actual); err != nil {
- return err
- }
- if err := checkEncryptedSecret(expected.CryptConfig.Passphrase, actual.CryptConfig.Passphrase); err != nil {
- return err
- }
- return compareSFTPFsConfig(expected, actual)
- }
- func compareS3Config(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.S3Config.Bucket != actual.S3Config.Bucket {
- return errors.New("fs S3 bucket mismatch")
- }
- if expected.S3Config.Region != actual.S3Config.Region {
- return errors.New("fs S3 region mismatch")
- }
- if expected.S3Config.AccessKey != actual.S3Config.AccessKey {
- return errors.New("fs S3 access key mismatch")
- }
- if err := checkEncryptedSecret(expected.S3Config.AccessSecret, actual.S3Config.AccessSecret); err != nil {
- return fmt.Errorf("fs S3 access secret mismatch: %v", err)
- }
- if expected.S3Config.Endpoint != actual.S3Config.Endpoint {
- return errors.New("fs S3 endpoint mismatch")
- }
- if expected.S3Config.StorageClass != actual.S3Config.StorageClass {
- return errors.New("fs S3 storage class mismatch")
- }
- if expected.S3Config.UploadPartSize != actual.S3Config.UploadPartSize {
- return errors.New("fs S3 upload part size mismatch")
- }
- if expected.S3Config.DownloadPartMaxTime != actual.S3Config.DownloadPartMaxTime {
- return errors.New("fs S3 download part max time mismatch")
- }
- if expected.S3Config.UploadConcurrency != actual.S3Config.UploadConcurrency {
- return errors.New("fs S3 upload concurrency mismatch")
- }
- if expected.S3Config.KeyPrefix != actual.S3Config.KeyPrefix &&
- expected.S3Config.KeyPrefix+"/" != actual.S3Config.KeyPrefix {
- return errors.New("fs S3 key prefix mismatch")
- }
- return nil
- }
- func compareGCSConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.GCSConfig.Bucket != actual.GCSConfig.Bucket {
- return errors.New("GCS bucket mismatch")
- }
- if expected.GCSConfig.StorageClass != actual.GCSConfig.StorageClass {
- return errors.New("GCS storage class mismatch")
- }
- if expected.GCSConfig.KeyPrefix != actual.GCSConfig.KeyPrefix &&
- expected.GCSConfig.KeyPrefix+"/" != actual.GCSConfig.KeyPrefix {
- return errors.New("GCS key prefix mismatch")
- }
- if expected.GCSConfig.AutomaticCredentials != actual.GCSConfig.AutomaticCredentials {
- return errors.New("GCS automatic credentials mismatch")
- }
- return nil
- }
- func compareSFTPFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.SFTPConfig.Endpoint != actual.SFTPConfig.Endpoint {
- return errors.New("SFTPFs endpoint mismatch")
- }
- if expected.SFTPConfig.Username != actual.SFTPConfig.Username {
- return errors.New("SFTPFs username mismatch")
- }
- if expected.SFTPConfig.DisableCouncurrentReads != actual.SFTPConfig.DisableCouncurrentReads {
- return errors.New("SFTPFs disable_concurrent_reads mismatch")
- }
- if expected.SFTPConfig.BufferSize != actual.SFTPConfig.BufferSize {
- return errors.New("SFTPFs buffer_size mismatch")
- }
- if err := checkEncryptedSecret(expected.SFTPConfig.Password, actual.SFTPConfig.Password); err != nil {
- return fmt.Errorf("SFTPFs password mismatch: %v", err)
- }
- if err := checkEncryptedSecret(expected.SFTPConfig.PrivateKey, actual.SFTPConfig.PrivateKey); err != nil {
- return fmt.Errorf("SFTPFs private key mismatch: %v", err)
- }
- if expected.SFTPConfig.Prefix != actual.SFTPConfig.Prefix {
- if expected.SFTPConfig.Prefix != "" && actual.SFTPConfig.Prefix != "/" {
- return errors.New("SFTPFs prefix mismatch")
- }
- }
- if len(expected.SFTPConfig.Fingerprints) != len(actual.SFTPConfig.Fingerprints) {
- return errors.New("SFTPFs fingerprints mismatch")
- }
- for _, value := range actual.SFTPConfig.Fingerprints {
- if !util.IsStringInSlice(value, expected.SFTPConfig.Fingerprints) {
- return errors.New("SFTPFs fingerprints mismatch")
- }
- }
- return nil
- }
- func compareAzBlobConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.AzBlobConfig.Container != actual.AzBlobConfig.Container {
- return errors.New("azure Blob container mismatch")
- }
- if expected.AzBlobConfig.AccountName != actual.AzBlobConfig.AccountName {
- return errors.New("azure Blob account name mismatch")
- }
- if err := checkEncryptedSecret(expected.AzBlobConfig.AccountKey, actual.AzBlobConfig.AccountKey); err != nil {
- return fmt.Errorf("azure Blob account key mismatch: %v", err)
- }
- if expected.AzBlobConfig.Endpoint != actual.AzBlobConfig.Endpoint {
- return errors.New("azure Blob endpoint mismatch")
- }
- if err := checkEncryptedSecret(expected.AzBlobConfig.SASURL, actual.AzBlobConfig.SASURL); err != nil {
- return fmt.Errorf("azure Blob SAS URL mismatch: %v", err)
- }
- if expected.AzBlobConfig.UploadPartSize != actual.AzBlobConfig.UploadPartSize {
- return errors.New("azure Blob upload part size mismatch")
- }
- if expected.AzBlobConfig.UploadConcurrency != actual.AzBlobConfig.UploadConcurrency {
- return errors.New("azure Blob upload concurrency mismatch")
- }
- if expected.AzBlobConfig.KeyPrefix != actual.AzBlobConfig.KeyPrefix &&
- expected.AzBlobConfig.KeyPrefix+"/" != actual.AzBlobConfig.KeyPrefix {
- return errors.New("azure Blob key prefix mismatch")
- }
- if expected.AzBlobConfig.UseEmulator != actual.AzBlobConfig.UseEmulator {
- return errors.New("azure Blob use emulator mismatch")
- }
- if expected.AzBlobConfig.AccessTier != actual.AzBlobConfig.AccessTier {
- return errors.New("azure Blob access tier mismatch")
- }
- return nil
- }
- func areSecretEquals(expected, actual *kms.Secret) bool {
- if expected == nil && actual == nil {
- return true
- }
- if expected != nil && expected.IsEmpty() && actual == nil {
- return true
- }
- if actual != nil && actual.IsEmpty() && expected == nil {
- return true
- }
- return false
- }
- func checkEncryptedSecret(expected, actual *kms.Secret) error {
- if areSecretEquals(expected, actual) {
- return nil
- }
- if expected == nil && actual != nil && !actual.IsEmpty() {
- return errors.New("secret mismatch")
- }
- if actual == nil && expected != nil && !expected.IsEmpty() {
- return errors.New("secret mismatch")
- }
- if expected.IsPlain() && actual.IsEncrypted() {
- if actual.GetPayload() == "" {
- return errors.New("invalid secret payload")
- }
- if actual.GetAdditionalData() != "" {
- return errors.New("invalid secret additional data")
- }
- if actual.GetKey() != "" {
- return errors.New("invalid secret key")
- }
- } else {
- if expected.GetStatus() != actual.GetStatus() || expected.GetPayload() != actual.GetPayload() {
- return errors.New("secret mismatch")
- }
- }
- return nil
- }
- func compareUserFilterSubStructs(expected *dataprovider.User, actual *dataprovider.User) error {
- for _, IPMask := range expected.Filters.AllowedIP {
- if !util.IsStringInSlice(IPMask, actual.Filters.AllowedIP) {
- return errors.New("allowed IP contents mismatch")
- }
- }
- for _, IPMask := range expected.Filters.DeniedIP {
- if !util.IsStringInSlice(IPMask, actual.Filters.DeniedIP) {
- return errors.New("denied IP contents mismatch")
- }
- }
- for _, method := range expected.Filters.DeniedLoginMethods {
- if !util.IsStringInSlice(method, actual.Filters.DeniedLoginMethods) {
- return errors.New("denied login methods contents mismatch")
- }
- }
- for _, protocol := range expected.Filters.DeniedProtocols {
- if !util.IsStringInSlice(protocol, actual.Filters.DeniedProtocols) {
- return errors.New("denied protocols contents mismatch")
- }
- }
- for _, options := range expected.Filters.WebClient {
- if !util.IsStringInSlice(options, actual.Filters.WebClient) {
- return errors.New("web client options contents mismatch")
- }
- }
- if expected.Filters.Hooks.ExternalAuthDisabled != actual.Filters.Hooks.ExternalAuthDisabled {
- return errors.New("external_auth_disabled hook mismatch")
- }
- if expected.Filters.Hooks.PreLoginDisabled != actual.Filters.Hooks.PreLoginDisabled {
- return errors.New("pre_login_disabled hook mismatch")
- }
- if expected.Filters.Hooks.CheckPasswordDisabled != actual.Filters.Hooks.CheckPasswordDisabled {
- return errors.New("check_password_disabled hook mismatch")
- }
- if expected.Filters.DisableFsChecks != actual.Filters.DisableFsChecks {
- return errors.New("disable_fs_checks mismatch")
- }
- return nil
- }
- func compareUserFilters(expected *dataprovider.User, actual *dataprovider.User) error {
- if len(expected.Filters.AllowedIP) != len(actual.Filters.AllowedIP) {
- return errors.New("allowed IP mismatch")
- }
- if len(expected.Filters.DeniedIP) != len(actual.Filters.DeniedIP) {
- return errors.New("denied IP mismatch")
- }
- if len(expected.Filters.DeniedLoginMethods) != len(actual.Filters.DeniedLoginMethods) {
- return errors.New("denied login methods mismatch")
- }
- if len(expected.Filters.DeniedProtocols) != len(actual.Filters.DeniedProtocols) {
- return errors.New("denied protocols mismatch")
- }
- if expected.Filters.MaxUploadFileSize != actual.Filters.MaxUploadFileSize {
- return errors.New("max upload file size mismatch")
- }
- if expected.Filters.TLSUsername != actual.Filters.TLSUsername {
- return errors.New("TLSUsername mismatch")
- }
- if len(expected.Filters.WebClient) != len(actual.Filters.WebClient) {
- return errors.New("WebClient filter mismatch")
- }
- if err := compareUserFilterSubStructs(expected, actual); err != nil {
- return err
- }
- return compareUserFilePatternsFilters(expected, actual)
- }
- func checkFilterMatch(expected []string, actual []string) bool {
- if len(expected) != len(actual) {
- return false
- }
- for _, e := range expected {
- if !util.IsStringInSlice(strings.ToLower(e), actual) {
- return false
- }
- }
- return true
- }
- func compareUserFilePatternsFilters(expected *dataprovider.User, actual *dataprovider.User) error {
- if len(expected.Filters.FilePatterns) != len(actual.Filters.FilePatterns) {
- return errors.New("file patterns mismatch")
- }
- for _, f := range expected.Filters.FilePatterns {
- found := false
- for _, f1 := range actual.Filters.FilePatterns {
- if path.Clean(f.Path) == path.Clean(f1.Path) {
- if !checkFilterMatch(f.AllowedPatterns, f1.AllowedPatterns) ||
- !checkFilterMatch(f.DeniedPatterns, f1.DeniedPatterns) {
- return errors.New("file patterns contents mismatch")
- }
- found = true
- }
- }
- if !found {
- return errors.New("file patterns contents mismatch")
- }
- }
- return nil
- }
- func compareEqualsUserFields(expected *dataprovider.User, actual *dataprovider.User) error {
- if expected.Username != actual.Username {
- return errors.New("username mismatch")
- }
- if expected.HomeDir != actual.HomeDir {
- return errors.New("home dir mismatch")
- }
- if expected.UID != actual.UID {
- return errors.New("UID mismatch")
- }
- if expected.GID != actual.GID {
- return errors.New("GID mismatch")
- }
- if expected.MaxSessions != actual.MaxSessions {
- return errors.New("MaxSessions mismatch")
- }
- if expected.QuotaSize != actual.QuotaSize {
- return errors.New("QuotaSize mismatch")
- }
- if expected.QuotaFiles != actual.QuotaFiles {
- return errors.New("QuotaFiles mismatch")
- }
- if len(expected.Permissions) != len(actual.Permissions) {
- return errors.New("permissions mismatch")
- }
- if expected.UploadBandwidth != actual.UploadBandwidth {
- return errors.New("UploadBandwidth mismatch")
- }
- if expected.DownloadBandwidth != actual.DownloadBandwidth {
- return errors.New("DownloadBandwidth mismatch")
- }
- if expected.Status != actual.Status {
- return errors.New("status mismatch")
- }
- if expected.ExpirationDate != actual.ExpirationDate {
- return errors.New("ExpirationDate mismatch")
- }
- if expected.AdditionalInfo != actual.AdditionalInfo {
- return errors.New("AdditionalInfo mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- return nil
- }
- func addLimitAndOffsetQueryParams(rawurl string, limit, offset int64) (*url.URL, error) {
- url, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
- q := url.Query()
- if limit > 0 {
- q.Add("limit", strconv.FormatInt(limit, 10))
- }
- if offset > 0 {
- q.Add("offset", strconv.FormatInt(offset, 10))
- }
- url.RawQuery = q.Encode()
- return url, err
- }
- func addModeQueryParam(rawurl, mode string) (*url.URL, error) {
- url, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
- q := url.Query()
- if len(mode) > 0 {
- q.Add("mode", mode)
- }
- url.RawQuery = q.Encode()
- return url, err
- }
- func addDisconnectQueryParam(rawurl, disconnect string) (*url.URL, error) {
- url, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
- q := url.Query()
- if len(disconnect) > 0 {
- q.Add("disconnect", disconnect)
- }
- url.RawQuery = q.Encode()
- return url, err
- }
|