cpp/src/IceBox/Service.cpp

   1 // **********************************************************************
   2 //
   3 // Copyright (c) 2003-2011 ZeroC, Inc. All rights reserved.
   4 //
   5 // This copy of Ice is licensed to you under the terms described in the
   6 // ICE_LICENSE file included in this distribution.
   7 //
   8 // **********************************************************************
   9
  10 #include <IceUtil/Options.h>
  11 #include <Ice/Ice.h>
  12 #include <Ice/Service.h>
  13 #include <IceBox/ServiceManagerI.h>
  14
  15 using namespace std;
  16 using namespace Ice;
  17 using namespace IceBox;
  18
  19 namespace IceBox
  20 {
  21
  22 class IceBoxService : public Ice::Service
  23 {
  24 public:
  25
  26     IceBoxService();
  27
  28 protected:
  29
  30     virtual bool start(int, char*[], int&);
  31     virtual bool stop();
  32
  33 private:
  34
  35     void usage(const std::string&);
  36     ServiceManagerIPtr _serviceManager;
  37 };
  38
  39 }
  40
  41 IceBox::IceBoxService::IceBoxService()
  42 {
  43 }
  44
  45 bool
  46 IceBox::IceBoxService::start(int argc, char* argv[], int& status)
  47 {
  48     // Run through the command line arguments removing all the service
  49     // properties.
  50     vector<string> args = Ice::argsToStringSeq(argc, argv);
  51     PropertiesPtr properties = communicator()->getProperties();
  52     const string prefix = "IceBox.Service.";
  53     PropertyDict services = properties->getPropertiesForPrefix(prefix);
  54     for(PropertyDict::const_iterator p = services.begin(); p != services.end(); ++p)
  55     {
  56         string name = p->first.substr(prefix.size());
  57         StringSeq::iterator q = args.begin();
  58         while(q != args.end())
  59         {
  60             if(q->find("--" + name + ".") == 0)
  61             {
  62                 q = args.erase(q);
  63                 continue;
  64             }
  65             ++q;
  66         }
  67     }
  68
  69     IceUtilInternal::Options opts;
  70     opts.addOpt("h", "help");
  71     opts.addOpt("v", "version");
  72
  73     try
  74     {
  75         args = opts.parse(args);
  76     }
  77     catch(const IceUtilInternal::BadOptException& e)
  78     {
  79         error(e.reason);
  80         usage(argv[0]);
  81         return false;
  82     }
  83
  84     if(opts.isSet("help"))
  85     {
  86         usage(argv[0]);
  87         status = EXIT_SUCCESS;
  88         return false;
  89     }
  90     if(opts.isSet("version"))
  91     {
  92         print(ICE_STRING_VERSION);
  93         status = EXIT_SUCCESS;
  94         return false;
  95     }
  96
  97     if(!args.empty())
  98     {
  99         usage(argv[0]);
 100         return false;
 101     }
 102
 103     _serviceManager = new ServiceManagerI(communicator(), argc, argv);
 104
 105     return _serviceManager->start();
 106 }
 107
 108 bool
 109 IceBox::IceBoxService::stop()
 110 {
 111     if(_serviceManager)
 112     {
 113         _serviceManager->stop();
 114         _serviceManager = 0;
 115     }
 116     return true;
 117 }
 118
 119 void
 120 IceBox::IceBoxService::usage(const string& appName)
 121 {
 122     string options =
 123         "Options:\n"
 124         "-h, --help           Show this message.\n"
 125         "-v, --version        Display the Ice version.";
 126 #ifndef _WIN32
 127     options.append(
 128         "\n"
 129         "\n"
 130         "--daemon             Run as a daemon.\n"
 131         "--pidfile FILE       Write process ID into FILE.\n"
 132         "--noclose            Do not close open file descriptors.\n"
 133         "--nochdir            Do not change the current working directory."
 134     );
 135 #endif
 136     print("Usage: " + appName + " [options]\n" + options);
 137 }
 138
 139 //COMPILERFIX: Borland C++ 2010 doesn't support wmain for console applications.
 140 #if defined(_WIN32 ) && !defined(__BCPLUSPLUS__)
 141
 142 int
 143 wmain(int argc, wchar_t* argv[])
 144
 145 #else
 146
 147 int
 148 main(int argc, char* argv[])
 149
 150 #endif
 151 {
 152     IceBox::IceBoxService svc;
 153
 154     InitializationData initData;
 155     initData.properties = createProperties();
 156     initData.properties->setProperty("Ice.Admin.DelayCreation", "1");
 157     return svc.main(argc, argv, initData);
 158 }
 159