Allow custom repository mirrors and configure chroot for distro updates.

build/install-chroot.sh

 #!/bin/bash -e
 
 # Copyright (c) 2010 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 # This script installs Debian-derived distributions in a chroot environment.
 # It can for example be used to have an accurate 32bit build and test
 # environment when otherwise working on a 64bit machine.
 # N. B. it is unlikely that this script will ever work on anything other than a
 # Debian-derived system.
 
+usage() {
+  echo "usage: ${0##*/} [-m mirror]"
+  echo "-m mirror  an alternate repository mirror for package downloads"
+  echo "-h         this help message"
+}
+
+process_opts() {
+  local OPTNAME OPTIND OPTERR OPTARG
+  while getopts ":m:h" OPTNAME; do
+    case "$OPTNAME" in
+      m)
+        if [ -n "${mirror}" ]; then
+          echo "You can only specify exactly one mirror location"
+          usage
+          exit 1
+        fi
+        mirror="$OPTARG"
+        ;;
+      h)
+        usage
+        exit 0
+        ;;
+      \:)
+        echo "'-$OPTARG' needs an argument."
+        usage
+        exit 1
+        ;;
+      *)
+        echo "invalid command-line option: $OPTARG"
+        usage
+        exit 1
+        ;;
+    esac
+  done
+
+  if [ $# -ge ${OPTIND} ]; then
+    eval echo "Unexpected command line argument: \${${OPTIND}}"
+    usage
+    exit 1
+  fi
+}
+
+
 # Check that we are running as a regular user
 [ "$(id -nu)" = root ] && {
   echo "Run this script as a regular user and provide your \"sudo\""           \
        "password if requested" >&2
   exit 1
 }
 mkdir -p "$HOME/chroot/"
 
+process_opts "$@"
+
 # Error handler
 trap 'exit 1' INT TERM QUIT
 trap 'sudo apt-get clean; tput bel; echo; echo Failed' EXIT
 
 # Install any missing applications that this script relies on. If these packages
 # are already installed, don't force another "apt-get install". That would
 # prevent them from being auto-removed, if they ever become eligible for that.
 # And as this script only needs the packages once, there is no good reason to
 # introduce a hard dependency on things such as dchroot and debootstrap.
 dep=
@@ skipping 43 matching lines @@
 target="${distname}${arch}"
 
 # Don't overwrite an existing installation
 [ -d /var/lib/chroot/"${target}" ] && {
   echo "This chroot already exists on your machine." >&2
   echo "Delete /var/lib/chroot/${target} if you want to start over." >&2
   exit 1
 }
 sudo mkdir -p /var/lib/chroot/"${target}"
 
+# Offer to include additional standard repositories for Ubuntu-based chroots.
+alt_repos=
+grep ubuntu.com /usr/share/debootstrap/scripts/"${distname}" >&/dev/null && {
+  while :; do
+    echo "Would you like to add ${distname}-updates and ${distname}-security "
+    echo -n "to the chroot's sources.list (y/n)? "
+    read alt_repos
+    case "${alt_repos}" in
+      y|Y)
+        alt_repos="y"
+        break
+      ;;
+      n|N)
+        break
+      ;;
+    esac
+  done
+}
+
 # Remove stale entry from /etc/schroot/schroot.conf. Entries start
 # with the target name in square brackets, followed by an arbitrary
 # number of lines. The entry stops when either the end of file has
 # been reached, or when the beginning of a new target is encountered.
 # This means, we cannot easily match for a range of lines in
 # "sed". Instead, we actually have to iterate over each line and check
 # whether it is the beginning of a new entry.
 sudo sed -ni '/^[[]'"${target%bit}"']$/,${:1;n;/^[[]/b2;b1;:2;p;n;b2};p'       \
          /etc/schroot/schroot.conf
 
 # Download base system. This takes some time
-grep ubuntu.com /usr/share/debootstrap/scripts/"${distname}" >&/dev/null &&
- mirror="http://archive.ubuntu.com/ubuntu" ||
- mirror="http://ftp.us.debian.org/debian"
+if [ -z "${mirror}" ]; then
+ grep ubuntu.com /usr/share/debootstrap/scripts/"${distname}" >&/dev/null &&
+   mirror="http://archive.ubuntu.com/ubuntu" ||
+   mirror="http://ftp.us.debian.org/debian"
+fi
  sudo debootstrap ${archflag} "${distname}" /var/lib/chroot/"${target}"        \
                   "$mirror"
 
 # Add new entry to /etc/schroot/schroot.conf
 grep ubuntu.com /usr/share/debootstrap/scripts/"${distname}" >&/dev/null &&
   brand="Ubuntu" || brand="Debian"
 sudo sh -c 'cat >>/etc/schroot/schroot.conf' <<EOF
 [${target%bit}]
 description=${brand} ${distname} ${arch}
 type=directory
@@ skipping 24 matching lines @@
   exec schroot -c ${target%bit} -p
 else
   p="\$1"; shift
   exec schroot -c ${target%bit} -p "\$p" -- "\$@"
 fi
 exit 1
 EOF
 sudo chown root:root /usr/local/bin/"${target%bit}"
 sudo chmod 755 /usr/local/bin/"${target%bit}"
 
+# Add the standard Ubuntu update repositories if requested.
+[ "${alt_repos}" = "y" -a -r "/var/lib/chroot/${target}/etc/apt/sources.list" ] &&
+sudo sed -i '/^deb .* [^ -]\+ main$/p
+             s/^\(deb .* [^ -]\+\) main/\1-security main/
+             p
+             t1
+             d
+             :1;s/-security main/-updates main/
+             t
+             d' "/var/lib/chroot/${target}/etc/apt/sources.list"
+
 # Add a few more repositories to the chroot
-[ -r /var/lib/chroot/${target}/etc/apt/sources.list ] &&
+[ -r "/var/lib/chroot/${target}/etc/apt/sources.list" ] &&
 sudo sed -i 's/ main$/ main restricted universe multiverse/
              p
              t1
              d
           :1;s/^deb/deb-src/
              t
-             d' /var/lib/chroot/${target}/etc/apt/sources.list
+             d' "/var/lib/chroot/${target}/etc/apt/sources.list"
 
 # Update packages
 sudo schroot -c "${target%bit}" -p -- /bin/sh -c '
   apt-get update; apt-get -y dist-upgrade' || :
 
 # Install a couple of missing packages
 for i in debian-keyring ubuntu-keyring locales sudo; do
   [ -d "/var/lib/chroot/${target}/usr/share/doc/$i" ] ||
     sudo schroot -c "${target%bit}" -p -- apt-get -y install "$i" || :
 done
@@ skipping 18 matching lines @@
 # If running a 32bit environment on a 64bit machine, install a few binaries
 # as 64bit.
 if [ "${arch}" = 32bit ] && file /bin/bash 2>/dev/null | grep -q x86-64; then
   sudo schroot -c "${target%bit}" -p -- apt-get -y install                     \
     lib64expat1 lib64ncurses5 lib64readline6 lib64z1
   dep=
   for i in binutils gdb strace; do
     [ -d /usr/share/doc/"$i" ] || dep="$dep $i"
   done
   [ -n "$dep" ] && sudo apt-get -y install $dep
-  sudo cp /usr/bin/gdb /var/lib/chroot/${target}/usr/local/bin/
-  sudo cp /usr/bin/ld /var/lib/chroot/${target}/usr/local/bin/
+  sudo cp /usr/bin/gdb "/var/lib/chroot/${target}/usr/local/bin/"
+  sudo cp /usr/bin/ld "/var/lib/chroot/${target}/usr/local/bin/"
   for i in libbfd libpython; do
     lib="$({ ldd /usr/bin/ld; ldd /usr/bin/gdb; } |
            grep "$i" | awk '{ print $3 }')"
     if [ -n "$lib" -a -r "$lib" ]; then
-      sudo cp "$lib" /var/lib/chroot/${target}/usr/lib64/
+      sudo cp "$lib" "/var/lib/chroot/${target}/usr/lib64/"
     fi
   done
   for lib in libssl libcrypt; do
-    sudo cp /usr/lib/$lib* /var/lib/chroot/${target}/usr/lib64/ || :
+    sudo cp /usr/lib/$lib* "/var/lib/chroot/${target}/usr/lib64/" || :
   done
 fi
 
 # Clean up package files
 sudo schroot -c "${target%bit}" -p -- apt-get clean
 sudo apt-get clean
 
 # Let the user know what we did
 trap '' INT TERM QUIT
 trap '' EXIT
 cat <<EOF
 
 
 Successfully installed ${distname} ${arch}
 
 You can run programs inside of the chroot by invoking the "${target%bit}"
 command.
 
 Your home directory is shared between the host and the chroot. But I configured
 $HOME/chroot to be private to the chroot environment. You can use it
 for files that need to differ between environments.
 EOF