WinMain.cpp 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173
  1. //---------------------------------------------------------------------------
  2. #include <vcl.h>
  3. #pragma hdrstop
  4. #include <CoreMain.h>
  5. #include <TextsWin.h>
  6. #include <TextsCore.h>
  7. #include <HelpWin.h>
  8. #include "CustomScpExplorer.h"
  9. #include "TerminalManager.h"
  10. #include "NonVisual.h"
  11. #include "Glyphs.h"
  12. #include "ProgParams.h"
  13. #include "Setup.h"
  14. #include "WinConfiguration.h"
  15. #include "GUITools.h"
  16. #include "Tools.h"
  17. #include "WinApi.h"
  18. #include <DateUtils.hpp>
  19. //---------------------------------------------------------------------------
  20. #pragma package(smart_init)
  21. //---------------------------------------------------------------------------
  22. void __fastcall GetLoginData(UnicodeString SessionName, TOptions * Options,
  23. TObjectList * DataList, UnicodeString & DownloadFile, bool NeedSession, TForm * LinkedForm)
  24. {
  25. bool DefaultsOnly = false;
  26. UnicodeString FolderOrWorkspaceName = DecodeUrlChars(SessionName);
  27. if (StoredSessions->IsFolder(FolderOrWorkspaceName) ||
  28. StoredSessions->IsWorkspace(FolderOrWorkspaceName))
  29. {
  30. StoredSessions->GetFolderOrWorkspace(FolderOrWorkspaceName, DataList);
  31. }
  32. else
  33. {
  34. TSessionData * SessionData =
  35. StoredSessions->ParseUrl(SessionName, Options, DefaultsOnly,
  36. &DownloadFile);
  37. DataList->Add(SessionData);
  38. if (DataList->Count == 1)
  39. {
  40. TSessionData * SessionData = DebugNotNull(dynamic_cast<TSessionData *>(DataList->Items[0]));
  41. if (SessionData->SaveOnly)
  42. {
  43. Configuration->Usage->Inc(L"CommandLineSessionSave");
  44. TSessionData * SavedSession = DoSaveSession(SessionData, NULL, true, NULL);
  45. if (SavedSession == NULL)
  46. {
  47. Abort();
  48. }
  49. WinConfiguration->LastStoredSession = SavedSession->Name;
  50. DataList->Clear();
  51. }
  52. else if (!SessionData->PuttyProtocol.IsEmpty())
  53. {
  54. // putty does not support resolving environment variables in session settings
  55. // though it's hardly of any use here.
  56. SessionData->ExpandEnvironmentVariables();
  57. OpenSessionInPutty(GUIConfiguration->PuttyPath, SessionData);
  58. DataList->Clear();
  59. Abort();
  60. }
  61. }
  62. }
  63. if (DefaultsOnly && !NeedSession)
  64. {
  65. // No URL specified on command-line and no explicit command-line parameter
  66. // that requires session was specified => noop
  67. DataList->Clear();
  68. }
  69. else if ((DataList->Count == 0) ||
  70. !dynamic_cast<TSessionData *>(DataList->Items[0])->CanLogin ||
  71. DefaultsOnly)
  72. {
  73. // Note that GetFolderOrWorkspace never returns sites that !CanLogin,
  74. // so we should not get here with more then one site.
  75. // Though we should be good, if we ever do.
  76. // We get here when:
  77. // - we need session for explicit command-line operation
  78. // - after we handle "save" URL.
  79. // - the specified session does not contain enough information to login [= not even hostname]
  80. DebugAssert(DataList->Count <= 1);
  81. if (!DoLoginDialog(StoredSessions, DataList, LinkedForm))
  82. {
  83. Abort();
  84. }
  85. }
  86. }
  87. //---------------------------------------------------------------------------
  88. void __fastcall Upload(TTerminal * Terminal, TStrings * FileList, bool UseDefaults)
  89. {
  90. UnicodeString TargetDirectory;
  91. TGUICopyParamType CopyParam = GUIConfiguration->DefaultCopyParam;
  92. TargetDirectory = UnixIncludeTrailingBackslash(Terminal->CurrentDirectory);
  93. std::unique_ptr<TSessionData> Data(Terminal->SessionData->Clone());
  94. Terminal->FillSessionDataForCode(Data.get());
  95. int Options = coDisableQueue;
  96. int CopyParamAttrs = Terminal->UsableCopyParamAttrs(0).Upload;
  97. if (UseDefaults ||
  98. DoCopyDialog(true, false, FileList, TargetDirectory, &CopyParam, Options,
  99. CopyParamAttrs, Data.get(), NULL))
  100. {
  101. Terminal->CopyToRemote(FileList, TargetDirectory, &CopyParam, 0, NULL);
  102. }
  103. }
  104. //---------------------------------------------------------------------------
  105. void __fastcall Download(TTerminal * Terminal, const UnicodeString FileName,
  106. bool UseDefaults)
  107. {
  108. TRemoteFile * File = NULL;
  109. try
  110. {
  111. Terminal->ExceptionOnFail = true;
  112. try
  113. {
  114. Terminal->ReadFile(FileName, File);
  115. }
  116. __finally
  117. {
  118. Terminal->ExceptionOnFail = false;
  119. }
  120. File->FullFileName = FileName;
  121. UnicodeString LocalDirectory = ExpandFileName(Terminal->SessionData->LocalDirectory);
  122. if (LocalDirectory.IsEmpty())
  123. {
  124. LocalDirectory = GetPersonalFolder();
  125. }
  126. UnicodeString TargetDirectory = IncludeTrailingBackslash(LocalDirectory);
  127. TGUICopyParamType CopyParam = GUIConfiguration->DefaultCopyParam;
  128. UnicodeString DisplayName = File->FileName;
  129. bool CustomDisplayName =
  130. !File->DisplayName.IsEmpty() &&
  131. (File->DisplayName != DisplayName);
  132. if (CustomDisplayName)
  133. {
  134. DisplayName = File->DisplayName;
  135. }
  136. UnicodeString FriendyFileName = UnixIncludeTrailingBackslash(UnixExtractFilePath(FileName)) + DisplayName;
  137. std::unique_ptr<TStrings> FileListFriendly(new TStringList());
  138. FileListFriendly->AddObject(FriendyFileName, File);
  139. int Options = coDisableQueue;
  140. int CopyParamAttrs = Terminal->UsableCopyParamAttrs(0).Download;
  141. if (UseDefaults ||
  142. DoCopyDialog(false, false, FileListFriendly.get(), TargetDirectory, &CopyParam,
  143. Options, CopyParamAttrs, NULL, NULL))
  144. {
  145. if (CustomDisplayName)
  146. {
  147. // Set only now, so that it is not redundantly displayed on the copy dialog.
  148. // We should escape the * and ?'s.
  149. CopyParam.FileMask = DisplayName;
  150. }
  151. std::unique_ptr<TStrings> FileList(new TStringList());
  152. FileList->AddObject(FileName, File);
  153. Terminal->CopyToLocal(FileList.get(), TargetDirectory, &CopyParam, 0, NULL);
  154. }
  155. UnicodeString Directory = UnixExtractFilePath(FileName);
  156. Terminal->AutoReadDirectory = true;
  157. Terminal->ChangeDirectory(Directory);
  158. }
  159. __finally
  160. {
  161. delete File;
  162. }
  163. }
  164. //---------------------------------------------------------------------------
  165. void __fastcall Edit(TCustomScpExplorerForm * ScpExplorer, TStrings * FileList)
  166. {
  167. ScpExplorer->StandaloneEdit(FileList->Strings[0]);
  168. Abort();
  169. }
  170. //---------------------------------------------------------------------------
  171. void __fastcall SynchronizeDirectories(TTerminal * Terminal,
  172. TStrings * CommandParams,
  173. UnicodeString & LocalDirectory, UnicodeString & RemoteDirectory)
  174. {
  175. if (CommandParams->Count >= 1)
  176. {
  177. LocalDirectory = CommandParams->Strings[0];
  178. }
  179. else if (!Terminal->SessionData->LocalDirectory.IsEmpty())
  180. {
  181. LocalDirectory = ExpandFileName(Terminal->SessionData->LocalDirectory);
  182. }
  183. else
  184. {
  185. LocalDirectory = WinConfiguration->ScpExplorer.LastLocalTargetDirectory;
  186. }
  187. if (CommandParams->Count >= 2)
  188. {
  189. RemoteDirectory = CommandParams->Strings[1];
  190. }
  191. else
  192. {
  193. RemoteDirectory = Terminal->CurrentDirectory;
  194. }
  195. }
  196. //---------------------------------------------------------------------------
  197. void __fastcall FullSynchronize(TTerminal * Terminal, TCustomScpExplorerForm * ScpExplorer,
  198. TStrings * CommandParams, bool UseDefaults)
  199. {
  200. UnicodeString LocalDirectory;
  201. UnicodeString RemoteDirectory;
  202. SynchronizeDirectories(Terminal, CommandParams, LocalDirectory, RemoteDirectory);
  203. bool SaveMode = true;
  204. // bit ugly
  205. TSynchronizeMode Mode = (TSynchronizeMode)GUIConfiguration->SynchronizeMode;
  206. if (ScpExplorer->DoFullSynchronizeDirectories(LocalDirectory,
  207. RemoteDirectory, Mode, SaveMode, UseDefaults))
  208. {
  209. if (SaveMode)
  210. {
  211. GUIConfiguration->SynchronizeMode = Mode;
  212. }
  213. Terminal->CloseOnCompletion();
  214. }
  215. else
  216. {
  217. Abort();
  218. }
  219. }
  220. //---------------------------------------------------------------------------
  221. void __fastcall Synchronize(TTerminal * Terminal, TCustomScpExplorerForm * ScpExplorer,
  222. TStrings * CommandParams, bool UseDefaults)
  223. {
  224. UnicodeString LocalDirectory;
  225. UnicodeString RemoteDirectory;
  226. SynchronizeDirectories(Terminal, CommandParams, LocalDirectory, RemoteDirectory);
  227. // Undocumented syntax for "Start in New Window"
  228. if (CommandParams->Count >= 4)
  229. {
  230. GUIConfiguration->SynchronizeParams = StrToIntDef(CommandParams->Strings[2], -1);
  231. GUIConfiguration->SynchronizeOptions = StrToIntDef(CommandParams->Strings[3], -1);
  232. Configuration->DontSave();
  233. }
  234. ScpExplorer->DoSynchronizeDirectories(LocalDirectory, RemoteDirectory, UseDefaults);
  235. Abort();
  236. }
  237. //---------------------------------------------------------------------------
  238. void __fastcall ImportSitesIfAny()
  239. {
  240. if (!WinConfiguration->AutoImportedFromPuttyOrFilezilla)
  241. {
  242. bool AnyPuttySession = GUIConfiguration->AnyPuttySessionForImport(StoredSessions);
  243. bool AnyFilezillaSession = GUIConfiguration->AnyFilezillaSessionForImport(StoredSessions);
  244. if (AnyPuttySession || AnyFilezillaSession)
  245. {
  246. UnicodeString PuttySource = LoadStrPart(IMPORT_SESSIONS2, 2);
  247. UnicodeString FilezillaSource = LoadStrPart(IMPORT_SESSIONS2, 3);
  248. UnicodeString Source;
  249. if (AnyPuttySession && AnyFilezillaSession)
  250. {
  251. Source = FORMAT(LoadStrPart(IMPORT_SESSIONS2, 4), (PuttySource, FilezillaSource));
  252. }
  253. else if (AnyPuttySession)
  254. {
  255. Source = PuttySource;
  256. }
  257. else if (AnyFilezillaSession)
  258. {
  259. Source = FilezillaSource;
  260. }
  261. else
  262. {
  263. DebugFail();
  264. }
  265. UnicodeString Message = FORMAT(LoadStrPart(IMPORT_SESSIONS2, 1), (Source));
  266. if (MessageDialog(Message, qtConfirmation,
  267. qaYes | qaNo, HELP_IMPORT_SESSIONS) == qaYes)
  268. {
  269. DoImportSessionsDialog(NULL);
  270. }
  271. WinConfiguration->AutoImportedFromPuttyOrFilezilla = true;
  272. }
  273. }
  274. }
  275. //---------------------------------------------------------------------------
  276. void __fastcall Usage(UnicodeString Param)
  277. {
  278. while (!Param.IsEmpty())
  279. {
  280. UnicodeString Pair = CutToChar(Param, L',', true);
  281. if (!Pair.IsEmpty())
  282. {
  283. if (Pair[Pair.Length()] == L'+')
  284. {
  285. UnicodeString Key = Pair.SubString(1, Pair.Length() - 1).Trim();
  286. Configuration->Usage->Inc(Key);
  287. }
  288. else
  289. {
  290. UnicodeString Key = CutToChar(Pair, L':', true);
  291. Configuration->Usage->Set(Key, Pair.Trim());
  292. }
  293. }
  294. }
  295. }
  296. //---------------------------------------------------------------------------
  297. void __fastcall RecordWrapperVersions(UnicodeString ConsoleVersion, UnicodeString DotNetVersion)
  298. {
  299. TUpdatesConfiguration Updates = WinConfiguration->Updates;
  300. if (!DotNetVersion.IsEmpty())
  301. {
  302. Updates.DotNetVersion = DotNetVersion;
  303. }
  304. if (!ConsoleVersion.IsEmpty())
  305. {
  306. Updates.ConsoleVersion = ConsoleVersion;
  307. }
  308. WinConfiguration->Updates = Updates;
  309. if (Configuration->Storage == stNul)
  310. {
  311. Configuration->SetDefaultStorage();
  312. try
  313. {
  314. THierarchicalStorage * Storage = Configuration->CreateConfigStorage();
  315. try
  316. {
  317. Storage->AccessMode = smReadWrite;
  318. if (Storage->OpenSubKey(Configuration->ConfigurationSubKey, true) &&
  319. Storage->OpenSubKey(L"Interface\\Updates", true, true))
  320. {
  321. if (!DotNetVersion.IsEmpty())
  322. {
  323. Storage->WriteString(L"DotNetVersion", DotNetVersion);
  324. }
  325. if (!ConsoleVersion.IsEmpty())
  326. {
  327. Storage->WriteString(L"ConsoleVersion", ConsoleVersion);
  328. }
  329. }
  330. }
  331. __finally
  332. {
  333. delete Storage;
  334. }
  335. }
  336. __finally
  337. {
  338. Configuration->SetNulStorage();
  339. }
  340. }
  341. }
  342. //---------------------------------------------------------------------------
  343. static UnicodeString ColorToRGBStr(TColor Color)
  344. {
  345. int RGB = ColorToRGB(Color);
  346. int R = GetRValue(RGB);
  347. int G = GetGValue(RGB);
  348. int B = GetBValue(RGB);
  349. UnicodeString Result = FORMAT(L"%.2x%.2x%.2x", (R, G, B));
  350. return Result;
  351. }
  352. //---------------------------------------------------------------------------
  353. void __fastcall UpdateStaticUsage()
  354. {
  355. Configuration->Usage->Inc(L"Runs");
  356. Configuration->Usage->UpdateCurrentVersion();
  357. Configuration->Usage->Set(L"WindowsVersion", (WindowsVersionLong()));
  358. Configuration->Usage->Set(L"WindowsProductName", (WindowsProductName()));
  359. DWORD Type;
  360. GetWindowsProductType(Type);
  361. Configuration->Usage->Set(L"WindowsProductType", (static_cast<int>(Type)));
  362. Configuration->Usage->Set(L"Windows64", IsWin64());
  363. Configuration->Usage->Set(L"UWP", IsUWP());
  364. Configuration->Usage->Set(L"DefaultLocale",
  365. // See TGUIConfiguration::GetAppliedLocaleHex()
  366. IntToHex(static_cast<int>(GetDefaultLCID()), 4));
  367. Configuration->Usage->Set(L"Locale", WinConfiguration->AppliedLocaleHex);
  368. Configuration->Usage->Set(L"EncodingMultiByteAnsi", !TEncoding::Default->IsSingleByte);
  369. Configuration->Usage->Set(L"PixelsPerInch", Screen->PixelsPerInch);
  370. bool PixelsPerInchSystemDiffers = false;
  371. bool PixelsPerInchMonitorsDiffer = false;
  372. bool PixelsPerInchAxesDiffer = false;
  373. HINSTANCE ShCoreLibrary = LoadLibrary(L"shcore.dll");
  374. if (ShCoreLibrary != NULL)
  375. {
  376. GetDpiForMonitorProc GetDpiForMonitor =
  377. (GetDpiForMonitorProc)GetProcAddress(ShCoreLibrary, "GetDpiForMonitor");
  378. if (GetDpiForMonitor != NULL)
  379. {
  380. unsigned int PrimaryDpiX;
  381. unsigned int PrimaryDpiY;
  382. for (int Index = 0; Index < Screen->MonitorCount; Index++)
  383. {
  384. unsigned int DpiX;
  385. unsigned int DpiY;
  386. GetDpiForMonitor(Screen->Monitors[Index]->Handle, MDT_Default, &DpiX, &DpiY);
  387. if (DpiX != DpiY)
  388. {
  389. PixelsPerInchAxesDiffer = true;
  390. }
  391. if (Index == 0)
  392. {
  393. PrimaryDpiX = DpiX;
  394. PrimaryDpiY = DpiY;
  395. // PixelsPerInch is GetDeviceCaps(DC, LOGPIXELSY)
  396. if (DpiY != (unsigned int)Screen->PixelsPerInch)
  397. {
  398. PixelsPerInchSystemDiffers = true;
  399. }
  400. }
  401. else
  402. {
  403. if ((DpiX != PrimaryDpiX) ||
  404. (DpiY != PrimaryDpiY))
  405. {
  406. PixelsPerInchMonitorsDiffer = true;
  407. }
  408. }
  409. }
  410. }
  411. }
  412. if (PixelsPerInchSystemDiffers)
  413. {
  414. Configuration->Usage->Inc(L"PixelsPerInchSystemDiffered");
  415. }
  416. Configuration->Usage->Set(L"PixelsPerInchMonitorsDiffer", PixelsPerInchMonitorsDiffer);
  417. Configuration->Usage->Set(L"PixelsPerInchAxesDiffer", PixelsPerInchAxesDiffer);
  418. Configuration->Usage->Set(L"WorkAreaWidth", Screen->WorkAreaWidth);
  419. Configuration->Usage->Set(L"WorkAreaHeight", Screen->WorkAreaHeight);
  420. HDC DC = GetDC(NULL);
  421. int Planes = GetDeviceCaps(DC, PLANES);
  422. int BitsPixel = GetDeviceCaps(DC, BITSPIXEL);
  423. Configuration->Usage->Set(L"ColorDepth", Planes * BitsPixel);
  424. Configuration->Usage->Set(L"MonitorCount", Screen->MonitorCount);
  425. Configuration->Usage->Set(L"NotUseThemes", !UseThemes());
  426. Configuration->Usage->Set(L"ThemeDefaultFontSize", Application->DefaultFont->Size);
  427. Configuration->Usage->Set(L"ThemeIconFontSize", Screen->IconFont->Size);
  428. Configuration->Usage->Set(L"SysColorWindow", ColorToRGBStr(clWindow));
  429. Configuration->Usage->Set(L"SysColorBtnFace", ColorToRGBStr(clBtnFace));
  430. Configuration->Usage->Set(L"SysColorWindowText", ColorToRGBStr(clWindowText));
  431. UnicodeString ProgramsFolder;
  432. ::SpecialFolderLocation(CSIDL_PROGRAM_FILES, ProgramsFolder);
  433. ProgramsFolder = IncludeTrailingBackslash(ExpandFileName(ProgramsFolder));
  434. UnicodeString ExeName = ExpandFileName(Application->ExeName);
  435. bool InProgramFiles = AnsiSameText(ExeName.SubString(1, ProgramsFolder.Length()), ProgramsFolder);
  436. Configuration->Usage->Set(L"InProgramFiles", InProgramFiles);
  437. Configuration->Usage->Set(L"IsInstalled", IsInstalled());
  438. Configuration->Usage->Set(L"Wine", IsWine());
  439. Configuration->Usage->Set(L"NetFrameworkVersion", GetNetVersionStr());
  440. Configuration->Usage->Set(L"PowerShellVersion", GetPowerShellVersionStr());
  441. WinConfiguration->UpdateStaticUsage();
  442. }
  443. //---------------------------------------------------------------------------
  444. void __fastcall MaintenanceTask()
  445. {
  446. CoreMaintenanceTask();
  447. }
  448. //---------------------------------------------------------------------------
  449. typedef std::vector<HWND> THandles;
  450. typedef std::map<unsigned long, THandles> TProcesses;
  451. //---------------------------------------------------------------------------
  452. BOOL __stdcall EnumOtherInstances(HWND Handle, LPARAM AParam)
  453. {
  454. TProcesses & Processes = *reinterpret_cast<TProcesses *>(AParam);
  455. unsigned long ProcessId;
  456. if (GetWindowThreadProcessId(Handle, &ProcessId) != 0)
  457. {
  458. Processes[ProcessId].push_back(Handle);
  459. }
  460. return TRUE;
  461. }
  462. //---------------------------------------------------------------------------
  463. static bool __fastcall SendCopyDataMessage(HWND Window, TCopyDataMessage & Message)
  464. {
  465. COPYDATASTRUCT CopyData;
  466. CopyData.cbData = sizeof(Message);
  467. CopyData.lpData = &Message;
  468. LRESULT SendResult =
  469. SendMessage(Window, WM_COPYDATA,
  470. reinterpret_cast<WPARAM>(HInstance), reinterpret_cast<LPARAM>(&CopyData));
  471. bool Result = (SendResult > 0);
  472. return Result;
  473. }
  474. //---------------------------------------------------------------------------
  475. static void __fastcall FindOtherInstances(THandles & OtherInstances)
  476. {
  477. TProcesses Processes;
  478. // FindWindow is optimization (if there's no hidden window, no point enumerating all windows to find some)
  479. if ((FindWindow(HIDDEN_WINDOW_NAME, NULL) != NULL) &&
  480. EnumWindows(EnumOtherInstances, reinterpret_cast<LPARAM>(&Processes)))
  481. {
  482. TCopyDataMessage Message;
  483. Message.Command = TCopyDataMessage::MainWindowCheck;
  484. TProcesses::const_iterator ProcessI = Processes.begin();
  485. while (ProcessI != Processes.end())
  486. {
  487. HWND HiddenWindow = NULL;
  488. THandles::const_iterator WindowI = ProcessI->second.begin();
  489. while ((HiddenWindow == NULL) && (WindowI != ProcessI->second.end()))
  490. {
  491. wchar_t ClassName[1024];
  492. if (GetClassName(*WindowI, ClassName, LENOF(ClassName)) != 0)
  493. {
  494. NULL_TERMINATE(ClassName);
  495. if (wcscmp(ClassName, HIDDEN_WINDOW_NAME) == 0)
  496. {
  497. HiddenWindow = *WindowI;
  498. }
  499. }
  500. WindowI++;
  501. }
  502. if (HiddenWindow != NULL)
  503. {
  504. WindowI = ProcessI->second.begin();
  505. while (WindowI != ProcessI->second.end())
  506. {
  507. if (*WindowI != HiddenWindow) // optimization
  508. {
  509. if (SendCopyDataMessage(*WindowI, Message))
  510. {
  511. OtherInstances.push_back(*WindowI);
  512. break;
  513. }
  514. }
  515. WindowI++;
  516. }
  517. }
  518. ProcessI++;
  519. }
  520. }
  521. }
  522. //---------------------------------------------------------------------------
  523. bool __fastcall SendToAnotherInstance()
  524. {
  525. THandles OtherInstances;
  526. FindOtherInstances(OtherInstances);
  527. bool Result = false;
  528. THandles::const_iterator I = OtherInstances.begin();
  529. while (!Result && (I != OtherInstances.end()))
  530. {
  531. HWND Handle = *I;
  532. TCopyDataMessage Message;
  533. Message.Command = TCopyDataMessage::CommandCanCommandLine;
  534. if (SendCopyDataMessage(Handle, Message))
  535. {
  536. // Restore window, if minimized
  537. ShowWindow(Handle, SW_RESTORE);
  538. // bring it to foreground
  539. SetForegroundWindow(Handle);
  540. Message.Command = TCopyDataMessage::CommandCommandLine;
  541. wcsncpy(Message.CommandLine, CmdLine, LENOF(Message.CommandLine));
  542. NULL_TERMINATE(Message.CommandLine);
  543. Result = SendCopyDataMessage(Handle, Message);
  544. }
  545. I++;
  546. }
  547. return Result;
  548. }
  549. //---------------------------------------------------------------------------
  550. void __fastcall Refresh(const UnicodeString & Session, const UnicodeString & Path)
  551. {
  552. THandles OtherInstances;
  553. FindOtherInstances(OtherInstances);
  554. THandles::const_iterator I = OtherInstances.begin();
  555. while (I != OtherInstances.end())
  556. {
  557. HWND Handle = *I;
  558. TCopyDataMessage Message;
  559. Message.Command = TCopyDataMessage::RefreshPanel;
  560. wcsncpy(Message.Refresh.Session, Session.c_str(), LENOF(Message.Refresh.Session));
  561. NULL_TERMINATE(Message.Refresh.Session);
  562. wcsncpy(Message.Refresh.Path, Path.c_str(), LENOF(Message.Refresh.Path));
  563. NULL_TERMINATE(Message.Refresh.Path);
  564. SendCopyDataMessage(Handle, Message);
  565. I++;
  566. }
  567. }
  568. //---------------------------------------------------------------------------
  569. bool __fastcall ShowUpdatesIfAvailable()
  570. {
  571. TUpdatesConfiguration Updates = WinConfiguration->Updates;
  572. int CurrentCompoundVer = Configuration->CompoundVersion;
  573. bool NoPopup = true;
  574. bool Result =
  575. !IsUWP() &&
  576. Updates.ShowOnStartup &&
  577. Updates.HaveValidResultsForVersion(CurrentCompoundVer) &&
  578. !Updates.Results.Disabled &&
  579. ((Updates.Results.Version > CurrentCompoundVer) || !Updates.Results.Message.IsEmpty()) &&
  580. !Updates.ShownResults;
  581. if (Result)
  582. {
  583. Configuration->Usage->Inc(L"UpdateStartup");
  584. Result = CheckForUpdates(true);
  585. if (Result)
  586. {
  587. Configuration->Usage->Inc(L"UpdateDownloadOpensStartup");
  588. }
  589. NoPopup = false;
  590. }
  591. else if (WinConfiguration->ShowTips)
  592. {
  593. int Days = DaysBetween(WinConfiguration->TipsShown, Now());
  594. if ((Days >= Updates.Results.TipsIntervalDays) &&
  595. (WinConfiguration->RunsSinceLastTip >= Updates.Results.TipsIntervalDays))
  596. {
  597. UnicodeString Tip = FirstUnshownTip();
  598. if (!Tip.IsEmpty())
  599. {
  600. AutoShowNewTip();
  601. NoPopup = false;
  602. }
  603. else
  604. {
  605. Configuration->Usage->Inc(L"TipsNoUnseen");
  606. }
  607. }
  608. }
  609. if (NoPopup)
  610. {
  611. WinConfiguration->RunsSinceLastTip = WinConfiguration->RunsSinceLastTip + 1;
  612. }
  613. return Result;
  614. }
  615. //---------------------------------------------------------------------------
  616. int __fastcall Execute()
  617. {
  618. DebugAssert(StoredSessions);
  619. TProgramParams * Params = TProgramParams::Instance();
  620. DebugAssert(Params);
  621. // do not flash message boxes on startup
  622. SetOnForeground(true);
  623. // let installer know, that some instance of application is running
  624. CreateMutex(NULL, False, AppName.c_str());
  625. bool OnlyInstance = (GetLastError() == 0);
  626. UpdateStaticUsage();
  627. UnicodeString KeyFile;
  628. if (Params->FindSwitch(L"PrivateKey", KeyFile))
  629. {
  630. WinConfiguration->DefaultKeyFile = KeyFile;
  631. }
  632. UnicodeString ConsoleVersion;
  633. UnicodeString DotNetVersion;
  634. Params->FindSwitch(L"Console", ConsoleVersion);
  635. Params->FindSwitch(L"DotNet", DotNetVersion);
  636. if (!ConsoleVersion.IsEmpty() || !DotNetVersion.IsEmpty())
  637. {
  638. RecordWrapperVersions(ConsoleVersion, DotNetVersion);
  639. }
  640. if (!DotNetVersion.IsEmpty())
  641. {
  642. Configuration->Usage->Inc(L"ConsoleDotNet");
  643. }
  644. UnicodeString SwitchValue;
  645. if (Params->FindSwitch(L"loglevel", SwitchValue))
  646. {
  647. int StarPos = SwitchValue.Pos(L"*");
  648. if (StarPos > 0)
  649. {
  650. bool LogSensitive = true;
  651. SwitchValue.Delete(StarPos, 1);
  652. if ((StarPos <= SwitchValue.Length()) &&
  653. (SwitchValue[StarPos] == L'-'))
  654. {
  655. LogSensitive = false;
  656. SwitchValue.Delete(StarPos, 1);
  657. }
  658. SwitchValue = SwitchValue.Trim();
  659. Configuration->TemporaryLogSensitive(LogSensitive);
  660. }
  661. int LogProtocol;
  662. if (!SwitchValue.IsEmpty() && TryStrToInt(SwitchValue, LogProtocol) && (LogProtocol >= -1))
  663. {
  664. Configuration->TemporaryLogProtocol(LogProtocol);
  665. }
  666. }
  667. if (Params->FindSwitch(LOGSIZE_SWITCH, SwitchValue))
  668. {
  669. int StarPos = SwitchValue.Pos(LOGSIZE_SEPARATOR);
  670. int LogMaxCount = 0;
  671. if (StarPos > 1)
  672. {
  673. if (!TryStrToInt(SwitchValue.SubString(1, StarPos - 1), LogMaxCount))
  674. {
  675. LogMaxCount = -1;
  676. }
  677. SwitchValue.Delete(1, StarPos);
  678. SwitchValue = SwitchValue.Trim();
  679. }
  680. __int64 LogMaxSize;
  681. if ((LogMaxCount >= 0) &&
  682. !SwitchValue.IsEmpty() &&
  683. TryStrToSize(SwitchValue, LogMaxSize))
  684. {
  685. Configuration->TemporaryLogMaxCount(LogMaxCount);
  686. Configuration->TemporaryLogMaxSize(LogMaxSize);
  687. }
  688. }
  689. TConsoleMode Mode = cmNone;
  690. if (Params->FindSwitch(L"help") || Params->FindSwitch(L"h") || Params->FindSwitch(L"?"))
  691. {
  692. Mode = cmHelp;
  693. }
  694. else if (Params->FindSwitch(L"batchsettings"))
  695. {
  696. Mode = cmBatchSettings;
  697. }
  698. else if (Params->FindSwitch(KEYGEN_SWITCH))
  699. {
  700. Mode = cmKeyGen;
  701. }
  702. else if (Params->FindSwitch(FINGERPRINTSCAN_SWITCH))
  703. {
  704. Mode = cmFingerprintScan;
  705. }
  706. // We have to check for /console only after the other options,
  707. // as the /console is always used when we are run by winscp.com
  708. // (ambiguous use to pass console version)
  709. else if (Params->FindSwitch(L"Console") || Params->FindSwitch(SCRIPT_SWITCH) ||
  710. Params->FindSwitch(COMMAND_SWITCH))
  711. {
  712. Mode = cmScripting;
  713. }
  714. if (Mode != cmNone)
  715. {
  716. return Console(Mode);
  717. }
  718. TTerminalManager * TerminalManager = NULL;
  719. GlyphsModule = NULL;
  720. NonVisualDataModule = NULL;
  721. TStrings * CommandParams = new TStringList;
  722. try
  723. {
  724. TerminalManager = TTerminalManager::Instance();
  725. HANDLE ResourceModule = GUIConfiguration->ChangeToDefaultResourceModule();
  726. try
  727. {
  728. GlyphsModule = new TGlyphsModule(Application);
  729. }
  730. __finally
  731. {
  732. GUIConfiguration->ChangeResourceModule(ResourceModule);
  733. }
  734. NonVisualDataModule = new TNonVisualDataModule(Application);
  735. // The default is 2.5s.
  736. // 20s is used by Office 2010 and Windows 10 Explorer.
  737. // Some applications use an infinite (Thunderbird, Firefox).
  738. // Overriden for some controls using THintInfo.HideTimeout
  739. Application->HintHidePause = 20000;
  740. HintWindowClass = __classid(TScreenTipHintWindow);
  741. UnicodeString IniFileName = Params->SwitchValue(INI_SWITCH);
  742. if (!IniFileName.IsEmpty())
  743. {
  744. UnicodeString IniFileNameExpanded = ExpandEnvironmentVariables(IniFileName);
  745. if (!FileExists(ApiPath(IniFileNameExpanded)))
  746. {
  747. // this should be displayed rather at the very beginning.
  748. // however for simplicity (GUI-only), we do it only here.
  749. MessageDialog(FMTLOAD(FILE_NOT_EXISTS, (IniFileNameExpanded)), qtError, qaOK);
  750. }
  751. }
  752. if (Params->FindSwitch(L"UninstallCleanup"))
  753. {
  754. MaintenanceTask();
  755. // The innosetup cannot skip UninstallCleanup run task for silent uninstalls,
  756. // workaround is that we create mutex in uninstaller, if it runs silent, and
  757. // ignore the UninstallCleanup, when the mutex exists.
  758. if ((OpenMutex(SYNCHRONIZE, false, L"WinSCPSilentUninstall") == NULL) &&
  759. (MessageDialog(MainInstructions(LoadStr(UNINSTALL_CLEANUP)), qtConfirmation,
  760. qaYes | qaNo, HELP_UNINSTALL_CLEANUP) == qaYes))
  761. {
  762. DoCleanupDialog(StoredSessions, Configuration);
  763. }
  764. }
  765. else if (Params->FindSwitch(L"RegisterForDefaultProtocols") ||
  766. Params->FindSwitch(L"RegisterAsUrlHandler")) // BACKWARD COMPATIBILITY
  767. {
  768. MaintenanceTask();
  769. if (CheckSafe(Params))
  770. {
  771. RegisterForDefaultProtocols();
  772. Configuration->DontSave();
  773. }
  774. }
  775. else if (Params->FindSwitch(L"UnregisterForProtocols"))
  776. {
  777. MaintenanceTask();
  778. if (CheckSafe(Params))
  779. {
  780. UnregisterForProtocols();
  781. Configuration->DontSave();
  782. }
  783. }
  784. else if (Params->FindSwitch(L"AddSearchPath"))
  785. {
  786. MaintenanceTask();
  787. if (CheckSafe(Params))
  788. {
  789. AddSearchPath(ExtractFilePath(Application->ExeName));
  790. Configuration->DontSave();
  791. }
  792. }
  793. else if (Params->FindSwitch(L"RemoveSearchPath"))
  794. {
  795. MaintenanceTask();
  796. if (CheckSafe(Params))
  797. {
  798. try
  799. {
  800. RemoveSearchPath(ExtractFilePath(Application->ExeName));
  801. }
  802. catch(...)
  803. {
  804. // ignore errors
  805. // (RemoveSearchPath is called always on uninstallation,
  806. // even if AddSearchPath was not used, so we would get the error
  807. // always for non-priviledged user)
  808. }
  809. Configuration->DontSave();
  810. }
  811. }
  812. else if (Params->FindSwitch(L"ImportSitesIfAny"))
  813. {
  814. MaintenanceTask();
  815. ImportSitesIfAny();
  816. }
  817. else if (Params->FindSwitch(L"Usage", SwitchValue))
  818. {
  819. MaintenanceTask();
  820. Usage(SwitchValue);
  821. }
  822. else if (Params->FindSwitch(L"Update"))
  823. {
  824. MaintenanceTask();
  825. CheckForUpdates(false);
  826. }
  827. else if (ShowUpdatesIfAvailable())
  828. {
  829. // noop
  830. }
  831. else if (Params->FindSwitch(L"Exit"))
  832. {
  833. // noop
  834. MaintenanceTask();
  835. Configuration->DontSave();
  836. }
  837. else if (Params->FindSwitch(L"MaintenanceTask"))
  838. {
  839. // Parameter /MaintenanceTask can be added to command-line when executing maintenance tasks
  840. // (e.g. from installer) just in case old version of WinSCP is called by mistake
  841. MaintenanceTask();
  842. Configuration->DontSave();
  843. }
  844. else
  845. {
  846. enum { pcNone, pcUpload, pcFullSynchronize, pcSynchronize, pcEdit, pcRefresh } ParamCommand;
  847. ParamCommand = pcNone;
  848. UnicodeString AutoStartSession;
  849. UnicodeString DownloadFile;
  850. bool UseDefaults = false;
  851. // do not check for temp dirs for service tasks (like RegisterAsUrlHandler)
  852. if (OnlyInstance &&
  853. WinConfiguration->TemporaryDirectoryCleanup)
  854. {
  855. TemporaryDirectoryCleanup();
  856. }
  857. WinConfiguration->CheckDefaultTranslation();
  858. if (!Params->Empty)
  859. {
  860. if (Params->FindSwitch(DEFAULTS_SWITCH) && CheckSafe(Params))
  861. {
  862. UseDefaults = true;
  863. }
  864. if (Params->FindSwitch(UPLOAD_SWITCH, CommandParams))
  865. {
  866. ParamCommand = pcUpload;
  867. if (CommandParams->Count == 0)
  868. {
  869. throw Exception(NO_UPLOAD_LIST_ERROR);
  870. }
  871. }
  872. if (Params->FindSwitch(UPLOAD_IF_ANY_SWITCH, CommandParams))
  873. {
  874. if (CommandParams->Count > 0)
  875. {
  876. ParamCommand = pcUpload;
  877. }
  878. }
  879. else if (Params->FindSwitch(L"Synchronize", CommandParams, 2))
  880. {
  881. ParamCommand = pcFullSynchronize;
  882. }
  883. else if (Params->FindSwitch(KEEP_UP_TO_DATE_SWITCH, CommandParams, 4))
  884. {
  885. ParamCommand = pcSynchronize;
  886. }
  887. else if (Params->FindSwitch(L"Edit", CommandParams, 1) &&
  888. (CommandParams->Count == 1))
  889. {
  890. ParamCommand = pcEdit;
  891. }
  892. else if (Params->FindSwitch(REFRESH_SWITCH, CommandParams, 1))
  893. {
  894. ParamCommand = pcRefresh;
  895. }
  896. }
  897. if (Params->ParamCount > 0)
  898. {
  899. AutoStartSession = Params->Param[1];
  900. Params->ParamsProcessed(1, 1);
  901. if ((ParamCommand == pcNone) &&
  902. (WinConfiguration->ExternalSessionInExistingInstance != OpenInNewWindow()) &&
  903. !Params->FindSwitch(NEWINSTANCE_SWICH) &&
  904. SendToAnotherInstance())
  905. {
  906. Configuration->Usage->Inc(L"SendToAnotherInstance");
  907. return 0;
  908. }
  909. UnicodeString CounterName;
  910. if (Params->FindSwitch(JUMPLIST_SWITCH))
  911. {
  912. CounterName = L"CommandLineJumpList";
  913. }
  914. else if (Params->FindSwitch(DESKTOP_SWITCH))
  915. {
  916. CounterName = L"CommandLineDesktop";
  917. }
  918. else if (Params->FindSwitch(SEND_TO_HOOK_SWITCH))
  919. {
  920. CounterName = L"CommandLineSendToHook";
  921. }
  922. else
  923. {
  924. CounterName = L"CommandLineSession2";
  925. }
  926. Configuration->Usage->Inc(CounterName);
  927. }
  928. else if (WinConfiguration->EmbeddedSessions && StoredSessions->Count)
  929. {
  930. AutoStartSession = StoredSessions->Sessions[0]->Name;
  931. }
  932. else
  933. {
  934. AutoStartSession = WinConfiguration->AutoStartSession;
  935. }
  936. if (ParamCommand == pcRefresh)
  937. {
  938. Refresh(AutoStartSession, (CommandParams->Count > 0 ? CommandParams->Strings[0] : UnicodeString()));
  939. return 0;
  940. }
  941. // from now flash message boxes on background
  942. SetOnForeground(false);
  943. bool NeedSession = (ParamCommand != pcNone);
  944. bool Retry;
  945. do
  946. {
  947. Retry = false;
  948. TObjectList * DataList = new TObjectList();
  949. try
  950. {
  951. GetLoginData(AutoStartSession, Params, DataList, DownloadFile, NeedSession, NULL);
  952. // GetLoginData now Aborts when session is needed and none is selected
  953. if (DebugAlwaysTrue(!NeedSession || (DataList->Count > 0)))
  954. {
  955. if (CheckSafe(Params))
  956. {
  957. UnicodeString LogFile;
  958. if (Params->FindSwitch(LOG_SWITCH, LogFile))
  959. {
  960. Configuration->TemporaryLogging(LogFile);
  961. }
  962. if (Params->FindSwitch(L"XmlLog", LogFile))
  963. {
  964. Configuration->TemporaryActionsLogging(LogFile);
  965. }
  966. }
  967. try
  968. {
  969. DebugAssert(!TerminalManager->ActiveTerminal);
  970. bool CanStart;
  971. if (DataList->Count > 0)
  972. {
  973. TTerminal * Terminal = TerminalManager->NewTerminals(DataList);
  974. if (!DownloadFile.IsEmpty())
  975. {
  976. Terminal->AutoReadDirectory = false;
  977. DownloadFile = UnixIncludeTrailingBackslash(Terminal->SessionData->RemoteDirectory) + DownloadFile;
  978. Terminal->SessionData->RemoteDirectory = L"";
  979. TManagedTerminal * ManagedTerminal = DebugNotNull(dynamic_cast<TManagedTerminal *>(Terminal));
  980. ManagedTerminal->StateData->RemoteDirectory = Terminal->SessionData->RemoteDirectory;
  981. }
  982. TerminalManager->ActiveTerminal = Terminal;
  983. CanStart = (TerminalManager->Count > 0);
  984. }
  985. else
  986. {
  987. DebugAssert(!NeedSession);
  988. CanStart = true;
  989. }
  990. if (!CanStart)
  991. {
  992. // do not prompt with login dialog, if connection of
  993. // auto-start session (typically from command line) failed
  994. if (AutoStartSession.IsEmpty())
  995. {
  996. Retry = true;
  997. }
  998. }
  999. else
  1000. {
  1001. // from now on, we do not support runtime interface change
  1002. CustomWinConfiguration->CanApplyInterfaceImmediately = false;
  1003. TCustomScpExplorerForm * ScpExplorer = CreateScpExplorer();
  1004. CustomWinConfiguration->AppliedInterface = CustomWinConfiguration->Interface;
  1005. try
  1006. {
  1007. // moved inside try .. __finally, because it can fail as well
  1008. TerminalManager->ScpExplorer = ScpExplorer;
  1009. if ((ParamCommand != pcNone) || !DownloadFile.IsEmpty())
  1010. {
  1011. Configuration->Usage->Inc(L"CommandLineOperation");
  1012. }
  1013. if (ParamCommand == pcUpload)
  1014. {
  1015. Upload(TerminalManager->ActiveTerminal, CommandParams, UseDefaults);
  1016. }
  1017. else if (ParamCommand == pcFullSynchronize)
  1018. {
  1019. FullSynchronize(TerminalManager->ActiveTerminal, ScpExplorer,
  1020. CommandParams, UseDefaults);
  1021. }
  1022. else if (ParamCommand == pcSynchronize)
  1023. {
  1024. Synchronize(TerminalManager->ActiveTerminal, ScpExplorer,
  1025. CommandParams, UseDefaults);
  1026. }
  1027. else if (ParamCommand == pcEdit)
  1028. {
  1029. Edit(ScpExplorer, CommandParams);
  1030. }
  1031. else if (!DownloadFile.IsEmpty())
  1032. {
  1033. Download(TerminalManager->ActiveTerminal, DownloadFile,
  1034. UseDefaults);
  1035. }
  1036. Application->Run();
  1037. }
  1038. __finally
  1039. {
  1040. TerminalManager->ScpExplorer = NULL;
  1041. SAFE_DESTROY(ScpExplorer);
  1042. }
  1043. }
  1044. }
  1045. catch (Exception &E)
  1046. {
  1047. ShowExtendedException(&E);
  1048. }
  1049. }
  1050. }
  1051. __finally
  1052. {
  1053. delete DataList;
  1054. }
  1055. }
  1056. while (Retry);
  1057. }
  1058. }
  1059. __finally
  1060. {
  1061. delete NonVisualDataModule;
  1062. NonVisualDataModule = NULL;
  1063. ReleaseImagesModules();
  1064. delete GlyphsModule;
  1065. GlyphsModule = NULL;
  1066. TTerminalManager::DestroyInstance();
  1067. delete CommandParams;
  1068. }
  1069. return 0;
  1070. }