usr.sbin/pc-sysinstall/backend/functions.sh

   1 #!/bin/sh
   2 #-
   3 # Copyright (c) 2010 iXsystems, Inc.  All rights reserved.
   4 #
   5 # Redistribution and use in source and binary forms, with or without
   6 # modification, are permitted provided that the following conditions
   7 # are met:
   8 # 1. Redistributions of source code must retain the above copyright
   9 #    notice, this list of conditions and the following disclaimer.
  10 # 2. Redistributions in binary form must reproduce the above copyright
  11 #    notice, this list of conditions and the following disclaimer in the
  12 #    documentation and/or other materials provided with the distribution.
  13 #
  14 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24 # SUCH DAMAGE.
  25 #
  26 # $FreeBSD$
  27
  28 # functions.sh
  29 # Library of functions which pc-sysinstall may call upon
  30
  31 # Function which displays the help-index file
  32 display_help()
  33 {
  34   if [ -e "${PROGDIR}/doc/help-index" ]
  35   then
  36     cat ${PROGDIR}/doc/help-index
  37   else
  38     echo "Error: ${PROGDIR}/doc/help-index not found"
  39     exit 1
  40   fi
  41 };
  42
  43 # Function which displays the help for a specified command
  44 display_command_help()
  45 {
  46   if [ -z "$1" ]
  47   then
  48     echo "Error: No command specified to display help for"
  49     exit 1
  50   fi
  51
  52   if [ -e "${PROGDIR}/doc/help-${1}" ]
  53   then
  54     cat ${PROGDIR}/doc/help-${1}
  55   else
  56     echo "Error: ${PROGDIR}/doc/help-${1} not found"
  57     exit 1
  58   fi
  59 };
  60
  61 # Function to convert bytes to megabytes
  62 convert_byte_to_megabyte()
  63 {
  64   if [ -z "${1}" ]
  65   then
  66     echo "Error: No bytes specified!"
  67     exit 1
  68   fi
  69
  70   expr -e ${1} / 1048576
  71 };
  72
  73 # Function to convert blocks to megabytes
  74 convert_blocks_to_megabyte()
  75 {
  76   if [ -z "${1}" ] ; then
  77     echo "Error: No blocks specified!"
  78     exit 1
  79   fi
  80
  81   expr -e ${1} / 2048
  82 };
  83
  84 # Takes $1 and strips the whitespace out of it, returns VAL
  85 strip_white_space()
  86 {
  87   if [ -z "${1}" ]
  88   then
  89     echo "Error: No value setup to strip whitespace from!"
  90
  91     exit 1
  92   fi
  93
  94   export VAL=`echo "$1" | tr -d ' '`
  95 };
  96
  97 # Displays an error message and exits with error 1
  98 exit_err()
  99 {
 100   # Echo the message for the users benefit
 101   echo "EXITERROR: $1"
 102
 103   # Save this error to the log file
 104   echo "EXITERROR: ${1}" >>$LOGOUT
 105
 106   # Check if we need to unmount any file-systems after this failure
 107   unmount_all_filesystems_failure
 108
 109   echo "For more details see log file: $LOGOUT"
 110
 111   exit 1
 112 };
 113
 114 # Run-command, don't halt if command exits with non-0
 115 rc_nohalt()
 116 {
 117   CMD="$1"
 118
 119   if [ -z "${CMD}" ]
 120   then
 121     exit_err "Error: missing argument in rc_nohalt()"
 122   fi
 123
 124   echo "Running: ${CMD}" >>${LOGOUT}
 125   ${CMD} >>${LOGOUT} 2>>${LOGOUT}
 126
 127 };
 128
 129 # Run-command, halt if command exits with non-0
 130 rc_halt()
 131 {
 132   CMD="$1"
 133
 134   if [ -z "${CMD}" ]
 135   then
 136     exit_err "Error: missing argument in rc_halt()"
 137   fi
 138
 139   echo "Running: ${CMD}" >>${LOGOUT}
 140   eval ${CMD} >>${LOGOUT} 2>>${LOGOUT}
 141   STATUS="$?"
 142   if [ "${STATUS}" != "0" ]
 143   then
 144     exit_err "Error ${STATUS}: ${CMD}"
 145   fi
 146 };
 147
 148 # Run-command w/echo to screen, halt if command exits with non-0
 149 rc_halt_echo()
 150 {
 151   CMD="$1"
 152
 153   if [ -z "${CMD}" ]
 154   then
 155     exit_err "Error: missing argument in rc_halt_echo()"
 156   fi
 157
 158   echo "Running: ${CMD}" >>${LOGOUT}
 159   ${CMD} 2>&1 | tee -a ${LOGOUT}
 160   STATUS="$?"
 161   if [ "$STATUS" != "0" ]
 162   then
 163     exit_err "Error ${STATUS}: $CMD"
 164   fi
 165
 166 };
 167
 168 # Run-command w/echo, don't halt if command exits with non-0
 169 rc_nohalt_echo()
 170 {
 171   CMD="$1"
 172
 173   if [ -z "${CMD}" ]
 174   then
 175     exit_err "Error: missing argument in rc_nohalt_echo()"
 176   fi
 177
 178   echo "Running: ${CMD}" >>${LOGOUT}
 179   ${CMD} 2>&1 | tee -a ${LOGOUT}
 180
 181 };
 182
 183 # Echo to the screen and to the log
 184 echo_log()
 185 {
 186   STR="$1"
 187
 188   if [ -z "${STR}" ]
 189   then
 190     exit_err "Error: missing argument in echo_log()"
 191   fi
 192
 193   echo "${STR}" | tee -a ${LOGOUT}
 194 };
 195
 196 # Make sure we have a numeric
 197 is_num()
 198 {
 199   expr $1 + 1 2>/dev/null
 200   return $?
 201 }
 202
 203 # Function which uses "fetch" to download a file, and display a progress report
 204 fetch_file()
 205 {
 206
 207   FETCHFILE="$1"
 208   FETCHOUTFILE="$2"
 209   EXITFAILED="$3"
 210
 211   SIZEFILE="${TMPDIR}/.fetchSize"
 212   EXITFILE="${TMPDIR}/.fetchExit"
 213
 214   rm ${SIZEFILE} 2>/dev/null >/dev/null
 215   rm ${FETCHOUTFILE} 2>/dev/null >/dev/null
 216
 217   fetch -s "${FETCHFILE}" >${SIZEFILE}
 218   SIZE="`cat ${SIZEFILE}`"
 219   SIZE="`expr ${SIZE} / 1024`"
 220   echo "FETCH: ${FETCHFILE}"
 221   echo "FETCH: ${FETCHOUTFILE}" >>${LOGOUT}
 222
 223   ( fetch -o ${FETCHOUTFILE} "${FETCHFILE}" >/dev/null 2>/dev/null ; echo "$?" > ${EXITFILE} ) &
 224   PID="$!"
 225   while
 226   z=1
 227   do
 228
 229     if [ -e "${FETCHOUTFILE}" ]
 230     then
 231       DSIZE=`du -k ${FETCHOUTFILE} | tr -d '\t' | cut -d '/' -f 1`
 232       if [ $(is_num "$DSIZE") ] ; then
 233       if [ $SIZE -lt $DSIZE ] ; then DSIZE="$SIZE"; fi
 234         echo "SIZE: ${SIZE} DOWNLOADED: ${DSIZE}"
 235         echo "SIZE: ${SIZE} DOWNLOADED: ${DSIZE}" >>${LOGOUT}
 236       fi
 237     fi
 238
 239     # Check if the download is finished
 240     ps -p ${PID} >/dev/null 2>/dev/null
 241     if [ $? -ne 0 ]
 242     then
 243       break;
 244     fi
 245
 246     sleep 2
 247   done
 248
 249   echo "FETCHDONE"
 250
 251   EXIT="`cat ${EXITFILE}`"
 252   if [ "${EXIT}" != "0" -a "$EXITFAILED" = "1" ]
 253   then
 254     exit_err "Error: Failed to download ${FETCHFILE}"
 255   fi
 256
 257   return $EXIT
 258
 259 };
 260
 261 # Function to return a the zpool name for this device
 262 get_zpool_name()
 263 {
 264   DEVICE="$1"
 265
 266   # Set the base name we use for zpools
 267   BASENAME="tank"
 268
 269   if [ ! -d "${TMPDIR}/.zpools" ] ; then
 270     mkdir -p ${TMPDIR}/.zpools
 271   fi
 272
 273   if [ -e "${TMPDIR}/.zpools/${DEVICE}" ] ; then
 274     cat ${TMPDIR}/.zpools/${DEVICE}
 275     return 0
 276   else
 277     # Need to generate a zpool name for this device
 278     NUM=`ls ${TMPDIR}/.zpools/ | wc -l | sed 's| ||g'`
 279     NEWNAME="${BASENAME}${NUM}"
 280     mkdir -p ${TMPDIR}/.zpools/`dirname $DEVICE`
 281     echo "$NEWNAME" >${TMPDIR}/.zpools/${DEVICE}
 282     echo "${NEWNAME}"
 283     return
 284   fi
 285 };
 286
 287 iscompressed()
 288 {
 289   local FILE
 290   local RES
 291
 292   FILE="$1"
 293   RES=1
 294
 295   if echo "${FILE}" | \
 296     grep -qiE '\.(Z|lzo|lzw|lzma|gz|bz2|xz|zip)$' 2>&1
 297   then
 298     RES=0
 299   fi
 300
 301   return ${RES}
 302 }
 303
 304 get_compression_type()
 305 {
 306   local FILE
 307   local SUFFIX
 308
 309   FILE="$1"
 310   SUFFIX=`echo "${FILE}" | sed -E 's|^(.+)\.(.+)$|\2|'`
 311
 312   VAL=""
 313   SUFFIX=`echo "${SUFFIX}" | tr A-Z a-z`
 314   case "${SUFFIX}" in
 315     z) VAL="lzw" ;;
 316     lzo) VAL="lzo" ;;
 317     lzw) VAL="lzw" ;;
 318     lzma) VAL="lzma" ;;
 319     gz) VAL="gzip" ;;
 320     bz2) VAL="bzip2" ;;
 321     xz) VAL="xz" ;;
 322     zip) VAL="zip" ;;
 323   esac
 324
 325   export VAL
 326 }
 327
 328 write_image()
 329 {
 330   local DEVICE_FILE
 331
 332   IMAGE_FILE="$1"
 333   DEVICE_FILE="$2"
 334
 335   if [ -z "${IMAGE_FILE}" ]
 336   then
 337     exit_err "ERROR: Image file not specified!"
 338   fi
 339
 340   if [ -z "${DEVICE_FILE}" ]
 341   then
 342     exit_err "ERROR: Device file not specified!"
 343   fi
 344
 345   if [ ! -f "${IMAGE_FILE}" ]
 346   then
 347     exit_err "ERROR: '${IMAGE_FILE}' does not exist!"
 348   fi
 349
 350   DEVICE_FILE="${DEVICE_FILE#/dev/}"
 351   DEVICE_FILE="/dev/${DEVICE_FILE}"
 352
 353   if [ ! -c "${DEVICE_FILE}" ]
 354   then
 355     exit_err "ERROR: '${DEVICE_FILE}' is not a character device!"
 356   fi
 357
 358   if iscompressed "${IMAGE_FILE}"
 359   then
 360         local COMPRESSION
 361
 362     get_compression_type "${IMAGE_FILE}"
 363         COMPRESSION="${VAL}"
 364
 365     case "${COMPRESSION}" in
 366       lzw)
 367         rc_halt "uncompress ${IMAGE_FILE} -c | dd of=${DEVICE_FILE}"
 368         IMAGE_FILE="${IMAGE_FILE%.Z}"
 369         ;;
 370
 371       lzo)
 372         rc_halt "lzop -d $IMAGE_{FILE} -c | dd of=${DEVICE_FILE}"
 373         IMAGE_FILE="${IMAGE_FILE%.lzo}"
 374         ;;
 375
 376       lzma)
 377         rc_halt "lzma -d ${IMAGE_FILE} -c | dd of=${DEVICE_FILE}"
 378         IMAGE_FILE="${IMAGE_FILE%.lzma}"
 379         ;;
 380
 381       gzip)
 382         rc_halt "gunzip ${IMAGE_FILE} -c | dd of=${DEVICE_FILE}"
 383         IMAGE_FILE="${IMAGE_FILE%.gz}"
 384         ;;
 385
 386       bzip2)
 387         rc_halt "bunzip2 ${IMAGE_FILE} -c | dd of=${DEVICE_FILE}"
 388         IMAGE_FILE="${IMAGE_FILE%.bz2}"
 389         ;;
 390
 391       xz)
 392         rc_halt "xz -d ${IMAGE_FILE} -c | dd of=${DEVICE_FILE}"
 393         IMAGE_FILE="${IMAGE_FILE%.xz}"
 394         ;;
 395
 396       zip)
 397         rc_halt "unzip ${IMAGE_FILE} -c | dd of=${DEVICE_FILE}"
 398         IMAGE_FILE="${IMAGE_FILE%.zip}"
 399         ;;
 400
 401       *)
 402         exit_err "ERROR: ${COMPRESSION} compression is not supported"
 403         ;;
 404     esac
 405
 406   else
 407     rc_halt "dd if=${IMAGE_FILE} of=${DEVICE_FILE}"
 408
 409   fi
 410 };
 411
 412 # Setup and install on a new disk / partition
 413 install_fresh()
 414 {
 415   # Lets start setting up the disk slices now
 416   setup_disk_slice
 417
 418   if [ -z "${ROOTIMAGE}" ]
 419   then
 420
 421     # Disk setup complete, now lets parse WORKINGSLICES and setup the bsdlabels
 422     setup_disk_label
 423
 424     # Now we've setup the bsdlabels, lets go ahead and run newfs / zfs
 425     # to setup the filesystems
 426     setup_filesystems
 427
 428     # Lets mount the partitions now
 429     mount_all_filesystems
 430
 431     # We are ready to begin extraction, lets start now
 432     init_extraction
 433
 434     # Check if we have any optional modules to load
 435     install_components
 436
 437     # Check if we have any packages to install
 438     install_packages
 439
 440     # Do any localization in configuration
 441     run_localize
 442
 443     # Save any networking config on the installed system
 444     save_networking_install
 445
 446     # Now add any users
 447     setup_users
 448
 449     # Do any last cleanup / setup before unmounting
 450     run_final_cleanup
 451
 452     # Now run any commands specified
 453     run_commands
 454
 455     # Unmount and finish up
 456     unmount_all_filesystems
 457   fi
 458
 459   echo_log "Installation finished!"
 460 };
 461
 462 # Extract the system to a pre-mounted directory
 463 install_extractonly()
 464 {
 465   # We are ready to begin extraction, lets start now
 466   init_extraction
 467
 468   # Check if we have any optional modules to load
 469   install_components
 470
 471   # Check if we have any packages to install
 472   install_packages
 473
 474   # Do any localization in configuration
 475   run_localize
 476
 477   # Save any networking config on the installed system
 478   save_networking_install
 479
 480   # Now add any users
 481   setup_users
 482
 483   # Now run any commands specified
 484   run_commands
 485
 486   # Set a hostname on the install system
 487   setup_hostname
 488
 489   # Set the root_pw if it is specified
 490   set_root_pw
 491
 492   echo_log "Installation finished!"
 493 };
 494
 495 install_image()
 496 {
 497   # We are ready to begin extraction, lets start now
 498   init_extraction
 499
 500   echo_log "Installation finished!"
 501 };
 502
 503 install_upgrade()
 504 {
 505   # We're going to do an upgrade, skip all the disk setup
 506   # and start by mounting the target drive/slices
 507   mount_upgrade
 508
 509   # Start the extraction process
 510   init_extraction
 511
 512   # Do any localization in configuration
 513   run_localize
 514
 515   # Now run any commands specified
 516   run_commands
 517
 518   # Merge any old configuration files
 519   merge_old_configs
 520
 521   # Check if we have any optional modules to load
 522   install_components
 523
 524   # Check if we have any packages to install
 525   install_packages
 526
 527   # All finished, unmount the file-systems
 528   unmount_upgrade
 529
 530   echo_log "Upgrade finished!"
 531 };