2006-10-11 22:57:54 +02:00
|
|
|
/*
|
|
|
|
|
|
|
|
Copyright (c) 2006, Arvid Norberg, Magnus Jonsson
|
|
|
|
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"
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
#include <ctime>
|
|
|
|
#include <iostream>
|
|
|
|
#include <fstream>
|
|
|
|
#include <iomanip>
|
|
|
|
#include <iterator>
|
|
|
|
#include <algorithm>
|
|
|
|
#include <set>
|
|
|
|
#include <cctype>
|
|
|
|
#include <algorithm>
|
|
|
|
|
|
|
|
#ifdef _MSC_VER
|
|
|
|
#pragma warning(push, 1)
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
#include <boost/filesystem/convenience.hpp>
|
|
|
|
#include <boost/filesystem/exception.hpp>
|
|
|
|
#include <boost/limits.hpp>
|
|
|
|
#include <boost/bind.hpp>
|
|
|
|
|
|
|
|
#ifdef _MSC_VER
|
|
|
|
#pragma warning(pop)
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#include "libtorrent/peer_id.hpp"
|
|
|
|
#include "libtorrent/torrent_info.hpp"
|
|
|
|
#include "libtorrent/tracker_manager.hpp"
|
|
|
|
#include "libtorrent/bencode.hpp"
|
|
|
|
#include "libtorrent/hasher.hpp"
|
|
|
|
#include "libtorrent/entry.hpp"
|
|
|
|
#include "libtorrent/session.hpp"
|
|
|
|
#include "libtorrent/fingerprint.hpp"
|
|
|
|
#include "libtorrent/entry.hpp"
|
|
|
|
#include "libtorrent/alert_types.hpp"
|
|
|
|
#include "libtorrent/invariant_check.hpp"
|
|
|
|
#include "libtorrent/file.hpp"
|
|
|
|
#include "libtorrent/bt_peer_connection.hpp"
|
|
|
|
#include "libtorrent/ip_filter.hpp"
|
|
|
|
#include "libtorrent/socket.hpp"
|
|
|
|
#include "libtorrent/aux_/session_impl.hpp"
|
|
|
|
#include "libtorrent/kademlia/dht_tracker.hpp"
|
2007-09-22 18:27:29 +02:00
|
|
|
#include "libtorrent/enum_net.hpp"
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-07-07 01:06:58 +02:00
|
|
|
#ifndef TORRENT_DISABLE_ENCRYPTION
|
|
|
|
|
|
|
|
#include <openssl/crypto.h>
|
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
// openssl requires this to clean up internal
|
|
|
|
// structures it allocates
|
|
|
|
struct openssl_cleanup
|
|
|
|
{
|
|
|
|
~openssl_cleanup() { CRYPTO_cleanup_all_ex_data(); }
|
|
|
|
} openssl_global_destructor;
|
|
|
|
}
|
|
|
|
|
2007-10-15 21:02:54 +02:00
|
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
|
|
// for ERROR_SEM_TIMEOUT
|
|
|
|
#include <winerror.h>
|
2007-07-07 01:06:58 +02:00
|
|
|
#endif
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
using boost::shared_ptr;
|
|
|
|
using boost::weak_ptr;
|
|
|
|
using boost::bind;
|
|
|
|
using boost::mutex;
|
|
|
|
using libtorrent::aux::session_impl;
|
|
|
|
|
2007-06-10 22:46:09 +02:00
|
|
|
namespace libtorrent {
|
|
|
|
|
|
|
|
namespace fs = boost::filesystem;
|
|
|
|
|
|
|
|
namespace detail
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
|
|
|
|
std::string generate_auth_string(std::string const& user
|
|
|
|
, std::string const& passwd)
|
|
|
|
{
|
|
|
|
if (user.empty()) return std::string();
|
|
|
|
return user + ":" + passwd;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
}
|
|
|
|
namespace aux {
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2006-11-15 22:39:58 +01:00
|
|
|
struct seed_random_generator
|
|
|
|
{
|
|
|
|
seed_random_generator()
|
2006-11-30 12:56:19 +01:00
|
|
|
{
|
2007-04-10 09:52:58 +02:00
|
|
|
std::srand(total_microseconds(time_now() - min_time()));
|
2006-11-30 12:56:19 +01:00
|
|
|
}
|
2006-11-15 22:39:58 +01:00
|
|
|
};
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
session_impl::session_impl(
|
|
|
|
std::pair<int, int> listen_port_range
|
|
|
|
, fingerprint const& cl_fprint
|
2007-11-16 22:21:28 +01:00
|
|
|
, char const* listen_interface
|
2008-02-17 23:51:03 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-11-16 22:21:28 +01:00
|
|
|
, fs::path const& logpath
|
|
|
|
#endif
|
|
|
|
)
|
2008-04-09 07:19:11 +02:00
|
|
|
:
|
|
|
|
#ifndef TORRENT_DISABLE_POOL_ALLOCATOR
|
|
|
|
m_send_buffers(send_buffer_size),
|
|
|
|
#endif
|
|
|
|
m_files(40)
|
2008-02-08 11:22:05 +01:00
|
|
|
, m_io_service()
|
|
|
|
, m_disk_thread(m_io_service)
|
2007-05-24 20:53:55 +02:00
|
|
|
, m_half_open(m_io_service)
|
2007-07-03 01:48:06 +02:00
|
|
|
, m_download_channel(m_io_service, peer_connection::download_channel)
|
2008-01-12 19:47:26 +01:00
|
|
|
#ifdef TORRENT_VERBOSE_BANDWIDTH_LIMIT
|
|
|
|
, m_upload_channel(m_io_service, peer_connection::upload_channel, true)
|
|
|
|
#else
|
2007-07-03 01:48:06 +02:00
|
|
|
, m_upload_channel(m_io_service, peer_connection::upload_channel)
|
2008-01-12 19:47:26 +01:00
|
|
|
#endif
|
2007-04-25 20:26:35 +02:00
|
|
|
, m_tracker_manager(m_settings, m_tracker_proxy)
|
2007-09-22 18:27:29 +02:00
|
|
|
, m_listen_port_retries(listen_port_range.second - listen_port_range.first)
|
2006-10-11 22:57:54 +02:00
|
|
|
, m_listen_interface(address::from_string(listen_interface), listen_port_range.first)
|
|
|
|
, m_abort(false)
|
2007-08-16 14:41:46 +02:00
|
|
|
, m_max_uploads(8)
|
2008-01-13 12:18:18 +01:00
|
|
|
, m_allowed_upload_slots(8)
|
2007-08-16 14:41:46 +02:00
|
|
|
, m_max_connections(200)
|
|
|
|
, m_num_unchoked(0)
|
|
|
|
, m_unchoke_time_scaler(0)
|
|
|
|
, m_optimistic_unchoke_time_scaler(0)
|
|
|
|
, m_disconnect_time_scaler(0)
|
2006-10-11 22:57:54 +02:00
|
|
|
, m_incoming_connection(false)
|
2007-04-05 00:27:36 +02:00
|
|
|
, m_last_tick(time_now())
|
2007-03-15 23:03:56 +01:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
, m_dht_same_port(true)
|
|
|
|
, m_external_udp_port(0)
|
2008-03-25 05:38:59 +01:00
|
|
|
, m_dht_socket(m_io_service, bind(&session_impl::on_receive_udp, this, _1, _2, _3, _4)
|
2007-12-09 05:15:24 +01:00
|
|
|
, m_half_open)
|
2007-03-15 23:03:56 +01:00
|
|
|
#endif
|
2006-12-15 18:47:21 +01:00
|
|
|
, m_timer(m_io_service)
|
2007-05-05 02:29:33 +02:00
|
|
|
, m_next_connect_torrent(0)
|
2008-02-17 23:51:03 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2007-11-16 22:21:28 +01:00
|
|
|
, m_logpath(logpath)
|
2008-04-05 06:53:22 +02:00
|
|
|
#endif
|
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
2008-04-11 10:46:43 +02:00
|
|
|
, m_asnum_db(0)
|
|
|
|
, m_country_db(0)
|
2007-11-16 22:21:28 +01:00
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2008-04-06 21:17:58 +02:00
|
|
|
m_tcp_mapping[0] = -1;
|
|
|
|
m_tcp_mapping[1] = -1;
|
|
|
|
m_udp_mapping[0] = -1;
|
|
|
|
m_udp_mapping[1] = -1;
|
2007-09-10 03:57:40 +02:00
|
|
|
#ifdef WIN32
|
2007-10-15 03:15:58 +02:00
|
|
|
// windows XP has a limit on the number of
|
|
|
|
// simultaneous half-open TCP connections
|
|
|
|
DWORD windows_version = ::GetVersion();
|
|
|
|
if ((windows_version & 0xff) >= 6)
|
|
|
|
{
|
|
|
|
// on vista the limit is 5 (in home edition)
|
|
|
|
m_half_open.limit(4);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// on XP SP2 it's 10
|
|
|
|
m_half_open.limit(8);
|
|
|
|
}
|
2007-09-10 03:57:40 +02:00
|
|
|
#endif
|
|
|
|
|
2007-07-03 01:48:06 +02:00
|
|
|
m_bandwidth_manager[peer_connection::download_channel] = &m_download_channel;
|
|
|
|
m_bandwidth_manager[peer_connection::upload_channel] = &m_upload_channel;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2008-02-17 23:51:03 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2006-11-15 22:39:58 +01:00
|
|
|
m_logger = create_log("main_session", listen_port(), false);
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << "\n";
|
2007-05-14 00:01:21 +02:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef TORRENT_STATS
|
2007-09-29 18:14:03 +02:00
|
|
|
m_stats_logger.open("session_stats.log", std::ios::trunc);
|
2007-05-14 00:01:21 +02:00
|
|
|
m_stats_logger <<
|
2006-11-21 16:51:28 +01:00
|
|
|
"1. second\n"
|
2007-05-14 00:01:21 +02:00
|
|
|
"2. upload rate\n"
|
|
|
|
"3. download rate\n"
|
|
|
|
"4. downloading torrents\n"
|
|
|
|
"5. seeding torrents\n"
|
|
|
|
"6. peers\n"
|
2007-05-14 06:01:30 +02:00
|
|
|
"7. connecting peers\n"
|
2007-09-29 18:14:03 +02:00
|
|
|
"8. disk block buffers\n"
|
2006-11-21 16:51:28 +01:00
|
|
|
"\n";
|
2007-09-29 18:14:03 +02:00
|
|
|
m_buffer_usage_logger.open("buffer_stats.log", std::ios::trunc);
|
2006-11-21 16:51:28 +01:00
|
|
|
m_second_counter = 0;
|
2008-02-12 09:34:57 +01:00
|
|
|
m_buffer_allocations = 0;
|
2006-10-11 22:57:54 +02:00
|
|
|
#endif
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
// ---- generate a peer id ----
|
2006-11-15 22:39:58 +01:00
|
|
|
static seed_random_generator seeder;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
m_key = rand() + (rand() << 15) + (rand() << 30);
|
|
|
|
std::string print = cl_fprint.to_string();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(print.length() <= 20);
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// the client's fingerprint
|
|
|
|
std::copy(
|
|
|
|
print.begin()
|
|
|
|
, print.begin() + print.length()
|
|
|
|
, m_peer_id.begin());
|
|
|
|
|
|
|
|
// http-accepted characters:
|
|
|
|
static char const printable[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
"abcdefghijklmnopqrstuvwxyz-_.!~*'()";
|
|
|
|
|
|
|
|
// the random number
|
|
|
|
for (unsigned char* i = m_peer_id.begin() + print.length();
|
|
|
|
i != m_peer_id.end(); ++i)
|
|
|
|
{
|
|
|
|
*i = printable[rand() % (sizeof(printable)-1)];
|
|
|
|
}
|
|
|
|
|
2008-04-07 03:22:26 +02:00
|
|
|
asio::error_code ec;
|
|
|
|
m_timer.expires_from_now(seconds(1), ec);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_timer.async_wait(
|
|
|
|
bind(&session_impl::second_tick, this, _1));
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
m_thread.reset(new boost::thread(boost::ref(*this)));
|
|
|
|
}
|
|
|
|
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
struct free_ptr
|
|
|
|
{
|
|
|
|
void* ptr_;
|
|
|
|
free_ptr(void* p): ptr_(p) {}
|
|
|
|
~free_ptr() { free(ptr_); }
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2008-04-11 10:46:43 +02:00
|
|
|
char const* session_impl::country_for_ip(address const& a)
|
|
|
|
{
|
|
|
|
if (!a.is_v4() || m_country_db == 0) return 0;
|
|
|
|
return GeoIP_country_code_by_ipnum(m_country_db, a.to_v4().to_ulong());
|
|
|
|
}
|
|
|
|
|
2008-04-05 06:53:22 +02:00
|
|
|
int session_impl::as_for_ip(address const& a)
|
|
|
|
{
|
2008-04-11 10:46:43 +02:00
|
|
|
if (!a.is_v4() || m_asnum_db == 0) return 0;
|
|
|
|
char* name = GeoIP_name_by_ipnum(m_asnum_db, a.to_v4().to_ulong());
|
2008-04-05 06:53:22 +02:00
|
|
|
if (name == 0) return 0;
|
|
|
|
free_ptr p(name);
|
|
|
|
// GeoIP returns the name as AS??? where ? is the AS-number
|
|
|
|
return atoi(name + 2);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string session_impl::as_name_for_ip(address const& a)
|
|
|
|
{
|
2008-04-11 10:46:43 +02:00
|
|
|
if (!a.is_v4() || m_asnum_db == 0) return std::string();
|
|
|
|
char* name = GeoIP_name_by_ipnum(m_asnum_db, a.to_v4().to_ulong());
|
2008-04-05 06:53:22 +02:00
|
|
|
if (name == 0) return std::string();
|
|
|
|
free_ptr p(name);
|
|
|
|
char* tmp = std::strchr(name, ' ');
|
|
|
|
if (tmp == 0) return std::string();
|
|
|
|
return tmp + 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::pair<const int, int>* session_impl::lookup_as(int as)
|
|
|
|
{
|
|
|
|
std::map<int, int>::iterator i = m_as_peak.lower_bound(as);
|
|
|
|
|
|
|
|
if (i == m_as_peak.end() || i->first != as)
|
|
|
|
{
|
|
|
|
// we don't have any data for this AS, insert a new entry
|
|
|
|
i = m_as_peak.insert(i, std::pair<int, int>(as, 0));
|
|
|
|
}
|
|
|
|
return &(*i);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool session_impl::load_asnum_db(char const* file)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2008-04-11 10:46:43 +02:00
|
|
|
if (m_asnum_db) GeoIP_delete(m_asnum_db);
|
|
|
|
m_asnum_db = GeoIP_open(file, GEOIP_STANDARD);
|
|
|
|
return m_asnum_db;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool session_impl::load_country_db(char const* file)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
if (m_country_db) GeoIP_delete(m_country_db);
|
|
|
|
m_country_db = GeoIP_open(file, GEOIP_STANDARD);
|
|
|
|
return m_country_db;
|
2008-04-05 06:53:22 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
void session_impl::load_state(entry const& ses_state)
|
|
|
|
{
|
|
|
|
if (ses_state.type() != entry::dictionary_t) return;
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
entry const* as_map = ses_state.find_key("AS map");
|
|
|
|
if (as_map && as_map->type() == entry::dictionary_t)
|
|
|
|
{
|
|
|
|
entry::dictionary_type const& as_peak = as_map->dict();
|
|
|
|
for (entry::dictionary_type::const_iterator i = as_peak.begin()
|
|
|
|
, end(as_peak.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
int as_num = atoi(i->first.c_str());
|
|
|
|
if (i->second.type() != entry::int_t || i->second.integer() == 0) continue;
|
|
|
|
int& peak = m_as_peak[as_num];
|
|
|
|
if (peak < i->second.integer()) peak = i->second.integer();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
entry session_impl::state() const
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
entry ret;
|
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
|
|
|
entry::dictionary_type& as_map = ret["AS map"].dict();
|
|
|
|
char buf[10];
|
|
|
|
for (std::map<int, int>::const_iterator i = m_as_peak.begin()
|
|
|
|
, end(m_as_peak.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if (i->second == 0) continue;
|
|
|
|
sprintf(buf, "%05d", i->first);
|
|
|
|
as_map[buf] = i->second;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
|
|
void session_impl::add_extension(
|
2007-09-14 02:11:33 +02:00
|
|
|
boost::function<boost::shared_ptr<torrent_plugin>(torrent*, void*)> ext)
|
2006-11-14 01:08:16 +01:00
|
|
|
{
|
|
|
|
m_extensions.push_back(ext);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2007-04-18 01:06:00 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2006-10-11 22:57:54 +02:00
|
|
|
void session_impl::add_dht_node(udp::endpoint n)
|
|
|
|
{
|
|
|
|
if (m_dht) m_dht->add_node(n);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
void session_impl::abort()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-10-07 20:06:56 +02:00
|
|
|
if (m_abort) return;
|
|
|
|
#if defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " *** ABORT CALLED ***\n";
|
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
// abort the main thread
|
|
|
|
m_abort = true;
|
2007-10-07 20:06:56 +02:00
|
|
|
if (m_lsd) m_lsd->close();
|
|
|
|
if (m_upnp) m_upnp->close();
|
|
|
|
if (m_natpmp) m_natpmp->close();
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
if (m_dht) m_dht->stop();
|
2007-12-09 05:15:24 +01:00
|
|
|
m_dht_socket.close();
|
2007-10-07 20:06:56 +02:00
|
|
|
#endif
|
2008-04-07 03:22:26 +02:00
|
|
|
asio::error_code ec;
|
|
|
|
m_timer.cancel(ec);
|
2007-10-26 09:14:19 +02:00
|
|
|
|
|
|
|
// close the listen sockets
|
|
|
|
for (std::list<listen_socket_t>::iterator i = m_listen_sockets.begin()
|
|
|
|
, end(m_listen_sockets.end()); i != end; ++i)
|
|
|
|
{
|
2008-04-07 03:22:26 +02:00
|
|
|
i->sock->close(ec);
|
2007-10-26 09:14:19 +02:00
|
|
|
}
|
|
|
|
|
2007-10-29 01:29:43 +01:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2008-01-27 23:39:50 +01:00
|
|
|
(*m_logger) << time_now_string() << " aborting all torrents (" << m_torrents.size() << ")\n";
|
2007-10-29 01:29:43 +01:00
|
|
|
#endif
|
2007-10-07 20:06:56 +02:00
|
|
|
// abort all torrents
|
|
|
|
for (torrent_map::iterator i = m_torrents.begin()
|
|
|
|
, end(m_torrents.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
i->second->abort();
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-10-29 01:29:43 +01:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-11-02 01:27:53 +01:00
|
|
|
(*m_logger) << time_now_string() << " aborting all tracker requests\n";
|
2007-10-29 01:29:43 +01:00
|
|
|
#endif
|
2007-11-02 01:27:53 +01:00
|
|
|
m_tracker_manager.abort_all_requests();
|
|
|
|
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " sending event=stopped to trackers\n";
|
|
|
|
int counter = 0;
|
|
|
|
#endif
|
|
|
|
for (torrent_map::iterator i = m_torrents.begin();
|
|
|
|
i != m_torrents.end(); ++i)
|
2007-10-29 01:29:43 +01:00
|
|
|
{
|
2007-11-02 01:27:53 +01:00
|
|
|
torrent& t = *i->second;
|
|
|
|
|
|
|
|
if ((!t.is_paused() || t.should_request())
|
|
|
|
&& !t.trackers().empty())
|
|
|
|
{
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
++counter;
|
|
|
|
#endif
|
|
|
|
tracker_request req = t.generate_tracker_request();
|
|
|
|
TORRENT_ASSERT(req.event == tracker_request::stopped);
|
|
|
|
req.listen_port = 0;
|
|
|
|
if (!m_listen_sockets.empty())
|
|
|
|
req.listen_port = m_listen_sockets.front().external_port;
|
|
|
|
req.key = m_key;
|
|
|
|
std::string login = i->second->tracker_login();
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
boost::shared_ptr<tracker_logger> tl(new tracker_logger(*this));
|
|
|
|
m_tracker_loggers.push_back(tl);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_tracker_manager.queue_request(m_io_service, m_half_open, req, login
|
2007-11-02 01:27:53 +01:00
|
|
|
, m_listen_interface.address(), tl);
|
|
|
|
#else
|
2008-01-08 06:47:43 +01:00
|
|
|
m_tracker_manager.queue_request(m_io_service, m_half_open, req, login
|
2007-11-02 01:27:53 +01:00
|
|
|
, m_listen_interface.address());
|
|
|
|
#endif
|
|
|
|
}
|
2007-10-29 01:29:43 +01:00
|
|
|
}
|
2007-10-26 09:14:19 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-11-02 01:27:53 +01:00
|
|
|
(*m_logger) << time_now_string() << " sent " << counter << " tracker stop requests\n";
|
2007-10-26 09:14:19 +02:00
|
|
|
#endif
|
2007-11-02 01:27:53 +01:00
|
|
|
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " aborting all connections (" << m_connections.size() << ")\n";
|
|
|
|
#endif
|
|
|
|
// abort all connections
|
|
|
|
while (!m_connections.empty())
|
|
|
|
{
|
|
|
|
#ifndef NDEBUG
|
|
|
|
int conn = m_connections.size();
|
|
|
|
#endif
|
2008-01-07 02:10:46 +01:00
|
|
|
(*m_connections.begin())->disconnect("stopping torrent");
|
2007-11-07 02:38:17 +01:00
|
|
|
TORRENT_ASSERT(conn == int(m_connections.size()) + 1);
|
2007-11-02 01:27:53 +01:00
|
|
|
}
|
2007-10-26 09:14:19 +02:00
|
|
|
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " shutting down connection queue\n";
|
|
|
|
#endif
|
|
|
|
m_half_open.close();
|
2007-10-08 01:20:38 +02:00
|
|
|
|
2007-11-08 08:52:41 +01:00
|
|
|
m_download_channel.close();
|
|
|
|
m_upload_channel.close();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-06-01 03:05:57 +02:00
|
|
|
void session_impl::set_port_filter(port_filter const& f)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
m_port_filter = f;
|
|
|
|
}
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
void session_impl::set_ip_filter(ip_filter const& f)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
m_ip_filter = f;
|
|
|
|
|
|
|
|
// Close connections whose endpoint is filtered
|
|
|
|
// by the new ip-filter
|
2007-07-26 09:04:35 +02:00
|
|
|
for (torrent_map::iterator i = m_torrents.begin()
|
|
|
|
, end(m_torrents.end()); i != end; ++i)
|
|
|
|
i->second->ip_filter_updated();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_settings(session_settings const& s)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(s.file_pool_size > 0);
|
2007-06-08 00:37:58 +02:00
|
|
|
|
|
|
|
// less than 5 seconds unchoke interval is insane
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(s.unchoke_interval >= 5);
|
2008-02-08 11:22:05 +01:00
|
|
|
if (m_settings.cache_size != s.cache_size)
|
|
|
|
m_disk_thread.set_cache_size(s.cache_size);
|
2008-02-10 01:58:25 +01:00
|
|
|
if (m_settings.cache_expiry != s.cache_expiry)
|
|
|
|
m_disk_thread.set_cache_size(s.cache_expiry);
|
2006-10-11 22:57:54 +02:00
|
|
|
m_settings = s;
|
2008-02-11 04:55:53 +01:00
|
|
|
if (m_settings.connection_speed <= 0) m_settings.connection_speed = 200;
|
|
|
|
|
2006-11-14 16:53:38 +01:00
|
|
|
m_files.resize(m_settings.file_pool_size);
|
2008-01-13 12:18:18 +01:00
|
|
|
if (!s.auto_upload_slots) m_allowed_upload_slots = m_max_uploads;
|
2006-10-11 22:57:54 +02:00
|
|
|
// replace all occurances of '\n' with ' '.
|
|
|
|
std::string::iterator i = m_settings.user_agent.begin();
|
|
|
|
while ((i = std::find(i, m_settings.user_agent.end(), '\n'))
|
|
|
|
!= m_settings.user_agent.end())
|
|
|
|
*i = ' ';
|
|
|
|
}
|
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
tcp::endpoint session_impl::get_ipv6_interface() const
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
return m_ipv6_interface;
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-11-23 23:14:33 +01:00
|
|
|
session_impl::listen_socket_t session_impl::setup_listener(tcp::endpoint ep
|
|
|
|
, int retries, bool v6_only)
|
2007-09-22 18:27:29 +02:00
|
|
|
{
|
|
|
|
asio::error_code ec;
|
|
|
|
listen_socket_t s;
|
|
|
|
s.sock.reset(new socket_acceptor(m_io_service));
|
|
|
|
s.sock->open(ep.protocol(), ec);
|
|
|
|
s.sock->set_option(socket_acceptor::reuse_address(true), ec);
|
2007-11-23 23:14:33 +01:00
|
|
|
if (ep.protocol() == tcp::v6()) s.sock->set_option(v6only(v6_only), ec);
|
2007-09-22 18:27:29 +02:00
|
|
|
s.sock->bind(ep, ec);
|
|
|
|
while (ec && retries > 0)
|
|
|
|
{
|
|
|
|
ec = asio::error_code();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!ec);
|
2007-09-22 18:27:29 +02:00
|
|
|
--retries;
|
|
|
|
ep.port(ep.port() + 1);
|
|
|
|
s.sock->bind(ep, ec);
|
|
|
|
}
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
// instead of giving up, try
|
|
|
|
// let the OS pick a port
|
|
|
|
ep.port(0);
|
|
|
|
ec = asio::error_code();
|
|
|
|
s.sock->bind(ep, ec);
|
|
|
|
}
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
// not even that worked, give up
|
|
|
|
if (m_alerts.should_post(alert::fatal))
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-09-24 00:50:25 +02:00
|
|
|
std::stringstream msg;
|
|
|
|
msg << "cannot bind to interface '";
|
|
|
|
print_endpoint(msg, ep) << "' " << ec.message();
|
|
|
|
m_alerts.post_alert(listen_failed_alert(ep, msg.str()));
|
2007-09-22 18:27:29 +02:00
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-09-24 00:50:25 +02:00
|
|
|
std::stringstream msg;
|
|
|
|
msg << "cannot bind to interface '";
|
|
|
|
print_endpoint(msg, ep) << "' " << ec.message();
|
|
|
|
(*m_logger) << msg.str() << "\n";
|
2006-10-11 22:57:54 +02:00
|
|
|
#endif
|
2007-09-22 18:27:29 +02:00
|
|
|
return listen_socket_t();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
2007-09-22 18:27:29 +02:00
|
|
|
s.external_port = s.sock->local_endpoint(ec).port();
|
|
|
|
s.sock->listen(0, ec);
|
|
|
|
if (ec)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
if (m_alerts.should_post(alert::fatal))
|
|
|
|
{
|
2007-09-24 00:50:25 +02:00
|
|
|
std::stringstream msg;
|
|
|
|
msg << "cannot listen on interface '";
|
|
|
|
print_endpoint(msg, ep) << "' " << ec.message();
|
|
|
|
m_alerts.post_alert(listen_failed_alert(ep, msg.str()));
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
2007-09-22 18:27:29 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-09-24 00:50:25 +02:00
|
|
|
std::stringstream msg;
|
|
|
|
msg << "cannot listen on interface '";
|
|
|
|
print_endpoint(msg, ep) << "' " << ec.message();
|
|
|
|
(*m_logger) << msg.str() << "\n";
|
2007-09-22 18:27:29 +02:00
|
|
|
#endif
|
|
|
|
return listen_socket_t();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
if (m_alerts.should_post(alert::fatal))
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
std::string msg = "listening on interface "
|
|
|
|
+ boost::lexical_cast<std::string>(ep);
|
|
|
|
m_alerts.post_alert(listen_succeeded_alert(ep, msg));
|
|
|
|
}
|
|
|
|
|
2007-09-10 21:10:38 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-09-24 00:50:25 +02:00
|
|
|
(*m_logger) << "listening on: " << ep
|
2007-09-22 18:27:29 +02:00
|
|
|
<< " external port: " << s.external_port << "\n";
|
2006-10-11 22:57:54 +02:00
|
|
|
#endif
|
2007-09-22 18:27:29 +02:00
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2008-01-07 02:10:46 +01:00
|
|
|
void session_impl::open_listen_port()
|
2007-09-22 18:27:29 +02:00
|
|
|
{
|
|
|
|
// close the open listen sockets
|
|
|
|
m_listen_sockets.clear();
|
|
|
|
m_incoming_connection = false;
|
|
|
|
|
|
|
|
if (is_any(m_listen_interface.address()))
|
|
|
|
{
|
|
|
|
// this means we should open two listen sockets
|
|
|
|
// one for IPv4 and one for IPv6
|
|
|
|
|
|
|
|
listen_socket_t s = setup_listener(
|
|
|
|
tcp::endpoint(address_v4::any(), m_listen_interface.port())
|
|
|
|
, m_listen_port_retries);
|
|
|
|
|
|
|
|
if (s.sock)
|
|
|
|
{
|
|
|
|
m_listen_sockets.push_back(s);
|
|
|
|
async_accept(s.sock);
|
|
|
|
}
|
|
|
|
|
|
|
|
s = setup_listener(
|
|
|
|
tcp::endpoint(address_v6::any(), m_listen_interface.port())
|
2007-11-23 23:14:33 +01:00
|
|
|
, m_listen_port_retries, true);
|
2007-09-22 18:27:29 +02:00
|
|
|
|
|
|
|
if (s.sock)
|
|
|
|
{
|
|
|
|
m_listen_sockets.push_back(s);
|
|
|
|
async_accept(s.sock);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// we should only open a single listen socket, that
|
|
|
|
// binds to the given interface
|
|
|
|
|
|
|
|
listen_socket_t s = setup_listener(
|
|
|
|
m_listen_interface, m_listen_port_retries);
|
|
|
|
|
|
|
|
if (s.sock)
|
|
|
|
{
|
|
|
|
m_listen_sockets.push_back(s);
|
|
|
|
async_accept(s.sock);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m_ipv6_interface = tcp::endpoint();
|
|
|
|
|
|
|
|
for (std::list<listen_socket_t>::const_iterator i = m_listen_sockets.begin()
|
|
|
|
, end(m_listen_sockets.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
asio::error_code ec;
|
|
|
|
tcp::endpoint ep = i->sock->local_endpoint(ec);
|
|
|
|
if (ec || ep.address().is_v4()) continue;
|
|
|
|
|
2007-09-22 19:07:55 +02:00
|
|
|
if (ep.address().to_v6() != address_v6::any())
|
2007-09-22 18:27:29 +02:00
|
|
|
{
|
2007-09-22 19:07:55 +02:00
|
|
|
// if we're listening on a specific address
|
|
|
|
// pick it
|
|
|
|
m_ipv6_interface = ep;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// if we're listening on any IPv6 address, enumerate them and
|
|
|
|
// pick the first non-local address
|
2008-01-11 07:49:37 +01:00
|
|
|
std::vector<ip_interface> const& ifs = enum_net_interfaces(m_io_service, ec);
|
|
|
|
for (std::vector<ip_interface>::const_iterator i = ifs.begin()
|
2007-09-22 19:07:55 +02:00
|
|
|
, end(ifs.end()); i != end; ++i)
|
|
|
|
{
|
2008-01-11 07:49:37 +01:00
|
|
|
if (i->interface_address.is_v4()
|
|
|
|
|| i->interface_address.to_v6().is_link_local()
|
|
|
|
|| i->interface_address.to_v6().is_loopback()) continue;
|
|
|
|
m_ipv6_interface = tcp::endpoint(i->interface_address, ep.port());
|
2007-09-22 19:07:55 +02:00
|
|
|
break;
|
|
|
|
}
|
2007-09-22 18:27:29 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!m_listen_sockets.empty())
|
|
|
|
{
|
|
|
|
asio::error_code ec;
|
|
|
|
tcp::endpoint local = m_listen_sockets.front().sock->local_endpoint(ec);
|
|
|
|
if (!ec)
|
|
|
|
{
|
2008-04-06 21:17:58 +02:00
|
|
|
if (m_natpmp.get())
|
|
|
|
{
|
|
|
|
if (m_tcp_mapping[0] != -1) m_natpmp->delete_mapping(m_tcp_mapping[0]);
|
|
|
|
m_tcp_mapping[0] = m_natpmp->add_mapping(natpmp::tcp
|
|
|
|
, local.port(), local.port());
|
|
|
|
}
|
|
|
|
if (m_upnp.get())
|
|
|
|
{
|
|
|
|
if (m_tcp_mapping[1] != -1) m_upnp->delete_mapping(m_tcp_mapping[1]);
|
|
|
|
m_tcp_mapping[1] = m_upnp->add_mapping(upnp::tcp
|
|
|
|
, local.port(), local.port());
|
|
|
|
}
|
2007-09-22 18:27:29 +02:00
|
|
|
}
|
2007-09-10 21:10:38 +02:00
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2008-01-18 03:02:18 +01:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
|
2008-03-25 05:38:59 +01:00
|
|
|
void session_impl::on_receive_udp(asio::error_code const& e
|
|
|
|
, udp::endpoint const& ep, char const* buf, int len)
|
2007-12-09 05:15:24 +01:00
|
|
|
{
|
2008-03-25 05:38:59 +01:00
|
|
|
if (e)
|
|
|
|
{
|
|
|
|
if (m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
std::string msg = "UDP socket error from '"
|
|
|
|
+ boost::lexical_cast<std::string>(ep) + "' " + e.message();
|
|
|
|
m_alerts.post_alert(udp_error_alert(ep, msg));
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-12-09 05:15:24 +01:00
|
|
|
if (len > 20 && *buf == 'd' && m_dht)
|
|
|
|
{
|
|
|
|
// this is probably a dht message
|
|
|
|
m_dht->on_receive(ep, buf, len);
|
|
|
|
}
|
|
|
|
}
|
2008-01-18 03:02:18 +01:00
|
|
|
|
|
|
|
#endif
|
2007-12-09 05:15:24 +01:00
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
void session_impl::async_accept(boost::shared_ptr<socket_acceptor> const& listener)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2008-01-27 23:39:50 +01:00
|
|
|
shared_ptr<socket_type> c(new socket_type(m_io_service));
|
2007-10-22 06:17:26 +02:00
|
|
|
c->instantiate<stream_socket>(m_io_service);
|
2007-09-22 18:27:29 +02:00
|
|
|
listener->async_accept(c->get<stream_socket>()
|
2006-10-11 22:57:54 +02:00
|
|
|
, bind(&session_impl::on_incoming_connection, this, c
|
2007-09-22 18:27:29 +02:00
|
|
|
, boost::weak_ptr<socket_acceptor>(listener), _1));
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-04-25 20:26:35 +02:00
|
|
|
void session_impl::on_incoming_connection(shared_ptr<socket_type> const& s
|
2008-04-07 03:22:26 +02:00
|
|
|
, weak_ptr<socket_acceptor> listen_socket, asio::error_code const& e)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
boost::shared_ptr<socket_acceptor> listener = listen_socket.lock();
|
|
|
|
if (!listener) return;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
if (e == asio::error::operation_aborted) return;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
if (m_abort) return;
|
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
asio::error_code ec;
|
2006-10-11 22:57:54 +02:00
|
|
|
if (e)
|
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
tcp::endpoint ep = listener->local_endpoint(ec);
|
2006-10-11 22:57:54 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
std::string msg = "error accepting connection on '"
|
2007-09-22 18:27:29 +02:00
|
|
|
+ boost::lexical_cast<std::string>(ep) + "' " + e.message();
|
2006-10-11 22:57:54 +02:00
|
|
|
(*m_logger) << msg << "\n";
|
2007-10-15 21:02:54 +02:00
|
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
|
|
// Windows sometimes generates this error. It seems to be
|
|
|
|
// non-fatal and we have to do another async_accept.
|
|
|
|
if (e.value() == ERROR_SEM_TIMEOUT)
|
|
|
|
{
|
|
|
|
async_accept(listener);
|
|
|
|
return;
|
|
|
|
}
|
2008-04-03 06:35:56 +02:00
|
|
|
#endif
|
|
|
|
#ifdef TORRENT_BSD
|
|
|
|
// Leopard sometimes generates an "invalid argument" error. It seems to be
|
|
|
|
// non-fatal and we have to do another async_accept.
|
|
|
|
if (e.value() == EINVAL)
|
|
|
|
{
|
|
|
|
async_accept(listener);
|
|
|
|
return;
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
#endif
|
2007-09-22 18:27:29 +02:00
|
|
|
if (m_alerts.should_post(alert::fatal))
|
|
|
|
{
|
|
|
|
std::string msg = "error accepting connection on '"
|
2008-03-16 13:33:39 +01:00
|
|
|
+ boost::lexical_cast<std::string>(ep) + "' " + e.message();
|
2007-09-22 18:27:29 +02:00
|
|
|
m_alerts.post_alert(listen_failed_alert(ep, msg));
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
return;
|
|
|
|
}
|
2007-09-22 18:27:29 +02:00
|
|
|
async_accept(listener);
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// we got a connection request!
|
2007-09-22 18:27:29 +02:00
|
|
|
tcp::endpoint endp = s->remote_endpoint(ec);
|
|
|
|
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << endp << " <== INCOMING CONNECTION FAILED, could "
|
|
|
|
"not retrieve remote endpoint " << ec.message() << "\n";
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-12-19 22:36:54 +01:00
|
|
|
(*m_logger) << time_now_string() << " <== INCOMING CONNECTION " << endp << "\n";
|
2006-10-11 22:57:54 +02:00
|
|
|
#endif
|
2007-11-19 19:05:47 +01:00
|
|
|
|
|
|
|
// local addresses do not count, since it's likely
|
|
|
|
// coming from our own client through local service discovery
|
|
|
|
// and it does not reflect whether or not a router is open
|
|
|
|
// for incoming connections or not.
|
|
|
|
if (!is_local(endp.address()))
|
|
|
|
m_incoming_connection = true;
|
|
|
|
|
2006-11-19 16:29:58 +01:00
|
|
|
if (m_ip_filter.access(endp.address()) & ip_filter::blocked)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << "filtered blocked ip\n";
|
|
|
|
#endif
|
2007-04-17 07:56:43 +02:00
|
|
|
if (m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_alerts.post_alert(peer_blocked_alert(endp.address()
|
|
|
|
, "incoming connection blocked by IP filter"));
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-10-27 23:21:44 +02:00
|
|
|
// don't allow more connections than the max setting
|
2008-01-18 02:29:28 +01:00
|
|
|
if (num_connections() >= max_connections())
|
2007-10-27 23:21:44 +02:00
|
|
|
{
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << "number of connections limit exceeded (conns: "
|
|
|
|
<< num_connections() << ", limit: " << max_connections()
|
|
|
|
<< "), connection rejected\n";
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-09-04 00:27:52 +02:00
|
|
|
// check if we have any active torrents
|
|
|
|
// if we don't reject the connection
|
2007-09-22 18:27:29 +02:00
|
|
|
if (m_torrents.empty()) return;
|
|
|
|
|
|
|
|
bool has_active_torrent = false;
|
|
|
|
for (torrent_map::iterator i = m_torrents.begin()
|
|
|
|
, end(m_torrents.end()); i != end; ++i)
|
2007-09-04 00:27:52 +02:00
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
if (!i->second->is_paused())
|
2007-09-04 00:27:52 +02:00
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
has_active_torrent = true;
|
|
|
|
break;
|
2007-09-04 00:27:52 +02:00
|
|
|
}
|
|
|
|
}
|
2007-09-22 18:27:29 +02:00
|
|
|
if (!has_active_torrent) return;
|
2007-09-04 00:27:52 +02:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
boost::intrusive_ptr<peer_connection> c(
|
2007-09-22 18:27:29 +02:00
|
|
|
new bt_peer_connection(*this, s, 0));
|
2006-10-11 22:57:54 +02:00
|
|
|
#ifndef NDEBUG
|
|
|
|
c->m_in_constructor = false;
|
|
|
|
#endif
|
|
|
|
|
2008-03-31 06:46:24 +02:00
|
|
|
if (!c->is_disconnecting())
|
|
|
|
{
|
|
|
|
m_connections.insert(c);
|
|
|
|
c->start();
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
2008-01-07 02:10:46 +01:00
|
|
|
void session_impl::close_connection(peer_connection const* p
|
|
|
|
, char const* message)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
|
2007-08-21 21:18:06 +02:00
|
|
|
// too expensive
|
|
|
|
// INVARIANT_CHECK;
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2008-01-07 02:10:46 +01:00
|
|
|
#ifndef NDEBUG
|
|
|
|
// for (aux::session_impl::torrent_map::const_iterator i = m_torrents.begin()
|
|
|
|
// , end(m_torrents.end()); i != end; ++i)
|
|
|
|
// TORRENT_ASSERT(!i->second->has_peer((peer_connection*)p));
|
|
|
|
#endif
|
|
|
|
|
2007-12-19 22:36:54 +01:00
|
|
|
#if defined(TORRENT_LOGGING)
|
2008-01-07 02:10:46 +01:00
|
|
|
(*m_logger) << time_now_string() << " CLOSING CONNECTION "
|
|
|
|
<< p->remote() << " : " << message << "\n";
|
2007-12-19 22:36:54 +01:00
|
|
|
#endif
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->is_disconnecting());
|
2008-01-07 02:10:46 +01:00
|
|
|
|
|
|
|
if (!p->is_choked()) --m_num_unchoked;
|
|
|
|
// connection_map::iterator i = std::lower_bound(m_connections.begin(), m_connections.end()
|
|
|
|
// , p, bind(&boost::intrusive_ptr<peer_connection>::get, _1) < p);
|
|
|
|
// if (i->get() != p) i == m_connections.end();
|
|
|
|
connection_map::iterator i = std::find_if(m_connections.begin(), m_connections.end()
|
|
|
|
, bind(&boost::intrusive_ptr<peer_connection>::get, _1) == p);
|
|
|
|
if (i != m_connections.end()) m_connections.erase(i);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_peer_id(peer_id const& id)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
m_peer_id = id;
|
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_key(int key)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
m_key = key;
|
|
|
|
}
|
|
|
|
|
2008-02-28 08:34:07 +01:00
|
|
|
int session_impl::next_port()
|
|
|
|
{
|
|
|
|
std::pair<int, int> const& out_ports = m_settings.outgoing_ports;
|
|
|
|
if (m_next_port < out_ports.first || m_next_port > out_ports.second)
|
|
|
|
m_next_port = out_ports.first;
|
|
|
|
|
|
|
|
int port = m_next_port;
|
|
|
|
++m_next_port;
|
|
|
|
if (m_next_port > out_ports.second) m_next_port = out_ports.first;
|
|
|
|
#if defined TORRENT_LOGGING
|
|
|
|
(*m_logger) << time_now_string() << " *** BINDING OUTGOING CONNECTION [ "
|
|
|
|
"port: " << port << " ]\n";
|
|
|
|
#endif
|
|
|
|
return port;
|
|
|
|
}
|
|
|
|
|
2008-04-07 03:22:26 +02:00
|
|
|
void session_impl::second_tick(asio::error_code const& e)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_mutex);
|
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
// too expensive
|
|
|
|
// INVARIANT_CHECK;
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2007-10-07 20:06:56 +02:00
|
|
|
if (m_abort) return;
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
if (e)
|
|
|
|
{
|
2008-02-28 08:34:07 +01:00
|
|
|
#if defined TORRENT_LOGGING
|
2006-11-14 01:08:16 +01:00
|
|
|
(*m_logger) << "*** SECOND TIMER FAILED " << e.message() << "\n";
|
2006-10-11 22:57:54 +02:00
|
|
|
#endif
|
2008-02-28 08:34:07 +01:00
|
|
|
::abort();
|
2006-10-11 22:57:54 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
float tick_interval = total_microseconds(time_now() - m_last_tick) / 1000000.f;
|
|
|
|
m_last_tick = time_now();
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2008-04-07 03:22:26 +02:00
|
|
|
asio::error_code ec;
|
|
|
|
m_timer.expires_from_now(seconds(1), ec);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_timer.async_wait(
|
|
|
|
bind(&session_impl::second_tick, this, _1));
|
2007-05-14 00:01:21 +02:00
|
|
|
|
|
|
|
#ifdef TORRENT_STATS
|
|
|
|
++m_second_counter;
|
|
|
|
int downloading_torrents = 0;
|
|
|
|
int seeding_torrents = 0;
|
|
|
|
for (torrent_map::iterator i = m_torrents.begin()
|
|
|
|
, end(m_torrents.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
if (i->second->is_seed())
|
|
|
|
++seeding_torrents;
|
|
|
|
else
|
|
|
|
++downloading_torrents;
|
|
|
|
}
|
2007-09-29 18:14:03 +02:00
|
|
|
int num_complete_connections = 0;
|
2007-05-14 06:01:30 +02:00
|
|
|
int num_half_open = 0;
|
|
|
|
for (connection_map::iterator i = m_connections.begin()
|
|
|
|
, end(m_connections.end()); i != end; ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
if ((*i)->is_connecting())
|
2007-05-14 06:01:30 +02:00
|
|
|
++num_half_open;
|
|
|
|
else
|
2007-09-29 18:14:03 +02:00
|
|
|
++num_complete_connections;
|
2007-05-14 06:01:30 +02:00
|
|
|
}
|
|
|
|
|
2007-05-14 00:01:21 +02:00
|
|
|
m_stats_logger
|
|
|
|
<< m_second_counter << "\t"
|
|
|
|
<< m_stat.upload_rate() << "\t"
|
|
|
|
<< m_stat.download_rate() << "\t"
|
|
|
|
<< downloading_torrents << "\t"
|
|
|
|
<< seeding_torrents << "\t"
|
2007-09-29 18:14:03 +02:00
|
|
|
<< num_complete_connections << "\t"
|
2007-05-14 06:01:30 +02:00
|
|
|
<< num_half_open << "\t"
|
2007-09-29 18:14:03 +02:00
|
|
|
<< m_disk_thread.disk_allocations() << "\t"
|
2007-05-14 06:01:30 +02:00
|
|
|
<< std::endl;
|
2007-05-14 00:01:21 +02:00
|
|
|
#endif
|
|
|
|
|
2008-03-29 20:39:24 +01:00
|
|
|
// --------------------------------------------------------------
|
|
|
|
// second_tick every torrent
|
|
|
|
// --------------------------------------------------------------
|
|
|
|
|
|
|
|
int congested_torrents = 0;
|
|
|
|
int uncongested_torrents = 0;
|
|
|
|
|
|
|
|
// count the number of seeding torrents vs. downloading
|
|
|
|
// torrents we are running
|
|
|
|
int num_seeds = 0;
|
|
|
|
int num_downloads = 0;
|
|
|
|
|
|
|
|
// count the number of peers of downloading torrents
|
|
|
|
int num_downloads_peers = 0;
|
|
|
|
|
|
|
|
// check each torrent for tracker updates
|
|
|
|
// TODO: do this in a timer-event in each torrent instead
|
|
|
|
for (torrent_map::iterator i = m_torrents.begin();
|
|
|
|
i != m_torrents.end();)
|
|
|
|
{
|
|
|
|
torrent& t = *i->second;
|
|
|
|
TORRENT_ASSERT(!t.is_aborted());
|
|
|
|
if (t.bandwidth_queue_size(peer_connection::upload_channel))
|
|
|
|
++congested_torrents;
|
|
|
|
else
|
|
|
|
++uncongested_torrents;
|
|
|
|
|
|
|
|
if (t.is_finished())
|
|
|
|
{
|
|
|
|
++num_seeds;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
++num_downloads;
|
|
|
|
num_downloads_peers += t.num_peers();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (t.should_request())
|
|
|
|
{
|
|
|
|
tracker_request req = t.generate_tracker_request();
|
|
|
|
req.listen_port = 0;
|
|
|
|
if (!m_listen_sockets.empty())
|
|
|
|
req.listen_port = m_listen_sockets.front().external_port;
|
|
|
|
req.key = m_key;
|
|
|
|
m_tracker_manager.queue_request(m_io_service, m_half_open, req
|
|
|
|
, t.tracker_login(), m_listen_interface.address(), i->second);
|
|
|
|
|
|
|
|
if (m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_alerts.post_alert(
|
|
|
|
tracker_announce_alert(
|
|
|
|
t.get_handle(), "tracker announce"));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
t.second_tick(m_stat, tick_interval);
|
|
|
|
++i;
|
|
|
|
}
|
|
|
|
|
|
|
|
m_stat.second_tick(tick_interval);
|
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
|
2008-03-29 20:39:24 +01:00
|
|
|
// --------------------------------------------------------------
|
|
|
|
// connect new peers
|
|
|
|
// --------------------------------------------------------------
|
|
|
|
|
2007-05-05 02:29:33 +02:00
|
|
|
// let torrents connect to peers if they want to
|
|
|
|
// if there are any torrents and any free slots
|
2007-05-25 23:00:35 +02:00
|
|
|
|
|
|
|
// this loop will "hand out" max(connection_speed
|
|
|
|
// , half_open.free_slots()) to the torrents, in a
|
|
|
|
// round robin fashion, so that every torrent is
|
|
|
|
// equallt likely to connect to a peer
|
|
|
|
|
2008-03-16 11:49:47 +01:00
|
|
|
int free_slots = m_half_open.free_slots();
|
2007-08-16 14:41:46 +02:00
|
|
|
if (!m_torrents.empty()
|
2008-03-16 11:49:47 +01:00
|
|
|
&& free_slots > -m_half_open.limit()
|
2007-08-16 14:41:46 +02:00
|
|
|
&& num_connections() < m_max_connections)
|
2007-05-05 02:29:33 +02:00
|
|
|
{
|
2007-05-25 23:00:35 +02:00
|
|
|
// this is the maximum number of connections we will
|
|
|
|
// attempt this tick
|
|
|
|
int max_connections = m_settings.connection_speed;
|
2008-03-29 20:39:24 +01:00
|
|
|
int average_peers = 0;
|
|
|
|
if (num_downloads > 0)
|
|
|
|
average_peers = num_downloads_peers / num_downloads;
|
2007-05-25 23:00:35 +02:00
|
|
|
|
|
|
|
torrent_map::iterator i = m_torrents.begin();
|
2007-05-05 02:29:33 +02:00
|
|
|
if (m_next_connect_torrent < int(m_torrents.size()))
|
2007-05-25 23:00:35 +02:00
|
|
|
std::advance(i, m_next_connect_torrent);
|
2007-05-05 02:29:33 +02:00
|
|
|
else
|
|
|
|
m_next_connect_torrent = 0;
|
2007-05-25 23:00:35 +02:00
|
|
|
int steps_since_last_connect = 0;
|
|
|
|
int num_torrents = int(m_torrents.size());
|
|
|
|
for (;;)
|
2007-05-05 02:29:33 +02:00
|
|
|
{
|
|
|
|
torrent& t = *i->second;
|
|
|
|
if (t.want_more_peers())
|
2007-08-16 14:41:46 +02:00
|
|
|
{
|
2008-03-29 20:39:24 +01:00
|
|
|
int connect_points = 100;
|
|
|
|
// have a bias against torrents with more peers
|
|
|
|
// than average
|
|
|
|
if (!t.is_seed() && t.num_peers() > average_peers)
|
|
|
|
connect_points /= 2;
|
|
|
|
// if this is a seed and there is a torrent that
|
|
|
|
// is downloading, lower the rate at which this
|
|
|
|
// torrent gets connections.
|
|
|
|
// dividing by num_seeds will have the effect
|
|
|
|
// that all seed will get as many connections
|
|
|
|
// together, as a single downloading torrent.
|
|
|
|
if (t.is_seed() && num_downloads > 0)
|
|
|
|
connect_points /= num_seeds + 1;
|
|
|
|
if (connect_points <= 0) connect_points = 1;
|
|
|
|
t.give_connect_points(connect_points);
|
2008-02-05 06:51:05 +01:00
|
|
|
try
|
2007-05-25 23:00:35 +02:00
|
|
|
{
|
2008-02-05 06:51:05 +01:00
|
|
|
if (t.try_connect_peer())
|
|
|
|
{
|
|
|
|
--max_connections;
|
2008-03-16 11:49:47 +01:00
|
|
|
--free_slots;
|
2008-02-05 06:51:05 +01:00
|
|
|
steps_since_last_connect = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (std::bad_alloc&)
|
|
|
|
{
|
|
|
|
// we ran out of memory trying to connect to a peer
|
|
|
|
// lower the global limit to the number of peers
|
|
|
|
// we already have
|
|
|
|
m_max_connections = num_connections();
|
|
|
|
if (m_max_connections < 2) m_max_connections = 2;
|
2007-05-25 23:00:35 +02:00
|
|
|
}
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
2007-05-05 02:29:33 +02:00
|
|
|
++m_next_connect_torrent;
|
2007-05-25 23:00:35 +02:00
|
|
|
++steps_since_last_connect;
|
2007-05-05 02:29:33 +02:00
|
|
|
++i;
|
|
|
|
if (i == m_torrents.end())
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_next_connect_torrent == num_torrents);
|
2007-05-05 02:29:33 +02:00
|
|
|
i = m_torrents.begin();
|
|
|
|
m_next_connect_torrent = 0;
|
|
|
|
}
|
2008-03-29 20:39:24 +01:00
|
|
|
// if we have gone two whole loops without
|
2007-05-25 23:00:35 +02:00
|
|
|
// handing out a single connection, break
|
2008-03-29 20:39:24 +01:00
|
|
|
if (steps_since_last_connect > num_torrents * 2) break;
|
2007-05-25 23:00:35 +02:00
|
|
|
// if there are no more free connection slots, abort
|
2008-03-16 11:49:47 +01:00
|
|
|
if (free_slots <= -m_half_open.limit()) break;
|
2007-05-25 23:00:35 +02:00
|
|
|
// if we should not make any more connections
|
|
|
|
// attempts this tick, abort
|
|
|
|
if (max_connections == 0) break;
|
2007-08-27 18:45:45 +02:00
|
|
|
// maintain the global limit on number of connections
|
|
|
|
if (num_connections() >= m_max_connections) break;
|
2007-05-25 23:00:35 +02:00
|
|
|
}
|
2007-05-05 02:29:33 +02:00
|
|
|
}
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
// do the second_tick() on each connection
|
|
|
|
// this will update their statistics (download and upload speeds)
|
|
|
|
// also purge sockets that have timed out
|
|
|
|
// and keep sockets open by keeping them alive.
|
|
|
|
for (connection_map::iterator i = m_connections.begin();
|
|
|
|
i != m_connections.end();)
|
|
|
|
{
|
|
|
|
// we need to do like this because j->second->disconnect() will
|
|
|
|
// erase the connection from the map we're iterating
|
|
|
|
connection_map::iterator j = i;
|
|
|
|
++i;
|
|
|
|
// if this socket has timed out
|
|
|
|
// close it.
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection& c = *j->get();
|
2006-10-11 22:57:54 +02:00
|
|
|
if (c.has_timed_out())
|
|
|
|
{
|
|
|
|
if (m_alerts.should_post(alert::debug))
|
|
|
|
{
|
|
|
|
m_alerts.post_alert(
|
|
|
|
peer_error_alert(
|
|
|
|
c.remote()
|
|
|
|
, c.pid()
|
|
|
|
, "connection timed out"));
|
|
|
|
}
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING)
|
|
|
|
(*c.m_logger) << "*** CONNECTION TIMED OUT\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
c.set_failed();
|
2008-01-07 02:10:46 +01:00
|
|
|
c.disconnect("timed out");
|
2006-10-11 22:57:54 +02:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
c.keep_alive();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
// --------------------------------------------------------------
|
|
|
|
// unchoke set and optimistic unchoke calculations
|
|
|
|
// --------------------------------------------------------------
|
|
|
|
m_unchoke_time_scaler--;
|
|
|
|
if (m_unchoke_time_scaler <= 0 && !m_connections.empty())
|
|
|
|
{
|
|
|
|
m_unchoke_time_scaler = settings().unchoke_interval;
|
|
|
|
|
|
|
|
std::vector<peer_connection*> peers;
|
|
|
|
for (connection_map::iterator i = m_connections.begin()
|
|
|
|
, end(m_connections.end()); i != end; ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* p = i->get();
|
2007-08-16 14:41:46 +02:00
|
|
|
torrent* t = p->associated_torrent().lock().get();
|
|
|
|
if (!p->peer_info_struct()
|
2007-08-17 09:37:08 +02:00
|
|
|
|| t == 0
|
2007-08-16 14:41:46 +02:00
|
|
|
|| !p->is_peer_interested()
|
|
|
|
|| p->is_disconnecting()
|
|
|
|
|| p->is_connecting()
|
|
|
|
|| (p->share_diff() < -free_upload_amount
|
2007-08-17 09:37:08 +02:00
|
|
|
&& !t->is_seed()))
|
2007-08-16 14:41:46 +02:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
if (!(*i)->is_choked() && t)
|
2007-08-22 20:50:53 +02:00
|
|
|
{
|
|
|
|
policy::peer* pi = p->peer_info_struct();
|
|
|
|
if (pi && pi->optimistically_unchoked)
|
|
|
|
{
|
|
|
|
pi->optimistically_unchoked = false;
|
|
|
|
// force a new optimistic unchoke
|
|
|
|
m_optimistic_unchoke_time_scaler = 0;
|
|
|
|
}
|
2007-10-31 10:48:20 +01:00
|
|
|
t->choke_peer(*(*i));
|
2007-08-22 20:50:53 +02:00
|
|
|
}
|
2007-08-16 14:41:46 +02:00
|
|
|
continue;
|
|
|
|
}
|
2007-10-31 10:48:20 +01:00
|
|
|
peers.push_back(i->get());
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
|
2008-03-10 04:30:01 +01:00
|
|
|
// sorts the peers that are eligible for unchoke by download rate and secondary
|
2007-08-16 14:41:46 +02:00
|
|
|
// by total upload. The reason for this is, if all torrents are being seeded,
|
|
|
|
// the download rate will be 0, and the peers we have sent the least to should
|
|
|
|
// be unchoked
|
|
|
|
std::sort(peers.begin(), peers.end()
|
2008-03-10 04:30:01 +01:00
|
|
|
, bind(&peer_connection::unchoke_compare, _1, _2));
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2008-03-10 04:30:01 +01:00
|
|
|
std::for_each(m_connections.begin(), m_connections.end()
|
|
|
|
, bind(&peer_connection::reset_choke_counters, _1));
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2008-01-13 12:18:18 +01:00
|
|
|
// auto unchoke
|
|
|
|
int upload_limit = m_bandwidth_manager[peer_connection::upload_channel]->throttle();
|
|
|
|
if (m_settings.auto_upload_slots && upload_limit != bandwidth_limit::inf)
|
|
|
|
{
|
|
|
|
// if our current upload rate is less than 90% of our
|
2008-01-17 18:40:46 +01:00
|
|
|
// limit AND most torrents are not "congested", i.e.
|
|
|
|
// they are not holding back because of a per-torrent
|
|
|
|
// limit
|
2008-01-13 12:18:18 +01:00
|
|
|
if (m_stat.upload_rate() < upload_limit * 0.9f
|
2008-01-15 09:37:48 +01:00
|
|
|
&& m_allowed_upload_slots <= m_num_unchoked + 1
|
|
|
|
&& congested_torrents < uncongested_torrents)
|
2008-01-13 12:18:18 +01:00
|
|
|
{
|
|
|
|
++m_allowed_upload_slots;
|
|
|
|
}
|
2008-01-17 22:34:00 +01:00
|
|
|
else if (m_upload_channel.queue_size() > 1
|
2008-01-13 12:18:18 +01:00
|
|
|
&& m_allowed_upload_slots > m_max_uploads)
|
|
|
|
{
|
|
|
|
--m_allowed_upload_slots;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
// reserve one upload slot for optimistic unchokes
|
2008-01-13 12:18:18 +01:00
|
|
|
int unchoke_set_size = m_allowed_upload_slots - 1;
|
2007-08-16 14:41:46 +02:00
|
|
|
|
|
|
|
m_num_unchoked = 0;
|
|
|
|
// go through all the peers and unchoke the first ones and choke
|
|
|
|
// all the other ones.
|
|
|
|
for (std::vector<peer_connection*>::iterator i = peers.begin()
|
|
|
|
, end(peers.end()); i != end; ++i)
|
|
|
|
{
|
|
|
|
peer_connection* p = *i;
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p);
|
2007-08-17 09:37:08 +02:00
|
|
|
torrent* t = p->associated_torrent().lock().get();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-08-16 14:41:46 +02:00
|
|
|
if (unchoke_set_size > 0)
|
|
|
|
{
|
2007-08-17 09:37:08 +02:00
|
|
|
if (p->is_choked())
|
|
|
|
{
|
2007-08-19 10:32:39 +02:00
|
|
|
if (!t->unchoke_peer(*p))
|
|
|
|
continue;
|
2007-08-17 09:37:08 +02:00
|
|
|
}
|
2007-08-19 10:32:39 +02:00
|
|
|
|
|
|
|
--unchoke_set_size;
|
|
|
|
++m_num_unchoked;
|
2007-08-17 09:37:08 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->peer_info_struct());
|
2007-08-16 14:41:46 +02:00
|
|
|
if (p->peer_info_struct()->optimistically_unchoked)
|
|
|
|
{
|
|
|
|
// force a new optimistic unchoke
|
|
|
|
m_optimistic_unchoke_time_scaler = 0;
|
|
|
|
p->peer_info_struct()->optimistically_unchoked = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p->peer_info_struct());
|
2007-08-16 14:41:46 +02:00
|
|
|
if (!p->is_choked() && !p->peer_info_struct()->optimistically_unchoked)
|
2007-08-17 09:37:08 +02:00
|
|
|
t->choke_peer(*p);
|
2007-08-21 23:51:29 +02:00
|
|
|
if (!p->is_choked())
|
|
|
|
++m_num_unchoked;
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m_optimistic_unchoke_time_scaler--;
|
|
|
|
if (m_optimistic_unchoke_time_scaler <= 0)
|
|
|
|
{
|
|
|
|
m_optimistic_unchoke_time_scaler
|
|
|
|
= settings().optimistic_unchoke_multiplier;
|
|
|
|
|
|
|
|
// find the peer that has been waiting the longest to be optimistically
|
|
|
|
// unchoked
|
|
|
|
connection_map::iterator current_optimistic_unchoke = m_connections.end();
|
|
|
|
connection_map::iterator optimistic_unchoke_candidate = m_connections.end();
|
|
|
|
ptime last_unchoke = max_time();
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
for (connection_map::iterator i = m_connections.begin()
|
|
|
|
, end(m_connections.end()); i != end; ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* p = i->get();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p);
|
2007-08-16 14:41:46 +02:00
|
|
|
policy::peer* pi = p->peer_info_struct();
|
|
|
|
if (!pi) continue;
|
2007-08-17 09:37:08 +02:00
|
|
|
torrent* t = p->associated_torrent().lock().get();
|
|
|
|
if (!t) continue;
|
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
if (pi->optimistically_unchoked)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!p->is_choked());
|
|
|
|
TORRENT_ASSERT(current_optimistic_unchoke == m_connections.end());
|
2007-08-16 14:41:46 +02:00
|
|
|
current_optimistic_unchoke = i;
|
|
|
|
}
|
2007-08-17 09:37:08 +02:00
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
if (pi->last_optimistically_unchoked < last_unchoke
|
|
|
|
&& !p->is_connecting()
|
|
|
|
&& !p->is_disconnecting()
|
2007-08-17 09:37:08 +02:00
|
|
|
&& p->is_peer_interested()
|
|
|
|
&& t->free_upload_slots()
|
|
|
|
&& p->is_choked())
|
2007-08-16 14:41:46 +02:00
|
|
|
{
|
|
|
|
last_unchoke = pi->last_optimistically_unchoked;
|
|
|
|
optimistic_unchoke_candidate = i;
|
|
|
|
}
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-08-16 14:41:46 +02:00
|
|
|
if (optimistic_unchoke_candidate != m_connections.end()
|
|
|
|
&& optimistic_unchoke_candidate != current_optimistic_unchoke)
|
|
|
|
{
|
|
|
|
if (current_optimistic_unchoke != m_connections.end())
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
torrent* t = (*current_optimistic_unchoke)->associated_torrent().lock().get();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-10-31 10:48:20 +01:00
|
|
|
(*current_optimistic_unchoke)->peer_info_struct()->optimistically_unchoked = false;
|
|
|
|
t->choke_peer(*current_optimistic_unchoke->get());
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
2007-08-21 23:51:29 +02:00
|
|
|
else
|
|
|
|
{
|
|
|
|
++m_num_unchoked;
|
|
|
|
}
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
torrent* t = (*optimistic_unchoke_candidate)->associated_torrent().lock().get();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(t);
|
2007-10-31 10:48:20 +01:00
|
|
|
bool ret = t->unchoke_peer(*optimistic_unchoke_candidate->get());
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(ret);
|
2007-10-31 10:48:20 +01:00
|
|
|
(*optimistic_unchoke_candidate)->peer_info_struct()->optimistically_unchoked = true;
|
2007-08-16 14:41:46 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// --------------------------------------------------------------
|
|
|
|
// disconnect peers when we have too many
|
|
|
|
// --------------------------------------------------------------
|
|
|
|
--m_disconnect_time_scaler;
|
|
|
|
if (m_disconnect_time_scaler <= 0)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-08-16 14:41:46 +02:00
|
|
|
m_disconnect_time_scaler = 60;
|
|
|
|
|
|
|
|
// every 60 seconds, disconnect the worst peer
|
|
|
|
// if we have reached the connection limit
|
|
|
|
if (num_connections() >= max_connections() && !m_torrents.empty())
|
|
|
|
{
|
|
|
|
torrent_map::iterator i = std::max_element(m_torrents.begin(), m_torrents.end()
|
2007-09-08 20:18:09 +02:00
|
|
|
, bind(&torrent::num_peers, bind(&torrent_map::value_type::second, _1))
|
|
|
|
< bind(&torrent::num_peers, bind(&torrent_map::value_type::second, _2)));
|
2007-08-16 14:41:46 +02:00
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i != m_torrents.end());
|
2007-08-16 14:41:46 +02:00
|
|
|
i->second->get_policy().disconnect_one_peer();
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
}
|
2006-12-21 23:20:28 +01:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
void session_impl::operator()()
|
|
|
|
{
|
|
|
|
eh_initializer();
|
|
|
|
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_mutex);
|
2007-09-22 18:27:29 +02:00
|
|
|
if (m_listen_interface.port() != 0) open_listen_port();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-04-05 00:27:36 +02:00
|
|
|
ptime timer = time_now();
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
do
|
|
|
|
{
|
2008-04-07 03:22:26 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-10-11 22:57:54 +02:00
|
|
|
try
|
|
|
|
{
|
2008-04-07 03:22:26 +02:00
|
|
|
#endif
|
2006-12-15 18:47:21 +01:00
|
|
|
m_io_service.run();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_abort == true);
|
2008-04-07 03:22:26 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
2007-02-12 19:21:14 +01:00
|
|
|
#ifndef NDEBUG
|
2006-10-11 22:57:54 +02:00
|
|
|
std::cerr << e.what() << "\n";
|
|
|
|
std::string err = e.what();
|
2007-02-12 19:21:14 +01:00
|
|
|
#endif
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(false);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
2008-04-07 03:22:26 +02:00
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
while (!m_abort);
|
|
|
|
|
2007-05-23 03:02:46 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " locking mutex\n";
|
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2008-01-27 23:39:50 +01:00
|
|
|
session_impl::mutex_t::scoped_lock l(m_mutex);
|
|
|
|
/*
|
2006-10-11 22:57:54 +02:00
|
|
|
#ifndef NDEBUG
|
|
|
|
for (torrent_map::iterator i = m_torrents.begin();
|
|
|
|
i != m_torrents.end(); ++i)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(i->second->num_peers() == 0);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
#endif
|
2008-01-27 23:39:50 +01:00
|
|
|
*/
|
2007-05-23 03:02:46 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " cleaning up torrents\n";
|
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
m_torrents.clear();
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrents.empty());
|
|
|
|
TORRENT_ASSERT(m_connections.empty());
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// the return value from this function is valid only as long as the
|
|
|
|
// session is locked!
|
|
|
|
boost::weak_ptr<torrent> session_impl::find_torrent(sha1_hash const& info_hash)
|
|
|
|
{
|
|
|
|
std::map<sha1_hash, boost::shared_ptr<torrent> >::iterator i
|
|
|
|
= m_torrents.find(info_hash);
|
|
|
|
#ifndef NDEBUG
|
|
|
|
for (std::map<sha1_hash, boost::shared_ptr<torrent> >::iterator j
|
|
|
|
= m_torrents.begin(); j != m_torrents.end(); ++j)
|
|
|
|
{
|
|
|
|
torrent* p = boost::get_pointer(j->second);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(p);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
if (i != m_torrents.end()) return i->second;
|
|
|
|
return boost::weak_ptr<torrent>();
|
|
|
|
}
|
|
|
|
|
2008-02-17 23:51:03 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2006-11-15 22:39:58 +01:00
|
|
|
boost::shared_ptr<logger> session_impl::create_log(std::string const& name
|
|
|
|
, int instance, bool append)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
// current options are file_logger, cout_logger and null_logger
|
2007-11-16 22:21:28 +01:00
|
|
|
return boost::shared_ptr<logger>(new logger(m_logpath, name + ".log", instance, append));
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
std::vector<torrent_handle> session_impl::get_torrents()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
std::vector<torrent_handle> ret;
|
2006-11-14 01:08:16 +01:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
for (session_impl::torrent_map::iterator i
|
|
|
|
= m_torrents.begin(), end(m_torrents.end());
|
|
|
|
i != end; ++i)
|
|
|
|
{
|
|
|
|
if (i->second->is_aborted()) continue;
|
2008-04-09 22:09:36 +02:00
|
|
|
ret.push_back(torrent_handle(i->second));
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2006-11-14 01:08:16 +01:00
|
|
|
torrent_handle session_impl::find_torrent_handle(sha1_hash const& info_hash)
|
|
|
|
{
|
2008-04-09 22:09:36 +02:00
|
|
|
return torrent_handle(find_torrent(info_hash));
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
torrent_handle session_impl::add_torrent(
|
2007-09-01 05:00:31 +02:00
|
|
|
boost::intrusive_ptr<torrent_info> ti
|
2007-06-10 22:46:09 +02:00
|
|
|
, fs::path const& save_path
|
2006-10-11 22:57:54 +02:00
|
|
|
, entry const& resume_data
|
2007-10-08 22:01:36 +02:00
|
|
|
, storage_mode_t storage_mode
|
2007-08-22 07:31:42 +02:00
|
|
|
, storage_constructor_type sc
|
2007-09-14 02:11:33 +02:00
|
|
|
, bool paused
|
|
|
|
, void* userdata)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!save_path.empty());
|
2007-09-22 18:27:29 +02:00
|
|
|
|
2007-09-01 05:00:31 +02:00
|
|
|
if (ti->begin_files() == ti->end_files())
|
2008-04-07 03:22:26 +02:00
|
|
|
{
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-10-11 22:57:54 +02:00
|
|
|
throw std::runtime_error("no files in torrent");
|
2008-04-07 03:22:26 +02:00
|
|
|
#else
|
|
|
|
return torrent_handle();
|
|
|
|
#endif
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// lock the session and the checker thread (the order is important!)
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
// INVARIANT_CHECK;
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
if (is_aborted())
|
2008-04-07 03:22:26 +02:00
|
|
|
{
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-10-11 22:57:54 +02:00
|
|
|
throw std::runtime_error("session is closing");
|
2008-04-07 03:22:26 +02:00
|
|
|
#else
|
|
|
|
return torrent_handle();
|
|
|
|
#endif
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// is the torrent already active?
|
2007-09-01 05:00:31 +02:00
|
|
|
if (!find_torrent(ti->info_hash()).expired())
|
2008-04-07 03:22:26 +02:00
|
|
|
{
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-10-11 22:57:54 +02:00
|
|
|
throw duplicate_torrent();
|
2008-04-07 03:22:26 +02:00
|
|
|
#else
|
|
|
|
return torrent_handle();
|
|
|
|
#endif
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// create the torrent and the data associated with
|
|
|
|
// the checker thread and store it before starting
|
|
|
|
// the thread
|
|
|
|
boost::shared_ptr<torrent> torrent_ptr(
|
2008-03-08 07:06:31 +01:00
|
|
|
new torrent(*this, ti, save_path
|
2007-10-08 22:01:36 +02:00
|
|
|
, m_listen_interface, storage_mode, 16 * 1024
|
2008-03-08 07:06:31 +01:00
|
|
|
, sc, paused, resume_data));
|
2007-04-11 19:44:15 +02:00
|
|
|
torrent_ptr->start();
|
2006-10-11 22:57:54 +02: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)
|
|
|
|
{
|
2007-09-14 02:11:33 +02:00
|
|
|
boost::shared_ptr<torrent_plugin> tp((*i)(torrent_ptr.get(), userdata));
|
2006-11-23 17:21:45 +01:00
|
|
|
if (tp) torrent_ptr->add_extension(tp);
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
if (m_dht)
|
|
|
|
{
|
2007-09-01 05:00:31 +02:00
|
|
|
torrent_info::nodes_t const& nodes = ti->nodes();
|
2006-10-11 22:57:54 +02:00
|
|
|
std::for_each(nodes.begin(), nodes.end(), bind(
|
|
|
|
(void(dht::dht_tracker::*)(std::pair<std::string, int> const&))
|
|
|
|
&dht::dht_tracker::add_node
|
|
|
|
, boost::ref(m_dht), _1));
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
m_torrents.insert(std::make_pair(ti->info_hash(), torrent_ptr));
|
|
|
|
|
2008-04-09 22:09:36 +02:00
|
|
|
return torrent_handle(torrent_ptr);
|
2008-03-08 07:06:31 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::check_torrent(boost::shared_ptr<torrent> const& t)
|
|
|
|
{
|
|
|
|
if (m_queued_for_checking.empty()) t->start_checking();
|
|
|
|
m_queued_for_checking.push_back(t);
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
void session_impl::done_checking(boost::shared_ptr<torrent> const& t)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(m_queued_for_checking.front() == t);
|
|
|
|
m_queued_for_checking.pop_front();
|
|
|
|
if (!m_queued_for_checking.empty())
|
|
|
|
m_queued_for_checking.front()->start_checking();
|
2008-04-16 09:02:39 +02:00
|
|
|
|
|
|
|
if (m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_alerts.post_alert(torrent_checked_alert(
|
|
|
|
t->get_handle()
|
|
|
|
, "torrent finished checking"));
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
torrent_handle session_impl::add_torrent(
|
|
|
|
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
|
2008-03-08 07:06:31 +01:00
|
|
|
, entry const& resume_data
|
2007-10-08 22:01:36 +02:00
|
|
|
, storage_mode_t storage_mode
|
2007-08-22 07:31:42 +02:00
|
|
|
, storage_constructor_type sc
|
2007-09-14 02:11:33 +02:00
|
|
|
, bool paused
|
|
|
|
, void* userdata)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
|
|
|
|
// TODO: support resume data in this case
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!save_path.empty());
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// lock the session
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_mutex);
|
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
// INVARIANT_CHECK;
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
// is the torrent already active?
|
|
|
|
if (!find_torrent(info_hash).expired())
|
2008-04-07 03:22:26 +02:00
|
|
|
{
|
2008-03-08 07:06:31 +01:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2006-10-11 22:57:54 +02:00
|
|
|
throw duplicate_torrent();
|
2008-03-08 07:06:31 +01:00
|
|
|
#else
|
|
|
|
return torrent_handle();
|
|
|
|
#endif
|
2008-04-07 03:22:26 +02:00
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// you cannot add new torrents to a session that is closing down
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(!is_aborted());
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// create the torrent and the data associated with
|
|
|
|
// the checker thread and store it before starting
|
|
|
|
// the thread
|
|
|
|
boost::shared_ptr<torrent> torrent_ptr(
|
2008-03-08 07:06:31 +01:00
|
|
|
new torrent(*this, tracker_url, info_hash, name
|
2007-10-08 22:01:36 +02:00
|
|
|
, save_path, m_listen_interface, storage_mode, 16 * 1024
|
2008-03-08 07:06:31 +01:00
|
|
|
, sc, paused, resume_data));
|
2007-04-11 19:44:15 +02:00
|
|
|
torrent_ptr->start();
|
2006-10-11 22:57:54 +02: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)
|
|
|
|
{
|
2007-09-14 02:11:33 +02:00
|
|
|
boost::shared_ptr<torrent_plugin> tp((*i)(torrent_ptr.get(), userdata));
|
2006-11-23 17:21:45 +01:00
|
|
|
if (tp) torrent_ptr->add_extension(tp);
|
2006-11-14 01:08:16 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-03-08 07:06:31 +01:00
|
|
|
m_torrents.insert(std::make_pair(info_hash, torrent_ptr));
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2008-04-09 22:09:36 +02:00
|
|
|
return torrent_handle(torrent_ptr);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-10-13 05:33:33 +02:00
|
|
|
void session_impl::remove_torrent(const torrent_handle& h, int options)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2008-04-09 22:09:36 +02:00
|
|
|
boost::shared_ptr<torrent> tptr = h.m_torrent.lock();
|
|
|
|
if (!tptr)
|
|
|
|
#ifdef BOOST_NO_EXCEPTIONS
|
|
|
|
return;
|
|
|
|
#else
|
|
|
|
throw invalid_handle();
|
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
session_impl::torrent_map::iterator i =
|
2008-04-09 22:09:36 +02:00
|
|
|
m_torrents.find(tptr->torrent_file().info_hash());
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
if (i != m_torrents.end())
|
|
|
|
{
|
|
|
|
torrent& t = *i->second;
|
2007-10-13 05:33:33 +02:00
|
|
|
if (options & session::delete_files)
|
|
|
|
t.delete_files();
|
2006-10-11 22:57:54 +02:00
|
|
|
t.abort();
|
|
|
|
|
2007-02-20 23:58:31 +01:00
|
|
|
if ((!t.is_paused() || t.should_request())
|
2007-12-07 22:53:33 +01:00
|
|
|
&& !t.trackers().empty())
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
|
|
|
tracker_request req = t.generate_tracker_request();
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(req.event == tracker_request::stopped);
|
2007-09-22 18:27:29 +02:00
|
|
|
req.listen_port = 0;
|
|
|
|
if (!m_listen_sockets.empty())
|
|
|
|
req.listen_port = m_listen_sockets.front().external_port;
|
2006-10-11 22:57:54 +02:00
|
|
|
req.key = m_key;
|
2006-11-15 22:39:58 +01:00
|
|
|
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
boost::shared_ptr<tracker_logger> tl(new tracker_logger(*this));
|
|
|
|
m_tracker_loggers.push_back(tl);
|
2008-01-08 06:47:43 +01:00
|
|
|
m_tracker_manager.queue_request(m_io_service, m_half_open, req
|
2007-03-02 19:40:02 +01:00
|
|
|
, t.tracker_login(), m_listen_interface.address(), tl);
|
2006-11-15 22:39:58 +01:00
|
|
|
#else
|
2008-01-08 06:47:43 +01:00
|
|
|
m_tracker_manager.queue_request(m_io_service, m_half_open, req
|
2007-03-02 19:40:02 +01:00
|
|
|
, t.tracker_login(), m_listen_interface.address());
|
2006-11-15 22:39:58 +01:00
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
if (m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
m_alerts.post_alert(
|
|
|
|
tracker_announce_alert(
|
|
|
|
t.get_handle(), "tracker announce, event=stopped"));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#ifndef NDEBUG
|
|
|
|
sha1_hash i_hash = t.torrent_file().info_hash();
|
|
|
|
#endif
|
|
|
|
m_torrents.erase(i);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrents.find(i_hash) == m_torrents.end());
|
2006-10-11 22:57:54 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool session_impl::listen_on(
|
|
|
|
std::pair<int, int> const& port_range
|
|
|
|
, const char* net_interface)
|
|
|
|
{
|
|
|
|
session_impl::mutex_t::scoped_lock l(m_mutex);
|
|
|
|
|
2007-08-21 20:33:28 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
tcp::endpoint new_interface;
|
|
|
|
if (net_interface && std::strlen(net_interface) > 0)
|
|
|
|
new_interface = tcp::endpoint(address::from_string(net_interface), port_range.first);
|
|
|
|
else
|
2007-09-22 18:27:29 +02:00
|
|
|
new_interface = tcp::endpoint(address_v4::any(), port_range.first);
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
m_listen_port_retries = port_range.second - port_range.first;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
// if the interface is the same and the socket is open
|
|
|
|
// don't do anything
|
|
|
|
if (new_interface == m_listen_interface
|
2007-09-22 18:27:29 +02:00
|
|
|
&& !m_listen_sockets.empty()) return true;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-03-16 22:45:29 +01:00
|
|
|
m_listen_interface = new_interface;
|
|
|
|
|
|
|
|
open_listen_port();
|
|
|
|
|
2007-04-04 04:06:07 +02:00
|
|
|
bool new_listen_address = m_listen_interface.address() != new_interface.address();
|
2007-04-14 19:30:57 +02:00
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2007-03-16 22:45:29 +01:00
|
|
|
if ((new_listen_address || m_dht_same_port) && m_dht)
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-03-15 23:03:56 +01:00
|
|
|
if (m_dht_same_port)
|
|
|
|
m_dht_settings.service_port = new_interface.port();
|
2006-10-11 22:57:54 +02:00
|
|
|
// the listen interface changed, rebind the dht listen socket as well
|
2007-12-09 05:15:24 +01:00
|
|
|
m_dht_socket.bind(m_dht_settings.service_port);
|
2007-05-31 02:21:54 +02:00
|
|
|
if (m_natpmp.get())
|
2008-04-06 21:17:58 +02:00
|
|
|
{
|
|
|
|
if (m_udp_mapping[0] != -1) m_natpmp->delete_mapping(m_udp_mapping[0]);
|
|
|
|
m_udp_mapping[0] = m_natpmp->add_mapping(natpmp::tcp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
|
|
|
}
|
2007-05-31 02:21:54 +02:00
|
|
|
if (m_upnp.get())
|
2008-04-06 21:17:58 +02:00
|
|
|
{
|
|
|
|
if (m_udp_mapping[1] != -1) m_upnp->delete_mapping(m_udp_mapping[1]);
|
|
|
|
m_udp_mapping[1] = m_upnp->add_mapping(upnp::tcp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-02-17 23:51:03 +01:00
|
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
2006-11-15 22:39:58 +01:00
|
|
|
m_logger = create_log("main_session", listen_port(), false);
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string() << "\n";
|
2006-11-15 22:39:58 +01:00
|
|
|
#endif
|
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
return !m_listen_sockets.empty();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
unsigned short session_impl::listen_port() const
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-09-22 18:27:29 +02:00
|
|
|
if (m_listen_sockets.empty()) return 0;
|
|
|
|
return m_listen_sockets.front().external_port;;
|
2007-03-15 23:03:56 +01:00
|
|
|
}
|
|
|
|
|
2007-04-04 04:06:07 +02:00
|
|
|
void session_impl::announce_lsd(sha1_hash const& ih)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-04-14 17:51:36 +02:00
|
|
|
// use internal listen port for local peers
|
2007-05-31 02:21:54 +02:00
|
|
|
if (m_lsd.get())
|
|
|
|
m_lsd->announce(ih, m_listen_interface.port());
|
2007-04-04 04:06:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::on_lsd_peer(tcp::endpoint peer, sha1_hash const& ih)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
|
2007-08-21 20:33:28 +02:00
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-04-04 04:06:07 +02:00
|
|
|
boost::shared_ptr<torrent> t = find_torrent(ih).lock();
|
|
|
|
if (!t) return;
|
2007-07-23 02:38:31 +02:00
|
|
|
// don't add peers from lsd to private torrents
|
|
|
|
if (t->torrent_file().priv()) return;
|
2007-04-04 04:06:07 +02:00
|
|
|
|
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
2007-04-10 09:52:58 +02:00
|
|
|
(*m_logger) << time_now_string()
|
2007-04-04 04:06:07 +02:00
|
|
|
<< ": added peer from local discovery: " << peer << "\n";
|
|
|
|
#endif
|
2007-04-10 23:23:13 +02:00
|
|
|
t->get_policy().peer_from_tracker(peer, peer_id(0), peer_info::lsd, 0);
|
2007-04-04 04:06:07 +02:00
|
|
|
}
|
|
|
|
|
2008-04-06 21:17:58 +02:00
|
|
|
void session_impl::on_port_mapping(int mapping, int port
|
|
|
|
, std::string const& errmsg, int map_transport)
|
2007-03-15 23:03:56 +01:00
|
|
|
{
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2008-04-06 21:17:58 +02:00
|
|
|
if (mapping == m_udp_mapping[map_transport] && port != 0)
|
2007-03-15 23:03:56 +01:00
|
|
|
{
|
2008-04-06 21:17:58 +02:00
|
|
|
m_external_udp_port = port;
|
|
|
|
m_dht_settings.service_port = port;
|
2007-03-16 09:25:08 +01:00
|
|
|
if (m_alerts.should_post(alert::info))
|
2008-04-06 21:17:58 +02:00
|
|
|
m_alerts.post_alert(portmap_alert(mapping, port
|
2008-04-07 01:18:35 +02:00
|
|
|
, map_transport, "successfully mapped UDP port"));
|
2008-04-06 21:17:58 +02:00
|
|
|
return;
|
2007-03-15 23:03:56 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2008-04-06 21:17:58 +02:00
|
|
|
if (mapping == m_tcp_mapping[map_transport] && port != 0)
|
2007-03-15 23:03:56 +01:00
|
|
|
{
|
2007-09-22 18:27:29 +02:00
|
|
|
if (!m_listen_sockets.empty())
|
2008-04-06 21:17:58 +02:00
|
|
|
m_listen_sockets.front().external_port = port;
|
2007-03-16 09:25:08 +01:00
|
|
|
if (m_alerts.should_post(alert::info))
|
2008-04-06 21:17:58 +02:00
|
|
|
m_alerts.post_alert(portmap_alert(mapping, port
|
2008-04-07 01:18:35 +02:00
|
|
|
, map_transport, "successfully mapped TCP port"));
|
2008-04-06 21:17:58 +02:00
|
|
|
return;
|
2007-03-15 23:03:56 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if (!errmsg.empty())
|
|
|
|
{
|
2007-03-16 09:25:08 +01:00
|
|
|
if (m_alerts.should_post(alert::warning))
|
2008-04-07 01:18:35 +02:00
|
|
|
m_alerts.post_alert(portmap_error_alert(mapping
|
|
|
|
, map_transport, errmsg));
|
2008-04-06 21:17:58 +02:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (m_alerts.should_post(alert::warning))
|
2008-04-07 01:18:35 +02:00
|
|
|
m_alerts.post_alert(portmap_alert(mapping, port
|
|
|
|
, map_transport, "successfully mapped port"));
|
2007-03-15 23:03:56 +01:00
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
session_status session_impl::status() const
|
|
|
|
{
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2007-09-22 18:27:29 +02:00
|
|
|
// INVARIANT_CHECK;
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
session_status s;
|
2008-01-13 12:18:18 +01:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
s.num_peers = (int)m_connections.size();
|
2008-01-13 12:18:18 +01:00
|
|
|
s.num_unchoked = m_num_unchoked;
|
|
|
|
s.allowed_upload_slots = m_allowed_upload_slots;
|
|
|
|
|
2008-01-17 18:40:46 +01:00
|
|
|
s.up_bandwidth_queue = m_upload_channel.queue_size();
|
|
|
|
s.down_bandwidth_queue = m_download_channel.queue_size();
|
|
|
|
|
2008-01-13 12:18:18 +01:00
|
|
|
s.has_incoming_connections = m_incoming_connection;
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
s.download_rate = m_stat.download_rate();
|
|
|
|
s.upload_rate = m_stat.upload_rate();
|
|
|
|
|
|
|
|
s.payload_download_rate = m_stat.download_payload_rate();
|
|
|
|
s.payload_upload_rate = m_stat.upload_payload_rate();
|
|
|
|
|
|
|
|
s.total_download = m_stat.total_protocol_download()
|
|
|
|
+ m_stat.total_payload_download();
|
|
|
|
|
|
|
|
s.total_upload = m_stat.total_protocol_upload()
|
|
|
|
+ m_stat.total_payload_upload();
|
|
|
|
|
|
|
|
s.total_payload_download = m_stat.total_payload_download();
|
|
|
|
s.total_payload_upload = m_stat.total_payload_upload();
|
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
if (m_dht)
|
|
|
|
{
|
|
|
|
m_dht->dht_status(s);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2006-11-28 19:18:37 +01:00
|
|
|
s.dht_nodes = 0;
|
|
|
|
s.dht_node_cache = 0;
|
|
|
|
s.dht_torrents = 0;
|
2007-05-12 03:52:25 +02:00
|
|
|
s.dht_global_nodes = 0;
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
|
|
|
|
|
|
void session_impl::start_dht(entry const& startup_state)
|
|
|
|
{
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-02-25 10:42:43 +01:00
|
|
|
if (m_dht)
|
|
|
|
{
|
|
|
|
m_dht->stop();
|
2007-03-02 02:16:59 +01:00
|
|
|
m_dht = 0;
|
2007-02-25 10:42:43 +01:00
|
|
|
}
|
2007-03-16 22:04:58 +01:00
|
|
|
if (m_dht_settings.service_port == 0
|
|
|
|
|| m_dht_same_port)
|
|
|
|
{
|
2007-03-15 23:03:56 +01:00
|
|
|
m_dht_same_port = true;
|
2007-03-16 22:45:29 +01:00
|
|
|
// if you hit this assert you are trying to start the
|
|
|
|
// DHT with the same port as the tcp listen port
|
|
|
|
// (which is default) _before_ you have opened the
|
|
|
|
// tcp listen port (so there is no configured port to use)
|
|
|
|
// basically, make sure you call listen_on() before
|
|
|
|
// start_dht(). See documentation for listen_on() for
|
|
|
|
// more information.
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_listen_interface.port() > 0);
|
2007-03-16 22:04:58 +01:00
|
|
|
m_dht_settings.service_port = m_listen_interface.port();
|
|
|
|
}
|
2007-03-15 23:03:56 +01:00
|
|
|
m_external_udp_port = m_dht_settings.service_port;
|
2008-04-06 21:17:58 +02:00
|
|
|
if (m_natpmp.get() && m_udp_mapping[0] == -1)
|
|
|
|
{
|
|
|
|
m_udp_mapping[0] = m_natpmp->add_mapping(natpmp::udp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
|
|
|
}
|
|
|
|
if (m_upnp.get() && m_udp_mapping[1] == -1)
|
|
|
|
{
|
|
|
|
m_udp_mapping[1] = m_upnp->add_mapping(upnp::udp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
|
|
|
}
|
2007-12-09 05:15:24 +01:00
|
|
|
m_dht = new dht::dht_tracker(m_dht_socket, m_dht_settings, startup_state);
|
|
|
|
if (!m_dht_socket.is_open() || m_dht_socket.local_port() != m_dht_settings.service_port)
|
|
|
|
{
|
|
|
|
m_dht_socket.bind(m_dht_settings.service_port);
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::stop_dht()
|
|
|
|
{
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-03-02 06:02:12 +01:00
|
|
|
if (!m_dht) return;
|
2007-02-25 10:42:43 +01:00
|
|
|
m_dht->stop();
|
2007-03-02 02:16:59 +01:00
|
|
|
m_dht = 0;
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_dht_settings(dht_settings const& settings)
|
|
|
|
{
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-03-15 23:03:56 +01:00
|
|
|
// only change the dht listen port in case the settings
|
|
|
|
// contains a vaiid port, and if it is different from
|
|
|
|
// the current setting
|
|
|
|
if (settings.service_port != 0)
|
|
|
|
m_dht_same_port = false;
|
2007-03-16 22:04:58 +01:00
|
|
|
else
|
|
|
|
m_dht_same_port = true;
|
2007-03-15 23:03:56 +01:00
|
|
|
if (!m_dht_same_port
|
|
|
|
&& settings.service_port != m_dht_settings.service_port
|
2006-10-11 22:57:54 +02:00
|
|
|
&& m_dht)
|
|
|
|
{
|
2007-12-09 05:15:24 +01:00
|
|
|
m_dht_socket.bind(settings.service_port);
|
|
|
|
|
2007-05-31 02:21:54 +02:00
|
|
|
if (m_natpmp.get())
|
2008-04-06 21:17:58 +02:00
|
|
|
{
|
|
|
|
if (m_udp_mapping[0] != -1) m_upnp->delete_mapping(m_udp_mapping[0]);
|
|
|
|
m_udp_mapping[0] = m_natpmp->add_mapping(natpmp::udp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
|
|
|
}
|
2007-05-31 02:21:54 +02:00
|
|
|
if (m_upnp.get())
|
2008-04-06 21:17:58 +02:00
|
|
|
{
|
|
|
|
if (m_udp_mapping[1] != -1) m_upnp->delete_mapping(m_udp_mapping[1]);
|
|
|
|
m_udp_mapping[1] = m_upnp->add_mapping(upnp::udp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
|
|
|
}
|
2007-03-15 23:03:56 +01:00
|
|
|
m_external_udp_port = settings.service_port;
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
m_dht_settings = settings;
|
2007-03-15 23:03:56 +01:00
|
|
|
if (m_dht_same_port)
|
|
|
|
m_dht_settings.service_port = m_listen_interface.port();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
entry session_impl::dht_state() const
|
|
|
|
{
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-10-29 17:18:37 +01:00
|
|
|
if (!m_dht) return entry();
|
2006-10-11 22:57:54 +02:00
|
|
|
return m_dht->state();
|
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::add_dht_node(std::pair<std::string, int> const& node)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_dht);
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2006-10-11 22:57:54 +02:00
|
|
|
m_dht->add_node(node);
|
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::add_dht_router(std::pair<std::string, int> const& node)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_dht);
|
2006-10-13 02:01:18 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2006-10-11 22:57:54 +02:00
|
|
|
m_dht->add_router_node(node);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
2007-06-06 02:41:20 +02:00
|
|
|
#ifndef TORRENT_DISABLE_ENCRYPTION
|
|
|
|
void session_impl::set_pe_settings(pe_settings const& settings)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
m_pe_settings = settings;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
bool session_impl::is_listening() const
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-09-22 18:27:29 +02:00
|
|
|
return !m_listen_sockets.empty();
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
session_impl::~session_impl()
|
|
|
|
{
|
2007-05-24 21:51:14 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << "\n\n *** shutting down session *** \n\n";
|
|
|
|
#endif
|
2007-10-26 09:14:19 +02:00
|
|
|
abort();
|
2007-03-02 06:02:12 +01:00
|
|
|
|
2008-04-05 06:53:22 +02:00
|
|
|
#ifndef TORRENT_DISABLE_GEO_IP
|
2008-04-11 10:46:43 +02:00
|
|
|
if (m_asnum_db) GeoIP_delete(m_asnum_db);
|
|
|
|
if (m_country_db) GeoIP_delete(m_country_db);
|
2008-04-05 06:53:22 +02:00
|
|
|
#endif
|
2007-05-24 21:51:14 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " waiting for main thread\n";
|
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
m_thread->join();
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrents.empty());
|
2007-04-02 22:00:24 +02:00
|
|
|
|
2007-12-24 09:15:10 +01:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " waiting for disk io thread\n";
|
|
|
|
#endif
|
|
|
|
m_disk_thread.join();
|
|
|
|
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_torrents.empty());
|
|
|
|
TORRENT_ASSERT(m_connections.empty());
|
2007-05-24 21:51:14 +02:00
|
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
|
|
(*m_logger) << time_now_string() << " shutdown complete!\n";
|
|
|
|
#endif
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_max_uploads(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit > 0 || limit == -1);
|
2006-10-11 22:57:54 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2008-01-15 09:37:48 +01:00
|
|
|
if (m_max_uploads == limit) return;
|
2006-10-11 22:57:54 +02:00
|
|
|
m_max_uploads = limit;
|
2008-01-15 09:37:48 +01:00
|
|
|
m_allowed_upload_slots = limit;
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_max_connections(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit > 0 || limit == -1);
|
2006-10-11 22:57:54 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2006-10-11 22:57:54 +02:00
|
|
|
m_max_connections = limit;
|
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::set_max_half_open_connections(int limit)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(limit > 0 || limit == -1);
|
2006-10-11 22:57:54 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-21 06:46:17 +02:00
|
|
|
if (limit <= 0) limit = (std::numeric_limits<int>::max)();
|
2007-05-05 02:29:33 +02:00
|
|
|
m_half_open.limit(limit);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
2007-07-03 01:48:06 +02:00
|
|
|
void session_impl::set_download_rate_limit(int bytes_per_second)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(bytes_per_second > 0 || bytes_per_second == -1);
|
2007-07-03 01:48:06 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-17 05:43:26 +02:00
|
|
|
if (bytes_per_second <= 0) bytes_per_second = bandwidth_limit::inf;
|
2007-07-03 01:48:06 +02:00
|
|
|
m_bandwidth_manager[peer_connection::download_channel]->throttle(bytes_per_second);
|
|
|
|
}
|
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
void session_impl::set_upload_rate_limit(int bytes_per_second)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(bytes_per_second > 0 || bytes_per_second == -1);
|
2006-10-11 22:57:54 +02:00
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-17 05:43:26 +02:00
|
|
|
if (bytes_per_second <= 0) bytes_per_second = bandwidth_limit::inf;
|
2007-07-03 01:48:06 +02:00
|
|
|
m_bandwidth_manager[peer_connection::upload_channel]->throttle(bytes_per_second);
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
std::auto_ptr<alert> session_impl::pop_alert()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2007-08-21 21:18:06 +02:00
|
|
|
// too expensive
|
|
|
|
// INVARIANT_CHECK;
|
2007-08-21 20:33:28 +02:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
if (m_alerts.pending())
|
|
|
|
return m_alerts.get();
|
|
|
|
return std::auto_ptr<alert>(0);
|
|
|
|
}
|
2007-11-25 09:18:57 +01:00
|
|
|
|
|
|
|
alert const* session_impl::wait_for_alert(time_duration max_wait)
|
|
|
|
{
|
|
|
|
return m_alerts.wait_for_alert(max_wait);
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
|
|
|
|
void session_impl::set_severity_level(alert::severity_t s)
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
m_alerts.set_severity(s);
|
|
|
|
}
|
|
|
|
|
2007-01-02 00:51:24 +01:00
|
|
|
int session_impl::upload_rate_limit() const
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-08-07 09:35:21 +02:00
|
|
|
int ret = m_bandwidth_manager[peer_connection::upload_channel]->throttle();
|
2007-08-21 06:46:17 +02:00
|
|
|
return ret == (std::numeric_limits<int>::max)() ? -1 : ret;
|
2007-01-02 00:51:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
int session_impl::download_rate_limit() const
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-07 09:35:21 +02:00
|
|
|
int ret = m_bandwidth_manager[peer_connection::download_channel]->throttle();
|
2007-08-21 06:46:17 +02:00
|
|
|
return ret == (std::numeric_limits<int>::max)() ? -1 : ret;
|
2007-01-02 00:51:24 +01:00
|
|
|
}
|
2007-05-31 02:21:54 +02:00
|
|
|
|
|
|
|
void session_impl::start_lsd()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2007-11-23 22:11:31 +01:00
|
|
|
if (m_lsd) return;
|
|
|
|
|
2007-09-29 23:31:51 +02:00
|
|
|
m_lsd = new lsd(m_io_service
|
2007-05-31 02:21:54 +02:00
|
|
|
, m_listen_interface.address()
|
2007-09-29 23:31:51 +02:00
|
|
|
, bind(&session_impl::on_lsd_peer, this, _1, _2));
|
2007-05-31 02:21:54 +02:00
|
|
|
}
|
|
|
|
|
2008-04-07 01:18:35 +02:00
|
|
|
natpmp* session_impl::start_natpmp()
|
2007-05-31 02:21:54 +02:00
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-04-07 01:18:35 +02:00
|
|
|
if (m_natpmp) return m_natpmp.get();
|
2007-11-23 22:11:31 +01:00
|
|
|
|
2007-09-29 23:31:51 +02:00
|
|
|
m_natpmp = new natpmp(m_io_service
|
2007-05-31 02:21:54 +02:00
|
|
|
, m_listen_interface.address()
|
|
|
|
, bind(&session_impl::on_port_mapping
|
2008-04-06 21:17:58 +02:00
|
|
|
, this, _1, _2, _3, 0));
|
2007-05-31 02:21:54 +02:00
|
|
|
|
2008-04-13 21:19:22 +02:00
|
|
|
if (m_listen_interface.port() > 0)
|
|
|
|
{
|
|
|
|
m_tcp_mapping[0] = m_natpmp->add_mapping(natpmp::tcp
|
|
|
|
, m_listen_interface.port(), m_listen_interface.port());
|
|
|
|
}
|
2007-05-31 02:21:54 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2008-04-06 21:17:58 +02:00
|
|
|
if (m_dht)
|
|
|
|
m_udp_mapping[0] = m_natpmp->add_mapping(natpmp::udp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
2007-05-31 02:21:54 +02:00
|
|
|
#endif
|
2008-04-07 01:18:35 +02:00
|
|
|
return m_natpmp.get();
|
2007-05-31 02:21:54 +02:00
|
|
|
}
|
|
|
|
|
2008-04-07 01:18:35 +02:00
|
|
|
upnp* session_impl::start_upnp()
|
2007-05-31 02:21:54 +02:00
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-08-21 20:33:28 +02:00
|
|
|
|
|
|
|
INVARIANT_CHECK;
|
|
|
|
|
2008-04-07 01:18:35 +02:00
|
|
|
if (m_upnp) return m_upnp.get();
|
2007-11-23 22:11:31 +01:00
|
|
|
|
2007-09-29 23:31:51 +02:00
|
|
|
m_upnp = new upnp(m_io_service, m_half_open
|
2007-05-31 02:21:54 +02:00
|
|
|
, m_listen_interface.address()
|
|
|
|
, m_settings.user_agent
|
|
|
|
, bind(&session_impl::on_port_mapping
|
2008-04-06 21:17:58 +02:00
|
|
|
, this, _1, _2, _3, 1)
|
2008-01-07 06:48:28 +01:00
|
|
|
, m_settings.upnp_ignore_nonrouters);
|
2007-05-31 02:21:54 +02:00
|
|
|
|
2007-12-24 09:18:53 +01:00
|
|
|
m_upnp->discover_device();
|
2008-04-13 21:19:22 +02:00
|
|
|
if (m_listen_interface.port() > 0)
|
|
|
|
{
|
|
|
|
m_tcp_mapping[1] = m_upnp->add_mapping(upnp::tcp
|
|
|
|
, m_listen_interface.port(), m_listen_interface.port());
|
|
|
|
}
|
2007-05-31 02:21:54 +02:00
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
2008-04-06 21:17:58 +02:00
|
|
|
if (m_dht)
|
|
|
|
m_udp_mapping[1] = m_upnp->add_mapping(upnp::udp
|
|
|
|
, m_dht_settings.service_port
|
|
|
|
, m_dht_settings.service_port);
|
2007-05-31 02:21:54 +02:00
|
|
|
#endif
|
2008-04-07 01:18:35 +02:00
|
|
|
return m_upnp.get();
|
2007-05-31 02:21:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::stop_lsd()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
2007-11-19 03:24:07 +01:00
|
|
|
if (m_lsd.get())
|
|
|
|
m_lsd->close();
|
2007-09-29 23:31:51 +02:00
|
|
|
m_lsd = 0;
|
2007-05-31 02:21:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::stop_natpmp()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
if (m_natpmp.get())
|
|
|
|
m_natpmp->close();
|
2007-09-29 23:31:51 +02:00
|
|
|
m_natpmp = 0;
|
2007-05-31 02:21:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::stop_upnp()
|
|
|
|
{
|
|
|
|
mutex_t::scoped_lock l(m_mutex);
|
|
|
|
if (m_upnp.get())
|
2008-04-06 21:17:58 +02:00
|
|
|
{
|
2007-05-31 02:21:54 +02:00
|
|
|
m_upnp->close();
|
2008-04-06 21:17:58 +02:00
|
|
|
m_udp_mapping[1] = -1;
|
|
|
|
m_tcp_mapping[1] = -1;
|
|
|
|
}
|
2007-09-29 23:31:51 +02:00
|
|
|
m_upnp = 0;
|
2007-05-31 02:21:54 +02:00
|
|
|
}
|
|
|
|
|
2008-03-29 23:45:55 +01:00
|
|
|
void session_impl::set_external_address(address const& ip)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(ip != address());
|
|
|
|
|
|
|
|
if (m_external_address == ip) return;
|
|
|
|
|
|
|
|
m_external_address = ip;
|
|
|
|
if (m_alerts.should_post(alert::info))
|
|
|
|
{
|
|
|
|
std::stringstream msg;
|
|
|
|
msg << "external address is '";
|
|
|
|
print_address(msg, ip) << "'";
|
|
|
|
m_alerts.post_alert(external_ip_alert(ip, msg.str()));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
void session_impl::free_disk_buffer(char* buf)
|
|
|
|
{
|
|
|
|
m_disk_thread.free_buffer(buf);
|
|
|
|
}
|
2008-04-10 12:03:23 +02:00
|
|
|
|
|
|
|
char* session_impl::allocate_disk_buffer()
|
|
|
|
{
|
|
|
|
return m_disk_thread.allocate_buffer();
|
|
|
|
}
|
2007-09-29 18:14:03 +02:00
|
|
|
|
|
|
|
std::pair<char*, int> session_impl::allocate_buffer(int size)
|
|
|
|
{
|
2007-12-27 22:43:11 +01:00
|
|
|
TORRENT_ASSERT(size > 0);
|
2007-09-29 18:14:03 +02:00
|
|
|
int num_buffers = (size + send_buffer_size - 1) / send_buffer_size;
|
2007-12-27 22:43:11 +01:00
|
|
|
TORRENT_ASSERT(num_buffers > 0);
|
|
|
|
|
|
|
|
boost::mutex::scoped_lock l(m_send_buffer_mutex);
|
2007-09-29 18:14:03 +02:00
|
|
|
#ifdef TORRENT_STATS
|
2008-02-14 04:48:20 +01:00
|
|
|
TORRENT_ASSERT(m_buffer_allocations >= 0);
|
2007-09-29 18:14:03 +02:00
|
|
|
m_buffer_allocations += num_buffers;
|
|
|
|
m_buffer_usage_logger << log_time() << " protocol_buffer: "
|
|
|
|
<< (m_buffer_allocations * send_buffer_size) << std::endl;
|
|
|
|
#endif
|
2008-04-09 07:19:11 +02:00
|
|
|
#ifdef TORRENT_DISABLE_POOL_ALLOCATOR
|
|
|
|
int num_bytes = num_buffers * send_buffer_size;
|
|
|
|
return std::make_pair((char*)malloc(num_bytes), num_bytes);
|
|
|
|
#else
|
2007-09-29 18:14:03 +02:00
|
|
|
return std::make_pair((char*)m_send_buffers.ordered_malloc(num_buffers)
|
|
|
|
, num_buffers * send_buffer_size);
|
2008-04-09 07:19:11 +02:00
|
|
|
#endif
|
2007-09-29 18:14:03 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void session_impl::free_buffer(char* buf, int size)
|
|
|
|
{
|
2007-12-27 22:43:11 +01:00
|
|
|
TORRENT_ASSERT(size > 0);
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(size % send_buffer_size == 0);
|
2007-09-29 18:14:03 +02:00
|
|
|
int num_buffers = size / send_buffer_size;
|
2007-12-27 22:43:11 +01:00
|
|
|
TORRENT_ASSERT(num_buffers > 0);
|
|
|
|
|
|
|
|
boost::mutex::scoped_lock l(m_send_buffer_mutex);
|
2007-09-29 18:14:03 +02:00
|
|
|
#ifdef TORRENT_STATS
|
|
|
|
m_buffer_allocations -= num_buffers;
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_buffer_allocations >= 0);
|
2007-09-29 18:14:03 +02:00
|
|
|
m_buffer_usage_logger << log_time() << " protocol_buffer: "
|
|
|
|
<< (m_buffer_allocations * send_buffer_size) << std::endl;
|
|
|
|
#endif
|
2008-04-09 07:19:11 +02:00
|
|
|
#ifdef TORRENT_DISABLE_POOL_ALLOCATOR
|
|
|
|
free(buf);
|
|
|
|
#else
|
2007-09-29 18:14:03 +02:00
|
|
|
m_send_buffers.ordered_free(buf, num_buffers);
|
2008-04-09 07:19:11 +02:00
|
|
|
#endif
|
2007-09-29 18:14:03 +02:00
|
|
|
}
|
2007-01-02 00:51:24 +01:00
|
|
|
|
2006-10-11 22:57:54 +02:00
|
|
|
#ifndef NDEBUG
|
2007-08-21 20:33:28 +02:00
|
|
|
void session_impl::check_invariant() const
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(m_max_connections > 0);
|
|
|
|
TORRENT_ASSERT(m_max_uploads > 0);
|
2008-01-13 12:18:18 +01:00
|
|
|
TORRENT_ASSERT(m_allowed_upload_slots >= m_max_uploads);
|
2007-08-16 14:41:46 +02:00
|
|
|
int unchokes = 0;
|
|
|
|
int num_optimistic = 0;
|
2007-08-21 20:33:28 +02:00
|
|
|
for (connection_map::const_iterator i = m_connections.begin();
|
2006-10-11 22:57:54 +02:00
|
|
|
i != m_connections.end(); ++i)
|
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
TORRENT_ASSERT(*i);
|
|
|
|
boost::shared_ptr<torrent> t = (*i)->associated_torrent().lock();
|
2006-10-11 22:57:54 +02:00
|
|
|
|
2007-10-31 10:48:20 +01:00
|
|
|
peer_connection* p = i->get();
|
2008-01-19 20:00:54 +01:00
|
|
|
TORRENT_ASSERT(!p->is_disconnecting());
|
2007-10-31 10:48:20 +01:00
|
|
|
if (!p->is_choked()) ++unchokes;
|
|
|
|
if (p->peer_info_struct()
|
|
|
|
&& p->peer_info_struct()->optimistically_unchoked)
|
2007-08-19 10:23:44 +02:00
|
|
|
{
|
2007-08-16 14:41:46 +02:00
|
|
|
++num_optimistic;
|
2007-10-31 10:48:20 +01:00
|
|
|
TORRENT_ASSERT(!p->is_choked());
|
2007-08-19 10:23:44 +02:00
|
|
|
}
|
2007-10-31 10:48:20 +01:00
|
|
|
if (t && p->peer_info_struct())
|
2006-10-11 22:57:54 +02:00
|
|
|
{
|
2007-10-31 10:48:20 +01:00
|
|
|
TORRENT_ASSERT(t->get_policy().has_connection(p));
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
}
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(num_optimistic == 0 || num_optimistic == 1);
|
2007-08-21 23:51:29 +02:00
|
|
|
if (m_num_unchoked != unchokes)
|
|
|
|
{
|
2007-10-05 02:30:00 +02:00
|
|
|
TORRENT_ASSERT(false);
|
2007-08-21 23:51:29 +02:00
|
|
|
}
|
2007-11-27 03:46:19 +01:00
|
|
|
for (std::map<sha1_hash, boost::shared_ptr<torrent> >::const_iterator j
|
|
|
|
= m_torrents.begin(); j != m_torrents.end(); ++j)
|
|
|
|
{
|
|
|
|
TORRENT_ASSERT(boost::get_pointer(j->second));
|
|
|
|
}
|
2006-10-11 22:57:54 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
}}
|
|
|
|
|