| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601 |
- // 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"
- apiKeysPath = "/api/v2/apikeys"
- retentionBasePath = "/api/v2/retention/users"
- retentionChecksPath = "/api/v2/retention/users/checks"
- )
- 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 admin 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
- }
- // GetAPIKeys returns a list of API keys 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 GetAPIKeys(limit, offset int64, expectedStatusCode int) ([]dataprovider.APIKey, []byte, error) {
- var apiKeys []dataprovider.APIKey
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(apiKeysPath), limit, offset)
- if err != nil {
- return apiKeys, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return apiKeys, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &apiKeys)
- } else {
- body, _ = getResponseBody(resp)
- }
- return apiKeys, body, err
- }
- // AddAPIKey adds a new API key and checks the received HTTP Status code against expectedStatusCode.
- func AddAPIKey(apiKey dataprovider.APIKey, expectedStatusCode int) (dataprovider.APIKey, []byte, error) {
- var newAPIKey dataprovider.APIKey
- var body []byte
- asJSON, _ := json.Marshal(apiKey)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(apiKeysPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newAPIKey, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newAPIKey, body, err
- }
- if err != nil {
- body, _ = getResponseBody(resp)
- return newAPIKey, body, err
- }
- response := make(map[string]string)
- err = render.DecodeJSON(resp.Body, &response)
- if err == nil {
- newAPIKey, body, err = GetAPIKeyByID(resp.Header.Get("X-Object-ID"), http.StatusOK)
- }
- if err == nil {
- err = checkAPIKey(&apiKey, &newAPIKey)
- }
- newAPIKey.Key = response["key"]
- return newAPIKey, body, err
- }
- // UpdateAPIKey updates an existing API key and checks the received HTTP Status code against expectedStatusCode
- func UpdateAPIKey(apiKey dataprovider.APIKey, expectedStatusCode int) (dataprovider.APIKey, []byte, error) {
- var newAPIKey dataprovider.APIKey
- var body []byte
- asJSON, _ := json.Marshal(apiKey)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(apiKeysPath, url.PathEscape(apiKey.KeyID)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newAPIKey, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newAPIKey, body, err
- }
- if err == nil {
- newAPIKey, body, err = GetAPIKeyByID(apiKey.KeyID, expectedStatusCode)
- }
- if err == nil {
- err = checkAPIKey(&apiKey, &newAPIKey)
- }
- return newAPIKey, body, err
- }
- // RemoveAPIKey removes an existing API key and checks the received HTTP Status code against expectedStatusCode.
- func RemoveAPIKey(apiKey dataprovider.APIKey, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(apiKeysPath, url.PathEscape(apiKey.KeyID)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetAPIKeyByID gets a API key by ID and checks the received HTTP Status code against expectedStatusCode.
- func GetAPIKeyByID(keyID string, expectedStatusCode int) (dataprovider.APIKey, []byte, error) {
- var apiKey dataprovider.APIKey
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(apiKeysPath, url.PathEscape(keyID)),
- nil, "", getDefaultToken())
- if err != nil {
- return apiKey, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &apiKey)
- } else {
- body, _ = getResponseBody(resp)
- }
- return apiKey, 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), "application/json",
- getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetRetentionChecks returns the active retention checks
- func GetRetentionChecks(expectedStatusCode int) ([]common.ActiveRetentionChecks, []byte, error) {
- var checks []common.ActiveRetentionChecks
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(retentionChecksPath), nil, "", getDefaultToken())
- if err != nil {
- return checks, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &checks)
- } else {
- body, _ = getResponseBody(resp)
- }
- return checks, body, err
- }
- // StartRetentionCheck starts a new retention check
- func StartRetentionCheck(username string, retention []common.FolderRetention, expectedStatusCode int) ([]byte, error) {
- var body []byte
- asJSON, _ := json.Marshal(retention)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(retentionBasePath, username, "check"),
- bytes.NewBuffer(asJSON), "application/json", 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 checkAPIKey(expected, actual *dataprovider.APIKey) error {
- if actual.Key != "" {
- return errors.New("key must not be visible")
- }
- if actual.KeyID == "" {
- return errors.New("actual key_id cannot be empty")
- }
- if expected.Name != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.Scope != actual.Scope {
- return errors.New("scope mismatch")
- }
- if actual.CreatedAt == 0 {
- return errors.New("created_at cannot be 0")
- }
- if actual.UpdatedAt == 0 {
- return errors.New("updated_at cannot be 0")
- }
- if expected.ExpiresAt != actual.ExpiresAt {
- return errors.New("expires_at mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if expected.User != actual.User {
- return errors.New("user mismatch")
- }
- if expected.Admin != actual.Admin {
- return errors.New("admin mismatch")
- }
- return nil
- }
- func checkAdmin(expected, 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 expected.CreatedAt > 0 {
- if expected.CreatedAt != actual.CreatedAt {
- return fmt.Errorf("created_at mismatch %v != %v", expected.CreatedAt, actual.CreatedAt)
- }
- }
- 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")
- }
- if expected.Filters.AllowAPIKeyAuth != actual.Filters.AllowAPIKeyAuth {
- return errors.New("allow_api_key_auth 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 expected.CreatedAt > 0 {
- if expected.CreatedAt != actual.CreatedAt {
- return fmt.Errorf("created_at mismatch %v != %v", expected.CreatedAt, actual.CreatedAt)
- }
- }
- 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.UploadConcurrency != actual.S3Config.UploadConcurrency {
- return errors.New("fs S3 upload concurrency mismatch")
- }
- if expected.S3Config.DownloadPartSize != actual.S3Config.DownloadPartSize {
- return errors.New("fs S3 download part size mismatch")
- }
- if expected.S3Config.DownloadConcurrency != actual.S3Config.DownloadConcurrency {
- return errors.New("fs S3 download concurrency mismatch")
- }
- if expected.S3Config.ForcePathStyle != actual.S3Config.ForcePathStyle {
- return errors.New("fs S3 force path style mismatch")
- }
- if expected.S3Config.DownloadPartMaxTime != actual.S3Config.DownloadPartMaxTime {
- return errors.New("fs S3 download part max time 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 expected.Filters.AllowAPIKeyAuth != actual.Filters.AllowAPIKeyAuth {
- return errors.New("allow_api_key_auth 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
- }
|