Squashed 'external/toxcore/c-toxcore/' changes from 1828c5356..c9cdae001
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
This commit is contained in:
Green Sky
299
testing/support/doubles/fake_network_stack.cc
Normal file
299
testing/support/doubles/fake_network_stack.cc
Normal file
@@ -0,0 +1,299 @@
|
||||
#include "fake_network_stack.hh"
|
||||
|
||||
#include <cerrno>
|
||||
#include <cstring>
|
||||
#include <iostream>
|
||||
|
||||
#include "../../../toxcore/mem.h"
|
||||
|
||||
namespace tox::test {
|
||||
|
||||
static const Network_Funcs kVtable = {
|
||||
.close
|
||||
= [](void *obj, Socket sock) { return static_cast<FakeNetworkStack *>(obj)->close(sock); },
|
||||
.accept
|
||||
= [](void *obj, Socket sock) { return static_cast<FakeNetworkStack *>(obj)->accept(sock); },
|
||||
.bind
|
||||
= [](void *obj, Socket sock,
|
||||
const IP_Port *addr) { return static_cast<FakeNetworkStack *>(obj)->bind(sock, addr); },
|
||||
.listen
|
||||
= [](void *obj, Socket sock,
|
||||
int backlog) { return static_cast<FakeNetworkStack *>(obj)->listen(sock, backlog); },
|
||||
.connect =
|
||||
[](void *obj, Socket sock, const IP_Port *addr) {
|
||||
return static_cast<FakeNetworkStack *>(obj)->connect(sock, addr);
|
||||
},
|
||||
.recvbuf
|
||||
= [](void *obj, Socket sock) { return static_cast<FakeNetworkStack *>(obj)->recvbuf(sock); },
|
||||
.recv = [](void *obj, Socket sock, uint8_t *buf,
|
||||
size_t len) { return static_cast<FakeNetworkStack *>(obj)->recv(sock, buf, len); },
|
||||
.recvfrom =
|
||||
[](void *obj, Socket sock, uint8_t *buf, size_t len, IP_Port *addr) {
|
||||
return static_cast<FakeNetworkStack *>(obj)->recvfrom(sock, buf, len, addr);
|
||||
},
|
||||
.send = [](void *obj, Socket sock, const uint8_t *buf,
|
||||
size_t len) { return static_cast<FakeNetworkStack *>(obj)->send(sock, buf, len); },
|
||||
.sendto =
|
||||
[](void *obj, Socket sock, const uint8_t *buf, size_t len, const IP_Port *addr) {
|
||||
return static_cast<FakeNetworkStack *>(obj)->sendto(sock, buf, len, addr);
|
||||
},
|
||||
.socket
|
||||
= [](void *obj, int domain, int type,
|
||||
int proto) { return static_cast<FakeNetworkStack *>(obj)->socket(domain, type, proto); },
|
||||
.socket_nonblock =
|
||||
[](void *obj, Socket sock, bool nonblock) {
|
||||
return static_cast<FakeNetworkStack *>(obj)->socket_nonblock(sock, nonblock);
|
||||
},
|
||||
.getsockopt =
|
||||
[](void *obj, Socket sock, int level, int optname, void *optval, size_t *optlen) {
|
||||
return static_cast<FakeNetworkStack *>(obj)->getsockopt(
|
||||
sock, level, optname, optval, optlen);
|
||||
},
|
||||
.setsockopt =
|
||||
[](void *obj, Socket sock, int level, int optname, const void *optval, size_t optlen) {
|
||||
return static_cast<FakeNetworkStack *>(obj)->setsockopt(
|
||||
sock, level, optname, optval, optlen);
|
||||
},
|
||||
.getaddrinfo =
|
||||
[](void *obj, const Memory *mem, const char *address, int family, int protocol,
|
||||
IP_Port **addrs) {
|
||||
FakeNetworkStack *self = static_cast<FakeNetworkStack *>(obj);
|
||||
if (self->universe().is_verbose()) {
|
||||
std::cerr << "[FakeNetworkStack] getaddrinfo for " << address << std::endl;
|
||||
}
|
||||
if (strcmp(address, "127.0.0.1") == 0 || strcmp(address, "localhost") == 0) {
|
||||
*addrs = static_cast<IP_Port *>(mem_alloc(mem, sizeof(IP_Port)));
|
||||
memset(&(*addrs)->ip, 0, sizeof(IP));
|
||||
ip_init(&(*addrs)->ip, false);
|
||||
(*addrs)->ip.ip.v4.uint32 = net_htonl(0x7F000001);
|
||||
(*addrs)->port = 0;
|
||||
return 1;
|
||||
}
|
||||
|
||||
IP ip;
|
||||
if (addr_parse_ip(address, &ip)) {
|
||||
*addrs = static_cast<IP_Port *>(mem_alloc(mem, sizeof(IP_Port)));
|
||||
(*addrs)->ip = ip;
|
||||
(*addrs)->port = 0;
|
||||
if (self->universe().is_verbose()) {
|
||||
std::cerr << "[FakeNetworkStack] resolved " << address << std::endl;
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
},
|
||||
.freeaddrinfo =
|
||||
[](void *obj, const Memory *mem, IP_Port *addrs) {
|
||||
mem_delete(mem, addrs);
|
||||
return 0;
|
||||
},
|
||||
};
|
||||
|
||||
FakeNetworkStack::FakeNetworkStack(NetworkUniverse &universe, const IP &node_ip)
|
||||
: universe_(universe)
|
||||
, node_ip_(node_ip)
|
||||
{
|
||||
}
|
||||
|
||||
FakeNetworkStack::~FakeNetworkStack() = default;
|
||||
|
||||
struct Network FakeNetworkStack::get_c_network() { return Network{&kVtable, this}; }
|
||||
|
||||
Socket FakeNetworkStack::socket(int domain, int type, int protocol)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
int fd = next_fd_++;
|
||||
|
||||
std::unique_ptr<FakeSocket> sock;
|
||||
if (type == SOCK_DGRAM) {
|
||||
if (universe_.is_verbose()) {
|
||||
std::cerr << "[FakeNetworkStack] create UDP socket fd=" << fd << std::endl;
|
||||
}
|
||||
sock = std::make_unique<FakeUdpSocket>(universe_);
|
||||
} else if (type == SOCK_STREAM) {
|
||||
if (universe_.is_verbose()) {
|
||||
std::cerr << "[FakeNetworkStack] create TCP socket fd=" << fd << std::endl;
|
||||
}
|
||||
sock = std::make_unique<FakeTcpSocket>(universe_);
|
||||
} else {
|
||||
// Unknown type
|
||||
return net_socket_from_native(-1);
|
||||
}
|
||||
|
||||
sockets_[fd] = std::move(sock);
|
||||
sockets_[fd]->set_ip(node_ip_);
|
||||
return net_socket_from_native(fd);
|
||||
}
|
||||
|
||||
FakeSocket *FakeNetworkStack::get_sock(Socket sock)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
auto it = sockets_.find(net_socket_to_native(sock));
|
||||
if (it != sockets_.end()) {
|
||||
return it->second.get();
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::close(Socket sock)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
int fd = net_socket_to_native(sock);
|
||||
auto it = sockets_.find(fd);
|
||||
if (it == sockets_.end()) {
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
it->second->close();
|
||||
sockets_.erase(it);
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Delegate all others
|
||||
int FakeNetworkStack::bind(Socket sock, const IP_Port *addr)
|
||||
{
|
||||
if (auto *s = get_sock(sock)) {
|
||||
int ret = s->bind(addr);
|
||||
if (universe_.is_verbose() && ret == 0) {
|
||||
char ip_str[TOX_INET_ADDRSTRLEN];
|
||||
ip_parse_addr(&s->ip_address(), ip_str, sizeof(ip_str));
|
||||
std::cerr << "[FakeNetworkStack] bound socket to " << ip_str << ":" << s->local_port()
|
||||
<< std::endl;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::connect(Socket sock, const IP_Port *addr)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->connect(addr);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::listen(Socket sock, int backlog)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->listen(backlog);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
Socket FakeNetworkStack::accept(Socket sock)
|
||||
{
|
||||
// This requires creating a new FD
|
||||
IP_Port addr;
|
||||
std::unique_ptr<FakeSocket> new_sock_obj;
|
||||
|
||||
{
|
||||
auto *s = get_sock(sock);
|
||||
if (!s) {
|
||||
errno = EBADF;
|
||||
return net_socket_from_native(-1);
|
||||
}
|
||||
new_sock_obj = s->accept(&addr);
|
||||
}
|
||||
|
||||
if (!new_sock_obj) {
|
||||
// errno set by accept
|
||||
return net_socket_from_native(-1);
|
||||
}
|
||||
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
int fd = next_fd_++;
|
||||
sockets_[fd] = std::move(new_sock_obj);
|
||||
return net_socket_from_native(fd);
|
||||
}
|
||||
|
||||
int FakeNetworkStack::send(Socket sock, const uint8_t *buf, size_t len)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->send(buf, len);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::recv(Socket sock, uint8_t *buf, size_t len)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->recv(buf, len);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::recvbuf(Socket sock)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->recv_buffer_size();
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::sendto(Socket sock, const uint8_t *buf, size_t len, const IP_Port *addr)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->sendto(buf, len, addr);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::recvfrom(Socket sock, uint8_t *buf, size_t len, IP_Port *addr)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->recvfrom(buf, len, addr);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::socket_nonblock(Socket sock, bool nonblock)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->socket_nonblock(nonblock);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::getsockopt(Socket sock, int level, int optname, void *optval, size_t *optlen)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->getsockopt(level, optname, optval, optlen);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int FakeNetworkStack::setsockopt(
|
||||
Socket sock, int level, int optname, const void *optval, size_t optlen)
|
||||
{
|
||||
if (auto *s = get_sock(sock))
|
||||
return s->setsockopt(level, optname, optval, optlen);
|
||||
errno = EBADF;
|
||||
return -1;
|
||||
}
|
||||
|
||||
FakeUdpSocket *FakeNetworkStack::get_udp_socket(Socket sock)
|
||||
{
|
||||
if (auto *s = get_sock(sock)) {
|
||||
if (s->type() == SOCK_DGRAM) {
|
||||
return static_cast<FakeUdpSocket *>(s);
|
||||
}
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
std::vector<FakeUdpSocket *> FakeNetworkStack::get_bound_udp_sockets()
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
std::vector<FakeUdpSocket *> result;
|
||||
for (const auto &pair : sockets_) {
|
||||
FakeSocket *s = pair.second.get();
|
||||
if (s->type() == SOCK_DGRAM && s->local_port() != 0) {
|
||||
result.push_back(static_cast<FakeUdpSocket *>(s));
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
} // namespace tox::test
|
||||
Reference in New Issue
Block a user