diku_mcs/moteserver.cc

   1 #include "FileDescriptor.h"
   2 #include "SessionListener.h"
   3 #include "HostListener.h"
   4 #include "database.h"
   5 #include "Configuration.h"
   6 #include "MMSException.h"
   7 #include "macros.h"
   8 #include <exception>
   9 #include <unistd.h>
  10 #include "libutil/File.h"
  11
  12 using namespace remote;
  13 using namespace remote::diku_mcs;
  14 using namespace remote::util;
  15
  16 void handleExit()
  17 {
  18         log("Shutting down\n");
  19         remove(Configuration::vm["pidFile"].as<std::string>().c_str());
  20 }
  21
  22 int main(int argc,char** argv)
  23 {
  24         std::ostringstream oss;
  25         std::string pid;
  26
  27         Configuration::read(argc,argv);
  28
  29         if (Configuration::vm["daemonize"].as<int>())
  30         {
  31                 printf("Daemonizing!\n");
  32                 if (fork()) exit(0);
  33                 setsid();
  34                 fclose(stdin);
  35                 close(1);
  36                 // reopen stdout
  37                 if (NULL == freopen(Configuration::vm["log-file"].as<std::string>().c_str(),"a",stdout))
  38                 {
  39                         fprintf(stderr,"Unable to open file %s for logging! Bailing out!\n",Configuration::vm["log-file"].as<std::string>().c_str());
  40                         return -1;
  41                 }
  42                 if (NULL == freopen(Configuration::vm["errorlog-file"].as<std::string>().c_str(),"a",stderr))
  43                 {
  44                         return -1;
  45                 }
  46         }
  47
  48         log("Starting mote server\n");
  49
  50         atexit(handleExit);
  51
  52         oss << getpid() << std::endl;
  53         pid = oss.str();
  54
  55         if (!File::writeFile(Configuration::vm["pidFile"].as<std::string>(),
  56                              pid.c_str(), pid.size()))
  57                 log("Failed to create .pid file\n");
  58
  59         do {
  60                         dbConn.connect( Configuration::vm["dbName"].as<std::string>(),
  61                                         Configuration::vm["dbHost"].as<std::string>(),
  62                                         Configuration::vm["dbUser"].as<std::string>(),
  63                                         Configuration::vm["dbPassword"].as<std::string>());
  64                         log("Connected to database\n");
  65                         Mote::resetDb();
  66                         log("Deleted old mote data\n");
  67                         Session::resetDb();
  68                         log("Deleted old session data\n");
  69                         HostListener hostListener(Configuration::vm["hostListenerPort"].as<unsigned int>());
  70                         SessionListener sessionListener(Configuration::vm["sessionListenerPort"].as<unsigned int>());
  71                         log("Entering service loop\n");
  72
  73         } while (FileDescriptor::serviceLoop());
  74 }