/*
 * Copyright 2019-present Infosys Limited
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <typeinfo>
#include "actionHandlers/actionHandlers.h"
#include "controlBlock.h"
#include "msgType.h"
#include "contextManager/subsDataGroupManager.h"
#include "contextManager/dataBlocks.h"
#include "procedureStats.h"
#include "log.h"
#include "secUtils.h"
#include "state.h"
#include <string.h>
#include <sstream>
#include <smTypes.h>

#include <ipcTypes.h>
#include <tipcTypes.h>
#include <msgBuffer.h>
#include <interfaces/mmeIpcInterface.h>
#include <utils/mmeContextManagerUtils.h>

using namespace SM;
using namespace mme;
using namespace cmn::utils;

extern MmeIpcInterface* mmeIpcIf_g;

ActStatus ActionHandlers::ni_detach_req_to_ue(SM::ControlBlock& cb)
{
	log_msg(LOG_DEBUG, "Inside ni_detach_req_to_ue \n");
	
	UEContext *ue_ctxt =  dynamic_cast<UEContext*>(cb.getPermDataBlock());
	
	if (ue_ctxt == NULL)
	{
		log_msg(LOG_DEBUG, "ni_detach_req_to_ue: ue context is NULL\n");
		return ActStatus::HALT;
	}
	
	ni_detach_request_Q_msg ni_detach_req;
	
	ni_detach_req.msg_type = ni_detach_request;
	ni_detach_req.enb_fd = ue_ctxt->getEnbFd();
	ni_detach_req.ue_idx = ue_ctxt->getContextID();
	ni_detach_req.enb_s1ap_ue_id =  ue_ctxt->getS1apEnbUeId();
	ni_detach_req.detach_type = 00000010;
	
	ue_ctxt->setDwnLnkSeqNo(ue_ctxt->getDwnLnkSeqNo()+1);
	ni_detach_req.dl_seq_no = ue_ctxt->getDwnLnkSeqNo();
	
	memcpy(&(ni_detach_req.int_key), &(ue_ctxt->getUeSecInfo().secinfo_m.int_key), NAS_INT_KEY_SIZE);
	
	/* Send message to S1app in S1q*/
	cmn::ipc::IpcAddress destAddr;
	destAddr.u32 = TipcServiceInstance::s1apAppInstanceNum_c;

	mmeIpcIf_g->dispatchIpcMsg((char *) &ni_detach_req, sizeof(ni_detach_req), destAddr);
	
	log_msg(LOG_DEBUG, "Leaving ni_detach_req_to_ue \n");

	ProcedureStats::num_of_clr_received ++;
	ProcedureStats::num_of_detach_req_to_ue_sent ++;

	return ActStatus::PROCEED;
}

ActStatus ActionHandlers::process_detach_accept_from_ue(SM::ControlBlock& cb)
{
	log_msg(LOG_DEBUG, "Inside process_detach_accept_from_ue \n");
		
	UEContext *ue_ctxt = dynamic_cast<UEContext*>(cb.getPermDataBlock());
	
	if (ue_ctxt == NULL)
	{
		log_msg(LOG_DEBUG, "process_detach_accept_from_ue: ue context is NULL\n");
		return ActStatus::HALT;
	}
		
	ue_ctxt->setUpLnkSeqNo(ue_ctxt->getUpLnkSeqNo()+1);
	
	log_msg(LOG_DEBUG, "Leaving process_detach_accept_from_ue \n");

	ProcedureStats::num_of_cla_sent ++;
	ProcedureStats::num_of_detach_accept_from_ue ++;

	return ActStatus::PROCEED;
}

/***************************************
* Action handler : send_s1_rel_cmd_to_ue_for_detach
***************************************/
ActStatus ActionHandlers::send_s1_rel_cmd_to_ue_for_detach(ControlBlock& cb)
{
    log_msg(LOG_DEBUG, "Inside send_s1_rel_cmd_to_ue_for_detach\n");

    UEContext *ue_ctxt = dynamic_cast<UEContext*>(cb.getPermDataBlock());
    if(ue_ctxt == NULL)
    {
            log_msg(LOG_DEBUG, "send_s1_rel_cmd_to_ue_for_detach: ue context is NULL \n");
            return ActStatus::HALT;
    }

    struct s1relcmd_info s1relcmd;

    s1relcmd.msg_type = s1_release_command;
    s1relcmd.ue_idx = ue_ctxt->getContextID();
    s1relcmd.enb_fd = ue_ctxt->getEnbFd();
    s1relcmd.enb_s1ap_ue_id = ue_ctxt->getS1apEnbUeId();
    s1relcmd.cause.present = s1apCause_PR_radioNetwork;
    s1relcmd.cause.choice.radioNetwork = s1apCauseRadioNetwork_user_inactivity;

    /*Send message to S1AP-APP*/
    cmn::ipc::IpcAddress destAddr;
    destAddr.u32 = TipcServiceInstance::s1apAppInstanceNum_c;
    mmeIpcIf_g->dispatchIpcMsg((char *) &s1relcmd, sizeof(s1relcmd), destAddr);

    log_msg(LOG_DEBUG,"Leaving send_s1_rel_cmd_to_ue \n");

    ProcedureStats::num_of_s1_rel_cmd_sent ++;
    return ActStatus::PROCEED;
}

/************************************************************
* Action handler : process_ue_ctxt_rel_comp_for_detach
**************************************************************/
ActStatus ActionHandlers::process_ue_ctxt_rel_comp_for_detach(ControlBlock& cb)
{
    log_msg(LOG_DEBUG, "Inside process_ue_ctxt_rel_comp_for_detach \n");

    UEContext *ueCtxt = dynamic_cast<UEContext*>(cb.getPermDataBlock());
    MmeDetachProcedureCtxt *procCtxt = dynamic_cast<MmeDetachProcedureCtxt*>(cb.getTempDataBlock());
    if (ueCtxt == NULL || procCtxt == NULL)
    {
    	log_msg(LOG_DEBUG, "UE context or  procedure context is NULL\n");
    	return ActStatus::HALT;
    }

    MmContext* mmCtxt = ueCtxt->getMmContext();
    if (mmCtxt == NULL)
    {
    	log_msg(LOG_DEBUG, "MM context is NULL \n");
    	return ActStatus::HALT;
    }

    if(procCtxt->getCancellationType() == SUBSCRIPTION_WITHDRAWAL)
    {
        MmeContextManagerUtils::deleteUEContext(cb.getCBIndex());
    }
    else
    {
    	mmCtxt->setMmState( EpsDetached );
    	MmeContextManagerUtils::deallocateProcedureCtxt(cb, detach_c);
    }

    ProcedureStats::num_of_subscribers_detached ++;
    ProcedureStats::num_of_subscribers_attached --;

    log_msg(LOG_DEBUG, "Leaving process_ue_ctxt_rel_comp_for_detach \n");

    return ActStatus::PROCEED;

}