Vladimir Golovnev
6 years ago
committed by
GitHub
37 changed files with 646 additions and 370 deletions
@ -0,0 +1,81 @@
@@ -0,0 +1,81 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2019 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#pragma once |
||||
|
||||
#include <libtorrent/version.hpp> |
||||
|
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
namespace libtorrent |
||||
{ |
||||
class entry; |
||||
class session; |
||||
struct ip_filter; |
||||
struct settings_pack; |
||||
struct torrent_handle; |
||||
|
||||
class alert; |
||||
struct add_torrent_alert; |
||||
struct external_ip_alert; |
||||
struct fastresume_rejected_alert; |
||||
struct file_completed_alert; |
||||
struct file_error_alert; |
||||
struct file_rename_failed_alert; |
||||
struct file_renamed_alert; |
||||
struct listen_failed_alert; |
||||
struct listen_succeeded_alert; |
||||
struct metadata_received_alert; |
||||
struct peer_ban_alert; |
||||
struct peer_blocked_alert; |
||||
struct portmap_alert; |
||||
struct portmap_error_alert; |
||||
struct save_resume_data_alert; |
||||
struct save_resume_data_failed_alert; |
||||
struct session_stats_alert; |
||||
struct state_update_alert; |
||||
struct stats_alert; |
||||
struct storage_moved_alert; |
||||
struct storage_moved_failed_alert; |
||||
struct torrent_alert; |
||||
struct torrent_checked_alert; |
||||
struct torrent_delete_failed_alert; |
||||
struct torrent_deleted_alert; |
||||
struct torrent_finished_alert; |
||||
struct torrent_paused_alert; |
||||
struct torrent_removed_alert; |
||||
struct torrent_resumed_alert; |
||||
struct tracker_error_alert; |
||||
struct tracker_reply_alert; |
||||
struct tracker_warning_alert; |
||||
struct url_seed_alert; |
||||
} |
||||
|
||||
namespace lt = libtorrent; |
||||
#else |
||||
#include <libtorrent/fwd.hpp> |
||||
#endif |
||||
@ -0,0 +1,114 @@
@@ -0,0 +1,114 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2019 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#include "portforwarderimpl.h" |
||||
|
||||
#include <QDebug> |
||||
|
||||
#include <libtorrent/session.hpp> |
||||
|
||||
#include "base/logger.h" |
||||
#include "base/settingsstorage.h" |
||||
|
||||
const QString KEY_ENABLED = QStringLiteral("Network/PortForwardingEnabled"); |
||||
|
||||
PortForwarderImpl::PortForwarderImpl(libtorrent::session *provider, QObject *parent) |
||||
: Net::PortForwarder {parent} |
||||
, m_active {SettingsStorage::instance()->loadValue(KEY_ENABLED, true).toBool()} |
||||
, m_provider {provider} |
||||
{ |
||||
if (m_active) |
||||
start(); |
||||
} |
||||
|
||||
PortForwarderImpl::~PortForwarderImpl() |
||||
{ |
||||
stop(); |
||||
} |
||||
|
||||
bool PortForwarderImpl::isEnabled() const |
||||
{ |
||||
return m_active; |
||||
} |
||||
|
||||
void PortForwarderImpl::setEnabled(const bool enabled) |
||||
{ |
||||
if (m_active != enabled) { |
||||
if (enabled) |
||||
start(); |
||||
else |
||||
stop(); |
||||
|
||||
m_active = enabled; |
||||
SettingsStorage::instance()->storeValue(KEY_ENABLED, enabled); |
||||
} |
||||
} |
||||
|
||||
void PortForwarderImpl::addPort(const quint16 port) |
||||
{ |
||||
if (!m_mappedPorts.contains(port)) { |
||||
m_mappedPorts.insert(port, {}); |
||||
if (isEnabled()) |
||||
m_mappedPorts[port] = {m_provider->add_port_mapping(lt::session::tcp, port, port)}; |
||||
} |
||||
} |
||||
|
||||
void PortForwarderImpl::deletePort(const quint16 port) |
||||
{ |
||||
if (m_mappedPorts.contains(port)) { |
||||
if (isEnabled()) { |
||||
for (const LTPortMapping &portMapping : m_mappedPorts[port]) |
||||
m_provider->delete_port_mapping(portMapping); |
||||
} |
||||
m_mappedPorts.remove(port); |
||||
} |
||||
} |
||||
|
||||
void PortForwarderImpl::start() |
||||
{ |
||||
qDebug("Enabling UPnP / NAT-PMP"); |
||||
lt::settings_pack settingsPack = m_provider->get_settings(); |
||||
settingsPack.set_bool(lt::settings_pack::enable_upnp, true); |
||||
settingsPack.set_bool(lt::settings_pack::enable_natpmp, true); |
||||
m_provider->apply_settings(settingsPack); |
||||
for (auto i = m_mappedPorts.begin(); i != m_mappedPorts.end(); ++i) { |
||||
// quint16 port = i.key();
|
||||
i.value() = {m_provider->add_port_mapping(lt::session::tcp, i.key(), i.key())}; |
||||
} |
||||
LogMsg(tr("UPnP / NAT-PMP support [ON]"), Log::INFO); |
||||
} |
||||
|
||||
void PortForwarderImpl::stop() |
||||
{ |
||||
qDebug("Disabling UPnP / NAT-PMP"); |
||||
lt::settings_pack settingsPack = m_provider->get_settings(); |
||||
settingsPack.set_bool(lt::settings_pack::enable_upnp, false); |
||||
settingsPack.set_bool(lt::settings_pack::enable_natpmp, false); |
||||
m_provider->apply_settings(settingsPack); |
||||
LogMsg(tr("UPnP / NAT-PMP support [OFF]"), Log::INFO); |
||||
} |
||||
@ -0,0 +1,65 @@
@@ -0,0 +1,65 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2019 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#pragma once |
||||
|
||||
#include <vector> |
||||
#include <QHash> |
||||
|
||||
#include "base/net/portforwarder.h" |
||||
#include "libtorrentfwd.h" |
||||
|
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
using LTPortMapping = int; |
||||
#else |
||||
#include <libtorrent/portmap.hpp> |
||||
using LTPortMapping = lt::port_mapping_t; |
||||
#endif |
||||
|
||||
class PortForwarderImpl : public Net::PortForwarder |
||||
{ |
||||
Q_DISABLE_COPY(PortForwarderImpl) |
||||
|
||||
public: |
||||
explicit PortForwarderImpl(lt::session *provider, QObject *parent = nullptr); |
||||
~PortForwarderImpl() override; |
||||
|
||||
bool isEnabled() const override; |
||||
void setEnabled(bool enabled) override; |
||||
|
||||
void addPort(quint16 port) override; |
||||
void deletePort(quint16 port) override; |
||||
|
||||
private: |
||||
void start(); |
||||
void stop(); |
||||
|
||||
bool m_active; |
||||
libtorrent::session *m_provider; |
||||
QHash<quint16, std::vector<LTPortMapping>> m_mappedPorts; |
||||
}; |
||||
Loading…
Reference in new issue