cNTService.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cNTService_H
#define _INC_cNTService_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "../GrayLibBase.h"
 
#if defined(_WIN32) && !defined(UNDER_CE)
#include "GrayCore/include/cHandlePtr.h"
#include "GrayCore/include/cOSHandle.h" // HANDLE_NULL
#include "GrayCore/include/FileName.h"
#include <winsvc.h>
 
namespace Gray
{
        template <> inline void cHandlePtr<SC_HANDLE>::CloseHandle(SC_HANDLE h) // static
        {
                ::CloseServiceHandle(h); // ignored BOOL return.
        }
}
 
namespace GrayLib
{
        class GRAYLIB_LINK cNTServiceHandle : public cHandlePtr < SC_HANDLE >
        {
 
                typedef cHandlePtr<SC_HANDLE> SUPER_t;
        public:
                cNTServiceHandle(SC_HANDLE hSC = HANDLE_NULL)
                : cHandlePtr<SC_HANDLE>(hSC)
                {
                }
                ~cNTServiceHandle()
                {
                }
                bool QueryServiceConfigX(QUERY_SERVICE_CONFIG* pCfg, DWORD dwSizeCfgMax, DWORD& rdwSizeCfgRet) const
                {
                        if (!_FNF(::QueryServiceConfig)(get_Handle(), pCfg, dwSizeCfgMax, &rdwSizeCfgRet))
                                return false;
                        return true;
                }
                bool DeleteService()
                {
                        return ::DeleteService(get_Handle());
                }
                bool ControlService(DWORD dwControl = SERVICE_CONTROL_STOP, LPSERVICE_STATUS lpServiceStatus = nullptr)
                {
                        return ::ControlService(get_Handle(), dwControl, lpServiceStatus) ? true : false;
                }
                bool OpenServiceX(SC_HANDLE hSCMgr, const FILECHAR_t* pszServiceName, DWORD dwDesiredAccess = SERVICE_ALL_ACCESS)
                {
                        AttachHandle(_FNF(::OpenService)(hSCMgr, pszServiceName, dwDesiredAccess));
                        return(isValidHandle());
                }
        };
 
        class GRAYLIB_LINK cNTServiceMgr : public cNTServiceHandle 
        {
        public:
                cNTServiceMgr(const FILECHAR_t* lpMachineName = nullptr, const FILECHAR_t* lpDatabaseName = nullptr, DWORD dwDesiredAccess = SC_MANAGER_ALL_ACCESS)
                : cNTServiceHandle(_FNF(::OpenSCManager)(lpMachineName, lpDatabaseName, dwDesiredAccess))
                {
                        // call isValidHandle();
                }
                ~cNTServiceMgr()
                {
                }
                void OpenServiceMgr(const FILECHAR_t* lpMachineName = nullptr, const FILECHAR_t* lpDatabaseName = nullptr, DWORD dwDesiredAccess = SC_MANAGER_ALL_ACCESS)
                {
                        // call isValidHandle(); after
                        AttachHandle(_FNF(::OpenSCManager)(lpMachineName, lpDatabaseName, dwDesiredAccess));
                }
                SC_HANDLE OpenServiceX(const FILECHAR_t* pszServiceName, DWORD dwDesiredAccess = SERVICE_ALL_ACCESS) const
                {
                        return(_FNF(::OpenService)(get_Handle(), pszServiceName, dwDesiredAccess));
                }
                bool OpenServiceX(cNTServiceHandle& sh, const FILECHAR_t* pszServiceName, DWORD dwDesiredAccess = SERVICE_ALL_ACCESS) const
                {
                        sh.AttachHandle( OpenServiceX(pszServiceName, dwDesiredAccess));
                        return(sh.isValidHandle());
                }
        };
};
#endif // _WIN32
#endif // _INC_CNTSERVICE_H