1#!/bin/sh
2#-
3# Copyright (c) 2018 Rebecca Cran
4# Copyright (c) 2017 Nathan Whitehorn
5# All rights reserved.
6#
7# Redistribution and use in source and binary forms, with or without
8# modification, are permitted provided that the following conditions
9# are met:
10# 1. Redistributions of source code must retain the above copyright
11#    notice, this list of conditions and the following disclaimer.
12# 2. Redistributions in binary form must reproduce the above copyright
13#    notice, this list of conditions and the following disclaimer in the
14#    documentation and/or other materials provided with the distribution.
15#
16# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26# SUCH DAMAGE.
27#
28# $FreeBSD$
29
30
31FREEBSD_BOOTLABEL="FreeBSD"
32
33BSDCFG_SHARE="/usr/share/bsdconfig"
34. $BSDCFG_SHARE/common.subr || exit 1
35f_dprintf "%s: loading_includes..." "$0"
36f_include $BSDCFG_SHARE/dialog.subr
37
38: ${TMPDIR:="/tmp"}
39
40die() {
41	echo $*
42	exit 1
43}
44
45dialog_uefi_entryname()
46{
47	local prompt="Please enter a name for the new entry"
48	local hline=
49	local value="$*"
50	local height width
51
52	f_dialog_inputbox_size height width \
53		"$DIALOG_TITLE" "$DIALOG_BACKTITLE" "$prompt" "$value" "$hline"
54
55	$DIALOG \
56		--title "$DIALOG_TITLE"         \
57		--backtitle "$DIALOG_BACKTITLE" \
58		--hline "$hline"                \
59		--ok-label "Ok"                 \
60		--no-cancel                     \
61		--inputbox "$prompt"            \
62		$height $width "$value"         \
63		2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
64}
65
66update_uefi_bootentry()
67{
68	nentries=$(efibootmgr | grep -c 'FreeBSD$')
69	# No entries so directly create one and return
70	if [ ${nentries} -eq 0 ]; then
71		f_dprintf "Creating UEFI boot entry"
72		efibootmgr --create --activate --label "$FREEBSD_BOOTLABEL" --loader "${mntpt}/${FREEBSD_BOOTNAME}" > /dev/null
73		return
74	fi
75
76	$DIALOG --backtitle 'FreeBSD Installer' --title 'Boot configuration' \
77	    --yesno 'There are multiple "FreeBSD" EFI boot entries. Would you like to remove them all and add a new one?' 0 0
78	if [ $? -eq $DIALOG_OK ]; then
79		for entry in $(efibootmgr | awk '$NF == "FreeBSD" { sub(/.*Boot/,"", $1); sub(/\*/,"", $1); print $1 }'); do
80			efibootmgr -B -b ${entry}
81		done
82		efibootmgr --create --activate --label "$FREEBSD_BOOTLABEL" --loader "${mntpt}/${FREEBSD_BOOTNAME}" > /dev/null
83		return
84	fi
85
86	FREEBSD_BOOTLABEL=$(dialog_uefi_entryname "${FREEBSD_BOOTLABEL}")
87	[ $? -eq $DIALOG_CANCEL ] && exit 1
88	efibootmgr --create --activate --label "$FREEBSD_BOOTLABEL" --loader "${mntpt}/${FREEBSD_BOOTNAME}" > /dev/null
89}
90
91f_dialog_title "Boot configuration"
92f_dialog_backtitle "FreeBSD Installer"
93
94if [ `uname -m` == powerpc ]; then
95	platform=`sysctl -n hw.platform`
96	if [ "$platform" == ps3 -o "$platform" == powernv ]; then
97		rootpart=$(awk '{ if($2 == "/") printf("%s:%s\n", $3, $1); }' $PATH_FSTAB)
98		kboot_conf=$BSDINSTALL_CHROOT/boot/etc/kboot.conf
99		mkdir -p $BSDINSTALL_CHROOT/boot/etc/
100		echo default=$FREEBSD_BOOTLABEL > $kboot_conf
101		echo $FREEBSD_BOOTLABEL=\'/kernel/kernel kernelname=/boot/kernel/kernel vfs.root.mountfrom=${rootpart}\' >> $kboot_conf
102	fi
103fi
104
105# Update the ESP (EFI System Partition) with the new bootloader if we have an ESP
106if [ -n "$(awk '{if ($2=="/boot/efi") printf("%s\n",$1);}' $PATH_FSTAB)" ]; then
107	case $(uname -m) in
108	    arm64)	ARCHBOOTNAME=aa64 ;;
109	    amd64)	ARCHBOOTNAME=x64 ;;
110	    riscv)	ARCHBOOTNAME=riscv64 ;;
111    #	    arm)	ARCHBOOTNAME=arm ;; # No other support for arm install
112    #	    i386)	ARCHBOOTNAME=ia32 ;; # no support for this in i386 kernels, rare machines
113	    *)		die "Unsupported arch $(uname -m) for UEFI install"
114	esac
115	BOOTDIR="/efi/boot"
116	BOOTNAME="${BOOTDIR}/boot${ARCHBOOTNAME}.efi"
117	FREEBSD_BOOTDIR="/efi/freebsd"
118	FREEBSD_BOOTNAME="${FREEBSD_BOOTDIR}/loader.efi"
119	mntpt="$BSDINSTALL_CHROOT/boot/efi"
120
121	f_dprintf "Installing loader.efi onto ESP"
122	mkdir -p "${mntpt}/${FREEBSD_BOOTDIR}" "${mntpt}/${BOOTDIR}"
123	cp "$BSDINSTALL_CHROOT/boot/loader.efi" "${mntpt}/${FREEBSD_BOOTNAME}"
124
125	#
126	# The following shouldn't be necessary. UEFI defines a way to
127	# specifically select what to boot (which we do via
128	# efibootmgr). However, virtual environments often times lack
129	# support for the NV variables efibootmgr sets. In addition,
130	# some UEFI implementations have features that interfere with
131	# the setting of these variables. To combat that, we install the
132	# default removable media boot file as a fallback if it doesn't
133	# exist. We don't install it all the time since that can
134	# interfere with other installations on the drive (like rEFInd).
135	#
136	if [ ! -f "${mntpt}/${BOOTNAME}" ]; then
137		cp "$BSDINSTALL_CHROOT/boot/loader.efi" "${mntpt}/${BOOTNAME}"
138	fi
139
140	if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
141		update_uefi_bootentry
142	fi
143
144	f_dprintf "Finished configuring ESP"
145fi
146
147# Add boot0cfg for MBR BIOS booting?
148