Subversion Repositories HomeAutomation

/*

 *

 *  Copyright(C) 2010  Mattias Runge

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License along

 *  with this program; if not, write to the Free Software Foundation, Inc.,

 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 */

#include <signal.h>

#include <sys/resource.h>

#include <boost/program_options.hpp>

#include <boost/thread/mutex.hpp>

#include <boost/thread/condition.hpp>

#include <boost/thread/locks.hpp>

#include "config.h"

#include "logging/Logger.h" // Deprecated

#include "common/log.h"

#include "net/Manager.h"

#include "timer/Manager.h"

#include "config/Manager.h"

#include "can/Protocol.h"

#include "broker/Manager.h"

#include "control/Manager.h"

#include "can/Network.h"

#include "can/Monitor.h"

#include "can/CanDaemon.h"

#include "vm/Manager.h"

#include "storage/Manager.h"

#include "net/GetFile.h"

#include "net/Url.h"

#include "vm/plugin/System.h"

#include "vm/plugin/Timer.h"

#include "vm/plugin/Module.h"

#include "vm/plugin/Node.h"

#include "vm/plugin/Console.h"

#include "vm/plugin/Storage.h"

#include "vm/plugin/Socket.h"

#ifdef USE_PLUGIN_XORG

#include "vm/plugin/Xorg.h"

#endif // USE_PLUGIN_XORG

#ifdef USE_PLUGIN_MYSQL

#include "vm/plugin/MySql.h"

#endif // USE_PLUGIN_MYSQL

using namespace atom;

logging::Logger LOG("Main"); // Deprecated

static const std::string log_module_ = "main";

std::vector<broker::Subscriber::Pointer> subscribers;

boost::condition on_message_condition;

boost::mutex guard_mutex;

net::GetFile::Pointer downloader;

void ContinueInitialization(std::string protocol_filename);

void Handler(int status);

void CleanUp();

void HandleProtocolFileDownload(bool success, net::Url url, std::string temporary_filepath)

{

  if (success)

  {

    log::Info(log_module_, "Successfully downloaded protocol file!");

    ContinueInitialization(temporary_filepath);

  }

  else

  {

    ContinueInitialization(url.GetRaw());

  }

}

int main(int argc, char **argv)

{

  /* Set core dump parameters */

  rlimit core_limit = { RLIM_INFINITY, RLIM_INFINITY };

  setrlimit( RLIMIT_CORE, &core_limit ); // enable core dumps

  /* Print startup information */

  log::Info(log_module_, "Atom daemon, version %s starting...", VERSION);

  log::Info(log_module_, "Released under %s.", LICENSE);

  log::Info(log_module_, "Written by Mattias Runge 2010-2012.");

  /* Register signal handlers */

  signal(SIGTERM, Handler);

  signal(SIGINT,  Handler);

  signal(SIGQUIT, Handler);

  signal(SIGABRT, Handler);

  signal(SIGPIPE, Handler);

  /* Check configuration */

  config::Manager::Create();

  if (!config::Manager::Instance()->Set(argc, argv))

  {

    CleanUp();

    return EXIT_SUCCESS;

  }

  /* Set up logging */

  if (config::Manager::Instance()->Exist("LogFile"))

  {

    log::OpenFile(config::Manager::Instance()->GetAsString("LogFile"));

  }

  if (config::Manager::Instance()->Exist("LogLevelMask"))

  {

    log::SetLevelByString(config::Manager::Instance()->GetAsString("LogLevelMask"));

  }

  /* Enter daemon mode if requested */

  if (config::Manager::Instance()->Exist("daemon"))

  {

    LOG.Info("Entering daemon mode...");

    if (daemon(0, 0) == -1)

    {

      log::Error(log_module_, "Could not enter daemon mode. Exiting...");

      CleanUp();

      return EXIT_FAILURE;

    }

    log::Info(log_module_, "Deamon mode entered successfully!");

  }

  /* Create singletons */

  storage::Manager::Create();

  timer::Manager::Create();

  broker::Manager::Create();

  net::Manager::Create();

  can::Protocol::Create();

  control::Manager::Create();

  vm::Manager::Create();

  std::string protocol_filename = config::Manager::Instance()->GetAsString("ProtocolFile");

  try

  {

    net::Url url(protocol_filename);

    downloader = net::GetFile::Pointer(new net::GetFile());

    log::Info(log_module_, "Will try to download protocol file from \"%s\"...", protocol_filename.data());

    downloader->Start(url, HandleProtocolFileDownload);

  }

  catch (std::exception& exception)

  {

    log::Debug(log_module_, "Could not parse ProtocolFile as a URL, \"%s\".", protocol_filename.data());

    ContinueInitialization(protocol_filename);

  }

  /* Lock main thread and wait for exit */

  boost::mutex::scoped_lock guard(guard_mutex);

  on_message_condition.wait(guard);

  /* Cleanup */

  log::Info(log_module_, "Cleaning up...");

  CleanUp();

  log::Info(log_module_, "Thank you for using Atom. Goodbye!");

  return EXIT_SUCCESS;

}

void ContinueInitialization(std::string protocol_filename)

{  

  /* Load protocol file */

  if (!can::Protocol::Instance()->Load(protocol_filename))

  {

    log::Error(log_module_, "Failed to load %s!", protocol_filename.data());

    on_message_condition.notify_all();

  }

  /* Set up storage path */

  storage::Manager::Instance()->SetRootPath(config::Manager::Instance()->GetAsString("StoragePath"));

  /* Start monitor server */

  if (config::Manager::Instance()->Exist("MonitorPort"))

  {

    subscribers.push_back(can::Monitor::Pointer(new can::Monitor(config::Manager::Instance()->GetAsInt("MonitorPort"))));

  }

  /* Start CanDaemon server */

  if (config::Manager::Instance()->Exist("DaemonPort"))

  {

    subscribers.push_back(can::CanDaemon::Pointer(new can::CanDaemon(config::Manager::Instance()->GetAsInt("DaemonPort"))));

  }

  /* Subscribe Can control manager to data */

  subscribers.push_back(control::Manager::Instance());

  /* Start VM plugins */

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::System(vm::Manager::Instance()->GetIoService())));

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Console(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("CommandPort"))));

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Storage(vm::Manager::Instance()->GetIoService())));

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Timer(vm::Manager::Instance()->GetIoService())));

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Module(vm::Manager::Instance()->GetIoService())));

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Node(vm::Manager::Instance()->GetIoService())));

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Socket(vm::Manager::Instance()->GetIoService())));

#ifdef USE_PLUGIN_XORG

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Xorg(vm::Manager::Instance()->GetIoService())));

#endif // USE_PLUGIN_XORG

#ifdef USE_PLUGIN_MYSQL

  vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::MySql(vm::Manager::Instance()->GetIoService())));

#endif // USE_PLUGIN_MYSQL

  /* Start VM */

  vm::Manager::Instance()->Start(config::Manager::Instance()->GetAsString("ScriptPath"), config::Manager::Instance()->GetAsString("UserScriptPath"));

  /* Connect to Can network */

  if (config::Manager::Instance()->Exist("CanNet"))

  {

    common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector("CanNet");

    for (uint n = 0; n < cannetworks.size(); n++)

    {

      subscribers.push_back(can::Network::Pointer(new can::Network(cannetworks[n])));

    }

  }

}

void CleanUp()

{

  downloader.reset();

  if (net::Manager::Instance().use_count() > 0)

  {

    net::Manager::Instance()->Stop();

  }

  subscribers.clear();

  config::Manager::Delete();

  timer::Manager::Delete();

  control::Manager::Delete();

  can::Protocol::Delete();

  broker::Manager::Delete();

  vm::Manager::Delete();

  net::Manager::Delete();

  storage::Manager::Delete();

  log::CloseFile();

}

void Handler(int status)

{

  std::string signal_name = "Unknown";

  switch (status)

  {

    case SIGTERM:

    {

      signal_name = "Terminate";

      break;

    }

    case SIGINT:

    {

      signal_name = "Interrupt";

      break;

    }

    case SIGQUIT:

    {

      signal_name = "Quit";

      break;

    }

    case SIGABRT:

    {

      signal_name = "Abort";

      break;

    }

    case SIGIO:

    {

      signal_name = "I/O";

      break;

    }

    case SIGPIPE:

    {

      signal_name = "Pipe";

      break;

    }

  }

  log::Info(log_module_, "Received signal %s (%d).", signal_name.c_str(), status);

  if (status != SIGPIPE)

  {

    on_message_condition.notify_all();

  }

}