summaryrefslogtreecommitdiff
path: root/buildlive
blob: ad607cb2eca21ed1c2f6b9ea040b1925fbb92375 (plain)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
#!/bin/bash
# Copyright (C)  2016-2018 Iru Cai <mytbk920423@gmail.com>
# SPDX-License-Identifier: GPL-3.0-or-later

set -e

DISTROLIST=()
ISOLIST=()
BOOTPATH=
ROOTPATH=
DEVNAME=
KERNELDIR=
DATADIR=
INSTALL_GRUB=1
GRUBCFG_ONLY=0
DOWNLOAD_ONLY=0
UMOUNT_BOOT=0
UMOUNT_ROOT=0
CLEAN_USB=0

. functions.sh

msg() {
	echo -e "$1" >&2
}

fatalerror() {
	msg "\x1b[1;31m$1\x1b[0m"
	exit 1
}

usage() {
	>&2 cat << EOF
$0 [--root <path>] [options] [distro 1] [distro 2] ...
use $0 -L to list available distros
options:
  --root <path>
  --root=<path>: set the path to put the live USB data files
  --boot <path>
  --boot=<path>: set the path to put the kernel and loader files
  --clean: clean the old live USB files before installing
  --no-grub: do not install GRUB loader (still generate grub.cfg)
  --grubcfg: do not do real install, only generate grub.cfg file
  --downloadonly: only download the ISO files
  --dev=<diskdev>: write boot sector to <diskdev> (default <diskdev> is autodetected from <rootpath>)
EOF
}

try_mount() {
	local mnt
	if [[ "$1" =~ ^/dev ]]
	then
		if ! findmnt "$1" > /dev/null
		then
			udevil mount "$1" > /dev/null
			mnt=$(findmnt -n -o TARGET "$1")
		else
			mnt=$(findmnt -n -o TARGET "$1")
		fi
	else
		mnt="$1"
	fi
	echo "$mnt"
}

unset ISOPATH
CFGFILE="$HOME/.liveusb-builder"
test -f "$CFGFILE" && source "$CFGFILE" || true
ISOPATH=${ISOPATH:-"$HOME/isofiles"}

while [[ -n "$1" ]]
do
	case "$1" in
		-L)
			exec ./listisos.sh
			;;
		--boot=*)
			BOOTPATH="${1/--boot=}"
			;;
		--boot)
			shift
			BOOTPATH="$1"
			;;
		--root=*)
			ROOTPATH="${1/--root=}"
			;;
		--root)
			shift
			ROOTPATH="$1"
			;;
		--dev=*)
			DEVNAME=${1/--dev=}
			;;
		--no-grub)
			INSTALL_GRUB=0
			;;
		--grubcfg)
			GRUBCFG_ONLY=1
			;;
		--downloadonly)
			DOWNLOAD_ONLY=1
			;;
		--clean)
			CLEAN_USB=1
			;;
		*=*|-*)
			usage
			exit 1
			;;
		*)
			if [ -f "distro/$1/isoinfo" ]
			then
				DISTROLIST=(${DISTROLIST[@]} "$1")
			else
				fatalerror "directory distro/$1/isoinfo not found"
			fi
			;;
	esac
	shift
done

if [[ -z "$ROOTPATH" && "$DOWNLOAD_ONLY" == 0 ]]; then
	usage
	exit 1
fi

_mnt="$(try_mount "$ROOTPATH")"
if [ "${_mnt}" != "$ROOTPATH" ]; then
	ROOTPATH="${_mnt}"
	UMOUNT_ROOT=1
fi

if [[ -z "$BOOTPATH" ]]; then
	BOOTPATH="$ROOTPATH"
else
	_mnt="$(try_mount "$BOOTPATH")"
	if [ "${_mnt}" != "$BOOTPATH" ]; then
		BOOTPATH="${_mnt}"
		UMOUNT_BOOT=1
	fi
fi

if [[ "$DOWNLOAD_ONLY" == 0 ]]; then
	UUID="$(findmnt -o UUID --raw --noheadings $ROOTPATH \
		|| fatalerror "UUID of $ROOTPATH not found, not a mountpoint?")"
	BOOTPART="$(findmnt -o SOURCE --raw --noheadings $BOOTPATH \
		|| fatalerror "$BOOTPATH seems not a mountpoint!")"
	# liveusb kernel and data should be placed on one disk
	# so get DEVNAME by the mountpoint of ROOTPATH is ok
	DEVNAME="/dev/$(getdiskbypart "/dev/disk/by-uuid/$UUID")"

	KERNELDIR="$BOOTPATH/liveusb-kernel"
	DATADIR="$ROOTPATH/liveusb-data"
	GRUBCFG="$BOOTPATH/grub/grub.cfg"
	SYSLINUXCFG="$BOOTPATH/syslinux/syslinux.cfg"
	install -d "$BOOTPATH/grub" "$BOOTPATH/syslinux"
	install -d "$KERNELDIR" "$DATADIR"
	msg "Files will be copy to $ROOTPATH"
fi

if [[ "${#DISTROLIST[@]}" == 0 ]]; then
	usage
	exit 1
fi

if [ ! -z "$DEVNAME" ]; then
	msg "Boot sector will be written to $DEVNAME"
fi

if [[ "$CLEAN_USB" = 1 ]]; then
	msg "Cleaning old live USB files."
	rm -rf "$KERNELDIR"/* "$DATADIR"/*
fi

for i in ${DISTROLIST[@]}
do
	process_isoinfo "$i"
	ISOLIST=("${ISOLIST[@]}" "$ISOFILE")
	ISONAMELIST=("${ISONAMELIST[@]}" "$ISONAME")

	if [ -f "$ISOPATH/$ISOFILE" ] && checksum_verify; then
		true
	else
		download_iso
	fi
done

if [[ "$DOWNLOAD_ONLY" == 1 ]]; then
	exit 0
fi

grubcfg_header > "$GRUBCFG"
syslinux_header > "$SYSLINUXCFG"

for i in `seq 1 ${#DISTROLIST[@]}`
do
	ISOFILE="${ISOLIST[$i-1]}"
	ISONAME="${ISONAMELIST[$i-1]}"
	DISTRO="${DISTROLIST[$i-1]}"
	ISO_FILEPATH="$ISOPATH/$ISOFILE"

	set_distro "$DISTRO"
	export DISTRONAME KEYWORD # for grub and syslinux generation

	process_distro "$DISTRO"

	if [ "$GRUBCFG_ONLY" == 0 ]; then
		install_live
	fi
	gen_grubcfg "$DISTRO" >> "$GRUBCFG"
	gen_syslinux "$DISTRO" >> "$SYSLINUXCFG"
done

if [ "$GRUBCFG_ONLY" == 1 ]; then
	msg 'Only generate grub.cfg and syslinux.cfg, will not install bootloader.'
elif [ "$INSTALL_GRUB" == 1 ]; then
	read -p 'Answer Y to install bootloader.[Y/n] '
	if [[ "$REPLY" =~ ^[Yy]|^$ ]]; then
		if ! as-root ./install_syslinux "$DEVNAME" "$BOOTPART" "$BOOTPATH"; then
			msg 'Failed to install syslinux, try to install GRUB as legacy BIOS bootloader.'
			as-root grub-install --force --boot-directory="$BOOTPATH" --target=i386-pc "$DEVNAME"
		fi
		as-root grub-install --boot-directory="$BOOTPATH" --efi-directory="$BOOTPATH" --bootloader-id=grub --target=x86_64-efi --removable --no-nvram "$DEVNAME"
	fi
fi

if [ "$UMOUNT_BOOT" == 1 ]; then
	msg 'Trying to umount the boot mountpoint, you may need to wait for sync() to complete.'
	udevil umount "$BOOTPATH"
fi
if [ "$UMOUNT_ROOT" == 1 -a "$ROOTPATH" != "$BOOTPATH" ]; then
	msg 'Trying to umount the root mountpoint, you may need to wait for sync() to complete.'
	udevil umount "$ROOTPATH"
fi