premiere
/
premiere-libtorrent
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

fix (broken) DHT node fuzzer

This commit is contained in:
arvidn 2019-04-30 09:16:56 +02:00 committed by Arvid Norberg
parent 7d1afa9b41
commit 94098566f6
1 changed files with 28 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
fuzzers/src/dht_node.cpp
View File

@ -34,6 +34,8 @@ POSSIBILITY OF SUCH DAMAGE.
#include "libtorrent/kademlia/dht_tracker.hpp" #include "libtorrent/kademlia/dht_tracker.hpp"
#include "libtorrent/performance_counters.hpp" #include "libtorrent/performance_counters.hpp"
#include "libtorrent/kademlia/dht_observer.hpp" #include "libtorrent/kademlia/dht_observer.hpp"
#include "libtorrent/ip_filter.hpp"
#include "libtorrent/aux_/session_impl.hpp"
#include <memory> #include <memory>
@ -97,34 +99,44 @@ struct obs : dht::dht_observer
}; };
obs o; obs o;
extern "C" int LLVMFuzzerTestOneInput(uint8_t const* data, size_t size)
{
#if LIBTORRENT_VERSION_NUM >= 10300 #if LIBTORRENT_VERSION_NUM >= 10300
io_context ios; io_context ios;
#else #else
io_service ios; io_service ios;
#endif #endif
#if LIBTORRENT_VERSION_NUM < 10200 #if LIBTORRENT_VERSION_NUM < 10200
rate_limited_udp_socket sock(ios); rate_limited_udp_socket sock(ios);
#endif #endif
dht::dht_tracker dht_node(&o
dht::dht_tracker(&o
#if LIBTORRENT_VERSION_NUM >= 10200 #if LIBTORRENT_VERSION_NUM >= 10200
, ios , ios
, [](aux::listen_socket_handle const&, udp::endpoint const& , [](aux::listen_socket_handle const&, udp::endpoint const&
, span<char const>, error_code&, udp_send_flags_t) {} , span<char const>, error_code&, udp_send_flags_t) {}
#else #else
, sock , sock
#endif #endif
, sett , sett
, cnt , cnt
#if LIBTORRENT_VERSION_NUM >= 10200 #if LIBTORRENT_VERSION_NUM >= 10200
, *dht_storage , *dht_storage
#else #else
, dht::dht_default_storage_constructor , dht::dht_default_storage_constructor
#endif #endif
, std::move(state)); , std::move(state));
auto listen_socket = std::make_shared<aux::listen_socket_t>();
aux::listen_socket_handle s(listen_socket);
error_code ignore;
lt::address_v4 src = make_address_v4("2.2.2.2", ignore);
udp::endpoint ep(src, 6881);
std::once_flag once_flag;
extern "C" int LLVMFuzzerTestOneInput(uint8_t const* data, size_t size)
{
ep.address(src);
src = lt::address_v4(detail::plus_one(src.to_bytes()));
std::call_once(once_flag, []{ dht_node.new_socket(s); });
dht_node.incoming_packet(s, ep, {reinterpret_cast<char const*>(data), int(size)});
return 0; return 0;
} }