功能:实现一个windows守护进程,设置有自定义图标,启动后隐藏黑窗口,获取exe所在目录,并设置开机自启动(需要以管理员权限运行)。如果需要打包成可执行的安装包程序,见另一篇博客:/qq_24977505/article/details/106415058。
隐藏控制台窗口:main函数前添加
#pragma comment(linker, "/subsystem:\"windows\" /entry:\"mainCRTStartup\"" ) 即可。
设置自定义图标:Damon.exe设置heworld.ico图标,HelloWorld.exe没有
1、需要先自制一个ico格式图标,点开/上传图片即可做好。
2、将图标放到代码路径下,在vs项目中添加.rc文件(选中项目->右键添加->新建项->资源文件.rc->确定)。
3、到源文件目录记事本打开这个.rc文件,末尾添加 IDI_ICON1 ICON DISCARDABLE "heworld.ico" ,heworld.ico是第一步自作好的ico文件,一定要同在源文件目录下。
4、重新编译即可。
设置开机自启动代码如下:
#include <stdio.h>
#include <stdlib.h>
#include <Windows.h>
#include <time.h>
// 设置黑窗隐藏
#pragma comment(linker, "/subsystem:\"windows\" /entry:\"mainCRTStartup\"" )
// 需要守护的进程可执行文件,和daemon.exe放一起
#define PROCCESS_NAME "HelloWorld.exe"
char g_workPath[MAX_PATH] = { 0 };
void WriteLogFile(char *msg);
BOOL SetSelfStart()
{
//获取程序完整名称
char pName[MAX_PATH] = { 0 };
GetModuleFileNameA(NULL, pName, MAX_PATH);
HKEY hKey = NULL;
LONG lRet = NULL;
const char * regeditPath = "Software\\Microsoft\\Windows\\CurrentVersion\\Run\\";
if (ERROR_SUCCESS != RegOpenKeyExA(HKEY_LOCAL_MACHINE, regeditPath, 0, KEY_ALL_ACCESS, &hKey))
{
return FALSE;
}
if (ERROR_SUCCESS != RegSetValueExA(hKey, "Heworld", 0, REG_SZ, (const unsigned char*)pName, strlen(pName) + sizeof(char)))
{
return FALSE;
}
RegCloseKey(hKey);
WriteLogFile("守护进程开机自启动成功\n");
return TRUE;
}
void GetWorkPath(char *argv)//获取exe所在目录
{
char drive[4];
char subdir[MAX_PATH];
char fn[MAX_PATH];
char exten[MAX_PATH];
_splitpath_s(argv, drive, subdir, fn, exten);
sprintf_s(g_workPath, "%s%s", drive, subdir);
}
int main(int argc, char *argv[])
{
GetWorkPath(argv[0]);
if (!SetSelfStart())
{
WriteLogFile("守护进程开机自启动失败\n");
}
STARTUPINFOA si;
PROCESS_INFORMATION pi;
//初始化
ZeroMemory(&si, sizeof(si));
si.wShowWindow = SW_HIDE;
si.cb = sizeof(si);
ZeroMemory(&pi, sizeof(pi));
//构造cmd执行守护进程的字符串
static char fileName[MAX_PATH] = { 0 };
snprintf(fileName, MAX_PATH, "%s%s", g_workPath, PROCCESS_NAME);
do {
if (!CreateProcessA(NULL, fileName, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
{
WriteLogFile("守护进程启动失败,程序即将退出\n");
return -1;
}
WriteLogFile("守护进程成功\n");
WaitForSingleObject(pi.hProcess, INFINITE);
WriteLogFile("守护进程退出......\n");
//关闭进程和句柄
CloseHandle(pi.hProcess);
CloseHandle(pi.hThread);
Sleep(2000);
} while (true);
return 0;
}
const char *GetNowStr()
{
static char nowstr[64];
time_t nowTime = 0;
time(&nowTime);
struct tm tmNow;
localtime_s(&tmNow, &nowTime);
sprintf_s(nowstr, "%02d%02d_%02d%02d%02d", tmNow.tm_mon + 1, tmNow.tm_mday,
tmNow.tm_hour, tmNow.tm_min, tmNow.tm_sec);
return nowstr;
}
void WriteLogFile(char *msg)
{
static int writeLine = 0;
static FILE* fp = NULL;
if (fp == NULL)
{
char buf[1024];
sprintf_s(buf, "%s%s", g_workPath, "daemon.log");
fopen_s(&fp, buf, "w");
if (fp == NULL)
{
printf("open log file fail\n");
return;
}
}
static char header[128];
sprintf_s(header, "[%s]:", GetNowStr());
fwrite(header, 1, strlen(header), fp);
fwrite(msg, 1, strlen(msg), fp);
fflush(fp);
printf(msg);
if (50000 < writeLine++)
{
fp = NULL;
writeLine = 0;
}
}
如果觉得《Windows守护进程实现/设置程序图标/隐藏黑窗/获取exe所在目录/设置开机自启动》对你有帮助,请点赞、收藏,并留下你的观点哦!