Revision control
Copy as Markdown
Other Tools
/*
* Copyright (C) 2014 - Julien Voisin <julien.voisin@dustri.org>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License, version 2 only, as
* published by the Free Software Foundation.
*
* 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.
*
* You should have received a copy of the GNU General Public License along with
* this program; if not, write to the Free Software Foundation, Inc., 51
* Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <gcrypt.h>
#include <pthread.h>
#include <proto.h>
#include <tlv.h>
#include <tap/tap.h>
GCRY_THREAD_OPTION_PTHREAD_IMPL;
#define NUM_TESTS 11
static void test_otrl_tlv_new()
{
const unsigned short type = OTRL_TLV_SMP1Q;
const char *data = "This is some test data";
const unsigned short len = strlen(data);
OtrlTLV* tlv = otrl_tlv_new(type, len, (unsigned char*)data);
ok(tlv->type == type &&
tlv->len == len &&
memcmp(tlv->data, data, len) == 0 &&
tlv->data[len] == '\0' &&
tlv->next == NULL,
"TLV created with success");
otrl_tlv_free(tlv);
}
static void test_otrl_tlv_parse()
{
const unsigned char serialized1[] =
{'\x01', '\x02', '\x00', '\x03', '1', '2', '3'};
const unsigned char serialized2[] = {'\x02', '\x04', '\x00', '\x00'};
const unsigned char serialized3[] =
{'\x04', '\x02', '\x00', '\x03', '1', '2', '3',
'\x02', '\x02', '\xff', '\xff', '1', '3', '3', '7'};
const unsigned char serialized4[] =
{'\x04', '\x02', '\x00', '\x03', '1', '2', '3',
'\x02', '\x02', '\x00', '\x04', '1', '3', '3', '7'};
OtrlTLV *tlv1 = otrl_tlv_parse(serialized1, sizeof(serialized1));
OtrlTLV *tlv2 = otrl_tlv_parse(serialized2, sizeof(serialized2));
OtrlTLV *tlv3 = otrl_tlv_parse(serialized3, sizeof(serialized3));
OtrlTLV *tlv4 = otrl_tlv_parse(serialized4, sizeof(serialized4));
ok(tlv1->type == 258 &&
tlv1->len == 3 &&
tlv1->next == NULL &&
memcmp(tlv1->data, "123", tlv1->len) == 0 &&
tlv1->data[tlv1->len] == 0,
"Single-TLV chain constructed with success");
otrl_tlv_free(tlv1);
ok(tlv2->type == 516 &&
tlv2->len == 0 &&
tlv2->next == NULL,
"tlv2 chain with no data constructed with success");
otrl_tlv_free(tlv2);
ok(tlv3->type == 1026 &&
tlv3->len == 3 &&
memcmp(tlv3->data, "123", tlv3->len) == 0 &&
tlv3->data[tlv3->len] == 0 &&
tlv3->next == NULL,
"tlv3 chain with overflow constructed with success");
otrl_tlv_free(tlv3);
ok(tlv4->type == 1026 &&
tlv4->len == 3 &&
memcmp(tlv4->data, "123", tlv4->len) == 0 &&
tlv4->data[tlv4->len] == 0 &&
tlv4->next != NULL,
"First part of the 2-part tlv chain build");
ok(tlv4->next != NULL &&
tlv4->next->type == 514 &&
tlv4->next->len == 4 &&
memcmp(tlv4->next->data, "1337", tlv4->next->len) == 0 &&
tlv4->next->data[tlv4->next->len] == 0 &&
tlv4->next->next == NULL,
"Second part of the 2-part tlv chain build");
otrl_tlv_free(tlv4);
}
static void test_otrl_tlv_seriallen()
{
const unsigned char serialized[] =
{'\x04', '\x02', '\x00', '\x03', '1', '2', '3',
'\x02', '\x02', '\x00', '\x04', '1', '3', '3', '7'};
OtrlTLV* tlv = otrl_tlv_parse(serialized, sizeof(serialized));
ok(otrl_tlv_seriallen(tlv) == 4 + 3 + 4 + 4,
"Size correctly guessed");
otrl_tlv_free(tlv);
}
static void test_otrl_tlv_serialize()
{
const unsigned char serialized[] =
{'\x04', '\x02', '\x00', '\x03', '1', '2', '3',
'\x02', '\x02', '\x00', '\x04', '1', '3', '3', '7'};
OtrlTLV *tlv = otrl_tlv_parse(serialized, sizeof(serialized));
unsigned char *buf = malloc(otrl_tlv_seriallen(tlv));
otrl_tlv_serialize(buf, tlv);
ok(memcmp(serialized, buf, sizeof(serialized)) == 0,
"tlv correctly serialized");
free(tlv);
tlv = otrl_tlv_parse(buf, sizeof(serialized));
otrl_tlv_serialize(buf, tlv);
ok(memcmp(serialized, buf, sizeof(serialized)) == 0,
"tlv correctly unserialized and serialized again");
otrl_tlv_free(tlv);
}
static void test_otrl_tlv_find()
{
const unsigned char serialized[] =
{'\x04', '\x02', '\x00', '\x03', '1', '2', '3',
'\x02', '\x02', '\x00', '\x04', '1', '3', '3', '7',
'\x01', '\x03', '\x0', '\x01', 'A',
'\x01', '\x02', '\x0', '\x01', 'B'};
OtrlTLV *tlv = otrl_tlv_parse(serialized, sizeof(serialized));
OtrlTLV *result = otrl_tlv_find(tlv, (1<<8) + 3);
ok(result == tlv->next->next, "TLV found");
result = otrl_tlv_find(tlv, 7);
ok(result == NULL, "Unexistent TLV not found");
otrl_tlv_free(tlv);
}
int main(int argc, char **argv)
{
plan_tests(NUM_TESTS);
gcry_control(GCRYCTL_SET_THREAD_CBS, &gcry_threads_pthread);
OTRL_INIT;
test_otrl_tlv_new();
test_otrl_tlv_parse();
test_otrl_tlv_seriallen();
test_otrl_tlv_serialize();
test_otrl_tlv_find();
return 0;
}