Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 1 | #!/usr/bin/python2 |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 2 | # |
| 3 | # Copyright 2010 Google Inc. All Rights Reserved. |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 4 | """Script to enter the ChromeOS chroot with mounted sources. |
| 5 | |
| 6 | This script enters the chroot with mounted sources. |
| 7 | """ |
| 8 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 9 | from __future__ import print_function |
| 10 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 11 | __author__ = 'asharif@google.com (Ahmad Sharif)' |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 12 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 13 | import argparse |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 14 | import getpass |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 15 | import os |
asharif | 556f4ff | 2013-02-15 04:50:35 +0000 | [diff] [blame] | 16 | import pwd |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 17 | import sys |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 18 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 19 | from cros_utils import command_executer |
| 20 | from cros_utils import logger |
| 21 | from cros_utils import misc |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 22 | |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 23 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 24 | class MountPoint(object): |
| 25 | """Mount point class""" |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 26 | |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 27 | def __init__(self, external_dir, mount_dir, owner, options=None): |
asharif | 6c61913 | 2013-02-15 21:55:28 +0000 | [diff] [blame] | 28 | self.external_dir = os.path.realpath(external_dir) |
| 29 | self.mount_dir = os.path.realpath(mount_dir) |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 30 | self.owner = owner |
| 31 | self.options = options |
| 32 | |
asharif | 556f4ff | 2013-02-15 04:50:35 +0000 | [diff] [blame] | 33 | def CreateAndOwnDir(self, dir_name): |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 34 | retv = 0 |
asharif | 556f4ff | 2013-02-15 04:50:35 +0000 | [diff] [blame] | 35 | if not os.path.exists(dir_name): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 36 | command = 'mkdir -p ' + dir_name |
| 37 | command += ' || sudo mkdir -p ' + dir_name |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 38 | retv = command_executer.GetCommandExecuter().RunCommand(command) |
| 39 | if retv != 0: |
| 40 | return retv |
asharif | 556f4ff | 2013-02-15 04:50:35 +0000 | [diff] [blame] | 41 | pw = pwd.getpwnam(self.owner) |
| 42 | if os.stat(dir_name).st_uid != pw.pw_uid: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 43 | command = 'sudo chown -f ' + self.owner + ' ' + dir_name |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 44 | retv = command_executer.GetCommandExecuter().RunCommand(command) |
| 45 | return retv |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 46 | |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 47 | def DoMount(self): |
asharif | 6c61913 | 2013-02-15 21:55:28 +0000 | [diff] [blame] | 48 | ce = command_executer.GetCommandExecuter() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 49 | mount_signature = '%s on %s' % (self.external_dir, self.mount_dir) |
| 50 | command = 'mount' |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 51 | retv, out, _ = ce.RunCommandWOutput(command) |
asharif | 6c61913 | 2013-02-15 21:55:28 +0000 | [diff] [blame] | 52 | if mount_signature not in out: |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 53 | retv = self.CreateAndOwnDir(self.mount_dir) |
| 54 | logger.GetLogger().LogFatalIf(retv, 'Cannot create mount_dir!') |
| 55 | retv = self.CreateAndOwnDir(self.external_dir) |
| 56 | logger.GetLogger().LogFatalIf(retv, 'Cannot create external_dir!') |
| 57 | retv = self.MountDir() |
| 58 | logger.GetLogger().LogFatalIf(retv, 'Cannot mount!') |
| 59 | return retv |
asharif | 6c61913 | 2013-02-15 21:55:28 +0000 | [diff] [blame] | 60 | else: |
| 61 | return 0 |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 62 | |
asharif | c97199a | 2013-02-15 22:48:45 +0000 | [diff] [blame] | 63 | def UnMount(self): |
| 64 | ce = command_executer.GetCommandExecuter() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 65 | return ce.RunCommand('sudo umount %s' % self.mount_dir) |
asharif | c97199a | 2013-02-15 22:48:45 +0000 | [diff] [blame] | 66 | |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 67 | def MountDir(self): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 68 | command = 'sudo mount --bind ' + self.external_dir + ' ' + self.mount_dir |
| 69 | if self.options == 'ro': |
| 70 | command += ' && sudo mount --bind -oremount,ro ' + self.mount_dir |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 71 | retv = command_executer.GetCommandExecuter().RunCommand(command) |
| 72 | return retv |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 73 | |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 74 | def __str__(self): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 75 | ret = '' |
| 76 | ret += self.external_dir + '\n' |
| 77 | ret += self.mount_dir + '\n' |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 78 | if self.owner: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 79 | ret += self.owner + '\n' |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 80 | if self.options: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 81 | ret += self.options + '\n' |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 82 | return ret |
| 83 | |
| 84 | |
| 85 | def Main(argv, return_output=False): |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 86 | """The main function.""" |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 87 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 88 | parser = argparse.ArgumentParser() |
| 89 | parser.add_argument('-c', |
| 90 | '--chromeos_root', |
| 91 | dest='chromeos_root', |
| 92 | default='../..', |
| 93 | help='ChromeOS root checkout directory.') |
| 94 | parser.add_argument('-t', |
| 95 | '--toolchain_root', |
| 96 | dest='toolchain_root', |
| 97 | help='Toolchain root directory.') |
| 98 | parser.add_argument('-o', |
| 99 | '--output', |
| 100 | dest='output', |
| 101 | help='Toolchain output directory') |
| 102 | parser.add_argument('--sudo', |
| 103 | dest='sudo', |
| 104 | action='store_true', |
| 105 | default=False, |
| 106 | help='Run the command with sudo.') |
| 107 | parser.add_argument('-r', |
| 108 | '--third_party', |
| 109 | dest='third_party', |
| 110 | help='The third_party directory to mount.') |
| 111 | parser.add_argument('-m', |
| 112 | '--other_mounts', |
| 113 | dest='other_mounts', |
| 114 | help='Other mount points in the form: ' |
| 115 | 'dir:mounted_dir:options') |
| 116 | parser.add_argument('-s', |
| 117 | '--mount-scripts-only', |
| 118 | dest='mount_scripts_only', |
| 119 | action='store_true', |
| 120 | default=False, |
| 121 | help='Mount only the scripts dir, and not the sources.') |
| 122 | parser.add_argument('passthrough_argv', nargs='*', |
| 123 | help='Command to be executed inside the chroot.') |
| 124 | |
| 125 | options = parser.parse_args(argv) |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 126 | |
asharif | 0b2f040 | 2013-02-15 04:50:25 +0000 | [diff] [blame] | 127 | chromeos_root = options.chromeos_root |
asharif | 1762130 | 2013-02-15 04:46:35 +0000 | [diff] [blame] | 128 | |
| 129 | chromeos_root = os.path.expanduser(chromeos_root) |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 130 | if options.toolchain_root: |
| 131 | options.toolchain_root = os.path.expanduser(options.toolchain_root) |
asharif | 1762130 | 2013-02-15 04:46:35 +0000 | [diff] [blame] | 132 | |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 133 | chromeos_root = os.path.abspath(chromeos_root) |
| 134 | |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 135 | tc_dirs = [] |
asharif | 642509c | 2013-02-15 09:19:32 +0000 | [diff] [blame] | 136 | if options.toolchain_root is None or options.mount_scripts_only: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 137 | m = 'toolchain_root not specified. Will not mount toolchain dirs.' |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 138 | logger.GetLogger().LogWarning(m) |
| 139 | else: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 140 | tc_dirs = [options.toolchain_root + '/google_vendor_src_branch/gcc', |
| 141 | options.toolchain_root + '/google_vendor_src_branch/binutils'] |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 142 | |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 143 | for tc_dir in tc_dirs: |
| 144 | if not os.path.exists(tc_dir): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 145 | logger.GetLogger().LogError('toolchain path ' + tc_dir + |
| 146 | ' does not exist!') |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 147 | parser.print_help() |
| 148 | sys.exit(1) |
asharif | 0b2f040 | 2013-02-15 04:50:25 +0000 | [diff] [blame] | 149 | |
| 150 | if not os.path.exists(chromeos_root): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 151 | logger.GetLogger().LogError('chromeos_root ' + options.chromeos_root + |
| 152 | ' does not exist!') |
asharif | 0b2f040 | 2013-02-15 04:50:25 +0000 | [diff] [blame] | 153 | parser.print_help() |
| 154 | sys.exit(1) |
| 155 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 156 | if not os.path.exists(chromeos_root + '/src/scripts/build_packages'): |
| 157 | logger.GetLogger( |
| 158 | ).LogError(options.chromeos_root + '/src/scripts/build_packages' |
| 159 | ' not found!') |
asharif | 0b2f040 | 2013-02-15 04:50:25 +0000 | [diff] [blame] | 160 | parser.print_help() |
| 161 | sys.exit(1) |
| 162 | |
asharif | b225e79 | 2013-02-15 21:20:11 +0000 | [diff] [blame] | 163 | version_dir = os.path.realpath(os.path.expanduser(os.path.dirname(__file__))) |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 164 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 165 | mounted_tc_root = '/usr/local/toolchain_root' |
| 166 | full_mounted_tc_root = chromeos_root + '/chroot/' + mounted_tc_root |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 167 | full_mounted_tc_root = os.path.abspath(full_mounted_tc_root) |
raymes | a7d219c | 2013-02-15 04:56:23 +0000 | [diff] [blame] | 168 | |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 169 | mount_points = [] |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 170 | for tc_dir in tc_dirs: |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 171 | last_dir = misc.GetRoot(tc_dir)[1] |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 172 | mount_point = MountPoint(tc_dir, full_mounted_tc_root + '/' + last_dir, |
| 173 | getpass.getuser(), 'ro') |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 174 | mount_points.append(mount_point) |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 175 | |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 176 | # Add the third_party mount point if it exists |
| 177 | if options.third_party: |
| 178 | third_party_dir = options.third_party |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 179 | logger.GetLogger().LogFatalIf( |
| 180 | not os.path.isdir(third_party_dir), |
| 181 | '--third_party option is not a valid dir.') |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 182 | else: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 183 | third_party_dir = os.path.abspath('%s/../../../third_party' % |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 184 | os.path.dirname(__file__)) |
| 185 | |
| 186 | if os.path.isdir(third_party_dir): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 187 | mount_point = MountPoint(third_party_dir, ('%s/%s' % ( |
| 188 | full_mounted_tc_root, os.path.basename(third_party_dir))), |
| 189 | getpass.getuser()) |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 190 | mount_points.append(mount_point) |
kbaclawski | 6999ada | 2013-02-15 19:57:09 +0000 | [diff] [blame] | 191 | |
asharif | 541b639 | 2013-02-15 04:50:38 +0000 | [diff] [blame] | 192 | output = options.output |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 193 | if output is None and options.toolchain_root: |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 194 | # Mount the output directory at /usr/local/toolchain_root/output |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 195 | output = options.toolchain_root + '/output' |
asharif | 01410cc | 2013-02-15 09:19:31 +0000 | [diff] [blame] | 196 | |
| 197 | if output: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 198 | mount_points.append(MountPoint(output, full_mounted_tc_root + '/output', |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 199 | getpass.getuser())) |
| 200 | |
| 201 | # Mount the other mount points |
raymes | a7d219c | 2013-02-15 04:56:23 +0000 | [diff] [blame] | 202 | mount_points += CreateMountPointsFromString(options.other_mounts, |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 203 | chromeos_root + '/chroot/') |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 204 | |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 205 | last_dir = misc.GetRoot(version_dir)[1] |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 206 | |
| 207 | # Mount the version dir (v14) at /usr/local/toolchain_root/v14 |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 208 | mount_point = MountPoint(version_dir, full_mounted_tc_root + '/' + last_dir, |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 209 | getpass.getuser()) |
| 210 | mount_points.append(mount_point) |
| 211 | |
| 212 | for mount_point in mount_points: |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 213 | retv = mount_point.DoMount() |
| 214 | if retv != 0: |
| 215 | return retv |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 216 | |
| 217 | # Finally, create the symlink to build-gcc. |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 218 | command = 'sudo chown ' + getpass.getuser() + ' ' + full_mounted_tc_root |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 219 | retv = command_executer.GetCommandExecuter().RunCommand(command) |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 220 | |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 221 | try: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 222 | CreateSymlink(last_dir + '/build-gcc', full_mounted_tc_root + '/build-gcc') |
| 223 | CreateSymlink(last_dir + '/build-binutils', |
| 224 | full_mounted_tc_root + '/build-binutils') |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 225 | except Exception as e: |
asharif | 0b2f040 | 2013-02-15 04:50:25 +0000 | [diff] [blame] | 226 | logger.GetLogger().LogError(str(e)) |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 227 | |
asharif | 3666653 | 2013-02-15 21:08:14 +0000 | [diff] [blame] | 228 | # Now call cros_sdk --enter with the rest of the arguments. |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 229 | command = 'cd %s/src/scripts && cros_sdk --enter' % chromeos_root |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 230 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 231 | if len(options.passthrough_argv) > 1: |
| 232 | inner_command = ' '.join(options.passthrough_argv[1:]) |
asharif | 51516da | 2013-02-15 04:56:12 +0000 | [diff] [blame] | 233 | inner_command = inner_command.strip() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 234 | if inner_command.startswith('-- '): |
asharif | 51516da | 2013-02-15 04:56:12 +0000 | [diff] [blame] | 235 | inner_command = inner_command[3:] |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 236 | command_file = 'tc_enter_chroot.cmd' |
| 237 | command_file_path = chromeos_root + '/src/scripts/' + command_file |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 238 | retv = command_executer.GetCommandExecuter().RunCommand('sudo rm -f ' + |
| 239 | command_file_path) |
| 240 | if retv != 0: |
| 241 | return retv |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 242 | f = open(command_file_path, 'w') |
asharif | c0f7193 | 2013-02-15 04:56:18 +0000 | [diff] [blame] | 243 | f.write(inner_command) |
| 244 | f.close() |
raymes | a7d219c | 2013-02-15 04:56:23 +0000 | [diff] [blame] | 245 | logger.GetLogger().LogCmd(inner_command) |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 246 | retv = command_executer.GetCommandExecuter().RunCommand('chmod +x ' + |
| 247 | command_file_path) |
| 248 | if retv != 0: |
| 249 | return retv |
asharif | 52284c6 | 2013-02-15 19:59:08 +0000 | [diff] [blame] | 250 | |
| 251 | if options.sudo: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 252 | command += ' sudo ./' + command_file |
asharif | 52284c6 | 2013-02-15 19:59:08 +0000 | [diff] [blame] | 253 | else: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 254 | command += ' ./' + command_file |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 255 | retv = command_executer.GetCommandExecuter().RunCommandGeneric( |
Luis Lozano | 036c923 | 2015-12-10 10:47:01 -0800 | [diff] [blame] | 256 | command, return_output) |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 257 | return retv |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 258 | else: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 259 | os.chdir('%s/src/scripts' % chromeos_root) |
asharif | 3666653 | 2013-02-15 21:08:14 +0000 | [diff] [blame] | 260 | ce = command_executer.GetCommandExecuter() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 261 | _, out, _ = ce.RunCommandWOutput('which cros_sdk') |
asharif | 3666653 | 2013-02-15 21:08:14 +0000 | [diff] [blame] | 262 | cros_sdk_binary = out.split()[0] |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 263 | return os.execv(cros_sdk_binary, ['', '--enter']) |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 264 | |
| 265 | |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 266 | def CreateMountPointsFromString(mount_strings, chroot_dir): |
| 267 | # String has options in the form dir:mount:options |
| 268 | mount_points = [] |
| 269 | if not mount_strings: |
| 270 | return mount_points |
| 271 | mount_list = mount_strings.split() |
| 272 | for mount_string in mount_list: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 273 | mount_values = mount_string.split(':') |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 274 | external_dir = mount_values[0] |
| 275 | mount_dir = mount_values[1] |
raymes | a7d219c | 2013-02-15 04:56:23 +0000 | [diff] [blame] | 276 | if len(mount_values) > 2: |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 277 | options = mount_values[2] |
| 278 | else: |
| 279 | options = None |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 280 | mount_point = MountPoint(external_dir, chroot_dir + '/' + mount_dir, |
asharif | da9ac65 | 2013-02-15 04:50:09 +0000 | [diff] [blame] | 281 | getpass.getuser(), options) |
| 282 | mount_points.append(mount_point) |
| 283 | return mount_points |
asharif | 252df0f | 2013-02-15 04:46:28 +0000 | [diff] [blame] | 284 | |
| 285 | |
asharif | 8a87387 | 2013-02-15 04:56:52 +0000 | [diff] [blame] | 286 | def CreateSymlink(target, link_name): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 287 | logger.GetLogger().LogFatalIf( |
| 288 | target.startswith('/'), "Can't create symlink to absolute path!") |
| 289 | real_from_file = misc.GetRoot(link_name)[0] + '/' + target |
asharif | 8a87387 | 2013-02-15 04:56:52 +0000 | [diff] [blame] | 290 | if os.path.realpath(real_from_file) != os.path.realpath(link_name): |
| 291 | if os.path.exists(link_name): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 292 | command = 'rm -rf ' + link_name |
asharif | 8a87387 | 2013-02-15 04:56:52 +0000 | [diff] [blame] | 293 | command_executer.GetCommandExecuter().RunCommand(command) |
| 294 | os.symlink(target, link_name) |
| 295 | |
| 296 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 297 | if __name__ == '__main__': |
asharif | 2198c51 | 2013-02-15 09:21:35 +0000 | [diff] [blame] | 298 | retval = Main(sys.argv) |
| 299 | sys.exit(retval) |