快速编写简单windows服务程序.docx
《快速编写简单windows服务程序.docx》由会员分享,可在线阅读,更多相关《快速编写简单windows服务程序.docx(9页珍藏版)》请在冰豆网上搜索。
快速编写简单windows服务程序
前不久,为完成项目中某个功能,需要写个windows服务程序,定时来调用另外的一个exe程序,完成过程遇到颇多问题,作为初学者,分享一下我的整个过程.参考了多位大神的文章.资料来自网上,最后给出链接.表示感谢
以下是一些基本知识,不得不耐下心去理解,这对程序理解和编写非常用帮助.
首先Microsoft Windows服务(即,以前的 NT服务)使您能够创建在它们自己的Windows会话中可长时间运行的可执行应用程序。
这些服务可以在计算机启动时自动启动,可以暂停和重新启动而且不显示任何用户界面。
这使服务非常适合在服务器上使用,或任何时候,为了不影响在同一台计算机上工作的其他用户,需要长时间运行功能时使用。
还可以在不同于登录用户的特定用户帐户或默认计算机帐户的安全上下文中运行服务。
服务是有状态的,当我们使用windows自带的服务管理程序sc.exe查看服务状态时可以显示服务的当前状态,这个状态是由我们在程序代码中进行控制的。
你最好在服务初始化的时候将服务设置为SERVICE_START_PENDING,当初始化完毕时设为SERVICE_RUNNING,这些状
态是系统自定义的状态,可通过msdn查看其他状态。
这个状态信息你会在sc.exe中看到。
在编写windows服务程序过程中你需要关注的函数有:
1.首先是main函数,由于windows服务不需要界面,所以大部分程序为win32控制台应用程序,所以程序主函数为main而不是WinMain()。
在主函数要做的主要工作就是初始化一个SERVICE_TABLE_ENTRY分派表结构体,然后调用StartServiceCtrlDispatcher();这将把调用进程的主线程转换为控制分派器。
该分派器启动一个新线程,该线程运行分派表中对应于你的服务的ServiceMain()函数。
ServiceMain()函数将在下面提到。
此过程示例代码如下:
SERVICE_TABLE_ENTRYentrytable[2];
entrytable[0].lpServiceName="testservice";
entrytable[0].lpServiceProc=(LPSERVICE_MAIN_FUNCTION)ServiceMain;
entrytable[1].lpServiceName=NULL;
entrytable[1].lpServiceProc=NULL;
StartServiceCtrlDispatcher(entrytable);
在这之后系统将自动创建一个线程去执行ServiceMain函数的内容,你应该将你要执行的任务
在ServiceMain中循环,这样服务就开始运行了。
2.ServiceMain函数为voidWINAPIServiceMain(intargc,char**argv)格式的函数,函数名字可以任意定义。
它的作用就是:
将你需要执行的任务放到该函数中循环执行即可。
这就是服务程序的工作函数。
在ServiceMain执行你的任务前,需要给SERVICE_TABLE_ENTRY分派
表结构体进行赋值,注意由于此时服务还没有开始执行你的任务所以我们将服务的状态设置为SERVICE_START_PENDING,即正在初始化。
我们进行如下赋值:
servicestatus.dwServiceType=SERVICE_WIN32;
servicestatus.dwCurrentState=SERVICE_START_PENDING;
servicestatus.dwControlsAccepted=SERVICE_ACCEPT_SHUTDOWN|SERVICE_ACCEPT_STOP;
//在本例中只接受系统关机和停止服务两种控制命令
servicestatus.dwWin32ExitCode=0;
servicestatus.dwServiceSpecificExitCode=0;
servicestatus.dwCheckPoint=0;
servicestatus.dwWaitHint=0;
hstatus=:
:
RegisterServiceCtrlHandler("testservice",CtrlHandler);
CtrlHandler为voidWINAPICtrlHandler(DWORDrequest)型的函数,函数名字可以任意设定。
将在下一点讲到。
Hstatus为SERVICE_STATUS_HANDLE类型的全局变量。
当需要改变服务状态时SetServiceStatus()函数需要它做为参数来标识一个服务。
3.voidWINAPICtrlHandler(DWORDrequest),函数的主要功能是,接收系统传递的控制命令,比如当你通过sc.exe关闭服务时,该函数会收到SERVICE_CONTROL_STOP消息,你就可以对服务进行必要的管理。
在本例子程序中就只接收SERVICE_ACCEPT_SHUTDOWN和
SERVICE_ACCEPT_STOP消息,这是通过前面给servicestatus赋值设定的。
这样一个基本的服务程序就完成了。
本文结束的时候会附上如
何安装服务。
当服务程序需要使用某些功能时,由于用户的关系而受到限制,比如访问注册表的HKEY_CURRENT_USER键,使用网络等等,这时候就需要以当前登陆用户的身份去进行操作,通常会创建一个进程来完成需要的功能。
如果使用CreateProcess,来创建进程的话,新创建的进程和服务程序依然是相同的用户身份,还是无法达到目的,只有使用CreateProcessAsUser了。
但CreateProcessAsUser的第一个参数是HANDLE hToken,该参数通常应该用LogonUser来获得,但是LogonUser又需要用户名和用户密码,这样就很不现实。
那应该怎么办呢?
我想到了一个方法可以绕过LogonUser直接获得hToken。
因为用户已经登陆,那么肯定有Shell(就是EXPLORER.EXE)运行了,我们可以通过遍历进程来取得Shell的hToken来运行进程。
因此需要
BOOLGetTokenByName(HANDLE&hToken,LPSTRlpName);
BOOLRunProcess(LPCSTRlpImage);两个函数
示例是关于基于opencv人脸识别,遍历样本文件夹,删除多余的图片保留10张, 然后执行外部自定义程序"GetFeatureDATA.exe"函数提取特征
GetFeatureDATA.exe中最头上加上#pragmacomment(linker,"/subsystem:
\"Windows\"/entry:
\"mainCRTStartup\"")就可以隐藏控制台窗口
开发环境vs2010,控制台应用程序.
//服务程序主函数。
#include"stdio.h"
#include"vector"
#include"Tlhelp32.h"
#include
#define_AFXDLL
//由于做的图像识别需要opencv头文件,需要什么文件自行更改
#include"cv.h"
#include"highgui.h"
usingnamespacestd;
//你的服务程序需要以下代码
SERVICE_STATUSservicestatus;
SERVICE_STATUS_HANDLEhstatus;//全局变量.是setServiceStatus()的参数,改变服务状态
voidWINAPIServiceMain(intargc,char**argv);
voidWINAPICtrlHandler(DWORDrequest);
boolbrun=false;//原来代码有的,我没有用,还是保留
//以下是以获取登录用户名
BOOLGetTokenByName(HANDLE&hToken,LPSTRlpName);
BOOLRunProcess(LPCSTRlpImage);
//自己添加的代码
inttrain_time;//以分钟计
vectorVec_Dir;//存放图片文件夹目录名称
vectorVec_Img;//
voidTraverseDir(CString&strDir,std:
:
vector&vecDir);
intTraverseImg(CString&strDir,std:
:
vector&vecFile);
voidTraverseDir(CString&strDir,std:
:
vector&vecDir)
{
WIN32_FIND_DATAFindFileData;
CStringstrDirTmp;
strDirTmp=strDir;
strDirTmp+="\\*.*";
HANDLEhFind=:
:
FindFirstFile(strDirTmp,&FindFileData);
if(INVALID_HANDLE_VALUE==hFind)
{
return;
}
while(TRUE)
{
if(FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)
{
if(FindFileData.cFileName[0]!
=_T('.'))
{
strDirTmp=strDir;
strDirTmp+="\\";
strDirTmp+=FindFileData.cFileName;
vecDir.push_back(strDirTmp);//保存所有目录
//TraverseDir(strDirTmp,vecFile);
}
}
else//是文件
{
/*strDirTmp=strDir;
strDirTmp+="\\";
strDirTmp+=FindFileData.cFileName;
vecFile.push_back(strDirTmp);*/
}
if(!
FindNextFile(hFind,&FindFileData))
break;
}
FindClose(hFind);
}
intTraverseImg(CString&strDir,std:
:
vector&vecFile)//输入路径,得到img路径文件名
{
intImgNum=0;
WIN32_FIND_DATAFindFileData;
CStringstrDirTmp;
strDirTmp=strDir;
strDirTmp+="\\*.*";
HANDLEhFind=:
:
FindFirstFile(strDirTmp,&FindFileData);
if(INVALID_HANDLE_VALUE==hFind)
{
//return;
}
while(TRUE)
{
if(FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)
{
if(FindFileData.cFileName[0]!
=_T('.'))
{
/*strDirTmp=strDir;
strDirTmp+="\\";
strDirTmp+=FindFileData.cFileName;
TraverseDir(strDirTmp,vecFile);*/
}
}
else
{
strDirTmp=strDir;
strDirTmp+="\\";
strDirTmp+=FindFileData.cFileName;
vecFile.push_back(strDirTmp);//将图片路径传入
ImgNum++;
}
if(!
FindNextFile(hFind,&FindFileData))
break;
}
FindClose(hFind);
returnImgNum;
}
voidWINAPIServiceMain(intargc,char**argv)
{
servicestatus.dwServiceType=SERVICE_WIN32;
servicestatus.dwCurrentState=SERVICE_START_PENDING;
servicestatus.dwControlsAccepted=SERVICE_ACCEPT_SHUTDOWN|SERVICE_ACCEPT_STOP;//在本例中只接受系统关机和停止服务两种控制命令
servicestatus.dwWin32ExitCode=0;
servicestatus.dwServiceSpecificExitCode=0;
servicestatus.dwCheckPoint=0;
servicestatus.dwWaitHint=0;
hstatus=:
:
RegisterServiceCtrlHandler("testservice",CtrlHandler);
if(hstatus==0)
{
//WriteToLog("RegisterServiceCtrlHandlerfailed");
return;
}
//WriteToLog("RegisterServiceCtrlHandlersuccess");
//向SCM报告运行状态
servicestatus.dwCurrentState=SERVICE_RUNNING;
SetServiceStatus(hstatus,&servicestatus);
//下面就开始任务循环了,你可以添加你自己希望服务做的工作
brun=true;
//MEMORYSTATUSmemstatus;
//charstr[100];
//memset(str,'\0',100);
//while(brun)
//{
//GlobalMemoryStatus(&memstatus);
//intavailmb=memstatus.dwAvailPhys/1024/1024;
//sprintf_s(str,100,"availablememoryis%dMB",availmb);
//WriteToLog(str);
//Sleep(SLEEP_TIME);
//}
//WriteToLog("servicestopped");
//
//以下是自己要写的代码的执行调用地方
//SYSTEMTIMEt;
//GetLocalTime(&t);
//inthour=t.wHour;//获取小时,可以在固定某个小时执行程序
while
(1)//
{
CStringSamplesDirPath=_T("G:
\\Samples");
TraverseDir(SamplesDirPath,Vec_Dir);//获取目录名称到vec_Dir
for(inti(0);i{
intImg_Num=TraverseImg(Vec_Dir[i],Vec_Img);//某个目录下的所有图片
if(Img_Num>10)
{
for(intj=Img_Num-10-1;j>=0;j--)
{
remove(Vec_Img[j]);
}
}
Vec_Img.clear();
}
//执行提取xml文件exe文件
//WinExec("G:
\\about_MFC\\GetFeatureDATA.exe",0);
//system("GetFeatureDATA.exe");
//ShellExecute(NULL,"open","G:
\\about_MFC\\GetFeatureDATA.exe",NULL,NULL,SW_SHOWNORMAL);
RunProcess("G:
\\about_MFC\\GetFeatureDATA.exe");
//从硬盘里读取时间来做个每隔多少时间进行
CvFileStorage*Threshold=cvOpenFileStorage("./service_time.xml",0,CV_STORAGE_READ);//读取预值
CvFileNode*ThresholdNode=cvGetFileNodeByName(Threshold,0,"circle_time");
doubleservice_time=cvReadRealByName(Threshold,ThresholdNode,"circle_time");
Sleep(service_time*60*1000);//sleep自定时间后再次执行操作
}
}
BOOLGetTokenByName(HANDLE&hToken,LPSTRlpName)
{
if(!
lpName)
{
returnFALSE;
}
HANDLEhProcessSnap=NULL;
BOOLbRet=FALSE;
PROCESSENTRY32pe32={0};
hProcessSnap=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);
if(hProcessSnap==INVALID_HANDLE_VALUE)
return(FALSE);
pe32.dwSize=sizeof(PROCESSENTRY32);
if(Process32First(hProcessSnap,&pe32))
{
do
{
if(!
strcmp(_strupr(pe32.szExeFile),_strupr(lpName)))
{
HANDLEhProcess=OpenProcess(PROCESS_QUERY_INFORMATION,
FALSE,pe32.th32ProcessID);
bRet=OpenProcessToken(hProcess,TOKEN_ALL_ACCESS,&hToken);
CloseHandle(hProcessSnap);
return(bRet);
}
}
while(Process32Next(hProcessSnap,&pe32));
bRet=TRUE;
}
else
bRet=FALSE;
CloseHandle(hProcessSnap);
return(bRet);
}
BOOLRunProcess(LPCSTRlpImage)
{
if(!
lpImage)
{
returnFALSE;
}
HANDLEhToken;
if(!
GetTokenByName(hToken,"EXPLORER.EXE"))
{
returnFALSE;
}
STARTUPINFOsi;
PROCESS_INFORMATIONpi;
ZeroMemory(&si,sizeof(STARTUPINFO));
si.cb=sizeof(STARTUPINFO);
si.lpDesktop=TEXT("winsta0\\default");
BOOLbResult=CreateProcessAsUser(hToken,lpImage,NULL,NULL,NULL,
FALSE,NORMAL_PRIORITY_CLASS,NULL,NULL,&si,&pi);
CloseHandle(hToken);
if(bResult)
{
OutputDebugString("CreateProcessAsUserok!
\r\n");
}
else
{
OutputDebugString("CreateProcessAsUserfalse!
\r\n");
}
returnbResult;
}
然后安装服务运行cmd.exe
输入以下:
sccreateyour_service_namebinpath=D:
\backup\GetXML.exe //这步注意等号右边有个空格
scstart your_service_name //启动服务,也可以启动任务管理器在服务一栏中找到你的服务启动或停止
scstoptestservicename //停止服务
scdeletetestservicename//删除服务,该服务将在下次重启后删除,在重启之前将不能注册
同一个名字的服务。