#include <Windows.h>
#include <Wbemidl.h>
#include <comdef.h>
#include <WinCrypt.h>
#include <string>
#include <algorithm>
#include <sstream>
#include <vector>
#include <commctrl.h>
#include <iomanip> // 添加缺失的头文件
#pragma comment(lib, "wbemuuid.lib")
#pragma comment(lib, "Advapi32.lib")
#pragma comment(lib, "Crypt32.lib")
#pragma comment(lib, "comctl32.lib")
// 资源ID定义
#define IDD_MAIN_DIALOG 101 // 添加缺失的定义
#define IDC_MACHINE_CODE 1001
#define IDC_DAYS_COMBO 1002
#define IDC_REG_CODE 1003
#define IDC_GENERATE_BTN 1004
#define IDC_REGISTER_BTN 1005
#define IDC_STATUS_TEXT 1006
#define IDC_VALIDATE_BTN 1007
// 全局变量
HWND g_hMainDlg = NULL;
char g_szMachineCode[256] = { 0 };
// 函数声明
const char* GetMachineCode();
const char* ComputeRegistration(const char* machineCode, int daysChoice);
BOOL SaveRegistration(const char* machineCode, const char* regCode, int daysChoice);
int ValidateLicense();
void UpdateStatus(const char* message);
void InitializeControls(HWND hDlg);
void GenerateRegistrationCode(HWND hDlg);
void RegisterProduct(HWND hDlg);
void ValidateLicenseOnStartup();
// 获取机器码函数
const char* GetMachineCode() {
if (g_szMachineCode[0] != '\0') {
return g_szMachineCode;
}
HRESULT hres;
std::string serials;
// 初始化COM
hres = CoInitializeEx(0, COINIT_MULTITHREADED);
if (FAILED(hres)) {
return "COM Init Failed";
}
// 设置COM安全级别
hres = CoInitializeSecurity(
NULL,
-1,
NULL,
NULL,
RPC_C_AUTHN_LEVEL_DEFAULT,
RPC_C_IMP_LEVEL_IMPERSONATE,
NULL,
EOAC_NONE,
NULL
);
if (FAILED(hres)) {
CoUninitialize();
return "CoInitializeSecurity Failed";
}
// 创建WMI实例
IWbemLocator* pLoc = NULL;
hres = CoCreateInstance(
CLSID_WbemLocator,
0,
CLSCTX_INPROC_SERVER,
IID_IWbemLocator,
(LPVOID*)&pLoc
);
if (FAILED(hres)) {
CoUninitialize();
return "WbemLocator Failed";
}
// 连接到WMI
IWbemServices* pSvc = NULL;
hres = pLoc->ConnectServer(
_bstr_t(L"ROOT\\CIMV2"),
NULL,
NULL,
0,
NULL,
0,
0,
&pSvc
);
if (FAILED(hres)) {
pLoc->Release();
CoUninitialize();
return "ConnectServer Failed";
}
// 设置代理安全级别
hres = CoSetProxyBlanket(
pSvc,
RPC_C_AUTHN_WINNT,
RPC_C_AUTHZ_NONE,
NULL,
RPC_C_AUTHN_LEVEL_CALL,
RPC_C_IMP_LEVEL_IMPERSONATE,
NULL,
EOAC_NONE
);
if (FAILED(hres)) {
pSvc->Release();
pLoc->Release();
CoUninitialize();
return "SetProxyBlanket Failed";
}
// 查询CPU信息
IEnumWbemClassObject* pEnumerator = NULL;
hres = pSvc->ExecQuery(
bstr_t("WQL"),
bstr_t("SELECT * FROM Win32_Processor"),
WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
NULL,
&pEnumerator
);
if (SUCCEEDED(hres)) {
IWbemClassObject* pclsObj = NULL;
ULONG uReturn = 0;
while (pEnumerator) {
HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1, &pclsObj, &uReturn);
if (0 == uReturn) break;
VARIANT vtProp;
hr = pclsObj->Get(L"ProcessorId", 0, &vtProp, 0, 0);
if (SUCCEEDED(hr)) {
serials += _bstr_t(vtProp.bstrVal);
VariantClear(&vtProp);
}
pclsObj->Release();
}
pEnumerator->Release();
}
// 查询磁盘信息
hres = pSvc->ExecQuery(
bstr_t("WQL"),
bstr_t("SELECT * FROM Win32_DiskDrive"),
WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
NULL,
&pEnumerator
);
if (SUCCEEDED(hres)) {
IWbemClassObject* pclsObj = NULL;
ULONG uReturn = 0;
while (pEnumerator) {
HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1, &pclsObj, &uReturn);
if (0 == uReturn) break;
VARIANT vtProp;
hr = pclsObj->Get(L"SerialNumber", 0, &vtProp, 0, 0);
if (SUCCEEDED(hr)) {
serials += _bstr_t(vtProp.bstrVal);
VariantClear(&vtProp);
}
pclsObj->Release();
}
pEnumerator->Release();
}
// 清理资源
pSvc->Release();
pLoc->Release();
CoUninitialize();
// 生成哈希值
HCRYPTPROV hProv = 0;
HCRYPTHASH hHash = 0;
BYTE rgbHash[16];
DWORD cbHash = 16;
CHAR rgbDigits[] = "0123456789abcdef";
if (!CryptAcquireContext(&hProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT)) {
return "CryptoContext Failed";
}
if (!CryptCreateHash(hProv, CALG_MD5, 0, 0, &hHash)) {
CryptReleaseContext(hProv, 0);
return "CreateHash Failed";
}
if (!CryptHashData(hHash, (BYTE*)serials.c_str(), (DWORD)serials.length(), 0)) {
CryptDestroyHash(hHash);
CryptReleaseContext(hProv, 0);
return "HashData Failed";
}
if (!CryptGetHashParam(hHash, HP_HASHVAL, rgbHash, &cbHash, 0)) {
CryptDestroyHash(hHash);
CryptReleaseContext(hProv, 0);
return "GetHashParam Failed";
}
std::string hashStr;
for (DWORD i = 0; i < cbHash; i++) {
char rgb[3];
sprintf_s(rgb, 3, "%c%c", rgbDigits[rgbHash[i] >> 4], rgbDigits[rgbHash[i] & 0xf]);
hashStr += rgb;
}
// 修正strcpy_s调用
strcpy_s(g_szMachineCode, sizeof(g_szMachineCode), hashStr.c_str());
return g_szMachineCode;
}
// 计算注册码函数
const char* ComputeRegistration(const char* machineCode, int daysChoice) {
static char regBuffer[256] = { 0 };
std::string input = machineCode;
std::string regCode;
// 简单转换算法（实际应使用加密算法）
for (char c : input) {
regCode += (c + daysChoice) ^ 0xAA;
}
// 修正strcpy_s调用
strcpy_s(regBuffer, sizeof(regBuffer), regCode.c_str());
return regBuffer;
}
// 保存注册信息函数
BOOL SaveRegistration(const char* machineCode, const char* regCode, int daysChoice) {
HKEY hKey;
LPCSTR subKey = "Software\\MyApp\\Registration";
// 创建注册表项
if (RegCreateKeyExA(HKEY_CURRENT_USER, subKey, 0, NULL,
REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hKey, NULL) != ERROR_SUCCESS) {
return FALSE;
}
// 保存机器码
RegSetValueExA(hKey, "MachineCode", 0, REG_SZ,
(const BYTE*)machineCode, (DWORD)strlen(machineCode) + 1);
// 保存注册码
RegSetValueExA(hKey, "RegCode", 0, REG_SZ,
(const BYTE*)regCode, (DWORD)strlen(regCode) + 1);
// 保存授权天数
DWORD days = 0;
switch (daysChoice) {
case 0: days = 7; break;
case 1: days = 15; break;
case 2: days = 30; break;
case 3: days = 0xFFFFFFFF; // 永久
}
RegSetValueExA(hKey, "Days", 0, REG_DWORD, (const BYTE*)&days, sizeof(days));
// 保存激活日期
SYSTEMTIME st;
GetSystemTime(&st);
FILETIME ft;
SystemTimeToFileTime(&st, &ft);
RegSetValueExA(hKey, "ActivationDate", 0, REG_QWORD, (const BYTE*)&ft, sizeof(ft));
RegCloseKey(hKey);
return TRUE;
}
// 验证授权函数
int ValidateLicense() {
HKEY hKey;
LPCSTR subKey = "Software\\MyApp\\Registration";
if (RegOpenKeyExA(HKEY_CURRENT_USER, subKey, 0, KEY_READ, &hKey) != ERROR_SUCCESS) {
return -1; // 未注册
}
// 检查机器码是否匹配
char storedMachineCode[256] = { 0 };
DWORD size = sizeof(storedMachineCode);
if (RegQueryValueExA(hKey, "MachineCode", NULL, NULL,
(LPBYTE)storedMachineCode, &size) != ERROR_SUCCESS) {
RegCloseKey(hKey);
return -1;
}
if (strcmp(storedMachineCode, GetMachineCode()) != 0) {
RegCloseKey(hKey);
return -1; // 机器码不匹配
}
// 检查授权天数
DWORD days = 0;
size = sizeof(days);
if (RegQueryValueExA(hKey, "Days", NULL, NULL, (LPBYTE)&days, &size) != ERROR_SUCCESS) {
RegCloseKey(hKey);
return -1;
}
// 永久授权
if (days == 0xFFFFFFFF) {
RegCloseKey(hKey);
return 1;
}
// 检查激活日期
FILETIME activationDate;
size = sizeof(activationDate);
if (RegQueryValueExA(hKey, "ActivationDate", NULL, NULL,
(LPBYTE)&activationDate, &size) != ERROR_SUCCESS) {
RegCloseKey(hKey);
return -1;
}
// 计算已过天数
SYSTEMTIME stNow;
FILETIME ftNow;
GetSystemTime(&stNow);
SystemTimeToFileTime(&stNow, &ftNow);
ULARGE_INTEGER uAct, uNow;
uAct.LowPart = activationDate.dwLowDateTime;
uAct.HighPart = activationDate.dwHighDateTime;
uNow.LowPart = ftNow.dwLowDateTime;
uNow.HighPart = ftNow.dwHighDateTime;
// 100纳秒 = 1 tick, 1天 = 864000000000 ticks
ULONGLONG diff = (uNow.QuadPart - uAct.QuadPart) / 864000000000;
RegCloseKey(hKey);
return (diff <= days) ? 1 : 0; // 1=有效, 0=过期
}
// 对话框消息处理
INT_PTR CALLBACK DialogProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) {
switch (uMsg) {
case WM_INITDIALOG:
g_hMainDlg = hDlg;
InitializeControls(hDlg);
ValidateLicenseOnStartup();
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam)) {
case IDC_GENERATE_BTN:
GenerateRegistrationCode(hDlg);
break;
case IDC_REGISTER_BTN:
RegisterProduct(hDlg);
break;
case IDC_VALIDATE_BTN:
ValidateLicenseOnStartup();
break;
case IDCANCEL:
EndDialog(hDlg, 0);
return TRUE;
}
break;
}
return FALSE;
}
// 初始化控件
void InitializeControls(HWND hDlg) {
// 设置字体
HFONT hFont = CreateFont(14, 0, 0, 0, FW_NORMAL, FALSE, FALSE, FALSE,
DEFAULT_CHARSET, OUT_DEFAULT_PRECIS,
CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY,
DEFAULT_PITCH | FF_DONTCARE, L"Arial");
// 获取机器码并显示
const char* mc = GetMachineCode();
SetDlgItemTextA(hDlg, IDC_MACHINE_CODE, mc);
// 设置控件字体
HWND hWnd = GetDlgItem(hDlg, IDC_MACHINE_CODE);
SendMessage(hWnd, WM_SETFONT, (WPARAM)hFont, TRUE);
hWnd = GetDlgItem(hDlg, IDC_REG_CODE);
SendMessage(hWnd, WM_SETFONT, (WPARAM)hFont, TRUE);
// 初始化授权天数下拉框
HWND hCombo = GetDlgItem(hDlg, IDC_DAYS_COMBO);
SendMessageW(hCombo, CB_ADDSTRING, 0, (LPARAM)L"7天试用");
SendMessageW(hCombo, CB_ADDSTRING, 0, (LPARAM)L"15天试用");
SendMessageW(hCombo, CB_ADDSTRING, 0, (LPARAM)L"30天正式版");
SendMessageW(hCombo, CB_ADDSTRING, 0, (LPARAM)L"永久授权");
SendMessage(hCombo, CB_SETCURSEL, 3, 0); // 默认选择永久授权
// 设置状态文本
UpdateStatus("请选择授权类型并生成注册码");
}
// 生成注册码
void GenerateRegistrationCode(HWND hDlg) {
// 获取选择的授权天数
HWND hCombo = GetDlgItem(hDlg, IDC_DAYS_COMBO);
int daysChoice = (int)SendMessage(hCombo, CB_GETCURSEL, 0, 0);
// 计算注册码
const char* mc = GetMachineCode();
const char* regCode = ComputeRegistration(mc, daysChoice);
// 显示注册码
SetDlgItemTextA(hDlg, IDC_REG_CODE, regCode);
UpdateStatus("注册码已生成，点击注册按钮完成激活");
}
// 注册产品
void RegisterProduct(HWND hDlg) {
// 获取选择的授权天数
HWND hCombo = GetDlgItem(hDlg, IDC_DAYS_COMBO);
int daysChoice = (int)SendMessage(hCombo, CB_GETCURSEL, 0, 0);
// 获取注册码
char regCode[256] = { 0 };
GetDlgItemTextA(hDlg, IDC_REG_CODE, regCode, sizeof(regCode));
// 保存注册信息
const char* mc = GetMachineCode();
if (SaveRegistration(mc, regCode, daysChoice)) {
UpdateStatus("产品注册成功！");
MessageBoxA(hDlg, "产品激活成功！", "注册成功", MB_OK | MB_ICONINFORMATION);
}
else {
UpdateStatus("注册失败，请重试");
MessageBoxA(hDlg, "注册失败，请检查系统权限", "错误", MB_OK | MB_ICONERROR);
}
}
// 验证授权
void ValidateLicenseOnStartup() {
int status = ValidateLicense();
switch (status) {
case 1:
UpdateStatus("授权状态: 已激活 (有效授权)");
break;
case 0:
UpdateStatus("授权状态: 已过期 (请重新注册)");
break;
case -1:
UpdateStatus("授权状态: 未注册 (请激活产品)");
break;
default:
UpdateStatus("授权状态: 未知错误");
}
}
// 更新状态文本
void UpdateStatus(const char* message) {
SetDlgItemTextA(g_hMainDlg, IDC_STATUS_TEXT, message);
}
// 主程序入口
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow) {
// 初始化COM
if (FAILED(CoInitializeEx(NULL, COINIT_APARTMENTTHREADED | COINIT_DISABLE_OLE1DDE))) {
MessageBoxA(NULL, "COM初始化失败", "错误", MB_OK | MB_ICONERROR);
return 1;
}
// 初始化通用控件
INITCOMMONCONTROLSEX icc;
icc.dwSize = sizeof(icc);
icc.dwICC = ICC_WIN95_CLASSES;
InitCommonControlsEx(&icc);
// 创建对话框
DialogBoxParam(hInstance, MAKEINTRESOURCE(IDD_MAIN_DIALOG), NULL, DialogProc, 0);
// 清理COM
CoUninitialize();
return 0;
}
[NK201!无法解析的外部符号 main，函数 "int cdecl invoke main(void)"(?invoke main@@YAHXZ)中引用了该符号
LNK112(1 个无法解析的外部命令
最新发布
扫一扫
2239
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
