TAO/orbsvcs/ImplRepo_Service/ImR_Activator.cpp

   1 /* -*- C++ -*- */
   2 #include "orbsvcs/Log_Macros.h"
   3 #include "Activator_Options.h"
   4 #include "ImR_Activator_i.h"
   5 #include "Activator_NT_Service.h"
   6
   7 #include "orbsvcs/Shutdown_Utilities.h"
   8
   9 class ImR_Activator_Shutdown : public Shutdown_Functor
  10 {
  11 public:
  12   ImR_Activator_Shutdown (ImR_Activator_i& act);
  13
  14   void operator() (int which_signal);
  15 private:
  16   ImR_Activator_i& act_;
  17 };
  18
  19 ImR_Activator_Shutdown::ImR_Activator_Shutdown (ImR_Activator_i &act)
  20   : act_(act)
  21 {
  22 }
  23
  24 void
  25 ImR_Activator_Shutdown::operator() (int /*which_signal*/)
  26 {
  27   try
  28   {
  29     this->act_.shutdown (true);
  30   }
  31   catch (const CORBA::Exception& ex)
  32   {
  33     ex._tao_print_exception ("ImR Activator: ");
  34   }
  35 }
  36
  37 int
  38 run_standalone (Activator_Options& opts)
  39 {
  40   ImR_Activator_i server;
  41
  42   ImR_Activator_Shutdown killer (server);
  43   Service_Shutdown kill_contractor (killer);
  44
  45   try
  46     {
  47       int status = server.init (opts);
  48
  49       if (status == -1)
  50         {
  51           return 1;
  52         }
  53       else
  54         {
  55           // Run the server if it is initialized correctly.
  56           server.run ();
  57
  58           // End the server after its work is done.
  59           status = server.fini ();
  60
  61           if (status == -1)
  62             return 1;
  63         }
  64       return 0;
  65     }
  66   catch (const CORBA::SystemException& sysex)
  67     {
  68       sysex._tao_print_exception ("System Exception");
  69     }
  70   catch (const CORBA::UserException& userex)
  71     {
  72       userex._tao_print_exception ("User Exception");
  73     }
  74   catch (const CORBA::Exception& ex)
  75     {
  76       ex._tao_print_exception ("Unknown Exception");
  77     }
  78
  79   return 1;
  80 }
  81
  82 #if defined (ACE_WIN32) && !defined (ACE_LACKS_WIN32_SERVICES)
  83 ACE_NT_SERVICE_DEFINE (service, Activator_NT_Service, IMR_ACTIVATOR_SERVICE_NAME);
  84 #endif /* ACE_WIN32 && !ACE_LACKS_WIN32_SERVICES */
  85
  86 int
  87 run_service ()
  88 {
  89 #if defined (ACE_WIN32) && !defined (ACE_LACKS_WIN32_SERVICES)
  90   SERVICE::instance()->name (IMR_ACTIVATOR_SERVICE_NAME, IMR_ACTIVATOR_DISPLAY_NAME);
  91   ACE_NT_SERVICE_RUN (service, SERVICE::instance (), ret);
  92
  93   if (ret == 0)
  94     ORBSVCS_ERROR ((LM_ERROR, "%p\n", "Couldn't start service"));
  95
  96   return ret;
  97 #else /* ACE_WIN32 */
  98   return 1;
  99 #endif /* ACE_WIN32 && !ACE_LACKS_WIN32_SERVICES */
 100 }
 101
 102 /**
 103  * Executes the various commands that are useful for a NT service.  Right
 104  * now these include 'install' and 'remove'.  Others, such as 'start' and
 105  * 'stop' can be added, but the 'net' program in Windows already handles
 106  * these commands.
 107  */
 108 static int
 109 run_service_command (Activator_Options& opts)
 110 {
 111   if (opts.service_command () == Activator_Options::SC_NONE)
 112     return 0;
 113
 114 #if defined (ACE_WIN32) && !defined (ACE_LACKS_WIN32_SERVICES)
 115   SERVICE::instance()->name (IMR_ACTIVATOR_SERVICE_NAME, IMR_ACTIVATOR_DISPLAY_NAME);
 116
 117   if (opts.service_command () == Activator_Options::SC_INSTALL ||
 118     opts.service_command () == Activator_Options::SC_INSTALL_NO_LOCATOR)
 119       {
 120         const DWORD MAX_PATH_LENGTH = 4096;
 121         ACE_TCHAR pathname[MAX_PATH_LENGTH];
 122
 123         DWORD length = ACE_TEXT_GetModuleFileName(0, pathname, MAX_PATH_LENGTH);
 124         if (length == 0 || length >= MAX_PATH_LENGTH - sizeof(" -s"))
 125           {
 126             ORBSVCS_ERROR ((LM_ERROR, "Error: Could not get module file name\n"));
 127             return -1;
 128           }
 129
 130         // Append the command used for running the implrepo as a service
 131         ACE_OS::strcat (pathname, ACE_TEXT (" -s"));
 132         int ret = -1;
 133         if (opts.service_command () == Activator_Options::SC_INSTALL)
 134           {
 135             // Must match Locator_NT_Service.h
 136             const ACE_TCHAR* DEPENDS_ON = ACE_TEXT("TAOImR");
 137
 138             ret =  SERVICE::instance ()->insert (SERVICE_DEMAND_START,
 139                                                  SERVICE_ERROR_NORMAL,
 140                                                  pathname,
 141                                                  0, // group
 142                                                  0, // tag
 143                                                  DEPENDS_ON);
 144           }
 145         else
 146           {
 147             ret =  SERVICE::instance ()->insert (SERVICE_DEMAND_START,
 148                                                 SERVICE_ERROR_NORMAL,
 149                                                 pathname);
 150           }
 151       if (ret != -1)
 152         {
 153           ORBSVCS_DEBUG ((LM_DEBUG, "ImR Activator: Service installed\n"));
 154           opts.save_registry_options ();
 155         }
 156       else
 157         {
 158           ORBSVCS_ERROR ((LM_ERROR, "Error: Failed to install service. errno %d, error <%p>\n", errno, "install"));
 159         }
 160       if (ret == 0)
 161         return 1;
 162     }
 163   else if (opts.service_command () == Activator_Options::SC_REMOVE)
 164     {
 165       int ret = SERVICE::instance ()->remove ();
 166       ORBSVCS_DEBUG ((LM_DEBUG, "ImR Activator: Service removed\n"));
 167       if (ret == 0)
 168         return 1; // If successful, then we don't want to continue.
 169     }
 170   else
 171     {
 172       ORBSVCS_ERROR ((LM_ERROR, "Error: Unknown service command :%d\n",
 173         opts.service_command ()));
 174       return -1;
 175     }
 176
 177   return -1;
 178
 179 #else /* ACE_WIN32 */
 180   ORBSVCS_ERROR ((LM_ERROR, "NT Service not supported on this platform"));
 181   return -1;
 182 #endif /* ACE_WIN32 && !ACE_LACKS_WIN32_SERVICES */
 183 }
 184
 185 int
 186 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 187 {
 188   Activator_Options opts;
 189
 190   int result = opts.init (argc, argv);
 191   if (result < 0)
 192     return 1;  // Error
 193   else if (result > 0)
 194     return 0;  // No error, but we should exit anyway.
 195
 196   result = run_service_command (opts);
 197   if (result < 0)
 198     return 1;  // Error
 199   else if (result > 0)
 200     return 0;  // No error, but we should exit anyway.
 201
 202   if (opts.service ())
 203     return run_service ();
 204
 205   return run_standalone (opts);
 206 }
 207