open5gs/src/init.c

/**
 * @file init.c
 */

/* Core library */
#define TRACE_MODULE _init
#include "core_debug.h"
#include "core_thread.h"

#include "s6a_app.h"

#include "context.h"
#include "event.h"

#define EVENT_WAIT_TIMEOUT 10000 /* 10 msec */

static thread_id mme_sm_thread;
static thread_id mme_net_thread;

status_t cellwire_initialize(char *config_path)
{
    status_t rv;

    srand(time(NULL)*getpid());

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

    return CORE_OK;
}

void cellwire_terminate(void)
{
    context_final();
}

void *THREAD_FUNC mme_sm_main(void *data)
{
    event_t event;
    msgq_id queue_id;
    mme_sm_t mme_sm;
    c_time_t prev_tm, now_tm;
    int r;

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

    queue_id = event_create();
    d_assert(queue_id, return NULL, "MME event queue creation failed");

    fsm_create(&mme_sm.fsm, mme_state_initial, mme_state_final);
    d_assert(&mme_sm.fsm, return NULL, "MME state machine creation failed");
    mme_sm.queue_id = queue_id;
    tm_service_init(&mme_sm.tm_service);

    fsm_init((fsm_t*)&mme_sm, 0);

    prev_tm = time_now();

    while ((!thread_should_stop()))
    {
        r = event_timedrecv(queue_id, &event, EVENT_WAIT_TIMEOUT);

        d_assert(r != 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)
        {
            event_timer_execute(&mme_sm.tm_service);

            prev_tm = now_tm;
        }

        if (r == CORE_TIMEUP)
        {
            continue;
        }

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

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

    event_delete(queue_id);

    return NULL;
}

void *THREAD_FUNC mme_net_main(void *data)
{
    while (!thread_should_stop())
    {
        net_fds_read_run(50); 
    }

    return NULL;
}

void threads_start()
{
    status_t rv;

    rv = thread_create(&mme_sm_thread, NULL, mme_sm_main, NULL);
    d_assert(rv == CORE_OK, return,
            "MME State machine thread creation failed");
    rv = thread_create(&mme_net_thread, NULL, mme_net_main, NULL);
    d_assert(rv == CORE_OK, return,
            "MME Network socket recv thread creation failed");
}

void threads_stop()
{
    thread_delete(mme_net_thread);
    thread_delete(mme_sm_thread);
}