289 lines
7.7 KiB
C++
289 lines
7.7 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::LoadGroups()
|
|
{
|
|
ui.treeGroups->clear();
|
|
|
|
QStringList ProcessGroups = m_pBox->GetTextList("ProcessGroup", m_Template);
|
|
foreach(const QString& Group, ProcessGroups)
|
|
{
|
|
QStringList Entries = Group.split(",");
|
|
QString GroupName = Entries.takeFirst();
|
|
|
|
QTreeWidgetItem* pItem = new QTreeWidgetItem();
|
|
pItem->setData(0, Qt::UserRole, GroupName);
|
|
if (GroupName.length() > 2)
|
|
GroupName = GroupName.mid(1, GroupName.length() - 2);
|
|
pItem->setText(0, GroupName);
|
|
for (int i = 0; i < Entries.count(); i++)
|
|
{
|
|
if (Entries[i].isEmpty())
|
|
continue;
|
|
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
|
|
SetProgramItem(Entries[i], pSubItem, 0);
|
|
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
|
|
pItem->addChild(pSubItem);
|
|
}
|
|
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
|
|
ui.treeGroups->addTopLevelItem(pItem);
|
|
}
|
|
|
|
LoadGroupsTmpl();
|
|
|
|
ui.treeGroups->expandAll();
|
|
|
|
m_GroupsChanged = false;
|
|
}
|
|
|
|
void COptionsWindow::LoadGroupsTmpl(bool bUpdate)
|
|
{
|
|
if (ui.chkShowGroupTmpl->isChecked())
|
|
{
|
|
foreach(const QString& Template, m_pBox->GetTemplates())
|
|
{
|
|
foreach(const QString& Group, m_pBox->GetTextListTmpl("ProcessGroup", Template))
|
|
{
|
|
QStringList Entries = Group.split(",");
|
|
QString GroupName = Entries.takeFirst();
|
|
|
|
|
|
QTreeWidgetItem* pItem = new QTreeWidgetItem();
|
|
if (GroupName.length() > 2)
|
|
GroupName = GroupName.mid(1, GroupName.length() - 2);
|
|
pItem->setText(0, GroupName + " (" + Template + ")");
|
|
for (int i = 0; i < Entries.count(); i++)
|
|
{
|
|
if (Entries[i].isEmpty())
|
|
continue;
|
|
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
|
|
SetProgramItem(Entries[i], pSubItem, 0);
|
|
pItem->addChild(pSubItem);
|
|
}
|
|
ui.treeGroups->addTopLevelItem(pItem);
|
|
}
|
|
}
|
|
|
|
if (bUpdate)
|
|
ui.treeGroups->expandAll();
|
|
}
|
|
else if (bUpdate)
|
|
{
|
|
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); )
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
|
|
QString GroupName = pItem->data(0, Qt::UserRole).toString();
|
|
if (GroupName.isEmpty()) {
|
|
delete pItem;
|
|
continue; // entry from template
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
}
|
|
|
|
QStringList COptionsWindow::GetCurrentGroups()
|
|
{
|
|
QStringList Groups;
|
|
|
|
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++)
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
|
|
QString GroupName = pItem->data(0, Qt::UserRole).toString();
|
|
Groups.append(GroupName);
|
|
}
|
|
|
|
foreach(const QString& Template, m_pBox->GetTemplates())
|
|
{
|
|
foreach(const QString& Group, m_pBox->GetTextListTmpl("ProcessGroup", Template))
|
|
{
|
|
QStringList Entries = Group.split(",");
|
|
QString GroupName = Entries.takeFirst();
|
|
if (!Groups.contains(GroupName))
|
|
Groups.append(GroupName);
|
|
}
|
|
}
|
|
|
|
return Groups;
|
|
}
|
|
|
|
void COptionsWindow::SaveGroups()
|
|
{
|
|
QStringList ProcessGroups;
|
|
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++)
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
|
|
QString GroupName = pItem->data(0, Qt::UserRole).toString();
|
|
if (GroupName.isEmpty())
|
|
continue; // this is a template entry, dont save
|
|
QStringList Programs;
|
|
for (int j = 0; j < pItem->childCount(); j++)
|
|
Programs.append(pItem->child(j)->data(0, Qt::UserRole).toString());
|
|
QString Group = GroupName + "," + Programs.join(",");
|
|
ProcessGroups.append(Group);
|
|
}
|
|
|
|
WriteTextList("ProcessGroup", ProcessGroups);
|
|
|
|
m_GroupsChanged = false;
|
|
}
|
|
|
|
void COptionsWindow::OnAddGroup()
|
|
{
|
|
QString Value = QInputDialog::getText(this, "Sandboxie-Plus", tr("Please enter a name for the new group"), QLineEdit::Normal, "NewGroup");
|
|
if (Value.isEmpty())
|
|
return;
|
|
|
|
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++) {
|
|
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
|
|
if (pItem->text(0).compare(Value, Qt::CaseInsensitive) == 0)
|
|
return;
|
|
}
|
|
|
|
QTreeWidgetItem* pItem = new QTreeWidgetItem();
|
|
pItem->setText(0, Value);
|
|
pItem->setData(0, Qt::UserRole, "<" + Value + ">");
|
|
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
|
|
ui.treeGroups->addTopLevelItem(pItem);
|
|
|
|
m_GroupsChanged = true;
|
|
OnOptChanged();
|
|
}
|
|
|
|
void COptionsWindow::AddProgramToGroup(const QString& Program, const QString& Group)
|
|
{
|
|
QTreeWidgetItem* pItem = FindGroupByName(Group, true);
|
|
|
|
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
|
|
SetProgramItem(Program, pSubItem, 0);
|
|
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
|
|
pItem->addChild(pSubItem);
|
|
|
|
m_GroupsChanged = true;
|
|
OnOptChanged();
|
|
}
|
|
|
|
bool COptionsWindow::DelProgramFromGroup(const QString& Program, const QString& Group)
|
|
{
|
|
QTreeWidgetItem* pItem = FindGroupByName(Group, false);
|
|
if (!pItem) return false;
|
|
|
|
bool bFound = false;
|
|
for (int j = 0; j < pItem->childCount(); j++){
|
|
QTreeWidgetItem* pProgItem = pItem->child(j);
|
|
if (pProgItem->data(0, Qt::UserRole).toString().compare(Program, Qt::CaseInsensitive) == 0) {
|
|
delete pProgItem;
|
|
bFound = true;
|
|
if (pItem->childCount() == 0)
|
|
delete pItem;
|
|
break;
|
|
}
|
|
}
|
|
|
|
m_GroupsChanged = true;
|
|
OnOptChanged();
|
|
|
|
return bFound;
|
|
}
|
|
|
|
QTreeWidgetItem* COptionsWindow::FindGroupByName(const QString& Group, bool bAdd)
|
|
{
|
|
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++) {
|
|
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
|
|
if (pItem->data(0, Qt::UserRole).toString().compare(Group, Qt::CaseInsensitive) == 0)
|
|
return pItem;
|
|
}
|
|
|
|
if (bAdd)
|
|
{
|
|
QString GroupName = Group;
|
|
QTreeWidgetItem* pItem = new QTreeWidgetItem();
|
|
pItem->setData(0, Qt::UserRole, GroupName);
|
|
if (GroupName.length() > 2)
|
|
GroupName = GroupName.mid(1, GroupName.length() - 2);
|
|
pItem->setText(0, GroupName);
|
|
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
|
|
ui.treeGroups->addTopLevelItem(pItem);
|
|
return pItem;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void COptionsWindow::OnAddProg()
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeGroups->currentItem();
|
|
while (pItem && pItem->parent())
|
|
pItem = pItem->parent();
|
|
|
|
if (!pItem)
|
|
{
|
|
QMessageBox::warning(this, "SandboxiePlus", tr("Please select group first."));
|
|
return;
|
|
}
|
|
|
|
if (pItem->data(0, Qt::UserRole).toString().isEmpty()) {
|
|
QMessageBox::warning(this, "SandboxiePlus", tr("Template values can not be edited."));
|
|
return;
|
|
}
|
|
|
|
QString Value = SelectProgram();
|
|
if (Value.isEmpty())
|
|
return;
|
|
|
|
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
|
|
SetProgramItem(Value, pSubItem, 0);
|
|
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
|
|
pItem->addChild(pSubItem);
|
|
|
|
m_GroupsChanged = true;
|
|
OnOptChanged();
|
|
}
|
|
|
|
void COptionsWindow::OnDelProg()
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeGroups->currentItem();
|
|
if (!pItem)
|
|
return;
|
|
|
|
if ((pItem->parent() ? pItem->parent() : pItem)->data(0, Qt::UserRole).toString().isEmpty()) {
|
|
QMessageBox::warning(this, "SandboxiePlus", tr("Template values can not be edited."));
|
|
return;
|
|
}
|
|
|
|
delete pItem;
|
|
|
|
m_GroupsChanged = true;
|
|
OnOptChanged();
|
|
}
|
|
|
|
void COptionsWindow::CopyGroupToList(const QString& Groupe, QTreeWidget* pTree, bool disabled)
|
|
{
|
|
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++)
|
|
{
|
|
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
|
|
if (pItem->data(0, Qt::UserRole).toString().compare(Groupe, Qt::CaseInsensitive) == 0)
|
|
{
|
|
for (int j = 0; j < pItem->childCount(); j++)
|
|
{
|
|
QString Value = pItem->child(j)->data(0, Qt::UserRole).toString();
|
|
|
|
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
|
|
pSubItem->setCheckState(0, disabled ? Qt::Unchecked : Qt::Checked);
|
|
SetProgramItem(Value, pSubItem, 0);
|
|
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
|
|
pTree->addTopLevelItem(pSubItem);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|