forked from premiere/premiere-libtorrent
380 lines
15 KiB
C++
380 lines
15 KiB
C++
/*
|
|
|
|
Copyright (c) 2006-2018, 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_SESSION_HPP_INCLUDED
|
|
#define TORRENT_SESSION_HPP_INCLUDED
|
|
|
|
#include <thread>
|
|
|
|
#include "libtorrent/config.hpp"
|
|
#include "libtorrent/io_service.hpp"
|
|
#include "libtorrent/settings_pack.hpp"
|
|
#include "libtorrent/session_handle.hpp"
|
|
#include "libtorrent/kademlia/dht_settings.hpp"
|
|
#include "libtorrent/kademlia/dht_state.hpp"
|
|
#include "libtorrent/kademlia/dht_storage.hpp"
|
|
|
|
#if TORRENT_ABI_VERSION == 1
|
|
#include "libtorrent/fingerprint.hpp"
|
|
#include <cstdio> // for snprintf
|
|
#endif
|
|
|
|
namespace libtorrent {
|
|
|
|
struct plugin;
|
|
|
|
// The default values of the session settings are set for a regular
|
|
// bittorrent client running on a desktop system. There are functions that
|
|
// can set the session settings to pre set settings for other environments.
|
|
// These can be used for the basis, and should be tweaked to fit your needs
|
|
// better.
|
|
//
|
|
// ``min_memory_usage`` returns settings that will use the minimal amount of
|
|
// RAM, at the potential expense of upload and download performance. It
|
|
// adjusts the socket buffer sizes, disables the disk cache, lowers the send
|
|
// buffer watermarks so that each connection only has at most one block in
|
|
// use at any one time. It lowers the outstanding blocks send to the disk
|
|
// I/O thread so that connections only have one block waiting to be flushed
|
|
// to disk at any given time. It lowers the max number of peers in the peer
|
|
// list for torrents. It performs multiple smaller reads when it hashes
|
|
// pieces, instead of reading it all into memory before hashing.
|
|
//
|
|
// This configuration is intended to be the starting point for embedded
|
|
// devices. It will significantly reduce memory usage.
|
|
//
|
|
// ``high_performance_seed`` returns settings optimized for a seed box,
|
|
// serving many peers and that doesn't do any downloading. It has a 128 MB
|
|
// disk cache and has a limit of 400 files in its file pool. It support fast
|
|
// upload rates by allowing large send buffers.
|
|
TORRENT_EXPORT settings_pack min_memory_usage();
|
|
TORRENT_EXPORT settings_pack high_performance_seed();
|
|
#if TORRENT_ABI_VERSION == 1
|
|
TORRENT_DEPRECATED
|
|
inline void min_memory_usage(settings_pack& set)
|
|
{ set = min_memory_usage(); }
|
|
TORRENT_DEPRECATED
|
|
inline void high_performance_seed(settings_pack& set)
|
|
{ set = high_performance_seed(); }
|
|
#endif
|
|
|
|
namespace aux {
|
|
|
|
struct session_impl;
|
|
}
|
|
|
|
struct disk_interface;
|
|
struct counters;
|
|
|
|
// this is a holder for the internal session implementation object. Once the
|
|
// session destruction is explicitly initiated, this holder is used to
|
|
// synchronize the completion of the shutdown. The lifetime of this object
|
|
// may outlive session, causing the session destructor to not block. The
|
|
// session_proxy destructor will block however, until the underlying session
|
|
// is done shutting down.
|
|
class TORRENT_EXPORT session_proxy
|
|
{
|
|
friend class session;
|
|
public:
|
|
// default constructor, does not refer to any session
|
|
// implementation object.
|
|
session_proxy();
|
|
~session_proxy();
|
|
session_proxy(session_proxy const&);
|
|
session_proxy& operator=(session_proxy const&);
|
|
session_proxy(session_proxy&&) noexcept;
|
|
session_proxy& operator=(session_proxy&&) noexcept;
|
|
private:
|
|
session_proxy(
|
|
std::shared_ptr<io_service> ios
|
|
, std::shared_ptr<std::thread> t
|
|
, std::shared_ptr<aux::session_impl> impl);
|
|
|
|
std::shared_ptr<io_service> m_io_service;
|
|
std::shared_ptr<std::thread> m_thread;
|
|
std::shared_ptr<aux::session_impl> m_impl;
|
|
};
|
|
|
|
// The session_params is a parameters pack for configuring the session
|
|
// before it's started.
|
|
struct TORRENT_EXPORT session_params
|
|
{
|
|
// This constructor can be used to start with the default plugins
|
|
// (ut_metadata, ut_pex and smart_ban). The default values in the
|
|
// settings is to start the default features like upnp, NAT-PMP,
|
|
// and dht for example.
|
|
explicit session_params(settings_pack&& sp);
|
|
explicit session_params(settings_pack const& sp);
|
|
session_params();
|
|
|
|
// This constructor helps to configure the set of initial plugins
|
|
// to be added to the session before it's started.
|
|
session_params(settings_pack&& sp
|
|
, std::vector<std::shared_ptr<plugin>> exts);
|
|
session_params(settings_pack const& sp
|
|
, std::vector<std::shared_ptr<plugin>> exts);
|
|
|
|
session_params(session_params const&) = default;
|
|
session_params(session_params&&) = default;
|
|
session_params& operator=(session_params const&) = default;
|
|
session_params& operator=(session_params&&) = default;
|
|
|
|
settings_pack settings;
|
|
|
|
std::vector<std::shared_ptr<plugin>> extensions;
|
|
|
|
dht::dht_settings dht_settings;
|
|
|
|
dht::dht_state dht_state;
|
|
|
|
dht::dht_storage_constructor_type dht_storage_constructor;
|
|
};
|
|
|
|
// This function helps to construct a ``session_params`` from a
|
|
// bencoded data generated by ``session_handle::save_state``
|
|
TORRENT_EXPORT session_params read_session_params(bdecode_node const& e
|
|
, save_state_flags_t flags = save_state_flags_t::all());
|
|
|
|
// The session holds all state that spans multiple torrents. Among other
|
|
// things it runs the network loop and manages all torrents. Once it's
|
|
// created, the session object will spawn the main thread that will do all
|
|
// the work. The main thread will be idle as long it doesn't have any
|
|
// torrents to participate in.
|
|
//
|
|
// You have some control over session configuration through the
|
|
// ``session_handle::apply_settings()`` member function. To change one or more
|
|
// configuration options, create a settings_pack. object and fill it with
|
|
// the settings to be set and pass it in to ``session::apply_settings()``.
|
|
//
|
|
// see apply_settings().
|
|
class TORRENT_EXPORT session : public session_handle
|
|
{
|
|
public:
|
|
|
|
// Constructs the session objects which acts as the container of torrents.
|
|
// In order to avoid a race condition between starting the session and
|
|
// configuring it, you can pass in a session_params object. Its settings
|
|
// will take effect before the session starts up.
|
|
explicit session(session_params const& params)
|
|
{ start(session_params(params), nullptr); }
|
|
explicit session(session_params&& params)
|
|
{ start(std::move(params), nullptr); }
|
|
session()
|
|
{
|
|
session_params params;
|
|
start(std::move(params), nullptr);
|
|
}
|
|
|
|
// Overload of the constructor that takes an external io_service to run
|
|
// the session object on. This is primarily useful for tests that may want
|
|
// to run multiple sessions on a single io_service, or low resource
|
|
// systems where additional threads are expensive and sharing an
|
|
// io_service with other events is fine.
|
|
//
|
|
// .. warning::
|
|
// The session object does not cleanly terminate with an external
|
|
// ``io_service``. The ``io_service::run()`` call _must_ have returned
|
|
// before it's safe to destruct the session. Which means you *MUST*
|
|
// call session::abort() and save the session_proxy first, then
|
|
// destruct the session object, then sync with the io_service, then
|
|
// destruct the session_proxy object.
|
|
session(session_params&& params, io_service& ios)
|
|
{ start(std::move(params), &ios); }
|
|
session(session_params const& params, io_service& ios)
|
|
{ start(session_params(params), &ios); }
|
|
|
|
// Constructs the session objects which acts as the container of torrents.
|
|
// It provides configuration options across torrents (such as rate limits,
|
|
// disk cache, ip filter etc.). In order to avoid a race condition between
|
|
// starting the session and configuring it, you can pass in a
|
|
// settings_pack object. Its settings will take effect before the session
|
|
// starts up.
|
|
//
|
|
// The ``flags`` parameter can be used to start default features (UPnP &
|
|
// NAT-PMP) and default plugins (ut_metadata, ut_pex and smart_ban). The
|
|
// default is to start those features. If you do not want them to start,
|
|
// pass 0 as the flags parameter.
|
|
session(settings_pack&& pack
|
|
, session_flags_t const flags = add_default_plugins)
|
|
{ start(flags, std::move(pack), nullptr); }
|
|
session(settings_pack const& pack
|
|
, session_flags_t const flags = add_default_plugins)
|
|
{ start(flags, settings_pack(pack), nullptr); }
|
|
|
|
// movable
|
|
session(session&&) = default;
|
|
session& operator=(session&&) = default;
|
|
|
|
// non-copyable
|
|
session(session const&) = delete;
|
|
session& operator=(session const&) = delete;
|
|
|
|
// overload of the constructor that takes an external io_service to run
|
|
// the session object on. This is primarily useful for tests that may want
|
|
// to run multiple sessions on a single io_service, or low resource
|
|
// systems where additional threads are expensive and sharing an
|
|
// io_service with other events is fine.
|
|
//
|
|
// .. warning::
|
|
// The session object does not cleanly terminate with an external
|
|
// ``io_service``. The ``io_service::run()`` call _must_ have returned
|
|
// before it's safe to destruct the session. Which means you *MUST*
|
|
// call session::abort() and save the session_proxy first, then
|
|
// destruct the session object, then sync with the io_service, then
|
|
// destruct the session_proxy object.
|
|
session(settings_pack&& pack
|
|
, io_service& ios
|
|
, session_flags_t const flags = add_default_plugins)
|
|
{ start(flags, std::move(pack), &ios); }
|
|
session(settings_pack const& pack
|
|
, io_service& ios
|
|
, session_flags_t const flags = add_default_plugins)
|
|
{ start(flags, settings_pack(pack), &ios); }
|
|
|
|
#if TORRENT_ABI_VERSION == 1
|
|
#ifdef __GNUC__
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
|
|
#endif
|
|
#ifdef __clang__
|
|
#pragma clang diagnostic push
|
|
#pragma clang diagnostic ignored "-Wdeprecated-declarations"
|
|
#endif
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push, 1)
|
|
#pragma warning(disable: 4996)
|
|
#endif
|
|
TORRENT_DEPRECATED
|
|
session(fingerprint const& print
|
|
, session_flags_t const flags = start_default_features | add_default_plugins
|
|
, alert_category_t const alert_mask = alert::error_notification)
|
|
{
|
|
settings_pack pack;
|
|
pack.set_int(settings_pack::alert_mask, int(alert_mask));
|
|
pack.set_str(settings_pack::peer_fingerprint, print.to_string());
|
|
if (!(flags & start_default_features))
|
|
{
|
|
pack.set_bool(settings_pack::enable_upnp, false);
|
|
pack.set_bool(settings_pack::enable_natpmp, false);
|
|
pack.set_bool(settings_pack::enable_lsd, false);
|
|
pack.set_bool(settings_pack::enable_dht, false);
|
|
}
|
|
|
|
start(flags, std::move(pack), nullptr);
|
|
}
|
|
|
|
TORRENT_DEPRECATED
|
|
session(fingerprint const& print
|
|
, std::pair<int, int> listen_port_range
|
|
, char const* listen_interface = "0.0.0.0"
|
|
, session_flags_t const flags = start_default_features | add_default_plugins
|
|
, alert_category_t const alert_mask = alert::error_notification)
|
|
{
|
|
TORRENT_ASSERT(listen_port_range.first > 0);
|
|
TORRENT_ASSERT(listen_port_range.first <= listen_port_range.second);
|
|
|
|
settings_pack pack;
|
|
pack.set_int(settings_pack::alert_mask, int(alert_mask));
|
|
pack.set_int(settings_pack::max_retry_port_bind, listen_port_range.second - listen_port_range.first);
|
|
pack.set_str(settings_pack::peer_fingerprint, print.to_string());
|
|
char if_string[100];
|
|
|
|
if (listen_interface == nullptr) listen_interface = "0.0.0.0";
|
|
std::snprintf(if_string, sizeof(if_string), "%s:%d", listen_interface, listen_port_range.first);
|
|
pack.set_str(settings_pack::listen_interfaces, if_string);
|
|
|
|
if (!(flags & start_default_features))
|
|
{
|
|
pack.set_bool(settings_pack::enable_upnp, false);
|
|
pack.set_bool(settings_pack::enable_natpmp, false);
|
|
pack.set_bool(settings_pack::enable_lsd, false);
|
|
pack.set_bool(settings_pack::enable_dht, false);
|
|
}
|
|
start(flags, std::move(pack), nullptr);
|
|
}
|
|
#ifdef __GNUC__
|
|
#pragma GCC diagnostic pop
|
|
#endif
|
|
#ifdef __clang__
|
|
#pragma clang diagnostic pop
|
|
#endif
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
#endif // TORRENT_ABI_VERSION
|
|
|
|
// The destructor of session will notify all trackers that our torrents
|
|
// have been shut down. If some trackers are down, they will time out.
|
|
// All this before the destructor of session returns. So, it's advised
|
|
// that any kind of interface (such as windows) are closed before
|
|
// destructing the session object. Because it can take a few second for
|
|
// it to finish. The timeout can be set with apply_settings().
|
|
~session();
|
|
|
|
// In case you want to destruct the session asynchronously, you can
|
|
// request a session destruction proxy. If you don't do this, the
|
|
// destructor of the session object will block while the trackers are
|
|
// contacted. If you keep one ``session_proxy`` to the session when
|
|
// destructing it, the destructor will not block, but start to close down
|
|
// the session, the destructor of the proxy will then synchronize the
|
|
// threads. So, the destruction of the session is performed from the
|
|
// ``session`` destructor call until the ``session_proxy`` destructor
|
|
// call. The ``session_proxy`` does not have any operations on it (since
|
|
// the session is being closed down, no operations are allowed on it).
|
|
// The only valid operation is calling the destructor::
|
|
//
|
|
// class session_proxy
|
|
// {
|
|
// public:
|
|
// session_proxy();
|
|
// ~session_proxy()
|
|
// };
|
|
session_proxy abort();
|
|
|
|
private:
|
|
|
|
void start(session_params&& params, io_service* ios);
|
|
void start(session_flags_t flags, settings_pack&& sp, io_service* ios);
|
|
|
|
void start(session_params const& params, io_service* ios) = delete;
|
|
void start(session_flags_t flags, settings_pack const& sp, io_service* ios) = delete;
|
|
|
|
// data shared between the main thread
|
|
// and the working thread
|
|
std::shared_ptr<io_service> m_io_service;
|
|
std::shared_ptr<std::thread> m_thread;
|
|
std::shared_ptr<aux::session_impl> m_impl;
|
|
};
|
|
|
|
}
|
|
|
|
#endif // TORRENT_SESSION_HPP_INCLUDED
|