1 /*
2 * Bittorrent Client using Qt and libtorrent.
3 * Copyright (C) 2016 Alexandr Milovantsev <dzmat@yandex.ru>
4 *
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU General Public License
7 * as published by the Free Software Foundation; either version 2
8 * of the License, or (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 *
19 * In addition, as a special exception, the copyright holders give permission to
20 * link this program with the OpenSSL project's "OpenSSL" library (or with
21 * modified versions of it that use the same license as the "OpenSSL" library),
22 * and distribute the linked executables. You must obey the GNU General Public
23 * License in all respects for all of the code used other than "OpenSSL". If you
24 * modify file(s), you may extend this exception to your version of the file(s),
25 * but you are not obligated to do so. If you do not wish to do so, delete this
26 * exception statement from your version.
27 */
28
29 #include "banlistoptionsdialog.h"
30
31 #include <QHostAddress>
32 #include <QMessageBox>
33 #include <QSortFilterProxyModel>
34 #include <QStringListModel>
35
36 #include "base/bittorrent/session.h"
37 #include "base/utils/net.h"
38 #include "ui_banlistoptionsdialog.h"
39 #include "utils.h"
40
41 #define SETTINGS_KEY(name) "BanListOptionsDialog/" name
42
BanListOptionsDialog(QWidget * parent)43 BanListOptionsDialog::BanListOptionsDialog(QWidget *parent)
44 : QDialog(parent)
45 , m_ui(new Ui::BanListOptionsDialog)
46 , m_storeDialogSize(SETTINGS_KEY("Size"))
47 , m_model(new QStringListModel(BitTorrent::Session::instance()->bannedIPs(), this))
48 {
49 m_ui->setupUi(this);
50
51 m_sortFilter = new QSortFilterProxyModel(this);
52 m_sortFilter->setDynamicSortFilter(true);
53 m_sortFilter->setSourceModel(m_model);
54
55 m_ui->bannedIPList->setModel(m_sortFilter);
56 m_ui->bannedIPList->sortByColumn(0, Qt::AscendingOrder);
57 m_ui->buttonBanIP->setEnabled(false);
58
59 Utils::Gui::resize(this, m_storeDialogSize);
60 }
61
~BanListOptionsDialog()62 BanListOptionsDialog::~BanListOptionsDialog()
63 {
64 m_storeDialogSize = size();
65 delete m_ui;
66 }
67
on_buttonBox_accepted()68 void BanListOptionsDialog::on_buttonBox_accepted()
69 {
70 if (m_modified)
71 {
72 // save to session
73 QStringList IPList;
74 // Operate on the m_sortFilter to grab the strings in sorted order
75 for (int i = 0; i < m_sortFilter->rowCount(); ++i)
76 {
77 QModelIndex index = m_sortFilter->index(i, 0);
78 IPList << index.data().toString();
79 }
80 BitTorrent::Session::instance()->setBannedIPs(IPList);
81 QDialog::accept();
82 }
83 else
84 {
85 QDialog::reject();
86 }
87 }
88
on_buttonBanIP_clicked()89 void BanListOptionsDialog::on_buttonBanIP_clicked()
90 {
91 QString ip = m_ui->txtIP->text();
92 if (!Utils::Net::isValidIP(ip))
93 {
94 QMessageBox::warning(this, tr("Warning"), tr("The entered IP address is invalid."));
95 return;
96 }
97 // the same IPv6 addresses could be written in different forms;
98 // QHostAddress::toString() result format follows RFC5952;
99 // thus we avoid duplicate entries pointing to the same address
100 ip = QHostAddress(ip).toString();
101 for (int i = 0; i < m_sortFilter->rowCount(); ++i)
102 {
103 QModelIndex index = m_sortFilter->index(i, 0);
104 if (ip == index.data().toString())
105 {
106 QMessageBox::warning(this, tr("Warning"), tr("The entered IP is already banned."));
107 return;
108 }
109 }
110
111 m_model->insertRow(m_model->rowCount());
112 m_model->setData(m_model->index(m_model->rowCount() - 1, 0), ip);
113 m_ui->txtIP->clear();
114 m_modified = true;
115 }
116
on_buttonDeleteIP_clicked()117 void BanListOptionsDialog::on_buttonDeleteIP_clicked()
118 {
119 QModelIndexList selection = m_ui->bannedIPList->selectionModel()->selectedIndexes();
120 std::sort(selection.begin(), selection.end(), [](const QModelIndex &left, const QModelIndex &right)
121 {
122 return (left.row() > right.row());
123 });
124 for (const QModelIndex &i : selection)
125 m_sortFilter->removeRow(i.row());
126
127 m_modified = true;
128 }
129
on_txtIP_textChanged(const QString & ip)130 void BanListOptionsDialog::on_txtIP_textChanged(const QString &ip)
131 {
132 m_ui->buttonBanIP->setEnabled(Utils::Net::isValidIP(ip));
133 }
134