Vladimir Golovnev (Glassez)
5 years ago
10 changed files with 350 additions and 139 deletions
@ -0,0 +1,74 @@
@@ -0,0 +1,74 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2020 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 "nativesessionextension.h" |
||||
|
||||
#include <libtorrent/alert_types.hpp> |
||||
|
||||
#include "nativetorrentextension.h" |
||||
|
||||
namespace |
||||
{ |
||||
void handleFastresumeRejectedAlert(const lt::fastresume_rejected_alert *alert) |
||||
{ |
||||
if (alert->error.value() == lt::errors::mismatching_file_size) { |
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
alert->handle.auto_managed(false); |
||||
#else |
||||
alert->handle.unset_flags(lt::torrent_flags::auto_managed); |
||||
#endif |
||||
alert->handle.pause(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
#if (LIBTORRENT_VERSION_NUM >= 10200) |
||||
lt::feature_flags_t NativeSessionExtension::implemented_features() |
||||
{ |
||||
return alert_feature; |
||||
} |
||||
|
||||
std::shared_ptr<lt::torrent_plugin> NativeSessionExtension::new_torrent(const lt::torrent_handle &torrentHandle, void *) |
||||
{ |
||||
return std::make_shared<NativeTorrentExtension>(torrentHandle); |
||||
} |
||||
#else |
||||
boost::shared_ptr<lt::torrent_plugin> NativeSessionExtension::new_torrent(const lt::torrent_handle &torrentHandle, void *) |
||||
{ |
||||
return boost::shared_ptr<lt::torrent_plugin> {new NativeTorrentExtension {torrentHandle}}; |
||||
} |
||||
#endif |
||||
|
||||
void NativeSessionExtension::on_alert(const lt::alert *alert) |
||||
{ |
||||
switch (alert->type()) { |
||||
case lt::fastresume_rejected_alert::alert_type: |
||||
handleFastresumeRejectedAlert(static_cast<const lt::fastresume_rejected_alert *>(alert)); |
||||
break; |
||||
} |
||||
} |
@ -0,0 +1,43 @@
@@ -0,0 +1,43 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2020 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/extensions.hpp> |
||||
#include <libtorrent/version.hpp> |
||||
|
||||
class NativeSessionExtension : public lt::plugin |
||||
{ |
||||
#if (LIBTORRENT_VERSION_NUM >= 10200) |
||||
lt::feature_flags_t implemented_features() override; |
||||
std::shared_ptr<lt::torrent_plugin> new_torrent(const lt::torrent_handle &torrentHandle, void *userData) override; |
||||
#else |
||||
boost::shared_ptr<lt::torrent_plugin> new_torrent(const lt::torrent_handle &torrentHandle, void *userData) override; |
||||
#endif |
||||
void on_alert(const lt::alert *alert) override; |
||||
}; |
@ -0,0 +1,94 @@
@@ -0,0 +1,94 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2020 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 "nativetorrentextension.h" |
||||
|
||||
#include <libtorrent/torrent_status.hpp> |
||||
|
||||
namespace |
||||
{ |
||||
bool isPaused(const lt::torrent_status &torrentStatus) |
||||
{ |
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
return (torrentStatus.paused && !torrentStatus.auto_managed); |
||||
#else |
||||
return ((torrentStatus.flags & lt::torrent_flags::paused) |
||||
&& !(torrentStatus.flags & lt::torrent_flags::auto_managed)); |
||||
#endif |
||||
} |
||||
|
||||
bool isAutoManaged(const lt::torrent_status &torrentStatus) |
||||
{ |
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
return torrentStatus.auto_managed; |
||||
#else |
||||
return bool {torrentStatus.flags & lt::torrent_flags::auto_managed}; |
||||
#endif |
||||
} |
||||
} |
||||
|
||||
NativeTorrentExtension::NativeTorrentExtension(const lt::torrent_handle &torrentHandle) |
||||
: m_torrentHandle {torrentHandle} |
||||
{ |
||||
} |
||||
|
||||
// This method is called when state of torrent is changed
|
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
void NativeTorrentExtension::on_state(const int state) |
||||
#else |
||||
void NativeTorrentExtension::on_state(const lt::torrent_status::state_t state) |
||||
#endif |
||||
{ |
||||
// When a torrent enters "checking files" state while paused, we temporarily resume it
|
||||
// (really we just allow libtorrent to resume it by enabling auto management for it).
|
||||
if (state == lt::torrent_status::checking_files) { |
||||
if (isPaused(m_torrentHandle.status({}))) { |
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
m_torrentHandle.stop_when_ready(true); |
||||
m_torrentHandle.auto_managed(true); |
||||
#else |
||||
m_torrentHandle.set_flags(lt::torrent_flags::stop_when_ready | lt::torrent_flags::auto_managed); |
||||
#endif |
||||
} |
||||
} |
||||
} |
||||
|
||||
bool NativeTorrentExtension::on_pause() |
||||
{ |
||||
if (!isAutoManaged(m_torrentHandle.status({}))) { |
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
m_torrentHandle.stop_when_ready(false); |
||||
#else |
||||
m_torrentHandle.unset_flags(lt::torrent_flags::stop_when_ready); |
||||
#endif |
||||
} |
||||
|
||||
// return `false` to allow standard handler
|
||||
// and other extensions to be also invoked.
|
||||
return false; |
||||
} |
@ -0,0 +1,49 @@
@@ -0,0 +1,49 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2020 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/extensions.hpp> |
||||
#include <libtorrent/torrent_handle.hpp> |
||||
#include <libtorrent/version.hpp> |
||||
|
||||
class NativeTorrentExtension : public lt::torrent_plugin |
||||
{ |
||||
public: |
||||
explicit NativeTorrentExtension(const lt::torrent_handle &torrentHandle); |
||||
|
||||
private: |
||||
#if (LIBTORRENT_VERSION_NUM < 10200) |
||||
void on_state(int state) override; |
||||
#else |
||||
void on_state(lt::torrent_status::state_t state) override; |
||||
#endif |
||||
bool on_pause() override; |
||||
|
||||
lt::torrent_handle m_torrentHandle; |
||||
}; |
Loading…
Reference in new issue