execute-all.sh

#!/usr/bin/env bash
#
#  Generate AOSP compatible vendor data for provided device & buildID
#

set -e # fail on unhandled error
set -u # fail on undefined variable
#set -x # debug

readonly SCRIPTS_ROOT="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

# Helper script to download Nexus factory images from web
readonly DOWNLOAD_SCRIPT="$SCRIPTS_ROOT/scripts/download-nexus-image.sh"

# Helper script to extract system & vendor images data
readonly EXTRACT_SCRIPT="$SCRIPTS_ROOT/scripts/extract-factory-images.sh"

# Helper script to de-optimize bytecode prebuilts
readonly REPAIR_SCRIPT="$SCRIPTS_ROOT/scripts/system-img-repair.sh"

# Helper script to generate vendor AOSP includes & makefiles
readonly VGEN_SCRIPT="$SCRIPTS_ROOT/scripts/generate-vendor.sh"

export JAVA_HOME=/usr/local/java/jdk1.8.0_71/bin/java
export PATH=/usr/local/java/jdk1.8.0_71/bin/:$PATH

declare -a sysTools=("mkdir" "java")

# angler is wip
declare -a availDevices=("bullhead" "flounder" "angler")

abort() {
  exit $1
}

usage() {
cat <<_EOF
  Usage: $(basename $0) [options]
    OPTIONS:
      -d|--device   : Device codename (angler, bullhead, etc.)
      -b|--buildID  : BuildID string (e.g. MMB29P)
      -o|--output   : Path to save generated vendor data
      -i|--imgs-tar : Read factory tar from file instead of downloading (optional)
      -k|--keep     : Keep all factory images extracted & de-optimized data (optional)
_EOF
  abort 1
}

command_exists() {
  type "$1" &> /dev/null ;
}

# Check that system tools exist
for i in "${sysTools[@]}"
do
  if ! command_exists $i; then
    echo "[-] '$i' command not found"
    abort 1
  fi
done

DEVICE=""
BUILDID=""
OUTPUT_DIR=""
INPUT_IMGS_TAR=""
KEEP_DATA=false
HOST_OS=""
DEV_ALIAS=""

while [[ $# > 0 ]]
do
  arg="$1"
  case $arg in
    -o|--output)
      OUTPUT_DIR=$(echo "$2" | sed 's:/*$::')
      shift
      ;;
    -d|--device)
      DEVICE=$(echo $2 | tr '[:upper:]' '[:lower:]')
      shift
      ;;
    -b|--buildID)
      BUILDID=$(echo $2 | tr '[:upper:]' '[:lower:]')
      shift
      ;;
    -i|--imgs-tar)
      INPUT_IMGS_TAR=$2
      shift
      ;;
    -k|--keep)
      KEEP_DATA=true
      ;;
    *)
      echo "[-] Invalid argument '$1'"
      usage
      ;;
  esac
  shift
done

if [[ "$DEVICE" == "" ]]; then
  echo "[-] device codename cannot be empty"
  usage
fi
if [[ "$BUILDID" == "" ]]; then
  echo "[-] buildID cannot be empty"
  usage
fi
if [[ "$OUTPUT_DIR" == "" || ! -d "$OUTPUT_DIR" ]]; then
  echo "[-] Output directory not found"
  usage
fi
if [[ "$INPUT_IMGS_TAR" != "" && ! -f "$INPUT_IMGS_TAR" ]]; then
  echo "[-] '$INPUT_IMGS_TAR' file not found"
  abort 1
fi

# Adjust hosts tools based on OS
HOST_OS=$(uname)
if [[ "$HOST_OS" != "Linux" && "$HOST_OS" != "Darwin" ]]; then
  echo '[-] '$HOST_OS' OS is not supported'
  abort 1
fi

# Check if supported device
deviceOK=false
for devNm in "${availDevices[@]}"  
do
  if [[ "$devNm" == "$DEVICE" ]]; then
    deviceOK=true
  fi
done
if [ "$deviceOK" = false ]; then
  echo "[-] '$DEVICE' is not supported"
  abort 1
fi

# Prepare output dir structure
OUT_BASE="$OUTPUT_DIR/$DEVICE/$BUILDID"
if [ ! -d "$OUT_BASE" ]; then
  mkdir -p $OUT_BASE
fi
FACTORY_IMGS_DATA="$OUT_BASE/factory_imgs_data"
FACTORY_IMGS_R_DATA="$OUT_BASE/factory_imgs_repaired_data"
echo "[*] Setting output base to '$OUT_BASE'"

# Factory image alias for devices with naming incompatibilities with AOSP
if [[ "$DEVICE" == "flounder" ]]; then
  DEV_ALIAS="volantis"
else
  DEV_ALIAS="$DEVICE"
fi

# Download images if not provided
if [[ "$INPUT_IMGS_TAR" == "" ]]; then
  if ! $DOWNLOAD_SCRIPT --device $DEVICE --alias $DEV_ALIAS --buildID $BUILDID --output "$OUT_BASE"; then
    echo "[-] Images download failed"
    abort 1
  fi
  archName="$(find $OUT_BASE -iname "*$DEV_ALIAS*$BUILDID*.tgz" | head -1)"
else
  archName="$INPUT_IMGS_TAR"
fi

# Clear old data if present & extract data from factory images
if [ -d "$FACTORY_IMGS_DATA" ]; then
  rm -rf "$FACTORY_IMGS_DATA"/*
else
  mkdir -p "$FACTORY_IMGS_DATA"
fi
if ! $EXTRACT_SCRIPT --input "$archName" --output "$FACTORY_IMGS_DATA" \
     --simg2img "$SCRIPTS_ROOT/hostTools/$HOST_OS/simg2img"; then
  echo "[-] Factory images data extract failed"
  abort 1
fi

# De-optimize bytecode from system partition
if [ -d "$FACTORY_IMGS_R_DATA" ]; then
  rm -rf "$FACTORY_IMGS_R_DATA"/*
else
  mkdir -p "$FACTORY_IMGS_R_DATA"
fi
if ! $REPAIR_SCRIPT --input "$FACTORY_IMGS_DATA/system" --output "$FACTORY_IMGS_R_DATA" \
     --oat2dex "$SCRIPTS_ROOT/hostTools/Java/oat2dex.jar"; then
  echo "[-] System partition de-optimization failed"
  abort 1
fi

# Bytecode under vendor partition doesn't require de-opt for (up to now)
# However, move it to repaired data directory to hava a single source for
# next script
mv "$FACTORY_IMGS_DATA/vendor" "$FACTORY_IMGS_R_DATA"

# Copy vendor partition image size as saved from $EXTRACT_SCRIPT script
# $VGEN_SCRIPT will fail over to last known working default if image size
# file not found when parsing data
cp "$FACTORY_IMGS_DATA/vendor_partition_size" "$FACTORY_IMGS_R_DATA"

if ! $VGEN_SCRIPT --input "$FACTORY_IMGS_R_DATA" --output "$OUT_BASE" \
  --blobs-list "$SCRIPTS_ROOT/$DEVICE/proprietary-blobs.txt"; then
  echo "[-] Vendor generation failed"
  abort 1
fi

if [ "$KEEP_DATA" = false ]; then
  rm -rf "$FACTORY_IMGS_DATA"
  rm -rf "$FACTORY_IMGS_R_DATA"
fi

echo "[*] All actions completed successfully"
echo "[*] Import "$OUT_BASE/vendor" to AOSP root"

abort 0