Jo, jeg har lavet en lille console-app, der ser sådan ud:
// ServiceTest.cpp : Defines the entry point for the console application.
//
#include <windows.h>
#include <tchar.h>
#include <iostream>
#include "cService.h"
using namespace std;
extern cService g_Service;
int _tmain(int argc, _TCHAR* argv[])
{
int nResult = 0;
// Check if there is a second argument (first one is .exe filename)
if(argc >= 2)
{
string arg(argv[1]);
if(arg == "/install") // If the argument is "/install", create the service
{
cout << "Installing service" << endl;
string strParameters = " /start";
// Collect any additional parameters
for(int i=2; i<argc; ++i)
strParameters += string(" ") + argv[i];
nResult = g_Service.Create(argv[0] + strParameters);
}
else if(arg == "/uninstall") // If the argument is "/uninstall", delete the service
{
cout << "Uninstalling service" << endl;
nResult = g_Service.Delete();
}
else if(arg == "/start") // If the argument is "/start", start the service
{
g_Service.Dispatch();
}
}
if(nResult == 0) // Check for error
{
cout << "Done..." << endl;
}
else
{
// Get result text
LPTSTR lpMsgBuf;
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS, NULL, nResult, 0, (LPTSTR)&lpMsgBuf, 0, NULL);
// Show result
cout << lpMsgBuf << endl;
// Free the buffer.
LocalFree(lpMsgBuf);
}
return nResult;
}
Den bruger klassen cService, som du kan se her:
// cService.h: interface for the cService class.
//
//////////////////////////////////////////////////////////////////////
#ifndef SERVICE_H
#define SERVICE_H
#include <windows.h>
#include "winsvc.h"
#include <string>
using namespace std;
const LPSTR ServiceName = "DemoService";
const LPSTR ServiceDisplayName = "A little Demo Service";
class cService
{
public:
cService();
virtual ~cService();
DWORD Create(string strCmd);
DWORD Delete();
DWORD Start();
DWORD Stop();
void Dispatch();
void MainStart(DWORD argc, LPTSTR *argv);
void ControlHandler(DWORD Opcode);
private:
SERVICE_STATUS_HANDLE hdlStat;
HANDLE hTimer;
};
#endif // SERVICE_H
// cService.cpp: implementation of the cService class.
//
//////////////////////////////////////////////////////////////////////
#include "cService.h"
#include <fstream>
// global instance of Service
cService g_Service;
// this function must be in global namespace (Windows API callback)
void ApiServiceMainStarter(DWORD argc, LPTSTR *argv)
{
g_Service.MainStart(argc, argv);
}
// this function must be in global namespace (Windows API callback)
void ApiServiceControlHandler(DWORD Opcode)
{
g_Service.ControlHandler(Opcode);
}
cService::cService()
{
}
cService::~cService()
{
}
DWORD cService::Create(string strCmd)
{
SC_HANDLE hdlSCM = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE);
if(hdlSCM == 0)
return ::GetLastError();
SC_HANDLE hdlServ = CreateService(
hdlSCM, // SCManager database
ServiceName, // name of service
ServiceDisplayName, // service name to display
STANDARD_RIGHTS_REQUIRED, // desired access
SERVICE_WIN32_OWN_PROCESS, // service type
SERVICE_DEMAND_START, // start type
SERVICE_ERROR_NORMAL, // error control type
strCmd.c_str(), // service's binary Path name
0, // no load ordering group
0, // no tag identifier
0, // no dependencies
0, // LocalSystem account
0); // no password
DWORD Ret = 0;
if(!hdlServ)
Ret = ::GetLastError();
CloseServiceHandle(hdlServ);
return Ret;
}
DWORD cService::Delete()
{
SC_HANDLE hdlSCM = OpenSCManager(0, 0, STANDARD_RIGHTS_REQUIRED);
if(!hdlSCM)
return ::GetLastError();
SC_HANDLE hdlServ = OpenService(hdlSCM, ServiceName, DELETE);
DWORD Ret = 0;
if(!DeleteService(hdlServ))
Ret = ::GetLastError();
CloseServiceHandle(hdlServ);
return Ret;
}
DWORD cService::Start()
{
SC_HANDLE hdlSCM = OpenSCManager(0, 0, STANDARD_RIGHTS_REQUIRED);
if(!hdlSCM)
return ::GetLastError();
SC_HANDLE hdlServ = OpenService(hdlSCM, ServiceName, SERVICE_START);
DWORD Ret = 0;
if(!StartService(hdlServ, 0, 0))
Ret = ::GetLastError();
CloseServiceHandle(hdlServ);
return Ret;
}
DWORD cService::Stop()
{
SC_HANDLE hdlSCM = OpenSCManager(0, 0, STANDARD_RIGHTS_REQUIRED);
if(!hdlSCM)
return ::GetLastError();
SC_HANDLE hdlServ = OpenService(hdlSCM, ServiceName, SERVICE_STOP);
SERVICE_STATUS ServStat;
DWORD Ret = 0;
if(!ControlService(hdlServ, SERVICE_CONTROL_STOP, &ServStat))
Ret = ::GetLastError();
CloseServiceHandle(hdlServ);
return Ret;
}
// Service Main Function called by Windows
void cService::MainStart(DWORD argc, LPTSTR *argv)
{
hdlStat = ::RegisterServiceCtrlHandler(ServiceName, (LPHANDLER_FUNCTION)ApiServiceControlHandler);
if(!hdlStat)
{
int Err = ::GetLastError();
return;
}
// Set status Running
ControlHandler(0);
// Do the service work
while(true)
{
SYSTEMTIME Clock;
::GetLocalTime(&Clock);
ofstream file;
file.open("c:\\Service.txt", ios_base::out|ios_base::app);
file << Clock.wHour << ":" << Clock.wMinute << ":" << Clock.wSecond << endl;
file.close();
Sleep(5000);
}
}
// Start the service (in the Exe which is running as Service!!)
void cService::Dispatch()
{
SERVICE_TABLE_ENTRY DispTbl[] =
{
{ ServiceName, (LPSERVICE_MAIN_FUNCTION)ApiServiceMainStarter},
{ 0, 0}
};
if(!::StartServiceCtrlDispatcher(DispTbl)) // does not return until service stopped
{
int Err = ::GetLastError();
}
}
// Control Handler handles Start / Stop events
void cService::ControlHandler(DWORD Opcode)
{
SERVICE_STATUS strctStat;
strctStat.dwServiceType = SERVICE_WIN32;
strctStat.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
strctStat.dwWin32ExitCode = NO_ERROR;
strctStat.dwServiceSpecificExitCode = 0; // returned Status code
strctStat.dwCheckPoint = 0; // returned Error code
strctStat.dwWaitHint = 0;
if(Opcode == SERVICE_CONTROL_STOP)
strctStat.dwCurrentState = SERVICE_STOPPED;
else if(Opcode == SERVICE_CONTROL_SHUTDOWN)
strctStat.dwCurrentState = SERVICE_STOPPED;
else if(Opcode == SERVICE_CONTROL_INTERROGATE)
strctStat.dwCurrentState = SERVICE_RUNNING;
else
strctStat.dwCurrentState = SERVICE_RUNNING;
SetServiceStatus(hdlStat, &strctStat);
}
Ny bruger
Nybegynder
Opret
Preview
