native_client_sdk/src/tools/run.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Launch a local server on an ephemeral port, then launch a executable that
   7 points to that server.
   8 """
   9
  10 import copy
  11 import getos
  12 import optparse
  13 import os
  14 import subprocess
  15 import sys
  16 import httpd
  17
  18
  19 if sys.version_info < (2, 6, 0):
  20   sys.stderr.write("python 2.6 or later is required run this script\n")
  21   sys.exit(1)
  22
  23
  24 def main(args):
  25   usage = """usage: %prog [options] -- executable args...
  26
  27   This command creates a local server on an ephemeral port, then runs:
  28     <executable> <args..> http://localhost:<port>/<page>.
  29
  30   Where <page> can be set by -P, or uses index.html by default."""
  31   parser = optparse.OptionParser(usage)
  32   parser.add_option('-C', '--serve-dir',
  33       help='Serve files out of this directory.',
  34       dest='serve_dir', default=os.path.abspath('.'))
  35   parser.add_option('-P', '--path', help='Path to load from local server.',
  36       dest='path', default='index.html')
  37   parser.add_option('-D',
  38       help='Add debug command-line when launching the chrome debug.',
  39       dest='debug', action='append', default=[])
  40   parser.add_option('-E',
  41       help='Add environment variables when launching the executable.',
  42       dest='environ', action='append', default=[])
  43   parser.add_option('--test-mode',
  44       help='Listen for posts to /ok or /fail and shut down the server with '
  45           ' errorcodes 0 and 1 respectively.',
  46       dest='test_mode', action='store_true')
  47   parser.add_option('-p', '--port',
  48       help='Port to run server on. Default is 5103, ephemeral is 0.',
  49       type='int', default=5103)
  50   options, args = parser.parse_args(args)
  51   if not args:
  52     parser.error('No executable given.')
  53
  54   # 0 means use an ephemeral port.
  55   server = httpd.LocalHTTPServer(options.serve_dir, options.port,
  56                                  options.test_mode)
  57   print 'Serving %s on %s...' % (options.serve_dir, server.GetURL(''))
  58
  59   env = copy.copy(os.environ)
  60   for e in options.environ:
  61     key, value = map(str.strip, e.split('='))
  62     env[key] = value
  63
  64   cmd = args + [server.GetURL(options.path)]
  65   print 'Running: %s...' % (' '.join(cmd),)
  66   process = subprocess.Popen(cmd, env=env)
  67
  68   # If any debug args are passed in, assume we want to debug
  69   if options.debug:
  70     if getos.GetPlatform() == 'linux':
  71       cmd = ['xterm', '-title', 'NaCl Debugger', '-e']
  72       cmd += options.debug
  73     elif getos.GetPlatform() == 'mac':
  74       cmd = ['osascript', '-e',
  75              'tell application "Terminal" to do script "%s"' %
  76                  ' '.join(r'\"%s\"' % x for x in options.debug)]
  77     else:
  78       cmd = []
  79     print 'Starting debugger: ' + ' '.join(cmd)
  80     debug_process = subprocess.Popen(cmd, env=env)
  81   else:
  82     debug_process = False
  83
  84   try:
  85     return server.ServeUntilSubprocessDies(process)
  86   finally:
  87     if process.returncode is None:
  88       process.kill()
  89     if debug_process and debug_process.returncode is None:
  90       debug_process.kill()
  91
  92 if __name__ == '__main__':
  93   sys.exit(main(sys.argv[1:]))