433 lines
10 KiB
C
433 lines
10 KiB
C
/*
|
|
* Copyright (C) 2013 Tobias Brunner
|
|
* Hochschule fuer Technik Rapperswil
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* for more details.
|
|
*/
|
|
|
|
#include "test_suite.h"
|
|
|
|
#include <library.h>
|
|
#include <utils/utils.h>
|
|
|
|
/*******************************************************************************
|
|
* object storage on lib
|
|
*/
|
|
|
|
START_TEST(test_objects)
|
|
{
|
|
char *k1 = "key1", *k2 = "key2";
|
|
char *v1 = "val1", *val;
|
|
|
|
ck_assert(lib->get(lib, k1) == NULL);
|
|
|
|
ck_assert(lib->set(lib, k1, v1));
|
|
ck_assert(!lib->set(lib, k1, v1));
|
|
|
|
val = lib->get(lib, k1);
|
|
ck_assert(val != NULL);
|
|
ck_assert(streq(val, v1));
|
|
|
|
ck_assert(lib->set(lib, k1, NULL));
|
|
ck_assert(!lib->set(lib, k2, NULL));
|
|
|
|
ck_assert(lib->get(lib, k1) == NULL);
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* test return_... functions
|
|
*/
|
|
|
|
START_TEST(test_return_functions)
|
|
{
|
|
ck_assert(return_null() == NULL);
|
|
ck_assert(return_null("asdf", 5, NULL, 1, "qwer") == NULL);
|
|
|
|
ck_assert(return_true() == TRUE);
|
|
ck_assert(return_true("asdf", 5, NULL, 1, "qwer") == TRUE);
|
|
|
|
ck_assert(return_false() == FALSE);
|
|
ck_assert(return_false("asdf", 5, NULL, 1, "qwer") == FALSE);
|
|
|
|
ck_assert(return_failed() == FAILED);
|
|
ck_assert(return_failed("asdf", 5, NULL, 1, "qwer") == FAILED);
|
|
|
|
ck_assert(return_success() == SUCCESS);
|
|
ck_assert(return_success("asdf", 5, NULL, 1, "qwer") == SUCCESS);
|
|
|
|
/* just make sure this works */
|
|
nop();
|
|
nop("asdf", 5, NULL, 1, "qwer");
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* timeval_add_ms
|
|
*/
|
|
|
|
START_TEST(test_timeval_add_ms)
|
|
{
|
|
timeval_t tv;
|
|
|
|
tv.tv_sec = 0;
|
|
tv.tv_usec = 0;
|
|
timeval_add_ms(&tv, 0);
|
|
ck_assert_int_eq(tv.tv_sec, 0);
|
|
ck_assert_int_eq(tv.tv_usec, 0);
|
|
|
|
timeval_add_ms(&tv, 1);
|
|
ck_assert_int_eq(tv.tv_sec, 0);
|
|
ck_assert_int_eq(tv.tv_usec, 1000);
|
|
|
|
timeval_add_ms(&tv, 0);
|
|
ck_assert_int_eq(tv.tv_sec, 0);
|
|
ck_assert_int_eq(tv.tv_usec, 1000);
|
|
|
|
timeval_add_ms(&tv, 999);
|
|
ck_assert_int_eq(tv.tv_sec, 1);
|
|
ck_assert_int_eq(tv.tv_usec, 0);
|
|
|
|
timeval_add_ms(&tv, 0);
|
|
ck_assert_int_eq(tv.tv_sec, 1);
|
|
ck_assert_int_eq(tv.tv_usec, 0);
|
|
|
|
timeval_add_ms(&tv, 1000);
|
|
ck_assert_int_eq(tv.tv_sec, 2);
|
|
ck_assert_int_eq(tv.tv_usec, 0);
|
|
|
|
timeval_add_ms(&tv, 1500);
|
|
ck_assert_int_eq(tv.tv_sec, 3);
|
|
ck_assert_int_eq(tv.tv_usec, 500000);
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* htoun/untoh
|
|
*/
|
|
|
|
START_TEST(test_htoun)
|
|
{
|
|
chunk_t net64, expected;
|
|
u_int16_t host16 = 513;
|
|
u_int32_t net16 = 0, host32 = 67305985;
|
|
u_int64_t net32 = 0, host64 = 578437695752307201;
|
|
|
|
net64 = chunk_alloca(16);
|
|
memset(net64.ptr, 0, net64.len);
|
|
|
|
expected = chunk_from_chars(0x00, 0x02, 0x01, 0x00);
|
|
htoun16((char*)&net16 + 1, host16);
|
|
ck_assert(chunk_equals(expected, chunk_from_thing(net16)));
|
|
|
|
expected = chunk_from_chars(0x00, 0x00, 0x04, 0x03, 0x02, 0x01, 0x00, 0x00);
|
|
htoun32((u_int16_t*)&net32 + 1, host32);
|
|
ck_assert(chunk_equals(expected, chunk_from_thing(net32)));
|
|
|
|
expected = chunk_from_chars(0x00, 0x00, 0x00, 0x00,
|
|
0x08, 0x07, 0x06, 0x05,
|
|
0x04, 0x03, 0x02, 0x01,
|
|
0x00, 0x00, 0x00, 0x00);
|
|
htoun64((u_int32_t*)net64.ptr + 1, host64);
|
|
ck_assert(chunk_equals(expected, net64));
|
|
}
|
|
END_TEST
|
|
|
|
START_TEST(test_untoh)
|
|
{
|
|
chunk_t net;
|
|
u_int16_t host16;
|
|
u_int32_t host32;
|
|
u_int64_t host64;
|
|
|
|
net = chunk_from_chars(0x00, 0x02, 0x01, 0x00);
|
|
host16 = untoh16(net.ptr + 1);
|
|
ck_assert(host16 == 513);
|
|
|
|
net = chunk_from_chars(0x00, 0x00, 0x04, 0x03, 0x02, 0x01, 0x00, 0x00);
|
|
host32 = untoh32(net.ptr + 2);
|
|
ck_assert(host32 == 67305985);
|
|
|
|
net = chunk_from_chars(0x00, 0x00, 0x00, 0x00, 0x08, 0x07, 0x06, 0x05,
|
|
0x04, 0x03, 0x02, 0x01, 0x00, 0x00, 0x00, 0x00);
|
|
host64 = untoh64(net.ptr + 4);
|
|
ck_assert(host64 == 578437695752307201);
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* memxor
|
|
*/
|
|
|
|
static void do_memxor(chunk_t a, chunk_t b, chunk_t exp)
|
|
{
|
|
chunk_t dst;
|
|
|
|
dst = chunk_clonea(a);
|
|
dst.len = b.len;
|
|
memxor(dst.ptr, b.ptr, b.len);
|
|
ck_assert(chunk_equals(dst, exp));
|
|
}
|
|
|
|
START_TEST(test_memxor)
|
|
{
|
|
chunk_t a, b, dst;
|
|
int i;
|
|
|
|
a = chunk_alloca(64);
|
|
memset(a.ptr, 0, a.len);
|
|
b = chunk_alloca(64);
|
|
for (i = 0; i < 64; i++)
|
|
{
|
|
b.ptr[i] = i;
|
|
b.len = i;
|
|
do_memxor(a, b, b);
|
|
}
|
|
b.len = 64;
|
|
do_memxor(a, b, b);
|
|
|
|
dst = chunk_clonea(a);
|
|
memxor(dst.ptr, b.ptr, b.len);
|
|
ck_assert(chunk_equals(dst, b));
|
|
|
|
memxor(dst.ptr, b.ptr, 0);
|
|
memxor(dst.ptr, b.ptr, 1);
|
|
memxor(dst.ptr + 1, b.ptr + 1, 1);
|
|
memxor(dst.ptr + 2, b.ptr + 2, b.len - 2);
|
|
ck_assert(chunk_equals(dst, a));
|
|
}
|
|
END_TEST
|
|
|
|
START_TEST(test_memxor_aligned)
|
|
{
|
|
u_int64_t a = 0, b = 0;
|
|
chunk_t ca, cb;
|
|
int i;
|
|
|
|
ca = chunk_from_thing(a);
|
|
cb = chunk_from_thing(b);
|
|
|
|
for (i = 0; i < 8; i++)
|
|
{
|
|
cb.ptr[i] = i + 1;
|
|
}
|
|
|
|
/* 64-bit aligned */
|
|
memxor(ca.ptr, cb.ptr, 8);
|
|
ck_assert(a == b);
|
|
/* 32-bit aligned source */
|
|
a = 0;
|
|
memxor(ca.ptr, cb.ptr + 4, 4);
|
|
ck_assert(chunk_equals(ca, chunk_from_chars(0x05, 0x06, 0x07, 0x08,
|
|
0x00, 0x00, 0x00, 0x00)));
|
|
/* 16-bit aligned source */
|
|
a = 0;
|
|
memxor(ca.ptr, cb.ptr + 2, 6);
|
|
ck_assert(chunk_equals(ca, chunk_from_chars(0x03, 0x04, 0x05, 0x06,
|
|
0x07, 0x08, 0x00, 0x00)));
|
|
/* 8-bit aligned source */
|
|
a = 0;
|
|
memxor(ca.ptr, cb.ptr + 1, 7);
|
|
ck_assert(chunk_equals(ca, chunk_from_chars(0x02, 0x03, 0x04, 0x05,
|
|
0x06, 0x07, 0x08, 0x00)));
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* memstr
|
|
*/
|
|
|
|
static struct {
|
|
char *haystack;
|
|
char *needle;
|
|
size_t n;
|
|
int offset;
|
|
} memstr_data[] = {
|
|
{NULL, NULL, 0, -1},
|
|
{NULL, NULL, 3, -1},
|
|
{NULL, "abc", 0, -1},
|
|
{NULL, "abc", 3, -1},
|
|
{"", "", 0, -1},
|
|
{"abc", NULL, 3, -1},
|
|
{"abc", "", 3, -1},
|
|
{"abc", "abc", 3, 0},
|
|
{" abc", "abc", 4, 1},
|
|
{" abc", "abc", 3, -1},
|
|
{"abcabc", "abc", 6, 0},
|
|
{" abc ", "abc", 5, 1},
|
|
};
|
|
|
|
START_TEST(test_memstr)
|
|
{
|
|
char *ret;
|
|
|
|
ret = memstr(memstr_data[_i].haystack, memstr_data[_i].needle, memstr_data[_i].n);
|
|
if (memstr_data[_i].offset >= 0)
|
|
{
|
|
ck_assert(ret == memstr_data[_i].haystack + memstr_data[_i].offset);
|
|
}
|
|
else
|
|
{
|
|
ck_assert(ret == NULL);
|
|
}
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* translate
|
|
*/
|
|
|
|
static struct {
|
|
char *in;
|
|
char *from;
|
|
char *to;
|
|
char *out;
|
|
} translate_data[] = {
|
|
{NULL, "", "", NULL},
|
|
{"abc", "", "", "abc"},
|
|
{"abc", "", "x", "abc"},
|
|
{"abc", "x", "", "abc"},
|
|
{"abc", "abc", "xyz", "xyz"},
|
|
{"aabbcc", "abc", "xyz", "xxyyzz"},
|
|
{"abbaccb", "abc", "xyz", "xyyxzzy"},
|
|
{"abxyzc", "abc", "xyz", "xyxyzz"},
|
|
{"abcdef", "abc", "xyz", "xyzdef"},
|
|
{"aaa", "abc", "xyz", "xxx"},
|
|
{"abc", "aaa", "xyz", "xbc"},
|
|
{"abc", "abc", "xxx", "xxx"},
|
|
};
|
|
|
|
START_TEST(test_translate)
|
|
{
|
|
char *str, *ret;
|
|
|
|
str = strdupnull(translate_data[_i].in);
|
|
ret = translate(str, translate_data[_i].from, translate_data[_i].to);
|
|
ck_assert(ret == str);
|
|
if (ret != translate_data[_i].out)
|
|
{
|
|
ck_assert_str_eq(str, translate_data[_i].out);
|
|
}
|
|
free(str);
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* time_printf_hook
|
|
*/
|
|
|
|
static struct {
|
|
time_t in;
|
|
bool utc;
|
|
char *out;
|
|
} time_data[] = {
|
|
{UNDEFINED_TIME, FALSE, "--- -- --:--:-- ----"},
|
|
{UNDEFINED_TIME, TRUE , "--- -- --:--:-- UTC ----"},
|
|
{1, FALSE, "Jan 01 01:00:01 1970"},
|
|
{1, TRUE , "Jan 01 00:00:01 UTC 1970"},
|
|
{1341150196, FALSE, "Jul 01 15:43:16 2012"},
|
|
{1341150196, TRUE , "Jul 01 13:43:16 UTC 2012"},
|
|
};
|
|
|
|
START_TEST(test_time_printf_hook)
|
|
{
|
|
char buf[32];
|
|
int len;
|
|
|
|
len = snprintf(buf, sizeof(buf), "%T", &time_data[_i].in, time_data[_i].utc);
|
|
ck_assert(len >= 0 && len < sizeof(buf));
|
|
ck_assert_str_eq(buf, time_data[_i].out);
|
|
}
|
|
END_TEST
|
|
|
|
/*******************************************************************************
|
|
* time_delta_printf_hook
|
|
*/
|
|
|
|
static struct {
|
|
time_t a;
|
|
time_t b;
|
|
char *out;
|
|
} time_delta_data[] = {
|
|
{0, 0, "0 seconds"},
|
|
{0, 1, "1 second"},
|
|
{0, -1, "1 second"},
|
|
{1, 0, "1 second"},
|
|
{0, 2, "2 seconds"},
|
|
{2, 0, "2 seconds"},
|
|
{0, 60, "60 seconds"},
|
|
{0, 120, "120 seconds"},
|
|
{0, 121, "2 minutes"},
|
|
{0, 3600, "60 minutes"},
|
|
{0, 7200, "120 minutes"},
|
|
{0, 7201, "2 hours"},
|
|
{0, 86400, "24 hours"},
|
|
{0, 172800, "48 hours"},
|
|
{0, 172801, "2 days"},
|
|
{172801, 86400, "24 hours"},
|
|
};
|
|
|
|
START_TEST(test_time_delta_printf_hook)
|
|
{
|
|
char buf[16];
|
|
int len;
|
|
|
|
len = snprintf(buf, sizeof(buf), "%V", &time_delta_data[_i].a, &time_delta_data[_i].b);
|
|
ck_assert(len >= 0 && len < sizeof(buf));
|
|
ck_assert_str_eq(buf, time_delta_data[_i].out);
|
|
}
|
|
END_TEST
|
|
|
|
Suite *utils_suite_create()
|
|
{
|
|
Suite *s;
|
|
TCase *tc;
|
|
|
|
s = suite_create("utils");
|
|
|
|
tc = tcase_create("objects");
|
|
tcase_add_test(tc, test_objects);
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("return functions");
|
|
tcase_add_test(tc, test_return_functions);
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("timeval_add_ms");
|
|
tcase_add_test(tc, test_timeval_add_ms);
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("htoun,untoh");
|
|
tcase_add_test(tc, test_htoun);
|
|
tcase_add_test(tc, test_untoh);
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("memxor");
|
|
tcase_add_test(tc, test_memxor);
|
|
tcase_add_test(tc, test_memxor_aligned);
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("memstr");
|
|
tcase_add_loop_test(tc, test_memstr, 0, countof(memstr_data));
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("translate");
|
|
tcase_add_loop_test(tc, test_translate, 0, countof(translate_data));
|
|
suite_add_tcase(s, tc);
|
|
|
|
tc = tcase_create("printf_hooks");
|
|
tcase_add_loop_test(tc, test_time_printf_hook, 0, countof(time_data));
|
|
tcase_add_loop_test(tc, test_time_delta_printf_hook, 0, countof(time_delta_data));
|
|
suite_add_tcase(s, tc);
|
|
|
|
return s;
|
|
}
|