634 lines
16 KiB
Plaintext
634 lines
16 KiB
Plaintext
if [ ! "$_SYSRC_SUBR" ]; then _SYSRC_SUBR=1
|
|
#
|
|
# Copyright (c) 2006-2012 Devin Teske
|
|
# All Rights Reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided 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 AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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.
|
|
#
|
|
# $FreeBSD$
|
|
#
|
|
############################################################ INCLUDES
|
|
|
|
BSDCFG_SHARE="/usr/share/bsdconfig"
|
|
[ "$_COMMON_SUBR" ] || . $BSDCFG_SHARE/common.subr || exit 1
|
|
|
|
BSDCFG_LIBE="/usr/libexec/bsdconfig"
|
|
if [ ! "$_SYSRC_JAILED" ]; then
|
|
f_dprintf "%s: loading includes..." sysrc.subr
|
|
f_include_lang $BSDCFG_LIBE/include/messages.subr
|
|
fi
|
|
|
|
############################################################ CONFIGURATION
|
|
|
|
#
|
|
# Standard pathnames (inherit values from shell if available)
|
|
#
|
|
: ${RC_DEFAULTS:="/etc/defaults/rc.conf"}
|
|
|
|
############################################################ GLOBALS
|
|
|
|
#
|
|
# Global exit status variables
|
|
#
|
|
SUCCESS=0
|
|
FAILURE=1
|
|
|
|
#
|
|
# Valid characters that can appear in an sh(1) variable name
|
|
#
|
|
# Please note that the character ranges A-Z and a-z should be avoided because
|
|
# these can include accent characters (which are not valid in a variable name).
|
|
# For example, A-Z matches any character that sorts after A but before Z,
|
|
# including A and Z. Although ASCII order would make more sense, that is not
|
|
# how it works.
|
|
#
|
|
VALID_VARNAME_CHARS="0-9ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz_"
|
|
|
|
############################################################ FUNCTIONS
|
|
|
|
# f_clean_env [ --except $varname ... ]
|
|
#
|
|
# Unset all environment variables in the current scope. An optional list of
|
|
# arguments can be passed, indicating which variables to avoid unsetting; the
|
|
# `--except' is required to enable the exclusion-list as the remainder of
|
|
# positional arguments.
|
|
#
|
|
# Be careful not to call this in a shell that you still expect to perform
|
|
# $PATH expansion in, because this will blow $PATH away. This is best used
|
|
# within a sub-shell block "(...)" or "$(...)" or "`...`".
|
|
#
|
|
f_clean_env()
|
|
{
|
|
local var arg except=
|
|
|
|
#
|
|
# Should we process an exclusion-list?
|
|
#
|
|
if [ "$1" = "--except" ]; then
|
|
except=1
|
|
shift 1
|
|
fi
|
|
|
|
#
|
|
# Loop over a list of variable names from set(1) built-in.
|
|
#
|
|
for var in $( set | awk -F= \
|
|
'/^[[:alpha:]_][[:alnum:]_]*=/ {print $1}' \
|
|
| grep -v '^except$'
|
|
); do
|
|
#
|
|
# In POSIX bourne-shell, attempting to unset(1) OPTIND results
|
|
# in "unset: Illegal number:" and causes abrupt termination.
|
|
#
|
|
[ "$var" = OPTIND ] && continue
|
|
|
|
#
|
|
# Process the exclusion-list?
|
|
#
|
|
if [ "$except" ]; then
|
|
for arg in "$@" ""; do
|
|
[ "$var" = "$arg" ] && break
|
|
done
|
|
[ "$arg" ] && continue
|
|
fi
|
|
|
|
unset "$var"
|
|
done
|
|
}
|
|
|
|
# f_sysrc_get $varname
|
|
#
|
|
# Get a system configuration setting from the collection of system-
|
|
# configuration files (in order: /etc/defaults/rc.conf /etc/rc.conf
|
|
# and /etc/rc.conf).
|
|
#
|
|
# NOTE: Additional shell parameter-expansion formats are supported. For
|
|
# example, passing an argument of "hostname%%.*" (properly quoted) will
|
|
# return the hostname up to (but not including) the first `.' (see sh(1),
|
|
# "Parameter Expansion" for more information on additional formats).
|
|
#
|
|
f_sysrc_get()
|
|
{
|
|
# Sanity check
|
|
[ -f "$RC_DEFAULTS" -a -r "$RC_DEFAULTS" ] || return $FAILURE
|
|
|
|
# Taint-check variable name
|
|
case "$1" in
|
|
[0-9]*)
|
|
# Don't expand possible positional parameters
|
|
return $FAILURE;;
|
|
*)
|
|
[ "$1" ] || return $FAILURE
|
|
esac
|
|
|
|
( # Execute within sub-shell to protect parent environment
|
|
|
|
#
|
|
# Clear the environment of all variables, preventing the
|
|
# expansion of normals such as `PS1', `TERM', etc.
|
|
#
|
|
f_clean_env --except IFS RC_CONFS RC_DEFAULTS
|
|
|
|
. "$RC_DEFAULTS" > /dev/null 2>&1
|
|
|
|
unset RC_DEFAULTS
|
|
# no longer needed
|
|
|
|
#
|
|
# If the query is for `rc_conf_files' then store the value that
|
|
# we inherited from sourcing RC_DEFAULTS (above) so that we may
|
|
# conditionally restore this value after source_rc_confs in the
|
|
# event that RC_CONFS does not customize the value.
|
|
#
|
|
if [ "$1" = "rc_conf_files" ]; then
|
|
_rc_conf_files="$rc_conf_files"
|
|
fi
|
|
|
|
#
|
|
# If RC_CONFS is defined, set $rc_conf_files to an explicit
|
|
# value, modifying the default behavior of source_rc_confs().
|
|
#
|
|
if [ "${RC_CONFS+set}" ]; then
|
|
rc_conf_files="$RC_CONFS"
|
|
_rc_confs_set=1
|
|
fi
|
|
|
|
source_rc_confs > /dev/null 2>&1
|
|
|
|
#
|
|
# If the query was for `rc_conf_files' AND after calling
|
|
# source_rc_confs the value has not changed, then we should
|
|
# restore the value to the one inherited from RC_DEFAULTS
|
|
# before performing the final query (preventing us from
|
|
# returning what was set via RC_CONFS when the intent was
|
|
# instead to query the value from the file(s) specified).
|
|
#
|
|
if [ "$1" = "rc_conf_files" -a \
|
|
"$_rc_confs_set" -a \
|
|
"$rc_conf_files" = "$RC_CONFS" \
|
|
]; then
|
|
rc_conf_files="$_rc_conf_files"
|
|
unset _rc_conf_files
|
|
unset _rc_confs_set
|
|
fi
|
|
|
|
unset RC_CONFS
|
|
# no longer needed
|
|
|
|
#
|
|
# This must be the last functional line for both the sub-shell
|
|
# and the function to preserve the return status from formats
|
|
# such as "${varname?}" and "${varname:?}" (see "Parameter
|
|
# Expansion" in sh(1) for more information).
|
|
#
|
|
eval echo '"${'"$1"'}"' 2> /dev/null
|
|
)
|
|
}
|
|
|
|
# f_sysrc_get_default $varname
|
|
#
|
|
# Get a system configuration default setting from the default rc.conf(5) file
|
|
# (or whatever RC_DEFAULTS points at).
|
|
#
|
|
f_sysrc_get_default()
|
|
{
|
|
# Sanity check
|
|
[ -f "$RC_DEFAULTS" -a -r "$RC_DEFAULTS" ] || return $FAILURE
|
|
|
|
# Taint-check variable name
|
|
case "$1" in
|
|
[0-9]*)
|
|
# Don't expand possible positional parameters
|
|
return $FAILURE;;
|
|
*)
|
|
[ "$1" ] || return $FAILURE
|
|
esac
|
|
|
|
( # Execute within sub-shell to protect parent environment
|
|
|
|
#
|
|
# Clear the environment of all variables, preventing the
|
|
# expansion of normals such as `PS1', `TERM', etc.
|
|
#
|
|
f_clean_env --except RC_DEFAULTS
|
|
|
|
. "$RC_DEFAULTS" > /dev/null 2>&1
|
|
|
|
unset RC_DEFAULTS
|
|
# no longer needed
|
|
|
|
#
|
|
# This must be the last functional line for both the sub-shell
|
|
# and the function to preserve the return status from formats
|
|
# such as "${varname?}" and "${varname:?}" (see "Parameter
|
|
# Expansion" in sh(1) for more information).
|
|
#
|
|
eval echo '"${'"$1"'}"' 2> /dev/null
|
|
)
|
|
}
|
|
|
|
# f_sysrc_find $varname
|
|
#
|
|
# Find which file holds the effective last-assignment to a given variable
|
|
# within the rc.conf(5) file(s).
|
|
#
|
|
# If the variable is found in any of the rc.conf(5) files, the function prints
|
|
# the filename it was found in and then returns success. Otherwise output is
|
|
# NULL and the function returns with error status.
|
|
#
|
|
f_sysrc_find()
|
|
{
|
|
local varname="${1%%[!$VALID_VARNAME_CHARS]*}"
|
|
local regex="^[[:space:]]*$varname="
|
|
local rc_conf_files="$( f_sysrc_get rc_conf_files )"
|
|
local conf_files=
|
|
local file
|
|
|
|
# Check parameters
|
|
case "$varname" in
|
|
""|[0-9]*) return $FAILURE
|
|
esac
|
|
|
|
#
|
|
# If RC_CONFS is defined, set $rc_conf_files to an explicit
|
|
# value, modifying the default behavior of source_rc_confs().
|
|
#
|
|
[ "${RC_CONFS+set}" ] && rc_conf_files="$RC_CONFS"
|
|
|
|
#
|
|
# Reverse the order of files in rc_conf_files (the boot process sources
|
|
# these in order, so we will search them in reverse-order to find the
|
|
# last-assignment -- the one that ultimately effects the environment).
|
|
#
|
|
for file in $rc_conf_files; do
|
|
conf_files="$file${conf_files:+ }$conf_files"
|
|
done
|
|
|
|
#
|
|
# Append the defaults file (since directives in the defaults file
|
|
# indeed affect the boot process, we'll want to know when a directive
|
|
# is found there).
|
|
#
|
|
conf_files="$conf_files${conf_files:+ }$RC_DEFAULTS"
|
|
|
|
#
|
|
# Find which file matches assignment to the given variable name.
|
|
#
|
|
for file in $conf_files; do
|
|
[ -f "$file" -a -r "$file" ] || continue
|
|
if grep -Eq "$regex" $file; then
|
|
echo $file
|
|
return $SUCCESS
|
|
fi
|
|
done
|
|
|
|
return $FAILURE # Not found
|
|
}
|
|
|
|
# f_sysrc_desc $varname
|
|
#
|
|
# Attempts to return the comments associated with varname from the rc.conf(5)
|
|
# defaults file `/etc/defaults/rc.conf' (or whatever RC_DEFAULTS points to).
|
|
#
|
|
# Multi-line comments are joined together. Results are NULL if no description
|
|
# could be found.
|
|
#
|
|
# This function is a two-parter. Below is the awk(1) portion of the function,
|
|
# afterward is the sh(1) function which utilizes the below awk script.
|
|
#
|
|
f_sysrc_desc_awk='
|
|
# Variables that should be defined on the invocation line:
|
|
# -v varname="varname"
|
|
#
|
|
BEGIN {
|
|
regex = "^[[:space:]]*"varname"="
|
|
found = 0
|
|
buffer = ""
|
|
}
|
|
{
|
|
if ( ! found )
|
|
{
|
|
if ( ! match($0, regex) ) next
|
|
|
|
found = 1
|
|
sub(/^[^#]*(#[[:space:]]*)?/, "")
|
|
buffer = $0
|
|
next
|
|
}
|
|
|
|
if ( !/^[[:space:]]*#/ ||
|
|
/^[[:space:]]*[[:alpha:]_][[:alnum:]_]*=/ ||
|
|
/^[[:space:]]*#[[:alpha:]_][[:alnum:]_]*=/ ||
|
|
/^[[:space:]]*$/ ) exit
|
|
|
|
sub(/(.*#)*[[:space:]]*/, "")
|
|
buffer = buffer" "$0
|
|
}
|
|
END {
|
|
# Clean up the buffer
|
|
sub(/^[[:space:]]*/, "", buffer)
|
|
sub(/[[:space:]]*$/, "", buffer)
|
|
|
|
print buffer
|
|
exit ! found
|
|
}
|
|
'
|
|
f_sysrc_desc()
|
|
{
|
|
awk -v varname="$1" "$f_sysrc_desc_awk" < "$RC_DEFAULTS"
|
|
}
|
|
|
|
# f_sysrc_set $varname $new_value
|
|
#
|
|
# Change a setting in the system configuration files (edits the files in-place
|
|
# to change the value in the last assignment to the variable). If the variable
|
|
# does not appear in the source file, it is appended to the end of the primary
|
|
# system configuration file `/etc/rc.conf'.
|
|
#
|
|
# This function is a two-parter. Below is the awk(1) portion of the function,
|
|
# afterward is the sh(1) function which utilizes the below awk script.
|
|
#
|
|
f_sysrc_set_awk='
|
|
# Variables that should be defined on the invocation line:
|
|
# -v varname="varname"
|
|
# -v new_value="new_value"
|
|
#
|
|
BEGIN {
|
|
regex = "^[[:space:]]*"varname"="
|
|
found = retval = 0
|
|
}
|
|
{
|
|
# If already found... just spew
|
|
if ( found ) { print; next }
|
|
|
|
# Does this line match an assignment to our variable?
|
|
if ( ! match($0, regex) ) { print; next }
|
|
|
|
# Save important match information
|
|
found = 1
|
|
matchlen = RSTART + RLENGTH - 1
|
|
|
|
# Store the value text for later munging
|
|
value = substr($0, matchlen + 1, length($0) - matchlen)
|
|
|
|
# Store the first character of the value
|
|
t1 = t2 = substr(value, 0, 1)
|
|
|
|
# Assignment w/ back-ticks, expression, or misc.
|
|
# We ignore these since we did not generate them
|
|
#
|
|
if ( t1 ~ /[`$\\]/ ) { retval = 1; print; next }
|
|
|
|
# Assignment w/ single-quoted value
|
|
else if ( t1 == "'\''" ) {
|
|
sub(/^'\''[^'\'']*/, "", value)
|
|
if ( length(value) == 0 ) t2 = ""
|
|
sub(/^'\''/, "", value)
|
|
}
|
|
|
|
# Assignment w/ double-quoted value
|
|
else if ( t1 == "\"" ) {
|
|
sub(/^"(.*\\\\+")*[^"]*/, "", value)
|
|
if ( length(value) == 0 ) t2 = ""
|
|
sub(/^"/, "", value)
|
|
}
|
|
|
|
# Assignment w/ non-quoted value
|
|
else if ( t1 ~ /[^[:space:];]/ ) {
|
|
t1 = t2 = "\""
|
|
sub(/^[^[:space:]]*/, "", value)
|
|
}
|
|
|
|
# Null-assignment
|
|
else if ( t1 ~ /[[:space:];]/ ) { t1 = t2 = "\"" }
|
|
|
|
printf "%s%c%s%c%s\n", substr($0, 0, matchlen), \
|
|
t1, new_value, t2, value
|
|
}
|
|
END { exit retval }
|
|
'
|
|
f_sysrc_set()
|
|
{
|
|
local varname="$1" new_value="$2"
|
|
|
|
# Check arguments
|
|
[ "$varname" ] || return $FAILURE
|
|
|
|
#
|
|
# Find which rc.conf(5) file contains the last-assignment
|
|
#
|
|
local not_found=
|
|
local file="$( f_sysrc_find "$varname" )"
|
|
if [ "$file" = "$RC_DEFAULTS" -o ! "$file" ]; then
|
|
#
|
|
# We either got a null response (not found) or the variable
|
|
# was only found in the rc.conf(5) defaults. In either case,
|
|
# let's instead modify the first file from $rc_conf_files.
|
|
#
|
|
|
|
not_found=1
|
|
|
|
#
|
|
# If RC_CONFS is defined, use $RC_CONFS
|
|
# rather than $rc_conf_files.
|
|
#
|
|
if [ "${RC_CONFS+set}" ]; then
|
|
file="${RC_CONFS%%[$IFS]*}"
|
|
else
|
|
file=$( f_sysrc_get 'rc_conf_files%%[$IFS]*' )
|
|
fi
|
|
fi
|
|
|
|
#
|
|
# If not found, append new value to last file and return.
|
|
#
|
|
if [ "$not_found" ]; then
|
|
echo "$varname=\"$new_value\"" >> "$file"
|
|
return $?
|
|
fi
|
|
|
|
#
|
|
# Perform sanity checks.
|
|
#
|
|
if [ ! -w "$file" ]; then
|
|
f_err "$msg_cannot_create_permission_denied\n" \
|
|
"$pgm" "$file"
|
|
return $FAILURE
|
|
fi
|
|
|
|
#
|
|
# Create a new temporary file to write to.
|
|
#
|
|
local tmpfile="$( mktemp -t "$pgm" )"
|
|
[ "$tmpfile" ] || return $FAILURE
|
|
|
|
#
|
|
# Fixup permissions (else we're in for a surprise, as mktemp(1) creates
|
|
# the temporary file with 0600 permissions, and if we simply mv(1) the
|
|
# temporary file over the destination, the destination will inherit the
|
|
# permissions from the temporary file).
|
|
#
|
|
local mode
|
|
mode=$( stat -f '%#Lp' "$file" 2> /dev/null )
|
|
f_quietly chmod "${mode:-0644}" "$tmpfile"
|
|
|
|
#
|
|
# Fixup ownership. The destination file _is_ writable (we tested
|
|
# earlier above). However, this will fail if we don't have sufficient
|
|
# permissions (so we throw stderr into the bit-bucket).
|
|
#
|
|
local owner
|
|
owner=$( stat -f '%u:%g' "$file" 2> /dev/null )
|
|
f_quietly chown "${owner:-root:wheel}" "$tmpfile"
|
|
|
|
#
|
|
# Operate on the matching file, replacing only the last occurrence.
|
|
#
|
|
local new_contents retval
|
|
new_contents=$( tail -r $file 2> /dev/null )
|
|
new_contents=$( echo "$new_contents" | awk -v varname="$varname" \
|
|
-v new_value="$new_value" "$f_sysrc_set_awk" )
|
|
retval=$?
|
|
|
|
#
|
|
# Write the temporary file contents.
|
|
#
|
|
echo "$new_contents" | tail -r > "$tmpfile" || return $FAILURE
|
|
if [ $retval -ne $SUCCESS ]; then
|
|
echo "$varname=\"$new_value\"" >> "$tmpfile"
|
|
fi
|
|
|
|
#
|
|
# Taint-check our results.
|
|
#
|
|
if ! /bin/sh -n "$tmpfile"; then
|
|
f_err "$msg_previous_syntax_errors\n" "$pgm" "$file"
|
|
rm -f "$tmpfile"
|
|
return $FAILURE
|
|
fi
|
|
|
|
#
|
|
# Finally, move the temporary file into place.
|
|
#
|
|
mv "$tmpfile" "$file"
|
|
}
|
|
|
|
# f_sysrc_delete $varname
|
|
#
|
|
# Remove a setting from the system configuration files (edits files in-place).
|
|
# Deletes all assignments to the given variable in all config files. If the
|
|
# `-f file' option is passed, the removal is restricted to only those files
|
|
# specified, otherwise the system collection of rc_conf_files is used.
|
|
#
|
|
# This function is a two-parter. Below is the awk(1) portion of the function,
|
|
# afterward is the sh(1) function which utilizes the below awk script.
|
|
#
|
|
f_sysrc_delete_awk='
|
|
# Variables that should be defined on the invocation line:
|
|
# -v varname="varname"
|
|
#
|
|
BEGIN {
|
|
regex = "^[[:space:]]*"varname"="
|
|
found = 0
|
|
}
|
|
{
|
|
if ( $0 ~ regex )
|
|
found = 1
|
|
else
|
|
print
|
|
}
|
|
END { exit ! found }
|
|
'
|
|
f_sysrc_delete()
|
|
{
|
|
local varname="$1"
|
|
local file
|
|
|
|
# Check arguments
|
|
[ "$varname" ] || return $FAILURE
|
|
|
|
#
|
|
# Operate on each of the specified files
|
|
#
|
|
for file in ${RC_CONFS-$( f_sysrc_get rc_conf_files )}; do
|
|
[ -e "$file" ] || continue
|
|
|
|
#
|
|
# Create a new temporary file to write to.
|
|
#
|
|
local tmpfile="$( mktemp -t "$pgm" )"
|
|
[ "$tmpfile" ] || return $FAILURE
|
|
|
|
#
|
|
# Fixup permissions and ownership (mktemp(1) defaults to 0600
|
|
# permissions) to instead match the destination file.
|
|
#
|
|
local mode owner
|
|
mode=$( stat -f '%#Lp' "$file" 2> /dev/null )
|
|
owner=$( stat -f '%u:%g' "$file" 2> /dev/null )
|
|
f_quietly chmod "${mode:-0644}" "$tmpfile"
|
|
f_quietly chown "${owner:-root:wheel}" "$tmpfile"
|
|
|
|
#
|
|
# Operate on the file, removing all occurrences, saving the
|
|
# output in our temporary file.
|
|
#
|
|
awk -v varname="$varname" "$f_sysrc_delete_awk" "$file" \
|
|
> "$tmpfile"
|
|
if [ $? -ne $SUCCESS ]; then
|
|
# The file didn't contain any assignments
|
|
rm -f "$tmpfile"
|
|
continue
|
|
fi
|
|
|
|
#
|
|
# Taint-check our results.
|
|
#
|
|
if ! /bin/sh -n "$tmpfile"; then
|
|
f_err "$msg_previous_syntax_errors\n" \
|
|
"$pgm" "$file"
|
|
rm -f "$tmpfile"
|
|
return $FAILURE
|
|
fi
|
|
|
|
#
|
|
# Perform sanity checks
|
|
#
|
|
if [ ! -w "$file" ]; then
|
|
f_err "$msg_permission_denied\n" "$pgm" "$file"
|
|
rm -f "$tmpfile"
|
|
return $FAILURE
|
|
fi
|
|
|
|
#
|
|
# Finally, move the temporary file into place.
|
|
#
|
|
mv "$tmpfile" "$file"
|
|
done
|
|
}
|
|
|
|
############################################################ MAIN
|
|
|
|
f_dprintf "%s: Successfully loaded." sysrc.subr
|
|
|
|
fi # ! $_SYSRC_SUBR
|