forked from acouzens/open5gs
320 lines
9.8 KiB
C
320 lines
9.8 KiB
C
#define TRACE_MODULE _mme_sm
|
|
#include "core_debug.h"
|
|
|
|
#include "s1ap_message.h"
|
|
#include "nas_message.h"
|
|
|
|
#include "mme_event.h"
|
|
|
|
#include "s1ap_path.h"
|
|
#include "nas_security.h"
|
|
#include "mme_s11_path.h"
|
|
#include "mme_s11_handler.h"
|
|
#include "emm_handler.h"
|
|
|
|
void mme_state_initial(fsm_t *s, event_t *e)
|
|
{
|
|
mme_sm_trace(1, e);
|
|
|
|
d_assert(s, return, "Null param");
|
|
|
|
FSM_TRAN(s, &mme_state_operational);
|
|
}
|
|
|
|
void mme_state_final(fsm_t *s, event_t *e)
|
|
{
|
|
mme_sm_trace(1, e);
|
|
|
|
d_assert(s, return, "Null param");
|
|
}
|
|
|
|
void mme_state_operational(fsm_t *s, event_t *e)
|
|
{
|
|
status_t rv;
|
|
char buf[INET_ADDRSTRLEN];
|
|
|
|
mme_sm_trace(1, e);
|
|
|
|
d_assert(s, return, "Null param");
|
|
|
|
switch (event_get(e))
|
|
{
|
|
case FSM_ENTRY_SIG:
|
|
{
|
|
rv = mme_s11_listen();
|
|
if (rv != CORE_OK)
|
|
{
|
|
d_error("Can't establish S11 path");
|
|
break;
|
|
}
|
|
|
|
rv = s1ap_listen();
|
|
if (rv != CORE_OK)
|
|
{
|
|
d_error("Can't establish S1AP path");
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
case FSM_EXIT_SIG:
|
|
{
|
|
rv = mme_s11_close();
|
|
if (rv != CORE_OK)
|
|
{
|
|
d_error("Can't close S11 path");
|
|
break;
|
|
}
|
|
|
|
rv = s1ap_close();
|
|
if (rv != CORE_OK)
|
|
{
|
|
d_error("Can't close S1AP path");
|
|
break;
|
|
}
|
|
|
|
break;
|
|
}
|
|
case MME_EVT_S1AP_ENB_LO_ACCEPT:
|
|
{
|
|
int rc;
|
|
|
|
net_sock_t *sock = (net_sock_t *)event_get_param1(e);
|
|
d_assert(sock, break, "Null param");
|
|
|
|
d_trace(1, "eNB-S1 accepted[%s] in master_sm module\n",
|
|
INET_NTOP(&sock->remote.sin_addr.s_addr, buf));
|
|
|
|
mme_enb_t *enb = mme_enb_find_by_sock(sock);
|
|
if (!enb)
|
|
{
|
|
rc = net_register_sock(sock, _s1ap_recv_cb, NULL);
|
|
d_assert(rc == 0, break, "register _s1ap_recv_cb failed");
|
|
|
|
mme_enb_t *enb = mme_enb_add(sock);
|
|
d_assert(enb, break, "Null param");
|
|
}
|
|
else
|
|
{
|
|
d_warn("eNB context duplicated with IP-address [%s]!!!",
|
|
INET_NTOP(&sock->remote.sin_addr.s_addr, buf));
|
|
net_close(sock);
|
|
d_warn("S1 Socket Closed");
|
|
}
|
|
|
|
break;
|
|
}
|
|
case MME_EVT_S1AP_ENB_LO_CONNREFUSED:
|
|
{
|
|
index_t index = event_get_param1(e);
|
|
mme_enb_t *enb = NULL;
|
|
|
|
d_assert(index, break, "Null param");
|
|
enb = mme_enb_find(index);
|
|
if (enb)
|
|
{
|
|
d_info("eNB-S1[%x] connection refused!!!", enb->enb_id);
|
|
mme_enb_remove(enb);
|
|
}
|
|
else
|
|
{
|
|
d_warn("Socket connection refused, Already Removed!");
|
|
}
|
|
|
|
break;
|
|
}
|
|
case MME_EVT_S1AP_ENB_MSG:
|
|
{
|
|
s1ap_message_t message;
|
|
index_t index = event_get_param1(e);
|
|
mme_enb_t *enb = NULL;
|
|
pkbuf_t *pkbuf = NULL;
|
|
|
|
d_assert(index, break, "Null param");
|
|
d_assert(enb = mme_enb_find(index), break, "No eNB context");
|
|
d_assert(FSM_STATE(&enb->sm), break, "No S1AP State Machine");
|
|
|
|
pkbuf = (pkbuf_t *)event_get_param2(e);
|
|
d_assert(pkbuf, break, "Null param");
|
|
d_assert(s1ap_decode_pdu(&message, pkbuf) == CORE_OK,
|
|
pkbuf_free(pkbuf); break, "Can't decode S1AP_PDU");
|
|
|
|
event_set_param3(e, (c_uintptr_t)&message);
|
|
fsm_dispatch(&enb->sm, (fsm_event_t*)e);
|
|
|
|
s1ap_free_pdu(&message);
|
|
pkbuf_free(pkbuf);
|
|
break;
|
|
}
|
|
case MME_EVT_EMM_UE_MSG:
|
|
{
|
|
nas_message_t message;
|
|
index_t index = event_get_param1(e);
|
|
pkbuf_t *pkbuf = (pkbuf_t *)event_get_param2(e);
|
|
enb_ue_t *enb_ue = NULL;
|
|
mme_ue_t *mme_ue = NULL;
|
|
|
|
enb_ue = enb_ue_find(index);
|
|
d_assert(enb_ue, break, "No ENB UE context");
|
|
|
|
d_assert(pkbuf, break, "Null param");
|
|
d_assert(nas_emm_decode(&message, pkbuf) == CORE_OK,
|
|
pkbuf_free(pkbuf); break, "Can't decode NAS_EMM");
|
|
|
|
mme_ue = enb_ue->mme_ue;
|
|
if (mme_ue == NULL)
|
|
{
|
|
/* Find MME UE by NAS message or create if needed */
|
|
mme_ue = emm_find_ue_by_message(enb_ue, &message);
|
|
}
|
|
|
|
d_assert(mme_ue, pkbuf_free(pkbuf);break, "No MME UE context");
|
|
d_assert(FSM_STATE(&mme_ue->sm), pkbuf_free(pkbuf);break,
|
|
"No EMM State Machine");
|
|
|
|
/* Set event */
|
|
event_set_param1(e, (c_uintptr_t)mme_ue->index);/* mme_ue index */
|
|
event_set_param3(e, (c_uintptr_t)&message);
|
|
|
|
fsm_dispatch(&mme_ue->sm, (fsm_event_t*)e);
|
|
|
|
pkbuf_free(pkbuf);
|
|
|
|
break;
|
|
}
|
|
case MME_EVT_EMM_UE_FROM_S6A:
|
|
case MME_EVT_EMM_UE_FROM_S11:
|
|
case MME_EVT_EMM_BEARER_FROM_S11:
|
|
case MME_EVT_EMM_UE_T3:
|
|
{
|
|
nas_message_t message;
|
|
index_t index = event_get_param1(e);
|
|
mme_ue_t *mme_ue = NULL;
|
|
mme_bearer_t *bearer = NULL;
|
|
pkbuf_t *pkbuf = NULL;
|
|
|
|
if (event_get(e) == MME_EVT_EMM_BEARER_FROM_S11)
|
|
{
|
|
d_assert(index, break, "Null param");
|
|
bearer = mme_bearer_find(index);
|
|
d_assert(bearer, break, "No ESM context");
|
|
mme_ue = bearer->mme_ue;
|
|
}
|
|
else
|
|
{
|
|
d_assert(index, break, "Null param");
|
|
mme_ue = mme_ue_find(index);
|
|
}
|
|
d_assert(mme_ue, break, "No UE context");
|
|
d_assert(FSM_STATE(&mme_ue->sm), break, "No EMM State Machine");
|
|
|
|
if (event_get(e) == MME_EVT_EMM_UE_MSG)
|
|
{
|
|
pkbuf = (pkbuf_t *)event_get_param2(e);
|
|
d_assert(pkbuf, break, "Null param");
|
|
d_assert(nas_emm_decode(&message, pkbuf) == CORE_OK,
|
|
pkbuf_free(pkbuf); break, "Can't decode NAS_EMM");
|
|
event_set_param3(e, (c_uintptr_t)&message);
|
|
}
|
|
|
|
fsm_dispatch(&mme_ue->sm, (fsm_event_t*)e);
|
|
|
|
if (event_get(e) == MME_EVT_EMM_UE_MSG)
|
|
{
|
|
pkbuf_free(pkbuf);
|
|
}
|
|
break;
|
|
}
|
|
case MME_EVT_ESM_BEARER_FROM_S6A:
|
|
case MME_EVT_ESM_BEARER_TO_S11:
|
|
case MME_EVT_ESM_BEARER_MSG:
|
|
{
|
|
nas_message_t message;
|
|
index_t index = event_get_param1(e);
|
|
mme_bearer_t *bearer = NULL;
|
|
mme_ue_t *mme_ue = NULL;
|
|
pkbuf_t *pkbuf = NULL;
|
|
|
|
d_assert(index, break, "Null param");
|
|
bearer = mme_bearer_find(index);
|
|
d_assert(bearer, break, "No ESM context");
|
|
d_assert(mme_ue = bearer->mme_ue, break, "No UE context");
|
|
d_assert(FSM_STATE(&bearer->sm), break, "No ESM State Machine");
|
|
|
|
if (event_get(e) == MME_EVT_ESM_BEARER_MSG)
|
|
{
|
|
pkbuf = (pkbuf_t *)event_get_param2(e);
|
|
d_assert(pkbuf, break, "Null param");
|
|
d_assert(nas_esm_decode(&message, pkbuf) == CORE_OK,
|
|
pkbuf_free(pkbuf); break, "Can't decode NAS_ESM");
|
|
|
|
event_set_param3(e, (c_uintptr_t)&message);
|
|
}
|
|
|
|
fsm_dispatch(&bearer->sm, (fsm_event_t*)e);
|
|
|
|
if (event_get(e) == MME_EVT_ESM_BEARER_MSG)
|
|
{
|
|
pkbuf_free(pkbuf);
|
|
}
|
|
|
|
break;
|
|
}
|
|
case MME_EVT_S11_UE_MSG:
|
|
{
|
|
status_t rv;
|
|
net_sock_t *sock = (net_sock_t *)event_get_param1(e);
|
|
gtp_node_t *gnode = (gtp_node_t *)event_get_param2(e);
|
|
pkbuf_t *pkbuf = (pkbuf_t *)event_get_param3(e);
|
|
gtp_xact_t *xact = NULL;
|
|
c_uint8_t type;
|
|
c_uint32_t teid;
|
|
gtp_message_t gtp_message;
|
|
mme_sess_t *sess = NULL;
|
|
|
|
d_assert(pkbuf, break, "Null param");
|
|
d_assert(sock, pkbuf_free(pkbuf); break, "Null param");
|
|
d_assert(gnode, pkbuf_free(pkbuf); break, "Null param");
|
|
|
|
rv = gtp_xact_receive(
|
|
&mme_self()->gtp_xact_ctx, sock, gnode,
|
|
&xact, &type, &teid, >p_message, pkbuf);
|
|
if (rv != CORE_OK)
|
|
break;
|
|
|
|
sess = mme_sess_find_by_teid(teid);
|
|
d_assert(sess, pkbuf_free(pkbuf); break,
|
|
"No Session Context(TEID:%d)", teid);
|
|
switch(type)
|
|
{
|
|
case GTP_CREATE_SESSION_RESPONSE_TYPE:
|
|
mme_s11_handle_create_session_response(
|
|
sess, >p_message.create_session_response);
|
|
break;
|
|
case GTP_MODIFY_BEARER_RESPONSE_TYPE:
|
|
mme_s11_handle_modify_bearer_response(
|
|
sess, >p_message.modify_bearer_response);
|
|
break;
|
|
case GTP_DELETE_SESSION_RESPONSE_TYPE:
|
|
mme_s11_handle_delete_session_response(
|
|
sess, >p_message.delete_session_response);
|
|
break;
|
|
default:
|
|
d_warn("Not implmeneted(type:%d)", type);
|
|
break;
|
|
}
|
|
pkbuf_free(pkbuf);
|
|
break;
|
|
}
|
|
case MME_EVT_S11_TRANSACTION_T3:
|
|
{
|
|
gtp_xact_timeout(event_get_param1(e));
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
d_error("No handler for event %s", mme_event_get_name(e));
|
|
break;
|
|
}
|
|
}
|
|
}
|