/*
 * 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 "test-common.h"

static void test1_func(abts_case *tc, void *data)
{
    int rv;
    ogs_socknode_t *s1ap;
    ogs_socknode_t *gtpu;
    ogs_pkbuf_t *emmbuf;
    ogs_pkbuf_t *esmbuf;
    ogs_pkbuf_t *sendbuf;
    ogs_pkbuf_t *recvbuf;
    ogs_s1ap_message_t message;

    ogs_nas_5gs_mobile_identity_suci_t mobile_identity_suci;
    test_ue_t *test_ue = NULL;
    test_sess_t *sess = NULL;
    test_bearer_t *bearer = NULL;

    S1AP_UE_associatedLogicalS1_ConnectionListRes_t *partOfS1_Interface = NULL;

    bson_t *doc = NULL;

    /* Setup Test UE & Session Context */
    memset(&mobile_identity_suci, 0, sizeof(mobile_identity_suci));

    mobile_identity_suci.h.supi_format = OGS_NAS_5GS_SUPI_FORMAT_IMSI;
    mobile_identity_suci.h.type = OGS_NAS_5GS_MOBILE_IDENTITY_SUCI;
    mobile_identity_suci.routing_indicator1 = 0;
    mobile_identity_suci.routing_indicator2 = 0xf;
    mobile_identity_suci.routing_indicator3 = 0xf;
    mobile_identity_suci.routing_indicator4 = 0xf;
    mobile_identity_suci.protection_scheme_id = OGS_PROTECTION_SCHEME_NULL;
    mobile_identity_suci.home_network_pki_value = 0;

    test_ue = test_ue_add_by_suci(&mobile_identity_suci, "3746000006");
    ogs_assert(test_ue);

    test_ue->e_cgi.cell_id = 0x54f6401;
    test_ue->nas.ksi = OGS_NAS_KSI_NO_KEY_IS_AVAILABLE;
    test_ue->nas.value = OGS_NAS_ATTACH_TYPE_COMBINED_EPS_IMSI_ATTACH;

    test_ue->k_string = "465b5ce8b199b49faa5f0a2ee238a6bc";
    test_ue->opc_string = "e8ed289deba952e4283b54e88e6183ca";

    sess = test_sess_add_by_apn(test_ue, "internet", OGS_GTP2_RAT_TYPE_EUTRAN);
    ogs_assert(sess);

    /* eNB connects to MME */
    s1ap = tests1ap_client(AF_INET);
    ABTS_PTR_NOTNULL(tc, s1ap);

    /* eNB connects to SGW */
    gtpu = test_gtpu_server(1, AF_INET);
    ABTS_PTR_NOTNULL(tc, gtpu);

    /* Send S1-Setup Reqeust */
    sendbuf = test_s1ap_build_s1_setup_request(
            S1AP_ENB_ID_PR_macroENB_ID, 0x54f64);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Setup Response */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(NULL, recvbuf);

    /********** Insert Subscriber in Database */
    doc = test_db_new_simple(test_ue);
    ABTS_PTR_NOTNULL(tc, doc);
    ABTS_INT_EQUAL(tc, OGS_OK, test_db_insert_ue(test_ue, doc));

    /* Send Service Request */
    emmbuf = testemm_build_service_request(test_ue);
    ABTS_PTR_NOTNULL(tc, emmbuf);
    sendbuf = test_s1ap_build_initial_ue_message(
            test_ue, emmbuf, S1AP_RRC_Establishment_Cause_mo_Data, true);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive Service reject */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Receive UE Context Release Command */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Send UE Context Release Complete */
    sendbuf = test_s1ap_build_ue_context_release_complete(test_ue);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Send Attach Request */
    memset(&sess->pdn_connectivity_param,
            0, sizeof(sess->pdn_connectivity_param));
    sess->pdn_connectivity_param.eit = 1;
    sess->pdn_connectivity_param.pco = 1;
    sess->pdn_connectivity_param.request_type =
        OGS_NAS_EPS_REQUEST_TYPE_INITIAL;
    esmbuf = testesm_build_pdn_connectivity_request(sess, false);
    ABTS_PTR_NOTNULL(tc, esmbuf);

    memset(&test_ue->attach_request_param,
            0, sizeof(test_ue->attach_request_param));
    test_ue->attach_request_param.drx_parameter = 1;
    test_ue->attach_request_param.ms_network_capability = 1;
    test_ue->attach_request_param.tmsi_status = 1;
    test_ue->attach_request_param.mobile_station_classmark_2 = 1;
    test_ue->attach_request_param.supported_codecs = 1;
    test_ue->attach_request_param.ue_usage_setting = 1;
    test_ue->attach_request_param.ms_network_feature_support = 1;
    emmbuf = testemm_build_attach_request(test_ue, esmbuf, false, false);
    ABTS_PTR_NOTNULL(tc, emmbuf);

    memset(&test_ue->initial_ue_param, 0, sizeof(test_ue->initial_ue_param));
    sendbuf = test_s1ap_build_initial_ue_message(
            test_ue, emmbuf, S1AP_RRC_Establishment_Cause_mo_Signalling, false);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive Authentication Request */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Send Authentication response */
    emmbuf = testemm_build_authentication_response(test_ue);
    ABTS_PTR_NOTNULL(tc, emmbuf);
    sendbuf = test_s1ap_build_uplink_nas_transport(test_ue, emmbuf);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive Security mode Command */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Send Security mode complete */
    test_ue->mobile_identity_imeisv_presence = true;
    emmbuf = testemm_build_security_mode_complete(test_ue);
    ABTS_PTR_NOTNULL(tc, emmbuf);
    sendbuf = test_s1ap_build_uplink_nas_transport(test_ue, emmbuf);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive ESM Information Request */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Send ESM Information Response */
    esmbuf = testesm_build_esm_information_response(sess);
    ABTS_PTR_NOTNULL(tc, esmbuf);
    sendbuf = test_s1ap_build_uplink_nas_transport(test_ue, esmbuf);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive Initial Context Setup Request +
     * Attach Accept +
     * Activate Default Bearer Context Request */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Send UE Capability Info Indication */
    sendbuf = tests1ap_build_ue_radio_capability_info_indication(test_ue);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Send Initial Context Setup Response */
    sendbuf = test_s1ap_build_initial_context_setup_response(test_ue);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Send Attach Complete + Activate default EPS bearer cotext accept */
    test_ue->nr_cgi.cell_id = 0x1234502;
    bearer = test_bearer_find_by_ue_ebi(test_ue, 5);
    ogs_assert(bearer);
    esmbuf = testesm_build_activate_default_eps_bearer_context_accept(
            bearer, false);
    ABTS_PTR_NOTNULL(tc, esmbuf);
    emmbuf = testemm_build_attach_complete(test_ue, esmbuf);
    ABTS_PTR_NOTNULL(tc, emmbuf);
    sendbuf = test_s1ap_build_uplink_nas_transport(test_ue, emmbuf);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive EMM information */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);

    /* Send S1-Reset */
    partOfS1_Interface = NULL;

    ogs_s1ap_build_part_of_s1_interface(
            &partOfS1_Interface,
            &test_ue->mme_ue_s1ap_id, &test_ue->enb_ue_s1ap_id);

    sendbuf = ogs_s1ap_build_s1_reset(
            S1AP_Cause_PR_radioNetwork,
            S1AP_CauseRadioNetwork_release_due_to_eutran_generated_reason,
            partOfS1_Interface);

    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Reset Acknowledge */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    ogs_pkbuf_free(recvbuf);

    /* Send UE Context Release Request */
    sendbuf = test_s1ap_build_ue_context_release_request(test_ue,
            S1AP_Cause_PR_radioNetwork, S1AP_CauseRadioNetwork_user_inactivity);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive UE Context Release Command */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(test_ue, recvbuf);
    ABTS_INT_EQUAL(tc,
            S1AP_ProcedureCode_id_ErrorIndication,
            test_ue->s1ap_procedure_code);

    ogs_msleep(300);

    /********** Remove Subscriber in Database */
    ABTS_INT_EQUAL(tc, OGS_OK, test_db_remove_ue(test_ue));

    /* eNB disonncect from MME */
    testenb_s1ap_close(s1ap);

    /* eNB disonncect from SGW */
    test_gtpu_close(gtpu);

    test_ue_remove(test_ue);
}

#define NUM_OF_TEST_UE 3

static void test2_func(abts_case *tc, void *data)
{
    int rv, i;
    ogs_socknode_t *s1ap;
    ogs_socknode_t *gtpu;
    ogs_pkbuf_t *emmbuf;
    ogs_pkbuf_t *esmbuf;
    ogs_pkbuf_t *sendbuf;
    ogs_pkbuf_t *recvbuf;
    ogs_s1ap_message_t message;

    ogs_nas_5gs_mobile_identity_suci_t mobile_identity_suci;
    test_ue_t *test_ue[NUM_OF_TEST_UE];
    test_sess_t *sess = NULL;
    test_bearer_t *bearer = NULL;

    bson_t *doc = NULL;

    /* eNB connects to MME */
    s1ap = tests1ap_client(AF_INET);
    ABTS_PTR_NOTNULL(tc, s1ap);

    /* eNB connects to SGW */
    gtpu = test_gtpu_server(1, AF_INET);
    ABTS_PTR_NOTNULL(tc, gtpu);

    /* Send S1-Setup Reqeust */
    sendbuf = test_s1ap_build_s1_setup_request(
            S1AP_ENB_ID_PR_macroENB_ID, 0x54f64);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Setup Response */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(NULL, recvbuf);

    for (i = 0; i < NUM_OF_TEST_UE; i++) {
        const char *scheme_output[] = {
            "0000000001",
            "0000000002",
            "0000000003",
            "0000000004",
            "0000000005",
        };

        /* Setup Test UE & Session Context */
        memset(&mobile_identity_suci, 0, sizeof(mobile_identity_suci));

        mobile_identity_suci.h.supi_format = OGS_NAS_5GS_SUPI_FORMAT_IMSI;
        mobile_identity_suci.h.type = OGS_NAS_5GS_MOBILE_IDENTITY_SUCI;
        mobile_identity_suci.routing_indicator1 = 0;
        mobile_identity_suci.routing_indicator2 = 0xf;
        mobile_identity_suci.routing_indicator3 = 0xf;
        mobile_identity_suci.routing_indicator4 = 0xf;
        mobile_identity_suci.protection_scheme_id = OGS_PROTECTION_SCHEME_NULL;
        mobile_identity_suci.home_network_pki_value = 0;

        test_ue[i] = test_ue_add_by_suci(
                &mobile_identity_suci, scheme_output[i]);
        ogs_assert(test_ue[i]);

        test_ue[i]->e_cgi.cell_id = 0x54f6401;
        test_ue[i]->nas.ksi = OGS_NAS_KSI_NO_KEY_IS_AVAILABLE;
        test_ue[i]->nas.value = OGS_NAS_ATTACH_TYPE_COMBINED_EPS_IMSI_ATTACH;

        test_ue[i]->k_string = "465b5ce8b199b49faa5f0a2ee238a6bc";
        test_ue[i]->opc_string = "e8ed289deba952e4283b54e88e6183ca";

        sess = test_sess_add_by_apn(
                test_ue[i], "internet", OGS_GTP2_RAT_TYPE_EUTRAN);
        ogs_assert(sess);

        /********** Insert Subscriber in Database */
        doc = test_db_new_simple(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, doc);
        ABTS_INT_EQUAL(tc, OGS_OK, test_db_insert_ue(test_ue[i], doc));
    }

    for (i = 0; i < NUM_OF_TEST_UE; i++) {
        if (i > 0)
            test_ue[i]->enb_ue_s1ap_id = test_ue[i-1]->enb_ue_s1ap_id;
        else
            test_ue[i]->enb_ue_s1ap_id = 0;

        sess = test_sess_find_by_apn(
                test_ue[i], "internet", OGS_GTP2_RAT_TYPE_EUTRAN);
        ogs_assert(sess);

        /* Send Attach Request */
        memset(&sess->pdn_connectivity_param,
                0, sizeof(sess->pdn_connectivity_param));
        sess->pdn_connectivity_param.eit = 1;
        sess->pdn_connectivity_param.pco = 1;
        sess->pdn_connectivity_param.request_type =
            OGS_NAS_EPS_REQUEST_TYPE_INITIAL;
        esmbuf = testesm_build_pdn_connectivity_request(sess, false);
        ABTS_PTR_NOTNULL(tc, esmbuf);

        memset(&test_ue[i]->attach_request_param,
                0, sizeof(test_ue[i]->attach_request_param));
        test_ue[i]->attach_request_param.drx_parameter = 1;
        test_ue[i]->attach_request_param.ms_network_capability = 1;
        test_ue[i]->attach_request_param.tmsi_status = 1;
        test_ue[i]->attach_request_param.mobile_station_classmark_2 = 1;
        test_ue[i]->attach_request_param.supported_codecs = 1;
        test_ue[i]->attach_request_param.ue_usage_setting = 1;
        test_ue[i]->attach_request_param.ms_network_feature_support = 1;
        emmbuf = testemm_build_attach_request(test_ue[i], esmbuf, false, false);
        ABTS_PTR_NOTNULL(tc, emmbuf);

        memset(&test_ue[i]->initial_ue_param, 0,
                sizeof(test_ue[i]->initial_ue_param));
        sendbuf = test_s1ap_build_initial_ue_message(
                test_ue[i], emmbuf,
                S1AP_RRC_Establishment_Cause_mo_Signalling, false);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive Authentication Request */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send Authentication response */
        emmbuf = testemm_build_authentication_response(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, emmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], emmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive Security mode Command */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send Security mode complete */
        test_ue[i]->mobile_identity_imeisv_presence = true;
        emmbuf = testemm_build_security_mode_complete(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, emmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], emmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive ESM Information Request */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send ESM Information Response */
        esmbuf = testesm_build_esm_information_response(sess);
        ABTS_PTR_NOTNULL(tc, esmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], esmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive Initial Context Setup Request +
         * Attach Accept +
         * Activate Default Bearer Context Request */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send UE Capability Info Indication */
        sendbuf =
            tests1ap_build_ue_radio_capability_info_indication(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Send Initial Context Setup Response */
        sendbuf = test_s1ap_build_initial_context_setup_response(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Send Attach Complete + Activate default EPS bearer cotext accept */
        test_ue[i]->nr_cgi.cell_id = 0x1234502;
        bearer = test_bearer_find_by_ue_ebi(test_ue[i], 5);
        ogs_assert(bearer);
        esmbuf = testesm_build_activate_default_eps_bearer_context_accept(
                bearer, false);
        ABTS_PTR_NOTNULL(tc, esmbuf);
        emmbuf = testemm_build_attach_complete(test_ue[i], esmbuf);
        ABTS_PTR_NOTNULL(tc, emmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], emmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive EMM information */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);
    }

    /* Send S1-Reset */
    sendbuf = ogs_s1ap_build_s1_reset(
            S1AP_Cause_PR_radioNetwork,
            S1AP_CauseRadioNetwork_release_due_to_eutran_generated_reason,
            NULL);

    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Reset Acknowledge */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    ogs_pkbuf_free(recvbuf);

    ogs_msleep(300);

    for (i = 0; i < NUM_OF_TEST_UE; i++) {
        /********** Remove Subscriber in Database */
        ABTS_INT_EQUAL(tc, OGS_OK, test_db_remove_ue(test_ue[i]));
    }

    /* eNB disonncect from MME */
    testenb_s1ap_close(s1ap);

    /* eNB disonncect from SGW */
    test_gtpu_close(gtpu);

    test_ue_remove_all();
}

static void test3_func(abts_case *tc, void *data)
{
    int rv, i;
    ogs_socknode_t *s1ap;
    ogs_socknode_t *gtpu;
    ogs_pkbuf_t *emmbuf;
    ogs_pkbuf_t *esmbuf;
    ogs_pkbuf_t *sendbuf;
    ogs_pkbuf_t *recvbuf;
    ogs_s1ap_message_t message;

    ogs_nas_5gs_mobile_identity_suci_t mobile_identity_suci;
    test_ue_t *test_ue[NUM_OF_TEST_UE];
    test_sess_t *sess = NULL;
    test_bearer_t *bearer = NULL;

    S1AP_UE_associatedLogicalS1_ConnectionListRes_t *partOfS1_Interface = NULL;

    bson_t *doc = NULL;

    /* eNB connects to MME */
    s1ap = tests1ap_client(AF_INET);
    ABTS_PTR_NOTNULL(tc, s1ap);

    /* eNB connects to SGW */
    gtpu = test_gtpu_server(1, AF_INET);
    ABTS_PTR_NOTNULL(tc, gtpu);

    /* Send S1-Setup Reqeust */
    sendbuf = test_s1ap_build_s1_setup_request(
            S1AP_ENB_ID_PR_macroENB_ID, 0x54f64);
    ABTS_PTR_NOTNULL(tc, sendbuf);
    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Setup Response */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    tests1ap_recv(NULL, recvbuf);

    for (i = 0; i < NUM_OF_TEST_UE; i++) {
        const char *scheme_output[] = {
            "0000000001",
            "0000000002",
            "0000000003",
            "0000000004",
            "0000000005",
        };

        /* Setup Test UE & Session Context */
        memset(&mobile_identity_suci, 0, sizeof(mobile_identity_suci));

        mobile_identity_suci.h.supi_format = OGS_NAS_5GS_SUPI_FORMAT_IMSI;
        mobile_identity_suci.h.type = OGS_NAS_5GS_MOBILE_IDENTITY_SUCI;
        mobile_identity_suci.routing_indicator1 = 0;
        mobile_identity_suci.routing_indicator2 = 0xf;
        mobile_identity_suci.routing_indicator3 = 0xf;
        mobile_identity_suci.routing_indicator4 = 0xf;
        mobile_identity_suci.protection_scheme_id = OGS_PROTECTION_SCHEME_NULL;
        mobile_identity_suci.home_network_pki_value = 0;

        test_ue[i] = test_ue_add_by_suci(
                &mobile_identity_suci, scheme_output[i]);
        ogs_assert(test_ue[i]);

        /* Multiple eNB-UE-S1AP-UD */
        test_ue[i]->enb_ue_s1ap_id = i;

        test_ue[i]->e_cgi.cell_id = 0x54f6401;
        test_ue[i]->nas.ksi = OGS_NAS_KSI_NO_KEY_IS_AVAILABLE;
        test_ue[i]->nas.value = OGS_NAS_ATTACH_TYPE_COMBINED_EPS_IMSI_ATTACH;

        test_ue[i]->k_string = "465b5ce8b199b49faa5f0a2ee238a6bc";
        test_ue[i]->opc_string = "e8ed289deba952e4283b54e88e6183ca";

        sess = test_sess_add_by_apn(
                test_ue[i], "internet", OGS_GTP2_RAT_TYPE_EUTRAN);
        ogs_assert(sess);

        /********** Insert Subscriber in Database */
        doc = test_db_new_simple(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, doc);
        ABTS_INT_EQUAL(tc, OGS_OK, test_db_insert_ue(test_ue[i], doc));
    }

    for (i = 0; i < NUM_OF_TEST_UE; i++) {
        if (i > 0)
            test_ue[i]->enb_ue_s1ap_id = test_ue[i-1]->enb_ue_s1ap_id;
        else
            test_ue[i]->enb_ue_s1ap_id = 0;

        sess = test_sess_find_by_apn(
                test_ue[i], "internet", OGS_GTP2_RAT_TYPE_EUTRAN);
        ogs_assert(sess);

        /* Send Attach Request */
        memset(&sess->pdn_connectivity_param,
                0, sizeof(sess->pdn_connectivity_param));
        sess->pdn_connectivity_param.eit = 1;
        sess->pdn_connectivity_param.pco = 1;
        sess->pdn_connectivity_param.request_type =
            OGS_NAS_EPS_REQUEST_TYPE_INITIAL;
        esmbuf = testesm_build_pdn_connectivity_request(sess, false);
        ABTS_PTR_NOTNULL(tc, esmbuf);

        memset(&test_ue[i]->attach_request_param,
                0, sizeof(test_ue[i]->attach_request_param));
        test_ue[i]->attach_request_param.drx_parameter = 1;
        test_ue[i]->attach_request_param.ms_network_capability = 1;
        test_ue[i]->attach_request_param.tmsi_status = 1;
        test_ue[i]->attach_request_param.mobile_station_classmark_2 = 1;
        test_ue[i]->attach_request_param.supported_codecs = 1;
        test_ue[i]->attach_request_param.ue_usage_setting = 1;
        test_ue[i]->attach_request_param.ms_network_feature_support = 1;
        emmbuf = testemm_build_attach_request(test_ue[i], esmbuf, false, false);
        ABTS_PTR_NOTNULL(tc, emmbuf);

        memset(&test_ue[i]->initial_ue_param, 0,
                sizeof(test_ue[i]->initial_ue_param));
        sendbuf = test_s1ap_build_initial_ue_message(
                test_ue[i], emmbuf,
                S1AP_RRC_Establishment_Cause_mo_Signalling, false);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive Authentication Request */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send Authentication response */
        emmbuf = testemm_build_authentication_response(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, emmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], emmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive Security mode Command */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send Security mode complete */
        test_ue[i]->mobile_identity_imeisv_presence = true;
        emmbuf = testemm_build_security_mode_complete(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, emmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], emmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive ESM Information Request */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send ESM Information Response */
        esmbuf = testesm_build_esm_information_response(sess);
        ABTS_PTR_NOTNULL(tc, esmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], esmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive Initial Context Setup Request +
         * Attach Accept +
         * Activate Default Bearer Context Request */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);

        /* Send UE Capability Info Indication */
        sendbuf =
            tests1ap_build_ue_radio_capability_info_indication(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Send Initial Context Setup Response */
        sendbuf = test_s1ap_build_initial_context_setup_response(test_ue[i]);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Send Attach Complete + Activate default EPS bearer cotext accept */
        test_ue[i]->nr_cgi.cell_id = 0x1234502;
        bearer = test_bearer_find_by_ue_ebi(test_ue[i], 5);
        ogs_assert(bearer);
        esmbuf = testesm_build_activate_default_eps_bearer_context_accept(
                bearer, false);
        ABTS_PTR_NOTNULL(tc, esmbuf);
        emmbuf = testemm_build_attach_complete(test_ue[i], esmbuf);
        ABTS_PTR_NOTNULL(tc, emmbuf);
        sendbuf = test_s1ap_build_uplink_nas_transport(test_ue[i], emmbuf);
        ABTS_PTR_NOTNULL(tc, sendbuf);
        rv = testenb_s1ap_send(s1ap, sendbuf);
        ABTS_INT_EQUAL(tc, OGS_OK, rv);

        /* Receive EMM information */
        recvbuf = testenb_s1ap_read(s1ap);
        ABTS_PTR_NOTNULL(tc, recvbuf);
        tests1ap_recv(test_ue[i], recvbuf);
    }

    /* Send S1-Reset */
    partOfS1_Interface = NULL;

    ogs_s1ap_build_part_of_s1_interface(
            &partOfS1_Interface,
            &test_ue[0]->mme_ue_s1ap_id, &test_ue[0]->enb_ue_s1ap_id);
    ogs_s1ap_build_part_of_s1_interface(
            &partOfS1_Interface,
            &test_ue[2]->mme_ue_s1ap_id, &test_ue[2]->enb_ue_s1ap_id);

    sendbuf = ogs_s1ap_build_s1_reset(
            S1AP_Cause_PR_radioNetwork,
            S1AP_CauseRadioNetwork_release_due_to_eutran_generated_reason,
            partOfS1_Interface);

    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Reset Acknowledge */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    ogs_pkbuf_free(recvbuf);

    /* Send S1-Reset */
    sendbuf = ogs_s1ap_build_s1_reset(
            S1AP_Cause_PR_radioNetwork,
            S1AP_CauseRadioNetwork_release_due_to_eutran_generated_reason,
            NULL);

    rv = testenb_s1ap_send(s1ap, sendbuf);
    ABTS_INT_EQUAL(tc, OGS_OK, rv);

    /* Receive S1-Reset Acknowledge */
    recvbuf = testenb_s1ap_read(s1ap);
    ABTS_PTR_NOTNULL(tc, recvbuf);
    ogs_pkbuf_free(recvbuf);

    ogs_msleep(300);

    for (i = 0; i < NUM_OF_TEST_UE; i++) {
        /********** Remove Subscriber in Database */
        ABTS_INT_EQUAL(tc, OGS_OK, test_db_remove_ue(test_ue[i]));
    }

    /* eNB disonncect from MME */
    testenb_s1ap_close(s1ap);

    /* eNB disonncect from SGW */
    test_gtpu_close(gtpu);

    test_ue_remove_all();
}

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

    abts_run_test(suite, test1_func, NULL);
    abts_run_test(suite, test2_func, NULL);
    abts_run_test(suite, test3_func, NULL);

    return suite;
}