qBittorrent/src/qtlibtorrent/qtorrenthandle.cpp

802 lines
25 KiB
C++
Raw Normal View History

/*
* Bittorrent Client using Qt4 and libtorrent.
* Copyright (C) 2006 Christophe Dumez
*
* 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.
*
* Contact : chris@qbittorrent.org
*/
#include <QString>
#include <QStringList>
#include <QFile>
#include <QDir>
#include <QByteArray>
#include <math.h>
#include "fs_utils.h"
#include "misc.h"
#include "preferences.h"
#include "qtorrenthandle.h"
#include "torrentpersistentdata.h"
#include <libtorrent/version.hpp>
#include <libtorrent/magnet_uri.hpp>
#include <libtorrent/torrent_info.hpp>
#include <libtorrent/bencode.hpp>
#include <libtorrent/entry.hpp>
2012-02-18 18:44:20 +04:00
#if LIBTORRENT_VERSION_MINOR < 15
#include <boost/date_time/posix_time/posix_time_types.hpp>
#endif
#ifdef Q_WS_WIN
#include <Windows.h>
#endif
using namespace libtorrent;
using namespace std;
2012-02-21 21:49:43 +04:00
static QPair<int, int> get_file_extremity_pieces(const torrent_info& t, int file_index)
{
const int num_pieces = t.num_pieces();
const int piece_size = t.piece_length();
const file_entry& file = t.file_at(file_index);
// Determine the first and last piece of the file
int first_piece = floor((file.offset + 1) / (float) piece_size);
Q_ASSERT(first_piece >= 0 && first_piece < num_pieces);
qDebug("First piece of the file is %d/%d", first_piece, num_pieces - 1);
int num_pieces_in_file = ceil(file.size / (float) piece_size);
int last_piece = first_piece + num_pieces_in_file - 1;
Q_ASSERT(last_piece >= 0 && last_piece < num_pieces);
qDebug("last piece of the file is %d/%d", last_piece, num_pieces - 1);
return qMakePair(first_piece, last_piece);
}
QTorrentHandle::QTorrentHandle(const torrent_handle& h): torrent_handle(h) {}
//
// Getters
//
QString QTorrentHandle::hash() const {
return misc::toQString(torrent_handle::info_hash());
}
QString QTorrentHandle::name() const {
QString name = TorrentPersistentData::getName(hash());
2012-02-20 21:30:53 +04:00
if (name.isEmpty()) {
name = misc::toQStringU(torrent_handle::name());
}
return name;
}
QString QTorrentHandle::creation_date() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
boost::optional<time_t> t = torrent_handle::get_torrent_info().creation_date();
2012-05-15 20:57:31 +04:00
return t ? misc::toQString(*t) : "";
#else
boost::optional<boost::posix_time::ptime> boostDate = torrent_handle::get_torrent_info().creation_date();
return boostDate ? misc::toQString(*boostDate) : "";
#endif
}
2010-05-18 13:35:55 +04:00
QString QTorrentHandle::next_announce() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return misc::userFriendlyDuration(torrent_handle::status(0x0).next_announce.total_seconds());
#else
return misc::userFriendlyDuration(torrent_handle::status().next_announce.total_seconds());
2011-04-17 14:29:44 +04:00
#endif
2010-05-18 13:35:55 +04:00
}
qlonglong QTorrentHandle::next_announce_s() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).next_announce.total_seconds();
#else
return torrent_handle::status().next_announce.total_seconds();
2011-04-17 14:29:44 +04:00
#endif
2010-05-18 13:35:55 +04:00
}
2012-02-21 21:49:43 +04:00
float QTorrentHandle::progress() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
torrent_status st = torrent_handle::status(query_accurate_download_counters);
#else
torrent_status st = torrent_handle::status();
#endif
2012-02-20 21:30:53 +04:00
if (!st.total_wanted)
return 0.;
2011-04-17 14:29:44 +04:00
if (st.total_wanted_done == st.total_wanted)
2008-08-17 07:47:52 +04:00
return 1.;
2012-02-21 21:49:43 +04:00
float progress = (float) st.total_wanted_done / (float) st.total_wanted;
2007-08-28 23:22:05 +04:00
Q_ASSERT(progress >= 0. && progress <= 1.);
return progress;
}
bitfield QTorrentHandle::pieces() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).pieces;
#else
return torrent_handle::status().pieces;
2011-04-17 14:29:44 +04:00
#endif
}
QString QTorrentHandle::current_tracker() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return misc::toQString(torrent_handle::status(0x0).current_tracker);
#else
return misc::toQString(torrent_handle::status().current_tracker);
2011-04-17 14:29:44 +04:00
#endif
}
bool QTorrentHandle::is_paused() const {
#if LIBTORRENT_VERSION_MINOR > 15
2011-04-17 14:29:44 +04:00
torrent_status st = torrent_handle::status(0x0);
return st.paused && !st.auto_managed;
#else
return torrent_handle::is_paused() && !torrent_handle::is_auto_managed();
#endif
}
bool QTorrentHandle::is_queued() const {
#if LIBTORRENT_VERSION_MINOR > 15
2011-04-17 14:29:44 +04:00
torrent_status st = torrent_handle::status(0x0);
return st.paused && st.auto_managed;
#else
return torrent_handle::is_paused() && torrent_handle::is_auto_managed();
#endif
}
size_type QTorrentHandle::total_size() const {
return torrent_handle::get_torrent_info().total_size();
}
size_type QTorrentHandle::piece_length() const {
return torrent_handle::get_torrent_info().piece_length();
}
int QTorrentHandle::num_pieces() const {
return torrent_handle::get_torrent_info().num_pieces();
2009-11-14 13:37:45 +03:00
}
bool QTorrentHandle::first_last_piece_first() const {
2012-02-21 21:49:43 +04:00
const torrent_info& t = get_torrent_info();
// Get int first media file
int index = 0;
2012-02-21 21:49:43 +04:00
for (index = 0; index < t.num_files(); ++index) {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
2012-02-21 21:49:43 +04:00
QString path = misc::toQStringU(t.file_at(index).path);
2011-04-17 14:29:44 +04:00
#else
2012-02-21 21:49:43 +04:00
QString path = misc::toQStringU(t.file_at(index).path.string());
2011-04-17 14:29:44 +04:00
#endif
const QString ext = fsutils::fileExtension(path);
2012-02-21 21:49:43 +04:00
if (misc::isPreviewable(ext) && torrent_handle::file_priority(index) > 0)
break;
}
2012-02-21 21:49:43 +04:00
if (index >= t.num_files()) // No media file
return false;
QPair<int, int> extremities = get_file_extremity_pieces (t, index);
return (torrent_handle::piece_priority(extremities.first) == 7)
&& (torrent_handle::piece_priority(extremities.second) == 7);
}
size_type QTorrentHandle::total_wanted_done() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(query_accurate_download_counters).total_wanted_done;
#else
return torrent_handle::status().total_wanted_done;
2011-04-17 14:29:44 +04:00
#endif
}
size_type QTorrentHandle::total_wanted() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).total_wanted;
#else
return torrent_handle::status().total_wanted;
2011-04-17 14:29:44 +04:00
#endif
}
2011-01-25 20:01:09 +03:00
qreal QTorrentHandle::download_payload_rate() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).download_payload_rate;
#else
return torrent_handle::status().download_payload_rate;
2011-04-17 14:29:44 +04:00
#endif
}
2011-01-25 20:01:09 +03:00
qreal QTorrentHandle::upload_payload_rate() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).upload_payload_rate;
#else
return torrent_handle::status().upload_payload_rate;
2011-04-17 14:29:44 +04:00
#endif
}
int QTorrentHandle::num_peers() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).num_peers;
#else
return torrent_handle::status().num_peers;
2011-04-17 14:29:44 +04:00
#endif
}
int QTorrentHandle::num_seeds() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).num_seeds;
#else
return torrent_handle::status().num_seeds;
2011-04-17 14:29:44 +04:00
#endif
}
int QTorrentHandle::num_complete() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).num_complete;
#else
return torrent_handle::status().num_complete;
2011-04-17 14:29:44 +04:00
#endif
}
int QTorrentHandle::num_incomplete() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).num_incomplete;
#else
return torrent_handle::status().num_incomplete;
2011-04-17 14:29:44 +04:00
#endif
}
QString QTorrentHandle::save_path() const {
#if LIBTORRENT_VERSION_MINOR > 15
return misc::toQStringU(torrent_handle::save_path()).replace("\\", "/");
#else
return misc::toQStringU(torrent_handle::save_path().string()).replace("\\", "/");
#endif
}
QStringList QTorrentHandle::url_seeds() const {
QStringList res;
try {
const std::set<std::string> existing_seeds = torrent_handle::url_seeds();
2010-10-10 19:39:08 +04:00
std::set<std::string>::const_iterator it;
2012-02-20 21:30:53 +04:00
for (it = existing_seeds.begin(); it != existing_seeds.end(); it++) {
2010-10-10 19:39:08 +04:00
qDebug("URL Seed: %s", it->c_str());
res << misc::toQString(*it);
}
2010-10-10 19:39:08 +04:00
} catch(std::exception e) {
std::cout << "ERROR: Failed to convert the URL seed" << std::endl;
}
return res;
}
// get the size of the torrent without the filtered files
2011-04-17 14:29:44 +04:00
size_type QTorrentHandle::actual_size() const {
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(query_accurate_download_counters).total_wanted;
#else
return torrent_handle::status().total_wanted;
2011-04-17 14:29:44 +04:00
#endif
}
bool QTorrentHandle::has_filtered_pieces() const {
2012-02-21 21:49:43 +04:00
const std::vector<int> piece_priorities = torrent_handle::piece_priorities();
foreach (const int priority, piece_priorities) {
if (priority == 0)
return true;
}
return false;
}
int QTorrentHandle::num_files() const {
return torrent_handle::get_torrent_info().num_files();
}
QString QTorrentHandle::filename_at(unsigned int index) const {
Q_ASSERT(index < (unsigned int)torrent_handle::get_torrent_info().num_files());
#if LIBTORRENT_VERSION_MINOR > 15
return fsutils::fileName(filepath_at(index));
#else
return misc::toQStringU(torrent_handle::get_torrent_info().file_at(index).path.leaf());
#endif
}
size_type QTorrentHandle::filesize_at(unsigned int index) const {
Q_ASSERT(index < (unsigned int)torrent_handle::get_torrent_info().num_files());
return torrent_handle::get_torrent_info().file_at(index).size;
}
QString QTorrentHandle::filepath_at(unsigned int index) const {
#if LIBTORRENT_VERSION_MINOR > 15
2011-04-17 14:29:44 +04:00
return misc::toQStringU(torrent_handle::get_torrent_info().file_at(index).path);
#else
return misc::toQStringU(torrent_handle::get_torrent_info().file_at(index).path.string());
#endif
}
QString QTorrentHandle::orig_filepath_at(unsigned int index) const {
#if LIBTORRENT_VERSION_MINOR > 15
2011-04-17 14:29:44 +04:00
return misc::toQStringU(torrent_handle::get_torrent_info().orig_files().at(index).path);
#else
return misc::toQStringU(torrent_handle::get_torrent_info().orig_files().at(index).path.string());
#endif
}
torrent_status::state_t QTorrentHandle::state() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).state;
#else
return torrent_handle::status().state;
2011-04-17 14:29:44 +04:00
#endif
}
QString QTorrentHandle::creator() const {
return misc::toQStringU(torrent_handle::get_torrent_info().creator());
}
QString QTorrentHandle::comment() const {
return misc::toQStringU(torrent_handle::get_torrent_info().comment());
}
size_type QTorrentHandle::total_failed_bytes() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).total_failed_bytes;
#else
return torrent_handle::status().total_failed_bytes;
2011-04-17 14:29:44 +04:00
#endif
}
size_type QTorrentHandle::total_redundant_bytes() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).total_redundant_bytes;
#else
return torrent_handle::status().total_redundant_bytes;
2011-04-17 14:29:44 +04:00
#endif
}
bool QTorrentHandle::is_checking() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
torrent_status st = torrent_handle::status(0x0);
#else
torrent_status st = torrent_handle::status();
#endif
return st.state == torrent_status::checking_files || st.state == torrent_status::checking_resume_data;
}
2010-03-03 20:27:25 +03:00
size_type QTorrentHandle::total_done() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).total_done;
#else
return torrent_handle::status().total_done;
2011-04-17 14:29:44 +04:00
#endif
}
2010-03-03 20:27:25 +03:00
size_type QTorrentHandle::all_time_download() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).all_time_download;
#else
return torrent_handle::status().all_time_download;
2011-04-17 14:29:44 +04:00
#endif
}
2010-03-03 20:27:25 +03:00
size_type QTorrentHandle::all_time_upload() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).all_time_upload;
#else
return torrent_handle::status().all_time_upload;
2011-04-17 14:29:44 +04:00
#endif
}
2010-03-03 20:27:25 +03:00
size_type QTorrentHandle::total_payload_download() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).total_payload_download;
#else
return torrent_handle::status().total_payload_download;
2011-04-17 14:29:44 +04:00
#endif
}
2010-03-03 20:27:25 +03:00
size_type QTorrentHandle::total_payload_upload() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).total_payload_upload;
#else
return torrent_handle::status().total_payload_upload;
2011-04-17 14:29:44 +04:00
#endif
}
// Return a list of absolute paths corresponding
// to all files in a torrent
2011-04-17 14:29:44 +04:00
QStringList QTorrentHandle::absolute_files_path() const {
QDir saveDir(save_path());
QStringList res;
2012-02-20 21:30:53 +04:00
for (int i = 0; i<num_files(); ++i) {
2011-04-17 14:29:44 +04:00
res << QDir::cleanPath(saveDir.absoluteFilePath(filepath_at(i)));
}
return res;
}
2011-04-17 14:29:44 +04:00
QStringList QTorrentHandle::absolute_files_path_uneeded() const {
QDir saveDir(save_path());
QStringList res;
std::vector<int> fp = torrent_handle::file_priorities();
2012-02-20 21:30:53 +04:00
for (uint i = 0; i < fp.size(); ++i) {
if (fp[i] == 0) {
2011-04-17 14:29:44 +04:00
const QString file_path = QDir::cleanPath(saveDir.absoluteFilePath(filepath_at(i)));
2012-02-20 21:30:53 +04:00
if (file_path.contains(".unwanted"))
res << file_path;
}
}
return res;
}
2010-08-22 00:22:59 +04:00
bool QTorrentHandle::has_missing_files() const {
2011-04-17 14:29:44 +04:00
const QStringList paths = absolute_files_path();
2012-02-20 21:30:53 +04:00
foreach (const QString &path, paths) {
if (!QFile::exists(path)) return true;
2010-08-22 00:22:59 +04:00
}
return false;
}
int QTorrentHandle::queue_position() const {
2012-02-20 21:30:53 +04:00
if (torrent_handle::queue_position() < 0)
return -1;
return torrent_handle::queue_position()+1;
}
int QTorrentHandle::num_uploads() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).num_uploads;
#else
return torrent_handle::status().num_uploads;
2011-04-17 14:29:44 +04:00
#endif
}
bool QTorrentHandle::is_seed() const {
// Affected by bug http://code.rasterbar.com/libtorrent/ticket/402
//return torrent_handle::is_seed();
// May suffer from approximation problems
//return (progress() == 1.);
// This looks safe
2012-02-21 21:49:43 +04:00
torrent_status::state_t st = state();
return (st == torrent_status::finished || st == torrent_status::seeding);
}
bool QTorrentHandle::is_auto_managed() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
torrent_status status = torrent_handle::status(0x0);
return status.auto_managed;
#else
return torrent_handle::is_auto_managed();
2011-04-17 14:29:44 +04:00
#endif
}
2011-04-17 14:36:50 +04:00
bool QTorrentHandle::is_sequential_download() const {
#if LIBTORRENT_VERSION_MINOR > 15
torrent_status status = torrent_handle::status(0x0);
return status.sequential_download;
#else
return torrent_handle::is_sequential_download();
#endif
}
qlonglong QTorrentHandle::active_time() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).active_time;
#else
return torrent_handle::status().active_time;
2011-04-17 14:29:44 +04:00
#endif
}
qlonglong QTorrentHandle::seeding_time() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).seeding_time;
#else
return torrent_handle::status().seeding_time;
2011-04-17 14:29:44 +04:00
#endif
}
int QTorrentHandle::num_connections() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).num_connections;
#else
return torrent_handle::status().num_connections;
2011-04-17 14:29:44 +04:00
#endif
}
int QTorrentHandle::connections_limit() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return torrent_handle::status(0x0).connections_limit;
#else
return torrent_handle::status().connections_limit;
2011-04-17 14:29:44 +04:00
#endif
}
2009-11-19 18:43:00 +03:00
bool QTorrentHandle::priv() const {
return torrent_handle::get_torrent_info().priv();
2009-11-19 18:43:00 +03:00
}
QString QTorrentHandle::firstFileSavePath() const {
Q_ASSERT(has_metadata());
QString fsave_path = TorrentPersistentData::getSavePath(hash());
2012-02-20 21:30:53 +04:00
if (fsave_path.isEmpty())
fsave_path = save_path();
2011-09-25 16:04:51 +04:00
fsave_path.replace("\\", "/");
2012-02-20 21:30:53 +04:00
if (!fsave_path.endsWith("/"))
fsave_path += "/";
fsave_path += filepath_at(0);
// Remove .!qB extension
2012-02-20 21:30:53 +04:00
if (fsave_path.endsWith(".!qB", Qt::CaseInsensitive))
fsave_path.chop(4);
return fsave_path;
}
bool QTorrentHandle::has_error() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
torrent_status st = torrent_handle::status(0x0);
return st.paused && !st.error.empty();
#else
return torrent_handle::is_paused() && !torrent_handle::status().error.empty();
2011-04-17 14:29:44 +04:00
#endif
}
QString QTorrentHandle::error() const {
2011-04-17 14:29:44 +04:00
#if LIBTORRENT_VERSION_MINOR > 15
return misc::toQString(torrent_handle::status(0x0).error);
#else
return misc::toQString(torrent_handle::status().error);
2011-04-17 14:29:44 +04:00
#endif
}
void QTorrentHandle::downloading_pieces(bitfield &bf) const {
std::vector<partial_piece_info> queue;
torrent_handle::get_download_queue(queue);
2012-02-21 21:49:43 +04:00
for (std::vector<partial_piece_info>::const_iterator it=queue.begin(); it!= queue.end(); it++) {
bf.set_bit(it->piece_index);
}
return;
}
bool QTorrentHandle::has_metadata() const {
#if LIBTORRENT_VERSION_MINOR > 15
2012-06-28 19:10:22 +04:00
return torrent_handle::status(0x0).has_metadata;
#else
return torrent_handle::has_metadata();
#endif
}
2011-04-17 19:00:48 +04:00
float QTorrentHandle::distributed_copies() const {
#if LIBTORRENT_VERSION_MINOR > 15
2012-06-28 19:10:22 +04:00
return torrent_handle::status(query_distributed_copies).distributed_copies;
2011-04-17 19:00:48 +04:00
#else
return torrent_handle::status().distributed_copies;
#endif
}
void QTorrentHandle::file_progress(std::vector<size_type>& fp) const {
2012-02-18 18:44:20 +04:00
torrent_handle::file_progress(fp, torrent_handle::piece_granularity);
}
//
// Setters
//
2010-11-24 23:31:14 +03:00
void QTorrentHandle::pause() const {
torrent_handle::auto_managed(false);
torrent_handle::pause();
torrent_handle::save_resume_data();
}
2010-11-24 23:31:14 +03:00
void QTorrentHandle::resume() const {
2012-02-21 21:49:43 +04:00
if (has_error())
torrent_handle::clear_error();
const QString torrent_hash = hash();
bool has_persistant_error = TorrentPersistentData::hasError(torrent_hash);
TorrentPersistentData::setErrorState(torrent_hash, false);
bool temp_path_enabled = Preferences().isTempPathEnabled();
2012-02-20 21:30:53 +04:00
if (has_persistant_error && temp_path_enabled) {
// Torrent was supposed to be seeding, checking again in final destination
qDebug("Resuming a torrent with error...");
const QString final_save_path = TorrentPersistentData::getSavePath(torrent_hash);
qDebug("Torrent final path is: %s", qPrintable(final_save_path));
2012-02-20 21:30:53 +04:00
if (!final_save_path.isEmpty())
move_storage(final_save_path);
}
torrent_handle::auto_managed(true);
torrent_handle::resume();
2012-02-20 21:30:53 +04:00
if (has_persistant_error && temp_path_enabled) {
// Force recheck
torrent_handle::force_recheck();
}
}
2012-02-21 21:49:43 +04:00
void QTorrentHandle::remove_url_seed(const QString& seed) const {
torrent_handle::remove_url_seed(seed.toStdString());
}
2012-02-21 21:49:43 +04:00
void QTorrentHandle::add_url_seed(const QString& seed) const {
2010-10-10 19:39:08 +04:00
const std::string str_seed = seed.toStdString();
qDebug("calling torrent_handle::add_url_seed(%s)", str_seed.c_str());
torrent_handle::add_url_seed(str_seed);
}
2012-02-21 21:49:43 +04:00
void QTorrentHandle::set_tracker_login(const QString& username, const QString& password) const {
torrent_handle::set_tracker_login(std::string(username.toLocal8Bit().constData()), std::string(password.toLocal8Bit().constData()));
2008-11-02 16:53:45 +03:00
}
2012-02-21 21:49:43 +04:00
void QTorrentHandle::move_storage(const QString& new_path) const {
if (QDir(save_path()) == QDir(new_path))
return;
2010-07-18 01:35:00 +04:00
TorrentPersistentData::setPreviousSavePath(hash(), save_path());
// Create destination directory if necessary
// or move_storage() will fail...
QDir().mkpath(new_path);
// Actually move the storage
torrent_handle::move_storage(new_path.toUtf8().constData());
}
2012-02-21 21:49:43 +04:00
bool QTorrentHandle::save_torrent_file(const QString& path) const {
2012-02-20 21:30:53 +04:00
if (!has_metadata()) return false;
2012-02-21 21:49:43 +04:00
const torrent_info& t = torrent_handle::get_torrent_info();
entry meta = bdecode(t.metadata().get(),
t.metadata().get() + t.metadata_size());
entry torrent_entry(entry::dictionary_t);
torrent_entry["info"] = meta;
2012-02-20 21:30:53 +04:00
if (!torrent_handle::trackers().empty())
torrent_entry["announce"] = torrent_handle::trackers().front().url;
vector<char> out;
bencode(back_inserter(out), torrent_entry);
QFile torrent_file(path);
if (!out.empty() && torrent_file.open(QIODevice::WriteOnly)) {
torrent_file.write(&out[0], out.size());
torrent_file.close();
return true;
}
return false;
}
void QTorrentHandle::file_priority(int index, int priority) const {
vector<int> priorities = torrent_handle::file_priorities();
2012-02-20 21:30:53 +04:00
if (priorities[index] != priority) {
priorities[index] = priority;
prioritize_files(priorities);
}
}
void QTorrentHandle::prioritize_files(const vector<int> &files) const {
2012-02-20 21:30:53 +04:00
if ((int)files.size() != torrent_handle::get_torrent_info().num_files()) return;
qDebug() << Q_FUNC_INFO;
bool was_seed = is_seed();
2011-01-08 16:48:29 +03:00
vector<size_type> progress;
file_progress(progress);
qDebug() << Q_FUNC_INFO << "Changing files priorities...";
torrent_handle::prioritize_files(files);
qDebug() << Q_FUNC_INFO << "Moving unwanted files to .unwanted folder...";
2012-02-21 21:49:43 +04:00
for (uint i = 0; i < files.size(); ++i) {
// Move unwanted files to a .unwanted subfolder
2012-02-20 21:30:53 +04:00
if (files[i] == 0 && progress[i] < filesize_at(i)) {
QString old_path = filepath_at(i);
// Make sure the file does not already exists
2012-02-20 21:30:53 +04:00
if (QFile::exists(QDir(save_path()).absoluteFilePath(old_path))) {
qWarning() << "File" << old_path << "already exists at destination.";
qWarning() << "We do not move it to .unwanted folder";
continue;
}
QString old_name = filename_at(i);
QString parent_path = fsutils::branchPath(old_path);
2012-02-20 21:30:53 +04:00
if (parent_path.isEmpty() || QDir(parent_path).dirName() != ".unwanted") {
QString unwanted_abspath = QDir::cleanPath(save_path()+"/"+parent_path+"/.unwanted");
qDebug() << "Unwanted path is" << unwanted_abspath;
bool created = QDir().mkpath(unwanted_abspath);
#ifdef Q_WS_WIN
qDebug() << "unwanted folder was created:" << created;
2012-02-20 21:30:53 +04:00
if (created) {
// Hide the folder on Windows
qDebug() << "Hiding folder (Windows)";
wstring win_path = unwanted_abspath.replace("/","\\").toStdWString();
DWORD dwAttrs = GetFileAttributesW(win_path.c_str());
bool ret = SetFileAttributesW(win_path.c_str(), dwAttrs|FILE_ATTRIBUTE_HIDDEN);
Q_ASSERT(ret != 0); Q_UNUSED(ret);
}
#else
Q_UNUSED(created);
#endif
2012-02-20 21:30:53 +04:00
if (!parent_path.isEmpty() && !parent_path.endsWith("/"))
parent_path += "/";
rename_file(i, parent_path+".unwanted/"+old_name);
}
}
// Move wanted files back to their original folder
qDebug() << Q_FUNC_INFO << "Moving wanted files back from .unwanted folder";
2012-02-20 21:30:53 +04:00
if (files[i] > 0) {
QString parent_relpath = fsutils::branchPath(filepath_at(i));
2012-02-20 21:30:53 +04:00
if (QDir(parent_relpath).dirName() == ".unwanted") {
QString old_name = filename_at(i);
QString new_relpath = fsutils::branchPath(parent_relpath);
if (new_relpath.isEmpty())
rename_file(i, old_name);
else
rename_file(i, QDir(new_relpath).filePath(old_name));
// Remove .unwanted directory if empty
qDebug() << "Attempting to remove .unwanted folder at " << QDir(save_path() + QDir::separator() + new_relpath).absoluteFilePath(".unwanted");
QDir(save_path() + QDir::separator() + new_relpath).rmdir(".unwanted");
}
}
}
2012-02-20 21:30:53 +04:00
if (was_seed && !is_seed()) {
qDebug() << "Torrent is no longer SEEDING";
// Save seed status
TorrentPersistentData::saveSeedStatus(*this);
// Move to temp folder if necessary
const Preferences pref;
2012-02-20 21:30:53 +04:00
if (pref.isTempPathEnabled()) {
QString tmp_path = pref.getTempPath();
qDebug() << "tmp folder is enabled, move torrent to " << tmp_path << " from " << save_path();
move_storage(tmp_path);
}
}
}
void QTorrentHandle::prioritize_first_last_piece(int file_index, bool b) const {
// Determine the priority to set
2012-02-21 21:49:43 +04:00
int prio = b ? 7 : torrent_handle::file_priority(file_index);
QPair<int, int> extremities = get_file_extremity_pieces (get_torrent_info(), file_index);
piece_priority(extremities.first, prio);
piece_priority(extremities.second, prio);
}
void QTorrentHandle::prioritize_first_last_piece(bool b) const {
2012-02-20 21:30:53 +04:00
if (!has_metadata()) return;
// Download first and last pieces first for all media files in the torrent
2012-02-21 21:49:43 +04:00
const uint nbfiles = num_files();
for (uint index = 0; index < nbfiles; ++index) {
2011-04-17 14:29:44 +04:00
const QString path = filepath_at(index);
const QString ext = fsutils::fileExtension(path);
2012-02-20 21:30:53 +04:00
if (misc::isPreviewable(ext) && torrent_handle::file_priority(index) > 0) {
2011-04-17 14:29:44 +04:00
qDebug() << "File" << path << "is previewable, toggle downloading of first/last pieces first";
prioritize_first_last_piece(index, b);
}
}
}
2012-02-21 21:49:43 +04:00
void QTorrentHandle::rename_file(int index, const QString& name) const {
qDebug() << Q_FUNC_INFO << index << name;
torrent_handle::rename_file(index, std::string(name.toUtf8().constData()));
}
//
// Operators
//
2012-02-20 23:32:58 +04:00
bool QTorrentHandle::operator ==(const QTorrentHandle& new_h) const {
2012-02-21 21:49:43 +04:00
return info_hash() == new_h.info_hash();
}