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

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