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 88 for geom in $(gpart status -sg | awk '{print $1}'); do 89 hasfreebsd=$(gpart show "${geom}" | cut -w -f 4,5 | grep "freebsd") 90 if [ -n "$hasfreebsd" ]; then 91 index=$(gpart show "${geom}" | cut -w -f 4,5 | grep "efi" | cut -w -f 1) 92 # Check that $index is a valid integer 93 [ -n "$index" ] && [ "$index" -eq "$index" ] && [ "$index" -ge 0 ] 2> /dev/null 94 if [ $? -ne 0 ]; then 95 continue 96 fi 97 98 mntpt=$(mktemp -d /tmp/stand-test.XXXXXX) 99 if [ -e "/dev/${geom}p${index}" ]; then 100 dev=${geom}p${index} 101 elif [ -e "/dev/${geom}s${index}" ]; then 102 dev=/${geom}s${index} 103 else 104 continue 105 fi 106 107 # Try and mount it. If it fails, assume it's 108 # unformatted and should be used. 109 mount -t msdosfs -o ro "/dev/${dev}" "${mntpt}" 110 if [ $? -ne 0 ]; then 111 ESPS="$ESPS ${dev}" 112 num_esps=$((num_esps + 1)) 113 else 114 umount "${mntpt}" 115 fi 116 rmdir "${mntpt}" 117 fi 118 done 119 fi 120 121 for esp in $ESPS; do 122 echo "Formatting /dev/${esp} as FAT32" 123 newfs_msdos -F 32 -c 1 -L EFISYS "/dev/$esp" > /dev/null 2>&1 124 if [ $? -ne 0 ]; then 125 die "Failed to format ESP $esp as FAT32" 126 fi 127 128 mntpt=$(mktemp -d /tmp/stand-test.XXXXXX) 129 echo "Mounting ESP /dev/${esp}" 130 mount -t msdosfs "/dev/${esp}" "${mntpt}" 131 if [ $? -ne 0 ]; then 132 die "Failed to mount ESP ${dev} on ${mntpt}" 133 fi 134 135 echo "Installing loader.efi onto ESP" 136 mkdir -p "$mntpt/EFI/freebsd" 137 cp "$BSDINSTALL_CHROOT/boot/loader.efi" "${mntpt}/EFI/freebsd/loader.efi" 138 139 if [ "$num_esps" -gt 1 ]; then 140 bootlabel="FreeBSD (${esp})" 141 else 142 bootlabel="FreeBSD" 143 fi 144 145 echo "Creating UEFI boot entry" 146 efibootmgr --create --activate --label "$bootlabel" --loader "${mntpt}/EFI/freebsd/loader.efi" > /dev/null 147 148 echo "Unmounting ESP" 149 umount "${mntpt}" 150 rmdir "${mntpt}" 151 152 echo "Finished configuring /dev/${esp} as ESP" 153 done 154fi 155 156# Add boot0cfg for MBR BIOS booting? 157