#!/bin/sh

sysctl kern.geom.debugflags=0x10

DISK_MD=mmcsd0

UBOOT_DIR="u-boot-khadas-v"		# Mainline U-boot
UBOOT_DIR="u-boot-kobol-helios64"	# Mainline U-boot
UBOOT_DIR="u-boot-pinebook-pro"		# 
UBOOT_DIR="u-boot-nanopc-t4"		# 
UBOOT_DIR="u-boot-nanopi-m4"		# 
UBOOT_DIR="u-boot-nanopi-r4s"		# 
UBOOT_DIR="u-boot-nanopi-r2s"		# 
UBOOT_DIR="u-boot-rock-pi-4"		# 
UBOOT_DIR="u-boot-rock-pi-4c"		# 
UBOOT_DIR="u-boot-rock-pi-e"		# 
UBOOT_DIR="u-boot-rock-pi-n10a"		# 
UBOOT_DIR="u-boot-roc-rk3328-pc"	# 
UBOOT_DIR="u-boot-roc-rk3399-pc-plus"	# 

# scp -r 192.168.1.111:/usr/local/share/u-boot .
# UBOOT_PATH="/root/u-boot/${UBOOT_DIR}"

UBOOT_PATH="/usr/local/share/u-boot/${UBOOT_DIR}"

rewrite_uboot ()
{
        # Mainline U-Boot for Rockchip
        if [ -f ${UBOOT_PATH}/u-boot.itb ]; then
        	echo "Installing U-Boot ${UBOOT_DIR} to ${DISK_MD}"
                dd if=${UBOOT_PATH}/idbloader.img of=/dev/${DISK_MD} conv=sync bs=512 seek=64
                dd if=${UBOOT_PATH}/u-boot.itb    of=/dev/${DISK_MD} conv=sync bs=512 seek=16384
        fi

        # BSP U-Boot for Rockchip
        if [ -f ${UBOOT_PATH}/trust.img ]; then
        	echo "Installing BSP U-Boot ${UBOOT_DIR} to ${DISK_MD}"
                dd if=${UBOOT_PATH}/idbloader.img of=/dev/${DISK_MD} conv=sync bs=512 seek=64
                dd if=${UBOOT_PATH}/uboot.img     of=/dev/${DISK_MD} conv=sync bs=512 seek=16384
                dd if=${UBOOT_PATH}/trust.img     of=/dev/${DISK_MD} conv=sync bs=512 seek=24576
        fi

        # Mainline U-Boot for Allwinner
        if [ -f ${UBOOT_PATH}/u-boot-sunxi-with-spl.bin ]; then
        	echo "Installing U-Boot ${UBOOT_DIR} to ${DISK_MD}"
		dd if=${UBOOT_PATH}/u-boot-sunxi-with-spl.bin of=/dev/${DISK_MD} bs=1k seek=8 conv=sync
	fi

        if [ -f ${UBOOT_PATH}/u-boot-spi.bin ]; then
		if [ -e /dev/${DISK_MD}p1 ]; then
			mount_msdosfs /dev/${DISK_MD}p1 /boot/efi
			cp ${UBOOT_PATH}/u-boot-spi.bin /boot/efi
			umount /boot/efi
		fi
	fi

        if [ -f ${UBOOT_PATH}/splash.bmp ]; then
		if [ -e /dev/${DISK_MD}p1 ]; then
			mount_msdosfs /dev/${DISK_MD}p1 /boot/efi
			cp ${UBOOT_PATH}/splash.bmp /boot/efi
			umount /boot/efi
		fi
	fi
}

# Main

if [ -e /dev/${DISK_MD} ] ; then
	rewrite_uboot
fi