Daemon.h

// $Id: Daemon.h 21 2010-09-05 04:18:17Z cschwarz1 $

#ifndef BASE_DAEMON_H
#define BASE_DAEMON_H

#include <iostream>
#include <ctime>
#include "Config.h"
#include "Global.h"
#include "Log.h"
#include "Mutex.h"
#include "Process.h"
#include "Socket.h"

template<class T> int daemon_main(int argc, const char **argv, const char *domain = NULL) {
    int ret;

    if (domain && *domain)
        base::Global::setDomain(domain);
    else
        base::Global::setDomainFromExecutable(*argv);
    try {
        T daemon(*argv);
        base::Socket::init();
        if (daemon.readArguments(argc, argv))
            ret = daemon.service(argv);
        else
            ret = 1;
        base::Socket::finit();
    } catch (const base::Exception &ex) {
        std::cerr << *argv << ": " << ex << std::endl;
        ret = 4;
    } catch (const std::exception &ex) {
        std::cerr << *argv << ": " << (ex.what() ? ex.what() : "(null)") << std::endl;
        ret = 4;
    }
    return ret;
}

namespace base {

    class PPBASE_EXPORT Daemon: public Mutex {
    public:
        virtual ~Daemon();


        Log *getLog();

        Config *getConfig();


        bool getForeground() const;


        const String &getName() const;


        Process::pid_type getPid() const;


        time_t getTime() const;


        void getUptime(String &s);


        void log(Log::levels level, const char *func, const char *fmt, ...)
        #ifdef __GNUC__
            __attribute__((format(printf, 4, 5)))
        #endif
            ;


        bool readArguments(int argc, const char **argv, const char *args =
        #ifdef _WIN32
            "c:d:fhqsvV"
        #else
            "c:d:fhqsvVw"
        #endif
        );


        int service(const char **argv);

protected:
    #ifdef _WIN32
        enum srvE {
            srvNone = 0,
            srvInstall,
            srvStart,
            srvStop,
            srvUninstall
        };
    #endif


        Daemon(const String &name);

        virtual void config();

        virtual void createThreads(bool init);

    #ifndef _WIN32


        Process::pid_type daemonize(bool createpid = true, bool exitparent = true);

        void dropRootPrivileges();
    #endif


        virtual bool dumpState(bool force);

    #ifndef _WIN32


        gid_t getGroupID() const;


        uid_t getUserID() const;
    #endif


        virtual bool handleArgument(int arg);


        bool hasExplicitUGID() const;


        virtual int loop() = 0;


        virtual bool needRootPrivileges();

        int serviceMain();

        virtual void shutdown();

        virtual void startup();

        virtual void terminateThreads();


        virtual void usage(const char *pgm, bool err) const;


        virtual void vlog(Log::levels level, const char *func, const char *fmt, va_list ap);

    #ifdef _WIN32
        int win32Service();
        static void WINAPI win32ServiceHandler(DWORD opcode);
        void win32ServiceMain(DWORD, LPWSTR *);
        static void WINAPI win32ServiceMainS(DWORD dw, LPWSTR *lpstr);
        void win32SetServiceStatus(DWORD status);
    #endif

        Config                 _cfg;              
        String                 _cfgfile;          
        bool                   _foreground;       
        u_longlong_t           _laststatedump;    
        Log                   *_log;              
        bool                   _logstderr;        
        int                    _maxopenfiles;     
        String                 _name;             
        Process::pid_type      _pid;              
        time_t                 _t_now;            
        time_t                 _t_start;          
        int                    _fdreserve;        

    #ifdef _WIN32
        enum srvE              _srvact;           
        WString                _srvname;          
        String                 _srvprefix;        
        SERVICE_STATUS_HANDLE  _srvsh;            
        SERVICE_STATUS         _srvst;            
    #else
        String                 _pidfile;          
        bool                   _watcher;          
    #endif
    };
}

#endif