Custom.cpp 60 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805
  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. AddEditLikeControl(PublicKeyLabel, NULL);
  1504. ValidityExpressionEdit = new TEdit(this);
  1505. AddEdit(ValidityExpressionEdit, CreateLabel(LoadStr(SSH_HOST_CA_PUBLIC_HOSTS)));
  1506. TLabel * Label = CreateLabel(LoadStr(SSH_HOST_CA_SIGNATURE_TYPES));
  1507. AddText(Label);
  1508. PermitRsaSha1Check = AddValidityCheckBox(1);
  1509. int PermitCheckBoxTop = Label->Top - (PermitRsaSha1Check->Height - Label->Height) / 2;
  1510. PermitRsaSha1Check->Left = OKButton->Left + 1;
  1511. PermitRsaSha1Check->Top = PermitCheckBoxTop;
  1512. PermitRsaSha256Check = AddValidityCheckBox(2);
  1513. PermitRsaSha256Check->Left = CancelButton->Left + 1;
  1514. PermitRsaSha256Check->Top = PermitCheckBoxTop;
  1515. PermitRsaSha512Check = AddValidityCheckBox(3);
  1516. PermitRsaSha512Check->Left = HelpButton->Left + 1;
  1517. PermitRsaSha512Check->Top = PermitCheckBoxTop;
  1518. }
  1519. //---------------------------------------------------------------------------
  1520. TCheckBox * TSshHostCADialog::AddValidityCheckBox(int CaptionStrPart)
  1521. {
  1522. TCheckBox * Result = new TCheckBox(this);
  1523. Result->Parent = this;
  1524. Result->Caption = LoadStrPart(SSH_HOST_CA_SIGNATURES, CaptionStrPart);
  1525. ScaleButtonControl(Result);
  1526. Result->Left = Left;
  1527. Result->OnClick = Change;
  1528. AddWinControl(Result);
  1529. return Result;
  1530. }
  1531. //---------------------------------------------------------------------------
  1532. bool TSshHostCADialog::ValidatePublicKey(UnicodeString & Status)
  1533. {
  1534. bool Result = false;
  1535. UnicodeString PublicKeyText = PublicKeyEdit->Text.Trim();
  1536. if (PublicKeyText.IsEmpty())
  1537. {
  1538. Status = LoadStr(SSH_HOST_CA_NO_KEY);
  1539. }
  1540. else
  1541. {
  1542. RawByteString PublicKeyDummy;
  1543. try
  1544. {
  1545. ParseCertificatePublicKey(PublicKeyText, PublicKeyDummy, Status);
  1546. Result = true;
  1547. }
  1548. catch (Exception & E)
  1549. {
  1550. Status = E.Message;
  1551. }
  1552. }
  1553. return Result;
  1554. }
  1555. //---------------------------------------------------------------------------
  1556. void __fastcall TSshHostCADialog::DoChange(bool & CanSubmit)
  1557. {
  1558. TCustomDialog::DoChange(CanSubmit);
  1559. UnicodeString PublicKeyStatus;
  1560. ValidatePublicKey(PublicKeyStatus);
  1561. // Should drop "SHA256:" prefix, the way we do in TSecureShell::VerifyHostKey
  1562. PublicKeyLabel->Text = PublicKeyStatus;
  1563. CanSubmit = !NameEdit->Text.Trim().IsEmpty() && !PublicKeyEdit->Text.Trim().IsEmpty();
  1564. }
  1565. //---------------------------------------------------------------------
  1566. void __fastcall TSshHostCADialog::DoValidate()
  1567. {
  1568. TCustomDialog::DoValidate();
  1569. UnicodeString PublicKeyStatus;
  1570. if (!ValidatePublicKey(PublicKeyStatus))
  1571. {
  1572. PublicKeyEdit->SetFocus();
  1573. throw Exception(MainInstructions(PublicKeyStatus));
  1574. }
  1575. if (ValidityExpressionEdit->Text.Trim().IsEmpty())
  1576. {
  1577. ValidityExpressionEdit->SetFocus();
  1578. throw Exception(MainInstructions(LoadStr(SSH_HOST_CA_NO_HOSTS)));
  1579. }
  1580. UnicodeString Error;
  1581. int ErrorStart, ErrorLen;
  1582. if (!IsCertificateValidityExpressionValid(ValidityExpressionEdit->Text, Error, ErrorStart, ErrorLen))
  1583. {
  1584. std::unique_ptr<TStrings> MoreMessages(TextToStringList(Error));
  1585. MoreMessageDialog(MainInstructions(LoadStr(SSH_HOST_CA_HOSTS_INVALID)), MoreMessages.get(), qtError, qaOK, HelpKeyword);
  1586. ValidityExpressionEdit->SetFocus();
  1587. ValidityExpressionEdit->SelStart = ErrorStart;
  1588. ValidityExpressionEdit->SelLength = ErrorLen;
  1589. Abort();
  1590. }
  1591. }
  1592. //---------------------------------------------------------------------------
  1593. bool TSshHostCADialog::Execute(TSshHostCA & SshHostCA)
  1594. {
  1595. NameEdit->Text = SshHostCA.Name;
  1596. RawByteString PublicKey = SshHostCA.PublicKey;
  1597. PublicKeyEdit->Text = EncodeStrToBase64(PublicKey);
  1598. ValidityExpressionEdit->Text = SshHostCA.ValidityExpression;
  1599. PermitRsaSha1Check->Checked = SshHostCA.PermitRsaSha1;
  1600. PermitRsaSha256Check->Checked = SshHostCA.PermitRsaSha256;
  1601. PermitRsaSha512Check->Checked = SshHostCA.PermitRsaSha512;
  1602. bool Result = TCustomDialog::Execute();
  1603. if (Result)
  1604. {
  1605. SshHostCA.Name = NameEdit->Text;
  1606. SshHostCA.PublicKey = DecodeBase64ToStr(PublicKeyEdit->Text);
  1607. SshHostCA.ValidityExpression = ValidityExpressionEdit->Text;
  1608. SshHostCA.PermitRsaSha1 = PermitRsaSha1Check->Checked;
  1609. SshHostCA.PermitRsaSha256 = PermitRsaSha256Check->Checked;
  1610. SshHostCA.PermitRsaSha512 = PermitRsaSha512Check->Checked;
  1611. }
  1612. return Result;
  1613. }
  1614. //---------------------------------------------------------------------------
  1615. void __fastcall TSshHostCADialog::BrowseButtonClick(TObject *)
  1616. {
  1617. std::unique_ptr<TOpenDialog> OpenDialog(new TOpenDialog(Application));
  1618. OpenDialog->Title = LoadStr(SSH_HOST_CA_BROWSE_TITLE);
  1619. OpenDialog->Filter = LoadStr(SSH_HOST_CA_BROWSE_FILTER);
  1620. bool Result = OpenDialog->Execute();
  1621. if (Result)
  1622. {
  1623. UnicodeString FileName = OpenDialog->FileName;
  1624. UnicodeString Algorithm, Comment;
  1625. bool HasCertificate;
  1626. RawByteString PublicKey;
  1627. try
  1628. {
  1629. PublicKey = LoadPublicKey(FileName, Algorithm, Comment, HasCertificate);
  1630. }
  1631. catch (Exception & E)
  1632. {
  1633. throw ExtException(&E, MainInstructions(FMTLOAD(SSH_HOST_CA_LOAD_ERROR, (FileName))));
  1634. }
  1635. if (NameEdit->Text.IsEmpty())
  1636. {
  1637. NameEdit->Text = Comment;
  1638. }
  1639. PublicKeyEdit->Text = EncodeStrToBase64(PublicKey);
  1640. }
  1641. }
  1642. //---------------------------------------------------------------------------
  1643. bool DoSshHostCADialog(bool Add, TSshHostCA & SshHostCA)
  1644. {
  1645. std::unique_ptr<TSshHostCADialog> Dialog(new TSshHostCADialog(Add));
  1646. return Dialog->Execute(SshHostCA);
  1647. }