cNTServiceImpl.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cNTServiceImpl_H
#define _INC_cNTServiceImpl_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cNTService.h"
 
#if defined(_WIN32) && !defined(UNDER_CE) && ! defined(_MFC_VER)
#include "../Thread/cSysEvent.h"
#include "GrayCore/include/cSingleton.h"
#include "GrayCore/include/cAppState.h"
 
namespace GrayLib
{
        enum NTS_CMD_TYPE
        {
#define CNTSERVICECMD(a,b,c) NTS_CMD_##a,
#include "cNTServiceCmd.tbl"
#undef CNTSERVICECMD
                NTS_CMD_QTY,
        };
 
        class GRAYLIB_LINK cNTServiceImpl
                : public cSingletonStatic< cNTServiceImpl >
        {
 
                typedef cSingletonStatic< cNTServiceImpl > SUPER_t;
 
        public:
                static const FILECHAR_t* const k_szCommandLines[NTS_CMD_QTY + 1];
                const FILECHAR_t* m_pszNameSymbolic;    
                const FILECHAR_t* m_pszNameDesc;                
 
        private:
                bool m_bIsNTService;    
                bool m_bDebugService;   
 
                cSysEvent m_StopEvent;          
                SERVICE_STATUS_HANDLE m_hStatusHandle; 
                SERVICE_STATUS m_Status;        
 
        private:
                HRESULT ServiceStart();
                void ServiceMainStart();
 
                // Our exported API.
                static void WINAPI ServiceHandler(DWORD dwCtrlCode);
                static void WINAPI ServiceMain(DWORD argc, LPSTR* lpszArgv);
 
        protected:
                // command line entries.
                HRESULT CmdServiceInstall();
                HRESULT CmdServiceRemove();
                HRESULT CmdServiceDebug();
                HRESULT CmdServiceRun();
 
                HRESULT CmdService();
 
        public:
                cNTServiceImpl( const FILECHAR_t* pszName, const FILECHAR_t* pszNameDesc = nullptr );
                virtual ~cNTServiceImpl();
 
                bool isRunningAsService() const
                {
                        return m_bIsNTService;
                }
                StrLen_t MakeRegKey(FILECHAR_t* pszKey, const FILECHAR_t* pszExt = nullptr) const;
 
                void ReportEventX(WORD wType, DWORD dwMsgID, const GChar_t* lpszMsg, const GChar_t* lpszArgs = nullptr);
                bool SetServiceStatus(DWORD dwCurrentState, LSTATUS dwWin32ExitCode, TIMESYSD_t nTimeHint = cTimeSys::k_FREQ);
                void ServiceStop();
 
                // ASSUME called from external source.
                bool onTickService();
 
                static cString GRAYCALL get_HelpText();
                static bool GRAYCALL HasServiceCommandLine();
                APP_EXITCODE_t Main();
        };
}
 
#ifdef GRAY_DLL // force implementation/instantiate for DLL/SO.
namespace Gray
{
        template class GRAYLIB_LINK cSingletonStatic< GrayLib::cNTServiceImpl >;
}
#endif
 
#endif // ! _MFC_VER
#endif // _INC_cNTServiceImpl_H