12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669567056715672567356745675567656775678567956805681568256835684568556865687568856895690569156925693569456955696569756985699570057015702570357045705570657075708570957105711571257135714571557165717571857195720572157225723572457255726572757285729573057315732573357345735573657375738573957405741574257435744574557465747574857495750575157525753575457555756575757585759576057615762576357645765576657675768576957705771577257735774577557765777577857795780578157825783578457855786578757885789579057915792579357945795579657975798579958005801580258035804580558065807580858095810581158125813581458155816581758185819582058215822582358245825582658275828582958305831583258335834583558365837583858395840584158425843584458455846584758485849585058515852585358545855585658575858585958605861586258635864586558665867586858695870587158725873587458755876587758785879588058815882588358845885588658875888588958905891589258935894589558965897589858995900590159025903590459055906590759085909591059115912591359145915591659175918591959205921592259235924592559265927592859295930593159325933593459355936593759385939594059415942594359445945594659475948594959505951595259535954595559565957595859595960596159625963596459655966596759685969597059715972597359745975597659775978597959805981598259835984598559865987598859895990599159925993599459955996599759985999600060016002600360046005600660076008600960106011601260136014601560166017601860196020602160226023602460256026602760286029603060316032603360346035603660376038603960406041604260436044604560466047604860496050605160526053605460556056605760586059606060616062606360646065606660676068606960706071607260736074607560766077607860796080608160826083608460856086608760886089609060916092609360946095609660976098609961006101610261036104610561066107610861096110611161126113611461156116611761186119612061216122612361246125612661276128612961306131613261336134613561366137613861396140614161426143614461456146614761486149615061516152615361546155615661576158615961606161616261636164616561666167616861696170617161726173617461756176617761786179618061816182618361846185618661876188618961906191619261936194619561966197619861996200620162026203620462056206620762086209621062116212621362146215621662176218621962206221622262236224622562266227622862296230623162326233623462356236623762386239624062416242624362446245624662476248624962506251625262536254625562566257625862596260626162626263626462656266626762686269627062716272627362746275627662776278627962806281628262836284628562866287628862896290629162926293629462956296629762986299630063016302630363046305630663076308630963106311631263136314631563166317631863196320632163226323632463256326632763286329633063316332633363346335633663376338633963406341634263436344634563466347634863496350635163526353635463556356635763586359636063616362636363646365636663676368636963706371637263736374637563766377637863796380638163826383638463856386638763886389639063916392639363946395639663976398639964006401640264036404640564066407640864096410641164126413641464156416641764186419642064216422642364246425642664276428642964306431643264336434643564366437643864396440644164426443644464456446644764486449645064516452645364546455645664576458645964606461646264636464646564666467646864696470647164726473647464756476647764786479648064816482648364846485648664876488648964906491649264936494649564966497649864996500650165026503650465056506650765086509651065116512651365146515651665176518651965206521652265236524652565266527652865296530653165326533653465356536653765386539654065416542654365446545654665476548654965506551655265536554655565566557655865596560656165626563656465656566656765686569657065716572657365746575657665776578657965806581658265836584658565866587658865896590659165926593659465956596659765986599660066016602660366046605660666076608660966106611661266136614661566166617661866196620662166226623662466256626662766286629663066316632663366346635663666376638663966406641664266436644664566466647664866496650665166526653665466556656665766586659666066616662666366646665666666676668666966706671667266736674667566766677667866796680668166826683668466856686668766886689669066916692669366946695669666976698669967006701670267036704670567066707670867096710671167126713671467156716671767186719672067216722672367246725672667276728672967306731673267336734673567366737673867396740674167426743674467456746674767486749675067516752675367546755675667576758675967606761676267636764676567666767676867696770677167726773677467756776677767786779678067816782678367846785678667876788678967906791679267936794679567966797679867996800680168026803680468056806680768086809681068116812681368146815681668176818681968206821682268236824682568266827682868296830683168326833683468356836683768386839684068416842684368446845684668476848684968506851685268536854685568566857685868596860686168626863686468656866686768686869687068716872687368746875687668776878687968806881688268836884688568866887688868896890689168926893689468956896689768986899690069016902690369046905690669076908690969106911691269136914691569166917691869196920692169226923692469256926692769286929693069316932693369346935693669376938693969406941694269436944694569466947694869496950695169526953695469556956695769586959696069616962696369646965696669676968696969706971697269736974697569766977697869796980698169826983698469856986698769886989699069916992699369946995699669976998699970007001700270037004700570067007700870097010701170127013701470157016701770187019702070217022702370247025702670277028702970307031703270337034703570367037703870397040704170427043704470457046704770487049705070517052705370547055705670577058705970607061706270637064706570667067706870697070707170727073707470757076707770787079708070817082708370847085708670877088708970907091709270937094709570967097709870997100710171027103710471057106710771087109711071117112711371147115711671177118711971207121712271237124712571267127712871297130713171327133713471357136713771387139714071417142714371447145714671477148714971507151715271537154715571567157715871597160716171627163716471657166716771687169717071717172717371747175717671777178717971807181718271837184718571867187718871897190719171927193719471957196719771987199720072017202720372047205720672077208720972107211721272137214721572167217721872197220722172227223722472257226722772287229723072317232723372347235723672377238723972407241724272437244724572467247724872497250725172527253725472557256725772587259726072617262726372647265726672677268726972707271727272737274727572767277727872797280728172827283728472857286728772887289729072917292729372947295729672977298729973007301730273037304730573067307730873097310731173127313731473157316731773187319732073217322732373247325732673277328732973307331733273337334733573367337733873397340734173427343734473457346734773487349735073517352735373547355735673577358735973607361736273637364736573667367736873697370737173727373737473757376737773787379738073817382738373847385738673877388738973907391739273937394739573967397739873997400740174027403740474057406740774087409741074117412741374147415741674177418741974207421742274237424742574267427742874297430743174327433743474357436743774387439744074417442744374447445744674477448744974507451745274537454745574567457745874597460746174627463746474657466746774687469747074717472747374747475747674777478747974807481748274837484748574867487748874897490749174927493749474957496749774987499750075017502750375047505750675077508750975107511751275137514751575167517751875197520752175227523752475257526752775287529753075317532753375347535753675377538753975407541754275437544754575467547754875497550755175527553755475557556755775587559756075617562756375647565756675677568756975707571757275737574757575767577757875797580758175827583758475857586758775887589759075917592759375947595759675977598759976007601760276037604760576067607760876097610761176127613761476157616761776187619762076217622762376247625762676277628762976307631763276337634763576367637763876397640764176427643764476457646764776487649765076517652765376547655765676577658765976607661766276637664766576667667766876697670767176727673767476757676767776787679768076817682768376847685768676877688768976907691769276937694769576967697769876997700770177027703770477057706770777087709771077117712771377147715771677177718771977207721772277237724772577267727772877297730773177327733773477357736773777387739774077417742774377447745774677477748774977507751775277537754775577567757775877597760776177627763776477657766776777687769777077717772777377747775777677777778777977807781778277837784778577867787778877897790779177927793779477957796779777987799780078017802780378047805780678077808780978107811781278137814781578167817781878197820782178227823782478257826782778287829783078317832783378347835783678377838783978407841784278437844784578467847784878497850785178527853785478557856785778587859786078617862786378647865786678677868786978707871787278737874787578767877787878797880788178827883788478857886788778887889789078917892789378947895789678977898789979007901790279037904790579067907790879097910791179127913791479157916791779187919792079217922792379247925792679277928792979307931793279337934793579367937793879397940794179427943794479457946794779487949795079517952795379547955795679577958795979607961796279637964796579667967796879697970797179727973797479757976797779787979798079817982798379847985798679877988798979907991799279937994799579967997799879998000800180028003800480058006800780088009801080118012801380148015801680178018801980208021802280238024802580268027802880298030803180328033803480358036803780388039804080418042804380448045804680478048804980508051805280538054805580568057805880598060806180628063806480658066806780688069807080718072807380748075807680778078807980808081808280838084808580868087808880898090809180928093809480958096809780988099810081018102810381048105810681078108810981108111811281138114811581168117811881198120812181228123812481258126812781288129813081318132813381348135813681378138813981408141814281438144814581468147814881498150815181528153815481558156815781588159816081618162816381648165816681678168816981708171817281738174817581768177817881798180818181828183818481858186818781888189819081918192819381948195819681978198819982008201820282038204820582068207820882098210821182128213821482158216821782188219822082218222822382248225822682278228822982308231823282338234823582368237823882398240824182428243824482458246824782488249825082518252825382548255825682578258825982608261826282638264826582668267826882698270827182728273827482758276827782788279828082818282828382848285828682878288828982908291829282938294829582968297829882998300830183028303830483058306830783088309831083118312831383148315831683178318831983208321832283238324832583268327832883298330833183328333833483358336833783388339834083418342834383448345834683478348834983508351835283538354835583568357835883598360836183628363836483658366836783688369837083718372837383748375837683778378837983808381838283838384838583868387838883898390839183928393839483958396839783988399840084018402840384048405840684078408840984108411841284138414841584168417841884198420842184228423842484258426842784288429843084318432843384348435843684378438843984408441844284438444844584468447844884498450845184528453845484558456845784588459846084618462846384648465846684678468846984708471847284738474847584768477847884798480848184828483848484858486848784888489849084918492849384948495849684978498849985008501850285038504850585068507850885098510851185128513851485158516851785188519852085218522852385248525852685278528852985308531853285338534853585368537853885398540854185428543854485458546854785488549855085518552855385548555855685578558855985608561856285638564856585668567856885698570857185728573857485758576857785788579858085818582858385848585858685878588858985908591859285938594859585968597859885998600860186028603860486058606860786088609861086118612861386148615861686178618861986208621862286238624862586268627862886298630863186328633863486358636863786388639864086418642864386448645864686478648864986508651865286538654865586568657865886598660866186628663866486658666866786688669867086718672867386748675867686778678867986808681868286838684868586868687868886898690869186928693869486958696869786988699870087018702870387048705870687078708870987108711871287138714871587168717871887198720872187228723872487258726872787288729873087318732873387348735873687378738873987408741874287438744874587468747874887498750875187528753875487558756875787588759876087618762876387648765876687678768876987708771877287738774877587768777877887798780878187828783878487858786878787888789879087918792879387948795879687978798879988008801880288038804880588068807880888098810881188128813881488158816881788188819882088218822882388248825882688278828882988308831883288338834883588368837883888398840884188428843884488458846884788488849885088518852885388548855885688578858885988608861886288638864886588668867886888698870887188728873887488758876887788788879888088818882888388848885888688878888888988908891889288938894889588968897889888998900890189028903890489058906890789088909891089118912891389148915891689178918891989208921892289238924892589268927892889298930893189328933893489358936893789388939894089418942894389448945894689478948894989508951895289538954895589568957895889598960896189628963896489658966896789688969897089718972897389748975897689778978897989808981898289838984898589868987898889898990899189928993899489958996899789988999900090019002900390049005900690079008900990109011901290139014901590169017901890199020902190229023902490259026902790289029903090319032903390349035903690379038903990409041904290439044904590469047904890499050905190529053905490559056905790589059906090619062906390649065906690679068906990709071907290739074907590769077907890799080908190829083908490859086908790889089909090919092909390949095909690979098909991009101910291039104910591069107910891099110911191129113911491159116911791189119912091219122912391249125912691279128912991309131913291339134913591369137913891399140914191429143914491459146914791489149915091519152915391549155915691579158915991609161916291639164916591669167916891699170917191729173917491759176917791789179918091819182918391849185918691879188918991909191919291939194919591969197919891999200920192029203920492059206920792089209921092119212921392149215921692179218921992209221922292239224922592269227922892299230923192329233923492359236923792389239924092419242924392449245924692479248924992509251925292539254925592569257925892599260926192629263926492659266926792689269927092719272927392749275927692779278927992809281928292839284928592869287928892899290929192929293929492959296929792989299930093019302930393049305930693079308930993109311931293139314931593169317931893199320932193229323932493259326932793289329933093319332933393349335933693379338933993409341934293439344934593469347934893499350935193529353935493559356935793589359936093619362936393649365936693679368936993709371937293739374937593769377937893799380938193829383938493859386938793889389939093919392939393949395939693979398939994009401940294039404940594069407940894099410941194129413941494159416941794189419942094219422942394249425942694279428942994309431943294339434943594369437943894399440944194429443944494459446944794489449945094519452945394549455945694579458945994609461946294639464946594669467946894699470947194729473947494759476947794789479948094819482948394849485948694879488948994909491949294939494949594969497949894999500950195029503950495059506950795089509951095119512951395149515951695179518951995209521952295239524952595269527952895299530953195329533953495359536953795389539954095419542954395449545954695479548954995509551955295539554955595569557955895599560956195629563956495659566956795689569957095719572957395749575957695779578957995809581958295839584958595869587958895899590959195929593959495959596959795989599960096019602960396049605960696079608960996109611961296139614961596169617961896199620962196229623962496259626962796289629963096319632963396349635963696379638963996409641964296439644964596469647964896499650965196529653965496559656965796589659966096619662966396649665966696679668966996709671967296739674967596769677967896799680968196829683968496859686968796889689969096919692969396949695969696979698969997009701970297039704970597069707970897099710971197129713971497159716971797189719972097219722972397249725972697279728972997309731973297339734973597369737973897399740974197429743974497459746974797489749975097519752975397549755975697579758975997609761976297639764976597669767976897699770977197729773977497759776977797789779978097819782978397849785978697879788978997909791979297939794979597969797979897999800980198029803980498059806980798089809981098119812981398149815981698179818981998209821982298239824982598269827982898299830983198329833983498359836983798389839984098419842984398449845984698479848984998509851985298539854985598569857985898599860986198629863986498659866986798689869987098719872987398749875987698779878987998809881988298839884988598869887988898899890989198929893989498959896989798989899990099019902990399049905990699079908990999109911991299139914991599169917991899199920992199229923992499259926992799289929993099319932993399349935993699379938993999409941994299439944994599469947994899499950995199529953995499559956995799589959996099619962996399649965996699679968996999709971997299739974997599769977997899799980998199829983998499859986998799889989999099919992999399949995999699979998999910000100011000210003100041000510006100071000810009100101001110012100131001410015100161001710018100191002010021100221002310024100251002610027100281002910030100311003210033100341003510036100371003810039100401004110042100431004410045100461004710048100491005010051100521005310054100551005610057100581005910060100611006210063100641006510066100671006810069100701007110072100731007410075100761007710078100791008010081100821008310084100851008610087100881008910090100911009210093100941009510096100971009810099101001010110102101031010410105101061010710108101091011010111101121011310114101151011610117101181011910120101211012210123101241012510126101271012810129101301013110132101331013410135101361013710138101391014010141101421014310144101451014610147101481014910150101511015210153101541015510156101571015810159101601016110162101631016410165101661016710168101691017010171101721017310174101751017610177101781017910180101811018210183101841018510186101871018810189101901019110192101931019410195101961019710198101991020010201102021020310204102051020610207102081020910210102111021210213102141021510216102171021810219102201022110222102231022410225102261022710228102291023010231102321023310234102351023610237102381023910240102411024210243102441024510246102471024810249102501025110252102531025410255102561025710258102591026010261102621026310264102651026610267102681026910270102711027210273102741027510276102771027810279102801028110282102831028410285102861028710288102891029010291102921029310294102951029610297102981029910300103011030210303103041030510306103071030810309103101031110312103131031410315103161031710318103191032010321103221032310324103251032610327103281032910330103311033210333103341033510336103371033810339103401034110342103431034410345103461034710348103491035010351103521035310354103551035610357103581035910360103611036210363103641036510366103671036810369103701037110372103731037410375103761037710378103791038010381103821038310384103851038610387103881038910390103911039210393103941039510396103971039810399104001040110402104031040410405104061040710408104091041010411104121041310414104151041610417104181041910420104211042210423104241042510426104271042810429104301043110432104331043410435104361043710438104391044010441104421044310444104451044610447104481044910450104511045210453104541045510456104571045810459104601046110462104631046410465104661046710468104691047010471104721047310474104751047610477104781047910480104811048210483104841048510486104871048810489104901049110492104931049410495104961049710498104991050010501105021050310504105051050610507105081050910510105111051210513105141051510516105171051810519105201052110522105231052410525105261052710528105291053010531105321053310534105351053610537105381053910540105411054210543105441054510546105471054810549105501055110552105531055410555105561055710558105591056010561105621056310564105651056610567105681056910570105711057210573105741057510576105771057810579105801058110582105831058410585105861058710588105891059010591105921059310594105951059610597105981059910600106011060210603106041060510606106071060810609106101061110612106131061410615106161061710618106191062010621106221062310624106251062610627106281062910630106311063210633106341063510636106371063810639106401064110642106431064410645106461064710648106491065010651106521065310654106551065610657106581065910660106611066210663106641066510666106671066810669106701067110672106731067410675106761067710678106791068010681106821068310684106851068610687106881068910690106911069210693106941069510696106971069810699107001070110702107031070410705107061070710708107091071010711107121071310714107151071610717107181071910720107211072210723107241072510726107271072810729107301073110732107331073410735107361073710738107391074010741107421074310744107451074610747107481074910750107511075210753107541075510756107571075810759107601076110762107631076410765107661076710768107691077010771107721077310774107751077610777107781077910780107811078210783107841078510786107871078810789107901079110792107931079410795107961079710798107991080010801108021080310804108051080610807108081080910810108111081210813108141081510816108171081810819108201082110822108231082410825108261082710828108291083010831108321083310834108351083610837108381083910840108411084210843108441084510846108471084810849108501085110852108531085410855108561085710858108591086010861108621086310864108651086610867108681086910870108711087210873108741087510876108771087810879108801088110882108831088410885108861088710888108891089010891108921089310894108951089610897108981089910900109011090210903109041090510906109071090810909109101091110912109131091410915109161091710918109191092010921109221092310924109251092610927109281092910930109311093210933109341093510936109371093810939109401094110942109431094410945109461094710948109491095010951109521095310954109551095610957109581095910960109611096210963109641096510966109671096810969109701097110972109731097410975109761097710978109791098010981109821098310984109851098610987109881098910990109911099210993109941099510996109971099810999110001100111002110031100411005110061100711008110091101011011110121101311014110151101611017110181101911020110211102211023110241102511026110271102811029110301103111032110331103411035110361103711038110391104011041110421104311044110451104611047110481104911050110511105211053110541105511056110571105811059110601106111062110631106411065110661106711068110691107011071110721107311074110751107611077110781107911080110811108211083110841108511086110871108811089110901109111092110931109411095110961109711098110991110011101111021110311104111051110611107111081110911110111111111211113111141111511116111171111811119111201112111122111231112411125111261112711128111291113011131111321113311134111351113611137111381113911140111411114211143111441114511146111471114811149111501115111152111531115411155111561115711158111591116011161111621116311164111651116611167111681116911170111711117211173111741117511176111771117811179111801118111182111831118411185111861118711188111891119011191111921119311194111951119611197111981119911200112011120211203112041120511206112071120811209112101121111212112131121411215112161121711218112191122011221112221122311224112251122611227112281122911230112311123211233112341123511236112371123811239112401124111242112431124411245112461124711248112491125011251112521125311254112551125611257112581125911260112611126211263112641126511266112671126811269112701127111272112731127411275112761127711278112791128011281112821128311284112851128611287112881128911290112911129211293112941129511296112971129811299113001130111302113031130411305113061130711308113091131011311113121131311314113151131611317113181131911320113211132211323113241132511326113271132811329113301133111332113331133411335113361133711338113391134011341113421134311344113451134611347113481134911350113511135211353113541135511356113571135811359113601136111362113631136411365113661136711368113691137011371113721137311374113751137611377113781137911380113811138211383113841138511386113871138811389113901139111392113931139411395113961139711398113991140011401114021140311404114051140611407114081140911410114111141211413114141141511416114171141811419114201142111422114231142411425114261142711428114291143011431114321143311434114351143611437114381143911440114411144211443114441144511446114471144811449114501145111452114531145411455114561145711458114591146011461114621146311464114651146611467114681146911470114711147211473114741147511476114771147811479114801148111482114831148411485114861148711488114891149011491114921149311494114951149611497114981149911500115011150211503115041150511506115071150811509115101151111512115131151411515115161151711518115191152011521115221152311524115251152611527115281152911530115311153211533115341153511536115371153811539115401154111542115431154411545115461154711548115491155011551115521155311554115551155611557115581155911560115611156211563115641156511566115671156811569115701157111572115731157411575115761157711578115791158011581115821158311584115851158611587115881158911590115911159211593115941159511596115971159811599116001160111602116031160411605116061160711608116091161011611116121161311614116151161611617116181161911620116211162211623116241162511626116271162811629116301163111632116331163411635116361163711638116391164011641116421164311644116451164611647116481164911650116511165211653116541165511656116571165811659116601166111662116631166411665116661166711668116691167011671116721167311674116751167611677116781167911680116811168211683116841168511686116871168811689116901169111692116931169411695116961169711698116991170011701117021170311704117051170611707117081170911710117111171211713117141171511716117171171811719117201172111722117231172411725117261172711728117291173011731117321173311734117351173611737117381173911740117411174211743117441174511746117471174811749117501175111752117531175411755117561175711758117591176011761117621176311764117651176611767117681176911770117711177211773117741177511776117771177811779117801178111782117831178411785117861178711788117891179011791117921179311794117951179611797117981179911800118011180211803118041180511806118071180811809118101181111812118131181411815118161181711818118191182011821118221182311824118251182611827118281182911830118311183211833118341183511836118371183811839118401184111842118431184411845118461184711848118491185011851118521185311854118551185611857118581185911860118611186211863118641186511866118671186811869118701187111872118731187411875118761187711878118791188011881118821188311884118851188611887118881188911890118911189211893118941189511896118971189811899119001190111902119031190411905119061190711908119091191011911119121191311914119151191611917119181191911920119211192211923119241192511926119271192811929119301193111932119331193411935119361193711938119391194011941119421194311944119451194611947119481194911950119511195211953119541195511956119571195811959119601196111962119631196411965119661196711968119691197011971119721197311974119751197611977119781197911980119811198211983119841198511986119871198811989119901199111992119931199411995119961199711998119991200012001120021200312004120051200612007120081200912010120111201212013120141201512016120171201812019120201202112022120231202412025120261202712028120291203012031120321203312034120351203612037120381203912040120411204212043120441204512046120471204812049120501205112052120531205412055120561205712058120591206012061120621206312064120651206612067120681206912070120711207212073120741207512076120771207812079120801208112082120831208412085120861208712088120891209012091120921209312094120951209612097120981209912100121011210212103121041210512106121071210812109121101211112112121131211412115121161211712118121191212012121121221212312124121251212612127121281212912130121311213212133121341213512136121371213812139121401214112142121431214412145121461214712148121491215012151121521215312154121551215612157121581215912160121611216212163121641216512166121671216812169121701217112172121731217412175121761217712178121791218012181121821218312184121851218612187121881218912190121911219212193121941219512196121971219812199122001220112202122031220412205122061220712208122091221012211122121221312214122151221612217122181221912220122211222212223122241222512226122271222812229122301223112232122331223412235122361223712238122391224012241122421224312244122451224612247122481224912250122511225212253122541225512256122571225812259122601226112262122631226412265122661226712268122691227012271122721227312274122751227612277122781227912280122811228212283122841228512286122871228812289122901229112292122931229412295122961229712298122991230012301123021230312304123051230612307123081230912310123111231212313123141231512316123171231812319123201232112322123231232412325123261232712328123291233012331123321233312334123351233612337123381233912340123411234212343123441234512346123471234812349123501235112352123531235412355123561235712358123591236012361123621236312364123651236612367123681236912370123711237212373123741237512376123771237812379123801238112382123831238412385123861238712388123891239012391123921239312394123951239612397123981239912400124011240212403124041240512406124071240812409124101241112412124131241412415124161241712418124191242012421124221242312424124251242612427124281242912430124311243212433124341243512436124371243812439124401244112442124431244412445124461244712448124491245012451124521245312454124551245612457124581245912460124611246212463124641246512466124671246812469124701247112472124731247412475124761247712478124791248012481124821248312484124851248612487124881248912490124911249212493124941249512496124971249812499125001250112502125031250412505125061250712508125091251012511125121251312514125151251612517125181251912520125211252212523125241252512526125271252812529125301253112532125331253412535125361253712538125391254012541125421254312544125451254612547125481254912550125511255212553125541255512556125571255812559125601256112562125631256412565125661256712568125691257012571125721257312574125751257612577125781257912580125811258212583125841258512586125871258812589125901259112592125931259412595125961259712598125991260012601126021260312604126051260612607126081260912610126111261212613126141261512616126171261812619126201262112622126231262412625126261262712628126291263012631126321263312634126351263612637126381263912640126411264212643126441264512646126471264812649126501265112652126531265412655126561265712658126591266012661126621266312664126651266612667126681266912670126711267212673126741267512676126771267812679126801268112682126831268412685126861268712688126891269012691126921269312694126951269612697126981269912700127011270212703127041270512706127071270812709127101271112712127131271412715127161271712718127191272012721127221272312724127251272612727127281272912730127311273212733127341273512736127371273812739127401274112742127431274412745127461274712748127491275012751127521275312754127551275612757127581275912760127611276212763127641276512766127671276812769127701277112772127731277412775127761277712778127791278012781127821278312784127851278612787127881278912790127911279212793127941279512796127971279812799128001280112802128031280412805128061280712808128091281012811128121281312814128151281612817128181281912820128211282212823128241282512826128271282812829128301283112832128331283412835128361283712838128391284012841128421284312844128451284612847128481284912850128511285212853128541285512856128571285812859128601286112862128631286412865128661286712868128691287012871128721287312874128751287612877128781287912880128811288212883128841288512886128871288812889128901289112892128931289412895128961289712898128991290012901129021290312904129051290612907129081290912910129111291212913129141291512916129171291812919129201292112922129231292412925129261292712928129291293012931129321293312934129351293612937129381293912940129411294212943129441294512946129471294812949129501295112952129531295412955129561295712958129591296012961129621296312964129651296612967129681296912970129711297212973129741297512976129771297812979129801298112982129831298412985129861298712988129891299012991129921299312994129951299612997129981299913000130011300213003130041300513006130071300813009130101301113012130131301413015130161301713018130191302013021130221302313024130251302613027130281302913030130311303213033130341303513036130371303813039130401304113042130431304413045130461304713048130491305013051130521305313054130551305613057130581305913060130611306213063130641306513066130671306813069130701307113072130731307413075130761307713078130791308013081130821308313084130851308613087130881308913090130911309213093130941309513096130971309813099131001310113102131031310413105131061310713108131091311013111131121311313114131151311613117131181311913120131211312213123131241312513126131271312813129131301313113132131331313413135131361313713138131391314013141131421314313144131451314613147131481314913150131511315213153131541315513156131571315813159131601316113162131631316413165131661316713168131691317013171131721317313174131751317613177131781317913180131811318213183131841318513186131871318813189131901319113192131931319413195131961319713198131991320013201132021320313204132051320613207132081320913210132111321213213132141321513216132171321813219132201322113222132231322413225132261322713228132291323013231132321323313234132351323613237132381323913240132411324213243132441324513246132471324813249132501325113252132531325413255132561325713258132591326013261132621326313264132651326613267132681326913270132711327213273132741327513276132771327813279132801328113282132831328413285132861328713288132891329013291132921329313294132951329613297132981329913300133011330213303133041330513306133071330813309133101331113312133131331413315133161331713318133191332013321133221332313324133251332613327133281332913330133311333213333133341333513336133371333813339133401334113342133431334413345133461334713348133491335013351133521335313354133551335613357133581335913360133611336213363133641336513366133671336813369133701337113372133731337413375133761337713378133791338013381133821338313384133851338613387133881338913390133911339213393133941339513396133971339813399134001340113402134031340413405134061340713408134091341013411134121341313414134151341613417134181341913420134211342213423134241342513426134271342813429134301343113432134331343413435134361343713438134391344013441134421344313444134451344613447134481344913450134511345213453134541345513456134571345813459134601346113462134631346413465134661346713468134691347013471134721347313474134751347613477134781347913480134811348213483134841348513486134871348813489134901349113492134931349413495134961349713498134991350013501135021350313504135051350613507135081350913510135111351213513135141351513516135171351813519135201352113522135231352413525135261352713528135291353013531135321353313534135351353613537135381353913540135411354213543135441354513546135471354813549135501355113552135531355413555135561355713558135591356013561135621356313564135651356613567135681356913570135711357213573135741357513576135771357813579135801358113582135831358413585135861358713588135891359013591135921359313594135951359613597135981359913600136011360213603136041360513606136071360813609136101361113612136131361413615136161361713618136191362013621136221362313624136251362613627136281362913630136311363213633136341363513636136371363813639136401364113642136431364413645136461364713648136491365013651136521365313654136551365613657136581365913660136611366213663136641366513666136671366813669136701367113672136731367413675136761367713678136791368013681136821368313684136851368613687136881368913690136911369213693136941369513696136971369813699137001370113702137031370413705137061370713708137091371013711137121371313714137151371613717137181371913720137211372213723137241372513726137271372813729137301373113732137331373413735137361373713738137391374013741137421374313744137451374613747137481374913750137511375213753137541375513756137571375813759137601376113762137631376413765137661376713768137691377013771137721377313774137751377613777137781377913780137811378213783137841378513786137871378813789137901379113792137931379413795137961379713798137991380013801138021380313804138051380613807138081380913810138111381213813138141381513816138171381813819138201382113822138231382413825138261382713828138291383013831138321383313834138351383613837138381383913840138411384213843138441384513846138471384813849138501385113852138531385413855138561385713858138591386013861138621386313864138651386613867138681386913870138711387213873138741387513876138771387813879138801388113882138831388413885138861388713888138891389013891138921389313894138951389613897138981389913900139011390213903139041390513906139071390813909139101391113912139131391413915139161391713918139191392013921139221392313924139251392613927139281392913930139311393213933139341393513936139371393813939139401394113942139431394413945139461394713948139491395013951139521395313954139551395613957139581395913960139611396213963139641396513966139671396813969139701397113972139731397413975139761397713978139791398013981139821398313984139851398613987139881398913990139911399213993139941399513996139971399813999140001400114002140031400414005140061400714008140091401014011140121401314014140151401614017140181401914020140211402214023140241402514026140271402814029140301403114032140331403414035140361403714038140391404014041140421404314044140451404614047140481404914050140511405214053140541405514056140571405814059140601406114062140631406414065140661406714068140691407014071140721407314074140751407614077140781407914080140811408214083140841408514086140871408814089140901409114092140931409414095140961409714098140991410014101141021410314104141051410614107141081410914110141111411214113141141411514116141171411814119141201412114122141231412414125141261412714128141291413014131141321413314134141351413614137141381413914140141411414214143141441414514146141471414814149141501415114152141531415414155141561415714158141591416014161141621416314164141651416614167141681416914170141711417214173141741417514176141771417814179141801418114182141831418414185141861418714188141891419014191141921419314194141951419614197141981419914200142011420214203142041420514206142071420814209142101421114212142131421414215142161421714218142191422014221142221422314224142251422614227142281422914230142311423214233142341423514236142371423814239142401424114242142431424414245142461424714248142491425014251142521425314254142551425614257142581425914260142611426214263142641426514266142671426814269142701427114272142731427414275142761427714278142791428014281142821428314284142851428614287142881428914290142911429214293142941429514296142971429814299143001430114302143031430414305143061430714308143091431014311143121431314314143151431614317143181431914320143211432214323143241432514326143271432814329143301433114332143331433414335143361433714338143391434014341143421434314344143451434614347143481434914350143511435214353143541435514356143571435814359143601436114362143631436414365143661436714368143691437014371143721437314374143751437614377143781437914380143811438214383143841438514386143871438814389143901439114392143931439414395143961439714398143991440014401144021440314404144051440614407144081440914410144111441214413144141441514416144171441814419144201442114422144231442414425144261442714428144291443014431144321443314434144351443614437144381443914440144411444214443144441444514446144471444814449144501445114452144531445414455144561445714458144591446014461144621446314464144651446614467144681446914470144711447214473144741447514476144771447814479144801448114482144831448414485144861448714488144891449014491144921449314494144951449614497144981449914500145011450214503145041450514506145071450814509145101451114512145131451414515145161451714518145191452014521145221452314524145251452614527145281452914530145311453214533145341453514536145371453814539145401454114542145431454414545145461454714548145491455014551145521455314554145551455614557145581455914560145611456214563145641456514566145671456814569145701457114572145731457414575145761457714578145791458014581145821458314584145851458614587145881458914590145911459214593145941459514596145971459814599146001460114602146031460414605146061460714608146091461014611146121461314614146151461614617146181461914620146211462214623146241462514626146271462814629146301463114632146331463414635146361463714638146391464014641146421464314644146451464614647146481464914650146511465214653146541465514656146571465814659146601466114662146631466414665146661466714668146691467014671146721467314674146751467614677146781467914680146811468214683146841468514686146871468814689146901469114692146931469414695146961469714698146991470014701147021470314704147051470614707147081470914710147111471214713147141471514716147171471814719147201472114722147231472414725147261472714728147291473014731147321473314734147351473614737147381473914740147411474214743147441474514746147471474814749147501475114752147531475414755147561475714758147591476014761147621476314764147651476614767147681476914770147711477214773147741477514776147771477814779147801478114782147831478414785147861478714788147891479014791147921479314794147951479614797147981479914800148011480214803148041480514806148071480814809148101481114812148131481414815148161481714818148191482014821148221482314824148251482614827148281482914830148311483214833148341483514836148371483814839148401484114842148431484414845148461484714848148491485014851148521485314854148551485614857148581485914860148611486214863148641486514866148671486814869148701487114872148731487414875148761487714878148791488014881148821488314884148851488614887148881488914890148911489214893148941489514896148971489814899149001490114902149031490414905149061490714908149091491014911149121491314914149151491614917149181491914920149211492214923149241492514926149271492814929149301493114932149331493414935149361493714938149391494014941149421494314944149451494614947149481494914950149511495214953149541495514956149571495814959149601496114962149631496414965149661496714968149691497014971149721497314974149751497614977149781497914980149811498214983149841498514986149871498814989149901499114992149931499414995149961499714998149991500015001150021500315004150051500615007150081500915010150111501215013150141501515016150171501815019150201502115022150231502415025150261502715028150291503015031150321503315034150351503615037150381503915040150411504215043150441504515046150471504815049150501505115052150531505415055150561505715058150591506015061150621506315064150651506615067150681506915070150711507215073150741507515076150771507815079150801508115082150831508415085150861508715088150891509015091150921509315094150951509615097150981509915100151011510215103151041510515106151071510815109151101511115112151131511415115151161511715118151191512015121151221512315124151251512615127151281512915130151311513215133151341513515136151371513815139151401514115142151431514415145151461514715148151491515015151151521515315154151551515615157151581515915160151611516215163151641516515166151671516815169151701517115172151731517415175151761517715178151791518015181151821518315184151851518615187151881518915190151911519215193151941519515196151971519815199152001520115202152031520415205152061520715208152091521015211152121521315214152151521615217152181521915220152211522215223152241522515226152271522815229152301523115232152331523415235152361523715238152391524015241152421524315244152451524615247152481524915250152511525215253152541525515256152571525815259152601526115262152631526415265152661526715268152691527015271152721527315274152751527615277152781527915280152811528215283152841528515286152871528815289152901529115292152931529415295152961529715298152991530015301153021530315304153051530615307153081530915310153111531215313153141531515316153171531815319153201532115322153231532415325153261532715328153291533015331153321533315334153351533615337153381533915340153411534215343153441534515346153471534815349153501535115352153531535415355153561535715358153591536015361153621536315364153651536615367153681536915370153711537215373153741537515376153771537815379153801538115382153831538415385153861538715388153891539015391153921539315394153951539615397153981539915400154011540215403154041540515406154071540815409154101541115412154131541415415154161541715418154191542015421154221542315424154251542615427154281542915430154311543215433154341543515436154371543815439154401544115442154431544415445154461544715448154491545015451154521545315454154551545615457154581545915460154611546215463154641546515466154671546815469154701547115472154731547415475154761547715478154791548015481154821548315484154851548615487154881548915490154911549215493154941549515496154971549815499155001550115502155031550415505155061550715508155091551015511155121551315514155151551615517155181551915520155211552215523155241552515526155271552815529155301553115532155331553415535155361553715538155391554015541155421554315544155451554615547155481554915550155511555215553155541555515556155571555815559155601556115562155631556415565155661556715568155691557015571155721557315574155751557615577155781557915580155811558215583155841558515586155871558815589155901559115592155931559415595155961559715598155991560015601156021560315604156051560615607156081560915610156111561215613156141561515616156171561815619156201562115622156231562415625156261562715628156291563015631156321563315634156351563615637156381563915640156411564215643156441564515646156471564815649156501565115652156531565415655156561565715658156591566015661156621566315664156651566615667156681566915670156711567215673156741567515676156771567815679156801568115682156831568415685156861568715688156891569015691156921569315694156951569615697156981569915700157011570215703157041570515706157071570815709157101571115712157131571415715157161571715718157191572015721157221572315724157251572615727157281572915730157311573215733157341573515736157371573815739157401574115742157431574415745157461574715748157491575015751157521575315754157551575615757157581575915760157611576215763157641576515766157671576815769157701577115772157731577415775157761577715778157791578015781157821578315784157851578615787157881578915790157911579215793157941579515796157971579815799158001580115802158031580415805158061580715808158091581015811158121581315814158151581615817158181581915820158211582215823158241582515826158271582815829158301583115832158331583415835158361583715838158391584015841158421584315844158451584615847158481584915850158511585215853158541585515856158571585815859158601586115862158631586415865158661586715868158691587015871158721587315874158751587615877158781587915880158811588215883158841588515886158871588815889158901589115892158931589415895158961589715898158991590015901159021590315904159051590615907159081590915910159111591215913159141591515916159171591815919159201592115922159231592415925159261592715928159291593015931159321593315934159351593615937159381593915940159411594215943159441594515946159471594815949159501595115952159531595415955159561595715958159591596015961159621596315964159651596615967159681596915970159711597215973159741597515976159771597815979159801598115982159831598415985159861598715988159891599015991159921599315994159951599615997159981599916000160011600216003160041600516006160071600816009160101601116012160131601416015160161601716018160191602016021160221602316024160251602616027160281602916030160311603216033160341603516036160371603816039160401604116042160431604416045160461604716048160491605016051160521605316054160551605616057160581605916060160611606216063160641606516066160671606816069160701607116072160731607416075160761607716078160791608016081160821608316084160851608616087160881608916090160911609216093160941609516096160971609816099161001610116102161031610416105161061610716108161091611016111161121611316114161151611616117161181611916120161211612216123161241612516126161271612816129161301613116132161331613416135161361613716138161391614016141161421614316144161451614616147161481614916150161511615216153161541615516156161571615816159161601616116162161631616416165161661616716168161691617016171161721617316174161751617616177161781617916180161811618216183161841618516186161871618816189161901619116192161931619416195161961619716198161991620016201162021620316204162051620616207162081620916210162111621216213162141621516216162171621816219162201622116222162231622416225162261622716228162291623016231162321623316234162351623616237162381623916240162411624216243162441624516246162471624816249162501625116252162531625416255162561625716258162591626016261162621626316264162651626616267162681626916270162711627216273162741627516276162771627816279162801628116282162831628416285162861628716288162891629016291162921629316294162951629616297162981629916300163011630216303163041630516306163071630816309163101631116312163131631416315163161631716318163191632016321163221632316324163251632616327163281632916330163311633216333163341633516336163371633816339163401634116342163431634416345163461634716348163491635016351163521635316354163551635616357163581635916360163611636216363163641636516366163671636816369163701637116372163731637416375163761637716378163791638016381163821638316384163851638616387163881638916390163911639216393163941639516396163971639816399164001640116402164031640416405164061640716408164091641016411164121641316414164151641616417164181641916420164211642216423164241642516426164271642816429164301643116432164331643416435164361643716438164391644016441164421644316444164451644616447164481644916450164511645216453164541645516456164571645816459164601646116462164631646416465164661646716468164691647016471164721647316474164751647616477164781647916480164811648216483164841648516486164871648816489164901649116492164931649416495164961649716498164991650016501165021650316504165051650616507165081650916510165111651216513165141651516516165171651816519165201652116522165231652416525165261652716528165291653016531165321653316534165351653616537165381653916540165411654216543165441654516546165471654816549165501655116552165531655416555165561655716558165591656016561165621656316564165651656616567165681656916570165711657216573165741657516576165771657816579165801658116582165831658416585165861658716588165891659016591165921659316594165951659616597165981659916600166011660216603166041660516606166071660816609166101661116612166131661416615166161661716618166191662016621166221662316624166251662616627166281662916630166311663216633166341663516636166371663816639166401664116642166431664416645166461664716648166491665016651166521665316654166551665616657166581665916660166611666216663166641666516666166671666816669166701667116672166731667416675166761667716678166791668016681166821668316684166851668616687166881668916690166911669216693166941669516696166971669816699167001670116702167031670416705167061670716708167091671016711167121671316714167151671616717167181671916720167211672216723167241672516726167271672816729167301673116732167331673416735167361673716738167391674016741167421674316744167451674616747167481674916750167511675216753167541675516756167571675816759167601676116762167631676416765167661676716768167691677016771167721677316774167751677616777167781677916780167811678216783167841678516786167871678816789167901679116792167931679416795167961679716798167991680016801168021680316804168051680616807168081680916810168111681216813168141681516816168171681816819168201682116822168231682416825168261682716828168291683016831168321683316834168351683616837168381683916840168411684216843168441684516846168471684816849168501685116852168531685416855168561685716858168591686016861168621686316864168651686616867168681686916870168711687216873168741687516876168771687816879168801688116882168831688416885168861688716888168891689016891168921689316894168951689616897168981689916900169011690216903169041690516906169071690816909169101691116912169131691416915169161691716918169191692016921169221692316924169251692616927169281692916930169311693216933169341693516936169371693816939169401694116942169431694416945169461694716948169491695016951169521695316954169551695616957169581695916960169611696216963169641696516966169671696816969169701697116972169731697416975169761697716978169791698016981169821698316984169851698616987169881698916990169911699216993169941699516996169971699816999170001700117002170031700417005170061700717008170091701017011170121701317014170151701617017170181701917020170211702217023170241702517026170271702817029170301703117032170331703417035170361703717038170391704017041170421704317044170451704617047170481704917050170511705217053170541705517056170571705817059170601706117062170631706417065170661706717068170691707017071170721707317074170751707617077170781707917080170811708217083170841708517086170871708817089170901709117092170931709417095170961709717098170991710017101171021710317104171051710617107171081710917110171111711217113171141711517116171171711817119171201712117122171231712417125171261712717128171291713017131171321713317134171351713617137171381713917140171411714217143171441714517146171471714817149171501715117152171531715417155171561715717158171591716017161171621716317164171651716617167171681716917170171711717217173171741717517176171771717817179171801718117182171831718417185171861718717188171891719017191171921719317194171951719617197171981719917200172011720217203172041720517206172071720817209172101721117212172131721417215172161721717218172191722017221172221722317224172251722617227172281722917230172311723217233172341723517236172371723817239172401724117242172431724417245172461724717248172491725017251172521725317254172551725617257172581725917260172611726217263172641726517266172671726817269172701727117272172731727417275172761727717278172791728017281172821728317284172851728617287172881728917290172911729217293172941729517296172971729817299173001730117302173031730417305173061730717308173091731017311173121731317314173151731617317173181731917320173211732217323173241732517326173271732817329173301733117332173331733417335173361733717338173391734017341173421734317344173451734617347173481734917350173511735217353173541735517356173571735817359173601736117362173631736417365173661736717368173691737017371173721737317374173751737617377173781737917380173811738217383173841738517386173871738817389173901739117392173931739417395173961739717398173991740017401174021740317404174051740617407174081740917410174111741217413174141741517416174171741817419174201742117422174231742417425174261742717428174291743017431174321743317434174351743617437174381743917440174411744217443174441744517446174471744817449174501745117452174531745417455174561745717458174591746017461174621746317464174651746617467174681746917470174711747217473174741747517476174771747817479174801748117482174831748417485174861748717488174891749017491174921749317494174951749617497174981749917500175011750217503175041750517506175071750817509175101751117512175131751417515175161751717518175191752017521175221752317524175251752617527175281752917530175311753217533175341753517536175371753817539175401754117542175431754417545175461754717548175491755017551175521755317554175551755617557175581755917560175611756217563175641756517566175671756817569175701757117572175731757417575175761757717578175791758017581175821758317584175851758617587175881758917590175911759217593175941759517596175971759817599176001760117602176031760417605176061760717608176091761017611176121761317614176151761617617176181761917620176211762217623176241762517626176271762817629176301763117632176331763417635176361763717638176391764017641176421764317644176451764617647176481764917650176511765217653176541765517656176571765817659176601766117662176631766417665176661766717668176691767017671176721767317674176751767617677176781767917680176811768217683176841768517686176871768817689176901769117692176931769417695176961769717698176991770017701177021770317704177051770617707177081770917710177111771217713177141771517716177171771817719177201772117722177231772417725177261772717728177291773017731177321773317734177351773617737177381773917740177411774217743177441774517746177471774817749177501775117752177531775417755177561775717758177591776017761177621776317764177651776617767177681776917770177711777217773177741777517776177771777817779177801778117782177831778417785177861778717788177891779017791177921779317794177951779617797177981779917800178011780217803178041780517806178071780817809178101781117812178131781417815178161781717818178191782017821178221782317824178251782617827178281782917830178311783217833178341783517836178371783817839178401784117842178431784417845178461784717848178491785017851178521785317854178551785617857178581785917860178611786217863178641786517866178671786817869178701787117872178731787417875178761787717878178791788017881178821788317884178851788617887178881788917890178911789217893178941789517896178971789817899179001790117902179031790417905179061790717908179091791017911179121791317914179151791617917179181791917920179211792217923179241792517926179271792817929179301793117932179331793417935179361793717938179391794017941179421794317944179451794617947179481794917950179511795217953179541795517956179571795817959179601796117962179631796417965179661796717968179691797017971179721797317974179751797617977179781797917980179811798217983179841798517986179871798817989179901799117992179931799417995179961799717998179991800018001180021800318004180051800618007180081800918010180111801218013180141801518016180171801818019180201802118022180231802418025180261802718028180291803018031180321803318034180351803618037180381803918040180411804218043180441804518046180471804818049180501805118052180531805418055180561805718058180591806018061180621806318064180651806618067180681806918070180711807218073180741807518076180771807818079180801808118082180831808418085180861808718088180891809018091180921809318094180951809618097180981809918100181011810218103181041810518106181071810818109181101811118112181131811418115181161811718118181191812018121181221812318124181251812618127181281812918130181311813218133181341813518136181371813818139181401814118142181431814418145181461814718148181491815018151181521815318154181551815618157181581815918160181611816218163181641816518166181671816818169181701817118172181731817418175181761817718178181791818018181181821818318184181851818618187181881818918190181911819218193181941819518196181971819818199182001820118202182031820418205182061820718208182091821018211182121821318214182151821618217182181821918220182211822218223182241822518226182271822818229182301823118232182331823418235182361823718238182391824018241182421824318244182451824618247182481824918250182511825218253182541825518256182571825818259182601826118262182631826418265182661826718268182691827018271182721827318274182751827618277182781827918280182811828218283182841828518286182871828818289182901829118292182931829418295182961829718298182991830018301183021830318304183051830618307183081830918310183111831218313183141831518316183171831818319183201832118322183231832418325183261832718328183291833018331183321833318334183351833618337183381833918340183411834218343183441834518346183471834818349183501835118352183531835418355183561835718358183591836018361183621836318364183651836618367183681836918370183711837218373183741837518376183771837818379183801838118382183831838418385183861838718388183891839018391183921839318394183951839618397183981839918400184011840218403184041840518406184071840818409184101841118412184131841418415184161841718418184191842018421184221842318424184251842618427184281842918430184311843218433184341843518436184371843818439184401844118442184431844418445184461844718448184491845018451184521845318454184551845618457184581845918460184611846218463184641846518466184671846818469184701847118472184731847418475184761847718478184791848018481184821848318484184851848618487184881848918490184911849218493184941849518496184971849818499185001850118502185031850418505185061850718508185091851018511185121851318514185151851618517185181851918520185211852218523185241852518526185271852818529185301853118532185331853418535185361853718538185391854018541185421854318544185451854618547185481854918550185511855218553185541855518556185571855818559185601856118562185631856418565185661856718568185691857018571185721857318574185751857618577185781857918580185811858218583185841858518586185871858818589185901859118592185931859418595185961859718598185991860018601186021860318604186051860618607186081860918610186111861218613186141861518616186171861818619186201862118622186231862418625186261862718628186291863018631186321863318634186351863618637186381863918640186411864218643186441864518646186471864818649186501865118652186531865418655186561865718658186591866018661186621866318664186651866618667186681866918670186711867218673186741867518676186771867818679186801868118682186831868418685186861868718688186891869018691186921869318694186951869618697186981869918700187011870218703187041870518706187071870818709187101871118712187131871418715187161871718718187191872018721187221872318724187251872618727187281872918730187311873218733187341873518736187371873818739187401874118742187431874418745187461874718748187491875018751187521875318754187551875618757187581875918760187611876218763187641876518766187671876818769187701877118772187731877418775187761877718778187791878018781187821878318784187851878618787187881878918790187911879218793187941879518796187971879818799188001880118802188031880418805188061880718808188091881018811188121881318814188151881618817188181881918820188211882218823188241882518826188271882818829188301883118832188331883418835188361883718838188391884018841188421884318844188451884618847188481884918850188511885218853188541885518856188571885818859188601886118862188631886418865188661886718868188691887018871188721887318874188751887618877188781887918880188811888218883188841888518886188871888818889188901889118892188931889418895188961889718898188991890018901189021890318904189051890618907189081890918910189111891218913189141891518916189171891818919189201892118922189231892418925189261892718928189291893018931189321893318934189351893618937189381893918940189411894218943189441894518946189471894818949189501895118952189531895418955189561895718958189591896018961189621896318964189651896618967189681896918970189711897218973189741897518976189771897818979189801898118982189831898418985189861898718988189891899018991189921899318994189951899618997189981899919000190011900219003190041900519006190071900819009190101901119012190131901419015190161901719018190191902019021190221902319024190251902619027190281902919030190311903219033190341903519036190371903819039190401904119042190431904419045190461904719048190491905019051190521905319054190551905619057190581905919060190611906219063190641906519066190671906819069190701907119072190731907419075190761907719078190791908019081190821908319084190851908619087190881908919090190911909219093190941909519096190971909819099191001910119102191031910419105191061910719108191091911019111191121911319114191151911619117191181911919120191211912219123191241912519126191271912819129191301913119132191331913419135191361913719138191391914019141191421914319144191451914619147191481914919150191511915219153191541915519156191571915819159191601916119162191631916419165191661916719168191691917019171191721917319174191751917619177191781917919180191811918219183191841918519186191871918819189191901919119192191931919419195191961919719198191991920019201192021920319204192051920619207192081920919210192111921219213192141921519216192171921819219192201922119222192231922419225192261922719228192291923019231192321923319234192351923619237192381923919240192411924219243192441924519246192471924819249192501925119252192531925419255192561925719258192591926019261192621926319264192651926619267192681926919270192711927219273192741927519276192771927819279192801928119282192831928419285192861928719288192891929019291192921929319294192951929619297192981929919300193011930219303193041930519306193071930819309193101931119312193131931419315193161931719318193191932019321193221932319324193251932619327193281932919330193311933219333193341933519336193371933819339193401934119342193431934419345193461934719348193491935019351193521935319354193551935619357193581935919360193611936219363193641936519366193671936819369193701937119372193731937419375193761937719378193791938019381193821938319384193851938619387193881938919390193911939219393193941939519396193971939819399194001940119402194031940419405194061940719408194091941019411194121941319414194151941619417194181941919420194211942219423194241942519426194271942819429194301943119432194331943419435194361943719438194391944019441194421944319444194451944619447194481944919450194511945219453194541945519456194571945819459194601946119462194631946419465194661946719468194691947019471194721947319474194751947619477194781947919480194811948219483194841948519486194871948819489194901949119492194931949419495194961949719498194991950019501195021950319504195051950619507195081950919510195111951219513195141951519516195171951819519195201952119522195231952419525195261952719528195291953019531195321953319534195351953619537195381953919540195411954219543195441954519546195471954819549195501955119552195531955419555195561955719558195591956019561195621956319564195651956619567195681956919570195711957219573195741957519576195771957819579195801958119582195831958419585195861958719588195891959019591195921959319594195951959619597195981959919600196011960219603196041960519606196071960819609196101961119612196131961419615196161961719618196191962019621196221962319624196251962619627196281962919630196311963219633196341963519636196371963819639196401964119642196431964419645196461964719648196491965019651196521965319654196551965619657196581965919660196611966219663196641966519666196671966819669196701967119672196731967419675196761967719678196791968019681196821968319684196851968619687196881968919690196911969219693196941969519696196971969819699197001970119702197031970419705197061970719708197091971019711197121971319714197151971619717197181971919720197211972219723197241972519726197271972819729197301973119732197331973419735197361973719738197391974019741197421974319744197451974619747197481974919750197511975219753197541975519756197571975819759197601976119762197631976419765197661976719768197691977019771197721977319774197751977619777197781977919780197811978219783197841978519786197871978819789197901979119792197931979419795197961979719798197991980019801198021980319804198051980619807198081980919810198111981219813198141981519816198171981819819198201982119822198231982419825198261982719828198291983019831198321983319834198351983619837198381983919840198411984219843198441984519846198471984819849198501985119852198531985419855198561985719858198591986019861198621986319864198651986619867198681986919870198711987219873198741987519876198771987819879198801988119882198831988419885198861988719888198891989019891198921989319894198951989619897198981989919900199011990219903199041990519906199071990819909199101991119912199131991419915199161991719918199191992019921199221992319924199251992619927199281992919930199311993219933199341993519936199371993819939199401994119942199431994419945199461994719948199491995019951199521995319954199551995619957199581995919960199611996219963199641996519966199671996819969199701997119972199731997419975199761997719978199791998019981199821998319984199851998619987199881998919990199911999219993199941999519996199971999819999200002000120002200032000420005200062000720008200092001020011200122001320014200152001620017200182001920020200212002220023200242002520026200272002820029200302003120032200332003420035200362003720038200392004020041200422004320044200452004620047200482004920050200512005220053200542005520056200572005820059200602006120062200632006420065200662006720068200692007020071200722007320074200752007620077200782007920080200812008220083200842008520086200872008820089200902009120092200932009420095200962009720098200992010020101201022010320104201052010620107201082010920110201112011220113201142011520116201172011820119201202012120122201232012420125201262012720128201292013020131201322013320134201352013620137201382013920140201412014220143201442014520146201472014820149201502015120152201532015420155201562015720158201592016020161201622016320164201652016620167201682016920170201712017220173201742017520176201772017820179201802018120182201832018420185201862018720188201892019020191201922019320194201952019620197201982019920200202012020220203202042020520206202072020820209202102021120212202132021420215202162021720218202192022020221202222022320224202252022620227202282022920230202312023220233202342023520236202372023820239202402024120242202432024420245202462024720248202492025020251202522025320254202552025620257202582025920260202612026220263202642026520266202672026820269202702027120272202732027420275202762027720278202792028020281202822028320284202852028620287202882028920290202912029220293202942029520296202972029820299203002030120302203032030420305203062030720308203092031020311203122031320314203152031620317203182031920320203212032220323203242032520326203272032820329203302033120332203332033420335203362033720338203392034020341203422034320344203452034620347203482034920350203512035220353203542035520356203572035820359203602036120362203632036420365203662036720368203692037020371203722037320374203752037620377203782037920380203812038220383203842038520386203872038820389203902039120392203932039420395203962039720398203992040020401204022040320404204052040620407204082040920410204112041220413204142041520416204172041820419204202042120422204232042420425204262042720428204292043020431204322043320434204352043620437204382043920440204412044220443204442044520446204472044820449204502045120452204532045420455204562045720458204592046020461204622046320464204652046620467204682046920470204712047220473204742047520476204772047820479204802048120482204832048420485204862048720488204892049020491204922049320494204952049620497204982049920500205012050220503205042050520506205072050820509205102051120512205132051420515205162051720518205192052020521205222052320524205252052620527205282052920530205312053220533205342053520536205372053820539205402054120542205432054420545205462054720548205492055020551205522055320554205552055620557205582055920560205612056220563205642056520566205672056820569205702057120572205732057420575205762057720578205792058020581205822058320584205852058620587205882058920590205912059220593205942059520596205972059820599206002060120602206032060420605206062060720608206092061020611206122061320614206152061620617206182061920620206212062220623206242062520626206272062820629206302063120632206332063420635206362063720638206392064020641206422064320644206452064620647206482064920650206512065220653206542065520656206572065820659206602066120662206632066420665206662066720668206692067020671206722067320674206752067620677206782067920680206812068220683206842068520686206872068820689206902069120692206932069420695206962069720698206992070020701207022070320704207052070620707207082070920710207112071220713207142071520716207172071820719207202072120722207232072420725207262072720728207292073020731207322073320734207352073620737207382073920740207412074220743207442074520746207472074820749207502075120752207532075420755207562075720758207592076020761207622076320764207652076620767207682076920770207712077220773207742077520776207772077820779207802078120782207832078420785207862078720788207892079020791207922079320794207952079620797207982079920800208012080220803208042080520806208072080820809208102081120812208132081420815208162081720818208192082020821208222082320824208252082620827208282082920830208312083220833208342083520836208372083820839208402084120842208432084420845208462084720848208492085020851208522085320854208552085620857208582085920860208612086220863208642086520866208672086820869208702087120872208732087420875208762087720878208792088020881208822088320884208852088620887208882088920890208912089220893208942089520896208972089820899209002090120902209032090420905209062090720908209092091020911209122091320914209152091620917209182091920920209212092220923209242092520926209272092820929209302093120932209332093420935209362093720938209392094020941209422094320944209452094620947209482094920950209512095220953209542095520956209572095820959209602096120962209632096420965209662096720968209692097020971209722097320974209752097620977209782097920980209812098220983209842098520986209872098820989209902099120992209932099420995209962099720998209992100021001210022100321004210052100621007210082100921010210112101221013210142101521016210172101821019210202102121022210232102421025210262102721028210292103021031210322103321034210352103621037210382103921040210412104221043210442104521046210472104821049210502105121052210532105421055210562105721058210592106021061210622106321064210652106621067210682106921070210712107221073210742107521076210772107821079210802108121082210832108421085210862108721088210892109021091210922109321094210952109621097210982109921100211012110221103211042110521106211072110821109211102111121112211132111421115211162111721118211192112021121211222112321124211252112621127211282112921130211312113221133211342113521136211372113821139211402114121142211432114421145211462114721148211492115021151211522115321154211552115621157211582115921160211612116221163211642116521166211672116821169211702117121172211732117421175211762117721178211792118021181211822118321184211852118621187211882118921190211912119221193211942119521196211972119821199212002120121202212032120421205212062120721208212092121021211212122121321214212152121621217212182121921220212212122221223212242122521226212272122821229212302123121232212332123421235212362123721238212392124021241212422124321244212452124621247212482124921250212512125221253212542125521256212572125821259212602126121262212632126421265212662126721268212692127021271212722127321274212752127621277212782127921280212812128221283212842128521286212872128821289212902129121292212932129421295212962129721298212992130021301213022130321304213052130621307213082130921310213112131221313213142131521316213172131821319213202132121322213232132421325213262132721328213292133021331213322133321334213352133621337213382133921340213412134221343213442134521346213472134821349213502135121352213532135421355213562135721358213592136021361213622136321364213652136621367213682136921370213712137221373213742137521376213772137821379213802138121382213832138421385213862138721388213892139021391213922139321394213952139621397213982139921400214012140221403214042140521406214072140821409214102141121412214132141421415214162141721418214192142021421214222142321424214252142621427214282142921430214312143221433214342143521436214372143821439214402144121442214432144421445214462144721448214492145021451214522145321454214552145621457214582145921460214612146221463214642146521466214672146821469214702147121472214732147421475214762147721478214792148021481214822148321484214852148621487214882148921490214912149221493214942149521496214972149821499215002150121502215032150421505215062150721508215092151021511215122151321514215152151621517215182151921520215212152221523215242152521526215272152821529215302153121532215332153421535215362153721538215392154021541215422154321544215452154621547215482154921550215512155221553215542155521556215572155821559215602156121562215632156421565215662156721568215692157021571215722157321574215752157621577215782157921580215812158221583215842158521586215872158821589215902159121592215932159421595215962159721598215992160021601216022160321604216052160621607216082160921610216112161221613216142161521616216172161821619216202162121622216232162421625216262162721628216292163021631216322163321634216352163621637216382163921640216412164221643216442164521646216472164821649216502165121652216532165421655216562165721658216592166021661216622166321664216652166621667216682166921670216712167221673216742167521676216772167821679216802168121682216832168421685216862168721688216892169021691216922169321694216952169621697216982169921700217012170221703217042170521706217072170821709217102171121712217132171421715217162171721718217192172021721217222172321724217252172621727217282172921730217312173221733217342173521736217372173821739217402174121742217432174421745217462174721748217492175021751217522175321754217552175621757217582175921760217612176221763217642176521766217672176821769217702177121772217732177421775217762177721778217792178021781217822178321784217852178621787217882178921790217912179221793217942179521796217972179821799218002180121802218032180421805218062180721808218092181021811218122181321814218152181621817218182181921820218212182221823218242182521826218272182821829218302183121832218332183421835218362183721838218392184021841218422184321844218452184621847218482184921850218512185221853218542185521856218572185821859218602186121862218632186421865218662186721868218692187021871218722187321874218752187621877218782187921880218812188221883218842188521886218872188821889218902189121892218932189421895218962189721898218992190021901219022190321904219052190621907219082190921910219112191221913219142191521916219172191821919219202192121922219232192421925219262192721928219292193021931219322193321934219352193621937219382193921940219412194221943219442194521946219472194821949219502195121952219532195421955219562195721958219592196021961219622196321964219652196621967219682196921970219712197221973219742197521976219772197821979219802198121982219832198421985219862198721988219892199021991219922199321994219952199621997219982199922000220012200222003220042200522006220072200822009220102201122012220132201422015220162201722018220192202022021220222202322024220252202622027220282202922030220312203222033220342203522036220372203822039220402204122042220432204422045220462204722048220492205022051220522205322054220552205622057220582205922060220612206222063220642206522066220672206822069220702207122072220732207422075220762207722078220792208022081220822208322084220852208622087220882208922090220912209222093220942209522096220972209822099221002210122102221032210422105221062210722108221092211022111221122211322114221152211622117221182211922120221212212222123221242212522126221272212822129221302213122132221332213422135221362213722138221392214022141221422214322144221452214622147221482214922150221512215222153221542215522156221572215822159221602216122162221632216422165221662216722168221692217022171221722217322174221752217622177221782217922180221812218222183221842218522186221872218822189221902219122192221932219422195221962219722198221992220022201222022220322204222052220622207222082220922210222112221222213222142221522216222172221822219222202222122222222232222422225222262222722228222292223022231222322223322234222352223622237222382223922240222412224222243222442224522246222472224822249222502225122252222532225422255222562225722258222592226022261222622226322264222652226622267222682226922270222712227222273222742227522276222772227822279222802228122282222832228422285222862228722288222892229022291222922229322294222952229622297222982229922300223012230222303223042230522306223072230822309223102231122312223132231422315223162231722318223192232022321223222232322324223252232622327223282232922330223312233222333223342233522336223372233822339223402234122342223432234422345223462234722348223492235022351223522235322354223552235622357223582235922360223612236222363223642236522366223672236822369223702237122372223732237422375223762237722378223792238022381223822238322384223852238622387223882238922390223912239222393223942239522396223972239822399224002240122402224032240422405224062240722408224092241022411224122241322414224152241622417224182241922420224212242222423224242242522426224272242822429224302243122432224332243422435224362243722438224392244022441224422244322444224452244622447224482244922450224512245222453224542245522456224572245822459224602246122462224632246422465224662246722468224692247022471224722247322474224752247622477224782247922480224812248222483224842248522486224872248822489224902249122492224932249422495224962249722498224992250022501225022250322504225052250622507225082250922510225112251222513225142251522516225172251822519225202252122522225232252422525225262252722528225292253022531225322253322534225352253622537225382253922540225412254222543225442254522546225472254822549225502255122552225532255422555225562255722558225592256022561225622256322564225652256622567225682256922570225712257222573225742257522576225772257822579225802258122582225832258422585225862258722588225892259022591225922259322594225952259622597225982259922600226012260222603226042260522606226072260822609226102261122612226132261422615226162261722618226192262022621226222262322624226252262622627226282262922630226312263222633226342263522636226372263822639226402264122642226432264422645226462264722648226492265022651226522265322654226552265622657226582265922660226612266222663226642266522666226672266822669226702267122672226732267422675226762267722678226792268022681226822268322684226852268622687226882268922690226912269222693226942269522696226972269822699227002270122702227032270422705227062270722708227092271022711227122271322714227152271622717227182271922720227212272222723227242272522726227272272822729227302273122732227332273422735227362273722738227392274022741227422274322744227452274622747227482274922750227512275222753227542275522756227572275822759227602276122762227632276422765227662276722768227692277022771227722277322774227752277622777227782277922780227812278222783227842278522786227872278822789227902279122792227932279422795227962279722798227992280022801228022280322804228052280622807228082280922810228112281222813228142281522816228172281822819228202282122822228232282422825228262282722828228292283022831228322283322834228352283622837228382283922840228412284222843228442284522846228472284822849228502285122852228532285422855228562285722858228592286022861228622286322864228652286622867228682286922870228712287222873228742287522876228772287822879228802288122882228832288422885228862288722888228892289022891228922289322894228952289622897228982289922900229012290222903229042290522906229072290822909229102291122912229132291422915229162291722918229192292022921229222292322924229252292622927229282292922930229312293222933229342293522936229372293822939229402294122942229432294422945229462294722948229492295022951229522295322954229552295622957229582295922960229612296222963229642296522966229672296822969229702297122972229732297422975229762297722978229792298022981229822298322984229852298622987229882298922990229912299222993229942299522996229972299822999230002300123002230032300423005230062300723008230092301023011230122301323014230152301623017230182301923020230212302223023230242302523026230272302823029230302303123032230332303423035230362303723038230392304023041230422304323044230452304623047230482304923050230512305223053230542305523056230572305823059230602306123062230632306423065230662306723068230692307023071230722307323074230752307623077230782307923080230812308223083230842308523086230872308823089230902309123092230932309423095230962309723098230992310023101231022310323104231052310623107231082310923110231112311223113231142311523116231172311823119231202312123122231232312423125231262312723128231292313023131231322313323134231352313623137231382313923140231412314223143231442314523146231472314823149231502315123152231532315423155231562315723158231592316023161231622316323164231652316623167231682316923170231712317223173231742317523176231772317823179231802318123182231832318423185231862318723188231892319023191231922319323194231952319623197231982319923200232012320223203232042320523206232072320823209232102321123212232132321423215232162321723218232192322023221232222322323224232252322623227232282322923230232312323223233232342323523236232372323823239232402324123242232432324423245232462324723248232492325023251232522325323254232552325623257232582325923260232612326223263232642326523266232672326823269232702327123272232732327423275232762327723278232792328023281232822328323284232852328623287232882328923290232912329223293232942329523296232972329823299233002330123302233032330423305233062330723308233092331023311233122331323314233152331623317233182331923320233212332223323233242332523326233272332823329233302333123332233332333423335233362333723338233392334023341233422334323344233452334623347233482334923350233512335223353233542335523356233572335823359233602336123362233632336423365233662336723368233692337023371233722337323374233752337623377233782337923380233812338223383233842338523386233872338823389233902339123392233932339423395233962339723398233992340023401234022340323404234052340623407234082340923410234112341223413234142341523416234172341823419234202342123422234232342423425234262342723428234292343023431234322343323434234352343623437234382343923440234412344223443234442344523446234472344823449234502345123452234532345423455234562345723458234592346023461234622346323464234652346623467234682346923470234712347223473234742347523476234772347823479234802348123482234832348423485234862348723488234892349023491234922349323494234952349623497234982349923500235012350223503235042350523506235072350823509235102351123512235132351423515235162351723518235192352023521235222352323524235252352623527235282352923530235312353223533235342353523536235372353823539235402354123542235432354423545235462354723548235492355023551235522355323554235552355623557235582355923560235612356223563235642356523566235672356823569235702357123572235732357423575235762357723578235792358023581235822358323584235852358623587235882358923590235912359223593235942359523596235972359823599236002360123602236032360423605236062360723608236092361023611236122361323614236152361623617236182361923620236212362223623236242362523626236272362823629236302363123632236332363423635236362363723638236392364023641236422364323644236452364623647236482364923650236512365223653236542365523656236572365823659236602366123662236632366423665236662366723668236692367023671236722367323674236752367623677236782367923680236812368223683236842368523686236872368823689236902369123692236932369423695236962369723698236992370023701237022370323704237052370623707237082370923710237112371223713237142371523716237172371823719237202372123722237232372423725237262372723728237292373023731237322373323734237352373623737237382373923740237412374223743237442374523746237472374823749237502375123752237532375423755237562375723758237592376023761237622376323764237652376623767237682376923770237712377223773237742377523776237772377823779237802378123782237832378423785237862378723788237892379023791237922379323794237952379623797237982379923800238012380223803238042380523806238072380823809238102381123812238132381423815238162381723818238192382023821238222382323824238252382623827238282382923830238312383223833238342383523836238372383823839238402384123842238432384423845238462384723848238492385023851238522385323854238552385623857238582385923860238612386223863238642386523866238672386823869238702387123872238732387423875238762387723878238792388023881238822388323884238852388623887238882388923890238912389223893238942389523896238972389823899239002390123902239032390423905239062390723908239092391023911239122391323914239152391623917239182391923920239212392223923239242392523926239272392823929239302393123932239332393423935239362393723938239392394023941239422394323944239452394623947239482394923950239512395223953239542395523956239572395823959239602396123962239632396423965239662396723968239692397023971239722397323974239752397623977239782397923980239812398223983239842398523986239872398823989239902399123992239932399423995239962399723998239992400024001240022400324004240052400624007240082400924010240112401224013240142401524016240172401824019240202402124022240232402424025240262402724028240292403024031240322403324034240352403624037240382403924040240412404224043240442404524046240472404824049240502405124052240532405424055240562405724058240592406024061240622406324064240652406624067240682406924070240712407224073240742407524076240772407824079240802408124082240832408424085240862408724088240892409024091240922409324094240952409624097240982409924100241012410224103241042410524106241072410824109241102411124112241132411424115241162411724118241192412024121241222412324124241252412624127241282412924130241312413224133241342413524136241372413824139241402414124142241432414424145241462414724148241492415024151241522415324154241552415624157241582415924160241612416224163241642416524166241672416824169241702417124172241732417424175241762417724178241792418024181241822418324184241852418624187241882418924190241912419224193241942419524196241972419824199242002420124202242032420424205242062420724208242092421024211242122421324214242152421624217242182421924220242212422224223242242422524226242272422824229242302423124232242332423424235242362423724238242392424024241242422424324244242452424624247242482424924250242512425224253242542425524256242572425824259242602426124262242632426424265242662426724268242692427024271242722427324274242752427624277242782427924280242812428224283242842428524286242872428824289242902429124292242932429424295242962429724298242992430024301243022430324304243052430624307243082430924310243112431224313243142431524316243172431824319243202432124322243232432424325243262432724328243292433024331243322433324334243352433624337243382433924340243412434224343243442434524346243472434824349243502435124352243532435424355243562435724358243592436024361243622436324364243652436624367243682436924370243712437224373243742437524376243772437824379243802438124382243832438424385243862438724388243892439024391243922439324394243952439624397243982439924400244012440224403244042440524406244072440824409244102441124412244132441424415244162441724418244192442024421244222442324424244252442624427244282442924430244312443224433244342443524436244372443824439244402444124442244432444424445244462444724448244492445024451244522445324454244552445624457244582445924460244612446224463244642446524466244672446824469244702447124472244732447424475244762447724478244792448024481244822448324484244852448624487244882448924490244912449224493244942449524496244972449824499245002450124502245032450424505245062450724508245092451024511245122451324514245152451624517245182451924520245212452224523245242452524526245272452824529245302453124532245332453424535245362453724538245392454024541245422454324544245452454624547245482454924550245512455224553245542455524556245572455824559245602456124562245632456424565245662456724568245692457024571245722457324574245752457624577245782457924580245812458224583245842458524586245872458824589245902459124592245932459424595245962459724598245992460024601246022460324604246052460624607246082460924610246112461224613246142461524616246172461824619246202462124622246232462424625246262462724628246292463024631246322463324634246352463624637246382463924640246412464224643246442464524646246472464824649246502465124652246532465424655246562465724658246592466024661246622466324664246652466624667246682466924670246712467224673246742467524676246772467824679246802468124682246832468424685246862468724688246892469024691246922469324694246952469624697246982469924700247012470224703247042470524706247072470824709247102471124712247132471424715247162471724718247192472024721247222472324724247252472624727247282472924730247312473224733247342473524736247372473824739247402474124742247432474424745247462474724748247492475024751247522475324754247552475624757247582475924760247612476224763247642476524766247672476824769247702477124772247732477424775247762477724778247792478024781247822478324784247852478624787247882478924790247912479224793247942479524796247972479824799248002480124802248032480424805248062480724808248092481024811248122481324814248152481624817248182481924820248212482224823248242482524826248272482824829248302483124832248332483424835248362483724838248392484024841248422484324844248452484624847248482484924850248512485224853248542485524856248572485824859248602486124862248632486424865248662486724868248692487024871248722487324874248752487624877248782487924880248812488224883248842488524886248872488824889248902489124892248932489424895248962489724898248992490024901249022490324904249052490624907249082490924910249112491224913249142491524916249172491824919249202492124922249232492424925249262492724928249292493024931249322493324934249352493624937249382493924940249412494224943249442494524946249472494824949249502495124952249532495424955249562495724958249592496024961249622496324964249652496624967249682496924970249712497224973249742497524976249772497824979249802498124982249832498424985249862498724988249892499024991249922499324994249952499624997249982499925000250012500225003250042500525006250072500825009250102501125012250132501425015250162501725018250192502025021250222502325024250252502625027250282502925030250312503225033250342503525036250372503825039250402504125042250432504425045250462504725048250492505025051250522505325054250552505625057250582505925060250612506225063250642506525066250672506825069250702507125072250732507425075250762507725078250792508025081250822508325084250852508625087250882508925090250912509225093250942509525096250972509825099251002510125102251032510425105251062510725108251092511025111251122511325114251152511625117251182511925120251212512225123251242512525126251272512825129251302513125132251332513425135251362513725138251392514025141251422514325144251452514625147251482514925150251512515225153251542515525156251572515825159251602516125162251632516425165251662516725168251692517025171251722517325174251752517625177251782517925180251812518225183251842518525186251872518825189251902519125192251932519425195251962519725198251992520025201252022520325204252052520625207252082520925210252112521225213252142521525216252172521825219252202522125222252232522425225 |
- // Copyright (C) 2019-2023 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"
- metadataBasePath = "/api/v2/metadata/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"
- robotsTxtPath = "/robots.txt"
- 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"
- 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"
- 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))
- 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.StatusInternalServerError)
- 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.StatusInternalServerError)
- 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.StatusInternalServerError)
- 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.StatusInternalServerError)
- 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.StatusInternalServerError, "")
- 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 TestBasicGroupHandling(t *testing.T) {
- g := getTestGroup()
- 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))
- 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.StatusInternalServerError)
- 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",
- })
- 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,
- },
- }
- 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)
- user, err = dataprovider.GetUserAfterIDPAuth(defaultUsername, "", common.ProtocolOIDC, nil)
- assert.NoError(t, err)
- assert.Len(t, user.VirtualFolders, 4)
- 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)
- 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/%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},
- "/%username%": {dataprovider.PermDelete},
- }
- group1.UserSettings.Filters.FilePatterns = []sdk.PatternsFilter{
- {
- Path: "/sub2/%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/"+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("/", 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/"+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)
- }
- 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.StatusInternalServerError)
- 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.StatusInternalServerError)
- 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.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.StatusInternalServerError)
- 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")
- }
- 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 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)
- 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 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.StatusInternalServerError)
- 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 TestMetadataAPIMock(t *testing.T) {
- user, _, err := httpdtest.AddUser(getTestUser(), http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPITokenFromTestServer(defaultTokenAuthUser, defaultTokenAuthPass)
- assert.NoError(t, err)
- req, err := http.NewRequest(http.MethodGet, path.Join(metadataBasePath, "/checks"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var resp []any
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- assert.Len(t, resp, 0)
- req, err = http.NewRequest(http.MethodPost, path.Join(metadataBasePath, user.Username, "/check"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusAccepted, rr)
- assert.Eventually(t, func() bool {
- req, err := http.NewRequest(http.MethodGet, path.Join(metadataBasePath, "/checks"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- var resp []any
- err = json.Unmarshal(rr.Body.Bytes(), &resp)
- assert.NoError(t, err)
- return len(resp) == 0
- }, 1000*time.Millisecond, 50*time.Millisecond)
- _, 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(metadataBasePath, user.Username, "/check"), nil)
- assert.NoError(t, err)
- setBearerForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusNotFound, rr)
- }
- 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.StatusInternalServerError)
- 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(), "the following characters are allowed")
- 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(), "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, 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")
- req, err = http.NewRequest(http.MethodPut, adminPath+"/"+admin.Username+"/2fa/disable", 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")
- // 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)
- 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(), "unable to set the new password")
- _, 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(), "unable to set the recovery code as used")
- 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)
- req, err = http.NewRequest(http.MethodGet, webClientSharesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, userWebToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- 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, webGroupsPath+"?qlimit=a", 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)
- req, err = http.NewRequest(http.MethodGet, webAdminEventActionsPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- 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)
- req, err = http.NewRequest(http.MethodGet, webAdminEventRulesPath+"?qlimit=10", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, testServerToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- 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.KeyAlgoDSA},
- },
- 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.KeyAlgoDSA}, configsGet.SFTPD.PublicKeyAlgos)
- assert.Len(t, configsGet.SFTPD.Moduli, 0)
- 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{
- Moduli: []string{"/moduli"},
- },
- }
- 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.Moduli, 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.Moduli = append(configs.SFTPD.Moduli, "/moduli_new")
- 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.Moduli, 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.Moduli, 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.StatusInternalServerError, 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)
- // 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(), "unable to verify form token")
- 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(), "Invalid authentication code")
- 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(), "Invalid credentials")
- 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(), "unable to verify form token")
- 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(), "Invalid credentials")
- 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(), "This recovery code was already used")
- 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(), "Invalid 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)
- // disable TOTP
- 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)
- 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(), "Two factory authentication is not enabled")
- 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(), "Two factory authentication is not enabled")
- _, err = httpdtest.RemoveAdmin(admin, 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(), "Invalid credentials")
- 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(), "Invalid credentials")
- 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)
- 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(), "Two factory authentication is not enabled")
- 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}
- 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)
- }
- // 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: could not parse key")
- user, _, err = httpdtest.GetUserByUsername(user.Username, http.StatusOK)
- assert.NoError(t, err)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientPubKeyChangeDisabled}
- 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}
- 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)
- }
- 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}
- _, _, 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 TestRobotsTxtCheck(t *testing.T) {
- req, _ := http.NewRequest(http.MethodGet, robotsTxtPath, nil)
- rr := executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- assert.Equal(t, "User-agent: *\nDisallow: /", 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(), "your IP address is blocked")
- 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")
- _, 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.CertAlgoDSAv01)
- form.Set("sftp_pub_key_algos", ssh.KeyAlgoDSA)
- form.Set("sftp_moduli", "path 1 , path 2")
- 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(), ssh.CertAlgoDSAv01) // 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.KeyAlgoDSA)
- form.Set("sftp_kex_algos", "diffie-hellman-group18-sha512")
- form.Add("sftp_kex_algos", "diffie-hellman-group16-sha512")
- 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(), "Configurations updated")
- // 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.KeyAlgoDSA)
- assert.Len(t, configs.SFTPD.Moduli, 2)
- assert.Contains(t, configs.SFTPD.Moduli, "path 1")
- assert.Contains(t, configs.SFTPD.Moduli, "path 2")
- assert.Len(t, configs.SFTPD.KexAlgorithms, 1)
- assert.Contains(t, configs.SFTPD.KexAlgorithms, "diffie-hellman-group16-sha512")
- // 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(), "unsupported form action")
- // 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(), "Validation error") // 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(), "Configurations updated")
- // 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.KeyAlgoDSA)
- assert.Len(t, configs.SFTPD.Moduli, 2)
- 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(), "Configurations updated")
- 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(), "Configurations updated")
- 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(), "invalid email address")
- 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(), "Configurations updated")
- // 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.KeyAlgoDSA)
- assert.Len(t, configs.SFTPD.Moduli, 2)
- 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(), "Configurations updated")
- 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.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")
- 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")
- user, _, err := httpdtest.AddUser(u, http.StatusCreated)
- assert.NoError(t, err)
- token, err := getJWTAPIUserTokenFromTestServer(defaultUsername, defaultPassword)
- assert.NoError(t, err)
- testFileName := "test.txt"
- 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)
- 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.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 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 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 contents")
- 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)
- 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")
- 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 remove step")
- _, 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 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(), "Please set a username")
- 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(), "Please set a password")
- 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(), "Passwords mismatch")
- 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, webUsersPath, 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(), "an admin user already exists")
- 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(), "invalid credentials")
- // 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(), "invalid credentials")
- // 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(), "login from IP 127.1.1.1 not allowed")
- 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(), "login from IP 127.0.1.1 not allowed")
- // 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(), "unable to verify form token")
- 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+"?qlimit=aa", 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+"?qlimit=1", nil) //nolint:goconst
- 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)
- 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)
- // 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.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 public key
- form.Set("email", email)
- 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.Equal(t, email, user.Email)
- assert.Equal(t, description, user.Description)
- user.Filters.WebClient = []string{sdk.WebClientAPIKeyAuthChangeDisabled, sdk.WebClientPubKeyChangeDisabled}
- _, _, 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)
- 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.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.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}
- _, _, 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(), "unable to verify form token")
- 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(), "Your profile has been successfully updated")
- 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(), "Your profile has been successfully updated")
- 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(), "unable to verify form token")
- 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(), "the new password must be different from the current one")
- 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+"?qlimit=ab", nil)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webUsersPath+"?qlimit=1", 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(), "View and manage auto blocklist")
- }
- 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(), "unable to verify form token")
- 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(), "invalid default users expiration")
- 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+"?qlimit=ac", nil)
- req.RemoteAddr = defaultRemoteAddr
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, _ = http.NewRequest(http.MethodGet, webAdminsPath+"?qlimit=1", 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(), "unable to verify form token")
- 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(), "unable to get the admin")
- _, 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("group1", group1.Name)
- form.Set("add_as_group_type1", "1")
- form.Set("group2", group2.Name)
- form.Set("add_as_group_type2", "2")
- form.Set("group3", 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(), "You cannot remove these permissions to yourself")
- 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(), "You cannot disable yourself")
- form.Set("status", "1")
- 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(), "You cannot add/change your role")
- }
- 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(), "unable to verify form token")
- 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(), "Your backup was successfully restored")
- 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("sub_perm_path0", "/subdir")
- form.Set("sub_perm_permissions0", "list")
- form.Add("sub_perm_permissions0", "download")
- form.Set("vfolder_path", " /vdir")
- form.Set("vfolder_name", folderName)
- form.Set("vfolder_quota_size", "1024")
- form.Set("vfolder_quota_files", "2")
- form.Set("pattern_path0", "/dir2")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "allowed")
- form.Set("pattern_policy0", "1")
- form.Set("pattern_path1", "/dir1")
- form.Set("patterns1", "*.png")
- form.Set("pattern_type1", "allowed")
- form.Set("pattern_path2", "/dir1")
- form.Set("patterns2", "*.zip")
- form.Set("pattern_type2", "denied")
- form.Set("pattern_path3", "/dir3")
- form.Set("patterns3", "*.rar")
- form.Set("pattern_type3", "denied")
- form.Set("pattern_path4", "/dir2")
- form.Set("patterns4", "*.mkv")
- form.Set("pattern_type4", "denied")
- 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(), "invalid default shares expiration")
- 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(), "invalid max shares expiration")
- 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(), "invalid password expiration")
- 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(), "invalid password strength")
- 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(), "Validation error: invalid TLS username")
- form.Set("tls_username", string(sdk.TLSUsernameNone))
- // invalid upload_bandwidth_source0
- form.Set("bandwidth_limit_sources0", "192.168.1.0/24, 192.168.2.0/25")
- form.Set("upload_bandwidth_source0", "a")
- form.Set("download_bandwidth_source0", "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(), "invalid upload_bandwidth_source")
- // invalid download_bandwidth_source0
- form.Set("upload_bandwidth_source0", "256")
- form.Set("download_bandwidth_source0", "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(), "invalid download_bandwidth_source")
- form.Set("download_bandwidth_source0", "512")
- form.Set("download_bandwidth_source1", "1024")
- form.Set("bandwidth_limit_sources1", "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(), "Validation error: could not parse bandwidth limit source")
- form.Set("bandwidth_limit_sources1", "127.0.0.1/32")
- form.Set("upload_bandwidth_source1", "-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(), "unable to verify form token")
- 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)
- }
- }
- }
- 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", testPubKey)
- 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("sub_perm_path0", "/otherdir")
- form.Set("sub_perm_permissions0", "list")
- form.Add("sub_perm_permissions0", "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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.zip")
- form.Set("pattern_type0", "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(), "unable to verify form token")
- 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)
- 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(), "unable to verify form token")
- 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("tpl_username", user1)
- form.Add("tpl_password", "password1")
- form.Add("tpl_public_keys", " ")
- form.Add("tpl_username", user2)
- form.Add("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(), "Cannot save the defined users")
- 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("allowed_extensions", "/dir1::.jpg,.png")
- form.Set("denied_extensions", "/dir2::.zip")
- 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("tpl_username", "user1")
- form.Set("tpl_password", "password1")
- form.Set("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(), "Error validating user")
- form.Set("tpl_username", " ")
- form.Set("tpl_password", "pwd")
- form.Set("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(), "No valid users defined, unable to complete the requested action")
- form.Set("tpl_username", "user1")
- form.Set("tpl_password", "password1")
- form.Set("tpl_public_keys", " ")
- form.Add("tpl_username", "user2")
- form.Add("tpl_password", "password2")
- form.Add("tpl_public_keys", testPubKey)
- form.Add("tpl_username", "")
- form.Add("tpl_password", "password3")
- form.Add("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)
- user1 := dump.Users[0]
- user2 := dump.Users[1]
- 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, 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", testPubKey)
- form.Add("public_keys", 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.Add("tpl_foldername", folder1)
- form.Add("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(), "Cannot save the defined folders")
- 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.Add("tpl_foldername", "folder1")
- form.Add("tpl_foldername", "folder2")
- form.Add("tpl_foldername", "folder3")
- form.Add("tpl_foldername", "folder1 ")
- form.Add("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(), "unable to verify form token")
- 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(), "Error parsing folders fields")
- 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)
- require.Equal(t, folder1, dump.Folders[0].Name)
- require.Equal(t, "desc folder folder1", dump.Folders[0].Description)
- require.True(t, strings.HasSuffix(dump.Folders[0].MappedPath, "folder1mappedfolder1path"))
- require.Equal(t, folder2, dump.Folders[1].Name)
- require.Equal(t, "desc folder folder2", dump.Folders[1].Description)
- require.True(t, strings.HasSuffix(dump.Folders[1].MappedPath, "folder2mappedfolder2path"))
- require.Equal(t, folder3, dump.Folders[2].Name)
- require.Equal(t, "desc folder folder3", dump.Folders[2].Description)
- require.True(t, strings.HasSuffix(dump.Folders[2].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(), "Error parsing folders fields")
- 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)
- require.Equal(t, folder1, dump.Folders[0].Name)
- require.Equal(t, folder1, dump.Folders[0].FsConfig.S3Config.AccessKey)
- err = dump.Folders[0].FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, fmt.Sprintf("pwd%s", folder1), dump.Folders[0].FsConfig.S3Config.AccessSecret.GetPayload())
- require.Equal(t, path.Join("base", folder1)+"/", dump.Folders[0].FsConfig.S3Config.KeyPrefix)
- require.Equal(t, folder2, dump.Folders[1].Name)
- require.Equal(t, folder2, dump.Folders[1].FsConfig.S3Config.AccessKey)
- err = dump.Folders[1].FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, "pwd"+folder2, dump.Folders[1].FsConfig.S3Config.AccessSecret.GetPayload())
- require.Equal(t, "base/"+folder2+"/", dump.Folders[1].FsConfig.S3Config.KeyPrefix)
- require.Equal(t, folder3, dump.Folders[2].Name)
- require.Equal(t, folder3, dump.Folders[2].FsConfig.S3Config.AccessKey)
- err = dump.Folders[2].FsConfig.S3Config.AccessSecret.Decrypt()
- require.NoError(t, err)
- require.Equal(t, "pwd"+folder3, dump.Folders[2].FsConfig.S3Config.AccessSecret.GetPayload())
- require.Equal(t, "base/"+folder3+"/", dump.Folders[2].FsConfig.S3Config.KeyPrefix)
- form.Set("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(), "No valid folders defined")
- form.Set("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(), "Error validating folder")
- }
- 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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "allowed")
- form.Set("pattern_policy0", "0")
- form.Set("pattern_path1", "/dir2")
- form.Set("patterns1", "*.zip")
- form.Set("pattern_type1", "denied")
- form.Set("pattern_policy1", "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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "allowed")
- form.Set("pattern_path1", "/dir2")
- form.Set("patterns1", "*.zip")
- form.Set("pattern_type1", "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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "allowed")
- form.Set("pattern_path1", "/dir2")
- form.Set("patterns1", "*.zip")
- form.Set("pattern_type1", "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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "allowed")
- form.Set("pattern_path1", "/dir2")
- form.Set("patterns1", "*.zip")
- form.Set("pattern_type1", "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(sftpPrivateKey)
- user.FsConfig.SFTPConfig.KeyPassphrase = kms.NewPlainSecret(testPrivateKeyPwd)
- 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("pattern_path0", "/dir1")
- form.Set("patterns0", "*.jpg,*.png")
- form.Set("pattern_type0", "allowed")
- form.Set("pattern_path1", "/dir2")
- form.Set("patterns1", "*.zip")
- form.Set("pattern_type1", "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(), "invalid action type")
- 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(), "invalid http timeout")
- 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_header_key0", action.Options.HTTPConfig.Headers[0].Key)
- form.Set("http_header_val0", action.Options.HTTPConfig.Headers[0].Value)
- form.Set("http_header_key1", action.Options.HTTPConfig.Headers[0].Key) // ignored
- form.Set("http_query_key0", action.Options.HTTPConfig.QueryParameters[0].Key)
- form.Set("http_query_val0", 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(), "unable to verify form token")
- 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(), "HTTP endpoint is required")
- 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+"?qlimit=aaa", 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_header_key0")
- form.Del("http_header_val0")
- form.Set("http_part_name0", "part1")
- form.Set("http_part_file0", "{{VirtualPath}}")
- form.Set("http_part_headers0", "X-MyHeader: a:b,c")
- form.Set("http_part_body0", "")
- form.Set("http_part_namea", "ignored")
- form.Set("http_part_filea", "{{VirtualPath}}")
- form.Set("http_part_headersa", "X-MyHeader: a:b,c")
- form.Set("http_part_bodya", "")
- form.Set("http_part_name12", "part2")
- form.Set("http_part_headers12", "Content-Type:application/json \r\n")
- form.Set("http_part_body12", "{{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(), "command is required")
- 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(), "invalid command timeout")
- form.Set("cmd_timeout", fmt.Sprintf("%d", action.Options.CmdConfig.Timeout))
- form.Set("cmd_env_key0", action.Options.CmdConfig.EnvVars[0].Key)
- form.Set("cmd_env_val0", 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(), "unable to verify form token")
- 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("folder_retention_path10", "p1")
- form.Set("folder_retention_val10", "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(), "invalid retention for path")
- form.Set("folder_retention_val10", "24")
- form.Set("folder_retention_options10", "1")
- form.Add("folder_retention_options10", "2")
- form.Set("folder_retention_path11", "../p2")
- form.Set("folder_retention_val11", "48")
- form.Set("folder_retention_options11", "1")
- form.Add("folder_retention_options12", "2") // 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)
- assert.True(t, folder.IgnoreUserPermissions)
- case "/p2":
- assert.Equal(t, 48, folder.Retention)
- assert.True(t, folder.DeleteEmptyDirs)
- assert.False(t, folder.IgnoreUserPermissions)
- 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(), "invalid fs action type")
- 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.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(), "invalid password expiration threshold")
- 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.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)
- }
- 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: "g*",
- 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(), "invalid status")
- 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(), "invalid trigger")
- form.Set("trigger", fmt.Sprintf("%d", rule.Trigger))
- form.Set("schedule_hour0", rule.Conditions.Schedules[0].Hours)
- form.Set("schedule_day_of_week0", rule.Conditions.Schedules[0].DayOfWeek)
- form.Set("schedule_day_of_month0", rule.Conditions.Schedules[0].DayOfMonth)
- form.Set("schedule_month0", rule.Conditions.Schedules[0].Month)
- form.Set("name_pattern0", rule.Conditions.Options.Names[0].Pattern)
- form.Set("type_name_pattern0", "inverse")
- form.Set("group_name_pattern0", rule.Conditions.Options.GroupNames[0].Pattern)
- form.Set("type_group_name_pattern0", "inverse")
- form.Set("role_name_pattern0", rule.Conditions.Options.RoleNames[0].Pattern)
- form.Set("type_role_name_pattern0", "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(), "invalid min file size")
- 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(), "invalid max file size")
- form.Set("fs_max_size", "0")
- form.Set("action_name0", action.Name)
- form.Set("action_order0", "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(), "invalid order")
- form.Set("action_order0", "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.StatusForbidden, rr)
- assert.Contains(t, rr.Body.String(), "unable to verify form token")
- 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)
- // 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: "g*",
- 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("fs_path_pattern0", 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(), "unable to verify form token")
- form.Set(csrfFormToken, csrfToken)
- // update with no action defined
- form.Del("action_name0")
- form.Del("action_order0")
- 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(), "at least one action is required")
- // 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(), "invalid trigger")
- 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(), "invalid list type")
- 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(), "unable to verify form token")
- 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(), "invalid mode")
- 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(), "invalid IP")
- 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(), "unable to verify form token")
- 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(), "invalid mode")
- 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(), "invalid list mode")
- _, 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(), "unable to verify form token")
- 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(), "name is mandatory")
- 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)
- // 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(), "invalid URL escape")
- // 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(), "unable to verify form token")
- // 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(), "invalid max sessions")
- 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(), "invalid quota size")
- 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(), "invalid upload bandwidth")
- 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(), "invalid download bandwidth")
- 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(), "invalid upload data transfer")
- 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(), "invalid expires in")
- 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(), "invalid max upload file size")
- 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(), "invalid external auth cache time")
- 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(), "unable to verify form token")
- 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)
- // 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(), "unable to verify form token")
- 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(), "invalid SFTP buffer size")
- 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(), "unable to verify form token")
- 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(), "credentials cannot be empty")
- 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(), "unable to verify form token")
- 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(), "invalid URL escape")
- 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+"?qlimit=a", nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, webToken)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusOK, rr)
- req, err = http.NewRequest(http.MethodGet, webFoldersPath+"?qlimit=1", 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
- 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(), "username is mandatory")
- 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(), "please set a password")
- // no code
- form.Set("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(), "please set a confirmation code")
- // 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(), "Error sending confirmation code via email")
- 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(), "Unable to render password reset template")
- 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)
- _, 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)
- 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.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)
- req, _ = http.NewRequest(http.MethodGet, webFoldersPath, nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- req, _ = http.NewRequest(http.MethodGet, webUsersPath, nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- 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)
- req, _ = http.NewRequest(http.MethodGet, webAdminsPath, nil)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- 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)
- req, err = http.NewRequest(http.MethodGet, webAdminRolesPath, nil)
- assert.NoError(t, err)
- setJWTCookieForReq(req, token)
- rr = executeRequest(req)
- checkResponseCode(t, http.StatusInternalServerError, rr)
- 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(req *http.Request, via []*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)
- }
- }
|