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 #   ifdef NDEBUG
  91     try {
  92 #   endif
  93
  94     configure();
  95     if (daemonize_)
  96         fork();
  97     if (! pidfile_.empty())
  98         pidfileCreate();
  99     main();
 100
 101 #   ifdef NDEBUG
 102     }
 103     catch (std::exception & e) {
 104         std::cerr << "\n*** Fatal exception: " << e.what() << std::endl;
 105         return 1;
 106     }
 107     catch (...) {
 108         std::cerr << "\n*** Fatal exception: (unknown)" << std::endl;
 109         return 1;
 110     }
 111 #   endif
 112
 113     return 0;
 114 }
 115
 116 ////////////////////////////////////////
 117 // protected members
 118
 119 prefix_ senf::Daemon::Daemon()
 120     : argc_(0), argv_(0), daemonize_(true), stdout_(-1), stderr_(-1), pidfile_(""), unique_(true),
 121       detached_(false)
 122 {}
 123
 124 ////////////////////////////////////////
 125 // private members
 126
 127 prefix_ void senf::Daemon::configure()
 128 {}
 129
 130 prefix_ void senf::Daemon::main()
 131 {
 132     init();
 133     detach();
 134     run();
 135 }
 136
 137 prefix_ void senf::Daemon::init()
 138 {}
 139
 140 prefix_ void senf::Daemon::run()
 141 {}
 142
 143 prefix_ void senf::Daemon::fork()
 144 {
 145
 146 }
 147
 148 prefix_ void senf::Daemon::pidfileCreate()
 149 {}
 150
 151 ///////////////////////////////////////////////////////////////////////////
 152
 153 prefix_ void senf::daemonize()
 154 {
 155     int pid = fork();
 156     if (pid < 0)
 157         throw senf::SystemException("fork",errno);
 158     if (pid > 0)
 159         ::_exit(0);
 160     if (::setsid() < 0)
 161         throw senf::SystemException("setsid",errno);
 162 }
 163
 164 prefix_ void senf::redirect_stdio(std::string const & path)
 165 {
 166     int fd = ::open(path.c_str(),O_RDWR);
 167     if (fd < 0) throw senf::SystemException("open",errno);
 168     if (dup2(fd,0) < 0) throw senf::SystemException("dup2",errno);
 169     if (dup2(fd,1) < 0) throw senf::SystemException("dup2",errno);
 170     if (dup2(fd,2) < 0) throw senf::SystemException("dup2",errno);
 171     if (::close(fd) < 0) throw senf::SystemException("close",errno);
 172 }
 173
 174 ///////////////////////////////cc.e////////////////////////////////////////
 175 #undef prefix_
 176 //#include "DaemonTools.mpp"
 177
 178 \f
 179 // Local Variables:
 180 // mode: c++
 181 // fill-column: 100
 182 // c-file-style: "senf"
 183 // indent-tabs-mode: nil
 184 // ispell-local-dictionary: "american"
 185 // compile-command: "scons -u test"
 186 // comment-column: 40
 187 // End: