/*****************************************************************************
            Eurecom OpenAirInterface 3
            Copyright(c) 2012 Eurecom

Source      emm_fsm.c

Version     0.1

Date        2012/10/03

Product     NAS stack

Subsystem   EPS Mobility Management

Author      Frederic Maurel

Description Defines the EPS Mobility Management procedures executed at
        the EMMREG Service Access Point.

*****************************************************************************/

#include "emm_fsm.h"
#include "commonDef.h"
#include "nas_log.h"

#include "mme_api.h"
#include "emmData.h"

#if defined(EPC_BUILD)
# include "assertions.h"
#endif

/****************************************************************************/
/****************  E X T E R N A L    D E F I N I T I O N S  ****************/
/****************************************************************************/

#ifdef NAS_UE
#define EMM_FSM_NB_UE_MAX   1
#endif
#ifdef NAS_MME
#define EMM_FSM_NB_UE_MAX   (MME_API_NB_UE_MAX + 1)
#endif

/****************************************************************************/
/*******************  L O C A L    D E F I N I T I O N S  *******************/
/****************************************************************************/

/*
 * -----------------------------------------------------------------------------
 *          Data used for trace logging
 * -----------------------------------------------------------------------------
 */

/* String representation of EMM events */
static const char *_emm_fsm_event_str[] = {
#ifdef NAS_UE
    "S1_ENABLED",
    "S1_DISABLED",
    "NO_IMSI",
    "NO_CELL",
    "REGISTER_REQ",
    "REGISTER_CNF",
    "REGISTER_REJ",
    "ATTACH_INIT",
    "ATTACH_REQ",
    "ATTACH_FAILED",
    "ATTACH_EXCEEDED",
    "AUTHENTICATION_REJ",
#endif
#ifdef NAS_MME
    "COMMON_PROC_REQ",
    "COMMON_PROC_CNF",
    "COMMON_PROC_REJ",
    "PROC_ABORT",
#endif
    "ATTACH_CNF",
    "ATTACH_REJ",
    "DETACH_INIT",
    "DETACH_REQ",
    "DETACH_FAILED",
    "DETACH_CNF",
    "TAU_REQ",
    "TAU_CNF",
    "TAU_REJ",
    "SERVICE_REQ",
    "SERVICE_CNF",
    "SERVICE_REJ",
    "LOWERLAYER_SUCCESS",
    "LOWERLAYER_FAILURE",
    "LOWERLAYER_RELEASE",
};

/* String representation of EMM status */
static const char *_emm_fsm_status_str[EMM_STATE_MAX] = {
#ifdef NAS_UE
    "NULL",
#endif
    "DEREGISTERED",
    "REGISTERED",
    "DEREGISTERED-INITIATED",
#ifdef NAS_UE
    "DEREGISTERED.NORMAL-SERVICE",
    "DEREGISTERED.LIMITED-SERVICE",
    "DEREGISTERED.ATTEMPTING-TO-ATTACH",
    "DEREGISTERED.PLMN-SEARCH",
    "DEREGISTERED.NO-IMSI",
    "DEREGISTERED.ATTACH-NEEDED",
    "DEREGISTERED.NO-CELL-AVAILABLE",
    "REGISTERED-INITIATED",
    "REGISTERED.NORMAL-SERVICE",
    "REGISTERED.ATTEMPTING-TO-PDATE",
    "REGISTERED.LIMITED-SERVICE",
    "REGISTERED.PLMN-SEARCH",
    "REGISTERED.UPDATE-NEEDED",
    "REGISTERED.NO-CELL-AVAILABLE",
    "REGISTERED.ATTEMPTING-TO-UPDATE-MM",
    "REGISTERED.IMSI-DETACH-INITIATED",
    "TRACKING-AREA-UPDATING-INITIATED",
    "SERVICE-REQUEST-INITIATED",
#endif
#ifdef NAS_MME
    "COMMON-PROCEDURE-INITIATED",
#endif
};

/*
 * -----------------------------------------------------------------------------
 *      EPS Mobility Management state machine handlers
 * -----------------------------------------------------------------------------
 */

/* Type of the EPS Mobility Management state machine handler */
typedef int(*emm_fsm_handler_t)(const emm_reg_t *);

#ifdef NAS_UE
int EmmNull(const emm_reg_t *);
#endif
int EmmDeregistered(const emm_reg_t *);
int EmmRegistered(const emm_reg_t *);
int EmmDeregisteredInitiated(const emm_reg_t *);
#ifdef NAS_UE
int EmmDeregisteredNormalService(const emm_reg_t *);
int EmmDeregisteredLimitedService(const emm_reg_t *);
int EmmDeregisteredAttemptingToAttach(const emm_reg_t *);
int EmmDeregisteredPlmnSearch(const emm_reg_t *);
int EmmDeregisteredNoImsi(const emm_reg_t *);
int EmmDeregisteredAttachNeeded(const emm_reg_t *);
int EmmDeregisteredNoCellAvailable(const emm_reg_t *);
int EmmRegisteredInitiated(const emm_reg_t *);
int EmmRegisteredNormalService(const emm_reg_t *);
int EmmRegisteredAttemptingToUpdate(const emm_reg_t *);
int EmmRegisteredLimitedService(const emm_reg_t *);
int EmmRegisteredPlmnSearch(const emm_reg_t *);
int EmmRegisteredUpdateNeeded(const emm_reg_t *);
int EmmRegisteredNoCellAvailable(const emm_reg_t *);
int EmmRegisteredAttemptingToUpdate(const emm_reg_t *);
int EmmRegisteredImsiDetachInitiated(const emm_reg_t *);
int EmmTrackingAreaUpdatingInitiated(const emm_reg_t *);
int EmmServiceRequestInitiated(const emm_reg_t *);
#endif
#ifdef NAS_MME
int EmmCommonProcedureInitiated(const emm_reg_t *);
#endif

/* EMM state machine handlers */
static const emm_fsm_handler_t _emm_fsm_handlers[EMM_STATE_MAX] = {
#ifdef NAS_UE
    EmmNull,
#endif
    EmmDeregistered,
    EmmRegistered,
    EmmDeregisteredInitiated,
#ifdef NAS_UE
    EmmDeregisteredNormalService,
    EmmDeregisteredLimitedService,
    EmmDeregisteredAttemptingToAttach,
    EmmDeregisteredPlmnSearch,
    EmmDeregisteredNoImsi,
    EmmDeregisteredAttachNeeded,
    EmmDeregisteredNoCellAvailable,
    EmmRegisteredInitiated,
    EmmRegisteredNormalService,
    EmmRegisteredAttemptingToUpdate,
    EmmRegisteredLimitedService,
    EmmRegisteredPlmnSearch,
    EmmRegisteredUpdateNeeded,
    EmmRegisteredNoCellAvailable,
    EmmRegisteredAttemptingToUpdate,
    EmmRegisteredImsiDetachInitiated,
    EmmTrackingAreaUpdatingInitiated,
    EmmServiceRequestInitiated,
#endif
#ifdef NAS_MME
    EmmCommonProcedureInitiated,
#endif
};

/*
 * -----------------------------------------------------------------------------
 *          Current EPS Mobility Management status
 * -----------------------------------------------------------------------------
 */

#if !defined(EPC_BUILD)
emm_fsm_state_t _emm_fsm_status[EMM_FSM_NB_UE_MAX];
#endif

/****************************************************************************/
/******************  E X P O R T E D    F U N C T I O N S  ******************/
/****************************************************************************/

/****************************************************************************
 **                                                                        **
 ** Name:    emm_fsm_initialize()                                      **
 **                                                                        **
 ** Description: Initializes the EMM state machine                         **
 **                                                                        **
 ** Inputs:  None                                                      **
 **      Others:    None                                       **
 **                                                                        **
 ** Outputs:     None                                                      **
 **      Return:    None                                       **
 **      Others:    _emm_fsm_status                            **
 **                                                                        **
 ***************************************************************************/
void emm_fsm_initialize(void)
{
    int ueid;
    LOG_FUNC_IN;

#ifdef NAS_UE
    _emm_fsm_status[0] = EMM_NULL;
#endif

#if defined(NAS_MME) && !defined(EPC_BUILD)
    for (ueid = 0; ueid < EMM_FSM_NB_UE_MAX; ueid++) {
        _emm_fsm_status[ueid] = EMM_DEREGISTERED;
    }
#endif

    LOG_FUNC_OUT;
}

/****************************************************************************
 **                                                                        **
 ** Name:    emm_fsm_set_status()                                      **
 **                                                                        **
 ** Description: Set the EPS Mobility Management status to the given state **
 **                                                                        **
 ** Inputs:  ueid:      Lower layers UE identifier                 **
 **      status:    The new EMM status                         **
 **      Others:    None                                       **
 **                                                                        **
 ** Outputs:     None                                                      **
 **      Return:    RETURNok, RETURNerror                      **
 **      Others:    _emm_fsm_status                            **
 **                                                                        **
 ***************************************************************************/
int emm_fsm_set_status(
#ifdef NAS_MME
    unsigned int     ueid,
    void            *ctx,
#endif
    emm_fsm_state_t  status)
{
    LOG_FUNC_IN;

#ifdef NAS_UE
    unsigned int ueid = 0;
#endif

#if defined(EPC_BUILD)
    emm_data_context_t *emm_ctx = (emm_data_context_t *)ctx;

    DevAssert(emm_ctx != NULL);

    if ((status < EMM_STATE_MAX) && (ueid > 0)) {
        LOG_TRACE(INFO, "EMM-FSM   - Status changed: %s ===> %s",
                  _emm_fsm_status_str[emm_ctx->_emm_fsm_status],
                  _emm_fsm_status_str[status]);
        if (status != emm_ctx->_emm_fsm_status) {
            emm_ctx->_emm_fsm_status = status;
        }
        LOG_FUNC_RETURN (RETURNok);
    }
#else
    if ( (status < EMM_STATE_MAX) && (ueid < EMM_FSM_NB_UE_MAX) ) {
        LOG_TRACE(INFO, "EMM-FSM   - Status changed: %s ===> %s",
                  _emm_fsm_status_str[_emm_fsm_status[ueid]],
                  _emm_fsm_status_str[status]);
        if (status != _emm_fsm_status[ueid]) {
            _emm_fsm_status[ueid] = status;
        }
        LOG_FUNC_RETURN (RETURNok);
    }
#endif

    LOG_FUNC_RETURN (RETURNerror);
}

/****************************************************************************
 **                                                                        **
 ** Name:    emm_fsm_get_status()                                      **
 **                                                                        **
 ** Description: Get the current value of the EPS Mobility Management      **
 **      status                                                    **
 **                                                                        **
 ** Inputs:  ueid:      Lower layers UE identifier                 **
 **      Others:    _emm_fsm_status                            **
 **                                                                        **
 ** Outputs:     None                                                      **
 **      Return:    The current value of the EMM status        **
 **      Others:    None                                       **
 **                                                                        **
 ***************************************************************************/
#ifdef NAS_UE
emm_fsm_state_t emm_fsm_get_status(void)
{
    return (_emm_fsm_status[0]);
}
#endif

#ifdef NAS_MME
emm_fsm_state_t emm_fsm_get_status(unsigned int ueid, void *ctx)
{
# if defined(EPC_BUILD)
    emm_data_context_t *emm_ctx = (emm_data_context_t *)ctx;

    if (emm_ctx != NULL) {
        return emm_ctx->_emm_fsm_status;
    }
# else
    if (ueid < EMM_FSM_NB_UE_MAX) {
        return (_emm_fsm_status[ueid]);
    }
#endif
    return EMM_STATE_MAX;
}
#endif

/****************************************************************************
 **                                                                        **
 ** Name:    emm_fsm_process()                                         **
 **                                                                        **
 ** Description: Executes the EMM state machine                            **
 **                                                                        **
 ** Inputs:  evt:       The EMMREG-SAP event to process            **
 **      Others:    _emm_fsm_status                            **
 **                                                                        **
 ** Outputs:     None                                                      **
 **      Return:    RETURNok, RETURNerror                      **
 **      Others:    None                                       **
 **                                                                        **
 ***************************************************************************/
int emm_fsm_process(const emm_reg_t *evt)
{
    emm_fsm_state_t status;

    LOG_FUNC_IN;

    int rc;
    emm_reg_primitive_t primitive = evt->primitive;

#ifdef NAS_UE
    status = _emm_fsm_status[0];
#endif
#ifdef NAS_MME
# if defined(EPC_BUILD)
    emm_data_context_t *emm_ctx = (emm_data_context_t *)evt->ctx;

    DevAssert(emm_ctx != NULL);

    status = emm_ctx->_emm_fsm_status;
# else
    if (evt->ueid >= EMM_FSM_NB_UE_MAX) {
        LOG_FUNC_RETURN (RETURNerror);
    }
    status = _emm_fsm_status[evt->ueid];
# endif
#endif

    LOG_TRACE(INFO, "EMM-FSM   - Received event %s (%d) in state %s",
              _emm_fsm_event_str[primitive - _EMMREG_START - 1], primitive,
              _emm_fsm_status_str[status]);

    /* Execute the EMM state machine */
    rc = (_emm_fsm_handlers[status])(evt);

    LOG_FUNC_RETURN (rc);
}

/****************************************************************************/
/*********************  L O C A L    F U N C T I O N S  *********************/
/****************************************************************************/