path: root/trunk/CIAO/DAnCE/NodeManager/Node_Manager.cpp

// $Id$

#include "NodeManager_Impl.h"
#include "orbsvcs/CosNamingC.h"
#include "tao/IORTable/IORTable.h"
#include "ace/SString.h"
#include "ace/Read_Buffer.h"
#include "ace/Get_Opt.h"
#include "ace/OS_NS_stdio.h"
#include "ace/OS_NS_unistd.h"
#include "ace/os_include/os_netdb.h"
#include "ciao/Server_init.h"
#include "ciao/CIAO_common.h"

const char *ior_file_name_ = "nodedaemon.ior";
char *nodeapp_location_ = 0;
const char *pid_file_name_ = 0;
bool write_to_ior_ = false;
bool register_with_ns_ = false;
int nodeapp_loc_ = 0;
int spawn_delay = 1;

ACE_CString nodeapp_options_;

int
parse_args (int argc, char *argv[])
{
  ACE_Get_Opt get_opts (argc, argv, "o:c:m:s:d:na:p:z:");
  int c;

  while ((c = get_opts ()) != -1)
    {
      switch (c)
        {
          case 'z':
            nodeapp_options_ += " -ORBDebugLevel ";
            nodeapp_options_ += get_opts.opt_arg ();
            break;

          case 'o':  // Get the file name to write to.
            ior_file_name_ = get_opts.opt_arg ();
            write_to_ior_ = true;
            break;

          case 'c':  // Get the default svc.conf filename.
            nodeapp_options_ += " -ORBSvcConf ";
            nodeapp_options_ += get_opts.opt_arg ();
            break;

          case 's': // Get the location to spawn the NodeApplication.
            nodeapp_location_ = get_opts.opt_arg ();
            nodeapp_loc_ = 1;
            break;

          case 'a': // Nodeapplication arguments.
            nodeapp_options_ = get_opts.opt_arg ();
            break;

          case 'd': // Get the spawn delay argument.
            spawn_delay = ACE_OS::atoi (get_opts.opt_arg ());
            break;

          case 'n':
            register_with_ns_ = true;
            break;

          case 'p':
            pid_file_name_ = get_opts.opt_arg ();
            break;

          case '?':  // Display help for use of the server.
          default:
            ACE_ERROR_RETURN ((LM_ERROR,
                               "usage:  %s\n"
                               "-o <ior_output_file>\n"
                               "-c <svc.conf file>\n"
                               "-n <use naming service>\n"
                               "-s <NodeApplication executable path>\n"
                               "-a <arguments to NodeApplication>\n"
                               "-d <spawn delay for nodeapplication>\n"
                               "-p <pid file>\n"
                               "\n",
                               argv [0]),
                              -1);
             break;
        }
    }

  return 0;
}

int
write_IOR (const char* ior)
{
  FILE* ior_output_file_ =
    ACE_OS::fopen (ior_file_name_, "w");

  if (ior_output_file_)
    {
      ACE_OS::fprintf (ior_output_file_,
                       "%s",
                       ior);
      ACE_OS::fclose (ior_output_file_);
    }

  return 0;
}

void
write_pid (void)
{
  if (pid_file_name_ == 0)
    {
      return;
    }

  FILE* pid_file = ACE_OS::fopen (pid_file_name_, "w");

  if (pid_file)
    {
      ACE_OS::fprintf (pid_file,
                       "%i",
                       ACE_OS::getpid ());
      ACE_OS::fclose (pid_file);
    }
}

bool
register_with_ns (const char * name_context,
                  CORBA::ORB_ptr orb,
                  CIAO::NodeManagerDaemon_ptr obj)
{
  // Naming Service related operations
  CORBA::Object_var naming_context_object =
    orb->resolve_initial_references ("NameService");

  CosNaming::NamingContext_var naming_context =
    CosNaming::NamingContext::_narrow (naming_context_object.in ());

  // Initialize the Naming Sequence
  CosNaming::Name name (1);
  name.length (1);
  name[0].id = name_context;

  try
    {
      // Register the servant with the Naming Service
      naming_context->bind (name, obj);
    }
  catch (CosNaming::NamingContext::AlreadyBound &)
    {
      ACE_DEBUG ((LM_DEBUG, "Node_Manager.cpp: Name already bound, rebinding....\n"));
      naming_context->rebind (name, obj);
    }

  return true;
}

int
main (int argc, char *argv[])
{
  try
    {
      // Initialize orb
      // @@TODO: Add error checking. There is absoluteley none.
      CORBA::ORB_var orb = CORBA::ORB_init (argc,
                                            argv,
                                            "");

      CIAO::Server_init (orb.in ());

      if (parse_args (argc, argv) != 0)
        return -1;

      // Check if NodeApplication executable location has been
      // specified
      if (! nodeapp_loc_)
        {
          ACE_DEBUG ((LM_DEBUG, "Use -s <NodeApplicationPath> to specify \
                                 executable path\n"));
          ACE_OS::exit (1);
        }

      // Get reference to Root POA.
      CORBA::Object_var obj
        = orb->resolve_initial_references ("RootPOA");

      PortableServer::POA_var poa
        = PortableServer::POA::_narrow (obj.in ());

      CORBA::Object_var table_object =
        orb->resolve_initial_references ("IORTable");

      IORTable::Table_var adapter =
        IORTable::Table::_narrow (table_object.in ());

      if (CORBA::is_nil (adapter.in ()))
        {
          ACE_ERROR_RETURN ((LM_ERROR, "Nil IORTable\n"), -1);
        }

      // Create and install the CIAO NodeManager servant
      CIAO::NodeManager_Impl *node_manager_servant = 0;
      ACE_NEW_RETURN (node_manager_servant,
                      CIAO::NodeManager_Impl("NodeManager",
                                             orb.in (),
                                             poa.in (),
                                             nodeapp_location_,
                                             nodeapp_options_.c_str (),
                                             spawn_delay),
                      -1);

      PortableServer::ServantBase_var safe_daemon (node_manager_servant);

      node_manager_servant->init ();

      // Implicit activation
      CIAO::NodeManagerDaemon_var node_manager =
        node_manager_servant->_this ();

      CORBA::String_var str =
        orb->object_to_string (node_manager.in ());

      adapter->bind ("NodeManager",
                     str.in ());

      if (write_to_ior_)
        {
          write_IOR (str.in ());
        }
      else if (register_with_ns_)
        {
          char name [MAXHOSTNAMELEN + 1];

          if (ACE_OS::hostname (name, MAXHOSTNAMELEN + 1) == -1)
            {
              ACE_ERROR ((LM_ERROR,
                          "gethostname call failed! \n"));
              ACE_OS::exit (1);
            }

          // Register this name with the Naming Service
          (void) register_with_ns (name,
                                   orb.in (),
                                   node_manager.in ());
        }

      ACE_DEBUG ((LM_INFO, "CIAO_NodeManager IOR: %s\n", str.in ()));

      // Activate POA manager
      PortableServer::POAManager_var mgr
        = poa->the_POAManager ();

      mgr->activate ();

      // Here start the Monitor
      /*
        MonitorController* monitor_controller
        = new MonitorController (orb);

        ACE_DEBUG ((LM_DEBUG , "Before Activate"));
        monitor_controller->activate ();
        ACE_DEBUG ((LM_DEBUG , "After Activate"));
      */

      // Finishing Deployment part
      ACE_DEBUG ((LM_DEBUG,
                  "CIAO_NodeManager is running...\n"));

      write_pid ();

      // Run the main event loop for the ORB.
      orb->run ();

      poa->destroy (1, 1);

      orb->destroy ();
    }
  catch (const CORBA::Exception& ex)
    {
      ex._tao_print_exception ("CIAO_NodeManager::main\t\n");
      return 1;
    }

  ACE_DEBUG ((LM_DEBUG,
              "CIAO_NodeManager has closed\n"));
  return 0;
}