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

View File

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