forked from Green-Sky/tomato
565efa4f39
c9cdae001 fix(toxav): remove extra copy of video frame on encode 4f6d4546b test: Improve the fake network library. a2581e700 refactor(toxcore): generate `Friend_Request` and `Dht_Nodes_Response` 2aaa11770 refactor(toxcore): use Tox_Memory in generated events 5c367452b test(toxcore): fix incorrect mutex in tox_scenario_get_time 8f92e710f perf: Add a timed limit of number of cookie requests. 695b6417a test: Add some more simulated network support. 815ae9ce9 test(toxcore): fix thread-safety in scenario framework 6d85c754e test(toxcore): add unit tests for net_crypto 9c22e79cc test(support): add SimulatedEnvironment for deterministic testing f34fcb195 chore: Update windows Dockerfile to debian stable (trixie). ece0e8980 fix(group_moderation): allow validating unsorted sanction list signatures a4fa754d7 refactor: rename struct Packet to struct Net_Packet d6f330f85 cleanup: Fix some warnings from coverity. e206bffa2 fix(group_chats): fix sync packets reverting topics 0e4715598 test: Add new scenario testing framework. 668291f44 refactor(toxcore): decouple Network_Funcs from sockaddr via IP_Port fc4396cef fix: potential division by zero in toxav and unsafe hex parsing 8e8b352ab refactor: Add nullable annotations to struct members. 7740bb421 refactor: decouple net_crypto from DHT 1936d4296 test: add benchmark for toxav audio and video 46bfdc2df fix: correct printf format specifiers for unsigned integers REVERT: 1828c5356 fix(toxav): remove extra copy of video frame on encode git-subtree-dir: external/toxcore/c-toxcore git-subtree-split: c9cdae001341e701fca980c9bb9febfeb95d2902
74 lines
2.5 KiB
C++
74 lines
2.5 KiB
C++
#include "DHT.h"
|
|
|
|
#include <cassert>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <vector>
|
|
|
|
#include "../testing/support/public/fuzz_data.hh"
|
|
#include "../testing/support/public/simulated_environment.hh"
|
|
|
|
namespace {
|
|
|
|
using tox::test::Fuzz_Data;
|
|
using tox::test::SimulatedEnvironment;
|
|
|
|
void TestHandleRequest(Fuzz_Data &input)
|
|
{
|
|
SimulatedEnvironment env;
|
|
auto c_mem = env.fake_memory().get_c_memory();
|
|
|
|
CONSUME_OR_RETURN(const uint8_t *self_public_key, input, CRYPTO_PUBLIC_KEY_SIZE);
|
|
CONSUME_OR_RETURN(const uint8_t *self_secret_key, input, CRYPTO_SECRET_KEY_SIZE);
|
|
|
|
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE];
|
|
uint8_t request[MAX_CRYPTO_REQUEST_SIZE];
|
|
uint8_t request_id;
|
|
handle_request(&c_mem, self_public_key, self_secret_key, public_key, request, &request_id,
|
|
input.data(), input.size());
|
|
}
|
|
|
|
void TestUnpackNodes(Fuzz_Data &input)
|
|
{
|
|
CONSUME1_OR_RETURN(const bool, tcp_enabled, input);
|
|
|
|
const uint16_t node_count = 5;
|
|
Node_format nodes[node_count];
|
|
uint16_t processed_data_len;
|
|
const int packed_count = unpack_nodes(
|
|
nodes, node_count, &processed_data_len, input.data(), input.size(), tcp_enabled);
|
|
if (packed_count > 0) {
|
|
SimulatedEnvironment env;
|
|
auto c_mem = env.fake_memory().get_c_memory();
|
|
Logger *logger = logger_new(&c_mem);
|
|
std::vector<uint8_t> packed(packed_count * PACKED_NODE_SIZE_IP6);
|
|
const int packed_size
|
|
= pack_nodes(logger, packed.data(), packed.size(), nodes, packed_count);
|
|
LOGGER_ASSERT(logger, packed_size == processed_data_len,
|
|
"packed size (%d) != unpacked size (%d)", packed_size, processed_data_len);
|
|
logger_kill(logger);
|
|
|
|
// Check that packed nodes can be unpacked again and result in the
|
|
// original unpacked nodes.
|
|
Node_format nodes2[node_count];
|
|
uint16_t processed_data_len2;
|
|
const int packed_count2 = unpack_nodes(
|
|
nodes2, node_count, &processed_data_len2, packed.data(), packed.size(), tcp_enabled);
|
|
(void)packed_count2;
|
|
#if 0
|
|
assert(processed_data_len2 == processed_data_len);
|
|
assert(packed_count2 == packed_count);
|
|
#endif
|
|
assert(memcmp(nodes, nodes2, sizeof(Node_format) * packed_count) == 0);
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size);
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
|
|
{
|
|
tox::test::fuzz_select_target<TestHandleRequest, TestUnpackNodes>(data, size);
|
|
return 0;
|
|
}
|