clean up of some test and fix warnings

This commit is contained in:
Arvid Norberg 2015-04-11 15:25:24 +00:00
parent fb40cb484e
commit b3e061ca94
4 changed files with 77 additions and 32 deletions

View File

@ -61,31 +61,37 @@ namespace libtorrent
// generated, they are put here to explicitly make them part // generated, they are put here to explicitly make them part
// of libtorrent and properly exported by the .dll. // of libtorrent and properly exported by the .dll.
file_storage::file_storage(file_storage const& f) file_storage::file_storage(file_storage const& f)
: m_files(f.m_files) : m_piece_length(f.m_piece_length)
, m_num_pieces(f.m_num_pieces)
, m_files(f.m_files)
, m_file_hashes(f.m_file_hashes) , m_file_hashes(f.m_file_hashes)
, m_symlinks(f.m_symlinks) , m_symlinks(f.m_symlinks)
, m_mtime(f.m_mtime) , m_mtime(f.m_mtime)
#ifndef TORRENT_NO_DEPRECATE
, m_file_base(f.m_file_base) , m_file_base(f.m_file_base)
#endif
, m_paths(f.m_paths) , m_paths(f.m_paths)
, m_name(f.m_name) , m_name(f.m_name)
, m_total_size(f.m_total_size) , m_total_size(f.m_total_size)
, m_num_pieces(f.m_num_pieces) , m_num_files(f.m_num_files)
, m_piece_length(f.m_piece_length)
{ {
} }
file_storage& file_storage::operator=(file_storage const& f) file_storage& file_storage::operator=(file_storage const& f)
{ {
m_piece_length = f.m_piece_length;
m_num_pieces = f.m_num_pieces;
m_files = f.m_files; m_files = f.m_files;
m_file_hashes = f.m_file_hashes; m_file_hashes = f.m_file_hashes;
m_symlinks = f.m_symlinks; m_symlinks = f.m_symlinks;
m_mtime = f.m_mtime; m_mtime = f.m_mtime;
#ifndef TORRENT_NO_DEPRECATE
m_file_base = f.m_file_base; m_file_base = f.m_file_base;
#endif
m_paths = f.m_paths; m_paths = f.m_paths;
m_name = f.m_name; m_name = f.m_name;
m_total_size = f.m_total_size; m_total_size = f.m_total_size;
m_num_pieces = f.m_num_pieces; m_num_files = f.m_num_files;
m_piece_length = f.m_piece_length;
return *this; return *this;
} }

View File

@ -45,6 +45,11 @@ POSSIBILITY OF SUCH DAMAGE.
using namespace libtorrent; using namespace libtorrent;
namespace lt = libtorrent; namespace lt = libtorrent;
void print_session_log(lt::session& ses)
{
print_alerts(ses, "ses", true, true);
}
int read_message(stream_socket& s, char* buffer) int read_message(stream_socket& s, char* buffer)
{ {
using namespace libtorrent::detail; using namespace libtorrent::detail;
@ -211,8 +216,8 @@ void do_handshake(stream_socket& s, sha1_hash const& ih, char* buffer)
, libtorrent::asio::transfer_all(), ec); , libtorrent::asio::transfer_all(), ec);
if (ec) if (ec)
{ {
std::cout << aux::time_now_string() << ": " << ec.message() << std::endl; TEST_ERROR(ec.message());
exit(1); return;
} }
std::cout << aux::time_now_string() << " <== handshake" << std::endl; std::cout << aux::time_now_string() << " <== handshake" << std::endl;
@ -236,11 +241,14 @@ void do_handshake(stream_socket& s, sha1_hash const& ih, char* buffer)
TEST_CHECK(std::memcmp(buffer + 28, ih.begin(), 20) == 0); TEST_CHECK(std::memcmp(buffer + 28, ih.begin(), 20) == 0);
} }
boost::shared_ptr<torrent_info> setup_peer(stream_socket& s, sha1_hash& ih, boost::shared_ptr<lt::session>& ses) boost::shared_ptr<torrent_info> setup_peer(stream_socket& s, sha1_hash& ih
, boost::shared_ptr<lt::session>& ses, torrent_handle* th = NULL)
{ {
boost::shared_ptr<torrent_info> t = ::create_torrent(); boost::shared_ptr<torrent_info> t = ::create_torrent();
ih = t->info_hash(); ih = t->info_hash();
ses.reset(new lt::session(fingerprint("LT", 0, 1, 0, 0), std::make_pair(48900, 49000), "0.0.0.0", 0)); ses.reset(new lt::session(fingerprint("LT", 0, 1, 0, 0)
, std::make_pair(48900, 49000), "0.0.0.0", 0, alert::all_categories));
error_code ec; error_code ec;
add_torrent_params p; add_torrent_params p;
p.flags &= ~add_torrent_params::flag_paused; p.flags &= ~add_torrent_params::flag_paused;
@ -251,11 +259,16 @@ boost::shared_ptr<torrent_info> setup_peer(stream_socket& s, sha1_hash& ih, boos
remove("./tmp1_fast/temporary", ec); remove("./tmp1_fast/temporary", ec);
if (ec) fprintf(stderr, "remove(): %s\n", ec.message().c_str()); if (ec) fprintf(stderr, "remove(): %s\n", ec.message().c_str());
ec.clear(); ec.clear();
ses->add_torrent(p, ec); torrent_handle ret = ses->add_torrent(p, ec);
if (th) *th = ret;
test_sleep(300); // wait for the torrent to be ready
wait_for_downloading(*ses, "ses");
s.connect(tcp::endpoint(address::from_string("127.0.0.1", ec), ses->listen_port()), ec); s.connect(tcp::endpoint(address::from_string("127.0.0.1", ec), ses->listen_port()), ec);
print_session_log(*ses);
return t; return t;
} }
@ -273,7 +286,9 @@ void test_reject_fast()
char recv_buffer[1000]; char recv_buffer[1000];
do_handshake(s, ih, recv_buffer); do_handshake(s, ih, recv_buffer);
print_session_log(*ses);
send_have_all(s); send_have_all(s);
print_session_log(*ses);
std::vector<int> allowed_fast; std::vector<int> allowed_fast;
allowed_fast.push_back(0); allowed_fast.push_back(0);
@ -283,9 +298,11 @@ void test_reject_fast()
std::for_each(allowed_fast.begin(), allowed_fast.end() std::for_each(allowed_fast.begin(), allowed_fast.end()
, boost::bind(&send_allow_fast, boost::ref(s), _1)); , boost::bind(&send_allow_fast, boost::ref(s), _1));
print_session_log(*ses);
while (!allowed_fast.empty()) while (!allowed_fast.empty())
{ {
print_session_log(*ses);
int len = read_message(s, recv_buffer); int len = read_message(s, recv_buffer);
print_message(recv_buffer, len); print_message(recv_buffer, len);
int msg = recv_buffer[0]; int msg = recv_buffer[0];
@ -309,8 +326,10 @@ void test_reject_fast()
libtorrent::asio::write(s, libtorrent::asio::buffer(recv_buffer, 13) libtorrent::asio::write(s, libtorrent::asio::buffer(recv_buffer, 13)
, libtorrent::asio::transfer_all(), ec); , libtorrent::asio::transfer_all(), ec);
} }
print_session_log(*ses);
s.close(); s.close();
test_sleep(500); test_sleep(500);
print_session_log(*ses);
} }
void test_respect_suggest() void test_respect_suggest()
@ -325,7 +344,9 @@ void test_respect_suggest()
char recv_buffer[1000]; char recv_buffer[1000];
do_handshake(s, ih, recv_buffer); do_handshake(s, ih, recv_buffer);
print_session_log(*ses);
send_have_all(s); send_have_all(s);
print_session_log(*ses);
std::vector<int> suggested; std::vector<int> suggested;
suggested.push_back(0); suggested.push_back(0);
@ -335,14 +356,18 @@ void test_respect_suggest()
std::for_each(suggested.begin(), suggested.end() std::for_each(suggested.begin(), suggested.end()
, boost::bind(&send_suggest_piece, boost::ref(s), _1)); , boost::bind(&send_suggest_piece, boost::ref(s), _1));
print_session_log(*ses);
send_unchoke(s); send_unchoke(s);
print_session_log(*ses);
send_keepalive(s); send_keepalive(s);
print_session_log(*ses);
int fail_counter = 100; int fail_counter = 100;
while (!suggested.empty() && fail_counter > 0) while (!suggested.empty() && fail_counter > 0)
{ {
print_session_log(*ses);
int len = read_message(s, recv_buffer); int len = read_message(s, recv_buffer);
print_message(recv_buffer, len); print_message(recv_buffer, len);
int msg = recv_buffer[0]; int msg = recv_buffer[0];
@ -367,10 +392,12 @@ void test_respect_suggest()
libtorrent::asio::write(s, libtorrent::asio::buffer(recv_buffer, 13) libtorrent::asio::write(s, libtorrent::asio::buffer(recv_buffer, 13)
, libtorrent::asio::transfer_all(), ec); , libtorrent::asio::transfer_all(), ec);
} }
print_session_log(*ses);
TEST_CHECK(fail_counter > 0); TEST_CHECK(fail_counter > 0);
s.close(); s.close();
test_sleep(500); test_sleep(500);
print_session_log(*ses);
} }
void test_multiple_bitfields() void test_multiple_bitfields()
@ -382,22 +409,29 @@ void test_multiple_bitfields()
io_service ios; io_service ios;
stream_socket s(ios); stream_socket s(ios);
boost::shared_ptr<torrent_info> ti = setup_peer(s, ih, ses); boost::shared_ptr<torrent_info> ti = setup_peer(s, ih, ses);
print_session_log(*ses);
char recv_buffer[1000]; char recv_buffer[1000];
do_handshake(s, ih, recv_buffer); do_handshake(s, ih, recv_buffer);
print_session_log(*ses);
std::string bitfield; std::string bitfield;
bitfield.resize(ti->num_pieces(), '0'); bitfield.resize(ti->num_pieces(), '0');
send_bitfield(s, bitfield.c_str()); send_bitfield(s, bitfield.c_str());
print_session_log(*ses);
bitfield[0] = '1'; bitfield[0] = '1';
send_bitfield(s, bitfield.c_str()); send_bitfield(s, bitfield.c_str());
print_session_log(*ses);
bitfield[1] = '1'; bitfield[1] = '1';
send_bitfield(s, bitfield.c_str()); send_bitfield(s, bitfield.c_str());
print_session_log(*ses);
bitfield[2] = '1'; bitfield[2] = '1';
send_bitfield(s, bitfield.c_str()); send_bitfield(s, bitfield.c_str());
print_session_log(*ses);
s.close(); s.close();
test_sleep(500); test_sleep(500);
print_session_log(*ses);
} }
void test_multiple_have_all() void test_multiple_have_all()
@ -413,13 +447,21 @@ void test_multiple_have_all()
char recv_buffer[1000]; char recv_buffer[1000];
do_handshake(s, ih, recv_buffer); do_handshake(s, ih, recv_buffer);
print_session_log(*ses);
send_have_all(s); send_have_all(s);
print_session_log(*ses);
send_have_all(s); send_have_all(s);
print_session_log(*ses);
send_have_none(s); send_have_none(s);
print_session_log(*ses);
send_have_all(s); send_have_all(s);
print_session_log(*ses);
s.close(); s.close();
print_session_log(*ses);
test_sleep(500); test_sleep(500);
print_session_log(*ses);
} }
// makes sure that pieces that are lost are not requested // makes sure that pieces that are lost are not requested
@ -429,32 +471,21 @@ void test_dont_have()
std::cerr << " === test dont_have ===" << std::endl; std::cerr << " === test dont_have ===" << std::endl;
boost::shared_ptr<torrent_info> t = ::create_torrent(); sha1_hash ih;
sha1_hash ih = t->info_hash(); torrent_handle th;
lt::session ses1(fingerprint("LT", 0, 1, 0, 0), std::make_pair(48950, 49050), "0.0.0.0", 0); boost::shared_ptr<lt::session> ses;
error_code ec;
add_torrent_params p;
p.flags &= ~add_torrent_params::flag_paused;
p.flags &= ~add_torrent_params::flag_auto_managed;
p.ti = t;
p.save_path = "./tmp1_dont_have";
remove("./tmp1_dont_have/temporary", ec);
if (ec) fprintf(stderr, "remove(): %s\n", ec.message().c_str());
ec.clear();
torrent_handle th = ses1.add_torrent(p, ec);
test_sleep(300);
io_service ios; io_service ios;
stream_socket s(ios); stream_socket s(ios);
s.connect(tcp::endpoint(address::from_string("127.0.0.1", ec), ses1.listen_port()), ec); boost::shared_ptr<torrent_info> ti = setup_peer(s, ih, ses, &th);
char recv_buffer[1000]; char recv_buffer[1000];
do_handshake(s, ih, recv_buffer); do_handshake(s, ih, recv_buffer);
print_session_log(*ses);
send_have_all(s); send_have_all(s);
print_session_log(*ses);
test_sleep(300); test_sleep(300);
print_session_log(*ses);
std::vector<peer_info> pi; std::vector<peer_info> pi;
th.get_peer_info(pi); th.get_peer_info(pi);
@ -466,8 +497,11 @@ void test_dont_have()
TEST_CHECK(pi[0].flags & peer_info::seed); TEST_CHECK(pi[0].flags & peer_info::seed);
int lt_dont_have = 0; int lt_dont_have = 0;
error_code ec;
while (lt_dont_have == 0) while (lt_dont_have == 0)
{ {
print_session_log(*ses);
int len = read_message(s, recv_buffer); int len = read_message(s, recv_buffer);
print_message(recv_buffer, len); print_message(recv_buffer, len);
if (len == 0) continue; if (len == 0) continue;
@ -477,7 +511,6 @@ void test_dont_have()
if (ext_msg != 0) continue; if (ext_msg != 0) continue;
bdecode_node e; bdecode_node e;
error_code ec;
int pos = 0; int pos = 0;
int ret = bdecode(recv_buffer + 2, recv_buffer + len, e, ec, &pos); int ret = bdecode(recv_buffer + 2, recv_buffer + len, e, ec, &pos);
if (ret != 0) if (ret != 0)
@ -497,6 +530,7 @@ void test_dont_have()
lt_dont_have = dont_have.int_value(); lt_dont_have = dont_have.int_value();
} }
print_session_log(*ses);
char* ptr = recv_buffer; char* ptr = recv_buffer;
write_uint32(6, ptr); write_uint32(6, ptr);
@ -507,8 +541,12 @@ void test_dont_have()
libtorrent::asio::write(s, libtorrent::asio::buffer(recv_buffer, 10) libtorrent::asio::write(s, libtorrent::asio::buffer(recv_buffer, 10)
, libtorrent::asio::transfer_all(), ec); , libtorrent::asio::transfer_all(), ec);
print_session_log(*ses);
test_sleep(1000); test_sleep(1000);
print_session_log(*ses);
th.get_peer_info(pi); th.get_peer_info(pi);
TEST_EQUAL(pi.size(), 1); TEST_EQUAL(pi.size(), 1);
@ -520,6 +558,8 @@ void test_dont_have()
TEST_EQUAL(pi[0].pieces[2], true); TEST_EQUAL(pi[0].pieces[2], true);
TEST_EQUAL(pi[0].pieces[1], true); TEST_EQUAL(pi[0].pieces[1], true);
TEST_EQUAL(pi[0].pieces[0], true); TEST_EQUAL(pi[0].pieces[0], true);
print_session_log(*ses);
} }
int test_main() int test_main()

View File

@ -81,7 +81,6 @@ void test_rules_invariant(std::vector<ip_range<T> > const& r, ip_filter const& f
} }
iterator i = r.begin(); iterator i = r.begin();
iterator j = boost::next(i);
for (iterator i(r.begin()), j(boost::next(r.begin())) for (iterator i(r.begin()), j(boost::next(r.begin()))
, end(r.end()); j != end; ++j, ++i) , end(r.end()); j != end; ++j, ++i)
{ {

View File

@ -533,7 +533,7 @@ int test_main()
// this will make the peer_list recalculate the connect candidates // this will make the peer_list recalculate the connect candidates
std::vector<torrent_peer*> peers; std::vector<torrent_peer*> peers;
torrent_peer* peer = p.connect_one_peer(1, &st); p.connect_one_peer(1, &st);
TEST_EQUAL(p.num_connect_candidates(), 50); TEST_EQUAL(p.num_connect_candidates(), 50);
TEST_EQUAL(p.num_peers(), 100); TEST_EQUAL(p.num_peers(), 100);