/*
 * Copyright (c) 2019, Infosys Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <msgHandlers/s1MsgHandler.h>

#include <event.h>
#include <ipcTypes.h>
#include <log.h>
#include <utils/mmeCommonUtils.h>
#include <contextManager/subsDataGroupManager.h>

using namespace SM;
using namespace mme;

S1MsgHandler::S1MsgHandler()
{

}

S1MsgHandler::~S1MsgHandler()
{

}

S1MsgHandler* S1MsgHandler::Instance()
{
	static S1MsgHandler msgHandler;
	return &msgHandler;
}

void S1MsgHandler::handleS1Message_v(const cmn::utils::MsgBuffer* buf)
{
	log_msg(LOG_INFO, "S1 - handleS1Message_v\n");

	if (buf == NULL)
		return;

	cmn::utils::MsgBuffer* msgBuf = const_cast<cmn::utils::MsgBuffer *>(buf);

	const s1_incoming_msg_data_t* msgData_p = (s1_incoming_msg_data_t*)(msgBuf->getDataPointer());

	switch (msgData_p->msg_type)
	{
		case msg_type_t::attach_request:
			handleInitUeAttachRequestMsg_v(msgBuf);
			break;

		case msg_type_t::id_response:
			handleIdentityResponseMsg_v(msgBuf, msgData_p->ue_idx);
			break;

		case msg_type_t::auth_response:
			handleAuthResponseMsg_v(msgBuf, msgData_p->ue_idx);
			break;

		case msg_type_t::sec_mode_complete:
			handleSecurityModeResponse_v(msgBuf, msgData_p->ue_idx);
			break;

		case msg_type_t::esm_info_response:
			handleEsmInfoResponse_v(msgBuf, msgData_p->ue_idx);
			break;

		case msg_type_t::init_ctxt_response:
			handleInitCtxtResponse_v(msgBuf, msgData_p->ue_idx);
			break;

		case msg_type_t::attach_complete:
			handleAttachComplete_v(msgBuf, msgData_p->ue_idx);
			break;
                
		case msg_type_t::detach_request:
			handleDetachRequest_v(msgBuf, msgData_p->ue_idx);
			break;
					
		case msg_type_t::s1_release_request:
			handleS1ReleaseRequestMsg_v(msgBuf, msgData_p->ue_idx);
			break;
			
		case msg_type_t::s1_release_complete:
			handleS1ReleaseComplete_v(msgBuf, msgData_p->ue_idx);
			break;
		
		case msg_type_t::detach_accept_from_ue:
			handleDetachAcceptFromUE_v(msgBuf, msgData_p->ue_idx);
			break;

		case  msg_type_t::service_request:
            		handleServiceRequest_v(msgBuf, msgData_p->ue_idx);
            		break;
					
		case msg_type_t::tau_request:
			handleTauRequestMsg_v(msgBuf, msgData_p->ue_idx);
			break;
		
		default:
			log_msg(LOG_INFO, "Unhandled S1 Message %d \n", msgData_p->msg_type);
			delete msgBuf;
	}
}

void S1MsgHandler::handleInitUeAttachRequestMsg_v(const cmn::utils::MsgBuffer* msgData_p)
{
	log_msg(LOG_INFO, "S1 - handleInitUeAttachRequestMsg_v\n");

	SM::ControlBlock* controlBlk_p = MmeCommonUtils::findControlBlock(
					const_cast<cmn::utils::MsgBuffer*>(msgData_p));
	if (controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "Failed to allocate ControlBlock \n");

        return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::ATTACH_REQ_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleIdentityResponseMsg_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleIdentityResponseMsg_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleIdentityResponseMsg_v: "
							"Failed to find UE context using idx %d\n",
							ueIdx);
		return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::IDENTITY_RESPONSE_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleAuthResponseMsg_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleAuthResponseMsg_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleAuthResponseMsg_v: "
							"Failed to find UE context using idx %d\n",
							ueIdx);
		return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::AUTH_RESP_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleSecurityModeResponse_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleSecurityModeResponse_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleSecurityModeResponse_v: "
							"Failed to find UE context using idx %d\n",
							ueIdx);
		return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::SEC_MODE_RESP_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleEsmInfoResponse_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleEsmInfoResponse_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleEsmInfoResponse_v: "
							"Failed to find UE context using idx %d\n",
							ueIdx);
		return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::ESM_INFO_RESP_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleInitCtxtResponse_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleInitCtxtResponse_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleInitCtxtResponse_v: "
							"Failed to find UE context using idx %d\n",
							ueIdx);
		return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::INIT_CTXT_RESP_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleAttachComplete_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleAttachComplete_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleAttachComplete_v: "
							"Failed to find UE context using idx %d\n",
							ueIdx);
		return;
	}

	// Fire attach-start event, insert cb to procedure queue
	SM::Event evt(Event_e::ATT_CMP_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleDetachRequest_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleDetachRequest_v\n");

	SM::ControlBlock* controlBlk_p = MmeCommonUtils::findControlBlock(
					const_cast<cmn::utils::MsgBuffer*>(msgData_p));
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleDetachRequest_v: "
				"Failed to find UE context using idx %d\n", ueIdx);
		return;
	}

	// Fire detach request event, insert cb to procedure queue
	SM::Event evt(Event_e::DETACH_REQ_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleS1ReleaseRequestMsg_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleS1ReleaseRequestMsg_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
    {
		log_msg(LOG_ERROR, ":handleS1ReleaseRequestMsg_v: "
                "Failed to find UE context using idx %d\n",
                 ueIdx);
        return;
    }

	// Fire s1 release event, insert cb to procedure queue
	SM::Event evt(Event_e::S1_REL_REQ_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleS1ReleaseComplete_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleS1ReleaseComplete_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
    {
		log_msg(LOG_ERROR, ":handleS1ReleaseComplete_v: "
                "Failed to find UE context using idx %d\n",
                 ueIdx);
        return;
    }

	// Fire s1 release complete event, insert cb to procedure queue
	SM::Event evt(Event_e::UE_CTXT_REL_COMP_FROM_ENB, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleDetachAcceptFromUE_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleDetachAcceptFromUE_v\n");

	SM::ControlBlock* controlBlk_p = SubsDataGroupManager::Instance()->findControlBlock(ueIdx);
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleDetachAcceptFromUE_v: "
				   "Failed to find UE Context using idx %d\n",
				   ueIdx);
		return;
	}

	//Fire NI_Detach Event, insert CB to procedure queue
	SM::Event evt(Event_e::DETACH_ACCEPT_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleServiceRequest_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleServiceRequest_v\n");

	SM::ControlBlock* controlBlk_p = MmeCommonUtils::findControlBlock(
					const_cast<cmn::utils::MsgBuffer*>(msgData_p));
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleServiceRequest_v: "
				   "Failed to find UE Context using idx %d\n",
				   ueIdx);
		return;
	}

	//Fire NI_Detach Event, insert CB to procedure queue
	SM::Event evt(Event_e::SERVICE_REQUEST_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}

void S1MsgHandler::handleTauRequestMsg_v(const cmn::utils::MsgBuffer* msgData_p, uint32_t ueIdx)
{
	log_msg(LOG_INFO, "S1 - handleTauRequestMsg_v\n");

	SM::ControlBlock* controlBlk_p = MmeCommonUtils::findControlBlock(
					const_cast<cmn::utils::MsgBuffer*>(msgData_p)); 
	if(controlBlk_p == NULL)
	{
		log_msg(LOG_ERROR, "handleTauRequestMsg_v: "
				   "Failed to find UE Context using idx %d\n",
				   ueIdx);
		return;
	}

	// Fire tau-start event, insert cb to procedure queue
	SM::Event evt(Event_e::TAU_REQUEST_FROM_UE, (void *)msgData_p);
	controlBlk_p->addEventToProcQ(evt);
}