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