#ifndef _CRYPTBASE_H #define _CRYPTBASE_H #pragma once //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // #if defined( _MOLA_LIB ) || defined( _SERVER ) # include "RandomSNKey.h" #else # include "RandomSNKey_4Client.h" #endif // //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ class ISecureWorkNode; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // // class EXPORTS_API CryptManager { public: typedef DWORD eCRYPT_T; public: VOID Release(); BOOL Encrypt( BYTE* IN OUT pSTREAM, INT iSize, BYTE& OUT rCRC ); BOOL Decrypt( BYTE* IN OUT pSTREAM, INT iSize, BYTE IN rCRC ); BOOL StartLogic( BOOL bSET ); // : SERVER/CLIENT // static BOOL Init1st(); static BOOL InitInfo( PVOID arg ); static BOOL InitSecureInfoParser(); VOID Update( DWORD dwCurTick ); BOOL IsCanDelete(); VOID DecRefCount(); // : CLIENT // 다음은 클라이언트에서만 사용할 것들... static DWORD GetHeartbeatSize(); static ISecureWorkNode* GetISecureWorkNode(); static DWORD GetMolaProtocolCRC(); static DWORD GetUserKeyInCryptInfo( BYTE* IN OUT pCryptINFO ); static CryptManager* SetCryptAlgorithmInfo( BYTE byLocalSNKey, BYTE* IN pCryptINFO, BYTE byEncodedIDX ); ////////////////////////////////////////////////////////////////////////// // #pragma pack(push, 4) // 24BYTEs private: BOOL m_Inited; PVOID m_pCryptMod; DWORD m_RefCount; BYTE m_byCRC; DWORD m_dwSolarCode; DWORD m_dwLunarCode; #pragma pack(pop) static BYTE ms_DecryptStream[256]; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // public: //ISecureWorkNode* ISecureWorkNodePtr(); // 각 CryptManager에 걸린 VOID SetCryptKey( PVOID pEncKey, PVOID pDecKey ); PVOID GetEncryptionKey(); PVOID GetDecryptionKey(); CHAR GetCRCKey(); // 암호화 정보, 키교환을 위한 스트림 생성, // cryptInfo는 MSG_CRYPTOGRAPHY_CMD::m_byCryptBuffer로 그 크기는 256이다. // bySymmetrySNKey는 보내고자 하는 SNKey과 쌍을 이루는 키를 입력한다. // // byPairSNKey%(MAX_KEY_SIZE) BYTE GetCryptAlgorithmInfo( BYTE byPairSNKey, BYTE* pCryptINFO, DWORD dwUserID ); // size 256 static BOOL SetGarbageBlock( BYTE* IN OUT pKeyBlock, DWORD dwGabageKey ); // : SERVER static DWORD GetSecureFaultAcceptSize(); public: VOID Initialize( BOOL bIsServer, eCRYPT_T eType = eCRYPT_NONE ); private: VOID _Initialize( BOOL bIsServer, eCRYPT_T eType ); ////////////////////////////////////////////////////////////////////////// // public: CryptManager(); ~CryptManager(); ////////////////////////////////////////////////////////////////////////// public: static const eCRYPT_T eCRYPT_HSEL = 0x00000000; static const eCRYPT_T eCRYPT_JSC = 0x00000001; static const eCRYPT_T eCRYPT_MASK = 0x00000007; static const eCRYPT_T eCRYPT_NONE = static_cast(-1); static const eCRYPT_T eCRYPT_T_COUNT = 8; public: static const DWORD VERSION_CODE = MAKE_MOLA_VERSION_(1,0,10); static const BOOL bClientSession = FALSE; static const BOOL bServerSession = TRUE; static const DWORD dwUSE_ONE = 0x1; static const DWORD dwUSE_TWO = 0x2; static const DWORD dwUSE_MASK = dwUSE_ONE | dwUSE_TWO; }; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // GameClient / CProtocolCryptography 에서 사용할 정보 class CryptManagerDeletator { public: CryptManagerDeletator() : m_POINTER(NULL) {} ~CryptManagerDeletator() { m_POINTER = NULL; } CryptManagerDeletator( const CryptManagerDeletator& rCryptManager ) : m_POINTER(rCryptManager.m_POINTER) {} CryptManagerDeletator( CryptManager* pManager ) : m_POINTER(pManager) {} VOID Release() { if( !m_POINTER ) return; if( m_POINTER->IsCanDelete() ) delete m_POINTER; else m_POINTER->DecRefCount(); m_POINTER = NULL; } inline BOOL operator == ( PVOID ptr ) const { return m_POINTER == ptr; } inline BOOL operator != ( PVOID ptr ) const { return !(m_POINTER == ptr); } inline CryptManagerDeletator* operator->() { return this; } inline BOOL Encrypt( BYTE* IN OUT pbySTREAM, int iSize, BYTE& OUT byCRC ) { if( m_POINTER ) return m_POINTER->Encrypt( pbySTREAM, iSize, byCRC ); return FALSE; } inline BOOL Decrypt( BYTE* IN OUT pbySTREAM, int iSize, BYTE IN byCRC ) { if( m_POINTER ) return m_POINTER->Decrypt( pbySTREAM, iSize, byCRC ); return FALSE; } inline BOOL StartLogic( BOOL bset ) { if( m_POINTER ) return m_POINTER->StartLogic( bset ); return FALSE; } ////////////////////////////////////////////////////////////////////////// private: // CryptManager* m_POINTER; }; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // Client CProtocolCryptography struct sCLIENT_CRYPT_INFO { BOOL m_bUseCryptography; ClientSNKey m_SNKey; CryptManagerDeletator m_pCryptManager; DWORD m_dwKey; }; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // #pragma pack(push, 1) // 패킷 버퍼 컨트롤 struct SPacketBuffer { #define PACKET_ALIGN32BYTE(size) (((size) + 0x1F) & ~0x1F) WORD __Dummy; union { WORD wSIZE; WORD NETWORK_PACKET_PTR[1]; }; BYTE PACKET_PTR[/*Network::MAX_CRYPTOGRAM_STREAM_SIZE*/ 16000]; inline VOID CopyFrom( const BYTE* pBuffer, const WORD wSize ) { MSG_CG_CRYPTOGRAPHY_PACKET_ENCODED* pEncodedMsg = (MSG_CG_CRYPTOGRAPHY_PACKET_ENCODED*)pBuffer; wSIZE = wSize - (WORD)pEncodedMsg->GetHeaderSize(); WORD wFSize = (WORD)PACKET_ALIGN32BYTE(wSIZE); CopyMemory( PACKET_PTR, pEncodedMsg->GetPacketStream(), wSIZE ); ZeroMemory( PACKET_PTR+wSIZE, wFSize-wSIZE ); } inline static SPacketBuffer* GetPacketBuffer( BYTE* pDecodedMsg ) { return (SPacketBuffer*)( pDecodedMsg-(sizeof(WORD)<<1) ); } }; #pragma pack(pop) //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // Client 초기화 루틴 class WzGResTable; struct eMOLASIG_MSG; struct EXPORTS_API sCLIENT_INIT_PRE_ENTRY { sCLIENT_INIT_PRE_ENTRY( WzGResTable* pGResTable ); }; struct eSECURE_QUERY { enum IDX_T { NON_OPERATION , SIMPLEX_PATTERN_SNK , QUERY_MOLA , CONFIGURE_MOLA , DUPLEX_PATTERN_SNK , RESOURCE_CHECK , BUILTIN_QUERY , MAX_SECURE_PACKET , // FIND_INFO_BY_WORK_KEY , DEL_INFO_BY_WORK_KEY , POP_INFO_BY_WORK_KEY , DEL_USER , }; }; struct MOLA_QUERY_MSG { DWORD m_dwUserKey; eSECURE_QUERY::IDX_T m_eSQT; union { DWORD m_dwWorkKey; // Normal WorkKey; BOOL m_bIsRandom; }; DWORD m_dwWorkKeyB; // Builtin WorkKey; ITimeout m_Timeout; BYTE* m_pBackupMsg; WORD m_wpBackupMsgSZ; MOLA_QUERY_MSG() : m_dwUserKey(0), m_eSQT(eSECURE_QUERY::NON_OPERATION), m_dwWorkKey(0), m_dwWorkKeyB(0), m_pBackupMsg(NULL), m_wpBackupMsgSZ(0) {} MOLA_QUERY_MSG( DWORD dwUserKey, eSECURE_QUERY::IDX_T eSQT, DWORD dwWorkKey = 0, DWORD dwWorkKeyB = 0, BYTE* pBackupMsg = NULL, WORD wpBackupMsgSZ = 0 ) : m_dwUserKey(dwUserKey), m_eSQT(eSQT), m_dwWorkKey(dwWorkKey), m_dwWorkKeyB(dwWorkKeyB), m_pBackupMsg(pBackupMsg), m_wpBackupMsgSZ(wpBackupMsgSZ) {} }; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // class EXPORTS_API ISecureWorkNode { ////////////////////////////////////////////////////////////////////////// // protected: DWORD m_dwRecvRoundKey; public: ISecureWorkNode() {} ~ISecureWorkNode() {} public: // [for CLIENT] Client::Application::Heartbeat에서 호출 (TRUE, PTR, SZ) 모두 참일 경우 패킷 전송 virtual BOOL DoEarlyProcess( BYTE*& rpSendingMsg, WORD& rwSendingSize ); // [for CLIENT] Client::Application::Heartbeat에서 호출 (TRUE, PTR, SZ) 모두 참일 경우 패킷 전송 virtual BOOL DoLatelyProcess( BYTE*& rpSendingMsg, WORD& rwSendingSize ); public: // [for CLIENT] Application::Heartbeat에서 호출 static VOID Update( DWORD dwDeltaTick ); public: // [for CLIENT] Scene 전환시 struct EXPORTS_API OnChange { // [for CLIENT] BattleScene일 경우 경우 호출 - Init 계열에서 한번 static VOID Scene( eZONETYPE ztZONE ); }; // [for CLIENT] 패킷 수신시 // pPacketMsg : 수신된 패킷 포인터 (이것은 SPacketBuffer를 사용한 BUFFER여야 한다.) // wSIZE : 클라는 wSIZE검사를 하지 않으므로 0 입력) struct EXPORTS_API OnMsgRecvSignal { // [for CLIENT] CG_SECURE_QUERY_BUILTIN_CHK_CMD static VOID BuiltinCodeCheck( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_QUERY_RESOURCE_CHK_T1_CMD static VOID ResourceCheckT1( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_QUERY_RESOURCE_CHK_T2_CMD static VOID ResourceCheckT2( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_QUERY_RESOURCE_CHK_T3_CMD static VOID ResourceCheckT3( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_QUERY_RESOURCE_CHK_T4_CMD static VOID ResourceCheckT4( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN01_SIMPLEX_SNK_CMD static VOID SimplexSNK_T1( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN02_SIMPLEX_SNK_CMD static VOID SimplexSNK_T2( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN03_SIMPLEX_SNK_CMD static VOID SimplexSNK_T3( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN04_SIMPLEX_SNK_CMD static VOID SimplexSNK_T4( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN08_DUPLEX_SNK_KEY static VOID DuplexSNK_T1( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN05_QUERY_MOLA_CMD static VOID MOLA_Query( PVOID pPacketMsg, WORD wSIZE ); // [for CLIENT] CG_SECURE_POLICY_PATTERN06_CONFIGURE_MOLA_CMD // [for CLIENT] CG_SECURE_POLICY_PATTERN07_CONFIGURE_HEARTBEAT_CMD static VOID MOLA_Configure( PVOID pPacketMsg, WORD wSIZE ); }; struct EXPORTS_API OnMsgSend { // [for CLIENT] CG_SECURE_QUERY_BUILTIN_CHK_CMD static BOOL BuiltinCodeCheck( BYTE*& prSendingMsg, WORD& rwSIZE ); }; struct EXPORTS_API OnGResInit { // [for CLIENT] 몇몇 CS공통 스크립트 관련 비교소스 데이터 유지 목적 static VOID RegisterData( eSCRIPT_CODE sccSCRIPT, DWORD dwKEY, PVOID pDATA ); }; //━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫ // static VOID OnMsg( eMOLASIG_MSG* pSIGMSG ); // 0 : FAIL, 1 : SUCCESS(NORMAL), 2 : SUCCESS(BUILTIN) DWORD OnRecvPacket( MOLA_QUERY_MSG* IN OUT pARG ); DWORD QueryProcess( MOLA_QUERY_MSG* IN OUT pARG ); // OLD VERSION PVOID MakePacketForClient( const BYTE byCate, const BYTE byProto, BYTE*& pSendingMsg, WORD& wSendingSize ); }; #endif //_CRYPTBASE_H