forked from premiere/premiere-libtorrent
1156 lines
34 KiB
C++
1156 lines
34 KiB
C++
/*
|
|
|
|
Copyright (c) 2003, Arvid Norberg
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in
|
|
the documentation and/or other materials provided with the distribution.
|
|
* Neither the name of the author nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#ifndef TORRENT_TORRENT_HPP_INCLUDE
|
|
#define TORRENT_TORRENT_HPP_INCLUDE
|
|
|
|
#include <algorithm>
|
|
#include <vector>
|
|
#include <set>
|
|
#include <list>
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push, 1)
|
|
#endif
|
|
|
|
#include <boost/limits.hpp>
|
|
#include <boost/filesystem/path.hpp>
|
|
#include <boost/tuple/tuple.hpp>
|
|
#include <boost/enable_shared_from_this.hpp>
|
|
#include <boost/scoped_ptr.hpp>
|
|
#include <boost/intrusive_ptr.hpp>
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
|
|
#include "libtorrent/torrent_handle.hpp"
|
|
#include "libtorrent/entry.hpp"
|
|
#include "libtorrent/torrent_info.hpp"
|
|
#include "libtorrent/socket.hpp"
|
|
#include "libtorrent/policy.hpp"
|
|
#include "libtorrent/tracker_manager.hpp"
|
|
#include "libtorrent/stat.hpp"
|
|
#include "libtorrent/alert.hpp"
|
|
#include "libtorrent/piece_picker.hpp"
|
|
#include "libtorrent/config.hpp"
|
|
#include "libtorrent/escape_string.hpp"
|
|
#include "libtorrent/bandwidth_limit.hpp"
|
|
#include "libtorrent/bandwidth_queue_entry.hpp"
|
|
#include "libtorrent/storage.hpp"
|
|
#include "libtorrent/hasher.hpp"
|
|
#include "libtorrent/assert.hpp"
|
|
#include "libtorrent/bitfield.hpp"
|
|
#include "libtorrent/aux_/session_impl.hpp"
|
|
#include "libtorrent/deadline_timer.hpp"
|
|
|
|
namespace libtorrent
|
|
{
|
|
#if defined(TORRENT_VERBOSE_LOGGING) || defined(TORRENT_LOGGING)
|
|
struct logger;
|
|
#endif
|
|
|
|
class piece_manager;
|
|
struct torrent_plugin;
|
|
struct bitfield;
|
|
struct announce_entry;
|
|
struct tracker_request;
|
|
struct add_torrent_params;
|
|
|
|
namespace aux
|
|
{
|
|
struct session_impl;
|
|
struct piece_checker_data;
|
|
}
|
|
|
|
struct web_seed_entry
|
|
{
|
|
std::string url;
|
|
// http seeds are different from url seeds in the
|
|
// protocol they use. http seeds follows the original
|
|
// http seed spec. by John Hoffman
|
|
enum type_t { url_seed, http_seed} type;
|
|
|
|
web_seed_entry(std::string const& url_, type_t type_)
|
|
: url(url_), type(type_) {}
|
|
|
|
bool operator==(web_seed_entry const& e) const
|
|
{ return url == e.url && type == e.type; }
|
|
|
|
bool operator<(web_seed_entry const& e) const
|
|
{ return url < e.url ? true : type < e.type; }
|
|
};
|
|
|
|
namespace fs = boost::filesystem;
|
|
|
|
// a torrent is a class that holds information
|
|
// for a specific download. It updates itself against
|
|
// the tracker
|
|
class TORRENT_EXPORT torrent: public request_callback
|
|
, public boost::enable_shared_from_this<torrent>
|
|
{
|
|
public:
|
|
|
|
torrent(aux::session_impl& ses, tcp::endpoint const& net_interface
|
|
, int block_size, int seq, add_torrent_params const& p);
|
|
~torrent();
|
|
|
|
#ifndef TORRENT_DISABLE_ENCRYPTION
|
|
sha1_hash const& obfuscated_hash() const
|
|
{ return m_obfuscated_hash; }
|
|
#endif
|
|
|
|
sha1_hash const& info_hash() const
|
|
{ return m_torrent_file->info_hash(); }
|
|
|
|
// starts the announce timer
|
|
void start();
|
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
void add_extension(boost::shared_ptr<torrent_plugin>);
|
|
void add_extension(boost::function<boost::shared_ptr<torrent_plugin>(torrent*, void*)> const& ext
|
|
, void* userdata);
|
|
#endif
|
|
|
|
#ifdef TORRENT_DEBUG
|
|
bool has_peer(peer_connection* p) const
|
|
{ return m_connections.find(p) != m_connections.end(); }
|
|
#endif
|
|
|
|
// this is called when the torrent has metadata.
|
|
// it will initialize the storage and the piece-picker
|
|
void init();
|
|
|
|
void on_resume_data_checked(int ret, disk_io_job const& j);
|
|
void on_force_recheck(int ret, disk_io_job const& j);
|
|
void on_piece_checked(int ret, disk_io_job const& j);
|
|
void files_checked_lock();
|
|
void files_checked(aux::session_impl::mutex_t::scoped_lock const&);
|
|
void start_checking();
|
|
|
|
void start_announcing();
|
|
void stop_announcing();
|
|
|
|
void set_upload_mode(bool b);
|
|
bool upload_mode() const { return m_upload_mode; }
|
|
|
|
int seed_rank(session_settings const& s) const;
|
|
|
|
enum flags_t { overwrite_existing = 1 };
|
|
void add_piece(int piece, char const* data, int flags = 0);
|
|
void on_disk_write_complete(int ret, disk_io_job const& j
|
|
, peer_request p);
|
|
|
|
struct read_piece_struct
|
|
{
|
|
boost::shared_array<char> piece_data;
|
|
int blocks_left;
|
|
bool fail;
|
|
};
|
|
void read_piece(int piece);
|
|
void on_disk_read_complete(int ret, disk_io_job const& j, peer_request r, read_piece_struct* rp);
|
|
|
|
storage_mode_t storage_mode() const { return m_storage_mode; }
|
|
storage_interface* get_storage()
|
|
{
|
|
if (!m_owning_storage) return 0;
|
|
return m_owning_storage->get_storage_impl();
|
|
}
|
|
|
|
// this will flag the torrent as aborted. The main
|
|
// loop in session_impl will check for this state
|
|
// on all torrents once every second, and take
|
|
// the necessary actions then.
|
|
void abort();
|
|
bool is_aborted() const { return m_abort; }
|
|
|
|
torrent_status::state_t state() const { return m_state; }
|
|
void set_state(torrent_status::state_t s);
|
|
|
|
session_settings const& settings() const;
|
|
|
|
aux::session_impl& session() { return m_ses; }
|
|
|
|
void set_sequential_download(bool sd);
|
|
bool is_sequential_download() const
|
|
{ return m_sequential_download; }
|
|
|
|
void set_queue_position(int p);
|
|
int queue_position() const { return m_sequence_number; }
|
|
|
|
void second_tick(stat& accumulator, int tick_interval_ms);
|
|
|
|
std::string name() const;
|
|
|
|
stat statistics() const { return m_stat; }
|
|
void add_stats(stat const& s) { m_stat += s; }
|
|
size_type bytes_left() const;
|
|
int block_bytes_wanted(piece_block const& p) const;
|
|
void bytes_done(torrent_status& st) const;
|
|
size_type quantized_bytes_done() const;
|
|
|
|
void ip_filter_updated() { m_policy.ip_filter_updated(); }
|
|
|
|
void handle_disk_error(disk_io_job const& j, peer_connection* c = 0);
|
|
void clear_error();
|
|
void set_error(error_code const& ec, std::string const& file);
|
|
bool has_error() const { return m_error; }
|
|
void pause();
|
|
void resume();
|
|
|
|
ptime started() const { return m_started; }
|
|
void do_pause();
|
|
void do_resume();
|
|
|
|
bool is_paused() const;
|
|
bool is_torrent_paused() const { return m_paused; }
|
|
void force_recheck();
|
|
void save_resume_data();
|
|
|
|
bool is_auto_managed() const { return m_auto_managed; }
|
|
void auto_managed(bool a);
|
|
|
|
bool should_check_files() const;
|
|
|
|
void delete_files();
|
|
|
|
// ============ start deprecation =============
|
|
void filter_piece(int index, bool filter);
|
|
void filter_pieces(std::vector<bool> const& bitmask);
|
|
bool is_piece_filtered(int index) const;
|
|
void filtered_pieces(std::vector<bool>& bitmask) const;
|
|
void filter_files(std::vector<bool> const& files);
|
|
#if !TORRENT_NO_FPU
|
|
void file_progress(std::vector<float>& fp) const;
|
|
#endif
|
|
// ============ end deprecation =============
|
|
|
|
void piece_availability(std::vector<int>& avail) const;
|
|
|
|
void set_piece_priority(int index, int priority);
|
|
int piece_priority(int index) const;
|
|
|
|
void prioritize_pieces(std::vector<int> const& pieces);
|
|
void piece_priorities(std::vector<int>&) const;
|
|
|
|
void set_file_priority(int index, int priority);
|
|
int file_priority(int index) const;
|
|
|
|
void prioritize_files(std::vector<int> const& files);
|
|
void file_priorities(std::vector<int>&) const;
|
|
|
|
void set_piece_deadline(int piece, time_duration t, int flags);
|
|
void update_piece_priorities();
|
|
|
|
torrent_status status() const;
|
|
|
|
void file_progress(std::vector<size_type>& fp, int flags = 0) const;
|
|
|
|
void use_interface(const char* net_interface);
|
|
tcp::endpoint const& get_interface() const { return m_net_interface; }
|
|
|
|
void connect_to_url_seed(web_seed_entry const& url);
|
|
bool connect_to_peer(policy::peer* peerinfo);
|
|
|
|
void set_ratio(float ratio)
|
|
{ TORRENT_ASSERT(ratio >= 0.0f); m_ratio = ratio; }
|
|
|
|
float ratio() const
|
|
{ return m_ratio; }
|
|
|
|
int priority() const { return m_priority; }
|
|
void set_priority(int prio)
|
|
{
|
|
TORRENT_ASSERT(prio <= 255 && prio >= 0);
|
|
if (prio > 255) prio = 255;
|
|
else if (prio < 0) prio = 0;
|
|
m_priority = prio;
|
|
}
|
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
|
void resolve_countries(bool r)
|
|
{ m_resolve_countries = r; }
|
|
|
|
bool resolving_countries() const { return m_resolve_countries; }
|
|
#endif
|
|
|
|
// --------------------------------------------
|
|
// BANDWIDTH MANAGEMENT
|
|
|
|
bandwidth_channel m_bandwidth_channel[2];
|
|
|
|
int bandwidth_throttle(int channel) const;
|
|
|
|
// --------------------------------------------
|
|
// PEER MANAGEMENT
|
|
|
|
// add or remove a url that will be attempted for
|
|
// finding the file(s) in this torrent.
|
|
void add_web_seed(std::string const& url, web_seed_entry::type_t type)
|
|
{ m_web_seeds.insert(web_seed_entry(url, type)); }
|
|
|
|
void remove_web_seed(std::string const& url, web_seed_entry::type_t type)
|
|
{ m_web_seeds.erase(web_seed_entry(url, type)); }
|
|
|
|
void retry_web_seed(std::string const& url, web_seed_entry::type_t type, int retry = 0);
|
|
|
|
std::set<web_seed_entry> web_seeds() const
|
|
{ return m_web_seeds; }
|
|
|
|
std::set<std::string> web_seeds(web_seed_entry::type_t type) const;
|
|
|
|
bool free_upload_slots() const
|
|
{ return m_num_uploads < m_max_uploads; }
|
|
|
|
bool choke_peer(peer_connection& c);
|
|
bool unchoke_peer(peer_connection& c);
|
|
|
|
// used by peer_connection to attach itself to a torrent
|
|
// since incoming connections don't know what torrent
|
|
// they're a part of until they have received an info_hash.
|
|
// false means attach failed
|
|
bool attach_peer(peer_connection* p);
|
|
|
|
// this will remove the peer and make sure all
|
|
// the pieces it had have their reference counter
|
|
// decreased in the piece_picker
|
|
void remove_peer(peer_connection* p);
|
|
|
|
void cancel_block(piece_block block);
|
|
|
|
bool want_more_peers() const;
|
|
bool try_connect_peer();
|
|
void give_connect_points(int points);
|
|
|
|
// the number of peers that belong to this torrent
|
|
int num_peers() const { return (int)m_connections.size(); }
|
|
int num_seeds() const;
|
|
|
|
typedef std::set<peer_connection*>::iterator peer_iterator;
|
|
typedef std::set<peer_connection*>::const_iterator const_peer_iterator;
|
|
|
|
const_peer_iterator begin() const { return m_connections.begin(); }
|
|
const_peer_iterator end() const { return m_connections.end(); }
|
|
|
|
peer_iterator begin() { return m_connections.begin(); }
|
|
peer_iterator end() { return m_connections.end(); }
|
|
|
|
void resolve_peer_country(boost::intrusive_ptr<peer_connection> const& p) const;
|
|
|
|
void get_full_peer_list(std::vector<peer_list_entry>& v) const;
|
|
void get_peer_info(std::vector<peer_info>& v);
|
|
void get_download_queue(std::vector<partial_piece_info>& queue);
|
|
|
|
// --------------------------------------------
|
|
// TRACKER MANAGEMENT
|
|
|
|
// these are callbacks called by the tracker_connection instance
|
|
// (either http_tracker_connection or udp_tracker_connection)
|
|
// when this torrent got a response from its tracker request
|
|
// or when a failure occured
|
|
virtual void tracker_response(
|
|
tracker_request const& r
|
|
, address const& tracker_ip
|
|
, std::list<address> const& ip_list
|
|
, std::vector<peer_entry>& e, int interval
|
|
, int complete, int incomplete, address const& external_ip);
|
|
virtual void tracker_request_timed_out(
|
|
tracker_request const& r);
|
|
virtual void tracker_request_error(tracker_request const& r
|
|
, int response_code, const std::string& str);
|
|
virtual void tracker_warning(tracker_request const& req
|
|
, std::string const& msg);
|
|
virtual void tracker_scrape_response(tracker_request const& req
|
|
, int complete, int incomplete, int downloaded);
|
|
|
|
// if no password and username is set
|
|
// this will return an empty string, otherwise
|
|
// it will concatenate the login and password
|
|
// ready to be sent over http (but without
|
|
// base64 encoding).
|
|
std::string tracker_login() const;
|
|
|
|
// returns the absolute time when the next tracker
|
|
// announce will take place.
|
|
ptime next_announce() const;
|
|
|
|
// forcefully sets next_announce to the current time
|
|
void force_tracker_request();
|
|
void force_tracker_request(ptime);
|
|
void scrape_tracker();
|
|
void announce_with_tracker(tracker_request::event_t e
|
|
= tracker_request::none
|
|
, address const& bind_interface = address_v4::any());
|
|
ptime const& last_scrape() const { return m_last_scrape; }
|
|
|
|
// sets the username and password that will be sent to
|
|
// the tracker
|
|
void set_tracker_login(std::string const& name, std::string const& pw);
|
|
|
|
// the tcp::endpoint of the tracker that we managed to
|
|
// announce ourself at the last time we tried to announce
|
|
const tcp::endpoint& current_tracker() const;
|
|
|
|
announce_entry* find_tracker(tracker_request const& r);
|
|
|
|
// --------------------------------------------
|
|
// PIECE MANAGEMENT
|
|
|
|
void update_sparse_piece_prio(int piece, int cursor, int reverse_cursor);
|
|
|
|
bool super_seeding() const
|
|
{ return m_super_seeding; }
|
|
|
|
void super_seeding(bool on);
|
|
int get_piece_to_super_seed(bitfield const&);
|
|
|
|
// returns true if we have downloaded the given piece
|
|
bool have_piece(int index) const
|
|
{
|
|
return has_picker()?m_picker->have_piece(index):true;
|
|
}
|
|
|
|
// called when we learn that we have a piece
|
|
// only once per piece
|
|
void we_have(int index);
|
|
|
|
int num_have() const
|
|
{
|
|
return has_picker()
|
|
?m_picker->num_have()
|
|
:m_torrent_file->num_pieces();
|
|
}
|
|
|
|
// when we get a have message, this is called for that piece
|
|
void peer_has(int index)
|
|
{
|
|
if (m_picker.get())
|
|
{
|
|
TORRENT_ASSERT(!is_seed());
|
|
m_picker->inc_refcount(index);
|
|
}
|
|
#ifdef TORRENT_DEBUG
|
|
else
|
|
{
|
|
TORRENT_ASSERT(is_seed());
|
|
}
|
|
#endif
|
|
}
|
|
|
|
// when we get a bitfield message, this is called for that piece
|
|
void peer_has(bitfield const& bits)
|
|
{
|
|
if (m_picker.get())
|
|
{
|
|
TORRENT_ASSERT(!is_seed());
|
|
m_picker->inc_refcount(bits);
|
|
}
|
|
#ifdef TORRENT_DEBUG
|
|
else
|
|
{
|
|
TORRENT_ASSERT(is_seed());
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void peer_has_all()
|
|
{
|
|
if (m_picker.get())
|
|
{
|
|
TORRENT_ASSERT(!is_seed());
|
|
m_picker->inc_refcount_all();
|
|
}
|
|
#ifdef TORRENT_DEBUG
|
|
else
|
|
{
|
|
TORRENT_ASSERT(is_seed());
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void peer_lost(int index)
|
|
{
|
|
if (m_picker.get())
|
|
{
|
|
TORRENT_ASSERT(!is_seed());
|
|
m_picker->dec_refcount(index);
|
|
}
|
|
#ifdef TORRENT_DEBUG
|
|
else
|
|
{
|
|
TORRENT_ASSERT(is_seed());
|
|
}
|
|
#endif
|
|
}
|
|
|
|
int block_size() const { TORRENT_ASSERT(m_block_size > 0); return m_block_size; }
|
|
peer_request to_req(piece_block const& p);
|
|
|
|
void disconnect_all(error_code const& ec);
|
|
int disconnect_peers(int num);
|
|
|
|
// this is called wheh the torrent has completed
|
|
// the download. It will post an event, disconnect
|
|
// all seeds and let the tracker know we're finished.
|
|
void completed();
|
|
|
|
#if TORRENT_USE_I2P
|
|
void on_i2p_resolve(error_code const& ec, char const* dest);
|
|
#endif
|
|
|
|
// this is the asio callback that is called when a name
|
|
// lookup for a PEER is completed.
|
|
void on_peer_name_lookup(error_code const& e, tcp::resolver::iterator i
|
|
, peer_id pid);
|
|
|
|
// this is the asio callback that is called when a name
|
|
// lookup for a WEB SEED is completed.
|
|
void on_name_lookup(error_code const& e, tcp::resolver::iterator i
|
|
, web_seed_entry url, tcp::endpoint proxy);
|
|
|
|
// this is the asio callback that is called when a name
|
|
// lookup for a proxy for a web seed is completed.
|
|
void on_proxy_name_lookup(error_code const& e, tcp::resolver::iterator i
|
|
, web_seed_entry url);
|
|
|
|
// this is called when the torrent has finished. i.e.
|
|
// all the pieces we have not filtered have been downloaded.
|
|
// If no pieces are filtered, this is called first and then
|
|
// completed() is called immediately after it.
|
|
void finished();
|
|
|
|
// This is the opposite of finished. It is called if we used
|
|
// to be finished but enabled some files for download so that
|
|
// we wasn't finished anymore.
|
|
void resume_download();
|
|
|
|
void async_verify_piece(int piece_index, boost::function<void(int)> const&);
|
|
|
|
// this is called from the peer_connection
|
|
// each time a piece has failed the hash
|
|
// test
|
|
void piece_finished(int index, int passed_hash_check);
|
|
|
|
// piece_passed is called when a piece passes the hash check
|
|
// this will tell all peers that we just got his piece
|
|
// and also let the piece picker know that we have this piece
|
|
// so it wont pick it for download
|
|
void piece_passed(int index);
|
|
|
|
// piece_failed is called when a piece fails the hash check
|
|
void piece_failed(int index);
|
|
|
|
// this will restore the piece picker state for a piece
|
|
// by re marking all the requests to blocks in this piece
|
|
// that are still outstanding in peers' download queues.
|
|
// this is done when a piece fails
|
|
void restore_piece_state(int index);
|
|
|
|
void add_redundant_bytes(int b);
|
|
void add_failed_bytes(int b);
|
|
|
|
// this is true if we have all the pieces
|
|
bool is_seed() const
|
|
{
|
|
return valid_metadata()
|
|
&& (!m_picker
|
|
|| m_state == torrent_status::seeding
|
|
|| m_picker->num_have() == m_picker->num_pieces());
|
|
}
|
|
|
|
// this is true if we have all the pieces that we want
|
|
bool is_finished() const
|
|
{
|
|
if (is_seed()) return true;
|
|
return valid_metadata() && m_torrent_file->num_pieces()
|
|
- m_picker->num_have() - m_picker->num_filtered() == 0;
|
|
}
|
|
|
|
fs::path save_path() const;
|
|
alert_manager& alerts() const;
|
|
piece_picker& picker()
|
|
{
|
|
TORRENT_ASSERT(m_picker.get());
|
|
return *m_picker;
|
|
}
|
|
bool has_picker() const
|
|
{
|
|
return m_picker.get() != 0;
|
|
}
|
|
policy& get_policy() { return m_policy; }
|
|
piece_manager& filesystem();
|
|
torrent_info const& torrent_file() const
|
|
{ return *m_torrent_file; }
|
|
|
|
std::vector<announce_entry> const& trackers() const
|
|
{ return m_trackers; }
|
|
|
|
void replace_trackers(std::vector<announce_entry> const& urls);
|
|
void add_tracker(announce_entry const& url);
|
|
|
|
torrent_handle get_handle();
|
|
|
|
void write_resume_data(entry& rd) const;
|
|
void read_resume_data(lazy_entry const& rd);
|
|
|
|
// LOGGING
|
|
#if defined TORRENT_VERBOSE_LOGGING || defined TORRENT_LOGGING || defined TORRENT_ERROR_LOGGING
|
|
virtual void debug_log(const std::string& line);
|
|
#endif
|
|
|
|
// DEBUG
|
|
#ifdef TORRENT_DEBUG
|
|
void check_invariant() const;
|
|
#endif
|
|
|
|
// --------------------------------------------
|
|
// RESOURCE MANAGEMENT
|
|
|
|
void add_free_upload(int diff) { m_available_free_upload += diff; }
|
|
|
|
void set_peer_upload_limit(tcp::endpoint ip, int limit);
|
|
void set_peer_download_limit(tcp::endpoint ip, int limit);
|
|
|
|
void set_upload_limit(int limit);
|
|
int upload_limit() const;
|
|
void set_download_limit(int limit);
|
|
int download_limit() const;
|
|
|
|
void set_max_uploads(int limit);
|
|
int max_uploads() const { return m_max_uploads; }
|
|
void set_max_connections(int limit);
|
|
int max_connections() const { return m_max_connections; }
|
|
|
|
void move_storage(fs::path const& save_path);
|
|
|
|
// renames the file with the given index to the new name
|
|
// the name may include a directory path
|
|
// returns false on failure
|
|
bool rename_file(int index, std::string const& name);
|
|
|
|
// unless this returns true, new connections must wait
|
|
// with their initialization.
|
|
bool ready_for_connections() const
|
|
{ return m_connections_initialized; }
|
|
bool valid_metadata() const
|
|
{ return m_torrent_file->is_valid(); }
|
|
bool are_files_checked() const
|
|
{ return m_files_checked; }
|
|
|
|
// parses the info section from the given
|
|
// bencoded tree and moves the torrent
|
|
// to the checker thread for initial checking
|
|
// of the storage.
|
|
// a return value of false indicates an error
|
|
bool set_metadata(char const* metadata_buf, int metadata_size);
|
|
|
|
int sequence_number() const { return m_sequence_number; }
|
|
|
|
bool seed_mode() const { return m_seed_mode; }
|
|
void leave_seed_mode(bool seed)
|
|
{
|
|
if (!m_seed_mode) return;
|
|
m_seed_mode = false;
|
|
// seed is false if we turned out not
|
|
// to be a seed after all
|
|
if (!seed) force_recheck();
|
|
m_num_verified = 0;
|
|
m_verified.free();
|
|
}
|
|
bool all_verified() const
|
|
{ return m_num_verified == m_torrent_file->num_pieces(); }
|
|
bool verified_piece(int piece) const
|
|
{
|
|
TORRENT_ASSERT(piece < int(m_verified.size()));
|
|
TORRENT_ASSERT(piece >= 0);
|
|
return m_verified.get_bit(piece);
|
|
}
|
|
void verified(int piece)
|
|
{
|
|
TORRENT_ASSERT(piece < int(m_verified.size()));
|
|
TORRENT_ASSERT(piece >= 0);
|
|
TORRENT_ASSERT(m_verified.get_bit(piece) == false);
|
|
++m_num_verified;
|
|
m_verified.set_bit(piece);
|
|
}
|
|
|
|
bool add_merkle_nodes(std::map<int, sha1_hash> const& n, int piece);
|
|
|
|
private:
|
|
|
|
void on_files_deleted(int ret, disk_io_job const& j);
|
|
void on_files_released(int ret, disk_io_job const& j);
|
|
void on_torrent_paused(int ret, disk_io_job const& j);
|
|
void on_storage_moved(int ret, disk_io_job const& j);
|
|
void on_save_resume_data(int ret, disk_io_job const& j);
|
|
void on_file_renamed(int ret, disk_io_job const& j);
|
|
|
|
void on_piece_verified(int ret, disk_io_job const& j
|
|
, boost::function<void(int)> f);
|
|
|
|
int prioritize_tracker(int tracker_index);
|
|
int deprioritize_tracker(int tracker_index);
|
|
|
|
void on_country_lookup(error_code const& error, tcp::resolver::iterator i
|
|
, boost::intrusive_ptr<peer_connection> p) const;
|
|
bool request_bandwidth_from_session(int channel) const;
|
|
|
|
void update_peer_interest(bool was_finished);
|
|
void prioritize_udp_trackers();
|
|
|
|
void queue_torrent_check();
|
|
void dequeue_torrent_check();
|
|
|
|
policy m_policy;
|
|
|
|
// total time we've been available on this torrent
|
|
// does not count when the torrent is stopped or paused
|
|
time_duration m_active_time;
|
|
|
|
// total time we've been available as a seed on this torrent
|
|
// does not count when the torrent is stopped or paused
|
|
time_duration m_seeding_time;
|
|
|
|
// all time totals of uploaded and downloaded payload
|
|
// stored in resume data
|
|
size_type m_total_uploaded;
|
|
size_type m_total_downloaded;
|
|
|
|
// if this torrent is running, this was the time
|
|
// when it was started. This is used to have a
|
|
// bias towards keeping seeding torrents that
|
|
// recently was started, to avoid oscillation
|
|
ptime m_started;
|
|
|
|
// the last time we initiated a scrape request to
|
|
// one of the trackers in this torrent
|
|
ptime m_last_scrape;
|
|
|
|
// the time when we switched to upload mode
|
|
ptime m_upload_mode_time;
|
|
|
|
boost::intrusive_ptr<torrent_info> m_torrent_file;
|
|
|
|
void parse_response(const entry& e, std::vector<peer_entry>& peer_list);
|
|
|
|
// if this pointer is 0, the torrent is in
|
|
// a state where the metadata hasn't been
|
|
// received yet.
|
|
// the piece_manager keeps the torrent object
|
|
// alive by holding a shared_ptr to it and
|
|
// the torrent keeps the piece manager alive
|
|
// with this intrusive_ptr. This cycle is
|
|
// broken when torrent::abort() is called
|
|
// Then the torrent releases the piece_manager
|
|
// and when the piece_manager is complete with all
|
|
// outstanding disk io jobs (that keeps
|
|
// the piece_manager alive) it will destruct
|
|
// and release the torrent file. The reason for
|
|
// this is that the torrent_info is used by
|
|
// the piece_manager, and stored in the
|
|
// torrent, so the torrent cannot destruct
|
|
// before the piece_manager.
|
|
boost::intrusive_ptr<piece_manager> m_owning_storage;
|
|
|
|
// this is a weak (non owninig) pointer to
|
|
// the piece_manager. This is used after the torrent
|
|
// has been aborted, and it can no longer own
|
|
// the object.
|
|
piece_manager* m_storage;
|
|
|
|
#ifdef TORRENT_DEBUG
|
|
public:
|
|
#endif
|
|
std::set<peer_connection*> m_connections;
|
|
#ifdef TORRENT_DEBUG
|
|
private:
|
|
#endif
|
|
|
|
// The list of web seeds in this torrent. Seeds
|
|
// with fatal errors are removed from the set
|
|
std::set<web_seed_entry> m_web_seeds;
|
|
|
|
// a list of web seeds that have failed and are
|
|
// waiting to be retried
|
|
std::map<web_seed_entry, ptime> m_web_seeds_next_retry;
|
|
|
|
// urls of the web seeds that we are currently
|
|
// resolving the address for
|
|
std::set<web_seed_entry> m_resolving_web_seeds;
|
|
|
|
#ifndef TORRENT_DISABLE_EXTENSIONS
|
|
typedef std::list<boost::shared_ptr<torrent_plugin> > extension_list_t;
|
|
extension_list_t m_extensions;
|
|
#endif
|
|
|
|
// used to resolve the names of web seeds
|
|
mutable tcp::resolver m_host_resolver;
|
|
|
|
// this announce timer is used both
|
|
// by Local service discovery and
|
|
// by the DHT.
|
|
deadline_timer m_lsd_announce_timer;
|
|
|
|
// used for tracker announces
|
|
deadline_timer m_tracker_timer;
|
|
|
|
void update_tracker_timer();
|
|
|
|
static void on_tracker_announce_disp(boost::weak_ptr<torrent> p
|
|
, error_code const& e);
|
|
|
|
void on_tracker_announce();
|
|
|
|
static void on_lsd_announce_disp(boost::weak_ptr<torrent> p
|
|
, error_code const& e);
|
|
|
|
// this is called once every 5 minutes for torrents
|
|
// that are not private
|
|
void on_lsd_announce();
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
static void on_dht_announce_response_disp(boost::weak_ptr<torrent> t
|
|
, std::vector<tcp::endpoint> const& peers);
|
|
void on_dht_announce_response(std::vector<tcp::endpoint> const& peers);
|
|
bool should_announce_dht() const;
|
|
|
|
// the time when the DHT was last announced of our
|
|
// presence on this torrent
|
|
ptime m_last_dht_announce;
|
|
#endif
|
|
|
|
// this is the upload and download statistics for the whole torrent.
|
|
// it's updated from all its peers once every second.
|
|
libtorrent::stat m_stat;
|
|
|
|
// -----------------------------
|
|
|
|
// a back reference to the session
|
|
// this torrent belongs to.
|
|
aux::session_impl& m_ses;
|
|
|
|
std::vector<boost::uint8_t> m_file_priority;
|
|
|
|
// this vector contains the number of bytes completely
|
|
// downloaded (as in passed-hash-check) in each file.
|
|
// this lets us trigger on individual files completing
|
|
std::vector<size_type> m_file_progress;
|
|
|
|
boost::scoped_ptr<piece_picker> m_picker;
|
|
|
|
std::vector<announce_entry> m_trackers;
|
|
// this is an index into m_trackers
|
|
|
|
struct time_critical_piece
|
|
{
|
|
// when this piece was first requested
|
|
ptime first_requested;
|
|
// when this piece was last requested
|
|
ptime last_requested;
|
|
// by what time we want this piece
|
|
ptime deadline;
|
|
// 1 = send alert with piece data when available
|
|
int flags;
|
|
// how many peers it's been requested from
|
|
int peers;
|
|
// the piece index
|
|
int piece;
|
|
bool operator<(time_critical_piece const& rhs) const
|
|
{ return deadline < rhs.deadline; }
|
|
};
|
|
|
|
void remove_time_critical_piece(int piece, bool finished = false);
|
|
void remove_time_critical_pieces(std::vector<int> const& priority);
|
|
void request_time_critical_pieces();
|
|
|
|
// this list is sorted by time_critical_piece::deadline
|
|
std::list<time_critical_piece> m_time_critical_pieces;
|
|
|
|
// the average time it takes to download one time critical piece
|
|
time_duration m_average_piece_time;
|
|
// the average piece download time deviation
|
|
time_duration m_piece_time_deviation;
|
|
|
|
// the number of bytes that has been
|
|
// downloaded that failed the hash-test
|
|
size_type m_total_failed_bytes;
|
|
size_type m_total_redundant_bytes;
|
|
|
|
// the number of bytes of padding files
|
|
int m_padding;
|
|
|
|
std::string m_username;
|
|
std::string m_password;
|
|
|
|
// the network interface all outgoing connections
|
|
// are opened through
|
|
tcp::endpoint m_net_interface;
|
|
|
|
fs::path m_save_path;
|
|
|
|
// each bit represents a piece. a set bit means
|
|
// the piece has had its hash verified. This
|
|
// is only used in seed mode (when m_seed_mode
|
|
// is true)
|
|
bitfield m_verified;
|
|
// m_num_verified = m_verified.count()
|
|
int m_num_verified;
|
|
|
|
// free download we have got that hasn't
|
|
// been distributed yet.
|
|
size_type m_available_free_upload;
|
|
|
|
// determines the storage state for this torrent.
|
|
storage_mode_t m_storage_mode;
|
|
|
|
// the state of this torrent (queued, checking, downloading, etc.)
|
|
torrent_status::state_t m_state;
|
|
|
|
// set if there's an error on this torrent
|
|
error_code m_error;
|
|
// if the error ocurred on a file, this is the file
|
|
std::string m_error_file;
|
|
|
|
// used if there is any resume data
|
|
std::vector<char> m_resume_data;
|
|
lazy_entry m_resume_entry;
|
|
|
|
// if the torrent is started without metadata, it may
|
|
// still be given a name until the metadata is received
|
|
// once the metadata is received this field will no
|
|
// longer be used and will be reset
|
|
boost::scoped_ptr<std::string> m_name;
|
|
|
|
#ifndef TORRENT_DISABLE_ENCRYPTION
|
|
// this is SHA1("req2" + info-hash), used for
|
|
// encrypted hand shakes
|
|
sha1_hash m_obfuscated_hash;
|
|
#endif
|
|
session_settings const& m_settings;
|
|
|
|
storage_constructor_type m_storage_constructor;
|
|
|
|
int m_progress_ppm;
|
|
|
|
// the upload/download ratio that each peer
|
|
// tries to maintain.
|
|
// 0 is infinite
|
|
float m_ratio;
|
|
|
|
// the maximum number of uploads for this torrent
|
|
int m_max_uploads;
|
|
|
|
// the number of unchoked peers in this torrent
|
|
int m_num_uploads;
|
|
|
|
// the maximum number of connections for this torrent
|
|
int m_max_connections;
|
|
|
|
// the size of a request block
|
|
// each piece is divided into these
|
|
// blocks when requested
|
|
int m_block_size;
|
|
|
|
// -----------------------------
|
|
// DATA FROM TRACKER RESPONSE
|
|
|
|
// the scrape data from the tracker response, this
|
|
// is optional and may be -1.
|
|
int m_complete;
|
|
int m_incomplete;
|
|
|
|
#ifdef TORRENT_DEBUG
|
|
// this is the amount downloaded when this torrent
|
|
// is started. i.e.
|
|
// total_done - m_initial_done <= total_payload_download
|
|
size_type m_initial_done;
|
|
#endif
|
|
// this is the deficit counter in the Deficit Round Robin
|
|
// used to determine which torrent gets the next
|
|
// connection attempt. See:
|
|
// http://www.ecs.umass.edu/ece/wolf/courses/ECE697J/papers/DRR.pdf
|
|
// The quanta assigned to each torrent depends on the torrents
|
|
// priority, whether it's seed and the number of connected
|
|
// peers it has. This has the effect that some torrents
|
|
// will have more connection attempts than other. Each
|
|
// connection attempt costs 100 points from the deficit
|
|
// counter. points are deducted in try_connect_peer and
|
|
// increased in give_connect_points. Outside of the
|
|
// torrent object, these points are called connect_points.
|
|
int m_deficit_counter;
|
|
|
|
// the sequence number for this torrent, this is a
|
|
// monotonically increasing number for each added torrent
|
|
boost::int16_t m_sequence_number;
|
|
|
|
// the index to the last tracker that worked
|
|
boost::int8_t m_last_working_tracker;
|
|
|
|
// the number of connection attempts that has
|
|
// failed in a row, this is currently used to
|
|
// determine the timeout until next try.
|
|
boost::int8_t m_failed_trackers;
|
|
|
|
// this is a counter that is decreased every
|
|
// second, and when it reaches 0, the policy::pulse()
|
|
// is called and the time scaler is reset to 10.
|
|
boost::int8_t m_time_scaler;
|
|
|
|
// this is the priority of the torrent. The higher
|
|
// the value is, the more bandwidth is assigned to
|
|
// the torrent's peers
|
|
boost::uint8_t m_priority;
|
|
|
|
// is set to true when the torrent has
|
|
// been aborted.
|
|
bool m_abort:1;
|
|
|
|
// is true if this torrent has been paused
|
|
bool m_paused:1;
|
|
|
|
// set to true when this torrent may not download anything
|
|
bool m_upload_mode:1;
|
|
|
|
// if this is true, libtorrent may pause and resume
|
|
// this torrent depending on queuing rules. Torrents
|
|
// started with auto_managed flag set may be added in
|
|
// a paused state in case there are no available
|
|
// slots.
|
|
bool m_auto_managed:1;
|
|
|
|
#ifndef TORRENT_DISABLE_RESOLVE_COUNTRIES
|
|
// this is true while there is a country
|
|
// resolution in progress. To avoid flodding
|
|
// the DNS request queue, only one ip is resolved
|
|
// at a time.
|
|
mutable bool m_resolving_country:1;
|
|
|
|
// this is true if the user has enabled
|
|
// country resolution in this torrent
|
|
bool m_resolve_countries:1;
|
|
#endif
|
|
|
|
// in case the piece picker hasn't been constructed
|
|
// when this settings is set, this variable will keep
|
|
// its value until the piece picker is created
|
|
bool m_sequential_download:1;
|
|
|
|
// is false by default and set to
|
|
// true when the first tracker reponse
|
|
// is received
|
|
bool m_got_tracker_response:1;
|
|
|
|
// this is set to false as long as the connections
|
|
// of this torrent hasn't been initialized. If we
|
|
// have metadata from the start, connections are
|
|
// initialized immediately, if we didn't have metadata,
|
|
// they are initialized right after files_checked().
|
|
// valid_resume_data() will return false as long as
|
|
// the connections aren't initialized, to avoid
|
|
// them from altering the piece-picker before it
|
|
// has been initialized with files_checked().
|
|
bool m_connections_initialized:1;
|
|
|
|
// if this is true, we're currently super seeding this
|
|
// torrent.
|
|
bool m_super_seeding:1;
|
|
|
|
// is set to true every time there is an incoming
|
|
// connection to this torrent
|
|
bool m_has_incoming:1;
|
|
|
|
// this is set to true when the files are checked
|
|
// before the files are checked, we don't try to
|
|
// connect to peers
|
|
bool m_files_checked:1;
|
|
|
|
// this is true if the torrent has been added to
|
|
// checking queue in the session
|
|
bool m_queued_for_checking:1;
|
|
|
|
// this is true while tracker announcing is enabled
|
|
// is is disabled while paused and checking files
|
|
bool m_announcing:1;
|
|
|
|
// this is true while the tracker deadline timer
|
|
// is in use. i.e. one or more trackers are waiting
|
|
// for a reannounce
|
|
bool m_waiting_tracker:1;
|
|
|
|
// this means we haven't verified the file content
|
|
// of the files we're seeding. the m_verified bitfield
|
|
// indicates which pieces have been verified and which
|
|
// haven't
|
|
bool m_seed_mode:1;
|
|
|
|
// this is set when we don't want to load seed_mode,
|
|
// paused or auto_managed from the resume data
|
|
bool m_override_resume_data:1;
|
|
};
|
|
|
|
inline ptime torrent::next_announce() const
|
|
{
|
|
return m_waiting_tracker?m_tracker_timer.expires_at():min_time();
|
|
}
|
|
|
|
inline void torrent::force_tracker_request()
|
|
{
|
|
if (is_paused()) return;
|
|
ptime now = time_now();
|
|
for (std::vector<announce_entry>::iterator i = m_trackers.begin()
|
|
, end(m_trackers.end()); i != end; ++i)
|
|
i->next_announce = now;
|
|
update_tracker_timer();
|
|
}
|
|
|
|
inline void torrent::force_tracker_request(ptime t)
|
|
{
|
|
if (is_paused()) return;
|
|
for (std::vector<announce_entry>::iterator i = m_trackers.begin()
|
|
, end(m_trackers.end()); i != end; ++i)
|
|
i->next_announce = t;
|
|
update_tracker_timer();
|
|
}
|
|
|
|
inline void torrent::set_tracker_login(
|
|
std::string const& name
|
|
, std::string const& pw)
|
|
{
|
|
m_username = name;
|
|
m_password = pw;
|
|
}
|
|
|
|
}
|
|
|
|
#endif // TORRENT_TORRENT_HPP_INCLUDED
|
|
|