Merge pull request #3224 from glassez/fastresume

Fix resume torrents without metadata.

src/core/bittorrent/session.cpp

@@ -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;
 }
 

src/core/bittorrent/session.h

@@ -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;

src/core/bittorrent/torrenthandle.cpp

@@ -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)

src/core/bittorrent/torrenthandle.h

@@ -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

src/core/bittorrent/torrentinfo.cpp

@@ -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;

src/core/bittorrent/torrentinfo.h

@@ -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;