myvtm/Module/mod_Initializer/mod_Initializer.h

#pragma once
#include <string>
using namespace std;

#include "RVCCrypt.h"
#include "InitializerConn.h"
#include "InitializerFSM.h"

#include "Initializer_server_g.h"
using namespace Initializer;

// 初始设定控制台 Initializer	0x509
class CInitializerEntity;

class CInitializerSession : public InitializerService_ServerSessionBase
{
public:
	CInitializerSession(CInitializerEntity *pEntity) : m_pEntity(pEntity) {}
	virtual ~CInitializerSession() {}
		
	virtual void Handle_Initialize(SpOnewayCallContext<InitializerService_Initialize_Info>::Pointer ctx);
	virtual void Handle_InitializeNew(SpReqAnsContext<InitializerService_InitializeNew_Req, InitializerService_InitializeNew_Ans>::Pointer ctx);
	virtual void Handle_GetAuthServerInfo(SpReqAnsContext<InitializerService_GetAuthServerInfo_Req, InitializerService_GetAuthServerInfo_Ans>::Pointer ctx);
	virtual void Handle_InitializeForBlueTooth(SpReqAnsContext<InitializerService_InitializeForBlueTooth_Req, InitializerService_InitializeForBlueTooth_Ans>::Pointer ctx);
private:
	CInitializerEntity *m_pEntity;
};

class CInitializerEntity : public CEntityBase, public ITimerListener, public ILogListener
{
public:
	CInitializerEntity()
		:m_pConnection(NULL), m_hGUIWindow(0), m_hGUIReadyEvent(0), m_hGUIClosedEvent(0)
	{
		memset(m_cTDESPukey, 0, 2048);
		memset(m_cSM2Pukey, 0, 2048);
		m_iTDESPukeyLen = 0;
		m_iSM2Pukeylen  = 0;

		memset(m_cTDESTMK, 0, 64);
		memset(m_cSM2TMK, 0, 64);

		m_eErrNum = Error_Succeed;
		m_afterSMIsFirst = false;
		m_autoTryInit = 0;

		m_bGetKMCKey = false; //false代表获取kmc密钥失败
	}
	
	virtual ~CInitializerEntity() 
	{
		if (m_hGUIReadyEvent != 0)
		{
			CloseHandle(m_hGUIReadyEvent);
			m_hGUIReadyEvent = 0;
		}

		if (m_hGUIClosedEvent != 0)
		{
			CloseHandle(m_hGUIClosedEvent);
			m_hGUIClosedEvent = 0;
		}

		//此处会引起初始化成功后，退出框架是报错
		/*if (NULL != m_pkeys)
		{
		delete m_pkeys;
		m_pkeys = NULL;
		}*/
	}

	virtual const char *GetEntityName() const { return "Initializer"; }
	virtual bool IsService()const{return true;}

	virtual void OnStarted();
	virtual void OnPreClose(EntityCloseCauseEnum eCloseCause,CSmartPointer<ITransactionContext> pTransactionContext);
	
	virtual CServerSessionBase *OnNewSession(const char* /*pszRemoteEntityName*/, const char * /*pszClass*/)
	{
		//LOG_FUNCTION();
		return new CInitializerSession(this);
	}
	virtual void OnLog(const CAutoArray<CUUID> &SubIDs, const CUUID nLogID,const LogTypeEnum eLogType, const SeverityLevelEnum eLevel,
		const DWORD dwSysError,const DWORD dwUserCode,const DWORD dwEntityInstanceID, const WORD wEntityDevelID, 
		const CAutoArray<DWORD> &Param, const char *pszEntityName, const char *pszModuleName,const char *pszMessage);
	// 初始化
	ErrorCodeEnum StartInitialize(const char *pszUserID, const char *pszPassword);
	ErrorCodeEnum StartInitializeNew(const char *pszAuthServer, int nAuthPort, const char *pszUserID, const char *pszPassword);

	// KMC初始化
	ErrorCodeEnum InitKMC(int nPinPadCapability);
		
	// 获取KMC错误
	CSimpleStringA GetKMCLastErrMsg();

	// 释放KMC
	ErrorCodeEnum ReleaseKMC();

	// 生成MK请求包
	ErrorCodeEnum CreateInitMKData(CBlob &blob, bool bSM);

	// 解析返回主密钥
	ErrorCodeEnum ParseInitMKResult(CBlob &blob, bool bSM);

	// 更新密码键盘主密钥
	DWORD LoadKeysToPinPad(bool bSM);

	//ErrorCodeEnum PrintAllKeys(bool bSM);
		
	bool HexStrToByteArray(const char* pHex, BYTE *pBuf, int *pBufLen);
	string ByteArrayToHexStr(BYTE *pBuf, int nBufLen);

	virtual void OnTimeout(DWORD dwTimerID);
	bool GetAuthAccessAddr(CSimpleStringA &strAddr, int &nPort);
	bool SetAuthAccessAddr(const char *pszAddr, int nPort);
	bool GetCenterSettingAddr(CSimpleStringA &strAddr, int &nPort);
	bool SetCenterSettingAddr(const char *pszAddr, int nPort);
	bool GetInitUkeyFlg(int &nUkeyFlg);
	bool GetInitializedFlg(int &nInitializedFlg);
	bool SetInitializedFlg(CSimpleStringA strInitState);
	bool IsIL();//是否是简化版	
	
	ErrorCodeEnum SecureClientConnect();
	ErrorCodeEnum SecureClientRelease();
	void OnSecureClientDisconnect();
	
	void BeginInitMK();
	void EndInitMK(DWORD rc, const char *pszErrMsg);
	void BeginLogin(const char *pszAuthServer, int nAuthPort, const char *pszUserID, const char *pszPassword);
	void AutoBeginLogin(const char *pszAuthServer, int nAuthPort);
	void EndLogin(DWORD rc, const char *pErrMsg);
	void BeginSyncPK();
	void BeginGetKMCKey();
	void EndGetKMCKey(DWORD rc, const char* pszErrMsg);
	void EndSyncPK(DWORD rc, const char *pszErrMsg);
	DWORD ReportInitMKState(const char *pszUserID, DWORD dwErrCode,  const char *pszErrMsg);
	ErrorCodeEnum DownloadCfg(const char *pszSettingServer, int nSettingPort);
	DWORD InitCardSwiper();
	const char* GetLastErrMsg();

	ErrorCodeEnum StartGUI();
	ErrorCodeEnum StartGUINew();
	ErrorCodeEnum CloseGUI();
	void CloseTask();

	bool HasPinPad();
	int GetPinPadCapability();
	bool GetTerminalFingerPrint(BYTE *pBuf, int &nBufLen);
	bool GetTerminalPublicKey(BYTE *pBuf, int &nBufLen);
	bool GetMD5Hash(const char *pStr, BYTE md5[16]);

    // 获取密码键盘ID和外设ID
	// 返回1：只有PinPadID；2：只有DeviceID；3：两者都有；0：失败
	int GetPinPadIDAndDeviceID(CSimpleStringA &strPinPadID, CSimpleStringA &strDeviceID);
	
	bool HasCkCodeFlg();

	int ConvertStr2Byte(string input, BYTE* output, int outputLen);

	//判断是否是国密改造之后的第一次初始化
	int GetOrSetIsFirstSM(int type);

	BYTE m_cTDESPukey[2048];
	BYTE m_cSM2Pukey[2048];
	char m_cTDESTMK[64];
	char m_cSM2TMK[64];
	int  m_iTDESPukeyLen;
	int  m_iSM2Pukeylen;
	CSimpleStringA m_strAuthorizer;
	CSimpleStringA m_strAuthTime;
	CSimpleStringA m_strLastErrMsg;
	DWORD m_eErrNum;
	SpReqAnsContext<InitializerService_InitializeNew_Req, InitializerService_InitializeNew_Ans>::Pointer m_ctx;
	SpReqAnsContext<InitializerService_InitializeForBlueTooth_Req, InitializerService_InitializeForBlueTooth_Ans>::Pointer m_ctx_blue;
	//type=0 :normal,type=1 :bluetooth
	int m_type;
	bool m_bHasCkCode;

	CSimpleStringA m_strUserID;
	CSimpleStringA m_strPassword;
	//判断是否为国密改造之后的第一次准入
	bool GetAccessauthIsFirst();
	//国密改造之后第一次准入的标志位
	bool m_afterSMIsFirst;
	//自动初始化的重试次数
	int m_autoTryInit;

	string m_TMK;
	string m_TPK;
	string m_EDK;
	string m_index;
	bool m_bGetKMCKey;
private:
	bool IsServiceAvailable(const char *pszServiceAddr, int nServerPort);
	ErrorCodeEnum ParseIPAddress(const char *str, CSimpleStringA &ip, int &port);

	int GUIThreadEntry();
	static unsigned int __stdcall __GUIThreadEntry(void *arg);

private:
	EncKeys* m_pkeys;
	CSimpleStringA m_strAuthServer;
	int m_nAuthPort;
	
	CInitializerFSM m_FSM;
	HWND m_hGUIWindow;
	HANDLE m_hGUIReadyEvent;
	HANDLE m_hGUIClosedEvent;

	
	CInitializerConn *m_pConnection;
	friend class CMKInitDlg;
	CSimpleStringA m_strInitUserID;

};