pfSense

#!/bin/sh

# Functions which perform the mounting / unmount for upgrades

mount_target_slice()

{

  MPART="${1}"

  # Import any zpools

  zpool import -a

  # Set a variable of files we want to make backups of before doing upgrade

  BKFILES="/etc/rc.conf /boot/loader.conf"

  if [ -e "/dev/${MPART}a.journal" ]

  then

    rc_halt "mount /dev/${MPART}.journal ${FSMNT}"

  elif [ -e "/dev/${MPART}a" ]

  then

    mount /dev/${MPART}a ${FSMNT}

    if [ "$?" != "0" ]

    then

      # Try ZFS on this slice

      mount -t zfs ${MPART}a ${FSMNT}

      if [ "${?}" != "0" ]

      then

        exit_err "Failed to mount ${MPART}"

      fi

    fi

  fi

  mount -t devfs devfs ${FSMNT}/dev

  chroot ${FSMNT} /sbin/mount -a >>${LOGOUT} 2>>${LOGOUT}

  chroot ${FSMNT} umount /proc >/dev/null 2>/dev/null 

  chroot ${FSMNT} umount /compat/linux/proc  >/dev/null 2>/dev/null

  # Save which partition was mounted, so we may unmount it later

  echo "umount -f /dev/${MPART}a" >>${TMPDIR}/.upgrade-unmount

  # Now before we start the upgrade, make sure we set our noschg flags

  echo_log "Cleaning up old filesystem... Please wait..."

  rc_halt "chflags -R noschg ${FSMNT}"

  # Make backup copies of some files

  for i in ${BKFILES}

  do

    cp ${FSMNT}${i} ${FSMNT}${i}.preUpgrade >/dev/null 2>/dev/null

  done

  # Remove some old dirs

  rm -rf ${FSMNT}/etc/rc.d >/dev/null 2>/dev/null

  # If we are doing PC-BSD install, lets cleanup old pkgs on disk

  if [ "$INSTALLTYPE" != "FreeBSD" ]

  then

    echo_log "Removing old packages, this may take a while... Please wait..."

    echo '#/bin/sh

for i in `pkg_info -E \*`

do

  echo "Uninstalling package: ${i}"

  pkg_delete -f ${i} >/dev/null 2>/dev/null

done

' >${FSMNT}/.cleanPkgs.sh

    chmod 755 ${FSMNT}/.cleanPkgs.sh

    chroot ${FSMNT} /.cleanPkgs.sh

    rm ${FSMNT}/.cleanPkgs.sh

    run_chroot_cmd "pkg_delete -f \*" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /usr/PCBSD" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /PCBSD" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /var/db/pkgs" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /usr/local32" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /usr/sbin" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /usr/lib" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /usr/bin" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /boot/kernel" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /sbin" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /bin" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /lib" >/dev/null 2>/dev/null

    run_chroot_cmd "rm -rf /libexec" >/dev/null 2>/dev/null

  fi

}

# Mount the target upgrade partitions

mount_upgrade()

{

  # Start with disk0

  disknum="0"

  # Make sure we remove the old upgrade-mount script

  rm -rf ${TMPDIR}/.upgrade-unmount >/dev/null 2>/dev/null

  # We are ready to start mounting, lets read the config and do it

  while read line

  do

     echo $line | grep "^disk${disknum}=" >/dev/null 2>/dev/null

     if [ "$?" = "0" ]

     then

       # Found a disk= entry, lets get the disk we are working on

       get_value_from_string "${line}"

       strip_white_space "$VAL"

       DISK="$VAL"

       # Before we go further, lets confirm this disk really exists

       if [ ! -e "/dev/${DISK}" ]

       then

         exit_err "ERROR: The disk ${DISK} does not exist!"

       fi

     fi

     echo $line | grep "^partition=" >/dev/null 2>/dev/null

     if [ "$?" = "0" ]

     then

       # Found a partition= entry, lets read / set it 

       get_value_from_string "${line}"

       strip_white_space "$VAL"

       PTYPE="$VAL"

       # We are using free space, figure out the slice number

       if [ "${PTYPE}" = "free" -o "${PTYPE}" = "all" ]

       then

           exit_err "ERROR: Invalid upgrade partition=${PTYPE}"

       fi

     fi

     echo $line | grep "^commitDiskPart" >/dev/null 2>/dev/null

     if [ "$?" = "0" ]

     then

       # Found our flag to commit this disk setup / lets do sanity check and do it

       if [ ! -z "${DISK}" -a ! -z "${PTYPE}" ]

       then

         case ${PTYPE} in

           s1|s2|s3|s4) tmpSLICE="${DISK}${PTYPE}a" ;;

                     *) exit_err "ERROR: Unknown PTYPE: $PTYPE" ;;

         esac

         if [ ! -e "/dev/${tmpSLICE}" ]

         then

           exit_err "ERROR: /dev/${tmpSLICE} does not exist! Mount failed!"

         fi

         # Start mounting this slice

         mount_target_slice "${DISK}${PTYPE}" 

         # Increment our disk counter to look for next disk and unset

         unset PTYPE DISK

         disknum="`expr $disknum + 1`"

       else

         exit_err "ERROR: commitDiskPart was called without procceding disk<num>= and partition= entries!!!" 

       fi

     fi

  done <${CFGF}

};

copy_skel_files_upgrade()

{

    # Now make sure we fix any user profile scripts, which cause problems from 7.x->8.x

    echo '#!/bin/sh

cd /home

for i in `ls`

do

  # Backup the old profile dirs

  if [ -d "${i}" ]

  then

    mv /home/${i}/.kde4 /home/${i}/.kde4.preUpgrade >/dev/null 2>/dev/null

    mv /home/${i}/.kde /home/${i}/.kde.preUpgrade >/dev/null 2>/dev/null

    mv /home/${i}/.fluxbox /home/${i}/.fluxbox.preUpgrade >/dev/null 2>/dev/null

    # Copy over the skel directories

    tar cv --exclude "./dot.*" -f - -C /usr/share/skel . 2>/dev/null | tar xvf - -C /home/${i} 2>/dev/null

    for j in `ls /usr/share/skel/dot*`

    do

      dname=`echo ${j} | sed s/dot//`

      cp /usr/share/skel/${j} /home/${i}/${dname}

    done

    chown -R ${i}:${i} /home/${i}

  fi

done

' >${FSMNT}/.fixUserProfile.sh  

    chmod 755 ${FSMNT}/.fixUserProfile.sh

    chroot ${FSMNT} /.fixUserProfile.sh >/dev/null 2>/dev/null

    rm ${FSMNT}/.fixUserProfile.sh

    # if the user wants to keep their original .kde4 profile

    ###########################################################################

    get_value_from_cfg "upgradeKeepDesktopProfile"

    if [ "$VAL" = "YES" -o "$VAL" = "yes" ] ; then

      echo '#!/bin/sh

      cd /home

for i in `ls`

do

  # Import the old config again

  if [ -d "${i}/.kde4.preUpgrade" ]

  then

    # Copy over the skel directories

    tar cv -f - -C /home/${i}/.kde4.preUpgrade . 2>/dev/null | tar xvf - -C /home/${i}/.kde4 2>/dev/null

    chown -R ${i}:${i} /home/${i}/.kde4

  fi

done

' >${FSMNT}/.fixUserProfile.sh

      chmod 755 ${FSMNT}/.fixUserProfile.sh

      chroot ${FSMNT} /.fixUserProfile.sh >/dev/null 2>/dev/null

      rm ${FSMNT}/.fixUserProfile.sh

    fi

};

# Function which merges some configuration files with the new defaults

merge_old_configs()

{

  # Merge the loader.conf with old

  cp ${FSMNT}/boot/loader.conf ${FSMNT}/boot/loader.conf.new

  merge_config "${FSMNT}/boot/loader.conf.preUpgrade" "${FSMNT}/boot/loader.conf.new" "${FSMNT}/boot/loader.conf"

  rm ${FSMNT}/boot/loader.conf.new

  # Merge the rc.conf with old

  cp ${FSMNT}/etc/rc.conf ${FSMNT}/etc/rc.conf.new

  merge_config "${FSMNT}/etc/rc.conf.preUpgrade" "${FSMNT}/etc/rc.conf.new" "${FSMNT}/etc/rc.conf"

  rm ${FSMNT}/etc/rc.conf.new

};

# Function which unmounts all the mounted file-systems

unmount_upgrade()

{

   # If on PC-BSD, make sure we copy any fixed skel files

   if [ "$INSTALLTYPE" != "FreeBSD" ] ; then

     copy_skel_files_upgrade

   fi

   cd /

   # Unmount FS

   chroot ${FSMNT} /sbin/umount -a >/dev/null 2>/dev/null

   umount ${FSMNT}/usr >/dev/null 2>/dev/null

   umount ${FSMNT}/var >/dev/null 2>/dev/null

   umount ${FSMNT}/dev >/dev/null 2>/dev/null

   umount ${FSMNT} >/dev/null 2>/dev/null

   # Run our saved unmount script for these file-systems

   rc_nohalt "sh ${TMPDIR}/.upgrade-unmount"

   umount ${CDMNT} 

};