Custom.cpp 60 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806
  1. //---------------------------------------------------------------------
  2. #include <vcl.h>
  3. #pragma hdrstop
  4. #include <Dialogs.hpp>
  5. //---------------------------------------------------------------------
  6. #include <Common.h>
  7. #include <CustomWinConfiguration.h>
  8. #include <WinInterface.h>
  9. #include <VCLCommon.h>
  10. #include <TextsWin.h>
  11. #include <HelpWin.h>
  12. #include <CoreMain.h>
  13. #include <PasTools.hpp>
  14. #include <ProgParams.h>
  15. #include <Tools.h>
  16. #include <GUITools.h>
  17. #include <PuttyTools.h>
  18. #include <HistoryComboBox.hpp>
  19. #include <Math.hpp>
  20. #include "Custom.h"
  21. //---------------------------------------------------------------------
  22. #pragma link "PasswordEdit"
  23. #pragma resource "*.dfm"
  24. //---------------------------------------------------------------------
  25. __fastcall TCustomDialog::TCustomDialog(UnicodeString AHelpKeyword)
  26. : TForm(GetFormOwner())
  27. {
  28. UseSystemSettings(this);
  29. FControlPadding = ScaleByTextHeight(this, 8);
  30. FPos = ScaleByTextHeight(this, 8);
  31. FPrePos = FPos;
  32. FHorizontalMargin = ScaleByTextHeight(this, 8);
  33. FIndent = FHorizontalMargin;
  34. FGroupBox = NULL;
  35. HelpKeyword = AHelpKeyword;
  36. TBorderIcons BI = BorderIcons;
  37. if (HelpKeyword.IsEmpty())
  38. {
  39. BI >> biHelp;
  40. OKButton->Left = CancelButton->Left;
  41. CancelButton->Left = HelpButton->Left;
  42. HelpButton->Visible = false;
  43. }
  44. else
  45. {
  46. BI << biHelp;
  47. }
  48. BorderIcons = BI;
  49. }
  50. //---------------------------------------------------------------------
  51. bool __fastcall TCustomDialog::Execute()
  52. {
  53. Changed();
  54. return (ShowModal() == DefaultResult(this));
  55. }
  56. //---------------------------------------------------------------------
  57. void __fastcall TCustomDialog::DoChange(bool & /*CanSubmit*/)
  58. {
  59. // noop
  60. }
  61. //---------------------------------------------------------------------
  62. void __fastcall TCustomDialog::Changed()
  63. {
  64. bool CanSubmit = true;
  65. DoChange(CanSubmit);
  66. EnableControl(OKButton, CanSubmit);
  67. }
  68. //---------------------------------------------------------------------
  69. void __fastcall TCustomDialog::Change(TObject * /*Sender*/)
  70. {
  71. Changed();
  72. }
  73. //---------------------------------------------------------------------------
  74. void __fastcall TCustomDialog::DoHelp()
  75. {
  76. FormHelp(this);
  77. }
  78. //---------------------------------------------------------------------------
  79. void __fastcall TCustomDialog::HelpButtonClick(TObject * /*Sender*/)
  80. {
  81. DoHelp();
  82. }
  83. //---------------------------------------------------------------------------
  84. void __fastcall TCustomDialog::DoShow()
  85. {
  86. OKButton->TabOrder = FCount;
  87. CancelButton->TabOrder = static_cast<short>(FCount + 1);
  88. HelpButton->TabOrder = static_cast<short>(FCount + 2);
  89. Changed();
  90. TForm::DoShow();
  91. }
  92. //---------------------------------------------------------------------------
  93. void __fastcall TCustomDialog::DoValidate()
  94. {
  95. // noop
  96. }
  97. //---------------------------------------------------------------------------
  98. bool __fastcall TCustomDialog::CloseQuery()
  99. {
  100. if (ModalResult == DefaultResult(this))
  101. {
  102. DoValidate();
  103. }
  104. return TForm::CloseQuery();
  105. }
  106. //---------------------------------------------------------------------------
  107. void __fastcall TCustomDialog::RemoveCancelButton()
  108. {
  109. CancelButton->Visible = false;
  110. OKButton->Left = CancelButton->Left;
  111. OKButton->Cancel = true;
  112. DebugAssert(OKButton->Width == CancelButton->Width);
  113. DebugAssert(OKButton->Top == CancelButton->Top);
  114. }
  115. //---------------------------------------------------------------------------
  116. void __fastcall TCustomDialog::AddDialogButton(TButton * Button)
  117. {
  118. Button->Parent = this;
  119. Button->Top = OKButton->Top;
  120. Button->Left = FHorizontalMargin;
  121. Button->Height = OKButton->Height;
  122. AddWinControl(Button);
  123. }
  124. //---------------------------------------------------------------------------
  125. void __fastcall TCustomDialog::AddImage(const UnicodeString & ImageName)
  126. {
  127. TImage * Image = new TImage(this);
  128. Image->Name = L"Image";
  129. Image->Parent = GetDefaultParent();
  130. LoadDialogImage(Image, ImageName);
  131. Image->SetBounds(FIndent, FPos + ScaleByTextHeight(this, 3), Image->Picture->Width, Image->Picture->Height);
  132. FIndent += Image->Width + ScaleByTextHeight(this, 12);
  133. }
  134. //---------------------------------------------------------------------------
  135. int __fastcall TCustomDialog::GetMaxControlWidth(TControl * Control)
  136. {
  137. return GetDefaultParent()->ClientWidth - Control->Left - FHorizontalMargin;
  138. }
  139. //---------------------------------------------------------------------------
  140. TWinControl * __fastcall TCustomDialog::GetDefaultParent()
  141. {
  142. return (FGroupBox != NULL) ? FGroupBox : static_cast<TWinControl *>(this);
  143. }
  144. //---------------------------------------------------------------------------
  145. void __fastcall TCustomDialog::AdjustHeight(TControl * Control)
  146. {
  147. FPos = Control->Top + Control->Height + FControlPadding;
  148. int Delta = (FPos - FPrePos);
  149. ClientHeight = ClientHeight + Delta;
  150. if (FGroupBox != NULL)
  151. {
  152. FGroupBox->Height = FGroupBox->Height + Delta;
  153. }
  154. FPrePos = FPos;
  155. }
  156. //---------------------------------------------------------------------------
  157. void __fastcall TCustomDialog::AddWinControl(TWinControl * Control)
  158. {
  159. Control->TabOrder = FCount;
  160. FCount++;
  161. }
  162. //---------------------------------------------------------------------------
  163. TCheckBox * __fastcall TCustomDialog::CreateAndAddCheckBox(const UnicodeString & Caption)
  164. {
  165. TCheckBox * CheckBox = new TCheckBox(this);
  166. CheckBox->Caption = Caption;
  167. AddButtonControl(CheckBox);
  168. return CheckBox;
  169. }
  170. //---------------------------------------------------------------------------
  171. TLabel * __fastcall TCustomDialog::CreateLabel(UnicodeString Label)
  172. {
  173. TLabel * Result = new TUIStateAwareLabel(this);
  174. Result->Caption = Label;
  175. return Result;
  176. }
  177. //---------------------------------------------------------------------------
  178. void __fastcall TCustomDialog::AddEditLikeControl(TWinControl * Edit, TLabel * Label, bool OneLine)
  179. {
  180. Edit->Parent = GetDefaultParent();
  181. // this updates Height property to real value
  182. Edit->HandleNeeded();
  183. if (Label != NULL)
  184. {
  185. Label->Parent = GetDefaultParent();
  186. Label->Left = FIndent;
  187. if (OneLine)
  188. {
  189. DebugAssert(Edit->Height > Label->Height);
  190. Label->Top = FPos + ((Edit->Height - Label->Height) / 2);
  191. }
  192. else
  193. {
  194. Label->Top = FPos;
  195. FPos += Label->Height + ScaleByTextHeight(this, 4);
  196. }
  197. }
  198. Edit->Top = FPos;
  199. if (OneLine)
  200. {
  201. Edit->Left = GetDefaultParent()->ClientWidth - FHorizontalMargin - Edit->Width;
  202. }
  203. else
  204. {
  205. Edit->Left = FIndent;
  206. Edit->Width = GetMaxControlWidth(Edit);
  207. }
  208. AdjustHeight(Edit);
  209. if (Label != NULL)
  210. {
  211. if (Label->FocusControl == NULL)
  212. {
  213. Label->FocusControl = Edit;
  214. }
  215. else
  216. {
  217. DebugAssert(Label->FocusControl == Edit);
  218. }
  219. }
  220. AddWinControl(Edit);
  221. }
  222. //---------------------------------------------------------------------------
  223. void __fastcall TCustomDialog::AddEdit(TCustomEdit * Edit, TLabel * Label, bool OneLine)
  224. {
  225. AddEditLikeControl(Edit, Label, OneLine);
  226. TEdit * PublicEdit = reinterpret_cast<TEdit *>(Edit);
  227. if (PublicEdit->OnChange == NULL)
  228. {
  229. PublicEdit->OnChange = Change;
  230. }
  231. }
  232. //---------------------------------------------------------------------------
  233. void __fastcall TCustomDialog::SetUpComboBox(TCustomCombo * Combo, TStrings * Items, bool OneLine)
  234. {
  235. if (Items != NULL)
  236. {
  237. Combo->Items = Items;
  238. }
  239. if (OneLine)
  240. {
  241. int Width = 0;
  242. for (int Index = 0; Index < Combo->Items->Count; Index++)
  243. {
  244. Width = Max(Width, Combo->Canvas->TextWidth(Combo->Items->Strings[Index]));
  245. }
  246. Width += ScaleByTextHeight(Combo, 4 + 16 + 14);
  247. Width = Max(Width, HelpButton->Width);
  248. Combo->Width = Width;
  249. Combo->Left = GetDefaultParent()->ClientWidth - FHorizontalMargin - Width;
  250. }
  251. TComboBox * PublicCombo = reinterpret_cast<TComboBox *>(Combo);
  252. if (PublicCombo->OnChange == NULL)
  253. {
  254. PublicCombo->OnChange = Change;
  255. }
  256. }
  257. //---------------------------------------------------------------------------
  258. void __fastcall TCustomDialog::AddComboBox(TCustomCombo * Combo, TLabel * Label, TStrings * Items, bool OneLine)
  259. {
  260. AddEditLikeControl(Combo, Label, OneLine);
  261. SetUpComboBox(Combo, Items, OneLine);
  262. }
  263. //---------------------------------------------------------------------------
  264. void __fastcall TCustomDialog::AddShortCutComboBox(TComboBox * Combo, TLabel * Label, const TShortCuts & ShortCuts)
  265. {
  266. AddEditLikeControl(Combo, Label, true);
  267. InitializeShortCutCombo(Combo, ShortCuts);
  268. SetUpComboBox(Combo, NULL, true);
  269. }
  270. //---------------------------------------------------------------------------
  271. void __fastcall TCustomDialog::ScaleButtonControl(TButtonControl * Control)
  272. {
  273. // this updates Height property to real value
  274. Control->HandleNeeded();
  275. // buttons do not scale with text on their own
  276. Control->Height = ScaleByTextHeight(Control, Control->Height);
  277. }
  278. //---------------------------------------------------------------------------
  279. void __fastcall TCustomDialog::AddButtonControl(TButtonControl * Control)
  280. {
  281. Control->Parent = GetDefaultParent();
  282. Control->Left = FIndent + ScaleByTextHeight(this, 6);
  283. Control->Top = FPos;
  284. Control->Width = GetMaxControlWidth(Control);
  285. ScaleButtonControl(Control);
  286. AdjustHeight(Control);
  287. AddWinControl(Control);
  288. TCheckBox * PublicControl = reinterpret_cast<TCheckBox *>(Control);
  289. if (PublicControl->OnClick == NULL)
  290. {
  291. PublicControl->OnClick = Change;
  292. }
  293. }
  294. //---------------------------------------------------------------------------
  295. void TCustomDialog::AddButtonNextToEdit(TButton * Button, TWinControl * Edit)
  296. {
  297. Button->Parent = GetDefaultParent();
  298. Button->Width = HelpButton->Width;
  299. Button->Left = GetDefaultParent()->ClientWidth - Button->Width - HorizontalMargin;
  300. Edit->Width = Button->Left - Edit->Left - ScaleByTextHeight(this, 6);
  301. Button->Top = Edit->Top - ScaleByTextHeight(this, 2);
  302. ScaleButtonControl(Button);
  303. AddWinControl(Button);
  304. }
  305. //---------------------------------------------------------------------------
  306. void __fastcall TCustomDialog::AddText(TLabel * Label)
  307. {
  308. Label->Parent = GetDefaultParent();
  309. DebugAssert(Label->AutoSize);
  310. Label->WordWrap = true;
  311. Label->Left = FIndent;
  312. Label->Width = GetMaxControlWidth(Label);
  313. Label->Top = FPos;
  314. Label->ShowAccelChar = false;
  315. TRect TextRect;
  316. SetRect(&TextRect, 0, 0, Label->Width, 0);
  317. DrawText(Label->Canvas->Handle, Label->Caption.c_str(), Label->Caption.Length() + 1, &TextRect,
  318. DT_EXPANDTABS | DT_CALCRECT | DT_WORDBREAK | DT_NOPREFIX |
  319. Label->DrawTextBiDiModeFlagsReadingOnly());
  320. if (TextRect.Height() > Label->Height)
  321. {
  322. Label->Height = TextRect.Height();
  323. Label->AutoSize = false;
  324. }
  325. else
  326. {
  327. Label->WordWrap = false;
  328. }
  329. AdjustHeight(Label);
  330. }
  331. //---------------------------------------------------------------------------
  332. void __fastcall TCustomDialog::AddText(TStaticText * Label)
  333. {
  334. Label->Parent = GetDefaultParent();
  335. DebugAssert(Label->AutoSize);
  336. Label->Left = FIndent;
  337. Label->Top = FPos;
  338. Label->ShowAccelChar = false;
  339. AdjustHeight(Label);
  340. AddWinControl(Label);
  341. }
  342. //---------------------------------------------------------------------------
  343. void __fastcall TCustomDialog::AddSeparator()
  344. {
  345. TBevel * Bevel = new TBevel(this);
  346. Bevel->Parent = GetDefaultParent();
  347. Bevel->Left = FIndent;
  348. Bevel->Top = FPos;
  349. Bevel->Height = 2;
  350. Bevel->Width = GetMaxControlWidth(Bevel);
  351. AdjustHeight(Bevel);
  352. }
  353. //---------------------------------------------------------------------------
  354. void __fastcall TCustomDialog::StartGroup(const UnicodeString & Caption)
  355. {
  356. if (FGroupBox != NULL)
  357. {
  358. FIndent = FGroupBox->Left;
  359. FPos = FGroupBox->Top + FGroupBox->Height + FControlPadding;
  360. FPrePos = FPos;
  361. FGroupBox = NULL;
  362. }
  363. TGroupBox * GroupBox = new TGroupBox(this);
  364. GroupBox->Parent = this;
  365. GroupBox->Caption = Caption;
  366. GroupBox->Left = FIndent;
  367. GroupBox->Top = FPos;
  368. GroupBox->Height = ScaleByTextHeight(GroupBox, 20);
  369. GroupBox->Width = GetMaxControlWidth(GroupBox);
  370. AdjustHeight(GroupBox);
  371. AddWinControl(GroupBox);
  372. FPos = ScaleByTextHeight(this, 16);
  373. FPrePos = FPos;
  374. FIndent = FHorizontalMargin;
  375. FGroupBox = GroupBox;
  376. }
  377. //---------------------------------------------------------------------------
  378. //---------------------------------------------------------------------------
  379. class TSaveSessionDialog : public TCustomDialog
  380. {
  381. public:
  382. __fastcall TSaveSessionDialog(TComponent* AOwner);
  383. void __fastcall Init(bool CanSavePassword, bool NotRecommendedSavingPassword,
  384. TStrings * AdditionalFolders);
  385. bool __fastcall Execute(UnicodeString & SessionName, bool & SavePassword,
  386. bool & CreateShortcut, const UnicodeString & OriginalSessionName);
  387. protected:
  388. virtual void __fastcall DoValidate();
  389. virtual void __fastcall DoChange(bool & CanSubmit);
  390. private:
  391. UnicodeString FOriginalSessionName;
  392. TEdit * SessionNameEdit;
  393. TComboBox * FolderCombo;
  394. TCheckBox * SavePasswordCheck;
  395. TCheckBox * CreateShortcutCheck;
  396. UnicodeString FRootFolder;
  397. UnicodeString __fastcall GetSessionName();
  398. };
  399. //---------------------------------------------------------------------------
  400. // Need to have an Owner argument for SafeFormCreate
  401. __fastcall TSaveSessionDialog::TSaveSessionDialog(TComponent* /*AOwner*/) :
  402. TCustomDialog(HELP_SESSION_SAVE)
  403. {
  404. }
  405. //---------------------------------------------------------------------------
  406. void __fastcall TSaveSessionDialog::Init(bool CanSavePassword,
  407. bool NotRecommendedSavingPassword, TStrings * AdditionalFolders)
  408. {
  409. Caption = LoadStr(SAVE_SESSION_CAPTION);
  410. SessionNameEdit = new TEdit(this);
  411. AddEdit(SessionNameEdit, CreateLabel(LoadStr(SAVE_SESSION_PROMPT)));
  412. FRootFolder = LoadStr(SAVE_SESSION_ROOT_FOLDER2);
  413. std::unique_ptr<TStringList> Folders(new TStringList());
  414. if (AdditionalFolders != NULL)
  415. {
  416. Folders->AddStrings(AdditionalFolders);
  417. }
  418. for (int Index = 0; Index < StoredSessions->Count; Index++)
  419. {
  420. TSessionData * Data = StoredSessions->Sessions[Index];
  421. if (!Data->Special && !Data->IsWorkspace)
  422. {
  423. UnicodeString Folder = Data->FolderName;
  424. if (!Folder.IsEmpty() && Folders->IndexOf(Folder) < 0)
  425. {
  426. Folders->Add(Folder);
  427. }
  428. }
  429. }
  430. DebugAssert(!Folders->CaseSensitive);
  431. Folders->Sort();
  432. FolderCombo = new TUIStateAwareComboBox(this);
  433. AddComboBox(FolderCombo, CreateLabel(LoadStr(SAVE_SESSION_FOLDER)));
  434. FolderCombo->DropDownCount = Max(FolderCombo->DropDownCount, 16);
  435. FolderCombo->Items->Add(FRootFolder);
  436. FolderCombo->Items->AddStrings(Folders.get());
  437. SavePasswordCheck = CreateAndAddCheckBox(
  438. LoadStr(NotRecommendedSavingPassword ? SAVE_SESSION_PASSWORD :
  439. (CustomWinConfiguration->UseMasterPassword ? SAVE_SESSION_PASSWORD_MASTER : SAVE_SESSION_PASSWORD_RECOMMENDED)));
  440. CreateShortcutCheck = CreateAndAddCheckBox(LoadStr(SAVE_SITE_WORKSPACE_SHORTCUT));
  441. EnableControl(SavePasswordCheck, CanSavePassword);
  442. }
  443. //---------------------------------------------------------------------------
  444. bool __fastcall TSaveSessionDialog::Execute(
  445. UnicodeString & SessionName, bool & SavePassword, bool & CreateShortcut,
  446. const UnicodeString & OriginalSessionName)
  447. {
  448. FOriginalSessionName = OriginalSessionName;
  449. SessionNameEdit->Text = TSessionData::ExtractLocalName(SessionName);
  450. UnicodeString Folder = TSessionData::ExtractFolderName(SessionName);
  451. if (Folder.IsEmpty())
  452. {
  453. FolderCombo->Text = FRootFolder;
  454. }
  455. else
  456. {
  457. FolderCombo->Text = Folder;
  458. }
  459. SavePasswordCheck->Checked = SavePassword;
  460. CreateShortcutCheck->Checked = CreateShortcut;
  461. bool Result = TCustomDialog::Execute();
  462. if (Result)
  463. {
  464. SessionName = GetSessionName();
  465. SavePassword = SavePasswordCheck->Checked;
  466. CreateShortcut = CreateShortcutCheck->Checked;
  467. }
  468. return Result;
  469. }
  470. //---------------------------------------------------------------------------
  471. UnicodeString __fastcall TSaveSessionDialog::GetSessionName()
  472. {
  473. UnicodeString Folder;
  474. if (FolderCombo->Text != FRootFolder)
  475. {
  476. Folder = FolderCombo->Text;
  477. }
  478. return TSessionData::ComposePath(Folder, SessionNameEdit->Text);
  479. }
  480. //---------------------------------------------------------------------------
  481. void __fastcall TSaveSessionDialog::DoValidate()
  482. {
  483. TSessionData::ValidateName(SessionNameEdit->Text);
  484. SessionNameValidate(GetSessionName(), FOriginalSessionName);
  485. UnicodeString Folder = TSessionData::ExtractFolderName(GetSessionName());
  486. if (!Folder.IsEmpty() && StoredSessions->IsWorkspace(Folder))
  487. {
  488. throw Exception(FMTLOAD(WORKSPACE_NOT_FOLDER, (Folder)));
  489. }
  490. if (SavePasswordCheck->Enabled && SavePasswordCheck->Checked &&
  491. CustomWinConfiguration->UseMasterPassword)
  492. {
  493. CustomWinConfiguration->AskForMasterPasswordIfNotSet();
  494. }
  495. TCustomDialog::DoValidate();
  496. }
  497. //---------------------------------------------------------------------------
  498. void __fastcall TSaveSessionDialog::DoChange(bool & CanSubmit)
  499. {
  500. CanSubmit = !SessionNameEdit->Text.IsEmpty();
  501. TCustomDialog::DoChange(CanSubmit);
  502. }
  503. //---------------------------------------------------------------------------
  504. TSessionData * __fastcall DoSaveSession(TSessionData * SessionData,
  505. TSessionData * OriginalSession, bool ForceDialog,
  506. TStrings * AdditionalFolders)
  507. {
  508. bool SavePassword = false;
  509. bool * PSavePassword;
  510. bool NotRecommendedSavingPassword =
  511. !CustomWinConfiguration->UseMasterPassword &&
  512. !SameText(SessionData->UserName, AnonymousUserName);
  513. if (Configuration->DisablePasswordStoring ||
  514. !SessionData->HasAnySessionPassword())
  515. {
  516. PSavePassword = NULL;
  517. }
  518. else
  519. {
  520. PSavePassword = &SavePassword;
  521. SavePassword =
  522. ((OriginalSession != NULL) && OriginalSession->HasAnySessionPassword()) ||
  523. !NotRecommendedSavingPassword;
  524. }
  525. UnicodeString SessionName = SessionData->SessionName;
  526. bool Result;
  527. bool CreateShortcut = false;
  528. if (!ForceDialog && ((PSavePassword == NULL) || SavePassword))
  529. {
  530. // This is probably here to ask before session is started saving.
  531. // Otherwise we would ask implicitly, when saving passwords, but at that moment,
  532. // part of the site is already saved and when the user cancel the prompt it's too late.
  533. CustomWinConfiguration->AskForMasterPasswordIfNotSetAndNeededToPersistSessionData(SessionData);
  534. Result = true;
  535. }
  536. else
  537. {
  538. // This can be a standalone dialog when used with save URL (from GetLoginData)
  539. TSaveSessionDialog * Dialog = SafeFormCreate<TSaveSessionDialog>();
  540. try
  541. {
  542. Dialog->Init((PSavePassword != NULL), NotRecommendedSavingPassword, AdditionalFolders);
  543. Result = Dialog->Execute(SessionName, SavePassword, CreateShortcut, SessionData->Name);
  544. }
  545. __finally
  546. {
  547. delete Dialog;
  548. }
  549. }
  550. TSessionData * NewSession = NULL;
  551. if (Result)
  552. {
  553. if ((PSavePassword != NULL) && !SavePassword)
  554. {
  555. SessionData->ClearSessionPasswords();
  556. }
  557. NewSession =
  558. StoredSessions->NewSession(SessionName, SessionData);
  559. // modified only, explicit
  560. StoredSessions->Save(false, true);
  561. if (!SessionData->HostKey.IsEmpty())
  562. {
  563. SessionData->CacheHostKeyIfNotCached();
  564. }
  565. if (CreateShortcut)
  566. {
  567. TOperationVisualizer Visualizer;
  568. UnicodeString AdditionalParams =
  569. TProgramParams::FormatSwitch(DESKTOP_SWITCH) + L" " +
  570. TProgramParams::FormatSwitch(UPLOAD_IF_ANY_SWITCH);
  571. CreateDesktopSessionShortCut(SessionName, L"", AdditionalParams, -1, SITE_ICON);
  572. }
  573. }
  574. return NewSession;
  575. }
  576. //---------------------------------------------------------------------------
  577. void __fastcall SessionNameValidate(const UnicodeString & Text,
  578. const UnicodeString & OriginalName)
  579. {
  580. TSessionData::ValidatePath(Text);
  581. DebugAssert(StoredSessions);
  582. TSessionData * Data = (TSessionData *)StoredSessions->FindByName(Text);
  583. if (Data && Data->Special)
  584. {
  585. MessageDialog(FMTLOAD(CANNOT_OVERWRITE_SPECIAL_SESSION, (Text)),
  586. qtError, qaOK, HELP_NONE);
  587. Abort();
  588. }
  589. else if ((Data != NULL) && !Data->IsSameName(OriginalName) &&
  590. MessageDialog(MainInstructions(FMTLOAD(CONFIRM_OVERWRITE_SESSION, (Text))),
  591. qtConfirmation, qaYes | qaNo, HELP_SESSION_SAVE_OVERWRITE) != qaYes)
  592. {
  593. Abort();
  594. }
  595. }
  596. //---------------------------------------------------------------------------
  597. //---------------------------------------------------------------------------
  598. class TSaveWorkspaceDialog : public TCustomDialog
  599. {
  600. public:
  601. __fastcall TSaveWorkspaceDialog(bool CanSavePasswords,
  602. bool NotRecommendedSavingPasswords);
  603. bool __fastcall Execute(
  604. UnicodeString & WorkspaceName, bool & SavePasswords, bool & CreateShortcut,
  605. bool & EnableAutoSave);
  606. protected:
  607. virtual void __fastcall DoValidate();
  608. virtual void __fastcall DoChange(bool & CanSubmit);
  609. private:
  610. TComboBox * WorkspaceNameCombo;
  611. TCheckBox * SavePasswordsCheck;
  612. TCheckBox * CreateShortcutCheck;
  613. TCheckBox * EnableAutoSaveCheck;
  614. };
  615. //---------------------------------------------------------------------------
  616. __fastcall TSaveWorkspaceDialog::TSaveWorkspaceDialog(
  617. bool CanSavePasswords, bool NotRecommendedSavingPasswords) :
  618. TCustomDialog(HELP_WORKSPACE_SAVE)
  619. {
  620. Caption = LoadStr(SAVE_WORKSPACE_CAPTION);
  621. WorkspaceNameCombo = new TUIStateAwareComboBox(this);
  622. WorkspaceNameCombo->AutoComplete = false;
  623. AddComboBox(WorkspaceNameCombo, CreateLabel(LoadStr(SAVE_WORKSPACE_PROMPT)));
  624. WorkspaceNameCombo->DropDownCount = Max(WorkspaceNameCombo->DropDownCount, 16);
  625. std::unique_ptr<TStrings> Workspaces(StoredSessions->GetWorkspaces());
  626. WorkspaceNameCombo->Items->AddStrings(Workspaces.get());
  627. SavePasswordsCheck = CreateAndAddCheckBox(
  628. LoadStr(NotRecommendedSavingPasswords ? SAVE_WORKSPACE_PASSWORDS :
  629. (CustomWinConfiguration->UseMasterPassword ?
  630. SAVE_WORKSPACE_PASSWORDS_MASTER : SAVE_WORKSPACE_PASSWORDS_RECOMMENDED)));
  631. EnableControl(SavePasswordsCheck, CanSavePasswords);
  632. CreateShortcutCheck = CreateAndAddCheckBox(LoadStr(SAVE_SITE_WORKSPACE_SHORTCUT));
  633. EnableAutoSaveCheck = CreateAndAddCheckBox(LoadStr(SAVE_WORKSPACE_AUTO));
  634. }
  635. //---------------------------------------------------------------------------
  636. bool __fastcall TSaveWorkspaceDialog::Execute(
  637. UnicodeString & WorkspaceName, bool & SavePasswords, bool & CreateShortcut,
  638. bool & EnableAutoSave)
  639. {
  640. WorkspaceNameCombo->Text = WorkspaceName;
  641. SavePasswordsCheck->Checked = SavePasswords;
  642. CreateShortcutCheck->Checked = CreateShortcut;
  643. EnableAutoSaveCheck->Checked = EnableAutoSave;
  644. bool Result = TCustomDialog::Execute();
  645. if (Result)
  646. {
  647. WorkspaceName = WorkspaceNameCombo->Text;
  648. SavePasswords = SavePasswordsCheck->Checked;
  649. CreateShortcut = CreateShortcutCheck->Checked;
  650. EnableAutoSave = EnableAutoSaveCheck->Checked;
  651. }
  652. return Result;
  653. }
  654. //---------------------------------------------------------------------------
  655. void __fastcall TSaveWorkspaceDialog::DoValidate()
  656. {
  657. TSessionData::ValidateName(WorkspaceNameCombo->Text);
  658. if (StoredSessions->IsFolder(WorkspaceNameCombo->Text))
  659. {
  660. throw Exception(FMTLOAD(FOLDER_NOT_WORKSPACE, (WorkspaceNameCombo->Text)));
  661. }
  662. if (SavePasswordsCheck->Enabled && SavePasswordsCheck->Checked &&
  663. CustomWinConfiguration->UseMasterPassword)
  664. {
  665. CustomWinConfiguration->AskForMasterPasswordIfNotSet();
  666. }
  667. TCustomDialog::DoValidate();
  668. }
  669. //---------------------------------------------------------------------------
  670. void __fastcall TSaveWorkspaceDialog::DoChange(bool & CanSubmit)
  671. {
  672. CanSubmit = !WorkspaceNameCombo->Text.IsEmpty();
  673. TCustomDialog::DoChange(CanSubmit);
  674. }
  675. //---------------------------------------------------------------------------
  676. bool __fastcall DoSaveWorkspaceDialog(UnicodeString & WorkspaceName,
  677. bool * SavePasswords, bool NotRecommendedSavingPasswords,
  678. bool & CreateShortcut, bool & EnableAutoSave)
  679. {
  680. std::unique_ptr<TSaveWorkspaceDialog> Dialog(
  681. new TSaveWorkspaceDialog((SavePasswords != NULL), NotRecommendedSavingPasswords));
  682. bool Dummy = false;
  683. if (SavePasswords == NULL)
  684. {
  685. SavePasswords = &Dummy;
  686. }
  687. return
  688. Dialog->Execute(
  689. WorkspaceName, *SavePasswords, CreateShortcut, EnableAutoSave);
  690. }
  691. //---------------------------------------------------------------------------
  692. //---------------------------------------------------------------------------
  693. class TShortCutDialog : public TCustomDialog
  694. {
  695. public:
  696. __fastcall TShortCutDialog(const TShortCuts & ShortCuts, UnicodeString HelpKeyword);
  697. bool __fastcall Execute(TShortCut & ShortCut);
  698. private:
  699. TComboBox * ShortCutCombo;
  700. };
  701. //---------------------------------------------------------------------------
  702. __fastcall TShortCutDialog::TShortCutDialog(const TShortCuts & ShortCuts, UnicodeString HelpKeyword) :
  703. TCustomDialog(HelpKeyword)
  704. {
  705. Caption = LoadStr(SHORTCUT_CAPTION);
  706. ShortCutCombo = new TUIStateAwareComboBox(this);
  707. AddShortCutComboBox(ShortCutCombo, CreateLabel(LoadStr(SHORTCUT_LABEL)), ShortCuts);
  708. }
  709. //---------------------------------------------------------------------------
  710. bool __fastcall TShortCutDialog::Execute(TShortCut & ShortCut)
  711. {
  712. SetShortCutCombo(ShortCutCombo, ShortCut);
  713. bool Result = TCustomDialog::Execute();
  714. if (Result)
  715. {
  716. ShortCut = GetShortCutCombo(ShortCutCombo);
  717. }
  718. return Result;
  719. }
  720. //---------------------------------------------------------------------------
  721. bool __fastcall DoShortCutDialog(TShortCut & ShortCut,
  722. const TShortCuts & ShortCuts, UnicodeString HelpKeyword)
  723. {
  724. bool Result;
  725. TShortCutDialog * Dialog = new TShortCutDialog(ShortCuts, HelpKeyword);
  726. try
  727. {
  728. Result = Dialog->Execute(ShortCut);
  729. }
  730. __finally
  731. {
  732. delete Dialog;
  733. }
  734. return Result;
  735. }
  736. //---------------------------------------------------------------------------
  737. //---------------------------------------------------------------------------
  738. class TRemoteMoveDialog : public TCustomDialog
  739. {
  740. public:
  741. __fastcall TRemoteMoveDialog(bool Multi, TDirectoryExistsEvent OnDirectoryExists);
  742. bool __fastcall Execute(UnicodeString & Target, UnicodeString & FileMask);
  743. protected:
  744. DYNAMIC void __fastcall DoShow();
  745. virtual void __fastcall DoValidate();
  746. UnicodeString __fastcall GetFileMask();
  747. private:
  748. THistoryComboBox * Combo;
  749. bool FMulti;
  750. TDirectoryExistsEvent FOnDirectoryExists;
  751. };
  752. //---------------------------------------------------------------------------
  753. __fastcall TRemoteMoveDialog::TRemoteMoveDialog(bool Multi, TDirectoryExistsEvent OnDirectoryExists) :
  754. TCustomDialog(HELP_REMOTE_MOVE)
  755. {
  756. Caption = LoadStr(REMOTE_MOVE_TITLE);
  757. // The same as TRemoteTransferDialog
  758. ClientWidth = ScaleByTextHeight(this, 420);
  759. FMulti = Multi;
  760. FOnDirectoryExists = OnDirectoryExists;
  761. AddImage(L"Move L to R");
  762. Combo = new THistoryComboBox(this);
  763. Combo->AutoComplete = false;
  764. AddComboBox(Combo, CreateLabel(LoadStr(REMOTE_TRANSFER_PROMPT2)));
  765. }
  766. //---------------------------------------------------------------------------
  767. bool __fastcall TRemoteMoveDialog::Execute(UnicodeString & Target, UnicodeString & FileMask)
  768. {
  769. Combo->Items = CustomWinConfiguration->History[L"RemoteTarget"];
  770. Combo->Text = UnixIncludeTrailingBackslash(Target) + FileMask;
  771. bool Result = TCustomDialog::Execute();
  772. if (Result)
  773. {
  774. Target = UnixExtractFilePath(Combo->Text);
  775. FileMask = GetFileMask();
  776. Combo->SaveToHistory();
  777. CustomWinConfiguration->History[L"RemoteTarget"] = Combo->Items;
  778. }
  779. return Result;
  780. }
  781. //---------------------------------------------------------------------------
  782. UnicodeString __fastcall TRemoteMoveDialog::GetFileMask()
  783. {
  784. return UnixExtractFileName(Combo->Text);
  785. }
  786. //---------------------------------------------------------------------------
  787. void __fastcall TRemoteMoveDialog::DoShow()
  788. {
  789. TCustomDialog::DoShow();
  790. InstallPathWordBreakProc(Combo);
  791. }
  792. //---------------------------------------------------------------------------
  793. void __fastcall TRemoteMoveDialog::DoValidate()
  794. {
  795. if (FOnDirectoryExists(NULL, Combo->Text))
  796. {
  797. Combo->Text = UnixCombinePaths(Combo->Text, AnyMask);
  798. }
  799. if (!IsFileNameMask(GetFileMask()) && FMulti)
  800. {
  801. UnicodeString Message =
  802. FormatMultiFilesToOneConfirmation(Combo->Text, true);
  803. if (MessageDialog(Message, qtConfirmation, qaOK | qaCancel, HELP_NONE) == qaCancel)
  804. {
  805. Abort();
  806. }
  807. }
  808. TCustomDialog::DoValidate();
  809. }
  810. //---------------------------------------------------------------------------
  811. bool __fastcall DoRemoteMoveDialog(
  812. bool Multi, UnicodeString & Target, UnicodeString & FileMask, TDirectoryExistsEvent OnDirectoryExists)
  813. {
  814. std::unique_ptr<TRemoteMoveDialog> Dialog(new TRemoteMoveDialog(Multi, OnDirectoryExists));
  815. return Dialog->Execute(Target, FileMask);
  816. }
  817. //---------------------------------------------------------------------------
  818. //---------------------------------------------------------------------------
  819. class TCustomCommandOptionsDialog : public TCustomDialog
  820. {
  821. public:
  822. __fastcall TCustomCommandOptionsDialog(
  823. const TCustomCommandType * Command, TStrings * CustomCommandOptions, unsigned int Flags,
  824. TCustomCommand * CustomCommandForOptions, const UnicodeString & Site, const TShortCuts * ShortCuts);
  825. bool __fastcall Execute(TShortCut * ShortCut);
  826. protected:
  827. virtual void __fastcall DoHelp();
  828. DYNAMIC void __fastcall DoShow();
  829. private:
  830. const TCustomCommandType * FCommand;
  831. TStrings * FCustomCommandOptions;
  832. std::vector<TControl *> FControls;
  833. std::vector<std::vector<UnicodeString> > FValues;
  834. unsigned int FFlags;
  835. UnicodeString FSite;
  836. TComboBox * FShortCutCombo;
  837. UnicodeString __fastcall HistoryKey(const TCustomCommandType::TOption & Option);
  838. THistoryComboBox * __fastcall CreateHistoryComboBox(const TCustomCommandType::TOption & Option, const UnicodeString & Value);
  839. void __fastcall BrowseButtonClick(TObject * Sender);
  840. void __fastcall LinkLabelClick(TObject * Sender);
  841. UnicodeString __fastcall SaveHistoryComboBoxValue(TControl * Control, const TCustomCommandType::TOption & Option);
  842. void __fastcall AddOptionComboBox(
  843. TComboBox * ComboBox, const UnicodeString & Value, const TCustomCommandType::TOption & Option,
  844. std::vector<UnicodeString> & Values);
  845. UnicodeString __fastcall GetComboBoxValue(TControl * Control, const UnicodeString & Default);
  846. int __fastcall GetOptionIndex(TControl * Control);
  847. int __fastcall GetControlIndex(TControl * Control);
  848. };
  849. //---------------------------------------------------------------------------
  850. __fastcall TCustomCommandOptionsDialog::TCustomCommandOptionsDialog(
  851. const TCustomCommandType * Command, TStrings * CustomCommandOptions,
  852. unsigned int Flags, TCustomCommand * CustomCommandForOptions,
  853. const UnicodeString & Site, const TShortCuts * ShortCuts) :
  854. TCustomDialog(HELP_EXTENSION_OPTIONS)
  855. {
  856. FCommand = Command;
  857. FFlags = Flags;
  858. FCustomCommandOptions = CustomCommandOptions;
  859. FSite = Site;
  860. Caption = StripEllipsis(StripHotkey(FCommand->Name));
  861. Width = ScaleByTextHeight(this, 400);
  862. bool HasGroups = false;
  863. int ControlIndex = 0;
  864. for (int OptionIndex = 0; OptionIndex < FCommand->OptionsCount; OptionIndex++)
  865. {
  866. const TCustomCommandType::TOption & Option = FCommand->GetOption(OptionIndex);
  867. if ((Option.Flags & FFlags) != 0)
  868. {
  869. UnicodeString OptionKey = FCommand->GetOptionKey(Option, FSite);
  870. UnicodeString Value;
  871. if ((CustomCommandForOptions != NULL) &&
  872. Option.HasPatterns(CustomCommandForOptions))
  873. {
  874. Value = CustomCommandForOptions->Complete(Option.Default, true);
  875. }
  876. else
  877. {
  878. if (FCustomCommandOptions->IndexOfName(OptionKey) >= 0)
  879. {
  880. Value = FCustomCommandOptions->Values[OptionKey];
  881. }
  882. else
  883. {
  884. Value = Option.Default;
  885. }
  886. }
  887. int Tag = (OptionIndex << 16) + ControlIndex;
  888. TControl * Control = NULL;
  889. std::vector<UnicodeString> Values;
  890. if (Option.Kind == TCustomCommandType::okUnknown)
  891. {
  892. Control = NULL;
  893. }
  894. else if (Option.Kind == TCustomCommandType::okLabel)
  895. {
  896. TLabel * Label = CreateLabel(Option.Caption);
  897. AddText(Label);
  898. Control = Label;
  899. }
  900. else if (Option.Kind == TCustomCommandType::okLink)
  901. {
  902. TStaticText * Label = new TStaticText(this);
  903. Label->Caption = Option.Caption;
  904. if (IsHttpOrHttpsUrl(Label->Caption))
  905. {
  906. Label->Caption = SecureUrl(Label->Caption);
  907. LinkLabel(Label);
  908. Label->TabStop = true;
  909. }
  910. else if (!Option.Default.IsEmpty() && IsHttpOrHttpsUrl(Option.Default))
  911. {
  912. Label->OnClick = LinkLabelClick;
  913. LinkLabel(Label);
  914. Label->TabStop = true;
  915. }
  916. else
  917. {
  918. // keep it plain text, as we have no URL
  919. }
  920. AddText(Label);
  921. Control = Label;
  922. }
  923. else if (Option.Kind == TCustomCommandType::okGroup)
  924. {
  925. StartGroup(Option.Caption);
  926. HasGroups = true;
  927. }
  928. else if (Option.Kind == TCustomCommandType::okSeparator)
  929. {
  930. AddSeparator();
  931. }
  932. else if (Option.Kind == TCustomCommandType::okTextBox)
  933. {
  934. Control = CreateHistoryComboBox(Option, Value);
  935. }
  936. else if (Option.Kind == TCustomCommandType::okFile)
  937. {
  938. THistoryComboBox * ComboBox = CreateHistoryComboBox(Option, Value);
  939. TButton * Button = new TButton(this);
  940. AddButtonNextToEdit(Button, ComboBox);
  941. Button->Tag = Tag;
  942. Button->Caption = LoadStr(EXTENSION_OPTIONS_BROWSE);
  943. Button->OnClick = BrowseButtonClick;
  944. Control = ComboBox;
  945. }
  946. else if (Option.Kind == TCustomCommandType::okDropDownList)
  947. {
  948. TComboBox * ComboBox = new TUIStateAwareComboBox(this);
  949. ComboBox->Style = csDropDownList;
  950. AddOptionComboBox(ComboBox, Value, Option, Values);
  951. Control = ComboBox;
  952. }
  953. else if (Option.Kind == TCustomCommandType::okComboBox)
  954. {
  955. TComboBox * ComboBox = new TUIStateAwareComboBox(this);
  956. ComboBox->Style = csDropDown;
  957. AddOptionComboBox(ComboBox, Value, Option, Values);
  958. if (ComboBox->ItemIndex < 0)
  959. {
  960. ComboBox->Text = Value;
  961. }
  962. Control = ComboBox;
  963. }
  964. else if (Option.Kind == TCustomCommandType::okCheckBox)
  965. {
  966. TCheckBox * CheckBox = CreateAndAddCheckBox(Option.Caption);
  967. CheckBox->Checked =
  968. (Option.Params.size() >= 1) &&
  969. (Value == Option.Params[0]);
  970. Control = CheckBox;
  971. }
  972. else
  973. {
  974. DebugFail();
  975. }
  976. if (Control != NULL)
  977. {
  978. Control->Tag = Tag;
  979. }
  980. FControls.push_back(Control);
  981. FValues.push_back(Values);
  982. ControlIndex++;
  983. DebugAssert(static_cast<int>(FControls.size()) == ControlIndex);
  984. }
  985. }
  986. if (ShortCuts != NULL)
  987. {
  988. if (HasGroups)
  989. {
  990. StartGroup(LoadStr(EXTENSION_GENERAL_GROUP));
  991. }
  992. else if (ControlIndex > 0)
  993. {
  994. AddSeparator();
  995. }
  996. FShortCutCombo = new TUIStateAwareComboBox(this);
  997. AddShortCutComboBox(FShortCutCombo, CreateLabel(LoadStr(EXTENSION_SHORTCUT)), *ShortCuts);
  998. }
  999. }
  1000. //---------------------------------------------------------------------------
  1001. void __fastcall TCustomCommandOptionsDialog::AddOptionComboBox(
  1002. TComboBox * ComboBox, const UnicodeString & Value, const TCustomCommandType::TOption & Option, std::vector<UnicodeString> & Values)
  1003. {
  1004. std::unique_ptr<TStringList> Items(new TStringList());
  1005. int ItemIndex = -1;
  1006. TCustomCommandType::TOption::TParams::const_iterator ParamI = Option.Params.begin();
  1007. while (ParamI != Option.Params.end())
  1008. {
  1009. UnicodeString Item = (*ParamI);
  1010. int P = Item.Pos(L"=");
  1011. UnicodeString ParamValue;
  1012. if (P > 0)
  1013. {
  1014. ParamValue = Item.SubString(1, P - 1);
  1015. Item.Delete(1, P);
  1016. }
  1017. else
  1018. {
  1019. ParamValue = Item;
  1020. }
  1021. Item = WinConfiguration->ExtensionStringTranslation(FCommand->Id, Item);
  1022. Items->Add(Item);
  1023. if (Value == ParamValue)
  1024. {
  1025. ItemIndex = Items->Count - 1;
  1026. }
  1027. Values.push_back(ParamValue);
  1028. ParamI++;
  1029. }
  1030. AddComboBox(ComboBox, CreateLabel(Option.Caption), Items.get(), true);
  1031. ComboBox->ItemIndex = ItemIndex;
  1032. }
  1033. //---------------------------------------------------------------------------
  1034. int __fastcall TCustomCommandOptionsDialog::GetOptionIndex(TControl * Control)
  1035. {
  1036. return (Control->Tag >> 16);
  1037. }
  1038. //---------------------------------------------------------------------------
  1039. int __fastcall TCustomCommandOptionsDialog::GetControlIndex(TControl * Control)
  1040. {
  1041. return (Control->Tag & 0xFFFF);
  1042. }
  1043. //---------------------------------------------------------------------------
  1044. void __fastcall TCustomCommandOptionsDialog::LinkLabelClick(TObject * Sender)
  1045. {
  1046. TStaticText * Label = DebugNotNull(dynamic_cast<TStaticText *>(Sender));
  1047. const TCustomCommandType::TOption & Option = FCommand->GetOption(GetOptionIndex(Label));
  1048. OpenBrowser(SecureUrl(Option.Default));
  1049. }
  1050. //---------------------------------------------------------------------------
  1051. void __fastcall TCustomCommandOptionsDialog::BrowseButtonClick(TObject * Sender)
  1052. {
  1053. TButton * Button = DebugNotNull(dynamic_cast<TButton *>(Sender));
  1054. int OptionIndex = GetOptionIndex(Button);
  1055. const TCustomCommandType::TOption & Option = FCommand->GetOption(OptionIndex);
  1056. int ControlIndex = GetControlIndex(Button);
  1057. THistoryComboBox * ComboBox = dynamic_cast<THistoryComboBox *>(FControls[ControlIndex]);
  1058. std::unique_ptr<TOpenDialog> OpenDialog(new TOpenDialog(Application));
  1059. UnicodeString Title;
  1060. if (!Option.FileCaption.IsEmpty())
  1061. {
  1062. Title = Option.FileCaption;
  1063. }
  1064. else
  1065. {
  1066. UnicodeString Caption = Option.Caption;
  1067. Caption = StripHotkey(Caption);
  1068. if (!Caption.IsEmpty() && (Caption[Caption.Length()] == L':'))
  1069. {
  1070. Caption.SetLength(Caption.Length() - 1);
  1071. }
  1072. Title = FMTLOAD(EXTENSION_OPTIONS_BROWSE_TITLE, (Caption));
  1073. }
  1074. OpenDialog->Title = Title;
  1075. UnicodeString Value;
  1076. if (ComboBox->Text.IsEmpty())
  1077. {
  1078. Value = Option.FileInitial;
  1079. }
  1080. else
  1081. {
  1082. Value = ComboBox->Text;
  1083. }
  1084. UnicodeString ExpandedValue = ExpandEnvironmentVariables(Value);
  1085. OpenDialog->FileName = ExpandedValue;
  1086. UnicodeString InitialDir = ExtractFilePath(ExpandedValue);
  1087. if (!InitialDir.IsEmpty())
  1088. {
  1089. OpenDialog->InitialDir = InitialDir;
  1090. }
  1091. OpenDialog->Filter = Option.FileFilter;
  1092. OpenDialog->DefaultExt = Option.FileExt;
  1093. if (OpenDialog->Execute())
  1094. {
  1095. if (OpenDialog->FileName != ExpandedValue)
  1096. {
  1097. ComboBox->Text = OpenDialog->FileName;
  1098. }
  1099. // If user just confirms the initial value, persist it
  1100. else if (ComboBox->Text.IsEmpty())
  1101. {
  1102. DebugAssert(Option.FileInitial == Value);
  1103. ComboBox->Text = Value;
  1104. }
  1105. }
  1106. }
  1107. //---------------------------------------------------------------------------
  1108. THistoryComboBox * __fastcall TCustomCommandOptionsDialog::CreateHistoryComboBox(
  1109. const TCustomCommandType::TOption & Option, const UnicodeString & Value)
  1110. {
  1111. THistoryComboBox * ComboBox = new THistoryComboBox(this);
  1112. ComboBox->AutoComplete = false;
  1113. AddComboBox(ComboBox, CreateLabel(Option.Caption));
  1114. ComboBox->Items = CustomWinConfiguration->History[HistoryKey(Option)];
  1115. ComboBox->Text = Value;
  1116. return ComboBox;
  1117. }
  1118. //---------------------------------------------------------------------------
  1119. UnicodeString __fastcall TCustomCommandOptionsDialog::HistoryKey(const TCustomCommandType::TOption & Option)
  1120. {
  1121. UnicodeString Result = FCommand->GetOptionKey(Option, FSite);
  1122. Result = CustomWinConfiguration->GetValidHistoryKey(Result);
  1123. return L"CustomCommandOption_" + Result;
  1124. }
  1125. //---------------------------------------------------------------------------
  1126. bool __fastcall TCustomCommandOptionsDialog::Execute(TShortCut * ShortCut)
  1127. {
  1128. if (ShortCut != NULL)
  1129. {
  1130. SetShortCutCombo(FShortCutCombo, *ShortCut);
  1131. }
  1132. bool Result = TCustomDialog::Execute();
  1133. if (Result)
  1134. {
  1135. int ControlIndex = 0;
  1136. for (int OptionIndex = 0; OptionIndex < FCommand->OptionsCount; OptionIndex++)
  1137. {
  1138. const TCustomCommandType::TOption & Option = FCommand->GetOption(OptionIndex);
  1139. if ((Option.Flags & FFlags) != 0)
  1140. {
  1141. if ((Option.Kind != TCustomCommandType::okUnknown) &&
  1142. Option.IsControl)
  1143. {
  1144. UnicodeString OptionKey = FCommand->GetOptionKey(Option, FSite);
  1145. TControl * Control = FControls[ControlIndex];
  1146. UnicodeString Value;
  1147. if (Option.Kind == TCustomCommandType::okTextBox)
  1148. {
  1149. Value = SaveHistoryComboBoxValue(Control, Option);
  1150. }
  1151. else if (Option.Kind == TCustomCommandType::okFile)
  1152. {
  1153. Value = SaveHistoryComboBoxValue(Control, Option);
  1154. }
  1155. else if (Option.Kind == TCustomCommandType::okDropDownList)
  1156. {
  1157. Value = GetComboBoxValue(Control, Option.Default);
  1158. }
  1159. else if (Option.Kind == TCustomCommandType::okComboBox)
  1160. {
  1161. TComboBox * ComboBox = DebugNotNull(dynamic_cast<TComboBox *>(Control));
  1162. Value = GetComboBoxValue(Control, ComboBox->Text);
  1163. }
  1164. else if (Option.Kind == TCustomCommandType::okCheckBox)
  1165. {
  1166. TCheckBox * CheckBox = DebugNotNull(dynamic_cast<TCheckBox *>(Control));
  1167. int Index = (CheckBox->Checked ? 0 : 1);
  1168. Value = (Index < static_cast<int>(Option.Params.size())) ? Option.Params[Index] : UnicodeString();
  1169. }
  1170. else
  1171. {
  1172. DebugFail();
  1173. }
  1174. // The default value setter deletes the "name" when the value is empty.
  1175. // It would cause us to fall back to the default value, but we want to remember the empty value.
  1176. SetStringValueEvenIfEmpty(FCustomCommandOptions, OptionKey, Value);
  1177. }
  1178. ControlIndex++;
  1179. }
  1180. }
  1181. if (ShortCut != NULL)
  1182. {
  1183. *ShortCut = GetShortCutCombo(FShortCutCombo);
  1184. }
  1185. }
  1186. return Result;
  1187. }
  1188. //---------------------------------------------------------------------------
  1189. UnicodeString __fastcall TCustomCommandOptionsDialog::GetComboBoxValue(
  1190. TControl * Control, const UnicodeString & Default)
  1191. {
  1192. TComboBox * ComboBox = DebugNotNull(dynamic_cast<TComboBox *>(Control));
  1193. UnicodeString Result;
  1194. if (ComboBox->ItemIndex < 0)
  1195. {
  1196. Result = Default;
  1197. }
  1198. else
  1199. {
  1200. Result = FValues[GetControlIndex(Control)][ComboBox->ItemIndex];
  1201. }
  1202. return Result;
  1203. }
  1204. //---------------------------------------------------------------------------
  1205. UnicodeString __fastcall TCustomCommandOptionsDialog::SaveHistoryComboBoxValue(
  1206. TControl * Control, const TCustomCommandType::TOption & Option)
  1207. {
  1208. THistoryComboBox * ComboBox = DebugNotNull(dynamic_cast<THistoryComboBox *>(Control));
  1209. ComboBox->SaveToHistory();
  1210. CustomWinConfiguration->History[HistoryKey(Option)] = ComboBox->Items;
  1211. return ComboBox->Text;
  1212. }
  1213. //---------------------------------------------------------------------------
  1214. void __fastcall TCustomCommandOptionsDialog::DoHelp()
  1215. {
  1216. UnicodeString HelpPage;
  1217. if (!FCommand->OptionsPage.IsEmpty())
  1218. {
  1219. HelpPage = FCommand->OptionsPage;
  1220. }
  1221. else
  1222. {
  1223. HelpPage = FCommand->HomePage;
  1224. }
  1225. if (!HelpPage.IsEmpty())
  1226. {
  1227. OpenBrowser(HelpPage);
  1228. }
  1229. else
  1230. {
  1231. TCustomDialog::DoHelp();
  1232. }
  1233. }
  1234. //---------------------------------------------------------------------------
  1235. void __fastcall TCustomCommandOptionsDialog::DoShow()
  1236. {
  1237. TCustomDialog::DoShow();
  1238. int ControlIndex = 0;
  1239. for (int OptionIndex = 0; OptionIndex < FCommand->OptionsCount; OptionIndex++)
  1240. {
  1241. const TCustomCommandType::TOption & Option = FCommand->GetOption(OptionIndex);
  1242. if ((Option.Flags & FFlags) != 0)
  1243. {
  1244. if (Option.Kind == TCustomCommandType::okFile)
  1245. {
  1246. TControl * Control = FControls[ControlIndex];
  1247. InstallPathWordBreakProc(DebugNotNull(dynamic_cast<TWinControl *>(Control)));
  1248. }
  1249. ControlIndex++;
  1250. }
  1251. }
  1252. }
  1253. //---------------------------------------------------------------------------
  1254. bool __fastcall DoCustomCommandOptionsDialog(
  1255. const TCustomCommandType * Command, TStrings * CustomCommandOptions, TShortCut * ShortCut,
  1256. unsigned int Flags, TCustomCommand * CustomCommandForOptions,
  1257. const UnicodeString & Site, const TShortCuts * ShortCuts)
  1258. {
  1259. std::unique_ptr<TCustomCommandOptionsDialog> Dialog(
  1260. new TCustomCommandOptionsDialog(Command, CustomCommandOptions, Flags, CustomCommandForOptions, Site, ShortCuts));
  1261. return Dialog->Execute(ShortCut);
  1262. }
  1263. //---------------------------------------------------------------------------
  1264. //---------------------------------------------------------------------------
  1265. class TUsageStatisticsDialog : public TCustomDialog
  1266. {
  1267. public:
  1268. __fastcall TUsageStatisticsDialog();
  1269. protected:
  1270. virtual void __fastcall DoChange(bool & CanSubmit);
  1271. private:
  1272. TEdit * FilterEdit;
  1273. TMemo * UsageMemo;
  1274. TButton * ClipboardButton;
  1275. void __fastcall ClipboardButtonClick(TObject * Sender);
  1276. };
  1277. //---------------------------------------------------------------------------
  1278. __fastcall TUsageStatisticsDialog::TUsageStatisticsDialog() :
  1279. TCustomDialog(HELP_USAGE)
  1280. {
  1281. Caption = LoadStr(USAGE_CAPTION);
  1282. Width = ScaleByTextHeight(this, 400);
  1283. // UnformatMessage is called, because previously, ** markup was used and translations may still contain that
  1284. AddText(CreateLabel(UnformatMessage(LoadStr(USAGE_DATA2))));
  1285. FilterEdit = new TEdit(this);
  1286. FilterEdit->Width = ScaleByTextHeight(this, 250);
  1287. AddEdit(FilterEdit, CreateLabel(LoadStr(USAGE_FILTER)), true);
  1288. UsageMemo = new TMemo(this);
  1289. UsageMemo->Height = ScaleByTextHeight(this, 300);
  1290. UsageMemo->ScrollBars = ssVertical;
  1291. AddEdit(UsageMemo, NULL);
  1292. ReadOnlyControl(UsageMemo);
  1293. ClipboardButton = new TButton(this);
  1294. ClipboardButton->Caption = LoadStr(USAGE_COPY);
  1295. ClipboardButton->Width = ScaleByTextHeight(this, 161);
  1296. ClipboardButton->OnClick = ClipboardButtonClick;
  1297. AddDialogButton(ClipboardButton);
  1298. RemoveCancelButton();
  1299. }
  1300. //---------------------------------------------------------------------------
  1301. void __fastcall TUsageStatisticsDialog::ClipboardButtonClick(TObject * /*Sender*/)
  1302. {
  1303. TInstantOperationVisualizer Visualizer;
  1304. CopyToClipboard(UsageMemo->Lines);
  1305. }
  1306. //---------------------------------------------------------------------------
  1307. void __fastcall TUsageStatisticsDialog::DoChange(bool & CanSubmit)
  1308. {
  1309. TCustomDialog::DoChange(CanSubmit);
  1310. UnicodeString Text = Configuration->Usage->Serialize(L"\n", FilterEdit->Text);
  1311. bool NoUsage = Text.IsEmpty();
  1312. ClipboardButton->Enabled = !NoUsage;
  1313. if (NoUsage)
  1314. {
  1315. Text = LoadStr(USAGE_DATA_NONE2);
  1316. }
  1317. UsageMemo->Lines->Text = Text;
  1318. }
  1319. //---------------------------------------------------------------------------
  1320. void __fastcall DoUsageStatisticsDialog()
  1321. {
  1322. std::unique_ptr<TUsageStatisticsDialog> Dialog(new TUsageStatisticsDialog());
  1323. Dialog->Execute();
  1324. }
  1325. //---------------------------------------------------------------------------
  1326. //---------------------------------------------------------------------------
  1327. class TSiteRawDialog : public TCustomDialog
  1328. {
  1329. public:
  1330. __fastcall TSiteRawDialog();
  1331. bool __fastcall Execute(TSessionData * Data);
  1332. protected:
  1333. DYNAMIC void __fastcall DoShow();
  1334. private:
  1335. TMemo * SettingsMemo;
  1336. void __fastcall AddButtonClick(TObject * Sender);
  1337. void __fastcall SettingsMemoKeyDown(TObject * Sender, WORD & Key, TShiftState Shift);
  1338. void DeleteNames(TStrings * Names, TStrings * Options);
  1339. };
  1340. //---------------------------------------------------------------------------
  1341. __fastcall TSiteRawDialog::TSiteRawDialog() :
  1342. TCustomDialog(HELP_SITE_RAW)
  1343. {
  1344. Caption = LoadStr(SITE_RAW_CAPTION);
  1345. Width = ScaleByTextHeight(this, 400);
  1346. SettingsMemo = new TMemo(this);
  1347. SettingsMemo->Height = ScaleByTextHeight(this, 300);
  1348. SettingsMemo->OnKeyDown = SettingsMemoKeyDown;
  1349. AddEdit(SettingsMemo, NULL);
  1350. TButton * AddButton = new TButton(this);
  1351. AddButton->Caption = LoadStr(SITE_RAW_ADD);
  1352. AddButton->Width = OKButton->Width;
  1353. AddButton->OnClick = AddButtonClick;
  1354. AddDialogButton(AddButton);
  1355. }
  1356. //---------------------------------------------------------------------------
  1357. void __fastcall TSiteRawDialog::DoShow()
  1358. {
  1359. TCustomDialog::DoShow();
  1360. InstallPathWordBreakProc(SettingsMemo);
  1361. }
  1362. //---------------------------------------------------------------------------
  1363. bool __fastcall TSiteRawDialog::Execute(TSessionData * Data)
  1364. {
  1365. std::unique_ptr<TSessionData> FactoryDefaults(new TSessionData(L""));
  1366. std::unique_ptr<TSessionData> RawData(new TSessionData(L""));
  1367. RawData->Assign(Data);
  1368. // SFTP-only is not reflected by the protocol prefix, we have to use rawsettings for that
  1369. if (RawData->FSProtocol != fsSFTPonly)
  1370. {
  1371. RawData->FSProtocol = FactoryDefaults->FSProtocol;
  1372. }
  1373. RawData->HostName = FactoryDefaults->HostName;
  1374. RawData->PortNumber = FactoryDefaults->PortNumber;
  1375. RawData->UserName = FactoryDefaults->UserName;
  1376. RawData->Password = FactoryDefaults->Password;
  1377. RawData->Ftps = FactoryDefaults->Ftps;
  1378. std::unique_ptr<TStrings> Options(RawData->SaveToOptions(FactoryDefaults.get(), false, false));
  1379. SettingsMemo->Lines = Options.get();
  1380. bool Result = TCustomDialog::Execute();
  1381. if (Result)
  1382. {
  1383. std::unique_ptr<TSessionData> BackupData(new TSessionData(L""));
  1384. BackupData->Assign(Data);
  1385. Data->DefaultSettings();
  1386. Data->FSProtocol = BackupData->FSProtocol;
  1387. Data->HostName = BackupData->HostName;
  1388. Data->PortNumber = BackupData->PortNumber;
  1389. Data->UserName = BackupData->UserName;
  1390. Data->Password = BackupData->Password;
  1391. Data->Ftps = BackupData->Ftps;
  1392. Data->ApplyRawSettings(SettingsMemo->Lines, false);
  1393. }
  1394. return Result;
  1395. }
  1396. //---------------------------------------------------------------------------
  1397. void __fastcall TSiteRawDialog::SettingsMemoKeyDown(TObject * Sender, WORD & Key, TShiftState Shift)
  1398. {
  1399. MemoKeyDown(Sender, Key, Shift);
  1400. }
  1401. //---------------------------------------------------------------------------
  1402. void __fastcall TSiteRawDialog::AddButtonClick(TObject *)
  1403. {
  1404. std::unique_ptr<TSessionData> FactoryDefaults(new TSessionData(L""));
  1405. std::unique_ptr<TSessionData> BasicData(new TSessionData(L""));
  1406. BasicData->FSProtocol = TFSProtocol(FactoryDefaults->FSProtocol + 1);
  1407. UnicodeString RandomAppendix(L"_");
  1408. BasicData->HostName = FactoryDefaults->HostName + RandomAppendix;
  1409. BasicData->Ftps = TFtps(FactoryDefaults->Ftps + 1);
  1410. BasicData->PortNumber = BasicData->GetDefaultPort() + 1;
  1411. BasicData->UserName = FactoryDefaults->UserName + RandomAppendix;
  1412. BasicData->Password = FactoryDefaults->Password + RandomAppendix;
  1413. std::unique_ptr<TStrings> BasicOptions(BasicData->SaveToOptions(FactoryDefaults.get(), false, false));
  1414. std::unique_ptr<TStrings> AllOptions(TSessionData::GetAllOptionNames(false));
  1415. std::unique_ptr<TStrings> Names(CreateSortedStringList());
  1416. for (int Index = 0; Index < AllOptions->Count; Index++)
  1417. {
  1418. Names->Add(AllOptions->Names[Index]);
  1419. }
  1420. DeleteNames(Names.get(), BasicOptions.get());
  1421. DeleteNames(Names.get(), SettingsMemo->Lines);
  1422. std::unique_ptr<TCustomDialog> AddDialog(new TCustomDialog(HelpKeyword));
  1423. AddDialog->Caption = LoadStr(SITE_RAW_ADD_CAPTION);
  1424. TComboBox * AddComboBox = new TUIStateAwareComboBox(AddDialog.get());
  1425. AddComboBox->Style = csDropDownList;
  1426. AddComboBox->DropDownCount = Max(AddComboBox->DropDownCount, 16);
  1427. AddDialog->AddComboBox(AddComboBox, CreateLabel(LoadStr(SITE_RAW_ADD_LABEL)), Names.get(), true);
  1428. AddComboBox->ItemIndex = 0;
  1429. if (AddDialog->Execute())
  1430. {
  1431. UnicodeString Name = AddComboBox->Items->Strings[AddComboBox->ItemIndex];
  1432. UnicodeString Value = AllOptions->Values[Name];
  1433. UnicodeString NameAndSeparator = Name + SettingsMemo->Lines->NameValueSeparator;
  1434. int Start = (SettingsMemo->Lines->Text + NameAndSeparator).Length();
  1435. SettingsMemo->Lines->Add(NameAndSeparator + Value);
  1436. SettingsMemo->SetFocus();
  1437. SettingsMemo->SelStart = Start;
  1438. SettingsMemo->SelLength = Value.Length();
  1439. }
  1440. }
  1441. //---------------------------------------------------------------------------
  1442. void TSiteRawDialog::DeleteNames(TStrings * Names, TStrings * Options)
  1443. {
  1444. for (int Index = 0; Index < Options->Count; Index++)
  1445. {
  1446. UnicodeString Name = Options->Names[Index];
  1447. if (!Name.IsEmpty())
  1448. {
  1449. int I = Names->IndexOf(Name);
  1450. if (I >= 0)
  1451. {
  1452. Names->Delete(I);
  1453. }
  1454. }
  1455. }
  1456. }
  1457. //---------------------------------------------------------------------------
  1458. void __fastcall DoSiteRawDialog(TSessionData * Data)
  1459. {
  1460. std::unique_ptr<TSiteRawDialog> Dialog(new TSiteRawDialog());
  1461. Dialog->Execute(Data);
  1462. }
  1463. //---------------------------------------------------------------------------
  1464. //---------------------------------------------------------------------------
  1465. class TSshHostCADialog : public TCustomDialog
  1466. {
  1467. public:
  1468. TSshHostCADialog(bool Add);
  1469. bool Execute(TSshHostCA & SshHostCA);
  1470. protected:
  1471. virtual void __fastcall DoChange(bool & CanSubmit);
  1472. virtual void __fastcall DoValidate();
  1473. private:
  1474. TEdit * NameEdit;
  1475. TEdit * PublicKeyEdit;
  1476. TEdit * PublicKeyLabel;
  1477. TEdit * ValidityExpressionEdit;
  1478. TCheckBox * PermitRsaSha1Check;
  1479. TCheckBox * PermitRsaSha256Check;
  1480. TCheckBox * PermitRsaSha512Check;
  1481. void __fastcall BrowseButtonClick(TObject * Sender);
  1482. TCheckBox * AddValidityCheckBox(int CaptionStrPart);
  1483. bool ValidatePublicKey(UnicodeString & Status);
  1484. };
  1485. //---------------------------------------------------------------------------
  1486. TSshHostCADialog::TSshHostCADialog(bool Add) :
  1487. TCustomDialog(HELP_SSH_HOST_CA)
  1488. {
  1489. ClientWidth = ScaleByTextHeight(this, 520);
  1490. Caption = LoadStr(Add ? SSH_HOST_CA_ADD : SSH_HOST_CA_EDIT);
  1491. NameEdit = new TEdit(this);
  1492. AddEdit(NameEdit, CreateLabel(LoadStr(SSH_HOST_CA_NAME)));
  1493. PublicKeyEdit = new TEdit(this);
  1494. AddEdit(PublicKeyEdit, CreateLabel(LoadStr(SSH_HOST_CA_PUBLIC_KEY)));
  1495. TButton * BrowseButton = new TButton(this);
  1496. BrowseButton->Caption = LoadStr(SSH_HOST_CA_BROWSE);
  1497. BrowseButton->OnClick = BrowseButtonClick;
  1498. AddButtonNextToEdit(BrowseButton, PublicKeyEdit);
  1499. NameEdit->Width = PublicKeyEdit->Width;
  1500. PublicKeyLabel = new TEdit(this);
  1501. ReadOnlyControl(PublicKeyLabel);
  1502. PublicKeyLabel->BorderStyle = bsNone;
  1503. PublicKeyLabel->TabStop = false;
  1504. AddEditLikeControl(PublicKeyLabel, NULL);
  1505. ValidityExpressionEdit = new TEdit(this);
  1506. AddEdit(ValidityExpressionEdit, CreateLabel(LoadStr(SSH_HOST_CA_PUBLIC_HOSTS)));
  1507. TLabel * Label = CreateLabel(LoadStr(SSH_HOST_CA_SIGNATURE_TYPES));
  1508. AddText(Label);
  1509. PermitRsaSha1Check = AddValidityCheckBox(1);
  1510. int PermitCheckBoxTop = Label->Top - (PermitRsaSha1Check->Height - Label->Height) / 2;
  1511. PermitRsaSha1Check->Left = OKButton->Left + 1;
  1512. PermitRsaSha1Check->Top = PermitCheckBoxTop;
  1513. PermitRsaSha256Check = AddValidityCheckBox(2);
  1514. PermitRsaSha256Check->Left = CancelButton->Left + 1;
  1515. PermitRsaSha256Check->Top = PermitCheckBoxTop;
  1516. PermitRsaSha512Check = AddValidityCheckBox(3);
  1517. PermitRsaSha512Check->Left = HelpButton->Left + 1;
  1518. PermitRsaSha512Check->Top = PermitCheckBoxTop;
  1519. }
  1520. //---------------------------------------------------------------------------
  1521. TCheckBox * TSshHostCADialog::AddValidityCheckBox(int CaptionStrPart)
  1522. {
  1523. TCheckBox * Result = new TCheckBox(this);
  1524. Result->Parent = this;
  1525. Result->Caption = LoadStrPart(SSH_HOST_CA_SIGNATURES, CaptionStrPart);
  1526. ScaleButtonControl(Result);
  1527. Result->Left = Left;
  1528. Result->OnClick = Change;
  1529. AddWinControl(Result);
  1530. return Result;
  1531. }
  1532. //---------------------------------------------------------------------------
  1533. bool TSshHostCADialog::ValidatePublicKey(UnicodeString & Status)
  1534. {
  1535. bool Result = false;
  1536. UnicodeString PublicKeyText = PublicKeyEdit->Text.Trim();
  1537. if (PublicKeyText.IsEmpty())
  1538. {
  1539. Status = LoadStr(SSH_HOST_CA_NO_KEY);
  1540. }
  1541. else
  1542. {
  1543. RawByteString PublicKeyDummy;
  1544. try
  1545. {
  1546. ParseCertificatePublicKey(PublicKeyText, PublicKeyDummy, Status);
  1547. Result = true;
  1548. }
  1549. catch (Exception & E)
  1550. {
  1551. Status = E.Message;
  1552. }
  1553. }
  1554. return Result;
  1555. }
  1556. //---------------------------------------------------------------------------
  1557. void __fastcall TSshHostCADialog::DoChange(bool & CanSubmit)
  1558. {
  1559. TCustomDialog::DoChange(CanSubmit);
  1560. UnicodeString PublicKeyStatus;
  1561. ValidatePublicKey(PublicKeyStatus);
  1562. // Should drop "SHA256:" prefix, the way we do in TSecureShell::VerifyHostKey
  1563. PublicKeyLabel->Text = PublicKeyStatus;
  1564. CanSubmit = !NameEdit->Text.Trim().IsEmpty() && !PublicKeyEdit->Text.Trim().IsEmpty();
  1565. }
  1566. //---------------------------------------------------------------------
  1567. void __fastcall TSshHostCADialog::DoValidate()
  1568. {
  1569. TCustomDialog::DoValidate();
  1570. UnicodeString PublicKeyStatus;
  1571. if (!ValidatePublicKey(PublicKeyStatus))
  1572. {
  1573. PublicKeyEdit->SetFocus();
  1574. throw Exception(MainInstructions(PublicKeyStatus));
  1575. }
  1576. if (ValidityExpressionEdit->Text.Trim().IsEmpty())
  1577. {
  1578. ValidityExpressionEdit->SetFocus();
  1579. throw Exception(MainInstructions(LoadStr(SSH_HOST_CA_NO_HOSTS)));
  1580. }
  1581. UnicodeString Error;
  1582. int ErrorStart, ErrorLen;
  1583. if (!IsCertificateValidityExpressionValid(ValidityExpressionEdit->Text, Error, ErrorStart, ErrorLen))
  1584. {
  1585. std::unique_ptr<TStrings> MoreMessages(TextToStringList(Error));
  1586. MoreMessageDialog(MainInstructions(LoadStr(SSH_HOST_CA_HOSTS_INVALID)), MoreMessages.get(), qtError, qaOK, HelpKeyword);
  1587. ValidityExpressionEdit->SetFocus();
  1588. ValidityExpressionEdit->SelStart = ErrorStart;
  1589. ValidityExpressionEdit->SelLength = ErrorLen;
  1590. Abort();
  1591. }
  1592. }
  1593. //---------------------------------------------------------------------------
  1594. bool TSshHostCADialog::Execute(TSshHostCA & SshHostCA)
  1595. {
  1596. NameEdit->Text = SshHostCA.Name;
  1597. RawByteString PublicKey = SshHostCA.PublicKey;
  1598. PublicKeyEdit->Text = EncodeStrToBase64(PublicKey);
  1599. ValidityExpressionEdit->Text = SshHostCA.ValidityExpression;
  1600. PermitRsaSha1Check->Checked = SshHostCA.PermitRsaSha1;
  1601. PermitRsaSha256Check->Checked = SshHostCA.PermitRsaSha256;
  1602. PermitRsaSha512Check->Checked = SshHostCA.PermitRsaSha512;
  1603. bool Result = TCustomDialog::Execute();
  1604. if (Result)
  1605. {
  1606. SshHostCA.Name = NameEdit->Text;
  1607. SshHostCA.PublicKey = DecodeBase64ToStr(PublicKeyEdit->Text);
  1608. SshHostCA.ValidityExpression = ValidityExpressionEdit->Text;
  1609. SshHostCA.PermitRsaSha1 = PermitRsaSha1Check->Checked;
  1610. SshHostCA.PermitRsaSha256 = PermitRsaSha256Check->Checked;
  1611. SshHostCA.PermitRsaSha512 = PermitRsaSha512Check->Checked;
  1612. }
  1613. return Result;
  1614. }
  1615. //---------------------------------------------------------------------------
  1616. void __fastcall TSshHostCADialog::BrowseButtonClick(TObject *)
  1617. {
  1618. std::unique_ptr<TOpenDialog> OpenDialog(new TOpenDialog(Application));
  1619. OpenDialog->Title = LoadStr(SSH_HOST_CA_BROWSE_TITLE);
  1620. OpenDialog->Filter = LoadStr(SSH_HOST_CA_BROWSE_FILTER);
  1621. bool Result = OpenDialog->Execute();
  1622. if (Result)
  1623. {
  1624. UnicodeString FileName = OpenDialog->FileName;
  1625. UnicodeString Algorithm, Comment;
  1626. bool HasCertificate;
  1627. RawByteString PublicKey;
  1628. try
  1629. {
  1630. PublicKey = LoadPublicKey(FileName, Algorithm, Comment, HasCertificate);
  1631. }
  1632. catch (Exception & E)
  1633. {
  1634. throw ExtException(&E, MainInstructions(FMTLOAD(SSH_HOST_CA_LOAD_ERROR, (FileName))));
  1635. }
  1636. if (NameEdit->Text.IsEmpty())
  1637. {
  1638. NameEdit->Text = Comment;
  1639. }
  1640. PublicKeyEdit->Text = EncodeStrToBase64(PublicKey);
  1641. }
  1642. }
  1643. //---------------------------------------------------------------------------
  1644. bool DoSshHostCADialog(bool Add, TSshHostCA & SshHostCA)
  1645. {
  1646. std::unique_ptr<TSshHostCADialog> Dialog(new TSshHostCADialog(Add));
  1647. return Dialog->Execute(SshHostCA);
  1648. }