forked from premiere/premiere-libtorrent
263 lines
6.9 KiB
C++
263 lines
6.9 KiB
C++
/*
|
|
|
|
Copyright (c) 2006, 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 NODE_HPP
|
|
#define NODE_HPP
|
|
|
|
#include <algorithm>
|
|
#include <map>
|
|
#include <set>
|
|
|
|
#include <libtorrent/kademlia/routing_table.hpp>
|
|
#include <libtorrent/kademlia/rpc_manager.hpp>
|
|
#include <libtorrent/kademlia/node_id.hpp>
|
|
#include <libtorrent/kademlia/msg.hpp>
|
|
|
|
#include <libtorrent/io.hpp>
|
|
#include <libtorrent/session_settings.hpp>
|
|
#include <libtorrent/assert.hpp>
|
|
|
|
#include <boost/cstdint.hpp>
|
|
#include <boost/optional.hpp>
|
|
#include <boost/iterator/transform_iterator.hpp>
|
|
#include <boost/ref.hpp>
|
|
|
|
#include "libtorrent/socket.hpp"
|
|
|
|
namespace libtorrent { namespace dht
|
|
{
|
|
|
|
using asio::ip::udp;
|
|
|
|
#ifdef TORRENT_DHT_VERBOSE_LOGGING
|
|
TORRENT_DECLARE_LOG(node);
|
|
#endif
|
|
|
|
// this is the entry for every peer
|
|
// the timestamp is there to make it possible
|
|
// to remove stale peers
|
|
struct peer_entry
|
|
{
|
|
tcp::endpoint addr;
|
|
ptime added;
|
|
};
|
|
|
|
// this is a group. It contains a set of group members
|
|
struct torrent_entry
|
|
{
|
|
std::set<peer_entry> peers;
|
|
};
|
|
|
|
inline bool operator<(peer_entry const& lhs, peer_entry const& rhs)
|
|
{
|
|
return lhs.addr.address() == rhs.addr.address()
|
|
? lhs.addr.port() < rhs.addr.port()
|
|
: lhs.addr.address() < rhs.addr.address();
|
|
}
|
|
|
|
struct null_type {};
|
|
|
|
class announce_observer : public observer
|
|
{
|
|
public:
|
|
announce_observer(boost::pool<>& allocator
|
|
, sha1_hash const& info_hash
|
|
, int listen_port
|
|
, entry const& write_token)
|
|
: observer(allocator)
|
|
, m_info_hash(info_hash)
|
|
, m_listen_port(listen_port)
|
|
, m_token(write_token)
|
|
{}
|
|
|
|
void send(msg& m)
|
|
{
|
|
m.port = m_listen_port;
|
|
m.info_hash = m_info_hash;
|
|
m.write_token = m_token;
|
|
}
|
|
|
|
void timeout() {}
|
|
void reply(msg const&) {}
|
|
void abort() {}
|
|
|
|
private:
|
|
sha1_hash m_info_hash;
|
|
int m_listen_port;
|
|
entry m_token;
|
|
};
|
|
|
|
class get_peers_observer : public observer
|
|
{
|
|
public:
|
|
get_peers_observer(sha1_hash const& info_hash
|
|
, int listen_port
|
|
, rpc_manager& rpc
|
|
, boost::function<void(std::vector<tcp::endpoint> const&, sha1_hash const&)> f)
|
|
: observer(rpc.allocator())
|
|
, m_info_hash(info_hash)
|
|
, m_listen_port(listen_port)
|
|
, m_rpc(rpc)
|
|
, m_fun(f)
|
|
{}
|
|
|
|
void send(msg& m)
|
|
{
|
|
m.port = m_listen_port;
|
|
m.info_hash = m_info_hash;
|
|
}
|
|
|
|
void timeout() {}
|
|
void reply(msg const& r)
|
|
{
|
|
observer_ptr o(new (m_rpc.allocator().malloc()) announce_observer(
|
|
m_rpc.allocator(), m_info_hash, m_listen_port, r.write_token));
|
|
#ifndef NDEBUG
|
|
o->m_in_constructor = false;
|
|
#endif
|
|
m_rpc.invoke(messages::announce_peer, r.addr, o);
|
|
m_fun(r.peers, m_info_hash);
|
|
}
|
|
void abort() {}
|
|
|
|
private:
|
|
sha1_hash m_info_hash;
|
|
int m_listen_port;
|
|
rpc_manager& m_rpc;
|
|
boost::function<void(std::vector<tcp::endpoint> const&, sha1_hash const&)> m_fun;
|
|
};
|
|
|
|
|
|
|
|
class node_impl : boost::noncopyable
|
|
{
|
|
typedef std::map<node_id, torrent_entry> table_t;
|
|
public:
|
|
node_impl(boost::function<void(msg const&)> const& f
|
|
, dht_settings const& settings, boost::optional<node_id> node_id);
|
|
|
|
virtual ~node_impl() {}
|
|
|
|
void refresh(node_id const& id, boost::function0<void> f);
|
|
void bootstrap(std::vector<udp::endpoint> const& nodes
|
|
, boost::function0<void> f);
|
|
void find_node(node_id const& id, boost::function<
|
|
void(std::vector<node_entry> const&)> f);
|
|
void add_router_node(udp::endpoint router);
|
|
|
|
void incoming(msg const& m);
|
|
|
|
void refresh();
|
|
void refresh_bucket(int bucket);
|
|
int bucket_size(int bucket);
|
|
|
|
typedef routing_table::iterator iterator;
|
|
|
|
iterator begin() const { return m_table.begin(); }
|
|
iterator end() const { return m_table.end(); }
|
|
|
|
typedef table_t::iterator data_iterator;
|
|
|
|
node_id const& nid() const { return m_id; }
|
|
boost::tuple<int, int> size() const{ return m_table.size(); }
|
|
size_type num_global_nodes() const
|
|
{ return m_table.num_global_nodes(); }
|
|
|
|
data_iterator begin_data() { return m_map.begin(); }
|
|
data_iterator end_data() { return m_map.end(); }
|
|
int data_size() const { return int(m_map.size()); }
|
|
|
|
#ifdef TORRENT_DHT_VERBOSE_LOGGING
|
|
void print_state(std::ostream& os) const
|
|
{ m_table.print_state(os); }
|
|
#endif
|
|
|
|
void announce(sha1_hash const& info_hash, int listen_port
|
|
, boost::function<void(std::vector<tcp::endpoint> const&
|
|
, sha1_hash const&)> f);
|
|
|
|
bool verify_token(msg const& m);
|
|
entry generate_token(msg const& m);
|
|
|
|
// the returned time is the delay until connection_timeout()
|
|
// should be called again the next time
|
|
time_duration connection_timeout();
|
|
time_duration refresh_timeout();
|
|
|
|
// generates a new secret number used to generate write tokens
|
|
void new_write_key();
|
|
|
|
// pings the given node, and adds it to
|
|
// the routing table if it respons and if the
|
|
// bucket is not full.
|
|
void add_node(udp::endpoint node);
|
|
|
|
void replacement_cache(bucket_t& nodes) const
|
|
{ m_table.replacement_cache(nodes); }
|
|
|
|
protected:
|
|
// is called when a find data request is received. Should
|
|
// return false if the data is not stored on this node. If
|
|
// the data is stored, it should be serialized into 'data'.
|
|
bool on_find(msg const& m, std::vector<tcp::endpoint>& peers) const;
|
|
|
|
// this is called when a store request is received. The data
|
|
// is store-parameters and the data to be stored.
|
|
void on_announce(msg const& m, msg& reply);
|
|
|
|
dht_settings const& m_settings;
|
|
|
|
// the maximum number of peers to send in a get_peers
|
|
// reply. Ordinary trackers usually limit this to 50.
|
|
// 50 => 6 * 50 = 250 bytes + packet overhead
|
|
int m_max_peers_reply;
|
|
|
|
private:
|
|
void incoming_request(msg const& h);
|
|
|
|
node_id m_id;
|
|
routing_table m_table;
|
|
rpc_manager m_rpc;
|
|
table_t m_map;
|
|
|
|
ptime m_last_tracker_tick;
|
|
|
|
// secret random numbers used to create write tokens
|
|
int m_secret[2];
|
|
};
|
|
|
|
|
|
} } // namespace libtorrent::dht
|
|
|
|
#endif // NODE_HPP
|
|
|