#!/bin/sh
#
# Copyright 2013 iXsystems (Kris Moore)
# Minor modifications by Joseph Mingrone
# All rights reserved
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted providing that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
# Script to detect FreeBSD disk installations and create grub
# entries for them
ROOTFS=`mount | awk '/ \/ / {print $1}'`
BEDS="$( echo ${ROOTFS} | awk -F '/' '{print $2}' )"
if [ "$BEDS" = "dev" ] ; then BEDS="ROOT"; fi
display_loaderopts()
{
# Optional ARG1, set to a ZFS dataset to mount and read values from
if [ -n "$1" ] ; then
fMnt="/mnt.$$"
mkdir $fMnt
mount -t zfs ${1} $fMnt >/dev/null
if [ $? -ne 0 ] ; then
echo "Failed to mount ${1}" >&2
return
fi
else
fMnt=""
fi
# Create our placeholder to save _load entries to parse
touch /tmp/.lRObjs.$$
touch /tmp/.lSysCtls.$$
# Loader files, in order of which to read them
lFiles="${fMnt}/boot/loader.conf.local ${fMnt}/boot/loader.conf"
for f in $lFiles
do
if [ ! -e "$f" ] ; then continue ; fi
# Lets parse any of the _load= lines
grep "_load=" ${f} | grep -v "^#" >/tmp/.lObjs.$$
while read line
do
loadVal="`echo $line | cut -d '=' -f 1`"
# Is this value already set in a higher priority file?
grep -q "^${loadVal}" /tmp/.lRObjs.$$
if [ $? -eq 0 ];then continue; fi
# Save this value for later
echo "$line" >> /tmp/.lRObjs.$$
haveObjs=1
done < /tmp/.lObjs.$$
rm /tmp/.lObjs.$$
# Lets look for any sysctls to set
grep "." ${f} | grep "=" | grep -v "^#" | grep -v "_load" >/tmp/.sObjs.$$
while read line
do
loadVal="`echo $line | cut -d '=' -f 1`"
# Is this value already set in a higher priority file?
grep -q "^${loadVal}" /tmp/.lSysCtls.$$
if [ $? -eq 0 ];then continue; fi
# Save this value for later
echo "$line" >> /tmp/.lSysCtls.$$
haveSysCtls=1
done < /tmp/.sObjs.$$
rm /tmp/.sObjs.$$
done
# Now lets echo out the modules to load
if [ "$haveObjs" = "1" ] ; then
while read line
do
echo "$line" | grep -q '"YES"'
if [ $? -ne 0 ] ; then continue ; fi
module="`echo $line | cut -d '=' -f 1 | sed 's|_load||g'`"
# Try to locate module now
if [ -e "${fMnt}/boot/kernel/${module}.ko" ] ; then
mPath="kernel"
elif [ -e "${fMnt}/boot/modules/${module}.ko" ] ; then
mPath="modules"
else
# This isn't a module that we can see, lets set it as a variable
#echo "No such module $line, setting as a variable" >&2
echo "$line" >> /tmp/.lSysCtls.$$
continue
fi
echo " kfreebsd_module_elf ${loadPrefix}/@/boot/${mPath}/${module}.ko"
done < /tmp/.lRObjs.$$
fi
# Any sysctls to set?
if [ "$haveSysCtls" = "1" ] ; then
while read line
do
# Strip out the vfs.root.mountfrom, we set that elsewhere
echo "$line" | grep -q "vfs.root.mountfrom"
if [ $? -eq 0 ] ; then continue ; fi
val="`echo $line | sed 's|"||g'`"
echo " set kFreeBSD.${val}"
done < /tmp/.lSysCtls.$$
fi
rm /tmp/.lRObjs.$$
rm /tmp/.lSysCtls.$$
if [ -n "$1" ] ; then
umount /mnt.$$ >/dev/null
rmdir /mnt.$$ >/dev/null
fi
}
detect_beadm()
{
which -s beadm
if [ $? -ne 0 ] ; then return 0; fi
# Check if we are running from the installer and use its beadm
if [ -e "/root/beadm.install" ] ; then
BEADM="/root/beadm.install"
else
BEADM="`which beadm`"
fi
${BEADM} list >/dev/null 2>/dev/null
if [ $? -ne 0 ] ; then return 0; fi
NICK="FreeBSD"
# Get list of beadm datasets
for b in `${BEADM} list 2>/dev/null| grep -v "Created" | cut -d ' ' -f 1`
do
# Got a beadm snapshot, lets get the complete dataset name
beLine=`${BEADM} list -a | grep "/$BEDS/${b}"`
cdataset=`echo $beLine | awk '{print $1}'`
cdatadate=`echo $beLine | awk '{print $5}'`
cdatatime=`echo $beLine | awk '{print $6}'`
ztank=`echo $cdataset | cut -d '/' -f 1`
shortdataset="/`echo $cdataset | cut -d '/' -f 2-5`"
# First part of this dataset
cat > /tmp/.grubdataset.$$.1 << EOF
insmod zfs
search --no-floppy -s -l $ztank
EOF
# Second part of loader to save
cat > /tmp/.grubdataset.$$.2 << EOF
kfreebsd_module ${shortdataset}/@/boot/zfs/zpool.cache type=/boot/zfs/zpool.cache
set kFreeBSD.vfs.root.mountfrom=zfs:$cdataset
EOF
# Now lets look for options in loader.conf to load
loadPrefix="${shortdataset}"
# If this is the current mounted dataset, we can skip mounting it
mount | grep -q -e "$cdataset on / (" -e "$cdataset on /mnt ("
if [ $? -eq 0 ] ; then
display_loaderopts >>/tmp/.grubdataset.$$.2
else
display_loaderopts $cdataset >> /tmp/.grubdataset.$$.2
fi
# Lets start a submenu for each BE
cat << EOF
# Start FreeBSD BE Submenu
submenu "${NICK} ($b) - ${cdatadate} ${cdatatime}" {
EOF
# Lets do the default entry first
#################################
cat << EOF
menuentry "Standard Bootup" {
EOF
# Get the dataset guts
cat /tmp/.grubdataset.$$.1
echo " kfreebsd ${shortdataset}/@/boot/kernel/kernel"
echo " kfreebsd_loadenv ${shortdataset}@/boot/device.hints"
cat /tmp/.grubdataset.$$.2
# Set any options
cat << EOF
}
EOF
# Next lets do single user mode
#################################
cat << EOF
menuentry "Single User Mode" {
EOF
# Get the dataset guts
cat /tmp/.grubdataset.$$.1
echo " kfreebsd ${shortdataset}/@/boot/kernel/kernel -s"
echo " kfreebsd_loadenv ${shortdataset}@/boot/device.hints"
cat /tmp/.grubdataset.$$.2
# Set any options
cat << EOF
}
EOF
# Close the submenu section
cat << EOF
}
# End FreeBSD BE
EOF
done
# Cleanup after ourselves
if [ -e "/tmp/.grubdataset.$$.1" ] ; then
rm /tmp/.grubdataset.$$.1
rm /tmp/.grubdataset.$$.2
fi
}
# Detect our types of disk layouts
detect_beadm
