Common.cpp 126 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248
  1. //---------------------------------------------------------------------------
  2. #define NO_WIN32_LEAN_AND_MEAN
  3. #include <vcl.h>
  4. #pragma hdrstop
  5. #include "Common.h"
  6. #include "Exceptions.h"
  7. #include "TextsCore.h"
  8. #include "Interface.h"
  9. #include <StrUtils.hpp>
  10. #include <DateUtils.hpp>
  11. #include <math.h>
  12. #include <shlobj.h>
  13. #include <limits>
  14. #include <algorithm>
  15. #include <shlwapi.h>
  16. #include <tlhelp32.h>
  17. #include <psapi.h>
  18. #include <CoreMain.h>
  19. #include <openssl/pkcs12.h>
  20. #include <openssl/pem.h>
  21. #include <openssl/err.h>
  22. #include <openssl/ssl.h>
  23. //---------------------------------------------------------------------------
  24. #pragma package(smart_init)
  25. //---------------------------------------------------------------------------
  26. const wchar_t * DSTModeNames = L"Win;Unix;Keep";
  27. //---------------------------------------------------------------------------
  28. //---------------------------------------------------------------------------
  29. const wchar_t EngShortMonthNames[12][4] =
  30. {L"Jan", L"Feb", L"Mar", L"Apr", L"May", L"Jun",
  31. L"Jul", L"Aug", L"Sep", L"Oct", L"Nov", L"Dec"};
  32. const char Bom[3] = "\xEF\xBB\xBF";
  33. const wchar_t TokenPrefix = L'%';
  34. const wchar_t NoReplacement = wchar_t(false);
  35. const wchar_t TokenReplacement = wchar_t(true);
  36. // Note similar list in MakeValidFileName
  37. const UnicodeString LocalInvalidChars(TraceInitStr(L"/\\:*?\"<>|"));
  38. const UnicodeString PasswordMask(TraceInitStr(L"***"));
  39. const UnicodeString Ellipsis(TraceInitStr(L"..."));
  40. //---------------------------------------------------------------------------
  41. UnicodeString ReplaceChar(UnicodeString Str, wchar_t A, wchar_t B)
  42. {
  43. for (Integer Index = 0; Index < Str.Length(); Index++)
  44. if (Str[Index+1] == A) Str[Index+1] = B;
  45. return Str;
  46. }
  47. //---------------------------------------------------------------------------
  48. UnicodeString DeleteChar(UnicodeString Str, wchar_t C)
  49. {
  50. int P;
  51. while ((P = Str.Pos(C)) > 0)
  52. {
  53. Str.Delete(P, 1);
  54. }
  55. return Str;
  56. }
  57. //---------------------------------------------------------------------------
  58. template<typename T>
  59. void DoPackStr(T & Str)
  60. {
  61. // Following will free unnecessary bytes
  62. Str = Str.c_str();
  63. }
  64. //---------------------------------------------------------------------------
  65. void PackStr(UnicodeString & Str)
  66. {
  67. DoPackStr(Str);
  68. }
  69. //---------------------------------------------------------------------------
  70. void PackStr(RawByteString & Str)
  71. {
  72. DoPackStr(Str);
  73. }
  74. //---------------------------------------------------------------------------
  75. void PackStr(AnsiString & Str)
  76. {
  77. DoPackStr(Str);
  78. }
  79. //---------------------------------------------------------------------------
  80. template<typename T>
  81. void __fastcall DoShred(T & Str)
  82. {
  83. if (!Str.IsEmpty())
  84. {
  85. Str.Unique();
  86. memset(Str.c_str(), 0, Str.Length() * sizeof(*Str.c_str()));
  87. Str = L"";
  88. }
  89. }
  90. //---------------------------------------------------------------------------
  91. void __fastcall Shred(UnicodeString & Str)
  92. {
  93. DoShred(Str);
  94. }
  95. //---------------------------------------------------------------------------
  96. void __fastcall Shred(UTF8String & Str)
  97. {
  98. DoShred(Str);
  99. }
  100. //---------------------------------------------------------------------------
  101. void __fastcall Shred(AnsiString & Str)
  102. {
  103. DoShred(Str);
  104. }
  105. //---------------------------------------------------------------------------
  106. void __fastcall Shred(RawByteString & Str)
  107. {
  108. DoShred(Str);
  109. }
  110. //---------------------------------------------------------------------------
  111. UnicodeString AnsiToString(const RawByteString & S)
  112. {
  113. return UnicodeString(AnsiString(S));
  114. }
  115. //---------------------------------------------------------------------------
  116. UnicodeString AnsiToString(const char * S, size_t Len)
  117. {
  118. return UnicodeString(AnsiString(S, Len));
  119. }
  120. //---------------------------------------------------------------------------
  121. // Note similar function ValidLocalFileName
  122. UnicodeString MakeValidFileName(UnicodeString FileName)
  123. {
  124. // Note similar list in LocalInvalidChars
  125. UnicodeString IllegalChars = L":;,=+<>|\"[] \\/?*";
  126. for (int Index = 0; Index < IllegalChars.Length(); Index++)
  127. {
  128. FileName = ReplaceChar(FileName, IllegalChars[Index+1], L'-');
  129. }
  130. return FileName;
  131. }
  132. //---------------------------------------------------------------------------
  133. UnicodeString RootKeyToStr(HKEY RootKey)
  134. {
  135. if (RootKey == HKEY_USERS) return L"HKU";
  136. else
  137. if (RootKey == HKEY_LOCAL_MACHINE) return L"HKLM";
  138. else
  139. if (RootKey == HKEY_CURRENT_USER) return L"HKCU";
  140. else
  141. if (RootKey == HKEY_CLASSES_ROOT) return L"HKCR";
  142. else
  143. if (RootKey == HKEY_CURRENT_CONFIG) return L"HKCC";
  144. else
  145. if (RootKey == HKEY_DYN_DATA) return L"HKDD";
  146. else
  147. { Abort(); return L""; };
  148. }
  149. //---------------------------------------------------------------------------
  150. UnicodeString BooleanToEngStr(bool B)
  151. {
  152. if (B)
  153. {
  154. return L"Yes";
  155. }
  156. else
  157. {
  158. return L"No";
  159. }
  160. }
  161. //---------------------------------------------------------------------------
  162. UnicodeString BooleanToStr(bool B)
  163. {
  164. if (B)
  165. {
  166. return LoadStr(YES_STR);
  167. }
  168. else
  169. {
  170. return LoadStr(NO_STR);
  171. }
  172. }
  173. //---------------------------------------------------------------------------
  174. UnicodeString DefaultStr(const UnicodeString & Str, const UnicodeString & Default)
  175. {
  176. if (!Str.IsEmpty())
  177. {
  178. return Str;
  179. }
  180. else
  181. {
  182. return Default;
  183. }
  184. }
  185. //---------------------------------------------------------------------------
  186. // For alternative with quoting support, see TFTPFileSystem::CutFeature
  187. UnicodeString CutToChar(UnicodeString &Str, wchar_t Ch, bool Trim)
  188. {
  189. Integer P = Str.Pos(Ch);
  190. UnicodeString Result;
  191. if (P)
  192. {
  193. Result = Str.SubString(1, P-1);
  194. Str.Delete(1, P);
  195. }
  196. else
  197. {
  198. Result = Str;
  199. Str = L"";
  200. }
  201. if (Trim)
  202. {
  203. Result = Result.TrimRight();
  204. Str = Str.TrimLeft();
  205. }
  206. return Result;
  207. }
  208. //---------------------------------------------------------------------------
  209. UnicodeString CopyToChars(const UnicodeString & Str, int & From, UnicodeString Chs, bool Trim,
  210. wchar_t * Delimiter, bool DoubleDelimiterEscapes)
  211. {
  212. UnicodeString Result;
  213. int P;
  214. for (P = From; P <= Str.Length(); P++)
  215. {
  216. if (IsDelimiter(Chs, Str, P))
  217. {
  218. if (DoubleDelimiterEscapes &&
  219. (P < Str.Length()) &&
  220. IsDelimiter(Chs, Str, P + 1) &&
  221. (Str[P + 1] == Str[P]))
  222. {
  223. Result += Str[P];
  224. P++;
  225. }
  226. else
  227. {
  228. break;
  229. }
  230. }
  231. else
  232. {
  233. Result += Str[P];
  234. }
  235. }
  236. if (P <= Str.Length())
  237. {
  238. if (Delimiter != NULL)
  239. {
  240. *Delimiter = Str[P];
  241. }
  242. }
  243. else
  244. {
  245. if (Delimiter != NULL)
  246. {
  247. *Delimiter = L'\0';
  248. }
  249. }
  250. // even if we reached the end, return index, as if there were the delimiter,
  251. // so caller can easily find index of the end of the piece by subtracting
  252. // 2 from From (as long as he did not asked for trimming)
  253. From = P+1;
  254. if (Trim)
  255. {
  256. Result = Result.TrimRight();
  257. while ((From <= Str.Length()) && (Str[From] == L' '))
  258. {
  259. From++;
  260. }
  261. }
  262. return Result;
  263. }
  264. //---------------------------------------------------------------------------
  265. UnicodeString CopyToChar(const UnicodeString & Str, wchar_t Ch, bool Trim)
  266. {
  267. int From = 1;
  268. return CopyToChars(Str, From, UnicodeString(Ch), Trim);
  269. }
  270. //---------------------------------------------------------------------------
  271. UnicodeString RemoveSuffix(const UnicodeString & Str, const UnicodeString & Suffix, bool RemoveNumbersAfterSuffix)
  272. {
  273. UnicodeString Result = Str;
  274. UnicodeString Buf = Str;
  275. if (RemoveNumbersAfterSuffix)
  276. {
  277. while (!Buf.IsEmpty() && IsDigit(Buf[Buf.Length()]))
  278. {
  279. Buf.SetLength(Buf.Length() - 1);
  280. }
  281. }
  282. if (EndsStr(Suffix, Buf))
  283. {
  284. Result.SetLength(Buf.Length() - Suffix.Length());
  285. }
  286. return Result;
  287. }
  288. //---------------------------------------------------------------------------
  289. UnicodeString DelimitStr(UnicodeString Str, UnicodeString Chars)
  290. {
  291. for (int i = 1; i <= Str.Length(); i++)
  292. {
  293. if (Str.IsDelimiter(Chars, i))
  294. {
  295. Str.Insert(L"\\", i);
  296. i++;
  297. }
  298. }
  299. return Str;
  300. }
  301. //---------------------------------------------------------------------------
  302. UnicodeString ShellDelimitStr(UnicodeString Str, wchar_t Quote)
  303. {
  304. UnicodeString Chars = L"$\\";
  305. if (Quote == L'"')
  306. {
  307. Chars += L"`\"";
  308. }
  309. return DelimitStr(Str, Chars);
  310. }
  311. //---------------------------------------------------------------------------
  312. UnicodeString ExceptionLogString(Exception *E)
  313. {
  314. DebugAssert(E);
  315. if (E->InheritsFrom(__classid(Exception)))
  316. {
  317. UnicodeString Msg;
  318. Msg = FORMAT(L"(%s) %s", (E->ClassName(), E->Message));
  319. if (E->InheritsFrom(__classid(ExtException)))
  320. {
  321. TStrings * MoreMessages = ((ExtException*)E)->MoreMessages;
  322. if (MoreMessages)
  323. {
  324. Msg += L"\n" +
  325. ReplaceStr(MoreMessages->Text, L"\r", L"");
  326. }
  327. }
  328. return Msg;
  329. }
  330. else
  331. {
  332. wchar_t Buffer[1024];
  333. ExceptionErrorMessage(ExceptObject(), ExceptAddr(), Buffer, LENOF(Buffer));
  334. return UnicodeString(Buffer);
  335. }
  336. }
  337. //---------------------------------------------------------------------------
  338. UnicodeString __fastcall MainInstructions(const UnicodeString & S)
  339. {
  340. UnicodeString MainMsgTag = LoadStr(MAIN_MSG_TAG);
  341. return MainMsgTag + S + MainMsgTag;
  342. }
  343. //---------------------------------------------------------------------------
  344. bool __fastcall HasParagraphs(const UnicodeString & S)
  345. {
  346. return (S.Pos(L"\n\n") > 0);
  347. }
  348. //---------------------------------------------------------------------------
  349. UnicodeString __fastcall MainInstructionsFirstParagraph(const UnicodeString & S)
  350. {
  351. // WORKAROUND, we consider it bad practice, the highlighting should better
  352. // be localized (but maybe we change our mind later)
  353. UnicodeString Result;
  354. int Pos = S.Pos(L"\n\n");
  355. // we would not be calling this on single paragraph message
  356. if (DebugAlwaysTrue(Pos > 0))
  357. {
  358. Result =
  359. MainInstructions(S.SubString(1, Pos - 1)) +
  360. S.SubString(Pos, S.Length() - Pos + 1);
  361. }
  362. else
  363. {
  364. Result = MainInstructions(S);
  365. }
  366. return Result;
  367. }
  368. //---------------------------------------------------------------------------
  369. bool ExtractMainInstructions(UnicodeString & S, UnicodeString & MainInstructions)
  370. {
  371. bool Result = false;
  372. UnicodeString MainMsgTag = LoadStr(MAIN_MSG_TAG);
  373. if (StartsStr(MainMsgTag, S))
  374. {
  375. int EndTagPos =
  376. S.SubString(MainMsgTag.Length() + 1, S.Length() - MainMsgTag.Length()).Pos(MainMsgTag);
  377. if (EndTagPos > 0)
  378. {
  379. MainInstructions = S.SubString(MainMsgTag.Length() + 1, EndTagPos - 1);
  380. S.Delete(1, EndTagPos + (2 * MainMsgTag.Length()) - 1);
  381. Result = true;
  382. }
  383. }
  384. DebugAssert(MainInstructions.Pos(MainMsgTag) == 0);
  385. DebugAssert(S.Pos(MainMsgTag) == 0);
  386. return Result;
  387. }
  388. //---------------------------------------------------------------------------
  389. static int FindInteractiveMsgStart(const UnicodeString & S)
  390. {
  391. int Result = 0;
  392. UnicodeString InteractiveMsgTag = LoadStr(INTERACTIVE_MSG_TAG);
  393. if (EndsStr(InteractiveMsgTag, S) &&
  394. (S.Length() >= 2 * InteractiveMsgTag.Length()))
  395. {
  396. Result = S.Length() - 2 * InteractiveMsgTag.Length() + 1;
  397. while ((Result > 0) && (S.SubString(Result, InteractiveMsgTag.Length()) != InteractiveMsgTag))
  398. {
  399. Result--;
  400. }
  401. }
  402. return Result;
  403. }
  404. //---------------------------------------------------------------------------
  405. UnicodeString RemoveMainInstructionsTag(UnicodeString S)
  406. {
  407. UnicodeString MainInstruction;
  408. if (ExtractMainInstructions(S, MainInstruction))
  409. {
  410. S = MainInstruction + S;
  411. }
  412. return S;
  413. }
  414. //---------------------------------------------------------------------------
  415. UnicodeString UnformatMessage(UnicodeString S)
  416. {
  417. S = RemoveMainInstructionsTag(S);
  418. int InteractiveMsgStart = FindInteractiveMsgStart(S);
  419. if (InteractiveMsgStart > 0)
  420. {
  421. S = S.SubString(1, InteractiveMsgStart - 1);
  422. }
  423. return S;
  424. }
  425. //---------------------------------------------------------------------------
  426. UnicodeString RemoveInteractiveMsgTag(UnicodeString S)
  427. {
  428. int InteractiveMsgStart = FindInteractiveMsgStart(S);
  429. if (InteractiveMsgStart > 0)
  430. {
  431. UnicodeString InteractiveMsgTag = LoadStr(INTERACTIVE_MSG_TAG);
  432. S.Delete(InteractiveMsgStart, InteractiveMsgTag.Length());
  433. S.Delete(S.Length() - InteractiveMsgTag.Length() + 1, InteractiveMsgTag.Length());
  434. }
  435. return S;
  436. }
  437. //---------------------------------------------------------------------------
  438. UnicodeString RemoveEmptyLines(const UnicodeString & S)
  439. {
  440. return
  441. ReplaceStr(
  442. ReplaceStr(S.TrimRight(), L"\n\n", L"\n"),
  443. L"\n \n", L"\n");
  444. }
  445. //---------------------------------------------------------------------------
  446. bool IsNumber(const UnicodeString Str)
  447. {
  448. int Value;
  449. return TryStrToInt(Str, Value);
  450. }
  451. //---------------------------------------------------------------------------
  452. UnicodeString Base64ToUrlSafe(const UnicodeString & S)
  453. {
  454. UnicodeString Result = S;
  455. while (EndsStr(L"=", Result))
  456. {
  457. Result.SetLength(Result.Length() - 1);
  458. }
  459. // See https://en.wikipedia.org/wiki/Base64#Implementations_and_history
  460. Result = ReplaceChar(Result, L'+', L'-');
  461. Result = ReplaceChar(Result, L'/', L'_');
  462. return Result;
  463. }
  464. //---------------------------------------------------------------------------
  465. const wchar_t NormalizedFingerprintSeparator = L'-';
  466. //---------------------------------------------------------------------------
  467. UnicodeString MD5ToUrlSafe(const UnicodeString & S)
  468. {
  469. return ReplaceChar(S, L':', NormalizedFingerprintSeparator);
  470. }
  471. //---------------------------------------------------------------------------
  472. bool SameChecksum(const UnicodeString & AChecksum1, const UnicodeString & AChecksum2, bool Base64)
  473. {
  474. UnicodeString Checksum1(AChecksum1);
  475. UnicodeString Checksum2(AChecksum2);
  476. bool Result;
  477. if (Base64)
  478. {
  479. Checksum1 = Base64ToUrlSafe(Checksum1);
  480. Checksum2 = Base64ToUrlSafe(Checksum2);
  481. Result = SameStr(Checksum1, Checksum2);
  482. }
  483. else
  484. {
  485. Checksum1 = MD5ToUrlSafe(Checksum1);
  486. Checksum2 = MD5ToUrlSafe(Checksum2);
  487. Result = SameText(Checksum1, Checksum2);
  488. }
  489. return Result;
  490. }
  491. //---------------------------------------------------------------------------
  492. UnicodeString __fastcall SystemTemporaryDirectory()
  493. {
  494. UnicodeString TempDir;
  495. TempDir.SetLength(MAX_PATH);
  496. TempDir.SetLength(GetTempPath(MAX_PATH, TempDir.c_str()));
  497. return TempDir;
  498. }
  499. //---------------------------------------------------------------------------
  500. UnicodeString __fastcall GetShellFolderPath(int CSIdl)
  501. {
  502. UnicodeString Result;
  503. wchar_t Path[2 * MAX_PATH + 10] = L"\0";
  504. if (SUCCEEDED(SHGetFolderPath(NULL, CSIdl, NULL, SHGFP_TYPE_CURRENT, Path)))
  505. {
  506. Result = Path;
  507. }
  508. return Result;
  509. }
  510. //---------------------------------------------------------------------------
  511. static UnicodeString __fastcall GetWineHomeFolder()
  512. {
  513. UnicodeString Result;
  514. UnicodeString WineHostHome = GetEnvironmentVariable(L"WINE_HOST_HOME");
  515. if (!WineHostHome.IsEmpty())
  516. {
  517. Result = L"Z:" + FromUnixPath(WineHostHome);
  518. }
  519. else
  520. {
  521. // Should we use WinAPI GetUserName() instead?
  522. UnicodeString UserName = GetEnvironmentVariable(L"USERNAME");
  523. if (!UserName.IsEmpty())
  524. {
  525. Result = L"Z:\\home\\" + UserName;
  526. }
  527. }
  528. if (!DirectoryExists(Result))
  529. {
  530. Result = L"";
  531. }
  532. return Result;
  533. }
  534. //---------------------------------------------------------------------------
  535. UnicodeString __fastcall GetPersonalFolder()
  536. {
  537. UnicodeString Result = GetShellFolderPath(CSIDL_PERSONAL);
  538. if (IsWine())
  539. {
  540. UnicodeString WineHome = GetWineHomeFolder();
  541. if (!WineHome.IsEmpty())
  542. {
  543. // if at least home exists, use it
  544. Result = WineHome;
  545. // but try to go deeper to "Documents"
  546. UnicodeString WineDocuments =
  547. IncludeTrailingBackslash(WineHome) + L"Documents";
  548. if (DirectoryExists(WineDocuments))
  549. {
  550. Result = WineDocuments;
  551. }
  552. }
  553. }
  554. return Result;
  555. }
  556. //---------------------------------------------------------------------------
  557. UnicodeString __fastcall GetDesktopFolder()
  558. {
  559. UnicodeString Result = GetShellFolderPath(CSIDL_DESKTOPDIRECTORY);
  560. if (IsWine())
  561. {
  562. UnicodeString WineHome = GetWineHomeFolder();
  563. if (!WineHome.IsEmpty())
  564. {
  565. UnicodeString WineDesktop =
  566. IncludeTrailingBackslash(WineHome) + L"Desktop";
  567. if (DirectoryExists(WineHome))
  568. {
  569. Result = WineDesktop;
  570. }
  571. }
  572. }
  573. return Result;
  574. }
  575. //---------------------------------------------------------------------------
  576. // Particularly needed when using file name selected by TFilenameEdit,
  577. // as it wraps a path to double-quotes, when there is a space in the path.
  578. UnicodeString __fastcall StripPathQuotes(const UnicodeString Path)
  579. {
  580. if ((Path.Length() >= 2) &&
  581. (Path[1] == L'\"') && (Path[Path.Length()] == L'\"'))
  582. {
  583. return Path.SubString(2, Path.Length() - 2);
  584. }
  585. else
  586. {
  587. return Path;
  588. }
  589. }
  590. //---------------------------------------------------------------------------
  591. UnicodeString __fastcall AddQuotes(UnicodeString Str)
  592. {
  593. if (Str.Pos(L" ") > 0)
  594. {
  595. Str = L"\"" + Str + L"\"";
  596. }
  597. return Str;
  598. }
  599. //---------------------------------------------------------------------------
  600. UnicodeString __fastcall AddPathQuotes(UnicodeString Path)
  601. {
  602. Path = StripPathQuotes(Path);
  603. return AddQuotes(Path);
  604. }
  605. //---------------------------------------------------------------------------
  606. static wchar_t * __fastcall ReplaceChar(
  607. UnicodeString & FileName, wchar_t * InvalidChar, wchar_t InvalidCharsReplacement)
  608. {
  609. int Index = InvalidChar - FileName.c_str() + 1;
  610. if (InvalidCharsReplacement == TokenReplacement)
  611. {
  612. // currently we do not support unicode chars replacement
  613. if (FileName[Index] > 0xFF)
  614. {
  615. EXCEPTION;
  616. }
  617. FileName.Insert(ByteToHex(static_cast<unsigned char>(FileName[Index])), Index + 1);
  618. FileName[Index] = TokenPrefix;
  619. InvalidChar = FileName.c_str() + Index + 2;
  620. }
  621. else
  622. {
  623. FileName[Index] = InvalidCharsReplacement;
  624. InvalidChar = FileName.c_str() + Index;
  625. }
  626. return InvalidChar;
  627. }
  628. //---------------------------------------------------------------------------
  629. // Note similar function MakeValidFileName
  630. UnicodeString __fastcall ValidLocalFileName(UnicodeString FileName)
  631. {
  632. return ValidLocalFileName(FileName, L'_', L"", LocalInvalidChars);
  633. }
  634. //---------------------------------------------------------------------------
  635. UnicodeString __fastcall ValidLocalFileName(
  636. UnicodeString FileName, wchar_t InvalidCharsReplacement,
  637. const UnicodeString & TokenizibleChars, const UnicodeString & LocalInvalidChars)
  638. {
  639. if (InvalidCharsReplacement != NoReplacement)
  640. {
  641. bool ATokenReplacement = (InvalidCharsReplacement == TokenReplacement);
  642. const wchar_t * Chars =
  643. (ATokenReplacement ? TokenizibleChars : LocalInvalidChars).c_str();
  644. wchar_t * InvalidChar = FileName.c_str();
  645. while ((InvalidChar = wcspbrk(InvalidChar, Chars)) != NULL)
  646. {
  647. int Pos = (InvalidChar - FileName.c_str() + 1);
  648. wchar_t Char;
  649. if (ATokenReplacement &&
  650. (*InvalidChar == TokenPrefix) &&
  651. (((FileName.Length() - Pos) <= 1) ||
  652. (((Char = static_cast<wchar_t>(HexToByte(FileName.SubString(Pos + 1, 2)))) == L'\0') ||
  653. (TokenizibleChars.Pos(Char) == 0))))
  654. {
  655. InvalidChar++;
  656. }
  657. else
  658. {
  659. InvalidChar = ReplaceChar(FileName, InvalidChar, InvalidCharsReplacement);
  660. }
  661. }
  662. // Windows trim trailing space or dot, hence we must encode it to preserve it
  663. if (!FileName.IsEmpty() &&
  664. ((FileName[FileName.Length()] == L' ') ||
  665. (FileName[FileName.Length()] == L'.')))
  666. {
  667. ReplaceChar(FileName, FileName.c_str() + FileName.Length() - 1, InvalidCharsReplacement);
  668. }
  669. if (IsReservedName(FileName))
  670. {
  671. int P = FileName.Pos(".");
  672. if (P == 0)
  673. {
  674. P = FileName.Length() + 1;
  675. }
  676. FileName.Insert(L"%00", P);
  677. }
  678. }
  679. return FileName;
  680. }
  681. //---------------------------------------------------------------------------
  682. void __fastcall SplitCommand(UnicodeString Command, UnicodeString &Program,
  683. UnicodeString & Params, UnicodeString & Dir)
  684. {
  685. Command = Command.Trim();
  686. Params = L"";
  687. Dir = L"";
  688. if (!Command.IsEmpty() && (Command[1] == L'\"'))
  689. {
  690. Command.Delete(1, 1);
  691. int P = Command.Pos(L'"');
  692. if (P)
  693. {
  694. Program = Command.SubString(1, P-1).Trim();
  695. Params = Command.SubString(P + 1, Command.Length() - P).Trim();
  696. }
  697. else
  698. {
  699. throw Exception(FMTLOAD(INVALID_SHELL_COMMAND, (L"\"" + Command)));
  700. }
  701. }
  702. else
  703. {
  704. int P = Command.Pos(L" ");
  705. if (P)
  706. {
  707. Program = Command.SubString(1, P).Trim();
  708. Params = Command.SubString(P + 1, Command.Length() - P).Trim();
  709. }
  710. else
  711. {
  712. Program = Command;
  713. }
  714. }
  715. int B = Program.LastDelimiter(L"\\/");
  716. if (B)
  717. {
  718. Dir = Program.SubString(1, B).Trim();
  719. }
  720. }
  721. //---------------------------------------------------------------------------
  722. UnicodeString __fastcall ExtractProgram(UnicodeString Command)
  723. {
  724. UnicodeString Program;
  725. UnicodeString Params;
  726. UnicodeString Dir;
  727. SplitCommand(Command, Program, Params, Dir);
  728. return Program;
  729. }
  730. //---------------------------------------------------------------------------
  731. UnicodeString __fastcall ExtractProgramName(UnicodeString Command)
  732. {
  733. UnicodeString Name = ExtractFileName(ExtractProgram(Command));
  734. int Dot = Name.LastDelimiter(L".");
  735. if (Dot > 0)
  736. {
  737. Name = Name.SubString(1, Dot - 1);
  738. }
  739. return Name;
  740. }
  741. //---------------------------------------------------------------------------
  742. UnicodeString __fastcall FormatCommand(UnicodeString Program, UnicodeString Params)
  743. {
  744. Program = Program.Trim();
  745. Params = Params.Trim();
  746. if (!Params.IsEmpty()) Params = L" " + Params;
  747. Program = AddQuotes(Program);
  748. return Program + Params;
  749. }
  750. //---------------------------------------------------------------------------
  751. const wchar_t ShellCommandFileNamePattern[] = L"!.!";
  752. //---------------------------------------------------------------------------
  753. void __fastcall ReformatFileNameCommand(UnicodeString & Command)
  754. {
  755. if (!Command.IsEmpty())
  756. {
  757. UnicodeString Program, Params, Dir;
  758. SplitCommand(Command, Program, Params, Dir);
  759. if (Params.Pos(ShellCommandFileNamePattern) == 0)
  760. {
  761. Params = Params + (Params.IsEmpty() ? L"" : L" ") + ShellCommandFileNamePattern;
  762. }
  763. Command = FormatCommand(Program, Params);
  764. }
  765. }
  766. //---------------------------------------------------------------------------
  767. UnicodeString __fastcall ExpandFileNameCommand(const UnicodeString Command,
  768. const UnicodeString FileName)
  769. {
  770. return AnsiReplaceStr(Command, ShellCommandFileNamePattern,
  771. AddPathQuotes(FileName));
  772. }
  773. //---------------------------------------------------------------------------
  774. UnicodeString __fastcall EscapeParam(const UnicodeString & Param)
  775. {
  776. // Make sure this won't break RTF syntax
  777. return ReplaceStr(Param, L"\"", L"\"\"");
  778. }
  779. //---------------------------------------------------------------------------
  780. UnicodeString __fastcall EscapePuttyCommandParam(UnicodeString Param)
  781. {
  782. bool Space = false;
  783. for (int i = 1; i <= Param.Length(); i++)
  784. {
  785. switch (Param[i])
  786. {
  787. case L'"':
  788. Param.Insert(L"\\", i);
  789. i++;
  790. break;
  791. case L' ':
  792. Space = true;
  793. break;
  794. case L'\\':
  795. int i2 = i;
  796. while ((i2 <= Param.Length()) && (Param[i2] == L'\\'))
  797. {
  798. i2++;
  799. }
  800. if ((i2 <= Param.Length()) && (Param[i2] == L'"'))
  801. {
  802. while (Param[i] == L'\\')
  803. {
  804. Param.Insert(L"\\", i);
  805. i += 2;
  806. }
  807. i--;
  808. }
  809. break;
  810. }
  811. }
  812. if (Space)
  813. {
  814. Param = L"\"" + Param + L'"';
  815. }
  816. return Param;
  817. }
  818. //---------------------------------------------------------------------------
  819. UnicodeString __fastcall StringsToParams(TStrings * Strings)
  820. {
  821. UnicodeString Result;
  822. for (int Index = 0; Index < Strings->Count; Index++)
  823. {
  824. UnicodeString Name = Strings->Names[Index];
  825. UnicodeString Value = Strings->ValueFromIndex[Index];
  826. // Do not quote if it is all-numeric
  827. if (IntToStr(StrToIntDef(Value, -1)) != Value)
  828. {
  829. Value = FORMAT(L"\"%s\"", (EscapeParam(Value)));
  830. }
  831. Result += FORMAT(L" %s=%s", (Name, Value));
  832. }
  833. return Result;
  834. }
  835. //---------------------------------------------------------------------------
  836. UnicodeString __fastcall ExpandEnvironmentVariables(const UnicodeString & Str)
  837. {
  838. UnicodeString Buf;
  839. unsigned int Size = 1024;
  840. Buf.SetLength(Size);
  841. Buf.Unique();
  842. unsigned int Len = ExpandEnvironmentStrings(Str.c_str(), Buf.c_str(), Size);
  843. if (Len > Size)
  844. {
  845. Buf.SetLength(Len);
  846. Buf.Unique();
  847. ExpandEnvironmentStrings(Str.c_str(), Buf.c_str(), Len);
  848. }
  849. PackStr(Buf);
  850. return Buf;
  851. }
  852. //---------------------------------------------------------------------------
  853. bool __fastcall IsPathToSameFile(const UnicodeString & Path1, const UnicodeString & Path2)
  854. {
  855. UnicodeString ShortPath1 = ExtractShortPathName(Path1);
  856. UnicodeString ShortPath2 = ExtractShortPathName(Path2);
  857. bool Result;
  858. // ExtractShortPathName returns empty string if file does not exist
  859. if (ShortPath1.IsEmpty() || ShortPath2.IsEmpty())
  860. {
  861. Result = AnsiSameText(Path1, Path2);
  862. }
  863. else
  864. {
  865. Result = AnsiSameText(ShortPath1, ShortPath2);
  866. }
  867. return Result;
  868. }
  869. //---------------------------------------------------------------------------
  870. bool __fastcall SamePaths(const UnicodeString & Path1, const UnicodeString & Path2)
  871. {
  872. // TODO: ExpandUNCFileName
  873. return AnsiSameText(IncludeTrailingBackslash(Path1), IncludeTrailingBackslash(Path2));
  874. }
  875. //---------------------------------------------------------------------------
  876. int __fastcall CompareLogicalText(
  877. const UnicodeString & S1, const UnicodeString & S2, bool NaturalOrderNumericalSorting)
  878. {
  879. // Keep in sync with CompareLogicalTextPas
  880. int Result;
  881. if (NaturalOrderNumericalSorting)
  882. {
  883. Result = StrCmpLogicalW(S1.c_str(), S2.c_str());
  884. }
  885. else
  886. {
  887. Result = lstrcmpi(S1.c_str(), S2.c_str());
  888. }
  889. // For deterministics results
  890. if (Result == 0)
  891. {
  892. Result = lstrcmp(S1.c_str(), S2.c_str());
  893. }
  894. return Result;
  895. }
  896. //---------------------------------------------------------------------------
  897. int __fastcall CompareNumber(__int64 Value1, __int64 Value2)
  898. {
  899. int Result;
  900. if (Value1 < Value2)
  901. {
  902. Result = -1;
  903. }
  904. else if (Value1 == Value2)
  905. {
  906. Result = 0;
  907. }
  908. else
  909. {
  910. Result = 1;
  911. }
  912. return Result;
  913. }
  914. //---------------------------------------------------------------------------
  915. bool ContainsTextSemiCaseSensitive(const UnicodeString & Text, const UnicodeString & SubText)
  916. {
  917. bool Result;
  918. if (AnsiLowerCase(SubText) == SubText)
  919. {
  920. Result = ContainsText(Text, SubText);
  921. }
  922. else
  923. {
  924. Result = ContainsStr(Text, SubText);
  925. }
  926. return Result;
  927. }
  928. //---------------------------------------------------------------------------
  929. bool __fastcall IsReservedName(UnicodeString FileName)
  930. {
  931. int P = FileName.Pos(L".");
  932. int Len = (P > 0) ? P - 1 : FileName.Length();
  933. if ((Len == 3) || (Len == 4))
  934. {
  935. if (P > 0)
  936. {
  937. FileName.SetLength(P - 1);
  938. }
  939. static UnicodeString Reserved[] = {
  940. L"CON", L"PRN", L"AUX", L"NUL",
  941. L"COM1", L"COM2", L"COM3", L"COM4", L"COM5", L"COM6", L"COM7", L"COM8", L"COM9",
  942. L"LPT1", L"LPT2", L"LPT3", L"LPT4", L"LPT5", L"LPT6", L"LPT7", L"LPT8", L"LPT9" };
  943. for (unsigned int Index = 0; Index < LENOF(Reserved); Index++)
  944. {
  945. if (SameText(FileName, Reserved[Index]))
  946. {
  947. return true;
  948. }
  949. }
  950. }
  951. return false;
  952. }
  953. //---------------------------------------------------------------------------
  954. // ApiPath support functions
  955. // Inspired by
  956. // https://stackoverflow.com/q/18580945/850848
  957. // This can be reimplemented using PathCchCanonicalizeEx on Windows 8 and later
  958. enum PATH_PREFIX_TYPE
  959. {
  960. PPT_UNKNOWN,
  961. PPT_ABSOLUTE, //Found absolute path that is none of the other types
  962. PPT_UNC, //Found \\server\share\ prefix
  963. PPT_LONG_UNICODE, //Found \\?\ prefix
  964. PPT_LONG_UNICODE_UNC, //Found \\?\UNC\ prefix
  965. };
  966. //---------------------------------------------------------------------------
  967. static int __fastcall PathRootLength(UnicodeString Path)
  968. {
  969. // Correction for PathSkipRoot API
  970. // Replace all /'s with \'s because PathSkipRoot can't handle /'s
  971. UnicodeString Result = ReplaceChar(Path, L'/', L'\\');
  972. // Now call the API
  973. LPCTSTR Buffer = PathSkipRoot(Result.c_str());
  974. return (Buffer != NULL) ? (Buffer - Result.c_str()) : -1;
  975. }
  976. //---------------------------------------------------------------------------
  977. static bool __fastcall PathIsRelative_CorrectedForMicrosoftStupidity(UnicodeString Path)
  978. {
  979. // Correction for PathIsRelative API
  980. // Replace all /'s with \'s because PathIsRelative can't handle /'s
  981. UnicodeString Result = ReplaceChar(Path, L'/', L'\\');
  982. //Now call the API
  983. return PathIsRelative(Result.c_str());
  984. }
  985. //---------------------------------------------------------------------------
  986. static int __fastcall GetOffsetAfterPathRoot(UnicodeString Path, PATH_PREFIX_TYPE & PrefixType)
  987. {
  988. // Checks if 'pPath' begins with the drive, share, prefix, etc
  989. // EXAMPLES:
  990. // Path Return: Points at: PrefixType:
  991. // Relative\Folder\File.txt 0 Relative\Folder\File.txt PPT_UNKNOWN
  992. // \RelativeToRoot\Folder 1 RelativeToRoot\Folder PPT_ABSOLUTE
  993. // C:\Windows\Folder 3 Windows\Folder PPT_ABSOLUTE
  994. // \\server\share\Desktop 15 Desktop PPT_UNC
  995. // \\?\C:\Windows\Folder 7 Windows\Folder PPT_LONG_UNICODE
  996. // \\?\UNC\server\share\Desktop 21 Desktop PPT_LONG_UNICODE_UNC
  997. // RETURN:
  998. // = Index in 'pPath' after the root, or
  999. // = 0 if no root was found
  1000. int Result = 0;
  1001. PrefixType = PPT_UNKNOWN;
  1002. if (!Path.IsEmpty())
  1003. {
  1004. int Len = Path.Length();
  1005. bool WinXPOnly = !IsWinVista();
  1006. // The PathSkipRoot() API doesn't work correctly on Windows XP
  1007. if (!WinXPOnly)
  1008. {
  1009. // Works since Vista and up, but still needs correction :)
  1010. int RootLength = PathRootLength(Path);
  1011. if (RootLength >= 0)
  1012. {
  1013. Result = RootLength + 1;
  1014. }
  1015. }
  1016. // Now determine the type of prefix
  1017. int IndCheckUNC = -1;
  1018. if ((Len >= 8) &&
  1019. (Path[1] == L'\\' || Path[1] == L'/') &&
  1020. (Path[2] == L'\\' || Path[2] == L'/') &&
  1021. (Path[3] == L'?') &&
  1022. (Path[4] == L'\\' || Path[4] == L'/') &&
  1023. (Path[5] == L'U' || Path[5] == L'u') &&
  1024. (Path[6] == L'N' || Path[6] == L'n') &&
  1025. (Path[7] == L'C' || Path[7] == L'c') &&
  1026. (Path[8] == L'\\' || Path[8] == L'/'))
  1027. {
  1028. // Found \\?\UNC\ prefix
  1029. PrefixType = PPT_LONG_UNICODE_UNC;
  1030. if (WinXPOnly)
  1031. {
  1032. //For older OS
  1033. Result += 8;
  1034. }
  1035. //Check for UNC share later
  1036. IndCheckUNC = 8;
  1037. }
  1038. else if ((Len >= 4) &&
  1039. (Path[1] == L'\\' || Path[1] == L'/') &&
  1040. (Path[2] == L'\\' || Path[2] == L'/') &&
  1041. (Path[3] == L'?') &&
  1042. (Path[4] == L'\\' || Path[4] == L'/'))
  1043. {
  1044. // Found \\?\ prefix
  1045. PrefixType = PPT_LONG_UNICODE;
  1046. if (WinXPOnly)
  1047. {
  1048. //For older OS
  1049. Result += 4;
  1050. }
  1051. }
  1052. else if ((Len >= 2) &&
  1053. (Path[1] == L'\\' || Path[1] == L'/') &&
  1054. (Path[2] == L'\\' || Path[2] == L'/'))
  1055. {
  1056. // Check for UNC share later
  1057. IndCheckUNC = 2;
  1058. }
  1059. if (IndCheckUNC >= 0)
  1060. {
  1061. // Check for UNC, i.e. \\server\share\ part
  1062. int Index = IndCheckUNC;
  1063. for (int SkipSlashes = 2; SkipSlashes > 0; SkipSlashes--)
  1064. {
  1065. for(; Index <= Len; Index++)
  1066. {
  1067. TCHAR z = Path[Index];
  1068. if ((z == L'\\') || (z == L'/') || (Index >= Len))
  1069. {
  1070. Index++;
  1071. if (SkipSlashes == 1)
  1072. {
  1073. if (PrefixType == PPT_UNKNOWN)
  1074. {
  1075. PrefixType = PPT_UNC;
  1076. }
  1077. if (WinXPOnly)
  1078. {
  1079. //For older OS
  1080. Result = Index;
  1081. }
  1082. }
  1083. break;
  1084. }
  1085. }
  1086. }
  1087. }
  1088. if (WinXPOnly)
  1089. {
  1090. // Only if we didn't determine any other type
  1091. if (PrefixType == PPT_UNKNOWN)
  1092. {
  1093. if (!PathIsRelative_CorrectedForMicrosoftStupidity(Path.SubString(Result, Path.Length() - Result + 1)))
  1094. {
  1095. PrefixType = PPT_ABSOLUTE;
  1096. }
  1097. }
  1098. // For older OS only
  1099. int RootLength = PathRootLength(Path.SubString(Result, Path.Length() - Result + 1));
  1100. if (RootLength >= 0)
  1101. {
  1102. Result = RootLength + 1;
  1103. }
  1104. }
  1105. else
  1106. {
  1107. // Only if we didn't determine any other type
  1108. if (PrefixType == PPT_UNKNOWN)
  1109. {
  1110. if (!PathIsRelative_CorrectedForMicrosoftStupidity(Path))
  1111. {
  1112. PrefixType = PPT_ABSOLUTE;
  1113. }
  1114. }
  1115. }
  1116. }
  1117. return Result;
  1118. }
  1119. //---------------------------------------------------------------------------
  1120. UnicodeString __fastcall MakeUnicodeLargePath(UnicodeString Path)
  1121. {
  1122. // Convert path from 'into a larger Unicode path, that allows up to 32,767 character length
  1123. UnicodeString Result;
  1124. if (!Path.IsEmpty())
  1125. {
  1126. // Determine the type of the existing prefix
  1127. PATH_PREFIX_TYPE PrefixType;
  1128. GetOffsetAfterPathRoot(Path, PrefixType);
  1129. // Assume path to be without change
  1130. Result = Path;
  1131. switch (PrefixType)
  1132. {
  1133. case PPT_ABSOLUTE:
  1134. {
  1135. // First we need to check if its an absolute path relative to the root
  1136. bool AddPrefix = true;
  1137. if ((Path.Length() >= 1) &&
  1138. ((Path[1] == L'\\') || (Path[1] == L'/')))
  1139. {
  1140. AddPrefix = FALSE;
  1141. // Get current root path
  1142. UnicodeString CurrentDir = GetCurrentDir();
  1143. PATH_PREFIX_TYPE PrefixType2; // unused
  1144. int Following = GetOffsetAfterPathRoot(CurrentDir, PrefixType2);
  1145. if (Following > 0)
  1146. {
  1147. AddPrefix = true;
  1148. Result = CurrentDir.SubString(1, Following - 1) + Result.SubString(2, Result.Length() - 1);
  1149. }
  1150. }
  1151. if (AddPrefix)
  1152. {
  1153. // Add \\?\ prefix
  1154. Result = L"\\\\?\\" + Result;
  1155. }
  1156. }
  1157. break;
  1158. case PPT_UNC:
  1159. // First we need to remove the opening slashes for UNC share
  1160. if ((Result.Length() >= 2) &&
  1161. ((Result[1] == L'\\') || (Result[1] == L'/')) &&
  1162. ((Result[2] == L'\\') || (Result[2] == L'/')))
  1163. {
  1164. Result = Result.SubString(3, Result.Length() - 2);
  1165. }
  1166. // Add \\?\UNC\ prefix
  1167. Result = L"\\\\?\\UNC\\" + Result;
  1168. break;
  1169. case PPT_LONG_UNICODE:
  1170. case PPT_LONG_UNICODE_UNC:
  1171. // nothing to do
  1172. break;
  1173. }
  1174. }
  1175. return Result;
  1176. }
  1177. //---------------------------------------------------------------------------
  1178. UnicodeString __fastcall ApiPath(UnicodeString Path)
  1179. {
  1180. UnicodeString Drive = ExtractFileDrive(Path);
  1181. // This may match even a path like "C:" or "\\server\\share", but we do not really care
  1182. if (Drive.IsEmpty() || (Path.SubString(Drive.Length() + 1, 1) != L"\\"))
  1183. {
  1184. Path = ExpandFileName(Path);
  1185. }
  1186. // Max path for directories is 12 characters shorter than max path for files
  1187. if (Path.Length() >= MAX_PATH - 12)
  1188. {
  1189. if (Configuration != NULL)
  1190. {
  1191. Configuration->Usage->Inc(L"LongPath");
  1192. }
  1193. Path = MakeUnicodeLargePath(Path);
  1194. }
  1195. return Path;
  1196. }
  1197. //---------------------------------------------------------------------------
  1198. UnicodeString __fastcall DisplayableStr(const RawByteString & Str)
  1199. {
  1200. bool Displayable = true;
  1201. int Index = 1;
  1202. while ((Index <= Str.Length()) && Displayable)
  1203. {
  1204. if (((Str[Index] < '\x20') || (static_cast<unsigned char>(Str[Index]) >= static_cast<unsigned char>('\x80'))) &&
  1205. (Str[Index] != '\n') && (Str[Index] != '\r') && (Str[Index] != '\t') && (Str[Index] != '\b'))
  1206. {
  1207. Displayable = false;
  1208. }
  1209. Index++;
  1210. }
  1211. UnicodeString Result;
  1212. if (Displayable)
  1213. {
  1214. Result = L"\"";
  1215. for (int Index = 1; Index <= Str.Length(); Index++)
  1216. {
  1217. switch (Str[Index])
  1218. {
  1219. case '\n':
  1220. Result += L"\\n";
  1221. break;
  1222. case '\r':
  1223. Result += L"\\r";
  1224. break;
  1225. case '\t':
  1226. Result += L"\\t";
  1227. break;
  1228. case '\b':
  1229. Result += L"\\b";
  1230. break;
  1231. case '\\':
  1232. Result += L"\\\\";
  1233. break;
  1234. case '"':
  1235. Result += L"\\\"";
  1236. break;
  1237. default:
  1238. Result += wchar_t(Str[Index]);
  1239. break;
  1240. }
  1241. }
  1242. Result += L"\"";
  1243. }
  1244. else
  1245. {
  1246. Result = L"0x" + BytesToHex(Str);
  1247. }
  1248. return Result;
  1249. }
  1250. //---------------------------------------------------------------------------
  1251. UnicodeString __fastcall ByteToHex(unsigned char B, bool UpperCase)
  1252. {
  1253. static wchar_t UpperDigits[] = L"0123456789ABCDEF";
  1254. static wchar_t LowerDigits[] = L"0123456789abcdef";
  1255. const wchar_t * Digits = (UpperCase ? UpperDigits : LowerDigits);
  1256. UnicodeString Result;
  1257. Result.SetLength(2);
  1258. Result[1] = Digits[(B & 0xF0) >> 4];
  1259. Result[2] = Digits[(B & 0x0F) >> 0];
  1260. return Result;
  1261. }
  1262. //---------------------------------------------------------------------------
  1263. UnicodeString __fastcall BytesToHex(const unsigned char * B, size_t Length, bool UpperCase, wchar_t Separator)
  1264. {
  1265. UnicodeString Result;
  1266. for (size_t i = 0; i < Length; i++)
  1267. {
  1268. Result += ByteToHex(B[i], UpperCase);
  1269. if ((Separator != L'\0') && (i < Length - 1))
  1270. {
  1271. Result += Separator;
  1272. }
  1273. }
  1274. return Result;
  1275. }
  1276. //---------------------------------------------------------------------------
  1277. UnicodeString __fastcall BytesToHex(RawByteString Str, bool UpperCase, wchar_t Separator)
  1278. {
  1279. return BytesToHex(reinterpret_cast<const unsigned char *>(Str.c_str()), Str.Length(), UpperCase, Separator);
  1280. }
  1281. //---------------------------------------------------------------------------
  1282. UnicodeString __fastcall CharToHex(wchar_t Ch, bool UpperCase)
  1283. {
  1284. return BytesToHex(reinterpret_cast<const unsigned char *>(&Ch), sizeof(Ch), UpperCase);
  1285. }
  1286. //---------------------------------------------------------------------------
  1287. RawByteString __fastcall HexToBytes(const UnicodeString Hex)
  1288. {
  1289. static UnicodeString Digits = L"0123456789ABCDEF";
  1290. RawByteString Result;
  1291. int L, P1, P2;
  1292. L = Hex.Length();
  1293. if (L % 2 == 0)
  1294. {
  1295. for (int i = 1; i <= Hex.Length(); i += 2)
  1296. {
  1297. P1 = Digits.Pos((wchar_t)toupper(Hex[i]));
  1298. P2 = Digits.Pos((wchar_t)toupper(Hex[i + 1]));
  1299. if (P1 <= 0 || P2 <= 0)
  1300. {
  1301. Result = L"";
  1302. break;
  1303. }
  1304. else
  1305. {
  1306. Result += static_cast<char>((P1 - 1) * 16 + P2 - 1);
  1307. }
  1308. }
  1309. }
  1310. return Result;
  1311. }
  1312. //---------------------------------------------------------------------------
  1313. unsigned char __fastcall HexToByte(const UnicodeString Hex)
  1314. {
  1315. static UnicodeString Digits = L"0123456789ABCDEF";
  1316. DebugAssert(Hex.Length() == 2);
  1317. int P1 = Digits.Pos((wchar_t)toupper(Hex[1]));
  1318. int P2 = Digits.Pos((wchar_t)toupper(Hex[2]));
  1319. return
  1320. static_cast<unsigned char>(((P1 <= 0) || (P2 <= 0)) ? 0 : (((P1 - 1) << 4) + (P2 - 1)));
  1321. }
  1322. //---------------------------------------------------------------------------
  1323. bool __fastcall IsLowerCaseLetter(wchar_t Ch)
  1324. {
  1325. return (Ch >= 'a') && (Ch <= 'z');
  1326. }
  1327. //---------------------------------------------------------------------------
  1328. bool __fastcall IsUpperCaseLetter(wchar_t Ch)
  1329. {
  1330. return (Ch >= 'A') && (Ch <= 'Z');
  1331. }
  1332. //---------------------------------------------------------------------------
  1333. bool __fastcall IsLetter(wchar_t Ch)
  1334. {
  1335. return IsLowerCaseLetter(Ch) || IsUpperCaseLetter(Ch);
  1336. }
  1337. //---------------------------------------------------------------------------
  1338. bool __fastcall IsDigit(wchar_t Ch)
  1339. {
  1340. return (Ch >= '0') && (Ch <= '9');
  1341. }
  1342. //---------------------------------------------------------------------------
  1343. bool __fastcall IsHex(wchar_t Ch)
  1344. {
  1345. return
  1346. IsDigit(Ch) ||
  1347. ((Ch >= 'A') && (Ch <= 'F')) ||
  1348. ((Ch >= 'a') && (Ch <= 'f'));
  1349. }
  1350. //---------------------------------------------------------------------------
  1351. TSearchRecSmart::TSearchRecSmart()
  1352. {
  1353. FLastWriteTimeSource.dwLowDateTime = 0;
  1354. FLastWriteTimeSource.dwHighDateTime = 0;
  1355. }
  1356. //---------------------------------------------------------------------------
  1357. void TSearchRecSmart::Clear()
  1358. {
  1359. Size = 0;
  1360. Attr = 0;
  1361. Name = TFileName();
  1362. ExcludeAttr = 0;
  1363. FindHandle = 0;
  1364. memset(&FindData, 0, sizeof(FindData));
  1365. FLastWriteTimeSource.dwLowDateTime = 0;
  1366. FLastWriteTimeSource.dwHighDateTime = 0;
  1367. }
  1368. //---------------------------------------------------------------------------
  1369. TDateTime TSearchRecSmart::GetLastWriteTime() const
  1370. {
  1371. if ((FindData.ftLastWriteTime.dwLowDateTime != FLastWriteTimeSource.dwLowDateTime) ||
  1372. (FindData.ftLastWriteTime.dwHighDateTime != FLastWriteTimeSource.dwHighDateTime))
  1373. {
  1374. FLastWriteTimeSource = FindData.ftLastWriteTime;
  1375. FLastWriteTime = FileTimeToDateTime(FLastWriteTimeSource);
  1376. }
  1377. return FLastWriteTime;
  1378. }
  1379. //---------------------------------------------------------------------------
  1380. bool TSearchRecSmart::IsRealFile() const
  1381. {
  1382. return ::IsRealFile(Name);
  1383. }
  1384. //---------------------------------------------------------------------------
  1385. bool TSearchRecSmart::IsDirectory() const
  1386. {
  1387. return FLAGSET(Attr, faDirectory);
  1388. }
  1389. //---------------------------------------------------------------------------
  1390. bool TSearchRecSmart::IsHidden() const
  1391. {
  1392. return FLAGSET(Attr, faHidden);
  1393. }
  1394. //---------------------------------------------------------------------------
  1395. TSearchRecOwned::~TSearchRecOwned()
  1396. {
  1397. if (Opened)
  1398. {
  1399. FindClose(*this);
  1400. }
  1401. }
  1402. //---------------------------------------------------------------------------
  1403. void TSearchRecOwned::Close()
  1404. {
  1405. FindClose(*this);
  1406. Opened = false;
  1407. }
  1408. //---------------------------------------------------------------------------
  1409. int __fastcall FindCheck(int Result, const UnicodeString & Path)
  1410. {
  1411. if ((Result != ERROR_SUCCESS) &&
  1412. (Result != ERROR_FILE_NOT_FOUND) &&
  1413. (Result != ERROR_NO_MORE_FILES))
  1414. {
  1415. throw EOSExtException(FMTLOAD(FIND_FILE_ERROR, (Path)), Result);
  1416. }
  1417. return Result;
  1418. }
  1419. //---------------------------------------------------------------------------
  1420. int __fastcall FindFirstUnchecked(const UnicodeString & Path, int Attr, TSearchRecChecked & F)
  1421. {
  1422. F.Path = Path;
  1423. int Result = FindFirst(ApiPath(Path), Attr, F);
  1424. F.Opened = (Result == 0);
  1425. return Result;
  1426. }
  1427. //---------------------------------------------------------------------------
  1428. int __fastcall FindFirstChecked(const UnicodeString & Path, int Attr, TSearchRecChecked & F)
  1429. {
  1430. int Result = FindFirstUnchecked(Path, Attr, F);
  1431. return FindCheck(Result, F.Path);
  1432. }
  1433. //---------------------------------------------------------------------------
  1434. // Equivalent to FindNext, just to complement to FindFirstUnchecked
  1435. int __fastcall FindNextUnchecked(TSearchRecChecked & F)
  1436. {
  1437. return FindNext(F);
  1438. }
  1439. //---------------------------------------------------------------------------
  1440. // It can make sense to use FindNextChecked, even if unchecked FindFirst is used.
  1441. // I.e. even if we do not care that FindFirst failed, if FindNext
  1442. // fails after successful FindFirst, it means some terrible problem
  1443. int __fastcall FindNextChecked(TSearchRecChecked & F)
  1444. {
  1445. return FindCheck(FindNextUnchecked(F), F.Path);
  1446. }
  1447. //---------------------------------------------------------------------------
  1448. bool __fastcall FileSearchRec(const UnicodeString FileName, TSearchRec & Rec)
  1449. {
  1450. int FindAttrs = faReadOnly | faHidden | faSysFile | faDirectory | faArchive;
  1451. bool Result = (FindFirst(ApiPath(FileName), FindAttrs, Rec) == 0);
  1452. if (Result)
  1453. {
  1454. FindClose(Rec);
  1455. }
  1456. return Result;
  1457. }
  1458. //---------------------------------------------------------------------------
  1459. void CopySearchRec(const TSearchRec & Source, TSearchRec & Dest)
  1460. {
  1461. // Strangely issues a compiler warning (W8111 due to TSearchRec::Time), when used in Script.cpp, but not here.
  1462. Dest = Source;
  1463. }
  1464. //---------------------------------------------------------------------------
  1465. bool __fastcall IsRealFile(const UnicodeString & FileName)
  1466. {
  1467. return (FileName != THISDIRECTORY) && (FileName != PARENTDIRECTORY);
  1468. }
  1469. //---------------------------------------------------------------------------
  1470. UnicodeString GetOSInfo()
  1471. {
  1472. UnicodeString Result = WindowsVersionLong();
  1473. AddToList(Result, WindowsProductName(), L" - ");
  1474. return Result;
  1475. }
  1476. //---------------------------------------------------------------------------
  1477. UnicodeString GetEnvironmentInfo()
  1478. {
  1479. UnicodeString Result = FORMAT(L"WinSCP %s (OS %s)", (Configuration->VersionStr, GetOSInfo()));
  1480. return Result;
  1481. }
  1482. //---------------------------------------------------------------------------
  1483. void __fastcall ProcessLocalDirectory(UnicodeString DirName,
  1484. TProcessLocalFileEvent CallBackFunc, void * Param,
  1485. int FindAttrs)
  1486. {
  1487. DebugAssert(CallBackFunc);
  1488. if (FindAttrs < 0)
  1489. {
  1490. FindAttrs = faReadOnly | faHidden | faSysFile | faDirectory | faArchive;
  1491. }
  1492. DirName = IncludeTrailingBackslash(DirName);
  1493. TSearchRecOwned SearchRec;
  1494. if (FindFirstChecked(DirName + L"*.*", FindAttrs, SearchRec) == 0)
  1495. {
  1496. do
  1497. {
  1498. if (SearchRec.IsRealFile())
  1499. {
  1500. CallBackFunc(DirName + SearchRec.Name, SearchRec, Param);
  1501. }
  1502. } while (FindNextChecked(SearchRec) == 0);
  1503. }
  1504. }
  1505. //---------------------------------------------------------------------------
  1506. int __fastcall FileGetAttrFix(const UnicodeString FileName)
  1507. {
  1508. // The default for FileGetAttr is to follow links
  1509. bool FollowLink = true;
  1510. // WORKAROUND:
  1511. // But the FileGetAttr when called for link with FollowLink set will always fail
  1512. // as it calls InternalGetFileNameFromSymLink, which test for CheckWin32Version(6, 0)
  1513. if (!IsWinVista())
  1514. {
  1515. FollowLink = false;
  1516. }
  1517. int Result = FileGetAttr(FileName, FollowLink);
  1518. if (Result < 0)
  1519. {
  1520. // When referring to files in some special symlinked locations
  1521. // (like a deduplicated drive or a commvault archive), the first call to GetFileAttributes fails.
  1522. Result = FileGetAttr(FileName, FollowLink);
  1523. }
  1524. return Result;
  1525. }
  1526. //---------------------------------------------------------------------------
  1527. TDateTime __fastcall EncodeDateVerbose(Word Year, Word Month, Word Day)
  1528. {
  1529. try
  1530. {
  1531. TDateTime DateTime = EncodeDate(Year, Month, Day);
  1532. return DateTime;
  1533. }
  1534. catch (EConvertError & E)
  1535. {
  1536. throw EConvertError(FORMAT(L"%s [%d-%d-%d]", (E.Message, int(Year), int(Month), int(Day))));
  1537. }
  1538. }
  1539. //---------------------------------------------------------------------------
  1540. TDateTime __fastcall EncodeTimeVerbose(Word Hour, Word Min, Word Sec, Word MSec)
  1541. {
  1542. try
  1543. {
  1544. TDateTime DateTime = EncodeTime(Hour, Min, Sec, MSec);
  1545. return DateTime;
  1546. }
  1547. catch (EConvertError & E)
  1548. {
  1549. throw EConvertError(FORMAT(L"%s [%d:%d:%d.%d]", (E.Message, int(Hour), int(Min), int(Sec), int(MSec))));
  1550. }
  1551. }
  1552. //---------------------------------------------------------------------------
  1553. TDateTime __fastcall SystemTimeToDateTimeVerbose(const SYSTEMTIME & SystemTime)
  1554. {
  1555. try
  1556. {
  1557. TDateTime DateTime = SystemTimeToDateTime(SystemTime);
  1558. return DateTime;
  1559. }
  1560. catch (EConvertError & E)
  1561. {
  1562. throw EConvertError(FORMAT(L"%s [%d-%2.2d-%2.2d %2.2d:%2.2d:%2.2d.%3.3d]", (E.Message, int(SystemTime.wYear), int(SystemTime.wMonth), int(SystemTime.wDay), int(SystemTime.wHour), int(SystemTime.wMinute), int(SystemTime.wSecond), int(SystemTime.wMilliseconds))));
  1563. }
  1564. }
  1565. //---------------------------------------------------------------------------
  1566. struct TDateTimeParams
  1567. {
  1568. TDateTime UnixEpoch;
  1569. double BaseDifference;
  1570. long BaseDifferenceSec;
  1571. // All Current* are actually global, not per-year and
  1572. // are valid for Year 0 (current) only
  1573. double CurrentDaylightDifference;
  1574. long CurrentDaylightDifferenceSec;
  1575. double CurrentDifference;
  1576. long CurrentDifferenceSec;
  1577. double StandardDifference;
  1578. long StandardDifferenceSec;
  1579. double DaylightDifference;
  1580. long DaylightDifferenceSec;
  1581. SYSTEMTIME SystemStandardDate;
  1582. SYSTEMTIME SystemDaylightDate;
  1583. TDateTime StandardDate;
  1584. TDateTime DaylightDate;
  1585. UnicodeString StandardName;
  1586. UnicodeString DaylightName;
  1587. // This is actually global, not per-year
  1588. bool DaylightHack;
  1589. bool HasDST() const
  1590. {
  1591. // On some systems it occurs that StandardDate is unset, while
  1592. // DaylightDate is set. MSDN states that this is invalid and
  1593. // should be treated as if there is no daylight saving.
  1594. // So check both.
  1595. return
  1596. (SystemStandardDate.wMonth != 0) &&
  1597. (SystemDaylightDate.wMonth != 0);
  1598. }
  1599. bool SummerDST() const
  1600. {
  1601. return HasDST() && (DaylightDate < StandardDate);
  1602. }
  1603. };
  1604. typedef std::map<int, TDateTimeParams> TYearlyDateTimeParams;
  1605. static TYearlyDateTimeParams YearlyDateTimeParams;
  1606. static std::unique_ptr<TCriticalSection> DateTimeParamsSection(TraceInitPtr(new TCriticalSection()));
  1607. static void __fastcall EncodeDSTMargin(const SYSTEMTIME & Date, unsigned short Year,
  1608. TDateTime & Result);
  1609. //---------------------------------------------------------------------------
  1610. static unsigned short __fastcall DecodeYear(const TDateTime & DateTime)
  1611. {
  1612. unsigned short Year, Month, Day;
  1613. DecodeDate(DateTime, Year, Month, Day);
  1614. return Year;
  1615. }
  1616. //---------------------------------------------------------------------------
  1617. static const TDateTimeParams * __fastcall GetDateTimeParams(unsigned short Year)
  1618. {
  1619. TGuard Guard(DateTimeParamsSection.get());
  1620. TDateTimeParams * Result;
  1621. TYearlyDateTimeParams::iterator i = YearlyDateTimeParams.find(Year);
  1622. if (i != YearlyDateTimeParams.end())
  1623. {
  1624. Result = &(*i).second;
  1625. }
  1626. else
  1627. {
  1628. // creates new entry as a side effect
  1629. Result = &YearlyDateTimeParams[Year];
  1630. TIME_ZONE_INFORMATION TZI;
  1631. unsigned long GTZI;
  1632. HINSTANCE Kernel32 = GetModuleHandle(kernel32);
  1633. typedef BOOL WINAPI (* TGetTimeZoneInformationForYear)(USHORT wYear, PDYNAMIC_TIME_ZONE_INFORMATION pdtzi, LPTIME_ZONE_INFORMATION ptzi);
  1634. TGetTimeZoneInformationForYear GetTimeZoneInformationForYear =
  1635. (TGetTimeZoneInformationForYear)GetProcAddress(Kernel32, "GetTimeZoneInformationForYear");
  1636. if ((Year == 0) || (GetTimeZoneInformationForYear == NULL))
  1637. {
  1638. GTZI = GetTimeZoneInformation(&TZI);
  1639. }
  1640. else
  1641. {
  1642. GetTimeZoneInformationForYear(Year, NULL, &TZI);
  1643. GTZI = TIME_ZONE_ID_UNKNOWN;
  1644. }
  1645. switch (GTZI)
  1646. {
  1647. case TIME_ZONE_ID_UNKNOWN:
  1648. Result->CurrentDaylightDifferenceSec = 0;
  1649. break;
  1650. case TIME_ZONE_ID_STANDARD:
  1651. Result->CurrentDaylightDifferenceSec = TZI.StandardBias;
  1652. break;
  1653. case TIME_ZONE_ID_DAYLIGHT:
  1654. Result->CurrentDaylightDifferenceSec = TZI.DaylightBias;
  1655. break;
  1656. case TIME_ZONE_ID_INVALID:
  1657. default:
  1658. throw Exception(TIMEZONE_ERROR);
  1659. }
  1660. Result->BaseDifferenceSec = TZI.Bias;
  1661. Result->BaseDifference = double(TZI.Bias) / MinsPerDay;
  1662. Result->BaseDifferenceSec *= SecsPerMin;
  1663. Result->CurrentDifferenceSec = TZI.Bias +
  1664. Result->CurrentDaylightDifferenceSec;
  1665. Result->CurrentDifference =
  1666. double(Result->CurrentDifferenceSec) / MinsPerDay;
  1667. Result->CurrentDifferenceSec *= SecsPerMin;
  1668. Result->CurrentDaylightDifference =
  1669. double(Result->CurrentDaylightDifferenceSec) / MinsPerDay;
  1670. Result->CurrentDaylightDifferenceSec *= SecsPerMin;
  1671. Result->DaylightDifferenceSec = TZI.DaylightBias * SecsPerMin;
  1672. Result->DaylightDifference = double(TZI.DaylightBias) / MinsPerDay;
  1673. Result->StandardDifferenceSec = TZI.StandardBias * SecsPerMin;
  1674. Result->StandardDifference = double(TZI.StandardBias) / MinsPerDay;
  1675. Result->SystemStandardDate = TZI.StandardDate;
  1676. Result->SystemDaylightDate = TZI.DaylightDate;
  1677. unsigned short AYear = (Year != 0) ? Year : DecodeYear(Now());
  1678. if (Result->HasDST())
  1679. {
  1680. EncodeDSTMargin(Result->SystemStandardDate, AYear, Result->StandardDate);
  1681. EncodeDSTMargin(Result->SystemDaylightDate, AYear, Result->DaylightDate);
  1682. }
  1683. Result->StandardName = TZI.StandardName;
  1684. Result->DaylightName = TZI.DaylightName;
  1685. Result->DaylightHack = !IsWin7();
  1686. }
  1687. return Result;
  1688. }
  1689. //---------------------------------------------------------------------------
  1690. static void __fastcall EncodeDSTMargin(const SYSTEMTIME & Date, unsigned short Year,
  1691. TDateTime & Result)
  1692. {
  1693. if (Date.wYear == 0)
  1694. {
  1695. TDateTime Temp = EncodeDateVerbose(Year, Date.wMonth, 1);
  1696. Result = Temp + ((Date.wDayOfWeek - DayOfWeek(Temp) + 8) % 7) +
  1697. (7 * (Date.wDay - 1));
  1698. // Day 5 means, the last occurence of day-of-week in month
  1699. if (Date.wDay == 5)
  1700. {
  1701. unsigned short Month = static_cast<unsigned short>(Date.wMonth + 1);
  1702. if (Month > 12)
  1703. {
  1704. Month = static_cast<unsigned short>(Month - 12);
  1705. Year++;
  1706. }
  1707. if (Result >= EncodeDateVerbose(Year, Month, 1))
  1708. {
  1709. Result -= 7;
  1710. }
  1711. }
  1712. Result += EncodeTimeVerbose(Date.wHour, Date.wMinute, Date.wSecond,
  1713. Date.wMilliseconds);
  1714. }
  1715. else
  1716. {
  1717. Result = EncodeDateVerbose(Year, Date.wMonth, Date.wDay) +
  1718. EncodeTimeVerbose(Date.wHour, Date.wMinute, Date.wSecond, Date.wMilliseconds);
  1719. }
  1720. }
  1721. //---------------------------------------------------------------------------
  1722. static bool __fastcall IsDateInDST(const TDateTime & DateTime)
  1723. {
  1724. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  1725. bool Result;
  1726. // On some systems it occurs that StandardDate is unset, while
  1727. // DaylightDate is set. MSDN states that this is invalid and
  1728. // should be treated as if there is no daylight saving.
  1729. // So check both.
  1730. if (!Params->HasDST())
  1731. {
  1732. Result = false;
  1733. }
  1734. else
  1735. {
  1736. if (Params->SummerDST())
  1737. {
  1738. Result =
  1739. (DateTime >= Params->DaylightDate) &&
  1740. (DateTime < Params->StandardDate);
  1741. }
  1742. else
  1743. {
  1744. Result =
  1745. (DateTime < Params->StandardDate) ||
  1746. (DateTime >= Params->DaylightDate);
  1747. }
  1748. }
  1749. return Result;
  1750. }
  1751. //---------------------------------------------------------------------------
  1752. bool __fastcall UsesDaylightHack()
  1753. {
  1754. return GetDateTimeParams(0)->DaylightHack;
  1755. }
  1756. //---------------------------------------------------------------------------
  1757. TDateTime __fastcall UnixToDateTime(__int64 TimeStamp, TDSTMode DSTMode)
  1758. {
  1759. DebugAssert(int(EncodeDateVerbose(1970, 1, 1)) == UnixDateDelta);
  1760. TDateTime Result = UnixDateDelta + (double(TimeStamp) / SecsPerDay);
  1761. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(Result));
  1762. if (Params->DaylightHack)
  1763. {
  1764. if ((DSTMode == dstmWin) || (DSTMode == dstmUnix))
  1765. {
  1766. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  1767. Result -= CurrentParams->CurrentDifference;
  1768. }
  1769. else if (DSTMode == dstmKeep)
  1770. {
  1771. Result -= Params->BaseDifference;
  1772. }
  1773. }
  1774. else
  1775. {
  1776. Result -= Params->BaseDifference;
  1777. }
  1778. if ((DSTMode == dstmUnix) || (DSTMode == dstmKeep))
  1779. {
  1780. Result -= DSTDifferenceForTime(Result);
  1781. }
  1782. return Result;
  1783. }
  1784. //---------------------------------------------------------------------------
  1785. __int64 __fastcall Round(double Number)
  1786. {
  1787. double Floor = floor(Number);
  1788. double Ceil = ceil(Number);
  1789. return static_cast<__int64>(((Number - Floor) > (Ceil - Number)) ? Ceil : Floor);
  1790. }
  1791. //---------------------------------------------------------------------------
  1792. bool __fastcall TryRelativeStrToDateTime(UnicodeString S, TDateTime & DateTime, bool Add)
  1793. {
  1794. S = S.Trim();
  1795. int Index = 1;
  1796. if (SameText(S, L"today"))
  1797. {
  1798. S = L"0DS";
  1799. }
  1800. else if (SameText(S, L"yesterday"))
  1801. {
  1802. S = L"1DS";
  1803. }
  1804. while ((Index <= S.Length()) && IsDigit(S[Index]))
  1805. {
  1806. Index++;
  1807. }
  1808. UnicodeString NumberStr = S.SubString(1, Index - 1);
  1809. int Number;
  1810. bool Result = TryStrToInt(NumberStr, Number);
  1811. if (Result)
  1812. {
  1813. if (!Add)
  1814. {
  1815. Number = -Number;
  1816. }
  1817. S.Delete(1, Index - 1);
  1818. S = S.Trim().UpperCase();
  1819. bool Start = (S.Length() == 2) && (S[2] == L'S');
  1820. if (Start)
  1821. {
  1822. S.SetLength(S.Length() - 1);
  1823. }
  1824. DateTime = Now();
  1825. // These may not overlap with ParseSize (K, M and G)
  1826. if (S == L"S")
  1827. {
  1828. DateTime = IncSecond(DateTime, Number);
  1829. if (Start)
  1830. {
  1831. DateTime = IncMilliSecond(DateTime, -static_cast<int>(MilliSecondOfTheSecond(DateTime)));
  1832. }
  1833. }
  1834. else if (S == L"N")
  1835. {
  1836. DateTime = IncMinute(DateTime, Number);
  1837. if (Start)
  1838. {
  1839. DateTime = IncMilliSecond(DateTime, -static_cast<int>(MilliSecondOfTheMinute(DateTime)));
  1840. }
  1841. }
  1842. else if (S == L"H")
  1843. {
  1844. DateTime = IncHour(DateTime, Number);
  1845. if (Start)
  1846. {
  1847. DateTime = IncMilliSecond(DateTime, -static_cast<int>(MilliSecondOfTheHour(DateTime)));
  1848. }
  1849. }
  1850. else if (S == L"D")
  1851. {
  1852. DateTime = IncDay(DateTime, Number);
  1853. if (Start)
  1854. {
  1855. DateTime = IncMilliSecond(DateTime, -static_cast<int>(MilliSecondOfTheDay(DateTime)));
  1856. }
  1857. }
  1858. else if (S == L"Y")
  1859. {
  1860. DateTime = IncYear(DateTime, Number);
  1861. if (Start)
  1862. {
  1863. DateTime = IncMilliSecond(DateTime, -MilliSecondOfTheYear(DateTime));
  1864. }
  1865. }
  1866. else
  1867. {
  1868. Result = false;
  1869. }
  1870. }
  1871. return Result;
  1872. }
  1873. //---------------------------------------------------------------------------
  1874. bool TryStrToDateTimeStandard(const UnicodeString & S, TDateTime & Value)
  1875. {
  1876. TFormatSettings FormatSettings = TFormatSettings::Create(GetDefaultLCID());
  1877. FormatSettings.DateSeparator = L'-';
  1878. FormatSettings.TimeSeparator = L':';
  1879. FormatSettings.ShortDateFormat = "yyyy/mm/dd";
  1880. FormatSettings.ShortTimeFormat = "hh:nn:ss";
  1881. return TryStrToDateTime(S, Value, FormatSettings);
  1882. }
  1883. //---------------------------------------------------------------------------
  1884. const wchar_t KiloSize = L'K';
  1885. const wchar_t MegaSize = L'M';
  1886. const wchar_t GigaSize = L'G';
  1887. //---------------------------------------------------------------------------
  1888. // Keep consistent with parse_blocksize
  1889. bool __fastcall TryStrToSize(UnicodeString SizeStr, __int64 & Size)
  1890. {
  1891. int Index = 0;
  1892. while ((Index + 1 <= SizeStr.Length()) && IsDigit(SizeStr[Index + 1]))
  1893. {
  1894. Index++;
  1895. }
  1896. bool Result =
  1897. (Index > 0) && TryStrToInt64(SizeStr.SubString(1, Index), Size);
  1898. if (Result)
  1899. {
  1900. SizeStr = SizeStr.SubString(Index + 1, SizeStr.Length() - Index).Trim();
  1901. if (!SizeStr.IsEmpty())
  1902. {
  1903. Result = (SizeStr.Length() == 1);
  1904. if (Result)
  1905. {
  1906. wchar_t Unit = (wchar_t)toupper(SizeStr[1]);
  1907. switch (Unit)
  1908. {
  1909. case GigaSize:
  1910. Size *= 1024;
  1911. // fallthru
  1912. case MegaSize:
  1913. Size *= 1024;
  1914. // fallthru
  1915. case KiloSize:
  1916. Size *= 1024;
  1917. break;
  1918. default:
  1919. Result = false;
  1920. }
  1921. }
  1922. }
  1923. }
  1924. return Result;
  1925. }
  1926. //---------------------------------------------------------------------------
  1927. UnicodeString __fastcall SizeToStr(__int64 Size)
  1928. {
  1929. UnicodeString Result;
  1930. if ((Size <= 0) || ((Size % 1024) != 0))
  1931. {
  1932. Result = IntToStr(Size);
  1933. }
  1934. else
  1935. {
  1936. Size /= 1024;
  1937. if ((Size % 1024) != 0)
  1938. {
  1939. Result = IntToStr(Size) + KiloSize;
  1940. }
  1941. else
  1942. {
  1943. Size /= 1024;
  1944. if ((Size % 1024) != 0)
  1945. {
  1946. Result = IntToStr(Size) + MegaSize;
  1947. }
  1948. else
  1949. {
  1950. Size /= 1024;
  1951. Result = IntToStr(Size) + GigaSize;
  1952. }
  1953. }
  1954. }
  1955. return Result;
  1956. }
  1957. //---------------------------------------------------------------------------
  1958. static __int64 __fastcall DateTimeToUnix(const TDateTime DateTime)
  1959. {
  1960. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  1961. DebugAssert(int(EncodeDateVerbose(1970, 1, 1)) == UnixDateDelta);
  1962. return Round(double(DateTime - UnixDateDelta) * SecsPerDay) +
  1963. CurrentParams->CurrentDifferenceSec;
  1964. }
  1965. //---------------------------------------------------------------------------
  1966. FILETIME __fastcall DateTimeToFileTime(const TDateTime DateTime,
  1967. TDSTMode /*DSTMode*/)
  1968. {
  1969. __int64 UnixTimeStamp = ::DateTimeToUnix(DateTime);
  1970. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  1971. if (!Params->DaylightHack)
  1972. {
  1973. // We should probably use reversed code of FileTimeToDateTime here instead of custom implementation
  1974. // We are incrementing and decrementing BaseDifferenceSec because it
  1975. // can actually change between years
  1976. // (as it did in Belarus from GMT+2 to GMT+3 between 2011 and 2012)
  1977. UnixTimeStamp += (IsDateInDST(DateTime) ?
  1978. Params->DaylightDifferenceSec : Params->StandardDifferenceSec) +
  1979. Params->BaseDifferenceSec;
  1980. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  1981. UnixTimeStamp -=
  1982. CurrentParams->CurrentDaylightDifferenceSec +
  1983. CurrentParams->BaseDifferenceSec;
  1984. }
  1985. FILETIME Result;
  1986. (*(__int64*)&(Result) = (__int64(UnixTimeStamp) + 11644473600LL) * 10000000LL);
  1987. return Result;
  1988. }
  1989. //---------------------------------------------------------------------------
  1990. TDateTime __fastcall FileTimeToDateTime(const FILETIME & FileTime)
  1991. {
  1992. // duplicated in DirView.pas
  1993. TDateTime Result;
  1994. // The 0xFFF... is sometime seen for invalid timestamps,
  1995. // it would cause failure in SystemTimeToDateTime below
  1996. if (FileTime.dwLowDateTime == std::numeric_limits<DWORD>::max())
  1997. {
  1998. Result = MinDateTime;
  1999. }
  2000. else
  2001. {
  2002. SYSTEMTIME SysTime;
  2003. if (!UsesDaylightHack())
  2004. {
  2005. SYSTEMTIME UniverzalSysTime;
  2006. FileTimeToSystemTime(&FileTime, &UniverzalSysTime);
  2007. SystemTimeToTzSpecificLocalTime(NULL, &UniverzalSysTime, &SysTime);
  2008. }
  2009. else
  2010. {
  2011. FILETIME LocalFileTime;
  2012. FileTimeToLocalFileTime(&FileTime, &LocalFileTime);
  2013. FileTimeToSystemTime(&LocalFileTime, &SysTime);
  2014. }
  2015. Result = SystemTimeToDateTimeVerbose(SysTime);
  2016. }
  2017. return Result;
  2018. }
  2019. //---------------------------------------------------------------------------
  2020. __int64 __fastcall ConvertTimestampToUnix(const FILETIME & FileTime,
  2021. TDSTMode DSTMode)
  2022. {
  2023. __int64 Result = ((*(__int64*)&(FileTime)) / 10000000LL - 11644473600LL);
  2024. if (UsesDaylightHack())
  2025. {
  2026. if ((DSTMode == dstmUnix) || (DSTMode == dstmKeep))
  2027. {
  2028. FILETIME LocalFileTime;
  2029. SYSTEMTIME SystemTime;
  2030. FileTimeToLocalFileTime(&FileTime, &LocalFileTime);
  2031. FileTimeToSystemTime(&LocalFileTime, &SystemTime);
  2032. TDateTime DateTime = SystemTimeToDateTimeVerbose(SystemTime);
  2033. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  2034. Result += (IsDateInDST(DateTime) ?
  2035. Params->DaylightDifferenceSec : Params->StandardDifferenceSec);
  2036. if (DSTMode == dstmKeep)
  2037. {
  2038. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  2039. Result -= CurrentParams->CurrentDaylightDifferenceSec;
  2040. }
  2041. }
  2042. }
  2043. else
  2044. {
  2045. if (DSTMode == dstmWin)
  2046. {
  2047. FILETIME LocalFileTime;
  2048. SYSTEMTIME SystemTime;
  2049. FileTimeToLocalFileTime(&FileTime, &LocalFileTime);
  2050. FileTimeToSystemTime(&LocalFileTime, &SystemTime);
  2051. TDateTime DateTime = SystemTimeToDateTimeVerbose(SystemTime);
  2052. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  2053. Result -= (IsDateInDST(DateTime) ?
  2054. Params->DaylightDifferenceSec : Params->StandardDifferenceSec);
  2055. }
  2056. }
  2057. return Result;
  2058. }
  2059. //---------------------------------------------------------------------------
  2060. TDateTime __fastcall ConvertTimestampToUTC(TDateTime DateTime)
  2061. {
  2062. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  2063. DateTime += DSTDifferenceForTime(DateTime);
  2064. DateTime += Params->BaseDifference;
  2065. if (Params->DaylightHack)
  2066. {
  2067. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  2068. DateTime += CurrentParams->CurrentDaylightDifference;
  2069. }
  2070. return DateTime;
  2071. }
  2072. //---------------------------------------------------------------------------
  2073. TDateTime __fastcall ConvertTimestampFromUTC(TDateTime DateTime)
  2074. {
  2075. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  2076. DateTime -= DSTDifferenceForTime(DateTime);
  2077. DateTime -= Params->BaseDifference;
  2078. if (Params->DaylightHack)
  2079. {
  2080. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  2081. DateTime -= CurrentParams->CurrentDaylightDifference;
  2082. }
  2083. return DateTime;
  2084. }
  2085. //---------------------------------------------------------------------------
  2086. __int64 __fastcall ConvertTimestampToUnixSafe(const FILETIME & FileTime,
  2087. TDSTMode DSTMode)
  2088. {
  2089. __int64 Result;
  2090. if ((FileTime.dwLowDateTime == 0) &&
  2091. (FileTime.dwHighDateTime == 0))
  2092. {
  2093. Result = ::DateTimeToUnix(Now());
  2094. }
  2095. else
  2096. {
  2097. Result = ConvertTimestampToUnix(FileTime, DSTMode);
  2098. }
  2099. return Result;
  2100. }
  2101. //---------------------------------------------------------------------------
  2102. double __fastcall DSTDifferenceForTime(TDateTime DateTime)
  2103. {
  2104. double Result;
  2105. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  2106. if (IsDateInDST(DateTime))
  2107. {
  2108. Result = Params->DaylightDifference;
  2109. }
  2110. else
  2111. {
  2112. Result = Params->StandardDifference;
  2113. }
  2114. return Result;
  2115. }
  2116. //---------------------------------------------------------------------------
  2117. TDateTime __fastcall AdjustDateTimeFromUnix(TDateTime DateTime, TDSTMode DSTMode)
  2118. {
  2119. const TDateTimeParams * Params = GetDateTimeParams(DecodeYear(DateTime));
  2120. if (Params->DaylightHack)
  2121. {
  2122. if ((DSTMode == dstmWin) || (DSTMode == dstmUnix))
  2123. {
  2124. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  2125. DateTime = DateTime - CurrentParams->CurrentDaylightDifference;
  2126. }
  2127. if (!IsDateInDST(DateTime))
  2128. {
  2129. if (DSTMode == dstmWin)
  2130. {
  2131. DateTime = DateTime - Params->DaylightDifference;
  2132. }
  2133. }
  2134. else
  2135. {
  2136. DateTime = DateTime - Params->StandardDifference;
  2137. }
  2138. }
  2139. else
  2140. {
  2141. if (DSTMode == dstmWin)
  2142. {
  2143. DateTime = DateTime + DSTDifferenceForTime(DateTime);
  2144. }
  2145. }
  2146. return DateTime;
  2147. }
  2148. //---------------------------------------------------------------------------
  2149. UnicodeString __fastcall FixedLenDateTimeFormat(const UnicodeString & Format)
  2150. {
  2151. UnicodeString Result = Format;
  2152. bool AsIs = false;
  2153. int Index = 1;
  2154. while (Index <= Result.Length())
  2155. {
  2156. wchar_t F = Result[Index];
  2157. if ((F == L'\'') || (F == L'\"'))
  2158. {
  2159. AsIs = !AsIs;
  2160. Index++;
  2161. }
  2162. else if (!AsIs && ((F == L'a') || (F == L'A')))
  2163. {
  2164. if (Result.SubString(Index, 5).LowerCase() == L"am/pm")
  2165. {
  2166. Index += 5;
  2167. }
  2168. else if (Result.SubString(Index, 3).LowerCase() == L"a/p")
  2169. {
  2170. Index += 3;
  2171. }
  2172. else if (Result.SubString(Index, 4).LowerCase() == L"ampm")
  2173. {
  2174. Index += 4;
  2175. }
  2176. else
  2177. {
  2178. Index++;
  2179. }
  2180. }
  2181. else
  2182. {
  2183. if (!AsIs && (wcschr(L"dDeEmMhHnNsS", F) != NULL) &&
  2184. ((Index == Result.Length()) || (Result[Index + 1] != F)))
  2185. {
  2186. Result.Insert(F, Index);
  2187. }
  2188. while ((Index <= Result.Length()) && (F == Result[Index]))
  2189. {
  2190. Index++;
  2191. }
  2192. }
  2193. }
  2194. return Result;
  2195. }
  2196. //---------------------------------------------------------------------------
  2197. UnicodeString __fastcall FormatTimeZone(long Sec)
  2198. {
  2199. TTimeSpan Span = TTimeSpan::FromSeconds(Sec);
  2200. UnicodeString Str;
  2201. if ((Span.Seconds == 0) && (Span.Minutes == 0))
  2202. {
  2203. Str = FORMAT(L"%d", (-Span.Hours));
  2204. }
  2205. else if (Span.Seconds == 0)
  2206. {
  2207. Str = FORMAT(L"%d:%2.2d", (-Span.Hours, abs(Span.Minutes)));
  2208. }
  2209. else
  2210. {
  2211. Str = FORMAT(L"%d:%2.2d:%2.2d", (-Span.Hours, abs(Span.Minutes), abs(Span.Seconds)));
  2212. }
  2213. Str = ((Span <= TTimeSpan::Zero) ? L"+" : L"") + Str;
  2214. return Str;
  2215. }
  2216. //---------------------------------------------------------------------------
  2217. UnicodeString __fastcall GetTimeZoneLogString()
  2218. {
  2219. const TDateTimeParams * CurrentParams = GetDateTimeParams(0);
  2220. UnicodeString Result =
  2221. FORMAT(L"Current: GMT%s", (FormatTimeZone(CurrentParams->CurrentDifferenceSec)));
  2222. if (!CurrentParams->HasDST())
  2223. {
  2224. Result += FORMAT(L" (%s), No DST", (CurrentParams->StandardName));
  2225. }
  2226. else
  2227. {
  2228. Result +=
  2229. FORMAT(L", Standard: GMT%s (%s), DST: GMT%s (%s), DST Start: %s, DST End: %s",
  2230. (FormatTimeZone(CurrentParams->BaseDifferenceSec + CurrentParams->StandardDifferenceSec),
  2231. CurrentParams->StandardName,
  2232. FormatTimeZone(CurrentParams->BaseDifferenceSec + CurrentParams->DaylightDifferenceSec),
  2233. CurrentParams->DaylightName,
  2234. CurrentParams->DaylightDate.DateString(),
  2235. CurrentParams->StandardDate.DateString()));
  2236. }
  2237. return Result;
  2238. }
  2239. //---------------------------------------------------------------------------
  2240. bool __fastcall AdjustClockForDSTEnabled()
  2241. {
  2242. // Windows XP deletes the DisableAutoDaylightTimeSet value when it is off
  2243. // (the later versions set it to DynamicDaylightTimeDisabled to 0)
  2244. bool DynamicDaylightTimeDisabled = false;
  2245. TRegistry * Registry = new TRegistry(KEY_READ);
  2246. try
  2247. {
  2248. Registry->RootKey = HKEY_LOCAL_MACHINE;
  2249. if (Registry->OpenKey(L"SYSTEM", false) &&
  2250. Registry->OpenKey(L"CurrentControlSet", false) &&
  2251. Registry->OpenKey(L"Control", false) &&
  2252. Registry->OpenKey(L"TimeZoneInformation", false))
  2253. {
  2254. if (Registry->ValueExists(L"DynamicDaylightTimeDisabled"))
  2255. {
  2256. DynamicDaylightTimeDisabled = Registry->ReadBool(L"DynamicDaylightTimeDisabled");
  2257. }
  2258. // WORKAROUND
  2259. // Windows XP equivalent
  2260. else if (Registry->ValueExists(L"DisableAutoDaylightTimeSet"))
  2261. {
  2262. DynamicDaylightTimeDisabled = Registry->ReadBool(L"DisableAutoDaylightTimeSet");
  2263. }
  2264. }
  2265. delete Registry;
  2266. }
  2267. catch(...)
  2268. {
  2269. }
  2270. return !DynamicDaylightTimeDisabled;
  2271. }
  2272. //---------------------------------------------------------------------------
  2273. UnicodeString __fastcall StandardDatestamp()
  2274. {
  2275. return FormatDateTime(L"yyyy'-'mm'-'dd", ConvertTimestampToUTC(Now()));
  2276. }
  2277. //---------------------------------------------------------------------------
  2278. UnicodeString __fastcall StandardTimestamp(const TDateTime & DateTime)
  2279. {
  2280. return FormatDateTime(L"yyyy'-'mm'-'dd'T'hh':'nn':'ss'.'zzz'Z'", ConvertTimestampToUTC(DateTime));
  2281. }
  2282. //---------------------------------------------------------------------------
  2283. UnicodeString __fastcall StandardTimestamp()
  2284. {
  2285. return StandardTimestamp(Now());
  2286. }
  2287. //---------------------------------------------------------------------------
  2288. static TDateTime TwoSeconds(0, 0, 2, 0);
  2289. int __fastcall CompareFileTime(TDateTime T1, TDateTime T2)
  2290. {
  2291. // "FAT" time precision
  2292. // (when one time is seconds-precision and other is millisecond-precision,
  2293. // we may have times like 12:00:00.000 and 12:00:01.999, which should
  2294. // be treated the same)
  2295. int Result;
  2296. if (T1 == T2)
  2297. {
  2298. // just optimization
  2299. Result = 0;
  2300. }
  2301. else if ((T1 < T2) && (T2 - T1 >= TwoSeconds))
  2302. {
  2303. Result = -1;
  2304. }
  2305. else if ((T1 > T2) && (T1 - T2 >= TwoSeconds))
  2306. {
  2307. Result = 1;
  2308. }
  2309. else
  2310. {
  2311. Result = 0;
  2312. }
  2313. return Result;
  2314. }
  2315. //---------------------------------------------------------------------------
  2316. int __fastcall TimeToMSec(TDateTime T)
  2317. {
  2318. return int(Round(double(T) * double(MSecsPerDay)));
  2319. }
  2320. //---------------------------------------------------------------------------
  2321. int __fastcall TimeToSeconds(TDateTime T)
  2322. {
  2323. return TimeToMSec(T) / MSecsPerSec;
  2324. }
  2325. //---------------------------------------------------------------------------
  2326. int __fastcall TimeToMinutes(TDateTime T)
  2327. {
  2328. return TimeToSeconds(T) / SecsPerMin;
  2329. }
  2330. //---------------------------------------------------------------------------
  2331. static bool __fastcall DoRecursiveDeleteFile(
  2332. const UnicodeString FileName, bool ToRecycleBin, UnicodeString & ErrorPath, int & Deleted)
  2333. {
  2334. bool Result;
  2335. Deleted = 0;
  2336. UnicodeString AErrorPath = FileName;
  2337. if (!ToRecycleBin)
  2338. {
  2339. TSearchRecChecked SearchRec;
  2340. Result = FileSearchRec(FileName, SearchRec);
  2341. if (Result)
  2342. {
  2343. if (!SearchRec.IsDirectory())
  2344. {
  2345. Result = DeleteFile(ApiPath(FileName));
  2346. if (Result)
  2347. {
  2348. Deleted++;
  2349. }
  2350. }
  2351. else
  2352. {
  2353. Result = (FindFirstUnchecked(FileName + L"\\*", faAnyFile, SearchRec) == 0);
  2354. if (Result)
  2355. {
  2356. try
  2357. {
  2358. do
  2359. {
  2360. UnicodeString FileName2 = FileName + L"\\" + SearchRec.Name;
  2361. if (SearchRec.IsDirectory())
  2362. {
  2363. if (SearchRec.IsRealFile())
  2364. {
  2365. Result = DoRecursiveDeleteFile(FileName2, DebugAlwaysFalse(ToRecycleBin), AErrorPath, Deleted);
  2366. }
  2367. }
  2368. else
  2369. {
  2370. Result = DeleteFile(ApiPath(FileName2));
  2371. if (!Result)
  2372. {
  2373. AErrorPath = FileName2;
  2374. }
  2375. else
  2376. {
  2377. Deleted++;
  2378. }
  2379. }
  2380. }
  2381. while (Result && (FindNextUnchecked(SearchRec) == 0));
  2382. }
  2383. __finally
  2384. {
  2385. FindClose(SearchRec);
  2386. }
  2387. if (Result)
  2388. {
  2389. Result = RemoveDir(ApiPath(FileName));
  2390. if (Result)
  2391. {
  2392. Deleted++;
  2393. }
  2394. }
  2395. }
  2396. }
  2397. }
  2398. }
  2399. else
  2400. {
  2401. SHFILEOPSTRUCT Data;
  2402. memset(&Data, 0, sizeof(Data));
  2403. Data.hwnd = NULL;
  2404. Data.wFunc = FO_DELETE;
  2405. // SHFileOperation does not support long paths anyway
  2406. UnicodeString FileList(ApiPath(FileName));
  2407. FileList.SetLength(FileList.Length() + 2);
  2408. FileList[FileList.Length() - 1] = L'\0';
  2409. FileList[FileList.Length()] = L'\0';
  2410. Data.pFrom = FileList.c_str();
  2411. Data.pTo = L"\0\0"; // this will actually give one null more than needed
  2412. Data.fFlags = FOF_NOCONFIRMATION | FOF_RENAMEONCOLLISION | FOF_NOCONFIRMMKDIR |
  2413. FOF_NOERRORUI | FOF_SILENT;
  2414. if (DebugAlwaysTrue(ToRecycleBin))
  2415. {
  2416. Data.fFlags |= FOF_ALLOWUNDO;
  2417. }
  2418. int ErrorCode = SHFileOperation(&Data);
  2419. Result = (ErrorCode == 0);
  2420. if (!Result)
  2421. {
  2422. // according to MSDN, SHFileOperation may return following non-Win32
  2423. // error codes
  2424. if (((ErrorCode >= 0x71) && (ErrorCode <= 0x88)) ||
  2425. (ErrorCode == 0xB7) || (ErrorCode == 0x402) || (ErrorCode == 0x10000) ||
  2426. (ErrorCode == 0x10074))
  2427. {
  2428. ErrorCode = 0;
  2429. }
  2430. SetLastError(ErrorCode);
  2431. }
  2432. if (Result)
  2433. {
  2434. Deleted = 1;
  2435. }
  2436. }
  2437. if (!Result)
  2438. {
  2439. ErrorPath = AErrorPath;
  2440. }
  2441. return Result;
  2442. }
  2443. //---------------------------------------------------------------------------
  2444. bool __fastcall RecursiveDeleteFile(const UnicodeString & FileName, bool ToRecycleBin)
  2445. {
  2446. UnicodeString ErrorPath; // unused
  2447. int Deleted;
  2448. bool Result = DoRecursiveDeleteFile(FileName, ToRecycleBin, ErrorPath, Deleted);
  2449. return Result;
  2450. }
  2451. //---------------------------------------------------------------------------
  2452. int __fastcall RecursiveDeleteFileChecked(const UnicodeString & FileName, bool ToRecycleBin)
  2453. {
  2454. UnicodeString ErrorPath;
  2455. int Deleted;
  2456. if (!DoRecursiveDeleteFile(FileName, ToRecycleBin, ErrorPath, Deleted))
  2457. {
  2458. throw EOSExtException(FMTLOAD(DELETE_LOCAL_FILE_ERROR, (ErrorPath)));
  2459. }
  2460. return Deleted;
  2461. }
  2462. //---------------------------------------------------------------------------
  2463. void __fastcall DeleteFileChecked(const UnicodeString & FileName)
  2464. {
  2465. if (!DeleteFile(ApiPath(FileName)))
  2466. {
  2467. throw EOSExtException(FMTLOAD(DELETE_LOCAL_FILE_ERROR, (FileName)));
  2468. }
  2469. }
  2470. //---------------------------------------------------------------------------
  2471. unsigned int __fastcall CancelAnswer(unsigned int Answers)
  2472. {
  2473. unsigned int Result;
  2474. if ((Answers & qaCancel) != 0)
  2475. {
  2476. Result = qaCancel;
  2477. }
  2478. else if ((Answers & qaNo) != 0)
  2479. {
  2480. Result = qaNo;
  2481. }
  2482. else if ((Answers & qaAbort) != 0)
  2483. {
  2484. Result = qaAbort;
  2485. }
  2486. else if ((Answers & qaOK) != 0)
  2487. {
  2488. Result = qaOK;
  2489. }
  2490. else
  2491. {
  2492. DebugFail();
  2493. Result = qaCancel;
  2494. }
  2495. return Result;
  2496. }
  2497. //---------------------------------------------------------------------------
  2498. unsigned int __fastcall AbortAnswer(unsigned int Answers)
  2499. {
  2500. unsigned int Result;
  2501. if (FLAGSET(Answers, qaAbort))
  2502. {
  2503. Result = qaAbort;
  2504. }
  2505. else
  2506. {
  2507. Result = CancelAnswer(Answers);
  2508. }
  2509. return Result;
  2510. }
  2511. //---------------------------------------------------------------------------
  2512. unsigned int __fastcall ContinueAnswer(unsigned int Answers)
  2513. {
  2514. unsigned int Result;
  2515. if (FLAGSET(Answers, qaSkip))
  2516. {
  2517. Result = qaSkip;
  2518. }
  2519. else if (FLAGSET(Answers, qaIgnore))
  2520. {
  2521. Result = qaIgnore;
  2522. }
  2523. else if (FLAGSET(Answers, qaYes))
  2524. {
  2525. Result = qaYes;
  2526. }
  2527. else if (FLAGSET(Answers, qaOK))
  2528. {
  2529. Result = qaOK;
  2530. }
  2531. else if (FLAGSET(Answers, qaRetry))
  2532. {
  2533. Result = qaRetry;
  2534. }
  2535. else
  2536. {
  2537. Result = CancelAnswer(Answers);
  2538. }
  2539. return Result;
  2540. }
  2541. //---------------------------------------------------------------------------
  2542. TLibModule * __fastcall FindModule(void * Instance)
  2543. {
  2544. TLibModule * CurModule;
  2545. CurModule = reinterpret_cast<TLibModule*>(LibModuleList);
  2546. while (CurModule)
  2547. {
  2548. if (CurModule->Instance == (unsigned)Instance)
  2549. {
  2550. break;
  2551. }
  2552. else
  2553. {
  2554. CurModule = CurModule->Next;
  2555. }
  2556. }
  2557. return CurModule;
  2558. }
  2559. //---------------------------------------------------------------------------
  2560. static UnicodeString __fastcall DoLoadStrFrom(HINSTANCE Module, int Ident, unsigned int MaxLength)
  2561. {
  2562. UnicodeString Result;
  2563. Result.SetLength(MaxLength);
  2564. int Length = LoadString(Module, Ident, Result.c_str(), MaxLength);
  2565. Result.SetLength(Length);
  2566. return Result;
  2567. }
  2568. //---------------------------------------------------------------------------
  2569. UnicodeString __fastcall LoadStrFrom(HINSTANCE Module, int Ident)
  2570. {
  2571. // 1024 = what VCL LoadStr limits the string to
  2572. return DoLoadStrFrom(Module, Ident, 1024);
  2573. }
  2574. //---------------------------------------------------------------------------
  2575. UnicodeString __fastcall LoadStr(int Ident, unsigned int MaxLength)
  2576. {
  2577. TLibModule * MainModule = FindModule(HInstance);
  2578. DebugAssert(MainModule != NULL);
  2579. return DoLoadStrFrom((HINSTANCE)MainModule->ResInstance, Ident, MaxLength);
  2580. }
  2581. //---------------------------------------------------------------------------
  2582. UnicodeString __fastcall LoadStrPart(int Ident, int Part)
  2583. {
  2584. UnicodeString Result;
  2585. UnicodeString Str = LoadStr(Ident);
  2586. while (Part > 0)
  2587. {
  2588. Result = CutToChar(Str, L'|', false);
  2589. Part--;
  2590. }
  2591. return Result;
  2592. }
  2593. //---------------------------------------------------------------------------
  2594. UnicodeString __fastcall DecodeUrlChars(UnicodeString S)
  2595. {
  2596. int i = 1;
  2597. while (i <= S.Length())
  2598. {
  2599. switch (S[i])
  2600. {
  2601. case L'+':
  2602. S[i] = ' ';
  2603. break;
  2604. case L'%':
  2605. {
  2606. UnicodeString Hex;
  2607. while ((i + 2 <= S.Length()) && (S[i] == L'%') &&
  2608. IsHex(S[i + 1]) && IsHex(S[i + 2]))
  2609. {
  2610. Hex += S.SubString(i + 1, 2);
  2611. S.Delete(i, 3);
  2612. }
  2613. if (!Hex.IsEmpty())
  2614. {
  2615. RawByteString Bytes = HexToBytes(Hex);
  2616. UnicodeString Chars(UTF8ToString(Bytes));
  2617. S.Insert(Chars, i);
  2618. i += Chars.Length() - 1;
  2619. }
  2620. }
  2621. break;
  2622. }
  2623. i++;
  2624. }
  2625. return S;
  2626. }
  2627. //---------------------------------------------------------------------------
  2628. UnicodeString __fastcall DoEncodeUrl(UnicodeString S, const UnicodeString & DoNotEncode)
  2629. {
  2630. int Index = 1;
  2631. while (Index <= S.Length())
  2632. {
  2633. wchar_t C = S[Index];
  2634. if (IsLetter(C) ||
  2635. IsDigit(C) ||
  2636. (C == L'_') || (C == L'-') || (C == L'.') || (C == L'*') ||
  2637. (DoNotEncode.Pos(C) > 0))
  2638. {
  2639. Index++;
  2640. }
  2641. else
  2642. {
  2643. UTF8String UtfS(S.SubString(Index, 1));
  2644. UnicodeString H;
  2645. for (int Index2 = 1; Index2 <= UtfS.Length(); Index2++)
  2646. {
  2647. H += L"%" + ByteToHex(static_cast<unsigned char>(UtfS[Index2]));
  2648. }
  2649. S.Delete(Index, 1);
  2650. S.Insert(H, Index);
  2651. Index += H.Length();
  2652. }
  2653. }
  2654. return S;
  2655. }
  2656. //---------------------------------------------------------------------------
  2657. UnicodeString __fastcall EncodeUrlString(UnicodeString S)
  2658. {
  2659. return DoEncodeUrl(S, UnicodeString());
  2660. }
  2661. //---------------------------------------------------------------------------
  2662. UnicodeString __fastcall EncodeUrlPath(UnicodeString S)
  2663. {
  2664. return DoEncodeUrl(S, L"/");
  2665. }
  2666. //---------------------------------------------------------------------------
  2667. UnicodeString __fastcall AppendUrlParams(UnicodeString AURL, UnicodeString Params)
  2668. {
  2669. // see also TWebHelpSystem::ShowHelp
  2670. const wchar_t FragmentSeparator = L'#';
  2671. UnicodeString URL = CutToChar(AURL, FragmentSeparator, false);
  2672. if (URL.Pos(L"?") == 0)
  2673. {
  2674. URL += L"?";
  2675. }
  2676. else
  2677. {
  2678. URL += L"&";
  2679. }
  2680. URL += Params;
  2681. AddToList(URL, AURL, FragmentSeparator);
  2682. return URL;
  2683. }
  2684. //---------------------------------------------------------------------------
  2685. UnicodeString __fastcall ExtractFileNameFromUrl(const UnicodeString & Url)
  2686. {
  2687. UnicodeString Result = Url;
  2688. int P = Result.Pos(L"?");
  2689. if (P > 0)
  2690. {
  2691. Result.SetLength(P - 1);
  2692. }
  2693. P = Result.LastDelimiter("/");
  2694. if (DebugAlwaysTrue(P > 0))
  2695. {
  2696. Result.Delete(1, P);
  2697. }
  2698. return Result;
  2699. }
  2700. //---------------------------------------------------------------------------
  2701. UnicodeString __fastcall EscapeHotkey(const UnicodeString & Caption)
  2702. {
  2703. return ReplaceStr(Caption, L"&", L"&&");
  2704. }
  2705. //---------------------------------------------------------------------------
  2706. // duplicated in console's Main.cpp
  2707. static bool __fastcall DoCutToken(UnicodeString & Str, UnicodeString & Token,
  2708. UnicodeString * RawToken, UnicodeString * Separator, bool EscapeQuotesInQuotesOnly)
  2709. {
  2710. bool Result;
  2711. Token = L"";
  2712. // inspired by Putty's sftp_getcmd() from PSFTP.C
  2713. int Index = 1;
  2714. while ((Index <= Str.Length()) &&
  2715. ((Str[Index] == L' ') || (Str[Index] == L'\t')))
  2716. {
  2717. Index++;
  2718. }
  2719. if (Index <= Str.Length())
  2720. {
  2721. bool Quoting = false;
  2722. while (Index <= Str.Length())
  2723. {
  2724. if (!Quoting && ((Str[Index] == L' ') || (Str[Index] == L'\t')))
  2725. {
  2726. break;
  2727. }
  2728. // With EscapeQuotesInQuotesOnly we escape quotes only within quotes
  2729. // otherwise the "" means " (quote), but it should mean empty string.
  2730. else if ((Str[Index] == L'"') && (Index + 1 <= Str.Length()) &&
  2731. (Str[Index + 1] == L'"') && (!EscapeQuotesInQuotesOnly || Quoting))
  2732. {
  2733. Index += 2;
  2734. Token += L'"';
  2735. }
  2736. else if (Str[Index] == L'"')
  2737. {
  2738. Index++;
  2739. Quoting = !Quoting;
  2740. }
  2741. else
  2742. {
  2743. Token += Str[Index];
  2744. Index++;
  2745. }
  2746. }
  2747. if (RawToken != NULL)
  2748. {
  2749. (*RawToken) = Str.SubString(1, Index - 1);
  2750. }
  2751. if (Index <= Str.Length())
  2752. {
  2753. if (Separator != NULL)
  2754. {
  2755. *Separator = Str.SubString(Index, 1);
  2756. }
  2757. Index++;
  2758. }
  2759. else
  2760. {
  2761. if (Separator != NULL)
  2762. {
  2763. *Separator = UnicodeString();
  2764. }
  2765. }
  2766. Str = Str.SubString(Index, Str.Length());
  2767. Result = true;
  2768. }
  2769. else
  2770. {
  2771. Result = false;
  2772. Str = L"";
  2773. }
  2774. return Result;
  2775. }
  2776. //---------------------------------------------------------------------------
  2777. bool __fastcall CutToken(UnicodeString & Str, UnicodeString & Token,
  2778. UnicodeString * RawToken, UnicodeString * Separator)
  2779. {
  2780. return DoCutToken(Str, Token, RawToken, Separator, false);
  2781. }
  2782. //---------------------------------------------------------------------------
  2783. bool __fastcall CutTokenEx(UnicodeString & Str, UnicodeString & Token,
  2784. UnicodeString * RawToken, UnicodeString * Separator)
  2785. {
  2786. return DoCutToken(Str, Token, RawToken, Separator, true);
  2787. }
  2788. //---------------------------------------------------------------------------
  2789. void __fastcall AddToList(UnicodeString & List, const UnicodeString & Value, const UnicodeString & Delimiter)
  2790. {
  2791. if (!Value.IsEmpty())
  2792. {
  2793. if (!List.IsEmpty() &&
  2794. ((List.Length() < Delimiter.Length()) ||
  2795. (List.SubString(List.Length() - Delimiter.Length() + 1, Delimiter.Length()) != Delimiter)))
  2796. {
  2797. List += Delimiter;
  2798. }
  2799. List += Value;
  2800. }
  2801. }
  2802. //---------------------------------------------------------------------------
  2803. bool __fastcall IsWinVista()
  2804. {
  2805. // Vista is 6.0
  2806. // Win XP is 5.1
  2807. // There also 5.2, what is Windows 2003 or Windows XP 64bit
  2808. // (we consider it WinXP for now)
  2809. return CheckWin32Version(6, 0);
  2810. }
  2811. //---------------------------------------------------------------------------
  2812. bool __fastcall IsWin7()
  2813. {
  2814. return CheckWin32Version(6, 1);
  2815. }
  2816. //---------------------------------------------------------------------------
  2817. bool __fastcall IsWin8()
  2818. {
  2819. return CheckWin32Version(6, 2);
  2820. }
  2821. //---------------------------------------------------------------------------
  2822. bool __fastcall IsWin10()
  2823. {
  2824. return CheckWin32Version(10, 0);
  2825. }
  2826. //---------------------------------------------------------------------------
  2827. static OSVERSIONINFO __fastcall GetWindowsVersion()
  2828. {
  2829. OSVERSIONINFO Result;
  2830. memset(&Result, 0, sizeof(Result));
  2831. Result.dwOSVersionInfoSize = sizeof(Result);
  2832. // Cannot use the VCL Win32MajorVersion+Win32MinorVersion+Win32BuildNumber as
  2833. // on Windows 10 due to some hacking in InitPlatformId, the Win32BuildNumber is lost
  2834. GetVersionEx(&Result);
  2835. return Result;
  2836. }
  2837. //---------------------------------------------------------------------------
  2838. bool __fastcall IsWin10Build(unsigned int BuildNumber)
  2839. {
  2840. // It might be enough to check the dwBuildNumber, as we do in TWinConfiguration::IsDDExtBroken()
  2841. OSVERSIONINFO OSVersionInfo = GetWindowsVersion();
  2842. return
  2843. (OSVersionInfo.dwMajorVersion > 10) ||
  2844. ((OSVersionInfo.dwMajorVersion == 10) && (OSVersionInfo.dwMinorVersion > 0)) ||
  2845. ((OSVersionInfo.dwMajorVersion == 10) && (OSVersionInfo.dwMinorVersion == 0) &&
  2846. (OSVersionInfo.dwBuildNumber >= BuildNumber));
  2847. }
  2848. //---------------------------------------------------------------------------
  2849. bool __fastcall IsWine()
  2850. {
  2851. HMODULE NtDll = GetModuleHandle(L"ntdll.dll");
  2852. return
  2853. DebugAlwaysTrue(NtDll != NULL) &&
  2854. (GetProcAddress(NtDll, "wine_get_version") != NULL);
  2855. }
  2856. //---------------------------------------------------------------------------
  2857. int GIsUWP = -1;
  2858. //---------------------------------------------------------------------------
  2859. bool __fastcall IsUWP()
  2860. {
  2861. if (GIsUWP < 0)
  2862. {
  2863. HINSTANCE Kernel32 = GetModuleHandle(kernel32);
  2864. typedef LONG WINAPI (* GetPackageFamilyNameProc)(HANDLE hProcess, UINT32 *packageFamilyNameLength, PWSTR packageFamilyName);
  2865. GetPackageFamilyNameProc GetPackageFamilyName =
  2866. (GetPackageFamilyNameProc)GetProcAddress(Kernel32, "GetPackageFamilyName");
  2867. UINT32 Len = 0;
  2868. bool Result =
  2869. (GetPackageFamilyName != NULL) &&
  2870. (GetPackageFamilyName(GetCurrentProcess(), &Len, NULL) == ERROR_INSUFFICIENT_BUFFER);
  2871. GIsUWP = (Result ? 1 : 0);
  2872. }
  2873. return (GIsUWP > 0);
  2874. }
  2875. //---------------------------------------------------------------------------
  2876. LCID __fastcall GetDefaultLCID()
  2877. {
  2878. return GetUserDefaultLCID();
  2879. }
  2880. //---------------------------------------------------------------------------
  2881. static UnicodeString ADefaultEncodingName;
  2882. UnicodeString __fastcall DefaultEncodingName()
  2883. {
  2884. if (ADefaultEncodingName.IsEmpty())
  2885. {
  2886. CPINFOEX Info;
  2887. GetCPInfoEx(CP_ACP, 0, &Info);
  2888. ADefaultEncodingName = Info.CodePageName;
  2889. }
  2890. return ADefaultEncodingName;
  2891. }
  2892. //---------------------------------------------------------------------------
  2893. bool _fastcall GetWindowsProductType(DWORD & Type)
  2894. {
  2895. bool Result;
  2896. HINSTANCE Kernel32 = GetModuleHandle(kernel32);
  2897. typedef BOOL WINAPI (* TGetProductInfo)(DWORD, DWORD, DWORD, DWORD, PDWORD);
  2898. TGetProductInfo GetProductInfo =
  2899. (TGetProductInfo)GetProcAddress(Kernel32, "GetProductInfo");
  2900. if (GetProductInfo == NULL)
  2901. {
  2902. Result = false;
  2903. }
  2904. else
  2905. {
  2906. GetProductInfo(Win32MajorVersion, Win32MinorVersion, 0, 0, &Type);
  2907. Result = true;
  2908. }
  2909. return Result;
  2910. }
  2911. //---------------------------------------------------------------------------
  2912. UnicodeString __fastcall WindowsProductName()
  2913. {
  2914. UnicodeString Result;
  2915. TRegistry * Registry = new TRegistry(KEY_READ);
  2916. try
  2917. {
  2918. Registry->RootKey = HKEY_LOCAL_MACHINE;
  2919. if (Registry->OpenKey(L"SOFTWARE", false) &&
  2920. Registry->OpenKey(L"Microsoft", false) &&
  2921. Registry->OpenKey(L"Windows NT", false) &&
  2922. Registry->OpenKey(L"CurrentVersion", false))
  2923. {
  2924. Result = Registry->ReadString(L"ProductName");
  2925. }
  2926. delete Registry;
  2927. }
  2928. catch(...)
  2929. {
  2930. }
  2931. return Result;
  2932. }
  2933. //---------------------------------------------------------------------------
  2934. int __fastcall GetWindowsBuild()
  2935. {
  2936. return GetWindowsVersion().dwBuildNumber;
  2937. }
  2938. //---------------------------------------------------------------------------
  2939. UnicodeString __fastcall WindowsVersion()
  2940. {
  2941. OSVERSIONINFO OSVersionInfo = GetWindowsVersion();
  2942. UnicodeString Result = FORMAT(L"%d.%d.%d", (int(OSVersionInfo.dwMajorVersion), int(OSVersionInfo.dwMinorVersion), int(OSVersionInfo.dwBuildNumber)));
  2943. return Result;
  2944. }
  2945. //---------------------------------------------------------------------------
  2946. UnicodeString __fastcall WindowsVersionLong()
  2947. {
  2948. UnicodeString Result = WindowsVersion();
  2949. AddToList(Result, Win32CSDVersion, L" ");
  2950. return Result;
  2951. }
  2952. //---------------------------------------------------------------------------
  2953. bool __fastcall IsDirectoryWriteable(const UnicodeString & Path)
  2954. {
  2955. UnicodeString FileName =
  2956. IncludeTrailingPathDelimiter(Path) +
  2957. FORMAT(L"wscp_%s_%d.tmp", (FormatDateTime(L"nnzzz", Now()), int(GetCurrentProcessId())));
  2958. HANDLE Handle = CreateFile(ApiPath(FileName).c_str(), GENERIC_READ | GENERIC_WRITE, 0, NULL,
  2959. CREATE_NEW, FILE_ATTRIBUTE_TEMPORARY | FILE_FLAG_DELETE_ON_CLOSE, 0);
  2960. bool Result = (Handle != INVALID_HANDLE_VALUE);
  2961. if (Result)
  2962. {
  2963. CloseHandle(Handle);
  2964. }
  2965. return Result;
  2966. }
  2967. //---------------------------------------------------------------------------
  2968. UnicodeString __fastcall FormatNumber(__int64 Number)
  2969. {
  2970. return FormatFloat(L"#,##0", Number);
  2971. }
  2972. //---------------------------------------------------------------------------
  2973. // simple alternative to FormatBytes
  2974. UnicodeString __fastcall FormatSize(__int64 Size)
  2975. {
  2976. return FormatNumber(Size);
  2977. }
  2978. //---------------------------------------------------------------------------
  2979. UnicodeString __fastcall FormatDateTimeSpan(const UnicodeString TimeFormat, TDateTime DateTime)
  2980. {
  2981. UnicodeString Result;
  2982. if (int(DateTime) > 0)
  2983. {
  2984. Result = IntToStr(int(DateTime)) + L", ";
  2985. }
  2986. // days are decremented, because when there are to many of them,
  2987. // "integer overflow" error occurs
  2988. Result += FormatDateTime(TimeFormat, DateTime - int(DateTime));
  2989. return Result;
  2990. }
  2991. //---------------------------------------------------------------------------
  2992. UnicodeString __fastcall ExtractFileBaseName(const UnicodeString & Path)
  2993. {
  2994. return ChangeFileExt(ExtractFileName(Path), L"");
  2995. }
  2996. //---------------------------------------------------------------------------
  2997. TStringList * __fastcall TextToStringList(const UnicodeString & Text)
  2998. {
  2999. std::unique_ptr<TStringList> List(new TStringList());
  3000. List->Text = Text;
  3001. return List.release();
  3002. }
  3003. //---------------------------------------------------------------------------
  3004. UnicodeString __fastcall StringsToText(TStrings * Strings)
  3005. {
  3006. UnicodeString Result;
  3007. if (Strings->Count == 1)
  3008. {
  3009. Result = Strings->Strings[0];
  3010. }
  3011. else
  3012. {
  3013. Result = Strings->Text;
  3014. }
  3015. return Result;
  3016. }
  3017. //---------------------------------------------------------------------------
  3018. TStrings * __fastcall CloneStrings(TStrings * Strings)
  3019. {
  3020. std::unique_ptr<TStringList> List(new TStringList());
  3021. List->AddStrings(Strings);
  3022. return List.release();
  3023. }
  3024. //---------------------------------------------------------------------------
  3025. UnicodeString __fastcall TrimVersion(UnicodeString Version)
  3026. {
  3027. while ((Version.Pos(L".") != Version.LastDelimiter(L".")) &&
  3028. (Version.SubString(Version.Length() - 1, 2) == L".0"))
  3029. {
  3030. Version.SetLength(Version.Length() - 2);
  3031. }
  3032. return Version;
  3033. }
  3034. //---------------------------------------------------------------------------
  3035. UnicodeString __fastcall FormatVersion(int MajovVersion, int MinorVersion, int Release)
  3036. {
  3037. return
  3038. TrimVersion(FORMAT(L"%d.%d.%d",
  3039. (MajovVersion, MinorVersion, Release)));
  3040. }
  3041. //---------------------------------------------------------------------------
  3042. TFormatSettings __fastcall GetEngFormatSettings()
  3043. {
  3044. return TFormatSettings::Create((TLocaleID)1033);
  3045. }
  3046. //---------------------------------------------------------------------------
  3047. int __fastcall ParseShortEngMonthName(const UnicodeString & MonthStr)
  3048. {
  3049. TFormatSettings FormatSettings = GetEngFormatSettings();
  3050. return IndexStr(MonthStr, FormatSettings.ShortMonthNames, FormatSettings.ShortMonthNames.Size()) + 1;
  3051. }
  3052. //---------------------------------------------------------------------------
  3053. TStringList * __fastcall CreateSortedStringList(bool CaseSensitive, System::Types::TDuplicates Duplicates)
  3054. {
  3055. TStringList * Result = new TStringList();
  3056. Result->CaseSensitive = CaseSensitive;
  3057. Result->Sorted = true;
  3058. Result->Duplicates = Duplicates;
  3059. return Result;
  3060. }
  3061. //---------------------------------------------------------------------------
  3062. static UnicodeString __fastcall NormalizeIdent(UnicodeString Ident)
  3063. {
  3064. int Index = 1;
  3065. while (Index <= Ident.Length())
  3066. {
  3067. if (Ident[Index] == L'-')
  3068. {
  3069. Ident.Delete(Index, 1);
  3070. }
  3071. else
  3072. {
  3073. Index++;
  3074. }
  3075. }
  3076. return Ident;
  3077. }
  3078. //---------------------------------------------------------------------------
  3079. UnicodeString __fastcall FindIdent(const UnicodeString & Ident, TStrings * Idents)
  3080. {
  3081. UnicodeString NormalizedIdent(NormalizeIdent(Ident));
  3082. for (int Index = 0; Index < Idents->Count; Index++)
  3083. {
  3084. if (SameText(NormalizedIdent, NormalizeIdent(Idents->Strings[Index])))
  3085. {
  3086. return Idents->Strings[Index];
  3087. }
  3088. }
  3089. return Ident;
  3090. }
  3091. //---------------------------------------------------------------------------
  3092. static UnicodeString __fastcall GetTlsErrorStr(int Err)
  3093. {
  3094. char * Buffer = new char[512];
  3095. ERR_error_string(Err, Buffer);
  3096. // not sure about the UTF8
  3097. return UnicodeString(UTF8String(Buffer));
  3098. }
  3099. //---------------------------------------------------------------------------
  3100. static FILE * __fastcall OpenCertificate(const UnicodeString & Path)
  3101. {
  3102. FILE * Result = _wfopen(ApiPath(Path).c_str(), L"rb");
  3103. if (Result == NULL)
  3104. {
  3105. int Error = errno;
  3106. throw EOSExtException(MainInstructions(FMTLOAD(CERTIFICATE_OPEN_ERROR, (Path))), Error);
  3107. }
  3108. return Result;
  3109. }
  3110. //---------------------------------------------------------------------------
  3111. struct TPemPasswordCallbackData
  3112. {
  3113. UnicodeString * Passphrase;
  3114. };
  3115. //---------------------------------------------------------------------------
  3116. static int PemPasswordCallback(char * Buf, int Size, int /*RWFlag*/, void * UserData)
  3117. {
  3118. TPemPasswordCallbackData & Data = *reinterpret_cast<TPemPasswordCallbackData *>(UserData);
  3119. UTF8String UtfPassphrase = UTF8String(*Data.Passphrase);
  3120. strncpy(Buf, UtfPassphrase.c_str(), Size);
  3121. Shred(UtfPassphrase);
  3122. Buf[Size - 1] = '\0';
  3123. return strlen(Buf);
  3124. }
  3125. //---------------------------------------------------------------------------
  3126. static bool __fastcall IsTlsPassphraseError(int Error, bool HasPassphrase)
  3127. {
  3128. int ErrorLib = ERR_GET_LIB(Error);
  3129. int ErrorReason = ERR_GET_REASON(Error);
  3130. bool Result =
  3131. ((ErrorLib == ERR_LIB_PKCS12) &&
  3132. (ErrorReason == PKCS12_R_MAC_VERIFY_FAILURE)) ||
  3133. ((ErrorLib == ERR_LIB_PEM) &&
  3134. (ErrorReason == PEM_R_BAD_PASSWORD_READ)) ||
  3135. (HasPassphrase && (ERR_LIB_EVP == ERR_LIB_EVP) &&
  3136. ((ErrorReason == PEM_R_BAD_DECRYPT) || (ErrorReason == PEM_R_BAD_BASE64_DECODE)));
  3137. return Result;
  3138. }
  3139. //---------------------------------------------------------------------------
  3140. static void __fastcall ThrowTlsCertificateErrorIgnorePassphraseErrors(const UnicodeString & Path, bool HasPassphrase)
  3141. {
  3142. int Error = ERR_get_error();
  3143. if (!IsTlsPassphraseError(Error, HasPassphrase))
  3144. {
  3145. throw ExtException(MainInstructions(FMTLOAD(CERTIFICATE_READ_ERROR, (Path))), GetTlsErrorStr(Error));
  3146. }
  3147. }
  3148. //---------------------------------------------------------------------------
  3149. void __fastcall ParseCertificate(const UnicodeString & Path,
  3150. const UnicodeString & Passphrase, X509 *& Certificate, EVP_PKEY *& PrivateKey,
  3151. bool & WrongPassphrase)
  3152. {
  3153. Certificate = NULL;
  3154. PrivateKey = NULL;
  3155. WrongPassphrase = false;
  3156. bool HasPassphrase = !Passphrase.IsEmpty();
  3157. FILE * File;
  3158. // Inspired by neon's ne_ssl_clicert_read
  3159. File = OpenCertificate(Path);
  3160. // openssl pkcs12 -inkey cert.pem -in cert.crt -export -out cert.pfx
  3161. // Binary file
  3162. PKCS12 * Pkcs12 = d2i_PKCS12_fp(File, NULL);
  3163. fclose(File);
  3164. if (Pkcs12 != NULL)
  3165. {
  3166. UTF8String PassphraseUtf(Passphrase);
  3167. bool Result =
  3168. (PKCS12_parse(Pkcs12, PassphraseUtf.c_str(), &PrivateKey, &Certificate, NULL) == 1);
  3169. PKCS12_free(Pkcs12);
  3170. if (!Result)
  3171. {
  3172. ThrowTlsCertificateErrorIgnorePassphraseErrors(Path, HasPassphrase);
  3173. WrongPassphrase = true;
  3174. }
  3175. }
  3176. else
  3177. {
  3178. ERR_clear_error();
  3179. TPemPasswordCallbackData CallbackUserData;
  3180. // PemPasswordCallback never writes to the .Passphrase
  3181. CallbackUserData.Passphrase = const_cast<UnicodeString *>(&Passphrase);
  3182. File = OpenCertificate(Path);
  3183. // Encrypted:
  3184. // openssl req -x509 -newkey rsa:2048 -keyout cert.pem -out cert.crt
  3185. // -----BEGIN ENCRYPTED PRIVATE KEY-----
  3186. // ...
  3187. // -----END ENCRYPTED PRIVATE KEY-----
  3188. // Not encrypted (add -nodes):
  3189. // -----BEGIN PRIVATE KEY-----
  3190. // ...
  3191. // -----END PRIVATE KEY-----
  3192. // Or (openssl genrsa -out client.key 1024 # used for certificate signing request)
  3193. // -----BEGIN RSA PRIVATE KEY-----
  3194. // ...
  3195. // -----END RSA PRIVATE KEY-----
  3196. PrivateKey = PEM_read_PrivateKey(File, NULL, PemPasswordCallback, &CallbackUserData);
  3197. fclose(File);
  3198. try
  3199. {
  3200. if (PrivateKey == NULL)
  3201. {
  3202. ThrowTlsCertificateErrorIgnorePassphraseErrors(Path, HasPassphrase);
  3203. WrongPassphrase = true;
  3204. }
  3205. File = OpenCertificate(Path);
  3206. // The file can contain both private and public key
  3207. // (basically cert.pem and cert.crt appended one to each other)
  3208. // -----BEGIN ENCRYPTED PRIVATE KEY-----
  3209. // ...
  3210. // -----END ENCRYPTED PRIVATE KEY-----
  3211. // -----BEGIN CERTIFICATE-----
  3212. // ...
  3213. // -----END CERTIFICATE-----
  3214. Certificate = PEM_read_X509(File, NULL, PemPasswordCallback, &CallbackUserData);
  3215. fclose(File);
  3216. if (Certificate == NULL)
  3217. {
  3218. int Error = ERR_get_error();
  3219. // unlikely
  3220. if (IsTlsPassphraseError(Error, HasPassphrase))
  3221. {
  3222. WrongPassphrase = true;
  3223. }
  3224. else
  3225. {
  3226. UnicodeString CertificatePath = ChangeFileExt(Path, L".cer");
  3227. if (!FileExists(CertificatePath))
  3228. {
  3229. CertificatePath = ChangeFileExt(Path, L".crt");
  3230. }
  3231. if (!FileExists(CertificatePath))
  3232. {
  3233. throw Exception(MainInstructions(FMTLOAD(CERTIFICATE_PUBLIC_KEY_NOT_FOUND, (Path))));
  3234. }
  3235. else
  3236. {
  3237. File = OpenCertificate(CertificatePath);
  3238. // -----BEGIN CERTIFICATE-----
  3239. // ...
  3240. // -----END CERTIFICATE-----
  3241. Certificate = PEM_read_X509(File, NULL, PemPasswordCallback, &CallbackUserData);
  3242. fclose(File);
  3243. if (Certificate == NULL)
  3244. {
  3245. int Base64Error = ERR_get_error();
  3246. File = OpenCertificate(CertificatePath);
  3247. // Binary DER-encoded certificate
  3248. // (as above, with BEGIN/END removed, and decoded from Base64 to binary)
  3249. // openssl x509 -in cert.crt -out client.der.crt -outform DER
  3250. Certificate = d2i_X509_fp(File, NULL);
  3251. fclose(File);
  3252. if (Certificate == NULL)
  3253. {
  3254. int DERError = ERR_get_error();
  3255. UnicodeString Message = MainInstructions(FMTLOAD(CERTIFICATE_READ_ERROR, (CertificatePath)));
  3256. UnicodeString MoreMessages =
  3257. FORMAT(L"Base64: %s\nDER: %s", (GetTlsErrorStr(Base64Error), GetTlsErrorStr(DERError)));
  3258. throw ExtException(Message, MoreMessages);
  3259. }
  3260. }
  3261. }
  3262. }
  3263. }
  3264. }
  3265. __finally
  3266. {
  3267. // We loaded private key, but failed to load certificate, discard the certificate
  3268. // (either exception was thrown or WrongPassphrase)
  3269. if ((PrivateKey != NULL) && (Certificate == NULL))
  3270. {
  3271. EVP_PKEY_free(PrivateKey);
  3272. PrivateKey = NULL;
  3273. }
  3274. // Certificate was verified, but passphrase was wrong when loading private key,
  3275. // so discard the certificate
  3276. else if ((Certificate != NULL) && (PrivateKey == NULL))
  3277. {
  3278. X509_free(Certificate);
  3279. Certificate = NULL;
  3280. }
  3281. }
  3282. }
  3283. }
  3284. //---------------------------------------------------------------------------
  3285. void __fastcall CheckCertificate(const UnicodeString & Path)
  3286. {
  3287. X509 * Certificate;
  3288. EVP_PKEY * PrivateKey;
  3289. bool WrongPassphrase;
  3290. ParseCertificate(Path, L"", Certificate, PrivateKey, WrongPassphrase);
  3291. if (PrivateKey != NULL)
  3292. {
  3293. EVP_PKEY_free(PrivateKey);
  3294. }
  3295. if (Certificate != NULL)
  3296. {
  3297. X509_free(Certificate);
  3298. }
  3299. }
  3300. //---------------------------------------------------------------------------
  3301. const UnicodeString HttpProtocol(L"http");
  3302. const UnicodeString HttpsProtocol(L"https");
  3303. const UnicodeString ProtocolSeparator(L"://");
  3304. //---------------------------------------------------------------------------
  3305. bool __fastcall IsHttpUrl(const UnicodeString & S)
  3306. {
  3307. return StartsText(HttpProtocol + ProtocolSeparator, S);
  3308. }
  3309. //---------------------------------------------------------------------------
  3310. bool __fastcall IsHttpOrHttpsUrl(const UnicodeString & S)
  3311. {
  3312. return
  3313. IsHttpUrl(S) ||
  3314. StartsText(HttpsProtocol + ProtocolSeparator, S);
  3315. }
  3316. //---------------------------------------------------------------------------
  3317. UnicodeString __fastcall ChangeUrlProtocol(const UnicodeString & S, const UnicodeString & Protocol)
  3318. {
  3319. int P = S.Pos(ProtocolSeparator);
  3320. DebugAssert(P > 0);
  3321. return Protocol + ProtocolSeparator + RightStr(S, S.Length() - P - ProtocolSeparator.Length() + 1);
  3322. }
  3323. //---------------------------------------------------------------------------
  3324. const UnicodeString RtfPara(TraceInitStr(L"\\par\n"));
  3325. const UnicodeString AssemblyNamespace(TraceInitStr(L"WinSCP"));
  3326. const UnicodeString TransferOptionsClassName(TraceInitStr(L"TransferOptions"));
  3327. const UnicodeString SessionClassName(TraceInitStr(L"Session"));
  3328. const UnicodeString RtfHyperlinkField(TraceInitStr(L"HYPERLINK"));
  3329. const UnicodeString RtfHyperlinkFieldPrefix(TraceInitStr(RtfHyperlinkField + L" \""));
  3330. const UnicodeString RtfHyperlinkFieldSuffix(TraceInitStr(L"\" "));
  3331. //---------------------------------------------------------------------
  3332. UnicodeString __fastcall RtfColor(int Index)
  3333. {
  3334. return FORMAT(L"\\cf%d", (Index));
  3335. }
  3336. //---------------------------------------------------------------------
  3337. UnicodeString __fastcall RtfText(const UnicodeString & Text, bool Rtf)
  3338. {
  3339. UnicodeString Result = Text;
  3340. if (Rtf)
  3341. {
  3342. int Index = 1;
  3343. while (Index <= Result.Length())
  3344. {
  3345. UnicodeString Replacement;
  3346. wchar_t Ch = Result[Index];
  3347. if ((Ch == L'\\') || (Ch == L'{') || (Ch == L'}'))
  3348. {
  3349. Replacement = FORMAT(L"\\%s", (Ch));
  3350. }
  3351. else if (Ch >= 0x0080)
  3352. {
  3353. Replacement = FORMAT(L"\\u%d?", (int(Ch)));
  3354. }
  3355. if (!Replacement.IsEmpty())
  3356. {
  3357. Result.Delete(Index, 1);
  3358. Result.Insert(Replacement, Index);
  3359. Index += Replacement.Length();
  3360. }
  3361. else
  3362. {
  3363. Index++;
  3364. }
  3365. }
  3366. }
  3367. return Result;
  3368. }
  3369. //---------------------------------------------------------------------
  3370. UnicodeString __fastcall RtfColorText(int Color, const UnicodeString & Text)
  3371. {
  3372. return RtfColor(Color) + L" " + RtfText(Text) + RtfColor(0) + L" ";
  3373. }
  3374. //---------------------------------------------------------------------
  3375. UnicodeString __fastcall RtfColorItalicText(int Color, const UnicodeString & Text)
  3376. {
  3377. return RtfColor(Color) + L"\\i " + RtfText(Text) + L"\\i0" + RtfColor(0) + L" ";
  3378. }
  3379. //---------------------------------------------------------------------
  3380. UnicodeString __fastcall RtfOverrideColorText(const UnicodeString & Text)
  3381. {
  3382. return RtfColorText(1, Text);
  3383. }
  3384. //---------------------------------------------------------------------
  3385. UnicodeString __fastcall RtfKeyword(const UnicodeString & Text)
  3386. {
  3387. return RtfColorText(5, Text);
  3388. }
  3389. //---------------------------------------------------------------------
  3390. UnicodeString __fastcall RtfParameter(const UnicodeString & Text)
  3391. {
  3392. return RtfColorText(6, Text);
  3393. }
  3394. //---------------------------------------------------------------------
  3395. UnicodeString __fastcall RtfString(const UnicodeString & Text)
  3396. {
  3397. return RtfColorText(4, Text);
  3398. }
  3399. //---------------------------------------------------------------------
  3400. UnicodeString __fastcall RtfLink(const UnicodeString & Link, const UnicodeString & RtfText)
  3401. {
  3402. return
  3403. L"{\\field{\\*\\fldinst{" + RtfHyperlinkFieldPrefix + Link + RtfHyperlinkFieldSuffix + L"}}{\\fldrslt{" +
  3404. RtfText + L"}}}";
  3405. }
  3406. //---------------------------------------------------------------------
  3407. UnicodeString __fastcall ScriptCommandLink(const UnicodeString & Command)
  3408. {
  3409. return L"scriptcommand_" + Command;
  3410. }
  3411. //---------------------------------------------------------------------
  3412. UnicodeString __fastcall RtfSwitch(
  3413. const UnicodeString & Switch, const UnicodeString & Link, bool Rtf)
  3414. {
  3415. UnicodeString Result = FORMAT(L"-%s", (Switch));
  3416. if (Rtf)
  3417. {
  3418. Result = RtfLink(Link + L"#" + Switch.LowerCase(), RtfParameter(Result));
  3419. }
  3420. return L" " + Result;
  3421. }
  3422. //---------------------------------------------------------------------
  3423. UnicodeString __fastcall RtfSwitchValue(
  3424. const UnicodeString & Name, const UnicodeString & Link, const UnicodeString & Value, bool Rtf)
  3425. {
  3426. return RtfSwitch(Name, Link, Rtf) + L"=" + Value;
  3427. }
  3428. //---------------------------------------------------------------------
  3429. UnicodeString __fastcall RtfSwitch(
  3430. const UnicodeString & Name, const UnicodeString & Link, const UnicodeString & Value, bool Rtf)
  3431. {
  3432. return RtfSwitchValue(Name, Link, RtfText(FORMAT("\"%s\"", (EscapeParam(Value))), Rtf), Rtf);
  3433. }
  3434. //---------------------------------------------------------------------
  3435. UnicodeString __fastcall RtfSwitch(
  3436. const UnicodeString & Name, const UnicodeString & Link, int Value, bool Rtf)
  3437. {
  3438. return RtfSwitchValue(Name, Link, RtfText(IntToStr(Value), Rtf), Rtf);
  3439. }
  3440. //---------------------------------------------------------------------
  3441. UnicodeString __fastcall RtfRemoveHyperlinks(UnicodeString Text)
  3442. {
  3443. // Remove all tags HYPERLINK "https://www.example.com".
  3444. // See also RtfEscapeParam
  3445. int Index = 1;
  3446. int P;
  3447. while ((P = PosEx(RtfHyperlinkFieldPrefix, Text, Index)) > 0)
  3448. {
  3449. int Index2 = P + RtfHyperlinkFieldPrefix.Length();
  3450. int P2 = PosEx(RtfHyperlinkFieldSuffix, Text, Index2);
  3451. if (P2 > 0)
  3452. {
  3453. Text.Delete(P, P2 - P + RtfHyperlinkFieldSuffix.Length());
  3454. }
  3455. else
  3456. {
  3457. Index = Index2;
  3458. }
  3459. }
  3460. return Text;
  3461. }
  3462. //---------------------------------------------------------------------
  3463. UnicodeString __fastcall RtfEscapeParam(UnicodeString Param, bool PowerShellEscape)
  3464. {
  3465. const UnicodeString Quote(L"\"");
  3466. UnicodeString Escape(Quote);
  3467. if (PowerShellEscape)
  3468. {
  3469. Escape = "`" + Escape + "`";
  3470. }
  3471. // Equivalent of EscapeParam, except that it does not double quotes in HYPERLINK.
  3472. // See also RtfRemoveHyperlinks.
  3473. int Index = 1;
  3474. while (true)
  3475. {
  3476. int P1 = PosEx(Quote, Param, Index);
  3477. if (P1 == 0)
  3478. {
  3479. // no more quotes
  3480. break;
  3481. }
  3482. else
  3483. {
  3484. int P2 = PosEx(RtfHyperlinkFieldPrefix, Param, Index);
  3485. int P3;
  3486. if ((P2 > 0) && (P2 < P1) && ((P3 = PosEx(RtfHyperlinkFieldSuffix, Param, P2)) > 0))
  3487. {
  3488. // skip HYPERLINK
  3489. Index = P3 + RtfHyperlinkFieldSuffix.Length();
  3490. }
  3491. else
  3492. {
  3493. Param.Insert(Escape, P1);
  3494. Index = P1 + (Escape.Length() + Quote.Length());
  3495. }
  3496. }
  3497. }
  3498. return Param;
  3499. }
  3500. //---------------------------------------------------------------------
  3501. static UnicodeString __fastcall RtfCodeComment(const UnicodeString & Text)
  3502. {
  3503. return RtfColorItalicText(2, Text);
  3504. }
  3505. //---------------------------------------------------------------------
  3506. UnicodeString __fastcall AssemblyCommentLine(TAssemblyLanguage Language, const UnicodeString & Text)
  3507. {
  3508. UnicodeString Prefix;
  3509. switch (Language)
  3510. {
  3511. case alCSharp:
  3512. Prefix = L"//";
  3513. break;
  3514. case alVBNET:
  3515. Prefix = L"'";
  3516. break;
  3517. case alPowerShell:
  3518. Prefix = L"#";
  3519. break;
  3520. }
  3521. return RtfCodeComment(Prefix + L" " + Text) + RtfPara;
  3522. }
  3523. //---------------------------------------------------------------------
  3524. UnicodeString __fastcall AssemblyString(TAssemblyLanguage Language, UnicodeString S)
  3525. {
  3526. switch (Language)
  3527. {
  3528. case alCSharp:
  3529. if (S.Pos(L"\\") > 0)
  3530. {
  3531. S = FORMAT(L"@\"%s\"", (ReplaceStr(S, L"\"", L"\"\"")));
  3532. }
  3533. else
  3534. {
  3535. S = FORMAT(L"\"%s\"", (ReplaceStr(S, L"\"", L"\\\"")));
  3536. }
  3537. break;
  3538. case alVBNET:
  3539. S = FORMAT(L"\"%s\"", (ReplaceStr(S, L"\"", L"\"\"")));
  3540. break;
  3541. case alPowerShell:
  3542. S = FORMAT(L"\"%s\"", (ReplaceStr(ReplaceStr(ReplaceStr(S, L"`", L"``"), L"$", L"`$"), L"\"", L"`\"")));
  3543. break;
  3544. default:
  3545. DebugFail();
  3546. break;
  3547. }
  3548. return RtfString(S);
  3549. }
  3550. //---------------------------------------------------------------------
  3551. static UnicodeString __fastcall RtfClass(const UnicodeString & Text)
  3552. {
  3553. return RtfColorText(3, Text);
  3554. }
  3555. //---------------------------------------------------------------------
  3556. UnicodeString __fastcall RtfLibraryClass(const UnicodeString & ClassName)
  3557. {
  3558. return RtfLink(L"library_" + ClassName.LowerCase(), RtfClass(ClassName));
  3559. }
  3560. //---------------------------------------------------------------------
  3561. UnicodeString __fastcall RtfLibraryMethod(const UnicodeString & ClassName, const UnicodeString & MethodName, bool InPage)
  3562. {
  3563. return RtfLink(L"library_" + ClassName.LowerCase() + (InPage ? L"#" : L"_") + MethodName.LowerCase(), RtfOverrideColorText(MethodName));
  3564. }
  3565. //---------------------------------------------------------------------
  3566. static UnicodeString __fastcall RtfLibraryProperty(const UnicodeString & ClassName, const UnicodeString & PropertyName)
  3567. {
  3568. return RtfLink(L"library_" + ClassName.LowerCase() + L"#" + PropertyName.LowerCase(), RtfOverrideColorText(PropertyName));
  3569. }
  3570. //---------------------------------------------------------------------
  3571. UnicodeString __fastcall AssemblyVariableName(TAssemblyLanguage Language, const UnicodeString & ClassName)
  3572. {
  3573. UnicodeString Result = ClassName.SubString(1, 1).LowerCase() + ClassName.SubString(2, ClassName.Length() - 1);
  3574. if (Language == alPowerShell)
  3575. {
  3576. Result = L"$" + Result;
  3577. }
  3578. return Result;
  3579. }
  3580. //---------------------------------------------------------------------
  3581. UnicodeString __fastcall AssemblyStatementSeparator(TAssemblyLanguage Language)
  3582. {
  3583. UnicodeString Result;
  3584. switch (Language)
  3585. {
  3586. case alCSharp:
  3587. Result = L";";
  3588. break;
  3589. case alVBNET:
  3590. case alPowerShell:
  3591. // noop
  3592. break;
  3593. }
  3594. return Result;
  3595. }
  3596. //---------------------------------------------------------------------
  3597. UnicodeString __fastcall AssemblyPropertyRaw(
  3598. TAssemblyLanguage Language, const UnicodeString & ClassName, const UnicodeString & Name,
  3599. const UnicodeString & Value, bool Inline)
  3600. {
  3601. UnicodeString Result;
  3602. UnicodeString RtfPropertyAndValue = RtfLibraryProperty(ClassName, Name) + L" = " + Value;
  3603. UnicodeString Indetation = (Inline ? L"" : L" ");
  3604. UnicodeString SpaceOrPara = (Inline ? UnicodeString(L" ") : RtfPara);
  3605. switch (Language)
  3606. {
  3607. case alCSharp:
  3608. Result = Indetation + RtfPropertyAndValue + (Inline ? L"" : L",") + SpaceOrPara;
  3609. break;
  3610. case alVBNET:
  3611. Result = Indetation + L"." + RtfPropertyAndValue + SpaceOrPara;
  3612. break;
  3613. case alPowerShell:
  3614. Result = Indetation + RtfPropertyAndValue + SpaceOrPara;
  3615. break;
  3616. }
  3617. return Result;
  3618. }
  3619. //---------------------------------------------------------------------
  3620. UnicodeString __fastcall AssemblyProperty(
  3621. TAssemblyLanguage Language, const UnicodeString & ClassName, const UnicodeString & Name,
  3622. const UnicodeString & Type, const UnicodeString & Member, bool Inline)
  3623. {
  3624. UnicodeString PropertyValue;
  3625. switch (Language)
  3626. {
  3627. case alCSharp:
  3628. case alVBNET:
  3629. PropertyValue = RtfClass(Type) + RtfText(L"." + Member);
  3630. break;
  3631. case alPowerShell:
  3632. PropertyValue = RtfText(L"[" + AssemblyNamespace + L".") + RtfClass(Type) + RtfText(L"]::" + Member);
  3633. break;
  3634. }
  3635. return AssemblyPropertyRaw(Language, ClassName, Name, PropertyValue, Inline);
  3636. }
  3637. //---------------------------------------------------------------------
  3638. UnicodeString __fastcall AssemblyProperty(
  3639. TAssemblyLanguage Language, const UnicodeString & ClassName,
  3640. const UnicodeString & Name, const UnicodeString & Value, bool Inline)
  3641. {
  3642. return AssemblyPropertyRaw(Language, ClassName, Name, AssemblyString(Language, Value), Inline);
  3643. }
  3644. //---------------------------------------------------------------------
  3645. UnicodeString __fastcall AssemblyProperty(
  3646. TAssemblyLanguage Language, const UnicodeString & ClassName,
  3647. const UnicodeString & Name, int Value, bool Inline)
  3648. {
  3649. return AssemblyPropertyRaw(Language, ClassName, Name, IntToStr(Value), Inline);
  3650. }
  3651. //---------------------------------------------------------------------
  3652. UnicodeString __fastcall AssemblyBoolean(TAssemblyLanguage Language, bool Value)
  3653. {
  3654. UnicodeString Result;
  3655. switch (Language)
  3656. {
  3657. case alCSharp:
  3658. Result = (Value ? L"true" : L"false");
  3659. break;
  3660. case alVBNET:
  3661. Result = (Value ? L"True" : L"False");
  3662. break;
  3663. case alPowerShell:
  3664. Result = (Value ? L"$True" : L"$False");
  3665. break;
  3666. }
  3667. return Result;
  3668. }
  3669. //---------------------------------------------------------------------
  3670. UnicodeString __fastcall AssemblyProperty(
  3671. TAssemblyLanguage Language, const UnicodeString & ClassName, const UnicodeString & Name, bool Value, bool Inline)
  3672. {
  3673. UnicodeString PropertyValue = AssemblyBoolean(Language, Value);
  3674. return AssemblyPropertyRaw(Language, ClassName, Name, PropertyValue, Inline);
  3675. }
  3676. //---------------------------------------------------------------------
  3677. UnicodeString __fastcall AssemblyNewClassInstance(TAssemblyLanguage Language, const UnicodeString & ClassName, bool Inline)
  3678. {
  3679. UnicodeString VariableName = AssemblyVariableName(Language, ClassName);
  3680. UnicodeString RtfClass = RtfLibraryClass(ClassName);
  3681. UnicodeString Result;
  3682. switch (Language)
  3683. {
  3684. case alCSharp:
  3685. if (!Inline)
  3686. {
  3687. Result += RtfClass + RtfText(L" " + VariableName + L" = ");
  3688. }
  3689. Result += RtfKeyword(L"new") + RtfText(L" ") + RtfClass;
  3690. break;
  3691. case alVBNET:
  3692. if (!Inline)
  3693. {
  3694. Result += RtfText(VariableName + L" ") + RtfKeyword(L"As") + RtfText(L" ");
  3695. }
  3696. Result += RtfKeyword(L"New") + RtfText(" ") + RtfClass;
  3697. break;
  3698. case alPowerShell:
  3699. if (!Inline)
  3700. {
  3701. Result += RtfText(VariableName + L" = ");
  3702. }
  3703. Result += RtfKeyword(L"New-Object") + RtfText(L" " + AssemblyNamespace + L".") + RtfClass;
  3704. break;
  3705. }
  3706. return Result;
  3707. }
  3708. //---------------------------------------------------------------------
  3709. UnicodeString __fastcall AssemblyVariableDeclaration(TAssemblyLanguage Language)
  3710. {
  3711. UnicodeString Result;
  3712. switch (Language)
  3713. {
  3714. case alVBNET:
  3715. Result = RtfKeyword(L"Dim") + RtfText(L" ");
  3716. break;
  3717. }
  3718. return Result;
  3719. }
  3720. //---------------------------------------------------------------------
  3721. UnicodeString __fastcall AssemblyNewClassInstanceStart(
  3722. TAssemblyLanguage Language, const UnicodeString & ClassName, bool Inline)
  3723. {
  3724. UnicodeString SpaceOrPara = (Inline ? UnicodeString(L" ") : RtfPara);
  3725. UnicodeString Result;
  3726. if (!Inline)
  3727. {
  3728. Result += AssemblyVariableDeclaration(Language);
  3729. }
  3730. Result += AssemblyNewClassInstance(Language, ClassName, Inline);
  3731. switch (Language)
  3732. {
  3733. case alCSharp:
  3734. Result += SpaceOrPara + RtfText(L"{") + SpaceOrPara;
  3735. break;
  3736. case alVBNET:
  3737. // Historically we use Dim .. With instead of object initilizer.
  3738. // But for inline use, we have to use object initialize.
  3739. // We should consistently always use object initilizers.
  3740. // Unfortunatelly VB.NET object initializer (contrary to C#) does not allow trailing comma.
  3741. Result += SpaceOrPara + RtfKeyword(L"With");
  3742. if (Inline)
  3743. {
  3744. Result += RtfText(L" { ");
  3745. }
  3746. else
  3747. {
  3748. Result += RtfText(L" " + AssemblyVariableName(Language, ClassName)) + RtfPara;
  3749. }
  3750. break;
  3751. case alPowerShell:
  3752. Result += RtfText(" -Property @{") + SpaceOrPara;
  3753. break;
  3754. }
  3755. return Result;
  3756. }
  3757. //---------------------------------------------------------------------
  3758. UnicodeString __fastcall AssemblyNewClassInstanceEnd(TAssemblyLanguage Language, bool Inline)
  3759. {
  3760. UnicodeString InlineEnd = RtfText(L"}");
  3761. UnicodeString Result;
  3762. switch (Language)
  3763. {
  3764. case alCSharp:
  3765. if (Inline)
  3766. {
  3767. Result = InlineEnd;
  3768. }
  3769. else
  3770. {
  3771. Result = RtfText(L"};") + RtfPara;
  3772. }
  3773. break;
  3774. case alVBNET:
  3775. if (Inline)
  3776. {
  3777. Result = InlineEnd;
  3778. }
  3779. else
  3780. {
  3781. Result = RtfKeyword(L"End With") + RtfPara;
  3782. }
  3783. break;
  3784. case alPowerShell:
  3785. if (Inline)
  3786. {
  3787. Result = InlineEnd;
  3788. }
  3789. else
  3790. {
  3791. Result = RtfText(L"}") + RtfPara;
  3792. }
  3793. break;
  3794. }
  3795. return Result;
  3796. }
  3797. //---------------------------------------------------------------------------
  3798. UnicodeString __fastcall AssemblyAddRawSettings(
  3799. TAssemblyLanguage Language, TStrings * RawSettings, const UnicodeString & ClassName,
  3800. const UnicodeString & MethodName)
  3801. {
  3802. UnicodeString Result;
  3803. for (int Index = 0; Index < RawSettings->Count; Index++)
  3804. {
  3805. UnicodeString Name = RawSettings->Names[Index];
  3806. UnicodeString Value = RawSettings->ValueFromIndex[Index];
  3807. UnicodeString AddRawSettingsMethod =
  3808. RtfLibraryMethod(ClassName, MethodName, false) +
  3809. FORMAT(L"(%s, %s)", (AssemblyString(Language, Name), AssemblyString(Language, Value)));
  3810. UnicodeString VariableName = AssemblyVariableName(Language, ClassName);
  3811. Result += RtfText(VariableName + L".") + AddRawSettingsMethod + AssemblyStatementSeparator(Language) + RtfPara;
  3812. }
  3813. return Result;
  3814. }
  3815. //---------------------------------------------------------------------------
  3816. void __fastcall LoadScriptFromFile(UnicodeString FileName, TStrings * Lines, bool FallbackToAnsi)
  3817. {
  3818. std::auto_ptr<TFileStream> Stream(new TFileStream(ApiPath(FileName), fmOpenRead | fmShareDenyWrite));
  3819. Lines->DefaultEncoding = TEncoding::UTF8;
  3820. try
  3821. {
  3822. Lines->LoadFromStream(Stream.get());
  3823. }
  3824. catch (EEncodingError & E)
  3825. {
  3826. if (FallbackToAnsi)
  3827. {
  3828. Lines->DefaultEncoding = TEncoding::ANSI;
  3829. Lines->Clear();
  3830. Stream->Position = 0;
  3831. Lines->LoadFromStream(Stream.get());
  3832. }
  3833. else
  3834. {
  3835. throw ExtException(LoadStr(TEXT_FILE_ENCODING), &E);
  3836. }
  3837. }
  3838. }
  3839. //---------------------------------------------------------------------------
  3840. UnicodeString __fastcall StripEllipsis(const UnicodeString & S)
  3841. {
  3842. UnicodeString Result = S;
  3843. if (Result.SubString(Result.Length() - Ellipsis.Length() + 1, Ellipsis.Length()) == Ellipsis)
  3844. {
  3845. Result.SetLength(Result.Length() - Ellipsis.Length());
  3846. Result = Result.TrimRight();
  3847. }
  3848. return Result;
  3849. }
  3850. //---------------------------------------------------------------------------
  3851. UnicodeString __fastcall GetFileMimeType(const UnicodeString & FileName)
  3852. {
  3853. wchar_t * MimeOut = NULL;
  3854. UnicodeString Result;
  3855. if (FindMimeFromData(NULL, FileName.c_str(), NULL, 0, NULL, FMFD_URLASFILENAME, &MimeOut, 0) == S_OK)
  3856. {
  3857. Result = MimeOut;
  3858. CoTaskMemFree(MimeOut);
  3859. }
  3860. return Result;
  3861. }
  3862. //---------------------------------------------------------------------------
  3863. TStrings * TlsCipherList()
  3864. {
  3865. // OpenSSL initialization happens in NeonInitialize
  3866. std::unique_ptr<TStrings> Result(new TStringList());
  3867. const SSL_METHOD * Method = DTLS_client_method();
  3868. SSL_CTX * Ctx = SSL_CTX_new(Method);
  3869. SSL * Ssl = SSL_new(Ctx);
  3870. int Index = 0;
  3871. const char * CipherName;
  3872. do
  3873. {
  3874. CipherName = SSL_get_cipher_list(Ssl, Index);
  3875. Index++;
  3876. if (CipherName != NULL)
  3877. {
  3878. Result->Add(UnicodeString(CipherName));
  3879. }
  3880. }
  3881. while (CipherName != NULL);
  3882. return Result.release();
  3883. }
  3884. //---------------------------------------------------------------------------
  3885. void SetStringValueEvenIfEmpty(TStrings * Strings, const UnicodeString & Name, const UnicodeString & Value)
  3886. {
  3887. if (Value.IsEmpty())
  3888. {
  3889. int Index = Strings->IndexOfName(Name);
  3890. if (Index < 0)
  3891. {
  3892. Index = Strings->Add(L"");
  3893. }
  3894. UnicodeString Line = Name + Strings->NameValueSeparator;
  3895. Strings->Strings[Index] = Line;
  3896. }
  3897. else
  3898. {
  3899. Strings->Values[Name] = Value;
  3900. }
  3901. }
  3902. //---------------------------------------------------------------------------
  3903. DWORD __fastcall GetParentProcessId(HANDLE Snapshot, DWORD ProcessId)
  3904. {
  3905. DWORD Result = 0;
  3906. PROCESSENTRY32 ProcessEntry;
  3907. memset(&ProcessEntry, sizeof(ProcessEntry), 0);
  3908. ProcessEntry.dwSize = sizeof(PROCESSENTRY32);
  3909. if (Process32First(Snapshot, &ProcessEntry))
  3910. {
  3911. do
  3912. {
  3913. if (ProcessEntry.th32ProcessID == ProcessId)
  3914. {
  3915. Result = ProcessEntry.th32ParentProcessID;
  3916. }
  3917. } while (Process32Next(Snapshot, &ProcessEntry));
  3918. }
  3919. return Result;
  3920. }
  3921. //---------------------------------------------------------------------------
  3922. static UnicodeString GetProcessName(DWORD ProcessId)
  3923. {
  3924. UnicodeString Result;
  3925. HANDLE Process = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, ProcessId);
  3926. // is common, when the parent process is installer, so we ignore it
  3927. if (Process)
  3928. {
  3929. Result.SetLength(MAX_PATH);
  3930. DWORD Len = GetModuleFileNameEx(Process, NULL, Result.c_str(), Result.Length());
  3931. Result.SetLength(Len);
  3932. // is common too, for some reason
  3933. if (!Result.IsEmpty())
  3934. {
  3935. Result = ExtractProgramName(FormatCommand(Result, UnicodeString()));
  3936. }
  3937. CloseHandle(Process);
  3938. }
  3939. return Result;
  3940. }
  3941. //---------------------------------------------------------------------------
  3942. UnicodeString ParentProcessName;
  3943. //---------------------------------------------------------------------------
  3944. UnicodeString __fastcall GetAncestorProcessName(int Levels)
  3945. {
  3946. UnicodeString Result;
  3947. bool Parent = (Levels == 1);
  3948. if (Parent && !ParentProcessName.IsEmpty())
  3949. {
  3950. Result = ParentProcessName;
  3951. }
  3952. else
  3953. {
  3954. try
  3955. {
  3956. HANDLE Snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  3957. DWORD ProcessId = GetCurrentProcessId();
  3958. typedef std::vector<DWORD> TProcesses;
  3959. TProcesses Processes;
  3960. // Either more to go (>0) or collecting all levels (-1 from GetAncestorProcessNames)
  3961. while ((Levels != 0) &&
  3962. (Levels > -20) && // prevent infinite loops
  3963. (ProcessId != 0))
  3964. {
  3965. const UnicodeString Sep(L", ");
  3966. ProcessId = GetParentProcessId(Snapshot, ProcessId);
  3967. // When ancestor process is terminated and another process reuses its ID, we may get a cycle.
  3968. TProcesses::const_iterator I = std::find(Processes.begin(), Processes.end(), ProcessId);
  3969. if (I != Processes.end())
  3970. {
  3971. int Index = I - Processes.begin();
  3972. AddToList(Result, FORMAT(L"cycle-%d", (Index)), Sep);
  3973. ProcessId = 0;
  3974. }
  3975. else
  3976. {
  3977. Processes.push_back(ProcessId);
  3978. if ((Levels < 0) && (ProcessId != 0))
  3979. {
  3980. UnicodeString Name = GetProcessName(ProcessId);
  3981. if (Name.IsEmpty())
  3982. {
  3983. Name = L"...";
  3984. ProcessId = 0;
  3985. }
  3986. AddToList(Result, Name, Sep);
  3987. }
  3988. Levels--;
  3989. }
  3990. }
  3991. if (Levels >= 0)
  3992. {
  3993. if (ProcessId == 0)
  3994. {
  3995. Result = L"err-notfound";
  3996. }
  3997. else
  3998. {
  3999. Result = GetProcessName(ProcessId);
  4000. }
  4001. }
  4002. else if (Result.IsEmpty())
  4003. {
  4004. Result = L"n/a";
  4005. }
  4006. CloseHandle(Snapshot);
  4007. }
  4008. catch (...)
  4009. {
  4010. Result = L"err-except";
  4011. }
  4012. if (Parent)
  4013. {
  4014. ParentProcessName = Result;
  4015. }
  4016. }
  4017. return Result;
  4018. }
  4019. //---------------------------------------------------------------------------
  4020. UnicodeString AncestorProcessNames;
  4021. //---------------------------------------------------------------------------
  4022. UnicodeString GetAncestorProcessNames()
  4023. {
  4024. if (AncestorProcessNames.IsEmpty())
  4025. {
  4026. AncestorProcessNames = GetAncestorProcessName(-1);
  4027. }
  4028. return AncestorProcessNames;
  4029. }