// autostart_
registry_manager.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <windows.h>
#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <win
reg.h>
#include <tlhelp32.h>
#include <psapi.h>
class Autostart
RegistryManager {
private:
std::map<std::string, std::string> startupLocations;
std::string serviceName;
std::string executablePath;
void InitializeStartupLocations() {
startupLocations["HKCU_Run"] = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run";
startupLocations["HKCU_RunOnce"] = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunOnce";
startupLocations["HKLM_Run"] = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run";
startupLocations["HKLM_RunOnce"] = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunOnce";
startupLocations["Startup_Folder"] = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders";
}
bool CreateServiceName() {
// 生成合法的服务名称
char computerName[MAX_COMPUTERNAME_LENGTH + 1];
DWORD size = sizeof(computerName);
if (GetComputerNameA(computerName, &size)) {
serviceName = "SystemUpdateTask_" + std::string(computerName);
}
else {
serviceName = "SystemUpdateTask_Default";
}
return true;
}
bool GetExecutablePath() {
char buffer[MAX_PATH];
if (GetModuleFileNameA(NULL, buffer, MAX_PATH)) {
executablePath = std::string(buffer);
return true;
}
return false;
}
bool AddTo
Registry(HKEY hKey, const std::string& subKey, const std::string& valueName, const std::string& valueData) {
HKEY hSubKey;
LONG result =
RegCreateKeyExA(hKey, subKey.c_str(), 0, NULL,
REG_
OPTION_NON_
VOLATILE,
KEY_WRITE, NULL, &hSubKey, NULL);
if (result != ERROR_SUCCESS) {
std::cout << "创建注册表项失败: " << subKey << std::endl;
return false;
}
result =
RegSetValueExA(hSubKey, valueName.c_str(), 0,
REG_SZ,
(const BYTE*)valueData.c_str(), valueData.length() + 1);
RegCloseKey(hSubKey);
if (result != ERROR_SUCCESS) {
std::cout << "设置注册表值失败: " << valueName << std::endl;
return false;
}
return true;
}
bool RemoveFrom
Registry(HKEY hKey, const std::string& subKey, const std::string& valueName) {
HKEY hSubKey;
LONG result =
RegOpenKeyExA(hKey, subKey.c_str(), 0, KEY_WRITE, &hSubKey);
if (result != ERROR_SUCCESS) {
std::cout << "打开注册表项失败: " << subKey << std::endl;
return false;
}
result =
RegDeleteValueA(hSubKey, valueName.c_str());
RegCloseKey(hSubKey);
if (result != ERROR_SUCCESS) {
std::cout << "删除注册表值失败: " << valueName << std::endl;
return false;
}
return true;
}
bool Check
RegistryValue(HKEY hKey, const std::string& subKey, const std::string& valueName) {
HKEY hSubKey;
LONG result =
RegOpenKeyExA(hKey, subKey.c_str(), 0, KEY_READ, &hSubKey);
if (result != ERROR_SUCCESS) {
return false;
}
char buffer[1024];
DWORD bufferSize = sizeof(buffer);
DWORD dataType;
result =
RegQueryValueExA(hSubKey, valueName.c_str(), NULL, &dataType, (LPBYTE)buffer, &bufferSize);
RegCloseKey(hSubKey);
return (result == ERROR_SUCCESS);
}
std::string GetStartupFolderPath() {
char buffer[MAX_PATH];
HKEY hKey;
if (
RegOpenKeyExA(HKEY_CURRENT_USER,
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders",
0, KEY_READ, &hKey) == ERROR_SUCCESS) {
DWORD bufferSize = sizeof(buffer);
if (
RegQueryValueExA(hKey, "Startup", NULL, NULL, (LPBYTE)buffer, &bufferSize) == ERROR_SUCCESS) {
RegCloseKey(hKey);
return std::string(buffer);
}
RegCloseKey(hKey);
}
return "";
}
bool CreateShortcutInStartupFolder() {
std::string startupFolder = GetStartupFolderPath();
if (startupFolder.empty()) {
std::cout << "无法获取启动文件夹路径" << std::endl;
return false;
}
std::string shortcutPath = startupFolder + "\\" + serviceName + ".lnk";
// 创建快捷方式文件
std::ofstream shortcut(shortcutPath);
if (!shortcut.is_open()) {
std::cout << "无法创建快捷方式文件" << std::endl;
return false;
}
// 写入快捷方式内容
shortcut << "[InternetShortcut]" << std::endl;
shortcut << "URL=file://" << executablePath << std::endl;
shortcut.close();
return true;
}
bool RemoveShortcutFromStartupFolder() {
std::string startupFolder = GetStartupFolderPath();
if (startupFolder.empty()) {
return false;
}
std::string shortcutPath = startupFolder + "\\" + serviceName + ".lnk";
return DeleteFileA(shortcutPath.c_str()) != 0;
}
bool CreateScheduledTask() {
std::string command = "schtasks /create /tn \"" + serviceName + "\" /tr \"" +
executablePath + "\" /sc onlogon /ru \"SYSTEM\" /f";
STARTUPINFOA si = { 0 };
PROCESS_INFORMATION pi = { 0 };
si.cb = sizeof(si);
if (CreateProcessA(NULL, (LPSTR)command.c_str(), NULL, NULL, FALSE,
CREATE_NO_WINDOW, NULL, NULL, &si, &pi)) {
WaitForSingleObject(pi.hProcess, 5000);
CloseHandle(pi.hProcess);
CloseHandle(pi.hThread);
return true;
}
return false;
}
bool RemoveScheduledTask() {
std::string command = "schtasks /delete /tn \"" + serviceName + "\" /f";
STARTUPINFOA si = { 0 };
PROCESS_INFORMATION pi = { 0 };
si.cb = sizeof(si);
if (CreateProcessA(NULL, (LPSTR)command.c_str(), NULL, NULL, FALSE,
CREATE_NO_WINDOW, NULL, NULL, &si, &pi)) {
WaitForSingleObject(pi.hProcess, 5000);
CloseHandle(pi.hProcess);
CloseHandle(pi.hThread);
return true;
}
return false;
}
public:
Autostart
RegistryManager() {
InitializeStartupLocations();
CreateServiceName();
GetExecutablePath();
}
bool
RegisterAutostart() {
std::cout << "正在注册自启动项..." << std::endl;
// 添加到HKCU\Run
if (!AddTo
Registry(HKEY_CURRENT_USER, startupLocations["HKCU_Run"],
serviceName, executablePath)) {
std::cout << "添加到HKCU\\Run失败" << std::endl;
return false;
}
// 添加到HKLM\Run (需要管理员权限)
if (!AddTo
Registry(HKEY_LOCAL_MACHINE, startupLocations["HKLM_Run"],
serviceName, executablePath)) {
std::cout << "添加到HKLM\\Run失败 (可能需要管理员权限)" << std::endl;
}
// 创建启动文件夹快捷方式
if (!CreateShortcutInStartupFolder()) {
std::cout << "创建启动文件夹快捷方式失败" << std::endl;
}
// 创建计划任务
if (!CreateScheduledTask()) {
std::cout << "创建计划任务失败" << std::endl;
}
std::cout << "自启动项注册完成" << std::endl;
return true;
}
bool Un
registerAutostart() {
std::cout << "正在移除自启动项..." << std::endl;
// 从HKCU\Run移除
RemoveFrom
Registry(HKEY_CURRENT_USER, startupLocations["HKCU_Run"], serviceName);
// 从HKLM\Run移除
RemoveFrom
Registry(HKEY_LOCAL_MACHINE, startupLocations["HKLM_Run"], serviceName);
// 移除启动文件夹快捷方式
RemoveShortcutFromStartupFolder();
// 移除计划任务
RemoveScheduledTask();
std::cout << "自启动项移除完成" << std::endl;
return true;
}
bool Is
Registered() {
// 检查HKCU\Run
if (Check
RegistryValue(HKEY_CURRENT_USER, startupLocations["HKCU_Run"], serviceName)) {
return true;
}
// 检查HKLM\Run
if (Check
RegistryValue(HKEY_LOCAL_MACHINE, startupLocations["HKLM_Run"], serviceName)) {
return true;
}
// 检查启动文件夹
std::string startupFolder = GetStartupFolderPath();
if (!startupFolder.empty()) {
std::string shortcutPath = startupFolder + "\\" + serviceName + ".lnk";
if (GetFileAttributesA(shortcutPath.c_str()) != INVALID_FILE_ATTRIBUTES) {
return true;
}
}
return false;
}
void ListAutostartItems() {
std::cout << "=== 当前自启动项 ===" << std::endl;
// 列出HKCU\Run项
HKEY hKey;
if (
RegOpenKeyExA(HKEY_CURRENT_USER, startupLocations["HKCU_Run"].c_str(),
0, KEY_READ, &hKey) == ERROR_SUCCESS) {
char valueName[256];
char valueData[1024];
DWORD valueNameSize = sizeof(valueName);
DWORD valueDataSize = sizeof(valueData);
DWORD index = 0;
while (
RegEnumValueA(hKey, index, valueName, &valueNameSize, NULL, NULL,
(LPBYTE)valueData, &valueDataSize) == ERROR_SUCCESS) {
std::cout << "HKCU\\Run: " << valueName << " = " << valueData << std::endl;
valueNameSize = sizeof(valueName);
valueDataSize = sizeof(valueData);
index++;
}
RegCloseKey(hKey);
}
// 列出HKLM\Run项
if (
RegOpenKeyExA(HKEY_LOCAL_MACHINE, startupLocations["HKLM_Run"].c_str(),
0, KEY_READ, &hKey) == ERROR_SUCCESS) {
char valueName[256];
char valueData[1024];
DWORD valueNameSize = sizeof(valueName);
DWORD valueDataSize = sizeof(valueData);
DWORD index = 0;
while (
RegEnumValueA(hKey, index, valueName, &valueNameSize, NULL, NULL,
(LPBYTE)valueData, &valueDataSize) == ERROR_SUCCESS) {
std::cout << "HKLM\\Run: " << valueName << " = " << valueData << std::endl;
valueNameSize = sizeof(valueName);
valueDataSize = sizeof(valueData);
index++;
}
RegCloseKey(hKey);
}
}
std::string GetServiceName() const {
return serviceName;
}
std::string GetExecutablePath() const {
return executablePath;
}
};
int main() {
SetConsoleOutputCP(CP_UTF8);
Autostart
RegistryManager manager;
std::cout << "自启动注册管理器" << std::endl;
std::cout << "服务名称: " << manager.GetServiceName() << std::endl;
std::cout << "可执行文件路径: " << manager.GetExecutablePath() << std::endl;
// 检查当前状态
if (manager.Is
Registered()) {
std::cout << "当前已注册自启动" << std::endl;
}
else {
std::cout << "当前未注册自启动" << std::endl;
}
// 列出当前自启动项
std::cout << "\n当前系统自启动项:" << std::endl;
manager.ListAutostartItems();
// 注册自启动
std::cout << "\n正在注册自启动..." << std::endl;
if (manager.
RegisterAutostart()) {
std::cout << "自启动注册成功" << std::endl;
}
else {
std::cout << "自启动注册失败" << std::endl;
}
// 再次检查状态
if (manager.Is
Registered()) {
std::cout << "注册后状态: 已注册" << std::endl;
}
else {
std::cout << "注册后状态: 未注册" << std::endl;
}
// 移除自启动
std::cout << "\n正在移除自启动..." << std::endl;
if (manager.Un
registerAutostart()) {
std::cout << "自启动移除成功" << std::endl;
}
else {
std::cout << "自启动移除失败" << std::endl;
}
std::cout << "\n自启动注册管理测试完成" << std::endl;
system("pause");
return 0;
}
用vs2015编译的时候报错了，修改代码并给出完整的C++代码
本文详细解释了在Windows注册表操作中两个关键选项的区别：REG_OPTION_NON_VOLATILE和REG_OPTION_VOLATILE。通过RegCreateKeyEx函数的使用，了解创建永久注册表项与临时注册表项的不同行为，以及它们如何影响注册表信息的持久性。
扫一扫
2127
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
