Fix resume torrents without metadata.

This commit is contained in:
Vladimir Golovnev (Glassez) 2015-06-04 11:03:19 +03:00
parent 0172ab1f50
commit 8cc9c64ff8
6 changed files with 59 additions and 54 deletions

View File

@ -95,7 +95,7 @@ namespace libt = libtorrent;
using namespace BitTorrent;
static bool readFile(const QString &path, QByteArray &buf);
static bool loadTorrentResumeData(const QByteArray &data, AddTorrentData &out);
static bool loadTorrentResumeData(const QByteArray &data, AddTorrentData &out, MagnetUri &magnetUri);
static void torrentQueuePositionUp(const libt::torrent_handle &handle);
static void torrentQueuePositionDown(const libt::torrent_handle &handle);
@ -1086,33 +1086,34 @@ bool Session::addTorrent_impl(const AddTorrentData &addData, const MagnetUri &ma
p = magnetUri.addTorrentParams();
hash = magnetUri.hash();
}
else {
if (!torrentInfo.isValid()) return false;
else if (torrentInfo.isValid()) {
// Metadata
p.ti = torrentInfo.nativeInfo();
hash = torrentInfo.hash();
}
else {
Q_ASSERT(false);
}
if (addData.resumed) {
// Set torrent fast resume data
if (addData.resumed && !fromMagnetUri) {
// Set torrent fast resume data
#if LIBTORRENT_VERSION_NUM < 10000
p.resume_data = &buf;
p.resume_data = &buf;
#else
p.resume_data = buf;
p.flags |= libt::add_torrent_params::flag_use_resume_save_path;
p.resume_data = buf;
p.flags |= libt::add_torrent_params::flag_use_resume_save_path;
#endif
p.flags |= libt::add_torrent_params::flag_merge_resume_trackers;
p.flags |= libt::add_torrent_params::flag_merge_resume_trackers;
}
else {
foreach (int prio, addData.filePriorities)
filePriorities.push_back(prio);
}
else {
foreach (int prio, addData.filePriorities)
filePriorities.push_back(prio);
#if LIBTORRENT_VERSION_NUM < 10000
p.file_priorities = &filePriorities;
p.file_priorities = &filePriorities;
#else
p.file_priorities = filePriorities;
p.file_priorities = filePriorities;
#endif
}
}
// We should not add torrent if it already
@ -1929,22 +1930,15 @@ void Session::startUpTorrents()
resumeDataDir.absoluteFilePath(QString("%1.fastresume.%2").arg(hash).arg(prio));
QByteArray data;
AddTorrentData resumeData;
if (readFile(fastresumePath, data) && loadTorrentResumeData(data, resumeData)) {
MagnetUri magnetUri;
if (readFile(fastresumePath, data) && loadTorrentResumeData(data, resumeData, magnetUri)) {
filePath = resumeDataDir.filePath(QString("%1.torrent").arg(hash));
if (QFile(filePath).exists()) {
qDebug("Starting up torrent %s ...", qPrintable(hash));
if (!addTorrent_impl(resumeData, MagnetUri(), TorrentInfo::loadFromFile(filePath), data))
logger->addMessage(tr("Unable to resume torrent '%1'.", "e.g: Unable to resume torrent 'hash'.")
.arg(Utils::Fs::toNativePath(hash)), Log::CRITICAL);
}
else {
logger->addMessage(tr("Unable to resume torrent '%1': torrent file not found.", "e.g: Unable to resume torrent 'hash': torrent file not found.")
qDebug("Starting up torrent %s ...", qPrintable(hash));
if (!addTorrent_impl(resumeData, magnetUri, TorrentInfo::loadFromFile(filePath), data))
logger->addMessage(tr("Unable to resume torrent '%1'.", "e.g: Unable to resume torrent 'hash'.")
.arg(Utils::Fs::toNativePath(hash)), Log::CRITICAL);
}
}
}
qDebug("Unfinished torrents resumed.");
}
quint64 Session::getAlltimeDL() const
@ -2121,6 +2115,9 @@ void Session::handleAddTorrentAlert(libtorrent::add_torrent_alert *p)
bool fromMagnetUri = !torrent->hasMetadata();
if (data.resumed) {
if (fromMagnetUri && !data.addPaused)
torrent->resume(data.addForced);
logger->addMessage(tr("'%1' resumed. (fast resume)", "'torrent name' was resumed. (fast resume)")
.arg(torrent->name()));
}
@ -2363,7 +2360,7 @@ bool readFile(const QString &path, QByteArray &buf)
return true;
}
bool loadTorrentResumeData(const QByteArray &data, AddTorrentData &out)
bool loadTorrentResumeData(const QByteArray &data, AddTorrentData &out, MagnetUri &magnetUri)
{
out = AddTorrentData();
out.resumed = true;
@ -2381,6 +2378,10 @@ bool loadTorrentResumeData(const QByteArray &data, AddTorrentData &out)
out.hasSeedStatus = fast.dict_find_int_value("qBt-seedStatus");
out.disableTempPath = fast.dict_find_int_value("qBt-tempPathDisabled");
magnetUri = MagnetUri(Utils::String::fromStdString(fast.dict_find_string_value("qBt-magnetUri")));
out.addPaused = fast.dict_find_int_value("qBt-paused");
out.addForced = fast.dict_find_int_value("qBt-forced");
return true;
}

View File

@ -113,6 +113,7 @@ namespace BitTorrent
QString savePath;
bool disableTempPath; // e.g. for imported torrents
bool sequential;
TriStateBool addForced;
TriStateBool addPaused;
QVector<int> filePriorities; // used if TorrentInfo is set
bool ignoreShareRatio;

View File

@ -40,6 +40,7 @@
#include <libtorrent/address.hpp>
#include <libtorrent/alert_types.hpp>
#include <libtorrent/create_torrent.hpp>
#include <libtorrent/magnet_uri.hpp>
#include <boost/bind.hpp>
#ifdef Q_OS_WIN
@ -130,6 +131,7 @@ AddTorrentData::AddTorrentData(const AddTorrentParams &in)
, disableTempPath(in.disableTempPath)
, sequential(in.sequential)
, hasSeedStatus(in.skipChecking)
, addForced(in.addForced)
, addPaused(in.addPaused)
, filePriorities(in.filePriorities)
, ratioLimit(TorrentHandle::USE_GLOBAL_RATIO)
@ -1437,26 +1439,34 @@ void TorrentHandle::handleTorrentResumedAlert(libtorrent::torrent_resumed_alert
void TorrentHandle::handleSaveResumeDataAlert(libtorrent::save_resume_data_alert *p)
{
if (p->resume_data) {
(*(p->resume_data))["qBt-addedTime"] = m_addedTime.toTime_t();
(*(p->resume_data))["qBt-savePath"] = m_useDefaultSavePath ? "" : m_savePath.toUtf8().constData();
(*(p->resume_data))["qBt-ratioLimit"] = QString::number(m_ratioLimit).toUtf8().constData();
(*(p->resume_data))["qBt-label"] = m_label.toUtf8().constData();
(*(p->resume_data))["qBt-name"] = m_name.toUtf8().constData();
(*(p->resume_data))["qBt-seedStatus"] = m_hasSeedStatus;
(*(p->resume_data))["qBt-tempPathDisabled"] = m_tempPathDisabled;
const bool useDummyResumeData = !(p && p->resume_data);
libtorrent::entry dummyEntry;
m_session->handleTorrentResumeDataReady(this, *(p->resume_data));
}
else {
m_session->handleTorrentResumeDataFailed(this);
libtorrent::entry &resumeData = useDummyResumeData ? dummyEntry : *(p->resume_data);
if (useDummyResumeData) {
resumeData["qBt-magnetUri"] = Utils::String::toStdString(toMagnetUri());
resumeData["qBt-paused"] = isPaused();
resumeData["qBt-forced"] = isForced();
}
resumeData["qBt-addedTime"] = m_addedTime.toTime_t();
resumeData["qBt-savePath"] = m_useDefaultSavePath ? "" : Utils::String::toStdString(m_savePath);
resumeData["qBt-ratioLimit"] = Utils::String::toStdString(QString::number(m_ratioLimit));
resumeData["qBt-label"] = Utils::String::toStdString(m_label);
resumeData["qBt-name"] = Utils::String::toStdString(m_name);
resumeData["qBt-seedStatus"] = m_hasSeedStatus;
resumeData["qBt-tempPathDisabled"] = m_tempPathDisabled;
m_session->handleTorrentResumeDataReady(this, resumeData);
}
void TorrentHandle::handleSaveResumeDataFailedAlert(libtorrent::save_resume_data_failed_alert *p)
{
Q_UNUSED(p);
m_session->handleTorrentResumeDataFailed(this);
// if torrent has no metadata we should save dummy fastresume data
// containing Magnet URI and qBittorrent own resume data only
if (p->error.value() == libt::errors::no_metadata)
handleSaveResumeDataAlert(0);
else
m_session->handleTorrentResumeDataFailed(this);
}
void TorrentHandle::handleFastResumeRejectedAlert(libtorrent::fastresume_rejected_alert *p)
@ -1808,7 +1818,7 @@ void TorrentHandle::flushCache()
QString TorrentHandle::toMagnetUri() const
{
return m_torrentInfo.toMagnetUri();
return Utils::String::fromStdString(libt::make_magnet_uri(m_nativeHandle));
}
void TorrentHandle::resolveCountries(bool b)

View File

@ -91,8 +91,9 @@ namespace BitTorrent
bool disableTempPath;
bool sequential;
bool hasSeedStatus;
// for new torrents
TriStateBool addForced;
TriStateBool addPaused;
// for new torrents
QVector<int> filePriorities;
bool ignoreShareRatio;
// for resumed torrents

View File

@ -32,7 +32,6 @@
#include <QDateTime>
#include <libtorrent/error_code.hpp>
#include <libtorrent/magnet_uri.hpp>
#include "core/utils/misc.h"
#include "core/utils/fs.h"
@ -212,12 +211,6 @@ QByteArray TorrentInfo::metadata() const
return QByteArray(m_nativeInfo->metadata().get(), m_nativeInfo->metadata_size());
}
QString TorrentInfo::toMagnetUri() const
{
if (!isValid()) return QString();
return Utils::String::fromStdString(libt::make_magnet_uri(*m_nativeInfo));
}
void TorrentInfo::renameFile(uint index, const QString &newPath)
{
if (!isValid()) return;

View File

@ -75,7 +75,6 @@ namespace BitTorrent
QList<TrackerEntry> trackers() const;
QList<QUrl> urlSeeds() const;
QByteArray metadata() const;
QString toMagnetUri() const;
void renameFile(uint index, const QString &newPath);
boost::intrusive_ptr<libtorrent::torrent_info> nativeInfo() const;