HardenedBSD/etc/rc.d/cfumass
Edward Tomasz Napierala a5efdbd0c9 Make the cfumass rc script support USB template 10.
MFC after:	2 weeks
Sponsored by:	The FreeBSD Foundation
2018-05-27 10:48:21 +00:00

150 lines
2.9 KiB
Bash
Executable File

#!/bin/sh
#
# $FreeBSD$
#
# PROVIDE: cfumass
# REQUIRE: var
# KEYWORD: nojail
. /etc/rc.subr
name="cfumass"
desc="Configure the LUN for device mode USB mass storage"
rcvar="cfumass_enable"
start_cmd="${name}_start"
stop_cmd="${name}_stop"
extra_commands="reload"
reload_cmd="${name}_start"
: ${cfumass_dir:=/var/cfumass}
: ${cfumass_image:=/var/tmp/cfumass.img}
: ${cfumass_vendor:="FreeBSD"}
: ${cfumass_product:="cfumass(4)"}
remove_luns()
{
local _lun _luns
_luns=`ctladm devlist -b block -v | awk '
$1 ~ /^[0-9]+$/ {
lun = $1
}
$1 == "file='"${cfumass_image}"'" {
print lun
}'`
for _lun in ${_luns}; do
ctladm remove -b block -l "${_lun}" > /dev/null
done
}
cfumass_start()
{
local err _files _template _new_template
if [ ! -d "${cfumass_dir}" ]; then
warn "${cfumass_dir} does not exist"
return 1
fi
_files=`find "${cfumass_dir}" -newer "${cfumass_image}" -print 2> /dev/null`
if [ ! -e "${cfumass_image}" -o -n "${_files}" ]; then
# The image doesn't exist or is out of date.
makefs -t cd9660 -o label="${cfumass_vendor}" \
-o rockridge "${cfumass_image}" "${cfumass_dir}"
err=$?
if [ "${err}" -ne 0 ]; then
warn "unable to create ${cfumass_image}"
return "${err}"
fi
fi
remove_luns
ctladm create -b block -o file="${cfumass_image}" -o readonly=on \
-o vendor="${cfumass_vendor}" -o product="${cfumass_product}" \
-S 0 > /dev/null
err=$?
if [ "${err}" -ne 0 ]; then
warn "unable to create CTL LUN"
return "${err}"
fi
load_kld -e cfumass cfumass
# If the template is already switched to Mass Storage, then reset
# it to -1 to force the host to reenumerate it; otherwise it might
# not notice the new LUN.
_template=`sysctl -n hw.usb.template`
if [ "${_template}" -eq 0 ]; then
sysctl hw.usb.template=-1 > /dev/null
err=$?
if [ "${err}" -ne 0 ]; then
warn "unable to set hw.usb.template sysctl"
return "${err}"
fi
fi
# Set the template number based on the current one.
_template=`sysctl -n hw.usb.template`
case "${_template}" in
-1)
_new_template="0"
;;
8)
_new_template="10"
;;
*)
warn "hw.usb.template sysctl set to neither -1 nor 8; not changing"
_new_template=""
;;
esac
if [ -n "${_new_template}" ]; then
sysctl hw.usb.template="${_new_template}" > /dev/null
err=$?
if [ "${err}" -ne 0 ]; then
warn "unable to set hw.usb.template sysctl to ${_new_template}"
return "${err}"
fi
fi
}
cfumass_stop()
{
local err _template _new_template
remove_luns
_template=`sysctl -n hw.usb.template`
case "${_template}" in
0)
_new_template="-1"
;;
10)
_new_template="8"
;;
*)
warn "hw.usb.template sysctl set to neither 0 nor 10; not changing"
_new_template=""
;;
esac
if [ -n "${_new_template}" ]; then
sysctl hw.usb.template="${_new_template}" > /dev/null
err=$?
if [ "${err}" -ne 0 ]; then
warn "unable to set hw.usb.template sysctl to ${_new_template}"
return "${err}"
fi
fi
}
load_rc_config $name
run_rc_command "$1"