main.sh

#!/bin/bash
#ChromeOS - Ubuntu install script for Chromebooks

#Variables definition
#Script variables
script_dir=`dirname "$BASH_SOURCE"`
current_dir="."

verbose=0
set_user=""
set_pass="codestarter"
no_part=false

#Script global directory variables
log_file="ubuntu-install.log"
log_dir="$current_dir/logs/"
tmp_dir="$current_dir/tmp/"
conf_dir="$current_dir/conf.d/"
profiles_dir="$current_dir/profiles/"
devices_dir="$profiles_dir/devices/"
scripts_dir="$current_dir/scripts/"
web_dl_dir="$tmp_dir/web_dl/"

#Default profile
default_profile_file="default.profile"
default_profile_dir="$profiles_dir/default/"
default_sys_dir="$default_profile_dir/system/"
default_scripts_dir="$default_profile_dir/scripts/"

#User profile
user_profile_file="user.profile"
user_profile_dir="$profiles_dir/user/"
user_sys_dir="$user_profile_dir/user/system/"
user_scripts_dir="$user_profile_dir/scripts/"

#Device specific variables
device_profile="none"
dev_profile_file="device.profile"

#External depenencies variables
#ChrUbuntu configuration file
chrubuntu_script="$scripts_dir/chrubuntu-chromeeos.sh"
chrubuntu_runonce="$tmp_dir/chrubuntu_runonce"
system_chroot="/tmp/urfs/"

#distro specific requirements
#A filesystem version of live ISO squashfs content
eos_sys_archive_url="https://s3-us-west-1.amazonaws.com/mojombo-codestarter/ubuntu_system.tar.gz"
if [ -e "$current_dir/../ubuntu_system.tar.gz" ];then
  eos_sys_archive="$current_dir/../ubuntu_system.tar.gz"
else
  eos_sys_archive="$tmp_dir/ubuntu_system.tar.gz"
fi
eos_sys_archive_md5="2a14cd56e0e116e921b064ee2959280a"

#Functions definition
usage(){
cat << EOF
usage: $0 [ OPTIONS ] [ DEVICE_PROFILE | ACTION ]

ChromeOS - Ubuntu installation script for Chromebooks

    OPTIONS:
    -h           Show help
    -v           Enable verbose mode
    -u USERNAME  Create a default user with this name and password "codestarter".
    -p PASSWORD  If a user is specified, use this password instead of default.
    -n           Do not run the partitioning step (useful for repeated installs).

    DEVICE_PROFILE:
        The device profile to load for your Chromebook

    ACTIONS:
        list    List all the elements for this option (ex: List all devices profile supported)
        search  Search for your critera in all devices profile
EOF
}

debug_msg(){
    debug_level="$1"
    msg="$2"
    case $debug_level in
        INFO)
            echo -e "\E[1;32m$msg"
            echo -e '\e[0m'
            ;;
        WARNING)
            echo -e "\E[1;33m$msg"
            echo -e '\e[0m'
            ;;
        ERROR)
            echo -e "\E[1;31m$msg"
            echo -e '\e[0m'
            ;;
        *)
            echo "$msg"
            echo -e '\e[0m'
            ;;
    esac
}

log_msg(){
    if [ -e "$log_dir" ];then
        debug_level="$1"
        msg="$2"
        log_format="$(date +%Y-%m-%dT%H:%M:%S) $debug_level $msg"
        echo "$log_format" >> "$log_dir/$log_file"
        if [ "$debug_level" != "COMMAND" ];then
          debug_msg "$debug_level" "$msg"
        fi
    else
        debug_msg "ERROR" "Log directory $log_dir does not exist...exiting"
        exit 1
    fi
}

run_command(){
    command="$1"
    log_msg "COMMAND" "$command"
    cmd_output=$($command 2>&1)
    if [ "$cmd_output" != "" ];then
        log_msg "COMMAND" "output: $cmd_output"
    fi
}

run_command_chroot(){
  command="$1"
  log_msg "COMMAND" "$command"
  cmd_output=$(sudo chroot $system_chroot /bin/bash -c "$command" 2>&1)
  if [ "$cmd_output" != "" ];then
    log_msg "COMMAND" "output: $cmd_output"
  fi
}

#Get command line arguments
#Required arguments

#Optional arguments
while getopts ":hvu:p:n" option; do
    case $option in
        h)
            usage
            exit 1
            ;;
        v)
            verbose=1
            ;;
        u)
            set_user=$OPTARG
            ;;
        p)
            if [ "$set_user" == "" ];then
              echo "Option -p requires option -u to be specified first." >&2
              exit 1
            fi
            set_pass=$OPTARG
            ;;
        n)
            no_part=true
            ;;
        \?)
            echo "Invalid option: -$OPTARG" >&2
            exit 1
            ;;
        :)
            echo "Option -$OPTARG requires an argument." >&2
            exit 1
            ;;
    esac
done

# Move the index past the options
shift $((OPTIND-1))

device_model="$1"
shift
device_search="$1"

# Default to "acer-720" if no other args are sent
if [ "$device_model" == "" ] && [ "$device_search" = "" ];then
  device_model="acer-c720"
fi

if [ "$device_model" == "search" ];then
    debug_msg "WARNING" "No search critera entered for device profile search...exiting"
    usage
    exit 1
fi

if [ "$device_search" == "search" ];then
    search_result=$(/bin/bash $0 list | tail -n +3 | grep -i "$device_model")
    if [ -z "$search_result" ] || [ "$search_result" == "" ];then
        debug_msg "WARNING" "No device profile found with search critera \"$device_model\""
    else
        debug_msg "INFO" "List of device profile matching search critera \"$device_model\""
        echo $search_result
    fi
    exit 1
fi

#Validate device model
case "$device_model" in
    list)
        debug_msg "INFO" "List of device profiles for supported devices..."
        for i in $(cd $devices_dir; ls -d */); do echo "- ${i%%/}"; done
        exit 0
        ;;
    *)
        device_profile="$devices_dir/$device_model/$dev_profile_file"
        device_profile_dir="$devices_dir/$device_model/"
        device_scripts_dir="$device_profile_dir/scripts/"
        device_sys_dir="$device_profile_dir/system/"
        if [ -z "$device_model" ]; then
            debug_msg "WARNING" "Device not specified...exiting"
            usage
            exit 1
        elif [ ! -e "$device_profile" ];then
            debug_msg "WARNING" "Device '$device_model' profile does not exist...exiting"
            usage
            exit 1
        fi
        ;;
esac

debug_msg "INFO" "Codestarter Chromebook/Ubuntu installer"
#Creating log files directory before using the log_msg function
if [ ! -e "$log_dir" ]; then
    mkdir $log_dir
fi

device_hwid=$(crossystem hwid)
log_msg "INFO" "Device model is $device_model"
log_msg "INFO" "Device hardware ID is $device_hwid"

if [ ! -e "$tmp_dir" ]; then
    log_msg "INFO" "Creating and downloading dependencies..."
    run_command "mkdir $tmp_dir"
fi

if [ $no_part == false ] && [ ! -e "$chrubuntu_runonce" ]; then
    log_msg "INFO" "Running ChrUbuntu to setup partitioning..."
    sudo bash $chrubuntu_script
    log_msg "INFO" "ChrUbuntu execution complete..."
    log_msg "INFO" "System will reboot in 10 seconds..."
    touch $chrubuntu_runonce
    sleep 10
    sudo reboot
    exit 0
else
    log_msg "INFO" "ChrUbuntu partitioning already done...skipping"
    log_msg "INFO" "Running ChrUbuntu to finish the formating process..."
    sudo bash $chrubuntu_script
fi

log_msg "INFO" "Importing device $device_model profile..."
. $device_profile

#Validating that required variables are defined in the device profile
if [ -z "$system_drive" ];then
    log_msg "ERROR" "System drive (system_drive) variable not defined in device profile $device_profile...exiting"
    exit 1
fi

if [ -z "$system_partition" ];then
    log_msg "ERROR" "System partition (system_partition) variable not defined in device profile $device_profile...exiting"
    exit 1
fi

#Verify if the swap file option in specified in the device profile
if [ -z "$swap_file_size" ];then
    log_msg "ERROR" "Swap file size (swap_file_size) variable is not defined in device profile $device_profile...exiting"
    exit 1
fi

if [ ! -e "$system_drive" ];then
    log_msg "ERROR" "System drive $system_drive does not exist...exiting"
    exit 1
fi

if [ ! -e "$system_partition" ];then
    log_msg "ERROR" "System drive $system_partition does not exist...exiting"
    exit 1
fi

log_msg "INFO" "Downloading Ubuntu system files..."
if [ ! -e "$eos_sys_archive" ];then
  curl -o "$eos_sys_archive" -L -O "$eos_sys_archive_url"
else
  log_msg "INFO" "Ubuntu system files are already downloaded...skipping"
fi

log_msg "INFO" "Validating Ubuntu system files archive md5sum..."
eos_sys_archive_dl_md5=$(md5sum $eos_sys_archive | awk '{print $1}')

#MD5 validation of Ubuntu system files archive
if [ "$eos_sys_archive_md5" != "$eos_sys_archive_dl_md5" ];then
  log_msg "ERROR" "Ubuntu system files archive MD5 does not match...exiting"
  run_command "rm $eos_sys_archive"
  log_msg "INFO" "Re-run this script to download the Ubuntu system files archive..."
  exit 1
else
  log_msg "INFO" "Ubuntu system files archive MD5 match...continuing"
fi

log_msg "INFO" "Installing Ubuntu system files to $system_chroot..."
run_command "tar -xf $eos_sys_archive -C $system_chroot"

if [ -e "$default_sys_dir" ];then
    log_msg "INFO" "Copying global system files to $system_chroot..."
    run_command "sudo cp -Rvu $default_sys_dir/. $system_chroot"
else
    log_msg "INFO" "No global system files found...skipping"
fi

if [ -e "$device_sys_dir" ];then
    log_msg "INFO" "Copying device system files to $system_chroot..."
    run_command "sudo cp -Rvu $device_sys_dir/. $system_chroot"
else
    log_msg "INFO" "No device system files found...skipping"
fi

if [ -e "$device_scripts_dir" ];then
    scripts_dir="/tmp/scripts/"
    chroot_dir_scripts="$system_chroot/tmp/scripts/"
    log_msg "INFO" "Copying device scripts to $chroot_dir_scripts..."
    run_command "mkdir -p $chroot_dir_scripts"
    run_command "sudo cp -Rvu $device_scripts_dir/. $chroot_dir_scripts"
else
    log_msg "INFO" "No device scripts found...skipping"
fi

log_msg "INFO" "Mounting dependencies for the chroot..."
run_command "sudo mount -o bind /dev/ $system_chroot/dev/"
run_command "sudo mount -o bind /dev/pts $system_chroot/dev/pts"
run_command "sudo mount -o bind /sys/ $system_chroot/sys/"
run_command "sudo mount -o bind /proc/ $system_chroot/proc/"

log_msg "INFO" "Recording device serial number..."
serial=$(/usr/sbin/dump_vpd_log --full --stdout | grep '"serial_number"' | sed -E 's/^"serial_number"="(.+)"$/\1/')
echo -e "$serial" > $tmp_dir/serialnumber
run_command "sudo mv $tmp_dir/serialnumber $system_chroot/etc/serialnumber"

log_msg "INFO" "Creating /etc/resolv.conf..."
echo -e "nameserver 208.67.222.123\nnameserver 208.67.220.123" > $tmp_dir/resolv.conf
run_command "sudo mv $tmp_dir/resolv.conf $system_chroot/etc/resolv.conf"
system_partition_uuid=$(sudo blkid $system_partition | sed -n 's/.*UUID=\"\([^\"]*\)\".*/\1/p')
log_msg "INFO" "Getting UUID from system partition..."
log_msg "INFO" "Creating /etc/fstab..."
echo -e "proc  /proc nodev,noexec,nosuid  0   0\nUUID=$system_partition_uuid  / ext4  noatime,nodiratime,errors=remount-ro  0   0\n/swap.img  none  swap  sw  0   0" > $tmp_dir/fstab
run_command "sudo mv $tmp_dir/fstab $system_chroot/etc/fstab"

log_msg "INFO" "Adding 14.04 source repo..."
run_command_chroot "echo -e 'deb-src http://archive.ubuntu.com/ubuntu/ trusty main restricted universe \ndeb-src http://archive.ubuntu.com/ubuntu/ trusty-security main restricted universe' |sudo tee -a /etc/apt/sources.list"
run_command_chroot "add-apt-repository universe"

log_msg "INFO" "Installing updates..."
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q update"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q upgrade"

#Device profile validation for the installation of kernel packages from an URL
if [ ! -z "$kernel_url_pkgs" ];then
    kernel_url_pkgs_array=($kernel_url_pkgs)
    kernel_dir="/tmp/kernel/"
    log_msg "INFO" "Downloading and installing kernel package(s) from URL"
    run_command_chroot "mkdir $kernel_dir"
    for kernel_pkg in "${kernel_url_pkgs_array[@]}";do
        run_command_chroot "wget -P $kernel_dir $kernel_pkg"
    done
    run_command_chroot "dpkg -i $kernel_dir/*.deb"
fi

#Device profile validation for the installation additional packages from PPA
if [ ! -z "$ppa_pkgs" ];then
    ppa_pkgs_array=($ppa_pkgs)
    log_msg "INFO" "Installing packages from PPA..."
    for ppa_pkg in "${ppa_pkgs_array[@]}";do
        run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q install $ppa_pkg"
    done
fi

log_msg "INFO" "Creating hosts file..."
echo -e "127.0.0.1  localhost\n127.0.1.1  $system_computer_name\n# The following lines are desirable for IPv6 capable hosts\n::1     ip6-localhost ip6-loopback\nfe00::0 ip6-localnet\nff00::0 ip6-mcastprefix\nff02::1 ip6-allnodes\nff02::2 ip6-allrouters" > $tmp_dir/hosts
run_command "sudo mv $tmp_dir/hosts $system_chroot/etc/hosts"

#Verification for the chroot scripts directory
if [ -e "$chroot_dir_scripts" ];then
    log_msg "INFO" "Executing device scripts..."
    for i in $(cd $chroot_dir_scripts; ls *.sh);do
        run_command_chroot "chmod a+x $scripts_dir/${i%%/}"
        run_command_chroot "/bin/bash -c $scripts_dir/${i%%/}"
    done
fi

log_msg "INFO" "Creating swap file..."
run_command_chroot "fallocate -l $swap_file_size /swap.img"
run_command_chroot "mkswap /swap.img"
run_command_chroot "chown root:root /swap.img"
run_command_chroot "chmod 0600 /swap.img"

log_msg "INFO" "Finishing configuration..."
run_command_chroot "chown root:messagebus /usr/lib/dbus-1.0/dbus-daemon-launch-helper"
run_command_chroot "chmod u+s /usr/lib/dbus-1.0/dbus-daemon-launch-helper"
run_command_chroot "rm /etc/skel/.config/plank/dock1/launchers/ubiquity.dockitem"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q remove gparted"
run_command_chroot "rm -rf /tmp/*"
run_command_chroot "chmod -R 777 /tmp/"

if [ "$set_user" == "" ];then
  # No user specified on command line, set up system to run configuration on
  # first boot.
  log_msg "INFO" "Enabling user and system configuration on first boot..."
  run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q update"
  run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q install oem-config"
  run_command_chroot "touch /var/lib/oem-config/run"
else
  # A user was specified on the command line, set it up.
  log_msg "INFO" "Configuring default user: $set_user..."
  run_command_chroot "useradd -m $set_user -s /bin/bash"
  run_command_chroot "echo $set_user:$set_pass | chpasswd"
  run_command_chroot "adduser $set_user adm"
  run_command_chroot "adduser $set_user sudo"
fi

log_msg "INFO" "Freeing up disk space"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q purge linux-headers-3.13.0-24 linux-headers-3.13.0-24-generic linux-image-3.13.0-24-generic linux-image-extra-3.13.0-24-generic linux-signed-image-3.13.0-24-generic"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q purge firefox firefox-locale-en firefox-locale-es firefox-locale-zh-hans unity-scope-firefoxbookmarks"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q purge gnome-mahjongg gnome-mines gnome-sudoku"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q purge thunderbird"
run_command_chroot "export DEBIAN_FRONTEND=noninteractive; apt-get -y -q clean"

log_msg "INFO" "Removing Memtest from installation."
run_command "sudo rm $system_chroot/boot/memtest86+*"

log_msg "INFO" "Running update-initramfs to create grub boot image..."
run_command_chroot "update-initramfs -c -k all"

log_msg "INFO" "Installing and updating grub to $system_drive..."
run_command_chroot "grub-install $system_drive --force"
run_command_chroot "update-grub"

log_msg "INFO" "Unmounting chroot dependencies and file system..."
run_command "sudo umount $system_chroot/dev/pts"
run_command "sudo umount $system_chroot/dev/"
run_command "sudo umount $system_chroot/sys"
run_command "sudo umount $system_chroot/proc"
run_command "sudo umount $system_chroot"

run_command "rm $tmp_dir/*"

log_msg "INFO" "Codestarter Chromebook/Ubuntu installation completed."
log_msg "INFO" "Press [ENTER] to reboot..."
read
run_command "sudo reboot"