ACE/examples/NT_Service/main.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    main.cpp
   5  *
   6  *  This is the main program - it just hands control off to the
   7  *  process instance to figure out what to do.  This program only
   8  *  runs on Win32.
   9  *
  10  *  @author Gonzalo Diethelm <gonzo@cs.wustl.edu> and Steve Huston <shuston@riverace.com>
  11  */
  12 //=============================================================================
  13
  14
  15 #include "ace/Get_Opt.h"
  16 #include "ace/Init_ACE.h"
  17 #include "ntsvc.h"
  18
  19 #if defined (ACE_WIN32) && !defined (ACE_LACKS_WIN32_SERVICES)
  20 // FUZZ: disable check_for_streams_include
  21 #include "ace/streams.h"
  22 #include "ace/OS_NS_errno.h"
  23
  24 // Default for the -i (install) option
  25 #define DEFAULT_SERVICE_INIT_STARTUP     SERVICE_AUTO_START
  26
  27 class Process
  28 {
  29 public:
  30   Process ();
  31   ~Process ();
  32
  33   int run(int argc, ACE_TCHAR* argv[]);
  34
  35 private:
  36   void parse_args (int argc,
  37                    ACE_TCHAR* argv[]);
  38   void print_usage_and_die ();
  39
  40 private:
  41   char progname[128];
  42
  43   int opt_install;
  44   int opt_remove;
  45   int opt_start;
  46   int opt_kill;
  47   int opt_type;
  48   int opt_debug;
  49
  50   int opt_startup;
  51 };
  52
  53 typedef ACE_Singleton<Process, ACE_Mutex> PROCESS;
  54
  55 Process::Process ()
  56   : opt_install (0),
  57     opt_remove (0),
  58     opt_start (0),
  59     opt_kill (0),
  60     opt_type (0),
  61     opt_debug (0),
  62     opt_startup (0)
  63 {
  64   ACE_OS::strcpy (progname,
  65                   "service");
  66   ACE::init ();
  67 }
  68
  69 Process::~Process ()
  70 {
  71   ACE::fini ();
  72 }
  73
  74 void
  75 Process::print_usage_and_die ()
  76 {
  77   ACE_DEBUG ((LM_INFO,
  78               "Usage: %s"
  79               " -in -r -s -k -tn -d\n"
  80               "  -i: Install this program as an NT service, with specified startup\n"
  81               "  -r: Remove this program from the Service Manager\n"
  82               "  -s: Start the service\n"
  83               "  -k: Kill the service\n"
  84               "  -t: Set startup for an existing service\n"
  85               "  -d: Debug; run as a regular application\n",
  86               progname,
  87               0));
  88   ACE_OS::exit(1);
  89 }
  90
  91 void
  92 Process::parse_args (int argc, ACE_TCHAR* argv[])
  93 {
  94   ACE_Get_Opt get_opt (argc, argv, ACE_TEXT ("i:rskt:d"));
  95   int c;
  96
  97   while ((c = get_opt ()) != -1)
  98     switch (c)
  99     {
 100     case 'i':
 101       opt_install = 1;
 102       opt_startup = ACE_OS::atoi (get_opt.opt_arg ());
 103       if (opt_startup <= 0)
 104          print_usage_and_die ();
 105       break;
 106     case 'r':
 107       opt_remove = 1;
 108       break;
 109     case 's':
 110       opt_start = 1;
 111       break;
 112     case 'k':
 113       opt_kill = 1;
 114       break;
 115     case 't':
 116       opt_type = 1;
 117       opt_startup = ACE_OS::atoi (get_opt.opt_arg ());
 118       if (opt_startup <= 0)
 119          print_usage_and_die ();
 120       break;
 121     case 'd':
 122       opt_debug = 1;
 123       break;
 124     default:
 125       // -i can also be given without a value - if so, it defaults
 126       // to defined value.
 127       if (ACE_OS::strcmp (get_opt.argv ()[get_opt.opt_ind () - 1], ACE_TEXT ("-i")) == 0)
 128       {
 129         opt_install = 1;
 130         opt_startup = DEFAULT_SERVICE_INIT_STARTUP;
 131       }
 132       else
 133       {
 134         print_usage_and_die ();
 135       }
 136       break;
 137   }
 138 }
 139
 140 // Define a function to handle Ctrl+C to cleanly shut this down.
 141
 142 static BOOL WINAPI
 143 ConsoleHandler (DWORD /*ctrlType*/)
 144 {
 145   SERVICE::instance ()->handle_control (SERVICE_CONTROL_STOP);
 146   return TRUE;
 147 }
 148
 149 ACE_NT_SERVICE_DEFINE (Beeper,
 150                        Service,
 151                        ACE_TEXT ("Annoying Beeper Service"));
 152
 153 int
 154 Process::run (int argc, ACE_TCHAR* argv[])
 155 {
 156   SERVICE::instance ()->name (ACE_TEXT ("Beeper"),
 157                               ACE_TEXT ("Annoying Beeper Service"));
 158
 159   parse_args (argc, argv);
 160
 161   if (opt_install && !opt_remove)
 162     {
 163       if (-1 == SERVICE::instance ()->insert (opt_startup))
 164         {
 165           ACE_ERROR ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("insert")));
 166           return -1;
 167         }
 168       return 0;
 169     }
 170
 171   if (opt_remove && !opt_install)
 172     {
 173       if (-1 == SERVICE::instance ()->remove ())
 174         {
 175           ACE_ERROR ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("remove")));
 176           return -1;
 177         }
 178       return 0;
 179     }
 180
 181   if (opt_start && opt_kill)
 182     print_usage_and_die ();
 183
 184   if (opt_start)
 185     {
 186       if (-1 == SERVICE::instance ()->start_svc ())
 187         {
 188           ACE_ERROR ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("start")));
 189           return -1;
 190         }
 191       return 0;
 192     }
 193
 194   if (opt_kill)
 195     {
 196       if (-1 == SERVICE::instance ()->stop_svc ())
 197         {
 198           ACE_ERROR ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("stop")));
 199           return -1;
 200         }
 201       return 0;
 202     }
 203
 204   if (opt_type)
 205     {
 206       if (-1 == SERVICE::instance ()->startup (opt_startup))
 207         {
 208           ACE_ERROR ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("set startup")));
 209           return -1;
 210         }
 211       return 0;
 212     }
 213
 214   // If we get here, we either run the app in debug mode (-d) or are
 215   // being called from the service manager to start the service.
 216
 217   if (opt_debug)
 218     {
 219       SetConsoleCtrlHandler (&ConsoleHandler, 1);
 220       SERVICE::instance ()->svc ();
 221     }
 222   else
 223     {
 224       ofstream *output_file = new ofstream("ntsvc.log", ios::out);
 225       if (output_file && output_file->rdstate() == ios::goodbit)
 226         ACE_LOG_MSG->msg_ostream(output_file, 1);
 227       ACE_LOG_MSG->open(argv[0],
 228                         ACE_Log_Msg::STDERR | ACE_Log_Msg::OSTREAM,
 229                         0);
 230       ACE_DEBUG ((LM_DEBUG, ACE_TEXT ("%T (%t): Starting service.\n")));
 231
 232       ACE_NT_SERVICE_RUN (Beeper,
 233                           SERVICE::instance (),
 234                           ret);
 235       if (ret == 0)
 236         ACE_ERROR ((LM_ERROR,
 237                     ACE_TEXT ("%p\n"),
 238                     ACE_TEXT ("Couldn't start service")));
 239       else
 240         ACE_DEBUG ((LM_DEBUG, ACE_TEXT ("%T (%t): Service stopped.\n")));
 241     }
 242
 243   return 0;
 244 }
 245
 246 int
 247 ACE_TMAIN (int argc, ACE_TCHAR* argv[])
 248 {
 249   return PROCESS::instance ()->run (argc, argv);
 250 }
 251
 252 #else
 253
 254 #include "ace/OS_main.h"
 255
 256 int
 257 ACE_TMAIN (int, ACE_TCHAR*[])
 258 {
 259   // This program needs Windows services.
 260   return 0;
 261 }
 262
 263 #endif /* ACE_WIN32 && !ACE_LACKS_WIN32_SERVICES */