1 files changed, 55 insertions, 0 deletions
diff --git a/target/linux/ipq806x/base-files/lib/upgrade/asrock.sh b/target/linux/ipq806x/base-files/lib/upgrade/asrock.sh
new file mode 100644
index 0000000000..71cac3faac
--- /dev/null
+++ b/target/linux/ipq806x/base-files/lib/upgrade/asrock.sh
@@ -0,0 +1,55 @@
+. /lib/functions.sh
+
+
+asrock_bootconfig_mangle() {
+	local mtdnum="$( find_mtd_index 0:BOOTCONFIG )"
+
+	if [ -z "$mtdnum" ]; then
+		echo "cannot find bootconfig mtd partition"
+		return 1
+	fi
+	dd if=/dev/mtd$mtdnum of=/tmp/mtd$mtdnum bs=1k
+
+	local partition_byte="$(dd if=/tmp/mtd$mtdnum bs=1 skip=52 count=1)"
+	local upgrade_byte="$(dd if=/tmp/mtd$mtdnum bs=1 skip=4 count=1)"
+
+	if [ $1 = "bootcheck" ]; then
+		if [ ! -s $upgrade_byte ]; then
+			dd if=/dev/mtd$mtdnum of=/tmp/mtd$mtdnum bs=1k
+			printf '\x00' | dd of=/tmp/mtd$mtdnum conv=notrunc bs=1 seek=4
+			printf '\x00' | dd of=/tmp/mtd$mtdnum conv=notrunc bs=1 seek=56
+		else
+			return 1
+		fi
+	elif [ $1 = "sysupgrade" ]; then
+		printf '\x01' | dd of=/tmp/mtd$mtdnum conv=notrunc bs=1 seek=4
+		printf '\x01' | dd of=/tmp/mtd$mtdnum conv=notrunc bs=1 seek=56
+	fi
+
+	if [ -s $partition_byte ]; then
+		printf '\x01' | dd of=/tmp/mtd$mtdnum conv=notrunc bs=1 seek=52
+	else
+		printf '\x00' | dd of=/tmp/mtd$mtdnum conv=notrunc bs=1 seek=52
+	fi
+
+	mtd write /tmp/mtd$mtdnum /dev/mtd$mtdnum
+	return 0
+}
+
+asrock_upgrade_prepare() {
+	local ubidev="$( nand_find_ubi ubi )"
+
+	#Set upgrade flag. If something goes wrong, router will boot with
+	#factory firmware.
+	asrock_bootconfig_mangle 'sysupgrade'
+
+	if [ $? -ne 0 ]; then
+		echo "cannot find bootconfig mtd partition"
+		exit 1
+	fi
+
+	# Just delete these partitions if present and use
+	# OpenWrt's standard names for those.
+	ubirmvol /dev/$ubidev -N ubi_rootfs &> /dev/null || true
+	ubirmvol /dev/$ubidev -N ubi_rootfs_data &> /dev/null || true
+}