mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-18 04:44:17 +00:00
6a55c03d43
--HG-- extra : rebase_source : 9f988171182a0abd54724e0cc2906fe678da318f extra : source : 7923a0c837b56efcd1376b95318fe4cdce03d5e8
149 lines
5.7 KiB
Bash
Executable File
149 lines
5.7 KiB
Bash
Executable File
#!/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 "mozinfo.json" file, we are in the objdir.
|
|
mozinfo_path = os.path.join(dir_path, 'mozinfo.json')
|
|
if 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.
|
|
return check_and_get_mach(os.path.dirname(__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:])
|