common_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667
  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. assert.True(t, Connections.IsNewConnectionAllowed())
  213. Connections.AddNetworkConnection()
  214. Connections.AddNetworkConnection()
  215. assert.False(t, Connections.IsNewConnectionAllowed())
  216. Connections.RemoveNetworkConnection()
  217. assert.True(t, Connections.IsNewConnectionAllowed())
  218. Connections.RemoveNetworkConnection()
  219. Config.MaxTotalConnections = oldValue
  220. }
  221. func TestIdleConnections(t *testing.T) {
  222. configCopy := Config
  223. Config.IdleTimeout = 1
  224. err := Initialize(Config)
  225. assert.NoError(t, err)
  226. conn1, conn2 := net.Pipe()
  227. customConn1 := &customNetConn{
  228. Conn: conn1,
  229. id: "id1",
  230. }
  231. customConn2 := &customNetConn{
  232. Conn: conn2,
  233. id: "id2",
  234. }
  235. sshConn1 := NewSSHConnection(customConn1.id, customConn1)
  236. sshConn2 := NewSSHConnection(customConn2.id, customConn2)
  237. username := "test_user"
  238. user := dataprovider.User{
  239. Username: username,
  240. }
  241. c := NewBaseConnection(sshConn1.id+"_1", ProtocolSFTP, user)
  242. c.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
  243. fakeConn := &fakeConnection{
  244. BaseConnection: c,
  245. }
  246. // both ssh connections are expired but they should get removed only
  247. // if there is no associated connection
  248. sshConn1.lastActivity = c.lastActivity
  249. sshConn2.lastActivity = c.lastActivity
  250. Connections.AddSSHConnection(sshConn1)
  251. Connections.Add(fakeConn)
  252. assert.Equal(t, Connections.GetActiveSessions(username), 1)
  253. c = NewBaseConnection(sshConn2.id+"_1", ProtocolSSH, user)
  254. fakeConn = &fakeConnection{
  255. BaseConnection: c,
  256. }
  257. Connections.AddSSHConnection(sshConn2)
  258. Connections.Add(fakeConn)
  259. assert.Equal(t, Connections.GetActiveSessions(username), 2)
  260. cFTP := NewBaseConnection("id2", ProtocolFTP, dataprovider.User{})
  261. cFTP.lastActivity = time.Now().UnixNano()
  262. fakeConn = &fakeConnection{
  263. BaseConnection: cFTP,
  264. }
  265. Connections.Add(fakeConn)
  266. assert.Equal(t, Connections.GetActiveSessions(username), 2)
  267. assert.Len(t, Connections.GetStats(), 3)
  268. Connections.RLock()
  269. assert.Len(t, Connections.sshConnections, 2)
  270. Connections.RUnlock()
  271. startIdleTimeoutTicker(100 * time.Millisecond)
  272. assert.Eventually(t, func() bool { return Connections.GetActiveSessions(username) == 1 }, 1*time.Second, 200*time.Millisecond)
  273. assert.Eventually(t, func() bool {
  274. Connections.RLock()
  275. defer Connections.RUnlock()
  276. return len(Connections.sshConnections) == 1
  277. }, 1*time.Second, 200*time.Millisecond)
  278. stopIdleTimeoutTicker()
  279. assert.Len(t, Connections.GetStats(), 2)
  280. c.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
  281. cFTP.lastActivity = time.Now().Add(-24 * time.Hour).UnixNano()
  282. sshConn2.lastActivity = c.lastActivity
  283. startIdleTimeoutTicker(100 * time.Millisecond)
  284. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 1*time.Second, 200*time.Millisecond)
  285. assert.Eventually(t, func() bool {
  286. Connections.RLock()
  287. defer Connections.RUnlock()
  288. return len(Connections.sshConnections) == 0
  289. }, 1*time.Second, 200*time.Millisecond)
  290. stopIdleTimeoutTicker()
  291. assert.True(t, customConn1.isClosed)
  292. assert.True(t, customConn2.isClosed)
  293. Config = configCopy
  294. }
  295. func TestCloseConnection(t *testing.T) {
  296. c := NewBaseConnection("id", ProtocolSFTP, dataprovider.User{})
  297. fakeConn := &fakeConnection{
  298. BaseConnection: c,
  299. }
  300. assert.True(t, Connections.IsNewConnectionAllowed())
  301. Connections.Add(fakeConn)
  302. assert.Len(t, Connections.GetStats(), 1)
  303. res := Connections.Close(fakeConn.GetID())
  304. assert.True(t, res)
  305. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
  306. res = Connections.Close(fakeConn.GetID())
  307. assert.False(t, res)
  308. Connections.Remove(fakeConn.GetID())
  309. }
  310. func TestSwapConnection(t *testing.T) {
  311. c := NewBaseConnection("id", ProtocolFTP, dataprovider.User{})
  312. fakeConn := &fakeConnection{
  313. BaseConnection: c,
  314. }
  315. Connections.Add(fakeConn)
  316. if assert.Len(t, Connections.GetStats(), 1) {
  317. assert.Equal(t, "", Connections.GetStats()[0].Username)
  318. }
  319. c = NewBaseConnection("id", ProtocolFTP, dataprovider.User{
  320. Username: userTestUsername,
  321. })
  322. fakeConn = &fakeConnection{
  323. BaseConnection: c,
  324. }
  325. err := Connections.Swap(fakeConn)
  326. assert.NoError(t, err)
  327. if assert.Len(t, Connections.GetStats(), 1) {
  328. assert.Equal(t, userTestUsername, Connections.GetStats()[0].Username)
  329. }
  330. res := Connections.Close(fakeConn.GetID())
  331. assert.True(t, res)
  332. assert.Eventually(t, func() bool { return len(Connections.GetStats()) == 0 }, 300*time.Millisecond, 50*time.Millisecond)
  333. err = Connections.Swap(fakeConn)
  334. assert.Error(t, err)
  335. }
  336. func TestAtomicUpload(t *testing.T) {
  337. configCopy := Config
  338. Config.UploadMode = UploadModeStandard
  339. assert.False(t, Config.IsAtomicUploadEnabled())
  340. Config.UploadMode = UploadModeAtomic
  341. assert.True(t, Config.IsAtomicUploadEnabled())
  342. Config.UploadMode = UploadModeAtomicWithResume
  343. assert.True(t, Config.IsAtomicUploadEnabled())
  344. Config = configCopy
  345. }
  346. func TestConnectionStatus(t *testing.T) {
  347. username := "test_user"
  348. user := dataprovider.User{
  349. Username: username,
  350. }
  351. fs := vfs.NewOsFs("", os.TempDir(), "")
  352. c1 := NewBaseConnection("id1", ProtocolSFTP, user)
  353. fakeConn1 := &fakeConnection{
  354. BaseConnection: c1,
  355. }
  356. t1 := NewBaseTransfer(nil, c1, nil, "/p1", "/r1", TransferUpload, 0, 0, 0, true, fs)
  357. t1.BytesReceived = 123
  358. t2 := NewBaseTransfer(nil, c1, nil, "/p2", "/r2", TransferDownload, 0, 0, 0, true, fs)
  359. t2.BytesSent = 456
  360. c2 := NewBaseConnection("id2", ProtocolSSH, user)
  361. fakeConn2 := &fakeConnection{
  362. BaseConnection: c2,
  363. command: "md5sum",
  364. }
  365. c3 := NewBaseConnection("id3", ProtocolWebDAV, user)
  366. fakeConn3 := &fakeConnection{
  367. BaseConnection: c3,
  368. command: "PROPFIND",
  369. }
  370. t3 := NewBaseTransfer(nil, c3, nil, "/p2", "/r2", TransferDownload, 0, 0, 0, true, fs)
  371. Connections.Add(fakeConn1)
  372. Connections.Add(fakeConn2)
  373. Connections.Add(fakeConn3)
  374. stats := Connections.GetStats()
  375. assert.Len(t, stats, 3)
  376. for _, stat := range stats {
  377. assert.Equal(t, stat.Username, username)
  378. assert.True(t, strings.HasPrefix(stat.GetConnectionInfo(), stat.Protocol))
  379. assert.True(t, strings.HasPrefix(stat.GetConnectionDuration(), "00:"))
  380. if stat.ConnectionID == "SFTP_id1" {
  381. assert.Len(t, stat.Transfers, 2)
  382. assert.Greater(t, len(stat.GetTransfersAsString()), 0)
  383. for _, tr := range stat.Transfers {
  384. if tr.OperationType == operationDownload {
  385. assert.True(t, strings.HasPrefix(tr.getConnectionTransferAsString(), "DL"))
  386. } else if tr.OperationType == operationUpload {
  387. assert.True(t, strings.HasPrefix(tr.getConnectionTransferAsString(), "UL"))
  388. }
  389. }
  390. } else if stat.ConnectionID == "DAV_id3" {
  391. assert.Len(t, stat.Transfers, 1)
  392. assert.Greater(t, len(stat.GetTransfersAsString()), 0)
  393. } else {
  394. assert.Equal(t, 0, len(stat.GetTransfersAsString()))
  395. }
  396. }
  397. err := t1.Close()
  398. assert.NoError(t, err)
  399. err = t2.Close()
  400. assert.NoError(t, err)
  401. err = fakeConn3.SignalTransfersAbort()
  402. assert.NoError(t, err)
  403. assert.Equal(t, int32(1), atomic.LoadInt32(&t3.AbortTransfer))
  404. err = t3.Close()
  405. assert.NoError(t, err)
  406. err = fakeConn3.SignalTransfersAbort()
  407. assert.Error(t, err)
  408. Connections.Remove(fakeConn1.GetID())
  409. stats = Connections.GetStats()
  410. assert.Len(t, stats, 2)
  411. assert.Equal(t, fakeConn3.GetID(), stats[0].ConnectionID)
  412. assert.Equal(t, fakeConn2.GetID(), stats[1].ConnectionID)
  413. Connections.Remove(fakeConn2.GetID())
  414. stats = Connections.GetStats()
  415. assert.Len(t, stats, 1)
  416. assert.Equal(t, fakeConn3.GetID(), stats[0].ConnectionID)
  417. Connections.Remove(fakeConn3.GetID())
  418. stats = Connections.GetStats()
  419. assert.Len(t, stats, 0)
  420. }
  421. func TestQuotaScans(t *testing.T) {
  422. username := "username"
  423. assert.True(t, QuotaScans.AddUserQuotaScan(username))
  424. assert.False(t, QuotaScans.AddUserQuotaScan(username))
  425. if assert.Len(t, QuotaScans.GetUsersQuotaScans(), 1) {
  426. assert.Equal(t, QuotaScans.GetUsersQuotaScans()[0].Username, username)
  427. }
  428. assert.True(t, QuotaScans.RemoveUserQuotaScan(username))
  429. assert.False(t, QuotaScans.RemoveUserQuotaScan(username))
  430. assert.Len(t, QuotaScans.GetUsersQuotaScans(), 0)
  431. folderName := "folder"
  432. assert.True(t, QuotaScans.AddVFolderQuotaScan(folderName))
  433. assert.False(t, QuotaScans.AddVFolderQuotaScan(folderName))
  434. if assert.Len(t, QuotaScans.GetVFoldersQuotaScans(), 1) {
  435. assert.Equal(t, QuotaScans.GetVFoldersQuotaScans()[0].Name, folderName)
  436. }
  437. assert.True(t, QuotaScans.RemoveVFolderQuotaScan(folderName))
  438. assert.False(t, QuotaScans.RemoveVFolderQuotaScan(folderName))
  439. assert.Len(t, QuotaScans.GetVFoldersQuotaScans(), 0)
  440. }
  441. func TestProxyProtocolVersion(t *testing.T) {
  442. c := Configuration{
  443. ProxyProtocol: 1,
  444. }
  445. proxyListener, err := c.GetProxyListener(nil)
  446. assert.NoError(t, err)
  447. assert.Nil(t, proxyListener.Policy)
  448. c.ProxyProtocol = 2
  449. proxyListener, err = c.GetProxyListener(nil)
  450. assert.NoError(t, err)
  451. assert.NotNil(t, proxyListener.Policy)
  452. c.ProxyProtocol = 1
  453. c.ProxyAllowed = []string{"invalid"}
  454. _, err = c.GetProxyListener(nil)
  455. assert.Error(t, err)
  456. c.ProxyProtocol = 2
  457. _, err = c.GetProxyListener(nil)
  458. assert.Error(t, err)
  459. }
  460. func TestStartupHook(t *testing.T) {
  461. Config.StartupHook = ""
  462. assert.NoError(t, Config.ExecuteStartupHook())
  463. Config.StartupHook = "http://foo\x7f.com/startup"
  464. assert.Error(t, Config.ExecuteStartupHook())
  465. Config.StartupHook = "http://invalid:5678/"
  466. assert.Error(t, Config.ExecuteStartupHook())
  467. Config.StartupHook = fmt.Sprintf("http://%v", httpAddr)
  468. assert.NoError(t, Config.ExecuteStartupHook())
  469. Config.StartupHook = "invalidhook"
  470. assert.Error(t, Config.ExecuteStartupHook())
  471. if runtime.GOOS != osWindows {
  472. hookCmd, err := exec.LookPath("true")
  473. assert.NoError(t, err)
  474. Config.StartupHook = hookCmd
  475. assert.NoError(t, Config.ExecuteStartupHook())
  476. }
  477. Config.StartupHook = ""
  478. }
  479. func TestPostConnectHook(t *testing.T) {
  480. Config.PostConnectHook = ""
  481. ipAddr := "127.0.0.1"
  482. assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  483. Config.PostConnectHook = "http://foo\x7f.com/"
  484. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  485. Config.PostConnectHook = "http://invalid:1234/"
  486. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  487. Config.PostConnectHook = fmt.Sprintf("http://%v/404", httpAddr)
  488. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  489. Config.PostConnectHook = fmt.Sprintf("http://%v", httpAddr)
  490. assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  491. Config.PostConnectHook = "invalid"
  492. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolFTP))
  493. if runtime.GOOS == osWindows {
  494. Config.PostConnectHook = "C:\\bad\\command"
  495. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  496. } else {
  497. Config.PostConnectHook = "/invalid/path"
  498. assert.Error(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  499. hookCmd, err := exec.LookPath("true")
  500. assert.NoError(t, err)
  501. Config.PostConnectHook = hookCmd
  502. assert.NoError(t, Config.ExecutePostConnectHook(ipAddr, ProtocolSFTP))
  503. }
  504. Config.PostConnectHook = ""
  505. }
  506. func TestCryptoConvertFileInfo(t *testing.T) {
  507. name := "name"
  508. fs, err := vfs.NewCryptFs("connID1", os.TempDir(), "", vfs.CryptFsConfig{Passphrase: kms.NewPlainSecret("secret")})
  509. require.NoError(t, err)
  510. cryptFs := fs.(*vfs.CryptFs)
  511. info := vfs.NewFileInfo(name, true, 48, time.Now(), false)
  512. assert.Equal(t, info, cryptFs.ConvertFileInfo(info))
  513. info = vfs.NewFileInfo(name, false, 48, time.Now(), false)
  514. assert.NotEqual(t, info.Size(), cryptFs.ConvertFileInfo(info).Size())
  515. info = vfs.NewFileInfo(name, false, 33, time.Now(), false)
  516. assert.Equal(t, int64(0), cryptFs.ConvertFileInfo(info).Size())
  517. info = vfs.NewFileInfo(name, false, 1, time.Now(), false)
  518. assert.Equal(t, int64(0), cryptFs.ConvertFileInfo(info).Size())
  519. }
  520. func TestFolderCopy(t *testing.T) {
  521. folder := vfs.BaseVirtualFolder{
  522. ID: 1,
  523. Name: "name",
  524. MappedPath: filepath.Clean(os.TempDir()),
  525. UsedQuotaSize: 4096,
  526. UsedQuotaFiles: 2,
  527. LastQuotaUpdate: utils.GetTimeAsMsSinceEpoch(time.Now()),
  528. Users: []string{"user1", "user2"},
  529. }
  530. folderCopy := folder.GetACopy()
  531. folder.ID = 2
  532. folder.Users = []string{"user3"}
  533. require.Len(t, folderCopy.Users, 2)
  534. require.True(t, utils.IsStringInSlice("user1", folderCopy.Users))
  535. require.True(t, utils.IsStringInSlice("user2", folderCopy.Users))
  536. require.Equal(t, int64(1), folderCopy.ID)
  537. require.Equal(t, folder.Name, folderCopy.Name)
  538. require.Equal(t, folder.MappedPath, folderCopy.MappedPath)
  539. require.Equal(t, folder.UsedQuotaSize, folderCopy.UsedQuotaSize)
  540. require.Equal(t, folder.UsedQuotaFiles, folderCopy.UsedQuotaFiles)
  541. require.Equal(t, folder.LastQuotaUpdate, folderCopy.LastQuotaUpdate)
  542. folder.FsConfig = vfs.Filesystem{
  543. CryptConfig: vfs.CryptFsConfig{
  544. Passphrase: kms.NewPlainSecret("crypto secret"),
  545. },
  546. }
  547. folderCopy = folder.GetACopy()
  548. folder.FsConfig.CryptConfig.Passphrase = kms.NewEmptySecret()
  549. require.Len(t, folderCopy.Users, 1)
  550. require.True(t, utils.IsStringInSlice("user3", folderCopy.Users))
  551. require.Equal(t, int64(2), folderCopy.ID)
  552. require.Equal(t, folder.Name, folderCopy.Name)
  553. require.Equal(t, folder.MappedPath, folderCopy.MappedPath)
  554. require.Equal(t, folder.UsedQuotaSize, folderCopy.UsedQuotaSize)
  555. require.Equal(t, folder.UsedQuotaFiles, folderCopy.UsedQuotaFiles)
  556. require.Equal(t, folder.LastQuotaUpdate, folderCopy.LastQuotaUpdate)
  557. require.Equal(t, "crypto secret", folderCopy.FsConfig.CryptConfig.Passphrase.GetPayload())
  558. }
  559. func TestCachedFs(t *testing.T) {
  560. user := dataprovider.User{
  561. HomeDir: filepath.Clean(os.TempDir()),
  562. }
  563. conn := NewBaseConnection("id", ProtocolSFTP, user)
  564. // changing the user should not affect the connection
  565. user.HomeDir = filepath.Join(os.TempDir(), "temp")
  566. err := os.Mkdir(user.HomeDir, os.ModePerm)
  567. assert.NoError(t, err)
  568. fs, err := user.GetFilesystem("")
  569. assert.NoError(t, err)
  570. p, err := fs.ResolvePath("/")
  571. assert.NoError(t, err)
  572. assert.Equal(t, user.GetHomeDir(), p)
  573. _, p, err = conn.GetFsAndResolvedPath("/")
  574. assert.NoError(t, err)
  575. assert.Equal(t, filepath.Clean(os.TempDir()), p)
  576. user.FsConfig.Provider = vfs.S3FilesystemProvider
  577. _, err = user.GetFilesystem("")
  578. assert.Error(t, err)
  579. conn.User.FsConfig.Provider = vfs.S3FilesystemProvider
  580. _, p, err = conn.GetFsAndResolvedPath("/")
  581. assert.NoError(t, err)
  582. assert.Equal(t, filepath.Clean(os.TempDir()), p)
  583. err = os.Remove(user.HomeDir)
  584. assert.NoError(t, err)
  585. }