tomato/auto_tests/netprof_test.c

/** Auto Tests: basic network profile functionality test (UDP only)
 */

#include <inttypes.h>
#include <stdint.h>
#include <stdio.h>

#include "../toxcore/tox_private.h"
#include "../toxcore/util.h"

#include "auto_test_support.h"
#include "check_compat.h"

#define NUM_TOXES 2

static void test_netprof(AutoTox *autotoxes)
{
    // Send some messages to create fake traffic
    for (size_t i = 0; i < 256; ++i) {
        for (uint32_t j = 0; j < NUM_TOXES; ++j) {
            tox_friend_send_message(autotoxes[j].tox, 0, TOX_MESSAGE_TYPE_NORMAL, (const uint8_t *)"test", 4, nullptr);
        }

        iterate_all_wait(autotoxes, NUM_TOXES, ITERATION_INTERVAL);
    }

    // idle traffic for a while
    for (size_t i = 0; i < 100; ++i) {
        iterate_all_wait(autotoxes, NUM_TOXES, ITERATION_INTERVAL);
    }

    const Tox *tox1 = autotoxes[0].tox;

    const uint64_t UDP_count_sent1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
                                     TOX_NETPROF_DIRECTION_SENT);
    const uint64_t UDP_count_recv1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
                                     TOX_NETPROF_DIRECTION_RECV);
    const uint64_t TCP_count_sent1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
                                     TOX_NETPROF_DIRECTION_SENT);
    const uint64_t TCP_count_recv1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
                                     TOX_NETPROF_DIRECTION_RECV);

    const uint64_t UDP_bytes_sent1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
                                     TOX_NETPROF_DIRECTION_SENT);
    const uint64_t UDP_bytes_recv1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
                                     TOX_NETPROF_DIRECTION_RECV);
    const uint64_t TCP_bytes_sent1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
                                     TOX_NETPROF_DIRECTION_SENT);
    const uint64_t TCP_bytes_recv1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
                                     TOX_NETPROF_DIRECTION_RECV);

    ck_assert(UDP_count_recv1 > 0 && UDP_count_sent1 > 0);
    ck_assert(UDP_bytes_recv1 > 0 && UDP_bytes_sent1 > 0);

    (void)TCP_count_sent1;
    (void)TCP_bytes_sent1;
    (void)TCP_bytes_recv1;
    (void)TCP_count_recv1;

    uint64_t total_sent_count = 0;
    uint64_t total_recv_count = 0;
    uint64_t total_sent_bytes = 0;
    uint64_t total_recv_bytes = 0;

    // tox1 makes sure the sum value of all packet ID's is equal to the totals
    for (size_t i = 0; i < 256; ++i) {
        // this id isn't valid for UDP packets but we still want to call the
        // functions and make sure they return some non-zero value
        if (i == TOX_NETPROF_PACKET_ID_TCP_DATA) {
            ck_assert(tox_netprof_get_packet_id_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                      TOX_NETPROF_DIRECTION_SENT) > 0);
            ck_assert(tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                      TOX_NETPROF_DIRECTION_SENT) > 0);
            ck_assert(tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                      TOX_NETPROF_DIRECTION_SENT) > 0);
            ck_assert(tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                      TOX_NETPROF_DIRECTION_RECV) > 0);
            continue;
        }

        total_sent_count += tox_netprof_get_packet_id_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                            TOX_NETPROF_DIRECTION_SENT);
        total_recv_count += tox_netprof_get_packet_id_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                            TOX_NETPROF_DIRECTION_RECV);

        total_sent_bytes += tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                            TOX_NETPROF_DIRECTION_SENT);
        total_recv_bytes += tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
                            TOX_NETPROF_DIRECTION_RECV);
    }

    const uint64_t total_packets = total_sent_count + total_recv_count;
    ck_assert_msg(total_packets == UDP_count_sent1 + UDP_count_recv1,
                  "%" PRIu64 "does not match %" PRIu64 "\n", total_packets, UDP_count_sent1 + UDP_count_recv1);

    ck_assert_msg(total_sent_count == UDP_count_sent1, "%" PRIu64 " does not match %" PRIu64 "\n", total_sent_count, UDP_count_sent1);
    ck_assert_msg(total_recv_count == UDP_count_recv1, "%" PRIu64 " does not match %" PRIu64"\n", total_recv_count, UDP_count_recv1);


    const uint64_t total_bytes = total_sent_bytes + total_recv_bytes;
    ck_assert_msg(total_bytes == UDP_bytes_sent1 + UDP_bytes_recv1,
                  "%" PRIu64 "does not match %" PRIu64 "\n", total_bytes, UDP_bytes_sent1 + UDP_bytes_recv1);

    ck_assert_msg(total_sent_bytes == UDP_bytes_sent1, "%" PRIu64 " does not match %" PRIu64 "\n", total_sent_bytes, UDP_bytes_sent1);
    ck_assert_msg(total_recv_bytes == UDP_bytes_recv1, "%" PRIu64 " does not match %" PRIu64 "\n", total_recv_bytes, UDP_bytes_recv1);
}

int main(void)
{
    setvbuf(stdout, nullptr, _IONBF, 0);

    Tox_Err_Options_New options_err;
    struct Tox_Options *tox_opts = tox_options_new(&options_err);

    ck_assert_msg(options_err == TOX_ERR_OPTIONS_NEW_OK, "Failed to initialize tox options: %u\n", options_err);

    tox_options_default(tox_opts);
    tox_options_set_udp_enabled(tox_opts, true);

    Run_Auto_Options autotox_opts = default_run_auto_options();
    autotox_opts.graph = GRAPH_COMPLETE;

    run_auto_test(tox_opts, NUM_TOXES, test_netprof, 0, &autotox_opts);

    // TODO(Jfreegman): uncomment this when TCP autotests are fixed
    // tox_options_set_udp_enabled(tox_opts, false);
    // run_auto_test(tox_opts, NUM_TOXES, test_netprof, 0, &autotox_opts);

    tox_options_free(tox_opts);

    return 0;
}

#undef NUM_TOXES