aboutsummaryrefslogtreecommitdiff
path: root/tc_enter_chroot.py
diff options
context:
space:
mode:
authorLuis Lozano <llozano@chromium.org>2015-12-15 13:49:30 -0800
committerLuis Lozano <llozano@chromium.org>2015-12-16 17:36:06 +0000
commitf2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbe (patch)
tree185d243c7eed7c7a0db6f0e640746cadc1479ea9 /tc_enter_chroot.py
parent2a66f70fef907c1cb15229cb58e5129cb620ac98 (diff)
downloadtoolchain-utils-f2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbe.tar.gz
Run pyformat on all the toolchain-utils files.
This gets rid of a lot of lint issues. Ran by doing this: for f in *.py; do echo -n "$f " ; if [ -x $f ]; then pyformat -i --remove_trailing_comma --yapf --force_quote_type=double $f ; else pyformat -i --remove_shebang --remove_trailing_comma --yapf --force_quote_type=double $f ; fi ; done BUG=chromium:567921 TEST=Ran simple crosperf run. Change-Id: I59778835fdaa5f706d2e1765924389f9e97433d1 Reviewed-on: https://chrome-internal-review.googlesource.com/242031 Reviewed-by: Luis Lozano <llozano@chromium.org> Commit-Queue: Luis Lozano <llozano@chromium.org> Tested-by: Luis Lozano <llozano@chromium.org> Reviewed-by: Yunlian Jiang <yunlian@google.com>
Diffstat (limited to 'tc_enter_chroot.py')
-rwxr-xr-xtc_enter_chroot.py195
1 files changed, 102 insertions, 93 deletions
diff --git a/tc_enter_chroot.py b/tc_enter_chroot.py
index c68c48a2..39bb7dc4 100755
--- a/tc_enter_chroot.py
+++ b/tc_enter_chroot.py
@@ -1,13 +1,12 @@
#!/usr/bin/python
#
# Copyright 2010 Google Inc. All Rights Reserved.
-
"""Script to enter the ChromeOS chroot with mounted sources.
This script enters the chroot with mounted sources.
"""
-__author__ = "asharif@google.com (Ahmad Sharif)"
+__author__ = 'asharif@google.com (Ahmad Sharif)'
import getpass
import optparse
@@ -22,93 +21,101 @@ from utils import misc
class MountPoint:
+
def __init__(self, external_dir, mount_dir, owner, options=None):
self.external_dir = os.path.realpath(external_dir)
self.mount_dir = os.path.realpath(mount_dir)
self.owner = owner
self.options = options
-
def CreateAndOwnDir(self, dir_name):
retval = 0
if not os.path.exists(dir_name):
- command = "mkdir -p " + dir_name
- command += " || sudo mkdir -p " + dir_name
+ command = 'mkdir -p ' + dir_name
+ command += ' || sudo mkdir -p ' + dir_name
retval = command_executer.GetCommandExecuter().RunCommand(command)
if retval != 0:
return retval
pw = pwd.getpwnam(self.owner)
if os.stat(dir_name).st_uid != pw.pw_uid:
- command = "sudo chown -f " + self.owner + " " + dir_name
+ command = 'sudo chown -f ' + self.owner + ' ' + dir_name
retval = command_executer.GetCommandExecuter().RunCommand(command)
return retval
-
def DoMount(self):
ce = command_executer.GetCommandExecuter()
- mount_signature = "%s on %s" % (self.external_dir, self.mount_dir)
- command = "mount"
+ mount_signature = '%s on %s' % (self.external_dir, self.mount_dir)
+ command = 'mount'
retval, out, err = ce.RunCommandWOutput(command)
if mount_signature not in out:
retval = self.CreateAndOwnDir(self.mount_dir)
- logger.GetLogger().LogFatalIf(retval, "Cannot create mount_dir!")
+ logger.GetLogger().LogFatalIf(retval, 'Cannot create mount_dir!')
retval = self.CreateAndOwnDir(self.external_dir)
- logger.GetLogger().LogFatalIf(retval, "Cannot create external_dir!")
+ logger.GetLogger().LogFatalIf(retval, 'Cannot create external_dir!')
retval = self.MountDir()
- logger.GetLogger().LogFatalIf(retval, "Cannot mount!")
+ logger.GetLogger().LogFatalIf(retval, 'Cannot mount!')
return retval
else:
return 0
-
def UnMount(self):
ce = command_executer.GetCommandExecuter()
- return ce.RunCommand("sudo umount %s" % self.mount_dir)
-
+ return ce.RunCommand('sudo umount %s' % self.mount_dir)
def MountDir(self):
- command = "sudo mount --bind " + self.external_dir + " " + self.mount_dir
- if self.options == "ro":
- command += " && sudo mount --bind -oremount,ro " + self.mount_dir
+ command = 'sudo mount --bind ' + self.external_dir + ' ' + self.mount_dir
+ if self.options == 'ro':
+ command += ' && sudo mount --bind -oremount,ro ' + self.mount_dir
retval = command_executer.GetCommandExecuter().RunCommand(command)
return retval
-
def __str__(self):
- ret = ""
- ret += self.external_dir + "\n"
- ret += self.mount_dir + "\n"
+ ret = ''
+ ret += self.external_dir + '\n'
+ ret += self.mount_dir + '\n'
if self.owner:
- ret += self.owner + "\n"
+ ret += self.owner + '\n'
if self.options:
- ret += self.options + "\n"
+ ret += self.options + '\n'
return ret
def Main(argv, return_output=False):
"""The main function."""
parser = optparse.OptionParser()
- parser.add_option("-c", "--chromeos_root", dest="chromeos_root",
- default="../..",
- help="ChromeOS root checkout directory.")
- parser.add_option("-t", "--toolchain_root", dest="toolchain_root",
- help="Toolchain root directory.")
- parser.add_option("-o", "--output", dest="output",
- help="Toolchain output directory")
- parser.add_option("--sudo", dest="sudo",
- action="store_true",
+ parser.add_option('-c',
+ '--chromeos_root',
+ dest='chromeos_root',
+ default='../..',
+ help='ChromeOS root checkout directory.')
+ parser.add_option('-t',
+ '--toolchain_root',
+ dest='toolchain_root',
+ help='Toolchain root directory.')
+ parser.add_option('-o',
+ '--output',
+ dest='output',
+ help='Toolchain output directory')
+ parser.add_option('--sudo',
+ dest='sudo',
+ action='store_true',
default=False,
- help="Run the command with sudo.")
- parser.add_option("-r", "--third_party", dest="third_party",
- help="The third_party directory to mount.")
- parser.add_option("-m", "--other_mounts", dest="other_mounts",
- help="Other mount points in the form: " +
- "dir:mounted_dir:options")
- parser.add_option("-s", "--mount-scripts-only",
- dest="mount_scripts_only",
- action="store_true",
+ help='Run the command with sudo.')
+ parser.add_option('-r',
+ '--third_party',
+ dest='third_party',
+ help='The third_party directory to mount.')
+ parser.add_option(
+ '-m',
+ '--other_mounts',
+ dest='other_mounts',
+ help='Other mount points in the form: ' + 'dir:mounted_dir:options')
+ parser.add_option('-s',
+ '--mount-scripts-only',
+ dest='mount_scripts_only',
+ action='store_true',
default=False,
- help="Mount only the scripts dir, and not the sources.")
+ help='Mount only the scripts dir, and not the sources.')
passthrough_argv = []
(options, passthrough_argv) = parser.parse_args(argv)
@@ -123,79 +130,78 @@ def Main(argv, return_output=False):
tc_dirs = []
if options.toolchain_root is None or options.mount_scripts_only:
- m = "toolchain_root not specified. Will not mount toolchain dirs."
+ m = 'toolchain_root not specified. Will not mount toolchain dirs.'
logger.GetLogger().LogWarning(m)
else:
- tc_dirs = [options.toolchain_root + "/google_vendor_src_branch/gcc",
- options.toolchain_root + "/google_vendor_src_branch/binutils"]
+ tc_dirs = [options.toolchain_root + '/google_vendor_src_branch/gcc',
+ options.toolchain_root + '/google_vendor_src_branch/binutils']
for tc_dir in tc_dirs:
if not os.path.exists(tc_dir):
- logger.GetLogger().LogError("toolchain path " +
- tc_dir + " does not exist!")
+ logger.GetLogger().LogError('toolchain path ' + tc_dir +
+ ' does not exist!')
parser.print_help()
sys.exit(1)
if not os.path.exists(chromeos_root):
- logger.GetLogger().LogError("chromeos_root " + options.chromeos_root +
- " does not exist!")
+ logger.GetLogger().LogError('chromeos_root ' + options.chromeos_root +
+ ' does not exist!')
parser.print_help()
sys.exit(1)
- if not os.path.exists(chromeos_root + "/src/scripts/build_packages"):
- logger.GetLogger().LogError(options.chromeos_root +
- "/src/scripts/build_packages"
- " not found!")
+ if not os.path.exists(chromeos_root + '/src/scripts/build_packages'):
+ logger.GetLogger(
+ ).LogError(options.chromeos_root + '/src/scripts/build_packages'
+ ' not found!')
parser.print_help()
sys.exit(1)
version_dir = os.path.realpath(os.path.expanduser(os.path.dirname(__file__)))
- mounted_tc_root = "/usr/local/toolchain_root"
- full_mounted_tc_root = chromeos_root + "/chroot/" + mounted_tc_root
+ mounted_tc_root = '/usr/local/toolchain_root'
+ full_mounted_tc_root = chromeos_root + '/chroot/' + mounted_tc_root
full_mounted_tc_root = os.path.abspath(full_mounted_tc_root)
mount_points = []
for tc_dir in tc_dirs:
last_dir = misc.GetRoot(tc_dir)[1]
- mount_point = MountPoint(tc_dir, full_mounted_tc_root + "/" + last_dir,
- getpass.getuser(), "ro")
+ mount_point = MountPoint(tc_dir, full_mounted_tc_root + '/' + last_dir,
+ getpass.getuser(), 'ro')
mount_points.append(mount_point)
# Add the third_party mount point if it exists
if options.third_party:
third_party_dir = options.third_party
- logger.GetLogger().LogFatalIf(not os.path.isdir(third_party_dir),
- "--third_party option is not a valid dir.")
+ logger.GetLogger().LogFatalIf(
+ not os.path.isdir(third_party_dir),
+ '--third_party option is not a valid dir.')
else:
- third_party_dir = os.path.abspath("%s/../../../third_party" %
+ third_party_dir = os.path.abspath('%s/../../../third_party' %
os.path.dirname(__file__))
if os.path.isdir(third_party_dir):
- mount_point = MountPoint(third_party_dir,
- ("%s/%s" %
- (full_mounted_tc_root,
- os.path.basename(third_party_dir))),
- getpass.getuser())
+ mount_point = MountPoint(third_party_dir, ('%s/%s' % (
+ full_mounted_tc_root, os.path.basename(third_party_dir))),
+ getpass.getuser())
mount_points.append(mount_point)
output = options.output
if output is None and options.toolchain_root:
# Mount the output directory at /usr/local/toolchain_root/output
- output = options.toolchain_root + "/output"
+ output = options.toolchain_root + '/output'
if output:
- mount_points.append(MountPoint(output, full_mounted_tc_root + "/output",
+ mount_points.append(MountPoint(output, full_mounted_tc_root + '/output',
getpass.getuser()))
# Mount the other mount points
mount_points += CreateMountPointsFromString(options.other_mounts,
- chromeos_root + "/chroot/")
+ chromeos_root + '/chroot/')
last_dir = misc.GetRoot(version_dir)[1]
# Mount the version dir (v14) at /usr/local/toolchain_root/v14
- mount_point = MountPoint(version_dir, full_mounted_tc_root + "/" + last_dir,
+ mount_point = MountPoint(version_dir, full_mounted_tc_root + '/' + last_dir,
getpass.getuser())
mount_points.append(mount_point)
@@ -205,49 +211,52 @@ def Main(argv, return_output=False):
return retval
# Finally, create the symlink to build-gcc.
- command = "sudo chown " + getpass.getuser() + " " + full_mounted_tc_root
+ command = 'sudo chown ' + getpass.getuser() + ' ' + full_mounted_tc_root
retval = command_executer.GetCommandExecuter().RunCommand(command)
try:
- CreateSymlink(last_dir + "/build-gcc", full_mounted_tc_root + "/build-gcc")
- CreateSymlink(last_dir + "/build-binutils", full_mounted_tc_root + "/build-binutils")
+ CreateSymlink(last_dir + '/build-gcc', full_mounted_tc_root + '/build-gcc')
+ CreateSymlink(last_dir + '/build-binutils',
+ full_mounted_tc_root + '/build-binutils')
except Exception as e:
logger.GetLogger().LogError(str(e))
# Now call cros_sdk --enter with the rest of the arguments.
- command = "cd %s/src/scripts && cros_sdk --enter" % chromeos_root
+ command = 'cd %s/src/scripts && cros_sdk --enter' % chromeos_root
if len(passthrough_argv) > 1:
- inner_command = " ".join(passthrough_argv[1:])
+ inner_command = ' '.join(passthrough_argv[1:])
inner_command = inner_command.strip()
- if inner_command.startswith("-- "):
+ if inner_command.startswith('-- '):
inner_command = inner_command[3:]
- command_file = "tc_enter_chroot.cmd"
- command_file_path = chromeos_root + "/src/scripts/" + command_file
- retval = command_executer.GetCommandExecuter().RunCommand("sudo rm -f " + command_file_path)
+ command_file = 'tc_enter_chroot.cmd'
+ command_file_path = chromeos_root + '/src/scripts/' + command_file
+ retval = command_executer.GetCommandExecuter().RunCommand('sudo rm -f ' +
+ command_file_path)
if retval != 0:
return retval
- f = open(command_file_path, "w")
+ f = open(command_file_path, 'w')
f.write(inner_command)
f.close()
logger.GetLogger().LogCmd(inner_command)
- retval = command_executer.GetCommandExecuter().RunCommand("chmod +x " + command_file_path)
+ retval = command_executer.GetCommandExecuter().RunCommand('chmod +x ' +
+ command_file_path)
if retval != 0:
return retval
if options.sudo:
- command += " sudo ./" + command_file
+ command += ' sudo ./' + command_file
else:
- command += " ./" + command_file
+ command += ' ./' + command_file
retval = command_executer.GetCommandExecuter().RunCommandGeneric(
command, return_output)
return retval
else:
- os.chdir("%s/src/scripts" % chromeos_root)
+ os.chdir('%s/src/scripts' % chromeos_root)
ce = command_executer.GetCommandExecuter()
- _, out, _ = ce.RunCommandWOutput("which cros_sdk")
+ _, out, _ = ce.RunCommandWOutput('which cros_sdk')
cros_sdk_binary = out.split()[0]
- return os.execv(cros_sdk_binary, ["", "--enter"])
+ return os.execv(cros_sdk_binary, ['', '--enter'])
def CreateMountPointsFromString(mount_strings, chroot_dir):
@@ -257,30 +266,30 @@ def CreateMountPointsFromString(mount_strings, chroot_dir):
return mount_points
mount_list = mount_strings.split()
for mount_string in mount_list:
- mount_values = mount_string.split(":")
+ mount_values = mount_string.split(':')
external_dir = mount_values[0]
mount_dir = mount_values[1]
if len(mount_values) > 2:
options = mount_values[2]
else:
options = None
- mount_point = MountPoint(external_dir, chroot_dir + "/" + mount_dir,
+ mount_point = MountPoint(external_dir, chroot_dir + '/' + mount_dir,
getpass.getuser(), options)
mount_points.append(mount_point)
return mount_points
def CreateSymlink(target, link_name):
- logger.GetLogger().LogFatalIf(target.startswith("/"),
- "Can't create symlink to absolute path!")
- real_from_file = misc.GetRoot(link_name)[0] + "/" + target
+ logger.GetLogger().LogFatalIf(
+ target.startswith('/'), "Can't create symlink to absolute path!")
+ real_from_file = misc.GetRoot(link_name)[0] + '/' + target
if os.path.realpath(real_from_file) != os.path.realpath(link_name):
if os.path.exists(link_name):
- command = "rm -rf " + link_name
+ command = 'rm -rf ' + link_name
command_executer.GetCommandExecuter().RunCommand(command)
os.symlink(target, link_name)
-if __name__ == "__main__":
+if __name__ == '__main__':
retval = Main(sys.argv)
sys.exit(retval)