haos-x728.sh

#!/bin/sh
#
# Hack the haos image to include my stuff...
#
# Create a custom handler so that RAUC OTA updates include
# customizations
#
set -euf -o pipefail
die() {
  echo "$@" 1>&2
  exit $1
}

usage() {
  if [ $# -gt 0 ] ; then
    echo "$@" 1>&2
    echo '---' 1>&2
  fi
  cat 1>&2 <<-_EOV_
	Usage: $0 [options] {orig-img} {modified-img}

	Options:
	  --[no-]ssh : Enable/disable ssh
	  --force : Force file overwrite
	  --rootfs : root directory or tarball
	_EOV_
  exit 1
}

( type sqfs2tar && type gensquashfs ) || die 21 "Need to install squashfs-tools-ng"
[ $(id -u) -ne 0 ] && usage "Must be run as root"

[ $# -eq 0 ] && usage

summarize() {
  set +x
  while read -r L
  do
    printf '\r'
    echo -n "$L"
    printf '\033[K'
  done
  printf '\rDone\033[K\r\n'
}
gsqfsout() {
  set +x
  while read -r L
  do
    if (echo "$L" | grep -q packing) ; then
      echo "$L"
    else
      exec 1>&2
      echo "$L"
      exec cat
    fi
  done
}


ssh=/
force=false
rootfs=''

while [ $# -gt 0 ] ; do
  case "$1" in
    --ssh) ssh=/ ;;
    --ssh=*) ssh=${1#--ssh=} ;;
    --no-ssh) ssh= ;;
    --force|-F) force=true ;;
    --rootfs=*) rootfs=${1#--rootfs=} ;;
    --root=*) rootfs=${1#--root=} ;;
    *) break ;;
  esac
  shift
done

[ $# -ne 2 ] && usage
[ -z "$rootfs" ] && usage "Not --rootfs specified"

srcimg="$1"
modimg="$2"

if [ -f "$modimg" ] ; then
  if $force ; then
    echo "$modimg: already exists.  Overwrite" 1>&2
  else
    die 84 "$modimg: already exists"
  fi
fi

w=$(mktemp -d) ; trap "umount $w/mnt || : ; rm -rf $w; kpartx -dv \$modimg" EXIT

if (echo "$srcimg" | grep -q '\.xz$' ) ; then
  xz -d -v < "$srcimg"  > "$modimg"
else
  cp -av "$srcimg" "$modimg"
fi

mkdir $w/mnt
kpartx -av "$modimg"

bootfs=/dev/mapper/loop0p1
mount -o loop $bootfs $w/mnt
if [ -f $w/mnt/config.txt ] ; then
  echo "Enable i2c bus"
  (
    echo '[all]'
    echo 'dtparam=i2c_arm=on'
  ) | tee -a $w/mnt/config.txt | summarize
fi
if [ -n "$ssh" ]; then
  auth_keys=""
  if [ x"$ssh" = x"/" ] ; then
    if [ -n "${SUDO_USER:-}" ] ; then
      userhome=$(getent passwd $SUDO_USER | cut -d: -f6)
      if [ -f "$userhome/.ssh/authorized_keys" ] ; then
        auth_keys="$userhome/.ssh/authorized_keys"
      fi
    fi
    if [ -z "$auth_keys" ] && [ -f "$HOME/.ssh/authorized_keys" ] ; then
      auth_keys="$HOME/.ssh/authorized_keys"
    fi
  elif [ -f "$ssh" ] ; then
    auth_keys="$ssh"
  elif [ -f "$ssh/.ssh/authorized_keys" ] ; then
    auth_keys="$ssh/.ssh/authorized_keys"
  elif [ -f "$ssh/authorized_keys" ] ; then
    auth_keys="$ssh/authorized_keys"
  fi
  if [ -z "$auth_keys" ] ; then
    echo "Unable to enable SSH as no authorized_keys found"
  else
    echo "Enabling SSH"
    mkdir -p $w/mnt/CONFIG
    cp "$auth_keys" $w/mnt/CONFIG
  fi
fi
umount $w/mnt

systemfs=/dev/mapper/loop0p3
mkdir -p $w/rootfs
echo "Unpack $systemfs (squashfs)"
sqfs2tar $systemfs | tar -C $w/rootfs -xvf - 2>&1 | summarize

# add the post-install handler
# sed -i -e '/^\[system\]$/a post-install=/lib/rauc/post-install'
echo 'Post install handler:'
( tee -a $w/rootfs/etc/rauc/system.conf ) <<-_EOF_
	[handlers]
	post-install=/lib/rauc/post-install
	_EOF_

if [ -f "$rootfs" ] ; then
  # So this is a tarball
  echo "Unpacking tarball"
  mkdir -p "$w/srcfs"
  tar -C "$w/srcfs" -xvf "$rootfs" 2>&1 | summarize
  rootfs="$w/srcfs"
fi

echo "Copying custom files"
find $rootfs '(' -type f -o -type l ')'| sed -e "s!$rootfs/!!" | tee $w/rootfs/etc/rauc/custom-files | (while read fpath
do
  dir=$(dirname "$fpath")
  mkdir -vp "$w/rootfs/$dir"
  cp -av "$rootfs/$fpath" "$w/rootfs/$dir" 2>&1
done) | summarize

echo "Re-squashing FS"
gensquashfs -D "$w/rootfs" "$w/new-rootfs.img" 2>&1 | gsqfsout | summarize
dd bs=64k if=$w/new-rootfs.img of=$systemfs