mach

   1 #!/bin/sh
   2 # This Source Code Form is subject to the terms of the Mozilla Public
   3 # License, v. 2.0. If a copy of the MPL was not distributed with this
   4 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   5
   6 # The beginning of this script is both valid shell and valid python,
   7 # such that the script starts with the shell and is reexecuted with
   8 # the right python.
   9 '''which' python2.7 > /dev/null && exec python2.7 "$0" "$@" || exec python "$0" "$@"
  10 '''
  11
  12 from __future__ import print_function, unicode_literals
  13
  14 import os
  15 import sys
  16
  17 def ancestors(path):
  18     while path:
  19         yield path
  20         (path, child) = os.path.split(path)
  21         if child == "":
  22             break
  23
  24 def load_mach(topsrcdir):
  25     sys.path[0:0] = [os.path.join(topsrcdir, "build")]
  26     import mach_bootstrap
  27     return mach_bootstrap.bootstrap(topsrcdir)
  28
  29 def main(args):
  30     # Check whether the current directory is within a mach src or obj dir.
  31     for dir_path in ancestors(os.getcwd()):
  32         # If we find a "mozinfo.json" file, we are in the objdir.
  33         mozinfo_path = os.path.join(dir_path, 'mozinfo.json')
  34         if os.path.isfile(mozinfo_path):
  35             import json
  36             info = json.load(open(mozinfo_path))
  37             if 'mozconfig' in info and 'MOZCONFIG' not in os.environ:
  38                 # If the MOZCONFIG environment variable is not already set, set it
  39                 # to the value from mozinfo.json.  This will tell the build system
  40                 # to look for a config file at the path in $MOZCONFIG rather than
  41                 # its default locations.
  42                 #
  43                 # Note: subprocess requires native strings in os.environ on Windows
  44                 os.environ[b'MOZCONFIG'] = str(info['mozconfig'])
  45
  46             if 'topsrcdir' in info:
  47                 # Continue searching for mach_bootstrap in the source directory.
  48                 dir_path = info['topsrcdir']
  49
  50         # If we find the mach bootstrap module, we are in the srcdir.
  51         mach_path = os.path.join(dir_path, 'build/mach_bootstrap.py')
  52         if os.path.isfile(mach_path):
  53             mach = load_mach(dir_path)
  54             sys.exit(mach.run(args[1:]))
  55
  56     print('Could not run mach: No mach source directory found.')
  57     sys.exit(1)
  58
  59
  60 if __name__ == '__main__':
  61     if sys.platform == 'win32':
  62         # This is a complete hack to work around the fact that Windows
  63         # multiprocessing needs to import the original module (ie: this
  64         # file), but only works if it has a .py extension.
  65         #
  66         # We do this by a sort of two-level function interposing. The first
  67         # level interposes forking.get_command_line() with our version defined
  68         # in my_get_command_line(). Our version of get_command_line will
  69         # replace the command string with the contents of the fork_interpose()
  70         # function to be used in the subprocess.
  71         #
  72         # The subprocess then gets an interposed imp.find_module(), which we
  73         # hack up to find 'mach' without the .py extension, since we already
  74         # know where it is (it's us!). If we're not looking for 'mach', then
  75         # the original find_module will suffice.
  76         #
  77         # See also: http://bugs.python.org/issue19946
  78         # And: https://bugzilla.mozilla.org/show_bug.cgi?id=914563
  79         import inspect
  80         from multiprocessing import forking
  81         global orig_command_line
  82
  83         def fork_interpose():
  84             import imp
  85             import os
  86             orig_find_module = imp.find_module
  87             def my_find_module(name, dirs):
  88                 if name == 'mach':
  89                     path = os.path.join(dirs[0], 'mach')
  90                     f = open(path)
  91                     return (f, path, ('', 'r', imp.PY_SOURCE))
  92                 return orig_find_module(name, dirs)
  93
  94             imp.find_module = my_find_module
  95             from multiprocessing.forking import main; main()
  96
  97         def my_get_command_line():
  98             fork_code, lineno = inspect.getsourcelines(fork_interpose)
  99             # Remove the first line (for 'def fork_interpose():') and the three
 100             # levels of indentation (12 spaces).
 101             fork_string = ''.join(x[12:] for x in fork_code[1:])
 102             cmdline = orig_command_line()
 103             cmdline[2] = fork_string
 104             return cmdline
 105         orig_command_line = forking.get_command_line
 106         forking.get_command_line = my_get_command_line
 107
 108     main(sys.argv)