scripts/live-helpers

   1 # live-boot helper functions, used by live-boot on boot and by live-snapshot
   2
   3 if [ ! -x "/bin/fstype" ]
   4 then
   5         # klibc not in path -> not in initramfs
   6         export PATH="${PATH}:/usr/lib/klibc/bin"
   7 fi
   8
   9 # handle upgrade path from old udev (using udevinfo) to
  10 # recent versions of udev (using udevadm info)
  11 if [ -x /sbin/udevadm ]
  12 then
  13         udevinfo='/sbin/udevadm info'
  14 else
  15         udevinfo='udevinfo'
  16 fi
  17
  18 sys2dev ()
  19 {
  20         sysdev=${1#/sys}
  21         echo "/dev/$($udevinfo -q name -p ${sysdev} 2>/dev/null|| echo ${sysdev##*/})"
  22 }
  23
  24 subdevices ()
  25 {
  26         sysblock=${1}
  27         r=""
  28
  29         for dev in "${sysblock}"/* "${sysblock}"
  30         do
  31                 if [ -e "${dev}/dev" ]
  32                 then
  33                         r="${r} ${dev}"
  34                 fi
  35         done
  36
  37         echo ${r}
  38 }
  39
  40 storage_devices()
  41 {
  42         black_listed_devices="${1}"
  43         white_listed_devices="${2}"
  44
  45         for sysblock in $(echo /sys/block/* | tr ' ' '\n' | grep -vE "loop|ram|fd")
  46         do
  47                 fulldevname=$(sys2dev "${sysblock}")
  48
  49                 if echo "${black_listed_devices}" | grep -qw "${fulldevname}" || \
  50                         [ -n "${white_listed_devices}" ] && \
  51                         echo "${white_listed_devices}" | grep -vqw "${fulldevname}"
  52                 then
  53                         # skip this device entirely
  54                         continue
  55                 fi
  56
  57                 for dev in $(subdevices "${sysblock}")
  58                 do
  59                         devname=$(sys2dev "${dev}")
  60
  61                         if echo "${black_listed_devices}" | grep -qw "${devname}"
  62                         then
  63                                 # skip this subdevice
  64                                 continue
  65                         else
  66                                 echo "${devname}"
  67                         fi
  68                 done
  69         done
  70 }
  71
  72 is_supported_fs ()
  73 {
  74         fstype="${1}"
  75
  76         # Validate input first
  77         if [ -z "${fstype}" ]
  78         then
  79                 return 1
  80         fi
  81
  82         # Try to look if it is already supported by the kernel
  83         if grep -q ${fstype} /proc/filesystems
  84         then
  85                 return 0
  86         else
  87                 # Then try to add support for it the gentle way using the initramfs capabilities
  88                 modprobe ${fstype}
  89                 if grep -q ${fstype} /proc/filesystems
  90                 then
  91                         return 0
  92                 # Then try the hard way if /root is already reachable
  93                 else
  94                         kmodule="/root/lib/modules/`uname -r`/${fstype}/${fstype}.ko"
  95                         if [ -e "${kmodule}" ]
  96                         then
  97                                 insmod "${kmodule}"
  98                                 if grep -q ${fstype} /proc/filesystems
  99                                 then
 100                                         return 0
 101                                 fi
 102                         fi
 103                 fi
 104         fi
 105
 106         return 1
 107 }
 108
 109 get_fstype ()
 110 {
 111         /sbin/blkid -s TYPE -o value $1 2>/dev/null
 112 }
 113
 114 where_is_mounted ()
 115 {
 116         device=${1}
 117
 118         if grep -q "^${device} " /proc/mounts
 119         then
 120                 # return the first found
 121                 grep -m1 "^${device} " /proc/mounts | cut -f2 -d ' '
 122         fi
 123 }
 124
 125 lastline ()
 126 {
 127         while read lines
 128         do
 129                 line=${lines}
 130         done
 131
 132         echo "${line}"
 133 }
 134
 135 base_path ()
 136 {
 137         testpath="${1}"
 138         mounts="$(awk '{print $2}' /proc/mounts)"
 139         testpath="$(busybox realpath ${testpath})"
 140
 141         while true
 142         do
 143                 if echo "${mounts}" | grep -qs "^${testpath}"
 144                 then
 145                         set -- $(echo "${mounts}" | grep "^${testpath}" | lastline)
 146                         echo ${1}
 147                         break
 148                 else
 149                         testpath=$(dirname $testpath)
 150                 fi
 151         done
 152 }
 153
 154 fs_size ()
 155 {
 156         # Returns used/free fs kbytes + 5% more
 157         # You could pass a block device as ${1} or the mount point as ${2}
 158
 159         dev="${1}"
 160         mountp="${2}"
 161         used="${3}"
 162
 163         if [ -z "${mountp}" ]
 164         then
 165                 mountp="$(where_is_mounted ${dev})"
 166
 167                 if [ -z "${mountp}" ]
 168                 then
 169                         mountp="/mnt/tmp_fs_size"
 170
 171                         mkdir -p "${mountp}"
 172                         mount -t $(get_fstype "${dev}") -o ro "${dev}" "${mountp}" || log_warning_msg "cannot mount -t $(get_fstype ${dev}) -o ro ${dev} ${mountp}"
 173
 174                         doumount=1
 175                 fi
 176         fi
 177
 178         if [ "${used}" = "used" ]
 179         then
 180                 size=$(du -ks ${mountp} | cut -f1)
 181                 size=$(expr ${size} + ${size} / 20 ) # FIXME: 5% more to be sure
 182         else
 183                 # free space
 184                 size="$(df -k | grep -s ${mountp} | awk '{print $4}')"
 185         fi
 186
 187         if [ -n "${doumount}" ]
 188         then
 189                 umount "${mountp}" || log_warning_msg "cannot umount ${mountp}"
 190                 rmdir "${mountp}"
 191         fi
 192
 193         echo "${size}"
 194 }
 195
 196 load_keymap ()
 197 {
 198         # Load custom keymap
 199         if [ -x /bin/loadkeys -a -r /etc/boottime.kmap.gz ]
 200         then
 201                 loadkeys /etc/boottime.kmap.gz
 202         fi
 203 }
 204
 205 setup_loop ()
 206 {
 207         local fspath=${1}
 208         local module=${2}
 209         local pattern=${3}
 210         local offset=${4}
 211         local encryption=${5}
 212         local readonly=${6}
 213
 214         # the output of setup_loop is evaluated in other functions,
 215         # modprobe leaks kernel options like "libata.dma=0"
 216         # as "options libata dma=0" on stdout, causing serious
 217         # problems therefor, so instead always avoid output to stdout
 218         modprobe -q -b "${module}" 1>/dev/null
 219
 220         udevadm settle
 221
 222         for loopdev in ${pattern}
 223         do
 224                 if [ "$(cat ${loopdev}/size)" -eq 0 ]
 225                 then
 226                         dev=$(sys2dev "${loopdev}")
 227                         options=''
 228
 229                         if [ -n "${readonly}" ]
 230                         then
 231                                 if losetup --help 2>&1 | grep -q -- "-r\b"
 232                                 then
 233                                         options="${options} -r"
 234                                 fi
 235                         fi
 236
 237                         if [ 0 -lt "${offset}" ]
 238                         then
 239                                 options="${options} -o ${offset}"
 240                         fi
 241
 242                         if [ -z "${encryption}" ]
 243                         then
 244                                 losetup ${options} "${dev}" "${fspath}"
 245                         else
 246                                 # Loop AES encryption
 247                                 while true
 248                                 do
 249                                         load_keymap
 250
 251                                         echo -n "Enter passphrase for root filesystem: " >&6
 252                                         read -s passphrase
 253                                         echo "${passphrase}" > /tmp/passphrase
 254                                         unset passphrase
 255                                         exec 9</tmp/passphrase
 256                                         /sbin/losetup ${options} -e "${encryption}" -p 9 "${dev}" "${fspath}"
 257                                         error=${?}
 258                                         exec 9<&-
 259                                         rm -f /tmp/passphrase
 260
 261                                         if [ 0 -eq ${error} ]
 262                                         then
 263                                                 unset error
 264                                                 break
 265                                         fi
 266
 267                                         echo
 268                                         echo -n "There was an error decrypting the root filesystem ... Retry? [Y/n] " >&6
 269                                         read answer
 270
 271                                         if [ "$(echo "${answer}" | cut -b1 | tr A-Z a-z)" = "n" ]
 272                                         then
 273                                                 unset answer
 274                                                 break
 275                                         fi
 276                                 done
 277                         fi
 278
 279                         echo "${dev}"
 280                         return 0
 281                 fi
 282         done
 283
 284         panic "No loop devices available"
 285 }
 286
 287 try_mount ()
 288 {
 289         dev="${1}"
 290         mountp="${2}"
 291         opts="${3}"
 292         fstype="${4}"
 293
 294         old_mountp="$(where_is_mounted ${dev})"
 295
 296         if [ -n "${old_mountp}" ]
 297         then
 298                 if [ "${opts}" != "ro" ]
 299                 then
 300                         mount -o remount,"${opts}" "${dev}" "${old_mountp}" || panic "Remounting ${dev} ${opts} on ${old_mountp} failed"
 301                 fi
 302
 303                 mount -o bind "${old_mountp}" "${mountp}" || panic "Cannot bind-mount ${old_mountp} on ${mountp}"
 304         else
 305                 if [ -z "${fstype}" ]
 306                 then
 307                         fstype=$(get_fstype "${dev}")
 308                 fi
 309                 mount -t "${fstype}" -o "${opts}" "${dev}" "${mountp}" || \
 310                 ( echo "SKIPPING: Cannot mount ${dev} on ${mountp}, fstype=${fstype}, options=${opts}" > live-boot.log && return 0 )
 311         fi
 312 }
 313
 314 find_cow_device ()
 315 {
 316         # Returns a device containing a partition labeled "${pers_label}" or containing a file named the same way
 317         #  in the latter case the partition containing the file is left mounted
 318         #  if is not in black_listed_devices.
 319         #  Additionally, if the white_listed_devices list is non-empty, the
 320         #  parent block device of the returned device must be part of this list.
 321         pers_label="${1}"
 322         cow_backing="/${pers_label}-backing"
 323         black_listed_devices="${2}"
 324         white_listed_devices="${3}"
 325
 326         if [ -z "${PERSISTENT_PATH}" ]
 327         then
 328                 pers_fpath=${cow_backing}/${pers_label}
 329         else
 330                 pers_fpath=${cow_backing}/${PERSISTENT_PATH}/${pers_label}
 331         fi
 332
 333         for dev in $(storage_devices "${black_listed_devices}" "${white_listed_devices}")
 334         do
 335                 # Checking for a luks device
 336                 if [ "${PERSISTENT_ENCRYPTION}" = "luks" ] && [ -e /sbin/cryptsetup ]
 337                 then
 338                         if ! modprobe dm-crypt
 339                         then
 340                                 log_warning_msg "Unable to load module dm-crypt"
 341                                 continue
 342                         fi
 343
 344                         if ! /sbin/cryptsetup isLuks ${dev}
 345                         then
 346                                 # we only look for encrypted subdevices
 347                                 continue
 348                         fi
 349
 350                         while true
 351                         do
 352                                 load_keymap
 353
 354                                 /lib/cryptsetup/askpass "Enter passphrase for ${pers_label} on ${dev}: " | /sbin/cryptsetup -T 1 luksOpen ${dev} $(basename ${dev}) --key-file=-
 355                                 error=${?}
 356
 357                                 dev="/dev/mapper/$(basename ${dev})"
 358
 359                                 if [ 0 -eq ${error} ]
 360                                 then
 361                                         unset error
 362                                         break
 363                                 fi
 364
 365                                 echo >&6
 366                                 echo -n "There was an error decrypting ${dev} ... Retry? [Y/n] " >&6
 367                                 read answer
 368
 369                                 if [ "$(echo "${answer}" | cut -b1 | tr A-Z a-z)" = "n" ]
 370                                 then
 371                                         unset answer
 372                                         # skip to next subdevice
 373                                         continue 2
 374                                 fi
 375                         done
 376                 fi
 377
 378                 if echo ${PERSISTENT_STORAGE} | grep -qw filesystem && [ "$(/sbin/blkid -s LABEL -o value $dev 2>/dev/null)" = "${pers_label}" ]
 379                 then
 380                         echo "${dev}"
 381                         return 0
 382                 fi
 383
 384                 if ! echo ${PERSISTENT_STORAGE} | grep -qw file
 385                 then
 386                         # do not mount the device to find for image files
 387                         # just skip this
 388                         continue
 389                 fi
 390
 391                 devfstype="$(get_fstype ${dev})"
 392
 393                 if is_supported_fs ${devfstype}
 394                 then
 395                         mkdir -p "${cow_backing}"
 396                         if try_mount "${dev}" "${cow_backing}" "rw"
 397                         then
 398                                 if [ -f "${pers_fpath}" ]
 399                                 then
 400                                         echo $(setup_loop "${pers_fpath}" "loop" "/sys/block/loop*")
 401                                         return 0
 402                                 else
 403                                         umount ${cow_backing} > /dev/null 2>&1 || true
 404                                 fi
 405                         fi
 406                 fi
 407         done
 408         return 1
 409 }
 410
 411 find_files ()
 412 {
 413         # return the a string composed by device name, mountpoint an the first of ${filenames} found on a supported partition
 414         #  if is not in black_listed_devices.
 415         #  Additionally, if the white_listed_devices list is non-empty, the
 416         #  parent block device of the returned device must be part of this list.
 417         # FIXME: merge with above function
 418
 419         filenames="${1}"
 420         snap_backing="/snap-backing"
 421         black_listed_devices="${2}"
 422         white_listed_devices="${3}"
 423
 424         for dev in $(storage_devices "${black_listed_devices}" "${white_listed_devices}")
 425         do
 426                 devfstype="$(get_fstype ${dev})"
 427
 428                 if is_supported_fs ${devfstype}
 429                 then
 430                         mkdir -p "${snap_backing}"
 431
 432                         if try_mount "${dev}" "${snap_backing}" "ro" "${devfstype}"
 433                         then
 434                                 for filename in ${filenames}
 435                                 do
 436                                         if [ -f "${snap_backing}/${filename}" ]
 437                                         then
 438                                                 echo "${dev} ${snap_backing} ${filename}"
 439                                                 umount ${snap_backing}
 440                                                 return 0
 441                                         fi
 442                                 done
 443                         fi
 444                         umount ${snap_backing}
 445                 fi
 446         done
 447 }
 448
 449 get_mac ()
 450 {
 451         mac=""
 452
 453         for adaptor in /sys/class/net/*
 454         do
 455                 status="$(cat ${adaptor}/iflink)"
 456
 457                 if [ "${status}" -eq 2 ]
 458                 then
 459                         mac="$(cat ${adaptor}/address)"
 460                         mac="$(echo ${mac} | sed 's/:/-/g' | tr '[a-z]' '[A-Z]')"
 461                 fi
 462         done
 463
 464         echo ${mac}
 465 }
 466
 467 is_luks()
 468 {
 469     devname="${1}"
 470     if [ -x /sbin/cryptsetup ]
 471     then
 472         /sbin/cryptsetup isLuks "${devname}" 2>/dev/null || ret=${?}
 473         return ${ret}
 474     else
 475         return 1
 476     fi
 477
 478 }
 479
 480 removable_dev ()
 481 {
 482         output_format="${1}"
 483         want_usb="${2}"
 484         ret=
 485
 486         for sysblock in $(echo /sys/block/* | tr ' ' '\n' | grep -vE "/(loop|ram|dm-|fd)")
 487         do
 488                 dev_ok=
 489                 if [ "$(cat ${sysblock}/removable)" = "1" ]
 490                 then
 491                         if [ -z "${want_usb}" ]
 492                         then
 493                                 dev_ok="yes"
 494                         else
 495                                 if readlink ${sysblock} | grep -q usb
 496                                 then
 497                                         dev_ok="yes"
 498                                 fi
 499                         fi
 500                 fi
 501
 502                 if [ "${dev_ok}" = "yes" ]
 503                 then
 504                         case "${output_format}" in
 505                                 sys)
 506                                         ret="${ret} ${sysblock}"
 507                                         ;;
 508                                 *)
 509                                         devname=$(sys2dev "${sysblock}")
 510                                         ret="${ret} ${devname}"
 511                                         ;;
 512                         esac
 513                 fi
 514         done
 515
 516         echo "${ret}"
 517 }
 518
 519 removable_usb_dev ()
 520 {
 521         output_format="${1}"
 522
 523         removable_dev "${output_format}" "want_usb"
 524 }
 525
 526 non_removable_dev ()
 527 {
 528         output_format="${1}"
 529         ret=
 530
 531         for sysblock in $(echo /sys/block/* | tr ' ' '\n' | grep -vE "/(loop|ram|dm-|fd)")
 532         do
 533                 if [ "$(cat ${sysblock}/removable)" = "0" ]
 534                 then
 535                         case "${output_format}" in
 536                                 sys)
 537                                         ret="${ret} ${sysblock}"
 538                                         ;;
 539                                 *)
 540                                         devname=$(sys2dev "${sysblock}")
 541                                         ret="${ret} ${devname}"
 542                                         ;;
 543                         esac
 544                 fi
 545         done
 546
 547         echo "${ret}"
 548 }