EdgeMax Recovery Kit Customized Script

emrk-reinstall-custom.sh

#!/bin/sh
#
# emrk-reinstall: EdgeOS installer
#
# Maintainer: Daniil Baturin <daniil at baturin dot org>
#
# Copyright (C) 2013 SO3Group
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
#

# It's time to reinstall EdgeOS!
# EdgeOS won't reinstall itself!

DEV=/dev/sdc
BOOT=/dev/sdc1
ROOT=/dev/sdc2
BOOT_MNT_DIR=/root/mnt/boot
ROOT_MNT_DIR=/root/mnt/root
TMP_DIR=$ROOT_MNT_DIR/tmp
W_DIR=w

# Release tarball file names
KERNEL_ORIG=vmlinux.tmp
KERNEL_ORIG_MD5=vmlinux.tmp.md5
SQUASHFS_ORIG=squashfs.tmp
SQUASHFS_MD5_ORIG=squashfs.tmp.md5
VERSION_ORIG=version.tmp

# Target file names
KERNEL=vmlinux.64
KERNEL_MD5=vmlinux.64.md5
SQUASHFS=squashfs.img
SQUASHFS_MD5=squashfs.img.md5
VERSION=version

PARTED=/sbin/parted
YESNO=/bin/yesno

## Scary disclaimer
##echo "WARNING: This script will reinstall EdgeOS from scratch"
##echo "If you have any usable data on your router storage,"
##echo "it will be irrecoverably destroyed!"
##echo "Do you want to continue?"
##$YESNO

##if [ $? == 1 ]; then
##    exit 0
##fi

# Umount USB stick filesystems that could be mounted at boot time
if mount | grep $BOOT > /dev/null; then
    echo "Unmounting boot partition"
    umount $BOOT
fi

if mount | grep $ROOT > /dev/null; then
    echo "Unmounting root partition"
    umount $ROOT
fi

## Repartition

# Remove everything
echo "Re-creating partition table"
$PARTED --script $DEV mktable msdos

# Boot
echo "Creating boot partition"
$PARTED --script $DEV mkpart primary fat32 1 150MB
echo "Formatting boot partition"
mkfs.vfat $BOOT

# Root
echo "Creating root partition"
$PARTED --script $DEV mkpart primary ext3 150MB 1900MB
echo "Formatting root partition"
mkfs.ext3 -q $ROOT

## Mount partitions
echo "Mounting boot parition"
mount -t vfat $BOOT $BOOT_MNT_DIR

echo "Mounting root partition"
mount -t ext3 $ROOT $ROOT_MNT_DIR

## Download image

##while true; do
##    read -p "Enter EdgeOS image url: " URL
##    curl -o $ROOT_MNT_DIR/tmp/edgeos.tar $URL
##    if [ $? == 0 ]; then
##        break
##    else
##        echo "Could not download EdgeOS image, try again!"
##    fi
##done
mkdir $ROOT_MNT_DIR/tmp
cp edgeos.tar $ROOT_MNT_DIR/tmp/edgeos.tar


## Reinstall

# Unpack image
echo "Unpacking EdgeOS release image"
tar xf $TMP_DIR/edgeos.tar -C $TMP_DIR

# The kernel
echo "Verifying EdgeOS kernel"
if [ `md5sum $TMP_DIR/$KERNEL_ORIG | awk -F ' ' '{print $1}'` != `cat $TMP_DIR/$KERNEL_ORIG_MD5` ]; then
    echo "Kernel from your image is corrupted! Check your image and start over."
    exit 1
fi

echo "Copying EdgeOS kernel to boot partition"
cp $TMP_DIR/$KERNEL_ORIG $BOOT_MNT_DIR/$KERNEL
cp $TMP_DIR/$KERNEL_ORIG_MD5 $BOOT_MNT_DIR/$KERNEL_MD5

# The image
echo "Verifying EdgeOS system image"
if [ `md5sum $TMP_DIR/$SQUASHFS_ORIG | awk -F ' ' '{print $1}'` != `cat $TMP_DIR/$SQUASHFS_MD5_ORIG` ]; then
    echo "System image from your image is corrupted! Check your image and start over."
    exit 1
fi

echo "Copying EdgeOS system image to root partition"
mv $TMP_DIR/$SQUASHFS_ORIG $ROOT_MNT_DIR/$SQUASHFS
mv $TMP_DIR/$SQUASHFS_MD5_ORIG $ROOT_MNT_DIR/$SQUASHFS_MD5

echo "Copying version file to the root partition"
mv $TMP_DIR/$VERSION_ORIG $ROOT_MNT_DIR/$VERSION

# Writable data dir
echo "Creating EdgeOS writable data directory"
mkdir $ROOT_MNT_DIR/$W_DIR

## Cleanup
echo "Cleaning up"
rm -rf $TMP_DIR

echo "Installation finished"

Replace DEV, BOOT, ROOT at the top with whatever your USB stick is. Save the edgemax firmware image as edgeos.tar in the same folder as this script and run it. Will re-image the USB stick without needing tftp.