324 lines
9.9 KiB
C++
324 lines
9.9 KiB
C++
#include "stdafx.h"
|
|
#include "OptionsWindow.h"
|
|
#include "SandMan.h"
|
|
#include "SettingsWindow.h"
|
|
#include "../MiscHelpers/Common/Settings.h"
|
|
#include "../MiscHelpers/Common/Common.h"
|
|
#include "../MiscHelpers/Common/ComboInputDialog.h"
|
|
#include "../MiscHelpers/Common/SettingsWidgets.h"
|
|
#include "Helpers/WinAdmin.h"
|
|
|
|
|
|
void COptionsWindow::LoadForced()
|
|
{
|
|
ui.treeForced->clear();
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("ForceProcess", m_Template))
|
|
AddForcedEntry(Value, (int)eProcess);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("ForceProcessDisabled", m_Template))
|
|
AddForcedEntry(Value, (int)eProcess, true);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("ForceFolder", m_Template))
|
|
AddForcedEntry(Value, (int)ePath);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("ForceFolderDisabled", m_Template))
|
|
AddForcedEntry(Value, (int)ePath, true);
|
|
|
|
ui.chkDisableForced->setChecked(m_pBox->GetBool("DisableForceRules", false));
|
|
|
|
ui.treeBreakout->clear();
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("BreakoutProcess", m_Template))
|
|
AddBreakoutEntry(Value, (int)eProcess);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("BreakoutProcessDisabled", m_Template))
|
|
AddBreakoutEntry(Value, (int)eProcess, true);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("BreakoutFolder", m_Template))
|
|
AddBreakoutEntry(Value, (int)ePath);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextList("BreakoutFolderDisabled", m_Template))
|
|
AddBreakoutEntry(Value, (int)ePath, true);
|
|
|
|
|
|
LoadForcedTmpl();
|
|
LoadBreakoutTmpl();
|
|
|
|
m_ForcedChanged = false;
|
|
}
|
|
|
|
void COptionsWindow::LoadForcedTmpl(bool bUpdate)
|
|
{
|
|
if (ui.chkShowForceTmpl->isChecked())
|
|
{
|
|
foreach(const QString& Template, m_pBox->GetTemplates())
|
|
{
|
|
foreach(const QString& Value, m_pBox->GetTextListTmpl("ForceProcess", Template))
|
|
AddForcedEntry(Value, (int)eProcess, false, Template);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextListTmpl("ForceFolder", Template))
|
|
AddForcedEntry(Value, (int)ePath, false, Template);
|
|
}
|
|
}
|
|
else if (bUpdate)
|
|
{
|
|
for (int i = 0; i < ui.treeForced->topLevelItemCount(); )
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeForced->topLevelItem(i);
|
|
int Type = pItem->data(0, Qt::UserRole).toInt();
|
|
if (Type == (int)eTemplate) {
|
|
delete pItem;
|
|
continue; // entry from template
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
}
|
|
|
|
void COptionsWindow::LoadBreakoutTmpl(bool bUpdate)
|
|
{
|
|
if (ui.chkShowBreakoutTmpl->isChecked())
|
|
{
|
|
foreach(const QString& Template, m_pBox->GetTemplates())
|
|
{
|
|
foreach(const QString& Value, m_pBox->GetTextListTmpl("BreakoutProcess", Template))
|
|
AddBreakoutEntry(Value, (int)eProcess, false, Template);
|
|
|
|
foreach(const QString& Value, m_pBox->GetTextListTmpl("BreakoutFolder", Template))
|
|
AddBreakoutEntry(Value, (int)ePath, false, Template);
|
|
}
|
|
}
|
|
else if (bUpdate)
|
|
{
|
|
for (int i = 0; i < ui.treeBreakout->topLevelItemCount(); )
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeBreakout->topLevelItem(i);
|
|
int Type = pItem->data(0, Qt::UserRole).toInt();
|
|
if (Type == (int)eTemplate) {
|
|
delete pItem;
|
|
continue; // entry from template
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
}
|
|
|
|
void COptionsWindow::AddForcedEntry(const QString& Name, int type, bool disabled, const QString& Template)
|
|
{
|
|
QTreeWidgetItem* pItem = new QTreeWidgetItem();
|
|
pItem->setCheckState(0, disabled ? Qt::Unchecked : Qt::Checked);
|
|
pItem->setText(0, (type == (int)eProcess ? tr("Process") : tr("Folder")) + (Template.isEmpty() ? "" : (" (" + Template + ")")));
|
|
pItem->setData(0, Qt::UserRole, Template.isEmpty() ? type : (int)eTemplate);
|
|
SetProgramItem(Name, pItem, (int)eProcess);
|
|
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
|
|
ui.treeForced->addTopLevelItem(pItem);
|
|
}
|
|
|
|
void COptionsWindow::AddBreakoutEntry(const QString& Name, int type, bool disabled, const QString& Template)
|
|
{
|
|
QTreeWidgetItem* pItem = new QTreeWidgetItem();
|
|
pItem->setCheckState(0, disabled ? Qt::Unchecked : Qt::Checked);
|
|
pItem->setText(0, (type == (int)eProcess ? tr("Process") : tr("Folder")) + (Template.isEmpty() ? "" : (" (" + Template + ")")));
|
|
pItem->setData(0, Qt::UserRole, Template.isEmpty() ? type : (int)eTemplate);
|
|
SetProgramItem(Name, pItem, (int)eProcess);
|
|
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
|
|
ui.treeBreakout->addTopLevelItem(pItem);
|
|
}
|
|
|
|
void COptionsWindow::SaveForced()
|
|
{
|
|
QStringList ForceProcess;
|
|
QStringList ForceProcessDisabled;
|
|
QStringList ForceFolder;
|
|
QStringList ForceFolderDisabled;
|
|
|
|
for (int i = 0; i < ui.treeForced->topLevelItemCount(); i++)
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeForced->topLevelItem(i);
|
|
int Type = pItem->data(0, Qt::UserRole).toInt();
|
|
if (Type == (int)eTemplate)
|
|
continue; // entry from template
|
|
|
|
if (pItem->checkState(0) == Qt::Checked) {
|
|
switch (Type) {
|
|
case eProcess: ForceProcess.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
case ePath: ForceFolder.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
}
|
|
}
|
|
else {
|
|
switch (Type) {
|
|
case eProcess: ForceProcessDisabled.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
case ePath: ForceFolderDisabled.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
}
|
|
}
|
|
}
|
|
|
|
WriteTextList("ForceProcess", ForceProcess);
|
|
WriteTextList("ForceProcessDisabled", ForceProcessDisabled);
|
|
WriteTextList("ForceFolder", ForceFolder);
|
|
WriteTextList("ForceFolderDisabled", ForceFolderDisabled);
|
|
|
|
WriteAdvancedCheck(ui.chkDisableForced, "DisableForceRules", "y", "");
|
|
|
|
|
|
QStringList BreakoutProcess;
|
|
QStringList BreakoutProcessDisabled;
|
|
QStringList BreakoutFolder;
|
|
QStringList BreakoutFolderDisabled;
|
|
|
|
for (int i = 0; i < ui.treeBreakout->topLevelItemCount(); i++)
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeBreakout->topLevelItem(i);
|
|
int Type = pItem->data(0, Qt::UserRole).toInt();
|
|
if (Type == (int)eTemplate)
|
|
continue; // entry from template
|
|
|
|
if (pItem->checkState(0) == Qt::Checked) {
|
|
switch (Type) {
|
|
case eProcess: BreakoutProcess.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
case ePath: BreakoutFolder.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
}
|
|
}
|
|
else {
|
|
switch (Type) {
|
|
case eProcess: BreakoutProcessDisabled.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
case ePath: BreakoutFolderDisabled.append(pItem->data(1, Qt::UserRole).toString()); break;
|
|
}
|
|
}
|
|
}
|
|
|
|
WriteTextList("BreakoutProcess", BreakoutProcess);
|
|
WriteTextList("BreakoutProcessDisabled", BreakoutProcessDisabled);
|
|
WriteTextList("BreakoutFolder", BreakoutFolder);
|
|
WriteTextList("BreakoutFolderDisabled", BreakoutFolderDisabled);
|
|
|
|
|
|
m_ForcedChanged = false;
|
|
}
|
|
|
|
void COptionsWindow::OnForceProg()
|
|
{
|
|
QString Value = SelectProgram();
|
|
if (Value.isEmpty())
|
|
return;
|
|
if (!CheckForcedItem(Value, (int)eProcess))
|
|
return;
|
|
AddForcedEntry(Value, (int)eProcess);
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnBreakoutProg()
|
|
{
|
|
QString Value = SelectProgram();
|
|
if (Value.isEmpty())
|
|
return;
|
|
AddBreakoutEntry(Value, (int)eProcess);
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnForceBrowse()
|
|
{
|
|
QString Value = QFileDialog::getOpenFileName(this, tr("Select Executable File"), "", tr("Executable Files (*.exe)"));
|
|
if (Value.isEmpty())
|
|
return;
|
|
if (!CheckForcedItem(Value, (int)eProcess))
|
|
return;
|
|
AddForcedEntry(Split2(Value, "/", true).second, (int)eProcess);
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnBreakoutBrowse()
|
|
{
|
|
QString Value = QFileDialog::getOpenFileName(this, tr("Select Executable File"), "", tr("Executable Files (*.exe)"));
|
|
if (Value.isEmpty())
|
|
return;
|
|
AddBreakoutEntry(Split2(Value, "/", true).second, (int)eProcess);
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnForceDir()
|
|
{
|
|
QString Value = QFileDialog::getExistingDirectory(this, tr("Select Directory")).replace("/", "\\");
|
|
if (Value.isEmpty())
|
|
return;
|
|
if (!CheckForcedItem(Value, (int)ePath))
|
|
return;
|
|
AddForcedEntry(Value, (int)ePath);
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnBreakoutDir()
|
|
{
|
|
QString Value = QFileDialog::getExistingDirectory(this, tr("Select Directory")).replace("/", "\\");
|
|
if (Value.isEmpty())
|
|
return;
|
|
AddBreakoutEntry(Value, (int)ePath);
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnDelForce()
|
|
{
|
|
DeleteAccessEntry(ui.treeForced->currentItem());
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnDelBreakout()
|
|
{
|
|
DeleteAccessEntry(ui.treeBreakout->currentItem());
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnForcedChanged()
|
|
{
|
|
m_ForcedChanged = true;
|
|
OnOptChanged();
|
|
}
|
|
|
|
bool COptionsWindow::CheckForcedItem(const QString& Value, int type)
|
|
{
|
|
bool bDangerous = false;
|
|
|
|
QString winPath = QString::fromUtf8(qgetenv("SystemRoot"));
|
|
|
|
if (type == eProcess)
|
|
{
|
|
if (Value.compare("explorer.exe", Qt::CaseInsensitive) == 0 || Value.compare(winPath + "\\explorer.exe", Qt::CaseInsensitive) == 0)
|
|
bDangerous = true;
|
|
else if (Value.compare("taskmgr.exe", Qt::CaseInsensitive) == 0 || Value.compare(winPath + "\\system32\\taskmgr.exe", Qt::CaseInsensitive) == 0)
|
|
bDangerous = true;
|
|
else if (Value.contains("sbiesvc.exe", Qt::CaseInsensitive))
|
|
bDangerous = true;
|
|
else if (Value.contains("sandman.exe", Qt::CaseInsensitive))
|
|
bDangerous = true;
|
|
}
|
|
else
|
|
{
|
|
if (Value.compare(winPath.left(3), Qt::CaseInsensitive) == 0)
|
|
bDangerous = true; // SystemDrive (C:\)
|
|
else if (Value.compare(winPath, Qt::CaseInsensitive) == 0)
|
|
bDangerous = true; // SystemRoot (C:\Windows)
|
|
else if (Value.left(winPath.length() + 1).compare(winPath + "\\", Qt::CaseInsensitive) == 0)
|
|
bDangerous = true; // sub path of C:\Windows
|
|
}
|
|
|
|
if (bDangerous && QMessageBox::warning(this, "Sandboxie-Plus", tr("Forcing the specified entry will most likely break Windows, are you sure you want to proceed?")
|
|
, QDialogButtonBox::Yes, QDialogButtonBox::No) != QDialogButtonBox::Yes)
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
void COptionsWindow::OnForcedChanged(QTreeWidgetItem *pItem, int)
|
|
{
|
|
QString Value = pItem->data(1, Qt::UserRole).toString();
|
|
if (pItem->checkState(0) == Qt::Checked && !CheckForcedItem(Value, pItem->data(0, Qt::UserRole).toInt()))
|
|
pItem->setCheckState(0, Qt::Unchecked);
|
|
//qDebug() << Test;
|
|
OnForcedChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnBreakoutChanged(QTreeWidgetItem *pItem, int)
|
|
{
|
|
OnForcedChanged();
|
|
}
|