Ice 3.4.2 Source Distributionv3.4.2

1 files changed, 218 insertions, 0 deletions

diff --git a/cpp/src/IceBox/Admin.cpp b/cpp/src/IceBox/Admin.cpp
new file mode 100644
index 00000000000..5bc96d0f084
--- /dev/null
+++ b/cpp/src/IceBox/Admin.cpp
@@ -0,0 +1,218 @@
+// **********************************************************************
+//
+// Copyright (c) 2003-2011 ZeroC, Inc. All rights reserved.
+//
+// This copy of Ice is licensed to you under the terms described in the
+// ICE_LICENSE file included in this distribution.
+//
+// **********************************************************************
+
+#include <Ice/Application.h>
+#include <Ice/SliceChecksums.h>
+#include <IceUtil/Options.h>
+#include <IceBox/IceBox.h>
+
+using namespace std;
+using namespace Ice;
+
+class Client : public Application
+{
+public:
+
+    void usage();
+    virtual int run(int, char*[]);
+};
+
+//COMPILERFIX: Borland C++ 2010 doesn't support wmain for console applications.
+#if defined(_WIN32 ) && !defined(__BCPLUSPLUS__)
+
+int
+wmain(int argc, wchar_t* argv[])
+
+#else
+
+int
+main(int argc, char* argv[])
+
+#endif
+{
+    Client app;
+    return app.main(argc, argv);
+}
+
+void
+Client::usage()
+{
+    cerr << "Usage: " << appName() << " [options] [command...]\n";
+    cerr <<        
+        "Options:\n"
+        "-h, --help           Show this message.\n"
+        "-v, --version        Display the Ice version.\n"
+        "\n"
+        "Commands:\n"
+        "start SERVICE        Start a service.\n"
+        "stop SERVICE         Stop a service.\n"
+        "shutdown             Shutdown the server.\n"
+        ;
+}
+
+int
+Client::run(int argc, char* argv[])
+{
+    IceUtilInternal::Options opts;
+    opts.addOpt("h", "help");
+    opts.addOpt("v", "version");
+
+    vector<string> commands;
+    try
+    {
+        commands = opts.parse(argc, (const char**)argv);
+    }
+    catch(const IceUtilInternal::BadOptException& e)
+    {
+        cerr << e.reason << endl;
+        usage();
+        return EXIT_FAILURE;
+    }
+
+    if(opts.isSet("help"))
+    {
+        usage();
+        return EXIT_SUCCESS;
+    }
+    if(opts.isSet("version"))
+    {
+        cout << ICE_STRING_VERSION << endl;
+        return EXIT_SUCCESS;
+    }
+
+    if(commands.empty())
+    {
+        usage();
+        return EXIT_FAILURE;
+    }
+
+  
+    ObjectPrx base = communicator()->propertyToProxy("IceBoxAdmin.ServiceManager.Proxy");
+
+    if(base == 0)
+    {
+        //
+        // The old deprecated way to retrieve the service manager proxy
+        //
+
+        PropertiesPtr properties = communicator()->getProperties();
+
+        Identity managerIdentity;
+        managerIdentity.category = properties->getPropertyWithDefault("IceBox.InstanceName", "IceBox");
+        managerIdentity.name = "ServiceManager";
+        
+        string managerProxy;
+        if(properties->getProperty("Ice.Default.Locator").empty())
+        {
+            string managerEndpoints = properties->getProperty("IceBox.ServiceManager.Endpoints");
+            if(managerEndpoints.empty())
+            {
+                cerr << appName() << ": property `IceBoxAdmin.ServiceManager.Proxy' is not set" << endl;
+                return EXIT_FAILURE;
+            }
+            
+            managerProxy = "\"" + communicator()->identityToString(managerIdentity) + "\" :" + managerEndpoints;
+        }
+        else
+        {
+            string managerAdapterId = properties->getProperty("IceBox.ServiceManager.AdapterId");
+            if(managerAdapterId.empty())
+            {
+                cerr << appName() << ": property `IceBoxAdmin.ServiceManager.Proxy' is not set" << endl;
+                return EXIT_FAILURE;
+            }
+            
+            managerProxy = "\"" + communicator()->identityToString(managerIdentity) + "\" @" + managerAdapterId;
+        }
+
+        base = communicator()->stringToProxy(managerProxy);
+    }
+
+    IceBox::ServiceManagerPrx manager = IceBox::ServiceManagerPrx::checkedCast(base);
+    if(!manager)
+    {
+        cerr << appName() << ": `" << base << "' is not an IceBox::ServiceManager" << endl;
+        return EXIT_FAILURE;
+    }
+
+    Ice::SliceChecksumDict serverChecksums = manager->getSliceChecksums();
+    Ice::SliceChecksumDict localChecksums = Ice::sliceChecksums();
+    for(Ice::SliceChecksumDict::const_iterator p = localChecksums.begin(); p != localChecksums.end(); ++p)
+    {
+        Ice::SliceChecksumDict::const_iterator q = serverChecksums.find(p->first);
+        if(q == serverChecksums.end())
+        {
+            cerr << appName() << ": server is using unknown Slice type `" << q->first << "'" << endl;
+        }
+        else if(p->second != q->second)
+        {
+            cerr << appName() << ": server is using a different Slice definition of `" << q->first << "'" << endl;
+        }
+    }
+
+    
+    vector<string>::const_iterator r;
+    for(r = commands.begin(); r != commands.end(); ++r)
+    {
+        if((*r) == "shutdown")
+        {
+            manager->shutdown();
+        }
+        else if((*r) == "start")
+        {
+            if(++r == commands.end())
+            {
+                cerr << appName() << ": no service name specified." << endl;
+                return EXIT_FAILURE;
+            }
+
+            try
+            {
+                manager->startService(*r);
+            }
+            catch(const IceBox::NoSuchServiceException&)
+            {
+                cerr << appName() << ": unknown service `" << *r << "'" << endl;
+            }
+            catch(const IceBox::AlreadyStartedException&)
+            {
+                cerr << appName() << ": service already started." << endl;
+            }
+        }
+        else if((*r) == "stop")
+        {
+            if(++r == commands.end())
+            {
+                cerr << appName() << ": no service name specified." << endl;
+                return EXIT_FAILURE;
+            }
+
+            try
+            {
+                manager->stopService(*r);
+            }
+            catch(const IceBox::NoSuchServiceException&)
+            {
+                cerr << appName() << ": unknown service `" << *r << "'" << endl;
+            }
+            catch(const IceBox::AlreadyStoppedException&)
+            {
+                cerr << appName() << ": service already stopped." << endl;
+            }
+        }
+        else
+        {
+            cerr << appName() << ": unknown command `" << *r << "'" << endl;
+            usage();
+            return EXIT_FAILURE;
+        }
+    }
+
+    return EXIT_SUCCESS;
+}