Index: tools/test-server.py |
diff --git a/tools/test-server.py b/tools/test-server.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..8ce7e40dad3de150771a26c208ed4ed09067b279 |
--- /dev/null |
+++ b/tools/test-server.py |
@@ -0,0 +1,217 @@ |
+#!/usr/bin/env python |
+# |
+# Copyright 2012 the V8 project authors. All rights reserved. |
+# Redistribution and use in source and binary forms, with or without |
+# modification, are permitted provided that the following conditions are |
+# met: |
+# |
+# * Redistributions of source code must retain the above copyright |
+# notice, this list of conditions and the following disclaimer. |
+# * Redistributions in binary form must reproduce the above |
+# copyright notice, this list of conditions and the following |
+# disclaimer in the documentation and/or other materials provided |
+# with the distribution. |
+# * Neither the name of Google Inc. nor the names of its |
+# contributors may be used to endorse or promote products derived |
+# from this software without specific prior written permission. |
+# |
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
+ |
+ |
+import os |
+import subprocess |
+import sys |
+ |
+ |
+PIDFILE = "/tmp/v8-distributed-testing-server.pid" |
+ROOT = os.path.abspath(os.path.dirname(sys.argv[0])) |
+ |
+ |
+def _PrintUsage(): |
+ print("""Usage: python %s COMMAND |
+ |
+Where COMMAND can be any of: |
+ start Starts the server. Forks to the background. |
+ stop Stops the server. |
+ restart Stops, then restarts the server. |
+ setup Creates or updates the environment for the server to run. |
+ update Alias for "setup". |
+ trust <keyfile> Adds the given public key to the list of trusted keys. |
+ help Displays this help text. |
+ """ % sys.argv[0]) |
+ |
+ |
+def _IsDaemonRunning(): |
+ return os.path.exists(PIDFILE) |
+ |
+ |
+def _Cmd(cmd): |
+ code = subprocess.call(cmd, shell=True) |
+ if code != 0: |
+ print("Command '%s' returned error code %d" % (cmd, code)) |
+ sys.exit(code) |
+ |
+ |
+def Update(): |
+ # Create directory for private data storage. |
+ data_dir = os.path.join(ROOT, "data") |
+ if not os.path.exists(data_dir): |
+ os.makedirs(data_dir) |
+ |
+ # Create directory for trusted public keys of peers (and self). |
+ trusted_dir = os.path.join(ROOT, "trusted") |
+ if not os.path.exists(trusted_dir): |
+ os.makedirs(trusted_dir) |
+ |
+ # Install UltraJSON. It is much faster than Python's builtin json. |
+ try: |
+ import ujson #@UnusedImport |
+ except ImportError: |
+ # Install pip if it doesn't exist. |
+ code = subprocess.call("which pip", shell=True) |
+ if code != 0: |
+ apt_get_code = subprocess.call("which apt-get", shell=True) |
+ if apt_get_code == 0: |
+ print("Installing pip...") |
+ _Cmd("sudo apt-get install python-pip") |
+ print("Updating pip using itself...") |
+ _Cmd("sudo pip install --upgrade pip") |
+ else: |
+ print("Please install pip on your machine. You can get it at: " |
+ "http://www.pip-installer.org/en/latest/installing.html " |
+ "or via your distro's package manager.") |
+ sys.exit(1) |
+ print("Using pip to install UltraJSON...") |
+ _Cmd("sudo pip install ujson") |
+ |
+ # Make sure we have a key pair for signing binaries. |
+ privkeyfile = os.path.expanduser("~/.ssh/v8_dtest") |
+ if not os.path.exists(privkeyfile): |
+ _Cmd("ssh-keygen -t rsa -f %s -N '' -q" % privkeyfile) |
+ fingerprint = subprocess.check_output("ssh-keygen -lf %s" % privkeyfile, |
+ shell=True) |
+ fingerprint = fingerprint.split(" ")[1].replace(":", "")[:16] |
+ pubkeyfile = os.path.join(trusted_dir, "%s.pem" % fingerprint) |
+ if (not os.path.exists(pubkeyfile) or |
+ os.path.getmtime(pubkeyfile) < os.path.getmtime(privkeyfile)): |
+ _Cmd("openssl rsa -in %s -out %s -pubout" % (privkeyfile, pubkeyfile)) |
+ with open(pubkeyfile, "a") as f: |
+ f.write(fingerprint + "\n") |
+ datafile = os.path.join(data_dir, "mypubkey") |
+ with open(datafile, "w") as f: |
+ f.write(fingerprint + "\n") |
+ |
+ # Check out or update the server implementation in the current directory. |
+ testrunner_dir = os.path.join(ROOT, "testrunner") |
+ if os.path.exists(os.path.join(testrunner_dir, "server/daemon.py")): |
+ _Cmd("cd %s; svn up" % testrunner_dir) |
+ else: |
+ path = ("http://v8.googlecode.com/svn/branches/bleeding_edge/" |
+ "tools/testrunner") |
+ _Cmd("svn checkout --force %s %s" % (path, testrunner_dir)) |
+ |
+ # Update this very script. |
+ path = ("http://v8.googlecode.com/svn/branches/bleeding_edge/" |
+ "tools/server.py") |
+ scriptname = os.path.abspath(sys.argv[0]) |
+ _Cmd("svn cat %s > %s" % (path, scriptname)) |
+ |
+ # Check out or update V8. |
+ v8_dir = os.path.join(ROOT, "v8") |
+ if os.path.exists(v8_dir): |
+ _Cmd("cd %s; git fetch" % v8_dir) |
+ else: |
+ _Cmd("git clone git://github.com/v8/v8.git %s" % v8_dir) |
+ |
+ print("Finished.") |
+ |
+ |
+# Handle "setup" here, because when executing that we can't import anything |
+# else yet. |
+if __name__ == "__main__" and len(sys.argv) == 2: |
+ if sys.argv[1] in ("setup", "update"): |
+ if _IsDaemonRunning(): |
+ print("Please stop the server before updating. Exiting.") |
+ sys.exit(1) |
+ Update() |
+ sys.exit(0) |
+ # Other parameters are handled below. |
+ |
+ |
+#========================================================== |
+# At this point we can assume that the implementation is available, |
+# so we can import it. |
+try: |
+ from testrunner.server import constants |
+ from testrunner.server import local_handler |
+ from testrunner.server import main |
+except Exception, e: |
+ print(e) |
+ print("Failed to import implementation. Have you run 'setup'?") |
+ sys.exit(1) |
+ |
+ |
+def _StartDaemon(daemon): |
+ if not os.path.isdir(os.path.join(ROOT, "v8")): |
+ print("No 'v8' working directory found. Have you run 'setup'?") |
+ sys.exit(1) |
+ daemon.start() |
+ |
+ |
+if __name__ == "__main__": |
+ if len(sys.argv) == 2: |
+ arg = sys.argv[1] |
+ if arg == "start": |
+ daemon = main.Server(PIDFILE, ROOT) |
+ _StartDaemon(daemon) |
+ elif arg == "stop": |
+ daemon = main.Server(PIDFILE, ROOT) |
+ daemon.stop() |
+ elif arg == "restart": |
+ daemon = main.Server(PIDFILE, ROOT) |
+ daemon.stop() |
+ _StartDaemon(daemon) |
+ elif arg in ("help", "-h", "--help"): |
+ _PrintUsage() |
+ elif arg == "status": |
+ if not _IsDaemonRunning(): |
+ print("Server not running.") |
+ else: |
+ print(local_handler.LocalQuery([constants.REQUEST_STATUS])) |
+ else: |
+ print("Unknown command") |
+ _PrintUsage() |
+ sys.exit(2) |
+ elif len(sys.argv) == 3: |
+ arg = sys.argv[1] |
+ if arg == "approve": |
+ filename = sys.argv[2] |
+ if not os.path.exists(filename): |
+ print("%s does not exist.") |
+ sys.exit(1) |
+ filename = os.path.abspath(filename) |
+ if _IsDaemonRunning(): |
+ response = local_handler.LocalQuery([constants.ADD_TRUSTED, filename]) |
+ else: |
+ daemon = main.Server(PIDFILE, ROOT) |
+ response = daemon.CopyToTrusted(filename) |
+ print("Added certificate %s to trusted certificates." % response) |
+ else: |
+ print("Unknown command") |
+ _PrintUsage() |
+ sys.exit(2) |
+ else: |
+ print("Unknown command") |
+ _PrintUsage() |
+ sys.exit(2) |
+ sys.exit(0) |