Sandboxie/SandboxiePlus/SandMan/SandMan.h

544 lines
15 KiB
C++

#pragma once
#include <QtWidgets/QMainWindow>
#include "../MiscHelpers/Common/Settings.h"
#include "../MiscHelpers/Common/TreeViewEx.h"
#include "../MiscHelpers/Common/PanelView.h"
#include "../MiscHelpers/Common/ProgressDialog.h"
#include <QTranslator>
#include "../version.h"
//#include "../QSbieAPI/SbieAPI.h"
#include "SbiePlusAPI.h"
class CPopUpWindow;
class CSbieView;
class CFileView;
class CBoxBorder;
class CSbieTemplatesEx;
class CTraceView;
class CScriptManager;
class CAddonManager;
struct ToolBarAction {
// Identifier of action stored in ini. Empty for separator.
QString scriptName = "";
// Not owned. Null for special cases.
QAction* action;
// Display name override for display in toolbar config menu. Empty if no override.
QString nameOverride = "";
};
class CSandMan : public QMainWindow
{
Q_OBJECT
public:
CSandMan(QWidget *parent = Q_NULLPTR);
virtual ~CSandMan();
CSbieTemplatesEx* GetCompat() { return m_SbieTemplates; }
void CheckCompat(QObject* receiver, const char* member);
CScriptManager* GetScripts() { return m_SbieScripts; }
CAddonManager* GetAddonManager() { return m_AddonManager; }
static QString GetVersion(bool bWithUpdates = false);
static void ShowMessageBox(QWidget* Widget, QMessageBox::Icon Icon, const QString& Message);
bool IsImDiskReady() const { return m_ImDiskReady; }
bool IsWFPEnabled() const;
SB_PROGRESS RecoverFiles(const QString& BoxName, const QList<QPair<QString, QString>>& FileList, QWidget* pParent, int Action = 0);
static QStringList GetFileCheckers(const CSandBoxPtr& pBox);
SB_PROGRESS CheckFiles(const QString& BoxName, const QStringList& Files);
enum EDelMode {
eCleanUp,
eAuto,
eForDelete
};
SB_STATUS DeleteBoxContent(const CSandBoxPtr& pBox, EDelMode Mode, bool DeleteSnapshots = true);
void UpdateProcesses();
SB_STATUS AddAsyncOp(const CSbieProgressPtr& pProgress, bool bWait = false, const QString& InitialMsg = QString(), QWidget* pParent = NULL);
static QString FormatError(const SB_STATUS& Error);
void CheckResults(QList<SB_STATUS> Results, QWidget* pParent, bool bAsync = false);
static QIcon GetIcon(const QString& Name, int iAction = 1);
bool IsFullyPortable();
bool IsShowHidden() { return m_pShowHidden && m_pShowHidden->isChecked(); }
bool KeepTerminated();
bool ShowAllSessions() { return m_pShowAllSessions && m_pShowAllSessions->isChecked(); }
bool IsSilentMode();
bool IsDisableRecovery() {return IsSilentMode() || m_pDisableRecovery && m_pDisableRecovery->isChecked();}
bool IsDisableMessages() {return IsSilentMode() || m_pDisableMessages && m_pDisableMessages->isChecked();}
CSbieView* GetBoxView() { return m_pBoxView; }
CFileView* GetFileView() { return m_pFileView; }
QString FormatSbieMessage(quint32 MsgCode, const QStringList& MsgData, QString ProcessName, QString* pLink = NULL);
QString MakeSbieMsgLink(quint32 MsgCode, const QStringList& MsgData, QString ProcessName);
static void SafeShow(QWidget* pWidget);
int SafeExec(QDialog* pDialog);
bool RunSandboxed(const QStringList& Commands, QString BoxName = QString(), const QString& WrkDir = QString(), bool bShowFCP = false);
SB_RESULT(quint32) RunStart(const QString& BoxName, const QString& Command, CSbieAPI::EStartFlags Flags = CSbieAPI::eStartDefault, const QString& WorkingDir = QString(), QProcess* pProcess = NULL);
SB_STATUS ImBoxMount(const CSandBoxPtr& pBox, bool bAutoUnmount = false);
void EditIni(const QString& IniPath, bool bPlus = false);
void UpdateDrives();
void UpdateForceUSB();
QIcon GetBoxIcon(int boxType, bool inUse = false);
QRgb GetBoxColor(int boxType) { return m_BoxColors[boxType]; }
QIcon GetColorIcon(QColor boxColor, bool inUse = false/*, bool bOut = false*/);
QIcon MakeIconBusy(const QIcon& Icon, int Index = 0);
QIcon IconAddOverlay(const QIcon& Icon, const QString& Name, int Size = 24);
QString GetBoxDescription(int boxType);
bool CheckCertificate(QWidget* pWidget, int iType = 0);
bool IsAlwaysOnTop() const;
void UpdateTheme();
void UpdateTitleTheme(const HWND& hwnd);
SB_STATUS ReloadCert(QWidget* pWidget = NULL);
void UpdateCertState();
void SaveMessageLog(QIODevice* pFile);
signals:
void DrivesChanged();
void CertUpdated();
void Closed();
protected:
friend class COnlineUpdater;
SB_RESULT(void*) ConnectSbie();
SB_STATUS ConnectSbieImpl();
SB_STATUS DisconnectSbie();
SB_RESULT(void*) StopSbie(bool andRemove = false);
static void RecoverFilesAsync(QPair<const CSbieProgressPtr&,QWidget*> pParam, const QString& BoxName, const QList<QPair<QString, QString>>& FileList, const QStringList& Checkers, int Action = 0);
static void CheckFilesAsync(const CSbieProgressPtr& pProgress, const QString& BoxName, const QStringList &Files, const QStringList& Checkers);
void AddLogMessage(const QDateTime& TimeStamp, const QString& Message, const QString& Link = QString());
QIcon GetTrayIcon(bool isConnected = true, bool bSun = false);
QString GetTrayText(bool isConnected = true);
void CheckSupport();
void closeEvent(QCloseEvent* e);
void changeEvent(QEvent* e);
void dragEnterEvent(QDragEnterEvent* e);
void dropEvent(QDropEvent* e);
void timerEvent(QTimerEvent* pEvent);
int m_uTimerID;
bool m_bConnectPending;
bool m_bStopPending;
CBoxBorder* m_pBoxBorder;
CSbieTemplatesEx* m_SbieTemplates;
CScriptManager* m_SbieScripts;
CAddonManager* m_AddonManager;
QMap<CSbieProgress*, QPair<CSbieProgressPtr, QPointer<QWidget>>> m_pAsyncProgress;
QMap<QString, QSet<QString>> m_MissingTemplates;
enum EBoxColors
{
eYellow = 0,
eRed,
eGreen,
eBlue,
eCyan,
eMagenta,
eOrang,
eWhite,
eMaxColor
};
QMap<int, QRgb> m_BoxColors;
class UGlobalHotkeys* m_pHotkeyManager;
bool m_ImDiskReady;
struct SSbieMsg {
QDateTime TimeStamp;
quint32 MsgCode;
QStringList MsgData;
QString ProcessName;
};
QVector<SSbieMsg> m_MessageLog;
public slots:
void OnBoxSelected();
void OnMessage(const QString& MsgData);
void OnStatusChanged();
void OnLogMessage(const QString& Message, bool bNotify = false);
void OnLogSbieMessage(quint32 MsgCode, const QStringList& MsgData, quint32 ProcessId);
void OnNotAuthorized(bool bLoginRequired, bool& bRetry);
void OnQueuedRequest(quint32 ClientPid, quint32 ClientTid, quint32 RequestId, const QVariantMap& Data);
void OnFileToRecover(const QString& BoxName, const QString& FilePath, const QString& BoxPath, quint32 ProcessId);
void OnFileRecovered(const QString& BoxName, const QString& FilePath, const QString& BoxPath);
bool OpenRecovery(const CSandBoxPtr& pBox, bool& DeleteSnapshots, bool bCloseEmpty = false);
class CRecoveryWindow* ShowRecovery(const CSandBoxPtr& pBox, bool bFind = true);
void TryFix(quint32 MsgCode, const QStringList& MsgData, const QString& ProcessName, const QString& BoxName);
void OpenCompat();
void UpdateSettings(bool bRebuildUI);
void RebuildUI();
void OnIniReloaded();
void SetupHotKeys();
void OnHotKey(size_t id);
void OnAsyncFinished();
void OnAsyncFinished(CSbieProgress* pProgress);
void OnAsyncMessage(const QString& Text);
void OnAsyncProgress(int Progress);
void OnCancelAsync();
void OnBoxAdded(const CSandBoxPtr& pBox);
void OnBoxClosed(const CSandBoxPtr& pBox);
void OnBoxCleaned(CSandBoxPlus* pBoxEx);
void OnStartMenuChanged();
void OpenUrl(const QString& url) { OpenUrl(QUrl(url)); }
void OpenUrl(const QUrl& url);
int ShowQuestion(const QString& question, const QString& checkBoxText, bool* checkBoxSetting, int buttons, int defaultButton, int type, QWidget* pParent);
void ShowMessage(const QString& message, int type);
void OnBoxMenu(const QPoint &);
void OnBoxDblClick(QTreeWidgetItem*);
void SyncStartMenu();
void ClearStartMenu();
void UpdateLabel();
private slots:
void OnMenuHover(QAction* action);
void OnBoxMenuHover(QAction* action);
void OnSandBoxAction();
void OnSettingsAction();
void OnPauseAll();
void OnEmptyAll();
void OnLockAll();
void OnWndFinder();
void OnBoxAssistant();
void OnDisableForce();
void OnDisableForce2();
void OnDisablePopUp();
void OnMaintenance();
void OnViewMode(QAction* action);
void OnAlwaysTop();
void OnView(QAction* action);
void OnRefresh();
void OnCleanUp();
void OnProcView();
void OnRecoveryLog();
void OnSettings();
void OnResetMsgs();
void OnResetGUI();
void OnEditIni();
void OnReloadIni();
void OnMonitoring();
void OnSymbolStatus(const QString& Message);
void CheckForUpdates(bool bManual = true);
void OnRestartAsAdmin();
void OnExit();
void OnHelp();
void OnAbout();
void OnShowHide();
void OnSysTray(QSystemTrayIcon::ActivationReason Reason);
void SetUITheme();
void SetTitleTheme(const HWND& hwnd);
void AddLogMessage(const QString& Message);
void AddFileRecovered(const QString& BoxName, const QString& FilePath);
void commitData(QSessionManager& manager);
private:
void CreateUI();
void CreateMenus(bool bAdvanced);
void CreateOldMenus();
void CreateMaintenanceMenu();
void CreateViewBaseMenu();
void CreateHelpMenu(bool bAdvanced);
void CreateToolBar(bool bRebuild);
void CreateLabel();
void CreateView(int iViewMode);
void CreateTrayIcon();
void CreateTrayMenu();
void CreateBoxMenu(QMenu* pMenu, int iOffset = 0, int iSysTrayFilter = 0);
void HandleMaintenance(SB_RESULT(void*) Status);
void LoadState(bool bFull = true);
void StoreState();
void UpdateState();
struct SBoxLink {
QString RelPath; // key
QString FullPath;
QString Target;
};
void EnumBoxLinks(QMap<QString, QMap<QString, SBoxLink> >& BoxLinks, const QString& Prefix, const QString& Folder, bool bWithSubDirs = true);
void CleanupShortcutPath(const QString& Path);
void DeleteShortcut(const QString& Path);
void CleanUpStartMenu(QMap<QString, QMap<QString, SBoxLink> >& BoxLinks);
QSet<QString> GetToolBarItemsConfig();
void SetToolBarItemsConfig(const QSet<QString>& items);
QList<ToolBarAction> GetAvailableToolBarActions();
void CreateToolBarConfigMenu(const QList<ToolBarAction>& actions, const QSet<QString>& currentSet);
void OnToolBarMenuItemClicked(const QString& scriptName);
void OnResetToolBarMenuConfig();
const QString ToolBarConfigKey = "UIConfig/ToolBarItems";
// per 1.9.3 menu. no whitespace!
const QStringList DefaultToolBarItems = QString(
"Settings,KeepTerminated,CleanUpMenu,BrowseFiles,EditIni,EnableMonitor"
).split(',');
QWidget* m_pMainWidget;
QVBoxLayout* m_pMainLayout;
QToolBar* m_pToolBar;
QSplitter* m_pPanelSplitter;
QSplitter* m_pLogSplitter;
QStackedLayout* m_pViewStack;
QComboBox* m_pBoxCombo;
CSbieView* m_pBoxView;
CFileView* m_pFileView;
QTabWidget* m_pLogTabs;
CPanelWidgetEx* m_pMessageLog;
CTraceView* m_pTraceView;
CPanelWidgetEx* m_pRecoveryLog;
class CRecoveryLogWnd* m_pRecoveryLogWnd;
QMenuBar* m_pMenuBar;
QHBoxLayout* m_pMenuLayout;
QMenu* m_pMenuFile;
QAction* m_pRestart;
QAction* m_pRunBoxed;
QAction* m_pNewBox;
QAction* m_pNewGroup;
QAction* m_pImportBox;
QAction* m_pPauseAll;
QAction* m_pEmptyAll;
QAction* m_pLockAll;
QAction* m_pWndFinder;
QAction* m_pDisableForce;
QAction* m_pDisableForce2;
QAction* m_pDisableRecovery;
QAction* m_pDisableMessages;
QAction* m_pDismissUpdate;
QMenu* m_pMaintenance;
QAction* m_pConnect;
QAction* m_pDisconnect;
QMenu* m_pMaintenanceItems;
QAction* m_pInstallDrv;
QAction* m_pStartDrv;
QAction* m_pStopDrv;
QAction* m_pUninstallDrv;
QAction* m_pInstallSvc;
QAction* m_pStartSvc;
QAction* m_pStopSvc;
QAction* m_pUninstallSvc;
QAction* m_pStopAll;
QAction* m_pImDiskCpl;
QAction* m_pUninstallAll;
QAction* m_pSetupWizard;
QAction* m_pExit;
QMenu* m_pMenuView;
QActionGroup* m_pViewMode;
QAction* m_pShowHidden;
QAction* m_pWndTopMost;
QAction* m_pMenuBrowse;
QAction* m_pRefreshAll;
QMenu* m_pCleanUpMenu;
QAction* m_pCleanUpProcesses;
QAction* m_pCleanUpMsgLog;
QAction* m_pCleanUpTrace;
QAction* m_pCleanUpRecovery;
QToolButton* m_pCleanUpButton;
QToolButton* m_pNewBoxButton;
QToolButton* m_pEditIniButton;
//QToolButton* m_pEditButton;
QAction* m_pKeepTerminated;
QAction* m_pShowAllSessions;
QAction* m_pArrangeGroups;
QMenu* m_pMenuOptions;
QAction* m_pMenuSettings;
QAction* m_pMenuResetMsgs;
QAction* m_pMenuResetGUI;
QAction* m_pEditIni;
QAction* m_pEditIni2;
QAction* m_pEditIni3;
QAction* m_pReloadIni;
QAction* m_pEnableMonitoring;
//QMenu* m_pMenuTools;
QAction* m_pBoxAssistant;
QAction* m_pSeparator;
QLabel* m_pLabel;
QMenu* m_pMenuHelp;
//QAction* m_pSupport;
QAction* m_pContribution;
QAction* m_pForum;
QAction* m_pManual;
QAction* m_pUpdate;
QAction* m_pAbout;
QAction* m_pAboutQt;
QLabel* m_pTraceInfo;
QLabel* m_pDisabledForce;
QLabel* m_pDisabledRecovery;
QLabel* m_pDisabledMessages;
QLabel* m_pRamDiskInfo;
// for old menu
QMenu* m_pSandbox;
QSystemTrayIcon* m_pTrayIcon;
QMenu* m_pTrayMenu;
QWidgetAction* m_pTrayList;
QTreeWidget* m_pTrayBoxes;
int m_iTrayPos;
//QMenu* m_pBoxMenu;
bool m_bIconEmpty;
int m_iIconDisabled;
bool m_bIconBusy;
bool m_bIconSun;
int m_iDeletingContent;
bool m_bOnTop;
bool m_bExit;
CProgressDialog* m_pProgressDialog;
bool m_pProgressModal;
CPopUpWindow* m_pPopUpWindow;
bool m_StartMenuUpdatePending;
bool m_ThemeUpdatePending;
QString m_DefaultStyle;
QPalette m_DefaultPalett;
double m_DefaultFontSize;
QPalette m_DarkPalett;
void LoadLanguage();
void LoadLanguage(const QString& Lang, const QString& Module, int Index);
QTranslator m_Translator[2];
public:
class COnlineUpdater*m_pUpdater;
QString m_Language;
quint32 m_LanguageId;
bool m_DarkTheme;
};
class CRecoveryLogWnd : public QDialog
{
Q_OBJECT
public:
CRecoveryLogWnd(QWidget *parent = Q_NULLPTR);
~CRecoveryLogWnd();
private slots:
void OnDblClick(QTreeWidgetItem* pItem);
signals:
void Closed();
protected:
friend class CSandMan;
void closeEvent(QCloseEvent *e);
CPanelWidgetEx* m_pRecoveryLog;
};
#include <QStyledItemDelegate>
class CTreeItemDelegate : public QStyledItemDelegate
{
void paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const
{
QStyleOptionViewItem opt(option);
opt.state &= ~QStyle::State_HasFocus;
QStyledItemDelegate::paint(painter, opt, index);
}
};
class CTreeItemDelegate2 : public CTreeItemDelegate
{
QSize sizeHint(const QStyleOptionViewItem &option, const QModelIndex &index) const
{
QSize size = QStyledItemDelegate::sizeHint(option, index);
size.setHeight(32);
return size;
}
};
extern CSandMan* theGUI;