wornvirtue
/
premiere-libtorrent
forked from premiere/premiere-libtorrent
1
0
Fork 0
Code Pull Requests Activity

improve connection tester a bit, to deal with being unchoked

This commit is contained in:
Arvid Norberg 2013-04-12 23:18:49 +00:00
parent 3d19ea4b18
commit 6e0ae9e959
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/connection_tester.cpp
View File

@ -96,6 +96,8 @@ struct peer_conn
, info_hash(ih) , info_hash(ih)
, outstanding_requests(0) , outstanding_requests(0)
, seed(seed_) , seed(seed_)
, fast_extension(false)
, choked(true)
, blocks_received(0) , blocks_received(0)
, blocks_sent(0) , blocks_sent(0)
, num_pieces(num_pieces) , num_pieces(num_pieces)
@ -150,6 +152,7 @@ struct peer_conn
// if this is true, this connection is a seed // if this is true, this connection is a seed
bool seed; bool seed;
bool fast_extension; bool fast_extension;
bool choked;
int blocks_received; int blocks_received;
int blocks_sent; int blocks_sent;
int num_pieces; int num_pieces;
@ -263,6 +266,7 @@ struct peer_conn
bool write_request() bool write_request()
{ {
if (choked) return false;
if (pieces.empty() && suggested_pieces.empty() && current_piece == -1) return false; if (pieces.empty() && suggested_pieces.empty() && current_piece == -1) return false;
if (current_piece == -1) if (current_piece == -1)
@ -483,6 +487,14 @@ struct peer_conn
++num_suggest; ++num_suggest;
} }
} }
else if (msg == 1) // unchoke
{
choked = false;
}
else if (msg == 0) // choke
{
choked = true;
}
work_download(); work_download();
} }
} }