flash: Add Pixel 8 (shiba)

HLOS_IMAGES_TARGET := boot.img \

    init_boot.img \

    dtbo.img \

    vendor_boot.img \

    vendor_kernel_boot.img \

    vbmeta_system.img \

    vbmeta_vendor.img \

    vbmeta.img \

    super.img

#!/usr/bin/env bash

##########

# This script is created and maintained by

#       Bharath(@teamb58).org

# Feel free to connect for any queries or suggestions.

##########

##########

# This script flashes a software release and complete wipes the device.

#

##########

set -e

set -u

# Target device info

PRODUCT="Pixel 8"

PRODUCT_ID="shiba"

# Paths/files

ROOT_DIR="$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"

IMAGES_DIR="${ROOT_DIR}"

# Abort the script (and wait for a key to be pressed).

abort_now() {

  echo ""

  read -rp "ERROR: Aborting now (press Enter to terminate)." a

  exit 1

}

# Check for connected phone

find_device() {

  echo "INFO: Looking for connected device(s)..."

  DEVICE_FOUND="false"

  while [ ${DEVICE_FOUND} = "false" ]

  do

    serial_numbers=

    for sn in $("${FASTBOOT_BIN}" devices | grep fastboot | grep -oE '^[[:alnum:]]+')

    do

      # Checking the product ID

      PRODUCT_STRING=$("${FASTBOOT_BIN}" -s "${sn}" getvar product 2>&1)

      # Add serial, if product matches

      if [[ ${PRODUCT_STRING} == *"${PRODUCT_ID}"* ]] || [[ ${PRODUCT_STRING} == *"${PRODUCT_ID_OLD}"* ]]

      then

        serial_numbers="${serial_numbers} $sn"

      fi

    done

    case $(echo "${serial_numbers}" | wc -w | grep -oE '[0-9]+') in

      0)

        echo ""

        echo "WARNING: No ${PRODUCT} found in fastboot mode."

        echo "WARNING: Make sure that a ${PRODUCT} is connected."

        ;;

      1)

          echo "INFO: One ${PRODUCT} in fastboot mode found (serial number: ${sn})."

        DEVICE_FOUND="true"

        break

        ;;

      *)

        echo ""

        echo "WARNING: Several ${PRODUCT}'s in fastboot mode connected."

        echo "WARNING: Please connect only one ${PRODUCT}."

        ;;

    esac

    echo ""

    while true

    do

      read -rp "Do you want to look for a ${PRODUCT} again? [(Y)es/(n)o]: " a

      if [ -z "${a}" ] || [ "${a}" = 'y' ] || [ "${a}" = 'Y' ]

      then

        break

      elif [ "${a}" = 'n' ] || [ "${a}" = 'N' ]

      then

        exit 0

      fi

    done

  done

}

# Flash (or manipulate) relevant partitions

flash_fw() {

  flash_image_ab_or_abort "${sn}" abl "${IMAGES_DIR}/abl.img"

  flash_image_ab_or_abort "${sn}" bl1 "${IMAGES_DIR}/bl1.img"

  flash_image_ab_or_abort "${sn}" bl2 "${IMAGES_DIR}/bl2.img"

  flash_image_ab_or_abort "${sn}" bl31 "${IMAGES_DIR}/bl31.img"

  flash_image_ab_or_abort "${sn}" gcf "${IMAGES_DIR}/gcf.img"

  flash_image_ab_or_abort "${sn}" gsa "${IMAGES_DIR}/gsa.img"

  flash_image_ab_or_abort "${sn}" gsa_bl1 "${IMAGES_DIR}/gsa_bl1.img"

  flash_image_ab_or_abort "${sn}" ldfw "${IMAGES_DIR}/ldfw.img"

  flash_image_ab_or_abort "${sn}" modem "${IMAGES_DIR}/modem.img"

  flash_image_ab_or_abort "${sn}" pbl "${IMAGES_DIR}/pbl.img"

  flash_image_ab_or_abort "${sn}" tzsw "${IMAGES_DIR}/tzsw.img"

  "$FASTBOOT_BIN" -s "${sn}" reboot bootloader

}

flash_device() {

  flash_image_ab_or_abort "${sn}" boot "${IMAGES_DIR}/boot.img"

  flash_image_ab_or_abort "${sn}" init_boot "${IMAGES_DIR}/init_boot.img"

  flash_image_ab_or_abort "${sn}" dtbo "${IMAGES_DIR}/dtbo.img"

  flash_image_ab_or_abort "${sn}" vendor_kernel_boot "${IMAGES_DIR}/vendor_kernel_boot.img"

  flash_image_ab_or_abort "${sn}" vendor_boot "${IMAGES_DIR}/vendor_boot.img"

  flash_image_ab_or_abort "${sn}" vbmeta_system "${IMAGES_DIR}/vbmeta_system.img"

  flash_image_ab_or_abort "${sn}" vbmeta_vendor "${IMAGES_DIR}/vbmeta_vendor.img"

  flash_image_ab_or_abort "${sn}" vbmeta "${IMAGES_DIR}/vbmeta.img"

  flash_image_or_abort "${sn}" super "${IMAGES_DIR}/super.img"

  "$FASTBOOT_BIN" -s "${sn}" erase userdata

  "$FASTBOOT_BIN" -s "${sn}" erase metadata

  "$FASTBOOT_BIN" -s "${sn}" --set-active=a

}

# Flash an image to a partition. Abort on failure.

# Arguments: <device serial number> <partition name> <image file>

flash_image_or_abort() {

  local retval=0

  "$FASTBOOT_BIN" -s "${1}" flash "${2}" "${3}" || retval=$?

  if [ "${retval}" -ne 0 ]

  then

    echo ""

    echo "ERROR: Could not flash the ${2} partition on device ${1}."

    echo ""

    echo "ERROR: Please unplug the phone, take the battery out, boot the device into"

    echo "ERROR: fastboot mode, and start this script again."

    echo "ERROR: (To get to fastboot mode, press Volume-Down and plug in the USB-C)"

    echo "ERROR: (cable until the fastboot menu appears.)"

    abort_now

  fi

}

# Flash an image to both A and B slot of partition. Abort on failure.

# Arguments: <device serial number> <partition name without slot> <image file>

flash_image_ab_or_abort() {

  flash_image_or_abort "${1}" "${2}_a" "${3}"

  flash_image_or_abort "${1}" "${2}_b" "${3}"

}

# Operating system checks and variable definition

os_checks() {

  case "$(uname -s 2> /dev/null)" in

    Linux|GNU/Linux)

      echo "INFO: You are using a Linux distribution."

      FASTBOOT_BIN="${ROOT_DIR}/bin-linux-x86/fastboot"

      ;;

    msys|MINGW*)

      echo "INFO: You are using MinGW on Windows"

      FASTBOOT_BIN="${ROOT_DIR}/bin-msys/fastboot.exe"

      ;;

    *)

      echo "ERROR: Unsupported operating system (${OSTYPE})."

      echo "ERROR: Only GNU/Linux, and MinGW on Windows are currently supported."

      abort_now

      ;;

  esac

}

# Control the reboot sequence

reboot_device() {

  echo "-----------"

  echo ""

  echo "INFO: Done. The device will reboot now."

  "${FASTBOOT_BIN}" -s "${sn}" reboot

  echo ""

  echo "INFO: You can unplug the USB cable now."

  echo ""

}

echo ""

echo "*** ${PRODUCT} flashing script ***"

echo ""

echo "INFO: The procedure will start soon. Please wait..."

echo "Note that this will detect and flash only on ${PRODUCT} device."

sleep 2

# Begin with some OS checks and variable definition

os_checks

# Call function to look for device(s)

# If only one device is found $sn will store its serial number

find_device

# Flash the fw

flash_fw

# Flash the device

flash_device

# Reboot device

reboot_device