´´½¨SvcHost.exe µ÷ÓõķþÎñÔ­ÀíÓëʵ¼ù(5)

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;
}