mirror of
https://github.com/openhwgroup/cvw
synced 2025-01-24 13:34:28 +00:00
128 lines
3.6 KiB
Bash
Executable File
128 lines
3.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Exit on any error (return code != 0)
|
|
# set -e
|
|
|
|
# Output colors
|
|
GREEN='\033[1;32m'
|
|
RED='\033[1;31m'
|
|
NC='\033[0m'
|
|
NAME="$GREEN"${0:2}"$NC"
|
|
|
|
# File location variables
|
|
RISCV=/opt/riscv
|
|
IMAGES=$RISCV/buildroot/output/images/
|
|
FW_JUMP=$IMAGES/fw_jump.bin
|
|
LINUX_KERNEL=$IMAGES/Image
|
|
DEVICE_TREE=$IMAGES/wally-vcu108.dtb
|
|
|
|
# Mount Directory
|
|
MNT_DIR=wallyimg
|
|
|
|
if [ "$#" -eq "0" ] ; then
|
|
echo "$NAME: $RED ERROR $NC: You must supply the SD card device."
|
|
echo "usage: ./flash-sd.sh <sd device> <mount directory>"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! -e "$1" ] ; then
|
|
echo "$NAME:$RED ERROR $NC: SD card device does not exist."
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! -z "$2" ] ; then
|
|
MNT_DIR=$2
|
|
fi
|
|
|
|
# If images are not built, exit
|
|
if [ ! -e $FW_JUMP ] || [ ! -e $LINUX_KERNEL ] ; then
|
|
echo 'ERROR: Missing images in buildroot output directory.'
|
|
echo ' Build images before running this script.'
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! -e $DEVICE_TREE ] ; then
|
|
echo 'ERROR: Missing device tree file'
|
|
exit 1
|
|
fi
|
|
|
|
# Size of OpenSBI and the Kernel in 512B blocks
|
|
DST_SIZE=$(ls -la --block-size=512 $DEVICE_TREE | cut -d' ' -f 5 )
|
|
FW_JUMP_SIZE=$(ls -la --block-size=512 $FW_JUMP | cut -d' ' -f 5 )
|
|
KERNEL_SIZE=$(ls -la --block-size=512 $LINUX_KERNEL | cut -d' ' -f 5 )
|
|
|
|
# Start sectors of OpenSBI and Kernel Partitions
|
|
FW_JUMP_START=$(( 34 + $DST_SIZE ))
|
|
KERNEL_START=$(( $FW_JUMP_START + $FW_JUMP_SIZE ))
|
|
FS_START=$(( $KERNEL_START + $KERNEL_SIZE ))
|
|
|
|
# Print out the sizes of the binaries in 512B blocks
|
|
echo -e "$NAME: Device tree block size: $DST_SIZE"
|
|
echo -e "$NAME: OpenSBI FW_JUMP block size: $FW_JUMP_SIZE"
|
|
echo -e "$NAME: Kernel block size: $KERNEL_SIZE"
|
|
|
|
read -p "Warning: Doing this will replace all data on this card.\nContinue? y/n: " -n 1 -r
|
|
echo
|
|
if [[ $REPLY =~ ^[Yy]$ ]] ; then
|
|
DEVBASENAME=$(basename $1)
|
|
CHECKMOUNT=$(lsblk | grep "sdb4" | tr -s ' ' | cut -d' ' -f 7)
|
|
|
|
if [ ! -z $CHECKMOUNT ] ; then
|
|
sudo umount -v $CHECKMOUNT
|
|
fi
|
|
|
|
#Make empty image
|
|
echo -e "$NAME: Creating blank image"
|
|
sudo dd if=/dev/zero of=$1 bs=64k status=progress && sync
|
|
|
|
# GUID Partition Tables (GPT)
|
|
# ===============================================
|
|
# -g Converts any existing mbr record to a gpt record
|
|
# --clear clears any GPT partition table that already exists.
|
|
# --set-alignment=1 that we want to align partition starting sectors
|
|
# to 1 sector boundaries I think? This would normally be set to 2048
|
|
# apparently.
|
|
|
|
sudo sgdisk -z $1
|
|
|
|
sleep 1
|
|
|
|
echo -e "$NAME: Creating GUID Partition Table"
|
|
sudo sgdisk -g --clear --set-alignment=1 \
|
|
--new=1:34:+$DST_SIZE: --change-name=1:'fdt' \
|
|
--new=2:$FW_JUMP_START:+$FW_JUMP_SIZE --change-name=2:'opensbi' --typecode=1:2E54B353-1271-4842-806F-E436D6AF6985 \
|
|
--new=3:$KERNEL_START:+$KERNEL_SIZE --change-name=3:'kernel' \
|
|
--new=4:$FS_START:-0 --change-name=4:'filesystem' \
|
|
$1
|
|
|
|
sudo partprobe $1
|
|
|
|
sleep 3
|
|
|
|
echo -e "$NAME: Copying binaries into their partitions."
|
|
DD_FLAGS="bs=4k iflag=fullblock oflag=direct conv=fsync status=progress"
|
|
|
|
echo -e "$NAME: Copying device tree"
|
|
sudo dd if=$DEVICE_TREE of="$1"1 $DD_FLAGS
|
|
|
|
echo -e "$NAME: Copying OpenSBI"
|
|
sudo dd if=$FW_JUMP of="$1"2 $DD_FLAGS
|
|
|
|
echo -e "$NAME: Copying Kernel"
|
|
sudo dd if=$LINUX_KERNEL of="$1"3 $DD_FLAGS
|
|
|
|
sudo mkfs.ext4 "$1"4
|
|
sudo mkdir /mnt/$MNT_DIR
|
|
|
|
sudo mount -v "$1"4 /mnt/$MNT_DIR
|
|
|
|
sudo umount -v /mnt/$MNT_DIR
|
|
|
|
sudo rmdir /mnt/$MNT_DIR
|
|
#sudo losetup -d $LOOPDEVICE
|
|
fi
|
|
|
|
echo
|
|
echo "GPT Information for $1 ==================================="
|
|
sudo sgdisk -p $1
|