// Copyright (C) 2019-2023 Nicola Murino // // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as published // by the Free Software Foundation, version 3. // // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU Affero General Public License for more details. // // You should have received a copy of the GNU Affero General Public License // along with this program. If not, see . // Package httpdtest provides utilities for testing the supported 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/sftpgo/sdk" "github.com/drakkan/sftpgo/v2/internal/common" "github.com/drakkan/sftpgo/v2/internal/dataprovider" "github.com/drakkan/sftpgo/v2/internal/httpclient" "github.com/drakkan/sftpgo/v2/internal/httpd" "github.com/drakkan/sftpgo/v2/internal/kms" "github.com/drakkan/sftpgo/v2/internal/util" "github.com/drakkan/sftpgo/v2/internal/version" "github.com/drakkan/sftpgo/v2/internal/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" groupPath = "/api/v2/groups" 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" adminPath = "/api/v2/admins" adminPwdPath = "/api/v2/admin/changepwd" apiKeysPath = "/api/v2/apikeys" retentionBasePath = "/api/v2/retention/users" retentionChecksPath = "/api/v2/retention/users/checks" eventActionsPath = "/api/v2/eventactions" eventRulesPath = "/api/v2/eventrules" rolesPath = "/api/v2/roles" ipListsPath = "/api/v2/iplists" ) 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]any, 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]any) 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 := addUpdateUserQueryParams(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 } // AddGroup adds a new group and checks the received HTTP Status code against expectedStatusCode. func AddGroup(group dataprovider.Group, expectedStatusCode int) (dataprovider.Group, []byte, error) { var newGroup dataprovider.Group var body []byte asJSON, _ := json.Marshal(group) resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(groupPath), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newGroup, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusCreated { body, _ = getResponseBody(resp) return newGroup, body, err } if err == nil { err = render.DecodeJSON(resp.Body, &newGroup) } else { body, _ = getResponseBody(resp) } if err == nil { err = checkGroup(group, newGroup) } return newGroup, body, err } // UpdateGroup updates an existing group and checks the received HTTP Status code against expectedStatusCode func UpdateGroup(group dataprovider.Group, expectedStatusCode int) (dataprovider.Group, []byte, error) { var newGroup dataprovider.Group var body []byte asJSON, _ := json.Marshal(group) resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(groupPath, url.PathEscape(group.Name)), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newGroup, body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusOK { return newGroup, body, err } if err == nil { newGroup, body, err = GetGroupByName(group.Name, expectedStatusCode) } if err == nil { err = checkGroup(group, newGroup) } return newGroup, body, err } // RemoveGroup removes an existing group and checks the received HTTP Status code against expectedStatusCode. func RemoveGroup(group dataprovider.Group, expectedStatusCode int) ([]byte, error) { var body []byte resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(groupPath, url.PathEscape(group.Name)), nil, "", getDefaultToken()) if err != nil { return body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) return body, checkResponse(resp.StatusCode, expectedStatusCode) } // GetGroupByName gets a group by name and checks the received HTTP Status code against expectedStatusCode. func GetGroupByName(name string, expectedStatusCode int) (dataprovider.Group, []byte, error) { var group dataprovider.Group var body []byte resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(groupPath, url.PathEscape(name)), nil, "", getDefaultToken()) if err != nil { return group, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &group) } else { body, _ = getResponseBody(resp) } return group, body, err } // GetGroups returns a list of groups 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 GetGroups(limit, offset int64, expectedStatusCode int) ([]dataprovider.Group, []byte, error) { var groups []dataprovider.Group var body []byte url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(groupPath), limit, offset) if err != nil { return groups, body, err } resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken()) if err != nil { return groups, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &groups) } else { body, _ = getResponseBody(resp) } return groups, body, err } // AddRole adds a new role and checks the received HTTP Status code against expectedStatusCode. func AddRole(role dataprovider.Role, expectedStatusCode int) (dataprovider.Role, []byte, error) { var newRole dataprovider.Role var body []byte asJSON, _ := json.Marshal(role) resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(rolesPath), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newRole, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusCreated { body, _ = getResponseBody(resp) return newRole, body, err } if err == nil { err = render.DecodeJSON(resp.Body, &newRole) } else { body, _ = getResponseBody(resp) } if err == nil { err = checkRole(role, newRole) } return newRole, body, err } // UpdateRole updates an existing role and checks the received HTTP Status code against expectedStatusCode func UpdateRole(role dataprovider.Role, expectedStatusCode int) (dataprovider.Role, []byte, error) { var newRole dataprovider.Role var body []byte asJSON, _ := json.Marshal(role) resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(rolesPath, url.PathEscape(role.Name)), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newRole, body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusOK { return newRole, body, err } if err == nil { newRole, body, err = GetRoleByName(role.Name, expectedStatusCode) } if err == nil { err = checkRole(role, newRole) } return newRole, body, err } // RemoveRole removes an existing role and checks the received HTTP Status code against expectedStatusCode. func RemoveRole(role dataprovider.Role, expectedStatusCode int) ([]byte, error) { var body []byte resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(rolesPath, url.PathEscape(role.Name)), nil, "", getDefaultToken()) if err != nil { return body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) return body, checkResponse(resp.StatusCode, expectedStatusCode) } // GetRoleByName gets a role by name and checks the received HTTP Status code against expectedStatusCode. func GetRoleByName(name string, expectedStatusCode int) (dataprovider.Role, []byte, error) { var role dataprovider.Role var body []byte resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(rolesPath, url.PathEscape(name)), nil, "", getDefaultToken()) if err != nil { return role, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &role) } else { body, _ = getResponseBody(resp) } return role, body, err } // GetRoles returns a list of roles 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 GetRoles(limit, offset int64, expectedStatusCode int) ([]dataprovider.Role, []byte, error) { var roles []dataprovider.Role var body []byte url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(rolesPath), limit, offset) if err != nil { return roles, body, err } resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken()) if err != nil { return roles, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &roles) } else { body, _ = getResponseBody(resp) } return roles, body, err } // AddIPListEntry adds a new IP list entry and checks the received HTTP Status code against expectedStatusCode. func AddIPListEntry(entry dataprovider.IPListEntry, expectedStatusCode int) (dataprovider.IPListEntry, []byte, error) { var newEntry dataprovider.IPListEntry var body []byte asJSON, _ := json.Marshal(entry) resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(ipListsPath, strconv.Itoa(int(entry.Type))), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newEntry, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusCreated { body, _ = getResponseBody(resp) return newEntry, body, err } if err == nil { newEntry, body, err = GetIPListEntry(entry.IPOrNet, entry.Type, http.StatusOK) } if err == nil { err = checkIPListEntry(entry, newEntry) } return newEntry, body, err } // UpdateIPListEntry updates an existing IP list entry and checks the received HTTP Status code against expectedStatusCode func UpdateIPListEntry(entry dataprovider.IPListEntry, expectedStatusCode int) (dataprovider.IPListEntry, []byte, error) { var newEntry dataprovider.IPListEntry var body []byte asJSON, _ := json.Marshal(entry) resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(ipListsPath, fmt.Sprintf("%d", entry.Type), url.PathEscape(entry.IPOrNet)), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newEntry, body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusOK { return newEntry, body, err } if err == nil { newEntry, body, err = GetIPListEntry(entry.IPOrNet, entry.Type, http.StatusOK) } if err == nil { err = checkIPListEntry(entry, newEntry) } return newEntry, body, err } // RemoveIPListEntry removes an existing IP list entry and checks the received HTTP Status code against expectedStatusCode. func RemoveIPListEntry(entry dataprovider.IPListEntry, expectedStatusCode int) ([]byte, error) { var body []byte resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(ipListsPath, fmt.Sprintf("%d", entry.Type), url.PathEscape(entry.IPOrNet)), nil, "", getDefaultToken()) if err != nil { return body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) return body, checkResponse(resp.StatusCode, expectedStatusCode) } // GetIPListEntry returns an IP list entry matching the specified parameters, if exists, // and checks the received HTTP Status code against expectedStatusCode. func GetIPListEntry(ipOrNet string, listType dataprovider.IPListType, expectedStatusCode int, ) (dataprovider.IPListEntry, []byte, error) { var entry dataprovider.IPListEntry var body []byte resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(ipListsPath, fmt.Sprintf("%d", listType), url.PathEscape(ipOrNet)), nil, "", getDefaultToken()) if err != nil { return entry, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &entry) } else { body, _ = getResponseBody(resp) } return entry, body, err } // GetIPListEntries returns a list of IP list entries and checks the received HTTP Status code against expectedStatusCode. func GetIPListEntries(listType dataprovider.IPListType, filter, from, order string, limit int64, expectedStatusCode int, ) ([]dataprovider.IPListEntry, []byte, error) { var entries []dataprovider.IPListEntry var body []byte url, err := url.Parse(buildURLRelativeToBase(ipListsPath, strconv.Itoa(int(listType)))) if err != nil { return entries, body, err } q := url.Query() q.Add("filter", filter) q.Add("from", from) q.Add("order", order) if limit > 0 { q.Add("limit", strconv.FormatInt(limit, 10)) } url.RawQuery = q.Encode() resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken()) if err != nil { return entries, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &entries) } else { body, _ = getResponseBody(resp) } return entries, 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 } // AddEventAction adds a new event action func AddEventAction(action dataprovider.BaseEventAction, expectedStatusCode int) (dataprovider.BaseEventAction, []byte, error) { var newAction dataprovider.BaseEventAction var body []byte asJSON, _ := json.Marshal(action) resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(eventActionsPath), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newAction, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusCreated { body, _ = getResponseBody(resp) return newAction, body, err } if err == nil { err = render.DecodeJSON(resp.Body, &newAction) } else { body, _ = getResponseBody(resp) } if err == nil { err = checkEventAction(action, newAction) } return newAction, body, err } // UpdateEventAction updates an existing event action func UpdateEventAction(action dataprovider.BaseEventAction, expectedStatusCode int) (dataprovider.BaseEventAction, []byte, error) { var newAction dataprovider.BaseEventAction var body []byte asJSON, _ := json.Marshal(action) resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(eventActionsPath, url.PathEscape(action.Name)), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newAction, body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusOK { return newAction, body, err } if err == nil { newAction, body, err = GetEventActionByName(action.Name, expectedStatusCode) } if err == nil { err = checkEventAction(action, newAction) } return newAction, body, err } // RemoveEventAction removes an existing action and checks the received HTTP Status code against expectedStatusCode. func RemoveEventAction(action dataprovider.BaseEventAction, expectedStatusCode int) ([]byte, error) { var body []byte resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(eventActionsPath, url.PathEscape(action.Name)), nil, "", getDefaultToken()) if err != nil { return body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) return body, checkResponse(resp.StatusCode, expectedStatusCode) } // GetEventActionByName gets an event action by name and checks the received HTTP Status code against expectedStatusCode. func GetEventActionByName(name string, expectedStatusCode int) (dataprovider.BaseEventAction, []byte, error) { var action dataprovider.BaseEventAction var body []byte resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(eventActionsPath, url.PathEscape(name)), nil, "", getDefaultToken()) if err != nil { return action, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &action) } else { body, _ = getResponseBody(resp) } return action, body, err } // GetEventActions returns a list of event actions 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 GetEventActions(limit, offset int64, expectedStatusCode int) ([]dataprovider.BaseEventAction, []byte, error) { var actions []dataprovider.BaseEventAction var body []byte url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(eventActionsPath), limit, offset) if err != nil { return actions, body, err } resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken()) if err != nil { return actions, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &actions) } else { body, _ = getResponseBody(resp) } return actions, body, err } // AddEventRule adds a new event rule func AddEventRule(rule dataprovider.EventRule, expectedStatusCode int) (dataprovider.EventRule, []byte, error) { var newRule dataprovider.EventRule var body []byte asJSON, _ := json.Marshal(rule) resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(eventRulesPath), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newRule, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusCreated { body, _ = getResponseBody(resp) return newRule, body, err } if err == nil { err = render.DecodeJSON(resp.Body, &newRule) } else { body, _ = getResponseBody(resp) } if err == nil { err = checkEventRule(rule, newRule) } return newRule, body, err } // UpdateEventRule updates an existing event rule func UpdateEventRule(rule dataprovider.EventRule, expectedStatusCode int) (dataprovider.EventRule, []byte, error) { var newRule dataprovider.EventRule var body []byte asJSON, _ := json.Marshal(rule) resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(eventRulesPath, url.PathEscape(rule.Name)), bytes.NewBuffer(asJSON), "application/json", getDefaultToken()) if err != nil { return newRule, body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) err = checkResponse(resp.StatusCode, expectedStatusCode) if expectedStatusCode != http.StatusOK { return newRule, body, err } if err == nil { newRule, body, err = GetEventRuleByName(rule.Name, expectedStatusCode) } if err == nil { err = checkEventRule(rule, newRule) } return newRule, body, err } // RemoveEventRule removes an existing rule and checks the received HTTP Status code against expectedStatusCode. func RemoveEventRule(rule dataprovider.EventRule, expectedStatusCode int) ([]byte, error) { var body []byte resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(eventRulesPath, url.PathEscape(rule.Name)), nil, "", getDefaultToken()) if err != nil { return body, err } defer resp.Body.Close() body, _ = getResponseBody(resp) return body, checkResponse(resp.StatusCode, expectedStatusCode) } // GetEventRuleByName gets an event rule by name and checks the received HTTP Status code against expectedStatusCode. func GetEventRuleByName(name string, expectedStatusCode int) (dataprovider.EventRule, []byte, error) { var rule dataprovider.EventRule var body []byte resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(eventRulesPath, url.PathEscape(name)), nil, "", getDefaultToken()) if err != nil { return rule, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &rule) } else { body, _ = getResponseBody(resp) } return rule, body, err } // GetEventRules returns a list of event rules 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 GetEventRules(limit, offset int64, expectedStatusCode int) ([]dataprovider.EventRule, []byte, error) { var rules []dataprovider.EventRule var body []byte url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(eventRulesPath), limit, offset) if err != nil { return rules, body, err } resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken()) if err != nil { return rules, body, err } defer resp.Body.Close() err = checkResponse(resp.StatusCode, expectedStatusCode) if err == nil && expectedStatusCode == http.StatusOK { err = render.DecodeJSON(resp.Body, &rules) } else { body, _ = getResponseBody(resp) } return rules, body, err } // RunOnDemandRule executes the specified on demand rule func RunOnDemandRule(name string, expectedStatusCode int) ([]byte, error) { resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(eventRulesPath, "run", url.PathEscape(name)), nil, "application/json", getDefaultToken()) if err != nil { return nil, err } defer resp.Body.Close() b, err := getResponseBody(resp) if err != nil { return b, err } if err := checkResponse(resp.StatusCode, expectedStatusCode); err != nil { return b, err } return b, nil } // 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) } // UpdateTransferQuotaUsage updates the user used transfer quota limits and checks the received // HTTP Status code against expectedStatusCode. func UpdateTransferQuotaUsage(user dataprovider.User, mode string, expectedStatusCode int) ([]byte, error) { var body []byte userAsJSON, _ := json.Marshal(user) url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "users", user.Username, "transfer-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 []dataprovider.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) ([]dataprovider.DefenderEntry, []byte, error) { var response []dataprovider.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) (dataprovider.DefenderEntry, []byte, error) { var host dataprovider.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) } // Dumpdata requests a backup to outputFile. // outputFile is relative to the configured backups_path func Dumpdata(outputFile, outputData, indent string, expectedStatusCode int, scopes ...string) (map[string]any, []byte, error) { var response map[string]any 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) } if len(scopes) > 0 { q.Add("scopes", strings.Join(scopes, ",")) } 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]any, []byte, error) { var response map[string]any 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]any, []byte, error) { var response map[string]any 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 checkEventAction(expected, actual dataprovider.BaseEventAction) error { if expected.ID <= 0 { if actual.ID <= 0 { return errors.New("actual action ID must be > 0") } } else { if actual.ID != expected.ID { return errors.New("action ID mismatch") } } if dataprovider.ConvertName(expected.Name) != actual.Name { return errors.New("name mismatch") } if expected.Description != actual.Description { return errors.New("description mismatch") } if expected.Type != actual.Type { return errors.New("type mismatch") } if expected.Options.PwdExpirationConfig.Threshold != actual.Options.PwdExpirationConfig.Threshold { return errors.New("password expiration threshold mismatch") } if err := compareEventActionIDPConfigFields(expected.Options.IDPConfig, actual.Options.IDPConfig); err != nil { return err } if err := compareEventActionCmdConfigFields(expected.Options.CmdConfig, actual.Options.CmdConfig); err != nil { return err } if err := compareEventActionEmailConfigFields(expected.Options.EmailConfig, actual.Options.EmailConfig); err != nil { return err } if err := compareEventActionDataRetentionFields(expected.Options.RetentionConfig, actual.Options.RetentionConfig); err != nil { return err } if err := compareEventActionFsConfigFields(expected.Options.FsConfig, actual.Options.FsConfig); err != nil { return err } return compareEventActionHTTPConfigFields(expected.Options.HTTPConfig, actual.Options.HTTPConfig) } func checkEventSchedules(expected, actual []dataprovider.Schedule) error { if len(expected) != len(actual) { return errors.New("schedules mismatch") } for _, ex := range expected { found := false for _, ac := range actual { if ac.DayOfMonth == ex.DayOfMonth && ac.DayOfWeek == ex.DayOfWeek && ac.Hours == ex.Hours && ac.Month == ex.Month { found = true break } } if !found { return errors.New("schedules content mismatch") } } return nil } func compareConditionPatternOptions(expected, actual []dataprovider.ConditionPattern) error { if len(expected) != len(actual) { return errors.New("condition pattern mismatch") } for _, ex := range expected { found := false for _, ac := range actual { if ac.Pattern == ex.Pattern && ac.InverseMatch == ex.InverseMatch { found = true break } } if !found { return errors.New("condition pattern content mismatch") } } return nil } func checkEventConditionOptions(expected, actual dataprovider.ConditionOptions) error { if err := compareConditionPatternOptions(expected.Names, actual.Names); err != nil { return errors.New("condition names mismatch") } if err := compareConditionPatternOptions(expected.GroupNames, actual.GroupNames); err != nil { return errors.New("condition group names mismatch") } if err := compareConditionPatternOptions(expected.RoleNames, actual.RoleNames); err != nil { return errors.New("condition role names mismatch") } if err := compareConditionPatternOptions(expected.FsPaths, actual.FsPaths); err != nil { return errors.New("condition fs_paths mismatch") } if len(expected.Protocols) != len(actual.Protocols) { return errors.New("condition protocols mismatch") } for _, v := range expected.Protocols { if !util.Contains(actual.Protocols, v) { return errors.New("condition protocols content mismatch") } } if len(expected.ProviderObjects) != len(actual.ProviderObjects) { return errors.New("condition provider objects mismatch") } for _, v := range expected.ProviderObjects { if !util.Contains(actual.ProviderObjects, v) { return errors.New("condition provider objects content mismatch") } } if expected.MinFileSize != actual.MinFileSize { return errors.New("condition min file size mismatch") } if expected.MaxFileSize != actual.MaxFileSize { return errors.New("condition max file size mismatch") } return nil } func checkEventConditions(expected, actual dataprovider.EventConditions) error { if len(expected.FsEvents) != len(actual.FsEvents) { return errors.New("fs events mismatch") } for _, v := range expected.FsEvents { if !util.Contains(actual.FsEvents, v) { return errors.New("fs events content mismatch") } } if len(expected.ProviderEvents) != len(actual.ProviderEvents) { return errors.New("provider events mismatch") } for _, v := range expected.ProviderEvents { if !util.Contains(actual.ProviderEvents, v) { return errors.New("provider events content mismatch") } } if err := checkEventConditionOptions(expected.Options, actual.Options); err != nil { return err } if expected.IDPLoginEvent != actual.IDPLoginEvent { return errors.New("IDP login event mismatch") } return checkEventSchedules(expected.Schedules, actual.Schedules) } func checkEventRuleActions(expected, actual []dataprovider.EventAction) error { if len(expected) != len(actual) { return errors.New("actions mismatch") } for _, ex := range expected { found := false for _, ac := range actual { if ex.Name == ac.Name && ex.Order == ac.Order && ex.Options.ExecuteSync == ac.Options.ExecuteSync && ex.Options.IsFailureAction == ac.Options.IsFailureAction && ex.Options.StopOnFailure == ac.Options.StopOnFailure { found = true break } } if !found { return errors.New("actions contents mismatch") } } return nil } func checkEventRule(expected, actual dataprovider.EventRule) error { if expected.ID <= 0 { if actual.ID <= 0 { return errors.New("actual group ID must be > 0") } } else { if actual.ID != expected.ID { return errors.New("group ID mismatch") } } if dataprovider.ConvertName(expected.Name) != actual.Name { return errors.New("name mismatch") } if expected.Status != actual.Status { return errors.New("status mismatch") } if expected.Description != actual.Description { return errors.New("description mismatch") } if actual.CreatedAt == 0 { return errors.New("created_at unset") } if actual.UpdatedAt == 0 { return errors.New("updated_at unset") } if expected.Trigger != actual.Trigger { return errors.New("trigger mismatch") } if err := checkEventConditions(expected.Conditions, actual.Conditions); err != nil { return err } return checkEventRuleActions(expected.Actions, actual.Actions) } func checkIPListEntry(expected, actual dataprovider.IPListEntry) error { if expected.IPOrNet != actual.IPOrNet { return errors.New("ipornet mismatch") } if expected.Description != actual.Description { return errors.New("description mismatch") } if expected.Type != actual.Type { return errors.New("type mismatch") } if expected.Mode != actual.Mode { return errors.New("mode mismatch") } if expected.Protocols != actual.Protocols { return errors.New("protocols mismatch") } if actual.CreatedAt == 0 { return errors.New("created_at unset") } if actual.UpdatedAt == 0 { return errors.New("updated_at unset") } return nil } func checkRole(expected, actual dataprovider.Role) error { if expected.ID <= 0 { if actual.ID <= 0 { return errors.New("actual role ID must be > 0") } } else { if actual.ID != expected.ID { return errors.New("role ID mismatch") } } if dataprovider.ConvertName(expected.Name) != actual.Name { return errors.New("name mismatch") } if expected.Description != actual.Description { return errors.New("description mismatch") } if actual.CreatedAt == 0 { return errors.New("created_at unset") } if actual.UpdatedAt == 0 { return errors.New("updated_at unset") } return nil } func checkGroup(expected, actual dataprovider.Group) error { if expected.ID <= 0 { if actual.ID <= 0 { return errors.New("actual group ID must be > 0") } } else { if actual.ID != expected.ID { return errors.New("group ID mismatch") } } if dataprovider.ConvertName(expected.Name) != actual.Name { return errors.New("name mismatch") } if expected.Description != actual.Description { return errors.New("description mismatch") } if actual.CreatedAt == 0 { return errors.New("created_at unset") } if actual.UpdatedAt == 0 { return errors.New("updated_at unset") } if err := compareEqualGroupSettingsFields(expected.UserSettings.BaseGroupUserSettings, actual.UserSettings.BaseGroupUserSettings); err != nil { return err } if err := compareVirtualFolders(expected.VirtualFolders, actual.VirtualFolders); err != nil { return err } if err := compareUserFilters(expected.UserSettings.Filters, actual.UserSettings.Filters); err != nil { return err } return compareFsConfig(&expected.UserSettings.FsConfig, &actual.UserSettings.FsConfig) } 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 dataprovider.ConvertName(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.Contains(actual.Permissions, p) { return errors.New("permissions content mismatch") } } if err := compareAdminFilters(expected.Filters, actual.Filters); err != nil { return err } return compareAdminGroups(expected, actual) } func compareAdminFilters(expected, actual dataprovider.AdminFilters) error { if expected.AllowAPIKeyAuth != actual.AllowAPIKeyAuth { return errors.New("allow_api_key_auth mismatch") } if len(expected.AllowList) != len(actual.AllowList) { return errors.New("allow list mismatch") } for _, v := range expected.AllowList { if !util.Contains(actual.AllowList, v) { return errors.New("allow list content mismatch") } } if expected.Preferences.HideUserPageSections != actual.Preferences.HideUserPageSections { return errors.New("hide user page sections mismatch") } if expected.Preferences.DefaultUsersExpiration != actual.Preferences.DefaultUsersExpiration { return errors.New("default users expiration mismatch") } return nil } func compareAdminEqualFields(expected *dataprovider.Admin, actual *dataprovider.Admin) error { if dataprovider.ConvertName(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") } if expected.Role != actual.Role { return errors.New("role 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 expected.Email != actual.Email { return errors.New("email mismatch") } if expected.Filters.RequirePasswordChange != actual.Filters.RequirePasswordChange { return errors.New("require_password_change mismatch") } if err := compareUserPermissions(expected.Permissions, actual.Permissions); err != nil { return err } if err := compareUserFilters(expected.Filters.BaseUserFilters, actual.Filters.BaseUserFilters); err != nil { return err } if err := compareFsConfig(&expected.FsConfig, &actual.FsConfig); err != nil { return err } if err := compareUserGroups(expected, actual); err != nil { return err } if err := compareVirtualFolders(expected.VirtualFolders, actual.VirtualFolders); err != nil { return err } return compareEqualsUserFields(expected, actual) } func compareUserPermissions(expected map[string][]string, actual map[string][]string) error { if len(expected) != len(actual) { return errors.New("permissions mismatch") } for dir, perms := range expected { if actualPerms, ok := actual[dir]; ok { for _, v := range actualPerms { if !util.Contains(perms, v) { return errors.New("permissions contents mismatch") } } } else { return errors.New("permissions directories mismatch") } } return nil } func compareAdminGroups(expected *dataprovider.Admin, actual *dataprovider.Admin) error { if len(actual.Groups) != len(expected.Groups) { return errors.New("groups len mismatch") } for _, g := range actual.Groups { found := false for _, g1 := range expected.Groups { if g1.Name == g.Name { found = true if g1.Options.AddToUsersAs != g.Options.AddToUsersAs { return fmt.Errorf("add to users as field mismatch for group %s", g.Name) } } } if !found { return errors.New("groups mismatch") } } return nil } func compareUserGroups(expected *dataprovider.User, actual *dataprovider.User) error { if len(actual.Groups) != len(expected.Groups) { return errors.New("groups len mismatch") } for _, g := range actual.Groups { found := false for _, g1 := range expected.Groups { if g1.Name == g.Name { found = true if g1.Type != g.Type { return fmt.Errorf("type mismatch for group %s", g.Name) } } } if !found { return errors.New("groups mismatch") } } return nil } func compareVirtualFolders(expected []vfs.VirtualFolder, actual []vfs.VirtualFolder) error { if len(actual) != len(expected) { return errors.New("virtual folders len mismatch") } for _, v := range actual { found := false for _, v1 := range expected { 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 expected.OSConfig.ReadBufferSize != actual.OSConfig.ReadBufferSize { return fmt.Errorf("read buffer size mismatch") } if expected.OSConfig.WriteBufferSize != actual.OSConfig.WriteBufferSize { return fmt.Errorf("write buffer size 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 } if expected.CryptConfig.ReadBufferSize != actual.CryptConfig.ReadBufferSize { return fmt.Errorf("crypt read buffer size mismatch") } if expected.CryptConfig.WriteBufferSize != actual.CryptConfig.WriteBufferSize { return fmt.Errorf("crypt write buffer size mismatch") } if err := compareSFTPFsConfig(expected, actual); err != nil { return err } return compareHTTPFsConfig(expected, actual) } func compareS3Config(expected *vfs.Filesystem, actual *vfs.Filesystem) error { //nolint:gocyclo 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 expected.S3Config.RoleARN != actual.S3Config.RoleARN { return errors.New("fs S3 role ARN 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.ACL != actual.S3Config.ACL { return errors.New("fs S3 ACL 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.UploadPartMaxTime != actual.S3Config.UploadPartMaxTime { return errors.New("fs S3 upload 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.ACL != actual.GCSConfig.ACL { return errors.New("GCS ACL 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") } if expected.GCSConfig.UploadPartSize != actual.GCSConfig.UploadPartSize { return errors.New("GCS upload part size mismatch") } if expected.GCSConfig.UploadPartMaxTime != actual.GCSConfig.UploadPartMaxTime { return errors.New("GCS upload part max time mismatch") } return nil } func compareHTTPFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error { if expected.HTTPConfig.Endpoint != actual.HTTPConfig.Endpoint { return errors.New("HTTPFs endpoint mismatch") } if expected.HTTPConfig.Username != actual.HTTPConfig.Username { return errors.New("HTTPFs username mismatch") } if expected.HTTPConfig.SkipTLSVerify != actual.HTTPConfig.SkipTLSVerify { return errors.New("HTTPFs skip_tls_verify mismatch") } if expected.SFTPConfig.EqualityCheckMode != actual.SFTPConfig.EqualityCheckMode { return errors.New("HTTPFs equality_check_mode mismatch") } if err := checkEncryptedSecret(expected.HTTPConfig.Password, actual.HTTPConfig.Password); err != nil { return fmt.Errorf("HTTPFs password mismatch: %v", err) } if err := checkEncryptedSecret(expected.HTTPConfig.APIKey, actual.HTTPConfig.APIKey); err != nil { return fmt.Errorf("HTTPFs API key mismatch: %v", err) } 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 expected.SFTPConfig.EqualityCheckMode != actual.SFTPConfig.EqualityCheckMode { return errors.New("SFTPFs equality_check_mode 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 err := checkEncryptedSecret(expected.SFTPConfig.KeyPassphrase, actual.SFTPConfig.KeyPassphrase); err != nil { return fmt.Errorf("SFTPFs private key passphrase 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.Contains(expected.SFTPConfig.Fingerprints, value) { 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.DownloadPartSize != actual.AzBlobConfig.DownloadPartSize { return errors.New("azure Blob download part size mismatch") } if expected.AzBlobConfig.DownloadConcurrency != actual.AzBlobConfig.DownloadConcurrency { return errors.New("azure Blob download 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 sdk.BaseUserFilters, actual sdk.BaseUserFilters) error { for _, IPMask := range expected.AllowedIP { if !util.Contains(actual.AllowedIP, IPMask) { return errors.New("allowed IP contents mismatch") } } for _, IPMask := range expected.DeniedIP { if !util.Contains(actual.DeniedIP, IPMask) { return errors.New("denied IP contents mismatch") } } for _, method := range expected.DeniedLoginMethods { if !util.Contains(actual.DeniedLoginMethods, method) { return errors.New("denied login methods contents mismatch") } } for _, protocol := range expected.DeniedProtocols { if !util.Contains(actual.DeniedProtocols, protocol) { return errors.New("denied protocols contents mismatch") } } for _, options := range expected.WebClient { if !util.Contains(actual.WebClient, options) { return errors.New("web client options contents mismatch") } } return compareUserFiltersEqualFields(expected, actual) } func compareUserFiltersEqualFields(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error { if expected.Hooks.ExternalAuthDisabled != actual.Hooks.ExternalAuthDisabled { return errors.New("external_auth_disabled hook mismatch") } if expected.Hooks.PreLoginDisabled != actual.Hooks.PreLoginDisabled { return errors.New("pre_login_disabled hook mismatch") } if expected.Hooks.CheckPasswordDisabled != actual.Hooks.CheckPasswordDisabled { return errors.New("check_password_disabled hook mismatch") } if expected.DisableFsChecks != actual.DisableFsChecks { return errors.New("disable_fs_checks mismatch") } if expected.StartDirectory != actual.StartDirectory { return errors.New("start_directory mismatch") } return nil } func compareBaseUserFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error { if len(expected.AllowedIP) != len(actual.AllowedIP) { return errors.New("allowed IP mismatch") } if len(expected.DeniedIP) != len(actual.DeniedIP) { return errors.New("denied IP mismatch") } if len(expected.DeniedLoginMethods) != len(actual.DeniedLoginMethods) { return errors.New("denied login methods mismatch") } if len(expected.DeniedProtocols) != len(actual.DeniedProtocols) { return errors.New("denied protocols mismatch") } if expected.MaxUploadFileSize != actual.MaxUploadFileSize { return errors.New("max upload file size mismatch") } if expected.TLSUsername != actual.TLSUsername { return errors.New("TLSUsername mismatch") } if len(expected.WebClient) != len(actual.WebClient) { return errors.New("WebClient filter mismatch") } if expected.AllowAPIKeyAuth != actual.AllowAPIKeyAuth { return errors.New("allow_api_key_auth mismatch") } if expected.ExternalAuthCacheTime != actual.ExternalAuthCacheTime { return errors.New("external_auth_cache_time mismatch") } if expected.FTPSecurity != actual.FTPSecurity { return errors.New("ftp_security mismatch") } if expected.IsAnonymous != actual.IsAnonymous { return errors.New("is_anonymous mismatch") } if expected.DefaultSharesExpiration != actual.DefaultSharesExpiration { return errors.New("default_shares_expiration mismatch") } if expected.PasswordExpiration != actual.PasswordExpiration { return errors.New("password_expiration mismatch") } if expected.PasswordStrength != actual.PasswordStrength { return errors.New("password_strength mismatch") } return nil } func compareUserFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error { if err := compareBaseUserFilters(expected, actual); err != nil { return err } if err := compareUserFilterSubStructs(expected, actual); err != nil { return err } if err := compareUserBandwidthLimitFilters(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.Contains(actual, strings.ToLower(e)) { return false } } return true } func compareUserBandwidthLimitFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error { if len(expected.BandwidthLimits) != len(actual.BandwidthLimits) { return errors.New("bandwidth limits filters mismatch") } for idx, l := range expected.BandwidthLimits { if actual.BandwidthLimits[idx].UploadBandwidth != l.UploadBandwidth { return errors.New("bandwidth filters upload_bandwidth mismatch") } if actual.BandwidthLimits[idx].DownloadBandwidth != l.DownloadBandwidth { return errors.New("bandwidth filters download_bandwidth mismatch") } if len(actual.BandwidthLimits[idx].Sources) != len(l.Sources) { return errors.New("bandwidth filters sources mismatch") } for _, source := range actual.BandwidthLimits[idx].Sources { if !util.Contains(l.Sources, source) { return errors.New("bandwidth filters source mismatch") } } } return nil } func compareUserFilePatternsFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error { if len(expected.FilePatterns) != len(actual.FilePatterns) { return errors.New("file patterns mismatch") } for _, f := range expected.FilePatterns { found := false for _, f1 := range actual.FilePatterns { if path.Clean(f.Path) == path.Clean(f1.Path) && f.DenyPolicy == f1.DenyPolicy { 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 compareKeyValues(expected, actual []dataprovider.KeyValue) error { if len(expected) != len(actual) { return errors.New("kay values mismatch") } for _, ex := range expected { found := false for _, ac := range actual { if ac.Key == ex.Key && ac.Value == ex.Value { found = true break } } if !found { return errors.New("kay values mismatch") } } return nil } func compareHTTPparts(expected, actual []dataprovider.HTTPPart) error { for _, p1 := range expected { found := false for _, p2 := range actual { if p1.Name == p2.Name { found = true if err := compareKeyValues(p1.Headers, p2.Headers); err != nil { return fmt.Errorf("http headers mismatch for part %q", p1.Name) } if p1.Body != p2.Body || p1.Filepath != p2.Filepath { return fmt.Errorf("http part %q mismatch", p1.Name) } } } if !found { return fmt.Errorf("expected http part %q not found", p1.Name) } } return nil } func compareEventActionHTTPConfigFields(expected, actual dataprovider.EventActionHTTPConfig) error { if expected.Endpoint != actual.Endpoint { return errors.New("http endpoint mismatch") } if expected.Username != actual.Username { return errors.New("http username mismatch") } if err := checkEncryptedSecret(expected.Password, actual.Password); err != nil { return err } if err := compareKeyValues(expected.Headers, actual.Headers); err != nil { return errors.New("http headers mismatch") } if expected.Timeout != actual.Timeout { return errors.New("http timeout mismatch") } if expected.SkipTLSVerify != actual.SkipTLSVerify { return errors.New("http skip TLS verify mismatch") } if expected.Method != actual.Method { return errors.New("http method mismatch") } if err := compareKeyValues(expected.QueryParameters, actual.QueryParameters); err != nil { return errors.New("http query parameters mismatch") } if expected.Body != actual.Body { return errors.New("http body mismatch") } if len(expected.Parts) != len(actual.Parts) { return errors.New("http parts mismatch") } return compareHTTPparts(expected.Parts, actual.Parts) } func compareEventActionEmailConfigFields(expected, actual dataprovider.EventActionEmailConfig) error { if len(expected.Recipients) != len(actual.Recipients) { return errors.New("email recipients mismatch") } for _, v := range expected.Recipients { if !util.Contains(actual.Recipients, v) { return errors.New("email recipients content mismatch") } } if len(expected.Bcc) != len(actual.Bcc) { return errors.New("email bcc mismatch") } for _, v := range expected.Bcc { if !util.Contains(actual.Bcc, v) { return errors.New("email bcc content mismatch") } } if expected.Subject != actual.Subject { return errors.New("email subject mismatch") } if expected.ContentType != actual.ContentType { return errors.New("email content type mismatch") } if expected.Body != actual.Body { return errors.New("email body mismatch") } if len(expected.Attachments) != len(actual.Attachments) { return errors.New("email attachments mismatch") } for _, v := range expected.Attachments { if !util.Contains(actual.Attachments, v) { return errors.New("email attachments content mismatch") } } return nil } func compareEventActionFsCompressFields(expected, actual dataprovider.EventActionFsCompress) error { if expected.Name != actual.Name { return errors.New("fs compress name mismatch") } if len(expected.Paths) != len(actual.Paths) { return errors.New("fs compress paths mismatch") } for _, v := range expected.Paths { if !util.Contains(actual.Paths, v) { return errors.New("fs compress paths content mismatch") } } return nil } func compareEventActionFsConfigFields(expected, actual dataprovider.EventActionFilesystemConfig) error { if expected.Type != actual.Type { return errors.New("fs type mismatch") } if err := compareKeyValues(expected.Renames, actual.Renames); err != nil { return errors.New("fs renames mismatch") } if err := compareKeyValues(expected.Copy, actual.Copy); err != nil { return errors.New("fs copy mismatch") } if len(expected.Deletes) != len(actual.Deletes) { return errors.New("fs deletes mismatch") } for _, v := range expected.Deletes { if !util.Contains(actual.Deletes, v) { return errors.New("fs deletes content mismatch") } } if len(expected.MkDirs) != len(actual.MkDirs) { return errors.New("fs mkdirs mismatch") } for _, v := range expected.MkDirs { if !util.Contains(actual.MkDirs, v) { return errors.New("fs mkdir content mismatch") } } if len(expected.Exist) != len(actual.Exist) { return errors.New("fs exist mismatch") } for _, v := range expected.Exist { if !util.Contains(actual.Exist, v) { return errors.New("fs exist content mismatch") } } return compareEventActionFsCompressFields(expected.Compress, actual.Compress) } func compareEventActionIDPConfigFields(expected, actual dataprovider.EventActionIDPAccountCheck) error { if expected.Mode != actual.Mode { return errors.New("mode mismatch") } if expected.TemplateAdmin != actual.TemplateAdmin { return errors.New("admin template mismatch") } if expected.TemplateUser != actual.TemplateUser { return errors.New("user template mismatch") } return nil } func compareEventActionCmdConfigFields(expected, actual dataprovider.EventActionCommandConfig) error { if expected.Cmd != actual.Cmd { return errors.New("command mismatch") } if expected.Timeout != actual.Timeout { return errors.New("cmd timeout mismatch") } if len(expected.Args) != len(actual.Args) { return errors.New("cmd args mismatch") } for _, v := range expected.Args { if !util.Contains(actual.Args, v) { return errors.New("cmd args content mismatch") } } if err := compareKeyValues(expected.EnvVars, actual.EnvVars); err != nil { return errors.New("cmd env vars mismatch") } return nil } func compareEventActionDataRetentionFields(expected, actual dataprovider.EventActionDataRetentionConfig) error { if len(expected.Folders) != len(actual.Folders) { return errors.New("retention folders mismatch") } for _, f1 := range expected.Folders { found := false for _, f2 := range actual.Folders { if f1.Path == f2.Path { found = true if f1.Retention != f2.Retention { return fmt.Errorf("retention mismatch for folder %s", f1.Path) } if f1.DeleteEmptyDirs != f2.DeleteEmptyDirs { return fmt.Errorf("delete_empty_dirs mismatch for folder %s", f1.Path) } if f1.IgnoreUserPermissions != f2.IgnoreUserPermissions { return fmt.Errorf("ignore_user_permissions mismatch for folder %s", f1.Path) } break } } if !found { return errors.New("retention folders mismatch") } } return nil } func compareEqualGroupSettingsFields(expected sdk.BaseGroupUserSettings, actual sdk.BaseGroupUserSettings) error { if expected.HomeDir != actual.HomeDir { return errors.New("home dir 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 expected.UploadBandwidth != actual.UploadBandwidth { return errors.New("UploadBandwidth mismatch") } if expected.DownloadBandwidth != actual.DownloadBandwidth { return errors.New("DownloadBandwidth mismatch") } if expected.UploadDataTransfer != actual.UploadDataTransfer { return errors.New("upload_data_transfer mismatch") } if expected.DownloadDataTransfer != actual.DownloadDataTransfer { return errors.New("download_data_transfer mismatch") } if expected.TotalDataTransfer != actual.TotalDataTransfer { return errors.New("total_data_transfer mismatch") } if expected.ExpiresIn != actual.ExpiresIn { return errors.New("expires_in mismatch") } return compareUserPermissions(expected.Permissions, actual.Permissions) } func compareEqualsUserFields(expected *dataprovider.User, actual *dataprovider.User) error { if dataprovider.ConvertName(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 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") } if expected.Role != actual.Role { return errors.New("role mismatch") } return compareQuotaUserFields(expected, actual) } func compareQuotaUserFields(expected *dataprovider.User, actual *dataprovider.User) error { if expected.QuotaSize != actual.QuotaSize { return errors.New("QuotaSize mismatch") } if expected.QuotaFiles != actual.QuotaFiles { return errors.New("QuotaFiles mismatch") } if expected.UploadDataTransfer != actual.UploadDataTransfer { return errors.New("upload_data_transfer mismatch") } if expected.DownloadDataTransfer != actual.DownloadDataTransfer { return errors.New("download_data_transfer mismatch") } if expected.TotalDataTransfer != actual.TotalDataTransfer { return errors.New("total_data_transfer 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 addUpdateUserQueryParams(rawurl, disconnect string) (*url.URL, error) { url, err := url.Parse(rawurl) if err != nil { return nil, err } q := url.Query() if disconnect != "" { q.Add("disconnect", disconnect) } url.RawQuery = q.Encode() return url, err }