建立SvcHost.exe呼叫的服務原理與實踐(2)

Just4life發表於2013-05-16
8. 程式碼

// SvcHostDLL.cpp : Demo for a service dll used by svchost.exe to host it.
//
// for detail comment see articles.
// by bingle_at_email.com.cn
//  www.BingleSite.net
//
/* save following as a .def file to export function, only ServiceMain is needed.
other used to install & uninstall service.
or use /EXPORT: link option to export them.
EXPORTS
 ServiceMain
 InstallService
 UninstallService
 RundllUninstallA
 RundllInstallA
*/
/*
To compile & link:
cl /MD /GX /LD svchostdll.cpp /link
advapi32.lib /DLL /base:0x71000000 /export:ServiceMain
/EXPORT:RundllUninstallA /EXPORT:RundllInstallA
/EXPORT:InstallService /EXPORT:UninstallService
*/
//
// Articles:
// 1. HOWTO Create a service dll used by svchost.exe by bingle, at: http://www.BingleSite.net/article/svchost-dll-service.html
// 2. Inside Win32 Services, Part 2 by: Mark Russinovich, at: http://www.winnetmag.com/Articles/Index.cfm?ArticleID=8943&pg=3
// 3. Platform SDK: Tools - Rundll32, at: http://msdn.microsoft.com/library/en-us/tools/tools/rundll32.asp
#include <stdio.h>
#include <time.h>
#include <assert.h>
#include <windows.h>
#define DEFAULT_SERVICE "IPRIP"
#define MY_EXECUTE_NAME "SvcHostDLL.exe"
//main service process function void __stdcall ServiceMain( int argc, wchar_t* argv[] );
//report service stat to the service control manager int TellSCM( DWORD dwState, DWORD dwExitCode, DWORD dwProgress );
//service control handler, call back by service control manager void __stdcall ServiceHandler( DWORD dwCommand );
//RealService just create a process int RealService(char *cmd, int bInteract);
//Install this dll as a Service host by svchost.exe, service name is given by caller int InstallService(char *name);
//unInstall a Service, be CARE FOR call this to delete a service int UninstallService(char *name);
//Install this dll as a Service host by svchost.exe, used by RUNDLL32.EXE to call void CALLBACK RundllInstallA(HWND hwnd, HINSTANCE hinst, char *param, int nCmdShow);
//unInstall a Service used by RUNDLL32.EXE to call, be CARE FOR call this to delete a service void CALLBACK RundllUninstallA(HWND hwnd, HINSTANCE hinst, char *param, int nCmdShow);
//output the debug infor into log file(or stderr if a console program call me) & DbgPrint void OutputString( char *lpFmt, ... );
//dll module handle used to get dll path in InstallService HANDLE hDll = NULL;
//Service HANDLE & STATUS used to get service state SERVICE_STATUS_HANDLE hSrv;
DWORD dwCurrState;
BOOL APIENTRY DllMain( HANDLE hModule,
DWORD ul_reason_for_call,
LPVOID lpReserved
  )
{
 switch (ul_reason_for_call)
 {
 case DLL_PROCESS_ATTACH:
hDll = hModule;
#ifdef _DEBUG
AllocConsole();
OutputString("SvcHostDLL: DllMain called DLL_PROCESS_ATTACH");
break;
 case DLL_THREAD_ATTACH:
OutputString("SvcHostDLL: DllMain called DLL_THREAD_ATTACH");
 case DLL_THREAD_DETACH:
OutputString("SvcHostDLL: DllMain called DLL_THREAD_DETACH");
 case DLL_PROCESS_DETACH:
TellSCM( SERVICE_STOP_PENDING, 0, 0 );
Sleep(1500);
TellSCM( SERVICE_STOPPED, 0, 0 );
OutputString("SvcHostDLL: DllMain called DLL_PROCESS_DETACH");
#endif
break;
 }
 return TRUE;
}
void __stdcall ServiceMain( int argc, wchar_t* argv[] )
{
// DebugBreak();
 char svcname[256];
 strncpy(svcname, (char*)argv[0], sizeof svcname); //it's should be unicode, but if it's ansi we do it well
 wcstombs(svcname, argv[0], sizeof svcname);
 OutputString("SvcHostDLL: ServiceMain(%d, %s) called", argc, svcname);
 hSrv = RegisterServiceCtrlHandler( svcname, (LPHANDLER_FUNCTION)ServiceHandler );
 if( hSrv == NULL )
 {
OutputString("SvcHostDLL: RegisterServiceCtrlHandler %S failed", argv[0]);
return;
 }else FreeConsole();
 TellSCM( SERVICE_START_PENDING, 0, 1 );
 TellSCM( SERVICE_RUNNING, 0, 0 );
 // call Real Service function noew
 if(argc > 1)
strncpy(svcname, (char*)argv[1], sizeof svcname),
wcstombs(svcname, argv[1], sizeof svcname);
 RealService(argc > 1 ? svcname : MY_EXECUTE_NAME, argc > 2 ? 1 : 0);
 do{
Sleep(10);//not quit until receive stop command, otherwise the service will stop
 }while(dwCurrState != SERVICE_STOP_PENDING && dwCurrState != SERVICE_STOPPED);
 OutputString("SvcHostDLL: ServiceMain done");
 return;
}
int TellSCM( DWORD dwState, DWORD dwExitCode, DWORD dwProgress )
{
 SERVICE_STATUS srvStatus;
 srvStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
 srvStatus.dwCurrentState = dwCurrState = dwState;
 srvStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE | SERVICE_ACCEPT_SHUTDOWN;
 srvStatus.dwWin32ExitCode = dwExitCode;
 srvStatus.dwServiceSpecificExitCode = 0;
 srvStatus.dwCheckPoint = dwProgress;
 srvStatus.dwWaitHint = 3000;
 return SetServiceStatus( hSrv, &srvStatus );
}
void __stdcall ServiceHandler( DWORD dwCommand )
{
 // not really necessary because the service stops quickly
 switch( dwCommand )
 {
 case SERVICE_CONTROL_STOP:
TellSCM( SERVICE_STOP_PENDING, 0, 1 );
OutputString("SvcHostDLL: ServiceHandler called SERVICE_CONTROL_STOP");
Sleep(10);
TellSCM( SERVICE_STOPPED, 0, 0 );
break;
 case SERVICE_CONTROL_PAUSE:
TellSCM( SERVICE_PAUSE_PENDING, 0, 1 );
OutputString("SvcHostDLL: ServiceHandler called SERVICE_CONTROL_PAUSE");
TellSCM( SERVICE_PAUSED, 0, 0 );
break;
 case SERVICE_CONTROL_CONTINUE:
TellSCM( SERVICE_CONTINUE_PENDING, 0, 1 );
OutputString("SvcHostDLL: ServiceHandler called SERVICE_CONTROL_CONTINUE");
TellSCM( SERVICE_RUNNING, 0, 0 );
break;
 case SERVICE_CONTROL_INTERROGATE:
OutputString("SvcHostDLL: ServiceHandler called SERVICE_CONTROL_INTERROGATE");
TellSCM( dwCurrState, 0, 0 );
break;
 case SERVICE_CONTROL_SHUTDOWN:
OutputString("SvcHostDLL: ServiceHandler called SERVICE_CONTROL_SHUTDOWN");
TellSCM( SERVICE_STOPPED, 0, 0 );
break;
 }
}
//RealService just create a process
int RealService(char *cmd, int bInteract)
{
 OutputString("SvcHostDLL: RealService called '%s' %s", cmd, bInteract ? "Interact" : "");
 STARTUPINFO si = {0};
 PROCESS_INFORMATION pi;
 si.cb = sizeof si;
 if(bInteract) si.lpDesktop = "WinSta0\\Default";
 if(!CreateProcess(NULL, cmd, NULL, NULL, false, 0, NULL, NULL, &si, &pi))
OutputString("SvcHostDLL: CreateProcess(%s) error:%d", cmd, GetLastError());
 else OutputString("SvcHostDLL: CreateProcess(%s) to %d", cmd, pi.dwProcessId);
 return 0;
}
int InstallService(char *name)
{
 // Open a handle to the SC Manager database.
 int rc = 0;
 HKEY hkRoot = HKEY_LOCAL_MACHINE, hkParam = 0;
 SC_HANDLE hscm = NULL, schService = NULL;
 try{
 char buff[500];
 char *svcname = DEFAULT_SERVICE;
 if(name && name[0]) svcname = name;
 //query svchost setting
 char *ptr, *pSvchost = "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Svchost";
 rc = RegOpenKeyEx(hkRoot, pSvchost, 0, KEY_QUERY_VALUE, &hkRoot);
 if(ERROR_SUCCESS != rc)
 {
OutputString("RegOpenKeyEx(%s) KEY_QUERY_VALUE error %d.", pSvchost, rc);
throw "";
 }
 DWORD type, size = sizeof buff;
 rc = RegQueryValueEx(hkRoot, "netsvcs", 0, &type, (unsigned char*)buff, &size);
 RegCloseKey(hkRoot);
 SetLastError(rc);
 if(ERROR_SUCCESS != rc)
throw "RegQueryValueEx(Svchost\\netsvcs)";
 for(ptr = buff; *ptr; ptr = strchr(ptr, 0)+1)
if(stricmp(ptr, svcname) == 0) break;
 if(*ptr == 0)
 {
OutputString("you specify service name not in Svchost\\netsvcs, must be one of following:");
for(ptr = buff; *ptr; ptr = strchr(ptr, 0)+1)
 OutputString(" - %s", ptr);
throw "";
 }
 //install service
 hscm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
 if (hscm == NULL)
throw "OpenSCManager()";
 char *bin = "%SystemRoot%\\System32\\svchost.exe -k netsvcs";
 schService = CreateService(
hscm,// SCManager database
svcname, // name of service
NULL, // service name to display
SERVICE_ALL_ACCESS,// desired access
SERVICE_WIN32_SHARE_PROCESS, // service type
SERVICE_AUTO_START,  // start type
SERVICE_ERROR_NORMAL,  // error control type
bin,// service's binary
NULL,  // no load ordering group
NULL,  // no tag identifier
NULL,  // no dependencies
NULL,  // LocalSystem account
NULL);  // no password
 if (schService == NULL)
 {
OutputString("CreateService(%s) error %d", svcname, rc = GetLastError());
throw "";
 }
 OutputString("CreateService(%s) SUCCESS. Config it", svcname);
 CloseServiceHandle(schService);
 CloseServiceHandle(hscm);
 //config service
 hkRoot = HKEY_LOCAL_MACHINE;
 strncpy(buff, "SYSTEM\\CurrentControlSet\\Services\\", sizeof buff);
 strncat(buff, svcname, 100);
 rc = RegOpenKeyEx(hkRoot, buff, 0, KEY_ALL_ACCESS, &hkRoot);
 if(ERROR_SUCCESS != rc)
 {
OutputString("RegOpenKeyEx(%s) KEY_SET_VALUE error %d.", svcname, rc);
throw "";
 }
 rc = RegCreateKey(hkRoot, "Parameters", &hkParam);
 SetLastError(rc);
 if(ERROR_SUCCESS != rc)
throw "RegCreateKey(Parameters)";
 if(!GetModuleFileName(HMODULE(hDll), buff, sizeof buff))
throw "GetModuleFileName() get dll path";
 rc = RegSetValueEx(hkParam, "ServiceDll", 0, REG_EXPAND_SZ, (unsigned char*)buff, strlen(buff)+1);
 SetLastError(rc);
 if(ERROR_SUCCESS != rc)
throw "RegSetValueEx(ServiceDll)";
 OutputString("Config service %s ok.", svcname);
 }catch(char *str)
 {
if(str && str[0])
{
 rc = GetLastError();
 OutputString("%s error %d", str, rc);
}
 }
 RegCloseKey(hkRoot);
 RegCloseKey(hkParam);
 CloseServiceHandle(schService);
 CloseServiceHandle(hscm);
 return rc;
}
/*
used to install by rundll32.exe
Platform SDK: Tools - Rundll32
The Run DLL utility (Rundll32.exe) included in Windows enables you to call functions exported from a 32-bit DLL. These functions must have the following syntax:
*/
void CALLBACK RundllInstallA(
HWND hwnd,// handle to owner window
HINSTANCE hinst, // instance handle for the DLL
char *param,// string the DLL will parse
int nCmdShow  // show state
)
{
 InstallService(param);
}
int UninstallService(char *name)
{
 int rc = 0;
 SC_HANDLE schService;
 SC_HANDLE hscm;
 __try{
 hscm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
 if (hscm == NULL)
 {
OutputString("OpenSCManager() error %d", rc = GetLastError() );
return rc;
 }
 char *svcname = DEFAULT_SERVICE;
 if(name && name[0]) svcname = name;
 schService = OpenService(hscm, svcname, DELETE);
 if (schService == NULL)
 {
OutputString("OpenService(%s) error %d", svcname, rc = GetLastError() );
return rc;
 }
 if (!DeleteService(schService) )
 {
OutputString("OpenService(%s) error %d", svcname, rc = GetLastError() );
return rc;
 }
 OutputString("DeleteService(%s) SUCCESS.", svcname);
 }__except(1)
 {
OutputString("Exception Catched 0x%X", GetExceptionCode());
 }
 CloseServiceHandle(schService);
 CloseServiceHandle(hscm);
 return rc;
}
/*
used to uninstall by rundll32.exe
Platform SDK: Tools - Rundll32
The Run DLL utility (Rundll32.exe) included in Windows enables you to call functions exported from a 32-bit DLL. These functions must have the following syntax:
*/
void CALLBACK RundllUninstallA(
HWND hwnd,// handle to owner window
HINSTANCE hinst, // instance handle for the DLL
char *param,// string the DLL will parse
int nCmdShow  // show state
)
{
 UninstallService(param);
}
//output the debug infor into log file & DbgPrint
void OutputString( char *lpFmt, ... )
{
 char buff[1024];
 va_list arglist;
 va_start( arglist, lpFmt );
 _vsnprintf( buff, sizeof buff, lpFmt, arglist );
 va_end( arglist );
 DWORD len;
 HANDLE herr = GetStdHandle(STD_OUTPUT_HANDLE);
 if(herr != INVALID_HANDLE_VALUE)
 {
WriteFile(herr, buff, strlen(buff), &len, NULL);
WriteFile(herr, "\r\n", 2, &len, NULL);
 }else
 {
FILE *fp = fopen("SvcHost.DLL.log", "a");
if(fp)
{
 char date[20], time[20];
 fprintf(fp, "%s %s - %s\n", _strdate(date), _strtime(time), buff);
 if(!stderr) fclose(fp);
}
 }
 OutputDebugString(buff);
}

相關文章