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

20
fuzzers/src/dht_node.cpp
View File

@ -34,6 +34,8 @@ POSSIBILITY OF SUCH DAMAGE.
#include "libtorrent/kademlia/dht_tracker.hpp"
#include "libtorrent/performance_counters.hpp"
#include "libtorrent/kademlia/dht_observer.hpp"
#include "libtorrent/ip_filter.hpp"
#include "libtorrent/aux_/session_impl.hpp"
#include <memory>
@ -97,8 +99,6 @@ struct obs : dht::dht_observer
};
obs o;
extern "C" int LLVMFuzzerTestOneInput(uint8_t const* data, size_t size)
{
#if LIBTORRENT_VERSION_NUM >= 10300
io_context ios;
#else
@ -107,8 +107,7 @@ extern "C" int LLVMFuzzerTestOneInput(uint8_t const* data, size_t size)
#if LIBTORRENT_VERSION_NUM < 10200
rate_limited_udp_socket sock(ios);
#endif
dht::dht_tracker(&o
dht::dht_tracker dht_node(&o
#if LIBTORRENT_VERSION_NUM >= 10200
, ios
, [](aux::listen_socket_handle const&, udp::endpoint const&
@ -124,7 +123,20 @@ extern "C" int LLVMFuzzerTestOneInput(uint8_t const* data, size_t size)
, dht::dht_default_storage_constructor
#endif
, 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;
}