FileOperationProgress.cpp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874
  1. //---------------------------------------------------------------------------
  2. #include <vcl.h>
  3. #pragma hdrstop
  4. #include "Common.h"
  5. #include "FileOperationProgress.h"
  6. #include "CoreMain.h"
  7. //---------------------------------------------------------------------------
  8. #define TRANSFER_BUF_SIZE 32768
  9. //---------------------------------------------------------------------------
  10. __fastcall TFileOperationProgressType::TFileOperationProgressType()
  11. {
  12. FOnProgress = NULL;
  13. FOnFinished = NULL;
  14. FParent = NULL;
  15. Init();
  16. Clear();
  17. }
  18. //---------------------------------------------------------------------------
  19. __fastcall TFileOperationProgressType::TFileOperationProgressType(
  20. TFileOperationProgressEvent AOnProgress, TFileOperationFinished AOnFinished,
  21. TFileOperationProgressType * Parent)
  22. {
  23. FOnProgress = AOnProgress;
  24. FOnFinished = AOnFinished;
  25. FParent = Parent;
  26. FReset = false;
  27. Init();
  28. Clear();
  29. }
  30. //---------------------------------------------------------------------------
  31. __fastcall TFileOperationProgressType::~TFileOperationProgressType()
  32. {
  33. DebugAssert(!InProgress || FReset);
  34. DebugAssert(!Suspended || FReset);
  35. SAFE_DESTROY(FSection);
  36. SAFE_DESTROY(FUserSelectionsSection);
  37. }
  38. //---------------------------------------------------------------------------
  39. void __fastcall TFileOperationProgressType::Init()
  40. {
  41. FSection = new TCriticalSection();
  42. FUserSelectionsSection = new TCriticalSection();
  43. }
  44. //---------------------------------------------------------------------------
  45. void __fastcall TFileOperationProgressType::Assign(const TFileOperationProgressType & Other)
  46. {
  47. TValueRestorer<TCriticalSection *> SectionRestorer(FSection);
  48. TValueRestorer<TCriticalSection *> UserSelectionsSectionRestorer(FUserSelectionsSection);
  49. TGuard Guard(FSection);
  50. TGuard OtherGuard(Other.FSection);
  51. *this = Other;
  52. }
  53. //---------------------------------------------------------------------------
  54. void __fastcall TFileOperationProgressType::AssignButKeepSuspendState(const TFileOperationProgressType & Other)
  55. {
  56. TGuard Guard(FSection);
  57. TValueRestorer<unsigned int> SuspendTimeRestorer(FSuspendTime);
  58. TValueRestorer<bool> SuspendedRestorer(FSuspended);
  59. Assign(Other);
  60. }
  61. //---------------------------------------------------------------------------
  62. void __fastcall TFileOperationProgressType::Clear()
  63. {
  64. FFileName = L"";
  65. FFullFileName = L"";
  66. FDirectory = L"";
  67. FAsciiTransfer = false;
  68. FCount = -1;
  69. FFilesFinished = 0;
  70. FFilesFinishedSuccessfully = 0;
  71. FStartTime = Now();
  72. FSuspended = false;
  73. FSuspendTime = 0;
  74. FInProgress = false;
  75. FDone = false;
  76. FFileInProgress = false;
  77. FTotalTransferred = 0;
  78. FTotalSkipped = 0;
  79. FTotalSize = 0;
  80. FSkippedSize = 0;
  81. FTotalSizeSet = false;
  82. FOperation = foNone;
  83. FTemp = false;
  84. FSkipToAll = false;
  85. FBatchOverwrite = boNo;
  86. // to bypass check in ClearTransfer()
  87. FTransferSize = 0;
  88. FCPSLimit = 0;
  89. FTicks.clear();
  90. FTotalTransferredThen.clear();
  91. FCounterSet = false;
  92. ClearTransfer();
  93. }
  94. //---------------------------------------------------------------------------
  95. void __fastcall TFileOperationProgressType::ClearTransfer()
  96. {
  97. if ((TransferSize > 0) && (TransferredSize < TransferSize))
  98. {
  99. TGuard Guard(FSection);
  100. __int64 RemainingSize = (TransferSize - TransferredSize);
  101. AddSkipped(RemainingSize);
  102. }
  103. FLocalSize = 0;
  104. FTransferSize = 0;
  105. FLocallyUsed = 0;
  106. FSkippedSize = 0;
  107. FTransferredSize = 0;
  108. FTransferringFile = false;
  109. FLastSecond = 0;
  110. }
  111. //---------------------------------------------------------------------------
  112. void __fastcall TFileOperationProgressType::Start(TFileOperation AOperation,
  113. TOperationSide ASide, int ACount)
  114. {
  115. Start(AOperation, ASide, ACount, false, L"", 0);
  116. }
  117. //---------------------------------------------------------------------------
  118. void __fastcall TFileOperationProgressType::Start(TFileOperation AOperation,
  119. TOperationSide ASide, int ACount, bool ATemp,
  120. const UnicodeString ADirectory, unsigned long ACPSLimit)
  121. {
  122. {
  123. TGuard Guard(FSection); // not really needed, just for consistency
  124. Clear();
  125. FOperation = AOperation;
  126. FSide = ASide;
  127. FCount = ACount;
  128. FInProgress = true;
  129. FCancel = csContinue;
  130. FDirectory = ADirectory;
  131. FTemp = ATemp;
  132. FCPSLimit = ACPSLimit;
  133. }
  134. try
  135. {
  136. DoProgress();
  137. }
  138. catch (...)
  139. {
  140. // connection can be lost during progress callbacks
  141. ClearTransfer();
  142. FInProgress = false;
  143. throw;
  144. }
  145. }
  146. //---------------------------------------------------------------------------
  147. void __fastcall TFileOperationProgressType::Reset()
  148. {
  149. FReset = true;
  150. }
  151. //---------------------------------------------------------------------------
  152. void __fastcall TFileOperationProgressType::Stop()
  153. {
  154. // added to include remaining bytes to TotalSkipped, in case
  155. // the progress happens to update before closing
  156. ClearTransfer();
  157. FInProgress = false;
  158. DoProgress();
  159. }
  160. //---------------------------------------------------------------------------
  161. void __fastcall TFileOperationProgressType::SetDone()
  162. {
  163. FDone = true;
  164. DoProgress();
  165. }
  166. //---------------------------------------------------------------------------
  167. void __fastcall TFileOperationProgressType::Suspend()
  168. {
  169. {
  170. TGuard Guard(FSection);
  171. DebugAssert(!Suspended);
  172. FSuspended = true;
  173. FSuspendTime = GetTickCount();
  174. }
  175. DoProgress();
  176. }
  177. //---------------------------------------------------------------------------
  178. void __fastcall TFileOperationProgressType::Resume()
  179. {
  180. {
  181. TGuard Guard(FSection);
  182. DebugAssert(Suspended);
  183. FSuspended = false;
  184. // shift timestamps for CPS calculation in advance
  185. // by the time the progress was suspended
  186. unsigned long Stopped = (GetTickCount() - FSuspendTime);
  187. size_t i = 0;
  188. while (i < FTicks.size())
  189. {
  190. FTicks[i] += Stopped;
  191. ++i;
  192. }
  193. }
  194. DoProgress();
  195. }
  196. //---------------------------------------------------------------------------
  197. int __fastcall TFileOperationProgressType::OperationProgress()
  198. {
  199. int Result;
  200. if (FCount > 0)
  201. {
  202. Result = (FFilesFinished * 100)/FCount;
  203. }
  204. else
  205. {
  206. Result = 0;
  207. }
  208. return Result;
  209. }
  210. //---------------------------------------------------------------------------
  211. int __fastcall TFileOperationProgressType::TransferProgress()
  212. {
  213. int Result;
  214. if (TransferSize)
  215. {
  216. Result = (int)((TransferredSize * 100)/TransferSize);
  217. }
  218. else
  219. {
  220. Result = 0;
  221. }
  222. return Result;
  223. }
  224. //---------------------------------------------------------------------------
  225. int __fastcall TFileOperationProgressType::TotalTransferProgress()
  226. {
  227. TGuard Guard(FSection);
  228. DebugAssert(TotalSizeSet);
  229. int Result = FTotalSize > 0 ? (int)(((FTotalTransferred + FTotalSkipped) * 100)/FTotalSize) : 0;
  230. return Result < 100 ? Result : 100;
  231. }
  232. //---------------------------------------------------------------------------
  233. int __fastcall TFileOperationProgressType::OverallProgress()
  234. {
  235. if (TotalSizeSet)
  236. {
  237. DebugAssert((Operation == foCopy) || (Operation == foMove));
  238. return TotalTransferProgress();
  239. }
  240. else
  241. {
  242. return OperationProgress();
  243. }
  244. }
  245. //---------------------------------------------------------------------------
  246. void __fastcall TFileOperationProgressType::Progress()
  247. {
  248. DoProgress();
  249. }
  250. //---------------------------------------------------------------------------
  251. void __fastcall TFileOperationProgressType::DoProgress()
  252. {
  253. SetThreadExecutionState(ES_SYSTEM_REQUIRED);
  254. FOnProgress(*this);
  255. }
  256. //---------------------------------------------------------------------------
  257. void __fastcall TFileOperationProgressType::Finish(UnicodeString FileName,
  258. bool Success, TOnceDoneOperation & OnceDoneOperation)
  259. {
  260. DebugAssert(InProgress);
  261. // Cancel reader is guarded
  262. FOnFinished(Operation, Side, Temp, FileName,
  263. Success && (Cancel == csContinue), OnceDoneOperation);
  264. FFilesFinished++;
  265. if (Success)
  266. {
  267. FFilesFinishedSuccessfully++;
  268. }
  269. DoProgress();
  270. }
  271. //---------------------------------------------------------------------------
  272. void __fastcall TFileOperationProgressType::SetFile(UnicodeString AFileName, bool AFileInProgress)
  273. {
  274. FFullFileName = AFileName;
  275. if (Side == osRemote)
  276. {
  277. // historically set were passing filename-only for remote site operations,
  278. // now we need to collect a full paths, so we pass in full path,
  279. // but still want to have filename-only in FileName
  280. AFileName = UnixExtractFileName(AFileName);
  281. }
  282. FFileName = AFileName;
  283. FFileInProgress = AFileInProgress;
  284. ClearTransfer();
  285. FFileStartTime = Now();
  286. DoProgress();
  287. }
  288. //---------------------------------------------------------------------------
  289. void __fastcall TFileOperationProgressType::SetFileInProgress()
  290. {
  291. DebugAssert(!FileInProgress);
  292. FFileInProgress = true;
  293. DoProgress();
  294. }
  295. //---------------------------------------------------------------------------
  296. void __fastcall TFileOperationProgressType::SetLocalSize(__int64 ASize)
  297. {
  298. FLocalSize = ASize;
  299. DoProgress();
  300. }
  301. //---------------------------------------------------------------------------
  302. void __fastcall TFileOperationProgressType::AddLocallyUsed(__int64 ASize)
  303. {
  304. FLocallyUsed += ASize;
  305. if (LocallyUsed > LocalSize)
  306. {
  307. FLocalSize = LocallyUsed;
  308. }
  309. DoProgress();
  310. }
  311. //---------------------------------------------------------------------------
  312. bool __fastcall TFileOperationProgressType::IsLocallyDone()
  313. {
  314. DebugAssert(LocallyUsed <= LocalSize);
  315. return (LocallyUsed == LocalSize);
  316. }
  317. //---------------------------------------------------------------------------
  318. void __fastcall TFileOperationProgressType::SetSpeedCounters()
  319. {
  320. if ((CPSLimit > 0) && !FCounterSet)
  321. {
  322. FCounterSet = true;
  323. Configuration->Usage->Inc(L"SpeedLimitUses");
  324. }
  325. }
  326. //---------------------------------------------------------------------------
  327. // Used in WebDAV protocol
  328. void __fastcall TFileOperationProgressType::ThrottleToCPSLimit(
  329. unsigned long Size)
  330. {
  331. unsigned long Remaining = Size;
  332. while (Remaining > 0)
  333. {
  334. Remaining -= AdjustToCPSLimit(Remaining);
  335. }
  336. }
  337. //---------------------------------------------------------------------------
  338. unsigned long __fastcall TFileOperationProgressType::AdjustToCPSLimit(
  339. unsigned long Size)
  340. {
  341. SetSpeedCounters();
  342. // CPSLimit reader is guarded, we cannot block whole method as it can last long.
  343. if (CPSLimit > 0)
  344. {
  345. // we must not return 0, hence, if we reach zero,
  346. // we wait until the next second
  347. do
  348. {
  349. unsigned int Second = (GetTickCount() / MSecsPerSec);
  350. if (Second != FLastSecond)
  351. {
  352. FRemainingCPS = CPSLimit;
  353. FLastSecond = Second;
  354. }
  355. if (FRemainingCPS == 0)
  356. {
  357. SleepEx(100, true);
  358. DoProgress();
  359. }
  360. }
  361. while ((CPSLimit > 0) && (FRemainingCPS == 0));
  362. // CPSLimit may have been dropped in DoProgress
  363. if (CPSLimit > 0)
  364. {
  365. if (FRemainingCPS < Size)
  366. {
  367. Size = FRemainingCPS;
  368. }
  369. FRemainingCPS -= Size;
  370. }
  371. }
  372. return Size;
  373. }
  374. //---------------------------------------------------------------------------
  375. // Use in SCP protocol only
  376. unsigned long __fastcall TFileOperationProgressType::LocalBlockSize()
  377. {
  378. unsigned long Result = TRANSFER_BUF_SIZE;
  379. if (LocallyUsed + Result > LocalSize)
  380. {
  381. Result = (unsigned long)(LocalSize - LocallyUsed);
  382. }
  383. Result = AdjustToCPSLimit(Result);
  384. return Result;
  385. }
  386. //---------------------------------------------------------------------------
  387. void __fastcall TFileOperationProgressType::SetTotalSize(__int64 ASize)
  388. {
  389. TGuard Guard(FSection); // not really needed, just for consistency
  390. FTotalSize = ASize;
  391. FTotalSizeSet = true;
  392. // parent has its own totals
  393. if (FParent != NULL)
  394. {
  395. DebugAssert(FParent->TotalSizeSet);
  396. }
  397. DoProgress();
  398. }
  399. //---------------------------------------------------------------------------
  400. void __fastcall TFileOperationProgressType::SetTransferSize(__int64 ASize)
  401. {
  402. FTransferSize = ASize;
  403. DoProgress();
  404. }
  405. //---------------------------------------------------------------------------
  406. void __fastcall TFileOperationProgressType::SetTransferringFile(bool ATransferringFile)
  407. {
  408. FTransferringFile = ATransferringFile;
  409. }
  410. //---------------------------------------------------------------------------
  411. bool __fastcall TFileOperationProgressType::PassCancelToParent(TCancelStatus ACancel)
  412. {
  413. bool Result;
  414. if (ACancel < csCancel)
  415. {
  416. // do not propagate csCancelFile,
  417. // though it's not supported for queue atm, so we do not expect it here
  418. DebugFail();
  419. Result = false;
  420. }
  421. else if (ACancel == csCancel)
  422. {
  423. Result = true;
  424. }
  425. else
  426. {
  427. // csCancelTransfer and csRemoteAbort are used with SCP only, which does not use parallel transfers
  428. DebugFail();
  429. Result = false;
  430. }
  431. return Result;
  432. }
  433. //---------------------------------------------------------------------------
  434. void __fastcall TFileOperationProgressType::SetCancel(TCancelStatus ACancel)
  435. {
  436. TGuard Guard(FSection);
  437. FCancel = ACancel;
  438. if ((FParent != NULL) && PassCancelToParent(ACancel))
  439. {
  440. FParent->SetCancel(ACancel);
  441. }
  442. }
  443. //---------------------------------------------------------------------------
  444. void __fastcall TFileOperationProgressType::SetCancelAtLeast(TCancelStatus ACancel)
  445. {
  446. TGuard Guard(FSection);
  447. if (FCancel < ACancel)
  448. {
  449. FCancel = ACancel;
  450. }
  451. if ((FParent != NULL) && PassCancelToParent(ACancel))
  452. {
  453. FParent->SetCancelAtLeast(ACancel);
  454. }
  455. }
  456. //---------------------------------------------------------------------------
  457. TCancelStatus __fastcall TFileOperationProgressType::GetCancel()
  458. {
  459. TCancelStatus Result = FCancel;
  460. if (FParent != NULL)
  461. {
  462. TGuard Guard(FSection);
  463. TCancelStatus ParentCancel = FParent->Cancel;
  464. if (ParentCancel > Result)
  465. {
  466. Result = ParentCancel;
  467. }
  468. }
  469. return Result;
  470. }
  471. //---------------------------------------------------------------------------
  472. bool __fastcall TFileOperationProgressType::ClearCancelFile()
  473. {
  474. TGuard Guard(FSection);
  475. // Not propagated to parent, as this is local flag, see also PassCancelToParent
  476. bool Result = (Cancel == csCancelFile);
  477. if (Result)
  478. {
  479. FCancel = csContinue;
  480. }
  481. return Result;
  482. }
  483. //---------------------------------------------------------------------------
  484. unsigned long __fastcall TFileOperationProgressType::GetCPSLimit()
  485. {
  486. unsigned int Result;
  487. if (FParent != NULL)
  488. {
  489. Result = FParent->CPSLimit;
  490. }
  491. else
  492. {
  493. TGuard Guard(FSection);
  494. Result = FCPSLimit;
  495. }
  496. return Result;
  497. }
  498. //---------------------------------------------------------------------------
  499. void __fastcall TFileOperationProgressType::SetCPSLimit(unsigned long ACPSLimit)
  500. {
  501. if (FParent != NULL)
  502. {
  503. FParent->SetCPSLimit(ACPSLimit);
  504. }
  505. else
  506. {
  507. TGuard Guard(FSection);
  508. FCPSLimit = ACPSLimit;
  509. }
  510. }
  511. //---------------------------------------------------------------------------
  512. TBatchOverwrite __fastcall TFileOperationProgressType::GetBatchOverwrite()
  513. {
  514. TBatchOverwrite Result;
  515. if (FParent != NULL)
  516. {
  517. Result = FParent->BatchOverwrite;
  518. }
  519. else
  520. {
  521. TGuard Guard(FSection); // not really needed
  522. Result = FBatchOverwrite;
  523. }
  524. return Result;
  525. }
  526. //---------------------------------------------------------------------------
  527. void __fastcall TFileOperationProgressType::SetBatchOverwrite(TBatchOverwrite ABatchOverwrite)
  528. {
  529. if (FParent != NULL)
  530. {
  531. FParent->SetBatchOverwrite(ABatchOverwrite);
  532. }
  533. else
  534. {
  535. TGuard Guard(FSection); // not really needed
  536. FBatchOverwrite = ABatchOverwrite;;
  537. }
  538. }
  539. //---------------------------------------------------------------------------
  540. bool __fastcall TFileOperationProgressType::GetSkipToAll()
  541. {
  542. bool Result;
  543. if (FParent != NULL)
  544. {
  545. Result = FParent->SkipToAll;
  546. }
  547. else
  548. {
  549. TGuard Guard(FSection); // not really needed
  550. Result = FSkipToAll;
  551. }
  552. return Result;
  553. }
  554. //---------------------------------------------------------------------------
  555. void __fastcall TFileOperationProgressType::SetSkipToAll()
  556. {
  557. if (FParent != NULL)
  558. {
  559. FParent->SetSkipToAll();
  560. }
  561. else
  562. {
  563. TGuard Guard(FSection); // not really needed
  564. FSkipToAll = true;
  565. }
  566. }
  567. //---------------------------------------------------------------------------
  568. void __fastcall TFileOperationProgressType::ChangeTransferSize(__int64 ASize)
  569. {
  570. // reflect change on file size (due to text transfer mode conversion particulary)
  571. // on total transfer size
  572. if (TotalSizeSet)
  573. {
  574. AddTotalSize(ASize - TransferSize);
  575. }
  576. FTransferSize = ASize;
  577. DoProgress();
  578. }
  579. //---------------------------------------------------------------------------
  580. void __fastcall TFileOperationProgressType::RollbackTransferFromTotals(__int64 ATransferredSize, __int64 ASkippedSize)
  581. {
  582. TGuard Guard(FSection);
  583. DebugAssert(ATransferredSize <= FTotalTransferred);
  584. DebugAssert(ASkippedSize <= FTotalSkipped);
  585. FTotalTransferred -= ATransferredSize;
  586. FTicks.clear();
  587. FTotalTransferredThen.clear();
  588. FTotalSkipped -= ASkippedSize;
  589. if (FParent != NULL)
  590. {
  591. FParent->RollbackTransferFromTotals(ATransferredSize, ASkippedSize);
  592. }
  593. }
  594. //---------------------------------------------------------------------------
  595. void __fastcall TFileOperationProgressType::RollbackTransfer()
  596. {
  597. FTransferredSize -= FSkippedSize;
  598. RollbackTransferFromTotals(FTransferredSize, FSkippedSize);
  599. FSkippedSize = 0;
  600. FTransferredSize = 0;
  601. FTransferSize = 0;
  602. FLocallyUsed = 0;
  603. }
  604. //---------------------------------------------------------------------------
  605. void __fastcall TFileOperationProgressType::AddTransferredToTotals(__int64 ASize)
  606. {
  607. TGuard Guard(FSection);
  608. FTotalTransferred += ASize;
  609. unsigned long Ticks = GetTickCount();
  610. if (FTicks.empty() ||
  611. (FTicks.back() > Ticks) || // ticks wrap after 49.7 days
  612. ((Ticks - FTicks.back()) >= MSecsPerSec))
  613. {
  614. FTicks.push_back(Ticks);
  615. FTotalTransferredThen.push_back(FTotalTransferred);
  616. }
  617. if (FTicks.size() > 10)
  618. {
  619. FTicks.erase(FTicks.begin());
  620. FTotalTransferredThen.erase(FTotalTransferredThen.begin());
  621. }
  622. if (FParent != NULL)
  623. {
  624. FParent->AddTransferredToTotals(ASize);
  625. }
  626. }
  627. //---------------------------------------------------------------------------
  628. void __fastcall TFileOperationProgressType::AddTotalSize(__int64 ASize)
  629. {
  630. if (ASize != 0)
  631. {
  632. TGuard Guard(FSection);
  633. FTotalSize += ASize;
  634. if (FParent != NULL)
  635. {
  636. FParent->AddTotalSize(ASize);
  637. }
  638. }
  639. }
  640. //---------------------------------------------------------------------------
  641. void __fastcall TFileOperationProgressType::AddTransferred(__int64 ASize,
  642. bool AddToTotals)
  643. {
  644. FTransferredSize += ASize;
  645. if (TransferredSize > TransferSize)
  646. {
  647. // this can happen with SFTP when downloading file that
  648. // grows while being downloaded
  649. if (TotalSizeSet)
  650. {
  651. // we should probably guard this with AddToTotals
  652. AddTotalSize(TransferredSize - TransferSize);
  653. }
  654. FTransferSize = TransferredSize;
  655. }
  656. if (AddToTotals)
  657. {
  658. AddTransferredToTotals(ASize);
  659. }
  660. DoProgress();
  661. }
  662. //---------------------------------------------------------------------------
  663. void __fastcall TFileOperationProgressType::AddSkipped(__int64 ASize)
  664. {
  665. TGuard Guard(FSection);
  666. FTotalSkipped += ASize;
  667. if (FParent != NULL)
  668. {
  669. FParent->AddSkipped(ASize);
  670. }
  671. }
  672. //---------------------------------------------------------------------------
  673. void __fastcall TFileOperationProgressType::AddResumed(__int64 ASize)
  674. {
  675. AddSkipped(ASize);
  676. FSkippedSize += ASize;
  677. AddTransferred(ASize, false);
  678. AddLocallyUsed(ASize);
  679. }
  680. //---------------------------------------------------------------------------
  681. void __fastcall TFileOperationProgressType::AddSkippedFileSize(__int64 ASize)
  682. {
  683. AddSkipped(ASize);
  684. DoProgress();
  685. }
  686. //---------------------------------------------------------------------------
  687. // Use in SCP protocol only
  688. unsigned long __fastcall TFileOperationProgressType::TransferBlockSize()
  689. {
  690. unsigned long Result = TRANSFER_BUF_SIZE;
  691. if (TransferredSize + Result > TransferSize)
  692. {
  693. Result = (unsigned long)(TransferSize - TransferredSize);
  694. }
  695. Result = AdjustToCPSLimit(Result);
  696. return Result;
  697. }
  698. //---------------------------------------------------------------------------
  699. unsigned long __fastcall TFileOperationProgressType::StaticBlockSize()
  700. {
  701. return TRANSFER_BUF_SIZE;
  702. }
  703. //---------------------------------------------------------------------------
  704. bool __fastcall TFileOperationProgressType::IsTransferDone()
  705. {
  706. DebugAssert(TransferredSize <= TransferSize);
  707. return (TransferredSize == TransferSize);
  708. }
  709. //---------------------------------------------------------------------------
  710. void __fastcall TFileOperationProgressType::SetAsciiTransfer(bool AAsciiTransfer)
  711. {
  712. FAsciiTransfer = AAsciiTransfer;
  713. DoProgress();
  714. }
  715. //---------------------------------------------------------------------------
  716. TDateTime __fastcall TFileOperationProgressType::TimeElapsed()
  717. {
  718. return Now() - StartTime;
  719. }
  720. //---------------------------------------------------------------------------
  721. unsigned int __fastcall TFileOperationProgressType::CPS()
  722. {
  723. TGuard Guard(FSection);
  724. return GetCPS();
  725. }
  726. //---------------------------------------------------------------------------
  727. // Has to be called from a guarded method
  728. unsigned int __fastcall TFileOperationProgressType::GetCPS()
  729. {
  730. unsigned int Result;
  731. if (FTicks.empty())
  732. {
  733. Result = 0;
  734. }
  735. else
  736. {
  737. unsigned long Ticks = (Suspended ? FSuspendTime : GetTickCount());
  738. unsigned long TimeSpan;
  739. if (Ticks < FTicks.front())
  740. {
  741. // clocks has wrapped, guess 10 seconds difference
  742. TimeSpan = 10000;
  743. }
  744. else
  745. {
  746. TimeSpan = (Ticks - FTicks.front());
  747. }
  748. if (TimeSpan == 0)
  749. {
  750. Result = 0;
  751. }
  752. else
  753. {
  754. __int64 Transferred = (FTotalTransferred - FTotalTransferredThen.front());
  755. Result = (unsigned int)(Transferred * MSecsPerSec / TimeSpan);
  756. }
  757. }
  758. return Result;
  759. }
  760. //---------------------------------------------------------------------------
  761. TDateTime __fastcall TFileOperationProgressType::TimeExpected()
  762. {
  763. unsigned int CurCps = CPS();
  764. if (CurCps)
  765. {
  766. return TDateTime((double)(((double)(TransferSize - TransferredSize)) / CurCps) / SecsPerDay);
  767. }
  768. else
  769. {
  770. return 0;
  771. }
  772. }
  773. //---------------------------------------------------------------------------
  774. TDateTime __fastcall TFileOperationProgressType::TotalTimeLeft()
  775. {
  776. TGuard Guard(FSection);
  777. DebugAssert(FTotalSizeSet);
  778. unsigned int CurCps = GetCPS();
  779. // sanity check
  780. if ((CurCps > 0) && (FTotalSize > FTotalSkipped + FTotalTransferred))
  781. {
  782. return TDateTime((double)((double)(FTotalSize - FTotalSkipped - FTotalTransferred) / CurCps) /
  783. SecsPerDay);
  784. }
  785. else
  786. {
  787. return 0;
  788. }
  789. }
  790. //---------------------------------------------------------------------------
  791. __int64 __fastcall TFileOperationProgressType::GetTotalTransferred()
  792. {
  793. TGuard Guard(FSection);
  794. return FTotalTransferred;
  795. }
  796. //---------------------------------------------------------------------------
  797. __int64 __fastcall TFileOperationProgressType::GetTotalSize()
  798. {
  799. TGuard Guard(FSection);
  800. return FTotalSize;
  801. }
  802. //---------------------------------------------------------------------------
  803. void __fastcall TFileOperationProgressType::LockUserSelections()
  804. {
  805. if (FParent != NULL)
  806. {
  807. FParent->LockUserSelections();
  808. }
  809. else
  810. {
  811. FUserSelectionsSection->Enter();
  812. }
  813. }
  814. //---------------------------------------------------------------------------
  815. void __fastcall TFileOperationProgressType::UnlockUserSelections()
  816. {
  817. if (FParent != NULL)
  818. {
  819. FParent->UnlockUserSelections();
  820. }
  821. else
  822. {
  823. FUserSelectionsSection->Leave();
  824. }
  825. }
  826. //---------------------------------------------------------------------------
  827. UnicodeString __fastcall TFileOperationProgressType::GetLogStr(bool Done)
  828. {
  829. UnicodeString Transferred = FormatSize(TotalTransferred);
  830. UnicodeString Left;
  831. TDateTime Time;
  832. UnicodeString TimeLabel;
  833. if (!Done && TotalSizeSet)
  834. {
  835. Time = TotalTimeLeft();
  836. TimeLabel = L"Left";
  837. }
  838. else
  839. {
  840. Time = TimeElapsed();
  841. TimeLabel = L"Elapsed";
  842. }
  843. UnicodeString TimeStr = FormatDateTimeSpan(Configuration->TimeFormat, Time);
  844. UnicodeString CPSStr = FormatSize(CPS());
  845. return FORMAT(L"Transferred: %s, %s: %s, CPS: %s/s", (Transferred, TimeLabel, TimeStr, CPSStr));
  846. }