54e69b8332a23372d7298106e509fdf696a4d735
[openwrt.git] / package / base-files / files / lib / functions.sh
1 #!/bin/sh
2 # Copyright (C) 2006-2011 OpenWrt.org
3 # Copyright (C) 2006 Fokus Fraunhofer <carsten.tittel@fokus.fraunhofer.de>
4
5
6 debug () {
7         ${DEBUG:-:} "$@"
8 }
9
10 # newline
11 N="
12 "
13
14 _C=0
15 NO_EXPORT=1
16 LOAD_STATE=1
17 LIST_SEP=" "
18
19 hotplug_dev() {
20         env -i ACTION=$1 INTERFACE=$2 /sbin/hotplug-call net
21 }
22
23 append() {
24         local var="$1"
25         local value="$2"
26         local sep="${3:- }"
27
28         eval "export ${NO_EXPORT:+-n} -- \"$var=\${$var:+\${$var}\${value:+\$sep}}\$value\""
29 }
30
31 list_contains() {
32         local var="$1"
33         local str="$2"
34         local val
35
36         eval "val=\" \${$var} \""
37         [ "${val%% $str *}" != "$val" ]
38 }
39
40 list_remove() {
41         local var="$1"
42         local remove="$2"
43         local val
44
45         eval "val=\" \${$var} \""
46         val1="${val%% $remove *}"
47         [ "$val1" = "$val" ] && return
48         val2="${val##* $remove }"
49         [ "$val2" = "$val" ] && return
50         val="${val1## } ${val2%% }"
51         val="${val%% }"
52         eval "export ${NO_EXPORT:+-n} -- \"$var=\$val\""
53 }
54
55 config_load() {
56         [ -n "$IPKG_INSTROOT" ] && return 0
57         uci_load "$@"
58 }
59
60 reset_cb() {
61         config_cb() { return 0; }
62         option_cb() { return 0; }
63         list_cb() { return 0; }
64 }
65 reset_cb
66
67 package() {
68         return 0
69 }
70
71 config () {
72         local cfgtype="$1"
73         local name="$2"
74
75         export ${NO_EXPORT:+-n} CONFIG_NUM_SECTIONS=$(($CONFIG_NUM_SECTIONS + 1))
76         name="${name:-cfg$CONFIG_NUM_SECTIONS}"
77         append CONFIG_SECTIONS "$name"
78         [ -n "$NO_CALLBACK" ] || config_cb "$cfgtype" "$name"
79         export ${NO_EXPORT:+-n} CONFIG_SECTION="$name"
80         export ${NO_EXPORT:+-n} "CONFIG_${CONFIG_SECTION}_TYPE=$cfgtype"
81 }
82
83 option () {
84         local varname="$1"; shift
85         local value="$*"
86
87         export ${NO_EXPORT:+-n} "CONFIG_${CONFIG_SECTION}_${varname}=$value"
88         [ -n "$NO_CALLBACK" ] || option_cb "$varname" "$*"
89 }
90
91 list() {
92         local varname="$1"; shift
93         local value="$*"
94         local len
95
96         config_get len "$CONFIG_SECTION" "${varname}_LENGTH" 0
97         [ $len = 0 ] && append CONFIG_LIST_STATE "${CONFIG_SECTION}_${varname}"
98         len=$(($len + 1))
99         config_set "$CONFIG_SECTION" "${varname}_ITEM$len" "$value"
100         config_set "$CONFIG_SECTION" "${varname}_LENGTH" "$len"
101         append "CONFIG_${CONFIG_SECTION}_${varname}" "$value" "$LIST_SEP"
102         list_cb "$varname" "$*"
103 }
104
105 config_rename() {
106         local OLD="$1"
107         local NEW="$2"
108         local oldvar
109         local newvar
110
111         [ -n "$OLD" -a -n "$NEW" ] || return
112         for oldvar in `set | grep ^CONFIG_${OLD}_ | \
113                 sed -e 's/\(.*\)=.*$/\1/'` ; do
114                 newvar="CONFIG_${NEW}_${oldvar##CONFIG_${OLD}_}"
115                 eval "export ${NO_EXPORT:+-n} \"$newvar=\${$oldvar}\""
116                 unset "$oldvar"
117         done
118         export ${NO_EXPORT:+-n} CONFIG_SECTIONS="$(echo " $CONFIG_SECTIONS " | sed -e "s, $OLD , $NEW ,")"
119
120         [ "$CONFIG_SECTION" = "$OLD" ] && export ${NO_EXPORT:+-n} CONFIG_SECTION="$NEW"
121 }
122
123 config_unset() {
124         config_set "$1" "$2" ""
125 }
126
127 config_clear() {
128         local SECTION="$1"
129         local oldvar
130
131         list_remove CONFIG_SECTIONS "$SECTION"
132         export ${NO_EXPORT:+-n} CONFIG_SECTIONS="${SECTION:+$CONFIG_SECTIONS}"
133
134         for oldvar in `set | grep ^CONFIG_${SECTION:+${SECTION}_} | \
135                 sed -e 's/\(.*\)=.*$/\1/'` ; do
136                 unset $oldvar
137         done
138 }
139
140 # config_get <variable> <section> <option> [<default>]
141 # config_get <section> <option>
142 config_get() {
143         case "$3" in
144                 "") eval echo "\${CONFIG_${1}_${2}:-\${4}}";;
145                 *)  eval export ${NO_EXPORT:+-n} -- "${1}=\${CONFIG_${2}_${3}:-\${4}}";;
146         esac
147 }
148
149 # config_get_bool <variable> <section> <option> [<default>]
150 config_get_bool() {
151         local _tmp
152         config_get _tmp "$2" "$3" "$4"
153         case "$_tmp" in
154                 1|on|true|enabled) _tmp=1;;
155                 0|off|false|disabled) _tmp=0;;
156                 *) _tmp="$4";;
157         esac
158         export ${NO_EXPORT:+-n} "$1=$_tmp"
159 }
160
161 config_set() {
162         local section="$1"
163         local option="$2"
164         local value="$3"
165         local old_section="$CONFIG_SECTION"
166
167         CONFIG_SECTION="$section"
168         option "$option" "$value"
169         CONFIG_SECTION="$old_section"
170 }
171
172 config_foreach() {
173         local ___function="$1"
174         [ "$#" -ge 1 ] && shift
175         local ___type="$1"
176         [ "$#" -ge 1 ] && shift
177         local section cfgtype
178
179         [ -z "$CONFIG_SECTIONS" ] && return 0
180         for section in ${CONFIG_SECTIONS}; do
181                 config_get cfgtype "$section" TYPE
182                 [ -n "$___type" -a "x$cfgtype" != "x$___type" ] && continue
183                 eval "$___function \"\$section\" \"\$@\""
184         done
185 }
186
187 config_list_foreach() {
188         [ "$#" -ge 3 ] || return 0
189         local section="$1"; shift
190         local option="$1"; shift
191         local function="$1"; shift
192         local val
193         local len
194         local c=1
195
196         config_get len "${section}" "${option}_LENGTH"
197         [ -z "$len" ] && return 0
198         while [ $c -le "$len" ]; do
199                 config_get val "${section}" "${option}_ITEM$c"
200                 eval "$function \"\$val\" \"$@\""
201                 c="$(($c + 1))"
202         done
203 }
204
205 load_modules() {
206         [ -d /etc/modules.d ] && {
207                 cd /etc/modules.d
208                 sed 's/^[^#]/insmod &/' $* | ash 2>&- || :
209         }
210 }
211
212 include() {
213         local file
214
215         for file in $(ls $1/*.sh 2>/dev/null); do
216                 . $file
217         done
218 }
219
220 find_mtd_index() {
221         local PART="$(grep "\"$1\"" /proc/mtd | awk -F: '{print $1}')"
222         local INDEX="${PART##mtd}"
223
224         echo ${INDEX}
225 }
226
227 find_mtd_part() {
228         local INDEX=$(find_mtd_index "$1")
229         local PREFIX=/dev/mtdblock
230
231         [ -d /dev/mtdblock ] && PREFIX=/dev/mtdblock/
232         echo "${INDEX:+$PREFIX$INDEX}"
233 }
234
235 find_mtd_chardev() {
236         local INDEX=$(find_mtd_index "$1")
237         local PREFIX=/dev/mtd
238
239         [ -d /dev/mtd ] && PREFIX=/dev/mtd/
240         echo "${INDEX:+$PREFIX$INDEX}"
241 }
242
243 strtok() { # <string> { <variable> [<separator>] ... }
244         local tmp
245         local val="$1"
246         local count=0
247
248         shift
249
250         while [ $# -gt 1 ]; do
251                 tmp="${val%%$2*}"
252
253                 [ "$tmp" = "$val" ] && break
254
255                 val="${val#$tmp$2}"
256
257                 export ${NO_EXPORT:+-n} "$1=$tmp"; count=$((count+1))
258                 shift 2
259         done
260
261         if [ $# -gt 0 -a -n "$val" ]; then
262                 export ${NO_EXPORT:+-n} "$1=$val"; count=$((count+1))
263         fi
264
265         return $count
266 }
267
268
269 jffs2_mark_erase() {
270         local part="$(find_mtd_part "$1")"
271         [ -z "$part" ] && {
272                 echo Partition not found.
273                 return 1
274         }
275         echo -e "\xde\xad\xc0\xde" | mtd -qq write - "$1"
276 }
277
278 uci_apply_defaults() {
279         cd /etc/uci-defaults || return 0
280         files="$(ls)"
281         [ -z "$files" ] && return 0
282         mkdir -p /tmp/.uci
283         for file in $files; do
284                 ( . "./$(basename $file)" ) && rm -f "$file"
285         done
286         uci commit
287 }
288
289 group_add() {
290         local name="$1"
291         local gid="$2"
292         local rc
293         [ -f "${IPKG_INSTROOT}/etc/group" ] || return 1
294         [ -n "$IPKG_INSTROOT" ] || lock /var/lock/group
295         echo "${name}:x:${gid}:" >> ${IPKG_INSTROOT}/etc/group
296         rc=$?
297         [ -n "$IPKG_INSTROOT" ] || lock -u /var/lock/group
298         return $rc
299 }
300
301 group_exists() {
302         grep -qs "^${1}:" ${IPKG_INSTROOT}/etc/group
303 }
304
305 user_add() {
306         local name="${1}"
307         local uid="${2}"
308         local gid="${3:-$2}"
309         local desc="${4:-$1}"
310         local home="${5:-/var/run/$1}"
311         local shell="${6:-/bin/false}"
312         local rc
313         [ -f "${IPKG_INSTROOT}/etc/passwd" ] || return 1
314         [ -n "$IPKG_INSTROOT" ] || lock /var/lock/passwd
315         echo "${name}:x:${uid}:${gid}:${desc}:${home}:${shell}" >> ${IPKG_INSTROOT}/etc/passwd
316         echo "${name}:x:0:0:99999:7:::" >> ${IPKG_INSTROOT}/etc/shadow
317         rc=$?
318         [ -n "$IPKG_INSTROOT" ] || lock -u /var/lock/passwd
319         return $rc
320 }
321
322 user_exists() {
323         grep -qs "^${1}:" ${IPKG_INSTROOT}/etc/passwd
324 }
325
326
327 pi_include() {
328         if [ -f "/tmp/overlay/$1" ]; then
329                 . "/tmp/overlay/$1"
330         elif [ -f "$1" ]; then
331                 . "$1"
332         elif [ -d "/tmp/overlay/$1" ]; then
333                 if [ -n "$(ls /tmp/overlay/$1/*.sh 2>/dev/null)" ]; then
334                         for src_script in /tmp/overlay/$1/*.sh; do
335                                 . "$src_script"
336                         done
337                 fi
338         elif [ -d "$1" ]; then
339                 if [ -n "$(ls $1/*.sh 2>/dev/null)" ]; then
340                         for src_script in $1/*.sh; do
341                                 . "$src_script"
342                         done
343                 fi
344         else
345                 echo "WARNING: $1 not found"
346                 return 1
347         fi
348         return 0
349 }
350
351 #!/bin/sh
352 # Copyright (C) 2006-2010 OpenWrt.org
353 # Copyright (C) 2010 Vertical Communications
354
355 boot_hook_splice_start() {
356         export -n PI_HOOK_SPLICE=1
357 }
358
359 boot_hook_splice_finish() {
360         local hook
361         for hook in $PI_STACK_LIST; do
362                 local v; eval "v=\${${hook}_splice:+\$${hook}_splice }$hook"
363                 export -n "${hook}=${v% }"
364                 export -n "${hook}_splice="
365         done
366         export -n PI_HOOK_SPLICE=
367 }
368
369 boot_hook_init() {
370         local hook="${1}_hook"
371         export -n "PI_STACK_LIST=${PI_STACK_LIST:+$PI_STACK_LIST }$hook"
372         export -n "$hook="
373 }
374
375 boot_hook_add() {
376         local hook="${1}_hook${PI_HOOK_SPLICE:+_splice}"
377         local func="${2}"
378
379         [ -n "$func" ] && {
380                 local v; eval "v=\$$hook"
381                 export -n "$hook=${v:+$v }$func"
382         }
383 }
384
385 boot_hook_shift() {
386         local hook="${1}_hook"
387         local rvar="${2}"
388
389         local v; eval "v=\$$hook"
390         [ -n "$v" ] && {
391                 local first="${v%% *}"
392
393                 [ "$v" != "${v#* }" ] && \
394                         export -n "$hook=${v#* }" || \
395                         export -n "$hook="
396
397                 export -n "$rvar=$first"
398                 return 0
399         }
400
401         return 1
402 }
403
404 boot_run_hook() {
405         local hook="$1"
406         local func
407
408         while boot_hook_shift "$hook" func; do
409                 local ran; eval "ran=\$PI_RAN_$func"
410                 [ -n "$ran" ] || {
411                         export -n "PI_RAN_$func=1"
412                         $func "$1" "$2"
413                 }
414         done
415 }
416
417 jffs2_ready () {
418         mtdpart="$(find_mtd_part rootfs_data)"
419         [ -z "$mtdpart" ] && return 1
420         magic=$(hexdump $mtdpart -n 4 -e '4/1 "%02x"')
421         [ "$magic" != "deadc0de" ]
422 }
423
424 dupe() { # <new_root> <old_root>
425         cd $1
426         echo -n "creating directories... "
427         {
428                 cd $2
429                 find . -xdev -type d
430                 echo "./dev ./overlay ./mnt ./proc ./tmp"
431                 # xdev skips mounted directories
432                 cd $1
433         } | xargs mkdir -p
434         echo "done"
435
436         echo -n "setting up symlinks... "
437         for file in $(cd $2; find . -xdev -type f;); do
438                 case "$file" in
439                 ./rom/note) ;; #nothing
440                 ./etc/config*|\
441                 ./usr/lib/opkg/info/*) cp -af $2/$file $file;;
442                 *) ln -sf /rom/${file#./*} $file;;
443                 esac
444         done
445         for file in $(cd $2; find . -xdev -type l;); do
446                 cp -af $2/${file#./*} $file
447         done
448         echo "done"
449 }
450
451 pivot() { # <new_root> <old_root>
452         mount -o noatime,move /proc $1/proc && \
453         pivot_root $1 $1$2 && {
454                 mount -o noatime,move $2/dev /dev
455                 mount -o noatime,move $2/tmp /tmp
456                 mount -o noatime,move $2/sys /sys 2>&-
457                 mount -o noatime,move $2/overlay /overlay 2>&-
458                 return 0
459         }
460 }
461
462 fopivot() { # <rw_root> <ro_root> <dupe?>
463         root=$1
464         {
465                 if grep -q overlay /proc/filesystems; then
466                         mount -o noatime,lowerdir=/,upperdir=$1 -t overlayfs "overlayfs:$1" /mnt && root=/mnt
467                 elif grep -q mini_fo /proc/filesystems; then
468                         mount -t mini_fo -o noatime,base=/,sto=$1 "mini_fo:$1" /mnt 2>&- && root=/mnt
469                 else
470                         mount --bind -o noatime / /mnt
471                         mount --bind -o noatime,union "$1" /mnt && root=/mnt
472                 fi
473         } || {
474                 [ "$3" = "1" ] && {
475                 mount | grep "on $1 type" 2>&- 1>&- || mount -o noatime,bind $1 $1
476                 dupe $1 $rom
477                 }
478         }
479         pivot $root $2
480 }
481
482 ramoverlay() {
483         mkdir -p /tmp/root
484         mount -t tmpfs -o noatime,mode=0755 root /tmp/root
485         fopivot /tmp/root /rom 1
486 }
487
488 [ -z "$IPKG_INSTROOT" -a -f /lib/config/uci.sh ] && . /lib/config/uci.sh