ftpd_test.go 142 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270
  1. // Copyright (C) 2019 Nicola Murino
  2. //
  3. // This program is free software: you can redistribute it and/or modify
  4. // it under the terms of the GNU Affero General Public License as published
  5. // by the Free Software Foundation, version 3.
  6. //
  7. // This program is distributed in the hope that it will be useful,
  8. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. // GNU Affero General Public License for more details.
  11. //
  12. // You should have received a copy of the GNU Affero General Public License
  13. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  14. package ftpd_test
  15. import (
  16. "crypto/rand"
  17. "crypto/sha256"
  18. "crypto/tls"
  19. "encoding/hex"
  20. "encoding/json"
  21. "errors"
  22. "fmt"
  23. "io"
  24. "io/fs"
  25. "net"
  26. "net/http"
  27. "os"
  28. "os/exec"
  29. "path"
  30. "path/filepath"
  31. "runtime"
  32. "strconv"
  33. "testing"
  34. "time"
  35. ftpserver "github.com/fclairamb/ftpserverlib"
  36. "github.com/jlaffaye/ftp"
  37. "github.com/pquerna/otp"
  38. "github.com/pquerna/otp/totp"
  39. "github.com/rs/zerolog"
  40. "github.com/sftpgo/sdk"
  41. sdkkms "github.com/sftpgo/sdk/kms"
  42. "github.com/stretchr/testify/assert"
  43. "github.com/stretchr/testify/require"
  44. "github.com/drakkan/sftpgo/v2/internal/common"
  45. "github.com/drakkan/sftpgo/v2/internal/config"
  46. "github.com/drakkan/sftpgo/v2/internal/dataprovider"
  47. "github.com/drakkan/sftpgo/v2/internal/ftpd"
  48. "github.com/drakkan/sftpgo/v2/internal/httpdtest"
  49. "github.com/drakkan/sftpgo/v2/internal/kms"
  50. "github.com/drakkan/sftpgo/v2/internal/logger"
  51. "github.com/drakkan/sftpgo/v2/internal/mfa"
  52. "github.com/drakkan/sftpgo/v2/internal/sftpd"
  53. "github.com/drakkan/sftpgo/v2/internal/vfs"
  54. )
  55. const (
  56. logSender = "ftpdTesting"
  57. ftpServerAddr = "127.0.0.1:2121"
  58. sftpServerAddr = "127.0.0.1:2122"
  59. ftpSrvAddrTLS = "127.0.0.1:2124" // ftp server with implicit tls
  60. ftpSrvAddrTLSResumption = "127.0.0.1:2126" // ftp server with implicit tls
  61. defaultUsername = "test_user_ftp"
  62. defaultPassword = "test_password"
  63. osWindows = "windows"
  64. ftpsCert = `-----BEGIN CERTIFICATE-----
  65. MIICHTCCAaKgAwIBAgIUHnqw7QnB1Bj9oUsNpdb+ZkFPOxMwCgYIKoZIzj0EAwIw
  66. RTELMAkGA1UEBhMCQVUxEzARBgNVBAgMClNvbWUtU3RhdGUxITAfBgNVBAoMGElu
  67. dGVybmV0IFdpZGdpdHMgUHR5IEx0ZDAeFw0yMDAyMDQwOTUzMDRaFw0zMDAyMDEw
  68. OTUzMDRaMEUxCzAJBgNVBAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYD
  69. VQQKDBhJbnRlcm5ldCBXaWRnaXRzIFB0eSBMdGQwdjAQBgcqhkjOPQIBBgUrgQQA
  70. IgNiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVqWvrJ51t5OxV0v25NsOgR82CA
  71. NXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIVCzgWkxiz7XE4lgUwX44FCXZM
  72. 3+JeUbKjUzBRMB0GA1UdDgQWBBRhLw+/o3+Z02MI/d4tmaMui9W16jAfBgNVHSME
  73. GDAWgBRhLw+/o3+Z02MI/d4tmaMui9W16jAPBgNVHRMBAf8EBTADAQH/MAoGCCqG
  74. SM49BAMCA2kAMGYCMQDqLt2lm8mE+tGgtjDmtFgdOcI72HSbRQ74D5rYTzgST1rY
  75. /8wTi5xl8TiFUyLMUsICMQC5ViVxdXbhuG7gX6yEqSkMKZICHpO8hqFwOD/uaFVI
  76. dV4vKmHUzwK/eIx+8Ay3neE=
  77. -----END CERTIFICATE-----`
  78. ftpsKey = `-----BEGIN EC PARAMETERS-----
  79. BgUrgQQAIg==
  80. -----END EC PARAMETERS-----
  81. -----BEGIN EC PRIVATE KEY-----
  82. MIGkAgEBBDCfMNsN6miEE3rVyUPwElfiJSWaR5huPCzUenZOfJT04GAcQdWvEju3
  83. UM2lmBLIXpGgBwYFK4EEACKhZANiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVq
  84. WvrJ51t5OxV0v25NsOgR82CANXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIV
  85. CzgWkxiz7XE4lgUwX44FCXZM3+JeUbI=
  86. -----END EC PRIVATE KEY-----`
  87. caCRT = `-----BEGIN CERTIFICATE-----
  88. MIIE5jCCAs6gAwIBAgIBATANBgkqhkiG9w0BAQsFADATMREwDwYDVQQDEwhDZXJ0
  89. QXV0aDAeFw0yNDAxMTAxODEyMDRaFw0zNDAxMTAxODIxNTRaMBMxETAPBgNVBAMT
  90. CENlcnRBdXRoMIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEA7WHW216m
  91. fi4uF8cx6HWf8wvAxaEWgCHTOi2MwFIzOrOtuT7xb64rkpdzx1aWetSiCrEyc3D1
  92. v03k0Akvlz1gtnDtO64+MA8bqlTnCydZJY4cCTvDOBUYZgtMqHZzpE6xRrqQ84zh
  93. yzjKQ5bR0st+XGfIkuhjSuf2n/ZPS37fge9j6AKzn/2uEVt33qmO85WtN3RzbSqL
  94. CdOJ6cQ216j3la1C5+NWvzIKC7t6NE1bBGI4+tRj7B5P5MeamkkogwbExUjdHp3U
  95. 4yasvoGcCHUQDoa4Dej1faywz6JlwB6rTV4ys4aZDe67V/Q8iB2May1k7zBz1Ztb
  96. KF5Em3xewP1LqPEowF1uc4KtPGcP4bxdaIpSpmObcn8AIfH6smLQrn0C3cs7CYfo
  97. NlFuTbwzENUhjz0X6EsoM4w4c87lO+dRNR7YpHLqR/BJTbbyXUB0imne1u00fuzb
  98. S7OtweiA9w7DRCkr2gU4lmHe7l0T+SA9pxIeVLb78x7ivdyXSF5LVQJ1JvhhWu6i
  99. M6GQdLHat/0fpRFUbEe34RQSDJ2eOBifMJqvsvpBP8d2jcRZVUVrSXGc2mAGuGOY
  100. /tmnCJGW8Fd+sgpCVAqM0pxCM+apqrvJYUqqQZ2ZxugCXULtRWJ9p4C9zUl40HEy
  101. OQ+AaiiwFll/doXELglcJdNg8AZPGhugfxMCAwEAAaNFMEMwDgYDVR0PAQH/BAQD
  102. AgEGMBIGA1UdEwEB/wQIMAYBAf8CAQAwHQYDVR0OBBYEFNoJhIvDZQrEf/VQbWuu
  103. XgNnt2m5MA0GCSqGSIb3DQEBCwUAA4ICAQCYhT5SRqk19hGrQ09hVSZOzynXAa5F
  104. sYkEWJzFyLg9azhnTPE1bFM18FScnkd+dal6mt+bQiJvdh24NaVkDghVB7GkmXki
  105. pAiZwEDHMqtbhiPxY8LtSeCBAz5JqXVU2Q0TpAgNSH4W7FbGWNThhxcJVOoIrXKE
  106. jbzhwl1Etcaf0DBKWliUbdlxQQs65DLy+rNBYtOeK0pzhzn1vpehUlJ4eTFzP9KX
  107. y2Mksuq9AspPbqnqpWW645MdTxMb5T57MCrY3GDKw63z5z3kz88LWJF3nOxZmgQy
  108. WFUhbLmZm7x6N5eiu6Wk8/B4yJ/n5UArD4cEP1i7nqu+mbbM/SZlq1wnGpg/sbRV
  109. oUF+a7pRcSbfxEttle4pLFhS+ErKatjGcNEab2OlU3bX5UoBs+TYodnCWGKOuBKV
  110. L/CYc65QyeYZ+JiwYn9wC8YkzOnnVIQjiCEkLgSL30h9dxpnTZDLrdAA8ItelDn5
  111. DvjuQq58CGDsaVqpSobiSC1DMXYWot4Ets1wwovUNEq1l0MERB+2olE+JU/8E23E
  112. eL1/aA7Kw/JibkWz1IyzClpFDKXf6kR2onJyxerdwUL+is7tqYFLysiHxZDL1bli
  113. SXbW8hMa5gvo0IilFP9Rznn8PplIfCsvBDVv6xsRr5nTAFtwKaMBVgznE2ghs69w
  114. kK8u1YiiVenmoQ==
  115. -----END CERTIFICATE-----`
  116. caCRL = `-----BEGIN X509 CRL-----
  117. MIICpzCBkAIBATANBgkqhkiG9w0BAQsFADATMREwDwYDVQQDEwhDZXJ0QXV0aBcN
  118. MjQwMTEwMTgyMjU4WhcNMjYwMTA5MTgyMjU4WjAkMCICEQDOaeHbjY4pEj8WBmqg
  119. ZuRRFw0yNDAxMTAxODIyNThaoCMwITAfBgNVHSMEGDAWgBTaCYSLw2UKxH/1UG1r
  120. rl4DZ7dpuTANBgkqhkiG9w0BAQsFAAOCAgEAZzZ4aBqCcAJigR9e/mqKpJa4B6FV
  121. +jZmnWXolGeUuVkjdiG9w614x7mB2S768iioJyALejjCZjqsp6ydxtn0epQw4199
  122. XSfPIxA9lxc7w79GLe0v3ztojvxDPh5V1+lwPzGf9i8AsGqb2BrcBqgxDeatndnE
  123. jF+18bY1saXOBpukNLjtRScUXzy5YcSuO6mwz4548v+1ebpF7W4Yh+yh0zldJKcF
  124. DouuirZWujJwTwxxfJ+2+yP7GAuefXUOhYs/1y9ylvUgvKFqSyokv6OaVgTooKYD
  125. MSADzmNcbRvwyAC5oL2yJTVVoTFeP6fXl/BdFH3sO/hlKXGy4Wh1AjcVE6T0CSJ4
  126. iYFX3gLFh6dbP9IQWMlIM5DKtAKSjmgOywEaWii3e4M0NFSf/Cy17p2E5/jXSLlE
  127. ypDileK0aALkx2twGWwogh6sY1dQ6R3GpKSRPD2muQxVOG6wXvuJce0E9WLx1Ud4
  128. hVUdUEMlKUvm77/15U5awarH2cCJQxzS/GMeIintQiG7hUlgRzRdmWVe3vOOvt94
  129. cp8+ZUH/QSDOo41ATTHpFeC/XqF5E2G/ahXqra+O5my52V/FP0bSJnkorJ8apy67
  130. sn6DFbkqX9khTXGtacczh2PcqVjcQjBniYl2sPO3qIrrrY3tic96tMnM/u3JRdcn
  131. w7bXJGfJcIMrrKs=
  132. -----END X509 CRL-----`
  133. client1Crt = `-----BEGIN CERTIFICATE-----
  134. MIIEITCCAgmgAwIBAgIRAJr32nHRlhyPiS7IfZ/ZWYowDQYJKoZIhvcNAQELBQAw
  135. EzERMA8GA1UEAxMIQ2VydEF1dGgwHhcNMjQwMTEwMTgxMjM3WhcNMzQwMTEwMTgy
  136. MTUzWjASMRAwDgYDVQQDEwdjbGllbnQxMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
  137. MIIBCgKCAQEAtuQFiqvdjd8WLxP0FgPDyDEJ1/uJ+Aoj6QllNV7svWxwW+kiJ3X6
  138. HUVNWhhCsNfly4pGW4erF4fZzmesElGx1PoWgQCWZKsa/N08bznelWgdmkyi85xE
  139. OkTj6e/cTWHFSOBURNJaXkGHZ0ROSh7qu0Ld+eqNo3k9W+NqZaqYvs2K7MLWeYl7
  140. Qie8Ctuq5Qaz/jm0XwR2PFBROVQSaCPCukancPQ21ftqHPhAbjxoxvvN5QP4ZdRf
  141. XlH/LDLhlFnJzPZdHnVy9xisSPPRfFApJiwyfjRYdtslpJOcNgP6oPlpX/dybbhO
  142. c9FEUgj/Q90Je8EfioBYFYsqVD6/dFv9SwIDAQABo3EwbzAOBgNVHQ8BAf8EBAMC
  143. A7gwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQUFBwMCMB0GA1UdDgQWBBRUh5Xo
  144. Gzjh6iReaPSOgGatqOw9bDAfBgNVHSMEGDAWgBTaCYSLw2UKxH/1UG1rrl4DZ7dp
  145. uTANBgkqhkiG9w0BAQsFAAOCAgEAyAK7cOTWqjyLgFM0kyyx1fNPvm2GwKep3MuU
  146. OrSnLuWjoxzb7WcbKNVMlnvnmSUAWuErxsY0PUJNfcuqWiGmEp4d/SWfWPigG6DC
  147. sDej35BlSfX8FCufYrfC74VNk4yBS2LVYmIqcpqUrfay0I2oZA8+ToLEpdUvEv2I
  148. l59eOhJO2jsC3JbOyZZmK2Kv7d94fR+1tg2Rq1Wbnmc9AZKq7KDReAlIJh4u2KHb
  149. BbtF79idusMwZyP777tqSQ4THBMa+VAEc2UrzdZqTIAwqlKQOvO2fRz2P+ARR+Tz
  150. MYJMdCdmPZ9qAc8U1OcFBG6qDDltO8wf/Nu/PsSI5LGCIhIuPPIuKfm0rRfTqCG7
  151. QPQPWjRoXtGGhwjdIuWbX9fIB+c+NpAEKHgLtV+Rxj8s5IVxqG9a5TtU9VkfVXJz
  152. J20naoz/G+vDsVINpd3kH0ziNvdrKfGRM5UgtnUOPCXB22fVmkIsMH2knI10CKK+
  153. offI56NTkLRu00xvg98/wdukhkwIAxg6PQI/BHY5mdvoacEHHHdOhMq+GSAh7DDX
  154. G8+HdbABM1ExkPnZLat15q706ztiuUpQv1C2DI8YviUVkMqCslj4cD4F8EFPo4kr
  155. kvme0Cuc9Qlf7N5rjdV3cjwavhFx44dyXj9aesft2Q1okPiIqbGNpcjHcIRlj4Au
  156. MU3Bo0A=
  157. -----END CERTIFICATE-----`
  158. client1Key = `-----BEGIN RSA PRIVATE KEY-----
  159. MIIEpAIBAAKCAQEAtuQFiqvdjd8WLxP0FgPDyDEJ1/uJ+Aoj6QllNV7svWxwW+ki
  160. J3X6HUVNWhhCsNfly4pGW4erF4fZzmesElGx1PoWgQCWZKsa/N08bznelWgdmkyi
  161. 85xEOkTj6e/cTWHFSOBURNJaXkGHZ0ROSh7qu0Ld+eqNo3k9W+NqZaqYvs2K7MLW
  162. eYl7Qie8Ctuq5Qaz/jm0XwR2PFBROVQSaCPCukancPQ21ftqHPhAbjxoxvvN5QP4
  163. ZdRfXlH/LDLhlFnJzPZdHnVy9xisSPPRfFApJiwyfjRYdtslpJOcNgP6oPlpX/dy
  164. bbhOc9FEUgj/Q90Je8EfioBYFYsqVD6/dFv9SwIDAQABAoIBAFjSHK7gENVZxphO
  165. hHg8k9ShnDo8eyDvK8l9Op3U3/yOsXKxolivvyx//7UFmz3vXDahjNHe7YScAXdw
  166. eezbqBXa7xrvghqZzp2HhFYwMJ0210mcdncBKVFzK4ztZHxgQ0PFTqet0R19jZjl
  167. X3A325/eNZeuBeOied4qb/24AD6JGc6A0J55f5/QUQtdwYwrL15iC/KZXDL90PPJ
  168. CFJyrSzcXvOMEvOfXIFxhDVKRCppyIYXG7c80gtNC37I6rxxMNQ4mxjwUI2IVhxL
  169. j+nZDu0JgRZ4NaGjOq2e79QxUVm/GG3z25XgmBFBrXkEVV+sCZE1VDyj6kQfv9FU
  170. NhOrwGECgYEAzq47r/HwXifuGYBV/mvInFw3BNLrKry+iUZrJ4ms4g+LfOi0BAgf
  171. sXsWXulpBo2YgYjFdO8G66f69GlB4B7iLscpABXbRtpDZEnchQpaF36/+4g3i8gB
  172. Z29XHNDB8+7t4wbXvlSnLv1tZWey2fS4hPosc2YlvS87DMmnJMJqhs8CgYEA4oiB
  173. LGQP6VNdX0Uigmh5fL1g1k95eC8GP1ylczCcIwsb2OkAq0MT7SHRXOlg3leEq4+g
  174. mCHk1NdjkSYxDL2ZeTKTS/gy4p1jlcDa6Ilwi4pVvatNvu4o80EYWxRNNb1mAn67
  175. T8TN9lzc6mEi+LepQM3nYJ3F+ZWTKgxH8uoJwMUCgYEArpumE1vbjUBAuEyi2eGn
  176. RunlFW83fBCfDAxw5KM8anNlja5uvuU6GU/6s06QCxg+2lh5MPPrLdXpfukZ3UVa
  177. Itjg+5B7gx1MSALaiY8YU7cibFdFThM3lHIM72wyH2ogkWcrh0GvSFSUQlJcWCSW
  178. asmMGiYXBgBL697FFZomMyMCgYEAkAnp0JcDQwHd4gDsk2zoqnckBsDb5J5J46n+
  179. DYNAFEww9bgZ08u/9MzG+cPu8xFE621U2MbcYLVfuuBE2ewIlPaij/COMmeO9Z59
  180. 0tPpOuDH6eTtd1SptxqR6P+8pEn8feOlKHBj4Z1kXqdK/EiTlwAVeep4Al2oCFls
  181. ujkz4F0CgYAe8vHnVFHlWi16zAqZx4ZZZhNuqPtgFkvPg9LfyNTA4dz7F9xgtUaY
  182. nXBPyCe/8NtgBfT79HkPiG3TM0xRZY9UZgsJKFtqAu5u4ManuWDnsZI9RK2QTLHe
  183. yEbH5r3Dg3n9k/3GbjXFIWdU9UaYsdnSKHHtMw9ZODc14LaAogEQug==
  184. -----END RSA PRIVATE KEY-----`
  185. // client 2 crt is revoked
  186. client2Crt = `-----BEGIN CERTIFICATE-----
  187. MIIEITCCAgmgAwIBAgIRAM5p4duNjikSPxYGaqBm5FEwDQYJKoZIhvcNAQELBQAw
  188. EzERMA8GA1UEAxMIQ2VydEF1dGgwHhcNMjQwMTEwMTgxMjUyWhcNMzQwMTEwMTgy
  189. MTUzWjASMRAwDgYDVQQDEwdjbGllbnQyMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
  190. MIIBCgKCAQEApNYpNZVmXZtAObpRRIuP2o/7z04H2E161vKZvJ3LSLlUTImVjm/b
  191. Qe6DTNCUVLnzQuanmUlu2rUnN3lDSfYoBcJWbvC3y1OCPRkCjDV6KiYMA9TPkZua
  192. eq6y3+bFFfEmyumsVEe0bSuzNHXCOIBT7PqYMdovECcwBh/RZCA5mqO5omEKh4LQ
  193. cr6+sVVkvD3nsyx0Alz/kTLFqc0mVflmpJq+0BpdetHRg4n5vy/I/08jZ81PQAmT
  194. A0kyl0Jh132JBGFdA8eyugPPP8n5edU4f3HXV/nR7XLwBrpSt8KgEg8cwfAu4Ic0
  195. 6tGzB0CH8lSGtU0tH2/cOlDuguDD7VvokQIDAQABo3EwbzAOBgNVHQ8BAf8EBAMC
  196. A7gwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQUFBwMCMB0GA1UdDgQWBBR5mf0f
  197. Zjf8ZCGXqU2+45th7VkkLDAfBgNVHSMEGDAWgBTaCYSLw2UKxH/1UG1rrl4DZ7dp
  198. uTANBgkqhkiG9w0BAQsFAAOCAgEARhFxNAouwbpEfN1M90+ao5rwyxEewerSoCCz
  199. PQzeUZ66MA/FkS/tFUGgGGG+wERN+WLbe1cN6q/XFr0FSMLuUxLXDNV02oUL/FnY
  200. xcyNLaZUZ0pP7sA+Hmx2AdTA6baIwQbyIY9RLAaz6hzo1YbI8yeis645F1bxgL2D
  201. EP5kXa3Obv0tqWByMZtrmJPv3p0W5GJKXVDn51GR/E5KI7pliZX2e0LmMX9mxfPB
  202. 4sXFUggMHXxWMMSAmXPVsxC2KX6gMnajO7JUraTwuGm+6V371FzEX+UKXHI+xSvO
  203. 78TseTIYsBGLjeiA8UjkKlD3T9qsQm2mb2PlKyqjvIm4i2ilM0E2w4JZmd45b925
  204. 7q/QLV3NZ/zZMi6AMyULu28DWKfAx3RLKwnHWSFcR4lVkxQrbDhEUMhAhLAX+2+e
  205. qc7qZm3dTabi7ZJiiOvYK/yNgFHa/XtZp5uKPB5tigPIa+34hbZF7s2/ty5X3O1N
  206. f5Ardz7KNsxJjZIt6HvB28E/PPOvBqCKJc1Y08J9JbZi8p6QS1uarGoR7l7rT1Hv
  207. /ZXkNTw2bw1VpcWdzDBLLVHYNnJmS14189LVk11PcJJpSmubwCqg+ZZULdgtVr3S
  208. ANas2dgMPVwXhnAalgkcc+lb2QqaEz06axfbRGBsgnyqR5/koKCg1Hr0+vThHSsR
  209. E0+r2+4=
  210. -----END CERTIFICATE-----`
  211. client2Key = `-----BEGIN RSA PRIVATE KEY-----
  212. MIIEowIBAAKCAQEApNYpNZVmXZtAObpRRIuP2o/7z04H2E161vKZvJ3LSLlUTImV
  213. jm/bQe6DTNCUVLnzQuanmUlu2rUnN3lDSfYoBcJWbvC3y1OCPRkCjDV6KiYMA9TP
  214. kZuaeq6y3+bFFfEmyumsVEe0bSuzNHXCOIBT7PqYMdovECcwBh/RZCA5mqO5omEK
  215. h4LQcr6+sVVkvD3nsyx0Alz/kTLFqc0mVflmpJq+0BpdetHRg4n5vy/I/08jZ81P
  216. QAmTA0kyl0Jh132JBGFdA8eyugPPP8n5edU4f3HXV/nR7XLwBrpSt8KgEg8cwfAu
  217. 4Ic06tGzB0CH8lSGtU0tH2/cOlDuguDD7VvokQIDAQABAoIBAQCMnEeg9uXQmdvq
  218. op4qi6bV+ZcDWvvkLwvHikFMnYpIaheYBpF2ZMKzdmO4xgCSWeFCQ4Hah8KxfHCM
  219. qLuWvw2bBBE5J8yQ/JaPyeLbec7RX41GQ2YhPoxDdP0PdErREdpWo4imiFhH/Ewt
  220. Rvq7ufRdpdLoS8dzzwnvX3r+H2MkHoC/QANW2AOuVoZK5qyCH5N8yEAAbWKaQaeL
  221. VBhAYEVKbAkWEtXw7bYXzxRR7WIM3f45v3ncRusDIG+Hf75ZjatoH0lF1gHQNofO
  222. qkCVZVzjkLFuzDic2KZqsNORglNs4J6t5Dahb9v3hnoK963YMnVSUjFvqQ+/RZZy
  223. VILFShilAoGBANucwZU61eJ0tLKBYEwmRY/K7Gu1MvvcYJIOoX8/BL3zNmNO0CLl
  224. NiABtNt9WOVwZxDsxJXdo1zvMtAegNqS6W11R1VAZbL6mQ/krScbLDE6JKA5DmA7
  225. 4nNi1gJOW1ziAfdBAfhe4cLbQOb94xkOK5xM1YpO0xgDJLwrZbehDMmPAoGBAMAl
  226. /owPDAvcXz7JFynT0ieYVc64MSFiwGYJcsmxSAnbEgQ+TR5FtkHYe91OSqauZcCd
  227. aoKXQNyrYKIhyounRPFTdYQrlx6KtEs7LU9wOxuphhpJtGjRnhmA7IqvX703wNvu
  228. khrEavn86G5boH8R80371SrN0Rh9UeAlQGuNBdvfAoGAEAmokW9Ug08miwqrr6Pz
  229. 3IZjMZJwALidTM1IufQuMnj6ddIhnQrEIx48yPKkdUz6GeBQkuk2rujA+zXfDxc/
  230. eMDhzrX/N0zZtLFse7ieR5IJbrH7/MciyG5lVpHGVkgjAJ18uVikgAhm+vd7iC7i
  231. vG1YAtuyysQgAKXircBTIL0CgYAHeTLWVbt9NpwJwB6DhPaWjalAug9HIiUjktiB
  232. GcEYiQnBWn77X3DATOA8clAa/Yt9m2HKJIHkU1IV3ESZe+8Fh955PozJJlHu3yVb
  233. Ap157PUHTriSnxyMF2Sb3EhX/rQkmbnbCqqygHC14iBy8MrKzLG00X6BelZV5n0D
  234. 8d85dwKBgGWY2nsaemPH/TiTVF6kW1IKSQoIyJChkngc+Xj/2aCCkkmAEn8eqncl
  235. RKjnkiEZeG4+G91Xu7+HmcBLwV86k5I+tXK9O1Okomr6Zry8oqVcxU5TB6VRS+rA
  236. ubwF00Drdvk2+kDZfxIM137nBiy7wgCJi2Ksm5ihN3dUF6Q0oNPl
  237. -----END RSA PRIVATE KEY-----`
  238. testFileName = "test_file_ftp.dat"
  239. testDLFileName = "test_download_ftp.dat"
  240. tlsClient1Username = "client1"
  241. tlsClient2Username = "client2"
  242. httpFsPort = 23456
  243. defaultHTTPFsUsername = "httpfs_ftp_user"
  244. emptyPwdPlaceholder = "empty"
  245. )
  246. var (
  247. configDir = filepath.Join(".", "..", "..")
  248. allPerms = []string{dataprovider.PermAny}
  249. homeBasePath string
  250. hookCmdPath string
  251. extAuthPath string
  252. preLoginPath string
  253. postConnectPath string
  254. preDownloadPath string
  255. preUploadPath string
  256. logFilePath string
  257. caCrtPath string
  258. caCRLPath string
  259. )
  260. func TestMain(m *testing.M) { //nolint:gocyclo
  261. logFilePath = filepath.Join(configDir, "sftpgo_ftpd_test.log")
  262. bannerFileName := "banner_file"
  263. bannerFile := filepath.Join(configDir, bannerFileName)
  264. logger.InitLogger(logFilePath, 5, 1, 28, false, false, zerolog.DebugLevel)
  265. err := os.WriteFile(bannerFile, []byte("SFTPGo test ready\nsimple banner line\n"), os.ModePerm)
  266. if err != nil {
  267. logger.ErrorToConsole("error creating banner file: %v", err)
  268. os.Exit(1)
  269. }
  270. // we run the test cases with UploadMode atomic and resume support. The non atomic code path
  271. // simply does not execute some code so if it works in atomic mode will
  272. // work in non atomic mode too
  273. os.Setenv("SFTPGO_COMMON__UPLOAD_MODE", "2")
  274. os.Setenv("SFTPGO_DATA_PROVIDER__CREATE_DEFAULT_ADMIN", "1")
  275. os.Setenv("SFTPGO_COMMON__ALLOW_SELF_CONNECTIONS", "1")
  276. os.Setenv("SFTPGO_DEFAULT_ADMIN_USERNAME", "admin")
  277. os.Setenv("SFTPGO_DEFAULT_ADMIN_PASSWORD", "password")
  278. err = config.LoadConfig(configDir, "")
  279. if err != nil {
  280. logger.ErrorToConsole("error loading configuration: %v", err)
  281. os.Exit(1)
  282. }
  283. providerConf := config.GetProviderConf()
  284. logger.InfoToConsole("Starting FTPD tests, provider: %v", providerConf.Driver)
  285. commonConf := config.GetCommonConfig()
  286. homeBasePath = os.TempDir()
  287. if runtime.GOOS != osWindows {
  288. commonConf.Actions.ExecuteOn = []string{"download", "upload", "rename", "delete"}
  289. commonConf.Actions.Hook = hookCmdPath
  290. hookCmdPath, err = exec.LookPath("true")
  291. if err != nil {
  292. logger.Warn(logSender, "", "unable to get hook command: %v", err)
  293. logger.WarnToConsole("unable to get hook command: %v", err)
  294. }
  295. }
  296. certPath := filepath.Join(os.TempDir(), "test_ftpd.crt")
  297. keyPath := filepath.Join(os.TempDir(), "test_ftpd.key")
  298. caCrtPath = filepath.Join(os.TempDir(), "test_ftpd_ca.crt")
  299. caCRLPath = filepath.Join(os.TempDir(), "test_ftpd_crl.crt")
  300. err = writeCerts(certPath, keyPath, caCrtPath, caCRLPath)
  301. if err != nil {
  302. os.Exit(1)
  303. }
  304. err = dataprovider.Initialize(providerConf, configDir, true)
  305. if err != nil {
  306. logger.ErrorToConsole("error initializing data provider: %v", err)
  307. os.Exit(1)
  308. }
  309. err = common.Initialize(commonConf, 0)
  310. if err != nil {
  311. logger.WarnToConsole("error initializing common: %v", err)
  312. os.Exit(1)
  313. }
  314. httpConfig := config.GetHTTPConfig()
  315. httpConfig.Initialize(configDir) //nolint:errcheck
  316. kmsConfig := config.GetKMSConfig()
  317. err = kmsConfig.Initialize()
  318. if err != nil {
  319. logger.ErrorToConsole("error initializing kms: %v", err)
  320. os.Exit(1)
  321. }
  322. mfaConfig := config.GetMFAConfig()
  323. err = mfaConfig.Initialize()
  324. if err != nil {
  325. logger.ErrorToConsole("error initializing MFA: %v", err)
  326. os.Exit(1)
  327. }
  328. httpdConf := config.GetHTTPDConfig()
  329. httpdConf.Bindings[0].Port = 8079
  330. httpdtest.SetBaseURL("http://127.0.0.1:8079")
  331. ftpdConf := config.GetFTPDConfig()
  332. ftpdConf.Bindings = []ftpd.Binding{
  333. {
  334. Port: 2121,
  335. ClientAuthType: 2,
  336. CertificateFile: certPath,
  337. CertificateKeyFile: keyPath,
  338. },
  339. }
  340. ftpdConf.PassivePortRange.Start = 0
  341. ftpdConf.PassivePortRange.End = 0
  342. ftpdConf.BannerFile = bannerFileName
  343. ftpdConf.CACertificates = []string{caCrtPath}
  344. ftpdConf.CARevocationLists = []string{caCRLPath}
  345. ftpdConf.EnableSite = true
  346. // required to test sftpfs
  347. sftpdConf := config.GetSFTPDConfig()
  348. sftpdConf.Bindings = []sftpd.Binding{
  349. {
  350. Port: 2122,
  351. },
  352. }
  353. hostKeyPath := filepath.Join(os.TempDir(), "id_ed25519")
  354. sftpdConf.HostKeys = []string{hostKeyPath}
  355. extAuthPath = filepath.Join(homeBasePath, "extauth.sh")
  356. preLoginPath = filepath.Join(homeBasePath, "prelogin.sh")
  357. postConnectPath = filepath.Join(homeBasePath, "postconnect.sh")
  358. preDownloadPath = filepath.Join(homeBasePath, "predownload.sh")
  359. preUploadPath = filepath.Join(homeBasePath, "preupload.sh")
  360. status := ftpd.GetStatus()
  361. if status.IsActive {
  362. logger.ErrorToConsole("ftpd is already active")
  363. os.Exit(1)
  364. }
  365. go func() {
  366. logger.Debug(logSender, "", "initializing FTP server with config %+v", ftpdConf)
  367. if err := ftpdConf.Initialize(configDir); err != nil {
  368. logger.ErrorToConsole("could not start FTP server: %v", err)
  369. os.Exit(1)
  370. }
  371. }()
  372. go func() {
  373. logger.Debug(logSender, "", "initializing SFTP server with config %+v", sftpdConf)
  374. if err := sftpdConf.Initialize(configDir); err != nil {
  375. logger.ErrorToConsole("could not start SFTP server: %v", err)
  376. os.Exit(1)
  377. }
  378. }()
  379. go func() {
  380. if err := httpdConf.Initialize(configDir, 0); err != nil {
  381. logger.ErrorToConsole("could not start HTTP server: %v", err)
  382. os.Exit(1)
  383. }
  384. }()
  385. waitTCPListening(ftpdConf.Bindings[0].GetAddress())
  386. waitTCPListening(httpdConf.Bindings[0].GetAddress())
  387. waitTCPListening(sftpdConf.Bindings[0].GetAddress())
  388. ftpd.ReloadCertificateMgr() //nolint:errcheck
  389. ftpdConf = config.GetFTPDConfig()
  390. ftpdConf.Bindings = []ftpd.Binding{
  391. {
  392. Port: 2124,
  393. TLSMode: 2,
  394. },
  395. }
  396. ftpdConf.CertificateFile = certPath
  397. ftpdConf.CertificateKeyFile = keyPath
  398. ftpdConf.CACertificates = []string{caCrtPath}
  399. ftpdConf.CARevocationLists = []string{caCRLPath}
  400. ftpdConf.EnableSite = false
  401. ftpdConf.DisableActiveMode = true
  402. ftpdConf.CombineSupport = 1
  403. ftpdConf.HASHSupport = 1
  404. go func() {
  405. logger.Debug(logSender, "", "initializing FTP server with config %+v", ftpdConf)
  406. if err := ftpdConf.Initialize(configDir); err != nil {
  407. logger.ErrorToConsole("could not start FTP server: %v", err)
  408. os.Exit(1)
  409. }
  410. }()
  411. waitTCPListening(ftpdConf.Bindings[0].GetAddress())
  412. ftpdConf = config.GetFTPDConfig()
  413. ftpdConf.Bindings = []ftpd.Binding{
  414. {
  415. Port: 2126,
  416. CertificateFile: certPath,
  417. CertificateKeyFile: keyPath,
  418. TLSMode: 1,
  419. TLSSessionReuse: 1,
  420. ClientAuthType: 2,
  421. },
  422. }
  423. ftpdConf.CACertificates = []string{caCrtPath}
  424. ftpdConf.CARevocationLists = []string{caCRLPath}
  425. go func() {
  426. logger.Debug(logSender, "", "initializing FTP server with config %+v", ftpdConf)
  427. if err := ftpdConf.Initialize(configDir); err != nil {
  428. logger.ErrorToConsole("could not start FTP server: %v", err)
  429. os.Exit(1)
  430. }
  431. }()
  432. waitTCPListening(ftpdConf.Bindings[0].GetAddress())
  433. waitNoConnections()
  434. startHTTPFs()
  435. exitCode := m.Run()
  436. os.Remove(logFilePath)
  437. os.Remove(bannerFile)
  438. os.Remove(extAuthPath)
  439. os.Remove(preLoginPath)
  440. os.Remove(postConnectPath)
  441. os.Remove(preDownloadPath)
  442. os.Remove(preUploadPath)
  443. os.Remove(certPath)
  444. os.Remove(keyPath)
  445. os.Remove(caCrtPath)
  446. os.Remove(caCRLPath)
  447. os.Remove(hostKeyPath)
  448. os.Remove(hostKeyPath + ".pub")
  449. os.Exit(exitCode)
  450. }
  451. func TestInitializationFailure(t *testing.T) {
  452. ftpdConf := config.GetFTPDConfig()
  453. ftpdConf.Bindings = []ftpd.Binding{}
  454. ftpdConf.CertificateFile = filepath.Join(os.TempDir(), "test_ftpd.crt")
  455. ftpdConf.CertificateKeyFile = filepath.Join(os.TempDir(), "test_ftpd.key")
  456. err := ftpdConf.Initialize(configDir)
  457. require.EqualError(t, err, common.ErrNoBinding.Error())
  458. ftpdConf.Bindings = []ftpd.Binding{
  459. {
  460. Port: 0,
  461. },
  462. {
  463. Port: 2121,
  464. },
  465. }
  466. ftpdConf.BannerFile = "a-missing-file"
  467. err = ftpdConf.Initialize(configDir)
  468. require.Error(t, err)
  469. ftpdConf.BannerFile = ""
  470. ftpdConf.Bindings[1].TLSMode = 10
  471. err = ftpdConf.Initialize(configDir)
  472. require.Error(t, err)
  473. ftpdConf.CertificateFile = ""
  474. ftpdConf.CertificateKeyFile = ""
  475. ftpdConf.Bindings[1].TLSMode = 1
  476. err = ftpdConf.Initialize(configDir)
  477. require.Error(t, err)
  478. certPath := filepath.Join(os.TempDir(), "test_ftpd.crt")
  479. keyPath := filepath.Join(os.TempDir(), "test_ftpd.key")
  480. ftpdConf.CertificateFile = certPath
  481. ftpdConf.CertificateKeyFile = keyPath
  482. ftpdConf.CACertificates = []string{"invalid ca cert"}
  483. err = ftpdConf.Initialize(configDir)
  484. require.Error(t, err)
  485. ftpdConf.CACertificates = nil
  486. ftpdConf.CARevocationLists = []string{""}
  487. err = ftpdConf.Initialize(configDir)
  488. require.Error(t, err)
  489. ftpdConf.CACertificates = []string{caCrtPath}
  490. ftpdConf.CARevocationLists = []string{caCRLPath}
  491. ftpdConf.Bindings[1].ForcePassiveIP = "127001"
  492. err = ftpdConf.Initialize(configDir)
  493. require.Error(t, err)
  494. require.Contains(t, err.Error(), "the provided passive IP \"127001\" is not valid")
  495. ftpdConf.Bindings[1].ForcePassiveIP = ""
  496. ftpdConf.Bindings[1].TLSMode = 2
  497. ftpdConf.Bindings[1].TLSSessionReuse = 1
  498. err = ftpdConf.Initialize(configDir)
  499. require.Error(t, err, "TLS session resumption should not be supported with implicit FTPS")
  500. ftpdConf.Bindings[1].TLSMode = 0
  501. ftpdConf.Bindings[1].TLSSessionReuse = 100
  502. err = ftpdConf.Initialize(configDir)
  503. require.Error(t, err)
  504. assert.Contains(t, err.Error(), "unsupported TLS reuse mode")
  505. ftpdConf.Bindings[1].TLSSessionReuse = 0
  506. err = ftpdConf.Initialize(configDir)
  507. require.Error(t, err)
  508. err = dataprovider.Close()
  509. assert.NoError(t, err)
  510. err = ftpdConf.Initialize(configDir)
  511. if assert.Error(t, err) {
  512. assert.Contains(t, err.Error(), "unable to load config from provider")
  513. }
  514. err = config.LoadConfig(configDir, "")
  515. assert.NoError(t, err)
  516. providerConf := config.GetProviderConf()
  517. err = dataprovider.Initialize(providerConf, configDir, true)
  518. assert.NoError(t, err)
  519. }
  520. func TestBasicFTPHandling(t *testing.T) {
  521. u := getTestUser()
  522. u.QuotaSize = 6553600
  523. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  524. assert.NoError(t, err)
  525. u = getTestSFTPUser()
  526. u.QuotaSize = 6553600
  527. sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  528. assert.NoError(t, err)
  529. for _, user := range []dataprovider.User{localUser, sftpUser} {
  530. client, err := getFTPClient(user, true, nil)
  531. if assert.NoError(t, err) {
  532. if user.Username == defaultUsername {
  533. assert.Len(t, common.Connections.GetStats(""), 1)
  534. } else {
  535. assert.Len(t, common.Connections.GetStats(""), 2)
  536. }
  537. testFilePath := filepath.Join(homeBasePath, testFileName)
  538. testFileSize := int64(65535)
  539. expectedQuotaSize := testFileSize
  540. expectedQuotaFiles := 1
  541. err = createTestFile(testFilePath, testFileSize)
  542. assert.NoError(t, err)
  543. err = checkBasicFTP(client)
  544. assert.NoError(t, err)
  545. err = ftpUploadFile(testFilePath, path.Join("/missing_dir", testFileName), testFileSize, client, 0)
  546. assert.Error(t, err)
  547. user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
  548. assert.NoError(t, err)
  549. assert.Equal(t, int64(0), user.FirstUpload)
  550. assert.Equal(t, int64(0), user.FirstDownload)
  551. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  552. assert.NoError(t, err)
  553. user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
  554. assert.NoError(t, err)
  555. assert.Greater(t, user.FirstUpload, int64(0))
  556. assert.Equal(t, int64(0), user.FirstDownload)
  557. // overwrite an existing file
  558. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  559. assert.NoError(t, err)
  560. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  561. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  562. assert.NoError(t, err)
  563. user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
  564. assert.NoError(t, err)
  565. assert.Equal(t, expectedQuotaFiles, user.UsedQuotaFiles)
  566. assert.Equal(t, expectedQuotaSize, user.UsedQuotaSize)
  567. assert.Greater(t, user.FirstUpload, int64(0))
  568. assert.Greater(t, user.FirstDownload, int64(0))
  569. err = client.Rename(testFileName, testFileName+"1")
  570. assert.NoError(t, err)
  571. err = client.Delete(testFileName)
  572. assert.Error(t, err)
  573. err = client.Delete(testFileName + "1")
  574. assert.NoError(t, err)
  575. user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
  576. assert.NoError(t, err)
  577. assert.Equal(t, expectedQuotaFiles-1, user.UsedQuotaFiles)
  578. assert.Equal(t, expectedQuotaSize-testFileSize, user.UsedQuotaSize)
  579. curDir, err := client.CurrentDir()
  580. if assert.NoError(t, err) {
  581. assert.Equal(t, "/", curDir)
  582. }
  583. testDir := "testDir"
  584. err = client.MakeDir(testDir)
  585. assert.NoError(t, err)
  586. err = client.ChangeDir(testDir)
  587. assert.NoError(t, err)
  588. curDir, err = client.CurrentDir()
  589. if assert.NoError(t, err) {
  590. assert.Equal(t, path.Join("/", testDir), curDir)
  591. }
  592. res, err := client.List(path.Join("/", testDir))
  593. assert.NoError(t, err)
  594. assert.Len(t, res, 0)
  595. res, err = client.List(path.Join("/"))
  596. assert.NoError(t, err)
  597. if assert.Len(t, res, 1) {
  598. assert.Equal(t, testDir, res[0].Name)
  599. }
  600. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  601. assert.NoError(t, err)
  602. _, err = client.FileSize(path.Join("/", testDir))
  603. assert.Error(t, err)
  604. size, err := client.FileSize(path.Join("/", testDir, testFileName))
  605. assert.NoError(t, err)
  606. assert.Equal(t, testFileSize, size)
  607. err = client.ChangeDirToParent()
  608. assert.NoError(t, err)
  609. curDir, err = client.CurrentDir()
  610. if assert.NoError(t, err) {
  611. assert.Equal(t, "/", curDir)
  612. }
  613. err = client.Delete(path.Join("/", testDir, testFileName))
  614. assert.NoError(t, err)
  615. err = client.Delete(testDir)
  616. assert.Error(t, err)
  617. err = client.RemoveDir(testDir)
  618. assert.NoError(t, err)
  619. err = os.Remove(testFilePath)
  620. assert.NoError(t, err)
  621. err = os.Remove(localDownloadPath)
  622. assert.NoError(t, err)
  623. err = client.Quit()
  624. assert.NoError(t, err)
  625. }
  626. }
  627. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  628. assert.NoError(t, err)
  629. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  630. assert.NoError(t, err)
  631. err = os.RemoveAll(localUser.GetHomeDir())
  632. assert.NoError(t, err)
  633. assert.Eventually(t, func() bool { return len(common.Connections.GetStats("")) == 0 }, 1*time.Second, 50*time.Millisecond)
  634. assert.Eventually(t, func() bool { return common.Connections.GetClientConnections() == 0 }, 1000*time.Millisecond,
  635. 50*time.Millisecond)
  636. }
  637. func TestHTTPFs(t *testing.T) {
  638. u := getTestUserWithHTTPFs()
  639. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  640. assert.NoError(t, err)
  641. client, err := getFTPClient(user, true, nil)
  642. if assert.NoError(t, err) {
  643. err = checkBasicFTP(client)
  644. assert.NoError(t, err)
  645. testFilePath := filepath.Join(homeBasePath, testFileName)
  646. testFileSize := int64(65535)
  647. err = createTestFile(testFilePath, testFileSize)
  648. assert.NoError(t, err)
  649. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  650. assert.NoError(t, err)
  651. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  652. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  653. assert.NoError(t, err)
  654. // test a download resume
  655. data := []byte("test data")
  656. err = os.WriteFile(testFilePath, data, os.ModePerm)
  657. assert.NoError(t, err)
  658. err = ftpUploadFile(testFilePath, testFileName, int64(len(data)), client, 0)
  659. assert.NoError(t, err)
  660. err = ftpDownloadFile(testFileName, localDownloadPath, int64(len(data)-5), client, 5)
  661. assert.NoError(t, err)
  662. readed, err := os.ReadFile(localDownloadPath)
  663. assert.NoError(t, err)
  664. assert.Equal(t, []byte("data"), readed, "readed data mismatch: %q", string(readed))
  665. err = os.Remove(testFilePath)
  666. assert.NoError(t, err)
  667. err = os.Remove(localDownloadPath)
  668. assert.NoError(t, err)
  669. err = client.Quit()
  670. assert.NoError(t, err)
  671. }
  672. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  673. assert.NoError(t, err)
  674. err = os.RemoveAll(user.GetHomeDir())
  675. assert.NoError(t, err)
  676. assert.Eventually(t, func() bool { return len(common.Connections.GetStats("")) == 0 }, 1*time.Second, 50*time.Millisecond)
  677. assert.Eventually(t, func() bool { return common.Connections.GetClientConnections() == 0 }, 1000*time.Millisecond,
  678. 50*time.Millisecond)
  679. }
  680. func TestListDirWithWildcards(t *testing.T) {
  681. localUser, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  682. assert.NoError(t, err)
  683. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  684. assert.NoError(t, err)
  685. defer func() {
  686. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  687. assert.NoError(t, err)
  688. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  689. assert.NoError(t, err)
  690. err = os.RemoveAll(localUser.GetHomeDir())
  691. assert.NoError(t, err)
  692. }()
  693. for _, user := range []dataprovider.User{localUser, sftpUser} {
  694. client, err := getFTPClient(user, true, nil, ftp.DialWithDisabledMLSD(true))
  695. if assert.NoError(t, err) {
  696. dir1 := "test.dir"
  697. dir2 := "test.dir1"
  698. err = client.MakeDir(dir1)
  699. assert.NoError(t, err)
  700. err = client.MakeDir(dir2)
  701. assert.NoError(t, err)
  702. testFilePath := filepath.Join(homeBasePath, testFileName)
  703. testFileSize := int64(65535)
  704. err = createTestFile(testFilePath, testFileSize)
  705. assert.NoError(t, err)
  706. fileName := "file[a-z]e.dat"
  707. err = ftpUploadFile(testFilePath, fileName, testFileSize, client, 0)
  708. assert.NoError(t, err)
  709. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  710. err = ftpDownloadFile(fileName, localDownloadPath, testFileSize, client, 0)
  711. assert.NoError(t, err)
  712. entries, err := client.List(fileName)
  713. require.NoError(t, err)
  714. require.Len(t, entries, 1)
  715. assert.Equal(t, fileName, entries[0].Name)
  716. nListEntries, err := client.NameList(fileName)
  717. require.NoError(t, err)
  718. require.Len(t, entries, 1)
  719. assert.Contains(t, nListEntries, fileName)
  720. entries, err = client.List(".")
  721. require.NoError(t, err)
  722. require.Len(t, entries, 3)
  723. nListEntries, err = client.NameList(".")
  724. require.NoError(t, err)
  725. require.Len(t, nListEntries, 3)
  726. entries, err = client.List("/test.*")
  727. require.NoError(t, err)
  728. require.Len(t, entries, 2)
  729. found := 0
  730. for _, e := range entries {
  731. switch e.Name {
  732. case dir1, dir2:
  733. found++
  734. }
  735. }
  736. assert.Equal(t, 2, found)
  737. nListEntries, err = client.NameList("/test.*")
  738. require.NoError(t, err)
  739. require.Len(t, entries, 2)
  740. assert.Contains(t, nListEntries, dir1)
  741. assert.Contains(t, nListEntries, dir2)
  742. entries, err = client.List("/*.dir?")
  743. require.NoError(t, err)
  744. assert.Len(t, entries, 1)
  745. assert.Equal(t, dir2, entries[0].Name)
  746. nListEntries, err = client.NameList("/*.dir?")
  747. require.NoError(t, err)
  748. require.Len(t, entries, 1)
  749. assert.Contains(t, nListEntries, dir2)
  750. entries, err = client.List("/test.???")
  751. require.NoError(t, err)
  752. require.Len(t, entries, 1)
  753. assert.Equal(t, dir1, entries[0].Name)
  754. nListEntries, err = client.NameList("/test.???")
  755. require.NoError(t, err)
  756. require.Len(t, entries, 1)
  757. assert.Contains(t, nListEntries, dir1)
  758. _, err = client.NameList("/missingdir/test.*")
  759. assert.Error(t, err)
  760. _, err = client.List("/missingdir/test.*")
  761. assert.Error(t, err)
  762. _, err = client.NameList("test[-]")
  763. if assert.Error(t, err) {
  764. assert.Contains(t, err.Error(), path.ErrBadPattern.Error())
  765. }
  766. _, err = client.List("test[-]")
  767. if assert.Error(t, err) {
  768. assert.Contains(t, err.Error(), path.ErrBadPattern.Error())
  769. }
  770. subDir := path.Join(dir1, "sub.d")
  771. err = client.MakeDir(subDir)
  772. assert.NoError(t, err)
  773. err = client.ChangeDir(path.Dir(subDir))
  774. assert.NoError(t, err)
  775. entries, err = client.List("sub.?")
  776. require.NoError(t, err)
  777. require.Len(t, entries, 1)
  778. assert.Contains(t, path.Base(subDir), entries[0].Name)
  779. nListEntries, err = client.NameList("sub.?")
  780. require.NoError(t, err)
  781. require.Len(t, entries, 1)
  782. assert.Contains(t, nListEntries, path.Base(subDir))
  783. entries, err = client.List("../*.dir?")
  784. require.NoError(t, err)
  785. require.Len(t, entries, 1)
  786. assert.Equal(t, path.Join("../", dir2), entries[0].Name)
  787. nListEntries, err = client.NameList("../*.dir?")
  788. require.NoError(t, err)
  789. require.Len(t, entries, 1)
  790. assert.Contains(t, nListEntries, path.Join("../", dir2))
  791. err = client.ChangeDir("/")
  792. assert.NoError(t, err)
  793. entries, err = client.List(path.Join(dir1, "sub.*"))
  794. require.NoError(t, err)
  795. require.Len(t, entries, 1)
  796. assert.Equal(t, path.Join(dir1, "sub.d"), entries[0].Name)
  797. nListEntries, err = client.NameList(path.Join(dir1, "sub.*"))
  798. require.NoError(t, err)
  799. require.Len(t, entries, 1)
  800. assert.Contains(t, nListEntries, path.Join(dir1, "sub.d"))
  801. err = client.RemoveDir(subDir)
  802. assert.NoError(t, err)
  803. err = client.RemoveDir(dir1)
  804. assert.NoError(t, err)
  805. err = client.RemoveDir(dir2)
  806. assert.NoError(t, err)
  807. err = os.Remove(testFilePath)
  808. assert.NoError(t, err)
  809. err = os.Remove(localDownloadPath)
  810. assert.NoError(t, err)
  811. err = client.Quit()
  812. assert.NoError(t, err)
  813. }
  814. }
  815. }
  816. func TestStartDirectory(t *testing.T) {
  817. startDir := "/start/dir"
  818. u := getTestUser()
  819. u.Filters.StartDirectory = startDir
  820. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  821. assert.NoError(t, err)
  822. u = getTestSFTPUser()
  823. u.Filters.StartDirectory = startDir
  824. sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  825. assert.NoError(t, err)
  826. for _, user := range []dataprovider.User{localUser, sftpUser} {
  827. client, err := getFTPClient(user, true, nil)
  828. if assert.NoError(t, err) {
  829. currentDir, err := client.CurrentDir()
  830. assert.NoError(t, err)
  831. assert.Equal(t, startDir, currentDir)
  832. testFilePath := filepath.Join(homeBasePath, testFileName)
  833. testFileSize := int64(65535)
  834. err = createTestFile(testFilePath, testFileSize)
  835. assert.NoError(t, err)
  836. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  837. assert.NoError(t, err)
  838. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  839. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  840. assert.NoError(t, err)
  841. entries, err := client.List(".")
  842. assert.NoError(t, err)
  843. if assert.Len(t, entries, 1) {
  844. assert.Equal(t, testFileName, entries[0].Name)
  845. }
  846. entries, err = client.List("/")
  847. assert.NoError(t, err)
  848. if assert.Len(t, entries, 1) {
  849. assert.Equal(t, "start", entries[0].Name)
  850. }
  851. err = client.ChangeDirToParent()
  852. assert.NoError(t, err)
  853. currentDir, err = client.CurrentDir()
  854. assert.NoError(t, err)
  855. assert.Equal(t, path.Dir(startDir), currentDir)
  856. err = client.ChangeDirToParent()
  857. assert.NoError(t, err)
  858. currentDir, err = client.CurrentDir()
  859. assert.NoError(t, err)
  860. assert.Equal(t, "/", currentDir)
  861. err = os.Remove(testFilePath)
  862. assert.NoError(t, err)
  863. err = os.Remove(localDownloadPath)
  864. assert.NoError(t, err)
  865. err = client.Quit()
  866. assert.NoError(t, err)
  867. }
  868. }
  869. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  870. assert.NoError(t, err)
  871. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  872. assert.NoError(t, err)
  873. err = os.RemoveAll(localUser.GetHomeDir())
  874. assert.NoError(t, err)
  875. }
  876. func TestLoginEmptyPassword(t *testing.T) {
  877. u := getTestUser()
  878. u.Password = ""
  879. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  880. assert.NoError(t, err)
  881. user.Password = emptyPwdPlaceholder
  882. _, err = getFTPClient(user, true, nil)
  883. assert.Error(t, err)
  884. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  885. assert.NoError(t, err)
  886. err = os.RemoveAll(user.GetHomeDir())
  887. assert.NoError(t, err)
  888. }
  889. func TestAnonymousUser(t *testing.T) {
  890. u := getTestUser()
  891. u.Password = ""
  892. u.Filters.IsAnonymous = true
  893. _, _, err := httpdtest.AddUser(u, http.StatusCreated)
  894. assert.Error(t, err)
  895. user, _, err := httpdtest.GetUserByUsername(u.Username, http.StatusOK)
  896. assert.NoError(t, err)
  897. assert.True(t, user.Filters.IsAnonymous)
  898. assert.Equal(t, []string{dataprovider.PermListItems, dataprovider.PermDownload}, user.Permissions["/"])
  899. assert.Equal(t, []string{common.ProtocolSSH, common.ProtocolHTTP}, user.Filters.DeniedProtocols)
  900. assert.Equal(t, []string{dataprovider.SSHLoginMethodPublicKey, dataprovider.SSHLoginMethodPassword,
  901. dataprovider.SSHLoginMethodKeyboardInteractive, dataprovider.SSHLoginMethodKeyAndPassword,
  902. dataprovider.SSHLoginMethodKeyAndKeyboardInt, dataprovider.LoginMethodTLSCertificate,
  903. dataprovider.LoginMethodTLSCertificateAndPwd}, user.Filters.DeniedLoginMethods)
  904. user.Password = emptyPwdPlaceholder
  905. client, err := getFTPClient(user, true, nil)
  906. if assert.NoError(t, err) {
  907. err = checkBasicFTP(client)
  908. assert.NoError(t, err)
  909. testFilePath := filepath.Join(homeBasePath, testFileName)
  910. testFileSize := int64(65535)
  911. err = createTestFile(testFilePath, testFileSize)
  912. assert.NoError(t, err)
  913. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  914. if assert.Error(t, err) {
  915. assert.Contains(t, err.Error(), "permission")
  916. }
  917. err = os.Rename(testFilePath, filepath.Join(user.GetHomeDir(), testFileName))
  918. assert.NoError(t, err)
  919. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  920. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  921. assert.NoError(t, err)
  922. err = client.MakeDir("adir")
  923. if assert.Error(t, err) {
  924. assert.Contains(t, err.Error(), "permission")
  925. }
  926. err = client.Quit()
  927. assert.NoError(t, err)
  928. err = os.Remove(localDownloadPath)
  929. assert.NoError(t, err)
  930. }
  931. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  932. assert.NoError(t, err)
  933. err = os.RemoveAll(user.GetHomeDir())
  934. assert.NoError(t, err)
  935. }
  936. func TestAnonymousGroupInheritance(t *testing.T) {
  937. g := getTestGroup()
  938. g.UserSettings.Filters.IsAnonymous = true
  939. g.UserSettings.Permissions = make(map[string][]string)
  940. g.UserSettings.Permissions["/"] = allPerms
  941. g.UserSettings.Permissions["/testsub"] = allPerms
  942. group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
  943. assert.NoError(t, err)
  944. u := getTestUser()
  945. u.Groups = []sdk.GroupMapping{
  946. {
  947. Name: group.Name,
  948. Type: sdk.GroupTypePrimary,
  949. },
  950. }
  951. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  952. assert.NoError(t, err)
  953. user.Password = emptyPwdPlaceholder
  954. client, err := getFTPClient(user, true, nil)
  955. if assert.NoError(t, err) {
  956. err = checkBasicFTP(client)
  957. assert.NoError(t, err)
  958. testFilePath := filepath.Join(homeBasePath, testFileName)
  959. testFileSize := int64(65535)
  960. err = createTestFile(testFilePath, testFileSize)
  961. assert.NoError(t, err)
  962. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  963. if assert.Error(t, err) {
  964. assert.Contains(t, err.Error(), "permission")
  965. }
  966. err = client.MakeDir("adir")
  967. if assert.Error(t, err) {
  968. assert.Contains(t, err.Error(), "permission")
  969. }
  970. err = client.MakeDir("/testsub/adir")
  971. if assert.Error(t, err) {
  972. assert.Contains(t, err.Error(), "permission")
  973. }
  974. err = os.Rename(testFilePath, filepath.Join(user.GetHomeDir(), testFileName))
  975. assert.NoError(t, err)
  976. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  977. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  978. assert.NoError(t, err)
  979. err = client.Quit()
  980. assert.NoError(t, err)
  981. err = os.Remove(localDownloadPath)
  982. assert.NoError(t, err)
  983. }
  984. user.Password = defaultPassword
  985. client, err = getFTPClient(user, true, nil)
  986. if assert.NoError(t, err) {
  987. err = checkBasicFTP(client)
  988. assert.NoError(t, err)
  989. err := client.Quit()
  990. assert.NoError(t, err)
  991. }
  992. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  993. assert.NoError(t, err)
  994. err = os.RemoveAll(user.GetHomeDir())
  995. assert.NoError(t, err)
  996. _, err = httpdtest.RemoveGroup(group, http.StatusOK)
  997. assert.NoError(t, err)
  998. }
  999. func TestMultiFactorAuth(t *testing.T) {
  1000. u := getTestUser()
  1001. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1002. assert.NoError(t, err)
  1003. configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
  1004. assert.NoError(t, err)
  1005. user.Password = defaultPassword
  1006. user.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
  1007. Enabled: true,
  1008. ConfigName: configName,
  1009. Secret: kms.NewPlainSecret(key.Secret()),
  1010. Protocols: []string{common.ProtocolFTP},
  1011. }
  1012. err = dataprovider.UpdateUser(&user, "", "", "")
  1013. assert.NoError(t, err)
  1014. user.Password = defaultPassword
  1015. _, err = getFTPClient(user, true, nil)
  1016. if assert.Error(t, err) {
  1017. assert.Contains(t, err.Error(), dataprovider.ErrInvalidCredentials.Error())
  1018. }
  1019. passcode, err := generateTOTPPasscode(key.Secret(), otp.AlgorithmSHA1)
  1020. assert.NoError(t, err)
  1021. user.Password = defaultPassword + passcode
  1022. client, err := getFTPClient(user, true, nil)
  1023. if assert.NoError(t, err) {
  1024. err = checkBasicFTP(client)
  1025. assert.NoError(t, err)
  1026. err := client.Quit()
  1027. assert.NoError(t, err)
  1028. }
  1029. // reusing the same passcode should not work
  1030. _, err = getFTPClient(user, true, nil)
  1031. if assert.Error(t, err) {
  1032. assert.Contains(t, err.Error(), dataprovider.ErrInvalidCredentials.Error())
  1033. }
  1034. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1035. assert.NoError(t, err)
  1036. err = os.RemoveAll(user.GetHomeDir())
  1037. assert.NoError(t, err)
  1038. }
  1039. func TestMustChangePasswordRequirement(t *testing.T) {
  1040. u := getTestUser()
  1041. u.Filters.RequirePasswordChange = true
  1042. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1043. assert.NoError(t, err)
  1044. _, err = getFTPClient(user, true, nil)
  1045. assert.Error(t, err)
  1046. err = dataprovider.UpdateUserPassword(user.Username, defaultPassword, "", "", "")
  1047. assert.NoError(t, err)
  1048. client, err := getFTPClient(user, true, nil)
  1049. if assert.NoError(t, err) {
  1050. err = checkBasicFTP(client)
  1051. assert.NoError(t, err)
  1052. err := client.Quit()
  1053. assert.NoError(t, err)
  1054. }
  1055. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1056. assert.NoError(t, err)
  1057. err = os.RemoveAll(user.GetHomeDir())
  1058. assert.NoError(t, err)
  1059. }
  1060. func TestSecondFactorRequirement(t *testing.T) {
  1061. u := getTestUser()
  1062. u.Filters.TwoFactorAuthProtocols = []string{common.ProtocolFTP}
  1063. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1064. assert.NoError(t, err)
  1065. _, err = getFTPClient(user, true, nil)
  1066. if assert.Error(t, err) {
  1067. assert.Contains(t, err.Error(), "second factor authentication is not set")
  1068. }
  1069. configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
  1070. assert.NoError(t, err)
  1071. user.Password = defaultPassword
  1072. user.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
  1073. Enabled: true,
  1074. ConfigName: configName,
  1075. Secret: kms.NewPlainSecret(key.Secret()),
  1076. Protocols: []string{common.ProtocolFTP},
  1077. }
  1078. err = dataprovider.UpdateUser(&user, "", "", "")
  1079. assert.NoError(t, err)
  1080. passcode, err := generateTOTPPasscode(key.Secret(), otp.AlgorithmSHA1)
  1081. assert.NoError(t, err)
  1082. user.Password = defaultPassword + passcode
  1083. client, err := getFTPClient(user, true, nil)
  1084. if assert.NoError(t, err) {
  1085. err = checkBasicFTP(client)
  1086. assert.NoError(t, err)
  1087. err := client.Quit()
  1088. assert.NoError(t, err)
  1089. }
  1090. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1091. assert.NoError(t, err)
  1092. err = os.RemoveAll(user.GetHomeDir())
  1093. assert.NoError(t, err)
  1094. }
  1095. func TestLoginInvalidCredentials(t *testing.T) {
  1096. u := getTestUser()
  1097. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1098. assert.NoError(t, err)
  1099. user.Username = "wrong username"
  1100. _, err = getFTPClient(user, false, nil)
  1101. if assert.Error(t, err) {
  1102. assert.Contains(t, err.Error(), dataprovider.ErrInvalidCredentials.Error())
  1103. }
  1104. user.Username = u.Username
  1105. user.Password = "wrong pwd"
  1106. _, err = getFTPClient(user, false, nil)
  1107. if assert.Error(t, err) {
  1108. assert.Contains(t, err.Error(), dataprovider.ErrInvalidCredentials.Error())
  1109. }
  1110. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1111. assert.NoError(t, err)
  1112. }
  1113. func TestLoginNonExistentUser(t *testing.T) {
  1114. user := getTestUser()
  1115. _, err := getFTPClient(user, false, nil)
  1116. assert.Error(t, err)
  1117. }
  1118. func TestFTPSecurity(t *testing.T) {
  1119. u := getTestUser()
  1120. u.Filters.FTPSecurity = 1
  1121. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1122. assert.NoError(t, err)
  1123. client, err := getFTPClient(user, true, nil)
  1124. if assert.NoError(t, err) {
  1125. err = checkBasicFTP(client)
  1126. assert.NoError(t, err)
  1127. err := client.Quit()
  1128. assert.NoError(t, err)
  1129. }
  1130. _, err = getFTPClient(user, false, nil)
  1131. if assert.Error(t, err) {
  1132. assert.Contains(t, err.Error(), "TLS is required")
  1133. }
  1134. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1135. assert.NoError(t, err)
  1136. err = os.RemoveAll(user.GetHomeDir())
  1137. assert.NoError(t, err)
  1138. }
  1139. func TestGroupFTPSecurity(t *testing.T) {
  1140. g := getTestGroup()
  1141. g.UserSettings.Filters.FTPSecurity = 1
  1142. group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
  1143. assert.NoError(t, err)
  1144. u := getTestUser()
  1145. u.Groups = []sdk.GroupMapping{
  1146. {
  1147. Name: group.Name,
  1148. Type: sdk.GroupTypePrimary,
  1149. },
  1150. }
  1151. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1152. assert.NoError(t, err)
  1153. client, err := getFTPClient(user, true, nil)
  1154. if assert.NoError(t, err) {
  1155. err = checkBasicFTP(client)
  1156. assert.NoError(t, err)
  1157. err := client.Quit()
  1158. assert.NoError(t, err)
  1159. }
  1160. _, err = getFTPClient(user, false, nil)
  1161. if assert.Error(t, err) {
  1162. assert.Contains(t, err.Error(), "TLS is required")
  1163. }
  1164. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1165. assert.NoError(t, err)
  1166. err = os.RemoveAll(user.GetHomeDir())
  1167. assert.NoError(t, err)
  1168. _, err = httpdtest.RemoveGroup(group, http.StatusOK)
  1169. assert.NoError(t, err)
  1170. }
  1171. func TestLoginExternalAuth(t *testing.T) {
  1172. if runtime.GOOS == osWindows {
  1173. t.Skip("this test is not available on Windows")
  1174. }
  1175. u := getTestUser()
  1176. err := dataprovider.Close()
  1177. assert.NoError(t, err)
  1178. err = config.LoadConfig(configDir, "")
  1179. assert.NoError(t, err)
  1180. providerConf := config.GetProviderConf()
  1181. err = os.WriteFile(extAuthPath, getExtAuthScriptContent(u), os.ModePerm)
  1182. assert.NoError(t, err)
  1183. providerConf.ExternalAuthHook = extAuthPath
  1184. providerConf.ExternalAuthScope = 0
  1185. err = dataprovider.Initialize(providerConf, configDir, true)
  1186. assert.NoError(t, err)
  1187. g := getTestGroup()
  1188. g.UserSettings.Filters.DeniedProtocols = []string{common.ProtocolFTP}
  1189. group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
  1190. assert.NoError(t, err)
  1191. client, err := getFTPClient(u, true, nil)
  1192. if assert.NoError(t, err) {
  1193. err = checkBasicFTP(client)
  1194. assert.NoError(t, err)
  1195. err := client.Quit()
  1196. assert.NoError(t, err)
  1197. }
  1198. u.Groups = []sdk.GroupMapping{
  1199. {
  1200. Name: group.Name,
  1201. Type: sdk.GroupTypePrimary,
  1202. },
  1203. }
  1204. err = os.WriteFile(extAuthPath, getExtAuthScriptContent(u), os.ModePerm)
  1205. assert.NoError(t, err)
  1206. _, err = getFTPClient(u, true, nil)
  1207. if !assert.Error(t, err) {
  1208. err := client.Quit()
  1209. assert.NoError(t, err)
  1210. } else {
  1211. assert.Contains(t, err.Error(), "protocol FTP is not allowed")
  1212. }
  1213. u.Groups = nil
  1214. err = os.WriteFile(extAuthPath, getExtAuthScriptContent(u), os.ModePerm)
  1215. assert.NoError(t, err)
  1216. u.Username = defaultUsername + "1"
  1217. client, err = getFTPClient(u, true, nil)
  1218. if !assert.Error(t, err) {
  1219. err := client.Quit()
  1220. assert.NoError(t, err)
  1221. } else {
  1222. assert.Contains(t, err.Error(), "invalid credentials")
  1223. }
  1224. user, _, err := httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
  1225. assert.NoError(t, err)
  1226. assert.Equal(t, defaultUsername, user.Username)
  1227. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1228. assert.NoError(t, err)
  1229. err = os.RemoveAll(user.GetHomeDir())
  1230. assert.NoError(t, err)
  1231. _, err = httpdtest.RemoveGroup(group, http.StatusOK)
  1232. assert.NoError(t, err)
  1233. err = dataprovider.Close()
  1234. assert.NoError(t, err)
  1235. err = config.LoadConfig(configDir, "")
  1236. assert.NoError(t, err)
  1237. providerConf = config.GetProviderConf()
  1238. err = dataprovider.Initialize(providerConf, configDir, true)
  1239. assert.NoError(t, err)
  1240. err = os.Remove(extAuthPath)
  1241. assert.NoError(t, err)
  1242. }
  1243. func TestPreLoginHook(t *testing.T) {
  1244. if runtime.GOOS == osWindows {
  1245. t.Skip("this test is not available on Windows")
  1246. }
  1247. u := getTestUser()
  1248. err := dataprovider.Close()
  1249. assert.NoError(t, err)
  1250. err = config.LoadConfig(configDir, "")
  1251. assert.NoError(t, err)
  1252. providerConf := config.GetProviderConf()
  1253. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(u, false), os.ModePerm)
  1254. assert.NoError(t, err)
  1255. providerConf.PreLoginHook = preLoginPath
  1256. err = dataprovider.Initialize(providerConf, configDir, true)
  1257. assert.NoError(t, err)
  1258. _, _, err = httpdtest.GetUserByUsername(defaultUsername, http.StatusNotFound)
  1259. assert.NoError(t, err)
  1260. client, err := getFTPClient(u, false, nil)
  1261. if assert.NoError(t, err) {
  1262. err = checkBasicFTP(client)
  1263. assert.NoError(t, err)
  1264. err := client.Quit()
  1265. assert.NoError(t, err)
  1266. }
  1267. user, _, err := httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
  1268. assert.NoError(t, err)
  1269. // test login with an existing user
  1270. client, err = getFTPClient(user, true, nil)
  1271. if assert.NoError(t, err) {
  1272. err = checkBasicFTP(client)
  1273. assert.NoError(t, err)
  1274. err := client.Quit()
  1275. assert.NoError(t, err)
  1276. }
  1277. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(user, true), os.ModePerm)
  1278. assert.NoError(t, err)
  1279. client, err = getFTPClient(u, false, nil)
  1280. if !assert.Error(t, err) {
  1281. err := client.Quit()
  1282. assert.NoError(t, err)
  1283. }
  1284. user.Status = 0
  1285. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(user, false), os.ModePerm)
  1286. assert.NoError(t, err)
  1287. client, err = getFTPClient(u, false, nil)
  1288. if !assert.Error(t, err, "pre-login script returned a disabled user, login must fail") {
  1289. err := client.Quit()
  1290. assert.NoError(t, err)
  1291. }
  1292. user.Status = 0
  1293. user.Filters.FTPSecurity = 1
  1294. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(user, false), os.ModePerm)
  1295. assert.NoError(t, err)
  1296. client, err = getFTPClient(u, true, nil)
  1297. if !assert.Error(t, err) {
  1298. err := client.Quit()
  1299. assert.NoError(t, err)
  1300. }
  1301. _, err = getFTPClient(user, false, nil)
  1302. if assert.Error(t, err) {
  1303. assert.Contains(t, err.Error(), "TLS is required")
  1304. }
  1305. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1306. assert.NoError(t, err)
  1307. err = os.RemoveAll(user.GetHomeDir())
  1308. assert.NoError(t, err)
  1309. err = dataprovider.Close()
  1310. assert.NoError(t, err)
  1311. err = config.LoadConfig(configDir, "")
  1312. assert.NoError(t, err)
  1313. providerConf = config.GetProviderConf()
  1314. err = dataprovider.Initialize(providerConf, configDir, true)
  1315. assert.NoError(t, err)
  1316. err = os.Remove(preLoginPath)
  1317. assert.NoError(t, err)
  1318. }
  1319. func TestPreLoginHookReturningAnonymousUser(t *testing.T) {
  1320. if runtime.GOOS == osWindows {
  1321. t.Skip("this test is not available on Windows")
  1322. }
  1323. u := getTestUser()
  1324. u.Filters.IsAnonymous = true
  1325. u.Filters.DeniedProtocols = []string{common.ProtocolSSH}
  1326. u.Password = ""
  1327. err := dataprovider.Close()
  1328. assert.NoError(t, err)
  1329. err = config.LoadConfig(configDir, "")
  1330. assert.NoError(t, err)
  1331. providerConf := config.GetProviderConf()
  1332. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(u, false), os.ModePerm)
  1333. assert.NoError(t, err)
  1334. providerConf.PreLoginHook = preLoginPath
  1335. err = dataprovider.Initialize(providerConf, configDir, true)
  1336. assert.NoError(t, err)
  1337. // the pre-login hook create the anonymous user
  1338. client, err := getFTPClient(u, false, nil)
  1339. if assert.NoError(t, err) {
  1340. err = checkBasicFTP(client)
  1341. assert.NoError(t, err)
  1342. testFilePath := filepath.Join(homeBasePath, testFileName)
  1343. testFileSize := int64(65535)
  1344. err = createTestFile(testFilePath, testFileSize)
  1345. assert.NoError(t, err)
  1346. err = client.MakeDir("tdiranonymous")
  1347. if assert.Error(t, err) {
  1348. assert.Contains(t, err.Error(), "permission")
  1349. }
  1350. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1351. if assert.Error(t, err) {
  1352. assert.Contains(t, err.Error(), "permission")
  1353. }
  1354. err = os.Rename(testFilePath, filepath.Join(u.GetHomeDir(), testFileName))
  1355. assert.NoError(t, err)
  1356. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1357. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  1358. assert.NoError(t, err)
  1359. err := client.Quit()
  1360. assert.NoError(t, err)
  1361. }
  1362. user, _, err := httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
  1363. assert.NoError(t, err)
  1364. assert.True(t, user.Filters.IsAnonymous)
  1365. assert.Equal(t, []string{dataprovider.PermListItems, dataprovider.PermDownload}, user.Permissions["/"])
  1366. assert.Equal(t, []string{common.ProtocolSSH, common.ProtocolHTTP}, user.Filters.DeniedProtocols)
  1367. assert.Equal(t, []string{dataprovider.SSHLoginMethodPublicKey, dataprovider.SSHLoginMethodPassword,
  1368. dataprovider.SSHLoginMethodKeyboardInteractive, dataprovider.SSHLoginMethodKeyAndPassword,
  1369. dataprovider.SSHLoginMethodKeyAndKeyboardInt, dataprovider.LoginMethodTLSCertificate,
  1370. dataprovider.LoginMethodTLSCertificateAndPwd}, user.Filters.DeniedLoginMethods)
  1371. // now the same with an existing user
  1372. client, err = getFTPClient(u, false, nil)
  1373. if assert.NoError(t, err) {
  1374. err = checkBasicFTP(client)
  1375. assert.NoError(t, err)
  1376. testFilePath := filepath.Join(homeBasePath, testFileName)
  1377. testFileSize := int64(65535)
  1378. err = createTestFile(testFilePath, testFileSize)
  1379. assert.NoError(t, err)
  1380. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1381. if assert.Error(t, err) {
  1382. assert.Contains(t, err.Error(), "permission")
  1383. }
  1384. err = os.Rename(testFilePath, filepath.Join(u.GetHomeDir(), testFileName))
  1385. assert.NoError(t, err)
  1386. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1387. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  1388. assert.NoError(t, err)
  1389. err := client.Quit()
  1390. assert.NoError(t, err)
  1391. }
  1392. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1393. assert.NoError(t, err)
  1394. err = os.RemoveAll(user.GetHomeDir())
  1395. assert.NoError(t, err)
  1396. err = dataprovider.Close()
  1397. assert.NoError(t, err)
  1398. err = config.LoadConfig(configDir, "")
  1399. assert.NoError(t, err)
  1400. providerConf = config.GetProviderConf()
  1401. err = dataprovider.Initialize(providerConf, configDir, true)
  1402. assert.NoError(t, err)
  1403. err = os.Remove(preLoginPath)
  1404. assert.NoError(t, err)
  1405. }
  1406. func TestPreDownloadHook(t *testing.T) {
  1407. if runtime.GOOS == osWindows {
  1408. t.Skip("this test is not available on Windows")
  1409. }
  1410. oldExecuteOn := common.Config.Actions.ExecuteOn
  1411. oldHook := common.Config.Actions.Hook
  1412. common.Config.Actions.ExecuteOn = []string{common.OperationPreDownload}
  1413. common.Config.Actions.Hook = preDownloadPath
  1414. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  1415. assert.NoError(t, err)
  1416. err = os.WriteFile(preDownloadPath, getExitCodeScriptContent(0), os.ModePerm)
  1417. assert.NoError(t, err)
  1418. testFilePath := filepath.Join(homeBasePath, testFileName)
  1419. testFileSize := int64(65535)
  1420. err = createTestFile(testFilePath, testFileSize)
  1421. assert.NoError(t, err)
  1422. client, err := getFTPClient(user, true, nil)
  1423. if assert.NoError(t, err) {
  1424. err = checkBasicFTP(client)
  1425. assert.NoError(t, err)
  1426. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1427. assert.NoError(t, err)
  1428. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1429. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  1430. assert.NoError(t, err)
  1431. err := client.Quit()
  1432. assert.NoError(t, err)
  1433. err = os.Remove(localDownloadPath)
  1434. assert.NoError(t, err)
  1435. }
  1436. // now return an error from the pre-download hook
  1437. err = os.WriteFile(preDownloadPath, getExitCodeScriptContent(1), os.ModePerm)
  1438. assert.NoError(t, err)
  1439. client, err = getFTPClient(user, true, nil)
  1440. if assert.NoError(t, err) {
  1441. err = checkBasicFTP(client)
  1442. assert.NoError(t, err)
  1443. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1444. assert.NoError(t, err)
  1445. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1446. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  1447. if assert.Error(t, err) {
  1448. assert.Contains(t, err.Error(), "permission denied")
  1449. }
  1450. err := client.Quit()
  1451. assert.NoError(t, err)
  1452. err = os.Remove(localDownloadPath)
  1453. assert.NoError(t, err)
  1454. }
  1455. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1456. assert.NoError(t, err)
  1457. err = os.RemoveAll(user.GetHomeDir())
  1458. assert.NoError(t, err)
  1459. err = os.Remove(testFilePath)
  1460. assert.NoError(t, err)
  1461. common.Config.Actions.ExecuteOn = oldExecuteOn
  1462. common.Config.Actions.Hook = oldHook
  1463. }
  1464. func TestPreUploadHook(t *testing.T) {
  1465. if runtime.GOOS == osWindows {
  1466. t.Skip("this test is not available on Windows")
  1467. }
  1468. oldExecuteOn := common.Config.Actions.ExecuteOn
  1469. oldHook := common.Config.Actions.Hook
  1470. common.Config.Actions.ExecuteOn = []string{common.OperationPreUpload}
  1471. common.Config.Actions.Hook = preUploadPath
  1472. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  1473. assert.NoError(t, err)
  1474. err = os.WriteFile(preUploadPath, getExitCodeScriptContent(0), os.ModePerm)
  1475. assert.NoError(t, err)
  1476. testFilePath := filepath.Join(homeBasePath, testFileName)
  1477. testFileSize := int64(65535)
  1478. err = createTestFile(testFilePath, testFileSize)
  1479. assert.NoError(t, err)
  1480. client, err := getFTPClient(user, true, nil)
  1481. if assert.NoError(t, err) {
  1482. err = checkBasicFTP(client)
  1483. assert.NoError(t, err)
  1484. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1485. assert.NoError(t, err)
  1486. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1487. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  1488. assert.NoError(t, err)
  1489. err := client.Quit()
  1490. assert.NoError(t, err)
  1491. err = os.Remove(localDownloadPath)
  1492. assert.NoError(t, err)
  1493. }
  1494. // now return an error from the pre-upload hook
  1495. err = os.WriteFile(preUploadPath, getExitCodeScriptContent(1), os.ModePerm)
  1496. assert.NoError(t, err)
  1497. client, err = getFTPClient(user, true, nil)
  1498. if assert.NoError(t, err) {
  1499. err = checkBasicFTP(client)
  1500. assert.NoError(t, err)
  1501. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1502. if assert.Error(t, err) {
  1503. assert.Contains(t, err.Error(), ftpserver.ErrFileNameNotAllowed.Error())
  1504. }
  1505. err = ftpUploadFile(testFilePath, testFileName+"1", testFileSize, client, 0)
  1506. if assert.Error(t, err) {
  1507. assert.Contains(t, err.Error(), ftpserver.ErrFileNameNotAllowed.Error())
  1508. }
  1509. err := client.Quit()
  1510. assert.NoError(t, err)
  1511. }
  1512. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1513. assert.NoError(t, err)
  1514. err = os.RemoveAll(user.GetHomeDir())
  1515. assert.NoError(t, err)
  1516. err = os.Remove(testFilePath)
  1517. assert.NoError(t, err)
  1518. common.Config.Actions.ExecuteOn = oldExecuteOn
  1519. common.Config.Actions.Hook = oldHook
  1520. }
  1521. func TestPostConnectHook(t *testing.T) {
  1522. if runtime.GOOS == osWindows {
  1523. t.Skip("this test is not available on Windows")
  1524. }
  1525. common.Config.PostConnectHook = postConnectPath
  1526. u := getTestUser()
  1527. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1528. assert.NoError(t, err)
  1529. err = os.WriteFile(postConnectPath, getExitCodeScriptContent(0), os.ModePerm)
  1530. assert.NoError(t, err)
  1531. client, err := getFTPClient(user, true, nil)
  1532. if assert.NoError(t, err) {
  1533. err = checkBasicFTP(client)
  1534. assert.NoError(t, err)
  1535. err := client.Quit()
  1536. assert.NoError(t, err)
  1537. }
  1538. err = os.WriteFile(postConnectPath, getExitCodeScriptContent(1), os.ModePerm)
  1539. assert.NoError(t, err)
  1540. client, err = getFTPClient(user, true, nil)
  1541. if !assert.Error(t, err) {
  1542. err := client.Quit()
  1543. assert.NoError(t, err)
  1544. }
  1545. common.Config.PostConnectHook = "http://127.0.0.1:8079/healthz"
  1546. client, err = getFTPClient(user, false, nil)
  1547. if assert.NoError(t, err) {
  1548. err = checkBasicFTP(client)
  1549. assert.NoError(t, err)
  1550. err := client.Quit()
  1551. assert.NoError(t, err)
  1552. }
  1553. common.Config.PostConnectHook = "http://127.0.0.1:8079/notfound"
  1554. client, err = getFTPClient(user, true, nil)
  1555. if !assert.Error(t, err) {
  1556. err := client.Quit()
  1557. assert.NoError(t, err)
  1558. }
  1559. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1560. assert.NoError(t, err)
  1561. err = os.RemoveAll(user.GetHomeDir())
  1562. assert.NoError(t, err)
  1563. common.Config.PostConnectHook = ""
  1564. }
  1565. //nolint:dupl
  1566. func TestMaxConnections(t *testing.T) {
  1567. oldValue := common.Config.MaxTotalConnections
  1568. common.Config.MaxTotalConnections = 1
  1569. assert.Eventually(t, func() bool {
  1570. return common.Connections.GetClientConnections() == 0
  1571. }, 1000*time.Millisecond, 50*time.Millisecond)
  1572. user := getTestUser()
  1573. err := dataprovider.AddUser(&user, "", "", "")
  1574. assert.NoError(t, err)
  1575. user.Password = ""
  1576. client, err := getFTPClient(user, true, nil)
  1577. if assert.NoError(t, err) {
  1578. err = checkBasicFTP(client)
  1579. assert.NoError(t, err)
  1580. _, err = getFTPClient(user, false, nil)
  1581. assert.Error(t, err)
  1582. err = client.Quit()
  1583. assert.NoError(t, err)
  1584. }
  1585. err = dataprovider.DeleteUser(user.Username, "", "", "")
  1586. assert.NoError(t, err)
  1587. err = os.RemoveAll(user.GetHomeDir())
  1588. assert.NoError(t, err)
  1589. common.Config.MaxTotalConnections = oldValue
  1590. }
  1591. //nolint:dupl
  1592. func TestMaxPerHostConnections(t *testing.T) {
  1593. oldValue := common.Config.MaxPerHostConnections
  1594. common.Config.MaxPerHostConnections = 1
  1595. assert.Eventually(t, func() bool {
  1596. return common.Connections.GetClientConnections() == 0
  1597. }, 1000*time.Millisecond, 50*time.Millisecond)
  1598. user := getTestUser()
  1599. err := dataprovider.AddUser(&user, "", "", "")
  1600. assert.NoError(t, err)
  1601. user.Password = ""
  1602. client, err := getFTPClient(user, true, nil)
  1603. if assert.NoError(t, err) {
  1604. err = checkBasicFTP(client)
  1605. assert.NoError(t, err)
  1606. _, err = getFTPClient(user, false, nil)
  1607. assert.Error(t, err)
  1608. err = client.Quit()
  1609. assert.NoError(t, err)
  1610. }
  1611. err = dataprovider.DeleteUser(user.Username, "", "", "")
  1612. assert.NoError(t, err)
  1613. err = os.RemoveAll(user.GetHomeDir())
  1614. assert.NoError(t, err)
  1615. common.Config.MaxPerHostConnections = oldValue
  1616. }
  1617. func TestRateLimiter(t *testing.T) {
  1618. oldConfig := config.GetCommonConfig()
  1619. cfg := config.GetCommonConfig()
  1620. cfg.DefenderConfig.Enabled = true
  1621. cfg.DefenderConfig.Threshold = 5
  1622. cfg.DefenderConfig.ScoreLimitExceeded = 3
  1623. cfg.RateLimitersConfig = []common.RateLimiterConfig{
  1624. {
  1625. Average: 1,
  1626. Period: 1000,
  1627. Burst: 1,
  1628. Type: 2,
  1629. Protocols: []string{common.ProtocolFTP},
  1630. GenerateDefenderEvents: true,
  1631. EntriesSoftLimit: 100,
  1632. EntriesHardLimit: 150,
  1633. },
  1634. }
  1635. err := common.Initialize(cfg, 0)
  1636. assert.NoError(t, err)
  1637. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  1638. assert.NoError(t, err)
  1639. client, err := getFTPClient(user, false, nil)
  1640. if assert.NoError(t, err) {
  1641. err = checkBasicFTP(client)
  1642. assert.NoError(t, err)
  1643. err = client.Quit()
  1644. assert.NoError(t, err)
  1645. }
  1646. _, err = getFTPClient(user, true, nil)
  1647. if assert.Error(t, err) {
  1648. assert.Contains(t, err.Error(), "rate limit exceed")
  1649. }
  1650. _, err = getFTPClient(user, false, nil)
  1651. if assert.Error(t, err) {
  1652. assert.Contains(t, err.Error(), "rate limit exceed")
  1653. }
  1654. _, err = getFTPClient(user, true, nil)
  1655. if assert.Error(t, err) {
  1656. assert.Contains(t, err.Error(), "banned client IP")
  1657. }
  1658. err = dataprovider.DeleteUser(user.Username, "", "", "")
  1659. assert.NoError(t, err)
  1660. err = os.RemoveAll(user.GetHomeDir())
  1661. assert.NoError(t, err)
  1662. err = common.Initialize(oldConfig, 0)
  1663. assert.NoError(t, err)
  1664. }
  1665. func TestDefender(t *testing.T) {
  1666. oldConfig := config.GetCommonConfig()
  1667. cfg := config.GetCommonConfig()
  1668. cfg.DefenderConfig.Enabled = true
  1669. cfg.DefenderConfig.Threshold = 4
  1670. cfg.DefenderConfig.ScoreLimitExceeded = 2
  1671. cfg.DefenderConfig.ScoreNoAuth = 1
  1672. cfg.DefenderConfig.ScoreValid = 1
  1673. err := common.Initialize(cfg, 0)
  1674. assert.NoError(t, err)
  1675. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  1676. assert.NoError(t, err)
  1677. client, err := getFTPClient(user, false, nil)
  1678. if assert.NoError(t, err) {
  1679. err = checkBasicFTP(client)
  1680. assert.NoError(t, err)
  1681. err = client.Quit()
  1682. assert.NoError(t, err)
  1683. }
  1684. // just dial without login
  1685. ftpOptions := []ftp.DialOption{ftp.DialWithTimeout(5 * time.Second)}
  1686. client, err = ftp.Dial(ftpServerAddr, ftpOptions...)
  1687. assert.NoError(t, err)
  1688. err = client.Quit()
  1689. assert.NoError(t, err)
  1690. hosts, _, err := httpdtest.GetDefenderHosts(http.StatusOK)
  1691. assert.NoError(t, err)
  1692. if assert.Len(t, hosts, 1) {
  1693. host := hosts[0]
  1694. assert.Empty(t, host.GetBanTime())
  1695. assert.Equal(t, 1, host.Score)
  1696. }
  1697. user.Password = "wrong_pwd"
  1698. _, err = getFTPClient(user, false, nil)
  1699. assert.Error(t, err)
  1700. hosts, _, err = httpdtest.GetDefenderHosts(http.StatusOK)
  1701. assert.NoError(t, err)
  1702. if assert.Len(t, hosts, 1) {
  1703. host := hosts[0]
  1704. assert.Empty(t, host.GetBanTime())
  1705. assert.Equal(t, 2, host.Score)
  1706. }
  1707. for i := 0; i < 2; i++ {
  1708. _, err = getFTPClient(user, false, nil)
  1709. assert.Error(t, err)
  1710. }
  1711. user.Password = defaultPassword
  1712. _, err = getFTPClient(user, false, nil)
  1713. if assert.Error(t, err) {
  1714. assert.Contains(t, err.Error(), "banned client IP")
  1715. }
  1716. err = dataprovider.DeleteUser(user.Username, "", "", "")
  1717. assert.NoError(t, err)
  1718. err = os.RemoveAll(user.GetHomeDir())
  1719. assert.NoError(t, err)
  1720. err = common.Initialize(oldConfig, 0)
  1721. assert.NoError(t, err)
  1722. }
  1723. func TestMaxSessions(t *testing.T) {
  1724. u := getTestUser()
  1725. u.MaxSessions = 1
  1726. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1727. assert.NoError(t, err)
  1728. client, err := getFTPClient(user, true, nil)
  1729. if assert.NoError(t, err) {
  1730. err = checkBasicFTP(client)
  1731. assert.NoError(t, err)
  1732. _, err = getFTPClient(user, false, nil)
  1733. assert.Error(t, err)
  1734. err = client.Quit()
  1735. assert.NoError(t, err)
  1736. }
  1737. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1738. assert.NoError(t, err)
  1739. err = os.RemoveAll(user.GetHomeDir())
  1740. assert.NoError(t, err)
  1741. }
  1742. func TestZeroBytesTransfers(t *testing.T) {
  1743. u := getTestUser()
  1744. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1745. assert.NoError(t, err)
  1746. for _, useTLS := range []bool{true, false} {
  1747. client, err := getFTPClient(user, useTLS, nil)
  1748. if assert.NoError(t, err) {
  1749. testFileName := "testfilename"
  1750. err = checkBasicFTP(client)
  1751. assert.NoError(t, err)
  1752. localDownloadPath := filepath.Join(homeBasePath, "empty_download")
  1753. err = os.WriteFile(localDownloadPath, []byte(""), os.ModePerm)
  1754. assert.NoError(t, err)
  1755. err = ftpUploadFile(localDownloadPath, testFileName, 0, client, 0)
  1756. assert.NoError(t, err)
  1757. size, err := client.FileSize(testFileName)
  1758. assert.NoError(t, err)
  1759. assert.Equal(t, int64(0), size)
  1760. err = os.Remove(localDownloadPath)
  1761. assert.NoError(t, err)
  1762. assert.NoFileExists(t, localDownloadPath)
  1763. err = ftpDownloadFile(testFileName, localDownloadPath, 0, client, 0)
  1764. assert.NoError(t, err)
  1765. assert.FileExists(t, localDownloadPath)
  1766. err = client.Quit()
  1767. assert.NoError(t, err)
  1768. err = os.Remove(localDownloadPath)
  1769. assert.NoError(t, err)
  1770. }
  1771. }
  1772. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1773. assert.NoError(t, err)
  1774. err = os.RemoveAll(user.GetHomeDir())
  1775. assert.NoError(t, err)
  1776. }
  1777. func TestDownloadErrors(t *testing.T) {
  1778. u := getTestUser()
  1779. u.QuotaFiles = 1
  1780. subDir1 := "sub1"
  1781. subDir2 := "sub2"
  1782. u.Permissions[path.Join("/", subDir1)] = []string{dataprovider.PermListItems}
  1783. u.Permissions[path.Join("/", subDir2)] = []string{dataprovider.PermListItems, dataprovider.PermUpload,
  1784. dataprovider.PermDelete, dataprovider.PermDownload}
  1785. u.Filters.FilePatterns = []sdk.PatternsFilter{
  1786. {
  1787. Path: "/sub2",
  1788. AllowedPatterns: []string{},
  1789. DeniedPatterns: []string{"*.jpg", "*.zip"},
  1790. },
  1791. }
  1792. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1793. assert.NoError(t, err)
  1794. client, err := getFTPClient(user, true, nil)
  1795. if assert.NoError(t, err) {
  1796. testFilePath1 := filepath.Join(user.HomeDir, subDir1, "file.zip")
  1797. testFilePath2 := filepath.Join(user.HomeDir, subDir2, "file.zip")
  1798. testFilePath3 := filepath.Join(user.HomeDir, subDir2, "file.jpg")
  1799. err = os.MkdirAll(filepath.Dir(testFilePath1), os.ModePerm)
  1800. assert.NoError(t, err)
  1801. err = os.MkdirAll(filepath.Dir(testFilePath2), os.ModePerm)
  1802. assert.NoError(t, err)
  1803. err = os.WriteFile(testFilePath1, []byte("file1"), os.ModePerm)
  1804. assert.NoError(t, err)
  1805. err = os.WriteFile(testFilePath2, []byte("file2"), os.ModePerm)
  1806. assert.NoError(t, err)
  1807. err = os.WriteFile(testFilePath3, []byte("file3"), os.ModePerm)
  1808. assert.NoError(t, err)
  1809. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1810. err = ftpDownloadFile(path.Join("/", subDir1, "file.zip"), localDownloadPath, 5, client, 0)
  1811. assert.Error(t, err)
  1812. err = ftpDownloadFile(path.Join("/", subDir2, "file.zip"), localDownloadPath, 5, client, 0)
  1813. assert.Error(t, err)
  1814. err = ftpDownloadFile(path.Join("/", subDir2, "file.jpg"), localDownloadPath, 5, client, 0)
  1815. assert.Error(t, err)
  1816. err = ftpDownloadFile("/missing.zip", localDownloadPath, 5, client, 0)
  1817. assert.Error(t, err)
  1818. err = client.Quit()
  1819. assert.NoError(t, err)
  1820. err = os.Remove(localDownloadPath)
  1821. assert.NoError(t, err)
  1822. }
  1823. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1824. assert.NoError(t, err)
  1825. err = os.RemoveAll(user.GetHomeDir())
  1826. assert.NoError(t, err)
  1827. }
  1828. func TestUploadErrors(t *testing.T) {
  1829. u := getTestUser()
  1830. u.QuotaSize = 65535
  1831. subDir1 := "sub1"
  1832. subDir2 := "sub2"
  1833. u.Permissions[path.Join("/", subDir1)] = []string{dataprovider.PermListItems}
  1834. u.Permissions[path.Join("/", subDir2)] = []string{dataprovider.PermListItems, dataprovider.PermUpload,
  1835. dataprovider.PermDelete}
  1836. u.Filters.FilePatterns = []sdk.PatternsFilter{
  1837. {
  1838. Path: "/sub2",
  1839. AllowedPatterns: []string{},
  1840. DeniedPatterns: []string{"*.zip"},
  1841. },
  1842. }
  1843. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1844. assert.NoError(t, err)
  1845. client, err := getFTPClient(user, true, nil)
  1846. if assert.NoError(t, err) {
  1847. testFilePath := filepath.Join(homeBasePath, testFileName)
  1848. testFileSize := user.QuotaSize
  1849. err = createTestFile(testFilePath, testFileSize)
  1850. assert.NoError(t, err)
  1851. err = client.MakeDir(subDir1)
  1852. assert.NoError(t, err)
  1853. err = client.MakeDir(subDir2)
  1854. assert.NoError(t, err)
  1855. err = client.ChangeDir(subDir1)
  1856. assert.NoError(t, err)
  1857. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1858. assert.Error(t, err)
  1859. err = client.ChangeDirToParent()
  1860. assert.NoError(t, err)
  1861. err = client.ChangeDir(subDir2)
  1862. assert.NoError(t, err)
  1863. err = ftpUploadFile(testFilePath, testFileName+".zip", testFileSize, client, 0)
  1864. assert.Error(t, err)
  1865. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1866. assert.NoError(t, err)
  1867. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1868. assert.Error(t, err)
  1869. err = client.ChangeDir("/")
  1870. assert.NoError(t, err)
  1871. err = ftpUploadFile(testFilePath, subDir1, testFileSize, client, 0)
  1872. assert.Error(t, err)
  1873. // overquota
  1874. err = ftpUploadFile(testFilePath, testFileName+"1", testFileSize, client, 0)
  1875. assert.Error(t, err)
  1876. err = client.Delete(path.Join("/", subDir2, testFileName))
  1877. assert.NoError(t, err)
  1878. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1879. assert.NoError(t, err)
  1880. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1881. assert.Error(t, err)
  1882. err = client.Quit()
  1883. assert.NoError(t, err)
  1884. err = os.Remove(testFilePath)
  1885. assert.NoError(t, err)
  1886. }
  1887. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  1888. assert.NoError(t, err)
  1889. err = os.RemoveAll(user.GetHomeDir())
  1890. assert.NoError(t, err)
  1891. }
  1892. func TestSFTPBuffered(t *testing.T) {
  1893. u := getTestUser()
  1894. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1895. assert.NoError(t, err)
  1896. u = getTestSFTPUser()
  1897. u.QuotaFiles = 100
  1898. u.FsConfig.SFTPConfig.BufferSize = 2
  1899. u.HomeDir = filepath.Join(os.TempDir(), u.Username)
  1900. sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1901. assert.NoError(t, err)
  1902. client, err := getFTPClient(sftpUser, true, nil)
  1903. if assert.NoError(t, err) {
  1904. testFilePath := filepath.Join(homeBasePath, testFileName)
  1905. testFileSize := int64(65535)
  1906. expectedQuotaSize := testFileSize
  1907. expectedQuotaFiles := 1
  1908. err = createTestFile(testFilePath, testFileSize)
  1909. assert.NoError(t, err)
  1910. err = checkBasicFTP(client)
  1911. assert.NoError(t, err)
  1912. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1913. assert.NoError(t, err)
  1914. // overwrite an existing file
  1915. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  1916. assert.NoError(t, err)
  1917. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  1918. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  1919. assert.NoError(t, err)
  1920. user, _, err := httpdtest.GetUserByUsername(sftpUser.Username, http.StatusOK)
  1921. assert.NoError(t, err)
  1922. assert.Equal(t, expectedQuotaFiles, user.UsedQuotaFiles)
  1923. assert.Equal(t, expectedQuotaSize, user.UsedQuotaSize)
  1924. data := []byte("test data")
  1925. err = os.WriteFile(testFilePath, data, os.ModePerm)
  1926. assert.NoError(t, err)
  1927. err = ftpUploadFile(testFilePath, testFileName, int64(len(data)), client, 0)
  1928. assert.NoError(t, err)
  1929. err = ftpUploadFile(testFilePath, testFileName, int64(len(data)+5), client, 5)
  1930. if assert.Error(t, err) {
  1931. assert.Contains(t, err.Error(), "operation unsupported")
  1932. }
  1933. err = ftpDownloadFile(testFileName, localDownloadPath, int64(4), client, 5)
  1934. assert.NoError(t, err)
  1935. readed, err := os.ReadFile(localDownloadPath)
  1936. assert.NoError(t, err)
  1937. assert.Equal(t, []byte("data"), readed)
  1938. // try to append to a file, it should fail
  1939. // now append to a file
  1940. srcFile, err := os.Open(testFilePath)
  1941. if assert.NoError(t, err) {
  1942. err = client.Append(testFileName, srcFile)
  1943. if assert.Error(t, err) {
  1944. assert.Contains(t, err.Error(), "operation unsupported")
  1945. }
  1946. err = srcFile.Close()
  1947. assert.NoError(t, err)
  1948. size, err := client.FileSize(testFileName)
  1949. assert.NoError(t, err)
  1950. assert.Equal(t, int64(len(data)), size)
  1951. err = ftpDownloadFile(testFileName, localDownloadPath, int64(len(data)), client, 0)
  1952. assert.NoError(t, err)
  1953. }
  1954. err = os.Remove(testFilePath)
  1955. assert.NoError(t, err)
  1956. err = os.Remove(localDownloadPath)
  1957. assert.NoError(t, err)
  1958. err = client.Quit()
  1959. assert.NoError(t, err)
  1960. }
  1961. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  1962. assert.NoError(t, err)
  1963. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  1964. assert.NoError(t, err)
  1965. err = os.RemoveAll(localUser.GetHomeDir())
  1966. assert.NoError(t, err)
  1967. err = os.RemoveAll(sftpUser.GetHomeDir())
  1968. assert.NoError(t, err)
  1969. }
  1970. func TestResume(t *testing.T) {
  1971. u := getTestUser()
  1972. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1973. assert.NoError(t, err)
  1974. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  1975. assert.NoError(t, err)
  1976. u = getTestUser()
  1977. u.FsConfig.OSConfig = sdk.OSFsConfig{
  1978. ReadBufferSize: 1,
  1979. WriteBufferSize: 1,
  1980. }
  1981. u.Username += "_buf"
  1982. u.HomeDir += "_buf"
  1983. bufferedUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  1984. assert.NoError(t, err)
  1985. for _, user := range []dataprovider.User{localUser, sftpUser, bufferedUser} {
  1986. client, err := getFTPClient(user, true, nil)
  1987. if assert.NoError(t, err) {
  1988. testFilePath := filepath.Join(homeBasePath, testFileName)
  1989. data := []byte("test data")
  1990. err = os.WriteFile(testFilePath, data, os.ModePerm)
  1991. assert.NoError(t, err)
  1992. err = ftpUploadFile(testFilePath, testFileName, int64(len(data)), client, 0)
  1993. assert.NoError(t, err)
  1994. err = ftpUploadFile(testFilePath, testFileName, int64(len(data)+5), client, 5)
  1995. assert.NoError(t, err)
  1996. readed, err := os.ReadFile(filepath.Join(user.GetHomeDir(), testFileName))
  1997. assert.NoError(t, err)
  1998. assert.Equal(t, "test test data", string(readed))
  1999. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  2000. err = ftpDownloadFile(testFileName, localDownloadPath, int64(len(data)), client, 5)
  2001. assert.NoError(t, err)
  2002. readed, err = os.ReadFile(localDownloadPath)
  2003. assert.NoError(t, err)
  2004. assert.Equal(t, data, readed)
  2005. err = client.Delete(testFileName)
  2006. assert.NoError(t, err)
  2007. err = ftpUploadFile(testFilePath, testFileName, int64(len(data)), client, 0)
  2008. assert.NoError(t, err)
  2009. // now append to a file
  2010. srcFile, err := os.Open(testFilePath)
  2011. if assert.NoError(t, err) {
  2012. err = client.Append(testFileName, srcFile)
  2013. assert.NoError(t, err)
  2014. err = srcFile.Close()
  2015. assert.NoError(t, err)
  2016. size, err := client.FileSize(testFileName)
  2017. assert.NoError(t, err)
  2018. assert.Equal(t, int64(2*len(data)), size)
  2019. err = ftpDownloadFile(testFileName, localDownloadPath, int64(2*len(data)), client, 0)
  2020. assert.NoError(t, err)
  2021. readed, err = os.ReadFile(localDownloadPath)
  2022. assert.NoError(t, err)
  2023. expected := append(data, data...)
  2024. assert.Equal(t, expected, readed)
  2025. }
  2026. // append to a new file
  2027. srcFile, err = os.Open(testFilePath)
  2028. if assert.NoError(t, err) {
  2029. newFileName := testFileName + "_new"
  2030. err = client.Append(newFileName, srcFile)
  2031. assert.NoError(t, err)
  2032. err = srcFile.Close()
  2033. assert.NoError(t, err)
  2034. size, err := client.FileSize(newFileName)
  2035. assert.NoError(t, err)
  2036. assert.Equal(t, int64(len(data)), size)
  2037. err = ftpDownloadFile(newFileName, localDownloadPath, int64(len(data)), client, 0)
  2038. assert.NoError(t, err)
  2039. readed, err = os.ReadFile(localDownloadPath)
  2040. assert.NoError(t, err)
  2041. assert.Equal(t, data, readed)
  2042. }
  2043. err = client.Quit()
  2044. assert.NoError(t, err)
  2045. err = os.Remove(testFilePath)
  2046. assert.NoError(t, err)
  2047. err = os.Remove(localDownloadPath)
  2048. assert.NoError(t, err)
  2049. if user.Username == defaultUsername {
  2050. err = os.RemoveAll(user.GetHomeDir())
  2051. assert.NoError(t, err)
  2052. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2053. assert.NoError(t, err)
  2054. user.Password = defaultPassword
  2055. user.ID = 0
  2056. user.CreatedAt = 0
  2057. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2058. assert.NoError(t, err, string(resp))
  2059. }
  2060. }
  2061. }
  2062. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2063. assert.NoError(t, err)
  2064. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2065. assert.NoError(t, err)
  2066. err = os.RemoveAll(localUser.GetHomeDir())
  2067. assert.NoError(t, err)
  2068. _, err = httpdtest.RemoveUser(bufferedUser, http.StatusOK)
  2069. assert.NoError(t, err)
  2070. err = os.RemoveAll(bufferedUser.GetHomeDir())
  2071. assert.NoError(t, err)
  2072. }
  2073. //nolint:dupl
  2074. func TestDeniedLoginMethod(t *testing.T) {
  2075. u := getTestUser()
  2076. u.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
  2077. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2078. assert.NoError(t, err)
  2079. _, err = getFTPClient(user, false, nil)
  2080. assert.Error(t, err)
  2081. user.Filters.DeniedLoginMethods = []string{dataprovider.SSHLoginMethodPublicKey, dataprovider.SSHLoginMethodKeyAndPassword}
  2082. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2083. assert.NoError(t, err)
  2084. client, err := getFTPClient(user, true, nil)
  2085. if assert.NoError(t, err) {
  2086. assert.NoError(t, checkBasicFTP(client))
  2087. err = client.Quit()
  2088. assert.NoError(t, err)
  2089. }
  2090. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2091. assert.NoError(t, err)
  2092. err = os.RemoveAll(user.GetHomeDir())
  2093. assert.NoError(t, err)
  2094. }
  2095. //nolint:dupl
  2096. func TestDeniedProtocols(t *testing.T) {
  2097. u := getTestUser()
  2098. u.Filters.DeniedProtocols = []string{common.ProtocolFTP}
  2099. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2100. assert.NoError(t, err)
  2101. _, err = getFTPClient(user, false, nil)
  2102. assert.Error(t, err)
  2103. user.Filters.DeniedProtocols = []string{common.ProtocolSSH, common.ProtocolWebDAV}
  2104. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2105. assert.NoError(t, err)
  2106. client, err := getFTPClient(user, true, nil)
  2107. if assert.NoError(t, err) {
  2108. assert.NoError(t, checkBasicFTP(client))
  2109. err = client.Quit()
  2110. assert.NoError(t, err)
  2111. }
  2112. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2113. assert.NoError(t, err)
  2114. err = os.RemoveAll(user.GetHomeDir())
  2115. assert.NoError(t, err)
  2116. }
  2117. func TestQuotaLimits(t *testing.T) {
  2118. u := getTestUser()
  2119. u.QuotaFiles = 1
  2120. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2121. assert.NoError(t, err)
  2122. u = getTestSFTPUser()
  2123. u.QuotaFiles = 1
  2124. sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2125. assert.NoError(t, err)
  2126. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2127. testFileSize := int64(65535)
  2128. testFilePath := filepath.Join(homeBasePath, testFileName)
  2129. err = createTestFile(testFilePath, testFileSize)
  2130. assert.NoError(t, err)
  2131. testFileSize1 := int64(131072)
  2132. testFileName1 := "test_file1.dat"
  2133. testFilePath1 := filepath.Join(homeBasePath, testFileName1)
  2134. err = createTestFile(testFilePath1, testFileSize1)
  2135. assert.NoError(t, err)
  2136. testFileSize2 := int64(32768)
  2137. testFileName2 := "test_file2.dat"
  2138. testFilePath2 := filepath.Join(homeBasePath, testFileName2)
  2139. err = createTestFile(testFilePath2, testFileSize2)
  2140. assert.NoError(t, err)
  2141. // test quota files
  2142. client, err := getFTPClient(user, false, nil)
  2143. if assert.NoError(t, err) {
  2144. err = ftpUploadFile(testFilePath, testFileName+".quota", testFileSize, client, 0) //nolint:goconst
  2145. assert.NoError(t, err)
  2146. err = ftpUploadFile(testFilePath, testFileName+".quota1", testFileSize, client, 0)
  2147. assert.Error(t, err)
  2148. err = client.Rename(testFileName+".quota", testFileName)
  2149. assert.NoError(t, err)
  2150. err = client.Quit()
  2151. assert.NoError(t, err)
  2152. }
  2153. // test quota size
  2154. user.QuotaSize = testFileSize - 1
  2155. user.QuotaFiles = 0
  2156. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2157. assert.NoError(t, err)
  2158. client, err = getFTPClient(user, true, nil)
  2159. if assert.NoError(t, err) {
  2160. err = ftpUploadFile(testFilePath, testFileName+".quota", testFileSize, client, 0)
  2161. assert.Error(t, err)
  2162. err = client.Rename(testFileName, testFileName+".quota")
  2163. assert.NoError(t, err)
  2164. err = client.Quit()
  2165. assert.NoError(t, err)
  2166. }
  2167. // now test quota limits while uploading the current file, we have 1 bytes remaining
  2168. user.QuotaSize = testFileSize + 1
  2169. user.QuotaFiles = 0
  2170. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2171. assert.NoError(t, err)
  2172. client, err = getFTPClient(user, false, nil)
  2173. if assert.NoError(t, err) {
  2174. err = ftpUploadFile(testFilePath1, testFileName1, testFileSize1, client, 0)
  2175. assert.Error(t, err)
  2176. _, err = client.FileSize(testFileName1)
  2177. assert.Error(t, err)
  2178. err = client.Rename(testFileName+".quota", testFileName)
  2179. assert.NoError(t, err)
  2180. // overwriting an existing file will work if the resulting size is lesser or equal than the current one
  2181. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2182. assert.NoError(t, err)
  2183. err = ftpUploadFile(testFilePath2, testFileName, testFileSize2, client, 0)
  2184. assert.NoError(t, err)
  2185. err = ftpUploadFile(testFilePath1, testFileName, testFileSize1, client, 0)
  2186. assert.Error(t, err)
  2187. err = ftpUploadFile(testFilePath1, testFileName, testFileSize1, client, 10)
  2188. assert.Error(t, err)
  2189. err = ftpUploadFile(testFilePath2, testFileName, testFileSize2, client, 0)
  2190. assert.NoError(t, err)
  2191. err = client.Quit()
  2192. assert.NoError(t, err)
  2193. }
  2194. err = os.Remove(testFilePath)
  2195. assert.NoError(t, err)
  2196. err = os.Remove(testFilePath1)
  2197. assert.NoError(t, err)
  2198. err = os.Remove(testFilePath2)
  2199. assert.NoError(t, err)
  2200. if user.Username == defaultUsername {
  2201. err = os.RemoveAll(user.GetHomeDir())
  2202. assert.NoError(t, err)
  2203. user.QuotaFiles = 0
  2204. user.QuotaSize = 0
  2205. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2206. assert.NoError(t, err)
  2207. user.Password = defaultPassword
  2208. user.QuotaSize = 0
  2209. user.ID = 0
  2210. user.CreatedAt = 0
  2211. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2212. assert.NoError(t, err, string(resp))
  2213. }
  2214. }
  2215. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2216. assert.NoError(t, err)
  2217. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2218. assert.NoError(t, err)
  2219. err = os.RemoveAll(localUser.GetHomeDir())
  2220. assert.NoError(t, err)
  2221. }
  2222. func TestUploadMaxSize(t *testing.T) {
  2223. testFileSize := int64(65535)
  2224. u := getTestUser()
  2225. u.Filters.MaxUploadFileSize = testFileSize + 1
  2226. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2227. assert.NoError(t, err)
  2228. u = getTestSFTPUser()
  2229. u.Filters.MaxUploadFileSize = testFileSize + 1
  2230. sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2231. assert.NoError(t, err)
  2232. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2233. testFilePath := filepath.Join(homeBasePath, testFileName)
  2234. err = createTestFile(testFilePath, testFileSize)
  2235. assert.NoError(t, err)
  2236. testFileSize1 := int64(131072)
  2237. testFileName1 := "test_file1.dat"
  2238. testFilePath1 := filepath.Join(homeBasePath, testFileName1)
  2239. err = createTestFile(testFilePath1, testFileSize1)
  2240. assert.NoError(t, err)
  2241. client, err := getFTPClient(user, false, nil)
  2242. if assert.NoError(t, err) {
  2243. err = ftpUploadFile(testFilePath1, testFileName1, testFileSize1, client, 0)
  2244. assert.Error(t, err)
  2245. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2246. assert.NoError(t, err)
  2247. // now test overwrite an existing file with a size bigger than the allowed one
  2248. err = createTestFile(filepath.Join(user.GetHomeDir(), testFileName1), testFileSize1)
  2249. assert.NoError(t, err)
  2250. err = ftpUploadFile(testFilePath1, testFileName1, testFileSize1, client, 0)
  2251. assert.Error(t, err)
  2252. err = client.Quit()
  2253. assert.NoError(t, err)
  2254. }
  2255. err = os.Remove(testFilePath)
  2256. assert.NoError(t, err)
  2257. err = os.Remove(testFilePath1)
  2258. assert.NoError(t, err)
  2259. if user.Username == defaultUsername {
  2260. err = os.RemoveAll(user.GetHomeDir())
  2261. assert.NoError(t, err)
  2262. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2263. assert.NoError(t, err)
  2264. user.Password = defaultPassword
  2265. user.Filters.MaxUploadFileSize = 65536000
  2266. user.ID = 0
  2267. user.CreatedAt = 0
  2268. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2269. assert.NoError(t, err, string(resp))
  2270. }
  2271. }
  2272. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2273. assert.NoError(t, err)
  2274. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2275. assert.NoError(t, err)
  2276. err = os.RemoveAll(localUser.GetHomeDir())
  2277. assert.NoError(t, err)
  2278. }
  2279. func TestLoginWithIPilters(t *testing.T) {
  2280. u := getTestUser()
  2281. u.Filters.DeniedIP = []string{"192.167.0.0/24", "172.18.0.0/16"}
  2282. u.Filters.AllowedIP = []string{"172.19.0.0/16"}
  2283. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2284. assert.NoError(t, err)
  2285. client, err := getFTPClient(user, true, nil)
  2286. if !assert.Error(t, err) {
  2287. err = client.Quit()
  2288. assert.NoError(t, err)
  2289. }
  2290. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2291. assert.NoError(t, err)
  2292. err = os.RemoveAll(user.GetHomeDir())
  2293. assert.NoError(t, err)
  2294. }
  2295. func TestLoginWithDatabaseCredentials(t *testing.T) {
  2296. u := getTestUser()
  2297. u.FsConfig.Provider = sdk.GCSFilesystemProvider
  2298. u.FsConfig.GCSConfig.Bucket = "test"
  2299. u.FsConfig.GCSConfig.Credentials = kms.NewPlainSecret(`{ "type": "service_account" }`)
  2300. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2301. assert.NoError(t, err)
  2302. assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.GCSConfig.Credentials.GetStatus())
  2303. assert.NotEmpty(t, user.FsConfig.GCSConfig.Credentials.GetPayload())
  2304. assert.Empty(t, user.FsConfig.GCSConfig.Credentials.GetAdditionalData())
  2305. assert.Empty(t, user.FsConfig.GCSConfig.Credentials.GetKey())
  2306. client, err := getFTPClient(user, false, nil)
  2307. if assert.NoError(t, err) {
  2308. err = client.Quit()
  2309. assert.NoError(t, err)
  2310. }
  2311. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2312. assert.NoError(t, err)
  2313. err = os.RemoveAll(user.GetHomeDir())
  2314. assert.NoError(t, err)
  2315. }
  2316. func TestLoginInvalidFs(t *testing.T) {
  2317. u := getTestUser()
  2318. u.FsConfig.Provider = sdk.GCSFilesystemProvider
  2319. u.FsConfig.GCSConfig.Bucket = "test"
  2320. u.FsConfig.GCSConfig.Credentials = kms.NewPlainSecret("invalid JSON for credentials")
  2321. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2322. assert.NoError(t, err)
  2323. client, err := getFTPClient(user, false, nil)
  2324. if !assert.Error(t, err) {
  2325. err = client.Quit()
  2326. assert.NoError(t, err)
  2327. }
  2328. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2329. assert.NoError(t, err)
  2330. err = os.RemoveAll(user.GetHomeDir())
  2331. assert.NoError(t, err)
  2332. }
  2333. func TestClientClose(t *testing.T) {
  2334. u := getTestUser()
  2335. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2336. assert.NoError(t, err)
  2337. client, err := getFTPClient(user, true, nil)
  2338. if assert.NoError(t, err) {
  2339. err = checkBasicFTP(client)
  2340. assert.NoError(t, err)
  2341. stats := common.Connections.GetStats("")
  2342. if assert.Len(t, stats, 1) {
  2343. common.Connections.Close(stats[0].ConnectionID, "")
  2344. assert.Eventually(t, func() bool { return len(common.Connections.GetStats("")) == 0 },
  2345. 1*time.Second, 50*time.Millisecond)
  2346. }
  2347. }
  2348. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2349. assert.NoError(t, err)
  2350. err = os.RemoveAll(user.GetHomeDir())
  2351. assert.NoError(t, err)
  2352. }
  2353. func TestRename(t *testing.T) {
  2354. u := getTestUser()
  2355. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2356. assert.NoError(t, err)
  2357. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  2358. assert.NoError(t, err)
  2359. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2360. testDir := "adir"
  2361. testFilePath := filepath.Join(homeBasePath, testFileName)
  2362. testFileSize := int64(65535)
  2363. err = createTestFile(testFilePath, testFileSize)
  2364. assert.NoError(t, err)
  2365. client, err := getFTPClient(user, false, nil)
  2366. if assert.NoError(t, err) {
  2367. err = checkBasicFTP(client)
  2368. assert.NoError(t, err)
  2369. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2370. assert.NoError(t, err)
  2371. err = client.MakeDir(testDir)
  2372. assert.NoError(t, err)
  2373. err = client.Rename(testFileName, path.Join("missing", testFileName))
  2374. assert.Error(t, err)
  2375. err = client.Rename(testFileName, path.Join(testDir, testFileName))
  2376. assert.NoError(t, err)
  2377. size, err := client.FileSize(path.Join(testDir, testFileName))
  2378. assert.NoError(t, err)
  2379. assert.Equal(t, testFileSize, size)
  2380. if runtime.GOOS != osWindows {
  2381. otherDir := "dir"
  2382. err = client.MakeDir(otherDir)
  2383. assert.NoError(t, err)
  2384. err = client.MakeDir(path.Join(otherDir, testDir))
  2385. assert.NoError(t, err)
  2386. code, response, err := client.SendCommand("SITE CHMOD 0001 %v", otherDir)
  2387. assert.NoError(t, err)
  2388. assert.Equal(t, ftp.StatusCommandOK, code)
  2389. assert.Equal(t, "SITE CHMOD command successful", response)
  2390. err = client.Rename(testDir, path.Join(otherDir, testDir))
  2391. assert.Error(t, err)
  2392. code, response, err = client.SendCommand("SITE CHMOD 755 %v", otherDir)
  2393. assert.NoError(t, err)
  2394. assert.Equal(t, ftp.StatusCommandOK, code)
  2395. assert.Equal(t, "SITE CHMOD command successful", response)
  2396. }
  2397. err = client.Quit()
  2398. assert.NoError(t, err)
  2399. }
  2400. user.Permissions[path.Join("/", testDir)] = []string{dataprovider.PermListItems}
  2401. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2402. assert.NoError(t, err)
  2403. client, err = getFTPClient(user, false, nil)
  2404. if assert.NoError(t, err) {
  2405. err = client.Rename(path.Join(testDir, testFileName), testFileName)
  2406. assert.Error(t, err)
  2407. err := client.Quit()
  2408. assert.NoError(t, err)
  2409. }
  2410. err = os.Remove(testFilePath)
  2411. assert.NoError(t, err)
  2412. if user.Username == defaultUsername {
  2413. err = os.RemoveAll(user.GetHomeDir())
  2414. assert.NoError(t, err)
  2415. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2416. assert.NoError(t, err)
  2417. user.Permissions = make(map[string][]string)
  2418. user.Permissions["/"] = allPerms
  2419. user.Password = defaultPassword
  2420. user.ID = 0
  2421. user.CreatedAt = 0
  2422. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2423. assert.NoError(t, err, string(resp))
  2424. }
  2425. }
  2426. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2427. assert.NoError(t, err)
  2428. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2429. assert.NoError(t, err)
  2430. err = os.RemoveAll(localUser.GetHomeDir())
  2431. assert.NoError(t, err)
  2432. }
  2433. func TestSymlink(t *testing.T) {
  2434. u := getTestUser()
  2435. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2436. assert.NoError(t, err)
  2437. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  2438. assert.NoError(t, err)
  2439. testFilePath := filepath.Join(homeBasePath, testFileName)
  2440. testFileSize := int64(65535)
  2441. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2442. err = createTestFile(testFilePath, testFileSize)
  2443. assert.NoError(t, err)
  2444. client, err := getFTPClient(user, false, nil)
  2445. if assert.NoError(t, err) {
  2446. err = checkBasicFTP(client)
  2447. assert.NoError(t, err)
  2448. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2449. assert.NoError(t, err)
  2450. code, _, err := client.SendCommand("SITE SYMLINK %v %v", testFileName, testFileName+".link")
  2451. assert.NoError(t, err)
  2452. assert.Equal(t, ftp.StatusCommandOK, code)
  2453. if runtime.GOOS != osWindows {
  2454. testDir := "adir"
  2455. otherDir := "dir"
  2456. err = client.MakeDir(otherDir)
  2457. assert.NoError(t, err)
  2458. err = client.MakeDir(path.Join(otherDir, testDir))
  2459. assert.NoError(t, err)
  2460. code, response, err := client.SendCommand("SITE CHMOD 0001 %v", otherDir)
  2461. assert.NoError(t, err)
  2462. assert.Equal(t, ftp.StatusCommandOK, code)
  2463. assert.Equal(t, "SITE CHMOD command successful", response)
  2464. code, _, err = client.SendCommand("SITE SYMLINK %v %v", testDir, path.Join(otherDir, testDir))
  2465. assert.NoError(t, err)
  2466. assert.Equal(t, ftp.StatusFileUnavailable, code)
  2467. code, response, err = client.SendCommand("SITE CHMOD 755 %v", otherDir)
  2468. assert.NoError(t, err)
  2469. assert.Equal(t, ftp.StatusCommandOK, code)
  2470. assert.Equal(t, "SITE CHMOD command successful", response)
  2471. }
  2472. err = client.Quit()
  2473. assert.NoError(t, err)
  2474. if user.Username == defaultUsername {
  2475. err = os.RemoveAll(user.GetHomeDir())
  2476. assert.NoError(t, err)
  2477. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2478. assert.NoError(t, err)
  2479. user.Password = defaultPassword
  2480. user.ID = 0
  2481. user.CreatedAt = 0
  2482. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2483. assert.NoError(t, err, string(resp))
  2484. }
  2485. }
  2486. err = os.Remove(testFilePath)
  2487. assert.NoError(t, err)
  2488. }
  2489. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2490. assert.NoError(t, err)
  2491. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2492. assert.NoError(t, err)
  2493. err = os.RemoveAll(localUser.GetHomeDir())
  2494. assert.NoError(t, err)
  2495. }
  2496. func TestStat(t *testing.T) {
  2497. u := getTestUser()
  2498. u.Permissions["/subdir"] = []string{dataprovider.PermUpload}
  2499. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2500. assert.NoError(t, err)
  2501. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  2502. assert.NoError(t, err)
  2503. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2504. client, err := getFTPClient(user, false, nil)
  2505. if assert.NoError(t, err) {
  2506. subDir := "subdir"
  2507. testFilePath := filepath.Join(homeBasePath, testFileName)
  2508. testFileSize := int64(65535)
  2509. err = createTestFile(testFilePath, testFileSize)
  2510. assert.NoError(t, err)
  2511. err = client.MakeDir(subDir)
  2512. assert.NoError(t, err)
  2513. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2514. assert.NoError(t, err)
  2515. err = ftpUploadFile(testFilePath, path.Join("/", subDir, testFileName), testFileSize, client, 0)
  2516. assert.Error(t, err)
  2517. size, err := client.FileSize(testFileName)
  2518. assert.NoError(t, err)
  2519. assert.Equal(t, testFileSize, size)
  2520. _, err = client.FileSize(path.Join("/", subDir, testFileName))
  2521. assert.Error(t, err)
  2522. _, err = client.FileSize("missing file")
  2523. assert.Error(t, err)
  2524. err = client.Quit()
  2525. assert.NoError(t, err)
  2526. err = os.Remove(testFilePath)
  2527. assert.NoError(t, err)
  2528. if user.Username == defaultUsername {
  2529. err = os.RemoveAll(user.GetHomeDir())
  2530. assert.NoError(t, err)
  2531. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2532. assert.NoError(t, err)
  2533. user.Password = defaultPassword
  2534. user.ID = 0
  2535. user.CreatedAt = 0
  2536. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2537. assert.NoError(t, err, string(resp))
  2538. }
  2539. }
  2540. }
  2541. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2542. assert.NoError(t, err)
  2543. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2544. assert.NoError(t, err)
  2545. err = os.RemoveAll(localUser.GetHomeDir())
  2546. assert.NoError(t, err)
  2547. }
  2548. func TestUploadOverwriteVfolder(t *testing.T) {
  2549. u := getTestUser()
  2550. vdir := "/vdir"
  2551. mappedPath := filepath.Join(os.TempDir(), "vdir")
  2552. folderName := filepath.Base(mappedPath)
  2553. f := vfs.BaseVirtualFolder{
  2554. Name: folderName,
  2555. MappedPath: mappedPath,
  2556. }
  2557. _, _, err := httpdtest.AddFolder(f, http.StatusCreated)
  2558. assert.NoError(t, err)
  2559. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  2560. BaseVirtualFolder: vfs.BaseVirtualFolder{
  2561. Name: folderName,
  2562. },
  2563. VirtualPath: vdir,
  2564. QuotaSize: -1,
  2565. QuotaFiles: -1,
  2566. })
  2567. err = os.MkdirAll(mappedPath, os.ModePerm)
  2568. assert.NoError(t, err)
  2569. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2570. assert.NoError(t, err)
  2571. client, err := getFTPClient(user, false, nil)
  2572. if assert.NoError(t, err) {
  2573. testFilePath := filepath.Join(homeBasePath, testFileName)
  2574. testFileSize := int64(65535)
  2575. err = createTestFile(testFilePath, testFileSize)
  2576. assert.NoError(t, err)
  2577. err = ftpUploadFile(testFilePath, path.Join(vdir, testFileName), testFileSize, client, 0)
  2578. assert.NoError(t, err)
  2579. folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
  2580. assert.NoError(t, err)
  2581. assert.Equal(t, testFileSize, folder.UsedQuotaSize)
  2582. assert.Equal(t, 1, folder.UsedQuotaFiles)
  2583. err = ftpUploadFile(testFilePath, path.Join(vdir, testFileName), testFileSize, client, 0)
  2584. assert.NoError(t, err)
  2585. folder, _, err = httpdtest.GetFolderByName(folderName, http.StatusOK)
  2586. assert.NoError(t, err)
  2587. assert.Equal(t, testFileSize, folder.UsedQuotaSize)
  2588. assert.Equal(t, 1, folder.UsedQuotaFiles)
  2589. err = client.Quit()
  2590. assert.NoError(t, err)
  2591. err = os.Remove(testFilePath)
  2592. assert.NoError(t, err)
  2593. }
  2594. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2595. assert.NoError(t, err)
  2596. _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
  2597. assert.NoError(t, err)
  2598. err = os.RemoveAll(user.GetHomeDir())
  2599. assert.NoError(t, err)
  2600. err = os.RemoveAll(mappedPath)
  2601. assert.NoError(t, err)
  2602. }
  2603. func TestTransferQuotaLimits(t *testing.T) {
  2604. u := getTestUser()
  2605. u.DownloadDataTransfer = 1
  2606. u.UploadDataTransfer = 1
  2607. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2608. assert.NoError(t, err)
  2609. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  2610. testFilePath := filepath.Join(homeBasePath, testFileName)
  2611. testFileSize := int64(524288)
  2612. err = createTestFile(testFilePath, testFileSize)
  2613. assert.NoError(t, err)
  2614. client, err := getFTPClient(user, false, nil)
  2615. if assert.NoError(t, err) {
  2616. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2617. assert.NoError(t, err)
  2618. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2619. assert.NoError(t, err)
  2620. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2621. if assert.Error(t, err) {
  2622. assert.Contains(t, err.Error(), ftpserver.ErrStorageExceeded.Error())
  2623. }
  2624. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  2625. assert.NoError(t, err)
  2626. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  2627. assert.NoError(t, err)
  2628. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  2629. if assert.Error(t, err) {
  2630. assert.Contains(t, err.Error(), common.ErrReadQuotaExceeded.Error())
  2631. }
  2632. err = client.Quit()
  2633. assert.NoError(t, err)
  2634. }
  2635. testFileSize = int64(600000)
  2636. err = createTestFile(testFilePath, testFileSize)
  2637. assert.NoError(t, err)
  2638. user.DownloadDataTransfer = 2
  2639. user.UploadDataTransfer = 2
  2640. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2641. assert.NoError(t, err)
  2642. client, err = getFTPClient(user, false, nil)
  2643. if assert.NoError(t, err) {
  2644. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2645. assert.NoError(t, err)
  2646. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  2647. assert.NoError(t, err)
  2648. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  2649. assert.Error(t, err)
  2650. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2651. assert.Error(t, err)
  2652. err = client.Quit()
  2653. assert.NoError(t, err)
  2654. }
  2655. err = os.Remove(localDownloadPath)
  2656. assert.NoError(t, err)
  2657. err = os.Remove(testFilePath)
  2658. assert.NoError(t, err)
  2659. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2660. assert.NoError(t, err)
  2661. err = os.RemoveAll(user.GetHomeDir())
  2662. assert.NoError(t, err)
  2663. }
  2664. func TestAllocateAvailable(t *testing.T) {
  2665. u := getTestUser()
  2666. mappedPath := filepath.Join(os.TempDir(), "vdir")
  2667. folderName := filepath.Base(mappedPath)
  2668. f := vfs.BaseVirtualFolder{
  2669. Name: folderName,
  2670. MappedPath: mappedPath,
  2671. }
  2672. _, _, err := httpdtest.AddFolder(f, http.StatusCreated)
  2673. assert.NoError(t, err)
  2674. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  2675. BaseVirtualFolder: vfs.BaseVirtualFolder{
  2676. Name: folderName,
  2677. },
  2678. VirtualPath: "/vdir",
  2679. QuotaSize: 110,
  2680. })
  2681. err = os.MkdirAll(mappedPath, os.ModePerm)
  2682. assert.NoError(t, err)
  2683. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2684. assert.NoError(t, err)
  2685. client, err := getFTPClient(user, false, nil)
  2686. if assert.NoError(t, err) {
  2687. code, response, err := client.SendCommand("allo 2000000")
  2688. assert.NoError(t, err)
  2689. assert.Equal(t, ftp.StatusCommandOK, code)
  2690. assert.Equal(t, "Done !", response)
  2691. code, response, err = client.SendCommand("AVBL /vdir")
  2692. assert.NoError(t, err)
  2693. assert.Equal(t, ftp.StatusFile, code)
  2694. assert.Equal(t, "110", response)
  2695. code, _, err = client.SendCommand("AVBL")
  2696. assert.NoError(t, err)
  2697. assert.Equal(t, ftp.StatusFile, code)
  2698. err = client.Quit()
  2699. assert.NoError(t, err)
  2700. }
  2701. user.QuotaSize = 100
  2702. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2703. assert.NoError(t, err)
  2704. client, err = getFTPClient(user, false, nil)
  2705. if assert.NoError(t, err) {
  2706. testFilePath := filepath.Join(homeBasePath, testFileName)
  2707. testFileSize := user.QuotaSize - 1
  2708. err = createTestFile(testFilePath, testFileSize)
  2709. assert.NoError(t, err)
  2710. code, response, err := client.SendCommand("allo 1000")
  2711. assert.NoError(t, err)
  2712. assert.Equal(t, ftp.StatusCommandOK, code)
  2713. assert.Equal(t, "Done !", response)
  2714. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2715. assert.NoError(t, err)
  2716. code, response, err = client.SendCommand("AVBL")
  2717. assert.NoError(t, err)
  2718. assert.Equal(t, ftp.StatusFile, code)
  2719. assert.Equal(t, "1", response)
  2720. err = client.Quit()
  2721. assert.NoError(t, err)
  2722. err = os.Remove(testFilePath)
  2723. assert.NoError(t, err)
  2724. }
  2725. user.TotalDataTransfer = 1
  2726. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2727. assert.NoError(t, err)
  2728. client, err = getFTPClient(user, false, nil)
  2729. if assert.NoError(t, err) {
  2730. code, response, err := client.SendCommand("AVBL")
  2731. assert.NoError(t, err)
  2732. assert.Equal(t, ftp.StatusFile, code)
  2733. assert.Equal(t, "1", response)
  2734. err = client.Quit()
  2735. assert.NoError(t, err)
  2736. }
  2737. user.TotalDataTransfer = 0
  2738. user.UploadDataTransfer = 5
  2739. user.QuotaSize = 6 * 1024 * 1024
  2740. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2741. assert.NoError(t, err)
  2742. client, err = getFTPClient(user, false, nil)
  2743. if assert.NoError(t, err) {
  2744. code, response, err := client.SendCommand("AVBL")
  2745. assert.NoError(t, err)
  2746. assert.Equal(t, ftp.StatusFile, code)
  2747. assert.Equal(t, "5242880", response)
  2748. err = client.Quit()
  2749. assert.NoError(t, err)
  2750. }
  2751. user.TotalDataTransfer = 0
  2752. user.UploadDataTransfer = 5
  2753. user.QuotaSize = 0
  2754. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2755. assert.NoError(t, err)
  2756. client, err = getFTPClient(user, false, nil)
  2757. if assert.NoError(t, err) {
  2758. code, response, err := client.SendCommand("AVBL")
  2759. assert.NoError(t, err)
  2760. assert.Equal(t, ftp.StatusFile, code)
  2761. assert.Equal(t, "5242880", response)
  2762. err = client.Quit()
  2763. assert.NoError(t, err)
  2764. }
  2765. user.Filters.MaxUploadFileSize = 100
  2766. user.QuotaSize = 0
  2767. user.TotalDataTransfer = 0
  2768. user.UploadDataTransfer = 0
  2769. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2770. assert.NoError(t, err)
  2771. client, err = getFTPClient(user, false, nil)
  2772. if assert.NoError(t, err) {
  2773. code, response, err := client.SendCommand("allo 10000")
  2774. assert.NoError(t, err)
  2775. assert.Equal(t, ftp.StatusCommandOK, code)
  2776. assert.Equal(t, "Done !", response)
  2777. code, response, err = client.SendCommand("AVBL")
  2778. assert.NoError(t, err)
  2779. assert.Equal(t, ftp.StatusFile, code)
  2780. assert.Equal(t, "100", response)
  2781. err = client.Quit()
  2782. assert.NoError(t, err)
  2783. }
  2784. user.QuotaSize = 50
  2785. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2786. assert.NoError(t, err)
  2787. client, err = getFTPClient(user, false, nil)
  2788. if assert.NoError(t, err) {
  2789. code, response, err := client.SendCommand("AVBL")
  2790. assert.NoError(t, err)
  2791. assert.Equal(t, ftp.StatusFile, code)
  2792. assert.Equal(t, "0", response)
  2793. }
  2794. user.QuotaSize = 1000
  2795. user.Filters.MaxUploadFileSize = 1
  2796. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  2797. assert.NoError(t, err)
  2798. client, err = getFTPClient(user, false, nil)
  2799. if assert.NoError(t, err) {
  2800. code, response, err := client.SendCommand("AVBL")
  2801. assert.NoError(t, err)
  2802. assert.Equal(t, ftp.StatusFile, code)
  2803. assert.Equal(t, "1", response)
  2804. }
  2805. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2806. assert.NoError(t, err)
  2807. _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
  2808. assert.NoError(t, err)
  2809. err = os.RemoveAll(user.GetHomeDir())
  2810. assert.NoError(t, err)
  2811. err = os.RemoveAll(mappedPath)
  2812. assert.NoError(t, err)
  2813. }
  2814. func TestAvailableSFTPFs(t *testing.T) {
  2815. u := getTestUser()
  2816. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2817. assert.NoError(t, err)
  2818. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  2819. assert.NoError(t, err)
  2820. client, err := getFTPClient(sftpUser, false, nil)
  2821. if assert.NoError(t, err) {
  2822. code, response, err := client.SendCommand("AVBL /")
  2823. assert.NoError(t, err)
  2824. assert.Equal(t, ftp.StatusFile, code)
  2825. avblSize, err := strconv.ParseInt(response, 10, 64)
  2826. assert.NoError(t, err)
  2827. assert.Greater(t, avblSize, int64(0))
  2828. err = client.Quit()
  2829. assert.NoError(t, err)
  2830. }
  2831. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2832. assert.NoError(t, err)
  2833. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2834. assert.NoError(t, err)
  2835. err = os.RemoveAll(localUser.GetHomeDir())
  2836. assert.NoError(t, err)
  2837. }
  2838. func TestChtimes(t *testing.T) {
  2839. u := getTestUser()
  2840. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2841. assert.NoError(t, err)
  2842. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  2843. assert.NoError(t, err)
  2844. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2845. client, err := getFTPClient(user, false, nil)
  2846. if assert.NoError(t, err) {
  2847. testFilePath := filepath.Join(homeBasePath, testFileName)
  2848. testFileSize := int64(65535)
  2849. err = createTestFile(testFilePath, testFileSize)
  2850. assert.NoError(t, err)
  2851. err = checkBasicFTP(client)
  2852. assert.NoError(t, err)
  2853. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2854. assert.NoError(t, err)
  2855. mtime := time.Now().Format("20060102150405")
  2856. code, response, err := client.SendCommand("MFMT %v %v", mtime, testFileName)
  2857. assert.NoError(t, err)
  2858. assert.Equal(t, ftp.StatusFile, code)
  2859. assert.Equal(t, fmt.Sprintf("Modify=%v; %v", mtime, testFileName), response)
  2860. err = client.Quit()
  2861. assert.NoError(t, err)
  2862. err = os.Remove(testFilePath)
  2863. assert.NoError(t, err)
  2864. if user.Username == defaultUsername {
  2865. err = os.RemoveAll(user.GetHomeDir())
  2866. assert.NoError(t, err)
  2867. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2868. assert.NoError(t, err)
  2869. user.Password = defaultPassword
  2870. user.ID = 0
  2871. user.CreatedAt = 0
  2872. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  2873. assert.NoError(t, err, string(resp))
  2874. }
  2875. }
  2876. }
  2877. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  2878. assert.NoError(t, err)
  2879. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  2880. assert.NoError(t, err)
  2881. err = os.RemoveAll(localUser.GetHomeDir())
  2882. assert.NoError(t, err)
  2883. }
  2884. func TestMODEType(t *testing.T) {
  2885. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  2886. assert.NoError(t, err)
  2887. client, err := getFTPClient(user, false, nil)
  2888. if assert.NoError(t, err) {
  2889. code, response, err := client.SendCommand("MODE s")
  2890. assert.NoError(t, err)
  2891. assert.Equal(t, ftp.StatusCommandOK, code)
  2892. assert.Equal(t, "Transfer mode set to 'S'", response)
  2893. code, response, err = client.SendCommand("MODE S")
  2894. assert.NoError(t, err)
  2895. assert.Equal(t, ftp.StatusCommandOK, code)
  2896. assert.Equal(t, "Transfer mode set to 'S'", response)
  2897. code, _, err = client.SendCommand("MODE Z")
  2898. assert.NoError(t, err)
  2899. assert.Equal(t, ftp.StatusNotImplementedParameter, code)
  2900. code, _, err = client.SendCommand("MODE SS")
  2901. assert.NoError(t, err)
  2902. assert.Equal(t, ftp.StatusBadArguments, code)
  2903. err = client.Quit()
  2904. assert.NoError(t, err)
  2905. }
  2906. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2907. assert.NoError(t, err)
  2908. err = os.RemoveAll(user.GetHomeDir())
  2909. assert.NoError(t, err)
  2910. }
  2911. func TestSTAT(t *testing.T) {
  2912. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  2913. assert.NoError(t, err)
  2914. client, err := getFTPClient(user, false, nil)
  2915. if assert.NoError(t, err) {
  2916. testFilePath := filepath.Join(homeBasePath, testFileName)
  2917. testFileSize := int64(131072)
  2918. err = createTestFile(testFilePath, testFileSize)
  2919. assert.NoError(t, err)
  2920. testDir := "testdir"
  2921. err = client.MakeDir(testDir)
  2922. assert.NoError(t, err)
  2923. err = ftpUploadFile(testFilePath, path.Join(testDir, testFileName), testFileSize, client, 0)
  2924. assert.NoError(t, err)
  2925. err = ftpUploadFile(testFilePath, path.Join(testDir, testFileName+"_1"), testFileSize, client, 0)
  2926. assert.NoError(t, err)
  2927. code, response, err := client.SendCommand("STAT %s", testDir)
  2928. assert.NoError(t, err)
  2929. assert.Equal(t, ftp.StatusDirectory, code)
  2930. assert.Contains(t, response, fmt.Sprintf("STAT %s", testDir))
  2931. assert.Contains(t, response, testFileName)
  2932. assert.Contains(t, response, testFileName+"_1")
  2933. assert.Contains(t, response, "End")
  2934. err = client.Quit()
  2935. assert.NoError(t, err)
  2936. }
  2937. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2938. assert.NoError(t, err)
  2939. err = os.RemoveAll(user.GetHomeDir())
  2940. assert.NoError(t, err)
  2941. }
  2942. func TestChown(t *testing.T) {
  2943. if runtime.GOOS == osWindows {
  2944. t.Skip("chown is not supported on Windows")
  2945. }
  2946. user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
  2947. assert.NoError(t, err)
  2948. client, err := getFTPClient(user, true, nil)
  2949. if assert.NoError(t, err) {
  2950. testFilePath := filepath.Join(homeBasePath, testFileName)
  2951. testFileSize := int64(131072)
  2952. err = createTestFile(testFilePath, testFileSize)
  2953. assert.NoError(t, err)
  2954. err = checkBasicFTP(client)
  2955. assert.NoError(t, err)
  2956. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2957. assert.NoError(t, err)
  2958. code, response, err := client.SendCommand("SITE CHOWN 1000:1000 %v", testFileName)
  2959. assert.NoError(t, err)
  2960. assert.Equal(t, ftp.StatusFileUnavailable, code)
  2961. assert.Equal(t, "Couldn't chown: operation unsupported", response)
  2962. err = client.Quit()
  2963. assert.NoError(t, err)
  2964. err = os.Remove(testFilePath)
  2965. assert.NoError(t, err)
  2966. }
  2967. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  2968. assert.NoError(t, err)
  2969. err = os.RemoveAll(user.GetHomeDir())
  2970. assert.NoError(t, err)
  2971. }
  2972. func TestChmod(t *testing.T) {
  2973. if runtime.GOOS == osWindows {
  2974. t.Skip("chmod is partially supported on Windows")
  2975. }
  2976. u := getTestUser()
  2977. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  2978. assert.NoError(t, err)
  2979. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  2980. assert.NoError(t, err)
  2981. for _, user := range []dataprovider.User{localUser, sftpUser} {
  2982. client, err := getFTPClient(user, true, nil)
  2983. if assert.NoError(t, err) {
  2984. testFilePath := filepath.Join(homeBasePath, testFileName)
  2985. testFileSize := int64(131072)
  2986. err = createTestFile(testFilePath, testFileSize)
  2987. assert.NoError(t, err)
  2988. err = checkBasicFTP(client)
  2989. assert.NoError(t, err)
  2990. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  2991. assert.NoError(t, err)
  2992. code, response, err := client.SendCommand("SITE CHMOD 600 %v", testFileName)
  2993. assert.NoError(t, err)
  2994. assert.Equal(t, ftp.StatusCommandOK, code)
  2995. assert.Equal(t, "SITE CHMOD command successful", response)
  2996. fi, err := os.Stat(filepath.Join(user.HomeDir, testFileName))
  2997. if assert.NoError(t, err) {
  2998. assert.Equal(t, os.FileMode(0600), fi.Mode().Perm())
  2999. }
  3000. err = client.Quit()
  3001. assert.NoError(t, err)
  3002. err = os.Remove(testFilePath)
  3003. assert.NoError(t, err)
  3004. if user.Username == defaultUsername {
  3005. err = os.RemoveAll(user.GetHomeDir())
  3006. assert.NoError(t, err)
  3007. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3008. assert.NoError(t, err)
  3009. user.Password = defaultPassword
  3010. user.ID = 0
  3011. user.CreatedAt = 0
  3012. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  3013. assert.NoError(t, err, string(resp))
  3014. }
  3015. }
  3016. }
  3017. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  3018. assert.NoError(t, err)
  3019. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  3020. assert.NoError(t, err)
  3021. err = os.RemoveAll(localUser.GetHomeDir())
  3022. assert.NoError(t, err)
  3023. }
  3024. func TestCombineDisabled(t *testing.T) {
  3025. u := getTestUser()
  3026. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3027. assert.NoError(t, err)
  3028. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  3029. assert.NoError(t, err)
  3030. for _, user := range []dataprovider.User{localUser, sftpUser} {
  3031. client, err := getFTPClient(user, true, nil)
  3032. if assert.NoError(t, err) {
  3033. err = checkBasicFTP(client)
  3034. assert.NoError(t, err)
  3035. code, response, err := client.SendCommand("COMB file file.1 file.2")
  3036. assert.NoError(t, err)
  3037. assert.Equal(t, ftp.StatusNotImplemented, code)
  3038. assert.Equal(t, "COMB support is disabled", response)
  3039. err = client.Quit()
  3040. assert.NoError(t, err)
  3041. }
  3042. }
  3043. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  3044. assert.NoError(t, err)
  3045. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  3046. assert.NoError(t, err)
  3047. err = os.RemoveAll(localUser.GetHomeDir())
  3048. assert.NoError(t, err)
  3049. }
  3050. func TestActiveModeDisabled(t *testing.T) {
  3051. u := getTestUser()
  3052. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3053. assert.NoError(t, err)
  3054. client, err := getFTPClientImplicitTLS(user)
  3055. if assert.NoError(t, err) {
  3056. err = checkBasicFTP(client)
  3057. assert.NoError(t, err)
  3058. code, response, err := client.SendCommand("PORT 10,2,0,2,4,31")
  3059. assert.NoError(t, err)
  3060. assert.Equal(t, ftp.StatusNotAvailable, code)
  3061. assert.Equal(t, "PORT command is disabled", response)
  3062. code, response, err = client.SendCommand("EPRT |1|132.235.1.2|6275|")
  3063. assert.NoError(t, err)
  3064. assert.Equal(t, ftp.StatusNotAvailable, code)
  3065. assert.Equal(t, "EPRT command is disabled", response)
  3066. err = client.Quit()
  3067. assert.NoError(t, err)
  3068. }
  3069. client, err = getFTPClient(user, false, nil)
  3070. if assert.NoError(t, err) {
  3071. code, response, err := client.SendCommand("PORT 10,2,0,2,4,31")
  3072. assert.NoError(t, err)
  3073. assert.Equal(t, ftp.StatusBadArguments, code)
  3074. assert.Equal(t, "Your request does not meet the configured security requirements", response)
  3075. code, response, err = client.SendCommand("EPRT |1|132.235.1.2|6275|")
  3076. assert.NoError(t, err)
  3077. assert.Equal(t, ftp.StatusBadArguments, code)
  3078. assert.Equal(t, "Your request does not meet the configured security requirements", response)
  3079. err = client.Quit()
  3080. assert.NoError(t, err)
  3081. }
  3082. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3083. assert.NoError(t, err)
  3084. err = os.RemoveAll(user.GetHomeDir())
  3085. assert.NoError(t, err)
  3086. }
  3087. func TestSITEDisabled(t *testing.T) {
  3088. u := getTestUser()
  3089. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3090. assert.NoError(t, err)
  3091. client, err := getFTPClientImplicitTLS(user)
  3092. if assert.NoError(t, err) {
  3093. err = checkBasicFTP(client)
  3094. assert.NoError(t, err)
  3095. code, response, err := client.SendCommand("SITE CHMOD 600 afile.txt")
  3096. assert.NoError(t, err)
  3097. assert.Equal(t, ftp.StatusBadCommand, code)
  3098. assert.Equal(t, "SITE support is disabled", response)
  3099. err = client.Quit()
  3100. assert.NoError(t, err)
  3101. }
  3102. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3103. assert.NoError(t, err)
  3104. err = os.RemoveAll(user.GetHomeDir())
  3105. assert.NoError(t, err)
  3106. }
  3107. func TestHASH(t *testing.T) {
  3108. u := getTestUser()
  3109. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3110. assert.NoError(t, err)
  3111. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  3112. assert.NoError(t, err)
  3113. u = getTestUserWithCryptFs()
  3114. u.Username += "_crypt"
  3115. cryptUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3116. assert.NoError(t, err)
  3117. for _, user := range []dataprovider.User{localUser, sftpUser, cryptUser} {
  3118. client, err := getFTPClientImplicitTLS(user)
  3119. if assert.NoError(t, err) {
  3120. testFilePath := filepath.Join(homeBasePath, testFileName)
  3121. testFileSize := int64(131072)
  3122. err = createTestFile(testFilePath, testFileSize)
  3123. assert.NoError(t, err)
  3124. err = checkBasicFTP(client)
  3125. assert.NoError(t, err)
  3126. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  3127. assert.NoError(t, err)
  3128. h := sha256.New()
  3129. f, err := os.Open(testFilePath)
  3130. assert.NoError(t, err)
  3131. _, err = io.Copy(h, f)
  3132. assert.NoError(t, err)
  3133. hash := hex.EncodeToString(h.Sum(nil))
  3134. err = f.Close()
  3135. assert.NoError(t, err)
  3136. code, response, err := client.SendCommand("XSHA256 %v", testFileName)
  3137. assert.NoError(t, err)
  3138. assert.Equal(t, ftp.StatusRequestedFileActionOK, code)
  3139. assert.Contains(t, response, hash)
  3140. code, response, err = client.SendCommand("HASH %v", testFileName)
  3141. assert.NoError(t, err)
  3142. assert.Equal(t, ftp.StatusFile, code)
  3143. assert.Contains(t, response, hash)
  3144. err = client.Quit()
  3145. assert.NoError(t, err)
  3146. err = os.Remove(testFilePath)
  3147. assert.NoError(t, err)
  3148. if user.Username == defaultUsername {
  3149. err = os.RemoveAll(user.GetHomeDir())
  3150. assert.NoError(t, err)
  3151. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3152. assert.NoError(t, err)
  3153. user.Password = defaultPassword
  3154. user.ID = 0
  3155. user.CreatedAt = 0
  3156. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  3157. assert.NoError(t, err, string(resp))
  3158. }
  3159. }
  3160. }
  3161. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  3162. assert.NoError(t, err)
  3163. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  3164. assert.NoError(t, err)
  3165. err = os.RemoveAll(localUser.GetHomeDir())
  3166. assert.NoError(t, err)
  3167. _, err = httpdtest.RemoveUser(cryptUser, http.StatusOK)
  3168. assert.NoError(t, err)
  3169. err = os.RemoveAll(cryptUser.GetHomeDir())
  3170. assert.NoError(t, err)
  3171. }
  3172. func TestCombine(t *testing.T) {
  3173. u := getTestUser()
  3174. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3175. assert.NoError(t, err)
  3176. sftpUser, _, err := httpdtest.AddUser(getTestSFTPUser(), http.StatusCreated)
  3177. assert.NoError(t, err)
  3178. for _, user := range []dataprovider.User{localUser, sftpUser} {
  3179. client, err := getFTPClientImplicitTLS(user)
  3180. if assert.NoError(t, err) {
  3181. testFilePath := filepath.Join(homeBasePath, testFileName)
  3182. testFileSize := int64(131072)
  3183. err = createTestFile(testFilePath, testFileSize)
  3184. assert.NoError(t, err)
  3185. err = checkBasicFTP(client)
  3186. assert.NoError(t, err)
  3187. err = ftpUploadFile(testFilePath, testFileName+".1", testFileSize, client, 0)
  3188. assert.NoError(t, err)
  3189. err = ftpUploadFile(testFilePath, testFileName+".2", testFileSize, client, 0)
  3190. assert.NoError(t, err)
  3191. code, response, err := client.SendCommand("COMB %v %v %v", testFileName, testFileName+".1", testFileName+".2")
  3192. assert.NoError(t, err)
  3193. if user.Username == defaultUsername {
  3194. assert.Equal(t, ftp.StatusRequestedFileActionOK, code)
  3195. assert.Equal(t, "COMB succeeded!", response)
  3196. } else {
  3197. assert.Equal(t, ftp.StatusFileUnavailable, code)
  3198. assert.Contains(t, response, "COMB is not supported for this filesystem")
  3199. }
  3200. err = client.Quit()
  3201. assert.NoError(t, err)
  3202. err = os.Remove(testFilePath)
  3203. assert.NoError(t, err)
  3204. if user.Username == defaultUsername {
  3205. err = os.RemoveAll(user.GetHomeDir())
  3206. assert.NoError(t, err)
  3207. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3208. assert.NoError(t, err)
  3209. user.Password = defaultPassword
  3210. user.ID = 0
  3211. user.CreatedAt = 0
  3212. _, resp, err := httpdtest.AddUser(user, http.StatusCreated)
  3213. assert.NoError(t, err, string(resp))
  3214. }
  3215. }
  3216. }
  3217. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  3218. assert.NoError(t, err)
  3219. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  3220. assert.NoError(t, err)
  3221. err = os.RemoveAll(localUser.GetHomeDir())
  3222. assert.NoError(t, err)
  3223. }
  3224. func TestTLSSessionReuse(t *testing.T) {
  3225. u := getTestUser()
  3226. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3227. assert.NoError(t, err)
  3228. client, err := getFTPClientWithSessionReuse(user, nil)
  3229. if assert.NoError(t, err) {
  3230. err = checkBasicFTP(client)
  3231. assert.NoError(t, err)
  3232. testFilePath := filepath.Join(homeBasePath, testFileName)
  3233. testFileSize := int64(65535)
  3234. err = createTestFile(testFilePath, testFileSize)
  3235. assert.NoError(t, err)
  3236. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  3237. assert.NoError(t, err)
  3238. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  3239. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  3240. assert.NoError(t, err)
  3241. entries, err := client.List("/")
  3242. assert.NoError(t, err)
  3243. assert.Len(t, entries, 1)
  3244. err = client.Quit()
  3245. assert.NoError(t, err)
  3246. err = os.Remove(testFilePath)
  3247. assert.NoError(t, err)
  3248. err = os.Remove(localDownloadPath)
  3249. assert.NoError(t, err)
  3250. }
  3251. // this TLS config does not support session resumption
  3252. tlsConfig := &tls.Config{
  3253. ServerName: "localhost",
  3254. InsecureSkipVerify: true, // use this for tests only
  3255. MinVersion: tls.VersionTLS12,
  3256. }
  3257. client, err = getFTPClientWithSessionReuse(user, tlsConfig)
  3258. if assert.NoError(t, err) {
  3259. err = checkBasicFTP(client)
  3260. assert.Error(t, err)
  3261. err = client.Quit()
  3262. assert.NoError(t, err)
  3263. }
  3264. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3265. assert.NoError(t, err)
  3266. err = os.RemoveAll(user.GetHomeDir())
  3267. assert.NoError(t, err)
  3268. }
  3269. func TestClientCertificateAuthRevokedCert(t *testing.T) {
  3270. u := getTestUser()
  3271. u.Username = tlsClient2Username
  3272. u.Filters.TLSUsername = sdk.TLSUsernameCN
  3273. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3274. assert.NoError(t, err)
  3275. tlsConfig := &tls.Config{
  3276. ServerName: "localhost",
  3277. InsecureSkipVerify: true, // use this for tests only
  3278. MinVersion: tls.VersionTLS12,
  3279. ClientSessionCache: tls.NewLRUClientSessionCache(0),
  3280. }
  3281. tlsCert, err := tls.X509KeyPair([]byte(client2Crt), []byte(client2Key))
  3282. assert.NoError(t, err)
  3283. tlsConfig.Certificates = append(tlsConfig.Certificates, tlsCert)
  3284. _, err = getFTPClientWithSessionReuse(user, tlsConfig)
  3285. if assert.Error(t, err) {
  3286. assert.Contains(t, err.Error(), "bad certificate")
  3287. }
  3288. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3289. assert.NoError(t, err)
  3290. err = os.RemoveAll(user.GetHomeDir())
  3291. assert.NoError(t, err)
  3292. }
  3293. func TestClientCertificateAuth(t *testing.T) {
  3294. u := getTestUser()
  3295. u.Username = tlsClient1Username
  3296. u.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
  3297. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3298. assert.NoError(t, err)
  3299. tlsConfig := &tls.Config{
  3300. ServerName: "localhost",
  3301. InsecureSkipVerify: true, // use this for tests only
  3302. MinVersion: tls.VersionTLS12,
  3303. }
  3304. tlsCert, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  3305. assert.NoError(t, err)
  3306. tlsConfig.Certificates = append(tlsConfig.Certificates, tlsCert)
  3307. // TLS username is not enabled, mutual TLS should fail
  3308. _, err = getFTPClient(user, true, tlsConfig)
  3309. if assert.Error(t, err) {
  3310. assert.Contains(t, err.Error(), "login method password is not allowed")
  3311. }
  3312. user.Filters.TLSUsername = sdk.TLSUsernameCN
  3313. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  3314. assert.NoError(t, err)
  3315. client, err := getFTPClient(user, true, tlsConfig)
  3316. if assert.NoError(t, err) {
  3317. err = checkBasicFTP(client)
  3318. assert.NoError(t, err)
  3319. err = client.Quit()
  3320. assert.NoError(t, err)
  3321. }
  3322. // now use a valid certificate with a CN different from username
  3323. u = getTestUser()
  3324. u.Username = tlsClient2Username
  3325. u.Filters.TLSUsername = sdk.TLSUsernameCN
  3326. u.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
  3327. user2, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3328. assert.NoError(t, err)
  3329. _, err = getFTPClient(user2, true, tlsConfig)
  3330. if assert.Error(t, err) {
  3331. assert.Contains(t, err.Error(), "does not match username")
  3332. }
  3333. // add the certs to the user
  3334. user2.Filters.TLSUsername = sdk.TLSUsernameNone
  3335. user2.Filters.TLSCerts = []string{client2Crt, client1Crt}
  3336. user2, _, err = httpdtest.UpdateUser(user2, http.StatusOK, "")
  3337. assert.NoError(t, err)
  3338. client, err = getFTPClient(user2, true, tlsConfig)
  3339. if assert.NoError(t, err) {
  3340. err = checkBasicFTP(client)
  3341. assert.NoError(t, err)
  3342. err = client.Quit()
  3343. assert.NoError(t, err)
  3344. }
  3345. user2.Filters.TLSCerts = []string{client2Crt}
  3346. user2, _, err = httpdtest.UpdateUser(user2, http.StatusOK, "")
  3347. assert.NoError(t, err)
  3348. _, err = getFTPClient(user2, true, tlsConfig)
  3349. if assert.Error(t, err) {
  3350. assert.Contains(t, err.Error(), "TLS certificate is not valid")
  3351. }
  3352. // now disable certificate authentication
  3353. user.Filters.DeniedLoginMethods = append(user.Filters.DeniedLoginMethods, dataprovider.LoginMethodTLSCertificate,
  3354. dataprovider.LoginMethodTLSCertificateAndPwd)
  3355. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  3356. assert.NoError(t, err)
  3357. _, err = getFTPClient(user, true, tlsConfig)
  3358. if assert.Error(t, err) {
  3359. assert.Contains(t, err.Error(), "login method TLSCertificate+password is not allowed")
  3360. }
  3361. // disable FTP protocol
  3362. user.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
  3363. user.Filters.DeniedProtocols = append(user.Filters.DeniedProtocols, common.ProtocolFTP)
  3364. user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
  3365. assert.NoError(t, err)
  3366. _, err = getFTPClient(user, true, tlsConfig)
  3367. if assert.Error(t, err) {
  3368. assert.Contains(t, err.Error(), "protocol FTP is not allowed")
  3369. }
  3370. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3371. assert.NoError(t, err)
  3372. err = os.RemoveAll(user.GetHomeDir())
  3373. assert.NoError(t, err)
  3374. _, err = httpdtest.RemoveUser(user2, http.StatusOK)
  3375. assert.NoError(t, err)
  3376. err = os.RemoveAll(user2.GetHomeDir())
  3377. assert.NoError(t, err)
  3378. _, err = getFTPClient(user, true, tlsConfig)
  3379. assert.Error(t, err)
  3380. }
  3381. func TestClientCertificateAndPwdAuth(t *testing.T) {
  3382. u := getTestUser()
  3383. u.Username = tlsClient1Username
  3384. u.Filters.TLSUsername = sdk.TLSUsernameCN
  3385. u.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword, dataprovider.LoginMethodTLSCertificate}
  3386. user, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3387. assert.NoError(t, err)
  3388. tlsConfig := &tls.Config{
  3389. ServerName: "localhost",
  3390. InsecureSkipVerify: true, // use this for tests only
  3391. MinVersion: tls.VersionTLS12,
  3392. }
  3393. tlsCert, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  3394. assert.NoError(t, err)
  3395. tlsConfig.Certificates = append(tlsConfig.Certificates, tlsCert)
  3396. client, err := getFTPClient(user, true, tlsConfig)
  3397. if assert.NoError(t, err) {
  3398. err = checkBasicFTP(client)
  3399. assert.NoError(t, err)
  3400. err = client.Quit()
  3401. assert.NoError(t, err)
  3402. }
  3403. _, err = getFTPClient(user, true, nil)
  3404. if assert.Error(t, err) {
  3405. assert.Contains(t, err.Error(), "login method password is not allowed")
  3406. }
  3407. user.Password = defaultPassword + "1"
  3408. _, err = getFTPClient(user, true, tlsConfig)
  3409. if assert.Error(t, err) {
  3410. assert.Contains(t, err.Error(), "invalid credentials")
  3411. }
  3412. tlsCert, err = tls.X509KeyPair([]byte(client2Crt), []byte(client2Key))
  3413. assert.NoError(t, err)
  3414. tlsConfig.Certificates = []tls.Certificate{tlsCert}
  3415. _, err = getFTPClient(user, true, tlsConfig)
  3416. if assert.Error(t, err) {
  3417. assert.Contains(t, err.Error(), "bad certificate")
  3418. }
  3419. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3420. assert.NoError(t, err)
  3421. err = os.RemoveAll(user.GetHomeDir())
  3422. assert.NoError(t, err)
  3423. }
  3424. func TestExternalAuthWithClientCert(t *testing.T) {
  3425. if runtime.GOOS == osWindows {
  3426. t.Skip("this test is not available on Windows")
  3427. }
  3428. u := getTestUser()
  3429. u.Username = tlsClient1Username
  3430. u.Filters.DeniedLoginMethods = append(u.Filters.DeniedLoginMethods, dataprovider.LoginMethodPassword)
  3431. u.Filters.TLSUsername = sdk.TLSUsernameCN
  3432. err := dataprovider.Close()
  3433. assert.NoError(t, err)
  3434. err = config.LoadConfig(configDir, "")
  3435. assert.NoError(t, err)
  3436. providerConf := config.GetProviderConf()
  3437. err = os.WriteFile(extAuthPath, getExtAuthScriptContent(u), os.ModePerm)
  3438. assert.NoError(t, err)
  3439. providerConf.ExternalAuthHook = extAuthPath
  3440. providerConf.ExternalAuthScope = 8
  3441. err = dataprovider.Initialize(providerConf, configDir, true)
  3442. assert.NoError(t, err)
  3443. // external auth not called, auth scope is 8
  3444. _, err = getFTPClient(u, true, nil)
  3445. assert.Error(t, err)
  3446. _, _, err = httpdtest.GetUserByUsername(u.Username, http.StatusNotFound)
  3447. assert.NoError(t, err)
  3448. tlsConfig := &tls.Config{
  3449. ServerName: "localhost",
  3450. InsecureSkipVerify: true, // use this for tests only
  3451. MinVersion: tls.VersionTLS12,
  3452. }
  3453. tlsCert, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  3454. assert.NoError(t, err)
  3455. tlsConfig.Certificates = append(tlsConfig.Certificates, tlsCert)
  3456. client, err := getFTPClient(u, true, tlsConfig)
  3457. if assert.NoError(t, err) {
  3458. err = checkBasicFTP(client)
  3459. assert.NoError(t, err)
  3460. err := client.Quit()
  3461. assert.NoError(t, err)
  3462. }
  3463. user, _, err := httpdtest.GetUserByUsername(u.Username, http.StatusOK)
  3464. assert.NoError(t, err)
  3465. assert.Equal(t, u.Username, user.Username)
  3466. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3467. assert.NoError(t, err)
  3468. err = os.RemoveAll(user.GetHomeDir())
  3469. assert.NoError(t, err)
  3470. u.Username = tlsClient2Username
  3471. _, err = getFTPClient(u, true, tlsConfig)
  3472. if assert.Error(t, err) {
  3473. assert.Contains(t, err.Error(), "invalid credentials")
  3474. }
  3475. err = dataprovider.Close()
  3476. assert.NoError(t, err)
  3477. err = config.LoadConfig(configDir, "")
  3478. assert.NoError(t, err)
  3479. providerConf = config.GetProviderConf()
  3480. err = dataprovider.Initialize(providerConf, configDir, true)
  3481. assert.NoError(t, err)
  3482. err = os.Remove(extAuthPath)
  3483. assert.NoError(t, err)
  3484. }
  3485. func TestPreLoginHookWithClientCert(t *testing.T) {
  3486. if runtime.GOOS == osWindows {
  3487. t.Skip("this test is not available on Windows")
  3488. }
  3489. u := getTestUser()
  3490. u.Username = tlsClient1Username
  3491. u.Filters.DeniedLoginMethods = append(u.Filters.DeniedLoginMethods, dataprovider.LoginMethodPassword)
  3492. u.Filters.TLSUsername = sdk.TLSUsernameCN
  3493. err := dataprovider.Close()
  3494. assert.NoError(t, err)
  3495. err = config.LoadConfig(configDir, "")
  3496. assert.NoError(t, err)
  3497. providerConf := config.GetProviderConf()
  3498. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(u, false), os.ModePerm)
  3499. assert.NoError(t, err)
  3500. providerConf.PreLoginHook = preLoginPath
  3501. err = dataprovider.Initialize(providerConf, configDir, true)
  3502. assert.NoError(t, err)
  3503. _, _, err = httpdtest.GetUserByUsername(tlsClient1Username, http.StatusNotFound)
  3504. assert.NoError(t, err)
  3505. tlsConfig := &tls.Config{
  3506. ServerName: "localhost",
  3507. InsecureSkipVerify: true, // use this for tests only
  3508. MinVersion: tls.VersionTLS12,
  3509. }
  3510. tlsCert, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  3511. assert.NoError(t, err)
  3512. tlsConfig.Certificates = append(tlsConfig.Certificates, tlsCert)
  3513. client, err := getFTPClient(u, true, tlsConfig)
  3514. if assert.NoError(t, err) {
  3515. err = checkBasicFTP(client)
  3516. assert.NoError(t, err)
  3517. err := client.Quit()
  3518. assert.NoError(t, err)
  3519. }
  3520. user, _, err := httpdtest.GetUserByUsername(tlsClient1Username, http.StatusOK)
  3521. assert.NoError(t, err)
  3522. // test login with an existing user
  3523. client, err = getFTPClient(user, true, tlsConfig)
  3524. if assert.NoError(t, err) {
  3525. err = checkBasicFTP(client)
  3526. assert.NoError(t, err)
  3527. err := client.Quit()
  3528. assert.NoError(t, err)
  3529. }
  3530. u.Username = tlsClient2Username
  3531. err = os.WriteFile(preLoginPath, getPreLoginScriptContent(u, false), os.ModePerm)
  3532. assert.NoError(t, err)
  3533. _, err = getFTPClient(u, true, tlsConfig)
  3534. if assert.Error(t, err) {
  3535. assert.Contains(t, err.Error(), "does not match username")
  3536. }
  3537. user2, _, err := httpdtest.GetUserByUsername(tlsClient2Username, http.StatusOK)
  3538. assert.NoError(t, err)
  3539. _, err = httpdtest.RemoveUser(user2, http.StatusOK)
  3540. assert.NoError(t, err)
  3541. err = os.RemoveAll(user2.GetHomeDir())
  3542. assert.NoError(t, err)
  3543. _, err = httpdtest.RemoveUser(user, http.StatusOK)
  3544. assert.NoError(t, err)
  3545. err = os.RemoveAll(user.GetHomeDir())
  3546. assert.NoError(t, err)
  3547. err = dataprovider.Close()
  3548. assert.NoError(t, err)
  3549. err = config.LoadConfig(configDir, "")
  3550. assert.NoError(t, err)
  3551. providerConf = config.GetProviderConf()
  3552. err = dataprovider.Initialize(providerConf, configDir, true)
  3553. assert.NoError(t, err)
  3554. err = os.Remove(preLoginPath)
  3555. assert.NoError(t, err)
  3556. }
  3557. func TestNestedVirtualFolders(t *testing.T) {
  3558. u := getTestUser()
  3559. localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3560. assert.NoError(t, err)
  3561. u = getTestSFTPUser()
  3562. mappedPathCrypt := filepath.Join(os.TempDir(), "crypt")
  3563. folderNameCrypt := filepath.Base(mappedPathCrypt)
  3564. vdirCryptPath := "/vdir/crypt"
  3565. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  3566. BaseVirtualFolder: vfs.BaseVirtualFolder{
  3567. Name: folderNameCrypt,
  3568. },
  3569. VirtualPath: vdirCryptPath,
  3570. })
  3571. mappedPath := filepath.Join(os.TempDir(), "local")
  3572. folderName := filepath.Base(mappedPath)
  3573. vdirPath := "/vdir/local"
  3574. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  3575. BaseVirtualFolder: vfs.BaseVirtualFolder{
  3576. Name: folderName,
  3577. },
  3578. VirtualPath: vdirPath,
  3579. })
  3580. mappedPathNested := filepath.Join(os.TempDir(), "nested")
  3581. folderNameNested := filepath.Base(mappedPathNested)
  3582. vdirNestedPath := "/vdir/crypt/nested"
  3583. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  3584. BaseVirtualFolder: vfs.BaseVirtualFolder{
  3585. Name: folderNameNested,
  3586. },
  3587. VirtualPath: vdirNestedPath,
  3588. QuotaFiles: -1,
  3589. QuotaSize: -1,
  3590. })
  3591. f1 := vfs.BaseVirtualFolder{
  3592. Name: folderNameCrypt,
  3593. FsConfig: vfs.Filesystem{
  3594. Provider: sdk.CryptedFilesystemProvider,
  3595. CryptConfig: vfs.CryptFsConfig{
  3596. Passphrase: kms.NewPlainSecret(defaultPassword),
  3597. },
  3598. },
  3599. MappedPath: mappedPathCrypt,
  3600. }
  3601. _, _, err = httpdtest.AddFolder(f1, http.StatusCreated)
  3602. assert.NoError(t, err)
  3603. f2 := vfs.BaseVirtualFolder{
  3604. Name: folderName,
  3605. MappedPath: mappedPath,
  3606. }
  3607. _, _, err = httpdtest.AddFolder(f2, http.StatusCreated)
  3608. assert.NoError(t, err)
  3609. f3 := vfs.BaseVirtualFolder{
  3610. Name: folderNameNested,
  3611. MappedPath: mappedPathNested,
  3612. }
  3613. _, _, err = httpdtest.AddFolder(f3, http.StatusCreated)
  3614. assert.NoError(t, err)
  3615. sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
  3616. assert.NoError(t, err)
  3617. client, err := getFTPClient(sftpUser, false, nil)
  3618. if assert.NoError(t, err) {
  3619. err = checkBasicFTP(client)
  3620. assert.NoError(t, err)
  3621. testFilePath := filepath.Join(homeBasePath, testFileName)
  3622. testFileSize := int64(65535)
  3623. err = createTestFile(testFilePath, testFileSize)
  3624. assert.NoError(t, err)
  3625. localDownloadPath := filepath.Join(homeBasePath, testDLFileName)
  3626. err = ftpUploadFile(testFilePath, testFileName, testFileSize, client, 0)
  3627. assert.NoError(t, err)
  3628. err = ftpDownloadFile(testFileName, localDownloadPath, testFileSize, client, 0)
  3629. assert.NoError(t, err)
  3630. err = ftpUploadFile(testFilePath, path.Join("/vdir", testFileName), testFileSize, client, 0)
  3631. assert.NoError(t, err)
  3632. err = ftpDownloadFile(path.Join("/vdir", testFileName), localDownloadPath, testFileSize, client, 0)
  3633. assert.NoError(t, err)
  3634. err = ftpUploadFile(testFilePath, path.Join(vdirPath, testFileName), testFileSize, client, 0)
  3635. assert.NoError(t, err)
  3636. err = ftpDownloadFile(path.Join(vdirPath, testFileName), localDownloadPath, testFileSize, client, 0)
  3637. assert.NoError(t, err)
  3638. err = ftpUploadFile(testFilePath, path.Join(vdirCryptPath, testFileName), testFileSize, client, 0)
  3639. assert.NoError(t, err)
  3640. err = ftpDownloadFile(path.Join(vdirCryptPath, testFileName), localDownloadPath, testFileSize, client, 0)
  3641. assert.NoError(t, err)
  3642. err = ftpUploadFile(testFilePath, path.Join(vdirNestedPath, testFileName), testFileSize, client, 0)
  3643. assert.NoError(t, err)
  3644. err = ftpDownloadFile(path.Join(vdirNestedPath, testFileName), localDownloadPath, testFileSize, client, 0)
  3645. assert.NoError(t, err)
  3646. err = client.Quit()
  3647. assert.NoError(t, err)
  3648. err = os.Remove(testFilePath)
  3649. assert.NoError(t, err)
  3650. err = os.Remove(localDownloadPath)
  3651. assert.NoError(t, err)
  3652. }
  3653. _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
  3654. assert.NoError(t, err)
  3655. _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
  3656. assert.NoError(t, err)
  3657. _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderNameCrypt}, http.StatusOK)
  3658. assert.NoError(t, err)
  3659. _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
  3660. assert.NoError(t, err)
  3661. _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderNameNested}, http.StatusOK)
  3662. assert.NoError(t, err)
  3663. err = os.RemoveAll(mappedPathCrypt)
  3664. assert.NoError(t, err)
  3665. err = os.RemoveAll(mappedPath)
  3666. assert.NoError(t, err)
  3667. err = os.RemoveAll(mappedPathNested)
  3668. assert.NoError(t, err)
  3669. err = os.RemoveAll(localUser.GetHomeDir())
  3670. assert.NoError(t, err)
  3671. assert.Eventually(t, func() bool { return len(common.Connections.GetStats("")) == 0 }, 1*time.Second, 50*time.Millisecond)
  3672. assert.Eventually(t, func() bool { return common.Connections.GetClientConnections() == 0 }, 1000*time.Millisecond,
  3673. 50*time.Millisecond)
  3674. }
  3675. func checkBasicFTP(client *ftp.ServerConn) error {
  3676. _, err := client.CurrentDir()
  3677. if err != nil {
  3678. return err
  3679. }
  3680. err = client.NoOp()
  3681. if err != nil {
  3682. return err
  3683. }
  3684. _, err = client.List(".")
  3685. if err != nil {
  3686. return err
  3687. }
  3688. return nil
  3689. }
  3690. func ftpUploadFile(localSourcePath string, remoteDestPath string, expectedSize int64, client *ftp.ServerConn, offset uint64) error {
  3691. srcFile, err := os.Open(localSourcePath)
  3692. if err != nil {
  3693. return err
  3694. }
  3695. defer srcFile.Close()
  3696. if offset > 0 {
  3697. err = client.StorFrom(remoteDestPath, srcFile, offset)
  3698. } else {
  3699. err = client.Stor(remoteDestPath, srcFile)
  3700. }
  3701. if err != nil {
  3702. return err
  3703. }
  3704. if expectedSize > 0 {
  3705. size, err := client.FileSize(remoteDestPath)
  3706. if err != nil {
  3707. return err
  3708. }
  3709. if size != expectedSize {
  3710. return fmt.Errorf("uploaded file size does not match, actual: %v, expected: %v", size, expectedSize)
  3711. }
  3712. }
  3713. return nil
  3714. }
  3715. func ftpDownloadFile(remoteSourcePath string, localDestPath string, expectedSize int64, client *ftp.ServerConn, offset uint64) error {
  3716. downloadDest, err := os.Create(localDestPath)
  3717. if err != nil {
  3718. return err
  3719. }
  3720. defer downloadDest.Close()
  3721. var r *ftp.Response
  3722. if offset > 0 {
  3723. r, err = client.RetrFrom(remoteSourcePath, offset)
  3724. } else {
  3725. r, err = client.Retr(remoteSourcePath)
  3726. }
  3727. if err != nil {
  3728. return err
  3729. }
  3730. defer r.Close()
  3731. written, err := io.Copy(downloadDest, r)
  3732. if err != nil {
  3733. return err
  3734. }
  3735. if written != expectedSize {
  3736. return fmt.Errorf("downloaded file size does not match, actual: %v, expected: %v", written, expectedSize)
  3737. }
  3738. return nil
  3739. }
  3740. func getFTPClientImplicitTLS(user dataprovider.User) (*ftp.ServerConn, error) {
  3741. ftpOptions := []ftp.DialOption{ftp.DialWithTimeout(5 * time.Second)}
  3742. tlsConfig := &tls.Config{
  3743. ServerName: "localhost",
  3744. InsecureSkipVerify: true, // use this for tests only
  3745. MinVersion: tls.VersionTLS12,
  3746. }
  3747. ftpOptions = append(ftpOptions, ftp.DialWithTLS(tlsConfig))
  3748. ftpOptions = append(ftpOptions, ftp.DialWithDisabledEPSV(true))
  3749. client, err := ftp.Dial(ftpSrvAddrTLS, ftpOptions...)
  3750. if err != nil {
  3751. return nil, err
  3752. }
  3753. pwd := defaultPassword
  3754. if user.Password != "" {
  3755. pwd = user.Password
  3756. }
  3757. err = client.Login(user.Username, pwd)
  3758. if err != nil {
  3759. return nil, err
  3760. }
  3761. return client, err
  3762. }
  3763. func getFTPClientWithSessionReuse(user dataprovider.User, tlsConfig *tls.Config, dialOptions ...ftp.DialOption,
  3764. ) (*ftp.ServerConn, error) {
  3765. ftpOptions := []ftp.DialOption{ftp.DialWithTimeout(5 * time.Second)}
  3766. ftpOptions = append(ftpOptions, dialOptions...)
  3767. if tlsConfig == nil {
  3768. tlsConfig = &tls.Config{
  3769. ServerName: "localhost",
  3770. InsecureSkipVerify: true, // use this for tests only
  3771. MinVersion: tls.VersionTLS12,
  3772. ClientSessionCache: tls.NewLRUClientSessionCache(0),
  3773. }
  3774. }
  3775. ftpOptions = append(ftpOptions, ftp.DialWithExplicitTLS(tlsConfig))
  3776. client, err := ftp.Dial(ftpSrvAddrTLSResumption, ftpOptions...)
  3777. if err != nil {
  3778. return nil, err
  3779. }
  3780. pwd := defaultPassword
  3781. if user.Password != "" {
  3782. if user.Password == emptyPwdPlaceholder {
  3783. pwd = ""
  3784. } else {
  3785. pwd = user.Password
  3786. }
  3787. }
  3788. err = client.Login(user.Username, pwd)
  3789. if err != nil {
  3790. return nil, err
  3791. }
  3792. return client, err
  3793. }
  3794. func getFTPClient(user dataprovider.User, useTLS bool, tlsConfig *tls.Config, dialOptions ...ftp.DialOption,
  3795. ) (*ftp.ServerConn, error) {
  3796. ftpOptions := []ftp.DialOption{ftp.DialWithTimeout(5 * time.Second)}
  3797. ftpOptions = append(ftpOptions, dialOptions...)
  3798. if useTLS {
  3799. if tlsConfig == nil {
  3800. tlsConfig = &tls.Config{
  3801. ServerName: "localhost",
  3802. InsecureSkipVerify: true, // use this for tests only
  3803. MinVersion: tls.VersionTLS12,
  3804. }
  3805. }
  3806. ftpOptions = append(ftpOptions, ftp.DialWithExplicitTLS(tlsConfig))
  3807. }
  3808. client, err := ftp.Dial(ftpServerAddr, ftpOptions...)
  3809. if err != nil {
  3810. return nil, err
  3811. }
  3812. pwd := defaultPassword
  3813. if user.Password != "" {
  3814. if user.Password == emptyPwdPlaceholder {
  3815. pwd = ""
  3816. } else {
  3817. pwd = user.Password
  3818. }
  3819. }
  3820. err = client.Login(user.Username, pwd)
  3821. if err != nil {
  3822. return nil, err
  3823. }
  3824. return client, err
  3825. }
  3826. func waitTCPListening(address string) {
  3827. for {
  3828. conn, err := net.Dial("tcp", address)
  3829. if err != nil {
  3830. logger.WarnToConsole("tcp server %v not listening: %v", address, err)
  3831. time.Sleep(100 * time.Millisecond)
  3832. continue
  3833. }
  3834. logger.InfoToConsole("tcp server %v now listening", address)
  3835. conn.Close()
  3836. break
  3837. }
  3838. }
  3839. func waitNoConnections() {
  3840. time.Sleep(50 * time.Millisecond)
  3841. for len(common.Connections.GetStats("")) > 0 {
  3842. time.Sleep(50 * time.Millisecond)
  3843. }
  3844. }
  3845. func getTestGroup() dataprovider.Group {
  3846. return dataprovider.Group{
  3847. BaseGroup: sdk.BaseGroup{
  3848. Name: "test_group",
  3849. Description: "test group description",
  3850. },
  3851. }
  3852. }
  3853. func getTestUser() dataprovider.User {
  3854. user := dataprovider.User{
  3855. BaseUser: sdk.BaseUser{
  3856. Username: defaultUsername,
  3857. Password: defaultPassword,
  3858. HomeDir: filepath.Join(homeBasePath, defaultUsername),
  3859. Status: 1,
  3860. ExpirationDate: 0,
  3861. },
  3862. }
  3863. user.Permissions = make(map[string][]string)
  3864. user.Permissions["/"] = allPerms
  3865. return user
  3866. }
  3867. func getTestSFTPUser() dataprovider.User {
  3868. u := getTestUser()
  3869. u.Username = u.Username + "_sftp"
  3870. u.FsConfig.Provider = sdk.SFTPFilesystemProvider
  3871. u.FsConfig.SFTPConfig.Endpoint = sftpServerAddr
  3872. u.FsConfig.SFTPConfig.Username = defaultUsername
  3873. u.FsConfig.SFTPConfig.Password = kms.NewPlainSecret(defaultPassword)
  3874. return u
  3875. }
  3876. func getTestUserWithHTTPFs() dataprovider.User {
  3877. u := getTestUser()
  3878. u.FsConfig.Provider = sdk.HTTPFilesystemProvider
  3879. u.FsConfig.HTTPConfig = vfs.HTTPFsConfig{
  3880. BaseHTTPFsConfig: sdk.BaseHTTPFsConfig{
  3881. Endpoint: fmt.Sprintf("http://127.0.0.1:%d/api/v1", httpFsPort),
  3882. Username: defaultHTTPFsUsername,
  3883. },
  3884. }
  3885. return u
  3886. }
  3887. func getExtAuthScriptContent(user dataprovider.User) []byte {
  3888. extAuthContent := []byte("#!/bin/sh\n\n")
  3889. extAuthContent = append(extAuthContent, []byte(fmt.Sprintf("if test \"$SFTPGO_AUTHD_USERNAME\" = \"%v\"; then\n", user.Username))...)
  3890. u, _ := json.Marshal(user)
  3891. extAuthContent = append(extAuthContent, []byte(fmt.Sprintf("echo '%v'\n", string(u)))...)
  3892. extAuthContent = append(extAuthContent, []byte("else\n")...)
  3893. extAuthContent = append(extAuthContent, []byte("echo '{\"username\":\"\"}'\n")...)
  3894. extAuthContent = append(extAuthContent, []byte("fi\n")...)
  3895. return extAuthContent
  3896. }
  3897. func getPreLoginScriptContent(user dataprovider.User, nonJSONResponse bool) []byte {
  3898. content := []byte("#!/bin/sh\n\n")
  3899. if nonJSONResponse {
  3900. content = append(content, []byte("echo 'text response'\n")...)
  3901. return content
  3902. }
  3903. if len(user.Username) > 0 {
  3904. u, _ := json.Marshal(user)
  3905. content = append(content, []byte(fmt.Sprintf("echo '%v'\n", string(u)))...)
  3906. }
  3907. return content
  3908. }
  3909. func getExitCodeScriptContent(exitCode int) []byte {
  3910. content := []byte("#!/bin/sh\n\n")
  3911. content = append(content, []byte(fmt.Sprintf("exit %v", exitCode))...)
  3912. return content
  3913. }
  3914. func createTestFile(path string, size int64) error {
  3915. baseDir := filepath.Dir(path)
  3916. if _, err := os.Stat(baseDir); errors.Is(err, fs.ErrNotExist) {
  3917. err = os.MkdirAll(baseDir, os.ModePerm)
  3918. if err != nil {
  3919. return err
  3920. }
  3921. }
  3922. content := make([]byte, size)
  3923. _, err := rand.Read(content)
  3924. if err != nil {
  3925. return err
  3926. }
  3927. return os.WriteFile(path, content, os.ModePerm)
  3928. }
  3929. func writeCerts(certPath, keyPath, caCrtPath, caCRLPath string) error {
  3930. err := os.WriteFile(certPath, []byte(ftpsCert), os.ModePerm)
  3931. if err != nil {
  3932. logger.ErrorToConsole("error writing FTPS certificate: %v", err)
  3933. return err
  3934. }
  3935. err = os.WriteFile(keyPath, []byte(ftpsKey), os.ModePerm)
  3936. if err != nil {
  3937. logger.ErrorToConsole("error writing FTPS private key: %v", err)
  3938. return err
  3939. }
  3940. err = os.WriteFile(caCrtPath, []byte(caCRT), os.ModePerm)
  3941. if err != nil {
  3942. logger.ErrorToConsole("error writing FTPS CA crt: %v", err)
  3943. return err
  3944. }
  3945. err = os.WriteFile(caCRLPath, []byte(caCRL), os.ModePerm)
  3946. if err != nil {
  3947. logger.ErrorToConsole("error writing FTPS CRL: %v", err)
  3948. return err
  3949. }
  3950. return nil
  3951. }
  3952. func generateTOTPPasscode(secret string, algo otp.Algorithm) (string, error) {
  3953. return totp.GenerateCodeCustom(secret, time.Now(), totp.ValidateOpts{
  3954. Period: 30,
  3955. Skew: 1,
  3956. Digits: otp.DigitsSix,
  3957. Algorithm: algo,
  3958. })
  3959. }
  3960. func startHTTPFs() {
  3961. go func() {
  3962. if err := httpdtest.StartTestHTTPFs(httpFsPort, nil); err != nil {
  3963. logger.ErrorToConsole("could not start HTTPfs test server: %v", err)
  3964. os.Exit(1)
  3965. }
  3966. }()
  3967. waitTCPListening(fmt.Sprintf(":%d", httpFsPort))
  3968. }