base-files: add mtd_get_mac_{ascii,binary} helpers
[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 mtd_get_mac_ascii()
244 {
245         local mtdname="$1"
246         local key="$2"
247         local part
248         local mac_dirty
249
250         . /lib/functions.sh
251
252         part=$(find_mtd_part "$mtdname")
253         if [ -z "$part" ]; then
254                 echo "mtd_get_mac_ascii: partition $mtdname not found!" >&2
255                 return
256         fi
257
258         mac_dirty=$(strings "$part" | sed -n 's/'"$key"'=//p')
259         # "canonicalize" mac
260         printf "%02x:%02x:%02x:%02x:%02x:%02x" 0x${mac_dirty//:/ 0x}
261 }
262
263 mtd_get_mac_binary() {
264         local mtdname="$1"
265         local offset="$2"
266         local part
267
268         part=$(find_mtd_part "$mtdname")
269         if [ -z "$part" ]; then
270                 echo "mtd_get_mac_binary: partition $mtdname not found!" >&2
271                 return
272         fi
273
274         dd bs=1 skip=$offset count=6 if=$part 2>/dev/null | hexdump -v -n 6 -e '5/1 "%02x:" 1/1 "%02x"'
275 }
276
277 strtok() { # <string> { <variable> [<separator>] ... }
278         local tmp
279         local val="$1"
280         local count=0
281
282         shift
283
284         while [ $# -gt 1 ]; do
285                 tmp="${val%%$2*}"
286
287                 [ "$tmp" = "$val" ] && break
288
289                 val="${val#$tmp$2}"
290
291                 export ${NO_EXPORT:+-n} "$1=$tmp"; count=$((count+1))
292                 shift 2
293         done
294
295         if [ $# -gt 0 -a -n "$val" ]; then
296                 export ${NO_EXPORT:+-n} "$1=$val"; count=$((count+1))
297         fi
298
299         return $count
300 }
301
302
303 jffs2_mark_erase() {
304         local part="$(find_mtd_part "$1")"
305         [ -z "$part" ] && {
306                 echo Partition not found.
307                 return 1
308         }
309         echo -e "\xde\xad\xc0\xde" | mtd -qq write - "$1"
310 }
311
312 uci_apply_defaults() {
313         cd /etc/uci-defaults || return 0
314         files="$(ls)"
315         [ -z "$files" ] && return 0
316         mkdir -p /tmp/.uci
317         for file in $files; do
318                 ( . "./$(basename $file)" ) && rm -f "$file"
319         done
320         uci commit
321 }
322
323 group_add() {
324         local name="$1"
325         local gid="$2"
326         local rc
327         [ -f "${IPKG_INSTROOT}/etc/group" ] || return 1
328         [ -n "$IPKG_INSTROOT" ] || lock /var/lock/group
329         echo "${name}:x:${gid}:" >> ${IPKG_INSTROOT}/etc/group
330         rc=$?
331         [ -n "$IPKG_INSTROOT" ] || lock -u /var/lock/group
332         return $rc
333 }
334
335 group_exists() {
336         grep -qs "^${1}:" ${IPKG_INSTROOT}/etc/group
337 }
338
339 user_add() {
340         local name="${1}"
341         local uid="${2}"
342         local gid="${3:-$2}"
343         local desc="${4:-$1}"
344         local home="${5:-/var/run/$1}"
345         local shell="${6:-/bin/false}"
346         local rc
347         [ -f "${IPKG_INSTROOT}/etc/passwd" ] || return 1
348         [ -n "$IPKG_INSTROOT" ] || lock /var/lock/passwd
349         echo "${name}:x:${uid}:${gid}:${desc}:${home}:${shell}" >> ${IPKG_INSTROOT}/etc/passwd
350         echo "${name}:x:0:0:99999:7:::" >> ${IPKG_INSTROOT}/etc/shadow
351         rc=$?
352         [ -n "$IPKG_INSTROOT" ] || lock -u /var/lock/passwd
353         return $rc
354 }
355
356 user_exists() {
357         grep -qs "^${1}:" ${IPKG_INSTROOT}/etc/passwd
358 }
359
360
361 pi_include() {
362         if [ -f "/tmp/overlay/$1" ]; then
363                 . "/tmp/overlay/$1"
364         elif [ -f "$1" ]; then
365                 . "$1"
366         elif [ -d "/tmp/overlay/$1" ]; then
367                 if [ -n "$(ls /tmp/overlay/$1/*.sh 2>/dev/null)" ]; then
368                         for src_script in /tmp/overlay/$1/*.sh; do
369                                 . "$src_script"
370                         done
371                 fi
372         elif [ -d "$1" ]; then
373                 if [ -n "$(ls $1/*.sh 2>/dev/null)" ]; then
374                         for src_script in $1/*.sh; do
375                                 . "$src_script"
376                         done
377                 fi
378         else
379                 echo "WARNING: $1 not found"
380                 return 1
381         fi
382         return 0
383 }
384
385 #!/bin/sh
386 # Copyright (C) 2006-2010 OpenWrt.org
387 # Copyright (C) 2010 Vertical Communications
388
389 boot_hook_splice_start() {
390         export -n PI_HOOK_SPLICE=1
391 }
392
393 boot_hook_splice_finish() {
394         local hook
395         for hook in $PI_STACK_LIST; do
396                 local v; eval "v=\${${hook}_splice:+\$${hook}_splice }$hook"
397                 export -n "${hook}=${v% }"
398                 export -n "${hook}_splice="
399         done
400         export -n PI_HOOK_SPLICE=
401 }
402
403 boot_hook_init() {
404         local hook="${1}_hook"
405         export -n "PI_STACK_LIST=${PI_STACK_LIST:+$PI_STACK_LIST }$hook"
406         export -n "$hook="
407 }
408
409 boot_hook_add() {
410         local hook="${1}_hook${PI_HOOK_SPLICE:+_splice}"
411         local func="${2}"
412
413         [ -n "$func" ] && {
414                 local v; eval "v=\$$hook"
415                 export -n "$hook=${v:+$v }$func"
416         }
417 }
418
419 boot_hook_shift() {
420         local hook="${1}_hook"
421         local rvar="${2}"
422
423         local v; eval "v=\$$hook"
424         [ -n "$v" ] && {
425                 local first="${v%% *}"
426
427                 [ "$v" != "${v#* }" ] && \
428                         export -n "$hook=${v#* }" || \
429                         export -n "$hook="
430
431                 export -n "$rvar=$first"
432                 return 0
433         }
434
435         return 1
436 }
437
438 boot_run_hook() {
439         local hook="$1"
440         local func
441
442         while boot_hook_shift "$hook" func; do
443                 local ran; eval "ran=\$PI_RAN_$func"
444                 [ -n "$ran" ] || {
445                         export -n "PI_RAN_$func=1"
446                         $func "$1" "$2"
447                 }
448         done
449 }
450
451 jffs2_ready () {
452         mtdpart="$(find_mtd_part rootfs_data)"
453         [ -z "$mtdpart" ] && return 1
454         magic=$(hexdump $mtdpart -n 4 -e '4/1 "%02x"')
455         [ "$magic" != "deadc0de" ]
456 }
457
458 dupe() { # <new_root> <old_root>
459         cd $1
460         echo -n "creating directories... "
461         {
462                 cd $2
463                 find . -xdev -type d
464                 echo "./dev ./overlay ./mnt ./proc ./tmp"
465                 # xdev skips mounted directories
466                 cd $1
467         } | xargs mkdir -p
468         echo "done"
469
470         echo -n "setting up symlinks... "
471         for file in $(cd $2; find . -xdev -type f;); do
472                 case "$file" in
473                 ./rom/note) ;; #nothing
474                 ./etc/config*|\
475                 ./usr/lib/opkg/info/*) cp -af $2/$file $file;;
476                 *) ln -sf /rom/${file#./*} $file;;
477                 esac
478         done
479         for file in $(cd $2; find . -xdev -type l;); do
480                 cp -af $2/${file#./*} $file
481         done
482         echo "done"
483 }
484
485 pivot() { # <new_root> <old_root>
486         mount -o noatime,move /proc $1/proc && \
487         pivot_root $1 $1$2 && {
488                 mount -o noatime,move $2/dev /dev
489                 mount -o noatime,move $2/tmp /tmp
490                 mount -o noatime,move $2/sys /sys 2>&-
491                 mount -o noatime,move $2/overlay /overlay 2>&-
492                 return 0
493         }
494 }
495
496 fopivot() { # <rw_root> <ro_root> <dupe?>
497         root=$1
498         {
499                 if grep -q overlay /proc/filesystems; then
500                         mount -o noatime,lowerdir=/,upperdir=$1 -t overlayfs "overlayfs:$1" /mnt && root=/mnt
501                 elif grep -q mini_fo /proc/filesystems; then
502                         mount -t mini_fo -o noatime,base=/,sto=$1 "mini_fo:$1" /mnt 2>&- && root=/mnt
503                 else
504                         mount --bind -o noatime / /mnt
505                         mount --bind -o noatime,union "$1" /mnt && root=/mnt
506                 fi
507         } || {
508                 [ "$3" = "1" ] && {
509                 mount | grep "on $1 type" 2>&- 1>&- || mount -o noatime,bind $1 $1
510                 dupe $1 $rom
511                 }
512         }
513         pivot $root $2
514 }
515
516 ramoverlay() {
517         mkdir -p /tmp/root
518         mount -t tmpfs -o noatime,mode=0755 root /tmp/root
519         fopivot /tmp/root /rom 1
520 }
521
522 [ -z "$IPKG_INSTROOT" -a -f /lib/config/uci.sh ] && . /lib/config/uci.sh