| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652 | 
							- package common
 
- import (
 
- 	"fmt"
 
- 	"net"
 
- 	"net/http"
 
- 	"os"
 
- 	"os/exec"
 
- 	"path/filepath"
 
- 	"runtime"
 
- 	"strings"
 
- 	"sync/atomic"
 
- 	"testing"
 
- 	"time"
 
- 	"github.com/rs/zerolog"
 
- 	"github.com/spf13/viper"
 
- 	"github.com/stretchr/testify/assert"
 
- 	"github.com/stretchr/testify/require"
 
- 	"github.com/drakkan/sftpgo/dataprovider"
 
- 	"github.com/drakkan/sftpgo/httpclient"
 
- 	"github.com/drakkan/sftpgo/kms"
 
- 	"github.com/drakkan/sftpgo/logger"
 
- 	"github.com/drakkan/sftpgo/utils"
 
- 	"github.com/drakkan/sftpgo/vfs"
 
- )
 
- const (
 
- 	logSenderTest    = "common_test"
 
- 	httpAddr         = "127.0.0.1:9999"
 
- 	httpProxyAddr    = "127.0.0.1:7777"
 
- 	configDir        = ".."
 
- 	osWindows        = "windows"
 
- 	userTestUsername = "common_test_username"
 
- 	userTestPwd      = "common_test_pwd"
 
- )
 
- type providerConf struct {
 
- 	Config dataprovider.Config `json:"data_provider" mapstructure:"data_provider"`
 
- }
 
- type fakeConnection struct {
 
- 	*BaseConnection
 
- 	command string
 
- }
 
- func (c *fakeConnection) AddUser(user dataprovider.User) error {
 
- 	fs, err := user.GetFilesystem(c.GetID())
 
- 	if err != nil {
 
- 		return err
 
- 	}
 
- 	c.BaseConnection.User = user
 
- 	c.BaseConnection.Fs = fs
 
- 	return nil
 
- }
 
- func (c *fakeConnection) Disconnect() error {
 
- 	Connections.Remove(c.GetID())
 
- 	return nil
 
- }
 
- func (c *fakeConnection) GetClientVersion() string {
 
- 	return ""
 
- }
 
- func (c *fakeConnection) GetCommand() string {
 
- 	return c.command
 
- }
 
- func (c *fakeConnection) GetRemoteAddress() string {
 
- 	return ""
 
- }
 
- type customNetConn struct {
 
- 	net.Conn
 
- 	id       string
 
- 	isClosed bool
 
- }
 
- func (c *customNetConn) Close() error {
 
- 	Connections.RemoveSSHConnection(c.id)
 
- 	c.isClosed = true
 
- 	return c.Conn.Close()
 
- }
 
- func TestMain(m *testing.M) {
 
- 	logfilePath := "common_test.log"
 
- 	logger.InitLogger(logfilePath, 5, 1, 28, false, zerolog.DebugLevel)
 
- 	viper.SetEnvPrefix("sftpgo")
 
- 	replacer := strings.NewReplacer(".", "__")
 
- 	viper.SetEnvKeyReplacer(replacer)
 
- 	viper.SetConfigName("sftpgo")
 
- 	viper.AutomaticEnv()
 
- 	viper.AllowEmptyEnv(true)
 
- 	driver, err := initializeDataprovider(-1)
 
- 	if err != nil {
 
- 		logger.WarnToConsole("error initializing data provider: %v", err)
 
- 		os.Exit(1)
 
- 	}
 
- 	logger.InfoToConsole("Starting COMMON tests, provider: %v", driver)
 
- 	err = Initialize(Configuration{})
 
- 	if err != nil {
 
- 		logger.WarnToConsole("error initializing common: %v", err)
 
- 		os.Exit(1)
 
- 	}
 
- 	httpConfig := httpclient.Config{
 
- 		Timeout: 5,
 
- 	}
 
- 	httpConfig.Initialize(configDir)
 
- 	go func() {
 
- 		// start a test HTTP server to receive action notifications
 
- 		http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 
- 			fmt.Fprintf(w, "OK\n")
 
- 		})
 
- 		http.HandleFunc("/404", func(w http.ResponseWriter, r *http.Request) {
 
- 			w.WriteHeader(http.StatusNotFound)
 
- 			fmt.Fprintf(w, "Not found\n")
 
- 		})
 
- 		if err := http.ListenAndServe(httpAddr, nil); err != nil {
 
- 			logger.ErrorToConsole("could not start HTTP notification server: %v", err)
 
- 			os.Exit(1)
 
- 		}
 
- 	}()
 
- 	go func() {
 
- 		Config.ProxyProtocol = 2
 
- 		listener, err := net.Listen("tcp", httpProxyAddr)
 
- 		if err != nil {
 
- 			logger.ErrorToConsole("error creating listener for proxy protocol server: %v", err)
 
- 			os.Exit(1)
 
- 		}
 
- 		proxyListener, err := Config.GetProxyListener(listener)
 
- 		if err != nil {
 
- 			logger.ErrorToConsole("error creating proxy protocol listener: %v", err)
 
- 			os.Exit(1)
 
- 		}
 
- 		Config.ProxyProtocol = 0
 
- 		s := &http.Server{}
 
- 		if err := s.Serve(proxyListener); err != nil {
 
- 			logger.ErrorToConsole("could not start HTTP proxy protocol server: %v", err)
 
- 			os.Exit(1)
 
- 		}
 
- 	}()
 
- 	waitTCPListening(httpAddr)
 
- 	waitTCPListening(httpProxyAddr)
 
- 	exitCode := m.Run()
 
- 	os.Remove(logfilePath) //nolint:errcheck
 
- 	os.Exit(exitCode)
 
- }
 
- func waitTCPListening(address string) {
 
- 	for {
 
- 		conn, err := net.Dial("tcp", address)
 
- 		if err != nil {
 
- 			logger.WarnToConsole("tcp server %v not listening: %v\n", address, err)
 
- 			time.Sleep(100 * time.Millisecond)
 
- 			continue
 
- 		}
 
- 		logger.InfoToConsole("tcp server %v now listening\n", address)
 
- 		conn.Close()
 
- 		break
 
- 	}
 
- }
 
- func initializeDataprovider(trackQuota int) (string, error) {
 
- 	configDir := ".."
 
- 	viper.AddConfigPath(configDir)
 
- 	if err := viper.ReadInConfig(); err != nil {
 
- 		return "", err
 
- 	}
 
- 	var cfg providerConf
 
- 	if err := viper.Unmarshal(&cfg); err != nil {
 
- 		return "", err
 
- 	}
 
- 	if trackQuota >= 0 && trackQuota <= 2 {
 
- 		cfg.Config.TrackQuota = trackQuota
 
- 	}
 
- 	return cfg.Config.Driver, dataprovider.Initialize(cfg.Config, configDir, true)
 
- }
 
- func closeDataprovider() error {
 
- 	return dataprovider.Close()
 
- }
 
- func TestSSHConnections(t *testing.T) {
 
- 	conn1, conn2 := net.Pipe()
 
- 	now := time.Now()
 
- 	sshConn1 := NewSSHConnection("id1", conn1)
 
- 	sshConn2 := NewSSHConnection("id2", conn2)
 
- 	sshConn3 := NewSSHConnection("id3", conn2)
 
- 	assert.Equal(t, "id1", sshConn1.GetID())
 
- 	assert.Equal(t, "id2", sshConn2.GetID())
 
- 	assert.Equal(t, "id3", sshConn3.GetID())
 
- 	sshConn1.UpdateLastActivity()
 
- 	assert.GreaterOrEqual(t, sshConn1.GetLastActivity().UnixNano(), now.UnixNano())
 
- 	Connections.AddSSHConnection(sshConn1)
 
- 	Connections.AddSSHConnection(sshConn2)
 
- 	Connections.AddSSHConnection(sshConn3)
 
- 	Connections.RLock()
 
- 	assert.Len(t, Connections.sshConnections, 3)
 
- 	Connections.RUnlock()
 
- 	Connections.RemoveSSHConnection(sshConn1.id)
 
- 	Connections.RLock()
 
- 	assert.Len(t, Connections.sshConnections, 2)
 
- 	assert.Equal(t, sshConn3.id, Connections.sshConnections[0].id)
 
- 	assert.Equal(t, sshConn2.id, Connections.sshConnections[1].id)
 
- 	Connections.RUnlock()
 
- 	Connections.RemoveSSHConnection(sshConn1.id)
 
- 	Connections.RLock()
 
- 	assert.Len(t, Connections.sshConnections, 2)
 
- 	assert.Equal(t, sshConn3.id, Connections.sshConnections[0].id)
 
- 	assert.Equal(t, sshConn2.id, Connections.sshConnections[1].id)
 
- 	Connections.RUnlock()
 
- 	Connections.RemoveSSHConnection(sshConn2.id)
 
- 	Connections.RLock()
 
- 	assert.Len(t, Connections.sshConnections, 1)
 
- 	assert.Equal(t, sshConn3.id, Connections.sshConnections[0].id)
 
- 	Connections.RUnlock()
 
- 	Connections.RemoveSSHConnection(sshConn3.id)
 
- 	Connections.RLock()
 
- 	assert.Len(t, Connections.sshConnections, 0)
 
- 	Connections.RUnlock()
 
- 	assert.NoError(t, sshConn1.Close())
 
- 	assert.NoError(t, sshConn2.Close())
 
- 	assert.NoError(t, sshConn3.Close())
 
- }
 
- func TestDefenderIntegration(t *testing.T) {
 
- 	// by default defender is nil
 
- 	configCopy := Config
 
- 	ip := "127.1.1.1"
 
- 	assert.Nil(t, ReloadDefender())
 
- 	AddDefenderEvent(ip, HostEventNoLoginTried)
 
- 	assert.False(t, IsBanned(ip))
 
- 	assert.Nil(t, GetDefenderBanTime(ip))
 
- 	assert.False(t, Unban(ip))
 
- 	assert.Equal(t, 0, GetDefenderScore(ip))
 
- 	Config.DefenderConfig = DefenderConfig{
 
- 		Enabled:          true,
 
- 		BanTime:          10,
 
- 		BanTimeIncrement: 50,
 
- 		Threshold:        0,
 
- 		ScoreInvalid:     2,
 
- 		ScoreValid:       1,
 
- 		ObservationTime:  15,
 
- 		EntriesSoftLimit: 100,
 
- 		EntriesHardLimit: 150,
 
- 	}
 
- 	err := Initialize(Config)
 
- 	assert.Error(t, err)
 
- 	Config.DefenderConfig.Threshold = 3
 
- 	err = Initialize(Config)
 
- 	assert.NoError(t, err)
 
- 	assert.Nil(t, ReloadDefender())
 
- 	AddDefenderEvent(ip, HostEventNoLoginTried)
 
- 	assert.False(t, IsBanned(ip))
 
- 	assert.Equal(t, 2, GetDefenderScore(ip))
 
- 	assert.False(t, Unban(ip))
 
- 	assert.Nil(t, GetDefenderBanTime(ip))
 
- 	AddDefenderEvent(ip, HostEventLoginFailed)
 
- 	assert.True(t, IsBanned(ip))
 
- 	assert.Equal(t, 0, GetDefenderScore(ip))
 
- 	assert.NotNil(t, GetDefenderBanTime(ip))
 
- 	assert.True(t, Unban(ip))
 
- 	assert.Nil(t, GetDefenderBanTime(ip))
 
- 	assert.False(t, Unban(ip))
 
- 	Config = configCopy
 
- }
 
- func TestMaxConnections(t *testing.T) {
 
- 	oldValue := Config.MaxTotalConnections
 
- 	Config.MaxTotalConnections = 1
 
- 	assert.True(t, Connections.IsNewConnectionAllowed())
 
- 	c := NewBaseConnection("id", ProtocolSFTP, dataprovider.User{}, nil)
 
- 	fakeConn := &fakeConnection{
 
- 		BaseConnection: c,
 
- 	}
 
- 	Connections.Add(fakeConn)
 
- 	assert.Len(t, Connections.GetStats(), 1)
 
- 	assert.False(t, Connections.IsNewConnectionAllowed())
 
- 	res := Connections.Close(fakeConn.GetID())
 
- 	assert.True(t, res)
 
- 	assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
 
- 	Config.MaxTotalConnections = oldValue
 
- }
 
- func TestIdleConnections(t *testing.T) {
 
- 	configCopy := Config
 
- 	Config.IdleTimeout = 1
 
- 	err := Initialize(Config)
 
- 	assert.NoError(t, err)
 
- 	conn1, conn2 := net.Pipe()
 
- 	customConn1 := &customNetConn{
 
- 		Conn: conn1,
 
- 		id:   "id1",
 
- 	}
 
- 	customConn2 := &customNetConn{
 
- 		Conn: conn2,
 
- 		id:   "id2",
 
- 	}
 
- 	sshConn1 := NewSSHConnection(customConn1.id, customConn1)
 
- 	sshConn2 := NewSSHConnection(customConn2.id, customConn2)
 
- 	username := "test_user"
 
- 	user := dataprovider.User{
 
- 		Username: username,
 
- 	}
 
- 	c := NewBaseConnection(sshConn1.id+"_1", ProtocolSFTP, user, nil)
 
- 	c.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
 
- 	fakeConn := &fakeConnection{
 
- 		BaseConnection: c,
 
- 	}
 
- 	// both ssh connections are expired but they should get removed only
 
- 	// if there is no associated connection
 
- 	sshConn1.lastActivity = c.lastActivity
 
- 	sshConn2.lastActivity = c.lastActivity
 
- 	Connections.AddSSHConnection(sshConn1)
 
- 	Connections.Add(fakeConn)
 
- 	assert.Equal(t, Connections.GetActiveSessions(username), 1)
 
- 	c = NewBaseConnection(sshConn2.id+"_1", ProtocolSSH, user, nil)
 
- 	fakeConn = &fakeConnection{
 
- 		BaseConnection: c,
 
- 	}
 
- 	Connections.AddSSHConnection(sshConn2)
 
- 	Connections.Add(fakeConn)
 
- 	assert.Equal(t, Connections.GetActiveSessions(username), 2)
 
- 	cFTP := NewBaseConnection("id2", ProtocolFTP, dataprovider.User{}, nil)
 
- 	cFTP.lastActivity = time.Now().UnixNano()
 
- 	fakeConn = &fakeConnection{
 
- 		BaseConnection: cFTP,
 
- 	}
 
- 	Connections.Add(fakeConn)
 
- 	assert.Equal(t, Connections.GetActiveSessions(username), 2)
 
- 	assert.Len(t, Connections.GetStats(), 3)
 
- 	Connections.RLock()
 
- 	assert.Len(t, Connections.sshConnections, 2)
 
- 	Connections.RUnlock()
 
- 	startIdleTimeoutTicker(100 * time.Millisecond)
 
- 	assert.Eventually(t, func() bool { return Connections.GetActiveSessions(username) == 1 }, 1*time.Second, 200*time.Millisecond)
 
- 	assert.Eventually(t, func() bool {
 
- 		Connections.RLock()
 
- 		defer Connections.RUnlock()
 
- 		return len(Connections.sshConnections) == 1
 
- 	}, 1*time.Second, 200*time.Millisecond)
 
- 	stopIdleTimeoutTicker()
 
- 	assert.Len(t, Connections.GetStats(), 2)
 
- 	c.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
 
- 	cFTP.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
 
- 	sshConn2.lastActivity = c.lastActivity
 
- 	startIdleTimeoutTicker(100 * time.Millisecond)
 
- 	assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 1*time.Second, 200*time.Millisecond)
 
- 	assert.Eventually(t, func() bool {
 
- 		Connections.RLock()
 
- 		defer Connections.RUnlock()
 
- 		return len(Connections.sshConnections) == 0
 
- 	}, 1*time.Second, 200*time.Millisecond)
 
- 	stopIdleTimeoutTicker()
 
- 	assert.True(t, customConn1.isClosed)
 
- 	assert.True(t, customConn2.isClosed)
 
- 	Config = configCopy
 
- }
 
- func TestCloseConnection(t *testing.T) {
 
- 	c := NewBaseConnection("id", ProtocolSFTP, dataprovider.User{}, nil)
 
- 	fakeConn := &fakeConnection{
 
- 		BaseConnection: c,
 
- 	}
 
- 	assert.True(t, Connections.IsNewConnectionAllowed())
 
- 	Connections.Add(fakeConn)
 
- 	assert.Len(t, Connections.GetStats(), 1)
 
- 	res := Connections.Close(fakeConn.GetID())
 
- 	assert.True(t, res)
 
- 	assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
 
- 	res = Connections.Close(fakeConn.GetID())
 
- 	assert.False(t, res)
 
- 	Connections.Remove(fakeConn.GetID())
 
- }
 
- func TestSwapConnection(t *testing.T) {
 
- 	c := NewBaseConnection("id", ProtocolFTP, dataprovider.User{}, nil)
 
- 	fakeConn := &fakeConnection{
 
- 		BaseConnection: c,
 
- 	}
 
- 	Connections.Add(fakeConn)
 
- 	if assert.Len(t, Connections.GetStats(), 1) {
 
- 		assert.Equal(t, "", Connections.GetStats()[0].Username)
 
- 	}
 
- 	c = NewBaseConnection("id", ProtocolFTP, dataprovider.User{
 
- 		Username: userTestUsername,
 
- 	}, nil)
 
- 	fakeConn = &fakeConnection{
 
- 		BaseConnection: c,
 
- 	}
 
- 	err := Connections.Swap(fakeConn)
 
- 	assert.NoError(t, err)
 
- 	if assert.Len(t, Connections.GetStats(), 1) {
 
- 		assert.Equal(t, userTestUsername, Connections.GetStats()[0].Username)
 
- 	}
 
- 	res := Connections.Close(fakeConn.GetID())
 
- 	assert.True(t, res)
 
- 	assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
 
- 	err = Connections.Swap(fakeConn)
 
- 	assert.Error(t, err)
 
- }
 
- func TestAtomicUpload(t *testing.T) {
 
- 	configCopy := Config
 
- 	Config.UploadMode = UploadModeStandard
 
- 	assert.False(t, Config.IsAtomicUploadEnabled())
 
- 	Config.UploadMode = UploadModeAtomic
 
- 	assert.True(t, Config.IsAtomicUploadEnabled())
 
- 	Config.UploadMode = UploadModeAtomicWithResume
 
- 	assert.True(t, Config.IsAtomicUploadEnabled())
 
- 	Config = configCopy
 
- }
 
- func TestConnectionStatus(t *testing.T) {
 
- 	username := "test_user"
 
- 	user := dataprovider.User{
 
- 		Username: username,
 
- 	}
 
- 	fs := vfs.NewOsFs("", os.TempDir(), nil)
 
- 	c1 := NewBaseConnection("id1", ProtocolSFTP, user, fs)
 
- 	fakeConn1 := &fakeConnection{
 
- 		BaseConnection: c1,
 
- 	}
 
- 	t1 := NewBaseTransfer(nil, c1, nil, "/p1", "/r1", TransferUpload, 0, 0, 0, true, fs)
 
- 	t1.BytesReceived = 123
 
- 	t2 := NewBaseTransfer(nil, c1, nil, "/p2", "/r2", TransferDownload, 0, 0, 0, true, fs)
 
- 	t2.BytesSent = 456
 
- 	c2 := NewBaseConnection("id2", ProtocolSSH, user, nil)
 
- 	fakeConn2 := &fakeConnection{
 
- 		BaseConnection: c2,
 
- 		command:        "md5sum",
 
- 	}
 
- 	c3 := NewBaseConnection("id3", ProtocolWebDAV, user, nil)
 
- 	fakeConn3 := &fakeConnection{
 
- 		BaseConnection: c3,
 
- 		command:        "PROPFIND",
 
- 	}
 
- 	t3 := NewBaseTransfer(nil, c3, nil, "/p2", "/r2", TransferDownload, 0, 0, 0, true, fs)
 
- 	Connections.Add(fakeConn1)
 
- 	Connections.Add(fakeConn2)
 
- 	Connections.Add(fakeConn3)
 
- 	stats := Connections.GetStats()
 
- 	assert.Len(t, stats, 3)
 
- 	for _, stat := range stats {
 
- 		assert.Equal(t, stat.Username, username)
 
- 		assert.True(t, strings.HasPrefix(stat.GetConnectionInfo(), stat.Protocol))
 
- 		assert.True(t, strings.HasPrefix(stat.GetConnectionDuration(), "00:"))
 
- 		if stat.ConnectionID == "SFTP_id1" {
 
- 			assert.Len(t, stat.Transfers, 2)
 
- 			assert.Greater(t, len(stat.GetTransfersAsString()), 0)
 
- 			for _, tr := range stat.Transfers {
 
- 				if tr.OperationType == operationDownload {
 
- 					assert.True(t, strings.HasPrefix(tr.getConnectionTransferAsString(), "DL"))
 
- 				} else if tr.OperationType == operationUpload {
 
- 					assert.True(t, strings.HasPrefix(tr.getConnectionTransferAsString(), "UL"))
 
- 				}
 
- 			}
 
- 		} else if stat.ConnectionID == "DAV_id3" {
 
- 			assert.Len(t, stat.Transfers, 1)
 
- 			assert.Greater(t, len(stat.GetTransfersAsString()), 0)
 
- 		} else {
 
- 			assert.Equal(t, 0, len(stat.GetTransfersAsString()))
 
- 		}
 
- 	}
 
- 	err := t1.Close()
 
- 	assert.NoError(t, err)
 
- 	err = t2.Close()
 
- 	assert.NoError(t, err)
 
- 	err = fakeConn3.SignalTransfersAbort()
 
- 	assert.NoError(t, err)
 
- 	assert.Equal(t, int32(1), atomic.LoadInt32(&t3.AbortTransfer))
 
- 	err = t3.Close()
 
- 	assert.NoError(t, err)
 
- 	err = fakeConn3.SignalTransfersAbort()
 
- 	assert.Error(t, err)
 
- 	Connections.Remove(fakeConn1.GetID())
 
- 	stats = Connections.GetStats()
 
- 	assert.Len(t, stats, 2)
 
- 	assert.Equal(t, fakeConn3.GetID(), stats[0].ConnectionID)
 
- 	assert.Equal(t, fakeConn2.GetID(), stats[1].ConnectionID)
 
- 	Connections.Remove(fakeConn2.GetID())
 
- 	stats = Connections.GetStats()
 
- 	assert.Len(t, stats, 1)
 
- 	assert.Equal(t, fakeConn3.GetID(), stats[0].ConnectionID)
 
- 	Connections.Remove(fakeConn3.GetID())
 
- 	stats = Connections.GetStats()
 
- 	assert.Len(t, stats, 0)
 
- }
 
- func TestQuotaScans(t *testing.T) {
 
- 	username := "username"
 
- 	assert.True(t, QuotaScans.AddUserQuotaScan(username))
 
- 	assert.False(t, QuotaScans.AddUserQuotaScan(username))
 
- 	if assert.Len(t, QuotaScans.GetUsersQuotaScans(), 1) {
 
- 		assert.Equal(t, QuotaScans.GetUsersQuotaScans()[0].Username, username)
 
- 	}
 
- 	assert.True(t, QuotaScans.RemoveUserQuotaScan(username))
 
- 	assert.False(t, QuotaScans.RemoveUserQuotaScan(username))
 
- 	assert.Len(t, QuotaScans.GetUsersQuotaScans(), 0)
 
- 	folderName := "folder"
 
- 	assert.True(t, QuotaScans.AddVFolderQuotaScan(folderName))
 
- 	assert.False(t, QuotaScans.AddVFolderQuotaScan(folderName))
 
- 	if assert.Len(t, QuotaScans.GetVFoldersQuotaScans(), 1) {
 
- 		assert.Equal(t, QuotaScans.GetVFoldersQuotaScans()[0].Name, folderName)
 
- 	}
 
- 	assert.True(t, QuotaScans.RemoveVFolderQuotaScan(folderName))
 
- 	assert.False(t, QuotaScans.RemoveVFolderQuotaScan(folderName))
 
- 	assert.Len(t, QuotaScans.GetVFoldersQuotaScans(), 0)
 
- }
 
- func TestProxyProtocolVersion(t *testing.T) {
 
- 	c := Configuration{
 
- 		ProxyProtocol: 1,
 
- 	}
 
- 	proxyListener, err := c.GetProxyListener(nil)
 
- 	assert.NoError(t, err)
 
- 	assert.Nil(t, proxyListener.Policy)
 
- 	c.ProxyProtocol = 2
 
- 	proxyListener, err = c.GetProxyListener(nil)
 
- 	assert.NoError(t, err)
 
- 	assert.NotNil(t, proxyListener.Policy)
 
- 	c.ProxyProtocol = 1
 
- 	c.ProxyAllowed = []string{"invalid"}
 
- 	_, err = c.GetProxyListener(nil)
 
- 	assert.Error(t, err)
 
- 	c.ProxyProtocol = 2
 
- 	_, err = c.GetProxyListener(nil)
 
- 	assert.Error(t, err)
 
- }
 
- func TestProxyProtocol(t *testing.T) {
 
- 	httpClient := httpclient.GetHTTPClient()
 
- 	resp, err := httpClient.Get(fmt.Sprintf("http://%v", httpProxyAddr))
 
- 	if assert.NoError(t, err) {
 
- 		defer resp.Body.Close()
 
- 		assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
 
- 	}
 
- }
 
- func TestPostConnectHook(t *testing.T) {
 
- 	Config.PostConnectHook = ""
 
- 	ipAddr := "127.0.0.1"
 
- 	assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
 
- 	Config.PostConnectHook = "http://foo\x7f.com/"
 
- 	assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
 
- 	Config.PostConnectHook = "http://invalid:1234/"
 
- 	assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
 
- 	Config.PostConnectHook = fmt.Sprintf("http://%v/404", httpAddr)
 
- 	assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
 
- 	Config.PostConnectHook = fmt.Sprintf("http://%v", httpAddr)
 
- 	assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
 
- 	Config.PostConnectHook = "invalid"
 
- 	assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
 
- 	if runtime.GOOS == osWindows {
 
- 		Config.PostConnectHook = "C:\\bad\\command"
 
- 		assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
 
- 	} else {
 
- 		Config.PostConnectHook = "/invalid/path"
 
- 		assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
 
- 		hookCmd, err := exec.LookPath("true")
 
- 		assert.NoError(t, err)
 
- 		Config.PostConnectHook = hookCmd
 
- 		assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
 
- 	}
 
- 	Config.PostConnectHook = ""
 
- }
 
- func TestCryptoConvertFileInfo(t *testing.T) {
 
- 	name := "name"
 
- 	fs, err := vfs.NewCryptFs("connID1", os.TempDir(), vfs.CryptFsConfig{Passphrase: kms.NewPlainSecret("secret")})
 
- 	require.NoError(t, err)
 
- 	cryptFs := fs.(*vfs.CryptFs)
 
- 	info := vfs.NewFileInfo(name, true, 48, time.Now(), false)
 
- 	assert.Equal(t, info, cryptFs.ConvertFileInfo(info))
 
- 	info = vfs.NewFileInfo(name, false, 48, time.Now(), false)
 
- 	assert.NotEqual(t, info.Size(), cryptFs.ConvertFileInfo(info).Size())
 
- 	info = vfs.NewFileInfo(name, false, 33, time.Now(), false)
 
- 	assert.Equal(t, int64(0), cryptFs.ConvertFileInfo(info).Size())
 
- 	info = vfs.NewFileInfo(name, false, 1, time.Now(), false)
 
- 	assert.Equal(t, int64(0), cryptFs.ConvertFileInfo(info).Size())
 
- }
 
- func TestFolderCopy(t *testing.T) {
 
- 	folder := vfs.BaseVirtualFolder{
 
- 		ID:              1,
 
- 		Name:            "name",
 
- 		MappedPath:      filepath.Clean(os.TempDir()),
 
- 		UsedQuotaSize:   4096,
 
- 		UsedQuotaFiles:  2,
 
- 		LastQuotaUpdate: utils.GetTimeAsMsSinceEpoch(time.Now()),
 
- 		Users:           []string{"user1", "user2"},
 
- 	}
 
- 	folderCopy := folder.GetACopy()
 
- 	folder.ID = 2
 
- 	folder.Users = []string{"user3"}
 
- 	require.Len(t, folderCopy.Users, 2)
 
- 	require.True(t, utils.IsStringInSlice("user1", folderCopy.Users))
 
- 	require.True(t, utils.IsStringInSlice("user2", folderCopy.Users))
 
- 	require.Equal(t, int64(1), folderCopy.ID)
 
- 	require.Equal(t, folder.Name, folderCopy.Name)
 
- 	require.Equal(t, folder.MappedPath, folderCopy.MappedPath)
 
- 	require.Equal(t, folder.UsedQuotaSize, folderCopy.UsedQuotaSize)
 
- 	require.Equal(t, folder.UsedQuotaFiles, folderCopy.UsedQuotaFiles)
 
- 	require.Equal(t, folder.LastQuotaUpdate, folderCopy.LastQuotaUpdate)
 
- }
 
 
  |