#include "Stdafx.h"
#include "AuthManager.h"
#include "DNPacket.h"
#include "Util.h"
#include "DNConfig.h"

#if defined(_LOGINSERVER)
extern TLoginConfig g_Config;
#elif defined(_VILLAGESERVER)
extern TVillageConfig g_Config;
#elif defined(_GAMESERVER)
extern TGameConfig g_Config;
#endif	// _GAMESERVER

CAuthManager::CAuthManager() : m_bResetAuthServer(false), m_ServerID(0), m_ResetAuthServerTick(0)
{
}

bool CAuthManager::Init()
{
	m_ServerID = CalculateServerID();
	DN_ASSERT(0 != m_ServerID,	"Check!");

#if !defined(_FINAL_BUILD)
	g_Config.nManagedID = m_ServerID;
#endif	// #if !defined(_FINAL_BUILD)

	return true;
}

void CAuthManager::Run(DWORD CurTick)
{
	if (!IsResetAuthServer() &&
		(RESETAUTHSERVERTERM < GetTickTerm(m_ResetAuthServerTick, CurTick)))
	{
		m_ResetAuthServerTick = CurTick;

		QueryResetAuthServer();
	}
}

INT64 CAuthManager::GetCertifyingKey()
{
	INT64 biCertifyingKey = 0;

	for (int iIndex = 0 ; 4 > iIndex ; ++iIndex) {
		WORD wRand = rand();
		wRand = (wRand)?(wRand):(1);
		biCertifyingKey |= static_cast<INT64>(wRand) << (16 * iIndex);
	}

	return(biCertifyingKey);
}

const int CAuthManager::CalculateServerID()
{
	int iServerID = 0;

	if (g_Config.bUseCmd) {
		DN_ASSERT(0 != g_Config.nManagedID,	"Invalid!");

		iServerID = g_Config.nManagedID;
	}
	else {
		DN_ASSERT(0 == g_Config.nManagedID,	"Invalid!");

		in_addr aInAddr;
		::memset(&aInAddr, 0, sizeof(aInAddr));
		GetLocalIp(&aInAddr);
		WORD aLowID = ::htons(static_cast<WORD>(aInAddr.s_imp));	// P.S.> 사용하는 IP 의 최하위 바이트가 0 이 되는 경우는 없음
		DN_ASSERT(0 != aLowID,	"Check!");

#if defined(_LOGINSERVER)
		iServerID = MAKELONG(aLowID, MAKEWORD((g_Config.nClientAcceptPort), SERVERTYPE_LOGIN+1));
#elif defined(_VILLAGESERVER)
		iServerID = MAKELONG(aLowID, MAKEWORD((g_Config.nVillageID), SERVERTYPE_VILLAGE+1));
#elif defined(_GAMESERVER)
		iServerID = MAKELONG(aLowID, MAKEWORD(0, SERVERTYPE_GAME+1));
#else
		iServerID = 0;
#endif
	}
	DN_ASSERT(0 != iServerID,	"Check!");

	return iServerID;
}

char* CAuthManager::GetLocalIp(OUT in_addr* pInAddr)
{
	WSADATA WsaData;
	int iResult = WSAStartup(MAKEWORD(2, 2), &WsaData);
	if (0 > iResult) {
		DN_RETURN(NULL);
	}

	char Name[256] = { 0, };

	gethostname(Name, sizeof(Name));
	PHOSTENT host = gethostbyname(Name);
	if (host) {
		if (pInAddr) {
			memcpy(pInAddr, host->h_addr_list[0], sizeof(in_addr));
		}
		return inet_ntoa(*(struct in_addr*)*host->h_addr_list);
	}

	WSACleanup();

	return NULL;
}