premiere-libtorrent/test/test_peer_list.cpp

707 lines
21 KiB
C++
Raw Normal View History

2014-07-06 21:18:00 +02:00
/*
Copyright (c) 2013, Arvid Norberg
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
are met:
* Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in
the documentation and/or other materials provided with the distribution.
* Neither the name of the author nor the names of its
contributors may be used to endorse or promote products derived
from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
*/
2014-10-26 08:34:31 +01:00
#include "libtorrent/peer_list.hpp"
2014-07-06 21:18:00 +02:00
#include "libtorrent/torrent_handle.hpp"
#include "libtorrent/torrent_peer_allocator.hpp"
#include "libtorrent/peer_connection_interface.hpp"
#include "libtorrent/stat.hpp"
#include "libtorrent/ip_voter.hpp"
#include "libtorrent/ip_filter.hpp"
#include "libtorrent/peer_info.hpp"
2014-07-19 04:17:19 +02:00
#include "libtorrent/random.hpp"
2014-07-06 21:18:00 +02:00
#include "test.hpp"
#include "setup_transfer.hpp"
#include <vector>
#include <stdarg.h>
using namespace libtorrent;
tcp::endpoint ep(char const* ip, int port)
{
return tcp::endpoint(address_v4::from_string(ip), port);
}
struct mock_peer_connection : peer_connection_interface
2015-04-12 08:39:16 +02:00
, boost::enable_shared_from_this<mock_peer_connection>
2014-07-06 21:18:00 +02:00
{
mock_peer_connection(bool out, tcp::endpoint const& ep)
: m_choked(false)
, m_outgoing(out)
, m_tp(NULL)
, m_remote(ep)
2015-04-12 08:39:16 +02:00
, m_disconnect_called(false)
2014-07-06 21:18:00 +02:00
{
2015-04-12 08:39:16 +02:00
fprintf(stderr, "construct %p\n", this);
2014-07-06 21:18:00 +02:00
for (int i = 0; i < 20; ++i) m_id[i] = rand();
}
2015-04-12 08:39:16 +02:00
virtual ~mock_peer_connection()
{
fprintf(stderr, "destruct %p\n", this);
}
#if defined TORRENT_LOGGING
2014-07-06 21:18:00 +02:00
virtual void peer_log(char const* fmt, ...) const
{
va_list v;
va_start(v, fmt);
vprintf(fmt, v);
va_end(v);
}
#endif
2015-04-12 08:39:16 +02:00
bool was_disconnected() const { return m_disconnect_called; }
2014-07-06 21:18:00 +02:00
libtorrent::stat m_stat;
bool m_choked;
bool m_outgoing;
torrent_peer* m_tp;
tcp::endpoint m_remote;
peer_id m_id;
2015-04-12 08:39:16 +02:00
bool m_disconnect_called;
2014-07-06 21:18:00 +02:00
virtual void get_peer_info(peer_info& p) const {}
virtual tcp::endpoint const& remote() const { return m_remote; }
virtual tcp::endpoint local_endpoint() const { return ep("127.0.0.1", 8080); }
virtual void disconnect(error_code const& ec
, operation_t op, int error = 0)
2015-04-12 08:39:16 +02:00
{
/* remove from mock_torrent list */ m_tp = 0;
m_disconnect_called = true;
}
2014-07-06 21:18:00 +02:00
virtual peer_id const& pid() const { return m_id; }
virtual void set_holepunch_mode() {}
virtual torrent_peer* peer_info_struct() const { return m_tp; }
virtual void set_peer_info(torrent_peer* pi) { m_tp = pi; }
virtual bool is_outgoing() const { return m_outgoing; }
virtual void add_stat(boost::int64_t downloaded, boost::int64_t uploaded)
2014-07-06 21:18:00 +02:00
{ m_stat.add_stat(downloaded, uploaded); }
virtual bool fast_reconnect() const { return true; }
virtual bool is_choked() const { return m_choked; }
virtual bool failed() const { return false; }
virtual libtorrent::stat const& statistics() const { return m_stat; }
};
struct mock_torrent
{
mock_torrent() : m_p(NULL) {}
virtual ~mock_torrent() {}
bool connect_to_peer(torrent_peer* peerinfo, bool ignore_limit = false)
{
TORRENT_ASSERT(peerinfo->connection == NULL);
if (peerinfo->connection) return false;
2015-04-12 08:39:16 +02:00
boost::shared_ptr<mock_peer_connection> c
= boost::make_shared<mock_peer_connection>(true, peerinfo->ip());
2014-07-06 21:18:00 +02:00
m_connections.push_back(c);
m_p->set_connection(peerinfo, c.get());
return true;
}
#if defined TORRENT_LOGGING
2014-07-06 21:18:00 +02:00
void debug_log(const char* fmt, ...) const
{
va_list v;
va_start(v, fmt);
vprintf(fmt, v);
va_end(v);
}
#endif
2014-10-26 08:34:31 +01:00
peer_list* m_p;
2014-07-06 21:18:00 +02:00
private:
std::vector<boost::shared_ptr<mock_peer_connection> > m_connections;
};
2015-04-12 08:39:16 +02:00
torrent_state init_state(torrent_peer_allocator& allocator
, external_ip& ext_ip)
2014-07-06 21:18:00 +02:00
{
torrent_state st;
st.is_finished = false;
st.is_paused = false;
st.max_peerlist_size = 1000;
st.allow_multiple_connections_per_ip = false;
st.peer_allocator = &allocator;
st.ip = &ext_ip;
st.port = 9999;
2015-04-12 08:39:16 +02:00
return st;
}
torrent_peer* add_peer(peer_list& p, torrent_state& st, tcp::endpoint const& ep)
{
int cc = p.num_connect_candidates();
torrent_peer* peer = p.add_peer(ep, 0, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), cc + 1);
TEST_EQUAL(peer->port, ep.port());
st.erased.clear();
return peer;
}
void connect_peer(peer_list& p, mock_torrent& t, torrent_state& st)
{
torrent_peer* tp = p.connect_one_peer(0, &st);
TEST_CHECK(tp);
t.connect_to_peer(tp);
st.erased.clear();
TEST_CHECK(tp->connection);
}
int test_main()
{
torrent_peer_allocator allocator;
external_ip ext_ip;
2014-07-06 21:18:00 +02:00
// test multiple peers with the same IP
// when disallowing it
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
TEST_EQUAL(p.num_connect_candidates(), 0);
torrent_peer* peer1 = p.add_peer(ep("10.0.0.2", 3000), 0, 0, &st);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(p.num_connect_candidates(), 1);
st.erased.clear();
torrent_peer* peer2 = p.add_peer(ep("10.0.0.2", 9020), 0, 0, &st);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(peer1, peer2);
TEST_EQUAL(p.num_connect_candidates(), 1);
st.erased.clear();
}
// test multiple peers with the same IP
// when allowing it
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = true;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
torrent_peer* peer1 = p.add_peer(ep("10.0.0.2", 3000), 0, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), 1);
TEST_EQUAL(p.num_peers(), 1);
st.erased.clear();
torrent_peer* peer2 = p.add_peer(ep("10.0.0.2", 9020), 0, 0, &st);
TEST_EQUAL(p.num_peers(), 2);
TEST_CHECK(peer1 != peer2);
TEST_EQUAL(p.num_connect_candidates(), 2);
st.erased.clear();
}
// test adding two peers with the same IP, but different ports, to
// make sure they can be connected at the same time
// with allow_multiple_connections_per_ip enabled
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = true;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
torrent_peer* peer1 = p.add_peer(ep("10.0.0.2", 3000), 0, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), 1);
st.erased.clear();
TEST_EQUAL(p.num_peers(), 1);
torrent_peer* tp = p.connect_one_peer(0, &st);
TEST_CHECK(tp);
t.connect_to_peer(tp);
st.erased.clear();
// we only have one peer, we can't
// connect another one
tp = p.connect_one_peer(0, &st);
TEST_CHECK(tp == NULL);
st.erased.clear();
torrent_peer* peer2 = p.add_peer(ep("10.0.0.2", 9020), 0, 0, &st);
TEST_EQUAL(p.num_peers(), 2);
TEST_CHECK(peer1 != peer2);
TEST_EQUAL(p.num_connect_candidates(), 1);
st.erased.clear();
tp = p.connect_one_peer(0, &st);
TEST_CHECK(tp);
t.connect_to_peer(tp);
TEST_EQUAL(p.num_connect_candidates(), 0);
st.erased.clear();
}
// test adding two peers with the same IP, but different ports, to
// make sure they can not be connected at the same time
// with allow_multiple_connections_per_ip disabled
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = false;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
torrent_peer* peer1 = p.add_peer(ep("10.0.0.2", 3000), 0, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), 1);
TEST_EQUAL(peer1->port, 3000);
st.erased.clear();
TEST_EQUAL(p.num_peers(), 1);
torrent_peer* tp = p.connect_one_peer(0, &st);
TEST_CHECK(tp);
t.connect_to_peer(tp);
st.erased.clear();
// we only have one peer, we can't
// connect another one
tp = p.connect_one_peer(0, &st);
TEST_CHECK(tp == NULL);
st.erased.clear();
torrent_peer* peer2 = p.add_peer(ep("10.0.0.2", 9020), 0, 0, &st);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(peer2->port, 9020);
TEST_CHECK(peer1 == peer2);
TEST_EQUAL(p.num_connect_candidates(), 0);
st.erased.clear();
}
// test incoming connection
// and update_peer_port
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = false;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
TEST_EQUAL(p.num_connect_candidates(), 0);
2015-04-12 08:39:16 +02:00
boost::shared_ptr<mock_peer_connection> c
= boost::make_shared<mock_peer_connection>(true, ep("10.0.0.1", 8080));
2014-07-06 21:18:00 +02:00
p.new_connection(*c, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), 0);
TEST_EQUAL(p.num_peers(), 1);
st.erased.clear();
p.update_peer_port(4000, c->peer_info_struct(), peer_info::incoming, &st);
TEST_EQUAL(p.num_connect_candidates(), 0);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(c->peer_info_struct()->port, 4000);
st.erased.clear();
}
// test incoming connection
// and update_peer_port, causing collission
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = true;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
torrent_peer* peer2 = p.add_peer(ep("10.0.0.1", 4000), 0, 0, &st);
TEST_CHECK(peer2);
TEST_EQUAL(p.num_connect_candidates(), 1);
2015-04-12 08:39:16 +02:00
boost::shared_ptr<mock_peer_connection> c
= boost::make_shared<mock_peer_connection>(true, ep("10.0.0.1", 8080));
2014-07-06 21:18:00 +02:00
p.new_connection(*c, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), 1);
// at this point we have two peers, because we think they have different
// ports
TEST_EQUAL(p.num_peers(), 2);
st.erased.clear();
// this peer will end up having the same port as the existing peer in the list
p.update_peer_port(4000, c->peer_info_struct(), peer_info::incoming, &st);
TEST_EQUAL(p.num_connect_candidates(), 0);
// the expected behavior is to replace that one
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(c->peer_info_struct()->port, 4000);
st.erased.clear();
}
// test ip filter
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = false;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
2015-04-12 08:39:16 +02:00
// add peer 1
torrent_peer* peer1 = add_peer(p, st, ep("10.0.0.2", 3000));
torrent_peer* peer2 = add_peer(p, st, ep("11.0.0.2", 9020));
2014-07-06 21:18:00 +02:00
TEST_CHECK(peer1 != peer2);
2015-04-12 08:39:16 +02:00
connect_peer(p, t, st);
connect_peer(p, t, st);
2014-07-06 21:18:00 +02:00
2015-04-12 08:39:16 +02:00
boost::shared_ptr<mock_peer_connection> con1
= static_cast<mock_peer_connection*>(peer1->connection)->shared_from_this();
TEST_EQUAL(con1->was_disconnected(), false);
boost::shared_ptr<mock_peer_connection> con2
= static_cast<mock_peer_connection*>(peer2->connection)->shared_from_this();
TEST_EQUAL(con2->was_disconnected(), false);
2014-07-06 21:18:00 +02:00
// now, filter one of the IPs and make sure the peer is removed
ip_filter filter;
filter.add_rule(address_v4::from_string("11.0.0.0"), address_v4::from_string("255.255.255.255"), 1);
std::vector<address> banned;
p.apply_ip_filter(filter, &st, banned);
// we just erased a peer, because it was filtered by the ip filter
TEST_EQUAL(st.erased.size(), 1);
TEST_EQUAL(p.num_connect_candidates(), 0);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(banned.size(), 1);
TEST_EQUAL(banned[0], address_v4::from_string("11.0.0.2"));
2015-04-12 08:39:16 +02:00
TEST_EQUAL(con2->was_disconnected(), true);
TEST_EQUAL(con1->was_disconnected(), false);
}
// test port filter
{
torrent_state st = init_state(allocator, ext_ip);
mock_torrent t;
st.allow_multiple_connections_per_ip = false;
peer_list p;
t.m_p = &p;
// add peer 1
torrent_peer* peer1 = add_peer(p, st, ep("10.0.0.2", 3000));
torrent_peer* peer2 = add_peer(p, st, ep("11.0.0.2", 9020));
TEST_CHECK(peer1 != peer2);
connect_peer(p, t, st);
connect_peer(p, t, st);
boost::shared_ptr<mock_peer_connection> con1
= static_cast<mock_peer_connection*>(peer1->connection)->shared_from_this();
TEST_EQUAL(con1->was_disconnected(), false);
boost::shared_ptr<mock_peer_connection> con2
= static_cast<mock_peer_connection*>(peer2->connection)->shared_from_this();
TEST_EQUAL(con2->was_disconnected(), false);
// now, filter one of the IPs and make sure the peer is removed
port_filter filter;
filter.add_rule(9000, 10000, 1);
std::vector<address> banned;
p.apply_port_filter(filter, &st, banned);
// we just erased a peer, because it was filtered by the ip filter
TEST_EQUAL(st.erased.size(), 1);
TEST_EQUAL(p.num_connect_candidates(), 0);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(banned.size(), 1);
TEST_EQUAL(banned[0], address_v4::from_string("11.0.0.2"));
TEST_EQUAL(con2->was_disconnected(), true);
TEST_EQUAL(con1->was_disconnected(), false);
2014-07-06 21:18:00 +02:00
}
// test banning peers
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.allow_multiple_connections_per_ip = false;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
2015-04-12 08:39:16 +02:00
torrent_peer* peer1 = add_peer(p, st, ep("10.0.0.1", 4000));
2014-07-06 21:18:00 +02:00
TEST_EQUAL(p.num_connect_candidates(), 1);
2015-04-12 08:39:16 +02:00
boost::shared_ptr<mock_peer_connection> c
= boost::make_shared<mock_peer_connection>(true, ep("10.0.0.1", 8080));
2014-07-06 21:18:00 +02:00
p.new_connection(*c, 0, &st);
TEST_EQUAL(p.num_connect_candidates(), 0);
TEST_EQUAL(p.num_peers(), 1);
st.erased.clear();
// now, ban the peer
bool ok = p.ban_peer(c->peer_info_struct());
TEST_EQUAL(ok, true);
TEST_EQUAL(peer1->banned, true);
// we still have it in the list
TEST_EQUAL(p.num_peers(), 1);
// it's just not a connect candidate, nor allowed to receive incoming connections
TEST_EQUAL(p.num_connect_candidates(), 0);
p.connection_closed(*c, 0, &st);
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(p.num_connect_candidates(), 0);
st.erased.clear();
2015-04-12 08:39:16 +02:00
c = boost::make_shared<mock_peer_connection>(true, ep("10.0.0.1", 8080));
2014-07-06 21:18:00 +02:00
ok = p.new_connection(*c, 0, &st);
// since it's banned, we should not allow this incoming connection
TEST_EQUAL(ok, false);
TEST_EQUAL(p.num_connect_candidates(), 0);
st.erased.clear();
}
// test erase_peers when we fill up the peer list
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2014-07-06 21:18:00 +02:00
mock_torrent t;
st.max_peerlist_size = 100;
st.allow_multiple_connections_per_ip = true;
2014-10-26 08:34:31 +01:00
peer_list p;
2014-07-06 21:18:00 +02:00
t.m_p = &p;
for (int i = 0; i < 100; ++i)
{
TEST_EQUAL(st.erased.size(), 0);
2015-04-12 08:39:16 +02:00
torrent_peer* peer = add_peer(p, st, rand_tcp_ep());
2014-07-06 21:18:00 +02:00
TEST_CHECK(peer);
if (peer == NULL || st.erased.size() > 0)
{
2015-04-12 08:39:16 +02:00
fprintf(stderr, "unexpected rejection of peer: %d in list. "
"added peer %p, erased %d peers\n"
2014-07-06 21:18:00 +02:00
, p.num_peers(), peer, int(st.erased.size()));
}
}
TEST_EQUAL(p.num_peers(), 100);
// trigger the eviction of one peer
torrent_peer* peer = p.add_peer(rand_tcp_ep(), 0, 0, &st);
// we either removed an existing peer, or rejected this one
// either is valid behavior when the list is full
2014-07-06 21:18:00 +02:00
TEST_CHECK(st.erased.size() == 1 || peer == NULL);
}
// test set_ip_filter
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
std::vector<address> banned;
mock_torrent t;
peer_list p;
t.m_p = &p;
for (int i = 0; i < 100; ++i)
{
p.add_peer(tcp::endpoint(
address_v4((10 << 24) + ((i + 10) << 16)), 353), 0, 0, &st);
TEST_EQUAL(st.erased.size(), 0);
st.erased.clear();
}
TEST_EQUAL(p.num_peers(), 100);
TEST_EQUAL(p.num_connect_candidates(), 100);
// trigger the removal of one peer
ip_filter filter;
filter.add_rule(address_v4::from_string("10.13.0.0")
, address_v4::from_string("10.13.255.255"), ip_filter::blocked);
p.apply_ip_filter(filter, &st, banned);
TEST_EQUAL(st.erased.size(), 1);
TEST_EQUAL(st.erased[0]->address(), address_v4::from_string("10.13.0.0"));
TEST_EQUAL(p.num_peers(), 99);
TEST_EQUAL(p.num_connect_candidates(), 99);
}
// test set_port_filter
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
std::vector<address> banned;
mock_torrent t;
peer_list p;
t.m_p = &p;
for (int i = 0; i < 100; ++i)
{
p.add_peer(tcp::endpoint(
address_v4((10 << 24) + ((i + 10) << 16)), i + 10), 0, 0, &st);
TEST_EQUAL(st.erased.size(), 0);
st.erased.clear();
}
TEST_EQUAL(p.num_peers(), 100);
TEST_EQUAL(p.num_connect_candidates(), 100);
// trigger the removal of one peer
port_filter filter;
filter.add_rule(13, 13, port_filter::blocked);
p.apply_port_filter(filter, &st, banned);
TEST_EQUAL(st.erased.size(), 1);
TEST_EQUAL(st.erased[0]->address(), address_v4::from_string("10.13.0.0"));
TEST_EQUAL(st.erased[0]->port, 13);
TEST_EQUAL(p.num_peers(), 99);
TEST_EQUAL(p.num_connect_candidates(), 99);
}
// test set_max_failcount
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
mock_torrent t;
peer_list p;
t.m_p = &p;
for (int i = 0; i < 100; ++i)
{
torrent_peer* peer = p.add_peer(tcp::endpoint(
address_v4((10 << 24) + ((i + 10) << 16)), i + 10), 0, 0, &st);
TEST_EQUAL(st.erased.size(), 0);
st.erased.clear();
// every other peer has a failcount of 1
if (i % 2) p.inc_failcount(peer);
}
TEST_EQUAL(p.num_peers(), 100);
TEST_EQUAL(p.num_connect_candidates(), 100);
// set the max failcount to 1 and observe how half the peers no longer
// are connect candidates
st.max_failcount = 1;
p.set_max_failcount(&st);
TEST_EQUAL(p.num_connect_candidates(), 50);
TEST_EQUAL(p.num_peers(), 100);
}
// test set_seed
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
mock_torrent t;
peer_list p;
t.m_p = &p;
for (int i = 0; i < 100; ++i)
{
torrent_peer* peer = p.add_peer(tcp::endpoint(
address_v4((10 << 24) + ((i + 10) << 16)), i + 10), 0, 0, &st);
TEST_EQUAL(st.erased.size(), 0);
st.erased.clear();
// make every other peer a seed
if (i % 2) p.set_seed(peer, true);
}
TEST_EQUAL(p.num_peers(), 100);
TEST_EQUAL(p.num_connect_candidates(), 100);
// now, the torrent completes and we're no longer interested in
// connecting to seeds. Make sure half the peers are no longer
// considered connect candidates
st.is_finished = true;
// this will make the peer_list recalculate the connect candidates
std::vector<torrent_peer*> peers;
2015-04-11 17:25:24 +02:00
p.connect_one_peer(1, &st);
TEST_EQUAL(p.num_connect_candidates(), 50);
TEST_EQUAL(p.num_peers(), 100);
}
2015-01-21 03:58:32 +01:00
// test has_peer
{
2015-04-12 08:39:16 +02:00
torrent_state st = init_state(allocator, ext_ip);
2015-01-21 03:58:32 +01:00
std::vector<address> banned;
mock_torrent t;
peer_list p;
t.m_p = &p;
2015-04-12 08:39:16 +02:00
torrent_peer* peer1 = add_peer(p, st, ep("10.10.0.1", 10));
torrent_peer* peer2 = add_peer(p, st, ep("10.10.0.2", 11));
2015-01-21 03:58:32 +01:00
TEST_EQUAL(p.num_peers(), 2);
TEST_EQUAL(p.num_connect_candidates(), 2);
TEST_EQUAL(p.has_peer(peer1), true);
TEST_EQUAL(p.has_peer(peer2), true);
ip_filter filter;
filter.add_rule(address_v4::from_string("10.10.0.1")
, address_v4::from_string("10.10.0.1"), ip_filter::blocked);
p.apply_ip_filter(filter, &st, banned);
TEST_EQUAL(st.erased.size(), 1);
st.erased.clear();
TEST_EQUAL(p.num_peers(), 1);
TEST_EQUAL(p.num_connect_candidates(), 1);
TEST_EQUAL(p.has_peer(peer1), false);
TEST_EQUAL(p.has_peer(peer2), true);
}
2015-04-12 08:39:16 +02:00
// test connect_candidates torrent_finish
{
torrent_state st = init_state(allocator, ext_ip);
std::vector<address> banned;
mock_torrent t;
peer_list p;
t.m_p = &p;
torrent_peer* peer1 = add_peer(p, st, ep("10.10.0.1", 10));
p.set_seed(peer1, true);
torrent_peer* peer2 = add_peer(p, st, ep("10.10.0.2", 11));
p.set_seed(peer2, true);
torrent_peer* peer3 = add_peer(p, st, ep("10.10.0.3", 11));
p.set_seed(peer3, true);
torrent_peer* peer4 = add_peer(p, st, ep("10.10.0.4", 11));
torrent_peer* peer5 = add_peer(p, st, ep("10.10.0.5", 11));
TEST_EQUAL(p.num_peers(), 5);
TEST_EQUAL(p.num_connect_candidates(), 5);
st.is_finished = true;
// we're finished downloading now, only the non-seeds are
// connect candidates
// connect to one of them
connect_peer(p, t, st);
TEST_EQUAL(p.num_peers(), 5);
// and there should be one left
TEST_EQUAL(p.num_connect_candidates(), 1);
}
2014-07-06 21:18:00 +02:00
// TODO: test erasing peers
// TODO: test logic for which connection to keep when receiving an incoming
// connection to the same peer as we just made an outgoing connection to
2015-01-21 03:58:32 +01:00
// TODO: test update_peer_port with allow_multiple_connections_per_ip and without
2014-07-06 21:18:00 +02:00
// TODO: test add i2p peers
// TODO: test allow_i2p_mixed
// TODO: test insert_peer failing with all error conditions
2014-07-06 21:18:00 +02:00
// TODO: test IPv6
// TODO: test connect_to_peer() failing
// TODO: test connection_closed
2015-04-12 08:39:16 +02:00
// TODO: connect candidates recalculation when finishing torrent
// TODO: connect candidates recalculation when incrementing failcount
// TODO: connect candidates recalculation when setting failcount
2014-07-06 21:18:00 +02:00
return 0;
}