Add a test to try and overflow the send queue in net_crypto.

1 files changed, 68 insertions, 0 deletions

diff --git a/auto_tests/overflow_recvq_test.c b/auto_tests/overflow_recvq_test.c
new file mode 100644
index 00000000..3227c5f4
--- /dev/null
+++ b/auto_tests/overflow_recvq_test.c
@@ -0,0 +1,68 @@
+/* Try to overflow the net_crypto packet buffer.
+ */
+
+#ifndef _XOPEN_SOURCE
+#define _XOPEN_SOURCE 600
+#endif
+
+#include "check_compat.h"
+
+#include "../toxcore/tox.h"
+
+typedef struct State {
+    uint32_t index;
+    uint32_t recv_count;
+} State;
+
+#include "run_auto_test.h"
+
+#define NUM_MSGS 40000
+
+static void handle_friend_message(Tox *tox, uint32_t friend_number, TOX_MESSAGE_TYPE type,
+                                  const uint8_t *message, size_t length, void *user_data)
+{
+    State *state = (State *)user_data;
+    state->recv_count++;
+}
+
+static void net_crypto_overflow_test(Tox **toxes, State *state)
+{
+    tox_callback_friend_message(toxes[0], handle_friend_message);
+
+    printf("sending many messages to tox0\n");
+
+    for (uint32_t tox_index = 1; tox_index < 3; tox_index++) {
+        for (uint32_t i = 0; i < NUM_MSGS; i++) {
+            uint8_t message[128] = {0};
+            snprintf((char *)message, sizeof(message), "%u-%u", tox_index, i);
+
+            TOX_ERR_FRIEND_SEND_MESSAGE err;
+            tox_friend_send_message(toxes[tox_index], 0, TOX_MESSAGE_TYPE_NORMAL, message, sizeof message, &err);
+
+            if (err == TOX_ERR_FRIEND_SEND_MESSAGE_SENDQ) {
+                printf("tox%u sent %u messages to friend 0\n", tox_index, i);
+                break;
+            }
+
+            ck_assert_msg(err == TOX_ERR_FRIEND_SEND_MESSAGE_OK,
+                          "tox%u failed to send message number %u: %d", tox_index, i, err);
+        }
+    }
+
+    // TODO(iphydf): Wait until all messages have arrived. Currently, not all
+    // messages arrive, so this test would always fail.
+    for (uint32_t i = 0; i < 200; i++) {
+        iterate_all(3, toxes, state);
+        c_sleep(ITERATION_INTERVAL);
+    }
+
+    printf("tox%u received %u messages\n", state[0].index, state[0].recv_count);
+}
+
+int main(void)
+{
+    setvbuf(stdout, nullptr, _IONBF, 0);
+
+    run_auto_test(3, net_crypto_overflow_test);
+    return 0;
+}