123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359736073617362736373647365736673677368736973707371737273737374737573767377737873797380738173827383738473857386738773887389739073917392739373947395739673977398739974007401740274037404740574067407740874097410741174127413741474157416741774187419742074217422742374247425742674277428742974307431743274337434743574367437743874397440744174427443744474457446744774487449745074517452745374547455745674577458745974607461746274637464746574667467746874697470747174727473747474757476747774787479748074817482748374847485748674877488748974907491749274937494749574967497749874997500750175027503750475057506750775087509751075117512751375147515751675177518751975207521752275237524752575267527752875297530753175327533753475357536753775387539754075417542754375447545754675477548754975507551755275537554755575567557755875597560756175627563756475657566756775687569757075717572757375747575757675777578757975807581758275837584758575867587758875897590759175927593759475957596759775987599760076017602760376047605760676077608760976107611761276137614761576167617761876197620762176227623762476257626762776287629763076317632763376347635763676377638763976407641764276437644764576467647764876497650765176527653765476557656765776587659766076617662766376647665766676677668766976707671767276737674767576767677767876797680768176827683768476857686768776887689769076917692769376947695769676977698769977007701770277037704770577067707770877097710771177127713771477157716771777187719772077217722772377247725772677277728772977307731773277337734773577367737773877397740774177427743774477457746774777487749775077517752775377547755775677577758775977607761776277637764776577667767776877697770777177727773777477757776777777787779778077817782778377847785778677877788778977907791779277937794779577967797779877997800780178027803780478057806780778087809781078117812781378147815781678177818781978207821782278237824782578267827782878297830783178327833783478357836783778387839784078417842784378447845784678477848784978507851785278537854785578567857785878597860786178627863786478657866786778687869787078717872787378747875787678777878787978807881788278837884788578867887788878897890789178927893789478957896789778987899790079017902790379047905790679077908790979107911791279137914791579167917791879197920792179227923792479257926792779287929793079317932793379347935793679377938793979407941794279437944794579467947794879497950795179527953795479557956795779587959796079617962796379647965796679677968796979707971797279737974797579767977797879797980798179827983798479857986798779887989799079917992799379947995799679977998799980008001800280038004800580068007800880098010801180128013801480158016801780188019802080218022802380248025802680278028802980308031803280338034803580368037803880398040804180428043804480458046804780488049805080518052805380548055805680578058805980608061806280638064806580668067806880698070807180728073807480758076807780788079808080818082808380848085808680878088808980908091809280938094809580968097809880998100810181028103810481058106810781088109811081118112811381148115811681178118811981208121812281238124812581268127812881298130813181328133813481358136813781388139814081418142814381448145814681478148814981508151815281538154815581568157815881598160816181628163816481658166816781688169817081718172817381748175817681778178817981808181818281838184818581868187818881898190819181928193819481958196819781988199820082018202820382048205820682078208820982108211821282138214821582168217821882198220822182228223822482258226822782288229823082318232823382348235823682378238823982408241824282438244824582468247824882498250825182528253825482558256825782588259826082618262826382648265826682678268826982708271827282738274827582768277827882798280828182828283828482858286828782888289829082918292829382948295829682978298829983008301830283038304830583068307830883098310831183128313831483158316831783188319832083218322832383248325832683278328832983308331833283338334833583368337833883398340834183428343834483458346834783488349835083518352835383548355835683578358835983608361836283638364836583668367836883698370837183728373837483758376837783788379838083818382838383848385838683878388838983908391839283938394839583968397839883998400840184028403840484058406840784088409841084118412841384148415841684178418841984208421842284238424842584268427842884298430843184328433843484358436843784388439844084418442844384448445844684478448844984508451845284538454845584568457845884598460846184628463846484658466846784688469847084718472847384748475847684778478847984808481848284838484848584868487848884898490849184928493849484958496849784988499850085018502850385048505850685078508850985108511851285138514851585168517851885198520852185228523852485258526852785288529853085318532853385348535853685378538853985408541854285438544854585468547854885498550855185528553855485558556855785588559856085618562856385648565856685678568856985708571857285738574857585768577857885798580858185828583858485858586858785888589859085918592859385948595859685978598859986008601860286038604860586068607860886098610861186128613861486158616861786188619862086218622862386248625862686278628862986308631863286338634863586368637863886398640864186428643864486458646864786488649865086518652865386548655865686578658865986608661866286638664866586668667866886698670867186728673867486758676867786788679868086818682868386848685868686878688868986908691869286938694869586968697869886998700870187028703870487058706870787088709871087118712871387148715871687178718871987208721872287238724872587268727872887298730873187328733873487358736873787388739874087418742874387448745874687478748874987508751875287538754875587568757875887598760876187628763876487658766876787688769877087718772877387748775877687778778877987808781878287838784878587868787878887898790879187928793879487958796879787988799880088018802880388048805880688078808880988108811881288138814881588168817881888198820882188228823882488258826882788288829883088318832883388348835883688378838883988408841884288438844884588468847884888498850885188528853885488558856885788588859886088618862886388648865886688678868886988708871887288738874887588768877887888798880888188828883888488858886888788888889889088918892889388948895889688978898889989008901890289038904890589068907890889098910891189128913891489158916891789188919892089218922892389248925892689278928892989308931893289338934893589368937893889398940894189428943894489458946894789488949895089518952895389548955895689578958895989608961896289638964896589668967896889698970897189728973897489758976897789788979898089818982898389848985898689878988898989908991899289938994899589968997899889999000900190029003900490059006900790089009901090119012901390149015901690179018901990209021902290239024902590269027902890299030903190329033903490359036903790389039904090419042904390449045904690479048904990509051905290539054905590569057905890599060906190629063906490659066906790689069907090719072907390749075907690779078907990809081908290839084908590869087908890899090909190929093909490959096909790989099910091019102910391049105910691079108910991109111911291139114911591169117911891199120912191229123912491259126912791289129913091319132913391349135913691379138913991409141914291439144914591469147914891499150915191529153915491559156915791589159916091619162916391649165916691679168916991709171917291739174917591769177917891799180918191829183918491859186918791889189919091919192919391949195919691979198919992009201920292039204920592069207920892099210921192129213921492159216921792189219922092219222922392249225922692279228922992309231923292339234923592369237923892399240924192429243924492459246924792489249925092519252925392549255925692579258925992609261926292639264926592669267926892699270927192729273927492759276927792789279928092819282928392849285928692879288928992909291929292939294929592969297929892999300930193029303930493059306930793089309931093119312931393149315931693179318931993209321932293239324932593269327932893299330933193329333933493359336933793389339934093419342934393449345934693479348934993509351935293539354935593569357935893599360936193629363936493659366936793689369937093719372937393749375937693779378937993809381938293839384938593869387938893899390939193929393939493959396939793989399940094019402940394049405940694079408940994109411941294139414941594169417941894199420942194229423942494259426942794289429943094319432943394349435943694379438943994409441944294439444944594469447944894499450945194529453945494559456945794589459946094619462946394649465946694679468946994709471947294739474947594769477947894799480948194829483948494859486948794889489949094919492949394949495949694979498949995009501950295039504950595069507950895099510951195129513951495159516951795189519952095219522952395249525952695279528952995309531953295339534953595369537953895399540954195429543954495459546954795489549955095519552955395549555955695579558955995609561956295639564956595669567956895699570957195729573957495759576957795789579958095819582958395849585958695879588958995909591959295939594959595969597959895999600960196029603960496059606960796089609961096119612961396149615961696179618961996209621962296239624962596269627962896299630963196329633963496359636963796389639964096419642964396449645964696479648964996509651965296539654965596569657965896599660966196629663966496659666966796689669967096719672967396749675967696779678967996809681968296839684968596869687968896899690969196929693969496959696969796989699970097019702970397049705970697079708970997109711971297139714971597169717971897199720972197229723972497259726972797289729973097319732973397349735973697379738973997409741974297439744974597469747974897499750975197529753975497559756975797589759976097619762976397649765976697679768976997709771977297739774977597769777977897799780978197829783978497859786978797889789979097919792979397949795979697979798979998009801980298039804980598069807980898099810981198129813981498159816981798189819982098219822982398249825982698279828982998309831983298339834983598369837983898399840984198429843984498459846984798489849985098519852985398549855985698579858985998609861986298639864986598669867986898699870987198729873987498759876987798789879988098819882988398849885988698879888988998909891989298939894989598969897989898999900990199029903990499059906990799089909991099119912991399149915991699179918991999209921992299239924992599269927992899299930993199329933993499359936993799389939994099419942994399449945994699479948994999509951995299539954995599569957995899599960996199629963996499659966996799689969997099719972997399749975997699779978997999809981998299839984998599869987998899899990999199929993999499959996999799989999100001000110002100031000410005100061000710008100091001010011100121001310014100151001610017100181001910020100211002210023100241002510026100271002810029100301003110032100331003410035100361003710038100391004010041100421004310044100451004610047100481004910050100511005210053100541005510056100571005810059100601006110062100631006410065100661006710068100691007010071100721007310074100751007610077100781007910080100811008210083100841008510086100871008810089100901009110092100931009410095100961009710098100991010010101101021010310104101051010610107101081010910110101111011210113101141011510116101171011810119101201012110122101231012410125101261012710128101291013010131101321013310134101351013610137101381013910140101411014210143101441014510146101471014810149101501015110152101531015410155101561015710158101591016010161101621016310164101651016610167101681016910170101711017210173101741017510176101771017810179101801018110182101831018410185101861018710188101891019010191101921019310194101951019610197101981019910200102011020210203102041020510206102071020810209102101021110212102131021410215102161021710218102191022010221102221022310224102251022610227102281022910230102311023210233102341023510236102371023810239102401024110242102431024410245102461024710248102491025010251102521025310254102551025610257102581025910260102611026210263102641026510266102671026810269102701027110272102731027410275102761027710278102791028010281102821028310284102851028610287102881028910290102911029210293102941029510296102971029810299103001030110302103031030410305103061030710308103091031010311103121031310314103151031610317103181031910320103211032210323103241032510326103271032810329103301033110332103331033410335103361033710338103391034010341103421034310344103451034610347103481034910350103511035210353103541035510356103571035810359103601036110362103631036410365103661036710368103691037010371103721037310374103751037610377103781037910380103811038210383103841038510386103871038810389103901039110392103931039410395103961039710398103991040010401104021040310404104051040610407104081040910410104111041210413104141041510416104171041810419104201042110422104231042410425104261042710428104291043010431104321043310434104351043610437104381043910440104411044210443104441044510446104471044810449104501045110452104531045410455104561045710458104591046010461104621046310464104651046610467104681046910470104711047210473104741047510476104771047810479104801048110482104831048410485104861048710488104891049010491104921049310494104951049610497104981049910500105011050210503105041050510506105071050810509105101051110512105131051410515105161051710518105191052010521105221052310524105251052610527105281052910530105311053210533105341053510536105371053810539105401054110542105431054410545105461054710548105491055010551105521055310554105551055610557105581055910560105611056210563105641056510566105671056810569105701057110572105731057410575105761057710578105791058010581105821058310584105851058610587105881058910590105911059210593105941059510596105971059810599106001060110602106031060410605106061060710608106091061010611106121061310614106151061610617106181061910620106211062210623106241062510626106271062810629106301063110632106331063410635106361063710638106391064010641106421064310644106451064610647106481064910650106511065210653106541065510656106571065810659106601066110662106631066410665106661066710668106691067010671106721067310674106751067610677106781067910680106811068210683106841068510686106871068810689106901069110692106931069410695106961069710698106991070010701107021070310704107051070610707107081070910710107111071210713107141071510716107171071810719107201072110722107231072410725107261072710728107291073010731107321073310734107351073610737107381073910740107411074210743107441074510746107471074810749107501075110752107531075410755107561075710758107591076010761107621076310764107651076610767107681076910770107711077210773107741077510776107771077810779107801078110782107831078410785107861078710788107891079010791107921079310794107951079610797107981079910800108011080210803108041080510806108071080810809108101081110812108131081410815108161081710818108191082010821108221082310824108251082610827108281082910830108311083210833108341083510836108371083810839108401084110842108431084410845108461084710848108491085010851108521085310854108551085610857108581085910860108611086210863108641086510866108671086810869108701087110872108731087410875108761087710878108791088010881108821088310884108851088610887108881088910890108911089210893108941089510896108971089810899109001090110902109031090410905109061090710908109091091010911109121091310914109151091610917109181091910920109211092210923109241092510926109271092810929109301093110932109331093410935109361093710938109391094010941109421094310944109451094610947109481094910950109511095210953109541095510956109571095810959109601096110962109631096410965109661096710968109691097010971109721097310974109751097610977109781097910980109811098210983109841098510986109871098810989109901099110992109931099410995109961099710998109991100011001110021100311004110051100611007110081100911010110111101211013110141101511016110171101811019110201102111022110231102411025110261102711028110291103011031110321103311034110351103611037110381103911040110411104211043110441104511046110471104811049110501105111052110531105411055110561105711058110591106011061110621106311064110651106611067110681106911070110711107211073110741107511076110771107811079110801108111082110831108411085110861108711088110891109011091110921109311094110951109611097110981109911100111011110211103111041110511106111071110811109111101111111112111131111411115111161111711118111191112011121111221112311124111251112611127111281112911130111311113211133111341113511136111371113811139111401114111142111431114411145111461114711148111491115011151111521115311154111551115611157111581115911160111611116211163111641116511166111671116811169111701117111172111731117411175111761117711178111791118011181111821118311184111851118611187111881118911190111911119211193111941119511196111971119811199112001120111202112031120411205112061120711208112091121011211112121121311214112151121611217112181121911220112211122211223112241122511226112271122811229112301123111232112331123411235112361123711238112391124011241112421124311244112451124611247112481124911250112511125211253112541125511256112571125811259112601126111262112631126411265112661126711268112691127011271112721127311274112751127611277112781127911280112811128211283112841128511286112871128811289112901129111292112931129411295112961129711298112991130011301113021130311304113051130611307113081130911310113111131211313113141131511316113171131811319113201132111322113231132411325113261132711328113291133011331113321133311334113351133611337113381133911340113411134211343113441134511346113471134811349113501135111352113531135411355113561135711358113591136011361113621136311364113651136611367113681136911370113711137211373113741137511376113771137811379113801138111382113831138411385113861138711388113891139011391113921139311394113951139611397113981139911400114011140211403114041140511406114071140811409114101141111412114131141411415114161141711418114191142011421114221142311424114251142611427114281142911430114311143211433114341143511436114371143811439114401144111442114431144411445114461144711448114491145011451114521145311454114551145611457114581145911460114611146211463114641146511466114671146811469114701147111472114731147411475114761147711478114791148011481114821148311484114851148611487114881148911490114911149211493114941149511496114971149811499115001150111502115031150411505115061150711508115091151011511115121151311514115151151611517115181151911520115211152211523115241152511526115271152811529115301153111532115331153411535115361153711538115391154011541115421154311544115451154611547115481154911550115511155211553115541155511556115571155811559115601156111562115631156411565115661156711568115691157011571115721157311574115751157611577115781157911580115811158211583115841158511586115871158811589115901159111592115931159411595115961159711598115991160011601116021160311604116051160611607116081160911610116111161211613116141161511616116171161811619116201162111622116231162411625116261162711628116291163011631116321163311634116351163611637116381163911640116411164211643116441164511646116471164811649116501165111652116531165411655116561165711658116591166011661116621166311664116651166611667116681166911670116711167211673116741167511676116771167811679116801168111682116831168411685116861168711688116891169011691116921169311694116951169611697116981169911700117011170211703117041170511706117071170811709117101171111712117131171411715117161171711718117191172011721117221172311724117251172611727117281172911730117311173211733117341173511736117371173811739117401174111742117431174411745117461174711748117491175011751117521175311754117551175611757117581175911760117611176211763117641176511766117671176811769117701177111772117731177411775117761177711778117791178011781117821178311784117851178611787117881178911790117911179211793117941179511796117971179811799118001180111802118031180411805118061180711808118091181011811118121181311814118151181611817118181181911820118211182211823118241182511826118271182811829118301183111832118331183411835118361183711838118391184011841118421184311844118451184611847118481184911850118511185211853118541185511856118571185811859118601186111862118631186411865118661186711868118691187011871118721187311874118751187611877118781187911880118811188211883118841188511886118871188811889118901189111892118931189411895118961189711898118991190011901119021190311904119051190611907119081190911910119111191211913119141191511916119171191811919119201192111922119231192411925119261192711928119291193011931119321193311934119351193611937119381193911940119411194211943119441194511946119471194811949119501195111952119531195411955119561195711958119591196011961119621196311964119651196611967119681196911970119711197211973119741197511976119771197811979119801198111982119831198411985119861198711988119891199011991119921199311994119951199611997119981199912000120011200212003120041200512006120071200812009120101201112012120131201412015120161201712018120191202012021120221202312024120251202612027120281202912030120311203212033120341203512036120371203812039120401204112042120431204412045120461204712048120491205012051120521205312054120551205612057120581205912060120611206212063120641206512066120671206812069120701207112072120731207412075120761207712078120791208012081120821208312084120851208612087120881208912090120911209212093120941209512096120971209812099121001210112102121031210412105121061210712108121091211012111121121211312114121151211612117121181211912120121211212212123121241212512126121271212812129121301213112132121331213412135121361213712138121391214012141121421214312144121451214612147121481214912150121511215212153121541215512156121571215812159121601216112162121631216412165121661216712168121691217012171121721217312174121751217612177121781217912180121811218212183121841218512186121871218812189121901219112192121931219412195121961219712198121991220012201122021220312204122051220612207122081220912210122111221212213122141221512216122171221812219122201222112222122231222412225122261222712228122291223012231122321223312234122351223612237122381223912240122411224212243122441224512246122471224812249122501225112252122531225412255122561225712258122591226012261122621226312264122651226612267122681226912270122711227212273122741227512276122771227812279122801228112282122831228412285122861228712288122891229012291122921229312294122951229612297122981229912300123011230212303123041230512306123071230812309123101231112312123131231412315123161231712318123191232012321123221232312324123251232612327123281232912330123311233212333123341233512336123371233812339123401234112342123431234412345123461234712348123491235012351123521235312354123551235612357123581235912360123611236212363123641236512366123671236812369123701237112372123731237412375123761237712378123791238012381123821238312384123851238612387123881238912390123911239212393123941239512396123971239812399124001240112402124031240412405124061240712408124091241012411124121241312414124151241612417124181241912420124211242212423124241242512426124271242812429124301243112432124331243412435124361243712438124391244012441124421244312444124451244612447124481244912450124511245212453124541245512456124571245812459124601246112462124631246412465124661246712468124691247012471124721247312474124751247612477124781247912480124811248212483124841248512486124871248812489124901249112492124931249412495124961249712498124991250012501125021250312504125051250612507125081250912510125111251212513125141251512516125171251812519125201252112522125231252412525125261252712528125291253012531125321253312534125351253612537125381253912540125411254212543125441254512546125471254812549125501255112552125531255412555125561255712558125591256012561125621256312564125651256612567125681256912570125711257212573125741257512576125771257812579125801258112582125831258412585125861258712588125891259012591125921259312594125951259612597125981259912600126011260212603126041260512606126071260812609126101261112612126131261412615126161261712618126191262012621126221262312624126251262612627126281262912630126311263212633126341263512636126371263812639126401264112642126431264412645126461264712648126491265012651126521265312654126551265612657126581265912660126611266212663126641266512666126671266812669126701267112672126731267412675126761267712678126791268012681126821268312684126851268612687126881268912690126911269212693126941269512696126971269812699127001270112702127031270412705127061270712708127091271012711127121271312714127151271612717127181271912720127211272212723127241272512726127271272812729127301273112732127331273412735127361273712738127391274012741127421274312744127451274612747127481274912750127511275212753127541275512756127571275812759127601276112762127631276412765127661276712768127691277012771127721277312774127751277612777127781277912780127811278212783127841278512786127871278812789127901279112792127931279412795127961279712798127991280012801128021280312804128051280612807128081280912810128111281212813128141281512816128171281812819128201282112822128231282412825128261282712828128291283012831128321283312834128351283612837128381283912840128411284212843128441284512846128471284812849128501285112852128531285412855128561285712858128591286012861128621286312864128651286612867128681286912870128711287212873128741287512876128771287812879128801288112882128831288412885128861288712888128891289012891128921289312894128951289612897128981289912900129011290212903129041290512906129071290812909129101291112912129131291412915129161291712918129191292012921129221292312924129251292612927129281292912930129311293212933129341293512936129371293812939129401294112942129431294412945129461294712948129491295012951129521295312954129551295612957129581295912960129611296212963129641296512966129671296812969129701297112972129731297412975129761297712978129791298012981129821298312984129851298612987129881298912990129911299212993129941299512996129971299812999130001300113002130031300413005130061300713008130091301013011130121301313014130151301613017130181301913020130211302213023130241302513026130271302813029130301303113032130331303413035130361303713038130391304013041130421304313044130451304613047130481304913050130511305213053130541305513056130571305813059130601306113062130631306413065130661306713068130691307013071130721307313074130751307613077130781307913080130811308213083130841308513086130871308813089130901309113092130931309413095130961309713098130991310013101131021310313104131051310613107131081310913110131111311213113131141311513116131171311813119131201312113122131231312413125131261312713128131291313013131131321313313134131351313613137131381313913140131411314213143131441314513146131471314813149131501315113152131531315413155131561315713158131591316013161131621316313164131651316613167131681316913170131711317213173131741317513176131771317813179131801318113182131831318413185131861318713188131891319013191131921319313194131951319613197131981319913200132011320213203132041320513206132071320813209132101321113212132131321413215132161321713218132191322013221132221322313224132251322613227132281322913230132311323213233132341323513236132371323813239132401324113242132431324413245132461324713248132491325013251132521325313254132551325613257132581325913260132611326213263132641326513266132671326813269132701327113272132731327413275132761327713278132791328013281132821328313284132851328613287132881328913290132911329213293132941329513296132971329813299133001330113302133031330413305133061330713308133091331013311133121331313314133151331613317133181331913320133211332213323133241332513326133271332813329133301333113332133331333413335133361333713338133391334013341133421334313344133451334613347133481334913350133511335213353133541335513356133571335813359133601336113362133631336413365133661336713368133691337013371133721337313374133751337613377133781337913380133811338213383133841338513386133871338813389133901339113392133931339413395133961339713398133991340013401134021340313404134051340613407134081340913410134111341213413134141341513416134171341813419134201342113422134231342413425134261342713428134291343013431134321343313434134351343613437134381343913440134411344213443134441344513446134471344813449134501345113452134531345413455134561345713458134591346013461134621346313464134651346613467134681346913470134711347213473134741347513476134771347813479134801348113482134831348413485134861348713488134891349013491134921349313494134951349613497134981349913500135011350213503135041350513506135071350813509135101351113512135131351413515135161351713518135191352013521135221352313524135251352613527135281352913530135311353213533135341353513536135371353813539135401354113542135431354413545135461354713548135491355013551135521355313554135551355613557135581355913560135611356213563135641356513566135671356813569135701357113572135731357413575135761357713578135791358013581135821358313584135851358613587135881358913590135911359213593135941359513596135971359813599136001360113602136031360413605136061360713608136091361013611136121361313614136151361613617136181361913620136211362213623136241362513626136271362813629136301363113632136331363413635136361363713638136391364013641136421364313644136451364613647136481364913650136511365213653136541365513656136571365813659136601366113662136631366413665136661366713668136691367013671136721367313674136751367613677136781367913680136811368213683136841368513686136871368813689136901369113692136931369413695136961369713698136991370013701137021370313704137051370613707137081370913710137111371213713137141371513716137171371813719137201372113722137231372413725137261372713728137291373013731137321373313734137351373613737137381373913740137411374213743137441374513746137471374813749137501375113752137531375413755137561375713758137591376013761137621376313764137651376613767137681376913770137711377213773137741377513776137771377813779137801378113782137831378413785137861378713788137891379013791137921379313794137951379613797137981379913800138011380213803138041380513806138071380813809138101381113812138131381413815138161381713818138191382013821138221382313824138251382613827138281382913830138311383213833138341383513836138371383813839138401384113842138431384413845138461384713848138491385013851138521385313854138551385613857138581385913860138611386213863138641386513866138671386813869138701387113872138731387413875138761387713878138791388013881138821388313884138851388613887138881388913890138911389213893138941389513896138971389813899139001390113902139031390413905139061390713908139091391013911139121391313914139151391613917139181391913920139211392213923139241392513926139271392813929139301393113932139331393413935139361393713938139391394013941139421394313944139451394613947139481394913950139511395213953139541395513956139571395813959139601396113962139631396413965139661396713968139691397013971139721397313974139751397613977139781397913980139811398213983139841398513986139871398813989139901399113992139931399413995139961399713998139991400014001140021400314004140051400614007140081400914010140111401214013140141401514016140171401814019140201402114022140231402414025140261402714028140291403014031140321403314034140351403614037140381403914040140411404214043140441404514046140471404814049140501405114052140531405414055140561405714058140591406014061140621406314064140651406614067140681406914070140711407214073140741407514076140771407814079140801408114082140831408414085140861408714088140891409014091140921409314094140951409614097140981409914100141011410214103141041410514106141071410814109141101411114112141131411414115141161411714118141191412014121141221412314124141251412614127141281412914130141311413214133141341413514136141371413814139141401414114142141431414414145141461414714148141491415014151141521415314154141551415614157141581415914160141611416214163141641416514166141671416814169141701417114172141731417414175141761417714178141791418014181141821418314184141851418614187141881418914190141911419214193141941419514196141971419814199142001420114202142031420414205142061420714208142091421014211142121421314214142151421614217142181421914220142211422214223142241422514226142271422814229142301423114232142331423414235142361423714238142391424014241142421424314244142451424614247142481424914250142511425214253142541425514256142571425814259142601426114262142631426414265142661426714268142691427014271142721427314274142751427614277142781427914280142811428214283142841428514286142871428814289142901429114292142931429414295142961429714298142991430014301143021430314304143051430614307143081430914310143111431214313143141431514316143171431814319143201432114322143231432414325143261432714328143291433014331143321433314334143351433614337143381433914340143411434214343143441434514346143471434814349143501435114352143531435414355143561435714358143591436014361143621436314364143651436614367143681436914370143711437214373143741437514376143771437814379143801438114382143831438414385143861438714388143891439014391143921439314394143951439614397143981439914400144011440214403144041440514406144071440814409144101441114412144131441414415144161441714418144191442014421144221442314424144251442614427144281442914430144311443214433144341443514436144371443814439144401444114442144431444414445144461444714448144491445014451144521445314454144551445614457144581445914460144611446214463144641446514466144671446814469144701447114472144731447414475144761447714478144791448014481144821448314484144851448614487144881448914490144911449214493144941449514496144971449814499145001450114502145031450414505145061450714508145091451014511145121451314514145151451614517145181451914520145211452214523145241452514526145271452814529145301453114532145331453414535145361453714538145391454014541145421454314544145451454614547145481454914550145511455214553145541455514556145571455814559145601456114562145631456414565145661456714568145691457014571145721457314574145751457614577145781457914580145811458214583145841458514586145871458814589145901459114592145931459414595145961459714598145991460014601146021460314604146051460614607146081460914610146111461214613146141461514616146171461814619146201462114622146231462414625146261462714628146291463014631146321463314634146351463614637146381463914640146411464214643146441464514646146471464814649146501465114652146531465414655146561465714658146591466014661146621466314664146651466614667146681466914670146711467214673146741467514676146771467814679146801468114682146831468414685146861468714688146891469014691146921469314694146951469614697146981469914700147011470214703147041470514706147071470814709147101471114712147131471414715147161471714718147191472014721147221472314724147251472614727147281472914730147311473214733147341473514736147371473814739147401474114742147431474414745147461474714748147491475014751147521475314754147551475614757147581475914760147611476214763147641476514766147671476814769147701477114772147731477414775147761477714778147791478014781147821478314784147851478614787147881478914790147911479214793147941479514796147971479814799148001480114802148031480414805148061480714808148091481014811148121481314814148151481614817148181481914820148211482214823148241482514826148271482814829148301483114832148331483414835148361483714838148391484014841148421484314844148451484614847148481484914850148511485214853148541485514856148571485814859148601486114862148631486414865148661486714868148691487014871148721487314874148751487614877148781487914880148811488214883148841488514886148871488814889148901489114892148931489414895148961489714898148991490014901149021490314904149051490614907149081490914910149111491214913149141491514916149171491814919149201492114922149231492414925149261492714928149291493014931149321493314934149351493614937149381493914940149411494214943149441494514946149471494814949149501495114952149531495414955149561495714958149591496014961149621496314964149651496614967149681496914970149711497214973149741497514976149771497814979149801498114982149831498414985149861498714988149891499014991149921499314994149951499614997149981499915000150011500215003150041500515006150071500815009150101501115012150131501415015150161501715018150191502015021150221502315024150251502615027150281502915030150311503215033150341503515036150371503815039150401504115042150431504415045150461504715048150491505015051150521505315054150551505615057150581505915060150611506215063150641506515066150671506815069150701507115072150731507415075150761507715078150791508015081150821508315084150851508615087150881508915090150911509215093150941509515096150971509815099151001510115102151031510415105151061510715108151091511015111151121511315114151151511615117151181511915120151211512215123151241512515126151271512815129151301513115132151331513415135151361513715138151391514015141151421514315144151451514615147151481514915150151511515215153151541515515156151571515815159151601516115162151631516415165151661516715168151691517015171151721517315174151751517615177151781517915180151811518215183151841518515186151871518815189151901519115192151931519415195151961519715198151991520015201152021520315204152051520615207152081520915210152111521215213152141521515216152171521815219152201522115222152231522415225152261522715228152291523015231152321523315234152351523615237152381523915240152411524215243152441524515246152471524815249152501525115252152531525415255152561525715258152591526015261152621526315264152651526615267152681526915270152711527215273152741527515276152771527815279152801528115282152831528415285152861528715288152891529015291152921529315294152951529615297152981529915300153011530215303153041530515306153071530815309153101531115312153131531415315153161531715318153191532015321153221532315324153251532615327153281532915330153311533215333153341533515336153371533815339153401534115342153431534415345153461534715348153491535015351153521535315354153551535615357153581535915360153611536215363153641536515366153671536815369153701537115372153731537415375153761537715378153791538015381153821538315384153851538615387153881538915390153911539215393153941539515396153971539815399154001540115402154031540415405154061540715408154091541015411154121541315414154151541615417154181541915420154211542215423154241542515426154271542815429154301543115432154331543415435154361543715438154391544015441154421544315444154451544615447154481544915450154511545215453154541545515456154571545815459154601546115462154631546415465154661546715468154691547015471154721547315474154751547615477154781547915480154811548215483154841548515486154871548815489154901549115492154931549415495154961549715498154991550015501155021550315504155051550615507155081550915510155111551215513155141551515516155171551815519155201552115522155231552415525155261552715528155291553015531155321553315534155351553615537155381553915540155411554215543155441554515546155471554815549155501555115552155531555415555155561555715558155591556015561155621556315564155651556615567155681556915570155711557215573155741557515576155771557815579155801558115582155831558415585155861558715588155891559015591155921559315594155951559615597155981559915600156011560215603156041560515606156071560815609156101561115612156131561415615156161561715618156191562015621156221562315624156251562615627156281562915630156311563215633156341563515636156371563815639156401564115642156431564415645156461564715648156491565015651156521565315654156551565615657156581565915660156611566215663156641566515666156671566815669156701567115672156731567415675156761567715678156791568015681156821568315684156851568615687156881568915690156911569215693156941569515696156971569815699157001570115702157031570415705157061570715708157091571015711157121571315714157151571615717157181571915720157211572215723157241572515726157271572815729157301573115732157331573415735157361573715738157391574015741157421574315744157451574615747157481574915750157511575215753157541575515756157571575815759157601576115762157631576415765157661576715768157691577015771157721577315774157751577615777157781577915780157811578215783157841578515786157871578815789157901579115792157931579415795157961579715798157991580015801158021580315804158051580615807158081580915810158111581215813158141581515816158171581815819158201582115822158231582415825158261582715828158291583015831158321583315834158351583615837158381583915840158411584215843158441584515846158471584815849158501585115852158531585415855158561585715858158591586015861158621586315864158651586615867158681586915870158711587215873158741587515876158771587815879158801588115882158831588415885158861588715888158891589015891158921589315894158951589615897158981589915900159011590215903159041590515906159071590815909159101591115912159131591415915159161591715918159191592015921159221592315924159251592615927159281592915930159311593215933159341593515936159371593815939159401594115942159431594415945159461594715948159491595015951159521595315954159551595615957159581595915960159611596215963159641596515966159671596815969159701597115972159731597415975159761597715978159791598015981159821598315984159851598615987159881598915990159911599215993159941599515996159971599815999160001600116002160031600416005160061600716008160091601016011160121601316014160151601616017160181601916020160211602216023160241602516026160271602816029160301603116032160331603416035160361603716038160391604016041160421604316044160451604616047160481604916050160511605216053160541605516056160571605816059160601606116062160631606416065160661606716068160691607016071160721607316074160751607616077160781607916080160811608216083160841608516086160871608816089160901609116092160931609416095160961609716098160991610016101161021610316104161051610616107161081610916110161111611216113161141611516116161171611816119161201612116122161231612416125161261612716128161291613016131161321613316134161351613616137161381613916140161411614216143161441614516146161471614816149161501615116152161531615416155161561615716158161591616016161161621616316164161651616616167161681616916170161711617216173161741617516176161771617816179161801618116182161831618416185161861618716188161891619016191161921619316194161951619616197161981619916200162011620216203162041620516206162071620816209162101621116212162131621416215162161621716218162191622016221162221622316224162251622616227162281622916230162311623216233162341623516236162371623816239162401624116242162431624416245162461624716248162491625016251162521625316254162551625616257162581625916260162611626216263162641626516266162671626816269162701627116272162731627416275162761627716278162791628016281162821628316284162851628616287162881628916290162911629216293162941629516296162971629816299163001630116302163031630416305163061630716308163091631016311163121631316314163151631616317163181631916320163211632216323163241632516326163271632816329163301633116332163331633416335163361633716338163391634016341163421634316344163451634616347163481634916350163511635216353163541635516356163571635816359163601636116362163631636416365163661636716368163691637016371163721637316374163751637616377163781637916380163811638216383163841638516386163871638816389163901639116392163931639416395163961639716398163991640016401164021640316404164051640616407164081640916410164111641216413164141641516416164171641816419164201642116422164231642416425164261642716428164291643016431164321643316434164351643616437164381643916440164411644216443164441644516446164471644816449164501645116452164531645416455164561645716458164591646016461164621646316464164651646616467164681646916470164711647216473164741647516476164771647816479164801648116482164831648416485164861648716488164891649016491164921649316494164951649616497164981649916500165011650216503165041650516506165071650816509165101651116512165131651416515165161651716518165191652016521165221652316524165251652616527165281652916530165311653216533165341653516536165371653816539165401654116542165431654416545165461654716548165491655016551165521655316554165551655616557165581655916560165611656216563165641656516566165671656816569165701657116572165731657416575165761657716578165791658016581165821658316584165851658616587165881658916590165911659216593165941659516596165971659816599166001660116602166031660416605166061660716608166091661016611166121661316614166151661616617166181661916620166211662216623166241662516626166271662816629166301663116632166331663416635166361663716638166391664016641166421664316644166451664616647166481664916650166511665216653166541665516656166571665816659166601666116662166631666416665166661666716668166691667016671166721667316674166751667616677166781667916680166811668216683166841668516686166871668816689166901669116692166931669416695166961669716698166991670016701167021670316704167051670616707167081670916710167111671216713167141671516716167171671816719167201672116722167231672416725167261672716728167291673016731167321673316734167351673616737167381673916740167411674216743167441674516746167471674816749167501675116752167531675416755167561675716758167591676016761167621676316764167651676616767167681676916770167711677216773167741677516776167771677816779167801678116782167831678416785167861678716788167891679016791167921679316794167951679616797167981679916800168011680216803168041680516806168071680816809168101681116812168131681416815168161681716818168191682016821168221682316824168251682616827168281682916830168311683216833168341683516836168371683816839168401684116842168431684416845168461684716848168491685016851168521685316854168551685616857168581685916860168611686216863168641686516866168671686816869168701687116872168731687416875168761687716878168791688016881168821688316884168851688616887168881688916890168911689216893168941689516896168971689816899169001690116902169031690416905169061690716908169091691016911169121691316914169151691616917169181691916920169211692216923169241692516926169271692816929169301693116932169331693416935169361693716938169391694016941169421694316944169451694616947169481694916950169511695216953169541695516956169571695816959169601696116962169631696416965169661696716968169691697016971169721697316974169751697616977169781697916980169811698216983169841698516986169871698816989169901699116992169931699416995169961699716998169991700017001170021700317004170051700617007170081700917010170111701217013170141701517016170171701817019170201702117022170231702417025170261702717028170291703017031170321703317034170351703617037170381703917040170411704217043170441704517046170471704817049170501705117052170531705417055170561705717058170591706017061170621706317064170651706617067170681706917070170711707217073170741707517076170771707817079170801708117082170831708417085170861708717088170891709017091170921709317094170951709617097170981709917100171011710217103171041710517106171071710817109171101711117112171131711417115171161711717118171191712017121171221712317124171251712617127171281712917130171311713217133171341713517136171371713817139171401714117142171431714417145171461714717148171491715017151171521715317154171551715617157171581715917160171611716217163171641716517166171671716817169171701717117172171731717417175171761717717178171791718017181171821718317184171851718617187171881718917190171911719217193171941719517196171971719817199172001720117202172031720417205172061720717208172091721017211172121721317214172151721617217172181721917220172211722217223172241722517226172271722817229172301723117232172331723417235172361723717238172391724017241172421724317244172451724617247172481724917250172511725217253172541725517256172571725817259172601726117262172631726417265172661726717268172691727017271172721727317274172751727617277172781727917280172811728217283172841728517286172871728817289172901729117292172931729417295172961729717298172991730017301173021730317304173051730617307173081730917310173111731217313173141731517316173171731817319173201732117322173231732417325173261732717328173291733017331173321733317334173351733617337173381733917340173411734217343173441734517346173471734817349173501735117352173531735417355173561735717358173591736017361173621736317364173651736617367173681736917370173711737217373173741737517376173771737817379173801738117382173831738417385173861738717388173891739017391173921739317394173951739617397173981739917400174011740217403174041740517406174071740817409174101741117412174131741417415174161741717418174191742017421174221742317424174251742617427174281742917430174311743217433174341743517436174371743817439174401744117442174431744417445174461744717448174491745017451174521745317454174551745617457174581745917460174611746217463174641746517466174671746817469174701747117472174731747417475174761747717478174791748017481174821748317484174851748617487174881748917490174911749217493174941749517496174971749817499175001750117502175031750417505175061750717508175091751017511175121751317514175151751617517175181751917520175211752217523175241752517526175271752817529175301753117532175331753417535175361753717538175391754017541175421754317544175451754617547175481754917550175511755217553175541755517556175571755817559175601756117562175631756417565175661756717568175691757017571175721757317574175751757617577175781757917580175811758217583175841758517586175871758817589175901759117592175931759417595175961759717598175991760017601176021760317604176051760617607176081760917610176111761217613176141761517616176171761817619176201762117622176231762417625176261762717628176291763017631176321763317634176351763617637176381763917640176411764217643176441764517646176471764817649176501765117652176531765417655176561765717658176591766017661176621766317664176651766617667176681766917670176711767217673176741767517676176771767817679176801768117682176831768417685176861768717688176891769017691176921769317694176951769617697176981769917700177011770217703177041770517706177071770817709177101771117712177131771417715177161771717718177191772017721177221772317724177251772617727177281772917730177311773217733177341773517736177371773817739177401774117742177431774417745177461774717748177491775017751177521775317754177551775617757177581775917760177611776217763177641776517766177671776817769177701777117772177731777417775177761777717778177791778017781177821778317784177851778617787177881778917790177911779217793177941779517796177971779817799178001780117802178031780417805178061780717808178091781017811178121781317814178151781617817178181781917820178211782217823178241782517826178271782817829178301783117832178331783417835178361783717838178391784017841178421784317844178451784617847178481784917850178511785217853178541785517856178571785817859178601786117862178631786417865178661786717868178691787017871178721787317874178751787617877178781787917880178811788217883178841788517886178871788817889178901789117892178931789417895178961789717898178991790017901179021790317904179051790617907179081790917910179111791217913179141791517916179171791817919179201792117922179231792417925179261792717928179291793017931179321793317934179351793617937179381793917940179411794217943179441794517946179471794817949179501795117952179531795417955179561795717958179591796017961179621796317964179651796617967179681796917970179711797217973179741797517976179771797817979179801798117982179831798417985179861798717988179891799017991179921799317994179951799617997179981799918000180011800218003180041800518006180071800818009180101801118012180131801418015180161801718018180191802018021180221802318024180251802618027180281802918030180311803218033180341803518036180371803818039180401804118042180431804418045180461804718048180491805018051180521805318054180551805618057180581805918060180611806218063180641806518066180671806818069180701807118072180731807418075180761807718078180791808018081180821808318084180851808618087180881808918090180911809218093180941809518096180971809818099181001810118102181031810418105181061810718108181091811018111181121811318114181151811618117181181811918120181211812218123181241812518126181271812818129181301813118132181331813418135181361813718138181391814018141181421814318144181451814618147181481814918150181511815218153181541815518156181571815818159181601816118162181631816418165181661816718168181691817018171181721817318174181751817618177181781817918180181811818218183181841818518186181871818818189181901819118192181931819418195181961819718198181991820018201182021820318204182051820618207182081820918210182111821218213182141821518216182171821818219182201822118222182231822418225182261822718228182291823018231182321823318234182351823618237182381823918240182411824218243182441824518246182471824818249182501825118252182531825418255182561825718258182591826018261182621826318264182651826618267182681826918270182711827218273182741827518276182771827818279182801828118282182831828418285182861828718288182891829018291182921829318294182951829618297182981829918300183011830218303183041830518306183071830818309183101831118312183131831418315183161831718318183191832018321183221832318324183251832618327183281832918330183311833218333183341833518336183371833818339183401834118342183431834418345183461834718348183491835018351183521835318354183551835618357183581835918360183611836218363183641836518366183671836818369183701837118372183731837418375183761837718378183791838018381183821838318384183851838618387183881838918390183911839218393183941839518396183971839818399184001840118402184031840418405184061840718408184091841018411184121841318414184151841618417184181841918420184211842218423184241842518426184271842818429184301843118432184331843418435184361843718438184391844018441184421844318444184451844618447184481844918450184511845218453184541845518456184571845818459184601846118462184631846418465184661846718468184691847018471184721847318474184751847618477184781847918480184811848218483184841848518486184871848818489184901849118492184931849418495184961849718498184991850018501185021850318504185051850618507185081850918510185111851218513185141851518516185171851818519185201852118522185231852418525185261852718528185291853018531185321853318534185351853618537185381853918540185411854218543185441854518546185471854818549185501855118552185531855418555185561855718558185591856018561185621856318564185651856618567185681856918570185711857218573185741857518576185771857818579185801858118582185831858418585185861858718588185891859018591185921859318594185951859618597185981859918600186011860218603186041860518606186071860818609186101861118612186131861418615186161861718618186191862018621186221862318624186251862618627186281862918630186311863218633186341863518636186371863818639186401864118642186431864418645186461864718648186491865018651186521865318654186551865618657186581865918660186611866218663186641866518666186671866818669186701867118672186731867418675186761867718678186791868018681186821868318684186851868618687186881868918690186911869218693186941869518696186971869818699187001870118702187031870418705187061870718708187091871018711187121871318714187151871618717187181871918720187211872218723187241872518726187271872818729187301873118732187331873418735187361873718738187391874018741187421874318744187451874618747187481874918750187511875218753187541875518756187571875818759187601876118762187631876418765187661876718768187691877018771187721877318774187751877618777187781877918780187811878218783187841878518786187871878818789187901879118792187931879418795187961879718798187991880018801188021880318804188051880618807188081880918810188111881218813188141881518816188171881818819188201882118822188231882418825188261882718828188291883018831188321883318834188351883618837188381883918840188411884218843188441884518846188471884818849188501885118852188531885418855188561885718858188591886018861188621886318864188651886618867188681886918870188711887218873188741887518876188771887818879188801888118882188831888418885188861888718888188891889018891188921889318894188951889618897188981889918900189011890218903189041890518906189071890818909189101891118912189131891418915189161891718918189191892018921189221892318924189251892618927189281892918930189311893218933189341893518936189371893818939189401894118942189431894418945189461894718948189491895018951189521895318954189551895618957189581895918960189611896218963189641896518966189671896818969189701897118972189731897418975189761897718978189791898018981189821898318984189851898618987189881898918990189911899218993189941899518996189971899818999190001900119002190031900419005190061900719008190091901019011190121901319014190151901619017190181901919020190211902219023190241902519026190271902819029190301903119032190331903419035190361903719038190391904019041190421904319044190451904619047190481904919050190511905219053190541905519056190571905819059190601906119062190631906419065190661906719068190691907019071190721907319074190751907619077190781907919080190811908219083190841908519086190871908819089190901909119092190931909419095190961909719098190991910019101191021910319104191051910619107191081910919110191111911219113191141911519116191171911819119191201912119122191231912419125191261912719128191291913019131191321913319134191351913619137191381913919140191411914219143191441914519146191471914819149191501915119152191531915419155191561915719158191591916019161191621916319164191651916619167191681916919170191711917219173191741917519176191771917819179191801918119182191831918419185191861918719188191891919019191191921919319194191951919619197191981919919200192011920219203192041920519206192071920819209192101921119212192131921419215192161921719218192191922019221192221922319224192251922619227192281922919230192311923219233192341923519236192371923819239192401924119242192431924419245192461924719248192491925019251192521925319254192551925619257192581925919260192611926219263192641926519266192671926819269192701927119272192731927419275192761927719278192791928019281192821928319284192851928619287192881928919290192911929219293192941929519296192971929819299193001930119302193031930419305193061930719308193091931019311193121931319314193151931619317193181931919320193211932219323193241932519326193271932819329193301933119332193331933419335193361933719338193391934019341193421934319344193451934619347193481934919350193511935219353193541935519356193571935819359193601936119362193631936419365193661936719368193691937019371193721937319374193751937619377193781937919380193811938219383193841938519386193871938819389193901939119392193931939419395193961939719398193991940019401194021940319404194051940619407194081940919410194111941219413194141941519416194171941819419194201942119422194231942419425194261942719428194291943019431194321943319434194351943619437194381943919440194411944219443194441944519446194471944819449194501945119452194531945419455194561945719458194591946019461194621946319464194651946619467194681946919470194711947219473194741947519476194771947819479194801948119482194831948419485194861948719488194891949019491194921949319494194951949619497194981949919500195011950219503195041950519506195071950819509195101951119512195131951419515195161951719518195191952019521195221952319524195251952619527195281952919530195311953219533195341953519536195371953819539195401954119542195431954419545195461954719548195491955019551195521955319554195551955619557195581955919560195611956219563195641956519566195671956819569195701957119572195731957419575195761957719578195791958019581195821958319584195851958619587195881958919590195911959219593195941959519596195971959819599196001960119602196031960419605196061960719608196091961019611196121961319614196151961619617196181961919620196211962219623196241962519626196271962819629196301963119632196331963419635196361963719638196391964019641196421964319644196451964619647196481964919650196511965219653196541965519656196571965819659196601966119662196631966419665196661966719668196691967019671196721967319674196751967619677196781967919680196811968219683196841968519686196871968819689196901969119692196931969419695196961969719698196991970019701197021970319704197051970619707197081970919710197111971219713197141971519716197171971819719197201972119722197231972419725197261972719728197291973019731197321973319734197351973619737197381973919740197411974219743197441974519746197471974819749197501975119752197531975419755197561975719758197591976019761197621976319764197651976619767197681976919770197711977219773197741977519776197771977819779197801978119782197831978419785197861978719788197891979019791197921979319794197951979619797197981979919800198011980219803198041980519806198071980819809198101981119812198131981419815198161981719818198191982019821198221982319824198251982619827198281982919830198311983219833198341983519836198371983819839198401984119842198431984419845198461984719848198491985019851198521985319854198551985619857198581985919860198611986219863198641986519866198671986819869198701987119872198731987419875198761987719878198791988019881198821988319884198851988619887198881988919890198911989219893198941989519896198971989819899199001990119902199031990419905199061990719908199091991019911199121991319914199151991619917199181991919920199211992219923199241992519926199271992819929199301993119932199331993419935199361993719938199391994019941199421994319944199451994619947199481994919950199511995219953199541995519956199571995819959199601996119962199631996419965199661996719968199691997019971199721997319974199751997619977199781997919980199811998219983199841998519986199871998819989199901999119992199931999419995199961999719998199992000020001200022000320004200052000620007200082000920010200112001220013200142001520016200172001820019200202002120022200232002420025200262002720028200292003020031200322003320034200352003620037200382003920040200412004220043200442004520046200472004820049200502005120052200532005420055200562005720058200592006020061200622006320064200652006620067200682006920070200712007220073200742007520076200772007820079200802008120082200832008420085200862008720088200892009020091200922009320094200952009620097200982009920100201012010220103201042010520106201072010820109201102011120112201132011420115201162011720118201192012020121201222012320124201252012620127201282012920130201312013220133201342013520136201372013820139201402014120142201432014420145201462014720148201492015020151201522015320154201552015620157201582015920160201612016220163201642016520166201672016820169201702017120172201732017420175201762017720178201792018020181201822018320184201852018620187201882018920190201912019220193201942019520196201972019820199202002020120202202032020420205202062020720208202092021020211202122021320214202152021620217202182021920220202212022220223202242022520226202272022820229202302023120232202332023420235202362023720238202392024020241202422024320244202452024620247202482024920250202512025220253202542025520256202572025820259202602026120262202632026420265202662026720268202692027020271202722027320274202752027620277202782027920280202812028220283202842028520286202872028820289202902029120292202932029420295202962029720298202992030020301203022030320304203052030620307203082030920310203112031220313203142031520316203172031820319203202032120322203232032420325203262032720328203292033020331203322033320334203352033620337203382033920340203412034220343203442034520346203472034820349203502035120352203532035420355203562035720358203592036020361203622036320364203652036620367203682036920370203712037220373203742037520376203772037820379203802038120382203832038420385203862038720388203892039020391203922039320394203952039620397203982039920400204012040220403204042040520406204072040820409204102041120412204132041420415204162041720418204192042020421204222042320424204252042620427204282042920430204312043220433204342043520436204372043820439204402044120442204432044420445204462044720448204492045020451204522045320454204552045620457204582045920460204612046220463204642046520466204672046820469204702047120472204732047420475204762047720478204792048020481204822048320484204852048620487204882048920490204912049220493204942049520496204972049820499205002050120502205032050420505205062050720508205092051020511205122051320514205152051620517205182051920520205212052220523205242052520526205272052820529205302053120532205332053420535205362053720538205392054020541205422054320544205452054620547205482054920550205512055220553205542055520556205572055820559205602056120562205632056420565205662056720568205692057020571205722057320574205752057620577205782057920580205812058220583205842058520586205872058820589205902059120592205932059420595205962059720598205992060020601206022060320604206052060620607206082060920610206112061220613206142061520616206172061820619206202062120622206232062420625206262062720628206292063020631206322063320634206352063620637206382063920640206412064220643206442064520646206472064820649206502065120652206532065420655206562065720658206592066020661206622066320664206652066620667206682066920670206712067220673206742067520676206772067820679206802068120682206832068420685206862068720688206892069020691206922069320694206952069620697206982069920700207012070220703207042070520706207072070820709207102071120712207132071420715207162071720718207192072020721207222072320724207252072620727207282072920730207312073220733207342073520736207372073820739207402074120742207432074420745207462074720748207492075020751207522075320754207552075620757207582075920760207612076220763207642076520766207672076820769207702077120772207732077420775207762077720778207792078020781207822078320784207852078620787207882078920790207912079220793207942079520796207972079820799208002080120802208032080420805208062080720808208092081020811208122081320814208152081620817208182081920820208212082220823208242082520826208272082820829208302083120832208332083420835208362083720838208392084020841208422084320844208452084620847208482084920850208512085220853208542085520856208572085820859208602086120862208632086420865208662086720868208692087020871208722087320874208752087620877208782087920880208812088220883208842088520886208872088820889208902089120892208932089420895208962089720898208992090020901209022090320904209052090620907209082090920910209112091220913209142091520916209172091820919209202092120922209232092420925209262092720928209292093020931209322093320934209352093620937209382093920940209412094220943209442094520946209472094820949209502095120952209532095420955209562095720958209592096020961209622096320964209652096620967209682096920970209712097220973209742097520976209772097820979209802098120982209832098420985209862098720988209892099020991209922099320994209952099620997209982099921000210012100221003210042100521006210072100821009210102101121012210132101421015210162101721018210192102021021210222102321024210252102621027210282102921030210312103221033210342103521036210372103821039210402104121042210432104421045210462104721048210492105021051210522105321054210552105621057210582105921060210612106221063210642106521066210672106821069210702107121072210732107421075210762107721078210792108021081210822108321084210852108621087210882108921090210912109221093210942109521096210972109821099211002110121102211032110421105211062110721108211092111021111211122111321114211152111621117211182111921120211212112221123211242112521126211272112821129211302113121132211332113421135211362113721138211392114021141211422114321144211452114621147211482114921150211512115221153211542115521156211572115821159211602116121162211632116421165211662116721168211692117021171211722117321174211752117621177211782117921180211812118221183211842118521186211872118821189211902119121192211932119421195211962119721198211992120021201212022120321204212052120621207212082120921210212112121221213212142121521216212172121821219212202122121222212232122421225212262122721228212292123021231212322123321234212352123621237212382123921240212412124221243212442124521246212472124821249212502125121252212532125421255212562125721258212592126021261212622126321264212652126621267212682126921270212712127221273212742127521276212772127821279212802128121282212832128421285212862128721288212892129021291212922129321294212952129621297212982129921300213012130221303213042130521306213072130821309213102131121312213132131421315213162131721318213192132021321213222132321324213252132621327213282132921330213312133221333213342133521336213372133821339213402134121342213432134421345213462134721348213492135021351213522135321354213552135621357213582135921360213612136221363213642136521366213672136821369213702137121372213732137421375213762137721378213792138021381213822138321384213852138621387213882138921390213912139221393213942139521396213972139821399214002140121402214032140421405214062140721408214092141021411214122141321414214152141621417214182141921420214212142221423214242142521426214272142821429214302143121432214332143421435214362143721438214392144021441214422144321444214452144621447214482144921450214512145221453214542145521456214572145821459214602146121462214632146421465214662146721468214692147021471214722147321474214752147621477214782147921480214812148221483214842148521486214872148821489214902149121492214932149421495214962149721498214992150021501215022150321504215052150621507215082150921510215112151221513215142151521516215172151821519215202152121522215232152421525215262152721528215292153021531215322153321534215352153621537215382153921540215412154221543215442154521546215472154821549215502155121552215532155421555215562155721558215592156021561215622156321564215652156621567215682156921570215712157221573215742157521576215772157821579215802158121582215832158421585215862158721588215892159021591215922159321594215952159621597215982159921600216012160221603216042160521606216072160821609216102161121612216132161421615216162161721618216192162021621216222162321624216252162621627216282162921630216312163221633216342163521636216372163821639216402164121642216432164421645216462164721648216492165021651216522165321654216552165621657216582165921660216612166221663216642166521666216672166821669216702167121672216732167421675216762167721678216792168021681216822168321684216852168621687216882168921690216912169221693216942169521696216972169821699217002170121702217032170421705217062170721708217092171021711217122171321714217152171621717217182171921720217212172221723217242172521726217272172821729217302173121732217332173421735217362173721738217392174021741217422174321744217452174621747217482174921750217512175221753217542175521756217572175821759217602176121762217632176421765217662176721768217692177021771217722177321774217752177621777217782177921780217812178221783217842178521786217872178821789217902179121792217932179421795217962179721798217992180021801218022180321804218052180621807218082180921810218112181221813218142181521816218172181821819218202182121822218232182421825218262182721828218292183021831218322183321834218352183621837218382183921840218412184221843218442184521846218472184821849218502185121852218532185421855218562185721858218592186021861218622186321864218652186621867218682186921870218712187221873218742187521876218772187821879218802188121882218832188421885218862188721888218892189021891218922189321894218952189621897218982189921900219012190221903219042190521906219072190821909219102191121912219132191421915219162191721918219192192021921219222192321924219252192621927219282192921930219312193221933219342193521936219372193821939219402194121942219432194421945219462194721948219492195021951219522195321954219552195621957219582195921960219612196221963219642196521966219672196821969219702197121972219732197421975219762197721978219792198021981219822198321984219852198621987219882198921990219912199221993219942199521996219972199821999220002200122002220032200422005220062200722008220092201022011220122201322014220152201622017220182201922020220212202222023220242202522026220272202822029220302203122032220332203422035220362203722038220392204022041220422204322044220452204622047220482204922050220512205222053220542205522056220572205822059220602206122062220632206422065220662206722068220692207022071220722207322074220752207622077220782207922080220812208222083220842208522086220872208822089220902209122092220932209422095220962209722098220992210022101221022210322104221052210622107221082210922110221112211222113221142211522116221172211822119221202212122122221232212422125221262212722128221292213022131221322213322134221352213622137221382213922140221412214222143221442214522146221472214822149221502215122152221532215422155221562215722158221592216022161221622216322164221652216622167221682216922170221712217222173221742217522176221772217822179221802218122182221832218422185221862218722188221892219022191221922219322194221952219622197221982219922200222012220222203222042220522206222072220822209222102221122212222132221422215222162221722218222192222022221222222222322224222252222622227222282222922230222312223222233222342223522236222372223822239222402224122242222432224422245222462224722248222492225022251222522225322254222552225622257222582225922260222612226222263222642226522266222672226822269222702227122272222732227422275222762227722278222792228022281222822228322284222852228622287222882228922290222912229222293222942229522296222972229822299223002230122302223032230422305223062230722308223092231022311223122231322314223152231622317223182231922320223212232222323223242232522326223272232822329223302233122332223332233422335223362233722338223392234022341223422234322344223452234622347223482234922350223512235222353223542235522356223572235822359223602236122362223632236422365223662236722368223692237022371223722237322374223752237622377223782237922380223812238222383223842238522386223872238822389223902239122392223932239422395223962239722398223992240022401224022240322404224052240622407224082240922410224112241222413224142241522416224172241822419224202242122422224232242422425224262242722428224292243022431224322243322434224352243622437224382243922440224412244222443224442244522446224472244822449224502245122452224532245422455224562245722458224592246022461224622246322464224652246622467224682246922470224712247222473224742247522476224772247822479224802248122482224832248422485224862248722488224892249022491224922249322494224952249622497224982249922500225012250222503225042250522506225072250822509225102251122512225132251422515225162251722518225192252022521225222252322524225252252622527225282252922530225312253222533225342253522536225372253822539225402254122542225432254422545225462254722548225492255022551225522255322554225552255622557225582255922560225612256222563225642256522566225672256822569225702257122572225732257422575225762257722578225792258022581225822258322584225852258622587225882258922590225912259222593225942259522596225972259822599226002260122602226032260422605226062260722608226092261022611226122261322614226152261622617226182261922620226212262222623226242262522626226272262822629226302263122632226332263422635226362263722638226392264022641226422264322644226452264622647226482264922650226512265222653226542265522656226572265822659226602266122662226632266422665226662266722668226692267022671226722267322674226752267622677226782267922680226812268222683226842268522686226872268822689226902269122692226932269422695226962269722698226992270022701227022270322704227052270622707227082270922710227112271222713227142271522716227172271822719227202272122722227232272422725227262272722728227292273022731227322273322734227352273622737227382273922740227412274222743227442274522746227472274822749227502275122752227532275422755227562275722758227592276022761227622276322764227652276622767227682276922770227712277222773227742277522776227772277822779227802278122782227832278422785227862278722788227892279022791227922279322794227952279622797227982279922800228012280222803228042280522806228072280822809228102281122812228132281422815228162281722818228192282022821228222282322824228252282622827228282282922830228312283222833228342283522836228372283822839228402284122842228432284422845228462284722848228492285022851228522285322854228552285622857228582285922860228612286222863228642286522866228672286822869228702287122872228732287422875228762287722878228792288022881228822288322884228852288622887228882288922890228912289222893228942289522896228972289822899229002290122902229032290422905229062290722908229092291022911229122291322914229152291622917229182291922920229212292222923229242292522926229272292822929229302293122932229332293422935229362293722938229392294022941229422294322944229452294622947229482294922950229512295222953229542295522956229572295822959229602296122962229632296422965229662296722968229692297022971229722297322974229752297622977229782297922980229812298222983229842298522986229872298822989229902299122992229932299422995229962299722998229992300023001230022300323004230052300623007230082300923010230112301223013230142301523016230172301823019230202302123022230232302423025230262302723028230292303023031230322303323034230352303623037230382303923040230412304223043230442304523046230472304823049230502305123052230532305423055230562305723058230592306023061230622306323064230652306623067230682306923070230712307223073230742307523076230772307823079230802308123082230832308423085230862308723088230892309023091230922309323094230952309623097230982309923100231012310223103231042310523106231072310823109231102311123112231132311423115231162311723118231192312023121231222312323124231252312623127231282312923130231312313223133231342313523136231372313823139231402314123142231432314423145231462314723148231492315023151231522315323154231552315623157231582315923160231612316223163231642316523166231672316823169231702317123172231732317423175231762317723178231792318023181231822318323184231852318623187231882318923190231912319223193231942319523196231972319823199232002320123202232032320423205232062320723208232092321023211232122321323214232152321623217232182321923220232212322223223232242322523226232272322823229232302323123232232332323423235232362323723238232392324023241232422324323244232452324623247232482324923250232512325223253232542325523256232572325823259232602326123262232632326423265232662326723268232692327023271232722327323274232752327623277232782327923280232812328223283232842328523286232872328823289232902329123292232932329423295232962329723298232992330023301233022330323304233052330623307233082330923310233112331223313233142331523316233172331823319233202332123322233232332423325233262332723328233292333023331233322333323334233352333623337233382333923340233412334223343233442334523346233472334823349233502335123352233532335423355233562335723358233592336023361233622336323364233652336623367233682336923370233712337223373233742337523376233772337823379233802338123382233832338423385233862338723388233892339023391233922339323394233952339623397233982339923400234012340223403234042340523406234072340823409234102341123412234132341423415234162341723418234192342023421234222342323424234252342623427234282342923430234312343223433234342343523436234372343823439234402344123442234432344423445234462344723448234492345023451234522345323454234552345623457234582345923460234612346223463234642346523466234672346823469234702347123472234732347423475234762347723478234792348023481234822348323484234852348623487234882348923490234912349223493234942349523496234972349823499235002350123502235032350423505235062350723508235092351023511235122351323514235152351623517235182351923520235212352223523235242352523526235272352823529235302353123532235332353423535235362353723538235392354023541235422354323544235452354623547235482354923550235512355223553235542355523556235572355823559235602356123562235632356423565235662356723568235692357023571235722357323574235752357623577235782357923580235812358223583235842358523586235872358823589235902359123592235932359423595235962359723598235992360023601236022360323604236052360623607236082360923610236112361223613236142361523616236172361823619236202362123622236232362423625236262362723628236292363023631236322363323634236352363623637236382363923640236412364223643236442364523646236472364823649236502365123652236532365423655236562365723658236592366023661236622366323664236652366623667236682366923670236712367223673236742367523676236772367823679236802368123682236832368423685236862368723688236892369023691236922369323694236952369623697236982369923700237012370223703237042370523706237072370823709237102371123712237132371423715237162371723718237192372023721237222372323724237252372623727237282372923730237312373223733237342373523736237372373823739237402374123742237432374423745237462374723748237492375023751237522375323754237552375623757237582375923760237612376223763237642376523766237672376823769237702377123772237732377423775237762377723778237792378023781237822378323784237852378623787237882378923790237912379223793237942379523796237972379823799238002380123802238032380423805238062380723808238092381023811238122381323814238152381623817238182381923820238212382223823238242382523826238272382823829238302383123832238332383423835238362383723838238392384023841238422384323844238452384623847238482384923850238512385223853238542385523856238572385823859238602386123862238632386423865238662386723868238692387023871238722387323874238752387623877238782387923880238812388223883238842388523886238872388823889238902389123892238932389423895238962389723898238992390023901239022390323904239052390623907239082390923910239112391223913239142391523916239172391823919239202392123922239232392423925239262392723928239292393023931239322393323934239352393623937239382393923940239412394223943239442394523946239472394823949239502395123952239532395423955239562395723958239592396023961239622396323964239652396623967239682396923970239712397223973239742397523976239772397823979239802398123982239832398423985239862398723988239892399023991239922399323994239952399623997239982399924000240012400224003240042400524006240072400824009240102401124012240132401424015240162401724018240192402024021240222402324024240252402624027240282402924030240312403224033240342403524036240372403824039240402404124042240432404424045240462404724048240492405024051240522405324054240552405624057240582405924060240612406224063240642406524066240672406824069240702407124072240732407424075240762407724078240792408024081240822408324084240852408624087240882408924090240912409224093240942409524096240972409824099241002410124102241032410424105241062410724108241092411024111241122411324114241152411624117241182411924120241212412224123241242412524126241272412824129241302413124132241332413424135241362413724138241392414024141241422414324144241452414624147241482414924150241512415224153241542415524156241572415824159241602416124162241632416424165241662416724168241692417024171241722417324174241752417624177241782417924180241812418224183241842418524186241872418824189241902419124192241932419424195241962419724198241992420024201242022420324204242052420624207242082420924210242112421224213242142421524216242172421824219242202422124222242232422424225242262422724228242292423024231242322423324234242352423624237242382423924240242412424224243242442424524246242472424824249242502425124252242532425424255242562425724258242592426024261242622426324264242652426624267242682426924270242712427224273242742427524276242772427824279242802428124282242832428424285242862428724288242892429024291242922429324294242952429624297242982429924300243012430224303243042430524306243072430824309243102431124312243132431424315243162431724318243192432024321243222432324324243252432624327243282432924330243312433224333243342433524336243372433824339243402434124342243432434424345243462434724348243492435024351243522435324354243552435624357243582435924360243612436224363243642436524366243672436824369243702437124372243732437424375243762437724378243792438024381243822438324384243852438624387243882438924390243912439224393243942439524396243972439824399244002440124402244032440424405244062440724408244092441024411244122441324414244152441624417244182441924420244212442224423244242442524426244272442824429244302443124432244332443424435244362443724438244392444024441244422444324444244452444624447244482444924450244512445224453244542445524456244572445824459244602446124462244632446424465244662446724468244692447024471244722447324474244752447624477244782447924480244812448224483244842448524486244872448824489244902449124492244932449424495244962449724498244992450024501245022450324504245052450624507245082450924510245112451224513245142451524516245172451824519245202452124522245232452424525245262452724528245292453024531245322453324534245352453624537245382453924540245412454224543245442454524546245472454824549245502455124552245532455424555245562455724558245592456024561245622456324564245652456624567245682456924570245712457224573245742457524576245772457824579245802458124582245832458424585245862458724588245892459024591245922459324594245952459624597245982459924600246012460224603246042460524606246072460824609246102461124612246132461424615246162461724618246192462024621246222462324624246252462624627246282462924630246312463224633246342463524636246372463824639246402464124642246432464424645246462464724648246492465024651246522465324654246552465624657246582465924660246612466224663246642466524666246672466824669246702467124672246732467424675246762467724678246792468024681246822468324684246852468624687246882468924690246912469224693246942469524696246972469824699247002470124702247032470424705247062470724708247092471024711247122471324714247152471624717247182471924720247212472224723247242472524726247272472824729247302473124732247332473424735247362473724738247392474024741247422474324744247452474624747247482474924750247512475224753247542475524756247572475824759247602476124762247632476424765247662476724768247692477024771247722477324774247752477624777247782477924780247812478224783247842478524786247872478824789247902479124792247932479424795247962479724798247992480024801248022480324804248052480624807248082480924810248112481224813248142481524816248172481824819248202482124822248232482424825248262482724828248292483024831248322483324834248352483624837248382483924840248412484224843248442484524846248472484824849248502485124852248532485424855248562485724858248592486024861248622486324864248652486624867248682486924870248712487224873248742487524876248772487824879248802488124882248832488424885248862488724888248892489024891248922489324894248952489624897248982489924900249012490224903249042490524906249072490824909249102491124912249132491424915249162491724918249192492024921249222492324924249252492624927249282492924930249312493224933249342493524936249372493824939249402494124942249432494424945249462494724948249492495024951249522495324954249552495624957249582495924960249612496224963249642496524966249672496824969249702497124972249732497424975249762497724978249792498024981249822498324984249852498624987249882498924990249912499224993249942499524996249972499824999250002500125002250032500425005250062500725008250092501025011250122501325014250152501625017250182501925020250212502225023250242502525026250272502825029250302503125032250332503425035250362503725038250392504025041250422504325044250452504625047250482504925050250512505225053250542505525056250572505825059250602506125062250632506425065250662506725068250692507025071250722507325074250752507625077250782507925080250812508225083250842508525086250872508825089250902509125092250932509425095250962509725098250992510025101251022510325104251052510625107251082510925110251112511225113251142511525116251172511825119251202512125122251232512425125251262512725128251292513025131251322513325134251352513625137251382513925140251412514225143251442514525146251472514825149251502515125152251532515425155251562515725158251592516025161251622516325164251652516625167251682516925170251712517225173251742517525176251772517825179251802518125182251832518425185251862518725188251892519025191251922519325194251952519625197251982519925200252012520225203252042520525206252072520825209252102521125212252132521425215252162521725218252192522025221252222522325224252252522625227252282522925230252312523225233252342523525236252372523825239252402524125242252432524425245252462524725248252492525025251252522525325254252552525625257252582525925260252612526225263252642526525266252672526825269252702527125272252732527425275252762527725278252792528025281252822528325284252852528625287252882528925290252912529225293252942529525296252972529825299253002530125302253032530425305253062530725308253092531025311253122531325314253152531625317253182531925320253212532225323253242532525326253272532825329253302533125332253332533425335253362533725338253392534025341253422534325344253452534625347253482534925350253512535225353253542535525356253572535825359253602536125362253632536425365253662536725368253692537025371253722537325374253752537625377253782537925380253812538225383253842538525386253872538825389253902539125392253932539425395253962539725398253992540025401254022540325404254052540625407254082540925410254112541225413254142541525416254172541825419254202542125422254232542425425254262542725428254292543025431254322543325434254352543625437254382543925440254412544225443254442544525446254472544825449254502545125452254532545425455254562545725458254592546025461254622546325464254652546625467254682546925470254712547225473254742547525476254772547825479254802548125482254832548425485254862548725488254892549025491254922549325494254952549625497254982549925500255012550225503255042550525506255072550825509255102551125512255132551425515255162551725518255192552025521255222552325524255252552625527255282552925530255312553225533255342553525536255372553825539255402554125542255432554425545255462554725548255492555025551255522555325554255552555625557255582555925560255612556225563255642556525566255672556825569255702557125572255732557425575255762557725578255792558025581255822558325584255852558625587255882558925590255912559225593255942559525596255972559825599256002560125602256032560425605256062560725608256092561025611256122561325614256152561625617256182561925620256212562225623256242562525626256272562825629256302563125632256332563425635256362563725638256392564025641256422564325644256452564625647256482564925650256512565225653256542565525656256572565825659256602566125662256632566425665256662566725668256692567025671256722567325674256752567625677256782567925680256812568225683256842568525686256872568825689256902569125692256932569425695256962569725698256992570025701257022570325704257052570625707257082570925710257112571225713257142571525716257172571825719257202572125722257232572425725257262572725728257292573025731257322573325734257352573625737257382573925740257412574225743257442574525746257472574825749257502575125752257532575425755257562575725758257592576025761257622576325764257652576625767257682576925770257712577225773257742577525776257772577825779257802578125782257832578425785257862578725788257892579025791257922579325794257952579625797257982579925800258012580225803258042580525806258072580825809258102581125812258132581425815258162581725818258192582025821258222582325824258252582625827258282582925830258312583225833258342583525836258372583825839258402584125842258432584425845258462584725848258492585025851258522585325854258552585625857258582585925860258612586225863258642586525866258672586825869258702587125872258732587425875258762587725878258792588025881258822588325884258852588625887258882588925890258912589225893258942589525896258972589825899259002590125902259032590425905259062590725908259092591025911259122591325914259152591625917259182591925920259212592225923259242592525926259272592825929259302593125932259332593425935259362593725938259392594025941259422594325944259452594625947259482594925950259512595225953259542595525956259572595825959259602596125962259632596425965259662596725968259692597025971259722597325974259752597625977259782597925980259812598225983259842598525986259872598825989259902599125992259932599425995259962599725998259992600026001260022600326004260052600626007260082600926010260112601226013260142601526016260172601826019260202602126022260232602426025260262602726028260292603026031260322603326034260352603626037260382603926040260412604226043260442604526046260472604826049260502605126052260532605426055260562605726058260592606026061260622606326064260652606626067260682606926070260712607226073260742607526076260772607826079260802608126082260832608426085260862608726088260892609026091260922609326094260952609626097260982609926100261012610226103261042610526106261072610826109261102611126112261132611426115261162611726118261192612026121261222612326124261252612626127261282612926130261312613226133261342613526136261372613826139261402614126142261432614426145261462614726148261492615026151261522615326154261552615626157261582615926160261612616226163261642616526166261672616826169261702617126172261732617426175261762617726178261792618026181261822618326184261852618626187261882618926190261912619226193261942619526196261972619826199262002620126202262032620426205262062620726208262092621026211262122621326214262152621626217262182621926220262212622226223262242622526226262272622826229262302623126232262332623426235262362623726238262392624026241262422624326244262452624626247262482624926250262512625226253262542625526256262572625826259262602626126262262632626426265262662626726268262692627026271262722627326274262752627626277262782627926280262812628226283262842628526286262872628826289262902629126292262932629426295262962629726298262992630026301263022630326304263052630626307263082630926310263112631226313263142631526316263172631826319263202632126322263232632426325263262632726328263292633026331263322633326334263352633626337263382633926340263412634226343263442634526346263472634826349263502635126352263532635426355263562635726358263592636026361263622636326364263652636626367263682636926370263712637226373263742637526376263772637826379263802638126382263832638426385263862638726388263892639026391263922639326394263952639626397263982639926400 |
- // Copyright (C) 2019 Nicola Murino
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published
- // by the Free Software Foundation, version 3.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- package httpd_test
- import (
- "bytes"
- "crypto/rand"
- "encoding/json"
- "errors"
- "fmt"
- "io"
- "io/fs"
- "math"
- "mime/multipart"
- "net"
- "net/http"
- "net/http/httptest"
- "net/url"
- "os"
- "path"
- "path/filepath"
- "regexp"
- "runtime"
- "strconv"
- "strings"
- "sync"
- "testing"
- "time"
- "github.com/go-chi/render"
- _ "github.com/go-sql-driver/mysql"
- _ "github.com/jackc/pgx/v5/stdlib"
- "github.com/lithammer/shortuuid/v3"
- _ "github.com/mattn/go-sqlite3"
- "github.com/mhale/smtpd"
- "github.com/pquerna/otp"
- "github.com/pquerna/otp/totp"
- "github.com/rs/xid"
- "github.com/rs/zerolog"
- "github.com/sftpgo/sdk"
- sdkkms "github.com/sftpgo/sdk/kms"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
- "golang.org/x/crypto/bcrypt"
- "golang.org/x/crypto/ssh"
- "golang.org/x/net/html"
- "github.com/drakkan/sftpgo/v2/internal/acme"
- "github.com/drakkan/sftpgo/v2/internal/common"
- "github.com/drakkan/sftpgo/v2/internal/config"
- "github.com/drakkan/sftpgo/v2/internal/dataprovider"
- "github.com/drakkan/sftpgo/v2/internal/httpclient"
- "github.com/drakkan/sftpgo/v2/internal/httpd"
- "github.com/drakkan/sftpgo/v2/internal/httpdtest"
- "github.com/drakkan/sftpgo/v2/internal/kms"
- "github.com/drakkan/sftpgo/v2/internal/logger"
- "github.com/drakkan/sftpgo/v2/internal/mfa"
- "github.com/drakkan/sftpgo/v2/internal/plugin"
- "github.com/drakkan/sftpgo/v2/internal/sftpd"
- "github.com/drakkan/sftpgo/v2/internal/smtp"
- "github.com/drakkan/sftpgo/v2/internal/util"
- "github.com/drakkan/sftpgo/v2/internal/vfs"
- )
- const (
- defaultUsername = "test_user"
- defaultPassword = "test_password"
- testPubKey = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQC03jj0D+djk7pxIf/0OhrxrchJTRZklofJ1NoIu4752Sq02mdXmarMVsqJ1cAjV5LBVy3D1F5U6XW4rppkXeVtd04Pxb09ehtH0pRRPaoHHlALiJt8CoMpbKYMA8b3KXPPriGxgGomvtU2T2RMURSwOZbMtpsugfjYSWenyYX+VORYhylWnSXL961LTyC21ehd6d6QnW9G7E5hYMITMY9TuQZz3bROYzXiTsgN0+g6Hn7exFQp50p45StUMfV/SftCMdCxlxuyGny2CrN/vfjO7xxOo2uv7q1qm10Q46KPWJQv+pgZ/OfL+EDjy07n5QVSKHlbx+2nT4Q0EgOSQaCTYwn3YjtABfIxWwgAFdyj6YlPulCL22qU4MYhDcA6PSBwDdf8hvxBfvsiHdM+JcSHvv8/VeJhk6CmnZxGY0fxBupov27z3yEO8nAg8k+6PaUiW1MSUfuGMF/ktB8LOstXsEPXSszuyXiOv4DaryOXUiSn7bmRqKcEFlJusO6aZP0= nicola@p1"
- testPubKey1 = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQCd60+/j+y8f0tLftihWV1YN9RSahMI9btQMDIMqts/jeNbD8jgoogM3nhF7KxfcaMKURuD47KC4Ey6iAJUJ0sWkSNNxOcIYuvA+5MlspfZDsa8Ag76Fe1vyz72WeHMHMeh/hwFo2TeIeIXg480T1VI6mzfDrVp2GzUx0SS0dMsQBjftXkuVR8YOiOwMCAH2a//M1OrvV7d/NBk6kBN0WnuIBb2jKm15PAA7+jQQG7tzwk2HedNH3jeL5GH31xkSRwlBczRK0xsCQXehAlx6cT/e/s44iJcJTHfpPKoSk6UAhPJYe7Z1QnuoawY9P9jQaxpyeImBZxxUEowhjpj2avBxKdRGBVK8R7EL8tSOeLbhdyWe5Mwc1+foEbq9Zz5j5Kd+hn3Wm1UnsGCrXUUUoZp1jnlNl0NakCto+5KmqnT9cHxaY+ix2RLUWAZyVFlRq71OYux1UHJnEJPiEI1/tr4jFBSL46qhQZv/TfpkfVW8FLz0lErfqu0gQEZnNHr3Fc= nicola@p1"
- defaultTokenAuthUser = "admin"
- defaultTokenAuthPass = "password"
- altAdminUsername = "newTestAdmin"
- altAdminPassword = "password1"
- csrfFormToken = "_form_token"
- tokenPath = "/api/v2/token"
- userTokenPath = "/api/v2/user/token"
- userLogoutPath = "/api/v2/user/logout"
- userPath = "/api/v2/users"
- adminPath = "/api/v2/admins"
- adminPwdPath = "/api/v2/admin/changepwd"
- folderPath = "/api/v2/folders"
- groupPath = "/api/v2/groups"
- activeConnectionsPath = "/api/v2/connections"
- serverStatusPath = "/api/v2/status"
- quotasBasePath = "/api/v2/quotas"
- quotaScanPath = "/api/v2/quotas/users/scans"
- quotaScanVFolderPath = "/api/v2/quotas/folders/scans"
- defenderHosts = "/api/v2/defender/hosts"
- versionPath = "/api/v2/version"
- logoutPath = "/api/v2/logout"
- userPwdPath = "/api/v2/user/changepwd"
- userDirsPath = "/api/v2/user/dirs"
- userFilesPath = "/api/v2/user/files"
- userFileActionsPath = "/api/v2/user/file-actions"
- userStreamZipPath = "/api/v2/user/streamzip"
- userUploadFilePath = "/api/v2/user/files/upload"
- userFilesDirsMetadataPath = "/api/v2/user/files/metadata"
- apiKeysPath = "/api/v2/apikeys"
- adminTOTPConfigsPath = "/api/v2/admin/totp/configs"
- adminTOTPGeneratePath = "/api/v2/admin/totp/generate"
- adminTOTPValidatePath = "/api/v2/admin/totp/validate"
- adminTOTPSavePath = "/api/v2/admin/totp/save"
- admin2FARecoveryCodesPath = "/api/v2/admin/2fa/recoverycodes"
- adminProfilePath = "/api/v2/admin/profile"
- userTOTPConfigsPath = "/api/v2/user/totp/configs"
- userTOTPGeneratePath = "/api/v2/user/totp/generate"
- userTOTPValidatePath = "/api/v2/user/totp/validate"
- userTOTPSavePath = "/api/v2/user/totp/save"
- user2FARecoveryCodesPath = "/api/v2/user/2fa/recoverycodes"
- userProfilePath = "/api/v2/user/profile"
- userSharesPath = "/api/v2/user/shares"
- retentionBasePath = "/api/v2/retention/users"
- fsEventsPath = "/api/v2/events/fs"
- providerEventsPath = "/api/v2/events/provider"
- logEventsPath = "/api/v2/events/logs"
- sharesPath = "/api/v2/shares"
- eventActionsPath = "/api/v2/eventactions"
- eventRulesPath = "/api/v2/eventrules"
- rolesPath = "/api/v2/roles"
- ipListsPath = "/api/v2/iplists"
- healthzPath = "/healthz"
- webBasePath = "/web"
- webBasePathAdmin = "/web/admin"
- webAdminSetupPath = "/web/admin/setup"
- webLoginPath = "/web/admin/login"
- webLogoutPath = "/web/admin/logout"
- webUsersPath = "/web/admin/users"
- webUserPath = "/web/admin/user"
- webGroupsPath = "/web/admin/groups"
- webGroupPath = "/web/admin/group"
- webFoldersPath = "/web/admin/folders"
- webFolderPath = "/web/admin/folder"
- webConnectionsPath = "/web/admin/connections"
- webStatusPath = "/web/admin/status"
- webAdminsPath = "/web/admin/managers"
- webAdminPath = "/web/admin/manager"
- webMaintenancePath = "/web/admin/maintenance"
- webRestorePath = "/web/admin/restore"
- webChangeAdminPwdPath = "/web/admin/changepwd"
- webAdminProfilePath = "/web/admin/profile"
- webTemplateUser = "/web/admin/template/user"
- webTemplateFolder = "/web/admin/template/folder"
- webDefenderPath = "/web/admin/defender"
- webIPListsPath = "/web/admin/ip-lists"
- webIPListPath = "/web/admin/ip-list"
- webAdminTwoFactorPath = "/web/admin/twofactor"
- webAdminTwoFactorRecoveryPath = "/web/admin/twofactor-recovery"
- webAdminMFAPath = "/web/admin/mfa"
- webAdminTOTPSavePath = "/web/admin/totp/save"
- webAdminForgotPwdPath = "/web/admin/forgot-password"
- webAdminResetPwdPath = "/web/admin/reset-password"
- webAdminEventRulesPath = "/web/admin/eventrules"
- webAdminEventRulePath = "/web/admin/eventrule"
- webAdminEventActionsPath = "/web/admin/eventactions"
- webAdminEventActionPath = "/web/admin/eventaction"
- webAdminRolesPath = "/web/admin/roles"
- webAdminRolePath = "/web/admin/role"
- webEventsPath = "/web/admin/events"
- webConfigsPath = "/web/admin/configs"
- webOAuth2TokenPath = "/web/admin/oauth2/token"
- webBasePathClient = "/web/client"
- webClientLoginPath = "/web/client/login"
- webClientFilesPath = "/web/client/files"
- webClientEditFilePath = "/web/client/editfile"
- webClientDirsPath = "/web/client/dirs"
- webClientDownloadZipPath = "/web/client/downloadzip"
- webChangeClientPwdPath = "/web/client/changepwd"
- webClientProfilePath = "/web/client/profile"
- webClientPingPath = "/web/client/ping"
- webClientTwoFactorPath = "/web/client/twofactor"
- webClientTwoFactorRecoveryPath = "/web/client/twofactor-recovery"
- webClientLogoutPath = "/web/client/logout"
- webClientMFAPath = "/web/client/mfa"
- webClientTOTPSavePath = "/web/client/totp/save"
- webClientSharesPath = "/web/client/shares"
- webClientSharePath = "/web/client/share"
- webClientPubSharesPath = "/web/client/pubshares"
- webClientForgotPwdPath = "/web/client/forgot-password"
- webClientResetPwdPath = "/web/client/reset-password"
- webClientViewPDFPath = "/web/client/viewpdf"
- webClientGetPDFPath = "/web/client/getpdf"
- webClientExistPath = "/web/client/exist"
- webClientTasksPath = "/web/client/tasks"
- webClientFileMovePath = "/web/client/file-actions/move"
- webClientFileCopyPath = "/web/client/file-actions/copy"
- jsonAPISuffix = "/json"
- httpBaseURL = "http://127.0.0.1:8081"
- defaultRemoteAddr = "127.0.0.1:1234"
- sftpServerAddr = "127.0.0.1:8022"
- smtpServerAddr = "127.0.0.1:3525"
- httpsCert = `-----BEGIN CERTIFICATE-----
- MIICHTCCAaKgAwIBAgIUHnqw7QnB1Bj9oUsNpdb+ZkFPOxMwCgYIKoZIzj0EAwIw
- RTELMAkGA1UEBhMCQVUxEzARBgNVBAgMClNvbWUtU3RhdGUxITAfBgNVBAoMGElu
- dGVybmV0IFdpZGdpdHMgUHR5IEx0ZDAeFw0yMDAyMDQwOTUzMDRaFw0zMDAyMDEw
- OTUzMDRaMEUxCzAJBgNVBAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYD
- VQQKDBhJbnRlcm5ldCBXaWRnaXRzIFB0eSBMdGQwdjAQBgcqhkjOPQIBBgUrgQQA
- IgNiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVqWvrJ51t5OxV0v25NsOgR82CA
- NXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIVCzgWkxiz7XE4lgUwX44FCXZM
- 3+JeUbKjUzBRMB0GA1UdDgQWBBRhLw+/o3+Z02MI/d4tmaMui9W16jAfBgNVHSME
- GDAWgBRhLw+/o3+Z02MI/d4tmaMui9W16jAPBgNVHRMBAf8EBTADAQH/MAoGCCqG
- SM49BAMCA2kAMGYCMQDqLt2lm8mE+tGgtjDmtFgdOcI72HSbRQ74D5rYTzgST1rY
- /8wTi5xl8TiFUyLMUsICMQC5ViVxdXbhuG7gX6yEqSkMKZICHpO8hqFwOD/uaFVI
- dV4vKmHUzwK/eIx+8Ay3neE=
- -----END CERTIFICATE-----`
- httpsKey = `-----BEGIN EC PARAMETERS-----
- BgUrgQQAIg==
- -----END EC PARAMETERS-----
- -----BEGIN EC PRIVATE KEY-----
- MIGkAgEBBDCfMNsN6miEE3rVyUPwElfiJSWaR5huPCzUenZOfJT04GAcQdWvEju3
- UM2lmBLIXpGgBwYFK4EEACKhZANiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVq
- WvrJ51t5OxV0v25NsOgR82CANXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIV
- CzgWkxiz7XE4lgUwX44FCXZM3+JeUbI=
- -----END EC PRIVATE KEY-----`
- sftpPrivateKey = `-----BEGIN OPENSSH PRIVATE KEY-----
- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
- QyNTUxOQAAACB+RB4yNTZz9mHOkawwUibNdemijVV3ErMeLxWUBlCN/gAAAJA7DjpfOw46
- XwAAAAtzc2gtZWQyNTUxOQAAACB+RB4yNTZz9mHOkawwUibNdemijVV3ErMeLxWUBlCN/g
- AAAEA0E24gi8ab/XRSvJ85TGZJMe6HVmwxSG4ExPfTMwwe2n5EHjI1NnP2Yc6RrDBSJs11
- 6aKNVXcSsx4vFZQGUI3+AAAACW5pY29sYUBwMQECAwQ=
- -----END OPENSSH PRIVATE KEY-----`
- sftpPkeyFingerprint = "SHA256:QVQ06XHZZbYZzqfrsZcf3Yozy2WTnqQPeLOkcJCdbP0"
- // password protected private key
- testPrivateKeyPwd = `-----BEGIN OPENSSH PRIVATE KEY-----
- b3BlbnNzaC1rZXktdjEAAAAACmFlczI1Ni1jdHIAAAAGYmNyeXB0AAAAGAAAABAvfwQQcs
- +PyMsCLTNFcKiQAAAAEAAAAAEAAAAzAAAAC3NzaC1lZDI1NTE5AAAAILqltfCL7IPuIQ2q
- +8w23flfgskjIlKViEwMfjJR4mrbAAAAkHp5xgG8J1XW90M/fT59ZUQht8sZzzP17rEKlX
- waYKvLzDxkPK6LFIYs55W1EX1eVt/2Maq+zQ7k2SOUmhPNknsUOlPV2gytX3uIYvXF7u2F
- FTBIJuzZ+UQ14wFbraunliE9yye9DajVG1kz2cz2wVgXUbee+gp5NyFVvln+TcTxXwMsWD
- qwlk5iw/jQekxThg==
- -----END OPENSSH PRIVATE KEY-----
- `
- testPubKeyPwd = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAILqltfCL7IPuIQ2q+8w23flfgskjIlKViEwMfjJR4mrb"
- privateKeyPwd = "password"
- rsa1024PrivKey = `-----BEGIN OPENSSH PRIVATE KEY-----
- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAlwAAAAdzc2gtcn
- NhAAAAAwEAAQAAAIEAxgrZ84gJyU7Qz8JbYuYh0fgTN29h4qVkqDkEE0lWZe7L4QRcQHrB
- vycJO5vjfitY5JTojV3nbDNHN6XGVX8QNurwXmxv0EmEbqPoNO/rTf1t7qqwMBBAfSJJ5H
- TXsO37vqcWSOt1Ki5yjRm232UfPo3AYXaZdOKDWKpzI12FfqkAAAIAondFqKJ3RagAAAAH
- c3NoLXJzYQAAAIEAxgrZ84gJyU7Qz8JbYuYh0fgTN29h4qVkqDkEE0lWZe7L4QRcQHrBvy
- cJO5vjfitY5JTojV3nbDNHN6XGVX8QNurwXmxv0EmEbqPoNO/rTf1t7qqwMBBAfSJJ5HTX
- sO37vqcWSOt1Ki5yjRm232UfPo3AYXaZdOKDWKpzI12FfqkAAAADAQABAAAAgC7V5COG+a
- GFJTbtJQWnnTn17D2A9upN6RcrnL4e6vLiXY8So+qP3YAicDmLrWpqP/SXDsRX/+ID4oTT
- jKstiJy5jTvXAozwBbFCvNDk1qifs8p/HKzel3t0172j6gLOa2h9+clJ4BYyCk6ue4f8fV
- yKTIc9chdJSpeINNY60CJxAAAAQQDhYpGXljD2Xy/CzqRXyoF+iMtOImLlbgQYswTXegk3
- 7JoCNvwqg8xP+JxGpvUGpX23VWh0nBhzcAKHGlssiYQuAAAAQQDwB6s7s1WIRZ2Jsz8f6l
- 7/ebpPrAMyKmWkXc7KyvR53zuMkMIdvujM5NkOWh1ON8jtNumArey2dWuGVh+pXbdVAAAA
- QQDTOAaMcyTfXMH/oSMsp+5obvT/RuewaRLHdBiCy0y1Jw0ykOcOCkswr/btDL26hImaHF
- SheorO+2We7dnFuUIFAAAACW5pY29sYUBwMQE=
- -----END OPENSSH PRIVATE KEY-----`
- rsa1024PubKey = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDGCtnziAnJTtDPwlti5iHR+BM3b2HipWSoOQQTSVZl7svhBFxAesG/Jwk7m+N+K1jklOiNXedsM0c3pcZVfxA26vBebG/QSYRuo+g07+tN/W3uqrAwEEB9IknkdNew7fu+pxZI63UqLnKNGbbfZR8+jcBhdpl04oNYqnMjXYV+qQ=="
- redactedSecret = "[**redacted**]"
- osWindows = "windows"
- oidcMockAddr = "127.0.0.1:11111"
- )
- var (
- configDir = filepath.Join(".", "..", "..")
- defaultPerms = []string{dataprovider.PermAny}
- homeBasePath string
- backupsPath string
- testServer *httptest.Server
- postConnectPath string
- preActionPath string
- lastResetCode string
- )
- type fakeConnection struct {
- *common.BaseConnection
- command string
- }
- func (c *fakeConnection) Disconnect() error {
- common.Connections.Remove(c.GetID())
- return nil
- }
- func (c *fakeConnection) GetClientVersion() string {
- return ""
- }
- func (c *fakeConnection) GetCommand() string {
- return c.command
- }
- func (c *fakeConnection) GetLocalAddress() string {
- return ""
- }
- func (c *fakeConnection) GetRemoteAddress() string {
- return ""
- }
- type generateTOTPRequest struct {
- ConfigName string `json:"config_name"`
- }
- type generateTOTPResponse struct {
- ConfigName string `json:"config_name"`
- Issuer string `json:"issuer"`
- Secret string `json:"secret"`
- QRCode []byte `json:"qr_code"`
- }
- type validateTOTPRequest struct {
- ConfigName string `json:"config_name"`
- Passcode string `json:"passcode"`
- Secret string `json:"secret"`
- }
- type recoveryCode struct {
- Code string `json:"code"`
- Used bool `json:"used"`
- }
- func TestMain(m *testing.M) {
- homeBasePath = os.TempDir()
- logfilePath := filepath.Join(configDir, "sftpgo_api_test.log")
- logger.InitLogger(logfilePath, 5, 1, 28, false, false, zerolog.DebugLevel)
- os.Setenv("SFTPGO_COMMON__UPLOAD_MODE", "2")
- os.Setenv("SFTPGO_DATA_PROVIDER__CREATE_DEFAULT_ADMIN", "1")
- os.Setenv("SFTPGO_COMMON__ALLOW_SELF_CONNECTIONS", "1")
- os.Setenv("SFTPGO_DATA_PROVIDER__NAMING_RULES", "0")
- os.Setenv("SFTPGO_DEFAULT_ADMIN_USERNAME", "admin")
- os.Setenv("SFTPGO_DEFAULT_ADMIN_PASSWORD", "password")
- os.Setenv("SFTPGO_HTTPD__MAX_UPLOAD_FILE_SIZE", "1048576000")
- err := config.LoadConfig(configDir, "")
- if err != nil {
- logger.WarnToConsole("error loading configuration: %v", err)
- os.Exit(1)
- }
- wdPath, err := os.Getwd()
- if err != nil {
- logger.WarnToConsole("error getting exe path: %v", err)
- os.Exit(1)
- }
- pluginsConfig := []plugin.Config{
- {
- Type: "eventsearcher",
- Cmd: filepath.Join(wdPath, "..", "..", "tests", "eventsearcher", "eventsearcher"),
- AutoMTLS: true,
- },
- }
- if runtime.GOOS == osWindows {
- pluginsConfig[0].Cmd += ".exe"
- }
- providerConf := config.GetProviderConf()
- logger.InfoToConsole("Starting HTTPD tests, provider: %v", providerConf.Driver)
- backupsPath = filepath.Join(os.TempDir(), "test_backups")
- providerConf.BackupsPath = backupsPath
- err = os.MkdirAll(backupsPath, os.ModePerm)
- if err != nil {
- logger.ErrorToConsole("error creating backups path: %v", err)
- os.Exit(1)
- }
- err = dataprovider.Initialize(providerConf, configDir, true)
- if err != nil {
- logger.WarnToConsole("error initializing data provider: %v", err)
- os.Exit(1)
- }
- err = common.Initialize(config.GetCommonConfig(), 0)
- if err != nil {
- logger.WarnToConsole("error initializing common: %v", err)
- os.Exit(1)
- }
- postConnectPath = filepath.Join(homeBasePath, "postconnect.sh")
- preActionPath = filepath.Join(homeBasePath, "preaction.sh")
- httpConfig := config.GetHTTPConfig()
- httpConfig.RetryMax = 1
- httpConfig.Timeout = 5
- httpConfig.Initialize(configDir) //nolint:errcheck
- kmsConfig := config.GetKMSConfig()
- err = kmsConfig.Initialize()
- if err != nil {
- logger.ErrorToConsole("error initializing kms: %v", err)
- os.Exit(1)
- }
- mfaConfig := config.GetMFAConfig()
- err = mfaConfig.Initialize()
- if err != nil {
- logger.ErrorToConsole("error initializing MFA: %v", err)
- os.Exit(1)
- }
- err = plugin.Initialize(pluginsConfig, "debug")
- if err != nil {
- logger.ErrorToConsole("error initializing plugin: %v", err)
- os.Exit(1)
- }
- httpdConf := config.GetHTTPDConfig()
- httpdConf.Bindings[0].Port = 8081
- httpdConf.Bindings[0].Security = httpd.SecurityConf{
- Enabled: true,
- HTTPSProxyHeaders: []httpd.HTTPSProxyHeader{
- {
- Key: "X-Forwarded-Proto",
- Value: "https",
- },
- },
- }
- httpdtest.SetBaseURL(httpBaseURL)
- // required to test sftpfs
- sftpdConf := config.GetSFTPDConfig()
- sftpdConf.Bindings = []sftpd.Binding{
- {
- Port: 8022,
- },
- }
- hostKeyPath := filepath.Join(os.TempDir(), "id_rsa")
- sftpdConf.HostKeys = []string{hostKeyPath}
- go func() {
- if err := httpdConf.Initialize(configDir, 0); err != nil {
- logger.ErrorToConsole("could not start HTTP server: %v", err)
- os.Exit(1)
- }
- }()
- go func() {
- if err := sftpdConf.Initialize(configDir); err != nil {
- logger.ErrorToConsole("could not start SFTP server: %v", err)
- os.Exit(1)
- }
- }()
- startSMTPServer()
- startOIDCMockServer()
- waitTCPListening(httpdConf.Bindings[0].GetAddress())
- waitTCPListening(sftpdConf.Bindings[0].GetAddress())
- httpd.ReloadCertificateMgr() //nolint:errcheck
- // now start an https server
- certPath := filepath.Join(os.TempDir(), "test.crt")
- keyPath := filepath.Join(os.TempDir(), "test.key")
- err = os.WriteFile(certPath, []byte(httpsCert), os.ModePerm)
- if err != nil {
- logger.ErrorToConsole("error writing HTTPS certificate: %v", err)
- os.Exit(1)
- }
- err = os.WriteFile(keyPath, []byte(httpsKey), os.ModePerm)
- if err != nil {
- logger.ErrorToConsole("error writing HTTPS private key: %v", err)
- os.Exit(1)
- }
- httpdConf.Bindings[0].Port = 8443
- httpdConf.Bindings[0].EnableHTTPS = true
- httpdConf.Bindings[0].CertificateFile = certPath
- httpdConf.Bindings[0].CertificateKeyFile = keyPath
- httpdConf.Bindings = append(httpdConf.Bindings, httpd.Binding{})
- go func() {
- if err := httpdConf.Initialize(configDir, 0); err != nil {
- logger.ErrorToConsole("could not start HTTPS server: %v", err)
- os.Exit(1)
- }
- }()
- waitTCPListening(httpdConf.Bindings[0].GetAddress())
- httpd.ReloadCertificateMgr() //nolint:errcheck
- testServer = httptest.NewServer(httpd.GetHTTPRouter(httpdConf.Bindings[0]))
- defer testServer.Close()
- exitCode := m.Run()
- os.Remove(logfilePath)
- os.RemoveAll(backupsPath)
- os.Remove(certPath)
- os.Remove(keyPath)
- os.Remove(hostKeyPath)
- os.Remove(hostKeyPath + ".pub")
- os.Remove(postConnectPath)
- os.Remove(preActionPath)
- os.Exit(exitCode)
- }
- func TestInitialization(t *testing.T) {
- isShared := 0
- err := config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- invalidFile := "invalid file"
- passphraseFile := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- err = os.WriteFile(passphraseFile, []byte("my secret"), 0600)
- assert.NoError(t, err)
- defer os.Remove(passphraseFile)
- httpdConf := config.GetHTTPDConfig()
- httpdConf.SigningPassphraseFile = invalidFile
- err = httpdConf.Initialize(configDir, isShared)
- assert.ErrorIs(t, err, fs.ErrNotExist)
- httpdConf.SigningPassphraseFile = passphraseFile
- defaultTemplatesPath := httpdConf.TemplatesPath
- defaultStaticPath := httpdConf.StaticFilesPath
- httpdConf.CertificateFile = invalidFile
- httpdConf.CertificateKeyFile = invalidFile
- err = httpdConf.Initialize(configDir, isShared)
- assert.Error(t, err)
- httpdConf.CertificateFile = ""
- httpdConf.CertificateKeyFile = ""
- httpdConf.TemplatesPath = "."
- err = httpdConf.Initialize(configDir, isShared)
- assert.Error(t, err)
- httpdConf = config.GetHTTPDConfig()
- httpdConf.TemplatesPath = defaultTemplatesPath
- httpdConf.CertificateFile = invalidFile
- httpdConf.CertificateKeyFile = invalidFile
- httpdConf.StaticFilesPath = ""
- httpdConf.TemplatesPath = ""
- err = httpdConf.Initialize(configDir, isShared)
- assert.Error(t, err)
- httpdConf.StaticFilesPath = defaultStaticPath
- httpdConf.TemplatesPath = defaultTemplatesPath
- httpdConf.CertificateFile = filepath.Join(os.TempDir(), "test.crt")
- httpdConf.CertificateKeyFile = filepath.Join(os.TempDir(), "test.key")
- httpdConf.CACertificates = append(httpdConf.CACertificates, invalidFile)
- err = httpdConf.Initialize(configDir, isShared)
- assert.Error(t, err)
- httpdConf.CACertificates = nil
- httpdConf.CARevocationLists = append(httpdConf.CARevocationLists, invalidFile)
- err = httpdConf.Initialize(configDir, isShared)
- assert.Error(t, err)
- httpdConf.CARevocationLists = nil
- httpdConf.SigningPassphraseFile = passphraseFile
- httpdConf.Bindings[0].ProxyAllowed = []string{"invalid ip/network"}
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "is not a valid IP range")
- }
- assert.Equal(t, "my secret", httpdConf.SigningPassphrase)
- httpdConf.Bindings[0].ProxyAllowed = nil
- httpdConf.Bindings[0].EnableWebAdmin = false
- httpdConf.Bindings[0].EnableWebClient = false
- httpdConf.Bindings[0].Port = 8081
- httpdConf.Bindings[0].EnableHTTPS = true
- httpdConf.Bindings[0].ClientAuthType = 1
- httpdConf.TokenValidation = 1
- err = httpdConf.Initialize(configDir, 0)
- assert.Error(t, err)
- httpdConf.TokenValidation = 0
- err = httpdConf.Initialize(configDir, 0)
- assert.Error(t, err)
- httpdConf.Bindings[0].OIDC = httpd.OIDC{
- ClientID: "123",
- ClientSecret: "secret",
- ConfigURL: "http://127.0.0.1:11111",
- }
- err = httpdConf.Initialize(configDir, 0)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "oidc")
- }
- httpdConf.Bindings[0].OIDC.UsernameField = "preferred_username"
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "oidc")
- }
- httpdConf.Bindings[0].OIDC = httpd.OIDC{}
- httpdConf.Bindings[0].EnableWebClient = true
- httpdConf.Bindings[0].EnableWebAdmin = true
- httpdConf.Bindings[0].EnabledLoginMethods = 1
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "no login method available for WebAdmin UI")
- }
- httpdConf.Bindings[0].EnabledLoginMethods = 2
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "no login method available for WebAdmin UI")
- }
- httpdConf.Bindings[0].EnabledLoginMethods = 6
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "no login method available for WebClient UI")
- }
- httpdConf.Bindings[0].EnabledLoginMethods = 4
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "no login method available for WebClient UI")
- }
- httpdConf.Bindings[0].EnabledLoginMethods = 3
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "no login method available for WebAdmin UI")
- }
- httpdConf.Bindings[0].EnableWebAdmin = false
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "no login method available for WebClient UI")
- }
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = httpdConf.Initialize(configDir, isShared)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "unable to load config from provider")
- }
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestBasicUserHandling(t *testing.T) {
- u := getTestUser()
- u.Email = "user@user.com"
- user, resp, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- _, resp, err = httpdtest.AddUser(u, http.StatusConflict)
- assert.NoError(t, err, string(resp))
- lastPwdChange := user.LastPasswordChange
- assert.Greater(t, lastPwdChange, int64(0))
- user.MaxSessions = 10
- user.QuotaSize = 4096
- user.QuotaFiles = 2
- user.UploadBandwidth = 128
- user.DownloadBandwidth = 64
- user.ExpirationDate = util.GetTimeAsMsSinceEpoch(time.Now())
- user.AdditionalInfo = "some free text"
- user.Filters.TLSUsername = sdk.TLSUsernameCN
- user.Email = "user@example.net"
- user.OIDCCustomFields = &map[string]any{
- "field1": "value1",
- }
- user.Filters.WebClient = append(user.Filters.WebClient, sdk.WebClientPubKeyChangeDisabled,
- sdk.WebClientWriteDisabled)
- originalUser := user
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, originalUser.ID, user.ID)
- assert.Equal(t, lastPwdChange, user.LastPasswordChange)
- user, _, err = httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Nil(t, user.OIDCCustomFields)
- assert.True(t, user.HasPassword)
- user.Email = "invalid@email"
- user.FsConfig.OSConfig = sdk.OSFsConfig{
- ReadBufferSize: 1,
- WriteBufferSize: 2,
- }
- _, body, err := httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- assert.Contains(t, string(body), "Validation error: email")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestBasicRoleHandling(t *testing.T) {
- r := getTestRole()
- role, resp, err := httpdtest.AddRole(r, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Greater(t, role.CreatedAt, int64(0))
- assert.Greater(t, role.UpdatedAt, int64(0))
- roleGet, _, err := httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role, roleGet)
- roles, _, err := httpdtest.GetRoles(0, 0, http.StatusOK)
- assert.NoError(t, err)
- if assert.GreaterOrEqual(t, len(roles), 1) {
- found := false
- for _, ro := range roles {
- if ro.Name == r.Name {
- assert.Equal(t, role, ro)
- found = true
- }
- }
- assert.True(t, found)
- }
- roles, _, err = httpdtest.GetRoles(0, int64(len(roles)), http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, roles, 0)
- role.Description = "updated desc"
- _, _, err = httpdtest.UpdateRole(role, http.StatusOK)
- assert.NoError(t, err)
- roleGet, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Description, roleGet.Description)
- _, _, err = httpdtest.GetRoleByName(role.Name+"_", http.StatusNotFound)
- assert.NoError(t, err)
- // adding the same role again should fail
- _, _, err = httpdtest.AddRole(r, http.StatusConflict)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestRoleRelations(t *testing.T) {
- r := getTestRole()
- role, resp, err := httpdtest.AddRole(r, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Role = role.Name
- _, resp, err = httpdtest.AddAdmin(a, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "a role admin cannot have the following permissions")
- a.Permissions = []string{dataprovider.PermAdminAddUsers, dataprovider.PermAdminChangeUsers,
- dataprovider.PermAdminDeleteUsers, dataprovider.PermAdminViewUsers}
- a.Role = "missing admin role"
- _, _, err = httpdtest.AddAdmin(a, http.StatusConflict)
- assert.NoError(t, err)
- a.Role = role.Name
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- admin.Role = "invalid role"
- _, resp, err = httpdtest.UpdateAdmin(admin, http.StatusConflict)
- assert.NoError(t, err, string(resp))
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, admin.Role)
- resp, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.Error(t, err, "removing a referenced role should fail")
- assert.Contains(t, string(resp), "is referenced")
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, role.Admins, 1) {
- assert.Equal(t, admin.Username, role.Admins[0])
- }
- u1 := getTestUser()
- u1.Username = defaultUsername + "1"
- u1.Role = "missing role"
- _, _, err = httpdtest.AddUser(u1, http.StatusConflict)
- assert.NoError(t, err)
- u1.Role = role.Name
- user1, _, err := httpdtest.AddUser(u1, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user1.Role)
- user1.Role = "missing"
- _, _, err = httpdtest.UpdateUser(user1, http.StatusConflict, "")
- assert.NoError(t, err)
- user1, _, err = httpdtest.GetUserByUsername(user1.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user1.Role)
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, role.Admins, 1) {
- assert.Equal(t, admin.Username, role.Admins[0])
- }
- if assert.Len(t, role.Users, 1) {
- assert.Equal(t, user1.Username, role.Users[0])
- }
- roles, _, err := httpdtest.GetRoles(0, 0, http.StatusOK)
- assert.NoError(t, err)
- for _, r := range roles {
- if r.Name == role.Name {
- if assert.Len(t, role.Admins, 1) {
- assert.Equal(t, admin.Username, role.Admins[0])
- }
- if assert.Len(t, role.Users, 1) {
- assert.Equal(t, user1.Username, role.Users[0])
- }
- }
- }
- u2 := getTestUser()
- user2, _, err := httpdtest.AddUser(u2, http.StatusCreated)
- assert.NoError(t, err)
- // the global admin can list all users
- users, _, err := httpdtest.GetUsers(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(users), 2)
- _, _, err = httpdtest.GetUserByUsername(user1.Username, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetUserByUsername(user2.Username, http.StatusOK)
- assert.NoError(t, err)
- // the role admin can only list users with its role
- token, _, err := httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- users, _, err = httpdtest.GetUsers(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, users, 1)
- _, _, err = httpdtest.GetUserByUsername(user1.Username, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetUserByUsername(user2.Username, http.StatusNotFound)
- assert.NoError(t, err)
- // the role admin can only update/delete users with its role
- _, _, err = httpdtest.UpdateUser(user1, http.StatusOK, "")
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateUser(user2, http.StatusNotFound, "")
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user2, http.StatusNotFound)
- assert.NoError(t, err)
- // new users created by a role admin have the same role
- u3 := getTestUser()
- u3.Username = defaultUsername + "3"
- _, _, err = httpdtest.AddUser(u3, http.StatusCreated)
- if assert.Error(t, err) {
- assert.Equal(t, err.Error(), "role mismatch")
- }
- user3, _, err := httpdtest.GetUserByUsername(u3.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user3.Role)
- _, err = httpdtest.RemoveUser(user3, http.StatusOK)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Admins, []string{altAdminUsername})
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- user1, _, err = httpdtest.GetUserByUsername(user1.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user1.Role)
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user2, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestRSAKeyInvalidSize(t *testing.T) {
- u := getTestUser()
- u.PublicKeys = append(u.PublicKeys, rsa1024PubKey)
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "invalid size")
- u = getTestSFTPUser()
- u.FsConfig.SFTPConfig.Password = nil
- u.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret(rsa1024PrivKey)
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "rsa key with size 1024 not accepted")
- }
- func TestTLSCert(t *testing.T) {
- u := getTestUser()
- u.Filters.TLSCerts = []string{"not a cert"}
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "invalid TLS certificate")
- u.Filters.TLSCerts = []string{httpsCert}
- user, resp, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- if assert.Len(t, user.Filters.TLSCerts, 1) {
- assert.Equal(t, httpsCert, user.Filters.TLSCerts[0])
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestBasicGroupHandling(t *testing.T) {
- g := getTestGroup()
- g.UserSettings.Filters.TLSCerts = []string{"invalid cert"} // ignored for groups
- group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
- assert.NoError(t, err)
- assert.Greater(t, group.CreatedAt, int64(0))
- assert.Greater(t, group.UpdatedAt, int64(0))
- assert.Len(t, group.UserSettings.Filters.TLSCerts, 0)
- groupGet, _, err := httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, group, groupGet)
- groups, _, err := httpdtest.GetGroups(0, 0, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, groups, 1) {
- assert.Equal(t, group, groups[0])
- }
- groups, _, err = httpdtest.GetGroups(0, 1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, groups, 0)
- _, _, err = httpdtest.GetGroupByName(group.Name+"_", http.StatusNotFound)
- assert.NoError(t, err)
- // adding the same group again should fail
- _, _, err = httpdtest.AddGroup(g, http.StatusConflict)
- assert.NoError(t, err)
- group.UserSettings.HomeDir = filepath.Join(os.TempDir(), "%username%")
- group.UserSettings.FsConfig.Provider = sdk.SFTPFilesystemProvider
- group.UserSettings.FsConfig.SFTPConfig.Endpoint = sftpServerAddr
- group.UserSettings.FsConfig.SFTPConfig.Username = defaultUsername
- group.UserSettings.FsConfig.SFTPConfig.Password = kms.NewPlainSecret(defaultPassword)
- group.UserSettings.Permissions = map[string][]string{
- "/": {dataprovider.PermAny},
- }
- group.UserSettings.Filters.AllowedIP = []string{"10.0.0.0/8"}
- group, _, err = httpdtest.UpdateGroup(group, http.StatusOK)
- assert.NoError(t, err)
- groupGet, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, groupGet.UserSettings.Permissions, 1)
- assert.Len(t, groupGet.UserSettings.Filters.AllowedIP, 1)
- // update again and check that the password was preserved
- dbGroup, err := dataprovider.GroupExists(group.Name)
- assert.NoError(t, err)
- group.UserSettings.FsConfig.SFTPConfig.Password = kms.NewSecret(
- dbGroup.UserSettings.FsConfig.SFTPConfig.Password.GetStatus(),
- dbGroup.UserSettings.FsConfig.SFTPConfig.Password.GetPayload(), "", "")
- group.UserSettings.Permissions = nil
- group.UserSettings.Filters.AllowedIP = nil
- group, _, err = httpdtest.UpdateGroup(group, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group.UserSettings.Permissions, 0)
- assert.Len(t, group.UserSettings.Filters.AllowedIP, 0)
- dbGroup, err = dataprovider.GroupExists(group.Name)
- assert.NoError(t, err)
- err = dbGroup.UserSettings.FsConfig.SFTPConfig.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, defaultPassword, dbGroup.UserSettings.FsConfig.SFTPConfig.Password.GetPayload())
- // check the group permissions
- groupGet, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, groupGet.UserSettings.Permissions, 0)
- group.UserSettings.HomeDir = "relative path"
- _, _, err = httpdtest.UpdateGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateGroup(group, http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestGroupRelations(t *testing.T) {
- mappedPath1 := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- folderName1 := filepath.Base(mappedPath1)
- mappedPath2 := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- folderName2 := filepath.Base(mappedPath2)
- _, _, err := httpdtest.AddFolder(vfs.BaseVirtualFolder{
- Name: folderName2,
- MappedPath: mappedPath2,
- FsConfig: vfs.Filesystem{
- OSConfig: sdk.OSFsConfig{
- ReadBufferSize: 3,
- WriteBufferSize: 5,
- },
- },
- }, http.StatusCreated)
- assert.NoError(t, err)
- g1 := getTestGroup()
- g1.Name += "_1"
- g1.VirtualFolders = append(g1.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir1",
- })
- g2 := getTestGroup()
- g2.Name += "_2"
- g2.VirtualFolders = append(g2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir2",
- })
- g3 := getTestGroup()
- g3.Name += "_3"
- g3.VirtualFolders = append(g3.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir3",
- })
- _, _, err = httpdtest.AddGroup(g1, http.StatusCreated)
- assert.Error(t, err, "adding a group with a missing folder must fail")
- _, _, err = httpdtest.AddFolder(vfs.BaseVirtualFolder{
- Name: folderName1,
- MappedPath: mappedPath1,
- }, http.StatusCreated)
- assert.NoError(t, err)
- group1, resp, err := httpdtest.AddGroup(g1, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, group1.VirtualFolders, 1)
- group2, resp, err := httpdtest.AddGroup(g2, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, group2.VirtualFolders, 1)
- group3, resp, err := httpdtest.AddGroup(g3, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, group3.VirtualFolders, 1)
- folder1, _, err := httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder1.Groups, 3)
- folder2, _, err := httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder2.Groups, 0)
- group1.VirtualFolders = append(group1.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: folder2,
- VirtualPath: "/vfolder2",
- })
- group1, _, err = httpdtest.UpdateGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.VirtualFolders, 2)
- folder2, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder2.Groups, 1)
- group1.VirtualFolders = []vfs.VirtualFolder{
- {
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folder1.Name,
- MappedPath: folder1.MappedPath,
- },
- VirtualPath: "/vpathmod",
- },
- }
- group1, _, err = httpdtest.UpdateGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.VirtualFolders, 1)
- folder2, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder2.Groups, 0)
- group1.VirtualFolders = append(group1.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: folder2,
- VirtualPath: "/vfolder2mod",
- })
- group1, _, err = httpdtest.UpdateGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.VirtualFolders, 2)
- folder2, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder2.Groups, 1)
- u := getTestUser()
- u.Groups = []sdk.GroupMapping{
- {
- Name: group1.Name,
- Type: sdk.GroupTypePrimary,
- },
- {
- Name: group2.Name,
- Type: sdk.GroupTypeSecondary,
- },
- {
- Name: group3.Name,
- Type: sdk.GroupTypeSecondary,
- },
- }
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- if assert.Len(t, user.Groups, 3) {
- for _, g := range user.Groups {
- if g.Name == group1.Name {
- assert.Equal(t, sdk.GroupTypePrimary, g.Type)
- } else {
- assert.Equal(t, sdk.GroupTypeSecondary, g.Type)
- }
- }
- }
- group1, _, err = httpdtest.GetGroupByName(group1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.Users, 1)
- group2, _, err = httpdtest.GetGroupByName(group2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group2.Users, 1)
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Users, 1)
- user.Groups = []sdk.GroupMapping{
- {
- Name: group3.Name,
- Type: sdk.GroupTypeSecondary,
- },
- }
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Len(t, user.Groups, 1)
- group1, _, err = httpdtest.GetGroupByName(group1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.Users, 0)
- group2, _, err = httpdtest.GetGroupByName(group2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group2.Users, 0)
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Users, 1)
- user.Groups = []sdk.GroupMapping{
- {
- Name: group1.Name,
- Type: sdk.GroupTypePrimary,
- },
- {
- Name: group2.Name,
- Type: sdk.GroupTypeSecondary,
- },
- {
- Name: group3.Name,
- Type: sdk.GroupTypeSecondary,
- },
- }
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Len(t, user.Groups, 3)
- group1, _, err = httpdtest.GetGroupByName(group1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.Users, 1)
- group2, _, err = httpdtest.GetGroupByName(group2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group2.Users, 1)
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Users, 1)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(folder1, http.StatusOK)
- assert.NoError(t, err)
- group1, _, err = httpdtest.GetGroupByName(group1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group1.Users, 0)
- assert.Len(t, group1.VirtualFolders, 1)
- group2, _, err = httpdtest.GetGroupByName(group2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group2.Users, 0)
- assert.Len(t, group2.VirtualFolders, 0)
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Users, 0)
- assert.Len(t, group3.VirtualFolders, 0)
- _, err = httpdtest.RemoveGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group3, http.StatusOK)
- assert.NoError(t, err)
- folder2, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder2.Groups, 0)
- _, err = httpdtest.RemoveFolder(folder2, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestGroupValidation(t *testing.T) {
- group := getTestGroup()
- group.VirtualFolders = []vfs.VirtualFolder{
- {
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: util.GenerateUniqueID(),
- MappedPath: filepath.Join(os.TempDir(), util.GenerateUniqueID()),
- },
- },
- }
- _, resp, err := httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "virtual path is mandatory")
- group.VirtualFolders = nil
- group.UserSettings.FsConfig.Provider = sdk.SFTPFilesystemProvider
- _, resp, err = httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "endpoint cannot be empty")
- group.UserSettings.FsConfig.Provider = sdk.LocalFilesystemProvider
- group.UserSettings.Permissions = map[string][]string{
- "a": nil,
- }
- _, resp, err = httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "cannot set permissions for non absolute path")
- group.UserSettings.Permissions = map[string][]string{
- "/": nil,
- }
- _, resp, err = httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "no permissions granted")
- group.UserSettings.Permissions = map[string][]string{
- "/..": nil,
- }
- _, resp, err = httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "cannot set permissions for invalid subdirectory")
- group.UserSettings.Permissions = map[string][]string{
- "/": {dataprovider.PermAny},
- }
- group.UserSettings.HomeDir = "relative"
- _, resp, err = httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "home_dir must be an absolute path")
- group.UserSettings.HomeDir = ""
- group.UserSettings.Filters.WebClient = []string{"invalid permission"}
- _, resp, err = httpdtest.AddGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid web client options")
- }
- func TestGroupSettingsOverride(t *testing.T) {
- mappedPath1 := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- folderName1 := filepath.Base(mappedPath1)
- mappedPath2 := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- folderName2 := filepath.Base(mappedPath2)
- mappedPath3 := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- folderName3 := filepath.Base(mappedPath3)
- g1 := getTestGroup()
- g1.Name += "_1"
- g1.VirtualFolders = append(g1.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir1",
- })
- g1.UserSettings.Permissions = map[string][]string{
- "/dir1": {dataprovider.PermUpload},
- "/dir2": {dataprovider.PermDownload, dataprovider.PermListItems},
- }
- g1.UserSettings.FsConfig.OSConfig = sdk.OSFsConfig{
- ReadBufferSize: 6,
- WriteBufferSize: 2,
- }
- g2 := getTestGroup()
- g2.Name += "_2"
- g2.UserSettings.Permissions = map[string][]string{
- "/dir1": {dataprovider.PermAny},
- "/dir3": {dataprovider.PermDownload, dataprovider.PermListItems, dataprovider.PermChtimes},
- }
- g2.VirtualFolders = append(g2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir2",
- })
- g2.VirtualFolders = append(g2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName2,
- },
- VirtualPath: "/vdir3",
- })
- g2.VirtualFolders = append(g2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName3,
- },
- VirtualPath: "/vdir4",
- })
- g2.UserSettings.Filters.AccessTime = []sdk.TimePeriod{
- {
- DayOfWeek: int(time.Now().UTC().Weekday()),
- From: "00:00",
- To: "23:59",
- },
- }
- f1 := vfs.BaseVirtualFolder{
- Name: folderName1,
- MappedPath: mappedPath1,
- FsConfig: vfs.Filesystem{
- OSConfig: sdk.OSFsConfig{
- ReadBufferSize: 3,
- WriteBufferSize: 5,
- },
- },
- }
- _, _, err := httpdtest.AddFolder(f1, http.StatusCreated)
- assert.NoError(t, err)
- f2 := vfs.BaseVirtualFolder{
- Name: folderName2,
- MappedPath: mappedPath2,
- }
- _, _, err = httpdtest.AddFolder(f2, http.StatusCreated)
- assert.NoError(t, err)
- f3 := vfs.BaseVirtualFolder{
- Name: folderName3,
- MappedPath: mappedPath3,
- FsConfig: vfs.Filesystem{
- OSConfig: sdk.OSFsConfig{
- ReadBufferSize: 1,
- WriteBufferSize: 2,
- },
- },
- }
- _, _, err = httpdtest.AddFolder(f3, http.StatusCreated)
- assert.NoError(t, err)
- group1, resp, err := httpdtest.AddGroup(g1, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- group2, resp, err := httpdtest.AddGroup(g2, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- u := getTestUser()
- u.Groups = []sdk.GroupMapping{
- {
- Name: group1.Name,
- Type: sdk.GroupTypePrimary,
- },
- {
- Name: group2.Name,
- Type: sdk.GroupTypeSecondary,
- },
- }
- r := getTestRole()
- role, _, err := httpdtest.AddRole(r, http.StatusCreated)
- assert.NoError(t, err)
- u.Role = role.Name
- user, resp, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, user.VirtualFolders, 0)
- assert.Len(t, user.Permissions, 1)
- user, err = dataprovider.CheckUserAndPass(defaultUsername, defaultPassword, "", common.ProtocolHTTP)
- assert.NoError(t, err)
- var folderNames []string
- if assert.Len(t, user.VirtualFolders, 4) {
- for _, f := range user.VirtualFolders {
- if !util.Contains(folderNames, f.Name) {
- folderNames = append(folderNames, f.Name)
- }
- switch f.Name {
- case folderName1:
- assert.Equal(t, mappedPath1, f.MappedPath)
- assert.Equal(t, 3, f.BaseVirtualFolder.FsConfig.OSConfig.ReadBufferSize)
- assert.Equal(t, 5, f.BaseVirtualFolder.FsConfig.OSConfig.WriteBufferSize)
- assert.True(t, util.Contains([]string{"/vdir1", "/vdir2"}, f.VirtualPath))
- case folderName2:
- assert.Equal(t, mappedPath2, f.MappedPath)
- assert.Equal(t, "/vdir3", f.VirtualPath)
- assert.Equal(t, 0, f.BaseVirtualFolder.FsConfig.OSConfig.ReadBufferSize)
- assert.Equal(t, 0, f.BaseVirtualFolder.FsConfig.OSConfig.WriteBufferSize)
- case folderName3:
- assert.Equal(t, mappedPath3, f.MappedPath)
- assert.Equal(t, "/vdir4", f.VirtualPath)
- assert.Equal(t, 1, f.BaseVirtualFolder.FsConfig.OSConfig.ReadBufferSize)
- assert.Equal(t, 2, f.BaseVirtualFolder.FsConfig.OSConfig.WriteBufferSize)
- }
- }
- }
- assert.Len(t, folderNames, 3)
- assert.Contains(t, folderNames, folderName1)
- assert.Contains(t, folderNames, folderName2)
- assert.Contains(t, folderNames, folderName3)
- assert.Len(t, user.Permissions, 4)
- assert.Equal(t, g1.UserSettings.Permissions["/dir1"], user.Permissions["/dir1"])
- assert.Equal(t, g1.UserSettings.Permissions["/dir2"], user.Permissions["/dir2"])
- assert.Equal(t, g2.UserSettings.Permissions["/dir3"], user.Permissions["/dir3"])
- assert.Equal(t, g1.UserSettings.FsConfig.OSConfig.ReadBufferSize, user.FsConfig.OSConfig.ReadBufferSize)
- assert.Equal(t, g1.UserSettings.FsConfig.OSConfig.WriteBufferSize, user.FsConfig.OSConfig.WriteBufferSize)
- assert.Len(t, user.Filters.AccessTime, 1)
- user, err = dataprovider.GetUserAfterIDPAuth(defaultUsername, "", common.ProtocolOIDC, nil)
- assert.NoError(t, err)
- assert.Len(t, user.VirtualFolders, 4)
- assert.Len(t, user.Filters.AccessTime, 1)
- user1, user2, err := dataprovider.GetUserVariants(defaultUsername, "")
- assert.NoError(t, err)
- assert.Len(t, user1.VirtualFolders, 0)
- assert.Len(t, user2.VirtualFolders, 4)
- assert.Equal(t, int64(0), user1.ExpirationDate)
- assert.Equal(t, int64(0), user2.ExpirationDate)
- assert.Len(t, user1.Filters.AccessTime, 0)
- assert.Len(t, user2.Filters.AccessTime, 1)
- group2.UserSettings.FsConfig = vfs.Filesystem{
- Provider: sdk.SFTPFilesystemProvider,
- SFTPConfig: vfs.SFTPFsConfig{
- BaseSFTPFsConfig: sdk.BaseSFTPFsConfig{
- Endpoint: sftpServerAddr,
- Username: defaultUsername,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- },
- }
- group2.UserSettings.Permissions = map[string][]string{
- "/": {dataprovider.PermListItems, dataprovider.PermDownload},
- "/%username%": {dataprovider.PermListItems},
- }
- group2.UserSettings.DownloadBandwidth = 128
- group2.UserSettings.UploadBandwidth = 256
- group2.UserSettings.Filters.PasswordStrength = 70
- group2.UserSettings.Filters.WebClient = []string{sdk.WebClientInfoChangeDisabled, sdk.WebClientMFADisabled}
- _, _, err = httpdtest.UpdateGroup(group2, http.StatusOK)
- assert.NoError(t, err)
- user, err = dataprovider.CheckUserAndPass(defaultUsername, defaultPassword, "", common.ProtocolHTTP)
- assert.NoError(t, err)
- assert.Len(t, user.VirtualFolders, 4)
- assert.Equal(t, sdk.LocalFilesystemProvider, user.FsConfig.Provider)
- assert.Equal(t, int64(0), user.DownloadBandwidth)
- assert.Equal(t, int64(0), user.UploadBandwidth)
- assert.Equal(t, 0, user.Filters.PasswordStrength)
- assert.Equal(t, []string{dataprovider.PermAny}, user.GetPermissionsForPath("/"))
- assert.Equal(t, []string{dataprovider.PermListItems}, user.GetPermissionsForPath("/"+defaultUsername))
- assert.Len(t, user.Filters.WebClient, 2)
- group1.UserSettings.FsConfig = vfs.Filesystem{
- Provider: sdk.SFTPFilesystemProvider,
- SFTPConfig: vfs.SFTPFsConfig{
- BaseSFTPFsConfig: sdk.BaseSFTPFsConfig{
- Endpoint: sftpServerAddr,
- Username: altAdminUsername,
- Prefix: "/dirs/%role%/%username%",
- },
- Password: kms.NewPlainSecret(defaultPassword),
- },
- }
- group1.UserSettings.MaxSessions = 2
- group1.UserSettings.QuotaFiles = 1000
- group1.UserSettings.UploadBandwidth = 512
- group1.UserSettings.DownloadBandwidth = 1024
- group1.UserSettings.TotalDataTransfer = 2048
- group1.UserSettings.ExpiresIn = 15
- group1.UserSettings.Filters.MaxUploadFileSize = 1024 * 1024
- group1.UserSettings.Filters.StartDirectory = "/startdir/%username%"
- group1.UserSettings.Filters.PasswordStrength = 70
- group1.UserSettings.Filters.WebClient = []string{sdk.WebClientInfoChangeDisabled}
- group1.UserSettings.Permissions = map[string][]string{
- "/": {dataprovider.PermListItems, dataprovider.PermUpload},
- "/sub/%username%": {dataprovider.PermRename},
- "/%role%/%username%": {dataprovider.PermDelete},
- }
- group1.UserSettings.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/sub2/%role%/%username%test",
- AllowedPatterns: []string{},
- DeniedPatterns: []string{"*.jpg", "*.zip"},
- },
- }
- _, _, err = httpdtest.UpdateGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- user, err = dataprovider.CheckUserAndPass(defaultUsername, defaultPassword, "", common.ProtocolHTTP)
- assert.NoError(t, err)
- assert.Len(t, user.VirtualFolders, 4)
- assert.Equal(t, user.CreatedAt+int64(group1.UserSettings.ExpiresIn)*86400000, user.ExpirationDate)
- assert.Equal(t, group1.UserSettings.Filters.PasswordStrength, user.Filters.PasswordStrength)
- assert.Equal(t, sdk.SFTPFilesystemProvider, user.FsConfig.Provider)
- assert.Equal(t, altAdminUsername, user.FsConfig.SFTPConfig.Username)
- assert.Equal(t, "/dirs/"+role.Name+"/"+defaultUsername, user.FsConfig.SFTPConfig.Prefix)
- assert.Equal(t, []string{dataprovider.PermListItems, dataprovider.PermUpload}, user.GetPermissionsForPath("/"))
- assert.Equal(t, []string{dataprovider.PermDelete}, user.GetPermissionsForPath(path.Join("/", role.Name, defaultUsername)))
- assert.Equal(t, []string{dataprovider.PermRename}, user.GetPermissionsForPath(path.Join("/sub", defaultUsername)))
- assert.Equal(t, group1.UserSettings.MaxSessions, user.MaxSessions)
- assert.Equal(t, group1.UserSettings.QuotaFiles, user.QuotaFiles)
- assert.Equal(t, group1.UserSettings.UploadBandwidth, user.UploadBandwidth)
- assert.Equal(t, group1.UserSettings.TotalDataTransfer, user.TotalDataTransfer)
- assert.Equal(t, group1.UserSettings.Filters.MaxUploadFileSize, user.Filters.MaxUploadFileSize)
- assert.Equal(t, "/startdir/"+defaultUsername, user.Filters.StartDirectory)
- if assert.Len(t, user.Filters.FilePatterns, 1) {
- assert.Equal(t, "/sub2/"+role.Name+"/"+defaultUsername+"test", user.Filters.FilePatterns[0].Path) //nolint:goconst
- }
- if assert.Len(t, user.Filters.WebClient, 2) {
- assert.Contains(t, user.Filters.WebClient, sdk.WebClientInfoChangeDisabled)
- assert.Contains(t, user.Filters.WebClient, sdk.WebClientMFADisabled)
- }
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName1}, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName2}, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName3}, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestConfigs(t *testing.T) {
- err := dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- configs, err := dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Equal(t, int64(0), configs.UpdatedAt)
- assert.Nil(t, configs.SFTPD)
- assert.Nil(t, configs.SMTP)
- configs = dataprovider.Configs{
- SFTPD: &dataprovider.SFTPDConfigs{},
- SMTP: &dataprovider.SMTPConfigs{},
- }
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- assert.NoError(t, err)
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Greater(t, configs.UpdatedAt, int64(0))
- configs = dataprovider.Configs{
- SFTPD: &dataprovider.SFTPDConfigs{
- Ciphers: []string{"unknown"},
- },
- SMTP: &dataprovider.SMTPConfigs{},
- }
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- assert.ErrorIs(t, err, util.ErrValidation)
- configs = dataprovider.Configs{
- SFTPD: &dataprovider.SFTPDConfigs{},
- SMTP: &dataprovider.SMTPConfigs{
- Host: "smtp.example.com",
- Port: -1,
- },
- }
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- assert.ErrorIs(t, err, util.ErrValidation)
- configs = dataprovider.Configs{
- SMTP: &dataprovider.SMTPConfigs{
- Host: "mail.example.com",
- Port: 587,
- User: "test@example.com",
- AuthType: 3,
- Encryption: 2,
- OAuth2: dataprovider.SMTPOAuth2{
- Provider: 1,
- Tenant: "",
- ClientID: "",
- },
- },
- }
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- if assert.ErrorIs(t, err, util.ErrValidation) {
- assert.Contains(t, err.Error(), "smtp oauth2: client id is required")
- }
- configs.SMTP.OAuth2 = dataprovider.SMTPOAuth2{
- Provider: 1,
- ClientID: "client id",
- ClientSecret: kms.NewPlainSecret("client secret"),
- RefreshToken: kms.NewPlainSecret("refresh token"),
- }
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- assert.NoError(t, err)
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Equal(t, 3, configs.SMTP.AuthType)
- assert.Equal(t, 1, configs.SMTP.OAuth2.Provider)
- err = dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- }
- func TestBasicIPListEntriesHandling(t *testing.T) {
- entry := dataprovider.IPListEntry{
- IPOrNet: "::ffff:12.34.56.78",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Description: "test desc",
- }
- _, _, err := httpdtest.GetIPListEntry(entry.IPOrNet, -1, http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateIPListEntry(entry, http.StatusNotFound)
- assert.NoError(t, err)
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusCreated)
- assert.Error(t, err)
- // IPv4 address in IPv6 will be converted to standard IPv4
- entry1, _, err := httpdtest.GetIPListEntry("12.34.56.78/32", dataprovider.IPListTypeAllowList, http.StatusOK)
- assert.NoError(t, err)
- entry = dataprovider.IPListEntry{
- IPOrNet: "192.168.0.0/24",
- Type: dataprovider.IPListTypeDefender,
- Mode: dataprovider.ListModeDeny,
- }
- entry2, _, err := httpdtest.AddIPListEntry(entry, http.StatusCreated)
- assert.NoError(t, err)
- // adding the same entry again should fail
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusConflict)
- assert.NoError(t, err)
- // adding an entry with an invalid IP should fail
- entry.IPOrNet = "not valid"
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- // adding an entry with an incompatible mode should fail
- entry.IPOrNet = entry2.IPOrNet
- entry.Mode = -1
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- entry.Type = -1
- _, _, err = httpdtest.UpdateIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- entry = dataprovider.IPListEntry{
- IPOrNet: "2001:4860:4860::8888/120",
- Type: dataprovider.IPListTypeRateLimiterSafeList,
- Mode: dataprovider.ListModeDeny,
- }
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- entry.Mode = dataprovider.ListModeAllow
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusCreated)
- assert.NoError(t, err)
- entry.Protocols = 3
- entry3, _, err := httpdtest.UpdateIPListEntry(entry, http.StatusOK)
- assert.NoError(t, err)
- entry.Mode = dataprovider.ListModeDeny
- _, _, err = httpdtest.UpdateIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- for _, tt := range []dataprovider.IPListType{dataprovider.IPListTypeAllowList, dataprovider.IPListTypeDefender, dataprovider.IPListTypeRateLimiterSafeList} {
- entries, _, err := httpdtest.GetIPListEntries(tt, "", "", dataprovider.OrderASC, 0, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, entries, 1) {
- switch tt {
- case dataprovider.IPListTypeAllowList:
- assert.Equal(t, entry1, entries[0])
- case dataprovider.IPListTypeDefender:
- assert.Equal(t, entry2, entries[0])
- case dataprovider.IPListTypeRateLimiterSafeList:
- assert.Equal(t, entry3, entries[0])
- }
- }
- }
- _, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "", "", "invalid order", 0, http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetIPListEntries(-1, "", "", dataprovider.OrderASC, 0, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry1, http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry2, http.StatusOK)
- assert.NoError(t, err)
- entry2.Type = -1
- _, err = httpdtest.RemoveIPListEntry(entry2, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry3, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestSearchIPListEntries(t *testing.T) {
- entries := []dataprovider.IPListEntry{
- {
- IPOrNet: "192.168.0.0/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 0,
- },
- {
- IPOrNet: "192.168.0.1/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 0,
- },
- {
- IPOrNet: "192.168.0.2/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 5,
- },
- {
- IPOrNet: "192.168.0.3/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 8,
- },
- {
- IPOrNet: "10.8.0.0/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 3,
- },
- {
- IPOrNet: "10.8.1.0/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 8,
- },
- {
- IPOrNet: "10.8.2.0/24",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 1,
- },
- }
- for _, e := range entries {
- _, _, err := httpdtest.AddIPListEntry(e, http.StatusCreated)
- assert.NoError(t, err)
- }
- results, _, err := httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "", "", dataprovider.OrderASC, 20, http.StatusOK)
- assert.NoError(t, err)
- if assert.Equal(t, len(entries), len(results)) {
- assert.Equal(t, "10.8.0.0/24", results[0].IPOrNet)
- }
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "", "", dataprovider.OrderDESC, 20, http.StatusOK)
- assert.NoError(t, err)
- if assert.Equal(t, len(entries), len(results)) {
- assert.Equal(t, "192.168.0.3/24", results[0].IPOrNet)
- }
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "", "192.168.0.1/24", dataprovider.OrderASC, 1, http.StatusOK)
- assert.NoError(t, err)
- if assert.Equal(t, 1, len(results), results) {
- assert.Equal(t, "192.168.0.2/24", results[0].IPOrNet)
- }
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "", "10.8.2.0/24", dataprovider.OrderDESC, 1, http.StatusOK)
- assert.NoError(t, err)
- if assert.Equal(t, 1, len(results), results) {
- assert.Equal(t, "10.8.1.0/24", results[0].IPOrNet)
- }
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "10.", "", dataprovider.OrderASC, 20, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 3, len(results))
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "192", "", dataprovider.OrderASC, 20, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 4, len(results))
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "1", "", dataprovider.OrderASC, 20, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 7, len(results))
- results, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeAllowList, "108", "", dataprovider.OrderASC, 20, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 0, len(results))
- for _, e := range entries {
- _, err := httpdtest.RemoveIPListEntry(e, http.StatusOK)
- assert.NoError(t, err)
- }
- }
- func TestIPListEntriesValidation(t *testing.T) {
- entry := dataprovider.IPListEntry{
- IPOrNet: "::ffff:34.56.78.90/120",
- Type: -1,
- Mode: dataprovider.ListModeDeny,
- }
- _, resp, err := httpdtest.AddIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid list type")
- entry.Type = dataprovider.IPListTypeRateLimiterSafeList
- _, resp, err = httpdtest.AddIPListEntry(entry, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid list mode")
- entry.Type = dataprovider.IPListTypeDefender
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusCreated)
- assert.Error(t, err)
- entry.IPOrNet = "34.56.78.0/24"
- _, err = httpdtest.RemoveIPListEntry(entry, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestBasicActionRulesHandling(t *testing.T) {
- actionName := "test action"
- a := dataprovider.BaseEventAction{
- Name: actionName,
- Description: "test description",
- Type: dataprovider.ActionTypeBackup,
- Options: dataprovider.BaseEventActionOptions{},
- }
- action, _, err := httpdtest.AddEventAction(a, http.StatusCreated)
- assert.NoError(t, err)
- // adding the same action should fail
- _, _, err = httpdtest.AddEventAction(a, http.StatusConflict)
- assert.NoError(t, err)
- actionGet, _, err := httpdtest.GetEventActionByName(actionName, http.StatusOK)
- assert.NoError(t, err)
- actions, _, err := httpdtest.GetEventActions(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, len(actions), 0)
- found := false
- for _, ac := range actions {
- if ac.Name == actionName {
- assert.Equal(t, actionGet, ac)
- found = true
- }
- }
- assert.True(t, found)
- a.Description = "new description"
- a.Type = dataprovider.ActionTypeDataRetentionCheck
- a.Options = dataprovider.BaseEventActionOptions{
- RetentionConfig: dataprovider.EventActionDataRetentionConfig{
- Folders: []dataprovider.FolderRetention{
- {
- Path: "/",
- Retention: 144,
- },
- {
- Path: "/p1",
- Retention: 0,
- },
- {
- Path: "/p2",
- Retention: 12,
- },
- },
- },
- }
- _, _, err = httpdtest.UpdateEventAction(a, http.StatusOK)
- assert.NoError(t, err)
- a.Type = dataprovider.ActionTypeCommand
- a.Options = dataprovider.BaseEventActionOptions{
- CmdConfig: dataprovider.EventActionCommandConfig{
- Cmd: filepath.Join(os.TempDir(), "test_cmd"),
- Timeout: 20,
- EnvVars: []dataprovider.KeyValue{
- {
- Key: "NAME",
- Value: "VALUE",
- },
- },
- },
- }
- _, _, err = httpdtest.UpdateEventAction(a, http.StatusOK)
- assert.NoError(t, err)
- // invalid type
- a.Type = 1000
- _, _, err = httpdtest.UpdateEventAction(a, http.StatusBadRequest)
- assert.NoError(t, err)
- a.Type = dataprovider.ActionTypeEmail
- a.Options = dataprovider.BaseEventActionOptions{
- EmailConfig: dataprovider.EventActionEmailConfig{
- Recipients: []string{"email@example.com"},
- Bcc: []string{"bcc@example.com"},
- Subject: "Event: {{Event}}",
- Body: "test mail body",
- Attachments: []string{"/{{VirtualPath}}"},
- },
- }
- _, _, err = httpdtest.UpdateEventAction(a, http.StatusOK)
- assert.NoError(t, err)
- a.Type = dataprovider.ActionTypeUserInactivityCheck
- a.Options = dataprovider.BaseEventActionOptions{
- UserInactivityConfig: dataprovider.EventActionUserInactivity{
- DisableThreshold: 10,
- DeleteThreshold: 20,
- },
- }
- _, _, err = httpdtest.UpdateEventAction(a, http.StatusOK)
- assert.NoError(t, err)
- a.Type = dataprovider.ActionTypeHTTP
- a.Options = dataprovider.BaseEventActionOptions{
- HTTPConfig: dataprovider.EventActionHTTPConfig{
- Endpoint: "https://localhost:1234",
- Username: defaultUsername,
- Password: kms.NewPlainSecret(defaultPassword),
- Headers: []dataprovider.KeyValue{
- {
- Key: "Content-Type",
- Value: "application/json",
- },
- },
- Timeout: 10,
- SkipTLSVerify: true,
- Method: http.MethodPost,
- QueryParameters: []dataprovider.KeyValue{
- {
- Key: "a",
- Value: "b",
- },
- },
- Body: `{"event":"{{Event}}","name":"{{Name}}"}`,
- },
- }
- action, _, err = httpdtest.UpdateEventAction(a, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, action.Options.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, action.Options.HTTPConfig.Password.GetPayload())
- assert.Empty(t, action.Options.HTTPConfig.Password.GetKey())
- assert.Empty(t, action.Options.HTTPConfig.Password.GetAdditionalData())
- // update again and check that the password was preserved
- dbAction, err := dataprovider.EventActionExists(actionName)
- assert.NoError(t, err)
- action.Options.HTTPConfig.Password = kms.NewSecret(
- dbAction.Options.HTTPConfig.Password.GetStatus(),
- dbAction.Options.HTTPConfig.Password.GetPayload(), "", "")
- action, _, err = httpdtest.UpdateEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- dbAction, err = dataprovider.EventActionExists(actionName)
- assert.NoError(t, err)
- err = dbAction.Options.HTTPConfig.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, defaultPassword, dbAction.Options.HTTPConfig.Password.GetPayload())
- r := dataprovider.EventRule{
- Name: "test rule name",
- Status: 1,
- Description: "",
- Trigger: dataprovider.EventTriggerFsEvent,
- Conditions: dataprovider.EventConditions{
- FsEvents: []string{"upload"},
- Options: dataprovider.ConditionOptions{
- MinFileSize: 1024 * 1024,
- },
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: actionName,
- },
- Order: 1,
- Options: dataprovider.EventActionOptions{
- IsFailureAction: false,
- StopOnFailure: true,
- ExecuteSync: true,
- },
- },
- },
- }
- rule, _, err := httpdtest.AddEventRule(r, http.StatusCreated)
- assert.NoError(t, err)
- // adding the same rule should fail
- _, _, err = httpdtest.AddEventRule(r, http.StatusConflict)
- assert.NoError(t, err)
- rule.Description = "new rule desc"
- rule.Trigger = 1000
- _, _, err = httpdtest.UpdateEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- rule.Trigger = dataprovider.EventTriggerFsEvent
- rule, _, err = httpdtest.UpdateEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- ruleGet, _, err := httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, ruleGet.Actions, 1) {
- if assert.NotNil(t, ruleGet.Actions[0].BaseEventAction.Options.HTTPConfig.Password) {
- assert.Equal(t, sdkkms.SecretStatusSecretBox, ruleGet.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, ruleGet.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetPayload())
- assert.Empty(t, ruleGet.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetKey())
- assert.Empty(t, ruleGet.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetAdditionalData())
- }
- }
- rules, _, err := httpdtest.GetEventRules(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, len(rules), 0)
- found = false
- for _, ru := range rules {
- if ru.Name == rule.Name {
- assert.Equal(t, ruleGet, ru)
- found = true
- }
- }
- assert.True(t, found)
- _, err = httpdtest.RemoveEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateEventRule(rule, http.StatusNotFound)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventRule(rule, http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateEventAction(action, http.StatusNotFound)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetEventActionByName(actionName, http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestActionRuleRelations(t *testing.T) {
- a1 := dataprovider.BaseEventAction{
- Name: "action1",
- Description: "test description",
- Type: dataprovider.ActionTypeBackup,
- Options: dataprovider.BaseEventActionOptions{},
- }
- a2 := dataprovider.BaseEventAction{
- Name: "action2",
- Type: dataprovider.ActionTypeTransferQuotaReset,
- Options: dataprovider.BaseEventActionOptions{},
- }
- a3 := dataprovider.BaseEventAction{
- Name: "action3",
- Type: dataprovider.ActionTypeEmail,
- Options: dataprovider.BaseEventActionOptions{
- EmailConfig: dataprovider.EventActionEmailConfig{
- Recipients: []string{"test@example.net"},
- ContentType: 1,
- Subject: "test subject",
- Body: "test body",
- },
- },
- }
- action1, _, err := httpdtest.AddEventAction(a1, http.StatusCreated)
- assert.NoError(t, err)
- action2, _, err := httpdtest.AddEventAction(a2, http.StatusCreated)
- assert.NoError(t, err)
- action3, _, err := httpdtest.AddEventAction(a3, http.StatusCreated)
- assert.NoError(t, err)
- r1 := dataprovider.EventRule{
- Name: "rule1",
- Description: "",
- Trigger: dataprovider.EventTriggerProviderEvent,
- Conditions: dataprovider.EventConditions{
- ProviderEvents: []string{"add"},
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action3.Name,
- },
- Order: 2,
- Options: dataprovider.EventActionOptions{
- IsFailureAction: true,
- },
- },
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action1.Name,
- },
- Order: 1,
- },
- },
- }
- rule1, _, err := httpdtest.AddEventRule(r1, http.StatusCreated)
- assert.NoError(t, err)
- if assert.Len(t, rule1.Actions, 2) {
- assert.Equal(t, action1.Name, rule1.Actions[0].Name)
- assert.Equal(t, 1, rule1.Actions[0].Order)
- assert.Equal(t, action3.Name, rule1.Actions[1].Name)
- assert.Equal(t, 2, rule1.Actions[1].Order)
- assert.True(t, rule1.Actions[1].Options.IsFailureAction)
- }
- r2 := dataprovider.EventRule{
- Name: "rule2",
- Description: "",
- Trigger: dataprovider.EventTriggerSchedule,
- Conditions: dataprovider.EventConditions{
- Schedules: []dataprovider.Schedule{
- {
- Hours: "1",
- DayOfWeek: "*",
- DayOfMonth: "*",
- Month: "*",
- },
- },
- Options: dataprovider.ConditionOptions{
- RoleNames: []dataprovider.ConditionPattern{
- {
- Pattern: "g*",
- InverseMatch: true,
- },
- },
- },
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action3.Name,
- },
- Order: 2,
- Options: dataprovider.EventActionOptions{
- IsFailureAction: true,
- },
- },
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action2.Name,
- },
- Order: 1,
- },
- },
- }
- rule2, _, err := httpdtest.AddEventRule(r2, http.StatusCreated)
- assert.NoError(t, err)
- if assert.Len(t, rule1.Actions, 2) {
- assert.Equal(t, action2.Name, rule2.Actions[0].Name)
- assert.Equal(t, 1, rule2.Actions[0].Order)
- assert.Equal(t, action3.Name, rule2.Actions[1].Name)
- assert.Equal(t, 2, rule2.Actions[1].Order)
- assert.True(t, rule2.Actions[1].Options.IsFailureAction)
- }
- // check the references
- action1, _, err = httpdtest.GetEventActionByName(action1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action1.Rules, 1)
- assert.True(t, util.Contains(action1.Rules, rule1.Name))
- action2, _, err = httpdtest.GetEventActionByName(action2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action2.Rules, 1)
- assert.True(t, util.Contains(action2.Rules, rule2.Name))
- action3, _, err = httpdtest.GetEventActionByName(action3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action3.Rules, 2)
- assert.True(t, util.Contains(action3.Rules, rule1.Name))
- assert.True(t, util.Contains(action3.Rules, rule2.Name))
- // referenced actions cannot be removed
- _, err = httpdtest.RemoveEventAction(action1, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action2, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action3, http.StatusBadRequest)
- assert.NoError(t, err)
- // remove action3 from rule2
- r2.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action2.Name,
- },
- Order: 10,
- },
- }
- rule2.Status = 1
- rule2, _, err = httpdtest.UpdateEventRule(r2, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, rule2.Actions, 1) {
- assert.Equal(t, action2.Name, rule2.Actions[0].Name)
- assert.Equal(t, 10, rule2.Actions[0].Order)
- }
- // check the updated relation
- action3, _, err = httpdtest.GetEventActionByName(action3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action3.Rules, 1)
- assert.True(t, util.Contains(action3.Rules, rule1.Name))
- _, err = httpdtest.RemoveEventRule(rule1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventRule(rule2, http.StatusOK)
- assert.NoError(t, err)
- // no relations anymore
- action1, _, err = httpdtest.GetEventActionByName(action1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action1.Rules, 0)
- action2, _, err = httpdtest.GetEventActionByName(action2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action2.Rules, 0)
- action3, _, err = httpdtest.GetEventActionByName(action3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, action3.Rules, 0)
- _, err = httpdtest.RemoveEventAction(action1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action3, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestOnDemandEventRules(t *testing.T) {
- ruleName := "test on demand rule"
- a := dataprovider.BaseEventAction{
- Name: "a",
- Type: dataprovider.ActionTypeBackup,
- Options: dataprovider.BaseEventActionOptions{},
- }
- action, _, err := httpdtest.AddEventAction(a, http.StatusCreated)
- assert.NoError(t, err)
- r := dataprovider.EventRule{
- Name: ruleName,
- Status: 1,
- Trigger: dataprovider.EventTriggerOnDemand,
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: a.Name,
- },
- },
- },
- }
- rule, _, err := httpdtest.AddEventRule(r, http.StatusCreated)
- assert.NoError(t, err)
- _, err = httpdtest.RunOnDemandRule(ruleName, http.StatusAccepted)
- assert.NoError(t, err)
- rule.Status = 0
- _, _, err = httpdtest.UpdateEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- resp, err := httpdtest.RunOnDemandRule(ruleName, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "is inactive")
- _, err = httpdtest.RemoveEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RunOnDemandRule(ruleName, http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestIDPLoginEventRule(t *testing.T) {
- ruleName := "test IDP login rule"
- a := dataprovider.BaseEventAction{
- Name: "a",
- Type: dataprovider.ActionTypeIDPAccountCheck,
- Options: dataprovider.BaseEventActionOptions{
- IDPConfig: dataprovider.EventActionIDPAccountCheck{
- Mode: 1,
- TemplateUser: `{"username": "user"}`,
- TemplateAdmin: `{"username": "admin"}`,
- },
- },
- }
- action, resp, err := httpdtest.AddEventAction(a, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- r := dataprovider.EventRule{
- Name: ruleName,
- Status: 1,
- Trigger: dataprovider.EventTriggerIDPLogin,
- Conditions: dataprovider.EventConditions{
- IDPLoginEvent: 1,
- Options: dataprovider.ConditionOptions{
- Names: []dataprovider.ConditionPattern{
- {
- Pattern: "username",
- },
- },
- },
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: a.Name,
- },
- Options: dataprovider.EventActionOptions{
- ExecuteSync: true,
- },
- },
- },
- }
- rule, _, err := httpdtest.AddEventRule(r, http.StatusCreated)
- assert.NoError(t, err)
- rule.Status = 0
- _, _, err = httpdtest.UpdateEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestEventActionValidation(t *testing.T) {
- action := dataprovider.BaseEventAction{
- Name: "",
- }
- _, resp, err := httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "name is mandatory")
- action = dataprovider.BaseEventAction{
- Name: "n",
- Type: -1,
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid action type")
- action.Type = dataprovider.ActionTypeHTTP
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "HTTP endpoint is required")
- action.Options.HTTPConfig.Endpoint = "abc"
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid HTTP endpoint schema")
- action.Options.HTTPConfig.Endpoint = "http://localhost"
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid HTTP timeout")
- action.Options.HTTPConfig.Timeout = 20
- action.Options.HTTPConfig.Headers = []dataprovider.KeyValue{
- {
- Key: "",
- Value: "",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid HTTP headers")
- action.Options.HTTPConfig.Headers = []dataprovider.KeyValue{
- {
- Key: "Content-Type",
- Value: "application/json",
- },
- }
- action.Options.HTTPConfig.Password = kms.NewSecret(sdkkms.SecretStatusRedacted, "payload", "", "")
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "cannot save HTTP configuration with a redacted secret")
- action.Options.HTTPConfig.Password = nil
- action.Options.HTTPConfig.Method = http.MethodTrace
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "unsupported HTTP method")
- action.Options.HTTPConfig.Method = http.MethodGet
- action.Options.HTTPConfig.QueryParameters = []dataprovider.KeyValue{
- {
- Key: "a",
- Value: "",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid HTTP query parameters")
- action.Options.HTTPConfig.QueryParameters = nil
- action.Options.HTTPConfig.Parts = []dataprovider.HTTPPart{
- {
- Name: "",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "HTTP part name is required")
- action.Options.HTTPConfig.Parts = []dataprovider.HTTPPart{
- {
- Name: "p1",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "HTTP part body is required if no file path is provided")
- action.Options.HTTPConfig.Parts = []dataprovider.HTTPPart{
- {
- Name: "p1",
- Filepath: "p",
- },
- }
- action.Options.HTTPConfig.Body = "b"
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "multipart requests require no body")
- action.Options.HTTPConfig.Body = ""
- action.Options.HTTPConfig.Headers = []dataprovider.KeyValue{
- {
- Key: "Content-Type",
- Value: "application/json",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "content type is automatically set for multipart requests")
- action.Type = dataprovider.ActionTypeCommand
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "command is required")
- action.Options.CmdConfig.Cmd = "relative"
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid command, it must be an absolute path")
- action.Options.CmdConfig.Cmd = filepath.Join(os.TempDir(), "cmd")
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid command action timeout")
- action.Options.CmdConfig.Timeout = 30
- action.Options.CmdConfig.EnvVars = []dataprovider.KeyValue{
- {
- Key: "k",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid command env vars")
- action.Options.CmdConfig.EnvVars = nil
- action.Options.CmdConfig.Args = []string{"arg1", ""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid command args")
- action.Type = dataprovider.ActionTypeEmail
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least one email recipient is required")
- action.Options.EmailConfig.Recipients = []string{""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid email recipients")
- action.Options.EmailConfig.Recipients = []string{"a@a.com"}
- action.Options.EmailConfig.Bcc = []string{""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid email bcc")
- action.Options.EmailConfig.Bcc = nil
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "email subject is required")
- action.Options.EmailConfig.Subject = "subject"
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "email body is required")
- action.Type = dataprovider.ActionTypeDataRetentionCheck
- action.Options.RetentionConfig = dataprovider.EventActionDataRetentionConfig{
- Folders: nil,
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "nothing to delete")
- action.Options.RetentionConfig = dataprovider.EventActionDataRetentionConfig{
- Folders: []dataprovider.FolderRetention{
- {
- Path: "/",
- Retention: 0,
- },
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "nothing to delete")
- action.Options.RetentionConfig = dataprovider.EventActionDataRetentionConfig{
- Folders: []dataprovider.FolderRetention{
- {
- Path: "../path",
- Retention: 1,
- },
- {
- Path: "/path",
- Retention: 10,
- },
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "duplicated folder path")
- action.Options.RetentionConfig = dataprovider.EventActionDataRetentionConfig{
- Folders: []dataprovider.FolderRetention{
- {
- Path: "p",
- Retention: -1,
- },
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid folder retention")
- action.Type = dataprovider.ActionTypeFilesystem
- action.Options.FsConfig = dataprovider.EventActionFilesystemConfig{
- Type: dataprovider.FilesystemActionRename,
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "no path to rename specified")
- action.Options.FsConfig.Renames = []dataprovider.KeyValue{
- {
- Key: "",
- Value: "/adir",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid paths to rename")
- action.Options.FsConfig.Renames = []dataprovider.KeyValue{
- {
- Key: "adir",
- Value: "/adir",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "rename source and target cannot be equal")
- action.Options.FsConfig.Renames = []dataprovider.KeyValue{
- {
- Key: "/",
- Value: "/dir",
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "renaming the root directory is not allowed")
- action.Options.FsConfig.Type = dataprovider.FilesystemActionMkdirs
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "no directory to create specified")
- action.Options.FsConfig.MkDirs = []string{"dir1", ""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid directory to create")
- action.Options.FsConfig.Type = dataprovider.FilesystemActionDelete
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "no path to delete specified")
- action.Options.FsConfig.Deletes = []string{"item1", ""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid path to delete")
- action.Options.FsConfig.Type = dataprovider.FilesystemActionExist
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "no path to check for existence specified")
- action.Options.FsConfig.Exist = []string{"item1", ""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid path to check for existence")
- action.Options.FsConfig.Type = dataprovider.FilesystemActionCompress
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "archive name is mandatory")
- action.Options.FsConfig.Compress.Name = "archive.zip"
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "no path to compress specified")
- action.Options.FsConfig.Compress.Paths = []string{"item1", ""}
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid path to compress")
- action.Type = dataprovider.ActionTypePasswordExpirationCheck
- action.Options.PwdExpirationConfig.Threshold = 0
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "threshold must be greater than 0")
- action.Type = dataprovider.ActionTypeIDPAccountCheck
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least a template must be set")
- action.Options.IDPConfig.TemplateAdmin = "{}"
- action.Options.IDPConfig.Mode = 100
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid account check mode")
- action.Type = dataprovider.ActionTypeUserInactivityCheck
- action.Options = dataprovider.BaseEventActionOptions{
- UserInactivityConfig: dataprovider.EventActionUserInactivity{
- DisableThreshold: 0,
- DeleteThreshold: 0,
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least a threshold must be defined")
- action.Options = dataprovider.BaseEventActionOptions{
- UserInactivityConfig: dataprovider.EventActionUserInactivity{
- DisableThreshold: 10,
- DeleteThreshold: 10,
- },
- }
- _, resp, err = httpdtest.AddEventAction(action, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "must be greater than deactivation threshold")
- }
- func TestEventRuleValidation(t *testing.T) {
- rule := dataprovider.EventRule{
- Name: "",
- }
- _, resp, err := httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "name is mandatory")
- rule.Name = "r"
- rule.Status = 100
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid event rule status")
- rule.Status = 1
- rule.Trigger = 1000
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid event rule trigger")
- rule.Trigger = dataprovider.EventTriggerFsEvent
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least one filesystem event is required")
- rule.Conditions.FsEvents = []string{""}
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "unsupported fs event")
- rule.Conditions.FsEvents = []string{"upload"}
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least one action is required")
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action1",
- },
- Order: 1,
- },
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "",
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "name not specified")
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action",
- },
- Order: 1,
- },
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action",
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "duplicated action")
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action11",
- },
- Order: 1,
- },
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action12",
- },
- Order: 1,
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "duplicated order")
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action111",
- },
- Order: 1,
- Options: dataprovider.EventActionOptions{
- IsFailureAction: true,
- },
- },
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action112",
- },
- Order: 2,
- Options: dataprovider.EventActionOptions{
- IsFailureAction: true,
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least a non-failure action is required")
- rule.Conditions.FsEvents = []string{"upload", "download"}
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action111",
- },
- Order: 1,
- Options: dataprovider.EventActionOptions{
- ExecuteSync: true,
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "sync execution is only supported for upload and pre-* events")
- rule.Conditions.FsEvents = []string{"pre-upload", "download"}
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action",
- },
- Order: 1,
- Options: dataprovider.EventActionOptions{
- ExecuteSync: false,
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "requires at least a sync action")
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action",
- },
- Order: 1,
- Options: dataprovider.EventActionOptions{
- ExecuteSync: true,
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "sync execution is only supported for upload and pre-* events")
- rule.Trigger = dataprovider.EventTriggerProviderEvent
- rule.Actions = []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "action1234",
- },
- Order: 1,
- Options: dataprovider.EventActionOptions{
- IsFailureAction: false,
- },
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least one provider event is required")
- rule.Conditions.ProviderEvents = []string{""}
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "unsupported provider event")
- rule.Conditions.ProviderEvents = []string{"add"}
- rule.Conditions.Options.RoleNames = []dataprovider.ConditionPattern{
- {
- Pattern: "",
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "empty condition pattern not allowed")
- rule.Conditions.Options.RoleNames = nil
- rule.Trigger = dataprovider.EventTriggerSchedule
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "at least one schedule is required")
- rule.Conditions.Schedules = []dataprovider.Schedule{
- {},
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid schedule")
- rule.Conditions.Schedules = []dataprovider.Schedule{
- {
- Hours: "3",
- DayOfWeek: "*",
- DayOfMonth: "*",
- Month: "*",
- },
- }
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusInternalServerError)
- assert.NoError(t, err, string(resp))
- rule.Trigger = dataprovider.EventTriggerIDPLogin
- rule.Conditions.IDPLoginEvent = 100
- _, resp, err = httpdtest.AddEventRule(rule, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid Identity Provider login event")
- }
- func TestUserBandwidthLimits(t *testing.T) {
- u := getTestUser()
- u.UploadBandwidth = 128
- u.DownloadBandwidth = 96
- u.Filters.BandwidthLimits = []sdk.BandwidthLimit{
- {
- Sources: []string{"1"},
- },
- }
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "Validation error: could not parse bandwidth limit source")
- u.Filters.BandwidthLimits = []sdk.BandwidthLimit{
- {
- Sources: nil,
- },
- }
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "Validation error: no bandwidth limit source specified")
- u.Filters.BandwidthLimits = []sdk.BandwidthLimit{
- {
- Sources: []string{"127.0.0.0/8", "::1/128"},
- UploadBandwidth: 256,
- },
- {
- Sources: []string{"10.0.0.0/8"},
- UploadBandwidth: 512,
- DownloadBandwidth: 256,
- },
- }
- user, resp, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, user.Filters.BandwidthLimits, 2)
- assert.Equal(t, u.Filters.BandwidthLimits, user.Filters.BandwidthLimits)
- connID := xid.New().String()
- localAddr := "127.0.0.1"
- up, down := user.GetBandwidthForIP("127.0.1.1", connID)
- assert.Equal(t, int64(256), up)
- assert.Equal(t, int64(0), down)
- conn := common.NewBaseConnection(connID, common.ProtocolHTTP, localAddr, "127.0.1.1", user)
- assert.Equal(t, int64(256), conn.User.UploadBandwidth)
- assert.Equal(t, int64(0), conn.User.DownloadBandwidth)
- up, down = user.GetBandwidthForIP("10.1.2.3", connID)
- assert.Equal(t, int64(512), up)
- assert.Equal(t, int64(256), down)
- conn = common.NewBaseConnection(connID, common.ProtocolHTTP, localAddr, "10.2.1.4:1234", user)
- assert.Equal(t, int64(512), conn.User.UploadBandwidth)
- assert.Equal(t, int64(256), conn.User.DownloadBandwidth)
- up, down = user.GetBandwidthForIP("192.168.1.2", connID)
- assert.Equal(t, int64(128), up)
- assert.Equal(t, int64(96), down)
- conn = common.NewBaseConnection(connID, common.ProtocolHTTP, localAddr, "172.16.0.1", user)
- assert.Equal(t, int64(128), conn.User.UploadBandwidth)
- assert.Equal(t, int64(96), conn.User.DownloadBandwidth)
- up, down = user.GetBandwidthForIP("invalid", connID)
- assert.Equal(t, int64(128), up)
- assert.Equal(t, int64(96), down)
- conn = common.NewBaseConnection(connID, common.ProtocolHTTP, localAddr, "172.16.0", user)
- assert.Equal(t, int64(128), conn.User.UploadBandwidth)
- assert.Equal(t, int64(96), conn.User.DownloadBandwidth)
- user.Filters.BandwidthLimits = []sdk.BandwidthLimit{
- {
- Sources: []string{"10.0.0.0/24"},
- UploadBandwidth: 256,
- DownloadBandwidth: 512,
- },
- }
- user, resp, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- if assert.Len(t, user.Filters.BandwidthLimits, 1) {
- bwLimit := user.Filters.BandwidthLimits[0]
- assert.Equal(t, []string{"10.0.0.0/24"}, bwLimit.Sources)
- assert.Equal(t, int64(256), bwLimit.UploadBandwidth)
- assert.Equal(t, int64(512), bwLimit.DownloadBandwidth)
- }
- up, down = user.GetBandwidthForIP("10.1.2.3", connID)
- assert.Equal(t, int64(128), up)
- assert.Equal(t, int64(96), down)
- conn = common.NewBaseConnection(connID, common.ProtocolHTTP, localAddr, "172.16.0.2", user)
- assert.Equal(t, int64(128), conn.User.UploadBandwidth)
- assert.Equal(t, int64(96), conn.User.DownloadBandwidth)
- up, down = user.GetBandwidthForIP("10.0.0.26", connID)
- assert.Equal(t, int64(256), up)
- assert.Equal(t, int64(512), down)
- conn = common.NewBaseConnection(connID, common.ProtocolHTTP, localAddr, "10.0.0.28", user)
- assert.Equal(t, int64(256), conn.User.UploadBandwidth)
- assert.Equal(t, int64(512), conn.User.DownloadBandwidth)
- // this works if we remove the omitempty tag from BandwidthLimits
- /*user.Filters.BandwidthLimits = nil
- user, resp, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- assert.Len(t, user.Filters.BandwidthLimits, 0)*/
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAccessTimeValidation(t *testing.T) {
- u := getTestUser()
- u.Filters.AccessTime = []sdk.TimePeriod{
- {
- DayOfWeek: 8,
- From: "10:00",
- To: "18:00",
- },
- }
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "invalid day of week")
- u.Filters.AccessTime = []sdk.TimePeriod{
- {
- DayOfWeek: 6,
- From: "10:00",
- To: "18",
- },
- }
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "invalid time of day")
- u.Filters.AccessTime = []sdk.TimePeriod{
- {
- DayOfWeek: 6,
- From: "11:00",
- To: "10:58",
- },
- }
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "The end time cannot be earlier than the start time")
- }
- func TestUserTimestamps(t *testing.T) {
- user, resp, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err, string(resp))
- createdAt := user.CreatedAt
- updatedAt := user.UpdatedAt
- assert.Equal(t, int64(0), user.LastLogin)
- assert.Equal(t, int64(0), user.FirstDownload)
- assert.Equal(t, int64(0), user.FirstUpload)
- assert.Greater(t, createdAt, int64(0))
- assert.Greater(t, updatedAt, int64(0))
- mappedPath := filepath.Join(os.TempDir(), "mapped_dir")
- folderName := filepath.Base(mappedPath)
- user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedPath,
- },
- VirtualPath: "/vdir",
- })
- f := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedPath,
- }
- _, _, err = httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err)
- time.Sleep(10 * time.Millisecond)
- user, resp, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- assert.Equal(t, int64(0), user.LastLogin)
- assert.Equal(t, int64(0), user.FirstDownload)
- assert.Equal(t, int64(0), user.FirstUpload)
- assert.Equal(t, createdAt, user.CreatedAt)
- assert.Greater(t, user.UpdatedAt, updatedAt)
- updatedAt = user.UpdatedAt
- // after a folder update or delete the user updated_at field should change
- folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- time.Sleep(10 * time.Millisecond)
- _, _, err = httpdtest.UpdateFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), user.LastLogin)
- assert.Equal(t, int64(0), user.FirstDownload)
- assert.Equal(t, int64(0), user.FirstUpload)
- assert.Equal(t, createdAt, user.CreatedAt)
- assert.Greater(t, user.UpdatedAt, updatedAt)
- updatedAt = user.UpdatedAt
- time.Sleep(10 * time.Millisecond)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), user.LastLogin)
- assert.Equal(t, int64(0), user.FirstDownload)
- assert.Equal(t, int64(0), user.FirstUpload)
- assert.Equal(t, createdAt, user.CreatedAt)
- assert.Greater(t, user.UpdatedAt, updatedAt)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAdminTimestamps(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- createdAt := admin.CreatedAt
- updatedAt := admin.UpdatedAt
- assert.Equal(t, int64(0), admin.LastLogin)
- assert.Greater(t, createdAt, int64(0))
- assert.Greater(t, updatedAt, int64(0))
- time.Sleep(10 * time.Millisecond)
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), admin.LastLogin)
- assert.Equal(t, createdAt, admin.CreatedAt)
- assert.Greater(t, admin.UpdatedAt, updatedAt)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestHTTPUserAuthEmptyPassword(t *testing.T) {
- u := getTestUser()
- u.Password = ""
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, "")
- c := httpclient.GetHTTPClient()
- resp, err := c.Do(req)
- c.CloseIdleConnections()
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, "")
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "unexpected status code 401")
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestHTTPAnonymousUser(t *testing.T) {
- u := getTestUser()
- u.Filters.IsAnonymous = true
- _, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.Error(t, err)
- user, _, err := httpdtest.GetUserByUsername(u.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.IsAnonymous)
- assert.Equal(t, []string{dataprovider.PermListItems, dataprovider.PermDownload}, user.Permissions["/"])
- assert.Equal(t, []string{common.ProtocolSSH, common.ProtocolHTTP}, user.Filters.DeniedProtocols)
- assert.Equal(t, []string{dataprovider.SSHLoginMethodPublicKey, dataprovider.SSHLoginMethodPassword,
- dataprovider.SSHLoginMethodKeyboardInteractive, dataprovider.SSHLoginMethodKeyAndPassword,
- dataprovider.SSHLoginMethodKeyAndKeyboardInt, dataprovider.LoginMethodTLSCertificate,
- dataprovider.LoginMethodTLSCertificateAndPwd}, user.Filters.DeniedLoginMethods)
- req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- c := httpclient.GetHTTPClient()
- resp, err := c.Do(req)
- c.CloseIdleConnections()
- assert.NoError(t, err)
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "unexpected status code 403")
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestHTTPUserAuthentication(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- c := httpclient.GetHTTPClient()
- resp, err := c.Do(req)
- c.CloseIdleConnections()
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- userToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, userToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- // login with wrong credentials
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, "")
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, "wrong pwd")
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- respBody, err := io.ReadAll(resp.Body)
- assert.NoError(t, err)
- assert.Contains(t, string(respBody), "invalid credentials")
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth("wrong username", defaultPassword)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- respBody, err = io.ReadAll(resp.Body)
- assert.NoError(t, err)
- assert.Contains(t, string(respBody), "invalid credentials")
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultTokenAuthUser, defaultTokenAuthPass)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder = make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- adminToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, adminToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, versionPath), nil)
- assert.NoError(t, err)
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", adminToken))
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- // using the user token should not work
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, versionPath), nil)
- assert.NoError(t, err)
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", userToken))
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userLogoutPath), nil)
- assert.NoError(t, err)
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", adminToken))
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userLogoutPath), nil)
- assert.NoError(t, err)
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", userToken))
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestPermMFADisabled(t *testing.T) {
- u := getTestUser()
- u.Filters.WebClient = []string{sdk.WebClientMFADisabled}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user.Filters.TwoFactorAuthProtocols = []string{common.ProtocolSSH}
- _, resp, err := httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "you cannot require two-factor authentication and at the same time disallow it")
- user.Filters.TwoFactorAuthProtocols = nil
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolSSH},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr) // MFA is disabled for this user
- user.Filters.WebClient = []string{sdk.WebClientWriteDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now we cannot disable MFA for this user
- user.Filters.WebClient = []string{sdk.WebClientMFADisabled}
- _, resp, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "two-factor authentication cannot be disabled for a user with an active configuration")
- saveReq := make(map[string]bool)
- saveReq["enabled"] = false
- asJSON, err = json.Marshal(saveReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestUpdateUserPassword(t *testing.T) {
- g := getTestGroup()
- g.UserSettings.Filters.PasswordStrength = 20
- g.UserSettings.MaxSessions = 10
- group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
- assert.NoError(t, err)
- u := getTestUser()
- u.Filters.RequirePasswordChange = true
- u.Groups = []sdk.GroupMapping{
- {
- Name: group.Name,
- Type: sdk.GroupTypePrimary,
- },
- }
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- lastPwdChange := user.LastPasswordChange
- time.Sleep(100 * time.Millisecond)
- newPwd := "uaCooGh3pheiShooghah"
- err = dataprovider.UpdateUserPassword(user.Username, newPwd, "", "", "")
- assert.NoError(t, err)
- _, err = dataprovider.CheckUserAndPass(user.Username, newPwd, "", common.ProtocolHTTP)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, user.Filters.RequirePasswordChange)
- assert.NotEqual(t, lastPwdChange, user.LastPasswordChange)
- // check that we don't save group overrides
- assert.Equal(t, 0, user.MaxSessions)
- assert.Equal(t, 0, user.Filters.PasswordStrength)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestLoginRedirectNext(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- uri := webClientFilesPath + "?path=%2F" //nolint:goconst
- req, err := http.NewRequest(http.MethodGet, uri, nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- redirectURI := rr.Header().Get("Location")
- assert.Equal(t, webClientLoginPath+"?next="+url.QueryEscape(uri), redirectURI) //nolint:goconst
- // render the login page
- req, err = http.NewRequest(http.MethodGet, redirectURI, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), fmt.Sprintf("action=%q", redirectURI))
- // now login the user and check the redirect
- csrfToken, err := getCSRFTokenMock(webClientLoginPath, defaultRemoteAddr)
- assert.NoError(t, err)
- form := getLoginForm(defaultUsername, defaultPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, redirectURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = redirectURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, uri, rr.Header().Get("Location"))
- // unsafe URI
- unsafeURI := webClientLoginPath + "?next=" + url.QueryEscape("http://example.net")
- req, err = http.NewRequest(http.MethodPost, unsafeURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = unsafeURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientFilesPath, rr.Header().Get("Location"))
- unsupportedURI := webClientLoginPath + "?next=" + url.QueryEscape(webClientProfilePath)
- req, err = http.NewRequest(http.MethodPost, unsupportedURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = unsupportedURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientFilesPath, rr.Header().Get("Location"))
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestMustChangePasswordRequirement(t *testing.T) {
- u := getTestUser()
- u.Filters.RequirePasswordChange = true
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.True(t, user.Filters.RequirePasswordChange)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, userFilesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Password change required. Please set a new password to continue to use your account")
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdRequired)
- // change pwd
- pwd := make(map[string]string)
- pwd["current_password"] = defaultPassword
- pwd["new_password"] = altAdminPassword
- asJSON, err := json.Marshal(pwd)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userPwdPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check that the change pwd bool is changed
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, user.Filters.RequirePasswordChange)
- // get a new token
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, altAdminPassword)
- assert.NoError(t, err)
- webToken, err = getJWTWebClientTokenFromTestServer(defaultUsername, altAdminPassword)
- assert.NoError(t, err)
- // the new token should work
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check the same as above but changing password from the WebClient UI
- user.Filters.RequirePasswordChange = true
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- webToken, err = getJWTWebClientTokenFromTestServer(defaultUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("current_password", altAdminPassword)
- form.Set("new_password1", defaultPassword)
- form.Set("new_password2", defaultPassword)
- req, err = http.NewRequest(http.MethodPost, webChangeClientPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestTwoFactorRequirements(t *testing.T) {
- u := getTestUser()
- u.Filters.TwoFactorAuthProtocols = []string{common.ProtocolHTTP, common.ProtocolFTP}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Two-factor authentication requirements not met, please configure two-factor authentication for the following protocols")
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError2FARequired)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolHTTP},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following protocols are required")
- userTOTPConfig.Protocols = []string{common.ProtocolHTTP, common.ProtocolFTP}
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now get new tokens and check that the two factor requirements are now met
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", passcode)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- userToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, userToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userDirsPath), nil)
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestTwoFactorRequirementsGroupLevel(t *testing.T) {
- g := getTestGroup()
- g.UserSettings.Filters.TwoFactorAuthProtocols = []string{common.ProtocolHTTP, common.ProtocolFTP}
- group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
- assert.NoError(t, err)
- u := getTestUser()
- u.Groups = []sdk.GroupMapping{
- {
- Name: group.Name,
- Type: sdk.GroupTypePrimary,
- },
- }
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError2FARequired)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Two-factor authentication requirements not met, please configure two-factor authentication for the following protocols")
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolHTTP},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following protocols are required")
- userTOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolFTP, common.ProtocolHTTP},
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now get new tokens and check that the two factor requirements are now met
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", passcode)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- userToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, userToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userDirsPath), nil)
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAdminMustChangePasswordRequirement(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin.Filters.RequirePasswordChange = true
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, _, err := httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- _, _, err = httpdtest.GetUsers(0, 0, http.StatusForbidden)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetStatus(http.StatusForbidden)
- assert.NoError(t, err)
- _, err = httpdtest.ChangeAdminPassword(altAdminPassword, defaultTokenAuthPass, http.StatusOK)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.RequirePasswordChange)
- // get a new token
- token, _, err = httpdtest.GetToken(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- _, _, err = httpdtest.GetUsers(0, 0, http.StatusOK)
- assert.NoError(t, err)
- desc := xid.New().String()
- admin.Filters.RequirePasswordChange = true
- admin.Filters.RequireTwoFactor = true
- admin.Description = desc
- _, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- if assert.Error(t, err) {
- assert.ErrorContains(t, err, "require password change mismatch")
- }
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.RequirePasswordChange)
- assert.False(t, admin.Filters.RequireTwoFactor)
- assert.Equal(t, desc, admin.Description)
- httpdtest.SetJWTToken("")
- admin.Filters.RequirePasswordChange = true
- _, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- // test the same for the WebAdmin
- webToken, err := getJWTWebTokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webUsersPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webUsersPath
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("current_password", defaultTokenAuthPass)
- form.Set("new_password1", altAdminPassword)
- form.Set("new_password2", altAdminPassword)
- req, err = http.NewRequest(http.MethodPost, webChangeAdminPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.RequirePasswordChange)
- webToken, err = getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webUsersPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webUsersPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAdminTwoFactorRequirements(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin.Filters.RequireTwoFactor = true
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Two-factor authentication requirements not met")
- webToken, err := getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webFoldersPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webFoldersPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError2FARequiredGeneric)
- // add TOTP config
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], altAdminUsername)
- assert.NoError(t, err)
- adminTOTPConfig := dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- }
- asJSON, err := json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", passcode)
- req.SetBasicAuth(altAdminUsername, altAdminPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- token = responseHolder["access_token"].(string)
- assert.NotEmpty(t, token)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, serverStatusPath), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- // try to disable 2FA
- disableReq := map[string]any{
- "enabled": false,
- }
- asJSON, err = json.Marshal(disableReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, fmt.Sprintf("%v%v", httpBaseURL, adminTOTPSavePath), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
- bodyResp, err := io.ReadAll(resp.Body)
- assert.NoError(t, err)
- assert.Contains(t, string(bodyResp), "two-factor authentication must be enabled")
- err = resp.Body.Close()
- assert.NoError(t, err)
- // try to disable 2FA using the dedicated API
- req, err = http.NewRequest(http.MethodPut, fmt.Sprintf("%v%v", httpBaseURL, path.Join(adminPath, altAdminUsername, "2fa", "disable")), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
- bodyResp, err = io.ReadAll(resp.Body)
- assert.NoError(t, err)
- assert.Contains(t, string(bodyResp), "two-factor authentication must be enabled")
- err = resp.Body.Close()
- assert.NoError(t, err)
- // disabling 2FA using another admin should work
- token, err = getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(adminPath, altAdminUsername, "2fa", "disable"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.TOTPConfig.Enabled)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestLoginUserAPITOTP(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolHTTP},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now require HTTP and SSH for TOTP
- user.Filters.TwoFactorAuthProtocols = []string{common.ProtocolHTTP, common.ProtocolSSH}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- // two factor auth cannot be disabled
- config := make(map[string]any)
- config["enabled"] = false
- asJSON, err = json.Marshal(config)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "two-factor authentication must be enabled")
- // all the required protocols must be enabled
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following protocols are required")
- // setting all the required protocols should work
- userTOTPConfig.Protocols = []string{common.ProtocolHTTP, common.ProtocolSSH}
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", passcode)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- userToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, userToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", passcode)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestLoginAdminAPITOTP(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], admin.Username)
- assert.NoError(t, err)
- altToken, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- adminTOTPConfig := dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- }
- asJSON, err := json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 12)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(altAdminUsername, altAdminPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", "passcode")
- req.SetBasicAuth(altAdminUsername, altAdminPassword)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", passcode)
- req.SetBasicAuth(altAdminUsername, altAdminPassword)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- adminToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, adminToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, versionPath), nil)
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- // get/set recovery codes
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPost, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // disable two-factor auth
- saveReq := make(map[string]bool)
- saveReq["enabled"] = false
- asJSON, err = json.Marshal(saveReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 0)
- // get/set recovery codes will not work
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestHTTPStreamZipError(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- userToken := responseHolder["access_token"].(string)
- assert.NotEmpty(t, userToken)
- err = resp.Body.Close()
- assert.NoError(t, err)
- filesList := []string{"missing"}
- asJSON, err := json.Marshal(filesList)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, fmt.Sprintf("%v%v", httpBaseURL, userStreamZipPath), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", userToken))
- resp, err = httpclient.GetHTTPClient().Do(req)
- if !assert.Error(t, err) { // the connection will be closed
- err = resp.Body.Close()
- assert.NoError(t, err)
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestBasicAdminHandling(t *testing.T) {
- // we have one admin by default
- admins, _, err := httpdtest.GetAdmins(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(admins), 1)
- admin := getTestAdmin()
- // the default admin already exists
- _, _, err = httpdtest.AddAdmin(admin, http.StatusConflict)
- assert.NoError(t, err)
- admin.Username = altAdminUsername
- admin.Filters.Preferences.HideUserPageSections = 1 + 4 + 8
- admin.Filters.Preferences.DefaultUsersExpiration = 30
- admin, _, err = httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.Preferences.HideGroups())
- assert.False(t, admin.Filters.Preferences.HideFilesystem())
- assert.True(t, admin.Filters.Preferences.HideVirtualFolders())
- assert.True(t, admin.Filters.Preferences.HideProfile())
- assert.False(t, admin.Filters.Preferences.HideACLs())
- assert.False(t, admin.Filters.Preferences.HideDiskQuotaAndBandwidthLimits())
- assert.False(t, admin.Filters.Preferences.HideAdvancedSettings())
- admin.AdditionalInfo = "test info"
- admin.Filters.Preferences.HideUserPageSections = 16 + 32 + 64
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, "test info", admin.AdditionalInfo)
- assert.False(t, admin.Filters.Preferences.HideGroups())
- assert.False(t, admin.Filters.Preferences.HideFilesystem())
- assert.False(t, admin.Filters.Preferences.HideVirtualFolders())
- assert.False(t, admin.Filters.Preferences.HideProfile())
- assert.True(t, admin.Filters.Preferences.HideACLs())
- assert.True(t, admin.Filters.Preferences.HideDiskQuotaAndBandwidthLimits())
- assert.True(t, admin.Filters.Preferences.HideAdvancedSettings())
- admins, _, err = httpdtest.GetAdmins(1, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, admins, 1)
- assert.NotEqual(t, admin.Username, admins[0].Username)
- admins, _, err = httpdtest.GetAdmins(1, 1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, admins, 1)
- assert.Equal(t, admin.Username, admins[0].Username)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusNotFound)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username+"123", http.StatusNotFound)
- assert.NoError(t, err)
- admin.Username = defaultTokenAuthUser
- _, err = httpdtest.RemoveAdmin(admin, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAdminGroups(t *testing.T) {
- group1 := getTestGroup()
- group1.Name += "_1"
- group1, _, err := httpdtest.AddGroup(group1, http.StatusCreated)
- assert.NoError(t, err)
- group2 := getTestGroup()
- group2.Name += "_2"
- group2, _, err = httpdtest.AddGroup(group2, http.StatusCreated)
- assert.NoError(t, err)
- group3 := getTestGroup()
- group3.Name += "_3"
- group3, _, err = httpdtest.AddGroup(group3, http.StatusCreated)
- assert.NoError(t, err)
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Groups = []dataprovider.AdminGroupMapping{
- {
- Name: group1.Name,
- Options: dataprovider.AdminGroupMappingOptions{
- AddToUsersAs: dataprovider.GroupAddToUsersAsPrimary,
- },
- },
- {
- Name: group2.Name,
- Options: dataprovider.AdminGroupMappingOptions{
- AddToUsersAs: dataprovider.GroupAddToUsersAsSecondary,
- },
- },
- {
- Name: group3.Name,
- Options: dataprovider.AdminGroupMappingOptions{
- AddToUsersAs: dataprovider.GroupAddToUsersAsMembership,
- },
- },
- }
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- assert.Len(t, admin.Groups, 3)
- groups, _, err := httpdtest.GetGroups(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, groups, 3)
- for _, g := range groups {
- if assert.Len(t, g.Admins, 1) {
- assert.Equal(t, admin.Username, g.Admins[0])
- }
- }
- admin, _, err = httpdtest.UpdateAdmin(a, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group2, http.StatusOK)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, admin.Groups, 1)
- // try to add a missing group
- admin.Groups = []dataprovider.AdminGroupMapping{
- {
- Name: group1.Name,
- Options: dataprovider.AdminGroupMappingOptions{
- AddToUsersAs: dataprovider.GroupAddToUsersAsPrimary,
- },
- },
- {
- Name: group2.Name,
- Options: dataprovider.AdminGroupMappingOptions{
- AddToUsersAs: dataprovider.GroupAddToUsersAsSecondary,
- },
- },
- }
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Admins, 1)
- _, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.Error(t, err)
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Admins, 1)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- group3, _, err = httpdtest.GetGroupByName(group3.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group3.Admins, 0)
- _, err = httpdtest.RemoveGroup(group3, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestChangeAdminPassword(t *testing.T) {
- _, err := httpdtest.ChangeAdminPassword("wrong", defaultTokenAuthPass, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.ChangeAdminPassword(defaultTokenAuthPass, defaultTokenAuthPass, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.ChangeAdminPassword(defaultTokenAuthPass, defaultTokenAuthPass+"1", http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.ChangeAdminPassword(defaultTokenAuthPass+"1", defaultTokenAuthPass, http.StatusUnauthorized)
- assert.NoError(t, err)
- admin, err := dataprovider.AdminExists(defaultTokenAuthUser)
- assert.NoError(t, err)
- admin.Password = defaultTokenAuthPass
- err = dataprovider.UpdateAdmin(&admin, "", "", "")
- assert.NoError(t, err)
- }
- func TestPasswordValidations(t *testing.T) {
- if config.GetProviderConf().Driver == dataprovider.MemoryDataProviderName {
- t.Skip("this test is not supported with the memory provider")
- }
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- providerConf := config.GetProviderConf()
- assert.NoError(t, err)
- providerConf.PasswordValidation.Admins.MinEntropy = 50
- providerConf.PasswordValidation.Users.MinEntropy = 70
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- _, resp, err := httpdtest.AddAdmin(a, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "insecure password")
- _, resp, err = httpdtest.AddUser(getTestUser(), http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "insecure password")
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestAdminPasswordHashing(t *testing.T) {
- if config.GetProviderConf().Driver == dataprovider.MemoryDataProviderName {
- t.Skip("this test is not supported with the memory provider")
- }
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- providerConf := config.GetProviderConf()
- assert.NoError(t, err)
- providerConf.PasswordHashing.Algo = dataprovider.HashingAlgoArgon2ID
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- currentAdmin, err := dataprovider.AdminExists(defaultTokenAuthUser)
- assert.NoError(t, err)
- assert.True(t, strings.HasPrefix(currentAdmin.Password, "$2a$"))
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- newAdmin, err := dataprovider.AdminExists(altAdminUsername)
- assert.NoError(t, err)
- assert.True(t, strings.HasPrefix(newAdmin.Password, "$argon2id$"))
- token, _, err := httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- _, _, err = httpdtest.GetStatus(http.StatusOK)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- _, _, err = httpdtest.GetStatus(http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- assert.NoError(t, err)
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestDefaultUsersExpiration(t *testing.T) {
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Filters.Preferences.DefaultUsersExpiration = 30
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- token, _, err := httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- _, _, err = httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.Error(t, err)
- user, _, err := httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, user.ExpirationDate, int64(0))
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- u := getTestUser()
- u.ExpirationDate = util.GetTimeAsMsSinceEpoch(time.Now().Add(1 * time.Minute))
- _, _, err = httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, u.ExpirationDate, user.ExpirationDate)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- _, _, err = httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- // render the user template page
- webToken, err := getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webTemplateUser, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webTemplateUser+fmt.Sprintf("?from=%s", user.Username), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- _, _, err = httpdtest.AddUser(u, http.StatusNotFound)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- }
- func TestAdminInvalidCredentials(t *testing.T) {
- req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultTokenAuthUser, defaultTokenAuthPass)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- // wrong password
- req.SetBasicAuth(defaultTokenAuthUser, "wrong pwd")
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- err = resp.Body.Close()
- assert.NoError(t, err)
- assert.Equal(t, dataprovider.ErrInvalidCredentials.Error(), responseHolder["error"].(string))
- // wrong username
- req.SetBasicAuth("wrong username", defaultTokenAuthPass)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusUnauthorized, resp.StatusCode)
- responseHolder = make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- assert.NoError(t, err)
- err = resp.Body.Close()
- assert.NoError(t, err)
- assert.Equal(t, dataprovider.ErrInvalidCredentials.Error(), responseHolder["error"].(string))
- }
- func TestAdminLastLogin(t *testing.T) {
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), admin.LastLogin)
- _, _, err = httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, admin.LastLogin, int64(0))
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAdminAllowList(t *testing.T) {
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- token, _, err := httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- _, _, err = httpdtest.GetStatus(http.StatusOK)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- admin.Password = altAdminPassword
- admin.Filters.AllowList = []string{"10.6.6.0/32"}
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetToken(altAdminUsername, altAdminPassword)
- assert.EqualError(t, err, "wrong status code: got 401 want 200")
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserStatus(t *testing.T) {
- u := getTestUser()
- u.Status = 3
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Status = 0
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user.Status = 2
- _, _, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- user.Status = 1
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUidGidLimits(t *testing.T) {
- u := getTestUser()
- u.UID = math.MaxInt32
- u.GID = math.MaxInt32
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, math.MaxInt32, user.GetUID())
- assert.Equal(t, math.MaxInt32, user.GetGID())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAddUserNoCredentials(t *testing.T) {
- u := getTestUser()
- u.Password = ""
- u.PublicKeys = []string{}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- // this user cannot login with an empty password but it still can use an SSH cert
- _, err = getJWTAPITokenFromTestServer(defaultTokenAuthUser, "")
- assert.Error(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAddUserNoUsername(t *testing.T) {
- u := getTestUser()
- u.Username = ""
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAddUserNoHomeDir(t *testing.T) {
- u := getTestUser()
- u.HomeDir = ""
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAddUserInvalidHomeDir(t *testing.T) {
- u := getTestUser()
- u.HomeDir = "relative_path" //nolint:goconst
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAddUserNoPerms(t *testing.T) {
- u := getTestUser()
- u.Permissions = make(map[string][]string)
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Permissions["/"] = []string{}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAddUserInvalidEmail(t *testing.T) {
- u := getTestUser()
- u.Email = "invalid_email"
- _, body, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(body), "Validation error: email")
- }
- func TestAddUserInvalidPerms(t *testing.T) {
- u := getTestUser()
- u.Permissions["/"] = []string{"invalidPerm"}
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- // permissions for root dir are mandatory
- u.Permissions["/"] = []string{}
- u.Permissions["/somedir"] = []string{dataprovider.PermAny}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Permissions["/"] = []string{dataprovider.PermAny}
- u.Permissions["/subdir/.."] = []string{dataprovider.PermAny}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAddUserInvalidFilters(t *testing.T) {
- u := getTestUser()
- u.Filters.AllowedIP = []string{"192.168.1.0/24", "192.168.2.0"}
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.AllowedIP = []string{}
- u.Filters.DeniedIP = []string{"192.168.3.0/16", "invalid"}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedIP = []string{}
- u.Filters.DeniedLoginMethods = []string{"invalid"}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedLoginMethods = dataprovider.ValidLoginMethods
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodTLSCertificateAndPwd}
- u.Filters.DeniedProtocols = dataprovider.ValidProtocols
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedProtocols = []string{common.ProtocolFTP}
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "relative",
- AllowedPatterns: []string{},
- DeniedPatterns: []string{},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/",
- AllowedPatterns: []string{},
- DeniedPatterns: []string{},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/subdir",
- AllowedPatterns: []string{"*.zip"},
- DeniedPatterns: []string{},
- },
- {
- Path: "/subdir",
- AllowedPatterns: []string{"*.rar"},
- DeniedPatterns: []string{"*.jpg"},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "relative",
- AllowedPatterns: []string{},
- DeniedPatterns: []string{},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/",
- AllowedPatterns: []string{},
- DeniedPatterns: []string{},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/subdir",
- AllowedPatterns: []string{"*.zip"},
- },
- {
- Path: "/subdir",
- AllowedPatterns: []string{"*.rar"},
- DeniedPatterns: []string{"*.jpg"},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/subdir",
- AllowedPatterns: []string{"a\\"},
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/subdir",
- AllowedPatterns: []string{"*.*"},
- DenyPolicy: 100,
- },
- }
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedProtocols = []string{"invalid"}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedProtocols = dataprovider.ValidProtocols
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.DeniedProtocols = nil
- u.Filters.TLSUsername = "not a supported attribute"
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.Filters.TLSUsername = ""
- u.Filters.WebClient = []string{"not a valid web client options"}
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestAddUserInvalidFsConfig(t *testing.T) {
- u := getTestUser()
- u.FsConfig.Provider = sdk.S3FilesystemProvider
- u.FsConfig.S3Config.Bucket = ""
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.S3Config.Bucket = "testbucket"
- u.FsConfig.S3Config.Region = "eu-west-1" //nolint:goconst
- u.FsConfig.S3Config.AccessKey = "access-key" //nolint:goconst
- u.FsConfig.S3Config.AccessSecret = kms.NewSecret(sdkkms.SecretStatusRedacted, "access-secret", "", "")
- u.FsConfig.S3Config.Endpoint = "http://127.0.0.1:9000/path?a=b"
- u.FsConfig.S3Config.StorageClass = "Standard" //nolint:goconst
- u.FsConfig.S3Config.KeyPrefix = "/adir/subdir/"
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.S3Config.AccessSecret.SetStatus(sdkkms.SecretStatusPlain)
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.S3Config.KeyPrefix = ""
- u.FsConfig.S3Config.UploadPartSize = 3
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.S3Config.UploadPartSize = 5001
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.S3Config.UploadPartSize = 0
- u.FsConfig.S3Config.UploadConcurrency = -1
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.S3Config.UploadConcurrency = 0
- u.FsConfig.S3Config.DownloadPartSize = -1
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "download_part_size cannot be")
- }
- u.FsConfig.S3Config.DownloadPartSize = 5001
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "download_part_size cannot be")
- }
- u.FsConfig.S3Config.DownloadPartSize = 0
- u.FsConfig.S3Config.DownloadConcurrency = 100
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid download concurrency")
- }
- u.FsConfig.S3Config.DownloadConcurrency = -1
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid download concurrency")
- }
- u.FsConfig.S3Config.DownloadConcurrency = 0
- u.FsConfig.S3Config.Endpoint = ""
- u.FsConfig.S3Config.Region = ""
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "region cannot be empty")
- }
- u = getTestUser()
- u.FsConfig.Provider = sdk.GCSFilesystemProvider
- u.FsConfig.GCSConfig.Bucket = ""
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.GCSConfig.Bucket = "abucket"
- u.FsConfig.GCSConfig.StorageClass = "Standard"
- u.FsConfig.GCSConfig.KeyPrefix = "/somedir/subdir/"
- u.FsConfig.GCSConfig.Credentials = kms.NewSecret(sdkkms.SecretStatusRedacted, "test", "", "") //nolint:goconst
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.GCSConfig.Credentials.SetStatus(sdkkms.SecretStatusPlain)
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.GCSConfig.KeyPrefix = "somedir/subdir/" //nolint:goconst
- u.FsConfig.GCSConfig.Credentials = kms.NewEmptySecret()
- u.FsConfig.GCSConfig.AutomaticCredentials = 0
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.GCSConfig.Credentials = kms.NewSecret(sdkkms.SecretStatusSecretBox, "invalid", "", "")
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u = getTestUser()
- u.FsConfig.Provider = sdk.AzureBlobFilesystemProvider
- u.FsConfig.AzBlobConfig.SASURL = kms.NewPlainSecret("http://foo\x7f.com/")
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.SASURL = kms.NewSecret(sdkkms.SecretStatusRedacted, "key", "", "")
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.SASURL = kms.NewEmptySecret()
- u.FsConfig.AzBlobConfig.AccountName = "name"
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.Container = "container"
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.AccountKey = kms.NewSecret(sdkkms.SecretStatusRedacted, "key", "", "")
- u.FsConfig.AzBlobConfig.KeyPrefix = "/amedir/subdir/"
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.AccountKey.SetStatus(sdkkms.SecretStatusPlain)
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.KeyPrefix = "amedir/subdir/"
- u.FsConfig.AzBlobConfig.UploadPartSize = -1
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.AzBlobConfig.UploadPartSize = 101
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u = getTestUser()
- u.FsConfig.Provider = sdk.CryptedFilesystemProvider
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.CryptConfig.Passphrase = kms.NewSecret(sdkkms.SecretStatusRedacted, "akey", "", "")
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u = getTestUser()
- u.FsConfig.Provider = sdk.SFTPFilesystemProvider
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.SFTPConfig.Password = kms.NewSecret(sdkkms.SecretStatusRedacted, "randompkey", "", "")
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.SFTPConfig.Password = kms.NewEmptySecret()
- u.FsConfig.SFTPConfig.PrivateKey = kms.NewSecret(sdkkms.SecretStatusRedacted, "keyforpkey", "", "")
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret("pk")
- u.FsConfig.SFTPConfig.Endpoint = "127.1.1.1:22"
- u.FsConfig.SFTPConfig.Username = defaultUsername
- u.FsConfig.SFTPConfig.BufferSize = -1
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid buffer_size")
- }
- u.FsConfig.SFTPConfig.BufferSize = 1000
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid buffer_size")
- }
- u = getTestUser()
- u.FsConfig.Provider = sdk.HTTPFilesystemProvider
- u.FsConfig.HTTPConfig.Endpoint = "http://foo\x7f.com/"
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid endpoint")
- }
- u.FsConfig.HTTPConfig.Endpoint = "http://127.0.0.1:9999/api/v1"
- u.FsConfig.HTTPConfig.Password = kms.NewSecret(sdkkms.SecretStatusSecretBox, "", "", "")
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid encrypted password")
- }
- u.FsConfig.HTTPConfig.Password = nil
- u.FsConfig.HTTPConfig.APIKey = kms.NewSecret(sdkkms.SecretStatusRedacted, redactedSecret, "", "")
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "cannot save a user with a redacted secret")
- }
- u.FsConfig.HTTPConfig.APIKey = nil
- u.FsConfig.HTTPConfig.Endpoint = "/api/v1"
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid endpoint schema")
- }
- u.FsConfig.HTTPConfig.Endpoint = "http://unix?api_prefix=v1"
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid unix domain socket path")
- }
- u.FsConfig.HTTPConfig.Endpoint = "http://unix?socket_path=test.sock"
- _, resp, err = httpdtest.AddUser(u, http.StatusBadRequest)
- if assert.NoError(t, err) {
- assert.Contains(t, string(resp), "invalid unix domain socket path")
- }
- }
- func TestUserRedactedPassword(t *testing.T) {
- u := getTestUser()
- u.FsConfig.Provider = sdk.S3FilesystemProvider
- u.FsConfig.S3Config.Bucket = "b"
- u.FsConfig.S3Config.Region = "eu-west-1"
- u.FsConfig.S3Config.AccessKey = "access-key"
- u.FsConfig.S3Config.RoleARN = "myRoleARN"
- u.FsConfig.S3Config.AccessSecret = kms.NewSecret(sdkkms.SecretStatusRedacted, "access-secret", "", "")
- u.FsConfig.S3Config.Endpoint = "http://127.0.0.1:9000/path?k=m"
- u.FsConfig.S3Config.StorageClass = "Standard"
- u.FsConfig.S3Config.ACL = "bucket-owner-full-control"
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err, string(resp))
- assert.Contains(t, string(resp), "cannot save a user with a redacted secret")
- err = dataprovider.AddUser(&u, "", "", "")
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "cannot save a user with a redacted secret")
- }
- u.FsConfig.S3Config.AccessSecret = kms.NewPlainSecret("secret")
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- folderName := "folderName"
- vfolder := vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: filepath.Join(os.TempDir(), "crypted"),
- FsConfig: vfs.Filesystem{
- Provider: sdk.CryptedFilesystemProvider,
- CryptConfig: vfs.CryptFsConfig{
- Passphrase: kms.NewSecret(sdkkms.SecretStatusRedacted, "crypted-secret", "", ""),
- },
- },
- },
- VirtualPath: "/avpath",
- }
- user.Password = defaultPassword
- user.VirtualFolders = append(user.VirtualFolders, vfolder)
- err = dataprovider.UpdateUser(&user, "", "", "")
- if assert.Error(t, err) {
- assert.Contains(t, err.Error(), "cannot save a user with a redacted secret")
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserType(t *testing.T) {
- u := getTestUser()
- u.Filters.UserType = string(sdk.UserTypeLDAP)
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, string(sdk.UserTypeLDAP), user.Filters.UserType)
- user.Filters.UserType = string(sdk.UserTypeOS)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, string(sdk.UserTypeOS), user.Filters.UserType)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestRetentionAPI(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- checks, _, err := httpdtest.GetRetentionChecks(http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, checks, 0)
- localFilePath := filepath.Join(user.HomeDir, "testdir", "testfile")
- err = os.MkdirAll(filepath.Dir(localFilePath), os.ModePerm)
- assert.NoError(t, err)
- err = os.WriteFile(localFilePath, []byte("test data"), os.ModePerm)
- assert.NoError(t, err)
- folderRetention := []dataprovider.FolderRetention{
- {
- Path: "/",
- Retention: 0,
- DeleteEmptyDirs: true,
- },
- }
- _, err = httpdtest.StartRetentionCheck(altAdminUsername, folderRetention, http.StatusNotFound)
- assert.NoError(t, err)
- resp, err := httpdtest.StartRetentionCheck(user.Username, folderRetention, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "Invalid retention check")
- folderRetention[0].Retention = 24
- _, err = httpdtest.StartRetentionCheck(user.Username, folderRetention, http.StatusAccepted)
- assert.NoError(t, err)
- assert.Eventually(t, func() bool {
- return len(common.RetentionChecks.Get("")) == 0
- }, 1000*time.Millisecond, 50*time.Millisecond)
- assert.FileExists(t, localFilePath)
- err = os.Chtimes(localFilePath, time.Now().Add(-48*time.Hour), time.Now().Add(-48*time.Hour))
- assert.NoError(t, err)
- _, err = httpdtest.StartRetentionCheck(user.Username, folderRetention, http.StatusAccepted)
- assert.NoError(t, err)
- assert.Eventually(t, func() bool {
- return len(common.RetentionChecks.Get("")) == 0
- }, 1000*time.Millisecond, 50*time.Millisecond)
- assert.NoFileExists(t, localFilePath)
- assert.NoDirExists(t, filepath.Dir(localFilePath))
- check := common.RetentionCheck{
- Folders: folderRetention,
- }
- c := common.RetentionChecks.Add(check, &user)
- assert.NotNil(t, c)
- _, err = httpdtest.StartRetentionCheck(user.Username, folderRetention, http.StatusConflict)
- assert.NoError(t, err)
- err = c.Start()
- assert.NoError(t, err)
- assert.Len(t, common.RetentionChecks.Get(""), 0)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err = httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, retentionBasePath+"/"+user.Username+"/check",
- bytes.NewBuffer([]byte("invalid json")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- asJSON, err := json.Marshal(folderRetention)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPost, retentionBasePath+"/"+user.Username+"/check?notifications=Email,",
- bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "to notify results via email")
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPost, retentionBasePath+"/"+user.Username+"/check?notifications=Email",
- bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestAddUserInvalidVirtualFolders(t *testing.T) {
- u := getTestUser()
- folderName := "fname"
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir"),
- Name: folderName,
- },
- VirtualPath: "/vdir",
- })
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir1"),
- Name: folderName + "1",
- },
- VirtualPath: "/vdir", // invalid, already defined
- })
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.VirtualFolders = nil
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir"),
- Name: folderName,
- },
- VirtualPath: "/vdir1",
- })
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir"),
- Name: folderName, // invalid, unique constraint (user.id, folder.id) violated
- },
- VirtualPath: "/vdir2",
- })
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.VirtualFolders = nil
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir1"),
- Name: folderName + "1",
- },
- VirtualPath: "/vdir1/",
- QuotaSize: -1,
- QuotaFiles: 1, // invvalid, we cannot have -1 and > 0
- })
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.VirtualFolders = nil
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir1"),
- Name: folderName + "1",
- },
- VirtualPath: "/vdir1/",
- QuotaSize: 1,
- QuotaFiles: -1,
- })
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.VirtualFolders = nil
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir1"),
- Name: folderName + "1",
- },
- VirtualPath: "/vdir1/",
- QuotaSize: -2, // invalid
- QuotaFiles: 0,
- })
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.VirtualFolders = nil
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir1"),
- Name: folderName + "1",
- },
- VirtualPath: "/vdir1/",
- QuotaSize: 0,
- QuotaFiles: -2, // invalid
- })
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.VirtualFolders = nil
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- MappedPath: filepath.Join(os.TempDir(), "mapped_dir"),
- },
- VirtualPath: "/vdir1",
- })
- // folder name is mandatory
- _, _, err = httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- }
- func TestUserPublicKey(t *testing.T) {
- u := getTestUser()
- u.Password = ""
- invalidPubKey := "invalid"
- u.PublicKeys = []string{invalidPubKey}
- _, _, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- u.PublicKeys = []string{testPubKey}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- dbUser, err := dataprovider.UserExists(u.Username, "")
- assert.NoError(t, err)
- assert.Empty(t, dbUser.Password)
- assert.False(t, dbUser.IsPasswordHashed())
- user.PublicKeys = []string{testPubKey, invalidPubKey}
- _, _, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- user.PublicKeys = []string{testPubKey, testPubKey, testPubKey}
- user.Password = defaultPassword
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- dbUser, err = dataprovider.UserExists(u.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUserEmptyPassword(t *testing.T) {
- u := getTestUser()
- u.PublicKeys = []string{testPubKey}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- // the password is not empty
- dbUser, err := dataprovider.UserExists(u.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- // now update the user and set an empty password
- data, err := json.Marshal(dbUser)
- assert.NoError(t, err)
- var customUser map[string]any
- err = json.Unmarshal(data, &customUser)
- assert.NoError(t, err)
- customUser["password"] = ""
- asJSON, err := json.Marshal(customUser)
- assert.NoError(t, err)
- userNoPwd, _, err := httpdtest.UpdateUserWithJSON(user, http.StatusOK, "", asJSON)
- assert.NoError(t, err)
- assert.Equal(t, user.Password, userNoPwd.Password) // the password is hidden
- // check the password within the data provider
- dbUser, err = dataprovider.UserExists(u.Username, "")
- assert.NoError(t, err)
- assert.Empty(t, dbUser.Password)
- assert.False(t, dbUser.IsPasswordHashed())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUserNoPassword(t *testing.T) {
- u := getTestUser()
- u.PublicKeys = []string{testPubKey}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- // the password is not empty
- dbUser, err := dataprovider.UserExists(u.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- // now update the user and remove the password field, old password should be preserved
- user.Password = "" // password has the omitempty tag
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- // the password is preserved
- dbUser, err = dataprovider.UserExists(u.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUser(t *testing.T) {
- u := getTestUser()
- u.UsedQuotaFiles = 1
- u.UsedQuotaSize = 2
- u.Filters.TLSUsername = sdk.TLSUsernameCN
- u.Filters.Hooks.CheckPasswordDisabled = true
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, 0, user.UsedQuotaFiles)
- assert.Equal(t, int64(0), user.UsedQuotaSize)
- user.HomeDir = filepath.Join(homeBasePath, "testmod")
- user.UID = 33
- user.GID = 101
- user.MaxSessions = 10
- user.QuotaSize = 4096
- user.QuotaFiles = 2
- user.Permissions["/"] = []string{dataprovider.PermCreateDirs, dataprovider.PermDelete, dataprovider.PermDownload}
- user.Permissions["/subdir"] = []string{dataprovider.PermListItems, dataprovider.PermUpload}
- user.Filters.AllowedIP = []string{"192.168.1.0/24", "192.168.2.0/24"}
- user.Filters.DeniedIP = []string{"192.168.3.0/24", "192.168.4.0/24"}
- user.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
- user.Filters.DeniedProtocols = []string{common.ProtocolWebDAV}
- user.Filters.TLSUsername = sdk.TLSUsernameNone
- user.Filters.Hooks.ExternalAuthDisabled = true
- user.Filters.Hooks.PreLoginDisabled = true
- user.Filters.Hooks.CheckPasswordDisabled = false
- user.Filters.DisableFsChecks = true
- user.Filters.FilePatterns = append(user.Filters.FilePatterns, sdk.PatternsFilter{
- Path: "/subdir",
- AllowedPatterns: []string{"*.zip", "*.rar"},
- DeniedPatterns: []string{"*.jpg", "*.png"},
- DenyPolicy: sdk.DenyPolicyHide,
- })
- user.Filters.MaxUploadFileSize = 4096
- user.UploadBandwidth = 1024
- user.DownloadBandwidth = 512
- user.VirtualFolders = nil
- mappedPath1 := filepath.Join(os.TempDir(), "mapped_dir1")
- mappedPath2 := filepath.Join(os.TempDir(), "mapped_dir2")
- folderName1 := filepath.Base(mappedPath1)
- folderName2 := filepath.Base(mappedPath2)
- user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir1",
- })
- user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName2,
- },
- VirtualPath: "/vdir12/subdir",
- QuotaSize: 123,
- QuotaFiles: 2,
- })
- f1 := vfs.BaseVirtualFolder{
- Name: folderName1,
- MappedPath: mappedPath1,
- }
- _, _, err = httpdtest.AddFolder(f1, http.StatusCreated)
- assert.NoError(t, err)
- f2 := vfs.BaseVirtualFolder{
- Name: folderName2,
- MappedPath: mappedPath2,
- }
- _, _, err = httpdtest.AddFolder(f2, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "invalid")
- assert.NoError(t, err)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "0")
- assert.NoError(t, err)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "1")
- assert.NoError(t, err)
- user.Permissions["/subdir"] = []string{}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Len(t, user.Permissions["/subdir"], 0)
- assert.Len(t, user.VirtualFolders, 2)
- for _, folder := range user.VirtualFolders {
- assert.Greater(t, folder.ID, int64(0))
- if folder.VirtualPath == "/vdir12/subdir" {
- assert.Equal(t, int64(123), folder.QuotaSize)
- assert.Equal(t, 2, folder.QuotaFiles)
- }
- }
- folder, _, err := httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user.Username)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- // removing the user must remove folder mapping
- folder, _, err = httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 0)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 0)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUserTransferQuotaUsage(t *testing.T) {
- u := getTestUser()
- usedDownloadDataTransfer := int64(2 * 1024 * 1024)
- usedUploadDataTransfer := int64(1024 * 1024)
- u.UsedDownloadDataTransfer = usedDownloadDataTransfer
- u.UsedUploadDataTransfer = usedUploadDataTransfer
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), user.UsedUploadDataTransfer)
- assert.Equal(t, int64(0), user.UsedDownloadDataTransfer)
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "invalid_mode", http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedUploadDataTransfer, user.UsedUploadDataTransfer)
- assert.Equal(t, usedDownloadDataTransfer, user.UsedDownloadDataTransfer)
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "add", http.StatusBadRequest)
- assert.NoError(t, err, "user has no transfer quota restrictions add mode should fail")
- user.TotalDataTransfer = 100
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "add", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 2*usedUploadDataTransfer, user.UsedUploadDataTransfer)
- assert.Equal(t, 2*usedDownloadDataTransfer, user.UsedDownloadDataTransfer)
- u.UsedDownloadDataTransfer = -1
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "add", http.StatusBadRequest)
- assert.NoError(t, err)
- u.UsedDownloadDataTransfer = usedDownloadDataTransfer
- u.Username += "1"
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "", http.StatusNotFound)
- assert.NoError(t, err)
- u.Username = defaultUsername
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedUploadDataTransfer, user.UsedUploadDataTransfer)
- assert.Equal(t, usedDownloadDataTransfer, user.UsedDownloadDataTransfer)
- u.UsedDownloadDataTransfer = 0
- u.UsedUploadDataTransfer = 1
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "add", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedUploadDataTransfer+1, user.UsedUploadDataTransfer)
- assert.Equal(t, usedDownloadDataTransfer, user.UsedDownloadDataTransfer)
- u.UsedDownloadDataTransfer = 1
- u.UsedUploadDataTransfer = 0
- _, err = httpdtest.UpdateTransferQuotaUsage(u, "add", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedUploadDataTransfer+1, user.UsedUploadDataTransfer)
- assert.Equal(t, usedDownloadDataTransfer+1, user.UsedDownloadDataTransfer)
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "users", u.Username, "transfer-usage"),
- bytes.NewBuffer([]byte(`not a json`)))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUserQuotaUsage(t *testing.T) {
- u := getTestUser()
- usedQuotaFiles := 1
- usedQuotaSize := int64(65535)
- u.UsedQuotaFiles = usedQuotaFiles
- u.UsedQuotaSize = usedQuotaSize
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 0, user.UsedQuotaFiles)
- assert.Equal(t, int64(0), user.UsedQuotaSize)
- _, err = httpdtest.UpdateQuotaUsage(u, "invalid_mode", http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.UpdateQuotaUsage(u, "", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, user.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize, user.UsedQuotaSize)
- _, err = httpdtest.UpdateQuotaUsage(u, "add", http.StatusBadRequest)
- assert.NoError(t, err, "user has no quota restrictions add mode should fail")
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, user.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize, user.UsedQuotaSize)
- user.QuotaFiles = 100
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = httpdtest.UpdateQuotaUsage(u, "add", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 2*usedQuotaFiles, user.UsedQuotaFiles)
- assert.Equal(t, 2*usedQuotaSize, user.UsedQuotaSize)
- u.UsedQuotaFiles = -1
- _, err = httpdtest.UpdateQuotaUsage(u, "", http.StatusBadRequest)
- assert.NoError(t, err)
- u.UsedQuotaFiles = usedQuotaFiles
- u.Username = u.Username + "1"
- _, err = httpdtest.UpdateQuotaUsage(u, "", http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserFolderMapping(t *testing.T) {
- mappedPath1 := filepath.Join(os.TempDir(), "mapped_dir1")
- mappedPath2 := filepath.Join(os.TempDir(), "mapped_dir2")
- folderName1 := filepath.Base(mappedPath1)
- folderName2 := filepath.Base(mappedPath2)
- u1 := getTestUser()
- u1.VirtualFolders = append(u1.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- },
- VirtualPath: "/vdir",
- QuotaSize: -1,
- QuotaFiles: -1,
- })
- f1 := vfs.BaseVirtualFolder{
- Name: folderName1,
- MappedPath: mappedPath1,
- UsedQuotaFiles: 2,
- UsedQuotaSize: 123,
- LastQuotaUpdate: 456,
- }
- _, _, err := httpdtest.AddFolder(f1, http.StatusCreated)
- assert.NoError(t, err)
- f2 := vfs.BaseVirtualFolder{
- Name: folderName2,
- MappedPath: mappedPath2,
- }
- _, _, err = httpdtest.AddFolder(f2, http.StatusCreated)
- assert.NoError(t, err)
- user1, _, err := httpdtest.AddUser(u1, http.StatusCreated)
- assert.NoError(t, err)
- // virtual folder must be auto created
- folder, _, err := httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user1.Username)
- assert.Equal(t, 2, folder.UsedQuotaFiles)
- assert.Equal(t, int64(123), folder.UsedQuotaSize)
- assert.Equal(t, int64(456), folder.LastQuotaUpdate)
- assert.Equal(t, 2, user1.VirtualFolders[0].UsedQuotaFiles)
- assert.Equal(t, int64(123), user1.VirtualFolders[0].UsedQuotaSize)
- assert.Equal(t, int64(456), user1.VirtualFolders[0].LastQuotaUpdate)
- u2 := getTestUser()
- u2.Username = defaultUsername + "2"
- u2.VirtualFolders = append(u2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- MappedPath: mappedPath1,
- },
- VirtualPath: "/vdir1",
- QuotaSize: 0,
- QuotaFiles: 0,
- })
- u2.VirtualFolders = append(u2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName2,
- MappedPath: mappedPath2,
- },
- VirtualPath: "/vdir2",
- QuotaSize: -1,
- QuotaFiles: -1,
- })
- user2, _, err := httpdtest.AddUser(u2, http.StatusCreated)
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user2.Username)
- folder, _, err = httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 2)
- assert.Contains(t, folder.Users, user1.Username)
- assert.Contains(t, folder.Users, user2.Username)
- // now update user2 removing mappedPath1
- user2.VirtualFolders = nil
- user2.VirtualFolders = append(user2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName2,
- MappedPath: mappedPath2,
- UsedQuotaFiles: 2,
- UsedQuotaSize: 123,
- },
- VirtualPath: "/vdir",
- QuotaSize: 0,
- QuotaFiles: 0,
- })
- user2, _, err = httpdtest.UpdateUser(user2, http.StatusOK, "")
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user2.Username)
- assert.Equal(t, 0, folder.UsedQuotaFiles)
- assert.Equal(t, int64(0), folder.UsedQuotaSize)
- folder, _, err = httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user1.Username)
- // add mappedPath1 again to user2
- user2.VirtualFolders = append(user2.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName1,
- MappedPath: mappedPath1,
- },
- VirtualPath: "/vdir1",
- })
- user2, _, err = httpdtest.UpdateUser(user2, http.StatusOK, "")
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(folderName2, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user2.Username)
- // removing virtual folders should clear relations on both side
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName2}, http.StatusOK)
- assert.NoError(t, err)
- user2, _, err = httpdtest.GetUserByUsername(user2.Username, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, user2.VirtualFolders, 1) {
- folder := user2.VirtualFolders[0]
- assert.Equal(t, mappedPath1, folder.MappedPath)
- assert.Equal(t, folderName1, folder.Name)
- }
- user1, _, err = httpdtest.GetUserByUsername(user1.Username, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, user2.VirtualFolders, 1) {
- folder := user2.VirtualFolders[0]
- assert.Equal(t, mappedPath1, folder.MappedPath)
- }
- folder, _, err = httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 2)
- // removing a user should clear virtual folder mapping
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(folderName1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Contains(t, folder.Users, user2.Username)
- // removing a folder should clear mapping on the user side too
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName1}, http.StatusOK)
- assert.NoError(t, err)
- user2, _, err = httpdtest.GetUserByUsername(user2.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, user2.VirtualFolders, 0)
- _, err = httpdtest.RemoveUser(user2, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserS3Config(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.S3FilesystemProvider
- user.FsConfig.S3Config.Bucket = "test" //nolint:goconst
- user.FsConfig.S3Config.AccessKey = "Server-Access-Key"
- user.FsConfig.S3Config.AccessSecret = kms.NewPlainSecret("Server-Access-Secret")
- user.FsConfig.S3Config.RoleARN = "myRoleARN"
- user.FsConfig.S3Config.Endpoint = "http://127.0.0.1:9000"
- user.FsConfig.S3Config.UploadPartSize = 8
- user.FsConfig.S3Config.DownloadPartMaxTime = 60
- user.FsConfig.S3Config.UploadPartMaxTime = 40
- user.FsConfig.S3Config.ForcePathStyle = true
- user.FsConfig.S3Config.SkipTLSVerify = true
- user.FsConfig.S3Config.DownloadPartSize = 6
- folderName := "vfolderName"
- user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- },
- VirtualPath: "/folderPath",
- })
- f := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: filepath.Join(os.TempDir(), "folderName"),
- FsConfig: vfs.Filesystem{
- Provider: sdk.CryptedFilesystemProvider,
- CryptConfig: vfs.CryptFsConfig{
- Passphrase: kms.NewPlainSecret("Crypted-Secret"),
- },
- },
- }
- _, _, err = httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err)
- user, body, err := httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(body))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.NotEmpty(t, user.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Empty(t, user.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.Empty(t, user.FsConfig.S3Config.AccessSecret.GetKey())
- assert.Equal(t, 60, user.FsConfig.S3Config.DownloadPartMaxTime)
- assert.Equal(t, 40, user.FsConfig.S3Config.UploadPartMaxTime)
- assert.True(t, user.FsConfig.S3Config.SkipTLSVerify)
- if assert.Len(t, user.VirtualFolders, 1) {
- folder := user.VirtualFolders[0]
- assert.Equal(t, sdkkms.SecretStatusSecretBox, folder.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, folder.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, folder.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, folder.FsConfig.CryptConfig.Passphrase.GetKey())
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, folder.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, folder.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, folder.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, folder.FsConfig.CryptConfig.Passphrase.GetKey())
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.ID = 0
- user.CreatedAt = 0
- user.VirtualFolders = nil
- secret := kms.NewSecret(sdkkms.SecretStatusSecretBox, "Server-Access-Secret", "", "")
- user.FsConfig.S3Config.AccessSecret = secret
- _, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.Error(t, err)
- user.FsConfig.S3Config.AccessSecret.SetStatus(sdkkms.SecretStatusPlain)
- user, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err)
- initialSecretPayload := user.FsConfig.S3Config.AccessSecret.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.NotEmpty(t, initialSecretPayload)
- assert.Empty(t, user.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.Empty(t, user.FsConfig.S3Config.AccessSecret.GetKey())
- user.FsConfig.Provider = sdk.S3FilesystemProvider
- user.FsConfig.S3Config.Bucket = "test-bucket"
- user.FsConfig.S3Config.Region = "us-east-1" //nolint:goconst
- user.FsConfig.S3Config.AccessKey = "Server-Access-Key1"
- user.FsConfig.S3Config.Endpoint = "http://localhost:9000"
- user.FsConfig.S3Config.KeyPrefix = "somedir/subdir" //nolint:goconst
- user.FsConfig.S3Config.UploadConcurrency = 5
- user.FsConfig.S3Config.DownloadConcurrency = 4
- user, bb, err := httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(bb))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.Equal(t, initialSecretPayload, user.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Empty(t, user.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.Empty(t, user.FsConfig.S3Config.AccessSecret.GetKey())
- // test user without access key and access secret (shared config state)
- user.FsConfig.Provider = sdk.S3FilesystemProvider
- user.FsConfig.S3Config.Bucket = "testbucket"
- user.FsConfig.S3Config.Region = "us-east-1"
- user.FsConfig.S3Config.AccessKey = ""
- user.FsConfig.S3Config.AccessSecret = kms.NewEmptySecret()
- user.FsConfig.S3Config.Endpoint = ""
- user.FsConfig.S3Config.KeyPrefix = "somedir/subdir"
- user.FsConfig.S3Config.UploadPartSize = 6
- user.FsConfig.S3Config.UploadConcurrency = 4
- user, body, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(body))
- assert.Nil(t, user.FsConfig.S3Config.AccessSecret)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.ID = 0
- user.CreatedAt = 0
- // shared credential test for add instead of update
- user, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err)
- assert.Nil(t, user.FsConfig.S3Config.AccessSecret)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestHTTPFsConfig(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.HTTPFilesystemProvider
- user.FsConfig.HTTPConfig = vfs.HTTPFsConfig{
- BaseHTTPFsConfig: sdk.BaseHTTPFsConfig{
- Endpoint: "http://127.0.0.1/httpfs",
- Username: defaultUsername,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- APIKey: kms.NewPlainSecret(defaultTokenAuthUser),
- }
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- initialPwdPayload := user.FsConfig.HTTPConfig.Password.GetPayload()
- initialAPIKeyPayload := user.FsConfig.HTTPConfig.APIKey.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, initialPwdPayload)
- assert.Empty(t, user.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Empty(t, user.FsConfig.HTTPConfig.Password.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.NotEmpty(t, initialAPIKeyPayload)
- assert.Empty(t, user.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.HTTPConfig.APIKey.GetKey())
- user.FsConfig.HTTPConfig.Password.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.HTTPConfig.Password.SetAdditionalData(util.GenerateUniqueID())
- user.FsConfig.HTTPConfig.Password.SetKey(util.GenerateUniqueID())
- user.FsConfig.HTTPConfig.APIKey.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.HTTPConfig.APIKey.SetAdditionalData(util.GenerateUniqueID())
- user.FsConfig.HTTPConfig.APIKey.SetKey(util.GenerateUniqueID())
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.HTTPConfig.Password.GetStatus())
- assert.Equal(t, initialPwdPayload, user.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, user.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Empty(t, user.FsConfig.HTTPConfig.Password.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.Equal(t, initialAPIKeyPayload, user.FsConfig.HTTPConfig.APIKey.GetPayload())
- assert.Empty(t, user.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.HTTPConfig.APIKey.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- // also test AddUser
- u := getTestUser()
- u.FsConfig.Provider = sdk.HTTPFilesystemProvider
- u.FsConfig.HTTPConfig = vfs.HTTPFsConfig{
- BaseHTTPFsConfig: sdk.BaseHTTPFsConfig{
- Endpoint: "http://127.0.0.1/httpfs",
- Username: defaultUsername,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- APIKey: kms.NewPlainSecret(defaultTokenAuthUser),
- }
- user, _, err = httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, user.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, user.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Empty(t, user.FsConfig.HTTPConfig.Password.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.NotEmpty(t, user.FsConfig.HTTPConfig.APIKey.GetPayload())
- assert.Empty(t, user.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.HTTPConfig.APIKey.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserAzureBlobConfig(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.AzureBlobFilesystemProvider
- user.FsConfig.AzBlobConfig.Container = "test"
- user.FsConfig.AzBlobConfig.AccountName = "Server-Account-Name"
- user.FsConfig.AzBlobConfig.AccountKey = kms.NewPlainSecret("Server-Account-Key")
- user.FsConfig.AzBlobConfig.Endpoint = "http://127.0.0.1:9000"
- user.FsConfig.AzBlobConfig.UploadPartSize = 8
- user.FsConfig.AzBlobConfig.DownloadPartSize = 6
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- initialPayload := user.FsConfig.AzBlobConfig.AccountKey.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetKey())
- user.FsConfig.AzBlobConfig.AccountKey.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.AzBlobConfig.AccountKey.SetAdditionalData("data")
- user.FsConfig.AzBlobConfig.AccountKey.SetKey("fake key")
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.Equal(t, initialPayload, user.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.ID = 0
- user.CreatedAt = 0
- secret := kms.NewSecret(sdkkms.SecretStatusSecretBox, "Server-Account-Key", "", "")
- user.FsConfig.AzBlobConfig.AccountKey = secret
- _, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.Error(t, err)
- user.FsConfig.AzBlobConfig.AccountKey = kms.NewPlainSecret("Server-Account-Key-Test")
- user, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err)
- initialPayload = user.FsConfig.AzBlobConfig.AccountKey.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetKey())
- user.FsConfig.Provider = sdk.AzureBlobFilesystemProvider
- user.FsConfig.AzBlobConfig.Container = "test-container"
- user.FsConfig.AzBlobConfig.Endpoint = "http://localhost:9001"
- user.FsConfig.AzBlobConfig.KeyPrefix = "somedir/subdir"
- user.FsConfig.AzBlobConfig.UploadConcurrency = 5
- user.FsConfig.AzBlobConfig.DownloadConcurrency = 4
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Equal(t, initialPayload, user.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.AzBlobConfig.AccountKey.GetKey())
- // test user without access key and access secret (SAS)
- user.FsConfig.Provider = sdk.AzureBlobFilesystemProvider
- user.FsConfig.AzBlobConfig.SASURL = kms.NewPlainSecret("https://myaccount.blob.core.windows.net/pictures/profile.jpg?sv=2012-02-12&st=2009-02-09&se=2009-02-10&sr=c&sp=r&si=YWJjZGVmZw%3d%3d&sig=dD80ihBh5jfNpymO5Hg1IdiJIEvHcJpCMiCMnN%2fRnbI%3d")
- user.FsConfig.AzBlobConfig.KeyPrefix = "somedir/subdir"
- user.FsConfig.AzBlobConfig.AccountName = ""
- user.FsConfig.AzBlobConfig.AccountKey = kms.NewEmptySecret()
- user.FsConfig.AzBlobConfig.UploadPartSize = 6
- user.FsConfig.AzBlobConfig.UploadConcurrency = 4
- user.FsConfig.AzBlobConfig.DownloadPartSize = 3
- user.FsConfig.AzBlobConfig.DownloadConcurrency = 5
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Nil(t, user.FsConfig.AzBlobConfig.AccountKey)
- assert.NotNil(t, user.FsConfig.AzBlobConfig.SASURL)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.ID = 0
- user.CreatedAt = 0
- // sas test for add instead of update
- user.FsConfig.AzBlobConfig = vfs.AzBlobFsConfig{
- BaseAzBlobFsConfig: sdk.BaseAzBlobFsConfig{
- Container: user.FsConfig.AzBlobConfig.Container,
- },
- SASURL: kms.NewPlainSecret("http://127.0.0.1/fake/sass/url"),
- }
- user, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err)
- assert.Nil(t, user.FsConfig.AzBlobConfig.AccountKey)
- initialPayload = user.FsConfig.AzBlobConfig.SASURL.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.AzBlobConfig.SASURL.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Empty(t, user.FsConfig.AzBlobConfig.SASURL.GetAdditionalData())
- assert.Empty(t, user.FsConfig.AzBlobConfig.SASURL.GetKey())
- user.FsConfig.AzBlobConfig.SASURL.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.AzBlobConfig.SASURL.SetAdditionalData("data")
- user.FsConfig.AzBlobConfig.SASURL.SetKey("fake key")
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.AzBlobConfig.SASURL.GetStatus())
- assert.Equal(t, initialPayload, user.FsConfig.AzBlobConfig.SASURL.GetPayload())
- assert.Empty(t, user.FsConfig.AzBlobConfig.SASURL.GetAdditionalData())
- assert.Empty(t, user.FsConfig.AzBlobConfig.SASURL.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserCryptFs(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.CryptedFilesystemProvider
- user.FsConfig.CryptConfig.Passphrase = kms.NewPlainSecret("crypt passphrase")
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- initialPayload := user.FsConfig.CryptConfig.Passphrase.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetKey())
- user.FsConfig.CryptConfig.Passphrase.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.CryptConfig.Passphrase.SetAdditionalData("data")
- user.FsConfig.CryptConfig.Passphrase.SetKey("fake pass key")
- user, bb, err := httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(bb))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.Equal(t, initialPayload, user.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.ID = 0
- user.CreatedAt = 0
- secret := kms.NewSecret(sdkkms.SecretStatusSecretBox, "invalid encrypted payload", "", "")
- user.FsConfig.CryptConfig.Passphrase = secret
- _, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.Error(t, err)
- user.FsConfig.CryptConfig.Passphrase = kms.NewPlainSecret("passphrase test")
- user, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err)
- initialPayload = user.FsConfig.CryptConfig.Passphrase.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetKey())
- user.FsConfig.Provider = sdk.CryptedFilesystemProvider
- user.FsConfig.CryptConfig.Passphrase.SetKey("pass")
- user, bb, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(bb))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, initialPayload)
- assert.Equal(t, initialPayload, user.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, user.FsConfig.CryptConfig.Passphrase.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserSFTPFs(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.SFTPFilesystemProvider
- user.FsConfig.SFTPConfig.Endpoint = "[::1]:22:22" // invalid endpoint
- user.FsConfig.SFTPConfig.Username = "sftp_user"
- user.FsConfig.SFTPConfig.Password = kms.NewPlainSecret("sftp_pwd")
- user.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret(sftpPrivateKey)
- user.FsConfig.SFTPConfig.Fingerprints = []string{sftpPkeyFingerprint}
- user.FsConfig.SFTPConfig.BufferSize = 2
- user.FsConfig.SFTPConfig.EqualityCheckMode = 1
- _, resp, err := httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "invalid endpoint")
- user.FsConfig.SFTPConfig.Endpoint = "127.0.0.1"
- _, _, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.Error(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, "127.0.0.1:22", user.FsConfig.SFTPConfig.Endpoint)
- user.FsConfig.SFTPConfig.Endpoint = "127.0.0.1:2022"
- user.FsConfig.SFTPConfig.DisableCouncurrentReads = true
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, "/", user.FsConfig.SFTPConfig.Prefix)
- assert.True(t, user.FsConfig.SFTPConfig.DisableCouncurrentReads)
- assert.Equal(t, int64(2), user.FsConfig.SFTPConfig.BufferSize)
- initialPwdPayload := user.FsConfig.SFTPConfig.Password.GetPayload()
- initialPkeyPayload := user.FsConfig.SFTPConfig.PrivateKey.GetPayload()
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.SFTPConfig.Password.GetStatus())
- assert.NotEmpty(t, initialPwdPayload)
- assert.Empty(t, user.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.Empty(t, user.FsConfig.SFTPConfig.Password.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.NotEmpty(t, initialPkeyPayload)
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetKey())
- user.FsConfig.SFTPConfig.Password.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.SFTPConfig.Password.SetAdditionalData("adata")
- user.FsConfig.SFTPConfig.Password.SetKey("fake pwd key")
- user.FsConfig.SFTPConfig.PrivateKey.SetStatus(sdkkms.SecretStatusSecretBox)
- user.FsConfig.SFTPConfig.PrivateKey.SetAdditionalData("adata")
- user.FsConfig.SFTPConfig.PrivateKey.SetKey("fake key")
- user.FsConfig.SFTPConfig.DisableCouncurrentReads = false
- user, bb, err := httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(bb))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.SFTPConfig.Password.GetStatus())
- assert.Equal(t, initialPwdPayload, user.FsConfig.SFTPConfig.Password.GetPayload())
- assert.Empty(t, user.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.Empty(t, user.FsConfig.SFTPConfig.Password.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.Equal(t, initialPkeyPayload, user.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetKey())
- assert.False(t, user.FsConfig.SFTPConfig.DisableCouncurrentReads)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.ID = 0
- user.CreatedAt = 0
- secret := kms.NewSecret(sdkkms.SecretStatusSecretBox, "invalid encrypted payload", "", "")
- user.FsConfig.SFTPConfig.Password = secret
- _, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.Error(t, err)
- user.FsConfig.SFTPConfig.Password = kms.NewEmptySecret()
- user.FsConfig.SFTPConfig.PrivateKey = secret
- _, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.Error(t, err)
- user.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret(sftpPrivateKey)
- user, _, err = httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err)
- initialPkeyPayload = user.FsConfig.SFTPConfig.PrivateKey.GetPayload()
- assert.Nil(t, user.FsConfig.SFTPConfig.Password)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.NotEmpty(t, initialPkeyPayload)
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetKey())
- user.FsConfig.Provider = sdk.SFTPFilesystemProvider
- user.FsConfig.SFTPConfig.PrivateKey.SetKey("k")
- user, bb, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(bb))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.NotEmpty(t, initialPkeyPayload)
- assert.Equal(t, initialPkeyPayload, user.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.Empty(t, user.FsConfig.SFTPConfig.PrivateKey.GetKey())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserHiddenFields(t *testing.T) {
- // sensitive data must be hidden but not deleted from the dataprovider
- usernames := []string{"user1", "user2", "user3", "user4", "user5", "user6"}
- u1 := getTestUser()
- u1.Username = usernames[0]
- u1.FsConfig.Provider = sdk.S3FilesystemProvider
- u1.FsConfig.S3Config.Bucket = "test"
- u1.FsConfig.S3Config.Region = "us-east-1"
- u1.FsConfig.S3Config.AccessKey = "S3-Access-Key"
- u1.FsConfig.S3Config.AccessSecret = kms.NewPlainSecret("S3-Access-Secret")
- user1, _, err := httpdtest.AddUser(u1, http.StatusCreated)
- assert.NoError(t, err)
- u2 := getTestUser()
- u2.Username = usernames[1]
- u2.FsConfig.Provider = sdk.GCSFilesystemProvider
- u2.FsConfig.GCSConfig.Bucket = "test"
- u2.FsConfig.GCSConfig.Credentials = kms.NewPlainSecret("fake credentials")
- u2.FsConfig.GCSConfig.ACL = "bucketOwnerRead"
- u2.FsConfig.GCSConfig.UploadPartSize = 5
- u2.FsConfig.GCSConfig.UploadPartMaxTime = 20
- user2, _, err := httpdtest.AddUser(u2, http.StatusCreated)
- assert.NoError(t, err)
- u3 := getTestUser()
- u3.Username = usernames[2]
- u3.FsConfig.Provider = sdk.AzureBlobFilesystemProvider
- u3.FsConfig.AzBlobConfig.Container = "test"
- u3.FsConfig.AzBlobConfig.AccountName = "Server-Account-Name"
- u3.FsConfig.AzBlobConfig.AccountKey = kms.NewPlainSecret("Server-Account-Key")
- user3, _, err := httpdtest.AddUser(u3, http.StatusCreated)
- assert.NoError(t, err)
- u4 := getTestUser()
- u4.Username = usernames[3]
- u4.FsConfig.Provider = sdk.CryptedFilesystemProvider
- u4.FsConfig.CryptConfig.Passphrase = kms.NewPlainSecret("test passphrase")
- user4, _, err := httpdtest.AddUser(u4, http.StatusCreated)
- assert.NoError(t, err)
- u5 := getTestUser()
- u5.Username = usernames[4]
- u5.FsConfig.Provider = sdk.SFTPFilesystemProvider
- u5.FsConfig.SFTPConfig.Endpoint = "127.0.0.1:2022"
- u5.FsConfig.SFTPConfig.Username = "sftp_user"
- u5.FsConfig.SFTPConfig.Password = kms.NewPlainSecret("apassword")
- u5.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret(sftpPrivateKey)
- u5.FsConfig.SFTPConfig.Fingerprints = []string{sftpPkeyFingerprint}
- u5.FsConfig.SFTPConfig.Prefix = "/prefix"
- user5, _, err := httpdtest.AddUser(u5, http.StatusCreated)
- assert.NoError(t, err)
- u6 := getTestUser()
- u6.Username = usernames[5]
- u6.FsConfig.Provider = sdk.HTTPFilesystemProvider
- u6.FsConfig.HTTPConfig = vfs.HTTPFsConfig{
- BaseHTTPFsConfig: sdk.BaseHTTPFsConfig{
- Endpoint: "http://127.0.0.1/api/v1",
- Username: defaultUsername,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- APIKey: kms.NewPlainSecret(defaultTokenAuthUser),
- }
- user6, _, err := httpdtest.AddUser(u6, http.StatusCreated)
- assert.NoError(t, err)
- users, _, err := httpdtest.GetUsers(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(users), 6)
- for _, username := range usernames {
- user, _, err := httpdtest.GetUserByUsername(username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user.Password)
- assert.True(t, user.HasPassword)
- }
- user1, _, err = httpdtest.GetUserByUsername(user1.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user1.Password)
- assert.Empty(t, user1.FsConfig.S3Config.AccessSecret.GetKey())
- assert.Empty(t, user1.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.NotEmpty(t, user1.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.NotEmpty(t, user1.FsConfig.S3Config.AccessSecret.GetPayload())
- user2, _, err = httpdtest.GetUserByUsername(user2.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user2.Password)
- assert.Empty(t, user2.FsConfig.GCSConfig.Credentials.GetKey())
- assert.Empty(t, user2.FsConfig.GCSConfig.Credentials.GetAdditionalData())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetStatus())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetPayload())
- user3, _, err = httpdtest.GetUserByUsername(user3.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user3.Password)
- assert.Empty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetKey())
- assert.Empty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- assert.NotEmpty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.NotEmpty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- user4, _, err = httpdtest.GetUserByUsername(user4.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user4.Password)
- assert.Empty(t, user4.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Empty(t, user4.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.NotEmpty(t, user4.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, user4.FsConfig.CryptConfig.Passphrase.GetPayload())
- user5, _, err = httpdtest.GetUserByUsername(user5.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user5.Password)
- assert.Empty(t, user5.FsConfig.SFTPConfig.Password.GetKey())
- assert.Empty(t, user5.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.Password.GetStatus())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.Password.GetPayload())
- assert.Empty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetKey())
- assert.Empty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- assert.Equal(t, "/prefix", user5.FsConfig.SFTPConfig.Prefix)
- user6, _, err = httpdtest.GetUserByUsername(user6.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user6.Password)
- assert.Empty(t, user6.FsConfig.HTTPConfig.Password.GetKey())
- assert.Empty(t, user6.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.NotEmpty(t, user6.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.NotEmpty(t, user6.FsConfig.HTTPConfig.APIKey.GetPayload())
- // finally check that we have all the data inside the data provider
- user1, err = dataprovider.UserExists(user1.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, user1.Password)
- assert.NotEmpty(t, user1.FsConfig.S3Config.AccessSecret.GetKey())
- assert.NotEmpty(t, user1.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.NotEmpty(t, user1.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.NotEmpty(t, user1.FsConfig.S3Config.AccessSecret.GetPayload())
- err = user1.FsConfig.S3Config.AccessSecret.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user1.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.Equal(t, u1.FsConfig.S3Config.AccessSecret.GetPayload(), user1.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Empty(t, user1.FsConfig.S3Config.AccessSecret.GetKey())
- assert.Empty(t, user1.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- user2, err = dataprovider.UserExists(user2.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, user2.Password)
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetKey())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetAdditionalData())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetStatus())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetPayload())
- err = user2.FsConfig.GCSConfig.Credentials.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user2.FsConfig.GCSConfig.Credentials.GetStatus())
- assert.Equal(t, u2.FsConfig.GCSConfig.Credentials.GetPayload(), user2.FsConfig.GCSConfig.Credentials.GetPayload())
- assert.Empty(t, user2.FsConfig.GCSConfig.Credentials.GetKey())
- assert.Empty(t, user2.FsConfig.GCSConfig.Credentials.GetAdditionalData())
- user3, err = dataprovider.UserExists(user3.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, user3.Password)
- assert.NotEmpty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetKey())
- assert.NotEmpty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- assert.NotEmpty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.NotEmpty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- err = user3.FsConfig.AzBlobConfig.AccountKey.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user3.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.Equal(t, u3.FsConfig.AzBlobConfig.AccountKey.GetPayload(), user3.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- assert.Empty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetKey())
- assert.Empty(t, user3.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- user4, err = dataprovider.UserExists(user4.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, user4.Password)
- assert.NotEmpty(t, user4.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.NotEmpty(t, user4.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.NotEmpty(t, user4.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, user4.FsConfig.CryptConfig.Passphrase.GetPayload())
- err = user4.FsConfig.CryptConfig.Passphrase.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user4.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.Equal(t, u4.FsConfig.CryptConfig.Passphrase.GetPayload(), user4.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, user4.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Empty(t, user4.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- user5, err = dataprovider.UserExists(user5.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, user5.Password)
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.Password.GetKey())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.Password.GetStatus())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.Password.GetPayload())
- err = user5.FsConfig.SFTPConfig.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user5.FsConfig.SFTPConfig.Password.GetStatus())
- assert.Equal(t, u5.FsConfig.SFTPConfig.Password.GetPayload(), user5.FsConfig.SFTPConfig.Password.GetPayload())
- assert.Empty(t, user5.FsConfig.SFTPConfig.Password.GetKey())
- assert.Empty(t, user5.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetKey())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.NotEmpty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- err = user5.FsConfig.SFTPConfig.PrivateKey.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user5.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.Equal(t, u5.FsConfig.SFTPConfig.PrivateKey.GetPayload(), user5.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- assert.Empty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetKey())
- assert.Empty(t, user5.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- user6, err = dataprovider.UserExists(user6.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, user6.Password)
- assert.NotEmpty(t, user6.FsConfig.HTTPConfig.Password.GetKey())
- assert.NotEmpty(t, user6.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.NotEmpty(t, user6.FsConfig.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, user6.FsConfig.HTTPConfig.Password.GetPayload())
- err = user6.FsConfig.HTTPConfig.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusPlain, user6.FsConfig.HTTPConfig.Password.GetStatus())
- assert.Equal(t, u6.FsConfig.HTTPConfig.Password.GetPayload(), user6.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, user6.FsConfig.HTTPConfig.Password.GetKey())
- assert.Empty(t, user6.FsConfig.HTTPConfig.Password.GetAdditionalData())
- // update the GCS user and check that the credentials are preserved
- user2.FsConfig.GCSConfig.Credentials = kms.NewEmptySecret()
- user2.FsConfig.GCSConfig.ACL = "private"
- _, _, err = httpdtest.UpdateUser(user2, http.StatusOK, "")
- assert.NoError(t, err)
- user2, _, err = httpdtest.GetUserByUsername(user2.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, user2.Password)
- assert.Empty(t, user2.FsConfig.GCSConfig.Credentials.GetKey())
- assert.Empty(t, user2.FsConfig.GCSConfig.Credentials.GetAdditionalData())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetStatus())
- assert.NotEmpty(t, user2.FsConfig.GCSConfig.Credentials.GetPayload())
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user3, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user4, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user5, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user6, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestSecretObject(t *testing.T) {
- s := kms.NewPlainSecret("test data")
- s.SetAdditionalData("username")
- require.True(t, s.IsValid())
- err := s.Encrypt()
- require.NoError(t, err)
- require.Equal(t, sdkkms.SecretStatusSecretBox, s.GetStatus())
- require.NotEmpty(t, s.GetPayload())
- require.NotEmpty(t, s.GetKey())
- require.True(t, s.IsValid())
- err = s.Decrypt()
- require.NoError(t, err)
- require.Equal(t, sdkkms.SecretStatusPlain, s.GetStatus())
- require.Equal(t, "test data", s.GetPayload())
- require.Empty(t, s.GetKey())
- }
- func TestSecretObjectCompatibility(t *testing.T) {
- // this is manually tested against vault too
- testPayload := "test payload"
- s := kms.NewPlainSecret(testPayload)
- require.True(t, s.IsValid())
- err := s.Encrypt()
- require.NoError(t, err)
- localAsJSON, err := json.Marshal(s)
- assert.NoError(t, err)
- for _, secretStatus := range []string{sdkkms.SecretStatusSecretBox} {
- kmsConfig := config.GetKMSConfig()
- assert.Empty(t, kmsConfig.Secrets.MasterKeyPath)
- if secretStatus == sdkkms.SecretStatusVaultTransit {
- os.Setenv("VAULT_SERVER_URL", "http://127.0.0.1:8200")
- os.Setenv("VAULT_SERVER_TOKEN", "s.9lYGq83MbgG5KR5kfebXVyhJ")
- kmsConfig.Secrets.URL = "hashivault://mykey"
- }
- err := kmsConfig.Initialize()
- assert.NoError(t, err)
- // encrypt without a master key
- secret := kms.NewPlainSecret(testPayload)
- secret.SetAdditionalData("add data")
- err = secret.Encrypt()
- assert.NoError(t, err)
- assert.Equal(t, 0, secret.GetMode())
- secretClone := secret.Clone()
- err = secretClone.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, testPayload, secretClone.GetPayload())
- if secretStatus == sdkkms.SecretStatusVaultTransit {
- // decrypt the local secret now that the provider is vault
- secretLocal := kms.NewEmptySecret()
- err = json.Unmarshal(localAsJSON, secretLocal)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, secretLocal.GetStatus())
- assert.Equal(t, 0, secretLocal.GetMode())
- err = secretLocal.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, testPayload, secretLocal.GetPayload())
- assert.Equal(t, sdkkms.SecretStatusPlain, secretLocal.GetStatus())
- err = secretLocal.Encrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, secretLocal.GetStatus())
- assert.Equal(t, 0, secretLocal.GetMode())
- }
- asJSON, err := json.Marshal(secret)
- assert.NoError(t, err)
- masterKeyPath := filepath.Join(os.TempDir(), "mkey")
- err = os.WriteFile(masterKeyPath, []byte("test key"), os.ModePerm)
- assert.NoError(t, err)
- config := kms.Configuration{
- Secrets: kms.Secrets{
- MasterKeyPath: masterKeyPath,
- },
- }
- if secretStatus == sdkkms.SecretStatusVaultTransit {
- config.Secrets.URL = "hashivault://mykey"
- }
- err = config.Initialize()
- assert.NoError(t, err)
- // now build the secret from JSON
- secret = kms.NewEmptySecret()
- err = json.Unmarshal(asJSON, secret)
- assert.NoError(t, err)
- assert.Equal(t, 0, secret.GetMode())
- err = secret.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, testPayload, secret.GetPayload())
- err = secret.Encrypt()
- assert.NoError(t, err)
- assert.Equal(t, 1, secret.GetMode())
- err = secret.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, testPayload, secret.GetPayload())
- if secretStatus == sdkkms.SecretStatusVaultTransit {
- // decrypt the local secret encryped without a master key now that
- // the provider is vault and a master key is set.
- // The provider will not change, the master key will be used
- secretLocal := kms.NewEmptySecret()
- err = json.Unmarshal(localAsJSON, secretLocal)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, secretLocal.GetStatus())
- assert.Equal(t, 0, secretLocal.GetMode())
- err = secretLocal.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, testPayload, secretLocal.GetPayload())
- assert.Equal(t, sdkkms.SecretStatusPlain, secretLocal.GetStatus())
- err = secretLocal.Encrypt()
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, secretLocal.GetStatus())
- assert.Equal(t, 1, secretLocal.GetMode())
- }
- err = kmsConfig.Initialize()
- assert.NoError(t, err)
- err = os.Remove(masterKeyPath)
- assert.NoError(t, err)
- if secretStatus == sdkkms.SecretStatusVaultTransit {
- os.Unsetenv("VAULT_SERVER_URL")
- os.Unsetenv("VAULT_SERVER_TOKEN")
- }
- }
- }
- func TestUpdateUserNoCredentials(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.Password = ""
- user.PublicKeys = []string{}
- // password and public key will be omitted from json serialization if empty and so they will remain unchanged
- // and no validation error will be raised
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUserEmptyHomeDir(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.HomeDir = ""
- _, _, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateUserInvalidHomeDir(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- user.HomeDir = "relative_path"
- _, _, err = httpdtest.UpdateUser(user, http.StatusBadRequest, "")
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateNonExistentUser(t *testing.T) {
- _, _, err := httpdtest.UpdateUser(getTestUser(), http.StatusNotFound, "")
- assert.NoError(t, err)
- }
- func TestGetNonExistentUser(t *testing.T) {
- _, _, err := httpdtest.GetUserByUsername("na", http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestDeleteNonExistentUser(t *testing.T) {
- _, err := httpdtest.RemoveUser(getTestUser(), http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestAddDuplicateUser(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- _, _, err = httpdtest.AddUser(getTestUser(), http.StatusConflict)
- assert.NoError(t, err)
- _, _, err = httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.Error(t, err, "adding a duplicate user must fail")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestGetUsers(t *testing.T) {
- user1, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- u := getTestUser()
- u.Username = defaultUsername + "1"
- user2, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- users, _, err := httpdtest.GetUsers(0, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(users), 2)
- for _, user := range users {
- if u.Username == user.Username {
- assert.True(t, user.HasPassword)
- }
- }
- users, _, err = httpdtest.GetUsers(1, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 1, len(users))
- users, _, err = httpdtest.GetUsers(1, 1, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 1, len(users))
- _, _, err = httpdtest.GetUsers(1, 1, http.StatusInternalServerError)
- assert.Error(t, err)
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user2, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestGetQuotaScans(t *testing.T) {
- _, _, err := httpdtest.GetQuotaScans(http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetQuotaScans(http.StatusInternalServerError)
- assert.Error(t, err)
- _, _, err = httpdtest.GetFoldersQuotaScans(http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetFoldersQuotaScans(http.StatusInternalServerError)
- assert.Error(t, err)
- }
- func TestStartQuotaScan(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- _, err = httpdtest.StartQuotaScan(user, http.StatusAccepted)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- folder := vfs.BaseVirtualFolder{
- Name: "vfolder",
- MappedPath: filepath.Join(os.TempDir(), "folder"),
- Description: "virtual folder",
- }
- _, _, err = httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err)
- _, err = httpdtest.StartFolderQuotaScan(folder, http.StatusAccepted)
- assert.NoError(t, err)
- for {
- quotaScan, _, err := httpdtest.GetFoldersQuotaScans(http.StatusOK)
- if !assert.NoError(t, err, "Error getting active scans") {
- break
- }
- if len(quotaScan) == 0 {
- break
- }
- time.Sleep(100 * time.Millisecond)
- }
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateFolderQuotaUsage(t *testing.T) {
- f := vfs.BaseVirtualFolder{
- Name: "vdir",
- MappedPath: filepath.Join(os.TempDir(), "folder"),
- }
- usedQuotaFiles := 1
- usedQuotaSize := int64(65535)
- f.UsedQuotaFiles = usedQuotaFiles
- f.UsedQuotaSize = usedQuotaSize
- folder, _, err := httpdtest.AddFolder(f, http.StatusCreated)
- if assert.NoError(t, err) {
- assert.Equal(t, usedQuotaFiles, folder.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize, folder.UsedQuotaSize)
- }
- _, err = httpdtest.UpdateFolderQuotaUsage(folder, "invalid mode", http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.UpdateFolderQuotaUsage(f, "reset", http.StatusOK)
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(f.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, folder.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize, folder.UsedQuotaSize)
- _, err = httpdtest.UpdateFolderQuotaUsage(f, "add", http.StatusOK)
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(f.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 2*usedQuotaFiles, folder.UsedQuotaFiles)
- assert.Equal(t, 2*usedQuotaSize, folder.UsedQuotaSize)
- f.UsedQuotaSize = -1
- _, err = httpdtest.UpdateFolderQuotaUsage(f, "", http.StatusBadRequest)
- assert.NoError(t, err)
- f.UsedQuotaSize = usedQuotaSize
- f.Name = f.Name + "1"
- _, err = httpdtest.UpdateFolderQuotaUsage(f, "", http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestGetVersion(t *testing.T) {
- _, _, err := httpdtest.GetVersion(http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetVersion(http.StatusInternalServerError)
- assert.Error(t, err, "get version request must succeed, we requested to check a wrong status code")
- }
- func TestGetStatus(t *testing.T) {
- _, _, err := httpdtest.GetStatus(http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetStatus(http.StatusBadRequest)
- assert.Error(t, err, "get provider status request must succeed, we requested to check a wrong status code")
- }
- func TestGetConnections(t *testing.T) {
- _, _, err := httpdtest.GetConnections(http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetConnections(http.StatusInternalServerError)
- assert.Error(t, err, "get sftp connections request must succeed, we requested to check a wrong status code")
- }
- func TestCloseActiveConnection(t *testing.T) {
- _, err := httpdtest.CloseConnection("non_existent_id", http.StatusNotFound)
- assert.NoError(t, err)
- user := getTestUser()
- c := common.NewBaseConnection("connID", common.ProtocolSFTP, "", "", user)
- fakeConn := &fakeConnection{
- BaseConnection: c,
- }
- err = common.Connections.Add(fakeConn)
- assert.NoError(t, err)
- _, err = httpdtest.CloseConnection(c.GetID(), http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 0)
- }
- func TestCloseConnectionAfterUserUpdateDelete(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- c := common.NewBaseConnection("connID", common.ProtocolFTP, "", "", user)
- fakeConn := &fakeConnection{
- BaseConnection: c,
- }
- err = common.Connections.Add(fakeConn)
- assert.NoError(t, err)
- c1 := common.NewBaseConnection("connID1", common.ProtocolSFTP, "", "", user)
- fakeConn1 := &fakeConnection{
- BaseConnection: c1,
- }
- err = common.Connections.Add(fakeConn1)
- assert.NoError(t, err)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "0")
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 2)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "1")
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 0)
- err = common.Connections.Add(fakeConn)
- assert.NoError(t, err)
- err = common.Connections.Add(fakeConn1)
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 2)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 0)
- }
- func TestAdminGenerateRecoveryCodesSaveError(t *testing.T) {
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.NamingRules = 7
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- a := getTestAdmin()
- a.Username = "adMiN@example.com "
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], admin.Username)
- assert.NoError(t, err)
- admin.Filters.TOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- }
- admin.Password = defaultTokenAuthPass
- err = dataprovider.UpdateAdmin(&admin, "", "", "")
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(a.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- adminAPIToken, err := getJWTAPITokenFromTestServerWithPasscode(a.Username, defaultTokenAuthPass, passcode)
- assert.NoError(t, err)
- assert.NotEmpty(t, adminAPIToken)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- if config.GetProviderConf().Driver == dataprovider.MemoryDataProviderName {
- return
- }
- req, err := http.NewRequest(http.MethodPost, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, adminAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following characters are allowed")
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestNamingRules(t *testing.T) {
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err := smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.NamingRules = 7
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- u := getTestUser()
- u.Username = " uSeR@user.me "
- u.Email = dataprovider.ConvertName(u.Username)
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, "user@user.me", user.Username)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- user.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolSSH},
- }
- user.Password = u.Password
- err = dataprovider.UpdateUser(&user, "", "", "")
- assert.NoError(t, err)
- user.Username = u.Username
- user.AdditionalInfo = "info"
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(u.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.TOTPConfig.Enabled)
- r := getTestRole()
- r.Name = "role@mycompany"
- role, _, err := httpdtest.AddRole(r, http.StatusCreated)
- assert.NoError(t, err)
- a := getTestAdmin()
- a.Username = "admiN@example.com "
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- assert.Equal(t, "admin@example.com", admin.Username)
- admin.Email = dataprovider.ConvertName(a.Username)
- admin.Username = a.Username
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(a.Username, http.StatusOK)
- assert.NoError(t, err)
- f := vfs.BaseVirtualFolder{
- Name: "文件夹AB",
- MappedPath: filepath.Clean(os.TempDir()),
- }
- folder, resp, err := httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, "文件夹ab", folder.Name)
- folder.Name = f.Name
- folder.Description = folder.Name
- _, resp, err = httpdtest.UpdateFolder(folder, http.StatusOK)
- assert.NoError(t, err, string(resp))
- folder, resp, err = httpdtest.GetFolderByName(f.Name, http.StatusOK)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, "文件夹AB", folder.Description)
- _, err = httpdtest.RemoveFolder(f, http.StatusOK)
- assert.NoError(t, err)
- token, err := getJWTWebClientTokenFromTestServer(u.Username, defaultPassword)
- assert.NoError(t, err)
- assert.NotEmpty(t, token)
- adminAPIToken, err := getJWTAPITokenFromTestServer(a.Username, defaultTokenAuthPass)
- assert.NoError(t, err)
- assert.NotEmpty(t, adminAPIToken)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- if config.GetProviderConf().Driver == dataprovider.MemoryDataProviderName {
- return
- }
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- token, err = getJWTWebClientTokenFromTestServer(user.Username, defaultPassword)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- req, err := http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidUser)
- // test user reset password. Setting the new password will fail because the username is not valid
- form = make(url.Values)
- form.Set("username", user.Username)
- form.Set(csrfFormToken, csrfToken)
- lastResetCode = ""
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("code", lastResetCode)
- form.Set("password", defaultPassword)
- form.Set("confirm_password", defaultPassword)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidUser)
- adminAPIToken, err = getJWTAPITokenFromTestServer(admin.Username, defaultTokenAuthPass)
- assert.NoError(t, err)
- userAPIToken, err := getJWTAPIUserTokenFromTestServer(user.Username, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userPath+"/"+user.Username+"/2fa/disable", nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, adminAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following characters are allowed")
- req, err = http.NewRequest(http.MethodPost, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following characters are allowed")
- apiKeyAuthReq := make(map[string]bool)
- apiKeyAuthReq["allow_api_key_auth"] = true
- asJSON, err := json.Marshal(apiKeyAuthReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following characters are allowed")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- token, err = getJWTWebTokenFromTestServer(admin.Username, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err = getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webAdminProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidUser)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminRolePath, role.Name), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidName)
- apiKeyAuthReq = make(map[string]bool)
- apiKeyAuthReq["allow_api_key_auth"] = true
- asJSON, err = json.Marshal(apiKeyAuthReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, adminProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, adminAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the following characters are allowed")
- // test admin reset password
- form = make(url.Values)
- form.Set("username", admin.Username)
- form.Set(csrfFormToken, csrfToken)
- lastResetCode = ""
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("code", lastResetCode)
- form.Set("password", defaultPassword)
- form.Set("confirm_password", defaultPassword)
- req, err = http.NewRequest(http.MethodPost, webAdminResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdGeneric)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- }
- func TestUserPassword(t *testing.T) {
- u := getTestUser()
- u.Password = ""
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.False(t, user.HasPassword)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.False(t, user.HasPassword)
- user.Password = defaultPassword
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.True(t, user.HasPassword)
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- rawUser := map[string]any{
- "username": user.Username,
- "home_dir": filepath.Join(homeBasePath, defaultUsername),
- "permissions": map[string][]string{
- "/": {"*"},
- },
- }
- userAsJSON, err := json.Marshal(rawUser)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // the previous password must be preserved
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.HasPassword)
- // update the user with an empty password field, the password will be unset
- rawUser["password"] = ""
- userAsJSON, err = json.Marshal(rawUser)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, user.HasPassword)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestSaveErrors(t *testing.T) {
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.NamingRules = 1
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- recCode := "recovery code"
- recoveryCodes := []dataprovider.RecoveryCode{
- {
- Secret: kms.NewPlainSecret(recCode),
- Used: false,
- },
- }
- u := getTestUser()
- u.Username = "user@example.com"
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- user.Password = u.Password
- user.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolSSH, common.ProtocolHTTP},
- }
- user.Filters.RecoveryCodes = recoveryCodes
- err = dataprovider.UpdateUser(&user, "", "", "")
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.TOTPConfig.Enabled)
- assert.Len(t, user.Filters.RecoveryCodes, 1)
- a := getTestAdmin()
- a.Username = "admin@example.com"
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- admin.Email = admin.Username
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- admin.Password = a.Password
- admin.Filters.TOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- }
- admin.Filters.RecoveryCodes = recoveryCodes
- err = dataprovider.UpdateAdmin(&admin, "", "", "")
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 1)
- r := getTestRole()
- r.Name = "role@mycompany"
- role, _, err := httpdtest.AddRole(r, http.StatusCreated)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- if config.GetProviderConf().Driver == dataprovider.MemoryDataProviderName {
- return
- }
- _, resp, err := httpdtest.UpdateRole(role, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "the following characters are allowed")
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := getLoginForm(a.Username, a.Password, csrfToken)
- req, err := http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr := executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- cookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("recovery_code", recCode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusInternalServerError, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- csrfToken, err = getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- form = getLoginForm(u.Username, u.Password, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("recovery_code", recCode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusInternalServerError, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserBaseDir(t *testing.T) {
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.UsersBaseDir = homeBasePath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- u := getTestUser()
- u.HomeDir = ""
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- if assert.Error(t, err) {
- assert.EqualError(t, err, "home dir mismatch")
- }
- assert.Equal(t, filepath.Join(providerConf.UsersBaseDir, u.Username), user.HomeDir)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestQuotaTrackingDisabled(t *testing.T) {
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.TrackQuota = 0
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- // user quota scan must fail
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- _, err = httpdtest.StartQuotaScan(user, http.StatusForbidden)
- assert.NoError(t, err)
- _, err = httpdtest.UpdateQuotaUsage(user, "", http.StatusForbidden)
- assert.NoError(t, err)
- _, err = httpdtest.UpdateTransferQuotaUsage(user, "", http.StatusForbidden)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- // folder quota scan must fail
- folder := vfs.BaseVirtualFolder{
- Name: "folder_quota_test",
- MappedPath: filepath.Clean(os.TempDir()),
- }
- folder, resp, err := httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- _, err = httpdtest.StartFolderQuotaScan(folder, http.StatusForbidden)
- assert.NoError(t, err)
- _, err = httpdtest.UpdateFolderQuotaUsage(folder, "", http.StatusForbidden)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestProviderErrors(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- userAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userWebToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- token, _, err := httpdtest.GetToken(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- testServerToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- httpdtest.SetJWTToken(token)
- err = dataprovider.Close()
- assert.NoError(t, err)
- _, _, err = httpdtest.GetUserByUsername("na", http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetUsers(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetGroups(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetAdmins(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetAPIKeys(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetEventActions(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetEventRules(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetIPListEntries(dataprovider.IPListTypeDefender, "", "", dataprovider.OrderASC, 10, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetRoles(1, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateRole(getTestRole(), http.StatusInternalServerError)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- // password reset errors
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("username", "username")
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorGetUser)
- getJSONShares := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err := http.NewRequest(http.MethodGet, webClientSharesPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, userWebToken)
- executeRequest(req)
- }
- getJSONShares()
- req, err = http.NewRequest(http.MethodGet, webClientSharePath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, userWebToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath+"/shareID", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, userWebToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/shareID", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, userWebToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- _, _, err = httpdtest.UpdateUser(dataprovider.User{BaseUser: sdk.BaseUser{Username: "auser"}}, http.StatusInternalServerError, "")
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(dataprovider.User{BaseUser: sdk.BaseUser{Username: "auser"}}, http.StatusInternalServerError)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: "aname"}, http.StatusInternalServerError)
- assert.NoError(t, err)
- status, _, err := httpdtest.GetStatus(http.StatusOK)
- if assert.NoError(t, err) {
- assert.False(t, status.DataProvider.IsActive)
- }
- _, _, err = httpdtest.Dumpdata("backup.json", "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetFolders(0, 0, http.StatusInternalServerError)
- assert.NoError(t, err)
- user = getTestUser()
- user.ID = 1
- backupData := dataprovider.BackupData{
- Version: dataprovider.DumpVersion,
- }
- backupData.Configs = &dataprovider.Configs{}
- backupData.Users = append(backupData.Users, user)
- backupContent, err := json.Marshal(backupData)
- assert.NoError(t, err)
- backupFilePath := filepath.Join(backupsPath, "backup.json")
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData.Configs = nil
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData.Folders = append(backupData.Folders, vfs.BaseVirtualFolder{Name: "testFolder", MappedPath: filepath.Clean(os.TempDir())})
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData.Users = nil
- backupData.Folders = nil
- backupData.Groups = append(backupData.Groups, getTestGroup())
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData.Groups = nil
- backupData.Admins = append(backupData.Admins, getTestAdmin())
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData.Users = nil
- backupData.Folders = nil
- backupData.Admins = nil
- backupData.APIKeys = append(backupData.APIKeys, dataprovider.APIKey{
- Name: "name",
- KeyID: util.GenerateUniqueID(),
- Key: fmt.Sprintf("%v.%v", util.GenerateUniqueID(), util.GenerateUniqueID()),
- Scope: dataprovider.APIKeyScopeUser,
- })
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData.APIKeys = nil
- backupData.Shares = append(backupData.Shares, dataprovider.Share{
- Name: util.GenerateUniqueID(),
- ShareID: util.GenerateUniqueID(),
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- Username: defaultUsername,
- })
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, resp, err := httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err, string(resp))
- backupData = dataprovider.BackupData{
- EventActions: []dataprovider.BaseEventAction{
- {
- Name: "quota reset",
- Type: dataprovider.ActionTypeFolderQuotaReset,
- },
- },
- Version: dataprovider.DumpVersion,
- }
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData = dataprovider.BackupData{
- EventRules: []dataprovider.EventRule{
- {
- Name: "quota reset",
- Trigger: dataprovider.EventTriggerSchedule,
- Conditions: dataprovider.EventConditions{
- Schedules: []dataprovider.Schedule{
- {
- Hours: "2",
- DayOfWeek: "1",
- DayOfMonth: "2",
- Month: "3",
- },
- },
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: "unknown action",
- },
- Order: 1,
- },
- },
- },
- },
- Version: dataprovider.DumpVersion,
- }
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData = dataprovider.BackupData{
- Roles: []dataprovider.Role{
- {
- Name: "role1",
- },
- },
- Version: dataprovider.DumpVersion,
- }
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- backupData = dataprovider.BackupData{
- IPLists: []dataprovider.IPListEntry{
- {
- IPOrNet: "192.168.1.1/24",
- Type: dataprovider.IPListTypeRateLimiterSafeList,
- Mode: dataprovider.ListModeAllow,
- },
- },
- Version: dataprovider.DumpVersion,
- }
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "", http.StatusInternalServerError)
- assert.NoError(t, err)
- err = os.Remove(backupFilePath)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webUserPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webTemplateUser+"?from=auser", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webGroupPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webTemplateUser, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webGroupPath, "groupname"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(webGroupPath, "grpname"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webTemplateFolder+"?from=afolder", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminEventActionPath, "actionname"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, "actionname"), bytes.NewBuffer(nil))
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- getJSONActions := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err := http.NewRequest(http.MethodGet, webAdminEventActionsPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- executeRequest(req)
- }
- getJSONActions()
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminEventRulePath, "rulename"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, "rulename"), bytes.NewBuffer(nil))
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- getJSONRules := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err := http.NewRequest(http.MethodGet, webAdminEventRulesPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- executeRequest(req)
- }
- getJSONRules()
- req, err = http.NewRequest(http.MethodGet, webAdminEventRulePath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- httpdtest.SetJWTToken("")
- }
- func TestFolders(t *testing.T) {
- folder := vfs.BaseVirtualFolder{
- Name: "name",
- MappedPath: "relative path",
- Users: []string{"1", "2", "3"},
- FsConfig: vfs.Filesystem{
- Provider: sdk.CryptedFilesystemProvider,
- CryptConfig: vfs.CryptFsConfig{
- Passphrase: kms.NewPlainSecret("asecret"),
- },
- },
- }
- _, _, err := httpdtest.AddFolder(folder, http.StatusBadRequest)
- assert.NoError(t, err)
- folder.MappedPath = filepath.Clean(os.TempDir())
- folder1, resp, err := httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, folder.Name, folder1.Name)
- assert.Equal(t, folder.MappedPath, folder1.MappedPath)
- assert.Equal(t, 0, folder1.UsedQuotaFiles)
- assert.Equal(t, int64(0), folder1.UsedQuotaSize)
- assert.Equal(t, int64(0), folder1.LastQuotaUpdate)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, folder1.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, folder1.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, folder1.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, folder1.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Len(t, folder1.Users, 0)
- // adding a duplicate folder must fail
- _, _, err = httpdtest.AddFolder(folder, http.StatusCreated)
- assert.Error(t, err)
- folder.MappedPath = filepath.Join(os.TempDir(), "vfolder")
- folder.Name = filepath.Base(folder.MappedPath)
- folder.UsedQuotaFiles = 1
- folder.UsedQuotaSize = 345
- folder.LastQuotaUpdate = 10
- folder2, _, err := httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, 1, folder2.UsedQuotaFiles)
- assert.Equal(t, int64(345), folder2.UsedQuotaSize)
- assert.Equal(t, int64(10), folder2.LastQuotaUpdate)
- assert.Len(t, folder2.Users, 0)
- folders, _, err := httpdtest.GetFolders(0, 0, http.StatusOK)
- assert.NoError(t, err)
- numResults := len(folders)
- assert.GreaterOrEqual(t, numResults, 2)
- found := false
- for _, f := range folders {
- if f.Name == folder1.Name {
- found = true
- assert.Equal(t, folder1.MappedPath, f.MappedPath)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, f.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, f.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, f.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, f.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Len(t, f.Users, 0)
- }
- }
- assert.True(t, found)
- folders, _, err = httpdtest.GetFolders(0, 1, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folders, numResults-1)
- folders, _, err = httpdtest.GetFolders(1, 0, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folders, 1)
- f, _, err := httpdtest.GetFolderByName(folder1.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, folder1.Name, f.Name)
- assert.Equal(t, folder1.MappedPath, f.MappedPath)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, f.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, f.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, f.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Empty(t, f.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Len(t, f.Users, 0)
- f, _, err = httpdtest.GetFolderByName(folder2.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, folder2.Name, f.Name)
- assert.Equal(t, folder2.MappedPath, f.MappedPath)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{
- Name: "invalid",
- }, http.StatusNotFound)
- assert.NoError(t, err)
- _, _, err = httpdtest.UpdateFolder(vfs.BaseVirtualFolder{Name: "notfound"}, http.StatusNotFound)
- assert.NoError(t, err)
- folder1.MappedPath = "a/relative/path"
- _, _, err = httpdtest.UpdateFolder(folder1, http.StatusBadRequest)
- assert.NoError(t, err)
- folder1.MappedPath = filepath.Join(os.TempDir(), "updated")
- folder1.Description = "updated folder description"
- f, resp, err = httpdtest.UpdateFolder(folder1, http.StatusOK)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, folder1.MappedPath, f.MappedPath)
- assert.Equal(t, folder1.Description, f.Description)
- _, err = httpdtest.RemoveFolder(folder1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(folder2, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestFolderRelations(t *testing.T) {
- mappedPath := filepath.Join(os.TempDir(), "mapped_path")
- name := filepath.Base(mappedPath)
- u := getTestUser()
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: name,
- },
- VirtualPath: "/mountu",
- })
- _, resp, err := httpdtest.AddUser(u, http.StatusInternalServerError)
- assert.NoError(t, err, string(resp))
- g := getTestGroup()
- g.VirtualFolders = append(g.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: name,
- },
- VirtualPath: "/mountg",
- })
- _, resp, err = httpdtest.AddGroup(g, http.StatusInternalServerError)
- assert.NoError(t, err, string(resp))
- f := vfs.BaseVirtualFolder{
- Name: name,
- MappedPath: mappedPath,
- }
- folder, _, err := httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 0)
- assert.Len(t, folder.Groups, 0)
- user, resp, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- group, resp, err := httpdtest.AddGroup(g, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- folder, _, err = httpdtest.GetFolderByName(folder.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Len(t, folder.Groups, 1)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, user.VirtualFolders, 1) {
- assert.Equal(t, mappedPath, user.VirtualFolders[0].MappedPath)
- }
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, group.VirtualFolders, 1) {
- assert.Equal(t, mappedPath, group.VirtualFolders[0].MappedPath)
- }
- // update the folder and check the modified field on user and group
- mappedPath = filepath.Join(os.TempDir(), "mapped_path")
- folder.MappedPath = mappedPath
- _, _, err = httpdtest.UpdateFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, user.VirtualFolders, 1) {
- assert.Equal(t, mappedPath, user.VirtualFolders[0].MappedPath)
- }
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, group.VirtualFolders, 1) {
- assert.Equal(t, mappedPath, group.VirtualFolders[0].MappedPath)
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- folder, _, err = httpdtest.GetFolderByName(folder.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 0)
- assert.Len(t, folder.Groups, 0)
- user, resp, err = httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, user.VirtualFolders, 1)
- group, resp, err = httpdtest.AddGroup(g, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Len(t, group.VirtualFolders, 1)
- folder, _, err = httpdtest.GetFolderByName(folder.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, folder.Users, 1)
- assert.Len(t, folder.Groups, 1)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, user.VirtualFolders, 0)
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group.VirtualFolders, 0)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestDumpdata(t *testing.T) {
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- _, rawResp, err := httpdtest.Dumpdata("", "", "", http.StatusBadRequest)
- assert.NoError(t, err, string(rawResp))
- _, _, err = httpdtest.Dumpdata(filepath.Join(backupsPath, "backup.json"), "", "", http.StatusBadRequest)
- assert.NoError(t, err)
- _, rawResp, err = httpdtest.Dumpdata("../backup.json", "", "", http.StatusBadRequest)
- assert.NoError(t, err, string(rawResp))
- _, rawResp, err = httpdtest.Dumpdata("backup.json", "", "0", http.StatusOK)
- assert.NoError(t, err, string(rawResp))
- response, _, err := httpdtest.Dumpdata("", "1", "0", http.StatusOK)
- assert.NoError(t, err)
- _, ok := response["admins"]
- assert.True(t, ok)
- _, ok = response["users"]
- assert.True(t, ok)
- _, ok = response["groups"]
- assert.True(t, ok)
- _, ok = response["folders"]
- assert.True(t, ok)
- _, ok = response["api_keys"]
- assert.True(t, ok)
- _, ok = response["shares"]
- assert.True(t, ok)
- _, ok = response["version"]
- assert.True(t, ok)
- _, rawResp, err = httpdtest.Dumpdata("backup.json", "", "1", http.StatusOK)
- assert.NoError(t, err, string(rawResp))
- err = os.Remove(filepath.Join(backupsPath, "backup.json"))
- assert.NoError(t, err)
- if runtime.GOOS != osWindows {
- err = os.Chmod(backupsPath, 0001)
- assert.NoError(t, err)
- _, _, err = httpdtest.Dumpdata("bck.json", "", "", http.StatusForbidden)
- assert.NoError(t, err)
- // subdir cannot be created
- _, _, err = httpdtest.Dumpdata(filepath.Join("subdir", "bck.json"), "", "", http.StatusForbidden)
- assert.NoError(t, err)
- err = os.Chmod(backupsPath, 0755)
- assert.NoError(t, err)
- }
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestDefenderAPI(t *testing.T) {
- oldConfig := config.GetCommonConfig()
- drivers := []string{common.DefenderDriverMemory}
- if isDbDefenderSupported() {
- drivers = append(drivers, common.DefenderDriverProvider)
- }
- for _, driver := range drivers {
- cfg := config.GetCommonConfig()
- cfg.DefenderConfig.Enabled = true
- cfg.DefenderConfig.Driver = driver
- cfg.DefenderConfig.Threshold = 3
- cfg.DefenderConfig.ScoreLimitExceeded = 2
- cfg.DefenderConfig.ScoreNoAuth = 0
- err := common.Initialize(cfg, 0)
- assert.NoError(t, err)
- ip := "::1"
- hosts, _, err := httpdtest.GetDefenderHosts(http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, hosts, 0)
- _, err = httpdtest.RemoveDefenderHostByIP(ip, http.StatusNotFound)
- assert.NoError(t, err)
- common.AddDefenderEvent(ip, common.ProtocolHTTP, common.HostEventNoLoginTried)
- hosts, _, err = httpdtest.GetDefenderHosts(http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, hosts, 0)
- common.AddDefenderEvent(ip, common.ProtocolHTTP, common.HostEventUserNotFound)
- hosts, _, err = httpdtest.GetDefenderHosts(http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, hosts, 1) {
- host := hosts[0]
- assert.Empty(t, host.GetBanTime())
- assert.Equal(t, 2, host.Score)
- assert.Equal(t, ip, host.IP)
- }
- host, _, err := httpdtest.GetDefenderHostByIP(ip, http.StatusOK)
- assert.NoError(t, err)
- assert.Empty(t, host.GetBanTime())
- assert.Equal(t, 2, host.Score)
- common.AddDefenderEvent(ip, common.ProtocolHTTP, common.HostEventUserNotFound)
- hosts, _, err = httpdtest.GetDefenderHosts(http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, hosts, 1) {
- host := hosts[0]
- assert.NotEmpty(t, host.GetBanTime())
- assert.Equal(t, 0, host.Score)
- assert.Equal(t, ip, host.IP)
- }
- host, _, err = httpdtest.GetDefenderHostByIP(ip, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEmpty(t, host.GetBanTime())
- assert.Equal(t, 0, host.Score)
- _, err = httpdtest.RemoveDefenderHostByIP(ip, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetDefenderHostByIP(ip, http.StatusNotFound)
- assert.NoError(t, err)
- common.AddDefenderEvent(ip, common.ProtocolHTTP, common.HostEventUserNotFound)
- common.AddDefenderEvent(ip, common.ProtocolHTTP, common.HostEventUserNotFound)
- hosts, _, err = httpdtest.GetDefenderHosts(http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, hosts, 1)
- _, err = httpdtest.RemoveDefenderHostByIP(ip, http.StatusOK)
- assert.NoError(t, err)
- host, _, err = httpdtest.GetDefenderHostByIP(ip, http.StatusNotFound)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveDefenderHostByIP(ip, http.StatusNotFound)
- assert.NoError(t, err)
- host, _, err = httpdtest.GetDefenderHostByIP("invalid_ip", http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveDefenderHostByIP("invalid_ip", http.StatusBadRequest)
- assert.NoError(t, err)
- if driver == common.DefenderDriverProvider {
- err = dataprovider.CleanupDefender(util.GetTimeAsMsSinceEpoch(time.Now().Add(1 * time.Hour)))
- assert.NoError(t, err)
- }
- }
- err := common.Initialize(oldConfig, 0)
- require.NoError(t, err)
- }
- func TestDefenderAPIErrors(t *testing.T) {
- if isDbDefenderSupported() {
- oldConfig := config.GetCommonConfig()
- cfg := config.GetCommonConfig()
- cfg.DefenderConfig.Enabled = true
- cfg.DefenderConfig.Driver = common.DefenderDriverProvider
- err := common.Initialize(cfg, 0)
- require.NoError(t, err)
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, defenderHosts, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- err = common.Initialize(oldConfig, 0)
- require.NoError(t, err)
- }
- }
- func TestRestoreShares(t *testing.T) {
- // shares should be restored preserving the UsedTokens, CreatedAt, LastUseAt, UpdatedAt,
- // and ExpiresAt, so an expired share can be restored while we cannot create an already
- // expired share
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- share := dataprovider.Share{
- ShareID: shortuuid.New(),
- Name: "share name",
- Description: "share description",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- Username: user.Username,
- CreatedAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(-144 * time.Hour)),
- UpdatedAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(-96 * time.Hour)),
- LastUseAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(-64 * time.Hour)),
- ExpiresAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(-48 * time.Hour)),
- MaxTokens: 10,
- UsedTokens: 8,
- AllowFrom: []string{"127.0.0.0/8"},
- }
- backupData := dataprovider.BackupData{
- Version: dataprovider.DumpVersion,
- }
- backupData.Shares = append(backupData.Shares, share)
- backupContent, err := json.Marshal(backupData)
- assert.NoError(t, err)
- _, _, err = httpdtest.LoaddataFromPostBody(backupContent, "0", "0", http.StatusOK)
- assert.NoError(t, err)
- shareGet, err := dataprovider.ShareExists(share.ShareID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, share, shareGet)
- share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(-142 * time.Hour))
- share.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(-92 * time.Hour))
- share.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(-62 * time.Hour))
- share.UsedTokens = 6
- backupData.Shares = []dataprovider.Share{share}
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- _, _, err = httpdtest.LoaddataFromPostBody(backupContent, "0", "0", http.StatusOK)
- assert.NoError(t, err)
- shareGet, err = dataprovider.ShareExists(share.ShareID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, share, shareGet)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestLoaddataFromPostBody(t *testing.T) {
- mappedPath := filepath.Join(os.TempDir(), "restored_folder")
- folderName := filepath.Base(mappedPath)
- role := getTestRole()
- role.ID = 1
- role.Name = "test_restored_role"
- group := getTestGroup()
- group.ID = 1
- group.Name = "test_group_restored"
- user := getTestUser()
- user.ID = 1
- user.Username = "test_user_restored"
- user.Groups = []sdk.GroupMapping{
- {
- Name: group.Name,
- Type: sdk.GroupTypePrimary,
- },
- }
- user.Role = role.Name
- admin := getTestAdmin()
- admin.ID = 1
- admin.Username = "test_admin_restored"
- admin.Permissions = []string{dataprovider.PermAdminAddUsers, dataprovider.PermAdminChangeUsers,
- dataprovider.PermAdminDeleteUsers, dataprovider.PermAdminViewUsers}
- admin.Role = role.Name
- backupData := dataprovider.BackupData{
- Version: dataprovider.DumpVersion,
- }
- backupData.Users = append(backupData.Users, user)
- backupData.Groups = append(backupData.Groups, group)
- backupData.Admins = append(backupData.Admins, admin)
- backupData.Roles = append(backupData.Roles, role)
- backupData.Folders = []vfs.BaseVirtualFolder{
- {
- Name: folderName,
- MappedPath: mappedPath,
- UsedQuotaSize: 123,
- UsedQuotaFiles: 456,
- LastQuotaUpdate: 789,
- Users: []string{"user"},
- },
- {
- Name: folderName,
- MappedPath: mappedPath + "1",
- },
- }
- backupData.APIKeys = append(backupData.APIKeys, dataprovider.APIKey{})
- backupData.Shares = append(backupData.Shares, dataprovider.Share{})
- backupContent, err := json.Marshal(backupData)
- assert.NoError(t, err)
- _, _, err = httpdtest.LoaddataFromPostBody(nil, "0", "0", http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.LoaddataFromPostBody(backupContent, "a", "0", http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.LoaddataFromPostBody([]byte("invalid content"), "0", "0", http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.LoaddataFromPostBody(backupContent, "0", "0", http.StatusInternalServerError)
- assert.NoError(t, err)
- keyID := util.GenerateUniqueID()
- backupData.APIKeys = []dataprovider.APIKey{
- {
- Name: "test key",
- Scope: dataprovider.APIKeyScopeAdmin,
- KeyID: keyID,
- Key: fmt.Sprintf("%v.%v", util.GenerateUniqueID(), util.GenerateUniqueID()),
- },
- }
- backupData.Shares = []dataprovider.Share{
- {
- ShareID: keyID,
- Name: keyID,
- Scope: dataprovider.ShareScopeWrite,
- Paths: []string{"/"},
- Username: user.Username,
- },
- }
- backupContent, err = json.Marshal(backupData)
- assert.NoError(t, err)
- _, resp, err := httpdtest.LoaddataFromPostBody(backupContent, "0", "0", http.StatusOK)
- assert.NoError(t, err, string(resp))
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user.Role)
- if assert.Len(t, user.Groups, 1) {
- assert.Equal(t, sdk.GroupTypePrimary, user.Groups[0].Type)
- assert.Equal(t, group.Name, user.Groups[0].Name)
- }
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, role.Admins, 1)
- assert.Len(t, role.Users, 1)
- _, err = dataprovider.ShareExists(keyID, user.Username)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, admin.Role)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, role.Admins, 0)
- assert.Len(t, role.Users, 0)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- apiKey, _, err := httpdtest.GetAPIKeyByID(keyID, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAPIKey(apiKey, http.StatusOK)
- assert.NoError(t, err)
- folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath+"1", folder.MappedPath)
- assert.Equal(t, int64(123), folder.UsedQuotaSize)
- assert.Equal(t, 456, folder.UsedQuotaFiles)
- assert.Equal(t, int64(789), folder.LastQuotaUpdate)
- assert.Len(t, folder.Users, 0)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestLoaddata(t *testing.T) {
- err := dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- mappedPath := filepath.Join(os.TempDir(), "restored_folder")
- folderName := filepath.Base(mappedPath)
- folderDesc := "restored folder desc"
- user := getTestUser()
- user.ID = 1
- user.Username = "test_user_restore"
- user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- },
- VirtualPath: "/vuserpath",
- })
- group := getTestGroup()
- group.ID = 1
- group.Name = "test_group_restore"
- group.VirtualFolders = append(group.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- },
- VirtualPath: "/vgrouppath",
- })
- role := getTestRole()
- role.ID = 1
- role.Name = "test_role_restore"
- user.Groups = append(user.Groups, sdk.GroupMapping{
- Name: group.Name,
- Type: sdk.GroupTypePrimary,
- })
- admin := getTestAdmin()
- admin.ID = 1
- admin.Username = "test_admin_restore"
- admin.Groups = []dataprovider.AdminGroupMapping{
- {
- Name: group.Name,
- },
- }
- ipListEntry := dataprovider.IPListEntry{
- IPOrNet: "172.16.2.4/32",
- Description: "entry desc",
- Type: dataprovider.IPListTypeDefender,
- Mode: dataprovider.ListModeDeny,
- Protocols: 3,
- }
- apiKey := dataprovider.APIKey{
- Name: util.GenerateUniqueID(),
- Scope: dataprovider.APIKeyScopeAdmin,
- KeyID: util.GenerateUniqueID(),
- Key: fmt.Sprintf("%v.%v", util.GenerateUniqueID(), util.GenerateUniqueID()),
- }
- share := dataprovider.Share{
- ShareID: util.GenerateUniqueID(),
- Name: util.GenerateUniqueID(),
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- Username: user.Username,
- }
- action := dataprovider.BaseEventAction{
- ID: 81,
- Name: "test restore action",
- Type: dataprovider.ActionTypeHTTP,
- Options: dataprovider.BaseEventActionOptions{
- HTTPConfig: dataprovider.EventActionHTTPConfig{
- Endpoint: "https://localhost:4567/action",
- Username: defaultUsername,
- Password: kms.NewPlainSecret(defaultPassword),
- Timeout: 10,
- SkipTLSVerify: true,
- Method: http.MethodPost,
- Body: `{"event":"{{Event}}","name":"{{Name}}"}`,
- },
- },
- }
- rule := dataprovider.EventRule{
- ID: 100,
- Name: "test rule restore",
- Description: "",
- Trigger: dataprovider.EventTriggerFsEvent,
- Conditions: dataprovider.EventConditions{
- FsEvents: []string{"download"},
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action.Name,
- },
- Order: 1,
- },
- },
- }
- configs := dataprovider.Configs{
- SFTPD: &dataprovider.SFTPDConfigs{
- HostKeyAlgos: []string{ssh.KeyAlgoRSA, ssh.CertAlgoRSAv01},
- PublicKeyAlgos: []string{ssh.InsecureKeyAlgoDSA},
- },
- SMTP: &dataprovider.SMTPConfigs{
- Host: "mail.example.com",
- Port: 587,
- From: "from@example.net",
- },
- }
- backupData := dataprovider.BackupData{
- Version: 14,
- }
- backupData.Configs = &configs
- backupData.Users = append(backupData.Users, user)
- backupData.Roles = append(backupData.Roles, role)
- backupData.Groups = append(backupData.Groups, group)
- backupData.Admins = append(backupData.Admins, admin)
- backupData.Folders = []vfs.BaseVirtualFolder{
- {
- Name: folderName,
- MappedPath: mappedPath + "1",
- UsedQuotaSize: 123,
- UsedQuotaFiles: 456,
- LastQuotaUpdate: 789,
- Users: []string{"user"},
- },
- {
- MappedPath: mappedPath,
- Name: folderName,
- Description: folderDesc,
- },
- }
- backupData.APIKeys = append(backupData.APIKeys, apiKey)
- backupData.Shares = append(backupData.Shares, share)
- backupData.EventActions = append(backupData.EventActions, action)
- backupData.EventRules = append(backupData.EventRules, rule)
- backupData.IPLists = append(backupData.IPLists, ipListEntry)
- backupContent, err := json.Marshal(backupData)
- assert.NoError(t, err)
- backupFilePath := filepath.Join(backupsPath, "backup.json")
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "a", "", http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "", "a", http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata("backup.json", "1", "", http.StatusBadRequest)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath+"a", "1", "", http.StatusBadRequest)
- assert.NoError(t, err)
- if runtime.GOOS != osWindows {
- err = os.Chmod(backupFilePath, 0111)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "1", "", http.StatusBadRequest)
- assert.NoError(t, err)
- err = os.Chmod(backupFilePath, 0644)
- assert.NoError(t, err)
- }
- // add objects from backup
- _, resp, err := httpdtest.Loaddata(backupFilePath, "1", "", http.StatusOK)
- assert.NoError(t, err, string(resp))
- // update from backup
- _, _, err = httpdtest.Loaddata(backupFilePath, "2", "", http.StatusOK)
- assert.NoError(t, err)
- configsGet, err := dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Equal(t, configs.SMTP, configsGet.SMTP)
- assert.Equal(t, []string{ssh.KeyAlgoRSA}, configsGet.SFTPD.HostKeyAlgos)
- assert.Equal(t, []string{ssh.InsecureKeyAlgoDSA}, configsGet.SFTPD.PublicKeyAlgos)
- assert.Len(t, configsGet.SFTPD.KexAlgorithms, 0)
- assert.Len(t, configsGet.SFTPD.Ciphers, 0)
- assert.Len(t, configsGet.SFTPD.MACs, 0)
- assert.Greater(t, configsGet.UpdatedAt, int64(0))
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, user.VirtualFolders, 1)
- assert.Len(t, user.Groups, 1)
- _, err = dataprovider.ShareExists(share.ShareID, user.Username)
- assert.NoError(t, err)
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group.VirtualFolders, 1)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, admin.Groups, 1)
- apiKey, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusOK)
- assert.NoError(t, err)
- action, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- entry, _, err := httpdtest.GetIPListEntry(ipListEntry.IPOrNet, ipListEntry.Type, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, entry.CreatedAt, int64(0))
- assert.Greater(t, entry.UpdatedAt, int64(0))
- assert.Equal(t, ipListEntry.Description, entry.Description)
- assert.Equal(t, ipListEntry.Protocols, entry.Protocols)
- assert.Equal(t, ipListEntry.Mode, entry.Mode)
- rule, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 1, rule.Status)
- if assert.Len(t, rule.Actions, 1) {
- if assert.NotNil(t, rule.Actions[0].BaseEventAction.Options.HTTPConfig.Password) {
- assert.Equal(t, sdkkms.SecretStatusSecretBox, rule.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, rule.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetPayload())
- assert.Empty(t, rule.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetKey())
- assert.Empty(t, rule.Actions[0].BaseEventAction.Options.HTTPConfig.Password.GetAdditionalData())
- }
- }
- response, _, err := httpdtest.Dumpdata("", "1", "0", http.StatusOK)
- assert.NoError(t, err)
- var dumpedData dataprovider.BackupData
- data, err := json.Marshal(response)
- assert.NoError(t, err)
- err = json.Unmarshal(data, &dumpedData)
- assert.NoError(t, err)
- found := false
- if assert.GreaterOrEqual(t, len(dumpedData.Users), 1) {
- for _, u := range dumpedData.Users {
- if u.Username == user.Username {
- found = true
- assert.Equal(t, len(user.VirtualFolders), len(u.VirtualFolders))
- assert.Equal(t, len(user.Groups), len(u.Groups))
- }
- }
- }
- assert.True(t, found)
- found = false
- if assert.GreaterOrEqual(t, len(dumpedData.Admins), 1) {
- for _, a := range dumpedData.Admins {
- if a.Username == admin.Username {
- found = true
- assert.Equal(t, len(admin.Groups), len(a.Groups))
- }
- }
- }
- assert.True(t, found)
- if assert.Len(t, dumpedData.Groups, 1) {
- assert.Equal(t, len(group.VirtualFolders), len(dumpedData.Groups[0].VirtualFolders))
- }
- if assert.Len(t, dumpedData.EventActions, 1) {
- assert.Equal(t, action.Name, dumpedData.EventActions[0].Name)
- }
- if assert.Len(t, dumpedData.EventRules, 1) {
- assert.Equal(t, rule.Name, dumpedData.EventRules[0].Name)
- assert.Len(t, dumpedData.EventRules[0].Actions, 1)
- }
- found = false
- for _, r := range dumpedData.Roles {
- if r.Name == role.Name {
- found = true
- }
- }
- assert.True(t, found)
- folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath, folder.MappedPath)
- assert.Equal(t, int64(123), folder.UsedQuotaSize)
- assert.Equal(t, 456, folder.UsedQuotaFiles)
- assert.Equal(t, int64(789), folder.LastQuotaUpdate)
- assert.Equal(t, folderDesc, folder.Description)
- assert.Len(t, folder.Users, 1)
- response, _, err = httpdtest.Dumpdata("", "1", "0", http.StatusOK, dataprovider.DumpScopeUsers)
- assert.NoError(t, err)
- dumpedData = dataprovider.BackupData{}
- data, err = json.Marshal(response)
- assert.NoError(t, err)
- err = json.Unmarshal(data, &dumpedData)
- assert.NoError(t, err)
- assert.Greater(t, len(dumpedData.Users), 0)
- assert.Len(t, dumpedData.Admins, 0)
- assert.Len(t, dumpedData.Folders, 0)
- assert.Len(t, dumpedData.Groups, 0)
- assert.Len(t, dumpedData.Roles, 0)
- assert.Len(t, dumpedData.EventRules, 0)
- assert.Len(t, dumpedData.EventActions, 0)
- assert.Len(t, dumpedData.IPLists, 0)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAPIKey(apiKey, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry, http.StatusOK)
- assert.NoError(t, err)
- err = os.Remove(backupFilePath)
- assert.NoError(t, err)
- err = createTestFile(backupFilePath, 20*1048576+1)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "1", "0", http.StatusBadRequest)
- assert.NoError(t, err)
- err = os.Remove(backupFilePath)
- assert.NoError(t, err)
- err = createTestFile(backupFilePath, 65535)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "1", "0", http.StatusBadRequest)
- assert.NoError(t, err)
- err = os.Remove(backupFilePath)
- assert.NoError(t, err)
- err = dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- }
- func TestLoaddataMode(t *testing.T) {
- err := dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- mappedPath := filepath.Join(os.TempDir(), "restored_fold")
- folderName := filepath.Base(mappedPath)
- configs := dataprovider.Configs{
- SFTPD: &dataprovider.SFTPDConfigs{
- PublicKeyAlgos: []string{ssh.KeyAlgoRSA},
- },
- }
- role := getTestRole()
- role.ID = 1
- role.Name = "test_role_load"
- role.Description = ""
- user := getTestUser()
- user.ID = 1
- user.Username = "test_user_restore"
- user.Role = role.Name
- group := getTestGroup()
- group.ID = 1
- group.Name = "test_group_restore"
- user.Groups = []sdk.GroupMapping{
- {
- Name: group.Name,
- Type: sdk.GroupTypePrimary,
- },
- }
- admin := getTestAdmin()
- admin.ID = 1
- admin.Username = "test_admin_restore"
- apiKey := dataprovider.APIKey{
- Name: util.GenerateUniqueID(),
- Scope: dataprovider.APIKeyScopeAdmin,
- KeyID: util.GenerateUniqueID(),
- Key: fmt.Sprintf("%v.%v", util.GenerateUniqueID(), util.GenerateUniqueID()),
- Description: "desc",
- }
- share := dataprovider.Share{
- ShareID: util.GenerateUniqueID(),
- Name: util.GenerateUniqueID(),
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- Username: user.Username,
- }
- action := dataprovider.BaseEventAction{
- ID: 81,
- Name: "test restore action data mode",
- Description: "action desc",
- Type: dataprovider.ActionTypeHTTP,
- Options: dataprovider.BaseEventActionOptions{
- HTTPConfig: dataprovider.EventActionHTTPConfig{
- Endpoint: "https://localhost:4567/mode",
- Username: defaultUsername,
- Password: kms.NewPlainSecret(defaultPassword),
- Timeout: 10,
- SkipTLSVerify: true,
- Method: http.MethodPost,
- Body: `{"event":"{{Event}}","name":"{{Name}}"}`,
- },
- },
- }
- rule := dataprovider.EventRule{
- ID: 100,
- Name: "test rule restore data mode",
- Description: "rule desc",
- Trigger: dataprovider.EventTriggerFsEvent,
- Conditions: dataprovider.EventConditions{
- FsEvents: []string{"mkdir"},
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action.Name,
- },
- Order: 1,
- },
- },
- }
- ipListEntry := dataprovider.IPListEntry{
- IPOrNet: "10.8.3.9/32",
- Description: "note",
- Type: dataprovider.IPListTypeDefender,
- Mode: dataprovider.ListModeDeny,
- Protocols: 7,
- }
- backupData := dataprovider.BackupData{
- Version: dataprovider.DumpVersion,
- }
- backupData.Configs = &configs
- backupData.Users = append(backupData.Users, user)
- backupData.Groups = append(backupData.Groups, group)
- backupData.Admins = append(backupData.Admins, admin)
- backupData.EventActions = append(backupData.EventActions, action)
- backupData.EventRules = append(backupData.EventRules, rule)
- backupData.Roles = append(backupData.Roles, role)
- backupData.Folders = []vfs.BaseVirtualFolder{
- {
- Name: folderName,
- MappedPath: mappedPath,
- UsedQuotaSize: 123,
- UsedQuotaFiles: 456,
- LastQuotaUpdate: 789,
- Users: []string{"user"},
- },
- {
- MappedPath: mappedPath + "1",
- Name: folderName,
- },
- }
- backupData.APIKeys = append(backupData.APIKeys, apiKey)
- backupData.Shares = append(backupData.Shares, share)
- backupData.IPLists = append(backupData.IPLists, ipListEntry)
- backupContent, _ := json.Marshal(backupData)
- backupFilePath := filepath.Join(backupsPath, "backup.json")
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- _, _, err = httpdtest.Loaddata(backupFilePath, "0", "0", http.StatusOK)
- assert.NoError(t, err)
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 1)
- folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath+"1", folder.MappedPath)
- assert.Equal(t, int64(123), folder.UsedQuotaSize)
- assert.Equal(t, 456, folder.UsedQuotaFiles)
- assert.Equal(t, int64(789), folder.LastQuotaUpdate)
- assert.Len(t, folder.Users, 0)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user.Role)
- oldUploadBandwidth := user.UploadBandwidth
- user.UploadBandwidth = oldUploadBandwidth + 128
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, role.Users, 1)
- assert.Len(t, role.Admins, 0)
- assert.Empty(t, role.Description)
- role.Description = "role desc"
- _, _, err = httpdtest.UpdateRole(role, http.StatusOK)
- assert.NoError(t, err)
- role.Description = ""
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, group.Users, 1)
- oldGroupDesc := group.Description
- group.Description = "new group description"
- group, _, err = httpdtest.UpdateGroup(group, http.StatusOK)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- oldInfo := admin.AdditionalInfo
- oldDesc := admin.Description
- admin.AdditionalInfo = "newInfo"
- admin.Description = "newDesc"
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- apiKey, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusOK)
- assert.NoError(t, err)
- oldAPIKeyDesc := apiKey.Description
- apiKey.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now())
- apiKey.Description = "new desc"
- apiKey, _, err = httpdtest.UpdateAPIKey(apiKey, http.StatusOK)
- assert.NoError(t, err)
- share.Description = "test desc"
- err = dataprovider.UpdateShare(&share, "", "", "")
- assert.NoError(t, err)
- action, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- oldActionDesc := action.Description
- action.Description = "new action description"
- action, _, err = httpdtest.UpdateEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- rule, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 0, rule.Status)
- oldRuleDesc := rule.Description
- rule.Description = "new rule description"
- rule, _, err = httpdtest.UpdateEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- entry, _, err := httpdtest.GetIPListEntry(ipListEntry.IPOrNet, ipListEntry.Type, http.StatusOK)
- assert.NoError(t, err)
- oldEntryDesc := entry.Description
- entry.Description = "new note"
- entry, _, err = httpdtest.UpdateIPListEntry(entry, http.StatusOK)
- assert.NoError(t, err)
- configs.SFTPD.PublicKeyAlgos = append(configs.SFTPD.PublicKeyAlgos, ssh.InsecureKeyAlgoDSA)
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- assert.NoError(t, err)
- backupData.Configs = &configs
- backupData.Folders = []vfs.BaseVirtualFolder{
- {
- MappedPath: mappedPath,
- Name: folderName,
- },
- }
- _, _, err = httpdtest.Loaddata(backupFilePath, "0", "1", http.StatusOK)
- assert.NoError(t, err)
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 2)
- group, _, err = httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, oldGroupDesc, group.Description)
- folder, _, err = httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath+"1", folder.MappedPath)
- assert.Equal(t, int64(123), folder.UsedQuotaSize)
- assert.Equal(t, 456, folder.UsedQuotaFiles)
- assert.Equal(t, int64(789), folder.LastQuotaUpdate)
- assert.Len(t, folder.Users, 0)
- action, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, oldActionDesc, action.Description)
- rule, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, oldRuleDesc, rule.Description)
- entry, _, err = httpdtest.GetIPListEntry(ipListEntry.IPOrNet, ipListEntry.Type, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, oldEntryDesc, entry.Description)
- c := common.NewBaseConnection("connID", common.ProtocolFTP, "", "", user)
- fakeConn := &fakeConnection{
- BaseConnection: c,
- }
- err = common.Connections.Add(fakeConn)
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 1)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, oldUploadBandwidth, user.UploadBandwidth)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, oldInfo, admin.AdditionalInfo)
- assert.NotEqual(t, oldDesc, admin.Description)
- apiKey, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusOK)
- assert.NoError(t, err)
- assert.NotEqual(t, int64(0), apiKey.ExpiresAt)
- assert.NotEqual(t, oldAPIKeyDesc, apiKey.Description)
- share, err = dataprovider.ShareExists(share.ShareID, user.Username)
- assert.NoError(t, err)
- assert.NotEmpty(t, share.Description)
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, role.Users, 1)
- assert.Len(t, role.Admins, 0)
- assert.NotEmpty(t, role.Description)
- _, _, err = httpdtest.Loaddata(backupFilePath, "0", "2", http.StatusOK)
- assert.NoError(t, err)
- // mode 2 will update the user and close the previous connection
- assert.Len(t, common.Connections.GetStats(""), 0)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, oldUploadBandwidth, user.UploadBandwidth)
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 1)
- // the group is referenced
- _, err = httpdtest.RemoveGroup(group, http.StatusBadRequest)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAPIKey(apiKey, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry, http.StatusOK)
- assert.NoError(t, err)
- err = os.Remove(backupFilePath)
- assert.NoError(t, err)
- err = dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- }
- func TestRateLimiter(t *testing.T) {
- oldConfig := config.GetCommonConfig()
- cfg := config.GetCommonConfig()
- cfg.RateLimitersConfig = []common.RateLimiterConfig{
- {
- Average: 1,
- Period: 1000,
- Burst: 1,
- Type: 1,
- Protocols: []string{common.ProtocolHTTP},
- },
- }
- err := common.Initialize(cfg, 0)
- assert.NoError(t, err)
- client := &http.Client{
- Timeout: 5 * time.Second,
- }
- resp, err := client.Get(httpBaseURL + healthzPath)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- resp, err = client.Get(httpBaseURL + healthzPath)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusTooManyRequests, resp.StatusCode)
- assert.Equal(t, "1", resp.Header.Get("Retry-After"))
- assert.NotEmpty(t, resp.Header.Get("X-Retry-In"))
- err = resp.Body.Close()
- assert.NoError(t, err)
- resp, err = client.Get(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusTooManyRequests, resp.StatusCode)
- assert.Equal(t, "1", resp.Header.Get("Retry-After"))
- assert.NotEmpty(t, resp.Header.Get("X-Retry-In"))
- err = resp.Body.Close()
- assert.NoError(t, err)
- resp, err = client.Get(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusTooManyRequests, resp.StatusCode)
- assert.Equal(t, "1", resp.Header.Get("Retry-After"))
- assert.NotEmpty(t, resp.Header.Get("X-Retry-In"))
- err = resp.Body.Close()
- assert.NoError(t, err)
- err = common.Initialize(oldConfig, 0)
- assert.NoError(t, err)
- }
- func TestHTTPSConnection(t *testing.T) {
- client := &http.Client{
- Timeout: 5 * time.Second,
- }
- resp, err := client.Get("https://localhost:8443" + healthzPath)
- if assert.Error(t, err) {
- if !strings.Contains(err.Error(), "certificate is not valid") &&
- !strings.Contains(err.Error(), "certificate signed by unknown authority") &&
- !strings.Contains(err.Error(), "certificate is not standards compliant") {
- assert.Fail(t, err.Error())
- }
- } else {
- resp.Body.Close()
- }
- }
- // test using mock http server
- func TestBasicUserHandlingMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, err := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusConflict, rr)
- user.MaxSessions = 10
- user.UploadBandwidth = 128
- user.Permissions["/"] = []string{dataprovider.PermAny, dataprovider.PermDelete, dataprovider.PermDownload}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, userPath+"/"+user.Username, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, userPath+"/"+user.Username, nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updatedUser)
- assert.NoError(t, err)
- assert.Equal(t, user.MaxSessions, updatedUser.MaxSessions)
- assert.Equal(t, user.UploadBandwidth, updatedUser.UploadBandwidth)
- assert.Equal(t, 1, len(updatedUser.Permissions["/"]))
- assert.True(t, util.Contains(updatedUser.Permissions["/"], dataprovider.PermAny))
- req, _ = http.NewRequest(http.MethodDelete, userPath+"/"+user.Username, nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestAddUserNoUsernameMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- user.Username = ""
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddUserInvalidHomeDirMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- user.HomeDir = "relative_path"
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddUserInvalidPermsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- user.Permissions["/"] = []string{}
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddFolderInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, folderPath, bytes.NewBuffer([]byte("invalid json")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddEventRuleInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, eventActionsPath, bytes.NewBuffer([]byte("invalid json")))
- require.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPost, eventRulesPath, bytes.NewBuffer([]byte("invalid json")))
- require.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddRoleInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, rolesPath, bytes.NewBuffer([]byte("{")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestIPListEntriesErrorsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, ipListsPath+"/a/b", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "invalid list type")
- req, err = http.NewRequest(http.MethodGet, ipListsPath+"/invalid", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "invalid list type")
- reqBody := bytes.NewBuffer([]byte("{"))
- req, err = http.NewRequest(http.MethodPost, ipListsPath+"/2", reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- entry := dataprovider.IPListEntry{
- IPOrNet: "172.120.1.1/32",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 0,
- }
- _, _, err = httpdtest.AddIPListEntry(entry, http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(ipListsPath, "1", url.PathEscape(entry.IPOrNet)), reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- _, err = httpdtest.RemoveIPListEntry(entry, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestRoleErrorsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- reqBody := bytes.NewBuffer([]byte("{"))
- req, err := http.NewRequest(http.MethodGet, rolesPath+"?limit=a", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- role, _, err := httpdtest.AddRole(getTestRole(), http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(rolesPath, role.Name), reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPut, path.Join(rolesPath, "missing_role"), reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestEventRuleErrorsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- reqBody := bytes.NewBuffer([]byte("invalid json body"))
- req, err := http.NewRequest(http.MethodGet, eventActionsPath+"?limit=b", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, eventRulesPath+"?limit=c", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- a := dataprovider.BaseEventAction{
- Name: "action name",
- Description: "test description",
- Type: dataprovider.ActionTypeBackup,
- Options: dataprovider.BaseEventActionOptions{},
- }
- action, _, err := httpdtest.AddEventAction(a, http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(eventActionsPath, action.Name), reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- r := dataprovider.EventRule{
- Name: "test event rule",
- Trigger: dataprovider.EventTriggerSchedule,
- Conditions: dataprovider.EventConditions{
- Schedules: []dataprovider.Schedule{
- {
- Hours: "2",
- DayOfWeek: "*",
- DayOfMonth: "*",
- Month: "*",
- },
- },
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action.Name,
- },
- Order: 1,
- },
- },
- }
- rule, _, err := httpdtest.AddEventRule(r, http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(eventRulesPath, rule.Name), reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- rule.Actions[0].Name = "misssing action name"
- asJSON, err := json.Marshal(rule)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(eventRulesPath, rule.Name), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- _, err = httpdtest.RemoveEventRule(rule, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveEventAction(action, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestGroupErrorsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- reqBody := bytes.NewBuffer([]byte("not a json string"))
- req, err := http.NewRequest(http.MethodPost, groupPath, reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, groupPath+"?limit=d", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- group, _, err := httpdtest.AddGroup(getTestGroup(), http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(groupPath, group.Name), reqBody)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUpdateFolderInvalidJsonMock(t *testing.T) {
- folder := vfs.BaseVirtualFolder{
- Name: "name",
- MappedPath: filepath.Clean(os.TempDir()),
- }
- folder, resp, err := httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPut, path.Join(folderPath, folder.Name), bytes.NewBuffer([]byte("not a json")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAddUserInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer([]byte("invalid json")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddAdminInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer([]byte("...")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestAddAdminNoPasswordMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Password = ""
- asJSON, err := json.Marshal(admin)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "please set a password")
- }
- func TestAdminTwoFactorLogin(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- admin1 := getTestAdmin()
- admin1.Username = altAdminUsername + "1"
- admin1.Password = altAdminPassword
- var permissions []string
- for _, p := range admin1.GetValidPerms() {
- if p != dataprovider.PermAdminAny && p != dataprovider.PermAdminDisableMFA {
- permissions = append(permissions, p)
- }
- }
- admin1.Permissions = permissions
- admin1, _, err = httpdtest.AddAdmin(admin1, http.StatusCreated)
- assert.NoError(t, err)
- // enable two factor authentication
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], admin.Username)
- assert.NoError(t, err)
- altToken, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- adminTOTPConfig := dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- }
- asJSON, err := json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var recCodes []recoveryCode
- err = json.Unmarshal(rr.Body.Bytes(), &recCodes)
- assert.NoError(t, err)
- assert.Len(t, recCodes, 12)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, admin.Filters.RecoveryCodes, 12)
- for _, c := range admin.Filters.RecoveryCodes {
- assert.Empty(t, c.Secret.GetAdditionalData())
- assert.Empty(t, c.Secret.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, c.Secret.GetStatus())
- assert.NotEmpty(t, c.Secret.GetPayload())
- }
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webAdminTwoFactorPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- cookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- // without a cookie
- req, err = http.NewRequest(http.MethodGet, webAdminTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminTwoFactorPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // any other page will be redirected to the two factor auth page
- req, err = http.NewRequest(http.MethodGet, webUsersPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- // a partial token cannot be used for user pages
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("passcode", passcode)
- // no csrf
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("passcode", "invalid_passcode")
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("passcode", "")
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("passcode", passcode)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webUsersPath, rr.Header().Get("Location"))
- // the same cookie cannot be reused
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusNotFound, rr.Code)
- // get a new cookie and login using a recovery code
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- recoveryCode := recCodes[0].Code
- form.Set("recovery_code", recoveryCode)
- // no csrf
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("recovery_code", "")
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("recovery_code", recoveryCode)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webUsersPath, rr.Header().Get("Location"))
- authenticatedCookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- //render MFA page
- req, err = http.NewRequest(http.MethodGet, webAdminMFAPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, authenticatedCookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check that the recovery code was marked as used
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- recCodes = nil
- err = json.Unmarshal(rr.Body.Bytes(), &recCodes)
- assert.NoError(t, err)
- assert.Len(t, recCodes, 12)
- found := false
- for _, rc := range recCodes {
- if rc.Code == recoveryCode {
- found = true
- assert.True(t, rc.Used)
- } else {
- assert.False(t, rc.Used)
- }
- }
- assert.True(t, found)
- // the same recovery code cannot be reused
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("recovery_code", recoveryCode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("recovery_code", "invalid_recovery_code")
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- // disable TOTP
- altToken1, err := getJWTAPITokenFromTestServer(admin1.Username, altAdminPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, adminPath+"/"+altAdminUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken1)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPut, adminPath+"/"+altAdminUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPut, adminPath+"/"+altAdminUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "two-factor authentication is not enabled")
- form = make(url.Values)
- form.Set("recovery_code", recoveryCode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18n2FADisabled)
- form.Set("passcode", passcode)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18n2FADisabled)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin1, http.StatusOK)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- req, err = http.NewRequest(http.MethodGet, webAdminMFAPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, authenticatedCookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- }
- func TestAdminTOTP(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- // TOTPConfig will be ignored on add
- admin.Filters.TOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: "config",
- Secret: kms.NewEmptySecret(),
- }
- asJSON, err := json.Marshal(admin)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 0)
- altToken, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, adminTOTPConfigsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var configs []mfa.TOTPConfig
- err = json.Unmarshal(rr.Body.Bytes(), &configs)
- assert.NoError(t, err, rr.Body.String())
- assert.Len(t, configs, len(mfa.GetAvailableTOTPConfigs()))
- totpConfig := configs[0]
- totpReq := generateTOTPRequest{
- ConfigName: totpConfig.Name,
- }
- asJSON, err = json.Marshal(totpReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPGeneratePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var totpGenResp generateTOTPResponse
- err = json.Unmarshal(rr.Body.Bytes(), &totpGenResp)
- assert.NoError(t, err)
- assert.NotEmpty(t, totpGenResp.Secret)
- assert.NotEmpty(t, totpGenResp.QRCode)
- passcode, err := generateTOTPPasscode(totpGenResp.Secret)
- assert.NoError(t, err)
- validateReq := validateTOTPRequest{
- ConfigName: totpGenResp.ConfigName,
- Passcode: passcode,
- Secret: totpGenResp.Secret,
- }
- asJSON, err = json.Marshal(validateReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPValidatePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // the same passcode cannot be reused
- req, err = http.NewRequest(http.MethodPost, adminTOTPValidatePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "this passcode was already used")
- adminTOTPConfig := dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: totpGenResp.ConfigName,
- Secret: kms.NewPlainSecret(totpGenResp.Secret),
- }
- asJSON, err = json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Equal(t, totpGenResp.ConfigName, admin.Filters.TOTPConfig.ConfigName)
- assert.Empty(t, admin.Filters.TOTPConfig.Secret.GetKey())
- assert.Empty(t, admin.Filters.TOTPConfig.Secret.GetAdditionalData())
- assert.NotEmpty(t, admin.Filters.TOTPConfig.Secret.GetPayload())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, admin.Filters.TOTPConfig.Secret.GetStatus())
- admin.Filters.TOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: false,
- ConfigName: util.GenerateUniqueID(),
- Secret: kms.NewEmptySecret(),
- }
- admin.Filters.RecoveryCodes = []dataprovider.RecoveryCode{
- {
- Secret: kms.NewEmptySecret(),
- },
- }
- admin, resp, err := httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err, string(resp))
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 12)
- // if we use token we cannot get or generate recovery codes
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- // now the same but with altToken
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var recCodes []recoveryCode
- err = json.Unmarshal(rr.Body.Bytes(), &recCodes)
- assert.NoError(t, err)
- assert.Len(t, recCodes, 12)
- // regenerate recovery codes
- req, err = http.NewRequest(http.MethodPost, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check that recovery codes are different
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var newRecCodes []recoveryCode
- err = json.Unmarshal(rr.Body.Bytes(), &newRecCodes)
- assert.NoError(t, err)
- assert.Len(t, newRecCodes, 12)
- assert.NotEqual(t, recCodes, newRecCodes)
- // disable 2FA, the update admin API should not work
- admin.Filters.TOTPConfig.Enabled = false
- admin.Filters.RecoveryCodes = nil
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, altAdminUsername, admin.Username)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 12)
- // use the dedicated API
- req, err = http.NewRequest(http.MethodPut, adminPath+"/"+altAdminUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.TOTPConfig.Enabled)
- assert.Len(t, admin.Filters.RecoveryCodes, 0)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(adminPath, altAdminUsername), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPut, adminPath+"/"+altAdminUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, admin2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestChangeAdminPwdInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPut, adminPwdPath, bytes.NewBuffer([]byte("{")))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- }
- func TestSMTPConfig(t *testing.T) {
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err := smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- smtpTestURL := path.Join(webConfigsPath, "smtp", "test")
- tokenHeader := "X-CSRF-TOKEN"
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, smtpTestURL, bytes.NewBuffer([]byte("{")))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- testReq := make(map[string]any)
- testReq["host"] = smtpCfg.Host
- testReq["port"] = 3525
- testReq["from"] = "from@example.com"
- asJSON, err := json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, smtpTestURL, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- testReq["recipient"] = "example@example.com"
- asJSON, err = json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, smtpTestURL, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- configs := dataprovider.Configs{
- SMTP: &dataprovider.SMTPConfigs{
- Host: "127.0.0.1",
- Port: 3535,
- User: "user@example.com",
- Password: kms.NewPlainSecret(defaultPassword),
- },
- }
- err = dataprovider.UpdateConfigs(&configs, "", "", "")
- assert.NoError(t, err)
- testReq["password"] = redactedSecret
- asJSON, err = json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, smtpTestURL, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), "server does not support SMTP AUTH")
- testReq["password"] = ""
- testReq["auth_type"] = 3
- testReq["oauth2"] = smtp.OAuth2Config{
- ClientSecret: redactedSecret,
- RefreshToken: redactedSecret,
- }
- asJSON, err = json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, smtpTestURL, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "smtp oauth2: client id is required")
- err = dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- }
- func TestOAuth2TokenRequest(t *testing.T) {
- tokenHeader := "X-CSRF-TOKEN"
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webOAuth2TokenPath, bytes.NewBuffer([]byte("{")))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- testReq := make(map[string]any)
- testReq["client_secret"] = redactedSecret
- asJSON, err := json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webOAuth2TokenPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "base redirect url is required")
- testReq["base_redirect_url"] = "http://localhost:8081"
- asJSON, err = json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webOAuth2TokenPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set(tokenHeader, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestMFAPermission(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientMFAPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientMFAPath
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user.Filters.WebClient = []string{sdk.WebClientMFADisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- webToken, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientMFAPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientMFAPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebUserTwoFactorLogin(t *testing.T) {
- u := getTestUser()
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- // enable two factor authentication
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- adminToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolHTTP},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var recCodes []recoveryCode
- err = json.Unmarshal(rr.Body.Bytes(), &recCodes)
- assert.NoError(t, err)
- assert.Len(t, recCodes, 12)
- user, _, err = httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Len(t, user.Filters.RecoveryCodes, 12)
- for _, c := range user.Filters.RecoveryCodes {
- assert.Empty(t, c.Secret.GetAdditionalData())
- assert.Empty(t, c.Secret.GetKey())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, c.Secret.GetStatus())
- assert.NotEmpty(t, c.Secret.GetPayload())
- }
- req, err = http.NewRequest(http.MethodGet, webClientTwoFactorPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webClientTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- form := getLoginForm(defaultUsername, defaultPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- cookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientTwoFactorPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // without a cookie
- req, err = http.NewRequest(http.MethodGet, webClientTwoFactorPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webClientTwoFactorRecoveryPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // any other page will be redirected to the two factor auth page
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- // a partial token cannot be used for admin pages
- req, err = http.NewRequest(http.MethodGet, webUsersPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("passcode", passcode)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("passcode", "invalid_user_passcode")
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("passcode", "")
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("passcode", passcode)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientFilesPath, rr.Header().Get("Location"))
- // the same cookie cannot be reused
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusNotFound, rr.Code)
- // get a new cookie and login using a recovery code
- form = getLoginForm(defaultUsername, defaultPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- recoveryCode := recCodes[0].Code
- form.Set("recovery_code", recoveryCode)
- // no csrf
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("recovery_code", "")
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("recovery_code", recoveryCode)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientFilesPath, rr.Header().Get("Location"))
- authenticatedCookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- //render MFA page
- req, err = http.NewRequest(http.MethodGet, webClientMFAPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, authenticatedCookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // get MFA qrcode
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientMFAPath, "qrcode?url="+url.QueryEscape(key.URL())), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, authenticatedCookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "image/png", rr.Header().Get("Content-Type"))
- // invalid MFA url
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientMFAPath, "qrcode?url="+url.QueryEscape("http://foo\x7f.eu")), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, authenticatedCookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- // check that the recovery code was marked as used
- req, err = http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- recCodes = nil
- err = json.Unmarshal(rr.Body.Bytes(), &recCodes)
- assert.NoError(t, err)
- assert.Len(t, recCodes, 12)
- found := false
- for _, rc := range recCodes {
- if rc.Code == recoveryCode {
- found = true
- assert.True(t, rc.Used)
- } else {
- assert.False(t, rc.Used)
- }
- }
- assert.True(t, found)
- // the same recovery code cannot be reused
- form = getLoginForm(defaultUsername, defaultPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("recovery_code", recoveryCode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form.Set("recovery_code", "invalid_user_recovery_code")
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form = getLoginForm(defaultUsername, defaultPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- // disable TOTP
- req, err = http.NewRequest(http.MethodPut, userPath+"/"+user.Username+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPut, userPath+"/"+user.Username+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "two-factor authentication is not enabled")
- form = make(url.Values)
- form.Set("recovery_code", recoveryCode)
- form.Set("passcode", passcode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18n2FADisabled)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18n2FADisabled)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- req, err = http.NewRequest(http.MethodGet, webClientMFAPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, authenticatedCookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- }
- func TestWebUserTwoFactoryLoginRedirect(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolHTTP},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- form := getLoginForm(defaultUsername, defaultPassword, csrfToken)
- uri := webClientFilesPath + "?path=%2F"
- loginURI := webClientLoginPath + "?next=" + url.QueryEscape(uri)
- expectedURI := webClientTwoFactorPath + "?next=" + url.QueryEscape(uri)
- req, err = http.NewRequest(http.MethodPost, loginURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = loginURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, expectedURI, rr.Header().Get("Location"))
- cookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- // test unsafe redirects
- externalURI := webClientLoginPath + "?next=" + url.QueryEscape("https://example.com")
- req, err = http.NewRequest(http.MethodPost, externalURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = externalURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- internalURI := webClientLoginPath + "?next=" + url.QueryEscape(webClientMFAPath)
- req, err = http.NewRequest(http.MethodPost, internalURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = internalURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- // render two factor page
- req, err = http.NewRequest(http.MethodGet, expectedURI, nil)
- assert.NoError(t, err)
- req.RequestURI = expectedURI
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), fmt.Sprintf("action=%q", expectedURI))
- // login with the passcode
- passcode, err := generateTOTPPasscode(key.Secret())
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set("passcode", passcode)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, expectedURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.RequestURI = expectedURI
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, uri, rr.Header().Get("Location"))
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestSearchEvents(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, fsEventsPath+"?limit=10&order=ASC&fs_provider=0", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- events := make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &events)
- assert.NoError(t, err)
- if assert.Len(t, events, 1) {
- ev := events[0]
- for _, field := range []string{"id", "timestamp", "action", "username", "fs_path", "status", "protocol",
- "ip", "session_id", "fs_provider", "bucket", "endpoint", "open_flags", "role", "instance_id"} {
- _, ok := ev[field]
- assert.True(t, ok, field)
- }
- }
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?limit=10&order=ASC&role=role1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- events = nil
- err = json.Unmarshal(rr.Body.Bytes(), &events)
- assert.NoError(t, err)
- assert.Len(t, events, 1)
- // CSV export
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?limit=10&order=ASC&csv_export=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "text/csv", rr.Header().Get("Content-Type"))
- // the test eventsearcher plugin returns error if start_timestamp < 0
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?start_timestamp=-1&end_timestamp=123456&statuses=1,2", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- // CSV export with error
- exportFunc := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?start_timestamp=-2&csv_export=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- }
- exportFunc()
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?limit=e", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, providerEventsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- events = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &events)
- assert.NoError(t, err)
- if assert.Len(t, events, 1) {
- ev := events[0]
- for _, field := range []string{"id", "timestamp", "action", "username", "object_type", "object_name",
- "object_data", "role", "instance_id"} {
- _, ok := ev[field]
- assert.True(t, ok, field)
- }
- }
- req, err = http.NewRequest(http.MethodGet, providerEventsPath+"?omit_object_data=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- events = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &events)
- assert.NoError(t, err)
- if assert.Len(t, events, 1) {
- ev := events[0]
- field := "object_data"
- _, ok := ev[field]
- assert.False(t, ok, field)
- }
- // CSV export
- req, err = http.NewRequest(http.MethodGet, providerEventsPath+"?csv_export=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "text/csv", rr.Header().Get("Content-Type"))
- // the test eventsearcher plugin returns error if start_timestamp < 0
- req, err = http.NewRequest(http.MethodGet, providerEventsPath+"?start_timestamp=-1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- // CSV export with error
- exportFunc = func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err = http.NewRequest(http.MethodGet, providerEventsPath+"?start_timestamp=-1&csv_export=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- }
- exportFunc()
- req, err = http.NewRequest(http.MethodGet, logEventsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- events = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &events)
- assert.NoError(t, err)
- if assert.Len(t, events, 1) {
- ev := events[0]
- for _, field := range []string{"id", "timestamp", "event", "ip", "message", "role", "instance_id"} {
- _, ok := ev[field]
- assert.True(t, ok, field)
- }
- }
- req, err = http.NewRequest(http.MethodGet, logEventsPath+"?events=a,1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // CSV export
- req, err = http.NewRequest(http.MethodGet, logEventsPath+"?csv_export=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "text/csv", rr.Header().Get("Content-Type"))
- // the test eventsearcher plugin returns error if start_timestamp < 0
- req, err = http.NewRequest(http.MethodGet, logEventsPath+"?start_timestamp=-1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- // CSV export with error
- exportFunc = func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err = http.NewRequest(http.MethodGet, logEventsPath+"?start_timestamp=-1&csv_export=true", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- }
- exportFunc()
- req, err = http.NewRequest(http.MethodGet, providerEventsPath+"?limit=2000", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, logEventsPath+"?limit=2000", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?start_timestamp=a", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?end_timestamp=a", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?order=ASSC", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?statuses=a,b", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, fsEventsPath+"?fs_provider=a", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, webEventsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestMFAErrors(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- assert.False(t, user.Filters.TOTPConfig.Enabled)
- userToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- adminToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- // invalid config name
- totpReq := generateTOTPRequest{
- ConfigName: "invalid config name",
- }
- asJSON, err := json.Marshal(totpReq)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userTOTPGeneratePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // invalid JSON
- invalidJSON := []byte("not a JSON")
- req, err = http.NewRequest(http.MethodPost, userTOTPGeneratePath, bytes.NewBuffer(invalidJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(invalidJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(invalidJSON))
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPost, adminTOTPValidatePath, bytes.NewBuffer(invalidJSON))
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // invalid TOTP config name
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: "missing name",
- Secret: kms.NewPlainSecret(xid.New().String()),
- Protocols: []string{common.ProtocolSSH},
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: config name")
- // invalid TOTP secret
- userTOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: nil,
- Protocols: []string{common.ProtocolSSH},
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: secret is mandatory")
- // no protocol
- userTOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: kms.NewPlainSecret(xid.New().String()),
- Protocols: nil,
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: specify at least one protocol")
- // invalid protocol
- userTOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: kms.NewPlainSecret(xid.New().String()),
- Protocols: []string{common.ProtocolWebDAV},
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: invalid protocol")
- adminTOTPConfig := dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: "",
- Secret: kms.NewPlainSecret("secret"),
- }
- asJSON, err = json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: config name is mandatory")
- adminTOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: nil,
- }
- asJSON, err = json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: secret is mandatory")
- // invalid TOTP secret status
- userTOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: kms.NewSecret(sdkkms.SecretStatusRedacted, "", "", ""),
- Protocols: []string{common.ProtocolSSH},
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // previous secret will be preserved and we have no secret saved
- assert.Contains(t, rr.Body.String(), "totp: secret is mandatory")
- req, err = http.NewRequest(http.MethodPost, adminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "totp: secret is mandatory")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestMFAInvalidSecret(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- userToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- user.Password = defaultPassword
- user.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: kms.NewSecret(sdkkms.SecretStatusSecretBox, "payload", "key", user.Username),
- Protocols: []string{common.ProtocolSSH, common.ProtocolHTTP},
- }
- user.Filters.RecoveryCodes = append(user.Filters.RecoveryCodes, dataprovider.RecoveryCode{
- Used: false,
- Secret: kms.NewSecret(sdkkms.SecretStatusSecretBox, "payload", "key", user.Username),
- })
- err = dataprovider.UpdateUser(&user, "", "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), "Unable to decrypt recovery codes")
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- form := getLoginForm(defaultUsername, defaultPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webClientTwoFactorPath, rr.Header().Get("Location"))
- cookie, err := getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("passcode", "123456")
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusInternalServerError, rr.Code)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("recovery_code", "RC-123456")
- req, err = http.NewRequest(http.MethodPost, webClientTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusInternalServerError, rr.Code)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, userTokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", "authcode")
- req.SetBasicAuth(defaultUsername, defaultPassword)
- resp, err := httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err = httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- admin.Password = altAdminPassword
- admin.Filters.TOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: mfa.GetAvailableTOTPConfigNames()[0],
- Secret: kms.NewSecret(sdkkms.SecretStatusSecretBox, "payload", "key", user.Username),
- }
- admin.Filters.RecoveryCodes = append(user.Filters.RecoveryCodes, dataprovider.RecoveryCode{
- Used: false,
- Secret: kms.NewSecret(sdkkms.SecretStatusSecretBox, "payload", "key", user.Username),
- })
- err = dataprovider.UpdateAdmin(&admin, "", "", "")
- assert.NoError(t, err)
- csrfToken, err = getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.Equal(t, webAdminTwoFactorPath, rr.Header().Get("Location"))
- cookie, err = getCookieFromResponse(rr)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("passcode", "123456")
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusInternalServerError, rr.Code)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("recovery_code", "RC-123456")
- req, err = http.NewRequest(http.MethodPost, webAdminTwoFactorRecoveryPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusInternalServerError, rr.Code)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v%v", httpBaseURL, tokenPath), nil)
- assert.NoError(t, err)
- req.Header.Set("X-SFTPGO-OTP", "auth-code")
- req.SetBasicAuth(altAdminUsername, altAdminPassword)
- resp, err = httpclient.GetHTTPClient().Do(req)
- assert.NoError(t, err)
- assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
- err = resp.Body.Close()
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebUserTOTP(t *testing.T) {
- u := getTestUser()
- // TOTPConfig will be ignored on add
- u.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: "",
- Secret: kms.NewEmptySecret(),
- Protocols: []string{common.ProtocolSSH},
- }
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- assert.False(t, user.Filters.TOTPConfig.Enabled)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, userTOTPConfigsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var configs []mfa.TOTPConfig
- err = json.Unmarshal(rr.Body.Bytes(), &configs)
- assert.NoError(t, err, rr.Body.String())
- assert.Len(t, configs, len(mfa.GetAvailableTOTPConfigs()))
- totpConfig := configs[0]
- totpReq := generateTOTPRequest{
- ConfigName: totpConfig.Name,
- }
- asJSON, err := json.Marshal(totpReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPGeneratePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var totpGenResp generateTOTPResponse
- err = json.Unmarshal(rr.Body.Bytes(), &totpGenResp)
- assert.NoError(t, err)
- assert.NotEmpty(t, totpGenResp.Secret)
- assert.NotEmpty(t, totpGenResp.QRCode)
- passcode, err := generateTOTPPasscode(totpGenResp.Secret)
- assert.NoError(t, err)
- validateReq := validateTOTPRequest{
- ConfigName: totpGenResp.ConfigName,
- Passcode: passcode,
- Secret: totpGenResp.Secret,
- }
- asJSON, err = json.Marshal(validateReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPValidatePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // the same passcode cannot be reused
- req, err = http.NewRequest(http.MethodPost, userTOTPValidatePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "this passcode was already used")
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: totpGenResp.ConfigName,
- Secret: kms.NewPlainSecret(totpGenResp.Secret),
- Protocols: []string{common.ProtocolSSH},
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- totpCfg := user.Filters.TOTPConfig
- assert.True(t, totpCfg.Enabled)
- secretPayload := totpCfg.Secret.GetPayload()
- assert.Equal(t, totpGenResp.ConfigName, totpCfg.ConfigName)
- assert.Empty(t, totpCfg.Secret.GetKey())
- assert.Empty(t, totpCfg.Secret.GetAdditionalData())
- assert.NotEmpty(t, secretPayload)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, totpCfg.Secret.GetStatus())
- assert.Len(t, totpCfg.Protocols, 1)
- assert.Contains(t, totpCfg.Protocols, common.ProtocolSSH)
- // update protocols only
- userTOTPConfig = dataprovider.UserTOTPConfig{
- Protocols: []string{common.ProtocolSSH, common.ProtocolFTP},
- Secret: kms.NewEmptySecret(),
- }
- asJSON, err = json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // update the user, TOTP should not be affected
- user.Filters.TOTPConfig = dataprovider.UserTOTPConfig{
- Enabled: false,
- Secret: kms.NewEmptySecret(),
- }
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.TOTPConfig.Enabled)
- assert.Equal(t, totpCfg.ConfigName, user.Filters.TOTPConfig.ConfigName)
- assert.Empty(t, user.Filters.TOTPConfig.Secret.GetKey())
- assert.Empty(t, user.Filters.TOTPConfig.Secret.GetAdditionalData())
- assert.Equal(t, secretPayload, user.Filters.TOTPConfig.Secret.GetPayload())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, user.Filters.TOTPConfig.Secret.GetStatus())
- assert.Len(t, user.Filters.TOTPConfig.Protocols, 2)
- assert.Contains(t, user.Filters.TOTPConfig.Protocols, common.ProtocolSSH)
- assert.Contains(t, user.Filters.TOTPConfig.Protocols, common.ProtocolFTP)
- req, err = http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var recCodes []recoveryCode
- err = json.Unmarshal(rr.Body.Bytes(), &recCodes)
- assert.NoError(t, err)
- assert.Len(t, recCodes, 12)
- // regenerate recovery codes
- req, err = http.NewRequest(http.MethodPost, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check that recovery codes are different
- req, err = http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var newRecCodes []recoveryCode
- err = json.Unmarshal(rr.Body.Bytes(), &newRecCodes)
- assert.NoError(t, err)
- assert.Len(t, newRecCodes, 12)
- assert.NotEqual(t, recCodes, newRecCodes)
- // disable 2FA, the update user API should not work
- adminToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user.Filters.TOTPConfig.Enabled = false
- user.Filters.RecoveryCodes = nil
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Equal(t, defaultUsername, user.Username)
- assert.True(t, user.Filters.TOTPConfig.Enabled)
- assert.Len(t, user.Filters.RecoveryCodes, 12)
- // use the dedicated API
- req, err = http.NewRequest(http.MethodPut, userPath+"/"+defaultUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, user.Filters.TOTPConfig.Enabled)
- assert.Len(t, user.Filters.RecoveryCodes, 0)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userPath+"/"+defaultUsername+"/2fa/disable", nil)
- assert.NoError(t, err)
- setBearerForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, user2FARecoveryCodesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, userTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestWebAPIChangeUserProfileMock(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- assert.False(t, user.Filters.AllowAPIKeyAuth)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- // invalid json
- req, err := http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer([]byte("{")))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- email := "userapi@example.com"
- description := "user API description"
- profileReq := make(map[string]any)
- profileReq["allow_api_key_auth"] = true
- profileReq["email"] = email
- profileReq["description"] = description
- profileReq["public_keys"] = []string{testPubKey, testPubKey1}
- profileReq["tls_certs"] = []string{httpsCert}
- asJSON, err := json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- profileReq = make(map[string]any)
- req, err = http.NewRequest(http.MethodGet, userProfilePath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = json.Unmarshal(rr.Body.Bytes(), &profileReq)
- assert.NoError(t, err)
- assert.Equal(t, email, profileReq["email"].(string))
- assert.Equal(t, description, profileReq["description"].(string))
- assert.True(t, profileReq["allow_api_key_auth"].(bool))
- val, ok := profileReq["public_keys"].([]any)
- if assert.True(t, ok, profileReq) {
- assert.Len(t, val, 2)
- }
- val, ok = profileReq["tls_certs"].([]any)
- if assert.True(t, ok, profileReq) {
- assert.Len(t, val, 1)
- }
- // set an invalid email
- profileReq = make(map[string]any)
- profileReq["email"] = "notavalidemail"
- asJSON, err = json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Validation error: email")
- // set an invalid public key
- profileReq = make(map[string]any)
- profileReq["public_keys"] = []string{"not a public key"}
- asJSON, err = json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Validation error: error parsing public key")
- // set an invalid TLS certificate
- profileReq = make(map[string]any)
- profileReq["tls_certs"] = []string{"not a TLS cert"}
- asJSON, err = json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Validation error: invalid TLS certificate")
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientPubKeyChangeDisabled,
- sdk.WebClientTLSCertChangeDisabled}
- user.Email = email
- user.Description = description
- user.Filters.AllowAPIKeyAuth = true
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- profileReq = make(map[string]any)
- profileReq["allow_api_key_auth"] = false
- profileReq["email"] = email
- profileReq["description"] = description + "_mod" //nolint:goconst
- profileReq["public_keys"] = []string{testPubKey}
- profileReq["tls_certs"] = []string{}
- asJSON, err = json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), "Profile updated")
- // check that api key auth and public keys were not changed
- profileReq = make(map[string]any)
- req, err = http.NewRequest(http.MethodGet, userProfilePath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = json.Unmarshal(rr.Body.Bytes(), &profileReq)
- assert.NoError(t, err)
- assert.Equal(t, email, profileReq["email"].(string))
- assert.Equal(t, description+"_mod", profileReq["description"].(string))
- assert.True(t, profileReq["allow_api_key_auth"].(bool))
- val, ok = profileReq["public_keys"].([]any)
- if assert.True(t, ok, profileReq) {
- assert.Len(t, val, 2)
- }
- val, ok = profileReq["tls_certs"].([]any)
- if assert.True(t, ok, profileReq) {
- assert.Len(t, val, 1)
- }
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientInfoChangeDisabled}
- user.Description = description + "_mod"
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- profileReq = make(map[string]any)
- profileReq["allow_api_key_auth"] = false
- profileReq["email"] = "newemail@apiuser.com"
- profileReq["description"] = description
- profileReq["public_keys"] = []string{testPubKey}
- asJSON, err = json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- profileReq = make(map[string]any)
- req, err = http.NewRequest(http.MethodGet, userProfilePath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = json.Unmarshal(rr.Body.Bytes(), &profileReq)
- assert.NoError(t, err)
- assert.Equal(t, email, profileReq["email"].(string))
- assert.Equal(t, description+"_mod", profileReq["description"].(string))
- assert.True(t, profileReq["allow_api_key_auth"].(bool))
- assert.Len(t, profileReq["public_keys"].([]any), 1)
- // finally disable all profile permissions
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientInfoChangeDisabled,
- sdk.WebClientPubKeyChangeDisabled, sdk.WebClientTLSCertChangeDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "You are not allowed to change anything")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userProfilePath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPut, userProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestPermGroupOverride(t *testing.T) {
- g := getTestGroup()
- g.UserSettings.Filters.WebClient = []string{sdk.WebClientPasswordChangeDisabled}
- group, _, err := httpdtest.AddGroup(g, http.StatusCreated)
- assert.NoError(t, err)
- u := getTestUser()
- u.Groups = []sdk.GroupMapping{
- {
- Name: group.Name,
- Type: sdk.GroupTypePrimary,
- },
- }
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- pwd := make(map[string]string)
- pwd["current_password"] = defaultPassword
- pwd["new_password"] = altAdminPassword
- asJSON, err := json.Marshal(pwd)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPut, userPwdPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- group.UserSettings.Filters.TwoFactorAuthProtocols = []string{common.ProtocolHTTP}
- group, _, err = httpdtest.UpdateGroup(group, http.StatusOK)
- assert.NoError(t, err)
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Two-factor authentication requirements not met, please configure two-factor authentication for the following protocols")
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError2FARequired)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebAPIChangeUserPwdMock(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- // invalid json
- req, err := http.NewRequest(http.MethodPut, userPwdPath, bytes.NewBuffer([]byte("{")))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- pwd := make(map[string]string)
- pwd["current_password"] = defaultPassword
- pwd["new_password"] = defaultPassword
- asJSON, err := json.Marshal(pwd)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userPwdPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the new password must be different from the current one")
- pwd["new_password"] = altAdminPassword
- asJSON, err = json.Marshal(pwd)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userPwdPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, altAdminPassword)
- assert.NoError(t, err)
- assert.NotEmpty(t, token)
- // remove the change password permission
- user.Filters.WebClient = []string{sdk.WebClientPasswordChangeDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Len(t, user.Filters.WebClient, 1)
- assert.Contains(t, user.Filters.WebClient, sdk.WebClientPasswordChangeDisabled)
- token, err = getJWTAPIUserTokenFromTestServer(defaultUsername, altAdminPassword)
- assert.NoError(t, err)
- assert.NotEmpty(t, token)
- pwd["current_password"] = altAdminPassword
- pwd["new_password"] = defaultPassword
- asJSON, err = json.Marshal(pwd)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userPwdPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestLoginInvalidPasswordMock(t *testing.T) {
- _, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass+"1")
- assert.Error(t, err)
- // now a login with no credentials
- req, _ := http.NewRequest(http.MethodGet, "/api/v2/token", nil)
- rr := executeRequest(req)
- assert.Equal(t, http.StatusUnauthorized, rr.Code)
- }
- func TestWebAPIChangeAdminProfileMock(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.AllowAPIKeyAuth)
- token, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- // invalid json
- req, err := http.NewRequest(http.MethodPut, adminProfilePath, bytes.NewBuffer([]byte("{")))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- email := "adminapi@example.com"
- description := "admin API description"
- profileReq := make(map[string]any)
- profileReq["allow_api_key_auth"] = true
- profileReq["email"] = email
- profileReq["description"] = description
- asJSON, err := json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, adminProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), "Profile updated")
- profileReq = make(map[string]any)
- req, err = http.NewRequest(http.MethodGet, adminProfilePath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = json.Unmarshal(rr.Body.Bytes(), &profileReq)
- assert.NoError(t, err)
- assert.Equal(t, email, profileReq["email"].(string))
- assert.Equal(t, description, profileReq["description"].(string))
- assert.True(t, profileReq["allow_api_key_auth"].(bool))
- // set an invalid email
- profileReq["email"] = "admin_invalid_email"
- asJSON, err = json.Marshal(profileReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, adminProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Validation error: email")
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, adminProfilePath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPut, adminProfilePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestChangeAdminPwdMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin.Permissions = []string{dataprovider.PermAdminAddUsers, dataprovider.PermAdminDeleteUsers}
- asJSON, err := json.Marshal(admin)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- altToken, err := getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- pwd := make(map[string]string)
- pwd["current_password"] = altAdminPassword
- pwd["new_password"] = defaultTokenAuthPass
- asJSON, err = json.Marshal(pwd)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, adminPwdPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = getJWTAPITokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.Error(t, err)
- altToken, err = getJWTAPITokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, adminPwdPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr) // current password does not match
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(adminPath, altAdminUsername), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestUpdateAdminMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- _, err = getJWTAPITokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.Error(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Permissions = []string{dataprovider.PermAdminManageAdmins}
- asJSON, err := json.Marshal(admin)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = getJWTAPITokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, "abc"), bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, altAdminUsername), bytes.NewBuffer([]byte("no json")))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- admin.Permissions = nil
- asJSON, err = json.Marshal(admin)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, altAdminUsername), bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- admin = getTestAdmin()
- admin.Status = 0
- asJSON, err = json.Marshal(admin)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, defaultTokenAuthUser), bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "you cannot disable yourself")
- admin.Status = 1
- admin.Permissions = []string{dataprovider.PermAdminAddUsers}
- asJSON, err = json.Marshal(admin)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, defaultTokenAuthUser), bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "you cannot remove these permissions to yourself")
- admin.Permissions = []string{dataprovider.PermAdminAny}
- admin.Role = "missing role"
- asJSON, err = json.Marshal(admin)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, defaultTokenAuthUser), bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "you cannot add/change your role")
- admin.Role = ""
- altToken, err := getJWTAPITokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin.Password = "" // it must remain unchanged
- admin.Permissions = []string{dataprovider.PermAdminManageAdmins, dataprovider.PermAdminCloseConnections}
- asJSON, err = json.Marshal(admin)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(adminPath, altAdminUsername), bytes.NewBuffer(asJSON))
- setBearerForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = getJWTAPITokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(adminPath, altAdminUsername), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestAdminLastLoginWithAPIKey(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Filters.AllowAPIKeyAuth = true
- admin, resp, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, int64(0), admin.LastLogin)
- apiKey := dataprovider.APIKey{
- Name: "admin API key",
- Scope: dataprovider.APIKeyScopeAdmin,
- Admin: altAdminUsername,
- LastUseAt: 123,
- }
- apiKey, resp, err = httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, int64(0), apiKey.LastUseAt)
- req, err := http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, admin.Username)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, admin.LastLogin, int64(0))
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserLastLoginWithAPIKey(t *testing.T) {
- user := getTestUser()
- user.Filters.AllowAPIKeyAuth = true
- user, resp, err := httpdtest.AddUser(user, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- assert.Equal(t, int64(0), user.LastLogin)
- apiKey := dataprovider.APIKey{
- Name: "user API key",
- Scope: dataprovider.APIKeyScopeUser,
- User: user.Username,
- }
- apiKey, resp, err = httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, user.LastLogin, int64(0))
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestAdminHandlingWithAPIKeys(t *testing.T) {
- sysAdmin, _, err := httpdtest.GetAdminByUsername(defaultTokenAuthUser, http.StatusOK)
- assert.NoError(t, err)
- sysAdmin.Filters.AllowAPIKeyAuth = true
- sysAdmin, _, err = httpdtest.UpdateAdmin(sysAdmin, http.StatusOK)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Name: "test admin API key",
- Scope: dataprovider.APIKeyScopeAdmin,
- Admin: defaultTokenAuthUser,
- }
- apiKey, resp, err := httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- asJSON, err := json.Marshal(admin)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = getJWTAPITokenFromTestServer(altAdminUsername, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin.Filters.AllowAPIKeyAuth = true
- asJSON, err = json.Marshal(admin)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(adminPath, altAdminUsername), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(adminPath, altAdminUsername), nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var adminGet dataprovider.Admin
- err = json.Unmarshal(rr.Body.Bytes(), &adminGet)
- assert.NoError(t, err)
- assert.True(t, adminGet.Filters.AllowAPIKeyAuth)
- req, err = http.NewRequest(http.MethodPut, path.Join(adminPath, defaultTokenAuthUser), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "updating the admin impersonated with an API key is not allowed")
- // changing the password for the impersonated admin is not allowed
- pwd := make(map[string]string)
- pwd["current_password"] = defaultTokenAuthPass
- pwd["new_password"] = altAdminPassword
- asJSON, err = json.Marshal(pwd)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, adminPwdPath, bytes.NewBuffer(asJSON))
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "API key authentication is not allowed")
- req, err = http.NewRequest(http.MethodDelete, path.Join(adminPath, defaultTokenAuthUser), nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "you cannot delete yourself")
- req, err = http.NewRequest(http.MethodDelete, path.Join(adminPath, altAdminUsername), nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveAPIKey(apiKey, http.StatusOK)
- assert.NoError(t, err)
- dbAdmin, err := dataprovider.AdminExists(defaultTokenAuthUser)
- assert.NoError(t, err)
- dbAdmin.Filters.AllowAPIKeyAuth = false
- err = dataprovider.UpdateAdmin(&dbAdmin, "", "", "")
- assert.NoError(t, err)
- sysAdmin, _, err = httpdtest.GetAdminByUsername(defaultTokenAuthUser, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, sysAdmin.Filters.AllowAPIKeyAuth)
- }
- func TestUserHandlingWithAPIKey(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Filters.AllowAPIKeyAuth = true
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Name: "test admin API key",
- Scope: dataprovider.APIKeyScopeAdmin,
- Admin: admin.Username,
- }
- apiKey, _, err = httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, err := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- user.Filters.DisableFsChecks = true
- user.Description = "desc"
- userAsJSON = getUserAsJSON(t, user)
- req, err = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updatedUser dataprovider.User
- err = json.Unmarshal(rr.Body.Bytes(), &updatedUser)
- assert.NoError(t, err)
- assert.True(t, updatedUser.Filters.DisableFsChecks)
- assert.Equal(t, user.Description, updatedUser.Description)
- req, err = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestUpdateUserQuotaUsageMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- var user dataprovider.User
- u := getTestUser()
- usedQuotaFiles := 1
- usedQuotaSize := int64(65535)
- u.UsedQuotaFiles = usedQuotaFiles
- u.UsedQuotaSize = usedQuotaSize
- u.QuotaFiles = 100
- userAsJSON := getUserAsJSON(t, u)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "users", u.Username, "usage"), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, user.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize, user.UsedQuotaSize)
- // now update only quota size
- u.UsedQuotaFiles = 0
- userAsJSON = getUserAsJSON(t, u)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "users", u.Username, "usage")+"?mode=add", bytes.NewBuffer(userAsJSON)) //nolint:goconst
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, user.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize*2, user.UsedQuotaSize)
- // only quota files
- u.UsedQuotaFiles = usedQuotaFiles
- u.UsedQuotaSize = 0
- userAsJSON = getUserAsJSON(t, u)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "users", u.Username, "usage")+"?mode=add", bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles*2, user.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize*2, user.UsedQuotaSize)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "users", u.Username, "usage"), bytes.NewBuffer([]byte("string")))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.True(t, common.QuotaScans.AddUserQuotaScan(user.Username, ""))
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "users", u.Username, "usage"), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusConflict, rr)
- assert.True(t, common.QuotaScans.RemoveUserQuotaScan(user.Username))
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestUserPermissionsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- user.Permissions = make(map[string][]string)
- user.Permissions["/somedir"] = []string{dataprovider.PermAny}
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- user.Permissions = make(map[string][]string)
- user.Permissions["/"] = []string{dataprovider.PermAny}
- user.Permissions[".."] = []string{dataprovider.PermAny}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- user.Permissions = make(map[string][]string)
- user.Permissions["/"] = []string{dataprovider.PermAny}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.Permissions["/somedir"] = []string{"invalid"}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- delete(user.Permissions, "/somedir")
- user.Permissions["/somedir/.."] = []string{dataprovider.PermAny}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- delete(user.Permissions, "/somedir/..")
- user.Permissions["not_abs_path"] = []string{dataprovider.PermAny}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- delete(user.Permissions, "not_abs_path")
- user.Permissions["/somedir/../otherdir/"] = []string{dataprovider.PermListItems}
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updatedUser)
- assert.NoError(t, err)
- if val, ok := updatedUser.Permissions["/otherdir"]; ok {
- assert.True(t, util.Contains(val, dataprovider.PermListItems))
- assert.Equal(t, 1, len(val))
- } else {
- assert.Fail(t, "expected dir not found in permissions")
- }
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestUpdateUserInvalidJsonMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer([]byte("Invalid json")))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestUpdateUserInvalidParamsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.HomeDir = ""
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- userID := user.ID
- user.ID = 0
- user.CreatedAt = 0
- userAsJSON = getUserAsJSON(t, user)
- req, _ = http.NewRequest(http.MethodPut, path.Join(userPath, user.Username), bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- user.ID = userID
- req, _ = http.NewRequest(http.MethodPut, userPath+"/0", bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestGetAdminsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- asJSON, err := json.Marshal(admin)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, adminPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, _ = http.NewRequest(http.MethodGet, adminPath+"?limit=510&offset=0&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var admins []dataprovider.Admin
- err = render.DecodeJSON(rr.Body, &admins)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(admins), 1)
- firtAdmin := admins[0].Username
- req, _ = http.NewRequest(http.MethodGet, adminPath+"?limit=510&offset=0&order=DESC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admins = nil
- err = render.DecodeJSON(rr.Body, &admins)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(admins), 1)
- assert.NotEqual(t, firtAdmin, admins[0].Username)
- req, _ = http.NewRequest(http.MethodGet, adminPath+"?limit=510&offset=1&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admins = nil
- err = render.DecodeJSON(rr.Body, &admins)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(admins), 1)
- assert.NotEqual(t, firtAdmin, admins[0].Username)
- req, _ = http.NewRequest(http.MethodGet, adminPath+"?limit=a&offset=0&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodGet, adminPath+"?limit=1&offset=aa&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodGet, adminPath+"?limit=1&offset=0&order=ASCa", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(adminPath, admin.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestGetUsersMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodGet, userPath+"?limit=510&offset=0&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var users []dataprovider.User
- err = render.DecodeJSON(rr.Body, &users)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(users), 1)
- req, _ = http.NewRequest(http.MethodGet, userPath+"?limit=aa&offset=0&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodGet, userPath+"?limit=1&offset=a&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodGet, userPath+"?limit=1&offset=0&order=ASCc", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestDeleteUserInvalidParamsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodDelete, userPath+"/0", nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestGetQuotaScansMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, quotaScanPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestStartQuotaScanMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- _, err = os.Stat(user.HomeDir)
- if err == nil {
- err = os.Remove(user.HomeDir)
- assert.NoError(t, err)
- }
- // simulate a duplicate quota scan
- common.QuotaScans.AddUserQuotaScan(user.Username, "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "users", user.Username, "scan"), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusConflict, rr)
- assert.True(t, common.QuotaScans.RemoveUserQuotaScan(user.Username))
- req, _ = http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "users", user.Username, "scan"), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- waitForUsersQuotaScan(t, token)
- _, err = os.Stat(user.HomeDir)
- if err != nil && errors.Is(err, fs.ErrNotExist) {
- err = os.MkdirAll(user.HomeDir, os.ModePerm)
- assert.NoError(t, err)
- }
- req, _ = http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "users", user.Username, "scan"), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- waitForUsersQuotaScan(t, token)
- req, _ = http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "users", user.Username, "scan"), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- waitForUsersQuotaScan(t, token)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestUpdateFolderQuotaUsageMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- mappedPath := filepath.Join(os.TempDir(), "vfolder")
- folderName := filepath.Base(mappedPath)
- f := vfs.BaseVirtualFolder{
- MappedPath: mappedPath,
- Name: folderName,
- }
- usedQuotaFiles := 1
- usedQuotaSize := int64(65535)
- f.UsedQuotaFiles = usedQuotaFiles
- f.UsedQuotaSize = usedQuotaSize
- var folder vfs.BaseVirtualFolder
- folderAsJSON, err := json.Marshal(f)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, folderPath, bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "folders", folder.Name, "usage"), bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var folderGet vfs.BaseVirtualFolder
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folderGet)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, folderGet.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize, folderGet.UsedQuotaSize)
- // now update only quota size
- f.UsedQuotaFiles = 0
- folderAsJSON, err = json.Marshal(f)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "folders", folder.Name, "usage")+"?mode=add",
- bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- folderGet = vfs.BaseVirtualFolder{}
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folderGet)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles, folderGet.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize*2, folderGet.UsedQuotaSize)
- // now update only quota files
- f.UsedQuotaSize = 0
- f.UsedQuotaFiles = 1
- folderAsJSON, err = json.Marshal(f)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "folders", folder.Name, "usage")+"?mode=add",
- bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- folderGet = vfs.BaseVirtualFolder{}
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folderGet)
- assert.NoError(t, err)
- assert.Equal(t, usedQuotaFiles*2, folderGet.UsedQuotaFiles)
- assert.Equal(t, usedQuotaSize*2, folderGet.UsedQuotaSize)
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "folders", folder.Name, "usage"),
- bytes.NewBuffer([]byte("not a json")))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.True(t, common.QuotaScans.AddVFolderQuotaScan(folderName))
- req, _ = http.NewRequest(http.MethodPut, path.Join(quotasBasePath, "folders", folder.Name, "usage"),
- bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusConflict, rr)
- assert.True(t, common.QuotaScans.RemoveVFolderQuotaScan(folderName))
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestStartFolderQuotaScanMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- mappedPath := filepath.Join(os.TempDir(), "vfolder")
- folderName := filepath.Base(mappedPath)
- folder := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedPath,
- }
- folderAsJSON, err := json.Marshal(folder)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, folderPath, bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = os.Stat(mappedPath)
- if err == nil {
- err = os.Remove(mappedPath)
- assert.NoError(t, err)
- }
- // simulate a duplicate quota scan
- common.QuotaScans.AddVFolderQuotaScan(folderName)
- req, _ = http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "folders", folder.Name, "scan"), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusConflict, rr)
- assert.True(t, common.QuotaScans.RemoveVFolderQuotaScan(folderName))
- // and now a real quota scan
- _, err = os.Stat(mappedPath)
- if err != nil && errors.Is(err, fs.ErrNotExist) {
- err = os.MkdirAll(mappedPath, os.ModePerm)
- assert.NoError(t, err)
- }
- req, _ = http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "folders", folder.Name, "scan"), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- waitForFoldersQuotaScanPath(t, token)
- // cleanup
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.RemoveAll(folderPath)
- assert.NoError(t, err)
- err = os.RemoveAll(mappedPath)
- assert.NoError(t, err)
- }
- func TestStartQuotaScanNonExistentUserMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- req, _ := http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "users", user.Username, "scan"), nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestStartQuotaScanNonExistentFolderMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- folder := vfs.BaseVirtualFolder{
- MappedPath: os.TempDir(),
- Name: "afolder",
- }
- req, _ := http.NewRequest(http.MethodPost, path.Join(quotasBasePath, "folders", folder.Name, "scan"), nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestGetFoldersMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- mappedPath := filepath.Join(os.TempDir(), "vfolder")
- folderName := filepath.Base(mappedPath)
- folder := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedPath,
- }
- folderAsJSON, err := json.Marshal(folder)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, folderPath, bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- var folders []vfs.BaseVirtualFolder
- url, err := url.Parse(folderPath + "?limit=510&offset=0&order=DESC")
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodGet, url.String(), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folders)
- assert.NoError(t, err)
- assert.GreaterOrEqual(t, len(folders), 1)
- req, _ = http.NewRequest(http.MethodGet, folderPath+"?limit=a&offset=0&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodGet, folderPath+"?limit=1&offset=a&order=ASC", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodGet, folderPath+"?limit=1&offset=0&order=ASCV", nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestGetVersionMock(t *testing.T) {
- req, _ := http.NewRequest(http.MethodGet, versionPath, nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodGet, versionPath, nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, versionPath, nil)
- setBearerForReq(req, "abcde")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- }
- func TestGetConnectionsMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, activeConnectionsPath, nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestGetStatusMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestDeleteActiveConnectionMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodDelete, activeConnectionsPath+"/connectionID", nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req.Header.Set(dataprovider.NodeTokenHeader, "Bearer abc")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "the provided token cannot be authenticated")
- req, err = http.NewRequest(http.MethodDelete, activeConnectionsPath+"/connectionID?node=node1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestNotFoundMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, "/non/existing/path", nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestMethodNotAllowedMock(t *testing.T) {
- req, _ := http.NewRequest(http.MethodPost, activeConnectionsPath, nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusMethodNotAllowed, rr)
- }
- func TestHealthCheck(t *testing.T) {
- req, _ := http.NewRequest(http.MethodGet, healthzPath, nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "ok", rr.Body.String())
- }
- func TestGetWebRootMock(t *testing.T) {
- req, _ := http.NewRequest(http.MethodGet, "/", nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webBasePath, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webBasePathAdmin, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webBasePathClient, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- }
- func TestWebNotFoundURI(t *testing.T) {
- urlString := httpBaseURL + webBasePath + "/a"
- req, err := http.NewRequest(http.MethodGet, urlString, nil)
- assert.NoError(t, err)
- resp, err := httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusNotFound, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, urlString, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, "invalid token")
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusNotFound, resp.StatusCode)
- urlString = httpBaseURL + webBasePathClient + "/a"
- req, err = http.NewRequest(http.MethodGet, urlString, nil)
- assert.NoError(t, err)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusNotFound, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, urlString, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, "invalid client token")
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusNotFound, resp.StatusCode)
- }
- func TestLogout(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, logoutPath, nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "Your token is no longer valid")
- }
- func TestDefenderAPIInvalidIDMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, path.Join(defenderHosts, "abc"), nil) // not hex id
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "invalid host id")
- }
- func TestTokenHeaderCookie(t *testing.T) {
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setJWTCookieForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "no token found")
- req, _ = http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- setBearerForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestTokenAudience(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "Your token audience is not valid")
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- setJWTCookieForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- }
- func TestWebAPILoginMock(t *testing.T) {
- _, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername+"1", defaultPassword)
- assert.Error(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword+"1")
- assert.Error(t, err)
- apiToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- // a web token is not valid for API usage
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "Your token audience is not valid")
- req, err = http.NewRequest(http.MethodGet, userDirsPath+"/?path=%2F", nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // API token is not valid for web usage
- req, _ = http.NewRequest(http.MethodGet, webClientProfilePath, nil)
- setJWTCookieForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webClientProfilePath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebClientLoginMock(t *testing.T) {
- _, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- // a web token is not valid for API or WebAdmin usage
- req, _ := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "Your token audience is not valid")
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- // bearer should not work
- req, _ = http.NewRequest(http.MethodGet, webClientProfilePath, nil)
- setBearerForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webClientPingPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setBearerForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- // now try to render client pages
- req, _ = http.NewRequest(http.MethodGet, webClientProfilePath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientPingPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now logout
- req, _ = http.NewRequest(http.MethodGet, webClientLogoutPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webClientProfilePath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webClientPingPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- // get a new token and use it after removing the user
- webToken, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- apiUserToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodGet, webClientProfilePath, nil)
- setJWTCookieForReq(req, webToken)
- req.RemoteAddr = defaultRemoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorGetUser)
- req, _ = http.NewRequest(http.MethodGet, webClientDirsPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorDirListUser)
- form := make(url.Values)
- form.Set("files", `[]`)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorGetUser)
- req, _ = http.NewRequest(http.MethodGet, userDirsPath, nil)
- setBearerForReq(req, apiUserToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to retrieve your user")
- req, _ = http.NewRequest(http.MethodGet, userFilesPath, nil)
- setBearerForReq(req, apiUserToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to retrieve your user")
- req, _ = http.NewRequest(http.MethodPost, userStreamZipPath, bytes.NewBuffer([]byte(`{}`)))
- setBearerForReq(req, apiUserToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to retrieve your user")
- form = make(url.Values)
- form.Set("public_keys", testPubKey)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- }
- func TestWebClientLoginErrorsMock(t *testing.T) {
- form := getLoginForm("", "", "")
- req, _ := http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr := executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- form = getLoginForm(defaultUsername, defaultPassword, "")
- req, _ = http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- }
- func TestWebClientMaxConnections(t *testing.T) {
- oldValue := common.Config.MaxTotalConnections
- common.Config.MaxTotalConnections = 1
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now add a fake connection
- fs := vfs.NewOsFs("id", os.TempDir(), "", nil)
- connection := &httpd.Connection{
- BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolHTTP, "", "", user),
- }
- err = common.Connections.Add(connection)
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), common.ErrConnectionDenied.Error())
- common.Connections.Remove(connection.GetID())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 0)
- common.Config.MaxTotalConnections = oldValue
- }
- func TestTokenInvalidIPAddress(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- req.RemoteAddr = "1.1.1.2"
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- apiToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath+"/?path=%2F", nil)
- assert.NoError(t, err)
- req.RemoteAddr = "2.2.2.2"
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "Your token is not valid")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestDefender(t *testing.T) {
- oldConfig := config.GetCommonConfig()
- cfg := config.GetCommonConfig()
- cfg.DefenderConfig.Enabled = true
- cfg.DefenderConfig.Threshold = 3
- cfg.DefenderConfig.ScoreLimitExceeded = 2
- err := common.Initialize(cfg, 0)
- assert.NoError(t, err)
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- remoteAddr := "172.16.5.6:9876"
- webAdminToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServerWithAddr(defaultUsername, defaultPassword, remoteAddr)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- req.RemoteAddr = remoteAddr
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- for i := 0; i < 3; i++ {
- _, err = getJWTWebClientTokenFromTestServerWithAddr(defaultUsername, "wrong pwd", remoteAddr)
- assert.Error(t, err)
- }
- _, err = getJWTWebClientTokenFromTestServerWithAddr(defaultUsername, defaultPassword, remoteAddr)
- assert.Error(t, err)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- req.RequestURI = webClientFilesPath
- setJWTCookieForReq(req, webToken)
- req.RemoteAddr = remoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorIPForbidden)
- req, _ = http.NewRequest(http.MethodGet, webUsersPath, nil)
- req.RequestURI = webUsersPath
- setJWTCookieForReq(req, webAdminToken)
- req.RemoteAddr = remoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorIPForbidden)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- req.Header.Set("X-Real-IP", "127.0.0.1:2345")
- setJWTCookieForReq(req, webToken)
- req.RemoteAddr = remoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "your IP address is blocked")
- // requests for static files should be always allowed
- req, err = http.NewRequest(http.MethodGet, "/static/favicon.ico", nil)
- assert.NoError(t, err)
- req.RemoteAddr = remoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, "/.well-known/acme-challenge/foo", nil)
- assert.NoError(t, err)
- req.RemoteAddr = remoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- err = common.Initialize(oldConfig, 0)
- assert.NoError(t, err)
- }
- func TestPostConnectHook(t *testing.T) {
- if runtime.GOOS == osWindows {
- t.Skip("this test is not available on Windows")
- }
- common.Config.PostConnectHook = postConnectPath
- u := getTestUser()
- u.Filters.AllowAPIKeyAuth = true
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- apiKey, _, err := httpdtest.AddAPIKey(dataprovider.APIKey{
- Name: "name",
- Scope: dataprovider.APIKeyScopeUser,
- User: user.Username,
- }, http.StatusCreated)
- assert.NoError(t, err)
- err = os.WriteFile(postConnectPath, getExitCodeScriptContent(0), os.ModePerm)
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.WriteFile(postConnectPath, getExitCodeScriptContent(1), os.ModePerm)
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- common.Config.PostConnectHook = ""
- }
- func TestMaxSessions(t *testing.T) {
- u := getTestUser()
- u.MaxSessions = 1
- u.Email = "user@session.com"
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- apiToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- // now add a fake connection
- fs := vfs.NewOsFs("id", os.TempDir(), "", nil)
- connection := &httpd.Connection{
- BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolHTTP, "", "", user),
- }
- err = common.Connections.Add(connection)
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- // try an user API call
- req, err := http.NewRequest(http.MethodGet, userDirsPath+"/?path=%2F", nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- // web client requests
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("files", `[]`)
- req, err = http.NewRequest(http.MethodPost, webClientDownloadZipPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- req, err = http.NewRequest(http.MethodGet, webClientDirsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorDirList429)
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath+"?path=p", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path=file", nil) //nolint:goconst
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=file", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- // test reset password
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err = smtpCfg.Initialize(configDir, true)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- lastResetCode = ""
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("password", defaultPassword)
- form.Set("confirm_password", defaultPassword)
- form.Set("code", lastResetCode)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- common.Connections.Remove(connection.GetID())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 0)
- }
- func TestWebConfigsMock(t *testing.T) {
- acmeConfig := config.GetACMEConfig()
- acmeConfig.CertsPath = filepath.Clean(os.TempDir())
- err := acme.Initialize(acmeConfig, configDir, true)
- require.NoError(t, err)
- err = dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webConfigsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form := make(url.Values)
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- // parse form error
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webConfigsPath+"?p=p%C3%AO%GH", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // save SFTP configs
- form.Set("sftp_host_key_algos", ssh.KeyAlgoRSA)
- form.Add("sftp_host_key_algos", ssh.InsecureCertAlgoDSAv01)
- form.Set("sftp_pub_key_algos", ssh.InsecureKeyAlgoDSA)
- form.Set("form_action", "sftp_submit")
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message) // invalid algo
- form.Set("sftp_host_key_algos", ssh.KeyAlgoRSA)
- form.Add("sftp_host_key_algos", ssh.CertAlgoRSAv01)
- form.Set("sftp_pub_key_algos", ssh.InsecureKeyAlgoDSA)
- form.Set("sftp_kex_algos", "diffie-hellman-group18-sha512")
- form.Add("sftp_kex_algos", ssh.KeyExchangeDH16SHA512)
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nConfigsOK)
- // check SFTP configs
- configs, err := dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.HostKeyAlgos, 1)
- assert.Contains(t, configs.SFTPD.HostKeyAlgos, ssh.KeyAlgoRSA)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 1)
- assert.Contains(t, configs.SFTPD.PublicKeyAlgos, ssh.InsecureKeyAlgoDSA)
- assert.Len(t, configs.SFTPD.KexAlgorithms, 1)
- assert.Contains(t, configs.SFTPD.KexAlgorithms, ssh.KeyExchangeDH16SHA512)
- // invalid form action
- form.Set("form_action", "")
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- // test SMTP configs
- form.Set("form_action", "smtp_submit")
- form.Set("smtp_host", "mail.example.net")
- form.Set("smtp_from", "Example <info@example.net>")
- form.Set("smtp_username", defaultUsername)
- form.Set("smtp_password", defaultPassword)
- form.Set("smtp_domain", "localdomain")
- form.Set("smtp_auth", "100")
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message) // invalid smtp_auth
- // set valid parameters
- form.Set("smtp_port", "a") // converted to 587
- form.Set("smtp_auth", "1")
- form.Set("smtp_encryption", "2")
- form.Set("smtp_debug", "checked")
- form.Set("smtp_oauth2_provider", "1")
- form.Set("smtp_oauth2_client_id", "123")
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nConfigsOK)
- // check
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.HostKeyAlgos, 1)
- assert.Contains(t, configs.SFTPD.HostKeyAlgos, ssh.KeyAlgoRSA)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 1)
- assert.Contains(t, configs.SFTPD.PublicKeyAlgos, ssh.InsecureKeyAlgoDSA)
- assert.Equal(t, "mail.example.net", configs.SMTP.Host)
- assert.Equal(t, 587, configs.SMTP.Port)
- assert.Equal(t, "Example <info@example.net>", configs.SMTP.From)
- assert.Equal(t, defaultUsername, configs.SMTP.User)
- assert.Equal(t, 1, configs.SMTP.Debug)
- assert.Equal(t, "", configs.SMTP.OAuth2.ClientID)
- err = configs.SMTP.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, defaultPassword, configs.SMTP.Password.GetPayload())
- assert.Equal(t, 1, configs.SMTP.AuthType)
- assert.Equal(t, 2, configs.SMTP.Encryption)
- assert.Equal(t, "localdomain", configs.SMTP.Domain)
- // set a redacted password, the current password must be preserved
- form.Set("smtp_password", redactedSecret)
- form.Set("smtp_auth", "")
- configs.SMTP.AuthType = 0 // empty will be converted to 0
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nConfigsOK)
- updatedConfigs, err := dataprovider.GetConfigs()
- assert.NoError(t, err)
- encryptedPayload := updatedConfigs.SMTP.Password.GetPayload()
- secretKey := updatedConfigs.SMTP.Password.GetKey()
- err = updatedConfigs.SMTP.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, configs.SFTPD, updatedConfigs.SFTPD)
- assert.Equal(t, configs.SMTP, updatedConfigs.SMTP)
- // now set an undecryptable password
- updatedConfigs.SMTP.Password = kms.NewSecret(sdkkms.SecretStatusSecretBox, encryptedPayload, secretKey, "")
- err = dataprovider.UpdateConfigs(&updatedConfigs, "", "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nConfigsOK)
- form.Set("form_action", "acme_submit")
- form.Set("acme_port", "") // on error will be set to 80
- form.Set("acme_protocols", "1")
- form.Add("acme_protocols", "2")
- form.Add("acme_protocols", "3")
- form.Set("acme_domain", "example.com")
- // no email set, validation will fail
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidEmail)
- form.Set("acme_domain", "")
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nConfigsOK)
- // check
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.HostKeyAlgos, 1)
- assert.Contains(t, configs.SFTPD.HostKeyAlgos, ssh.KeyAlgoRSA)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 1)
- assert.Contains(t, configs.SFTPD.PublicKeyAlgos, ssh.InsecureKeyAlgoDSA)
- assert.Equal(t, 80, configs.ACME.HTTP01Challenge.Port)
- assert.Equal(t, 7, configs.ACME.Protocols)
- assert.Empty(t, configs.ACME.Domain)
- assert.Empty(t, configs.ACME.Email)
- assert.True(t, configs.ACME.HasProtocol(common.ProtocolFTP))
- assert.True(t, configs.ACME.HasProtocol(common.ProtocolWebDAV))
- assert.True(t, configs.ACME.HasProtocol(common.ProtocolHTTP))
- // create certificate files, so no real ACME call is done
- domain := "acme.example.com"
- crtPath := filepath.Join(acmeConfig.CertsPath, util.SanitizeDomain(domain)+".crt")
- keyPath := filepath.Join(acmeConfig.CertsPath, util.SanitizeDomain(domain)+".key")
- err = os.WriteFile(crtPath, nil, 0666)
- assert.NoError(t, err)
- err = os.WriteFile(keyPath, nil, 0666)
- assert.NoError(t, err)
- form.Set("acme_port", "402")
- form.Set("acme_protocols", "1")
- form.Add("acme_protocols", "1000")
- form.Set("acme_domain", domain)
- form.Set("acme_email", "email@example.com")
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nConfigsOK)
- configs, err = dataprovider.GetConfigs()
- assert.NoError(t, err)
- assert.Len(t, configs.SFTPD.HostKeyAlgos, 1)
- assert.Len(t, configs.SFTPD.PublicKeyAlgos, 1)
- assert.Equal(t, 402, configs.ACME.HTTP01Challenge.Port)
- assert.Equal(t, 1, configs.ACME.Protocols)
- assert.Equal(t, domain, configs.ACME.Domain)
- assert.Equal(t, "email@example.com", configs.ACME.Email)
- assert.False(t, configs.ACME.HasProtocol(common.ProtocolFTP))
- assert.False(t, configs.ACME.HasProtocol(common.ProtocolWebDAV))
- assert.True(t, configs.ACME.HasProtocol(common.ProtocolHTTP))
- err = os.Remove(crtPath)
- assert.NoError(t, err)
- err = os.Remove(keyPath)
- assert.NoError(t, err)
- err = dataprovider.UpdateConfigs(nil, "", "", "")
- assert.NoError(t, err)
- }
- func TestSFTPLoopError(t *testing.T) {
- user1 := getTestUser()
- user2 := getTestUser()
- user1.Username += "1"
- user1.Email = "user1@test.com"
- user2.Username += "2"
- user1.FsConfig = vfs.Filesystem{
- Provider: sdk.SFTPFilesystemProvider,
- SFTPConfig: vfs.SFTPFsConfig{
- BaseSFTPFsConfig: sdk.BaseSFTPFsConfig{
- Endpoint: sftpServerAddr,
- Username: user2.Username,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- },
- }
- user2.FsConfig.Provider = sdk.SFTPFilesystemProvider
- user2.FsConfig.SFTPConfig = vfs.SFTPFsConfig{
- BaseSFTPFsConfig: sdk.BaseSFTPFsConfig{
- Endpoint: sftpServerAddr,
- Username: user1.Username,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- }
- user1, resp, err := httpdtest.AddUser(user1, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- user2, resp, err = httpdtest.AddUser(user2, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- // test reset password
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err = smtpCfg.Initialize(configDir, true)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user1.Username)
- lastResetCode = ""
- req, err := http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr := executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("password", defaultPassword)
- form.Set("confirm_password", defaultPassword)
- form.Set("code", lastResetCode)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorLoginAfterReset)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user1.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user2, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user2.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestLoginInvalidFs(t *testing.T) {
- u := getTestUser()
- u.Filters.AllowAPIKeyAuth = true
- u.FsConfig.Provider = sdk.GCSFilesystemProvider
- u.FsConfig.GCSConfig.Bucket = "test"
- u.FsConfig.GCSConfig.UploadPartSize = 1
- u.FsConfig.GCSConfig.UploadPartMaxTime = 10
- u.FsConfig.GCSConfig.Credentials = kms.NewPlainSecret("invalid JSON for credentials")
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- apiKey, _, err := httpdtest.AddAPIKey(dataprovider.APIKey{
- Name: "testk",
- Scope: dataprovider.APIKeyScopeUser,
- User: u.Username,
- }, http.StatusCreated)
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- _, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebClientChangePwd(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webChangeClientPwdPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form := make(url.Values)
- form.Set("current_password", defaultPassword)
- form.Set("new_password1", defaultPassword)
- form.Set("new_password2", defaultPassword)
- // no csrf token
- req, err = http.NewRequest(http.MethodPost, webChangeClientPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webChangeClientPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdNoDifferent)
- form.Set("current_password", defaultPassword+"2")
- form.Set("new_password1", defaultPassword+"1")
- form.Set("new_password2", defaultPassword+"1")
- req, _ = http.NewRequest(http.MethodPost, webChangeClientPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdCurrentNoMatch)
- form.Set("current_password", defaultPassword)
- form.Set("new_password1", defaultPassword+"1")
- form.Set("new_password2", defaultPassword+"1")
- req, _ = http.NewRequest(http.MethodPost, webChangeClientPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webClientLoginPath, rr.Header().Get("Location"))
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.Error(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername+"1", defaultPassword+"1")
- assert.Error(t, err)
- _, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword+"1")
- assert.NoError(t, err)
- // remove the change password permission
- user.Filters.WebClient = []string{sdk.WebClientPasswordChangeDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- assert.Len(t, user.Filters.WebClient, 1)
- assert.Contains(t, user.Filters.WebClient, sdk.WebClientPasswordChangeDisabled)
- webToken, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword+"1")
- assert.NoError(t, err)
- form.Set("current_password", defaultPassword+"1")
- form.Set("new_password1", defaultPassword)
- form.Set("new_password2", defaultPassword)
- req, _ = http.NewRequest(http.MethodPost, webChangeClientPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestPreDownloadHook(t *testing.T) {
- if runtime.GOOS == osWindows {
- t.Skip("this test is not available on Windows")
- }
- oldExecuteOn := common.Config.Actions.ExecuteOn
- oldHook := common.Config.Actions.Hook
- common.Config.Actions.ExecuteOn = []string{common.OperationPreDownload}
- common.Config.Actions.Hook = preActionPath
- u := getTestUser()
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- err = os.WriteFile(preActionPath, getExitCodeScriptContent(0), os.ModePerm)
- assert.NoError(t, err)
- testFileName := "testfile"
- testFileContents := []byte("file contents")
- err = os.MkdirAll(filepath.Join(user.GetHomeDir()), os.ModePerm)
- assert.NoError(t, err)
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), testFileName), testFileContents, os.ModePerm)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil) //nolint:goconst
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, testFileContents, rr.Body.Bytes())
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, testFileContents, rr.Body.Bytes())
- err = os.WriteFile(preActionPath, getExitCodeScriptContent(1), os.ModePerm)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError403Message)
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "permission denied")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- common.Config.Actions.ExecuteOn = oldExecuteOn
- common.Config.Actions.Hook = oldHook
- }
- func TestPreUploadHook(t *testing.T) {
- if runtime.GOOS == osWindows {
- t.Skip("this test is not available on Windows")
- }
- oldExecuteOn := common.Config.Actions.ExecuteOn
- oldHook := common.Config.Actions.Hook
- common.Config.Actions.ExecuteOn = []string{common.OperationPreUpload}
- common.Config.Actions.Hook = preActionPath
- u := getTestUser()
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- err = os.WriteFile(preActionPath, getExitCodeScriptContent(0), os.ModePerm)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "filepre")
- assert.NoError(t, err)
- _, err = part.Write([]byte("file content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=filepre",
- bytes.NewBuffer([]byte("single upload content")))
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = os.WriteFile(preActionPath, getExitCodeScriptContent(1), os.ModePerm)
- assert.NoError(t, err)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=filepre",
- bytes.NewBuffer([]byte("single upload content")))
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- common.Config.Actions.ExecuteOn = oldExecuteOn
- common.Config.Actions.Hook = oldHook
- }
- func TestShareUsage(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "testfile.dat"
- testFileSize := int64(65536)
- testFilePath := filepath.Join(user.GetHomeDir(), testFileName)
- err = createTestFile(testFilePath, testFileSize)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- Password: defaultPassword,
- MaxTokens: 2,
- ExpiresAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(1 * time.Second)),
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/unknownid", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- req.SetBasicAuth(defaultUsername, "wrong password")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- time.Sleep(2 * time.Second)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"_mod", nil)
- assert.NoError(t, err)
- req.RequestURI = webClientPubSharesPath + "/" + objectID + "_mod"
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- share.ExpiresAt = 0
- jsonReq := make(map[string]any)
- jsonReq["name"] = share.Name
- jsonReq["scope"] = share.Scope
- jsonReq["paths"] = share.Paths
- jsonReq["password"] = share.Password
- jsonReq["max_tokens"] = share.MaxTokens
- jsonReq["expires_at"] = share.ExpiresAt
- asJSON, err = json.Marshal(jsonReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userSharesPath+"/"+objectID, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "invalid share scope")
- share.MaxTokens = 3
- share.Scope = dataprovider.ShareScopeWrite
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userSharesPath+"/"+objectID, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part1, err := writer.CreateFormFile("filenames", "file1.txt")
- assert.NoError(t, err)
- _, err = part1.Write([]byte("file1 content"))
- assert.NoError(t, err)
- part2, err := writer.CreateFormFile("filenames", "file2.txt")
- assert.NoError(t, err)
- _, err = part2.Write([]byte("file2 content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Unable to parse multipart form")
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- // set the proper content type
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Allowed usage exceeded")
- share.MaxTokens = 6
- share.Scope = dataprovider.ShareScopeWrite
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, userSharesPath+"/"+objectID, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientPubSharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- share, err = dataprovider.ShareExists(objectID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, 6, share.UsedTokens)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- share.MaxTokens = 0
- err = dataprovider.UpdateShare(&share, user.Username, "", "")
- assert.NoError(t, err)
- user.Permissions["/"] = []string{dataprovider.PermListItems, dataprovider.PermDownload}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "permission denied")
- user.Permissions["/"] = []string{dataprovider.PermAny}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- body = new(bytes.Buffer)
- writer = multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filename", "file1.txt")
- assert.NoError(t, err)
- _, err = part.Write([]byte("file content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader = bytes.NewReader(body.Bytes())
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "No files uploaded!")
- user.Filters.WebClient = []string{sdk.WebClientSharesDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- user.Filters.WebClient = []string{sdk.WebClientShareNoPasswordDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- share.Password = ""
- err = dataprovider.UpdateShare(&share, user.Username, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "sharing without a password was disabled")
- user.Filters.WebClient = []string{sdk.WebClientInfoChangeDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- share.Scope = dataprovider.ShareScopeRead
- share.Paths = []string{"/missing1", "/missing2"}
- err = dataprovider.UpdateShare(&share, user.Username, "", "")
- assert.NoError(t, err)
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- share, err = dataprovider.ShareExists(objectID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, 6, share.UsedTokens)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }()
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- executeRequest(req)
- }
- func TestShareMaxExpiration(t *testing.T) {
- u := getTestUser()
- u.Filters.MaxSharesExpiration = 5
- u.Filters.DefaultSharesExpiration = 10
- _, resp, err := httpdtest.AddUser(u, http.StatusBadRequest)
- assert.NoError(t, err)
- assert.Contains(t, string(resp), "must be less than or equal to max shares expiration")
- u.Filters.DefaultSharesExpiration = 0
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webClientToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- s := dataprovider.Share{
- Name: "test share",
- Scope: dataprovider.ShareScopeRead,
- Password: defaultPassword,
- Paths: []string{"/"},
- ExpiresAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(24 * time.Hour * time.Duration(u.Filters.MaxSharesExpiration+2))),
- }
- asJSON, err := json.Marshal(s)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "share must expire before")
- req, err = http.NewRequest(http.MethodPut, path.Join(userSharesPath, "shareID"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // expiresAt is mandatory
- s.ExpiresAt = 0
- asJSON, err = json.Marshal(s)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "share must expire before")
- s.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(2 * time.Hour))
- asJSON, err = json.Marshal(s)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- shareID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, shareID)
- s.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(24 * time.Hour * time.Duration(u.Filters.MaxSharesExpiration+2)))
- asJSON, err = json.Marshal(s)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, path.Join(userSharesPath, shareID), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "share must expire before")
- form := make(url.Values)
- form.Set("name", s.Name)
- form.Set("scope", strconv.Itoa(int(s.Scope)))
- form.Set("max_tokens", "0")
- form.Set("paths[0][path]", "/")
- form.Set("expiration_date", time.Now().Add(24*time.Hour*time.Duration(u.Filters.MaxSharesExpiration+2)).Format("2006-01-02 15:04:05"))
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webClientToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareExpirationOutOfRange)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientSharePath, shareID), bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webClientToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareExpirationOutOfRange)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webClientToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorGetUser)
- }
- func TestWebClientShareCredentials(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- shareRead := dataprovider.Share{
- Name: "test share read",
- Scope: dataprovider.ShareScopeRead,
- Password: defaultPassword,
- Paths: []string{"/"},
- }
- shareWrite := dataprovider.Share{
- Name: "test share write",
- Scope: dataprovider.ShareScopeReadWrite,
- Password: defaultPassword,
- Paths: []string{"/"},
- }
- asJSON, err := json.Marshal(shareRead)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- shareReadID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, shareReadID)
- asJSON, err = json.Marshal(shareWrite)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- shareWriteID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, shareWriteID)
- uri := path.Join(webClientPubSharesPath, shareReadID, "browse")
- req, err = http.NewRequest(http.MethodGet, uri, nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- location := rr.Header().Get("Location")
- assert.Contains(t, location, url.QueryEscape(uri))
- // get the login form
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now set the user token, it is not valid for the share
- req, err = http.NewRequest(http.MethodGet, uri, nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- // get a share token
- form := make(url.Values)
- form.Set("share_password", defaultPassword)
- loginURI := path.Join(webClientPubSharesPath, shareReadID, "login")
- req, err = http.NewRequest(http.MethodPost, loginURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- // set the CSRF token
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, loginURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nShareLoginOK)
- cookie := rr.Header().Get("Set-Cookie")
- cookie = strings.TrimPrefix(cookie, "jwt=")
- assert.NotEmpty(t, cookie)
- req, err = http.NewRequest(http.MethodGet, uri, nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // get the download page
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, shareReadID, "download?a=b"), nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // get the download page for a missing share
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, "invalidshareid", "download"), nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // the same cookie will not work for the other share
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, shareWriteID, "browse"), nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- setJWTCookieForReq(req, cookie)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- // IP address does not match
- req, err = http.NewRequest(http.MethodGet, uri, nil)
- assert.NoError(t, err)
- req.RequestURI = uri
- setJWTCookieForReq(req, cookie)
- req.RemoteAddr = "1.2.3.4:1234"
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- // try to login with invalid credentials
- form.Set("share_password", "")
- req, err = http.NewRequest(http.MethodPost, loginURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- // login with the next param set
- form.Set("share_password", defaultPassword)
- nextURI := path.Join(webClientPubSharesPath, shareReadID, "browse")
- loginURI = path.Join(webClientPubSharesPath, shareReadID, fmt.Sprintf("login?next=%s", url.QueryEscape(nextURI)))
- req, err = http.NewRequest(http.MethodPost, loginURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, nextURI, rr.Header().Get("Location"))
- // try to login to a missing share
- loginURI = path.Join(webClientPubSharesPath, "missing", "login")
- req, err = http.NewRequest(http.MethodPost, loginURI, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestShareMaxSessions(t *testing.T) {
- u := getTestUser()
- u.MaxSessions = 1
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share max sessions read",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID+"/dirs", nil) //nolint:goconst
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // add a fake connection
- fs := vfs.NewOsFs("id", os.TempDir(), "", nil)
- connection := &httpd.Connection{
- BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolHTTP, "", "", user),
- }
- err = common.Connections.Add(connection)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID+"/dirs", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"/dirs", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=file.pdf"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"/browse", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- req, err = http.NewRequest(http.MethodPost, webClientPubSharesPath+"/"+objectID+"/browse/exist", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "invalid share scope")
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID+"/files?path=afile", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- form := make(url.Values)
- form.Set("files", `[]`)
- req, err = http.NewRequest(http.MethodPost, webClientPubSharesPath+"/"+objectID+"/partial",
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError429Message)
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- req, err = http.NewRequest(http.MethodDelete, userSharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now test a write share
- share = dataprovider.Share{
- Name: "test share max sessions write",
- Scope: dataprovider.ShareScopeWrite,
- Paths: []string{"/"},
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "file.txt"), bytes.NewBuffer([]byte("content")))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part1, err := writer.CreateFormFile("filenames", "file1.txt")
- assert.NoError(t, err)
- _, err = part1.Write([]byte("file1 content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- share = dataprovider.Share{
- Name: "test share max sessions read/write",
- Scope: dataprovider.ShareScopeReadWrite,
- Paths: []string{"/"},
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodPost, webClientPubSharesPath+"/"+objectID+"/browse/exist", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusTooManyRequests, rr)
- assert.Contains(t, rr.Body.String(), "too many open sessions")
- common.Connections.Remove(connection.GetID())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- assert.Len(t, common.Connections.GetStats(""), 0)
- }
- func TestShareUploadSingle(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share",
- Scope: dataprovider.ShareScopeWrite,
- Paths: []string{"/"},
- Password: defaultPassword,
- MaxTokens: 0,
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- content := []byte("shared file content")
- modTime := time.Now().Add(-12 * time.Hour)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "file.txt"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- req.Header.Set("X-SFTPGO-MTIME", strconv.FormatInt(util.GetTimeAsMsSinceEpoch(modTime), 10))
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- info, err := os.Stat(filepath.Join(user.GetHomeDir(), "file.txt"))
- if assert.NoError(t, err) {
- assert.InDelta(t, util.GetTimeAsMsSinceEpoch(modTime), util.GetTimeAsMsSinceEpoch(info.ModTime()), float64(1000))
- }
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "upload"), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "file.txt"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- info, err = os.Stat(filepath.Join(user.GetHomeDir(), "file.txt"))
- if assert.NoError(t, err) {
- assert.InDelta(t, util.GetTimeAsMsSinceEpoch(time.Now()), util.GetTimeAsMsSinceEpoch(info.ModTime()), float64(3000))
- }
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "dir", "file.dat"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "%2F"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- err = os.MkdirAll(filepath.Join(user.GetHomeDir(), "dir"), os.ModePerm)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "dir"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- // only uploads to the share root dir are allowed
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "dir", "file.dat"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- share, err = dataprovider.ShareExists(objectID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, 2, share.UsedTokens)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, "file1.txt"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestShareReadWrite(t *testing.T) {
- u := getTestUser()
- u.Filters.StartDirectory = path.Join("/start", "dir")
- u.Permissions["/start/dir/limited"] = []string{dataprovider.PermListItems}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- testFileName := "test.txt"
- testSubDirs := "/sub/dir"
- share := dataprovider.Share{
- Name: "test share rw",
- Scope: dataprovider.ShareScopeReadWrite,
- Paths: []string{user.Filters.StartDirectory},
- Password: defaultPassword,
- MaxTokens: 0,
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- filesToCheck := make(map[string]any)
- filesToCheck["files"] = []string{testFileName}
- asJSON, err = json.Marshal(filesToCheck)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "/browse/exist?path=%2F"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var fileList []any
- err = json.Unmarshal(rr.Body.Bytes(), &fileList)
- assert.NoError(t, err)
- assert.Len(t, fileList, 0)
- content := []byte("shared rw content")
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID, testFileName), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- assert.FileExists(t, filepath.Join(user.GetHomeDir(), user.Filters.StartDirectory, testFileName))
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID)+"/"+url.PathEscape(path.Join(testSubDirs, testFileName)), bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID)+"/"+url.PathEscape(path.Join(testSubDirs, testFileName))+"?mkdir_parents=true",
- bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- assert.FileExists(t, filepath.Join(user.GetHomeDir(), user.Filters.StartDirectory, testSubDirs, testFileName))
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID)+"/"+url.PathEscape(path.Join("limited", "sub", testFileName))+"?mkdir_parents=true",
- bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "/browse/exist?path=%2F"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- fileList = nil
- err = json.Unmarshal(rr.Body.Bytes(), &fileList)
- assert.NoError(t, err)
- assert.Len(t, fileList, 1)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path="+testFileName), nil) //nolint:goconst
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contentDisposition := rr.Header().Get("Content-Disposition")
- assert.NotEmpty(t, contentDisposition)
- form := make(url.Values)
- form.Set("files", fmt.Sprintf(`["%s"]`, testFileName))
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contentDisposition = rr.Header().Get("Content-Disposition")
- assert.NotEmpty(t, contentDisposition)
- assert.Equal(t, "application/zip", rr.Header().Get("Content-Type"))
- // parse form error
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial?path=p%C3%AO%GK"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // invalid files list
- form.Set("files", fmt.Sprintf(`[%s]`, testFileName))
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- // missing directory
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial?path=missing"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID)+"/"+url.PathEscape("../"+testFileName),
- bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Uploading outside the share is not allowed")
- req, err = http.NewRequest(http.MethodPost, path.Join(sharesPath, objectID)+"/"+url.PathEscape("/../../"+testFileName),
- bytes.NewBuffer(content))
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Uploading outside the share is not allowed")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestShareUncompressed(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "testfile.dat"
- testFileSize := int64(65536)
- testFilePath := filepath.Join(user.GetHomeDir(), testFileName)
- err = createTestFile(testFilePath, testFileSize)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- Password: defaultPassword,
- MaxTokens: 0,
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- s, err := dataprovider.ShareExists(objectID, defaultUsername)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), s.ExpiresAt)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "application/zip", rr.Header().Get("Content-Type"))
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"?compress=false", nil) //nolint:goconst
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "application/zip", rr.Header().Get("Content-Type"))
- share = dataprovider.Share{
- Name: "test share1",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{testFileName},
- Password: defaultPassword,
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "application/zip", rr.Header().Get("Content-Type"))
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"?compress=false", nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "application/octet-stream", rr.Header().Get("Content-Type"))
- share, err = dataprovider.ShareExists(objectID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, 2, share.UsedTokens)
- user.Permissions["/"] = []string{dataprovider.PermListItems, dataprovider.PermUpload}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"?compress=false", nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- share, err = dataprovider.ShareExists(objectID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, 2, share.UsedTokens)
- user.Permissions["/"] = []string{dataprovider.PermAny}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- err = os.Remove(testFilePath)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"?compress=false", nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestDownloadFromShareError(t *testing.T) {
- u := getTestUser()
- u.DownloadDataTransfer = 1
- u.Filters.DefaultSharesExpiration = 10
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user.UsedDownloadDataTransfer = 1024*1024 - 32768
- _, err = httpdtest.UpdateTransferQuotaUsage(user, "add", http.StatusOK)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(1024*1024-32768), user.UsedDownloadDataTransfer)
- testFileName := "test_share_file.dat"
- testFileSize := int64(524288)
- testFilePath := filepath.Join(user.GetHomeDir(), testFileName)
- err = createTestFile(testFilePath, testFileSize)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share root browse",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- MaxTokens: 2,
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- s, err := dataprovider.ShareExists(objectID, defaultUsername)
- assert.NoError(t, err)
- assert.Greater(t, s.ExpiresAt, int64(0))
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- share, err = dataprovider.ShareExists(objectID, user.Username)
- assert.NoError(t, err)
- assert.Equal(t, 0, share.UsedTokens)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }()
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path="+testFileName), nil)
- assert.NoError(t, err)
- executeRequest(req)
- }
- func TestBrowseShares(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "testsharefile.dat"
- testFileNameLink := "testsharefile.link"
- shareDir := "share"
- subDir := "sub"
- testFileSize := int64(65536)
- testFilePath := filepath.Join(user.GetHomeDir(), shareDir, testFileName)
- testLinkPath := filepath.Join(user.GetHomeDir(), shareDir, testFileNameLink)
- err = createTestFile(testFilePath, testFileSize)
- assert.NoError(t, err)
- err = createTestFile(filepath.Join(user.GetHomeDir(), shareDir, subDir, testFileName), testFileSize)
- assert.NoError(t, err)
- err = os.Symlink(testFilePath, testLinkPath)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share browse",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{shareDir},
- MaxTokens: 0,
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "upload"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "invalid share scope")
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Please set the path to a valid file")
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents := make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=%2F"+subDir), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 1)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F.."), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPathInvalid)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "dirs?path=%2F.."), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=%2F.."), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path=%2F..%2F"+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path="+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contentDisposition := rr.Header().Get("Content-Disposition")
- assert.NotEmpty(t, contentDisposition)
- form := make(url.Values)
- form.Set("files", `[]`)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial?path=%2F.."),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPathInvalid)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path="+testFileName), nil) //nolint:goconst
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contentDisposition = rr.Header().Get("Content-Disposition")
- assert.NotEmpty(t, contentDisposition)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path="+subDir+"%2F"+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contentDisposition = rr.Header().Get("Content-Disposition")
- assert.NotEmpty(t, contentDisposition)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=missing"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFsGeneric)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path=missing"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "dirs?path=missing"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=missing"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path="+testFileNameLink), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFsGeneric)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path="+testFileNameLink), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "non regular files are not supported for shares")
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path="+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPDFMessage)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=missing"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFsGeneric)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPDFMessage)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=%2F.."), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPathInvalid)
- fakePDF := []byte(`%PDF-1.6`)
- for i := 0; i < 128; i++ {
- fakePDF = append(fakePDF, []byte(fmt.Sprintf("%d", i))...)
- }
- pdfPath := filepath.Join(user.GetHomeDir(), shareDir, "test.pdf")
- pdfLinkPath := filepath.Join(user.GetHomeDir(), shareDir, "link.pdf")
- err = os.WriteFile(pdfPath, fakePDF, 0666)
- assert.NoError(t, err)
- err = os.Symlink(pdfPath, pdfLinkPath)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "viewpdf?path=test.pdf"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=test.pdf"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- s, err := dataprovider.ShareExists(objectID, defaultUsername)
- assert.NoError(t, err)
- usedTokens := s.UsedTokens
- assert.Greater(t, usedTokens, 0)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=link.pdf"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // downloading a symlink will fail, usage should not change
- s, err = dataprovider.ShareExists(objectID, defaultUsername)
- assert.NoError(t, err)
- assert.Equal(t, usedTokens, s.UsedTokens)
- // share a symlink
- share = dataprovider.Share{
- Name: "test share browse",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{path.Join(shareDir, testFileNameLink)},
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- // uncompressed download should not work
- req, err = http.NewRequest(http.MethodGet, webClientPubSharesPath+"/"+objectID+"?compress=false", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "application/zip", rr.Header().Get("Content-Type"))
- // this share is not browsable, it does not contains a directory
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- form = make(url.Values)
- form.Set("files", `[]`)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareBrowseNoDir)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path="+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareBrowseNoDir)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path="+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the shared object is not a directory and so it is not browsable")
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareBrowseNoDir)
- // now test a missing shareID
- objectID = "123456"
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "files?path="+testFileName), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- form = make(url.Values)
- form.Set("files", `[]`)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "partial?path=%2F"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "viewpdf?path=p"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "getpdf?path=p"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // share a missing base path
- share = dataprovider.Share{
- Name: "test share",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{path.Join(shareDir, "missingdir")},
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "unable to check the share directory")
- // share multiple paths
- share = dataprovider.Share{
- Name: "test share",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{shareDir, "/anotherdir"},
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareBrowsePaths)
- share = dataprovider.Share{
- Name: "test share rw",
- Scope: dataprovider.ShareScopeReadWrite,
- Paths: []string{"/missingdir"},
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "/browse/exist?path=%2F"), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "unable to check the share directory")
- share = dataprovider.Share{
- Name: "test share rw",
- Scope: dataprovider.ShareScopeReadWrite,
- Paths: []string{shareDir},
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "/browse/exist?path=%2F.."), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Invalid path")
- // share the root path
- share = dataprovider.Share{
- Name: "test share root",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- MaxTokens: 0,
- }
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 1)
- // if we require two-factor auth for HTTP protocol the share should not work anymore
- user.Filters.TwoFactorAuthProtocols = []string{common.ProtocolSSH}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user.Filters.TwoFactorAuthProtocols = []string{common.ProtocolSSH, common.ProtocolHTTP}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, path.Join(sharesPath, objectID, "dirs?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "two-factor authentication requirements not met")
- user.Filters.TwoFactorAuthProtocols = []string{common.ProtocolSSH}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- // share read/write
- share.Scope = dataprovider.ShareScopeReadWrite
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "browse?path=%2F"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // on upload we should be redirected
- req, err = http.NewRequest(http.MethodGet, path.Join(webClientPubSharesPath, objectID, "upload"), nil)
- assert.NoError(t, err)
- req.SetBasicAuth(defaultUsername, defaultPassword)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- location := rr.Header().Get("Location")
- assert.Equal(t, path.Join(webClientPubSharesPath, objectID, "browse"), location)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestUserAPIShareErrors(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Scope: 1000,
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "invalid scope")
- // invalid json
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer([]byte("{")))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- share.Scope = dataprovider.ShareScopeWrite
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "at least a shared path is required")
- share.Paths = []string{"path1", "../path1", "/path2"}
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the write share scope requires exactly one path")
- share.Paths = []string{"", ""}
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "at least a shared path is required")
- share.Paths = []string{"path1", "../path1", "/path1"}
- share.Password = redactedSecret
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "cannot save a share with a redacted password")
- share.Password = "newpass"
- share.AllowFrom = []string{"not valid"}
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "could not parse allow from entry")
- share.AllowFrom = []string{"127.0.0.1/8"}
- share.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(-12 * time.Hour))
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "expiration must be in the future")
- share.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(12 * time.Hour))
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- location := rr.Header().Get("Location")
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "name is mandatory")
- // invalid json
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer([]byte("}")))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, userSharesPath+"?limit=a", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestUserAPIShares(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- u := getTestUser()
- u.Username = altAdminUsername
- user1, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token1, err := getJWTAPIUserTokenFromTestServer(user1.Username, defaultPassword)
- assert.NoError(t, err)
- // the share username will be set from the claims
- share := dataprovider.Share{
- Name: "share1",
- Description: "description1",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- CreatedAt: 1,
- UpdatedAt: 2,
- LastUseAt: 3,
- ExpiresAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(2 * time.Hour)),
- Password: defaultPassword,
- MaxTokens: 10,
- UsedTokens: 2,
- AllowFrom: []string{"192.168.1.0/24"},
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- location := rr.Header().Get("Location")
- assert.NotEmpty(t, location)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- assert.Equal(t, fmt.Sprintf("%v/%v", userSharesPath, objectID), location)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var shareGet dataprovider.Share
- err = json.Unmarshal(rr.Body.Bytes(), &shareGet)
- assert.NoError(t, err)
- assert.Equal(t, objectID, shareGet.ShareID)
- assert.Equal(t, share.Name, shareGet.Name)
- assert.Equal(t, share.Description, shareGet.Description)
- assert.Equal(t, share.Scope, shareGet.Scope)
- assert.Equal(t, share.Paths, shareGet.Paths)
- assert.Equal(t, int64(0), shareGet.LastUseAt)
- assert.Greater(t, shareGet.CreatedAt, share.CreatedAt)
- assert.Greater(t, shareGet.UpdatedAt, share.UpdatedAt)
- assert.Equal(t, share.ExpiresAt, shareGet.ExpiresAt)
- assert.Equal(t, share.MaxTokens, shareGet.MaxTokens)
- assert.Equal(t, 0, shareGet.UsedTokens)
- assert.Equal(t, share.Paths, shareGet.Paths)
- assert.Equal(t, redactedSecret, shareGet.Password)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token1)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- s, err := dataprovider.ShareExists(objectID, defaultUsername)
- assert.NoError(t, err)
- match, err := s.CheckCredentials(defaultPassword)
- assert.True(t, match)
- assert.NoError(t, err)
- match, err = s.CheckCredentials(defaultPassword + "mod")
- assert.False(t, match)
- assert.Error(t, err)
- shareGet.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(3 * time.Hour))
- asJSON, err = json.Marshal(shareGet)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- s, err = dataprovider.ShareExists(objectID, defaultUsername)
- assert.NoError(t, err)
- match, err = s.CheckCredentials(defaultPassword)
- assert.True(t, match)
- assert.NoError(t, err)
- match, err = s.CheckCredentials(defaultPassword + "mod")
- assert.False(t, match)
- assert.Error(t, err)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var shareGetNew dataprovider.Share
- err = json.Unmarshal(rr.Body.Bytes(), &shareGetNew)
- assert.NoError(t, err)
- assert.NotEqual(t, shareGet.UpdatedAt, shareGetNew.UpdatedAt)
- shareGet.UpdatedAt = shareGetNew.UpdatedAt
- assert.Equal(t, shareGet, shareGetNew)
- req, err = http.NewRequest(http.MethodGet, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var shares []dataprovider.Share
- err = json.Unmarshal(rr.Body.Bytes(), &shares)
- assert.NoError(t, err)
- if assert.Len(t, shares, 1) {
- assert.Equal(t, shareGetNew, shares[0])
- }
- err = dataprovider.UpdateShareLastUse(&shareGetNew, 2)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- shareGetNew = dataprovider.Share{}
- err = json.Unmarshal(rr.Body.Bytes(), &shareGetNew)
- assert.NoError(t, err)
- assert.Equal(t, 2, shareGetNew.UsedTokens, "share: %v", shareGetNew)
- assert.Greater(t, shareGetNew.LastUseAt, int64(0), "share: %v", shareGetNew)
- req, err = http.NewRequest(http.MethodGet, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token1)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- shares = nil
- err = json.Unmarshal(rr.Body.Bytes(), &shares)
- assert.NoError(t, err)
- assert.Len(t, shares, 0)
- // set an empty password
- shareGet.Password = ""
- asJSON, err = json.Marshal(shareGet)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- shareGetNew = dataprovider.Share{}
- err = json.Unmarshal(rr.Body.Bytes(), &shareGetNew)
- assert.NoError(t, err)
- assert.Empty(t, shareGetNew.Password)
- req, err = http.NewRequest(http.MethodDelete, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- share.Name = ""
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- location = rr.Header().Get("Location")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- // the share should be deleted with the associated user
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodDelete, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user1.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestUsersAPISharesNoPasswordDisabled(t *testing.T) {
- u := getTestUser()
- u.Filters.WebClient = []string{sdk.WebClientShareNoPasswordDisabled}
- u.Filters.PasswordStrength = 70
- u.Password = "ahpoo8baa6EeZieshies"
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, u.Password)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "s",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- }
- asJSON, err := json.Marshal(share)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "You are not authorized to share files/folders without a password")
- share.Password = defaultPassword
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- share.Password = "vi5eiJoovee5ya9yahpi"
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- location := rr.Header().Get("Location")
- assert.NotEmpty(t, location)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- assert.Equal(t, fmt.Sprintf("%v/%v", userSharesPath, objectID), location)
- share.Password = ""
- asJSON, err = json.Marshal(share)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "You are not authorized to share files/folders without a password")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestUserAPIKey(t *testing.T) {
- u := getTestUser()
- u.Filters.AllowAPIKeyAuth = true
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Name: "testkey",
- User: user.Username + "1",
- Scope: dataprovider.APIKeyScopeUser,
- }
- _, _, err = httpdtest.AddAPIKey(apiKey, http.StatusBadRequest)
- assert.NoError(t, err)
- apiKey.User = user.Username
- apiKey, _, err = httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err)
- adminAPIKey := dataprovider.APIKey{
- Name: "testadminkey",
- Scope: dataprovider.APIKeyScopeAdmin,
- }
- adminAPIKey, _, err = httpdtest.AddAPIKey(adminAPIKey, http.StatusCreated)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "filenametest")
- assert.NoError(t, err)
- _, err = part.Write([]byte("test file content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var dirEntries []map[string]any
- err = json.Unmarshal(rr.Body.Bytes(), &dirEntries)
- assert.NoError(t, err)
- assert.Len(t, dirEntries, 1)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, adminAPIKey.Key, user.Username)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- user.Status = 0
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- user.Status = 1
- user.Filters.DeniedProtocols = []string{common.ProtocolHTTP}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- user.Filters.DeniedProtocols = []string{common.ProtocolFTP}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- apiKeyNew := dataprovider.APIKey{
- Name: apiKey.Name,
- Scope: dataprovider.APIKeyScopeUser,
- }
- apiKeyNew, _, err = httpdtest.AddAPIKey(apiKeyNew, http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKeyNew.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- // now associate a user
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKeyNew.Key, user.Username)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now with a missing user
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKeyNew.Key, user.Username+"1")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- // empty user and key not associated to any user
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKeyNew.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- apiKeyNew.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(-24 * time.Hour))
- _, _, err = httpdtest.UpdateAPIKey(apiKeyNew, http.StatusOK)
- assert.NoError(t, err)
- // expired API key
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKeyNew.Key, user.Username)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAPIKey(apiKeyNew, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAPIKey(adminAPIKey, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebClientExistenceCheck(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webClientExistPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr) // no CSRF header
- req, err = http.NewRequest(http.MethodPost, webClientExistPath, bytes.NewBuffer([]byte(`[]`)))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- filesToCheck := make(map[string]any)
- filesToCheck["files"] = nil
- asJSON, err := json.Marshal(filesToCheck)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "files to be checked are mandatory")
- testFileName := "file.dat"
- testDirName := "adirname"
- filesToCheck["files"] = []string{testFileName}
- asJSON, err = json.Marshal(filesToCheck)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath+"?path=%2Fmissingdir", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath+"?path=%2F", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var fileList []any
- err = json.Unmarshal(rr.Body.Bytes(), &fileList)
- assert.NoError(t, err)
- assert.Len(t, fileList, 0)
- err = createTestFile(filepath.Join(user.GetHomeDir(), testFileName), 100)
- assert.NoError(t, err)
- err = os.Mkdir(filepath.Join(user.GetHomeDir(), testDirName), 0755)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath+"?path=%2F", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- fileList = nil
- err = json.Unmarshal(rr.Body.Bytes(), &fileList)
- assert.NoError(t, err)
- assert.Len(t, fileList, 1)
- filesToCheck["files"] = []string{testFileName, testDirName}
- asJSON, err = json.Marshal(filesToCheck)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath+"?path=%2F", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- fileList = nil
- err = json.Unmarshal(rr.Body.Bytes(), &fileList)
- assert.NoError(t, err)
- assert.Len(t, fileList, 2)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath+"?path=%2F"+testDirName, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- fileList = nil
- err = json.Unmarshal(rr.Body.Bytes(), &fileList)
- assert.NoError(t, err)
- assert.Len(t, fileList, 0)
- user.Filters.DeniedProtocols = []string{common.ProtocolHTTP}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientExistPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebClientViewPDF(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientViewPDFPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, webClientViewPDFPath+"?path=test.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=test.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFsGeneric)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2F", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPDFMessage)
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), "test.pdf"), []byte("some text data"), 0666)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2Ftest.pdf", nil) //nolint:goconst
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPDFMessage)
- err = createTestFile(filepath.Join(user.GetHomeDir(), "test.pdf"), 1024)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2Ftest.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPDFMessage)
- fakePDF := []byte(`%PDF-1.6`)
- for i := 0; i < 128; i++ {
- fakePDF = append(fakePDF, []byte(fmt.Sprintf("%d", i))...)
- }
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), "test.pdf"), fakePDF, 0666)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2Ftest.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/",
- DeniedPatterns: []string{"*.pdf"},
- },
- }
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2Ftest.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError403Message)
- user.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/",
- DeniedPatterns: []string{"*.txt"},
- },
- }
- user.Filters.DeniedProtocols = []string{common.ProtocolHTTP}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2Ftest.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientGetPDFPath+"?path=%2Ftest.pdf", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestWebEditFile(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- testFile1 := "testfile1.txt"
- testFile2 := "testfile2"
- file1Size := int64(65536)
- file2Size := int64(1048576 * 5)
- err = createTestFile(filepath.Join(user.GetHomeDir(), testFile1), file1Size)
- assert.NoError(t, err)
- err = createTestFile(filepath.Join(user.GetHomeDir(), testFile2), file2Size)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientEditFilePath+"?path="+testFile1, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path="+testFile2, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorEditSize)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path=missing", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFsGeneric)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path=%2F", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorEditDir)
- user.Filters.DeniedProtocols = []string{common.ProtocolHTTP}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path="+testFile1, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- user.Filters.DeniedProtocols = []string{common.ProtocolFTP}
- user.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/",
- DeniedPatterns: []string{"*.txt"},
- },
- }
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path="+testFile1, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError403Message)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientEditFilePath+"?path="+testFile1, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestWebGetFiles(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "testfile"
- testDir := "testdir"
- testFileContents := []byte("file contents")
- err = os.MkdirAll(filepath.Join(user.GetHomeDir(), testDir), os.ModePerm)
- assert.NoError(t, err)
- extensions := []string{"", ".doc", ".ppt", ".xls", ".pdf", ".mkv", ".png", ".go", ".zip", ".txt"}
- for _, ext := range extensions {
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), testFileName+ext), testFileContents, os.ModePerm)
- assert.NoError(t, err)
- }
- err = os.Symlink(filepath.Join(user.GetHomeDir(), testFileName+".doc"), filepath.Join(user.GetHomeDir(), testDir, testFileName+".link"))
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testDir, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientDirsPath+"?path="+testDir, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var dirContents []map[string]any
- err = json.Unmarshal(rr.Body.Bytes(), &dirContents)
- assert.NoError(t, err)
- assert.Len(t, dirContents, 1)
- req, _ = http.NewRequest(http.MethodGet, webClientDirsPath+"?dirtree=1&path="+testDir, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- dirContents = make([]map[string]any, 0)
- err = json.Unmarshal(rr.Body.Bytes(), &dirContents)
- assert.NoError(t, err)
- assert.Len(t, dirContents, 0)
- req, _ = http.NewRequest(http.MethodGet, userDirsPath+"?path="+testDir, nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var dirEntries []map[string]any
- err = json.Unmarshal(rr.Body.Bytes(), &dirEntries)
- assert.NoError(t, err)
- assert.Len(t, dirEntries, 1)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("files", fmt.Sprintf(`["%s","%s","%s"]`, testFileName, testDir, testFileName+extensions[2]))
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath+"?path="+url.QueryEscape("/"),
- bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- // add csrf token
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath+"?path="+url.QueryEscape("/"),
- bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // parse form error
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath+"?path=p%C3%AO%GK",
- bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- filesList := []string{testFileName, testDir, testFileName + extensions[2]}
- asJSON, err := json.Marshal(filesList)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPost, userStreamZipPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPost, userStreamZipPath, bytes.NewBuffer([]byte(`file`)))
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("files", fmt.Sprintf(`["%v"]`, testDir))
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath+"?path="+url.QueryEscape("/"),
- bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("files", "notalist")
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath+"?path="+url.QueryEscape("/"),
- bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- req, _ = http.NewRequest(http.MethodGet, webClientDirsPath+"?path=/", nil) //nolint:goconst
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- dirContents = nil
- err = json.Unmarshal(rr.Body.Bytes(), &dirContents)
- assert.NoError(t, err)
- assert.Len(t, dirContents, len(extensions)+1)
- req, _ = http.NewRequest(http.MethodGet, userDirsPath+"?path=/", nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- dirEntries = nil
- err = json.Unmarshal(rr.Body.Bytes(), &dirEntries)
- assert.NoError(t, err)
- assert.Len(t, dirEntries, len(extensions)+1)
- req, _ = http.NewRequest(http.MethodGet, webClientDirsPath+"?path=/missing", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorDirListGeneric)
- req, _ = http.NewRequest(http.MethodGet, userDirsPath+"?path=missing", nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to get directory lister")
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, testFileContents, rr.Body.Bytes())
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, testFileContents, rr.Body.Bytes())
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path=", nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Please set the path to a valid file")
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testDir, nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "is a directory")
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path=notafile", nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to stat the requested file")
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=2-")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- assert.Equal(t, testFileContents[2:], rr.Body.Bytes())
- lastModified, err := http.ParseTime(rr.Header().Get("Last-Modified"))
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=2-")
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- assert.Equal(t, testFileContents[2:], rr.Body.Bytes())
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=-2")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- assert.Equal(t, testFileContents[11:], rr.Body.Bytes())
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=-2,")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestedRangeNotSatisfiable, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=1a-")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestedRangeNotSatisfiable, rr)
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=2b-")
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestedRangeNotSatisfiable, rr)
- req, _ = http.NewRequest(http.MethodHead, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=2-")
- req.Header.Set("If-Range", lastModified.UTC().Format(http.TimeFormat))
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- req, _ = http.NewRequest(http.MethodHead, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=2-")
- req.Header.Set("If-Range", lastModified.UTC().Add(-120*time.Second).Format(http.TimeFormat))
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodHead, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("If-Modified-Since", lastModified.UTC().Add(-120*time.Second).Format(http.TimeFormat))
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodHead, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("If-Modified-Since", lastModified.UTC().Add(120*time.Second).Format(http.TimeFormat))
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotModified, rr)
- req, _ = http.NewRequest(http.MethodHead, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("If-Unmodified-Since", lastModified.UTC().Add(-120*time.Second).Format(http.TimeFormat))
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPreconditionFailed, rr)
- req, _ = http.NewRequest(http.MethodHead, userFilesPath+"?path="+testFileName, nil)
- req.Header.Set("If-Unmodified-Since", lastModified.UTC().Add(-120*time.Second).Format(http.TimeFormat))
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPreconditionFailed, rr)
- req, _ = http.NewRequest(http.MethodHead, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("If-Unmodified-Since", lastModified.UTC().Add(120*time.Second).Format(http.TimeFormat))
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user.Filters.DeniedProtocols = []string{common.ProtocolHTTP}
- _, resp, err := httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientDirsPath+"?path=/", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, _ = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, _ = http.NewRequest(http.MethodGet, userDirsPath+"?path="+testDir, nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- filesList = []string{testDir}
- asJSON, err = json.Marshal(filesList)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPost, userStreamZipPath, bytes.NewBuffer(asJSON))
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- user.Filters.DeniedProtocols = []string{common.ProtocolFTP}
- user.Filters.DeniedLoginMethods = []string{dataprovider.LoginMethodPassword}
- _, resp, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- form = make(url.Values)
- form.Set("files", `[]`)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientDownloadZipPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, _ = http.NewRequest(http.MethodGet, userDirsPath+"?path="+testDir, nil)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestRenameDifferentResource(t *testing.T) {
- folderName := "foldercryptfs"
- f := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: filepath.Join(os.TempDir(), "folderName"),
- FsConfig: vfs.Filesystem{
- Provider: sdk.CryptedFilesystemProvider,
- CryptConfig: vfs.CryptFsConfig{
- Passphrase: kms.NewPlainSecret("super secret"),
- },
- },
- }
- _, _, err := httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err)
- u := getTestUser()
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- },
- VirtualPath: "/folderPath",
- })
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "file.txt"
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- getStatusResponse := func(taskID string) int {
- req, _ := http.NewRequest(http.MethodGet, webClientTasksPath+"/"+url.PathEscape(taskID), nil)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- if rr.Code != http.StatusOK {
- return -1
- }
- resp := make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- if err != nil {
- return -1
- }
- return int(resp["status"].(float64))
- }
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testFileName+"&target="+url.QueryEscape(path.Join("/", "folderPath", testFileName)), nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Cannot perform copy step")
- req, err = http.NewRequest(http.MethodPost, webClientFileMovePath+"?path="+testFileName+"&target="+url.QueryEscape(path.Join("/", "folderPath", testFileName)), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- taskResp := make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &taskResp)
- assert.NoError(t, err)
- taskID := taskResp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusNotFound
- }, 1000*time.Millisecond, 100*time.Millisecond)
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), testFileName), []byte("just a test"), os.ModePerm)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testFileName+"&target="+url.QueryEscape(path.Join("/", "folderPath", testFileName)), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // recreate the file and remove the delete permission
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), testFileName), []byte("just another test"), os.ModePerm)
- assert.NoError(t, err)
- u.Permissions = map[string][]string{
- "/": {dataprovider.PermUpload, dataprovider.PermListItems, dataprovider.PermCreateDirs,
- dataprovider.PermDownload, dataprovider.PermOverwrite},
- }
- _, resp, err := httpdtest.UpdateUser(u, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- webAPIToken, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testFileName+"&target="+url.QueryEscape(path.Join("/", "folderPath", testFileName)), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Cannot perform copy step")
- u.Permissions = map[string][]string{
- "/": {dataprovider.PermUpload, dataprovider.PermListItems, dataprovider.PermCreateDirs,
- dataprovider.PermDownload, dataprovider.PermOverwrite, dataprovider.PermCopy},
- }
- _, resp, err = httpdtest.UpdateUser(u, http.StatusOK, "")
- assert.NoError(t, err, string(resp))
- webAPIToken, err = getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testFileName+"&target="+url.QueryEscape(path.Join("/", "folderPath", testFileName)), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Cannot perform remove step")
- req, err = http.NewRequest(http.MethodPost, webClientFileMovePath+"?path="+testFileName+"&target="+url.QueryEscape(path.Join("/", "folderPath", testFileName)), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- taskResp = make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &taskResp)
- assert.NoError(t, err)
- taskID = taskResp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusForbidden
- }, 1000*time.Millisecond, 100*time.Millisecond)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebDirsAPI(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- testDir := "testdir"
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var contents []map[string]any
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 0)
- // rename a missing folder
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testDir+"&target="+testDir+"new", nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // copy a missing folder
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/copy?path="+testDir+"%2F&target="+testDir+"new%2F", nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // delete a missing folder
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // create a dir
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- // check the dir was created
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- if assert.Len(t, contents, 1) {
- assert.Equal(t, testDir, contents[0]["name"])
- }
- // rename a dir with the same source and target name
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testDir+"&target="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testDir+"&target=%2F"+testDir+"%2F", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- // copy a dir with the same source and target name
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/copy?path="+testDir+"&target="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- // create a dir with missing parents
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path="+url.QueryEscape(path.Join("/sub/dir", testDir)), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // setting the mkdir_parents param will work
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?mkdir_parents=true&path="+url.QueryEscape(path.Join("/sub/dir", testDir)), nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- // copy the dir
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/copy?path="+testDir+"&target="+testDir+"copy", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // rename the dir
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testDir+"&target="+testDir+"new", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // delete the dir
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path="+testDir+"new", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path="+testDir+"copy", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // the root dir cannot be created
- req, err = http.NewRequest(http.MethodPost, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- user.Permissions["/"] = []string{dataprovider.PermListItems}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- // the user has no more the permission to create the directory
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- // the user is deleted, any API call should fail
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path="+testDir+"&target="+testDir+"new", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/copy?path="+testDir+"&target="+testDir+"new", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path="+testDir+"new", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestWebUploadSingleFile(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- content := []byte("test content")
- req, err := http.NewRequest(http.MethodPost, userUploadFilePath, bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "please set a file path")
- modTime := time.Now().Add(-24 * time.Hour)
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=file.txt", bytes.NewBuffer(content)) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- req.Header.Set("X-SFTPGO-MTIME", strconv.FormatInt(util.GetTimeAsMsSinceEpoch(modTime), 10))
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- info, err := os.Stat(filepath.Join(user.GetHomeDir(), "file.txt"))
- if assert.NoError(t, err) {
- assert.InDelta(t, util.GetTimeAsMsSinceEpoch(modTime), util.GetTimeAsMsSinceEpoch(info.ModTime()), float64(1000))
- }
- // invalid modification time will be ignored
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=file.txt", bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- req.Header.Set("X-SFTPGO-MTIME", "123abc")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- info, err = os.Stat(filepath.Join(user.GetHomeDir(), "file.txt"))
- if assert.NoError(t, err) {
- assert.InDelta(t, util.GetTimeAsMsSinceEpoch(time.Now()), util.GetTimeAsMsSinceEpoch(info.ModTime()), float64(3000))
- }
- // upload to a missing dir will fail without the mkdir_parents param
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path="+url.QueryEscape("/subdir/file.txt"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?mkdir_parents=true&path="+url.QueryEscape("/subdir/file.txt"), bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- metadataReq := make(map[string]int64)
- metadataReq["modification_time"] = util.GetTimeAsMsSinceEpoch(modTime)
- asJSON, err := json.Marshal(metadataReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPatch, userFilesDirsMetadataPath+"?path=file.txt", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- info, err = os.Stat(filepath.Join(user.GetHomeDir(), "file.txt"))
- if assert.NoError(t, err) {
- assert.InDelta(t, util.GetTimeAsMsSinceEpoch(modTime), util.GetTimeAsMsSinceEpoch(info.ModTime()), float64(1000))
- }
- // missing file
- req, err = http.NewRequest(http.MethodPatch, userFilesDirsMetadataPath+"?path=file2.txt", bytes.NewBuffer(asJSON)) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to set metadata for path")
- // invalid JSON
- req, err = http.NewRequest(http.MethodPatch, userFilesDirsMetadataPath+"?path=file.txt", bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // missing mandatory parameter
- req, err = http.NewRequest(http.MethodPatch, userFilesDirsMetadataPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "please set a modification_time and a path")
- metadataReq = make(map[string]int64)
- asJSON, err = json.Marshal(metadataReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPatch, userFilesDirsMetadataPath+"?path=file.txt", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "please set a modification_time and a path")
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=%2Fdir%2Ffile.txt", bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to write file")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=file.txt", bytes.NewBuffer(content))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to retrieve your user")
- metadataReq["modification_time"] = util.GetTimeAsMsSinceEpoch(modTime)
- asJSON, err = json.Marshal(metadataReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPatch, userFilesDirsMetadataPath+"?path=file.txt", bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to retrieve your user")
- }
- func TestWebFilesAPI(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part1, err := writer.CreateFormFile("filenames", "file1.txt")
- assert.NoError(t, err)
- _, err = part1.Write([]byte("file1 content"))
- assert.NoError(t, err)
- part2, err := writer.CreateFormFile("filenames", "file2.txt")
- assert.NoError(t, err)
- _, err = part2.Write([]byte("file2 content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Unable to parse multipart form")
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), user.FirstUpload)
- assert.Equal(t, int64(0), user.FirstDownload)
- // set the proper content type
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, user.FirstUpload, int64(0))
- assert.Equal(t, int64(0), user.FirstDownload)
- // check we have 2 files
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var contents []map[string]any
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- // download a file
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path=file1.txt", nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "file1 content", rr.Body.String())
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Greater(t, user.FirstUpload, int64(0))
- assert.Greater(t, user.FirstDownload, int64(0))
- // overwrite the existing files
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- // now create a dir and upload to that dir
- testDir := "tdir"
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path="+testDir, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- // upload to a missing subdir will fail without the mkdir_parents param
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path="+url.QueryEscape("/sub/"+testDir), reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?mkdir_parents=true&path="+url.QueryEscape("/sub/"+testDir), reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 4)
- req, err = http.NewRequest(http.MethodGet, userDirsPath+"?path="+testDir, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- // copy a file
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/copy?path=file1.txt&target=%2Ftdir%2Ffile_copy.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // rename a file
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?target=%2Ftdir%2Ffile3.txt&path=file1.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // rename a missing file
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path=file1.txt&target=%2Ftdir%2Ffile3.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // rename a file with target name equal to source name
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path=file1.txt&target=file1.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- // delete a file
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file2.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // delete a missing file
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file2.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // delete a directory
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=tdir", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // make a symlink outside the home dir and then try to delete it
- extPath := filepath.Join(os.TempDir(), "file")
- err = os.WriteFile(extPath, []byte("contents"), os.ModePerm)
- assert.NoError(t, err)
- err = os.Symlink(extPath, filepath.Join(user.GetHomeDir(), "file"))
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- err = os.Remove(extPath)
- assert.NoError(t, err)
- // remove delete and overwrite permissions
- user.Permissions["/"] = []string{dataprovider.PermListItems, dataprovider.PermDownload, dataprovider.PermUpload}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path=tdir", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=%2Ftdir%2Ffile1.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- // the user is deleted, any API call should fail
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path=file1.txt&target=%2Ftdir%2Ffile3.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file2.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestBufferedWebFilesAPI(t *testing.T) {
- u := getTestUser()
- u.FsConfig.OSConfig = sdk.OSFsConfig{
- ReadBufferSize: 1,
- WriteBufferSize: 1,
- }
- vdirPath := "/crypted"
- mappedPath := filepath.Join(os.TempDir(), util.GenerateUniqueID())
- folderName := filepath.Base(mappedPath)
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- },
- VirtualPath: vdirPath,
- QuotaFiles: -1,
- QuotaSize: -1,
- })
- f := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedPath,
- FsConfig: vfs.Filesystem{
- Provider: sdk.CryptedFilesystemProvider,
- CryptConfig: vfs.CryptFsConfig{
- OSFsConfig: sdk.OSFsConfig{
- WriteBufferSize: 3,
- ReadBufferSize: 2,
- },
- Passphrase: kms.NewPlainSecret(defaultPassword),
- },
- },
- }
- _, _, err := httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part1, err := writer.CreateFormFile("filenames", "file1.txt")
- assert.NoError(t, err)
- _, err = part1.Write([]byte("file1 content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path="+url.QueryEscape(vdirPath), reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path=file1.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "file1 content", rr.Body.String())
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+url.QueryEscape(vdirPath+"/file1.txt"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "file1 content", rr.Body.String())
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path=file1.txt", nil)
- assert.NoError(t, err)
- req.Header.Set("Range", "bytes=2-")
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- assert.Equal(t, "le1 content", rr.Body.String())
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+url.QueryEscape(vdirPath+"/file1.txt"), nil)
- assert.NoError(t, err)
- req.Header.Set("Range", "bytes=3-6")
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- assert.Equal(t, "e1 c", rr.Body.String())
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(mappedPath)
- assert.NoError(t, err)
- }
- func TestWebClientTasksAPI(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- u1 := getTestUser()
- u1.Username = xid.New().String()
- user1, _, err := httpdtest.AddUser(u1, http.StatusCreated)
- assert.NoError(t, err)
- testDir := "subdir"
- testFileData := []byte("data")
- testFilePath := filepath.Join(user.GetHomeDir(), testDir, "file.txt")
- testFileName := filepath.Base(testFilePath)
- err = os.MkdirAll(filepath.Dir(testFilePath), os.ModePerm)
- assert.NoError(t, err)
- err = os.WriteFile(testFilePath, testFileData, 0666)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- webToken1, err := getJWTWebClientTokenFromTestServer(user1.Username, defaultPassword)
- assert.NoError(t, err)
- getStatusResponse := func(taskID string) int {
- req, _ := http.NewRequest(http.MethodGet, webClientTasksPath+"/"+url.PathEscape(taskID), nil)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- if rr.Code != http.StatusOK {
- return -1
- }
- resp := make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- if err != nil {
- return -1
- }
- return int(resp["status"].(float64))
- }
- // missing task
- assert.Equal(t, -1, getStatusResponse("missing"))
- req, err := http.NewRequest(http.MethodPost, webClientFileCopyPath+"?path="+
- url.QueryEscape(path.Join(testDir, testFileName))+"&target="+url.QueryEscape(testFileName), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- resp := make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- taskID := resp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusOK
- }, 1000*time.Millisecond, 100*time.Millisecond)
- // cannot get the task with a different user
- req, err = http.NewRequest(http.MethodGet, webClientTasksPath+"/"+url.PathEscape(taskID), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken1)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, webClientFileMovePath+"?path="+
- url.QueryEscape(path.Join(testDir, testFileName))+"&target="+url.QueryEscape(testFileName), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- resp = make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- taskID = resp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusOK
- }, 1000*time.Millisecond, 100*time.Millisecond)
- req, err = http.NewRequest(http.MethodDelete, webClientDirsPath+"?path="+
- url.QueryEscape(testDir), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- resp = make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- taskID = resp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusOK
- }, 1000*time.Millisecond, 100*time.Millisecond)
- req, err = http.NewRequest(http.MethodDelete, webClientDirsPath+"?path="+
- url.QueryEscape(testDir), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- resp = make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- taskID = resp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusNotFound
- }, 1000*time.Millisecond, 100*time.Millisecond)
- req, err = http.NewRequest(http.MethodPost, webClientFileMovePath+"?path="+
- url.QueryEscape(path.Join(testDir, testFileName))+"&target="+url.QueryEscape(testFileName), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- resp = make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- taskID = resp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusNotFound
- }, 1000*time.Millisecond, 100*time.Millisecond)
- req, err = http.NewRequest(http.MethodPost, webClientFileCopyPath+"?path="+
- url.QueryEscape(path.Join(testDir, testFileName)+"/")+"&target="+url.QueryEscape(testFileName+"/"), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- resp = make(map[string]any)
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- taskID = resp["message"].(string)
- assert.NotEmpty(t, taskID)
- assert.Eventually(t, func() bool {
- status := getStatusResponse(taskID)
- return status == http.StatusNotFound
- }, 1000*time.Millisecond, 100*time.Millisecond)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user1, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user1.GetHomeDir())
- assert.NoError(t, err)
- // user deleted
- req, err = http.NewRequest(http.MethodDelete, webClientDirsPath+"?path="+
- url.QueryEscape(testDir), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webClientFileMovePath+"?path="+
- url.QueryEscape(path.Join(testDir, testFileName))+"&target="+url.QueryEscape(testFileName), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webClientFileCopyPath+"?path="+
- url.QueryEscape(path.Join(testDir, testFileName))+"&target="+url.QueryEscape(testFileName), nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestStartDirectory(t *testing.T) {
- u := getTestUser()
- u.Filters.StartDirectory = "/start/dir"
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- filename := "file1.txt"
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part1, err := writer.CreateFormFile("filenames", filename)
- assert.NoError(t, err)
- _, err = part1.Write([]byte("test content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- // check we have 2 files in the defined start dir
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var contents []map[string]any
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- if assert.Len(t, contents, 1) {
- assert.Equal(t, filename, contents[0]["name"].(string))
- }
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=file2.txt",
- bytes.NewBuffer([]byte("single upload content")))
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path=testdir", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path=testdir&target=testdir1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path=%2Ftestdirroot", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath+"?path="+url.QueryEscape(u.Filters.StartDirectory), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 3)
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+filename, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path=%2F"+filename, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPatch, userFilesPath+"?path="+filename+"&target="+filename+"_rename", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path=testdir1", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- req, err = http.NewRequest(http.MethodGet, webClientDirsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 2)
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path="+filename+"_rename", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath+"?path="+url.QueryEscape(u.Filters.StartDirectory), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- contents = nil
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 1)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebFilesTransferQuotaLimits(t *testing.T) {
- u := getTestUser()
- u.UploadDataTransfer = 1
- u.DownloadDataTransfer = 1
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- testFileName := "file.data"
- testFileSize := 550000
- testFileContents := make([]byte, testFileSize)
- n, err := io.ReadFull(rand.Reader, testFileContents)
- assert.NoError(t, err)
- assert.Equal(t, testFileSize, n)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", testFileName)
- assert.NoError(t, err)
- _, err = part.Write(testFileContents)
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, testFileContents, rr.Body.Bytes())
- // error while download is active
- downloadFunc := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- downloadFunc()
- // error before starting the download
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path="+testFileName, nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- // error while upload is active
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestEntityTooLarge, rr)
- // error before starting the upload
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestEntityTooLarge, rr)
- // now test upload/download to/from shares
- share1 := dataprovider.Share{
- Name: "share1",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- }
- asJSON, err := json.Marshal(share1)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- req, err = http.NewRequest(http.MethodGet, sharesPath+"/"+objectID, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- form := make(url.Values)
- form.Set("files", `[]`)
- req, err = http.NewRequest(http.MethodPost, path.Join(webClientPubSharesPath, objectID, "/partial"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorQuotaRead)
- share2 := dataprovider.Share{
- Name: "share2",
- Scope: dataprovider.ShareScopeWrite,
- Paths: []string{"/"},
- }
- asJSON, err = json.Marshal(share2)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userSharesPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- objectID = rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, sharesPath+"/"+objectID, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestEntityTooLarge, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebUploadErrors(t *testing.T) {
- u := getTestUser()
- u.QuotaSize = 65535
- subDir1 := "sub1"
- subDir2 := "sub2"
- u.Permissions[path.Join("/", subDir1)] = []string{dataprovider.PermListItems}
- u.Permissions[path.Join("/", subDir2)] = []string{dataprovider.PermListItems, dataprovider.PermUpload,
- dataprovider.PermDelete}
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/sub2",
- AllowedPatterns: []string{},
- DeniedPatterns: []string{"*.zip"},
- },
- }
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "file.zip")
- assert.NoError(t, err)
- _, err = part.Write([]byte("file content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- // zip file are not allowed within sub2
- req, err := http.NewRequest(http.MethodPost, userFilesPath+"?path=sub2", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- // we have no upload permissions within sub1
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path=sub1", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- // we cannot create dirs in sub2
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?mkdir_parents=true&path="+url.QueryEscape("/sub2/dir"), reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "unable to check/create missing parent dir")
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?mkdir_parents=true&path="+url.QueryEscape("/sub2/dir/test"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Error checking parent directories")
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?mkdir_parents=true&path="+url.QueryEscape("/sub2/dir1/file.txt"), bytes.NewBuffer([]byte("")))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Error checking parent directories")
- // create a dir and try to overwrite it with a file
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path=file.zip", nil) //nolint:goconst
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "operation unsupported")
- // try to upload to a missing parent directory
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path=missingdir", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path=file.zip", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // upload will work now
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- // overwrite the file
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- vfs.SetTempPath(filepath.Join(os.TempDir(), "missingpath"))
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- if runtime.GOOS != osWindows {
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file.zip", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- vfs.SetTempPath(filepath.Clean(os.TempDir()))
- err = os.Chmod(user.GetHomeDir(), 0555)
- assert.NoError(t, err)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Error closing file")
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=file.zip", bytes.NewBuffer(nil))
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Error closing file")
- err = os.Chmod(user.GetHomeDir(), os.ModePerm)
- assert.NoError(t, err)
- }
- vfs.SetTempPath("")
- // upload a multipart form with no files
- body = new(bytes.Buffer)
- writer = multipart.NewWriter(body)
- err = writer.Close()
- assert.NoError(t, err)
- reader = bytes.NewReader(body.Bytes())
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path=sub2", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "No files uploaded!")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebAPIVFolder(t *testing.T) {
- u := getTestUser()
- u.QuotaSize = 65535
- vdir := "/vdir"
- mappedPath := filepath.Join(os.TempDir(), "vdir")
- folderName := filepath.Base(mappedPath)
- u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
- BaseVirtualFolder: vfs.BaseVirtualFolder{
- Name: folderName,
- },
- VirtualPath: vdir,
- QuotaSize: -1,
- QuotaFiles: -1,
- })
- f := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedPath,
- }
- _, _, err := httpdtest.AddFolder(f, http.StatusCreated)
- assert.NoError(t, err)
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(user.Username, defaultPassword)
- assert.NoError(t, err)
- fileContents := []byte("test contents")
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "file.txt")
- assert.NoError(t, err)
- _, err = part.Write(fileContents)
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath+"?path=vdir", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(len(fileContents)), user.UsedQuotaSize)
- folder, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(len(fileContents)), folder.UsedQuotaSize)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath+"?path=vdir", reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(len(fileContents)), user.UsedQuotaSize)
- folder, _, err = httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, int64(len(fileContents)), folder.UsedQuotaSize)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- err = os.RemoveAll(mappedPath)
- assert.NoError(t, err)
- }
- func TestWebAPIWritePermission(t *testing.T) {
- u := getTestUser()
- u.Filters.WebClient = append(u.Filters.WebClient, sdk.WebClientWriteDisabled)
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "file.txt")
- assert.NoError(t, err)
- _, err = part.Write([]byte(""))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path=a&target=b", nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=a", nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodGet, userFilesPath+"?path=a.txt", nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodPost, userDirsPath+"?path=dir", nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodPost, userFileActionsPath+"/move?path=dir&target=dir1", nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodDelete, userDirsPath+"?path=dir", nil)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebAPICryptFs(t *testing.T) {
- u := getTestUser()
- u.QuotaSize = 65535
- u.FsConfig.Provider = sdk.CryptedFilesystemProvider
- u.FsConfig.CryptConfig.Passphrase = kms.NewPlainSecret(defaultPassword)
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "file.txt")
- assert.NoError(t, err)
- _, err = part.Write([]byte("content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebUploadSFTP(t *testing.T) {
- u := getTestUser()
- localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- u = getTestSFTPUser()
- u.QuotaFiles = 100
- u.FsConfig.SFTPConfig.BufferSize = 2
- u.HomeDir = filepath.Join(os.TempDir(), u.Username)
- sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(sftpUser.Username, defaultPassword)
- assert.NoError(t, err)
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "file.txt")
- assert.NoError(t, err)
- _, err = part.Write([]byte("test file content"))
- assert.NoError(t, err)
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- expectedQuotaSize := int64(17)
- expectedQuotaFiles := 1
- user, _, err := httpdtest.GetUserByUsername(sftpUser.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, expectedQuotaFiles, user.UsedQuotaFiles)
- assert.Equal(t, expectedQuotaSize, user.UsedQuotaSize)
- user.QuotaSize = 10
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- // we are now overquota on overwrite
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestEntityTooLarge, rr)
- assert.Contains(t, rr.Body.String(), "denying write due to space limit")
- assert.Contains(t, rr.Body.String(), "Unable to write file")
- // delete the file
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(sftpUser.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 0, user.UsedQuotaFiles)
- assert.Equal(t, int64(0), user.UsedQuotaSize)
- req, err = http.NewRequest(http.MethodPost, userUploadFilePath+"?path=file.txt",
- bytes.NewBuffer([]byte("test upload single file content")))
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestEntityTooLarge, rr)
- assert.Contains(t, rr.Body.String(), "denying write due to space limit")
- assert.Contains(t, rr.Body.String(), "Error saving file")
- // delete the file
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = reader.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusRequestEntityTooLarge, rr)
- assert.Contains(t, rr.Body.String(), "denying write due to space limit")
- assert.Contains(t, rr.Body.String(), "Error saving file")
- // delete the file
- req, err = http.NewRequest(http.MethodDelete, userFilesPath+"?path=file.txt", nil)
- assert.NoError(t, err)
- setBearerForReq(req, webAPIToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(sftpUser.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, 0, user.UsedQuotaFiles)
- assert.Equal(t, int64(0), user.UsedQuotaSize)
- _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(localUser.GetHomeDir())
- assert.NoError(t, err)
- err = os.RemoveAll(sftpUser.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebAPISFTPPasswordProtectedPrivateKey(t *testing.T) {
- u := getTestUser()
- u.Password = ""
- u.PublicKeys = []string{testPubKeyPwd}
- localUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- u = getTestSFTPUser()
- u.FsConfig.SFTPConfig.Password = kms.NewEmptySecret()
- u.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret(testPrivateKeyPwd)
- u.FsConfig.SFTPConfig.KeyPassphrase = kms.NewPlainSecret(privateKeyPwd)
- u.HomeDir = filepath.Join(os.TempDir(), u.Username)
- sftpUser, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(sftpUser.Username, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // update the user, the key must be preserved
- assert.Equal(t, sdkkms.SecretStatusSecretBox, sftpUser.FsConfig.SFTPConfig.KeyPassphrase.GetStatus())
- _, _, err = httpdtest.UpdateUser(sftpUser, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // using a wrong passphrase or no passphrase should fail
- sftpUser.FsConfig.SFTPConfig.KeyPassphrase = kms.NewPlainSecret("wrong")
- _, _, err = httpdtest.UpdateUser(sftpUser, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(sftpUser.Username, defaultPassword)
- assert.Error(t, err)
- sftpUser.FsConfig.SFTPConfig.KeyPassphrase = kms.NewEmptySecret()
- _, _, err = httpdtest.UpdateUser(sftpUser, http.StatusOK, "")
- assert.NoError(t, err)
- _, err = getJWTWebClientTokenFromTestServer(sftpUser.Username, defaultPassword)
- assert.Error(t, err)
- _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(localUser, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(localUser.GetHomeDir())
- assert.NoError(t, err)
- err = os.RemoveAll(sftpUser.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebUploadMultipartFormReadError(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, userFilesPath, nil)
- assert.NoError(t, err)
- mpartForm := &multipart.Form{
- File: make(map[string][]*multipart.FileHeader),
- }
- mpartForm.File["filenames"] = append(mpartForm.File["filenames"], &multipart.FileHeader{Filename: "missing"})
- req.MultipartForm = mpartForm
- req.Header.Add("Content-Type", "multipart/form-data")
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- assert.Contains(t, rr.Body.String(), "Unable to read uploaded file")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestCompressionErrorMock(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- _, err := httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }()
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("files", `["missing"]`)
- req, _ := http.NewRequest(http.MethodPost, webClientDownloadZipPath+"?path=%2F",
- bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- executeRequest(req)
- }
- func TestGetFilesSFTPBackend(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- u := getTestSFTPUser()
- u.HomeDir = filepath.Clean(os.TempDir())
- u.FsConfig.SFTPConfig.BufferSize = 2
- u.Permissions["/adir"] = nil
- u.Permissions["/adir1"] = []string{dataprovider.PermListItems}
- u.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/adir2",
- DeniedPatterns: []string{"*.txt"},
- },
- }
- sftpUserBuffered, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- u.Username += "_unbuffered"
- u.FsConfig.SFTPConfig.BufferSize = 0
- sftpUserUnbuffered, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "testsftpfile"
- testDir := "testsftpdir"
- testFileContents := []byte("sftp file contents")
- err = os.MkdirAll(filepath.Join(user.GetHomeDir(), testDir, "sub"), os.ModePerm)
- assert.NoError(t, err)
- err = os.MkdirAll(filepath.Join(user.GetHomeDir(), "adir1"), os.ModePerm)
- assert.NoError(t, err)
- err = os.MkdirAll(filepath.Join(user.GetHomeDir(), "adir2"), os.ModePerm)
- assert.NoError(t, err)
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), testFileName), testFileContents, os.ModePerm)
- assert.NoError(t, err)
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), "adir1", "afile"), testFileContents, os.ModePerm)
- assert.NoError(t, err)
- err = os.WriteFile(filepath.Join(user.GetHomeDir(), "adir2", "afile.txt"), testFileContents, os.ModePerm)
- assert.NoError(t, err)
- for _, sftpUser := range []dataprovider.User{sftpUserBuffered, sftpUserUnbuffered} {
- webToken, err := getJWTWebClientTokenFromTestServer(sftpUser.Username, defaultPassword)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webClientFilesPath, nil)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+path.Join(testDir, "sub"), nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- htmlErrFrag := `div id="errorMsg" class="rounded border-warning border border-dashed bg-light-warning`
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+path.Join(testDir, "missing"), nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), htmlErrFrag)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path=adir/sub", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), htmlErrFrag)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path=adir1/afile", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), htmlErrFrag)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path=adir2/afile.txt", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), htmlErrFrag)
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, testFileContents, rr.Body.Bytes())
- req, _ = http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- req.Header.Set("Range", "bytes=2-")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusPartialContent, rr)
- assert.Equal(t, testFileContents[2:], rr.Body.Bytes())
- _, err = httpdtest.RemoveUser(sftpUser, http.StatusOK)
- assert.NoError(t, err)
- }
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestClientUserClose(t *testing.T) {
- u := getTestUser()
- u.UploadBandwidth = 32
- u.DownloadBandwidth = 32
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- testFileName := "file.dat"
- testFileSize := int64(524288)
- testFilePath := filepath.Join(user.GetHomeDir(), testFileName)
- err = createTestFile(testFilePath, testFileSize)
- assert.NoError(t, err)
- uploadContent := make([]byte, testFileSize)
- _, err = rand.Read(uploadContent)
- assert.NoError(t, err)
- webToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- webAPIToken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- var wg sync.WaitGroup
- wg.Add(1)
- go func() {
- defer wg.Done()
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, _ := http.NewRequest(http.MethodGet, webClientFilesPath+"?path="+testFileName, nil)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }()
- wg.Add(1)
- go func() {
- defer wg.Done()
- req, _ := http.NewRequest(http.MethodGet, webClientEditFilePath+"?path="+testFileName, nil)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- }()
- wg.Add(1)
- go func() {
- defer wg.Done()
- body := new(bytes.Buffer)
- writer := multipart.NewWriter(body)
- part, err := writer.CreateFormFile("filenames", "upload.dat")
- assert.NoError(t, err)
- n, err := part.Write(uploadContent)
- assert.NoError(t, err)
- assert.Equal(t, testFileSize, int64(n))
- err = writer.Close()
- assert.NoError(t, err)
- reader := bytes.NewReader(body.Bytes())
- req, err := http.NewRequest(http.MethodPost, userFilesPath, reader)
- assert.NoError(t, err)
- req.Header.Add("Content-Type", writer.FormDataContentType())
- setBearerForReq(req, webAPIToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "transfer aborted")
- }()
- // wait for the transfers
- assert.Eventually(t, func() bool {
- stats := common.Connections.GetStats("")
- if len(stats) == 3 {
- if len(stats[0].Transfers) > 0 && len(stats[1].Transfers) > 0 {
- return true
- }
- }
- return false
- }, 1*time.Second, 50*time.Millisecond)
- for _, stat := range common.Connections.GetStats("") {
- // close all the active transfers
- common.Connections.Close(stat.ConnectionID, "")
- }
- wg.Wait()
- assert.Eventually(t, func() bool { return len(common.Connections.GetStats("")) == 0 },
- 1*time.Second, 100*time.Millisecond)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestWebAdminSetupMock(t *testing.T) {
- req, err := http.NewRequest(http.MethodGet, webAdminSetupPath, nil)
- assert.NoError(t, err)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- // now delete all the admins
- admins, err := dataprovider.GetAdmins(100, 0, dataprovider.OrderASC)
- assert.NoError(t, err)
- for _, admin := range admins {
- err = dataprovider.DeleteAdmin(admin.Username, "", "", "")
- assert.NoError(t, err)
- }
- // close the provider and initializes it without creating the default admin
- os.Setenv("SFTPGO_DATA_PROVIDER__CREATE_DEFAULT_ADMIN", "0")
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- // now the setup page must be rendered
- req, err = http.NewRequest(http.MethodGet, webAdminSetupPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check redirects to the setup page
- req, err = http.NewRequest(http.MethodGet, "/", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminSetupPath, rr.Header().Get("Location"))
- req, err = http.NewRequest(http.MethodGet, webBasePath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminSetupPath, rr.Header().Get("Location"))
- req, err = http.NewRequest(http.MethodGet, webBasePathAdmin, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminSetupPath, rr.Header().Get("Location"))
- req, err = http.NewRequest(http.MethodGet, webLoginPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminSetupPath, rr.Header().Get("Location"))
- req, err = http.NewRequest(http.MethodGet, webClientLoginPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminSetupPath, rr.Header().Get("Location"))
- csrfToken, err := getCSRFToken(httpBaseURL + webAdminSetupPath)
- assert.NoError(t, err)
- form := make(url.Values)
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("username", defaultTokenAuthUser)
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("password", defaultTokenAuthPass)
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdNoMatch)
- form.Set("confirm_password", defaultTokenAuthPass)
- // test a parse form error
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath+"?param=p%C3%AO%GH", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test a dataprovider error
- err = dataprovider.Close()
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // finally initialize the provider and create the default admin
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webAdminMFAPath, rr.Header().Get("Location"))
- // if we resubmit the form we get a bad request, an admin already exists
- req, err = http.NewRequest(http.MethodPost, webAdminSetupPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- os.Setenv("SFTPGO_DATA_PROVIDER__CREATE_DEFAULT_ADMIN", "1")
- }
- func TestAllowList(t *testing.T) {
- configCopy := common.Config
- entries := []dataprovider.IPListEntry{
- {
- IPOrNet: "172.120.1.1/32",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 0,
- },
- {
- IPOrNet: "172.120.1.2/32",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 0,
- },
- {
- IPOrNet: "192.8.7.0/22",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 8,
- },
- }
- for _, e := range entries {
- _, _, err := httpdtest.AddIPListEntry(e, http.StatusCreated)
- assert.NoError(t, err)
- }
- common.Config.MaxTotalConnections = 1
- common.Config.AllowListStatus = 1
- err := common.Initialize(common.Config, 0)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webLoginPath, nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), common.ErrConnectionDenied.Error())
- req.RemoteAddr = "172.120.1.1"
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- testIP := "172.120.1.3"
- req.RemoteAddr = testIP
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), common.ErrConnectionDenied.Error())
- req.RemoteAddr = "192.8.7.1"
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- entry := dataprovider.IPListEntry{
- IPOrNet: "172.120.1.3/32",
- Type: dataprovider.IPListTypeAllowList,
- Mode: dataprovider.ListModeAllow,
- Protocols: 8,
- }
- err = dataprovider.AddIPListEntry(&entry, "", "", "")
- assert.NoError(t, err)
- req.RemoteAddr = testIP
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = dataprovider.DeleteIPListEntry(entry.IPOrNet, entry.Type, "", "", "")
- assert.NoError(t, err)
- req.RemoteAddr = testIP
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), common.ErrConnectionDenied.Error())
- common.Config = configCopy
- err = common.Initialize(common.Config, 0)
- assert.NoError(t, err)
- for _, e := range entries {
- _, err := httpdtest.RemoveIPListEntry(e, http.StatusOK)
- assert.NoError(t, err)
- }
- }
- func TestWebAdminLoginMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webStatusPath+"notfound", nil)
- req.RequestURI = webStatusPath + "notfound"
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webLogoutPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- cookie := rr.Header().Get("Cookie")
- assert.Empty(t, cookie)
- req, _ = http.NewRequest(http.MethodGet, logoutPath, nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, serverStatusPath, nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "Your token is no longer valid")
- req, _ = http.NewRequest(http.MethodGet, webStatusPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- // now try using wrong password
- form := getLoginForm(defaultTokenAuthUser, "wrong pwd", csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- // wrong username
- form = getLoginForm("wrong username", defaultTokenAuthPass, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- // try from an ip not allowed
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Filters.AllowList = []string{"10.0.0.0/8"}
- _, _, err = httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- rAddr := "127.1.1.1:1234"
- csrfToken, err = getCSRFTokenMock(webLoginPath, rAddr)
- assert.NoError(t, err)
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.RemoteAddr = rAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- rAddr = "10.9.9.9:1234"
- csrfToken, err = getCSRFTokenMock(webLoginPath, rAddr)
- assert.NoError(t, err)
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.RemoteAddr = rAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- rAddr = "127.0.1.1:4567"
- csrfToken, err = getCSRFTokenMock(webLoginPath, rAddr)
- assert.NoError(t, err)
- form = getLoginForm(altAdminUsername, altAdminPassword, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.RemoteAddr = rAddr
- req.Header.Set("X-Forwarded-For", "10.9.9.9")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCredentials)
- // invalid csrf token
- form = getLoginForm(altAdminUsername, altAdminPassword, "invalid csrf")
- req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- req.RemoteAddr = "10.9.9.8:1234"
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- req, _ = http.NewRequest(http.MethodGet, webLoginPath, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveAdmin(a, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAdminNoToken(t *testing.T) {
- req, _ := http.NewRequest(http.MethodGet, webAdminProfilePath, nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, webUserPath, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodGet, userPath, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- req, _ = http.NewRequest(http.MethodGet, activeConnectionsPath, nil)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- }
- func TestWebUserShare(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- token, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userAPItoken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "test share",
- Description: "test share desc",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- ExpiresAt: util.GetTimeAsMsSinceEpoch(time.Now().Add(24 * time.Hour)),
- MaxTokens: 100,
- AllowFrom: []string{"127.0.0.0/8", "172.16.0.0/16"},
- Password: defaultPassword,
- }
- form := make(url.Values)
- form.Set("name", share.Name)
- form.Set("scope", strconv.Itoa(int(share.Scope)))
- form.Set("paths[0][path]", "/")
- form.Set("max_tokens", strconv.Itoa(share.MaxTokens))
- form.Set("allowed_ip", strings.Join(share.AllowFrom, ","))
- form.Set("description", share.Description)
- form.Set("password", share.Password)
- form.Set("expiration_date", "123")
- // invalid expiration date
- req, err := http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareExpiration)
- form.Set("expiration_date", util.GetTimeFromMsecSinceEpoch(share.ExpiresAt).UTC().Format("2006-01-02 15:04:05"))
- form.Set("scope", "")
- // invalid scope
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareScope)
- form.Set("scope", strconv.Itoa(int(share.Scope)))
- // invalid max tokens
- form.Set("max_tokens", "t")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareMaxTokens)
- form.Set("max_tokens", strconv.Itoa(share.MaxTokens))
- // no csrf token
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("scope", "100")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareScope)
- form.Set("scope", strconv.Itoa(int(share.Scope)))
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, err = http.NewRequest(http.MethodGet, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userAPItoken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var shares []dataprovider.Share
- err = json.Unmarshal(rr.Body.Bytes(), &shares)
- assert.NoError(t, err)
- if assert.Len(t, shares, 1) {
- s := shares[0]
- assert.Equal(t, share.Name, s.Name)
- assert.Equal(t, share.Description, s.Description)
- assert.Equal(t, share.Scope, s.Scope)
- assert.Equal(t, share.Paths, s.Paths)
- assert.InDelta(t, share.ExpiresAt, s.ExpiresAt, 999)
- assert.Equal(t, share.MaxTokens, s.MaxTokens)
- assert.Equal(t, share.AllowFrom, s.AllowFrom)
- assert.Equal(t, redactedSecret, s.Password)
- share.ShareID = s.ShareID
- }
- form.Set("password", redactedSecret)
- form.Set("expiration_date", "123")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/unknowid", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/"+share.ShareID, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareExpiration)
- form.Set("expiration_date", "")
- form.Set(csrfFormToken, "")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/"+share.ShareID, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("allowed_ip", "1.1.1")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/"+share.ShareID, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidIPMask)
- form.Set("allowed_ip", "")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/"+share.ShareID, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, err = http.NewRequest(http.MethodGet, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userAPItoken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- shares = nil
- err = json.Unmarshal(rr.Body.Bytes(), &shares)
- assert.NoError(t, err)
- if assert.Len(t, shares, 1) {
- s := shares[0]
- assert.Equal(t, share.Name, s.Name)
- assert.Equal(t, share.Description, s.Description)
- assert.Equal(t, share.Scope, s.Scope)
- assert.Equal(t, share.Paths, s.Paths)
- assert.Equal(t, int64(0), s.ExpiresAt)
- assert.Equal(t, share.MaxTokens, s.MaxTokens)
- assert.Empty(t, s.AllowFrom)
- }
- // check the password
- s, err := dataprovider.ShareExists(share.ShareID, user.Username)
- assert.NoError(t, err)
- match, err := s.CheckCredentials(defaultPassword)
- assert.NoError(t, err)
- assert.True(t, match)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath+"?path=%2F&files=a", nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath+"?path=%2F&files=%5B\"adir\"%5D", nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath+"/unknown", nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath+"/"+share.ShareID, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientSharesPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientSharesPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebUserShareNoPasswordDisabled(t *testing.T) {
- u := getTestUser()
- u.Filters.WebClient = []string{sdk.WebClientShareNoPasswordDisabled}
- u.Filters.DefaultSharesExpiration = 15
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- user.Filters.DefaultSharesExpiration = 30
- user, _, err = httpdtest.UpdateUser(u, http.StatusOK, "")
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- token, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userAPItoken, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- share := dataprovider.Share{
- Name: "s",
- Scope: dataprovider.ShareScopeRead,
- Paths: []string{"/"},
- }
- form := make(url.Values)
- form.Set("name", share.Name)
- form.Set("scope", strconv.Itoa(int(share.Scope)))
- form.Set("paths[0][path]", "/")
- form.Set("max_tokens", "0")
- form.Set(csrfFormToken, csrfToken)
- req, err := http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareNoPwd)
- form.Set("password", defaultPassword)
- req, err = http.NewRequest(http.MethodPost, webClientSharePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, userSharesPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, userAPItoken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var shares []dataprovider.Share
- err = json.Unmarshal(rr.Body.Bytes(), &shares)
- assert.NoError(t, err)
- if assert.Len(t, shares, 1) {
- s := shares[0]
- assert.Equal(t, share.Name, s.Name)
- assert.Equal(t, share.Scope, s.Scope)
- assert.Equal(t, share.Paths, s.Paths)
- share.ShareID = s.ShareID
- }
- assert.NotEmpty(t, share.ShareID)
- form.Set("password", "")
- req, err = http.NewRequest(http.MethodPost, webClientSharePath+"/"+share.ShareID, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorShareNoPwd)
- user.Filters.DefaultSharesExpiration = 0
- user.Filters.MaxSharesExpiration = 30
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientSharePath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebUserProfile(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- assert.NoError(t, err)
- token, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- email := "user@user.com"
- description := "User"
- form := make(url.Values)
- form.Set("allow_api_key_auth", "1")
- form.Set("email", email)
- form.Set("description", description)
- form.Set("public_keys[0][public_key]", testPubKey)
- form.Set("public_keys[1][public_key]", testPubKey1)
- form.Set("tls_certs[0][tls_cert]", httpsCert)
- // no csrf token
- req, err := http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nProfileUpdated)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.AllowAPIKeyAuth)
- assert.Len(t, user.PublicKeys, 2)
- assert.Len(t, user.Filters.TLSCerts, 1)
- assert.Equal(t, email, user.Email)
- assert.Equal(t, description, user.Description)
- // set an invalid email
- form.Set("email", "not an email")
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidEmail)
- // invalid tls cert
- form.Set("email", email)
- form.Set("tls_certs[0][tls_cert]", "not a TLS cert")
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidTLSCert)
- // invalid public key
- form.Set("tls_certs[0][tls_cert]", httpsCert)
- form.Set("public_keys[0][public_key]", "invalid")
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPubKeyInvalid)
- // now remove permissions
- form.Set("public_keys[0][public_key]", testPubKey)
- form.Del("public_keys[1][public_key]")
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- form.Set("allow_api_key_auth", "0")
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nProfileUpdated)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.AllowAPIKeyAuth)
- assert.Len(t, user.PublicKeys, 1)
- assert.Len(t, user.Filters.TLSCerts, 1)
- assert.Equal(t, email, user.Email)
- assert.Equal(t, description, user.Description)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled,
- sdk.WebClientPubKeyChangeDisabled, sdk.WebClientTLSCertChangeDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- form.Set("public_keys[0][public_key]", testPubKey)
- form.Set("public_keys[1][public_key]", testPubKey1)
- form.Set("tls_certs[0][tls_cert]", "")
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nProfileUpdated)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.AllowAPIKeyAuth)
- assert.Len(t, user.PublicKeys, 1)
- assert.Len(t, user.Filters.TLSCerts, 1)
- assert.Equal(t, email, user.Email)
- assert.Equal(t, description, user.Description)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientInfoChangeDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- form.Set("email", "newemail@user.com")
- form.Set("description", "new description")
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nProfileUpdated)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.AllowAPIKeyAuth)
- assert.Len(t, user.PublicKeys, 2)
- assert.Len(t, user.Filters.TLSCerts, 0)
- assert.Equal(t, email, user.Email)
- assert.Equal(t, description, user.Description)
- // finally disable all profile permissions
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientInfoChangeDisabled,
- sdk.WebClientPubKeyChangeDisabled, sdk.WebClientTLSCertChangeDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- token, err = getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webClientProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- }
- func TestWebAdminProfile(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTWebTokenFromTestServer(admin.Username, altAdminPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webAdminProfilePath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form := make(url.Values)
- form.Set("allow_api_key_auth", "1")
- form.Set("email", "admin@example.com")
- form.Set("description", "admin desc")
- // no csrf token
- req, err = http.NewRequest(http.MethodPost, webAdminProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webAdminProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nProfileUpdated)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.AllowAPIKeyAuth)
- assert.Equal(t, "admin@example.com", admin.Email)
- assert.Equal(t, "admin desc", admin.Description)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webAdminProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nProfileUpdated)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.AllowAPIKeyAuth)
- assert.Empty(t, admin.Email)
- assert.Empty(t, admin.Description)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webAdminProfilePath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- }
- func TestWebAdminPwdChange(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin.Filters.Preferences.HideUserPageSections = 16 + 32
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTWebTokenFromTestServer(admin.Username, altAdminPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webChangeAdminPwdPath, nil)
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form := make(url.Values)
- form.Set("current_password", altAdminPassword)
- form.Set("new_password1", altAdminPassword)
- form.Set("new_password2", altAdminPassword)
- // no csrf token
- req, _ = http.NewRequest(http.MethodPost, webChangeAdminPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, _ = http.NewRequest(http.MethodPost, webChangeAdminPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdNoDifferent)
- form.Set("new_password1", altAdminPassword+"1")
- form.Set("new_password2", altAdminPassword+"1")
- req, _ = http.NewRequest(http.MethodPost, webChangeAdminPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAPIKeysManagement(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Name: "test key",
- Scope: dataprovider.APIKeyScopeAdmin,
- }
- asJSON, err := json.Marshal(apiKey)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, apiKeysPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- location := rr.Header().Get("Location")
- assert.NotEmpty(t, location)
- objectID := rr.Header().Get("X-Object-ID")
- assert.NotEmpty(t, objectID)
- assert.Equal(t, fmt.Sprintf("%v/%v", apiKeysPath, objectID), location)
- apiKey.KeyID = objectID
- response := make(map[string]string)
- err = json.Unmarshal(rr.Body.Bytes(), &response)
- assert.NoError(t, err)
- key := response["key"]
- assert.NotEmpty(t, key)
- assert.True(t, strings.HasPrefix(key, apiKey.KeyID+"."))
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var keyGet dataprovider.APIKey
- err = json.Unmarshal(rr.Body.Bytes(), &keyGet)
- assert.NoError(t, err)
- assert.Empty(t, keyGet.Key)
- assert.Equal(t, apiKey.KeyID, keyGet.KeyID)
- assert.Equal(t, apiKey.Scope, keyGet.Scope)
- assert.Equal(t, apiKey.Name, keyGet.Name)
- assert.Equal(t, int64(0), keyGet.ExpiresAt)
- assert.Equal(t, int64(0), keyGet.LastUseAt)
- assert.Greater(t, keyGet.CreatedAt, int64(0))
- assert.Greater(t, keyGet.UpdatedAt, int64(0))
- assert.Empty(t, keyGet.Description)
- assert.Empty(t, keyGet.User)
- assert.Empty(t, keyGet.Admin)
- // API key is not enabled for the admin user so this request should fail
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, admin.Username)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "the admin associated with the provided api key cannot be authenticated")
- admin.Filters.AllowAPIKeyAuth = true
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, admin.Username)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, admin.Username+"1")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- // now associate the key directly to the admin
- apiKey.Admin = admin.Username
- apiKey.Description = "test description"
- asJSON, err = json.Marshal(apiKey)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, apiKeysPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var keys []dataprovider.APIKey
- err = json.Unmarshal(rr.Body.Bytes(), &keys)
- assert.NoError(t, err)
- if assert.GreaterOrEqual(t, len(keys), 1) {
- found := false
- for _, k := range keys {
- if k.KeyID == apiKey.KeyID {
- found = true
- assert.Empty(t, k.Key)
- assert.Equal(t, apiKey.Scope, k.Scope)
- assert.Equal(t, apiKey.Name, k.Name)
- assert.Equal(t, int64(0), k.ExpiresAt)
- assert.Greater(t, k.LastUseAt, int64(0))
- assert.Equal(t, k.CreatedAt, keyGet.CreatedAt)
- assert.Greater(t, k.UpdatedAt, keyGet.UpdatedAt)
- assert.Equal(t, apiKey.Description, k.Description)
- assert.Empty(t, k.User)
- assert.Equal(t, admin.Username, k.Admin)
- }
- }
- assert.True(t, found)
- }
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // invalid API keys
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key+"invalid", "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- assert.Contains(t, rr.Body.String(), "the provided api key cannot be authenticated")
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, "invalid", "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // using an API key we cannot modify/get API keys
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- admin.Filters.AllowList = []string{"172.16.18.0/24"}
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- req, err = http.NewRequest(http.MethodDelete, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, versionPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "the provided api key is not valid")
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAPIKeySearch(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Scope: dataprovider.APIKeyScopeAdmin,
- }
- for i := 1; i < 5; i++ {
- apiKey.Name = fmt.Sprintf("testapikey%v", i)
- asJSON, err := json.Marshal(apiKey)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, apiKeysPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- }
- req, err := http.NewRequest(http.MethodGet, apiKeysPath+"?limit=1&order=ASC", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var keys []dataprovider.APIKey
- err = json.Unmarshal(rr.Body.Bytes(), &keys)
- assert.NoError(t, err)
- assert.Len(t, keys, 1)
- firstKey := keys[0]
- req, err = http.NewRequest(http.MethodGet, apiKeysPath+"?limit=1&order=DESC", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- keys = nil
- err = json.Unmarshal(rr.Body.Bytes(), &keys)
- assert.NoError(t, err)
- if assert.Len(t, keys, 1) {
- assert.NotEqual(t, firstKey.KeyID, keys[0].KeyID)
- }
- req, err = http.NewRequest(http.MethodGet, apiKeysPath+"?limit=1&offset=100", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- keys = nil
- err = json.Unmarshal(rr.Body.Bytes(), &keys)
- assert.NoError(t, err)
- assert.Len(t, keys, 0)
- req, err = http.NewRequest(http.MethodGet, apiKeysPath+"?limit=f", nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("%v/%v", apiKeysPath, "missingid"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, apiKeysPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- keys = nil
- err = json.Unmarshal(rr.Body.Bytes(), &keys)
- assert.NoError(t, err)
- counter := 0
- for _, k := range keys {
- if strings.HasPrefix(k.Name, "testapikey") {
- req, err = http.NewRequest(http.MethodDelete, fmt.Sprintf("%v/%v", apiKeysPath, k.KeyID), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- counter++
- }
- }
- assert.Equal(t, 4, counter)
- }
- func TestAPIKeyErrors(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Name: "testkey",
- Scope: dataprovider.APIKeyScopeUser,
- }
- asJSON, err := json.Marshal(apiKey)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, apiKeysPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- location := rr.Header().Get("Location")
- assert.NotEmpty(t, location)
- // invalid API scope
- apiKey.Scope = 1000
- asJSON, err = json.Marshal(apiKey)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, apiKeysPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // invalid JSON
- req, err = http.NewRequest(http.MethodPost, apiKeysPath, bytes.NewBuffer([]byte(`invalid JSON`)))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer([]byte(`invalid JSON`)))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodDelete, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodDelete, location, nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodPut, location, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestAPIKeyOnDeleteCascade(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin, _, err = httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- apiKey := dataprovider.APIKey{
- Name: "user api key",
- Scope: dataprovider.APIKeyScopeUser,
- User: user.Username,
- }
- apiKey, _, err = httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusUnauthorized, rr)
- user.Filters.AllowAPIKeyAuth = true
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, userDirsPath, nil)
- assert.NoError(t, err)
- setAPIKeyForReq(req, apiKey.Key, "")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var contents []map[string]any
- err = json.NewDecoder(rr.Body).Decode(&contents)
- assert.NoError(t, err)
- assert.Len(t, contents, 0)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusNotFound)
- assert.NoError(t, err)
- apiKey.User = ""
- apiKey.Admin = admin.Username
- apiKey.Scope = dataprovider.APIKeyScopeAdmin
- apiKey, _, err = httpdtest.AddAPIKey(apiKey, http.StatusCreated)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusNotFound)
- assert.NoError(t, err)
- }
- func TestBasicWebUsersMock(t *testing.T) {
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user1 := getTestUser()
- user1.Username += "1"
- user1AsJSON := getUserAsJSON(t, user1)
- req, _ = http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(user1AsJSON))
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user1)
- assert.NoError(t, err)
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodGet, webUsersPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webUsersPath+jsonAPISuffix, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webUserPath, nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(webUserPath, user.Username), nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webUserPath+"/0", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("username", user.Username)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath+"/0", &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodPost, webUserPath+"/aaa", &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webUserPath, user.Username), nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webUserPath, user.Username), nil)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webUserPath, user1.Username), nil)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestRenderDefenderPageMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webDefenderPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nDefenderTitle)
- }
- func TestWebAdminBasicMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("username", admin.Username)
- form.Set("password", "")
- form.Set("status", "1")
- form.Set("permissions", "*")
- form.Set("description", admin.Description)
- form.Set("user_page_hidden_sections", "1")
- form.Add("user_page_hidden_sections", "2")
- form.Add("user_page_hidden_sections", "3")
- form.Add("user_page_hidden_sections", "4")
- form.Add("user_page_hidden_sections", "5")
- form.Add("user_page_hidden_sections", "6")
- form.Add("user_page_hidden_sections", "7")
- form.Set("default_users_expiration", "10")
- req, _ := http.NewRequest(http.MethodPost, webAdminPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("status", "a")
- req, _ = http.NewRequest(http.MethodPost, webAdminPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("status", "1")
- form.Set("default_users_expiration", "a")
- req, _ = http.NewRequest(http.MethodPost, webAdminPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("default_users_expiration", "10")
- req, _ = http.NewRequest(http.MethodPost, webAdminPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("password", admin.Password)
- req, _ = http.NewRequest(http.MethodPost, webAdminPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // add TOTP config
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], altAdminUsername)
- assert.NoError(t, err)
- altToken, err := getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- adminTOTPConfig := dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- }
- asJSON, err := json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- // no CSRF token
- req, err = http.NewRequest(http.MethodPost, webAdminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- req, err = http.NewRequest(http.MethodPost, webAdminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, altToken)
- setCSRFHeaderForReq(req, csrfToken)
- req.RemoteAddr = defaultRemoteAddr
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- secretPayload := admin.Filters.TOTPConfig.Secret.GetPayload()
- assert.NotEmpty(t, secretPayload)
- assert.Equal(t, 1+2+4+8+16+32+64, admin.Filters.Preferences.HideUserPageSections)
- assert.Equal(t, 10, admin.Filters.Preferences.DefaultUsersExpiration)
- adminTOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewEmptySecret(),
- }
- asJSON, err = json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webAdminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, altToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Equal(t, secretPayload, admin.Filters.TOTPConfig.Secret.GetPayload())
- adminTOTPConfig = dataprovider.AdminTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: nil,
- }
- asJSON, err = json.Marshal(adminTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webAdminTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, altToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webAdminsPath+jsonAPISuffix, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webAdminsPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webAdminPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("password", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, altAdminUsername), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- form.Set(csrfFormToken, "invalid csrf")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, altAdminUsername), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- form.Set("email", "not-an-email")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, altAdminUsername), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("email", "")
- form.Set("status", "b")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, altAdminUsername), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("email", "admin@example.com")
- form.Set("status", "0")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, altAdminUsername), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- admin, _, err = httpdtest.GetAdminByUsername(altAdminUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, admin.Filters.TOTPConfig.Enabled)
- assert.Equal(t, "admin@example.com", admin.Email)
- assert.Equal(t, 0, admin.Status)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, altAdminUsername+"1"), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(webAdminPath, altAdminUsername), nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(webAdminPath, altAdminUsername+"1"), nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodGet, webUserPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webAdminPath, altAdminUsername), nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webUserPath, nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusNotFound)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webAdminPath, defaultTokenAuthUser), nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "you cannot delete yourself")
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webAdminPath, defaultTokenAuthUser), nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- }
- func TestWebAdminGroupsMock(t *testing.T) {
- group1 := getTestGroup()
- group1.Name += "_1"
- group1, _, err := httpdtest.AddGroup(group1, http.StatusCreated)
- assert.NoError(t, err)
- group2 := getTestGroup()
- group2.Name += "_2"
- group2, _, err = httpdtest.AddGroup(group2, http.StatusCreated)
- assert.NoError(t, err)
- group3 := getTestGroup()
- group3.Name += "_3"
- group3, _, err = httpdtest.AddGroup(group3, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", admin.Username)
- form.Set("password", "")
- form.Set("status", "1")
- form.Set("permissions", "*")
- form.Set("description", admin.Description)
- form.Set("password", admin.Password)
- form.Set("groups[0][group]", group1.Name)
- form.Set("groups[0][group_type]", "1")
- form.Set("groups[1][group]", group2.Name)
- form.Set("groups[1][group_type]", "2")
- form.Set("groups[2][group]", group3.Name)
- req, err := http.NewRequest(http.MethodPost, webAdminPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- if assert.Len(t, admin.Groups, 3) {
- for _, g := range admin.Groups {
- switch g.Name {
- case group1.Name:
- assert.Equal(t, dataprovider.GroupAddToUsersAsPrimary, g.Options.AddToUsersAs)
- case group2.Name:
- assert.Equal(t, dataprovider.GroupAddToUsersAsSecondary, g.Options.AddToUsersAs)
- case group3.Name:
- assert.Equal(t, dataprovider.GroupAddToUsersAsMembership, g.Options.AddToUsersAs)
- default:
- t.Errorf("unexpected group %q", g.Name)
- }
- }
- }
- adminToken, err := getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webUserPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webUserPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, adminToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- _, err = httpdtest.RemoveGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group3, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebAdminPermissions(t *testing.T) {
- admin := getTestAdmin()
- admin.Username = altAdminUsername
- admin.Password = altAdminPassword
- admin.Permissions = []string{dataprovider.PermAdminAddUsers}
- _, _, err := httpdtest.AddAdmin(admin, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTWebToken(altAdminUsername, altAdminPassword)
- require.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, httpBaseURL+webUserPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err := httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusOK, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, httpBaseURL+path.Join(webUserPath, "auser"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, httpBaseURL+webFolderPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, httpBaseURL+webStatusPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, httpBaseURL+webConnectionsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, httpBaseURL+webAdminPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- req, err = http.NewRequest(http.MethodGet, httpBaseURL+path.Join(webAdminPath, "a"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- resp, err = httpclient.GetHTTPClient().Do(req)
- require.NoError(t, err)
- defer resp.Body.Close()
- assert.Equal(t, http.StatusForbidden, resp.StatusCode)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestAdminUpdateSelfMock(t *testing.T) {
- admin, _, err := httpdtest.GetAdminByUsername(defaultTokenAuthUser, http.StatusOK)
- assert.NoError(t, err)
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("username", admin.Username)
- form.Set("password", admin.Password)
- form.Set("status", "0")
- form.Set("permissions", dataprovider.PermAdminAddUsers)
- form.Set("permissions", dataprovider.PermAdminCloseConnections)
- form.Set(csrfFormToken, csrfToken)
- req, _ := http.NewRequest(http.MethodPost, path.Join(webAdminPath, defaultTokenAuthUser), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorAdminSelfPerms)
- form.Set("permissions", dataprovider.PermAdminAny)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, defaultTokenAuthUser), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorAdminSelfDisable)
- form.Set("status", "1")
- form.Set("require_two_factor", "1")
- form.Set("require_password_change", "1")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, defaultTokenAuthUser), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- admin, _, err = httpdtest.GetAdminByUsername(defaultTokenAuthUser, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.RequirePasswordChange)
- assert.False(t, admin.Filters.RequireTwoFactor)
- form.Set("role", "my role")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, defaultTokenAuthUser), bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorAdminSelfRole)
- }
- func TestWebMaintenanceMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webMaintenancePath, nil)
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("mode", "a")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("mode", "0")
- form.Set("quota", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("quota", "0")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodPost, webRestorePath+"?a=%3", &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- backupFilePath := filepath.Join(os.TempDir(), "backup.json")
- err = createTestFile(backupFilePath, 0)
- assert.NoError(t, err)
- b, contentType, _ = getMultipartFormData(form, "backup_file", backupFilePath)
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = createTestFile(backupFilePath, 10)
- assert.NoError(t, err)
- b, contentType, _ = getMultipartFormData(form, "backup_file", backupFilePath)
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user := getTestUser()
- user.ID = 1
- user.Username = "test_user_web_restore"
- admin := getTestAdmin()
- admin.ID = 1
- admin.Username = "test_admin_web_restore"
- apiKey := dataprovider.APIKey{
- Name: "key name",
- KeyID: util.GenerateUniqueID(),
- Key: fmt.Sprintf("%v.%v", util.GenerateUniqueID(), util.GenerateUniqueID()),
- Scope: dataprovider.APIKeyScopeAdmin,
- }
- backupData := dataprovider.BackupData{
- Version: dataprovider.DumpVersion,
- }
- backupData.Users = append(backupData.Users, user)
- backupData.Admins = append(backupData.Admins, admin)
- backupData.APIKeys = append(backupData.APIKeys, apiKey)
- backupContent, err := json.Marshal(backupData)
- assert.NoError(t, err)
- err = os.WriteFile(backupFilePath, backupContent, os.ModePerm)
- assert.NoError(t, err)
- b, contentType, _ = getMultipartFormData(form, "backup_file", backupFilePath)
- req, _ = http.NewRequest(http.MethodPost, webRestorePath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nBackupOK)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetAPIKeyByID(apiKey.KeyID, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAPIKey(apiKey, http.StatusOK)
- assert.NoError(t, err)
- err = os.Remove(backupFilePath)
- assert.NoError(t, err)
- }
- func TestWebUserAddMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- group1 := getTestGroup()
- group1.Name += "_1"
- group1, _, err = httpdtest.AddGroup(group1, http.StatusCreated)
- assert.NoError(t, err)
- group2 := getTestGroup()
- group2.Name += "_2"
- group2, _, err = httpdtest.AddGroup(group2, http.StatusCreated)
- assert.NoError(t, err)
- group3 := getTestGroup()
- group3.Name += "_3"
- group3, _, err = httpdtest.AddGroup(group3, http.StatusCreated)
- assert.NoError(t, err)
- user := getTestUser()
- user.UploadBandwidth = 32
- user.DownloadBandwidth = 64
- user.UploadDataTransfer = 1000
- user.DownloadDataTransfer = 2000
- user.UID = 1000
- user.AdditionalInfo = "info"
- user.Description = "user dsc"
- user.Email = "test@test.com"
- mappedDir := filepath.Join(os.TempDir(), "mapped")
- folderName := filepath.Base(mappedDir)
- f := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: mappedDir,
- }
- folderAsJSON, err := json.Marshal(f)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodPost, folderPath, bytes.NewBuffer(folderAsJSON))
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("email", user.Email)
- form.Set("home_dir", user.HomeDir)
- form.Set("osfs_read_buffer_size", "2")
- form.Set("osfs_write_buffer_size", "3")
- form.Set("password", user.Password)
- form.Set("primary_group", group1.Name)
- form.Set("secondary_groups", group2.Name)
- form.Set("membership_groups", group3.Name)
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "")
- form.Set("permissions", "*")
- form.Set("directory_permissions[0][sub_perm_path]", "/subdir")
- form.Set("directory_permissions[0][sub_perm_permissions][]", "list")
- form.Add("directory_permissions[0][sub_perm_permissions][]", "download")
- form.Set("virtual_folders[0][vfolder_path]", " /vdir")
- form.Set("virtual_folders[0][vfolder_name]", folderName)
- form.Set("virtual_folders[0][vfolder_quota_files]", "2")
- form.Set("virtual_folders[0][vfolder_quota_size]", "1024")
- form.Set("directory_patterns[0][pattern_path]", "/dir2")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("directory_patterns[0][pattern_policy]", "1")
- form.Set("directory_patterns[1][pattern_path]", "/dir1")
- form.Set("directory_patterns[1][patterns]", "*.png")
- form.Set("directory_patterns[1][pattern_type]", "allowed")
- form.Set("directory_patterns[2][pattern_path]", "/dir1")
- form.Set("directory_patterns[2][patterns]", "*.zip")
- form.Set("directory_patterns[2][pattern_type]", "denied")
- form.Set("directory_patterns[3][pattern_path]", "/dir3")
- form.Set("directory_patterns[3][patterns]", "*.rar")
- form.Set("directory_patterns[3][pattern_type]", "denied")
- form.Set("directory_patterns[4][pattern_path]", "/dir2")
- form.Set("directory_patterns[4][patterns]", "*.mkv")
- form.Set("directory_patterns[4][pattern_type]", "denied")
- form.Set("access_time_restrictions[0][access_time_day_of_week]", "2")
- form.Set("access_time_restrictions[0][access_time_start]", "10") // invalid and no end, ignored
- form.Set("access_time_restrictions[1][access_time_day_of_week]", "3")
- form.Set("access_time_restrictions[1][access_time_start]", "12:00")
- form.Set("access_time_restrictions[1][access_time_end]", "14:09")
- form.Set("additional_info", user.AdditionalInfo)
- form.Set("description", user.Description)
- form.Add("hooks", "external_auth_disabled")
- form.Set("disable_fs_checks", "checked")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("start_directory", "start/dir")
- form.Set("require_password_change", "1")
- b, contentType, _ := getMultipartFormData(form, "", "")
- // test invalid url escape
- req, _ = http.NewRequest(http.MethodPost, webUserPath+"?a=%2", &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("public_keys", testPubKey)
- form.Add("public_keys", testPubKey1)
- form.Set("uid", strconv.FormatInt(int64(user.UID), 10))
- form.Set("gid", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid gid
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("gid", "0")
- form.Set("max_sessions", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid max sessions
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("max_sessions", "0")
- form.Set("quota_size", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid quota size
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("quota_size", "0")
- form.Set("quota_files", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid quota files
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("quota_files", "0")
- form.Set("upload_bandwidth", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid upload bandwidth
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("upload_bandwidth", strconv.FormatInt(user.UploadBandwidth, 10))
- form.Set("download_bandwidth", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid download bandwidth
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("download_bandwidth", strconv.FormatInt(user.DownloadBandwidth, 10))
- form.Set("upload_data_transfer", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid upload data transfer
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("upload_data_transfer", strconv.FormatInt(user.UploadDataTransfer, 10))
- form.Set("download_data_transfer", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid download data transfer
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("download_data_transfer", strconv.FormatInt(user.DownloadDataTransfer, 10))
- form.Set("total_data_transfer", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid total data transfer
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("total_data_transfer", strconv.FormatInt(user.TotalDataTransfer, 10))
- form.Set("status", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid status
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "123")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid expiration date
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("expiration_date", "")
- form.Set("allowed_ip", "invalid,ip")
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid allowed_ip
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "192.168.1.2") // it should be 192.168.1.2/32
- b, contentType, _ = getMultipartFormData(form, "", "")
- // test invalid denied_ip
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("denied_ip", "")
- // test invalid max file upload size
- form.Set("max_upload_file_size", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("max_upload_file_size", "1 KB")
- // test invalid default shares expiration
- form.Set("default_shares_expiration", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("default_shares_expiration", "10")
- // test invalid max shares expiration
- form.Set("max_shares_expiration", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("max_shares_expiration", "30")
- // test invalid password expiration
- form.Set("password_expiration", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("password_expiration", "90")
- // test invalid password strength
- form.Set("password_strength", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("password_strength", "60")
- // test invalid tls username
- form.Set("tls_username", "username")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("tls_username", string(sdk.TLSUsernameNone))
- // invalid upload bandwidth source
- form.Set("src_bandwidth_limits[0][bandwidth_limit_sources]", "192.168.1.0/24, 192.168.2.0/25")
- form.Set("src_bandwidth_limits[0][upload_bandwidth_source]", "a")
- form.Set("src_bandwidth_limits[0][download_bandwidth_source]", "0")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- // invalid download bandwidth source
- form.Set("src_bandwidth_limits[0][upload_bandwidth_source]", "256")
- form.Set("src_bandwidth_limits[0][download_bandwidth_source]", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("src_bandwidth_limits[0][download_bandwidth_source]", "512")
- form.Set("src_bandwidth_limits[1][download_bandwidth_source]", "1024")
- form.Set("src_bandwidth_limits[1][bandwidth_limit_sources]", "1.1.1")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorSourceBWLimitInvalid)
- form.Set("src_bandwidth_limits[1][bandwidth_limit_sources]", "127.0.0.1/32")
- form.Set("src_bandwidth_limits[1][upload_bandwidth_source]", "-1")
- // invalid external auth cache size
- form.Set("external_auth_cache_time", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("external_auth_cache_time", "0")
- form.Set(csrfFormToken, "invalid form token")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- dbUser, err := dataprovider.UserExists(user.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- // the user already exists, was created with the above request
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- newUser := dataprovider.User{}
- err = render.DecodeJSON(rr.Body, &newUser)
- assert.NoError(t, err)
- assert.Equal(t, user.UID, newUser.UID)
- assert.Equal(t, 2, newUser.FsConfig.OSConfig.ReadBufferSize)
- assert.Equal(t, 3, newUser.FsConfig.OSConfig.WriteBufferSize)
- assert.Equal(t, user.UploadBandwidth, newUser.UploadBandwidth)
- assert.Equal(t, user.DownloadBandwidth, newUser.DownloadBandwidth)
- assert.Equal(t, user.UploadDataTransfer, newUser.UploadDataTransfer)
- assert.Equal(t, user.DownloadDataTransfer, newUser.DownloadDataTransfer)
- assert.Equal(t, user.TotalDataTransfer, newUser.TotalDataTransfer)
- assert.Equal(t, int64(1000), newUser.Filters.MaxUploadFileSize)
- assert.Equal(t, user.AdditionalInfo, newUser.AdditionalInfo)
- assert.Equal(t, user.Description, newUser.Description)
- assert.True(t, newUser.Filters.Hooks.ExternalAuthDisabled)
- assert.False(t, newUser.Filters.Hooks.PreLoginDisabled)
- assert.False(t, newUser.Filters.Hooks.CheckPasswordDisabled)
- assert.True(t, newUser.Filters.DisableFsChecks)
- assert.False(t, newUser.Filters.AllowAPIKeyAuth)
- assert.Equal(t, user.Email, newUser.Email)
- assert.Equal(t, "/start/dir", newUser.Filters.StartDirectory)
- assert.Equal(t, 0, newUser.Filters.FTPSecurity)
- assert.Equal(t, 10, newUser.Filters.DefaultSharesExpiration)
- assert.Equal(t, 30, newUser.Filters.MaxSharesExpiration)
- assert.Equal(t, 90, newUser.Filters.PasswordExpiration)
- assert.Equal(t, 60, newUser.Filters.PasswordStrength)
- assert.Greater(t, newUser.LastPasswordChange, int64(0))
- assert.True(t, newUser.Filters.RequirePasswordChange)
- assert.True(t, util.Contains(newUser.PublicKeys, testPubKey))
- if val, ok := newUser.Permissions["/subdir"]; ok {
- assert.True(t, util.Contains(val, dataprovider.PermListItems))
- assert.True(t, util.Contains(val, dataprovider.PermDownload))
- } else {
- assert.Fail(t, "user permissions must contain /somedir", "actual: %v", newUser.Permissions)
- }
- assert.Len(t, newUser.PublicKeys, 2)
- assert.Len(t, newUser.VirtualFolders, 1)
- for _, v := range newUser.VirtualFolders {
- assert.Equal(t, v.VirtualPath, "/vdir")
- assert.Equal(t, v.Name, folderName)
- assert.Equal(t, v.MappedPath, mappedDir)
- assert.Equal(t, v.QuotaFiles, 2)
- assert.Equal(t, v.QuotaSize, int64(1024))
- }
- assert.Len(t, newUser.Filters.FilePatterns, 3)
- for _, filter := range newUser.Filters.FilePatterns {
- switch filter.Path {
- case "/dir1":
- assert.Len(t, filter.DeniedPatterns, 1)
- assert.Len(t, filter.AllowedPatterns, 1)
- assert.True(t, util.Contains(filter.AllowedPatterns, "*.png"))
- assert.True(t, util.Contains(filter.DeniedPatterns, "*.zip"))
- assert.Equal(t, sdk.DenyPolicyDefault, filter.DenyPolicy)
- case "/dir2":
- assert.Len(t, filter.DeniedPatterns, 1)
- assert.Len(t, filter.AllowedPatterns, 2)
- assert.True(t, util.Contains(filter.AllowedPatterns, "*.jpg"))
- assert.True(t, util.Contains(filter.AllowedPatterns, "*.png"))
- assert.True(t, util.Contains(filter.DeniedPatterns, "*.mkv"))
- assert.Equal(t, sdk.DenyPolicyHide, filter.DenyPolicy)
- case "/dir3":
- assert.Len(t, filter.DeniedPatterns, 1)
- assert.Len(t, filter.AllowedPatterns, 0)
- assert.True(t, util.Contains(filter.DeniedPatterns, "*.rar"))
- assert.Equal(t, sdk.DenyPolicyDefault, filter.DenyPolicy)
- }
- }
- if assert.Len(t, newUser.Filters.BandwidthLimits, 2) {
- for _, bwLimit := range newUser.Filters.BandwidthLimits {
- if len(bwLimit.Sources) == 2 {
- assert.Equal(t, "192.168.1.0/24", bwLimit.Sources[0])
- assert.Equal(t, "192.168.2.0/25", bwLimit.Sources[1])
- assert.Equal(t, int64(256), bwLimit.UploadBandwidth)
- assert.Equal(t, int64(512), bwLimit.DownloadBandwidth)
- } else {
- assert.Equal(t, []string{"127.0.0.1/32"}, bwLimit.Sources)
- assert.Equal(t, int64(0), bwLimit.UploadBandwidth)
- assert.Equal(t, int64(1024), bwLimit.DownloadBandwidth)
- }
- }
- }
- if assert.Len(t, newUser.Filters.AccessTime, 1) {
- assert.Equal(t, 3, newUser.Filters.AccessTime[0].DayOfWeek)
- assert.Equal(t, "12:00", newUser.Filters.AccessTime[0].From)
- assert.Equal(t, "14:09", newUser.Filters.AccessTime[0].To)
- }
- assert.Len(t, newUser.Groups, 3)
- assert.Equal(t, sdk.TLSUsernameNone, newUser.Filters.TLSUsername)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, newUser.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- _, err = httpdtest.RemoveGroup(group1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveGroup(group3, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebUserUpdateMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- user.Filters.BandwidthLimits = []sdk.BandwidthLimit{
- {
- Sources: []string{"10.8.0.0/16", "192.168.1.0/25"},
- UploadBandwidth: 256,
- DownloadBandwidth: 512,
- },
- }
- user.TotalDataTransfer = 4000
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- lastPwdChange := user.LastPasswordChange
- assert.Greater(t, lastPwdChange, int64(0))
- // add TOTP config
- configName, key, _, err := mfa.GenerateTOTPSecret(mfa.GetAvailableTOTPConfigNames()[0], user.Username)
- assert.NoError(t, err)
- userToken, err := getJWTWebClientTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- userTOTPConfig := dataprovider.UserTOTPConfig{
- Enabled: true,
- ConfigName: configName,
- Secret: kms.NewPlainSecret(key.Secret()),
- Protocols: []string{common.ProtocolSSH, common.ProtocolFTP},
- }
- asJSON, err := json.Marshal(userTOTPConfig)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webClientTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, userToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- req, err = http.NewRequest(http.MethodPost, webClientTOTPSavePath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, userToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.True(t, user.Filters.TOTPConfig.Enabled)
- assert.Equal(t, int64(4000), user.TotalDataTransfer)
- assert.Equal(t, lastPwdChange, user.LastPasswordChange)
- if assert.Len(t, user.Filters.BandwidthLimits, 1) {
- if assert.Len(t, user.Filters.BandwidthLimits[0].Sources, 2) {
- assert.Equal(t, "10.8.0.0/16", user.Filters.BandwidthLimits[0].Sources[0])
- assert.Equal(t, "192.168.1.0/25", user.Filters.BandwidthLimits[0].Sources[1])
- }
- assert.Equal(t, int64(256), user.Filters.BandwidthLimits[0].UploadBandwidth)
- assert.Equal(t, int64(512), user.Filters.BandwidthLimits[0].DownloadBandwidth)
- }
- dbUser, err := dataprovider.UserExists(user.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.MaxSessions = 1
- user.QuotaFiles = 2
- user.QuotaSize = 1000 * 1000 * 1000
- user.GID = 1000
- user.Filters.AllowAPIKeyAuth = true
- user.AdditionalInfo = "new additional info"
- user.Email = "user@example.com"
- form := make(url.Values)
- form.Set("username", user.Username)
- form.Set("email", user.Email)
- form.Set("password", "")
- form.Set("public_keys[0][public_key]", testPubKey)
- form.Set("tls_certs[0][tls_cert]", httpsCert)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", "1 GB")
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("permissions", "*")
- form.Set("directory_permissions[0][sub_perm_path]", "/otherdir")
- form.Set("directory_permissions[0][sub_perm_permissions][]", "list")
- form.Add("directory_permissions[0][sub_perm_permissions][]", "upload")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", " 192.168.1.3/32, 192.168.2.0/24 ")
- form.Set("denied_ip", " 10.0.0.2/32 ")
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.zip")
- form.Set("directory_patterns[0][pattern_type]", "denied")
- form.Set("denied_login_methods", dataprovider.SSHLoginMethodKeyboardInteractive)
- form.Set("denied_protocols", common.ProtocolFTP)
- form.Set("max_upload_file_size", "100")
- form.Set("default_shares_expiration", "30")
- form.Set("max_shares_expiration", "60")
- form.Set("password_expiration", "60")
- form.Set("password_strength", "40")
- form.Set("disconnect", "1")
- form.Set("additional_info", user.AdditionalInfo)
- form.Set("description", user.Description)
- form.Set("tls_username", string(sdk.TLSUsernameCN))
- form.Set("allow_api_key_auth", "1")
- form.Set("require_password_change", "1")
- form.Set("external_auth_cache_time", "120")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- dbUser, err = dataprovider.UserExists(user.Username, "")
- assert.NoError(t, err)
- assert.Empty(t, dbUser.Password)
- assert.False(t, dbUser.IsPasswordHashed())
- form.Set("password", defaultPassword)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- dbUser, err = dataprovider.UserExists(user.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- prevPwd := dbUser.Password
- form.Set("password", redactedSecret)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- dbUser, err = dataprovider.UserExists(user.Username, "")
- assert.NoError(t, err)
- assert.NotEmpty(t, dbUser.Password)
- assert.True(t, dbUser.IsPasswordHashed())
- assert.Equal(t, prevPwd, dbUser.Password)
- assert.True(t, dbUser.Filters.TOTPConfig.Enabled)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, user.Email, updateUser.Email)
- assert.Equal(t, user.HomeDir, updateUser.HomeDir)
- assert.Equal(t, user.MaxSessions, updateUser.MaxSessions)
- assert.Equal(t, user.QuotaFiles, updateUser.QuotaFiles)
- assert.Equal(t, user.QuotaSize, updateUser.QuotaSize)
- assert.Equal(t, user.UID, updateUser.UID)
- assert.Equal(t, user.GID, updateUser.GID)
- assert.Equal(t, user.AdditionalInfo, updateUser.AdditionalInfo)
- assert.Equal(t, user.Description, updateUser.Description)
- assert.Equal(t, int64(100), updateUser.Filters.MaxUploadFileSize)
- assert.Equal(t, sdk.TLSUsernameCN, updateUser.Filters.TLSUsername)
- assert.True(t, updateUser.Filters.AllowAPIKeyAuth)
- assert.True(t, updateUser.Filters.TOTPConfig.Enabled)
- assert.Equal(t, int64(0), updateUser.TotalDataTransfer)
- assert.Equal(t, int64(0), updateUser.DownloadDataTransfer)
- assert.Equal(t, int64(0), updateUser.UploadDataTransfer)
- assert.Equal(t, int64(0), updateUser.Filters.ExternalAuthCacheTime)
- assert.Equal(t, 30, updateUser.Filters.DefaultSharesExpiration)
- assert.Equal(t, 60, updateUser.Filters.MaxSharesExpiration)
- assert.Equal(t, 60, updateUser.Filters.PasswordExpiration)
- assert.Equal(t, 40, updateUser.Filters.PasswordStrength)
- assert.True(t, updateUser.Filters.RequirePasswordChange)
- if val, ok := updateUser.Permissions["/otherdir"]; ok {
- assert.True(t, util.Contains(val, dataprovider.PermListItems))
- assert.True(t, util.Contains(val, dataprovider.PermUpload))
- } else {
- assert.Fail(t, "user permissions must contains /otherdir", "actual: %v", updateUser.Permissions)
- }
- assert.True(t, util.Contains(updateUser.Filters.AllowedIP, "192.168.1.3/32"))
- assert.True(t, util.Contains(updateUser.Filters.DeniedIP, "10.0.0.2/32"))
- assert.True(t, util.Contains(updateUser.Filters.DeniedLoginMethods, dataprovider.SSHLoginMethodKeyboardInteractive))
- assert.True(t, util.Contains(updateUser.Filters.DeniedProtocols, common.ProtocolFTP))
- assert.True(t, util.Contains(updateUser.Filters.FilePatterns[0].DeniedPatterns, "*.zip"))
- assert.Len(t, updateUser.Filters.BandwidthLimits, 0)
- assert.Len(t, updateUser.Filters.TLSCerts, 1)
- req, err = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestRenderFolderTemplateMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webTemplateFolder, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- folder := vfs.BaseVirtualFolder{
- Name: "templatefolder",
- MappedPath: filepath.Join(os.TempDir(), "mapped"),
- Description: "template folder desc",
- }
- folder, _, err = httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webTemplateFolder+fmt.Sprintf("?from=%v", folder.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webTemplateFolder+"?from=unknown-folder", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestRenderUserTemplateMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webTemplateUser, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webTemplateUser+fmt.Sprintf("?from=%v", user.Username), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webTemplateUser+"?from=unknown", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserTemplateWithFoldersMock(t *testing.T) {
- folder := vfs.BaseVirtualFolder{
- Name: "vfolder",
- MappedPath: filepath.Join(os.TempDir(), "mapped"),
- Description: "vfolder desc with spéciàl ch@rs",
- }
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- form := make(url.Values)
- form.Set("username", user.Username)
- form.Set("home_dir", filepath.Join(os.TempDir(), "%username%"))
- form.Set("uid", strconv.FormatInt(int64(user.UID), 10))
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("fs_provider", "0")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("ftp_security", "1")
- form.Set("external_auth_cache_time", "0")
- form.Set("description", "desc %username% %password%")
- form.Set("start_directory", "/base/%username%")
- form.Set("vfolder_path", "/vdir%username%")
- form.Set("vfolder_name", folder.Name)
- form.Set("vfolder_quota_size", "-1")
- form.Set("vfolder_quota_files", "-1")
- form.Add("tpl_username", "auser1")
- form.Add("tpl_password", "password1")
- form.Add("tpl_public_keys", " ")
- form.Add("tpl_username", "auser2")
- form.Add("tpl_password", "password2")
- form.Add("tpl_public_keys", testPubKey)
- form.Add("tpl_username", "auser1")
- form.Add("tpl_password", "password")
- form.Add("tpl_public_keys", "")
- form.Set("form_action", "export_from_template")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ := http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- require.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- folder, resp, err := httpdtest.AddFolder(folder, http.StatusCreated)
- assert.NoError(t, err, string(resp))
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var dump dataprovider.BackupData
- err = json.Unmarshal(rr.Body.Bytes(), &dump)
- assert.NoError(t, err)
- assert.Len(t, dump.Users, 2)
- assert.Len(t, dump.Folders, 1)
- user1 := dump.Users[0]
- user2 := dump.Users[1]
- folder1 := dump.Folders[0]
- assert.Equal(t, "auser1", user1.Username)
- assert.Equal(t, "auser2", user2.Username)
- assert.Equal(t, "desc auser1 password1", user1.Description)
- assert.Equal(t, "desc auser2 password2", user2.Description)
- assert.Equal(t, filepath.Join(os.TempDir(), user1.Username), user1.HomeDir)
- assert.Equal(t, filepath.Join(os.TempDir(), user2.Username), user2.HomeDir)
- assert.Equal(t, path.Join("/base", user1.Username), user1.Filters.StartDirectory)
- assert.Equal(t, path.Join("/base", user2.Username), user2.Filters.StartDirectory)
- assert.Equal(t, 0, user2.Filters.DefaultSharesExpiration)
- assert.Equal(t, folder.Name, folder1.Name)
- assert.Len(t, user1.PublicKeys, 0)
- assert.Len(t, user2.PublicKeys, 1)
- assert.Len(t, user1.VirtualFolders, 1)
- assert.Len(t, user2.VirtualFolders, 1)
- assert.Equal(t, "/vdirauser1", user1.VirtualFolders[0].VirtualPath)
- assert.Equal(t, "/vdirauser2", user2.VirtualFolders[0].VirtualPath)
- assert.Equal(t, 1, user1.Filters.FTPSecurity)
- assert.Equal(t, 1, user2.Filters.FTPSecurity)
- _, err = httpdtest.RemoveFolder(folder, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserSaveFromTemplateMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user1 := "u1"
- user2 := "u2"
- form := make(url.Values)
- form.Set("username", "")
- form.Set("home_dir", filepath.Join(os.TempDir(), "%username%"))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("uid", "0")
- form.Set("gid", "0")
- form.Set("max_sessions", "0")
- form.Set("quota_size", "0")
- form.Set("quota_files", "0")
- form.Set("permissions", "*")
- form.Set("status", "1")
- form.Set("expiration_date", "")
- form.Set("fs_provider", "0")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("external_auth_cache_time", "0")
- form.Add("template_users[0][tpl_username]", user1)
- form.Add("template_users[0][tpl_password]", "password1")
- form.Add("template_users[0][tpl_public_keys]", " ")
- form.Add("template_users[1][tpl_username]", user2)
- form.Add("template_users[1][tpl_public_keys]", testPubKey)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ := http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- u1, _, err := httpdtest.GetUserByUsername(user1, http.StatusOK)
- assert.NoError(t, err)
- u2, _, err := httpdtest.GetUserByUsername(user2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(u1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(u2, http.StatusOK)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, err = http.NewRequest(http.MethodPost, webTemplateUser, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestUserTemplateMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- user := getTestUser()
- user.FsConfig.Provider = sdk.S3FilesystemProvider
- user.FsConfig.S3Config.Bucket = "test"
- user.FsConfig.S3Config.Region = "eu-central-1"
- user.FsConfig.S3Config.AccessKey = "%username%"
- user.FsConfig.S3Config.KeyPrefix = "somedir/subdir/"
- user.FsConfig.S3Config.UploadPartSize = 5
- user.FsConfig.S3Config.UploadConcurrency = 4
- user.FsConfig.S3Config.DownloadPartSize = 6
- user.FsConfig.S3Config.DownloadConcurrency = 3
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("home_dir", filepath.Join(os.TempDir(), "%username%"))
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "1")
- form.Set("s3_bucket", user.FsConfig.S3Config.Bucket)
- form.Set("s3_region", user.FsConfig.S3Config.Region)
- form.Set("s3_access_key", "%username%")
- form.Set("s3_access_secret", "%password%")
- form.Set("s3_key_prefix", "base/%username%")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Add("hooks", "external_auth_disabled")
- form.Add("hooks", "check_password_disabled")
- form.Set("disable_fs_checks", "checked")
- form.Set("s3_download_part_max_time", "0")
- form.Set("s3_upload_part_max_time", "0")
- // test invalid s3_upload_part_size
- form.Set("s3_upload_part_size", "a")
- form.Set("form_action", "export_from_template")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ := http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- form.Set("s3_upload_part_size", strconv.FormatInt(user.FsConfig.S3Config.UploadPartSize, 10))
- form.Set("s3_upload_concurrency", strconv.Itoa(user.FsConfig.S3Config.UploadConcurrency))
- form.Set("s3_download_part_size", strconv.FormatInt(user.FsConfig.S3Config.DownloadPartSize, 10))
- form.Set("s3_download_concurrency", strconv.Itoa(user.FsConfig.S3Config.DownloadConcurrency))
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- form.Set("template_users[0][tpl_username]", "user1")
- form.Set("template_users[0][tpl_password]", "password1")
- form.Set("template_users[0][tpl_public_keys]", "invalid-pkey")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- require.Contains(t, rr.Body.String(), util.I18nErrorPubKeyInvalid)
- form.Set("template_users[0][tpl_username]", " ")
- form.Set("template_users[0][tpl_password]", "pwd")
- form.Set("template_users[0][tpl_public_keys]", testPubKey)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- require.Contains(t, rr.Body.String(), util.I18nErrorUserTemplate)
- form.Set("template_users[0][tpl_username]", "user1")
- form.Set("template_users[0][tpl_password]", "password1")
- form.Set("template_users[0][tpl_public_keys]", " ")
- form.Set("template_users[1][tpl_username]", "user2")
- form.Set("template_users[1][tpl_password]", "password2")
- form.Set("template_users[1][tpl_public_keys]", testPubKey)
- form.Set("template_users[2][tpl_username]", "")
- form.Set("template_users[2][tpl_password]", "password3")
- form.Set("template_users[2][tpl_public_keys]", testPubKey)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateUser, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var dump dataprovider.BackupData
- err = json.Unmarshal(rr.Body.Bytes(), &dump)
- require.NoError(t, err)
- require.Len(t, dump.Users, 2)
- require.Len(t, dump.Admins, 0)
- require.Len(t, dump.Folders, 0)
- var user1, user2 dataprovider.User
- for _, u := range dump.Users {
- switch u.Username {
- case "user1":
- user1 = u
- default:
- user2 = u
- }
- }
- require.Equal(t, "user1", user1.Username)
- require.Equal(t, sdk.S3FilesystemProvider, user1.FsConfig.Provider)
- require.Equal(t, "user2", user2.Username)
- require.Equal(t, sdk.S3FilesystemProvider, user2.FsConfig.Provider)
- require.Len(t, user1.PublicKeys, 0)
- require.Len(t, user2.PublicKeys, 1)
- require.Equal(t, filepath.Join(os.TempDir(), user1.Username), user1.HomeDir)
- require.Equal(t, filepath.Join(os.TempDir(), user2.Username), user2.HomeDir)
- require.Equal(t, user1.Username, user1.FsConfig.S3Config.AccessKey)
- require.Equal(t, user2.Username, user2.FsConfig.S3Config.AccessKey)
- require.Equal(t, path.Join("base", user1.Username)+"/", user1.FsConfig.S3Config.KeyPrefix)
- require.Equal(t, path.Join("base", user2.Username)+"/", user2.FsConfig.S3Config.KeyPrefix)
- require.True(t, user1.FsConfig.S3Config.AccessSecret.IsEncrypted())
- err = user1.FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, "password1", user1.FsConfig.S3Config.AccessSecret.GetPayload())
- require.True(t, user2.FsConfig.S3Config.AccessSecret.IsEncrypted())
- err = user2.FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, "password2", user2.FsConfig.S3Config.AccessSecret.GetPayload())
- require.True(t, user1.Filters.Hooks.ExternalAuthDisabled)
- require.True(t, user1.Filters.Hooks.CheckPasswordDisabled)
- require.False(t, user1.Filters.Hooks.PreLoginDisabled)
- require.True(t, user2.Filters.Hooks.ExternalAuthDisabled)
- require.True(t, user2.Filters.Hooks.CheckPasswordDisabled)
- require.False(t, user2.Filters.Hooks.PreLoginDisabled)
- require.True(t, user1.Filters.DisableFsChecks)
- require.True(t, user2.Filters.DisableFsChecks)
- }
- func TestUserPlaceholders(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- u := getTestUser()
- u.HomeDir = filepath.Join(os.TempDir(), "%username%_%password%")
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", u.Username)
- form.Set("home_dir", u.HomeDir)
- form.Set("password", u.Password)
- form.Set("status", strconv.Itoa(u.Status))
- form.Set("expiration_date", "")
- form.Set("permissions", "*")
- form.Set("public_keys[0][public_key]", testPubKey)
- form.Set("public_keys[1][public_key]", testPubKey1)
- form.Set("uid", "0")
- form.Set("gid", "0")
- form.Set("max_sessions", "0")
- form.Set("quota_size", "0")
- form.Set("quota_files", "0")
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("total_data_transfer", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ := http.NewRequest(http.MethodPost, webUserPath, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- user, _, err := httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, filepath.Join(os.TempDir(), fmt.Sprintf("%v_%v", defaultUsername, defaultPassword)), user.HomeDir)
- dbUser, err := dataprovider.UserExists(defaultUsername, "")
- assert.NoError(t, err)
- assert.True(t, dbUser.IsPasswordHashed())
- hashedPwd := dbUser.Password
- form.Set("password", redactedSecret)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, err = http.NewRequest(http.MethodPost, path.Join(webUserPath, defaultUsername), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- user, _, err = httpdtest.GetUserByUsername(defaultUsername, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, filepath.Join(os.TempDir(), defaultUsername+"_%password%"), user.HomeDir)
- // check that the password was unchanged
- dbUser, err = dataprovider.UserExists(defaultUsername, "")
- assert.NoError(t, err)
- assert.True(t, dbUser.IsPasswordHashed())
- assert.Equal(t, hashedPwd, dbUser.Password)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestFolderPlaceholders(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- folderName := "folderName"
- form := make(url.Values)
- form.Set("name", folderName)
- form.Set("mapped_path", filepath.Join(os.TempDir(), "%name%"))
- form.Set("description", "desc folder %name%")
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, err := http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- folderGet, _, err := httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, filepath.Join(os.TempDir(), folderName), folderGet.MappedPath)
- assert.Equal(t, fmt.Sprintf("desc folder %v", folderName), folderGet.Description)
- form.Set("mapped_path", filepath.Join(os.TempDir(), "%name%_%name%"))
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- folderGet, _, err = httpdtest.GetFolderByName(folderName, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, filepath.Join(os.TempDir(), fmt.Sprintf("%v_%v", folderName, folderName)), folderGet.MappedPath)
- assert.Equal(t, fmt.Sprintf("desc folder %v", folderName), folderGet.Description)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folderName}, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestFolderSaveFromTemplateMock(t *testing.T) {
- folder1 := "f1"
- folder2 := "f2"
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("name", "name")
- form.Set("mapped_path", filepath.Join(os.TempDir(), "%name%"))
- form.Set("description", "desc folder %name%")
- form.Set("template_folders[0][tpl_foldername]", folder1)
- form.Set("template_folders[1][tpl_foldername]", folder2)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, err := http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- _, _, err = httpdtest.GetFolderByName(folder1, http.StatusOK)
- assert.NoError(t, err)
- _, _, err = httpdtest.GetFolderByName(folder2, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folder1}, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveFolder(vfs.BaseVirtualFolder{Name: folder2}, http.StatusOK)
- assert.NoError(t, err)
- err = dataprovider.Close()
- assert.NoError(t, err)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, err = http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestFolderTemplateMock(t *testing.T) {
- folderName := "vfolder-template"
- mappedPath := filepath.Join(os.TempDir(), "%name%mapped%name%path")
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("name", folderName)
- form.Set("mapped_path", mappedPath)
- form.Set("description", "desc folder %name%")
- form.Set("template_folders[0][tpl_foldername]", "folder1")
- form.Set("template_folders[1][tpl_foldername]", "folder2")
- form.Set("template_folders[2][tpl_foldername]", "folder3")
- form.Set("template_folders[3][tpl_foldername]", "folder1 ")
- form.Add("template_folders[3][tpl_foldername]", " ")
- form.Set("form_action", "export_from_template")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ := http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateFolder+"?param=p%C3%AO%GG", &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidForm)
- folder1 := "folder1"
- folder2 := "folder2"
- folder3 := "folder3"
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var dump dataprovider.BackupData
- err = json.Unmarshal(rr.Body.Bytes(), &dump)
- require.NoError(t, err)
- require.Len(t, dump.Users, 0)
- require.Len(t, dump.Admins, 0)
- require.Len(t, dump.Folders, 3)
- for _, folder := range dump.Folders {
- switch folder.Name {
- case folder1:
- require.Equal(t, "desc folder folder1", folder.Description)
- require.True(t, strings.HasSuffix(folder.MappedPath, "folder1mappedfolder1path"))
- case folder2:
- require.Equal(t, "desc folder folder2", folder.Description)
- require.True(t, strings.HasSuffix(folder.MappedPath, "folder2mappedfolder2path"))
- default:
- require.Equal(t, "desc folder folder3", folder.Description)
- require.True(t, strings.HasSuffix(folder.MappedPath, "folder3mappedfolder3path"))
- }
- }
- form.Set("fs_provider", "1")
- form.Set("s3_bucket", "bucket")
- form.Set("s3_region", "us-east-1")
- form.Set("s3_access_key", "%name%")
- form.Set("s3_access_secret", "pwd%name%")
- form.Set("s3_key_prefix", "base/%name%")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("s3_upload_part_size", "5")
- form.Set("s3_upload_concurrency", "4")
- form.Set("s3_download_part_max_time", "0")
- form.Set("s3_upload_part_max_time", "0")
- form.Set("s3_download_part_size", "6")
- form.Set("s3_download_concurrency", "2")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- dump = dataprovider.BackupData{
- Version: dataprovider.DumpVersion,
- }
- err = json.Unmarshal(rr.Body.Bytes(), &dump)
- require.NoError(t, err)
- require.Len(t, dump.Users, 0)
- require.Len(t, dump.Admins, 0)
- require.Len(t, dump.Folders, 3)
- for _, folder := range dump.Folders {
- switch folder.Name {
- case folder1:
- require.Equal(t, folder1, folder.FsConfig.S3Config.AccessKey)
- err = folder.FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, fmt.Sprintf("pwd%s", folder1), folder.FsConfig.S3Config.AccessSecret.GetPayload())
- require.Equal(t, path.Join("base", folder1)+"/", folder.FsConfig.S3Config.KeyPrefix)
- case folder2:
- require.Equal(t, folder2, folder.FsConfig.S3Config.AccessKey)
- err = folder.FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, "pwd"+folder2, folder.FsConfig.S3Config.AccessSecret.GetPayload())
- require.Equal(t, "base/"+folder2+"/", folder.FsConfig.S3Config.KeyPrefix)
- default:
- require.Equal(t, folder3, folder.FsConfig.S3Config.AccessKey)
- err = folder.FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, "pwd"+folder3, folder.FsConfig.S3Config.AccessSecret.GetPayload())
- require.Equal(t, "base/"+folder3+"/", folder.FsConfig.S3Config.KeyPrefix)
- }
- }
- form.Set("template_folders[0][tpl_foldername]", " ")
- form.Set("template_folders[1][tpl_foldername]", "")
- form.Set("template_folders[2][tpl_foldername]", "")
- form.Set("template_folders[3][tpl_foldername]", " ")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFolderTemplate)
- form.Set("template_folders[0][tpl_foldername]", "name")
- form.Set("mapped_path", "relative-path")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, webTemplateFolder, &b)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidHomeDir)
- }
- func TestWebUserS3Mock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- lastPwdChange := user.LastPasswordChange
- assert.Greater(t, lastPwdChange, int64(0))
- user.FsConfig.Provider = sdk.S3FilesystemProvider
- user.FsConfig.S3Config.Bucket = "test"
- user.FsConfig.S3Config.Region = "eu-west-1"
- user.FsConfig.S3Config.AccessKey = "access-key"
- user.FsConfig.S3Config.AccessSecret = kms.NewPlainSecret("access-secret")
- user.FsConfig.S3Config.RoleARN = "arn:aws:iam::123456789012:user/Development/product_1234/*"
- user.FsConfig.S3Config.Endpoint = "http://127.0.0.1:9000/path?a=b"
- user.FsConfig.S3Config.StorageClass = "Standard"
- user.FsConfig.S3Config.KeyPrefix = "somedir/subdir/"
- user.FsConfig.S3Config.UploadPartSize = 5
- user.FsConfig.S3Config.UploadConcurrency = 4
- user.FsConfig.S3Config.DownloadPartMaxTime = 60
- user.FsConfig.S3Config.UploadPartMaxTime = 120
- user.FsConfig.S3Config.DownloadPartSize = 6
- user.FsConfig.S3Config.DownloadConcurrency = 3
- user.FsConfig.S3Config.ForcePathStyle = true
- user.FsConfig.S3Config.SkipTLSVerify = true
- user.FsConfig.S3Config.ACL = "public-read"
- user.Description = "s3 tèst user"
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("password", redactedSecret)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "1")
- form.Set("s3_bucket", user.FsConfig.S3Config.Bucket)
- form.Set("s3_region", user.FsConfig.S3Config.Region)
- form.Set("s3_access_key", user.FsConfig.S3Config.AccessKey)
- form.Set("s3_access_secret", user.FsConfig.S3Config.AccessSecret.GetPayload())
- form.Set("s3_role_arn", user.FsConfig.S3Config.RoleARN)
- form.Set("s3_storage_class", user.FsConfig.S3Config.StorageClass)
- form.Set("s3_acl", user.FsConfig.S3Config.ACL)
- form.Set("s3_endpoint", user.FsConfig.S3Config.Endpoint)
- form.Set("s3_key_prefix", user.FsConfig.S3Config.KeyPrefix)
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("directory_patterns[0][pattern_policy]", "0")
- form.Set("directory_patterns[1][pattern_path]", "/dir2")
- form.Set("directory_patterns[1][patterns]", "*.zip")
- form.Set("directory_patterns[1][pattern_type]", "denied")
- form.Set("directory_patterns[1][pattern_policy]", "1")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("ftp_security", "1")
- form.Set("s3_force_path_style", "checked")
- form.Set("s3_skip_tls_verify", "checked")
- form.Set("description", user.Description)
- form.Add("hooks", "pre_login_disabled")
- form.Add("allow_api_key_auth", "1")
- // test invalid s3_upload_part_size
- form.Set("s3_upload_part_size", "a")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid s3_upload_concurrency
- form.Set("s3_upload_part_size", strconv.FormatInt(user.FsConfig.S3Config.UploadPartSize, 10))
- form.Set("s3_upload_concurrency", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid s3_download_part_size
- form.Set("s3_upload_concurrency", strconv.Itoa(user.FsConfig.S3Config.UploadConcurrency))
- form.Set("s3_download_part_size", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid s3_download_concurrency
- form.Set("s3_download_part_size", strconv.FormatInt(user.FsConfig.S3Config.DownloadPartSize, 10))
- form.Set("s3_download_concurrency", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid s3_download_part_max_time
- form.Set("s3_download_concurrency", strconv.Itoa(user.FsConfig.S3Config.DownloadConcurrency))
- form.Set("s3_download_part_max_time", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid s3_upload_part_max_time
- form.Set("s3_download_part_max_time", strconv.Itoa(user.FsConfig.S3Config.DownloadPartMaxTime))
- form.Set("s3_upload_part_max_time", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now add the user
- form.Set("s3_upload_part_max_time", strconv.Itoa(user.FsConfig.S3Config.UploadPartMaxTime))
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, int64(1577836800000), updateUser.ExpirationDate)
- assert.Equal(t, updateUser.FsConfig.S3Config.Bucket, user.FsConfig.S3Config.Bucket)
- assert.Equal(t, updateUser.FsConfig.S3Config.Region, user.FsConfig.S3Config.Region)
- assert.Equal(t, updateUser.FsConfig.S3Config.AccessKey, user.FsConfig.S3Config.AccessKey)
- assert.Equal(t, updateUser.FsConfig.S3Config.RoleARN, user.FsConfig.S3Config.RoleARN)
- assert.Equal(t, updateUser.FsConfig.S3Config.StorageClass, user.FsConfig.S3Config.StorageClass)
- assert.Equal(t, updateUser.FsConfig.S3Config.ACL, user.FsConfig.S3Config.ACL)
- assert.Equal(t, updateUser.FsConfig.S3Config.Endpoint, user.FsConfig.S3Config.Endpoint)
- assert.Equal(t, updateUser.FsConfig.S3Config.KeyPrefix, user.FsConfig.S3Config.KeyPrefix)
- assert.Equal(t, updateUser.FsConfig.S3Config.UploadPartSize, user.FsConfig.S3Config.UploadPartSize)
- assert.Equal(t, updateUser.FsConfig.S3Config.UploadConcurrency, user.FsConfig.S3Config.UploadConcurrency)
- assert.Equal(t, updateUser.FsConfig.S3Config.DownloadPartMaxTime, user.FsConfig.S3Config.DownloadPartMaxTime)
- assert.Equal(t, updateUser.FsConfig.S3Config.UploadPartMaxTime, user.FsConfig.S3Config.UploadPartMaxTime)
- assert.Equal(t, updateUser.FsConfig.S3Config.DownloadPartSize, user.FsConfig.S3Config.DownloadPartSize)
- assert.Equal(t, updateUser.FsConfig.S3Config.DownloadConcurrency, user.FsConfig.S3Config.DownloadConcurrency)
- assert.Equal(t, lastPwdChange, updateUser.LastPasswordChange)
- assert.True(t, updateUser.FsConfig.S3Config.ForcePathStyle)
- assert.True(t, updateUser.FsConfig.S3Config.SkipTLSVerify)
- if assert.Equal(t, 2, len(updateUser.Filters.FilePatterns)) {
- for _, filter := range updateUser.Filters.FilePatterns {
- switch filter.Path {
- case "/dir1":
- assert.Equal(t, sdk.DenyPolicyDefault, filter.DenyPolicy)
- case "/dir2":
- assert.Equal(t, sdk.DenyPolicyHide, filter.DenyPolicy)
- }
- }
- }
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Empty(t, updateUser.FsConfig.S3Config.AccessSecret.GetKey())
- assert.Empty(t, updateUser.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.Equal(t, user.Description, updateUser.Description)
- assert.True(t, updateUser.Filters.Hooks.PreLoginDisabled)
- assert.False(t, updateUser.Filters.Hooks.ExternalAuthDisabled)
- assert.False(t, updateUser.Filters.Hooks.CheckPasswordDisabled)
- assert.False(t, updateUser.Filters.DisableFsChecks)
- assert.True(t, updateUser.Filters.AllowAPIKeyAuth)
- assert.Equal(t, 1, updateUser.Filters.FTPSecurity)
- // now check that a redacted password is not saved
- form.Set("s3_access_secret", redactedSecret)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var lastUpdatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &lastUpdatedUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.S3Config.AccessSecret.GetStatus())
- assert.Equal(t, updateUser.FsConfig.S3Config.AccessSecret.GetPayload(), lastUpdatedUser.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.S3Config.AccessSecret.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.S3Config.AccessSecret.GetAdditionalData())
- assert.Equal(t, lastPwdChange, lastUpdatedUser.LastPasswordChange)
- // now clear credentials
- form.Set("s3_access_key", "")
- form.Set("s3_access_secret", "")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var userGet dataprovider.User
- err = render.DecodeJSON(rr.Body, &userGet)
- assert.NoError(t, err)
- assert.Nil(t, userGet.FsConfig.S3Config.AccessSecret)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebUserGCSMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, err := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- credentialsFilePath := filepath.Join(os.TempDir(), "gcs.json")
- err = createTestFile(credentialsFilePath, 0)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.GCSFilesystemProvider
- user.FsConfig.GCSConfig.Bucket = "test"
- user.FsConfig.GCSConfig.KeyPrefix = "somedir/subdir/"
- user.FsConfig.GCSConfig.StorageClass = "standard"
- user.FsConfig.GCSConfig.ACL = "publicReadWrite"
- user.FsConfig.GCSConfig.UploadPartSize = 16
- user.FsConfig.GCSConfig.UploadPartMaxTime = 32
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("password", redactedSecret)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "2")
- form.Set("gcs_bucket", user.FsConfig.GCSConfig.Bucket)
- form.Set("gcs_storage_class", user.FsConfig.GCSConfig.StorageClass)
- form.Set("gcs_acl", user.FsConfig.GCSConfig.ACL)
- form.Set("gcs_key_prefix", user.FsConfig.GCSConfig.KeyPrefix)
- form.Set("gcs_upload_part_size", strconv.FormatInt(user.FsConfig.GCSConfig.UploadPartSize, 10))
- form.Set("gcs_upload_part_max_time", strconv.FormatInt(int64(user.FsConfig.GCSConfig.UploadPartMaxTime), 10))
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("ftp_security", "1")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- b, contentType, _ = getMultipartFormData(form, "gcs_credential_file", credentialsFilePath)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = createTestFile(credentialsFilePath, 4096)
- assert.NoError(t, err)
- b, contentType, _ = getMultipartFormData(form, "gcs_credential_file", credentialsFilePath)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, int64(1577836800000), updateUser.ExpirationDate)
- assert.Equal(t, user.FsConfig.Provider, updateUser.FsConfig.Provider)
- assert.Equal(t, user.FsConfig.GCSConfig.Bucket, updateUser.FsConfig.GCSConfig.Bucket)
- assert.Equal(t, user.FsConfig.GCSConfig.StorageClass, updateUser.FsConfig.GCSConfig.StorageClass)
- assert.Equal(t, user.FsConfig.GCSConfig.ACL, updateUser.FsConfig.GCSConfig.ACL)
- assert.Equal(t, user.FsConfig.GCSConfig.KeyPrefix, updateUser.FsConfig.GCSConfig.KeyPrefix)
- assert.Equal(t, user.FsConfig.GCSConfig.UploadPartSize, updateUser.FsConfig.GCSConfig.UploadPartSize)
- assert.Equal(t, user.FsConfig.GCSConfig.UploadPartMaxTime, updateUser.FsConfig.GCSConfig.UploadPartMaxTime)
- if assert.Len(t, updateUser.Filters.FilePatterns, 1) {
- assert.Equal(t, "/dir1", updateUser.Filters.FilePatterns[0].Path)
- assert.Len(t, updateUser.Filters.FilePatterns[0].AllowedPatterns, 2)
- assert.Contains(t, updateUser.Filters.FilePatterns[0].AllowedPatterns, "*.png")
- assert.Contains(t, updateUser.Filters.FilePatterns[0].AllowedPatterns, "*.jpg")
- }
- assert.Equal(t, 1, updateUser.Filters.FTPSecurity)
- form.Set("gcs_auto_credentials", "on")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- updateUser = dataprovider.User{}
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, 1, updateUser.FsConfig.GCSConfig.AutomaticCredentials)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = os.Remove(credentialsFilePath)
- assert.NoError(t, err)
- }
- func TestWebUserHTTPFsMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, err := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.HTTPFilesystemProvider
- user.FsConfig.HTTPConfig = vfs.HTTPFsConfig{
- BaseHTTPFsConfig: sdk.BaseHTTPFsConfig{
- Endpoint: "https://127.0.0.1:9999/api/v1",
- Username: defaultUsername,
- SkipTLSVerify: true,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- APIKey: kms.NewPlainSecret(defaultTokenAuthPass),
- }
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("password", redactedSecret)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "6")
- form.Set("http_endpoint", user.FsConfig.HTTPConfig.Endpoint)
- form.Set("http_username", user.FsConfig.HTTPConfig.Username)
- form.Set("http_password", user.FsConfig.HTTPConfig.Password.GetPayload())
- form.Set("http_api_key", user.FsConfig.HTTPConfig.APIKey.GetPayload())
- form.Set("http_skip_tls_verify", "checked")
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("directory_patterns[1][pattern_path]", "/dir2")
- form.Set("directory_patterns[1][patterns]", "*.zip")
- form.Set("directory_patterns[1][pattern_type]", "denied")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("http_equality_check_mode", "true")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the updated user
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, int64(1577836800000), updateUser.ExpirationDate)
- assert.Equal(t, 2, len(updateUser.Filters.FilePatterns))
- assert.Equal(t, user.FsConfig.HTTPConfig.Endpoint, updateUser.FsConfig.HTTPConfig.Endpoint)
- assert.Equal(t, user.FsConfig.HTTPConfig.Username, updateUser.FsConfig.HTTPConfig.Username)
- assert.Equal(t, user.FsConfig.HTTPConfig.SkipTLSVerify, updateUser.FsConfig.HTTPConfig.SkipTLSVerify)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, updateUser.FsConfig.HTTPConfig.Password.GetKey())
- assert.Empty(t, updateUser.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.HTTPConfig.APIKey.GetPayload())
- assert.Empty(t, updateUser.FsConfig.HTTPConfig.APIKey.GetKey())
- assert.Empty(t, updateUser.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- assert.Equal(t, 1, updateUser.FsConfig.HTTPConfig.EqualityCheckMode)
- // now check that a redacted password is not saved
- form.Set("http_equality_check_mode", "")
- form.Set("http_password", " "+redactedSecret+" ")
- form.Set("http_api_key", redactedSecret)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var lastUpdatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &lastUpdatedUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.HTTPConfig.Password.GetStatus())
- assert.Equal(t, updateUser.FsConfig.HTTPConfig.Password.GetPayload(), lastUpdatedUser.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.HTTPConfig.Password.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.Equal(t, updateUser.FsConfig.HTTPConfig.APIKey.GetPayload(), lastUpdatedUser.FsConfig.HTTPConfig.APIKey.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.HTTPConfig.APIKey.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- assert.Equal(t, 0, lastUpdatedUser.FsConfig.HTTPConfig.EqualityCheckMode)
- req, err = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebUserAzureBlobMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.AzureBlobFilesystemProvider
- user.FsConfig.AzBlobConfig.Container = "container"
- user.FsConfig.AzBlobConfig.AccountName = "aname"
- user.FsConfig.AzBlobConfig.AccountKey = kms.NewPlainSecret("access-skey")
- user.FsConfig.AzBlobConfig.Endpoint = "http://127.0.0.1:9000/path?b=c"
- user.FsConfig.AzBlobConfig.KeyPrefix = "somedir/subdir/"
- user.FsConfig.AzBlobConfig.UploadPartSize = 5
- user.FsConfig.AzBlobConfig.UploadConcurrency = 4
- user.FsConfig.AzBlobConfig.DownloadPartSize = 3
- user.FsConfig.AzBlobConfig.DownloadConcurrency = 6
- user.FsConfig.AzBlobConfig.UseEmulator = true
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("password", redactedSecret)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "3")
- form.Set("az_container", user.FsConfig.AzBlobConfig.Container)
- form.Set("az_account_name", user.FsConfig.AzBlobConfig.AccountName)
- form.Set("az_account_key", user.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- form.Set("az_endpoint", user.FsConfig.AzBlobConfig.Endpoint)
- form.Set("az_key_prefix", user.FsConfig.AzBlobConfig.KeyPrefix)
- form.Set("az_use_emulator", "checked")
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("directory_patterns[1][pattern_path]", "/dir2")
- form.Set("directory_patterns[1][patterns]", "*.zip")
- form.Set("directory_patterns[1][pattern_type]", "denied")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- // test invalid az_upload_part_size
- form.Set("az_upload_part_size", "a")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid az_upload_concurrency
- form.Set("az_upload_part_size", strconv.FormatInt(user.FsConfig.AzBlobConfig.UploadPartSize, 10))
- form.Set("az_upload_concurrency", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid az_download_part_size
- form.Set("az_upload_concurrency", strconv.Itoa(user.FsConfig.AzBlobConfig.UploadConcurrency))
- form.Set("az_download_part_size", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // test invalid az_download_concurrency
- form.Set("az_download_part_size", strconv.FormatInt(user.FsConfig.AzBlobConfig.DownloadPartSize, 10))
- form.Set("az_download_concurrency", "a")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now add the user
- form.Set("az_download_concurrency", strconv.Itoa(user.FsConfig.AzBlobConfig.DownloadConcurrency))
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, int64(1577836800000), updateUser.ExpirationDate)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.Container, user.FsConfig.AzBlobConfig.Container)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.AccountName, user.FsConfig.AzBlobConfig.AccountName)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.Endpoint, user.FsConfig.AzBlobConfig.Endpoint)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.KeyPrefix, user.FsConfig.AzBlobConfig.KeyPrefix)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.UploadPartSize, user.FsConfig.AzBlobConfig.UploadPartSize)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.UploadConcurrency, user.FsConfig.AzBlobConfig.UploadConcurrency)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.DownloadPartSize, user.FsConfig.AzBlobConfig.DownloadPartSize)
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.DownloadConcurrency, user.FsConfig.AzBlobConfig.DownloadConcurrency)
- assert.Equal(t, 2, len(updateUser.Filters.FilePatterns))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- assert.Empty(t, updateUser.FsConfig.AzBlobConfig.AccountKey.GetKey())
- assert.Empty(t, updateUser.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- // now check that a redacted password is not saved
- form.Set("az_account_key", redactedSecret+" ")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var lastUpdatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &lastUpdatedUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.AzBlobConfig.AccountKey.GetStatus())
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.AccountKey.GetPayload(), lastUpdatedUser.FsConfig.AzBlobConfig.AccountKey.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.AzBlobConfig.AccountKey.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.AzBlobConfig.AccountKey.GetAdditionalData())
- // test SAS url
- user.FsConfig.AzBlobConfig.SASURL = kms.NewPlainSecret("sasurl")
- form.Set("az_account_name", "")
- form.Set("az_account_key", "")
- form.Set("az_container", "")
- form.Set("az_sas_url", user.FsConfig.AzBlobConfig.SASURL.GetPayload())
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- updateUser = dataprovider.User{}
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.AzBlobConfig.SASURL.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.AzBlobConfig.SASURL.GetPayload())
- assert.Empty(t, updateUser.FsConfig.AzBlobConfig.SASURL.GetKey())
- assert.Empty(t, updateUser.FsConfig.AzBlobConfig.SASURL.GetAdditionalData())
- // now check that a redacted sas url is not saved
- form.Set("az_sas_url", redactedSecret)
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- lastUpdatedUser = dataprovider.User{}
- err = render.DecodeJSON(rr.Body, &lastUpdatedUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.AzBlobConfig.SASURL.GetStatus())
- assert.Equal(t, updateUser.FsConfig.AzBlobConfig.SASURL.GetPayload(), lastUpdatedUser.FsConfig.AzBlobConfig.SASURL.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.AzBlobConfig.SASURL.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.AzBlobConfig.SASURL.GetAdditionalData())
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebUserCryptMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.CryptedFilesystemProvider
- user.FsConfig.CryptConfig.Passphrase = kms.NewPlainSecret("crypted passphrase")
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("password", redactedSecret)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "4")
- form.Set("crypt_passphrase", "")
- form.Set("cryptfs_read_buffer_size", "1")
- form.Set("cryptfs_write_buffer_size", "2")
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("directory_patterns[1][pattern_path]", "/dir2")
- form.Set("directory_patterns[1][patterns]", "*.zip")
- form.Set("directory_patterns[1][pattern_type]", "denied")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- // passphrase cannot be empty
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("crypt_passphrase", user.FsConfig.CryptConfig.Passphrase.GetPayload())
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, int64(1577836800000), updateUser.ExpirationDate)
- assert.Equal(t, 2, len(updateUser.Filters.FilePatterns))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, updateUser.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Empty(t, updateUser.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- assert.Equal(t, 1, updateUser.FsConfig.CryptConfig.ReadBufferSize)
- assert.Equal(t, 2, updateUser.FsConfig.CryptConfig.WriteBufferSize)
- // now check that a redacted password is not saved
- form.Set("crypt_passphrase", redactedSecret+" ")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var lastUpdatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &lastUpdatedUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.CryptConfig.Passphrase.GetStatus())
- assert.Equal(t, updateUser.FsConfig.CryptConfig.Passphrase.GetPayload(), lastUpdatedUser.FsConfig.CryptConfig.Passphrase.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.CryptConfig.Passphrase.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.CryptConfig.Passphrase.GetAdditionalData())
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebUserSFTPFsMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- userAsJSON := getUserAsJSON(t, user)
- req, _ := http.NewRequest(http.MethodPost, userPath, bytes.NewBuffer(userAsJSON))
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- err = render.DecodeJSON(rr.Body, &user)
- assert.NoError(t, err)
- user.FsConfig.Provider = sdk.SFTPFilesystemProvider
- user.FsConfig.SFTPConfig.Endpoint = "127.0.0.1:22"
- user.FsConfig.SFTPConfig.Username = "sftpuser"
- user.FsConfig.SFTPConfig.Password = kms.NewPlainSecret("pwd")
- user.FsConfig.SFTPConfig.PrivateKey = kms.NewPlainSecret(testPrivateKeyPwd)
- user.FsConfig.SFTPConfig.KeyPassphrase = kms.NewPlainSecret(privateKeyPwd)
- user.FsConfig.SFTPConfig.Fingerprints = []string{sftpPkeyFingerprint}
- user.FsConfig.SFTPConfig.Prefix = "/home/sftpuser"
- user.FsConfig.SFTPConfig.DisableCouncurrentReads = true
- user.FsConfig.SFTPConfig.BufferSize = 5
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("password", redactedSecret)
- form.Set("home_dir", user.HomeDir)
- form.Set("uid", "0")
- form.Set("gid", strconv.FormatInt(int64(user.GID), 10))
- form.Set("max_sessions", strconv.FormatInt(int64(user.MaxSessions), 10))
- form.Set("quota_size", strconv.FormatInt(user.QuotaSize, 10))
- form.Set("quota_files", strconv.FormatInt(int64(user.QuotaFiles), 10))
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("permissions", "*")
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("expiration_date", "2020-01-01 00:00:00")
- form.Set("allowed_ip", "")
- form.Set("denied_ip", "")
- form.Set("fs_provider", "5")
- form.Set("crypt_passphrase", "")
- form.Set("directory_patterns[0][pattern_path]", "/dir1")
- form.Set("directory_patterns[0][patterns]", "*.jpg,*.png")
- form.Set("directory_patterns[0][pattern_type]", "allowed")
- form.Set("directory_patterns[1][pattern_path]", "/dir2")
- form.Set("directory_patterns[1][patterns]", "*.zip")
- form.Set("directory_patterns[1][pattern_type]", "denied")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- // empty sftpconfig
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("sftp_endpoint", user.FsConfig.SFTPConfig.Endpoint)
- form.Set("sftp_username", user.FsConfig.SFTPConfig.Username)
- form.Set("sftp_password", user.FsConfig.SFTPConfig.Password.GetPayload())
- form.Set("sftp_private_key", user.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- form.Set("sftp_key_passphrase", user.FsConfig.SFTPConfig.KeyPassphrase.GetPayload())
- form.Set("sftp_fingerprints", user.FsConfig.SFTPConfig.Fingerprints[0])
- form.Set("sftp_prefix", user.FsConfig.SFTPConfig.Prefix)
- form.Set("sftp_disable_concurrent_reads", "true")
- form.Set("sftp_equality_check_mode", "true")
- form.Set("sftp_buffer_size", strconv.FormatInt(user.FsConfig.SFTPConfig.BufferSize, 10))
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var updateUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &updateUser)
- assert.NoError(t, err)
- assert.Equal(t, int64(1577836800000), updateUser.ExpirationDate)
- assert.Equal(t, 2, len(updateUser.Filters.FilePatterns))
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.SFTPConfig.Password.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.SFTPConfig.Password.GetPayload())
- assert.Empty(t, updateUser.FsConfig.SFTPConfig.Password.GetKey())
- assert.Empty(t, updateUser.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- assert.Empty(t, updateUser.FsConfig.SFTPConfig.PrivateKey.GetKey())
- assert.Empty(t, updateUser.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateUser.FsConfig.SFTPConfig.KeyPassphrase.GetStatus())
- assert.NotEmpty(t, updateUser.FsConfig.SFTPConfig.KeyPassphrase.GetPayload())
- assert.Empty(t, updateUser.FsConfig.SFTPConfig.KeyPassphrase.GetKey())
- assert.Empty(t, updateUser.FsConfig.SFTPConfig.KeyPassphrase.GetAdditionalData())
- assert.Equal(t, updateUser.FsConfig.SFTPConfig.Prefix, user.FsConfig.SFTPConfig.Prefix)
- assert.Equal(t, updateUser.FsConfig.SFTPConfig.Username, user.FsConfig.SFTPConfig.Username)
- assert.Equal(t, updateUser.FsConfig.SFTPConfig.Endpoint, user.FsConfig.SFTPConfig.Endpoint)
- assert.True(t, updateUser.FsConfig.SFTPConfig.DisableCouncurrentReads)
- assert.Len(t, updateUser.FsConfig.SFTPConfig.Fingerprints, 1)
- assert.Equal(t, user.FsConfig.SFTPConfig.BufferSize, updateUser.FsConfig.SFTPConfig.BufferSize)
- assert.Contains(t, updateUser.FsConfig.SFTPConfig.Fingerprints, sftpPkeyFingerprint)
- assert.Equal(t, 1, updateUser.FsConfig.SFTPConfig.EqualityCheckMode)
- // now check that a redacted credentials are not saved
- form.Set("sftp_password", redactedSecret+" ")
- form.Set("sftp_private_key", redactedSecret)
- form.Set("sftp_key_passphrase", redactedSecret)
- form.Set("sftp_equality_check_mode", "")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var lastUpdatedUser dataprovider.User
- err = render.DecodeJSON(rr.Body, &lastUpdatedUser)
- assert.NoError(t, err)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.SFTPConfig.Password.GetStatus())
- assert.Equal(t, updateUser.FsConfig.SFTPConfig.Password.GetPayload(), lastUpdatedUser.FsConfig.SFTPConfig.Password.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.SFTPConfig.Password.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.SFTPConfig.Password.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.SFTPConfig.PrivateKey.GetStatus())
- assert.Equal(t, updateUser.FsConfig.SFTPConfig.PrivateKey.GetPayload(), lastUpdatedUser.FsConfig.SFTPConfig.PrivateKey.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.SFTPConfig.PrivateKey.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.SFTPConfig.PrivateKey.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, lastUpdatedUser.FsConfig.SFTPConfig.KeyPassphrase.GetStatus())
- assert.Equal(t, updateUser.FsConfig.SFTPConfig.KeyPassphrase.GetPayload(), lastUpdatedUser.FsConfig.SFTPConfig.KeyPassphrase.GetPayload())
- assert.Empty(t, lastUpdatedUser.FsConfig.SFTPConfig.KeyPassphrase.GetKey())
- assert.Empty(t, lastUpdatedUser.FsConfig.SFTPConfig.KeyPassphrase.GetAdditionalData())
- assert.Equal(t, 0, lastUpdatedUser.FsConfig.SFTPConfig.EqualityCheckMode)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(userPath, user.Username), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebUserRole(t *testing.T) {
- role, resp, err := httpdtest.AddRole(getTestRole(), http.StatusCreated)
- assert.NoError(t, err, string(resp))
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Role = role.Name
- a.Permissions = []string{dataprovider.PermAdminAddUsers, dataprovider.PermAdminChangeUsers,
- dataprovider.PermAdminDeleteUsers, dataprovider.PermAdminViewUsers}
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- webToken, err := getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- user := getTestUser()
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- form.Set("home_dir", user.HomeDir)
- form.Set("password", user.Password)
- form.Set("status", strconv.Itoa(user.Status))
- form.Set("permissions", "*")
- form.Set("external_auth_cache_time", "0")
- form.Set("uid", "0")
- form.Set("gid", "0")
- form.Set("max_sessions", "0")
- form.Set("quota_size", "0")
- form.Set("quota_files", "0")
- form.Set("upload_bandwidth", strconv.FormatInt(user.UploadBandwidth, 10))
- form.Set("download_bandwidth", strconv.FormatInt(user.DownloadBandwidth, 10))
- form.Set("upload_data_transfer", strconv.FormatInt(user.UploadDataTransfer, 10))
- form.Set("download_data_transfer", strconv.FormatInt(user.DownloadDataTransfer, 10))
- form.Set("total_data_transfer", strconv.FormatInt(user.TotalDataTransfer, 10))
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "10")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- b, contentType, _ := getMultipartFormData(form, "", "")
- req, err := http.NewRequest(http.MethodPost, webUserPath, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user.Role)
- form.Set("role", "")
- b, contentType, _ = getMultipartFormData(form, "", "")
- req, _ = http.NewRequest(http.MethodPost, path.Join(webUserPath, user.Username), &b)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, role.Name, user.Role)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebEventAction(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- action := dataprovider.BaseEventAction{
- ID: 81,
- Name: "web_action_http",
- Description: "http web action",
- Type: dataprovider.ActionTypeHTTP,
- Options: dataprovider.BaseEventActionOptions{
- HTTPConfig: dataprovider.EventActionHTTPConfig{
- Endpoint: "https://localhost:4567/action",
- Username: defaultUsername,
- Headers: []dataprovider.KeyValue{
- {
- Key: "Content-Type",
- Value: "application/json",
- },
- },
- Password: kms.NewPlainSecret(defaultPassword),
- Timeout: 10,
- SkipTLSVerify: true,
- Method: http.MethodPost,
- QueryParameters: []dataprovider.KeyValue{
- {
- Key: "param1",
- Value: "value1",
- },
- },
- Body: `{"event":"{{Event}}","name":"{{Name}}"}`,
- },
- },
- }
- form := make(url.Values)
- form.Set("name", action.Name)
- form.Set("description", action.Description)
- form.Set("fs_action_type", "0")
- form.Set("type", "a")
- req, err := http.NewRequest(http.MethodPost, webAdminEventActionPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("http_timeout", "b")
- req, err = http.NewRequest(http.MethodPost, webAdminEventActionPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("cmd_timeout", "20")
- form.Set("pwd_expiration_threshold", "10")
- form.Set("http_timeout", fmt.Sprintf("%d", action.Options.HTTPConfig.Timeout))
- form.Set("http_headers[0][http_header_key]", action.Options.HTTPConfig.Headers[0].Key)
- form.Set("http_headers[0][http_header_value]", action.Options.HTTPConfig.Headers[0].Value)
- form.Set("http_headers[1][http_header_key]", action.Options.HTTPConfig.Headers[0].Key) // ignored
- form.Set("query_parameters[0][http_query_key]", action.Options.HTTPConfig.QueryParameters[0].Key)
- form.Set("query_parameters[0][http_query_value]", action.Options.HTTPConfig.QueryParameters[0].Value)
- form.Set("http_body", action.Options.HTTPConfig.Body)
- form.Set("http_skip_tls_verify", "1")
- form.Set("http_username", action.Options.HTTPConfig.Username)
- form.Set("http_password", action.Options.HTTPConfig.Password.GetPayload())
- form.Set("http_method", action.Options.HTTPConfig.Method)
- req, err = http.NewRequest(http.MethodPost, webAdminEventActionPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminEventActionPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorURLRequired)
- form.Set("http_endpoint", action.Options.HTTPConfig.Endpoint)
- req, err = http.NewRequest(http.MethodPost, webAdminEventActionPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // a new add will fail
- req, err = http.NewRequest(http.MethodPost, webAdminEventActionPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // list actions
- req, err = http.NewRequest(http.MethodGet, webAdminEventActionsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminEventActionsPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render add page
- req, err = http.NewRequest(http.MethodGet, webAdminEventActionPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render action page
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminEventActionPath, action.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // missing action
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminEventActionPath, action.Name+"1"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // check the action
- actionGet, _, err := httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Equal(t, action.Description, actionGet.Description)
- assert.Equal(t, action.Options.HTTPConfig.Body, actionGet.Options.HTTPConfig.Body)
- assert.Equal(t, action.Options.HTTPConfig.Endpoint, actionGet.Options.HTTPConfig.Endpoint)
- assert.Equal(t, action.Options.HTTPConfig.Headers, actionGet.Options.HTTPConfig.Headers)
- assert.Equal(t, action.Options.HTTPConfig.Method, actionGet.Options.HTTPConfig.Method)
- assert.Equal(t, action.Options.HTTPConfig.SkipTLSVerify, actionGet.Options.HTTPConfig.SkipTLSVerify)
- assert.Equal(t, action.Options.HTTPConfig.Timeout, actionGet.Options.HTTPConfig.Timeout)
- assert.Equal(t, action.Options.HTTPConfig.Username, actionGet.Options.HTTPConfig.Username)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, actionGet.Options.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, actionGet.Options.HTTPConfig.Password.GetPayload())
- assert.Empty(t, actionGet.Options.HTTPConfig.Password.GetKey())
- assert.Empty(t, actionGet.Options.HTTPConfig.Password.GetAdditionalData())
- // update and check that the password is preserved and the multipart fields
- form.Set("http_password", redactedSecret)
- form.Set("http_body", "")
- form.Set("http_timeout", "0")
- form.Del("http_headers[0][http_header_key]")
- form.Del("http_headers[0][http_header_val]")
- form.Set("multipart_body[0][http_part_name]", "part1")
- form.Set("multipart_body[0][http_part_file]", "{{VirtualPath}}")
- form.Set("multipart_body[0][http_part_body]", "")
- form.Set("multipart_body[0][http_part_headers]", "X-MyHeader: a:b,c")
- form.Set("multipart_body[12][http_part_name]", "part2")
- form.Set("multipart_body[12][http_part_headers]", "Content-Type:application/json \r\n")
- form.Set("multipart_body[12][http_part_body]", "{{ObjectData}}")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- dbAction, err := dataprovider.EventActionExists(action.Name)
- assert.NoError(t, err)
- err = dbAction.Options.HTTPConfig.Password.Decrypt()
- assert.NoError(t, err)
- assert.Equal(t, defaultPassword, dbAction.Options.HTTPConfig.Password.GetPayload())
- assert.Empty(t, dbAction.Options.HTTPConfig.Body)
- assert.Equal(t, 0, dbAction.Options.HTTPConfig.Timeout)
- if assert.Len(t, dbAction.Options.HTTPConfig.Parts, 2) {
- assert.Equal(t, "part1", dbAction.Options.HTTPConfig.Parts[0].Name)
- assert.Equal(t, "/{{VirtualPath}}", dbAction.Options.HTTPConfig.Parts[0].Filepath)
- assert.Empty(t, dbAction.Options.HTTPConfig.Parts[0].Body)
- assert.Equal(t, "X-MyHeader", dbAction.Options.HTTPConfig.Parts[0].Headers[0].Key)
- assert.Equal(t, "a:b,c", dbAction.Options.HTTPConfig.Parts[0].Headers[0].Value)
- assert.Equal(t, "part2", dbAction.Options.HTTPConfig.Parts[1].Name)
- assert.Equal(t, "{{ObjectData}}", dbAction.Options.HTTPConfig.Parts[1].Body)
- assert.Empty(t, dbAction.Options.HTTPConfig.Parts[1].Filepath)
- assert.Equal(t, "Content-Type", dbAction.Options.HTTPConfig.Parts[1].Headers[0].Key)
- assert.Equal(t, "application/json", dbAction.Options.HTTPConfig.Parts[1].Headers[0].Value)
- }
- // change action type
- action.Type = dataprovider.ActionTypeCommand
- action.Options.CmdConfig = dataprovider.EventActionCommandConfig{
- Cmd: filepath.Join(os.TempDir(), "cmd"),
- Args: []string{"arg1", "arg2"},
- Timeout: 20,
- EnvVars: []dataprovider.KeyValue{
- {
- Key: "key",
- Value: "val",
- },
- },
- }
- form.Set("type", fmt.Sprintf("%d", action.Type))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorCommandRequired)
- form.Set("cmd_path", action.Options.CmdConfig.Cmd)
- form.Set("cmd_timeout", "a")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("cmd_timeout", fmt.Sprintf("%d", action.Options.CmdConfig.Timeout))
- form.Set("env_vars[0][cmd_env_key]", action.Options.CmdConfig.EnvVars[0].Key)
- form.Set("env_vars[0][cmd_env_value]", action.Options.CmdConfig.EnvVars[0].Value)
- form.Set("cmd_arguments", "arg1 ,arg2 ")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // update a missing action
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name+"1"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // update with no csrf token
- form.Del(csrfFormToken)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Equal(t, action.Options.CmdConfig.Cmd, actionGet.Options.CmdConfig.Cmd)
- assert.Equal(t, action.Options.CmdConfig.Args, actionGet.Options.CmdConfig.Args)
- assert.Equal(t, action.Options.CmdConfig.Timeout, actionGet.Options.CmdConfig.Timeout)
- assert.Equal(t, action.Options.CmdConfig.EnvVars, actionGet.Options.CmdConfig.EnvVars)
- assert.Equal(t, dataprovider.EventActionHTTPConfig{}, actionGet.Options.HTTPConfig)
- assert.Equal(t, dataprovider.EventActionPasswordExpiration{}, actionGet.Options.PwdExpirationConfig)
- // change action type again
- action.Type = dataprovider.ActionTypeEmail
- action.Options.EmailConfig = dataprovider.EventActionEmailConfig{
- Recipients: []string{"address1@example.com", "address2@example.com"},
- Bcc: []string{"address3@example.com"},
- Subject: "subject",
- ContentType: 1,
- Body: "body",
- Attachments: []string{"/file1.txt", "/file2.txt"},
- }
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("email_recipients", "address1@example.com, address2@example.com")
- form.Set("email_bcc", "address3@example.com")
- form.Set("email_subject", action.Options.EmailConfig.Subject)
- form.Set("email_content_type", fmt.Sprintf("%d", action.Options.EmailConfig.ContentType))
- form.Set("email_body", action.Options.EmailConfig.Body)
- form.Set("email_attachments", "file1.txt, file2.txt")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Equal(t, action.Options.EmailConfig.Recipients, actionGet.Options.EmailConfig.Recipients)
- assert.Equal(t, action.Options.EmailConfig.Bcc, actionGet.Options.EmailConfig.Bcc)
- assert.Equal(t, action.Options.EmailConfig.Subject, actionGet.Options.EmailConfig.Subject)
- assert.Equal(t, action.Options.EmailConfig.ContentType, actionGet.Options.EmailConfig.ContentType)
- assert.Equal(t, action.Options.EmailConfig.Body, actionGet.Options.EmailConfig.Body)
- assert.Equal(t, action.Options.EmailConfig.Attachments, actionGet.Options.EmailConfig.Attachments)
- assert.Equal(t, dataprovider.EventActionHTTPConfig{}, actionGet.Options.HTTPConfig)
- assert.Empty(t, actionGet.Options.CmdConfig.Cmd)
- assert.Equal(t, 0, actionGet.Options.CmdConfig.Timeout)
- assert.Len(t, actionGet.Options.CmdConfig.EnvVars, 0)
- // change action type to data retention check
- action.Type = dataprovider.ActionTypeDataRetentionCheck
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("data_retention[10][folder_retention_path]", "p1")
- form.Set("data_retention[10][folder_retention_val]", "a")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("data_retention[10][folder_retention_val]", "24")
- form.Set("data_retention[10][folder_retention_options][]", "1")
- form.Set("data_retention[11][folder_retention_path]", "../p2")
- form.Set("data_retention[11][folder_retention_val]", "48")
- form.Set("data_retention[11][folder_retention_options][]", "1")
- form.Set("data_retention[13][folder_retention_options][]", "1") // ignored
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- if assert.Len(t, actionGet.Options.RetentionConfig.Folders, 2) {
- for _, folder := range actionGet.Options.RetentionConfig.Folders {
- switch folder.Path {
- case "/p1":
- assert.Equal(t, 24, folder.Retention)
- assert.True(t, folder.DeleteEmptyDirs)
- case "/p2":
- assert.Equal(t, 48, folder.Retention)
- assert.True(t, folder.DeleteEmptyDirs)
- default:
- t.Errorf("unexpected folder path %v", folder.Path)
- }
- }
- }
- action.Type = dataprovider.ActionTypeFilesystem
- action.Options.FsConfig = dataprovider.EventActionFilesystemConfig{
- Type: dataprovider.FilesystemActionMkdirs,
- MkDirs: []string{"a ", " a/b"},
- }
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("fs_mkdir_paths", strings.Join(action.Options.FsConfig.MkDirs, ","))
- form.Set("fs_action_type", "invalid")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("fs_action_type", fmt.Sprintf("%d", action.Options.FsConfig.Type))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- if assert.Len(t, actionGet.Options.FsConfig.MkDirs, 2) {
- for _, dir := range actionGet.Options.FsConfig.MkDirs {
- switch dir {
- case "/a":
- case "/a/b":
- default:
- t.Errorf("unexpected dir path %v", dir)
- }
- }
- }
- action.Options.FsConfig = dataprovider.EventActionFilesystemConfig{
- Type: dataprovider.FilesystemActionExist,
- Exist: []string{"b ", " c/d"},
- }
- form.Set("fs_action_type", fmt.Sprintf("%d", action.Options.FsConfig.Type))
- form.Set("fs_exist_paths", strings.Join(action.Options.FsConfig.Exist, ","))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- if assert.Len(t, actionGet.Options.FsConfig.Exist, 2) {
- for _, p := range actionGet.Options.FsConfig.Exist {
- switch p {
- case "/b":
- case "/c/d":
- default:
- t.Errorf("unexpected path %v", p)
- }
- }
- }
- action.Options.FsConfig = dataprovider.EventActionFilesystemConfig{
- Type: dataprovider.FilesystemActionRename,
- Renames: []dataprovider.KeyValue{
- {
- Key: "/src",
- Value: "/target",
- },
- },
- }
- form.Set("fs_action_type", fmt.Sprintf("%d", action.Options.FsConfig.Type))
- form.Set("fs_rename[0][fs_rename_source]", action.Options.FsConfig.Renames[0].Key)
- form.Set("fs_rename[0][fs_rename_target]", action.Options.FsConfig.Renames[0].Value)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Len(t, actionGet.Options.FsConfig.Renames, 1)
- action.Options.FsConfig = dataprovider.EventActionFilesystemConfig{
- Type: dataprovider.FilesystemActionCopy,
- Copy: []dataprovider.KeyValue{
- {
- Key: "/copy_src",
- Value: "/copy_target",
- },
- },
- }
- form.Set("fs_action_type", fmt.Sprintf("%d", action.Options.FsConfig.Type))
- form.Set("fs_copy[0][fs_copy_source]", action.Options.FsConfig.Copy[0].Key)
- form.Set("fs_copy[0][fs_copy_target]", action.Options.FsConfig.Copy[0].Value)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the update
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Len(t, actionGet.Options.FsConfig.Copy, 1)
- action.Type = dataprovider.ActionTypePasswordExpirationCheck
- action.Options.PwdExpirationConfig.Threshold = 15
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("pwd_expiration_threshold", "a")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("pwd_expiration_threshold", strconv.Itoa(action.Options.PwdExpirationConfig.Threshold))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Equal(t, action.Options.PwdExpirationConfig.Threshold, actionGet.Options.PwdExpirationConfig.Threshold)
- assert.Equal(t, 0, actionGet.Options.CmdConfig.Timeout)
- assert.Len(t, actionGet.Options.CmdConfig.EnvVars, 0)
- action.Type = dataprovider.ActionTypeUserInactivityCheck
- action.Options.UserInactivityConfig = dataprovider.EventActionUserInactivity{
- DisableThreshold: 10,
- DeleteThreshold: 15,
- }
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("inactivity_disable_threshold", strconv.Itoa(action.Options.UserInactivityConfig.DisableThreshold))
- form.Set("inactivity_delete_threshold", strconv.Itoa(action.Options.UserInactivityConfig.DeleteThreshold))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Equal(t, 0, actionGet.Options.PwdExpirationConfig.Threshold)
- assert.Equal(t, action.Options.UserInactivityConfig.DisableThreshold, actionGet.Options.UserInactivityConfig.DisableThreshold)
- assert.Equal(t, action.Options.UserInactivityConfig.DeleteThreshold, actionGet.Options.UserInactivityConfig.DeleteThreshold)
- action.Type = dataprovider.ActionTypeIDPAccountCheck
- form.Set("type", fmt.Sprintf("%d", action.Type))
- form.Set("idp_mode", "1")
- form.Set("idp_user", `{"username":"user"}`)
- form.Set("idp_admin", `{"username":"admin"}`)
- form.Set("pwd_expiration_threshold", strconv.Itoa(action.Options.PwdExpirationConfig.Threshold))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventActionPath, action.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- actionGet, _, err = httpdtest.GetEventActionByName(action.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, action.Type, actionGet.Type)
- assert.Equal(t, 1, actionGet.Options.IDPConfig.Mode)
- assert.Contains(t, actionGet.Options.IDPConfig.TemplateUser, `"user"`)
- assert.Contains(t, actionGet.Options.IDPConfig.TemplateAdmin, `"admin"`)
- req, err = http.NewRequest(http.MethodDelete, path.Join(webAdminEventActionPath, action.Name), nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- req, err = http.NewRequest(http.MethodDelete, path.Join(webAdminEventActionPath, action.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminEventActionsPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, `[]`, rr.Body.String())
- }
- func TestWebEventRule(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- a := dataprovider.BaseEventAction{
- Name: "web_action",
- Type: dataprovider.ActionTypeFilesystem,
- Options: dataprovider.BaseEventActionOptions{
- FsConfig: dataprovider.EventActionFilesystemConfig{
- Type: dataprovider.FilesystemActionExist,
- Exist: []string{"/dir1"},
- },
- },
- }
- action, _, err := httpdtest.AddEventAction(a, http.StatusCreated)
- assert.NoError(t, err)
- rule := dataprovider.EventRule{
- Name: "test_web_rule",
- Status: 1,
- Description: "rule added using web API",
- Trigger: dataprovider.EventTriggerSchedule,
- Conditions: dataprovider.EventConditions{
- Schedules: []dataprovider.Schedule{
- {
- Hours: "0",
- DayOfWeek: "*",
- DayOfMonth: "*",
- Month: "*",
- },
- },
- Options: dataprovider.ConditionOptions{
- Names: []dataprovider.ConditionPattern{
- {
- Pattern: "u*",
- InverseMatch: true,
- },
- },
- GroupNames: []dataprovider.ConditionPattern{
- {
- Pattern: "g*",
- InverseMatch: true,
- },
- },
- RoleNames: []dataprovider.ConditionPattern{
- {
- Pattern: "r*",
- InverseMatch: true,
- },
- },
- },
- },
- Actions: []dataprovider.EventAction{
- {
- BaseEventAction: dataprovider.BaseEventAction{
- Name: action.Name,
- },
- Order: 1,
- },
- },
- }
- form := make(url.Values)
- form.Set("name", rule.Name)
- form.Set("description", rule.Description)
- form.Set("status", "a")
- req, err := http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("status", fmt.Sprintf("%d", rule.Status))
- form.Set("trigger", "a")
- req, err = http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("trigger", fmt.Sprintf("%d", rule.Trigger))
- form.Set("schedules[0][schedule_hour]", rule.Conditions.Schedules[0].Hours)
- form.Set("schedules[0][schedule_day_of_week]", rule.Conditions.Schedules[0].DayOfWeek)
- form.Set("schedules[0][schedule_day_of_month]", rule.Conditions.Schedules[0].DayOfMonth)
- form.Set("schedules[0][schedule_month]", rule.Conditions.Schedules[0].Month)
- form.Set("name_filters[0][name_pattern]", rule.Conditions.Options.Names[0].Pattern)
- form.Set("name_filters[0][type_name_pattern]", "inverse")
- form.Set("group_name_filters[0][group_name_pattern]", rule.Conditions.Options.GroupNames[0].Pattern)
- form.Set("group_name_filters[0][type_group_name_pattern]", "inverse")
- form.Set("role_name_filters[0][role_name_pattern]", rule.Conditions.Options.RoleNames[0].Pattern)
- form.Set("role_name_filters[0][type_role_name_pattern]", "inverse")
- req, err = http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidMinSize)
- form.Set("fs_min_size", "0")
- req, err = http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidMaxSize)
- form.Set("fs_max_size", "0")
- form.Set("actions[0][action_name]", action.Name)
- req, err = http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // a new add will fail
- req, err = http.NewRequest(http.MethodPost, webAdminEventRulePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // list rules
- req, err = http.NewRequest(http.MethodGet, webAdminEventRulesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminEventRulesPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render add page
- req, err = http.NewRequest(http.MethodGet, webAdminEventRulePath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render rule page
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminEventRulePath, rule.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // missing rule
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminEventRulePath, rule.Name+"1"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // check the rule
- ruleGet, _, err := httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, rule.Trigger, ruleGet.Trigger)
- assert.Equal(t, rule.Status, ruleGet.Status)
- assert.Equal(t, rule.Description, ruleGet.Description)
- assert.Equal(t, rule.Conditions, ruleGet.Conditions)
- if assert.Len(t, ruleGet.Actions, 1) {
- assert.Equal(t, rule.Actions[0].Name, ruleGet.Actions[0].Name)
- assert.Equal(t, rule.Actions[0].Order, ruleGet.Actions[0].Order)
- }
- // change rule trigger and status
- rule.Status = 0
- rule.Trigger = dataprovider.EventTriggerFsEvent
- rule.Conditions = dataprovider.EventConditions{
- FsEvents: []string{"upload", "download"},
- Options: dataprovider.ConditionOptions{
- Names: []dataprovider.ConditionPattern{
- {
- Pattern: "u*",
- InverseMatch: true,
- },
- },
- GroupNames: []dataprovider.ConditionPattern{
- {
- Pattern: "g*",
- InverseMatch: true,
- },
- },
- RoleNames: []dataprovider.ConditionPattern{
- {
- Pattern: "r*",
- InverseMatch: true,
- },
- },
- FsPaths: []dataprovider.ConditionPattern{
- {
- Pattern: "/subdir/*.txt",
- },
- },
- Protocols: []string{common.ProtocolSFTP, common.ProtocolHTTP},
- MinFileSize: 1024 * 1024,
- MaxFileSize: 5 * 1024 * 1024,
- },
- }
- form.Set("status", fmt.Sprintf("%d", rule.Status))
- form.Set("trigger", fmt.Sprintf("%d", rule.Trigger))
- for _, event := range rule.Conditions.FsEvents {
- form.Add("fs_events", event)
- }
- form.Set("path_filters[0][fs_path_pattern]", rule.Conditions.Options.FsPaths[0].Pattern)
- for _, protocol := range rule.Conditions.Options.Protocols {
- form.Add("fs_protocols", protocol)
- }
- form.Set("fs_min_size", fmt.Sprintf("%d", rule.Conditions.Options.MinFileSize))
- form.Set("fs_max_size", fmt.Sprintf("%d", rule.Conditions.Options.MaxFileSize))
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the rule
- ruleGet, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, rule.Status, ruleGet.Status)
- assert.Equal(t, rule.Trigger, ruleGet.Trigger)
- assert.Equal(t, rule.Description, ruleGet.Description)
- assert.Equal(t, rule.Conditions, ruleGet.Conditions)
- if assert.Len(t, ruleGet.Actions, 1) {
- assert.Equal(t, rule.Actions[0].Name, ruleGet.Actions[0].Name)
- assert.Equal(t, rule.Actions[0].Order, ruleGet.Actions[0].Order)
- }
- rule.Trigger = dataprovider.EventTriggerIDPLogin
- form.Set("trigger", fmt.Sprintf("%d", rule.Trigger))
- form.Set("idp_login_event", "1")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the rule
- ruleGet, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, rule.Trigger, ruleGet.Trigger)
- assert.Equal(t, 1, ruleGet.Conditions.IDPLoginEvent)
- form.Set("idp_login_event", "2")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the rule
- ruleGet, _, err = httpdtest.GetEventRuleByName(rule.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, rule.Trigger, ruleGet.Trigger)
- assert.Equal(t, 2, ruleGet.Conditions.IDPLoginEvent)
- // update a missing rule
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name+"1"),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // update with no csrf token
- form.Del(csrfFormToken)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- // update with no action defined
- form.Del("actions[0][action_name]")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorRuleActionRequired)
- // invalid trigger
- form.Set("trigger", "a")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminEventRulePath, rule.Name),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- req, err = http.NewRequest(http.MethodDelete, path.Join(webAdminEventRulePath, rule.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodDelete, path.Join(webAdminEventActionPath, action.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebIPListEntries(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webIPListPath+"/mode", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, webIPListPath+"/mode/a", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webIPListPath, "/1/a"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webIPListPath+"/1", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webIPListsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- entry := dataprovider.IPListEntry{
- IPOrNet: "12.34.56.78/20",
- Type: dataprovider.IPListTypeDefender,
- Mode: dataprovider.ListModeDeny,
- Description: "note",
- Protocols: 5,
- }
- form := make(url.Values)
- form.Set("ipornet", entry.IPOrNet)
- form.Set("description", entry.Description)
- form.Set("mode", "a")
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/mode", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError400Message)
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/1", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/2", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("mode", "2")
- form.Set("protocols", "a")
- form.Add("protocols", "1")
- form.Add("protocols", "4")
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/2", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- entry1, _, err := httpdtest.GetIPListEntry(entry.IPOrNet, dataprovider.IPListTypeDefender, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, entry.Description, entry1.Description)
- assert.Equal(t, entry.Mode, entry1.Mode)
- assert.Equal(t, entry.Protocols, entry1.Protocols)
- form.Set("ipornet", "1111.11.11.11")
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/1", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorIPInvalid)
- form.Set("ipornet", entry.IPOrNet)
- form.Set("mode", "invalid") // ignored for list type 1
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/1", bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- entry2, _, err := httpdtest.GetIPListEntry(entry.IPOrNet, dataprovider.IPListTypeAllowList, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, entry.Description, entry2.Description)
- assert.Equal(t, dataprovider.ListModeAllow, entry2.Mode)
- assert.Equal(t, entry.Protocols, entry2.Protocols)
- req, err = http.NewRequest(http.MethodGet, path.Join(webIPListPath, "1", url.PathEscape(entry2.IPOrNet)), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("protocols", "1")
- req, err = http.NewRequest(http.MethodPost, path.Join(webIPListPath, "1", url.PathEscape(entry.IPOrNet)),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- entry2, _, err = httpdtest.GetIPListEntry(entry.IPOrNet, dataprovider.IPListTypeAllowList, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, entry.Description, entry2.Description)
- assert.Equal(t, dataprovider.ListModeAllow, entry2.Mode)
- assert.Equal(t, 1, entry2.Protocols)
- form.Del(csrfFormToken)
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/1/"+url.PathEscape(entry.IPOrNet),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/a/"+url.PathEscape(entry.IPOrNet),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/1/"+url.PathEscape(entry.IPOrNet)+"a",
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- form.Set("mode", "a")
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/2/"+url.PathEscape(entry.IPOrNet),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("mode", "100")
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/2/"+url.PathEscape(entry.IPOrNet),
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- _, err = httpdtest.RemoveIPListEntry(entry1, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveIPListEntry(entry2, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestWebRole(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- role := getTestRole()
- form := make(url.Values)
- form.Set("name", "")
- form.Set("description", role.Description)
- req, err := http.NewRequest(http.MethodPost, webAdminRolePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminRolePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorNameRequired)
- form.Set("name", role.Name)
- req, err = http.NewRequest(http.MethodPost, webAdminRolePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // a new add will fail
- req, err = http.NewRequest(http.MethodPost, webAdminRolePath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // list roles
- req, err = http.NewRequest(http.MethodGet, webAdminRolesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminRolesPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render the new role page
- req, err = http.NewRequest(http.MethodGet, webAdminRolePath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminRolePath, role.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminRolePath, "missing_role"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- // parse form error
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminRolePath, role.Name)+"?param=p%C4%AO%GH",
- bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidForm)
- // update role
- form.Set("description", "new desc")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminRolePath, role.Name), bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check the changes
- role, _, err = httpdtest.GetRoleByName(role.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, "new desc", role.Description)
- // no CSRF token
- form.Set(csrfFormToken, "")
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminRolePath, role.Name), bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- // missing role
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminRolePath, "missing"), bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestNameParamSingleSlash(t *testing.T) {
- err := dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.NamingRules = 5
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- group := getTestGroup()
- group.Name = "/"
- form := make(url.Values)
- form.Set("name", group.Name)
- form.Set("description", group.Description)
- form.Set("max_sessions", "0")
- form.Set("quota_files", "0")
- form.Set("quota_size", "0")
- form.Set("upload_bandwidth", "0")
- form.Set("download_bandwidth", "0")
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("expires_in", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- groupGet, _, err := httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, "/", groupGet.Name)
- // check strip slash
- req, err = http.NewRequest(http.MethodGet, webGroupsPath+"/", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // cleanup
- req, err = http.NewRequest(http.MethodDelete, groupPath+"/"+url.PathEscape(group.Name), nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = dataprovider.Close()
- assert.NoError(t, err)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf = config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- }
- func TestAddWebGroup(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- group := getTestGroup()
- group.UserSettings = dataprovider.GroupUserSettings{
- BaseGroupUserSettings: sdk.BaseGroupUserSettings{
- HomeDir: filepath.Join(os.TempDir(), util.GenerateUniqueID()),
- Permissions: make(map[string][]string),
- MaxSessions: 2,
- QuotaSize: 123,
- QuotaFiles: 10,
- UploadBandwidth: 128,
- DownloadBandwidth: 256,
- ExpiresIn: 10,
- },
- }
- form := make(url.Values)
- form.Set("name", group.Name)
- form.Set("description", group.Description)
- form.Set("home_dir", group.UserSettings.HomeDir)
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("max_sessions", strconv.FormatInt(int64(group.UserSettings.MaxSessions), 10))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidQuotaSize)
- form.Set("quota_files", strconv.FormatInt(int64(group.UserSettings.QuotaFiles), 10))
- form.Set("quota_size", strconv.FormatInt(group.UserSettings.QuotaSize, 10))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("upload_bandwidth", strconv.FormatInt(group.UserSettings.UploadBandwidth, 10))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("download_bandwidth", strconv.FormatInt(group.UserSettings.DownloadBandwidth, 10))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("expires_in", strconv.Itoa(group.UserSettings.ExpiresIn))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidMaxFilesize)
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("external_auth_cache_time", "0")
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath+"?b=%2", &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr) // error parsing the multipart form
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // a new add will fail
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webGroupPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // list groups
- req, err = http.NewRequest(http.MethodGet, webGroupsPath, nil)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webGroupsPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render the new group page
- req, err = http.NewRequest(http.MethodGet, path.Join(webGroupPath, group.Name), nil)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // check the added group
- groupGet, _, err := httpdtest.GetGroupByName(group.Name, http.StatusOK)
- assert.NoError(t, err)
- assert.Equal(t, group.UserSettings, groupGet.UserSettings)
- assert.Equal(t, group.Name, groupGet.Name)
- assert.Equal(t, group.Description, groupGet.Description)
- // cleanup
- req, err = http.NewRequest(http.MethodDelete, path.Join(groupPath, group.Name), nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webGroupPath, group.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestAddWebFoldersMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- mappedPath := filepath.Clean(os.TempDir())
- folderName := filepath.Base(mappedPath)
- folderDesc := "a simple desc"
- form := make(url.Values)
- form.Set("mapped_path", mappedPath)
- form.Set("name", folderName)
- form.Set("description", folderDesc)
- form.Set("osfs_read_buffer_size", "3")
- form.Set("osfs_write_buffer_size", "4")
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // adding the same folder will fail since the name must be unique
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // invalid form
- req, err = http.NewRequest(http.MethodPost, webFolderPath, strings.NewReader(form.Encode()))
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", "text/plain; boundary=")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // now render the add folder page
- req, err = http.NewRequest(http.MethodGet, webFolderPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var folder vfs.BaseVirtualFolder
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath, folder.MappedPath)
- assert.Equal(t, folderName, folder.Name)
- assert.Equal(t, folderDesc, folder.Description)
- assert.Equal(t, 3, folder.FsConfig.OSConfig.ReadBufferSize)
- assert.Equal(t, 4, folder.FsConfig.OSConfig.WriteBufferSize)
- // cleanup
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestHTTPFsWebFolderMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- mappedPath := filepath.Clean(os.TempDir())
- folderName := filepath.Base(mappedPath)
- httpfsConfig := vfs.HTTPFsConfig{
- BaseHTTPFsConfig: sdk.BaseHTTPFsConfig{
- Endpoint: "https://127.0.0.1:9998/api/v1",
- Username: folderName,
- SkipTLSVerify: true,
- },
- Password: kms.NewPlainSecret(defaultPassword),
- APIKey: kms.NewPlainSecret(defaultTokenAuthPass),
- }
- form := make(url.Values)
- form.Set("mapped_path", mappedPath)
- form.Set("name", folderName)
- form.Set("fs_provider", "6")
- form.Set("http_endpoint", httpfsConfig.Endpoint)
- form.Set("http_username", "%name%")
- form.Set("http_password", httpfsConfig.Password.GetPayload())
- form.Set("http_api_key", httpfsConfig.APIKey.GetPayload())
- form.Set("http_skip_tls_verify", "checked")
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check
- var folder vfs.BaseVirtualFolder
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath, folder.MappedPath)
- assert.Equal(t, folderName, folder.Name)
- assert.Equal(t, sdk.HTTPFilesystemProvider, folder.FsConfig.Provider)
- assert.Equal(t, httpfsConfig.Endpoint, folder.FsConfig.HTTPConfig.Endpoint)
- assert.Equal(t, httpfsConfig.Username, folder.FsConfig.HTTPConfig.Username)
- assert.Equal(t, httpfsConfig.SkipTLSVerify, folder.FsConfig.HTTPConfig.SkipTLSVerify)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, folder.FsConfig.HTTPConfig.Password.GetStatus())
- assert.NotEmpty(t, folder.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, folder.FsConfig.HTTPConfig.Password.GetKey())
- assert.Empty(t, folder.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, folder.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.NotEmpty(t, folder.FsConfig.HTTPConfig.APIKey.GetPayload())
- assert.Empty(t, folder.FsConfig.HTTPConfig.APIKey.GetKey())
- assert.Empty(t, folder.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- // update
- form.Set("http_password", redactedSecret)
- form.Set("http_api_key", redactedSecret)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- // check
- var updateFolder vfs.BaseVirtualFolder
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &updateFolder)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath, updateFolder.MappedPath)
- assert.Equal(t, folderName, updateFolder.Name)
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateFolder.FsConfig.HTTPConfig.Password.GetStatus())
- assert.Equal(t, folder.FsConfig.HTTPConfig.Password.GetPayload(), updateFolder.FsConfig.HTTPConfig.Password.GetPayload())
- assert.Empty(t, updateFolder.FsConfig.HTTPConfig.Password.GetKey())
- assert.Empty(t, updateFolder.FsConfig.HTTPConfig.Password.GetAdditionalData())
- assert.Equal(t, sdkkms.SecretStatusSecretBox, updateFolder.FsConfig.HTTPConfig.APIKey.GetStatus())
- assert.Equal(t, folder.FsConfig.HTTPConfig.APIKey.GetPayload(), updateFolder.FsConfig.HTTPConfig.APIKey.GetPayload())
- assert.Empty(t, updateFolder.FsConfig.HTTPConfig.APIKey.GetKey())
- assert.Empty(t, updateFolder.FsConfig.HTTPConfig.APIKey.GetAdditionalData())
- // cleanup
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestS3WebFolderMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- mappedPath := filepath.Clean(os.TempDir())
- folderName := filepath.Base(mappedPath)
- folderDesc := "a simple desc"
- S3Bucket := "test"
- S3Region := "eu-west-1"
- S3AccessKey := "access-key"
- S3AccessSecret := kms.NewPlainSecret("folder-access-secret")
- S3SessionToken := "fake session token"
- S3RoleARN := "arn:aws:iam::123456789012:user/Development/product_1234/*"
- S3Endpoint := "http://127.0.0.1:9000/path?b=c"
- S3StorageClass := "Standard"
- S3ACL := "public-read-write"
- S3KeyPrefix := "somedir/subdir/"
- S3UploadPartSize := 5
- S3UploadConcurrency := 4
- S3MaxPartDownloadTime := 120
- S3MaxPartUploadTime := 60
- S3DownloadPartSize := 6
- S3DownloadConcurrency := 3
- form := make(url.Values)
- form.Set("mapped_path", mappedPath)
- form.Set("name", folderName)
- form.Set("description", folderDesc)
- form.Set("fs_provider", "1")
- form.Set("s3_bucket", S3Bucket)
- form.Set("s3_region", S3Region)
- form.Set("s3_access_key", S3AccessKey)
- form.Set("s3_access_secret", S3AccessSecret.GetPayload())
- form.Set("s3_session_token", S3SessionToken)
- form.Set("s3_role_arn", S3RoleARN)
- form.Set("s3_storage_class", S3StorageClass)
- form.Set("s3_acl", S3ACL)
- form.Set("s3_endpoint", S3Endpoint)
- form.Set("s3_key_prefix", S3KeyPrefix)
- form.Set("s3_upload_part_size", strconv.Itoa(S3UploadPartSize))
- form.Set("s3_download_part_max_time", strconv.Itoa(S3MaxPartDownloadTime))
- form.Set("s3_download_part_size", strconv.Itoa(S3DownloadPartSize))
- form.Set("s3_download_concurrency", strconv.Itoa(S3DownloadConcurrency))
- form.Set("s3_upload_part_max_time", strconv.Itoa(S3MaxPartUploadTime))
- form.Set("s3_upload_concurrency", "a")
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("s3_upload_concurrency", strconv.Itoa(S3UploadConcurrency))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webFolderPath, &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- var folder vfs.BaseVirtualFolder
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath, folder.MappedPath)
- assert.Equal(t, folderName, folder.Name)
- assert.Equal(t, folderDesc, folder.Description)
- assert.Equal(t, sdk.S3FilesystemProvider, folder.FsConfig.Provider)
- assert.Equal(t, S3Bucket, folder.FsConfig.S3Config.Bucket)
- assert.Equal(t, S3Region, folder.FsConfig.S3Config.Region)
- assert.Equal(t, S3AccessKey, folder.FsConfig.S3Config.AccessKey)
- assert.NotEmpty(t, folder.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Equal(t, S3Endpoint, folder.FsConfig.S3Config.Endpoint)
- assert.Equal(t, S3StorageClass, folder.FsConfig.S3Config.StorageClass)
- assert.Equal(t, S3ACL, folder.FsConfig.S3Config.ACL)
- assert.Equal(t, S3KeyPrefix, folder.FsConfig.S3Config.KeyPrefix)
- assert.Equal(t, S3UploadConcurrency, folder.FsConfig.S3Config.UploadConcurrency)
- assert.Equal(t, int64(S3UploadPartSize), folder.FsConfig.S3Config.UploadPartSize)
- assert.Equal(t, S3MaxPartDownloadTime, folder.FsConfig.S3Config.DownloadPartMaxTime)
- assert.Equal(t, S3MaxPartUploadTime, folder.FsConfig.S3Config.UploadPartMaxTime)
- assert.Equal(t, S3DownloadConcurrency, folder.FsConfig.S3Config.DownloadConcurrency)
- assert.Equal(t, int64(S3DownloadPartSize), folder.FsConfig.S3Config.DownloadPartSize)
- assert.False(t, folder.FsConfig.S3Config.ForcePathStyle)
- assert.False(t, folder.FsConfig.S3Config.SkipTLSVerify)
- // update
- S3UploadConcurrency = 10
- form.Set("s3_upload_concurrency", "b")
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form.Set("s3_upload_concurrency", strconv.Itoa(S3UploadConcurrency))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- folder = vfs.BaseVirtualFolder{}
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- assert.Equal(t, mappedPath, folder.MappedPath)
- assert.Equal(t, folderName, folder.Name)
- assert.Equal(t, folderDesc, folder.Description)
- assert.Equal(t, sdk.S3FilesystemProvider, folder.FsConfig.Provider)
- assert.Equal(t, S3Bucket, folder.FsConfig.S3Config.Bucket)
- assert.Equal(t, S3Region, folder.FsConfig.S3Config.Region)
- assert.Equal(t, S3AccessKey, folder.FsConfig.S3Config.AccessKey)
- assert.Equal(t, S3RoleARN, folder.FsConfig.S3Config.RoleARN)
- assert.NotEmpty(t, folder.FsConfig.S3Config.AccessSecret.GetPayload())
- assert.Equal(t, S3Endpoint, folder.FsConfig.S3Config.Endpoint)
- assert.Equal(t, S3StorageClass, folder.FsConfig.S3Config.StorageClass)
- assert.Equal(t, S3KeyPrefix, folder.FsConfig.S3Config.KeyPrefix)
- assert.Equal(t, S3UploadConcurrency, folder.FsConfig.S3Config.UploadConcurrency)
- assert.Equal(t, int64(S3UploadPartSize), folder.FsConfig.S3Config.UploadPartSize)
- // cleanup
- req, _ = http.NewRequest(http.MethodDelete, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestUpdateWebGroupMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- group, _, err := httpdtest.AddGroup(getTestGroup(), http.StatusCreated)
- assert.NoError(t, err)
- group.UserSettings = dataprovider.GroupUserSettings{
- BaseGroupUserSettings: sdk.BaseGroupUserSettings{
- HomeDir: filepath.Join(os.TempDir(), util.GenerateUniqueID()),
- Permissions: make(map[string][]string),
- },
- FsConfig: vfs.Filesystem{
- Provider: sdk.SFTPFilesystemProvider,
- SFTPConfig: vfs.SFTPFsConfig{
- BaseSFTPFsConfig: sdk.BaseSFTPFsConfig{
- Endpoint: sftpServerAddr,
- Username: defaultUsername,
- BufferSize: 1,
- },
- },
- },
- }
- form := make(url.Values)
- form.Set("name", group.Name)
- form.Set("description", group.Description)
- form.Set("home_dir", group.UserSettings.HomeDir)
- form.Set("max_sessions", strconv.FormatInt(int64(group.UserSettings.MaxSessions), 10))
- form.Set("quota_files", strconv.FormatInt(int64(group.UserSettings.QuotaFiles), 10))
- form.Set("quota_size", strconv.FormatInt(group.UserSettings.QuotaSize, 10))
- form.Set("upload_bandwidth", strconv.FormatInt(group.UserSettings.UploadBandwidth, 10))
- form.Set("download_bandwidth", strconv.FormatInt(group.UserSettings.DownloadBandwidth, 10))
- form.Set("upload_data_transfer", "0")
- form.Set("download_data_transfer", "0")
- form.Set("total_data_transfer", "0")
- form.Set("max_upload_file_size", "0")
- form.Set("default_shares_expiration", "0")
- form.Set("max_shares_expiration", "0")
- form.Set("expires_in", "0")
- form.Set("password_expiration", "0")
- form.Set("password_strength", "0")
- form.Set("external_auth_cache_time", "0")
- form.Set("fs_provider", strconv.FormatInt(int64(group.UserSettings.FsConfig.Provider), 10))
- form.Set("sftp_endpoint", group.UserSettings.FsConfig.SFTPConfig.Endpoint)
- form.Set("sftp_username", group.UserSettings.FsConfig.SFTPConfig.Username)
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, path.Join(webGroupPath, group.Name), &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nError500Message)
- form.Set("sftp_buffer_size", strconv.FormatInt(group.UserSettings.FsConfig.SFTPConfig.BufferSize, 10))
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webGroupPath, group.Name), &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webGroupPath, group.Name), &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorFsCredentialsRequired)
- form.Set("sftp_password", defaultPassword)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webGroupPath, group.Name), &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, err = http.NewRequest(http.MethodDelete, path.Join(groupPath, group.Name), nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webGroupPath, group.Name), &b)
- assert.NoError(t, err)
- req.Header.Set("Content-Type", contentType)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestUpdateWebFolderMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- folderName := "vfolderupdate"
- folderDesc := "updated desc"
- folder := vfs.BaseVirtualFolder{
- Name: folderName,
- MappedPath: filepath.Join(os.TempDir(), "folderupdate"),
- Description: "dsc",
- }
- _, _, err = httpdtest.AddFolder(folder, http.StatusCreated)
- newMappedPath := folder.MappedPath + "1"
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("mapped_path", newMappedPath)
- form.Set("name", folderName)
- form.Set("description", folderDesc)
- form.Set(csrfFormToken, "")
- b, contentType, err := getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidCSRF)
- form.Set(csrfFormToken, csrfToken)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusSeeOther, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(folderPath, folderName), nil)
- setBearerForReq(req, apiToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = render.DecodeJSON(rr.Body, &folder)
- assert.NoError(t, err)
- assert.Equal(t, newMappedPath, folder.MappedPath)
- assert.Equal(t, folderName, folder.Name)
- assert.Equal(t, folderDesc, folder.Description)
- // parse form error
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName)+"??a=a%B3%A2%G3", &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Contains(t, rr.Body.String(), util.I18nErrorInvalidForm)
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName+"1"), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- form.Set("mapped_path", "arelative/path")
- b, contentType, err = getMultipartFormData(form, "", "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(webFolderPath, folderName), &b)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- req.Header.Set("Content-Type", contentType)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // render update folder page
- req, err = http.NewRequest(http.MethodGet, path.Join(webFolderPath, folderName), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, path.Join(webFolderPath, folderName+"1"), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webFolderPath, folderName), nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webFolderPath, folderName), nil)
- setJWTCookieForReq(req, apiToken) // api token is not accepted
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusFound, rr)
- assert.Equal(t, webLoginPath, rr.Header().Get("Location"))
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webFolderPath, folderName), nil)
- setJWTCookieForReq(req, webToken)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestWebFoldersMock(t *testing.T) {
- webToken, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- apiToken, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- mappedPath1 := filepath.Join(os.TempDir(), "vfolder1")
- mappedPath2 := filepath.Join(os.TempDir(), "vfolder2")
- folderName1 := filepath.Base(mappedPath1)
- folderName2 := filepath.Base(mappedPath2)
- folderDesc1 := "vfolder1 desc"
- folderDesc2 := "vfolder2 desc"
- folders := []vfs.BaseVirtualFolder{
- {
- Name: folderName1,
- MappedPath: mappedPath1,
- Description: folderDesc1,
- },
- {
- Name: folderName2,
- MappedPath: mappedPath2,
- Description: folderDesc2,
- },
- }
- for _, folder := range folders {
- folderAsJSON, err := json.Marshal(folder)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, folderPath, bytes.NewBuffer(folderAsJSON))
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusCreated, rr)
- }
- req, err := http.NewRequest(http.MethodGet, folderPath, nil)
- assert.NoError(t, err)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var foldersGet []vfs.BaseVirtualFolder
- err = render.DecodeJSON(rr.Body, &foldersGet)
- assert.NoError(t, err)
- numFound := 0
- for _, f := range foldersGet {
- if f.Name == folderName1 {
- assert.Equal(t, mappedPath1, f.MappedPath)
- assert.Equal(t, folderDesc1, f.Description)
- numFound++
- }
- if f.Name == folderName2 {
- assert.Equal(t, mappedPath2, f.MappedPath)
- assert.Equal(t, folderDesc2, f.Description)
- numFound++
- }
- }
- assert.Equal(t, 2, numFound)
- req, err = http.NewRequest(http.MethodGet, webFoldersPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webFoldersPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- for _, folder := range folders {
- req, _ := http.NewRequest(http.MethodDelete, path.Join(folderPath, folder.Name), nil)
- setBearerForReq(req, apiToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- }
- func TestAdminForgotPassword(t *testing.T) {
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err := smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Filters.RequirePasswordChange = true
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webAdminForgotPwdPath, nil)
- assert.NoError(t, err)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminResetPwdPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webLoginPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form := make(url.Values)
- form.Set("username", "")
- // no csrf token
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusForbidden, rr.Code)
- // empty username
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorUsernameRequired)
- lastResetCode = ""
- form.Set("username", altAdminUsername)
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- form = make(url.Values)
- req, err = http.NewRequest(http.MethodPost, webAdminResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusForbidden, rr.Code)
- // no password
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdGeneric)
- // no code
- form.Set("password", defaultPassword)
- form.Set("confirm_password", defaultPassword)
- req, err = http.NewRequest(http.MethodPost, webAdminResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdGeneric)
- // ok
- form.Set("code", lastResetCode)
- req, err = http.NewRequest(http.MethodPost, webAdminResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- form.Set("username", altAdminUsername)
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- // not working smtp server
- smtpCfg = smtp.Config{
- Host: "127.0.0.1",
- Port: 3526,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- form = make(url.Values)
- form.Set("username", altAdminUsername)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPwdResetSendEmail)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- form.Set("username", altAdminUsername)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webAdminForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPwdResetGeneric)
- req, err = http.NewRequest(http.MethodGet, webAdminForgotPwdPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webAdminResetPwdPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- admin, _, err = httpdtest.GetAdminByUsername(admin.Username, http.StatusOK)
- assert.NoError(t, err)
- assert.False(t, admin.Filters.RequirePasswordChange)
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- }
- func TestUserForgotPassword(t *testing.T) {
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err := smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- u := getTestUser()
- u.Email = "user@test.com"
- u.Filters.WebClient = []string{sdk.WebClientPasswordResetDisabled}
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, webClientForgotPwdPath, nil)
- assert.NoError(t, err)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientResetPwdPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webClientLoginPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- form := make(url.Values)
- form.Set("username", "")
- // no csrf token
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusForbidden, rr.Code)
- // empty username
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- form.Set(csrfFormToken, csrfToken)
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorUsernameRequired)
- // user cannot reset the password
- form.Set("username", user.Username)
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorPwdResetForbidded)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled}
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- lastResetCode = ""
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- // no csrf token
- form = make(url.Values)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusForbidden, rr.Code)
- // no password
- form.Set(csrfFormToken, csrfToken)
- form.Set("password", "")
- form.Set("confirm_password", "")
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdGeneric)
- // passwords mismatch
- form.Set("password", altAdminPassword)
- form.Set("code", lastResetCode)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdNoMatch)
- // no code
- form.Del("code")
- form.Set("confirm_password", altAdminPassword)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdGeneric)
- // ok
- form.Set("code", lastResetCode)
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", user.Username)
- lastResetCode = ""
- req, err = http.NewRequest(http.MethodPost, webClientForgotPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusFound, rr.Code)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- req, err = http.NewRequest(http.MethodGet, webClientForgotPwdPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, webClientResetPwdPath, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- // user does not exist anymore
- form = make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("code", lastResetCode)
- form.Set("password", "pwd")
- form.Set("confirm_password", "pwd")
- req, err = http.NewRequest(http.MethodPost, webClientResetPwdPath, bytes.NewBuffer([]byte(form.Encode())))
- assert.NoError(t, err)
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- assert.Equal(t, http.StatusOK, rr.Code)
- assert.Contains(t, rr.Body.String(), util.I18nErrorChangePwdGeneric)
- }
- func TestAPIForgotPassword(t *testing.T) {
- smtpCfg := smtp.Config{
- Host: "127.0.0.1",
- Port: 3525,
- From: "notification@example.com",
- TemplatesPath: "templates",
- }
- err := smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Email = ""
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- // no email, forgot pwd will not work
- lastResetCode = ""
- req, err := http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Your account does not have an email address")
- admin.Email = "admin@test.com"
- admin, _, err = httpdtest.UpdateAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- // invalid JSON
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/reset-password"), bytes.NewBuffer([]byte(`{`)))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- resetReq := make(map[string]string)
- resetReq["code"] = lastResetCode
- resetReq["password"] = defaultPassword
- asJSON, err := json.Marshal(resetReq)
- assert.NoError(t, err)
- // a user cannot use an admin code
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "invalid confirmation code")
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // the same code cannot be reused
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "confirmation code not found")
- admin, err = dataprovider.AdminExists(altAdminUsername)
- assert.NoError(t, err)
- match, err := admin.CheckPassword(defaultPassword)
- assert.NoError(t, err)
- assert.True(t, match)
- lastResetCode = ""
- // now the same for a user
- u := getTestUser()
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "Your account does not have an email address")
- user.Email = "user@test.com"
- user, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- // invalid JSON
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/reset-password"), bytes.NewBuffer([]byte(`{`)))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- // remove the reset password permission
- user.Filters.WebClient = []string{sdk.WebClientPasswordResetDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- resetReq["code"] = lastResetCode
- resetReq["password"] = altAdminPassword
- asJSON, err = json.Marshal(resetReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "you are not allowed to reset your password")
- user.Filters.WebClient = []string{sdk.WebClientSharesDisabled}
- _, _, err = httpdtest.UpdateUser(user, http.StatusOK, "")
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- // the same code cannot be reused
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "confirmation code not found")
- user, err = dataprovider.UserExists(defaultUsername, "")
- assert.NoError(t, err)
- err = bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(altAdminPassword))
- assert.NoError(t, err)
- lastResetCode = ""
- // a request for a missing admin/user will be silently ignored
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, "missing-admin", "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Empty(t, lastResetCode)
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, "missing-user", "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Empty(t, lastResetCode)
- lastResetCode = ""
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.GreaterOrEqual(t, len(lastResetCode), 20)
- smtpCfg = smtp.Config{}
- err = smtpCfg.Initialize(configDir, true)
- require.NoError(t, err)
- // without an smtp configuration reset password is not available
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "No SMTP configuration")
- req, err = http.NewRequest(http.MethodPost, path.Join(userPath, defaultUsername, "/forgot-password"), nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "No SMTP configuration")
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- // the admin does not exist anymore
- resetReq["code"] = lastResetCode
- resetReq["password"] = altAdminPassword
- asJSON, err = json.Marshal(resetReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, path.Join(adminPath, altAdminUsername, "/reset-password"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusBadRequest, rr)
- assert.Contains(t, rr.Body.String(), "unable to associate the confirmation code with an existing admin")
- _, err = httpdtest.RemoveUser(user, http.StatusOK)
- assert.NoError(t, err)
- err = os.RemoveAll(user.GetHomeDir())
- assert.NoError(t, err)
- }
- func TestProviderClosedMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- // create a role admin
- role, resp, err := httpdtest.AddRole(getTestRole(), http.StatusCreated)
- assert.NoError(t, err, string(resp))
- a := getTestAdmin()
- a.Username = altAdminUsername
- a.Password = altAdminPassword
- a.Role = role.Name
- a.Permissions = []string{dataprovider.PermAdminAddUsers, dataprovider.PermAdminChangeUsers,
- dataprovider.PermAdminDeleteUsers, dataprovider.PermAdminViewUsers}
- admin, _, err := httpdtest.AddAdmin(a, http.StatusCreated)
- assert.NoError(t, err)
- altToken, err := getJWTWebTokenFromTestServer(altAdminUsername, altAdminPassword)
- assert.NoError(t, err)
- dataprovider.Close()
- testReq := make(map[string]any)
- testReq["password"] = redactedSecret
- asJSON, err := json.Marshal(testReq)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodPost, path.Join(webConfigsPath, "smtp", "test"), bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- testReq["base_redirect_url"] = "http://localhost"
- testReq["client_secret"] = redactedSecret
- asJSON, err = json.Marshal(testReq)
- assert.NoError(t, err)
- req, err = http.NewRequest(http.MethodPost, webOAuth2TokenPath, bytes.NewBuffer(asJSON))
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webConfigsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, webConfigsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- getJSONFolders := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, _ := http.NewRequest(http.MethodGet, webFoldersPath+jsonAPISuffix, nil)
- setJWTCookieForReq(req, token)
- executeRequest(req)
- }
- getJSONFolders()
- getJSONGroups := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, _ := http.NewRequest(http.MethodGet, webGroupsPath+jsonAPISuffix, nil)
- setJWTCookieForReq(req, token)
- executeRequest(req)
- }
- getJSONGroups()
- getJSONUsers := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, _ := http.NewRequest(http.MethodGet, webUsersPath+jsonAPISuffix, nil)
- setJWTCookieForReq(req, token)
- executeRequest(req)
- }
- getJSONUsers()
- req, _ = http.NewRequest(http.MethodGet, webUserPath+"/0", nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- form := make(url.Values)
- form.Set(csrfFormToken, csrfToken)
- form.Set("username", "test")
- req, _ = http.NewRequest(http.MethodPost, webUserPath+"/0", strings.NewReader(form.Encode()))
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodGet, path.Join(webAdminPath, defaultTokenAuthUser), nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webAdminPath, defaultTokenAuthUser), strings.NewReader(form.Encode()))
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- getJSONAdmins := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, _ := http.NewRequest(http.MethodGet, webAdminsPath+jsonAPISuffix, nil)
- setJWTCookieForReq(req, token)
- executeRequest(req)
- }
- getJSONAdmins()
- req, _ = http.NewRequest(http.MethodGet, path.Join(webFolderPath, defaultTokenAuthUser), nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodPost, path.Join(webFolderPath, defaultTokenAuthUser), strings.NewReader(form.Encode()))
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodPost, webUserPath, strings.NewReader(form.Encode()))
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodPost, webUserPath, strings.NewReader(form.Encode()))
- setJWTCookieForReq(req, altToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodGet, webIPListPath+"/1/a", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, webIPListPath+"/1/a", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- getJSONRoles := func() {
- defer func() {
- rcv := recover()
- assert.Equal(t, http.ErrAbortHandler, rcv)
- }()
- req, err := http.NewRequest(http.MethodGet, webAdminRolesPath+jsonAPISuffix, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- executeRequest(req)
- }
- getJSONRoles()
- req, err = http.NewRequest(http.MethodGet, path.Join(webAdminRolePath, role.Name), nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, err = http.NewRequest(http.MethodPost, path.Join(webAdminRolePath, role.Name), strings.NewReader(form.Encode()))
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- err = config.LoadConfig(configDir, "")
- assert.NoError(t, err)
- providerConf := config.GetProviderConf()
- providerConf.BackupsPath = backupsPath
- err = dataprovider.Initialize(providerConf, configDir, true)
- assert.NoError(t, err)
- if config.GetProviderConf().Driver != dataprovider.MemoryDataProviderName {
- _, err = httpdtest.RemoveAdmin(admin, http.StatusOK)
- assert.NoError(t, err)
- _, err = httpdtest.RemoveRole(role, http.StatusOK)
- assert.NoError(t, err)
- }
- }
- func TestWebConnectionsMock(t *testing.T) {
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webConnectionsPath, nil)
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webConnectionsPath, "id"), nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webConnectionsPath, "id"), nil)
- setJWTCookieForReq(req, token)
- setCSRFHeaderForReq(req, "csrfToken")
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "Invalid token")
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- assert.NoError(t, err)
- req, _ = http.NewRequest(http.MethodDelete, path.Join(webConnectionsPath, "id"), nil)
- setJWTCookieForReq(req, token)
- setCSRFHeaderForReq(req, csrfToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- func TestGetWebStatusMock(t *testing.T) {
- oldConfig := config.GetCommonConfig()
- cfg := config.GetCommonConfig()
- cfg.RateLimitersConfig = []common.RateLimiterConfig{
- {
- Average: 1,
- Period: 1000,
- Burst: 1,
- Type: 1,
- Protocols: []string{common.ProtocolFTP},
- },
- }
- err := common.Initialize(cfg, 0)
- assert.NoError(t, err)
- token, err := getJWTWebTokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, _ := http.NewRequest(http.MethodGet, webStatusPath, nil)
- setJWTCookieForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err = common.Initialize(oldConfig, 0)
- assert.NoError(t, err)
- }
- func TestStaticFilesMock(t *testing.T) {
- req, err := http.NewRequest(http.MethodGet, "/static/favicon.ico", nil)
- assert.NoError(t, err)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, "/openapi/openapi.yaml", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, "/static", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusMovedPermanently, rr)
- location := rr.Header().Get("Location")
- assert.Equal(t, "/static/", location)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- req, err = http.NewRequest(http.MethodGet, "/openapi", nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusMovedPermanently, rr)
- location = rr.Header().Get("Location")
- assert.Equal(t, "/openapi/", location)
- req, err = http.NewRequest(http.MethodGet, location, nil)
- assert.NoError(t, err)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- }
- func TestPasswordChangeRequired(t *testing.T) {
- user := getTestUser()
- assert.False(t, user.MustChangePassword())
- user.Filters.RequirePasswordChange = true
- assert.True(t, user.MustChangePassword())
- user.Filters.RequirePasswordChange = false
- assert.False(t, user.MustChangePassword())
- user.Filters.PasswordExpiration = 2
- user.LastPasswordChange = util.GetTimeAsMsSinceEpoch(time.Now())
- assert.False(t, user.MustChangePassword())
- user.LastPasswordChange = util.GetTimeAsMsSinceEpoch(time.Now().Add(49 * time.Hour))
- assert.False(t, user.MustChangePassword())
- user.LastPasswordChange = util.GetTimeAsMsSinceEpoch(time.Now().Add(-49 * time.Hour))
- assert.True(t, user.MustChangePassword())
- }
- func TestPasswordExpiresIn(t *testing.T) {
- user := getTestUser()
- user.Filters.PasswordExpiration = 30
- user.LastPasswordChange = util.GetTimeAsMsSinceEpoch(time.Now().Add(-15*24*time.Hour + 1*time.Hour))
- res := user.PasswordExpiresIn()
- assert.Equal(t, 15, res)
- user.Filters.PasswordExpiration = 15
- res = user.PasswordExpiresIn()
- assert.Equal(t, 1, res)
- user.LastPasswordChange = util.GetTimeAsMsSinceEpoch(time.Now().Add(-15*24*time.Hour - 1*time.Hour))
- res = user.PasswordExpiresIn()
- assert.Equal(t, 0, res)
- user.Filters.PasswordExpiration = 5
- res = user.PasswordExpiresIn()
- assert.Equal(t, -10, res)
- }
- func TestSecondFactorRequirements(t *testing.T) {
- user := getTestUser()
- user.Filters.TwoFactorAuthProtocols = []string{common.ProtocolHTTP, common.ProtocolSSH}
- assert.True(t, user.MustSetSecondFactor())
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolFTP))
- assert.True(t, user.MustSetSecondFactorForProtocol(common.ProtocolHTTP))
- assert.True(t, user.MustSetSecondFactorForProtocol(common.ProtocolSSH))
- user.Filters.TOTPConfig.Enabled = true
- assert.True(t, user.MustSetSecondFactor())
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolFTP))
- assert.True(t, user.MustSetSecondFactorForProtocol(common.ProtocolHTTP))
- assert.True(t, user.MustSetSecondFactorForProtocol(common.ProtocolSSH))
- user.Filters.TOTPConfig.Protocols = []string{common.ProtocolHTTP}
- assert.True(t, user.MustSetSecondFactor())
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolFTP))
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolHTTP))
- assert.True(t, user.MustSetSecondFactorForProtocol(common.ProtocolSSH))
- user.Filters.TOTPConfig.Protocols = []string{common.ProtocolHTTP, common.ProtocolSSH}
- assert.False(t, user.MustSetSecondFactor())
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolFTP))
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolHTTP))
- assert.False(t, user.MustSetSecondFactorForProtocol(common.ProtocolSSH))
- }
- func startOIDCMockServer() {
- go func() {
- http.HandleFunc("/", func(w http.ResponseWriter, _ *http.Request) {
- fmt.Fprintf(w, "OK\n")
- })
- http.HandleFunc("/auth/realms/sftpgo/.well-known/openid-configuration", func(w http.ResponseWriter, _ *http.Request) {
- fmt.Fprintf(w, `{"issuer":"http://127.0.0.1:11111/auth/realms/sftpgo","authorization_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/auth","token_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/token","introspection_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/token/introspect","userinfo_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/userinfo","end_session_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/logout","frontchannel_logout_session_supported":true,"frontchannel_logout_supported":true,"jwks_uri":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/certs","check_session_iframe":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/login-status-iframe.html","grant_types_supported":["authorization_code","implicit","refresh_token","password","client_credentials","urn:ietf:params:oauth:grant-type:device_code","urn:openid:params:grant-type:ciba"],"response_types_supported":["code","none","id_token","token","id_token token","code id_token","code token","code id_token token"],"subject_types_supported":["public","pairwise"],"id_token_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512"],"id_token_encryption_alg_values_supported":["RSA-OAEP","RSA-OAEP-256","RSA1_5"],"id_token_encryption_enc_values_supported":["A256GCM","A192GCM","A128GCM","A128CBC-HS256","A192CBC-HS384","A256CBC-HS512"],"userinfo_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512","none"],"request_object_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512","none"],"request_object_encryption_alg_values_supported":["RSA-OAEP","RSA-OAEP-256","RSA1_5"],"request_object_encryption_enc_values_supported":["A256GCM","A192GCM","A128GCM","A128CBC-HS256","A192CBC-HS384","A256CBC-HS512"],"response_modes_supported":["query","fragment","form_post","query.jwt","fragment.jwt","form_post.jwt","jwt"],"registration_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/clients-registrations/openid-connect","token_endpoint_auth_methods_supported":["private_key_jwt","client_secret_basic","client_secret_post","tls_client_auth","client_secret_jwt"],"token_endpoint_auth_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512"],"introspection_endpoint_auth_methods_supported":["private_key_jwt","client_secret_basic","client_secret_post","tls_client_auth","client_secret_jwt"],"introspection_endpoint_auth_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512"],"authorization_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512"],"authorization_encryption_alg_values_supported":["RSA-OAEP","RSA-OAEP-256","RSA1_5"],"authorization_encryption_enc_values_supported":["A256GCM","A192GCM","A128GCM","A128CBC-HS256","A192CBC-HS384","A256CBC-HS512"],"claims_supported":["aud","sub","iss","auth_time","name","given_name","family_name","preferred_username","email","acr"],"claim_types_supported":["normal"],"claims_parameter_supported":true,"scopes_supported":["openid","phone","email","web-origins","offline_access","microprofile-jwt","profile","address","roles"],"request_parameter_supported":true,"request_uri_parameter_supported":true,"require_request_uri_registration":true,"code_challenge_methods_supported":["plain","S256"],"tls_client_certificate_bound_access_tokens":true,"revocation_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/revoke","revocation_endpoint_auth_methods_supported":["private_key_jwt","client_secret_basic","client_secret_post","tls_client_auth","client_secret_jwt"],"revocation_endpoint_auth_signing_alg_values_supported":["PS384","ES384","RS384","HS256","HS512","ES256","RS256","HS384","ES512","PS256","PS512","RS512"],"backchannel_logout_supported":true,"backchannel_logout_session_supported":true,"device_authorization_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/auth/device","backchannel_token_delivery_modes_supported":["poll","ping"],"backchannel_authentication_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/ext/ciba/auth","backchannel_authentication_request_signing_alg_values_supported":["PS384","ES384","RS384","ES256","RS256","ES512","PS256","PS512","RS512"],"require_pushed_authorization_requests":false,"pushed_authorization_request_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/ext/par/request","mtls_endpoint_aliases":{"token_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/token","revocation_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/revoke","introspection_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/token/introspect","device_authorization_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/auth/device","registration_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/clients-registrations/openid-connect","userinfo_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/userinfo","pushed_authorization_request_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/ext/par/request","backchannel_authentication_endpoint":"http://127.0.0.1:11111/auth/realms/sftpgo/protocol/openid-connect/ext/ciba/auth"}}`)
- })
- http.HandleFunc("/404", func(w http.ResponseWriter, _ *http.Request) {
- w.WriteHeader(http.StatusNotFound)
- fmt.Fprintf(w, "Not found\n")
- })
- if err := http.ListenAndServe(oidcMockAddr, nil); err != nil {
- logger.ErrorToConsole("could not start HTTP notification server: %v", err)
- os.Exit(1)
- }
- }()
- waitTCPListening(oidcMockAddr)
- }
- func waitForUsersQuotaScan(t *testing.T, token string) {
- for {
- var scans []common.ActiveQuotaScan
- req, _ := http.NewRequest(http.MethodGet, quotaScanPath, nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err := render.DecodeJSON(rr.Body, &scans)
- if !assert.NoError(t, err, "Error getting active scans") {
- break
- }
- if len(scans) == 0 {
- break
- }
- time.Sleep(100 * time.Millisecond)
- }
- }
- func waitForFoldersQuotaScanPath(t *testing.T, token string) {
- var scans []common.ActiveVirtualFolderQuotaScan
- for {
- req, _ := http.NewRequest(http.MethodGet, quotaScanVFolderPath, nil)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- err := render.DecodeJSON(rr.Body, &scans)
- if !assert.NoError(t, err, "Error getting active folders scans") {
- break
- }
- if len(scans) == 0 {
- break
- }
- time.Sleep(100 * time.Millisecond)
- }
- }
- func waitTCPListening(address string) {
- for {
- conn, err := net.Dial("tcp", address)
- if err != nil {
- logger.WarnToConsole("tcp server %v not listening: %v", address, err)
- time.Sleep(100 * time.Millisecond)
- continue
- }
- logger.InfoToConsole("tcp server %v now listening", address)
- conn.Close()
- break
- }
- }
- func startSMTPServer() {
- go func() {
- if err := smtpd.ListenAndServe(smtpServerAddr, func(_ net.Addr, _ string, _ []string, data []byte) error {
- re := regexp.MustCompile(`code is ".*?"`)
- code := strings.TrimPrefix(string(re.Find(data)), "code is ")
- lastResetCode = strings.ReplaceAll(code, "\"", "")
- return nil
- }, "SFTPGo test", "localhost"); err != nil {
- logger.ErrorToConsole("could not start SMTP server: %v", err)
- os.Exit(1)
- }
- }()
- waitTCPListening(smtpServerAddr)
- }
- func getTestAdmin() dataprovider.Admin {
- return dataprovider.Admin{
- Username: defaultTokenAuthUser,
- Password: defaultTokenAuthPass,
- Status: 1,
- Permissions: []string{dataprovider.PermAdminAny},
- Email: "admin@example.com",
- Description: "test admin",
- }
- }
- func getTestGroup() dataprovider.Group {
- return dataprovider.Group{
- BaseGroup: sdk.BaseGroup{
- Name: "test_group",
- Description: "test group description",
- },
- }
- }
- func getTestRole() dataprovider.Role {
- return dataprovider.Role{
- Name: "test_role",
- Description: "test role description",
- }
- }
- func getTestUser() dataprovider.User {
- user := dataprovider.User{
- BaseUser: sdk.BaseUser{
- Username: defaultUsername,
- Password: defaultPassword,
- HomeDir: filepath.Join(homeBasePath, defaultUsername),
- Status: 1,
- Description: "test user",
- },
- }
- user.Permissions = make(map[string][]string)
- user.Permissions["/"] = defaultPerms
- return user
- }
- func getTestSFTPUser() dataprovider.User {
- u := getTestUser()
- u.Username = u.Username + "_sftp"
- u.FsConfig.Provider = sdk.SFTPFilesystemProvider
- u.FsConfig.SFTPConfig.Endpoint = sftpServerAddr
- u.FsConfig.SFTPConfig.Username = defaultUsername
- u.FsConfig.SFTPConfig.Password = kms.NewPlainSecret(defaultPassword)
- return u
- }
- func getUserAsJSON(t *testing.T, user dataprovider.User) []byte {
- json, err := json.Marshal(user)
- assert.NoError(t, err)
- return json
- }
- func getCSRFTokenMock(loginURLPath, remoteAddr string) (string, error) {
- req, err := http.NewRequest(http.MethodGet, loginURLPath, nil)
- if err != nil {
- return "", err
- }
- req.RemoteAddr = remoteAddr
- rr := executeRequest(req)
- return getCSRFTokenFromBody(bytes.NewBuffer(rr.Body.Bytes()))
- }
- func getCSRFToken(url string) (string, error) {
- req, err := http.NewRequest(http.MethodGet, url, nil)
- if err != nil {
- return "", err
- }
- resp, err := httpclient.GetHTTPClient().Do(req)
- if err != nil {
- return "", err
- }
- defer resp.Body.Close()
- return getCSRFTokenFromBody(resp.Body)
- }
- func getCSRFTokenFromBody(body io.Reader) (string, error) {
- doc, err := html.Parse(body)
- if err != nil {
- return "", err
- }
- var csrfToken string
- var f func(*html.Node)
- f = func(n *html.Node) {
- if n.Type == html.ElementNode && n.Data == "input" {
- var name, value string
- for _, attr := range n.Attr {
- if attr.Key == "value" {
- value = attr.Val
- }
- if attr.Key == "name" {
- name = attr.Val
- }
- }
- if name == csrfFormToken {
- csrfToken = value
- return
- }
- }
- for c := n.FirstChild; c != nil; c = c.NextSibling {
- f(c)
- }
- }
- f(doc)
- return csrfToken, nil
- }
- func getLoginForm(username, password, csrfToken string) url.Values {
- form := make(url.Values)
- form.Set("username", username)
- form.Set("password", password)
- form.Set(csrfFormToken, csrfToken)
- return form
- }
- func setCSRFHeaderForReq(req *http.Request, csrfToken string) {
- req.Header.Set("X-CSRF-TOKEN", csrfToken)
- }
- func setBearerForReq(req *http.Request, jwtToken string) {
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", jwtToken))
- }
- func setAPIKeyForReq(req *http.Request, apiKey, username string) {
- if username != "" {
- apiKey += "." + username
- }
- req.Header.Set("X-SFTPGO-API-KEY", apiKey)
- }
- func setJWTCookieForReq(req *http.Request, jwtToken string) {
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Cookie", fmt.Sprintf("jwt=%v", jwtToken))
- }
- func getJWTAPITokenFromTestServer(username, password string) (string, error) {
- return getJWTAPITokenFromTestServerWithPasscode(username, password, "")
- }
- func getJWTAPITokenFromTestServerWithPasscode(username, password, passcode string) (string, error) {
- req, _ := http.NewRequest(http.MethodGet, tokenPath, nil)
- req.SetBasicAuth(username, password)
- if passcode != "" {
- req.Header.Set("X-SFTPGO-OTP", passcode)
- }
- rr := executeRequest(req)
- if rr.Code != http.StatusOK {
- return "", fmt.Errorf("unexpected status code %v", rr.Code)
- }
- responseHolder := make(map[string]any)
- err := render.DecodeJSON(rr.Body, &responseHolder)
- if err != nil {
- return "", err
- }
- return responseHolder["access_token"].(string), nil
- }
- func getJWTAPIUserTokenFromTestServer(username, password string) (string, error) {
- req, _ := http.NewRequest(http.MethodGet, userTokenPath, nil)
- req.SetBasicAuth(username, password)
- rr := executeRequest(req)
- if rr.Code != http.StatusOK {
- return "", fmt.Errorf("unexpected status code %v", rr.Code)
- }
- responseHolder := make(map[string]any)
- err := render.DecodeJSON(rr.Body, &responseHolder)
- if err != nil {
- return "", err
- }
- return responseHolder["access_token"].(string), nil
- }
- func getJWTWebToken(username, password string) (string, error) {
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- if err != nil {
- return "", err
- }
- form := getLoginForm(username, password, csrfToken)
- req, _ := http.NewRequest(http.MethodPost, httpBaseURL+webLoginPath,
- bytes.NewBuffer([]byte(form.Encode())))
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- client := &http.Client{
- Timeout: 10 * time.Second,
- CheckRedirect: func(_ *http.Request, _ []*http.Request) error {
- return http.ErrUseLastResponse
- },
- }
- resp, err := client.Do(req)
- if err != nil {
- return "", err
- }
- defer resp.Body.Close()
- if resp.StatusCode != http.StatusFound {
- return "", fmt.Errorf("unexpected status code %v", resp.StatusCode)
- }
- cookie := resp.Header.Get("Set-Cookie")
- if strings.HasPrefix(cookie, "jwt=") {
- return cookie[4:], nil
- }
- return "", errors.New("no cookie found")
- }
- func getCookieFromResponse(rr *httptest.ResponseRecorder) (string, error) {
- cookie := strings.Split(rr.Header().Get("Set-Cookie"), ";")
- if strings.HasPrefix(cookie[0], "jwt=") {
- return cookie[0][4:], nil
- }
- return "", errors.New("no cookie found")
- }
- func getJWTWebClientTokenFromTestServerWithAddr(username, password, remoteAddr string) (string, error) {
- csrfToken, err := getCSRFTokenMock(webClientLoginPath, remoteAddr)
- if err != nil {
- return "", err
- }
- form := getLoginForm(username, password, csrfToken)
- req, _ := http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = remoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr := executeRequest(req)
- if rr.Code != http.StatusFound {
- return "", fmt.Errorf("unexpected status code %v", rr)
- }
- return getCookieFromResponse(rr)
- }
- func getJWTWebClientTokenFromTestServer(username, password string) (string, error) {
- csrfToken, err := getCSRFToken(httpBaseURL + webClientLoginPath)
- if err != nil {
- return "", err
- }
- form := getLoginForm(username, password, csrfToken)
- req, _ := http.NewRequest(http.MethodPost, webClientLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr := executeRequest(req)
- if rr.Code != http.StatusFound {
- return "", fmt.Errorf("unexpected status code %v", rr)
- }
- return getCookieFromResponse(rr)
- }
- func getJWTWebTokenFromTestServer(username, password string) (string, error) {
- csrfToken, err := getCSRFToken(httpBaseURL + webLoginPath)
- if err != nil {
- return "", err
- }
- form := getLoginForm(username, password, csrfToken)
- req, _ := http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
- req.RemoteAddr = defaultRemoteAddr
- req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
- rr := executeRequest(req)
- if rr.Code != http.StatusFound {
- return "", fmt.Errorf("unexpected status code %v", rr)
- }
- return getCookieFromResponse(rr)
- }
- func executeRequest(req *http.Request) *httptest.ResponseRecorder {
- rr := httptest.NewRecorder()
- testServer.Config.Handler.ServeHTTP(rr, req)
- return rr
- }
- func checkResponseCode(t *testing.T, expected int, rr *httptest.ResponseRecorder) {
- assert.Equal(t, expected, rr.Code, rr.Body.String())
- }
- func createTestFile(path string, size int64) error {
- baseDir := filepath.Dir(path)
- if _, err := os.Stat(baseDir); errors.Is(err, fs.ErrNotExist) {
- err = os.MkdirAll(baseDir, os.ModePerm)
- if err != nil {
- return err
- }
- }
- content := make([]byte, size)
- if size > 0 {
- _, err := rand.Read(content)
- if err != nil {
- return err
- }
- }
- return os.WriteFile(path, content, os.ModePerm)
- }
- func getExitCodeScriptContent(exitCode int) []byte {
- content := []byte("#!/bin/sh\n\n")
- content = append(content, []byte(fmt.Sprintf("exit %v", exitCode))...)
- return content
- }
- func getMultipartFormData(values url.Values, fileFieldName, filePath string) (bytes.Buffer, string, error) {
- var b bytes.Buffer
- w := multipart.NewWriter(&b)
- for k, v := range values {
- for _, s := range v {
- if err := w.WriteField(k, s); err != nil {
- return b, "", err
- }
- }
- }
- if len(fileFieldName) > 0 && len(filePath) > 0 {
- fw, err := w.CreateFormFile(fileFieldName, filepath.Base(filePath))
- if err != nil {
- return b, "", err
- }
- f, err := os.Open(filePath)
- if err != nil {
- return b, "", err
- }
- defer f.Close()
- if _, err = io.Copy(fw, f); err != nil {
- return b, "", err
- }
- }
- err := w.Close()
- return b, w.FormDataContentType(), err
- }
- func generateTOTPPasscode(secret string) (string, error) {
- return totp.GenerateCodeCustom(secret, time.Now(), totp.ValidateOpts{
- Period: 30,
- Skew: 1,
- Digits: otp.DigitsSix,
- Algorithm: otp.AlgorithmSHA1,
- })
- }
- func isDbDefenderSupported() bool {
- // SQLite shares the implementation with other SQL-based provider but it makes no sense
- // to use it outside test cases
- switch dataprovider.GetProviderStatus().Driver {
- case dataprovider.MySQLDataProviderName, dataprovider.PGSQLDataProviderName,
- dataprovider.CockroachDataProviderName, dataprovider.SQLiteDataProviderName:
- return true
- default:
- return false
- }
- }
- func BenchmarkSecretDecryption(b *testing.B) {
- s := kms.NewPlainSecret("test data")
- s.SetAdditionalData("username")
- err := s.Encrypt()
- require.NoError(b, err)
- for i := 0; i < b.N; i++ {
- err = s.Clone().Decrypt()
- require.NoError(b, err)
- }
- }
|