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"
|
|
|
|
|
2003-10-23 01:00:57 +02:00
|
|
|
#include <ctime>
|
|
|
|
#include <iostream>
|
|
|
|
#include <fstream>
|
|
|
|
#include <iomanip>
|
|
|
|
#include <iterator>
|
|
|
|
#include <algorithm>
|
|
|
|
#include <set>
|
|
|
|
#include <cctype>
|
2003-10-30 00:28:09 +01:00
|
|
|
#include <numeric>
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-25 19:18:36 +01:00
|
|
|
#ifdef _MSC_VER
|
|
|
|
#pragma warning(push, 1)
|
|
|
|
#endif
|
|
|
|
|
2003-10-23 01:00:57 +02:00
|
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
#include <boost/filesystem/convenience.hpp>
|
2004-08-05 15:56:26 +02:00
|
|
|
#include <boost/bind.hpp>
|
2006-04-25 23:04:48 +02:00
|
|
|
#include <boost/thread/mutex.hpp>
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-25 19:18:36 +01:00
|
|
|
#ifdef _MSC_VER
|
|
|
|
#pragma warning(pop)
|
|
|
|
#endif
|
|
|
|
|
2003-10-26 18:35:23 +01:00
|
|
|
#include "libtorrent/torrent_handle.hpp"
|
2003-10-23 01:00:57 +02:00
|
|
|
#include "libtorrent/session.hpp"
|
|
|
|
#include "libtorrent/torrent_info.hpp"
|
2004-01-31 11:46:15 +01:00
|
|
|
#include "libtorrent/tracker_manager.hpp"
|
2003-10-23 01:00:57 +02:00
|
|
|
#include "libtorrent/bencode.hpp"
|
|
|
|
#include "libtorrent/hasher.hpp"
|
|
|
|
#include "libtorrent/entry.hpp"
|
|
|
|
#include "libtorrent/peer.hpp"
|
2006-04-25 23:04:48 +02:00
|
|
|
#include "libtorrent/bt_peer_connection.hpp"
|
|
|
|
#include "libtorrent/web_peer_connection.hpp"
|
2003-10-23 01:00:57 +02:00
|
|
|
#include "libtorrent/peer_id.hpp"
|
2003-12-22 08:14:35 +01:00
|
|
|
#include "libtorrent/alert.hpp"
|
|
|
|
#include "libtorrent/identify_client.hpp"
|
2004-01-18 20:12:18 +01:00
|
|
|
#include "libtorrent/alert_types.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-25 20:26:35 +02:00
|
|
|
#include "libtorrent/instantiate_connection.hpp"
|
2007-09-10 08:12:41 +02:00
|
|
|
#include "libtorrent/assert.hpp"
|
2003-10-23 01:00:57 +02:00
|
|
|
|
|
|
|
using namespace libtorrent;
|
2006-01-07 14:48:14 +01:00
|
|
|
using boost::tuples::tuple;
|
|
|
|
using boost::tuples::get;
|
|
|
|
using boost::tuples::make_tuple;
|
2006-01-09 01:07:00 +01:00
|
|
|
using boost::bind;
|
2006-04-25 23:04:48 +02:00
|
|
|
using boost::mutex;
|
2006-10-11 16:02:21 +02:00
|
|
|
using libtorrent::aux::session_impl;
|
2003-10-23 01:00:57 +02:00
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
|
|
|
|
enum
|
|
|
|
{
|
|
|
|
// wait 60 seconds before retrying a failed tracker
|
2004-04-02 00:29:51 +02:00
|
|
|
tracker_retry_delay_min = 60
|
|
|
|
// when tracker_failed_max trackers
|
|
|
|
// has failed, wait 10 minutes instead
|
|
|
|
, tracker_retry_delay_max = 10 * 60
|
|
|
|
, tracker_failed_max = 5
|
2003-10-23 01:00:57 +02:00
|
|
|
};
|
|
|
|
|
2005-07-10 12:42:00 +02:00
|
|
|
int calculate_block_size(const torrent_info& i, int default_block_size)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2005-07-10 12:42:00 +02:00
|
|
|
if (default_block_size < 1024) default_block_size = 1024;
|
2003-12-21 18:28:27 +01:00
|
|
|
|
|
|
|
// if pieces are too small, adjust the block size
|
|
|
|
if (i.piece_length() < default_block_size)
|
|
|
|
{
|
2008-04-05 23:18:27 +02:00
|
|
|
return i.piece_length();
|
2003-12-21 18:28:27 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// otherwise, go with the default
|
|
|
|
return default_block_size;
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2004-01-12 04:05:10 +01:00
|
|
|
struct find_peer_by_ip
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
find_peer_by_ip(tcp::endpoint const& a, const torrent* t)
|
2004-01-25 13:37:15 +01:00
|
|
|
: ip(a)
|
|
|
|
, tor(t)
|
2007-10-05 02:30:00 +02:00
|
|
|
{ TORRENT_ASSERT(t != 0); }
|
2004-01-12 04:05:10 +01:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
bool operator()(session_impl::connection_map::value_type const& c) const
|
2004-01-12 04:05:10 +01:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
tcp::endpoint const& sender = c->remote();
|
2006-04-25 23:04:48 +02:00
|
|
|
if (sender.address() != ip.address()) return false;
|
2007-10-31 10:48:20 +01:00
|
|
|
if (tor != c->associated_torrent().lock().get()) return false;
|
2004-01-12 04:05:10 +01:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
tcp::endpoint const& ip;
|
|
|
|
torrent const* tor;
|
2004-01-12 04:05:10 +01:00
|
|
|
};
|
|
|
|
|
2004-01-13 04:08:59 +01:00
|
|
|
struct peer_by_id
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
peer_by_id(const peer_id& i): pid(i) {}
|
2004-01-13 04:08:59 +01:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
bool operator()(session_impl::connection_map::value_type const& p) const
|
2004-01-13 04:08:59 +01:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
if (p->pid() != pid) return false;
|
2004-01-13 04:08:59 +01:00
|
|
|
// have a special case for all zeros. We can have any number
|
2006-04-25 23:04:48 +02:00
|
|
|
// of peers with that pid, since it's used to indicate no pid.
|
|
|
|
if (std::count(pid.begin(), pid.end(), 0) == 20) return false;
|
2004-01-13 04:08:59 +01:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
peer_id const& pid;
|
2004-01-13 04:08:59 +01:00
|
|
|
};
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
namespace libtorrent
|
|
|
|
{
|
2007-06-10 22:46:09 +02:00
|
|
|
|
2003-12-07 06:53:04 +01:00
|
|
|
torrent::torrent(
|
2006-10-11 16:02:21 +02:00
|
|
|
session_impl& ses
|
2007-09-01 05:00:31 +02:00
|
|
|
, boost::intrusive_ptr<torrent_info> tf
|
2007-06-10 22:46:09 +02:00
|
|
|
, fs::path const& save_path
|
2006-04-25 23:04:48 +02:00
|
|
|
, tcp::endpoint const& net_interface
|
2007-10-08 22:01:36 +02:00
|
|
|
, storage_mode_t storage_mode
|
2006-04-25 23:04:48 +02:00
|
|
|
, int block_size
|
2007-08-22 07:31:42 +02:00
|
|
|
, storage_constructor_type sc
|
2008-03-08 07:06:31 +01:00
|
|
|
, bool paused
|
|
|
|
, entry const& resume_data)
|
2005-10-16 01:07:39 +02:00
|
|
|
: m_torrent_file(tf)
|
2003-10-23 01:00:57 +02:00
|
|
|
, m_abort(false)
|
2007-08-22 07:31:42 +02:00
|
|
|
, m_paused(paused)
|
2004-07-24 13:54:17 +02:00
|
|
|
, m_just_paused(false)
|
2004-01-21 14:16:11 +01:00
|
|
|
, m_event(tracker_request::started)
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_block_size(0)
|
|
|
|
, m_storage(0)
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_next_request(time_now())
|
2003-10-23 01:00:57 +02:00
|
|
|
, m_duration(1800)
|
2005-02-21 14:59:24 +01:00
|
|
|
, m_complete(-1)
|
|
|
|
, m_incomplete(-1)
|
2006-12-15 18:47:21 +01:00
|
|
|
, m_host_resolver(ses.m_io_service)
|
2007-05-02 21:47:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
2007-01-29 08:39:33 +01:00
|
|
|
, m_resolving_country(false)
|
|
|
|
, m_resolve_countries(false)
|
2007-05-02 21:47:38 +02:00
|
|
|
#endif
|
2007-04-04 04:06:07 +02:00
|
|
|
, m_announce_timer(ses.m_io_service)
|
2007-05-31 02:21:54 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2007-05-17 02:01:51 +02:00
|
|
|
, m_last_dht_announce(time_now() - minutes(15))
|
2007-05-31 02:21:54 +02:00
|
|
|
#endif
|
2003-10-23 01:00:57 +02:00
|
|
|
, m_ses(ses)
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_picker(0)
|
2007-09-01 05:00:31 +02:00
|
|
|
, m_trackers(m_torrent_file->trackers())
|
2004-01-31 11:20:19 +01:00
|
|
|
, m_last_working_tracker(-1)
|
2003-10-23 01:00:57 +02:00
|
|
|
, m_currently_trying_tracker(0)
|
2004-04-02 00:29:51 +02:00
|
|
|
, m_failed_trackers(0)
|
2003-12-01 06:01:40 +01:00
|
|
|
, m_time_scaler(0)
|
2003-12-07 06:53:04 +01:00
|
|
|
, m_num_pieces(0)
|
2008-01-31 18:52:29 +01:00
|
|
|
, m_sequential_download(false)
|
2003-12-22 08:14:35 +01:00
|
|
|
, m_got_tracker_response(false)
|
2004-01-12 04:05:10 +01:00
|
|
|
, m_ratio(0.f)
|
2004-04-18 14:28:02 +02:00
|
|
|
, m_total_failed_bytes(0)
|
2006-04-25 23:04:48 +02:00
|
|
|
, m_total_redundant_bytes(0)
|
2006-05-20 17:30:40 +02:00
|
|
|
, m_net_interface(net_interface.address(), 0)
|
2004-10-18 12:36:47 +02:00
|
|
|
, m_save_path(complete(save_path))
|
2007-10-08 22:01:36 +02:00
|
|
|
, m_storage_mode(storage_mode)
|
2008-03-08 07:06:31 +01:00
|
|
|
, m_state(torrent_status::queued_for_checking)
|
|
|
|
, m_progress(0.f)
|
|
|
|
, m_resume_data(resume_data)
|
2005-07-10 12:42:00 +02:00
|
|
|
, m_default_block_size(block_size)
|
2005-10-16 11:15:46 +02:00
|
|
|
, m_connections_initialized(true)
|
2007-08-22 07:31:42 +02:00
|
|
|
, m_settings(ses.settings())
|
2007-03-16 06:29:23 +01:00
|
|
|
, m_storage_constructor(sc)
|
2007-08-21 06:46:17 +02:00
|
|
|
, m_max_uploads((std::numeric_limits<int>::max)())
|
2007-08-17 09:37:08 +02:00
|
|
|
, m_num_uploads(0)
|
2007-08-21 06:46:17 +02:00
|
|
|
, m_max_connections((std::numeric_limits<int>::max)())
|
2008-03-29 20:39:24 +01:00
|
|
|
, m_deficit_counter(0)
|
2007-10-31 10:48:20 +01:00
|
|
|
, m_policy(this)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2007-12-14 07:09:44 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
m_files_checked = false;
|
|
|
|
#endif
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
torrent::torrent(
|
2006-10-11 16:02:21 +02:00
|
|
|
session_impl& ses
|
2004-06-14 01:30:42 +02:00
|
|
|
, char const* tracker_url
|
|
|
|
, sha1_hash const& info_hash
|
2006-11-14 01:08:16 +01:00
|
|
|
, char const* name
|
2007-06-10 22:46:09 +02:00
|
|
|
, fs::path const& save_path
|
2006-04-25 23:04:48 +02:00
|
|
|
, tcp::endpoint const& net_interface
|
2007-10-08 22:01:36 +02:00
|
|
|
, storage_mode_t storage_mode
|
2006-04-25 23:04:48 +02:00
|
|
|
, int block_size
|
2007-08-22 07:31:42 +02:00
|
|
|
, storage_constructor_type sc
|
2008-03-08 07:06:31 +01:00
|
|
|
, bool paused
|
|
|
|
, entry const& resume_data)
|
2007-09-01 05:00:31 +02:00
|
|
|
: m_torrent_file(new torrent_info(info_hash))
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_abort(false)
|
2007-08-22 07:31:42 +02:00
|
|
|
, m_paused(paused)
|
2004-07-24 13:54:17 +02:00
|
|
|
, m_just_paused(false)
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_event(tracker_request::started)
|
|
|
|
, m_block_size(0)
|
|
|
|
, m_storage(0)
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_next_request(time_now())
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_duration(1800)
|
2005-02-21 14:59:24 +01:00
|
|
|
, m_complete(-1)
|
|
|
|
, m_incomplete(-1)
|
2006-12-15 18:47:21 +01:00
|
|
|
, m_host_resolver(ses.m_io_service)
|
2007-05-02 21:47:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
2007-01-29 08:39:33 +01:00
|
|
|
, m_resolving_country(false)
|
|
|
|
, m_resolve_countries(false)
|
2007-05-02 21:47:38 +02:00
|
|
|
#endif
|
2007-04-04 04:06:07 +02:00
|
|
|
, m_announce_timer(ses.m_io_service)
|
2007-05-31 02:21:54 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2007-05-17 02:01:51 +02:00
|
|
|
, m_last_dht_announce(time_now() - minutes(15))
|
2007-05-31 02:21:54 +02:00
|
|
|
#endif
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_ses(ses)
|
|
|
|
, m_picker(0)
|
|
|
|
, m_last_working_tracker(-1)
|
|
|
|
, m_currently_trying_tracker(0)
|
|
|
|
, m_failed_trackers(0)
|
|
|
|
, m_time_scaler(0)
|
|
|
|
, m_num_pieces(0)
|
2008-01-31 18:52:29 +01:00
|
|
|
, m_sequential_download(false)
|
2004-06-14 01:30:42 +02:00
|
|
|
, m_got_tracker_response(false)
|
|
|
|
, m_ratio(0.f)
|
|
|
|
, m_total_failed_bytes(0)
|
2006-04-25 23:04:48 +02:00
|
|
|
, m_total_redundant_bytes(0)
|
2006-05-20 17:30:40 +02:00
|
|
|
, m_net_interface(net_interface.address(), 0)
|
2004-10-18 12:36:47 +02:00
|
|
|
, m_save_path(complete(save_path))
|
2007-10-08 22:01:36 +02:00
|
|
|
, m_storage_mode(storage_mode)
|
2008-03-08 07:06:31 +01:00
|
|
|
, m_state(torrent_status::queued_for_checking)
|
|
|
|
, m_progress(0.f)
|
|
|
|
, m_resume_data(resume_data)
|
2005-07-10 12:42:00 +02:00
|
|
|
, m_default_block_size(block_size)
|
2005-10-16 11:15:46 +02:00
|
|
|
, m_connections_initialized(false)
|
2007-08-22 07:31:42 +02:00
|
|
|
, m_settings(ses.settings())
|
2007-03-16 06:29:23 +01:00
|
|
|
, m_storage_constructor(sc)
|
2007-08-21 06:46:17 +02:00
|
|
|
, m_max_uploads((std::numeric_limits<int>::max)())
|
2007-08-17 09:37:08 +02:00
|
|
|
, m_num_uploads(0)
|
2007-08-21 06:46:17 +02:00
|
|
|
, m_max_connections((std::numeric_limits<int>::max)())
|
2008-03-29 20:39:24 +01:00
|
|
|
, m_deficit_counter(0)
|
2007-10-31 10:48:20 +01:00
|
|
|
, m_policy(this)
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2007-12-14 07:09:44 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
m_files_checked = false;
|
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
if (name) m_name.reset(new std::string(name));
|
|
|
|
|
2007-02-21 18:15:47 +01:00
|
|
|
if (tracker_url)
|
|
|
|
{
|
|
|
|
m_trackers.push_back(announce_entry(tracker_url));
|
2007-09-01 05:00:31 +02:00
|
|
|
m_torrent_file->add_tracker(tracker_url);
|
2007-02-21 18:15:47 +01:00
|
|
|
}
|
2007-04-11 19:44:15 +02:00
|
|
|
}
|
2007-04-11 19:22:19 +02:00
|
|
|
|
2007-04-11 19:44:15 +02:00
|
|
|
void torrent::start()
|
|
|
|
{
|
2007-04-11 19:22:19 +02:00
|
|
|
boost::weak_ptr<torrent> self(shared_from_this());
|
2007-09-01 05:00:31 +02:00
|
|
|
if (m_torrent_file->is_valid()) init();
|
2007-11-11 20:09:29 +01:00
|
|
|
if (m_abort) return;
|
2008-04-07 04:51:21 +02:00
|
|
|
asio::error_code ec;
|
|
|
|
m_announce_timer.expires_from_now(seconds(1), ec);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_announce_timer.async_wait(
|
|
|
|
bind(&torrent::on_announce_disp, self, _1));
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2007-02-12 10:20:49 +01:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
bool torrent::should_announce_dht() const
|
|
|
|
{
|
2007-11-25 10:33:26 +01:00
|
|
|
if (m_ses.m_listen_sockets.empty()) return false;
|
|
|
|
|
|
|
|
if (!m_ses.m_dht) return false;
|
|
|
|
|
2007-02-12 10:20:49 +01:00
|
|
|
// don't announce private torrents
|
2007-09-01 05:00:31 +02:00
|
|
|
if (m_torrent_file->is_valid() && m_torrent_file->priv()) return false;
|
2007-02-12 10:20:49 +01:00
|
|
|
|
|
|
|
if (m_trackers.empty()) return true;
|
|
|
|
|
|
|
|
return m_failed_trackers > 0 || !m_ses.settings().use_dht_as_fallback;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2003-12-14 06:56:12 +01:00
|
|
|
torrent::~torrent()
|
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
// The invariant can't be maintained here, since the torrent
|
|
|
|
// is being destructed, all weak references to it have been
|
|
|
|
// reset, which means that all its peers already have an
|
|
|
|
// invalidated torrent pointer (so it cannot be verified to be correct)
|
|
|
|
|
|
|
|
// i.e. the invariant can only be maintained if all connections have
|
|
|
|
// been closed by the time the torrent is destructed. And they are
|
|
|
|
// supposed to be closed. So we can still do the invariant check.
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_connections.empty());
|
2006-05-28 21:03:54 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-08-16 14:41:46 +02:00
|
|
|
for (peer_iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
(*(*i)->m_logger) << "*** DESTRUCTING TORRENT\n";
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_abort);
|
2005-11-08 01:56:26 +01:00
|
|
|
if (!m_connections.empty())
|
|
|
|
disconnect_all();
|
2003-12-14 06:56:12 +01:00
|
|
|
}
|
|
|
|
|
2007-10-22 22:58:42 +02:00
|
|
|
peer_request torrent::to_req(piece_block const& p)
|
|
|
|
{
|
|
|
|
int block_offset = p.block_index * m_block_size;
|
2008-04-05 23:18:27 +02:00
|
|
|
int block_size = (std::min)(torrent_file().piece_size(
|
2007-10-22 22:58:42 +02:00
|
|
|
p.piece_index) - block_offset, m_block_size);
|
|
|
|
TORRENT_ASSERT(block_size > 0);
|
|
|
|
TORRENT_ASSERT(block_size <= m_block_size);
|
|
|
|
|
|
|
|
peer_request r;
|
|
|
|
r.piece = p.piece_index;
|
|
|
|
r.start = block_offset;
|
|
|
|
r.length = block_size;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
std::string torrent::name() const
|
|
|
|
{
|
2007-09-01 05:00:31 +02:00
|
|
|
if (valid_metadata()) return m_torrent_file->name();
|
2006-11-14 01:08:16 +01:00
|
|
|
if (m_name) return *m_name;
|
|
|
|
return "";
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
2007-11-08 02:45:35 +01:00
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
void torrent::add_extension(boost::shared_ptr<torrent_plugin> ext)
|
|
|
|
{
|
|
|
|
m_extensions.push_back(ext);
|
|
|
|
}
|
2007-11-08 02:45:35 +01:00
|
|
|
|
|
|
|
void torrent::add_extension(boost::function<boost::shared_ptr<torrent_plugin>(torrent*, void*)> const& ext
|
|
|
|
, void* userdata)
|
|
|
|
{
|
|
|
|
boost::shared_ptr<torrent_plugin> tp(ext(this, userdata));
|
|
|
|
if (!tp) return;
|
|
|
|
|
|
|
|
add_extension(tp);
|
|
|
|
|
|
|
|
for (peer_iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
|
|
|
peer_connection* p = *i;
|
|
|
|
boost::shared_ptr<peer_plugin> pp(tp->new_connection(p));
|
|
|
|
if (pp) p->add_extension(pp);
|
|
|
|
}
|
|
|
|
|
|
|
|
// if files are checked for this torrent, call the extension
|
|
|
|
// to let it initialize itself
|
|
|
|
if (m_connections_initialized)
|
|
|
|
tp->on_files_checked();
|
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#endif
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
// this may not be called from a constructor because of the call to
|
|
|
|
// shared_from_this()
|
2004-06-14 01:30:42 +02:00
|
|
|
void torrent::init()
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrent_file->is_valid());
|
|
|
|
TORRENT_ASSERT(m_torrent_file->num_files() > 0);
|
|
|
|
TORRENT_ASSERT(m_torrent_file->total_size() >= 0);
|
2004-06-14 01:30:42 +02:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
m_have_pieces.resize(m_torrent_file->num_pieces(), false);
|
2007-06-10 22:46:09 +02:00
|
|
|
// the shared_from_this() will create an intentional
|
|
|
|
// cycle of ownership, se the hpp file for description.
|
|
|
|
m_owning_storage = new piece_manager(shared_from_this(), m_torrent_file
|
2008-03-08 07:06:31 +01:00
|
|
|
, m_save_path, m_ses.m_files, m_ses.m_disk_thread, m_storage_constructor
|
|
|
|
, m_storage_mode);
|
2007-06-10 22:46:09 +02:00
|
|
|
m_storage = m_owning_storage.get();
|
2007-09-01 05:00:31 +02:00
|
|
|
m_block_size = calculate_block_size(*m_torrent_file, m_default_block_size);
|
2007-03-17 18:28:59 +01:00
|
|
|
m_picker.reset(new piece_picker(
|
2008-04-05 23:18:27 +02:00
|
|
|
m_torrent_file->piece_length() / m_block_size
|
|
|
|
, int((m_torrent_file->total_size()+m_block_size-1)/m_block_size)));
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
std::vector<std::string> const& url_seeds = m_torrent_file->url_seeds();
|
2006-04-25 23:04:48 +02:00
|
|
|
std::copy(url_seeds.begin(), url_seeds.end(), std::inserter(m_web_seeds
|
|
|
|
, m_web_seeds.begin()));
|
2008-03-08 07:06:31 +01:00
|
|
|
|
|
|
|
m_state = torrent_status::queued_for_checking;
|
|
|
|
|
|
|
|
m_storage->async_check_fastresume(&m_resume_data
|
|
|
|
, bind(&torrent::on_resume_data_checked
|
|
|
|
, shared_from_this(), _1, _2));
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_resume_data_checked(int ret, disk_io_job const& j)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
if (ret == piece_manager::fatal_disk_error)
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::fatal))
|
|
|
|
{
|
2008-04-13 00:08:07 +02:00
|
|
|
m_ses.m_alerts.post_alert(file_error_alert(j.error_file, get_handle(), j.str));
|
2008-03-08 07:06:31 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_ses.m_logger) << time_now_string() << ": fatal disk error ["
|
|
|
|
" error: " << j.str <<
|
|
|
|
" torrent: " << torrent_file().name() <<
|
|
|
|
" ]\n";
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
std::fill(m_have_pieces.begin(), m_have_pieces.end(), false);
|
|
|
|
m_num_pieces = 0;
|
|
|
|
pause();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// parse out "peers" from the resume data and add them to the peer list
|
|
|
|
entry const* peers_entry = m_resume_data.find_key("peers");
|
|
|
|
if (peers_entry && peers_entry->type() == entry::list_t)
|
|
|
|
{
|
|
|
|
peer_id id;
|
|
|
|
std::fill(id.begin(), id.end(), 0);
|
|
|
|
entry::list_type const& peer_list = peers_entry->list();
|
|
|
|
|
|
|
|
for (entry::list_type::const_iterator i = peer_list.begin();
|
|
|
|
i != peer_list.end(); ++i)
|
|
|
|
{
|
|
|
|
if (i->type() != entry::dictionary_t) continue;
|
|
|
|
entry const* ip = i->find_key("ip");
|
|
|
|
entry const* port = i->find_key("port");
|
|
|
|
if (ip == 0 || port == 0
|
|
|
|
|| ip->type() != entry::string_t
|
|
|
|
|| port->type() != entry::int_t)
|
|
|
|
continue;
|
|
|
|
tcp::endpoint a(
|
|
|
|
address::from_string(ip->string())
|
|
|
|
, (unsigned short)port->integer());
|
|
|
|
m_policy.peer_from_tracker(a, id, peer_info::resume_data, 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// parse out "banned_peers" and add them as banned
|
|
|
|
entry const* banned_peers_entry = m_resume_data.find_key("banned_peers");
|
|
|
|
if (banned_peers_entry != 0 && banned_peers_entry->type() == entry::list_t)
|
|
|
|
{
|
|
|
|
peer_id id;
|
|
|
|
std::fill(id.begin(), id.end(), 0);
|
|
|
|
entry::list_type const& peer_list = banned_peers_entry->list();
|
|
|
|
|
|
|
|
for (entry::list_type::const_iterator i = peer_list.begin();
|
|
|
|
i != peer_list.end(); ++i)
|
|
|
|
{
|
|
|
|
if (i->type() != entry::dictionary_t) continue;
|
|
|
|
entry const* ip = i->find_key("ip");
|
|
|
|
entry const* port = i->find_key("port");
|
|
|
|
if (ip == 0 || port == 0
|
|
|
|
|| ip->type() != entry::string_t
|
|
|
|
|| port->type() != entry::int_t)
|
|
|
|
continue;
|
|
|
|
tcp::endpoint a(
|
|
|
|
address::from_string(ip->string())
|
|
|
|
, (unsigned short)port->integer());
|
|
|
|
policy::peer* p = m_policy.peer_from_tracker(a, id, peer_info::resume_data, 0);
|
|
|
|
if (p) p->banned = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool fastresume_rejected = !j.str.empty();
|
|
|
|
|
|
|
|
if (fastresume_rejected && m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(fastresume_rejected_alert(get_handle(), j.str));
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_ses.m_logger) << "fastresume data for "
|
|
|
|
<< torrent_file().name() << " rejected: "
|
|
|
|
<< j.str << "\n";
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ret == 0)
|
|
|
|
{
|
|
|
|
// there are either no files for this torrent
|
|
|
|
// or the resume_data was accepted
|
|
|
|
|
|
|
|
m_num_pieces = 0;
|
|
|
|
std::fill(m_have_pieces.begin(), m_have_pieces.end(), false);
|
|
|
|
if (!fastresume_rejected)
|
|
|
|
{
|
2008-04-13 00:08:07 +02:00
|
|
|
TORRENT_ASSERT(m_resume_data.type() == entry::dictionary_t);
|
2008-03-08 07:06:31 +01:00
|
|
|
// parse slots
|
|
|
|
entry const* slots_ent = m_resume_data.find_key("slots");
|
|
|
|
if (slots_ent != 0 && slots_ent->type() == entry::list_t)
|
|
|
|
{
|
|
|
|
entry::list_type const& slots = slots_ent->list();
|
|
|
|
|
|
|
|
for (entry::list_type::const_iterator i = slots.begin();
|
|
|
|
i != slots.end(); ++i)
|
|
|
|
{
|
|
|
|
if (i->type() != entry::int_t) continue;
|
|
|
|
int piece_index = int(i->integer());
|
|
|
|
if (piece_index < 0 || piece_index >= torrent_file().num_pieces())
|
|
|
|
continue;
|
|
|
|
m_have_pieces[piece_index] = true;
|
|
|
|
++m_num_pieces;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// parse unfinished pieces
|
|
|
|
int num_blocks_per_piece =
|
|
|
|
static_cast<int>(torrent_file().piece_length()) / block_size();
|
|
|
|
|
|
|
|
entry const* unfinished_ent = m_resume_data.find_key("unfinished");
|
|
|
|
if (unfinished_ent != 0 && unfinished_ent->type() == entry::list_t)
|
|
|
|
{
|
|
|
|
entry::list_type const& unfinished = unfinished_ent->list();
|
|
|
|
int index = 0;
|
|
|
|
for (entry::list_type::const_iterator i = unfinished.begin();
|
|
|
|
i != unfinished.end(); ++i, ++index)
|
|
|
|
{
|
|
|
|
if (i->type() != entry::dictionary_t) continue;
|
|
|
|
entry const* piece = i->find_key("piece");
|
|
|
|
if (piece == 0 || piece->type() != entry::int_t) continue;
|
|
|
|
int piece_index = int(piece->integer());
|
|
|
|
if (piece_index < 0 || piece_index >= torrent_file().num_pieces())
|
|
|
|
continue;
|
|
|
|
|
|
|
|
// if this assert is hit, the resume data file was corrupt
|
|
|
|
TORRENT_ASSERT(m_have_pieces[piece_index]);
|
|
|
|
m_have_pieces[piece_index] = false;
|
|
|
|
--m_num_pieces;
|
|
|
|
|
|
|
|
entry const* bitmask_ent = i->find_key("bitmask");
|
|
|
|
if (bitmask_ent == 0 || bitmask_ent->type() != entry::string_t) break;
|
|
|
|
std::string const& bitmask = bitmask_ent->string();
|
|
|
|
|
|
|
|
const int num_bitmask_bytes = (std::max)(num_blocks_per_piece / 8, 1);
|
|
|
|
if ((int)bitmask.size() != num_bitmask_bytes) continue;
|
|
|
|
for (int j = 0; j < num_bitmask_bytes; ++j)
|
|
|
|
{
|
|
|
|
unsigned char bits = bitmask[j];
|
|
|
|
int num_bits = (std::min)(num_blocks_per_piece - j*8, 8);
|
|
|
|
for (int k = 0; k < num_bits; ++k)
|
|
|
|
{
|
|
|
|
const int bit = j * 8 + k;
|
|
|
|
if (bits & (1 << k))
|
|
|
|
{
|
|
|
|
m_picker->mark_as_finished(piece_block(piece_index, bit), 0);
|
|
|
|
if (m_picker->is_piece_finished(piece_index))
|
|
|
|
async_verify_piece(piece_index, bind(&torrent::piece_finished
|
|
|
|
, shared_from_this(), piece_index, _1));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
files_checked();
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// either the fastresume data was rejected or there are
|
|
|
|
// some files
|
|
|
|
m_ses.check_torrent(shared_from_this());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::start_checking()
|
|
|
|
{
|
|
|
|
m_state = torrent_status::checking_files;
|
|
|
|
|
|
|
|
m_storage->async_check_files(bind(
|
|
|
|
&torrent::on_piece_checked
|
|
|
|
, shared_from_this(), _1, _2));
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_piece_checked(int ret, disk_io_job const& j)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
if (ret == piece_manager::fatal_disk_error)
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::fatal))
|
|
|
|
{
|
2008-04-13 00:08:07 +02:00
|
|
|
m_ses.m_alerts.post_alert(file_error_alert(j.error_file, get_handle(), j.str));
|
2008-03-08 07:06:31 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
|
|
(*m_ses.m_logger) << time_now_string() << ": fatal disk error ["
|
|
|
|
" error: " << j.str <<
|
|
|
|
" torrent: " << torrent_file().name() <<
|
|
|
|
" ]\n";
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
std::fill(m_have_pieces.begin(), m_have_pieces.end(), false);
|
|
|
|
m_num_pieces = 0;
|
|
|
|
pause();
|
|
|
|
m_ses.done_checking(shared_from_this());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
m_progress = j.piece / float(torrent_file().num_pieces());
|
|
|
|
|
|
|
|
if (j.offset >= 0 && !m_have_pieces[j.offset])
|
|
|
|
{
|
|
|
|
m_have_pieces[j.offset] = true;
|
|
|
|
++m_num_pieces;
|
|
|
|
}
|
|
|
|
|
|
|
|
// we're not done checking yet
|
|
|
|
// this handler will be called repeatedly until
|
|
|
|
// we're done, or encounter a failure
|
|
|
|
if (ret == piece_manager::need_full_check) return;
|
|
|
|
|
|
|
|
m_ses.done_checking(shared_from_this());
|
|
|
|
files_checked();
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
2004-02-26 01:27:06 +01:00
|
|
|
void torrent::use_interface(const char* net_interface)
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-05-20 17:30:40 +02:00
|
|
|
m_net_interface = tcp::endpoint(address::from_string(net_interface), 0);
|
2004-02-26 01:27:06 +01:00
|
|
|
}
|
|
|
|
|
2007-04-11 19:22:19 +02:00
|
|
|
void torrent::on_announce_disp(boost::weak_ptr<torrent> p
|
|
|
|
, asio::error_code const& e)
|
2007-04-04 04:06:07 +02:00
|
|
|
{
|
|
|
|
if (e) return;
|
2007-04-11 19:22:19 +02:00
|
|
|
boost::shared_ptr<torrent> t = p.lock();
|
|
|
|
if (!t) return;
|
|
|
|
t->on_announce();
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_announce()
|
|
|
|
{
|
2007-11-11 20:09:29 +01:00
|
|
|
if (m_abort) return;
|
|
|
|
|
2007-04-11 19:22:19 +02:00
|
|
|
boost::weak_ptr<torrent> self(shared_from_this());
|
2007-04-04 04:06:07 +02:00
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
asio::error_code ec;
|
2007-09-01 05:00:31 +02:00
|
|
|
if (!m_torrent_file->priv())
|
2007-07-23 02:38:31 +02:00
|
|
|
{
|
|
|
|
// announce on local network every 5 minutes
|
2008-04-07 04:51:21 +02:00
|
|
|
m_announce_timer.expires_from_now(minutes(5), ec);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_announce_timer.async_wait(
|
|
|
|
bind(&torrent::on_announce_disp, self, _1));
|
2007-04-04 04:06:07 +02:00
|
|
|
|
2007-07-23 02:38:31 +02:00
|
|
|
// announce with the local discovery service
|
2007-11-25 10:33:26 +01:00
|
|
|
if (!m_paused)
|
|
|
|
m_ses.announce_lsd(m_torrent_file->info_hash());
|
2007-07-23 02:38:31 +02:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
m_announce_timer.expires_from_now(minutes(15), ec);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_announce_timer.async_wait(
|
|
|
|
bind(&torrent::on_announce_disp, self, _1));
|
2007-07-23 02:38:31 +02:00
|
|
|
}
|
2007-04-04 04:06:07 +02:00
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2007-11-25 10:33:26 +01:00
|
|
|
if (m_paused) return;
|
2007-04-04 04:06:07 +02:00
|
|
|
if (!m_ses.m_dht) return;
|
2007-05-17 02:01:51 +02:00
|
|
|
ptime now = time_now();
|
|
|
|
if (should_announce_dht() && now - m_last_dht_announce > minutes(14))
|
2007-04-04 04:06:07 +02:00
|
|
|
{
|
2007-05-17 02:01:51 +02:00
|
|
|
m_last_dht_announce = now;
|
2007-09-01 05:00:31 +02:00
|
|
|
m_ses.m_dht->announce(m_torrent_file->info_hash()
|
2007-09-22 18:27:29 +02:00
|
|
|
, m_ses.m_listen_sockets.front().external_port
|
2008-01-08 06:47:43 +01:00
|
|
|
, bind(&torrent::on_dht_announce_response_disp, self, _1));
|
2007-04-04 04:06:07 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2006-08-01 17:27:08 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
|
2006-08-29 03:15:24 +02:00
|
|
|
void torrent::on_dht_announce_response_disp(boost::weak_ptr<libtorrent::torrent> t
|
|
|
|
, std::vector<tcp::endpoint> const& peers)
|
|
|
|
{
|
|
|
|
boost::shared_ptr<libtorrent::torrent> tor = t.lock();
|
|
|
|
if (!tor) return;
|
|
|
|
tor->on_dht_announce_response(peers);
|
|
|
|
}
|
|
|
|
|
2006-08-01 17:27:08 +02:00
|
|
|
void torrent::on_dht_announce_response(std::vector<tcp::endpoint> const& peers)
|
|
|
|
{
|
2007-02-01 08:33:04 +01:00
|
|
|
if (peers.empty()) return;
|
|
|
|
|
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(tracker_reply_alert(
|
|
|
|
get_handle(), peers.size(), "Got peers from DHT"));
|
|
|
|
}
|
2006-08-01 17:27:08 +02:00
|
|
|
std::for_each(peers.begin(), peers.end(), bind(
|
2007-04-10 23:23:13 +02:00
|
|
|
&policy::peer_from_tracker, boost::ref(m_policy), _1, peer_id(0)
|
|
|
|
, peer_info::dht, 0));
|
2006-08-01 17:27:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
2007-11-20 23:46:27 +01:00
|
|
|
void torrent::scrape_tracker()
|
|
|
|
{
|
|
|
|
if (m_trackers.empty()) return;
|
|
|
|
|
|
|
|
TORRENT_ASSERT(m_currently_trying_tracker >= 0);
|
|
|
|
TORRENT_ASSERT(m_currently_trying_tracker < int(m_trackers.size()));
|
|
|
|
|
|
|
|
tracker_request req;
|
|
|
|
req.info_hash = m_torrent_file->info_hash();
|
|
|
|
req.kind = tracker_request::scrape_request;
|
|
|
|
req.url = m_trackers[m_currently_trying_tracker].url;
|
2008-01-08 06:47:43 +01:00
|
|
|
m_ses.m_tracker_manager.queue_request(m_ses.m_io_service, m_ses.m_half_open, req
|
2007-11-20 23:46:27 +01:00
|
|
|
, tracker_login(), m_ses.m_listen_interface.address(), shared_from_this());
|
|
|
|
}
|
|
|
|
|
2004-07-24 13:54:17 +02:00
|
|
|
// returns true if it is time for this torrent to make another
|
|
|
|
// tracker request
|
|
|
|
bool torrent::should_request()
|
|
|
|
{
|
2008-01-27 23:39:50 +01:00
|
|
|
// INVARIANT_CHECK;
|
2006-08-01 17:27:08 +02:00
|
|
|
|
2007-11-16 20:35:09 +01:00
|
|
|
if (m_trackers.empty()) return false;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2004-07-24 13:54:17 +02:00
|
|
|
if (m_just_paused)
|
|
|
|
{
|
|
|
|
m_just_paused = false;
|
|
|
|
return true;
|
|
|
|
}
|
2007-04-05 00:27:36 +02:00
|
|
|
return !m_paused && m_next_request < time_now();
|
2004-07-24 13:54:17 +02:00
|
|
|
}
|
|
|
|
|
2005-08-11 01:32:39 +02:00
|
|
|
void torrent::tracker_warning(std::string const& msg)
|
|
|
|
{
|
2007-03-31 00:00:26 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2005-08-11 01:32:39 +02:00
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(tracker_warning_alert(get_handle(), msg));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-11-20 23:46:27 +01:00
|
|
|
void torrent::tracker_scrape_response(tracker_request const& req
|
|
|
|
, int complete, int incomplete, int downloaded)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
TORRENT_ASSERT(req.kind == tracker_request::scrape_request);
|
|
|
|
|
|
|
|
if (complete >= 0) m_complete = complete;
|
|
|
|
if (incomplete >= 0) m_incomplete = incomplete;
|
|
|
|
|
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
std::stringstream s;
|
|
|
|
s << "Got scrape response from tracker: " << req.url;
|
|
|
|
m_ses.m_alerts.post_alert(scrape_reply_alert(
|
|
|
|
get_handle(), m_incomplete, m_complete, s.str()));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
void torrent::tracker_response(
|
2007-11-21 00:19:30 +01:00
|
|
|
tracker_request const& r
|
2005-03-24 13:13:47 +01:00
|
|
|
, std::vector<peer_entry>& peer_list
|
2005-02-21 14:59:24 +01:00
|
|
|
, int interval
|
|
|
|
, int complete
|
2008-03-29 23:45:55 +01:00
|
|
|
, int incomplete
|
|
|
|
, address const& external_ip)
|
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);
|
|
|
|
|
2007-03-31 00:00:26 +02:00
|
|
|
INVARIANT_CHECK;
|
2007-11-20 23:46:27 +01:00
|
|
|
TORRENT_ASSERT(r.kind == tracker_request::announce_request);
|
2007-03-31 00:00:26 +02:00
|
|
|
|
2008-03-29 23:45:55 +01:00
|
|
|
if (external_ip != address())
|
|
|
|
m_ses.set_external_address(external_ip);
|
|
|
|
|
2004-04-02 00:29:51 +02:00
|
|
|
m_failed_trackers = 0;
|
2006-11-19 16:29:58 +01:00
|
|
|
// announce intervals less than 5 minutes
|
2004-01-25 13:37:15 +01:00
|
|
|
// are insane.
|
2004-09-16 19:18:10 +02:00
|
|
|
if (interval < 60 * 5) interval = 60 * 5;
|
2004-01-25 13:37:15 +01:00
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
m_last_working_tracker
|
2004-09-12 12:12:16 +02:00
|
|
|
= prioritize_tracker(m_currently_trying_tracker);
|
2004-01-21 14:16:11 +01:00
|
|
|
m_currently_trying_tracker = 0;
|
2003-10-23 18:55:52 +02:00
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
m_duration = interval;
|
2007-04-05 00:27:36 +02:00
|
|
|
m_next_request = time_now() + seconds(m_duration);
|
2003-10-23 18:55:52 +02:00
|
|
|
|
2005-02-21 14:59:24 +01:00
|
|
|
if (complete >= 0) m_complete = complete;
|
|
|
|
if (incomplete >= 0) m_incomplete = incomplete;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
// connect to random peers from the list
|
|
|
|
std::random_shuffle(peer_list.begin(), peer_list.end());
|
2003-10-23 18:55:52 +02:00
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING
|
2004-01-21 14:16:11 +01:00
|
|
|
std::stringstream s;
|
2004-01-22 23:45:52 +01:00
|
|
|
s << "TRACKER RESPONSE:\n"
|
|
|
|
"interval: " << m_duration << "\n"
|
|
|
|
"peers:\n";
|
2004-01-21 14:16:11 +01:00
|
|
|
for (std::vector<peer_entry>::const_iterator i = peer_list.begin();
|
2005-07-06 02:58:23 +02:00
|
|
|
i != peer_list.end(); ++i)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2004-01-21 14:16:11 +01:00
|
|
|
s << " " << std::setfill(' ') << std::setw(16) << i->ip
|
2004-01-22 23:45:52 +01:00
|
|
|
<< " " << std::setw(5) << std::dec << i->port << " ";
|
2006-04-25 23:04:48 +02:00
|
|
|
if (!i->pid.is_all_zeros()) s << " " << i->pid << " " << identify_client(i->pid);
|
2004-01-22 23:45:52 +01:00
|
|
|
s << "\n";
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
2008-03-29 23:45:55 +01:00
|
|
|
s << "external ip: " << external_ip << "\n";
|
2004-01-21 14:16:11 +01:00
|
|
|
debug_log(s.str());
|
|
|
|
#endif
|
|
|
|
// for each of the peers we got from the tracker
|
|
|
|
for (std::vector<peer_entry>::iterator i = peer_list.begin();
|
2005-07-06 02:58:23 +02:00
|
|
|
i != peer_list.end(); ++i)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2004-01-21 14:16:11 +01:00
|
|
|
// don't make connections to ourself
|
2006-04-25 23:04:48 +02:00
|
|
|
if (i->pid == m_ses.get_peer_id())
|
2004-01-21 14:16:11 +01:00
|
|
|
continue;
|
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
asio::error_code ec;
|
|
|
|
tcp::endpoint a(address::from_string(i->ip, ec), i->port);
|
|
|
|
|
|
|
|
if (ec)
|
2005-07-06 02:58:23 +02:00
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
// assume this is because we got a hostname instead of
|
|
|
|
// an ip address from the tracker
|
2007-03-07 19:50:38 +01:00
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
tcp::resolver::query q(i->ip, boost::lexical_cast<std::string>(i->port));
|
|
|
|
m_host_resolver.async_resolve(q,
|
|
|
|
bind(&torrent::on_peer_name_lookup, shared_from_this(), _1, _2, i->pid));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2007-03-07 19:50:38 +01:00
|
|
|
if (m_ses.m_ip_filter.access(a.address()) & ip_filter::blocked)
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-03-07 19:50:38 +01:00
|
|
|
debug_log("blocked ip from tracker: " + i->ip);
|
2005-07-06 02:58:23 +02:00
|
|
|
#endif
|
2007-04-17 07:56:43 +02:00
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
2008-04-07 04:51:21 +02:00
|
|
|
{
|
2007-04-17 07:56:43 +02:00
|
|
|
m_ses.m_alerts.post_alert(peer_blocked_alert(a.address()
|
|
|
|
, "peer from tracker blocked by IP filter"));
|
|
|
|
}
|
|
|
|
|
2007-03-07 19:50:38 +01:00
|
|
|
continue;
|
|
|
|
}
|
2008-04-07 04:51:21 +02:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
m_policy.peer_from_tracker(a, i->pid, peer_info::tracker, 0);
|
2007-03-07 19:50:38 +01:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2005-03-10 10:59:12 +01:00
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
std::stringstream s;
|
2007-11-20 23:46:27 +01:00
|
|
|
s << "Got response from tracker: " << r.url;
|
2005-03-10 10:59:12 +01:00
|
|
|
m_ses.m_alerts.post_alert(tracker_reply_alert(
|
2007-02-01 08:33:04 +01:00
|
|
|
get_handle(), peer_list.size(), s.str()));
|
2005-03-10 10:59:12 +01:00
|
|
|
}
|
2003-12-22 08:14:35 +01:00
|
|
|
m_got_tracker_response = true;
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2007-03-07 19:50:38 +01:00
|
|
|
void torrent::on_peer_name_lookup(asio::error_code const& e, tcp::resolver::iterator host
|
2008-04-07 04:51:21 +02:00
|
|
|
, peer_id pid)
|
2007-03-07 19:50:38 +01:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
if (e || host == tcp::resolver::iterator() ||
|
|
|
|
m_ses.is_aborted()) return;
|
|
|
|
|
|
|
|
if (m_ses.m_ip_filter.access(host->endpoint().address()) & ip_filter::blocked)
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-03-07 19:50:38 +01:00
|
|
|
debug_log("blocked ip from tracker: " + host->endpoint().address().to_string());
|
|
|
|
#endif
|
2007-04-17 07:56:43 +02:00
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(peer_blocked_alert(host->endpoint().address()
|
|
|
|
, "peer from tracker blocked by IP filter"));
|
|
|
|
}
|
|
|
|
|
2007-03-07 19:50:38 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
m_policy.peer_from_tracker(*host, pid, peer_info::tracker, 0);
|
2007-03-07 19:50:38 +01:00
|
|
|
}
|
|
|
|
|
2004-01-18 02:58:33 +01:00
|
|
|
size_type torrent::bytes_left() const
|
2003-12-07 06:53:04 +01:00
|
|
|
{
|
2004-06-14 01:30:42 +02:00
|
|
|
// if we don't have the metadata yet, we
|
|
|
|
// cannot tell how big the torrent is.
|
|
|
|
if (!valid_metadata()) return -1;
|
2007-09-01 05:00:31 +02:00
|
|
|
return m_torrent_file->total_size()
|
2006-09-24 02:48:31 +02:00
|
|
|
- quantized_bytes_done();
|
|
|
|
}
|
|
|
|
|
|
|
|
size_type torrent::quantized_bytes_done() const
|
|
|
|
{
|
2006-12-11 13:48:33 +01:00
|
|
|
// INVARIANT_CHECK;
|
2006-09-24 02:48:31 +02:00
|
|
|
|
|
|
|
if (!valid_metadata()) return 0;
|
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (m_torrent_file->num_pieces() == 0)
|
2006-09-24 02:48:31 +02:00
|
|
|
return 0;
|
2006-12-04 13:20:34 +01:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (is_seed()) return m_torrent_file->total_size();
|
2006-12-04 13:20:34 +01:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
const int last_piece = m_torrent_file->num_pieces() - 1;
|
2006-09-24 02:48:31 +02:00
|
|
|
|
|
|
|
size_type total_done
|
2007-11-30 07:14:45 +01:00
|
|
|
= size_type(m_num_pieces) * m_torrent_file->piece_length();
|
2006-09-24 02:48:31 +02:00
|
|
|
|
|
|
|
// if we have the last piece, we have to correct
|
|
|
|
// the amount we have, since the first calculation
|
|
|
|
// assumed all pieces were of equal size
|
|
|
|
if (m_have_pieces[last_piece])
|
|
|
|
{
|
2007-09-01 05:00:31 +02:00
|
|
|
int corr = m_torrent_file->piece_size(last_piece)
|
|
|
|
- m_torrent_file->piece_length();
|
2006-09-24 02:48:31 +02:00
|
|
|
total_done += corr;
|
|
|
|
}
|
|
|
|
return total_done;
|
2004-01-15 17:45:34 +01:00
|
|
|
}
|
|
|
|
|
2005-05-30 19:43:03 +02:00
|
|
|
// the first value is the total number of bytes downloaded
|
|
|
|
// the second value is the number of bytes of those that haven't
|
|
|
|
// been filtered as not wanted we have downloaded
|
2006-01-07 14:48:14 +01:00
|
|
|
tuple<size_type, size_type> torrent::bytes_done() const
|
2004-01-15 17:45:34 +01:00
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (!valid_metadata() || m_torrent_file->num_pieces() == 0)
|
2006-01-07 14:48:14 +01:00
|
|
|
return tuple<size_type, size_type>(0,0);
|
2007-03-17 00:28:26 +01:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
const int last_piece = m_torrent_file->num_pieces() - 1;
|
2007-11-27 00:08:59 +01:00
|
|
|
const int piece_size = m_torrent_file->piece_length();
|
2004-01-15 17:45:34 +01:00
|
|
|
|
2006-12-04 13:20:34 +01:00
|
|
|
if (is_seed())
|
2007-09-01 05:00:31 +02:00
|
|
|
return make_tuple(m_torrent_file->total_size()
|
|
|
|
, m_torrent_file->total_size());
|
2006-12-04 13:20:34 +01:00
|
|
|
|
2007-12-09 20:38:17 +01:00
|
|
|
TORRENT_ASSERT(m_num_pieces >= m_picker->num_have_filtered());
|
2007-11-30 07:14:45 +01:00
|
|
|
size_type wanted_done = size_type(m_num_pieces - m_picker->num_have_filtered())
|
2007-11-27 00:08:59 +01:00
|
|
|
* piece_size;
|
2007-12-09 20:38:17 +01:00
|
|
|
TORRENT_ASSERT(wanted_done >= 0);
|
2005-05-30 19:43:03 +02:00
|
|
|
|
2004-01-18 02:58:33 +01:00
|
|
|
size_type total_done
|
2007-11-30 07:14:45 +01:00
|
|
|
= size_type(m_num_pieces) * piece_size;
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_num_pieces < m_torrent_file->num_pieces());
|
2004-01-15 17:45:34 +01:00
|
|
|
|
|
|
|
// if we have the last piece, we have to correct
|
|
|
|
// the amount we have, since the first calculation
|
|
|
|
// assumed all pieces were of equal size
|
2003-12-07 06:53:04 +01:00
|
|
|
if (m_have_pieces[last_piece])
|
|
|
|
{
|
2007-11-27 00:08:59 +01:00
|
|
|
TORRENT_ASSERT(total_done >= piece_size);
|
2007-09-01 05:00:31 +02:00
|
|
|
int corr = m_torrent_file->piece_size(last_piece)
|
2007-11-27 00:08:59 +01:00
|
|
|
- piece_size;
|
|
|
|
TORRENT_ASSERT(corr <= 0);
|
|
|
|
TORRENT_ASSERT(corr > -piece_size);
|
2005-05-30 19:43:03 +02:00
|
|
|
total_done += corr;
|
2007-03-15 23:03:56 +01:00
|
|
|
if (m_picker->piece_priority(last_piece) != 0)
|
2007-11-27 00:08:59 +01:00
|
|
|
{
|
|
|
|
TORRENT_ASSERT(wanted_done >= piece_size);
|
2005-05-30 19:43:03 +02:00
|
|
|
wanted_done += corr;
|
2007-11-27 00:08:59 +01:00
|
|
|
}
|
2003-12-07 06:53:04 +01:00
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done <= m_torrent_file->total_size());
|
|
|
|
TORRENT_ASSERT(wanted_done <= m_torrent_file->total_size());
|
2007-12-09 20:38:17 +01:00
|
|
|
TORRENT_ASSERT(total_done >= wanted_done);
|
2006-12-11 13:48:33 +01:00
|
|
|
|
2004-01-15 17:45:34 +01:00
|
|
|
const std::vector<piece_picker::downloading_piece>& dl_queue
|
2004-06-14 01:30:42 +02:00
|
|
|
= m_picker->get_download_queue();
|
2004-01-15 17:45:34 +01:00
|
|
|
|
2008-04-05 23:18:27 +02:00
|
|
|
const int blocks_per_piece = piece_size / m_block_size;
|
2004-01-15 17:45:34 +01:00
|
|
|
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator i =
|
2005-05-30 19:43:03 +02:00
|
|
|
dl_queue.begin(); i != dl_queue.end(); ++i)
|
2004-01-15 17:45:34 +01:00
|
|
|
{
|
2005-05-30 19:43:03 +02:00
|
|
|
int corr = 0;
|
2006-12-31 15:48:18 +01:00
|
|
|
int index = i->index;
|
2007-10-05 19:31:02 +02:00
|
|
|
if (m_have_pieces[index]) continue;
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i->finished <= m_picker->blocks_in_piece(index));
|
2006-12-14 17:12:31 +01:00
|
|
|
|
|
|
|
#ifndef NDEBUG
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator j = boost::next(i);
|
|
|
|
j != dl_queue.end(); ++j)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(j->index != index);
|
2006-12-14 17:12:31 +01:00
|
|
|
}
|
|
|
|
#endif
|
2004-01-15 17:45:34 +01:00
|
|
|
|
|
|
|
for (int j = 0; j < blocks_per_piece; ++j)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker->is_finished(piece_block(index, j)) == (i->info[j].state == piece_picker::block_info::state_finished));
|
2007-06-10 22:46:09 +02:00
|
|
|
corr += (i->info[j].state == piece_picker::block_info::state_finished) * m_block_size;
|
2007-11-27 00:08:59 +01:00
|
|
|
TORRENT_ASSERT(corr >= 0);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(index != last_piece || j < m_picker->blocks_in_last_piece()
|
2007-06-10 22:46:09 +02:00
|
|
|
|| i->info[j].state != piece_picker::block_info::state_finished);
|
2004-01-15 17:45:34 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// correction if this was the last piece
|
|
|
|
// and if we have the last block
|
|
|
|
if (i->index == last_piece
|
2007-06-10 22:46:09 +02:00
|
|
|
&& i->info[m_picker->blocks_in_last_piece()-1].state
|
|
|
|
== piece_picker::block_info::state_finished)
|
2004-01-15 17:45:34 +01:00
|
|
|
{
|
2005-05-30 19:43:03 +02:00
|
|
|
corr -= m_block_size;
|
2007-09-01 05:00:31 +02:00
|
|
|
corr += m_torrent_file->piece_size(last_piece) % m_block_size;
|
2004-01-15 17:45:34 +01:00
|
|
|
}
|
2005-05-30 19:43:03 +02:00
|
|
|
total_done += corr;
|
2007-03-15 23:03:56 +01:00
|
|
|
if (m_picker->piece_priority(index) != 0)
|
2005-05-30 19:43:03 +02:00
|
|
|
wanted_done += corr;
|
2004-01-15 17:45:34 +01:00
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done <= m_torrent_file->total_size());
|
|
|
|
TORRENT_ASSERT(wanted_done <= m_torrent_file->total_size());
|
2006-12-15 13:29:47 +01:00
|
|
|
|
2004-09-16 03:14:16 +02:00
|
|
|
std::map<piece_block, int> downloading_piece;
|
|
|
|
for (const_peer_iterator i = begin(); i != end(); ++i)
|
2004-01-15 17:45:34 +01:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* pc = *i;
|
2004-01-15 17:45:34 +01:00
|
|
|
boost::optional<piece_block_progress> p
|
2006-04-25 23:04:48 +02:00
|
|
|
= pc->downloading_piece_progress();
|
2004-01-15 17:45:34 +01:00
|
|
|
if (p)
|
|
|
|
{
|
|
|
|
if (m_have_pieces[p->piece_index])
|
|
|
|
continue;
|
2004-09-16 03:14:16 +02:00
|
|
|
|
|
|
|
piece_block block(p->piece_index, p->block_index);
|
|
|
|
if (m_picker->is_finished(block))
|
2004-01-15 17:45:34 +01:00
|
|
|
continue;
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2004-09-16 03:14:16 +02:00
|
|
|
std::map<piece_block, int>::iterator dp
|
|
|
|
= downloading_piece.find(block);
|
|
|
|
if (dp != downloading_piece.end())
|
|
|
|
{
|
|
|
|
if (dp->second < p->bytes_downloaded)
|
|
|
|
dp->second = p->bytes_downloaded;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
downloading_piece[block] = p->bytes_downloaded;
|
|
|
|
}
|
2006-12-18 02:23:30 +01:00
|
|
|
#ifndef NDEBUG
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->bytes_downloaded <= p->full_block_bytes);
|
2007-09-01 05:00:31 +02:00
|
|
|
int last_piece = m_torrent_file->num_pieces() - 1;
|
2006-12-18 02:23:30 +01:00
|
|
|
if (p->piece_index == last_piece
|
2007-09-01 05:00:31 +02:00
|
|
|
&& p->block_index == m_torrent_file->piece_size(last_piece) / block_size())
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->full_block_bytes == m_torrent_file->piece_size(last_piece) % block_size());
|
2006-12-18 02:23:30 +01:00
|
|
|
else
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->full_block_bytes == block_size());
|
2006-12-18 02:23:30 +01:00
|
|
|
#endif
|
2004-01-15 17:45:34 +01:00
|
|
|
}
|
|
|
|
}
|
2004-09-16 03:14:16 +02:00
|
|
|
for (std::map<piece_block, int>::iterator i = downloading_piece.begin();
|
|
|
|
i != downloading_piece.end(); ++i)
|
2005-05-30 19:43:03 +02:00
|
|
|
{
|
2004-09-16 03:14:16 +02:00
|
|
|
total_done += i->second;
|
2007-03-15 23:03:56 +01:00
|
|
|
if (m_picker->piece_priority(i->first.piece_index) != 0)
|
2005-05-30 19:43:03 +02:00
|
|
|
wanted_done += i->second;
|
|
|
|
}
|
2006-12-15 13:29:47 +01:00
|
|
|
|
2006-12-18 02:23:30 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (total_done >= m_torrent_file->total_size())
|
2006-12-18 02:23:30 +01:00
|
|
|
{
|
2007-10-01 04:09:12 +02:00
|
|
|
// Thist happens when a piece has been downloaded completely
|
|
|
|
// but not yet verified against the hash
|
2006-12-18 02:23:30 +01:00
|
|
|
std::copy(m_have_pieces.begin(), m_have_pieces.end()
|
|
|
|
, std::ostream_iterator<bool>(std::cerr, " "));
|
|
|
|
std::cerr << std::endl;
|
|
|
|
std::cerr << "num_pieces: " << m_num_pieces << std::endl;
|
|
|
|
|
|
|
|
std::cerr << "unfinished:" << std::endl;
|
|
|
|
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator i =
|
|
|
|
dl_queue.begin(); i != dl_queue.end(); ++i)
|
|
|
|
{
|
|
|
|
std::cerr << " " << i->index << " ";
|
|
|
|
for (int j = 0; j < blocks_per_piece; ++j)
|
|
|
|
{
|
2007-06-10 22:46:09 +02:00
|
|
|
std::cerr << (i->info[j].state == piece_picker::block_info::state_finished ? "1" : "0");
|
2006-12-18 02:23:30 +01:00
|
|
|
}
|
|
|
|
std::cerr << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::cerr << "downloading pieces:" << std::endl;
|
|
|
|
|
|
|
|
for (std::map<piece_block, int>::iterator i = downloading_piece.begin();
|
|
|
|
i != downloading_piece.end(); ++i)
|
|
|
|
{
|
|
|
|
std::cerr << " " << i->first.piece_index << ":" << i->first.block_index
|
|
|
|
<< " " << i->second << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done <= m_torrent_file->total_size());
|
|
|
|
TORRENT_ASSERT(wanted_done <= m_torrent_file->total_size());
|
2006-12-18 02:23:30 +01:00
|
|
|
|
|
|
|
#endif
|
2006-12-15 13:29:47 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done >= wanted_done);
|
2006-01-07 14:48:14 +01:00
|
|
|
return make_tuple(total_done, wanted_done);
|
2004-01-15 17:45:34 +01:00
|
|
|
}
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2008-04-13 00:08:07 +02:00
|
|
|
// passed_hash_check
|
|
|
|
// 0: success, piece passed check
|
|
|
|
// -1: disk failure
|
|
|
|
// -2: piece failed check
|
|
|
|
void torrent::piece_finished(int index, int passed_hash_check)
|
2007-06-10 22:46:09 +02:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2007-10-18 02:32:16 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_ses.m_logger) << time_now_string() << " *** PIECE_FINISHED [ p: "
|
|
|
|
<< index << " chk: " << (passed_hash_check?"passed":"failed") << " ]\n";
|
|
|
|
#endif
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
bool was_seed = is_seed();
|
|
|
|
bool was_finished = m_picker->num_filtered() + num_pieces()
|
|
|
|
== torrent_file().num_pieces();
|
|
|
|
|
2008-04-13 00:08:07 +02:00
|
|
|
if (passed_hash_check == 0)
|
2007-06-10 22:46:09 +02:00
|
|
|
{
|
2007-10-04 01:15:13 +02:00
|
|
|
if (m_ses.m_alerts.should_post(alert::debug))
|
2007-08-01 08:14:16 +02:00
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(piece_finished_alert(get_handle()
|
|
|
|
, index, "piece finished"));
|
|
|
|
}
|
2007-06-10 22:46:09 +02:00
|
|
|
// the following call may cause picker to become invalid
|
|
|
|
// in case we just became a seed
|
|
|
|
announce_piece(index);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2007-06-10 22:46:09 +02:00
|
|
|
// if we just became a seed, picker is now invalid, since it
|
|
|
|
// is deallocated by the torrent once it starts seeding
|
|
|
|
if (!was_finished
|
|
|
|
&& (is_seed()
|
|
|
|
|| m_picker->num_filtered() + num_pieces()
|
|
|
|
== torrent_file().num_pieces()))
|
|
|
|
{
|
|
|
|
// torrent finished
|
|
|
|
// i.e. all the pieces we're interested in have
|
|
|
|
// been downloaded. Release the files (they will open
|
|
|
|
// in read only mode if needed)
|
2008-04-07 04:51:21 +02:00
|
|
|
finished();
|
2007-06-10 22:46:09 +02:00
|
|
|
}
|
|
|
|
}
|
2008-04-13 00:08:07 +02:00
|
|
|
else if (passed_hash_check == -2)
|
2007-06-10 22:46:09 +02:00
|
|
|
{
|
|
|
|
piece_failed(index);
|
|
|
|
}
|
2008-04-13 00:08:07 +02:00
|
|
|
else
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(passed_hash_check == -1);
|
|
|
|
m_picker->restore_piece(index);
|
|
|
|
}
|
2007-06-10 22:46:09 +02:00
|
|
|
|
2008-04-13 00:08:07 +02:00
|
|
|
m_policy.piece_finished(index, passed_hash_check == 0);
|
2007-06-10 22:46:09 +02:00
|
|
|
|
|
|
|
if (!was_seed && is_seed())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(passed_hash_check);
|
2007-06-10 22:46:09 +02:00
|
|
|
completed();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-12-01 22:27:27 +01:00
|
|
|
void torrent::piece_failed(int index)
|
|
|
|
{
|
2006-12-14 17:12:31 +01:00
|
|
|
// if the last piece fails the peer connection will still
|
|
|
|
// think that it has received all of it until this function
|
|
|
|
// resets the download queue. So, we cannot do the
|
|
|
|
// invariant check here since it assumes:
|
2007-09-01 05:00:31 +02:00
|
|
|
// (total_done == m_torrent_file->total_size()) => is_seed()
|
2007-10-31 10:48:20 +01:00
|
|
|
INVARIANT_CHECK;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_storage);
|
|
|
|
TORRENT_ASSERT(m_storage->refcount() > 0);
|
|
|
|
TORRENT_ASSERT(m_picker.get());
|
|
|
|
TORRENT_ASSERT(index >= 0);
|
|
|
|
TORRENT_ASSERT(index < m_torrent_file->num_pieces());
|
2004-01-25 13:37:15 +01:00
|
|
|
|
2003-12-22 08:14:35 +01:00
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
std::stringstream s;
|
|
|
|
s << "hash for piece " << index << " failed";
|
2004-01-07 01:48:02 +01:00
|
|
|
m_ses.m_alerts.post_alert(hash_failed_alert(get_handle(), index, s.str()));
|
2003-12-22 08:14:35 +01:00
|
|
|
}
|
2004-04-18 14:28:02 +02:00
|
|
|
// increase the total amount of failed bytes
|
2007-09-01 05:00:31 +02:00
|
|
|
m_total_failed_bytes += m_torrent_file->piece_size(index);
|
2004-04-18 14:28:02 +02:00
|
|
|
|
2007-07-04 04:16:49 +02:00
|
|
|
std::vector<void*> downloaders;
|
2004-06-14 01:30:42 +02:00
|
|
|
m_picker->get_downloaders(downloaders, index);
|
2003-12-01 22:27:27 +01:00
|
|
|
|
|
|
|
// decrease the trust point of all peers that sent
|
|
|
|
// parts of this piece.
|
2004-09-12 12:12:16 +02:00
|
|
|
// first, build a set of all peers that participated
|
2007-07-04 04:16:49 +02:00
|
|
|
std::set<void*> peers;
|
2004-09-12 12:12:16 +02:00
|
|
|
std::copy(downloaders.begin(), downloaders.end(), std::inserter(peers, peers.begin()));
|
|
|
|
|
2007-12-18 07:04:54 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
for (std::vector<void*>::iterator i = downloaders.begin()
|
|
|
|
, end(downloaders.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
policy::peer* p = (policy::peer*)*i;
|
|
|
|
if (p && p->connection)
|
|
|
|
{
|
|
|
|
p->connection->piece_failed = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
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-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
|
|
|
(*i)->on_piece_failed(index);
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2007-07-04 04:16:49 +02:00
|
|
|
for (std::set<void*>::iterator i = peers.begin()
|
2004-09-12 12:12:16 +02:00
|
|
|
, end(peers.end()); i != end; ++i)
|
2003-12-01 22:27:27 +01:00
|
|
|
{
|
2007-07-04 04:16:49 +02:00
|
|
|
policy::peer* p = static_cast<policy::peer*>(*i);
|
|
|
|
if (p == 0) continue;
|
|
|
|
if (p->connection) p->connection->received_invalid_data(index);
|
2003-12-14 06:56:12 +01:00
|
|
|
|
2004-09-12 12:12:16 +02:00
|
|
|
// either, we have received too many failed hashes
|
|
|
|
// or this was the only peer that sent us this piece.
|
2006-04-25 23:04:48 +02:00
|
|
|
// TODO: make this a changable setting
|
2007-07-04 04:16:49 +02:00
|
|
|
if (p->trust_points <= -7
|
2007-05-25 21:42:10 +02:00
|
|
|
|| peers.size() == 1)
|
2003-12-01 22:27:27 +01:00
|
|
|
{
|
2003-12-14 06:56:12 +01:00
|
|
|
// we don't trust this peer anymore
|
|
|
|
// ban it.
|
2004-03-21 03:03:37 +01:00
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(peer_ban_alert(
|
2007-07-04 04:16:49 +02:00
|
|
|
p->ip
|
2004-03-21 03:03:37 +01:00
|
|
|
, get_handle()
|
|
|
|
, "banning peer because of too many corrupt pieces"));
|
|
|
|
}
|
2007-04-13 03:53:25 +02:00
|
|
|
|
|
|
|
// mark the peer as banned
|
2007-07-04 04:16:49 +02:00
|
|
|
p->banned = true;
|
2005-09-01 23:04:21 +02:00
|
|
|
|
2007-07-04 04:16:49 +02:00
|
|
|
if (p->connection)
|
|
|
|
{
|
2007-12-19 22:36:54 +01:00
|
|
|
#ifdef TORRENT_LOGGING
|
|
|
|
(*m_ses.m_logger) << time_now_string() << " *** BANNING PEER [ " << p->ip
|
|
|
|
<< " ] 'too many corrupt pieces'\n";
|
2008-02-07 08:09:52 +01:00
|
|
|
#endif
|
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-07-04 04:16:49 +02:00
|
|
|
(*p->connection->m_logger) << "*** BANNING PEER [ " << p->ip
|
|
|
|
<< " ] 'too many corrupt pieces'\n";
|
2005-09-01 23:04:21 +02:00
|
|
|
#endif
|
2008-01-07 02:10:46 +01:00
|
|
|
p->connection->disconnect("too many corrupt pieces, banning peer");
|
2007-07-04 04:16:49 +02:00
|
|
|
}
|
2003-12-01 22:27:27 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// we have to let the piece_picker know that
|
|
|
|
// this piece failed the check as it can restore it
|
|
|
|
// and mark it as being interesting for download
|
|
|
|
// TODO: do this more intelligently! and keep track
|
|
|
|
// of how much crap (data that failed hash-check) and
|
|
|
|
// how much redundant data we have downloaded
|
|
|
|
// if some clients has sent more than one piece
|
|
|
|
// start with redownloading the pieces that the client
|
|
|
|
// that has sent the least number of pieces
|
2004-06-14 01:30:42 +02:00
|
|
|
m_picker->restore_piece(index);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_storage);
|
2004-01-09 11:50:22 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_have_pieces[index] == false);
|
2007-12-18 07:04:54 +01:00
|
|
|
|
|
|
|
#ifndef NDEBUG
|
|
|
|
for (std::vector<void*>::iterator i = downloaders.begin()
|
|
|
|
, end(downloaders.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
policy::peer* p = (policy::peer*)*i;
|
|
|
|
if (p && p->connection)
|
|
|
|
{
|
|
|
|
p->connection->piece_failed = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2003-12-01 22:27:27 +01:00
|
|
|
}
|
|
|
|
|
2005-03-05 15:17:17 +01:00
|
|
|
void torrent::abort()
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2005-03-05 15:17:17 +01:00
|
|
|
m_abort = true;
|
2006-04-25 23:04:48 +02:00
|
|
|
// if the torrent is paused, it doesn't need
|
|
|
|
// to announce with even=stopped again.
|
|
|
|
if (!m_paused)
|
|
|
|
m_event = tracker_request::stopped;
|
2005-03-05 15:17:17 +01:00
|
|
|
// disconnect all peers and close all
|
2005-08-15 00:04:58 +02:00
|
|
|
// files belonging to the torrents
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-08-16 14:41:46 +02:00
|
|
|
for (peer_iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
(*(*i)->m_logger) << "*** ABORTING TORRENT\n";
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2005-03-05 15:17:17 +01:00
|
|
|
disconnect_all();
|
2007-10-02 20:34:34 +02:00
|
|
|
if (m_owning_storage.get())
|
|
|
|
m_storage->async_release_files(
|
|
|
|
bind(&torrent::on_files_released, shared_from_this(), _1, _2));
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
m_owning_storage = 0;
|
2008-04-07 04:51:21 +02:00
|
|
|
asio::error_code ec;
|
|
|
|
m_announce_timer.cancel(ec);
|
2007-11-11 20:09:29 +01:00
|
|
|
m_host_resolver.cancel();
|
2005-03-05 15:17:17 +01:00
|
|
|
}
|
2003-12-01 22:27:27 +01:00
|
|
|
|
2007-10-13 05:33:33 +02:00
|
|
|
void torrent::on_files_deleted(int ret, disk_io_job const& j)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
if (alerts().should_post(alert::warning))
|
|
|
|
{
|
2007-11-16 20:04:43 +01:00
|
|
|
if (ret != 0)
|
|
|
|
{
|
|
|
|
alerts().post_alert(torrent_deleted_alert(get_handle(), "delete files failed: " + j.str));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
alerts().post_alert(torrent_deleted_alert(get_handle(), "files deleted"));
|
|
|
|
}
|
2007-10-13 05:33:33 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-06-11 23:24:14 +02:00
|
|
|
void torrent::on_files_released(int ret, disk_io_job const& j)
|
2007-10-02 20:34:34 +02:00
|
|
|
{
|
|
|
|
/*
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
if (alerts().should_post(alert::warning))
|
|
|
|
{
|
|
|
|
alerts().post_alert(torrent_paused_alert(get_handle(), "torrent paused"));
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_torrent_paused(int ret, disk_io_job const& j)
|
2007-06-11 23:24:14 +02:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
if (alerts().should_post(alert::warning))
|
|
|
|
{
|
|
|
|
alerts().post_alert(torrent_paused_alert(get_handle(), "torrent paused"));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-10-23 01:00:57 +02:00
|
|
|
void torrent::announce_piece(int index)
|
|
|
|
{
|
2006-12-31 15:48:18 +01:00
|
|
|
// INVARIANT_CHECK;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(index >= 0);
|
|
|
|
TORRENT_ASSERT(index < m_torrent_file->num_pieces());
|
2004-01-25 13:37:15 +01:00
|
|
|
|
2007-07-04 04:16:49 +02:00
|
|
|
std::vector<void*> downloaders;
|
2004-06-14 01:30:42 +02:00
|
|
|
m_picker->get_downloaders(downloaders, index);
|
2003-12-01 22:27:27 +01:00
|
|
|
|
|
|
|
// increase the trust point of all peers that sent
|
|
|
|
// parts of this piece.
|
2007-07-04 04:16:49 +02:00
|
|
|
std::set<void*> peers;
|
2004-09-12 12:12:16 +02:00
|
|
|
std::copy(downloaders.begin(), downloaders.end(), std::inserter(peers, peers.begin()));
|
|
|
|
|
2006-12-21 03:44:00 +01:00
|
|
|
if (!m_have_pieces[index])
|
|
|
|
m_num_pieces++;
|
|
|
|
m_have_pieces[index] = true;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(std::accumulate(m_have_pieces.begin(), m_have_pieces.end(), 0)
|
2006-12-21 03:44:00 +01:00
|
|
|
== m_num_pieces);
|
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
m_picker->we_have(index);
|
2004-01-13 04:08:59 +01:00
|
|
|
for (peer_iterator i = m_connections.begin(); i != m_connections.end(); ++i)
|
2008-04-07 04:51:21 +02:00
|
|
|
(*i)->announce_piece(index);
|
2006-11-14 01:08:16 +01:00
|
|
|
|
2007-07-04 04:16:49 +02:00
|
|
|
for (std::set<void*>::iterator i = peers.begin()
|
2006-12-31 15:48:18 +01:00
|
|
|
, end(peers.end()); i != end; ++i)
|
|
|
|
{
|
2007-07-04 04:16:49 +02:00
|
|
|
policy::peer* p = static_cast<policy::peer*>(*i);
|
|
|
|
if (p == 0) continue;
|
|
|
|
p->on_parole = false;
|
|
|
|
++p->trust_points;
|
|
|
|
// TODO: make this limit user settable
|
|
|
|
if (p->trust_points > 20) p->trust_points = 20;
|
|
|
|
if (p->connection) p->connection->received_valid_data(index);
|
2006-12-31 15:48:18 +01:00
|
|
|
}
|
|
|
|
|
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-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
|
|
|
(*i)->on_piece_pass(index);
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
2007-05-10 00:54:26 +02:00
|
|
|
if (is_seed())
|
|
|
|
{
|
|
|
|
m_picker.reset();
|
2007-11-19 05:58:52 +01:00
|
|
|
if (m_ses.settings().free_torrent_hashes)
|
|
|
|
m_torrent_file->seed_free();
|
2007-05-10 00:54:26 +02:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2004-03-28 19:45:37 +02:00
|
|
|
std::string torrent::tracker_login() const
|
2004-02-22 23:40:45 +01:00
|
|
|
{
|
|
|
|
if (m_username.empty() && m_password.empty()) return "";
|
|
|
|
return m_username + ":" + m_password;
|
|
|
|
}
|
|
|
|
|
2007-05-30 08:52:59 +02:00
|
|
|
void torrent::piece_availability(std::vector<int>& avail) const
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
2007-03-20 02:59:00 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2007-05-30 08:52:59 +02:00
|
|
|
if (is_seed())
|
|
|
|
{
|
|
|
|
avail.clear();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
m_picker->get_availability(avail);
|
|
|
|
}
|
2007-03-20 02:59:00 +01:00
|
|
|
|
|
|
|
void torrent::set_piece_priority(int index, int priority)
|
|
|
|
{
|
2007-08-09 00:38:30 +02:00
|
|
|
// INVARIANT_CHECK;
|
2007-03-20 02:59:00 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2007-03-20 02:59:00 +01:00
|
|
|
if (is_seed()) return;
|
|
|
|
|
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
|
|
|
TORRENT_ASSERT(index >= 0);
|
|
|
|
TORRENT_ASSERT(index < m_torrent_file->num_pieces());
|
2007-03-20 02:59:00 +01:00
|
|
|
|
2007-08-10 23:13:35 +02:00
|
|
|
bool filter_updated = m_picker->set_piece_priority(index, priority);
|
2007-12-15 23:51:03 +01:00
|
|
|
TORRENT_ASSERT(m_num_pieces >= m_picker->num_have_filtered());
|
2007-08-10 23:13:35 +02:00
|
|
|
if (filter_updated) update_peer_interest();
|
2007-03-20 02:59:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
int torrent::piece_priority(int index) const
|
|
|
|
{
|
2007-08-09 00:38:30 +02:00
|
|
|
// INVARIANT_CHECK;
|
2007-03-20 02:59:00 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2007-03-20 02:59:00 +01:00
|
|
|
if (is_seed()) return 1;
|
|
|
|
|
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
|
|
|
TORRENT_ASSERT(index >= 0);
|
|
|
|
TORRENT_ASSERT(index < m_torrent_file->num_pieces());
|
2007-03-20 02:59:00 +01:00
|
|
|
|
|
|
|
return m_picker->piece_priority(index);
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::prioritize_pieces(std::vector<int> const& pieces)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2007-03-20 02:59:00 +01:00
|
|
|
if (is_seed()) return;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
2007-03-20 02:59:00 +01:00
|
|
|
|
|
|
|
int index = 0;
|
2007-08-10 23:13:35 +02:00
|
|
|
bool filter_updated = false;
|
2007-03-20 02:59:00 +01:00
|
|
|
for (std::vector<int>::const_iterator i = pieces.begin()
|
|
|
|
, end(pieces.end()); i != end; ++i, ++index)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(*i >= 0);
|
|
|
|
TORRENT_ASSERT(*i <= 7);
|
2007-08-10 23:13:35 +02:00
|
|
|
filter_updated |= m_picker->set_piece_priority(index, *i);
|
2007-12-15 23:51:03 +01:00
|
|
|
TORRENT_ASSERT(m_num_pieces >= m_picker->num_have_filtered());
|
2007-03-20 02:59:00 +01:00
|
|
|
}
|
2007-08-10 23:13:35 +02:00
|
|
|
if (filter_updated) update_peer_interest();
|
2007-03-20 02:59:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::piece_priorities(std::vector<int>& pieces) const
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2007-03-20 02:59:00 +01:00
|
|
|
if (is_seed())
|
|
|
|
{
|
|
|
|
pieces.clear();
|
2007-09-01 05:00:31 +02:00
|
|
|
pieces.resize(m_torrent_file->num_pieces(), 1);
|
2007-03-20 02:59:00 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
2007-03-20 02:59:00 +01:00
|
|
|
m_picker->piece_priorities(pieces);
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
void set_if_greater(int& piece_prio, int file_prio)
|
|
|
|
{
|
|
|
|
if (file_prio > piece_prio) piece_prio = file_prio;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::prioritize_files(std::vector<int> const& files)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
// this call is only valid on torrents with metadata
|
|
|
|
if (!valid_metadata() || is_seed()) return;
|
|
|
|
|
|
|
|
// the bitmask need to have exactly one bit for every file
|
|
|
|
// in the torrent
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(int(files.size()) == m_torrent_file->num_files());
|
2007-03-20 02:59:00 +01:00
|
|
|
|
|
|
|
size_type position = 0;
|
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (m_torrent_file->num_pieces() == 0) return;
|
2007-03-21 03:09:50 +01:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
int piece_length = m_torrent_file->piece_length();
|
2007-03-21 03:09:50 +01:00
|
|
|
// initialize the piece priorities to 0, then only allow
|
|
|
|
// setting higher priorities
|
2007-09-01 05:00:31 +02:00
|
|
|
std::vector<int> pieces(m_torrent_file->num_pieces(), 0);
|
2007-03-21 03:09:50 +01:00
|
|
|
for (int i = 0; i < int(files.size()); ++i)
|
2007-03-20 02:59:00 +01:00
|
|
|
{
|
2007-03-21 03:09:50 +01:00
|
|
|
size_type start = position;
|
2007-09-01 05:00:31 +02:00
|
|
|
size_type size = m_torrent_file->file_at(i).size;
|
2007-04-18 00:56:14 +02:00
|
|
|
if (size == 0) continue;
|
|
|
|
position += size;
|
2007-03-21 03:09:50 +01:00
|
|
|
// mark all pieces of the file with this file's priority
|
|
|
|
// but only if the priority is higher than the pieces
|
|
|
|
// already set (to avoid problems with overlapping pieces)
|
|
|
|
int start_piece = int(start / piece_length);
|
2007-04-18 00:56:14 +02:00
|
|
|
int last_piece = int((position - 1) / piece_length);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(last_piece <= int(pieces.size()));
|
2007-03-21 03:09:50 +01:00
|
|
|
// if one piece spans several files, we might
|
|
|
|
// come here several times with the same start_piece, end_piece
|
|
|
|
std::for_each(pieces.begin() + start_piece
|
|
|
|
, pieces.begin() + last_piece + 1
|
|
|
|
, bind(&set_if_greater, _1, files[i]));
|
2007-03-20 02:59:00 +01:00
|
|
|
}
|
2007-03-21 03:09:50 +01:00
|
|
|
prioritize_pieces(pieces);
|
|
|
|
update_peer_interest();
|
2007-03-20 02:59:00 +01:00
|
|
|
}
|
|
|
|
|
2007-03-21 03:09:50 +01:00
|
|
|
// updates the interested flag in peers
|
|
|
|
void torrent::update_peer_interest()
|
|
|
|
{
|
|
|
|
for (peer_iterator i = begin(); i != end(); ++i)
|
2007-10-31 10:48:20 +01:00
|
|
|
(*i)->update_interest();
|
2008-03-29 19:47:24 +01:00
|
|
|
|
|
|
|
// if we used to be finished, but we aren't anymore
|
|
|
|
// we may need to connect to peers again
|
|
|
|
if (!is_finished())
|
|
|
|
m_policy.recalculate_connect_candidates();
|
2007-03-21 03:09:50 +01:00
|
|
|
}
|
2007-03-20 02:59:00 +01:00
|
|
|
|
2005-05-25 12:01:01 +02:00
|
|
|
void torrent::filter_piece(int index, bool filter)
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2006-12-11 13:48:33 +01:00
|
|
|
if (is_seed()) return;
|
|
|
|
|
2005-05-25 12:01:01 +02:00
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
|
|
|
TORRENT_ASSERT(index >= 0);
|
|
|
|
TORRENT_ASSERT(index < m_torrent_file->num_pieces());
|
2005-05-25 12:01:01 +02:00
|
|
|
|
2007-03-15 23:03:56 +01:00
|
|
|
m_picker->set_piece_priority(index, filter ? 1 : 0);
|
2007-03-29 00:22:24 +02:00
|
|
|
update_peer_interest();
|
2005-05-25 12:01:01 +02:00
|
|
|
}
|
|
|
|
|
2005-06-23 01:04:37 +02:00
|
|
|
void torrent::filter_pieces(std::vector<bool> const& bitmask)
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2005-06-23 01:04:37 +02:00
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2006-12-11 13:48:33 +01:00
|
|
|
if (is_seed()) return;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
2005-06-23 01:04:37 +02:00
|
|
|
|
|
|
|
int index = 0;
|
|
|
|
for (std::vector<bool>::const_iterator i = bitmask.begin()
|
|
|
|
, end(bitmask.end()); i != end; ++i, ++index)
|
|
|
|
{
|
2007-03-15 23:03:56 +01:00
|
|
|
if ((m_picker->piece_priority(index) == 0) == *i) continue;
|
2005-09-01 23:04:21 +02:00
|
|
|
if (*i)
|
2007-03-15 23:03:56 +01:00
|
|
|
m_picker->set_piece_priority(index, 0);
|
2005-09-01 23:04:21 +02:00
|
|
|
else
|
2007-03-15 23:03:56 +01:00
|
|
|
m_picker->set_piece_priority(index, 1);
|
2005-06-23 01:04:37 +02:00
|
|
|
}
|
2007-03-29 00:22:24 +02:00
|
|
|
update_peer_interest();
|
2005-06-23 01:04:37 +02:00
|
|
|
}
|
|
|
|
|
2005-05-25 12:01:01 +02:00
|
|
|
bool torrent::is_piece_filtered(int index) const
|
|
|
|
{
|
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2006-12-11 13:48:33 +01:00
|
|
|
if (is_seed()) return false;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
|
|
|
TORRENT_ASSERT(index >= 0);
|
|
|
|
TORRENT_ASSERT(index < m_torrent_file->num_pieces());
|
2005-05-25 12:01:01 +02:00
|
|
|
|
2007-03-15 23:03:56 +01:00
|
|
|
return m_picker->piece_priority(index) == 0;
|
2005-05-25 12:01:01 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::filtered_pieces(std::vector<bool>& bitmask) const
|
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2005-05-25 12:01:01 +02:00
|
|
|
// this call is only valid on torrents with metadata
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2006-12-11 13:48:33 +01:00
|
|
|
if (is_seed())
|
|
|
|
{
|
|
|
|
bitmask.clear();
|
2007-09-01 05:00:31 +02:00
|
|
|
bitmask.resize(m_torrent_file->num_pieces(), false);
|
2006-12-11 13:48:33 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_picker.get());
|
2005-05-25 12:01:01 +02:00
|
|
|
m_picker->filtered_pieces(bitmask);
|
|
|
|
}
|
|
|
|
|
2005-07-04 01:33:47 +02:00
|
|
|
void torrent::filter_files(std::vector<bool> const& bitmask)
|
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2005-07-04 01:33:47 +02:00
|
|
|
// this call is only valid on torrents with metadata
|
2006-12-11 13:48:33 +01:00
|
|
|
if (!valid_metadata() || is_seed()) return;
|
2005-07-04 18:27:14 +02:00
|
|
|
|
2005-07-04 18:33:54 +02:00
|
|
|
// the bitmask need to have exactly one bit for every file
|
|
|
|
// in the torrent
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT((int)bitmask.size() == m_torrent_file->num_files());
|
2005-07-04 18:33:54 +02:00
|
|
|
|
2005-07-08 16:04:14 +02:00
|
|
|
size_type position = 0;
|
2005-07-04 18:27:14 +02:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (m_torrent_file->num_pieces())
|
2005-07-04 18:33:54 +02:00
|
|
|
{
|
2007-09-01 05:00:31 +02:00
|
|
|
int piece_length = m_torrent_file->piece_length();
|
2005-07-04 18:33:54 +02:00
|
|
|
// mark all pieces as filtered, then clear the bits for files
|
|
|
|
// that should be downloaded
|
2007-09-01 05:00:31 +02:00
|
|
|
std::vector<bool> piece_filter(m_torrent_file->num_pieces(), true);
|
2005-07-06 02:58:23 +02:00
|
|
|
for (int i = 0; i < (int)bitmask.size(); ++i)
|
2005-07-04 18:27:14 +02:00
|
|
|
{
|
2005-07-08 16:04:14 +02:00
|
|
|
size_type start = position;
|
2007-09-01 05:00:31 +02:00
|
|
|
position += m_torrent_file->file_at(i).size;
|
2005-07-04 18:33:54 +02:00
|
|
|
// is the file selected for download?
|
|
|
|
if (!bitmask[i])
|
|
|
|
{
|
|
|
|
// mark all pieces of the file as downloadable
|
|
|
|
int start_piece = int(start / piece_length);
|
|
|
|
int last_piece = int(position / piece_length);
|
|
|
|
// if one piece spans several files, we might
|
|
|
|
// come here several times with the same start_piece, end_piece
|
|
|
|
std::fill(piece_filter.begin() + start_piece, piece_filter.begin()
|
|
|
|
+ last_piece + 1, false);
|
2005-07-04 18:27:14 +02:00
|
|
|
}
|
|
|
|
}
|
2005-07-04 18:33:54 +02:00
|
|
|
filter_pieces(piece_filter);
|
2005-07-04 18:27:14 +02:00
|
|
|
}
|
2005-07-04 01:33:47 +02:00
|
|
|
}
|
|
|
|
|
2004-09-12 12:12:16 +02:00
|
|
|
void torrent::replace_trackers(std::vector<announce_entry> const& urls)
|
|
|
|
{
|
|
|
|
m_trackers = urls;
|
|
|
|
if (m_currently_trying_tracker >= (int)m_trackers.size())
|
2004-09-12 15:53:00 +02:00
|
|
|
m_currently_trying_tracker = (int)m_trackers.size()-1;
|
2004-09-12 12:12:16 +02:00
|
|
|
m_last_working_tracker = -1;
|
|
|
|
}
|
|
|
|
|
2004-03-21 03:03:37 +01:00
|
|
|
tracker_request torrent::generate_tracker_request()
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_trackers.empty());
|
2007-02-20 02:42:12 +01:00
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
m_next_request = time_now() + seconds(tracker_retry_delay_max);
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-01-20 23:59:21 +01:00
|
|
|
tracker_request req;
|
2007-09-01 05:00:31 +02:00
|
|
|
req.info_hash = m_torrent_file->info_hash();
|
2006-04-25 23:04:48 +02:00
|
|
|
req.pid = m_ses.get_peer_id();
|
2004-01-20 23:59:21 +01:00
|
|
|
req.downloaded = m_stat.total_payload_download();
|
|
|
|
req.uploaded = m_stat.total_payload_upload();
|
|
|
|
req.left = bytes_left();
|
2006-09-24 13:22:25 +02:00
|
|
|
if (req.left == -1) req.left = 16*1024;
|
2004-01-21 14:16:11 +01:00
|
|
|
req.event = m_event;
|
2007-09-22 18:27:29 +02:00
|
|
|
tcp::endpoint ep = m_ses.get_ipv6_interface();
|
|
|
|
if (ep != tcp::endpoint())
|
|
|
|
req.ipv6 = ep.address().to_string();
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
if (m_event != tracker_request::stopped)
|
|
|
|
m_event = tracker_request::none;
|
2004-09-12 12:12:16 +02:00
|
|
|
req.url = m_trackers[m_currently_trying_tracker].url;
|
2007-07-02 20:44:33 +02:00
|
|
|
req.num_want = m_settings.num_want;
|
2005-03-08 15:16:14 +01:00
|
|
|
// if we are aborting. we don't want any new peers
|
|
|
|
if (req.event == tracker_request::stopped)
|
|
|
|
req.num_want = 0;
|
2004-03-21 03:03:37 +01:00
|
|
|
|
|
|
|
// default initialize, these should be set by caller
|
|
|
|
// before passing the request to the tracker_manager
|
|
|
|
req.listen_port = 0;
|
|
|
|
req.key = 0;
|
|
|
|
|
2004-01-20 23:59:21 +01:00
|
|
|
return req;
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2007-08-17 09:37:08 +02:00
|
|
|
void torrent::choke_peer(peer_connection& c)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!c.is_choked());
|
|
|
|
TORRENT_ASSERT(m_num_uploads > 0);
|
2007-08-17 09:37:08 +02:00
|
|
|
c.send_choke();
|
|
|
|
--m_num_uploads;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool torrent::unchoke_peer(peer_connection& c)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(c.is_choked());
|
2007-08-17 09:37:08 +02:00
|
|
|
if (m_num_uploads >= m_max_uploads) return false;
|
|
|
|
c.send_unchoke();
|
|
|
|
++m_num_uploads;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2007-07-06 19:15:35 +02:00
|
|
|
void torrent::cancel_block(piece_block block)
|
|
|
|
{
|
2008-03-14 11:17:27 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-07-06 19:15:35 +02:00
|
|
|
for (peer_iterator i = m_connections.begin()
|
|
|
|
, end(m_connections.end()); i != end; ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
(*i)->cancel_request(block);
|
2007-07-06 19:15:35 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
void torrent::remove_peer(peer_connection* p)
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2007-08-17 04:15:01 +02:00
|
|
|
// INVARIANT_CHECK;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p != 0);
|
2004-01-25 13:37:15 +01:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_iterator i = m_connections.find(p);
|
2007-04-12 12:21:55 +02:00
|
|
|
if (i == m_connections.end())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(false);
|
2007-04-12 12:21:55 +02:00
|
|
|
return;
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2005-10-16 23:14:08 +02:00
|
|
|
if (ready_for_connections())
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->associated_torrent().lock().get() == this);
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2007-04-15 04:14:02 +02:00
|
|
|
if (p->is_seed())
|
|
|
|
{
|
|
|
|
if (m_picker.get())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!is_seed());
|
2007-04-15 04:14:02 +02:00
|
|
|
m_picker->dec_refcount_all();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
if (m_picker.get())
|
2007-04-15 04:14:02 +02:00
|
|
|
{
|
|
|
|
const std::vector<bool>& pieces = p->get_bitfield();
|
2008-01-31 18:52:29 +01:00
|
|
|
TORRENT_ASSERT(std::count(pieces.begin(), pieces.end(), true)
|
|
|
|
< int(pieces.size()));
|
|
|
|
m_picker->dec_refcount(pieces);
|
2007-04-15 04:14:02 +02:00
|
|
|
}
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2007-08-17 09:37:08 +02:00
|
|
|
if (!p->is_choked())
|
|
|
|
--m_num_uploads;
|
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
m_policy.connection_closed(*p);
|
2007-04-12 12:21:55 +02:00
|
|
|
p->set_peer_info(0);
|
2007-10-31 10:48:20 +01:00
|
|
|
TORRENT_ASSERT(i != m_connections.end());
|
2003-10-23 01:00:57 +02:00
|
|
|
m_connections.erase(i);
|
2008-01-06 01:58:51 +01:00
|
|
|
|
|
|
|
// remove from bandwidth request-queue
|
|
|
|
for (int c = 0; c < 2; ++c)
|
|
|
|
{
|
|
|
|
for (queue_t::iterator i = m_bandwidth_queue[c].begin()
|
|
|
|
, end(m_bandwidth_queue[c].end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if (i->peer != p) continue;
|
|
|
|
m_bandwidth_queue[c].erase(i);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
void torrent::connect_to_url_seed(std::string const& url)
|
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING
|
2007-10-24 00:11:02 +02:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " resolving web seed: " << url << "\n";
|
2006-06-29 01:27:44 +02:00
|
|
|
#endif
|
|
|
|
|
2008-04-09 09:51:41 +02:00
|
|
|
std::string protocol;
|
|
|
|
std::string auth;
|
|
|
|
std::string hostname;
|
|
|
|
int port;
|
|
|
|
std::string path;
|
|
|
|
boost::tie(protocol, auth, hostname, port, path)
|
|
|
|
= parse_url_components(url);
|
|
|
|
|
|
|
|
#ifdef TORRENT_USE_OPENSSL
|
|
|
|
if (protocol != "http" && protocol != "https")
|
|
|
|
#else
|
|
|
|
if (protocol != "http")
|
|
|
|
#endif
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(
|
|
|
|
url_seed_alert(get_handle(), url, "unknown protocol"));
|
|
|
|
}
|
|
|
|
// never try it again
|
|
|
|
remove_url_seed(url);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (hostname.empty())
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(
|
|
|
|
url_seed_alert(get_handle(), url, "invalid hostname"));
|
|
|
|
}
|
|
|
|
// never try it again
|
|
|
|
remove_url_seed(url);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (port == 0)
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(
|
|
|
|
url_seed_alert(get_handle(), url, "invalid port"));
|
|
|
|
}
|
|
|
|
// never try it again
|
|
|
|
remove_url_seed(url);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2006-05-20 17:30:40 +02:00
|
|
|
m_resolving_web_seeds.insert(url);
|
2007-04-25 20:26:35 +02:00
|
|
|
proxy_settings const& ps = m_ses.web_seed_proxy();
|
|
|
|
if (ps.type == proxy_settings::http
|
|
|
|
|| ps.type == proxy_settings::http_pw)
|
|
|
|
{
|
|
|
|
// use proxy
|
|
|
|
tcp::resolver::query q(ps.hostname
|
|
|
|
, boost::lexical_cast<std::string>(ps.port));
|
2008-01-08 06:47:43 +01:00
|
|
|
m_host_resolver.async_resolve(q,
|
|
|
|
bind(&torrent::on_proxy_name_lookup, shared_from_this(), _1, _2, url));
|
2007-04-25 20:26:35 +02:00
|
|
|
}
|
|
|
|
else
|
2006-07-27 20:07:51 +02:00
|
|
|
{
|
2006-08-01 17:27:08 +02:00
|
|
|
tcp::resolver::query q(hostname, boost::lexical_cast<std::string>(port));
|
2008-01-08 06:47:43 +01:00
|
|
|
m_host_resolver.async_resolve(q,
|
2007-02-12 06:46:29 +01:00
|
|
|
bind(&torrent::on_name_lookup, shared_from_this(), _1, _2, url
|
2008-01-08 06:47:43 +01:00
|
|
|
, tcp::endpoint()));
|
2006-07-27 20:07:51 +02:00
|
|
|
}
|
2007-02-12 06:46:29 +01:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_proxy_name_lookup(asio::error_code const& e, tcp::resolver::iterator host
|
2008-04-07 04:51:21 +02:00
|
|
|
, std::string url)
|
2007-02-12 06:46:29 +01:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " completed resolve proxy hostname for: " << url << "\n";
|
2007-02-12 06:46:29 +01:00
|
|
|
#endif
|
|
|
|
|
|
|
|
if (e || host == tcp::resolver::iterator())
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
std::stringstream msg;
|
|
|
|
msg << "HTTP seed proxy hostname lookup failed: " << e.message();
|
|
|
|
m_ses.m_alerts.post_alert(
|
|
|
|
url_seed_alert(get_handle(), url, msg.str()));
|
|
|
|
}
|
|
|
|
|
|
|
|
// the name lookup failed for the http host. Don't try
|
|
|
|
// this host again
|
|
|
|
remove_url_seed(url);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (m_ses.is_aborted()) return;
|
|
|
|
|
|
|
|
tcp::endpoint a(host->endpoint());
|
2007-05-22 22:44:18 +02:00
|
|
|
|
|
|
|
using boost::tuples::ignore;
|
2007-02-12 06:46:29 +01:00
|
|
|
std::string hostname;
|
|
|
|
int port;
|
2007-05-22 22:44:18 +02:00
|
|
|
boost::tie(ignore, ignore, hostname, port, ignore)
|
2007-02-12 06:46:29 +01:00
|
|
|
= parse_url_components(url);
|
|
|
|
|
2007-04-17 07:56:43 +02:00
|
|
|
if (m_ses.m_ip_filter.access(a.address()) & ip_filter::blocked)
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(peer_blocked_alert(a.address()
|
|
|
|
, "proxy (" + hostname + ") blocked by IP filter"));
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-02-12 06:46:29 +01:00
|
|
|
tcp::resolver::query q(hostname, boost::lexical_cast<std::string>(port));
|
2008-01-08 06:47:43 +01:00
|
|
|
m_host_resolver.async_resolve(q,
|
|
|
|
bind(&torrent::on_name_lookup, shared_from_this(), _1, _2, url, a));
|
2007-02-12 06:46:29 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_name_lookup(asio::error_code const& e, tcp::resolver::iterator host
|
2008-04-07 04:51:21 +02:00
|
|
|
, std::string url, tcp::endpoint proxy)
|
2007-02-12 06:46:29 +01:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " completed resolve: " << url << "\n";
|
2007-02-12 06:46:29 +01:00
|
|
|
#endif
|
|
|
|
|
|
|
|
std::set<std::string>::iterator i = m_resolving_web_seeds.find(url);
|
|
|
|
if (i != m_resolving_web_seeds.end()) m_resolving_web_seeds.erase(i);
|
|
|
|
|
|
|
|
if (e || host == tcp::resolver::iterator())
|
|
|
|
{
|
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
std::stringstream msg;
|
|
|
|
msg << "HTTP seed hostname lookup failed: " << e.message();
|
|
|
|
m_ses.m_alerts.post_alert(
|
|
|
|
url_seed_alert(get_handle(), url, msg.str()));
|
|
|
|
}
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-02-12 06:46:29 +01:00
|
|
|
(*m_ses.m_logger) << " ** HOSTNAME LOOKUP FAILED!**: " << url << "\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// the name lookup failed for the http host. Don't try
|
|
|
|
// this host again
|
|
|
|
remove_url_seed(url);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (m_ses.is_aborted()) return;
|
|
|
|
|
|
|
|
tcp::endpoint a(host->endpoint());
|
|
|
|
|
|
|
|
if (m_ses.m_ip_filter.access(a.address()) & ip_filter::blocked)
|
|
|
|
{
|
2007-04-17 07:56:43 +02:00
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(peer_blocked_alert(a.address()
|
|
|
|
, "web seed (" + url + ") blocked by IP filter"));
|
|
|
|
}
|
2007-02-12 06:46:29 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
boost::shared_ptr<socket_type> s(new (std::nothrow) socket_type(m_ses.m_io_service));
|
|
|
|
if (!s) return;
|
2007-10-22 06:17:26 +02:00
|
|
|
|
|
|
|
bool ret = instantiate_connection(m_ses.m_io_service, m_ses.web_seed_proxy(), *s);
|
|
|
|
TORRENT_ASSERT(ret);
|
|
|
|
|
2007-04-25 20:26:35 +02:00
|
|
|
if (m_ses.web_seed_proxy().type == proxy_settings::http
|
|
|
|
|| m_ses.web_seed_proxy().type == proxy_settings::http_pw)
|
|
|
|
{
|
|
|
|
// the web seed connection will talk immediately to
|
|
|
|
// the proxy, without requiring CONNECT support
|
|
|
|
s->get<http_stream>().set_no_connect(true);
|
|
|
|
}
|
2008-02-28 08:34:07 +01:00
|
|
|
|
|
|
|
std::pair<int, int> const& out_ports = m_settings.outgoing_ports;
|
|
|
|
asio::error_code ec;
|
|
|
|
if (out_ports.first > 0 && out_ports.second >= out_ports.first)
|
|
|
|
s->bind(tcp::endpoint(address(), m_ses.next_port()), ec);
|
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
boost::intrusive_ptr<peer_connection> c(new (std::nothrow) web_peer_connection(
|
2007-04-25 20:26:35 +02:00
|
|
|
m_ses, shared_from_this(), s, a, url, 0));
|
2008-04-07 04:51:21 +02:00
|
|
|
if (!c) return;
|
2007-02-12 06:46:29 +01:00
|
|
|
|
|
|
|
#ifndef NDEBUG
|
|
|
|
c->m_in_constructor = false;
|
|
|
|
#endif
|
|
|
|
|
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)
|
|
|
|
{
|
|
|
|
boost::shared_ptr<peer_plugin> pp((*i)->new_connection(c.get()));
|
|
|
|
if (pp) c->add_extension(pp);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
// add the newly connected peer to this torrent's peer list
|
|
|
|
m_connections.insert(boost::get_pointer(c));
|
|
|
|
m_ses.m_connections.insert(c);
|
|
|
|
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2007-02-12 06:46:29 +01:00
|
|
|
try
|
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
#endif
|
2007-02-12 06:46:29 +01:00
|
|
|
// add the newly connected peer to this torrent's peer list
|
2007-10-31 10:48:20 +01:00
|
|
|
m_connections.insert(boost::get_pointer(c));
|
|
|
|
m_ses.m_connections.insert(c);
|
2008-04-01 21:17:09 +02:00
|
|
|
c->start();
|
2007-02-12 06:46:29 +01:00
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
m_ses.m_half_open.enqueue(
|
|
|
|
bind(&peer_connection::connect, c, _1)
|
|
|
|
, bind(&peer_connection::timed_out, c)
|
|
|
|
, seconds(settings().peer_connect_timeout));
|
2008-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-07-27 20:07:51 +02:00
|
|
|
}
|
2007-02-12 06:46:29 +01:00
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-02-12 06:46:29 +01:00
|
|
|
(*m_ses.m_logger) << " ** HOSTNAME LOOKUP FAILED!**: " << e.what() << "\n";
|
|
|
|
#endif
|
2006-07-27 20:07:51 +02:00
|
|
|
|
2007-02-12 06:46:29 +01:00
|
|
|
// TODO: post an error alert!
|
2008-01-07 02:10:46 +01:00
|
|
|
c->disconnect(e.what());
|
2007-02-12 06:46:29 +01:00
|
|
|
}
|
2007-05-10 03:50:11 +02:00
|
|
|
#endif
|
2008-04-07 04:51:21 +02:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2007-05-02 21:47:38 +02:00
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
2007-08-07 09:17:13 +02:00
|
|
|
namespace
|
|
|
|
{
|
|
|
|
unsigned long swap_bytes(unsigned long a)
|
|
|
|
{
|
|
|
|
return (a >> 24) | ((a & 0xff0000) >> 8) | ((a & 0xff00) << 8) | (a << 24);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-01-29 08:39:33 +01:00
|
|
|
void torrent::resolve_peer_country(boost::intrusive_ptr<peer_connection> const& p) const
|
|
|
|
{
|
|
|
|
if (m_resolving_country
|
|
|
|
|| p->has_country()
|
|
|
|
|| p->is_connecting()
|
|
|
|
|| p->is_queued()
|
2007-08-07 09:17:13 +02:00
|
|
|
|| p->in_handshake()
|
|
|
|
|| p->remote().address().is_v6()) return;
|
2007-01-29 08:39:33 +01:00
|
|
|
|
|
|
|
m_resolving_country = true;
|
2007-08-07 09:17:13 +02:00
|
|
|
asio::ip::address_v4 reversed(swap_bytes(p->remote().address().to_v4().to_ulong()));
|
|
|
|
tcp::resolver::query q(reversed.to_string() + ".zz.countries.nerd.dk", "0");
|
2008-01-08 06:47:43 +01:00
|
|
|
m_host_resolver.async_resolve(q,
|
|
|
|
bind(&torrent::on_country_lookup, shared_from_this(), _1, _2, p));
|
2007-01-29 08:39:33 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
2007-05-06 00:55:34 +02:00
|
|
|
struct country_entry
|
2007-01-29 08:39:33 +01:00
|
|
|
{
|
2007-05-06 00:55:34 +02:00
|
|
|
int code;
|
|
|
|
char const* name;
|
|
|
|
};
|
2007-01-29 08:39:33 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::on_country_lookup(asio::error_code const& error, tcp::resolver::iterator i
|
|
|
|
, intrusive_ptr<peer_connection> p) const
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
m_resolving_country = false;
|
|
|
|
|
|
|
|
// must be ordered in increasing order
|
2007-09-26 02:03:22 +02:00
|
|
|
static const country_entry country_map[] =
|
2007-01-29 08:39:33 +01:00
|
|
|
{
|
2007-05-06 00:55:34 +02:00
|
|
|
{ 4, "AF"}, { 8, "AL"}, { 10, "AQ"}, { 12, "DZ"}, { 16, "AS"}
|
|
|
|
, { 20, "AD"}, { 24, "AO"}, { 28, "AG"}, { 31, "AZ"}, { 32, "AR"}
|
|
|
|
, { 36, "AU"}, { 40, "AT"}, { 44, "BS"}, { 48, "BH"}, { 50, "BD"}
|
|
|
|
, { 51, "AM"}, { 52, "BB"}, { 56, "BE"}, { 60, "BM"}, { 64, "BT"}
|
|
|
|
, { 68, "BO"}, { 70, "BA"}, { 72, "BW"}, { 74, "BV"}, { 76, "BR"}
|
|
|
|
, { 84, "BZ"}, { 86, "IO"}, { 90, "SB"}, { 92, "VG"}, { 96, "BN"}
|
|
|
|
, {100, "BG"}, {104, "MM"}, {108, "BI"}, {112, "BY"}, {116, "KH"}
|
|
|
|
, {120, "CM"}, {124, "CA"}, {132, "CV"}, {136, "KY"}, {140, "CF"}
|
|
|
|
, {144, "LK"}, {148, "TD"}, {152, "CL"}, {156, "CN"}, {158, "TW"}
|
|
|
|
, {162, "CX"}, {166, "CC"}, {170, "CO"}, {174, "KM"}, {175, "YT"}
|
|
|
|
, {178, "CG"}, {180, "CD"}, {184, "CK"}, {188, "CR"}, {191, "HR"}
|
|
|
|
, {192, "CU"}, {203, "CZ"}, {204, "BJ"}, {208, "DK"}, {212, "DM"}
|
|
|
|
, {214, "DO"}, {218, "EC"}, {222, "SV"}, {226, "GQ"}, {231, "ET"}
|
|
|
|
, {232, "ER"}, {233, "EE"}, {234, "FO"}, {238, "FK"}, {239, "GS"}
|
|
|
|
, {242, "FJ"}, {246, "FI"}, {248, "AX"}, {250, "FR"}, {254, "GF"}
|
|
|
|
, {258, "PF"}, {260, "TF"}, {262, "DJ"}, {266, "GA"}, {268, "GE"}
|
|
|
|
, {270, "GM"}, {275, "PS"}, {276, "DE"}, {288, "GH"}, {292, "GI"}
|
|
|
|
, {296, "KI"}, {300, "GR"}, {304, "GL"}, {308, "GD"}, {312, "GP"}
|
|
|
|
, {316, "GU"}, {320, "GT"}, {324, "GN"}, {328, "GY"}, {332, "HT"}
|
|
|
|
, {334, "HM"}, {336, "VA"}, {340, "HN"}, {344, "HK"}, {348, "HU"}
|
|
|
|
, {352, "IS"}, {356, "IN"}, {360, "ID"}, {364, "IR"}, {368, "IQ"}
|
|
|
|
, {372, "IE"}, {376, "IL"}, {380, "IT"}, {384, "CI"}, {388, "JM"}
|
|
|
|
, {392, "JP"}, {398, "KZ"}, {400, "JO"}, {404, "KE"}, {408, "KP"}
|
|
|
|
, {410, "KR"}, {414, "KW"}, {417, "KG"}, {418, "LA"}, {422, "LB"}
|
|
|
|
, {426, "LS"}, {428, "LV"}, {430, "LR"}, {434, "LY"}, {438, "LI"}
|
|
|
|
, {440, "LT"}, {442, "LU"}, {446, "MO"}, {450, "MG"}, {454, "MW"}
|
|
|
|
, {458, "MY"}, {462, "MV"}, {466, "ML"}, {470, "MT"}, {474, "MQ"}
|
|
|
|
, {478, "MR"}, {480, "MU"}, {484, "MX"}, {492, "MC"}, {496, "MN"}
|
|
|
|
, {498, "MD"}, {500, "MS"}, {504, "MA"}, {508, "MZ"}, {512, "OM"}
|
|
|
|
, {516, "NA"}, {520, "NR"}, {524, "NP"}, {528, "NL"}, {530, "AN"}
|
|
|
|
, {533, "AW"}, {540, "NC"}, {548, "VU"}, {554, "NZ"}, {558, "NI"}
|
|
|
|
, {562, "NE"}, {566, "NG"}, {570, "NU"}, {574, "NF"}, {578, "NO"}
|
|
|
|
, {580, "MP"}, {581, "UM"}, {583, "FM"}, {584, "MH"}, {585, "PW"}
|
|
|
|
, {586, "PK"}, {591, "PA"}, {598, "PG"}, {600, "PY"}, {604, "PE"}
|
|
|
|
, {608, "PH"}, {612, "PN"}, {616, "PL"}, {620, "PT"}, {624, "GW"}
|
|
|
|
, {626, "TL"}, {630, "PR"}, {634, "QA"}, {634, "QA"}, {638, "RE"}
|
|
|
|
, {642, "RO"}, {643, "RU"}, {646, "RW"}, {654, "SH"}, {659, "KN"}
|
|
|
|
, {660, "AI"}, {662, "LC"}, {666, "PM"}, {670, "VC"}, {674, "SM"}
|
|
|
|
, {678, "ST"}, {682, "SA"}, {686, "SN"}, {690, "SC"}, {694, "SL"}
|
|
|
|
, {702, "SG"}, {703, "SK"}, {704, "VN"}, {705, "SI"}, {706, "SO"}
|
|
|
|
, {710, "ZA"}, {716, "ZW"}, {724, "ES"}, {732, "EH"}, {736, "SD"}
|
|
|
|
, {740, "SR"}, {744, "SJ"}, {748, "SZ"}, {752, "SE"}, {756, "CH"}
|
|
|
|
, {760, "SY"}, {762, "TJ"}, {764, "TH"}, {768, "TG"}, {772, "TK"}
|
|
|
|
, {776, "TO"}, {780, "TT"}, {784, "AE"}, {788, "TN"}, {792, "TR"}
|
|
|
|
, {795, "TM"}, {796, "TC"}, {798, "TV"}, {800, "UG"}, {804, "UA"}
|
|
|
|
, {807, "MK"}, {818, "EG"}, {826, "GB"}, {834, "TZ"}, {840, "US"}
|
|
|
|
, {850, "VI"}, {854, "BF"}, {858, "UY"}, {860, "UZ"}, {862, "VE"}
|
|
|
|
, {876, "WF"}, {882, "WS"}, {887, "YE"}, {891, "CS"}, {894, "ZM"}
|
2007-01-29 08:39:33 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
if (error || i == tcp::resolver::iterator())
|
|
|
|
{
|
|
|
|
// this is used to indicate that we shouldn't
|
|
|
|
// try to resolve it again
|
|
|
|
p->set_country("--");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
while (i != tcp::resolver::iterator()
|
|
|
|
&& !i->endpoint().address().is_v4()) ++i;
|
|
|
|
if (i != tcp::resolver::iterator())
|
|
|
|
{
|
|
|
|
// country is an ISO 3166 country code
|
|
|
|
int country = i->endpoint().address().to_v4().to_ulong() & 0xffff;
|
|
|
|
|
|
|
|
// look up the country code in the map
|
|
|
|
const int size = sizeof(country_map)/sizeof(country_map[0]);
|
2007-05-06 00:55:34 +02:00
|
|
|
country_entry tmp = {country, ""};
|
2007-09-26 17:32:49 +02:00
|
|
|
country_entry const* i =
|
2007-05-06 00:55:34 +02:00
|
|
|
std::lower_bound(country_map, country_map + size, tmp
|
|
|
|
, bind(&country_entry::code, _1) < bind(&country_entry::code, _2));
|
2007-01-29 08:39:33 +01:00
|
|
|
if (i == country_map + size
|
2007-05-06 00:55:34 +02:00
|
|
|
|| i->code != country)
|
2007-01-29 08:39:33 +01:00
|
|
|
{
|
|
|
|
// unknown country!
|
|
|
|
p->set_country("!!");
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-01-29 08:39:33 +01:00
|
|
|
(*m_ses.m_logger) << "IP " << p->remote().address() << " was mapped to unknown country: " << country << "\n";
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-05-06 00:55:34 +02:00
|
|
|
p->set_country(i->name);
|
2007-01-29 08:39:33 +01:00
|
|
|
}
|
|
|
|
}
|
2007-05-02 21:47:38 +02:00
|
|
|
#endif
|
2007-01-29 08:39:33 +01:00
|
|
|
|
2007-11-25 19:48:43 +01:00
|
|
|
void torrent::get_peer_info(std::vector<peer_info>& v)
|
|
|
|
{
|
|
|
|
v.clear();
|
|
|
|
for (peer_iterator i = begin();
|
|
|
|
i != end(); ++i)
|
|
|
|
{
|
|
|
|
peer_connection* peer = *i;
|
|
|
|
|
|
|
|
// incoming peers that haven't finished the handshake should
|
|
|
|
// not be included in this list
|
|
|
|
if (peer->associated_torrent().expired()) continue;
|
|
|
|
|
|
|
|
v.push_back(peer_info());
|
|
|
|
peer_info& p = v.back();
|
|
|
|
|
|
|
|
peer->get_peer_info(p);
|
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
|
|
|
if (resolving_countries())
|
|
|
|
resolve_peer_country(intrusive_ptr<peer_connection>(peer));
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::get_download_queue(std::vector<partial_piece_info>& queue)
|
|
|
|
{
|
|
|
|
queue.clear();
|
|
|
|
if (!valid_metadata() || is_seed()) return;
|
|
|
|
piece_picker const& p = picker();
|
|
|
|
std::vector<piece_picker::downloading_piece> const& q
|
|
|
|
= p.get_download_queue();
|
|
|
|
|
|
|
|
for (std::vector<piece_picker::downloading_piece>::const_iterator i
|
|
|
|
= q.begin(); i != q.end(); ++i)
|
|
|
|
{
|
|
|
|
partial_piece_info pi;
|
|
|
|
pi.piece_state = (partial_piece_info::state_t)i->state;
|
|
|
|
pi.blocks_in_piece = p.blocks_in_piece(i->index);
|
|
|
|
pi.finished = (int)i->finished;
|
|
|
|
pi.writing = (int)i->writing;
|
|
|
|
pi.requested = (int)i->requested;
|
2008-04-05 23:18:27 +02:00
|
|
|
int piece_size = int(torrent_file().piece_size(i->index));
|
2007-11-25 19:48:43 +01:00
|
|
|
for (int j = 0; j < pi.blocks_in_piece; ++j)
|
|
|
|
{
|
|
|
|
block_info& bi = pi.blocks[j];
|
|
|
|
bi.state = i->info[j].state;
|
|
|
|
bi.block_size = j < pi.blocks_in_piece - 1 ? m_block_size
|
|
|
|
: piece_size - (j * m_block_size);
|
|
|
|
bool complete = bi.state == block_info::writing
|
|
|
|
|| bi.state == block_info::finished;
|
|
|
|
if (i->info[j].peer == 0)
|
|
|
|
{
|
|
|
|
bi.peer = tcp::endpoint();
|
|
|
|
bi.bytes_progress = complete ? bi.block_size : 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
policy::peer* p = static_cast<policy::peer*>(i->info[j].peer);
|
|
|
|
if (p->connection)
|
|
|
|
{
|
|
|
|
bi.peer = p->connection->remote();
|
|
|
|
if (bi.state == block_info::requested)
|
|
|
|
{
|
|
|
|
boost::optional<piece_block_progress> pbp
|
|
|
|
= p->connection->downloading_piece_progress();
|
|
|
|
if (pbp && pbp->piece_index == i->index && pbp->block_index == j)
|
|
|
|
{
|
|
|
|
bi.bytes_progress = pbp->bytes_downloaded;
|
|
|
|
TORRENT_ASSERT(bi.bytes_progress <= bi.block_size);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
bi.bytes_progress = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
bi.bytes_progress = complete ? bi.block_size : 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
bi.peer = p->ip;
|
|
|
|
bi.bytes_progress = complete ? bi.block_size : 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pi.blocks[j].num_peers = i->info[j].num_peers;
|
|
|
|
}
|
|
|
|
pi.piece_index = i->index;
|
|
|
|
queue.push_back(pi);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
bool torrent::connect_to_peer(policy::peer* peerinfo)
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
|
|
|
INVARIANT_CHECK;
|
2007-05-02 19:38:37 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(peerinfo);
|
|
|
|
TORRENT_ASSERT(peerinfo->connection == 0);
|
2008-03-29 20:39:24 +01:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
peerinfo->connected = time_now();
|
2007-05-12 22:25:44 +02:00
|
|
|
#ifndef NDEBUG
|
2007-05-24 20:53:55 +02:00
|
|
|
// this asserts that we don't have duplicates in the policy's peer list
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_iterator i_ = std::find_if(m_connections.begin(), m_connections.end()
|
|
|
|
, bind(&peer_connection::remote, _1) == peerinfo->ip);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i_ == m_connections.end()
|
2007-10-31 10:48:20 +01:00
|
|
|
|| dynamic_cast<bt_peer_connection*>(*i_) == 0);
|
2007-05-12 22:25:44 +02:00
|
|
|
#endif
|
2007-05-04 18:35:29 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(want_more_peers());
|
|
|
|
TORRENT_ASSERT(m_ses.num_connections() < m_ses.max_connections());
|
2007-05-02 19:38:37 +02:00
|
|
|
|
2007-04-10 23:23:13 +02:00
|
|
|
tcp::endpoint const& a(peerinfo->ip);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT((m_ses.m_ip_filter.access(a.address()) & ip_filter::blocked) == 0);
|
2006-11-24 15:22:52 +01:00
|
|
|
|
2008-01-27 23:39:50 +01:00
|
|
|
boost::shared_ptr<socket_type> s(new socket_type(m_ses.m_io_service));
|
2007-10-22 06:17:26 +02:00
|
|
|
|
|
|
|
bool ret = instantiate_connection(m_ses.m_io_service, m_ses.peer_proxy(), *s);
|
|
|
|
TORRENT_ASSERT(ret);
|
2008-02-28 08:34:07 +01:00
|
|
|
std::pair<int, int> const& out_ports = m_ses.settings().outgoing_ports;
|
|
|
|
asio::error_code ec;
|
|
|
|
if (out_ports.first > 0 && out_ports.second >= out_ports.first)
|
|
|
|
s->bind(tcp::endpoint(address(), m_ses.next_port()), ec);
|
2007-10-22 06:17:26 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
boost::intrusive_ptr<peer_connection> c(new bt_peer_connection(
|
2007-04-10 23:23:13 +02:00
|
|
|
m_ses, shared_from_this(), s, a, peerinfo));
|
2007-06-20 20:41:53 +02:00
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
#ifndef NDEBUG
|
|
|
|
c->m_in_constructor = false;
|
|
|
|
#endif
|
|
|
|
|
2008-04-07 10:15:31 +02:00
|
|
|
c->add_stat(peerinfo->prev_amount_download, peerinfo->prev_amount_upload);
|
|
|
|
peerinfo->prev_amount_download = 0;
|
|
|
|
peerinfo->prev_amount_upload = 0;
|
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
2008-04-07 04:51:21 +02:00
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
2007-10-31 10:48:20 +01:00
|
|
|
boost::shared_ptr<peer_plugin> pp((*i)->new_connection(c.get()));
|
|
|
|
if (pp) c->add_extension(pp);
|
2008-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
|
|
|
}
|
2007-10-31 10:48:20 +01:00
|
|
|
#endif
|
2007-09-19 08:05:14 +02:00
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
// add the newly connected peer to this torrent's peer list
|
|
|
|
m_connections.insert(boost::get_pointer(c));
|
|
|
|
m_ses.m_connections.insert(c);
|
|
|
|
c->start();
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
int timeout = settings().peer_connect_timeout;
|
|
|
|
if (peerinfo) timeout += 3 * peerinfo->failcount;
|
2007-10-15 07:03:29 +02:00
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try
|
|
|
|
{
|
|
|
|
#endif
|
2007-05-05 02:29:33 +02:00
|
|
|
m_ses.m_half_open.enqueue(
|
|
|
|
bind(&peer_connection::connect, c, _1)
|
|
|
|
, bind(&peer_connection::timed_out, c)
|
2007-10-15 07:03:29 +02:00
|
|
|
, seconds(timeout));
|
2008-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
std::set<peer_connection*>::iterator i
|
|
|
|
= m_connections.find(boost::get_pointer(c));
|
2006-04-25 23:04:48 +02:00
|
|
|
if (i != m_connections.end()) m_connections.erase(i);
|
2008-01-07 02:10:46 +01:00
|
|
|
c->disconnect(e.what());
|
2007-10-31 10:48:20 +01:00
|
|
|
return false;
|
2006-01-06 21:20:20 +01:00
|
|
|
}
|
2008-04-07 04:51:21 +02:00
|
|
|
#endif
|
2007-10-31 10:48:20 +01:00
|
|
|
peerinfo->connection = c.get();
|
|
|
|
return true;
|
2003-12-01 06:01:40 +01:00
|
|
|
}
|
|
|
|
|
2007-12-28 21:11:10 +01:00
|
|
|
bool torrent::set_metadata(entry const& metadata, std::string& error)
|
2006-11-14 01:08:16 +01:00
|
|
|
{
|
2007-06-10 22:46:09 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!m_torrent_file->is_valid());
|
2007-12-28 21:11:10 +01:00
|
|
|
if (!m_torrent_file->parse_info_section(metadata, error))
|
|
|
|
{
|
|
|
|
// parse failed
|
|
|
|
return false;
|
|
|
|
}
|
2006-11-14 01:08:16 +01:00
|
|
|
|
|
|
|
if (m_ses.m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(metadata_received_alert(
|
|
|
|
get_handle(), "metadata successfully received from swarm"));
|
|
|
|
}
|
2008-03-08 07:06:31 +01:00
|
|
|
|
|
|
|
init();
|
|
|
|
|
2007-12-29 08:16:36 +01:00
|
|
|
return true;
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
|
2008-04-07 04:51:21 +02:00
|
|
|
bool torrent::attach_peer(peer_connection* p)
|
2003-12-01 06:01:40 +01:00
|
|
|
{
|
2007-08-16 14:41:46 +02:00
|
|
|
// INVARIANT_CHECK;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p != 0);
|
|
|
|
TORRENT_ASSERT(!p->is_local());
|
2004-01-13 04:08:59 +01:00
|
|
|
|
2008-03-14 11:17:27 +01:00
|
|
|
if ((m_state == torrent_status::queued_for_checking
|
2008-03-08 07:06:31 +01:00
|
|
|
|| m_state == torrent_status::checking_files)
|
2008-03-14 11:17:27 +01:00
|
|
|
&& valid_metadata())
|
2008-03-08 07:06:31 +01:00
|
|
|
{
|
|
|
|
p->disconnect("torrent is not ready to accept peers");
|
2008-04-07 04:51:21 +02:00
|
|
|
return false;
|
2008-03-08 07:06:31 +01:00
|
|
|
}
|
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
if (m_ses.m_connections.find(p) == m_ses.m_connections.end())
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
p->disconnect("peer is not properly constructed");
|
2008-04-07 04:51:21 +02:00
|
|
|
return false;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
2006-10-11 16:02:21 +02:00
|
|
|
if (m_ses.is_aborted())
|
2006-05-28 21:03:54 +02:00
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
p->disconnect("session is closing");
|
2008-04-07 04:51:21 +02:00
|
|
|
return false;
|
2006-05-28 21:03:54 +02:00
|
|
|
}
|
|
|
|
|
2007-12-27 11:19:36 +01:00
|
|
|
if (int(m_connections.size()) >= m_max_connections)
|
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
p->disconnect("reached connection limit");
|
2008-04-07 04:51:21 +02:00
|
|
|
return false;
|
2007-12-27 11:19:36 +01:00
|
|
|
}
|
|
|
|
|
2008-01-31 18:52:29 +01:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-04-25 23:04:48 +02:00
|
|
|
try
|
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
#endif
|
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)
|
|
|
|
{
|
|
|
|
boost::shared_ptr<peer_plugin> pp((*i)->new_connection(p));
|
|
|
|
if (pp) p->add_extension(pp);
|
|
|
|
}
|
|
|
|
#endif
|
2008-01-19 20:00:54 +01:00
|
|
|
if (!m_policy.new_connection(*p))
|
2008-04-07 04:51:21 +02:00
|
|
|
return false;
|
2008-01-31 18:52:29 +01:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
#if defined TORRENT_LOGGING
|
2007-12-19 22:36:54 +01:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " CLOSING CONNECTION "
|
|
|
|
<< p->remote() << " policy::new_connection threw: " << e.what() << "\n";
|
|
|
|
#endif
|
2008-01-31 18:52:29 +01:00
|
|
|
p->disconnect(e.what());
|
2008-04-07 04:51:21 +02:00
|
|
|
return false;
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2008-01-31 18:52:29 +01:00
|
|
|
#endif
|
2008-01-19 20:00:54 +01:00
|
|
|
TORRENT_ASSERT(m_connections.find(p) == m_connections.end());
|
|
|
|
peer_iterator ci = m_connections.insert(p).first;
|
2005-10-01 13:20:47 +02:00
|
|
|
#ifndef NDEBUG
|
2008-01-31 18:52:29 +01:00
|
|
|
asio::error_code ec;
|
|
|
|
TORRENT_ASSERT(p->remote() == p->get_socket()->remote_endpoint(ec) || ec);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if !defined NDEBUG && !defined TORRENT_DISABLE_INVARIANT_CHECKS
|
2007-10-31 10:48:20 +01:00
|
|
|
m_policy.check_invariant();
|
2005-10-01 13:20:47 +02:00
|
|
|
#endif
|
2008-04-07 04:51:21 +02:00
|
|
|
return true;
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2007-05-02 19:38:37 +02:00
|
|
|
bool torrent::want_more_peers() const
|
|
|
|
{
|
2007-08-16 14:41:46 +02:00
|
|
|
return int(m_connections.size()) < m_max_connections
|
2008-03-08 07:06:31 +01:00
|
|
|
&& !m_paused
|
|
|
|
&& m_state != torrent_status::checking_files
|
2008-03-29 19:47:24 +01:00
|
|
|
&& m_state != torrent_status::queued_for_checking
|
|
|
|
&& m_policy.num_connect_candidates() > 0;
|
2007-05-02 19:38:37 +02:00
|
|
|
}
|
|
|
|
|
2004-01-20 23:59:21 +01:00
|
|
|
void torrent::disconnect_all()
|
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);
|
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
|
|
|
while (!m_connections.empty())
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* p = *m_connections.begin();
|
|
|
|
TORRENT_ASSERT(p->associated_torrent().lock().get() == this);
|
2005-09-01 23:04:21 +02:00
|
|
|
|
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
|
|
|
if (m_abort)
|
2007-10-31 10:48:20 +01:00
|
|
|
(*p->m_logger) << "*** CLOSING CONNECTION 'aborting'\n";
|
2006-04-25 23:04:48 +02:00
|
|
|
else
|
2007-10-31 10:48:20 +01:00
|
|
|
(*p->m_logger) << "*** CLOSING CONNECTION 'pausing'\n";
|
2005-09-01 23:04:21 +02:00
|
|
|
#endif
|
2006-04-25 23:04:48 +02:00
|
|
|
#ifndef NDEBUG
|
|
|
|
std::size_t size = m_connections.size();
|
|
|
|
#endif
|
2008-01-17 21:03:59 +01:00
|
|
|
if (p->is_disconnecting())
|
2008-01-17 22:25:19 +01:00
|
|
|
m_connections.erase(m_connections.begin());
|
2008-01-17 21:03:59 +01:00
|
|
|
else
|
|
|
|
p->disconnect(m_abort?"stopping torrent":"pausing torrent");
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_connections.size() <= size);
|
2004-01-20 12:01:50 +01:00
|
|
|
}
|
|
|
|
}
|
2004-01-13 04:08:59 +01:00
|
|
|
|
2007-02-20 00:06:20 +01:00
|
|
|
int torrent::bandwidth_throttle(int channel) const
|
|
|
|
{
|
|
|
|
return m_bandwidth_limit[channel].throttle();
|
2007-02-19 01:01:39 +01:00
|
|
|
}
|
|
|
|
|
2008-01-15 09:37:48 +01:00
|
|
|
int torrent::bandwidth_queue_size(int channel) const
|
|
|
|
{
|
|
|
|
return (int)m_bandwidth_queue[channel].size();
|
|
|
|
}
|
|
|
|
|
2007-01-10 16:02:25 +01:00
|
|
|
void torrent::request_bandwidth(int channel
|
2007-07-03 01:48:06 +02:00
|
|
|
, boost::intrusive_ptr<peer_connection> const& p
|
2008-01-17 18:40:46 +01:00
|
|
|
, int max_block_size, int priority)
|
2007-01-10 16:02:25 +01:00
|
|
|
{
|
2008-01-07 05:47:20 +01:00
|
|
|
TORRENT_ASSERT(max_block_size > 0);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_bandwidth_limit[channel].throttle() > 0);
|
2007-12-31 10:41:50 +01:00
|
|
|
TORRENT_ASSERT(p->max_assignable_bandwidth(channel) > 0);
|
2008-01-14 00:46:43 +01:00
|
|
|
TORRENT_ASSERT(p->m_channel_state[channel] == peer_info::bw_torrent);
|
2008-01-07 05:47:20 +01:00
|
|
|
int block_size = (std::min)(m_bandwidth_limit[channel].throttle() / 10
|
|
|
|
, max_block_size);
|
2007-08-27 01:40:58 +02:00
|
|
|
if (block_size <= 0) block_size = 1;
|
2007-07-03 01:48:06 +02:00
|
|
|
|
|
|
|
if (m_bandwidth_limit[channel].max_assignable() > 0)
|
2007-01-10 16:02:25 +01:00
|
|
|
{
|
2008-01-17 18:40:46 +01:00
|
|
|
perform_bandwidth_request(channel, p, block_size, priority);
|
2007-01-10 16:02:25 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2007-07-03 01:48:06 +02:00
|
|
|
// skip forward in the queue until we find a prioritized peer
|
|
|
|
// or hit the front of it.
|
|
|
|
queue_t::reverse_iterator i = m_bandwidth_queue[channel].rbegin();
|
2008-01-17 18:40:46 +01:00
|
|
|
while (i != m_bandwidth_queue[channel].rend() && priority > i->priority)
|
|
|
|
{
|
|
|
|
++i->priority;
|
|
|
|
++i;
|
|
|
|
}
|
2008-01-27 22:03:33 +01:00
|
|
|
m_bandwidth_queue[channel].insert(i.base(), bw_queue_entry<peer_connection, torrent>(
|
2008-01-17 18:40:46 +01:00
|
|
|
p, block_size, priority));
|
2007-01-10 16:02:25 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::expire_bandwidth(int channel, int amount)
|
|
|
|
{
|
2007-01-30 18:56:42 +01:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2008-01-10 23:13:23 +01:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(amount > 0);
|
2007-01-10 16:02:25 +01:00
|
|
|
m_bandwidth_limit[channel].expire(amount);
|
2007-12-31 10:41:50 +01:00
|
|
|
queue_t tmp;
|
2007-07-03 01:48:06 +02:00
|
|
|
while (!m_bandwidth_queue[channel].empty())
|
2007-01-10 16:02:25 +01:00
|
|
|
{
|
2008-01-27 22:03:33 +01:00
|
|
|
bw_queue_entry<peer_connection, torrent> qe = m_bandwidth_queue[channel].front();
|
2007-07-03 01:48:06 +02:00
|
|
|
if (m_bandwidth_limit[channel].max_assignable() == 0)
|
|
|
|
break;
|
2007-01-10 16:02:25 +01:00
|
|
|
m_bandwidth_queue[channel].pop_front();
|
2007-12-31 10:41:50 +01:00
|
|
|
if (qe.peer->max_assignable_bandwidth(channel) <= 0)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(m_ses.m_bandwidth_manager[channel]->is_in_history(qe.peer.get()));
|
|
|
|
if (!qe.peer->is_disconnecting()) tmp.push_back(qe);
|
|
|
|
continue;
|
|
|
|
}
|
2007-07-03 01:48:06 +02:00
|
|
|
perform_bandwidth_request(channel, qe.peer
|
2008-01-17 18:40:46 +01:00
|
|
|
, qe.max_block_size, qe.priority);
|
2007-01-10 16:02:25 +01:00
|
|
|
}
|
2007-12-31 10:41:50 +01:00
|
|
|
m_bandwidth_queue[channel].insert(m_bandwidth_queue[channel].begin(), tmp.begin(), tmp.end());
|
2007-01-10 16:02:25 +01:00
|
|
|
}
|
|
|
|
|
2007-07-03 01:48:06 +02:00
|
|
|
void torrent::perform_bandwidth_request(int channel
|
|
|
|
, boost::intrusive_ptr<peer_connection> const& p
|
|
|
|
, int block_size
|
2008-01-17 18:40:46 +01:00
|
|
|
, int priority)
|
2007-07-03 01:48:06 +02:00
|
|
|
{
|
2008-01-14 00:46:43 +01:00
|
|
|
TORRENT_ASSERT(p->m_channel_state[channel] == peer_info::bw_torrent);
|
|
|
|
p->m_channel_state[channel] = peer_info::bw_global;
|
2007-07-03 01:48:06 +02:00
|
|
|
m_ses.m_bandwidth_manager[channel]->request_bandwidth(p
|
2008-01-17 18:40:46 +01:00
|
|
|
, block_size, priority);
|
2007-07-03 01:48:06 +02:00
|
|
|
m_bandwidth_limit[channel].assign(block_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::assign_bandwidth(int channel, int amount, int blk)
|
2007-01-10 16:02:25 +01:00
|
|
|
{
|
2007-01-30 18:56:42 +01:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(amount > 0);
|
|
|
|
TORRENT_ASSERT(amount <= blk);
|
2007-07-03 01:48:06 +02:00
|
|
|
if (amount < blk)
|
|
|
|
expire_bandwidth(channel, blk - amount);
|
2007-01-10 16:02:25 +01:00
|
|
|
}
|
|
|
|
|
2007-10-18 02:32:16 +02:00
|
|
|
// called when torrent is finished (all interesting
|
|
|
|
// pieces have been downloaded)
|
2005-06-11 01:12:50 +02:00
|
|
|
void torrent::finished()
|
2004-01-20 12:01:50 +01:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
if (alerts().should_post(alert::info))
|
|
|
|
{
|
|
|
|
alerts().post_alert(torrent_finished_alert(
|
|
|
|
get_handle()
|
2004-04-02 00:29:51 +02:00
|
|
|
, "torrent has finished downloading"));
|
2004-01-21 14:16:11 +01:00
|
|
|
}
|
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
m_state = torrent_status::finished;
|
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
// disconnect all seeds
|
2006-12-14 17:12:31 +01:00
|
|
|
// TODO: should disconnect all peers that have the pieces we have
|
|
|
|
// not just seeds
|
2006-04-25 23:04:48 +02:00
|
|
|
std::vector<peer_connection*> seeds;
|
2004-01-20 12:01:50 +01:00
|
|
|
for (peer_iterator i = m_connections.begin();
|
2005-06-11 01:12:50 +02:00
|
|
|
i != m_connections.end(); ++i)
|
2004-01-20 12:01:50 +01:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* p = *i;
|
|
|
|
TORRENT_ASSERT(p->associated_torrent().lock().get() == this);
|
|
|
|
if (p->is_seed())
|
2005-09-01 23:04:21 +02:00
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-10-31 10:48:20 +01:00
|
|
|
(*p->m_logger) << "*** SEED, CLOSING CONNECTION\n";
|
2005-09-01 23:04:21 +02:00
|
|
|
#endif
|
2007-10-31 10:48:20 +01:00
|
|
|
seeds.push_back(p);
|
2005-09-01 23:04:21 +02:00
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
std::for_each(seeds.begin(), seeds.end()
|
2008-01-07 02:10:46 +01:00
|
|
|
, bind(&peer_connection::disconnect, _1, "torrent finished, disconnecting seed"));
|
2004-01-21 14:16:11 +01:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_storage);
|
2007-10-02 20:34:34 +02:00
|
|
|
// we need to keep the object alive during this operation
|
|
|
|
m_storage->async_release_files(
|
|
|
|
bind(&torrent::on_files_released, shared_from_this(), _1, _2));
|
2005-06-11 01:12:50 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// called when torrent is complete (all pieces downloaded)
|
|
|
|
void torrent::completed()
|
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2004-01-21 14:16:11 +01:00
|
|
|
// make the next tracker request
|
|
|
|
// be a completed-event
|
|
|
|
m_event = tracker_request::completed;
|
2008-03-08 07:06:31 +01:00
|
|
|
m_state = torrent_status::seeding;
|
2004-01-21 14:16:11 +01:00
|
|
|
force_tracker_request();
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2004-09-12 12:12:16 +02:00
|
|
|
// this will move the tracker with the given index
|
|
|
|
// to a prioritized position in the list (move it towards
|
|
|
|
// the begining) and return the new index to the tracker.
|
|
|
|
int torrent::prioritize_tracker(int index)
|
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(index >= 0);
|
2004-09-12 12:12:16 +02:00
|
|
|
if (index >= (int)m_trackers.size()) return (int)m_trackers.size()-1;
|
|
|
|
|
|
|
|
while (index > 0 && m_trackers[index].tier == m_trackers[index-1].tier)
|
|
|
|
{
|
|
|
|
std::swap(m_trackers[index].url, m_trackers[index-1].url);
|
|
|
|
--index;
|
|
|
|
}
|
|
|
|
return index;
|
|
|
|
}
|
2003-10-23 01:00:57 +02:00
|
|
|
|
|
|
|
void torrent::try_next_tracker()
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2004-04-02 00:29:51 +02:00
|
|
|
++m_currently_trying_tracker;
|
2003-10-23 01:00:57 +02:00
|
|
|
|
2004-09-12 12:12:16 +02:00
|
|
|
if ((unsigned)m_currently_trying_tracker >= m_trackers.size())
|
2003-10-23 01:00:57 +02:00
|
|
|
{
|
2004-04-02 00:29:51 +02:00
|
|
|
int delay = tracker_retry_delay_min
|
2007-08-17 00:14:17 +02:00
|
|
|
+ (std::min)(m_failed_trackers, (int)tracker_failed_max)
|
2004-04-02 00:29:51 +02:00
|
|
|
* (tracker_retry_delay_max - tracker_retry_delay_min)
|
|
|
|
/ tracker_failed_max;
|
|
|
|
|
|
|
|
++m_failed_trackers;
|
2003-10-23 01:00:57 +02:00
|
|
|
// if we've looped the tracker list, wait a bit before retrying
|
|
|
|
m_currently_trying_tracker = 0;
|
2007-04-05 00:27:36 +02:00
|
|
|
m_next_request = time_now() + seconds(delay);
|
2007-02-12 10:20:49 +01:00
|
|
|
|
2007-02-19 01:01:39 +01:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2007-11-25 10:33:26 +01:00
|
|
|
if (m_abort) return;
|
|
|
|
|
2007-04-04 04:06:07 +02:00
|
|
|
// only start the announce if we want to announce with the dht
|
2007-11-25 10:33:26 +01:00
|
|
|
ptime now = time_now();
|
|
|
|
if (should_announce_dht() && now - m_last_dht_announce > minutes(14))
|
2007-02-12 10:20:49 +01:00
|
|
|
{
|
2007-05-17 02:01:51 +02:00
|
|
|
// force the DHT to reannounce
|
2007-11-25 10:33:26 +01:00
|
|
|
m_last_dht_announce = now;
|
2007-04-11 19:22:19 +02:00
|
|
|
boost::weak_ptr<torrent> self(shared_from_this());
|
2007-11-25 10:33:26 +01:00
|
|
|
m_ses.m_dht->announce(m_torrent_file->info_hash()
|
|
|
|
, m_ses.m_listen_sockets.front().external_port
|
2008-01-08 06:47:43 +01:00
|
|
|
, bind(&torrent::on_dht_announce_response_disp, self, _1));
|
2007-02-12 10:20:49 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// don't delay before trying the next tracker
|
2007-04-05 00:27:36 +02:00
|
|
|
m_next_request = time_now();
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
2004-09-12 12:12:16 +02:00
|
|
|
|
2003-10-23 01:00:57 +02:00
|
|
|
}
|
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
void torrent::files_checked()
|
2005-10-13 09:59:05 +02:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrent_file->is_valid());
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
2003-12-07 16:03:06 +01:00
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
m_state = torrent_status::connecting_to_tracker;
|
|
|
|
|
2006-12-04 13:20:34 +01:00
|
|
|
if (!is_seed())
|
|
|
|
{
|
2008-03-08 07:06:31 +01:00
|
|
|
m_picker->init(m_have_pieces);
|
2008-01-31 18:52:29 +01:00
|
|
|
if (m_sequential_download)
|
|
|
|
picker().sequential_download(m_sequential_download);
|
2006-12-04 13:20:34 +01:00
|
|
|
}
|
2007-05-10 00:54:26 +02:00
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
for (extension_list_t::iterator i = m_extensions.begin()
|
|
|
|
, end(m_extensions.end()); i != end; ++i)
|
|
|
|
{
|
2008-02-14 04:48:20 +01:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
|
|
|
(*i)->on_files_checked();
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
2007-05-10 00:54:26 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (is_seed())
|
2006-12-04 13:20:34 +01:00
|
|
|
{
|
2008-03-08 07:06:31 +01:00
|
|
|
m_state = torrent_status::seeding;
|
2007-03-17 18:28:59 +01:00
|
|
|
m_picker.reset();
|
2007-11-19 05:58:52 +01:00
|
|
|
if (m_ses.settings().free_torrent_hashes)
|
|
|
|
m_torrent_file->seed_free();
|
2006-12-04 13:20:34 +01:00
|
|
|
}
|
2007-03-17 18:28:59 +01:00
|
|
|
|
2005-10-16 11:15:46 +02:00
|
|
|
if (!m_connections_initialized)
|
|
|
|
{
|
|
|
|
m_connections_initialized = true;
|
|
|
|
// all peer connections have to initialize themselves now that the metadata
|
|
|
|
// is available
|
2007-10-31 10:48:20 +01:00
|
|
|
for (torrent::peer_iterator i = m_connections.begin()
|
2006-04-25 23:04:48 +02:00
|
|
|
, end(m_connections.end()); i != end;)
|
2005-10-16 11:15:46 +02:00
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
boost::intrusive_ptr<peer_connection> pc = *i;
|
|
|
|
++i;
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-11-14 01:08:16 +01:00
|
|
|
try
|
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
#endif
|
|
|
|
pc->on_metadata();
|
|
|
|
pc->init();
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
2006-04-25 23:04:48 +02:00
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
pc->disconnect(e.what());
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
2008-04-07 04:51:21 +02:00
|
|
|
#endif
|
2005-10-16 11:15:46 +02:00
|
|
|
}
|
|
|
|
}
|
2007-12-14 07:09:44 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
m_files_checked = true;
|
|
|
|
#endif
|
2003-11-07 02:44:30 +01:00
|
|
|
}
|
|
|
|
|
2004-01-07 01:48:02 +01:00
|
|
|
alert_manager& torrent::alerts() const
|
|
|
|
{
|
|
|
|
return m_ses.m_alerts;
|
|
|
|
}
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
fs::path torrent::save_path() const
|
2004-06-14 01:30:42 +02:00
|
|
|
{
|
2008-03-08 07:06:31 +01:00
|
|
|
return m_save_path;
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
void torrent::move_storage(fs::path const& save_path)
|
2004-07-18 02:39:58 +02:00
|
|
|
{
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
if (m_owning_storage.get())
|
2005-01-10 12:14:22 +01:00
|
|
|
{
|
2007-06-11 23:24:14 +02:00
|
|
|
m_owning_storage->async_move_storage(save_path
|
|
|
|
, bind(&torrent::on_storage_moved, shared_from_this(), _1, _2));
|
2005-01-10 12:14:22 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
m_save_path = save_path;
|
|
|
|
}
|
2004-07-18 02:39:58 +02:00
|
|
|
}
|
|
|
|
|
2007-06-11 23:24:14 +02:00
|
|
|
void torrent::on_storage_moved(int ret, disk_io_job const& j)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
|
|
|
if (alerts().should_post(alert::warning))
|
|
|
|
{
|
|
|
|
alerts().post_alert(storage_moved_alert(get_handle(), j.str));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
piece_manager& torrent::filesystem()
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_owning_storage.get());
|
2007-06-10 22:46:09 +02:00
|
|
|
return *m_owning_storage;
|
2004-06-14 01:30:42 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-04-09 22:09:36 +02:00
|
|
|
torrent_handle torrent::get_handle()
|
2004-01-07 01:48:02 +01:00
|
|
|
{
|
2008-04-09 22:09:36 +02:00
|
|
|
return torrent_handle(shared_from_this());
|
2004-01-07 01:48:02 +01:00
|
|
|
}
|
|
|
|
|
2006-05-15 00:30:05 +02:00
|
|
|
session_settings const& torrent::settings() const
|
|
|
|
{
|
2006-10-11 16:02:21 +02:00
|
|
|
return m_ses.settings();
|
2006-05-15 00:30:05 +02:00
|
|
|
}
|
2004-01-07 01:48:02 +01:00
|
|
|
|
2003-12-14 06:56:12 +01:00
|
|
|
#ifndef NDEBUG
|
2004-06-14 01:30:42 +02:00
|
|
|
void torrent::check_invariant() const
|
2003-12-14 06:56:12 +01:00
|
|
|
{
|
2007-10-03 19:40:18 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2008-04-13 00:08:07 +02:00
|
|
|
TORRENT_ASSERT(m_resume_data.type() == entry::dictionary_t
|
|
|
|
|| m_resume_data.type() == entry::undefined_t);
|
|
|
|
|
2008-01-06 01:58:51 +01:00
|
|
|
TORRENT_ASSERT(m_bandwidth_queue[0].size() <= m_connections.size());
|
|
|
|
TORRENT_ASSERT(m_bandwidth_queue[1].size() <= m_connections.size());
|
|
|
|
|
2008-01-17 18:40:46 +01:00
|
|
|
for (int c = 0; c < 2; ++c)
|
|
|
|
{
|
|
|
|
queue_t::const_iterator j = m_bandwidth_queue[c].begin();
|
2008-01-19 19:12:02 +01:00
|
|
|
if (j == m_bandwidth_queue[c].end()) continue;
|
2008-01-17 18:40:46 +01:00
|
|
|
++j;
|
|
|
|
for (queue_t::const_iterator i = m_bandwidth_queue[c].begin()
|
|
|
|
, end(m_bandwidth_queue[c].end()); i != end && j != end; ++i, ++j)
|
|
|
|
TORRENT_ASSERT(i->priority >= j->priority);
|
|
|
|
}
|
|
|
|
|
2007-08-17 09:37:08 +02:00
|
|
|
int num_uploads = 0;
|
2007-07-08 22:45:42 +02:00
|
|
|
std::map<piece_block, int> num_requests;
|
2006-04-25 23:04:48 +02:00
|
|
|
for (const_peer_iterator i = begin(); i != end(); ++i)
|
2006-05-28 21:03:54 +02:00
|
|
|
{
|
2008-01-07 02:10:46 +01:00
|
|
|
// make sure this peer is not a dangling pointer
|
|
|
|
TORRENT_ASSERT(m_ses.has_peer(*i));
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection const& p = *(*i);
|
2007-07-08 22:45:42 +02:00
|
|
|
for (std::deque<piece_block>::const_iterator i = p.request_queue().begin()
|
|
|
|
, end(p.request_queue().end()); i != end; ++i)
|
|
|
|
++num_requests[*i];
|
|
|
|
for (std::deque<piece_block>::const_iterator i = p.download_queue().begin()
|
|
|
|
, end(p.download_queue().end()); i != end; ++i)
|
|
|
|
++num_requests[*i];
|
2007-08-17 09:37:08 +02:00
|
|
|
if (!p.is_choked()) ++num_uploads;
|
2006-05-28 21:03:54 +02:00
|
|
|
torrent* associated_torrent = p.associated_torrent().lock().get();
|
|
|
|
if (associated_torrent != this)
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(false);
|
2006-05-28 21:03:54 +02:00
|
|
|
}
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(num_uploads == m_num_uploads);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-07-08 22:45:42 +02:00
|
|
|
if (has_picker())
|
|
|
|
{
|
|
|
|
for (std::map<piece_block, int>::iterator i = num_requests.begin()
|
|
|
|
, end(num_requests.end()); i != end; ++i)
|
|
|
|
{
|
2008-01-02 04:18:29 +01:00
|
|
|
if (!m_picker->is_downloaded(i->first))
|
|
|
|
TORRENT_ASSERT(m_picker->num_peers(i->first) == i->second);
|
2007-07-08 22:45:42 +02:00
|
|
|
}
|
2007-12-15 23:51:03 +01:00
|
|
|
TORRENT_ASSERT(m_num_pieces >= m_picker->num_have_filtered());
|
2007-07-08 22:45:42 +02:00
|
|
|
}
|
|
|
|
|
2006-12-11 13:48:33 +01:00
|
|
|
if (valid_metadata())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_abort || int(m_have_pieces.size()) == m_torrent_file->num_pieces());
|
2006-12-11 13:48:33 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_abort || m_have_pieces.empty());
|
2006-12-11 13:48:33 +01:00
|
|
|
}
|
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
for (policy::const_iterator i = m_policy.begin_peer()
|
|
|
|
, end(m_policy.end_peer()); i != end; ++i)
|
2007-09-19 08:05:14 +02:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
TORRENT_ASSERT(i->second.ip.address() == i->first);
|
2007-09-19 08:05:14 +02:00
|
|
|
}
|
2007-10-31 10:48:20 +01:00
|
|
|
|
2006-12-11 13:48:33 +01:00
|
|
|
size_type total_done = quantized_bytes_done();
|
2007-09-01 05:00:31 +02:00
|
|
|
if (m_torrent_file->is_valid())
|
2007-02-15 04:03:50 +01:00
|
|
|
{
|
|
|
|
if (is_seed())
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done == m_torrent_file->total_size());
|
2007-02-15 04:03:50 +01:00
|
|
|
else
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done != m_torrent_file->total_size());
|
2007-02-15 04:03:50 +01:00
|
|
|
}
|
2006-12-11 13:48:33 +01:00
|
|
|
else
|
2007-02-15 04:03:50 +01:00
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(total_done == 0);
|
2007-02-15 04:03:50 +01:00
|
|
|
}
|
2006-12-11 13:48:33 +01:00
|
|
|
|
2007-10-18 06:18:09 +02:00
|
|
|
if (m_picker && !m_abort)
|
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
|
|
|
|
= m_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 = m_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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// This check is very expensive.
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_num_pieces
|
2006-12-18 02:23:30 +01:00
|
|
|
== std::count(m_have_pieces.begin(), m_have_pieces.end(), true));
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!valid_metadata() || m_block_size > 0);
|
|
|
|
TORRENT_ASSERT(!valid_metadata() || (m_torrent_file->piece_length() % m_block_size) == 0);
|
|
|
|
// if (is_seed()) TORRENT_ASSERT(m_picker.get() == 0);
|
2003-12-14 06:56:12 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-01-31 18:52:29 +01:00
|
|
|
void torrent::set_sequential_download(bool sd)
|
2006-09-04 19:17:45 +02:00
|
|
|
{
|
2007-03-17 00:28:26 +01:00
|
|
|
if (has_picker())
|
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
picker().sequential_download(sd);
|
2007-03-17 00:28:26 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2008-01-31 18:52:29 +01:00
|
|
|
m_sequential_download = sd;
|
2007-03-17 00:28:26 +01:00
|
|
|
}
|
2006-09-04 19:17:45 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2004-10-29 15:21:09 +02:00
|
|
|
void torrent::set_max_uploads(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2007-08-16 14:41:46 +02:00
|
|
|
m_max_uploads = limit;
|
2004-10-29 15:21:09 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::set_max_connections(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2007-08-16 14:41:46 +02:00
|
|
|
m_max_connections = limit;
|
2004-10-29 15:21:09 +02:00
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void torrent::set_peer_upload_limit(tcp::endpoint ip, int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_iterator i = std::find_if(m_connections.begin(), m_connections.end()
|
|
|
|
, bind(&peer_connection::remote, _1) == ip);
|
|
|
|
if (i == m_connections.end()) return;
|
|
|
|
(*i)->set_upload_limit(limit);
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::set_peer_download_limit(tcp::endpoint ip, int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_iterator i = std::find_if(m_connections.begin(), m_connections.end()
|
|
|
|
, bind(&peer_connection::remote, _1) == ip);
|
|
|
|
if (i == m_connections.end()) return;
|
|
|
|
(*i)->set_download_limit(limit);
|
2006-04-25 23:04:48 +02:00
|
|
|
}
|
|
|
|
|
2004-07-24 13:54:17 +02:00
|
|
|
void torrent::set_upload_limit(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2004-07-24 13:54:17 +02:00
|
|
|
if (limit < num_peers() * 10) limit = num_peers() * 10;
|
2007-01-10 16:02:25 +01:00
|
|
|
m_bandwidth_limit[peer_connection::upload_channel].throttle(limit);
|
2004-07-24 13:54:17 +02:00
|
|
|
}
|
|
|
|
|
2007-04-10 11:25:17 +02:00
|
|
|
int torrent::upload_limit() const
|
|
|
|
{
|
|
|
|
int limit = m_bandwidth_limit[peer_connection::upload_channel].throttle();
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit == (std::numeric_limits<int>::max)()) limit = -1;
|
2007-04-10 11:25:17 +02:00
|
|
|
return limit;
|
|
|
|
}
|
|
|
|
|
2004-07-24 13:54:17 +02:00
|
|
|
void torrent::set_download_limit(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit >= -1);
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2004-07-24 13:54:17 +02:00
|
|
|
if (limit < num_peers() * 10) limit = num_peers() * 10;
|
2007-01-10 16:02:25 +01:00
|
|
|
m_bandwidth_limit[peer_connection::download_channel].throttle(limit);
|
2004-07-24 13:54:17 +02:00
|
|
|
}
|
|
|
|
|
2007-04-10 11:25:17 +02:00
|
|
|
int torrent::download_limit() const
|
|
|
|
{
|
|
|
|
int limit = m_bandwidth_limit[peer_connection::download_channel].throttle();
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit == (std::numeric_limits<int>::max)()) limit = -1;
|
2007-04-10 11:25:17 +02:00
|
|
|
return limit;
|
|
|
|
}
|
|
|
|
|
2007-10-13 05:33:33 +02:00
|
|
|
void torrent::delete_files()
|
|
|
|
{
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING
|
2007-10-13 05:33:33 +02:00
|
|
|
for (peer_iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
(*(*i)->m_logger) << "*** DELETING FILES IN TORRENT\n";
|
2007-10-13 05:33:33 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
disconnect_all();
|
2007-12-07 04:09:21 +01:00
|
|
|
if (!m_paused)
|
|
|
|
m_just_paused = true;
|
2007-10-13 05:33:33 +02:00
|
|
|
m_paused = true;
|
|
|
|
// tell the tracker that we stopped
|
|
|
|
m_event = tracker_request::stopped;
|
|
|
|
|
|
|
|
if (m_owning_storage.get())
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(m_storage);
|
|
|
|
m_storage->async_delete_files(
|
|
|
|
bind(&torrent::on_files_deleted, shared_from_this(), _1, _2));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-03-21 03:03:37 +01:00
|
|
|
void torrent::pause()
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2004-12-21 13:30:09 +01:00
|
|
|
if (m_paused) return;
|
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-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
|
|
|
if ((*i)->on_pause()) return;
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-08-16 14:41:46 +02:00
|
|
|
for (peer_iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
(*(*i)->m_logger) << "*** PAUSING TORRENT\n";
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2004-03-21 03:03:37 +01:00
|
|
|
disconnect_all();
|
|
|
|
m_paused = true;
|
2004-07-24 13:54:17 +02:00
|
|
|
// tell the tracker that we stopped
|
|
|
|
m_event = tracker_request::stopped;
|
|
|
|
m_just_paused = true;
|
2005-03-05 15:17:17 +01:00
|
|
|
// this will make the storage close all
|
|
|
|
// files and flush all cached data
|
2007-06-10 22:46:09 +02:00
|
|
|
if (m_owning_storage.get())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_storage);
|
2007-06-11 23:24:14 +02:00
|
|
|
m_storage->async_release_files(
|
2007-10-02 20:34:34 +02:00
|
|
|
bind(&torrent::on_torrent_paused, shared_from_this(), _1, _2));
|
2007-06-10 22:46:09 +02:00
|
|
|
}
|
2008-03-30 17:44:31 +02:00
|
|
|
else
|
|
|
|
{
|
|
|
|
if (alerts().should_post(alert::warning))
|
|
|
|
{
|
|
|
|
alerts().post_alert(torrent_paused_alert(get_handle(), "torrent paused"));
|
|
|
|
}
|
|
|
|
}
|
2004-03-21 03:03:37 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::resume()
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2004-12-21 13:30:09 +01:00
|
|
|
if (!m_paused) return;
|
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-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
|
|
|
if ((*i)->on_resume()) return;
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2004-03-21 03:03:37 +01:00
|
|
|
m_paused = false;
|
2004-07-24 13:54:17 +02:00
|
|
|
|
|
|
|
// tell the tracker that we're back
|
|
|
|
m_event = tracker_request::started;
|
|
|
|
force_tracker_request();
|
|
|
|
|
2004-03-21 03:03:37 +01:00
|
|
|
// make pulse be called as soon as possible
|
|
|
|
m_time_scaler = 0;
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
void torrent::second_tick(stat& accumulator, float tick_interval)
|
2003-12-01 06:01:40 +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-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try {
|
|
|
|
#endif
|
|
|
|
(*i)->tick();
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
} catch (std::exception&) {}
|
|
|
|
#endif
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2005-03-30 01:47:13 +02:00
|
|
|
if (m_paused)
|
|
|
|
{
|
|
|
|
// let the stats fade out to 0
|
2006-04-25 23:04:48 +02:00
|
|
|
m_stat.second_tick(tick_interval);
|
2005-03-30 01:47:13 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
// ---- WEB SEEDS ----
|
|
|
|
|
2007-11-28 03:54:46 +01:00
|
|
|
// re-insert urls that are to be retries into the m_web_seeds
|
|
|
|
typedef std::map<std::string, ptime>::iterator iter_t;
|
|
|
|
for (iter_t i = m_web_seeds_next_retry.begin(); i != m_web_seeds_next_retry.end();)
|
2006-04-25 23:04:48 +02:00
|
|
|
{
|
2007-11-28 03:54:46 +01:00
|
|
|
iter_t erase_element = i++;
|
|
|
|
if (erase_element->second <= time_now())
|
2007-11-27 04:37:47 +01:00
|
|
|
{
|
2007-11-28 03:54:46 +01:00
|
|
|
m_web_seeds.insert(erase_element->first);
|
|
|
|
m_web_seeds_next_retry.erase(erase_element);
|
2007-11-27 04:37:47 +01:00
|
|
|
}
|
2007-11-28 03:54:46 +01:00
|
|
|
}
|
2007-11-27 04:37:47 +01:00
|
|
|
|
2007-11-28 03:54:46 +01:00
|
|
|
// if we have everything we want we don't need to connect to any web-seed
|
|
|
|
if (!is_finished() && !m_web_seeds.empty())
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
// keep trying web-seeds if there are any
|
|
|
|
// first find out which web seeds we are connected to
|
|
|
|
std::set<std::string> web_seeds;
|
|
|
|
for (peer_iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
|
|
|
web_peer_connection* p
|
2007-10-31 10:48:20 +01:00
|
|
|
= dynamic_cast<web_peer_connection*>(*i);
|
2006-04-25 23:04:48 +02:00
|
|
|
if (!p) continue;
|
|
|
|
web_seeds.insert(p->url());
|
|
|
|
}
|
|
|
|
|
2006-05-20 17:30:40 +02:00
|
|
|
for (std::set<std::string>::iterator i = m_resolving_web_seeds.begin()
|
|
|
|
, end(m_resolving_web_seeds.end()); i != end; ++i)
|
|
|
|
web_seeds.insert(web_seeds.begin(), *i);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
|
|
|
// from the list of available web seeds, subtract the ones we are
|
|
|
|
// already connected to.
|
|
|
|
std::vector<std::string> not_connected_web_seeds;
|
|
|
|
std::set_difference(m_web_seeds.begin(), m_web_seeds.end(), web_seeds.begin()
|
|
|
|
, web_seeds.end(), std::back_inserter(not_connected_web_seeds));
|
|
|
|
|
|
|
|
// connect to all of those that we aren't connected to
|
|
|
|
std::for_each(not_connected_web_seeds.begin(), not_connected_web_seeds.end()
|
|
|
|
, bind(&torrent::connect_to_url_seed, this, _1));
|
|
|
|
}
|
|
|
|
|
2004-01-13 04:08:59 +01:00
|
|
|
for (peer_iterator i = m_connections.begin();
|
2007-05-24 19:07:43 +02:00
|
|
|
i != m_connections.end();)
|
2003-12-07 06:53:04 +01:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* p = *i;
|
2007-05-24 19:07:43 +02:00
|
|
|
++i;
|
2004-03-23 23:58:18 +01:00
|
|
|
m_stat += p->statistics();
|
2004-03-28 19:45:37 +02:00
|
|
|
// updates the peer connection's ul/dl bandwidth
|
|
|
|
// resource requests
|
2008-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2007-05-24 19:07:43 +02:00
|
|
|
try
|
|
|
|
{
|
2008-04-07 04:51:21 +02:00
|
|
|
#endif
|
2007-05-24 19:07:43 +02:00
|
|
|
p->second_tick(tick_interval);
|
2008-04-07 04:51:21 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2007-05-24 19:07:43 +02:00
|
|
|
}
|
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
2008-02-17 21:57:16 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-05-24 19:07:43 +02:00
|
|
|
(*p->m_logger) << "**ERROR**: " << e.what() << "\n";
|
|
|
|
#endif
|
|
|
|
p->set_failed();
|
2008-01-07 02:10:46 +01:00
|
|
|
p->disconnect(e.what());
|
2007-05-24 19:07:43 +02:00
|
|
|
}
|
2008-04-07 04:51:21 +02:00
|
|
|
#endif
|
2003-12-07 06:53:04 +01:00
|
|
|
}
|
2004-04-18 15:41:08 +02:00
|
|
|
accumulator += m_stat;
|
2006-07-26 12:21:25 +02:00
|
|
|
m_stat.second_tick(tick_interval);
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2004-10-29 15:21:09 +02:00
|
|
|
m_time_scaler--;
|
|
|
|
if (m_time_scaler <= 0)
|
|
|
|
{
|
2007-08-16 14:41:46 +02:00
|
|
|
m_time_scaler = 10;
|
2007-10-31 10:48:20 +01:00
|
|
|
m_policy.pulse();
|
2004-10-29 15:21:09 +02:00
|
|
|
}
|
2004-03-23 23:58:18 +01:00
|
|
|
}
|
|
|
|
|
2007-11-27 04:37:47 +01:00
|
|
|
void torrent::retry_url_seed(std::string const& url)
|
|
|
|
{
|
|
|
|
m_web_seeds_next_retry[url] = time_now()
|
|
|
|
+ seconds(m_ses.settings().urlseed_wait_retry);
|
|
|
|
}
|
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
bool torrent::try_connect_peer()
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(want_more_peers());
|
2008-03-29 20:39:24 +01:00
|
|
|
if (m_deficit_counter < 100) return false;
|
|
|
|
m_deficit_counter -= 100;
|
|
|
|
bool ret = m_policy.connect_one_peer();
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
void torrent::give_connect_points(int points)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(points <= 100);
|
|
|
|
TORRENT_ASSERT(points > 0);
|
|
|
|
TORRENT_ASSERT(want_more_peers());
|
|
|
|
m_deficit_counter += points;
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
|
2008-04-13 00:08:07 +02:00
|
|
|
void torrent::async_verify_piece(int piece_index, boost::function<void(int)> const& f)
|
2003-12-07 06:53:04 +01:00
|
|
|
{
|
2007-10-18 02:32:16 +02:00
|
|
|
// INVARIANT_CHECK;
|
2006-04-25 23:04:48 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_storage);
|
|
|
|
TORRENT_ASSERT(m_storage->refcount() > 0);
|
|
|
|
TORRENT_ASSERT(piece_index >= 0);
|
|
|
|
TORRENT_ASSERT(piece_index < m_torrent_file->num_pieces());
|
|
|
|
TORRENT_ASSERT(piece_index < (int)m_have_pieces.size());
|
2008-02-18 04:07:14 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
if (m_picker)
|
|
|
|
{
|
|
|
|
int blocks_in_piece = m_picker->blocks_in_piece(piece_index);
|
|
|
|
for (int i = 0; i < blocks_in_piece; ++i)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(m_picker->num_peers(piece_block(piece_index, i)) == 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2004-01-25 13:37:15 +01:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
m_storage->async_hash(piece_index, bind(&torrent::on_piece_verified
|
|
|
|
, shared_from_this(), _1, _2, f));
|
2008-01-31 18:52:29 +01:00
|
|
|
#if !defined NDEBUG && !defined TORRENT_DISABLE_INVARIANT_CHECKS
|
2007-10-18 02:32:16 +02:00
|
|
|
check_invariant();
|
|
|
|
#endif
|
2007-06-10 22:46:09 +02:00
|
|
|
}
|
2003-12-07 06:53:04 +01:00
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
void torrent::on_piece_verified(int ret, disk_io_job const& j
|
2008-04-13 00:08:07 +02:00
|
|
|
, boost::function<void(int)> f)
|
2007-06-10 22:46:09 +02:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
2008-04-13 00:08:07 +02:00
|
|
|
|
|
|
|
// return value:
|
|
|
|
// 0: success, piece passed hash check
|
|
|
|
// -1: disk failure
|
|
|
|
// -2: hash check failed
|
|
|
|
|
|
|
|
if (ret == -1)
|
|
|
|
{
|
|
|
|
if (alerts().should_post(alert::fatal))
|
|
|
|
{
|
|
|
|
alerts().post_alert(file_error_alert(j.error_file, get_handle(), j.str));
|
|
|
|
}
|
|
|
|
pause();
|
|
|
|
}
|
|
|
|
f(ret);
|
2003-12-01 06:01:40 +01:00
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
const tcp::endpoint& torrent::current_tracker() const
|
2004-03-01 01:50:00 +01:00
|
|
|
{
|
|
|
|
return m_tracker_address;
|
|
|
|
}
|
|
|
|
|
2006-06-12 01:24:36 +02:00
|
|
|
void torrent::file_progress(std::vector<float>& fp) const
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(valid_metadata());
|
2006-06-12 01:24:36 +02:00
|
|
|
|
|
|
|
fp.clear();
|
2007-09-01 05:00:31 +02:00
|
|
|
fp.resize(m_torrent_file->num_files(), 0.f);
|
2006-06-12 01:24:36 +02:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
for (int i = 0; i < m_torrent_file->num_files(); ++i)
|
2006-06-12 01:24:36 +02:00
|
|
|
{
|
2007-09-01 05:00:31 +02:00
|
|
|
peer_request ret = m_torrent_file->map_file(i, 0, 0);
|
|
|
|
size_type size = m_torrent_file->file_at(i).size;
|
2006-10-08 22:36:05 +02:00
|
|
|
|
|
|
|
// zero sized files are considered
|
|
|
|
// 100% done all the time
|
|
|
|
if (size == 0)
|
|
|
|
{
|
|
|
|
fp[i] = 1.f;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2006-06-12 01:24:36 +02:00
|
|
|
size_type done = 0;
|
2006-08-27 20:38:30 +02:00
|
|
|
while (size > 0)
|
2006-06-12 01:24:36 +02:00
|
|
|
{
|
2008-02-25 11:28:53 +01:00
|
|
|
size_type bytes_step = (std::min)(size_type(m_torrent_file->piece_size(ret.piece)
|
|
|
|
- ret.start), size);
|
2006-06-12 01:24:36 +02:00
|
|
|
if (m_have_pieces[ret.piece]) done += bytes_step;
|
|
|
|
++ret.piece;
|
|
|
|
ret.start = 0;
|
2006-08-27 20:38:30 +02:00
|
|
|
size -= bytes_step;
|
2006-06-12 01:24:36 +02:00
|
|
|
}
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(size == 0);
|
2006-10-08 22:36:05 +02:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
fp[i] = static_cast<float>(done) / m_torrent_file->file_at(i).size;
|
2006-06-12 01:24:36 +02:00
|
|
|
}
|
|
|
|
}
|
2005-10-16 18:58:41 +02:00
|
|
|
|
2003-10-31 05:02:51 +01:00
|
|
|
torrent_status torrent::status() const
|
2003-10-26 18:35:23 +01:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(std::accumulate(
|
2004-01-15 17:45:34 +01:00
|
|
|
m_have_pieces.begin()
|
|
|
|
, m_have_pieces.end()
|
|
|
|
, 0) == m_num_pieces);
|
2003-10-30 00:28:09 +01:00
|
|
|
|
2004-01-15 17:45:34 +01:00
|
|
|
torrent_status st;
|
2003-10-30 00:28:09 +01:00
|
|
|
|
2008-01-17 18:40:46 +01:00
|
|
|
st.up_bandwidth_queue = (int)m_bandwidth_queue[peer_connection::upload_channel].size();
|
|
|
|
st.down_bandwidth_queue = (int)m_bandwidth_queue[peer_connection::download_channel].size();
|
|
|
|
|
2007-11-20 23:46:27 +01:00
|
|
|
st.num_peers = (int)std::count_if(m_connections.begin(), m_connections.end()
|
|
|
|
, !boost::bind(&peer_connection::is_connecting, _1));
|
|
|
|
|
2008-04-01 19:38:19 +02:00
|
|
|
st.list_peers = m_policy.num_peers();
|
|
|
|
st.list_seeds = m_policy.num_seeds();
|
2008-03-29 19:47:24 +01:00
|
|
|
st.connect_candidates = m_policy.num_connect_candidates();
|
2005-11-05 11:56:47 +01:00
|
|
|
|
2007-10-08 22:01:36 +02:00
|
|
|
st.storage_mode = m_storage_mode;
|
2007-07-23 06:36:56 +02:00
|
|
|
|
2005-02-23 21:38:29 +01:00
|
|
|
st.num_complete = m_complete;
|
|
|
|
st.num_incomplete = m_incomplete;
|
2004-03-21 03:03:37 +01:00
|
|
|
st.paused = m_paused;
|
2005-05-30 19:43:03 +02:00
|
|
|
boost::tie(st.total_done, st.total_wanted_done) = bytes_done();
|
2007-12-09 20:38:17 +01:00
|
|
|
TORRENT_ASSERT(st.total_done >= st.total_wanted_done);
|
2003-10-31 13:07:07 +01:00
|
|
|
|
2003-12-22 08:14:35 +01:00
|
|
|
// payload transfer
|
|
|
|
st.total_payload_download = m_stat.total_payload_download();
|
|
|
|
st.total_payload_upload = m_stat.total_payload_upload();
|
|
|
|
|
|
|
|
// total transfer
|
|
|
|
st.total_download = m_stat.total_payload_download()
|
|
|
|
+ m_stat.total_protocol_download();
|
|
|
|
st.total_upload = m_stat.total_payload_upload()
|
|
|
|
+ m_stat.total_protocol_upload();
|
|
|
|
|
2004-04-18 14:28:02 +02:00
|
|
|
// failed bytes
|
|
|
|
st.total_failed_bytes = m_total_failed_bytes;
|
2006-04-25 23:04:48 +02:00
|
|
|
st.total_redundant_bytes = m_total_redundant_bytes;
|
2004-04-18 14:28:02 +02:00
|
|
|
|
2003-12-22 08:14:35 +01:00
|
|
|
// transfer rate
|
2003-12-07 06:53:04 +01:00
|
|
|
st.download_rate = m_stat.download_rate();
|
|
|
|
st.upload_rate = m_stat.upload_rate();
|
2004-04-18 14:28:02 +02:00
|
|
|
st.download_payload_rate = m_stat.download_payload_rate();
|
|
|
|
st.upload_payload_rate = m_stat.upload_payload_rate();
|
2003-12-22 08:14:35 +01:00
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
st.next_announce = boost::posix_time::seconds(
|
|
|
|
total_seconds(next_announce() - time_now()));
|
|
|
|
if (st.next_announce.is_negative())
|
|
|
|
st.next_announce = boost::posix_time::seconds(0);
|
|
|
|
|
2004-01-17 21:04:19 +01:00
|
|
|
st.announce_interval = boost::posix_time::seconds(m_duration);
|
|
|
|
|
2005-03-11 18:21:56 +01:00
|
|
|
if (m_last_working_tracker >= 0)
|
|
|
|
{
|
|
|
|
st.current_tracker
|
|
|
|
= m_trackers[m_last_working_tracker].url;
|
|
|
|
}
|
|
|
|
|
2007-08-17 09:37:08 +02:00
|
|
|
st.num_uploads = m_num_uploads;
|
2007-08-16 14:41:46 +02:00
|
|
|
st.uploads_limit = m_max_uploads;
|
|
|
|
st.num_connections = int(m_connections.size());
|
|
|
|
st.connections_limit = m_max_connections;
|
2004-06-14 01:30:42 +02:00
|
|
|
// if we don't have any metadata, stop here
|
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
st.state = m_state;
|
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
if (!valid_metadata())
|
|
|
|
{
|
|
|
|
if (m_got_tracker_response == false)
|
|
|
|
st.state = torrent_status::connecting_to_tracker;
|
|
|
|
else
|
|
|
|
st.state = torrent_status::downloading_metadata;
|
2004-09-16 03:14:16 +02:00
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
st.progress = m_progress;
|
2006-04-25 23:04:48 +02:00
|
|
|
st.block_size = 0;
|
2004-06-14 01:30:42 +02:00
|
|
|
return st;
|
|
|
|
}
|
|
|
|
|
2006-04-25 23:04:48 +02:00
|
|
|
st.block_size = block_size();
|
|
|
|
|
2004-06-14 01:30:42 +02:00
|
|
|
// fill in status that depends on metadata
|
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
st.total_wanted = m_torrent_file->total_size();
|
2005-05-30 19:43:03 +02:00
|
|
|
|
|
|
|
if (m_picker.get() && (m_picker->num_filtered() > 0
|
|
|
|
|| m_picker->num_have_filtered() > 0))
|
|
|
|
{
|
|
|
|
int filtered_pieces = m_picker->num_filtered()
|
|
|
|
+ m_picker->num_have_filtered();
|
2007-09-01 05:00:31 +02:00
|
|
|
int last_piece_index = m_torrent_file->num_pieces() - 1;
|
2007-03-15 23:03:56 +01:00
|
|
|
if (m_picker->piece_priority(last_piece_index) == 0)
|
2005-05-30 19:43:03 +02:00
|
|
|
{
|
2007-09-01 05:00:31 +02:00
|
|
|
st.total_wanted -= m_torrent_file->piece_size(last_piece_index);
|
2005-05-30 19:43:03 +02:00
|
|
|
--filtered_pieces;
|
|
|
|
}
|
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
st.total_wanted -= filtered_pieces * m_torrent_file->piece_length();
|
2005-05-30 19:43:03 +02:00
|
|
|
}
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(st.total_wanted >= st.total_wanted_done);
|
2005-08-15 00:04:58 +02:00
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
if (m_state == torrent_status::checking_files)
|
|
|
|
st.progress = m_progress;
|
|
|
|
else if (st.total_wanted == 0) st.progress = 1.f;
|
2005-08-15 00:04:58 +02:00
|
|
|
else st.progress = st.total_wanted_done
|
2008-04-05 23:18:27 +02:00
|
|
|
/ static_cast<float>(st.total_wanted);
|
2004-01-15 02:01:09 +01:00
|
|
|
|
2004-01-15 17:45:34 +01:00
|
|
|
st.pieces = &m_have_pieces;
|
2006-04-25 23:04:48 +02:00
|
|
|
st.num_pieces = m_num_pieces;
|
2004-08-05 15:56:26 +02:00
|
|
|
st.num_seeds = num_seeds();
|
2006-12-04 13:20:34 +01:00
|
|
|
if (m_picker.get())
|
|
|
|
st.distributed_copies = m_picker->distributed_copies();
|
|
|
|
else
|
|
|
|
st.distributed_copies = -1;
|
2003-10-31 05:02:51 +01:00
|
|
|
return st;
|
2003-10-26 18:35:23 +01:00
|
|
|
}
|
|
|
|
|
2004-08-05 15:56:26 +02:00
|
|
|
int torrent::num_seeds() const
|
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-06-03 07:57:05 +02:00
|
|
|
return (int)std::count_if(m_connections.begin(), m_connections.end()
|
2007-10-31 10:48:20 +01:00
|
|
|
, boost::bind(&peer_connection::is_seed, _1));
|
2004-09-12 15:53:00 +02:00
|
|
|
}
|
|
|
|
|
2005-03-24 13:13:47 +01:00
|
|
|
void torrent::tracker_request_timed_out(
|
2007-11-20 23:46:27 +01:00
|
|
|
tracker_request const& r)
|
2003-12-22 08:14:35 +01:00
|
|
|
{
|
2006-04-25 23:04:48 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
2007-03-31 00:00:26 +02:00
|
|
|
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2003-12-22 08:14:35 +01:00
|
|
|
debug_log("*** tracker timed out");
|
|
|
|
#endif
|
2006-05-28 21:03:54 +02:00
|
|
|
|
2003-12-22 08:14:35 +01:00
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
std::stringstream s;
|
2007-11-20 23:46:27 +01:00
|
|
|
s << "tracker: \"" << r.url << "\" timed out";
|
|
|
|
if (r.kind == tracker_request::announce_request)
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(tracker_alert(get_handle()
|
|
|
|
, m_failed_trackers + 1, 0, s.str()));
|
|
|
|
}
|
|
|
|
else if (r.kind == tracker_request::scrape_request)
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(scrape_failed_alert(get_handle(), s.str()));
|
|
|
|
}
|
2003-12-22 08:14:35 +01:00
|
|
|
}
|
2007-11-20 23:46:27 +01:00
|
|
|
|
|
|
|
if (r.kind == tracker_request::announce_request)
|
|
|
|
try_next_tracker();
|
2003-12-22 08:14:35 +01:00
|
|
|
}
|
|
|
|
|
2004-01-26 11:29:00 +01:00
|
|
|
// TODO: with some response codes, we should just consider
|
2003-12-22 08:14:35 +01:00
|
|
|
// the tracker as a failure and not retry
|
|
|
|
// it anymore
|
2007-11-20 23:46:27 +01:00
|
|
|
void torrent::tracker_request_error(tracker_request const& r
|
2005-03-24 13:13:47 +01:00
|
|
|
, int response_code, const std::string& str)
|
2003-12-22 08:14:35 +01:00
|
|
|
{
|
2007-03-31 00:00:26 +02:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_ses.m_mutex);
|
|
|
|
|
2006-05-28 21:03:54 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2003-12-22 08:14:35 +01:00
|
|
|
debug_log(std::string("*** tracker error: ") + str);
|
|
|
|
#endif
|
|
|
|
if (m_ses.m_alerts.should_post(alert::warning))
|
|
|
|
{
|
|
|
|
std::stringstream s;
|
2007-11-20 23:46:27 +01:00
|
|
|
s << "tracker: \"" << r.url << "\" " << str;
|
|
|
|
if (r.kind == tracker_request::announce_request)
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(tracker_alert(get_handle()
|
|
|
|
, m_failed_trackers + 1, response_code, s.str()));
|
|
|
|
}
|
|
|
|
else if (r.kind == tracker_request::scrape_request)
|
|
|
|
{
|
|
|
|
m_ses.m_alerts.post_alert(scrape_failed_alert(get_handle(), s.str()));
|
|
|
|
}
|
2003-12-22 08:14:35 +01:00
|
|
|
}
|
|
|
|
|
2007-11-20 23:46:27 +01:00
|
|
|
if (r.kind == tracker_request::announce_request)
|
|
|
|
try_next_tracker();
|
2003-12-22 08:14:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-02-07 08:09:52 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2003-11-20 20:58:29 +01:00
|
|
|
void torrent::debug_log(const std::string& line)
|
|
|
|
{
|
2007-10-26 09:14:19 +02:00
|
|
|
(*m_ses.m_logger) << time_now_string() << " " << line << "\n";
|
2003-11-20 20:58:29 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2005-06-15 14:54:35 +02:00
|
|
|
}
|
|
|
|
|