Beesnest/Windows/DcDemon.cpp

Go to the documentation of this file.

/*
Copyright 2007 Erez Bibi (erezbibi@users.sourceforge.net)
This file is part of Beesnest.

Beesnest is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

Beesnest is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Beesnest; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/


/**
 * DcDemon.cpp: implementation of the CdcDemon class.
 * for documentation see the header file.
 *
 * The code here is based on:
 * http://www.codersource.net/win32_nt_service.html
 * http://www.devx.com/cplus/Article/9857
 */

#include "DcDemon.h"

/* Help function to handle calls from Windows. */
void ControlHandler (DWORD request)
{
    switch(request)
    {
        case SERVICE_CONTROL_STOP:
        case SERVICE_CONTROL_SHUTDOWN:
            CdcDemon::Stop (0);
            break;
    }
    return;
}


SERVICE_STATUS CdcDemon::ServiceStatus;
SERVICE_STATUS_HANDLE CdcDemon::hStatus;
char* CdcDemon::gsServiceName = "beesnest";

void CdcDemon::Install ()
{
    SC_HANDLE serviceControlManager = OpenSCManager (0, 0,
        SC_MANAGER_CREATE_SERVICE);

    if (serviceControlManager)
    {
        char path[ _MAX_PATH + 1 ];
        if (GetModuleFileName (0, path, sizeof (path) / sizeof (path[0])) > 0)
        {
            SC_HANDLE service = CreateService (serviceControlManager,
                gsServiceName, gsServiceName,
                SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
                SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL, path,
                0, 0, 0, 0, 0);
            if (service)
            {
                CloseServiceHandle (service);
                cout<<gsServiceName<<" Installed Successfully\n";
            }
            else
            {
                if (GetLastError () == ERROR_SERVICE_EXISTS)
                    cout<<gsServiceName<<" Already Exists.\n";
                else
                    cout<<gsServiceName
                        <<" Was not Installed Successfully. Error Code "
                        <<GetLastError()<<endl;
            }
        }
        CloseServiceHandle (serviceControlManager);
    }
}

void CdcDemon::Uninstall ()
{
    SC_HANDLE serviceControlManager = OpenSCManager (0, 0,
        SC_MANAGER_CONNECT);

    if (serviceControlManager)
    {
        SC_HANDLE service = OpenService (serviceControlManager,
            gsServiceName, SERVICE_QUERY_STATUS | DELETE);
        if (service)
        {
            SERVICE_STATUS serviceStatus;
            if ( QueryServiceStatus (service, &serviceStatus))
                if (serviceStatus.dwCurrentState == SERVICE_STOPPED)
                {
                    if (DeleteService (service))
                        cout<<gsServiceName<<" Removed Successfully\n";
                    else
                    {
                        DWORD dwError;
                        dwError = GetLastError ();
                        if (dwError == ERROR_ACCESS_DENIED)
                            cout<<"Access Denied While trying to Remove "
                                <<gsServiceName<<endl;
                        else if (dwError == ERROR_INVALID_HANDLE)
                            cout<<"Handle invalid while trying to Remove "
                                <<gsServiceName<<endl;
                        else if (dwError == ERROR_SERVICE_MARKED_FOR_DELETE)
                            cout<<gsServiceName
                                <<" already marked for deletion\n";
                    }
                }
                else
                    cout<<gsServiceName<<" is still Running.\n";
            CloseServiceHandle (service);
        }
        CloseServiceHandle (serviceControlManager);
    }
}

bool CdcDemon::StartPending ()
{
    ServiceStatus.dwServiceType = SERVICE_WIN32;
    ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
    ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP |
        SERVICE_ACCEPT_SHUTDOWN;
    ServiceStatus.dwWin32ExitCode = 0;
    ServiceStatus.dwServiceSpecificExitCode = 0;
    ServiceStatus.dwCheckPoint = 0;
    ServiceStatus.dwWaitHint = 0;

    hStatus = RegisterServiceCtrlHandler (gsServiceName,
        (LPHANDLER_FUNCTION)ControlHandler);
    if (hStatus == (SERVICE_STATUS_HANDLE)0)
        return false;

    SetServiceStatus(hStatus, &ServiceStatus);
    return true;
}

bool CdcDemon::MarkRun ()
{
    ServiceStatus.dwCurrentState = SERVICE_RUNNING;
    SetServiceStatus(hStatus, &ServiceStatus);
    return true;
}

bool CdcDemon::Stop (int code)
{
    ServiceStatus.dwCurrentState = SERVICE_STOPPED;
    ServiceStatus.dwControlsAccepted &= ~(SERVICE_ACCEPT_STOP |
            SERVICE_ACCEPT_SHUTDOWN);
    ServiceStatus.dwWin32ExitCode = code;
    SetServiceStatus(hStatus, &ServiceStatus);
    return true;
}


void CdcDemon::RunMain ()
{
    SERVICE_TABLE_ENTRY ServiceTable[2];
    ServiceTable[0].lpServiceName = gsServiceName;
    ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ProgramMain;
    ServiceTable[1].lpServiceName = NULL;
    ServiceTable[1].lpServiceProc = NULL;
    /* Start the control dispatcher thread for the service. */
    StartServiceCtrlDispatcher (ServiceTable);
}