parch-riscv/mkimg

#!/usr/bin/bash
#
# SPDX-FileCopyrightText: 2022 Celeste Liu <CoelacanthusHex@gmail.com>
# SPDX-License-Identifier: GPL-3.0-or-later

# shellcheck disable=1091
. /usr/share/makepkg/util.sh
colorize

verbose=0
use_fixed_password=0
build_firmware=1
varbose_arg=
rootfs="archriscv-$(date --rfc-3339=date).tar.zst"

show_help() {
cat << EOF
Usage: ${0##*/} [-hvfd] [-p PASSWORD] [-r ROOTFS] [FILENAME]
Create Arch RISC-V distro image.

    FILENAME    generated image file name
                default: 'archriscv-$(date --rfc-3339=date).qcow2'
                unless the extension is qcow2, implies raw disk format

    -h          display this help and exit
    -f          use fixed password instead of using systemd-firstboot to ask
    -d          only build the disk image and omit building OpenSBI/U-Boot
    -p PASSWORD set root password to PASSWORD instead of passwd in rootfs
    -r ROOTFS   specify rootfs file name
    -v          verbose mode
EOF
}

parse-args() {
    local OPTIND=1
    while getopts 'hvfdr:p:' opt; do
      case $opt in
        h)  show_help
            exit 0
            ;;
        v)  verbose=$((verbose+1))
            varbose_arg="--verbose"
            ;;
        f)  use_fixed_password=1
            ;;
        d)  build_firmware=0
            ;;
        p)  password=$OPTARG
            ;;
        r)  rootfs=$OPTARG
            ;;
        *)  show_help >&2
            exit 1
            ;;
      esac
  done
  shift "$(( OPTIND-1 ))"
  filename=${1:-archriscv-$(date --rfc-3339=date).qcow2}
}

toggle-systemd-firstboot() {
    msg2 "Toggle systemd-firstboot..."
    sudo rm -f mnt/etc/{machine-id,hostname,shadow}
    sudo systemd-nspawn -D mnt systemctl enable systemd-firstboot.service
}

use-fixed-password() {
    msg2 "Using fixed password... $password"
    [[ -n $password ]] && sudo usermod --root $(realpath ./mnt) --password $(openssl passwd -6 "$password") root
}

parse-args "$@"

if [ $build_firmware == 1 ]
then
    msg "Building U-Boot..."

    if [[ -d u-boot ]]; then
        pushd u-boot
        git checkout master -- ':(top)'
        git checkout master
        git pull --rebase
        git checkout v2023.04
        popd
    else
        git clone --depth 1 -b v2023.04 https://github.com/u-boot/u-boot.git
    fi
    pushd u-boot

    make \
        CROSS_COMPILE=riscv64-linux-gnu- \
        qemu-riscv64_smode_defconfig
    make CROSS_COMPILE=riscv64-linux-gnu-
    popd

    msg "Building OpenSBI..."

    if [[ -d opensbi ]]; then
        pushd opensbi
        git checkout master -- ':(top)'
        git checkout master
        git pull --rebase
        git checkout v2023.04
        popd
    else
        git clone --depth 1 -b v1.2 https://github.com/riscv-software-src/opensbi
    fi
    pushd opensbi

    make \
        CROSS_COMPILE=riscv64-linux-gnu- \
        PLATFORM=generic \
        FW_PAYLOAD_PATH=../u-boot/u-boot.bin
    popd

    cp ./opensbi/build/platform/generic/firmware/fw_payload.bin opensbi_fw_payload.bin
fi

if [[ $filename == *.qcow2 ]]
then
    msg "Create QCOW2 image..."
    qemu-img create -f qcow2 "$filename" 10G
    sudo modprobe nbd max_part=16 || exit 1
    # Possible NBD device collision?
    sudo qemu-nbd -c /dev/nbd0 "$filename"
    loopdev=/dev/nbd0
else
    msg "Create raw image..."
    fallocate "$filename" -l 10G
    loopdev=$(sudo losetup --show -P -f "$filename")
fi

msg "Partitioning..."
(echo n; echo p; echo 1; echo ''; echo ''; echo w) | sudo fdisk "$loopdev"

sudo partprobe "$loopdev"
partdev="$loopdev"p1

sudo mkfs.ext4 "$partdev"
sudo e2label "$partdev" rootfs

sudo mkdir -p mnt
sudo mount "$partdev" mnt
sudo chown root:root mnt

partuuid=$(sudo findmnt mnt -o UUID -n)

msg "Extract rootfs..."

pushd mnt
sudo bsdtar $varbose_arg -kpxf "../$rootfs"
popd

msg "Install kernel package..."

sudo systemd-nspawn -D mnt pacman \
    --noconfirm --needed \
    -Syu linux linux-firmware

sudo mkdir -p mnt/boot/extlinux
cat << EOF | sudo tee mnt/boot/extlinux/extlinux.conf
menu title Arch RISC-V Boot Menu
timeout 100
default linux-fallback

label linux
    menu label Linux linux
    kernel /boot/vmlinuz-linux
    initrd /boot/initramfs-linux.img
    append earlyprintk rw root=UUID=$partuuid rootwait rootfstype=ext4 LANG=en_US.UTF-8 console=ttyS0

label linux-fallback
    menu label Linux linux (fallback initramfs)
    kernel /boot/vmlinuz-linux
    initrd /boot/initramfs-linux-fallback.img
    append earlyprintk rw root=UUID=$partuuid rootwait rootfstype=ext4 LANG=en_US.UTF-8 console=ttyS0
EOF

cat << EOF | sudo tee mnt/etc/systemd/network/default.network
[Match]
Name=en*

[Network]
DHCP=yes
EOF
sudo systemd-nspawn -D mnt systemctl enable systemd-networkd.service

msg "Clean up..."
msg2 "Clean up pacman package cache..."
yes y | sudo pacman \
    --sysroot ./mnt \
    --sync --clean --clean

(( use_fixed_password==0 )) && toggle-systemd-firstboot || use-fixed-password

msg2 "Unmount and sync..."
sudo umount mnt
if [[ $filename == *.qcow2 ]]
then
    sudo qemu-nbd -d "$loopdev"
else
    sudo losetup -d "$loopdev"
fi
sudo sync