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
30die() {
31	echo $*
32	exit 1
33}
34
35if [ `uname -m` == powerpc ]; then
36	platform=`sysctl -n hw.platform`
37	if [ "$platform" == ps3 -o "$platform" == powernv ]; then
38		rootpart=$(awk '{ if($2 == "/") printf("%s:%s\n", $3, $1); }' $PATH_FSTAB)
39		mkdir -p $BSDINSTALL_CHROOT/boot/etc/
40		echo FreeBSD=\'/kernel/kernel kernelname=/boot/kernel/kernel vfs.root.mountfrom=${rootpart}\' > $BSDINSTALL_CHROOT/boot/etc/kboot.conf
41	fi
42fi
43
44# Update the ESP (EFI System Partition) with the new bootloader
45if [ "$(uname -m)" = "amd64" ] || [ "$(uname -m)" = "i386" ]; then
46	X86_BOOTMETHOD=$(sysctl -n machdep.bootmethod)
47fi
48
49if [ "$(uname -m)" = "arm64" ] || [ "$X86_BOOTMETHOD" = "UEFI" ]; then
50	UFSBOOT_ESPS=$(cat /tmp/bsdinstall-esps)
51	num_esps=0
52
53	if [ -n "$ZFSBOOT_DISKS" ]; then
54		# We're in a ZFS install environment
55		for disk in $ZFSBOOT_DISKS; do
56			index=$(gpart show "$disk" | cut -w -f 4,5 | grep "efi" | cut -w -f 1)
57			# Check that $index is an integer
58			[ -n "$index" ] && [ "$index" -eq "$index" ] && [ "$index" -ge 0 ] 2> /dev/null
59			if [ $? -ne 0 ]; then
60				continue
61			fi
62
63			if [ -e "/dev/${disk}p${index}" ]; then
64				ESPS="$ESPS ${disk}p${index}"
65			elif [ -e "/dev/${disk}s${index}" ]; then
66				ESPS="$ESPS ${disk}s${index}"
67			else
68				continue
69			fi
70
71			num_esps=$((num_esps + 1))
72		done
73	fi
74
75	if [ -n "$UFSBOOT_ESPS" ]; then
76		# We're in a UFS install environment
77		for partition in $UFSBOOT_ESPS; do
78			ESPS="$ESPS $partition"
79			num_esps=$((num_esps + 1))
80		done
81	fi
82
83	if [ -z "$ESPS" ]; then
84		# The installer hasn't given us any ESPs to use.
85		# Try and figure out which to use by looking for an
86		# unformatted efi partition
87		for disk in $(sysctl -n kern.disks); do
88			hasfreebsd=$(gpart show "$disk" | cut -w -f 4,5 | grep "freebsd")
89			if [ -n "$hasfreebsd" ]; then
90				index=$(gpart show "$disk" | cut -w -f 4,5 | grep "efi" | cut -w -f 1)
91				# Check that $index is a valid integer
92				[ -n "$index" ] && [ "$index" -eq "$index" ] && [ "$index" -ge 0 ] 2> /dev/null
93				if [ $? -ne 0 ]; then
94					continue
95				fi
96
97				mntpt=$(mktemp -d /tmp/stand-test.XXXXXX)
98				if [ -e "/dev/${disk}p${index}" ]; then
99					dev=${disk}p${index}
100				elif [ -e "/dev/${disk}s${index}" ]; then
101					dev=/${disk}s${index}
102				else
103					continue
104				fi
105
106				# Try and mount it. If it fails, assume it's
107				# unformatted and should be used.
108				mount -t msdosfs "/dev/${dev}" "${mntpt}"
109				if [ $? -ne 0 ]; then
110					ESPS="$ESPS ${dev}"
111					num_esps=$((num_esps + 1))
112				else
113					umount "${mntpt}"
114				fi
115				rmdir "${mntpt}"
116			fi
117		done
118	fi
119
120	for esp in $ESPS; do
121		newfs_msdos -F 32 -c 1 -L EFISYS "/dev/$esp" > /dev/null 2>&1
122		if [ $? -ne 0 ]; then
123			die "Failed to format ESP $esp as FAT32"
124		fi
125
126		mntpt=$(mktemp -d /tmp/stand-test.XXXXXX)
127		mount -t msdosfs "/dev/${esp}" "${mntpt}"
128		if [ $? -ne 0 ]; then
129			die "Failed to mount ESP ${dev} on ${mntpt}"
130		fi
131
132		mkdir -p "$mntpt/EFI/freebsd"
133		cp "$BSDINSTALL_CHROOT/boot/loader.efi" "${mntpt}/EFI/freebsd/loader.efi"
134
135		if [ "$num_esps" -gt 1 ]; then
136			bootlabel="FreeBSD (${esp})"
137		else
138			bootlabel="FreeBSD"
139		fi
140
141		efibootmgr --create --label "$bootlabel" --loader "${mntpt}/EFI/freebsd/loader.efi" > /dev/null
142
143		umount "${mntpt}"
144		rmdir "${mntpt}"
145
146		# When creating new entries, efibootmgr doesn't mark them active, so we need to
147		# do so. It doesn't make it easy to find which entry it just added, so rely on
148		# the fact that it places the new entry first in BootOrder.
149		bootorder=$(efivar --name 8be4df61-93ca-11d2-aa0d-00e098032b8c-BootOrder --print --no-name --hex | head -1)
150		bootentry=$(echo "$bootorder" | cut -w -f 3)$(echo "$bootorder" | cut -w -f 2)
151		efibootmgr --activate "$bootentry" > /dev/null
152	done
153fi
154
155# Add boot0cfg for MBR BIOS booting?
156