common_test.go 20 KB

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