base-files: fix failsafe via button press
[openwrt.git] / package / base-files / files / lib / preinit / 30_failsafe_wait
1 #!/bin/sh
2 # Copyright (C) 2006-2010 OpenWrt.org
3 # Copyright (C) 2010 Vertical Communications
4
5 fs_wait_for_key () {
6     local timeout=$3
7     local timer
8     local do_failsafe
9     local keypress_true="$(mktemp)"
10     local keypress_wait="$(mktemp)"
11     local keypress_sec="$(mktemp)"
12     if [ -z "$keypress_wait" ]; then
13         keypress_wait=/tmp/.keypress_wait
14         touch $keypress_wait
15     fi
16     if [ -z "$keypress_true" ]; then
17         keypress_true=/tmp/.keypress_true
18         touch $keypress_true
19     fi
20     if [ -z "$keypress_sec" ]; then
21         keypress_sec=/tmp/.keypress_sec
22         touch $keypress_sec
23     fi
24
25     trap "echo 'true' >$keypress_true; lock -u $keypress_wait ; rm -f $keypress_wait" INT
26     trap "echo 'true' >$keypress_true; lock -u $keypress_wait ; rm -f $keypress_wait" USR1
27
28     [ -n "$timeout" ] || timeout=1
29     [ $timeout -ge 1 ] || timeout=1
30     timer=$timeout
31     lock $keypress_wait
32     {
33         while [ $timer -gt 0 ]; do
34             echo "$timer" >$keypress_sec
35             timer=$(($timer - 1))
36             sleep 1
37         done
38         lock -u $keypress_wait
39         rm -f $keypress_wait
40     } &
41
42     echo "Press the [$1] key and hit [enter] $2"
43     # if we're on the console we wait for input
44     { 
45         while [ -r $keypress_wait ]; do
46             timer="$(cat $keypress_sec)"
47
48             [ -n "$timer" ] || timer=1
49             timer="${timer%%\ *}"
50             [ $timer -ge 1 ] || timer=1
51             do_failsafe=""
52             {
53                 read -t "$timer" do_failsafe
54                 if [ "$do_failsafe" = "$1" ]; then
55                     echo "true" >$keypress_true
56                     lock -u $keypress_wait
57                     rm -f $keypress_wait
58                 fi
59             }
60         done
61     }
62     lock -w $keypress_wait
63
64     keypressed=1
65     [ "$(cat $keypress_true)" = "true" ] && keypressed=0
66     
67     rm -f $keypress_true
68     rm -f $keypress_wait
69     rm -f $keypress_sec
70     
71     return $keypressed
72 }
73
74 failsafe_wait() {
75     FAILSAFE=
76     grep -q 'failsafe=' /proc/cmdline && FAILSAFE=true && export FAILSAFE
77     if [ "$FAILSAFE" != "true" ]; then
78       pi_failsafe_net_message=true
79       preinit_net_echo "Please press button now to enter failsafe"
80       pi_failsafe_net_message=false
81       fs_wait_for_key f 'to enter failsafe mode' $fs_failsafe_wait_timeout && FAILSAFE=true && export FAILSAFE
82       [ -f "/tmp/failsafe_button" ] && FAILSAFE=true && export FAILSAFE && rm /tmp/failsafe_button
83     fi
84 }
85
86 boot_hook_add preinit_main failsafe_wait
87