mirror of
https://github.com/drakkan/sftpgo.git
synced 2024-11-22 07:30:25 +00:00
78bf808322
This way we no longer depend on the local file system path and so we can add support for cloud backends in future updates
652 lines
19 KiB
Go
652 lines
19 KiB
Go
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)
|
|
}
|