1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825 |
- //---------------------------------------------------------------------------
- #define NO_WIN32_LEAN_AND_MEAN
- #include <vcl.h>
- #pragma hdrstop
- #include "TerminalManager.h"
- #include <Authenticate.h>
- #include "CustomScpExplorer.h"
- #include "NonVisual.h"
- #include "WinConfiguration.h"
- #include "Tools.h"
- #include <Common.h>
- #include <CoreMain.h>
- #include <GUITools.h>
- #include <TextsWin.h>
- #include <TextsCore.h>
- #include <Progress.h>
- #include <Exceptions.h>
- #include <VCLCommon.h>
- #include <WinApi.h>
- #include <PuttyTools.h>
- #include <HelpWin.h>
- #include <System.IOUtils.hpp>
- #include <StrUtils.hpp>
- //---------------------------------------------------------------------------
- #pragma package(smart_init)
- //---------------------------------------------------------------------------
- TTerminalManager * TTerminalManager::FInstance = NULL;
- //---------------------------------------------------------------------------
- __fastcall TManagedTerminal::TManagedTerminal(TSessionData * SessionData,
- TConfiguration * Configuration) :
- TTerminal(SessionData, Configuration),
- LocalExplorerState(NULL), RemoteExplorerState(NULL),
- ReopenStart(0), DirectoryLoaded(Now()), TerminalThread(NULL)
- {
- StateData = new TSessionData(L"");
- StateData->Assign(SessionData);
- StateData->LocalDirectory = ::ExpandFileName(StateData->LocalDirectory);
- }
- //---------------------------------------------------------------------------
- __fastcall TManagedTerminal::~TManagedTerminal()
- {
- delete StateData;
- delete LocalExplorerState;
- delete RemoteExplorerState;
- }
- //---------------------------------------------------------------------------
- //---------------------------------------------------------------------------
- TTerminalManager * __fastcall TTerminalManager::Instance(bool ForceCreation)
- {
- if (!FInstance && ForceCreation)
- {
- FInstance = new TTerminalManager();
- }
- return FInstance;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DestroyInstance()
- {
- DebugAssert(FInstance);
- SAFE_DESTROY(FInstance);
- }
- //---------------------------------------------------------------------------
- __fastcall TTerminalManager::TTerminalManager() :
- TTerminalList(Configuration)
- {
- FQueueSection = new TCriticalSection();
- FActiveTerminal = NULL;
- FScpExplorer = NULL;
- FDestroying = false;
- FTerminalPendingAction = tpNull;
- FDirectoryReadingStart = 0;
- FAuthenticateForm = NULL;
- FTaskbarList = NULL;
- FAuthenticating = 0;
- FMainThread = GetCurrentThreadId();
- FChangeSection.reset(new TCriticalSection());
- FPendingConfigurationChange = 0;
- FKeepAuthenticateForm = false;
- FApplicationsEvents.reset(new TApplicationEvents(Application));
- FApplicationsEvents->OnException = ApplicationException;
- FApplicationsEvents->OnShowHint = ApplicationShowHint;
- FApplicationsEvents->OnMessage = ApplicationMessage;
- FApplicationsEvents->OnModalBegin = ApplicationModalBegin;
- FApplicationsEvents->OnModalEnd = ApplicationModalEnd;
- DebugAssert(WinConfiguration->OnMasterPasswordPrompt == NULL);
- WinConfiguration->OnMasterPasswordPrompt = MasterPasswordPrompt;
- InitTaskbarButtonCreatedMessage();
- DebugAssert(Configuration && !Configuration->OnChange);
- Configuration->OnChange = ConfigurationChange;
- FOnLastTerminalClosed = NULL;
- FOnTerminalListChanged = NULL;
- FTerminalList = new TStringList();
- FQueues = new TList();
- FTerminationMessages = new TStringList();
- std::unique_ptr<TSessionData> DummyData(new TSessionData(L""));
- FLocalTerminal = CreateTerminal(DummyData.get());
- SetupTerminal(FLocalTerminal);
- }
- //---------------------------------------------------------------------------
- __fastcall TTerminalManager::~TTerminalManager()
- {
- FreeAll();
- DebugAssert(!ScpExplorer);
- DebugAssert(Configuration->OnChange == ConfigurationChange);
- Configuration->OnChange = NULL;
- FApplicationsEvents.reset(NULL);
- DebugAssert(WinConfiguration->OnMasterPasswordPrompt == MasterPasswordPrompt);
- WinConfiguration->OnMasterPasswordPrompt = NULL;
- delete FLocalTerminal;
- delete FQueues;
- delete FTerminationMessages;
- delete FTerminalList;
- CloseAutheticateForm();
- delete FQueueSection;
- ReleaseTaskbarList();
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SetQueueConfiguration(TTerminalQueue * Queue)
- {
- Queue->TransfersLimit = GUIConfiguration->QueueTransfersLimit;
- Queue->KeepDoneItemsFor =
- (GUIConfiguration->QueueKeepDoneItems ? GUIConfiguration->QueueKeepDoneItemsFor : 0);
- }
- //---------------------------------------------------------------------------
- TTerminalQueue * __fastcall TTerminalManager::NewQueue(TTerminal * Terminal)
- {
- TTerminalQueue * Queue = new TTerminalQueue(Terminal, Configuration);
- SetQueueConfiguration(Queue);
- Queue->Enabled = WinConfiguration->EnableQueueByDefault;
- Queue->OnQueryUser = TerminalQueryUser;
- Queue->OnPromptUser = TerminalPromptUser;
- Queue->OnShowExtendedException = TerminalShowExtendedException;
- Queue->OnEvent = QueueEvent;
- return Queue;
- }
- //---------------------------------------------------------------------------
- TTerminal * __fastcall TTerminalManager::CreateTerminal(TSessionData * Data)
- {
- return new TManagedTerminal(Data, Configuration);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SetupTerminal(TTerminal * Terminal)
- {
- Terminal->OnQueryUser = TerminalQueryUser;
- Terminal->OnPromptUser = TerminalPromptUser;
- Terminal->OnDisplayBanner = TerminalDisplayBanner;
- Terminal->OnShowExtendedException = TerminalShowExtendedException;
- Terminal->OnProgress = OperationProgress;
- Terminal->OnFinished = OperationFinished;
- Terminal->OnDeleteLocalFile = DeleteLocalFile;
- Terminal->OnReadDirectoryProgress = TerminalReadDirectoryProgress;
- Terminal->OnInformation = TerminalInformation;
- Terminal->OnCustomCommand = TerminalCustomCommand;
- }
- //---------------------------------------------------------------------------
- TTerminal * __fastcall TTerminalManager::DoNewTerminal(TSessionData * Data)
- {
- TTerminal * Terminal = TTerminalList::NewTerminal(Data);
- try
- {
- FQueues->Add(NewQueue(Terminal));
- FTerminationMessages->Add(L"");
- SetupTerminal(Terminal);
- }
- catch(...)
- {
- if (Terminal != NULL)
- {
- FreeTerminal(Terminal);
- }
- throw;
- }
- return Terminal;
- }
- //---------------------------------------------------------------------------
- TTerminal * __fastcall TTerminalManager::NewTerminal(TSessionData * Data)
- {
- TTerminal * Terminal = DoNewTerminal(Data);
- DoTerminalListChanged();
- return Terminal;
- }
- //---------------------------------------------------------------------------
- TTerminal * __fastcall TTerminalManager::NewTerminals(TList * DataList)
- {
- TTerminal * Result = NULL;
- for (int Index = 0; Index < DataList->Count; Index++)
- {
- TTerminal * Terminal =
- NewTerminal(reinterpret_cast<TSessionData *>(DataList->Items[Index]));
- if (Index == 0)
- {
- Result = Terminal;
- }
- }
- DoTerminalListChanged();
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::FreeActiveTerminal()
- {
- if (FTerminalPendingAction == tpNull)
- {
- DebugAssert(ActiveTerminal);
- FreeTerminal(ActiveTerminal);
- }
- else
- {
- DebugAssert(FTerminalPendingAction == ::tpNone);
- FTerminalPendingAction = tpFree;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DoConnectTerminal(TTerminal * Terminal, bool Reopen, bool AdHoc)
- {
- TManagedTerminal * ManagedTerminal = dynamic_cast<TManagedTerminal *>(Terminal);
- // it must be managed terminal, unless it is secondary terminal (of managed terminal)
- DebugAssert((ManagedTerminal != NULL) || (dynamic_cast<TSecondaryTerminal *>(Terminal) != NULL));
- // particularly when we are reconnecting RemoteDirectory of managed terminal
- // hold the last used remote directory as opposite to session data, which holds
- // the default remote directory.
- // make sure the last used directory is used, but the default is preserved too
- UnicodeString OrigRemoteDirectory = Terminal->SessionData->RemoteDirectory;
- try
- {
- TTerminalThread * TerminalThread = new TTerminalThread(Terminal);
- TerminalThread->AllowAbandon = (Terminal == FActiveTerminal);
- try
- {
- if (ManagedTerminal != NULL)
- {
- Terminal->SessionData->RemoteDirectory = ManagedTerminal->StateData->RemoteDirectory;
- if ((double)ManagedTerminal->ReopenStart == 0)
- {
- ManagedTerminal->ReopenStart = Now();
- }
- DebugAssert(ManagedTerminal->TerminalThread == NULL);
- ManagedTerminal->TerminalThread = TerminalThread;
- }
- TNotifyEvent OnIdle;
- ((TMethod*)&OnIdle)->Code = TerminalThreadIdle;
- TerminalThread->OnIdle = OnIdle;
- if (Reopen)
- {
- TerminalThread->TerminalReopen();
- }
- else
- {
- TerminalThread->TerminalOpen();
- }
- }
- __finally
- {
- TerminalThread->OnIdle = NULL;
- if (!TerminalThread->Release())
- {
- if (!AdHoc && (DebugAlwaysTrue(Terminal == FActiveTerminal)))
- {
- // terminal was abandoned, must create a new one to replace it
- Terminal = CreateTerminal(new TSessionData(L""));
- SetupTerminal(Terminal);
- OwnsObjects = false;
- Items[ActiveTerminalIndex] = Terminal;
- OwnsObjects = true;
- FActiveTerminal = Terminal;
- }
- // When abandoning cancelled terminal, DoInformation(Phase = 0) does not make it to TerminalInformation handler.
- if (DebugAlwaysTrue(FAuthenticating > 0))
- {
- FKeepAuthenticateForm = false;
- AuthenticatingDone();
- }
- }
- else
- {
- if (ManagedTerminal != NULL)
- {
- ManagedTerminal->TerminalThread = NULL;
- }
- }
- }
- }
- __finally
- {
- Terminal->SessionData->RemoteDirectory = OrigRemoteDirectory;
- if (Terminal->Active && (ManagedTerminal != NULL))
- {
- ManagedTerminal->ReopenStart = 0;
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::CloseAutheticateForm()
- {
- SAFE_DESTROY(FAuthenticateForm);
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::ConnectTerminal(TTerminal * Terminal)
- {
- bool Result = true;
- // were it an active terminal, it would allow abandoning, what this API cannot deal with
- DebugAssert(Terminal != FActiveTerminal);
- try
- {
- DoConnectTerminal(Terminal, false, false);
- }
- catch (Exception & E)
- {
- ShowExtendedExceptionEx(Terminal, &E);
- Result = false;
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalThreadIdle(void * /*Data*/, TObject * /*Sender*/)
- {
- Application->ProcessMessages();
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::ConnectActiveTerminalImpl(bool Reopen)
- {
- TTerminalPendingAction Action;
- bool Result;
- do
- {
- Action = tpNull;
- Result = false;
- try
- {
- DebugAssert(ActiveTerminal);
- DoConnectTerminal(ActiveTerminal, Reopen, false);
- if (ScpExplorer)
- {
- DebugAssert(ActiveTerminal->Status == ssOpened);
- TerminalReady();
- }
- WinConfiguration->ClearTemporaryLoginData();
- Result = true;
- }
- catch (Exception & E)
- {
- DebugAssert(FTerminalPendingAction == tpNull);
- FTerminalPendingAction = ::tpNone;
- try
- {
- DebugAssert(ActiveTerminal != NULL);
- ActiveTerminal->ShowExtendedException(&E);
- Action = FTerminalPendingAction;
- }
- __finally
- {
- FTerminalPendingAction = tpNull;
- }
- }
- }
- while (Action == tpReconnect);
- if (Action == tpFree)
- {
- FreeActiveTerminal();
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::ConnectActiveTerminal()
- {
- ActiveTerminal->CollectUsage();
- // add only stored sessions to the jump list,
- // ad-hoc session cannot be reproduced from just a session name
- if (StoredSessions->FindSame(ActiveTerminal->SessionData) != NULL)
- {
- WinConfiguration->AddSessionToJumpList(ActiveTerminal->SessionData->SessionName);
- }
- FAuthenticationCancelled = false;
- bool Result = ConnectActiveTerminalImpl(false);
- UnicodeString DateStamp = StandardDatestamp();
- if (Result)
- {
- if (Configuration->Usage->Get(L"OpenedSessionsFailedLastDate") == DateStamp)
- {
- Configuration->Usage->Inc(L"OpenedSessionsFailedRecovered");
- }
- }
- else
- {
- Configuration->Usage->Inc(L"OpenedSessionsFailed");
- Configuration->Usage->Set(L"OpenedSessionsFailedLastDate", DateStamp);
- if (FAuthenticationCancelled)
- {
- Configuration->Usage->Inc(L"OpenedSessionsFailedAfterCancel");
- }
- }
- if (Result && WinConfiguration->AutoOpenInPutty && CanOpenInPutty())
- {
- try
- {
- OpenInPutty();
- }
- catch(Exception & E)
- {
- ShowExtendedExceptionEx(NULL, &E);
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DisconnectActiveTerminal()
- {
- DebugAssert(ActiveTerminal);
- int Index = IndexOf(ActiveTerminal);
- TTerminalQueue * OldQueue;
- TTerminalQueue * NewQueue;
- OldQueue = reinterpret_cast<TTerminalQueue *>(FQueues->Items[Index]);
- NewQueue = this->NewQueue(ActiveTerminal);
- FQueues->Items[Index] = NewQueue;
- ScpExplorer->Queue = NewQueue;
- delete OldQueue;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ReconnectActiveTerminal()
- {
- DebugAssert(ActiveTerminal);
- if (ScpExplorer)
- {
- if (ScpExplorer->Terminal == ActiveTerminal)
- {
- ScpExplorer->UpdateTerminal(ActiveTerminal);
- }
- }
- try
- {
- if (FTerminalPendingAction == tpNull)
- {
- ConnectActiveTerminalImpl(true);
- }
- else
- {
- FTerminalPendingAction = tpReconnect;
- }
- }
- catch(...)
- {
- FreeActiveTerminal();
- throw;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::FreeAll()
- {
- FDestroying = true;
- try
- {
- while (Count)
- {
- FreeTerminal(Terminals[0]);
- }
- }
- __finally
- {
- FDestroying = false;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::FreeTerminal(TTerminal * Terminal)
- {
- try
- {
- // we want the Login dialog to open on auto-workspace name,
- // as set in TCustomScpExplorerForm::FormClose
- if (!FDestroying || !WinConfiguration->AutoSaveWorkspace)
- {
- if (StoredSessions->FindSame(Terminal->SessionData) != NULL)
- {
- WinConfiguration->LastStoredSession = Terminal->SessionData->Name;
- }
- }
- if (ScpExplorer != NULL)
- {
- ScpExplorer->TerminalRemoved(Terminal);
- }
- if (Terminal->Active)
- {
- Terminal->Close();
- }
- }
- __finally
- {
- int Index = IndexOf(Terminal);
- Extract(Terminal);
- TTerminalQueue * Queue;
- Queue = reinterpret_cast<TTerminalQueue *>(FQueues->Items[Index]);
- FQueues->Delete(Index);
- FTerminationMessages->Delete(Index);
- if (ActiveTerminal && (Terminal == ActiveTerminal))
- {
- if ((Count > 0) && !FDestroying)
- {
- ActiveTerminal = Terminals[Index < Count ? Index : Index - 1];
- }
- else
- {
- ActiveTerminal = NULL;
- }
- }
- else
- {
- SaveTerminal(Terminal);
- }
- // only now all references to/from queue (particularly events to explorer)
- // are cleared
- delete Queue;
- delete Terminal;
- DoTerminalListChanged();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SetScpExplorer(TCustomScpExplorerForm * value)
- {
- if (ScpExplorer != value)
- {
- // changing explorer is not supported yet
- DebugAssert(!ScpExplorer || !value);
- FScpExplorer = value;
- if (FScpExplorer)
- {
- FScpExplorer->Terminal = ActiveTerminal;
- FScpExplorer->Queue = ActiveQueue;
- FOnLastTerminalClosed = FScpExplorer->LastTerminalClosed;
- FOnTerminalListChanged = FScpExplorer->TerminalListChanged;
- UpdateTaskbarList();
- }
- else
- {
- FOnLastTerminalClosed = NULL;
- FOnTerminalListChanged = NULL;
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SetActiveTerminal(TTerminal * value)
- {
- DoSetActiveTerminal(value, false);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SetActiveTerminalWithAutoReconnect(TTerminal * value)
- {
- DoSetActiveTerminal(value, true);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DoSetActiveTerminal(TTerminal * value, bool AutoReconnect)
- {
- if (ActiveTerminal != value)
- {
- // here used to be call to TCustomScpExporer::UpdateSessionData (now UpdateTerminal)
- // but it seems to be duplicate to call from TCustomScpExporer::TerminalChanging
- TTerminal * PActiveTerminal = ActiveTerminal;
- FActiveTerminal = value;
- // moved from else block of next if (ActiveTerminal) statement
- // so ScpExplorer can update its caption
- UpdateAppTitle();
- if (ScpExplorer)
- {
- if (ActiveTerminal && (ActiveTerminal->Status == ssOpened))
- {
- TerminalReady();
- }
- else
- {
- ScpExplorer->Terminal = NULL;
- ScpExplorer->Queue = NULL;
- }
- }
- if (PActiveTerminal && !PActiveTerminal->Active)
- {
- SaveTerminal(PActiveTerminal);
- }
- if (ActiveTerminal)
- {
- int Index = ActiveTerminalIndex;
- if (!ActiveTerminal->Active && !FTerminationMessages->Strings[Index].IsEmpty())
- {
- UnicodeString Message = FTerminationMessages->Strings[Index];
- FTerminationMessages->Strings[Index] = L"";
- if (AutoReconnect)
- {
- ReconnectActiveTerminal();
- }
- else
- {
- Exception * E = new ESshFatal(NULL, Message);
- try
- {
- // finally show pending terminal message,
- // this gives user also possibility to reconnect
- ActiveTerminal->ShowExtendedException(E);
- }
- __finally
- {
- delete E;
- }
- }
- }
- }
- else
- {
- if (OnLastTerminalClosed)
- {
- OnLastTerminalClosed(this);
- }
- }
- if ((ActiveTerminal != NULL) && !ActiveTerminal->Active)
- {
- ConnectActiveTerminal();
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::QueueStatusUpdated()
- {
- UpdateAppTitle();
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::ShouldDisplayQueueStatusOnAppTitle()
- {
- bool Result = IsApplicationMinimized();
- if (!Result && (ScpExplorer != NULL))
- {
- HWND Window = GetActiveWindow();
- Window = GetAncestor(Window, GA_ROOTOWNER);
- Result = (ScpExplorer->Handle != Window);
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::UpdateAppTitle()
- {
- if (ScpExplorer)
- {
- UnicodeString NewTitle = FormatMainFormCaption(GetActiveTerminalTitle(false));
- UnicodeString QueueProgressTitle;
- if (!FForegroundProgressTitle.IsEmpty())
- {
- NewTitle = FForegroundProgressTitle + L" - " + NewTitle;
- }
- else if (!FProgressTitle.IsEmpty() && !ForegroundTask())
- {
- NewTitle = FProgressTitle + L" - " + NewTitle;
- }
- else if (ShouldDisplayQueueStatusOnAppTitle() &&
- !(QueueProgressTitle = ScpExplorer->GetQueueProgressTitle()).IsEmpty())
- {
- NewTitle = QueueProgressTitle + L" - " + NewTitle;
- }
- else if (ActiveTerminal && (ScpExplorer != NULL))
- {
- AddToList(NewTitle, ScpExplorer->PathForCaption(), L" - ");
- }
- ScpExplorer->Caption = NewTitle;
- ScpExplorer->ApplicationTitleChanged();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SaveTerminal(TTerminal * Terminal)
- {
- TSessionData * Data = StoredSessions->FindSame(Terminal->SessionData);
- if (Data != NULL)
- {
- TManagedTerminal * ManagedTerminal = dynamic_cast<TManagedTerminal *>(Terminal);
- DebugAssert(ManagedTerminal != NULL);
- bool Changed = false;
- if (Terminal->SessionData->UpdateDirectories)
- {
- Data->CopyDirectoriesStateData(ManagedTerminal->StateData);
- Changed = true;
- }
- if (Changed)
- {
- // modified only, implicit
- StoredSessions->Save(false, false);
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::HandleException(Exception * E)
- {
- // can be null for example when exception is thrown on login dialog
- if (ActiveTerminal != NULL)
- {
- ActiveTerminal->ShowExtendedException(E);
- }
- else
- {
- ShowExtendedException(E);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ApplicationException(TObject * /*Sender*/,
- Exception * E)
- {
- HandleException(E);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ApplicationShowHint(UnicodeString & HintStr,
- bool & /*CanShow*/, THintInfo & HintInfo)
- {
- HintInfo.HintData = HintInfo.HintControl;
- if (HasLabelHintPopup(HintInfo.HintControl, HintStr))
- {
- // Hack for transfer setting labels.
- // Should be converted to something like HintLabel()
- HintInfo.HideTimeout = 100000; // "almost" never
- }
- else if (dynamic_cast<TProgressBar *>(HintInfo.HintControl) != NULL)
- {
- // Hint is forcibly hidden in TProgressForm::FormHide
- HintInfo.HideTimeout = 100000; // "almost" never
- HintInfo.ReshowTimeout = 500; // updated each 0.5s
- }
- else
- {
- int HintMaxWidth = 300;
- TControl * ScaleControl = HintInfo.HintControl;
- if (DebugAlwaysFalse(HintInfo.HintControl == NULL) ||
- (GetParentForm(HintInfo.HintControl) == NULL))
- {
- ScaleControl = ScpExplorer;
- }
- HintMaxWidth = ScaleByTextHeight(ScaleControl, HintMaxWidth);
- HintInfo.HintMaxWidth = HintMaxWidth;
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::HandleMouseWheel(WPARAM WParam, LPARAM LParam)
- {
- // WORKAROUND This is no longer necessary on Windows 10 (except for WM_WANTS_MOUSEWHEEL_INACTIVE part)
- bool Result = false;
- if (Application->Active)
- {
- TPoint Point(LOWORD(LParam), HIWORD(LParam));
- TWinControl * Control = FindVCLWindow(Point);
- if (Control != NULL)
- {
- TCustomForm * Form = GetParentForm(Control);
- // Only case we expect the parent form to be NULL is on the Find/Replace dialog,
- // which is owned by VCL's internal TRedirectorWindow.
- DebugAssert((Form != NULL) || (Control->ClassName() == L"TRedirectorWindow"));
- if (Form != NULL)
- {
- // Send it only to windows we tested it with.
- // Though we should sooner or later remove this test and pass it to all our windows.
- if (Form->Active && (Form->Perform(WM_WANTS_MOUSEWHEEL, 0, 0) == 1))
- {
- SendMessage(Control->Handle, WM_MOUSEWHEEL, WParam, LParam);
- Result = true;
- }
- else if (!Form->Active && (Form->Perform(WM_WANTS_MOUSEWHEEL_INACTIVE, 0, 0) == 1))
- {
- TWinControl * Control2;
- // FindVCLWindow stops on window level, when the window is not active? or when there's a modal window over it?
- // (but in any case, when we have operation running on top of Synchronization checklist).
- // WORKAROUND: The while loop does what AllLevels parameter of ControlAtPos should do, but it's broken.
- // See http://qc.embarcadero.com/wc/qcmain.aspx?d=82143
- while ((Control2 = dynamic_cast<TWinControl *>(Control->ControlAtPos(Control->ScreenToClient(Point), false, true))) != NULL)
- {
- Control = Control2;
- }
- SendMessage(Control->Handle, WM_MOUSEWHEEL, WParam, LParam);
- Result = true;
- }
- }
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ApplicationMessage(TMsg & Msg, bool & Handled)
- {
- if (Msg.message == FTaskbarButtonCreatedMessage)
- {
- CreateTaskbarList();
- }
- if (Msg.message == WM_MOUSEWHEEL)
- {
- Handled = HandleMouseWheel(Msg.wParam, Msg.lParam);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ApplicationModalBegin(TObject * /*Sender*/)
- {
- NonVisualDataModule->StartBusy();
- if (ScpExplorer != NULL)
- {
- ScpExplorer->SuspendWindowLock();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ApplicationModalEnd(TObject * /*Sender*/)
- {
- NonVisualDataModule->EndBusy();
- if (ScpExplorer != NULL)
- {
- ScpExplorer->ResumeWindowLock();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::InitTaskbarButtonCreatedMessage()
- {
- // XE6 VCL already handles TaskbarButtonCreated, but does not call ChangeWindowMessageFilterEx.
- // So we keep our implementation.
- // See also https://stackoverflow.com/q/14614823/850848#14618587
- FTaskbarButtonCreatedMessage = RegisterWindowMessage(L"TaskbarButtonCreated");
- HINSTANCE User32Library = LoadLibrary(L"user32.dll");
- ChangeWindowMessageFilterExProc ChangeWindowMessageFilterEx =
- (ChangeWindowMessageFilterExProc)GetProcAddress(User32Library, "ChangeWindowMessageFilterEx");
- if (ChangeWindowMessageFilterEx != NULL)
- {
- // without this we won't get TaskbarButtonCreated, when app is running elevated
- ChangeWindowMessageFilterEx(
- Application->Handle, FTaskbarButtonCreatedMessage, MSGFLT_ALLOW, NULL);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::CreateTaskbarList()
- {
- ReleaseTaskbarList();
- if(SUCCEEDED(CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_ALL,
- IID_ITaskbarList3, (void **) &FTaskbarList)))
- {
- if (ScpExplorer != NULL)
- {
- UpdateTaskbarList();
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ReleaseTaskbarList()
- {
- if (FTaskbarList != NULL)
- {
- FTaskbarList->Release();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::UpdateTaskbarList()
- {
- ScpExplorer->UpdateTaskbarList(FTaskbarList);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DeleteLocalFile(const UnicodeString FileName, bool Alternative)
- {
- RecursiveDeleteFileChecked(FileName, (WinConfiguration->DeleteToRecycleBin != Alternative));
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalQueryUser(TObject * Sender,
- const UnicodeString Query, TStrings * MoreMessages, unsigned int Answers,
- const TQueryParams * Params, unsigned int & Answer, TQueryType Type, void * /*Arg*/)
- {
- UnicodeString HelpKeyword;
- TMessageParams MessageParams(Params);
- UnicodeString AQuery = Query;
- if (Params != NULL)
- {
- HelpKeyword = Params->HelpKeyword;
- if (FLAGSET(Params->Params, qpFatalAbort))
- {
- AQuery = FMTLOAD(WARN_FATAL_ERROR, (AQuery));
- if (!MessageParams.TimerMessage.IsEmpty())
- {
- MessageParams.TimerMessage = FMTLOAD(WARN_FATAL_ERROR, (MessageParams.TimerMessage));
- }
- }
- }
- if (ScpExplorer)
- {
- Answer = ScpExplorer->MoreMessageDialog(AQuery, MoreMessages, Type, Answers,
- HelpKeyword, &MessageParams, dynamic_cast<TTerminal *>(Sender));
- }
- else
- {
- Answer = MoreMessageDialog(AQuery, MoreMessages, Type, Answers, HelpKeyword,
- &MessageParams);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::AuthenticateFormCancel(TObject * Sender)
- {
- TAuthenticateForm * Form = dynamic_cast<TAuthenticateForm *>(Sender);
- DebugAssert(Form != NULL);
- TManagedTerminal * ManagedTerminal = dynamic_cast<TManagedTerminal *>(Form->Terminal);
- // will be null e.g. for background transfers
- if (ManagedTerminal != NULL)
- {
- TTerminalThread * TerminalThread = ManagedTerminal->TerminalThread;
- // can be NULL for reconnects from transfers
- if ((TerminalThread != NULL) && !TerminalThread->Cancelling)
- {
- Form->Log(LoadStr(AUTH_CANCELLING));
- TerminalThread->Cancel();
- }
- }
- FAuthenticationCancelled = true;
- }
- //---------------------------------------------------------------------------
- TAuthenticateForm * __fastcall TTerminalManager::MakeAuthenticateForm(
- TTerminal * Terminal)
- {
- TAuthenticateForm * Dialog = SafeFormCreate<TAuthenticateForm>();
- Dialog->Init(Terminal);
- DebugAssert(Dialog->OnCancel == NULL);
- Dialog->OnCancel = AuthenticateFormCancel;
- return Dialog;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::FileNameInputDialogInitializeRenameBaseName(
- TObject * /*Sender*/, TInputDialogData * Data)
- {
- EditSelectBaseName(Data->Edit->Handle);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalPromptUser(
- TTerminal * Terminal, TPromptKind Kind, UnicodeString Name, UnicodeString Instructions,
- TStrings * Prompts, TStrings * Results, bool & Result, void * /*Arg*/)
- {
- if (((Kind == pkPrompt) || (Kind == pkFileName)) && (FAuthenticateForm == NULL) &&
- (Terminal->Status != ssOpening))
- {
- DebugAssert(Instructions.IsEmpty());
- DebugAssert(Prompts->Count == 1);
- DebugAssert(FLAGSET(int(Prompts->Objects[0]), pupEcho));
- UnicodeString AResult = Results->Strings[0];
- TInputDialogInitialize InputDialogInitialize = NULL;
- if ((Kind == pkFileName) && !WinConfiguration->RenameWholeName)
- {
- InputDialogInitialize = FileNameInputDialogInitializeRenameBaseName;
- }
- Result = InputDialog(Name, Prompts->Strings[0], AResult, L"", NULL, false, InputDialogInitialize);
- if (Result)
- {
- Results->Strings[0] = AResult;
- }
- }
- else
- {
- TAuthenticateForm * AuthenticateForm = FAuthenticateForm;
- if (AuthenticateForm == NULL)
- {
- AuthenticateForm = MakeAuthenticateForm(Terminal);
- }
- try
- {
- Result = AuthenticateForm->PromptUser(Kind, Name, Instructions, Prompts, Results,
- (FAuthenticateForm != NULL), Terminal->StoredCredentialsTried);
- }
- __finally
- {
- if (FAuthenticateForm == NULL)
- {
- delete AuthenticateForm;
- }
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalDisplayBanner(
- TTerminal * Terminal, UnicodeString SessionName,
- const UnicodeString & Banner, bool & NeverShowAgain, int Options, unsigned int & Params)
- {
- DebugAssert(FAuthenticateForm != NULL);
- TAuthenticateForm * AuthenticateForm = FAuthenticateForm;
- if (AuthenticateForm == NULL)
- {
- AuthenticateForm = MakeAuthenticateForm(Terminal);
- }
- try
- {
- AuthenticateForm->Banner(Banner, NeverShowAgain, Options, Params);
- }
- __finally
- {
- if (FAuthenticateForm == NULL)
- {
- delete AuthenticateForm;
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalShowExtendedException(
- TTerminal * Terminal, Exception * E, void * /*Arg*/)
- {
- if (ScpExplorer)
- {
- ScpExplorer->ShowExtendedException(Terminal, E);
- }
- else
- {
- ShowExtendedExceptionEx(Terminal, E);
- }
- }
- //---------------------------------------------------------------------------
- static TDateTime DirectoryReadingProgressDelay(0, 0, 1, 500);
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalReadDirectoryProgress(
- TObject * /*Sender*/, int Progress, int ResolvedLinks, bool & Cancel)
- {
- if (Progress == 0)
- {
- if (ScpExplorer != NULL)
- {
- // See also TCustomScpExplorerForm::RemoteDirViewBusy
- ScpExplorer->LockWindow();
- }
- FDirectoryReadingStart = Now();
- if (!FProgressTitle.IsEmpty() || !FForegroundProgressTitle.IsEmpty())
- {
- FProgressTitle = L"";
- FForegroundProgressTitle = L"";
- UpdateAppTitle();
- }
- // Reset "was ESC ever pressed?" state
- GetAsyncKeyState(VK_ESCAPE);
- }
- else if (Progress < 0)
- {
- if (Progress == -2)
- {
- // cancelled
- if (ScpExplorer != NULL)
- {
- ScpExplorer->ReadDirectoryCancelled();
- }
- }
- else
- {
- if (ScpExplorer != NULL)
- {
- ScpExplorer->UnlockWindow();
- }
- FProgressTitle = L"";
- FForegroundProgressTitle = L"";
- UpdateAppTitle();
- }
- }
- else
- {
- // If the least significant bit is set,
- // the key was pressed after the previous call to GetAsyncKeyState.
- int KeyState = GetAsyncKeyState(VK_ESCAPE);
- if (FLAGSET(KeyState, 0x01))
- {
- Cancel = true;
- }
- if ((Now() - FDirectoryReadingStart) >= DirectoryReadingProgressDelay)
- {
- // 4 is arbitrary number
- FForegroundProgressTitle =
- FMTLOAD(ResolvedLinks >= 4 ? DIRECTORY_READING_AND_RESOLVING_PROGRESS : DIRECTORY_READING_PROGRESS,
- (Progress));
- UpdateAppTitle();
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalCustomCommand(
- TTerminal * /*Terminal*/, const UnicodeString & Command, bool & Handled)
- {
- // Implementation has to be thread-safe
- Handled = CopyCommandToClipboard(Command);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::AuthenticatingDone()
- {
- FAuthenticating--;
- if (FAuthenticating == 0)
- {
- BusyEnd(FBusyToken);
- FBusyToken = NULL;
- }
- if (!FKeepAuthenticateForm)
- {
- CloseAutheticateForm();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalInformation(
- TTerminal * Terminal, const UnicodeString & Str, bool /*Status*/, int Phase)
- {
- if (Phase == 1)
- {
- if (FAuthenticating == 0)
- {
- FBusyToken = BusyStart();
- }
- FAuthenticating++;
- }
- else if (Phase == 0)
- {
- DebugAssert(FAuthenticating > 0);
- AuthenticatingDone();
- }
- else
- {
- if (FAuthenticating > 0)
- {
- bool ShowPending = false;
- if (FAuthenticateForm == NULL)
- {
- FAuthenticateForm = MakeAuthenticateForm(Terminal);
- ShowPending = true;
- }
- FAuthenticateForm->Log(Str);
- if (ShowPending)
- {
- FAuthenticateForm->ShowAsModal();
- }
- }
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::OperationFinished(::TFileOperation Operation,
- TOperationSide Side, bool Temp, const UnicodeString & FileName, bool Success,
- TOnceDoneOperation & OnceDoneOperation)
- {
- DebugAssert(ScpExplorer);
- ScpExplorer->OperationFinished(Operation, Side, Temp, FileName, Success,
- OnceDoneOperation);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::OperationProgress(
- TFileOperationProgressType & ProgressData)
- {
- if (ProgressData.InProgress)
- {
- FProgressTitle = ScpExplorer->GetProgressTitle(ProgressData);
- }
- else
- {
- FProgressTitle = L"";
- }
- UpdateAppTitle();
- DebugAssert(ScpExplorer);
- ScpExplorer->OperationProgress(ProgressData);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::QueueEvent(TTerminalQueue * Queue, TQueueEvent Event)
- {
- TGuard Guard(FQueueSection);
- FQueueEvents.push_back(std::make_pair(Queue, Event));
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DoConfigurationChange()
- {
- DebugAssert(Configuration);
- DebugAssert(Configuration == WinConfiguration);
- TTerminalQueue * Queue;
- for (int Index = 0; Index < Count; Index++)
- {
- DebugAssert(Terminals[Index]->Log);
- Terminals[Index]->Log->ReflectSettings();
- Terminals[Index]->ActionLog->ReflectSettings();
- Queue = reinterpret_cast<TTerminalQueue *>(FQueues->Items[Index]);
- SetQueueConfiguration(Queue);
- }
- if (ScpExplorer)
- {
- ScpExplorer->ConfigurationChanged();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::ConfigurationChange(TObject * /*Sender*/)
- {
- if (FMainThread == GetCurrentThreadId())
- {
- DoConfigurationChange();
- }
- else
- {
- TGuard Guard(FChangeSection.get());
- FPendingConfigurationChange++;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::TerminalReady()
- {
- ScpExplorer->Terminal = ActiveTerminal;
- ScpExplorer->Queue = ActiveQueue;
- ScpExplorer->TerminalReady();
- }
- //---------------------------------------------------------------------------
- TStrings * __fastcall TTerminalManager::GetTerminalList()
- {
- FTerminalList->Clear();
- for (int i = 0; i < Count; i++)
- {
- TTerminal * Terminal = Terminals[i];
- UnicodeString Name = GetTerminalTitle(Terminal, true);
- FTerminalList->AddObject(Name, Terminal);
- }
- return FTerminalList;
- }
- //---------------------------------------------------------------------------
- int __fastcall TTerminalManager::GetActiveTerminalIndex()
- {
- return ActiveTerminal ? IndexOf(ActiveTerminal) : -1;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SetActiveTerminalIndex(int value)
- {
- ActiveTerminal = Terminals[value];
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall TTerminalManager::GetTerminalShortPath(TTerminal * Terminal)
- {
- UnicodeString Result = UnixExtractFileName(Terminal->CurrentDirectory);
- if (Result.IsEmpty())
- {
- Result = Terminal->CurrentDirectory;
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall TTerminalManager::GetTerminalTitle(TTerminal * Terminal, bool Unique)
- {
- UnicodeString Result = Terminal->SessionData->SessionName;
- if (Unique)
- {
- int Index = IndexOf(Terminal);
- // not for background transfer sessions
- if (Index >= 0)
- {
- for (int Index2 = 0; Index2 < Count; Index2++)
- {
- UnicodeString Name = Terminals[Index2]->SessionData->SessionName;
- if ((Terminals[Index2] != Terminal) &&
- SameText(Name, Result))
- {
- UnicodeString Path = GetTerminalShortPath(Terminal);
- if (!Path.IsEmpty())
- {
- Result = FORMAT(L"%s (%s)", (Result, Path));
- }
- break;
- }
- }
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- UnicodeString __fastcall TTerminalManager::GetActiveTerminalTitle(bool Unique)
- {
- UnicodeString Result;
- if (ActiveTerminal != NULL)
- {
- Result = GetTerminalTitle(ActiveTerminal, Unique);
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- TTerminalQueue * __fastcall TTerminalManager::GetActiveQueue()
- {
- TTerminalQueue * Result = NULL;
- if (ActiveTerminal != NULL)
- {
- Result = reinterpret_cast<TTerminalQueue *>(FQueues->Items[ActiveTerminalIndex]);
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::CycleTerminals(bool Forward)
- {
- if (Count > 0)
- {
- int Index = ActiveTerminalIndex;
- Index += Forward ? 1 : -1;
- if (Index < 0)
- {
- Index = Count-1;
- }
- else if (Index >= Count)
- {
- Index = 0;
- }
- ActiveTerminalIndex = Index;
- }
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::CanOpenInPutty()
- {
- return (ActiveTerminal != NULL) && !GUIConfiguration->PuttyPath.Trim().IsEmpty();
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::UpdateSessionCredentials(TSessionData * Data)
- {
- Data->UserName = ActiveTerminal->UserName;
- Data->Password = ActiveTerminal->Password;
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::OpenInPutty()
- {
- Configuration->Usage->Inc(L"OpenInPutty");
- TSessionData * Data = NULL;
- try
- {
- // Is NULL on the first session when called from ConnectActiveTerminal()
- // due to WinConfiguration->AutoOpenInPutty
- if (ScpExplorer != NULL)
- {
- Data = ScpExplorer->CloneCurrentSessionData();
- }
- else
- {
- Data = new TSessionData(L"");
- DebugAssert(ActiveTerminal != NULL);
- Data->Assign(ActiveTerminal->SessionData);
- UpdateSessionCredentials(Data);
- }
- // putty does not support resolving environment variables in session settings
- Data->ExpandEnvironmentVariables();
- if (ActiveTerminal->TunnelLocalPortNumber != 0)
- {
- Data->ConfigureTunnel(ActiveTerminal->TunnelLocalPortNumber);
- }
- OpenSessionInPutty(GUIConfiguration->PuttyPath, Data);
- }
- __finally
- {
- delete Data;
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::NewSession(bool /*FromSite*/, const UnicodeString & SessionUrl, bool ReloadSessions, TForm * LinkedForm)
- {
- if (ReloadSessions)
- {
- StoredSessions->Reload();
- }
- UnicodeString DownloadFile; // unused
- std::unique_ptr<TObjectList> DataList(new TObjectList());
- GetLoginData(SessionUrl, NULL, DataList.get(), DownloadFile, true, LinkedForm);
- if (DataList->Count > 0)
- {
- ActiveTerminal = NewTerminals(DataList.get());
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::Idle(bool SkipCurrentTerminal)
- {
- if (FPendingConfigurationChange > 0) // optimization
- {
- bool Changed = false;
- {
- TGuard Guard(FChangeSection.get());
- if (DebugAlwaysTrue(FPendingConfigurationChange > 0))
- {
- FPendingConfigurationChange--;
- Changed = true;
- }
- }
- if (Changed)
- {
- DoConfigurationChange();
- }
- }
- for (int Index = 0; Index < Count; Index++)
- {
- TTerminal * Terminal = Terminals[Index];
- try
- {
- if (!SkipCurrentTerminal || (Terminal != ActiveTerminal))
- {
- TManagedTerminal * ManagedTerminal = dynamic_cast<TManagedTerminal *>(Terminal);
- DebugAssert(ManagedTerminal != NULL);
- // make sure Idle is called on the thread that runs the terminal
- if (ManagedTerminal->TerminalThread != NULL)
- {
- ManagedTerminal->TerminalThread->Idle();
- }
- else
- {
- if (Terminal->Active)
- {
- Terminal->Idle();
- }
- }
- if (Terminal->Active)
- {
- DebugAssert(Index < FQueues->Count);
- if (Index < FQueues->Count)
- {
- reinterpret_cast<TTerminalQueue *>(FQueues->Items[Index])->Idle();
- }
- }
- }
- }
- catch(Exception & E)
- {
- if (Terminal == ActiveTerminal)
- {
- // throw further, so that the exception is handled in proper place
- // (particularly in broken-transfer reconnect handler, bug 72)
- throw;
- }
- else
- {
- // we may not have inactive terminal, unless there is a explorer,
- // also Idle is called from explorer anyway
- DebugAssert(ScpExplorer != NULL);
- if (ScpExplorer != NULL)
- {
- ScpExplorer->InactiveTerminalException(Terminal, &E);
- }
- if (!Terminal->Active)
- {
- // if session is lost, save the error message and rethrow it
- // once the terminal gets activated
- FTerminationMessages->Strings[Index] = E.Message;
- }
- }
- }
- }
- TTerminalQueue * QueueWithEvent;
- TQueueEvent QueueEvent;
- do
- {
- QueueWithEvent = NULL;
- {
- TGuard Guard(FQueueSection);
- if (!FQueueEvents.empty())
- {
- QueueWithEvent = FQueueEvents[0].first;
- QueueEvent = FQueueEvents[0].second;
- FQueueEvents.erase(FQueueEvents.begin());
- }
- }
- if (QueueWithEvent != NULL)
- {
- int Index = FQueues->IndexOf(QueueWithEvent);
- // the session may not exist anymore
- if (Index >= 0)
- {
- TTerminal * Terminal = Terminals[Index];
- // we can hardly have a queue event without explorer
- DebugAssert(ScpExplorer != NULL);
- if (ScpExplorer != NULL)
- {
- ScpExplorer->QueueEvent(Terminal, QueueWithEvent, QueueEvent);
- }
- }
- }
- }
- while (QueueWithEvent != NULL);
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::MasterPasswordPrompt()
- {
- if (GetCurrentThreadId() == MainThreadID)
- {
- if (!DoMasterPasswordDialog())
- {
- Abort();
- }
- }
- else
- {
- // this can happen only when we keep cancelling all master password prompts
- // as long as the sessing finally connects (session password has to be
- // explictly typed in), and background transfer is started
- Abort();
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::Move(TTerminal * Source, TTerminal * Target)
- {
- int SourceIndex = IndexOf(Source);
- int TargetIndex = IndexOf(Target);
- TTerminalList::Move(SourceIndex, TargetIndex);
- FQueues->Move(SourceIndex, TargetIndex);
- DoTerminalListChanged();
- // when there are indexed sessions with the same name,
- // the index may change when reordering the sessions
- UpdateAppTitle();
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::DoTerminalListChanged()
- {
- if (OnTerminalListChanged)
- {
- OnTerminalListChanged(this);
- }
- }
- //---------------------------------------------------------------------------
- void __fastcall TTerminalManager::SaveWorkspace(TList * DataList)
- {
- for (int Index = 0; Index < Count; Index++)
- {
- TManagedTerminal * ManagedTerminal = dynamic_cast<TManagedTerminal *>(Terminals[Index]);
- TSessionData * Data = StoredSessions->SaveWorkspaceData(ManagedTerminal->StateData);
- DataList->Add(Data);
- Data->Name = IntToHex(Index, 4);
- }
- }
- //---------------------------------------------------------------------------
- TTerminal * __fastcall TTerminalManager::FindActiveTerminalForSite(TSessionData * Data)
- {
- TTerminal * Result = NULL;
- for (int Index = 0; (Result == NULL) && (Index < Count); Index++)
- {
- TTerminal * Terminal = Terminals[Index];
- if (Terminal->Active &&
- Terminal->SessionData->IsSameSite(Data))
- {
- Result = Terminal;
- }
- }
- return Result;
- }
- //---------------------------------------------------------------------------
- TTerminalQueue * __fastcall TTerminalManager::FindQueueForTerminal(TTerminal * Terminal)
- {
- int Index = IndexOf(Terminal);
- return reinterpret_cast<TTerminalQueue *>(FQueues->Items[Index]);
- }
- //---------------------------------------------------------------------------
- TRemoteFile * __fastcall TTerminalManager::CheckRights(
- TTerminal * Terminal, const UnicodeString & EntryType, const UnicodeString & FileName, bool & WrongRights)
- {
- std::unique_ptr<TRemoteFile> FileOwner;
- TRemoteFile * File;
- try
- {
- Terminal->LogEvent(FORMAT(L"Checking %s \"%s\"...", (LowerCase(EntryType), FileName)));
- Terminal->ReadFile(FileName, File);
- FileOwner.reset(File);
- int ForbiddenRights = TRights::rfGroupWrite | TRights::rfOtherWrite;
- if ((File->Rights->Number & ForbiddenRights) != 0)
- {
- Terminal->LogEvent(FORMAT(L"%s \"%s\" exists, but has incorrect permissions %s.", (EntryType, FileName, File->Rights->Octal)));
- WrongRights = true;
- }
- else
- {
- Terminal->LogEvent(FORMAT(L"%s \"%s\" exists and has correct permissions %s.", (EntryType, FileName, File->Rights->Octal)));
- }
- }
- catch (Exception & E)
- {
- }
- return FileOwner.release();
- }
- //---------------------------------------------------------------------------
- bool __fastcall TTerminalManager::UploadPublicKey(
- TTerminal * Terminal, TSessionData * Data, UnicodeString & FileName)
- {
- std::unique_ptr<TOpenDialog> OpenDialog(new TOpenDialog(Application));
- OpenDialog->Title = LoadStr(LOGIN_PUBLIC_KEY_TITLE);
- OpenDialog->Filter = LoadStr(LOGIN_PUBLIC_KEY_FILTER);
- OpenDialog->DefaultExt = PuttyKeyExt;
- OpenDialog->FileName = FileName;
- bool Result = OpenDialog->Execute();
- if (Result)
- {
- Configuration->Usage->Inc(L"PublicKeyInstallation");
- FileName = OpenDialog->FileName;
- bool AutoReadDirectory;
- bool ExceptionOnFail;
- UnicodeString TemporaryDir;
- bool WrongRights = false;
- const UnicodeString SshFolder = L".ssh";
- const UnicodeString AuthorizedKeysFile = L"authorized_keys";
- UnicodeString AuthorizedKeysFilePath = FORMAT(L"%s/%s", (SshFolder, AuthorizedKeysFile));
- VerifyAndConvertKey(FileName, ssh2only, false);
- UnicodeString Comment;
- UnicodeString Line = GetPublicKeyLine(FileName, Comment);
- bool AdHocTerminal = (Terminal == NULL);
- std::unique_ptr<TTerminal> TerminalOwner;
- if (AdHocTerminal)
- {
- DebugAssert(Data != NULL);
- TAutoFlag KeepAuthenticateFormFlag(FKeepAuthenticateForm);
- try
- {
- TerminalOwner.reset(CreateTerminal(Data));
- Terminal = TerminalOwner.get();
- SetupTerminal(Terminal);
- Terminal->OnProgress = NULL;
- Terminal->OnFinished = NULL;
- DoConnectTerminal(Terminal, false, true);
- }
- catch (Exception & E)
- {
- CloseAutheticateForm();
- throw;
- }
- }
- AutoReadDirectory = Terminal->AutoReadDirectory;
- ExceptionOnFail = Terminal->ExceptionOnFail;
- try
- {
- Terminal->AutoReadDirectory = false;
- Terminal->ExceptionOnFail = true;
- UnicodeString SshImplementation = Terminal->GetSessionInfo().SshImplementation;
- UnicodeString NotOpenSSHMessage = FMTLOAD(LOGIN_NOT_OPENSSH, (SshImplementation));
- if (IsOpenSSH(SshImplementation) ||
- (MessageDialog(NotOpenSSHMessage, qtConfirmation, qaOK | qaCancel, HELP_LOGIN_AUTHORIZED_KEYS) == qaOK))
- {
- Terminal->Log->AddSeparator();
- Terminal->LogEvent(FORMAT(L"Adding public key line to \"%s\" file:\n%s", (AuthorizedKeysFilePath, Line)));
- // Ad-hoc terminal
- if (FAuthenticateForm != NULL)
- {
- FAuthenticateForm->Log(FMTLOAD(LOGIN_PUBLIC_KEY_UPLOAD, (Comment)));
- }
- UnicodeString SshFolderAbsolutePath = UnixIncludeTrailingBackslash(Terminal->GetHomeDirectory()) + SshFolder;
- std::unique_ptr<TRemoteFile> SshFolderFile(CheckRights(Terminal, L"Folder", SshFolderAbsolutePath, WrongRights));
- if (SshFolderFile.get() == NULL)
- {
- TRights SshFolderRights;
- SshFolderRights.Number = TRights::rfUserRead | TRights::rfUserWrite | TRights::rfUserExec;
- TRemoteProperties SshFolderProperties;
- SshFolderProperties.Rights = SshFolderRights;
- SshFolderProperties.Valid = TValidProperties() << vpRights;
- Terminal->LogEvent(FORMAT(L"Trying to create \"%s\" folder with permissions %s...", (SshFolder, SshFolderRights.Octal)));
- Terminal->CreateDirectory(SshFolderAbsolutePath, &SshFolderProperties);
- }
- TemporaryDir = ExcludeTrailingBackslash(WinConfiguration->TemporaryDir());
- if (!ForceDirectories(ApiPath(TemporaryDir)))
- {
- throw EOSExtException(FMTLOAD(CREATE_TEMP_DIR_ERROR, (TemporaryDir)));
- }
- UnicodeString TemporaryAuthorizedKeysFile = IncludeTrailingBackslash(TemporaryDir) + AuthorizedKeysFile;
- UnicodeString AuthorizedKeysFileAbsolutePath = UnixIncludeTrailingBackslash(SshFolderAbsolutePath) + AuthorizedKeysFile;
- bool Updated = true;
- TCopyParamType CopyParam; // Use factory defaults
- CopyParam.ResumeSupport = rsOff; // not to break the permissions
- CopyParam.PreserveTime = false; // not needed
- UnicodeString AuthorizedKeys;
- std::unique_ptr<TRemoteFile> AuthorizedKeysFileFile(CheckRights(Terminal, L"File", AuthorizedKeysFileAbsolutePath, WrongRights));
- if (AuthorizedKeysFileFile.get() != NULL)
- {
- AuthorizedKeysFileFile->FullFileName = AuthorizedKeysFileAbsolutePath;
- std::unique_ptr<TStrings> Files(new TStringList());
- Files->AddObject(AuthorizedKeysFileAbsolutePath, AuthorizedKeysFileFile.get());
- Terminal->LogEvent(FORMAT(L"Downloading current \"%s\" file...", (AuthorizedKeysFile)));
- Terminal->CopyToLocal(Files.get(), TemporaryDir, &CopyParam, cpNoConfirmation, NULL);
- // Overload with Encoding parameter work incorrectly, when used on a file without BOM
- AuthorizedKeys = TFile::ReadAllText(TemporaryAuthorizedKeysFile);
- std::unique_ptr<TStrings> AuthorizedKeysLines(TextToStringList(AuthorizedKeys));
- int P = Line.Pos(L" ");
- if (DebugAlwaysTrue(P > 0))
- {
- P = PosEx(L" ", Line, P + 1);
- }
- UnicodeString Prefix = Line.SubString(1, P); // including the space
- for (int Index = 0; Index < AuthorizedKeysLines->Count; Index++)
- {
- if (StartsStr(Prefix, AuthorizedKeysLines->Strings[Index]))
- {
- Terminal->LogEvent(FORMAT(L"\"%s\" file already contains public key line:\n%s", (AuthorizedKeysFile, AuthorizedKeysLines->Strings[Index])));
- Updated = false;
- }
- }
- if (Updated)
- {
- Terminal->LogEvent(FORMAT(L"\"%s\" file does not contain the public key line yet.", (AuthorizedKeysFile)));
- if (!EndsStr(L"\n", AuthorizedKeys))
- {
- Terminal->LogEvent(FORMAT(L"Adding missing trailing new line to \"%s\" file...", (AuthorizedKeysFile)));
- AuthorizedKeys += L"\n";
- }
- }
- }
- else
- {
- Terminal->LogEvent(FORMAT(L"Creating new \"%s\" file...", (AuthorizedKeysFile)));
- CopyParam.PreserveRights = true;
- CopyParam.Rights.Number = TRights::rfUserRead | TRights::rfUserWrite;
- }
- if (Updated)
- {
- AuthorizedKeys += Line + L"\n";
- // Overload without Encoding parameter uses TEncoding::UTF8, but does not write BOM, what we want
- TFile::WriteAllText(TemporaryAuthorizedKeysFile, AuthorizedKeys);
- std::unique_ptr<TStrings> Files(new TStringList());
- Files->Add(TemporaryAuthorizedKeysFile);
- Terminal->LogEvent(FORMAT(L"Uploading updated \"%s\" file...", (AuthorizedKeysFile)));
- Terminal->CopyToRemote(Files.get(), SshFolderAbsolutePath, &CopyParam, cpNoConfirmation, NULL);
- }
- }
- }
- __finally
- {
- Terminal->AutoReadDirectory = AutoReadDirectory;
- Terminal->ExceptionOnFail = ExceptionOnFail;
- if (!TemporaryDir.IsEmpty())
- {
- RecursiveDeleteFile(ExcludeTrailingBackslash(TemporaryDir), false);
- }
- CloseAutheticateForm(); // When uploading from Login dialog
- }
- Terminal->LogEvent(L"Public key installation done.");
- if (AdHocTerminal)
- {
- TerminalOwner.reset(NULL);
- }
- else
- {
- Terminal->Log->AddSeparator();
- }
- UnicodeString Message = FMTLOAD(LOGIN_PUBLIC_KEY_UPLOADED, (Comment));
- if (WrongRights)
- {
- Message += L"\n\n" + FMTLOAD(LOGIN_PUBLIC_KEY_PERMISSIONS, (AuthorizedKeysFilePath));
- }
- MessageDialog(Message, qtInformation, qaOK, HELP_LOGIN_AUTHORIZED_KEYS);
- }
- return Result;
- }
|