myvtm/Module/mod_pinpad/PinPadFSM.h

#ifndef __PINPAD_FSM_H
#define __PINPAD_FSM_H

#pragma once

#include "SpFSM.h"
#include "toolkit.h"
#include "SpUtility.h"
#include "DevFSMCommBase.hpp"
#include "CommEntityUtil.hpp"
#include "PinPad_UserErrorCode.h"
#include "json/json.h"
enum EvtType
{
	USER_EVT_TEST =	EVT_USER+1,
	USER_EVT_ANY_INPUT_TIMER,
	USER_EVT_QUIT,
	USER_EVT_INIT,
	USER_EVT_INITFINISHED,
	USER_EVT_SETACCOUNT,
	USER_EVT_SETACCOUNTFINISHED,
	USER_EVT_INPUTCANCEL,
	USER_EVT_INPUTWAITINGMORE,
	USER_EVT_EXIT,
	USER_EVT_GETINPUT_SM,
	USER_EVT_GETINPUT_SM_FINISHED,
	USER_EVT_LOADKEY_SM,
	USER_EVT_LOADKEY_SM_FINISHED,
	USER_EVT_ENCRYPT_SM,
	USER_EVT_ENCRYPT_SM_FINISHED,
	USER_EVT_GET_CHECKCODE,
	USER_EVT_GET_CHECKCODE_FINISHED,
	USER_EVT_TODO_INIT,
	USER_EVT_TODO_INIT_FINISHED,
};
#include "PinPad_server_g.h"
#include "PinPad_msg_g.h"

#include "PinPadClass.h"
#include "DeviceBaseHelper.h"

//using namespace PinPad;
using PinPad::PinPadService_InputWaitMore_Info;
using PinPad::PinPadService_InputCancel_Info;
using PinPad::PinPadService_Exit_Info;
using PinPad::PinPadService_GetInputSM_Req;
using PinPad::PinPadService_GetInputSM_Ans;
using PinPad::PinPadService_LoadKeysSM_Req;
using PinPad::PinPadService_LoadKeysSM_Ans;
using PinPad::PinPadService_EncryptDataSM_Req;
using PinPad::PinPadService_EncryptDataSM_Ans;
using PinPad::PinPadService_QueryFunc_Req;
using PinPad::PinPadService_QueryFunc_Ans;
using PinPad::PinPadService_GetCheckCode_Req;
using PinPad::PinPadService_GetCheckCode_Ans;
using PinPad::PinPadService_GetDevInfo_Req;
using PinPad::PinPadService_GetDevInfo_Ans;
using PinPad::InputContent;

#include <map>
using namespace std;

#include "DeviceBaseHelper.h"

struct CtxInfo
{
	int dataSize;
	LPVOID pCtx;
};

class CPinPadEntity;
class CPinPadFSM;

unsigned int __stdcall DoWork(void* pData);

class PinPadInitFinishedEvent : public FSMEvent
{
public:
	PinPadInitFinishedEvent() : FSMEvent(USER_EVT_INITFINISHED){}
	~PinPadInitFinishedEvent(){}
};
class InputCancelEvent : public FSMEvent
{
public:
	InputCancelEvent() : FSMEvent(USER_EVT_INPUTCANCEL){}
	~InputCancelEvent(){}
};
class InputWaitingMoreEvent : public FSMEvent
{
public:
	InputWaitingMoreEvent() : FSMEvent(USER_EVT_INPUTWAITINGMORE){}
	~InputWaitingMoreEvent(){}
};

class GetInputSMEvent : public FSMEvent
{
public:
	GetInputSMEvent() : FSMEvent(USER_EVT_GETINPUT_SM){}
	~GetInputSMEvent(){}
	SpReqAnsContext<PinPadService_GetInputSM_Req, PinPadService_GetInputSM_Ans>::Pointer ctx;
	virtual void OnUnhandled() 
	{
		if (ctx != NULL)
			ctx->Answer(Error_InvalidState);
	}
};
class LoadKeySMEvent : public FSMEvent
{
public:
	LoadKeySMEvent() : FSMEvent(USER_EVT_LOADKEY_SM){}
	~LoadKeySMEvent(){}
	SpReqAnsContext<PinPadService_LoadKeysSM_Req, PinPadService_LoadKeysSM_Ans>::Pointer ctx;
	virtual void OnUnhandled() 
	{
		if (ctx != NULL)
			ctx->Answer(Error_InvalidState);
	}
};
class EncryptSMEvent : public FSMEvent
{
public:
	EncryptSMEvent() : FSMEvent(USER_EVT_ENCRYPT_SM){}
	~EncryptSMEvent(){}
	SpReqAnsContext<PinPadService_EncryptDataSM_Req, PinPadService_EncryptDataSM_Ans>::Pointer ctx;
	virtual void OnUnhandled() 
	{
		if (ctx != NULL)
			ctx->Answer(Error_InvalidState);
	}
};
class GetCheckCodeEvent : public FSMEvent
{
public:
	GetCheckCodeEvent() : FSMEvent(USER_EVT_GET_CHECKCODE){}
	~GetCheckCodeEvent(){}
	SpReqAnsContext<PinPadService_GetCheckCode_Req, PinPadService_GetCheckCode_Ans>::Pointer ctx;
	virtual void OnUnhandled()
	{
		if (ctx != NULL)
			ctx->Answer(Error_InvalidState);
	}
};

class CPinPadFSM : public CCommDevFSM<CPinPadFSM, PinPadClass>
{
public:
	enum {s0,s1,s2,s3,s4,s5,s6};
	BEGIN_FSM_STATE(CPinPadFSM)
		FSM_STATE_ENTRY(s0, "Init", s0_on_entry, s0_on_exit, s0_on_event)
		FSM_STATE_ENTRY(s1, "Initializing", s1_on_entry, s1_on_exit, s1_on_event)
		FSM_STATE_ENTRY(s2, "Idle", s2_on_entry, s2_on_exit, s2_on_event)
		FSM_STATE_ENTRY(s3, "Input", s3_on_entry, s3_on_exit, s3_on_event)
		FSM_STATE_ENTRY(s4, "AccessAuth", s4_on_entry, s4_on_exit, s4_on_event)
		FSM_STATE_ENTRY(s5, "Failed", s5_on_entry, s5_on_exit, s5_on_event)
		FSM_STATE_ENTRY(s6, "InitFailed", s6_on_entry, s6_on_exit, s6_on_event)
		END_FSM_STATE()

		BEGIN_FSM_RULE(CPinPadFSM, s0)
		FSM_RULE_ENTRY(s0, s1, USER_EVT_INIT, 0)
		FSM_RULE_ENTRY(s1, s2, USER_EVT_INITFINISHED, 0)
		FSM_RULE_ENTRY(s1, s6, USER_EVT_INITFINISHED, 1)
		FSM_RULE_ENTRY(s2, s3, USER_EVT_GETINPUT_SM, 0)
		FSM_RULE_ENTRY(s2, s4, USER_EVT_LOADKEY_SM, 2)
		FSM_RULE_ENTRY(s2, s4, USER_EVT_ENCRYPT_SM, 2)
		FSM_RULE_ENTRY(s2, s0, USER_EVT_TODO_INIT_FINISHED, 0)
		FSM_RULE_ENTRY(s3, s2, USER_EVT_GETINPUT_SM_FINISHED, 0)
		FSM_RULE_ENTRY(s3, s2, USER_EVT_EXIT, 0)
		FSM_RULE_ENTRY(s4, s2, USER_EVT_LOADKEY_SM_FINISHED, 0)
		FSM_RULE_ENTRY(s4, s2, USER_EVT_ENCRYPT_SM_FINISHED, 0)
		FSM_RULE_ENTRY(s5, s0, USER_EVT_TODO_INIT_FINISHED, 0)
		FSM_RULE_ENTRY(s6, s0, USER_EVT_TODO_INIT_FINISHED, 0)
		END_FSM_RULE()

		CPinPadFSM() : m_bFrontCancel(false),
		m_bWaitingMore(false), m_bExit(false), m_bPlainPin(true),
		m_bEntityExit(false), m_bPinInput(false), m_bLoadKey(false), m_bEncrypt(false)
		, m_dwDevCommFailCount(0), m_dwPinPadRunCount(0), m_eDevState(DEVICE_STATUS_NOT_READY), m_encryptkey(1)
		, m_bSM(false), m_b3DESLoaded(false), m_bSMLoaded(false), m_iInWhatPage(PageType_Other), m_szModel(""), m_szType("")
		, m_szVendor(""),m_csMachineType(true), m_csSite(true), m_terminalNo(true), m_port(true), m_Baudrate(true)
		, m_devSN(""){
			HARDWARE_ENTITY_RESET_ENTITYID(m_entCode, 0x206);
			ZeroMemory(&m_adapterInfo, sizeof(m_adapterInfo));
	}
	~CPinPadFSM(){
		m_bEntityExit = true;
	}
	virtual ErrorCodeEnum OnInit();
	virtual ErrorCodeEnum OnExit();

	//void DoWork();


	void s0_on_entry();
	void s0_on_exit();
	unsigned int s0_on_event(FSMEvent* event);
	void s1_on_entry();
	void s1_on_exit();
	unsigned int s1_on_event(FSMEvent* event);
	void s2_on_entry();
	void s2_on_exit();
	unsigned int s2_on_event(FSMEvent* event);
	void s3_on_entry();
	void s3_on_exit();
	unsigned int s3_on_event(FSMEvent* event);
	void s4_on_entry();
	void s4_on_exit();
	unsigned int s4_on_event(FSMEvent* event);
	void s5_on_entry();
	void s5_on_exit();
	unsigned int s5_on_event(FSMEvent* event);
	void s6_on_entry();
	void s6_on_exit();
	unsigned int s6_on_event(FSMEvent* event);

	int Initial();
	void push_char(char *buf, int *len, int c);
	void PushInputChar(char *buf, int c);
	void pop_char(char *buf, int *len,bool bClear=true);
	void clear_char(char *buf, int *len);
	bool Get12Account(char *szAcc);
	int GetInput(SpReqAnsContext<PinPadService_GetInputSM_Req, PinPadService_GetInputSM_Ans>::Pointer ctxSM);
	ErrorCodeEnum GetEncryptText(SpReqAnsContext<PinPadService_GetInputSM_Req, PinPadService_GetInputSM_Ans>::Pointer ctxSM);
	bool GetDevInitFlag(){ return m_bOpened;}
	void ResetDevInitFlag() { m_bOpened = false; }
	void SetExitFlag(){m_bExit = true;}
	bool GetEntityExit() { return m_bEntityExit;}
	void SetCtx(SpReqAnsContext<PinPadService_GetInputSM_Req, PinPadService_GetInputSM_Ans>::Pointer ctxSM)
	{
		m_inputSMCtx = ctxSM;
		DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("sm ctx,confirm,encrypt,len,timeout:%d,%d,%d,%d,smflag:%d", ctxSM->Req.confirm
			, ctxSM->Req.encrypt, ctxSM->Req.length, ctxSM->Req.timeout, ctxSM->Req.smflag);
		m_bPinInput = true;
	}
	bool GetCtx(SpReqAnsContext<PinPadService_GetInputSM_Req, PinPadService_GetInputSM_Ans>::Pointer& ctxSM)
	{
		if (m_inputSMCtx == NULL)
		{
			DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM)("ctxSM is null");
			return false;
		}
		ctxSM = m_inputSMCtx;
		return true;
	}
	template <class TReq, class TAns>
	void SaveCtx(int methodID, CSmartPointer<SpReqAnsContext<TReq, TAns> > ctx)
	{
		m_mapCtx[methodID].pCtx = static_cast<void*>(ctx.GetRawPointer());
	}
	LPVOID  GetCtx(int methodID)
	{
		if (m_mapCtx.find(methodID) != m_mapCtx.end())
			return m_mapCtx[methodID].pCtx;
		else
			return NULL;
	}
	bool GetDevInfo(DevCategoryInfo &devInfo);
	int LoadKeySM(SpReqAnsContext<PinPadService_LoadKeysSM_Req, PinPadService_LoadKeysSM_Ans>::Pointer ctx);
	int EncryptSM(SpReqAnsContext<PinPadService_EncryptDataSM_Req, PinPadService_EncryptDataSM_Ans>::Pointer ctx);
	ErrorCodeEnum GetCheckCode(SpReqAnsContext<PinPadService_GetCheckCode_Req, PinPadService_GetCheckCode_Ans>::Pointer ctx);
	int GetEncryptKey(int &devFunc,int &loadFunc)
	{
		devFunc = loadFunc = 0;
		devFunc = m_encryptkey;
		int tmp = 0;
		if (m_b3DESLoaded)
			tmp = 1;
		if (m_bSMLoaded)
			tmp += 2;
		
		if (m_bSMLoaded)
			loadFunc = 2;
		else
			loadFunc = tmp;
		//20161216 oilyang add 为了兼容密钥更新有问题的设备
		if (loadFunc == 0)
			loadFunc = 1;
		return 0;
	}

	void SetInWhatPage(int iPageType);
	void SelfTest(EntityTestEnum eTestType,CSmartPointer<ITransactionContext> pTransactionContext);
	void ToLogWarnInfoAboutTermCustom();
	int ProcFWBReConn();
public:
	SP::Toolkit::CConditionVarPlus m_hInputConVar;
	SP::Toolkit::CConditionVarPlus m_hInitConVar;
private:

	SpReqAnsContext<PinPadService_GetInputSM_Req, PinPadService_GetInputSM_Ans>::Pointer m_inputSMCtx;
	char m_szAccount[MAX_ACCOUNT_LEN];
	bool m_bFrontCancel,m_bWaitingMore,m_bExit,m_bPlainPin,m_bEntityExit
		,m_bPinInput,m_bLoadKey,m_bEncrypt,m_bSM,m_b3DESLoaded,m_bSMLoaded;
	ULLINT m_ullBeginTime, m_ullEndTime;
	CSimpleStringA m_deviceNo, m_port, m_Baudrate, m_devCheckData,m_keySN,m_keySNSM,m_szModel,m_szType,m_szVendor,m_devSN/*fwb SN*/;
	char m_buf[4];
	ErrorCodeEnum m_testResult;
	DWORD m_dwDevCommFailCount,m_dwPinPadRunCount;
	DevStateEnum m_eDevState;
	int m_encryptkey;
	int m_iInWhatPage/*在哪个页面，区分首页，用户桌面，其他页*/;
	CSimpleStringA m_csMachineType, m_csSite, m_terminalNo,m_lastCheckCode;

	CSimpleStringA m_csAlarmMsg;
	map<int, CtxInfo> m_mapCtx;
	bool IsInBusiness() { return (m_iInWhatPage == PageType_Other); }
};

struct InitTask : public ITaskSp
{
	CPinPadFSM *fsm;
	
	InitTask(CPinPadFSM* f) : fsm(f) {}

	void Process()
	{
		FSMEvent *e = new FSMEvent(USER_EVT_INITFINISHED);
		e->param1 = fsm->Initial();
		fsm->PostEventFIFO(e);
	}
};

struct LoadKeySMTask : public ITaskSp
{
	CPinPadFSM* fsm;
	SpReqAnsContext<PinPadService_LoadKeysSM_Req, PinPadService_LoadKeysSM_Ans>::Pointer ctx;
	LoadKeySMTask(CPinPadFSM* f) : fsm(f) {}
	void Process()
	{
		LOG_FUNCTION();
		FSMEvent *e = new FSMEvent(USER_EVT_LOADKEY_SM_FINISHED);
		e->param1 = fsm->LoadKeySM(ctx);
		fsm->PostEventFIFO(e);
	}
};
struct EncryptSMTask : public ITaskSp
{
	CPinPadFSM* fsm;
	SpReqAnsContext<PinPadService_EncryptDataSM_Req, PinPadService_EncryptDataSM_Ans>::Pointer ctx;
	EncryptSMTask(CPinPadFSM* f) : fsm(f) {}
	void Process()
	{
		LOG_FUNCTION();
		FSMEvent *e = new FSMEvent(USER_EVT_ENCRYPT_SM_FINISHED);
		e->param1 = fsm->EncryptSM(ctx);
		fsm->PostEventFIFO(e);
	}
};
struct GetCheckCodeTask : public ITaskSp
{
	CPinPadFSM* fsm;
	SpReqAnsContext<PinPadService_GetCheckCode_Req, PinPadService_GetCheckCode_Ans>::Pointer ctx;
	GetCheckCodeTask(CPinPadFSM* f) : fsm(f) {}
	void Process()
	{
		FSMEvent *e = new FSMEvent(USER_EVT_GET_CHECKCODE_FINISHED);
		e->param1 = fsm->GetCheckCode(ctx);
		fsm->PostEventFIFO(e);
	}
};
struct ProcFWBReConnTask : public ITaskSp
{
	CPinPadFSM* fsm;
	ProcFWBReConnTask(CPinPadFSM* f) : fsm(f) {}
	void Process()
	{
		FSMEvent* e = new FSMEvent(USER_EVT_TODO_INIT_FINISHED);
		e->param1 = fsm->ProcFWBReConn();
		fsm->PostEventFIFO(e);
	}
};
template<class T>
class TimerOutHelper : public ITimerListener
{
public:
	typedef void (T::*FuncTimer)(void *pUserdata);

	TimerOutHelper(T *p, FuncTimer pTimerFunc, void *pUserData, bool bDeleteSelf = false) 
		: m_pObject(p), m_pUserData(pUserData), m_pTimer(pTimerFunc), m_bDeleteSelf(bDeleteSelf) {}

	virtual void OnTimeout(DWORD dwTimerID)
	{
		(m_pObject->*m_pTimer)(m_pUserData);
		if (m_bDeleteSelf)
			delete this;
	}
private:
	void *m_pUserData;
	T *m_pObject;
	FuncTimer m_pTimer;
	bool m_bDeleteSelf;
};
#endif //__PINPAD_FSM_H