1 files changed, 76 insertions, 0 deletions
diff --git a/scripts/Glacier2Util.py b/scripts/Glacier2Util.py
new file mode 100644
index 00000000000..81bdc050f67
--- /dev/null
+++ b/scripts/Glacier2Util.py
@@ -0,0 +1,76 @@
+# **********************************************************************
+#
+# Copyright (c) 2003-2016 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.
+#
+# **********************************************************************
+
+import sys, os
+from Util import *
+
+class Glacier2Router(ProcessFromBinDir, Server):
+
+    def __init__(self, portnum=10, passwords={"userid": "abc123"}, *args, **kargs):
+        Server.__init__(self, "glacier2router", mapping=Mapping.getByName("cpp"), desc="Glacier2 router",
+                        readyCount=2, *args, **kargs)
+        self.portnum = portnum
+        self.passwords = passwords
+
+    def setup(self, current):
+        if self.passwords:
+            path = os.path.join(current.testcase.getPath(), "passwords")
+            with open(path, "w") as file:
+                command = "%s %s" % (sys.executable,
+                                     os.path.abspath(os.path.join(toplevel, "scripts", "icehashpassword.py")))
+
+                #
+                # For Linux ARM default rounds makes test slower (Usually runs on embbeded boards)
+                #
+                if current.config.buildPlatform.find("arm") >= 0:
+                    command += " --rounds 100000"
+
+                for user, password in self.passwords.items():
+                    p = subprocess.Popen(command,
+                                         shell=True,
+                                         stdout=subprocess.PIPE,
+                                         stderr=subprocess.STDOUT,
+                                         stdin=subprocess.PIPE)
+                    p.stdin.write(password.encode('UTF-8'))
+                    p.stdin.write('\r\n'.encode('UTF-8'))
+                    p.stdin.flush()
+                    if(p.wait() != 0):
+                        raise RuntimeError("icehashpassword.py failed:\n" + p.stdout.read().decode('UTF-8').strip())
+                    file.write("%s %s\n" % (user, p.stdout.readline().decode('UTF-8').strip()))
+            current.testcase.files.append(path)
+
+    def getProps(self, current):
+        props = Server.getProps(self, current)
+        props.update({
+            "Glacier2.Client.Endpoints" : current.getTestEndpoint(self.portnum),
+            "Glacier2.Server.Endpoints" : "tcp",
+            "Ice.Admin.Endpoints" : current.getTestEndpoint(self.portnum + 1, "tcp"),
+            "Ice.Admin.InstanceName" : "Glacier2",
+        })
+        if self.passwords:
+            props["Glacier2.CryptPasswords"] = os.path.join("{testdir}", "passwords")
+        if isinstance(current.testcase.getTestSuite(), Glacier2TestSuite):
+            # Add the properties provided by the Glacier2TestSuite routerProps parameter.
+            props.update(current.testcase.getTestSuite().getRouterProps(self, current))
+        return props
+
+    def getClientProxy(self, current):
+        return "Glacier2/router:{0}".format(current.getTestEndpoint(self.portnum))
+
+class Glacier2TestSuite(TestSuite):
+
+    def __init__(self, path, routerProps={}, testcases=None, *args, **kargs):
+        if testcases is None:
+            testcases = [ClientServerTestCase(servers=[Glacier2Router(), Server()])]
+        TestSuite.__init__(self, path, testcases, *args, **kargs)
+        self.routerProps = routerProps
+
+    def getRouterProps(self, process, current):
+        return self.routerProps(process, current) if callable(self.routerProps) else self.routerProps.copy()
+