merged RC_1_1 into master
This commit is contained in:
commit
0fa24faec4
|
@ -33,6 +33,9 @@ POSSIBILITY OF SUCH DAMAGE.
|
||||||
#include "test.hpp"
|
#include "test.hpp"
|
||||||
#include "settings.hpp"
|
#include "settings.hpp"
|
||||||
#include "setup_swarm.hpp"
|
#include "setup_swarm.hpp"
|
||||||
|
#include "setup_transfer.hpp" // for addr()
|
||||||
|
#include "utils.hpp" // for print_alerts
|
||||||
|
#include "create_torrent.hpp"
|
||||||
#include "simulator/simulator.hpp"
|
#include "simulator/simulator.hpp"
|
||||||
#include "simulator/http_server.hpp"
|
#include "simulator/http_server.hpp"
|
||||||
#include "simulator/utils.hpp"
|
#include "simulator/utils.hpp"
|
||||||
|
@ -59,6 +62,12 @@ bool eq(Tp1 const lhs, Tp2 const rhs)
|
||||||
return std::abs(lt::duration_cast<seconds>(lhs - rhs).count()) <= 1;
|
return std::abs(lt::duration_cast<seconds>(lhs - rhs).count()) <= 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
std::shared_ptr<T> clone_ptr(std::shared_ptr<T> const& ptr)
|
||||||
|
{
|
||||||
|
return std::make_shared<T>(*ptr);
|
||||||
|
}
|
||||||
|
|
||||||
void test_interval(int interval)
|
void test_interval(int interval)
|
||||||
{
|
{
|
||||||
using sim::asio::ip::address_v4;
|
using sim::asio::ip::address_v4;
|
||||||
|
@ -942,7 +951,7 @@ std::shared_ptr<torrent_info> make_torrent(bool priv)
|
||||||
{
|
{
|
||||||
file_storage fs;
|
file_storage fs;
|
||||||
fs.add_file("foobar", 13241);
|
fs.add_file("foobar", 13241);
|
||||||
create_torrent ct(fs);
|
lt::create_torrent ct(fs);
|
||||||
|
|
||||||
ct.add_tracker("http://tracker.com:8080/announce");
|
ct.add_tracker("http://tracker.com:8080/announce");
|
||||||
|
|
||||||
|
@ -1128,6 +1137,112 @@ TORRENT_TEST(tracker_user_agent_privacy_mode_private_torrent)
|
||||||
TEST_EQUAL(got_announce, true);
|
TEST_EQUAL(got_announce, true);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// This test sets up two peers, one seed an one downloader. The downloader has
|
||||||
|
// two trackers, both in tier 0. The behavior we expect is that it picks one of
|
||||||
|
// the trackers at random and announces to it. Since both trackers are working,
|
||||||
|
// it should not announce to the tracker it did not initially pick.
|
||||||
|
|
||||||
|
// #error parameterize this test over adding the trackers into different tiers
|
||||||
|
// and setting "announce_to_all_tiers"
|
||||||
|
|
||||||
|
TORRENT_TEST(tracker_tiers)
|
||||||
|
{
|
||||||
|
using namespace libtorrent;
|
||||||
|
|
||||||
|
char const* peer0_ip = "50.0.0.1";
|
||||||
|
char const* peer1_ip = "50.0.0.2";
|
||||||
|
|
||||||
|
using asio::ip::address;
|
||||||
|
address peer0 = addr(peer0_ip);
|
||||||
|
address peer1 = addr(peer1_ip);
|
||||||
|
|
||||||
|
// setup the simulation
|
||||||
|
sim::default_config network_cfg;
|
||||||
|
sim::simulation sim{network_cfg};
|
||||||
|
sim::asio::io_service ios0 { sim, peer0 };
|
||||||
|
sim::asio::io_service ios1 { sim, peer1 };
|
||||||
|
|
||||||
|
sim::asio::io_service tracker1(sim, address_v4::from_string("3.0.0.1"));
|
||||||
|
sim::asio::io_service tracker2(sim, address_v4::from_string("3.0.0.2"));
|
||||||
|
sim::http_server http1(tracker1, 8080);
|
||||||
|
sim::http_server http2(tracker2, 8080);
|
||||||
|
|
||||||
|
bool received_announce[2] = {false, false};
|
||||||
|
http1.register_handler("/announce"
|
||||||
|
, [&](std::string method, std::string req
|
||||||
|
, std::map<std::string, std::string>&)
|
||||||
|
{
|
||||||
|
received_announce[0] = true;
|
||||||
|
std::string const ret = "d8:intervali60e5:peers0:e";
|
||||||
|
return sim::send_response(200, "OK", static_cast<int>(ret.size())) + ret;
|
||||||
|
});
|
||||||
|
|
||||||
|
http2.register_handler("/announce"
|
||||||
|
, [&](std::string method, std::string req
|
||||||
|
, std::map<std::string, std::string>&)
|
||||||
|
{
|
||||||
|
received_announce[1] = true;
|
||||||
|
std::string const ret = "d8:intervali60e5:peers0:e";
|
||||||
|
return sim::send_response(200, "OK", static_cast<int>(ret.size())) + ret;
|
||||||
|
});
|
||||||
|
|
||||||
|
lt::session_proxy zombie[2];
|
||||||
|
|
||||||
|
// setup settings pack to use for the session (customization point)
|
||||||
|
lt::settings_pack pack = settings();
|
||||||
|
// create session
|
||||||
|
std::shared_ptr<lt::session> ses[2];
|
||||||
|
pack.set_str(settings_pack::listen_interfaces, peer0_ip + std::string(":6881"));
|
||||||
|
ses[0] = std::make_shared<lt::session>(pack, ios0);
|
||||||
|
|
||||||
|
pack.set_str(settings_pack::listen_interfaces, peer1_ip + std::string(":6881"));
|
||||||
|
ses[1] = std::make_shared<lt::session>(pack, ios1);
|
||||||
|
|
||||||
|
// only monitor alerts for session 0 (the downloader)
|
||||||
|
print_alerts(*ses[0], [=](lt::session& ses, lt::alert const* a) {
|
||||||
|
if (auto ta = alert_cast<lt::add_torrent_alert>(a))
|
||||||
|
{
|
||||||
|
ta->handle.connect_peer(lt::tcp::endpoint(peer1, 6881));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
print_alerts(*ses[1]);
|
||||||
|
|
||||||
|
// the first peer is a downloader, the second peer is a seed
|
||||||
|
lt::add_torrent_params params = ::create_torrent(1);
|
||||||
|
auto ti2 = clone_ptr(params.ti);
|
||||||
|
params.flags &= ~lt::torrent_flags::auto_managed;
|
||||||
|
params.flags &= ~lt::torrent_flags::paused;
|
||||||
|
|
||||||
|
// These trackers are in the same tier. libtorrent is expected to pick one at
|
||||||
|
// random and stick to it, never announce to the other one.
|
||||||
|
params.ti->add_tracker("http://3.0.0.1:8080/announce", 0);
|
||||||
|
params.ti->add_tracker("http://3.0.0.2:8080/announce", 0);
|
||||||
|
params.save_path = save_path(0);
|
||||||
|
ses[0]->async_add_torrent(params);
|
||||||
|
|
||||||
|
params.ti = ti2;
|
||||||
|
params.save_path = save_path(1);
|
||||||
|
ses[1]->async_add_torrent(params);
|
||||||
|
|
||||||
|
sim::timer t(sim, lt::minutes(30), [&](boost::system::error_code const& ec)
|
||||||
|
{
|
||||||
|
TEST_CHECK(received_announce[0] != received_announce[1]);
|
||||||
|
TEST_CHECK(ses[0]->get_torrents()[0].status().is_seeding);
|
||||||
|
TEST_CHECK(ses[1]->get_torrents()[0].status().is_seeding);
|
||||||
|
|
||||||
|
// shut down
|
||||||
|
int idx = 0;
|
||||||
|
for (auto& s : ses)
|
||||||
|
{
|
||||||
|
zombie[idx++] = s->abort();
|
||||||
|
s.reset();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
sim.run();
|
||||||
|
}
|
||||||
|
|
||||||
// TODO: test external IP
|
// TODO: test external IP
|
||||||
// TODO: test with different queuing settings
|
// TODO: test with different queuing settings
|
||||||
// TODO: test when a torrent transitions from downloading to finished and
|
// TODO: test when a torrent transitions from downloading to finished and
|
||||||
|
|
|
@ -117,7 +117,8 @@ namespace libtorrent {
|
||||||
// event, we need to let this announce through
|
// event, we need to let this announce through
|
||||||
bool const need_send_complete = is_seed && !complete_sent;
|
bool const need_send_complete = is_seed && !complete_sent;
|
||||||
|
|
||||||
return now >= next_announce
|
// add some slack here for rounding errors
|
||||||
|
return now + seconds(1) >= next_announce
|
||||||
&& (now >= min_announce || need_send_complete)
|
&& (now >= min_announce || need_send_complete)
|
||||||
&& (fails < fail_limit || fail_limit == 0)
|
&& (fails < fail_limit || fail_limit == 0)
|
||||||
&& !updating;
|
&& !updating;
|
||||||
|
|
|
@ -2722,8 +2722,7 @@ namespace libtorrent {
|
||||||
// exclude redundant bytes if we should
|
// exclude redundant bytes if we should
|
||||||
if (!settings().get_bool(settings_pack::report_true_downloaded))
|
if (!settings().get_bool(settings_pack::report_true_downloaded))
|
||||||
req.downloaded -= m_total_redundant_bytes;
|
req.downloaded -= m_total_redundant_bytes;
|
||||||
if (settings().get_bool(settings_pack::report_redundant_bytes))
|
req.redundant = m_total_redundant_bytes;
|
||||||
req.redundant = m_total_redundant_bytes;
|
|
||||||
if (req.downloaded < 0) req.downloaded = 0;
|
if (req.downloaded < 0) req.downloaded = 0;
|
||||||
|
|
||||||
req.event = e;
|
req.event = e;
|
||||||
|
|
Loading…
Reference in New Issue