Moved TrackerList and TrackerController out of the public API.

This commit is contained in:
Jari Sundell
2025-05-28 10:11:29 +02:00
committed by GitHub
parent 5fb61b2857
commit de6c48ca3f
12 changed files with 87 additions and 73 deletions
+18 -16
View File
@@ -13,8 +13,6 @@
#include <rak/regex.h> #include <rak/regex.h>
#include <torrent/rate.h> #include <torrent/rate.h>
#include <torrent/throttle.h> #include <torrent/throttle.h>
#include <torrent/tracker_controller.h>
#include <torrent/tracker_list.h>
#include <torrent/tracker/tracker.h> #include <torrent/tracker/tracker.h>
#include <torrent/connection_manager.h> #include <torrent/connection_manager.h>
#include <torrent/data/download_data.h> #include <torrent/data/download_data.h>
@@ -390,21 +388,24 @@ t_multicall(core::Download* download, const torrent::Object::list_type& args) {
// Add some pre-parsing of the commands, so we don't spend time // Add some pre-parsing of the commands, so we don't spend time
// parsing and searching command map for every single call. // parsing and searching command map for every single call.
torrent::Object resultRaw = torrent::Object::create_list(); torrent::Object result_raw = torrent::Object::create_list();
torrent::Object::list_type& result = resultRaw.as_list(); torrent::Object::list_type& result = result_raw.as_list();
for (int itr = 0, last = download->tracker_list()->size(); itr != last; itr++) { for (uint32_t idx = 0, last = download->tracker_list_size(); idx < last; idx++) {
torrent::Object::list_type& row = result.insert(result.end(), torrent::Object::create_list())->as_list(); auto& row = result.insert(result.end(), torrent::Object::create_list())->as_list();
auto tracker = download->tracker_controller().at(idx);
if (!tracker.is_valid())
continue;
for (torrent::Object::list_const_iterator cItr = ++args.begin(); cItr != args.end(); cItr++) { for (torrent::Object::list_const_iterator cItr = ++args.begin(); cItr != args.end(); cItr++) {
const std::string& cmd = cItr->as_string(); const std::string& cmd = cItr->as_string();
auto t = download->tracker_list()->at(itr);
row.push_back(rpc::parse_command(rpc::make_target(&t), cmd.c_str(), cmd.c_str() + cmd.size()).first); row.push_back(rpc::parse_command(rpc::make_target(&tracker), cmd.c_str(), cmd.c_str() + cmd.size()).first);
} }
} }
return resultRaw; return result_raw;
} }
torrent::Object torrent::Object
@@ -485,7 +486,8 @@ download_tracker_insert(core::Download* download, const torrent::Object::list_ty
if (group < 0 || group > 32) if (group < 0 || group > 32)
throw torrent::input_error("Tracker group number invalid."); throw torrent::input_error("Tracker group number invalid.");
download->download()->tracker_list()->insert_url(group, args.back().as_string(), true); download->tracker_controller().add_extra_tracker(group, args.back().as_string());
return torrent::Object(); return torrent::Object();
} }
@@ -589,7 +591,7 @@ d_list_remove(core::Download* download, const torrent::Object& rawArgs, const ch
#define CMD2_BIND_CL std::bind(&core::Download::connection_list, std::placeholders::_1) #define CMD2_BIND_CL std::bind(&core::Download::connection_list, std::placeholders::_1)
#define CMD2_BIND_FL std::bind(&core::Download::file_list, std::placeholders::_1) #define CMD2_BIND_FL std::bind(&core::Download::file_list, std::placeholders::_1)
#define CMD2_BIND_PL std::bind(&core::Download::c_peer_list, std::placeholders::_1) #define CMD2_BIND_PL std::bind(&core::Download::c_peer_list, std::placeholders::_1)
#define CMD2_BIND_TL std::bind(&core::Download::tracker_list, std::placeholders::_1) #define CMD2_BIND_TC std::bind(&core::Download::tracker_controller, std::placeholders::_1)
#define CMD2_BIND_INFO std::bind(&core::Download::info, std::placeholders::_1) #define CMD2_BIND_INFO std::bind(&core::Download::info, std::placeholders::_1)
#define CMD2_BIND_DATA std::bind(&core::Download::data, std::placeholders::_1) #define CMD2_BIND_DATA std::bind(&core::Download::data, std::placeholders::_1)
@@ -836,15 +838,15 @@ initialize_command_download() {
CMD2_DL_V ("d.tracker_announce", std::bind(&torrent::Download::manual_request, CMD2_BIND_DL, false)); CMD2_DL_V ("d.tracker_announce", std::bind(&torrent::Download::manual_request, CMD2_BIND_DL, false));
CMD2_DL_V ("d.tracker_announce.force", std::bind(&torrent::Download::manual_request, CMD2_BIND_DL, true)); CMD2_DL_V ("d.tracker_announce.force", std::bind(&torrent::Download::manual_request, CMD2_BIND_DL, true));
CMD2_DL ("d.tracker_numwant", std::bind(&torrent::TrackerList::numwant, CMD2_BIND_TL)); CMD2_DL ("d.tracker_numwant", std::bind(&torrent::tracker::TrackerControllerWrapper::numwant, CMD2_BIND_TC));
CMD2_DL_VALUE_V ("d.tracker_numwant.set", std::bind(&torrent::TrackerList::set_numwant, CMD2_BIND_TL, std::placeholders::_2)); CMD2_DL_VALUE_V ("d.tracker_numwant.set", std::bind(&torrent::tracker::TrackerControllerWrapper::set_numwant, CMD2_BIND_TC, std::placeholders::_2));
// TODO: Deprecate 'd.tracker_focus'. // TODO: Deprecate 'd.tracker_focus'.
CMD2_DL ("d.tracker_focus", std::bind(&core::Download::tracker_list_size, std::placeholders::_1)); CMD2_DL ("d.tracker_focus", std::bind(&core::Download::tracker_list_size, std::placeholders::_1));
CMD2_DL ("d.tracker_size", std::bind(&core::Download::tracker_list_size, std::placeholders::_1)); CMD2_DL ("d.tracker_size", std::bind(&core::Download::tracker_list_size, std::placeholders::_1));
CMD2_DL ("d.tracker.has_active", std::bind(&torrent::TrackerList::has_active, CMD2_BIND_TL)); CMD2_DL ("d.tracker.has_active", std::bind(&torrent::tracker::TrackerControllerWrapper::has_active_trackers, CMD2_BIND_TC));
CMD2_DL ("d.tracker.has_active_not_scrape", std::bind(&torrent::TrackerList::has_active_not_scrape, CMD2_BIND_TL)); CMD2_DL ("d.tracker.has_active_not_scrape", std::bind(&torrent::tracker::TrackerControllerWrapper::has_active_trackers_not_scrape, CMD2_BIND_TC));
CMD2_DL ("d.tracker.has_usable", std::bind(&torrent::TrackerList::has_usable, CMD2_BIND_TL)); CMD2_DL ("d.tracker.has_usable", std::bind(&torrent::tracker::TrackerControllerWrapper::has_usable_trackers, CMD2_BIND_TC));
CMD2_DL_LIST ("d.tracker.insert", std::bind(&download_tracker_insert, std::placeholders::_1, std::placeholders::_2)); CMD2_DL_LIST ("d.tracker.insert", std::bind(&download_tracker_insert, std::placeholders::_1, std::placeholders::_2));
CMD2_DL_VALUE_V ("d.tracker.send_scrape", [](auto download, uint64_t arg) { download->tracker_controller().scrape_request(arg); }); CMD2_DL_VALUE_V ("d.tracker.send_scrape", [](auto download, uint64_t arg) { download->tracker_controller().scrape_request(arg); });
+4 -3
View File
@@ -6,7 +6,6 @@
#include <rak/address_info.h> #include <rak/address_info.h>
#include <rak/path.h> #include <rak/path.h>
#include <torrent/connection_manager.h> #include <torrent/connection_manager.h>
#include <torrent/tracker_list.h>
#include <torrent/tracker/tracker.h> #include <torrent/tracker/tracker.h>
#include <torrent/torrent.h> #include <torrent/torrent.h>
#include <torrent/rate.h> #include <torrent/rate.h>
@@ -73,10 +72,12 @@ initialize_rpc() {
return (*d->file_list())[index].get(); return (*d->file_list())[index].get();
}; };
rpc::rpc.slot_find_tracker() = [](core::Download* d, uint32_t index) -> torrent::tracker::Tracker { rpc::rpc.slot_find_tracker() = [](core::Download* d, uint32_t index) -> torrent::tracker::Tracker {
if (index >= d->tracker_list()->size()) if (index >= d->tracker_controller().size())
throw torrent::input_error("invalid parameters: index not found"); throw torrent::input_error("invalid parameters: index not found");
return d->tracker_list()->at(index); // TODO: This should be rewritten to check if the tracker is valid and use a different
// function.
return d->tracker_controller().at(index);
}; };
rpc::rpc.slot_find_peer() = [](core::Download* d, const torrent::HashString& hash) -> torrent::Peer* { rpc::rpc.slot_find_peer() = [](core::Download* d, const torrent::HashString& hash) -> torrent::Peer* {
auto itr = d->connection_list()->find(hash.c_str()); auto itr = d->connection_list()->find(hash.c_str());
+16 -4
View File
@@ -62,11 +62,23 @@ apply_dht_add_node(const std::string& arg) {
torrent::Object torrent::Object
apply_enable_trackers(int64_t arg) { apply_enable_trackers(int64_t arg) {
for (core::Manager::DListItr itr = control->core()->download_list()->begin(), last = control->core()->download_list()->end(); itr != last; ++itr) { if (arg == 0) {
std::for_each((*itr)->tracker_list()->begin(), (*itr)->tracker_list()->end(), std::mem_fn(arg ? &torrent::tracker::Tracker::enable : &torrent::tracker::Tracker::disable)); for (auto itr : *control->core()->download_list())
itr->tracker_controller().for_each([](auto& tracker) { tracker.disable(); });
if (arg && !rpc::call_command_value("trackers.use_udp")) } else if (rpc::call_command_value("trackers.use_udp") == 0) {
(*itr)->enable_udp_trackers(false); for (auto itr : *control->core()->download_list()) {
itr->tracker_controller().for_each([](auto& tracker) {
if (tracker.type() == torrent::TRACKER_UDP)
tracker.disable();
else
tracker.enable();
});
}
} else {
for (auto itr : *control->core()->download_list())
itr->tracker_controller().for_each([](auto& tracker) { tracker.enable(); });
} }
return torrent::Object(); return torrent::Object();
+11 -8
View File
@@ -6,7 +6,6 @@
#include <torrent/exceptions.h> #include <torrent/exceptions.h>
#include <torrent/rate.h> #include <torrent/rate.h>
#include <torrent/torrent.h> #include <torrent/torrent.h>
#include <torrent/tracker_list.h>
#include <torrent/tracker/tracker.h> #include <torrent/tracker/tracker.h>
#include <torrent/data/file_list.h> #include <torrent/data/file_list.h>
@@ -38,13 +37,17 @@ Download::~Download() {
void void
Download::enable_udp_trackers(bool state) { Download::enable_udp_trackers(bool state) {
for (torrent::TrackerList::iterator itr = m_download.tracker_list()->begin(), last = m_download.tracker_list()->end(); itr != last; ++itr) for (int idx = 0, end = m_download.tracker_controller().size(); idx < end; ++idx) {
if (itr->type() == torrent::TRACKER_UDP) { auto tracker = m_download.tracker_controller().at(idx);
if (state)
itr->enable(); if (tracker.type() != torrent::TRACKER_UDP)
else continue;
itr->disable();
} if (state)
tracker.enable();
else
tracker.disable();
}
} }
uint32_t uint32_t
+2 -3
View File
@@ -5,9 +5,9 @@
#include <torrent/download.h> #include <torrent/download.h>
#include <torrent/download_info.h> #include <torrent/download_info.h>
#include <torrent/hash_string.h> #include <torrent/hash_string.h>
#include <torrent/tracker_list.h>
#include <torrent/data/file_list.h> #include <torrent/data/file_list.h>
#include <torrent/peer/connection_list.h> #include <torrent/peer/connection_list.h>
#include <torrent/tracker/wrappers.h>
#include "globals.h" #include "globals.h"
@@ -62,8 +62,7 @@ public:
torrent::Object* bencode() { return m_download.bencode(); } torrent::Object* bencode() { return m_download.bencode(); }
auto tracker_controller() { return m_download.tracker_controller(); } auto tracker_controller() { return m_download.tracker_controller(); }
tracker_list_type* tracker_list() { return m_download.tracker_list(); } uint32_t tracker_list_size() const { return m_download.c_tracker_controller().size(); }
uint32_t tracker_list_size() const { return m_download.tracker_list()->size(); }
auto connection_list() { return m_download.connection_list(); } auto connection_list() { return m_download.connection_list(); }
uint32_t connection_list_size() const; uint32_t connection_list_size() const;
+1 -1
View File
@@ -225,7 +225,7 @@ DownloadFactory::receive_success() {
torrent::Object* rtorrent = &root->insert_preserve_copy("rtorrent", torrent::Object::create_map()).first->second; torrent::Object* rtorrent = &root->insert_preserve_copy("rtorrent", torrent::Object::create_map()).first->second;
torrent::Object& resumeObject = root->insert_preserve_copy("libtorrent_resume", torrent::Object::create_map()).first->second; torrent::Object& resumeObject = root->insert_preserve_copy("libtorrent_resume", torrent::Object::create_map()).first->second;
rtorrent->insert_key("key", download->tracker_list()->key()); rtorrent->insert_key("key", download->tracker_controller().key());
initialize_rtorrent(download, rtorrent); initialize_rtorrent(download, rtorrent);
-1
View File
@@ -17,7 +17,6 @@
#include <torrent/error.h> #include <torrent/error.h>
#include <torrent/exceptions.h> #include <torrent/exceptions.h>
#include <torrent/object_stream.h> #include <torrent/object_stream.h>
#include <torrent/tracker_list.h>
#include <torrent/throttle.h> #include <torrent/throttle.h>
#include <torrent/utils/log.h> #include <torrent/utils/log.h>
+11 -8
View File
@@ -10,7 +10,6 @@
#include <torrent/rate.h> #include <torrent/rate.h>
#include <torrent/throttle.h> #include <torrent/throttle.h>
#include <torrent/torrent.h> #include <torrent/torrent.h>
#include <torrent/tracker_list.h>
#include <torrent/tracker/tracker.h> #include <torrent/tracker/tracker.h>
#include <torrent/data/file_list.h> #include <torrent/data/file_list.h>
#include <torrent/data/file_manager.h> #include <torrent/data/file_manager.h>
@@ -160,14 +159,18 @@ print_download_status(char* first, char* last, core::Download* d) {
first = print_buffer(first, last, "Checking hash [%2i%%]", first = print_buffer(first, last, "Checking hash [%2i%%]",
(d->download()->chunks_hashed() * 100) / d->download()->file_list()->size_chunks()); (d->download()->chunks_hashed() * 100) / d->download()->file_list()->size_chunks());
} else if (d->tracker_list()->has_active_not_scrape()) { } else if (d->tracker_controller().has_active_trackers_not_scrape()) {
torrent::TrackerList::iterator itr = auto tracker = d->tracker_controller().find_if([](const auto& t) {
std::find_if(d->tracker_list()->begin(), d->tracker_list()->end(), return t.is_busy_not_scrape();
std::mem_fn(&torrent::tracker::Tracker::is_busy_not_scrape)); });
auto status = itr->status();
first = print_buffer(first, last, "Tracker[%i:%i]: Connecting to %s %s", if (tracker.is_valid()) {
itr->group(), std::distance(d->tracker_list()->begin(), itr), itr->url().c_str(), status.c_str()); auto status = tracker.status();
first = print_buffer(first, last, "Tracker[%i]: Connecting to %s %s", tracker.group(), tracker.url().c_str(), status.c_str());
} else {
first = print_buffer(first, last, "Tracker: Connecting ...");
}
} else if (!d->message().empty()) { } else if (!d->message().empty()) {
first = print_buffer(first, last, "%s", d->message().c_str()); first = print_buffer(first, last, "%s", d->message().c_str());
+3 -5
View File
@@ -1,17 +1,15 @@
#include "config.h" #include "config.h"
#include "window_download_statusbar.h"
#include <torrent/rate.h> #include <torrent/rate.h>
#include <torrent/data/transfer_list.h> #include <torrent/data/transfer_list.h>
#include <torrent/tracker_controller.h>
#include <torrent/tracker_list.h>
#include <torrent/peer/connection_list.h> #include <torrent/peer/connection_list.h>
#include <torrent/peer/peer_list.h> #include <torrent/peer/peer_list.h>
#include "canvas.h" #include "canvas.h"
#include "globals.h" #include "globals.h"
#include "utils.h" #include "utils.h"
#include "window_download_statusbar.h"
#include "core/download.h" #include "core/download.h"
namespace display { namespace display {
@@ -57,7 +55,7 @@ WindowDownloadStatusbar::redraw() {
print_download_status(buffer.data(), last, m_download); print_download_status(buffer.data(), last, m_download);
m_canvas->print(0, 2, "[%c:%i] %s", m_canvas->print(0, 2, "[%c:%i] %s",
m_download->tracker_list()->has_active() ? 'C' : ' ', m_download->tracker_controller().has_active_trackers() ? 'C' : ' ',
(int)(m_download->download()->tracker_controller().seconds_to_next_timeout()), (int)(m_download->download()->tracker_controller().seconds_to_next_timeout()),
buffer.data()); buffer.data());
} }
+10 -14
View File
@@ -1,16 +1,14 @@
#include "config.h" #include "config.h"
#include "window_tracker_list.h"
#include <rak/algorithm.h> #include <rak/algorithm.h>
#include <rak/string_manip.h> #include <rak/string_manip.h>
#include <torrent/exceptions.h> #include <torrent/exceptions.h>
#include <torrent/tracker_list.h>
#include <torrent/tracker_controller.h>
#include <torrent/tracker/tracker.h> #include <torrent/tracker/tracker.h>
#include "core/download.h" #include "core/download.h"
#include "window_tracker_list.h"
namespace display { namespace display {
WindowTrackerList::WindowTrackerList(core::Download* d, unsigned int* focus) : WindowTrackerList::WindowTrackerList(core::Download* d, unsigned int* focus) :
@@ -25,27 +23,25 @@ WindowTrackerList::redraw() {
schedule_update(10); schedule_update(10);
m_canvas->erase(); m_canvas->erase();
unsigned int pos = 0; auto pos = 0u;
auto tl = m_download->tracker_list();
auto tc = m_download->tracker_controller(); auto tc = m_download->tracker_controller();
auto tc_size = tc.size();
m_canvas->print(2, pos, "Trackers: [Key: %08x] [%s %s %s]", m_canvas->print(2, pos, "Trackers: [Key: %08x] [%s %s %s]",
tl->key(), tc.key(),
tc.is_requesting() ? "req" : " ", tc.is_requesting() ? "req" : " ",
tc.is_promiscuous_mode() ? "prom" : " ", tc.is_promiscuous_mode() ? "prom" : " ",
tc.is_failure_mode() ? "fail" : " "); tc.is_failure_mode() ? "fail" : " ");
++pos; ++pos;
if (tl->size() == 0 || *m_focus >= tl->size()) if (tc_size == 0 || *m_focus >= tc_size)
return; return;
typedef std::pair<unsigned int, unsigned int> Range; auto range = rak::advance_bidirectional<unsigned int>(0, *m_focus, tc_size, (m_canvas->height() - 1) / 2);
auto group = tc.at(range.first).group();
Range range = rak::advance_bidirectional<unsigned int>(0, *m_focus, tl->size(), (m_canvas->height() - 1) / 2);
unsigned int group = tl->at(range.first).group();
while (range.first != range.second) { while (range.first != range.second) {
auto tracker = tl->at(range.first); auto tracker = tc.at(range.first);
if (tracker.group() == group) if (tracker.group() == group)
m_canvas->print(0, pos, "%2i:", group++); m_canvas->print(0, pos, "%2i:", group++);
@@ -93,7 +89,7 @@ WindowTrackerList::redraw() {
// If we're at the end of the range, check if we can // If we're at the end of the range, check if we can
// show one more line for the following tracker. // show one more line for the following tracker.
if (range.first == range.second && pos < m_canvas->height() && range.first < tl->size()) if (range.first == range.second && pos < m_canvas->height() && range.first < tc_size)
range.second++; range.second++;
} }
} }
-1
View File
@@ -8,7 +8,6 @@
#include <torrent/connection_manager.h> #include <torrent/connection_manager.h>
#include <torrent/throttle.h> #include <torrent/throttle.h>
#include <torrent/torrent.h> #include <torrent/torrent.h>
#include <torrent/tracker_list.h>
#include <torrent/data/file_list.h> #include <torrent/data/file_list.h>
#include <torrent/peer/connection_list.h> #include <torrent/peer/connection_list.h>
+11 -9
View File
@@ -1,7 +1,6 @@
#include "config.h" #include "config.h"
#include <torrent/exceptions.h> #include <torrent/exceptions.h>
#include <torrent/tracker_list.h>
#include <torrent/tracker/tracker.h> #include <torrent/tracker/tracker.h>
#include "display/frame.h" #include "display/frame.h"
@@ -67,7 +66,10 @@ ElementTrackerList::receive_disable() {
if (m_window == NULL) if (m_window == NULL)
throw torrent::internal_error("ui::ElementTrackerList::receive_disable(...) called on a disabled object"); throw torrent::internal_error("ui::ElementTrackerList::receive_disable(...) called on a disabled object");
auto t = m_download->download()->tracker_list()->at(m_focus); auto t = m_download->download()->tracker_controller().at(m_focus);
if (!t.is_valid())
return;
if (t.is_enabled()) if (t.is_enabled())
t.disable(); t.disable();
@@ -82,7 +84,7 @@ ElementTrackerList::receive_next() {
if (m_window == NULL) if (m_window == NULL)
throw torrent::internal_error("ui::ElementTrackerList::receive_next(...) called on a disabled object"); throw torrent::internal_error("ui::ElementTrackerList::receive_next(...) called on a disabled object");
if (++m_focus >= m_download->download()->tracker_list()->size()) if (++m_focus >= m_download->download()->tracker_controller().size())
m_focus = 0; m_focus = 0;
m_window->mark_dirty(); m_window->mark_dirty();
@@ -93,13 +95,13 @@ ElementTrackerList::receive_prev() {
if (m_window == NULL) if (m_window == NULL)
throw torrent::internal_error("ui::ElementTrackerList::receive_prev(...) called on a disabled object"); throw torrent::internal_error("ui::ElementTrackerList::receive_prev(...) called on a disabled object");
if (m_download->download()->tracker_list()->size() == 0) if (m_download->download()->tracker_controller().size() == 0)
return; return;
if (m_focus != 0) if (m_focus != 0)
--m_focus; --m_focus;
else else
m_focus = m_download->download()->tracker_list()->size() - 1; m_focus = m_download->download()->tracker_controller().size() - 1;
m_window->mark_dirty(); m_window->mark_dirty();
} }
@@ -109,12 +111,12 @@ ElementTrackerList::receive_cycle_group() {
if (m_window == NULL) if (m_window == NULL)
throw torrent::internal_error("ui::ElementTrackerList::receive_group_cycle(...) called on a disabled object"); throw torrent::internal_error("ui::ElementTrackerList::receive_group_cycle(...) called on a disabled object");
torrent::TrackerList* tl = m_download->tracker_list(); auto tracker = m_download->download()->tracker_controller().at(m_focus);
if (m_focus >= tl->size()) if (!tracker.is_valid())
throw torrent::internal_error("ui::ElementTrackerList::receive_group_cycle(...) called with an invalid focus"); return;
tl->cycle_group(tl->at(m_focus).group()); m_download->download()->tracker_controller().cycle_group(tracker.group());
m_window->mark_dirty(); m_window->mark_dirty();
} }