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