1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708 |
- //---------------------------------------------------------------------------
- #include <vcl.h>
- #pragma hdrstop
- #include <Consts.hpp>
- #include <shlobj.h>
- #include <stdio.h>
- #define INITGUID
- #include <propkey.h>
- #include <powrprof.h>
- #include <Common.h>
- #include <TextsWin.h>
- #include <TextsCore.h>
- #include <HelpWin.h>
- #include <Exceptions.h>
- #include <CoreMain.h>
- #include <RemoteFiles.h>
- #include <PuttyTools.h>
- #include "GUITools.h"
- #include "VCLCommon.h"
- #include "Setup.h"
- #include "Tools.h"
- #include <WinHelpViewer.hpp>
- #include <PasTools.hpp>
- #include <System.Win.ComObj.hpp>
- #include <StrUtils.hpp>
- #include <WinConfiguration.h>
- #include <ProgParams.h>
- //---------------------------------------------------------------------------
- // WORKAROUND
- // VCL includes wininet.h (even with NO_WIN32_LEAN_AND_MEAN)
- // and it cannot be combined with winhttp.h as of current Windows SDK.
- // This is hack to allow that.
- // https://social.msdn.microsoft.com/Forums/windowsdesktop/en-US/8f468d9f-3f15-452c-803d-fc63ab3f684e/cannot-use-both-winineth-and-winhttph
- #undef BOOLAPI
- #undef SECURITY_FLAG_IGNORE_CERT_DATE_INVALID
- #undef SECURITY_FLAG_IGNORE_CERT_CN_INVALID
- #define URL_COMPONENTS URL_COMPONENTS_ANOTHER
- #define URL_COMPONENTSA URL_COMPONENTSA_ANOTHER
- #define URL_COMPONENTSW URL_COMPONENTSW_ANOTHER
- #define LPURL_COMPONENTS LPURL_COMPONENTS_ANOTHER
- #define LPURL_COMPONENTSA LPURL_COMPONENTS_ANOTHER
- #define LPURL_COMPONENTSW LPURL_COMPONENTS_ANOTHER
- #define INTERNET_SCHEME INTERNET_SCHEME_ANOTHER
- #define LPINTERNET_SCHEME LPINTERNET_SCHEME_ANOTHER
- #define HTTP_VERSION_INFO HTTP_VERSION_INFO_ANOTHER
- #define LPHTTP_VERSION_INFO LPHTTP_VERSION_INFO_ANOTHER
- #include <winhttp.h>
- #undef URL_COMPONENTS
- #undef URL_COMPONENTSA
- #undef URL_COMPONENTSW
- #undef LPURL_COMPONENTS
- #undef LPURL_COMPONENTSA
- #undef LPURL_COMPONENTSW
- #undef INTERNET_SCHEME
- #undef LPINTERNET_SCHEME
- #undef HTTP_VERSION_INFO
- #undef LPHTTP_VERSION_INFO
- //---------------------------------------------------------------------------
- #pragma package(smart_init)
- //---------------------------------------------------------------------------
- TFontStyles __fastcall IntToFontStyles(int value)
- {
- TFontStyles Result;
- for (int i = fsBold; i <= fsStrikeOut; i++)
- {
- if (value & 1)
- {
- Result << (TFontStyle)i;
- }
- value >>= 1;
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- int __fastcall FontStylesToInt(const TFontStyles value)
- {
- int Result = 0;
- for (int i = fsStrikeOut; i >= fsBold; i--)
- {
- Result <<= 1;
- if (value.Contains((TFontStyle)i))
- {
- Result |= 1;
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- bool __fastcall SameFont(TFont * Font1, TFont * Font2)
- {
- // keep in sync with TFontConfiguration::operator !=
- return
- SameText(Font1->Name, Font2->Name) && (Font1->Height == Font2->Height) &&
- (Font1->Charset == Font2->Charset) && (Font1->Style == Font2->Style);
- }
- //---------------------------------------------------------------------------
- TColor __fastcall GetWindowTextColor(TColor BackgroundColor, TColor Color)
- {
- if (Color == TColor(0))
- {
- // Could use current theme TMT_TEXTCOLOR - see https://github.com/ysc3839/win32-darkmode
- Color = (IsDarkColor(BackgroundColor) ? clWhite : clWindowText);
- SetContrast(Color, BackgroundColor, 180);
- }
- return Color;
- }
- //---------------------------------------------------------------------------
- TColor __fastcall GetWindowColor(TColor Color)
- {
- if (Color == TColor(0))
- {
- // Could use current theme TMT_FILLCOLOR - see https://github.com/ysc3839/win32-darkmode
- Color = (WinConfiguration->UseDarkTheme() ? static_cast<TColor>(RGB(0x20, 0x20, 0x20)) : clWindow);
- }
- return Color;
- }
- //---------------------------------------------------------------------------
- TColor __fastcall GetBtnFaceColor()
- {
- return WinConfiguration->UseDarkTheme() ? TColor(RGB(43, 43, 43)) : clBtnFace;
- }
- //---------------------------------------------------------------------------
- TColor __fastcall GetNonZeroColor(TColor Color)
- {
- // 0,0,0 is "default color"
- if (Color == TColor(0))
- {
- // use near-black instead
- Color = TColor(RGB(1, 1, 1));
- }
- return Color;
- }
- //---------------------------------------------------------------------------
- void __fastcall CenterFormOn(TForm * Form, TControl * CenterOn)
- {
- TPoint ScreenPoint = CenterOn->ClientToScreen(TPoint(0, 0));
- Form->Left = ScreenPoint.x + (CenterOn->Width / 2) - (Form->Width / 2);
- Form->Top = ScreenPoint.y + (CenterOn->Height / 2) - (Form->Height / 2);
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall GetListViewStr(TListView * ListView)
- {
- UnicodeString Result;
- for (int Index = 0; Index < ListView->Columns->Count; Index++)
- {
- AddToList(Result, IntToStr(ListView->Column[Index]->Width), L",");
- }
- // WORKAROUND
- // Adding an additional comma after the list,
- // to ensure that old versions that did not expect the pixels-per-inch part,
- // stop at the comma, otherwise they try to parse the
- // "last-column-width;pixels-per-inch" as integer and throw.
- // For the other instance of this hack, see TCustomListViewColProperties.GetParamsStr
- Result += L",;" + SavePixelsPerInch(ListView);
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall LoadListViewStr(TListView * ListView, UnicodeString ALayoutStr)
- {
- UnicodeString LayoutStr = CutToChar(ALayoutStr, L';', true);
- int PixelsPerInch = LoadPixelsPerInch(CutToChar(ALayoutStr, L';', true), ListView);
- int Index = 0;
- while (!LayoutStr.IsEmpty() && (Index < ListView->Columns->Count))
- {
- int Width;
- if (TryStrToInt(CutToChar(LayoutStr, L',', true), Width))
- {
- ListView->Column[Index]->Width = LoadDimension(Width, PixelsPerInch, ListView);
- }
- Index++;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall LoadFormDimensions(
- const UnicodeString & LeftStr, const UnicodeString & TopStr, const UnicodeString & RightStr, const UnicodeString & BottomStr,
- int PixelsPerInch, Forms::TMonitor * Monitor, TForm * Form, TRect & Bounds, bool & DefaultPos)
- {
- DefaultPos = (StrToIntDef(LeftStr, 0) == -1) && (StrToIntDef(TopStr, 0) == -1);
- if (!DefaultPos)
- {
- Bounds.Left = StrToDimensionDef(LeftStr, PixelsPerInch, Form, Bounds.Left);
- Bounds.Top = StrToDimensionDef(TopStr, PixelsPerInch, Form, Bounds.Top);
- }
- else
- {
- Bounds.Left = 0;
- Bounds.Top = 0;
- }
- Bounds.Right = StrToDimensionDef(RightStr, PixelsPerInch, Form, Bounds.Right);
- Bounds.Bottom = StrToDimensionDef(BottomStr, PixelsPerInch, Form, Bounds.Bottom);
- // move to the target monitor
- OffsetRect(Bounds, Monitor->Left, Monitor->Top);
- // reduce window size to that of monitor size
- // (this does not cut window into monitor!)
- if (Bounds.Width() > Monitor->WorkareaRect.Width())
- {
- Bounds.Right -= (Bounds.Width() - Monitor->WorkareaRect.Width());
- }
- if (Bounds.Height() > Monitor->WorkareaRect.Height())
- {
- Bounds.Bottom -= (Bounds.Height() - Monitor->WorkareaRect.Height());
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall RestoreForm(UnicodeString Data, TForm * Form, bool PositionOnly)
- {
- DebugAssert(Form);
- if (!Data.IsEmpty())
- {
- Forms::TMonitor * Monitor = FormMonitor(Form);
- UnicodeString LeftStr = CutToChar(Data, L';', true);
- UnicodeString TopStr = CutToChar(Data, L';', true);
- UnicodeString RightStr = CutToChar(Data, L';', true);
- UnicodeString BottomStr = CutToChar(Data, L';', true);
- TWindowState State = (TWindowState)StrToIntDef(CutToChar(Data, L';', true), (int)wsNormal);
- int PixelsPerInch = LoadPixelsPerInch(CutToChar(Data, L';', true), Form);
- TRect OriginalBounds = Form->BoundsRect;
- int OriginalPixelsPerInch = Form->PixelsPerInch;
- Form->WindowState = State;
- if (State == wsNormal)
- {
- bool DefaultPos;
- TRect Bounds = OriginalBounds;
- LoadFormDimensions(LeftStr, TopStr, RightStr, BottomStr, PixelsPerInch, Monitor, Form, Bounds, DefaultPos);
- int Padding = ScaleByPixelsPerInch(20, Monitor);
- if (DefaultPos ||
- ((Bounds.Left < Monitor->Left - Padding) ||
- (Bounds.Left > Monitor->Left + Monitor->WorkareaRect.Width() - Padding) ||
- (Bounds.Top < Monitor->Top - Padding) ||
- (Bounds.Top > Monitor->Top + Monitor->WorkareaRect.Height() - Padding)))
- {
- bool ExplicitPlacing = !Monitor->Primary;
- if (!ExplicitPlacing)
- {
- TPosition Position;
- if ((Application->MainForm == NULL) || (Application->MainForm == Form))
- {
- Position = poDefaultPosOnly;
- }
- else
- {
- Position = poOwnerFormCenter;
- }
- // If handle is allocated already, changing Position reallocates it, what brings lot of nasty side effects.
- if (Form->HandleAllocated() && (Form->Position != Position))
- {
- ExplicitPlacing = true;
- }
- else
- {
- Form->Width = Bounds.Width();
- Form->Height = Bounds.Height();
- }
- }
- if (ExplicitPlacing)
- {
- // when positioning on non-primary monitor, we need
- // to handle that ourselves, so place window to center
- if (!PositionOnly)
- {
- Form->SetBounds(Monitor->Left + ((Monitor->Width - Bounds.Width()) / 2),
- Monitor->Top + ((Monitor->Height - Bounds.Height()) / 2),
- Bounds.Width(), Bounds.Height());
- }
- if (!Form->HandleAllocated())
- {
- Form->Position = poDesigned;
- }
- }
- }
- else
- {
- Form->Position = poDesigned;
- if (!PositionOnly)
- {
- Form->BoundsRect = Bounds;
- // If setting bounds moved the form to another monitor with non-default DPI,
- // recalculate the size to avoid rounding issues
- // (as the form was very likely moved to the monitor, where the sizes were saved)
- // See also TCustomScpExplorerForm::DoShow()
- if (OriginalPixelsPerInch != Form->PixelsPerInch)
- {
- TRect Bounds2 = OriginalBounds;
- LoadFormDimensions(LeftStr, TopStr, RightStr, BottomStr, PixelsPerInch, Monitor, Form, Bounds2, DefaultPos);
- DebugAssert(!DefaultPos);
- Form->BoundsRect = Bounds2;
- }
- }
- }
- }
- else if (State == wsMaximized)
- {
- Form->Position = poDesigned;
- if (!PositionOnly)
- {
- TRect Bounds2 = Form->BoundsRect;
- OffsetRect(Bounds2, Monitor->Left, Monitor->Top);
- Form->BoundsRect = Bounds2;
- }
- }
- }
- else if (Form->Position == poDesigned)
- {
- Form->Position = poDefaultPosOnly;
- }
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall StoreForm(TCustomForm * Form)
- {
- DebugAssert(Form);
- TRect Bounds = Form->BoundsRect;
- OffsetRect(Bounds, -Form->Monitor->Left, -Form->Monitor->Top);
- UnicodeString Result =
- FORMAT(L"%d;%d;%d;%d;%d;%s", (SaveDimension(Bounds.Left), SaveDimension(Bounds.Top),
- SaveDimension(Bounds.Right), SaveDimension(Bounds.Bottom),
- // we do not want WinSCP to start minimized next time (we cannot handle that anyway).
- // note that WindowState is wsNormal when window in minimized for some reason.
- // actually it is wsMinimized only when minimized by MSVDM
- (int)(Form->WindowState == wsMinimized ? wsNormal : Form->WindowState),
- SavePixelsPerInch(Form)));
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall RestoreFormSize(UnicodeString Data, TForm * Form)
- {
- // This has to be called only after DoFormWindowProc(CM_SHOWINGCHANGED).
- // See comment in ResizeForm.
- UnicodeString WidthStr = CutToChar(Data, L',', true);
- UnicodeString HeightStr = CutToChar(Data, L',', true);
- int PixelsPerInch = LoadPixelsPerInch(CutToChar(Data, L',', true), Form);
- int Width = StrToDimensionDef(WidthStr, PixelsPerInch, Form, Form->Width);
- int Height = StrToDimensionDef(HeightStr, PixelsPerInch, Form, Form->Height);
- ResizeForm(Form, Width, Height);
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall StoreFormSize(TForm * Form)
- {
- return FORMAT(L"%d,%d,%s", (Form->Width, Form->Height, SavePixelsPerInch(Form)));
- }
- //---------------------------------------------------------------------------
- static void __fastcall ExecuteProcessAndReadOutput(const
- UnicodeString & Command, const UnicodeString & HelpKeyword, UnicodeString & Output)
- {
- if (!CopyCommandToClipboard(Command))
- {
- SECURITY_ATTRIBUTES SecurityAttributes;
- ZeroMemory(&SecurityAttributes, sizeof(SecurityAttributes));
- SecurityAttributes.nLength = sizeof(SECURITY_ATTRIBUTES);
- SecurityAttributes.bInheritHandle = TRUE;
- SecurityAttributes.lpSecurityDescriptor = NULL;
- HANDLE PipeRead = INVALID_HANDLE_VALUE;
- HANDLE PipeWrite = INVALID_HANDLE_VALUE;
- if (!CreatePipe(&PipeRead, &PipeWrite, &SecurityAttributes, 0) ||
- !SetHandleInformation(PipeRead, HANDLE_FLAG_INHERIT, 0))
- {
- throw EOSExtException(FMTLOAD(EXECUTE_APP_ERROR, (Command)));
- }
- PROCESS_INFORMATION ProcessInformation;
- ZeroMemory(&ProcessInformation, sizeof(ProcessInformation));
- try
- {
- try
- {
- STARTUPINFO StartupInfo;
- ZeroMemory(&StartupInfo, sizeof(StartupInfo));
- StartupInfo.cb = sizeof(STARTUPINFO);
- StartupInfo.hStdError = PipeWrite;
- StartupInfo.hStdOutput = PipeWrite;
- StartupInfo.wShowWindow = SW_HIDE;
- StartupInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
- if (!CreateProcess(NULL, Command.c_str(), NULL, NULL, TRUE, 0, NULL, NULL, &StartupInfo, &ProcessInformation))
- {
- throw EOSExtException(FMTLOAD(EXECUTE_APP_ERROR, (Command)));
- }
- }
- __finally
- {
- // If we do not close the handle here, the ReadFile below would get stuck once the app finishes writting,
- // as it still sees that someone "can" write to the pipe.
- CloseHandle(PipeWrite);
- }
- DWORD BytesAvail;
- while (PeekNamedPipe(PipeRead, NULL, 0, NULL, &BytesAvail, NULL))
- {
- if (BytesAvail > 0)
- {
- char Buffer[4096];
- DWORD BytesToRead = std::min(BytesAvail, static_cast<unsigned long>(sizeof(Buffer)));
- DWORD BytesRead;
- if (ReadFile(PipeRead, Buffer, BytesToRead, &BytesRead, NULL))
- {
- Output += UnicodeString(UTF8String(Buffer, BytesRead));
- }
- }
- // Same as in ExecuteShellCheckedAndWait
- Sleep(50);
- Application->ProcessMessages();
- }
- DWORD ExitCode;
- if (DebugAlwaysTrue(GetExitCodeProcess(ProcessInformation.hProcess, &ExitCode)) &&
- (ExitCode != 0))
- {
- UnicodeString Buf = Output;
- UnicodeString Buf2;
- if (ExtractMainInstructions(Buf, Buf2))
- {
- throw ExtException(Output, UnicodeString(), HelpKeyword);
- }
- else
- {
- throw ExtException(MainInstructions(FMTLOAD(COMMAND_FAILED_CODEONLY, (static_cast<int>(ExitCode)))), Output, HelpKeyword);
- }
- }
- }
- __finally
- {
- CloseHandle(ProcessInformation.hProcess);
- CloseHandle(ProcessInformation.hThread);
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall ExecuteProcessChecked(
- const UnicodeString & Command, const UnicodeString & HelpKeyword, UnicodeString * Output)
- {
- if (Output == NULL)
- {
- ExecuteShellChecked(Command);
- }
- else
- {
- ExecuteProcessAndReadOutput(Command, HelpKeyword, *Output);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall ExecuteProcessCheckedAndWait(
- const UnicodeString & Command, const UnicodeString & HelpKeyword, UnicodeString * Output)
- {
- if (Output == NULL)
- {
- ExecuteShellCheckedAndWait(Command, &Application->ProcessMessages);
- }
- else
- {
- ExecuteProcessAndReadOutput(Command, HelpKeyword, *Output);
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall IsKeyPressed(int VirtualKey)
- {
- return FLAGSET(GetAsyncKeyState(VirtualKey), 0x8000);
- }
- //---------------------------------------------------------------------------
- bool __fastcall UseAlternativeFunction()
- {
- return IsKeyPressed(VK_SHIFT);
- }
- //---------------------------------------------------------------------------
- bool __fastcall OpenInNewWindow()
- {
- return UseAlternativeFunction();
- }
- //---------------------------------------------------------------------------
- void ExecuteSelf(const UnicodeString & Params)
- {
- ExecuteShellChecked(Application->ExeName, Params);
- }
- //---------------------------------------------------------------------------
- void __fastcall ExecuteNewInstance(const UnicodeString & Param, const UnicodeString & AdditionalParams)
- {
- UnicodeString Arg;
- if (!Param.IsEmpty())
- {
- Arg = FORMAT(L"\"%s\"", (Param));
- }
- UnicodeString Space(L" ");
- AddToList(Arg, TProgramParams::FormatSwitch(NEWINSTANCE_SWICH), Space);
- AddToList(Arg, AdditionalParams, Space);
- ExecuteSelf(Arg);
- }
- //---------------------------------------------------------------------------
- IShellLink * __fastcall CreateDesktopShortCut(const UnicodeString & Name,
- const UnicodeString &File, const UnicodeString & Params, const UnicodeString & Description,
- int SpecialFolder, int IconIndex, bool Return)
- {
- IShellLink* pLink = NULL;
- if (SpecialFolder < 0)
- {
- SpecialFolder = CSIDL_DESKTOPDIRECTORY;
- }
- try
- {
- OleCheck(CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (void **) &pLink));
- try
- {
- pLink->SetPath(File.c_str());
- pLink->SetDescription(Description.c_str());
- pLink->SetArguments(Params.c_str());
- pLink->SetShowCmd(SW_SHOW);
- // Explicitly setting icon file,
- // without this icons are not shown at least in Windows 7 jumplist
- pLink->SetIconLocation(File.c_str(), IconIndex);
- IPersistFile* pPersistFile;
- if (!Return &&
- SUCCEEDED(pLink->QueryInterface(IID_IPersistFile, (void **)&pPersistFile)))
- {
- try
- {
- LPMALLOC ShellMalloc;
- LPITEMIDLIST DesktopPidl;
- wchar_t DesktopDir[MAX_PATH];
- OleCheck(SHGetMalloc(&ShellMalloc));
- try
- {
- OleCheck(SHGetSpecialFolderLocation(NULL, SpecialFolder, &DesktopPidl));
- OleCheck(SHGetPathFromIDList(DesktopPidl, DesktopDir));
- }
- __finally
- {
- ShellMalloc->Free(DesktopPidl);
- ShellMalloc->Release();
- }
- WideString strShortCutLocation(DesktopDir);
- // Name can contain even path (e.g. to create quick launch icon)
- strShortCutLocation += UnicodeString(L"\\") + Name + L".lnk";
- OleCheck(pPersistFile->Save(strShortCutLocation.c_bstr(), TRUE));
- }
- __finally
- {
- pPersistFile->Release();
- }
- }
- // this is necessary for Windows 7 taskbar jump list links
- IPropertyStore * PropertyStore;
- if (SUCCEEDED(pLink->QueryInterface(IID_IPropertyStore, (void**)&PropertyStore)))
- {
- PROPVARIANT Prop;
- Prop.vt = VT_LPWSTR;
- Prop.pwszVal = Name.c_str();
- PropertyStore->SetValue(PKEY_Title, Prop);
- PropertyStore->Commit();
- PropertyStore->Release();
- }
- }
- catch(...)
- {
- pLink->Release();
- throw;
- }
- if (!Return)
- {
- pLink->Release();
- pLink = NULL;
- }
- }
- catch(Exception & E)
- {
- throw ExtException(&E, LoadStr(CREATE_SHORTCUT_ERROR));
- }
- return pLink;
- }
- //---------------------------------------------------------------------------
- IShellLink * __fastcall CreateAppDesktopShortCut(
- const UnicodeString & Name, const UnicodeString & AParams, const UnicodeString & Description,
- int SpecialFolder, int IconIndex, bool Return)
- {
- UnicodeString ParamValue = Configuration->GetIniFileParamValue();
- UnicodeString Params;
- if (!ParamValue.IsEmpty())
- {
- Params = TProgramParams::FormatSwitch(INI_SWITCH) + L"=" + AddQuotes(ParamValue);
- }
- AddToList(Params, AParams, L" ");
- return CreateDesktopShortCut(Name, Application->ExeName, Params, Description, SpecialFolder, IconIndex, Return);
- }
- //---------------------------------------------------------------------------
- IShellLink * __fastcall CreateDesktopSessionShortCut(
- const UnicodeString & SessionName, UnicodeString Name,
- const UnicodeString & AdditionalParams, int SpecialFolder, int IconIndex,
- bool Return)
- {
- bool DefaultsOnly;
- UnicodeString InfoTip;
- bool IsFolder = StoredSessions->IsFolder(SessionName);
- bool IsWorkspace = StoredSessions->IsWorkspace(SessionName);
- if (IsFolder || IsWorkspace)
- {
- InfoTip = FMTLOAD(
- (IsFolder ? SHORTCUT_INFO_TIP_FOLDER : SHORTCUT_INFO_TIP_WORKSPACE),
- (SessionName));
- if (Name.IsEmpty())
- {
- // no slashes in filename
- Name = UnixExtractFileName(SessionName);
- }
- }
- else
- {
- // this should not be done for workspaces and folders
- TSessionData * SessionData =
- StoredSessions->ParseUrl(EncodeUrlString(SessionName), NULL, DefaultsOnly);
- InfoTip =
- FMTLOAD(SHORTCUT_INFO_TIP, (SessionName, SessionData->InfoTip));
- if (Name.IsEmpty())
- {
- // no slashes in filename
- Name = SessionData->LocalName;
- }
- delete SessionData;
- }
- return
- CreateAppDesktopShortCut(ValidLocalFileName(Name),
- FORMAT(L"\"%s\"%s%s", (EncodeUrlString(SessionName), (AdditionalParams.IsEmpty() ? L"" : L" "), AdditionalParams)),
- InfoTip, SpecialFolder, IconIndex, Return);
- }
- //---------------------------------------------------------------------------
- void ValidateMask(const UnicodeString & Mask, int ForceDirectoryMasks)
- {
- TFileMasks Masks(ForceDirectoryMasks);
- Masks = Mask;
- }
- //---------------------------------------------------------------------------
- template<class TEditControl>
- void __fastcall ValidateMaskEditT(const UnicodeString & Mask, TEditControl * Edit, int ForceDirectoryMasks)
- {
- DebugAssert(Edit != NULL);
- if (!IsCancelButtonBeingClicked(Edit))
- {
- try
- {
- ValidateMask(Mask, ForceDirectoryMasks);
- }
- catch(EFileMasksException & E)
- {
- ShowExtendedException(&E);
- Edit->SetFocus();
- // This does not work for TEdit and TMemo (descendants of TCustomEdit) anymore,
- // as it re-selects whole text on exception in TCustomEdit.CMExit
- Edit->SelStart = E.ErrorStart - 1;
- Edit->SelLength = E.ErrorLen;
- Abort();
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall ValidateMaskEdit(TComboBox * Edit)
- {
- ValidateMaskEditT(Edit->Text, Edit, -1);
- }
- //---------------------------------------------------------------------------
- void __fastcall ValidateMaskEdit(TEdit * Edit)
- {
- ValidateMaskEditT(Edit->Text, Edit, -1);
- }
- //---------------------------------------------------------------------------
- void __fastcall ValidateMaskEdit(TMemo * Edit, bool Directory)
- {
- if (!IsCancelButtonBeingClicked(Edit))
- {
- UnicodeString Mask = TFileMasks::ComposeMaskStr(GetUnwrappedMemoLines(Edit), Directory);
- ValidateMaskEditT(Mask, Edit, Directory ? 1 : 0);
- }
- }
- //---------------------------------------------------------------------------
- TStrings * __fastcall GetUnwrappedMemoLines(TMemo * Memo)
- {
- // This removes soft linebreaks when text in memo wraps
- // (Memo->Lines includes soft linebreaks, while Memo->Text does not)
- return TextToStringList(Memo->Text);
- }
- //---------------------------------------------------------------------------
- void __fastcall ExitActiveControl(TForm * Form)
- {
- if (Form->ActiveControl != NULL)
- {
- TNotifyEvent OnExit = ((TEdit*)Form->ActiveControl)->OnExit;
- if (OnExit != NULL)
- {
- OnExit(Form->ActiveControl);
- }
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall IsWinSCPUrl(const UnicodeString & Url)
- {
- UnicodeString HomePageUrl = LoadStr(HOMEPAGE_URL);
- UnicodeString HttpHomePageUrl = ChangeUrlProtocol(HomePageUrl, HttpProtocol);
- DebugAssert(HomePageUrl != HttpHomePageUrl);
- return
- StartsText(HomePageUrl, Url) ||
- StartsText(HttpHomePageUrl, Url);
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall SecureUrl(const UnicodeString & Url)
- {
- UnicodeString Result = Url;
- if (IsWinSCPUrl(Url) && IsHttpUrl(Url))
- {
- Result = ChangeUrlProtocol(Result, HttpsProtocol);
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall OpenBrowser(UnicodeString URL)
- {
- if (IsWinSCPUrl(URL))
- {
- DebugAssert(!IsHttpUrl(URL));
- URL = CampaignUrl(URL);
- }
- if (!CopyCommandToClipboard(URL))
- {
- // Rather arbitrary limit. Opening a URL over approx. 5 KB fails in Chrome, Firefox and Edge.
- const int URLLimit = 4*1024;
- if (URL.Length() > URLLimit)
- {
- URL.SetLength(URLLimit);
- }
- ShellExecute(Application->Handle, L"open", URL.c_str(), NULL, NULL, SW_SHOWNORMAL);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall OpenFolderInExplorer(const UnicodeString & Path)
- {
- if ((int)ShellExecute(Application->Handle, L"explore",
- (wchar_t*)Path.data(), NULL, NULL, SW_SHOWNORMAL) <= 32)
- {
- throw Exception(FMTLOAD(EXPLORE_LOCAL_DIR_ERROR, (Path)));
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall OpenFileInExplorer(const UnicodeString & Path)
- {
- PCIDLIST_ABSOLUTE Folder = ILCreateFromPathW(ApiPath(Path).c_str());
- SHOpenFolderAndSelectItems(Folder, 0, NULL, 0);
- }
- //---------------------------------------------------------------------------
- void __fastcall ShowHelp(const UnicodeString & AHelpKeyword)
- {
- // see also AppendUrlParams
- UnicodeString HelpKeyword = AHelpKeyword;
- const wchar_t FragmentSeparator = L'#';
- UnicodeString HelpPath = CutToChar(HelpKeyword, FragmentSeparator, false);
- UnicodeString HelpUrl = FMTLOAD(DOCUMENTATION_KEYWORD_URL2, (HelpPath, Configuration->ProductVersion, GUIConfiguration->AppliedLocaleHex));
- AddToList(HelpUrl, HelpKeyword, FragmentSeparator);
- OpenBrowser(HelpUrl);
- }
- //---------------------------------------------------------------------------
- bool __fastcall IsFormatInClipboard(unsigned int Format)
- {
- bool Result = OpenClipboard(0);
- if (Result)
- {
- Result = IsClipboardFormatAvailable(Format);
- CloseClipboard();
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- HANDLE __fastcall OpenTextFromClipboard(const wchar_t *& Text)
- {
- UnicodeString ErrorContext;
- try
- {
- HANDLE Result = NULL;
- ErrorContext = L"open";
- if (OpenClipboard(0))
- {
- // Check also for CF_TEXT?
- ErrorContext = L"getdata";
- Result = GetClipboardData(CF_UNICODETEXT);
- if (Result != NULL)
- {
- ErrorContext = L"lock";
- Text = static_cast<const wchar_t*>(GlobalLock(Result));
- }
- else
- {
- ErrorContext = L"close";
- CloseClipboard();
- }
- }
- return Result;
- }
- catch (EAccessViolation & E)
- {
- throw EAccessViolation(AddContextToExceptionMessage(E, ErrorContext));
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall CloseTextFromClipboard(HANDLE Handle)
- {
- UnicodeString ErrorContext;
- try
- {
- if (Handle != NULL)
- {
- ErrorContext = "unlock";
- GlobalUnlock(Handle);
- }
- ErrorContext = "close";
- CloseClipboard();
- }
- catch (EAccessViolation & E)
- {
- throw EAccessViolation(AddContextToExceptionMessage(E, ErrorContext));
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall TextFromClipboard(UnicodeString & Text, bool Trim)
- {
- UnicodeString ErrorContext;
- try
- {
- const wchar_t * AText = NULL;
- ErrorContext = L"open";
- HANDLE Handle = OpenTextFromClipboard(AText);
- bool Result = (Handle != NULL);
- if (Result)
- {
- // For all current uses (URL pasting, key/fingerprint pasting, known_hosts pasting, "more messages" copying,
- // permissions pasting), 64KB is large enough.
- const int Limit = 64*1024;
- ErrorContext = L"size";
- size_t Size = GlobalSize(Handle);
- int Len = (Size / sizeof(*AText)) - 1;
- if (Len > Limit)
- {
- ErrorContext = FORMAT(L"substring(%d,%d)", (int(Size), Len));
- Text = UnicodeString(AText, Limit);
- }
- else
- {
- ErrorContext = FORMAT(L"string(%d,%d)", (int(Size), Len));
- Text = AText;
- }
- if (Trim)
- {
- ErrorContext = L"trim";
- Text = Text.Trim();
- }
- ErrorContext = L"close";
- CloseTextFromClipboard(Handle);
- }
- return Result;
- }
- catch (EAccessViolation & E)
- {
- throw EAccessViolation(AddContextToExceptionMessage(E, ErrorContext));
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall NonEmptyTextFromClipboard(UnicodeString & Text)
- {
- return
- TextFromClipboard(Text, true) &&
- !Text.IsEmpty();
- }
- //---------------------------------------------------------------------------
- static bool __fastcall GetResource(
- const UnicodeString ResName, void *& Content, unsigned long & Size)
- {
- HRSRC Resource = FindResourceEx(HInstance, RT_RCDATA, ResName.c_str(),
- MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL));
- bool Result = (Resource != NULL);
- if (Result)
- {
- Size = SizeofResource(HInstance, Resource);
- if (!Size)
- {
- throw Exception(FORMAT(L"Cannot get size of resource %s", (ResName)));
- }
- Content = LoadResource(HInstance, Resource);
- if (!Content)
- {
- throw Exception(FORMAT(L"Cannot read resource %s", (ResName)));
- }
- Content = LockResource(Content);
- if (!Content)
- {
- throw Exception(FORMAT(L"Cannot lock resource %s", (ResName)));
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- bool __fastcall DumpResourceToFile(const UnicodeString ResName,
- const UnicodeString FileName)
- {
- void * Content;
- unsigned long Size;
- bool Result = GetResource(ResName, Content, Size);
- if (Result)
- {
- FILE * f = _wfopen(ApiPath(FileName).c_str(), L"wb");
- if (!f)
- {
- throw Exception(FORMAT(L"Cannot create file %s", (FileName)));
- }
- if (fwrite(Content, 1, Size, f) != Size)
- {
- throw Exception(FORMAT(L"Cannot write to file %s", (FileName)));
- }
- fclose(f);
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall ReadResource(const UnicodeString ResName)
- {
- void * Content;
- unsigned long Size;
- UnicodeString Result;
- if (GetResource(ResName, Content, Size))
- {
- Result = UnicodeString(UTF8String(static_cast<char*>(Content), Size));
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- template <class T>
- void __fastcall BrowseForExecutableT(T * Control, UnicodeString Title,
- UnicodeString Filter, bool FileNameCommand, bool Escape)
- {
- UnicodeString Executable, Program, Params, Dir;
- Executable = Control->Text;
- if (FileNameCommand)
- {
- ReformatFileNameCommand(Executable);
- }
- SplitCommand(Executable, Program, Params, Dir);
- TOpenDialog * FileDialog = new TOpenDialog(Application);
- try
- {
- if (Escape)
- {
- Program = ReplaceStr(Program, L"\\\\", L"\\");
- }
- UnicodeString ExpandedProgram = ExpandEnvironmentVariables(Program);
- FileDialog->FileName = ExpandedProgram;
- UnicodeString InitialDir = ExtractFilePath(ExpandedProgram);
- if (!InitialDir.IsEmpty())
- {
- FileDialog->InitialDir = InitialDir;
- }
- FileDialog->Filter = Filter;
- FileDialog->Title = Title;
- if (FileDialog->Execute())
- {
- TNotifyEvent PrevOnChange = Control->OnChange;
- Control->OnChange = NULL;
- try
- {
- // preserve unexpanded file, if the destination has not changed actually
- if (!IsPathToSameFile(ExpandedProgram, FileDialog->FileName))
- {
- Program = FileDialog->FileName;
- if (Escape)
- {
- Program = ReplaceStr(Program, L"\\", L"\\\\");
- }
- }
- Control->Text = FormatCommand(Program, Params);
- }
- __finally
- {
- Control->OnChange = PrevOnChange;
- }
- if (Control->OnExit != NULL)
- {
- Control->OnExit(Control);
- }
- }
- }
- __finally
- {
- delete FileDialog;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall BrowseForExecutable(TEdit * Control, UnicodeString Title,
- UnicodeString Filter, bool FileNameCommand, bool Escape)
- {
- BrowseForExecutableT(Control, Title, Filter, FileNameCommand, Escape);
- }
- //---------------------------------------------------------------------------
- void __fastcall BrowseForExecutable(TComboBox * Control, UnicodeString Title,
- UnicodeString Filter, bool FileNameCommand, bool Escape)
- {
- BrowseForExecutableT(Control, Title, Filter, FileNameCommand, Escape);
- }
- //---------------------------------------------------------------------------
- bool __fastcall FontDialog(TFont * Font)
- {
- bool Result;
- TFontDialog * Dialog = new TFontDialog(Application);
- try
- {
- Dialog->Device = fdScreen;
- Dialog->Options = TFontDialogOptions() << fdForceFontExist;
- Dialog->Font = Font;
- Result = Dialog->Execute();
- if (Result)
- {
- Font->Assign(Dialog->Font);
- }
- }
- __finally
- {
- delete Dialog;
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- bool __fastcall SaveDialog(UnicodeString Title, UnicodeString Filter,
- UnicodeString DefaultExt, UnicodeString & FileName)
- {
- bool Result;
- #if 0
- TFileSaveDialog * Dialog = new TFileSaveDialog(Application);
- try
- {
- Dialog->Title = Title;
- FilterToFileTypes(Filter, Dialog->FileTypes);
- Dialog->DefaultExtension = DefaultExt;
- Dialog->FileName = FileName;
- UnicodeString DefaultFolder = ExtractFilePath(FileName);
- if (!DefaultFolder.IsEmpty())
- {
- Dialog->DefaultFolder = DefaultFolder;
- }
- Dialog->Options = Dialog->Options << fdoOverWritePrompt << fdoForceFileSystem <<
- fdoPathMustExist << fdoNoReadOnlyReturn;
- Result = Dialog->Execute();
- if (Result)
- {
- FileName = Dialog->FileName;
- }
- }
- __finally
- {
- delete Dialog;
- }
- #else
- TSaveDialog * Dialog = new TSaveDialog(Application);
- try
- {
- Dialog->Title = Title;
- Dialog->Filter = Filter;
- Dialog->DefaultExt = DefaultExt;
- Dialog->FileName = FileName;
- UnicodeString InitialDir = ExtractFilePath(FileName);
- if (!InitialDir.IsEmpty())
- {
- Dialog->InitialDir = InitialDir;
- }
- Dialog->Options = Dialog->Options << ofOverwritePrompt << ofPathMustExist <<
- ofNoReadOnlyReturn;
- Result = Dialog->Execute();
- if (Result)
- {
- FileName = Dialog->FileName;
- }
- }
- __finally
- {
- delete Dialog;
- }
- #endif
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall CopyToClipboard(UnicodeString Text)
- {
- HANDLE Data;
- void * DataPtr;
- AppLogFmt("Copying text to clipboard [%d]", (Text.Length()));
- if (OpenClipboard(0))
- {
- try
- {
- size_t Size = (Text.Length() + 1) * sizeof(wchar_t);
- Data = GlobalAlloc(GMEM_MOVEABLE + GMEM_DDESHARE, Size);
- try
- {
- DataPtr = GlobalLock(Data);
- try
- {
- memcpy(DataPtr, Text.c_str(), Size);
- EmptyClipboard();
- SetClipboardData(CF_UNICODETEXT, Data);
- }
- __finally
- {
- GlobalUnlock(Data);
- }
- }
- catch(...)
- {
- GlobalFree(Data);
- throw;
- }
- }
- __finally
- {
- CloseClipboard();
- }
- AppLogFmt("Copied text to clipboard [%d]", (Text.Length()));
- }
- else
- {
- throw Exception(Vcl_Consts_SCannotOpenClipboard);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall CopyToClipboard(TStrings * Strings)
- {
- if (Strings->Count > 0)
- {
- CopyToClipboard(StringsToText(Strings));
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall IsWin64()
- {
- static int Result = -1;
- if (Result < 0)
- {
- Result = 0;
- BOOL Wow64Process = FALSE;
- if (IsWow64Process(GetCurrentProcess(), &Wow64Process))
- {
- if (Wow64Process)
- {
- Result = 1;
- }
- }
- }
- return (Result > 0);
- }
- //---------------------------------------------------------------------------
- static void __fastcall AcquireShutDownPrivileges()
- {
- HANDLE Token;
- // Get a token for this process.
- Win32Check(OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &Token));
- TOKEN_PRIVILEGES Priv;
- ZeroMemory(&Priv, sizeof(Priv));
- // Get the LUID for the shutdown privilege.
- // For hibernate/suspend, you need the same:
- // https://stackoverflow.com/q/959589/850848
- Win32Check(LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &Priv.Privileges[0].Luid));
- Priv.PrivilegeCount = 1; // one privilege to set
- Priv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
- // Get the shutdown privilege for this process.
- Win32Check(AdjustTokenPrivileges(Token, FALSE, &Priv, 0, (PTOKEN_PRIVILEGES)NULL, 0));
- }
- //---------------------------------------------------------------------------
- void __fastcall ShutDownWindows()
- {
- AcquireShutDownPrivileges();
- // Shut down the system and force all applications to close.
- Win32Check(ExitWindowsEx(EWX_SHUTDOWN | EWX_POWEROFF,
- SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER | SHTDN_REASON_FLAG_PLANNED));
- }
- //---------------------------------------------------------------------------
- void __fastcall SuspendWindows()
- {
- AcquireShutDownPrivileges();
- // https://learn.microsoft.com/en-us/windows/win32/api/powrprof/nf-powrprof-setsuspendstate
- Win32Check(SetSuspendState(false, false, false));
- }
- //---------------------------------------------------------------------------
- void __fastcall EditSelectBaseName(HWND Edit)
- {
- UnicodeString Text;
- Text.SetLength(GetWindowTextLength(Edit) + 1);
- GetWindowText(Edit, Text.c_str(), Text.Length());
- int P = Text.LastDelimiter(L".");
- if (P > 0)
- {
- // SendMessage does not work, if edit control is not fully
- // initialized yet
- PostMessage(Edit, EM_SETSEL, 0, P - 1);
- }
- }
- //---------------------------------------------------------------------------
- UnicodeString GetConvertedKeyFileName(const UnicodeString & FileName)
- {
- return ChangeFileExt(FileName, FORMAT(L".%s", (PuttyKeyExt)));
- }
- //---------------------------------------------------------------------------
- static void __fastcall ConvertKey(UnicodeString & FileName, TKeyType Type)
- {
- UnicodeString Passphrase;
- UnicodeString Comment;
- if (IsKeyEncrypted(Type, FileName, Comment))
- {
- if (!InputDialog(
- LoadStr(PASSPHRASE_TITLE),
- FORMAT(LoadStr(PROMPT_KEY_PASSPHRASE), (Comment)),
- Passphrase, HELP_NONE, NULL, false, NULL, false))
- {
- Abort();
- }
- }
- TPrivateKey * PrivateKey = LoadKey(Type, FileName, Passphrase);
- try
- {
- FileName = GetConvertedKeyFileName(FileName);
- if (!SaveDialog(LoadStr(CONVERTKEY_SAVE_TITLE), LoadStr(CONVERTKEY_SAVE_FILTER), PuttyKeyExt, FileName))
- {
- Abort();
- }
- SaveKey(ktSSH2, FileName, Passphrase, PrivateKey);
- MessageDialog(MainInstructions(FMTLOAD(CONVERTKEY_SAVED, (FileName))), qtInformation, qaOK);
- }
- __finally
- {
- FreeKey(PrivateKey);
- }
- }
- //---------------------------------------------------------------------------
- void DoVerifyKey(UnicodeString & FileName, bool Convert, UnicodeString & Message, TStrings *& MoreMessages, UnicodeString & HelpKeyword)
- {
- std::unique_ptr<TStrings> AMoreMessages;
- if (!FileName.Trim().IsEmpty())
- {
- FileName = ExpandEnvironmentVariables(FileName);
- TKeyType Type = KeyType(FileName);
- // reason _wfopen failed
- int Error = errno;
- HelpKeyword = HELP_LOGIN_KEY_TYPE;
- UnicodeString PuttygenPath;
- switch (Type)
- {
- case ktOpenSSHPEM:
- case ktOpenSSHNew:
- case ktSSHCom:
- {
- UnicodeString TypeName = ((Type == ktOpenSSHPEM) || (Type == ktOpenSSHNew)) ? L"OpenSSH" : L"ssh.com";
- Message = FMTLOAD(KEY_TYPE_UNSUPPORTED2, (FileName, TypeName));
- if (Convert)
- {
- Configuration->Usage->Inc(L"PrivateKeyConvertSuggestionsNative");
- UnicodeString ConvertMessage = FMTLOAD(KEY_TYPE_CONVERT4, (TypeName, RemoveMainInstructionsTag(Message)));
- Message = EmptyStr;
- if (MoreMessageDialog(ConvertMessage, NULL, qtConfirmation, qaOK | qaCancel, HelpKeyword) == qaOK)
- {
- ConvertKey(FileName, Type);
- Configuration->Usage->Inc(L"PrivateKeyConverted");
- }
- else
- {
- Abort();
- }
- }
- else
- {
- HelpKeyword = HELP_KEY_TYPE_UNSUPPORTED;
- }
- }
- break;
- case ktSSH1:
- Message = MainInstructions(FMTLOAD(KEY_TYPE_SSH1, (FileName)));
- break;
- case ktSSH2:
- Message = TestKey(Type, FileName);
- break;
- case ktSSH1Public:
- case ktSSH2PublicRFC4716:
- case ktSSH2PublicOpenSSH:
- // noop
- break;
- case ktUnopenable:
- Message = MainInstructions(FMTLOAD(KEY_TYPE_UNOPENABLE, (FileName)));
- if (Error != ERROR_SUCCESS)
- {
- AMoreMessages.reset(TextToStringList(SysErrorMessageForError(Error)));
- }
- break;
- default:
- DebugFail();
- // fallthru
- case ktUnknown:
- Message = MainInstructions(FMTLOAD(KEY_TYPE_UNKNOWN2, (FileName)));
- break;
- }
- }
- MoreMessages = AMoreMessages.release();
- }
- //---------------------------------------------------------------------------
- static void __fastcall DoVerifyKey(UnicodeString & FileName, bool Convert, bool CanIgnore)
- {
- TStrings * AMoreMessages;
- UnicodeString Message;
- UnicodeString HelpKeyword;
- DoVerifyKey(FileName, Convert, Message, AMoreMessages, HelpKeyword);
- std::unique_ptr<TStrings> MoreMessages(AMoreMessages);
- if (!Message.IsEmpty())
- {
- Configuration->Usage->Inc(L"PrivateKeySelectErrors");
- unsigned int Answers = (CanIgnore ? (qaIgnore | qaAbort) : qaOK);
- if (MoreMessageDialog(Message, MoreMessages.get(), qtWarning, Answers, HelpKeyword) != qaIgnore)
- {
- Abort();
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall VerifyAndConvertKey(UnicodeString & FileName, bool CanIgnore)
- {
- DoVerifyKey(FileName, true, CanIgnore);
- }
- //---------------------------------------------------------------------------
- void __fastcall VerifyKey(const UnicodeString & FileName)
- {
- UnicodeString AFileName(FileName);
- DoVerifyKey(AFileName, false, true);
- }
- //---------------------------------------------------------------------------
- void __fastcall VerifyCertificate(const UnicodeString & FileName)
- {
- if (!FileName.Trim().IsEmpty())
- {
- try
- {
- CheckCertificate(FileName);
- }
- catch (Exception & E)
- {
- if (ExceptionMessageDialog(&E, qtWarning, L"", qaIgnore | qaAbort) == qaAbort)
- {
- Abort();
- }
- }
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall DetectSystemExternalEditor(
- bool AllowDefaultEditor,
- UnicodeString & Executable, UnicodeString & ExecutableDescription,
- UnicodeString & UsageState, bool & TryNextTime)
- {
- bool Result = false;
- UnicodeString TempName = ExcludeTrailingBackslash(WinConfiguration->TemporaryDir()) + L".txt";
- if (FileExists(ApiPath(TempName)))
- {
- TryNextTime = true;
- UsageState = "F";
- }
- else
- {
- unsigned int File = FileCreate(ApiPath(TempName));
- if (File == (unsigned int)INVALID_HANDLE_VALUE)
- {
- TryNextTime = true;
- UsageState = "F";
- }
- else
- {
- FileClose(File);
- try
- {
- wchar_t ExecutableBuf[MAX_PATH];
- if (!SUCCEEDED(FindExecutable(TempName.c_str(), NULL, ExecutableBuf)))
- {
- UsageState = "N";
- }
- else
- {
- Executable = ExecutableBuf;
- if (Executable.IsEmpty() ||
- !FileExists(ApiPath(Executable)))
- {
- UsageState = "N";
- }
- else
- {
- UnicodeString ExecutableName = ExtractFileName(Executable);
- if (!AllowDefaultEditor &&
- SameText(ExecutableName, TEditorPreferences::GetDefaultExternalEditor()))
- {
- UsageState = "P";
- Executable = L"";
- }
- else if (SameText(ExecutableName, "openwith.exe"))
- {
- UsageState = "W";
- Executable = L"";
- }
- else
- {
- try
- {
- ExecutableDescription = Configuration->GetFileDescription(Executable);
- }
- catch(...)
- {
- }
- if (ExecutableDescription.IsEmpty())
- {
- ExecutableDescription = ExecutableName;
- }
- Result = true;
- }
- }
- }
- }
- __finally
- {
- DeleteFile(ApiPath(TempName));
- }
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- // this was moved to global scope in past in some attempt to fix crashes,
- // not sure it really helped
- WINHTTP_CURRENT_USER_IE_PROXY_CONFIG IEProxyInfo;
- //---------------------------------------------------------------------------
- static bool __fastcall GetProxyUrlFromIE(UnicodeString & Proxy)
- {
- bool Result = false;
- memset(&IEProxyInfo, 0, sizeof(IEProxyInfo));
- if (WinHttpGetIEProxyConfigForCurrentUser(&IEProxyInfo))
- {
- if (IEProxyInfo.lpszProxy != NULL)
- {
- UnicodeString IEProxy = IEProxyInfo.lpszProxy;
- Proxy = L"";
- while (Proxy.IsEmpty() && !IEProxy.IsEmpty())
- {
- UnicodeString Str = CutToChar(IEProxy, L';', true);
- if (Str.Pos(L"=") == 0)
- {
- Proxy = Str;
- }
- else
- {
- UnicodeString Protocol = CutToChar(Str, L'=', true);
- if (SameText(Protocol, L"http"))
- {
- Proxy = Str;
- }
- }
- }
- GlobalFree(IEProxyInfo.lpszProxy);
- Result = true;
- }
- if (IEProxyInfo.lpszAutoConfigUrl != NULL)
- {
- GlobalFree(IEProxyInfo.lpszAutoConfigUrl);
- }
- if (IEProxyInfo.lpszProxyBypass != NULL)
- {
- GlobalFree(IEProxyInfo.lpszProxyBypass);
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- bool __fastcall AutodetectProxy(UnicodeString & HostName, int & PortNumber)
- {
- bool Result = false;
- /* First we try for proxy info direct from the registry if
- it's available. */
- UnicodeString Proxy;
- WINHTTP_PROXY_INFO ProxyInfo;
- memset(&ProxyInfo, 0, sizeof(ProxyInfo));
- if (WinHttpGetDefaultProxyConfiguration(&ProxyInfo))
- {
- if (ProxyInfo.lpszProxy != NULL)
- {
- Proxy = ProxyInfo.lpszProxy;
- GlobalFree(ProxyInfo.lpszProxy);
- Result = true;
- }
- if (ProxyInfo.lpszProxyBypass != NULL)
- {
- GlobalFree(ProxyInfo.lpszProxyBypass);
- }
- }
- /* The next fallback is to get the proxy info from MSIE. This is also
- usually much quicker than WinHttpGetProxyForUrl(), although sometimes
- it seems to fall back to that, based on the longish delay involved.
- Another issue with this is that it won't work in a service process
- that isn't impersonating an interactive user (since there isn't a
- current user), but in that case we just fall back to
- WinHttpGetProxyForUrl() */
- if (!Result)
- {
- Result = GetProxyUrlFromIE(Proxy);
- }
- if (Result)
- {
- if (Proxy.Trim().IsEmpty())
- {
- Result = false;
- }
- else
- {
- HostName = CutToChar(Proxy, L':', true);
- PortNumber = StrToIntDef(Proxy, ProxyPortNumber);
- }
- }
- // We can also use WinHttpGetProxyForUrl, but it is lengthy
- // See the source address of the code for example
- return Result;
- }
- //---------------------------------------------------------------------------
- //---------------------------------------------------------------------------
- class TWinHelpTester : public TInterfacedObject, public IWinHelpTester
- {
- public:
- virtual bool __fastcall CanShowALink(const UnicodeString ALink, const UnicodeString FileName);
- virtual bool __fastcall CanShowTopic(const UnicodeString Topic, const UnicodeString FileName);
- virtual bool __fastcall CanShowContext(const int Context, const UnicodeString FileName);
- virtual TStringList * __fastcall GetHelpStrings(const UnicodeString ALink);
- virtual UnicodeString __fastcall GetHelpPath();
- virtual UnicodeString __fastcall GetDefaultHelpFile();
- IUNKNOWN
- };
- //---------------------------------------------------------------------------
- class TCustomHelpSelector : public TInterfacedObject, public IHelpSelector
- {
- public:
- __fastcall TCustomHelpSelector(const UnicodeString & Name);
- virtual int __fastcall SelectKeyword(TStrings * Keywords);
- virtual int __fastcall TableOfContents(TStrings * Contents);
- IUNKNOWN
- private:
- UnicodeString FName;
- };
- //---------------------------------------------------------------------------
- void __fastcall AssignHelpSelector(IHelpSelector * HelpSelector)
- {
- _di_IHelpSystem HelpSystem;
- if (GetHelpSystem(HelpSystem))
- {
- HelpSystem->AssignHelpSelector(HelpSelector);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall InitializeCustomHelp(ICustomHelpViewer * HelpViewer)
- {
- _di_IHelpManager HelpManager;
- RegisterViewer(HelpViewer, HelpManager);
- // Register dummy tester that disables win help
- WinHelpTester = new TWinHelpTester();
- AssignHelpSelector(new TCustomHelpSelector(HelpViewer->GetViewerName()));
- }
- //---------------------------------------------------------------------------
- void __fastcall FinalizeCustomHelp()
- {
- AssignHelpSelector(NULL);
- }
- //---------------------------------------------------------------------------
- //---------------------------------------------------------------------------
- bool __fastcall TWinHelpTester::CanShowALink(const UnicodeString ALink,
- const UnicodeString FileName)
- {
- return !Application->HelpFile.IsEmpty();
- }
- //---------------------------------------------------------------------------
- bool __fastcall TWinHelpTester::CanShowTopic(const UnicodeString Topic,
- const UnicodeString FileName)
- {
- DebugFail();
- return !Application->HelpFile.IsEmpty();
- }
- //---------------------------------------------------------------------------
- bool __fastcall TWinHelpTester::CanShowContext(const int /*Context*/,
- const UnicodeString FileName)
- {
- DebugFail();
- return !Application->HelpFile.IsEmpty();
- }
- //---------------------------------------------------------------------------
- TStringList * __fastcall TWinHelpTester::GetHelpStrings(const UnicodeString ALink)
- {
- DebugFail();
- TStringList * Result = new TStringList();
- Result->Add(ViewerName + L": " + ALink);
- return Result;
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall TWinHelpTester::GetHelpPath()
- {
- // never called on windows anyway
- return ExtractFilePath(Application->HelpFile);
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall TWinHelpTester::GetDefaultHelpFile()
- {
- return Application->HelpFile;
- }
- //---------------------------------------------------------------------------
- //---------------------------------------------------------------------------
- __fastcall TCustomHelpSelector::TCustomHelpSelector(const UnicodeString & Name) :
- FName(Name)
- {
- }
- //---------------------------------------------------------------------------
- int __fastcall TCustomHelpSelector::SelectKeyword(TStrings * /*Keywords*/)
- {
- DebugFail();
- return 0;
- }
- //---------------------------------------------------------------------------
- int __fastcall TCustomHelpSelector::TableOfContents(TStrings * Contents)
- {
- return Contents->IndexOf(FName);
- }
|