| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763 |
- package config_test
- import (
- "encoding/json"
- "io/ioutil"
- "os"
- "path/filepath"
- "strings"
- "testing"
- "github.com/spf13/viper"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
- "github.com/drakkan/sftpgo/common"
- "github.com/drakkan/sftpgo/config"
- "github.com/drakkan/sftpgo/dataprovider"
- "github.com/drakkan/sftpgo/ftpd"
- "github.com/drakkan/sftpgo/httpclient"
- "github.com/drakkan/sftpgo/httpd"
- "github.com/drakkan/sftpgo/sftpd"
- "github.com/drakkan/sftpgo/utils"
- "github.com/drakkan/sftpgo/webdavd"
- )
- const (
- tempConfigName = "temp"
- )
- func reset() {
- viper.Reset()
- config.Init()
- }
- func TestLoadConfigTest(t *testing.T) {
- reset()
- configDir := ".."
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- assert.NotEqual(t, httpd.Conf{}, config.GetHTTPConfig())
- assert.NotEqual(t, dataprovider.Config{}, config.GetProviderConf())
- assert.NotEqual(t, sftpd.Configuration{}, config.GetSFTPDConfig())
- assert.NotEqual(t, httpclient.Config{}, config.GetHTTPConfig())
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, []byte("{invalid json}"), os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, []byte("{\"sftpd\": {\"bind_port\": \"a\"}}"), os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.Error(t, err)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestLoadConfigFileNotFound(t *testing.T) {
- reset()
- viper.SetConfigName("configfile")
- err := config.LoadConfig(os.TempDir(), "")
- assert.NoError(t, err)
- }
- func TestEmptyBanner(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- sftpdConf := config.GetSFTPDConfig()
- sftpdConf.Banner = " "
- c := make(map[string]sftpd.Configuration)
- c["sftpd"] = sftpdConf
- jsonConf, _ := json.Marshal(c)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- sftpdConf = config.GetSFTPDConfig()
- assert.NotEmpty(t, strings.TrimSpace(sftpdConf.Banner))
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- ftpdConf := config.GetFTPDConfig()
- ftpdConf.Banner = " "
- c1 := make(map[string]ftpd.Configuration)
- c1["ftpd"] = ftpdConf
- jsonConf, _ = json.Marshal(c1)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- ftpdConf = config.GetFTPDConfig()
- assert.NotEmpty(t, strings.TrimSpace(ftpdConf.Banner))
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestInvalidUploadMode(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- commonConf := config.GetCommonConfig()
- commonConf.UploadMode = 10
- c := make(map[string]common.Configuration)
- c["common"] = commonConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- assert.Equal(t, 0, config.GetCommonConfig().UploadMode)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestInvalidExternalAuthScope(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.ExternalAuthScope = 10
- c := make(map[string]dataprovider.Config)
- c["data_provider"] = providerConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- assert.Equal(t, 0, config.GetProviderConf().ExternalAuthScope)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestInvalidCredentialsPath(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.CredentialsPath = ""
- c := make(map[string]dataprovider.Config)
- c["data_provider"] = providerConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- assert.Equal(t, "credentials", config.GetProviderConf().CredentialsPath)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestInvalidProxyProtocol(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- commonConf := config.GetCommonConfig()
- commonConf.ProxyProtocol = 10
- c := make(map[string]common.Configuration)
- c["common"] = commonConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- assert.Equal(t, 0, config.GetCommonConfig().ProxyProtocol)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestInvalidUsersBaseDir(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.UsersBaseDir = "."
- c := make(map[string]dataprovider.Config)
- c["data_provider"] = providerConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- assert.Empty(t, config.GetProviderConf().UsersBaseDir)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestCommonParamsCompatibility(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- sftpdConf := config.GetSFTPDConfig()
- sftpdConf.IdleTimeout = 21 //nolint:staticcheck
- sftpdConf.Actions.Hook = "http://hook"
- sftpdConf.Actions.ExecuteOn = []string{"upload"}
- sftpdConf.SetstatMode = 1 //nolint:staticcheck
- sftpdConf.UploadMode = common.UploadModeAtomicWithResume //nolint:staticcheck
- sftpdConf.ProxyProtocol = 1 //nolint:staticcheck
- sftpdConf.ProxyAllowed = []string{"192.168.1.1"} //nolint:staticcheck
- c := make(map[string]sftpd.Configuration)
- c["sftpd"] = sftpdConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- commonConf := config.GetCommonConfig()
- assert.Equal(t, 21, commonConf.IdleTimeout)
- assert.Equal(t, "http://hook", commonConf.Actions.Hook)
- assert.Len(t, commonConf.Actions.ExecuteOn, 1)
- assert.True(t, utils.IsStringInSlice("upload", commonConf.Actions.ExecuteOn))
- assert.Equal(t, 1, commonConf.SetstatMode)
- assert.Equal(t, 1, commonConf.ProxyProtocol)
- assert.Len(t, commonConf.ProxyAllowed, 1)
- assert.True(t, utils.IsStringInSlice("192.168.1.1", commonConf.ProxyAllowed))
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestHostKeyCompatibility(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- sftpdConf := config.GetSFTPDConfig()
- sftpdConf.Keys = []sftpd.Key{ //nolint:staticcheck
- {
- PrivateKey: "rsa",
- },
- {
- PrivateKey: "ecdsa",
- },
- }
- c := make(map[string]sftpd.Configuration)
- c["sftpd"] = sftpdConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- sftpdConf = config.GetSFTPDConfig()
- assert.Equal(t, 2, len(sftpdConf.HostKeys))
- assert.True(t, utils.IsStringInSlice("rsa", sftpdConf.HostKeys))
- assert.True(t, utils.IsStringInSlice("ecdsa", sftpdConf.HostKeys))
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestSetGetConfig(t *testing.T) {
- reset()
- sftpdConf := config.GetSFTPDConfig()
- sftpdConf.MaxAuthTries = 10
- config.SetSFTPDConfig(sftpdConf)
- assert.Equal(t, sftpdConf.MaxAuthTries, config.GetSFTPDConfig().MaxAuthTries)
- dataProviderConf := config.GetProviderConf()
- dataProviderConf.Host = "test host"
- config.SetProviderConf(dataProviderConf)
- assert.Equal(t, dataProviderConf.Host, config.GetProviderConf().Host)
- httpdConf := config.GetHTTPDConfig()
- httpdConf.Bindings = append(httpdConf.Bindings, httpd.Binding{Address: "0.0.0.0"})
- config.SetHTTPDConfig(httpdConf)
- assert.Equal(t, httpdConf.Bindings[0].Address, config.GetHTTPDConfig().Bindings[0].Address)
- commonConf := config.GetCommonConfig()
- commonConf.IdleTimeout = 10
- config.SetCommonConfig(commonConf)
- assert.Equal(t, commonConf.IdleTimeout, config.GetCommonConfig().IdleTimeout)
- ftpdConf := config.GetFTPDConfig()
- ftpdConf.CertificateFile = "cert"
- ftpdConf.CertificateKeyFile = "key"
- config.SetFTPDConfig(ftpdConf)
- assert.Equal(t, ftpdConf.CertificateFile, config.GetFTPDConfig().CertificateFile)
- assert.Equal(t, ftpdConf.CertificateKeyFile, config.GetFTPDConfig().CertificateKeyFile)
- webDavConf := config.GetWebDAVDConfig()
- webDavConf.CertificateFile = "dav_cert"
- webDavConf.CertificateKeyFile = "dav_key"
- config.SetWebDAVDConfig(webDavConf)
- assert.Equal(t, webDavConf.CertificateFile, config.GetWebDAVDConfig().CertificateFile)
- assert.Equal(t, webDavConf.CertificateKeyFile, config.GetWebDAVDConfig().CertificateKeyFile)
- kmsConf := config.GetKMSConfig()
- kmsConf.Secrets.MasterKeyPath = "apath"
- kmsConf.Secrets.URL = "aurl"
- config.SetKMSConfig(kmsConf)
- assert.Equal(t, kmsConf.Secrets.MasterKeyPath, config.GetKMSConfig().Secrets.MasterKeyPath)
- assert.Equal(t, kmsConf.Secrets.URL, config.GetKMSConfig().Secrets.URL)
- telemetryConf := config.GetTelemetryConfig()
- telemetryConf.BindPort = 10001
- telemetryConf.BindAddress = "0.0.0.0"
- config.SetTelemetryConfig(telemetryConf)
- assert.Equal(t, telemetryConf.BindPort, config.GetTelemetryConfig().BindPort)
- assert.Equal(t, telemetryConf.BindAddress, config.GetTelemetryConfig().BindAddress)
- }
- func TestServiceToStart(t *testing.T) {
- reset()
- configDir := ".."
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- assert.True(t, config.HasServicesToStart())
- sftpdConf := config.GetSFTPDConfig()
- sftpdConf.Bindings[0].Port = 0
- config.SetSFTPDConfig(sftpdConf)
- assert.False(t, config.HasServicesToStart())
- ftpdConf := config.GetFTPDConfig()
- ftpdConf.Bindings[0].Port = 2121
- config.SetFTPDConfig(ftpdConf)
- assert.True(t, config.HasServicesToStart())
- ftpdConf.Bindings[0].Port = 0
- config.SetFTPDConfig(ftpdConf)
- webdavdConf := config.GetWebDAVDConfig()
- webdavdConf.Bindings[0].Port = 9000
- config.SetWebDAVDConfig(webdavdConf)
- assert.True(t, config.HasServicesToStart())
- webdavdConf.Bindings[0].Port = 0
- config.SetWebDAVDConfig(webdavdConf)
- assert.False(t, config.HasServicesToStart())
- sftpdConf.Bindings[0].Port = 2022
- config.SetSFTPDConfig(sftpdConf)
- assert.True(t, config.HasServicesToStart())
- }
- func TestSFTPDBindingsCompatibility(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- sftpdConf := config.GetSFTPDConfig()
- require.Len(t, sftpdConf.Bindings, 1)
- sftpdConf.Bindings = nil
- sftpdConf.BindPort = 9022 //nolint:staticcheck
- sftpdConf.BindAddress = "127.0.0.1" //nolint:staticcheck
- c := make(map[string]sftpd.Configuration)
- c["sftpd"] = sftpdConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- sftpdConf = config.GetSFTPDConfig()
- // the default binding should be replaced with the deprecated configuration
- require.Len(t, sftpdConf.Bindings, 1)
- require.Equal(t, 9022, sftpdConf.Bindings[0].Port)
- require.Equal(t, "127.0.0.1", sftpdConf.Bindings[0].Address)
- require.True(t, sftpdConf.Bindings[0].ApplyProxyConfig)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- sftpdConf = config.GetSFTPDConfig()
- require.Len(t, sftpdConf.Bindings, 1)
- require.Equal(t, 9022, sftpdConf.Bindings[0].Port)
- require.Equal(t, "127.0.0.1", sftpdConf.Bindings[0].Address)
- require.True(t, sftpdConf.Bindings[0].ApplyProxyConfig)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestFTPDBindingsCompatibility(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- ftpdConf := config.GetFTPDConfig()
- require.Len(t, ftpdConf.Bindings, 1)
- ftpdConf.Bindings = nil
- ftpdConf.BindPort = 9022 //nolint:staticcheck
- ftpdConf.BindAddress = "127.1.0.1" //nolint:staticcheck
- ftpdConf.ForcePassiveIP = "127.1.1.1" //nolint:staticcheck
- ftpdConf.TLSMode = 2 //nolint:staticcheck
- c := make(map[string]ftpd.Configuration)
- c["ftpd"] = ftpdConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- ftpdConf = config.GetFTPDConfig()
- // the default binding should be replaced with the deprecated configuration
- require.Len(t, ftpdConf.Bindings, 1)
- require.Equal(t, 9022, ftpdConf.Bindings[0].Port)
- require.Equal(t, "127.1.0.1", ftpdConf.Bindings[0].Address)
- require.True(t, ftpdConf.Bindings[0].ApplyProxyConfig)
- require.Equal(t, 2, ftpdConf.Bindings[0].TLSMode)
- require.Equal(t, "127.1.1.1", ftpdConf.Bindings[0].ForcePassiveIP)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestWebDAVDBindingsCompatibility(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- webdavConf := config.GetWebDAVDConfig()
- require.Len(t, webdavConf.Bindings, 1)
- webdavConf.Bindings = nil
- webdavConf.BindPort = 9080 //nolint:staticcheck
- webdavConf.BindAddress = "127.0.0.1" //nolint:staticcheck
- c := make(map[string]webdavd.Configuration)
- c["webdavd"] = webdavConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- webdavConf = config.GetWebDAVDConfig()
- // the default binding should be replaced with the deprecated configuration
- require.Len(t, webdavConf.Bindings, 1)
- require.Equal(t, 9080, webdavConf.Bindings[0].Port)
- require.Equal(t, "127.0.0.1", webdavConf.Bindings[0].Address)
- require.False(t, webdavConf.Bindings[0].EnableHTTPS)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestHTTPDBindingsCompatibility(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- httpdConf := config.GetHTTPDConfig()
- require.Len(t, httpdConf.Bindings, 1)
- httpdConf.Bindings = nil
- httpdConf.BindPort = 9080 //nolint:staticcheck
- httpdConf.BindAddress = "127.1.1.1" //nolint:staticcheck
- c := make(map[string]httpd.Conf)
- c["httpd"] = httpdConf
- jsonConf, err := json.Marshal(c)
- assert.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- assert.NoError(t, err)
- httpdConf = config.GetHTTPDConfig()
- // the default binding should be replaced with the deprecated configuration
- require.Len(t, httpdConf.Bindings, 1)
- require.Equal(t, 9080, httpdConf.Bindings[0].Port)
- require.Equal(t, "127.1.1.1", httpdConf.Bindings[0].Address)
- require.False(t, httpdConf.Bindings[0].EnableHTTPS)
- require.True(t, httpdConf.Bindings[0].EnableWebAdmin)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- }
- func TestSFTPDBindingsFromEnv(t *testing.T) {
- reset()
- os.Setenv("SFTPGO_SFTPD__BINDINGS__0__ADDRESS", "127.0.0.1")
- os.Setenv("SFTPGO_SFTPD__BINDINGS__0__PORT", "2200")
- os.Setenv("SFTPGO_SFTPD__BINDINGS__0__APPLY_PROXY_CONFIG", "false")
- os.Setenv("SFTPGO_SFTPD__BINDINGS__3__ADDRESS", "127.0.1.1")
- os.Setenv("SFTPGO_SFTPD__BINDINGS__3__PORT", "2203")
- os.Setenv("SFTPGO_SFTPD__BINDINGS__3__APPLY_PROXY_CONFIG", "1")
- t.Cleanup(func() {
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__0__ADDRESS")
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__0__PORT")
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__0__APPLY_PROXY_CONFIG")
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__3__ADDRESS")
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__3__PORT")
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__3__APPLY_PROXY_CONFIG")
- })
- configDir := ".."
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- bindings := config.GetSFTPDConfig().Bindings
- require.Len(t, bindings, 2)
- require.Equal(t, 2200, bindings[0].Port)
- require.Equal(t, "127.0.0.1", bindings[0].Address)
- require.False(t, bindings[0].ApplyProxyConfig)
- require.Equal(t, 2203, bindings[1].Port)
- require.Equal(t, "127.0.1.1", bindings[1].Address)
- require.True(t, bindings[1].ApplyProxyConfig)
- }
- func TestFTPDBindingsFromEnv(t *testing.T) {
- reset()
- os.Setenv("SFTPGO_FTPD__BINDINGS__0__ADDRESS", "127.0.0.1")
- os.Setenv("SFTPGO_FTPD__BINDINGS__0__PORT", "2200")
- os.Setenv("SFTPGO_FTPD__BINDINGS__0__APPLY_PROXY_CONFIG", "f")
- os.Setenv("SFTPGO_FTPD__BINDINGS__0__TLS_MODE", "2")
- os.Setenv("SFTPGO_FTPD__BINDINGS__0__FORCE_PASSIVE_IP", "127.0.1.2")
- os.Setenv("SFTPGO_FTPD__BINDINGS__9__ADDRESS", "127.0.1.1")
- os.Setenv("SFTPGO_FTPD__BINDINGS__9__PORT", "2203")
- os.Setenv("SFTPGO_FTPD__BINDINGS__9__APPLY_PROXY_CONFIG", "t")
- os.Setenv("SFTPGO_FTPD__BINDINGS__9__TLS_MODE", "1")
- os.Setenv("SFTPGO_FTPD__BINDINGS__9__FORCE_PASSIVE_IP", "127.0.1.1")
- os.Setenv("SFTPGO_FTPD__BINDINGS__9__CLIENT_AUTH_TYPE", "1")
- t.Cleanup(func() {
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__0__ADDRESS")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__0__PORT")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__0__APPLY_PROXY_CONFIG")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__0__TLS_MODE")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__0__FORCE_PASSIVE_IP")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__9__ADDRESS")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__9__PORT")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__9__APPLY_PROXY_CONFIG")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__9__TLS_MODE")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__9__FORCE_PASSIVE_IP")
- os.Unsetenv("SFTPGO_FTPD__BINDINGS__9__CLIENT_AUTH_TYPE")
- })
- configDir := ".."
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- bindings := config.GetFTPDConfig().Bindings
- require.Len(t, bindings, 2)
- require.Equal(t, 2200, bindings[0].Port)
- require.Equal(t, "127.0.0.1", bindings[0].Address)
- require.False(t, bindings[0].ApplyProxyConfig)
- require.Equal(t, 2, bindings[0].TLSMode)
- require.Equal(t, "127.0.1.2", bindings[0].ForcePassiveIP)
- require.Equal(t, 0, bindings[0].ClientAuthType)
- require.Equal(t, 2203, bindings[1].Port)
- require.Equal(t, "127.0.1.1", bindings[1].Address)
- require.True(t, bindings[1].ApplyProxyConfig)
- require.Equal(t, 1, bindings[1].TLSMode)
- require.Equal(t, "127.0.1.1", bindings[1].ForcePassiveIP)
- require.Equal(t, 1, bindings[1].ClientAuthType)
- }
- func TestWebDAVBindingsFromEnv(t *testing.T) {
- reset()
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__1__ADDRESS", "127.0.0.1")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__1__PORT", "8000")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__1__ENABLE_HTTPS", "0")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__2__ADDRESS", "127.0.1.1")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__2__PORT", "9000")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__2__ENABLE_HTTPS", "1")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__2__CLIENT_AUTH_TYPE", "1")
- t.Cleanup(func() {
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__1__ADDRESS")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__1__PORT")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__1__ENABLE_HTTPS")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__2__ADDRESS")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__2__PORT")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__2__ENABLE_HTTPS")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__2__CLIENT_AUTH_TYPE")
- })
- configDir := ".."
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- bindings := config.GetWebDAVDConfig().Bindings
- require.Len(t, bindings, 3)
- require.Equal(t, 0, bindings[0].Port)
- require.Empty(t, bindings[0].Address)
- require.False(t, bindings[0].EnableHTTPS)
- require.Equal(t, 8000, bindings[1].Port)
- require.Equal(t, "127.0.0.1", bindings[1].Address)
- require.False(t, bindings[1].EnableHTTPS)
- require.Equal(t, 0, bindings[1].ClientAuthType)
- require.Equal(t, 9000, bindings[2].Port)
- require.Equal(t, "127.0.1.1", bindings[2].Address)
- require.True(t, bindings[2].EnableHTTPS)
- require.Equal(t, 1, bindings[2].ClientAuthType)
- }
- func TestHTTPDBindingsFromEnv(t *testing.T) {
- reset()
- sockPath := filepath.Clean(os.TempDir())
- os.Setenv("SFTPGO_HTTPD__BINDINGS__0__ADDRESS", sockPath)
- os.Setenv("SFTPGO_HTTPD__BINDINGS__0__PORT", "0")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__1__ADDRESS", "127.0.0.1")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__1__PORT", "8000")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__1__ENABLE_HTTPS", "0")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__1__ENABLE_WEB_ADMIN", "1")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__2__ADDRESS", "127.0.1.1")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__2__PORT", "9000")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__2__ENABLE_WEB_ADMIN", "0")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__2__ENABLE_HTTPS", "1")
- os.Setenv("SFTPGO_HTTPD__BINDINGS__2__CLIENT_AUTH_TYPE", "1")
- t.Cleanup(func() {
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__0__ADDRESS")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__0__PORT")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__1__ADDRESS")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__1__PORT")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__1__ENABLE_HTTPS")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__1__ENABLE_WEB_ADMIN")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__2__ADDRESS")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__2__PORT")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__2__ENABLE_HTTPS")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__2__ENABLE_WEB_ADMIN")
- os.Unsetenv("SFTPGO_HTTPD__BINDINGS__2__CLIENT_AUTH_TYPE")
- })
- configDir := ".."
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- bindings := config.GetHTTPDConfig().Bindings
- require.Len(t, bindings, 3)
- require.Equal(t, 0, bindings[0].Port)
- require.Equal(t, sockPath, bindings[0].Address)
- require.False(t, bindings[0].EnableHTTPS)
- require.True(t, bindings[0].EnableWebAdmin)
- require.Equal(t, 8000, bindings[1].Port)
- require.Equal(t, "127.0.0.1", bindings[1].Address)
- require.False(t, bindings[1].EnableHTTPS)
- require.True(t, bindings[1].EnableWebAdmin)
- require.Equal(t, 9000, bindings[2].Port)
- require.Equal(t, "127.0.1.1", bindings[2].Address)
- require.True(t, bindings[2].EnableHTTPS)
- require.False(t, bindings[2].EnableWebAdmin)
- require.Equal(t, 1, bindings[2].ClientAuthType)
- }
- func TestHTTPClientCertificatesFromEnv(t *testing.T) {
- reset()
- configDir := ".."
- confName := tempConfigName + ".json"
- configFilePath := filepath.Join(configDir, confName)
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- httpConf := config.GetHTTPConfig()
- httpConf.Certificates = append(httpConf.Certificates, httpclient.TLSKeyPair{
- Cert: "cert",
- Key: "key",
- })
- c := make(map[string]httpclient.Config)
- c["http"] = httpConf
- jsonConf, err := json.Marshal(c)
- require.NoError(t, err)
- err = ioutil.WriteFile(configFilePath, jsonConf, os.ModePerm)
- require.NoError(t, err)
- err = config.LoadConfig(configDir, confName)
- require.NoError(t, err)
- require.Len(t, config.GetHTTPConfig().Certificates, 1)
- require.Equal(t, "cert", config.GetHTTPConfig().Certificates[0].Cert)
- require.Equal(t, "key", config.GetHTTPConfig().Certificates[0].Key)
- os.Setenv("SFTPGO_HTTP__CERTIFICATES__0__CERT", "cert0")
- os.Setenv("SFTPGO_HTTP__CERTIFICATES__0__KEY", "key0")
- os.Setenv("SFTPGO_HTTP__CERTIFICATES__8__CERT", "cert8")
- os.Setenv("SFTPGO_HTTP__CERTIFICATES__9__CERT", "cert9")
- os.Setenv("SFTPGO_HTTP__CERTIFICATES__9__KEY", "key9")
- t.Cleanup(func() {
- os.Unsetenv("SFTPGO_HTTP__CERTIFICATES__0__CERT")
- os.Unsetenv("SFTPGO_HTTP__CERTIFICATES__0__KEY")
- os.Unsetenv("SFTPGO_HTTP__CERTIFICATES__8__CERT")
- os.Unsetenv("SFTPGO_HTTP__CERTIFICATES__9__CERT")
- os.Unsetenv("SFTPGO_HTTP__CERTIFICATES__9__KEY")
- })
- err = config.LoadConfig(configDir, confName)
- require.NoError(t, err)
- require.Len(t, config.GetHTTPConfig().Certificates, 2)
- require.Equal(t, "cert0", config.GetHTTPConfig().Certificates[0].Cert)
- require.Equal(t, "key0", config.GetHTTPConfig().Certificates[0].Key)
- require.Equal(t, "cert9", config.GetHTTPConfig().Certificates[1].Cert)
- require.Equal(t, "key9", config.GetHTTPConfig().Certificates[1].Key)
- err = os.Remove(configFilePath)
- assert.NoError(t, err)
- config.Init()
- err = config.LoadConfig(configDir, "")
- require.NoError(t, err)
- require.Len(t, config.GetHTTPConfig().Certificates, 2)
- require.Equal(t, "cert0", config.GetHTTPConfig().Certificates[0].Cert)
- require.Equal(t, "key0", config.GetHTTPConfig().Certificates[0].Key)
- require.Equal(t, "cert9", config.GetHTTPConfig().Certificates[1].Cert)
- require.Equal(t, "key9", config.GetHTTPConfig().Certificates[1].Key)
- }
- func TestConfigFromEnv(t *testing.T) {
- reset()
- os.Setenv("SFTPGO_SFTPD__BINDINGS__0__ADDRESS", "127.0.0.1")
- os.Setenv("SFTPGO_WEBDAVD__BINDINGS__0__PORT", "12000")
- os.Setenv("SFTPGO_DATA_PROVIDER__PASSWORD_HASHING__ARGON2_OPTIONS__ITERATIONS", "41")
- os.Setenv("SFTPGO_DATA_PROVIDER__POOL_SIZE", "10")
- os.Setenv("SFTPGO_DATA_PROVIDER__ACTIONS__EXECUTE_ON", "add")
- os.Setenv("SFTPGO_KMS__SECRETS__URL", "local")
- os.Setenv("SFTPGO_KMS__SECRETS__MASTER_KEY_PATH", "path")
- t.Cleanup(func() {
- os.Unsetenv("SFTPGO_SFTPD__BINDINGS__0__ADDRESS")
- os.Unsetenv("SFTPGO_WEBDAVD__BINDINGS__0__PORT")
- os.Unsetenv("SFTPGO_DATA_PROVIDER__PASSWORD_HASHING__ARGON2_OPTIONS__ITERATIONS")
- os.Unsetenv("SFTPGO_DATA_PROVIDER__POOL_SIZE")
- os.Unsetenv("SFTPGO_DATA_PROVIDER__ACTIONS__EXECUTE_ON")
- os.Unsetenv("SFTPGO_KMS__SECRETS__URL")
- os.Unsetenv("SFTPGO_KMS__SECRETS__MASTER_KEY_PATH")
- })
- err := config.LoadConfig(".", "invalid config")
- assert.NoError(t, err)
- sftpdConfig := config.GetSFTPDConfig()
- assert.Equal(t, "127.0.0.1", sftpdConfig.Bindings[0].Address)
- assert.Equal(t, 12000, config.GetWebDAVDConfig().Bindings[0].Port)
- dataProviderConf := config.GetProviderConf()
- assert.Equal(t, uint32(41), dataProviderConf.PasswordHashing.Argon2Options.Iterations)
- assert.Equal(t, 10, dataProviderConf.PoolSize)
- assert.Len(t, dataProviderConf.Actions.ExecuteOn, 1)
- assert.Contains(t, dataProviderConf.Actions.ExecuteOn, "add")
- kmsConfig := config.GetKMSConfig()
- assert.Equal(t, "local", kmsConfig.Secrets.URL)
- assert.Equal(t, "path", kmsConfig.Secrets.MasterKeyPath)
- }
|