Plugins
Author: | Arvid Norberg, arvid@rasterbar.com |
---|---|
Version: | 1.0.0 |
Table of contents
libtorrent has a plugin interface for implementing extensions to the protocol. These can be general extensions for transferring metadata or peer exchange extensions, or it could be used to provide a way to customize the protocol to fit a particular (closed) network.
In short, the plugin interface makes it possible to:
- register extension messages (sent in the extension handshake), see extensions.
- add data and parse data from the extension handshake.
- send extension messages and standard bittorrent messages.
- override or block the handling of standard bittorrent messages.
- save and restore state via the session state
- see all alerts that are posted
a word of caution
Writing your own plugin is a very easy way to introduce serious bugs such as dead locks and race conditions. Since a plugin has access to internal structures it is also quite easy to sabotage libtorrent's operation.
All the callbacks in this interface are called with the main libtorrent thread mutex locked. And they are always called from the libtorrent network thread. In case portions of your plugin are called from other threads, typically the main thread, you cannot use any of the member functions on the internal structures in libtorrent, since those require the mutex to be locked. Futhermore, you would also need to have a mutex on your own shared data within the plugin, to make sure it is not accessed at the same time from the libtorrent thread (through a callback). See boost thread's mutex. If you need to send out a message from another thread, it is advised to use an internal queue, and do the actual sending in tick().
Since the plugin interface gives you easy access to internal structures, it is not supported as a stable API. Plugins should be considered spcific to a specific version of libtorrent. Although, in practice the internals mostly don't change that dramatically.
plugin-interface
The plugin interface consists of three base classes that the plugin may implement. These are called plugin, torrent_plugin and peer_plugin. They are found in the <libtorrent/extensions.hpp> header.
These plugins are instantiated for each session, torrent and possibly each peer, respectively.
For plugins that only need per torrent state, it is enough to only implement torrent_plugin and pass a constructor function or function object to session::add_extension() or torrent_handle::add_extension() (if the torrent has already been started and you want to hook in the extension at run-time).
The signature of the function is:
boost::shared_ptr<torrent_plugin> (*)(torrent*, void*);
The first argument is the internal torrent object, the second argument is the userdata passed to session::add_torrent() or torrent_handle::add_extension().
The function should return a boost::shared_ptr<torrent_plugin> which may or may not be 0. If it is a null pointer, the extension is simply ignored for this torrent. If it is a valid pointer (to a class inheriting torrent_plugin), it will be associated with this torrent and callbacks will be made on torrent events.
For more elaborate plugins which require session wide state, you would implement plugin, construct an object (in a boost::shared_ptr) and pass it in to session::add_extension().
custom alerts
Since plugins are running within internal libtorrent threads, one convenient way to communicate with the client is to post custom alerts.
The expected interface of any alert, apart from deriving from the alert base class, looks like this:
const static int alert_type = <unique alert ID>; virtual int type() const { return alert_type; } virtual std::string message() const; virtual std::auto_ptr<alert> clone() const { return std::auto_ptr<alert>(new name(*this)); } const static int static_category = <bitmask of alert::category_t flags>; virtual int category() const { return static_category; } virtual char const* what() const { return <string literal of the name of this alert>; }
The alert_type is used for the type-checking in alert_cast. It must not collide with any other alert. The built-in alerts in libtorrent will not use alert type IDs greater than user_alert_id. When defining your own alert, make sure it's greater than this constant.
type() is the run-time equivalence of the alert_type.
The message() virtual function is expected to construct a useful string representation of the alert and the event or data it represents. Something convenient to put in a log file for instance.
clone() is used internally to copy alerts. The suggested implementation of simply allocating a new instance as a copy of *this is all that's expected.
The static category is required for checking wether or not the category for a specific alert is enabled or not, without instantiating the alert. The category virtual function is the run-time equivalence.
The what() virtual function may simply be a string literal of the class name of your alert.
For more information, see the alert section.
plugin
Declared in "libtorrent/extensions.hpp"
this is the base class for a session plugin. One primary feature is that it is notified of all torrents that are added to the session, and can add its own torrent_plugins.
struct plugin { virtual boost::shared_ptr<torrent_plugin> new_torrent (torrent*, void*); virtual void added (aux::session_impl*); virtual void on_alert (alert const*); virtual void on_tick (); virtual bool on_optimistic_unchoke (std::vector<policy::peer*>& /* peers */); virtual void save_state (entry&) const; virtual void load_state (lazy_entry const&); };
new_torrent()
virtual boost::shared_ptr<torrent_plugin> new_torrent (torrent*, void*);
this is called by the session every time a new torrent is added. The torrent* points to the internal torrent object created for the new torrent. The void* is the userdata pointer as passed in via add_torrent_params.
If the plugin returns a torrent_plugin instance, it will be added to the new torrent. Otherwise, return an empty shared_ptr to a torrent_plugin (the default).
on_alert()
virtual void on_alert (alert const*);
called when an alert is posted alerts that are filtered are not posted
on_optimistic_unchoke()
virtual bool on_optimistic_unchoke (std::vector<policy::peer*>& /* peers */);
called when choosing peers to optimisticly unchoke peer's will be unchoked in the order they appear in the given vector which is initiallity sorted by when they were last optimistically unchoked. if the plugin returns true then the ordering provided will be used and no other plugin will be allowed to change it.
torrent_plugin
Declared in "libtorrent/extensions.hpp"
Torrent plugins are associated with a single torrent and have a number of functions called at certain events. Many of its functions have the ability to change or override the default libtorrent behavior.
struct torrent_plugin { virtual boost::shared_ptr<peer_plugin> new_connection (peer_connection*); virtual void on_piece_pass (int /*index*/); virtual void on_piece_failed (int /*index*/); virtual void tick (); virtual bool on_resume (); virtual bool on_pause (); virtual void on_files_checked (); virtual void on_state (int /*s*/); virtual void on_add_peer (tcp::endpoint const&, int /*src*/, int /*flags*/); };
new_connection()
virtual boost::shared_ptr<peer_plugin> new_connection (peer_connection*);
This function is called each time a new peer is connected to the torrent. You may choose to ignore this by just returning a default constructed shared_ptr (in which case you don't need to override this member function).
If you need an extension to the peer connection (which most plugins do) you are supposed to return an instance of your peer_plugin class. Which in turn will have its hook functions called on event specific to that peer.
The peer_connection will be valid as long as the shared_ptr is being held by the torrent object. So, it is generally a good idea to not keep a shared_ptr to your own peer_plugin. If you want to keep references to it, use weak_ptr.
If this function throws an exception, the connection will be closed.
on_piece_failed() on_piece_pass()
virtual void on_piece_pass (int /*index*/); virtual void on_piece_failed (int /*index*/);
These hooks are called when a piece passes the hash check or fails the hash check, respectively. The index is the piece index that was downloaded. It is possible to access the list of peers that participated in sending the piece through the torrent and the piece_picker.
tick()
virtual void tick ();
This hook is called approximately once per second. It is a way of making it easy for plugins to do timed events, for sending messages or whatever.
on_resume() on_pause()
virtual bool on_resume (); virtual bool on_pause ();
These hooks are called when the torrent is paused and unpaused respectively. The return value indicates if the event was handled. A return value of true indicates that it was handled, and no other plugin after this one will have this hook function called, and the standard handler will also not be invoked. So, returning true effectively overrides the standard behavior of pause or unpause.
Note that if you call pause() or resume() on the torrent from your handler it will recurse back into your handler, so in order to invoke the standard handler, you have to keep your own state on whether you want standard behavior or overridden behavior.
on_files_checked()
virtual void on_files_checked ();
This function is called when the initial files of the torrent have been checked. If there are no files to check, this function is called immediately.
i.e. This function is always called when the torrent is in a state where it can start downloading.
on_state()
virtual void on_state (int /*s*/);
called when the torrent changes state the state is one of torrent_status::state_t enum members
on_add_peer()
virtual void on_add_peer (tcp::endpoint const&, int /*src*/, int /*flags*/);
called every time a new peer is added to the peer list. This is before the peer is connected to. For flags, see torrent_plugin::flags_t. The source argument refers to the source where we learned about this peer from. It's a bitmask, because many sources may have told us about the same peer. For peer source flags, see peer_info::peer_source_flags.
peer_plugin
Declared in "libtorrent/extensions.hpp"
peer plugins are associated with a specific peer. A peer could be both a regular bittorrent peer (bt_peer_connection) or one of the web seed connections (web_peer_connection or http_seed_connection). In order to only attach to certain peers, make your torrent_plugin::new_connection only return a plugin for certain peer connection types
struct peer_plugin { virtual char const* type () const; virtual void add_handshake (entry&); virtual void on_disconnect (error_code const& /*ec*/); virtual void on_connected (); virtual bool on_handshake (char const* /*reserved_bits*/); virtual bool on_extension_handshake (lazy_entry const&); virtual bool on_have (int /*index*/); virtual bool on_bitfield (bitfield const& /*bitfield*/); virtual bool on_have_all (); virtual bool on_reject (peer_request const&); virtual bool on_request (peer_request const&); virtual bool on_unchoke (); virtual bool on_interested (); virtual bool on_allowed_fast (int /*index*/); virtual bool on_have_none (); virtual bool on_choke (); virtual bool on_not_interested (); virtual bool on_piece (peer_request const& /*piece*/ , disk_buffer_holder& /*data*/); virtual bool on_suggest (int /*index*/); virtual bool on_cancel (peer_request const&); virtual bool on_dont_have (int /*index*/); virtual void sent_unchoke (); virtual bool can_disconnect (error_code const& /*ec*/); virtual bool on_extended (int /*length*/, int /*msg*/, buffer::const_interval /*body*/); virtual bool on_unknown_message (int /*length*/, int /*msg*/, buffer::const_interval /*body*/); virtual void on_piece_pass (int /*index*/); virtual void on_piece_failed (int /*index*/); virtual void tick (); virtual bool write_request (peer_request const&); };
type()
virtual char const* type () const;
This function is expected to return the name of the plugin.
add_handshake()
virtual void add_handshake (entry&);
can add entries to the extension handshake this is not called for web seeds
on_disconnect()
virtual void on_disconnect (error_code const& /*ec*/);
called when the peer is being disconnected.
on_connected()
virtual void on_connected ();
called when the peer is successfully connected. Note that incoming connections will have been connected by the time the peer plugin is attached to it, and won't have this hook called.
on_handshake()
virtual bool on_handshake (char const* /*reserved_bits*/);
this is called when the initial BT handshake is received. Returning false means that the other end doesn't support this extension and will remove it from the list of plugins. this is not called for web seeds
on_extension_handshake()
virtual bool on_extension_handshake (lazy_entry const&);
called when the extension handshake from the other end is received if this returns false, it means that this extension isn't supported by this peer. It will result in this peer_plugin being removed from the peer_connection and destructed. this is not called for web seeds
on_bitfield() on_have_none() on_suggest() on_unchoke() on_cancel() on_have() on_choke() on_piece() on_request() on_reject() on_not_interested() on_interested() on_allowed_fast() on_have_all() on_dont_have()
virtual bool on_have (int /*index*/); virtual bool on_bitfield (bitfield const& /*bitfield*/); virtual bool on_have_all (); virtual bool on_reject (peer_request const&); virtual bool on_request (peer_request const&); virtual bool on_unchoke (); virtual bool on_interested (); virtual bool on_allowed_fast (int /*index*/); virtual bool on_have_none (); virtual bool on_choke (); virtual bool on_not_interested (); virtual bool on_piece (peer_request const& /*piece*/ , disk_buffer_holder& /*data*/); virtual bool on_suggest (int /*index*/); virtual bool on_cancel (peer_request const&); virtual bool on_dont_have (int /*index*/);
returning true from any of the message handlers indicates that the plugin has handeled the message. it will break the plugin chain traversing and not let anyone else handle the message, including the default handler.
can_disconnect()
virtual bool can_disconnect (error_code const& /*ec*/);
called when libtorrent think this peer should be disconnected. if the plugin returns false, the peer will not be disconnected.
on_extended()
virtual bool on_extended (int /*length*/, int /*msg*/, buffer::const_interval /*body*/);
called when an extended message is received. If returning true, the message is not processed by any other plugin and if false is returned the next plugin in the chain will receive it to be able to handle it this is not called for web seeds
on_unknown_message()
virtual bool on_unknown_message (int /*length*/, int /*msg*/, buffer::const_interval /*body*/);
this is not called for web seeds
on_piece_failed() on_piece_pass()
virtual void on_piece_pass (int /*index*/); virtual void on_piece_failed (int /*index*/);
called when a piece that this peer participated in either fails or passes the hash_check
write_request()
virtual bool write_request (peer_request const&);
called each time a request message is to be sent. If true is returned, the original request message won't be sent and no other plugin will have this function called.
create_lt_trackers_plugin()
Declared in "libtorrent/extensions/lt_trackers.hpp"
boost::shared_ptr<torrent_plugin> create_lt_trackers_plugin (torrent*, void*);
constructor function for the trackers exchange extension. This can either be passed in the add_torrent_params::extensions field, or via torrent_handle::add_extension().
create_smart_ban_plugin()
Declared in "libtorrent/extensions/smart_ban.hpp"
boost::shared_ptr<torrent_plugin> create_smart_ban_plugin (torrent*, void*);
constructor function for the smart ban extension. The extension keeps track of the data peers have sent us for failing pieces and once the piece completes and passes the hash check bans the peers that turned out to have sent corrupt data. This function can either be passed in the add_torrent_params::extensions field, or via torrent_handle::add_extension().
create_ut_metadata_plugin()
Declared in "libtorrent/extensions/ut_metadata.hpp"
boost::shared_ptr<torrent_plugin> create_ut_metadata_plugin (torrent*, void*);
constructor function for the ut_metadata extension. The ut_metadata extension allows peers to request the .torrent file (or more specifically the 'info'-dictionary of the .torrent file) from each other. This is the main building block in making magnet links work. This extension is enabled by default unless explicitly disabled in the session constructor.
This can either be passed in the add_torrent_params::extensions field, or via torrent_handle::add_extension().
create_ut_pex_plugin()
Declared in "libtorrent/extensions/ut_pex.hpp"
boost::shared_ptr<torrent_plugin> create_ut_pex_plugin (torrent*, void*);
constructor function for the ut_pex extension. The ut_pex extension allows peers to gossip about their connections, allowing the swarm stay well connected and peers aware of more peers in the swarm. This extension is enabled by default unless explicitly disabled in the session constructor.
This can either be passed in the add_torrent_params::extensions field, or via torrent_handle::add_extension().