如何编写windows服务程序

如何编写windows服务程序,第1张

Windows提供了一套后台服务程序编程接口,用户在编写后台服务程序时需要遵循一定的编程框架,否则服务程序不能正常运行。

服务程序通常编写成控制台类型的应用程序,总的来说,一个遵守服务控制管理程序接口要求的程序 包含下面三个函数

1)服务程序主函数(main):调用系统函数 StartServiceCtrlDispatcher 连接程序主线程到服务控制管理程序。

和其它进程一样,Main函数是服务进程的入口函数,服务控制管理器(SCM)在启动服务程序时,会从服务程序的main函数开始执行。在进入点函数里面要完成ServiceMain的初始化,准确点说是初始化一个SERVICE_TABLE_ENTRY结构数组,这个结构记录了这个服务程序里面所包含的所有服务的名称和服毁备务的进入点函数。然后再调用接口StartServiceCtrlDispatcher 。

Main函数的函数框架如下:

int _tmain(int argc, _TCHAR* argv[])

{

//服务入口点函数表

SERVICE_TABLE_ENTRY dispatchTable[]=

{

{TEXT(SZSERVICENAME),(LPSERVICE_MAIN_FUNCTION)Service_Main},

{ NULL,NULL}

}

if((argc>1)&&((*argv[1]==‘-‘)||(argv[1]=="/")))

{

/*

参数个数大于1是安装或者删除服务,该 *** 作是由用户来执行的

当然也可以讲这一部分功能另写一个程序来实现

*/

if(_stricmp("install",argv[1]+1)==0)

{

installService()

}

else if(_stricmp("remove",argv[1]+1)==0)

{

removeService()

}

else if(_stricmp("debug",argv[1]+1)==0)

{

bDebugServer=true

debugService(argc,argv)

}

}

else

{

/*

如果未能和上面的如何参数匹配,则可能是服务控制管理程序来启动该程序。 立即调用StartServiceCtrlDispatcher 函数

*/

g_logout.Logout("%s\n", "enter StartServiceCtrlDispatcher...")

//通知服务管理器为每一个服务创建服务线程

if(!StartServiceCtrlDispatcher(dispatchTable))

g_logout.Logout("%s\n", "StartServiceCtrlDispatcher failed.")

else

g_logout.Logout("%s\n", "StartServiceCtrlDispatcher OK.")

}

return 0

}

SCM启动一个服务程序之后,它会等待该程序的主线程去调StartServiceCtrlDispatcher。如果那个函数在两分钟内没有被调用,SCM将会认为这个服务有问题,并调用TerminateProcess去杀死这个进程。这就要求你的主厅橘线程要尽可能快的调用StartServiceCtrlDispatcher。

2)服务入口点函数(ServiceMain):执行服务初始化任务,同时纤伏毁执行多个服务的服务进程有多个服务入口函数。

在服务入口函数里,必须立即注册服务控制回调函数。然后调用函数SetServiceStatus 通知SCM 服务现在的状态,否则SCM会认为服务启动失败。

ServiceMain函数框架如下:

void WINAPI Service_Main(DWORD dwArgc, LPTSTR *lpszArgv)

{

//注册服务控制处理函数

sshStatusHandle=RegisterServiceCtrlHandler(TEXT(SZSERVICENAME),Service_Ctrl)

//如果注册失败

if(!sshStatusHandle)

{

g_logout.Logout("%s\n", "RegisterServiceCtrlHandler failed...")

return

}

//初始化 SERVICE_STATUS 结构中的成员

ssStatus.dwServiceType=SERVICE_WIN32_OWN_PROCESS//可执行文件中只有一个单独的服务

ssStatus.dwServiceSpecificExitCode=0

ssStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP//允许用SCP去停止该服务

//更新服务状态

if(ReportStatusToSCMgr(SERVICE_START_PENDING,//服务状态,服务仍在初始化

NO_ERROR,

3000)) //等待时间

SvcInit( dwArgc, lpszArgv )//服务初始化函数

else

g_logout.Logout("%s\n", "ReportStatusToSCMgr SERVICE_START_PENDING failed...")

}

服务初始化函数SvcInit:

该函数的写法比较重要。在函数中创建一个等待事件,然后一直等待该事件。该线程在服务接到请求之前一直处于挂起状态,直到接到服务停止消息。

VOID SvcInit( DWORD dwArgc, LPTSTR *lpszArgv)

{

/*创建事件*/

ghSvcStopEvent = CreateEvent(

NULL,// default security attributes

TRUE,// manual reset event

FALSE, // not signaled

NULL) // no name

if ( ghSvcStopEvent == NULL)

{

ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 )

return

}

// Report running status when initialization is complete.

ReportSvcStatus( SERVICE_RUNNING, NO_ERROR, 0 )

// 在这里执行服务线程的创建...

while(1)

{

// 等待停止事件被触发

WaitForSingleObject(ghSvcStopEvent, INFINITE)

ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 )

return

}

}

3)控制服务处理程序函数(Handler):在服务程序收到控制请求时由控制分发线程引用。(此处是Service_Ctrl)。

void WINAPI Service_Ctrl(DWORD dwCtrlCode)

{

//处理控制请求码

switch(dwCtrlCode)

{

//先更新服务状态为 SERVICDE_STOP_PENDING,再停止服务。

case SERVICE_CONTROL_STOP:

ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500)

ServiceStop()//由具体的服务程序实现

/*ssStatus.dwCurrentState=SERVICE_STOPPED*/

//其它控制请求...

default:

break

}

}

3. 注意事项

1)安装服务可以另写一个程序,也可以并在服务程序当中,比较简单,这里就不介绍了。

2)服务初始化函数SvcInit里创建等待事件比较重要,在服务接收到服务停止消息后,触发该事件。

3)Service_Main在等待事件触发后立即返回,服务进程就会退出了

环境: vs2010,多字符集

以下是一些基本知识, 不得不耐下心去理解,这对程序理解和编写非常用帮助.

首先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_ENTRY entrytable[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函数为void WINAPI ServiceMain(int argc, 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为void WINAPI CtrlHandler(DWORD request)型的函数,函数名字可以任意设定。将在下一点讲到。

Hstatus 为SERVICE_STATUS_HANDLE 类 型 的 全 局 变 量 。 当 需 要 改 变 服 务 状 态 时SetServiceStatus()函数需要它做为参数来标识一个服务。

3. void WINAPI CtrlHandler(DWORD request),函数的主要功能是,接收系统传递的控制命令,比如当你通过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来运行进程。

因此需要

BOOL GetTokenByName(HANDLE &hToken,LPSTR lpName)

BOOL RunProcess(LPCSTR lpImage)两个函数

示例是关于基于opencv人脸识别, 遍历样本文件夹,删除多余的图片保留10张, 然后执行 外部自定义程序"GetFeatureDATA.exe "函数提取特征

GetFeatureDATA.exe中最头上加上#pragma comment(linker,"/subsystem:\"Windows\" /entry:\"mainCRTStartup\"")就可以隐藏控制台窗口

开发环境vs2010, 控制台应用程序 一个.cpp文件.

//服务程序主函数。

[cpp] view plaincopy

#include"stdio.h"

#include"vector"

#include"Tlhelp32.h"

#include<afx.h>

#define_AFXDLL

//由于做的图像识别需要opencv头文件, 需要什么文件自行更改

#include"cv.h"

#include"highgui.h"

usingnamespacestd

//你的服务程序需要以下代码

SERVICE_STATUS servicestatus

SERVICE_STATUS_HANDLE hstatus//全局变量. 是setServiceStatus()的参数, 改变服务状态

voidWINAPI ServiceMain(intargc,char**argv)

voidWINAPI CtrlHandler(DWORD request)

boolbrun=false//原来代码有的,我没有用,还是保留

//以下是以获取登录用户名

BOOL GetTokenByName(HANDLE &hToken,LPSTR lpName)

BOOL RunProcess(LPCSTR lpImage)

//自己添加的代码

inttrain_time//以分钟计

vector<CString>Vec_Dir//存放图片文件夹目录名称

vector<CString>Vec_Img//

voidTraverseDir(CString&strDir,std::vector<CString>&vecDir)

intTraverseImg(CString&strDir,std::vector<CString>&vecFile)

voidTraverseDir(CString&strDir,std::vector<CString>&vecDir)

{

WIN32_FIND_DATA FindFileData

CStringstrDirTmp

strDirTmp =strDir

strDirTmp +="\\*.*"

HANDLE hFind=::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<CString>&vecFile)//输入路径,得到img路径文件名 不用的请忽视

{

intImgNum=0

WIN32_FIND_DATA FindFileData

CStringstrDirTmp

strDirTmp =strDir

strDirTmp +="\\*.*"

HANDLE hFind=::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

}

voidWINAPI ServiceMain(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)

{

return

}

//向SCM 报告运行状态

servicestatus.dwCurrentState =SERVICE_RUNNING

SetServiceStatus(hstatus,&servicestatus)

//下面就

brun=true

//以下是自己要写的代码的执行调用地方.开始任务循环了,你可以添加你自己希望服务做的工作

//SYSTEMTIME t

//GetLocalTime(&t)

//int hour = t.wHour //获取小时, 可以在固定某个小时执行程序

while(1)//

{

CStringSamplesDirPath=_T("G:\\Samples")

TraverseDir(SamplesDirPath,Vec_Dir)//获取目录名称到vec_Dir

for(inti(0)i<Vec_Dir.size()i++)//

{

intImg_Num=TraverseImg(Vec_Dir[i],Vec_Img)//某个目录下的所有图片

if(Img_Num>10)

{

for(intj=Img_Num-10-1j>=0j--)

{

remove(Vec_Img[j])

}

}

Vec_Img.clear()

}

//几种调用外部程序的方法,但除了RunProcess其他都是以system身份打开程序.

//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自定时间后再次执行 *** 作

}

}

BOOL GetTokenByName(HANDLE &hToken,LPSTR lpName)

{

if(!lpName)

{

returnFALSE

}

HANDLE hProcessSnap =NULL

BOOL bRet =FALSE

PROCESSENTRY32 pe32 ={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)))

{

HANDLE hProcess =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)

}

BOOL RunProcess(LPCSTR lpImage)

{

if(!lpImage)

{

returnFALSE

}

HANDLE hToken

if(!GetTokenByName(hToken,"EXPLORER.EXE"))

{

returnFALSE

}

STARTUPINFO si

PROCESS_INFORMATION pi

ZeroMemory(&si,sizeof(STARTUPINFO))

si.cb=sizeof(STARTUPINFO)

si.lpDesktop =TEXT("winsta0\\default")

BOOL bResult =CreateProcessAsUser(hToken,lpImage,NULL,NULL,NULL,

FALSE,NORMAL_PRIORITY_CLASS,NULL,NULL,&si,&pi)

CloseHandle(hToken)

if(bResult)

{

OutputDebugString("CreateProcessAsUser ok!\r\n")

}

else

{

OutputDebugString("CreateProcessAsUser false!\r\n")

}

returnbResult

}

然后安装服务

点开始运行cmd.exe

输入以下:

sc create your_service_name binpath= D:\backup\GetXML.exe //这步注意等号右边有个空格

sc start your_service_name //启动服务, 也可以启动任务管理器在服务一栏中找到你的服务启动或停止

sc stop testservicename //停止服务

sc delete testservicename //删除服务,该服务将在下次重启后删除,在重启之前将不能注册

同一个名字的服务。

启动服务, 停止服务,等也可以在windows任务管理器中管理


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/yw/12211920.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023-05-21
下一篇 2023-05-21

发表评论

登录后才能评论

评论列表(0条)

保存