1 if [ ! "$_STARTUP_RCVAR_SUBR" ]; then _STARTUP_RCVAR_SUBR=1
3 # Copyright (c) 2006-2013 Devin Teske
6 # Redistribution and use in source and binary forms, with or without
7 # modification, are permitted provided that the following conditions
9 # 1. Redistributions of source code must retain the above copyright
10 # notice, this list of conditions and the following disclaimer.
11 # 2. Redistributions in binary form must reproduce the above copyright
12 # notice, this list of conditions and the following disclaimer in the
13 # documentation and/or other materials provided with the distribution.
15 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 ############################################################ INCLUDES
31 BSDCFG_SHARE="/usr/share/bsdconfig"
32 . $BSDCFG_SHARE/common.subr || exit 1
33 f_dprintf "%s: loading includes..." startup/rcvar.subr
34 f_include $BSDCFG_SHARE/sysrc.subr
36 ############################################################ CONFIGURATION
39 # Default path to the `/etc/rc.d' directory where service(8) scripts are stored
41 : ${ETC_RC_D:=/etc/rc.d}
44 # Default path to `/etc/rc.subr' (for find_local_scripts_new())
46 : ${ETC_RC_SUBR:=/etc/rc.subr}
48 ############################################################ GLOBALS
51 # Initialize in-memory cache variables
57 # Define what an rcvar looks like
59 STARTUP_RCVAR_REGEX='[[:alpha:]_][[:alnum:]_]*="([Yy][Ee][Ss]|[Nn][Oo])"'
62 # Default path to on-disk cache file(s)
64 STARTUP_RCVAR_MAP_CACHEFILE="/var/run/bsdconfig/startup_rcvar_map.cache"
66 ############################################################ FUNCTIONS
68 # f_startup_rcvar_map [$var_to_set]
70 # Produce a map (beit from in-memory cache or on-disk cache) of rc.d scripts
71 # and their associated rcvar's. The map returned has the following format:
73 # rcvar default script description
75 # With each as follows:
77 # rcvar the variable used to enable this rc.d script
78 # default default value for this variable
79 # script the rc.d script in-question
80 # description description of the variable from rc.conf(5) defaults
82 # If $var_to_set is missing or NULL, the map is printed to standard output for
83 # capturing in a sub-shell (which is less-recommended because of performance
84 # degredation; for example, when called in a loop).
88 local __var_to_set="$1"
90 # If the in-memory cached value is available, return it immediately
91 if [ "$_STARTUP_RCVAR_MAP" ]; then
92 if [ "$__var_to_set" ]; then
93 setvar "$__var_to_set" "$STARTUP_RCVAR_MAP"
95 echo "$STARTUP_RCVAR_MAP"
101 # create the in-memory cache (potentially from validated on-disk cache)
104 # Get a list of /etc/rc.d scripts ...
105 local __file __rc_script_list=
106 for __file in "$ETC_RC_D"/*; do
107 [ -f "$__file" ] || continue
108 [ -x "$__file" ] || continue
109 __rc_script_list="$__rc_script_list $__file"
111 # ... and /usr/local/etc/rc.d scripts
112 __rc_script_list="$__rc_script_list $(
113 local_startup=$( f_sysrc_get local_startup )
114 f_include "$ETC_RC_SUBR"
115 find_local_scripts_new
118 __rc_script_list="${__rc_script_list# }" # Trim leading space
121 # Calculate a digest given the checksums of all dependencies (scripts
122 # and the defaults file). This digest will be used to determine if an
123 # on-disk global persistant cache file (containg this digest on the
124 # first line) is valid and can be used to quickly populate the cache
125 # value for immediate return.
127 local __rc_script_list_digest
128 __rc_script_list_digest=$( cd "$ETC_RC_D" &&
129 cksum "$RC_DEFAULTS" $__rc_script_list | md5 )
132 # Check to see if the global persistant cache file exists
134 if [ -f "$STARTUP_RCVAR_MAP_CACHEFILE" ]; then
136 # Attempt to populate the in-memory cache with the (soon to be)
137 # validated on-disk cache. If validation fails, fall-back to
138 # the current value and return error.
141 ( # Get digest as first word on first line
142 read digest rest_ignored
145 # If the stored digest matches the calculated-
146 # one populate the in-memory cache from the on-
147 # disk cache and return success.
149 if [ "$digest" = "$__rc_script_list_digest" ]
154 # Otherwise, return the current value
155 echo "$STARTUP_RCVAR_MAP"
158 ) < "$STARTUP_RCVAR_MAP_CACHEFILE"
161 export STARTUP_RCVAR_MAP # Make children faster (export cache)
162 if [ $__retval -eq $SUCCESS ]; then
163 export _STARTUP_RCVAR_MAP=1
164 if [ "$__var_to_set" ]; then
165 setvar "$__var_to_set" "$STARTUP_RCVAR_MAP"
167 echo "$STARTUP_RCVAR_MAP"
171 # Otherwise, fall-thru to create in-memory cache from scratch
175 # If we reach this point, we need to generate the data from scratch
176 # (and after we do, we'll attempt to create the global persistant
177 # cache file to speed up future executions).
181 for script in $__rc_script_list; do
182 rcvar_list=$( $script rcvar | awk -F= \
183 -v script="$script" '
184 /^'"$STARTUP_RCVAR_REGEX"'/ {
185 if ( $2 ~ /^"[Yy][Ee][Ss]"$/ )
190 for entry in $rcvar_list; do
192 rcvar_default=$( f_sysrc_get_default "$rcvar" )
193 [ "$rcvar_default" ] ||
194 rcvar_default="${entry#*,}"
195 rcvar_desc=$( f_sysrc_desc "$rcvar" )
196 echo $rcvar ${rcvar_default:-NO} \
197 $script "$rcvar_desc"
201 export STARTUP_RCVAR_MAP
202 export _STARTUP_RCVAR_MAP=1
203 if [ "$__var_to_set" ]; then
204 setvar "$__var_to_set" "$STARTUP_RCVAR_MAP"
206 echo "$STARTUP_RCVAR_MAP"
210 # Attempt to create/update the persistant global cache
213 # Create a new temporary file to write to
214 local __tmpfile="$( mktemp -t "$pgm" )"
215 [ "$__tmpfile" ] || return $FAILURE
217 # Write the temporary file contents
218 echo "$__rc_script_list_digest" > "$__tmpfile"
219 echo "$STARTUP_RCVAR_MAP" >> "$__tmpfile"
221 # Finally, move the temporary file into place
222 case "$STARTUP_RCVAR_MAP_CACHEFILE" in
223 */*) f_quietly mkdir -p "${STARTUP_RCVAR_MAP_CACHEFILE%/*}"
225 mv "$__tmpfile" "$STARTUP_RCVAR_MAP_CACHEFILE"
228 ############################################################ MAIN
230 f_dprintf "%s: Successfully loaded." startup/rcvar.subr
232 fi # ! $_STARTUP_RCVAR_SUBR