forked from Green-Sky/tomato
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:
102
auto_tests/scenarios/scenario_friend_request_spam_test.c
Normal file
102
auto_tests/scenarios/scenario_friend_request_spam_test.c
Normal file
@@ -0,0 +1,102 @@
|
||||
#include "framework/framework.h"
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#define FR_MESSAGE "Gentoo"
|
||||
#define FR_TOX_COUNT 33
|
||||
|
||||
typedef struct {
|
||||
uint32_t requests_received;
|
||||
} ReceiverState;
|
||||
|
||||
static void on_friend_request(const Tox_Event_Friend_Request *event, void *user_data)
|
||||
{
|
||||
ToxNode *self = (ToxNode *)user_data;
|
||||
ReceiverState *state = (ReceiverState *)tox_node_get_script_ctx(self);
|
||||
|
||||
const uint8_t *public_key = tox_event_friend_request_get_public_key(event);
|
||||
const uint8_t *data = tox_event_friend_request_get_message(event);
|
||||
const size_t length = tox_event_friend_request_get_message_length(event);
|
||||
|
||||
if (length == sizeof(FR_MESSAGE) && memcmp(FR_MESSAGE, data, sizeof(FR_MESSAGE)) == 0) {
|
||||
tox_friend_add_norequest(tox_node_get_tox(self), public_key, nullptr);
|
||||
state->requests_received++;
|
||||
tox_node_log(self, "Friend request received: %u", state->requests_received);
|
||||
}
|
||||
}
|
||||
|
||||
static void receiver_script(ToxNode *self, void *ctx)
|
||||
{
|
||||
const ReceiverState *state = (const ReceiverState *)ctx;
|
||||
tox_events_callback_friend_request(tox_node_get_dispatch(self), on_friend_request);
|
||||
|
||||
WAIT_UNTIL(tox_node_is_self_connected(self));
|
||||
|
||||
// Wait for all requests to be received and friends connected
|
||||
WAIT_UNTIL(state->requests_received == FR_TOX_COUNT - 1);
|
||||
|
||||
// Also wait until they are all connected as friends
|
||||
bool all_connected = false;
|
||||
while (!all_connected && tox_scenario_is_running(self)) {
|
||||
all_connected = true;
|
||||
for (uint32_t i = 0; i < FR_TOX_COUNT - 1; ++i) {
|
||||
if (!tox_node_is_friend_connected(self, i)) {
|
||||
all_connected = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!all_connected) {
|
||||
tox_scenario_yield(self);
|
||||
}
|
||||
}
|
||||
|
||||
tox_node_log(self, "All friends connected");
|
||||
}
|
||||
|
||||
static void sender_script(ToxNode *self, void *ctx)
|
||||
{
|
||||
WAIT_UNTIL(tox_node_is_self_connected(self));
|
||||
|
||||
ToxNode *receiver = (ToxNode *)ctx;
|
||||
uint8_t address[TOX_ADDRESS_SIZE];
|
||||
tox_node_get_address(receiver, address);
|
||||
|
||||
Tox_Err_Friend_Add err;
|
||||
tox_friend_add(tox_node_get_tox(self), address, (const uint8_t *)FR_MESSAGE, sizeof(FR_MESSAGE), &err);
|
||||
if (err != TOX_ERR_FRIEND_ADD_OK) {
|
||||
tox_node_log(self, "Failed to add friend: %u", err);
|
||||
return;
|
||||
}
|
||||
|
||||
WAIT_UNTIL(tox_node_is_friend_connected(self, 0));
|
||||
tox_node_log(self, "Connected to receiver");
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
ToxScenario *s = tox_scenario_new(argc, argv, 300000);
|
||||
ReceiverState receiver_state = {0};
|
||||
|
||||
ToxNode *receiver = tox_scenario_add_node(s, "Receiver", receiver_script, &receiver_state, sizeof(ReceiverState));
|
||||
ToxNode *senders[FR_TOX_COUNT - 1];
|
||||
|
||||
for (int i = 0; i < FR_TOX_COUNT - 1; ++i) {
|
||||
char alias[16];
|
||||
snprintf(alias, sizeof(alias), "Sender%d", i);
|
||||
senders[i] = tox_scenario_add_node(s, alias, sender_script, receiver, 0);
|
||||
tox_node_bootstrap(senders[i], receiver);
|
||||
tox_node_bootstrap(receiver, senders[i]);
|
||||
}
|
||||
|
||||
ToxScenarioStatus res = tox_scenario_run(s);
|
||||
if (res != TOX_SCENARIO_DONE) {
|
||||
tox_scenario_log(s, "Test failed with status %u", res);
|
||||
return 1;
|
||||
}
|
||||
|
||||
tox_scenario_free(s);
|
||||
return 0;
|
||||
}
|
||||
|
||||
#undef FR_MESSAGE
|
||||
Reference in New Issue
Block a user