Vladimir Golovnev
6 years ago
committed by
GitHub
37 changed files with 646 additions and 370 deletions
@ -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 @@ |
|||||||
|
/*
|
||||||
|
* 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 @@ |
|||||||
|
/*
|
||||||
|
* 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