forked from Green-Sky/tomato
Green Sky
8eb4892b49
6d634674a9 cleanup: Remove old type-ordered event getters. d1d48d1dfc feat: add ngc events 994ffecc6b refactor: Make event dispatch ordered by receive time. 812f931d5f fix: Make sure there's enough space for CONSUME1 in fuzzers. 50f1b30fa9 test: Add fuzz tests to the coverage run. df76f5cf47 chore: Move from gcov to llvm source-based coverage. 072e3beb3f fix: issues with packet broadcast error reporting 6b6718e4d2 cleanup: Make group packet entry creation less error-prone 5b9c420ce1 refactor: packet broadcast functions now return errors af4cb31028 refactor: Use `operator==` for equality tests of `Node_format`. 9592d590cf refactor(test): Slightly nicer C++ interface to tox Random. c66e10fb7a refactor: Minor refactoring of get_close_nodes functions. ebc9643862 fix: don't pass garbage data buffer to packet send functions 32b68cffca cleanup: Some more test cleanups, removing overly smart code. 0426624dcb refactor: Assign malloc return to a local variable first. afc38f2458 test: Add more unit tests for `add_to_list`. 05ce5c1ab9 test: Add "infer" CI check to github, remove from circle. REVERT: 8f0d505f9a feat: add ngc events REVERT: 9b8216e70c refactor: Make event dispatch ordered by receive time. git-subtree-dir: external/toxcore/c-toxcore git-subtree-split: 6d634674a929edb0ab70689dcbcb195b3547be13
62 lines
2.1 KiB
C++
62 lines
2.1 KiB
C++
#include "DHT.h"
|
|
|
|
#include <cassert>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <vector>
|
|
|
|
#include "../testing/fuzzing/fuzz_support.h"
|
|
|
|
namespace {
|
|
|
|
void TestHandleRequest(Fuzz_Data &input)
|
|
{
|
|
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(
|
|
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) {
|
|
Logger *logger = logger_new();
|
|
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);
|
|
assert(processed_data_len2 == processed_data_len);
|
|
assert(packed_count2 == packed_count);
|
|
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)
|
|
{
|
|
fuzz_select_target(data, size, TestHandleRequest, TestUnpackNodes);
|
|
return 0;
|
|
}
|