1 files changed, 79 insertions, 0 deletions
diff --git a/alt-rootfs-installer/socs.d/StarFive-riscv64.sh b/alt-rootfs-installer/socs.d/StarFive-riscv64.sh
new file mode 100644
index 0000000..faae8d0
--- /dev/null
+++ b/alt-rootfs-installer/socs.d/StarFive-riscv64.sh
@@ -0,0 +1,79 @@
+#!/bin/bash
+# vim:sw=8:noet
+
+SPL_UID=2E54B353-1271-4842-806F-E436D6AF6985
+UBOOT_UID=BC13C2FF-59E6-4262-A352-B275FD6F7172
+FSBL=$SPL_UID
+BBL=$UBOOT_UID
+
+MEDIATABLE=gpt
+ROOTPART_NUM=1
+START_SECTOR=16 # MiB
+CONSOLE=1
+
+soc_write_bootloader() {
+	unmount_partition
+	[ -n "$ROOTFS" ] || clear_riscv64_bootloader_partition
+
+	print_message "Creating partitions for U-Boot SPL and OpenSBI + U-Boot..."
+	sgdisk \
+		--set-alignment=2 \
+		--new=0:4096:8191 \
+			--change-name=0:spl \
+			--typecode=0:"$SPL_UID" \
+		--new=0:8192:16383 \
+			--change-name=0:uboot \
+			--typecode=0:"$UBOOT_UID" \
+		"$MEDIA" ||
+		fatal_error "failed to create U-Boot SPL and OpenSBI + U-Boot partitions"
+	print_done
+
+	print_message "Checking partition numbers..."
+	SPLPART_NUM="" UBOOTPART_NUM=""
+	local PART_NUMS=$(ls "$MEDIA"* |sed "s:$MEDIA::" |sed 's/^p//')
+	local PART_NUM
+	for PART_NUM in $PART_NUMS; do
+		FIND_PART_CODE=$(sgdisk -i "$PART_NUM" "$MEDIA" |
+			grep 'Partition GUID code:' | cut -d ' ' -f 4)
+		case "$FIND_PART_CODE" in
+			"$SPL_UID") SPLPART_NUM="$PART_NUM";;
+			"$UBOOT_UID") UBOOTPART_NUM="$PART_NUM";;
+		esac
+	done
+	if [[ "$UBOOTPART_NUM" != 2 && "$SPLPART_NUM" == 2 ]]; then
+		sgdisk --transpose=$SPLPART_NUM:$UBOOTPART_NUM "$MEDIA" ||
+			fatal_error "failed to swap U-Boot SPL and OpenSBI + U-Boot partitions"
+		SPLPART_NUM=$UBOOTPART_NUM
+		UBOOTPART_NUM=2
+	elif [[ "$UBOOTPART_NUM" != 2 && "$SPLPART_NUM" != 2 ]]; then
+		print_message "WARNING: Swap $UBOOTPART_NUM with 2-nd partition or else it won't boot!"
+		print_message "         # sgdisk --transpose=2:$UBOOTPART_NUM \"$MEDIA\""
+	fi
+	print_done
+
+	sync_partitions
+	mount_partition
+
+	print_message "Searching for U-Boot files..."
+	# All starfive boards use same binary
+	TARGET=starfive_visionfive2 set_path_to_uboot
+	SPL_BLOB="$(readlink -e "$UBOOT/u-boot-spl.bin.normal.out")"
+	[ -f "${SPL_BLOB-}" ] || fatal_error "${SPL_BLOB-} was not found"
+	UBOOT_BLOB="$(readlink -e "$UBOOT/u-boot.itb")"
+	[ -f "${UBOOT_BLOB-}" ] || fatal_error "${UBOOT_BLOB-}} was not found"
+	print_done
+
+	print_message "Writing U-Boot SPL for ${TARGET-}..."
+	SPLPART="${MEDIA}${partsuffix-}${SPLPART_NUM}"
+	[ -b "${SPLPART-}" ] || fatal_error "${SPLPART-} is not a block device or does not exist"
+	dd if="$SPL_BLOB" of="$SPLPART" bs=4M status=none || fatal_error "failed to write U-Boot SPL"
+	sync
+	print_done
+
+	print_message "Writing OpenSBI + U-Boot for ${TARGET-}..."
+	UBOOTPART="${MEDIA}${partsuffix-}${UBOOTPART_NUM}"
+	[ -b "${UBOOTPART-}" ] || fatal_error "${UBOOTPART-} is not a block device or does not exist"
+	dd if="$UBOOT_BLOB" of="$UBOOTPART" bs=4M status=none || fatal_error "failed to write OpenSBI+U-Boot"
+	sync
+	print_done
+}