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
91 lines
2.3 KiB
C++
91 lines
2.3 KiB
C++
#ifndef C_TOXCORE_TESTING_SUPPORT_DOUBLES_NETWORK_UNIVERSE_H
|
|
#define C_TOXCORE_TESTING_SUPPORT_DOUBLES_NETWORK_UNIVERSE_H
|
|
|
|
#include <cstdint>
|
|
#include <functional>
|
|
#include <map>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <queue>
|
|
#include <vector>
|
|
|
|
#include "../../../toxcore/network.h"
|
|
|
|
namespace tox::test {
|
|
|
|
class FakeUdpSocket;
|
|
class FakeTcpSocket;
|
|
|
|
struct Packet {
|
|
IP_Port from;
|
|
IP_Port to;
|
|
std::vector<uint8_t> data;
|
|
uint64_t delivery_time;
|
|
bool is_tcp = false;
|
|
|
|
// TCP Simulation Fields
|
|
uint8_t tcp_flags = 0; // 0x01=FIN, 0x02=SYN, 0x10=ACK
|
|
uint32_t seq = 0;
|
|
uint32_t ack = 0;
|
|
|
|
bool operator>(const Packet &other) const { return delivery_time > other.delivery_time; }
|
|
};
|
|
|
|
/**
|
|
* @brief The God Object for the network simulation.
|
|
* Manages routing, latency, and connectivity.
|
|
*/
|
|
class NetworkUniverse {
|
|
public:
|
|
using PacketFilter = std::function<bool(Packet &)>;
|
|
using PacketSink = std::function<void(const Packet &)>;
|
|
|
|
NetworkUniverse();
|
|
~NetworkUniverse();
|
|
|
|
// Registration
|
|
// Returns true if binding succeeded
|
|
bool bind_udp(IP ip, uint16_t port, FakeUdpSocket *socket);
|
|
void unbind_udp(IP ip, uint16_t port);
|
|
|
|
bool bind_tcp(IP ip, uint16_t port, FakeTcpSocket *socket);
|
|
void unbind_tcp(IP ip, uint16_t port, FakeTcpSocket *socket);
|
|
|
|
// Routing
|
|
void send_packet(Packet p);
|
|
|
|
// Simulation
|
|
void process_events(uint64_t current_time_ms);
|
|
void set_latency(uint64_t ms);
|
|
void set_verbose(bool verbose);
|
|
bool is_verbose() const;
|
|
void add_filter(PacketFilter filter);
|
|
void add_observer(PacketSink sink);
|
|
|
|
// Helpers
|
|
// Finds a free port starting from 'start'
|
|
uint16_t find_free_port(IP ip, uint16_t start = 33445);
|
|
|
|
struct IP_Port_Key {
|
|
IP ip;
|
|
uint16_t port;
|
|
bool operator<(const IP_Port_Key &other) const;
|
|
};
|
|
|
|
private:
|
|
std::map<IP_Port_Key, FakeUdpSocket *> udp_bindings_;
|
|
std::multimap<IP_Port_Key, FakeTcpSocket *> tcp_bindings_;
|
|
|
|
std::priority_queue<Packet, std::vector<Packet>, std::greater<Packet>> event_queue_;
|
|
std::vector<PacketFilter> filters_;
|
|
std::vector<PacketSink> observers_;
|
|
|
|
uint64_t global_latency_ms_ = 0;
|
|
bool verbose_ = false;
|
|
std::recursive_mutex mutex_;
|
|
};
|
|
|
|
} // namespace tox::test
|
|
|
|
#endif // C_TOXCORE_TESTING_SUPPORT_DOUBLES_NETWORK_UNIVERSE_H
|