mirror of
https://github.com/transmission/transmission
synced 2024-12-27 18:18:10 +00:00
3008a992ca
* fix: Bitfield.getRaw() regression
1047 lines
29 KiB
C++
1047 lines
29 KiB
C++
/*
|
|
* This file Copyright (C) 2008-2014 Mnemosyne LLC
|
|
*
|
|
* It may be used under the GNU GPL versions 2 or 3
|
|
* or any future license endorsed by Mnemosyne LLC.
|
|
*
|
|
*/
|
|
|
|
#include <algorithm>
|
|
#include <cstring>
|
|
#include <string_view>
|
|
|
|
#include "transmission.h"
|
|
#include "completion.h"
|
|
#include "error.h"
|
|
#include "file.h"
|
|
#include "log.h"
|
|
#include "metainfo.h" /* tr_metainfoGetBasename() */
|
|
#include "peer-mgr.h" /* pex */
|
|
#include "platform.h" /* tr_getResumeDir() */
|
|
#include "resume.h"
|
|
#include "session.h"
|
|
#include "torrent.h"
|
|
#include "tr-assert.h"
|
|
#include "utils.h" /* tr_buildPath */
|
|
#include "variant.h"
|
|
|
|
using namespace std::literals;
|
|
|
|
namespace
|
|
{
|
|
|
|
constexpr int MAX_REMEMBERED_PEERS = 200;
|
|
|
|
} // unnamed namespace
|
|
|
|
static char* getResumeFilename(tr_torrent const* tor, enum tr_metainfo_basename_format format)
|
|
{
|
|
char* base = tr_metainfoGetBasename(tr_torrentInfo(tor), format);
|
|
char* filename = tr_strdup_printf("%s" TR_PATH_DELIMITER_STR "%s.resume", tr_getResumeDir(tor->session), base);
|
|
tr_free(base);
|
|
return filename;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void savePeers(tr_variant* dict, tr_torrent const* tor)
|
|
{
|
|
tr_pex* pex = nullptr;
|
|
int count = tr_peerMgrGetPeers(tor, &pex, TR_AF_INET, TR_PEERS_INTERESTING, MAX_REMEMBERED_PEERS);
|
|
|
|
if (count > 0)
|
|
{
|
|
tr_variantDictAddRaw(dict, TR_KEY_peers2, pex, sizeof(tr_pex) * count);
|
|
}
|
|
|
|
tr_free(pex);
|
|
|
|
count = tr_peerMgrGetPeers(tor, &pex, TR_AF_INET6, TR_PEERS_INTERESTING, MAX_REMEMBERED_PEERS);
|
|
|
|
if (count > 0)
|
|
{
|
|
tr_variantDictAddRaw(dict, TR_KEY_peers2_6, pex, sizeof(tr_pex) * count);
|
|
}
|
|
|
|
tr_free(pex);
|
|
}
|
|
|
|
static size_t addPeers(tr_torrent* tor, uint8_t const* buf, size_t buflen)
|
|
{
|
|
size_t const n_in = buflen / sizeof(tr_pex);
|
|
size_t const n_pex = std::min(n_in, size_t{ MAX_REMEMBERED_PEERS });
|
|
|
|
tr_pex pex[MAX_REMEMBERED_PEERS];
|
|
memcpy(pex, buf, sizeof(tr_pex) * n_pex);
|
|
return tr_peerMgrAddPex(tor, TR_PEER_FROM_RESUME, pex, n_pex);
|
|
}
|
|
|
|
static uint64_t loadPeers(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
uint8_t const* str = nullptr;
|
|
auto len = size_t{};
|
|
if (tr_variantDictFindRaw(dict, TR_KEY_peers2, &str, &len))
|
|
{
|
|
size_t const numAdded = addPeers(tor, str, len);
|
|
tr_logAddTorDbg(tor, "Loaded %zu IPv4 peers from resume file", numAdded);
|
|
ret = TR_FR_PEERS;
|
|
}
|
|
|
|
if (tr_variantDictFindRaw(dict, TR_KEY_peers2_6, &str, &len))
|
|
{
|
|
size_t const numAdded = addPeers(tor, str, len);
|
|
tr_logAddTorDbg(tor, "Loaded %zu IPv6 peers from resume file", numAdded);
|
|
ret = TR_FR_PEERS;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void saveLabels(tr_variant* dict, tr_torrent const* tor)
|
|
{
|
|
auto const& labels = tor->labels;
|
|
tr_variant* list = tr_variantDictAddList(dict, TR_KEY_labels, std::size(labels));
|
|
for (auto const& label : labels)
|
|
{
|
|
tr_variantListAddStr(list, label);
|
|
}
|
|
}
|
|
|
|
static uint64_t loadLabels(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
tr_variant* list = nullptr;
|
|
if (tr_variantDictFindList(dict, TR_KEY_labels, &list))
|
|
{
|
|
int const n = tr_variantListSize(list);
|
|
for (int i = 0; i < n; ++i)
|
|
{
|
|
char const* str = nullptr;
|
|
auto str_len = size_t{};
|
|
if (tr_variantGetStr(tr_variantListChild(list, i), &str, &str_len) && str != nullptr && str_len != 0)
|
|
{
|
|
tor->labels.emplace(str, str_len);
|
|
}
|
|
}
|
|
|
|
ret = TR_FR_LABELS;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void saveDND(tr_variant* dict, tr_torrent const* tor)
|
|
{
|
|
tr_info const* const inf = tr_torrentInfo(tor);
|
|
tr_file_index_t const n = inf->fileCount;
|
|
|
|
tr_variant* const list = tr_variantDictAddList(dict, TR_KEY_dnd, n);
|
|
|
|
for (tr_file_index_t i = 0; i < n; ++i)
|
|
{
|
|
tr_variantListAddBool(list, inf->files[i].dnd);
|
|
}
|
|
}
|
|
|
|
static uint64_t loadDND(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
uint64_t ret = 0;
|
|
tr_variant* list = nullptr;
|
|
tr_file_index_t const n = tor->info.fileCount;
|
|
|
|
if (tr_variantDictFindList(dict, TR_KEY_dnd, &list) && tr_variantListSize(list) == n)
|
|
{
|
|
tr_file_index_t* dl = tr_new(tr_file_index_t, n);
|
|
tr_file_index_t* dnd = tr_new(tr_file_index_t, n);
|
|
tr_file_index_t dlCount = 0;
|
|
tr_file_index_t dndCount = 0;
|
|
|
|
for (tr_file_index_t i = 0; i < n; ++i)
|
|
{
|
|
auto tmp = false;
|
|
if (tr_variantGetBool(tr_variantListChild(list, i), &tmp) && tmp)
|
|
{
|
|
dnd[dndCount++] = i;
|
|
}
|
|
else
|
|
{
|
|
dl[dlCount++] = i;
|
|
}
|
|
}
|
|
|
|
if (dndCount != 0)
|
|
{
|
|
tr_torrentInitFileDLs(tor, dnd, dndCount, false);
|
|
tr_logAddTorDbg(tor, "Resume file found %d files listed as dnd", dndCount);
|
|
}
|
|
|
|
if (dlCount != 0)
|
|
{
|
|
tr_torrentInitFileDLs(tor, dl, dlCount, true);
|
|
tr_logAddTorDbg(tor, "Resume file found %d files marked for download", dlCount);
|
|
}
|
|
|
|
tr_free(dnd);
|
|
tr_free(dl);
|
|
ret = TR_FR_DND;
|
|
}
|
|
else
|
|
{
|
|
tr_logAddTorDbg(
|
|
tor,
|
|
"Couldn't load DND flags. DND list (%p) has %zu"
|
|
" children; torrent has %d files",
|
|
(void*)list,
|
|
tr_variantListSize(list),
|
|
(int)n);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void saveFilePriorities(tr_variant* dict, tr_torrent const* tor)
|
|
{
|
|
tr_info const* const inf = tr_torrentInfo(tor);
|
|
tr_file_index_t const n = inf->fileCount;
|
|
|
|
tr_variant* const list = tr_variantDictAddList(dict, TR_KEY_priority, n);
|
|
for (tr_file_index_t i = 0; i < n; ++i)
|
|
{
|
|
tr_variantListAddInt(list, inf->files[i].priority);
|
|
}
|
|
}
|
|
|
|
static uint64_t loadFilePriorities(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
tr_file_index_t const n = tor->info.fileCount;
|
|
tr_variant* list = nullptr;
|
|
if (tr_variantDictFindList(dict, TR_KEY_priority, &list) && tr_variantListSize(list) == n)
|
|
{
|
|
for (tr_file_index_t i = 0; i < n; ++i)
|
|
{
|
|
auto priority = int64_t{};
|
|
if (tr_variantGetInt(tr_variantListChild(list, i), &priority))
|
|
{
|
|
tr_torrentInitFilePriority(tor, i, priority);
|
|
}
|
|
}
|
|
|
|
ret = TR_FR_FILE_PRIORITIES;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void saveSingleSpeedLimit(tr_variant* d, tr_torrent* tor, tr_direction dir)
|
|
{
|
|
tr_variantDictReserve(d, 3);
|
|
tr_variantDictAddInt(d, TR_KEY_speed_Bps, tr_torrentGetSpeedLimit_Bps(tor, dir));
|
|
tr_variantDictAddBool(d, TR_KEY_use_global_speed_limit, tr_torrentUsesSessionLimits(tor));
|
|
tr_variantDictAddBool(d, TR_KEY_use_speed_limit, tr_torrentUsesSpeedLimit(tor, dir));
|
|
}
|
|
|
|
static void saveSpeedLimits(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
saveSingleSpeedLimit(tr_variantDictAddDict(dict, TR_KEY_speed_limit_down, 0), tor, TR_DOWN);
|
|
saveSingleSpeedLimit(tr_variantDictAddDict(dict, TR_KEY_speed_limit_up, 0), tor, TR_UP);
|
|
}
|
|
|
|
static void saveRatioLimits(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
tr_variant* d = tr_variantDictAddDict(dict, TR_KEY_ratio_limit, 2);
|
|
tr_variantDictAddReal(d, TR_KEY_ratio_limit, tr_torrentGetRatioLimit(tor));
|
|
tr_variantDictAddInt(d, TR_KEY_ratio_mode, tr_torrentGetRatioMode(tor));
|
|
}
|
|
|
|
static void saveIdleLimits(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
tr_variant* d = tr_variantDictAddDict(dict, TR_KEY_idle_limit, 2);
|
|
tr_variantDictAddInt(d, TR_KEY_idle_limit, tr_torrentGetIdleLimit(tor));
|
|
tr_variantDictAddInt(d, TR_KEY_idle_mode, tr_torrentGetIdleMode(tor));
|
|
}
|
|
|
|
static void loadSingleSpeedLimit(tr_variant* d, tr_direction dir, tr_torrent* tor)
|
|
{
|
|
auto i = int64_t{};
|
|
auto boolVal = false;
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_speed_Bps, &i))
|
|
{
|
|
tr_torrentSetSpeedLimit_Bps(tor, dir, i);
|
|
}
|
|
else if (tr_variantDictFindInt(d, TR_KEY_speed, &i))
|
|
{
|
|
tr_torrentSetSpeedLimit_Bps(tor, dir, i * 1024);
|
|
}
|
|
|
|
if (tr_variantDictFindBool(d, TR_KEY_use_speed_limit, &boolVal))
|
|
{
|
|
tr_torrentUseSpeedLimit(tor, dir, boolVal);
|
|
}
|
|
|
|
if (tr_variantDictFindBool(d, TR_KEY_use_global_speed_limit, &boolVal))
|
|
{
|
|
tr_torrentUseSessionLimits(tor, boolVal);
|
|
}
|
|
}
|
|
|
|
static uint64_t loadSpeedLimits(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
tr_variant* d = nullptr;
|
|
if (tr_variantDictFindDict(dict, TR_KEY_speed_limit_up, &d))
|
|
{
|
|
loadSingleSpeedLimit(d, TR_UP, tor);
|
|
ret = TR_FR_SPEEDLIMIT;
|
|
}
|
|
|
|
if (tr_variantDictFindDict(dict, TR_KEY_speed_limit_down, &d))
|
|
{
|
|
loadSingleSpeedLimit(d, TR_DOWN, tor);
|
|
ret = TR_FR_SPEEDLIMIT;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static uint64_t loadRatioLimits(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
tr_variant* d = nullptr;
|
|
if (tr_variantDictFindDict(dict, TR_KEY_ratio_limit, &d))
|
|
{
|
|
auto dratio = double{};
|
|
if (tr_variantDictFindReal(d, TR_KEY_ratio_limit, &dratio))
|
|
{
|
|
tr_torrentSetRatioLimit(tor, dratio);
|
|
}
|
|
|
|
auto i = int64_t{};
|
|
if (tr_variantDictFindInt(d, TR_KEY_ratio_mode, &i))
|
|
{
|
|
tr_torrentSetRatioMode(tor, tr_ratiolimit(i));
|
|
}
|
|
|
|
ret = TR_FR_RATIOLIMIT;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static uint64_t loadIdleLimits(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
tr_variant* d = nullptr;
|
|
if (tr_variantDictFindDict(dict, TR_KEY_idle_limit, &d))
|
|
{
|
|
auto imin = int64_t{};
|
|
if (tr_variantDictFindInt(d, TR_KEY_idle_limit, &imin))
|
|
{
|
|
tr_torrentSetIdleLimit(tor, imin);
|
|
}
|
|
|
|
auto i = int64_t{};
|
|
if (tr_variantDictFindInt(d, TR_KEY_idle_mode, &i))
|
|
{
|
|
tr_torrentSetIdleMode(tor, tr_idlelimit(i));
|
|
}
|
|
|
|
ret = TR_FR_IDLELIMIT;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void saveName(tr_variant* dict, tr_torrent const* tor)
|
|
{
|
|
tr_variantDictAddStr(dict, TR_KEY_name, tr_torrentName(tor));
|
|
}
|
|
|
|
static uint64_t loadName(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
char const* name = nullptr;
|
|
if (tr_variantDictFindStr(dict, TR_KEY_name, &name, nullptr))
|
|
{
|
|
ret = TR_FR_NAME;
|
|
|
|
if (tr_strcmp0(tr_torrentName(tor), name) != 0)
|
|
{
|
|
tr_free(tor->info.name);
|
|
tor->info.name = tr_strdup(name);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void saveFilenames(tr_variant* dict, tr_torrent const* tor)
|
|
{
|
|
tr_file_index_t const n = tor->info.fileCount;
|
|
tr_file const* files = tor->info.files;
|
|
|
|
bool any_renamed = false;
|
|
|
|
for (tr_file_index_t i = 0; !any_renamed && i < n; ++i)
|
|
{
|
|
any_renamed = files[i].is_renamed;
|
|
}
|
|
|
|
if (any_renamed)
|
|
{
|
|
tr_variant* list = tr_variantDictAddList(dict, TR_KEY_files, n);
|
|
|
|
for (tr_file_index_t i = 0; i < n; ++i)
|
|
{
|
|
tr_variantListAddStr(list, files[i].is_renamed ? files[i].name : "");
|
|
}
|
|
}
|
|
}
|
|
|
|
static uint64_t loadFilenames(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
|
|
tr_variant* list = nullptr;
|
|
if (tr_variantDictFindList(dict, TR_KEY_files, &list))
|
|
{
|
|
size_t const n = tr_variantListSize(list);
|
|
tr_file* files = tor->info.files;
|
|
|
|
for (size_t i = 0; i < tor->info.fileCount && i < n; ++i)
|
|
{
|
|
char const* str = nullptr;
|
|
auto str_len = size_t{};
|
|
if (tr_variantGetStr(tr_variantListChild(list, i), &str, &str_len) && str != nullptr && str_len != 0)
|
|
{
|
|
tr_free(files[i].name);
|
|
files[i].name = tr_strndup(str, str_len);
|
|
files[i].is_renamed = true;
|
|
}
|
|
}
|
|
|
|
ret = TR_FR_FILENAMES;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
// TODO: Refactor this into a constructor for tr_variant
|
|
static void bitfieldToBenc(tr_bitfield const* b, tr_variant* benc)
|
|
{
|
|
if (b->hasAll())
|
|
{
|
|
tr_variantInitStr(benc, "all"sv);
|
|
}
|
|
else if (b->hasNone())
|
|
{
|
|
tr_variantInitStr(benc, "none"sv);
|
|
}
|
|
else
|
|
{
|
|
auto const raw = b->raw();
|
|
tr_variantInitRaw(benc, raw.data(), std::size(raw));
|
|
}
|
|
}
|
|
|
|
static void saveProgress(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
tr_info const* inf = tr_torrentInfo(tor);
|
|
time_t const now = tr_time();
|
|
|
|
tr_variant* const prog = tr_variantDictAddDict(dict, TR_KEY_progress, 3);
|
|
|
|
/* add the file/piece check timestamps... */
|
|
tr_variant* const l = tr_variantDictAddList(prog, TR_KEY_time_checked, inf->fileCount);
|
|
|
|
for (tr_file_index_t fi = 0; fi < inf->fileCount; ++fi)
|
|
{
|
|
time_t oldest_nonzero = now;
|
|
time_t newest = 0;
|
|
bool has_zero = false;
|
|
time_t const mtime = tr_torrentGetFileMTime(tor, fi);
|
|
tr_file const* f = &inf->files[fi];
|
|
|
|
/* get the oldest and newest nonzero timestamps for pieces in this file */
|
|
for (tr_piece_index_t i = f->firstPiece; i <= f->lastPiece; ++i)
|
|
{
|
|
tr_piece const* const p = &inf->pieces[i];
|
|
|
|
if (p->timeChecked == 0)
|
|
{
|
|
has_zero = true;
|
|
}
|
|
else if (oldest_nonzero > p->timeChecked)
|
|
{
|
|
oldest_nonzero = p->timeChecked;
|
|
}
|
|
|
|
if (newest < p->timeChecked)
|
|
{
|
|
newest = p->timeChecked;
|
|
}
|
|
}
|
|
|
|
/* If some of a file's pieces have been checked more recently than
|
|
the file's mtime, and some less recently, then that file will
|
|
have a list containing timestamps for each piece.
|
|
|
|
However, the most common use case is that the file doesn't change
|
|
after it's downloaded. To reduce overhead in the .resume file,
|
|
only a single timestamp is saved for the file if *all* or *none*
|
|
of the pieces were tested more recently than the file's mtime. */
|
|
|
|
if (!has_zero && mtime <= oldest_nonzero) /* all checked */
|
|
{
|
|
tr_variantListAddInt(l, oldest_nonzero);
|
|
}
|
|
else if (newest < mtime) /* none checked */
|
|
{
|
|
tr_variantListAddInt(l, newest);
|
|
}
|
|
else /* some are checked, some aren't... so list piece by piece */
|
|
{
|
|
int const offset = oldest_nonzero - 1;
|
|
tr_variant* ll = tr_variantListAddList(l, 2 + f->lastPiece - f->firstPiece);
|
|
tr_variantListAddInt(ll, offset);
|
|
|
|
for (tr_piece_index_t i = f->firstPiece; i <= f->lastPiece; ++i)
|
|
{
|
|
tr_piece const* const p = &inf->pieces[i];
|
|
|
|
tr_variantListAddInt(ll, p->timeChecked != 0 ? p->timeChecked - offset : 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* add the progress */
|
|
if (tor->completeness == TR_SEED)
|
|
{
|
|
tr_variantDictAddStr(prog, TR_KEY_have, "all"sv);
|
|
}
|
|
|
|
/* add the blocks bitfield */
|
|
bitfieldToBenc(tor->completion.blockBitfield, tr_variantDictAdd(prog, TR_KEY_blocks));
|
|
}
|
|
|
|
static uint64_t loadProgress(tr_variant* dict, tr_torrent* tor)
|
|
{
|
|
auto ret = uint64_t{};
|
|
tr_info const* inf = tr_torrentInfo(tor);
|
|
|
|
for (size_t i = 0; i < inf->pieceCount; ++i)
|
|
{
|
|
inf->pieces[i].timeChecked = 0;
|
|
}
|
|
|
|
tr_variant* prog = nullptr;
|
|
if (tr_variantDictFindDict(dict, TR_KEY_progress, &prog))
|
|
{
|
|
tr_variant* l = nullptr;
|
|
if (tr_variantDictFindList(prog, TR_KEY_time_checked, &l))
|
|
{
|
|
/* per-piece timestamps were added in 2.20.
|
|
|
|
If some of a file's pieces have been checked more recently than
|
|
the file's mtime, and some lest recently, then that file will
|
|
have a list containing timestamps for each piece.
|
|
|
|
However, the most common use case is that the file doesn't change
|
|
after it's downloaded. To reduce overhead in the .resume file,
|
|
only a single timestamp is saved for the file if *all* or *none*
|
|
of the pieces were tested more recently than the file's mtime. */
|
|
|
|
for (tr_file_index_t fi = 0; fi < inf->fileCount; ++fi)
|
|
{
|
|
tr_variant* b = tr_variantListChild(l, fi);
|
|
tr_file const* f = &inf->files[fi];
|
|
|
|
if (tr_variantIsInt(b))
|
|
{
|
|
auto t = int64_t{};
|
|
tr_variantGetInt(b, &t);
|
|
|
|
for (tr_piece_index_t i = f->firstPiece; i <= f->lastPiece; ++i)
|
|
{
|
|
inf->pieces[i].timeChecked = (time_t)t;
|
|
}
|
|
}
|
|
else if (tr_variantIsList(b))
|
|
{
|
|
int64_t offset = 0;
|
|
int const pieces = f->lastPiece + 1 - f->firstPiece;
|
|
|
|
tr_variantGetInt(tr_variantListChild(b, 0), &offset);
|
|
|
|
for (int i = 0; i < pieces; ++i)
|
|
{
|
|
int64_t t = 0;
|
|
tr_variantGetInt(tr_variantListChild(b, i + 1), &t);
|
|
inf->pieces[f->firstPiece + i].timeChecked = (time_t)(t != 0 ? t + offset : 0);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else if (tr_variantDictFindList(prog, TR_KEY_mtimes, &l))
|
|
{
|
|
/* Before 2.20, we stored the files' mtimes in the .resume file.
|
|
When loading the .resume file, a torrent's file would be flagged
|
|
as untested if its stored mtime didn't match its real mtime. */
|
|
|
|
for (tr_file_index_t fi = 0; fi < inf->fileCount; ++fi)
|
|
{
|
|
auto t = int64_t{};
|
|
|
|
if (tr_variantGetInt(tr_variantListChild(l, fi), &t))
|
|
{
|
|
tr_file const* f = &inf->files[fi];
|
|
time_t const mtime = tr_torrentGetFileMTime(tor, fi);
|
|
time_t const timeChecked = mtime == t ? mtime : 0;
|
|
|
|
for (tr_piece_index_t i = f->firstPiece; i <= f->lastPiece; ++i)
|
|
{
|
|
inf->pieces[i].timeChecked = timeChecked;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
auto blocks = tr_bitfield{ tor->blockCount };
|
|
|
|
auto rawlen = size_t{};
|
|
char const* err = nullptr;
|
|
char const* str = nullptr;
|
|
uint8_t const* raw = nullptr;
|
|
tr_variant const* const b = tr_variantDictFind(prog, TR_KEY_blocks);
|
|
if (b != nullptr)
|
|
{
|
|
uint8_t const* buf = nullptr;
|
|
auto buflen = size_t{};
|
|
|
|
if (!tr_variantGetRaw(b, &buf, &buflen))
|
|
{
|
|
err = "Invalid value for \"blocks\"";
|
|
}
|
|
else if (buflen == 3 && memcmp(buf, "all", 3) == 0)
|
|
{
|
|
blocks.setHasAll();
|
|
}
|
|
else if (buflen == 4 && memcmp(buf, "none", 4) == 0)
|
|
{
|
|
blocks.setHasNone();
|
|
}
|
|
else
|
|
{
|
|
blocks.setRaw(buf, buflen, true);
|
|
}
|
|
}
|
|
else if (tr_variantDictFindStr(prog, TR_KEY_have, &str, nullptr))
|
|
{
|
|
if (strcmp(str, "all") == 0)
|
|
{
|
|
blocks.setHasAll();
|
|
}
|
|
else
|
|
{
|
|
err = "Invalid value for HAVE";
|
|
}
|
|
}
|
|
else if (tr_variantDictFindRaw(prog, TR_KEY_bitfield, &raw, &rawlen))
|
|
{
|
|
blocks.setRaw(raw, rawlen, true);
|
|
}
|
|
else
|
|
{
|
|
err = "Couldn't find 'pieces' or 'have' or 'bitfield'";
|
|
}
|
|
|
|
if (err != nullptr)
|
|
{
|
|
tr_logAddTorDbg(tor, "Torrent needs to be verified - %s", err);
|
|
}
|
|
else
|
|
{
|
|
tr_cpBlockInit(&tor->completion, blocks);
|
|
}
|
|
|
|
ret = TR_FR_PROGRESS;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
void tr_torrentSaveResume(tr_torrent* tor)
|
|
{
|
|
tr_variant top;
|
|
|
|
if (!tr_isTorrent(tor))
|
|
{
|
|
return;
|
|
}
|
|
|
|
tr_variantInitDict(&top, 50); /* arbitrary "big enough" number */
|
|
tr_variantDictAddInt(&top, TR_KEY_seeding_time_seconds, tor->secondsSeeding);
|
|
tr_variantDictAddInt(&top, TR_KEY_downloading_time_seconds, tor->secondsDownloading);
|
|
tr_variantDictAddInt(&top, TR_KEY_activity_date, tor->activityDate);
|
|
tr_variantDictAddInt(&top, TR_KEY_added_date, tor->addedDate);
|
|
tr_variantDictAddInt(&top, TR_KEY_corrupt, tor->corruptPrev + tor->corruptCur);
|
|
tr_variantDictAddInt(&top, TR_KEY_done_date, tor->doneDate);
|
|
tr_variantDictAddStr(&top, TR_KEY_destination, tor->downloadDir);
|
|
|
|
if (tor->incompleteDir != nullptr)
|
|
{
|
|
tr_variantDictAddStr(&top, TR_KEY_incomplete_dir, tor->incompleteDir);
|
|
}
|
|
|
|
tr_variantDictAddInt(&top, TR_KEY_downloaded, tor->downloadedPrev + tor->downloadedCur);
|
|
tr_variantDictAddInt(&top, TR_KEY_uploaded, tor->uploadedPrev + tor->uploadedCur);
|
|
tr_variantDictAddInt(&top, TR_KEY_max_peers, tor->maxConnectedPeers);
|
|
tr_variantDictAddInt(&top, TR_KEY_bandwidth_priority, tr_torrentGetPriority(tor));
|
|
tr_variantDictAddBool(&top, TR_KEY_paused, !tor->isRunning && !tor->isQueued);
|
|
savePeers(&top, tor);
|
|
|
|
if (tr_torrentHasMetadata(tor))
|
|
{
|
|
saveFilePriorities(&top, tor);
|
|
saveDND(&top, tor);
|
|
saveProgress(&top, tor);
|
|
}
|
|
|
|
saveSpeedLimits(&top, tor);
|
|
saveRatioLimits(&top, tor);
|
|
saveIdleLimits(&top, tor);
|
|
saveFilenames(&top, tor);
|
|
saveName(&top, tor);
|
|
saveLabels(&top, tor);
|
|
|
|
char* const filename = getResumeFilename(tor, TR_METAINFO_BASENAME_HASH);
|
|
int const err = tr_variantToFile(&top, TR_VARIANT_FMT_BENC, filename);
|
|
if (err != 0)
|
|
{
|
|
tr_torrentSetLocalError(tor, "Unable to save resume file: %s", tr_strerror(err));
|
|
}
|
|
tr_free(filename);
|
|
|
|
tr_variantFree(&top);
|
|
}
|
|
|
|
static uint64_t loadFromFile(tr_torrent* tor, uint64_t fieldsToLoad, bool* didRenameToHashOnlyName)
|
|
{
|
|
TR_ASSERT(tr_isTorrent(tor));
|
|
|
|
auto boolVal = false;
|
|
auto const wasDirty = tor->isDirty;
|
|
auto fieldsLoaded = uint64_t{};
|
|
auto i = int64_t{};
|
|
auto len = size_t{};
|
|
auto top = tr_variant{};
|
|
char const* str = nullptr;
|
|
tr_error* error = nullptr;
|
|
|
|
if (didRenameToHashOnlyName != nullptr)
|
|
{
|
|
*didRenameToHashOnlyName = false;
|
|
}
|
|
|
|
char* const filename = getResumeFilename(tor, TR_METAINFO_BASENAME_HASH);
|
|
|
|
if (!tr_variantFromFile(&top, TR_VARIANT_FMT_BENC, filename, &error))
|
|
{
|
|
tr_logAddTorDbg(tor, "Couldn't read \"%s\": %s", filename, error->message);
|
|
tr_error_clear(&error);
|
|
|
|
char* old_filename = getResumeFilename(tor, TR_METAINFO_BASENAME_NAME_AND_PARTIAL_HASH);
|
|
|
|
if (!tr_variantFromFile(&top, TR_VARIANT_FMT_BENC, old_filename, &error))
|
|
{
|
|
tr_logAddTorDbg(tor, "Couldn't read \"%s\" either: %s", old_filename, error->message);
|
|
tr_error_free(error);
|
|
|
|
tr_free(old_filename);
|
|
tr_free(filename);
|
|
return fieldsLoaded;
|
|
}
|
|
|
|
if (tr_sys_path_rename(old_filename, filename, nullptr))
|
|
{
|
|
tr_logAddTorDbg(tor, "Migrated resume file from \"%s\" to \"%s\"", old_filename, filename);
|
|
|
|
if (didRenameToHashOnlyName != nullptr)
|
|
{
|
|
*didRenameToHashOnlyName = true;
|
|
}
|
|
}
|
|
|
|
tr_free(old_filename);
|
|
}
|
|
|
|
tr_logAddTorDbg(tor, "Read resume file \"%s\"", filename);
|
|
|
|
if ((fieldsToLoad & TR_FR_CORRUPT) != 0 && tr_variantDictFindInt(&top, TR_KEY_corrupt, &i))
|
|
{
|
|
tor->corruptPrev = i;
|
|
fieldsLoaded |= TR_FR_CORRUPT;
|
|
}
|
|
|
|
if ((fieldsToLoad & (TR_FR_PROGRESS | TR_FR_DOWNLOAD_DIR)) != 0 &&
|
|
tr_variantDictFindStr(&top, TR_KEY_destination, &str, &len) && !tr_str_is_empty(str))
|
|
{
|
|
bool const is_current_dir = tor->currentDir == tor->downloadDir;
|
|
tr_free(tor->downloadDir);
|
|
tor->downloadDir = tr_strndup(str, len);
|
|
|
|
if (is_current_dir)
|
|
{
|
|
tor->currentDir = tor->downloadDir;
|
|
}
|
|
|
|
fieldsLoaded |= TR_FR_DOWNLOAD_DIR;
|
|
}
|
|
|
|
if ((fieldsToLoad & (TR_FR_PROGRESS | TR_FR_INCOMPLETE_DIR)) != 0 &&
|
|
tr_variantDictFindStr(&top, TR_KEY_incomplete_dir, &str, &len) && !tr_str_is_empty(str))
|
|
{
|
|
bool const is_current_dir = tor->currentDir == tor->incompleteDir;
|
|
tr_free(tor->incompleteDir);
|
|
tor->incompleteDir = tr_strndup(str, len);
|
|
|
|
if (is_current_dir)
|
|
{
|
|
tor->currentDir = tor->incompleteDir;
|
|
}
|
|
|
|
fieldsLoaded |= TR_FR_INCOMPLETE_DIR;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_DOWNLOADED) != 0 && tr_variantDictFindInt(&top, TR_KEY_downloaded, &i))
|
|
{
|
|
tor->downloadedPrev = i;
|
|
fieldsLoaded |= TR_FR_DOWNLOADED;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_UPLOADED) != 0 && tr_variantDictFindInt(&top, TR_KEY_uploaded, &i))
|
|
{
|
|
tor->uploadedPrev = i;
|
|
fieldsLoaded |= TR_FR_UPLOADED;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_MAX_PEERS) != 0 && tr_variantDictFindInt(&top, TR_KEY_max_peers, &i))
|
|
{
|
|
tor->maxConnectedPeers = i;
|
|
fieldsLoaded |= TR_FR_MAX_PEERS;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_RUN) != 0 && tr_variantDictFindBool(&top, TR_KEY_paused, &boolVal))
|
|
{
|
|
tor->isRunning = !boolVal;
|
|
fieldsLoaded |= TR_FR_RUN;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_ADDED_DATE) != 0 && tr_variantDictFindInt(&top, TR_KEY_added_date, &i))
|
|
{
|
|
tor->addedDate = i;
|
|
fieldsLoaded |= TR_FR_ADDED_DATE;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_DONE_DATE) != 0 && tr_variantDictFindInt(&top, TR_KEY_done_date, &i))
|
|
{
|
|
tor->doneDate = i;
|
|
fieldsLoaded |= TR_FR_DONE_DATE;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_ACTIVITY_DATE) != 0 && tr_variantDictFindInt(&top, TR_KEY_activity_date, &i))
|
|
{
|
|
tr_torrentSetDateActive(tor, i);
|
|
fieldsLoaded |= TR_FR_ACTIVITY_DATE;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_TIME_SEEDING) != 0 && tr_variantDictFindInt(&top, TR_KEY_seeding_time_seconds, &i))
|
|
{
|
|
tor->secondsSeeding = i;
|
|
fieldsLoaded |= TR_FR_TIME_SEEDING;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_TIME_DOWNLOADING) != 0 && tr_variantDictFindInt(&top, TR_KEY_downloading_time_seconds, &i))
|
|
{
|
|
tor->secondsDownloading = i;
|
|
fieldsLoaded |= TR_FR_TIME_DOWNLOADING;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_BANDWIDTH_PRIORITY) != 0 && tr_variantDictFindInt(&top, TR_KEY_bandwidth_priority, &i) &&
|
|
tr_isPriority(i))
|
|
{
|
|
tr_torrentSetPriority(tor, i);
|
|
fieldsLoaded |= TR_FR_BANDWIDTH_PRIORITY;
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_PEERS) != 0)
|
|
{
|
|
fieldsLoaded |= loadPeers(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_FILE_PRIORITIES) != 0)
|
|
{
|
|
fieldsLoaded |= loadFilePriorities(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_PROGRESS) != 0)
|
|
{
|
|
fieldsLoaded |= loadProgress(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_DND) != 0)
|
|
{
|
|
fieldsLoaded |= loadDND(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_SPEEDLIMIT) != 0)
|
|
{
|
|
fieldsLoaded |= loadSpeedLimits(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_RATIOLIMIT) != 0)
|
|
{
|
|
fieldsLoaded |= loadRatioLimits(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_IDLELIMIT) != 0)
|
|
{
|
|
fieldsLoaded |= loadIdleLimits(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_FILENAMES) != 0)
|
|
{
|
|
fieldsLoaded |= loadFilenames(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_NAME) != 0)
|
|
{
|
|
fieldsLoaded |= loadName(&top, tor);
|
|
}
|
|
|
|
if ((fieldsToLoad & TR_FR_LABELS) != 0)
|
|
{
|
|
fieldsLoaded |= loadLabels(&top, tor);
|
|
}
|
|
|
|
/* loading the resume file triggers of a lot of changes,
|
|
* but none of them needs to trigger a re-saving of the
|
|
* same resume information... */
|
|
tor->isDirty = wasDirty;
|
|
|
|
tr_variantFree(&top);
|
|
tr_free(filename);
|
|
return fieldsLoaded;
|
|
}
|
|
|
|
static uint64_t setFromCtor(tr_torrent* tor, uint64_t fields, tr_ctor const* ctor, tr_ctorMode mode)
|
|
{
|
|
uint64_t ret = 0;
|
|
|
|
if ((fields & TR_FR_RUN) != 0)
|
|
{
|
|
auto isPaused = bool{};
|
|
if (tr_ctorGetPaused(ctor, mode, &isPaused))
|
|
{
|
|
tor->isRunning = !isPaused;
|
|
ret |= TR_FR_RUN;
|
|
}
|
|
}
|
|
|
|
if (((fields & TR_FR_MAX_PEERS) != 0) && tr_ctorGetPeerLimit(ctor, mode, &tor->maxConnectedPeers))
|
|
{
|
|
ret |= TR_FR_MAX_PEERS;
|
|
}
|
|
|
|
if ((fields & TR_FR_DOWNLOAD_DIR) != 0)
|
|
{
|
|
char const* path = nullptr;
|
|
if (tr_ctorGetDownloadDir(ctor, mode, &path) && !tr_str_is_empty(path))
|
|
{
|
|
ret |= TR_FR_DOWNLOAD_DIR;
|
|
tr_free(tor->downloadDir);
|
|
tor->downloadDir = tr_strdup(path);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static uint64_t useManditoryFields(tr_torrent* tor, uint64_t fields, tr_ctor const* ctor)
|
|
{
|
|
return setFromCtor(tor, fields, ctor, TR_FORCE);
|
|
}
|
|
|
|
static uint64_t useFallbackFields(tr_torrent* tor, uint64_t fields, tr_ctor const* ctor)
|
|
{
|
|
return setFromCtor(tor, fields, ctor, TR_FALLBACK);
|
|
}
|
|
|
|
uint64_t tr_torrentLoadResume(tr_torrent* tor, uint64_t fieldsToLoad, tr_ctor const* ctor, bool* didRenameToHashOnlyName)
|
|
{
|
|
TR_ASSERT(tr_isTorrent(tor));
|
|
|
|
uint64_t ret = 0;
|
|
|
|
ret |= useManditoryFields(tor, fieldsToLoad, ctor);
|
|
fieldsToLoad &= ~ret;
|
|
ret |= loadFromFile(tor, fieldsToLoad, didRenameToHashOnlyName);
|
|
fieldsToLoad &= ~ret;
|
|
ret |= useFallbackFields(tor, fieldsToLoad, ctor);
|
|
|
|
return ret;
|
|
}
|
|
|
|
void tr_torrentRemoveResume(tr_torrent const* tor)
|
|
{
|
|
char* filename = getResumeFilename(tor, TR_METAINFO_BASENAME_HASH);
|
|
tr_sys_path_remove(filename, nullptr);
|
|
tr_free(filename);
|
|
|
|
filename = getResumeFilename(tor, TR_METAINFO_BASENAME_NAME_AND_PARTIAL_HASH);
|
|
tr_sys_path_remove(filename, nullptr);
|
|
tr_free(filename);
|
|
}
|