/*
 * Copyright (C) 2019 by Sukchan Lee <acetcom@gmail.com>
 *
 * This file is part of Open5GS.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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, see <https://www.gnu.org/licenses/>.
 */

#include "ogs-core.h"
#include "core/abts.h"

#define TLV_0_LEN 10
#define TLV_1_LEN 20
#define TLV_2_LEN 100
#define TLV_3_LEN 200
#define TLV_4_LEN 2
#define TLV_5_LEN 2000
#define TLV_0_INSTANCE 0
#define TLV_1_INSTANCE 1
#define TLV_2_INSTANCE 2
#define TLV_3_INSTANCE 3
#define TLV_4_INSTANCE 4
#define TLV_5_INSTANCE 5
#define TLV_VALUE_ARRAY_SIZE    3000

#define EMBED_TLV_TYPE      20
#define EMBED_TLV_INSTANCE  7

typedef struct _test_tlv_element
{
    uint8_t type;
    uint32_t length;
    uint8_t instance;
    uint8_t *value;
    uint8_t val_char;
} test_tlv_element;

uint8_t test_ogs_tlv_value[TLV_VALUE_ARRAY_SIZE];

test_tlv_element tlv_element[] ={
    {1, TLV_0_LEN, TLV_0_INSTANCE, 0, 0x0a},
    {50, TLV_1_LEN, TLV_1_INSTANCE, 0, 0x0b},
    {255, TLV_2_LEN, TLV_2_INSTANCE, 0, 0x0c},
    {254, TLV_3_LEN, TLV_3_INSTANCE, 0, 0x0d},
    {5, TLV_4_LEN, TLV_4_INSTANCE, 0, 0x0e},
    {30, TLV_5_LEN, TLV_5_INSTANCE, 0, 0x0f}
};

void ogs_tlv_test_set_ogs_tlv_value(void)
{
    uint32_t inc = 0;

    /* set test tlv value */
    tlv_element[0].value = test_ogs_tlv_value;
    memset(tlv_element[0].value, tlv_element[0].val_char, tlv_element[0].length);
    inc += tlv_element[0].length;

    tlv_element[1].value = test_ogs_tlv_value + inc;
    memset(tlv_element[1].value,  tlv_element[1].val_char, tlv_element[1].length);
    inc += tlv_element[1].length;

    tlv_element[2].value = test_ogs_tlv_value + inc;
    memset(tlv_element[2].value,  tlv_element[2].val_char, tlv_element[2].length);
    inc += tlv_element[2].length;

    tlv_element[3].value = test_ogs_tlv_value + inc;
    memset(tlv_element[3].value,  tlv_element[3].val_char, tlv_element[3].length);
    inc += tlv_element[3].length;

    tlv_element[4].value = test_ogs_tlv_value + inc;
    memset(tlv_element[4].value,  tlv_element[4].val_char, tlv_element[4].length);
    inc += tlv_element[4].length;

    return;
}

void ogs_tlv_test_check_embed_ogs_tlv_test(abts_case *tc, ogs_tlv_t *root_tlv, int mode)
{
    uint32_t m;
    uint32_t parent_block_len;
    ogs_tlv_t *pTlv;
    ogs_tlv_t *embed_tlv = NULL, *parent_tlv = NULL, *parsed_tlv = NULL;
    uint8_t parent_block[4000];
    uint8_t *pos = NULL;
    int result;

    memset(parent_block, 0x00, sizeof(parent_block));

    parent_block_len = ogs_tlv_render(root_tlv, parent_block,
        sizeof(parent_block));

    ogs_tlv_free_all(root_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    pos = parent_block;

    switch (mode) {
    case OGS_TLV_MODE_T2_L2:
        ABTS_INT_EQUAL(tc, 332, parent_block_len);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].type >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].length & 0xFF);
        for(m = 0; m < tlv_element[0].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].val_char);

        ABTS_INT_EQUAL(tc, *(pos++), EMBED_TLV_TYPE >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), EMBED_TLV_TYPE & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), (308 >> 8));
        ABTS_INT_EQUAL(tc, *(pos++), 308 & 0xFF);

        /* embedded ogs_tlv_t */
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].type >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].length & 0xFF);
        for(m = 0; m < tlv_element[2].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].val_char);


        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].type >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].length & 0xFF);
        for(m = 0; m < tlv_element[3].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].val_char);


        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].type >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].length & 0xFF);

        for(m = 0; m < tlv_element[4].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), 0x0e);
        break;
    case OGS_TLV_MODE_T1_L2:
        ABTS_INT_EQUAL(tc, 327, parent_block_len);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].length & 0xFF);
        for(m = 0; m < tlv_element[0].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].val_char);

        ABTS_INT_EQUAL(tc, *(pos++), EMBED_TLV_TYPE & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), (306 >> 8));
        ABTS_INT_EQUAL(tc, *(pos++), 306 & 0xFF);

        /* embedded ogs_tlv_t */
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].length & 0xFF);
        for(m = 0; m < tlv_element[2].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].val_char);


        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].length & 0xFF);
        for(m = 0; m < tlv_element[3].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].val_char);


        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].length & 0xFF);

        for(m = 0; m < tlv_element[4].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), 0x0e);
        break;
    case OGS_TLV_MODE_T1_L2_I1:
        ABTS_INT_EQUAL(tc, 332, parent_block_len);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].length & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].instance & 0xFF);
        for(m = 0; m < tlv_element[0].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[0].val_char);

        ABTS_INT_EQUAL(tc, *(pos++), EMBED_TLV_TYPE & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), (308 >> 8));
        ABTS_INT_EQUAL(tc, *(pos++), 308 & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), EMBED_TLV_INSTANCE);

        /* embedded ogs_tlv_t */
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].length & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].instance & 0xFF);
        for(m = 0; m < tlv_element[2].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[2].val_char);


        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].length & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].instance & 0xFF);
        for(m = 0; m < tlv_element[3].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), tlv_element[3].val_char);


        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].type & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].length >> 8);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].length & 0xFF);
        ABTS_INT_EQUAL(tc, *(pos++), tlv_element[4].instance & 0xFF);

        for(m = 0; m < tlv_element[4].length; m++)
            ABTS_INT_EQUAL(tc, *(pos++), 0x0e);
        break;
    default:
        ABTS_TRUE(tc, 0);
        break;
    }

    parsed_tlv = ogs_tlv_parse_block(parent_block_len,
        parent_block, mode);
    ABTS_PTR_NOTNULL(tc, parsed_tlv);

    pTlv = parsed_tlv;

    ABTS_INT_EQUAL(tc, pTlv->type, tlv_element[0].type);
    ABTS_INT_EQUAL(tc, pTlv->length, tlv_element[0].length);
    ABTS_INT_EQUAL(tc, pTlv->instance, tlv_element[0].instance);
    result = memcmp(pTlv->value, tlv_element[0].value, tlv_element[0].length);
    ABTS_INT_EQUAL(tc, result, 0);
    pTlv = pTlv->next;

    ABTS_INT_EQUAL(tc, pTlv->type, 20);
    switch (mode) {
    case OGS_TLV_MODE_T2_L2:
    case OGS_TLV_MODE_T1_L2_I1:
        ABTS_INT_EQUAL(tc, pTlv->length, 308);
        break;
    case OGS_TLV_MODE_T1_L2:
        ABTS_INT_EQUAL(tc, pTlv->length, 306);
        break;
    default:
        ABTS_TRUE(tc, 0);
        break;
    }
    pTlv = pTlv->next;

    ABTS_INT_EQUAL(tc, pTlv->type, tlv_element[4].type);
    ABTS_INT_EQUAL(tc, pTlv->length, tlv_element[4].length);
    if (mode == OGS_TLV_MODE_T1_L2_I1)
        ABTS_INT_EQUAL(tc, pTlv->instance, tlv_element[4].instance);
    else
        ABTS_INT_EQUAL(tc, pTlv->instance, 0);
    result = memcmp(pTlv->value, tlv_element[4].value, tlv_element[4].length);
    ABTS_INT_EQUAL(tc, result, 0);
    pTlv = pTlv->next;

    ABTS_TRUE(tc, pTlv == NULL);

    parent_tlv = ogs_tlv_find(parsed_tlv,20);
    ABTS_PTR_NOTNULL(tc, parent_tlv);

    ogs_tlv_parse_embedded_block(parent_tlv, mode);
    embed_tlv = parent_tlv->embedded;
    ABTS_PTR_NOTNULL(tc, embed_tlv);

    ABTS_INT_EQUAL(tc, embed_tlv->type, tlv_element[2].type);
    ABTS_INT_EQUAL(tc, embed_tlv->length, tlv_element[2].length);
    if (mode == OGS_TLV_MODE_T1_L2_I1)
        ABTS_INT_EQUAL(tc, embed_tlv->instance, tlv_element[2].instance);
    else
        ABTS_INT_EQUAL(tc, embed_tlv->instance, 0);
    for(m = 0; m < tlv_element[2].length; m++) {
        ABTS_INT_EQUAL(tc, *((uint8_t*)embed_tlv->value+m),
            tlv_element[2].val_char);
    }
    embed_tlv = embed_tlv->next;

    ABTS_INT_EQUAL(tc, embed_tlv->type, tlv_element[3].type);
    ABTS_INT_EQUAL(tc, embed_tlv->length, tlv_element[3].length);
    if (mode == OGS_TLV_MODE_T1_L2_I1)
        ABTS_INT_EQUAL(tc, embed_tlv->instance, tlv_element[3].instance);
    else
        ABTS_INT_EQUAL(tc, embed_tlv->instance, 0);
    for(m = 0; m < tlv_element[3].length; m++) {
        ABTS_INT_EQUAL(tc, *((uint8_t*)embed_tlv->value+m),
            tlv_element[3].val_char);
    }
    embed_tlv = embed_tlv->next;

    ABTS_TRUE(tc, embed_tlv == NULL);

    embed_tlv = ogs_tlv_find(parsed_tlv,254);
    ABTS_PTR_NOTNULL(tc, embed_tlv);

    embed_tlv = ogs_tlv_find(parsed_tlv,253);
    ABTS_TRUE(tc, embed_tlv == NULL);

    ogs_tlv_free_all(parsed_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    return;
}

/* basic encoding/decoding/finding Test */
static void test1_func(abts_case *tc, void *data)
{
    uint32_t idx, m, parent_block_len;
    ogs_tlv_t *root_tlv = NULL, *parsed_tlv = NULL, *pTlv;
    uint8_t parent_block[4000];
    uint8_t *pos = NULL;
    uint8_t mode = (uintptr_t)data;

    ogs_tlv_test_set_ogs_tlv_value();

    /* tlv encoding for test */
    root_tlv = ogs_tlv_add(NULL, mode, tlv_element[0].type,
        tlv_element[0].length, tlv_element[0].instance, tlv_element[0].value);

    for (idx = 1; idx < 4; idx++) {
        ogs_tlv_add(root_tlv, mode, tlv_element[idx].type,
            tlv_element[idx].length, tlv_element[idx].instance,
            tlv_element[idx].value);
    }

    memset(parent_block, 0x00, sizeof(parent_block));
    parent_block_len = ogs_tlv_render(root_tlv, parent_block,
        sizeof(parent_block));

    switch (mode) {
    case OGS_TLV_MODE_T2_L2:
    case OGS_TLV_MODE_T1_L2_I1:
        ABTS_INT_EQUAL(tc, 346, parent_block_len);
        break;
    case OGS_TLV_MODE_T1_L2:
        ABTS_INT_EQUAL(tc, 342, parent_block_len);
        break;
    case OGS_TLV_MODE_T1_L1:
        ABTS_INT_EQUAL(tc, 338, parent_block_len);
        break;
    default:
        ABTS_TRUE(tc, 0);
        break;
    }

    ogs_tlv_free_all(root_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    pos = parent_block;

    for (idx = 0; idx < 4; idx++) {
        switch (mode) {
        case OGS_TLV_MODE_T2_L2:
            ABTS_INT_EQUAL(tc, (tlv_element[idx].type >> 8), *(pos++));
            ABTS_INT_EQUAL(tc, tlv_element[idx].type & 0xFF, *(pos++));
            ABTS_INT_EQUAL(tc, (tlv_element[idx].length >> 8), *(pos++));
            ABTS_INT_EQUAL(tc, tlv_element[idx].length & 0xFF, *(pos++));
            break;
        case OGS_TLV_MODE_T1_L2:
            ABTS_INT_EQUAL(tc, tlv_element[idx].type & 0xFF, *(pos++));
            ABTS_INT_EQUAL(tc, (tlv_element[idx].length >> 8), *(pos++));
            ABTS_INT_EQUAL(tc, tlv_element[idx].length & 0xFF, *(pos++));
            break;
        case OGS_TLV_MODE_T1_L1:
            ABTS_INT_EQUAL(tc, tlv_element[idx].type & 0xFF, *(pos++));
            ABTS_INT_EQUAL(tc, tlv_element[idx].length & 0xFF, *(pos++));
            break;
        case OGS_TLV_MODE_T1_L2_I1:
            ABTS_INT_EQUAL(tc, tlv_element[idx].type & 0xFF, *(pos++));
            ABTS_INT_EQUAL(tc, (tlv_element[idx].length >> 8), *(pos++));
            ABTS_INT_EQUAL(tc, tlv_element[idx].length & 0xFF, *(pos++));
            ABTS_INT_EQUAL(tc, tlv_element[idx].instance & 0xFF, *(pos++));
            break;
        default:
            ABTS_TRUE(tc, 0);
            break;
        }

        for(m = 0; m < tlv_element[idx].length; m++)
            ABTS_INT_EQUAL(tc, tlv_element[idx].val_char, *(pos++));
    }

    parsed_tlv = ogs_tlv_parse_block(parent_block_len,parent_block, mode);
    ABTS_PTR_NOTNULL(tc, parsed_tlv);

    pTlv = parsed_tlv;

    idx = 0;
    while (pTlv) {
        int result;
        ABTS_INT_EQUAL(tc, pTlv->type, tlv_element[idx].type);
        ABTS_INT_EQUAL(tc, pTlv->length, tlv_element[idx].length);
        if (mode == OGS_TLV_MODE_T1_L2_I1)
            ABTS_INT_EQUAL(tc, pTlv->instance, tlv_element[idx].instance);
        else
            ABTS_INT_EQUAL(tc, pTlv->instance, 0);
        result = memcmp(pTlv->value, tlv_element[idx].value, tlv_element[idx].length);
        ABTS_INT_EQUAL(tc, result, 0);

        pTlv = pTlv->next;
        idx++;
    }

    ABTS_INT_EQUAL(tc, idx, 4);

    pTlv = ogs_tlv_find(parsed_tlv, 255);
    ABTS_PTR_NOTNULL(tc, pTlv);

    pTlv = ogs_tlv_find(parsed_tlv, 253);
    ABTS_TRUE(tc, pTlv == NULL);

    ogs_tlv_free_all(parsed_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    return;
}

/* embedded ogs_tlv_t test :  first, make embedded tlv block  for embeded tlv
    and then make whole tlv block using embedded tlv block previously made*/
static void test2_func(abts_case *tc, void *data)
{
    uint32_t embed_block_len;
    ogs_tlv_t *root_tlv = NULL;
    ogs_tlv_t *embed_tlv = NULL;
    uint8_t embed_block[1000];;
    uint8_t mode = (uintptr_t)data;

    ogs_tlv_test_set_ogs_tlv_value();

    /* Tlv Encoding for embeded ogs_tlv_t */
    embed_tlv = ogs_tlv_add(NULL, mode, tlv_element[2].type,
        tlv_element[2].length, tlv_element[2].instance, tlv_element[2].value);
    ogs_tlv_add(embed_tlv, mode, tlv_element[3].type,
        tlv_element[3].length, tlv_element[3].instance, tlv_element[3].value);

    embed_block_len = ogs_tlv_render(embed_tlv, embed_block,
        sizeof(embed_block));
    switch (mode) {
    case OGS_TLV_MODE_T2_L2:
    case OGS_TLV_MODE_T1_L2_I1:
        ABTS_INT_EQUAL(tc, embed_block_len, 308);
        break;
    case OGS_TLV_MODE_T1_L2:
        ABTS_INT_EQUAL(tc, embed_block_len, 306);
        break;
    default:
        ABTS_TRUE(tc, 0);
        break;
    }

    ogs_tlv_free_all(embed_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    root_tlv = ogs_tlv_add(NULL, mode, tlv_element[0].type,
        tlv_element[0].length, tlv_element[0].instance, tlv_element[0].value);

    ogs_tlv_add(root_tlv, mode, EMBED_TLV_TYPE, embed_block_len,
            EMBED_TLV_INSTANCE, embed_block);
    ogs_tlv_add(root_tlv, mode, tlv_element[4].type,
        tlv_element[4].length, tlv_element[4].instance, tlv_element[4].value);

    ogs_tlv_test_check_embed_ogs_tlv_test(tc, root_tlv, mode);

    return;
}


/* embedded ogs_tlv_t test :  make parent and embedded tlv and then ogs_tlv_render
    functions makes whole tlv block.  the value member of each tlv must
    point to vaild address at the time using ogs_tlv_rendering function */
static void test3_func(abts_case *tc, void *data)
{
    ogs_tlv_t *root_tlv = NULL, *parent_tlv = NULL;
    uint8_t mode = (uintptr_t)data;

    ogs_tlv_test_set_ogs_tlv_value();

    /* Tlv Encoding for embeded ogs_tlv_t */
    root_tlv = ogs_tlv_add(NULL, mode, tlv_element[0].type,
        tlv_element[0].length, tlv_element[0].instance, tlv_element[0].value);
    parent_tlv= ogs_tlv_add(root_tlv, mode, EMBED_TLV_TYPE, 0, EMBED_TLV_INSTANCE, NULL);
    ogs_tlv_add(root_tlv, mode, tlv_element[4].type,
        tlv_element[4].length, tlv_element[4].instance, tlv_element[4].value);

    ogs_tlv_embed(parent_tlv, mode, tlv_element[2].type,
        tlv_element[2].length, tlv_element[2].instance, tlv_element[2].value);
    ogs_tlv_embed(parent_tlv, mode, tlv_element[3].type,
        tlv_element[3].length, tlv_element[3].instance, tlv_element[3].value);

    ogs_tlv_test_check_embed_ogs_tlv_test(tc, root_tlv, mode);

    return;
}


/* embedded ogs_tlv_t test :  make parent and embedded tlv and then ogs_tlv_render
    functions makes whole tlv block.  The value member of each tlv is copied
    to the buff of the root ogs_tlv_t, so the allocated resource for the value
    member of each tlv can be deallocated after executing ogs_tlv_add or tlv embed
    function*/
static void test4_func(abts_case *tc, void *data)
{
    ogs_tlv_t *root_tlv = NULL, *parent_tlv = NULL;
    uint8_t tlv_buff[2000];
    uint8_t mode = (uintptr_t)data;

    ogs_tlv_test_set_ogs_tlv_value();

    root_tlv = ogs_tlv_copy(tlv_buff, sizeof(tlv_buff), mode,
        tlv_element[0].type, tlv_element[0].length,
        tlv_element[0].instance, tlv_element[0].value);
    parent_tlv = ogs_tlv_add(root_tlv, mode, EMBED_TLV_TYPE, 0,
                        EMBED_TLV_INSTANCE, NULL);
    ogs_tlv_add(root_tlv, mode, tlv_element[4].type, tlv_element[4].length,
        tlv_element[4].instance, tlv_element[4].value);

    ogs_tlv_embed(parent_tlv, mode, tlv_element[2].type,
        tlv_element[2].length, tlv_element[2].instance, tlv_element[2].value);
    ogs_tlv_embed(parent_tlv, mode, tlv_element[3].type,
        tlv_element[3].length, tlv_element[3].instance, tlv_element[3].value);

    memset(tlv_element[2].value, 0x00, tlv_element[2].length);
    memset(tlv_element[3].value, 0xf0, tlv_element[3].length);

    ogs_tlv_test_check_embed_ogs_tlv_test(tc, root_tlv, mode);

    return;
}

/* endian check test */
static void test5_func(abts_case *tc, void *data)
{
    uint32_t parent_block_len;
    ogs_tlv_t *root_tlv = NULL, *parsed_tlv = NULL, *p_tlv;
    uint8_t parent_block[4000];
    uint8_t *pos = NULL;
    uint16_t c_16 = 0x1122;
    uint32_t c_32 = 0x11223344;
    uint8_t mode = (uintptr_t)data;

    /* tlv encoding for test */
    c_16 = htobe16(c_16);
    root_tlv = ogs_tlv_add(NULL, mode, 10, 2, 0, (uint8_t*)&c_16);
    c_32 = htobe32(c_32);
    ogs_tlv_add(root_tlv, mode, 20, 4, 0, (uint8_t*)&c_32);

    memset(parent_block, 0x00, sizeof(parent_block));
    parent_block_len = ogs_tlv_render(root_tlv, parent_block,
        sizeof(parent_block));

    ogs_tlv_free_all(root_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    pos = parent_block;

    switch (mode) {
    case OGS_TLV_MODE_T2_L2:
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 10);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 2);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 20);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 4);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 0x33);
        ABTS_INT_EQUAL(tc, *(pos++), 0x44);
        break;
    case OGS_TLV_MODE_T1_L2:
        ABTS_INT_EQUAL(tc, *(pos++), 10);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 2);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 20);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 4);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 0x33);
        ABTS_INT_EQUAL(tc, *(pos++), 0x44);
        break;
    case OGS_TLV_MODE_T1_L2_I1:
        ABTS_INT_EQUAL(tc, *(pos++), 10);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 2);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 20);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 4);
        ABTS_INT_EQUAL(tc, *(pos++), 0);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 0x33);
        ABTS_INT_EQUAL(tc, *(pos++), 0x44);
        break;
    case OGS_TLV_MODE_T1_L1:
        ABTS_INT_EQUAL(tc, *(pos++), 10);
        ABTS_INT_EQUAL(tc, *(pos++), 2);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 20);
        ABTS_INT_EQUAL(tc, *(pos++), 4);
        ABTS_INT_EQUAL(tc, *(pos++), 0x11);
        ABTS_INT_EQUAL(tc, *(pos++), 0x22);
        ABTS_INT_EQUAL(tc, *(pos++), 0x33);
        ABTS_INT_EQUAL(tc, *(pos++), 0x44);
        break;
    default:
        ABTS_TRUE(tc, 0);
        break;
    }

    p_tlv = parsed_tlv = ogs_tlv_parse_block(parent_block_len,parent_block,
        mode);
    ABTS_INT_EQUAL(tc, ogs_tlv_value_16(p_tlv), 0x1122);
    p_tlv = parsed_tlv->next;
    ABTS_INT_EQUAL(tc, ogs_tlv_value_32(p_tlv), 0x11223344);

    ogs_tlv_free_all(parsed_tlv);
    ABTS_INT_EQUAL(tc, ogs_tlv_pool_avail(), ogs_core()->tlv.pool);

    return;
}

/* Sample header for tlv_msg */
#define TLV_AUTHORIZATION_POLICY_SUPPORT_TYPE 21
#define TLV_AUTHORIZATION_POLICY_SUPPORT_LEN 1
typedef ogs_tlv_uint8_t tlv_authorization_policy_support_t;
extern ogs_tlv_desc_t tlv_desc_authorization_policy_support;

#define TLV_CLIENT_SECURITY_HISTORY_TYPE 108
#define TLV_CLIENT_SECURITY_HISTORY_LEN OGS_TLV_VARIABLE_LEN
typedef struct _tlv_client_security_history_t {
    ogs_tlv_presence_t presence;
    tlv_authorization_policy_support_t authorization_policy_support0;
    tlv_authorization_policy_support_t authorization_policy_support2;
} tlv_client_security_history_t;
extern ogs_tlv_desc_t tlv_desc_client_security_history;

#define TLV_CLIENT_INFO_TYPE 103
#define TLV_CLIENT_INFO_LEN OGS_TLV_VARIABLE_LEN
typedef struct _tlv_client_info_t {
    ogs_tlv_presence_t presence;
    tlv_client_security_history_t client_security_history;
} tlv_client_info_t;
extern ogs_tlv_desc_t tlv_desc_client_info;

#define TLV_SERVER_NAME_TYPE 25
#define TLV_SERVER_NAME_LEN OGS_TLV_VARIABLE_LEN
typedef ogs_tlv_octet_t tlv_server_name_t;
extern ogs_tlv_desc_t tlv_desc_server_name;

#define TLV_SERVER_INFO_TYPE 26
#define TLV_SERVER_INFO_LEN OGS_TLV_VARIABLE_LEN
typedef struct _tlv_server_info_t {
    ogs_tlv_presence_t presence;
    tlv_server_name_t OGS_TLV_1_OR_MORE(server_name);
} tlv_server_info_t;
extern ogs_tlv_desc_t tlv_desc_server_info;

typedef struct _tlv_attach_req {
    tlv_client_info_t client_info;
    tlv_server_info_t server_info;
} tlv_attach_req;

extern ogs_tlv_desc_t tlv_desc_attach_req;

/* Sample source for tlv_msg */
ogs_tlv_desc_t tlv_desc_authorization_policy_support0 =
{
    OGS_TLV_UINT8,
    "Auth Policy0",
    TLV_AUTHORIZATION_POLICY_SUPPORT_TYPE,
    TLV_AUTHORIZATION_POLICY_SUPPORT_LEN,
    0,
    sizeof(tlv_authorization_policy_support_t),
    { NULL }
};
ogs_tlv_desc_t tlv_desc_authorization_policy_support2 =
{
    OGS_TLV_UINT8,
    "Auth Policy2",
    TLV_AUTHORIZATION_POLICY_SUPPORT_TYPE,
    TLV_AUTHORIZATION_POLICY_SUPPORT_LEN,
    2,
    sizeof(tlv_authorization_policy_support_t),
    { NULL }
};

ogs_tlv_desc_t tlv_desc_client_security_history =
{
    OGS_TLV_COMPOUND,
    "Sec History",
    TLV_CLIENT_SECURITY_HISTORY_TYPE,
    TLV_CLIENT_SECURITY_HISTORY_LEN,
    0,
    sizeof(tlv_client_security_history_t),
    {
        &tlv_desc_authorization_policy_support0,
        &tlv_desc_authorization_policy_support2,
        NULL,
    }
};

ogs_tlv_desc_t tlv_desc_client_info =
{
    OGS_TLV_COMPOUND,
    "Client Info",
    TLV_CLIENT_INFO_TYPE,
    TLV_CLIENT_INFO_LEN,
    0,
    sizeof(tlv_client_info_t),
    {
        &tlv_desc_client_security_history,
        NULL,
    }
};

ogs_tlv_desc_t tlv_desc_server_name =
{
    OGS_TLV_VAR_STR,
    "Server Name",
    TLV_SERVER_NAME_TYPE,
    TLV_SERVER_NAME_LEN,
    0,
    sizeof(tlv_server_name_t),
    { NULL }
};

ogs_tlv_desc_t tlv_desc_server_info =
{
    OGS_TLV_COMPOUND,
    "Server Info",
    TLV_SERVER_INFO_TYPE,
    TLV_SERVER_INFO_LEN,
    0,
    sizeof(tlv_server_info_t),
    {
        &tlv_desc_server_name, &ogs_tlv_desc_more16,
        NULL,
    }
};

ogs_tlv_desc_t tlv_desc_attach_req = {
    OGS_TLV_MESSAGE, "Attach Req", 0, 0, 0, 0, {
    &tlv_desc_client_info,
    &tlv_desc_server_info,
    NULL,
}};

static void test6_func(abts_case *tc, void *data)
{
    tlv_attach_req reqv;
    tlv_attach_req reqv2;

    ogs_pkbuf_t *req = NULL;
    char testbuf[1024];

    int i;

    /* Initialize message value structure */
    memset(&reqv, 0, sizeof(tlv_attach_req));

    /* Set nessary members of message */
    reqv.client_info.presence = 1;
    reqv.client_info.client_security_history.presence = 1;
    reqv.client_info.client_security_history.
            authorization_policy_support0.presence = 1;
    reqv.client_info.client_security_history.
            authorization_policy_support0.u8 = 0x3;
    reqv.client_info.client_security_history.
            authorization_policy_support2.presence = 1;
    reqv.client_info.client_security_history.
            authorization_policy_support2.u8 = 0x9;

    reqv.server_info.presence = 1;
    for (i = 0; i < 16; i++) {
        reqv.server_info.server_name[i].presence = 1;
        reqv.server_info.server_name[i].data =
            (uint8_t*)"\x11\x22\x33\x44\x55\x66";
        reqv.server_info.server_name[i].len = 6;

        req = ogs_tlv_build_msg(
                &tlv_desc_attach_req, &reqv, OGS_TLV_MODE_T1_L2_I1);
        ABTS_INT_EQUAL(tc, 32+i*10, req->len);
        ogs_pkbuf_free(req);
    }

    /* Build message */
    req = ogs_tlv_build_msg(&tlv_desc_attach_req, &reqv, OGS_TLV_MODE_T1_L2_I1);
#define TEST_TLV_BUILD_MSG \
    "67000e00 6c000a00 15000100 03150001" \
    "02091a00 a0001900 06001122 33445566" \
    "19000600 11223344 55661900 06001122" \
    "33445566 19000600 11223344 55661900" \
    "06001122 33445566 19000600 11223344" \
    "55661900 06001122 33445566 19000600" \
    "11223344 55661900 06001122 33445566" \
    "19000600 11223344 55661900 06001122" \
    "33445566 19000600 11223344 55661900" \
    "06001122 33445566 19000600 11223344" \
    "55661900 06001122 33445566 19000600" \
    "11223344 5566"

    ABTS_INT_EQUAL(tc, 182, req->len);
    ABTS_TRUE(tc, memcmp(req->data,
        ogs_hex_from_string(TEST_TLV_BUILD_MSG, testbuf, sizeof(testbuf)),
        req->len) == 0);

    /* Initialize message value structure */
    memset(&reqv2, 0, sizeof(tlv_attach_req));

    /* Parse message */
    ogs_tlv_parse_msg(&reqv2, &tlv_desc_attach_req, req,
            OGS_TLV_MODE_T1_L2_I1);

    ABTS_INT_EQUAL(tc, 1, reqv2.client_info.presence);
    ABTS_INT_EQUAL(tc, 1,
            reqv2.client_info.client_security_history.presence);
    ABTS_INT_EQUAL(tc, 1,
            reqv2.client_info.client_security_history.
                        authorization_policy_support0.presence);
    ABTS_INT_EQUAL(tc, 0x3, reqv2.client_info.
            client_security_history.authorization_policy_support0.u8);
    ABTS_INT_EQUAL(tc, 1,
            reqv2.client_info.client_security_history.
                        authorization_policy_support2.presence);
    ABTS_INT_EQUAL(tc, 0x9, reqv2.client_info.
            client_security_history.authorization_policy_support2.u8);

    for (i = 0; i < 16; i++) {
        ABTS_INT_EQUAL(tc, 1, reqv2.server_info.presence);
        ABTS_INT_EQUAL(tc, 1, reqv2.server_info.server_name[i].presence);
        ABTS_INT_EQUAL(tc, 6, reqv2.server_info.server_name[i].len);
        ABTS_TRUE(tc, memcmp(reqv2.server_info.server_name[i].data,
                    (uint8_t*)"\x11\x22\x33\x44\x55\x66", 6) == 0);
    }

    ogs_pkbuf_free(req);
}

abts_suite *test_tlv(abts_suite *suite)
{
    suite = ADD_SUITE(suite)

    abts_run_test(suite, test1_func, (void*)OGS_TLV_MODE_T2_L2);
    abts_run_test(suite, test2_func, (void*)OGS_TLV_MODE_T2_L2);
    abts_run_test(suite, test3_func, (void*)OGS_TLV_MODE_T2_L2);
    abts_run_test(suite, test4_func, (void*)OGS_TLV_MODE_T2_L2);
    abts_run_test(suite, test5_func, (void*)OGS_TLV_MODE_T2_L2);
    abts_run_test(suite, test1_func, (void*)OGS_TLV_MODE_T1_L1);
    abts_run_test(suite, test5_func, (void*)OGS_TLV_MODE_T1_L1);
    abts_run_test(suite, test1_func, (void*)OGS_TLV_MODE_T1_L2);
    abts_run_test(suite, test2_func, (void*)OGS_TLV_MODE_T1_L2);
    abts_run_test(suite, test3_func, (void*)OGS_TLV_MODE_T1_L2);
    abts_run_test(suite, test4_func, (void*)OGS_TLV_MODE_T1_L2);
    abts_run_test(suite, test5_func, (void*)OGS_TLV_MODE_T1_L2);
    abts_run_test(suite, test1_func, (void*)OGS_TLV_MODE_T1_L2_I1);
    abts_run_test(suite, test2_func, (void*)OGS_TLV_MODE_T1_L2_I1);
    abts_run_test(suite, test3_func, (void*)OGS_TLV_MODE_T1_L2_I1);
    abts_run_test(suite, test4_func, (void*)OGS_TLV_MODE_T1_L2_I1);
    abts_run_test(suite, test5_func, (void*)OGS_TLV_MODE_T1_L2_I1);

    abts_run_test(suite, test6_func, NULL);

    return suite;
}