修复BUG 新增黑名单

修复在添加白名单时添加一条‘目标端口:任意’的BUG
新增黑名单增删功能,增加黑名单时会先将其从白名单中删除
This commit is contained in:
筱傑
2019-02-19 11:37:40 +08:00
committed by GitHub
parent 6a70501baa
commit 64ee5e54d0
13 changed files with 802 additions and 46 deletions

25
dal.cpp
View File

@@ -88,6 +88,16 @@ QList<WhiteListItem> DAL::getWhiteList(QString ip)
return whitelist;
}
void DAL::removeFromWhiteList(QString ip)
{
QString sql = QString("DELETE FROM whitelist WHERE IP=?");
QSqlQuery query;
query.prepare(sql);
query.addBindValue(ip);
if(!query.exec())
qCritical()<<query.lastError();
}
void DAL::removeFromWhiteList(QString ip, QList<int> ports)
{
QString portStr = QString("%1").arg(ports[0]);
@@ -130,7 +140,10 @@ bool DAL::addItemToBlackList(int port)
{
return addItemToBlackList("any", port);
}
bool DAL::addItemToBlackList(QString ip)
{
return addItemToBlackList(ip, -1);
}
bool DAL::addItemToBlackList(QString ip, int port)
{
QSqlQuery query;
@@ -145,6 +158,16 @@ bool DAL::addItemToBlackList(QString ip, int port)
return true;
}
void DAL::removeFromBlackList(QString ip)
{
QString sql = QString("DELETE FROM blacklist WHERE IP=?");
QSqlQuery query;
query.prepare(sql);
query.addBindValue(ip);
if(!query.exec())
qCritical()<<query.lastError();
}
QString DAL::getPortList()
{
QSqlQuery query;