forked from acouzens/open5gs
139 lines
3.0 KiB
C
139 lines
3.0 KiB
C
#define TRACE_MODULE _mme_init
|
|
|
|
#include "core_debug.h"
|
|
#include "core_thread.h"
|
|
#include "core_msgq.h"
|
|
#include "core_fsm.h"
|
|
|
|
#include "gtp/gtp_xact.h"
|
|
|
|
#include "mme_event.h"
|
|
|
|
#include "mme_fd_path.h"
|
|
#include "s1ap_path.h"
|
|
|
|
#include "mme_sm.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;
|
|
|
|
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;
|
|
|
|
rv = mme_m_tmsi_pool_generate();
|
|
if (rv != CORE_OK) return rv;
|
|
|
|
rv = mme_fd_init();
|
|
if (rv != CORE_OK) return CORE_ERROR;
|
|
|
|
#define USRSCTP_LOCAL_UDP_PORT 9899
|
|
rv = s1ap_init(USRSCTP_LOCAL_UDP_PORT);
|
|
if (rv != CORE_OK) return rv;
|
|
|
|
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_fd_final();
|
|
|
|
mme_context_final();
|
|
|
|
s1ap_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(MSGQ_O_BLOCK);
|
|
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()->tm_service,
|
|
MME_EVT_S11_T3_RESPONSE, MME_EVT_S11_T3_HOLDING);
|
|
|
|
fsm_create(&mme_sm, mme_state_initial, mme_state_final);
|
|
fsm_init(&mme_sm, 0);
|
|
|
|
prev_tm = time_now();
|
|
|
|
#define EVENT_LOOP_TIMEOUT 50 /* 50ms */
|
|
while ((!thread_should_stop()))
|
|
{
|
|
rv = event_timedrecv(mme_self()->queue_id, &event, EVENT_LOOP_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_LOOP_TIMEOUT * 1000)
|
|
{
|
|
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())
|
|
{
|
|
sock_select_loop(EVENT_LOOP_TIMEOUT);
|
|
}
|
|
|
|
return NULL;
|
|
}
|