Utils/DaemonTools.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2006 Stefan Bund <g0dil@senf.berlios.de>
   4 //
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation; either version 2 of the License, or
   8 // (at your option) any later version.
   9 //
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14 //
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the
  17 // Free Software Foundation, Inc.,
  18 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19
  20 /** \file
  21     \brief DaemonTools  non-inline non-template implementation */
  22
  23 #include "DaemonTools.hh"
  24 //#include "DaemonTools.ih"
  25
  26 // Custom includes
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #include <unistd.h>
  30 #include <fcntl.h>
  31 #include <errno.h>
  32 #include "Exception.hh"
  33
  34 //#include "DaemonTools.mpp"
  35 #define prefix_
  36 ///////////////////////////////cc.p////////////////////////////////////////
  37
  38 ///////////////////////////////////////////////////////////////////////////
  39 // senf::Daemon
  40
  41 prefix_ senf::Daemon::~Daemon()
  42 {}
  43
  44 prefix_ void senf::Daemon::daemonize(bool v)
  45 {
  46     daemonize_ = v;
  47 }
  48
  49 prefix_ bool senf::Daemon::daemon()
  50 {
  51     return daemonize_;
  52 }
  53
  54 prefix_ void senf::Daemon::consoleLog(std::string path, StdStream which)
  55 {
  56     int fd (-1);
  57     if (! path.empty()) {
  58         int fd (::open(path.c_str(), O_WRONLY | O_APPEND));
  59         if (fd < 0)
  60             throwErrno("open()");
  61     }
  62     switch (which) {
  63     case StdOut:
  64         stdout_ = fd;
  65         break;
  66     case StdErr:
  67         stderr_ = fd;
  68         break;
  69     case Both:
  70         stdout_ = fd;
  71         stderr_ = fd;
  72         break;
  73     }
  74 }
  75
  76 prefix_ void senf::Daemon::pidFile(std::string f, bool unique)
  77 {
  78     pidfile_ = f;
  79     unique_ = unique;
  80 }
  81
  82 prefix_ void senf::Daemon::detach()
  83 {}
  84
  85 prefix_ int senf::Daemon::start(int argc, char const ** argv)
  86 {
  87     argc_ = argc;
  88     argv_ = argv;
  89
  90     configure();
  91     if (daemonize_)
  92         fork();
  93     if (! pidfile_.empty())
  94         pidfileCreate();
  95     main();
  96     return 0;
  97 }
  98
  99 ////////////////////////////////////////
 100 // protected members
 101
 102 prefix_ senf::Daemon::Daemon()
 103     : argc_(0), argv_(0), daemonize_(true), stdout_(-1), stderr_(-1), pidfile_(""), unique_(true),
 104       detached_(false)
 105 {}
 106
 107 ////////////////////////////////////////
 108 // private members
 109
 110 prefix_ void senf::Daemon::configure()
 111 {}
 112
 113 prefix_ void senf::Daemon::main()
 114 {
 115     init();
 116     detach();
 117     run();
 118 }
 119
 120 prefix_ void senf::Daemon::init()
 121 {}
 122
 123 prefix_ void senf::Daemon::run()
 124 {}
 125
 126 prefix_ void senf::Daemon::pidfileCreate()
 127 {}
 128
 129 ///////////////////////////////////////////////////////////////////////////
 130
 131 prefix_ void senf::daemonize()
 132 {
 133     int pid = fork();
 134     if (pid < 0)
 135         throw senf::SystemException("fork",errno);
 136     if (pid > 0)
 137         ::_exit(0);
 138     if (::setsid() < 0)
 139         throw senf::SystemException("setsid",errno);
 140 }
 141
 142 prefix_ void senf::redirect_stdio(std::string const & path)
 143 {
 144     int fd = ::open(path.c_str(),O_RDWR);
 145     if (fd < 0) throw senf::SystemException("open",errno);
 146     if (dup2(fd,0) < 0) throw senf::SystemException("dup2",errno);
 147     if (dup2(fd,1) < 0) throw senf::SystemException("dup2",errno);
 148     if (dup2(fd,2) < 0) throw senf::SystemException("dup2",errno);
 149     if (::close(fd) < 0) throw senf::SystemException("close",errno);
 150 }
 151
 152 ///////////////////////////////cc.e////////////////////////////////////////
 153 #undef prefix_
 154 //#include "DaemonTools.mpp"
 155
 156 \f
 157 // Local Variables:
 158 // mode: c++
 159 // fill-column: 100
 160 // c-file-style: "senf"
 161 // indent-tabs-mode: nil
 162 // ispell-local-dictionary: "american"
 163 // compile-command: "scons -u test"
 164 // comment-column: 40
 165 // End: