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
126 lines
3.6 KiB
C++
126 lines
3.6 KiB
C++
#include "fake_sockets.hh"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "network_universe.hh"
|
|
|
|
namespace tox::test {
|
|
namespace {
|
|
|
|
class FakeTcpSocketTest : public ::testing::Test {
|
|
public:
|
|
~FakeTcpSocketTest() override;
|
|
|
|
protected:
|
|
NetworkUniverse universe;
|
|
FakeTcpSocket server{universe};
|
|
FakeTcpSocket client{universe};
|
|
};
|
|
|
|
FakeTcpSocketTest::~FakeTcpSocketTest() = default;
|
|
|
|
TEST_F(FakeTcpSocketTest, ConnectAndAccept)
|
|
{
|
|
IP_Port server_addr;
|
|
ip_init(&server_addr.ip, false);
|
|
server_addr.ip.ip.v4.uint32 = net_htonl(0x7F000001);
|
|
server_addr.port = net_htons(8080);
|
|
|
|
ASSERT_EQ(server.bind(&server_addr), 0);
|
|
ASSERT_EQ(server.listen(5), 0);
|
|
|
|
// Client connects
|
|
ASSERT_EQ(client.connect(&server_addr), -1);
|
|
ASSERT_EQ(errno, EINPROGRESS);
|
|
|
|
// Process events (Client SYN -> Server)
|
|
universe.process_events(0);
|
|
|
|
// Server accepts (SYN-ACK -> Client)
|
|
universe.process_events(0);
|
|
|
|
// Client receives SYN-ACK, sends ACK (ACK -> Server)
|
|
universe.process_events(0);
|
|
|
|
// Server receives ACK, connection established
|
|
IP_Port client_addr;
|
|
auto accepted = server.accept(&client_addr);
|
|
ASSERT_NE(accepted, nullptr);
|
|
auto *accepted_tcp = static_cast<FakeTcpSocket *>(accepted.get());
|
|
EXPECT_EQ(accepted_tcp->state(), FakeTcpSocket::ESTABLISHED);
|
|
|
|
EXPECT_EQ(client.state(), FakeTcpSocket::ESTABLISHED);
|
|
}
|
|
|
|
TEST_F(FakeTcpSocketTest, DataTransfer)
|
|
{
|
|
IP_Port server_addr;
|
|
ip_init(&server_addr.ip, false);
|
|
server_addr.ip.ip.v4.uint32 = net_htonl(0x7F000001);
|
|
server_addr.port = net_htons(8081);
|
|
|
|
server.bind(&server_addr);
|
|
server.listen(5);
|
|
client.connect(&server_addr);
|
|
|
|
// Handshake
|
|
universe.process_events(0); // SYN
|
|
universe.process_events(0); // SYN-ACK
|
|
universe.process_events(0); // ACK
|
|
|
|
auto accepted = server.accept(nullptr);
|
|
ASSERT_NE(accepted, nullptr);
|
|
|
|
// Send data Client -> Server
|
|
uint8_t send_buf[] = "Hello";
|
|
ASSERT_EQ(client.send(send_buf, 5), 5);
|
|
|
|
universe.process_events(0); // Data packet
|
|
|
|
uint8_t recv_buf[10];
|
|
ASSERT_EQ(accepted->recv(recv_buf, 10), 5);
|
|
EXPECT_EQ(std::string(reinterpret_cast<char *>(recv_buf), 5), "Hello");
|
|
}
|
|
|
|
class FakeUdpSocketTest : public ::testing::Test {
|
|
public:
|
|
~FakeUdpSocketTest() override;
|
|
|
|
protected:
|
|
NetworkUniverse universe;
|
|
FakeUdpSocket server{universe};
|
|
FakeUdpSocket client{universe};
|
|
};
|
|
|
|
FakeUdpSocketTest::~FakeUdpSocketTest() = default;
|
|
|
|
TEST_F(FakeUdpSocketTest, BindAndSendTo)
|
|
{
|
|
IP_Port server_addr;
|
|
ip_init(&server_addr.ip, false);
|
|
server_addr.ip.ip.v4.uint32 = net_htonl(0x7F000001);
|
|
server_addr.port = net_htons(9000);
|
|
|
|
ASSERT_EQ(server.bind(&server_addr), 0);
|
|
|
|
const char *message = "UDP Packet";
|
|
ASSERT_EQ(client.sendto(
|
|
reinterpret_cast<const uint8_t *>(message), strlen(message), &server_addr),
|
|
strlen(message));
|
|
|
|
universe.process_events(0);
|
|
|
|
IP_Port sender_addr;
|
|
uint8_t recv_buf[100];
|
|
int len = server.recvfrom(recv_buf, sizeof(recv_buf), &sender_addr);
|
|
|
|
ASSERT_GT(len, 0);
|
|
EXPECT_EQ(std::string(reinterpret_cast<char *>(recv_buf), len), message);
|
|
EXPECT_EQ(sender_addr.port, net_htons(client.local_port()));
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace tox::test
|
|
|
|
// end of file
|