#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: case MME_EVT_S1AP_UE_FROM_EMM: case MME_EVT_S1AP_UE_FROM_S11: { 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"); if (event_get(e) == MME_EVT_S1AP_ENB_MSG) { 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); if (event_get(e) == MME_EVT_S1AP_ENB_MSG) { 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) { mme_ue = mme_ue_find_by_message(&message); if (!mme_ue) { mme_ue = mme_ue_add(enb_ue); d_assert(mme_ue, break, "Null param"); } } 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: { index_t index = event_get_param1(e); mme_ue_t *mme_ue = NULL; mme_bearer_t *bearer = 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"); fsm_dispatch(&mme_ue->sm, (fsm_event_t*)e); 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; case GTP_RELEASE_ACCESS_BEARERS_RESPONSE_TYPE: mme_s11_handle_release_access_bearers_response( sess, >p_message.release_access_bearers_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; } } }