2003-10-23 01:00:57 +02:00
|
|
|
/*
|
|
|
|
|
|
|
|
Copyright (c) 2003, 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.
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
2007-03-17 18:15:16 +01:00
|
|
|
#include "libtorrent/pch.hpp"
|
|
|
|
|
2004-11-30 12:17:32 +01:00
|
|
|
#include <vector>
|
2004-01-12 21:31:27 +01:00
|
|
|
#include <limits>
|
2006-04-25 23:04:48 +02:00
|
|
|
#include <boost/bind.hpp>
|
2004-01-12 21:31:27 +01:00
|
|
|
|
2003-10-23 01:00:57 +02:00
|
|
|
#include "libtorrent/peer_connection.hpp"
|
2003-12-22 08:14:35 +01:00
|
|
|
#include "libtorrent/identify_client.hpp"
|
2004-01-04 05:29:13 +01:00
|
|
|
#include "libtorrent/entry.hpp"
|
|
|
|
#include "libtorrent/bencode.hpp"
|
2004-01-18 20:12:18 +01:00
|
|
|
#include "libtorrent/alert_types.hpp"
|
2004-01-25 19:18:36 +01:00
|
|
|
#include "libtorrent/invariant_check.hpp"
|
2004-01-26 18:39:44 +01:00
|
|
|
#include "libtorrent/io.hpp"
|
2006-07-08 21:41:39 +02:00
|
|
|
#include "libtorrent/file.hpp"
|
2004-04-17 14:29:35 +02:00
|
|
|
#include "libtorrent/version.hpp"
|
2006-11-14 01:08:16 +01:00
|
|
|
#include "libtorrent/extensions.hpp"
|
2006-10-11 16:02:21 +02:00
|
|
|
#include "libtorrent/aux_/session_impl.hpp"
|
2007-04-10 23:23:13 +02:00
|
|
|
#include "libtorrent/policy.hpp"
|
2007-04-25 20:26:35 +02:00
|
|
|
#include "libtorrent/socket_type.hpp"
|
2007-09-01 06:08:39 +02:00
|
|
|
#include "libtorrent/assert.hpp"
|
2008-12-26 08:00:21 +01:00
|
|
|
#include "libtorrent/broadcast_socket.hpp"
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-06-23 17:37:24 +02:00
|
|
|
//#define TORRENT_CORRUPT_DATA
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
using boost::bind;
|
|
|
|
using boost::shared_ptr;
|
2006-10-11 16:02:21 +02:00
|
|
|
using libtorrent::aux::session_impl;
|
2004-10-14 03:17:04 +02:00
|
|
|
|
2004-01-04 05:29:13 +01:00
|
|
|
namespace libtorrent
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2007-05-10 03:50:11 +02:00
|
|
|
// outbound connection
|
2004-01-04 05:29:13 +01:00
|
|
|
peer_connection::peer_connection(
|
2006-10-11 16:02:21 +02:00
|
|
|
session_impl& ses
|
2006-04-25 23:04:48 +02:00
|
|
|
, boost::weak_ptr<torrent> tor
|
2007-04-23 23:36:21 +02:00
|
|
|
, shared_ptr<socket_type> s
|
2008-05-12 12:10:39 +02:00
|
|
|
, tcp::endpoint const& endp
|
2007-04-10 23:23:13 +02:00
|
|
|
, policy::peer* peerinfo)
|
2005-08-19 01:55:32 +02:00
|
|
|
:
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_choke(time_now() - hours(1))
|
2006-04-25 23:04:48 +02:00
|
|
|
,
|
2005-08-19 01:55:32 +02:00
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
m_ses(ses)
|
2006-10-11 16:02:21 +02:00
|
|
|
, m_max_out_request_queue(m_ses.settings().max_out_request_queue)
|
2009-03-31 10:02:25 +02:00
|
|
|
, m_work(ses.m_io_service)
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_last_piece(time_now())
|
2007-05-25 21:42:10 +02:00
|
|
|
, m_last_request(time_now())
|
2007-09-01 09:38:10 +02:00
|
|
|
, m_last_incoming_request(min_time())
|
2008-12-11 00:07:44 +01:00
|
|
|
, m_last_unchoke(time_now())
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_last_receive(time_now())
|
|
|
|
, m_last_sent(time_now())
|
2008-06-29 11:50:42 +02:00
|
|
|
, m_requested(min_time())
|
2008-07-07 14:04:06 +02:00
|
|
|
, m_timeout_extend(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_remote_dl_update(time_now())
|
2008-07-10 12:58:30 +02:00
|
|
|
, m_connect(time_now())
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_became_uninterested(time_now())
|
|
|
|
, m_became_uninteresting(time_now())
|
|
|
|
, m_free_upload(0)
|
|
|
|
, m_downloaded_at_last_unchoke(0)
|
2008-12-11 00:07:44 +01:00
|
|
|
, m_uploaded_at_last_unchoke(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_disk_recv_buffer(ses, 0)
|
2004-01-04 05:29:13 +01:00
|
|
|
, m_socket(s)
|
2008-05-12 12:10:39 +02:00
|
|
|
, m_remote(endp)
|
2006-04-25 23:04:48 +02:00
|
|
|
, m_torrent(tor)
|
2009-03-17 10:34:44 +01:00
|
|
|
, m_outstanding_bytes(0)
|
|
|
|
, m_queued_time_critical(0)
|
2004-01-28 12:37:46 +01:00
|
|
|
, m_num_pieces(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_timeout(m_ses.settings().peer_timeout)
|
|
|
|
, m_packet_size(0)
|
2009-03-13 07:09:39 +01:00
|
|
|
, m_soft_packet_size(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_recv_pos(0)
|
|
|
|
, m_disk_recv_buffer_size(0)
|
|
|
|
, m_reading_bytes(0)
|
2004-01-12 21:31:27 +01:00
|
|
|
, m_num_invalid_requests(0)
|
2008-01-17 18:40:46 +01:00
|
|
|
, m_priority(1)
|
2009-04-26 02:21:59 +02:00
|
|
|
, m_upload_limit(0)
|
|
|
|
, m_download_limit(0)
|
2007-04-10 23:23:13 +02:00
|
|
|
, m_peer_info(peerinfo)
|
2007-04-27 02:27:37 +02:00
|
|
|
, m_speed(slow)
|
2007-05-05 02:29:33 +02:00
|
|
|
, m_connection_ticket(-1)
|
2008-12-08 07:36:22 +01:00
|
|
|
, m_superseed_piece(-1)
|
2007-06-07 02:05:18 +02:00
|
|
|
, m_remote_bytes_dled(0)
|
|
|
|
, m_remote_dl_rate(0)
|
2007-08-01 07:22:34 +02:00
|
|
|
, m_outstanding_writing_bytes(0)
|
2008-04-03 08:11:21 +02:00
|
|
|
, m_download_rate_peak(0)
|
|
|
|
, m_upload_rate_peak(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_rtt(0)
|
|
|
|
, m_prefer_whole_pieces(0)
|
|
|
|
, m_desired_queue_size(2)
|
2009-03-12 18:06:41 +01:00
|
|
|
, m_choke_rejects(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_fast_reconnect(false)
|
|
|
|
, m_active(true)
|
|
|
|
, m_peer_interested(false)
|
|
|
|
, m_peer_choked(true)
|
|
|
|
, m_interesting(false)
|
|
|
|
, m_choked(true)
|
|
|
|
, m_failed(false)
|
|
|
|
, m_ignore_bandwidth_limits(false)
|
2009-01-28 07:14:21 +01:00
|
|
|
, m_ignore_unchoke_slots(false)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_have_all(false)
|
|
|
|
, m_disconnecting(false)
|
|
|
|
, m_connecting(true)
|
|
|
|
, m_queued(true)
|
|
|
|
, m_request_large_blocks(false)
|
|
|
|
, m_upload_only(false)
|
2008-06-29 11:50:42 +02:00
|
|
|
, m_snubbed(false)
|
2008-07-18 12:03:42 +02:00
|
|
|
, m_bitfield_received(false)
|
2008-11-06 09:34:56 +01:00
|
|
|
, m_no_download(false)
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-04-25 23:04:48 +02:00
|
|
|
, m_in_constructor(true)
|
2008-07-18 12:03:42 +02:00
|
|
|
, m_disconnect_started(false)
|
2008-08-29 19:21:56 +02:00
|
|
|
, m_initialized(false)
|
2009-03-17 10:34:44 +01:00
|
|
|
, m_received_in_piece(0)
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
2008-01-14 00:46:43 +01:00
|
|
|
m_channel_state[upload_channel] = peer_info::bw_idle;
|
|
|
|
m_channel_state[download_channel] = peer_info::bw_idle;
|
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
m_quota[0] = 0;
|
|
|
|
m_quota[1] = 0;
|
|
|
|
|
2007-12-21 00:53:03 +01:00
|
|
|
TORRENT_ASSERT(peerinfo == 0 || peerinfo->banned == false);
|
2007-05-02 21:47:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
2007-01-29 08:39:33 +01:00
|
|
|
std::fill(m_country, m_country + 2, 0);
|
2008-04-11 10:46:43 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
if (m_ses.has_country_db())
|
|
|
|
{
|
|
|
|
char const *country = m_ses.country_for_ip(m_remote.address());
|
|
|
|
if (country != 0)
|
|
|
|
{
|
|
|
|
m_country[0] = country[0];
|
|
|
|
m_country[1] = country[1];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2007-05-02 21:47:38 +02:00
|
|
|
#endif
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-10-23 06:10:23 +02:00
|
|
|
error_code ec;
|
|
|
|
m_logger = m_ses.create_log(m_remote.address().to_string(ec) + "_"
|
2009-01-27 07:17:55 +01:00
|
|
|
+ to_string(m_remote.port()).elems, m_ses.listen_port());
|
2006-04-25 23:04:48 +02:00
|
|
|
(*m_logger) << "*** OUTGOING CONNECTION\n";
|
|
|
|
#endif
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2007-12-18 07:04:54 +01:00
|
|
|
piece_failed = false;
|
2007-12-14 19:02:06 +01:00
|
|
|
#endif
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
m_inet_as_name = m_ses.as_name_for_ip(m_remote.address());
|
|
|
|
#endif
|
2004-01-25 19:18:36 +01:00
|
|
|
|
2004-01-15 17:45:34 +01:00
|
|
|
std::fill(m_peer_id.begin(), m_peer_id.end(), 0);
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2007-05-10 03:50:11 +02:00
|
|
|
// incoming connection
|
2004-01-04 05:29:13 +01:00
|
|
|
peer_connection::peer_connection(
|
2006-10-11 16:02:21 +02:00
|
|
|
session_impl& ses
|
2008-06-29 11:50:42 +02:00
|
|
|
, shared_ptr<socket_type> s
|
2008-05-12 12:10:39 +02:00
|
|
|
, tcp::endpoint const& endp
|
2007-04-10 23:23:13 +02:00
|
|
|
, policy::peer* peerinfo)
|
2005-08-19 01:55:32 +02:00
|
|
|
:
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_choke(time_now() - hours(1))
|
2006-04-25 23:04:48 +02:00
|
|
|
,
|
2005-08-19 01:55:32 +02:00
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
m_ses(ses)
|
2006-10-11 16:02:21 +02:00
|
|
|
, m_max_out_request_queue(m_ses.settings().max_out_request_queue)
|
2009-03-31 10:02:25 +02:00
|
|
|
, m_work(ses.m_io_service)
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_last_piece(time_now())
|
2007-05-25 21:42:10 +02:00
|
|
|
, m_last_request(time_now())
|
2007-09-01 09:38:10 +02:00
|
|
|
, m_last_incoming_request(min_time())
|
2008-12-11 00:07:44 +01:00
|
|
|
, m_last_unchoke(time_now())
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_last_receive(time_now())
|
|
|
|
, m_last_sent(time_now())
|
2008-06-29 11:50:42 +02:00
|
|
|
, m_requested(min_time())
|
2008-07-07 14:04:06 +02:00
|
|
|
, m_timeout_extend(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_remote_dl_update(time_now())
|
2008-07-10 12:58:30 +02:00
|
|
|
, m_connect(time_now())
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_became_uninterested(time_now())
|
|
|
|
, m_became_uninteresting(time_now())
|
|
|
|
, m_free_upload(0)
|
|
|
|
, m_downloaded_at_last_unchoke(0)
|
2008-12-11 00:07:44 +01:00
|
|
|
, m_uploaded_at_last_unchoke(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_disk_recv_buffer(ses, 0)
|
2004-01-04 05:29:13 +01:00
|
|
|
, m_socket(s)
|
2008-05-12 12:10:39 +02:00
|
|
|
, m_remote(endp)
|
2009-03-17 10:34:44 +01:00
|
|
|
, m_outstanding_bytes(0)
|
|
|
|
, m_queued_time_critical(0)
|
2004-01-28 12:37:46 +01:00
|
|
|
, m_num_pieces(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_timeout(m_ses.settings().peer_timeout)
|
|
|
|
, m_packet_size(0)
|
2009-03-13 07:09:39 +01:00
|
|
|
, m_soft_packet_size(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_recv_pos(0)
|
|
|
|
, m_disk_recv_buffer_size(0)
|
|
|
|
, m_reading_bytes(0)
|
2004-01-12 21:31:27 +01:00
|
|
|
, m_num_invalid_requests(0)
|
2008-01-17 18:40:46 +01:00
|
|
|
, m_priority(1)
|
2009-04-26 02:21:59 +02:00
|
|
|
, m_upload_limit(0)
|
|
|
|
, m_download_limit(0)
|
2007-04-10 23:23:13 +02:00
|
|
|
, m_peer_info(peerinfo)
|
2007-04-27 02:27:37 +02:00
|
|
|
, m_speed(slow)
|
2007-08-25 16:52:48 +02:00
|
|
|
, m_connection_ticket(-1)
|
2008-12-08 07:36:22 +01:00
|
|
|
, m_superseed_piece(-1)
|
2007-06-07 02:05:18 +02:00
|
|
|
, m_remote_bytes_dled(0)
|
|
|
|
, m_remote_dl_rate(0)
|
2007-08-01 07:22:34 +02:00
|
|
|
, m_outstanding_writing_bytes(0)
|
2008-04-03 08:11:21 +02:00
|
|
|
, m_download_rate_peak(0)
|
|
|
|
, m_upload_rate_peak(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_rtt(0)
|
|
|
|
, m_prefer_whole_pieces(0)
|
|
|
|
, m_desired_queue_size(2)
|
2009-03-12 18:06:41 +01:00
|
|
|
, m_choke_rejects(0)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_fast_reconnect(false)
|
|
|
|
, m_active(false)
|
|
|
|
, m_peer_interested(false)
|
|
|
|
, m_peer_choked(true)
|
|
|
|
, m_interesting(false)
|
|
|
|
, m_choked(true)
|
|
|
|
, m_failed(false)
|
|
|
|
, m_ignore_bandwidth_limits(false)
|
2009-01-28 07:14:21 +01:00
|
|
|
, m_ignore_unchoke_slots(false)
|
2008-05-19 04:52:32 +02:00
|
|
|
, m_have_all(false)
|
|
|
|
, m_disconnecting(false)
|
|
|
|
, m_connecting(false)
|
|
|
|
, m_queued(false)
|
|
|
|
, m_request_large_blocks(false)
|
|
|
|
, m_upload_only(false)
|
2008-06-29 11:50:42 +02:00
|
|
|
, m_snubbed(false)
|
2008-07-18 12:03:42 +02:00
|
|
|
, m_bitfield_received(false)
|
2008-11-06 09:34:56 +01:00
|
|
|
, m_no_download(false)
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-04-25 23:04:48 +02:00
|
|
|
, m_in_constructor(true)
|
2008-07-18 12:03:42 +02:00
|
|
|
, m_disconnect_started(false)
|
2008-08-29 19:21:56 +02:00
|
|
|
, m_initialized(false)
|
2009-03-17 10:34:44 +01:00
|
|
|
, m_received_in_piece(0)
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
2008-01-14 00:46:43 +01:00
|
|
|
m_channel_state[upload_channel] = peer_info::bw_idle;
|
|
|
|
m_channel_state[download_channel] = peer_info::bw_idle;
|
2008-04-09 06:09:40 +02:00
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
m_quota[0] = 0;
|
|
|
|
m_quota[1] = 0;
|
|
|
|
|
2007-05-02 21:47:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
2007-01-29 08:39:33 +01:00
|
|
|
std::fill(m_country, m_country + 2, 0);
|
2008-04-11 10:46:43 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
if (m_ses.has_country_db())
|
|
|
|
{
|
|
|
|
char const *country = m_ses.country_for_ip(m_remote.address());
|
|
|
|
if (country != 0)
|
|
|
|
{
|
|
|
|
m_country[0] = country[0];
|
|
|
|
m_country[1] = country[1];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2007-05-02 21:47:38 +02:00
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-05-03 18:05:42 +02:00
|
|
|
error_code ec;
|
2008-05-12 12:10:39 +02:00
|
|
|
TORRENT_ASSERT(m_socket->remote_endpoint(ec) == m_remote || ec);
|
2008-01-07 02:10:46 +01:00
|
|
|
m_logger = m_ses.create_log(remote().address().to_string(ec) + "_"
|
2009-01-27 07:17:55 +01:00
|
|
|
+ to_string(remote().port()).elems, m_ses.listen_port());
|
2006-04-25 23:04:48 +02:00
|
|
|
(*m_logger) << "*** INCOMING CONNECTION\n";
|
|
|
|
#endif
|
2007-01-10 16:02:25 +01:00
|
|
|
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
m_inet_as_name = m_ses.as_name_for_ip(m_remote.address());
|
|
|
|
#endif
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2007-12-18 07:04:54 +01:00
|
|
|
piece_failed = false;
|
2007-12-14 19:02:06 +01:00
|
|
|
#endif
|
2004-01-15 17:45:34 +01:00
|
|
|
std::fill(m_peer_id.begin(), m_peer_id.end(), 0);
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-03-10 04:30:01 +01:00
|
|
|
bool peer_connection::unchoke_compare(boost::intrusive_ptr<peer_connection const> const& p) const
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(p);
|
|
|
|
peer_connection const& rhs = *p;
|
|
|
|
|
2008-10-15 04:18:59 +02:00
|
|
|
size_type c1;
|
|
|
|
size_type c2;
|
|
|
|
|
2008-10-15 06:52:07 +02:00
|
|
|
// first compare how many bytes they've sent us
|
|
|
|
c1 = m_statistics.total_payload_download() - m_downloaded_at_last_unchoke;
|
|
|
|
c2 = rhs.m_statistics.total_payload_download() - rhs.m_downloaded_at_last_unchoke;
|
|
|
|
if (c1 > c2) return true;
|
|
|
|
if (c1 < c2) return false;
|
2008-03-10 04:30:01 +01:00
|
|
|
|
|
|
|
// if they are equal, compare how much we have uploaded
|
2008-12-11 00:07:44 +01:00
|
|
|
c1 = m_statistics.total_payload_upload() - m_uploaded_at_last_unchoke;
|
|
|
|
c2 = rhs.m_statistics.total_payload_upload() - rhs.m_uploaded_at_last_unchoke;
|
2008-03-10 04:30:01 +01:00
|
|
|
|
2008-10-14 02:57:58 +02:00
|
|
|
// in order to not switch back and forth too often,
|
|
|
|
// unchoked peers must be at least one piece ahead
|
|
|
|
// of a choked peer to be sorted at a lower unchoke-priority
|
2008-10-15 06:52:07 +02:00
|
|
|
boost::shared_ptr<torrent> t1 = m_torrent.lock();
|
|
|
|
TORRENT_ASSERT(t1);
|
2008-10-15 04:18:59 +02:00
|
|
|
boost::shared_ptr<torrent> t2 = rhs.associated_torrent().lock();
|
|
|
|
TORRENT_ASSERT(t2);
|
2008-12-11 00:07:44 +01:00
|
|
|
int pieces = m_ses.settings().seeding_piece_quota;
|
|
|
|
bool c1_done = is_choked() || c1 > (std::max)(t1->torrent_file().piece_length() * pieces, 256 * 1024);
|
|
|
|
bool c2_done = rhs.is_choked() || c2 > (std::max)(t2->torrent_file().piece_length() * pieces, 256 * 1024);
|
|
|
|
|
|
|
|
if (!c1_done && c2_done) return true;
|
|
|
|
if (c1_done && !c2_done) return false;
|
2008-10-14 02:57:58 +02:00
|
|
|
|
2008-12-11 00:07:44 +01:00
|
|
|
// if both peers have are still in their send quota or not in their send quota
|
|
|
|
// prioritize the one that has waited the longest to be unchoked
|
|
|
|
return m_last_unchoke < rhs.m_last_unchoke;
|
2008-03-10 04:30:01 +01:00
|
|
|
}
|
|
|
|
|
2009-04-04 09:55:34 +02:00
|
|
|
bool peer_connection::upload_rate_compare(peer_connection const* p) const
|
|
|
|
{
|
|
|
|
size_type c1;
|
|
|
|
size_type c2;
|
|
|
|
|
|
|
|
c1 = m_statistics.total_payload_upload() - m_uploaded_at_last_unchoke;
|
|
|
|
c2 = p->m_statistics.total_payload_upload() - p->m_uploaded_at_last_unchoke;
|
|
|
|
|
|
|
|
return c1 > c2;
|
|
|
|
}
|
|
|
|
|
2008-03-10 04:30:01 +01:00
|
|
|
void peer_connection::reset_choke_counters()
|
|
|
|
{
|
|
|
|
m_downloaded_at_last_unchoke = m_statistics.total_payload_download();
|
|
|
|
}
|
|
|
|
|
2008-03-31 06:46:24 +02:00
|
|
|
void peer_connection::start()
|
|
|
|
{
|
2008-05-15 04:29:26 +02:00
|
|
|
TORRENT_ASSERT(m_peer_info == 0 || m_peer_info->connection == this);
|
2008-03-31 06:46:24 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
|
2008-04-09 06:09:40 +02:00
|
|
|
if (!t)
|
|
|
|
{
|
|
|
|
tcp::socket::non_blocking_io ioc(true);
|
2008-05-03 18:05:42 +02:00
|
|
|
error_code ec;
|
2008-04-09 06:09:40 +02:00
|
|
|
m_socket->io_control(ioc, ec);
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
m_remote = m_socket->remote_endpoint(ec);
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if (m_remote.address().is_v4())
|
|
|
|
m_socket->set_option(type_of_service(m_ses.settings().peer_tos), ec);
|
|
|
|
}
|
|
|
|
else if (t->ready_for_connections())
|
|
|
|
{
|
2008-03-31 06:46:24 +02:00
|
|
|
init();
|
2008-04-09 06:09:40 +02:00
|
|
|
}
|
2008-03-31 06:46:24 +02:00
|
|
|
}
|
|
|
|
|
2007-03-21 03:09:50 +01:00
|
|
|
void peer_connection::update_interest()
|
|
|
|
{
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-03-21 03:09:50 +01:00
|
|
|
|
2008-09-18 22:17:49 +02:00
|
|
|
// if m_have_piece is 0, it means the connections
|
|
|
|
// have not been initialized yet. The interested
|
|
|
|
// flag will be updated once they are.
|
|
|
|
if (m_have_piece.size() == 0) return;
|
2008-09-19 09:14:24 +02:00
|
|
|
if (!t->ready_for_connections()) return;
|
2008-09-18 22:17:49 +02:00
|
|
|
|
2007-03-21 03:09:50 +01:00
|
|
|
bool interested = false;
|
2008-06-07 04:58:28 +02:00
|
|
|
if (!t->is_finished())
|
2007-03-21 03:09:50 +01:00
|
|
|
{
|
2008-06-07 04:58:28 +02:00
|
|
|
piece_picker const& p = t->picker();
|
|
|
|
int num_pieces = p.num_pieces();
|
|
|
|
for (int j = 0; j != num_pieces; ++j)
|
2007-03-21 03:09:50 +01:00
|
|
|
{
|
2008-06-07 04:58:28 +02:00
|
|
|
if (!p.have_piece(j)
|
|
|
|
&& t->piece_priority(j) > 0
|
|
|
|
&& m_have_piece[j])
|
|
|
|
{
|
|
|
|
interested = true;
|
|
|
|
break;
|
|
|
|
}
|
2007-03-21 03:09:50 +01:00
|
|
|
}
|
|
|
|
}
|
2008-10-21 19:10:11 +02:00
|
|
|
if (!interested) send_not_interested();
|
|
|
|
else t->get_policy().peer_is_interesting(*this);
|
2007-03-21 03:09:50 +01:00
|
|
|
|
2008-09-19 04:37:05 +02:00
|
|
|
TORRENT_ASSERT(in_handshake() || is_interesting() == interested);
|
2007-03-21 03:09:50 +01:00
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
void peer_connection::add_extension(boost::shared_ptr<peer_plugin> ext)
|
|
|
|
{
|
|
|
|
m_extensions.push_back(ext);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
void peer_connection::send_allowed_set()
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-08-14 19:47:48 +02:00
|
|
|
|
2008-12-08 07:36:22 +01:00
|
|
|
if (t->super_seeding())
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " *** SKIPPING ALLOWED SET BECAUSE OF SUPER SEEDING\n";
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
int num_allowed_pieces = m_ses.settings().allowed_fast_set_size;
|
|
|
|
int num_pieces = t->torrent_file().num_pieces();
|
|
|
|
|
|
|
|
if (num_allowed_pieces >= num_pieces)
|
|
|
|
{
|
|
|
|
for (int i = 0; i < num_pieces; ++i)
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> ALLOWED_FAST [ " << i << " ]\n";
|
|
|
|
#endif
|
|
|
|
write_allow_fast(i);
|
|
|
|
m_accept_fast.insert(i);
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string x;
|
|
|
|
address const& addr = m_remote.address();
|
|
|
|
if (addr.is_v4())
|
|
|
|
{
|
|
|
|
address_v4::bytes_type bytes = addr.to_v4().to_bytes();
|
|
|
|
x.assign((char*)&bytes[0], bytes.size());
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
address_v6::bytes_type bytes = addr.to_v6().to_bytes();
|
|
|
|
x.assign((char*)&bytes[0], bytes.size());
|
|
|
|
}
|
|
|
|
x.append((char*)&t->torrent_file().info_hash()[0], 20);
|
|
|
|
|
|
|
|
sha1_hash hash = hasher(&x[0], x.size()).final();
|
|
|
|
for (;;)
|
|
|
|
{
|
|
|
|
char* p = (char*)&hash[0];
|
|
|
|
for (int i = 0; i < 5; ++i)
|
|
|
|
{
|
|
|
|
int piece = detail::read_uint32(p) % num_pieces;
|
|
|
|
if (m_accept_fast.find(piece) == m_accept_fast.end())
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> ALLOWED_FAST [ " << piece << " ]\n";
|
|
|
|
#endif
|
|
|
|
write_allow_fast(piece);
|
|
|
|
m_accept_fast.insert(piece);
|
|
|
|
if (int(m_accept_fast.size()) >= num_allowed_pieces
|
|
|
|
|| int(m_accept_fast.size()) == num_pieces) return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
hash = hasher((char*)&hash[0], 20).final();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
void peer_connection::on_metadata_impl()
|
|
|
|
{
|
|
|
|
boost::shared_ptr<torrent> t = associated_torrent().lock();
|
|
|
|
m_have_piece.resize(t->torrent_file().num_pieces(), m_have_all);
|
|
|
|
m_num_pieces = m_have_piece.count();
|
|
|
|
if (m_num_pieces == int(m_have_piece.size()))
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " *** on_metadata(): THIS IS A SEED ***\n";
|
|
|
|
#endif
|
|
|
|
// if this is a web seed. we don't have a peer_info struct
|
|
|
|
if (m_peer_info) m_peer_info->seed = true;
|
|
|
|
m_upload_only = true;
|
|
|
|
|
|
|
|
t->peer_has_all();
|
|
|
|
disconnect_if_redundant();
|
|
|
|
if (m_disconnecting) return;
|
|
|
|
|
|
|
|
on_metadata();
|
|
|
|
if (m_disconnecting) return;
|
|
|
|
|
|
|
|
if (!t->is_finished())
|
|
|
|
t->get_policy().peer_is_interesting(*this);
|
|
|
|
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
TORRENT_ASSERT(!m_have_all);
|
|
|
|
|
|
|
|
on_metadata();
|
|
|
|
if (m_disconnecting) return;
|
|
|
|
|
|
|
|
// let the torrent know which pieces the
|
|
|
|
// peer has
|
|
|
|
// if we're a seed, we don't keep track of piece availability
|
|
|
|
bool interesting = false;
|
|
|
|
if (!t->is_seed())
|
|
|
|
{
|
|
|
|
t->peer_has(m_have_piece);
|
|
|
|
|
|
|
|
for (int i = 0; i < (int)m_have_piece.size(); ++i)
|
|
|
|
{
|
|
|
|
if (m_have_piece[i])
|
|
|
|
{
|
|
|
|
if (!t->have_piece(i) && t->picker().piece_priority(i) != 0)
|
|
|
|
interesting = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (interesting) t->get_policy().peer_is_interesting(*this);
|
|
|
|
else if (upload_only()) disconnect("upload to upload connections");
|
|
|
|
}
|
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
void peer_connection::init()
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
TORRENT_ASSERT(t->valid_metadata());
|
|
|
|
TORRENT_ASSERT(t->ready_for_connections());
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
m_have_piece.resize(t->torrent_file().num_pieces(), m_have_all);
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
if (m_have_all) m_num_pieces = t->torrent_file().num_pieces();
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-08-29 19:21:56 +02:00
|
|
|
m_initialized = true;
|
|
|
|
#endif
|
2004-06-14 01:30:42 +02:00
|
|
|
// now that we have a piece_picker,
|
2008-01-31 18:52:29 +01:00
|
|
|
// update it with this peer's pieces
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
TORRENT_ASSERT(m_num_pieces == m_have_piece.count());
|
2008-01-31 18:52:29 +01:00
|
|
|
|
|
|
|
if (m_num_pieces == int(m_have_piece.size()))
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2004-06-14 01:30:42 +02:00
|
|
|
(*m_logger) << " *** THIS IS A SEED ***\n";
|
|
|
|
#endif
|
2007-04-15 04:14:02 +02:00
|
|
|
// if this is a web seed. we don't have a peer_info struct
|
|
|
|
if (m_peer_info) m_peer_info->seed = true;
|
2008-09-20 01:02:45 +02:00
|
|
|
m_upload_only = true;
|
2008-07-18 12:03:42 +02:00
|
|
|
|
2007-04-15 04:14:02 +02:00
|
|
|
t->peer_has_all();
|
2008-07-18 12:03:42 +02:00
|
|
|
if (t->is_finished()) send_not_interested();
|
|
|
|
else t->get_policy().peer_is_interesting(*this);
|
2007-04-15 04:14:02 +02:00
|
|
|
return;
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
2007-04-15 04:14:02 +02:00
|
|
|
// if we're a seed, we don't keep track of piece availability
|
|
|
|
if (!t->is_seed())
|
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
t->peer_has(m_have_piece);
|
2007-04-15 04:14:02 +02:00
|
|
|
bool interesting = false;
|
|
|
|
for (int i = 0; i < int(m_have_piece.size()); ++i)
|
|
|
|
{
|
|
|
|
if (m_have_piece[i])
|
|
|
|
{
|
|
|
|
// if the peer has a piece and we don't, the peer is interesting
|
|
|
|
if (!t->have_piece(i)
|
|
|
|
&& t->picker().piece_priority(i) != 0)
|
|
|
|
interesting = true;
|
|
|
|
}
|
|
|
|
}
|
2008-07-18 12:03:42 +02:00
|
|
|
if (interesting) t->get_policy().peer_is_interesting(*this);
|
|
|
|
else send_not_interested();
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
update_interest();
|
2007-04-15 04:14:02 +02:00
|
|
|
}
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
2004-01-04 05:29:13 +01:00
|
|
|
peer_connection::~peer_connection()
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2006-06-12 01:24:36 +02:00
|
|
|
// INVARIANT_CHECK;
|
2008-01-19 20:00:54 +01:00
|
|
|
TORRENT_ASSERT(!m_in_constructor);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_disconnecting);
|
2008-07-18 12:03:42 +02:00
|
|
|
TORRENT_ASSERT(m_disconnect_started);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-05-05 08:25:22 +02:00
|
|
|
m_disk_recv_buffer_size = 0;
|
2008-04-10 12:03:23 +02:00
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2005-01-10 12:14:22 +01:00
|
|
|
if (m_logger)
|
|
|
|
{
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2005-01-10 12:14:22 +01:00
|
|
|
<< " *** CONNECTION CLOSED\n";
|
|
|
|
}
|
|
|
|
#endif
|
2008-01-07 02:10:46 +01:00
|
|
|
TORRENT_ASSERT(!m_ses.has_peer(this));
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-01-07 02:10:46 +01:00
|
|
|
for (aux::session_impl::torrent_map::const_iterator i = m_ses.m_torrents.begin()
|
|
|
|
, end(m_ses.m_torrents.end()); i != end; ++i)
|
|
|
|
TORRENT_ASSERT(!i->second->has_peer(this));
|
2007-04-12 12:21:55 +02:00
|
|
|
if (m_peer_info)
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_peer_info->connection == 0);
|
2007-04-12 12:21:55 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2005-10-01 17:12:10 +02:00
|
|
|
#endif
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-09-06 23:04:57 +02:00
|
|
|
int peer_connection::picker_options() const
|
|
|
|
{
|
|
|
|
int ret = 0;
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
if (!t) return 0;
|
|
|
|
|
|
|
|
if (t->is_sequential_download())
|
|
|
|
{
|
2009-01-14 10:07:27 +01:00
|
|
|
ret |= piece_picker::sequential | piece_picker::ignore_whole_pieces;
|
2008-09-06 23:04:57 +02:00
|
|
|
}
|
|
|
|
else if (t->num_have() < t->settings().initial_picker_threshold)
|
|
|
|
{
|
|
|
|
// if we have fewer pieces than a certain threshols
|
|
|
|
// don't pick rare pieces, just pick random ones,
|
|
|
|
// and prioritize finishing them
|
|
|
|
ret |= piece_picker::prioritize_partials;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2009-01-14 10:07:27 +01:00
|
|
|
ret |= piece_picker::rarest_first | piece_picker::speed_affinity;
|
2008-09-06 23:04:57 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (m_snubbed)
|
|
|
|
{
|
|
|
|
// snubbed peers should request
|
|
|
|
// the common pieces first, just to make
|
|
|
|
// it more likely for all snubbed peers to
|
|
|
|
// request blocks from the same piece
|
|
|
|
ret |= piece_picker::reverse;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (t->settings().prioritize_partial_pieces)
|
|
|
|
ret |= piece_picker::prioritize_partials;
|
|
|
|
|
|
|
|
if (on_parole()) ret |= piece_picker::on_parole
|
|
|
|
| piece_picker::prioritize_partials;
|
|
|
|
|
|
|
|
// only one of rarest_first, common_first and sequential can be set.
|
|
|
|
TORRENT_ASSERT(bool(ret & piece_picker::rarest_first)
|
|
|
|
+ bool(ret & piece_picker::sequential) <= 1);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2007-10-15 07:03:29 +02:00
|
|
|
void peer_connection::fast_reconnect(bool r)
|
|
|
|
{
|
2008-04-01 19:38:19 +02:00
|
|
|
if (!peer_info_struct() || peer_info_struct()->fast_reconnects > 1)
|
|
|
|
return;
|
2007-10-15 07:03:29 +02:00
|
|
|
m_fast_reconnect = r;
|
|
|
|
peer_info_struct()->connected = time_now()
|
|
|
|
- seconds(m_ses.settings().min_reconnect_time
|
|
|
|
* m_ses.settings().max_failcount);
|
2008-04-01 19:38:19 +02:00
|
|
|
++peer_info_struct()->fast_reconnects;
|
2007-10-15 07:03:29 +02:00
|
|
|
}
|
|
|
|
|
2004-03-11 15:56:48 +01:00
|
|
|
void peer_connection::announce_piece(int index)
|
|
|
|
{
|
2007-06-06 02:41:20 +02:00
|
|
|
// dont announce during handshake
|
|
|
|
if (in_handshake()) return;
|
2007-09-01 09:38:10 +02:00
|
|
|
|
|
|
|
// remove suggested pieces that we have
|
|
|
|
std::vector<int>::iterator i = std::find(
|
|
|
|
m_suggested_pieces.begin(), m_suggested_pieces.end(), index);
|
|
|
|
if (i != m_suggested_pieces.end()) m_suggested_pieces.erase(i);
|
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
if (has_piece(index))
|
2008-04-20 19:17:58 +02:00
|
|
|
{
|
2008-07-18 12:03:42 +02:00
|
|
|
// if we got a piece that this peer has
|
|
|
|
// it might have been the last interesting
|
|
|
|
// piece this peer had. We might not be
|
|
|
|
// interested anymore
|
|
|
|
update_interest();
|
|
|
|
if (is_disconnecting()) return;
|
|
|
|
|
|
|
|
// optimization, don't send have messages
|
|
|
|
// to peers that already have the piece
|
|
|
|
if (!m_ses.settings().send_redundant_have)
|
|
|
|
{
|
2008-04-20 19:17:58 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2008-07-18 12:03:42 +02:00
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> HAVE [ piece: " << index << " ] SUPRESSED\n";
|
2008-04-20 19:17:58 +02:00
|
|
|
#endif
|
2008-07-18 12:03:42 +02:00
|
|
|
return;
|
|
|
|
}
|
2008-04-20 19:17:58 +02:00
|
|
|
}
|
2006-11-19 16:29:58 +01:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-11-19 16:29:58 +01:00
|
|
|
<< " ==> HAVE [ piece: " << index << "]\n";
|
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
write_have(index);
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-11-19 16:29:58 +01:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
TORRENT_ASSERT(t->have_piece(index));
|
2006-11-19 16:29:58 +01:00
|
|
|
#endif
|
2004-03-11 15:56:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
bool peer_connection::has_piece(int i) const
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
TORRENT_ASSERT(t->valid_metadata());
|
|
|
|
TORRENT_ASSERT(i >= 0);
|
|
|
|
TORRENT_ASSERT(i < t->torrent_file().num_pieces());
|
2004-03-11 15:56:48 +01:00
|
|
|
return m_have_piece[i];
|
|
|
|
}
|
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
std::vector<piece_block> const& peer_connection::request_queue() const
|
2005-09-27 10:07:24 +02:00
|
|
|
{
|
|
|
|
return m_request_queue;
|
|
|
|
}
|
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
std::vector<pending_block> const& peer_connection::download_queue() const
|
2004-03-11 15:56:48 +01:00
|
|
|
{
|
|
|
|
return m_download_queue;
|
|
|
|
}
|
|
|
|
|
2005-09-27 10:07:24 +02:00
|
|
|
std::deque<peer_request> const& peer_connection::upload_queue() const
|
2004-03-11 15:56:48 +01:00
|
|
|
{
|
|
|
|
return m_requests;
|
|
|
|
}
|
|
|
|
|
2009-02-17 03:33:45 +01:00
|
|
|
time_duration peer_connection::download_queue_time(int extra_bytes) const
|
|
|
|
{
|
|
|
|
int rate = m_statistics.transfer_rate(stat::download_payload)
|
|
|
|
+ m_statistics.transfer_rate(stat::download_protocol);
|
|
|
|
// avoid division by zero
|
|
|
|
if (rate < 50) rate = 50;
|
2009-03-17 10:34:44 +01:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
return seconds((m_outstanding_bytes + m_queued_time_critical * t->block_size()) / rate);
|
2009-02-17 03:33:45 +01:00
|
|
|
}
|
|
|
|
|
2004-03-11 15:56:48 +01:00
|
|
|
void peer_connection::add_stat(size_type downloaded, size_type uploaded)
|
|
|
|
{
|
|
|
|
m_statistics.add_stat(downloaded, uploaded);
|
|
|
|
}
|
|
|
|
|
2008-05-28 04:35:02 +02:00
|
|
|
bitfield const& peer_connection::get_bitfield() const
|
2004-03-11 15:56:48 +01:00
|
|
|
{
|
|
|
|
return m_have_piece;
|
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
void peer_connection::received_valid_data(int index)
|
2004-03-11 15:56:48 +01:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
2008-01-07 02:10:46 +01:00
|
|
|
#ifdef BOOST_NO_EXCEPTIONS
|
|
|
|
(*i)->on_piece_pass(index);
|
|
|
|
#else
|
2006-11-14 01:08:16 +01:00
|
|
|
try { (*i)->on_piece_pass(index); } catch (std::exception&) {}
|
2008-01-07 02:10:46 +01:00
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
2004-03-11 15:56:48 +01:00
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
void peer_connection::received_invalid_data(int index)
|
2004-03-11 15:56:48 +01:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
2008-01-07 02:10:46 +01:00
|
|
|
#ifdef BOOST_NO_EXCEPTIONS
|
|
|
|
(*i)->on_piece_failed(index);
|
|
|
|
#else
|
2006-11-14 01:08:16 +01:00
|
|
|
try { (*i)->on_piece_failed(index); } catch (std::exception&) {}
|
2008-01-07 02:10:46 +01:00
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2006-11-14 01:08:16 +01:00
|
|
|
|
2007-05-25 21:42:10 +02:00
|
|
|
if (peer_info_struct())
|
|
|
|
{
|
2008-07-11 11:23:22 +02:00
|
|
|
if (m_ses.settings().use_parole_mode)
|
|
|
|
peer_info_struct()->on_parole = true;
|
|
|
|
|
2007-05-25 21:42:10 +02:00
|
|
|
++peer_info_struct()->hashfails;
|
2008-04-01 19:38:19 +02:00
|
|
|
boost::int8_t& trust_points = peer_info_struct()->trust_points;
|
2007-05-16 06:12:13 +02:00
|
|
|
|
2007-05-25 21:42:10 +02:00
|
|
|
// we decrease more than we increase, to keep the
|
|
|
|
// allowed failed/passed ratio low.
|
|
|
|
// TODO: make this limit user settable
|
|
|
|
trust_points -= 2;
|
|
|
|
if (trust_points < -7) trust_points = -7;
|
|
|
|
}
|
2004-03-11 15:56:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
size_type peer_connection::total_free_upload() const
|
|
|
|
{
|
|
|
|
return m_free_upload;
|
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::add_free_upload(size_type free_upload)
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2004-03-11 15:56:48 +01:00
|
|
|
m_free_upload += free_upload;
|
|
|
|
}
|
|
|
|
|
2004-01-08 14:03:38 +01:00
|
|
|
// verifies a piece to see if it is valid (is within a valid range)
|
|
|
|
// and if it can correspond to a request generated by libtorrent.
|
|
|
|
bool peer_connection::verify_piece(const peer_request& p) const
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t->valid_metadata());
|
2007-09-03 23:16:24 +02:00
|
|
|
torrent_info const& ti = t->torrent_file();
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2004-01-08 14:03:38 +01:00
|
|
|
return p.piece >= 0
|
2006-04-25 23:04:48 +02:00
|
|
|
&& p.piece < t->torrent_file().num_pieces()
|
2009-03-17 10:34:44 +01:00
|
|
|
&& p.length >= 0
|
2004-01-08 14:03:38 +01:00
|
|
|
&& p.start >= 0
|
2006-04-25 23:04:48 +02:00
|
|
|
&& (p.length == t->block_size()
|
|
|
|
|| (p.length < t->block_size()
|
2007-09-03 23:16:24 +02:00
|
|
|
&& p.piece == ti.num_pieces()-1
|
|
|
|
&& p.start + p.length == ti.piece_size(p.piece))
|
2006-11-14 01:08:16 +01:00
|
|
|
|| (m_request_large_blocks
|
2009-04-04 18:19:19 +02:00
|
|
|
&& p.length <= ti.piece_length() * (m_prefer_whole_pieces == 0 ?
|
|
|
|
1 : m_prefer_whole_pieces)))
|
2007-09-03 23:20:28 +02:00
|
|
|
&& p.piece * size_type(ti.piece_length()) + p.start + p.length
|
2007-09-03 23:16:24 +02:00
|
|
|
<= ti.total_size()
|
2006-11-14 01:08:16 +01:00
|
|
|
&& (p.start % t->block_size() == 0);
|
2004-01-08 14:03:38 +01:00
|
|
|
}
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::attach_to_torrent(sha1_hash const& ih)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_disconnecting);
|
|
|
|
TORRENT_ASSERT(m_torrent.expired());
|
2007-08-16 14:41:46 +02:00
|
|
|
boost::weak_ptr<torrent> wpt = m_ses.find_torrent(ih);
|
2007-08-17 00:13:35 +02:00
|
|
|
boost::shared_ptr<torrent> t = wpt.lock();
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2006-05-28 21:03:54 +02:00
|
|
|
if (t && t->is_aborted())
|
2007-08-17 00:13:35 +02:00
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-08-17 00:13:35 +02:00
|
|
|
(*m_logger) << " *** the torrent has been aborted\n";
|
|
|
|
#endif
|
2006-05-28 21:03:54 +02:00
|
|
|
t.reset();
|
2007-08-17 00:13:35 +02:00
|
|
|
}
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (!t)
|
|
|
|
{
|
|
|
|
// we couldn't find the torrent!
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-08-17 00:13:35 +02:00
|
|
|
(*m_logger) << " *** couldn't find a torrent with the given info_hash: " << ih << "\n";
|
|
|
|
(*m_logger) << " torrents:\n";
|
|
|
|
session_impl::torrent_map const& torrents = m_ses.m_torrents;
|
|
|
|
for (session_impl::torrent_map::const_iterator i = torrents.begin()
|
|
|
|
, end(torrents.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
(*m_logger) << " " << i->second->torrent_file().info_hash() << "\n";
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
2008-05-12 08:03:31 +02:00
|
|
|
disconnect("got invalid info-hash", 2);
|
2008-01-07 02:10:46 +01:00
|
|
|
return;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (t->is_paused())
|
|
|
|
{
|
|
|
|
// paused torrents will not accept
|
|
|
|
// incoming connections
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2006-04-25 23:04:48 +02:00
|
|
|
(*m_logger) << " rejected connection to paused torrent\n";
|
|
|
|
#endif
|
2008-01-07 02:10:46 +01:00
|
|
|
disconnect("connection rejected bacause torrent is paused");
|
|
|
|
return;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrent.expired());
|
2006-04-25 23:04:48 +02:00
|
|
|
// check to make sure we don't have another connection with the same
|
|
|
|
// info_hash and peer_id. If we do. close this connection.
|
|
|
|
t->attach_peer(this);
|
2008-01-19 20:00:54 +01:00
|
|
|
if (m_disconnecting) return;
|
2007-08-16 14:41:46 +02:00
|
|
|
m_torrent = wpt;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_torrent.expired());
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// if the torrent isn't ready to accept
|
|
|
|
// connections yet, we'll have to wait with
|
|
|
|
// our initialization
|
|
|
|
if (t->ready_for_connections()) init();
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_torrent.expired());
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// assume the other end has no pieces
|
|
|
|
// if we don't have valid metadata yet,
|
|
|
|
// leave the vector unallocated
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_num_pieces == 0);
|
2008-05-28 04:35:02 +02:00
|
|
|
m_have_piece.clear_all();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_torrent.expired());
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// message handlers
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// -----------------------------
|
|
|
|
// --------- KEEPALIVE ---------
|
|
|
|
// -----------------------------
|
|
|
|
|
|
|
|
void peer_connection::incoming_keepalive()
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== KEEPALIVE\n";
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ----------- CHOKE -----------
|
|
|
|
// -----------------------------
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_choke()
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2004-01-05 00:51:54 +01:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_choke()) return;
|
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-04-02 22:00:24 +02:00
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== CHOKE\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
|
|
|
m_peer_choked = true;
|
2006-12-16 03:19:53 +01:00
|
|
|
|
2007-10-10 04:27:55 +02:00
|
|
|
if (peer_info_struct() == 0 || !peer_info_struct()->on_parole)
|
2004-01-04 06:28:24 +01:00
|
|
|
{
|
2007-10-10 04:27:55 +02:00
|
|
|
// if the peer is not in parole mode, clear the queued
|
|
|
|
// up block requests
|
|
|
|
if (!t->is_seed())
|
2006-12-04 13:20:34 +01:00
|
|
|
{
|
2007-10-10 04:27:55 +02:00
|
|
|
piece_picker& p = t->picker();
|
2009-03-17 10:34:44 +01:00
|
|
|
for (std::vector<piece_block>::const_iterator i = m_request_queue.begin()
|
2007-10-10 04:27:55 +02:00
|
|
|
, end(m_request_queue.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
p.abort_download(*i);
|
|
|
|
}
|
2006-12-04 13:20:34 +01:00
|
|
|
}
|
2007-10-10 04:27:55 +02:00
|
|
|
m_request_queue.clear();
|
2005-09-27 10:07:24 +02:00
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2004-01-04 06:28:24 +01:00
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
bool match_request(peer_request const& r, piece_block const& b, int block_size)
|
|
|
|
{
|
|
|
|
if (b.piece_index != r.piece) return false;
|
|
|
|
if (b.block_index != r.start / block_size) return false;
|
|
|
|
if (r.start % block_size != 0) return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// -----------------------------
|
|
|
|
// -------- REJECT PIECE -------
|
|
|
|
// -----------------------------
|
|
|
|
|
|
|
|
void peer_connection::incoming_reject_request(peer_request const& r)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-08-14 19:47:48 +02:00
|
|
|
|
2007-09-14 05:38:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_reject(r)) return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
std::vector<pending_block>::iterator i = std::find_if(
|
2007-08-14 19:47:48 +02:00
|
|
|
m_download_queue.begin(), m_download_queue.end()
|
2008-07-07 14:04:06 +02:00
|
|
|
, bind(match_request, boost::cref(r), bind(&pending_block::block, _1)
|
|
|
|
, t->block_size()));
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " <== REJECT_PIECE [ piece: " << r.piece << " | s: " << r.start << " | l: " << r.length << " ]\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (i != m_download_queue.end())
|
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
piece_block b = i->block;
|
|
|
|
bool remove_from_picker = !i->timed_out && !i->not_wanted;
|
2007-08-14 19:47:48 +02:00
|
|
|
m_download_queue.erase(i);
|
2009-03-17 10:34:44 +01:00
|
|
|
m_outstanding_bytes -= r.length;
|
2007-10-10 04:27:55 +02:00
|
|
|
|
|
|
|
// if the peer is in parole mode, keep the request
|
|
|
|
if (peer_info_struct() && peer_info_struct()->on_parole)
|
2007-08-14 19:47:48 +02:00
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
m_request_queue.insert(m_request_queue.begin(), b);
|
2007-10-10 04:27:55 +02:00
|
|
|
}
|
2009-03-17 10:34:44 +01:00
|
|
|
else if (!t->is_seed() && remove_from_picker)
|
2007-10-10 04:27:55 +02:00
|
|
|
{
|
|
|
|
piece_picker& p = t->picker();
|
|
|
|
p.abort_download(b);
|
2007-08-14 19:47:48 +02:00
|
|
|
}
|
2009-03-17 10:34:44 +01:00
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_DEBUG
|
|
|
|
check_invariant();
|
|
|
|
#endif
|
2007-08-14 19:47:48 +02:00
|
|
|
}
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-08-14 19:47:48 +02:00
|
|
|
else
|
|
|
|
{
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " *** PIECE NOT IN REQUEST QUEUE\n";
|
|
|
|
}
|
|
|
|
#endif
|
2007-10-03 18:57:20 +02:00
|
|
|
if (has_peer_choked())
|
|
|
|
{
|
|
|
|
// if we're choked and we got a rejection of
|
|
|
|
// a piece in the allowed fast set, remove it
|
|
|
|
// from the allow fast set.
|
|
|
|
std::vector<int>::iterator i = std::find(
|
|
|
|
m_allowed_fast.begin(), m_allowed_fast.end(), r.piece);
|
|
|
|
if (i != m_allowed_fast.end()) m_allowed_fast.erase(i);
|
|
|
|
}
|
2007-10-04 11:32:09 +02:00
|
|
|
else
|
|
|
|
{
|
|
|
|
std::vector<int>::iterator i = std::find(m_suggested_pieces.begin()
|
|
|
|
, m_suggested_pieces.end(), r.piece);
|
|
|
|
if (i != m_suggested_pieces.end())
|
|
|
|
m_suggested_pieces.erase(i);
|
|
|
|
}
|
|
|
|
|
2008-07-08 23:33:36 +02:00
|
|
|
if (m_request_queue.empty() && m_download_queue.size() < 2)
|
2007-08-14 19:47:48 +02:00
|
|
|
{
|
2008-07-08 23:33:36 +02:00
|
|
|
request_a_block(*t, *this);
|
2007-08-14 19:47:48 +02:00
|
|
|
send_block_requests();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-09-01 09:38:10 +02:00
|
|
|
// -----------------------------
|
2008-01-02 04:18:29 +01:00
|
|
|
// ------- SUGGEST PIECE -------
|
2007-09-01 09:38:10 +02:00
|
|
|
// -----------------------------
|
|
|
|
|
|
|
|
void peer_connection::incoming_suggest(int index)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
2007-09-14 05:38:38 +02:00
|
|
|
|
2007-09-01 09:38:10 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " <== SUGGEST_PIECE [ piece: " << index << " ]\n";
|
|
|
|
#endif
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
if (!t) return;
|
|
|
|
|
2007-09-14 05:38:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_suggest(index)) return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-09-01 09:38:10 +02:00
|
|
|
if (t->have_piece(index)) return;
|
|
|
|
|
|
|
|
if (m_suggested_pieces.size() > 9)
|
|
|
|
m_suggested_pieces.erase(m_suggested_pieces.begin());
|
|
|
|
m_suggested_pieces.push_back(index);
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ** SUGGEST_PIECE [ piece: " << index << " added to set: " << m_suggested_pieces.size() << " ]\n";
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ---------- UNCHOKE ----------
|
|
|
|
// -----------------------------
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_unchoke()
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_unchoke()) return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== UNCHOKE\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
|
|
|
m_peer_choked = false;
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
t->get_policy().unchoked(*this);
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// -------- INTERESTED ---------
|
|
|
|
// -----------------------------
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_interested()
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_interested()) return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== INTERESTED\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
|
|
|
m_peer_interested = true;
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2009-03-12 18:06:41 +01:00
|
|
|
if (ignore_unchoke_slots()) send_unchoke();
|
2006-04-25 23:04:48 +02:00
|
|
|
t->get_policy().interested(*this);
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ------ NOT INTERESTED -------
|
|
|
|
// -----------------------------
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_not_interested()
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_not_interested()) return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
m_became_uninterested = time_now();
|
2004-02-01 14:48:30 +01:00
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== NOT_INTERESTED\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
m_peer_interested = false;
|
|
|
|
if (is_disconnecting()) return;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2009-01-28 07:14:21 +01:00
|
|
|
if (!is_choked() && !ignore_unchoke_slots())
|
2008-10-19 00:12:05 +02:00
|
|
|
{
|
|
|
|
if (m_peer_info && m_peer_info->optimistically_unchoked)
|
|
|
|
{
|
|
|
|
m_peer_info->optimistically_unchoked = false;
|
|
|
|
m_ses.m_optimistic_unchoke_time_scaler = 0;
|
|
|
|
}
|
|
|
|
t->choke_peer(*this);
|
|
|
|
--m_ses.m_num_unchoked;
|
|
|
|
m_ses.m_unchoke_time_scaler = 0;
|
|
|
|
}
|
|
|
|
|
2009-03-12 18:06:41 +01:00
|
|
|
if (ignore_unchoke_slots()) send_unchoke();
|
2006-04-25 23:04:48 +02:00
|
|
|
t->get_policy().not_interested(*this);
|
2008-12-08 07:36:22 +01:00
|
|
|
|
|
|
|
if (t->super_seeding() && m_superseed_piece != -1)
|
|
|
|
{
|
|
|
|
// assume the peer has the piece we're superseeding to it
|
|
|
|
// and give it another one
|
|
|
|
if (!m_have_piece[m_superseed_piece]) incoming_have(m_superseed_piece);
|
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ----------- HAVE ------------
|
|
|
|
// -----------------------------
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_have(int index)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_have(index)) return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
// if we haven't received a bitfield, it was
|
|
|
|
// probably omitted, which is the same as 'have_none'
|
|
|
|
if (!m_bitfield_received) incoming_have_none();
|
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2004-06-28 22:23:42 +02:00
|
|
|
<< " <== HAVE [ piece: " << index << "]\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
|
|
|
|
2008-09-07 12:00:58 +02:00
|
|
|
if (is_disconnecting()) return;
|
|
|
|
|
2007-12-21 00:58:58 +01:00
|
|
|
if (!t->valid_metadata() && index > int(m_have_piece.size()))
|
|
|
|
{
|
|
|
|
if (index < 65536)
|
|
|
|
{
|
|
|
|
// if we don't have metadata
|
|
|
|
// and we might not have received a bitfield
|
|
|
|
// extend the bitmask to fit the new
|
|
|
|
// have message
|
|
|
|
m_have_piece.resize(index + 1, false);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// unless the index > 64k, in which case
|
|
|
|
// we just ignore it
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// if we got an invalid message, abort
|
2007-12-21 00:58:58 +01:00
|
|
|
if (index >= int(m_have_piece.size()) || index < 0)
|
2008-01-07 02:10:46 +01:00
|
|
|
{
|
2008-05-12 08:03:31 +02:00
|
|
|
disconnect("got 'have'-message with higher index than the number of pieces", 2);
|
2008-01-07 02:10:46 +01:00
|
|
|
return;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-12-09 08:56:37 +01:00
|
|
|
if (t->super_seeding() && !m_ses.settings().strict_super_seeding)
|
2008-12-08 07:36:22 +01:00
|
|
|
{
|
|
|
|
// if we're superseeding and the peer just told
|
|
|
|
// us that it completed the piece we're superseeding
|
|
|
|
// to it, change the superseeding piece for this peer
|
|
|
|
// if the peer optimizes out redundant have messages
|
|
|
|
// this will be handled when the peer sends not-interested
|
|
|
|
// instead.
|
|
|
|
if (m_superseed_piece == index)
|
|
|
|
{
|
|
|
|
superseed_piece(t->get_piece_to_super_seed(m_have_piece));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
if (m_have_piece[index])
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2006-04-25 23:04:48 +02:00
|
|
|
(*m_logger) << " got redundant HAVE message for index: " << index << "\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2008-05-28 04:35:02 +02:00
|
|
|
m_have_piece.set_bit(index);
|
2008-08-29 19:21:56 +02:00
|
|
|
++m_num_pieces;
|
2004-01-28 12:37:46 +01:00
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
// only update the piece_picker if
|
2006-12-04 13:20:34 +01:00
|
|
|
// we have the metadata and if
|
|
|
|
// we're not a seed (in which case
|
|
|
|
// we won't have a piece picker)
|
2006-04-25 23:04:48 +02:00
|
|
|
if (t->valid_metadata())
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
t->peer_has(index);
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (!t->have_piece(index)
|
2006-12-04 13:20:34 +01:00
|
|
|
&& !t->is_seed()
|
2005-05-25 12:01:01 +02:00
|
|
|
&& !is_interesting()
|
2007-03-15 23:03:56 +01:00
|
|
|
&& t->picker().piece_priority(index) != 0)
|
2006-04-25 23:04:48 +02:00
|
|
|
t->get_policy().peer_is_interesting(*this);
|
2007-06-07 02:53:48 +02:00
|
|
|
|
2007-06-14 23:47:00 +02:00
|
|
|
// this will disregard all have messages we get within
|
|
|
|
// the first two seconds. Since some clients implements
|
|
|
|
// lazy bitfields, these will not be reliable to use
|
|
|
|
// for an estimated peer download rate.
|
|
|
|
if (!peer_info_struct() || time_now() - peer_info_struct()->connected > seconds(2))
|
|
|
|
{
|
|
|
|
// update bytes downloaded since last timer
|
|
|
|
m_remote_bytes_dled += t->torrent_file().piece_size(index);
|
|
|
|
}
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
2007-06-07 02:05:18 +02:00
|
|
|
|
2008-12-09 08:56:37 +01:00
|
|
|
// if we're super seeding, this might mean that somebody
|
|
|
|
// forwarded this piece. In which case we need to give
|
|
|
|
// a new piece to that peer
|
|
|
|
if (t->super_seeding()
|
|
|
|
&& m_ses.settings().strict_super_seeding
|
|
|
|
&& (index != m_superseed_piece || t->num_peers() == 1))
|
|
|
|
{
|
|
|
|
for (torrent::peer_iterator i = t->begin()
|
|
|
|
, end(t->end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
peer_connection* p = *i;
|
|
|
|
if (p->superseed_piece() != index) continue;
|
|
|
|
if (!p->has_piece(index)) continue;
|
|
|
|
p->superseed_piece(t->get_piece_to_super_seed(p->get_bitfield()));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
if (is_seed())
|
2004-01-12 21:31:27 +01:00
|
|
|
{
|
2008-07-18 12:03:42 +02:00
|
|
|
m_peer_info->seed = true;
|
|
|
|
m_upload_only = true;
|
|
|
|
disconnect_if_redundant();
|
|
|
|
if (is_disconnecting()) return;
|
2004-01-12 21:31:27 +01:00
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
|
|
|
}
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// --------- BITFIELD ----------
|
|
|
|
// -----------------------------
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2008-05-28 04:35:02 +02:00
|
|
|
void peer_connection::incoming_bitfield(bitfield const& bits)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
2008-05-28 04:35:02 +02:00
|
|
|
if ((*i)->on_bitfield(bits)) return;
|
2007-04-02 22:00:24 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== BITFIELD ";
|
2007-03-28 22:54:51 +02:00
|
|
|
|
2008-05-28 04:35:02 +02:00
|
|
|
for (int i = 0; i < int(bits.size()); ++i)
|
2007-03-28 22:54:51 +02:00
|
|
|
{
|
2008-05-28 04:35:02 +02:00
|
|
|
if (bits[i]) (*m_logger) << "1";
|
2007-03-28 22:54:51 +02:00
|
|
|
else (*m_logger) << "0";
|
|
|
|
}
|
|
|
|
(*m_logger) << "\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// if we don't have the metedata, we cannot
|
|
|
|
// verify the bitfield size
|
|
|
|
if (t->valid_metadata()
|
2008-08-29 19:21:56 +02:00
|
|
|
&& (bits.size() + 7) / 8 != (m_have_piece.size() + 7) / 8)
|
2008-01-07 02:10:46 +01:00
|
|
|
{
|
2009-04-04 11:52:25 +02:00
|
|
|
char msg[200];
|
|
|
|
snprintf(msg, 200, "got bitfield with invalid size: %d bytes. expected: %d bytes"
|
|
|
|
, int((bits.size() + 7) / 8), int((m_have_piece.size() + 7) / 8));
|
|
|
|
disconnect(msg, 2);
|
2008-01-07 02:10:46 +01:00
|
|
|
return;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
m_bitfield_received = true;
|
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
// if we don't have metadata yet
|
|
|
|
// just remember the bitmask
|
|
|
|
// don't update the piecepicker
|
|
|
|
// (since it doesn't exist yet)
|
2007-03-17 00:28:26 +01:00
|
|
|
if (!t->ready_for_connections())
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2008-05-28 04:35:02 +02:00
|
|
|
m_have_piece = bits;
|
|
|
|
m_num_pieces = bits.count();
|
|
|
|
if (m_peer_info) m_peer_info->seed = (m_num_pieces == int(bits.size()));
|
2004-06-14 01:30:42 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t->valid_metadata());
|
2007-08-14 19:47:48 +02:00
|
|
|
|
2008-05-28 04:35:02 +02:00
|
|
|
int num_pieces = bits.count();
|
2007-04-13 03:53:25 +02:00
|
|
|
if (num_pieces == int(m_have_piece.size()))
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << " *** THIS IS A SEED ***\n";
|
|
|
|
#endif
|
2007-04-13 03:57:16 +02:00
|
|
|
// if this is a web seed. we don't have a peer_info struct
|
|
|
|
if (m_peer_info) m_peer_info->seed = true;
|
2008-07-18 12:03:42 +02:00
|
|
|
m_upload_only = true;
|
2007-04-15 04:14:02 +02:00
|
|
|
|
2008-05-28 04:35:02 +02:00
|
|
|
m_have_piece.set_all();
|
2007-04-15 04:14:02 +02:00
|
|
|
m_num_pieces = num_pieces;
|
|
|
|
t->peer_has_all();
|
|
|
|
if (!t->is_finished())
|
|
|
|
t->get_policy().peer_is_interesting(*this);
|
2008-08-12 10:18:05 +02:00
|
|
|
|
|
|
|
disconnect_if_redundant();
|
|
|
|
|
2007-04-15 04:14:02 +02:00
|
|
|
return;
|
2007-04-13 03:53:25 +02:00
|
|
|
}
|
|
|
|
|
2007-02-12 06:46:29 +01:00
|
|
|
// let the torrent know which pieces the
|
|
|
|
// peer has
|
2007-04-15 04:14:02 +02:00
|
|
|
// if we're a seed, we don't keep track of piece availability
|
2008-01-31 18:52:29 +01:00
|
|
|
bool interesting = false;
|
2007-04-15 04:14:02 +02:00
|
|
|
if (!t->is_seed())
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2008-05-28 04:35:02 +02:00
|
|
|
t->peer_has(bits);
|
2008-01-31 18:52:29 +01:00
|
|
|
|
2007-04-15 04:14:02 +02:00
|
|
|
for (int i = 0; i < (int)m_have_piece.size(); ++i)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2008-05-28 04:35:02 +02:00
|
|
|
bool have = bits[i];
|
2007-04-15 04:14:02 +02:00
|
|
|
if (have && !m_have_piece[i])
|
|
|
|
{
|
|
|
|
if (!t->have_piece(i) && t->picker().piece_priority(i) != 0)
|
|
|
|
interesting = true;
|
|
|
|
}
|
|
|
|
else if (!have && m_have_piece[i])
|
|
|
|
{
|
|
|
|
// this should probably not be allowed
|
|
|
|
t->peer_lost(i);
|
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
|
|
|
}
|
2008-01-31 18:52:29 +01:00
|
|
|
|
2008-05-28 04:35:02 +02:00
|
|
|
m_have_piece = bits;
|
2008-01-31 18:52:29 +01:00
|
|
|
m_num_pieces = num_pieces;
|
|
|
|
|
|
|
|
if (interesting) t->get_policy().peer_is_interesting(*this);
|
2008-05-18 07:59:47 +02:00
|
|
|
else if (upload_only()) disconnect("upload to upload connections");
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-12-17 20:03:23 +01:00
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
void peer_connection::disconnect_if_redundant()
|
|
|
|
{
|
|
|
|
if (!m_ses.settings().close_redundant_connections) return;
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
if (m_upload_only && t->is_finished())
|
2008-11-01 11:14:11 +01:00
|
|
|
{
|
2008-07-18 12:03:42 +02:00
|
|
|
disconnect("seed to seed");
|
2008-11-01 11:14:11 +01:00
|
|
|
return;
|
|
|
|
}
|
2008-07-18 12:03:42 +02:00
|
|
|
|
|
|
|
if (m_upload_only
|
|
|
|
&& !m_interesting
|
|
|
|
&& m_bitfield_received
|
|
|
|
&& t->are_files_checked())
|
2008-11-01 11:14:11 +01:00
|
|
|
{
|
2008-07-18 12:03:42 +02:00
|
|
|
disconnect("uninteresting upload-only peer");
|
2008-11-01 11:14:11 +01:00
|
|
|
return;
|
|
|
|
}
|
2008-07-18 12:03:42 +02:00
|
|
|
}
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ---------- REQUEST ----------
|
|
|
|
// -----------------------------
|
2003-12-17 20:03:23 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_request(peer_request const& r)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2004-01-05 00:51:54 +01:00
|
|
|
|
2008-12-08 07:36:22 +01:00
|
|
|
if (m_superseed_piece != -1
|
|
|
|
&& r.piece != m_superseed_piece)
|
|
|
|
{
|
|
|
|
++m_num_invalid_requests;
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " <== INVALID_SUPER_SEED_REQUEST [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " | "
|
|
|
|
"i: " << m_peer_interested << " | "
|
|
|
|
"t: " << (int)t->torrent_file().piece_size(r.piece) << " | "
|
|
|
|
"n: " << t->torrent_file().num_pieces() << " | "
|
|
|
|
"h: " << t->have_piece(r.piece) << " | "
|
|
|
|
"ss: " << m_superseed_piece << " ]\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (t->alerts().should_post<invalid_request_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(invalid_request_alert(
|
|
|
|
t->get_handle(), m_remote, m_peer_id, r));
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
// if we haven't received a bitfield, it was
|
|
|
|
// probably omitted, which is the same as 'have_none'
|
|
|
|
if (!m_bitfield_received) incoming_have_none();
|
2008-09-07 12:00:58 +02:00
|
|
|
if (is_disconnecting()) return;
|
2008-07-18 12:03:42 +02:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_request(r)) return;
|
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-04-02 22:00:24 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (!t->valid_metadata())
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
|
|
|
// if we don't have valid metadata yet,
|
|
|
|
// we shouldn't get a request
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2004-06-28 22:23:42 +02:00
|
|
|
<< " <== UNEXPECTED_REQUEST [ "
|
2004-06-14 01:30:42 +02:00
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " | "
|
|
|
|
"i: " << m_peer_interested << " | "
|
2008-04-05 23:18:27 +02:00
|
|
|
"t: " << t->torrent_file().piece_size(r.piece) << " | "
|
2006-04-25 23:04:48 +02:00
|
|
|
"n: " << t->torrent_file().num_pieces() << " ]\n";
|
2008-01-02 04:18:29 +01:00
|
|
|
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> REJECT_PIECE [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " ]\n";
|
2004-06-14 01:30:42 +02:00
|
|
|
#endif
|
2007-08-14 19:47:48 +02:00
|
|
|
write_reject_request(r);
|
2004-06-14 01:30:42 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2006-10-11 16:02:21 +02:00
|
|
|
if (int(m_requests.size()) > m_ses.settings().max_allowed_in_request_queue)
|
2004-02-20 16:22:23 +01:00
|
|
|
{
|
|
|
|
// don't allow clients to abuse our
|
|
|
|
// memory consumption.
|
|
|
|
// ignore requests if the client
|
|
|
|
// is making too many of them.
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2004-06-28 22:23:42 +02:00
|
|
|
<< " <== TOO MANY REQUESTS [ "
|
2004-06-14 01:30:42 +02:00
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " | "
|
|
|
|
"i: " << m_peer_interested << " | "
|
2008-04-05 23:18:27 +02:00
|
|
|
"t: " << t->torrent_file().piece_size(r.piece) << " | "
|
2006-04-25 23:04:48 +02:00
|
|
|
"n: " << t->torrent_file().num_pieces() << " ]\n";
|
2008-01-02 04:18:29 +01:00
|
|
|
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> REJECT_PIECE [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " ]\n";
|
2004-06-14 01:30:42 +02:00
|
|
|
#endif
|
2007-08-14 19:47:48 +02:00
|
|
|
write_reject_request(r);
|
2004-02-20 16:22:23 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// make sure this request
|
2004-11-30 12:17:32 +01:00
|
|
|
// is legal and that the peer
|
2004-01-05 00:51:54 +01:00
|
|
|
// is not choked
|
|
|
|
if (r.piece >= 0
|
2006-04-25 23:04:48 +02:00
|
|
|
&& r.piece < t->torrent_file().num_pieces()
|
|
|
|
&& t->have_piece(r.piece)
|
2004-01-05 00:51:54 +01:00
|
|
|
&& r.start >= 0
|
2006-04-25 23:04:48 +02:00
|
|
|
&& r.start < t->torrent_file().piece_size(r.piece)
|
2004-01-05 00:51:54 +01:00
|
|
|
&& r.length > 0
|
2006-04-25 23:04:48 +02:00
|
|
|
&& r.length + r.start <= t->torrent_file().piece_size(r.piece)
|
2007-08-03 18:12:32 +02:00
|
|
|
&& m_peer_interested
|
|
|
|
&& r.length <= t->block_size())
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2006-11-19 16:29:58 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-11-19 16:29:58 +01:00
|
|
|
<< " <== REQUEST [ piece: " << r.piece << " | s: " << r.start << " | l: " << r.length << " ]\n";
|
|
|
|
#endif
|
2004-01-05 00:51:54 +01:00
|
|
|
// if we have choked the client
|
|
|
|
// ignore the request
|
2007-08-14 19:47:48 +02:00
|
|
|
if (m_choked && m_accept_fast.find(r.piece) == m_accept_fast.end())
|
|
|
|
{
|
|
|
|
write_reject_request(r);
|
2009-03-12 18:06:41 +01:00
|
|
|
++m_choke_rejects;
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-01-02 04:18:29 +01:00
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " *** REJECTING REQUEST [ peer choked and piece not in allowed fast set ]\n";
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> REJECT_PIECE [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " ]\n";
|
2007-08-14 19:47:48 +02:00
|
|
|
#endif
|
2009-03-12 18:06:41 +01:00
|
|
|
|
|
|
|
if (m_choke_rejects > m_ses.settings().max_rejects)
|
|
|
|
{
|
|
|
|
disconnect("too many piece requests while choked");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
else if ((m_choke_rejects & 0xf) == 0)
|
|
|
|
{
|
|
|
|
// tell the peer it's choked again
|
|
|
|
// every 16 requests in a row
|
|
|
|
write_choke();
|
|
|
|
}
|
2007-08-14 19:47:48 +02:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2009-03-12 18:06:41 +01:00
|
|
|
m_choke_rejects = 0;
|
2007-08-14 19:47:48 +02:00
|
|
|
m_requests.push_back(r);
|
2007-09-01 09:38:10 +02:00
|
|
|
m_last_incoming_request = time_now();
|
2007-08-14 19:47:48 +02:00
|
|
|
fill_send_buffer();
|
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2004-06-28 22:23:42 +02:00
|
|
|
<< " <== INVALID_REQUEST [ "
|
2004-01-05 02:30:34 +01:00
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " | "
|
|
|
|
"i: " << m_peer_interested << " | "
|
2008-04-05 23:18:27 +02:00
|
|
|
"t: " << t->torrent_file().piece_size(r.piece) << " | "
|
2006-11-19 16:29:58 +01:00
|
|
|
"n: " << t->torrent_file().num_pieces() << " | "
|
2007-08-03 18:12:32 +02:00
|
|
|
"h: " << t->have_piece(r.piece) << " | "
|
|
|
|
"block_limit: " << t->block_size() << " ]\n";
|
2008-01-02 04:18:29 +01:00
|
|
|
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> REJECT_PIECE [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " ]\n";
|
2004-01-05 02:30:34 +01:00
|
|
|
#endif
|
2004-01-12 21:31:27 +01:00
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
write_reject_request(r);
|
2004-01-12 21:31:27 +01:00
|
|
|
++m_num_invalid_requests;
|
|
|
|
|
2008-07-06 14:22:56 +02:00
|
|
|
if (t->alerts().should_post<invalid_request_alert>())
|
2004-01-08 14:03:38 +01:00
|
|
|
{
|
2008-07-08 11:30:10 +02:00
|
|
|
t->alerts().post_alert(invalid_request_alert(
|
|
|
|
t->get_handle(), m_remote, m_peer_id, r));
|
2004-01-08 14:03:38 +01:00
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
void peer_connection::incoming_piece_fragment(int bytes)
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_piece = time_now();
|
2009-03-17 10:34:44 +01:00
|
|
|
m_outstanding_bytes -= bytes;
|
|
|
|
#ifdef TORRENT_DEBUG
|
2009-04-11 22:45:14 +02:00
|
|
|
boost::shared_ptr<torrent> t = associated_torrent().lock();
|
|
|
|
TORRENT_ASSERT(m_received_in_piece + bytes <= t->block_size());
|
2009-03-17 10:34:44 +01:00
|
|
|
m_received_in_piece += bytes;
|
|
|
|
#endif
|
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_DEBUG
|
|
|
|
check_invariant();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::start_receive_piece(peer_request const& r)
|
|
|
|
{
|
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_DEBUG
|
|
|
|
check_invariant();
|
|
|
|
#endif
|
|
|
|
#ifdef TORRENT_DEBUG
|
|
|
|
buffer::const_interval recv_buffer = receive_buffer();
|
|
|
|
int recv_pos = recv_buffer.end - recv_buffer.begin;
|
|
|
|
TORRENT_ASSERT(recv_pos >= 9);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = associated_torrent().lock();
|
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
piece_block b(r.piece, r.start / t->block_size());
|
|
|
|
|
|
|
|
if (!verify_piece(r))
|
|
|
|
{
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
2009-03-27 18:14:15 +01:00
|
|
|
<< " <== INVALID_PIECE [ piece: " << r.piece << " | "
|
|
|
|
"start: " << r.start << " | "
|
|
|
|
"length: " << r.length << " ]\n";
|
2009-03-17 10:34:44 +01:00
|
|
|
#endif
|
|
|
|
disconnect("got invalid piece packet", 2);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool in_req_queue = false;
|
|
|
|
for (std::vector<pending_block>::const_iterator i = m_download_queue.begin()
|
|
|
|
, end(m_download_queue.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if (i->block != b) continue;
|
|
|
|
in_req_queue = true;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
// if this is not in the request queue, we have to
|
|
|
|
// assume our outstanding bytes includes this piece too
|
|
|
|
if (!in_req_queue)
|
|
|
|
{
|
|
|
|
if (t->alerts().should_post<unwanted_block_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(unwanted_block_alert(t->get_handle(), m_remote
|
2009-04-25 10:19:21 +02:00
|
|
|
, m_peer_id, b.block_index, b.piece_index));
|
2009-03-17 10:34:44 +01:00
|
|
|
}
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_logger) << " *** The block we just got was not in the "
|
|
|
|
"request queue ***\n";
|
|
|
|
#endif
|
|
|
|
m_download_queue.insert(m_download_queue.begin(), b);
|
|
|
|
m_download_queue.front().not_wanted = true;
|
|
|
|
m_outstanding_bytes += r.length;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2006-12-22 01:45:43 +01:00
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-12-22 01:45:43 +01:00
|
|
|
struct check_postcondition
|
|
|
|
{
|
|
|
|
check_postcondition(boost::shared_ptr<torrent> const& t_
|
|
|
|
, bool init_check = true): t(t_) { if (init_check) check(); }
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2006-12-22 01:45:43 +01:00
|
|
|
~check_postcondition() { check(); }
|
|
|
|
|
|
|
|
void check()
|
|
|
|
{
|
|
|
|
if (!t->is_seed())
|
|
|
|
{
|
|
|
|
const int blocks_per_piece = static_cast<int>(
|
|
|
|
t->torrent_file().piece_length() / t->block_size());
|
|
|
|
|
|
|
|
std::vector<piece_picker::downloading_piece> const& dl_queue
|
|
|
|
= t->picker().get_download_queue();
|
|
|
|
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator i =
|
|
|
|
dl_queue.begin(); i != dl_queue.end(); ++i)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i->finished <= blocks_per_piece);
|
2006-12-22 01:45:43 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
shared_ptr<torrent> t;
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ----------- PIECE -----------
|
|
|
|
// -----------------------------
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_piece(peer_request const& p, char const* data)
|
2008-04-10 12:03:23 +02:00
|
|
|
{
|
2009-01-23 10:13:31 +01:00
|
|
|
char* buffer = m_ses.allocate_disk_buffer("receive buffer");
|
2008-04-10 12:03:23 +02:00
|
|
|
if (buffer == 0)
|
|
|
|
{
|
|
|
|
disconnect("out of memory");
|
|
|
|
return;
|
2008-06-29 11:50:42 +02:00
|
|
|
}
|
2008-04-10 12:03:23 +02:00
|
|
|
disk_buffer_holder holder(m_ses, buffer);
|
2008-04-29 05:12:14 +02:00
|
|
|
std::memcpy(buffer, data, p.length);
|
2008-04-10 12:03:23 +02:00
|
|
|
incoming_piece(p, holder);
|
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::incoming_piece(peer_request const& p, disk_buffer_holder& data)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-04-02 22:00:24 +02:00
|
|
|
|
2008-05-05 08:25:22 +02:00
|
|
|
TORRENT_ASSERT(!m_disk_recv_buffer);
|
2008-04-10 12:03:23 +02:00
|
|
|
TORRENT_ASSERT(m_disk_recv_buffer_size == 0);
|
|
|
|
|
2008-06-23 17:37:24 +02:00
|
|
|
#ifdef TORRENT_CORRUPT_DATA
|
|
|
|
// corrupt all pieces from certain peers
|
|
|
|
if (m_remote.address().is_v4()
|
|
|
|
&& (m_remote.address().to_v4().to_ulong() & 0xf) == 0)
|
|
|
|
{
|
|
|
|
data.get()[0] = ~data.get()[0];
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
// if we haven't received a bitfield, it was
|
|
|
|
// probably omitted, which is the same as 'have_none'
|
|
|
|
if (!m_bitfield_received) incoming_have_none();
|
2008-09-07 12:00:58 +02:00
|
|
|
if (is_disconnecting()) return;
|
2008-07-18 12:03:42 +02:00
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
if ((*i)->on_piece(p, data))
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_DEBUG
|
2009-04-11 22:45:14 +02:00
|
|
|
TORRENT_ASSERT(m_received_in_piece == p.length);
|
2009-03-17 10:34:44 +01:00
|
|
|
m_received_in_piece = 0;
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
2007-04-02 22:00:24 +02:00
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-04-02 22:00:24 +02:00
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-12-22 01:45:43 +01:00
|
|
|
check_postcondition post_checker_(t);
|
2008-10-10 07:25:55 +02:00
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2006-12-11 13:48:33 +01:00
|
|
|
t->check_invariant();
|
|
|
|
#endif
|
2008-01-31 18:52:29 +01:00
|
|
|
#endif
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-04-25 23:04:48 +02:00
|
|
|
<< " <== PIECE [ piece: " << p.piece << " | "
|
|
|
|
"s: " << p.start << " | "
|
|
|
|
"l: " << p.length << " | "
|
|
|
|
"ds: " << statistics().download_rate() << " | "
|
2008-09-04 16:39:02 +02:00
|
|
|
"qs: " << int(m_desired_queue_size) << " ]\n";
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2008-07-09 19:37:38 +02:00
|
|
|
if (p.length == 0)
|
|
|
|
{
|
|
|
|
if (t->alerts().should_post<peer_error_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(peer_error_alert(t->get_handle(), m_remote
|
2008-11-17 20:12:34 +01:00
|
|
|
, m_peer_id, "peer sent 0 length piece"));
|
2008-07-09 19:37:38 +02:00
|
|
|
}
|
2008-11-17 20:29:14 +01:00
|
|
|
// This is used as a reject-request by bitcomet
|
|
|
|
incoming_reject_request(p);
|
2008-07-09 19:37:38 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2006-12-04 21:42:47 +01:00
|
|
|
// if we're already seeding, don't bother,
|
|
|
|
// just ignore it
|
2006-12-16 03:19:53 +01:00
|
|
|
if (t->is_seed())
|
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2009-04-11 22:45:14 +02:00
|
|
|
TORRENT_ASSERT(m_received_in_piece == p.length);
|
2009-03-17 10:34:44 +01:00
|
|
|
m_received_in_piece = 0;
|
|
|
|
#endif
|
|
|
|
if (!m_download_queue.empty()) m_download_queue.erase(m_download_queue.begin());
|
2008-07-11 09:30:04 +02:00
|
|
|
t->add_redundant_bytes(p.length);
|
2006-12-16 03:19:53 +01:00
|
|
|
return;
|
|
|
|
}
|
2006-12-04 21:42:47 +01:00
|
|
|
|
2008-07-10 21:31:22 +02:00
|
|
|
ptime now = time_now();
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
piece_picker& picker = t->picker();
|
2006-12-11 13:48:33 +01:00
|
|
|
piece_manager& fs = t->filesystem();
|
2004-01-13 04:08:59 +01:00
|
|
|
|
2006-12-11 16:43:27 +01:00
|
|
|
std::vector<piece_block> finished_blocks;
|
2006-04-25 23:04:48 +02:00
|
|
|
piece_block block_finished(p.piece, p.start / t->block_size());
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p.start % t->block_size() == 0);
|
|
|
|
TORRENT_ASSERT(p.length == t->block_size()
|
2006-12-21 12:37:03 +01:00
|
|
|
|| p.length == t->torrent_file().total_size() % t->block_size());
|
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
std::vector<pending_block>::iterator b
|
2008-07-07 14:04:06 +02:00
|
|
|
= std::find_if(
|
2006-12-21 12:37:03 +01:00
|
|
|
m_download_queue.begin()
|
|
|
|
, m_download_queue.end()
|
2008-07-07 14:04:06 +02:00
|
|
|
, has_block(block_finished));
|
2003-12-08 22:59:48 +01:00
|
|
|
|
2008-03-14 11:17:27 +01:00
|
|
|
if (b == m_download_queue.end())
|
2006-12-21 12:37:03 +01:00
|
|
|
{
|
2008-07-09 19:40:03 +02:00
|
|
|
if (t->alerts().should_post<unwanted_block_alert>())
|
2006-11-14 01:08:16 +01:00
|
|
|
{
|
2008-07-08 20:41:04 +02:00
|
|
|
t->alerts().post_alert(unwanted_block_alert(t->get_handle(), m_remote
|
|
|
|
, m_peer_id, block_finished.block_index, block_finished.piece_index));
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-07-06 19:15:35 +02:00
|
|
|
(*m_logger) << " *** The block we just got was not in the "
|
|
|
|
"request queue ***\n";
|
2009-03-17 10:34:44 +01:00
|
|
|
#endif
|
|
|
|
#ifdef TORRENT_DEBUG
|
2009-04-11 22:45:14 +02:00
|
|
|
TORRENT_ASSERT(m_received_in_piece == p.length);
|
2009-03-17 10:34:44 +01:00
|
|
|
m_received_in_piece = 0;
|
2006-12-21 12:37:03 +01:00
|
|
|
#endif
|
2008-07-11 09:30:04 +02:00
|
|
|
t->add_redundant_bytes(p.length);
|
2007-07-06 19:15:35 +02:00
|
|
|
return;
|
2006-12-21 12:37:03 +01:00
|
|
|
}
|
2009-03-17 10:34:44 +01:00
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-09-25 22:12:53 +02:00
|
|
|
pending_block pending_b = *b;
|
|
|
|
#endif
|
2003-12-08 22:59:48 +01:00
|
|
|
|
2008-09-01 19:41:13 +02:00
|
|
|
int block_index = b - m_download_queue.begin() - 1;
|
2008-08-25 17:25:46 +02:00
|
|
|
for (int i = 0; i < block_index; ++i)
|
2008-03-14 11:17:27 +01:00
|
|
|
{
|
2008-08-25 17:25:46 +02:00
|
|
|
pending_block& qe = m_download_queue[i];
|
2008-07-07 14:04:06 +02:00
|
|
|
|
2008-03-14 11:17:27 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-07-07 14:04:06 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2008-08-25 23:10:13 +02:00
|
|
|
<< " *** SKIPPED_PIECE [ piece: " << qe.block.piece_index << " | "
|
|
|
|
"b: " << qe.block.block_index << " ] ***\n";
|
2008-03-14 11:17:27 +01:00
|
|
|
#endif
|
2008-07-07 14:04:06 +02:00
|
|
|
|
2008-08-25 17:25:46 +02:00
|
|
|
++qe.skipped;
|
2008-07-07 14:04:06 +02:00
|
|
|
// if the number of times a block is skipped by out of order
|
|
|
|
// blocks exceeds the size of the outstanding queue, assume that
|
|
|
|
// the other end dropped the request.
|
2009-03-17 10:34:44 +01:00
|
|
|
if (qe.skipped > m_desired_queue_size * 2)
|
2008-07-07 14:04:06 +02:00
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post<request_dropped_alert>())
|
|
|
|
m_ses.m_alerts.post_alert(request_dropped_alert(t->get_handle()
|
2008-08-25 17:25:46 +02:00
|
|
|
, remote(), pid(), qe.block.block_index, qe.block.piece_index));
|
2009-03-17 10:34:44 +01:00
|
|
|
|
|
|
|
if (!qe.timed_out && !qe.not_wanted)
|
|
|
|
picker.abort_download(qe.block);
|
|
|
|
|
2009-04-25 10:19:21 +02:00
|
|
|
TORRENT_ASSERT(m_download_queue[block_index] == pending_b);
|
2008-08-25 17:25:46 +02:00
|
|
|
m_download_queue.erase(m_download_queue.begin() + i);
|
|
|
|
--i;
|
|
|
|
--block_index;
|
2009-03-17 10:34:44 +01:00
|
|
|
m_outstanding_bytes -= t->block_size();
|
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_DEBUG
|
|
|
|
check_invariant();
|
|
|
|
#endif
|
2008-03-14 11:17:27 +01:00
|
|
|
}
|
|
|
|
}
|
2008-09-25 22:12:53 +02:00
|
|
|
TORRENT_ASSERT(int(m_download_queue.size()) > block_index + 1);
|
|
|
|
b = m_download_queue.begin() + (block_index + 1);
|
|
|
|
TORRENT_ASSERT(b->block == pending_b.block);
|
2008-07-07 14:04:06 +02:00
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
|
|
|
TORRENT_ASSERT(m_received_in_piece == p.length);
|
|
|
|
m_received_in_piece = 0;
|
|
|
|
#endif
|
2006-12-21 12:37:03 +01:00
|
|
|
// if the block we got is already finished, then ignore it
|
2007-06-10 22:46:09 +02:00
|
|
|
if (picker.is_downloaded(block_finished))
|
2006-12-21 12:37:03 +01:00
|
|
|
{
|
2008-07-11 09:30:04 +02:00
|
|
|
t->add_redundant_bytes(p.length);
|
2007-06-10 22:46:09 +02:00
|
|
|
|
2008-03-14 11:17:27 +01:00
|
|
|
m_download_queue.erase(b);
|
2008-07-07 14:04:06 +02:00
|
|
|
m_timeout_extend = 0;
|
2008-06-29 11:50:42 +02:00
|
|
|
|
|
|
|
if (!m_download_queue.empty())
|
2008-07-10 21:31:22 +02:00
|
|
|
m_requested = now;
|
2008-06-29 11:50:42 +02:00
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
request_a_block(*t, *this);
|
|
|
|
send_block_requests();
|
2006-12-21 12:37:03 +01:00
|
|
|
return;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2006-12-21 12:37:03 +01:00
|
|
|
|
2008-07-10 21:31:22 +02:00
|
|
|
if (total_seconds(now - m_requested)
|
2008-07-07 14:04:06 +02:00
|
|
|
< m_ses.settings().request_timeout
|
|
|
|
&& m_snubbed)
|
|
|
|
{
|
|
|
|
m_snubbed = false;
|
|
|
|
if (m_ses.m_alerts.should_post<peer_unsnubbed_alert>())
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(peer_unsnubbed_alert(t->get_handle()
|
|
|
|
, m_remote, m_peer_id));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
fs.async_write(p, data, bind(&peer_connection::on_disk_write_complete
|
|
|
|
, self(), _1, _2, p, t));
|
2007-08-01 07:22:34 +02:00
|
|
|
m_outstanding_writing_bytes += p.length;
|
2008-02-17 21:40:21 +01:00
|
|
|
TORRENT_ASSERT(m_channel_state[download_channel] == peer_info::bw_idle);
|
2008-03-14 11:17:27 +01:00
|
|
|
m_download_queue.erase(b);
|
2008-03-14 18:43:38 +01:00
|
|
|
|
2008-08-19 17:04:14 +02:00
|
|
|
if (m_outstanding_writing_bytes >= m_ses.settings().max_outstanding_disk_bytes_per_connection
|
|
|
|
&& t->alerts().should_post<performance_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(performance_alert(t->get_handle()
|
|
|
|
, performance_alert::outstanding_disk_buffer_limit_reached));
|
|
|
|
}
|
|
|
|
|
2008-06-29 11:50:42 +02:00
|
|
|
if (!m_download_queue.empty())
|
2008-07-10 21:31:22 +02:00
|
|
|
{
|
|
|
|
m_timeout_extend = (std::max)(m_timeout_extend
|
|
|
|
- m_ses.settings().request_timeout, 0);
|
|
|
|
m_requested += seconds(m_ses.settings().request_timeout);
|
|
|
|
if (m_requested > now) m_requested = now;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
m_timeout_extend = 0;
|
|
|
|
}
|
2008-06-29 11:50:42 +02:00
|
|
|
|
2008-03-14 18:43:38 +01:00
|
|
|
// did we request this block from any other peers?
|
|
|
|
bool multi = picker.num_peers(block_finished) > 1;
|
2007-07-04 04:16:49 +02:00
|
|
|
picker.mark_as_writing(block_finished, peer_info_struct());
|
2008-03-14 18:43:38 +01:00
|
|
|
|
2008-09-06 23:04:57 +02:00
|
|
|
TORRENT_ASSERT(picker.num_peers(block_finished) == 0);
|
2008-03-14 18:43:38 +01:00
|
|
|
// if we requested this block from other peers, cancel it now
|
|
|
|
if (multi) t->cancel_block(block_finished);
|
|
|
|
|
2008-09-06 23:04:57 +02:00
|
|
|
TORRENT_ASSERT(picker.num_peers(block_finished) == 0);
|
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#if defined TORRENT_DEBUG && !defined TORRENT_DISABLE_INVARIANT_CHECKS \
|
2008-10-10 07:25:55 +02:00
|
|
|
&& defined TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2008-01-02 04:18:29 +01:00
|
|
|
t->check_invariant();
|
|
|
|
#endif
|
2008-02-27 21:37:41 +01:00
|
|
|
request_a_block(*t, *this);
|
|
|
|
send_block_requests();
|
2007-06-10 22:46:09 +02:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
void peer_connection::on_disk_write_complete(int ret, disk_io_job const& j
|
|
|
|
, peer_request p, boost::shared_ptr<torrent> t)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-18 02:32:16 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-01 07:22:34 +02:00
|
|
|
m_outstanding_writing_bytes -= p.length;
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_outstanding_writing_bytes >= 0);
|
2007-08-01 07:22:34 +02:00
|
|
|
|
2007-10-18 02:32:16 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2008-01-07 02:10:46 +01:00
|
|
|
// (*m_ses.m_logger) << time_now_string() << " *** DISK_WRITE_COMPLETE [ p: "
|
|
|
|
// << p.piece << " o: " << p.start << " ]\n";
|
2007-08-03 10:19:10 +02:00
|
|
|
#endif
|
2007-08-01 07:22:34 +02:00
|
|
|
// in case the outstanding bytes just dropped down
|
|
|
|
// to allow to receive more data
|
|
|
|
setup_receive();
|
|
|
|
|
2007-10-18 02:32:16 +02:00
|
|
|
piece_block block_finished(p.piece, p.start / t->block_size());
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
if (ret == -1 || !t)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2008-02-18 04:07:14 +01:00
|
|
|
if (t->has_picker()) t->picker().write_failed(block_finished);
|
2007-10-18 02:32:16 +02:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
if (!t)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2008-01-07 02:10:46 +01:00
|
|
|
disconnect(j.str.c_str());
|
2007-06-10 22:46:09 +02:00
|
|
|
return;
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2007-06-10 22:46:09 +02:00
|
|
|
|
2008-07-06 14:22:56 +02:00
|
|
|
if (t->alerts().should_post<file_error_alert>())
|
2008-04-13 00:08:07 +02:00
|
|
|
t->alerts().post_alert(file_error_alert(j.error_file, t->get_handle(), j.str));
|
2009-02-23 02:21:19 +01:00
|
|
|
t->set_error(j.error, j.error_file);
|
2007-06-10 22:46:09 +02:00
|
|
|
t->pause();
|
|
|
|
return;
|
|
|
|
}
|
2006-12-21 23:20:28 +01:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
if (t->is_seed()) return;
|
2007-03-27 09:04:31 +02:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
piece_picker& picker = t->picker();
|
2004-01-18 11:22:18 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p.piece == j.piece);
|
|
|
|
TORRENT_ASSERT(p.start == j.offset);
|
2008-09-06 23:04:57 +02:00
|
|
|
TORRENT_ASSERT(picker.num_peers(block_finished) == 0);
|
2007-07-04 04:16:49 +02:00
|
|
|
picker.mark_as_finished(block_finished, peer_info_struct());
|
2008-07-06 14:22:56 +02:00
|
|
|
if (t->alerts().should_post<block_finished_alert>())
|
2007-08-01 08:14:16 +02:00
|
|
|
{
|
|
|
|
t->alerts().post_alert(block_finished_alert(t->get_handle(),
|
2008-07-08 11:30:10 +02:00
|
|
|
remote(), pid(), block_finished.block_index, block_finished.piece_index));
|
2007-08-01 08:14:16 +02:00
|
|
|
}
|
2007-03-27 09:04:31 +02:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
// did we just finish the piece?
|
|
|
|
if (picker.is_piece_finished(p.piece))
|
|
|
|
{
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2007-06-10 22:46:09 +02:00
|
|
|
check_postcondition post_checker2_(t, false);
|
|
|
|
#endif
|
|
|
|
t->async_verify_piece(p.piece, bind(&torrent::piece_finished, t
|
|
|
|
, p.piece, _1));
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2006-12-31 15:48:18 +01:00
|
|
|
|
2007-10-18 02:32:16 +02:00
|
|
|
if (!t->is_seed() && !m_torrent.expired())
|
|
|
|
{
|
|
|
|
// this is a free function defined in policy.cpp
|
|
|
|
request_a_block(*t, *this);
|
|
|
|
send_block_requests();
|
|
|
|
}
|
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
// -----------------------------
|
|
|
|
// ---------- CANCEL -----------
|
|
|
|
// -----------------------------
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_cancel(peer_request const& r)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-04-02 22:00:24 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_cancel(r)) return;
|
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-04-02 22:00:24 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-04-25 23:04:48 +02:00
|
|
|
<< " <== CANCEL [ piece: " << r.piece << " | s: " << r.start << " | l: " << r.length << " ]\n";
|
|
|
|
#endif
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
std::deque<peer_request>::iterator i
|
|
|
|
= std::find(m_requests.begin(), m_requests.end(), r);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2004-01-05 00:51:54 +01:00
|
|
|
if (i != m_requests.end())
|
|
|
|
{
|
|
|
|
m_requests.erase(i);
|
2008-01-02 04:18:29 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> REJECT_PIECE [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " ]\n";
|
|
|
|
#endif
|
|
|
|
write_reject_request(r);
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
else
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " *** GOT CANCEL NOT IN THE QUEUE\n";
|
2004-01-05 00:51:54 +01:00
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-11-05 00:27:06 +01:00
|
|
|
|
2005-09-15 00:45:22 +02:00
|
|
|
// -----------------------------
|
|
|
|
// --------- DHT PORT ----------
|
|
|
|
// -----------------------------
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::incoming_dht_port(int listen_port)
|
2005-09-15 00:45:22 +02:00
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2005-09-15 00:45:22 +02:00
|
|
|
<< " <== DHT_PORT [ p: " << listen_port << " ]\n";
|
2006-08-01 17:27:08 +02:00
|
|
|
#endif
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2006-10-11 16:02:21 +02:00
|
|
|
m_ses.add_dht_node(udp::endpoint(
|
2006-08-01 17:27:08 +02:00
|
|
|
m_remote.address(), listen_port));
|
2005-09-15 00:45:22 +02:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
// -----------------------------
|
|
|
|
// --------- HAVE ALL ----------
|
|
|
|
// -----------------------------
|
|
|
|
|
|
|
|
void peer_connection::incoming_have_all()
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " <== HAVE_ALL\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_have_all()) return;
|
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
m_have_all = true;
|
|
|
|
|
|
|
|
if (m_peer_info) m_peer_info->seed = true;
|
2008-07-18 12:03:42 +02:00
|
|
|
m_upload_only = true;
|
|
|
|
m_bitfield_received = true;
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << " *** THIS IS A SEED ***\n";
|
|
|
|
#endif
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
// if we don't have metadata yet
|
|
|
|
// just remember the bitmask
|
|
|
|
// don't update the piecepicker
|
|
|
|
// (since it doesn't exist yet)
|
|
|
|
if (!t->ready_for_connections())
|
|
|
|
{
|
2008-08-12 10:18:05 +02:00
|
|
|
// assume seeds are interesting when we
|
|
|
|
// don't even have the metadata
|
|
|
|
t->get_policy().peer_is_interesting(*this);
|
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
disconnect_if_redundant();
|
2007-08-14 19:47:48 +02:00
|
|
|
// TODO: this might need something more
|
|
|
|
// so that once we have the metadata
|
|
|
|
// we can construct a full bitfield
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_have_piece.empty());
|
2008-05-28 04:35:02 +02:00
|
|
|
m_have_piece.set_all();
|
2007-08-21 10:16:41 +02:00
|
|
|
m_num_pieces = m_have_piece.size();
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
t->peer_has_all();
|
2008-07-18 12:03:42 +02:00
|
|
|
|
|
|
|
// if we're finished, we're not interested
|
|
|
|
if (t->is_finished()) send_not_interested();
|
|
|
|
else t->get_policy().peer_is_interesting(*this);
|
|
|
|
|
|
|
|
disconnect_if_redundant();
|
2007-08-14 19:47:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// -----------------------------
|
|
|
|
// --------- HAVE NONE ---------
|
|
|
|
// -----------------------------
|
|
|
|
|
|
|
|
void peer_connection::incoming_have_none()
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " <== HAVE_NONE\n";
|
|
|
|
#endif
|
|
|
|
|
2008-05-12 05:05:27 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_have_none()) return;
|
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-08-14 19:47:48 +02:00
|
|
|
if (m_peer_info) m_peer_info->seed = false;
|
2008-07-18 12:03:42 +02:00
|
|
|
m_bitfield_received = true;
|
|
|
|
|
|
|
|
// we're never interested in a peer that doesn't have anything
|
|
|
|
send_not_interested();
|
2008-05-12 05:05:27 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_have_piece.empty() || !t->ready_for_connections());
|
2008-07-18 12:03:42 +02:00
|
|
|
disconnect_if_redundant();
|
2007-08-14 19:47:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// -----------------------------
|
|
|
|
// ------- ALLOWED FAST --------
|
|
|
|
// -----------------------------
|
|
|
|
|
|
|
|
void peer_connection::incoming_allowed_fast(int index)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " <== ALLOWED_FAST [ " << index << " ]\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if ((*i)->on_allowed_fast(index)) return;
|
|
|
|
}
|
|
|
|
#endif
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-08-14 19:47:48 +02:00
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
if (t->valid_metadata())
|
2007-10-01 03:12:00 +02:00
|
|
|
{
|
2008-08-29 19:21:56 +02:00
|
|
|
if (index < 0 || index >= int(m_have_piece.size()))
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-08-29 19:21:56 +02:00
|
|
|
(*m_logger) << time_now_string() << " <== INVALID_ALLOWED_FAST [ " << index << " | s: "
|
|
|
|
<< int(m_have_piece.size()) << " ]\n";
|
2007-10-01 03:12:00 +02:00
|
|
|
#endif
|
2008-08-29 19:21:56 +02:00
|
|
|
return;
|
|
|
|
}
|
2007-10-01 03:12:00 +02:00
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
// if we already have the piece, we can
|
|
|
|
// ignore this message
|
|
|
|
if (t->have_piece(index))
|
|
|
|
return;
|
|
|
|
}
|
2007-12-18 05:40:19 +01:00
|
|
|
|
2007-08-14 19:47:48 +02:00
|
|
|
m_allowed_fast.push_back(index);
|
|
|
|
|
|
|
|
// if the peer has the piece and we want
|
|
|
|
// to download it, request it
|
2007-08-16 14:41:46 +02:00
|
|
|
if (int(m_have_piece.size()) > index
|
2007-08-14 19:47:48 +02:00
|
|
|
&& m_have_piece[index]
|
2008-08-29 19:21:56 +02:00
|
|
|
&& t->valid_metadata()
|
2007-08-14 19:47:48 +02:00
|
|
|
&& t->has_picker()
|
|
|
|
&& t->picker().piece_priority(index) > 0)
|
|
|
|
{
|
|
|
|
t->get_policy().peer_is_interesting(*this);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<int> const& peer_connection::allowed_fast()
|
|
|
|
{
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-08-14 19:47:48 +02:00
|
|
|
|
2007-08-21 08:57:12 +02:00
|
|
|
m_allowed_fast.erase(std::remove_if(m_allowed_fast.begin()
|
|
|
|
, m_allowed_fast.end(), bind(&torrent::have_piece, t, _1))
|
|
|
|
, m_allowed_fast.end());
|
2007-08-14 19:47:48 +02:00
|
|
|
|
|
|
|
// TODO: sort the allowed fast set in priority order
|
|
|
|
return m_allowed_fast;
|
|
|
|
}
|
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
bool peer_connection::can_request_time_critical() const
|
|
|
|
{
|
|
|
|
if (has_peer_choked() || !is_interesting()) return false;
|
|
|
|
if (m_desired_queue_size * 2 <
|
|
|
|
- (int)m_download_queue.size()
|
|
|
|
- (int)m_request_queue.size()) return false;
|
|
|
|
if (on_parole()) return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::make_time_critical(piece_block const& block)
|
|
|
|
{
|
|
|
|
std::vector<piece_block>::iterator rit = std::find(m_request_queue.begin()
|
|
|
|
, m_request_queue.end(), block);
|
|
|
|
if (rit == m_request_queue.end()) return;
|
|
|
|
// ignore it if it's already time critical
|
|
|
|
if (rit - m_request_queue.begin() < m_queued_time_critical) return;
|
|
|
|
piece_block b = *rit;
|
|
|
|
m_request_queue.erase(rit);
|
|
|
|
m_request_queue.insert(m_request_queue.begin() + m_queued_time_critical, b);
|
|
|
|
++m_queued_time_critical;
|
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::add_request(piece_block const& block, bool time_critical)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2008-07-08 10:13:45 +02:00
|
|
|
// INVARIANT_CHECK;
|
2004-01-25 19:18:36 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2003-12-01 22:27:27 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t->valid_metadata());
|
|
|
|
TORRENT_ASSERT(block.piece_index >= 0);
|
|
|
|
TORRENT_ASSERT(block.piece_index < t->torrent_file().num_pieces());
|
|
|
|
TORRENT_ASSERT(block.block_index >= 0);
|
|
|
|
TORRENT_ASSERT(block.block_index < t->torrent_file().piece_size(block.piece_index));
|
|
|
|
TORRENT_ASSERT(!t->picker().is_requested(block) || (t->picker().num_peers(block) > 0));
|
|
|
|
TORRENT_ASSERT(!t->have_piece(block.piece_index));
|
2008-07-07 14:04:06 +02:00
|
|
|
TORRENT_ASSERT(std::find_if(m_download_queue.begin(), m_download_queue.end()
|
|
|
|
, has_block(block)) == m_download_queue.end());
|
|
|
|
TORRENT_ASSERT(std::find(m_request_queue.begin(), m_request_queue.end()
|
|
|
|
, block) == m_request_queue.end());
|
2004-01-05 00:51:54 +01:00
|
|
|
|
2007-04-27 02:27:37 +02:00
|
|
|
piece_picker::piece_state_t state;
|
|
|
|
peer_speed_t speed = peer_speed();
|
2007-09-06 20:33:15 +02:00
|
|
|
char const* speedmsg = 0;
|
2007-08-01 08:14:16 +02:00
|
|
|
if (speed == fast)
|
|
|
|
{
|
|
|
|
speedmsg = "fast";
|
|
|
|
state = piece_picker::fast;
|
|
|
|
}
|
|
|
|
else if (speed == medium)
|
|
|
|
{
|
|
|
|
speedmsg = "medium";
|
|
|
|
state = piece_picker::medium;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
speedmsg = "slow";
|
|
|
|
state = piece_picker::slow;
|
|
|
|
}
|
2007-04-27 02:27:37 +02:00
|
|
|
|
2007-09-15 22:20:07 +02:00
|
|
|
if (!t->picker().mark_as_downloading(block, peer_info_struct(), state))
|
|
|
|
return;
|
|
|
|
|
2008-07-06 14:22:56 +02:00
|
|
|
if (t->alerts().should_post<block_downloading_alert>())
|
2007-08-01 08:14:16 +02:00
|
|
|
{
|
|
|
|
t->alerts().post_alert(block_downloading_alert(t->get_handle(),
|
2008-07-08 11:30:10 +02:00
|
|
|
remote(), pid(), speedmsg, block.block_index, block.piece_index));
|
2007-08-01 08:14:16 +02:00
|
|
|
}
|
2007-04-27 02:27:37 +02:00
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
if (time_critical)
|
|
|
|
{
|
|
|
|
m_request_queue.insert(m_request_queue.begin() + m_queued_time_critical
|
|
|
|
, block);
|
|
|
|
++m_queued_time_critical;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
m_request_queue.push_back(block);
|
|
|
|
}
|
2004-01-05 00:51:54 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::cancel_request(piece_block const& block)
|
2004-01-05 00:51:54 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2008-01-16 22:07:04 +01:00
|
|
|
// this peer might be disconnecting
|
|
|
|
if (!t) return;
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t->valid_metadata());
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(block.piece_index >= 0);
|
|
|
|
TORRENT_ASSERT(block.piece_index < t->torrent_file().num_pieces());
|
|
|
|
TORRENT_ASSERT(block.block_index >= 0);
|
|
|
|
TORRENT_ASSERT(block.block_index < t->torrent_file().piece_size(block.piece_index));
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2007-07-06 19:15:35 +02:00
|
|
|
// if all the peers that requested this block has been
|
|
|
|
// cancelled, then just ignore the cancel.
|
|
|
|
if (!t->picker().is_requested(block)) return;
|
2004-01-07 01:48:02 +01:00
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
std::vector<pending_block>::iterator it
|
2008-07-07 14:04:06 +02:00
|
|
|
= std::find_if(m_download_queue.begin(), m_download_queue.end(), has_block(block));
|
2006-04-25 23:04:48 +02:00
|
|
|
if (it == m_download_queue.end())
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
std::vector<piece_block>::iterator rit = std::find(m_request_queue.begin()
|
2008-07-07 14:04:06 +02:00
|
|
|
, m_request_queue.end(), block);
|
|
|
|
|
2007-07-06 19:15:35 +02:00
|
|
|
// when a multi block is received, it is cancelled
|
|
|
|
// from all peers, so if this one hasn't requested
|
|
|
|
// the block, just ignore to cancel it.
|
2008-07-07 14:04:06 +02:00
|
|
|
if (rit == m_request_queue.end()) return;
|
2007-07-06 19:15:35 +02:00
|
|
|
|
|
|
|
t->picker().abort_download(block);
|
2008-07-07 14:04:06 +02:00
|
|
|
m_request_queue.erase(rit);
|
2006-04-25 23:04:48 +02:00
|
|
|
// since we found it in the request queue, it means it hasn't been
|
|
|
|
// sent yet, so we don't have to send a cancel.
|
|
|
|
return;
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
int block_offset = block.block_index * t->block_size();
|
|
|
|
int block_size
|
2008-04-05 23:18:27 +02:00
|
|
|
= (std::min)(t->torrent_file().piece_size(block.piece_index)-block_offset,
|
2006-04-25 23:04:48 +02:00
|
|
|
t->block_size());
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(block_size > 0);
|
|
|
|
TORRENT_ASSERT(block_size <= t->block_size());
|
2004-01-08 14:03:38 +01:00
|
|
|
|
|
|
|
peer_request r;
|
|
|
|
r.piece = block.piece_index;
|
|
|
|
r.start = block_offset;
|
|
|
|
r.length = block_size;
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-04-25 23:04:48 +02:00
|
|
|
<< " ==> CANCEL [ piece: " << block.piece_index << " | s: "
|
|
|
|
<< block_offset << " | l: " << block_size << " | " << block.block_index << " ]\n";
|
2005-03-19 13:22:40 +01:00
|
|
|
#endif
|
2008-01-02 04:18:29 +01:00
|
|
|
write_cancel(r);
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
|
|
|
|
2004-01-12 21:31:27 +01:00
|
|
|
void peer_connection::send_choke()
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_peer_info || !m_peer_info->optimistically_unchoked);
|
2007-08-21 19:45:28 +02:00
|
|
|
|
2004-01-04 05:29:13 +01:00
|
|
|
if (m_choked) return;
|
2006-04-25 23:04:48 +02:00
|
|
|
write_choke();
|
2004-01-04 05:29:13 +01:00
|
|
|
m_choked = true;
|
2005-03-19 13:22:40 +01:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " ==> CHOKE\n";
|
2005-08-19 01:55:32 +02:00
|
|
|
#endif
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_choke = time_now();
|
2005-03-19 13:22:40 +01:00
|
|
|
#endif
|
2004-01-12 21:31:27 +01:00
|
|
|
m_num_invalid_requests = 0;
|
2007-10-06 00:45:24 +02:00
|
|
|
|
|
|
|
// reject the requests we have in the queue
|
2008-07-09 19:45:37 +02:00
|
|
|
// except the allowed fast pieces
|
|
|
|
for (std::deque<peer_request>::iterator i = m_requests.begin();
|
|
|
|
i != m_requests.end();)
|
2008-01-02 04:18:29 +01:00
|
|
|
{
|
2008-07-09 19:45:37 +02:00
|
|
|
if (m_accept_fast.count(i->piece))
|
|
|
|
{
|
|
|
|
++i;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2008-01-02 04:18:29 +01:00
|
|
|
peer_request const& r = *i;
|
2008-07-09 19:45:37 +02:00
|
|
|
write_reject_request(r);
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2008-01-02 04:18:29 +01:00
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> REJECT_PIECE [ "
|
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " ]\n";
|
|
|
|
#endif
|
2008-07-09 19:45:37 +02:00
|
|
|
i = m_requests.erase(i);
|
|
|
|
}
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
bool peer_connection::send_unchoke()
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
if (!m_choked) return false;
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
if (!t->ready_for_connections()) return false;
|
2007-09-01 09:38:10 +02:00
|
|
|
m_last_unchoke = time_now();
|
2006-04-25 23:04:48 +02:00
|
|
|
write_unchoke();
|
2004-01-04 05:29:13 +01:00
|
|
|
m_choked = false;
|
2005-03-19 13:22:40 +01:00
|
|
|
|
2008-12-11 00:07:44 +01:00
|
|
|
m_uploaded_at_last_unchoke = m_statistics.total_payload_upload();
|
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " ==> UNCHOKE\n";
|
2005-03-19 13:22:40 +01:00
|
|
|
#endif
|
2008-08-29 19:21:56 +02:00
|
|
|
return true;
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
|
|
|
|
2004-01-12 21:31:27 +01:00
|
|
|
void peer_connection::send_interested()
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
|
|
|
if (m_interesting) return;
|
2008-08-17 00:07:15 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2008-08-29 19:21:56 +02:00
|
|
|
if (!t->ready_for_connections()) return;
|
|
|
|
m_interesting = true;
|
2008-07-18 12:03:42 +02:00
|
|
|
write_interested();
|
2005-03-19 13:22:40 +01:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " ==> INTERESTED\n";
|
2005-03-19 13:22:40 +01:00
|
|
|
#endif
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
|
|
|
|
2004-01-12 21:31:27 +01:00
|
|
|
void peer_connection::send_not_interested()
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
|
|
|
if (!m_interesting) return;
|
2008-08-17 00:07:15 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2008-08-29 19:21:56 +02:00
|
|
|
if (!t->ready_for_connections()) return;
|
|
|
|
m_interesting = false;
|
2008-07-18 12:03:42 +02:00
|
|
|
write_not_interested();
|
2004-02-01 14:48:30 +01:00
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
m_became_uninteresting = time_now();
|
2004-02-01 14:48:30 +01:00
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << " ==> NOT_INTERESTED\n";
|
2005-03-19 13:22:40 +01:00
|
|
|
#endif
|
2008-07-18 12:03:42 +02:00
|
|
|
disconnect_if_redundant();
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::send_block_requests()
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
2007-01-01 12:54:31 +01:00
|
|
|
INVARIANT_CHECK;
|
2006-12-27 16:14:17 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2006-12-27 16:14:17 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if ((int)m_download_queue.size() >= m_desired_queue_size) return;
|
|
|
|
|
2008-06-29 11:50:42 +02:00
|
|
|
bool empty_download_queue = m_download_queue.empty();
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
while (!m_request_queue.empty()
|
2009-03-17 10:34:44 +01:00
|
|
|
&& ((int)m_download_queue.size() < m_desired_queue_size
|
|
|
|
|| m_queued_time_critical > 0))
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
|
|
|
piece_block block = m_request_queue.front();
|
|
|
|
|
|
|
|
int block_offset = block.block_index * t->block_size();
|
2008-04-05 23:18:27 +02:00
|
|
|
int block_size = (std::min)(t->torrent_file().piece_size(
|
2006-04-25 23:04:48 +02:00
|
|
|
block.piece_index) - block_offset, t->block_size());
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(block_size > 0);
|
|
|
|
TORRENT_ASSERT(block_size <= t->block_size());
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
peer_request r;
|
|
|
|
r.piece = block.piece_index;
|
|
|
|
r.start = block_offset;
|
|
|
|
r.length = block_size;
|
|
|
|
|
2009-03-17 10:34:44 +01:00
|
|
|
m_request_queue.erase(m_request_queue.begin());
|
|
|
|
if (m_queued_time_critical) --m_queued_time_critical;
|
2008-07-09 20:01:00 +02:00
|
|
|
if (t->is_seed()) continue;
|
|
|
|
// this can happen if a block times out, is re-requested and
|
|
|
|
// then arrives "unexpectedly"
|
|
|
|
if (t->picker().is_finished(block) || t->picker().is_downloaded(block))
|
|
|
|
continue;
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
m_download_queue.push_back(block);
|
2009-03-17 10:34:44 +01:00
|
|
|
m_outstanding_bytes += block_size;
|
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_DEBUG
|
|
|
|
check_invariant();
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
/*
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-11-14 01:08:16 +01:00
|
|
|
<< " *** REQUEST-QUEUE** [ "
|
|
|
|
"piece: " << block.piece_index << " | "
|
|
|
|
"block: " << block.block_index << " ]\n";
|
|
|
|
#endif
|
|
|
|
*/
|
|
|
|
// if we are requesting large blocks, merge the smaller
|
|
|
|
// blocks that are in the same piece into larger requests
|
|
|
|
if (m_request_large_blocks)
|
|
|
|
{
|
2007-09-03 23:16:24 +02:00
|
|
|
int blocks_per_piece = t->torrent_file().piece_length() / t->block_size();
|
|
|
|
|
|
|
|
while (!m_request_queue.empty())
|
2006-11-14 01:08:16 +01:00
|
|
|
{
|
2007-09-03 23:16:24 +02:00
|
|
|
// check to see if this block is connected to the previous one
|
|
|
|
// if it is, merge them, otherwise, break this merge loop
|
|
|
|
piece_block const& front = m_request_queue.front();
|
|
|
|
if (front.piece_index * blocks_per_piece + front.block_index
|
|
|
|
!= block.piece_index * blocks_per_piece + block.block_index + 1)
|
|
|
|
break;
|
2006-11-14 01:08:16 +01:00
|
|
|
block = m_request_queue.front();
|
2009-03-17 10:34:44 +01:00
|
|
|
m_request_queue.erase(m_request_queue.begin());
|
2006-11-14 01:08:16 +01:00
|
|
|
m_download_queue.push_back(block);
|
2009-03-17 10:34:44 +01:00
|
|
|
if (m_queued_time_critical) --m_queued_time_critical;
|
2007-09-03 23:16:24 +02:00
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2007-09-03 23:16:24 +02:00
|
|
|
<< " *** MERGING REQUEST ** [ "
|
2006-11-14 01:08:16 +01:00
|
|
|
"piece: " << block.piece_index << " | "
|
|
|
|
"block: " << block.block_index << " ]\n";
|
|
|
|
#endif
|
2007-09-03 23:16:24 +02:00
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
block_offset = block.block_index * t->block_size();
|
2008-04-05 23:18:27 +02:00
|
|
|
block_size = (std::min)(t->torrent_file().piece_size(
|
2006-11-14 01:08:16 +01:00
|
|
|
block.piece_index) - block_offset, t->block_size());
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(block_size > 0);
|
|
|
|
TORRENT_ASSERT(block_size <= t->block_size());
|
2006-11-14 01:08:16 +01:00
|
|
|
|
|
|
|
r.length += block_size;
|
2009-03-17 10:34:44 +01:00
|
|
|
m_outstanding_bytes += block_size;
|
|
|
|
#if !defined TORRENT_DISABLE_INVARIANT_CHECKS && defined TORRENT_DEBUG
|
|
|
|
check_invariant();
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(verify_piece(r));
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
bool handled = false;
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
2009-04-04 18:19:19 +02:00
|
|
|
if ((handled = (*i)->write_request(r))) break;
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2007-05-25 21:42:10 +02:00
|
|
|
if (!handled)
|
2009-04-04 18:19:19 +02:00
|
|
|
#endif
|
2007-05-25 21:42:10 +02:00
|
|
|
{
|
|
|
|
write_request(r);
|
|
|
|
m_last_request = time_now();
|
|
|
|
}
|
2006-11-14 01:08:16 +01:00
|
|
|
|
2005-03-19 13:22:40 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-04-25 23:04:48 +02:00
|
|
|
<< " ==> REQUEST [ "
|
2006-11-14 01:08:16 +01:00
|
|
|
"piece: " << r.piece << " | "
|
|
|
|
"s: " << r.start << " | "
|
|
|
|
"l: " << r.length << " | "
|
|
|
|
"ds: " << statistics().download_rate() << " B/s | "
|
2008-09-04 16:39:02 +02:00
|
|
|
"qs: " << int(m_desired_queue_size) << " "
|
2007-10-18 06:18:09 +02:00
|
|
|
"blk: " << (m_request_large_blocks?"large":"single") << " ]\n";
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
|
|
|
}
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_piece = time_now();
|
2008-06-29 11:50:42 +02:00
|
|
|
|
|
|
|
if (!m_download_queue.empty()
|
|
|
|
&& empty_download_queue)
|
|
|
|
{
|
|
|
|
// This means we just added a request to this connection
|
|
|
|
m_requested = time_now();
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
void peer_connection::timed_out()
|
|
|
|
{
|
2008-01-07 05:47:20 +01:00
|
|
|
TORRENT_ASSERT(m_connecting);
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-10-23 06:10:23 +02:00
|
|
|
error_code ec;
|
|
|
|
(*m_ses.m_logger) << time_now_string() << " CONNECTION TIMED OUT: " << m_remote.address().to_string(ec)
|
2007-05-05 02:29:33 +02:00
|
|
|
<< "\n";
|
|
|
|
#endif
|
2008-05-12 08:03:31 +02:00
|
|
|
disconnect("timed out: connect", 1);
|
2007-05-05 02:29:33 +02:00
|
|
|
}
|
|
|
|
|
2008-05-12 08:03:31 +02:00
|
|
|
// the error argument defaults to 0, which means deliberate disconnect
|
|
|
|
// 1 means unexpected disconnect/error
|
|
|
|
// 2 protocol error (client sent something invalid)
|
|
|
|
void peer_connection::disconnect(char const* message, int error)
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
2007-06-10 22:46:09 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-07-18 12:03:42 +02:00
|
|
|
m_disconnect_started = true;
|
|
|
|
#endif
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-05-12 08:03:31 +02:00
|
|
|
switch (error)
|
|
|
|
{
|
|
|
|
case 0:
|
|
|
|
(*m_logger) << "*** CONNECTION CLOSED " << message << "\n";
|
|
|
|
break;
|
|
|
|
case 1:
|
|
|
|
(*m_logger) << "*** CONNECTION FAILED " << message << "\n";
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
(*m_logger) << "*** PEER ERROR " << message << "\n";
|
|
|
|
break;
|
|
|
|
}
|
2008-01-07 02:10:46 +01:00
|
|
|
#endif
|
2008-04-09 06:09:40 +02:00
|
|
|
// we cannot do this in a constructor
|
|
|
|
TORRENT_ASSERT(m_in_constructor == false);
|
2008-05-12 08:03:31 +02:00
|
|
|
if (error > 0) m_failed = true;
|
|
|
|
if (m_disconnecting) return;
|
2008-04-09 06:09:40 +02:00
|
|
|
boost::intrusive_ptr<peer_connection> me(this);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-01-19 20:00:54 +01:00
|
|
|
if (m_connecting && m_connection_ticket >= 0)
|
|
|
|
{
|
2007-05-05 02:29:33 +02:00
|
|
|
m_ses.m_half_open.done(m_connection_ticket);
|
2008-01-19 20:00:54 +01:00
|
|
|
m_connection_ticket = -1;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2008-07-06 14:22:56 +02:00
|
|
|
torrent_handle handle;
|
|
|
|
if (t) handle = t->get_handle();
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-05-12 08:03:31 +02:00
|
|
|
if (message)
|
2008-01-07 02:10:46 +01:00
|
|
|
{
|
2008-07-06 14:22:56 +02:00
|
|
|
if (error > 1 && m_ses.m_alerts.should_post<peer_error_alert>())
|
2008-05-12 08:03:31 +02:00
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(
|
2008-07-06 14:22:56 +02:00
|
|
|
peer_error_alert(handle, remote(), pid(), message));
|
2008-05-12 08:03:31 +02:00
|
|
|
}
|
2008-07-06 14:22:56 +02:00
|
|
|
else if (error <= 1 && m_ses.m_alerts.should_post<peer_disconnected_alert>())
|
2008-05-12 08:03:31 +02:00
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(
|
2008-07-06 14:22:56 +02:00
|
|
|
peer_disconnected_alert(handle, remote(), pid(), message));
|
2008-05-12 08:03:31 +02:00
|
|
|
}
|
2008-01-07 02:10:46 +01:00
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (t)
|
|
|
|
{
|
2008-06-03 07:21:00 +02:00
|
|
|
// make sure we keep all the stats!
|
|
|
|
t->add_stats(statistics());
|
|
|
|
|
2007-04-12 12:21:55 +02:00
|
|
|
if (t->has_picker())
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
|
|
|
piece_picker& picker = t->picker();
|
2007-04-12 12:21:55 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
while (!m_download_queue.empty())
|
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
pending_block& qe = m_download_queue.back();
|
|
|
|
if (!qe.timed_out && !qe.not_wanted) picker.abort_download(qe.block);
|
2006-04-25 23:04:48 +02:00
|
|
|
m_download_queue.pop_back();
|
|
|
|
}
|
|
|
|
while (!m_request_queue.empty())
|
|
|
|
{
|
|
|
|
picker.abort_download(m_request_queue.back());
|
|
|
|
m_request_queue.pop_back();
|
|
|
|
}
|
|
|
|
}
|
2006-12-11 16:43:27 +01:00
|
|
|
|
|
|
|
t->remove_peer(this);
|
2006-04-25 23:04:48 +02:00
|
|
|
m_torrent.reset();
|
|
|
|
}
|
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#if defined TORRENT_DEBUG && defined TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2008-01-31 18:52:29 +01:00
|
|
|
// since this connection doesn't have a torrent reference
|
|
|
|
// no torrent should have a reference to this connection either
|
|
|
|
for (aux::session_impl::torrent_map::const_iterator i = m_ses.m_torrents.begin()
|
|
|
|
, end(m_ses.m_torrents.end()); i != end; ++i)
|
|
|
|
TORRENT_ASSERT(!i->second->has_peer(this));
|
|
|
|
#endif
|
|
|
|
|
2008-01-19 20:00:54 +01:00
|
|
|
m_disconnecting = true;
|
2008-05-03 18:05:42 +02:00
|
|
|
error_code ec;
|
2008-04-09 06:09:40 +02:00
|
|
|
m_socket->close(ec);
|
2008-01-07 02:10:46 +01:00
|
|
|
m_ses.close_connection(this, message);
|
2009-01-25 03:45:31 +01:00
|
|
|
|
|
|
|
// we should only disconnect while we still have
|
|
|
|
// at least one reference left to the connection
|
|
|
|
TORRENT_ASSERT(refcount() > 0);
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::set_upload_limit(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2009-04-26 02:21:59 +02:00
|
|
|
if (limit < 0) limit = 0;
|
|
|
|
if (limit < 10 && limit > 0) limit = 10;
|
2007-04-03 01:10:11 +02:00
|
|
|
m_upload_limit = limit;
|
2009-04-26 02:21:59 +02:00
|
|
|
m_bandwidth_channel[upload_channel].throttle(m_upload_limit);
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void peer_connection::set_download_limit(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2009-04-26 02:21:59 +02:00
|
|
|
if (limit < 0) limit = 0;
|
|
|
|
if (limit < 10 && limit > 0) limit = 10;
|
2007-04-03 01:10:11 +02:00
|
|
|
m_download_limit = limit;
|
2009-04-26 02:21:59 +02:00
|
|
|
m_bandwidth_channel[download_channel].throttle(m_download_limit);
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-25 19:18:36 +01:00
|
|
|
size_type peer_connection::share_diff() const
|
2004-01-12 04:05:10 +01:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
float ratio = t->ratio();
|
2004-01-12 04:05:10 +01:00
|
|
|
|
|
|
|
// if we have an infinite ratio, just say we have downloaded
|
|
|
|
// much more than we have uploaded. And we'll keep uploading.
|
2004-11-18 23:33:50 +01:00
|
|
|
if (ratio == 0.f)
|
2007-08-21 06:46:17 +02:00
|
|
|
return (std::numeric_limits<size_type>::max)();
|
2004-01-12 04:05:10 +01:00
|
|
|
|
|
|
|
return m_free_upload
|
2004-01-25 19:18:36 +01:00
|
|
|
+ static_cast<size_type>(m_statistics.total_payload_download() * ratio)
|
2004-01-12 04:05:10 +01:00
|
|
|
- m_statistics.total_payload_upload();
|
|
|
|
}
|
|
|
|
|
2009-03-01 08:20:08 +01:00
|
|
|
bool peer_connection::ignore_unchoke_slots() const
|
|
|
|
{
|
|
|
|
return m_ignore_unchoke_slots
|
|
|
|
|| (m_ses.settings().ignore_limits_on_local_network && on_local_network());
|
|
|
|
}
|
|
|
|
|
2007-05-25 19:06:30 +02:00
|
|
|
// defined in upnp.cpp
|
|
|
|
bool is_local(address const& a);
|
|
|
|
|
|
|
|
bool peer_connection::on_local_network() const
|
|
|
|
{
|
2008-01-11 06:40:19 +01:00
|
|
|
if (libtorrent::is_local(m_remote.address())
|
|
|
|
|| is_loopback(m_remote.address())) return true;
|
2007-05-25 19:06:30 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2007-05-25 21:42:10 +02:00
|
|
|
void peer_connection::get_peer_info(peer_info& p) const
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!associated_torrent().expired());
|
2007-05-25 21:42:10 +02:00
|
|
|
|
2008-06-29 11:50:42 +02:00
|
|
|
ptime now = time_now();
|
|
|
|
|
2008-04-03 08:11:21 +02:00
|
|
|
p.download_rate_peak = m_download_rate_peak;
|
|
|
|
p.upload_rate_peak = m_upload_rate_peak;
|
2008-02-09 23:42:56 +01:00
|
|
|
p.rtt = m_rtt;
|
2007-05-25 21:42:10 +02:00
|
|
|
p.down_speed = statistics().download_rate();
|
|
|
|
p.up_speed = statistics().upload_rate();
|
|
|
|
p.payload_down_speed = statistics().download_payload_rate();
|
|
|
|
p.payload_up_speed = statistics().upload_payload_rate();
|
|
|
|
p.pid = pid();
|
|
|
|
p.ip = remote();
|
2007-08-01 08:11:11 +02:00
|
|
|
p.pending_disk_bytes = m_outstanding_writing_bytes;
|
2009-04-26 02:21:59 +02:00
|
|
|
p.send_quota = m_quota[upload_channel];
|
|
|
|
p.receive_quota = m_quota[download_channel];
|
2008-10-24 02:15:39 +02:00
|
|
|
p.num_pieces = m_num_pieces;
|
2008-06-29 11:50:42 +02:00
|
|
|
if (m_download_queue.empty()) p.request_timeout = -1;
|
2008-07-07 14:04:06 +02:00
|
|
|
else p.request_timeout = total_seconds(m_requested - now) + m_ses.settings().request_timeout
|
|
|
|
+ m_timeout_extend;
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
p.inet_as_name = m_inet_as_name;
|
|
|
|
#endif
|
2009-02-17 03:33:45 +01:00
|
|
|
|
|
|
|
p.download_queue_time = download_queue_time();
|
2009-03-17 10:34:44 +01:00
|
|
|
p.queue_bytes = m_outstanding_bytes;
|
2007-05-25 21:42:10 +02:00
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
|
|
|
p.country[0] = m_country[0];
|
|
|
|
p.country[1] = m_country[1];
|
|
|
|
#endif
|
|
|
|
|
|
|
|
p.total_download = statistics().total_payload_download();
|
|
|
|
p.total_upload = statistics().total_payload_upload();
|
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
if (m_bandwidth_channel[upload_channel].throttle() == 0)
|
2007-05-25 21:42:10 +02:00
|
|
|
p.upload_limit = -1;
|
|
|
|
else
|
2009-04-26 02:21:59 +02:00
|
|
|
p.upload_limit = m_bandwidth_channel[upload_channel].throttle();
|
2007-05-25 21:42:10 +02:00
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
if (m_bandwidth_channel[download_channel].throttle() == 0)
|
2007-05-25 21:42:10 +02:00
|
|
|
p.download_limit = -1;
|
|
|
|
else
|
2009-04-26 02:21:59 +02:00
|
|
|
p.download_limit = m_bandwidth_channel[download_channel].throttle();
|
2007-05-25 21:42:10 +02:00
|
|
|
|
|
|
|
p.load_balancing = total_free_upload();
|
|
|
|
|
2007-09-10 10:07:18 +02:00
|
|
|
p.download_queue_length = int(download_queue().size() + m_request_queue.size());
|
2008-07-08 02:03:08 +02:00
|
|
|
p.requests_in_buffer = int(m_requests_in_buffer.size());
|
2007-09-10 10:07:18 +02:00
|
|
|
p.target_dl_queue_length = int(desired_queue_size());
|
|
|
|
p.upload_queue_length = int(upload_queue().size());
|
2007-05-25 21:42:10 +02:00
|
|
|
|
|
|
|
if (boost::optional<piece_block_progress> ret = downloading_piece_progress())
|
|
|
|
{
|
|
|
|
p.downloading_piece_index = ret->piece_index;
|
|
|
|
p.downloading_block_index = ret->block_index;
|
|
|
|
p.downloading_progress = ret->bytes_downloaded;
|
|
|
|
p.downloading_total = ret->full_block_bytes;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
p.downloading_piece_index = -1;
|
|
|
|
p.downloading_block_index = -1;
|
|
|
|
p.downloading_progress = 0;
|
|
|
|
p.downloading_total = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
p.pieces = get_bitfield();
|
|
|
|
p.last_request = now - m_last_request;
|
2007-08-17 00:13:35 +02:00
|
|
|
p.last_active = now - (std::max)(m_last_sent, m_last_receive);
|
2007-05-25 21:42:10 +02:00
|
|
|
|
|
|
|
// this will set the flags so that we can update them later
|
|
|
|
p.flags = 0;
|
|
|
|
get_specific_peer_info(p);
|
|
|
|
|
|
|
|
p.flags |= is_seed() ? peer_info::seed : 0;
|
2008-06-29 11:50:42 +02:00
|
|
|
p.flags |= m_snubbed ? peer_info::snubbed : 0;
|
2008-07-19 09:57:43 +02:00
|
|
|
p.flags |= m_upload_only ? peer_info::upload_only : 0;
|
2007-05-25 21:42:10 +02:00
|
|
|
if (peer_info_struct())
|
|
|
|
{
|
2008-04-05 06:53:22 +02:00
|
|
|
policy::peer* pi = peer_info_struct();
|
|
|
|
p.source = pi->source;
|
|
|
|
p.failcount = pi->failcount;
|
|
|
|
p.num_hashfails = pi->hashfails;
|
|
|
|
p.flags |= pi->on_parole ? peer_info::on_parole : 0;
|
|
|
|
p.flags |= pi->optimistically_unchoked ? peer_info::optimistic_unchoke : 0;
|
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
p.inet_as = pi->inet_as->first;
|
|
|
|
#endif
|
2007-05-25 21:42:10 +02:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
p.source = 0;
|
|
|
|
p.failcount = 0;
|
|
|
|
p.num_hashfails = 0;
|
2007-06-07 02:05:18 +02:00
|
|
|
p.remote_dl_rate = 0;
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
p.inet_as = 0xffff;
|
|
|
|
#endif
|
2007-05-25 21:42:10 +02:00
|
|
|
}
|
|
|
|
|
2008-04-05 06:53:22 +02:00
|
|
|
p.remote_dl_rate = m_remote_dl_rate;
|
2007-09-29 18:14:03 +02:00
|
|
|
p.send_buffer_size = m_send_buffer.capacity();
|
2008-01-10 23:13:23 +01:00
|
|
|
p.used_send_buffer = m_send_buffer.size();
|
2008-04-10 12:03:23 +02:00
|
|
|
p.receive_buffer_size = m_recv_buffer.capacity() + m_disk_recv_buffer_size;
|
|
|
|
p.used_receive_buffer = m_recv_pos;
|
2008-01-14 00:46:43 +01:00
|
|
|
p.write_state = m_channel_state[upload_channel];
|
|
|
|
p.read_state = m_channel_state[download_channel];
|
2008-09-25 22:39:06 +02:00
|
|
|
|
|
|
|
p.progress = (float)p.pieces.count() / (float)p.pieces.size();
|
2007-05-25 21:42:10 +02:00
|
|
|
}
|
2007-05-25 19:06:30 +02:00
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
// allocates a disk buffer of size 'disk_buffer_size' and replaces the
|
|
|
|
// end of the current receive buffer with it. i.e. the receive pos
|
|
|
|
// must be <= packet_size - disk_buffer_size
|
|
|
|
// the disk buffer can be accessed through release_disk_receive_buffer()
|
|
|
|
// when it is queried, the responsibility to free it is transferred
|
|
|
|
// to the caller
|
|
|
|
bool peer_connection::allocate_disk_receive_buffer(int disk_buffer_size)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
TORRENT_ASSERT(m_packet_size > 0);
|
|
|
|
TORRENT_ASSERT(m_recv_pos <= m_packet_size - disk_buffer_size);
|
2008-05-05 08:25:22 +02:00
|
|
|
TORRENT_ASSERT(!m_disk_recv_buffer);
|
2008-04-10 12:03:23 +02:00
|
|
|
TORRENT_ASSERT(disk_buffer_size <= 16 * 1024);
|
|
|
|
|
2008-10-06 05:23:47 +02:00
|
|
|
if (disk_buffer_size == 0) return true;
|
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
if (disk_buffer_size > 16 * 1024)
|
|
|
|
{
|
2008-05-12 08:03:31 +02:00
|
|
|
disconnect("invalid piece size", 2);
|
2008-04-10 12:03:23 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2009-01-23 10:13:31 +01:00
|
|
|
m_disk_recv_buffer.reset(m_ses.allocate_disk_buffer("receive buffer"));
|
2008-05-05 08:25:22 +02:00
|
|
|
if (!m_disk_recv_buffer)
|
2008-04-10 12:03:23 +02:00
|
|
|
{
|
|
|
|
disconnect("out of memory");
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
m_disk_recv_buffer_size = disk_buffer_size;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
char* peer_connection::release_disk_receive_buffer()
|
|
|
|
{
|
|
|
|
m_disk_recv_buffer_size = 0;
|
2008-05-05 08:25:22 +02:00
|
|
|
return m_disk_recv_buffer.release();
|
2008-04-10 12:03:23 +02:00
|
|
|
}
|
|
|
|
|
2009-04-21 21:27:52 +02:00
|
|
|
// size = the packet size to remove from the receive buffer
|
|
|
|
// packet_size = the next packet size to receive in the buffer
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::cut_receive_buffer(int size, int packet_size)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(packet_size > 0);
|
|
|
|
TORRENT_ASSERT(int(m_recv_buffer.size()) >= size);
|
|
|
|
TORRENT_ASSERT(int(m_recv_buffer.size()) >= m_recv_pos);
|
|
|
|
TORRENT_ASSERT(m_recv_pos >= size);
|
2007-06-06 02:41:20 +02:00
|
|
|
|
|
|
|
if (size > 0)
|
|
|
|
std::memmove(&m_recv_buffer[0], &m_recv_buffer[0] + size, m_recv_pos - size);
|
2007-05-11 20:40:22 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
m_recv_pos -= size;
|
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-04-25 23:04:48 +02:00
|
|
|
std::fill(m_recv_buffer.begin() + m_recv_pos, m_recv_buffer.end(), 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
m_packet_size = packet_size;
|
|
|
|
}
|
|
|
|
|
2008-12-08 07:36:22 +01:00
|
|
|
void peer_connection::superseed_piece(int index)
|
|
|
|
{
|
|
|
|
if (index == -1)
|
|
|
|
{
|
|
|
|
if (m_superseed_piece == -1) return;
|
|
|
|
m_superseed_piece = -1;
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " *** ending super seed mode\n";
|
|
|
|
#endif
|
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
assert(t);
|
|
|
|
|
|
|
|
for (int i = 0; i < m_have_piece.size(); ++i)
|
|
|
|
{
|
|
|
|
if (m_have_piece[i] || !t->have_piece(i)) continue;
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << " ==> HAVE [ piece: " << i << "] (ending super seed)\n";
|
|
|
|
#endif
|
|
|
|
write_have(i);
|
|
|
|
}
|
|
|
|
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(!has_piece(index));
|
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> HAVE [ piece: " << index << "] (super seed)\n";
|
|
|
|
#endif
|
|
|
|
write_have(index);
|
|
|
|
m_superseed_piece = index;
|
|
|
|
}
|
|
|
|
|
2008-01-07 02:10:46 +01:00
|
|
|
void peer_connection::second_tick(float tick_interval)
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
2007-04-05 00:27:36 +02:00
|
|
|
ptime now(time_now());
|
2008-05-12 05:05:27 +02:00
|
|
|
boost::intrusive_ptr<peer_connection> me(self());
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-06-30 09:58:50 +02:00
|
|
|
// the invariant check must be run before me is destructed
|
|
|
|
// in case the peer got disconnected
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2009-04-26 02:21:59 +02:00
|
|
|
|
|
|
|
// drain the IP overhead from the bandwidth limiters
|
|
|
|
if (m_ses.m_settings.rate_limit_ip_overhead)
|
|
|
|
{
|
|
|
|
int download_overhead = m_statistics.download_ip_overhead();
|
|
|
|
int upload_overhead = m_statistics.upload_ip_overhead();
|
|
|
|
m_bandwidth_channel[download_channel].use_quota(download_overhead);
|
|
|
|
m_bandwidth_channel[upload_channel].use_quota(upload_overhead);
|
|
|
|
|
|
|
|
int up_limit = m_bandwidth_channel[upload_channel].throttle();
|
|
|
|
int down_limit = m_bandwidth_channel[download_channel].throttle();
|
|
|
|
|
|
|
|
if (t)
|
|
|
|
{
|
|
|
|
t->m_bandwidth_channel[download_channel].use_quota(download_overhead);
|
|
|
|
t->m_bandwidth_channel[upload_channel].use_quota(upload_overhead);
|
|
|
|
|
|
|
|
if (down_limit > 0
|
|
|
|
&& download_overhead >= down_limit
|
|
|
|
&& t->alerts().should_post<performance_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(performance_alert(t->get_handle()
|
|
|
|
, performance_alert::download_limit_too_low));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (up_limit > 0
|
|
|
|
&& upload_overhead >= up_limit
|
|
|
|
&& t->alerts().should_post<performance_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(performance_alert(t->get_handle()
|
|
|
|
, performance_alert::upload_limit_too_low));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
m_ses.m_download_channel.use_quota(download_overhead);
|
|
|
|
m_ses.m_upload_channel.use_quota(upload_overhead);
|
|
|
|
}
|
|
|
|
|
2008-04-09 22:12:52 +02:00
|
|
|
if (!t || m_disconnecting)
|
|
|
|
{
|
|
|
|
m_ses.m_half_open.done(m_connection_ticket);
|
|
|
|
m_connecting = false;
|
2008-04-10 07:40:54 +02:00
|
|
|
disconnect("torrent aborted");
|
2008-04-09 22:12:52 +02:00
|
|
|
return;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
on_tick();
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
(*i)->tick();
|
|
|
|
}
|
2008-06-29 07:35:48 +02:00
|
|
|
if (is_disconnecting()) return;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// if the peer hasn't said a thing for a certain
|
|
|
|
// time, it is considered to have timed out
|
|
|
|
time_duration d;
|
|
|
|
d = now - m_last_receive;
|
|
|
|
if (d > seconds(m_timeout) && !m_connecting)
|
|
|
|
{
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " *** LAST ACTIVITY [ "
|
|
|
|
<< total_seconds(d) << " seconds ago ] ***\n";
|
|
|
|
#endif
|
|
|
|
disconnect("timed out: inactivity");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// do not stall waiting for a handshake
|
|
|
|
if (!m_connecting
|
|
|
|
&& in_handshake()
|
|
|
|
&& d > seconds(m_ses.settings().handshake_timeout))
|
|
|
|
{
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " *** NO HANDSHAKE [ waited "
|
|
|
|
<< total_seconds(d) << " seconds ] ***\n";
|
|
|
|
#endif
|
2008-06-29 08:01:20 +02:00
|
|
|
disconnect("timed out: no handshake");
|
2008-06-29 07:35:48 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// disconnect peers that we unchoked, but
|
|
|
|
// they didn't send a request within 20 seconds.
|
|
|
|
// but only if we're a seed
|
|
|
|
d = now - (std::max)(m_last_unchoke, m_last_incoming_request);
|
|
|
|
if (!m_connecting
|
|
|
|
&& m_requests.empty()
|
|
|
|
&& !m_choked
|
|
|
|
&& m_peer_interested
|
|
|
|
&& t && t->is_finished()
|
|
|
|
&& d > seconds(20))
|
|
|
|
{
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " *** NO REQUEST [ t: "
|
|
|
|
<< total_seconds(d) << " ] ***\n";
|
2006-11-14 01:08:16 +01:00
|
|
|
#endif
|
2008-06-29 07:35:48 +02:00
|
|
|
disconnect("timed out: no request when unchoked");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// if the peer hasn't become interested and we haven't
|
|
|
|
// become interested in the peer for 10 minutes, it
|
|
|
|
// has also timed out.
|
|
|
|
time_duration d1;
|
|
|
|
time_duration d2;
|
|
|
|
d1 = now - m_became_uninterested;
|
|
|
|
d2 = now - m_became_uninteresting;
|
|
|
|
time_duration time_limit = seconds(
|
|
|
|
m_ses.settings().inactivity_timeout);
|
|
|
|
|
|
|
|
// don't bother disconnect peers we haven't been interested
|
|
|
|
// in (and that hasn't been interested in us) for a while
|
|
|
|
// unless we have used up all our connection slots
|
|
|
|
if (!m_interesting
|
|
|
|
&& !m_peer_interested
|
|
|
|
&& d1 > time_limit
|
|
|
|
&& d2 > time_limit
|
|
|
|
&& (m_ses.num_connections() >= m_ses.max_connections()
|
|
|
|
|| (t && t->num_peers() >= t->max_connections())))
|
|
|
|
{
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " *** MUTUAL NO INTEREST [ "
|
|
|
|
"t1: " << total_seconds(d1) << " | "
|
|
|
|
"t2: " << total_seconds(d2) << " ] ***\n";
|
|
|
|
#endif
|
|
|
|
disconnect("timed out: no interest");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2008-06-29 11:50:42 +02:00
|
|
|
if (!m_download_queue.empty()
|
2008-07-07 14:04:06 +02:00
|
|
|
&& now > m_requested + seconds(m_ses.settings().request_timeout
|
|
|
|
+ m_timeout_extend))
|
2008-06-29 11:50:42 +02:00
|
|
|
{
|
2008-07-07 14:04:06 +02:00
|
|
|
snub_peer();
|
2008-06-29 11:50:42 +02:00
|
|
|
}
|
2008-07-07 14:04:06 +02:00
|
|
|
|
2008-06-29 07:35:48 +02:00
|
|
|
// if we haven't sent something in too long, send a keep-alive
|
|
|
|
keep_alive();
|
2006-11-14 01:08:16 +01:00
|
|
|
|
2007-05-25 19:06:30 +02:00
|
|
|
m_ignore_bandwidth_limits = m_ses.settings().ignore_limits_on_local_network
|
|
|
|
&& on_local_network();
|
|
|
|
|
2007-02-25 06:20:14 +01:00
|
|
|
m_statistics.second_tick(tick_interval);
|
|
|
|
|
2008-04-03 08:11:21 +02:00
|
|
|
if (m_statistics.upload_payload_rate() > m_upload_rate_peak)
|
|
|
|
{
|
|
|
|
m_upload_rate_peak = m_statistics.upload_payload_rate();
|
|
|
|
}
|
|
|
|
if (m_statistics.download_payload_rate() > m_download_rate_peak)
|
|
|
|
{
|
|
|
|
m_download_rate_peak = m_statistics.download_payload_rate();
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
if (peer_info_struct())
|
|
|
|
{
|
|
|
|
std::pair<const int, int>* as_stats = peer_info_struct()->inet_as;
|
|
|
|
if (as_stats && as_stats->second < m_download_rate_peak)
|
|
|
|
as_stats->second = m_download_rate_peak;
|
|
|
|
}
|
|
|
|
#endif
|
2008-04-03 08:11:21 +02:00
|
|
|
}
|
2008-05-12 05:05:27 +02:00
|
|
|
if (is_disconnecting()) return;
|
2008-04-03 08:11:21 +02:00
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
if (!t->ready_for_connections()) return;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
// calculate the desired download queue size
|
2006-10-11 16:02:21 +02:00
|
|
|
const float queue_time = m_ses.settings().request_queue_time;
|
2006-04-25 23:04:48 +02:00
|
|
|
// (if the latency is more than this, the download will stall)
|
|
|
|
// so, the queue size is queue_time * down_rate / 16 kiB
|
|
|
|
// (16 kB is the size of each request)
|
|
|
|
// the minimum number of requests is 2 and the maximum is 48
|
|
|
|
// the block size doesn't have to be 16. So we first query the
|
|
|
|
// torrent for it
|
2008-11-12 01:54:44 +01:00
|
|
|
const int block_size = t->block_size();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(block_size > 0);
|
2005-09-21 23:44:38 +02:00
|
|
|
|
2008-06-29 11:50:42 +02:00
|
|
|
if (m_snubbed)
|
|
|
|
{
|
|
|
|
m_desired_queue_size = 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
m_desired_queue_size = static_cast<int>(queue_time
|
|
|
|
* statistics().download_rate() / block_size);
|
|
|
|
if (m_desired_queue_size > m_max_out_request_queue)
|
|
|
|
m_desired_queue_size = m_max_out_request_queue;
|
|
|
|
if (m_desired_queue_size < min_request_queue)
|
|
|
|
m_desired_queue_size = min_request_queue;
|
2008-08-19 17:04:14 +02:00
|
|
|
|
|
|
|
if (m_desired_queue_size == m_max_out_request_queue
|
|
|
|
&& t->alerts().should_post<performance_alert>())
|
|
|
|
{
|
|
|
|
t->alerts().post_alert(performance_alert(t->get_handle()
|
|
|
|
, performance_alert::outstanding_request_limit_reached));
|
|
|
|
}
|
2008-06-29 11:50:42 +02:00
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2005-09-21 23:44:38 +02:00
|
|
|
if (!m_download_queue.empty()
|
2008-07-07 14:04:06 +02:00
|
|
|
&& now - m_last_piece > seconds(m_ses.settings().piece_timeout
|
|
|
|
+ m_timeout_extend))
|
2005-09-21 23:44:38 +02:00
|
|
|
{
|
|
|
|
// this peer isn't sending the pieces we've
|
|
|
|
// requested (this has been observed by BitComet)
|
|
|
|
// in this case we'll clear our download queue and
|
|
|
|
// re-request the blocks.
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2006-04-25 23:04:48 +02:00
|
|
|
<< " *** PIECE_REQUESTS TIMED OUT [ " << (int)m_download_queue.size()
|
2007-04-05 00:27:36 +02:00
|
|
|
<< " " << total_seconds(now - m_last_piece) << "] ***\n";
|
2005-09-21 23:44:38 +02:00
|
|
|
#endif
|
|
|
|
|
2008-07-07 14:04:06 +02:00
|
|
|
snub_peer();
|
2004-08-08 23:26:40 +02:00
|
|
|
}
|
|
|
|
|
2004-01-04 05:29:13 +01:00
|
|
|
// If the client sends more data
|
|
|
|
// we send it data faster, otherwise, slower.
|
|
|
|
// It will also depend on how much data the
|
|
|
|
// client has sent us. This is the mean to
|
2004-01-14 13:53:17 +01:00
|
|
|
// maintain the share ratio given by m_ratio
|
|
|
|
// with all peers.
|
2004-03-07 21:50:56 +01:00
|
|
|
|
2007-09-04 00:57:09 +02:00
|
|
|
if (t->is_finished() || is_choked() || t->ratio() == 0.0f)
|
2004-02-23 23:54:54 +01:00
|
|
|
{
|
|
|
|
// if we have downloaded more than one piece more
|
|
|
|
// than we have uploaded OR if we are a seed
|
|
|
|
// have an unlimited upload rate
|
2009-04-26 02:21:59 +02:00
|
|
|
m_bandwidth_channel[upload_channel].throttle(m_upload_limit);
|
2004-02-23 23:54:54 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2006-11-20 22:03:58 +01:00
|
|
|
size_type bias = 0x10000 + 2 * t->block_size() + m_free_upload;
|
2004-02-23 23:54:54 +01:00
|
|
|
|
2004-03-07 21:50:56 +01:00
|
|
|
double break_even_time = 15; // seconds.
|
|
|
|
size_type have_uploaded = m_statistics.total_payload_upload();
|
|
|
|
size_type have_downloaded = m_statistics.total_payload_download();
|
2004-02-23 23:54:54 +01:00
|
|
|
double download_speed = m_statistics.download_rate();
|
|
|
|
|
2004-03-07 21:50:56 +01:00
|
|
|
size_type soon_downloaded =
|
|
|
|
have_downloaded + (size_type)(download_speed * break_even_time*1.5);
|
|
|
|
|
2006-12-04 21:42:47 +01:00
|
|
|
if (t->ratio() != 1.f)
|
2006-04-25 23:04:48 +02:00
|
|
|
soon_downloaded = (size_type)(soon_downloaded*(double)t->ratio());
|
2004-03-07 21:50:56 +01:00
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
double upload_speed_limit = (soon_downloaded - have_uploaded
|
|
|
|
+ bias) / break_even_time;
|
|
|
|
|
|
|
|
if (m_upload_limit > 0 && m_upload_limit < upload_speed_limit)
|
|
|
|
upload_speed_limit = m_upload_limit;
|
2004-02-23 23:54:54 +01:00
|
|
|
|
2007-08-17 00:13:35 +02:00
|
|
|
upload_speed_limit = (std::min)(upload_speed_limit,
|
2007-08-21 06:46:17 +02:00
|
|
|
(double)(std::numeric_limits<int>::max)());
|
2004-02-23 23:54:54 +01:00
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
m_bandwidth_channel[upload_channel].throttle(
|
|
|
|
(std::min)((std::max)((int)upload_speed_limit, 10)
|
2007-01-10 16:02:25 +01:00
|
|
|
, m_upload_limit));
|
2004-02-23 23:54:54 +01:00
|
|
|
}
|
2006-12-04 21:42:47 +01:00
|
|
|
|
2007-06-14 23:47:00 +02:00
|
|
|
// update once every minute
|
|
|
|
if (now - m_remote_dl_update >= seconds(60))
|
|
|
|
{
|
|
|
|
float factor = 0.6666666666667f;
|
2007-06-07 02:05:18 +02:00
|
|
|
|
2007-06-15 01:15:30 +02:00
|
|
|
if (m_remote_dl_rate == 0) factor = 0.0f;
|
2007-06-07 02:53:48 +02:00
|
|
|
|
2007-07-03 01:48:06 +02:00
|
|
|
m_remote_dl_rate = int((m_remote_dl_rate * factor) +
|
|
|
|
((m_remote_bytes_dled * (1.0f-factor)) / 60.f));
|
2007-06-07 02:05:18 +02:00
|
|
|
|
2007-06-14 23:47:00 +02:00
|
|
|
m_remote_bytes_dled = 0;
|
|
|
|
m_remote_dl_update = now;
|
|
|
|
}
|
2007-06-07 02:05:18 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
fill_send_buffer();
|
2003-12-08 02:37:30 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-07-07 14:04:06 +02:00
|
|
|
void peer_connection::snub_peer()
|
|
|
|
{
|
2008-07-08 10:13:45 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-07-07 14:04:06 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
TORRENT_ASSERT(t);
|
|
|
|
|
|
|
|
if (!m_snubbed)
|
|
|
|
{
|
|
|
|
m_snubbed = true;
|
|
|
|
if (m_ses.m_alerts.should_post<peer_snubbed_alert>())
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(peer_snubbed_alert(t->get_handle()
|
|
|
|
, m_remote, m_peer_id));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
m_desired_queue_size = 1;
|
|
|
|
|
2008-07-11 11:25:03 +02:00
|
|
|
if (on_parole())
|
|
|
|
{
|
|
|
|
m_timeout_extend += m_ses.settings().request_timeout;
|
|
|
|
return;
|
|
|
|
}
|
2008-07-07 14:04:06 +02:00
|
|
|
if (!t->has_picker()) return;
|
|
|
|
piece_picker& picker = t->picker();
|
|
|
|
|
|
|
|
piece_block r(-1, -1);
|
|
|
|
// time out the last request in the queue
|
|
|
|
if (!m_request_queue.empty())
|
|
|
|
{
|
|
|
|
r = m_request_queue.back();
|
|
|
|
m_request_queue.pop_back();
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(!m_download_queue.empty());
|
2009-03-17 10:34:44 +01:00
|
|
|
pending_block& qe = m_download_queue.back();
|
|
|
|
if (!qe.timed_out && !qe.not_wanted)
|
|
|
|
r = qe.block;
|
2008-07-07 14:04:06 +02:00
|
|
|
|
|
|
|
// only time out a request if it blocks the piece
|
|
|
|
// from being completed (i.e. no free blocks to
|
|
|
|
// request from it)
|
|
|
|
piece_picker::downloading_piece p;
|
2009-03-17 10:34:44 +01:00
|
|
|
picker.piece_info(qe.block.piece_index, p);
|
|
|
|
int free_blocks = picker.blocks_in_piece(qe.block.piece_index)
|
2008-07-07 14:04:06 +02:00
|
|
|
- p.finished - p.writing - p.requested;
|
|
|
|
if (free_blocks > 0)
|
|
|
|
{
|
|
|
|
m_timeout_extend += m_ses.settings().request_timeout;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (m_ses.m_alerts.should_post<block_timeout_alert>())
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(block_timeout_alert(t->get_handle()
|
2009-03-17 10:34:44 +01:00
|
|
|
, remote(), pid(), qe.block.block_index, qe.block.piece_index));
|
2008-07-07 14:04:06 +02:00
|
|
|
}
|
2009-03-17 10:34:44 +01:00
|
|
|
qe.timed_out = true;
|
2008-07-07 14:04:06 +02:00
|
|
|
}
|
|
|
|
if (!m_download_queue.empty() || !m_request_queue.empty())
|
|
|
|
m_timeout_extend += m_ses.settings().request_timeout;
|
|
|
|
|
2008-07-09 13:11:06 +02:00
|
|
|
m_desired_queue_size = 2;
|
2008-07-07 14:04:06 +02:00
|
|
|
request_a_block(*t, *this);
|
2008-07-09 13:11:06 +02:00
|
|
|
m_desired_queue_size = 1;
|
2008-07-07 14:04:06 +02:00
|
|
|
|
|
|
|
// abort the block after the new one has
|
|
|
|
// been requested in order to prevent it from
|
|
|
|
// picking the same block again, stalling the
|
|
|
|
// same piece indefinitely.
|
|
|
|
if (r != piece_block(-1, -1))
|
|
|
|
picker.abort_download(r);
|
2008-07-08 10:13:45 +02:00
|
|
|
|
|
|
|
send_block_requests();
|
2008-07-07 14:04:06 +02:00
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::fill_send_buffer()
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2008-10-10 09:08:46 +02:00
|
|
|
#ifdef TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
2008-10-10 09:08:46 +02:00
|
|
|
#endif
|
2004-01-25 19:18:36 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
if (!t) return;
|
2005-03-12 13:18:07 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// only add new piece-chunks if the send buffer is small enough
|
|
|
|
// otherwise there will be no end to how large it will be!
|
2007-05-10 08:20:29 +02:00
|
|
|
|
2007-05-14 08:06:28 +02:00
|
|
|
int buffer_size_watermark = int(m_statistics.upload_rate()) / 2;
|
2008-01-08 02:16:30 +01:00
|
|
|
if (buffer_size_watermark < 512) buffer_size_watermark = 512;
|
|
|
|
else if (buffer_size_watermark > m_ses.settings().send_buffer_watermark)
|
|
|
|
buffer_size_watermark = m_ses.settings().send_buffer_watermark;
|
2007-05-10 08:20:29 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
while (!m_requests.empty()
|
2008-01-13 10:33:00 +01:00
|
|
|
&& (send_buffer_size() + m_reading_bytes < buffer_size_watermark))
|
2003-10-30 00:28:09 +01:00
|
|
|
{
|
2008-08-29 19:21:56 +02:00
|
|
|
TORRENT_ASSERT(t->ready_for_connections());
|
2006-04-25 23:04:48 +02:00
|
|
|
peer_request& r = m_requests.front();
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(r.piece >= 0);
|
|
|
|
TORRENT_ASSERT(r.piece < (int)m_have_piece.size());
|
|
|
|
TORRENT_ASSERT(t->have_piece(r.piece));
|
|
|
|
TORRENT_ASSERT(r.start + r.length <= t->torrent_file().piece_size(r.piece));
|
|
|
|
TORRENT_ASSERT(r.length > 0 && r.start >= 0);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2009-02-03 08:46:24 +01:00
|
|
|
if (!t->seed_mode() || t->verified_piece(r.piece))
|
|
|
|
{
|
|
|
|
t->filesystem().async_read(r, bind(&peer_connection::on_disk_read_complete
|
|
|
|
, self(), _1, _2, r));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// this means we're in seed mode and we haven't yet
|
|
|
|
// verified this piece (r.piece)
|
|
|
|
t->filesystem().async_read_and_hash(r, bind(&peer_connection::on_disk_read_complete
|
|
|
|
, self(), _1, _2, r));
|
|
|
|
t->verified(r.piece);
|
|
|
|
}
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
m_reading_bytes += r.length;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
m_requests.erase(m_requests.begin());
|
|
|
|
}
|
|
|
|
}
|
2003-10-30 00:28:09 +01:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
void peer_connection::on_disk_read_complete(int ret, disk_io_job const& j, peer_request r)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
m_reading_bytes -= r.length;
|
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
disk_buffer_holder buffer(m_ses, j.buffer);
|
|
|
|
|
2009-02-03 08:46:24 +01:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2007-06-10 22:46:09 +02:00
|
|
|
if (ret != r.length || m_torrent.expired())
|
|
|
|
{
|
|
|
|
if (!t)
|
|
|
|
{
|
2008-01-07 02:10:46 +01:00
|
|
|
disconnect(j.str.c_str());
|
2007-06-10 22:46:09 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2009-02-03 08:46:24 +01:00
|
|
|
if (ret == -3)
|
|
|
|
{
|
|
|
|
if (t->seed_mode()) t->leave_seed_mode(false);
|
|
|
|
write_reject_request(r);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (t->alerts().should_post<file_error_alert>())
|
|
|
|
t->alerts().post_alert(file_error_alert(j.error_file, t->get_handle(), j.str));
|
2009-02-23 02:21:19 +01:00
|
|
|
t->set_error(j.error, j.error_file);
|
2009-02-03 08:46:24 +01:00
|
|
|
t->pause();
|
|
|
|
}
|
2007-06-10 22:46:09 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2009-02-03 08:46:24 +01:00
|
|
|
if (t)
|
|
|
|
{
|
|
|
|
if (t->seed_mode() && t->all_verified())
|
|
|
|
t->leave_seed_mode(true);
|
|
|
|
}
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string()
|
|
|
|
<< " ==> PIECE [ piece: " << r.piece << " | s: " << r.start
|
|
|
|
<< " | l: " << r.length << " ]\n";
|
|
|
|
#endif
|
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
write_piece(r, buffer);
|
2007-06-10 22:46:09 +02:00
|
|
|
setup_send();
|
|
|
|
}
|
|
|
|
|
2007-01-10 16:02:25 +01:00
|
|
|
void peer_connection::assign_bandwidth(int channel, int amount)
|
|
|
|
{
|
2007-01-30 18:56:42 +01:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2007-01-10 16:02:25 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << "bandwidth [ " << channel << " ] + " << amount << "\n";
|
|
|
|
#endif
|
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
TORRENT_ASSERT(amount > 0);
|
|
|
|
m_quota[channel] += amount;
|
|
|
|
TORRENT_ASSERT(m_channel_state[channel] == peer_info::bw_limit);
|
2008-01-14 00:46:43 +01:00
|
|
|
m_channel_state[channel] = peer_info::bw_idle;
|
2007-01-10 16:02:25 +01:00
|
|
|
if (channel == upload_channel)
|
|
|
|
{
|
|
|
|
setup_send();
|
|
|
|
}
|
|
|
|
else if (channel == download_channel)
|
|
|
|
{
|
|
|
|
setup_receive();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::setup_send()
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-01-14 00:46:43 +01:00
|
|
|
if (m_channel_state[upload_channel] != peer_info::bw_idle) return;
|
2007-01-10 16:02:25 +01:00
|
|
|
|
|
|
|
shared_ptr<torrent> t = m_torrent.lock();
|
2007-05-25 19:06:30 +02:00
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
if (m_quota[upload_channel] == 0
|
2007-09-29 18:14:03 +02:00
|
|
|
&& !m_send_buffer.empty()
|
2007-01-10 16:02:25 +01:00
|
|
|
&& !m_connecting
|
2007-05-25 19:06:30 +02:00
|
|
|
&& t
|
|
|
|
&& !m_ignore_bandwidth_limits)
|
2007-01-10 16:02:25 +01:00
|
|
|
{
|
|
|
|
// in this case, we have data to send, but no
|
|
|
|
// bandwidth. So, we simply request bandwidth
|
|
|
|
// from the torrent
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2009-04-26 02:21:59 +02:00
|
|
|
int priority = 1 + is_interesting() * 2 + m_requests_in_buffer.size();
|
|
|
|
// peers that we are not interested in are non-prioritized
|
|
|
|
m_channel_state[upload_channel] = peer_info::bw_limit;
|
|
|
|
m_ses.m_upload_rate.request_bandwidth(self()
|
|
|
|
, m_send_buffer.size(), priority
|
|
|
|
, &m_ses.m_upload_channel
|
|
|
|
, &t->m_bandwidth_channel[upload_channel]
|
|
|
|
, &m_bandwidth_channel[upload_channel]);
|
2008-07-08 02:03:08 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2009-04-26 02:21:59 +02:00
|
|
|
(*m_logger) << time_now_string() << " *** REQUEST_BANDWIDTH [ "
|
|
|
|
"upload: " << m_send_buffer.size()
|
|
|
|
<< " prio: " << priority << "]\n";
|
2008-07-08 02:03:08 +02:00
|
|
|
#endif
|
2007-01-10 16:02:25 +01:00
|
|
|
return;
|
|
|
|
}
|
2007-06-06 02:41:20 +02:00
|
|
|
|
2008-02-27 21:37:41 +01:00
|
|
|
if (!can_write())
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2008-11-12 00:45:43 +01:00
|
|
|
if (m_send_buffer.empty())
|
|
|
|
{
|
|
|
|
(*m_logger) << time_now_string() << " *** SEND BUFFER DEPLETED ["
|
2009-04-26 02:21:59 +02:00
|
|
|
" quota: " << m_quota[upload_channel] <<
|
2008-11-12 00:45:43 +01:00
|
|
|
" ignore: " << (m_ignore_bandwidth_limits?"yes":"no") <<
|
|
|
|
" buf: " << m_send_buffer.size() <<
|
|
|
|
" connecting: " << (m_connecting?"yes":"no") <<
|
|
|
|
" ]\n";
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
(*m_logger) << time_now_string() << " *** CANNOT WRITE ["
|
2009-04-26 02:21:59 +02:00
|
|
|
" quota: " << m_quota[upload_channel] <<
|
2008-11-12 00:45:43 +01:00
|
|
|
" ignore: " << (m_ignore_bandwidth_limits?"yes":"no") <<
|
|
|
|
" buf: " << m_send_buffer.size() <<
|
|
|
|
" connecting: " << (m_connecting?"yes":"no") <<
|
|
|
|
" ]\n";
|
|
|
|
}
|
2008-02-27 21:37:41 +01:00
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// send the actual buffer
|
2007-09-29 18:14:03 +02:00
|
|
|
if (!m_send_buffer.empty())
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
2007-09-29 18:14:03 +02:00
|
|
|
int amount_to_send = m_send_buffer.size();
|
2009-04-26 02:21:59 +02:00
|
|
|
int quota_left = m_quota[upload_channel];
|
2007-05-25 19:06:30 +02:00
|
|
|
if (!m_ignore_bandwidth_limits && amount_to_send > quota_left)
|
|
|
|
amount_to_send = quota_left;
|
2003-10-30 00:28:09 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(amount_to_send > 0);
|
2006-04-30 02:39:18 +02:00
|
|
|
|
2007-05-25 19:06:30 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2008-02-27 21:37:41 +01:00
|
|
|
(*m_logger) << time_now_string() << " *** ASYNC_WRITE [ bytes: " << amount_to_send << " ]\n";
|
2007-05-25 19:06:30 +02:00
|
|
|
#endif
|
2007-09-29 18:14:03 +02:00
|
|
|
std::list<asio::const_buffer> const& vec = m_send_buffer.build_iovec(amount_to_send);
|
|
|
|
m_socket->async_write_some(vec, bind(&peer_connection::on_send_data, self(), _1, _2));
|
2006-04-30 02:39:18 +02:00
|
|
|
|
2008-01-14 00:46:43 +01:00
|
|
|
m_channel_state[upload_channel] = peer_info::bw_network;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
}
|
2003-12-22 08:14:35 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::setup_receive()
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
2003-12-08 22:59:48 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2008-01-14 00:46:43 +01:00
|
|
|
if (m_channel_state[download_channel] != peer_info::bw_idle) return;
|
2007-01-10 16:02:25 +01:00
|
|
|
|
|
|
|
shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
if (m_quota[download_channel] == 0
|
2007-01-10 16:02:25 +01:00
|
|
|
&& !m_connecting
|
2007-05-25 19:06:30 +02:00
|
|
|
&& t
|
|
|
|
&& !m_ignore_bandwidth_limits)
|
2007-01-10 16:02:25 +01:00
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2009-04-26 02:21:59 +02:00
|
|
|
(*m_logger) << time_now_string() << " *** REQUEST_BANDWIDTH [ "
|
|
|
|
"download: " << (m_download_queue.size() * 16 * 1024 + 30)
|
|
|
|
<< " prio: " << m_priority << " ]\n";
|
|
|
|
#endif
|
|
|
|
TORRENT_ASSERT(m_channel_state[download_channel] == peer_info::bw_idle);
|
2009-04-26 21:44:08 +02:00
|
|
|
TORRENT_ASSERT(m_outstanding_bytes >= 0);
|
2009-04-26 02:21:59 +02:00
|
|
|
m_channel_state[download_channel] = peer_info::bw_limit;
|
|
|
|
m_ses.m_download_rate.request_bandwidth(self()
|
|
|
|
, m_outstanding_bytes + 30, m_priority
|
|
|
|
, &m_ses.m_download_channel
|
|
|
|
, &t->m_bandwidth_channel[download_channel]
|
|
|
|
, &m_bandwidth_channel[download_channel]);
|
2007-01-10 16:02:25 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2008-02-27 21:37:41 +01:00
|
|
|
if (!can_read())
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " *** CANNOT READ ["
|
2009-04-26 02:21:59 +02:00
|
|
|
" quota: " << m_quota[download_channel] <<
|
2008-02-27 21:37:41 +01:00
|
|
|
" ignore: " << (m_ignore_bandwidth_limits?"yes":"no") <<
|
|
|
|
" outstanding: " << m_outstanding_writing_bytes <<
|
|
|
|
" outstanding-limit: " << m_ses.settings().max_outstanding_disk_bytes_per_connection <<
|
|
|
|
" ]\n";
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_packet_size > 0);
|
2007-05-25 19:06:30 +02:00
|
|
|
int max_receive = m_packet_size - m_recv_pos;
|
2009-03-13 07:09:39 +01:00
|
|
|
if (m_recv_pos >= m_soft_packet_size) m_soft_packet_size = 0;
|
|
|
|
if (m_soft_packet_size && max_receive > m_soft_packet_size - m_recv_pos)
|
|
|
|
max_receive = m_soft_packet_size - m_recv_pos;
|
2009-04-26 02:21:59 +02:00
|
|
|
int quota_left = m_quota[download_channel];
|
2007-05-25 19:06:30 +02:00
|
|
|
if (!m_ignore_bandwidth_limits && max_receive > quota_left)
|
|
|
|
max_receive = quota_left;
|
|
|
|
|
2007-08-01 17:48:55 +02:00
|
|
|
if (max_receive == 0) return;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_recv_pos >= 0);
|
|
|
|
TORRENT_ASSERT(m_packet_size > 0);
|
|
|
|
TORRENT_ASSERT(can_read());
|
2007-05-25 19:06:30 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
2008-02-27 21:37:41 +01:00
|
|
|
(*m_logger) << time_now_string() << " *** ASYNC_READ [ max: " << max_receive << " bytes ]\n";
|
2007-05-25 19:06:30 +02:00
|
|
|
#endif
|
2008-04-10 12:03:23 +02:00
|
|
|
|
|
|
|
int regular_buffer_size = m_packet_size - m_disk_recv_buffer_size;
|
|
|
|
|
|
|
|
if (int(m_recv_buffer.size()) < regular_buffer_size)
|
|
|
|
m_recv_buffer.resize(regular_buffer_size);
|
|
|
|
|
2008-05-05 08:25:22 +02:00
|
|
|
if (!m_disk_recv_buffer || regular_buffer_size >= m_recv_pos + max_receive)
|
2008-04-10 12:03:23 +02:00
|
|
|
{
|
|
|
|
// only receive into regular buffer
|
|
|
|
TORRENT_ASSERT(m_recv_pos + max_receive <= int(m_recv_buffer.size()));
|
|
|
|
m_socket->async_read_some(asio::buffer(&m_recv_buffer[m_recv_pos]
|
|
|
|
, max_receive), bind(&peer_connection::on_receive_data, self(), _1, _2));
|
|
|
|
}
|
|
|
|
else if (m_recv_pos >= regular_buffer_size)
|
|
|
|
{
|
|
|
|
// only receive into disk buffer
|
|
|
|
TORRENT_ASSERT(m_recv_pos - regular_buffer_size >= 0);
|
|
|
|
TORRENT_ASSERT(m_recv_pos - regular_buffer_size + max_receive <= m_disk_recv_buffer_size);
|
2008-05-05 08:25:22 +02:00
|
|
|
m_socket->async_read_some(asio::buffer(m_disk_recv_buffer.get() + m_recv_pos - regular_buffer_size
|
2008-04-10 12:03:23 +02:00
|
|
|
, max_receive)
|
|
|
|
, bind(&peer_connection::on_receive_data, self(), _1, _2));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// receive into both regular and disk buffer
|
|
|
|
TORRENT_ASSERT(max_receive + m_recv_pos > regular_buffer_size);
|
|
|
|
TORRENT_ASSERT(m_recv_pos < regular_buffer_size);
|
|
|
|
TORRENT_ASSERT(max_receive - regular_buffer_size
|
|
|
|
+ m_recv_pos <= m_disk_recv_buffer_size);
|
|
|
|
|
|
|
|
boost::array<asio::mutable_buffer, 2> vec;
|
|
|
|
vec[0] = asio::buffer(&m_recv_buffer[m_recv_pos]
|
|
|
|
, regular_buffer_size - m_recv_pos);
|
2008-05-05 08:25:22 +02:00
|
|
|
vec[1] = asio::buffer(m_disk_recv_buffer.get()
|
2008-04-10 12:03:23 +02:00
|
|
|
, max_receive - regular_buffer_size + m_recv_pos);
|
|
|
|
m_socket->async_read_some(vec, bind(&peer_connection::on_receive_data
|
|
|
|
, self(), _1, _2));
|
|
|
|
}
|
2008-01-14 00:46:43 +01:00
|
|
|
m_channel_state[download_channel] = peer_info::bw_network;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
#ifndef TORRENT_DISABLE_ENCRYPTION
|
|
|
|
|
|
|
|
// returns the last 'bytes' from the receive buffer
|
|
|
|
std::pair<buffer::interval, buffer::interval> peer_connection::wr_recv_buffers(int bytes)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(bytes <= m_recv_pos);
|
|
|
|
|
|
|
|
std::pair<buffer::interval, buffer::interval> vec;
|
|
|
|
int regular_buffer_size = m_packet_size - m_disk_recv_buffer_size;
|
|
|
|
TORRENT_ASSERT(regular_buffer_size >= 0);
|
2008-05-05 08:25:22 +02:00
|
|
|
if (!m_disk_recv_buffer || regular_buffer_size >= m_recv_pos)
|
2008-04-10 12:03:23 +02:00
|
|
|
{
|
|
|
|
vec.first = buffer::interval(&m_recv_buffer[0]
|
|
|
|
+ m_recv_pos - bytes, &m_recv_buffer[0] + m_recv_pos);
|
|
|
|
vec.second = buffer::interval(0,0);
|
|
|
|
}
|
|
|
|
else if (m_recv_pos - bytes >= regular_buffer_size)
|
|
|
|
{
|
2008-05-05 08:25:22 +02:00
|
|
|
vec.first = buffer::interval(m_disk_recv_buffer.get() + m_recv_pos
|
|
|
|
- regular_buffer_size - bytes, m_disk_recv_buffer.get() + m_recv_pos
|
2008-04-10 12:03:23 +02:00
|
|
|
- regular_buffer_size);
|
|
|
|
vec.second = buffer::interval(0,0);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(m_recv_pos - bytes < regular_buffer_size);
|
|
|
|
TORRENT_ASSERT(m_recv_pos > regular_buffer_size);
|
|
|
|
vec.first = buffer::interval(&m_recv_buffer[0] + m_recv_pos - bytes
|
|
|
|
, &m_recv_buffer[0] + regular_buffer_size);
|
2008-05-05 08:25:22 +02:00
|
|
|
vec.second = buffer::interval(m_disk_recv_buffer.get()
|
|
|
|
, m_disk_recv_buffer.get() + m_recv_pos - regular_buffer_size);
|
2008-04-10 12:03:23 +02:00
|
|
|
}
|
|
|
|
TORRENT_ASSERT(vec.first.left() + vec.second.left() == bytes);
|
|
|
|
return vec;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::reset_recv_buffer(int packet_size)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(packet_size > 0);
|
2007-02-12 06:46:29 +01:00
|
|
|
if (m_recv_pos > m_packet_size)
|
|
|
|
{
|
|
|
|
cut_receive_buffer(m_packet_size, packet_size);
|
|
|
|
return;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
m_recv_pos = 0;
|
|
|
|
m_packet_size = packet_size;
|
|
|
|
}
|
2007-06-06 02:41:20 +02:00
|
|
|
|
2008-07-08 02:03:08 +02:00
|
|
|
void peer_connection::send_buffer(char const* buf, int size, int flags)
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
2008-07-08 02:03:08 +02:00
|
|
|
if (flags == message_type_request)
|
|
|
|
m_requests_in_buffer.push_back(m_send_buffer.size() + size);
|
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
int free_space = m_send_buffer.space_in_last_buffer();
|
|
|
|
if (free_space > size) free_space = size;
|
|
|
|
if (free_space > 0)
|
|
|
|
{
|
|
|
|
m_send_buffer.append(buf, free_space);
|
|
|
|
size -= free_space;
|
|
|
|
buf += free_space;
|
|
|
|
#ifdef TORRENT_STATS
|
|
|
|
m_ses.m_buffer_usage_logger << log_time() << " send_buffer: "
|
|
|
|
<< free_space << std::endl;
|
|
|
|
m_ses.log_buffer_usage();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
if (size <= 0) return;
|
|
|
|
|
|
|
|
std::pair<char*, int> buffer = m_ses.allocate_buffer(size);
|
2008-02-05 06:51:05 +01:00
|
|
|
if (buffer.first == 0)
|
|
|
|
{
|
|
|
|
disconnect("out of memory");
|
|
|
|
return;
|
|
|
|
}
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(buffer.second >= size);
|
2007-09-29 18:14:03 +02:00
|
|
|
std::memcpy(buffer.first, buf, size);
|
|
|
|
m_send_buffer.append_buffer(buffer.first, buffer.second, size
|
|
|
|
, bind(&session_impl::free_buffer, boost::ref(m_ses), _1, buffer.second));
|
|
|
|
#ifdef TORRENT_STATS
|
|
|
|
m_ses.m_buffer_usage_logger << log_time() << " send_buffer_alloc: " << size << std::endl;
|
|
|
|
m_ses.log_buffer_usage();
|
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
setup_send();
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: change this interface to automatically call setup_send() when the
|
|
|
|
// return value is destructed
|
|
|
|
buffer::interval peer_connection::allocate_send_buffer(int size)
|
|
|
|
{
|
2007-12-27 22:43:11 +01:00
|
|
|
TORRENT_ASSERT(size > 0);
|
2007-09-29 18:14:03 +02:00
|
|
|
char* insert = m_send_buffer.allocate_appendix(size);
|
|
|
|
if (insert == 0)
|
|
|
|
{
|
|
|
|
std::pair<char*, int> buffer = m_ses.allocate_buffer(size);
|
2008-02-05 06:51:05 +01:00
|
|
|
if (buffer.first == 0)
|
|
|
|
{
|
|
|
|
disconnect("out of memory");
|
|
|
|
return buffer::interval(0, 0);
|
|
|
|
}
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(buffer.second >= size);
|
2007-09-29 18:14:03 +02:00
|
|
|
m_send_buffer.append_buffer(buffer.first, buffer.second, size
|
|
|
|
, bind(&session_impl::free_buffer, boost::ref(m_ses), _1, buffer.second));
|
|
|
|
buffer::interval ret(buffer.first, buffer.first + size);
|
|
|
|
#ifdef TORRENT_STATS
|
|
|
|
m_ses.m_buffer_usage_logger << log_time() << " allocate_buffer_alloc: " << size << std::endl;
|
|
|
|
m_ses.log_buffer_usage();
|
|
|
|
#endif
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
#ifdef TORRENT_STATS
|
|
|
|
m_ses.m_buffer_usage_logger << log_time() << " allocate_buffer: " << size << std::endl;
|
|
|
|
m_ses.log_buffer_usage();
|
|
|
|
#endif
|
|
|
|
buffer::interval ret(insert, insert + size);
|
|
|
|
return ret;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
template<class T>
|
|
|
|
struct set_to_zero
|
|
|
|
{
|
|
|
|
set_to_zero(T& v, bool cond): m_val(v), m_cond(cond) {}
|
|
|
|
void fire() { if (!m_cond) return; m_cond = false; m_val = 0; }
|
|
|
|
~set_to_zero() { if (m_cond) m_val = 0; }
|
|
|
|
private:
|
|
|
|
T& m_val;
|
|
|
|
bool m_cond;
|
|
|
|
};
|
|
|
|
|
|
|
|
// --------------------------
|
|
|
|
// RECEIVE DATA
|
|
|
|
// --------------------------
|
|
|
|
|
2008-05-03 18:05:42 +02:00
|
|
|
void peer_connection::on_receive_data(const error_code& error
|
2008-04-07 10:39:01 +02:00
|
|
|
, std::size_t bytes_transferred)
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2009-01-25 03:45:31 +01:00
|
|
|
// keep ourselves alive in until this function exits in
|
|
|
|
// case we disconnect
|
|
|
|
boost::intrusive_ptr<peer_connection> me(self());
|
|
|
|
|
2008-01-14 00:46:43 +01:00
|
|
|
TORRENT_ASSERT(m_channel_state[download_channel] == peer_info::bw_network);
|
|
|
|
m_channel_state[download_channel] = peer_info::bw_idle;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-12-28 21:06:59 +01:00
|
|
|
int bytes_in_loop = bytes_transferred;
|
2008-11-14 21:51:49 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (error)
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-12-19 22:36:54 +01:00
|
|
|
(*m_logger) << time_now_string() << " **ERROR**: "
|
|
|
|
<< error.message() << "[in peer_connection::on_receive_data]\n";
|
2005-03-19 13:22:40 +01:00
|
|
|
#endif
|
2008-12-28 21:06:59 +01:00
|
|
|
m_statistics.trancieve_ip_packet(bytes_in_loop, m_remote.address().is_v6());
|
2006-04-25 23:04:48 +02:00
|
|
|
on_receive(error, bytes_transferred);
|
2008-01-07 02:10:46 +01:00
|
|
|
disconnect(error.message().c_str());
|
|
|
|
return;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2004-03-28 19:45:37 +02:00
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
int max_receive = 0;
|
2007-05-10 03:50:11 +02:00
|
|
|
do
|
|
|
|
{
|
2007-05-25 21:42:10 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << "read " << bytes_transferred << " bytes\n";
|
|
|
|
#endif
|
2007-05-10 03:50:11 +02:00
|
|
|
// correct the dl quota usage, if not all of the buffer was actually read
|
2007-05-25 19:06:30 +02:00
|
|
|
if (!m_ignore_bandwidth_limits)
|
2009-04-26 02:21:59 +02:00
|
|
|
{
|
|
|
|
TORRENT_ASSERT(bytes_transferred <= m_quota[download_channel]);
|
|
|
|
m_quota[download_channel] -= bytes_transferred;
|
|
|
|
}
|
2007-05-10 03:50:11 +02:00
|
|
|
|
2008-12-28 21:06:59 +01:00
|
|
|
if (m_disconnecting)
|
|
|
|
{
|
|
|
|
m_statistics.trancieve_ip_packet(bytes_in_loop, m_remote.address().is_v6());
|
|
|
|
return;
|
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_packet_size > 0);
|
|
|
|
TORRENT_ASSERT(bytes_transferred > 0);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-05-10 03:50:11 +02:00
|
|
|
m_last_receive = time_now();
|
|
|
|
m_recv_pos += bytes_transferred;
|
2008-04-10 12:03:23 +02:00
|
|
|
TORRENT_ASSERT(m_recv_pos <= int(m_recv_buffer.size()
|
|
|
|
+ m_disk_recv_buffer_size));
|
2007-12-19 22:36:54 +01:00
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-09-21 19:12:26 +02:00
|
|
|
size_type cur_payload_dl = m_statistics.last_payload_downloaded();
|
|
|
|
size_type cur_protocol_dl = m_statistics.last_protocol_downloaded();
|
|
|
|
#endif
|
2007-08-01 17:48:55 +02:00
|
|
|
on_receive(error, bytes_transferred);
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-09-21 19:12:26 +02:00
|
|
|
TORRENT_ASSERT(m_statistics.last_payload_downloaded() - cur_payload_dl >= 0);
|
|
|
|
TORRENT_ASSERT(m_statistics.last_protocol_downloaded() - cur_protocol_dl >= 0);
|
|
|
|
size_type stats_diff = m_statistics.last_payload_downloaded() - cur_payload_dl +
|
|
|
|
m_statistics.last_protocol_downloaded() - cur_protocol_dl;
|
|
|
|
TORRENT_ASSERT(stats_diff == bytes_transferred);
|
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_packet_size > 0);
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2007-05-10 03:50:11 +02:00
|
|
|
if (m_peer_choked
|
|
|
|
&& m_recv_pos == 0
|
|
|
|
&& (m_recv_buffer.capacity() - m_packet_size) > 128)
|
|
|
|
{
|
2007-09-23 02:51:45 +02:00
|
|
|
buffer(m_packet_size).swap(m_recv_buffer);
|
2007-05-10 03:50:11 +02:00
|
|
|
}
|
|
|
|
|
2009-03-13 07:09:39 +01:00
|
|
|
if (m_recv_pos >= m_soft_packet_size) m_soft_packet_size = 0;
|
2008-04-10 12:03:23 +02:00
|
|
|
max_receive = m_packet_size - m_recv_pos;
|
2009-03-13 07:09:39 +01:00
|
|
|
if (m_soft_packet_size && max_receive > m_soft_packet_size - m_recv_pos)
|
|
|
|
max_receive = m_soft_packet_size - m_recv_pos;
|
2009-04-26 02:21:59 +02:00
|
|
|
int quota_left = m_quota[download_channel];
|
2007-05-25 19:06:30 +02:00
|
|
|
if (!m_ignore_bandwidth_limits && max_receive > quota_left)
|
|
|
|
max_receive = quota_left;
|
2007-05-16 20:21:08 +02:00
|
|
|
|
|
|
|
if (max_receive == 0) break;
|
|
|
|
|
2008-04-10 12:03:23 +02:00
|
|
|
int regular_buffer_size = m_packet_size - m_disk_recv_buffer_size;
|
|
|
|
|
|
|
|
if (int(m_recv_buffer.size()) < regular_buffer_size)
|
|
|
|
m_recv_buffer.resize(regular_buffer_size);
|
|
|
|
|
2008-05-03 18:05:42 +02:00
|
|
|
error_code ec;
|
2008-05-05 08:25:22 +02:00
|
|
|
if (!m_disk_recv_buffer || regular_buffer_size >= m_recv_pos + max_receive)
|
2008-04-10 12:03:23 +02:00
|
|
|
{
|
|
|
|
// only receive into regular buffer
|
|
|
|
TORRENT_ASSERT(m_recv_pos + max_receive <= int(m_recv_buffer.size()));
|
|
|
|
bytes_transferred = m_socket->read_some(asio::buffer(&m_recv_buffer[m_recv_pos]
|
|
|
|
, max_receive), ec);
|
|
|
|
}
|
|
|
|
else if (m_recv_pos >= regular_buffer_size)
|
|
|
|
{
|
|
|
|
// only receive into disk buffer
|
|
|
|
TORRENT_ASSERT(m_recv_pos - regular_buffer_size >= 0);
|
|
|
|
TORRENT_ASSERT(m_recv_pos - regular_buffer_size + max_receive <= m_disk_recv_buffer_size);
|
2008-05-05 08:25:22 +02:00
|
|
|
bytes_transferred = m_socket->read_some(asio::buffer(m_disk_recv_buffer.get()
|
2008-04-10 12:03:23 +02:00
|
|
|
+ m_recv_pos - regular_buffer_size, (std::min)(m_packet_size
|
|
|
|
- m_recv_pos, max_receive)), ec);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// receive into both regular and disk buffer
|
|
|
|
TORRENT_ASSERT(max_receive + m_recv_pos > regular_buffer_size);
|
|
|
|
TORRENT_ASSERT(m_recv_pos < regular_buffer_size);
|
|
|
|
TORRENT_ASSERT(max_receive - regular_buffer_size
|
|
|
|
+ m_recv_pos <= m_disk_recv_buffer_size);
|
|
|
|
|
|
|
|
boost::array<asio::mutable_buffer, 2> vec;
|
|
|
|
vec[0] = asio::buffer(&m_recv_buffer[m_recv_pos]
|
|
|
|
, regular_buffer_size - m_recv_pos);
|
2008-05-05 08:25:22 +02:00
|
|
|
vec[1] = asio::buffer(m_disk_recv_buffer.get()
|
2008-04-10 12:03:23 +02:00
|
|
|
, (std::min)(m_disk_recv_buffer_size
|
|
|
|
, max_receive - regular_buffer_size + m_recv_pos));
|
|
|
|
bytes_transferred = m_socket->read_some(vec, ec);
|
|
|
|
}
|
2007-05-10 03:50:11 +02:00
|
|
|
if (ec && ec != asio::error::would_block)
|
2008-01-07 02:10:46 +01:00
|
|
|
{
|
2008-12-28 21:06:59 +01:00
|
|
|
m_statistics.trancieve_ip_packet(bytes_in_loop, m_remote.address().is_v6());
|
2008-01-07 02:10:46 +01:00
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
2008-04-10 12:03:23 +02:00
|
|
|
if (ec == asio::error::would_block) break;
|
2008-12-28 21:06:59 +01:00
|
|
|
bytes_in_loop += bytes_transferred;
|
2007-04-25 19:36:12 +02:00
|
|
|
}
|
2007-05-10 03:50:11 +02:00
|
|
|
while (bytes_transferred > 0);
|
2007-04-25 19:36:12 +02:00
|
|
|
|
2008-12-28 21:06:59 +01:00
|
|
|
m_statistics.trancieve_ip_packet(bytes_in_loop, m_remote.address().is_v6());
|
2006-04-25 23:04:48 +02:00
|
|
|
setup_receive();
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-03-28 19:45:37 +02:00
|
|
|
bool peer_connection::can_write() const
|
2004-01-04 05:29:13 +01:00
|
|
|
{
|
|
|
|
// if we have requests or pending data to be sent or announcements to be made
|
|
|
|
// we want to send data
|
2007-09-29 18:14:03 +02:00
|
|
|
return !m_send_buffer.empty()
|
2009-04-26 02:21:59 +02:00
|
|
|
&& (m_quota[upload_channel] > 0
|
2007-05-25 19:06:30 +02:00
|
|
|
|| m_ignore_bandwidth_limits)
|
2006-04-25 23:04:48 +02:00
|
|
|
&& !m_connecting;
|
2004-03-28 19:45:37 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool peer_connection::can_read() const
|
|
|
|
{
|
2009-04-26 02:21:59 +02:00
|
|
|
bool ret = (m_quota[download_channel] > 0
|
2007-05-25 19:06:30 +02:00
|
|
|
|| m_ignore_bandwidth_limits)
|
2007-08-01 07:22:34 +02:00
|
|
|
&& !m_connecting
|
|
|
|
&& m_outstanding_writing_bytes <
|
|
|
|
m_ses.settings().max_outstanding_disk_bytes_per_connection;
|
2007-08-03 10:19:10 +02:00
|
|
|
|
|
|
|
return ret;
|
2004-01-04 05:29:13 +01:00
|
|
|
}
|
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
void peer_connection::connect(int ticket)
|
2005-11-02 17:28:39 +01:00
|
|
|
{
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-10-06 01:36:50 +02:00
|
|
|
// in case we disconnect here, we need to
|
|
|
|
// keep the connection alive until the
|
|
|
|
// exit invariant check is run
|
|
|
|
boost::intrusive_ptr<peer_connection> me(self());
|
|
|
|
#endif
|
2005-11-02 17:28:39 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-05-03 18:05:42 +02:00
|
|
|
error_code ec;
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-01-07 02:10:46 +01:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " CONNECTING: " << m_remote.address().to_string(ec)
|
2007-05-25 19:06:30 +02:00
|
|
|
<< ":" << m_remote.port() << "\n";
|
2005-11-02 17:28:39 +01:00
|
|
|
#endif
|
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
m_connection_ticket = ticket;
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
|
2005-11-02 17:28:39 +01:00
|
|
|
m_queued = false;
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_connecting);
|
2008-03-16 11:49:47 +01:00
|
|
|
|
|
|
|
if (!t)
|
|
|
|
{
|
|
|
|
disconnect("torrent aborted");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2009-03-22 21:34:57 +01:00
|
|
|
m_socket->open(m_remote.protocol(), ec);
|
2008-01-07 02:10:46 +01:00
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
2007-05-10 03:50:11 +02:00
|
|
|
|
|
|
|
// set the socket to non-blocking, so that we can
|
|
|
|
// read the entire buffer on each read event we get
|
|
|
|
tcp::socket::non_blocking_io ioc(true);
|
2008-01-07 02:10:46 +01:00
|
|
|
m_socket->io_control(ioc, ec);
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
2008-09-25 09:40:55 +02:00
|
|
|
|
|
|
|
tcp::endpoint bind_interface = t->get_interface();
|
|
|
|
|
|
|
|
std::pair<int, int> const& out_ports = m_ses.settings().outgoing_ports;
|
|
|
|
if (out_ports.first > 0 && out_ports.second >= out_ports.first)
|
|
|
|
{
|
|
|
|
m_socket->set_option(socket_acceptor::reuse_address(true), ec);
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
bind_interface.port(m_ses.next_port());
|
|
|
|
}
|
|
|
|
|
2009-04-04 12:08:07 +02:00
|
|
|
// if we're not binding to a specific interface, bind
|
|
|
|
// to the same protocol family as the target endpoint
|
|
|
|
if (is_any(bind_interface.address()))
|
|
|
|
{
|
|
|
|
if (m_remote.address().is_v4())
|
|
|
|
bind_interface.address(address_v4::any());
|
|
|
|
else
|
|
|
|
bind_interface.address(address_v6::any());
|
|
|
|
}
|
|
|
|
|
2008-09-25 09:40:55 +02:00
|
|
|
m_socket->bind(bind_interface, ec);
|
2008-01-07 02:10:46 +01:00
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
disconnect(ec.message().c_str());
|
|
|
|
return;
|
|
|
|
}
|
2007-04-25 20:26:35 +02:00
|
|
|
m_socket->async_connect(m_remote
|
|
|
|
, bind(&peer_connection::on_connection_complete, self(), _1));
|
2008-02-09 23:42:56 +01:00
|
|
|
m_connect = time_now();
|
2008-11-14 21:51:49 +01:00
|
|
|
m_statistics.sent_syn(m_remote.address().is_v6());
|
2005-11-02 17:28:39 +01:00
|
|
|
|
2008-07-06 14:22:56 +02:00
|
|
|
if (t->alerts().should_post<peer_connect_alert>())
|
2005-11-02 17:28:39 +01:00
|
|
|
{
|
2008-07-06 14:22:56 +02:00
|
|
|
t->alerts().post_alert(peer_connect_alert(
|
2008-07-08 11:30:10 +02:00
|
|
|
t->get_handle(), remote(), pid()));
|
2005-11-02 17:28:39 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-05-03 18:05:42 +02:00
|
|
|
void peer_connection::on_connection_complete(error_code const& e)
|
2005-11-02 17:28:39 +01:00
|
|
|
{
|
2008-02-09 23:42:56 +01:00
|
|
|
ptime completed = time_now();
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2005-11-02 17:28:39 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-09 23:42:56 +01:00
|
|
|
m_rtt = total_milliseconds(completed - m_connect);
|
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
if (m_disconnecting) return;
|
|
|
|
|
|
|
|
m_connecting = false;
|
|
|
|
m_ses.m_half_open.done(m_connection_ticket);
|
|
|
|
|
2008-10-23 06:10:23 +02:00
|
|
|
error_code ec;
|
2006-04-25 23:04:48 +02:00
|
|
|
if (e)
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2008-10-23 06:10:23 +02:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " CONNECTION FAILED: " << m_remote.address().to_string(ec)
|
2006-11-15 22:39:58 +01:00
|
|
|
<< ": " << e.message() << "\n";
|
2005-11-02 17:28:39 +01:00
|
|
|
#endif
|
2008-05-12 08:03:31 +02:00
|
|
|
disconnect(e.message().c_str(), 1);
|
2006-04-25 23:04:48 +02:00
|
|
|
return;
|
|
|
|
}
|
2005-11-02 17:28:39 +01:00
|
|
|
|
2006-05-29 19:50:57 +02:00
|
|
|
if (m_disconnecting) return;
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_receive = time_now();
|
2006-05-28 21:03:54 +02:00
|
|
|
|
2005-11-02 17:28:39 +01:00
|
|
|
// this means the connection just succeeded
|
|
|
|
|
2008-11-14 21:51:49 +01:00
|
|
|
m_statistics.received_synack(m_remote.address().is_v6());
|
2008-09-26 08:10:21 +02:00
|
|
|
|
2008-06-16 15:54:14 +02:00
|
|
|
TORRENT_ASSERT(m_socket);
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING
|
2008-10-23 06:10:23 +02:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " COMPLETED: " << m_remote.address().to_string(ec)
|
2008-02-09 23:42:56 +01:00
|
|
|
<< " rtt = " << m_rtt << "\n";
|
2006-04-25 23:04:48 +02:00
|
|
|
#endif
|
|
|
|
|
2008-06-16 15:54:14 +02:00
|
|
|
if (m_remote == m_socket->local_endpoint(ec))
|
|
|
|
{
|
|
|
|
// if the remote endpoint is the same as the local endpoint, we're connected
|
|
|
|
// to ourselves
|
|
|
|
disconnect("connected to ourselves", 1);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2008-03-12 17:58:23 +01:00
|
|
|
if (m_remote.address().is_v4())
|
|
|
|
{
|
2008-05-03 18:05:42 +02:00
|
|
|
error_code ec;
|
2008-03-12 17:58:23 +01:00
|
|
|
m_socket->set_option(type_of_service(m_ses.settings().peer_tos), ec);
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
on_connected();
|
2006-05-02 01:34:37 +02:00
|
|
|
setup_send();
|
2007-01-10 16:02:25 +01:00
|
|
|
setup_receive();
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2005-11-02 17:28:39 +01:00
|
|
|
|
2004-01-04 05:29:13 +01:00
|
|
|
// --------------------------
|
|
|
|
// SEND DATA
|
|
|
|
// --------------------------
|
|
|
|
|
2008-05-03 18:05:42 +02:00
|
|
|
void peer_connection::on_send_data(error_code const& error
|
2008-04-07 10:39:01 +02:00
|
|
|
, std::size_t bytes_transferred)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
2004-01-25 19:18:36 +01:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2009-01-25 03:45:31 +01:00
|
|
|
// keep ourselves alive in until this function exits in
|
|
|
|
// case we disconnect
|
|
|
|
boost::intrusive_ptr<peer_connection> me(self());
|
|
|
|
|
2008-01-14 00:46:43 +01:00
|
|
|
TORRENT_ASSERT(m_channel_state[upload_channel] == peer_info::bw_network);
|
2007-09-29 18:14:03 +02:00
|
|
|
|
|
|
|
m_send_buffer.pop_front(bytes_transferred);
|
2008-07-08 02:03:08 +02:00
|
|
|
|
|
|
|
for (std::vector<int>::iterator i = m_requests_in_buffer.begin()
|
|
|
|
, end(m_requests_in_buffer.end()); i != end; ++i)
|
|
|
|
*i -= bytes_transferred;
|
|
|
|
|
|
|
|
while (!m_requests_in_buffer.empty()
|
|
|
|
&& m_requests_in_buffer.front() <= 0)
|
|
|
|
m_requests_in_buffer.erase(m_requests_in_buffer.begin());
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2008-01-14 00:46:43 +01:00
|
|
|
m_channel_state[upload_channel] = peer_info::bw_idle;
|
2007-01-10 16:02:25 +01:00
|
|
|
|
2007-05-25 19:06:30 +02:00
|
|
|
if (!m_ignore_bandwidth_limits)
|
2009-04-26 02:21:59 +02:00
|
|
|
{
|
|
|
|
TORRENT_ASSERT(bytes_transferred <= m_quota[upload_channel]);
|
|
|
|
m_quota[upload_channel] -= bytes_transferred;
|
|
|
|
}
|
2007-05-25 19:06:30 +02:00
|
|
|
|
2008-11-14 21:51:49 +01:00
|
|
|
m_statistics.trancieve_ip_packet(bytes_transferred, m_remote.address().is_v6());
|
|
|
|
|
2007-05-25 19:06:30 +02:00
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << "wrote " << bytes_transferred << " bytes\n";
|
|
|
|
#endif
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (error)
|
2006-05-29 19:50:57 +02:00
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-06-06 02:41:20 +02:00
|
|
|
(*m_logger) << "**ERROR**: " << error.message() << " [in peer_connection::on_send_data]\n";
|
2006-05-29 19:50:57 +02:00
|
|
|
#endif
|
2008-01-07 02:10:46 +01:00
|
|
|
disconnect(error.message().c_str());
|
|
|
|
return;
|
2006-05-29 19:50:57 +02:00
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
if (m_disconnecting) return;
|
2005-09-14 21:33:16 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_connecting);
|
|
|
|
TORRENT_ASSERT(bytes_transferred > 0);
|
2004-01-04 05:29:13 +01:00
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
m_last_sent = time_now();
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-09-21 19:12:26 +02:00
|
|
|
size_type cur_payload_ul = m_statistics.last_payload_uploaded();
|
|
|
|
size_type cur_protocol_ul = m_statistics.last_protocol_uploaded();
|
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
on_sent(error, bytes_transferred);
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2008-09-21 19:12:26 +02:00
|
|
|
TORRENT_ASSERT(m_statistics.last_payload_uploaded() - cur_payload_ul >= 0);
|
|
|
|
TORRENT_ASSERT(m_statistics.last_protocol_uploaded() - cur_protocol_ul >= 0);
|
|
|
|
size_type stats_diff = m_statistics.last_payload_uploaded() - cur_payload_ul
|
|
|
|
+ m_statistics.last_protocol_uploaded() - cur_protocol_ul;
|
|
|
|
TORRENT_ASSERT(stats_diff == bytes_transferred);
|
|
|
|
#endif
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
fill_send_buffer();
|
2007-04-25 19:36:12 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
setup_send();
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2008-11-29 22:33:21 +01:00
|
|
|
#ifdef TORRENT_DEBUG
|
2006-04-25 23:04:48 +02:00
|
|
|
void peer_connection::check_invariant() const
|
|
|
|
{
|
2008-05-05 08:25:22 +02:00
|
|
|
TORRENT_ASSERT(bool(m_disk_recv_buffer) == (m_disk_recv_buffer_size > 0));
|
2008-04-10 12:03:23 +02:00
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
TORRENT_ASSERT(m_upload_limit >= 0);
|
|
|
|
TORRENT_ASSERT(m_download_limit >= 0);
|
|
|
|
|
2008-01-31 18:52:29 +01:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
2008-01-19 20:00:54 +01:00
|
|
|
if (m_disconnecting)
|
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
TORRENT_ASSERT(!t);
|
2008-07-18 12:03:42 +02:00
|
|
|
TORRENT_ASSERT(m_disconnect_started);
|
2008-01-19 20:00:54 +01:00
|
|
|
}
|
|
|
|
else if (!m_in_constructor)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(m_ses.has_peer((peer_connection*)this));
|
|
|
|
}
|
|
|
|
|
2009-04-26 21:44:08 +02:00
|
|
|
TORRENT_ASSERT(m_outstanding_bytes >= 0);
|
2009-03-31 20:01:12 +02:00
|
|
|
if (t && t->valid_metadata() && !m_disconnecting)
|
2009-03-17 10:34:44 +01:00
|
|
|
{
|
|
|
|
boost::optional<piece_block_progress> p = t?downloading_piece_progress():boost::optional<piece_block_progress>();
|
|
|
|
torrent_info const& ti = t->torrent_file();
|
|
|
|
// if the piece is fully downloaded, we might have popped it from the
|
|
|
|
// download queue already
|
|
|
|
int outstanding_bytes = 0;
|
|
|
|
bool in_download_queue = false;
|
|
|
|
int block_size = t->block_size();
|
|
|
|
piece_block last_block(ti.num_pieces()-1, (ti.total_size() % ti.piece_length())
|
|
|
|
/ block_size);
|
|
|
|
int last_block_size = t->torrent_file().total_size() & (block_size-1);
|
|
|
|
if (last_block_size == 0) last_block_size = block_size;
|
|
|
|
for (std::vector<pending_block>::const_iterator i = m_download_queue.begin()
|
|
|
|
, end(m_download_queue.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if (p && i->block == piece_block(p->piece_index, p->block_index))
|
|
|
|
{
|
|
|
|
in_download_queue = true;
|
|
|
|
outstanding_bytes += p->full_block_bytes - m_received_in_piece;
|
|
|
|
}
|
|
|
|
else if (i->block == last_block)
|
|
|
|
{
|
|
|
|
outstanding_bytes += last_block_size;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
outstanding_bytes += block_size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
//if (p && p->bytes_downloaded < p->full_block_bytes) TORRENT_ASSERT(in_download_queue);
|
|
|
|
|
|
|
|
TORRENT_ASSERT(m_outstanding_bytes == outstanding_bytes);
|
|
|
|
}
|
|
|
|
|
2009-04-26 02:21:59 +02:00
|
|
|
if (m_channel_state[download_channel] == peer_info::bw_limit)
|
|
|
|
TORRENT_ASSERT(m_quota[download_channel] == 0);
|
|
|
|
if (m_channel_state[upload_channel] == peer_info::bw_limit)
|
|
|
|
TORRENT_ASSERT(m_quota[upload_channel] == 0);
|
2008-01-10 23:13:23 +01:00
|
|
|
|
2008-01-02 04:18:29 +01:00
|
|
|
std::set<piece_block> unique;
|
2008-07-07 14:04:06 +02:00
|
|
|
std::transform(m_download_queue.begin(), m_download_queue.end()
|
|
|
|
, std::inserter(unique, unique.begin()), boost::bind(&pending_block::block, _1));
|
2008-01-02 04:18:29 +01:00
|
|
|
std::copy(m_request_queue.begin(), m_request_queue.end(), std::inserter(unique, unique.begin()));
|
|
|
|
TORRENT_ASSERT(unique.size() == m_download_queue.size() + m_request_queue.size());
|
2007-04-12 12:21:55 +02:00
|
|
|
if (m_peer_info)
|
2007-08-21 19:45:28 +02:00
|
|
|
{
|
2008-04-07 10:15:31 +02:00
|
|
|
TORRENT_ASSERT(m_peer_info->prev_amount_upload == 0);
|
|
|
|
TORRENT_ASSERT(m_peer_info->prev_amount_download == 0);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_peer_info->connection == this
|
2007-04-12 12:21:55 +02:00
|
|
|
|| m_peer_info->connection == 0);
|
2007-08-21 19:45:28 +02:00
|
|
|
|
|
|
|
if (m_peer_info->optimistically_unchoked)
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!is_choked());
|
2007-08-21 19:45:28 +02:00
|
|
|
}
|
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
TORRENT_ASSERT(m_have_piece.count() == m_num_pieces);
|
|
|
|
|
2008-01-31 18:52:29 +01:00
|
|
|
if (!t)
|
|
|
|
{
|
2008-07-12 19:51:59 +02:00
|
|
|
#ifdef TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2008-01-31 18:52:29 +01:00
|
|
|
// since this connection doesn't have a torrent reference
|
|
|
|
// no torrent should have a reference to this connection either
|
|
|
|
for (aux::session_impl::torrent_map::const_iterator i = m_ses.m_torrents.begin()
|
|
|
|
, end(m_ses.m_torrents.end()); i != end; ++i)
|
|
|
|
TORRENT_ASSERT(!i->second->has_peer((peer_connection*)this));
|
2008-07-12 19:51:59 +02:00
|
|
|
#endif
|
2008-01-31 18:52:29 +01:00
|
|
|
return;
|
|
|
|
}
|
2007-10-31 10:48:20 +01:00
|
|
|
|
2008-08-29 19:21:56 +02:00
|
|
|
if (t->ready_for_connections() && m_initialized)
|
2008-09-25 22:12:53 +02:00
|
|
|
TORRENT_ASSERT(t->torrent_file().num_pieces() == int(m_have_piece.size()));
|
2008-08-29 19:21:56 +02:00
|
|
|
|
2008-07-18 12:03:42 +02:00
|
|
|
if (m_ses.settings().close_redundant_connections)
|
|
|
|
{
|
|
|
|
// make sure upload only peers are disconnected
|
|
|
|
if (t->is_finished() && m_upload_only)
|
|
|
|
TORRENT_ASSERT(m_disconnect_started);
|
|
|
|
if (m_upload_only
|
|
|
|
&& !m_interesting
|
|
|
|
&& m_bitfield_received
|
|
|
|
&& t->are_files_checked())
|
|
|
|
TORRENT_ASSERT(m_disconnect_started);
|
|
|
|
}
|
|
|
|
|
2008-10-06 00:09:33 +02:00
|
|
|
if (!m_disconnect_started && m_initialized)
|
2008-08-29 19:21:56 +02:00
|
|
|
{
|
|
|
|
// none of this matters if we're disconnecting anyway
|
|
|
|
if (t->is_finished())
|
|
|
|
TORRENT_ASSERT(!m_interesting);
|
|
|
|
if (is_seed())
|
|
|
|
TORRENT_ASSERT(m_upload_only);
|
|
|
|
}
|
2008-07-18 12:03:42 +02:00
|
|
|
|
2008-03-12 09:36:22 +01:00
|
|
|
if (t->has_picker())
|
|
|
|
{
|
|
|
|
std::map<piece_block, int> num_requests;
|
|
|
|
for (torrent::const_peer_iterator i = t->begin(); i != t->end(); ++i)
|
|
|
|
{
|
|
|
|
// make sure this peer is not a dangling pointer
|
2008-07-12 19:51:59 +02:00
|
|
|
#ifdef TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2008-03-12 09:36:22 +01:00
|
|
|
TORRENT_ASSERT(m_ses.has_peer(*i));
|
2008-07-12 19:51:59 +02:00
|
|
|
#endif
|
2008-03-12 09:36:22 +01:00
|
|
|
peer_connection const& p = *(*i);
|
2009-03-17 10:34:44 +01:00
|
|
|
for (std::vector<piece_block>::const_iterator i = p.request_queue().begin()
|
2008-03-12 09:36:22 +01:00
|
|
|
, end(p.request_queue().end()); i != end; ++i)
|
|
|
|
++num_requests[*i];
|
2009-03-17 10:34:44 +01:00
|
|
|
for (std::vector<pending_block>::const_iterator i = p.download_queue().begin()
|
2008-03-12 09:36:22 +01:00
|
|
|
, end(p.download_queue().end()); i != end; ++i)
|
2009-03-17 10:34:44 +01:00
|
|
|
if (!i->not_wanted && !i->timed_out) ++num_requests[i->block];
|
2008-03-12 09:36:22 +01:00
|
|
|
}
|
|
|
|
for (std::map<piece_block, int>::iterator i = num_requests.begin()
|
|
|
|
, end(num_requests.end()); i != end; ++i)
|
|
|
|
{
|
2009-03-17 10:34:44 +01:00
|
|
|
piece_block b = i->first;
|
|
|
|
int count = i->second;
|
|
|
|
int picker_count = t->picker().num_peers(b);
|
|
|
|
if (!t->picker().is_downloaded(b))
|
|
|
|
TORRENT_ASSERT(picker_count == count);
|
2008-03-12 09:36:22 +01:00
|
|
|
}
|
|
|
|
}
|
2008-07-12 19:51:59 +02:00
|
|
|
#ifdef TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2007-10-31 10:48:20 +01:00
|
|
|
if (m_peer_info)
|
2006-05-28 21:03:54 +02:00
|
|
|
{
|
2008-07-30 08:42:08 +02:00
|
|
|
policy::const_iterator i = t->get_policy().begin_peer();
|
|
|
|
policy::const_iterator end = t->get_policy().end_peer();
|
|
|
|
for (; i != end; ++i)
|
2006-05-28 21:03:54 +02:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
if (&i->second == m_peer_info) break;
|
2006-05-28 21:03:54 +02:00
|
|
|
}
|
2008-07-30 08:42:08 +02:00
|
|
|
TORRENT_ASSERT(i != end);
|
2006-05-28 21:03:54 +02:00
|
|
|
}
|
2008-07-12 19:51:59 +02:00
|
|
|
#endif
|
2007-10-18 06:18:09 +02:00
|
|
|
if (t->has_picker() && !t->is_aborted())
|
2007-10-18 02:32:16 +02:00
|
|
|
{
|
|
|
|
// make sure that pieces that have completed the download
|
|
|
|
// of all their blocks are in the disk io thread's queue
|
|
|
|
// to be checked.
|
|
|
|
const std::vector<piece_picker::downloading_piece>& dl_queue
|
|
|
|
= t->picker().get_download_queue();
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator i =
|
|
|
|
dl_queue.begin(); i != dl_queue.end(); ++i)
|
|
|
|
{
|
|
|
|
const int blocks_per_piece = t->picker().blocks_in_piece(i->index);
|
|
|
|
|
|
|
|
bool complete = true;
|
|
|
|
for (int j = 0; j < blocks_per_piece; ++j)
|
|
|
|
{
|
|
|
|
if (i->info[j].state == piece_picker::block_info::state_finished)
|
|
|
|
continue;
|
|
|
|
complete = false;
|
|
|
|
break;
|
|
|
|
}
|
2008-02-08 11:22:05 +01:00
|
|
|
/*
|
|
|
|
// this invariant is not valid anymore since the completion event
|
|
|
|
// might be queued in the io service
|
2007-12-18 07:04:54 +01:00
|
|
|
if (complete && !piece_failed)
|
2007-10-18 02:32:16 +02:00
|
|
|
{
|
|
|
|
disk_io_job ret = m_ses.m_disk_thread.find_job(
|
|
|
|
&t->filesystem(), -1, i->index);
|
|
|
|
TORRENT_ASSERT(ret.action == disk_io_job::hash || ret.action == disk_io_job::write);
|
|
|
|
TORRENT_ASSERT(ret.piece == i->index);
|
|
|
|
}
|
2008-02-08 11:22:05 +01:00
|
|
|
*/
|
2007-10-18 02:32:16 +02:00
|
|
|
}
|
|
|
|
}
|
2006-12-16 04:34:56 +01:00
|
|
|
|
2006-12-18 02:24:35 +01:00
|
|
|
// extremely expensive invariant check
|
|
|
|
/*
|
2006-12-16 04:34:56 +01:00
|
|
|
if (!t->is_seed())
|
|
|
|
{
|
|
|
|
piece_picker& p = t->picker();
|
|
|
|
const std::vector<piece_picker::downloading_piece>& dlq = p.get_download_queue();
|
|
|
|
const int blocks_per_piece = static_cast<int>(
|
|
|
|
t->torrent_file().piece_length() / t->block_size());
|
|
|
|
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator i =
|
|
|
|
dlq.begin(); i != dlq.end(); ++i)
|
|
|
|
{
|
|
|
|
for (int j = 0; j < blocks_per_piece; ++j)
|
|
|
|
{
|
|
|
|
if (std::find(m_request_queue.begin(), m_request_queue.end()
|
|
|
|
, piece_block(i->index, j)) != m_request_queue.end()
|
|
|
|
||
|
|
|
|
std::find(m_download_queue.begin(), m_download_queue.end()
|
|
|
|
, piece_block(i->index, j)) != m_download_queue.end())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i->info[j].peer == m_remote);
|
2006-12-16 04:34:56 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i->info[j].peer != m_remote || i->info[j].finished);
|
2006-12-16 04:34:56 +01:00
|
|
|
}
|
2006-12-16 04:36:33 +01:00
|
|
|
}
|
2006-12-16 04:34:56 +01:00
|
|
|
}
|
|
|
|
}
|
2006-12-18 02:24:35 +01:00
|
|
|
*/
|
2003-11-05 00:27:06 +01:00
|
|
|
}
|
2004-01-25 19:18:36 +01:00
|
|
|
#endif
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2007-04-27 02:27:37 +02:00
|
|
|
peer_connection::peer_speed_t peer_connection::peer_speed()
|
|
|
|
{
|
|
|
|
shared_ptr<torrent> t = m_torrent.lock();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-04-27 02:27:37 +02:00
|
|
|
|
2007-04-27 18:26:30 +02:00
|
|
|
int download_rate = int(statistics().download_payload_rate());
|
|
|
|
int torrent_download_rate = int(t->statistics().download_payload_rate());
|
2007-04-27 02:27:37 +02:00
|
|
|
|
|
|
|
if (download_rate > 512 && download_rate > torrent_download_rate / 16)
|
|
|
|
m_speed = fast;
|
|
|
|
else if (download_rate > 4096 && download_rate > torrent_download_rate / 64)
|
|
|
|
m_speed = medium;
|
|
|
|
else if (download_rate < torrent_download_rate / 15 && m_speed == fast)
|
|
|
|
m_speed = medium;
|
2008-09-06 23:04:57 +02:00
|
|
|
else
|
2007-04-27 02:27:37 +02:00
|
|
|
m_speed = slow;
|
|
|
|
|
|
|
|
return m_speed;
|
|
|
|
}
|
2004-02-01 14:48:30 +01:00
|
|
|
|
2004-01-04 05:29:13 +01:00
|
|
|
void peer_connection::keep_alive()
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2008-10-10 09:08:46 +02:00
|
|
|
#ifdef TORRENT_EXPENSIVE_INVARIANT_CHECKS
|
2004-01-25 19:18:36 +01:00
|
|
|
INVARIANT_CHECK;
|
2008-10-10 09:08:46 +02:00
|
|
|
#endif
|
2004-01-25 19:18:36 +01:00
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
time_duration d;
|
|
|
|
d = time_now() - m_last_sent;
|
|
|
|
if (total_seconds(d) < m_timeout / 2) return;
|
2007-06-06 02:41:20 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
if (m_connecting) return;
|
2007-06-06 02:41:20 +02:00
|
|
|
if (in_handshake()) return;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-06-06 02:41:20 +02:00
|
|
|
// if the last send has not completed yet, do not send a keep
|
|
|
|
// alive
|
2008-01-14 00:46:43 +01:00
|
|
|
if (m_channel_state[upload_channel] != peer_info::bw_idle) return;
|
2007-06-06 02:41:20 +02:00
|
|
|
|
|
|
|
#ifdef TORRENT_VERBOSE_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " ==> KEEPALIVE\n";
|
|
|
|
#endif
|
2007-10-22 06:56:09 +02:00
|
|
|
|
2007-08-03 10:19:10 +02:00
|
|
|
m_last_sent = time_now();
|
2006-04-25 23:04:48 +02:00
|
|
|
write_keepalive();
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
2004-01-12 21:31:27 +01:00
|
|
|
|
|
|
|
bool peer_connection::is_seed() const
|
|
|
|
{
|
2007-05-27 00:27:40 +02:00
|
|
|
// if m_num_pieces == 0, we probably don't have the
|
2004-10-18 00:23:08 +02:00
|
|
|
// metadata yet.
|
2008-08-29 19:21:56 +02:00
|
|
|
boost::shared_ptr<torrent> t = m_torrent.lock();
|
|
|
|
return m_num_pieces == (int)m_have_piece.size() && m_num_pieces > 0 && t && t->valid_metadata();
|
2004-01-12 21:31:27 +01:00
|
|
|
}
|
2004-01-04 06:53:01 +01:00
|
|
|
}
|
2005-03-19 13:22:40 +01:00
|
|
|
|