debian/live-initramfs.init

   1 #!/bin/sh
   2
   3 ### BEGIN INIT INFO
   4 # Provides:             live-initramfs
   5 # Required-Start:       $syslog
   6 # Required-Stop:        $syslog
   7 # Should-Start:         $local_fs
   8 # Should-Stop:          $local_fs
   9 # Default-Start:        1 2 3 4 5
  10 # Default-Stop:         0 6
  11 # Short-Description:    live-initramfs init script
  12 # Description:          Resyncs snapshots, evantually caches files in order to
  13 #                       let remove the media.
  14 ### END INIT INFO
  15
  16 # Authors: Tollef Fog Heen <tfheen@canonical.com>
  17 #          Marco Amadori <marco.amadori@gmail.com>
  18
  19 PATH=/usr/sbin:/usr/bin:/sbin:/bin
  20 NAME=live-initramfs
  21 SCRIPTNAME=/etc/init.d/${NAME}
  22 DO_SNAPSHOT=/sbin/live-snapshot
  23
  24 # Exit if system was not booted by live-initramfs
  25 grep -qs boot=live /proc/cmdline || exit 0
  26
  27 # Exit if the system was booted from an ISO image rather than a physical CD
  28 grep -qs find_iso= /proc/cmdline && exit 0
  29
  30 # Read configuration variable file if it is present
  31 [ -r /etc/live.conf ] && . /etc/live.conf
  32
  33 # Load the VERBOSE setting and other rcS variables
  34 [ -f /etc/default/rcS ] && . /etc/default/rcS
  35
  36 # Define LSB log_* functions.
  37 # Depend on lsb-base (>= 3.0-6) to ensure that this file is present.
  38 . /lib/lsb/init-functions
  39
  40 # Try to cache everything we're likely to need after ejecting.  This
  41 # is fragile and simple-minded, but our options are limited.
  42 cache_path()
  43 {
  44         path="${1}"
  45
  46         if [ -d "${path}" ]
  47         then
  48                 find "${path}" -type f | xargs cat > /dev/null 2>&1
  49         elif [ -f "${path}" ]
  50         then
  51                 if [ -x "${path}" ]
  52                 then
  53                         if file "${path}" | grep -q 'dynamically linked'
  54                         then
  55                                 for lib in $(ldd "${path}" | awk '{ print $3 }')
  56                                 do
  57                                         cache_path "${lib}"
  58                                 done
  59                         fi
  60                 fi
  61
  62                 cat "${path}" >/dev/null 2>&1
  63         fi
  64 }
  65
  66 get_boot_device()
  67 {
  68         # search in /proc/mounts for the device that is mounted at /live/image
  69         while read DEVICE MOUNT REST
  70         do
  71                 if [ "${MOUNT}" = "/live/image" ]
  72                 then
  73                         echo "${DEVICE}"
  74                         exit 0
  75                 fi
  76         done < /proc/mounts
  77 }
  78
  79 device_is_USB_flash_drive()
  80 {
  81         # remove leading "/dev/" and all trailing numbers from input
  82         DEVICE=$(expr substr ${1} 6 3)
  83
  84         # check that device starts with "sd"
  85         [ "$(expr substr ${DEVICE} 1 2)" != "sd" ] && return 1
  86
  87         # check that the device is an USB device
  88         if readlink /sys/block/${DEVICE} | grep -q usb ||
  89            readlink /sys/block/${DEVICE}/device | grep -q usb # linux < 2.6.29
  90         then
  91                 return 0
  92         fi
  93
  94         return 1
  95 }
  96
  97 do_stop ()
  98 {
  99         if ! grep -qs nopersistent /proc/cmdline && grep -qs persistent /proc/cmdline
 100         then
 101                 if [ ! -z "${ROOTSNAP}" ]
 102                 then
 103                         ${DO_SNAPSHOT} --resync-string="${ROOTSNAP}"
 104                 fi
 105
 106                 if [ ! -z "${HOMESNAP}" ]
 107                 then
 108                         ${DO_SNAPSHOT} --resync-string="${HOMESNAP}"
 109                 fi
 110         fi
 111
 112         # check for netboot
 113         if [ ! -z "${NETBOOT}" ] || grep -qs netboot /proc/cmdline || grep -qsi root=/dev/nfs /proc/cmdline  || grep -qsi root=/dev/cifs /proc/cmdline
 114         then
 115                 return 0
 116         fi
 117
 118         prompt=1
 119         if grep -qs noprompt /proc/cmdline
 120         then
 121                 prompt=
 122         fi
 123
 124         for path in $(which halt) $(which reboot) /etc/rc?.d /etc/default $(which stty)
 125         do
 126                 cache_path "${path}"
 127         done
 128
 129         for x in $(cat /proc/cmdline)
 130         do
 131                 case ${x} in
 132                         quickreboot)
 133                                 QUICKREBOOT="Yes"
 134                                 ;;
 135                 esac
 136         done
 137
 138         if [ -z ${QUICKREBOOT} ]
 139         then
 140                 # in order to deal with the aufs unmount problem, we do a sync here.
 141                 sync
 142                 sleep 1
 143                 sync
 144
 145                 # TODO: i18n
 146                 BOOT_DEVICE="$(get_boot_device)"
 147
 148                 if device_is_USB_flash_drive ${BOOT_DEVICE}
 149                 then
 150                         # do NOT eject USB flash drives!
 151                         # otherwise rebooting with most USB flash drives
 152                         # failes because they actually remember the
 153                         # "ejected" state even after reboot
 154                         MESSAGE="Please remove the USB flash drive"
 155                 else
 156                         # ejecting is a very good idea here
 157                         MESSAGE="Please remove the disc, close the the tray (if any)"
 158
 159                         if [ -x /usr/bin/eject ]
 160                         then
 161                                 eject -p -m /live/image >/dev/null 2>&1
 162                         fi
 163
 164                 fi
 165
 166                 [ "$prompt" ] || return 0
 167
 168                 stty sane < /dev/console
 169
 170                 printf "\n\n${MESSAGE} and press ENTER to continue:" > /dev/console
 171
 172                 if [ -x /sbin/usplash_write ]
 173                 then
 174                         /sbin/usplash_write "TIMEOUT 86400"
 175                         /sbin/usplash_write "TEXT-URGENT ${MESSAGE}"
 176                         /sbin/usplash_write "TEXT-URGENT and press ENTER to continue"
 177                 fi
 178
 179                 read x < /dev/console
 180         fi
 181 }
 182
 183 case "${1}" in
 184         start|restart|reload|force-reload|status)
 185                 [ "${VERBOSE}" != no ] && log_end_msg 0
 186                 ;;
 187
 188         stop)
 189                 log_begin_msg "${NAME} is resyncing snapshots and caching reboot files..."
 190                 do_stop
 191
 192                 case "${?}" in
 193                         0|1)
 194                                 [ "${VERBOSE}" != no ] && log_end_msg 0
 195                                 ;;
 196
 197                         2)
 198                                 [ "${VERBOSE}" != no ] && log_end_msg 1
 199                                 ;;
 200                 esac
 201
 202                 mount -o remount,ro /live/cow
 203                 ;;
 204
 205         *)
 206                 log_success_msg "Usage: ${SCRIPTNAME} {start|stop|restart|force-reload}" >&2
 207                 exit 3
 208                 ;;
 209 esac