mirror of
https://github.com/irungentoo/toxcore.git
synced 2024-03-22 13:30:51 +08:00
a549807df7
This will allow us to do more interesting things with memory allocation within toxcore, and allow fuzzers to explore various allocation failure paths.
97 lines
2.5 KiB
C++
97 lines
2.5 KiB
C++
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
* Copyright © 2022 The TokTok team.
|
|
*/
|
|
|
|
#ifndef C_TOXCORE_TESTING_FUZZING_FUZZ_TOX_H
|
|
#define C_TOXCORE_TESTING_FUZZING_FUZZ_TOX_H
|
|
|
|
#include <cassert>
|
|
#include <memory>
|
|
|
|
#include "../../toxcore/DHT.h"
|
|
#include "../../toxcore/logger.h"
|
|
#include "../../toxcore/network.h"
|
|
#include "fuzz_support.h"
|
|
|
|
constexpr uint16_t SIZE_IP_PORT = SIZE_IP6 + sizeof(uint16_t);
|
|
|
|
template <typename T>
|
|
using Ptr = std::unique_ptr<T, void (*)(T *)>;
|
|
|
|
/** @brief Construct any Tox resource using fuzzer input data.
|
|
*
|
|
* Constructs (or fails by returning) a valid object of type T and passes it to
|
|
* a function specified on the rhs of `>>`. Takes care of cleaning up the
|
|
* resource after the specified function returns.
|
|
*
|
|
* Some `with` instances require additional inputs such as the `Fuzz_Data`
|
|
* reference or a logger.
|
|
*/
|
|
template <typename T>
|
|
struct with;
|
|
|
|
/** @brief Construct a Logger without logging callback.
|
|
*/
|
|
template <>
|
|
struct with<Logger> {
|
|
template <typename F>
|
|
void operator>>(F &&f)
|
|
{
|
|
Ptr<Logger> logger(logger_new(), logger_kill);
|
|
assert(logger != nullptr);
|
|
f(std::move(logger));
|
|
}
|
|
};
|
|
|
|
/** @brief Construct an IP_Port by unpacking fuzzer input with `unpack_ip_port`.
|
|
*/
|
|
template <>
|
|
struct with<IP_Port> {
|
|
Fuzz_Data &input_;
|
|
|
|
template <typename F>
|
|
void operator>>(F &&f)
|
|
{
|
|
CONSUME_OR_RETURN(const uint8_t *ipp_packed, input_, SIZE_IP_PORT);
|
|
IP_Port ipp;
|
|
unpack_ip_port(&ipp, ipp_packed, SIZE_IP6, true);
|
|
|
|
f(ipp);
|
|
}
|
|
};
|
|
|
|
/** @brief Construct a Networking_Core object using the Network vtable passed.
|
|
*
|
|
* Use `with<Logger>{} >> with<Networking_Core>{input, ns, mem} >> ...` to construct
|
|
* a logger and pass it to the Networking_Core constructor function.
|
|
*/
|
|
template <>
|
|
struct with<Networking_Core> {
|
|
Fuzz_Data &input_;
|
|
const Network *ns_;
|
|
const Memory *mem_;
|
|
Ptr<Logger> logger_{nullptr, logger_kill};
|
|
|
|
friend with operator>>(with<Logger> f, with self)
|
|
{
|
|
f >> [&self](Ptr<Logger> logger) { self.logger_ = std::move(logger); };
|
|
return self;
|
|
}
|
|
|
|
template <typename F>
|
|
void operator>>(F &&f)
|
|
{
|
|
with<IP_Port>{input_} >> [&f, this](const IP_Port &ipp) {
|
|
Ptr<Networking_Core> net(
|
|
new_networking_ex(logger_.get(), mem_, ns_, &ipp.ip, ipp.port, ipp.port + 100, nullptr),
|
|
kill_networking);
|
|
if (net == nullptr) {
|
|
return;
|
|
}
|
|
f(std::move(net));
|
|
};
|
|
}
|
|
};
|
|
|
|
#endif // C_TOXCORE_TESTING_FUZZING_FUZZ_TOX_H
|