source: src-sh/pc-sysinstall/backend/functions-cleanup.sh @ 1620346

9.1-release9.2-releasereleng/10.0releng/10.0.1releng/10.0.2releng/10.0.3releng/10.1
Last change on this file since 1620346 was 1620346, checked in by Kris Moore <kris@…>, 21 months ago

Initial import of PC-BSD /current/ SVN repo

  • Property mode set to 100755
File size: 11.5 KB
Line 
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: head/usr.sbin/pc-sysinstall/backend/functions-cleanup.sh 240165 2012-09-06 14:59:53Z jpaetzel $
27
28# Functions which perform the final cleanup after an install
29
30# Finishes up with ZFS setup before unmounting
31zfs_cleanup_unmount()
32{
33  # Loop through our FS and see if we have any ZFS partitions to cleanup
34  for PART in `ls ${PARTDIR}`
35  do
36    PARTDEV=`echo $PART | sed 's|-|/|g'`
37    PARTFS="`cat ${PARTDIR}/${PART} | cut -d '#' -f 1`"
38    PARTMNT="`cat ${PARTDIR}/${PART} | cut -d '#' -f 2`"
39    ZPOOLNAME=$(get_zpool_name "${PARTDEV}")
40
41    if [ "$PARTFS" = "ZFS" ]
42    then
43      # Check if we have multiple zfs mounts specified
44      for ZMNT in `echo ${PARTMNT} | sed 's|,| |g'`
45      do
46        if [ "${ZMNT}" = "/" ]
47        then
48          # Make sure we haven't already added the zfs boot line when
49          # Creating a dedicated "/boot" partition
50          cat ${FSMNT}/boot/loader.conf 2>/dev/null | grep -q "vfs.root.mountfrom=" 2>/dev/null
51          if [ $? -ne 0 ] ; then
52            echo "vfs.root.mountfrom=\"zfs:${ZPOOLNAME}/ROOT/default\"" >> ${FSMNT}/boot/loader.conf
53          fi
54          export FOUNDZFSROOT="${ZPOOLNAME}"
55        fi
56      done
57      FOUNDZFS="1"
58    fi
59  done
60
61  if [ -n "${FOUNDZFS}" ]
62  then
63    # Check if we need to add our ZFS flags to rc.conf, src.conf and loader.conf
64    cat ${FSMNT}/boot/loader.conf 2>/dev/null | grep -q 'zfs_load="YES"' 2>/dev/null
65    if [ $? -ne 0 ]
66    then
67      echo 'zfs_load="YES"' >>${FSMNT}/boot/loader.conf
68    fi
69    cat ${FSMNT}/etc/rc.conf 2>/dev/null | grep -q 'zfs_enable="YES"' 2>/dev/null
70    if [ $? -ne 0 ]
71    then
72      echo 'zfs_enable="YES"' >>${FSMNT}/etc/rc.conf
73    fi
74
75    sleep 2
76    # Copy over any ZFS cache data
77    cp /boot/zfs/* ${FSMNT}/boot/zfs/
78
79    # Copy the hostid so that our zfs cache works
80    cp /etc/hostid ${FSMNT}/etc/hostid
81  fi
82
83  # Loop through our FS and see if we have any ZFS partitions to cleanup
84  for PART in `ls ${PARTDIR}`
85  do
86    PARTDEV=`echo $PART | sed 's|-|/|g'`
87    PARTFS="`cat ${PARTDIR}/${PART} | cut -d '#' -f 1`"
88    PARTMNT="`cat ${PARTDIR}/${PART} | cut -d '#' -f 2`"
89    PARTENC="`cat ${PARTDIR}/${PART} | cut -d '#' -f 3`"
90    ZPOOLNAME=$(get_zpool_name "${PARTDEV}")
91
92    if [ "$PARTFS" = "ZFS" ]
93    then
94
95      # Create a list of zpool names we can export
96      echo $ZPOOLEXPORTS | grep -q "$ZPOOLNAME "
97      if [ $? -ne 0 ] ; then
98        export ZPOOLEXPORTS="$ZPOOLNAME $ZPOOLEXPORTS"
99      fi
100
101      # Check if we have multiple zfs mounts specified
102      for ZMNT in `echo ${PARTMNT} | sed 's|,| |g'`
103      do
104        ZMNT="`echo $ZMNT | cut -d '(' -f 1`"
105        PARTMNTREV="${ZMNT} ${PARTMNTREV}"
106      done
107
108      for ZMNT in ${PARTMNTREV}
109      do
110        if [ "${ZMNT}" = "/" ] ; then continue ; fi
111        # Some ZFS like /swap aren't mounted, and dont need unmounting
112        mount | grep -q "${FSMNT}${ZMNT}"
113        if [ $? -eq 0 ] ; then
114          rc_halt "zfs unmount ${ZPOOLNAME}${ZMNT}"
115          rc_halt "zfs set mountpoint=${ZMNT} ${ZPOOLNAME}${ZMNT}"
116        fi
117        sleep 2
118      done
119    fi
120  done
121
122};
123
124# Function which performs the specific setup for using a /boot partition
125setup_dedicated_boot_part()
126{
127  ROOTFS="${1}"
128  ROOTFSTYPE="${2}"
129  BOOTFS="${3}"
130  BOOTMNT="${4}"
131
132  # Set the root mount in loader.conf
133  echo "vfs.root.mountfrom=\"${ROOTFSTYPE}:${ROOTFS}\"" >> ${FSMNT}/boot/loader.conf
134  rc_halt "mkdir -p ${FSMNT}/${BOOTMNT}/boot"
135  rc_halt "mv ${FSMNT}/boot/* ${FSMNT}${BOOTMNT}/boot/"
136  rc_halt "mv ${FSMNT}${BOOTMNT}/boot ${FSMNT}/boot/"
137  rc_halt "umount ${BOOTFS}"
138  rc_halt "mount ${BOOTFS} ${FSMNT}${BOOTMNT}"
139  rc_halt "rmdir ${FSMNT}/boot"
140
141  # Strip the '/' from BOOTMNT before making symlink
142  BOOTMNTNS="`echo ${BOOTMNT} | sed 's|/||g'`"
143  rc_halt "chroot ${FSMNT} ln -s ${BOOTMNTNS}/boot /boot"
144 
145};
146
147# Function which creates the /etc/fstab for the installed system
148setup_fstab()
149{
150  FSTAB="${FSMNT}/etc/fstab"
151  rm ${FSTAB} >/dev/null 2>/dev/null
152
153  # Create the header
154  echo "# Device                Mountpoint              FStype          Options Dump Pass" >> ${FSTAB}
155
156  # Loop through the partitions, and start creating /etc/fstab
157  for PART in `ls ${PARTDIR}`
158  do
159    PARTDEV=`echo $PART | sed 's|-|/|g'`
160    PARTFS="`cat ${PARTDIR}/${PART} | cut -d '#' -f 1`"
161    PARTMNT="`cat ${PARTDIR}/${PART} | cut -d '#' -f 2`"
162    PARTENC="`cat ${PARTDIR}/${PART} | cut -d '#' -f 3`"
163    PARTLABEL="`cat ${PARTDIR}/${PART} | cut -d '#' -f 4`"
164
165    # Unset EXT
166    EXT=""
167
168    # Set mount options for file-systems
169    case $PARTFS in
170      UFS+J) MNTOPTS="rw,noatime,async" ;;
171      SWAP) MNTOPTS="sw" ;;
172      *) MNTOPTS="rw,noatime" ;;
173    esac
174
175
176    # Figure out if we are using a glabel, or the raw name for this entry
177    if [ -n "${PARTLABEL}" ]
178    then
179      DEVICE="label/${PARTLABEL}"
180    else
181      # Check if using encryption
182      if [ "${PARTENC}" = "ON" ] ; then
183        EXT=".eli"
184      fi
185
186      if [ "${PARTFS}" = "UFS+J" ] ; then
187        EXT="${EXT}.journal"
188      fi
189      DEVICE="${PARTDEV}${EXT}"
190    fi
191
192
193    # Set our ROOTFSTYPE for loader.conf if necessary
194    check_for_mount "${PARTMNT}" "/"
195    if [ $? -eq 0 ] ; then
196      if [ "${PARTFS}" = "ZFS" ] ; then
197        ROOTFSTYPE="zfs"
198        ZPOOLNAME=$(get_zpool_name "${PARTDEV}")
199        ROOTFS="${ZPOOLNAME}/ROOT/default"
200      else
201        ROOTFS="${DEVICE}"
202        ROOTFSTYPE="ufs"
203      fi
204    fi
205
206    # Only create non-zfs partitions
207    if [ "${PARTFS}" != "ZFS" ]
208    then
209
210      # Make sure geom_journal is loaded
211      if [ "${PARTFS}" = "UFS+J" ] ; then
212        setup_gjournal
213      fi
214
215      # Save the BOOTFS for call at the end
216      if [ "${PARTMNT}" = "/boot" ] ; then
217        BOOTFS="${PARTDEV}${EXT}"
218        BOOTMNT="${BOOT_PART_MOUNT}"
219        PARTMNT="${BOOTMNT}"
220      fi
221
222      # Echo out the fstab entry now
223      if [ "${PARTFS}" = "SWAP" ]
224      then
225        echo "/dev/${DEVICE}    none            swap    ${MNTOPTS}      0       0" >> ${FSTAB}
226      else
227        echo "/dev/${DEVICE}    ${PARTMNT}              ufs     ${MNTOPTS}      1       1" >> ${FSTAB}
228      fi
229
230    fi # End of ZFS Check
231  done
232
233  # Setup some specific PC-BSD fstab options
234  if [ "$INSTALLTYPE" != "FreeBSD" ]
235  then
236    echo "procfs                        /proc                   procfs          rw              0       0" >> ${FSTAB}
237    echo "linprocfs             /compat/linux/proc      linprocfs       rw              0       0" >> ${FSTAB}
238  fi
239
240  # If we have a dedicated /boot, run the post-install setup of it now
241  if [ ! -z "${BOOTMNT}" ] ; then
242    setup_dedicated_boot_part "${ROOTFS}" "${ROOTFSTYPE}" "${BOOTFS}" "${BOOTMNT}"
243  fi
244
245};
246
247# Setup our disk mirroring with gmirror
248setup_gmirror()
249{
250  cat ${FSMNT}/boot/loader.conf 2>/dev/null | grep -q 'geom_mirror_load="YES"' 2>/dev/null
251  if [ $? -ne 0 ]
252  then
253    echo 'geom_mirror_load="YES"' >>${FSMNT}/boot/loader.conf
254  fi
255
256};
257
258# Function which saves geli keys and sets up loading of them at boot
259setup_geli_loading()
260{
261
262  # Make our keys dir
263  mkdir -p ${FSMNT}/boot/keys >/dev/null 2>/dev/null
264
265  cd ${GELIKEYDIR}
266  for KEYFILE in `ls`
267  do
268     # Figure out the partition name based on keyfile name removing .key
269     PART="`echo ${KEYFILE} | cut -d '.' -f 1`"
270     PARTDEV="`echo ${PART} | sed 's|-|/|g'`"
271     PARTNAME="`echo ${PART} | sed 's|-dev-||g'`"
272
273     rc_halt "geli configure -b ${PARTDEV}"
274
275     # If no passphrase, setup key files
276     if [ ! -e "${PARTDIR}-enc/${PART}-encpass" ] ; then
277       echo "geli_${PARTNAME}_keyfile0_load=\"YES\"" >> ${FSMNT}/boot/loader.conf
278       echo "geli_${PARTNAME}_keyfile0_type=\"${PARTNAME}:geli_keyfile0\"" >> ${FSMNT}/boot/loader.conf
279       echo "geli_${PARTNAME}_keyfile0_name=\"/boot/keys/${PARTNAME}.key\"" >> ${FSMNT}/boot/loader.conf
280
281       # Copy the key to the disk
282       rc_halt "cp ${GELIKEYDIR}/${KEYFILE} ${FSMNT}/boot/keys/${PARTNAME}.key"
283     fi
284
285  done
286
287  # Make sure we have geom_eli set to load at boot
288  cat ${FSMNT}/boot/loader.conf 2>/dev/null | grep -q 'geom_eli_load="YES"' 2>/dev/null
289  if [ $? -ne 0 ]
290  then
291    echo 'geom_eli_load="YES"' >>${FSMNT}/boot/loader.conf
292  fi
293
294};
295
296
297# Function to generate a random hostname if none was specified
298gen_hostname()
299{
300  RAND="`jot -r 1 1 9000`"
301
302  if [ "$INSTALLTYPE" = "FreeBSD" ]
303  then
304    VAL="freebsd-${RAND}" 
305  else
306    VAL="pcbsd-${RAND}" 
307  fi
308
309  export VAL
310
311};
312
313# Function which sets up the hostname for the system
314setup_hostname()
315{
316
317  get_value_from_cfg hostname
318  HOSTNAME="${VAL}"
319
320  # If we don't have a hostname, make one up
321  if [ -z "${HOSTNAME}" ]
322  then
323    gen_hostname
324    HOSTNAME="${VAL}"
325  fi
326
327  # Clean up any saved hostname
328  cat ${FSMNT}/etc/rc.conf | grep -v "hostname=" >${FSMNT}/etc/rc.conf.new
329  mv ${FSMNT}/etc/rc.conf.new ${FSMNT}/etc/rc.conf
330
331  # Set the hostname now
332  echo_log "Setting hostname: ${HOSTNAME}"
333  echo "hostname=\"${HOSTNAME}\"" >> ${FSMNT}/etc/rc.conf
334  sed -i -e "s|my.domain|${HOSTNAME} ${HOSTNAME}|g" ${FSMNT}/etc/hosts
335
336};
337
338
339# Check and make sure geom_journal is enabled on the system
340setup_gjournal()
341{
342
343  # Make sure we have geom_journal set to load at boot
344  cat ${FSMNT}/boot/loader.conf 2>/dev/null | grep -q 'geom_journal_load="YES"' 2>/dev/null
345  if [ $? -ne 0 ]
346  then
347    echo 'geom_journal_load="YES"' >>${FSMNT}/boot/loader.conf
348  fi
349
350};
351
352# Function which sets the root password from the install config
353set_root_pw()
354{
355  # Get the plaintext string
356  get_value_from_cfg_with_spaces rootPass
357  local PW="${VAL}"
358
359  # Get the encrypted string
360  get_value_from_cfg_with_spaces rootEncPass
361  local ENCPW="${VAL}"
362
363  # If we don't have a root pass, return
364  if [ -z "${PW}" -a -z "${ENCPW}" ] ; then return 0 ; fi
365
366  echo_log "Setting root password"
367
368  # Check if setting plaintext password
369  if [ -n "${PW}" ] ; then
370    echo "${PW}" > ${FSMNT}/.rootpw
371    run_chroot_cmd "cat /.rootpw | pw usermod root -h 0"
372    rc_halt "rm ${FSMNT}/.rootpw"
373  fi
374
375  # Check if setting encrypted password
376  if [ -n "${ENCPW}" ] ; then
377    echo "${ENCPW}" > ${FSMNT}/.rootpw
378    run_chroot_cmd "cat /.rootpw | pw usermod root -H 0"
379    rc_halt "rm ${FSMNT}/.rootpw"
380  fi
381
382};
383
384
385run_final_cleanup()
386{
387  # Check if we need to run any gmirror setup
388  ls ${MIRRORCFGDIR}/* >/dev/null 2>/dev/null
389  if [ $? -eq 0 ]
390  then
391    # Lets setup gmirror now
392    setup_gmirror
393  fi
394
395  # Check if we need to save any geli keys
396  ls ${GELIKEYDIR}/* >/dev/null 2>/dev/null
397  if [ $? -eq 0 ]
398  then
399    # Lets setup geli loading
400    setup_geli_loading
401  fi
402
403  # Set a hostname on the install system
404  setup_hostname
405
406  # Set the root_pw if it is specified
407  set_root_pw
408
409  # Generate the fstab for the installed system
410  setup_fstab
411};
Note: See TracBrowser for help on using the repository browser.