This repository has been archived on 2023-08-20. You can view files and clone it, but cannot push or open issues or pull requests.
lsts_glued/mkdisk.bash
2019-08-14 12:29:36 +01:00

345 lines
9.1 KiB
Bash
Executable File

#! /bin/bash
###########################################################################
# GLUED: GNU/Linux Uniform Environment Distribution #
# Copyright (C) 2007-2019 Universidade do Porto - Faculdade de Engenharia #
# Laboratório de Sistemas e Tecnologia Subaquática (LSTS) #
###########################################################################
# This program is free software; you can redistribute it and/or modify #
# it under the terms of the GNU General Public License as published by #
# the Free Software Foundation; either version 2 of the License, or (at #
# your option) any later version. #
# #
# This program is distributed in the hope that it will be useful, but #
# WITHOUT ANY WARRANTY; without even the implied warranty of #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU #
# General Public License for more details. #
# #
# You should have received a copy of the GNU General Public License #
# along with this program; if not, write to the Free Software #
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA #
# 02110-1301 USA. #
###########################################################################
# Author: Ricardo Martins #
###########################################################################
cmd_parted()
{
bdev="$1"; shift
"$cmd_parted" "$bdev" -a cylinder -s -- $@
}
cmd_mount()
{
fs="$1"
bdev="$2"
mkdir -p mount &&
mount -t "$fs" "$bdev" mount
}
cmd_unmount()
{
while [ 1 ]; do
(umount mount && rmdir mount) > /dev/null 2>&1
[ $? -eq 0 ] && break
sleep 1
done
}
die()
{
umount -v mount > /dev/null 2>&1
rmdir mount > /dev/null 2>&1
[ -n "$dev_loop" ] && losetup -d "$dev_loop"
exit 1
}
create_part_xboot()
{
nfo1 "X-Boot partition ($part_label)"
nfo2 "Creating partition"
cmd_parted "$dev_loop" \
mkpart primary fat32 "$part_start" "$part_end" \
set "$part_nr" boot on \
align-check minimal "$part_nr" \
|| die
nfo2 "Creating filesystem"
$cmd_mkdosfs -n "$part_label" "$part_dev" > /dev/null || die
nfo2 "Populating filesystem"
cmd_mount vfat "$part_dev" || die
for f in MLO u-boot.img uEnv.txt kernel; do
if [ -f "$cfg_sys_family/rootfs/boot/$f" ]; then
nfo2 installing $f to boot partition
cp "$cfg_sys_family/rootfs/boot/$f" mount || die
fi
done
cmd_parted "$dev_loop" \
set "$part_nr" lba on \
|| die
cmd_parted "$dev_loop" print > /dev/null
}
create_part_rpiboot()
{
nfo1 "RPi-Boot partition ($part_label)"
nfo2 "Creating partition"
cmd_parted "$dev_loop" \
mkpart primary fat32 "$part_start" "$part_end" \
set "$part_nr" boot on \
align-check minimal "$part_nr" \
|| die
nfo2 "Creating filesystem"
$cmd_mkdosfs -n "$part_label" "$part_dev" > /dev/null || die
nfo2 "Populating filesystem"
cmd_mount vfat "$part_dev" || die
for f in "$cfg_sys_family/rootfs/boot/"*; do
nfo2 installing $f to boot partition
cp -r "$f" mount || die
done
nfo2 renaming kernel to kernel7.img
mv mount/kernel mount/kernel7.img
if [ -f mount/board.dtb ]; then
dtb=$(basename "$cfg_target_linux_dtb")
nfo2 renaming board.dtb to "$dtb"
mv "mount/board.dtb" "mount/$dtb"
fi
cmd_parted "$dev_loop" \
set "$part_nr" lba on \
|| die
cmd_parted "$dev_loop" print > /dev/null
}
create_part_root()
{
nfo1 "Root partition ($part_label)"
nfo2 "Creating partition"
cmd_parted "$dev_loop" \
mkpart primary ext2 "$part_start" "$part_end" \
align-check minimal "$part_nr" \
|| die
nfo2 "Creating filesystem"
"$cmd_mkfs_ext2" -q -F -L "$part_label" "$part_dev" > /dev/null 2>&1 || die
nfo2 "Populating filesystem"
cmd_mount ext2 "$part_dev" || die
tar -C mount -pxf "$cfg_rootfs_tar" || die
# Install bootloader.
if [ -f "$cfg_sys_family/toolchain/bin/extlinux" ]; then
nfo2 "Installing extlinux to root partition"
"$cfg_sys_family/toolchain/bin/extlinux" -S 63 -H 255 -i mount/boot/extlinux || die
fi
cmd_unmount || die
# Mark partition bootable if needed.
if [ -f "$cfg_sys_family/toolchain/bin/extlinux" ]; then
cmd_parted "$dev_loop" \
set "$part_nr" boot on \
|| die
fi
}
create_part_data()
{
nfo1 "Data partition ($part_label)"
nfo2 "Creating partition"
cmd_parted "$dev_loop" \
mkpart primary ext4 "$part_start" "$part_end" \
align-check minimal "$part_nr" \
|| die
nfo2 "Creating filesystem"
"$cmd_mkfs_ext4" -q -F -L "$part_label" -O dir_index "$part_dev" > /dev/null 2>&1 || die
nfo2 "Populating filesystem"
cmd_mount ext4 "$part_dev" || die
mkdir mount/lsts || die
cmd_unmount || die
}
export LC_ALL=C
if [ -z "$BASH_VERSION" ]; then
echo "ERROR: you must use bash to run this script."
exit 1
fi
if [ "$(whoami)" != 'root' ]; then
echo "ERROR: you must be root to execute this target."
exit 1
fi
if [ $# -lt 2 ]; then
echo "Usage: $0 <config> <disk> [disk_size]"
exit 1
fi
source "$1"
if ! [ -f "$cfg_rootfs_tar" ]; then
echo "ERROR: you must run pkrootfs.bash first."
exit 1
fi
if [ -x "$cfg_dir_toolchain/sbin/parted" ]; then
cmd_parted="$cfg_dir_toolchain/sbin/parted"
else
echo "Warning: using system parted, this might lead to errors"
cmd_parted="parted"
fi
if [ -x "$cfg_dir_toolchain/sbin/fdisk" ]; then
cmd_fdisk="$cfg_dir_toolchain/sbin/fdisk"
else
echo "Warning: using system fdisk, this might lead to errors"
cmd_fdisk="fdisk"
fi
if [ -x "$cfg_dir_toolchain/sbin/mkdosfs" ]; then
cmd_mkdosfs="$cfg_dir_toolchain/sbin/mkdosfs"
else
echo "Warning: using system mkdosfs, this might lead to errors"
cmd_mkdosfs="mkdosfs"
fi
if [ -x "$cfg_dir_toolchain/sbin/mkfs.ext2" ]; then
cmd_mkfs_ext2="$cfg_dir_toolchain/sbin/mkfs.ext2"
else
echo "Warning: using system mkfs.ext2, this might lead to errors"
cmd_mkfs_ext2="mkfs.ext2"
fi
if [ -x "$cfg_dir_toolchain/sbin/mkfs.ext4" ]; then
cmd_mkfs_ext4="$cfg_dir_toolchain/sbin/mkfs.ext4"
else
echo "Warning: using system mkfs.ext4, this might lead to errors"
cmd_mkfs_ext4="mkfs.ext4"
fi
source "functions.bash"
dev="$2"
# unmounting partitions on target device
mounted_partition=$(mount | grep $dev | cut -f1 -d " ");
if [ ${#mounted_partition} -gt 0 ]; then
echo "Warning: some partitions of $hd_dev are mounted:"
for p in ${mounted_partition};
do
echo " unmounting $p";
umount $p;
done
fi
if [ -z "$cfg_partitions" ]; then
cfg_partitions=( \
'root' 'root0' '512B' '532MiB'
'data' 'data0' '532MiB' '-1'
)
fi
cmd_loop_attach()
{
export dev="$1"
export dev_loop="$(losetup -f)"
nfo1 "Attaching $dev to $dev_loop"
losetup -v "$dev_loop" "$dev"
if [ $? -ne 0 ]; then
unset dev_loop
die
fi
}
cmd_loop_detach()
{
nfo1 "Detaching $dev_loop"
sync && sync && sync && sync
blockdev --flushbufs "$dev_loop"
losetup -v -d "$dev_loop"
sync && sync && sync && sync
unset dev_loop
}
nfo1 "Obliterating partition table"
cmd_loop_attach "$dev"
dd if=/dev/zero of="$dev_loop" bs=1 count=1M count=32 > /dev/null 2>&1 || die
losetup -a
cmd_loop_detach
losetup -a
nfo1 "Creating empty partition table"
cmd_loop_attach "$dev"
cmd_parted "$dev_loop" mklabel msdos || die
cmd_loop_detach
part_nr=1
xboot_system_id=0
for ((i = 0; i < ${#cfg_partitions[@]}; i += 4)); do
cmd_loop_attach "$dev"
part_type="${cfg_partitions[$i+0]}"
part_label="${cfg_partitions[$i+1]}"
part_start="${cfg_partitions[$i+2]}"
part_end="${cfg_partitions[$i+3]}"
part_dev="${dev_loop}p${part_nr}"
case $part_type in
'x-boot')
create_part_xboot
xboot_system_id=1
;;
'rpi-boot')
create_part_rpiboot
xboot_system_id=1
;;
'root')
create_part_root
;;
'data')
create_part_data
;;
*)
echo "ERROR: unknown partition type '$part_type'"
die
;;
esac
cmd_loop_detach
let part_nr++
done
if [ -f "$cfg_sys_family/rootfs/boot/extlinux/mbr.bin" ]; then
nfo1 "Installing bootloader in MBR..."
cmd_loop_attach "$dev"
dd if="$cfg_sys_family/rootfs/boot/extlinux/mbr.bin" of="$dev_loop" > /dev/null 2>&1
cmd_loop_detach
fi
nfo1 "Synchronizing caches"
sync && sync && sync && sync && sync && sync
# Make sure the partition has 0x0e type.
if [ $xboot_system_id -eq 1 ]; then
cmd_loop_attach "$dev"
echo -en "t\n1\ne\nw\n" | $cmd_fdisk "$dev_loop" > /dev/null 2>&1
cmd_loop_detach
fi