modules/admin-core/root/sbin/luci-flash

   1 #!/bin/sh
   2 . /etc/functions.sh
   3
   4 trap '' INT
   5 trap '' TERM
   6 trap '' PIPE
   7
   8 # initialize defaults
   9 RAMFS_COPY_BIN="/usr/bin/awk"       # extra programs for temporary ramfs root
  10 RAMFS_COPY_DATA=""      # extra data files
  11 export KEEP_PATTERN=""
  12 export VERBOSE=1
  13
  14 # parse options
  15 while [ -n "$1" ]; do
  16         case "$1" in
  17                 -k)
  18                                 shift
  19                                 export SAVE_CONFIG=1
  20                                 export KEEP_PATTERN="$1"
  21                 ;;
  22                 -*)
  23                         echo "Invalid option: $1"
  24                         exit 1
  25                 ;;
  26                 *) break;;
  27         esac
  28         shift;
  29 done
  30
  31 export CONFFILES=/tmp/sysupgrade.conffiles
  32 export CONF_TAR=/tmp/sysupgrade.tgz
  33
  34 [ -f $CONFFILES ] && rm $CONFFILES
  35 [ -f $CONF_TAR ]  && rm $CONF_TAR
  36
  37 export ARGV="$*"
  38 export ARGC="$#"
  39
  40 [ -z "$ARGV" ] && {
  41         cat <<EOF
  42 Usage: $0 [options] <image file or URL>
  43
  44 Options:
  45         -k      <"file 1, file 2, ..."> Files to be kept
  46 EOF
  47         exit 1
  48 }
  49
  50 add_pattern_conffiles() {
  51         local file="$1"
  52         find $KEEP_PATTERN >> "$file" 2>/dev/null
  53         return 0
  54 }
  55
  56 # hooks
  57 sysupgrade_image_check="platform_check_image"
  58 sysupgrade_init_conffiles=""
  59
  60 [ -n "$KEEP_PATTERN" ] && append sysupgrade_init_conffiles "add_pattern_conffiles"
  61
  62 include /lib/upgrade
  63
  64 do_save_conffiles() {
  65         [ -z "$(rootfs_type)" ] && {
  66                 echo "Cannot save config while running from ramdisk."
  67                 exit 3
  68                 return 0
  69         }
  70         run_hooks "$CONFFILES" $sysupgrade_init_conffiles
  71
  72         v "Saving config files..."
  73         [ "$VERBOSE" -gt 1 ] && TAR_V="v" || TAR_V=""
  74         tar c${TAR_V}zf "$CONF_TAR" -T "$CONFFILES" 2>/dev/null
  75 }
  76
  77 type platform_check_image >/dev/null 2>/dev/null || {
  78         echo "Firmware upgrade is not implemented for this platform."
  79         exit 1
  80 }
  81
  82 for check in $sysupgrade_image_check; do
  83         ( eval "$check \"\$ARGV\"" ) || {
  84                 echo "Image check '$check' failed."
  85                 exit 2
  86         }
  87 done
  88
  89 [ -n "$sysupgrade_init_conffiles" ] && do_save_conffiles
  90 run_hooks "" $sysupgrade_pre_upgrade
  91
  92 if [ -n "$(rootfs_type)" ]; then
  93         v "Switching to ramdisk..."
  94         run_ramfs '. /etc/functions.sh; include /lib/upgrade; do_upgrade'
  95 else
  96         do_upgrade
  97 fi