toxcore/auto_tests/messenger_test.c

385 lines
13 KiB
C
Raw Normal View History

2013-08-07 10:07:13 +08:00
/* unit tests for /core/Messenger.c
* Design:
* Just call every non-static function in Messenger.c, checking that
* they return as they should with check calls. "Bad" calls of the type
* function(bad_data, good_length) are _not_ checked for, this type
* of call is the fault of the client code.
*
* Note:
* None of the functions here test things that rely on the network, i.e.
* checking that status changes are received, messages can be sent, etc.
2013-08-07 10:07:13 +08:00
* All of that is done in a separate test, with two local clients running. */
2013-09-11 05:15:26 +08:00
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "check_compat.h"
2016-12-19 23:27:46 +08:00
#include "helpers.h"
#include "../testing/misc_tools.c" // hex_string_to_bin
#include "../toxcore/Messenger.h"
2016-12-19 23:27:46 +08:00
2013-08-07 10:07:13 +08:00
#include <stdint.h>
#include <string.h>
2016-09-01 07:33:20 +08:00
#include <sys/types.h>
2013-08-07 10:07:13 +08:00
2016-12-19 23:27:46 +08:00
#if VANILLA_NACL
#include <crypto_box.h> // crypto_box_PUBLICKEYBYTES and other defines.
#else
#include <sodium.h>
#endif
2013-08-07 10:07:13 +08:00
#define REALLY_BIG_NUMBER ((1) << (sizeof(uint16_t) * 7))
#define STRINGS_EQUAL(X, Y) (strcmp(X, Y) == 0)
2016-12-19 23:27:46 +08:00
static bool enable_broken_tests = false;
static const char *friend_id_str = "e4b3d5030bc99494605aecc33ceec8875640c1d74aa32790e821b17e98771c4a00000000f1db";
2013-08-07 10:07:13 +08:00
/* in case we need more than one ID for a test */
static const char *good_id_a_str = "DB9B569D14850ED8364C3744CAC2C8FF78985D213E980C7C508D0E91E8E45441";
static const char *good_id_b_str = "d3f14b6d384d8f5f2a66cff637e69f28f539c5de61bc29744785291fa4ef4d64";
2013-08-07 10:07:13 +08:00
static const char *bad_id_str = "9B569D14ff637e69f2";
2013-08-07 10:07:13 +08:00
static unsigned char *friend_id = NULL;
static unsigned char *good_id_a = NULL;
static unsigned char *good_id_b = NULL;
static unsigned char *bad_id = NULL;
2013-08-07 10:07:13 +08:00
static int friend_id_num = 0;
2013-08-07 10:07:13 +08:00
static Messenger *m;
2013-08-07 10:07:13 +08:00
START_TEST(test_m_sendmesage)
{
const char *message = "h-hi :3";
2013-08-07 10:07:13 +08:00
int good_len = strlen(message);
2014-05-03 09:44:09 +08:00
int bad_len = MAX_CRYPTO_PACKET_SIZE;
2013-08-07 10:07:13 +08:00
ck_assert(m_send_message_generic(m, -1, MESSAGE_NORMAL, (const uint8_t *)message, good_len, 0) == -1);
ck_assert(m_send_message_generic(m, REALLY_BIG_NUMBER, MESSAGE_NORMAL, (const uint8_t *)message, good_len, 0) == -1);
ck_assert(m_send_message_generic(m, 17, MESSAGE_NORMAL, (const uint8_t *)message, good_len, 0) == -1);
ck_assert(m_send_message_generic(m, friend_id_num, MESSAGE_NORMAL, (const uint8_t *)message, bad_len, 0) == -2);
2013-08-07 10:07:13 +08:00
}
END_TEST
START_TEST(test_m_get_userstatus_size)
{
int rc = 0;
ck_assert_msg((m_get_statusmessage_size(m, -1) == -1),
2013-08-17 01:11:09 +08:00
"m_get_statusmessage_size did NOT catch an argument of -1");
ck_assert_msg((m_get_statusmessage_size(m, REALLY_BIG_NUMBER) == -1),
2013-08-17 01:11:09 +08:00
"m_get_statusmessage_size did NOT catch the following argument: %d\n",
REALLY_BIG_NUMBER);
rc = m_get_statusmessage_size(m, friend_id_num);
2013-08-07 10:07:13 +08:00
2013-08-07 16:04:16 +08:00
/* this WILL error if the original m_addfriend_norequest() failed */
2015-02-26 10:09:14 +08:00
ck_assert_msg((rc >= 0 && rc <= MAX_STATUSMESSAGE_LENGTH),
"m_get_statusmessage_size is returning out of range values! (%i)\n"
2013-08-17 01:11:09 +08:00
"(this can be caused by the error of m_addfriend_norequest"
2015-02-26 10:09:14 +08:00
" in the beginning of the suite)\n", rc);
2013-08-07 10:07:13 +08:00
}
END_TEST
START_TEST(test_m_set_userstatus)
{
const char *status = "online!";
2013-08-07 10:07:13 +08:00
uint16_t good_length = strlen(status);
uint16_t bad_length = REALLY_BIG_NUMBER;
ck_assert_msg((m_set_statusmessage(m, (const uint8_t *)status, bad_length) == -1),
2013-08-18 16:48:36 +08:00
"m_set_userstatus did NOT catch the following length: %d\n",
REALLY_BIG_NUMBER);
2013-08-07 10:07:13 +08:00
ck_assert_msg((m_set_statusmessage(m, (const uint8_t *)status, good_length) == 0),
2013-08-18 16:48:36 +08:00
"m_set_userstatus did NOT return 0 on the following length: %d\n"
"MAX_STATUSMESSAGE_LENGTH: %d\n", good_length, MAX_STATUSMESSAGE_LENGTH);
2013-08-07 10:07:13 +08:00
}
END_TEST
2013-09-08 04:05:48 +08:00
START_TEST(test_m_get_friend_connectionstatus)
2013-08-07 10:07:13 +08:00
{
2013-09-08 04:05:48 +08:00
ck_assert_msg((m_get_friend_connectionstatus(m, -1) == -1),
"m_get_friend_connectionstatus did NOT catch an argument of -1.\n");
ck_assert_msg((m_get_friend_connectionstatus(m, REALLY_BIG_NUMBER) == -1),
"m_get_friend_connectionstatus did NOT catch an argument of %d.\n",
REALLY_BIG_NUMBER);
}
END_TEST
START_TEST(test_m_friend_exists)
{
ck_assert_msg((m_friend_exists(m, -1) == 0),
"m_friend_exists did NOT catch an argument of -1.\n");
ck_assert_msg((m_friend_exists(m, REALLY_BIG_NUMBER) == 0),
"m_friend_exists did NOT catch an argument of %d.\n",
2013-08-17 01:11:09 +08:00
REALLY_BIG_NUMBER);
2013-08-07 10:07:13 +08:00
}
END_TEST
START_TEST(test_m_delfriend)
{
ck_assert_msg((m_delfriend(m, -1) == -1),
2013-08-17 01:11:09 +08:00
"m_delfriend did NOT catch an argument of -1\n");
ck_assert_msg((m_delfriend(m, REALLY_BIG_NUMBER) == -1),
2013-08-17 01:11:09 +08:00
"m_delfriend did NOT catch the following number: %d\n",
REALLY_BIG_NUMBER);
2013-08-07 10:07:13 +08:00
}
END_TEST
2013-08-07 10:07:13 +08:00
START_TEST(test_m_addfriend)
{
2016-12-19 23:27:46 +08:00
const char *good_data = "test";
const char *bad_data = "";
2013-08-07 10:07:13 +08:00
int good_len = strlen(good_data);
int bad_len = strlen(bad_data);
2014-05-03 09:44:09 +08:00
int really_bad_len = (MAX_CRYPTO_PACKET_SIZE - crypto_box_PUBLICKEYBYTES
- crypto_box_NONCEBYTES - crypto_box_BOXZEROBYTES
+ crypto_box_ZEROBYTES + 100);
/* TODO(irungentoo): Update this properly to latest master */
2016-12-19 23:27:46 +08:00
if (m_addfriend(m, (const uint8_t *)friend_id, (const uint8_t *)good_data, really_bad_len) != FAERR_TOOLONG) {
2013-08-07 10:07:13 +08:00
ck_abort_msg("m_addfriend did NOT catch the following length: %d\n", really_bad_len);
}
/* this will return an error if the original m_addfriend_norequest() failed */
2016-12-19 23:27:46 +08:00
if (m_addfriend(m, (const uint8_t *)friend_id, (const uint8_t *)good_data, good_len) != FAERR_ALREADYSENT) {
2013-08-07 16:04:16 +08:00
ck_abort_msg("m_addfriend did NOT catch adding a friend we already have.\n"
"(this can be caused by the error of m_addfriend_norequest in"
" the beginning of the suite)\n");
}
2013-08-07 10:07:13 +08:00
2016-12-19 23:27:46 +08:00
if (m_addfriend(m, (const uint8_t *)good_id_b, (const uint8_t *)bad_data, bad_len) != FAERR_NOMESSAGE) {
2013-08-07 10:07:13 +08:00
ck_abort_msg("m_addfriend did NOT catch the following length: %d\n", bad_len);
}
/* this should REALLY return an error */
/* TODO(irungentoo): validate client_id in m_addfriend? */
2016-12-19 23:27:46 +08:00
if (m_addfriend(m, (const uint8_t *)bad_id, (const uint8_t *)good_data, good_len) >= 0) {
ck_abort_msg("The following ID passed through "
"m_addfriend without an error:\n'%s'\n", bad_id_str);
}
2013-08-07 10:07:13 +08:00
}
END_TEST
2013-08-07 10:07:13 +08:00
START_TEST(test_setname)
{
const char *good_name = "consensualCorn";
2013-08-07 10:07:13 +08:00
int good_length = strlen(good_name);
int bad_length = REALLY_BIG_NUMBER;
ck_assert_msg((setname(m, (const uint8_t *)good_name, bad_length) == -1),
2013-08-18 16:48:36 +08:00
"setname() did NOT error on %d as a length argument!\n", bad_length);
2013-08-17 01:11:09 +08:00
ck_assert_msg((setname(m, (const uint8_t *)good_name, good_length) == 0),
2013-08-18 16:48:36 +08:00
"setname() did NOT return 0 on good arguments!\n");
2013-08-07 10:07:13 +08:00
}
END_TEST
START_TEST(test_getself_name)
{
const char *nickname = "testGallop";
2013-08-07 10:07:13 +08:00
int len = strlen(nickname);
VLA(char, nick_check, len);
2013-08-07 10:07:13 +08:00
setname(m, (const uint8_t *)nickname, len);
2014-02-23 00:52:46 +08:00
getself_name(m, (uint8_t *)nick_check);
2013-08-07 10:07:13 +08:00
2013-09-01 00:49:22 +08:00
ck_assert_msg((memcmp(nickname, nick_check, len) == 0),
2013-08-17 01:11:09 +08:00
"getself_name failed to return the known name!\n"
"known name: %s\nreturned: %s\n", nickname, nick_check);
2013-08-07 10:07:13 +08:00
}
END_TEST
/* this test is excluded for now, due to lack of a way
* to set a friend's status for now.
* ideas:
* if we have access to the friends list, we could
* just add a status manually ourselves. */
2016-12-19 23:27:46 +08:00
#if 0
2013-08-07 10:07:13 +08:00
START_TEST(test_m_copy_userstatus)
{
assert(m_copy_userstatus(-1, buf, MAX_USERSTATUS_LENGTH) == -1);
assert(m_copy_userstatus(REALLY_BIG_NUMBER, buf, MAX_USERSTATUS_LENGTH) == -1);
m_copy_userstatus(friend_id_num, buf, MAX_USERSTATUS_LENGTH + 6);
assert(STRINGS_EQUAL(name_buf, friend_id_status));
}
END_TEST
2016-12-19 23:27:46 +08:00
#endif
2013-08-07 10:07:13 +08:00
START_TEST(test_getname)
{
uint8_t name_buf[MAX_NAME_LENGTH];
2013-08-18 05:23:54 +08:00
uint8_t test_name[] = {'f', 'o', 'o'};
ck_assert(getname(m, -1, name_buf) == -1);
ck_assert(getname(m, REALLY_BIG_NUMBER, name_buf) == -1);
2013-08-07 10:07:13 +08:00
2013-08-18 05:23:54 +08:00
memcpy(m->friendlist[0].name, &test_name[0], 3);
m->friendlist[0].name_length = 4;
ck_assert(getname(m, 0, &name_buf[0]) == 4);
2013-08-07 10:07:13 +08:00
2013-08-18 05:23:54 +08:00
ck_assert(strcmp((char *)&name_buf[0], "foo") == 0);
2013-08-07 10:07:13 +08:00
}
END_TEST
START_TEST(test_dht_state_saveloadsave)
{
/* validate that:
* a) saving stays within the confined space
* b) a save()d state can be load()ed back successfully
* c) a second save() is of equal size
* d) the second save() is of equal content */
size_t i, extra = 64;
size_t size = DHT_size(m->dht);
VLA(uint8_t, buffer, size + 2 * extra);
memset(buffer, 0xCD, extra);
memset(buffer + extra + size, 0xCD, extra);
DHT_save(m->dht, buffer + extra);
2013-09-18 04:28:39 +08:00
for (i = 0; i < extra; i++) {
ck_assert_msg(buffer[i] == 0xCD, "Buffer underwritten from DHT_save() @%u", i);
ck_assert_msg(buffer[extra + size + i] == 0xCD, "Buffer overwritten from DHT_save() @%u", i);
}
2013-11-11 08:45:24 +08:00
int res = DHT_load(m->dht, buffer + extra, size);
2013-09-18 04:28:39 +08:00
2016-09-01 02:12:19 +08:00
if (res == -1) {
ck_assert_msg(res == 0, "Failed to load back stored buffer: res == -1");
2016-09-01 02:12:19 +08:00
} else {
char msg[128];
size_t offset = res >> 4;
uint8_t *ptr = buffer + extra + offset;
sprintf(msg, "Failed to load back stored buffer: 0x%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx @%zu/%zu, code %d",
ptr[-2], ptr[-1], ptr[0], ptr[1], ptr[2], ptr[3], ptr[4], ptr[5], offset, size, res & 0x0F);
ck_assert_msg(res == 0, msg);
}
size_t size2 = DHT_size(m->dht);
ck_assert_msg(size == size2, "Messenger \"grew\" in size from a store/load cycle: %u -> %u", size, size2);
VLA(uint8_t, buffer2, size2);
DHT_save(m->dht, buffer2);
ck_assert_msg(!memcmp(buffer + extra, buffer2, size), "DHT state changed by store/load/store cycle");
}
END_TEST
START_TEST(test_messenger_state_saveloadsave)
{
/* validate that:
* a) saving stays within the confined space
* b) a save()d state can be load()ed back successfully
* c) a second save() is of equal size
* d) the second save() is of equal content */
size_t i, extra = 64;
2013-10-24 02:49:59 +08:00
size_t size = messenger_size(m);
VLA(uint8_t, buffer, size + 2 * extra);
memset(buffer, 0xCD, extra);
memset(buffer + extra + size, 0xCD, extra);
2013-10-24 02:49:59 +08:00
messenger_save(m, buffer + extra);
2013-09-18 04:28:39 +08:00
for (i = 0; i < extra; i++) {
2013-10-24 02:49:59 +08:00
ck_assert_msg(buffer[i] == 0xCD, "Buffer underwritten from messenger_save() @%u", i);
ck_assert_msg(buffer[extra + size + i] == 0xCD, "Buffer overwritten from messenger_save() @%u", i);
}
2013-10-24 02:49:59 +08:00
int res = messenger_load(m, buffer + extra, size);
2013-09-18 04:28:39 +08:00
2016-09-01 02:12:19 +08:00
if (res == -1) {
ck_assert_msg(res == 0, "Failed to load back stored buffer: res == -1");
2016-09-01 02:12:19 +08:00
} else {
char msg[128];
size_t offset = res >> 4;
uint8_t *ptr = buffer + extra + offset;
sprintf(msg, "Failed to load back stored buffer: 0x%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx @%zu/%zu, code %d",
ptr[-2], ptr[-1], ptr[0], ptr[1], ptr[2], ptr[3], ptr[4], ptr[5], offset, size, res & 0x0F);
ck_assert_msg(res == 0, msg);
}
2013-10-24 02:49:59 +08:00
size_t size2 = messenger_size(m);
ck_assert_msg(size == size2, "Messenger \"grew\" in size from a store/load cycle: %u -> %u", size, size2);
VLA(uint8_t, buffer2, size2);
2013-10-24 02:49:59 +08:00
messenger_save(m, buffer2);
ck_assert_msg(!memcmp(buffer + extra, buffer2, size), "Messenger state changed by store/load/store cycle");
}
END_TEST
static Suite *messenger_suite(void)
2013-08-07 10:07:13 +08:00
{
Suite *s = suite_create("Messenger");
2013-09-27 17:24:33 +08:00
DEFTESTCASE(dht_state_saveloadsave);
DEFTESTCASE(messenger_state_saveloadsave);
DEFTESTCASE(getself_name);
DEFTESTCASE(m_get_userstatus_size);
DEFTESTCASE(m_set_userstatus);
2016-12-19 23:27:46 +08:00
if (enable_broken_tests) {
DEFTESTCASE(m_addfriend);
}
2013-09-27 17:24:33 +08:00
DEFTESTCASE(m_friend_exists);
DEFTESTCASE(m_get_friend_connectionstatus);
DEFTESTCASE(m_delfriend);
DEFTESTCASE(setname);
DEFTESTCASE(getname);
DEFTESTCASE(m_sendmesage);
2013-08-07 10:07:13 +08:00
return s;
}
int main(int argc, char *argv[])
{
Suite *messenger = messenger_suite();
SRunner *test_runner = srunner_create(messenger);
int number_failed = 0;
friend_id = hex_string_to_bin(friend_id_str);
good_id_a = hex_string_to_bin(good_id_a_str);
good_id_b = hex_string_to_bin(good_id_b_str);
bad_id = hex_string_to_bin(bad_id_str);
/* IPv6 status from global define */
Messenger_Options options = {0};
options.ipv6enabled = TOX_ENABLE_IPV6_DEFAULT;
options.log_callback = (logger_cb *)print_debug_log;
m = new_messenger(&options, 0);
2013-08-07 10:07:13 +08:00
/* setup a default friend and friendnum */
2016-09-01 02:12:19 +08:00
if (m_addfriend_norequest(m, (uint8_t *)friend_id) < 0) {
2013-08-07 16:04:16 +08:00
fputs("m_addfriend_norequest() failed on a valid ID!\n"
"this was CRITICAL to the test, and the build WILL fail.\n"
"the tests will continue now...\n\n", stderr);
2016-09-01 02:12:19 +08:00
}
2013-08-07 16:04:16 +08:00
2016-09-01 02:12:19 +08:00
if ((friend_id_num = getfriend_id(m, (uint8_t *)friend_id)) < 0) {
2013-08-07 16:04:16 +08:00
fputs("getfriend_id() failed on a valid ID!\n"
"this was CRITICAL to the test, and the build WILL fail.\n"
"the tests will continue now...\n\n", stderr);
2016-09-01 02:12:19 +08:00
}
2013-08-07 10:07:13 +08:00
srunner_run_all(test_runner, CK_NORMAL);
number_failed = srunner_ntests_failed(test_runner);
srunner_free(test_runner);
free(friend_id);
free(good_id_a);
free(good_id_b);
free(bad_id);
2013-10-24 02:49:59 +08:00
kill_messenger(m);
2013-08-07 10:07:13 +08:00
return number_failed;
}