hacks/boot_img.bash at main · MichaIng/hacks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#!/bin/bash
{
. /boot/dietpi/func/dietpi-globals

# Inputs:
# - "-e", "--edit": drop to shell for edit
# - "-b", "--boot-part": has boot partition: if set, assumes partition 1 to be boot partition, and partition 2 to be rootfs
# - $1: img path
# - $2: img size (GiB, optional, defaults to 2)
EDIT=0
FP_IMG=
FS_IMG=
ROOT_PART=1
BOOT_PART=
while (( $# ))
do
	case $1 in
		'-e'|'--edit') EDIT=1;;
		'-b'|'--boot-part') BOOT_PART=1 ROOT_PART=2;;
		*) [[ $FP_IMG ]] && FS_IMG=$1 || FP_IMG=$1;;
	esac
	shift
done
G_EXEC test -f "$FP_IMG"
disable_error=1 G_CHECK_VALIDINT "$FS_IMG" 0 || FS_IMG=2

# Install dependencies
# - Install FAT fsck if boot partition is used
[[ $BOOT_PART ]] && dosfstools=('dosfstools') || dosfstools=()
# - Install QEMU emulation support when running from x86_64 host
(( $G_HW_ARCH == 10 )) && emulation_packages=('qemu-user-static') || emulation_packages=()
G_AG_CHECK_INSTALL_PREREQ fdisk dbus systemd-container "${dosfstools[@]}" "${emulation_packages[@]}"
G_EXEC systemctl start dbus
[[ ${emulation_packages[0]} ]] && G_EXEC systemctl restart systemd-binfmt

# Obtain device paths
LOOP_DEV=$(losetup -f)
ROOT_DEV="${LOOP_DEV}p$ROOT_PART"
[[ $BOOT_PART ]] && BOOT_DEV="${LOOP_DEV}p$BOOT_PART"

# Exit trap for cleanup
G_EXIT_CUSTOM()
{
	# Revert workarounds
	G_EXEC rm -f rootfs/etc/systemd/system/dropbear.service
	for i in rootfs/etc/systemd/system/*.service.d/no-credentials.conf
	do
		[[ $i == 'rootfs/etc/systemd/system/*.service.d/no-credentials.conf' ]] && break
		G_EXEC rm "$i"
		G_EXEC rmdir --ignore-fail-on-non-empty "${i%/no-credentials.conf}"
	done
	# Cleanup
	findmnt -M rootfs > /dev/null && G_EXEC umount -Rl rootfs
	G_EXEC losetup -d "$LOOP_DEV"
	[[ -d 'rootfs' ]] && G_EXEC rmdir rootfs
}
trap G_EXIT_CUSTOM EXIT

# Create loop device and fsck
G_EXEC losetup -P "$LOOP_DEV" "$FP_IMG"
G_EXEC_OUTPUT=1 G_EXEC e2fsck -fyD "$ROOT_DEV"
[[ $BOOT_DEV ]] && G_EXEC_OUTPUT=1 G_EXEC fsck -y "$BOOT_DEV"

# Raise image+partition+fs size if required
if (( $FS_IMG && $(stat -c '%s' "$FP_IMG") < $FS_IMG*1024**3 ))
then
	G_EXEC truncate -s $(($FS_IMG*1024**3)) "$FP_IMG"
	G_EXEC losetup -c "$LOOP_DEV"
	G_EXEC_OUTPUT=1 G_EXEC eval "sfdisk -N'${ROOT_DEV: -1}' '$LOOP_DEV' <<< ',+'"
	G_EXEC_OUTPUT=1 G_EXEC resize2fs "$ROOT_DEV"
fi

# Mount
G_EXEC mkdir -p rootfs
findmnt -M rootfs &> /dev/null && G_EXEC umount -R rootfs
G_EXEC mount "$ROOT_DEV" rootfs
if [[ $BOOT_DEV ]]
then
	if [[ -d '/boot/firmware' ]]
	then
		G_EXEC mount "$BOOT_DEV" rootfs/boot/firmware
	else
		G_EXEC mount "$BOOT_DEV" rootfs/boot
	fi
fi

# Workarounds
# - Mask Dropbear to prevent its package install from failing
G_EXEC ln -s /dev/null rootfs/etc/systemd/system/dropbear.service
# - Prevent failing services and missing login prompt by disabling credentials in systemd units
if [[ ${emulation_packages[0]} ]]
then
	for i in rootfs/lib/systemd/system/*.service
	do
		[[ -f $i ]] || continue
		grep -q '^ImportCredential=' "$i" && G_EXEC mkdir -p "${i/lib/etc}.d" && G_EXEC eval "echo -e '[Service]\nImportCredential=' > '${i/lib/etc}.d/no-credentials.conf'"
		grep -q '^LoadCredential=' "$i" && G_EXEC mkdir -p "${i/lib/etc}.d" && G_EXEC eval "echo -e '[Service]\nLoadCredential=' > '${i/lib/etc}.d/no-credentials.conf'"
	done
fi

# Drop to shell for edit
(( $EDIT )) && bash

# Boot
# - Bind mounts required to allow container reading its own root device info: --bind="$LOOP_DEV" --bind="$ROOT_DEV" ${BOOT_DEV:+--bind="$BOOT_DEV"} --bind=/dev/disk
systemd-nspawn -bD rootfs || exit 1

exit 0
}