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 $1<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     trap - INT
  65     trap - USR1
  66
  67     keypressed=1
  68     [ "$(cat $keypress_true)" = "true" ] && keypressed=0
  69     rm -f $keypress_true
  70     rm -f $keypress_wait
  71     rm -f $keypress_sec
  72
  73     return $keypressed
  74 }
  75
  76 failsafe_wait() {
  77     FAILSAFE=
  78     pi_failsafe_net_message=true
  79     preinit_net_echo "Please press button now to enter failsafe"
  80     pi_failsafe_net_message=false
  81     echo -n "Press CTRL-C or "
  82     fs_wait_for_key f 'to enter failsafe mode' $fs_failsafe_wait_timeout && FAILSAFE=true && export FAILSAFE
  83 }
  84
  85 boot_hook_add preinit_main failsafe_wait
  86