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
  30 def check_and_run_mach(dir_path, args):
  31     # If we find the mach bootstrap module, we are in the srcdir.
  32     mach_path = os.path.join(dir_path, 'build/mach_bootstrap.py')
  33     if os.path.isfile(mach_path):
  34         mach = load_mach(dir_path)
  35         sys.exit(mach.run(args))
  36
  37
  38 def main(args):
  39     # Check whether the current directory is within a mach src or obj dir.
  40     for dir_path in ancestors(os.getcwd()):
  41         # If we find a "mozinfo.json" file, we are in the objdir.
  42         mozinfo_path = os.path.join(dir_path, 'mozinfo.json')
  43         if os.path.isfile(mozinfo_path):
  44             import json
  45             info = json.load(open(mozinfo_path))
  46             if 'mozconfig' in info and 'MOZCONFIG' not in os.environ:
  47                 # If the MOZCONFIG environment variable is not already set, set it
  48                 # to the value from mozinfo.json.  This will tell the build system
  49                 # to look for a config file at the path in $MOZCONFIG rather than
  50                 # its default locations.
  51                 #
  52                 # Note: subprocess requires native strings in os.environ on Windows
  53                 os.environ[b'MOZCONFIG'] = str(info['mozconfig'])
  54
  55             if 'topsrcdir' in info:
  56                 # Continue searching for mach_bootstrap in the source directory.
  57                 dir_path = info['topsrcdir']
  58
  59         check_and_run_mach(dir_path, args)
  60
  61     # If we didn't find a source path by scanning for a mozinfo.json, check
  62     # whether the directory containing this script is a source directory.
  63     check_and_run_mach(os.path.dirname(__file__), args)
  64
  65     print('Could not run mach: No mach source directory found.')
  66     sys.exit(1)
  67
  68
  69 if __name__ == '__main__':
  70     if sys.platform == 'win32':
  71         # This is a complete hack to work around the fact that Windows
  72         # multiprocessing needs to import the original module (ie: this
  73         # file), but only works if it has a .py extension.
  74         #
  75         # We do this by a sort of two-level function interposing. The first
  76         # level interposes forking.get_command_line() with our version defined
  77         # in my_get_command_line(). Our version of get_command_line will
  78         # replace the command string with the contents of the fork_interpose()
  79         # function to be used in the subprocess.
  80         #
  81         # The subprocess then gets an interposed imp.find_module(), which we
  82         # hack up to find 'mach' without the .py extension, since we already
  83         # know where it is (it's us!). If we're not looking for 'mach', then
  84         # the original find_module will suffice.
  85         #
  86         # See also: http://bugs.python.org/issue19946
  87         # And: https://bugzilla.mozilla.org/show_bug.cgi?id=914563
  88         import inspect
  89         from multiprocessing import forking
  90         global orig_command_line
  91
  92         def fork_interpose():
  93             import imp
  94             import os
  95             import sys
  96             orig_find_module = imp.find_module
  97             def my_find_module(name, dirs):
  98                 if name == 'mach':
  99                     path = os.path.join(dirs[0], 'mach')
 100                     f = open(path)
 101                     return (f, path, ('', 'r', imp.PY_SOURCE))
 102                 return orig_find_module(name, dirs)
 103
 104             # Don't allow writing bytecode file for mach module.
 105             orig_load_module = imp.load_module
 106             def my_load_module(name, file, path, description):
 107                 # multiprocess.forking invokes imp.load_module manually and
 108                 # hard-codes the name __parents_main__ as the module name.
 109                 if name == '__parents_main__':
 110                     old_bytecode = sys.dont_write_bytecode
 111                     sys.dont_write_bytecode = True
 112                     try:
 113                         return orig_load_module(name, file, path, description)
 114                     finally:
 115                         sys.dont_write_bytecode = old_bytecode
 116
 117                 return orig_load_module(name, file, path, description)
 118
 119             imp.find_module = my_find_module
 120             imp.load_module = my_load_module
 121             from multiprocessing.forking import main; main()
 122
 123         def my_get_command_line():
 124             fork_code, lineno = inspect.getsourcelines(fork_interpose)
 125             # Remove the first line (for 'def fork_interpose():') and the three
 126             # levels of indentation (12 spaces).
 127             fork_string = ''.join(x[12:] for x in fork_code[1:])
 128             cmdline = orig_command_line()
 129             cmdline[2] = fork_string
 130             return cmdline
 131         orig_command_line = forking.get_command_line
 132         forking.get_command_line = my_get_command_line
 133
 134     main(sys.argv[1:])