octeon: add new target and support for Ubiquiti EdgeRouter 6P

[openwrt/openwrt.git] / target / linux / octeon / base-files / lib / upgrade / platform.sh

diff --git a/target/linux/octeon/base-files/lib/upgrade/platform.sh b/target/linux/octeon/base-files/lib/upgrade/platform.sh
index 4cb03bc984409680277a8c544fd5ff3644bd203b..255699bb31978725f1d213907b68bb535dee1ba6 100755 (executable)
--- a/target/linux/octeon/base-files/lib/upgrade/platform.sh
+++ b/target/linux/octeon/base-files/lib/upgrade/platform.sh
@@ -2,17 +2,11 @@
 # Copyright (C) 2014 OpenWrt.org
 #
 
-. /lib/functions/octeon.sh
-
 platform_get_rootfs() {
        local rootfsdev
 
        if read cmdline < /proc/cmdline; then
                case "$cmdline" in
-                       *block2mtd=*)
-                               rootfsdev="${cmdline##*block2mtd=}"
-                               rootfsdev="${rootfsdev%%,*}"
-                       ;;
                        *root=*)
                                rootfsdev="${cmdline##*root=}"
                                rootfsdev="${rootfsdev%% *}"
@@ -24,12 +18,21 @@ platform_get_rootfs() {
 }
 
 platform_copy_config() {
-       local board="$(octeon_board_name)"
-
-       case "$board" in
+       case "$(board_name)" in
        erlite)
                mount -t vfat /dev/sda1 /mnt
-               cp -af "$CONF_TAR" /mnt/
+               cp -af "$UPGRADE_BACKUP" "/mnt/$BACKUP_FILE"
+               umount /mnt
+               ;;
+       itus,shield-router)
+               mount -t vfat /dev/mmcblk1p1 /mnt
+               cp -af "$UPGRADE_BACKUP" "/mnt/$BACKUP_FILE"
+               umount /mnt
+               ;;
+       ubnt,edgerouter-4 | \
+       ubnt,edgerouter-6p)
+               mount -t vfat /dev/mmcblk0p1 /mnt
+               cp -af "$UPGRADE_BACKUP" "/mnt/$BACKUP_FILE"
                umount /mnt
                ;;
        esac
@@ -41,36 +44,56 @@ platform_do_flash() {
        local kernel=$3
        local rootfs=$4
 
+       local board_dir=$(tar tf "$tar_file" | grep -m 1 '^sysupgrade-.*/$')
+       board_dir=${board_dir%/}
+       [ -n "$board_dir" ] || return 1
+
        mkdir -p /boot
-       mount -t vfat /dev/$kernel /boot
 
-       [ -f /boot/vmlinux.64 -a ! -L /boot/vmlinux.64 ] && {
-               mv /boot/vmlinux.64 /boot/vmlinux.64.previous
-               mv /boot/vmlinux.64.md5 /boot/vmlinux.64.md5.previous
-       }
+       if [ $board = "itus,shield-router" ]; then
+               # mmcblk1p1 (fat) contains all ELF-bin images for the Shield
+               mount /dev/mmcblk1p1 /boot
+
+               echo "flashing Itus Kernel to /boot/$kernel (/dev/mmblk1p1)"
+               tar -Oxf $tar_file "$board_dir/kernel" > /boot/$kernel
+       else
+               mount -t vfat /dev/$kernel /boot
+
+               [ -f /boot/vmlinux.64 -a ! -L /boot/vmlinux.64 ] && {
+                       mv /boot/vmlinux.64 /boot/vmlinux.64.previous
+                       mv /boot/vmlinux.64.md5 /boot/vmlinux.64.md5.previous
+               }
+
+               echo "flashing kernel to /dev/$kernel"
+               tar xf $tar_file $board_dir/kernel -O > /boot/vmlinux.64
+               md5sum /boot/vmlinux.64 | cut -f1 -d " " > /boot/vmlinux.64.md5
+       fi
 
-       echo "flashing kernel to /dev/$kernel"
-       tar xf $tar_file sysupgrade-$board/kernel -O > /boot/vmlinux.64
-       md5sum /boot/vmlinux.64 | cut -f1 -d " " > /boot/vmlinux.64.md5
        echo "flashing rootfs to ${rootfs}"
-       tar xf $tar_file sysupgrade-$board/root -O | dd of="${rootfs}" bs=4096
+       tar xf $tar_file $board_dir/root -O | dd of="${rootfs}" bs=4096
+
        sync
        umount /boot
 }
 
 platform_do_upgrade() {
        local tar_file="$1"
-       local board=$(octeon_board_name)
+       local board=$(board_name)
        local rootfs="$(platform_get_rootfs)"
        local kernel=
 
        [ -b "${rootfs}" ] || return 1
        case "$board" in
+       er | \
+       ubnt,edgerouter-4 | \
+       ubnt,edgerouter-6p)
+               kernel=mmcblk0p1
+               ;;
        erlite)
                kernel=sda1
                ;;
-       er)
-               kernel=mmcblk0p1
+       itus,shield-router)
+               kernel=ItusrouterImage
                ;;
        *)
                return 1
@@ -79,20 +102,26 @@ platform_do_upgrade() {
        platform_do_flash $tar_file $board $kernel $rootfs
 
        return 0
-       
 }
 
 platform_check_image() {
-       local board=$(octeon_board_name)
+       local board=$(board_name)
+       local tar_file="$1"
+
+       local board_dir=$(tar tf "$tar_file" | grep -m 1 '^sysupgrade-.*/$')
+       board_dir=${board_dir%/}
+       [ -n "$board_dir" ] || return 1
 
        case "$board" in
+       er | \
        erlite | \
-       er)
-               local tar_file="$1"
-               local kernel_length=`(tar xf $tar_file sysupgrade-$board/kernel -O | wc -c) 2> /dev/null`
-               local rootfs_length=`(tar xf $tar_file sysupgrade-$board/root -O | wc -c) 2> /dev/null`
+       itus,shield-router | \
+       ubnt,edgerouter-4 | \
+       ubnt,edgerouter-6p)
+               local kernel_length=$(tar xf $tar_file $board_dir/kernel -O | wc -c 2> /dev/null)
+               local rootfs_length=$(tar xf $tar_file $board_dir/root -O | wc -c 2> /dev/null)
                [ "$kernel_length" = 0 -o "$rootfs_length" = 0 ] && {
-                       echo "The upgarde image is corrupt."
+                       echo "The upgrade image is corrupt."
                        return 1
                }
                return 0