open5gs/src/mme/mme_init.c

#define TRACE_MODULE _mme_init

#include "core_debug.h"
#include "core_thread.h"

#include "mme_event.h"

#include "mme_s6a_handler.h"

static thread_id sm_thread;
static void *THREAD_FUNC sm_main(thread_id id, void *data);

static thread_id net_thread;
static void *THREAD_FUNC net_main(thread_id id, void *data);

static int initialized = 0;

status_t mme_initialize()
{
    status_t rv;
    int ret;

    rv = mme_context_init();
    if (rv != CORE_OK) return rv;

    rv = mme_context_parse_config();
    if (rv != CORE_OK) return rv;

    rv = mme_context_setup_trace_module();
    if (rv != CORE_OK) return rv;

    ret = mme_s6a_init();
    if (ret != 0) return CORE_ERROR;

    rv = thread_create(&sm_thread, NULL, sm_main, NULL);
    if (rv != CORE_OK) return rv;
    rv = thread_create(&net_thread, NULL, net_main, NULL);
    if (rv != CORE_OK) return rv;
    
    initialized = 1;

    return CORE_OK;
}

void mme_terminate(void)
{
    if (!initialized) return;

    thread_delete(net_thread);
    thread_delete(sm_thread);

    mme_s6a_final();

    mme_context_final();

    gtp_xact_final();
}

static void *THREAD_FUNC sm_main(thread_id id, void *data)
{
    event_t event;
    fsm_t mme_sm;
    c_time_t prev_tm, now_tm;
    status_t rv;

    memset(&event, 0, sizeof(event_t));

    mme_self()->queue_id = event_create();
    d_assert(mme_self()->queue_id, return NULL, 
            "MME event queue creation failed");
    tm_service_init(&mme_self()->tm_service);
    gtp_xact_init(&mme_self()->gtp_xact_ctx, 
            &mme_self()->tm_service, MME_EVT_S11_TRANSACTION_T3);

    fsm_create(&mme_sm, mme_state_initial, mme_state_final);
    fsm_init(&mme_sm, 0);

    prev_tm = time_now();

    while ((!thread_should_stop()))
    {
        rv = event_timedrecv(mme_self()->queue_id, &event, EVENT_WAIT_TIMEOUT);

        d_assert(rv != CORE_ERROR, continue,
                "While receiving a event message, error occurs");

        now_tm = time_now();

        /* if the gap is over 10 ms, execute preriodic jobs */
        if (now_tm - prev_tm > EVENT_WAIT_TIMEOUT)
        {
            tm_execute_tm_service(
                    &mme_self()->tm_service, mme_self()->queue_id);

            prev_tm = now_tm;
        }

        if (rv == CORE_TIMEUP)
        {
            continue;
        }

        fsm_dispatch(&mme_sm, (fsm_event_t*)&event);
    }

    fsm_final(&mme_sm, 0);
    fsm_clear(&mme_sm);

    event_delete(mme_self()->queue_id);

    return NULL;
}

static void *THREAD_FUNC net_main(thread_id id, void *data)
{
    while (!thread_should_stop())
    {
        net_fds_read_run(50); 
    }

    return NULL;
}