mirror of
https://git.hardenedbsd.org/hardenedbsd/HardenedBSD.git
synced 2024-11-15 14:56:13 +01:00
150 lines
2.9 KiB
Bash
Executable File
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"
|