source: trunk/openwrt/install.sh

#!/bin/bash
#
# Install DebWrt helper script
#
# Copyright 2016 Johan van Zoomeren <amain@debwrt.net>
#

BASE_DIR=$(dirname ${0})

if [ ! $(id -u) -eq 0 ]
then
    echo "Must be root"
    exit 1
fi

dev=${1}
shift

if [ -n "$@" ]
then
   eval "$@" 2>/dev/null
   if [ ! $? -eq 0 ]
   then
       echo "Incorrect argument"
       exit 1
   fi
fi

swap=${swap:-512}

if [ ! -b "${dev}" ]
then
    cat <<EOF
usage: $(basename ${0}) <block device> [swap=<size>] [root=<size>]

Install DebWrt on <block devices:
* Create Linux partition
* Create swap partition
* Create an ext3 filesystem on the Linux partition
* Create a swap filesysttem on the swap partition
* Extract the root file system image (debwrt-rootfs-*.tar.bz2)
* Extract the modules image (debwrt-modules-*.tar.gz)
* Extract the headers image (debwrt-headers-*.tar.gz)
* Copy firmware image to /boot (*-squashfs-factory.*)

Default swap size: 512 (MB), rest will be used for the root fs partition.

example: $(basename ${0}) /dev/sdb

WARNING: will overwrite everything on <block device>!

EOF
    exit 1
fi

if echo ${dev} | grep -q "[0-9]"
then
    echo "Partition number found in ${dev}, please specify 'raw' block device"
    exit 1
fi

if mount | grep -q ${dev}
then
    echo "Device ${dev} already mounted:"
    mount | grep ${dev}
    exit 1
fi

function get()
{
    local pat=${1}

    find ${BASE_DIR} -name "${pat}" >/tmp/.install.pat.$$
    c=$(wc -l /tmp/.install.pat.$$ | awk '{print $1}')

    if [ ${c} -gt 1 ]
    then
        rm -f /tmp/.install.pat.$$
        echo "Found multiple images for '${pat}', while we can only install one" 1>&2
        exit 1
    fi

    file=$(cat /tmp/.install.pat.$$)

    if [ "${file}" = "" ]
    then
        rm -f /tmp/.install.pat.$$
        echo "No image found for '${pat}'" 1>&2
        exit 1
    fi

    echo $(basename ${file})

    rm -f /tmp/.install.pat.$$
}

function extract()
{
    local image=${1}
    local to=${2}

    ft=$(file -b ${BASE_DIR}/${image} | awk '{print $1}')
    echo "Extracting (${ft}) '${image}'..."

    case ${ft} in
        bzip2) x=j;;
        gzip)  x=z;;
        *)     x=;;
    esac

    if type -p pv &>/dev/null
    then
        pv ${BASE_DIR}/${image} | tar x${x}f - -C ${to}
    else
        tar x${x}f ${BASE_DIR}/${image} -C ${to}
    fi
}

echo "Looking for images in '${BASE_DIR}' ..."
rootfs=$(get 'debwrt-rootfs-*.tar.bz2')  e1=$?
modules=$(get 'debwrt-modules-*.tar.gz') e2=$?
headers=$(get 'debwrt-headers-*.tar.gz') e3=$?
firmware=$(get '*-squashfs-factory.*')   e4=$?
[ $((e1+e2+e3+e4)) -gt 0 ] && exit 1

if [ "${root}" = "" ]
then
    size=$(sfdisk -l ${dev} 2>&1 \
             | grep '^Disk.*bytes' \
             | sed 's/.* \([0-9]\+\) bytes.*/\1/')
    swapb=$((swap * 1024 * 1024))
    root=$(((size - swapb) / 1024 / 1024))
fi

cat <<EOF

Intalling to   : ${dev}
      root(MiB): ${root}
      swap(MiB): ${swap}

Root filesystem: ${rootfs}
Modules        : ${modules}
Headers        : ${headers}
Firmware       : ${firmware}

EOF

read -p "Are you sure to install DebWrt to ${dev} [Y/N]? " a

if [ ! "${a^^}" = "Y" ]
then
    echo "Aborting"
    exit 1
fi

echo
echo "Installing DebWrt...."
echo

cat >/tmp/.install_ptable.$$ <<EOF
,${root}Mib,L
,,L
EOF
sfdisk ${dev} </tmp/.install_ptable.$$
rm -f /tmp/.install_ptable.$$

partprobe -s ${dev}

mkfs.ext3 -F -L DEBWRT_ROOT ${dev}1
mkswap -f -L DEBWRT_SWAP ${dev}2

echo

function _umount()
{
    echo
    echo "Unmounting ${dev}1, this may take a while...."
    umount ${BASE_DIR}/mnt &>/dev/null
    rmdir ${BASE_DIR}/mnt &>/dev/null
    echo
    echo "done"
}

trap _umount EXIT
mkdir -p ${BASE_DIR}/mnt
mount ${dev}1 ${BASE_DIR}/mnt

extract ${rootfs} ${BASE_DIR}/mnt   ; echo
extract ${modules} ${BASE_DIR}/mnt  ; echo
extract ${headers} ${BASE_DIR}/mnt  ; echo
echo -n "Copy firmware (${firmware}) to /boot..."
cp ${firmware} ${BASE_DIR}/mnt/boot
echo "done"

echo
echo -n "Calculate module dependencies (depmod)..."
depmod -b ${BASE_DIR}/mnt \
       -a $(find ${BASE_DIR}/mnt/lib/modules/ \
                 -maxdepth 1 \
                 -mindepth 1 \
                 -type d \
                 -printf "%f\n" \
             | head -1)
echo "done"