2021-10-16 16:19:51 +01:00
# include "stdafx.h"
# include "RecoveryWindow.h"
# include "SandMan.h"
# include "../MiscHelpers/Common/Settings.h"
# include "../MiscHelpers/Common/TreeItemModel.h"
# include "../MiscHelpers/Common/Common.h"
2022-08-10 19:14:37 +01:00
# include "SettingsWindow.h"
2021-10-16 16:19:51 +01:00
# if defined(Q_OS_WIN)
# include <wtypes.h>
# include <QAbstractNativeEventFilter>
# include <dbt.h>
# endif
2022-06-14 19:58:13 +01:00
CRecoveryWindow : : CRecoveryWindow ( const CSandBoxPtr & pBox , bool bImmediate , QWidget * parent )
2021-10-16 16:19:51 +01:00
: QDialog ( parent )
{
2022-06-14 19:58:13 +01:00
m_bImmediate = bImmediate ;
2022-06-13 19:12:53 +01:00
2021-10-16 16:19:51 +01:00
Qt : : WindowFlags flags = windowFlags ( ) ;
flags | = Qt : : CustomizeWindowHint ;
//flags &= ~Qt::WindowContextHelpButtonHint;
//flags &= ~Qt::WindowSystemMenuHint;
2021-12-08 21:40:25 +00:00
flags | = Qt : : WindowMinMaxButtonsHint ;
2021-10-16 16:19:51 +01:00
flags | = Qt : : WindowMinimizeButtonHint ;
//flags &= ~Qt::WindowCloseButtonHint;
setWindowFlags ( flags ) ;
//setWindowState(Qt::WindowActive);
SetForegroundWindow ( ( HWND ) QWidget : : winId ( ) ) ;
2022-06-14 19:58:13 +01:00
bool bAlwaysOnTop = theConf - > GetBool ( " Options/AlwaysOnTop " , false ) | | ( bImmediate & & theConf - > GetBool ( " Options/RecoveryOnTop " , true ) ) ;
2021-10-16 16:19:51 +01:00
this - > setWindowFlag ( Qt : : WindowStaysOnTopHint , bAlwaysOnTop ) ;
if ( ! bAlwaysOnTop ) {
HWND hWnd = ( HWND ) this - > winId ( ) ;
SetWindowPos ( hWnd , HWND_TOPMOST , 0 , 0 , 0 , 0 , SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE ) ;
QTimer : : singleShot ( 100 , this , [ hWnd ] ( ) {
SetWindowPos ( hWnd , HWND_NOTOPMOST , 0 , 0 , 0 , 0 , SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE ) ;
} ) ;
}
ui . setupUi ( this ) ;
this - > setWindowTitle ( tr ( " %1 - File Recovery " ) . arg ( pBox - > GetName ( ) ) ) ;
2022-08-10 19:14:37 +01:00
FixTriStateBoxPallete ( this ) ;
2022-07-29 09:24:32 +01:00
ui . treeFiles - > setAlternatingRowColors ( theConf - > GetBool ( " Options/AltRowColors " , false ) ) ;
2021-10-16 16:19:51 +01:00
m_pBox = pBox ;
m_pCounter = NULL ;
m_LastTargetIndex = 0 ;
m_bTargetsChanged = false ;
m_bReloadPending = false ;
2023-02-03 07:00:11 +00:00
m_DeleteSnapshots = false ;
2021-10-16 16:19:51 +01:00
QStyle * pStyle = QStyleFactory : : create ( " windows " ) ;
ui . treeFiles - > setStyle ( pStyle ) ;
2022-07-31 19:05:22 +01:00
ui . treeFiles - > setItemDelegate ( new CTreeItemDelegate ( ) ) ;
2021-10-16 16:19:51 +01:00
ui . treeFiles - > setExpandsOnDoubleClick ( false ) ;
ui . btnDeleteAll - > setVisible ( false ) ;
2023-01-12 22:10:50 +00:00
m_pFileModel = new CSimpleTreeModel ( this ) ;
2021-10-16 16:19:51 +01:00
m_pFileModel - > SetUseIcons ( true ) ;
m_pFileModel - > AddColumn ( tr ( " File Name " ) , " FileName " ) ;
m_pFileModel - > AddColumn ( tr ( " File Size " ) , " FileSize " ) ;
m_pFileModel - > AddColumn ( tr ( " Full Path " ) , " DiskPath " ) ;
2022-07-29 09:24:32 +01:00
m_pSortProxy = new CSortFilterProxyModel ( this ) ;
2021-10-16 16:19:51 +01:00
m_pSortProxy - > setSortRole ( Qt : : EditRole ) ;
m_pSortProxy - > setSourceModel ( m_pFileModel ) ;
m_pSortProxy - > setDynamicSortFilter ( true ) ;
//ui.treeFiles->setItemDelegate(theGUI->GetItemDelegate());
ui . treeFiles - > setModel ( m_pSortProxy ) ;
ui . treeFiles - > setSelectionMode ( QAbstractItemView : : ExtendedSelection ) ;
ui . treeFiles - > setSortingEnabled ( true ) ;
//ui.treeFiles->setUniformRowHeights(true);
2023-01-12 22:10:50 +00:00
CFinder * pFinder = new CFinder ( m_pSortProxy , this ) ;
ui . gridLayout - > addWidget ( pFinder , 3 , 0 , 1 , 5 ) ;
pFinder - > SetTree ( ui . treeFiles ) ;
2021-10-16 16:19:51 +01:00
ui . finder - > deleteLater ( ) ; // remove place holder
//connect(ui.treeFiles, SIGNAL(clicked(const QModelIndex&)), this, SLOT(UpdateSnapshot(const QModelIndex&)));
//connect(ui.treeFiles->selectionModel(), SIGNAL(currentChanged(QModelIndex, QModelIndex)), this, SLOT(UpdateSnapshot(const QModelIndex&)));
2021-12-26 09:19:31 +00:00
connect ( ui . treeFiles , SIGNAL ( doubleClicked ( const QModelIndex & ) ) , this , SLOT ( OnRecover ( ) ) ) ;
2021-10-16 16:19:51 +01:00
connect ( ui . btnAddFolder , SIGNAL ( clicked ( bool ) ) , this , SLOT ( OnAddFolder ( ) ) ) ;
connect ( ui . chkShowAll , SIGNAL ( clicked ( bool ) ) , this , SLOT ( FindFiles ( ) ) ) ;
connect ( ui . btnRefresh , SIGNAL ( clicked ( bool ) ) , this , SLOT ( FindFiles ( ) ) ) ;
connect ( ui . btnRecover , SIGNAL ( clicked ( bool ) ) , this , SLOT ( OnRecover ( ) ) ) ;
2022-07-14 09:46:13 +01:00
connect ( ui . btnDelete , SIGNAL ( clicked ( bool ) ) , this , SLOT ( OnDelete ( ) ) ) ;
2021-10-16 16:19:51 +01:00
connect ( ui . cmbRecover , SIGNAL ( currentIndexChanged ( int ) ) , this , SLOT ( OnTargetChanged ( ) ) ) ;
connect ( ui . btnDeleteAll , SIGNAL ( clicked ( bool ) ) , this , SLOT ( OnDeleteAll ( ) ) ) ;
connect ( ui . btnClose , SIGNAL ( clicked ( bool ) ) , this , SLOT ( close ( ) ) ) ;
QMenu * pRecMenu = new QMenu ( ui . btnRecover ) ;
m_pRemember = pRecMenu - > addAction ( tr ( " Remember target selection " ) ) ;
m_pRemember - > setCheckable ( true ) ;
ui . btnRecover - > setPopupMode ( QToolButton : : MenuButtonPopup ) ;
ui . btnRecover - > setMenu ( pRecMenu ) ;
2022-01-30 14:53:37 +00:00
QMenu * pDelMenu = new QMenu ( ui . btnDeleteAll ) ;
2022-01-30 16:56:20 +00:00
pDelMenu - > addAction ( tr ( " Delete everything, including all snapshots " ) , this , SLOT ( OnDeleteEverything ( ) ) ) ;
2022-01-30 14:53:37 +00:00
ui . btnDeleteAll - > setPopupMode ( QToolButton : : MenuButtonPopup ) ;
ui . btnDeleteAll - > setMenu ( pDelMenu ) ;
2021-10-16 16:19:51 +01:00
restoreGeometry ( theConf - > GetBlob ( " RecoveryWindow/Window_Geometry " ) ) ;
QByteArray Columns = theConf - > GetBlob ( " RecoveryWindow/TreeView_Columns " ) ;
if ( ! Columns . isEmpty ( ) )
ui . treeFiles - > header ( ) - > restoreState ( Columns ) ;
for ( int i = 0 ; i < m_pFileModel - > columnCount ( ) ; i + + )
m_pFileModel - > SetColumnEnabled ( i , true ) ;
foreach ( const QString & NtFolder , m_pBox - > GetTextList ( " RecoverFolder " , true , true ) )
{
2022-05-08 14:26:59 +01:00
QString RecFolder = theAPI - > ResolveAbsolutePath ( NtFolder ) ;
2021-10-16 16:19:51 +01:00
bool bOk ;
2022-05-08 14:26:59 +01:00
QString Folder = theAPI - > Nt2DosPath ( RecFolder , & bOk ) ;
2021-10-16 16:19:51 +01:00
if ( bOk )
m_RecoveryFolders . append ( Folder ) ;
2022-05-08 14:26:59 +01:00
else if ( RecFolder . left ( 11 ) = = " \\ Device \\ Mup " )
m_RecoveryFolders . append ( " \\ " + RecFolder . mid ( 11 ) ) ;
2021-10-16 16:19:51 +01:00
}
ui . cmbRecover - > addItem ( tr ( " Original location " ) , 0 ) ;
ui . cmbRecover - > addItem ( tr ( " Browse for location " ) , 1 ) ;
ui . cmbRecover - > addItem ( tr ( " Clear folder list " ) , - 1 ) ;
QStringList RecoverTargets = theAPI - > GetUserSettings ( ) - > GetTextList ( " SbieCtrl_RecoverTarget " , true ) ;
ui . cmbRecover - > insertItems ( ui . cmbRecover - > count ( ) - 1 , RecoverTargets ) ;
m_LastTargetIndex = theConf - > GetInt ( " RecoveryWindow/LastTarget " , - 1 ) ;
m_pRemember - > setChecked ( m_LastTargetIndex ! = - 1 ) ;
if ( m_LastTargetIndex = = - 1 )
m_LastTargetIndex = 0 ;
ui . cmbRecover - > setCurrentIndex ( m_LastTargetIndex ) ;
}
CRecoveryWindow : : ~ CRecoveryWindow ( )
{
theConf - > SetBlob ( " RecoveryWindow/Window_Geometry " , saveGeometry ( ) ) ;
theConf - > SetBlob ( " RecoveryWindow/TreeView_Columns " , ui . treeFiles - > header ( ) - > saveState ( ) ) ;
}
int CRecoveryWindow : : exec ( )
{
//QDialog::setWindowModality(Qt::WindowModal);
ui . btnDeleteAll - > setVisible ( true ) ;
2022-01-30 14:53:37 +00:00
SafeShow ( this ) ;
2021-10-16 16:19:51 +01:00
return QDialog : : exec ( ) ;
}
2022-12-21 09:14:31 +00:00
bool CRecoveryWindow : : IsDeleteDialog ( ) const
{
return ui . btnDeleteAll - > isVisible ( ) ;
}
2021-10-16 16:19:51 +01:00
void CRecoveryWindow : : closeEvent ( QCloseEvent * e )
{
emit Closed ( ) ;
this - > deleteLater ( ) ;
}
void CRecoveryWindow : : OnAddFolder ( )
{
2021-12-20 11:55:02 +00:00
QString Folder = QFileDialog : : getExistingDirectory ( this , tr ( " Select Directory " ) ) . replace ( " / " , " \\ " ) ;
2021-10-16 16:19:51 +01:00
if ( Folder . isEmpty ( ) )
return ;
if ( m_RecoveryFolders . contains ( Folder ) )
return ;
m_RecoveryFolders . append ( Folder ) ;
m_pBox - > AppendText ( " RecoverFolder " , Folder ) ;
FindFiles ( Folder ) ;
m_pFileModel - > Sync ( m_FileMap ) ;
ui . treeFiles - > expandAll ( ) ;
}
void CRecoveryWindow : : OnTargetChanged ( )
{
int op = ui . cmbRecover - > currentData ( ) . toInt ( ) ;
if ( op = = 1 )
{
QString Folder = QFileDialog : : getExistingDirectory ( this , tr ( " Select Directory " ) ) . replace ( " / " , " \\ " ) ;
if ( Folder . isEmpty ( ) ) {
ui . cmbRecover - > setCurrentIndex ( m_LastTargetIndex ) ;
return ;
}
m_LastTargetIndex = ui . cmbRecover - > count ( ) - 1 ;
ui . cmbRecover - > insertItem ( m_LastTargetIndex , Folder ) ;
ui . cmbRecover - > setCurrentIndex ( m_LastTargetIndex ) ;
m_bTargetsChanged = true ;
}
else if ( op = = - 1 )
{
while ( ui . cmbRecover - > count ( ) > 3 )
ui . cmbRecover - > removeItem ( 2 ) ;
ui . cmbRecover - > setCurrentIndex ( 0 ) ;
m_bTargetsChanged = true ;
}
else {
m_LastTargetIndex = ui . cmbRecover - > currentIndex ( ) ;
}
}
void CRecoveryWindow : : OnRecover ( )
{
if ( m_bTargetsChanged ) {
QStringList RecoverTargets ;
for ( int i = 2 ; i < ui . cmbRecover - > count ( ) - 1 ; i + + )
RecoverTargets . append ( ui . cmbRecover - > itemText ( i ) ) ;
theAPI - > GetUserSettings ( ) - > UpdateTextList ( " SbieCtrl_RecoverTarget " , RecoverTargets , true ) ;
}
theConf - > SetValue ( " RecoveryWindow/LastTarget " , m_pRemember - > isChecked ( ) ? m_LastTargetIndex : - 1 ) ;
QString RecoveryFolder ;
if ( ui . cmbRecover - > currentIndex ( ) > 0 )
RecoveryFolder = ui . cmbRecover - > currentText ( ) ;
RecoverFiles ( false , RecoveryFolder ) ;
}
2022-07-14 09:46:13 +01:00
void CRecoveryWindow : : OnDelete ( )
{
QMap < QString , SRecItem > FileMap = GetFiles ( ) ;
2023-03-27 12:23:01 +01:00
if ( FileMap . isEmpty ( ) )
return ;
if ( QMessageBox : : Yes ! = QMessageBox ( " Sandboxie-Plus " , tr ( " Do you really want to delete %1 selected files? " ) . arg ( FileMap . count ( ) ) , QMessageBox : : Question , QMessageBox : : Yes , QMessageBox : : No | QMessageBox : : Default | QMessageBox : : Escape , QMessageBox : : NoButton , this ) . exec ( ) )
2022-07-14 09:46:13 +01:00
return ;
foreach ( const QString & FilePath , FileMap . keys ( ) )
QFile : : remove ( FilePath ) ;
FindFiles ( ) ;
}
2021-10-16 16:19:51 +01:00
void CRecoveryWindow : : OnDeleteAll ( )
{
2022-07-22 20:56:52 +01:00
this - > setResult ( 1 ) ;
this - > close ( ) ;
2021-10-16 16:19:51 +01:00
}
2022-01-30 14:53:37 +00:00
void CRecoveryWindow : : OnDeleteEverything ( )
{
2023-02-03 07:00:11 +00:00
m_DeleteSnapshots = true ;
2022-01-30 14:53:37 +00:00
OnDeleteAll ( ) ;
}
2021-10-16 16:19:51 +01:00
void CRecoveryWindow : : AddFile ( const QString & FilePath , const QString & BoxPath )
{
ui . chkShowAll - > setTristate ( true ) ;
if ( m_FileMap . isEmpty ( ) ) {
ui . chkShowAll - > setCheckState ( Qt : : PartiallyChecked ) ;
QMenu * pCloseMenu = new QMenu ( ui . btnClose ) ;
pCloseMenu - > addAction ( tr ( " Close until all programs stop in this box " ) , this , SLOT ( OnCloseUntil ( ) ) ) ;
2022-06-04 20:07:04 +01:00
pCloseMenu - > addAction ( tr ( " Close and Disable Immediate Recovery for this box " ) , this , SLOT ( OnAutoDisable ( ) ) ) ;
2021-10-16 16:19:51 +01:00
ui . btnClose - > setPopupMode ( QToolButton : : MenuButtonPopup ) ;
ui . btnClose - > setMenu ( pCloseMenu ) ;
}
m_NewFiles . insert ( FilePath ) ;
if ( m_FileMap . isEmpty ( ) )
FindFiles ( ) ;
else if ( ! m_bReloadPending )
{
m_bReloadPending = true ;
QTimer : : singleShot ( 500 , this , SLOT ( FindFiles ( ) ) ) ;
}
}
int CRecoveryWindow : : FindFiles ( )
{
m_bReloadPending = false ;
if ( ! m_NewFiles . isEmpty ( ) ) {
ui . lblInfo - > setText ( tr ( " There are %1 new files available to recover. " ) . arg ( m_NewFiles . count ( ) ) ) ;
}
else if ( m_pCounter = = NULL ) {
m_pCounter = new CRecoveryCounter ( m_pBox - > GetFileRoot ( ) , this ) ;
connect ( m_pCounter , SIGNAL ( Count ( quint32 , quint32 , quint64 ) ) , this , SLOT ( OnCount ( quint32 , quint32 , quint64 ) ) ) ;
}
m_FileMap . clear ( ) ;
int Count = 0 ;
if ( ui . chkShowAll - > checkState ( ) = = Qt : : Checked )
{
//for(char drive = 'A'; drive <= 'Z'; drive++)
QDir Dir ( m_pBox - > GetFileRoot ( ) + " \\ drive \\ " ) ;
foreach ( const QFileInfo & Info , Dir . entryInfoList ( QDir : : Dirs | QDir : : NoDotAndDotDot ) )
Count + = FindBoxFiles ( " \\ drive \\ " + Info . fileName ( ) ) ;
2022-01-13 22:52:58 +00:00
if ( m_pBox - > GetBool ( " SeparateUserFolders " , true , true ) ) {
2021-10-16 16:19:51 +01:00
Count + = FindBoxFiles ( " \\ user \\ current " ) ;
Count + = FindBoxFiles ( " \\ user \\ all " ) ;
Count + = FindBoxFiles ( " \\ user \\ public " ) ;
}
//Count += FindBoxFiles("\\share");
QDir DirSvr ( m_pBox - > GetFileRoot ( ) + " \\ share \\ " ) ;
foreach ( const QFileInfo & InfoSrv , DirSvr . entryInfoList ( QDir : : Dirs | QDir : : NoDotAndDotDot ) ) {
QDir DirPub ( m_pBox - > GetFileRoot ( ) + " \\ share \\ " + InfoSrv . fileName ( ) ) ;
foreach ( const QFileInfo & InfoPub , DirPub . entryInfoList ( QDir : : Dirs | QDir : : NoDotAndDotDot ) )
Count + = FindBoxFiles ( " \\ share \\ " + InfoSrv . fileName ( ) + " \\ " + InfoPub . fileName ( ) ) ;
}
}
else
{
foreach ( const QString & Folder , m_RecoveryFolders )
Count + = FindFiles ( Folder ) ;
}
2022-06-14 19:58:13 +01:00
if ( m_bImmediate & & m_FileMap . isEmpty ( ) )
2022-06-13 19:12:53 +01:00
this - > close ( ) ;
2021-10-16 16:19:51 +01:00
m_pFileModel - > Sync ( m_FileMap ) ;
ui . treeFiles - > expandAll ( ) ;
2022-09-22 09:27:04 +01:00
if ( m_bImmediate )
SelectFiles ( ) ;
2022-09-22 08:28:49 +01:00
2022-09-22 09:27:04 +01:00
return Count ;
}
void CRecoveryWindow : : SelectFiles ( )
{
//QModelIndex Index = m_pFileModel->index(0, 0);
QModelIndex Index ;
for ( int i = 0 ; i < m_pFileModel - > rowCount ( ) ; + + i )
2022-09-22 08:28:49 +01:00
{
QModelIndex ModelIndex = m_pFileModel - > index ( i , 0 ) ;
QVariant ID = m_pFileModel - > GetItemID ( ModelIndex ) ;
2022-09-22 09:27:04 +01:00
//QVariant ID = m_pFileModel->GetItemID(Index);
2022-09-22 08:28:49 +01:00
2022-09-22 09:27:04 +01:00
QVariantMap File = m_FileMap . value ( ID ) ;
if ( File . isEmpty ( ) )
continue ;
2022-09-22 08:28:49 +01:00
2022-09-22 09:27:04 +01:00
if ( File [ " IsDir " ] . toBool ( ) = = false )
{
Index = ModelIndex ;
goto SelectFile ;
}
else
{
QList < QModelIndex > Folders ;
Folders . append ( ModelIndex ) ;
do
{
QModelIndex CurIndex = Folders . takeFirst ( ) ;
for ( int i = 0 ; i < m_pFileModel - > rowCount ( CurIndex ) ; i + + )
{
QModelIndex ChildIndex = m_pFileModel - > index ( i , 0 , CurIndex ) ;
QVariant ChildID = m_pFileModel - > GetItemID ( ChildIndex ) ;
QVariantMap File = m_FileMap . value ( ChildID ) ;
if ( File . isEmpty ( ) )
continue ;
if ( File [ " IsDir " ] . toBool ( ) = = false )
{
Index = ChildIndex ;
goto SelectFile ;
}
else
Folders . append ( ChildIndex ) ;
}
} while ( ! Folders . isEmpty ( ) ) ;
}
}
SelectFile :
if ( Index . isValid ( ) ) {
QModelIndex ModelIndex = m_pSortProxy - > mapFromSource ( Index ) ;
ui . treeFiles - > selectionModel ( ) - > setCurrentIndex ( ModelIndex , QItemSelectionModel : : SelectCurrent ) ;
ui . treeFiles - > setCurrentIndex ( ModelIndex ) ;
ui . treeFiles - > setFocus ( ) ;
}
2021-10-16 16:19:51 +01:00
}
int CRecoveryWindow : : FindFiles ( const QString & Folder )
{
//int Count = 0;
//foreach(const QString & Path, theAPI->GetBoxedPath(m_pBox, Folder))
// Count += FindFiles(Folder, Path, Folder);
//return Count;
2022-05-19 18:53:51 +01:00
return FindFiles ( theAPI - > GetBoxedPath ( m_pBox . data ( ) , Folder ) , Folder , Folder ) . first ;
2021-10-16 16:19:51 +01:00
}
int CRecoveryWindow : : FindBoxFiles ( const QString & Folder )
{
2022-05-19 18:53:51 +01:00
QString RealFolder = theAPI - > GetRealPath ( m_pBox . data ( ) , m_pBox - > GetFileRoot ( ) + Folder ) ;
2021-10-16 16:19:51 +01:00
if ( RealFolder . isEmpty ( ) )
return 0 ;
return FindFiles ( m_pBox - > GetFileRoot ( ) + Folder , RealFolder , RealFolder ) . first ;
}
QPair < int , quint64 > CRecoveryWindow : : FindFiles ( const QString & BoxedFolder , const QString & RealFolder , const QString & Name , const QString & ParentID )
{
int Count = 0 ;
quint64 Size = 0 ;
QDir Dir ( BoxedFolder ) ;
foreach ( const QFileInfo & Info , Dir . entryInfoList ( QDir : : AllEntries ) )
{
QString Name = Info . fileName ( ) ;
if ( Name = = " . " | | Name = = " .. " )
continue ;
QString Path = Info . filePath ( ) . replace ( " / " , " \\ " ) ;
if ( Info . isFile ( ) )
{
QString RealPath = RealFolder + Path . mid ( BoxedFolder . length ( ) ) ;
if ( ! m_NewFiles . contains ( RealPath ) & & ui . chkShowAll - > checkState ( ) = = Qt : : PartiallyChecked )
continue ;
Count + + ;
Size + = Info . size ( ) ;
QVariantMap RecFile ;
RecFile [ " ID " ] = RealPath ;
RecFile [ " ParentID " ] = RealFolder ;
RecFile [ " FileName " ] = Name ;
RecFile [ " FileSize " ] = FormatSize ( Info . size ( ) ) ;
RecFile [ " DiskPath " ] = RealPath ;
RecFile [ " BoxPath " ] = Path ;
RecFile [ " Icon " ] = m_IconProvider . icon ( Info ) ;
RecFile [ " IsDir " ] = false ;
m_FileMap . insert ( RealPath , RecFile ) ;
}
else
{
auto CountSize = FindFiles ( Path , RealFolder + " \\ " + Name , Name , RealFolder ) ;
Count + = CountSize . first ;
Size + = CountSize . second ;
}
}
if ( Count > 0 )
{
QVariantMap RecFolder ;
RecFolder [ " ID " ] = RealFolder ;
RecFolder [ " ParentID " ] = ParentID ;
RecFolder [ " FileName " ] = Name ;
RecFolder [ " FileSize " ] = FormatSize ( Size ) ;
RecFolder [ " DiskPath " ] = RealFolder ;
RecFolder [ " BoxPath " ] = BoxedFolder ;
RecFolder [ " Icon " ] = m_IconProvider . icon ( QFileIconProvider : : Folder ) ;
RecFolder [ " IsDir " ] = true ;
m_FileMap . insert ( RealFolder , RecFolder ) ;
}
return qMakePair ( Count , Size ) ;
}
2022-07-14 09:46:13 +01:00
QMap < QString , CRecoveryWindow : : SRecItem > CRecoveryWindow : : GetFiles ( )
2021-10-16 16:19:51 +01:00
{
//bool HasShare = false;
2022-05-08 14:26:59 +01:00
QMap < QString , SRecItem > FileMap ;
2021-10-16 16:19:51 +01:00
foreach ( const QModelIndex & Index , ui . treeFiles - > selectionModel ( ) - > selectedIndexes ( ) )
{
QModelIndex ModelIndex = m_pSortProxy - > mapToSource ( Index ) ;
QVariant ID = m_pFileModel - > GetItemID ( ModelIndex ) ;
//QVariant ID = m_pFileModel->GetItemID(Index);
QVariantMap File = m_FileMap . value ( ID ) ;
if ( File . isEmpty ( ) )
continue ;
if ( File [ " IsDir " ] . toBool ( ) = = false )
{
//if (File["DiskPath"].toString().indexOf("\\device\\mup", 0, Qt::CaseInsensitive) == 0)
// HasShare = true;
2022-05-08 14:26:59 +01:00
QString CurPath = File [ " DiskPath " ] . toString ( ) ; ;
FileMap [ File [ " BoxPath " ] . toString ( ) ] . FullPath = CurPath ;
FileMap [ File [ " BoxPath " ] . toString ( ) ] . RelPath = CurPath . mid ( CurPath . lastIndexOf ( " \\ " ) ) ;
2021-10-16 16:19:51 +01:00
}
else
{
2022-05-08 14:26:59 +01:00
QString DirPath = File [ " DiskPath " ] . toString ( ) ;
//if(ModelIndex.parent().isValid())
// DirPath = Split2(DirPath, "\\", true).first;
2021-10-16 16:19:51 +01:00
QList < QModelIndex > Folders ;
Folders . append ( ModelIndex ) ;
do
{
QModelIndex CurIndex = Folders . takeFirst ( ) ;
for ( int i = 0 ; i < m_pFileModel - > rowCount ( CurIndex ) ; i + + )
{
QModelIndex ChildIndex = m_pFileModel - > index ( i , 0 , CurIndex ) ;
QVariant ChildID = m_pFileModel - > GetItemID ( ChildIndex ) ;
QVariantMap File = m_FileMap . value ( ChildID ) ;
if ( File . isEmpty ( ) )
continue ;
if ( File [ " IsDir " ] . toBool ( ) = = false )
{
//if (File["DiskPath"].toString().indexOf("\\device\\mup") == 0)
// HasShare = true;
2022-05-08 14:26:59 +01:00
QString CurPath = File [ " DiskPath " ] . toString ( ) ;
FileMap [ File [ " BoxPath " ] . toString ( ) ] . FullPath = CurPath ;
2022-05-31 18:05:00 +01:00
QString RelPath = CurPath . mid ( Split2 ( DirPath , " \\ " , true ) . first . length ( ) ) ;
2022-05-08 14:26:59 +01:00
if ( RelPath . length ( ) > FileMap [ File [ " BoxPath " ] . toString ( ) ] . RelPath . length ( ) )
FileMap [ File [ " BoxPath " ] . toString ( ) ] . RelPath = RelPath ;
2021-10-16 16:19:51 +01:00
}
else
Folders . append ( ChildIndex ) ;
}
} while ( ! Folders . isEmpty ( ) ) ;
}
}
2022-07-14 09:46:13 +01:00
return FileMap ;
}
void CRecoveryWindow : : RecoverFiles ( bool bBrowse , QString RecoveryFolder )
{
QMap < QString , SRecItem > FileMap = GetFiles ( ) ;
2021-10-16 16:19:51 +01:00
/*if (HasShare && !bBrowse) {
QMessageBox : : warning ( this , " Sandboxie-Plus " , tr ( " One or more selected files are located on a network share, and must be recovered to a local drive, please select a folder to recover all selected files to. " ) ) ;
bBrowse = true ;
} */
if ( bBrowse & & RecoveryFolder . isEmpty ( ) ) {
RecoveryFolder = QFileDialog : : getExistingDirectory ( this , tr ( " Select Directory " ) ) . replace ( " / " , " \\ " ) ;
if ( RecoveryFolder . isEmpty ( ) )
return ;
QStringList RecoverTargets = theAPI - > GetUserSettings ( ) - > GetTextList ( " SbieCtrl_RecoverTarget " , true ) ;
if ( ! RecoverTargets . contains ( RecoveryFolder ) )
theAPI - > GetUserSettings ( ) - > UpdateTextList ( " SbieCtrl_RecoverTarget " , RecoverTargets , true ) ;
}
QList < QPair < QString , QString > > FileList ;
2022-05-08 14:26:59 +01:00
for ( QMap < QString , SRecItem > : : const_iterator I = FileMap . begin ( ) ; I ! = FileMap . end ( ) ; + + I )
2021-10-16 16:19:51 +01:00
{
QString BoxedFilePath = I . key ( ) ;
2022-05-08 14:26:59 +01:00
QString RecoveryPath = I . value ( ) . FullPath ;
2021-10-16 16:19:51 +01:00
if ( ! RecoveryFolder . isEmpty ( ) )
{
2022-05-08 14:26:59 +01:00
//QString FileName = RecoveryPath.mid(RecoveryPath.lastIndexOf("\\") + 1);
//RecoveryPath = RecoveryFolder + "\\" + FileName;
RecoveryPath = RecoveryFolder + I . value ( ) . RelPath ;
2021-10-16 16:19:51 +01:00
}
FileList . append ( qMakePair ( BoxedFilePath , RecoveryPath ) ) ;
}
2023-05-08 20:33:50 +01:00
SB_PROGRESS Status = theGUI - > RecoverFiles ( m_pBox - > GetName ( ) , FileList , this ) ;
2021-10-16 16:19:51 +01:00
if ( Status . GetStatus ( ) = = OP_ASYNC )
{
connect ( Status . GetValue ( ) . data ( ) , SIGNAL ( Finished ( ) ) , this , SLOT ( FindFiles ( ) ) ) ;
theGUI - > AddAsyncOp ( Status . GetValue ( ) ) ;
}
}
void CRecoveryWindow : : OnCount ( quint32 fileCount , quint32 folderCount , quint64 totalSize )
{
ui . lblInfo - > setText ( tr ( " There are %1 files and %2 folders in the sandbox, occupying %3 of disk space. " ) . arg ( fileCount ) . arg ( folderCount ) . arg ( FormatSize ( totalSize ) ) ) ;
2021-12-08 21:40:25 +00:00
m_pCounter - > deleteLater ( ) ;
m_pCounter = NULL ;
2021-10-16 16:19:51 +01:00
}
void CRecoveryWindow : : OnCloseUntil ( )
{
m_pBox . objectCast < CSandBoxPlus > ( ) - > SetSuspendRecovery ( ) ;
close ( ) ;
}
2022-06-04 20:07:04 +01:00
void CRecoveryWindow : : OnAutoDisable ( )
{
2022-06-13 19:12:53 +01:00
m_pBox . objectCast < CSandBoxPlus > ( ) - > SetSuspendRecovery ( ) ;
2022-11-29 15:49:16 +00:00
m_pBox - > SetBoolSafe ( " AutoRecover " , false ) ;
2022-06-04 20:07:04 +01:00
close ( ) ;
}
2021-10-16 16:19:51 +01:00
void CRecoveryCounter : : run ( )
{
quint32 fileCount = 0 ;
quint32 folderCount = 0 ;
quint64 totalSize = 0 ;
QStringList Folders ;
Folders . append ( m_BoxPath ) ;
do {
if ( ! m_run ) break ;
QDir Dir ( Folders . takeFirst ( ) ) ;
foreach ( const QFileInfo & Info , Dir . entryInfoList ( QDir : : AllEntries ) )
{
if ( ! m_run ) break ;
QString Name = Info . fileName ( ) ;
if ( Name = = " . " | | Name = = " .. " )
continue ;
QString Path = Info . filePath ( ) . replace ( " / " , " \\ " ) ;
if ( Info . isFile ( ) )
{
fileCount + + ;
totalSize + = Info . size ( ) ;
}
else
{
Folders . append ( Path ) ;
folderCount + + ;
}
}
} while ( ! Folders . isEmpty ( ) ) ;
2021-12-08 21:40:25 +00:00
emit Count ( fileCount , folderCount , totalSize ) ;
2022-06-14 19:58:13 +01:00
}