ResourceWatcherFSM.cpp 213 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979
  1. #include "stdafx.h"
  2. #include "ResourceWatcherFSM.h"
  3. #include "DeviceBaseClass.h"
  4. #include <string.h>
  5. #include <string>
  6. #include <assert.h>
  7. #include <algorithm>
  8. #include "fileutil.h"
  9. #include <ctime>
  10. #include "CommEntityUtil.hpp"
  11. #include <stdio.h>
  12. #include <stdlib.h>
  13. #include "ResourceWatcher_UserCode.h"
  14. #ifdef RVC_OS_LINUX
  15. #include <vector>
  16. #include <dirent.h>
  17. #include <sstream>
  18. #include <fstream>
  19. #include <iostream>
  20. #include "publicFunExport.h"
  21. #include "toolkit.h"
  22. #include "iniutil.h"
  23. #include "osutil.h"
  24. #include "SpUtility.h"
  25. #include "RestfulFunc.h"
  26. #else
  27. #include <intrin.h>
  28. #include <ShlObj.h>
  29. #include <ShellAPI.h>
  30. #include <Pdh.h>
  31. #include <io.h>
  32. #include <direct.h>
  33. #include "memutil.h"
  34. #include "array.h"
  35. #include <Windows.h>
  36. #include <wintrust.h>
  37. #include <direct.h>
  38. #include <Mscat.h>
  39. #include <Softpub.h>
  40. #include <wlanapi.h>
  41. #include <Netlistmgr.h> //网络检测添加
  42. #include <psapi.h>
  43. #include <tlhelp32.h>
  44. #include "Mmdeviceapi.h"
  45. #include "PolicyConfig.h"
  46. #include "Propidl.h"
  47. #include "Functiondiscoverykeys_devpkey.h"
  48. #include "NetConfigCommon.h"
  49. #include "NetworkInfo.h"
  50. #include "NetworkProbe.h"
  51. #include "MbnEventImpl.h"
  52. #pragma comment(lib, "pdh.lib")
  53. #pragma comment(lib,"winmm.lib" )
  54. #pragma comment(lib, "shell32.lib")
  55. #pragma comment(lib, "Wintrust.lib")
  56. #pragma comment(lib, "crypt32.lib")
  57. #pragma comment(lib, "wlanapi.lib")
  58. #endif
  59. //硬件信息搜集参数
  60. #define DEFAULT_HARDWARE_CHECK_TIME 60 * 1000 //CPU、内存扫描默认间隔,1分钟
  61. #define TIMER_HARDWARE_CHECK 2
  62. //硬盘扫描
  63. #define DEFAULT_DISK_CHECK_TIME 4 * 60 * 60 * 1000 //硬盘默认4小时扫描一次
  64. #define TIMER_DISK_CHECK 3
  65. //保留版本数目
  66. #define DEFAULT_VERSION_SAVED_COUNT 2
  67. #define MAX_VERSION_PATH 256
  68. const int MAX_DAY_OF_BACKWARD = 365;
  69. const char* DEFAULT_DELETE_FILE_SUFFIX = "*";
  70. LPCTSTR UPLOAD_VIDEO_PATH = "UploadVideo";
  71. CRITICAL_SECTION g_csVideoMoreClear;
  72. const int AD0_DAY_OF_BACKWARD = 7;
  73. const int TIMER_INTERVAL_CLEARVIDEO_ENHANCE = 60 * 1000;
  74. #ifdef RVC_OS_LINUX
  75. #define PROCESS_ITEM 14 //进程CPU信息读取位置参数
  76. const int MAX_MAINLINK_CHECK_TIME = 5000;
  77. const int TIMER_MAINLINK_CHECK = 4;
  78. #else
  79. const int TIMER_ID_CLEARVIDEO_ENHANCE = 1;
  80. const int THOUSAND = 1024;
  81. const int MILLION = 1048576;
  82. #endif // RVC_OS_LINUX
  83. static int hardwareCheckTime = 60 * 1000; //初始化硬件检测间隔
  84. static int hardwareCheckFlag = 1;
  85. static int cpuWarnTime = 0;
  86. static int cpuWarnThreshold = 10;
  87. static int memWarnTime = 0;
  88. static int memWarnThreshold = 10; //初始化CUP、内存告警阈值次数
  89. static int diskCheckTime = 4 * 60 * 60 * 1000; //初始化硬盘检测间隔
  90. static int diskCheckFlag = 1;
  91. #define MAX_VOLUME_COUNT 26
  92. #define BUFSIZE 512
  93. static BOOL sBDiskValStatus[MAX_VOLUME_COUNT + 1] = { FALSE };
  94. #define MAX_SUBFILES_COUNT 1000
  95. #define DIV (1024 * 1024)
  96. #define DAY_DIV (24 * 60 * 60)
  97. #define HOURS_DIV (60 * 60)
  98. #define MINUS_DIV (60)
  99. const char* SystemElapsedQuery = "\\System\\System Up Time";
  100. #ifndef RVC_OS_LINUX
  101. #define ENCODING (X509_ASN_ENCODING | PKCS_7_ASN_ENCODING)
  102. #define SafeDeleteArraySize(pData) { if(pData){delete []pData;pData=NULL;} }
  103. #endif // !RVC_OS_LINUX
  104. template<class T>
  105. class TimerOutHelper : public ITimerListener
  106. {
  107. public:
  108. typedef void (T::* FuncTimer)(void* pUserdata);
  109. TimerOutHelper(T* p, FuncTimer pTimerFunc, void* pData, bool bDeleteSelf = false)
  110. : m_pObject(p), m_pUserData(pData), m_pTimer(pTimerFunc), m_bDeleteSelf(bDeleteSelf)
  111. {
  112. }
  113. virtual void OnTimeout(DWORD dwTimerID)
  114. {
  115. (m_pObject->*m_pTimer)(m_pUserData);
  116. if (m_bDeleteSelf)
  117. delete this;
  118. }
  119. private:
  120. void* m_pUserData;
  121. T* m_pObject;
  122. FuncTimer m_pTimer;
  123. bool m_bDeleteSelf;
  124. };
  125. #ifdef RVC_OS_WIN
  126. ResourceWatcherFSM::ResourceWatcherFSM(void)
  127. :m_IsPadDevice(FALSE),
  128. m_csUploadVideoPath(true),
  129. m_nDayOfBackward(DEFAULT_DAY_OF_BACKWARD),
  130. m_bReadyFlag(false),
  131. m_bVideoClearReady(false),
  132. m_nMinSavedDay(DEFAULT_SAVE_DATE_COUNT),
  133. m_strTerminalNo(true),
  134. m_bFirstRunAfterBoot(FALSE),
  135. m_csDelFileSuffix(true),
  136. m_failDelCnt(0),
  137. m_cpuTop(0)
  138. {
  139. m_iNonSignedTotal = 0;
  140. m_xIdlePre = 0;
  141. m_xKernelPre = 0;
  142. m_xUserPre = 0;
  143. m_bNeedForceDiskCheck = false;
  144. m_DoneDetectAutostart = true;
  145. mUiRequireBytes.QuadPart = 0ui64;
  146. mUiCalibration.QuadPart = 0ui64;
  147. }
  148. #endif // RVC_OS_WIN
  149. ResourceWatcherFSM::~ResourceWatcherFSM(void)
  150. {
  151. }
  152. #ifdef RVC_OS_LINUX
  153. void ResourceWatcherFSM::ClearVideoQueue()
  154. {
  155. while (!m_VideoFiles.empty()) {
  156. TmpFileInfo* pNodeInfo = m_VideoFiles.top();
  157. m_VideoFiles.pop();
  158. if (pNodeInfo != NULL) {
  159. delete pNodeInfo;
  160. }
  161. }
  162. memset(mszMaxSizeFilePath, 0, sizeof(mszMaxSizeFilePath));
  163. mAllVideoFileSizeCal = 0;
  164. mMaxFileSize = 0;
  165. mUiRequireBytes = 0;
  166. VideoDailyRecord.clear();
  167. }
  168. void ResourceWatcherFSM::PrintVideEmurateResult()
  169. {
  170. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  171. ByteSprintf(szResult, (double)mMaxFileSize);
  172. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Max Video file: %s size: %s", mszMaxSizeFilePath, szResult);
  173. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("mAllVideoFileSizeCal : %llu Bytes", mAllVideoFileSizeCal);
  174. ByteSprintf(szResult, (double)mAllVideoFileSizeCal);
  175. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Totally(size: %d) directory size: %s", m_VideoFiles.size(), szResult);
  176. }
  177. #else
  178. void ResourceWatcherFSM::ClearVideoQueue()
  179. {
  180. while (!m_VideoFiles.empty()) {
  181. TmpFileInfo* pNodeInfo = m_VideoFiles.top();
  182. m_VideoFiles.pop();
  183. if (pNodeInfo != NULL) {
  184. delete pNodeInfo;
  185. }
  186. }
  187. memset(mszMaxSizeFilePath, 0, sizeof(mszMaxSizeFilePath));
  188. mAllVideoFileSizeCal.QuadPart = 0ui64;
  189. mMaxFileSize.QuadPart = 0ui64;
  190. mUiRequireBytes.QuadPart = 0ui64;
  191. VideoDailyRecord.clear();
  192. }
  193. #endif // RVC_OS_LINUX
  194. void ResourceWatcherFSM::s0_on_entry()
  195. {
  196. LOG_FUNCTION();
  197. #if defined(RVC_OS_WIN)
  198. this->PostEventFIFO(new RunEvent());
  199. BOOL fInstallBySetup(FALSE);
  200. ErrorCodeEnum rc = DetectVTMInstalledBySetup(fInstallBySetup);
  201. if (Error_Succeed == rc && !fInstallBySetup) {
  202. GetEntityBase()->GetFunction()->PostThreadPoolTask(new CreateLinkTask(this));
  203. }
  204. else {
  205. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("detect install type return %s, setup type: %d", SpStrError(rc), fInstallBySetup);
  206. }
  207. #endif //RVC_OS_WIN
  208. }
  209. unsigned int ResourceWatcherFSM::s0_on_event(FSMEvent* e)
  210. {
  211. LOG_FUNCTION();
  212. unsigned int unRes = 0;
  213. switch (e->iEvt)
  214. {
  215. case USER_EVT_RUN:
  216. {
  217. #ifdef RVC_OS_LINUX
  218. RunTask* videoTask = new RunTask(this);
  219. videoTask->SetFlag(m_bReadyFlag ? 1 : 0);
  220. GetEntityBase()->GetFunction()->PostThreadPoolTask(videoTask);
  221. #else
  222. ClearVideoFilesTask* videoTask = new ClearVideoFilesTask(this);
  223. videoTask->SetFlag(m_bReadyFlag ? 1 : 0);
  224. GetEntityBase()->GetFunction()->PostThreadPoolTask(videoTask);
  225. ClearAd0Task* ad0Task = new ClearAd0Task(this);
  226. GetEntityBase()->GetFunction()->PostThreadPoolTask(ad0Task);
  227. #endif // RVC_OS_LINUX
  228. e->SetHandled();
  229. }
  230. break;
  231. default:
  232. {
  233. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Unexpecetd Event(%d)", e->iEvt);
  234. }
  235. break;
  236. }
  237. return unRes;
  238. }
  239. void ResourceWatcherFSM::s0_on_exit()
  240. {
  241. }
  242. void ResourceWatcherFSM::s1_on_entry()
  243. {
  244. LOG_FUNCTION();
  245. }
  246. unsigned int ResourceWatcherFSM::s1_on_event(FSMEvent* e)
  247. {
  248. LOG_FUNCTION();
  249. unsigned int unRes = 0;
  250. switch (e->iEvt)
  251. {
  252. case USER_EVT_CLEAR_UPLOADEDVIDEOFILES_FINISHED:
  253. {
  254. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("%s::There are %d file(s), successfully delete %d file(s) in path %s", __FUNCTION__,
  255. e->param2, e->param1, (LPCTSTR)m_csUploadVideoPath);
  256. if (m_failDelCnt != 0)
  257. {
  258. CSimpleStringA strMsg = CSimpleStringA::Format("There are %d matched files should be deleted", m_failDelCnt);
  259. LogWarn(Severity_Middle, Error_Unexpect, 0, (LPCTSTR)strMsg);
  260. }
  261. ClearVideoFilesEnhanceTask* task = new ClearVideoFilesEnhanceTask(this);
  262. GetEntityBase()->GetFunction()->PostThreadPoolTask(task);
  263. }
  264. e->SetHandled();
  265. break;
  266. #ifndef RVC_OS_LINUX
  267. case EVT_TIMER:
  268. {
  269. if (e->param1 == TIMER_ID_CLEARVIDEO_ENHANCE)
  270. {
  271. ClearVideoFilesEnhanceTask* task = new ClearVideoFilesEnhanceTask(this);
  272. GetEntityBase()->GetFunction()->PostThreadPoolTask(task);
  273. }
  274. }
  275. e->SetHandled();
  276. break;
  277. #endif // !RVC_OS_LINUX
  278. case USER_EVT_CLEAR_VIDEOFILE_ENHANCE_FINISHED:
  279. e->SetHandled();
  280. break;
  281. case USER_EVT_VERSION_CLEAR_FINISHED:
  282. e->SetHandled();
  283. break;
  284. default:
  285. {
  286. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Unexpecetd Event(%d)", e->iEvt);
  287. }
  288. break;
  289. }
  290. return unRes;
  291. }
  292. void ResourceWatcherFSM::s1_on_exit()
  293. {
  294. }
  295. void ResourceWatcherFSM::s2_on_entry()
  296. {
  297. LOG_FUNCTION();
  298. }
  299. unsigned int ResourceWatcherFSM::s2_on_event(FSMEvent* e)
  300. {
  301. LOG_FUNCTION();
  302. unsigned int unRes = 0;
  303. switch (e->iEvt)
  304. {
  305. case USER_EVT_RUN:
  306. {
  307. e->SetHandled();
  308. }
  309. break;
  310. #ifndef RVC_OS_LINUX
  311. case EVT_TIMER:
  312. {
  313. if (e->param1 == TIMER_ID_CLEARVIDEO_ENHANCE)
  314. {
  315. ClearVideoFilesEnhanceTask* task = new ClearVideoFilesEnhanceTask(this);
  316. GetEntityBase()->GetFunction()->PostThreadPoolTask(task);
  317. }
  318. }
  319. e->SetHandled();
  320. break;
  321. #endif // !RVC_OS_LINUX
  322. case USER_EVT_CLEAR_VIDEOFILE_ENHANCE_FINISHED:
  323. {
  324. }
  325. e->SetHandled();
  326. break;
  327. }
  328. return unRes;
  329. }
  330. void ResourceWatcherFSM::s2_on_exit()
  331. {
  332. }
  333. void ResourceWatcherFSM::s3_on_entry()
  334. {
  335. LOG_FUNCTION();
  336. m_bVideoClearReady = true;
  337. }
  338. unsigned int ResourceWatcherFSM::s3_on_event(FSMEvent* e)
  339. {
  340. unsigned int unRes = 0;
  341. switch (e->iEvt)
  342. {
  343. case USER_EVT_RUN:
  344. {
  345. e->SetHandled();
  346. }
  347. break;
  348. #ifndef RVC_OS_LINUX
  349. case EVT_TIMER:
  350. {
  351. if (e->param1 == TIMER_ID_CLEARVIDEO_ENHANCE)
  352. {
  353. ClearVideoFilesEnhanceTask* task = new ClearVideoFilesEnhanceTask(this);
  354. GetEntityBase()->GetFunction()->PostThreadPoolTask(task);
  355. }
  356. }
  357. e->SetHandled();
  358. break;
  359. #endif // !RVC_OS_LINUX
  360. case USER_EVT_CLEAR_VIDEOFILE_ENHANCE:
  361. {
  362. ClearVideoFilesEnhanceTask* task = new ClearVideoFilesEnhanceTask(this);
  363. GetEntityBase()->GetFunction()->PostThreadPoolTask(task);
  364. }
  365. e->SetHandled();
  366. break;
  367. case USER_EVT_CLEAR_VIDEOFILE_ENHANCE_FINISHED:
  368. {
  369. }
  370. e->SetHandled();
  371. break;
  372. }
  373. return unRes;
  374. }
  375. void ResourceWatcherFSM::s3_on_exit()
  376. {
  377. m_bVideoClearReady = false;
  378. }
  379. #ifdef RVC_OS_LINUX
  380. bool ResourceWatcherFSM::GetMonitorInfo(CSimpleStringA& outInfo)
  381. {
  382. std::string sucContent, failedContent;
  383. CSimpleStringA strCmd("xrandr | grep \" connected\" | sed -e \"s/\([A-Z0-9]\+\) connected.*/\1/\"");
  384. bool ret = SP::Module::Util::ShellExecute(strCmd.GetData(), sucContent, failedContent);
  385. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("{%s}:{%s}{%s}", strCmd.GetData(), sucContent.c_str(), failedContent.c_str());
  386. std::string msg = SP::Utility::ToTrim(sucContent);
  387. SP::Utility::replaceInPlace(msg, "\n", "|");
  388. outInfo = msg.c_str();
  389. return ret;
  390. }
  391. #endif // RVC_OS_LINUX
  392. void ResourceWatcherFSM::readVideoStorePath()
  393. {
  394. CSmartPointer<IConfigInfo> spCtSettingConfig;
  395. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  396. // From CenterSetting --Josephus at 10:02:10 201759
  397. spCtSettingConfig->ReadConfigValueInt(GetEntityBase()->GetEntityName(), "BackwardDays", m_nDayOfBackward);
  398. if (m_nDayOfBackward <= 0) {
  399. m_nDayOfBackward = 90;
  400. }
  401. else
  402. {
  403. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Detect BackwardDays from CentralSetting %d", m_nDayOfBackward);
  404. }
  405. if (m_nDayOfBackward <= 0 || m_nDayOfBackward > MAX_DAY_OF_BACKWARD) {
  406. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Detect BackwardDays(%d) is illegal and Set default %d", m_nDayOfBackward, DEFAULT_DAY_OF_BACKWARD);
  407. m_nDayOfBackward = DEFAULT_DAY_OF_BACKWARD;
  408. }
  409. int minSavedDay = 0;
  410. spCtSettingConfig->ReadConfigValueInt(GetEntityBase()->GetEntityName(), "MinSavedDays", minSavedDay);
  411. if (minSavedDay >= 1 && minSavedDay < DEFAULT_SAVE_DATE_COUNT) {
  412. m_nMinSavedDay = minSavedDay;
  413. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Detect MinSavedDays from CentralSetting %d", m_nMinSavedDay);
  414. }
  415. }
  416. void ResourceWatcherFSM::TriggerAtStatusChanged(bool bMStatus)
  417. {
  418. if (firstMpage)
  419. {
  420. firstMpage = false;
  421. CSmartPointer<IConfigInfo> spRunConfig;
  422. CSmartPointer<IConfigInfo> spCtSettingConfig;
  423. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  424. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  425. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "HardwareCheckTime", hardwareCheckTime);
  426. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "HardwareCheckFlag", hardwareCheckFlag);
  427. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "DiskCheckTime", diskCheckTime);
  428. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "DiskCheckFlag", diskCheckFlag);
  429. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "CpuTooHighPercent", m_cpuHighPercent);
  430. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "CpuWarnThreshold", cpuWarnThreshold);
  431. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "MemoryTooHighPercent", m_memHighPercent);
  432. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "MemWarnThreshold", memWarnThreshold);
  433. spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "HardDiskTooHighPercent", m_diskHighPercent);
  434. spCtSettingConfig->ReadConfigValueInt(m_pEntity->GetEntityName(), "VerifySignature", m_iVerify);
  435. #if defined(RVC_OS_WIN)
  436. if (m_iVerify == 1) {//文件签名校验
  437. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Verify digitsign...");
  438. VerifyDigitSignTask* task = new VerifyDigitSignTask(this);
  439. GetEntityBase()->GetFunction()->PostThreadPoolTask(task);
  440. }
  441. if (IsHasSetAutoStartupByFolder()) {
  442. /** 因为安装时已经设置为开机自启动了,所以要把注册表的开机自启动取消掉(如果有) Gifur@20221017]*/
  443. m_forceStartupWithExeFlag = 3;
  444. }
  445. else {
  446. InitialAutoStartupSetType();
  447. }
  448. #endif //RVC_OS_WIN
  449. if (hardwareCheckTime <= 0) { //非法数值情况下设置为默认一分钟
  450. hardwareCheckTime = DEFAULT_HARDWARE_CHECK_TIME;
  451. }
  452. void* pTmpData = NULL;
  453. ITimerListener* pListener = new TimerOutHelper<ResourceWatcherFSM>(this, &ResourceWatcherFSM::HardwareInfoTimer, pTmpData);
  454. GetEntityBase()->GetFunction()->SetTimer(TIMER_HARDWARE_CHECK, pListener, hardwareCheckTime);
  455. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Set Hardware Timer!");
  456. if (diskCheckFlag > 0) {
  457. GetSystemDiskStatus(); //仅启动时检测一次硬盘状态即可
  458. }
  459. if (diskCheckTime <= 0) {
  460. diskCheckTime = DEFAULT_DISK_CHECK_TIME;
  461. }
  462. void* pTmpData2 = NULL;
  463. ITimerListener* pListener2 = new TimerOutHelper<ResourceWatcherFSM>(this, &ResourceWatcherFSM::DiskCheckTimer, pTmpData2);
  464. GetEntityBase()->GetFunction()->SetTimer(TIMER_DISK_CHECK, pListener2, diskCheckTime);
  465. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Set DISK CHECK Timer!");
  466. DeleteFileTask* deleteFilesTask = new DeleteFileTask(this);
  467. GetEntityBase()->GetFunction()->PostThreadPoolTask(deleteFilesTask);
  468. #if defined(RVC_OS_LINUX)
  469. if (m_bFirstRunAfterBoot) {
  470. UINT sysVer = 0;
  471. UploadSysVersionInfo(sysVer);
  472. ConfirmWindowEffectHasBeenOpen();
  473. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("fetch os version: %u", sysVer);
  474. if (1031 < sysVer) {
  475. ConfirmDDEClipboardDisable();
  476. ConfirmNotificationCenterDisable();
  477. }
  478. else {
  479. RecoverDDEClipboardEnable();
  480. }
  481. GetEntityBase()->GetFunction()->PostThreadPoolTask(new DetectSoftwareInstallStatusTask(this));
  482. GetEntityBase()->GetFunction()->PostThreadPoolTask(new UploadMonitorInfoTask(this));
  483. AlarmSystemBasicInfo();
  484. }
  485. #endif //RVC_OS_LINUX
  486. }
  487. }
  488. ErrorCodeEnum ResourceWatcherFSM::OnInit()
  489. {
  490. firstMpage = true; //是否首次进入首页
  491. ErrorCodeEnum erroCode = Error_Succeed;
  492. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Complied at: %s %s", __DATE__, __TIME__);
  493. #ifdef WIDE_CONDITION
  494. setlocale(LC_ALL, "chs");
  495. #endif
  496. CSmartPointer<IConfigInfo> spRunConfig;
  497. CSmartPointer<IConfigInfo> spCtSettingConfig;
  498. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  499. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  500. erroCode = GetEntityBase()->GetFunction()->GetSystemStaticInfo(m_RvcSysinfo);
  501. m_strTerminalNo = m_RvcSysinfo.strTerminalID;
  502. if (m_RvcSysinfo.strMachineType.Compare("RVC.Pad", true)) {
  503. m_IsPadDevice = FALSE;
  504. }
  505. readVideoStorePath();
  506. erroCode = GetEntityBase()->GetFunction()->GetPath("LocalVideo", m_csUploadVideoPath);
  507. if (m_csUploadVideoPath.GetLength() > 0) {
  508. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)(m_csUploadVideoPath.GetData());
  509. m_bReadyFlag = true;
  510. }
  511. else
  512. {
  513. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetPath returned empty string.");
  514. return Error_NotInit;
  515. }
  516. erroCode = spCtSettingConfig->ReadConfigValueInt("TOPProcess", "cpu", m_cpuTop); //读取需要监控的CPU进程个数
  517. if (erroCode != Error_Succeed) {
  518. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Get CPU TOP NUM failed: %s.", SpStrError(erroCode));
  519. }
  520. if (m_cpuTop == 0) {
  521. m_cpuTop = 10; //From static config
  522. }
  523. else if (m_cpuTop == -1) {
  524. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("No need for TOP PROCESS!");
  525. m_cpuTop = 0;
  526. }
  527. m_bReadyFlag = IsNeedToFirstClear();
  528. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Retrieve uploadedVideo path[%s], backwardDays(%d) done(%d)",
  529. (LPCTSTR)m_csUploadVideoPath, m_nDayOfBackward, !m_bReadyFlag);
  530. m_bFirstRunAfterBoot = DetectIsFirstRunAtBoot();
  531. if (m_bFirstRunAfterBoot) {
  532. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("First time to run after system boot.");
  533. #if defined(RVC_OS_WIN)
  534. m_DoneDetectAutostart = false;
  535. #endif //RVC_OS_WIN
  536. }
  537. InitializeCriticalSection(&g_csVideoMoreClear);
  538. if (m_bFirstRunAfterBoot) {
  539. #if defined(RVC_OS_WIN)
  540. DetectDestopFileAndWarn();
  541. //GetSystemMetrics (SM_CMONITORS) 计数仅显示可见的显示器
  542. const int screenNums = GetSystemMetrics(SM_CMONITORS);
  543. const int aimScreenNums4Stand2S = 2;
  544. const int aimScreenNums4Other = 1;
  545. if (0 == m_RvcSysinfo.strMachineType.Compare("RVC.Stand2S", true) && screenNums != aimScreenNums4Stand2S) {
  546. LogWarn(Severity_Low, Error_Unexpect, LOG_RESOURCEWATCHER_LACK_OF_MONITOR
  547. , CSimpleStringA::Format("{\"subject\":\"screen_count\",\"machine_type\":\"%s\",\"expect\":%d,\"actual\":%d}"
  548. , m_RvcSysinfo.strMachineType.GetData(), aimScreenNums4Stand2S, screenNums));
  549. }
  550. else if (0 != m_RvcSysinfo.strMachineType.Compare("RVC.Stand2S", true) && screenNums != aimScreenNums4Other) {
  551. LogWarn(Severity_Low, Error_Unexpect, LOG_RESOURCEWATCHER_LACK_OF_MONITOR
  552. , CSimpleStringA::Format("{\"subject\":\"screen_count\",\"machine_type\":\"%s\",\"expect\":%d,\"actual\":%d}"
  553. , m_RvcSysinfo.strMachineType.GetData(), aimScreenNums4Other, screenNums));
  554. }
  555. #endif //RVC_OS_WIN
  556. }
  557. spCtSettingConfig->ReadConfigValueInt(m_pEntity->GetEntityName(), "DisplayCnt", m_iNonSignedDisplay);
  558. if (m_iNonSignedDisplay <= 0)
  559. m_iNonSignedDisplay = 10;
  560. spRunConfig->ReadConfigValueInt("WarnRecord", "disk", m_diskLastWarnHour);
  561. return Error_Succeed;
  562. }
  563. ErrorCodeEnum ResourceWatcherFSM::OnExit()
  564. {
  565. #ifdef RVC_OS_LINUX
  566. GetEntityBase()->GetFunction()->KillTimer(TIMER_HARDWARE_CHECK);
  567. #endif // RVC_OS_LINUX
  568. ClearVideoQueue();
  569. DeleteCriticalSection(&g_csVideoMoreClear);
  570. return Error_Succeed;
  571. }
  572. void ResourceWatcherFSM::AfterInit()
  573. {
  574. }
  575. #ifdef RVC_OS_LINUX
  576. void ResourceWatcherFSM::AppendVideoSizeTogether()
  577. {
  578. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Add UploadFile size together...");
  579. unsigned long long uiAdditionBytes;
  580. CalculateWillUploadDirSize(uiAdditionBytes);
  581. mAllVideoFileSizeCal += uiAdditionBytes;
  582. diskInfo.SetUploadVideoDirSize(uiAdditionBytes);
  583. }
  584. void ResourceWatcherFSM::CalculateFreeDiskNeeded()
  585. {
  586. const_map_cu_iter citer = VideoDailyRecord.cbegin();
  587. UINT uCount = 0;
  588. while (citer != VideoDailyRecord.cend()) {
  589. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("date[%d]: %d", (citer->first), citer->second);
  590. uCount += citer->second;
  591. citer++;
  592. }
  593. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("count %d and video file size#%d", uCount, m_VideoFiles.size());
  594. diskInfo.CalAverRequreSpaceEachDay(mAllVideoFileSizeCal,
  595. uCount, VideoDailyRecord.size(),
  596. BASE_AVAILALBE_DATE_COUNT, mUiRequireBytes);
  597. }
  598. void ResourceWatcherFSM::RecordFreeDiskNeededToRunCfg()
  599. {
  600. CSmartPointer<IConfigInfo> spConfigRun;
  601. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  602. spConfigRun->WriteConfigValueHexInt("Video", "RequiredBytes", mUiRequireBytes);
  603. spConfigRun->WriteConfigValue("LastClearTime", "Video", (const char*)CSimpleStringA::Format("0x%08X", (DWORD)CSmallDateTime::GetNow()));
  604. m_bReadyFlag = false;
  605. }
  606. void ResourceWatcherFSM::GetFreeDiskNeededFromRunCfg()
  607. {
  608. CSmartPointer<IConfigInfo> spConfigRun;
  609. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  610. UINT64 value(0);
  611. spConfigRun->ReadConfigValueHexInt("Video", "RequiredBytes", value);
  612. mUiRequireBytes = value;
  613. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Retrieve from runcfg about required bytes: %llu", value);
  614. }
  615. #endif // RVC_OS_LINUX
  616. int ResourceWatcherFSM::CleanVideoFiles(int& newDeletedFilesNum, int initFlag)
  617. {
  618. #ifdef RVC_OS_LINUX
  619. ClearVideoQueue();
  620. newDeletedFilesNum = 0;
  621. CSimpleStringA filePath = m_csUploadVideoPath;
  622. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("(InitFlag:%d) filePath %s", initFlag, (LPCTSTR)filePath);
  623. newDeletedFilesNum = 0;
  624. m_failDelCnt = 0;
  625. const int nRet = RemoveExpireVideoFileAndRecord((LPCTSTR)filePath, newDeletedFilesNum, m_failDelCnt, initFlag);
  626. PrintVideEmurateResult();
  627. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("nRet: %d, new Deleted Files Num:%d, video counts:%d", nRet, newDeletedFilesNum, m_VideoFiles.size());
  628. LOG_ASSERT(nRet - newDeletedFilesNum == m_VideoFiles.size());
  629. if (!!initFlag) {
  630. AppendVideoSizeTogether();
  631. CalculateFreeDiskNeeded();
  632. RecordFreeDiskNeededToRunCfg();
  633. }
  634. else {
  635. GetFreeDiskNeededFromRunCfg();
  636. }
  637. return nRet;
  638. #else
  639. ClearVideoQueue();
  640. CSimpleStringA filePath = m_csUploadVideoPath + "\\*";
  641. newDeletedFilesNum = 0;
  642. m_failDelCnt = 0;
  643. int nRet = ProcessVideoFile((LPCTSTR)filePath, newDeletedFilesNum, m_failDelCnt, initFlag);
  644. PrintResult();
  645. assert(nRet - newDeletedFilesNum == m_VideoFiles.size());
  646. CSmartPointer<IConfigInfo> spConfigRun;
  647. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  648. if (initFlag == 1)
  649. {
  650. ULARGE_INTEGER uiAdditionBytes;
  651. CalCulateUploadFileAddition(uiAdditionBytes);
  652. mAllVideoFileSizeCal.QuadPart += uiAdditionBytes.QuadPart;
  653. diskInfo.SetUploadVideoDirSize(uiAdditionBytes);
  654. const_map_cu_iter citer = VideoDailyRecord.cbegin();
  655. UINT uCount = 0;
  656. while (citer != VideoDailyRecord.cend())
  657. {
  658. uCount += citer->second;
  659. citer++;
  660. }
  661. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("uCount#%d, size#%d", uCount, m_VideoFiles.size());
  662. diskInfo.CalAverRequreSpaceEachDay(mAllVideoFileSizeCal,
  663. uCount,
  664. VideoDailyRecord.size(),
  665. BASE_AVAILALBE_DATE_COUNT,
  666. mUiRequireBytes);
  667. FILETIME ftCurTime;
  668. GetSystemTimeAsFileTime(&ftCurTime);
  669. SYSTEMTIME stUTC, stLocal;
  670. FileTimeToSystemTime(&ftCurTime, &stUTC);
  671. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  672. int nYear = stLocal.wYear;
  673. int nMonth = stLocal.wMonth;
  674. int nDay = stLocal.wDay;
  675. char displayTime[MAX_PATH] = { 0 };
  676. GetTimeFormatStr(displayTime, MAX_PATH, &ftCurTime);
  677. CSimpleStringA csRecord = displayTime;
  678. spConfigRun->WriteConfigValueHexInt("Video", "RequiredBytes", mUiRequireBytes.QuadPart);
  679. spConfigRun->WriteConfigValueInt("Video", "FilesNum", nRet);
  680. spConfigRun->WriteConfigValueInt("Video", "Deleted", newDeletedFilesNum);
  681. spConfigRun->WriteConfigValueInt("Video", "FailedDel", m_failDelCnt);
  682. spConfigRun->WriteConfigValue("Video", "LastTime", csRecord);
  683. spConfigRun->WriteConfigValueInt("Video", "Year", nYear);
  684. spConfigRun->WriteConfigValueInt("Video", "Month", nMonth);
  685. spConfigRun->WriteConfigValueInt("Video", "Day", nDay);
  686. m_bReadyFlag = false;
  687. }
  688. else
  689. {
  690. ErrorCodeEnum erroCode = spConfigRun->ReadConfigValueHexInt("Video", "RequiredBytes",
  691. mUiRequireBytes.QuadPart);
  692. if (erroCode != Error_Succeed)
  693. {
  694. LogWarn(Severity_Middle, Error_Unexpect, 0, "Get RequiredBytes failed !");
  695. mUiRequireBytes.QuadPart = 0ui64;
  696. }
  697. }
  698. return nRet;
  699. #endif // RVC_OS_LINUX
  700. }
  701. #ifdef RVC_OS_LINUX
  702. void ResourceWatcherFSM::CalculateBackTime(time_t* ftResultTime, int days)
  703. {
  704. time_t ftCurTime = time(NULL);//获取当前时间
  705. time_t ftBackTime = ftCurTime - (60 * 60 * 24 * days);
  706. *ftResultTime = ftBackTime;
  707. return;
  708. }
  709. BOOL ResourceWatcherFSM::CalculateWillUploadDirSize(unsigned long long& uiBytesCal)
  710. {
  711. uiBytesCal = 0;
  712. BOOL bRet = FALSE;
  713. CSimpleStringA uploadVideoPath;
  714. CSimpleStringA tmpValue;
  715. ErrorCodeEnum ec = GetEntityBase()->GetFunction()->GetPath(UPLOAD_VIDEO_PATH, uploadVideoPath);
  716. if (uploadVideoPath.GetLength() <= 0) {
  717. ec = Error_Param;
  718. }
  719. if (ec == Error_Succeed) {
  720. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("UploadVideo path: %s", (LPCTSTR)uploadVideoPath);
  721. uiBytesCal = 0;
  722. const int nFileCnt = CalculateAllVideoSize(uploadVideoPath, &uiBytesCal);
  723. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  724. ByteSprintf(szResult, (double)uiBytesCal);
  725. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Result: Files(%d), Size: %s", nFileCnt, szResult);
  726. }
  727. return (ec == Error_Succeed);
  728. }
  729. int ResourceWatcherFSM::RemoveExpireVideoFileAndRecord(LPCTSTR dirName, int& deletedFileCnt, int& deleteFailedCnt, const int nFlag)
  730. {
  731. int fileCnt = 0;
  732. int fileLength = strlen(dirName);
  733. // =>=> /*foundFileData.cFileName[0] != '.'*/ !!! be careful for backdate call for the function.
  734. char searchFilePath[MAX_PATH] = { 0 };
  735. char tempFilePath[MAX_PATH] = { 0 };
  736. char displayTime[MAX_PATH] = { 0 };
  737. time_t tmp_time;
  738. CalculateBackTime(&tmp_time, m_nDayOfBackward);
  739. deletedFileCnt = deleteFailedCnt = 0;
  740. GetTimeFormatStr(displayTime, MAX_PATH, &tmp_time);
  741. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("video backtrace time: %s", displayTime);
  742. struct stat st;
  743. if (stat(dirName, &st) < 0 || !S_ISDIR(st.st_mode)) {
  744. return 0;
  745. }
  746. DIR* d = opendir(dirName);
  747. if (!d) {
  748. return 0;
  749. }
  750. struct dirent* dp = NULL;
  751. while ((dp = readdir(d)) != NULL) {
  752. if ((!strcmp(dp->d_name, ".")) || (!strcmp(dp->d_name, ".."))) {
  753. continue;
  754. }
  755. strcpy(tempFilePath, dirName);
  756. if (strlen(tempFilePath) + strlen(dp->d_name) + 3 >= MAX_PATH) {
  757. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  758. continue;
  759. }
  760. strcat(tempFilePath, "/");
  761. strcat(tempFilePath, dp->d_name);
  762. if (-1 == stat(tempFilePath, &st)) {
  763. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("stat video file: %s error: %d", tempFilePath, errno);
  764. continue;
  765. }
  766. const bool isdir = !!(S_ISDIR(st.st_mode));
  767. if (isdir) {
  768. fileCnt += RemoveExpireVideoFileAndRecord((LPCTSTR)tempFilePath, deletedFileCnt, deleteFailedCnt, nFlag);
  769. }
  770. else {
  771. if (nFlag == 1 && st.st_mtime < tmp_time) {
  772. GetTimeFormatStr(displayTime, MAX_PATH, &st.st_mtime);
  773. if (RemoveFileA(tempFilePath)) {
  774. deletedFileCnt++;
  775. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete matched %s(%s) suc.", dp->d_name, displayTime);
  776. }
  777. else {
  778. deleteFailedCnt++;
  779. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Delete matched %s(%s) failed: %d", dp->d_name, displayTime, errno);
  780. }
  781. }
  782. RecordVideoFileInfomation(tempFilePath, &st);
  783. fileCnt++;
  784. }
  785. }
  786. closedir(d);
  787. return fileCnt;
  788. }
  789. void ResourceWatcherFSM::RecordVideoFileInfomation(LPCTSTR lpFilePath, struct stat* pWfdata)
  790. {
  791. m_VideoFiles.push(new TmpFileInfo(lpFilePath, pWfdata->st_mtime, pWfdata->st_size));
  792. mAllVideoFileSizeCal += pWfdata->st_size;
  793. if (mMaxFileSize < pWfdata->st_size) {
  794. strcpy(mszMaxSizeFilePath, lpFilePath);
  795. mMaxFileSize = pWfdata->st_size;
  796. }
  797. SyncUpdateVideoCreateDaily(&pWfdata->st_mtime, 1);
  798. }
  799. #else
  800. void ResourceWatcherFSM::CalculateBackTime(PFILETIME ftResultTime, int days)
  801. {
  802. //Contains a 64-bit value representing the number of 100-nanosecond intervals since January 1, 1601 (UTC).
  803. FILETIME ftCurTime, ftBackTime;
  804. GetSystemTimeAsFileTime(&ftCurTime);
  805. ULARGE_INTEGER uliCurTime, uliOffset, uliBackTime;
  806. uliCurTime.HighPart = ftCurTime.dwHighDateTime;
  807. uliCurTime.LowPart = ftCurTime.dwLowDateTime;
  808. if (days > 0)
  809. {
  810. uliOffset.QuadPart = UInt32x32To64(days * 24 * 60 * 60, 1e7);
  811. uliBackTime.QuadPart = uliCurTime.QuadPart - uliOffset.QuadPart;
  812. }
  813. else
  814. {
  815. uliOffset.QuadPart = UInt32x32To64(-1 * days * 24 * 60 * 60, 1e7);
  816. uliBackTime.QuadPart = uliCurTime.QuadPart + uliOffset.QuadPart;
  817. }
  818. ftBackTime.dwHighDateTime = uliBackTime.HighPart;
  819. ftBackTime.dwLowDateTime = uliBackTime.LowPart;
  820. *ftResultTime = ftBackTime;
  821. return;
  822. }
  823. int ResourceWatcherFSM::ProcessVideoFile(
  824. LPCTSTR dirName
  825. , int& deletedFileCnt
  826. , int& deleteFailedCnt
  827. , const int nFlag)
  828. {
  829. int fileCnt = 0;
  830. int fileLength = strlen(dirName);
  831. // =>=> /*foundFileData.cFileName[0] != '.'*/ !!! be careful for backdate call for the function.
  832. char searchFilePath[MAX_PATH] = { 0 };
  833. char tempFilePath[MAX_PATH] = { 0 };
  834. char displayTime[MAX_PATH] = { 0 };
  835. WIN32_FIND_DATA foundFileData;
  836. HANDLE hFind;
  837. FILETIME ftRemoveTime;
  838. CalculateBackTime(&ftRemoveTime, m_nDayOfBackward);
  839. GetTimeFormatStr(displayTime, MAX_PATH, &ftRemoveTime);
  840. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("The date backward to delete: %s", displayTime);
  841. hFind = FindFirstFile(dirName, &foundFileData);
  842. if (hFind == INVALID_HANDLE_VALUE)
  843. {
  844. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM) ("FindFirstFile failed (%u)", GetLastError());
  845. return 0;
  846. }
  847. if ((foundFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
  848. {
  849. if (/*foundFileData.cFileName[0] != '.'*/FALSE)
  850. {
  851. strcpy_s(tempFilePath, dirName);
  852. tempFilePath[fileLength - 1] = '\0';
  853. if (strlen(tempFilePath) + strlen(foundFileData.cFileName) + 3 >= MAX_PATH)
  854. {
  855. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  856. }
  857. else
  858. {
  859. strcat_s(tempFilePath, foundFileData.cFileName);
  860. strcat_s(tempFilePath, "\\*");
  861. fileCnt += ProcessVideoFile((LPCTSTR)tempFilePath, deletedFileCnt, deleteFailedCnt, nFlag);
  862. }
  863. }
  864. }
  865. else
  866. {
  867. strcpy_s(searchFilePath, dirName);
  868. searchFilePath[fileLength - 1] = '\0';
  869. strcat_s(searchFilePath, foundFileData.cFileName);
  870. if (nFlag == 1 && CompareVideoFileTime(&ftRemoveTime, &foundFileData.ftLastWriteTime) >= 0)
  871. {
  872. GetTimeFormatStr(displayTime, MAX_PATH, &foundFileData.ftLastWriteTime);
  873. if (DeleteFile(searchFilePath) != 0)
  874. {
  875. deletedFileCnt++;
  876. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete matched %s(%s) suc.", foundFileData.cFileName, displayTime);
  877. }
  878. else
  879. {
  880. deleteFailedCnt++;
  881. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete matched %s(%s) failed(%d).", foundFileData.cFileName, displayTime, GetLastError());
  882. CalCulateFileInfo(searchFilePath, &foundFileData);
  883. }
  884. }
  885. else
  886. {
  887. CalCulateFileInfo(searchFilePath, &foundFileData);
  888. }
  889. fileCnt++;
  890. }
  891. while (FindNextFile(hFind, &foundFileData))
  892. {
  893. if ((foundFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
  894. {
  895. if (/*foundFileData.cFileName[0] != '.'*/FALSE)
  896. {
  897. strcpy_s(tempFilePath, dirName);
  898. tempFilePath[fileLength - 1] = '\0';
  899. strcat_s(tempFilePath, foundFileData.cFileName);
  900. strcat_s(tempFilePath, "\\*");
  901. fileCnt += ProcessVideoFile((LPCTSTR)tempFilePath, deletedFileCnt, deleteFailedCnt, nFlag);
  902. }
  903. }
  904. else
  905. {
  906. strcpy_s(searchFilePath, dirName);
  907. searchFilePath[fileLength - 1] = '\0';
  908. strcat_s(searchFilePath, foundFileData.cFileName);
  909. if (nFlag == 1 && CompareVideoFileTime(&ftRemoveTime, &foundFileData.ftLastWriteTime) >= 0)
  910. {
  911. GetTimeFormatStr(displayTime, MAX_PATH, &foundFileData.ftLastWriteTime);
  912. if (DeleteFile(searchFilePath) != 0)
  913. {
  914. deletedFileCnt++;
  915. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete matched %s(%s) suc.", foundFileData.cFileName, displayTime);
  916. }
  917. else
  918. {
  919. deleteFailedCnt++;
  920. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Delete matched %s(%s) failed(%d).",
  921. foundFileData.cFileName, displayTime, GetLastError());
  922. CalCulateFileInfo(searchFilePath, &foundFileData);
  923. }
  924. }
  925. else
  926. {
  927. CalCulateFileInfo(searchFilePath, &foundFileData);
  928. }
  929. fileCnt++;
  930. }
  931. }
  932. FindClose(hFind);
  933. return fileCnt;
  934. }
  935. #endif // RVC_OS_LINUX
  936. void ResourceWatcherFSM::LinkDetect(int detectType, const char* url, bool& status, unsigned int& delay, int warnFlag)
  937. {
  938. string tUrl = url;
  939. string host = "http://";
  940. int flag = 0;
  941. unsigned int tInterval = 0;
  942. #if defined(RVC_OS_LINUX)
  943. string msg;
  944. if (tUrl.find("http") == string::npos) {
  945. tUrl = host + tUrl;
  946. }
  947. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Check url:[%s].", tUrl.c_str());
  948. timeval begin, end;
  949. gettimeofday(&begin, NULL);
  950. int curFlag = HttpProbe(tUrl, msg, 5);
  951. gettimeofday(&end, NULL);
  952. tInterval = (end.tv_sec - begin.tv_sec) * 1000 + (end.tv_usec - begin.tv_usec) / 1000;
  953. if (curFlag > 0 && curFlag < 400) {
  954. flag = true;
  955. }
  956. else {
  957. flag = false;
  958. }
  959. #else
  960. DWORD start, finish;
  961. if (tUrl.find("http") == string::npos) {
  962. tUrl = host + tUrl;
  963. }
  964. char tmpUrl[MAX_PATH] = { 0 };
  965. strcpy_s(tmpUrl, tUrl.c_str());
  966. start = timeGetTime();
  967. flag = http_probe(tmpUrl, 5000, 0, warnFlag);
  968. finish = timeGetTime();
  969. tInterval = finish - start;
  970. #endif //RVC_OS_LINUX
  971. if (flag == false) {
  972. status = false;
  973. }
  974. else {
  975. status = true;
  976. }
  977. delay = tInterval;
  978. }
  979. ErrorCodeEnum ResourceWatcherFSM::BizLinkDetect(
  980. SpReqAnsContext<ResourceWatcherService_BizLinkDetect_Req, ResourceWatcherService_BizLinkDetect_Ans>::Pointer ctx)
  981. {
  982. LOG_FUNCTION();
  983. ErrorCodeEnum ec = Error_Succeed;
  984. int detectType = ctx->Req.protocol;
  985. CSimpleStringA url = ctx->Req.bizLink;
  986. bool linkStatus = false;
  987. unsigned int delay = 0;
  988. LinkDetect(detectType, url.GetData(), linkStatus, delay);
  989. ctx->Ans.bizLinkStatus = linkStatus;
  990. ctx->Ans.bizLinkDelayMS = delay;
  991. ctx->Ans.intParam = 0;
  992. CSimpleStringA warn;
  993. if (!linkStatus) {
  994. warn = CSimpleStringA::Format("BizLink disconnected[%s].", url.GetData());
  995. }
  996. else
  997. {
  998. warn = CSimpleStringA::Format("BizLink connected[%s], Delay time [%d]ms.", url.GetData(), delay);
  999. }
  1000. LogWarn(Severity_Low, Error_Param, LOG_WARN_BIZLINK_CONNECT_CHECK, warn.GetData());
  1001. if (ctx != NULL) {
  1002. ctx->Answer(ec);
  1003. }
  1004. return ec;
  1005. }
  1006. #if defined(RVC_OS_LINUX)
  1007. int ResourceWatcherFSM::RemoveSpecifiedFile(DealFileType fileType, LPCTSTR szFilePath, LPCTSTR szFileName)
  1008. {
  1009. CSimpleStringA ssFileName = szFileName;
  1010. if (ssFileName.IsNullOrEmpty()) return FALSE;
  1011. if ((fileType == DFT_CenterSetting && ssFileName.IsStartWith("centersetting", true) && !ssFileName.IsEndWith(".ini", true))
  1012. || fileType == DFT_Ad0
  1013. ) {
  1014. if (RemoveFileA(szFilePath)) {
  1015. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("%s delete suc.", szFileName);
  1016. return 1;
  1017. }
  1018. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("%s delete failed %d.", szFileName, GetLastError());
  1019. return -1;
  1020. }
  1021. return 0;
  1022. }
  1023. #else
  1024. int ResourceWatcherFSM::DoSomethingWith(DealFileType fileType, LPCTSTR szFilePath, LPCTSTR szFileName)
  1025. {
  1026. CSimpleStringA ssFileName = szFileName;
  1027. if (ssFileName.GetLength() <= 0)
  1028. return FALSE;
  1029. if ((fileType == DFT_CenterSetting && ssFileName.IsStartWith("centersetting", true) && !ssFileName.IsEndWith(".ini", true))
  1030. ||
  1031. fileType == DFT_Ad0
  1032. ) {
  1033. if (DeleteFileA(szFilePath) != 0) {
  1034. return 1;
  1035. }
  1036. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("%s delete failed %d.", szFileName, GetLastError());
  1037. return -1;
  1038. }
  1039. return 0;
  1040. }
  1041. void ResourceWatcherFSM::CalCulateFileInfo(LPCTSTR lpFilePath, PWIN32_FIND_DATA pWfdata)
  1042. {
  1043. m_VideoFiles.push(new TmpFileInfo(lpFilePath,
  1044. &(pWfdata->ftLastWriteTime),
  1045. pWfdata->nFileSizeHigh, pWfdata->nFileSizeLow));
  1046. ULARGE_INTEGER uiTemp;
  1047. uiTemp.HighPart = pWfdata->nFileSizeHigh;
  1048. uiTemp.LowPart = pWfdata->nFileSizeLow;
  1049. mAllVideoFileSizeCal.QuadPart += uiTemp.QuadPart;
  1050. if (mMaxFileSize.QuadPart < uiTemp.QuadPart) {
  1051. strcpy_s(mszMaxSizeFilePath, lpFilePath);
  1052. mMaxFileSize.QuadPart = uiTemp.QuadPart;
  1053. }
  1054. SYSTEMTIME stUTC, stLocal;
  1055. FileTimeToSystemTime(&(pWfdata->ftLastWriteTime), &stUTC);
  1056. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  1057. UINT uDate = stLocal.wYear * 10000 + stLocal.wMonth * 100 + stLocal.wDay;
  1058. VideoDailyRecord[uDate]++;
  1059. }
  1060. #endif //RVC_OS_LINUX
  1061. int ResourceWatcherFSM::ClearSpecifieFile(
  1062. DealFileType fileType,
  1063. LPCTSTR dirName,
  1064. int& deletedFileCnt,
  1065. int& deleteFailedCnt)
  1066. {
  1067. #if defined(RVC_OS_WIN)
  1068. int nRes;
  1069. int nfileCnt = 0;
  1070. int fileLength = strlen(dirName);
  1071. char searchFilePath[MAX_PATH] = { 0 };
  1072. char tempFilePath[MAX_PATH] = { 0 };
  1073. char displayTime[MAX_PATH] = { 0 };
  1074. WIN32_FIND_DATA fndFileData;
  1075. HANDLE hFind;
  1076. hFind = FindFirstFile(dirName, &fndFileData);
  1077. if (hFind == INVALID_HANDLE_VALUE) {
  1078. return -2;
  1079. }
  1080. if ((fndFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY) {
  1081. if (fndFileData.cFileName[0] != '.') {
  1082. memset(tempFilePath, 0, sizeof(tempFilePath));
  1083. strcpy_s(tempFilePath, dirName);
  1084. tempFilePath[fileLength - 1] = '\0';
  1085. if (strlen(tempFilePath) + strlen(fndFileData.cFileName) + 3 >= MAX_PATH) {
  1086. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  1087. }
  1088. else
  1089. {
  1090. strcat_s(tempFilePath, fndFileData.cFileName);
  1091. strcat_s(tempFilePath, "\\*");
  1092. int nRes = ClearSpecifieFile(fileType, (LPCTSTR)tempFilePath, deletedFileCnt, deleteFailedCnt);
  1093. if (nRes >= 0) nfileCnt += nRes;
  1094. }
  1095. }
  1096. }
  1097. else
  1098. {
  1099. memset((void*)searchFilePath, 0, MAX_PATH);
  1100. strcpy_s(searchFilePath, dirName);
  1101. searchFilePath[fileLength - 1] = '\0';
  1102. strcat_s(searchFilePath, fndFileData.cFileName);
  1103. if (fileType == DFT_Ad0) {
  1104. if (CompareFileTime(&mftAd0RemoveTime, &fndFileData.ftLastWriteTime) >= 0) {
  1105. nRes = DoSomethingWith(fileType, searchFilePath, fndFileData.cFileName);
  1106. }
  1107. else
  1108. {
  1109. nRes = 0;
  1110. }
  1111. }
  1112. else
  1113. {
  1114. nRes = DoSomethingWith(fileType, searchFilePath, fndFileData.cFileName);
  1115. }
  1116. if (nRes == -1) deleteFailedCnt++;
  1117. else if (nRes == 1) deletedFileCnt++;
  1118. nfileCnt++;
  1119. }
  1120. while (FindNextFile(hFind, &fndFileData)) {
  1121. if ((fndFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY) {
  1122. if (fndFileData.cFileName[0] != '.') {
  1123. memset(tempFilePath, 0, sizeof(tempFilePath));
  1124. strcpy_s(tempFilePath, dirName);
  1125. tempFilePath[fileLength - 1] = '\0';
  1126. strcat_s(tempFilePath, fndFileData.cFileName);
  1127. strcat_s(tempFilePath, "\\*");
  1128. int nRes = ClearSpecifieFile(fileType, (LPCTSTR)tempFilePath, deletedFileCnt, deleteFailedCnt);
  1129. if (nRes >= 0) nfileCnt += nRes;
  1130. }
  1131. }
  1132. else
  1133. {
  1134. memset((void*)searchFilePath, 0, MAX_PATH);
  1135. strcpy_s(searchFilePath, dirName);
  1136. searchFilePath[fileLength - 1] = '\0';
  1137. strcat_s(searchFilePath, fndFileData.cFileName);
  1138. if (fileType == DFT_Ad0) {
  1139. if (CompareFileTime(&mftAd0RemoveTime, &fndFileData.ftLastWriteTime) >= 0) {
  1140. nRes = DoSomethingWith(fileType, searchFilePath, fndFileData.cFileName);
  1141. }
  1142. else
  1143. {
  1144. nRes = 0;
  1145. }
  1146. }
  1147. else
  1148. {
  1149. nRes = DoSomethingWith(fileType, searchFilePath, fndFileData.cFileName);
  1150. }
  1151. if (nRes == -1) deleteFailedCnt++;
  1152. else if (nRes == 1) deletedFileCnt++;
  1153. nfileCnt++;
  1154. }
  1155. }
  1156. FindClose(hFind);
  1157. return nfileCnt;
  1158. #else
  1159. int nRes;
  1160. int nfileCnt = 0;
  1161. int fileLength = strlen(dirName);
  1162. char searchFilePath[MAX_PATH] = { 0 };
  1163. char tempFilePath[MAX_PATH] = { 0 };
  1164. char displayTime[MAX_PATH] = { 0 };
  1165. struct stat st;
  1166. if (stat(dirName, &st) < 0) {
  1167. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("stat for %s failed: %d %s", dirName, errno, strerror(errno));
  1168. return -2;
  1169. }
  1170. if (!S_ISDIR(st.st_mode)) {
  1171. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)(" %s is not directory", dirName);
  1172. return -2;
  1173. }
  1174. DIR* d = opendir(dirName);
  1175. if (!d) {
  1176. return -3;
  1177. }
  1178. struct dirent* dp = NULL;
  1179. while ((dp = readdir(d)) != NULL) {
  1180. if ((!strcmp(dp->d_name, ".")) || (!strcmp(dp->d_name, ".."))) {
  1181. continue;
  1182. }
  1183. strcpy(tempFilePath, dirName);
  1184. if (strlen(tempFilePath) + strlen(dp->d_name) + 3 >= MAX_PATH) {
  1185. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  1186. continue;
  1187. }
  1188. strcat(tempFilePath, "/");
  1189. strcat(tempFilePath, dp->d_name);
  1190. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("fileName: %s", dp->d_name);
  1191. struct stat stFile;
  1192. stat(tempFilePath, &stFile);
  1193. const bool isdir = !!(S_ISDIR(stFile.st_mode));
  1194. if (isdir && fileType != DFT_CenterSetting) {
  1195. int nRes = ClearSpecifieFile(fileType, (LPCTSTR)tempFilePath, deletedFileCnt, deleteFailedCnt);
  1196. if (nRes >= 0) nfileCnt += nRes;
  1197. }
  1198. else if (!isdir)
  1199. {
  1200. if (fileType != DFT_Ad0 || (CompareVideoFileTime(&mftAd0RemoveTime, &stFile.st_mtime) >= 0)) {
  1201. nRes = RemoveSpecifiedFile(fileType, tempFilePath, dp->d_name);
  1202. }
  1203. else
  1204. {
  1205. nRes = 0;
  1206. }
  1207. if (nRes == -1) deleteFailedCnt++;
  1208. else if (nRes == 1) deletedFileCnt++;
  1209. nfileCnt++;
  1210. }
  1211. }
  1212. closedir(d);
  1213. return nfileCnt;
  1214. #endif //RVC_OS_WIN
  1215. }
  1216. void ResourceWatcherFSM::ClearMoreVideoFiles()
  1217. {
  1218. #ifdef RVC_OS_LINUX
  1219. EnterCriticalSection(&g_csVideoMoreClear);
  1220. ErrorCodeEnum erroCode = Error_Succeed;
  1221. BOOL bOK = FALSE;
  1222. char szResult3[DEFAULT_OUTPUT_FORMAT_SIZE * 2] = { 0 };
  1223. CSmartPointer<IConfigInfo> spConfigRun;
  1224. ErrorCodeEnum ecRunCfg = GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  1225. if (m_VideoFiles.empty() || mUiRequireBytes == 0) {
  1226. int reseved = 0;
  1227. int nRes = CleanVideoFiles(reseved, IsNeedToFirstClear() ? 1 : 0);
  1228. if (nRes - reseved == 0) {
  1229. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("There are no any video files in facts.");
  1230. //SetSysValAndBroadcast(0, "There are no any video files in facts.");
  1231. LeaveCriticalSection(&g_csVideoMoreClear);
  1232. return;
  1233. }
  1234. }
  1235. int CurRank = 3; //default
  1236. int BfRank = 0;
  1237. if (spConfigRun->ReadConfigValueInt("Video", "RequiredRank", BfRank) != Error_Succeed || BfRank < 0) {
  1238. BfRank = 0;
  1239. }
  1240. if (BfRank > 0) {
  1241. unsigned long long uiTotalBytes;
  1242. unsigned long long uiFreeDiskBytes;
  1243. if (!DiskInfo::GetDiskSpace((LPCTSTR)m_csUploadVideoPath, uiTotalBytes, uiFreeDiskBytes)) {
  1244. LeaveCriticalSection(&g_csVideoMoreClear);
  1245. return;
  1246. }
  1247. //对比上次,剩余的可用磁盘空间更小了。
  1248. if (mUiCalibration >= uiFreeDiskBytes) {
  1249. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Previous Set CurRank from %d to %d.", CurRank, BfRank);
  1250. CurRank = BfRank == 4 ? 1 : BfRank;
  1251. }
  1252. mUiCalibration = uiFreeDiskBytes;
  1253. }
  1254. const int InitialRank = CurRank;
  1255. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  1256. int nEnoughLevel = 0;
  1257. unsigned long long uiFactor;
  1258. char displayTime[MAX_PATH] = { 0 };
  1259. time_t ftSavedTime;
  1260. CalculateBackTime(&ftSavedTime, m_nMinSavedDay);
  1261. GetTimeFormatStr(displayTime, MAX_PATH, &ftSavedTime);
  1262. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("It's SavedTime backtrace: %s", displayTime);
  1263. do {
  1264. uiFactor = DEFAULT_AVAILALBE_DATE_COUNT; // 7天
  1265. if (CurRank == 2) uiFactor = LESS_AVAILALBE_DATE_COUNT; // 3天
  1266. else if (CurRank == 1) uiFactor = LEAST_AVAILALBE_DATE_COUNT; // 1天
  1267. const unsigned long long uiSpaceRequired = mUiRequireBytes * uiFactor;
  1268. ByteSprintf(szResult, (double)uiSpaceRequired);
  1269. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Required free disk size: %s, current rank: %d", szResult, CurRank);
  1270. unsigned long long uiMoreSpaceRealRequired(0);
  1271. unsigned long long uiAdjustment(0);
  1272. if (diskInfo.IsNeedToDoClearJobEx((LPCTSTR)m_csUploadVideoPath, uiSpaceRequired, uiAdjustment, &uiMoreSpaceRealRequired)
  1273. && uiMoreSpaceRealRequired) {
  1274. unsigned long long uiNewFreeBytes(0);
  1275. UINT uDeletedCnt = 0, uFailedDeleteCnt = 0;
  1276. BOOL bTrackBackStop = FALSE;
  1277. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("the more space required: %llu.", uiMoreSpaceRealRequired);
  1278. while (!m_VideoFiles.empty() && uiNewFreeBytes < uiMoreSpaceRealRequired) {
  1279. TmpFileInfo* pCurFileInfo = m_VideoFiles.top();
  1280. if (pCurFileInfo != NULL) {
  1281. GetTimeFormatStr(displayTime, MAX_PATH, &(pCurFileInfo->mftEditTime));
  1282. if (CompareVideoFileTime(&ftSavedTime, &(pCurFileInfo->mftEditTime)) < 0) {
  1283. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("stop at %s: %s", pCurFileInfo->msFileName.c_str(), displayTime);
  1284. bTrackBackStop = TRUE;
  1285. break;
  1286. }
  1287. if (RemoveFileA(pCurFileInfo->msFileName.c_str()) != 0) {
  1288. uiNewFreeBytes += pCurFileInfo->ftSize;
  1289. uDeletedCnt++;
  1290. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete suc: %s: %s", pCurFileInfo->msFileName.c_str(), displayTime);
  1291. SyncUpdateVideoCreateDaily(&(pCurFileInfo->mftEditTime), -1);
  1292. }
  1293. else {
  1294. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("[DelFail] File(%s) %d", pCurFileInfo->msFileName.c_str(), errno);
  1295. uFailedDeleteCnt++;
  1296. }
  1297. }
  1298. m_VideoFiles.pop();
  1299. if (pCurFileInfo != NULL) delete pCurFileInfo;
  1300. }
  1301. char szOutput[DEFAULT_OUTPUT_FORMAT_SIZE] = { 0 };
  1302. char szOutput2[DEFAULT_OUTPUT_FORMAT_SIZE] = { 0 };
  1303. ByteSprintf(szOutput, (double)uiNewFreeBytes);
  1304. ByteSprintf(szOutput2, (double)uiMoreSpaceRealRequired);
  1305. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Result(Rank#%d): Succeed deleted<%d>, Failed deleted<%d>, Deleted Size<%s> Real Required Size<%s>",
  1306. CurRank, uDeletedCnt, uFailedDeleteCnt, szOutput, szOutput2);
  1307. if (uiNewFreeBytes >= uiMoreSpaceRealRequired) {
  1308. break;
  1309. }
  1310. else {
  1311. --CurRank;
  1312. }
  1313. #ifdef WITH_DEBUG
  1314. UINT uCount = 0;
  1315. for (const_map_cu_iter citer = VideoDailyRecord.cbegin(); citer != VideoDailyRecord.cend(); citer++) { uCount += citer->second; }
  1316. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("FileCount(%d)(%d)", uCount, m_VideoFiles.size());
  1317. #endif
  1318. }
  1319. else if (InitialRank > CurRank) {
  1320. break;
  1321. }
  1322. else {
  1323. LeaveCriticalSection(&g_csVideoMoreClear);
  1324. return;
  1325. }
  1326. } while (CurRank > 0);
  1327. if (CurRank < 3) {
  1328. unsigned long long uiTotalBytes;
  1329. unsigned long long uiFreeDiskBytes;
  1330. if (DiskInfo::GetDiskSpace((LPCTSTR)m_csUploadVideoPath, uiTotalBytes, uiFreeDiskBytes)) {
  1331. char szResult2[DEFAULT_OUTPUT_FORMAT_SIZE];
  1332. ByteSprintf(szResult, (double)uiTotalBytes);
  1333. ByteSprintf(szResult2, (double)uiFreeDiskBytes);
  1334. double dFreeRate = (uiFreeDiskBytes * kConversion[3]) / (uiTotalBytes * kConversion[3]);
  1335. sprintf(szResult3, "[Disk# %s] total: %s, free: %s, rate: %.1lf%%.",
  1336. m_csUploadVideoPath.GetData(), szResult, szResult2, dFreeRate * 100);
  1337. mUiCalibration = uiFreeDiskBytes;
  1338. }
  1339. else {
  1340. strcpy(szResult3, "It's not enough disk space for recording video !");
  1341. }
  1342. }
  1343. else {
  1344. strcpy(szResult3, "Disk space is enough for recording.");
  1345. }
  1346. if (CurRank == 0) { CurRank = 4; }
  1347. if (BfRank != CurRank) {
  1348. erroCode = spConfigRun->WriteConfigValueInt("Video", "RequiredRank", CurRank);
  1349. }
  1350. SetSysValAndBroadcast(CurRank, szResult3);
  1351. LeaveCriticalSection(&g_csVideoMoreClear);
  1352. return;
  1353. #else
  1354. EnterCriticalSection(&g_csVideoMoreClear);
  1355. ErrorCodeEnum erroCode = Error_Succeed;
  1356. BOOL bOK = FALSE;
  1357. CSmartPointer<IConfigInfo> spConfigRun;
  1358. ErrorCodeEnum ecRunCfg = GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  1359. if (m_VideoFiles.empty())
  1360. {
  1361. int reseved = 0;
  1362. IsNeedToFirstClear();
  1363. int nRes = CleanVideoFiles(reseved, m_bReadyFlag ? 1 : 0);
  1364. if (nRes - reseved == 0)
  1365. {
  1366. SetSysValAndBroadcast(0, "There are no any video files in facts.");
  1367. LeaveCriticalSection(&g_csVideoMoreClear);
  1368. return;
  1369. }
  1370. }
  1371. if (mUiRequireBytes.QuadPart == 0ui64)
  1372. {//更新记录运行时,并等待下一次的自检执行。
  1373. int reseved = 0;
  1374. CleanVideoFiles(reseved);
  1375. LeaveCriticalSection(&g_csVideoMoreClear);
  1376. return;
  1377. }
  1378. ULARGE_INTEGER uiSpaceRequired;
  1379. ULARGE_INTEGER uiSpaceRealRequired;
  1380. ULARGE_INTEGER uiFactor;
  1381. int CurRank = 3;
  1382. int BfRank = 0;
  1383. if (spConfigRun->ReadConfigValueInt("Video", "RequiredRank", BfRank) != Error_Succeed || BfRank < 0)
  1384. BfRank = 0;
  1385. if (BfRank > 0)
  1386. {
  1387. ULARGE_INTEGER uiTotalBytes;
  1388. ULARGE_INTEGER uiFreeDiskBytes;
  1389. if (DiskInfo::GetDiskSpace((LPCTSTR)m_csUploadVideoPath, uiTotalBytes, uiFreeDiskBytes))
  1390. {
  1391. //对比上次,剩余的可用磁盘空间更小了。
  1392. if (mUiCalibration.QuadPart >= uiFreeDiskBytes.QuadPart)
  1393. {
  1394. CurRank = BfRank;
  1395. }
  1396. mUiCalibration.QuadPart = uiFreeDiskBytes.QuadPart;
  1397. }
  1398. }
  1399. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  1400. int nEnoughLevel = 0;
  1401. do
  1402. {
  1403. uiFactor.QuadPart = 0;
  1404. switch (CurRank)
  1405. {
  1406. case 3:
  1407. uiSpaceRequired.QuadPart = mUiRequireBytes.QuadPart * DEFAULT_AVAILALBE_DATE_COUNT;
  1408. break;
  1409. case 2:
  1410. uiSpaceRequired.QuadPart = mUiRequireBytes.QuadPart * LESS_AVAILALBE_DATE_COUNT;
  1411. break;
  1412. case 1:
  1413. uiSpaceRequired.QuadPart = mUiRequireBytes.QuadPart * LEAST_AVAILALBE_DATE_COUNT;
  1414. break;
  1415. default:
  1416. CurRank = 3;
  1417. uiSpaceRequired.QuadPart = mUiRequireBytes.QuadPart * DEFAULT_AVAILALBE_DATE_COUNT;
  1418. }
  1419. ByteSprintf(szResult, (double)uiSpaceRequired.QuadPart);
  1420. if (bOK = diskInfo.IsNeedToDoClearJobEx((LPCTSTR)m_csUploadVideoPath,
  1421. uiSpaceRequired, uiFactor, &uiSpaceRealRequired))
  1422. {
  1423. if (uiSpaceRealRequired.QuadPart == 0)
  1424. {
  1425. break;
  1426. }
  1427. char displayTime[MAX_PATH] = { 0 };
  1428. FILETIME ftSavedTime;
  1429. //TODO: 暂时设置为至少保存两星期的录像文件,如果一天都不保留,那么 m_VideoFiles 要做即时更新 。2017-5-10
  1430. CalculateBackTime(&ftSavedTime, m_nMinSavedDay);
  1431. GetTimeFormatStr(displayTime, MAX_PATH, &ftSavedTime);
  1432. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("SavedTime backtrace: %s", displayTime);
  1433. ULARGE_INTEGER uiFreeBytes;
  1434. uiFreeBytes.QuadPart = 0;
  1435. UINT uDeletedCnt = 0, uFailedDeleteCnt = 0;
  1436. BOOL bTrackBackStop = FALSE;
  1437. while (!m_VideoFiles.empty() && uiFreeBytes.QuadPart < uiSpaceRealRequired.QuadPart)
  1438. {
  1439. TmpFileInfo* pCurFileInfo = m_VideoFiles.top();
  1440. if (pCurFileInfo != NULL)
  1441. {
  1442. SYSTEMTIME stUTC, stLocal;
  1443. FileTimeToSystemTime(&(pCurFileInfo->mftEditTime), &stUTC);
  1444. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  1445. if (CompareVideoFileTime(&ftSavedTime, &(pCurFileInfo->mftEditTime)) < 0)
  1446. {
  1447. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("yet %s: %02d/%02d/%d %02d:%02d", pCurFileInfo->msFileName.c_str(),
  1448. stLocal.wMonth, stLocal.wDay, stLocal.wYear,
  1449. stLocal.wHour, stLocal.wMinute);
  1450. bTrackBackStop = TRUE;
  1451. break;
  1452. }
  1453. if (DeleteFile(pCurFileInfo->msFileName.c_str()) != 0)
  1454. {
  1455. ULARGE_INTEGER tmpBytes;
  1456. tmpBytes.HighPart = pCurFileInfo->mDwFileSizeHeight;
  1457. tmpBytes.LowPart = pCurFileInfo->mDwFileSizeLow;
  1458. uiFreeBytes.QuadPart += tmpBytes.QuadPart;
  1459. uDeletedCnt++;
  1460. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete suc: %s: %02d/%02d/%d %02d:%02d", pCurFileInfo->msFileName.c_str(),
  1461. stLocal.wMonth, stLocal.wDay, stLocal.wYear,
  1462. stLocal.wHour, stLocal.wMinute);
  1463. UINT uDate = stLocal.wYear * 10000 + stLocal.wMonth * 100 + stLocal.wDay;
  1464. VideoDailyRecord[uDate]--;
  1465. }
  1466. else
  1467. {
  1468. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("[DelFail] File(%s) %u", pCurFileInfo->msFileName.c_str(), GetLastError());
  1469. uFailedDeleteCnt++;
  1470. }
  1471. }
  1472. m_VideoFiles.pop();
  1473. if (pCurFileInfo != NULL) delete pCurFileInfo;
  1474. }
  1475. char szOutput[DEFAULT_OUTPUT_FORMAT_SIZE] = { 0 };
  1476. char szOutput2[DEFAULT_OUTPUT_FORMAT_SIZE] = { 0 };
  1477. ByteSprintf(szOutput, (double)uiFreeBytes.QuadPart);
  1478. ByteSprintf(szOutput2, (double)uiSpaceRealRequired.QuadPart);
  1479. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Result(Rank#%d): Succeed deleted<%d>, Failed deleted<%d>, Deleted Size<%s> Real Required Size<%s>",
  1480. CurRank, uDeletedCnt, uFailedDeleteCnt, szOutput, szOutput2);
  1481. if (uiFreeBytes.QuadPart >= uiSpaceRealRequired.QuadPart)
  1482. {
  1483. break;
  1484. }
  1485. --CurRank;
  1486. #if 1
  1487. UINT uCount = 0;
  1488. for (const_map_cu_iter citer = VideoDailyRecord.cbegin();
  1489. citer != VideoDailyRecord.cend(); citer++)
  1490. {
  1491. uCount += citer->second;
  1492. }
  1493. #endif
  1494. }
  1495. else
  1496. {
  1497. LogWarn(Severity_Middle, Error_Unexpect, 0, "IsNeedToDoClearJobEx invoked failed !");
  1498. LeaveCriticalSection(&g_csVideoMoreClear);
  1499. return;
  1500. }
  1501. } while (CurRank > 0);
  1502. char szResult3[DEFAULT_OUTPUT_FORMAT_SIZE * 2] = { 0 };
  1503. if (CurRank < 3)
  1504. {
  1505. ULARGE_INTEGER uiTotalBytes;
  1506. ULARGE_INTEGER uiFreeDiskBytes;
  1507. if (DiskInfo::GetDiskSpace((LPCTSTR)m_csUploadVideoPath, uiTotalBytes, uiFreeDiskBytes))
  1508. {
  1509. char szResult2[DEFAULT_OUTPUT_FORMAT_SIZE];
  1510. ByteSprintf(szResult, (double)uiTotalBytes.QuadPart);
  1511. ByteSprintf(szResult2, (double)uiFreeDiskBytes.QuadPart);
  1512. double dFreeRate = (uiFreeDiskBytes.QuadPart * kConversion[3]) / (uiTotalBytes.QuadPart * kConversion[3]);
  1513. sprintf_s(szResult3, "[Disk# %c:\\] total: %s, free: %s, rate: %.1lf%%.",
  1514. m_csUploadVideoPath[0], szResult, szResult2, dFreeRate * 100);
  1515. mUiCalibration.QuadPart = uiFreeDiskBytes.QuadPart;
  1516. }
  1517. else
  1518. {
  1519. strcpy_s(szResult3, "It's not enough disk space for recording video !");
  1520. }
  1521. if (CurRank == 0)
  1522. {
  1523. CurRank = 4;
  1524. }
  1525. }
  1526. else
  1527. {
  1528. strcpy_s(szResult3, "Disk space is enough for recording.");
  1529. }
  1530. if (BfRank != CurRank)
  1531. {
  1532. erroCode = spConfigRun->WriteConfigValueInt("Video", "RequiredRank", CurRank);
  1533. }
  1534. SetSysValAndBroadcast(CurRank, szResult3);
  1535. LeaveCriticalSection(&g_csVideoMoreClear);
  1536. return;
  1537. #endif // RVC_OS_LINUX
  1538. }
  1539. #ifdef RVC_OS_LINUX
  1540. BOOL ResourceWatcherFSM::RemoveDuplicateCenterSettingFiles()
  1541. {
  1542. LOG_FUNCTION();
  1543. ErrorCodeEnum erroCode = Error_Succeed;
  1544. CSmartPointer<IConfigInfo> spConfigRun;
  1545. erroCode = GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  1546. if (erroCode != Error_Succeed) {
  1547. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("OpenConfig of 'Config_Run' failed 0x%x(%d).", erroCode, erroCode);
  1548. return FALSE;
  1549. }
  1550. int nLastTaskTime(0);
  1551. spConfigRun->ReadConfigValueInt("LastClearTime", "CenterSetting", nLastTaskTime);
  1552. if (IsTodayDone(nLastTaskTime)) {
  1553. return TRUE;
  1554. }
  1555. CSimpleStringA ssFilePath(true);
  1556. erroCode = GetEntityBase()->GetFunction()->GetPath("cfg", ssFilePath);
  1557. if (erroCode != Error_Succeed || ssFilePath.IsNullOrEmpty()) {
  1558. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Get path of 'cfg' failed 0x%x(%d).", erroCode, erroCode);
  1559. return FALSE;
  1560. }
  1561. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("filePath %s", (LPCTSTR)ssFilePath);
  1562. int nDelSuc = 0;
  1563. int nDelFailed = 0;
  1564. int nfileSum = ClearSpecifieFile(DFT_CenterSetting, ssFilePath, nDelSuc, nDelFailed);
  1565. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("#CenterSetting#There are %d file(s), delete: %d, failed: %d.", nfileSum, nDelSuc, nDelFailed);
  1566. spConfigRun->WriteConfigValue("LastClearTime", "CenterSetting", (const char*)CSimpleStringA::Format("0x%08X", (DWORD)CSmallDateTime::GetNow()));
  1567. return TRUE;
  1568. }
  1569. #endif // RVC_OS_LINUX
  1570. bool ResourceWatcherFSM::IsNeedToFirstClear()
  1571. {
  1572. #ifdef RVC_OS_LINUX
  1573. int nLastTaskTime(0);
  1574. CSmartPointer<IConfigInfo> spRunConfig;
  1575. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  1576. spRunConfig->ReadConfigValueInt("LastClearTime", "Video", nLastTaskTime);
  1577. return !IsTodayDone(nLastTaskTime);
  1578. #else
  1579. bool result = true;
  1580. CSimpleStringA csLstExecTime;
  1581. CSmartPointer<IConfigInfo> spRunConfig;
  1582. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  1583. spRunConfig->ReadConfigValue("Video", "LastTime", csLstExecTime);
  1584. if (csLstExecTime.GetLength() > 1) {
  1585. int nYear = 0, nMonth = 0, nDay = 0;
  1586. spRunConfig->ReadConfigValueInt("Video", "Year", nYear);
  1587. spRunConfig->ReadConfigValueInt("Video", "Month", nMonth);
  1588. spRunConfig->ReadConfigValueInt("Video", "Day", nDay);
  1589. if (nYear != 0 && nMonth != 0 && nDay != 0) {
  1590. SYSTEMTIME stUTC, stLocal;
  1591. GetSystemTime(&stUTC);
  1592. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  1593. if (nYear == stLocal.wYear && nMonth == stLocal.wMonth && nDay == stLocal.wDay) {
  1594. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Video Clear job has been done today in %s", (LPCTSTR)csLstExecTime);
  1595. result = false;
  1596. }
  1597. }
  1598. }
  1599. return result;
  1600. #endif // RVC_OS_LINUX
  1601. }
  1602. void ResourceWatcherFSM::DeleteFiles()
  1603. {
  1604. string p;
  1605. CSmartPointer<IConfigInfo> spCtSettingConfig;
  1606. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  1607. CSimpleStringA upPath(""), tempPath(""), tPath(""), upgradePath("");
  1608. spCtSettingConfig->ReadConfigValue("ResourceWatcher", "DocumentCleanPath", tempPath);
  1609. if (tempPath.GetLength() == 0) {
  1610. return;
  1611. }
  1612. ErrorCodeEnum erroCode = Error_Succeed;
  1613. CSmartPointer<IEntityFunction> pFunc = GetEntityBase()->GetFunction();
  1614. CSmartPointer<IConfigInfo> spConfigRun;
  1615. //判断是否需要被清理
  1616. erroCode = pFunc->OpenConfig(Config_Run, spConfigRun);
  1617. if (erroCode != Error_Succeed) {
  1618. LogWarn(Severity_Middle, erroCode, 0, "Open Run Config failed");
  1619. return;
  1620. }
  1621. int nLastTaskTime = 0;
  1622. int nLstFlag = 0;
  1623. bool bFailFlag = false;
  1624. CSimpleStringA optFileName("");
  1625. UINT64 utVersion = 0;
  1626. spConfigRun->ReadConfigValueHexInt("DeleteDocumentTask", "OptVer", utVersion);
  1627. spConfigRun->ReadConfigValue("DeleteDocumentTask", "DocumentName", optFileName);
  1628. spConfigRun->ReadConfigValueInt("DeleteDocumentTask", "LastCondi", nLstFlag);
  1629. spConfigRun->ReadConfigValueInt("DeleteDocumentTask", "LastTime", nLastTaskTime);
  1630. CVersion optVer(utVersion);
  1631. SYSTEMTIME stTaskTime = CSmallDateTime(nLastTaskTime).ToSystemTime();
  1632. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("LastTime<%04d-%02d-%02d %02d:%02d:%02d>: OptVer: %s, LastCondi: %d",
  1633. stTaskTime.wYear, stTaskTime.wMonth, stTaskTime.wDay,
  1634. stTaskTime.wHour, stTaskTime.wMinute, stTaskTime.wSecond,
  1635. (LPCTSTR)optVer.ToString(), nLstFlag);
  1636. if (optVer.IsValid() && optVer == m_RvcSysinfo.InstallVersion && !nLstFlag && optFileName == tempPath)
  1637. {
  1638. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Certain document cleaning has been done before.");
  1639. return;
  1640. }
  1641. #ifdef RVC_OS_WIN
  1642. //删除upPath目录下的文件
  1643. struct _finddata_t upfileinfo;
  1644. long upFile = 0;
  1645. WIN32_FIND_DATA wfd;
  1646. HANDLE hFind;
  1647. upPath = tempPath;
  1648. upPath.Append("\\*");
  1649. hFind = FindFirstFile(upPath.GetData(), &wfd);
  1650. if (hFind != INVALID_HANDLE_VALUE) {
  1651. if ((wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY) {
  1652. int nDelSuc = 0, nDelFail = 0, nFileCount = 0;
  1653. nFileCount = ProcessFileDelete(upPath.GetData(), nDelSuc, nDelFail);
  1654. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Process result(%s): count(%d), suc(%d), failed(%d).", upPath.GetData(), nFileCount, nDelSuc, nDelFail);
  1655. if (nDelFail != 0) {
  1656. bFailFlag = true;
  1657. //break;
  1658. }
  1659. FindClose(hFind);
  1660. }
  1661. else
  1662. {
  1663. if (DeleteFile(tempPath.GetData()) != 0) {
  1664. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("删除文件%s成功。", tempPath.GetData());
  1665. //break;
  1666. }
  1667. else
  1668. {
  1669. bFailFlag = true;
  1670. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("删除文件%s失败。", tempPath.GetData());
  1671. }
  1672. }
  1673. }
  1674. else
  1675. {
  1676. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("FindFirstFile failed (%s)(%u)", upPath.GetData(), GetLastError());
  1677. }
  1678. #else //LINUX
  1679. struct stat buffer;
  1680. upPath = tempPath;
  1681. if (stat(upPath.GetData(), &buffer) == 0 && S_ISDIR(buffer.st_mode)) //判断是不是文件夹
  1682. {
  1683. int nDelSuc = 0, nDelFail = 0, nFileCount = 0;
  1684. nFileCount = ProcessFileDelete(upPath.GetData(), nDelSuc, nDelFail);
  1685. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Process result(%s): count(%d), suc(%d), failed(%d).", upPath.GetData(), nFileCount, nDelSuc, nDelFail);
  1686. if (nDelFail != 0) {
  1687. bFailFlag = true;
  1688. //break;
  1689. }
  1690. }
  1691. else
  1692. {
  1693. if (RemoveFileA(tempPath.GetData()) != 0) {
  1694. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("删除文件%s成功。", tempPath.GetData());
  1695. //break;
  1696. }
  1697. else
  1698. {
  1699. bFailFlag = true;
  1700. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("删除文件%s失败。", tempPath.GetData());
  1701. }
  1702. }
  1703. #endif // RVC_OS_WIN
  1704. nLstFlag = bFailFlag ? 1 : 0;
  1705. spConfigRun->WriteConfigValueHexInt("DeleteDocumentTask", "OptVer", m_RvcSysinfo.InstallVersion.GetVersion64());
  1706. spConfigRun->WriteConfigValue("DeleteDocumentTask", "DocumentName", tempPath);
  1707. spConfigRun->WriteConfigValueInt("DeleteDocumentTask", "LastCondi", nLstFlag);
  1708. spConfigRun->WriteConfigValue("DeleteDocumentTask", "LastTime",
  1709. (LPCTSTR)CSimpleStringA::Format("0x%08X", (DWORD)CSmallDateTime::GetNow()));
  1710. if (!bFailFlag) {
  1711. LogWarn(Severity_Low, Error_Debug, LOG_WARN_DIR_DELETE, CSimpleStringA::Format("Delete files in [%s] success", tempPath.GetData()));
  1712. }
  1713. else
  1714. {
  1715. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Delete files in [%s] failed.", tempPath.GetData());
  1716. }
  1717. }
  1718. int ResourceWatcherFSM::ProcessFileDelete(LPCTSTR lpszPath, int& nDelSucCnt, int& nDelFailedCnt)
  1719. {
  1720. #ifdef RVC_OS_LINUX
  1721. int fileCnt = 0;
  1722. int fileLength = strlen(lpszPath);
  1723. char* searchFilePath = new char[MAX_PATH];
  1724. char* tempFilePath = new char[MAX_PATH];
  1725. if (searchFilePath == NULL || tempFilePath == NULL) {
  1726. if (searchFilePath) delete[] searchFilePath;
  1727. if (tempFilePath) delete[] tempFilePath;
  1728. return fileCnt;
  1729. }
  1730. ZeroMemory(searchFilePath, MAX_PATH);
  1731. ZeroMemory(tempFilePath, MAX_PATH);
  1732. struct stat st;
  1733. if (stat(lpszPath, &st) < 0 || !S_ISDIR(st.st_mode)) {
  1734. if (searchFilePath) delete[] searchFilePath;
  1735. if (tempFilePath) delete[] tempFilePath;
  1736. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("stat failed or is not dir.");
  1737. return fileCnt;
  1738. }
  1739. DIR* d = opendir(lpszPath);
  1740. if (!d) {
  1741. if (searchFilePath) delete[] searchFilePath;
  1742. if (tempFilePath) delete[] tempFilePath;
  1743. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("opendir failed: %d", errno);
  1744. return fileCnt;
  1745. }
  1746. struct dirent* dp = NULL;
  1747. while ((dp = readdir(d)) != NULL) {
  1748. if ((!strcmp(dp->d_name, ".")) || (!strcmp(dp->d_name, ".."))) {
  1749. continue;
  1750. }
  1751. strcpy(tempFilePath, lpszPath);
  1752. if (strlen(tempFilePath) + strlen(dp->d_name) + 3 >= MAX_PATH) {
  1753. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  1754. continue;
  1755. }
  1756. strcat(tempFilePath, "/");
  1757. strcat(tempFilePath, dp->d_name);
  1758. stat(tempFilePath, &st);
  1759. bool isdir = !!(S_ISDIR(st.st_mode));
  1760. if (isdir) {
  1761. int tDelSucCnt = 0, tDelFailedCnt = 0;
  1762. fileCnt += ProcessFileDelete((LPCTSTR)tempFilePath, tDelSucCnt, tDelFailedCnt);
  1763. if (tDelFailedCnt == 0 && rmdir(tempFilePath) != 0) {
  1764. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("rmdir(%s) failed, GLE = %u.", (LPCTSTR)tempFilePath, errno);
  1765. nDelFailedCnt++;
  1766. }
  1767. nDelFailedCnt += tDelFailedCnt;
  1768. nDelSucCnt += tDelSucCnt;
  1769. }
  1770. else {
  1771. if (RemoveFileA(tempFilePath)) {
  1772. nDelSucCnt++;
  1773. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Delete [%s] suc.", dp->d_name);
  1774. }
  1775. else {
  1776. nDelFailedCnt++;
  1777. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RemoveFileA [%s] failed, GLE = %u.", tempFilePath, errno);
  1778. }
  1779. fileCnt++;
  1780. }
  1781. }
  1782. closedir(d);
  1783. if (searchFilePath) delete[] searchFilePath;
  1784. if (tempFilePath) delete[] tempFilePath;
  1785. searchFilePath = NULL;
  1786. tempFilePath = NULL;
  1787. return fileCnt;
  1788. #else
  1789. int fileCnt = 0;
  1790. int fileLength = strlen(lpszPath);
  1791. WIN32_FIND_DATA wfd;
  1792. HANDLE hFind;
  1793. hFind = FindFirstFile(lpszPath, &wfd);
  1794. if (hFind == INVALID_HANDLE_VALUE)
  1795. {
  1796. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("FindFirstFile failed GLE = %u.", GetLastError());
  1797. return 0;
  1798. }
  1799. char* searchFilePath = new char[MAX_PATH];
  1800. char* tempFilePath = new char[MAX_PATH];
  1801. if (searchFilePath == NULL || tempFilePath == NULL)
  1802. {
  1803. if (searchFilePath) delete[] searchFilePath;
  1804. if (tempFilePath) delete[] tempFilePath;
  1805. searchFilePath = NULL;
  1806. tempFilePath = NULL;
  1807. return fileCnt;
  1808. }
  1809. ZeroMemory(searchFilePath, MAX_PATH);
  1810. ZeroMemory(tempFilePath, MAX_PATH);
  1811. do
  1812. {
  1813. if ((wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
  1814. {
  1815. if (wfd.cFileName[0] != '.')
  1816. {
  1817. strcpy_s(tempFilePath, MAX_PATH, lpszPath);
  1818. tempFilePath[fileLength - 1] = '\0';
  1819. if (strlen(tempFilePath) + strlen(wfd.cFileName) + 3 >= MAX_PATH)
  1820. {
  1821. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  1822. }
  1823. else
  1824. {
  1825. strcat_s(tempFilePath, MAX_PATH, wfd.cFileName);
  1826. CSimpleStringA tmpPath = tempFilePath;
  1827. strcat_s(tempFilePath, MAX_PATH, "\\*");
  1828. int tDelSucCnt = 0, tDelFailedCnt = 0;
  1829. fileCnt += ProcessFileDelete((LPCTSTR)tempFilePath, tDelSucCnt, tDelFailedCnt);
  1830. if (tDelFailedCnt == 0 && !RemoveDirectoryA(tmpPath))
  1831. {
  1832. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RemoveDirectoryA(%s) failed, GLE = %u.", (LPCTSTR)tmpPath, GetLastError());
  1833. nDelFailedCnt++;
  1834. }
  1835. nDelFailedCnt += tDelFailedCnt;
  1836. nDelSucCnt += tDelSucCnt;
  1837. }
  1838. }
  1839. }
  1840. else
  1841. {
  1842. strcpy_s(searchFilePath, MAX_PATH, lpszPath);
  1843. searchFilePath[fileLength - 1] = '\0';
  1844. strcat_s(searchFilePath, MAX_PATH, wfd.cFileName);
  1845. if (DeleteFile(searchFilePath) != 0)
  1846. {
  1847. nDelSucCnt++;
  1848. }
  1849. else
  1850. {
  1851. nDelFailedCnt++;
  1852. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Delete [%s] failed, GLE = %u.", wfd.cFileName, GetLastError());
  1853. }
  1854. fileCnt++;
  1855. }
  1856. } while (FindNextFileA(hFind, &wfd));
  1857. FindClose(hFind);
  1858. if (searchFilePath) delete[] searchFilePath;
  1859. if (tempFilePath) delete[] tempFilePath;
  1860. searchFilePath = NULL;
  1861. tempFilePath = NULL;
  1862. return fileCnt;
  1863. #endif // RVC_OS_LINUX
  1864. }
  1865. BOOL ResourceWatcherFSM::DeleteVersionPackage()
  1866. {
  1867. BOOL bRet = FALSE;
  1868. ErrorCodeEnum erroCode = Error_Succeed;
  1869. CSmartPointer<IEntityFunction> pFunc = GetEntityBase()->GetFunction();
  1870. CSmartPointer<IConfigInfo> spConfigRun;
  1871. CSmartPointer<IConfigInfo> spConfig;
  1872. CSystemStaticInfo info;
  1873. if (pFunc->GetSystemStaticInfo(info) != Error_Succeed) {
  1874. LogWarn(Severity_Middle, erroCode, 0, "Get system static info error");
  1875. return FALSE;
  1876. }
  1877. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Current version: %s", (LPCTSTR)info.InstallVersion.ToString());
  1878. //判断是否需要被清理
  1879. int nLastTaskTime = 0;
  1880. int nLstFlag = 0;
  1881. int nDelCount = 0;
  1882. UINT64 utVersion = 0;
  1883. int verSaved = DEFAULT_VERSION_SAVED_COUNT;
  1884. int tmpSaved = 0;
  1885. pFunc->OpenConfig(Config_CenterSetting, spConfig);
  1886. spConfig->ReadConfigValueInt("VersionSaved", "default", tmpSaved);
  1887. if (tmpSaved > DEFAULT_VERSION_SAVED_COUNT) {
  1888. verSaved = tmpSaved;
  1889. }
  1890. pFunc->OpenConfig(Config_Run, spConfigRun);
  1891. spConfigRun->ReadConfigValueHexInt("VersionClear", "OptVer", utVersion);
  1892. spConfigRun->ReadConfigValueInt("VersionClear", "DelVerCnt", nDelCount);
  1893. spConfigRun->ReadConfigValueInt("VersionClear", "LastCondi", nLstFlag);
  1894. spConfigRun->ReadConfigValueInt("VersionClear", "LastTime", nLastTaskTime);
  1895. CVersion optVer(utVersion);
  1896. SYSTEMTIME stTaskTime = CSmallDateTime(nLastTaskTime).ToSystemTime();
  1897. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("LastTime<%04d-%02d-%02d %02d:%02d:%02d>: OptVer: %s, LastCondi: %d",
  1898. stTaskTime.wYear, stTaskTime.wMonth, stTaskTime.wDay,
  1899. stTaskTime.wHour, stTaskTime.wMinute, stTaskTime.wSecond,
  1900. (LPCTSTR)optVer.ToString(), nLstFlag);
  1901. //判断是否需要被清理
  1902. if (optVer.IsValid() && optVer == info.InstallVersion && !nLstFlag) {
  1903. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("VersionPackage cleaning has been done before.");
  1904. return TRUE;
  1905. }
  1906. CSimpleStringA strRunInfo, strStartTime;
  1907. if (m_pEntity->GetFunction()->GetPath("RunInfo", strRunInfo) == Error_Succeed) {
  1908. strStartTime = strRunInfo + SPLIT_SLASH_STR + "runcfg" + SPLIT_SLASH_STR + "starttime.dat";
  1909. if (!ExistsFileA(strStartTime.GetData())) {
  1910. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("upgrade prpcess done, try to delete version files.");//完成升级过程,通过
  1911. }
  1912. else
  1913. {
  1914. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("upgrade process is not end ,should not delete version files.");
  1915. return FALSE;//升级未完成,不进行删除
  1916. }
  1917. }
  1918. else
  1919. {
  1920. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("upgrade process is not end ,get runinfo path is error, should not delete version files.");
  1921. return FALSE;//失败,继续等待
  1922. }
  1923. //获取当前目录下的文件夹名称, 将符合命名规则的终端版本名称push_back
  1924. CSimpleStringA csPath;
  1925. ErrorCodeEnum Error = m_pEntity->GetFunction()->GetPath("RootVer", csPath); //获取当前版本路劲 例如:C:\Run
  1926. if (Error_Succeed == Error) {
  1927. if (csPath.IsNullOrEmpty()) {
  1928. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("get currVer path is null");
  1929. return FALSE;
  1930. }
  1931. else
  1932. {
  1933. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("CUR VERSION PATH = %s.", csPath.GetData());
  1934. }
  1935. }
  1936. else
  1937. {
  1938. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("get Ver path error");
  1939. return FALSE;
  1940. }
  1941. vector<CVersion> intallInfoVects;
  1942. vector<string> files;
  1943. CSimpleStringA upPath;
  1944. ErrorCodeEnum upError = pFunc->GetPath("Upgraded", upPath);
  1945. if (Error_Succeed != upError) {
  1946. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("get Upgraded path error");
  1947. return FALSE;
  1948. }
  1949. #ifdef RVC_OS_LINUX
  1950. DIR* dp;
  1951. struct dirent* dirp;
  1952. if ((dp = opendir(csPath.GetData())) != NULL) {
  1953. while ((dirp = readdir(dp)) != NULL) {
  1954. //files.push_back(string(dirp->d_name));
  1955. CInstallInfo verInfo = {};
  1956. DWORD dwMajor(0), dwMinor(0), dwRevision(0), dwBuild(0);
  1957. int n = sscanf(dirp->d_name, "%d.%d.%d.%d", &dwMajor, &dwMinor, &dwRevision, &dwBuild);
  1958. if (n != 4)
  1959. {
  1960. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not version file.[%s]", dirp->d_name);
  1961. continue;
  1962. }
  1963. else {
  1964. CVersion verName(dwMajor, dwMinor, dwRevision, dwBuild);
  1965. if (verName.IsValid() && verName < info.InstallVersion) //对高于自身版本的文件不予删除
  1966. {
  1967. intallInfoVects.push_back(verName);
  1968. }
  1969. else {
  1970. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not version file.[%s]", dirp->d_name);
  1971. }
  1972. }
  1973. }
  1974. closedir(dp);
  1975. }
  1976. //删除rvc/Upgraded目录下的文件
  1977. DIR* updp;
  1978. struct dirent* updirp;
  1979. if ((updp = opendir(upPath.GetData())) != NULL) {
  1980. while ((updirp = readdir(updp)) != NULL) {
  1981. char szPath[MAX_VERSION_PATH] = { 0 };
  1982. strcat(szPath, upPath.GetData());
  1983. CSimpleStringA tmpDirPath = szPath;
  1984. strcat(szPath, SPLIT_SLASH_STR);
  1985. strcat(szPath, updirp->d_name);
  1986. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("operating Upgrade path: [%s][%s]", (LPCTSTR)tmpDirPath, szPath);
  1987. RemoveFileA(szPath);
  1988. }
  1989. closedir(updp);
  1990. }
  1991. #else
  1992. long hFile = 0;
  1993. struct _finddata_t fileinfo;//文件信息读取结构
  1994. string p;
  1995. if ((hFile = _findfirst(p.assign(string(csPath.GetData())).append("\\*").c_str(), &fileinfo)) != -1) {
  1996. do {
  1997. CInstallInfo verInfo = {};
  1998. DWORD dwMajor(0), dwMinor(0), dwRevision(0), dwBuild(0);
  1999. int n = sscanf(fileinfo.name, "%d.%d.%d.%d", &dwMajor, &dwMinor, &dwRevision, &dwBuild);
  2000. if (n != 4)
  2001. {
  2002. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not version file.[%s]", fileinfo.name);
  2003. continue;
  2004. }
  2005. else {
  2006. CVersion verName(dwMajor, dwMinor, dwRevision, dwBuild);
  2007. if (verName.IsValid() && verName < info.InstallVersion)
  2008. {
  2009. intallInfoVects.push_back(verName);
  2010. }
  2011. else {
  2012. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not version file.[%s]", fileinfo.name);
  2013. }
  2014. }
  2015. } while (_findnext(hFile, &fileinfo) == 0);
  2016. _findclose(hFile);
  2017. }
  2018. //删除rvc/Upgraded目录下的文件
  2019. struct _finddata_t upfileinfo;
  2020. long upFile = 0;
  2021. if ((upFile = _findfirst(p.assign(string(upPath.GetData())).append("\\*").c_str(), &upfileinfo)) != -1) {
  2022. do {
  2023. char szPath[MAX_VERSION_PATH] = { 0 };
  2024. strcat_s(szPath, upPath.GetData());
  2025. CSimpleStringA tmpDirPath = szPath;
  2026. strcat_s(szPath, SPLIT_SLASH_STR);
  2027. strcat_s(szPath, upfileinfo.name);
  2028. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("operating Upgrade path: [%s][%s]", (LPCTSTR)tmpDirPath, szPath);
  2029. RemoveFileA(szPath);
  2030. } while (_findnext(upFile, &upfileinfo) == 0);
  2031. _findclose(upFile);
  2032. }
  2033. #endif // RVC_OS_LINUX
  2034. int nPackCount = intallInfoVects.size();
  2035. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Install Package's count: %d", nPackCount);
  2036. bool bFailFlag = false;
  2037. int nDelVersionCnt = 0;
  2038. std::string tInfo("");
  2039. if (nPackCount != 0) {
  2040. int nSavedCount = 0;
  2041. sort(intallInfoVects.begin(), intallInfoVects.end());
  2042. vector<CVersion>::reverse_iterator riter = intallInfoVects.rbegin();
  2043. while (riter != intallInfoVects.rend()) {
  2044. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("InstallPackage Info: %s", (LPCTSTR)(*riter).ToString());
  2045. if (++nSavedCount <= verSaved) {
  2046. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Save above version");
  2047. riter++;
  2048. continue;
  2049. }
  2050. if (!DeleteSpecificeVersionDir(*riter)) bFailFlag = true;
  2051. else {
  2052. nDelVersionCnt++;
  2053. tInfo = tInfo + "|" + std::string((*riter).ToString().GetData());
  2054. }
  2055. riter++;
  2056. }
  2057. }
  2058. nLstFlag = bFailFlag ? 1 : 0;
  2059. spConfigRun->WriteConfigValueHexInt("VersionClear", "OptVer", info.InstallVersion.GetVersion64());
  2060. spConfigRun->WriteConfigValueInt("VersionClear", "DelVerCnt", nDelCount + nDelVersionCnt);
  2061. spConfigRun->WriteConfigValueInt("VersionClear", "LastCondi", nLstFlag);
  2062. spConfigRun->WriteConfigValue("VersionClear", "LastTime",
  2063. (LPCTSTR)CSimpleStringA::Format("0x%08X", (DWORD)CSmallDateTime::GetNow()));
  2064. if (nDelVersionCnt > 0) {
  2065. std::string warn = "Delete Vesion file count : " + std::to_string((long long)nDelVersionCnt) + ". ----- " + tInfo + ".";
  2066. LogWarn(Severity_Low, Error_Debug, LOG_WARN_VER_DELETE, CSimpleStringA::Format("%s", warn.c_str()));
  2067. }
  2068. return TRUE;
  2069. }
  2070. BOOL ResourceWatcherFSM::DeleteSpecificeVersionDir(CVersion version)
  2071. {
  2072. #ifdef RVC_OS_LINUX
  2073. ErrorCodeEnum erroCode = Error_Unexpect;
  2074. CSimpleStringA strPath;
  2075. if (GetEntityBase()->GetFunction()->GetPath("RootVer", strPath) != Error_Succeed)
  2076. return FALSE;
  2077. strPath = strPath + SPLIT_SLASH_STR + version.ToString();
  2078. if (!ExistsDirA(strPath)) {
  2079. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Directory(%s) Not Existed, Maybe delete before !!", strPath.GetData());
  2080. return TRUE;
  2081. }
  2082. CSimpleStringA tmpDirPath = strPath;
  2083. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("operating version path: [%s][%s]", (LPCTSTR)tmpDirPath, (LPCTSTR)strPath);
  2084. int nDelSuc = 0, nDelFail = 0, nFileCount = 0;
  2085. nFileCount = ProcessFileDelete((LPCTSTR)tmpDirPath, nDelSuc, nDelFail);
  2086. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Process result(%s): count(%d), suc(%d), failed(%d).", tmpDirPath.GetData(), nFileCount, nDelSuc, nDelFail);
  2087. if (nDelFail != 0 || rmdir(tmpDirPath) != 0) {
  2088. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Unexpect condition happen: (%d)", nDelFail);
  2089. return FALSE;
  2090. }
  2091. return TRUE;
  2092. #else
  2093. ErrorCodeEnum erroCode = Error_Unexpect;
  2094. char szPath[MAX_VERSION_PATH] = { 0 };
  2095. ZeroMemory(szPath, MAX_VERSION_PATH);
  2096. DWORD dwLen = GetModuleFileNameA(NULL, szPath, MAX_VERSION_PATH);
  2097. if (dwLen != 0)
  2098. {
  2099. char* pch = strstr(szPath, "\\bin");
  2100. if (pch != NULL)
  2101. {
  2102. pch[0] = '\0';
  2103. CSimpleStringA strPath = szPath;
  2104. pch = strrchr(szPath, '\\');
  2105. if (pch != NULL)
  2106. {
  2107. pch[1] = '\0';
  2108. strcat_s(szPath, (LPCTSTR)version.ToString());
  2109. if (!ExistsDirA(szPath))
  2110. {
  2111. return TRUE;
  2112. }
  2113. CSimpleStringA tmpDirPath = szPath;
  2114. strcat_s(szPath, "\\*");
  2115. int nDelSuc = 0, nDelFail = 0, nFileCount = 0;
  2116. nFileCount = ProcessFileDelete(szPath, nDelSuc, nDelFail);
  2117. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Process result(%s): count(%d), suc(%d), failed(%d).", szPath, nFileCount, nDelSuc, nDelFail);
  2118. if (nDelFail != 0 || !RemoveDirectoryA(tmpDirPath))
  2119. {
  2120. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Unexpect condition happen: (%d)(%u).", nDelFail, GetLastError());
  2121. return FALSE;
  2122. }
  2123. erroCode = Error_Succeed;
  2124. }
  2125. }
  2126. }
  2127. else
  2128. {
  2129. LogWarn(Severity_Middle, Error_Unexpect, 0, "GetModuleFileNameA failed !");
  2130. }
  2131. return (erroCode == Error_Succeed);
  2132. #endif // RVC_OS_LINUX
  2133. }
  2134. #ifdef RVC_OS_LINUX
  2135. int ResourceWatcherFSM::CalculateAllVideoSize(LPCTSTR lpszPath, unsigned long long* puliSpaceBytes)
  2136. {
  2137. int fileCnt = 0;
  2138. int fileLength = strlen(lpszPath);
  2139. char* searchFilePath = new char[MAX_PATH];
  2140. char* tempFilePath = new char[MAX_PATH];
  2141. if (searchFilePath == NULL || tempFilePath == NULL) {
  2142. if (searchFilePath) delete[] searchFilePath;
  2143. if (tempFilePath) delete[] tempFilePath;
  2144. return fileCnt;
  2145. }
  2146. ZeroMemory(searchFilePath, MAX_PATH);
  2147. ZeroMemory(tempFilePath, MAX_PATH);
  2148. struct stat st;
  2149. if (stat(lpszPath, &st) < 0 || !S_ISDIR(st.st_mode)) {
  2150. if (searchFilePath) delete[] searchFilePath;
  2151. if (tempFilePath) delete[] tempFilePath;
  2152. return fileCnt;
  2153. }
  2154. DIR* d = opendir(lpszPath);
  2155. if (!d) {
  2156. if (searchFilePath) delete[] searchFilePath;
  2157. if (tempFilePath) delete[] tempFilePath;
  2158. return fileCnt;
  2159. }
  2160. struct dirent* dp = NULL;
  2161. while ((dp = readdir(d)) != NULL) {
  2162. if ((!strcmp(dp->d_name, ".")) || (!strcmp(dp->d_name, ".."))) {
  2163. continue;
  2164. }
  2165. strcpy(tempFilePath, lpszPath);
  2166. if (strlen(tempFilePath) + strlen(dp->d_name) + 3 >= MAX_PATH) {
  2167. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  2168. continue;
  2169. }
  2170. strcat(tempFilePath, "/");
  2171. strcat(tempFilePath, dp->d_name);
  2172. stat(tempFilePath, &st);
  2173. const bool isdir = !!(S_ISDIR(st.st_mode));
  2174. if (isdir) {
  2175. fileCnt += CalculateAllVideoSize((LPCTSTR)tempFilePath, puliSpaceBytes);
  2176. }
  2177. else {
  2178. *puliSpaceBytes += st.st_size;
  2179. SyncUpdateVideoCreateDaily(&st.st_mtime, 1);
  2180. fileCnt++;
  2181. }
  2182. }
  2183. closedir(d);
  2184. if (searchFilePath) delete[] searchFilePath;
  2185. if (tempFilePath) delete[] tempFilePath;
  2186. return fileCnt;
  2187. }
  2188. #endif // RVC_OS_LINUX
  2189. void ResourceWatcherFSM::SelfTest(EntityTestEnum eTestType,
  2190. CSmartPointer<ITransactionContext> pTransactionContext)
  2191. {
  2192. #ifdef RVC_OS_LINUX
  2193. static uint32_t times = 0;
  2194. pTransactionContext->SendAnswer(Error_Succeed);
  2195. times++;
  2196. if (m_bVideoClearReady && times > 60 * 2) { //2 hour per times
  2197. times = 0;
  2198. PostEventFIFO(new FSMEvent(USER_EVT_CLEAR_VIDEOFILE_ENHANCE));
  2199. }
  2200. #else
  2201. pTransactionContext->SendAnswer(Error_Succeed);
  2202. if (m_bVideoClearReady)
  2203. {
  2204. PostEventFIFO(new FSMEvent(USER_EVT_CLEAR_VIDEOFILE_ENHANCE));
  2205. }
  2206. #endif // RVC_OS_LINUX
  2207. }
  2208. void ResourceWatcherFSM::SetSysValAndBroadcast(DWORD dwVal, LPCTSTR lpszMessage)
  2209. {
  2210. static DWORD dwOldVal = (DWORD)-1;
  2211. CSmartPointer<IEntityFunction> spFunction = this->GetEntityBase()->GetFunction();
  2212. CSimpleStringA csVPStatus = VIDEO_SPACE_UNKNOWN;
  2213. CSimpleStringA csNewVPSatus = VIDEO_SPACE_UNKNOWN;
  2214. spFunction->GetSysVar(SYSVAR_DISKSTATEUS_FOR_VIDEO, csVPStatus);
  2215. if (dwVal == 4)
  2216. {
  2217. csNewVPSatus = VIDEO_SPACE_INSUFF_FATAL;
  2218. LogWarn(Severity_High, Error_TooSmallBuffer, LOG_ERR_RES_VIDEO_INSUFFICIENT, lpszMessage);
  2219. }
  2220. else if (dwVal == 2)
  2221. {
  2222. csNewVPSatus = VIDEO_SPACE_INSUFF_LITTLE;
  2223. if (dwVal != dwOldVal)
  2224. LogWarn(Severity_Middle, Error_TooSmallBuffer, LOG_WARN_RES_VIDEO_INSUFFICIENT, lpszMessage);
  2225. }
  2226. else if (dwVal == 1)
  2227. {
  2228. csNewVPSatus = VIDEO_SPACE_INSUFF_LESS;
  2229. if (dwVal != dwOldVal)
  2230. LogWarn(Severity_Low, Error_TooSmallBuffer, LOG_WARN_RES_VIDEO_INSUFFICIENT, lpszMessage);
  2231. }
  2232. else if (dwVal == 3)
  2233. {
  2234. csNewVPSatus = VIDEO_SPACE_SUFFICIENT;
  2235. if (dwVal != dwOldVal)
  2236. LogEvent(Severity_High, LOG_EVT_RES_VIDEO_SUFFICIENT, lpszMessage);
  2237. }
  2238. else if (dwVal == 0)
  2239. {
  2240. if (dwVal != dwOldVal)
  2241. LogEvent(Severity_High, LOG_EVT_RES_VIDEO_SUFFICIENT, lpszMessage);
  2242. }
  2243. if (csNewVPSatus != csVPStatus)
  2244. {
  2245. ErrorCodeEnum eRet = spFunction->SetSysVar(SYSVAR_DISKSTATEUS_FOR_VIDEO, csNewVPSatus);
  2246. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("SetSysVar(%s) from '%s' to '%s' returned %d",
  2247. SYSVAR_DISKSTATEUS_FOR_VIDEO
  2248. , (LPCTSTR)csVPStatus
  2249. , (LPCTSTR)csNewVPSatus
  2250. , eRet);
  2251. }
  2252. dwOldVal = dwVal;
  2253. }
  2254. void ResourceWatcherFSM::HardwareInfoTimer(void* pData)
  2255. {
  2256. if (hardwareCheckFlag > 0) {
  2257. GetSystemCPUStatus();
  2258. GetSystemMemoryStatus();
  2259. }
  2260. GetEntityBase()->GetFunction()->ResetTimer(TIMER_HARDWARE_CHECK, hardwareCheckTime);
  2261. }
  2262. void ResourceWatcherFSM::DiskCheckTimer(void* pData)
  2263. {
  2264. if (diskCheckFlag > 0) {
  2265. GetSystemDiskStatus();
  2266. #ifdef RVC_OS_LINUX
  2267. GetOperationDiskStatus();
  2268. #endif // RVC_OS_LINUX
  2269. }
  2270. GetEntityBase()->GetFunction()->ResetTimer(TIMER_DISK_CHECK, diskCheckTime);
  2271. }
  2272. #define CNNTYPE_UNKONWN 0 //0:未知
  2273. #define CNNTYPE_LAN 1 //网线已连接
  2274. #define CNNTYPE_LAN_DISCONNECT 2 //网线未连接(行内PAD,本应连接网线的)
  2275. #define CNNTYPE_DONGLE_CNTED 3 //外接卡托已连接(这种情况几乎没有)
  2276. #define CNNTYPE_DONGLE_DISCONNECT 4 //外接卡托未连接(这种情况几乎没有)
  2277. #define CNNTYPE_LTE_CNTED 5 //5:LTE设备,处于已连接状态
  2278. #define CNNTYPE_LTE_DISCONNECT 6 //LTE设备,处于未连接状态
  2279. #define CNNTYPE_MIFI_WIFI_SIM 7 //MIFI连接模式,WIFI已连接,SIM卡已连接
  2280. #define CNNTYPE_MIFI_WIFI_CNTED 8 //MIFI连接模式,WIFI已连接,SIM卡未连接
  2281. #define CNNTYPE_MIFI_WIFI_DISCONNECT 9 //MIFI连接模式,WIFI未连接
  2282. #define CNNTYPE_WIFI_CONNECT 10 //WIFI已连接(非MIFI模式,手工连接等其他场景)
  2283. #define CNNTYPE_NO_ANY_CONNECT 11 //无任何其他内容连接
  2284. ErrorCodeEnum ResourceWatcherFSM::CheckNetType(
  2285. SpReqAnsContext<ResourceWatcherService_CheckNetType_Req, ResourceWatcherService_CheckNetType_Ans>::Pointer ctx)
  2286. {
  2287. ErrorCodeEnum ec = Error_Succeed;
  2288. int netType = 0; //默认未知
  2289. #ifdef RVC_OS_LINUX
  2290. int i = 0;
  2291. int sockfd;
  2292. struct ifconf ifconf;
  2293. struct ifreq* ifreq;
  2294. char buf[1024];
  2295. //初始化ifconf
  2296. ifconf.ifc_len = 1024;
  2297. ifconf.ifc_buf = buf;
  2298. if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
  2299. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("socket error");
  2300. }
  2301. //获取所有接口信息
  2302. ioctl(sockfd, SIOCGIFCONF, &ifconf);
  2303. //逐个获取Ip地址
  2304. int size = ifconf.ifc_len / sizeof(struct ifreq);
  2305. ifreq = (struct ifreq*)buf;
  2306. string netName(ifreq->ifr_name);
  2307. if (size == 1 && netName == "lo") //只有逻辑地址 -- 网线被拔出
  2308. {
  2309. netType = 1;
  2310. }
  2311. else
  2312. {
  2313. for (i = size; i > 0; i--) {
  2314. string netName(ifreq->ifr_name); //有其他网卡 -- 有线
  2315. if (netName != "lo") {
  2316. netType = 2;
  2317. }
  2318. ifreq++;
  2319. }
  2320. }
  2321. #else
  2322. CSystemStaticInfo info;
  2323. GetEntityBase()->GetFunction()->GetSystemStaticInfo(info);
  2324. if (info.strMachineType.Compare("RVC.PAD") == 0) //PAD机型
  2325. {
  2326. CMbnImplHelper mbnHelper;
  2327. ResetDongleInfos();
  2328. const int nMBN = mbnHelper.EnumerateInterfaces();
  2329. const int cnt = g_DongleInfos.GetCount();
  2330. for (int i = 0; i < cnt; ++i) {
  2331. if (g_DongleInfos[i].state == MBD_ACTIVATION_CONNECTED) {
  2332. netType = 4; //移动网络
  2333. }
  2334. }
  2335. if (netType == 0 && 1 == IsWifiConnected()) {
  2336. netType = 3; //无线网络
  2337. }
  2338. if (netType == 0) {
  2339. if (SP::Module::Net::CheckLANConnectStatus()) {
  2340. netType = 2; //有线
  2341. }
  2342. else {
  2343. netType = 1; //未连接
  2344. }
  2345. }
  2346. }
  2347. else //大机、低柜、卡库
  2348. {
  2349. CoInitialize(NULL);
  2350. // 通过NLA接口获取网络状态
  2351. IUnknown* pUnknown = NULL;
  2352. bool bOnline = true;//是否在线
  2353. HRESULT Result = CoCreateInstance(CLSID_NetworkListManager, NULL, CLSCTX_ALL,
  2354. IID_IUnknown, (void**)&pUnknown);
  2355. if (SUCCEEDED(Result)) {
  2356. INetworkListManager* pNetworkListManager = NULL;
  2357. if (pUnknown)
  2358. Result = pUnknown->QueryInterface(IID_INetworkListManager, (void
  2359. **)&pNetworkListManager);
  2360. if (SUCCEEDED(Result)) {
  2361. VARIANT_BOOL IsConnect = VARIANT_FALSE;
  2362. if (pNetworkListManager) {
  2363. //Result = pNetworkListManager->get_IsConnectedToInternet(&IsConnect); //检测互联网连接
  2364. Result = pNetworkListManager->get_IsConnected(&IsConnect); //检测本地网络
  2365. }
  2366. if (SUCCEEDED(Result)) {
  2367. bOnline = (IsConnect == VARIANT_TRUE) ? true : false;
  2368. }
  2369. }
  2370. if (pNetworkListManager)
  2371. pNetworkListManager->Release();
  2372. }
  2373. if (pUnknown)
  2374. pUnknown->Release();
  2375. if (bOnline == true) {
  2376. netType = 2;
  2377. }
  2378. else
  2379. {
  2380. netType = 1;
  2381. }
  2382. }
  2383. #endif // RVC_OS_LINUX
  2384. if (ctx != NULL) {
  2385. ctx->Ans.netType = netType;
  2386. ctx->Answer(ec);
  2387. }
  2388. return ec;
  2389. }
  2390. ErrorCodeEnum ResourceWatcherFSM::GetBizLinks(
  2391. SpReqAnsContext<ResourceWatcherService_GetBizLinks_Req, ResourceWatcherService_GetBizLinks_Ans>::Pointer ctx)
  2392. {
  2393. LOG_FUNCTION();
  2394. ErrorCodeEnum ec = Error_Succeed;
  2395. int filter = ctx->Req.filter; //后续过滤使用
  2396. CAutoArray<CSimpleStringA> bizLinks;
  2397. CAutoArray<CSimpleStringA> bizNames;
  2398. CSmartPointer<IConfigInfo> spCtSettingConfig;
  2399. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  2400. if (filter == 0) //用户管理桌面业务网址
  2401. {
  2402. CSimpleStringA mLink(""), adLink(""), tLink("");
  2403. spCtSettingConfig->ReadConfigValue("Chromium", "UserMgrUrlFulture", mLink);
  2404. if (mLink.GetLength() != 0) {
  2405. CSimpleStringA tName = CSimpleStringA("主站点");
  2406. bizNames.Append(&tName, 0, 1);
  2407. bizLinks.Append(&mLink, 0, 1);
  2408. }
  2409. spCtSettingConfig->ReadConfigValue("Chromium", "UserMgrAd", adLink);
  2410. if (adLink.GetLength() != 0) {
  2411. CSimpleStringA tName = CSimpleStringA("广告站点");
  2412. bizNames.Append(&tName, 0, 1);
  2413. bizLinks.Append(&adLink, 0, 1);
  2414. }
  2415. int bizCnt = 0, outCnt = 0;
  2416. CSimpleStringA getLink("");
  2417. do {
  2418. bizCnt++;
  2419. getLink = CSimpleStringA("UserBizLink").Append(std::to_string(bizCnt).c_str());
  2420. spCtSettingConfig->ReadConfigValue("Chromium", (LPCTSTR)getLink, tLink);
  2421. if (tLink.GetLength() != 0) {
  2422. CAutoArray<CSimpleStringA> arr = tLink.Split('|');
  2423. if (arr.GetCount() == 3) {
  2424. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not bizlink info.");
  2425. continue;
  2426. }
  2427. outCnt++;
  2428. CSimpleStringA tName = CSimpleStringA("业务站点").Append(std::to_string(outCnt).c_str());
  2429. bizNames.Append(&tName, 0, 1);
  2430. bizLinks.Append(&tLink, 0, 1);
  2431. }
  2432. } while (tLink.GetLength() != 0);
  2433. }
  2434. else if (filter == 1) //错误页集中配置、准入网址
  2435. {
  2436. int bizCnt = 0;
  2437. CSimpleStringA info(""), getLink("");
  2438. do {
  2439. bizCnt++;
  2440. CSimpleStringA chineseName(""), entityName(""), key("");
  2441. CSimpleStringA url("");
  2442. getLink = CSimpleStringA("UserBizLink").Append(std::to_string(bizCnt).c_str());
  2443. spCtSettingConfig->ReadConfigValue("Chromium", (LPCTSTR)getLink, info);
  2444. if (info.GetLength() != 0) {
  2445. CAutoArray<CSimpleStringA> arr = info.Split('|');
  2446. int len = arr.GetCount();
  2447. if (len != 3) {
  2448. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not errPage Check config info.");
  2449. }
  2450. else
  2451. {
  2452. chineseName = arr[0];
  2453. entityName = arr[1];
  2454. key = arr[2];
  2455. spCtSettingConfig->ReadConfigValue(entityName.GetData(), key.GetData(), url);
  2456. if (url.GetLength() != 0) {
  2457. bizNames.Append(&chineseName, 0, 1);
  2458. bizLinks.Append(&url, 0, 1);
  2459. }
  2460. else
  2461. {
  2462. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Get errPage detecet url failed. EntityName:(%s), Key:(%s).",
  2463. entityName.GetData(), key.GetData());
  2464. }
  2465. }
  2466. }
  2467. } while (info.GetLength() != 0);
  2468. }
  2469. ctx->Ans.bizLinks = bizLinks;
  2470. ctx->Ans.bizNames = bizNames;
  2471. if (ctx != NULL) {
  2472. ctx->Answer(ec);
  2473. }
  2474. return ec;
  2475. }
  2476. #ifdef _MSC_VER
  2477. BOOL ResourceWatcherFSM::CalCulateUploadFileAddition(ULARGE_INTEGER& uiBytesCal)
  2478. {
  2479. uiBytesCal.QuadPart = 0;
  2480. BOOL bRet = FALSE;
  2481. ErrorCodeEnum ec = Error_Unexpect;
  2482. CSimpleStringA uploadVideoPath;
  2483. ec = GetEntityBase()->GetFunction()->GetPath(UPLOAD_VIDEO_PATH, uploadVideoPath);
  2484. if (uploadVideoPath.GetLength() <= 0) ec = Error_Param;
  2485. if (ec == Error_Succeed) {
  2486. CSimpleStringA filePath = uploadVideoPath + "\\*";
  2487. ULARGE_INTEGER uliSpaceSize;
  2488. uliSpaceSize.QuadPart = 0;
  2489. int nFileCnt = ProcessFileSpaceCalculation(filePath, &uliSpaceSize);
  2490. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  2491. ByteSprintf(szResult, (double)uliSpaceSize.QuadPart);
  2492. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Result: Files(%d), Size(%s).", nFileCnt, szResult);
  2493. uiBytesCal.QuadPart = uliSpaceSize.QuadPart;
  2494. }
  2495. return (ec == Error_Succeed);
  2496. }
  2497. int ResourceWatcherFSM::ProcessFileSpaceCalculation(LPCTSTR lpszPath, PULARGE_INTEGER puliSpaceBytes)
  2498. {
  2499. int fileCnt = 0;
  2500. int fileLength = strlen(lpszPath);
  2501. char* searchFilePath = new char[MAX_PATH];
  2502. char* tempFilePath = new char[MAX_PATH];
  2503. if (searchFilePath == NULL || tempFilePath == NULL)
  2504. {
  2505. if (searchFilePath) delete[] searchFilePath;
  2506. if (tempFilePath) delete[] tempFilePath;
  2507. return fileCnt;
  2508. }
  2509. ZeroMemory(searchFilePath, MAX_PATH);
  2510. ZeroMemory(tempFilePath, MAX_PATH);
  2511. WIN32_FIND_DATA wfd;
  2512. HANDLE hFind;
  2513. hFind = FindFirstFile(lpszPath, &wfd);
  2514. if (hFind == INVALID_HANDLE_VALUE)
  2515. {
  2516. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("FindFirstFile failed GLE = %u.", GetLastError());
  2517. if (searchFilePath) delete[] searchFilePath;
  2518. if (tempFilePath) delete[] tempFilePath;
  2519. return 0;
  2520. }
  2521. do
  2522. {
  2523. if ((wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
  2524. {
  2525. if (wfd.cFileName[0] != '.')
  2526. {
  2527. strcpy_s(tempFilePath, MAX_PATH, lpszPath);
  2528. tempFilePath[fileLength - 1] = '\0';
  2529. if (strlen(tempFilePath) + strlen(wfd.cFileName) + 3 >= MAX_PATH)
  2530. {
  2531. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("filePath is too long for current.");
  2532. }
  2533. else
  2534. {
  2535. strcat_s(tempFilePath, MAX_PATH, wfd.cFileName);
  2536. strcat_s(tempFilePath, MAX_PATH, "\\*");
  2537. fileCnt += ProcessFileSpaceCalculation((LPCTSTR)tempFilePath, puliSpaceBytes);
  2538. }
  2539. }
  2540. }
  2541. else
  2542. {
  2543. strcpy_s(searchFilePath, MAX_PATH, lpszPath);
  2544. searchFilePath[fileLength - 1] = '\0';
  2545. strcat_s(searchFilePath, MAX_PATH, wfd.cFileName);
  2546. ULARGE_INTEGER uiTemp;
  2547. uiTemp.HighPart = wfd.nFileSizeHigh;
  2548. uiTemp.LowPart = wfd.nFileSizeLow;
  2549. puliSpaceBytes->QuadPart += uiTemp.QuadPart;
  2550. SYSTEMTIME stUTC, stLocal;
  2551. FileTimeToSystemTime(&wfd.ftLastWriteTime, &stUTC);
  2552. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  2553. UINT uDate = stLocal.wYear * 10000 + stLocal.wMonth * 100 + stLocal.wDay;
  2554. VideoDailyRecord[uDate]++;
  2555. fileCnt++;
  2556. }
  2557. } while (FindNextFileA(hFind, &wfd));
  2558. FindClose(hFind);
  2559. if (searchFilePath) delete[] searchFilePath;
  2560. if (tempFilePath) delete[] tempFilePath;
  2561. return fileCnt;
  2562. }
  2563. #endif // _MSC_VER
  2564. DWORD ResourceWatcherFSM::ExecuteFile(LPCTSTR lpszFilePath, LPCTSTR lpParam)
  2565. {
  2566. DWORD dwRes = 0;
  2567. #ifdef _MSC_VER
  2568. SHELLEXECUTEINFOA shell;
  2569. memset(&shell, 0, sizeof(SHELLEXECUTEINFOA));
  2570. shell.cbSize = sizeof(SHELLEXECUTEINFOA);
  2571. shell.lpFile = lpszFilePath;
  2572. shell.lpParameters = lpParam;
  2573. shell.fMask = SEE_MASK_INVOKEIDLIST;
  2574. shell.lpVerb = NULL;
  2575. shell.nShow = SW_SHOWDEFAULT;
  2576. if (!ShellExecuteExA(&shell))
  2577. {
  2578. dwRes = GetLastError();
  2579. int nRes = (int)shell.hInstApp;
  2580. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("GLE=%u, InstApp=%d", dwRes, nRes);
  2581. switch (dwRes) {
  2582. case ERROR_NO_ACE_CONDITION:
  2583. LogWarn(Severity_Middle, Error_MethodNotFound, dwRes,
  2584. CSimpleStringA::Format("Target file(%s) association not available !", lpszFilePath));
  2585. break;
  2586. case ERROR_ACCESS_DENIED:
  2587. LogWarn(Severity_Middle, Error_NoPrivilege, dwRes,
  2588. CSimpleStringA::Format("Target file(%s) not privilege !", lpszFilePath));
  2589. break;
  2590. case ERROR_PATH_NOT_FOUND:
  2591. case ERROR_FILE_NOT_FOUND:
  2592. LogWarn(Severity_Middle, Error_NotExist, dwRes,
  2593. CSimpleStringA::Format("Target file(%s) not existed !", lpszFilePath));
  2594. break;
  2595. case ERROR_DLL_NOT_FOUND:
  2596. LogWarn(Severity_Middle, Error_NotExist, dwRes,
  2597. CSimpleStringA::Format("One of library file for Target(%s) not existed !", lpszFilePath));
  2598. break;
  2599. case ERROR_NOT_ENOUGH_MEMORY:
  2600. LogWarn(Severity_Middle, Error_Resource, dwRes,
  2601. CSimpleStringA::Format("Not enough memory for Target(%s) to perform !", lpszFilePath));
  2602. break;
  2603. case ERROR_SHARING_VIOLATION:
  2604. LogWarn(Severity_Middle, Error_Overflow, dwRes,
  2605. CSimpleStringA::Format("Violation occurred for Target(%s) to share !", lpszFilePath));
  2606. break;
  2607. default:
  2608. LogWarn(Severity_Middle, Error_Unexpect, dwRes,
  2609. CSimpleStringA::Format("Unpected condtion for Target(%s) to execute(%u) !"
  2610. , lpszFilePath, dwRes));
  2611. break;
  2612. }
  2613. }
  2614. #else
  2615. dwRes = 1;
  2616. #endif //_MSC_VER
  2617. return dwRes;
  2618. }
  2619. #ifdef _MSC_VER
  2620. BOOL ResourceWatcherFSM::ClearDirRecursiveEx(LPCSTR lpDirPath)
  2621. {
  2622. int len = strlen(lpDirPath);
  2623. if (lpDirPath == NULL || len == 0)
  2624. return FALSE;
  2625. if (len + 2 >= MAX_PATH)
  2626. {
  2627. return FALSE;
  2628. }
  2629. char* path = new char[MAX_PATH];
  2630. if (path == NULL) return FALSE;
  2631. memset(path, 0, sizeof(path));
  2632. strcpy_s(path, MAX_PATH, lpDirPath);
  2633. int pos = len - 1;
  2634. for (; pos >= 0 && (path[pos] == '/' || path[pos] == '\\'); --pos)
  2635. path[pos] = '\0';
  2636. path[pos + 1] = '\0';
  2637. strcat_s(path, MAX_PATH, "\\*");
  2638. len = strlen(path);
  2639. char tempFilePath[MAX_PATH];
  2640. ZeroMemory(tempFilePath, MAX_PATH);
  2641. WIN32_FIND_DATA wfd;
  2642. HANDLE hFind;
  2643. hFind = FindFirstFile(path, &wfd);
  2644. if (hFind == INVALID_HANDLE_VALUE)
  2645. {
  2646. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("FindFirstFile failed GLE = %u.", GetLastError());
  2647. delete[] path;
  2648. return FALSE;
  2649. }
  2650. BOOL bResult = TRUE;
  2651. do
  2652. {
  2653. if ((wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
  2654. {
  2655. if (wfd.cFileName[0] != '.')
  2656. {
  2657. strcpy_s(tempFilePath, path);
  2658. tempFilePath[len - 1] = '\0';
  2659. if (len + strlen(wfd.cFileName) > MAX_PATH)
  2660. {
  2661. bResult = FALSE;
  2662. break;
  2663. }
  2664. else
  2665. {
  2666. strcat_s(tempFilePath, wfd.cFileName);
  2667. if (!RemoveDirRecursiveA(tempFilePath))
  2668. {
  2669. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RemoveDirRecursiveA(%s) failed", tempFilePath);
  2670. bResult = FALSE;
  2671. break;
  2672. }
  2673. }
  2674. }
  2675. }
  2676. else
  2677. {
  2678. strcpy_s(tempFilePath, path);
  2679. tempFilePath[len - 1] = '\0';
  2680. strcat_s(tempFilePath, wfd.cFileName);
  2681. if (!RemoveFileA(tempFilePath))
  2682. {
  2683. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RemoveFileA(%s) failed", tempFilePath);
  2684. bResult = FALSE;
  2685. break;
  2686. }
  2687. }
  2688. } while (FindNextFileA(hFind, &wfd));
  2689. FindClose(hFind);
  2690. if (path) delete[] path;
  2691. return bResult;
  2692. }
  2693. #endif // _MSC_VER
  2694. BOOL ResourceWatcherFSM::ClearAd0Folder()
  2695. {
  2696. LOG_FUNCTION();
  2697. ErrorCodeEnum ec;
  2698. CSimpleStringA strAd0Path;
  2699. CSmartPointer<IConfigInfo> spConfigRun;
  2700. int nLastTaskTime(0);
  2701. CSimpleStringA ssFilePath(true);
  2702. ec = GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  2703. spConfigRun->ReadConfigValueInt("LastClearTime", "Ad0Task", nLastTaskTime);
  2704. if (IsTodayDone(nLastTaskTime)) {
  2705. return TRUE;
  2706. }
  2707. ec = GetEntityBase()->GetFunction()->GetPath("Ad0", strAd0Path);
  2708. if (ec != Error_Succeed || strAd0Path.IsNullOrEmpty()) {
  2709. return FALSE;
  2710. }
  2711. if (!ExistsDirA(strAd0Path)) {
  2712. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("The Folder is not existed");
  2713. spConfigRun->WriteConfigValue("LastClearTime", "Ad0Task",
  2714. (const char*)CSimpleStringA::Format("0x%08X", (DWORD)CSmallDateTime::GetNow()));
  2715. return TRUE;
  2716. }
  2717. char displayTime[MAX_PATH] = { 0 };
  2718. ZeroMemory(&mftAd0RemoveTime, sizeof(FILETIME));
  2719. CalculateBackTime(&mftAd0RemoveTime, AD0_DAY_OF_BACKWARD);
  2720. GetTimeFormatStr(displayTime, MAX_PATH, &mftAd0RemoveTime);
  2721. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("The date backward to delete: %s", displayTime);
  2722. #ifdef RVC_OS_LINUX
  2723. ssFilePath = strAd0Path;
  2724. #else
  2725. ssFilePath = strAd0Path + "\\*";
  2726. #endif // RVC_OS_LINUX
  2727. int nDelSuc = 0;
  2728. int nDelFailed = 0;
  2729. int nfileSum = ClearSpecifieFile(DFT_Ad0, (LPCTSTR)ssFilePath, nDelSuc, nDelFailed);
  2730. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("#Ad0#There are %d file(s), delete: %d, failed: %d.", nfileSum, nDelSuc, nDelFailed);
  2731. spConfigRun->WriteConfigValue("LastClearTime", "Ad0Task",
  2732. (const char*)CSimpleStringA::Format("0x%08X", (DWORD)CSmallDateTime::GetNow()));
  2733. return TRUE;
  2734. }
  2735. BOOL ResourceWatcherFSM::IsTodayDone(int nRecordTime)
  2736. {
  2737. if (nRecordTime <= 0) {
  2738. return FALSE;
  2739. }
  2740. SYSTEMTIME stNow = {};
  2741. SYSTEMTIME stTaskTime = CSmallDateTime(nRecordTime).ToSystemTime();
  2742. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Last record time: %04d-%02d-%02d %02d:%02d:%02d",
  2743. stTaskTime.wYear, stTaskTime.wMonth, stTaskTime.wDay,
  2744. stTaskTime.wHour, stTaskTime.wMinute, stTaskTime.wSecond);
  2745. #ifdef RVC_OS_LINUX
  2746. stNow = CSmallDateTime::GetNow().ToSystemTime();
  2747. #else
  2748. GetLocalTime(&stNow);
  2749. #endif // RVC_OS_LINUX
  2750. if (stTaskTime.wYear == stNow.wYear
  2751. && stTaskTime.wMonth == stNow.wMonth
  2752. && stTaskTime.wDay == stNow.wDay)
  2753. {
  2754. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("task has been executed today, last clear time: %s",
  2755. (const char*)CSmallDateTime(nRecordTime).ToTimeString());
  2756. return TRUE;
  2757. }
  2758. return FALSE;
  2759. }
  2760. ///*TODO: 使用共用库的接口 (80374374@11/24/2023)*/
  2761. BOOL ResourceWatcherFSM::GetSystemBootTime(CSmallDateTime& systemBootTime)
  2762. {
  2763. #ifdef RVC_OS_LINUX
  2764. DWORD ticks = 0;
  2765. SYSTEMTIME systemTime;
  2766. struct sysinfo info;
  2767. time_t curTime = 0;
  2768. time_t bootTime = 0;
  2769. struct tm* ptm = NULL;
  2770. if (sysinfo(&info)) {
  2771. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Failed to get sysinfo, errno:%u, reason:%s", errno, strerror(errno));
  2772. return FALSE;
  2773. }
  2774. time(&curTime);
  2775. if (curTime > info.uptime) {
  2776. bootTime = curTime - info.uptime;
  2777. }
  2778. else
  2779. {
  2780. bootTime = info.uptime - curTime;
  2781. }
  2782. ptm = localtime(&bootTime);
  2783. struct timespec ts;
  2784. if (!clock_gettime(CLOCK_MONOTONIC_RAW, &ts))
  2785. ticks = (ts.tv_sec * 1000) + (ts.tv_nsec / 1000000);
  2786. systemTime.wYear = (WORD)(ptm->tm_year + 1900);
  2787. systemTime.wMonth = (WORD)(ptm->tm_mon + 1);
  2788. systemTime.wDayOfWeek = (WORD)ptm->tm_wday;
  2789. systemTime.wDay = (WORD)ptm->tm_mday;
  2790. systemTime.wHour = (WORD)ptm->tm_hour;
  2791. systemTime.wMinute = (WORD)ptm->tm_min;
  2792. systemTime.wSecond = (WORD)ptm->tm_sec;
  2793. systemTime.wMilliseconds = (WORD)(ticks % 1000);
  2794. systemBootTime.FromSystemTime(systemTime);
  2795. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("OSStartTime: %s", systemBootTime.ToTimeString().GetData());
  2796. return TRUE;
  2797. #else
  2798. PDH_STATUS Status;
  2799. HQUERY Query = NULL;
  2800. HCOUNTER hcElapsedTimeCount;
  2801. BOOL fSuc = FALSE;
  2802. /** 杀毒软件会对附件这几个系统调用函数有所监控,导致有时会耗费5s时间 Gifur@20221019]*/
  2803. Status = PdhOpenQuery(NULL, NULL, &Query);
  2804. PDH_FMT_COUNTERVALUE counterValue;
  2805. if (Status != ERROR_SUCCESS) {
  2806. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("PdhOpenQuery failed with status 0x%x.", Status);
  2807. goto Cleanup;
  2808. }
  2809. Status = PdhAddCounter(Query, SystemElapsedQuery, NULL, &hcElapsedTimeCount);
  2810. if (Status != ERROR_SUCCESS) {
  2811. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("PdhAddCounter for SystemElapsedQuery failed with status 0x%x.", Status);
  2812. goto Cleanup;
  2813. }
  2814. // 查询性能监视器数据
  2815. Status = PdhCollectQueryData(Query);
  2816. if (Status != ERROR_SUCCESS) {
  2817. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("PdhCollectQueryData failed with 0x%x.", Status);
  2818. goto Cleanup;
  2819. }
  2820. Status = PdhGetFormattedCounterValue(hcElapsedTimeCount, PDH_FMT_LARGE, NULL, &counterValue);
  2821. if (Status == ERROR_SUCCESS) {
  2822. ULONGLONG ulSinceSeconds = counterValue.largeValue;
  2823. ULONG days = 0, hours = 0, minutes = 0, seconds = 0;
  2824. days = ULONG(ulSinceSeconds / DAY_DIV);
  2825. ulSinceSeconds %= DAY_DIV;
  2826. hours = ULONG(ulSinceSeconds / HOURS_DIV);
  2827. ulSinceSeconds %= HOURS_DIV;
  2828. minutes = ULONG(ulSinceSeconds / MINUS_DIV);
  2829. ulSinceSeconds %= MINUS_DIV;
  2830. seconds = ULONG(ulSinceSeconds);
  2831. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("SystemElapseTime: %u %02u:%02u:%02u", days, hours, minutes, seconds);
  2832. FILETIME ftCurTime, ftStartTime;
  2833. GetSystemTimeAsFileTime(&ftCurTime);
  2834. ULARGE_INTEGER uliCurTime;
  2835. uliCurTime.HighPart = ftCurTime.dwHighDateTime;
  2836. uliCurTime.LowPart = ftCurTime.dwLowDateTime;
  2837. uliCurTime.QuadPart -= counterValue.largeValue * 1e7;
  2838. ftStartTime.dwHighDateTime = uliCurTime.HighPart;
  2839. ftStartTime.dwLowDateTime = uliCurTime.LowPart;
  2840. SYSTEMTIME stUTC, stLocal;
  2841. FileTimeToSystemTime(&ftStartTime, &stUTC);
  2842. char temp[22];
  2843. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  2844. sprintf_s(temp, 22, "%d-%02d-%02d %02d:%02d:%02d",
  2845. stLocal.wYear, stLocal.wMonth, stLocal.wDay, stLocal.wHour, stLocal.wMinute, stLocal.wSecond);
  2846. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("OSStartTime: %s", temp);
  2847. systemBootTime.FromSystemTime(stLocal);
  2848. fSuc = TRUE;
  2849. }
  2850. Cleanup:
  2851. Status = PdhRemoveCounter(hcElapsedTimeCount);
  2852. if (Query) {
  2853. PdhCloseQuery(Query);
  2854. }
  2855. return fSuc;
  2856. #endif // RVC_OS_LINUX
  2857. }
  2858. #ifdef _MSC_VER
  2859. bool ResourceWatcherFSM::CreateLinkFile(const CSimpleStringA& szStartAppPath, const CSimpleStringA& szAddCmdLine,
  2860. const CSimpleStringA& szDestLnkPath, const CSimpleStringA& szIconPath)
  2861. {
  2862. HRESULT hr = CoInitialize(NULL);
  2863. bool bRet = false;
  2864. if (SUCCEEDED(hr))
  2865. {
  2866. IShellLink* pShellLink;
  2867. hr = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (void**)&pShellLink);
  2868. if (SUCCEEDED(hr))
  2869. {
  2870. pShellLink->SetPath(szStartAppPath);
  2871. string strTmp = szStartAppPath;
  2872. int nStart = strTmp.find_last_of("/\\");
  2873. pShellLink->SetWorkingDirectory(strTmp.substr(0, nStart).c_str());
  2874. pShellLink->SetArguments(szAddCmdLine);
  2875. if (szIconPath)
  2876. {
  2877. pShellLink->SetIconLocation(szIconPath, 0);
  2878. }
  2879. IPersistFile* pPersistFile;
  2880. hr = pShellLink->QueryInterface(IID_IPersistFile, (void**)&pPersistFile);
  2881. if (SUCCEEDED(hr))
  2882. {
  2883. hr = pPersistFile->Save(CSimpleStringA2W(szDestLnkPath).GetData(), FALSE);
  2884. if (SUCCEEDED(hr))
  2885. {
  2886. bRet = true;
  2887. }
  2888. pPersistFile->Release();
  2889. }
  2890. pShellLink->Release();
  2891. }
  2892. CoUninitialize();
  2893. }
  2894. return bRet;
  2895. }
  2896. void ResourceWatcherFSM::CreateLink(const CSimpleStringA& exePath, const CSimpleStringA& csVersion)
  2897. {
  2898. LPITEMIDLIST lp;
  2899. SHGetSpecialFolderLocation(0, CSIDL_DESKTOPDIRECTORY, &lp);
  2900. CHAR lstr[100] = "";
  2901. SHGetPathFromIDList(lp, lstr);
  2902. CSimpleStringA s = lstr;
  2903. s += "\\可视柜台.lnk";
  2904. CSimpleStringA icoPath;
  2905. ErrorCodeEnum Error = GetEntityBase()->GetFunction()->GetPath("RootVer", icoPath);
  2906. icoPath = icoPath + "\\" + csVersion.GetData() + "\\bin\\VTM.ico";
  2907. if (!CreateLinkFile(exePath, "", s, icoPath))
  2908. {
  2909. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("CreateLinkFile DESKTOP fail");
  2910. }
  2911. SHGetSpecialFolderLocation(0, CSIDL_PROGRAMS, &lp);
  2912. memset(lstr, 0, strlen(lstr));
  2913. SHGetPathFromIDList(lp, lstr);
  2914. s = lstr;
  2915. s += "\\可视柜台.lnk";
  2916. if (!CreateLinkFile(exePath, "", s, icoPath))
  2917. {
  2918. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("CreateLinkFile CSIDL_PROGRAMS fail");
  2919. }
  2920. }
  2921. bool ResourceWatcherFSM::CopyExeToRoot(const CSimpleStringA& csVersion)
  2922. {
  2923. CSimpleStringA rootVerPath, exePath;
  2924. ErrorCodeEnum rc = GetEntityBase()->GetFunction()->GetPath("RootVer", rootVerPath);
  2925. exePath = rootVerPath + "\\VTM.exe";
  2926. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("exePath:%s", exePath.GetData());
  2927. if (rc == Error_Succeed)
  2928. {
  2929. CSimpleStringA srcExePath, csPath = rootVerPath;
  2930. srcExePath = rootVerPath + "\\" + csVersion.GetData() + "\\bin\\VTM.exe";
  2931. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("srcExePath:%s exePath:%s", srcExePath.GetData(), exePath.GetData());
  2932. if (!ExistsFileA(srcExePath)) {
  2933. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("%s is not exist!", srcExePath.GetData());
  2934. return false;
  2935. }
  2936. if (!CopyFileA(srcExePath.GetData(), exePath.GetData(), FALSE))
  2937. {
  2938. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("CopyFileA Fail error :%d ", GetLastError());
  2939. return false;
  2940. }
  2941. }
  2942. else {
  2943. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetPath 4 RootVer failed: %s", SpStrError(rc));
  2944. return false;
  2945. }
  2946. return true;
  2947. }
  2948. bool ResourceWatcherFSM::GetVtmVersionFromActiveTxt(CSimpleStringA& csVersion)
  2949. {
  2950. bool bRet = false;
  2951. CSimpleStringA rootVerPath;
  2952. ErrorCodeEnum rc = GetEntityBase()->GetFunction()->GetPath("RootVer", rootVerPath);//获取version根路径
  2953. if (rc != Error_Succeed) {
  2954. LogWarn(Severity_Middle, Error_Exception, LOG_RESOURCEWATCHER_OPEN_ACTIVE_FAIL, "testActive fail , get RootVer path is fai");
  2955. return false;
  2956. }
  2957. CSimpleStringA strActiveFile = CSimpleStringA::Format("%s\\active.txt", rootVerPath.GetData());
  2958. FILE* pFile = fopen(strActiveFile.GetData(), "rb+");
  2959. if (pFile != NULL)
  2960. {
  2961. char szTemp[256] = {};
  2962. char szVersion[32] = {};
  2963. int n = fread(szTemp, 1, 256, pFile);
  2964. strncpy_s(szVersion, sizeof(szVersion), szTemp, _TRUNCATE);
  2965. csVersion = szVersion;
  2966. fclose(pFile);
  2967. bRet = true;
  2968. }
  2969. else
  2970. {
  2971. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Open active.txt failed.");
  2972. bRet = false;
  2973. }
  2974. return bRet;
  2975. }
  2976. bool ResourceWatcherFSM::GetRegValue(HKEY hKey, LPCTSTR lpcszParam,
  2977. CHAR* pSzValue, DWORD* pDwSizeOfSz)
  2978. {
  2979. DWORD dwType = REG_SZ;
  2980. TCHAR szValue[MAX_PATH + 1] = { 0 };
  2981. DWORD dwSize = MAX_PATH * sizeof(CHAR);
  2982. LONG lResult = RegQueryValueEx(hKey, lpcszParam, NULL, &dwType, (LPBYTE)szValue, pDwSizeOfSz);
  2983. if (lResult == ERROR_SUCCESS) {
  2984. strcpy_s(pSzValue, *pDwSizeOfSz, szValue);
  2985. return true;
  2986. }
  2987. else {
  2988. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RegQueryValueEx for \"%s\" error, result=%ld.", lpcszParam, lResult);
  2989. return false;
  2990. }
  2991. }
  2992. void ResourceWatcherFSM::DetectAutoStartupCover()
  2993. {
  2994. int bStartupByVbs(0);
  2995. DetectAutoStartup(bStartupByVbs);
  2996. if (m_forceStartupWithExeFlag == 1 && bStartupByVbs == 1 && !m_IsPadDevice) {
  2997. ChangeAutoStartupWithExe(); //test succ
  2998. }
  2999. else if (m_forceStartupWithExeFlag == 2 && (bStartupByVbs == 0 || bStartupByVbs == 1) && !m_IsPadDevice)
  3000. {
  3001. ChangeAutoStartupWithExe(); //test succ || test succ
  3002. }
  3003. else if (m_forceStartupWithExeFlag == 3 && (bStartupByVbs == 1 || bStartupByVbs == 2))
  3004. {
  3005. ChangeAutoStartupWithExe(true); //test succ || test succ
  3006. }
  3007. }
  3008. static int Is32R64Platform()
  3009. {
  3010. static int isWow64 = -1;
  3011. typedef BOOL(WINAPI* LPFN_ISWOW64PROCESS)(HANDLE, PBOOL);
  3012. if (isWow64 == -1) {
  3013. BOOL bIsWow64 = FALSE;
  3014. LPFN_ISWOW64PROCESS fnIsWow64Process =
  3015. (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandle("kernel32"), "IsWow64Process");
  3016. if (NULL != fnIsWow64Process) {
  3017. if (!fnIsWow64Process(GetCurrentProcess(), &bIsWow64)) {
  3018. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("detect is running with 64bit or not failed: %u", GetLastError());
  3019. return -1;
  3020. }
  3021. else
  3022. {
  3023. isWow64 = bIsWow64 ? 1 : 0;
  3024. }
  3025. }
  3026. }
  3027. return isWow64;
  3028. }
  3029. void ResourceWatcherFSM::DetectAutoStartup(int& vbsType)
  3030. {
  3031. LOG_FUNCTION();
  3032. vbsType = -1;
  3033. if (m_DoneDetectAutostart) {
  3034. vbsType = -2;
  3035. return;
  3036. }
  3037. DWORD dwFlag = KEY_READ | KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS;
  3038. if (Is32R64Platform() != 0) {
  3039. dwFlag |= KEY_WOW64_64KEY;
  3040. }
  3041. HKEY hKey;
  3042. LONG lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
  3043. TEXT("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"),
  3044. 0, dwFlag, &hKey);
  3045. if (lResult == ERROR_SUCCESS) {
  3046. DWORD dwType = REG_SZ;
  3047. DWORD dwSize = MAX_PATH * sizeof(TCHAR);
  3048. TCHAR szValue[MAX_PATH + 1] = { 0 };
  3049. lResult = RegQueryValueEx(hKey, "Shell", NULL, &dwType, (LPBYTE)szValue, &dwSize);
  3050. if (lResult == ERROR_SUCCESS) {
  3051. CSimpleStringA value(szValue);
  3052. if (value.IsNullOrEmpty() || value.Compare("explorer.exe") != 0) {
  3053. if (value.IsEndWith(".vbs", true)) {
  3054. LogWarn(Severity_Low, Error_Debug, LOG_WARN_AUTO_STARTUP_FROM_REGIST, value);
  3055. vbsType = 1;
  3056. }
  3057. else if (value.IsEndWith("VTM.exe", true)) {
  3058. LogWarn(Severity_Low, Error_Debug, LOG_WARN_CHECK_AUTO_STARTUP_WITH_EXE, value);
  3059. vbsType = 2;
  3060. }
  3061. }
  3062. else {
  3063. LogWarn(Severity_Low, Error_Debug, LOG_WARN_NOT_AUTO_STARTUP_FROM_REGIST, value);
  3064. if (value.Compare("explorer.exe") == 0) {
  3065. vbsType = 0;
  3066. }
  3067. }
  3068. }
  3069. else {
  3070. LogWarn(Severity_Low, Error_Unexpect, LOG_WARN_CHECK_AUTO_STARTUP_FROM_REGIST_FAILED,
  3071. CSimpleStringA::Format("RegQueryValueEx error returned %u, GLE=%u.", lResult, GetLastError()));
  3072. }
  3073. }
  3074. else {
  3075. LogWarn(Severity_Middle, Error_Unexpect, LOG_WARN_CHECK_AUTO_STARTUP_FROM_REGIST_FAILED,
  3076. CSimpleStringA::Format("RegOpenKeyEx error, GLE=%u.", GetLastError()));
  3077. }
  3078. RegCloseKey(hKey);
  3079. m_DoneDetectAutostart = true;
  3080. return;
  3081. }
  3082. ErrorCodeEnum ResourceWatcherFSM::ChangeAutoStartupWithExe(bool resetExplorer)
  3083. {
  3084. const char* exploreExe = "explorer.exe";
  3085. ErrorCodeEnum result(Error_Succeed);
  3086. CSimpleStringA exePath(true);
  3087. if (resetExplorer) {
  3088. exePath = exploreExe;
  3089. }
  3090. else
  3091. {
  3092. result = GetEntityBase()->GetFunction()->GetPath("RootVer", exePath);
  3093. if (exePath.IsNullOrEmpty() || result != Error_Succeed) {
  3094. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Get RootVer path failed: %d", result);
  3095. return Error_Unexpect;
  3096. }
  3097. exePath.Append("\\VTM.exe");
  3098. if (!ExistsFileA(exePath)) {
  3099. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("[%s] is not existed!", exePath.GetData());
  3100. return Error_NotExist;
  3101. }
  3102. }
  3103. DWORD dwFlag = KEY_WRITE;
  3104. if (Is32R64Platform() != 0) {
  3105. dwFlag |= KEY_WOW64_64KEY;
  3106. }
  3107. HKEY hKey;
  3108. LONG lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
  3109. TEXT("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"), 0, dwFlag, &hKey);
  3110. if (lResult == ERROR_SUCCESS) {
  3111. lResult = RegSetValueExA(hKey, "Shell", 0, REG_SZ, (const BYTE*)exePath.GetData(), exePath.GetLength());
  3112. if (lResult == ERROR_SUCCESS) {
  3113. LogWarn(Severity_Low, Error_Debug, LOG_WARN_UPDATE_AUTO_STARTUP_WITH_EXE_SUCC
  3114. , CSimpleStringA::Format("RegSetValueExA for Shell with %s succ!", exePath.GetData()));
  3115. result = Error_Succeed;
  3116. }
  3117. else
  3118. {
  3119. LogWarn(Severity_Middle, Error_Unexpect, LOG_WARN_UPDATE_AUTO_STARTUP_WITH_EXE_FAILED
  3120. , CSimpleStringA::Format("RegSetValueExA for Shell with %s failed: %d", exePath.GetData(), lResult));
  3121. result = Error_Unexpect;
  3122. }
  3123. }
  3124. else
  3125. {
  3126. LogWarn(Severity_Middle, Error_Unexpect, LOG_WARN_UPDATE_AUTO_STARTUP_WITH_EXE_FAILED
  3127. , CSimpleStringA::Format("RegOpenKeyEx for write %s failed: %d", exePath.GetData(), lResult));
  3128. result = Error_Unexpect;
  3129. }
  3130. RegCloseKey(hKey);
  3131. return result;
  3132. }
  3133. void ResourceWatcherFSM::InitialAutoStartupSetType()
  3134. {
  3135. CSmartPointer<IConfigInfo> spConfig;
  3136. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spConfig);
  3137. int value(0);
  3138. spConfig->ReadConfigValueInt(GetEntityBase()->GetEntityName(), "AutoStartCMD", value);
  3139. if (value != 0) {
  3140. m_forceStartupWithExeFlag = value;
  3141. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Get AutoStartCMD from CS: %d", m_forceStartupWithExeFlag);
  3142. }
  3143. }
  3144. BOOL ResourceWatcherFSM::IsHasSetAutoStartupByFolder()
  3145. {
  3146. BOOL result(FALSE);
  3147. const char* STARTUP_DIR_FULL_PATH = "C:\\ProgramData\\Microsoft\\Windows\\Start Menu\\Programs\\StartUp";
  3148. #ifdef DEVOPS_ON_ST /*DevOps流水线编译,ST环境*/
  3149. const char* CMB_LINK_FILE_NAME = "招商银行可视柜台(ST测试).lnk";
  3150. #elif defined(DEVOPS_ON_UAT)/*DevOps流水线编译,UAT环境*/
  3151. const char* CMB_LINK_FILE_NAME = "招商银行可视柜台(UAT测试).lnk";
  3152. #elif defined(DEVOPS_ON_PRD)/*DevOps流水线编译,PRD环境*/
  3153. const char* CMB_LINK_FILE_NAME = "招商银行可视柜台.lnk";
  3154. #elif defined(DEVOPS_ON_DEV)/*DevOps流水线编译,Dev环境*/
  3155. const char* CMB_LINK_FILE_NAME = "招商银行可视柜台(开发).lnk";
  3156. #else/*本地编译等非DevOps环境编译的版本*/
  3157. const char* CMB_LINK_FILE_NAME = "招商银行可视柜台.lnk";
  3158. #endif
  3159. if (!ExistsDirA(STARTUP_DIR_FULL_PATH)) {
  3160. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("[%s] dir is not exist!", STARTUP_DIR_FULL_PATH);
  3161. return result;
  3162. }
  3163. auto arr = fileutil_get_sub_files(STARTUP_DIR_FULL_PATH);
  3164. if (arr != NULL) {
  3165. for (int i = 0; i < arr->nelts; ++i) {
  3166. char* file = ARRAY_IDX(arr, i, char*);
  3167. char* filename = strrchr(file, '\\');
  3168. if (filename != NULL) { filename += 1; }
  3169. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("autostartup[%s]", filename);
  3170. if (filename != NULL/* && strcmp(filename, CMB_LINK_FILE_NAME) == 0*/) {
  3171. CSimpleStringA strfilename(filename);
  3172. if (strfilename.IsStartWith("招商银行可视柜台") && strfilename.IsEndWith(".lnk")) {
  3173. result = TRUE;
  3174. break;
  3175. }
  3176. }
  3177. }
  3178. toolkit_array_free2(arr);
  3179. }
  3180. return result;
  3181. }
  3182. void ResourceWatcherFSM::DetectVersionHasChangedAndWarn()
  3183. {
  3184. CSmartPointer<IConfigInfo> spRunConfig;
  3185. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  3186. UINT64 utVersion = 0;
  3187. spRunConfig->ReadConfigValueHexInt("VersionMonitor", "LastVer", utVersion);
  3188. CVersion lastVer(utVersion);
  3189. if (lastVer.IsValid() && lastVer > m_RvcSysinfo.InstallVersion) {
  3190. CSimpleStringA rootVerPath;
  3191. CSimpleStringA warnInfo = CSimpleStringA::Format("version changed from %s to %s", lastVer.ToString().GetData(), m_RvcSysinfo.InstallVersion.ToString().GetData());
  3192. GetEntityBase()->GetFunction()->GetPath("RootVer", rootVerPath);
  3193. CSimpleStringA csPath = rootVerPath + "\\active.txt";
  3194. if (!ExistsFileA(csPath)) {
  3195. LogWarn(Severity_Low, Error_Debug, LOG_WARN_ACTIVE_FILE_CHANGE, CSimpleStringA::Format("%s: active.txt not exist!", warnInfo.GetData()));
  3196. }
  3197. else
  3198. {
  3199. CSimpleFileComponent activeFileInfo;
  3200. HANDLE hFile = CreateFileA(
  3201. csPath,
  3202. GENERIC_READ,
  3203. FILE_SHARE_READ,
  3204. NULL,
  3205. OPEN_EXISTING,
  3206. FILE_ATTRIBUTE_NORMAL | FILE_FLAG_BACKUP_SEMANTICS,
  3207. NULL);
  3208. if (hFile == INVALID_HANDLE_VALUE) {
  3209. LogWarn(Severity_Low, Error_Debug, LOG_WARN_ACTIVE_FILE_CHANGE, CSimpleStringA::Format("%s: active.txt open failed %u", warnInfo.GetData(), GetLastError()));
  3210. }
  3211. else
  3212. {
  3213. GetFileTime(hFile, (LPFILETIME)&activeFileInfo.mftCreate,
  3214. (LPFILETIME)&activeFileInfo.mftAccess, (LPFILETIME)&activeFileInfo.mftModified);
  3215. CloseHandle(hFile);
  3216. char szFormat1[128] = { 0 };
  3217. GetTimeFormatStr(szFormat1, 128, (FILETIME*)&activeFileInfo.mftCreate);
  3218. char szFormat2[128] = { 0 };
  3219. GetTimeFormatStr(szFormat2, 128, (FILETIME*)&activeFileInfo.mftAccess);
  3220. char szFormat3[128] = { 0 };
  3221. GetTimeFormatStr(szFormat3, 128, (FILETIME*)&activeFileInfo.mftModified);
  3222. LogWarn(Severity_Low, Error_Debug, LOG_WARN_ACTIVE_FILE_CHANGE, CSimpleStringA::Format(
  3223. "%s: active.txt create at %s, access at %s, modified at %s", warnInfo.GetData(), szFormat1, szFormat2, szFormat3));
  3224. }
  3225. }
  3226. }
  3227. if (!lastVer.IsValid() || lastVer != m_RvcSysinfo.InstallVersion) {
  3228. spRunConfig->WriteConfigValueHexInt("VersionMonitor", "LastVer", m_RvcSysinfo.InstallVersion.GetVersion64());
  3229. }
  3230. }
  3231. void ResourceWatcherFSM::DetectDestopFileAndWarn()
  3232. {
  3233. const DWORD INFO_BUFFER_SIZE = 2048;
  3234. char infoBuf[INFO_BUFFER_SIZE];
  3235. DWORD bufCharCount = INFO_BUFFER_SIZE;
  3236. char sys_disk = '\0';
  3237. bufCharCount = INFO_BUFFER_SIZE;
  3238. if (!GetComputerName(infoBuf, &bufCharCount))
  3239. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetComputerName failed, GLE=%u", GetLastError());
  3240. else DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Computer name: %s", infoBuf);
  3241. // Get and display the user name.
  3242. ZeroMemory(infoBuf, INFO_BUFFER_SIZE);
  3243. bufCharCount = INFO_BUFFER_SIZE;
  3244. if (!GetUserName(infoBuf, &bufCharCount)) {
  3245. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetUserName failed, GLE=%u", GetLastError());
  3246. }
  3247. else
  3248. {
  3249. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("User name: %s", infoBuf);
  3250. CSimpleStringA desktopFullPath = CSimpleStringA::Format("C:\\Users\\%s\\Desktop", infoBuf);
  3251. auto arr = fileutil_get_sub_files(desktopFullPath);
  3252. CSimpleStringA filelist(true);
  3253. int dirCount(0), fileCount(0);
  3254. if (arr != NULL) {
  3255. for (int i = 0; i < arr->nelts; ++i) {
  3256. char* file = ARRAY_IDX(arr, i, char*);
  3257. char* filename = strrchr(file, '\\');
  3258. if (filename != NULL) { filename += 1; }
  3259. else { filename = file; }
  3260. fileCount++;
  3261. if (!filelist.IsNullOrEmpty()) {
  3262. filelist += "|";
  3263. }
  3264. filelist += filename;
  3265. }
  3266. toolkit_array_free2(arr);
  3267. }
  3268. filelist += ";";
  3269. arr = fileutil_get_sub_dirs(desktopFullPath);
  3270. if (arr != NULL) {
  3271. for (int i = 0; i < arr->nelts; ++i) {
  3272. char* dir = ARRAY_IDX(arr, i, char*);
  3273. char* dirname = strrchr(dir, '\\');
  3274. if (dirname != NULL) { dirname += 1; }
  3275. else { dirname = dir; }
  3276. dirCount++;
  3277. if (!filelist.IsNullOrEmpty() && i != 0) {
  3278. filelist += "|";
  3279. }
  3280. filelist += dirname;
  3281. }
  3282. toolkit_array_free2(arr);
  3283. }
  3284. LogWarn(Severity_Low, Error_Debug, LOG_INFO_DESKTOP_FILESTATUS, CSimpleStringA::Format(
  3285. "%s's deskop: file(%d),dir(%d), list:(%s)", infoBuf, fileCount, dirCount,
  3286. filelist.GetLength() < 950 ? filelist.GetData() : (filelist.SubString(0, 950) + "....").GetData()));
  3287. }
  3288. }
  3289. ErrorCodeEnum ResourceWatcherFSM::DetectVTMInstalledBySetup(BOOL& fYes)
  3290. {
  3291. HKEY key;
  3292. ErrorCodeEnum result(Error_Succeed);
  3293. fYes = FALSE;
  3294. CSimpleStringA regPath = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\{123BBC1D-8555-4E90-96CA-70E678FF5C24}_is1";
  3295. LONG lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath, 0, KEY_ALL_ACCESS, &key);
  3296. if (ERROR_SUCCESS == lResult) {
  3297. fYes = TRUE;
  3298. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Found {123BBC1D-8555-4E90-96CA-70E678FF5C24}");
  3299. LogWarn(Severity_Low, Error_Debug, LOG_INFO_INSTALL_BY_SETUP, "Intalled by Setup");
  3300. }
  3301. else if (ERROR_FILE_NOT_FOUND == lResult)
  3302. {
  3303. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Not Found {123BBC1D-8555-4E90-96CA-70E678FF5C24}");
  3304. fYes = FALSE;
  3305. }
  3306. else
  3307. {
  3308. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RegOpenKeyEx For{123BBC1D-8555-4E90-96CA-70E678FF5C24} error, Result=%ld.", lResult);
  3309. fYes = FALSE;
  3310. result = Error_Unexpect;
  3311. }
  3312. RegCloseKey(key);
  3313. return result;
  3314. }
  3315. bool ResourceWatcherFSM::RegRdVtmVersion(CSimpleStringA& VTMpath, CSimpleStringA& vtmVersion)
  3316. {
  3317. bool bRet = false;
  3318. HKEY key;
  3319. CSimpleStringA regPath = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths\\VTM";
  3320. CSimpleStringA regParaPath = "path";
  3321. CSimpleStringA regParaVersion = "version";
  3322. LONG lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath, 0, KEY_ALL_ACCESS, &key);
  3323. char arrVTMpath[MAX_PATH] = { 0 }, arrVTMversion[MAX_PATH] = { 0 };
  3324. if (ERROR_SUCCESS == lResult)
  3325. {
  3326. DWORD dwValue = MAX_PATH * sizeof(CHAR);
  3327. bool res1 = GetRegValue(key, regParaPath, arrVTMpath, &dwValue);
  3328. if (res1) {
  3329. bRet = true;
  3330. VTMpath = arrVTMpath;
  3331. }
  3332. else {
  3333. bRet = false;
  3334. RegCloseKey(key);
  3335. return bRet;
  3336. }
  3337. dwValue = MAX_PATH * sizeof(CHAR);
  3338. bool res2 = GetRegValue(key, regParaVersion, arrVTMversion, &dwValue);
  3339. if (res2) {
  3340. bRet = true;
  3341. vtmVersion = arrVTMversion;
  3342. }
  3343. else {
  3344. bRet = false;
  3345. RegCloseKey(key);
  3346. return bRet;
  3347. }
  3348. }
  3349. else
  3350. {
  3351. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("%s::RegOpenKeyEx error, Result=%ld.", __FUNCTION__, lResult);
  3352. bRet = false;
  3353. }
  3354. RegCloseKey(key);
  3355. return bRet;
  3356. }
  3357. bool ResourceWatcherFSM::RegWtVtmPath()
  3358. {
  3359. bool bRet = true;
  3360. HKEY key;
  3361. CSimpleStringA regPath = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths";
  3362. if (ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath, 0, KEY_ALL_ACCESS, &key))
  3363. {
  3364. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Open Reg failed.");
  3365. bRet = false;
  3366. }
  3367. DWORD dwDisposition;
  3368. regPath += "\\VTM";
  3369. if (ERROR_SUCCESS != RegCreateKeyEx(HKEY_LOCAL_MACHINE, regPath.GetData(), 0, NULL,
  3370. REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &key, &dwDisposition))
  3371. {
  3372. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Create key failed");
  3373. bRet = false;
  3374. }
  3375. if (ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath.GetData(), 0, KEY_ALL_ACCESS, &key))
  3376. {
  3377. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Set value failed");
  3378. bRet = false;
  3379. }
  3380. CSimpleStringA para = "path";
  3381. CSimpleStringA exePath;
  3382. ErrorCodeEnum Error = GetEntityBase()->GetFunction()->GetPath("RootVer", exePath);
  3383. if (Error_Succeed == Error)
  3384. {
  3385. exePath += "\\VTM.exe";
  3386. if (ERROR_SUCCESS != RegSetValueEx(key, para.GetData(), 0, REG_SZ, (BYTE*)exePath.GetData(), MAX_PATH))
  3387. {
  3388. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Set path value failed");
  3389. bRet = false;
  3390. }
  3391. }
  3392. else
  3393. {
  3394. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Get path of VTM.exe failed.");
  3395. }
  3396. ::RegCloseKey(key);
  3397. return bRet;
  3398. }
  3399. bool ResourceWatcherFSM::RegWtVtmVersion()
  3400. {
  3401. bool bRet = true;
  3402. HKEY key;
  3403. CSimpleStringA regPath = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths";
  3404. if (ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath, 0, KEY_ALL_ACCESS, &key))
  3405. {
  3406. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Open Reg failed.");
  3407. bRet = false;
  3408. }
  3409. DWORD dwDisposition;
  3410. regPath += "\\VTM";
  3411. if (ERROR_SUCCESS != RegCreateKeyEx(HKEY_LOCAL_MACHINE, regPath.GetData(), 0, NULL,
  3412. REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &key, &dwDisposition))
  3413. {
  3414. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Create key failed");
  3415. bRet = false;
  3416. }
  3417. if (ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath.GetData(), 0, KEY_ALL_ACCESS, &key))
  3418. {
  3419. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Set value failed");
  3420. bRet = false;
  3421. }
  3422. CSimpleStringA para = "version";
  3423. CSimpleStringA csVersion;
  3424. if (GetVtmVersionFromActiveTxt(csVersion))
  3425. {
  3426. //RegSetValueEx(key, para.GetData(), 0, REG_SZ, (BYTE*)csVersion.GetData(), MAX_PATH);
  3427. if (ERROR_SUCCESS != RegSetValueEx(key, para.GetData(), 0, REG_SZ, (BYTE*)csVersion.GetData(), MAX_PATH))
  3428. {
  3429. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Set value failed");
  3430. bRet = false;
  3431. }
  3432. }
  3433. else
  3434. {
  3435. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Get version of VTM.exe failed.");
  3436. }
  3437. ::RegCloseKey(key);
  3438. return bRet;
  3439. }
  3440. bool ResourceWatcherFSM::RegOperation()
  3441. {
  3442. bool bRet = true;
  3443. CSimpleStringA csVersion, VtmRegPath, VtmRegVersion;
  3444. bool bGetActive = GetVtmVersionFromActiveTxt(csVersion);
  3445. bool bRegRead = RegRdVtmVersion(VtmRegPath, VtmRegVersion);
  3446. CSimpleStringA rootVerPath, exePath;
  3447. ErrorCodeEnum rc = GetEntityBase()->GetFunction()->GetPath("RootVer", rootVerPath);
  3448. exePath = rootVerPath + "\\VTM.exe";
  3449. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("csVersion :%s ,VtmRegVersion:%s ", csVersion.GetData(), VtmRegVersion.GetData());
  3450. if (!ExistsFileA(exePath) || !bGetActive || !bRegRead || (0 != VtmRegVersion.Compare(csVersion, true)))
  3451. {
  3452. if (!ExistsFileA(exePath) && !CopyExeToRoot(csVersion))
  3453. {
  3454. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Copy VTM.exe to version root fail.");
  3455. bRet = false;
  3456. }
  3457. else
  3458. {
  3459. if (!RegWtVtmPath())
  3460. {
  3461. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RegWtVtmPath failed.");
  3462. }
  3463. if (!RegWtVtmVersion())
  3464. {
  3465. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("RegWtVtmVersion failed.");
  3466. }
  3467. CreateLink(exePath, csVersion);
  3468. }
  3469. }
  3470. return bRet;
  3471. }
  3472. bool ResourceWatcherFSM::UpdateExe()
  3473. {
  3474. int updateExe = 0;
  3475. CSmartPointer<IConfigInfo> spCtSettingConfig;
  3476. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  3477. spCtSettingConfig->ReadConfigValueInt("NonExclusive", "UpdateExe", updateExe);
  3478. if (updateExe == 1)
  3479. return true;
  3480. return false;
  3481. }
  3482. void ResourceWatcherFSM::VerifySignature()
  3483. {
  3484. LOG_FUNCTION();
  3485. CSimpleStringA strMsg(true);
  3486. char* buffer = new char[MAX_PATH];
  3487. getcwd(buffer, MAX_PATH);
  3488. strcat(buffer, "\\*");
  3489. DWORD elapsed = 0;
  3490. DWORD dwStart = GetTickCount();
  3491. DWORD dwEnd = GetTickCount();
  3492. VerifySignature(buffer);
  3493. if (m_nonSignedFiles.size() > 0)
  3494. {
  3495. CSimpleStringA fileList = "";
  3496. for (size_t i = 0; i < m_nonSignedFiles.size(); i++)
  3497. {
  3498. fileList += GetFileName(m_nonSignedFiles[i]);
  3499. if (i < m_nonSignedFiles.size() - 1)
  3500. fileList += ",";
  3501. }
  3502. int toDisplay = m_iNonSignedTotal < m_iNonSignedDisplay ? m_iNonSignedTotal : m_iNonSignedDisplay;
  3503. strMsg = CSimpleStringA::Format("共有%d个文件未签名,其中%d个为:%s.",
  3504. m_iNonSignedTotal, toDisplay, fileList.GetData());
  3505. }
  3506. dwEnd = GetTickCount();
  3507. elapsed = dwEnd - dwStart;
  3508. strMsg = strMsg + CSimpleStringA::Format("签名耗时:%dms", elapsed);
  3509. LogWarn(Severity_Low, Error_Unexpect, LOG_WARN_FILE_NOT_TRUSTED, (LPCTSTR)strMsg);
  3510. delete buffer;
  3511. }
  3512. void ResourceWatcherFSM::VerifySignature(const CSimpleStringA& filePath)
  3513. {
  3514. HANDLE hFind = NULL;
  3515. WIN32_FIND_DATA fileInfo;
  3516. DWORD64 dwSize = 0;
  3517. hFind = FindFirstFile(filePath, &fileInfo);
  3518. if (hFind == INVALID_HANDLE_VALUE)
  3519. return;
  3520. string strDirPath = filePath;
  3521. do
  3522. {
  3523. if (fileInfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
  3524. {
  3525. if (!strcmp(fileInfo.cFileName, ".") || !strcmp(fileInfo.cFileName, ".."))
  3526. {
  3527. //Do nothing for "." and ".." folders
  3528. }
  3529. else
  3530. {
  3531. strDirPath.insert(strDirPath.find("*"), string(fileInfo.cFileName) + "\\");
  3532. VerifySignature(strDirPath.c_str());
  3533. strDirPath = filePath;
  3534. }
  3535. }
  3536. else
  3537. {
  3538. CSimpleStringA fileName = fileInfo.cFileName;
  3539. string strFileName = fileName;
  3540. string strFilePath;
  3541. if (strFileName.find(".dll") == string::npos && strFileName.find(".exe") == string::npos)
  3542. continue;
  3543. strFilePath = strDirPath;
  3544. strFilePath.replace(strFilePath.find("*"), strFilePath.length(), string(fileInfo.cFileName));
  3545. if (!RetrieveDigitalSignatureInfo(strFilePath.c_str()))
  3546. {
  3547. if (m_iNonSignedTotal < m_iNonSignedDisplay)
  3548. {
  3549. m_nonSignedFiles.push_back(strFilePath.c_str());
  3550. }
  3551. m_iNonSignedTotal++;
  3552. }
  3553. }
  3554. } while (FindNextFile(hFind, &fileInfo));
  3555. FindClose(hFind);
  3556. }
  3557. CSimpleStringA ResourceWatcherFSM::GetFileName(const CSimpleStringA& filePath)
  3558. {
  3559. CSimpleStringA tmp = filePath;
  3560. auto arrPath = tmp.Split('\\');
  3561. int size = arrPath.GetCount();
  3562. return arrPath[size - 1];
  3563. }
  3564. bool ResourceWatcherFSM::RetrieveDigitalSignatureInfo(const char* pFilePath)
  3565. {
  3566. HCERTSTORE hStore = NULL;
  3567. HCRYPTMSG hMsg = NULL;
  3568. WCHAR* wpFilePath = NULL;
  3569. int len = ::MultiByteToWideChar(CP_ACP, 0, pFilePath, -1, NULL, 0);
  3570. if (len > 0) {
  3571. wpFilePath = new wchar_t[len + 1];
  3572. std::memset(wpFilePath, 0, (len + 1) * sizeof(wchar_t));
  3573. ::MultiByteToWideChar(CP_ACP, 0, pFilePath, -1, &wpFilePath[0], len);
  3574. }
  3575. if (CryptQueryObject(CERT_QUERY_OBJECT_FILE,
  3576. wpFilePath,
  3577. CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED_EMBED,
  3578. CERT_QUERY_FORMAT_FLAG_BINARY,
  3579. 0,
  3580. NULL,
  3581. NULL,
  3582. NULL,
  3583. &hStore,
  3584. &hMsg,
  3585. NULL))
  3586. {
  3587. DWORD dwCountSigners = 0;
  3588. DWORD dwcbSz = sizeof(dwCountSigners);
  3589. if (CryptMsgGetParam(hMsg, CMSG_SIGNER_COUNT_PARAM, 0, &dwCountSigners, &dwcbSz))
  3590. {
  3591. if (dwCountSigners != 0)
  3592. {
  3593. dwcbSz = 0;
  3594. CryptMsgGetParam(hMsg, CMSG_SIGNER_INFO_PARAM, 0, NULL, &dwcbSz);
  3595. if (dwcbSz)
  3596. {
  3597. PCMSG_SIGNER_INFO pSignerInfo = (PCMSG_SIGNER_INFO)new (std::nothrow) BYTE[dwcbSz];
  3598. if (pSignerInfo)
  3599. {
  3600. DWORD dwcbSz2 = dwcbSz;
  3601. if (CryptMsgGetParam(hMsg, CMSG_SIGNER_INFO_PARAM, 0, pSignerInfo, &dwcbSz) &&
  3602. dwcbSz == dwcbSz2)
  3603. {
  3604. if (VerifyFirstCertificate(hStore, pSignerInfo)
  3605. || VerifyOtherCertificate(pSignerInfo))
  3606. return true;
  3607. }
  3608. delete[] pSignerInfo;
  3609. pSignerInfo = NULL;
  3610. }
  3611. }
  3612. }
  3613. }
  3614. }
  3615. if (hStore != NULL)
  3616. {
  3617. CertCloseStore(hStore, 0);
  3618. hStore = NULL;
  3619. }
  3620. if (hMsg != NULL)
  3621. {
  3622. CryptMsgClose(hMsg);
  3623. hMsg = NULL;
  3624. }
  3625. return false;
  3626. }
  3627. bool ResourceWatcherFSM::VerifyFirstCertificate(HCERTSTORE hStore, PCMSG_SIGNER_INFO pSignerInfo)
  3628. {
  3629. CERT_INFO ci = { 0 };
  3630. ci.Issuer = pSignerInfo->Issuer;
  3631. ci.SerialNumber = pSignerInfo->SerialNumber;
  3632. PCCERT_CONTEXT pCertContext = NULL;
  3633. int c = 0;
  3634. bool ret = false;
  3635. for (;; c++)
  3636. {
  3637. pCertContext = CertFindCertificateInStore(hStore,
  3638. ENCODING, 0, CERT_FIND_SUBJECT_CERT,
  3639. (PVOID)&ci, pCertContext);
  3640. if (!pCertContext)
  3641. {
  3642. break;
  3643. }
  3644. DWORD dwcbSz;
  3645. char* pBuff;
  3646. dwcbSz = CertGetNameString(pCertContext, CERT_NAME_SIMPLE_DISPLAY_TYPE, 0, NULL, NULL, 0);
  3647. if (dwcbSz != 0)
  3648. {
  3649. pBuff = new (std::nothrow) char[dwcbSz];
  3650. if (pBuff)
  3651. {
  3652. if (CertGetNameString(pCertContext, CERT_NAME_SIMPLE_DISPLAY_TYPE, 0, NULL, pBuff, dwcbSz) == dwcbSz)
  3653. {
  3654. if (!strcmp(pBuff, "China Merchants Bank Co., Ltd") || !strcmp(pBuff, "Microsoft Corporation"))
  3655. ret = true;
  3656. }
  3657. delete[] pBuff;
  3658. pBuff = NULL;
  3659. }
  3660. }
  3661. }
  3662. if (pCertContext)
  3663. {
  3664. CertFreeCertificateContext(pCertContext);
  3665. pCertContext = NULL;
  3666. }
  3667. return ret;
  3668. }
  3669. bool ResourceWatcherFSM::VerifyOtherCertificate(PCMSG_SIGNER_INFO pSignerInfo)
  3670. {
  3671. int ret = false;
  3672. for (DWORD a = 0; a < pSignerInfo->UnauthAttrs.cAttr; a++)
  3673. {
  3674. #ifndef szOID_NESTED_SIGNATURE
  3675. #define szOID_NESTED_SIGNATURE "1.3.6.1.4.1.311.2.4.1"
  3676. #endif
  3677. if (pSignerInfo->UnauthAttrs.rgAttr[a].pszObjId &&
  3678. lstrcmpA(pSignerInfo->UnauthAttrs.rgAttr[a].pszObjId, szOID_NESTED_SIGNATURE) == 0)
  3679. {
  3680. HCRYPTMSG hMsg = ::CryptMsgOpenToDecode(X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, 0, 0, NULL, NULL, NULL);
  3681. if (hMsg)
  3682. {
  3683. if (::CryptMsgUpdate(hMsg,
  3684. pSignerInfo->UnauthAttrs.rgAttr[a].rgValue->pbData,
  3685. pSignerInfo->UnauthAttrs.rgAttr[a].rgValue->cbData,
  3686. TRUE))
  3687. {
  3688. DWORD dwSignerInfo = 0;
  3689. ::CryptMsgGetParam(hMsg, CMSG_SIGNER_INFO_PARAM, 0, NULL, &dwSignerInfo);
  3690. if (dwSignerInfo != 0)
  3691. {
  3692. PCMSG_SIGNER_INFO pSignerInfo2 = (PCMSG_SIGNER_INFO)new (std::nothrow) BYTE[dwSignerInfo];
  3693. if (pSignerInfo2)
  3694. {
  3695. if (::CryptMsgGetParam(hMsg, CMSG_SIGNER_INFO_PARAM,
  3696. 0, (PVOID)pSignerInfo2, &dwSignerInfo))
  3697. {
  3698. CRYPT_DATA_BLOB c7Data;
  3699. c7Data.cbData = pSignerInfo->UnauthAttrs.rgAttr[a].rgValue->cbData;
  3700. c7Data.pbData = pSignerInfo->UnauthAttrs.rgAttr[a].rgValue->pbData;
  3701. //ret = FindAppropriateStoreAndVerifyFirstCertificate(pSignerInfo2, &c7Data);
  3702. HCERTSTORE hStore = NULL;
  3703. hStore = CertOpenStore(CERT_STORE_PROV_PKCS7, X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, NULL, 0, &c7Data);
  3704. ret = hStore && VerifyFirstCertificate(hStore, pSignerInfo2);
  3705. if (hStore)
  3706. {
  3707. ::CertCloseStore(hStore, CERT_CLOSE_STORE_FORCE_FLAG);
  3708. hStore = NULL;
  3709. }
  3710. }
  3711. delete[] pSignerInfo2;
  3712. pSignerInfo2 = NULL;
  3713. }
  3714. }
  3715. }
  3716. ::CryptMsgClose(hMsg);
  3717. }
  3718. }
  3719. }
  3720. return ret;
  3721. }
  3722. void ResourceWatcherFSM::ClearDumpFiles()
  3723. {
  3724. int deleteSucCnt = 0;
  3725. int deleteFailedCnt = 0;
  3726. int fileCnt = 0;
  3727. CSimpleStringA dmpPath;
  3728. ErrorCodeEnum dmpError = GetEntityBase()->GetFunction()->GetPath("Temp", dmpPath);
  3729. if (Error_Succeed != dmpError) {
  3730. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("get Dump path error");
  3731. return;
  3732. }
  3733. dmpPath += CSimpleStringA::Format("\\..\\dmp");
  3734. // =>=> /*foundFileData.cFileName[0] != '.'*/ !!! be careful for backdate call for the function.
  3735. char searchFilePath[MAX_PATH] = { 0 };
  3736. char tempFilePath[MAX_PATH] = { 0 };
  3737. char displayTime[MAX_PATH] = { 0 };
  3738. WIN32_FIND_DATA foundFileData;
  3739. HANDLE hFind;
  3740. CSmartPointer<IConfigInfo> spCtSettingConfig;
  3741. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  3742. int backDay = 0;
  3743. ErrorCodeEnum re = spCtSettingConfig->ReadConfigValueInt("ResourceWatcher", "DumpClearBackDay", backDay);
  3744. if (backDay < 7 || backDay >365 || re != Error_Succeed)
  3745. {
  3746. backDay = 7; //保留至少七天的dump文件
  3747. }
  3748. FILETIME ftRemoveTime;
  3749. CalculateBackTime(&ftRemoveTime, backDay);
  3750. GetTimeFormatStr(displayTime, MAX_PATH, &ftRemoveTime);
  3751. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("The date DUMPFILE backward to delete: %s", displayTime);
  3752. string p;
  3753. hFind = FindFirstFile(p.assign(string(dmpPath.GetData())).append("\\*").c_str(), &foundFileData);
  3754. if (hFind != INVALID_HANDLE_VALUE) {
  3755. do {
  3756. if (CSimpleStringA::Format("%s", &foundFileData.cFileName) == "." || CSimpleStringA::Format("%s", &foundFileData.cFileName) == "..")
  3757. {
  3758. continue;
  3759. }
  3760. if (CompareFileTime(&ftRemoveTime, &foundFileData.ftLastWriteTime) >= 0)
  3761. {
  3762. GetTimeFormatStr(displayTime, MAX_PATH, &foundFileData.ftLastWriteTime);
  3763. //strcpy_s(searchFilePath, (dmpPath + CSimpleStringA::Format("\\%s", &foundFileData.cFileName)).GetData());
  3764. if (DeleteFile((dmpPath + CSimpleStringA::Format("\\%s", &foundFileData.cFileName)).GetData()) != 0)
  3765. {
  3766. deleteSucCnt++;
  3767. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Delete DUMPFILE matched %s(%s) suc.", foundFileData.cFileName, displayTime);
  3768. }
  3769. else
  3770. {
  3771. deleteFailedCnt++;
  3772. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Delete DUMPFILE matched %s(%s) failed(%d).", foundFileData.cFileName, displayTime, GetLastError());
  3773. }
  3774. }
  3775. fileCnt++;
  3776. } while (FindNextFile(hFind, &foundFileData));
  3777. FindClose(hFind);
  3778. }
  3779. else
  3780. {
  3781. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM) ("FindFirstFile failed (%u)", GetLastError());
  3782. return;
  3783. }
  3784. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Total [%d] dump files, delete suc[%d], failed[%d].", fileCnt, deleteSucCnt, deleteFailedCnt);
  3785. return;
  3786. }
  3787. long long GetDirSizeFsm(string dirPath)
  3788. {
  3789. long long fileLen = 0;
  3790. long hFile = 0;
  3791. //文件信息
  3792. struct _finddata_t fileinfo;
  3793. string p;
  3794. if ((hFile = _findfirst(p.assign(dirPath).append("\\*").c_str(), &fileinfo)) != -1)
  3795. {
  3796. do
  3797. {
  3798. //如果是目录,迭代之
  3799. //如果不是,加入列表
  3800. if ((fileinfo.attrib & _A_SUBDIR))
  3801. {
  3802. if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
  3803. fileLen += GetDirSizeFsm(p.assign(dirPath).append("\\").append(fileinfo.name));
  3804. }
  3805. else
  3806. {
  3807. fileLen += fileinfo.size;
  3808. }
  3809. } while (_findnext(hFile, &fileinfo) == 0);
  3810. _findclose(hFile);
  3811. }
  3812. return fileLen;
  3813. }
  3814. struct CmpByValueFromBiggest //从最大的排起
  3815. {
  3816. bool operator()(const pair<string, long>& l, const pair<string, long>& r) {
  3817. return l.second > r.second;
  3818. }
  3819. };
  3820. //unordered_map<string, long> fileSpace;
  3821. unordered_map<string, long> fileIncreasedSpace;
  3822. vector<pair<string, long>> sortFileIncreasedSpace;
  3823. unordered_map<string, long> fileSpace;
  3824. vector<pair<string, long>> sortFileSpace;
  3825. static bool spaceLock = false;
  3826. void ResourceWatcherFSM::CheckDiskFileSpace()
  3827. {
  3828. if (spaceLock)
  3829. {
  3830. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("当前已有硬盘扫描进行中,跳过。");
  3831. return;
  3832. }
  3833. else
  3834. {
  3835. spaceLock = true; //加锁
  3836. }
  3837. CSmartPointer<IConfigInfo> spConfigRun;
  3838. CSmartPointer<IConfigInfo> spCtSettingConfig;
  3839. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  3840. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  3841. int lastDiskUsed = 0;
  3842. unsigned long long lastVer = 0;
  3843. int spaceIncrease = 0; //硬盘增长阈值
  3844. spConfigRun->ReadConfigValueHexInt("SpaceInfo", "OptVer", lastVer);
  3845. spConfigRun->ReadConfigValueInt("SpaceInfo", "DiskTotalUsed(GB)", lastDiskUsed);
  3846. spCtSettingConfig->ReadConfigValueInt(GetEntityBase()->GetEntityName(), "DiskUsedIncreaseInt(GB)", spaceIncrease);
  3847. if (spaceIncrease == 0)
  3848. {
  3849. spaceIncrease = 5; //默认阈值5GB
  3850. }
  3851. if (!m_bNeedForceDiskCheck) //是否需要强制全盘扫描 - 硬盘使用过高是会强制
  3852. {
  3853. if (lastVer == m_RvcSysinfo.InstallVersion.GetVersion64())
  3854. {
  3855. if (lastDiskUsed != 0 && ((int)(GetDiskUsedByte() / (1024 * 1024 * 1024)) - lastDiskUsed) < spaceIncrease) //增长值判定
  3856. {
  3857. //DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("硬盘空间增长未超过阈值,无需扫描。");
  3858. if (spaceLock)
  3859. {
  3860. spaceLock = false; //释放锁
  3861. }
  3862. return;
  3863. }
  3864. else
  3865. {
  3866. //降低磁盘扫描频率
  3867. SYSTEMTIME localTime;
  3868. GetLocalTime(&localTime);
  3869. if (m_diskLastWarnHour != localTime.wHour)
  3870. {
  3871. m_diskLastWarnHour = localTime.wHour;
  3872. CSmartPointer<IConfigInfo> spConfigRun;
  3873. ErrorCodeEnum eErr = GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  3874. if (eErr == Error_Succeed) {
  3875. spConfigRun->WriteConfigValueInt("WarnRecord", "disk", m_diskLastWarnHour);
  3876. }
  3877. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("硬盘增长超过阈值,重点目录扫描!");
  3878. }
  3879. else
  3880. {
  3881. //DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("磁盘增长最近1小时已扫描一次,暂不扫描。");
  3882. if (spaceLock)
  3883. {
  3884. spaceLock = false; //释放锁
  3885. }
  3886. return;
  3887. }
  3888. }
  3889. }
  3890. else
  3891. {
  3892. spConfigRun->WriteConfigValueHexInt("SpaceInfo", "OptVer", m_RvcSysinfo.InstallVersion.GetVersion64());
  3893. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("新版本,重点目录扫描");
  3894. }
  3895. }
  3896. else
  3897. {
  3898. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("强制扫描。");
  3899. }
  3900. CSystemStaticInfo sysInfo;
  3901. string path("");
  3902. GetEntityBase()->GetFunction()->GetSystemStaticInfo(sysInfo);
  3903. if (sysInfo.strMachineType == "RVC.PAD")
  3904. {
  3905. path = "C:";
  3906. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("pad设备。扫描C盘。");
  3907. }
  3908. else
  3909. {
  3910. path = "D:";
  3911. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("非pad设备。扫描D盘。");
  3912. }
  3913. if (path.size() == 0)
  3914. {
  3915. if (spaceLock)
  3916. {
  3917. spaceLock = false; //释放锁
  3918. }
  3919. return;
  3920. }
  3921. long long fileLen = 0;
  3922. long hFile = 0;
  3923. //文件信息
  3924. struct _finddata_t fileinfo;
  3925. string p;
  3926. if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1)
  3927. {
  3928. do
  3929. {
  3930. //如果是目录,迭代之
  3931. //如果不是,加入列表
  3932. if ((fileinfo.attrib & _A_SUBDIR))
  3933. {
  3934. if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
  3935. fileLen = GetDirSizeFsm(p.assign(path).append("\\").append(fileinfo.name));
  3936. else
  3937. {
  3938. continue;
  3939. }
  3940. }
  3941. else
  3942. {
  3943. fileLen = fileinfo.size;
  3944. }
  3945. CSimpleStringA resLen = CSimpleStringA::Format("%lld B", fileLen);
  3946. if (fileLen >= 1024 && fileLen < (1024 * 1024))
  3947. {
  3948. DOUBLE len = (DOUBLE)fileLen / 1024;
  3949. resLen = CSimpleStringA::Format("%.2f KB", len);
  3950. }
  3951. else if (fileLen >= (1024 * 1024) && fileLen < (1024 * 1024 * 1024))
  3952. {
  3953. DOUBLE len = (DOUBLE)fileLen / (1024 * 1024);
  3954. resLen = CSimpleStringA::Format("%.2f MB", len);
  3955. }
  3956. else if (fileLen >= (1024 * 1024 * 1024))
  3957. {
  3958. DOUBLE len = (DOUBLE)fileLen / (1024 * 1024 * 1024);
  3959. resLen = CSimpleStringA::Format("%.2f GB", len);
  3960. }
  3961. CSimpleStringA oldLenStr("");
  3962. long oldLen = 0;
  3963. spConfigRun->ReadConfigValue("SpaceInfo",
  3964. p.assign(path).append("\\").append(fileinfo.name).c_str(), oldLenStr);
  3965. oldLen = atol(oldLenStr.GetData());
  3966. fileIncreasedSpace[p.assign(path).append("\\").append(fileinfo.name)] = (long)(fileLen / (1024 * 1024)) - oldLen; //保存在unordered_map中
  3967. fileSpace[p.assign(path).append("\\").append(fileinfo.name)] = (long)(fileLen / (1024 * 1024));
  3968. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("[%s] 大小为 [%s]",
  3969. p.assign(path).append("\\").append(fileinfo.name).c_str(), resLen.GetData());
  3970. if (lastVer != m_RvcSysinfo.InstallVersion.GetVersion64())
  3971. {
  3972. spConfigRun->WriteConfigValue("SpaceInfo", p.assign(path).append("\\").append(fileinfo.name).c_str(),
  3973. CSimpleStringA::Format("%ld MB", (long)(fileLen / (1024 * 1024))));
  3974. }
  3975. } while (_findnext(hFile, &fileinfo) == 0);
  3976. _findclose(hFile);
  3977. }
  3978. //获取version目录下的文件夹大小
  3979. CSimpleStringA csPath;
  3980. ErrorCodeEnum Error = m_pEntity->GetFunction()->GetPath("RootVer", csPath); //获取当前版本路劲 例如:C:\Run
  3981. fileLen = 0;
  3982. hFile = 0;
  3983. p = "";
  3984. if ((hFile = _findfirst(p.assign(csPath.GetData()).append("\\*").c_str(), &fileinfo)) != -1)
  3985. {
  3986. do
  3987. {
  3988. //如果是目录,迭代之
  3989. //如果不是,加入列表
  3990. if ((fileinfo.attrib & _A_SUBDIR))
  3991. {
  3992. if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
  3993. fileLen = GetDirSizeFsm(p.assign(csPath.GetData()).append("\\").append(fileinfo.name));
  3994. else
  3995. {
  3996. continue;
  3997. }
  3998. }
  3999. else
  4000. {
  4001. fileLen = fileinfo.size;
  4002. }
  4003. CSimpleStringA resLen = CSimpleStringA::Format("%lld B", fileLen);
  4004. if (fileLen >= 1024 && fileLen < (1024 * 1024))
  4005. {
  4006. DOUBLE len = (DOUBLE)fileLen / 1024;
  4007. resLen = CSimpleStringA::Format("%.2f KB", len);
  4008. }
  4009. else if (fileLen >= (1024 * 1024) && fileLen < (1024 * 1024 * 1024))
  4010. {
  4011. DOUBLE len = (DOUBLE)fileLen / (1024 * 1024);
  4012. resLen = CSimpleStringA::Format("%.2f MB", len);
  4013. }
  4014. else if (fileLen >= (1024 * 1024 * 1024))
  4015. {
  4016. DOUBLE len = (DOUBLE)fileLen / (1024 * 1024 * 1024);
  4017. resLen = CSimpleStringA::Format("%.2f GB", len);
  4018. }
  4019. CSimpleStringA oldLenStr(""); //MB
  4020. long oldLen = 0;
  4021. spConfigRun->ReadConfigValue("SpaceInfo",
  4022. p.assign(csPath.GetData()).append("\\").append(fileinfo.name).c_str(), oldLenStr);
  4023. oldLen = atol(oldLenStr.GetData());
  4024. fileIncreasedSpace[p.assign(csPath.GetData()).append("\\").append(fileinfo.name)] = (long)(fileLen / (1024 * 1024)) - oldLen; //保存在unordered_map中
  4025. fileSpace[p.assign(csPath.GetData()).append("\\").append(fileinfo.name)] = (long)(fileLen / (1024 * 1024));
  4026. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("[%s] 大小为 [%s]",
  4027. p.assign(csPath.GetData()).append("\\").append(fileinfo.name).c_str(), resLen.GetData());
  4028. if (lastVer != m_RvcSysinfo.InstallVersion.GetVersion64())
  4029. {
  4030. spConfigRun->WriteConfigValue("SpaceInfo", p.assign(csPath.GetData()).append("\\").append(fileinfo.name).c_str(),
  4031. CSimpleStringA::Format("%ld MB", (long)(fileLen / (1024 * 1024))));
  4032. }
  4033. } while (_findnext(hFile, &fileinfo) == 0);
  4034. _findclose(hFile);
  4035. }
  4036. //获取rvc目录下的文件夹大小
  4037. fileLen = 0;
  4038. hFile = 0;
  4039. p = "";
  4040. if ((hFile = _findfirst(p.assign(path).append("\\rvc\\*").c_str(), &fileinfo)) != -1)
  4041. {
  4042. do
  4043. {
  4044. //如果是目录,迭代之
  4045. //如果不是,加入列表
  4046. if ((fileinfo.attrib & _A_SUBDIR))
  4047. {
  4048. if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
  4049. fileLen = GetDirSizeFsm(p.assign(path).append("\\rvc\\").append(fileinfo.name));
  4050. else
  4051. {
  4052. continue;
  4053. }
  4054. }
  4055. else
  4056. {
  4057. fileLen = fileinfo.size;
  4058. }
  4059. CSimpleStringA resLen = CSimpleStringA::Format("%lld B", fileLen);
  4060. if (fileLen >= 1024 && fileLen < (1024 * 1024))
  4061. {
  4062. DOUBLE len = (DOUBLE)fileLen / 1024;
  4063. resLen = CSimpleStringA::Format("%.2f KB", len);
  4064. }
  4065. else if (fileLen >= (1024 * 1024) && fileLen < (1024 * 1024 * 1024))
  4066. {
  4067. DOUBLE len = (DOUBLE)fileLen / (1024 * 1024);
  4068. resLen = CSimpleStringA::Format("%.2f MB", len);
  4069. }
  4070. else if (fileLen >= (1024 * 1024 * 1024))
  4071. {
  4072. DOUBLE len = (DOUBLE)fileLen / (1024 * 1024 * 1024);
  4073. resLen = CSimpleStringA::Format("%.2f GB", len);
  4074. }
  4075. CSimpleStringA oldLenStr("");
  4076. long oldLen = 0;
  4077. spConfigRun->ReadConfigValue("SpaceInfo",
  4078. p.assign(path).append("\\rvc\\").append(fileinfo.name).c_str(), oldLenStr);
  4079. oldLen = atol(oldLenStr.GetData());
  4080. fileIncreasedSpace[p.assign(path).append("\\rvc\\").append(fileinfo.name)] = (long)(fileLen / (1024 * 1024)) - oldLen; //保存在unordered_map中
  4081. fileSpace[p.assign(path).append("\\rvc\\").append(fileinfo.name)] = (long)(fileLen / (1024 * 1024));
  4082. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("[%s] 大小为 [%s]",
  4083. p.assign(path).append("\\rvc\\").append(fileinfo.name).c_str(), resLen.GetData());
  4084. if (lastVer != m_RvcSysinfo.InstallVersion.GetVersion64())
  4085. {
  4086. spConfigRun->WriteConfigValue("SpaceInfo", p.assign(path).append("\\rvc\\").append(fileinfo.name).c_str(),
  4087. CSimpleStringA::Format("%ld MB", (long)(fileLen / (1024 * 1024))));
  4088. }
  4089. } while (_findnext(hFile, &fileinfo) == 0);
  4090. _findclose(hFile);
  4091. }
  4092. if (lastVer != m_RvcSysinfo.InstallVersion.GetVersion64())
  4093. {
  4094. //写入硬盘量
  4095. spConfigRun->WriteConfigValueInt("SpaceInfo", "DiskTotalUsed(GB)", GetDiskUsedByte() / (1024 * 1024 * 1024));
  4096. if (spaceLock)
  4097. {
  4098. spaceLock = false; //释放锁
  4099. }
  4100. return; //新版本无需上报文件信息,更新完运行时文件后即可结束
  4101. }
  4102. //获取增量最大的前十文件夹信息
  4103. sortFileIncreasedSpace = vector<pair<string, long>>(fileIncreasedSpace.begin(), fileIncreasedSpace.end());
  4104. sort(sortFileIncreasedSpace.begin(), sortFileIncreasedSpace.end(), CmpByValueFromBiggest());
  4105. int topFileNum = sortFileIncreasedSpace.size();
  4106. if (topFileNum > 10)
  4107. {
  4108. topFileNum = 10;
  4109. }
  4110. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("变化最大的%d个文件夹信息。", topFileNum);
  4111. fileLen = 0;
  4112. for (int i = 0; i < topFileNum; i++)
  4113. {
  4114. fileLen = sortFileIncreasedSpace[i].second;
  4115. if (fileLen == 0) break;
  4116. CSimpleStringA warn1 = CSimpleStringA::Format("TOP %d: [%s], 增量[%ld]MB",
  4117. i + 1, sortFileIncreasedSpace[i].first.c_str(), sortFileIncreasedSpace[i].second);
  4118. LogWarn(Severity_Low, Error_Resource, LOG_RESOURCEWATCHER_FILE_TOPINCREASED_SPACE, warn1.GetData());
  4119. }
  4120. if (m_bNeedForceDiskCheck)
  4121. {
  4122. //获取空间占用最大前十文件夹信息
  4123. sortFileSpace = vector<pair<string, long>>(fileSpace.begin(), fileSpace.end());
  4124. sort(sortFileSpace.begin(), sortFileSpace.end(), CmpByValueFromBiggest());
  4125. int topFileSize = sortFileSpace.size();
  4126. if (topFileSize > 10)
  4127. {
  4128. topFileSize = 10;
  4129. }
  4130. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("空间占用最大的%d个文件夹信息。", topFileSize);
  4131. fileLen = 0;
  4132. for (int i = 0; i < topFileSize; i++)
  4133. {
  4134. fileLen = sortFileSpace[i].second;
  4135. if (fileLen == 0) break;
  4136. CSimpleStringA warn2 = CSimpleStringA::Format("空间占用排名 %d: [%s], 大小[%ld]MB",
  4137. i + 1, sortFileSpace[i].first.c_str(), sortFileSpace[i].second);
  4138. LogWarn(Severity_Low, Error_Resource, LOG_RESOURCEWATCHER_FILE_TOPUSED_SPACE, warn2.GetData());
  4139. }
  4140. m_bNeedForceDiskCheck = false;
  4141. }
  4142. if (spaceLock)
  4143. {
  4144. spaceLock = false; //释放锁
  4145. }
  4146. }
  4147. void ResourceWatcherFSM::GetSystemDiskStatus()
  4148. {
  4149. if (m_diskHighPercent <= 0)
  4150. {
  4151. m_diskHighPercent = 90; //默认硬盘告警百分比90%
  4152. }
  4153. if (m_RvcSysinfo.strMachineType.Compare("RVC.PAD") != 0) //非pad设备扫描一下C盘
  4154. {
  4155. ULARGE_INTEGER ulAvailFreeTemp, ulTotalBytesTemp, ulTotalFreeTemp;
  4156. BOOL retTemp = GetDiskFreeSpaceEx(TEXT("c:"), &ulAvailFreeTemp, &ulTotalBytesTemp, &ulTotalFreeTemp);
  4157. if (retTemp == 0)
  4158. {
  4159. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetSystemDiskStatus.GetDiskFreeSpaceEx failed(%d).", GetLastError());
  4160. return;
  4161. }
  4162. DWORD dwAvFreeTemp = ulAvailFreeTemp.QuadPart / MILLION;
  4163. DWORD dwTotalTemp = ulTotalBytesTemp.QuadPart / MILLION;
  4164. DWORD dwTotalFreeTemp = ulTotalFreeTemp.QuadPart / MILLION;
  4165. int ratioTemp = dwTotalFreeTemp * 100 / dwTotalTemp;
  4166. if ((100 - ratioTemp) > m_diskHighPercent)
  4167. {
  4168. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("The disk has %d MB(%d%%) available.\n", dwTotalFreeTemp, ratioTemp);
  4169. char diskMessageTemp[1024];
  4170. sprintf(diskMessageTemp, "{\"Disk\":\"C:\",\"total\":\"%0.2f GByte\",\"free ratio\":\"%d%%\"}",
  4171. (double)dwTotalTemp / 1024, ratioTemp);
  4172. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_HARDDISK_TOO_HIGH, diskMessageTemp);
  4173. }
  4174. }
  4175. ULARGE_INTEGER ulAvailFree, ulTotalBytes, ulTotalFree;
  4176. BOOL ret = GetDiskFreeSpaceEx(NULL, &ulAvailFree, &ulTotalBytes, &ulTotalFree);
  4177. if (ret == 0)
  4178. {
  4179. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetSystemDiskStatus.GetDiskFreeSpaceEx failed(%d).", GetLastError());
  4180. return;
  4181. }
  4182. CSimpleStringA csBinPath;
  4183. ErrorCodeEnum Error = m_pEntity->GetFunction()->GetPath("Bin", csBinPath); //获取当前终端运行磁盘位置
  4184. DWORD dwAvFree = ulAvailFree.QuadPart / MILLION;
  4185. DWORD dwTotal = ulTotalBytes.QuadPart / MILLION;
  4186. DWORD dwTotalFree = ulTotalFree.QuadPart / MILLION;
  4187. int ratio = dwTotalFree * 100 / dwTotal;
  4188. if ((100 - ratio) > m_diskHighPercent)
  4189. {
  4190. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("The disk has %d MB(%d%%) available.\n", dwTotalFree, ratio);
  4191. string test(csBinPath.GetData());
  4192. char diskMessage[1024];
  4193. sprintf(diskMessage, "{\"Disk\":\"%c:\",\"total\":\"%0.2f GByte\",\"free ratio\":\"%d%%\"}",
  4194. csBinPath.GetData()[0], (double)dwTotal / 1024, ratio);
  4195. //oilyang@20200526 根据wq建议,降低磁盘空间偏少的告警频率
  4196. //SYSTEMTIME localTime;
  4197. //GetLocalTime(&localTime);
  4198. //if (m_diskLastWarnHour != localTime.wHour) //需要告警时才强制全盘扫描
  4199. //{
  4200. // m_diskLastWarnHour = localTime.wHour;
  4201. // CSmartPointer<IConfigInfo> spConfigRun;
  4202. // ErrorCodeEnum eErr = GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spConfigRun);
  4203. // if (eErr == Error_Succeed) {
  4204. // spConfigRun->WriteConfigValueInt("WarnRecord", "disk", m_diskLastWarnHour);
  4205. // }
  4206. // //LogWarn(Severity_Low, Error_Resource, LOG_EVT_SELFCHECK_HARDDISK_TOO_HIGH, "Harddisk free space is few.");
  4207. // LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_HARDDISK_TOO_HIGH, diskMessage);
  4208. // m_bNeedForceDiskCheck = true;
  4209. //}
  4210. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_HARDDISK_TOO_HIGH, diskMessage);
  4211. m_bNeedForceDiskCheck = true;
  4212. }
  4213. else
  4214. {
  4215. m_bNeedForceDiskCheck = false;
  4216. }
  4217. CheckDiskFileSpaceTask* task4 = new CheckDiskFileSpaceTask(this);
  4218. GetEntityBase()->GetFunction()->PostThreadPoolTask(task4);
  4219. }
  4220. ULONGLONG ResourceWatcherFSM::GetDiskUsedByte()
  4221. {
  4222. ULARGE_INTEGER ulAvailFree, ulTotalBytes, ulTotalFree;
  4223. BOOL ret = GetDiskFreeSpaceEx(NULL, &ulAvailFree, &ulTotalBytes, &ulTotalFree);
  4224. if (ret == 0)
  4225. {
  4226. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("GetSystemDiskStatus.GetDiskFreeSpaceEx failed(%d).", GetLastError());
  4227. return -1;
  4228. }
  4229. CSimpleStringA csBinPath;
  4230. ErrorCodeEnum Error = m_pEntity->GetFunction()->GetPath("Bin", csBinPath); //获取当前终端运行磁盘位置
  4231. return ulTotalBytes.QuadPart - ulTotalFree.QuadPart;
  4232. }
  4233. __int64 Filetime2Int64(const FILETIME& ftime)
  4234. {
  4235. LARGE_INTEGER li;
  4236. li.LowPart = ftime.dwLowDateTime;
  4237. li.HighPart = ftime.dwHighDateTime;
  4238. return li.QuadPart;
  4239. }
  4240. __int64 CompareFileTime2(const FILETIME& preTime, const FILETIME& nowTime)
  4241. {
  4242. return Filetime2Int64(nowTime) - Filetime2Int64(preTime);
  4243. }
  4244. unordered_map<int, long long> oldProcessTime;
  4245. unordered_map<int, long long> newProcessTime;
  4246. unordered_map<int, CSimpleStringA> processName;
  4247. unordered_map<int, double> processCpu;
  4248. bool cmp(const pair<int, double>& a, const pair<int, double>& b) {
  4249. return a.second > b.second;
  4250. }
  4251. void ResourceWatcherFSM::GetSystemCPUStatus()
  4252. {
  4253. FILETIME preIdleTime;
  4254. FILETIME preKernelTime;
  4255. FILETIME preUserTime;
  4256. FILETIME preProcCreateTime;
  4257. FILETIME preProcExitTime;
  4258. FILETIME preProcKernelTime;
  4259. FILETIME preProcUserTime;
  4260. GetSystemTimes(&preIdleTime, &preKernelTime, &preUserTime);
  4261. PROCESSENTRY32 pe32Old;
  4262. pe32Old.dwSize = sizeof(pe32Old);
  4263. //获得系统进程快照的句柄
  4264. HANDLE hProcessSnapOld = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  4265. if (hProcessSnapOld == INVALID_HANDLE_VALUE)
  4266. {
  4267. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("CreateToolhelp32Snapshot error.");
  4268. return;
  4269. }
  4270. //首先获得第一个进程
  4271. BOOL bProcessOld = Process32First(hProcessSnapOld, &pe32Old);
  4272. //循环获得所有进程
  4273. while (bProcessOld)
  4274. {
  4275. //进程名和进程ID
  4276. HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, false, pe32Old.th32ProcessID);
  4277. GetProcessTimes(hProcess, &preProcCreateTime, &preProcExitTime, &preProcKernelTime, &preProcUserTime);
  4278. oldProcessTime[pe32Old.th32ProcessID] = Filetime2Int64(preProcKernelTime) + Filetime2Int64(preProcUserTime);
  4279. bProcessOld = Process32Next(hProcessSnapOld, &pe32Old);
  4280. }
  4281. CloseHandle(hProcessSnapOld);
  4282. Sleep(1000);
  4283. FILETIME idleTime;
  4284. FILETIME kernelTime;
  4285. FILETIME userTime;
  4286. FILETIME ProcCreateTime;
  4287. FILETIME ProcExitTime;
  4288. FILETIME procKernelTime;
  4289. FILETIME procUserTime;
  4290. GetSystemTimes(&idleTime, &kernelTime, &userTime);
  4291. PROCESSENTRY32 pe32New;
  4292. pe32New.dwSize = sizeof(pe32New);
  4293. //获得系统进程快照的句柄
  4294. HANDLE hProcessSnapNew = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  4295. if (hProcessSnapNew == INVALID_HANDLE_VALUE)
  4296. {
  4297. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("CreateToolhelp32Snapshot error.");
  4298. return;
  4299. }
  4300. //首先获得第一个进程
  4301. BOOL bProcessNew = Process32First(hProcessSnapNew, &pe32New);
  4302. //循环获得所有进程
  4303. while (bProcessNew)
  4304. {
  4305. //进程名和进程ID
  4306. HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, false, pe32New.th32ProcessID);
  4307. GetProcessTimes(hProcess, &ProcCreateTime, &ProcExitTime, &procKernelTime, &procUserTime);
  4308. newProcessTime[pe32New.th32ProcessID] = Filetime2Int64(procKernelTime) + Filetime2Int64(procUserTime);
  4309. processName[pe32New.th32ProcessID] = CSimpleStringA(pe32New.szExeFile);
  4310. bProcessNew = Process32Next(hProcessSnapNew, &pe32New);
  4311. }
  4312. CloseHandle(hProcessSnapNew);
  4313. auto idle = CompareFileTime2(preIdleTime, idleTime);
  4314. auto kernel = CompareFileTime2(preKernelTime, kernelTime);
  4315. auto user = CompareFileTime2(preUserTime, userTime);
  4316. if (kernel + user == 0)
  4317. return;
  4318. CSystemRunInfo runInfo = { 0 };
  4319. GetEntityBase()->GetFunction()->GetSystemRunInfo(runInfo);
  4320. for (int i = 0; i < runInfo.strRunningEntityNames.GetCount(); i++) //实体进程的名称默认全为sphost,需转换为对应实体名
  4321. {
  4322. runInfo.strRunningEntityNames[i];
  4323. CEntityRunInfo entityInfo = { 0 };
  4324. GetEntityBase()->GetFunction()->GetEntityRunInfo(runInfo.strRunningEntityNames[i].GetData(), entityInfo);
  4325. processName[entityInfo.dwProcessID] = runInfo.strRunningEntityNames[i];
  4326. }
  4327. unordered_map<int, long long>::iterator it;
  4328. for (it = newProcessTime.begin(); it != newProcessTime.end(); ++it)
  4329. {
  4330. int tPid = it->first;
  4331. //进程的占用率 = 单位时间间隔里进程的CPU时间片占用 / 单位时间间隔里CPU的整体时间片
  4332. if (oldProcessTime.find(tPid) == oldProcessTime.end())
  4333. {
  4334. continue;
  4335. }
  4336. double tRatio = 100.0 * (newProcessTime[tPid] - oldProcessTime[tPid]) / (kernel + user);
  4337. if (tRatio > 0.0)
  4338. {
  4339. processCpu[tPid] = tRatio;
  4340. }
  4341. }
  4342. vector<pair<int, double>> vec(processCpu.begin(), processCpu.end());
  4343. sort(vec.begin(), vec.end(), cmp); //根据CPU使用率从大到小进行排序
  4344. int len = min(10, vec.size());
  4345. CSimpleStringA procWarn = "";
  4346. for (int i = 0; i < len; ++i) //构建进程的CPU使用告警信息
  4347. {
  4348. procWarn = procWarn + ",\"" + processName[vec[i].first] + "\":\"" + CSimpleStringA::Format("%.2f", vec[i].second) + "%\"";
  4349. }
  4350. double cpuRatio = 100.0 * (kernel + user - idle) / (kernel + user);
  4351. if (m_cpuHighPercent <= 0)
  4352. {
  4353. m_cpuHighPercent = 80; //默认告警百分比80%
  4354. }
  4355. if (cpuWarnThreshold <= 0)
  4356. {
  4357. cpuWarnThreshold = 10; //默认满10次告一次警
  4358. }
  4359. if (cpuRatio > (double)m_cpuHighPercent)
  4360. {
  4361. cpuWarnTime = cpuWarnTime % cpuWarnThreshold;
  4362. if (cpuWarnTime == 0) //每达到一次阈值告警一次
  4363. {
  4364. CSimpleStringA warn = CSimpleStringA::Format("{\"type\":\"cpu\", \"total_used\":\" %.2f", cpuRatio);
  4365. warn = warn + "%\"" + procWarn + "}";
  4366. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_CPU_TOO_HIGH, warn.GetData());
  4367. if (cpuRatio > 98) //cpu极端异常情况
  4368. {
  4369. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_CPU_ERROR, warn.GetData());
  4370. }
  4371. }
  4372. cpuWarnTime++;
  4373. }
  4374. }
  4375. unsigned long long convert_time_format(const FILETIME* ftime)
  4376. {
  4377. LARGE_INTEGER li;
  4378. li.LowPart = ftime->dwLowDateTime;
  4379. li.HighPart = ftime->dwHighDateTime;
  4380. return li.QuadPart;
  4381. }
  4382. /// 时间转换
  4383. static uint64_t file_time_2_utc(const FILETIME* ftime)
  4384. {
  4385. LARGE_INTEGER li;
  4386. li.LowPart = ftime->dwLowDateTime;
  4387. li.HighPart = ftime->dwHighDateTime;
  4388. return li.QuadPart;
  4389. }
  4390. /// 获得CPU的核数
  4391. static int get_processor_number()
  4392. {
  4393. SYSTEM_INFO info;
  4394. GetSystemInfo(&info);
  4395. return (int)info.dwNumberOfProcessors;
  4396. }
  4397. int ResourceWatcherFSM::GetCpuUsageRatio(int pid)
  4398. {
  4399. //cpu数量
  4400. static int processor_count_ = -1;
  4401. //上一次的时间
  4402. static int64_t last_time_ = 0;
  4403. static int64_t last_system_time_ = 0;
  4404. FILETIME now;
  4405. FILETIME creation_time;
  4406. FILETIME exit_time;
  4407. FILETIME kernel_time;
  4408. FILETIME user_time;
  4409. int64_t system_time;
  4410. int64_t time;
  4411. int64_t system_time_delta;
  4412. int64_t time_delta;
  4413. int cpu = -1;
  4414. if (processor_count_ == -1)
  4415. {
  4416. processor_count_ = get_processor_number();
  4417. }
  4418. GetSystemTimeAsFileTime(&now);
  4419. HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, false, pid);
  4420. if (!GetProcessTimes(hProcess, &creation_time, &exit_time, &kernel_time, &user_time))
  4421. {
  4422. return -1;
  4423. }
  4424. system_time = (file_time_2_utc(&kernel_time) + file_time_2_utc(&user_time))
  4425. / processor_count_;
  4426. time = file_time_2_utc(&now);
  4427. if ((last_system_time_ == 0) || (last_time_ == 0))
  4428. {
  4429. last_system_time_ = system_time;
  4430. last_time_ = time;
  4431. return -1;
  4432. }
  4433. system_time_delta = system_time - last_system_time_;
  4434. time_delta = time - last_time_;
  4435. if (time_delta == 0)
  4436. return -1;
  4437. cpu = (int)((system_time_delta * 100 + time_delta / 2) / time_delta);
  4438. last_system_time_ = system_time;
  4439. last_time_ = time;
  4440. return cpu;
  4441. }
  4442. double GetMemoryUsageGB(int pid)
  4443. {
  4444. #ifdef WIN32
  4445. DWORDLONG mem = 0, vmem = 0;
  4446. PROCESS_MEMORY_COUNTERS pmc;
  4447. // get process hanlde by pid
  4448. HANDLE process = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);
  4449. if (GetProcessMemoryInfo(process, &pmc, sizeof(pmc)))
  4450. {
  4451. mem = pmc.WorkingSetSize;
  4452. vmem = pmc.PagefileUsage;
  4453. }
  4454. CloseHandle(process);
  4455. // use GetCurrentProcess() can get current process and no need to close handle
  4456. // convert mem from B to GB
  4457. double memGB = mem / (1024.0 * 1024.0 * 1024.0);
  4458. return memGB;
  4459. #else
  4460. char file_name[64] = { 0 };
  4461. FILE* fd;
  4462. char line_buff[512] = { 0 };
  4463. sprintf(file_name, "/proc/%d/status", pid);
  4464. fd = fopen(file_name, "r");
  4465. if (nullptr == fd)
  4466. return 0;
  4467. char name[64];
  4468. int vmrss = 0;
  4469. for (int i = 0; i < VMRSS_LINE - 1; i++)
  4470. fgets(line_buff, sizeof(line_buff), fd);
  4471. fgets(line_buff, sizeof(line_buff), fd);
  4472. sscanf(line_buff, "%s %d", name, &vmrss);
  4473. fclose(fd);
  4474. // cnvert VmRSS from KB to MB
  4475. return vmrss / 1024.0;
  4476. #endif
  4477. }
  4478. unordered_map<int, double> processMem;
  4479. void ResourceWatcherFSM::GetSystemMemoryStatus()
  4480. {
  4481. // Specify the width of the field in which to print the numbers.
  4482. // The asterisk in the format specifier "%*I64d" takes an integer
  4483. // argument and uses it to pad and right justify the number.
  4484. #define WIDTH 7
  4485. MEMORYSTATUSEX statex;
  4486. statex.dwLength = sizeof(statex);
  4487. GlobalMemoryStatusEx(&statex);
  4488. double totalMem = statex.ullTotalPhys / (1024.0 * 1024.0 * 1024.0); //GB
  4489. unordered_map<int, CSimpleStringA>::iterator it;
  4490. for (it = processName.begin(); it != processName.end(); ++it)
  4491. {
  4492. int tPid = it->first;
  4493. double tmp = GetMemoryUsageGB(it->first);
  4494. if (tmp > 0.0)
  4495. {
  4496. processMem[tPid] = 100 * (tmp / totalMem);
  4497. }
  4498. }
  4499. //return;
  4500. vector<pair<int, double>> vec(processMem.begin(), processMem.end());
  4501. sort(vec.begin(), vec.end(), cmp); //根据CPU使用率从大到小进行排序
  4502. int len = min(10, vec.size());
  4503. CSimpleStringA procWarn = "";
  4504. for (int i = 0; i < len; ++i) //构建进程的CPU使用告警信息
  4505. {
  4506. procWarn = procWarn + ",\"" + processName[vec[i].first] + "\":\"" + CSimpleStringA::Format("%.1f", vec[i].second) + "%\"";
  4507. }
  4508. if (m_memHighPercent <= 0)
  4509. {
  4510. m_memHighPercent = 80; //默认告警百分比80%
  4511. }
  4512. if (memWarnThreshold <= 0)
  4513. {
  4514. memWarnThreshold = 10; //默认告警阈值10次
  4515. }
  4516. if (statex.dwMemoryLoad > m_memHighPercent)
  4517. {
  4518. memWarnTime = memWarnTime % memWarnThreshold;
  4519. if (memWarnTime == 0) //每达到阈值告警一次
  4520. {
  4521. CSimpleStringA warn = CSimpleStringA::Format("{\"type\":\"mem\", \"total\":\"%.1f GByte\",\"total_used\":\" %d%%\"", totalMem, statex.dwMemoryLoad);
  4522. warn = warn + procWarn + "}";
  4523. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_MEMORY_TOO_HIGH, warn.GetData());
  4524. }
  4525. memWarnTime++;
  4526. }
  4527. }
  4528. int ResourceWatcherFSM::IsWifiConnected()
  4529. {
  4530. DWORD service_ver;
  4531. HANDLE handle = NULL;
  4532. int result = -1;
  4533. DWORD err = WlanOpenHandle(WLAN_API_VERSION, NULL, &service_ver, &handle);
  4534. if (err != 0) {
  4535. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("open wlan handle failed %d", err);
  4536. return result;
  4537. }
  4538. PWLAN_INTERFACE_INFO_LIST info_list = NULL;
  4539. PWLAN_INTERFACE_INFO interface_info = NULL;
  4540. err = WlanEnumInterfaces(handle, NULL, &info_list);
  4541. if (err != ERROR_SUCCESS) {
  4542. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("Enum wlan interface failed %d", err);
  4543. WlanCloseHandle(handle, NULL);
  4544. return -1;
  4545. }
  4546. const int count = info_list->dwNumberOfItems;
  4547. for (int i = 0; i < count; ++i) {
  4548. interface_info = &info_list->InterfaceInfo[i];
  4549. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("wireless interface name: %s", SP::Module::Util::W2S(interface_info->strInterfaceDescription).c_str());
  4550. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("state: %d", interface_info->isState);
  4551. if (interface_info->isState == wlan_interface_state_connected) {
  4552. result = 1;
  4553. }
  4554. DWORD radio_state_info_size = sizeof(WLAN_RADIO_STATE);
  4555. PWLAN_RADIO_STATE radio_state_info_ptr = NULL;
  4556. WLAN_OPCODE_VALUE_TYPE op_code = wlan_opcode_value_type_invalid;
  4557. err = WlanQueryInterface(handle, &interface_info->InterfaceGuid, wlan_intf_opcode_radio_state,
  4558. NULL, &radio_state_info_size, (PVOID*)&radio_state_info_ptr, &op_code);
  4559. if (err != ERROR_SUCCESS) {
  4560. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("WlanQueryInterface failed %d", err);
  4561. }
  4562. else
  4563. {
  4564. const uint32_t count_of_phys = radio_state_info_ptr->dwNumberOfPhys;
  4565. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("Number of valid PHYs: %d", count_of_phys);
  4566. if (count_of_phys > 0 && result == -1) {
  4567. if (dot11_radio_state_on == radio_state_info_ptr->PhyRadioState[0].dot11SoftwareRadioState) {
  4568. result = 0;
  4569. }
  4570. else
  4571. {
  4572. /*
  4573. for (int j = 0; j < count_of_phys; ++j) {
  4574. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("software radio state: %d, %d", j , radio_state_info_ptr->PhyRadioState[j].dot11SoftwareRadioState);
  4575. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("hardware radio state: %d, %d", j, radio_state_info_ptr->PhyRadioState[j].dot11HardwareRadioState);
  4576. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("**********************");
  4577. }
  4578. */
  4579. }
  4580. }
  4581. }
  4582. }
  4583. if (info_list != NULL) {
  4584. WlanFreeMemory(info_list);
  4585. }
  4586. WlanCloseHandle(handle, NULL);
  4587. return result;
  4588. }
  4589. #endif // _MSC_VER end硬件资源监控相关系统功能 CPU、内存、硬盘、网络wifi
  4590. #ifdef RVC_OS_LINUX
  4591. static bool spaceLock = false;
  4592. void ResourceWatcherFSM::CheckDiskFileSpace()
  4593. {
  4594. if (spaceLock)
  4595. {
  4596. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("当前已有硬盘扫描进行中,跳过。");
  4597. return;
  4598. }
  4599. else
  4600. {
  4601. spaceLock = true; //加锁
  4602. }
  4603. //获取opt文件夹下的文件占用信息
  4604. FILE* fp = NULL;
  4605. CSimpleStringA cmd = "du -h -s /opt/* | sort -rn | head";
  4606. if ((fp = popen(cmd.GetData(), "r")) == NULL) {
  4607. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("popen failed:%d", GetLastError());
  4608. return;
  4609. }
  4610. char buf[1024] = { 0 };
  4611. CSimpleStringA warnMsg(true);
  4612. while (fgets(buf, sizeof(buf), fp))
  4613. {
  4614. warnMsg = warnMsg + "|" + buf;
  4615. }
  4616. if (warnMsg.GetLength() > 0)
  4617. {
  4618. LogWarn(Severity_Low, Error_Resource, LOG_RESOURCEWATCHER_FILE_TOPUSED_SPACE, warnMsg.GetData()); //告警opt下的文件夹
  4619. }
  4620. pclose(fp);
  4621. //获取opt/rvc文件夹下的文件占用信息
  4622. FILE* fp2 = NULL;
  4623. CSimpleStringA cmd2 = "du -h -s /opt/rvc/* | sort -rn | head";
  4624. if ((fp2 = popen(cmd2.GetData(), "r")) == NULL) {
  4625. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("popen failed:%d", GetLastError());
  4626. return;
  4627. }
  4628. CSimpleStringA warnMsg2(true);
  4629. char buf2[1024] = { 0 };
  4630. while (fgets(buf2, sizeof(buf2), fp2))
  4631. {
  4632. warnMsg2 = warnMsg2 + "|" + buf2;
  4633. }
  4634. if (warnMsg2.GetLength() > 0)
  4635. {
  4636. LogWarn(Severity_Low, Error_Resource, LOG_RESOURCEWATCHER_FILE_TOPUSED_SPACE, warnMsg2.GetData()); //告警opt/rvc下的文件夹
  4637. }
  4638. pclose(fp2);
  4639. if (spaceLock)
  4640. {
  4641. spaceLock = false; //释放锁
  4642. }
  4643. }
  4644. #endif // RVC_OS_LINUX
  4645. ErrorCodeEnum ResourceWatcherFSM::CatchSystemBasicInfo(SystemBasicInfo& info)
  4646. {
  4647. #if defined(RVC_OS_WIN)
  4648. ///**TODO(Gifur@10/10/2023): 待实现 */
  4649. return Error_NotImpl;
  4650. #else
  4651. CSimpleStringA runCfgPath;
  4652. ErrorCodeEnum errCode = GetEntityBase()->GetFunction()->GetPath("RunCfg", runCfgPath);
  4653. if (runCfgPath.IsNullOrEmpty()) return Error_Null;
  4654. CSimpleStringA storeFilePath = runCfgPath + SPLIT_SLASH_STR + "SMBIOSData.txt";
  4655. if (!ExistsFileA(storeFilePath)) {
  4656. RemoveFileA(storeFilePath);
  4657. }
  4658. std::string succStr, errStr;
  4659. std::string runStr("dmidecode -t system > ");
  4660. runStr += storeFilePath.GetData();
  4661. if (!SP::Module::Util::ShellExecute(runStr, succStr, errStr)) {
  4662. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("execute '%s' failed!", runStr.c_str());
  4663. if (!ExistsFileA(storeFilePath)) {
  4664. RemoveFileA(storeFilePath);
  4665. }
  4666. return Error_Unexpect;
  4667. }
  4668. do {
  4669. std::string succStr, errStr;
  4670. std::string runStr("awk -F ':' '(NR>=7)&&(NR<=14)&&($0~\"Manufacturer\"){print$2}' ");
  4671. runStr += storeFilePath.GetData();
  4672. if (SP::Module::Util::ShellExecute(runStr, succStr, errStr)) {
  4673. info.strManufacturer = SP::Utility::ToTrim(succStr).c_str();
  4674. }
  4675. else
  4676. {
  4677. succStr = "";
  4678. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("fetch 'Manufacturer' failed!");
  4679. }
  4680. } while (false);
  4681. do {
  4682. std::string succStr, errStr;
  4683. std::string runStr("awk -F ':' '(NR>=7)&&(NR<=14)&&($0~\"Product Name\"){print$2}' ");
  4684. runStr += storeFilePath.GetData();
  4685. if (SP::Module::Util::ShellExecute(runStr, succStr, errStr)) {
  4686. info.strProductName = SP::Utility::ToTrim(succStr).c_str();
  4687. }
  4688. else
  4689. {
  4690. succStr = "";
  4691. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("fetch 'Product Name' failed!");
  4692. }
  4693. } while (false);
  4694. do {
  4695. std::string succStr, errStr;
  4696. std::string runStr("awk -F ':' '(NR>=7)&&(NR<=14)&&($0~\"Serial Number\"){print$2}' ");
  4697. runStr += storeFilePath.GetData();
  4698. if (SP::Module::Util::ShellExecute(runStr, succStr, errStr)) {
  4699. info.strSerialNumber = SP::Utility::ToTrim(succStr).c_str();
  4700. }
  4701. else
  4702. {
  4703. succStr = "";
  4704. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("fetch 'Serial Number' failed!");
  4705. }
  4706. } while (false);
  4707. if (!ExistsFileA(storeFilePath)) {
  4708. RemoveFileA(storeFilePath);
  4709. }
  4710. return Error_Succeed;
  4711. #endif //RVC_OS_WIN
  4712. }
  4713. void ResourceWatcherFSM::AlarmSystemBasicInfo()
  4714. {
  4715. SystemBasicInfo info;
  4716. if (Error_Succeed == CatchSystemBasicInfo(info)) {
  4717. CSimpleStringA strResult("{");
  4718. if (!info.strManufacturer.IsNullOrEmpty()) {
  4719. strResult += CSimpleStringA::Format("\"Manufacturer\":\"%s\"", info.strManufacturer.GetData());
  4720. }
  4721. if (!info.strProductName.IsNullOrEmpty()) {
  4722. if (strResult.GetLength() > 1) {
  4723. strResult += ", ";
  4724. }
  4725. strResult += CSimpleStringA::Format("\"Product Name\":\"%s\"", info.strProductName.GetData());
  4726. }
  4727. if (!info.strSerialNumber.IsNullOrEmpty()) {
  4728. if (strResult.GetLength() > 1) {
  4729. strResult += ", ";
  4730. }
  4731. strResult += CSimpleStringA::Format("\"Serial Number\":\"%s\"", info.strSerialNumber.GetData());
  4732. }
  4733. strResult += "}";
  4734. LogWarn(Severity_Low, Error_Debug, LOG_INFO_DEVICE_SMBIOS_GET, strResult);
  4735. }
  4736. }
  4737. BOOL ResourceWatcherFSM::DetectIsFirstRunAtBoot()
  4738. {
  4739. CSystemRunInfo runInfo = { 0 };
  4740. GetEntityBase()->GetFunction()->GetSystemRunInfo(runInfo);
  4741. BOOL bSet = FALSE;
  4742. CBootInfo bootInfo = { 0 };
  4743. CSmallDateTime dateTime;
  4744. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("systemRunInfo time: %s", (LPCTSTR)runInfo.tmStart.ToTimeString());
  4745. ErrorCodeEnum erroCode = GetEntityBase()->GetFunction()->GetRebootInfo(/*runInfo.tmStart*/dateTime, bootInfo);
  4746. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("bootInfor time: %s", (LPCTSTR)bootInfo.tmStart.ToTimeString());
  4747. CSmallDateTime systemBootTime;
  4748. BOOL bRet = GetSystemBootTime(systemBootTime);
  4749. if (bRet && systemBootTime > bootInfo.tmStart) {
  4750. ErrorCodeEnum eRet = GetEntityBase()->GetFunction()->SetSysVar(SYSVAR_FRAMEWORK_FIRST_BOOT, SYSVAR_FRAMEWORK_FIRST_BOOT_YES);
  4751. if (eRet != Error_Succeed) {
  4752. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Set %s with %s failed: %d", SYSVAR_FRAMEWORK_FIRST_BOOT, SYSVAR_FRAMEWORK_FIRST_BOOT_YES, eRet);
  4753. }
  4754. bSet = TRUE;
  4755. std::map<std::string, std::string> srcData;
  4756. CSimpleStringA strData = CSimpleStringA::Format("{\"AppLastStartTime\":\"%s\",\"AppStartTime\":\"%s\",\"OSStartTime\":\"%s\"}"
  4757. , (LPCTSTR)bootInfo.tmStart.ToTimeString(), (LPCTSTR)runInfo.tmStart.ToTimeString(), systemBootTime.ToTimeString().GetData());
  4758. LogWarn(Severity_Low, Error_Debug, LOG_RESOURCEWATCHER_FIRST_RUN_AFTER_BOOT, strData.GetData());
  4759. }
  4760. else
  4761. {
  4762. GetEntityBase()->GetFunction()->SetSysVar(SYSVAR_FRAMEWORK_FIRST_BOOT, SYSVAR_FRAMEWORK_FIRST_BOOT_NO);
  4763. bSet = FALSE;
  4764. }
  4765. return bSet;
  4766. }
  4767. #ifdef RVC_OS_LINUX
  4768. void ResourceWatcherFSM::UploadMonitorSettings()
  4769. {
  4770. CSmartPointer<IConfigInfo> spCtSettingConfig;
  4771. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  4772. int nSwitchOff(0);
  4773. spCtSettingConfig->ReadConfigValueInt(GetEntityBase()->GetEntityName(), "MonitorListenOFF", nSwitchOff);
  4774. if (!!nSwitchOff) {
  4775. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Cancel monitor change listen!");
  4776. if (m_bFirstRunAfterBoot) {
  4777. CSimpleStringA value(true);
  4778. bool res = GetMonitorInfo(value);
  4779. LogWarn(Severity_Low, Error_Debug, LOG_INFO_MONITOR_SETTINGS, value);
  4780. }
  4781. }
  4782. else
  4783. {
  4784. int nInterval(10000);
  4785. int nValue(0);
  4786. spCtSettingConfig->ReadConfigValueInt(GetEntityBase()->GetEntityName(), "MonitorListenInterval", nValue);
  4787. if (nValue >= 10000) {
  4788. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Update monitor listen interval from %d to %d", nInterval, nValue);
  4789. nInterval = nValue;
  4790. }
  4791. CSimpleStringA strLastMonitor(true);
  4792. bool readFromCfgFalg(true);
  4793. CSmartPointer<IConfigInfo> spRunConfig;
  4794. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  4795. spRunConfig->ReadConfigValue("MonitorAttribute", "LastInfo", strLastMonitor);
  4796. do
  4797. {
  4798. CSimpleStringA value(true);
  4799. bool res = GetMonitorInfo(value);
  4800. if (!value.IsNullOrEmpty() && (strLastMonitor.IsNullOrEmpty() || (strLastMonitor.Compare(value) != 0))) {
  4801. LogWarn(Severity_Low, Error_Debug, LOG_INFO_MONITOR_SETTINGS, value);
  4802. CSmartPointer<IConfigInfo> spRunConfig;
  4803. GetEntityBase()->GetFunction()->OpenConfig(Config_Run, spRunConfig);
  4804. spRunConfig->WriteConfigValue("MonitorAttribute", "LastInfo", value);
  4805. strLastMonitor = value;
  4806. if (readFromCfgFalg) {
  4807. readFromCfgFalg = false;
  4808. }
  4809. }
  4810. else if (readFromCfgFalg && !strLastMonitor.IsNullOrEmpty() && strLastMonitor.Compare(value) == 0)
  4811. {
  4812. LogWarn(Severity_Low, Error_Debug, LOG_INFO_MONITOR_SETTINGS, value);
  4813. readFromCfgFalg = false;
  4814. }
  4815. Sleep(nInterval);
  4816. } while (true);
  4817. }
  4818. }
  4819. void ResourceWatcherFSM::DetectSoftwareInstallStatus()
  4820. {
  4821. CSmartPointer<IConfigInfo> spCtSettingConfig;
  4822. GetEntityBase()->GetFunction()->OpenConfig(Config_CenterSetting, spCtSettingConfig);
  4823. CSimpleStringA strSoftwaresNames[MAX_SOFTWARE_INSTALLED_DETECT_COUNT];
  4824. CSimpleStringA strSoftwaresVers[MAX_SOFTWARE_INSTALLED_DETECT_COUNT];
  4825. int nNameCounts = 0;
  4826. for (int i = 0; i < MAX_SOFTWARE_INSTALLED_DETECT_COUNT; i++) {
  4827. strSoftwaresNames[i] = "";
  4828. strSoftwaresVers[i] = "";
  4829. CSimpleStringA item = CSimpleStringA::Format("SoftwareInstallDetect%d", i);
  4830. CSimpleStringA value(true);
  4831. spCtSettingConfig->ReadConfigValue(GetEntityBase()->GetEntityName(), item, value);
  4832. if (value.IsNullOrEmpty()) {
  4833. break;
  4834. }
  4835. if (value.IndexOf("#") != -1) {
  4836. CAutoArray<CSimpleStringA> items = value.Split('#');
  4837. strSoftwaresNames[i] = items[0];
  4838. strSoftwaresVers[i] = items[1];
  4839. }
  4840. else
  4841. {
  4842. strSoftwaresNames[i] = value;
  4843. }
  4844. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Detect SoftDetect from CentralSetting, %d: %s,%s,%s."
  4845. , i, item.GetData(), strSoftwaresNames[i].GetData(), strSoftwaresVers[i].GetData());
  4846. nNameCounts++;
  4847. }
  4848. if (nNameCounts > 0) {
  4849. for (int i = 0; i < nNameCounts; ++i) {
  4850. CSimpleStringA runItem = CSimpleStringA::Format("dpkg -l | grep %s | awk '{print $3}'", strSoftwaresNames[i].GetData());
  4851. std::string additional("");
  4852. std::string succStr, errStr;
  4853. std::string runStr(runItem.GetData());
  4854. if (SP::Module::Util::ShellExecute(runStr, succStr, errStr)) {
  4855. if (succStr.empty()) {
  4856. LogWarn(Severity_Low, Error_Debug, LOG_WARN_SOFTWARE_DETECT_NOT_EXISTS_BASE + i,
  4857. CSimpleStringA::Format("{\"software_name\":\"%s\", \"version_status\":\"not installed\"}", strSoftwaresNames[i].GetData()));
  4858. }
  4859. else
  4860. {
  4861. succStr = SP::Utility::ToTrim(succStr);
  4862. if (strSoftwaresVers[i].IsNullOrEmpty() || strSoftwaresVers[i].Compare(succStr.c_str()) == 0) {
  4863. LogWarn(Severity_Low, Error_Debug, LOG_WARN_SOFTWARE_DETECT_EXISTS_BASE + i,
  4864. CSimpleStringA::Format("{\"software_name\":\"%s\", \"version_status\":\"%s\"}"
  4865. , strSoftwaresNames[i].GetData(), succStr.c_str(), strSoftwaresVers[i].GetData()));
  4866. }
  4867. else
  4868. {
  4869. LogWarn(Severity_Low, Error_Debug, LOG_WARN_SOFTWARE_DETECT_NOT_EXISTS_BASE + i,
  4870. CSimpleStringA::Format("{\"software_name\":\"%s\", \"version_status\":\"%s\", \"expected\":\"%s\"}"
  4871. , strSoftwaresNames[i].GetData(), succStr.c_str(), strSoftwaresVers[i].GetData()));
  4872. }
  4873. }
  4874. }
  4875. else
  4876. {
  4877. LogWarn(Severity_Low, Error_Debug, LOG_WARN_SOFTWARE_DETECT_EXISTS_STATUS_FAILED,
  4878. CSimpleStringA::Format("{\"software_name\":\"%s\", \"version_status\":\"detect failed\", \"expected\":\"%s\"}"
  4879. , strSoftwaresNames[i].GetData(), strSoftwaresVers[i].GetData()));
  4880. }
  4881. }
  4882. }
  4883. }
  4884. UINT ResourceWatcherFSM::GetSystemDisplayVersion(CSimpleStringA& strVersion)
  4885. {
  4886. UINT result = 0;
  4887. const char filePath[] = "/etc/os-version";
  4888. strVersion.Clear();
  4889. char tmp[33];
  4890. memset(tmp, 0, 33);
  4891. inifile_read_str_s("Version", "MajorVersion", "unknown", tmp, 32, filePath);
  4892. strVersion = tmp;
  4893. strVersion += ".";
  4894. memset(tmp, 0, 33);
  4895. inifile_read_str_s("Version", "MinorVersion", "unknown", tmp, 32, filePath);
  4896. strVersion += tmp;
  4897. strVersion += ".";
  4898. result = atoi(tmp);
  4899. memset(tmp, 0, 33);
  4900. inifile_read_str_s("Version", "OsBuild", "unknown", tmp, 32, filePath);
  4901. strVersion += tmp;
  4902. return result;
  4903. }
  4904. void ResourceWatcherFSM::UploadSysVersionInfo(UINT& ver)
  4905. {
  4906. std::map<std::string, std::string> srcData;
  4907. CSimpleStringA osver;
  4908. ver = GetSystemDisplayVersion(osver);
  4909. srcData.insert(std::make_pair("os-version", osver)); ////当前操作系统版本(系统信息)
  4910. tk_utsname_t t;
  4911. osutil_uname(&t);
  4912. srcData.insert(std::make_pair("sysname", t.sysname)); ////当前操作系统名
  4913. srcData.insert(std::make_pair("release", t.release)); ////当前发布级别
  4914. srcData.insert(std::make_pair("version", t.version)); ////当前发布版本
  4915. srcData.insert(std::make_pair("machine", t.machine)); ////当前硬件体系类型
  4916. do {
  4917. std::string sucContent, failedContent;
  4918. std::string cmd("cat /proc/cpuinfo |grep \"model name\" | awk 'NR==1'");
  4919. bool ret = SP::Module::Util::ShellExecute(cmd, sucContent, failedContent);
  4920. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("{%s}:{%s}{%s}", cmd.c_str(), sucContent.c_str(), failedContent.c_str());
  4921. auto elems = SP::Utility::Split(sucContent, ':');
  4922. if (elems.size() > 1) {
  4923. sucContent = SP::Utility::ToTrim(elems[1]);
  4924. }
  4925. srcData.insert(std::make_pair("cpu-type", sucContent));
  4926. } while (false);
  4927. //#if defined(RVC_OS_LINUX)
  4928. // CSystemStaticInfo sysInfo;
  4929. // ErrorCodeEnum eErr = GetEntityBase()->GetFunction()->GetSystemStaticInfo(sysInfo);
  4930. // if (eErr == Error_Succeed) {
  4931. // termInfo["MachineMode"] = (LPCTSTR)sysInfo.strMachineModel;
  4932. // termInfo["MachineSN"] = (LPCTSTR)sysInfo.strMachineSN;
  4933. // }
  4934. //#endif //RVC_OS_LINUX
  4935. auto ret = generateJsonStr(srcData);
  4936. LogWarn(Severity_Low, Error_Hardware, LOG_RESOURCEWATCHER_SYSTEMINFO, ret.second.c_str());
  4937. }
  4938. string ResourceWatcherFSM::MemoryProcessStatus()
  4939. {
  4940. CSmartPointer<IEntityFunction> spFunction = this->GetEntityBase()->GetFunction();
  4941. CSmartPointer<IConfigInfo> spConfig;
  4942. ErrorCodeEnum erroCode = Error_Unexpect;
  4943. ///*TODO: 不要每次都进来读取配置文件 (80374374@5/17/2023)*/
  4944. spFunction->OpenConfig(Config_CenterSetting, spConfig);
  4945. int memTop = 0;
  4946. erroCode = spConfig->ReadConfigValueInt("TOPProcess", "mem", memTop);
  4947. if (memTop == 0) {
  4948. memTop = 10;
  4949. }
  4950. else if (memTop == -1)
  4951. {
  4952. return "";
  4953. }
  4954. map<string, string> nameProcess;
  4955. CSystemRunInfo runInfo = { 0 };
  4956. GetEntityBase()->GetFunction()->GetSystemRunInfo(runInfo);
  4957. for (int i = 0; i < runInfo.strRunningEntityNames.GetCount(); i++)
  4958. {
  4959. string entityName = runInfo.strRunningEntityNames[i].GetData();
  4960. CEntityRunInfo entityInfo = { 0 };
  4961. GetEntityBase()->GetFunction()->GetEntityRunInfo(entityName.c_str(), entityInfo);
  4962. nameProcess[to_string(entityInfo.dwProcessID)] = entityName;
  4963. }
  4964. FILE* fp;
  4965. char buffer[2048];
  4966. string cmd = "ps c -eo pid,%mem,rss,command | sort -k3 -rn | head -" + to_string(memTop);
  4967. fp = popen(cmd.c_str(), "r");
  4968. int i = 0;
  4969. string s;
  4970. string str[4];
  4971. //mem
  4972. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("MEM TOP %d PROCESS!", memTop);
  4973. string topMessage = "";
  4974. while (i < memTop)
  4975. {
  4976. fgets(buffer, sizeof(buffer), fp);
  4977. s = buffer;
  4978. istringstream is(s);
  4979. is >> str[0] >> str[1] >> str[2] >> str[3];
  4980. double temp;
  4981. stringstream ss1;
  4982. ss1 << str[2];
  4983. ss1 >> temp;
  4984. temp = temp / (1024 * 1024);
  4985. stringstream ss2;
  4986. ss2 << std::setiosflags(std::ios::fixed) << std::setprecision(2) << temp;
  4987. str[2] = ss2.str();
  4988. if (nameProcess.find(str[0]) != nameProcess.end())
  4989. {
  4990. str[3] = nameProcess[str[0]];
  4991. }
  4992. topMessage = topMessage + "\"" + str[3] + "\"" + ":" + "\"" + str[1] + "%\"";
  4993. if (i != memTop - 1)
  4994. {
  4995. topMessage += ",";
  4996. }
  4997. ++i;
  4998. }
  4999. topMessage += "}";
  5000. pclose(fp);
  5001. return topMessage;
  5002. }
  5003. long ResourceWatcherFSM::GetCPURunTime(CPUInfo* cpu) //获取CPU的当前信息,同时返回CPU使用时间
  5004. {
  5005. FILE* fd;
  5006. char buff[1024] = { 0 };
  5007. CPUInfo* cpuInfo = cpu;
  5008. fd = fopen("/proc/stat", "r");
  5009. if (!fd) {
  5010. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Read CPU Info FAILED!");
  5011. return -1;
  5012. }
  5013. fgets(buff, sizeof(buff), fd);
  5014. char name[64] = { 0 };
  5015. sscanf(buff, "%s %u %u %u %u %u %u %u",
  5016. cpuInfo->name, &cpuInfo->user, &cpuInfo->nice, &cpuInfo->system, &cpuInfo->idle, &cpuInfo->lowait, &cpuInfo->irq, &cpuInfo->softirq);
  5017. fclose(fd);
  5018. return (cpuInfo->user + cpuInfo->nice + cpuInfo->system + cpuInfo->idle + cpuInfo->lowait + cpuInfo->irq + cpuInfo->softirq);
  5019. }
  5020. long ResourceWatcherFSM::ReadProcCPURunTime(DWORD pid, ProcCPUInfo* pInfo) { //根据pid读取进程占用的CPU时间
  5021. char file_name[64] = { 0 };
  5022. FILE* fd;
  5023. char line_buff[1024] = { 0 };
  5024. sprintf(file_name, "/proc/%d/stat", pid);
  5025. fd = fopen(file_name, "r");
  5026. if (nullptr == fd) {
  5027. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Read PROCESS[%d] CPU Info FAILED!", pid);
  5028. return -1;
  5029. }
  5030. fgets(line_buff, sizeof(line_buff), fd);
  5031. sscanf(line_buff, "%u %s", &pInfo->pid, pInfo->pname);
  5032. const char* q = GetItems(line_buff, PROCESS_ITEM);
  5033. sscanf(q, "%ld %ld %ld %ld", &pInfo->utime, &pInfo->stime, &pInfo->cutime, &pInfo->cstime);
  5034. fclose(fd);
  5035. return (pInfo->utime + pInfo->stime + pInfo->cutime + pInfo->cstime);
  5036. }
  5037. //获取第N项开始的指针
  5038. const char* ResourceWatcherFSM::GetItems(const char* buffer, unsigned int item) {
  5039. const char* p = buffer;
  5040. int len = strlen(buffer);
  5041. int count = 0;
  5042. for (int i = 0; i < len; i++) {
  5043. if (' ' == *p) {
  5044. count++;
  5045. if (count == item - 1) {
  5046. p++;
  5047. break;
  5048. }
  5049. }
  5050. p++;
  5051. }
  5052. return p;
  5053. }
  5054. bool cmp(const pair<DWORD, DOUBLE>& a, const pair<DWORD, DOUBLE>& b) {
  5055. return a.second > b.second;
  5056. }
  5057. /* -----------------------
  5058. 根据先后两个时间的CPU信息,
  5059. 计算CPU的总占用量和各个进程的占用量
  5060. ----------------------- */
  5061. void ResourceWatcherFSM::GetSystemCPUStatus()
  5062. {
  5063. //int cpu_num = sysconf(_SC_NPROCESSORS_CONF); //CPU核数
  5064. CPUInfo* cpuOld = new CPUInfo;
  5065. CPUInfo* cpuNew = new CPUInfo;
  5066. const char* tPath = "/proc";
  5067. map<DWORD, string> nameProcess; // <pid, 进程名>
  5068. map<DWORD, DOUBLE> usageProcess; // <pid, CPU占用率>
  5069. long oldTotalTime = GetCPURunTime(cpuOld); //旧CPU运行总时间
  5070. if (oldTotalTime == -1)
  5071. {
  5072. return;
  5073. }
  5074. map<DWORD, long> oldProcCPUTime; // <pid, 旧时刻进程的CPU占用时间>
  5075. struct dirent* oldDirp;
  5076. DIR* oldDp = opendir(tPath);
  5077. if (oldDp != NULL) {
  5078. while ((oldDirp = readdir(oldDp)) != NULL) { //获取旧时刻 每一个进程的pid、CPU运行时间、进程名
  5079. string fileName = oldDirp->d_name;
  5080. BOOL isPid = TRUE;
  5081. for (int i = 0; i < fileName.length(); ++i) //判断进程id的文件夹 是否文件名是否纯数字
  5082. {
  5083. if (!isdigit(fileName[i]))
  5084. {
  5085. isPid = FALSE;
  5086. break;
  5087. }
  5088. }
  5089. if (!isPid) continue;
  5090. DWORD pid;
  5091. long oldTime;
  5092. ProcCPUInfo* oldProcCPU = new ProcCPUInfo;
  5093. sscanf(oldDirp->d_name, "%d", &pid); //读取文件夹名
  5094. oldTime = ReadProcCPURunTime(pid, oldProcCPU);
  5095. if (oldTime == -1)
  5096. {
  5097. return;
  5098. }
  5099. oldProcCPUTime[pid] = oldTime;
  5100. nameProcess[pid] = oldProcCPU->pname;
  5101. delete oldProcCPU;
  5102. }
  5103. closedir(oldDp);
  5104. }
  5105. sleep(1);
  5106. long newTotalTime = GetCPURunTime(cpuNew); //新CPU运行总时间
  5107. if (newTotalTime == -1)
  5108. {
  5109. return;
  5110. }
  5111. map<DWORD, long> newProcCPUTime; // <pid, 新时刻进程的CPU占用时间>
  5112. struct dirent* newDirp;
  5113. DIR* newDp = opendir(tPath);
  5114. if (newDp != NULL) {
  5115. while ((newDirp = readdir(newDp)) != NULL) { //获取新时刻 每一个进程的pid、CPU运行时间、进程名
  5116. string fileName = newDirp->d_name;
  5117. BOOL isPid = TRUE;
  5118. for (int i = 0; i < fileName.length(); ++i) //判断进程id的文件夹 是否文件名是否纯数字
  5119. {
  5120. if (!isdigit(fileName[i]))
  5121. {
  5122. isPid = FALSE;
  5123. break;
  5124. }
  5125. }
  5126. if (!isPid) continue;
  5127. DWORD pid;
  5128. long newTime;
  5129. ProcCPUInfo* newProcCPU = new ProcCPUInfo;
  5130. sscanf(newDirp->d_name, "%d", &pid); //读取文件夹名
  5131. newTime = ReadProcCPURunTime(pid, newProcCPU);
  5132. if (newTime == -1)
  5133. {
  5134. return;
  5135. }
  5136. newProcCPUTime[pid] = newTime;
  5137. string tName = newProcCPU->pname;
  5138. tName.erase(tName.begin());
  5139. tName.erase(tName.begin() + (tName.length() - 1));
  5140. nameProcess[pid] = tName;
  5141. delete newProcCPU;
  5142. }
  5143. closedir(newDp);
  5144. }
  5145. map<DWORD, long>::iterator it;
  5146. for (it = newProcCPUTime.begin(); it != newProcCPUTime.end(); ++it)
  5147. {
  5148. DWORD tPid = it->first;
  5149. //进程的占用率 = 单位时间间隔里进程的CPU时间片占用 / 单位时间间隔里CPU的整体时间片
  5150. DOUBLE tRatio = 100.0 * (newProcCPUTime[tPid] - oldProcCPUTime[tPid]) / (newTotalTime - oldTotalTime);
  5151. usageProcess[tPid] = tRatio;
  5152. }
  5153. CSystemRunInfo runInfo = { 0 };
  5154. GetEntityBase()->GetFunction()->GetSystemRunInfo(runInfo);
  5155. for (int i = 0; i < runInfo.strRunningEntityNames.GetCount(); i++) //实体进程的名称默认全为sphost,需转换为对应实体名
  5156. {
  5157. string entityName = runInfo.strRunningEntityNames[i].GetData();
  5158. CEntityRunInfo entityInfo = { 0 };
  5159. GetEntityBase()->GetFunction()->GetEntityRunInfo(entityName.c_str(), entityInfo);
  5160. nameProcess[entityInfo.dwProcessID] = entityName;
  5161. }
  5162. vector<pair<DWORD, DOUBLE>> vec(usageProcess.begin(), usageProcess.end());
  5163. sort(vec.begin(), vec.end(), cmp); //根据CPU使用率从大到小进行排序
  5164. int len = MIN(m_cpuTop, vec.size());
  5165. string procWarn = "";
  5166. for (int i = 0; i < len; ++i) //构建进程的CPU使用告警信息
  5167. {
  5168. char* tUsage = new char[20];
  5169. sprintf(tUsage, "%.2lf", vec[i].second);
  5170. procWarn = procWarn + ",\"" + nameProcess[vec[i].first] + "\":\"" + string(tUsage) + "%\"";
  5171. delete tUsage;
  5172. }
  5173. unsigned long oldInfo, newInfo;
  5174. double cUsedRate = 0;
  5175. //第一次(用户+优先级+系统+空闲)的时间
  5176. //oldInfo = (unsigned long)(cpuOld->user + cpuOld->nice + cpuOld->system + cpuOld->idle + cpuOld->lowait + cpuOld->irq + cpuOld->softirq);
  5177. //第二次(用户+优先级+系统+空闲)的时间
  5178. //newInfo = (unsigned long)(cpuNew->user + cpuNew->nice + cpuNew->system + cpuNew->idle + cpuNew->lowait + cpuNew->irq + cpuNew->softirq);
  5179. double totalTime = newTotalTime - oldTotalTime;
  5180. double freeTime = cpuNew->idle - cpuOld->idle;
  5181. cUsedRate = (totalTime - freeTime) / totalTime;
  5182. if (m_cpuHighPercent <= 0)
  5183. {
  5184. m_cpuHighPercent = 80; //默认告警百分比80%
  5185. }
  5186. if (cpuWarnThreshold <= 0)
  5187. {
  5188. cpuWarnThreshold = 10; //默认满10次告一次警
  5189. }
  5190. if ((cUsedRate * 100) > m_cpuHighPercent)
  5191. {
  5192. cpuWarnTime = cpuWarnTime % cpuWarnThreshold;
  5193. if (cpuWarnTime == 0) //每达到一次阈值告警一次
  5194. {
  5195. stringstream tss;
  5196. tss << std::setiosflags(std::ios::fixed) << std::setprecision(2) << cUsedRate * 100;
  5197. string warn = "{ \"total_used\":\"" + tss.str() + "%\"" + procWarn + "}";
  5198. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_CPU_TOO_HIGH, CSimpleStringA::Format("%s", warn.c_str()));
  5199. }
  5200. cpuWarnTime++;
  5201. }
  5202. delete cpuNew;
  5203. delete cpuOld;
  5204. }
  5205. void ResourceWatcherFSM::GetSystemMemoryStatus()
  5206. {
  5207. MemInfo* memInfo = new MemInfo();
  5208. FILE* fd;
  5209. char buff[256];
  5210. fd = fopen("/proc/meminfo", "r");
  5211. if (!fd) {
  5212. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("Read Memory INFO FAILED!");
  5213. return;
  5214. }
  5215. fgets(buff, sizeof(buff), fd);
  5216. sscanf(buff, "%s %lu ", memInfo->infoName1, &memInfo->memTotal);
  5217. fgets(buff, sizeof(buff), fd);
  5218. sscanf(buff, "%s %lu ", memInfo->infoName2, &memInfo->memFree);
  5219. fgets(buff, sizeof(buff), fd);
  5220. sscanf(buff, "%s %lu ", memInfo->infoName3, &memInfo->memAvail);
  5221. fclose(fd);
  5222. double mUsedRate = 0;
  5223. mUsedRate = (double)(memInfo->memTotal - memInfo->memAvail) / (double)memInfo->memTotal;
  5224. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  5225. char szResult2[DEFAULT_OUTPUT_FORMAT_SIZE];
  5226. char szResult3[DEFAULT_OUTPUT_FORMAT_SIZE];
  5227. ByteSprintf(szResult, (double)(memInfo->memTotal) * 1024, 'G'); // 内存信息在 /proc/meminfo 中的单位为 KB
  5228. ByteSprintf(szResult2, (double)(memInfo->memFree) * 1024, 'G');
  5229. ByteSprintf(szResult3, (double)(memInfo->memAvail) * 1024, 'G');
  5230. if (m_memHighPercent <= 0)
  5231. {
  5232. m_memHighPercent = 80; //默认告警百分比80%
  5233. }
  5234. if (memWarnThreshold <= 0)
  5235. {
  5236. memWarnThreshold = 10; //默认告警阈值10次
  5237. }
  5238. if ((mUsedRate * 100) > m_memHighPercent)
  5239. {
  5240. memWarnTime = memWarnTime % memWarnThreshold;
  5241. if (memWarnTime == 0) //每达到阈值告警一次
  5242. {
  5243. char usedResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  5244. ByteSprintf(usedResult, (double)(memInfo->memTotal) * 1024 * mUsedRate, 'G');
  5245. stringstream ss1;
  5246. ss1 << std::setiosflags(std::ios::fixed) << std::setprecision(2) << mUsedRate * 100;
  5247. string warn = "{ \"total\":\"" + string(szResult) + "\"," +
  5248. "\"used\":\"" + ss1.str() + "%\"," + MemoryProcessStatus();
  5249. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_MEMORY_TOO_HIGH, CSimpleStringA::Format("%s", warn.c_str()));
  5250. }
  5251. memWarnTime++;
  5252. }
  5253. }
  5254. void ResourceWatcherFSM::GetDiskStatusFrom(LPCTSTR path)
  5255. {
  5256. if (m_diskHighPercent <= 0)
  5257. {
  5258. m_diskHighPercent = 90; //默认硬盘告警百分比90%
  5259. }
  5260. unsigned long long uiTotalBytes;
  5261. unsigned long long uiFreeDiskBytes;
  5262. char szResult[DEFAULT_OUTPUT_FORMAT_SIZE];
  5263. char szResult2[DEFAULT_OUTPUT_FORMAT_SIZE];
  5264. BOOL readFlag = DiskInfo::GetDiskSpace(path, uiTotalBytes, uiFreeDiskBytes);
  5265. double dUsedRate = 1 - (uiFreeDiskBytes * kConversion[3]) / (uiTotalBytes * kConversion[3]);
  5266. ByteSprintf(szResult, (double)uiTotalBytes);
  5267. ByteSprintf(szResult2, (double)uiFreeDiskBytes);
  5268. if (readFlag && (dUsedRate * 100 < m_diskHighPercent)) {
  5269. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("[Disk# %s] total: %s, free: %s, [Disk] used rate: %.1lf%%.", path, szResult, szResult2, dUsedRate * 100);
  5270. }
  5271. else
  5272. {
  5273. LogWarn(Severity_Middle, Error_Resource, LOG_EVT_RESOURCE_HARDDISK_TOO_HIGH,
  5274. CSimpleStringA::Format("DISK TOO HIGH! -- [Disk# %s] total: %s, free: %s, [Disk] used rate: %.1lf%%.",
  5275. path, szResult, szResult2, dUsedRate * 100));
  5276. CheckDiskFileSpaceTask* task4 = new CheckDiskFileSpaceTask(this);
  5277. GetEntityBase()->GetFunction()->PostThreadPoolTask(task4);
  5278. }
  5279. }
  5280. void ResourceWatcherFSM::GetSystemDiskStatus()
  5281. {
  5282. GetDiskStatusFrom("/");
  5283. }
  5284. void ResourceWatcherFSM::GetOperationDiskStatus()
  5285. {
  5286. GetDiskStatusFrom(m_csUploadVideoPath);
  5287. }
  5288. void ResourceWatcherFSM::switchWindowsEffect(const char* cfgFilePath, bool enable)
  5289. {
  5290. char tmp[33];
  5291. memset(tmp, 0, 33);
  5292. inifile_read_str_s("Compositing", "Enabled", "", tmp, 32, cfgFilePath);
  5293. if (strlen(tmp) == 0 || stricmp(tmp, "false") == 0) {
  5294. if (enable) {
  5295. inifile_write_str(cfgFilePath, "Compositing", "Enabled", "true");
  5296. LogWarn(Severity_Low, Error_InvalidState, LOG_RESOURCEWATCHER_COMPOSITING_CHANGE_ON, "to active window effect");
  5297. }
  5298. }
  5299. else if (stricmp(tmp, "true") == 0) {
  5300. if (!enable) {
  5301. inifile_write_str(cfgFilePath, "Compositing", "Enabled", "false");
  5302. LogWarn(Severity_Low, Error_InvalidState, LOG_RESOURCEWATCHER_COMPOSITING_CHANGE_OFF, "to inactive window effect");
  5303. }
  5304. }
  5305. }
  5306. void ResourceWatcherFSM::RecoverDDEClipboardEnable()
  5307. {
  5308. const std::string clipboardPath = "/usr/bin/dde-clipboard";
  5309. const std::string clipboardPathBak = "/usr/bin/dde-clipboard.bak";
  5310. if (!ExistsFileA(clipboardPath.c_str()) && ExistsFileA(clipboardPathBak.c_str())) {
  5311. fileutil_copy_file(clipboardPath.c_str(), clipboardPathBak.c_str());
  5312. //fileutil_delete_file(clipboardPathBak.c_str());
  5313. LogWarn(Severity_Low, Error_Debug, LOG_RESOURCEWATCHER_CLIPBOARD_RECOVER_ENABLE, "recover clipboard succ.");
  5314. }
  5315. else {
  5316. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("%s exists.", clipboardPath.c_str());
  5317. }
  5318. }
  5319. void ResourceWatcherFSM::ConfirmDDEClipboardDisable()
  5320. {
  5321. const std::string clipboardPath = "/usr/bin/dde-clipboard";
  5322. const std::string clipboardPathBak = "/usr/bin/dde-clipboard.bak";
  5323. char* clipboards[] = { "dde-clipboard", "dde-clipboardloader" };
  5324. if (!ExistsFileA(clipboardPath.c_str())) {
  5325. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("There are no any %s files.", clipboardPath.c_str());
  5326. if (ExistsFileA(clipboardPathBak.c_str())) {
  5327. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("%s exists.", clipboardPathBak.c_str());
  5328. }
  5329. return;
  5330. }
  5331. if (0 != fileutil_copy_file(clipboardPathBak.c_str(), clipboardPath.c_str())) {
  5332. LogWarn(Severity_Low, Error_Unexpect, LOG_RESOURCEWATCHER_CLIPBOARD_REMOVE_FAILED, "remove clipboard failed");
  5333. }
  5334. else {
  5335. fileutil_delete_file(clipboardPath.c_str());
  5336. osutil_terminate_related_process(clipboards, array_size(clipboards));
  5337. LogWarn(Severity_Low, Error_Process, LOG_RESOURCEWATCHER_CLIPBOARD_REMOVE_SUCC, "remove clipboard succ.");
  5338. }
  5339. }
  5340. void ResourceWatcherFSM::ConfirmNotificationCenterDisable()
  5341. {
  5342. const char* osd_file_path = "/usr/lib/deepin-daemon/dde-osd";
  5343. const char* osd_bak_file_path = "/usr/lib/deepin-daemon/dde-osd.bak";
  5344. CSmartPointer<IConfigInfo> spConfig;
  5345. GetEntityBase()->GetFunction()->OpenConfig(Config_Cache, spConfig);
  5346. int flag = 0;
  5347. spConfig->ReadConfigValueInt("UOSFeatures", "NotifyCenterSupport", flag);
  5348. if (flag == 1) {
  5349. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("allow the dde-osd process alive.");
  5350. if (!ExistsFileA(osd_file_path) && ExistsFileA(osd_bak_file_path)) {
  5351. fileutil_copy_file(osd_file_path, osd_bak_file_path);
  5352. LogWarn(Severity_Low, Error_Debug, LOG_RESOURCEWATCHER_OSD_RECOVER_ENABLE, "recover dde osd succ.");
  5353. }
  5354. }
  5355. else {
  5356. if (ExistsFileA(osd_file_path) && !ExistsFileA(osd_bak_file_path)) {
  5357. char* relates[] = { "dde-osd" };
  5358. if (0 != fileutil_copy_file(osd_bak_file_path, osd_file_path)) {
  5359. LogWarn(Severity_Low, Error_Unexpect, LOG_RESOURCEWATCHER_CLIPBOARD_REMOVE_FAILED, "remove dde osd failed");
  5360. }
  5361. else {
  5362. osutil_terminate_related_process(relates, array_size(relates));
  5363. fileutil_delete_file(osd_file_path);
  5364. LogWarn(Severity_Low, Error_Process, LOG_RESOURCEWATCHER_OSD_REMOVE_SUCC, "remove dde osd succ.");
  5365. }
  5366. }
  5367. }
  5368. }
  5369. void ResourceWatcherFSM::ConfirmWindowEffectHasBeenOpen()
  5370. {
  5371. array_header_t* arr;
  5372. arr = fileutil_get_sub_dirs_a("/home");
  5373. if (arr) {
  5374. int i;
  5375. for (i = 0; i < arr->nelts; ++i) {
  5376. char szDestSubDir[256] = { 0 };
  5377. char* dir = ARRAY_IDX(arr, i, char*);
  5378. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("sub dir: %s", dir);
  5379. strcpy(szDestSubDir, dir);
  5380. strcat(szDestSubDir, SPLIT_SLASH_STR);
  5381. strcat(szDestSubDir, ".config/kwinrc");
  5382. if (ExistsFileA(szDestSubDir)) {
  5383. switchWindowsEffect(szDestSubDir, true);
  5384. }
  5385. }
  5386. toolkit_array_free2(arr);
  5387. }
  5388. }
  5389. #endif // RVC_OS_LINUX