package/base-files/files/lib/functions.sh

   1 #!/bin/sh
   2 # Copyright (C) 2006 OpenWrt.org
   3 # Copyright (C) 2006 Fokus Fraunhofer <carsten.tittel@fokus.fraunhofer.de>
   4
   5
   6 debug () {
   7         ${DEBUG:-:} "$@"
   8 }
   9 mount() {
  10         busybox mount "$@"
  11 }
  12
  13 # newline
  14 N="
  15 "
  16
  17 _C=0
  18 NO_EXPORT=1
  19 LOAD_STATE=1
  20 LIST_SEP=" "
  21
  22 hotplug_dev() {
  23         env -i ACTION=$1 INTERFACE=$2 /sbin/hotplug-call net
  24 }
  25
  26 append() {
  27         local var="$1"
  28         local value="$2"
  29         local sep="${3:- }"
  30
  31         eval "export ${NO_EXPORT:+-n} -- \"$var=\${$var:+\${$var}\${value:+\$sep}}\$value\""
  32 }
  33
  34 list_contains() {
  35         local var="$1"
  36         local str="$2"
  37         local val
  38
  39         eval "val=\" \${$var} \""
  40         [ "${val%% $str *}" != "$val" ]
  41 }
  42
  43 list_remove() {
  44         local var="$1"
  45         local remove="$2"
  46         local val
  47
  48         eval "val=\" \${$var} \""
  49         val1="${val%% $remove *}"
  50         [ "$val1" = "$val" ] && return
  51         val2="${val##* $remove }"
  52         [ "$val2" = "$val" ] && return
  53         val="${val1## } ${val2%% }"
  54         val="${val%% }"
  55         eval "export ${NO_EXPORT:+-n} -- \"$var=\$val\""
  56 }
  57
  58 config_load() {
  59         [ -n "$IPKG_INSTROOT" ] && return 0
  60         uci_load "$@"
  61 }
  62
  63 reset_cb() {
  64         config_cb() { return 0; }
  65         option_cb() { return 0; }
  66         list_cb() { return 0; }
  67 }
  68 reset_cb
  69
  70 package() {
  71         return 0
  72 }
  73
  74 config () {
  75         local cfgtype="$1"
  76         local name="$2"
  77
  78         export ${NO_EXPORT:+-n} CONFIG_NUM_SECTIONS=$(($CONFIG_NUM_SECTIONS + 1))
  79         name="${name:-cfg$CONFIG_NUM_SECTIONS}"
  80         append CONFIG_SECTIONS "$name"
  81         [ -n "$NO_CALLBACK" ] || config_cb "$cfgtype" "$name"
  82         export ${NO_EXPORT:+-n} CONFIG_SECTION="$name"
  83         export ${NO_EXPORT:+-n} "CONFIG_${CONFIG_SECTION}_TYPE=$cfgtype"
  84 }
  85
  86 option () {
  87         local varname="$1"; shift
  88         local value="$*"
  89
  90         export ${NO_EXPORT:+-n} "CONFIG_${CONFIG_SECTION}_${varname}=$value"
  91         [ -n "$NO_CALLBACK" ] || option_cb "$varname" "$*"
  92 }
  93
  94 list() {
  95         local varname="$1"; shift
  96         local value="$*"
  97         local len
  98
  99         config_get len "$CONFIG_SECTION" "${varname}_LENGTH" 0
 100         [ $len = 0 ] && append CONFIG_LIST_STATE "${CONFIG_SECTION}_${varname}"
 101         len=$(($len + 1))
 102         config_set "$CONFIG_SECTION" "${varname}_ITEM$len" "$value"
 103         config_set "$CONFIG_SECTION" "${varname}_LENGTH" "$len"
 104         append "CONFIG_${CONFIG_SECTION}_${varname}" "$value" "$LIST_SEP"
 105         list_cb "$varname" "$*"
 106 }
 107
 108 config_rename() {
 109         local OLD="$1"
 110         local NEW="$2"
 111         local oldvar
 112         local newvar
 113
 114         [ -n "$OLD" -a -n "$NEW" ] || return
 115         for oldvar in `set | grep ^CONFIG_${OLD}_ | \
 116                 sed -e 's/\(.*\)=.*$/\1/'` ; do
 117                 newvar="CONFIG_${NEW}_${oldvar##CONFIG_${OLD}_}"
 118                 eval "export ${NO_EXPORT:+-n} \"$newvar=\${$oldvar}\""
 119                 unset "$oldvar"
 120         done
 121         export ${NO_EXPORT:+-n} CONFIG_SECTIONS="$(echo " $CONFIG_SECTIONS " | sed -e "s, $OLD , $NEW ,")"
 122
 123         [ "$CONFIG_SECTION" = "$OLD" ] && export ${NO_EXPORT:+-n} CONFIG_SECTION="$NEW"
 124 }
 125
 126 config_unset() {
 127         config_set "$1" "$2" ""
 128 }
 129
 130 config_clear() {
 131         local SECTION="$1"
 132         local oldvar
 133
 134         list_remove CONFIG_SECTIONS "$SECTION"
 135         export ${NO_EXPORT:+-n} CONFIG_SECTIONS="${SECTION:+$CONFIG_SECTIONS}"
 136
 137         for oldvar in `set | grep ^CONFIG_${SECTION:+${SECTION}_} | \
 138                 sed -e 's/\(.*\)=.*$/\1/'` ; do
 139                 unset $oldvar
 140         done
 141 }
 142
 143 # config_get <variable> <section> <option> [<default>]
 144 # config_get <section> <option>
 145 config_get() {
 146         case "$3" in
 147                 "") eval echo "\${CONFIG_${1}_${2}:-\${4}}";;
 148                 *)  eval export ${NO_EXPORT:+-n} -- "${1}=\${CONFIG_${2}_${3}:-\${4}}";;
 149         esac
 150 }
 151
 152 # config_get_bool <variable> <section> <option> [<default>]
 153 config_get_bool() {
 154         local _tmp
 155         config_get _tmp "$2" "$3" "$4"
 156         case "$_tmp" in
 157                 1|on|true|enabled) _tmp=1;;
 158                 0|off|false|disabled) _tmp=0;;
 159                 *) _tmp="$4";;
 160         esac
 161         export ${NO_EXPORT:+-n} "$1=$_tmp"
 162 }
 163
 164 config_set() {
 165         local section="$1"
 166         local option="$2"
 167         local value="$3"
 168         local old_section="$CONFIG_SECTION"
 169
 170         CONFIG_SECTION="$section"
 171         option "$option" "$value"
 172         CONFIG_SECTION="$old_section"
 173 }
 174
 175 config_foreach() {
 176         local function="$1"
 177         [ "$#" -ge 1 ] && shift
 178         local type="$1"
 179         [ "$#" -ge 1 ] && shift
 180         local section cfgtype
 181
 182         [ -z "$CONFIG_SECTIONS" ] && return 0
 183         for section in ${CONFIG_SECTIONS}; do
 184                 config_get cfgtype "$section" TYPE
 185                 [ -n "$type" -a "x$cfgtype" != "x$type" ] && continue
 186                 eval "$function \"\$section\" \"\$@\""
 187         done
 188 }
 189
 190 config_list_foreach() {
 191         [ "$#" -ge 3 ] || return 0
 192         local section="$1"; shift
 193         local option="$1"; shift
 194         local function="$1"; shift
 195         local val
 196         local len
 197         local c=1
 198
 199         config_get len "${section}" "${option}_LENGTH"
 200         [ -z "$len" ] && return 0
 201         while [ $c -le "$len" ]; do
 202                 config_get val "${section}" "${option}_ITEM$c"
 203                 eval "$function \"\$val\" \"$@\""
 204                 c="$(($c + 1))"
 205         done
 206 }
 207
 208 load_modules() {
 209         [ -d /etc/modules.d ] && {
 210                 cd /etc/modules.d
 211                 sed 's/^[^#]/insmod &/' $* | ash 2>&- || :
 212         }
 213 }
 214
 215 include() {
 216         local file
 217
 218         for file in $(ls $1/*.sh 2>/dev/null); do
 219                 . $file
 220         done
 221 }
 222
 223 find_mtd_part() {
 224         local PART="$(grep "\"$1\"" /proc/mtd | awk -F: '{print $1}')"
 225         local PREFIX=/dev/mtdblock
 226
 227         PART="${PART##mtd}"
 228         [ -d /dev/mtdblock ] && PREFIX=/dev/mtdblock/
 229         echo "${PART:+$PREFIX$PART}"
 230 }
 231
 232 strtok() { # <string> { <variable> [<separator>] ... }
 233         local tmp
 234         local val="$1"
 235         local count=0
 236
 237         shift
 238
 239         while [ $# -gt 1 ]; do
 240                 tmp="${val%%$2*}"
 241
 242                 [ "$tmp" = "$val" ] && break
 243
 244                 val="${val#$tmp$2}"
 245
 246                 export ${NO_EXPORT:+-n} "$1=$tmp"; count=$((count+1))
 247                 shift 2
 248         done
 249
 250         if [ $# -gt 0 -a -n "$val" ]; then
 251                 export ${NO_EXPORT:+-n} "$1=$val"; count=$((count+1))
 252         fi
 253
 254         return $count
 255 }
 256
 257
 258 jffs2_mark_erase() {
 259         local part="$(find_mtd_part "$1")"
 260         [ -z "$part" ] && {
 261                 echo Partition not found.
 262                 return 1
 263         }
 264         echo -e "\xde\xad\xc0\xde" | mtd -qq write - "$1"
 265 }
 266
 267 uci_apply_defaults() {
 268         cd /etc/uci-defaults || return 0
 269         files="$(ls)"
 270         [ -z "$files" ] && return 0
 271         mkdir -p /tmp/.uci
 272         for file in $files; do
 273                 ( . "./$(basename $file)" ) && rm -f "$file"
 274         done
 275         uci commit
 276 }
 277
 278 #
 279 # service: simple wrapper around start-stop-daemon
 280 #
 281 # Usage: service ACTION EXEC ARGS...
 282 #
 283 # Action:
 284 #   -S,--start           start EXEC, passing it ARGS as its arguments
 285 #   -K,--stop            stop EXEC (send it a $SERVICE_SIG_STOP signal)
 286 #   -R,--reload          reload EXEC (send it a $SERVICE_SIG_RELOAD signal)
 287 #
 288 # Environment variables used:
 289 #   SERVICE_DAEMONIZE    run EXEC in background
 290 #   SERVICE_WRITE_PID    create a pid file and use it
 291 #   SERVICE_USE_PID      assume EXEC creates its own pid file and use it
 292 #   SERVICE_PID_FILE     pid file to use (default to /var/run/EXEC.pid)
 293 #   SERVICE_SIG_RELOAD   signal used for reloading
 294 #   SERVICE_SIG_STOP     signal used for stopping
 295 #   SERVICE_UID          user EXEC should be run as
 296 #   SERVICE_GID          group EXEC should be run as
 297 #
 298 #   SERVICE_DEBUG        don't do anything, but show what would be done
 299 #   SERVICE_QUIET        don't print anything
 300 #
 301
 302 SERVICE_QUIET=1
 303 SERVICE_SIG_RELOAD="HUP"
 304 SERVICE_SIG_STOP="TERM"
 305
 306 service() {
 307         local ssd
 308         local ssd_pid_file
 309         local ssd_sig
 310         local ssd_start
 311         ssd="${SERVICE_DEBUG:+echo }start-stop-daemon${SERVICE_QUIET:+ -q}"
 312         case "$1" in
 313           -S|--start|start)
 314                 ssd="$ssd -S"
 315                 ssd_start=1
 316                 ;;
 317           -K|--stop|stop)
 318                 ssd="$ssd -K"
 319                 ssd_sig="$SERVICE_SIG_STOP"
 320                 ;;
 321           -R|--reload|reload)
 322                 ssd="$ssd -K"
 323                 ssd_sig="$SERVICE_SIG_RELOAD"
 324                 ;;
 325           *)
 326                 echo "ssd: unknow action '$1'" 1>&2
 327                 return 1
 328         esac
 329         shift
 330         if [ -z "$1" ]; then
 331                 echo "ssd: missing arguments" 1>&2
 332                 return 1
 333         fi
 334         ssd="$ssd -x $1"
 335         if [ -n "$SERVICE_PID_FILE$SERVICE_USE_PID$SERVICE_WRITE_PID" ]; then
 336                 ssd="$ssd -p ${SERVICE_PID_FILE:-/var/run/${1##*/}.pid}"
 337         fi
 338         ssd="$ssd${SERVICE_UID:+ -c $SERVICE_UID${SERVICE_GID:+:$SERVICE_GID}}"
 339         if [ -n "$ssd_start" ]; then
 340                 ssd="$ssd${SERVICE_DAEMONIZE:+ -b}${SERVICE_WRITE_PID:+ -m}"
 341         else
 342                 ssd="$ssd${ssd_sig:+ -s $ssd_sig}"
 343         fi
 344         shift
 345         $ssd${1:+ -- "$@"}
 346 }
 347
 348 service_start() {
 349         service -S "$@"
 350 }
 351
 352 service_stop() {
 353         service -K "$@"
 354 }
 355
 356 service_reload() {
 357         service -R "$@"
 358 }
 359
 360 service_kill() {
 361         local name="${1}"
 362         local pid="${2:-$(pidof "$name")}"
 363         local grace="${3:-5}"
 364
 365         [ -f "$pid" ] && pid="$(head -n1 "$pid" 2>/dev/null)"
 366
 367         for pid in $pid; do
 368                 [ -d "/proc/$pid" ] || continue
 369                 local try=0
 370                 kill -TERM $pid 2>/dev/null && \
 371                         while grep -qs "$name" "/proc/$pid/cmdline" && [ $((try++)) -lt $grace ]; do sleep 1; done
 372                 kill -KILL $pid 2>/dev/null && \
 373                         while grep -qs "$name" "/proc/$pid/cmdline"; do sleep 1; done
 374         done
 375 }
 376
 377
 378 pi_include() {
 379         if [ -f "/tmp/overlay/$1" ]; then
 380                 . "/tmp/overlay/$1"
 381         elif [ -f "$1" ]; then
 382                 . "$1"
 383         elif [ -d "/tmp/overlay/$1" ]; then
 384                 if [ -n "$(ls /tmp/overlay/$1/*.sh 2>/dev/null)" ]; then
 385                         for src_script in /tmp/overlay/$1/*.sh; do
 386                                 . "$src_script"
 387                         done
 388                 fi
 389         elif [ -d "$1" ]; then
 390                 if [ -n "$(ls $1/*.sh 2>/dev/null)" ]; then
 391                         for src_script in $1/*.sh; do
 392                                 . "$src_script"
 393                         done
 394                 fi
 395         else
 396                 echo "WARNING: $1 not found"
 397                 return 1
 398         fi
 399         return 0
 400 }
 401
 402 [ -z "$IPKG_INSTROOT" -a -f /lib/config/uci.sh ] && . /lib/config/uci.sh