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 30BSDCFG_SHARE="/usr/share/bsdconfig" 31. $BSDCFG_SHARE/common.subr || exit 1 32 33: ${TMPDIR:="/tmp"} 34 35die() { 36 echo $* 37 exit 1 38} 39 40if [ `uname -m` == powerpc ]; then 41 platform=`sysctl -n hw.platform` 42 if [ "$platform" == ps3 -o "$platform" == powernv ]; then 43 rootpart=$(awk '{ if($2 == "/") printf("%s:%s\n", $3, $1); }' $PATH_FSTAB) 44 mkdir -p $BSDINSTALL_CHROOT/boot/etc/ 45 echo FreeBSD=\'/kernel/kernel kernelname=/boot/kernel/kernel vfs.root.mountfrom=${rootpart}\' > $BSDINSTALL_CHROOT/boot/etc/kboot.conf 46 fi 47fi 48 49# Update the ESP (EFI System Partition) with the new bootloader 50if [ "$(uname -m)" = "amd64" ] || [ "$(uname -m)" = "i386" ]; then 51 X86_BOOTMETHOD=$(sysctl -n machdep.bootmethod) 52fi 53 54if [ "$(uname -m)" = "arm64" ] || [ "$X86_BOOTMETHOD" = "UEFI" ]; then 55 UFSBOOT_ESPS=$(cat $TMPDIR/bsdinstall-esps 2>/dev/null) 56 ZFSBOOT_DISKS=$(cat $TMPDIR/bsdinstall-zfsboot 2>/dev/null) 57 num_esps=0 58 59 if [ -n "$ZFSBOOT_DISKS" ]; then 60 # We're in a ZFS install environment 61 for disk in $ZFSBOOT_DISKS; do 62 index=$(gpart show "$disk" | cut -w -f 4,5 | grep "efi" | cut -w -f 1) 63 # Check that $index is an integer 64 [ -n "$index" ] && [ "$index" -eq "$index" ] && [ "$index" -ge 0 ] 2> /dev/null 65 if [ $? -ne 0 ]; then 66 continue 67 fi 68 69 if [ -e "/dev/${disk}p${index}" ]; then 70 ESPS="$ESPS ${disk}p${index}" 71 elif [ -e "/dev/${disk}s${index}" ]; then 72 ESPS="$ESPS ${disk}s${index}" 73 else 74 continue 75 fi 76 77 num_esps=$((num_esps + 1)) 78 done 79 fi 80 81 if [ -n "$UFSBOOT_ESPS" ]; then 82 # We're in a UFS install environment 83 for partition in $UFSBOOT_ESPS; do 84 ESPS="$ESPS $partition" 85 num_esps=$((num_esps + 1)) 86 done 87 fi 88 89 if [ -z "$ESPS" ]; then 90 # The installer hasn't given us any ESPs to use. 91 # Try and figure out which to use by looking for an 92 # unformatted efi partition 93 94 for geom in $(gpart status -sg | awk '{print $1}'); do 95 hasfreebsd=$(gpart show "${geom}" | cut -w -f 4,5 | grep "freebsd") 96 if [ -n "$hasfreebsd" ]; then 97 index=$(gpart show "${geom}" | cut -w -f 4,5 | grep "efi" | cut -w -f 1) 98 # Check that $index is a valid integer 99 [ -n "$index" ] && [ "$index" -eq "$index" ] && [ "$index" -ge 0 ] 2> /dev/null 100 if [ $? -ne 0 ]; then 101 continue 102 fi 103 104 mntpt=$(mktemp -d $TMPDIR/stand-test.XXXXXX) 105 if [ -e "/dev/${geom}p${index}" ]; then 106 dev=${geom}p${index} 107 elif [ -e "/dev/${geom}s${index}" ]; then 108 dev=/${geom}s${index} 109 else 110 continue 111 fi 112 113 # Try and mount it. If it fails, assume it's 114 # unformatted and should be used. 115 mount -t msdosfs -o ro "/dev/${dev}" "${mntpt}" 116 if [ $? -ne 0 ]; then 117 ESPS="$ESPS ${dev}" 118 num_esps=$((num_esps + 1)) 119 else 120 umount "${mntpt}" 121 fi 122 rmdir "${mntpt}" 123 fi 124 done 125 fi 126 127 case $(uname -m) in 128 arm64) ARCHBOOTNAME=aa64 ;; 129 amd64) ARCHBOOTNAME=x64 ;; 130 # arm) ARCHBOOTNAME=arm ;; # No other support for arm install 131 # i386) ARCHBOOTNAME=ia32 ;; # no support for this in i386 kernels, rare machines 132 *) die "Unsupported arch $(uname -m) for UEFI install" 133 esac 134 BOOTNAME="/EFI/BOOT/BOOT${ARCHBOOTNAME}.EFI" 135 136 for esp in $ESPS; do 137 f_dprintf "Formatting /dev/${esp} as FAT32" 138 newfs_msdos -F 32 -c 1 -L EFISYS "/dev/$esp" > /dev/null 2>&1 139 if [ $? -ne 0 ]; then 140 die "Failed to format ESP $esp as FAT32" 141 fi 142 143 mntpt=$(mktemp -d $TMPDIR/stand-test.XXXXXX) 144 f_dprintf "Mounting ESP /dev/${esp}" 145 mount -t msdosfs "/dev/${esp}" "${mntpt}" 146 if [ $? -ne 0 ]; then 147 die "Failed to mount ESP ${dev} on ${mntpt}" 148 fi 149 150 f_dprintf "Installing loader.efi onto ESP" 151 mkdir -p "$mntpt/EFI/freebsd" 152 cp "$BSDINSTALL_CHROOT/boot/loader.efi" "${mntpt}/EFI/freebsd/loader.efi" 153 154 # 155 # The following shouldn't be necessary. UEFI defines a way to 156 # specifically select what to boot (which we do via 157 # efibootmgr). However, virtual environments often times lack 158 # support for the NV variables efibootmgr sets. In addition, 159 # some UEFI implementations have features that interfere with 160 # the setting of these variables. To combat that, we install the 161 # default removable media boot file as a fallback if it doesn't 162 # exist. We don't install it all the time since that can 163 # interfere with other installations on the drive (like rEFInd). 164 # 165 if [ ! -f "${mntpt}/${BOOTNAME}" ]; then 166 cp "$BSDINSTALL_CHROOT/boot/loader.efi" "${mntpt}/${BOOTNAME}" 167 fi 168 169 if [ "$num_esps" -gt 1 ]; then 170 bootlabel="FreeBSD (${esp})" 171 else 172 bootlabel="FreeBSD" 173 fi 174 175 f_dprintf "Creating UEFI boot entry" 176 efibootmgr --create --activate --label "$bootlabel" --loader "${mntpt}/EFI/freebsd/loader.efi" > /dev/null 177 178 f_dprintf "Unmounting ESP" 179 umount "${mntpt}" 180 rmdir "${mntpt}" 181 182 f_dprintf "Finished configuring /dev/${esp} as ESP" 183 done 184fi 185 186# Add boot0cfg for MBR BIOS booting? 187