diff options
Diffstat (limited to 'mach')
| -rwxr-xr-x | mach | 150 |
1 files changed, 150 insertions, 0 deletions
@@ -0,0 +1,150 @@ +#!/bin/sh +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. + +# The beginning of this script is both valid shell and valid python, +# such that the script starts with the shell and is reexecuted with +# the right python. +'''which' python2.7 > /dev/null && exec python2.7 "$0" "$@" || exec python "$0" "$@" +''' + +from __future__ import print_function, unicode_literals + +import os +import sys + +def ancestors(path): + while path: + yield path + (path, child) = os.path.split(path) + if child == "": + break + +def load_mach(dir_path, mach_path): + import imp + with open(mach_path, 'r') as fh: + imp.load_module('mach_bootstrap', fh, mach_path, + ('.py', 'r', imp.PY_SOURCE)) + import mach_bootstrap + return mach_bootstrap.bootstrap(dir_path) + + +def check_and_get_mach(dir_path): + bootstrap_paths = ( + 'build/mach_bootstrap.py', + # test package bootstrap + 'tools/mach_bootstrap.py', + ) + for bootstrap_path in bootstrap_paths: + mach_path = os.path.join(dir_path, bootstrap_path) + if os.path.isfile(mach_path): + return load_mach(dir_path, mach_path) + return None + + +def get_mach(): + # Check whether the current directory is within a mach src or obj dir. + for dir_path in ancestors(os.getcwd()): + # If we find a "config.status" and "mozinfo.json" file, we are in the objdir. + config_status_path = os.path.join(dir_path, 'config.status') + mozinfo_path = os.path.join(dir_path, 'mozinfo.json') + if os.path.isfile(config_status_path) and os.path.isfile(mozinfo_path): + import json + info = json.load(open(mozinfo_path)) + if 'mozconfig' in info and 'MOZCONFIG' not in os.environ: + # If the MOZCONFIG environment variable is not already set, set it + # to the value from mozinfo.json. This will tell the build system + # to look for a config file at the path in $MOZCONFIG rather than + # its default locations. + # + # Note: subprocess requires native strings in os.environ on Windows + os.environ[b'MOZCONFIG'] = str(info['mozconfig']) + + if 'topsrcdir' in info: + # Continue searching for mach_bootstrap in the source directory. + dir_path = info['topsrcdir'] + + mach = check_and_get_mach(dir_path) + if mach: + return mach + + # If we didn't find a source path by scanning for a mozinfo.json, check + # whether the directory containing this script is a source directory. We + # follow symlinks so mach can be run even if cwd is outside the srcdir. + return check_and_get_mach(os.path.dirname(os.path.realpath(__file__))) + +def main(args): + mach = get_mach() + if not mach: + print('Could not run mach: No mach source directory found.') + sys.exit(1) + sys.exit(mach.run(args)) + + +if __name__ == '__main__': + if sys.platform == 'win32': + # This is a complete hack to work around the fact that Windows + # multiprocessing needs to import the original module (ie: this + # file), but only works if it has a .py extension. + # + # We do this by a sort of two-level function interposing. The first + # level interposes forking.get_command_line() with our version defined + # in my_get_command_line(). Our version of get_command_line will + # replace the command string with the contents of the fork_interpose() + # function to be used in the subprocess. + # + # The subprocess then gets an interposed imp.find_module(), which we + # hack up to find 'mach' without the .py extension, since we already + # know where it is (it's us!). If we're not looking for 'mach', then + # the original find_module will suffice. + # + # See also: http://bugs.python.org/issue19946 + # And: https://bugzilla.mozilla.org/show_bug.cgi?id=914563 + import inspect + from multiprocessing import forking + global orig_command_line + + def fork_interpose(): + import imp + import os + import sys + orig_find_module = imp.find_module + def my_find_module(name, dirs): + if name == 'mach': + path = os.path.join(dirs[0], 'mach') + f = open(path) + return (f, path, ('', 'r', imp.PY_SOURCE)) + return orig_find_module(name, dirs) + + # Don't allow writing bytecode file for mach module. + orig_load_module = imp.load_module + def my_load_module(name, file, path, description): + # multiprocess.forking invokes imp.load_module manually and + # hard-codes the name __parents_main__ as the module name. + if name == '__parents_main__': + old_bytecode = sys.dont_write_bytecode + sys.dont_write_bytecode = True + try: + return orig_load_module(name, file, path, description) + finally: + sys.dont_write_bytecode = old_bytecode + + return orig_load_module(name, file, path, description) + + imp.find_module = my_find_module + imp.load_module = my_load_module + from multiprocessing.forking import main; main() + + def my_get_command_line(): + fork_code, lineno = inspect.getsourcelines(fork_interpose) + # Remove the first line (for 'def fork_interpose():') and the three + # levels of indentation (12 spaces). + fork_string = ''.join(x[12:] for x in fork_code[1:]) + cmdline = orig_command_line() + cmdline[2] = fork_string + return cmdline + orig_command_line = forking.get_command_line + forking.get_command_line = my_get_command_line + + main(sys.argv[1:]) |