common_test.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652
  1. package common
  2. import (
  3. "fmt"
  4. "net"
  5. "net/http"
  6. "os"
  7. "os/exec"
  8. "path/filepath"
  9. "runtime"
  10. "strings"
  11. "sync/atomic"
  12. "testing"
  13. "time"
  14. "github.com/rs/zerolog"
  15. "github.com/spf13/viper"
  16. "github.com/stretchr/testify/assert"
  17. "github.com/stretchr/testify/require"
  18. "github.com/drakkan/sftpgo/dataprovider"
  19. "github.com/drakkan/sftpgo/httpclient"
  20. "github.com/drakkan/sftpgo/kms"
  21. "github.com/drakkan/sftpgo/logger"
  22. "github.com/drakkan/sftpgo/utils"
  23. "github.com/drakkan/sftpgo/vfs"
  24. )
  25. const (
  26. logSenderTest = "common_test"
  27. httpAddr = "127.0.0.1:9999"
  28. httpProxyAddr = "127.0.0.1:7777"
  29. configDir = ".."
  30. osWindows = "windows"
  31. userTestUsername = "common_test_username"
  32. userTestPwd = "common_test_pwd"
  33. )
  34. type providerConf struct {
  35. Config dataprovider.Config `json:"data_provider" mapstructure:"data_provider"`
  36. }
  37. type fakeConnection struct {
  38. *BaseConnection
  39. command string
  40. }
  41. func (c *fakeConnection) AddUser(user dataprovider.User) error {
  42. fs, err := user.GetFilesystem(c.GetID())
  43. if err != nil {
  44. return err
  45. }
  46. c.BaseConnection.User = user
  47. c.BaseConnection.Fs = fs
  48. return nil
  49. }
  50. func (c *fakeConnection) Disconnect() error {
  51. Connections.Remove(c.GetID())
  52. return nil
  53. }
  54. func (c *fakeConnection) GetClientVersion() string {
  55. return ""
  56. }
  57. func (c *fakeConnection) GetCommand() string {
  58. return c.command
  59. }
  60. func (c *fakeConnection) GetRemoteAddress() string {
  61. return ""
  62. }
  63. type customNetConn struct {
  64. net.Conn
  65. id string
  66. isClosed bool
  67. }
  68. func (c *customNetConn) Close() error {
  69. Connections.RemoveSSHConnection(c.id)
  70. c.isClosed = true
  71. return c.Conn.Close()
  72. }
  73. func TestMain(m *testing.M) {
  74. logfilePath := "common_test.log"
  75. logger.InitLogger(logfilePath, 5, 1, 28, false, zerolog.DebugLevel)
  76. viper.SetEnvPrefix("sftpgo")
  77. replacer := strings.NewReplacer(".", "__")
  78. viper.SetEnvKeyReplacer(replacer)
  79. viper.SetConfigName("sftpgo")
  80. viper.AutomaticEnv()
  81. viper.AllowEmptyEnv(true)
  82. driver, err := initializeDataprovider(-1)
  83. if err != nil {
  84. logger.WarnToConsole("error initializing data provider: %v", err)
  85. os.Exit(1)
  86. }
  87. logger.InfoToConsole("Starting COMMON tests, provider: %v", driver)
  88. err = Initialize(Configuration{})
  89. if err != nil {
  90. logger.WarnToConsole("error initializing common: %v", err)
  91. os.Exit(1)
  92. }
  93. httpConfig := httpclient.Config{
  94. Timeout: 5,
  95. }
  96. httpConfig.Initialize(configDir)
  97. go func() {
  98. // start a test HTTP server to receive action notifications
  99. http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  100. fmt.Fprintf(w, "OK\n")
  101. })
  102. http.HandleFunc("/404", func(w http.ResponseWriter, r *http.Request) {
  103. w.WriteHeader(http.StatusNotFound)
  104. fmt.Fprintf(w, "Not found\n")
  105. })
  106. if err := http.ListenAndServe(httpAddr, nil); err != nil {
  107. logger.ErrorToConsole("could not start HTTP notification server: %v", err)
  108. os.Exit(1)
  109. }
  110. }()
  111. go func() {
  112. Config.ProxyProtocol = 2
  113. listener, err := net.Listen("tcp", httpProxyAddr)
  114. if err != nil {
  115. logger.ErrorToConsole("error creating listener for proxy protocol server: %v", err)
  116. os.Exit(1)
  117. }
  118. proxyListener, err := Config.GetProxyListener(listener)
  119. if err != nil {
  120. logger.ErrorToConsole("error creating proxy protocol listener: %v", err)
  121. os.Exit(1)
  122. }
  123. Config.ProxyProtocol = 0
  124. s := &http.Server{}
  125. if err := s.Serve(proxyListener); err != nil {
  126. logger.ErrorToConsole("could not start HTTP proxy protocol server: %v", err)
  127. os.Exit(1)
  128. }
  129. }()
  130. waitTCPListening(httpAddr)
  131. waitTCPListening(httpProxyAddr)
  132. exitCode := m.Run()
  133. os.Remove(logfilePath) //nolint:errcheck
  134. os.Exit(exitCode)
  135. }
  136. func waitTCPListening(address string) {
  137. for {
  138. conn, err := net.Dial("tcp", address)
  139. if err != nil {
  140. logger.WarnToConsole("tcp server %v not listening: %v\n", address, err)
  141. time.Sleep(100 * time.Millisecond)
  142. continue
  143. }
  144. logger.InfoToConsole("tcp server %v now listening\n", address)
  145. conn.Close()
  146. break
  147. }
  148. }
  149. func initializeDataprovider(trackQuota int) (string, error) {
  150. configDir := ".."
  151. viper.AddConfigPath(configDir)
  152. if err := viper.ReadInConfig(); err != nil {
  153. return "", err
  154. }
  155. var cfg providerConf
  156. if err := viper.Unmarshal(&cfg); err != nil {
  157. return "", err
  158. }
  159. if trackQuota >= 0 && trackQuota <= 2 {
  160. cfg.Config.TrackQuota = trackQuota
  161. }
  162. return cfg.Config.Driver, dataprovider.Initialize(cfg.Config, configDir, true)
  163. }
  164. func closeDataprovider() error {
  165. return dataprovider.Close()
  166. }
  167. func TestSSHConnections(t *testing.T) {
  168. conn1, conn2 := net.Pipe()
  169. now := time.Now()
  170. sshConn1 := NewSSHConnection("id1", conn1)
  171. sshConn2 := NewSSHConnection("id2", conn2)
  172. sshConn3 := NewSSHConnection("id3", conn2)
  173. assert.Equal(t, "id1", sshConn1.GetID())
  174. assert.Equal(t, "id2", sshConn2.GetID())
  175. assert.Equal(t, "id3", sshConn3.GetID())
  176. sshConn1.UpdateLastActivity()
  177. assert.GreaterOrEqual(t, sshConn1.GetLastActivity().UnixNano(), now.UnixNano())
  178. Connections.AddSSHConnection(sshConn1)
  179. Connections.AddSSHConnection(sshConn2)
  180. Connections.AddSSHConnection(sshConn3)
  181. Connections.RLock()
  182. assert.Len(t, Connections.sshConnections, 3)
  183. Connections.RUnlock()
  184. Connections.RemoveSSHConnection(sshConn1.id)
  185. Connections.RLock()
  186. assert.Len(t, Connections.sshConnections, 2)
  187. assert.Equal(t, sshConn3.id, Connections.sshConnections[0].id)
  188. assert.Equal(t, sshConn2.id, Connections.sshConnections[1].id)
  189. Connections.RUnlock()
  190. Connections.RemoveSSHConnection(sshConn1.id)
  191. Connections.RLock()
  192. assert.Len(t, Connections.sshConnections, 2)
  193. assert.Equal(t, sshConn3.id, Connections.sshConnections[0].id)
  194. assert.Equal(t, sshConn2.id, Connections.sshConnections[1].id)
  195. Connections.RUnlock()
  196. Connections.RemoveSSHConnection(sshConn2.id)
  197. Connections.RLock()
  198. assert.Len(t, Connections.sshConnections, 1)
  199. assert.Equal(t, sshConn3.id, Connections.sshConnections[0].id)
  200. Connections.RUnlock()
  201. Connections.RemoveSSHConnection(sshConn3.id)
  202. Connections.RLock()
  203. assert.Len(t, Connections.sshConnections, 0)
  204. Connections.RUnlock()
  205. assert.NoError(t, sshConn1.Close())
  206. assert.NoError(t, sshConn2.Close())
  207. assert.NoError(t, sshConn3.Close())
  208. }
  209. func TestDefenderIntegration(t *testing.T) {
  210. // by default defender is nil
  211. configCopy := Config
  212. ip := "127.1.1.1"
  213. assert.Nil(t, ReloadDefender())
  214. AddDefenderEvent(ip, HostEventNoLoginTried)
  215. assert.False(t, IsBanned(ip))
  216. assert.Nil(t, GetDefenderBanTime(ip))
  217. assert.False(t, Unban(ip))
  218. assert.Equal(t, 0, GetDefenderScore(ip))
  219. Config.DefenderConfig = DefenderConfig{
  220. Enabled: true,
  221. BanTime: 10,
  222. BanTimeIncrement: 50,
  223. Threshold: 0,
  224. ScoreInvalid: 2,
  225. ScoreValid: 1,
  226. ObservationTime: 15,
  227. EntriesSoftLimit: 100,
  228. EntriesHardLimit: 150,
  229. }
  230. err := Initialize(Config)
  231. assert.Error(t, err)
  232. Config.DefenderConfig.Threshold = 3
  233. err = Initialize(Config)
  234. assert.NoError(t, err)
  235. assert.Nil(t, ReloadDefender())
  236. AddDefenderEvent(ip, HostEventNoLoginTried)
  237. assert.False(t, IsBanned(ip))
  238. assert.Equal(t, 2, GetDefenderScore(ip))
  239. assert.False(t, Unban(ip))
  240. assert.Nil(t, GetDefenderBanTime(ip))
  241. AddDefenderEvent(ip, HostEventLoginFailed)
  242. assert.True(t, IsBanned(ip))
  243. assert.Equal(t, 0, GetDefenderScore(ip))
  244. assert.NotNil(t, GetDefenderBanTime(ip))
  245. assert.True(t, Unban(ip))
  246. assert.Nil(t, GetDefenderBanTime(ip))
  247. assert.False(t, Unban(ip))
  248. Config = configCopy
  249. }
  250. func TestMaxConnections(t *testing.T) {
  251. oldValue := Config.MaxTotalConnections
  252. Config.MaxTotalConnections = 1
  253. assert.True(t, Connections.IsNewConnectionAllowed())
  254. c := NewBaseConnection("id", ProtocolSFTP, dataprovider.User{}, nil)
  255. fakeConn := &fakeConnection{
  256. BaseConnection: c,
  257. }
  258. Connections.Add(fakeConn)
  259. assert.Len(t, Connections.GetStats(), 1)
  260. assert.False(t, Connections.IsNewConnectionAllowed())
  261. res := Connections.Close(fakeConn.GetID())
  262. assert.True(t, res)
  263. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
  264. Config.MaxTotalConnections = oldValue
  265. }
  266. func TestIdleConnections(t *testing.T) {
  267. configCopy := Config
  268. Config.IdleTimeout = 1
  269. err := Initialize(Config)
  270. assert.NoError(t, err)
  271. conn1, conn2 := net.Pipe()
  272. customConn1 := &customNetConn{
  273. Conn: conn1,
  274. id: "id1",
  275. }
  276. customConn2 := &customNetConn{
  277. Conn: conn2,
  278. id: "id2",
  279. }
  280. sshConn1 := NewSSHConnection(customConn1.id, customConn1)
  281. sshConn2 := NewSSHConnection(customConn2.id, customConn2)
  282. username := "test_user"
  283. user := dataprovider.User{
  284. Username: username,
  285. }
  286. c := NewBaseConnection(sshConn1.id+"_1", ProtocolSFTP, user, nil)
  287. c.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
  288. fakeConn := &fakeConnection{
  289. BaseConnection: c,
  290. }
  291. // both ssh connections are expired but they should get removed only
  292. // if there is no associated connection
  293. sshConn1.lastActivity = c.lastActivity
  294. sshConn2.lastActivity = c.lastActivity
  295. Connections.AddSSHConnection(sshConn1)
  296. Connections.Add(fakeConn)
  297. assert.Equal(t, Connections.GetActiveSessions(username), 1)
  298. c = NewBaseConnection(sshConn2.id+"_1", ProtocolSSH, user, nil)
  299. fakeConn = &fakeConnection{
  300. BaseConnection: c,
  301. }
  302. Connections.AddSSHConnection(sshConn2)
  303. Connections.Add(fakeConn)
  304. assert.Equal(t, Connections.GetActiveSessions(username), 2)
  305. cFTP := NewBaseConnection("id2", ProtocolFTP, dataprovider.User{}, nil)
  306. cFTP.lastActivity = time.Now().UnixNano()
  307. fakeConn = &fakeConnection{
  308. BaseConnection: cFTP,
  309. }
  310. Connections.Add(fakeConn)
  311. assert.Equal(t, Connections.GetActiveSessions(username), 2)
  312. assert.Len(t, Connections.GetStats(), 3)
  313. Connections.RLock()
  314. assert.Len(t, Connections.sshConnections, 2)
  315. Connections.RUnlock()
  316. startIdleTimeoutTicker(100 * time.Millisecond)
  317. assert.Eventually(t, func() bool { return Connections.GetActiveSessions(username) == 1 }, 1*time.Second, 200*time.Millisecond)
  318. assert.Eventually(t, func() bool {
  319. Connections.RLock()
  320. defer Connections.RUnlock()
  321. return len(Connections.sshConnections) == 1
  322. }, 1*time.Second, 200*time.Millisecond)
  323. stopIdleTimeoutTicker()
  324. assert.Len(t, Connections.GetStats(), 2)
  325. c.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
  326. cFTP.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
  327. sshConn2.lastActivity = c.lastActivity
  328. startIdleTimeoutTicker(100 * time.Millisecond)
  329. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 1*time.Second, 200*time.Millisecond)
  330. assert.Eventually(t, func() bool {
  331. Connections.RLock()
  332. defer Connections.RUnlock()
  333. return len(Connections.sshConnections) == 0
  334. }, 1*time.Second, 200*time.Millisecond)
  335. stopIdleTimeoutTicker()
  336. assert.True(t, customConn1.isClosed)
  337. assert.True(t, customConn2.isClosed)
  338. Config = configCopy
  339. }
  340. func TestCloseConnection(t *testing.T) {
  341. c := NewBaseConnection("id", ProtocolSFTP, dataprovider.User{}, nil)
  342. fakeConn := &fakeConnection{
  343. BaseConnection: c,
  344. }
  345. assert.True(t, Connections.IsNewConnectionAllowed())
  346. Connections.Add(fakeConn)
  347. assert.Len(t, Connections.GetStats(), 1)
  348. res := Connections.Close(fakeConn.GetID())
  349. assert.True(t, res)
  350. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
  351. res = Connections.Close(fakeConn.GetID())
  352. assert.False(t, res)
  353. Connections.Remove(fakeConn.GetID())
  354. }
  355. func TestSwapConnection(t *testing.T) {
  356. c := NewBaseConnection("id", ProtocolFTP, dataprovider.User{}, nil)
  357. fakeConn := &fakeConnection{
  358. BaseConnection: c,
  359. }
  360. Connections.Add(fakeConn)
  361. if assert.Len(t, Connections.GetStats(), 1) {
  362. assert.Equal(t, "", Connections.GetStats()[0].Username)
  363. }
  364. c = NewBaseConnection("id", ProtocolFTP, dataprovider.User{
  365. Username: userTestUsername,
  366. }, nil)
  367. fakeConn = &fakeConnection{
  368. BaseConnection: c,
  369. }
  370. err := Connections.Swap(fakeConn)
  371. assert.NoError(t, err)
  372. if assert.Len(t, Connections.GetStats(), 1) {
  373. assert.Equal(t, userTestUsername, Connections.GetStats()[0].Username)
  374. }
  375. res := Connections.Close(fakeConn.GetID())
  376. assert.True(t, res)
  377. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
  378. err = Connections.Swap(fakeConn)
  379. assert.Error(t, err)
  380. }
  381. func TestAtomicUpload(t *testing.T) {
  382. configCopy := Config
  383. Config.UploadMode = UploadModeStandard
  384. assert.False(t, Config.IsAtomicUploadEnabled())
  385. Config.UploadMode = UploadModeAtomic
  386. assert.True(t, Config.IsAtomicUploadEnabled())
  387. Config.UploadMode = UploadModeAtomicWithResume
  388. assert.True(t, Config.IsAtomicUploadEnabled())
  389. Config = configCopy
  390. }
  391. func TestConnectionStatus(t *testing.T) {
  392. username := "test_user"
  393. user := dataprovider.User{
  394. Username: username,
  395. }
  396. fs := vfs.NewOsFs("", os.TempDir(), nil)
  397. c1 := NewBaseConnection("id1", ProtocolSFTP, user, fs)
  398. fakeConn1 := &fakeConnection{
  399. BaseConnection: c1,
  400. }
  401. t1 := NewBaseTransfer(nil, c1, nil, "/p1", "/r1", TransferUpload, 0, 0, 0, true, fs)
  402. t1.BytesReceived = 123
  403. t2 := NewBaseTransfer(nil, c1, nil, "/p2", "/r2", TransferDownload, 0, 0, 0, true, fs)
  404. t2.BytesSent = 456
  405. c2 := NewBaseConnection("id2", ProtocolSSH, user, nil)
  406. fakeConn2 := &fakeConnection{
  407. BaseConnection: c2,
  408. command: "md5sum",
  409. }
  410. c3 := NewBaseConnection("id3", ProtocolWebDAV, user, nil)
  411. fakeConn3 := &fakeConnection{
  412. BaseConnection: c3,
  413. command: "PROPFIND",
  414. }
  415. t3 := NewBaseTransfer(nil, c3, nil, "/p2", "/r2", TransferDownload, 0, 0, 0, true, fs)
  416. Connections.Add(fakeConn1)
  417. Connections.Add(fakeConn2)
  418. Connections.Add(fakeConn3)
  419. stats := Connections.GetStats()
  420. assert.Len(t, stats, 3)
  421. for _, stat := range stats {
  422. assert.Equal(t, stat.Username, username)
  423. assert.True(t, strings.HasPrefix(stat.GetConnectionInfo(), stat.Protocol))
  424. assert.True(t, strings.HasPrefix(stat.GetConnectionDuration(), "00:"))
  425. if stat.ConnectionID == "SFTP_id1" {
  426. assert.Len(t, stat.Transfers, 2)
  427. assert.Greater(t, len(stat.GetTransfersAsString()), 0)
  428. for _, tr := range stat.Transfers {
  429. if tr.OperationType == operationDownload {
  430. assert.True(t, strings.HasPrefix(tr.getConnectionTransferAsString(), "DL"))
  431. } else if tr.OperationType == operationUpload {
  432. assert.True(t, strings.HasPrefix(tr.getConnectionTransferAsString(), "UL"))
  433. }
  434. }
  435. } else if stat.ConnectionID == "DAV_id3" {
  436. assert.Len(t, stat.Transfers, 1)
  437. assert.Greater(t, len(stat.GetTransfersAsString()), 0)
  438. } else {
  439. assert.Equal(t, 0, len(stat.GetTransfersAsString()))
  440. }
  441. }
  442. err := t1.Close()
  443. assert.NoError(t, err)
  444. err = t2.Close()
  445. assert.NoError(t, err)
  446. err = fakeConn3.SignalTransfersAbort()
  447. assert.NoError(t, err)
  448. assert.Equal(t, int32(1), atomic.LoadInt32(&t3.AbortTransfer))
  449. err = t3.Close()
  450. assert.NoError(t, err)
  451. err = fakeConn3.SignalTransfersAbort()
  452. assert.Error(t, err)
  453. Connections.Remove(fakeConn1.GetID())
  454. stats = Connections.GetStats()
  455. assert.Len(t, stats, 2)
  456. assert.Equal(t, fakeConn3.GetID(), stats[0].ConnectionID)
  457. assert.Equal(t, fakeConn2.GetID(), stats[1].ConnectionID)
  458. Connections.Remove(fakeConn2.GetID())
  459. stats = Connections.GetStats()
  460. assert.Len(t, stats, 1)
  461. assert.Equal(t, fakeConn3.GetID(), stats[0].ConnectionID)
  462. Connections.Remove(fakeConn3.GetID())
  463. stats = Connections.GetStats()
  464. assert.Len(t, stats, 0)
  465. }
  466. func TestQuotaScans(t *testing.T) {
  467. username := "username"
  468. assert.True(t, QuotaScans.AddUserQuotaScan(username))
  469. assert.False(t, QuotaScans.AddUserQuotaScan(username))
  470. if assert.Len(t, QuotaScans.GetUsersQuotaScans(), 1) {
  471. assert.Equal(t, QuotaScans.GetUsersQuotaScans()[0].Username, username)
  472. }
  473. assert.True(t, QuotaScans.RemoveUserQuotaScan(username))
  474. assert.False(t, QuotaScans.RemoveUserQuotaScan(username))
  475. assert.Len(t, QuotaScans.GetUsersQuotaScans(), 0)
  476. folderName := "folder"
  477. assert.True(t, QuotaScans.AddVFolderQuotaScan(folderName))
  478. assert.False(t, QuotaScans.AddVFolderQuotaScan(folderName))
  479. if assert.Len(t, QuotaScans.GetVFoldersQuotaScans(), 1) {
  480. assert.Equal(t, QuotaScans.GetVFoldersQuotaScans()[0].Name, folderName)
  481. }
  482. assert.True(t, QuotaScans.RemoveVFolderQuotaScan(folderName))
  483. assert.False(t, QuotaScans.RemoveVFolderQuotaScan(folderName))
  484. assert.Len(t, QuotaScans.GetVFoldersQuotaScans(), 0)
  485. }
  486. func TestProxyProtocolVersion(t *testing.T) {
  487. c := Configuration{
  488. ProxyProtocol: 1,
  489. }
  490. proxyListener, err := c.GetProxyListener(nil)
  491. assert.NoError(t, err)
  492. assert.Nil(t, proxyListener.Policy)
  493. c.ProxyProtocol = 2
  494. proxyListener, err = c.GetProxyListener(nil)
  495. assert.NoError(t, err)
  496. assert.NotNil(t, proxyListener.Policy)
  497. c.ProxyProtocol = 1
  498. c.ProxyAllowed = []string{"invalid"}
  499. _, err = c.GetProxyListener(nil)
  500. assert.Error(t, err)
  501. c.ProxyProtocol = 2
  502. _, err = c.GetProxyListener(nil)
  503. assert.Error(t, err)
  504. }
  505. func TestProxyProtocol(t *testing.T) {
  506. httpClient := httpclient.GetHTTPClient()
  507. resp, err := httpClient.Get(fmt.Sprintf("http://%v", httpProxyAddr))
  508. if assert.NoError(t, err) {
  509. defer resp.Body.Close()
  510. assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
  511. }
  512. }
  513. func TestPostConnectHook(t *testing.T) {
  514. Config.PostConnectHook = ""
  515. ipAddr := "127.0.0.1"
  516. assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  517. Config.PostConnectHook = "http://foo\x7f.com/"
  518. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  519. Config.PostConnectHook = "http://invalid:1234/"
  520. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  521. Config.PostConnectHook = fmt.Sprintf("http://%v/404", httpAddr)
  522. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  523. Config.PostConnectHook = fmt.Sprintf("http://%v", httpAddr)
  524. assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  525. Config.PostConnectHook = "invalid"
  526. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  527. if runtime.GOOS == osWindows {
  528. Config.PostConnectHook = "C:\\bad\\command"
  529. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  530. } else {
  531. Config.PostConnectHook = "/invalid/path"
  532. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  533. hookCmd, err := exec.LookPath("true")
  534. assert.NoError(t, err)
  535. Config.PostConnectHook = hookCmd
  536. assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  537. }
  538. Config.PostConnectHook = ""
  539. }
  540. func TestCryptoConvertFileInfo(t *testing.T) {
  541. name := "name"
  542. fs, err := vfs.NewCryptFs("connID1", os.TempDir(), vfs.CryptFsConfig{Passphrase: kms.NewPlainSecret("secret")})
  543. require.NoError(t, err)
  544. cryptFs := fs.(*vfs.CryptFs)
  545. info := vfs.NewFileInfo(name, true, 48, time.Now(), false)
  546. assert.Equal(t, info, cryptFs.ConvertFileInfo(info))
  547. info = vfs.NewFileInfo(name, false, 48, time.Now(), false)
  548. assert.NotEqual(t, info.Size(), cryptFs.ConvertFileInfo(info).Size())
  549. info = vfs.NewFileInfo(name, false, 33, time.Now(), false)
  550. assert.Equal(t, int64(0), cryptFs.ConvertFileInfo(info).Size())
  551. info = vfs.NewFileInfo(name, false, 1, time.Now(), false)
  552. assert.Equal(t, int64(0), cryptFs.ConvertFileInfo(info).Size())
  553. }
  554. func TestFolderCopy(t *testing.T) {
  555. folder := vfs.BaseVirtualFolder{
  556. ID: 1,
  557. Name: "name",
  558. MappedPath: filepath.Clean(os.TempDir()),
  559. UsedQuotaSize: 4096,
  560. UsedQuotaFiles: 2,
  561. LastQuotaUpdate: utils.GetTimeAsMsSinceEpoch(time.Now()),
  562. Users: []string{"user1", "user2"},
  563. }
  564. folderCopy := folder.GetACopy()
  565. folder.ID = 2
  566. folder.Users = []string{"user3"}
  567. require.Len(t, folderCopy.Users, 2)
  568. require.True(t, utils.IsStringInSlice("user1", folderCopy.Users))
  569. require.True(t, utils.IsStringInSlice("user2", folderCopy.Users))
  570. require.Equal(t, int64(1), folderCopy.ID)
  571. require.Equal(t, folder.Name, folderCopy.Name)
  572. require.Equal(t, folder.MappedPath, folderCopy.MappedPath)
  573. require.Equal(t, folder.UsedQuotaSize, folderCopy.UsedQuotaSize)
  574. require.Equal(t, folder.UsedQuotaFiles, folderCopy.UsedQuotaFiles)
  575. require.Equal(t, folder.LastQuotaUpdate, folderCopy.LastQuotaUpdate)
  576. }