xref: /freebsd/usr.sbin/freebsd-update/freebsd-update.sh (revision bc0c6c9cf3a9f9a54dbdd92dd8f1f65ff8092d17)
148ffe56aSColin Percival#!/bin/sh
248ffe56aSColin Percival
348ffe56aSColin Percival#-
44d846d26SWarner Losh# SPDX-License-Identifier: BSD-2-Clause
51de7b4b8SPedro F. Giffuni#
62328d598SColin Percival# Copyright 2004-2007 Colin Percival
748ffe56aSColin Percival# All rights reserved
848ffe56aSColin Percival#
948ffe56aSColin Percival# Redistribution and use in source and binary forms, with or without
1048ffe56aSColin Percival# modification, are permitted providing that the following conditions
1148ffe56aSColin Percival# are met:
1248ffe56aSColin Percival# 1. Redistributions of source code must retain the above copyright
1348ffe56aSColin Percival#    notice, this list of conditions and the following disclaimer.
1448ffe56aSColin Percival# 2. Redistributions in binary form must reproduce the above copyright
1548ffe56aSColin Percival#    notice, this list of conditions and the following disclaimer in the
1648ffe56aSColin Percival#    documentation and/or other materials provided with the distribution.
1748ffe56aSColin Percival#
1848ffe56aSColin Percival# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
1948ffe56aSColin Percival# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
2048ffe56aSColin Percival# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2148ffe56aSColin Percival# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
2248ffe56aSColin Percival# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
2348ffe56aSColin Percival# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
2448ffe56aSColin Percival# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
2548ffe56aSColin Percival# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
2648ffe56aSColin Percival# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
2748ffe56aSColin Percival# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
2848ffe56aSColin Percival# POSSIBILITY OF SUCH DAMAGE.
2948ffe56aSColin Percival
3048ffe56aSColin Percival#### Usage function -- called from command-line handling code.
3148ffe56aSColin Percival
3248ffe56aSColin Percival# Usage instructions.  Options not listed:
3348ffe56aSColin Percival# --debug	-- don't filter output from utilities
3448ffe56aSColin Percival# --no-stats	-- don't show progress statistics while fetching files
3548ffe56aSColin Percivalusage () {
3648ffe56aSColin Percival	cat <<EOF
37c76da1f0SFaraz Vahediusage: `basename $0` [options] command ...
3848ffe56aSColin Percival
3948ffe56aSColin PercivalOptions:
4048ffe56aSColin Percival  -b basedir   -- Operate on a system mounted at basedir
4148ffe56aSColin Percival                  (default: /)
4248ffe56aSColin Percival  -d workdir   -- Store working files in workdir
4348ffe56aSColin Percival                  (default: /var/db/freebsd-update/)
4448ffe56aSColin Percival  -f conffile  -- Read configuration options from conffile
4548ffe56aSColin Percival                  (default: /etc/freebsd-update.conf)
46df7e2e0dSEd Maste  -F           -- Force a fetch operation to proceed in the
47df7e2e0dSEd Maste                  case of an unfinished upgrade
48c76da1f0SFaraz Vahedi  -j jail      -- Operate on the given jail specified by jid or name
4948ffe56aSColin Percival  -k KEY       -- Trust an RSA key with SHA256 hash of KEY
50e0e5bf4dSPoul-Henning Kamp  -r release   -- Target for upgrade (e.g., 13.2-RELEASE)
5148ffe56aSColin Percival  -s server    -- Server from which to fetch updates
5248ffe56aSColin Percival                  (default: update.FreeBSD.org)
5348ffe56aSColin Percival  -t address   -- Mail output of cron command, if any, to address
5448ffe56aSColin Percival                  (default: root)
558935f242SAllan Jude  --not-running-from-cron
568935f242SAllan Jude               -- Run without a tty, for use by automated tools
57b39ce43eSColin Percival  --currently-running release
58b39ce43eSColin Percival               -- Update as if currently running this release
5948ffe56aSColin PercivalCommands:
6048ffe56aSColin Percival  fetch        -- Fetch updates from server
6148ffe56aSColin Percival  cron         -- Sleep rand(3600) seconds, fetch updates, and send an
6248ffe56aSColin Percival                  email if updates were found
63db6b0a61SColin Percival  upgrade      -- Fetch upgrades to FreeBSD version specified via -r option
648cfda118SMichael Gmelin  updatesready -- Check if there are fetched updates ready to install
65db6b0a61SColin Percival  install      -- Install downloaded updates or upgrades
6648ffe56aSColin Percival  rollback     -- Uninstall most recently installed updates
6775cb6429SEd Maste  IDS          -- Compare the system against an index of "known good" files
688cfda118SMichael Gmelin  showconfig   -- Show configuration
6948ffe56aSColin PercivalEOF
7048ffe56aSColin Percival	exit 0
7148ffe56aSColin Percival}
7248ffe56aSColin Percival
7348ffe56aSColin Percival#### Configuration processing functions
7448ffe56aSColin Percival
7548ffe56aSColin Percival#-
7648ffe56aSColin Percival# Configuration options are set in the following order of priority:
7748ffe56aSColin Percival# 1. Command line options
7848ffe56aSColin Percival# 2. Configuration file options
7948ffe56aSColin Percival# 3. Default options
8048ffe56aSColin Percival# In addition, certain options (e.g., IgnorePaths) can be specified multiple
8148ffe56aSColin Percival# times and (as long as these are all in the same place, e.g., inside the
8248ffe56aSColin Percival# configuration file) they will accumulate.  Finally, because the path to the
8348ffe56aSColin Percival# configuration file can be specified at the command line, the entire command
8448ffe56aSColin Percival# line must be processed before we start reading the configuration file.
8548ffe56aSColin Percival#
8648ffe56aSColin Percival# Sound like a mess?  It is.  Here's how we handle this:
8748ffe56aSColin Percival# 1. Initialize CONFFILE and all the options to "".
8848ffe56aSColin Percival# 2. Process the command line.  Throw an error if a non-accumulating option
8948ffe56aSColin Percival#    is specified twice.
9048ffe56aSColin Percival# 3. If CONFFILE is "", set CONFFILE to /etc/freebsd-update.conf .
9148ffe56aSColin Percival# 4. For all the configuration options X, set X_saved to X.
9248ffe56aSColin Percival# 5. Initialize all the options to "".
9348ffe56aSColin Percival# 6. Read CONFFILE line by line, parsing options.
9448ffe56aSColin Percival# 7. For each configuration option X, set X to X_saved iff X_saved is not "".
9548ffe56aSColin Percival# 8. Repeat steps 4-7, except setting options to their default values at (6).
9648ffe56aSColin Percival
9748ffe56aSColin PercivalCONFIGOPTIONS="KEYPRINT WORKDIR SERVERNAME MAILTO ALLOWADD ALLOWDELETE
9848ffe56aSColin Percival    KEEPMODIFIEDMETADATA COMPONENTS IGNOREPATHS UPDATEIFUNMODIFIED
9908e23beeSColin Percival    BASEDIR VERBOSELEVEL TARGETRELEASE STRICTCOMPONENTS MERGECHANGES
10023d827efSSimon L. B. Nielsen    IDSIGNOREPATHS BACKUPKERNEL BACKUPKERNELDIR BACKUPKERNELSYMBOLFILES"
10148ffe56aSColin Percival
10248ffe56aSColin Percival# Set all the configuration options to "".
10348ffe56aSColin Percivalnullconfig () {
10448ffe56aSColin Percival	for X in ${CONFIGOPTIONS}; do
10548ffe56aSColin Percival		eval ${X}=""
10648ffe56aSColin Percival	done
10748ffe56aSColin Percival}
10848ffe56aSColin Percival
10948ffe56aSColin Percival# For each configuration option X, set X_saved to X.
11048ffe56aSColin Percivalsaveconfig () {
11148ffe56aSColin Percival	for X in ${CONFIGOPTIONS}; do
11248ffe56aSColin Percival		eval ${X}_saved=\$${X}
11348ffe56aSColin Percival	done
11448ffe56aSColin Percival}
11548ffe56aSColin Percival
11648ffe56aSColin Percival# For each configuration option X, set X to X_saved if X_saved is not "".
11748ffe56aSColin Percivalmergeconfig () {
11848ffe56aSColin Percival	for X in ${CONFIGOPTIONS}; do
11948ffe56aSColin Percival		eval _=\$${X}_saved
12048ffe56aSColin Percival		if ! [ -z "${_}" ]; then
12148ffe56aSColin Percival			eval ${X}=\$${X}_saved
12248ffe56aSColin Percival		fi
12348ffe56aSColin Percival	done
12448ffe56aSColin Percival}
12548ffe56aSColin Percival
12648ffe56aSColin Percival# Set the trusted keyprint.
12748ffe56aSColin Percivalconfig_KeyPrint () {
12848ffe56aSColin Percival	if [ -z ${KEYPRINT} ]; then
12948ffe56aSColin Percival		KEYPRINT=$1
13048ffe56aSColin Percival	else
13148ffe56aSColin Percival		return 1
13248ffe56aSColin Percival	fi
13348ffe56aSColin Percival}
13448ffe56aSColin Percival
13548ffe56aSColin Percival# Set the working directory.
13648ffe56aSColin Percivalconfig_WorkDir () {
13748ffe56aSColin Percival	if [ -z ${WORKDIR} ]; then
13848ffe56aSColin Percival		WORKDIR=$1
13948ffe56aSColin Percival	else
14048ffe56aSColin Percival		return 1
14148ffe56aSColin Percival	fi
14248ffe56aSColin Percival}
14348ffe56aSColin Percival
14448ffe56aSColin Percival# Set the name of the server (pool) from which to fetch updates
14548ffe56aSColin Percivalconfig_ServerName () {
14648ffe56aSColin Percival	if [ -z ${SERVERNAME} ]; then
14748ffe56aSColin Percival		SERVERNAME=$1
14848ffe56aSColin Percival	else
14948ffe56aSColin Percival		return 1
15048ffe56aSColin Percival	fi
15148ffe56aSColin Percival}
15248ffe56aSColin Percival
15348ffe56aSColin Percival# Set the address to which 'cron' output will be mailed.
15448ffe56aSColin Percivalconfig_MailTo () {
15548ffe56aSColin Percival	if [ -z ${MAILTO} ]; then
15648ffe56aSColin Percival		MAILTO=$1
15748ffe56aSColin Percival	else
15848ffe56aSColin Percival		return 1
15948ffe56aSColin Percival	fi
16048ffe56aSColin Percival}
16148ffe56aSColin Percival
16248ffe56aSColin Percival# Set whether FreeBSD Update is allowed to add files (or directories, or
16348ffe56aSColin Percival# symlinks) which did not previously exist.
16448ffe56aSColin Percivalconfig_AllowAdd () {
16548ffe56aSColin Percival	if [ -z ${ALLOWADD} ]; then
16648ffe56aSColin Percival		case $1 in
16748ffe56aSColin Percival		[Yy][Ee][Ss])
16848ffe56aSColin Percival			ALLOWADD=yes
16948ffe56aSColin Percival			;;
17048ffe56aSColin Percival		[Nn][Oo])
17148ffe56aSColin Percival			ALLOWADD=no
17248ffe56aSColin Percival			;;
17348ffe56aSColin Percival		*)
17448ffe56aSColin Percival			return 1
17548ffe56aSColin Percival			;;
17648ffe56aSColin Percival		esac
17748ffe56aSColin Percival	else
17848ffe56aSColin Percival		return 1
17948ffe56aSColin Percival	fi
18048ffe56aSColin Percival}
18148ffe56aSColin Percival
18248ffe56aSColin Percival# Set whether FreeBSD Update is allowed to remove files/directories/symlinks.
18348ffe56aSColin Percivalconfig_AllowDelete () {
18448ffe56aSColin Percival	if [ -z ${ALLOWDELETE} ]; then
18548ffe56aSColin Percival		case $1 in
18648ffe56aSColin Percival		[Yy][Ee][Ss])
18748ffe56aSColin Percival			ALLOWDELETE=yes
18848ffe56aSColin Percival			;;
18948ffe56aSColin Percival		[Nn][Oo])
19048ffe56aSColin Percival			ALLOWDELETE=no
19148ffe56aSColin Percival			;;
19248ffe56aSColin Percival		*)
19348ffe56aSColin Percival			return 1
19448ffe56aSColin Percival			;;
19548ffe56aSColin Percival		esac
19648ffe56aSColin Percival	else
19748ffe56aSColin Percival		return 1
19848ffe56aSColin Percival	fi
19948ffe56aSColin Percival}
20048ffe56aSColin Percival
20148ffe56aSColin Percival# Set whether FreeBSD Update should keep existing inode ownership,
20248ffe56aSColin Percival# permissions, and flags, in the event that they have been modified locally
20348ffe56aSColin Percival# after the release.
20448ffe56aSColin Percivalconfig_KeepModifiedMetadata () {
20548ffe56aSColin Percival	if [ -z ${KEEPMODIFIEDMETADATA} ]; then
20648ffe56aSColin Percival		case $1 in
20748ffe56aSColin Percival		[Yy][Ee][Ss])
20848ffe56aSColin Percival			KEEPMODIFIEDMETADATA=yes
20948ffe56aSColin Percival			;;
21048ffe56aSColin Percival		[Nn][Oo])
21148ffe56aSColin Percival			KEEPMODIFIEDMETADATA=no
21248ffe56aSColin Percival			;;
21348ffe56aSColin Percival		*)
21448ffe56aSColin Percival			return 1
21548ffe56aSColin Percival			;;
21648ffe56aSColin Percival		esac
21748ffe56aSColin Percival	else
21848ffe56aSColin Percival		return 1
21948ffe56aSColin Percival	fi
22048ffe56aSColin Percival}
22148ffe56aSColin Percival
22248ffe56aSColin Percival# Add to the list of components which should be kept updated.
22348ffe56aSColin Percivalconfig_Components () {
22448ffe56aSColin Percival	for C in $@; do
22512294db4SMichael Gmelin		COMPONENTS="${COMPONENTS} ${C}"
22612294db4SMichael Gmelin	done
22712294db4SMichael Gmelin}
22812294db4SMichael Gmelin
22912294db4SMichael Gmelin# Remove src component from list if it isn't installed
23012294db4SMichael Gmelinfinalize_components_config () {
23112294db4SMichael Gmelin	COMPONENTS=""
23212294db4SMichael Gmelin	for C in $@; do
2335a74378cSXin LI		if [ "$C" = "src" ]; then
234cfd9be9cSEd Maste			if [ -e "${BASEDIR}/usr/src/COPYRIGHT" ]; then
23548ffe56aSColin Percival				COMPONENTS="${COMPONENTS} ${C}"
2365a74378cSXin LI			else
2375a74378cSXin LI				echo "src component not installed, skipped"
2385a74378cSXin LI			fi
2395a74378cSXin LI		else
2405a74378cSXin LI			COMPONENTS="${COMPONENTS} ${C}"
2415a74378cSXin LI		fi
24248ffe56aSColin Percival	done
24348ffe56aSColin Percival}
24448ffe56aSColin Percival
24548ffe56aSColin Percival# Add to the list of paths under which updates will be ignored.
24648ffe56aSColin Percivalconfig_IgnorePaths () {
24748ffe56aSColin Percival	for C in $@; do
24848ffe56aSColin Percival		IGNOREPATHS="${IGNOREPATHS} ${C}"
24948ffe56aSColin Percival	done
25048ffe56aSColin Percival}
25148ffe56aSColin Percival
25208e23beeSColin Percival# Add to the list of paths which IDS should ignore.
25308e23beeSColin Percivalconfig_IDSIgnorePaths () {
25408e23beeSColin Percival	for C in $@; do
25508e23beeSColin Percival		IDSIGNOREPATHS="${IDSIGNOREPATHS} ${C}"
25608e23beeSColin Percival	done
25708e23beeSColin Percival}
25808e23beeSColin Percival
25948ffe56aSColin Percival# Add to the list of paths within which updates will be performed only if the
26048ffe56aSColin Percival# file on disk has not been modified locally.
26148ffe56aSColin Percivalconfig_UpdateIfUnmodified () {
26248ffe56aSColin Percival	for C in $@; do
26348ffe56aSColin Percival		UPDATEIFUNMODIFIED="${UPDATEIFUNMODIFIED} ${C}"
26448ffe56aSColin Percival	done
26548ffe56aSColin Percival}
26648ffe56aSColin Percival
267db6b0a61SColin Percival# Add to the list of paths within which updates to text files will be merged
268db6b0a61SColin Percival# instead of overwritten.
269db6b0a61SColin Percivalconfig_MergeChanges () {
270db6b0a61SColin Percival	for C in $@; do
271db6b0a61SColin Percival		MERGECHANGES="${MERGECHANGES} ${C}"
272db6b0a61SColin Percival	done
273db6b0a61SColin Percival}
274db6b0a61SColin Percival
27548ffe56aSColin Percival# Work on a FreeBSD installation mounted under $1
27648ffe56aSColin Percivalconfig_BaseDir () {
27748ffe56aSColin Percival	if [ -z ${BASEDIR} ]; then
27848ffe56aSColin Percival		BASEDIR=$1
27948ffe56aSColin Percival	else
28048ffe56aSColin Percival		return 1
28148ffe56aSColin Percival	fi
28248ffe56aSColin Percival}
28348ffe56aSColin Percival
284db6b0a61SColin Percival# When fetching upgrades, should we assume the user wants exactly the
285db6b0a61SColin Percival# components listed in COMPONENTS, rather than trying to guess based on
286db6b0a61SColin Percival# what's currently installed?
287db6b0a61SColin Percivalconfig_StrictComponents () {
288db6b0a61SColin Percival	if [ -z ${STRICTCOMPONENTS} ]; then
289db6b0a61SColin Percival		case $1 in
290db6b0a61SColin Percival		[Yy][Ee][Ss])
291db6b0a61SColin Percival			STRICTCOMPONENTS=yes
292db6b0a61SColin Percival			;;
293db6b0a61SColin Percival		[Nn][Oo])
294db6b0a61SColin Percival			STRICTCOMPONENTS=no
295db6b0a61SColin Percival			;;
296db6b0a61SColin Percival		*)
297db6b0a61SColin Percival			return 1
298db6b0a61SColin Percival			;;
299db6b0a61SColin Percival		esac
300db6b0a61SColin Percival	else
301db6b0a61SColin Percival		return 1
302db6b0a61SColin Percival	fi
303db6b0a61SColin Percival}
304db6b0a61SColin Percival
305db6b0a61SColin Percival# Upgrade to FreeBSD $1
306db6b0a61SColin Percivalconfig_TargetRelease () {
307db6b0a61SColin Percival	if [ -z ${TARGETRELEASE} ]; then
308db6b0a61SColin Percival		TARGETRELEASE=$1
309db6b0a61SColin Percival	else
310db6b0a61SColin Percival		return 1
311db6b0a61SColin Percival	fi
312d23dc1eeSColin Percival	if echo ${TARGETRELEASE} | grep -qE '^[0-9.]+$'; then
313d23dc1eeSColin Percival		TARGETRELEASE="${TARGETRELEASE}-RELEASE"
314d23dc1eeSColin Percival	fi
315db6b0a61SColin Percival}
316db6b0a61SColin Percival
3170d5c5243SEd Maste# Pretend current release is FreeBSD $1
3180d5c5243SEd Masteconfig_SourceRelease () {
3190d5c5243SEd Maste	UNAME_r=$1
3200d5c5243SEd Maste	if echo ${UNAME_r} | grep -qE '^[0-9.]+$'; then
3210d5c5243SEd Maste		UNAME_r="${UNAME_r}-RELEASE"
3220d5c5243SEd Maste	fi
323b958d4b2SEd Maste	export UNAME_r
3240d5c5243SEd Maste}
3250d5c5243SEd Maste
326c76da1f0SFaraz Vahedi# Get the Jail's path and the version of its installed userland
327c76da1f0SFaraz Vahediconfig_TargetJail () {
328c76da1f0SFaraz Vahedi	JAIL=$1
329c76da1f0SFaraz Vahedi	UNAME_r=$(freebsd-version -j ${JAIL})
330c76da1f0SFaraz Vahedi	BASEDIR=$(jls -j ${JAIL} -h path | awk 'NR == 2 {print}')
331c76da1f0SFaraz Vahedi	if [ -z ${BASEDIR} ] || [ -z ${UNAME_r} ]; then
332c76da1f0SFaraz Vahedi		echo "The specified jail either doesn't exist or" \
333c76da1f0SFaraz Vahedi		      "does not have freebsd-version."
334c76da1f0SFaraz Vahedi		exit 1
335c76da1f0SFaraz Vahedi	fi
336c76da1f0SFaraz Vahedi	export UNAME_r
337c76da1f0SFaraz Vahedi}
338c76da1f0SFaraz Vahedi
33948ffe56aSColin Percival# Define what happens to output of utilities
34048ffe56aSColin Percivalconfig_VerboseLevel () {
34148ffe56aSColin Percival	if [ -z ${VERBOSELEVEL} ]; then
34248ffe56aSColin Percival		case $1 in
34348ffe56aSColin Percival		[Dd][Ee][Bb][Uu][Gg])
34448ffe56aSColin Percival			VERBOSELEVEL=debug
34548ffe56aSColin Percival			;;
34648ffe56aSColin Percival		[Nn][Oo][Ss][Tt][Aa][Tt][Ss])
34748ffe56aSColin Percival			VERBOSELEVEL=nostats
34848ffe56aSColin Percival			;;
34948ffe56aSColin Percival		[Ss][Tt][Aa][Tt][Ss])
35048ffe56aSColin Percival			VERBOSELEVEL=stats
35148ffe56aSColin Percival			;;
35248ffe56aSColin Percival		*)
35348ffe56aSColin Percival			return 1
35448ffe56aSColin Percival			;;
35548ffe56aSColin Percival		esac
35648ffe56aSColin Percival	else
35748ffe56aSColin Percival		return 1
35848ffe56aSColin Percival	fi
35948ffe56aSColin Percival}
36048ffe56aSColin Percival
36123d827efSSimon L. B. Nielsenconfig_BackupKernel () {
36223d827efSSimon L. B. Nielsen	if [ -z ${BACKUPKERNEL} ]; then
36323d827efSSimon L. B. Nielsen		case $1 in
36423d827efSSimon L. B. Nielsen		[Yy][Ee][Ss])
36523d827efSSimon L. B. Nielsen			BACKUPKERNEL=yes
36623d827efSSimon L. B. Nielsen			;;
36723d827efSSimon L. B. Nielsen		[Nn][Oo])
36823d827efSSimon L. B. Nielsen			BACKUPKERNEL=no
36923d827efSSimon L. B. Nielsen			;;
37023d827efSSimon L. B. Nielsen		*)
37123d827efSSimon L. B. Nielsen			return 1
37223d827efSSimon L. B. Nielsen			;;
37323d827efSSimon L. B. Nielsen		esac
37423d827efSSimon L. B. Nielsen	else
37523d827efSSimon L. B. Nielsen		return 1
37623d827efSSimon L. B. Nielsen	fi
37723d827efSSimon L. B. Nielsen}
37823d827efSSimon L. B. Nielsen
37923d827efSSimon L. B. Nielsenconfig_BackupKernelDir () {
38023d827efSSimon L. B. Nielsen	if [ -z ${BACKUPKERNELDIR} ]; then
38123d827efSSimon L. B. Nielsen		if [ -z "$1" ]; then
38223d827efSSimon L. B. Nielsen			echo "BackupKernelDir set to empty dir"
38323d827efSSimon L. B. Nielsen			return 1
38423d827efSSimon L. B. Nielsen		fi
38523d827efSSimon L. B. Nielsen
38623d827efSSimon L. B. Nielsen		# We check for some paths which would be extremely odd
38723d827efSSimon L. B. Nielsen		# to use, but which could cause a lot of problems if
38823d827efSSimon L. B. Nielsen		# used.
38923d827efSSimon L. B. Nielsen		case $1 in
39023d827efSSimon L. B. Nielsen		/|/bin|/boot|/etc|/lib|/libexec|/sbin|/usr|/var)
39123d827efSSimon L. B. Nielsen			echo "BackupKernelDir set to invalid path $1"
39223d827efSSimon L. B. Nielsen			return 1
39323d827efSSimon L. B. Nielsen			;;
39423d827efSSimon L. B. Nielsen		/*)
39523d827efSSimon L. B. Nielsen			BACKUPKERNELDIR=$1
39623d827efSSimon L. B. Nielsen			;;
39723d827efSSimon L. B. Nielsen		*)
39823d827efSSimon L. B. Nielsen			echo "BackupKernelDir ($1) is not an absolute path"
39923d827efSSimon L. B. Nielsen			return 1
40023d827efSSimon L. B. Nielsen			;;
40123d827efSSimon L. B. Nielsen		esac
40223d827efSSimon L. B. Nielsen	else
40323d827efSSimon L. B. Nielsen		return 1
40423d827efSSimon L. B. Nielsen	fi
40523d827efSSimon L. B. Nielsen}
40623d827efSSimon L. B. Nielsen
40723d827efSSimon L. B. Nielsenconfig_BackupKernelSymbolFiles () {
40823d827efSSimon L. B. Nielsen	if [ -z ${BACKUPKERNELSYMBOLFILES} ]; then
40923d827efSSimon L. B. Nielsen		case $1 in
41023d827efSSimon L. B. Nielsen		[Yy][Ee][Ss])
41123d827efSSimon L. B. Nielsen			BACKUPKERNELSYMBOLFILES=yes
41223d827efSSimon L. B. Nielsen			;;
41323d827efSSimon L. B. Nielsen		[Nn][Oo])
41423d827efSSimon L. B. Nielsen			BACKUPKERNELSYMBOLFILES=no
41523d827efSSimon L. B. Nielsen			;;
41623d827efSSimon L. B. Nielsen		*)
41723d827efSSimon L. B. Nielsen			return 1
41823d827efSSimon L. B. Nielsen			;;
41923d827efSSimon L. B. Nielsen		esac
42023d827efSSimon L. B. Nielsen	else
42123d827efSSimon L. B. Nielsen		return 1
42223d827efSSimon L. B. Nielsen	fi
42323d827efSSimon L. B. Nielsen}
42423d827efSSimon L. B. Nielsen
425f28f1389SDave Fullardconfig_CreateBootEnv () {
426f28f1389SDave Fullard	if [ -z ${BOOTENV} ]; then
427f28f1389SDave Fullard		case $1 in
428f28f1389SDave Fullard		[Yy][Ee][Ss])
429f28f1389SDave Fullard			BOOTENV=yes
430f28f1389SDave Fullard			;;
431f28f1389SDave Fullard		[Nn][Oo])
432f28f1389SDave Fullard			BOOTENV=no
433f28f1389SDave Fullard			;;
434f28f1389SDave Fullard		*)
435f28f1389SDave Fullard			return 1
436f28f1389SDave Fullard			;;
437f28f1389SDave Fullard		esac
438f28f1389SDave Fullard	else
439f28f1389SDave Fullard		return 1
440f28f1389SDave Fullard	fi
441f28f1389SDave Fullard}
44248ffe56aSColin Percival# Handle one line of configuration
44348ffe56aSColin Percivalconfigline () {
44448ffe56aSColin Percival	if [ $# -eq 0 ]; then
44548ffe56aSColin Percival		return
44648ffe56aSColin Percival	fi
44748ffe56aSColin Percival
44848ffe56aSColin Percival	OPT=$1
44948ffe56aSColin Percival	shift
45048ffe56aSColin Percival	config_${OPT} $@
45148ffe56aSColin Percival}
45248ffe56aSColin Percival
45348ffe56aSColin Percival#### Parameter handling functions.
45448ffe56aSColin Percival
45548ffe56aSColin Percival# Initialize parameters to null, just in case they're
45648ffe56aSColin Percival# set in the environment.
45748ffe56aSColin Percivalinit_params () {
45848ffe56aSColin Percival	# Configration settings
45948ffe56aSColin Percival	nullconfig
46048ffe56aSColin Percival
46148ffe56aSColin Percival	# No configuration file set yet
46248ffe56aSColin Percival	CONFFILE=""
46348ffe56aSColin Percival
46448ffe56aSColin Percival	# No commands specified yet
46548ffe56aSColin Percival	COMMANDS=""
4668935f242SAllan Jude
4678935f242SAllan Jude	# Force fetch to proceed
4688935f242SAllan Jude	FORCEFETCH=0
4698935f242SAllan Jude
4708935f242SAllan Jude	# Run without a TTY
4718935f242SAllan Jude	NOTTYOK=0
47233bd05c3SGuangyuan Yang
47333bd05c3SGuangyuan Yang	# Fetched first in a chain of commands
47433bd05c3SGuangyuan Yang	ISFETCHED=0
47548ffe56aSColin Percival}
47648ffe56aSColin Percival
47748ffe56aSColin Percival# Parse the command line
47848ffe56aSColin Percivalparse_cmdline () {
47948ffe56aSColin Percival	while [ $# -gt 0 ]; do
48048ffe56aSColin Percival		case "$1" in
48148ffe56aSColin Percival		# Location of configuration file
48248ffe56aSColin Percival		-f)
48348ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi
48448ffe56aSColin Percival			if [ ! -z "${CONFFILE}" ]; then usage; fi
48548ffe56aSColin Percival			shift; CONFFILE="$1"
48648ffe56aSColin Percival			;;
4878935f242SAllan Jude		-F)
4888935f242SAllan Jude			FORCEFETCH=1
4898935f242SAllan Jude			;;
4908935f242SAllan Jude		--not-running-from-cron)
4918935f242SAllan Jude			NOTTYOK=1
4928935f242SAllan Jude			;;
493b39ce43eSColin Percival		--currently-running)
4940d5c5243SEd Maste			shift
4950d5c5243SEd Maste			config_SourceRelease $1 || usage
496b39ce43eSColin Percival			;;
49748ffe56aSColin Percival
49848ffe56aSColin Percival		# Configuration file equivalents
49948ffe56aSColin Percival		-b)
50048ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
50148ffe56aSColin Percival			config_BaseDir $1 || usage
50248ffe56aSColin Percival			;;
50348ffe56aSColin Percival		-d)
50448ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
50548ffe56aSColin Percival			config_WorkDir $1 || usage
50648ffe56aSColin Percival			;;
507c76da1f0SFaraz Vahedi		-j)
508c76da1f0SFaraz Vahedi			if [ $# -eq 1 ]; then usage; fi; shift
509c76da1f0SFaraz Vahedi			config_TargetJail $1 || usage
510c76da1f0SFaraz Vahedi			;;
51148ffe56aSColin Percival		-k)
51248ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
51348ffe56aSColin Percival			config_KeyPrint $1 || usage
51448ffe56aSColin Percival			;;
51548ffe56aSColin Percival		-s)
51648ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
51748ffe56aSColin Percival			config_ServerName $1 || usage
51848ffe56aSColin Percival			;;
519db6b0a61SColin Percival		-r)
520db6b0a61SColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
521db6b0a61SColin Percival			config_TargetRelease $1 || usage
522db6b0a61SColin Percival			;;
52348ffe56aSColin Percival		-t)
52448ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
52548ffe56aSColin Percival			config_MailTo $1 || usage
52648ffe56aSColin Percival			;;
52748ffe56aSColin Percival		-v)
52848ffe56aSColin Percival			if [ $# -eq 1 ]; then usage; fi; shift
52948ffe56aSColin Percival			config_VerboseLevel $1 || usage
53048ffe56aSColin Percival			;;
53148ffe56aSColin Percival
53248ffe56aSColin Percival		# Aliases for "-v debug" and "-v nostats"
53348ffe56aSColin Percival		--debug)
53448ffe56aSColin Percival			config_VerboseLevel debug || usage
53548ffe56aSColin Percival			;;
53648ffe56aSColin Percival		--no-stats)
53748ffe56aSColin Percival			config_VerboseLevel nostats || usage
53848ffe56aSColin Percival			;;
53948ffe56aSColin Percival
54048ffe56aSColin Percival		# Commands
5418cfda118SMichael Gmelin		cron | fetch | upgrade | updatesready | install | rollback |\
5428cfda118SMichael Gmelin		IDS | showconfig)
54348ffe56aSColin Percival			COMMANDS="${COMMANDS} $1"
54448ffe56aSColin Percival			;;
54548ffe56aSColin Percival
54648ffe56aSColin Percival		# Anything else is an error
54748ffe56aSColin Percival		*)
54848ffe56aSColin Percival			usage
54948ffe56aSColin Percival			;;
55048ffe56aSColin Percival		esac
55148ffe56aSColin Percival		shift
55248ffe56aSColin Percival	done
55348ffe56aSColin Percival
55448ffe56aSColin Percival	# Make sure we have at least one command
55548ffe56aSColin Percival	if [ -z "${COMMANDS}" ]; then
55648ffe56aSColin Percival		usage
55748ffe56aSColin Percival	fi
55848ffe56aSColin Percival}
55948ffe56aSColin Percival
56048ffe56aSColin Percival# Parse the configuration file
56148ffe56aSColin Percivalparse_conffile () {
56248ffe56aSColin Percival	# If a configuration file was specified on the command line, check
56348ffe56aSColin Percival	# that it exists and is readable.
56448ffe56aSColin Percival	if [ ! -z "${CONFFILE}" ] && [ ! -r "${CONFFILE}" ]; then
56548ffe56aSColin Percival		echo -n "File does not exist "
56648ffe56aSColin Percival		echo -n "or is not readable: "
56748ffe56aSColin Percival		echo ${CONFFILE}
56848ffe56aSColin Percival		exit 1
56948ffe56aSColin Percival	fi
57048ffe56aSColin Percival
57148ffe56aSColin Percival	# If a configuration file was not specified on the command line,
57248ffe56aSColin Percival	# use the default configuration file path.  If that default does
57348ffe56aSColin Percival	# not exist, give up looking for any configuration.
57448ffe56aSColin Percival	if [ -z "${CONFFILE}" ]; then
57548ffe56aSColin Percival		CONFFILE="/etc/freebsd-update.conf"
57648ffe56aSColin Percival		if [ ! -r "${CONFFILE}" ]; then
57748ffe56aSColin Percival			return
57848ffe56aSColin Percival		fi
57948ffe56aSColin Percival	fi
58048ffe56aSColin Percival
58148ffe56aSColin Percival	# Save the configuration options specified on the command line, and
58248ffe56aSColin Percival	# clear all the options in preparation for reading the config file.
58348ffe56aSColin Percival	saveconfig
58448ffe56aSColin Percival	nullconfig
58548ffe56aSColin Percival
58648ffe56aSColin Percival	# Read the configuration file.  Anything after the first '#' is
58748ffe56aSColin Percival	# ignored, and any blank lines are ignored.
58848ffe56aSColin Percival	L=0
58948ffe56aSColin Percival	while read LINE; do
59048ffe56aSColin Percival		L=$(($L + 1))
59148ffe56aSColin Percival		LINEX=`echo "${LINE}" | cut -f 1 -d '#'`
59248ffe56aSColin Percival		if ! configline ${LINEX}; then
59348ffe56aSColin Percival			echo "Error processing configuration file, line $L:"
59448ffe56aSColin Percival			echo "==> ${LINE}"
59548ffe56aSColin Percival			exit 1
59648ffe56aSColin Percival		fi
59748ffe56aSColin Percival	done < ${CONFFILE}
59848ffe56aSColin Percival
59948ffe56aSColin Percival	# Merge the settings read from the configuration file with those
60048ffe56aSColin Percival	# provided at the command line.
60148ffe56aSColin Percival	mergeconfig
60248ffe56aSColin Percival}
60348ffe56aSColin Percival
60448ffe56aSColin Percival# Provide some default parameters
60548ffe56aSColin Percivaldefault_params () {
60648ffe56aSColin Percival	# Save any parameters already configured, and clear the slate
60748ffe56aSColin Percival	saveconfig
60848ffe56aSColin Percival	nullconfig
60948ffe56aSColin Percival
61048ffe56aSColin Percival	# Default configurations
61148ffe56aSColin Percival	config_WorkDir /var/db/freebsd-update
61248ffe56aSColin Percival	config_MailTo root
61348ffe56aSColin Percival	config_AllowAdd yes
61448ffe56aSColin Percival	config_AllowDelete yes
61548ffe56aSColin Percival	config_KeepModifiedMetadata yes
61648ffe56aSColin Percival	config_BaseDir /
61748ffe56aSColin Percival	config_VerboseLevel stats
618db6b0a61SColin Percival	config_StrictComponents no
61923d827efSSimon L. B. Nielsen	config_BackupKernel yes
62023d827efSSimon L. B. Nielsen	config_BackupKernelDir /boot/kernel.old
62123d827efSSimon L. B. Nielsen	config_BackupKernelSymbolFiles no
622f28f1389SDave Fullard	config_CreateBootEnv yes
62348ffe56aSColin Percival
62448ffe56aSColin Percival	# Merge these defaults into the earlier-configured settings
62548ffe56aSColin Percival	mergeconfig
62648ffe56aSColin Percival}
62748ffe56aSColin Percival
62848ffe56aSColin Percival# Set utility output filtering options, based on ${VERBOSELEVEL}
62948ffe56aSColin Percivalfetch_setup_verboselevel () {
63048ffe56aSColin Percival	case ${VERBOSELEVEL} in
63148ffe56aSColin Percival	debug)
63248ffe56aSColin Percival		QUIETREDIR="/dev/stderr"
63348ffe56aSColin Percival		QUIETFLAG=" "
63448ffe56aSColin Percival		STATSREDIR="/dev/stderr"
63548ffe56aSColin Percival		DDSTATS=".."
63648ffe56aSColin Percival		XARGST="-t"
63748ffe56aSColin Percival		NDEBUG=" "
63848ffe56aSColin Percival		;;
63948ffe56aSColin Percival	nostats)
64048ffe56aSColin Percival		QUIETREDIR=""
64148ffe56aSColin Percival		QUIETFLAG=""
64248ffe56aSColin Percival		STATSREDIR="/dev/null"
64348ffe56aSColin Percival		DDSTATS=".."
64448ffe56aSColin Percival		XARGST=""
64548ffe56aSColin Percival		NDEBUG=""
64648ffe56aSColin Percival		;;
64748ffe56aSColin Percival	stats)
64848ffe56aSColin Percival		QUIETREDIR="/dev/null"
64948ffe56aSColin Percival		QUIETFLAG="-q"
65048ffe56aSColin Percival		STATSREDIR="/dev/stdout"
65148ffe56aSColin Percival		DDSTATS=""
65248ffe56aSColin Percival		XARGST=""
65348ffe56aSColin Percival		NDEBUG="-n"
65448ffe56aSColin Percival		;;
65548ffe56aSColin Percival	esac
65648ffe56aSColin Percival}
65748ffe56aSColin Percival
658*bc0c6c9cSFernando Apesteguía# Check if there are any kernel modules installed from ports.
659*bc0c6c9cSFernando Apesteguía# In that case warn the user that a rebuild from ports (i.e. not from
660*bc0c6c9cSFernando Apesteguía# packages) might need necessary for the modules to work in the new release.
661*bc0c6c9cSFernando Apesteguíaupgrade_check_kmod_ports() {
662*bc0c6c9cSFernando Apesteguía	local mod_name
663*bc0c6c9cSFernando Apesteguía	local modules
664*bc0c6c9cSFernando Apesteguía	local pattern
665*bc0c6c9cSFernando Apesteguía	local pkg_name
666*bc0c6c9cSFernando Apesteguía	local port_name
667*bc0c6c9cSFernando Apesteguía	local report
668*bc0c6c9cSFernando Apesteguía	local w
669*bc0c6c9cSFernando Apesteguía
670*bc0c6c9cSFernando Apesteguía	if ! command -v pkg >/dev/null; then
671*bc0c6c9cSFernando Apesteguía		echo "Skipping kernel modules check. pkg(8) not present."
672*bc0c6c9cSFernando Apesteguía		return
673*bc0c6c9cSFernando Apesteguía	fi
674*bc0c6c9cSFernando Apesteguía
675*bc0c6c9cSFernando Apesteguía	# Most modules are in /boot/modules but we should actually look
676*bc0c6c9cSFernando Apesteguía	# in every path configured in module_path
677*bc0c6c9cSFernando Apesteguía	search_files="/boot/defaults/loader.conf /boot/loader.conf"
678*bc0c6c9cSFernando Apesteguía	pattern=$(grep -shE '^module_path=' ${search_files} |
679*bc0c6c9cSFernando Apesteguía		tail -1 |
680*bc0c6c9cSFernando Apesteguía		cut -f2 -d\" |
681*bc0c6c9cSFernando Apesteguía		tr ";" "|")
682*bc0c6c9cSFernando Apesteguía
683*bc0c6c9cSFernando Apesteguía	if [ -z "${pattern}" ]; then
684*bc0c6c9cSFernando Apesteguía		# Not having module_path in loader.conf is probably an error.
685*bc0c6c9cSFernando Apesteguía		# Check at least the most common path
686*bc0c6c9cSFernando Apesteguía		pattern="/boot/modules"
687*bc0c6c9cSFernando Apesteguía	fi
688*bc0c6c9cSFernando Apesteguía
689*bc0c6c9cSFernando Apesteguía	# Check the pkg database for modules installed in those directories
690*bc0c6c9cSFernando Apesteguía	modules=$(pkg query '%Fp' | grep -E "${pattern}")
691*bc0c6c9cSFernando Apesteguía
692*bc0c6c9cSFernando Apesteguía	if [ -z "${modules}" ]; then
693*bc0c6c9cSFernando Apesteguía		return
694*bc0c6c9cSFernando Apesteguía	fi
695*bc0c6c9cSFernando Apesteguía
696*bc0c6c9cSFernando Apesteguía	echo -e "\n"
697*bc0c6c9cSFernando Apesteguía	echo "The following modules have been installed from packages."
698*bc0c6c9cSFernando Apesteguía	echo "As a consequence they might not work when performing a major or minor upgrade."
699*bc0c6c9cSFernando Apesteguía	echo -e "It is advised to rebuild these ports:\n"
700*bc0c6c9cSFernando Apesteguía
701*bc0c6c9cSFernando Apesteguía
702*bc0c6c9cSFernando Apesteguía	report="Module Package Port\n------ ------- ----\n"
703*bc0c6c9cSFernando Apesteguía	for module in ${modules}; do
704*bc0c6c9cSFernando Apesteguía		w=$(pkg which "${module}")
705*bc0c6c9cSFernando Apesteguía		mod_name=$(echo "${w}" | awk '{print $1;}')
706*bc0c6c9cSFernando Apesteguía		pkg_name=$(echo "${w}" | awk '{print $6;}')
707*bc0c6c9cSFernando Apesteguía		port_name=$(pkg info -o "${pkg_name}" | awk '{print $2;}')
708*bc0c6c9cSFernando Apesteguía		report="${report}${mod_name} ${pkg_name} ${port_name}\n"
709*bc0c6c9cSFernando Apesteguía	done
710*bc0c6c9cSFernando Apesteguía
711*bc0c6c9cSFernando Apesteguía	echo -e "${report}" | column -t
712*bc0c6c9cSFernando Apesteguía	echo -e "\n"
713*bc0c6c9cSFernando Apesteguía}
714*bc0c6c9cSFernando Apesteguía
71548ffe56aSColin Percival# Perform sanity checks and set some final parameters
71648ffe56aSColin Percival# in preparation for fetching files.  Figure out which
71748ffe56aSColin Percival# set of updates should be downloaded: If the user is
71848ffe56aSColin Percival# running *-p[0-9]+, strip off the last part; if the
71948ffe56aSColin Percival# user is running -SECURITY, call it -RELEASE.  Chdir
72048ffe56aSColin Percival# into the working directory.
721211f2ba0SColin Percivalfetchupgrade_check_params () {
72248ffe56aSColin Percival	export HTTP_USER_AGENT="freebsd-update (${COMMAND}, `uname -r`)"
72348ffe56aSColin Percival
72448ffe56aSColin Percival	_SERVERNAME_z=\
72548ffe56aSColin Percival"SERVERNAME must be given via command line or configuration file."
72648ffe56aSColin Percival	_KEYPRINT_z="Key must be given via -k option or configuration file."
72748ffe56aSColin Percival	_KEYPRINT_bad="Invalid key fingerprint: "
72848ffe56aSColin Percival	_WORKDIR_bad="Directory does not exist or is not writable: "
729f88076f0SMark Felder	_WORKDIR_bad2="Directory is not on a persistent filesystem: "
73048ffe56aSColin Percival
73148ffe56aSColin Percival	if [ -z "${SERVERNAME}" ]; then
73248ffe56aSColin Percival		echo -n "`basename $0`: "
73348ffe56aSColin Percival		echo "${_SERVERNAME_z}"
73448ffe56aSColin Percival		exit 1
73548ffe56aSColin Percival	fi
73648ffe56aSColin Percival	if [ -z "${KEYPRINT}" ]; then
73748ffe56aSColin Percival		echo -n "`basename $0`: "
73848ffe56aSColin Percival		echo "${_KEYPRINT_z}"
73948ffe56aSColin Percival		exit 1
74048ffe56aSColin Percival	fi
74148ffe56aSColin Percival	if ! echo "${KEYPRINT}" | grep -qE "^[0-9a-f]{64}$"; then
74248ffe56aSColin Percival		echo -n "`basename $0`: "
74348ffe56aSColin Percival		echo -n "${_KEYPRINT_bad}"
74448ffe56aSColin Percival		echo ${KEYPRINT}
74548ffe56aSColin Percival		exit 1
74648ffe56aSColin Percival	fi
74748ffe56aSColin Percival	if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then
74848ffe56aSColin Percival		echo -n "`basename $0`: "
74948ffe56aSColin Percival		echo -n "${_WORKDIR_bad}"
75048ffe56aSColin Percival		echo ${WORKDIR}
75148ffe56aSColin Percival		exit 1
75248ffe56aSColin Percival	fi
753dfe9215bSMark Felder	case `df -T ${WORKDIR}` in */dev/md[0-9]* | *tmpfs*)
754f88076f0SMark Felder		echo -n "`basename $0`: "
755f88076f0SMark Felder		echo -n "${_WORKDIR_bad2}"
756f88076f0SMark Felder		echo ${WORKDIR}
757f88076f0SMark Felder		exit 1
758dfe9215bSMark Felder		;;
759dfe9215bSMark Felder	esac
760a2356430SColin Percival	chmod 700 ${WORKDIR}
76148ffe56aSColin Percival	cd ${WORKDIR} || exit 1
76248ffe56aSColin Percival
76348ffe56aSColin Percival	# Generate release number.  The s/SECURITY/RELEASE/ bit exists
76448ffe56aSColin Percival	# to provide an upgrade path for FreeBSD Update 1.x users, since
76548ffe56aSColin Percival	# the kernels provided by FreeBSD Update 1.x are always labelled
76648ffe56aSColin Percival	# as X.Y-SECURITY.
76748ffe56aSColin Percival	RELNUM=`uname -r |
76848ffe56aSColin Percival	    sed -E 's,-p[0-9]+,,' |
76948ffe56aSColin Percival	    sed -E 's,-SECURITY,-RELEASE,'`
77048ffe56aSColin Percival	ARCH=`uname -m`
77148ffe56aSColin Percival	FETCHDIR=${RELNUM}/${ARCH}
772db6b0a61SColin Percival	PATCHDIR=${RELNUM}/${ARCH}/bp
77348ffe56aSColin Percival
774d308a8bfSEd Maste	# Disallow upgrade from a version that is not a release
775d308a8bfSEd Maste	case ${RELNUM} in
776d308a8bfSEd Maste	*-RELEASE | *-ALPHA*  | *-BETA* | *-RC*)
777d308a8bfSEd Maste		;;
778d308a8bfSEd Maste	*)
7790d5c5243SEd Maste		echo -n "`basename $0`: "
7800d5c5243SEd Maste		cat <<- EOF
781d308a8bfSEd Maste			Cannot upgrade from a version that is not a release
782d308a8bfSEd Maste			(including alpha, beta and release candidates)
783d308a8bfSEd Maste			using `basename $0`. Instead, FreeBSD can be directly
784d308a8bfSEd Maste			upgraded by source or upgraded to a RELEASE/RELENG version
785d308a8bfSEd Maste			prior to running `basename $0`.
786d308a8bfSEd Maste			Currently running: ${RELNUM}
7870d5c5243SEd Maste		EOF
7880d5c5243SEd Maste		exit 1
789d308a8bfSEd Maste		;;
790d308a8bfSEd Maste	esac
7910d5c5243SEd Maste
79248ffe56aSColin Percival	# Figure out what directory contains the running kernel
79348ffe56aSColin Percival	BOOTFILE=`sysctl -n kern.bootfile`
79448ffe56aSColin Percival	KERNELDIR=${BOOTFILE%/kernel}
79548ffe56aSColin Percival	if ! [ -d ${KERNELDIR} ]; then
79648ffe56aSColin Percival		echo "Cannot identify running kernel"
79748ffe56aSColin Percival		exit 1
79848ffe56aSColin Percival	fi
79948ffe56aSColin Percival
8002c434b2cSColin Percival	# Figure out what kernel configuration is running.  We start with
8012c434b2cSColin Percival	# the output of `uname -i`, and then make the following adjustments:
8022c434b2cSColin Percival	# 1. Replace "SMP-GENERIC" with "SMP".  Why the SMP kernel config
8032c434b2cSColin Percival	# file says "ident SMP-GENERIC", I don't know...
8042c434b2cSColin Percival	# 2. If the kernel claims to be GENERIC _and_ ${ARCH} is "amd64"
8052c434b2cSColin Percival	# _and_ `sysctl kern.version` contains a line which ends "/SMP", then
8062c434b2cSColin Percival	# we're running an SMP kernel.  This mis-identification is a bug
8072c434b2cSColin Percival	# which was fixed in 6.2-STABLE.
8082c434b2cSColin Percival	KERNCONF=`uname -i`
8092c434b2cSColin Percival	if [ ${KERNCONF} = "SMP-GENERIC" ]; then
8102c434b2cSColin Percival		KERNCONF=SMP
8112c434b2cSColin Percival	fi
8122c434b2cSColin Percival	if [ ${KERNCONF} = "GENERIC" ] && [ ${ARCH} = "amd64" ]; then
8132c434b2cSColin Percival		if sysctl kern.version | grep -qE '/SMP$'; then
8142c434b2cSColin Percival			KERNCONF=SMP
8152c434b2cSColin Percival		fi
8162c434b2cSColin Percival	fi
8172c434b2cSColin Percival
81848ffe56aSColin Percival	# Define some paths
81948ffe56aSColin Percival	BSPATCH=/usr/bin/bspatch
82048ffe56aSColin Percival	SHA256=/sbin/sha256
82148ffe56aSColin Percival	PHTTPGET=/usr/libexec/phttpget
82248ffe56aSColin Percival
82348ffe56aSColin Percival	# Set up variables relating to VERBOSELEVEL
82448ffe56aSColin Percival	fetch_setup_verboselevel
82548ffe56aSColin Percival
82648ffe56aSColin Percival	# Construct a unique name from ${BASEDIR}
82748ffe56aSColin Percival	BDHASH=`echo ${BASEDIR} | sha256 -q`
82848ffe56aSColin Percival}
82948ffe56aSColin Percival
830211f2ba0SColin Percival# Perform sanity checks etc. before fetching updates.
831211f2ba0SColin Percivalfetch_check_params () {
832211f2ba0SColin Percival	fetchupgrade_check_params
833211f2ba0SColin Percival
834211f2ba0SColin Percival	if ! [ -z "${TARGETRELEASE}" ]; then
835211f2ba0SColin Percival		echo -n "`basename $0`: "
83659b02bb4SMichael Osipov		echo -n "'-r' option is meaningless with 'fetch' command.  "
837211f2ba0SColin Percival		echo "(Did you mean 'upgrade' instead?)"
838211f2ba0SColin Percival		exit 1
839211f2ba0SColin Percival	fi
8408935f242SAllan Jude
8418935f242SAllan Jude	# Check that we have updates ready to install
8428bf2dcceSAllan Jude	if [ -f ${BDHASH}-install/kerneldone -a $FORCEFETCH -eq 0 ]; then
8438935f242SAllan Jude		echo "You have a partially completed upgrade pending"
84459b02bb4SMichael Osipov		echo "Run '`basename $0` [options] install' first."
84559b02bb4SMichael Osipov		echo "Run '`basename $0` [options] fetch -F' to proceed anyway."
8468935f242SAllan Jude		exit 1
8478935f242SAllan Jude	fi
848211f2ba0SColin Percival}
849211f2ba0SColin Percival
850db6b0a61SColin Percival# Perform sanity checks etc. before fetching upgrades.
851db6b0a61SColin Percivalupgrade_check_params () {
852211f2ba0SColin Percival	fetchupgrade_check_params
853db6b0a61SColin Percival
854db6b0a61SColin Percival	# Unless set otherwise, we're upgrading to the same kernel config.
855db6b0a61SColin Percival	NKERNCONF=${KERNCONF}
856db6b0a61SColin Percival
857db6b0a61SColin Percival	# We need TARGETRELEASE set
85859b02bb4SMichael Osipov	_TARGETRELEASE_z="Release target must be specified via '-r' option."
859db6b0a61SColin Percival	if [ -z "${TARGETRELEASE}" ]; then
860db6b0a61SColin Percival		echo -n "`basename $0`: "
861db6b0a61SColin Percival		echo "${_TARGETRELEASE_z}"
862db6b0a61SColin Percival		exit 1
863db6b0a61SColin Percival	fi
864db6b0a61SColin Percival
865db6b0a61SColin Percival	# The target release should be != the current release.
866db6b0a61SColin Percival	if [ "${TARGETRELEASE}" = "${RELNUM}" ]; then
867db6b0a61SColin Percival		echo -n "`basename $0`: "
868db6b0a61SColin Percival		echo "Cannot upgrade from ${RELNUM} to itself"
869db6b0a61SColin Percival		exit 1
870db6b0a61SColin Percival	fi
871db6b0a61SColin Percival
872db6b0a61SColin Percival	# Turning off AllowAdd or AllowDelete is a bad idea for upgrades.
873db6b0a61SColin Percival	if [ "${ALLOWADD}" = "no" ]; then
874db6b0a61SColin Percival		echo -n "`basename $0`: "
875db6b0a61SColin Percival		echo -n "WARNING: \"AllowAdd no\" is a bad idea "
876db6b0a61SColin Percival		echo "when upgrading between releases."
877db6b0a61SColin Percival		echo
878db6b0a61SColin Percival	fi
879db6b0a61SColin Percival	if [ "${ALLOWDELETE}" = "no" ]; then
880db6b0a61SColin Percival		echo -n "`basename $0`: "
881db6b0a61SColin Percival		echo -n "WARNING: \"AllowDelete no\" is a bad idea "
882db6b0a61SColin Percival		echo "when upgrading between releases."
883db6b0a61SColin Percival		echo
884db6b0a61SColin Percival	fi
885db6b0a61SColin Percival
886db6b0a61SColin Percival	# Set EDITOR to /usr/bin/vi if it isn't already set
887db6b0a61SColin Percival	: ${EDITOR:='/usr/bin/vi'}
888db6b0a61SColin Percival}
889db6b0a61SColin Percival
89048ffe56aSColin Percival# Perform sanity checks and set some final parameters in
89148ffe56aSColin Percival# preparation for installing updates.
89248ffe56aSColin Percivalinstall_check_params () {
89348ffe56aSColin Percival	# Check that we are root.  All sorts of things won't work otherwise.
89448ffe56aSColin Percival	if [ `id -u` != 0 ]; then
89548ffe56aSColin Percival		echo "You must be root to run this."
89648ffe56aSColin Percival		exit 1
89748ffe56aSColin Percival	fi
89848ffe56aSColin Percival
8992328d598SColin Percival	# Check that securelevel <= 0.  Otherwise we can't update schg files.
9002328d598SColin Percival	if [ `sysctl -n kern.securelevel` -gt 0 ]; then
9012328d598SColin Percival		echo "Updates cannot be installed when the system securelevel"
9022328d598SColin Percival		echo "is greater than zero."
9032328d598SColin Percival		exit 1
9042328d598SColin Percival	fi
9052328d598SColin Percival
90648ffe56aSColin Percival	# Check that we have a working directory
90748ffe56aSColin Percival	_WORKDIR_bad="Directory does not exist or is not writable: "
90848ffe56aSColin Percival	if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then
90948ffe56aSColin Percival		echo -n "`basename $0`: "
91048ffe56aSColin Percival		echo -n "${_WORKDIR_bad}"
91148ffe56aSColin Percival		echo ${WORKDIR}
91248ffe56aSColin Percival		exit 1
91348ffe56aSColin Percival	fi
91448ffe56aSColin Percival	cd ${WORKDIR} || exit 1
91548ffe56aSColin Percival
91648ffe56aSColin Percival	# Construct a unique name from ${BASEDIR}
91748ffe56aSColin Percival	BDHASH=`echo ${BASEDIR} | sha256 -q`
91848ffe56aSColin Percival
91948ffe56aSColin Percival	# Check that we have updates ready to install
92048ffe56aSColin Percival	if ! [ -L ${BDHASH}-install ]; then
92148ffe56aSColin Percival		echo "No updates are available to install."
92233bd05c3SGuangyuan Yang		if [ $ISFETCHED -eq 0 ]; then
92359b02bb4SMichael Osipov			echo "Run '`basename $0` [options] fetch' first."
9248cfda118SMichael Gmelin			exit 2
92533bd05c3SGuangyuan Yang		fi
92633bd05c3SGuangyuan Yang		exit 0
92748ffe56aSColin Percival	fi
92848ffe56aSColin Percival	if ! [ -f ${BDHASH}-install/INDEX-OLD ] ||
92948ffe56aSColin Percival	    ! [ -f ${BDHASH}-install/INDEX-NEW ]; then
93048ffe56aSColin Percival		echo "Update manifest is corrupt -- this should never happen."
93159b02bb4SMichael Osipov		echo "Re-run '`basename $0` [options] fetch'."
93248ffe56aSColin Percival		exit 1
93348ffe56aSColin Percival	fi
93423d827efSSimon L. B. Nielsen
93523d827efSSimon L. B. Nielsen	# Figure out what directory contains the running kernel
93623d827efSSimon L. B. Nielsen	BOOTFILE=`sysctl -n kern.bootfile`
93723d827efSSimon L. B. Nielsen	KERNELDIR=${BOOTFILE%/kernel}
93823d827efSSimon L. B. Nielsen	if ! [ -d ${KERNELDIR} ]; then
93923d827efSSimon L. B. Nielsen		echo "Cannot identify running kernel"
94023d827efSSimon L. B. Nielsen		exit 1
94123d827efSSimon L. B. Nielsen	fi
94248ffe56aSColin Percival}
94348ffe56aSColin Percival
944f28f1389SDave Fullard# Creates a new boot environment
945f28f1389SDave Fullardinstall_create_be () {
946f28f1389SDave Fullard	# Figure out if we're running in a jail and return if we are
947f28f1389SDave Fullard	if [ `sysctl -n security.jail.jailed` = 1 ]; then
948f28f1389SDave Fullard		return 1
949f28f1389SDave Fullard	fi
950e01e8f91SKyle Evans	# Operating on roots that aren't located at / will, more often than not,
951e01e8f91SKyle Evans	# not touch the boot environment.
952e01e8f91SKyle Evans	if [ "$BASEDIR" != "/" ]; then
953e01e8f91SKyle Evans		return 1
954e01e8f91SKyle Evans	fi
955f28f1389SDave Fullard	# Create a boot environment if enabled
956f28f1389SDave Fullard	if [ ${BOOTENV} = yes ]; then
957f28f1389SDave Fullard		bectl check 2>/dev/null
958f28f1389SDave Fullard		case $? in
959f28f1389SDave Fullard			0)
960f28f1389SDave Fullard				# Boot environment are supported
961f28f1389SDave Fullard				CREATEBE=yes
962f28f1389SDave Fullard				;;
963f28f1389SDave Fullard			255)
964f28f1389SDave Fullard				# Boot environments are not supported
965f28f1389SDave Fullard				CREATEBE=no
966f28f1389SDave Fullard				;;
967f28f1389SDave Fullard			*)
968f28f1389SDave Fullard				# If bectl returns an unexpected exit code, don't create a BE
969f28f1389SDave Fullard				CREATEBE=no
970f28f1389SDave Fullard				;;
971f28f1389SDave Fullard		esac
972f28f1389SDave Fullard		if [ ${CREATEBE} = yes ]; then
973f28f1389SDave Fullard			echo -n "Creating snapshot of existing boot environment... "
974e01e8f91SKyle Evans			VERSION=`freebsd-version -ku | sort -V | tail -n 1`
975f28f1389SDave Fullard			TIMESTAMP=`date +"%Y-%m-%d_%H%M%S"`
976989c5f6dSKyle Evans			bectl create -r ${VERSION}_${TIMESTAMP}
977f28f1389SDave Fullard			if [ $? -eq 0 ]; then
978f28f1389SDave Fullard				echo "done.";
979f28f1389SDave Fullard			else
980f28f1389SDave Fullard				echo "failed."
981f28f1389SDave Fullard				exit 1
982f28f1389SDave Fullard			fi
983f28f1389SDave Fullard		fi
984f28f1389SDave Fullard	fi
985f28f1389SDave Fullard}
986f28f1389SDave Fullard
98748ffe56aSColin Percival# Perform sanity checks and set some final parameters in
98848ffe56aSColin Percival# preparation for UNinstalling updates.
98948ffe56aSColin Percivalrollback_check_params () {
99048ffe56aSColin Percival	# Check that we are root.  All sorts of things won't work otherwise.
99148ffe56aSColin Percival	if [ `id -u` != 0 ]; then
99248ffe56aSColin Percival		echo "You must be root to run this."
99348ffe56aSColin Percival		exit 1
99448ffe56aSColin Percival	fi
99548ffe56aSColin Percival
99648ffe56aSColin Percival	# Check that we have a working directory
99748ffe56aSColin Percival	_WORKDIR_bad="Directory does not exist or is not writable: "
99848ffe56aSColin Percival	if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then
99948ffe56aSColin Percival		echo -n "`basename $0`: "
100048ffe56aSColin Percival		echo -n "${_WORKDIR_bad}"
100148ffe56aSColin Percival		echo ${WORKDIR}
100248ffe56aSColin Percival		exit 1
100348ffe56aSColin Percival	fi
100448ffe56aSColin Percival	cd ${WORKDIR} || exit 1
100548ffe56aSColin Percival
100648ffe56aSColin Percival	# Construct a unique name from ${BASEDIR}
100748ffe56aSColin Percival	BDHASH=`echo ${BASEDIR} | sha256 -q`
100848ffe56aSColin Percival
100948ffe56aSColin Percival	# Check that we have updates ready to rollback
101048ffe56aSColin Percival	if ! [ -L ${BDHASH}-rollback ]; then
101148ffe56aSColin Percival		echo "No rollback directory found."
101248ffe56aSColin Percival		exit 1
101348ffe56aSColin Percival	fi
101448ffe56aSColin Percival	if ! [ -f ${BDHASH}-rollback/INDEX-OLD ] ||
101548ffe56aSColin Percival	    ! [ -f ${BDHASH}-rollback/INDEX-NEW ]; then
101648ffe56aSColin Percival		echo "Update manifest is corrupt -- this should never happen."
101748ffe56aSColin Percival		exit 1
101848ffe56aSColin Percival	fi
101948ffe56aSColin Percival}
102048ffe56aSColin Percival
102108e23beeSColin Percival# Perform sanity checks and set some final parameters
102208e23beeSColin Percival# in preparation for comparing the system against the
102308e23beeSColin Percival# published index.  Figure out which index we should
102408e23beeSColin Percival# compare against: If the user is running *-p[0-9]+,
102508e23beeSColin Percival# strip off the last part; if the user is running
102608e23beeSColin Percival# -SECURITY, call it -RELEASE.  Chdir into the working
102708e23beeSColin Percival# directory.
102808e23beeSColin PercivalIDS_check_params () {
102908e23beeSColin Percival	export HTTP_USER_AGENT="freebsd-update (${COMMAND}, `uname -r`)"
103008e23beeSColin Percival
103108e23beeSColin Percival	_SERVERNAME_z=\
103208e23beeSColin Percival"SERVERNAME must be given via command line or configuration file."
103359b02bb4SMichael Osipov	_KEYPRINT_z="Key must be given via '-k' option or configuration file."
103408e23beeSColin Percival	_KEYPRINT_bad="Invalid key fingerprint: "
103508e23beeSColin Percival	_WORKDIR_bad="Directory does not exist or is not writable: "
103608e23beeSColin Percival
103708e23beeSColin Percival	if [ -z "${SERVERNAME}" ]; then
103808e23beeSColin Percival		echo -n "`basename $0`: "
103908e23beeSColin Percival		echo "${_SERVERNAME_z}"
104008e23beeSColin Percival		exit 1
104108e23beeSColin Percival	fi
104208e23beeSColin Percival	if [ -z "${KEYPRINT}" ]; then
104308e23beeSColin Percival		echo -n "`basename $0`: "
104408e23beeSColin Percival		echo "${_KEYPRINT_z}"
104508e23beeSColin Percival		exit 1
104608e23beeSColin Percival	fi
104708e23beeSColin Percival	if ! echo "${KEYPRINT}" | grep -qE "^[0-9a-f]{64}$"; then
104808e23beeSColin Percival		echo -n "`basename $0`: "
104908e23beeSColin Percival		echo -n "${_KEYPRINT_bad}"
105008e23beeSColin Percival		echo ${KEYPRINT}
105108e23beeSColin Percival		exit 1
105208e23beeSColin Percival	fi
105308e23beeSColin Percival	if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then
105408e23beeSColin Percival		echo -n "`basename $0`: "
105508e23beeSColin Percival		echo -n "${_WORKDIR_bad}"
105608e23beeSColin Percival		echo ${WORKDIR}
105708e23beeSColin Percival		exit 1
105808e23beeSColin Percival	fi
105908e23beeSColin Percival	cd ${WORKDIR} || exit 1
106008e23beeSColin Percival
106108e23beeSColin Percival	# Generate release number.  The s/SECURITY/RELEASE/ bit exists
106208e23beeSColin Percival	# to provide an upgrade path for FreeBSD Update 1.x users, since
106308e23beeSColin Percival	# the kernels provided by FreeBSD Update 1.x are always labelled
106408e23beeSColin Percival	# as X.Y-SECURITY.
106508e23beeSColin Percival	RELNUM=`uname -r |
106608e23beeSColin Percival	    sed -E 's,-p[0-9]+,,' |
106708e23beeSColin Percival	    sed -E 's,-SECURITY,-RELEASE,'`
106808e23beeSColin Percival	ARCH=`uname -m`
106908e23beeSColin Percival	FETCHDIR=${RELNUM}/${ARCH}
107008e23beeSColin Percival	PATCHDIR=${RELNUM}/${ARCH}/bp
107108e23beeSColin Percival
107208e23beeSColin Percival	# Figure out what directory contains the running kernel
107308e23beeSColin Percival	BOOTFILE=`sysctl -n kern.bootfile`
107408e23beeSColin Percival	KERNELDIR=${BOOTFILE%/kernel}
107508e23beeSColin Percival	if ! [ -d ${KERNELDIR} ]; then
107608e23beeSColin Percival		echo "Cannot identify running kernel"
107708e23beeSColin Percival		exit 1
107808e23beeSColin Percival	fi
107908e23beeSColin Percival
108008e23beeSColin Percival	# Figure out what kernel configuration is running.  We start with
108108e23beeSColin Percival	# the output of `uname -i`, and then make the following adjustments:
108208e23beeSColin Percival	# 1. Replace "SMP-GENERIC" with "SMP".  Why the SMP kernel config
108308e23beeSColin Percival	# file says "ident SMP-GENERIC", I don't know...
108408e23beeSColin Percival	# 2. If the kernel claims to be GENERIC _and_ ${ARCH} is "amd64"
108508e23beeSColin Percival	# _and_ `sysctl kern.version` contains a line which ends "/SMP", then
108608e23beeSColin Percival	# we're running an SMP kernel.  This mis-identification is a bug
108708e23beeSColin Percival	# which was fixed in 6.2-STABLE.
108808e23beeSColin Percival	KERNCONF=`uname -i`
108908e23beeSColin Percival	if [ ${KERNCONF} = "SMP-GENERIC" ]; then
109008e23beeSColin Percival		KERNCONF=SMP
109108e23beeSColin Percival	fi
109208e23beeSColin Percival	if [ ${KERNCONF} = "GENERIC" ] && [ ${ARCH} = "amd64" ]; then
109308e23beeSColin Percival		if sysctl kern.version | grep -qE '/SMP$'; then
109408e23beeSColin Percival			KERNCONF=SMP
109508e23beeSColin Percival		fi
109608e23beeSColin Percival	fi
109708e23beeSColin Percival
109808e23beeSColin Percival	# Define some paths
109908e23beeSColin Percival	SHA256=/sbin/sha256
110008e23beeSColin Percival	PHTTPGET=/usr/libexec/phttpget
110108e23beeSColin Percival
110208e23beeSColin Percival	# Set up variables relating to VERBOSELEVEL
110308e23beeSColin Percival	fetch_setup_verboselevel
110408e23beeSColin Percival}
110508e23beeSColin Percival
110648ffe56aSColin Percival#### Core functionality -- the actual work gets done here
110748ffe56aSColin Percival
110848ffe56aSColin Percival# Use an SRV query to pick a server.  If the SRV query doesn't provide
110948ffe56aSColin Percival# a useful answer, use the server name specified by the user.
111048ffe56aSColin Percival# Put another way... look up _http._tcp.${SERVERNAME} and pick a server
111148ffe56aSColin Percival# from that; or if no servers are returned, use ${SERVERNAME}.
111248ffe56aSColin Percival# This allows a user to specify "portsnap.freebsd.org" (in which case
111348ffe56aSColin Percival# portsnap will select one of the mirrors) or "portsnap5.tld.freebsd.org"
111448ffe56aSColin Percival# (in which case portsnap will use that particular server, since there
111548ffe56aSColin Percival# won't be an SRV entry for that name).
111648ffe56aSColin Percival#
111748ffe56aSColin Percival# We ignore the Port field, since we are always going to use port 80.
111848ffe56aSColin Percival
111948ffe56aSColin Percival# Fetch the mirror list, but do not pick a mirror yet.  Returns 1 if
112048ffe56aSColin Percival# no mirrors are available for any reason.
112148ffe56aSColin Percivalfetch_pick_server_init () {
112248ffe56aSColin Percival	: > serverlist_tried
112348ffe56aSColin Percival
112448ffe56aSColin Percival# Check that host(1) exists (i.e., that the system wasn't built with the
112548ffe56aSColin Percival# WITHOUT_BIND set) and don't try to find a mirror if it doesn't exist.
112648ffe56aSColin Percival	if ! which -s host; then
112748ffe56aSColin Percival		: > serverlist_full
112848ffe56aSColin Percival		return 1
112948ffe56aSColin Percival	fi
113048ffe56aSColin Percival
113148ffe56aSColin Percival	echo -n "Looking up ${SERVERNAME} mirrors... "
113248ffe56aSColin Percival
113348ffe56aSColin Percival# Issue the SRV query and pull out the Priority, Weight, and Target fields.
113448ffe56aSColin Percival# BIND 9 prints "$name has SRV record ..." while BIND 8 prints
113548ffe56aSColin Percival# "$name server selection ..."; we allow either format.
113648ffe56aSColin Percival	MLIST="_http._tcp.${SERVERNAME}"
113748ffe56aSColin Percival	host -t srv "${MLIST}" |
1138e7fd266eSColin Percival	    sed -nE "s/${MLIST} (has SRV record|server selection) //Ip" |
113948ffe56aSColin Percival	    cut -f 1,2,4 -d ' ' |
114048ffe56aSColin Percival	    sed -e 's/\.$//' |
114148ffe56aSColin Percival	    sort > serverlist_full
114248ffe56aSColin Percival
114348ffe56aSColin Percival# If no records, give up -- we'll just use the server name we were given.
114448ffe56aSColin Percival	if [ `wc -l < serverlist_full` -eq 0 ]; then
114548ffe56aSColin Percival		echo "none found."
114648ffe56aSColin Percival		return 1
114748ffe56aSColin Percival	fi
114848ffe56aSColin Percival
114948ffe56aSColin Percival# Report how many mirrors we found.
115048ffe56aSColin Percival	echo `wc -l < serverlist_full` "mirrors found."
115148ffe56aSColin Percival
115248ffe56aSColin Percival# Generate a random seed for use in picking mirrors.  If HTTP_PROXY
115348ffe56aSColin Percival# is set, this will be used to generate the seed; otherwise, the seed
115448ffe56aSColin Percival# will be random.
115548ffe56aSColin Percival	if [ -n "${HTTP_PROXY}${http_proxy}" ]; then
115648ffe56aSColin Percival		RANDVALUE=`sha256 -qs "${HTTP_PROXY}${http_proxy}" |
115748ffe56aSColin Percival		    tr -d 'a-f' |
115848ffe56aSColin Percival		    cut -c 1-9`
115948ffe56aSColin Percival	else
116048ffe56aSColin Percival		RANDVALUE=`jot -r 1 0 999999999`
116148ffe56aSColin Percival	fi
116248ffe56aSColin Percival}
116348ffe56aSColin Percival
116448ffe56aSColin Percival# Pick a mirror.  Returns 1 if we have run out of mirrors to try.
116548ffe56aSColin Percivalfetch_pick_server () {
116648ffe56aSColin Percival# Generate a list of not-yet-tried mirrors
116748ffe56aSColin Percival	sort serverlist_tried |
116848ffe56aSColin Percival	    comm -23 serverlist_full - > serverlist
116948ffe56aSColin Percival
117048ffe56aSColin Percival# Have we run out of mirrors?
117148ffe56aSColin Percival	if [ `wc -l < serverlist` -eq 0 ]; then
11729e8c28fcSEd Maste		cat <<- EOF
11739e8c28fcSEd Maste			No mirrors remaining, giving up.
11749e8c28fcSEd Maste
11759e8c28fcSEd Maste			This may be because upgrading from this platform (${ARCH})
11769e8c28fcSEd Maste			or release (${RELNUM}) is unsupported by `basename $0`. Only
11779e8c28fcSEd Maste			platforms with Tier 1 support can be upgraded by `basename $0`.
117886d0d3aaSLi-Wen Hsu			See https://www.freebsd.org/platforms/ for more info.
11799e8c28fcSEd Maste
11809e8c28fcSEd Maste			If unsupported, FreeBSD must be upgraded by source.
11819e8c28fcSEd Maste		EOF
118248ffe56aSColin Percival		return 1
118348ffe56aSColin Percival	fi
118448ffe56aSColin Percival
118548ffe56aSColin Percival# Find the highest priority level (lowest numeric value).
118648ffe56aSColin Percival	SRV_PRIORITY=`cut -f 1 -d ' ' serverlist | sort -n | head -1`
118748ffe56aSColin Percival
118848ffe56aSColin Percival# Add up the weights of the response lines at that priority level.
118948ffe56aSColin Percival	SRV_WSUM=0;
119048ffe56aSColin Percival	while read X; do
119148ffe56aSColin Percival		case "$X" in
119248ffe56aSColin Percival		${SRV_PRIORITY}\ *)
119348ffe56aSColin Percival			SRV_W=`echo $X | cut -f 2 -d ' '`
119448ffe56aSColin Percival			SRV_WSUM=$(($SRV_WSUM + $SRV_W))
119548ffe56aSColin Percival			;;
119648ffe56aSColin Percival		esac
119748ffe56aSColin Percival	done < serverlist
119848ffe56aSColin Percival
119948ffe56aSColin Percival# If all the weights are 0, pretend that they are all 1 instead.
120048ffe56aSColin Percival	if [ ${SRV_WSUM} -eq 0 ]; then
120148ffe56aSColin Percival		SRV_WSUM=`grep -E "^${SRV_PRIORITY} " serverlist | wc -l`
120248ffe56aSColin Percival		SRV_W_ADD=1
120348ffe56aSColin Percival	else
120448ffe56aSColin Percival		SRV_W_ADD=0
120548ffe56aSColin Percival	fi
120648ffe56aSColin Percival
120748ffe56aSColin Percival# Pick a value between 0 and the sum of the weights - 1
120848ffe56aSColin Percival	SRV_RND=`expr ${RANDVALUE} % ${SRV_WSUM}`
120948ffe56aSColin Percival
121048ffe56aSColin Percival# Read through the list of mirrors and set SERVERNAME.  Write the line
121148ffe56aSColin Percival# corresponding to the mirror we selected into serverlist_tried so that
121248ffe56aSColin Percival# we won't try it again.
121348ffe56aSColin Percival	while read X; do
121448ffe56aSColin Percival		case "$X" in
121548ffe56aSColin Percival		${SRV_PRIORITY}\ *)
121648ffe56aSColin Percival			SRV_W=`echo $X | cut -f 2 -d ' '`
121748ffe56aSColin Percival			SRV_W=$(($SRV_W + $SRV_W_ADD))
121848ffe56aSColin Percival			if [ $SRV_RND -lt $SRV_W ]; then
121948ffe56aSColin Percival				SERVERNAME=`echo $X | cut -f 3 -d ' '`
122048ffe56aSColin Percival				echo "$X" >> serverlist_tried
122148ffe56aSColin Percival				break
122248ffe56aSColin Percival			else
122348ffe56aSColin Percival				SRV_RND=$(($SRV_RND - $SRV_W))
122448ffe56aSColin Percival			fi
122548ffe56aSColin Percival			;;
122648ffe56aSColin Percival		esac
122748ffe56aSColin Percival	done < serverlist
122848ffe56aSColin Percival}
122948ffe56aSColin Percival
123048ffe56aSColin Percival# Take a list of ${oldhash}|${newhash} and output a list of needed patches,
123148ffe56aSColin Percival# i.e., those for which we have ${oldhash} and don't have ${newhash}.
123248ffe56aSColin Percivalfetch_make_patchlist () {
123348ffe56aSColin Percival	grep -vE "^([0-9a-f]{64})\|\1$" |
123448ffe56aSColin Percival	    tr '|' ' ' |
123548ffe56aSColin Percival		while read X Y; do
123648ffe56aSColin Percival			if [ -f "files/${Y}.gz" ] ||
123748ffe56aSColin Percival			    [ ! -f "files/${X}.gz" ]; then
123848ffe56aSColin Percival				continue
123948ffe56aSColin Percival			fi
124048ffe56aSColin Percival			echo "${X}|${Y}"
1241f6e21461SEd Maste		done | sort -u
124248ffe56aSColin Percival}
124348ffe56aSColin Percival
124448ffe56aSColin Percival# Print user-friendly progress statistics
124548ffe56aSColin Percivalfetch_progress () {
124648ffe56aSColin Percival	LNC=0
124748ffe56aSColin Percival	while read x; do
124848ffe56aSColin Percival		LNC=$(($LNC + 1))
124948ffe56aSColin Percival		if [ $(($LNC % 10)) = 0 ]; then
125048ffe56aSColin Percival			echo -n $LNC
125148ffe56aSColin Percival		elif [ $(($LNC % 2)) = 0 ]; then
125248ffe56aSColin Percival			echo -n .
125348ffe56aSColin Percival		fi
125448ffe56aSColin Percival	done
125548ffe56aSColin Percival	echo -n " "
125648ffe56aSColin Percival}
125748ffe56aSColin Percival
1258db6b0a61SColin Percival# Function for asking the user if everything is ok
1259db6b0a61SColin Percivalcontinuep () {
1260db6b0a61SColin Percival	while read -p "Does this look reasonable (y/n)? " CONTINUE; do
1261db6b0a61SColin Percival		case "${CONTINUE}" in
126239f4633bSJuraj Lutter		[yY]*)
1263db6b0a61SColin Percival			return 0
1264db6b0a61SColin Percival			;;
126539f4633bSJuraj Lutter		[nN]*)
1266db6b0a61SColin Percival			return 1
1267db6b0a61SColin Percival			;;
1268db6b0a61SColin Percival		esac
1269db6b0a61SColin Percival	done
1270db6b0a61SColin Percival}
1271db6b0a61SColin Percival
127248ffe56aSColin Percival# Initialize the working directory
127348ffe56aSColin Percivalworkdir_init () {
127448ffe56aSColin Percival	mkdir -p files
127548ffe56aSColin Percival	touch tINDEX.present
127648ffe56aSColin Percival}
127748ffe56aSColin Percival
127848ffe56aSColin Percival# Check that we have a public key with an appropriate hash, or
127948ffe56aSColin Percival# fetch the key if it doesn't exist.  Returns 1 if the key has
128048ffe56aSColin Percival# not yet been fetched.
128148ffe56aSColin Percivalfetch_key () {
128248ffe56aSColin Percival	if [ -r pub.ssl ] && [ `${SHA256} -q pub.ssl` = ${KEYPRINT} ]; then
128348ffe56aSColin Percival		return 0
128448ffe56aSColin Percival	fi
128548ffe56aSColin Percival
128648ffe56aSColin Percival	echo -n "Fetching public key from ${SERVERNAME}... "
128748ffe56aSColin Percival	rm -f pub.ssl
128848ffe56aSColin Percival	fetch ${QUIETFLAG} http://${SERVERNAME}/${FETCHDIR}/pub.ssl \
128948ffe56aSColin Percival	    2>${QUIETREDIR} || true
129048ffe56aSColin Percival	if ! [ -r pub.ssl ]; then
129148ffe56aSColin Percival		echo "failed."
129248ffe56aSColin Percival		return 1
129348ffe56aSColin Percival	fi
129448ffe56aSColin Percival	if ! [ `${SHA256} -q pub.ssl` = ${KEYPRINT} ]; then
129548ffe56aSColin Percival		echo "key has incorrect hash."
129648ffe56aSColin Percival		rm -f pub.ssl
129748ffe56aSColin Percival		return 1
129848ffe56aSColin Percival	fi
129948ffe56aSColin Percival	echo "done."
130048ffe56aSColin Percival}
130148ffe56aSColin Percival
130248ffe56aSColin Percival# Fetch metadata signature, aka "tag".
130348ffe56aSColin Percivalfetch_tag () {
1304db6b0a61SColin Percival	echo -n "Fetching metadata signature "
1305db6b0a61SColin Percival	echo ${NDEBUG} "for ${RELNUM} from ${SERVERNAME}... "
130648ffe56aSColin Percival	rm -f latest.ssl
130748ffe56aSColin Percival	fetch ${QUIETFLAG} http://${SERVERNAME}/${FETCHDIR}/latest.ssl	\
130848ffe56aSColin Percival	    2>${QUIETREDIR} || true
130948ffe56aSColin Percival	if ! [ -r latest.ssl ]; then
131048ffe56aSColin Percival		echo "failed."
131148ffe56aSColin Percival		return 1
131248ffe56aSColin Percival	fi
131348ffe56aSColin Percival
131448ffe56aSColin Percival	openssl rsautl -pubin -inkey pub.ssl -verify		\
131548ffe56aSColin Percival	    < latest.ssl > tag.new 2>${QUIETREDIR} || true
131648ffe56aSColin Percival	rm latest.ssl
131748ffe56aSColin Percival
131848ffe56aSColin Percival	if ! [ `wc -l < tag.new` = 1 ] ||
131948ffe56aSColin Percival	    ! grep -qE	\
132048ffe56aSColin Percival    "^freebsd-update\|${ARCH}\|${RELNUM}\|[0-9]+\|[0-9a-f]{64}\|[0-9]{10}" \
132148ffe56aSColin Percival		tag.new; then
132248ffe56aSColin Percival		echo "invalid signature."
132348ffe56aSColin Percival		return 1
132448ffe56aSColin Percival	fi
132548ffe56aSColin Percival
132648ffe56aSColin Percival	echo "done."
132748ffe56aSColin Percival
132848ffe56aSColin Percival	RELPATCHNUM=`cut -f 4 -d '|' < tag.new`
132948ffe56aSColin Percival	TINDEXHASH=`cut -f 5 -d '|' < tag.new`
133048ffe56aSColin Percival	EOLTIME=`cut -f 6 -d '|' < tag.new`
133148ffe56aSColin Percival}
133248ffe56aSColin Percival
133348ffe56aSColin Percival# Sanity-check the patch number in a tag, to make sure that we're not
133448ffe56aSColin Percival# going to "update" backwards and to prevent replay attacks.
133548ffe56aSColin Percivalfetch_tagsanity () {
133648ffe56aSColin Percival	# Check that we're not going to move from -pX to -pY with Y < X.
133748ffe56aSColin Percival	RELPX=`uname -r | sed -E 's,.*-,,'`
133848ffe56aSColin Percival	if echo ${RELPX} | grep -qE '^p[0-9]+$'; then
133948ffe56aSColin Percival		RELPX=`echo ${RELPX} | cut -c 2-`
134048ffe56aSColin Percival	else
134148ffe56aSColin Percival		RELPX=0
134248ffe56aSColin Percival	fi
134348ffe56aSColin Percival	if [ "${RELPATCHNUM}" -lt "${RELPX}" ]; then
134448ffe56aSColin Percival		echo
134548ffe56aSColin Percival		echo -n "Files on mirror (${RELNUM}-p${RELPATCHNUM})"
134648ffe56aSColin Percival		echo " appear older than what"
134748ffe56aSColin Percival		echo "we are currently running (`uname -r`)!"
134848ffe56aSColin Percival		echo "Cowardly refusing to proceed any further."
134948ffe56aSColin Percival		return 1
135048ffe56aSColin Percival	fi
135148ffe56aSColin Percival
135248ffe56aSColin Percival	# If "tag" exists and corresponds to ${RELNUM}, make sure that
135348ffe56aSColin Percival	# it contains a patch number <= RELPATCHNUM, in order to protect
135448ffe56aSColin Percival	# against rollback (replay) attacks.
135548ffe56aSColin Percival	if [ -f tag ] &&
135648ffe56aSColin Percival	    grep -qE	\
135748ffe56aSColin Percival    "^freebsd-update\|${ARCH}\|${RELNUM}\|[0-9]+\|[0-9a-f]{64}\|[0-9]{10}" \
135848ffe56aSColin Percival		tag; then
135948ffe56aSColin Percival		LASTRELPATCHNUM=`cut -f 4 -d '|' < tag`
136048ffe56aSColin Percival
136148ffe56aSColin Percival		if [ "${RELPATCHNUM}" -lt "${LASTRELPATCHNUM}" ]; then
136248ffe56aSColin Percival			echo
136348ffe56aSColin Percival			echo -n "Files on mirror (${RELNUM}-p${RELPATCHNUM})"
136448ffe56aSColin Percival			echo " are older than the"
136548ffe56aSColin Percival			echo -n "most recently seen updates"
136648ffe56aSColin Percival			echo " (${RELNUM}-p${LASTRELPATCHNUM})."
136748ffe56aSColin Percival			echo "Cowardly refusing to proceed any further."
136848ffe56aSColin Percival			return 1
136948ffe56aSColin Percival		fi
137048ffe56aSColin Percival	fi
137148ffe56aSColin Percival}
137248ffe56aSColin Percival
137348ffe56aSColin Percival# Fetch metadata index file
137448ffe56aSColin Percivalfetch_metadata_index () {
137548ffe56aSColin Percival	echo ${NDEBUG} "Fetching metadata index... "
137648ffe56aSColin Percival	rm -f ${TINDEXHASH}
137748ffe56aSColin Percival	fetch ${QUIETFLAG} http://${SERVERNAME}/${FETCHDIR}/t/${TINDEXHASH}
137848ffe56aSColin Percival	    2>${QUIETREDIR}
137948ffe56aSColin Percival	if ! [ -f ${TINDEXHASH} ]; then
138048ffe56aSColin Percival		echo "failed."
138148ffe56aSColin Percival		return 1
138248ffe56aSColin Percival	fi
138348ffe56aSColin Percival	if [ `${SHA256} -q ${TINDEXHASH}` != ${TINDEXHASH} ]; then
138448ffe56aSColin Percival		echo "update metadata index corrupt."
138548ffe56aSColin Percival		return 1
138648ffe56aSColin Percival	fi
138748ffe56aSColin Percival	echo "done."
138848ffe56aSColin Percival}
138948ffe56aSColin Percival
139048ffe56aSColin Percival# Print an error message about signed metadata being bogus.
139148ffe56aSColin Percivalfetch_metadata_bogus () {
139248ffe56aSColin Percival	echo
139348ffe56aSColin Percival	echo "The update metadata$1 is correctly signed, but"
139448ffe56aSColin Percival	echo "failed an integrity check."
139548ffe56aSColin Percival	echo "Cowardly refusing to proceed any further."
139648ffe56aSColin Percival	return 1
139748ffe56aSColin Percival}
139848ffe56aSColin Percival
139948ffe56aSColin Percival# Construct tINDEX.new by merging the lines named in $1 from ${TINDEXHASH}
140048ffe56aSColin Percival# with the lines not named in $@ from tINDEX.present (if that file exists).
140148ffe56aSColin Percivalfetch_metadata_index_merge () {
140248ffe56aSColin Percival	for METAFILE in $@; do
140348ffe56aSColin Percival		if [ `grep -E "^${METAFILE}\|" ${TINDEXHASH} | wc -l`	\
140448ffe56aSColin Percival		    -ne 1 ]; then
140548ffe56aSColin Percival			fetch_metadata_bogus " index"
140648ffe56aSColin Percival			return 1
140748ffe56aSColin Percival		fi
140848ffe56aSColin Percival
140948ffe56aSColin Percival		grep -E "${METAFILE}\|" ${TINDEXHASH}
141048ffe56aSColin Percival	done |
141148ffe56aSColin Percival	    sort > tINDEX.wanted
141248ffe56aSColin Percival
141348ffe56aSColin Percival	if [ -f tINDEX.present ]; then
141448ffe56aSColin Percival		join -t '|' -v 2 tINDEX.wanted tINDEX.present |
141548ffe56aSColin Percival		    sort -m - tINDEX.wanted > tINDEX.new
141648ffe56aSColin Percival		rm tINDEX.wanted
141748ffe56aSColin Percival	else
141848ffe56aSColin Percival		mv tINDEX.wanted tINDEX.new
141948ffe56aSColin Percival	fi
142048ffe56aSColin Percival}
142148ffe56aSColin Percival
142248ffe56aSColin Percival# Sanity check all the lines of tINDEX.new.  Even if more metadata lines
142348ffe56aSColin Percival# are added by future versions of the server, this won't cause problems,
142448ffe56aSColin Percival# since the only lines which appear in tINDEX.new are the ones which we
142548ffe56aSColin Percival# specifically grepped out of ${TINDEXHASH}.
142648ffe56aSColin Percivalfetch_metadata_index_sanity () {
142748ffe56aSColin Percival	if grep -qvE '^[0-9A-Z.-]+\|[0-9a-f]{64}$' tINDEX.new; then
142848ffe56aSColin Percival		fetch_metadata_bogus " index"
142948ffe56aSColin Percival		return 1
143048ffe56aSColin Percival	fi
143148ffe56aSColin Percival}
143248ffe56aSColin Percival
143348ffe56aSColin Percival# Sanity check the metadata file $1.
143448ffe56aSColin Percivalfetch_metadata_sanity () {
143548ffe56aSColin Percival	# Some aliases to save space later: ${P} is a character which can
143648ffe56aSColin Percival	# appear in a path; ${M} is the four numeric metadata fields; and
143748ffe56aSColin Percival	# ${H} is a sha256 hash.
14387c06c7c5SKris Moore	P="[-+./:=,%@_[~[:alnum:]]"
143948ffe56aSColin Percival	M="[0-9]+\|[0-9]+\|[0-9]+\|[0-9]+"
144048ffe56aSColin Percival	H="[0-9a-f]{64}"
144148ffe56aSColin Percival
144248ffe56aSColin Percival	# Check that the first four fields make sense.
144348ffe56aSColin Percival	if gunzip -c < files/$1.gz |
1444823c0d5fSXin LI	    grep -qvE "^[a-z]+\|[0-9a-z-]+\|${P}+\|[fdL-]\|"; then
144548ffe56aSColin Percival		fetch_metadata_bogus ""
144648ffe56aSColin Percival		return 1
144748ffe56aSColin Percival	fi
144848ffe56aSColin Percival
144948ffe56aSColin Percival	# Remove the first three fields.
145048ffe56aSColin Percival	gunzip -c < files/$1.gz |
145148ffe56aSColin Percival	    cut -f 4- -d '|' > sanitycheck.tmp
145248ffe56aSColin Percival
145348ffe56aSColin Percival	# Sanity check entries with type 'f'
145448ffe56aSColin Percival	if grep -E '^f' sanitycheck.tmp |
145548ffe56aSColin Percival	    grep -qvE "^f\|${M}\|${H}\|${P}*\$"; then
145648ffe56aSColin Percival		fetch_metadata_bogus ""
145748ffe56aSColin Percival		return 1
145848ffe56aSColin Percival	fi
145948ffe56aSColin Percival
146048ffe56aSColin Percival	# Sanity check entries with type 'd'
146148ffe56aSColin Percival	if grep -E '^d' sanitycheck.tmp |
146248ffe56aSColin Percival	    grep -qvE "^d\|${M}\|\|\$"; then
146348ffe56aSColin Percival		fetch_metadata_bogus ""
146448ffe56aSColin Percival		return 1
146548ffe56aSColin Percival	fi
146648ffe56aSColin Percival
146748ffe56aSColin Percival	# Sanity check entries with type 'L'
146848ffe56aSColin Percival	if grep -E '^L' sanitycheck.tmp |
146948ffe56aSColin Percival	    grep -qvE "^L\|${M}\|${P}*\|\$"; then
147048ffe56aSColin Percival		fetch_metadata_bogus ""
147148ffe56aSColin Percival		return 1
147248ffe56aSColin Percival	fi
147348ffe56aSColin Percival
147448ffe56aSColin Percival	# Sanity check entries with type '-'
147548ffe56aSColin Percival	if grep -E '^-' sanitycheck.tmp |
147648ffe56aSColin Percival	    grep -qvE "^-\|\|\|\|\|\|"; then
147748ffe56aSColin Percival		fetch_metadata_bogus ""
147848ffe56aSColin Percival		return 1
147948ffe56aSColin Percival	fi
148048ffe56aSColin Percival
148148ffe56aSColin Percival	# Clean up
148248ffe56aSColin Percival	rm sanitycheck.tmp
148348ffe56aSColin Percival}
148448ffe56aSColin Percival
148548ffe56aSColin Percival# Fetch the metadata index and metadata files listed in $@,
148648ffe56aSColin Percival# taking advantage of metadata patches where possible.
148748ffe56aSColin Percivalfetch_metadata () {
148848ffe56aSColin Percival	fetch_metadata_index || return 1
148948ffe56aSColin Percival	fetch_metadata_index_merge $@ || return 1
149048ffe56aSColin Percival	fetch_metadata_index_sanity || return 1
149148ffe56aSColin Percival
149248ffe56aSColin Percival	# Generate a list of wanted metadata patches
149348ffe56aSColin Percival	join -t '|' -o 1.2,2.2 tINDEX.present tINDEX.new |
149448ffe56aSColin Percival	    fetch_make_patchlist > patchlist
149548ffe56aSColin Percival
149648ffe56aSColin Percival	if [ -s patchlist ]; then
149748ffe56aSColin Percival		# Attempt to fetch metadata patches
149848ffe56aSColin Percival		echo -n "Fetching `wc -l < patchlist | tr -d ' '` "
149948ffe56aSColin Percival		echo ${NDEBUG} "metadata patches.${DDSTATS}"
150048ffe56aSColin Percival		tr '|' '-' < patchlist |
150148ffe56aSColin Percival		    lam -s "${FETCHDIR}/tp/" - -s ".gz" |
150248ffe56aSColin Percival		    xargs ${XARGST} ${PHTTPGET} ${SERVERNAME}	\
150348ffe56aSColin Percival			2>${STATSREDIR} | fetch_progress
150448ffe56aSColin Percival		echo "done."
150548ffe56aSColin Percival
150648ffe56aSColin Percival		# Attempt to apply metadata patches
150748ffe56aSColin Percival		echo -n "Applying metadata patches... "
150848ffe56aSColin Percival		tr '|' ' ' < patchlist |
150948ffe56aSColin Percival		    while read X Y; do
151048ffe56aSColin Percival			if [ ! -f "${X}-${Y}.gz" ]; then continue; fi
151148ffe56aSColin Percival			gunzip -c < ${X}-${Y}.gz > diff
151248ffe56aSColin Percival			gunzip -c < files/${X}.gz > diff-OLD
151348ffe56aSColin Percival
151448ffe56aSColin Percival			# Figure out which lines are being added and removed
151548ffe56aSColin Percival			grep -E '^-' diff |
151648ffe56aSColin Percival			    cut -c 2- |
151748ffe56aSColin Percival			    while read PREFIX; do
151848ffe56aSColin Percival				look "${PREFIX}" diff-OLD
151948ffe56aSColin Percival			    done |
152048ffe56aSColin Percival			    sort > diff-rm
152148ffe56aSColin Percival			grep -E '^\+' diff |
152248ffe56aSColin Percival			    cut -c 2- > diff-add
152348ffe56aSColin Percival
152448ffe56aSColin Percival			# Generate the new file
152548ffe56aSColin Percival			comm -23 diff-OLD diff-rm |
152648ffe56aSColin Percival			    sort - diff-add > diff-NEW
152748ffe56aSColin Percival
152848ffe56aSColin Percival			if [ `${SHA256} -q diff-NEW` = ${Y} ]; then
152948ffe56aSColin Percival				mv diff-NEW files/${Y}
153048ffe56aSColin Percival				gzip -n files/${Y}
153148ffe56aSColin Percival			else
153248ffe56aSColin Percival				mv diff-NEW ${Y}.bad
153348ffe56aSColin Percival			fi
153448ffe56aSColin Percival			rm -f ${X}-${Y}.gz diff
153548ffe56aSColin Percival			rm -f diff-OLD diff-NEW diff-add diff-rm
153648ffe56aSColin Percival		done 2>${QUIETREDIR}
153748ffe56aSColin Percival		echo "done."
153848ffe56aSColin Percival	fi
153948ffe56aSColin Percival
154048ffe56aSColin Percival	# Update metadata without patches
154148ffe56aSColin Percival	cut -f 2 -d '|' < tINDEX.new |
154248ffe56aSColin Percival	    while read Y; do
154348ffe56aSColin Percival		if [ ! -f "files/${Y}.gz" ]; then
154448ffe56aSColin Percival			echo ${Y};
154548ffe56aSColin Percival		fi
1546bce02f98SColin Percival	    done |
1547bce02f98SColin Percival	    sort -u > filelist
154848ffe56aSColin Percival
154948ffe56aSColin Percival	if [ -s filelist ]; then
155048ffe56aSColin Percival		echo -n "Fetching `wc -l < filelist | tr -d ' '` "
155148ffe56aSColin Percival		echo ${NDEBUG} "metadata files... "
155248ffe56aSColin Percival		lam -s "${FETCHDIR}/m/" - -s ".gz" < filelist |
155348ffe56aSColin Percival		    xargs ${XARGST} ${PHTTPGET} ${SERVERNAME}	\
155448ffe56aSColin Percival		    2>${QUIETREDIR}
155548ffe56aSColin Percival
155648ffe56aSColin Percival		while read Y; do
155748ffe56aSColin Percival			if ! [ -f ${Y}.gz ]; then
155848ffe56aSColin Percival				echo "failed."
155948ffe56aSColin Percival				return 1
156048ffe56aSColin Percival			fi
156148ffe56aSColin Percival			if [ `gunzip -c < ${Y}.gz |
156248ffe56aSColin Percival			    ${SHA256} -q` = ${Y} ]; then
156348ffe56aSColin Percival				mv ${Y}.gz files/${Y}.gz
156448ffe56aSColin Percival			else
156548ffe56aSColin Percival				echo "metadata is corrupt."
156648ffe56aSColin Percival				return 1
156748ffe56aSColin Percival			fi
156848ffe56aSColin Percival		done < filelist
156948ffe56aSColin Percival		echo "done."
157048ffe56aSColin Percival	fi
157148ffe56aSColin Percival
157248ffe56aSColin Percival# Sanity-check the metadata files.
157348ffe56aSColin Percival	cut -f 2 -d '|' tINDEX.new > filelist
157448ffe56aSColin Percival	while read X; do
157548ffe56aSColin Percival		fetch_metadata_sanity ${X} || return 1
157648ffe56aSColin Percival	done < filelist
157748ffe56aSColin Percival
157848ffe56aSColin Percival# Remove files which are no longer needed
157948ffe56aSColin Percival	cut -f 2 -d '|' tINDEX.present |
158048ffe56aSColin Percival	    sort > oldfiles
158148ffe56aSColin Percival	cut -f 2 -d '|' tINDEX.new |
158248ffe56aSColin Percival	    sort |
158348ffe56aSColin Percival	    comm -13 - oldfiles |
158448ffe56aSColin Percival	    lam -s "files/" - -s ".gz" |
158548ffe56aSColin Percival	    xargs rm -f
158648ffe56aSColin Percival	rm patchlist filelist oldfiles
158748ffe56aSColin Percival	rm ${TINDEXHASH}
158848ffe56aSColin Percival
158948ffe56aSColin Percival# We're done!
159048ffe56aSColin Percival	mv tINDEX.new tINDEX.present
159148ffe56aSColin Percival	mv tag.new tag
159248ffe56aSColin Percival
159348ffe56aSColin Percival	return 0
159448ffe56aSColin Percival}
159548ffe56aSColin Percival
1596db6b0a61SColin Percival# Extract a subset of a downloaded metadata file containing only the parts
1597db6b0a61SColin Percival# which are listed in COMPONENTS.
1598db6b0a61SColin Percivalfetch_filter_metadata_components () {
1599db6b0a61SColin Percival	METAHASH=`look "$1|" tINDEX.present | cut -f 2 -d '|'`
1600db6b0a61SColin Percival	gunzip -c < files/${METAHASH}.gz > $1.all
1601db6b0a61SColin Percival
1602db6b0a61SColin Percival	# Fish out the lines belonging to components we care about.
1603db6b0a61SColin Percival	for C in ${COMPONENTS}; do
1604db6b0a61SColin Percival		look "`echo ${C} | tr '/' '|'`|" $1.all
1605db6b0a61SColin Percival	done > $1
1606db6b0a61SColin Percival
1607db6b0a61SColin Percival	# Remove temporary file.
1608db6b0a61SColin Percival	rm $1.all
1609db6b0a61SColin Percival}
1610db6b0a61SColin Percival
1611b698a3abSColin Percival# Generate a filtered version of the metadata file $1 from the downloaded
161248ffe56aSColin Percival# file, by fishing out the lines corresponding to components we're trying
161348ffe56aSColin Percival# to keep updated, and then removing lines corresponding to paths we want
161448ffe56aSColin Percival# to ignore.
161548ffe56aSColin Percivalfetch_filter_metadata () {
161648ffe56aSColin Percival	# Fish out the lines belonging to components we care about.
1617db6b0a61SColin Percival	fetch_filter_metadata_components $1
1618db6b0a61SColin Percival
161948ffe56aSColin Percival	# Canonicalize directory names by removing any trailing / in
162048ffe56aSColin Percival	# order to avoid listing directories multiple times if they
162148ffe56aSColin Percival	# belong to multiple components.  Turning "/" into "" doesn't
162248ffe56aSColin Percival	# matter, since we add a leading "/" when we use paths later.
1623db6b0a61SColin Percival	cut -f 3- -d '|' $1 |
162448ffe56aSColin Percival	    sed -e 's,/|d|,|d|,' |
16257e654612SColin Percival	    sed -e 's,/|-|,|-|,' |
162648ffe56aSColin Percival	    sort -u > $1.tmp
162748ffe56aSColin Percival
162848ffe56aSColin Percival	# Figure out which lines to ignore and remove them.
162948ffe56aSColin Percival	for X in ${IGNOREPATHS}; do
163048ffe56aSColin Percival		grep -E "^${X}" $1.tmp
163148ffe56aSColin Percival	done |
163248ffe56aSColin Percival	    sort -u |
163348ffe56aSColin Percival	    comm -13 - $1.tmp > $1
163448ffe56aSColin Percival
163548ffe56aSColin Percival	# Remove temporary files.
1636db6b0a61SColin Percival	rm $1.tmp
163748ffe56aSColin Percival}
163848ffe56aSColin Percival
1639db6b0a61SColin Percival# Filter the metadata file $1 by adding lines with "/boot/$2"
1640bce02f98SColin Percival# replaced by ${KERNELDIR} (which is `sysctl -n kern.bootfile` minus the
1641db6b0a61SColin Percival# trailing "/kernel"); and if "/boot/$2" does not exist, remove
1642bce02f98SColin Percival# the original lines which start with that.
1643bce02f98SColin Percival# Put another way: Deal with the fact that the FOO kernel is sometimes
1644bce02f98SColin Percival# installed in /boot/FOO/ and is sometimes installed elsewhere.
164548ffe56aSColin Percivalfetch_filter_kernel_names () {
1646db6b0a61SColin Percival	grep ^/boot/$2 $1 |
1647db6b0a61SColin Percival	    sed -e "s,/boot/$2,${KERNELDIR},g" |
164848ffe56aSColin Percival	    sort - $1 > $1.tmp
164948ffe56aSColin Percival	mv $1.tmp $1
1650bce02f98SColin Percival
1651db6b0a61SColin Percival	if ! [ -d /boot/$2 ]; then
1652db6b0a61SColin Percival		grep -v ^/boot/$2 $1 > $1.tmp
1653bce02f98SColin Percival		mv $1.tmp $1
1654bce02f98SColin Percival	fi
165548ffe56aSColin Percival}
165648ffe56aSColin Percival
165748ffe56aSColin Percival# For all paths appearing in $1 or $3, inspect the system
165848ffe56aSColin Percival# and generate $2 describing what is currently installed.
165948ffe56aSColin Percivalfetch_inspect_system () {
166048ffe56aSColin Percival	# No errors yet...
166148ffe56aSColin Percival	rm -f .err
166248ffe56aSColin Percival
166348ffe56aSColin Percival	# Tell the user why his disk is suddenly making lots of noise
166448ffe56aSColin Percival	echo -n "Inspecting system... "
166548ffe56aSColin Percival
166648ffe56aSColin Percival	# Generate list of files to inspect
166748ffe56aSColin Percival	cat $1 $3 |
166848ffe56aSColin Percival	    cut -f 1 -d '|' |
166948ffe56aSColin Percival	    sort -u > filelist
167048ffe56aSColin Percival
167148ffe56aSColin Percival	# Examine each file and output lines of the form
167248ffe56aSColin Percival	# /path/to/file|type|device-inum|user|group|perm|flags|value
167348ffe56aSColin Percival	# sorted by device and inode number.
167448ffe56aSColin Percival	while read F; do
167548ffe56aSColin Percival		# If the symlink/file/directory does not exist, record this.
167648ffe56aSColin Percival		if ! [ -e ${BASEDIR}/${F} ]; then
167748ffe56aSColin Percival			echo "${F}|-||||||"
167848ffe56aSColin Percival			continue
167948ffe56aSColin Percival		fi
168048ffe56aSColin Percival		if ! [ -r ${BASEDIR}/${F} ]; then
168148ffe56aSColin Percival			echo "Cannot read file: ${BASEDIR}/${F}"	\
168248ffe56aSColin Percival			    >/dev/stderr
168348ffe56aSColin Percival			touch .err
168448ffe56aSColin Percival			return 1
168548ffe56aSColin Percival		fi
168648ffe56aSColin Percival
168748ffe56aSColin Percival		# Otherwise, output an index line.
168848ffe56aSColin Percival		if [ -L ${BASEDIR}/${F} ]; then
168948ffe56aSColin Percival			echo -n "${F}|L|"
169048ffe56aSColin Percival			stat -n -f '%d-%i|%u|%g|%Mp%Lp|%Of|' ${BASEDIR}/${F};
169148ffe56aSColin Percival			readlink ${BASEDIR}/${F};
169248ffe56aSColin Percival		elif [ -f ${BASEDIR}/${F} ]; then
169348ffe56aSColin Percival			echo -n "${F}|f|"
169448ffe56aSColin Percival			stat -n -f '%d-%i|%u|%g|%Mp%Lp|%Of|' ${BASEDIR}/${F};
169548ffe56aSColin Percival			sha256 -q ${BASEDIR}/${F};
169648ffe56aSColin Percival		elif [ -d ${BASEDIR}/${F} ]; then
169748ffe56aSColin Percival			echo -n "${F}|d|"
169848ffe56aSColin Percival			stat -f '%d-%i|%u|%g|%Mp%Lp|%Of|' ${BASEDIR}/${F};
169948ffe56aSColin Percival		else
170048ffe56aSColin Percival			echo "Unknown file type: ${BASEDIR}/${F}"	\
170148ffe56aSColin Percival			    >/dev/stderr
170248ffe56aSColin Percival			touch .err
170348ffe56aSColin Percival			return 1
170448ffe56aSColin Percival		fi
170548ffe56aSColin Percival	done < filelist |
170648ffe56aSColin Percival	    sort -k 3,3 -t '|' > $2.tmp
170748ffe56aSColin Percival	rm filelist
170848ffe56aSColin Percival
17096dcc68c8SBenedict Reuschling	# Check if an error occurred during system inspection
171048ffe56aSColin Percival	if [ -f .err ]; then
171148ffe56aSColin Percival		return 1
171248ffe56aSColin Percival	fi
171348ffe56aSColin Percival
171448ffe56aSColin Percival	# Convert to the form
171548ffe56aSColin Percival	# /path/to/file|type|user|group|perm|flags|value|hlink
171648ffe56aSColin Percival	# by resolving identical device and inode numbers into hard links.
171748ffe56aSColin Percival	cut -f 1,3 -d '|' $2.tmp |
171848ffe56aSColin Percival	    sort -k 1,1 -t '|' |
171948ffe56aSColin Percival	    sort -s -u -k 2,2 -t '|' |
172048ffe56aSColin Percival	    join -1 2 -2 3 -t '|' - $2.tmp |
172148ffe56aSColin Percival	    awk -F \| -v OFS=\|		\
172248ffe56aSColin Percival		'{
172348ffe56aSColin Percival		    if (($2 == $3) || ($4 == "-"))
172448ffe56aSColin Percival			print $3,$4,$5,$6,$7,$8,$9,""
172548ffe56aSColin Percival		    else
172648ffe56aSColin Percival			print $3,$4,$5,$6,$7,$8,$9,$2
172748ffe56aSColin Percival		}' |
172848ffe56aSColin Percival	    sort > $2
172948ffe56aSColin Percival	rm $2.tmp
173048ffe56aSColin Percival
173148ffe56aSColin Percival	# We're finished looking around
173248ffe56aSColin Percival	echo "done."
173348ffe56aSColin Percival}
173448ffe56aSColin Percival
1735c55b7e52SColin Percival# For any paths matching ${MERGECHANGES}, compare $2 against $1 and $3 and
1736c55b7e52SColin Percival# find any files with values unique to $2; generate $4 containing these paths
1737c55b7e52SColin Percival# and their corresponding hashes from $1.
1738db6b0a61SColin Percivalfetch_filter_mergechanges () {
1739db6b0a61SColin Percival	# Pull out the paths and hashes of the files matching ${MERGECHANGES}.
1740c55b7e52SColin Percival	for F in $1 $2 $3; do
1741db6b0a61SColin Percival		for X in ${MERGECHANGES}; do
1742db6b0a61SColin Percival			grep -E "^${X}" ${F}
1743db6b0a61SColin Percival		done |
1744db6b0a61SColin Percival		    cut -f 1,2,7 -d '|' |
1745db6b0a61SColin Percival		    sort > ${F}-values
1746db6b0a61SColin Percival	done
1747db6b0a61SColin Percival
1748c55b7e52SColin Percival	# Any line in $2-values which doesn't appear in $1-values or $3-values
1749c55b7e52SColin Percival	# and is a file means that we should list the path in $3.
1750c55b7e52SColin Percival	sort $1-values $3-values |
1751c55b7e52SColin Percival	    comm -13 - $2-values |
1752db6b0a61SColin Percival	    fgrep '|f|' |
1753db6b0a61SColin Percival	    cut -f 1 -d '|' > $2-paths
1754db6b0a61SColin Percival
1755db6b0a61SColin Percival	# For each path, pull out one (and only one!) entry from $1-values.
1756db6b0a61SColin Percival	# Note that we cannot distinguish which "old" version the user made
1757db6b0a61SColin Percival	# changes to; but hopefully any changes which occur due to security
1758db6b0a61SColin Percival	# updates will exist in both the "new" version and the version which
1759db6b0a61SColin Percival	# the user has installed, so the merging will still work.
1760db6b0a61SColin Percival	while read X; do
1761db6b0a61SColin Percival		look "${X}|" $1-values |
1762db6b0a61SColin Percival		    head -1
1763c55b7e52SColin Percival	done < $2-paths > $4
1764db6b0a61SColin Percival
1765db6b0a61SColin Percival	# Clean up
1766c55b7e52SColin Percival	rm $1-values $2-values $3-values $2-paths
1767db6b0a61SColin Percival}
1768db6b0a61SColin Percival
176948ffe56aSColin Percival# For any paths matching ${UPDATEIFUNMODIFIED}, remove lines from $[123]
1770db6b0a61SColin Percival# which correspond to lines in $2 with hashes not matching $1 or $3, unless
1771db6b0a61SColin Percival# the paths are listed in $4.  For entries in $2 marked "not present"
1772db6b0a61SColin Percival# (aka. type -), remove lines from $[123] unless there is a corresponding
1773db6b0a61SColin Percival# entry in $1.
177448ffe56aSColin Percivalfetch_filter_unmodified_notpresent () {
177548ffe56aSColin Percival	# Figure out which lines of $1 and $3 correspond to bits which
177648ffe56aSColin Percival	# should only be updated if they haven't changed, and fish out
177748ffe56aSColin Percival	# the (path, type, value) tuples.
177848ffe56aSColin Percival	# NOTE: We don't consider a file to be "modified" if it matches
177948ffe56aSColin Percival	# the hash from $3.
178048ffe56aSColin Percival	for X in ${UPDATEIFUNMODIFIED}; do
178148ffe56aSColin Percival		grep -E "^${X}" $1
178248ffe56aSColin Percival		grep -E "^${X}" $3
178348ffe56aSColin Percival	done |
178448ffe56aSColin Percival	    cut -f 1,2,7 -d '|' |
178548ffe56aSColin Percival	    sort > $1-values
178648ffe56aSColin Percival
178748ffe56aSColin Percival	# Do the same for $2.
178848ffe56aSColin Percival	for X in ${UPDATEIFUNMODIFIED}; do
178948ffe56aSColin Percival		grep -E "^${X}" $2
179048ffe56aSColin Percival	done |
179148ffe56aSColin Percival	    cut -f 1,2,7 -d '|' |
179248ffe56aSColin Percival	    sort > $2-values
179348ffe56aSColin Percival
179448ffe56aSColin Percival	# Any entry in $2-values which is not in $1-values corresponds to
1795db6b0a61SColin Percival	# a path which we need to remove from $1, $2, and $3, unless it
1796db6b0a61SColin Percival	# that path appears in $4.
1797db6b0a61SColin Percival	comm -13 $1-values $2-values |
1798db6b0a61SColin Percival	    sort -t '|' -k 1,1 > mlines.tmp
1799db6b0a61SColin Percival	cut -f 1 -d '|' $4 |
1800db6b0a61SColin Percival	    sort |
1801db6b0a61SColin Percival	    join -v 2 -t '|' - mlines.tmp |
1802db6b0a61SColin Percival	    sort > mlines
1803db6b0a61SColin Percival	rm $1-values $2-values mlines.tmp
180448ffe56aSColin Percival
180548ffe56aSColin Percival	# Any lines in $2 which are not in $1 AND are "not present" lines
180648ffe56aSColin Percival	# also belong in mlines.
180748ffe56aSColin Percival	comm -13 $1 $2 |
180848ffe56aSColin Percival	    cut -f 1,2,7 -d '|' |
180948ffe56aSColin Percival	    fgrep '|-|' >> mlines
181048ffe56aSColin Percival
181148ffe56aSColin Percival	# Remove lines from $1, $2, and $3
181248ffe56aSColin Percival	for X in $1 $2 $3; do
181348ffe56aSColin Percival		sort -t '|' -k 1,1 ${X} > ${X}.tmp
181448ffe56aSColin Percival		cut -f 1 -d '|' < mlines |
181548ffe56aSColin Percival		    sort |
181648ffe56aSColin Percival		    join -v 2 -t '|' - ${X}.tmp |
181748ffe56aSColin Percival		    sort > ${X}
181848ffe56aSColin Percival		rm ${X}.tmp
181948ffe56aSColin Percival	done
182048ffe56aSColin Percival
182148ffe56aSColin Percival	# Store a list of the modified files, for future reference
182248ffe56aSColin Percival	fgrep -v '|-|' mlines |
182348ffe56aSColin Percival	    cut -f 1 -d '|' > modifiedfiles
182448ffe56aSColin Percival	rm mlines
182548ffe56aSColin Percival}
182648ffe56aSColin Percival
182748ffe56aSColin Percival# For each entry in $1 of type -, remove any corresponding
182848ffe56aSColin Percival# entry from $2 if ${ALLOWADD} != "yes".  Remove all entries
182948ffe56aSColin Percival# of type - from $1.
183048ffe56aSColin Percivalfetch_filter_allowadd () {
183148ffe56aSColin Percival	cut -f 1,2 -d '|' < $1 |
183248ffe56aSColin Percival	    fgrep '|-' |
183348ffe56aSColin Percival	    cut -f 1 -d '|' > filesnotpresent
183448ffe56aSColin Percival
183548ffe56aSColin Percival	if [ ${ALLOWADD} != "yes" ]; then
183648ffe56aSColin Percival		sort < $2 |
183748ffe56aSColin Percival		    join -v 1 -t '|' - filesnotpresent |
183848ffe56aSColin Percival		    sort > $2.tmp
183948ffe56aSColin Percival		mv $2.tmp $2
184048ffe56aSColin Percival	fi
184148ffe56aSColin Percival
184248ffe56aSColin Percival	sort < $1 |
184348ffe56aSColin Percival	    join -v 1 -t '|' - filesnotpresent |
184448ffe56aSColin Percival	    sort > $1.tmp
184548ffe56aSColin Percival	mv $1.tmp $1
184648ffe56aSColin Percival	rm filesnotpresent
184748ffe56aSColin Percival}
184848ffe56aSColin Percival
184948ffe56aSColin Percival# If ${ALLOWDELETE} != "yes", then remove any entries from $1
185048ffe56aSColin Percival# which don't correspond to entries in $2.
185148ffe56aSColin Percivalfetch_filter_allowdelete () {
185248ffe56aSColin Percival	# Produce a lists ${PATH}|${TYPE}
185348ffe56aSColin Percival	for X in $1 $2; do
185448ffe56aSColin Percival		cut -f 1-2 -d '|' < ${X} |
185548ffe56aSColin Percival		    sort -u > ${X}.nodes
185648ffe56aSColin Percival	done
185748ffe56aSColin Percival
185848ffe56aSColin Percival	# Figure out which lines need to be removed from $1.
185948ffe56aSColin Percival	if [ ${ALLOWDELETE} != "yes" ]; then
186048ffe56aSColin Percival		comm -23 $1.nodes $2.nodes > $1.badnodes
186148ffe56aSColin Percival	else
186248ffe56aSColin Percival		: > $1.badnodes
186348ffe56aSColin Percival	fi
186448ffe56aSColin Percival
186548ffe56aSColin Percival	# Remove the relevant lines from $1
186648ffe56aSColin Percival	while read X; do
186748ffe56aSColin Percival		look "${X}|" $1
186848ffe56aSColin Percival	done < $1.badnodes |
186948ffe56aSColin Percival	    comm -13 - $1 > $1.tmp
187048ffe56aSColin Percival	mv $1.tmp $1
187148ffe56aSColin Percival
187248ffe56aSColin Percival	rm $1.badnodes $1.nodes $2.nodes
187348ffe56aSColin Percival}
187448ffe56aSColin Percival
187548ffe56aSColin Percival# If ${KEEPMODIFIEDMETADATA} == "yes", then for each entry in $2
187648ffe56aSColin Percival# with metadata not matching any entry in $1, replace the corresponding
187748ffe56aSColin Percival# line of $3 with one having the same metadata as the entry in $2.
187848ffe56aSColin Percivalfetch_filter_modified_metadata () {
187948ffe56aSColin Percival	# Fish out the metadata from $1 and $2
188048ffe56aSColin Percival	for X in $1 $2; do
188148ffe56aSColin Percival		cut -f 1-6 -d '|' < ${X} > ${X}.metadata
188248ffe56aSColin Percival	done
188348ffe56aSColin Percival
188448ffe56aSColin Percival	# Find the metadata we need to keep
188548ffe56aSColin Percival	if [ ${KEEPMODIFIEDMETADATA} = "yes" ]; then
188648ffe56aSColin Percival		comm -13 $1.metadata $2.metadata > keepmeta
188748ffe56aSColin Percival	else
188848ffe56aSColin Percival		: > keepmeta
188948ffe56aSColin Percival	fi
189048ffe56aSColin Percival
189148ffe56aSColin Percival	# Extract the lines which we need to remove from $3, and
189248ffe56aSColin Percival	# construct the lines which we need to add to $3.
189348ffe56aSColin Percival	: > $3.remove
189448ffe56aSColin Percival	: > $3.add
189548ffe56aSColin Percival	while read LINE; do
189648ffe56aSColin Percival		NODE=`echo "${LINE}" | cut -f 1-2 -d '|'`
189748ffe56aSColin Percival		look "${NODE}|" $3 >> $3.remove
189848ffe56aSColin Percival		look "${NODE}|" $3 |
189948ffe56aSColin Percival		    cut -f 7- -d '|' |
190048ffe56aSColin Percival		    lam -s "${LINE}|" - >> $3.add
190148ffe56aSColin Percival	done < keepmeta
190248ffe56aSColin Percival
190348ffe56aSColin Percival	# Remove the specified lines and add the new lines.
190448ffe56aSColin Percival	sort $3.remove |
190548ffe56aSColin Percival	    comm -13 - $3 |
190648ffe56aSColin Percival	    sort -u - $3.add > $3.tmp
190748ffe56aSColin Percival	mv $3.tmp $3
190848ffe56aSColin Percival
190948ffe56aSColin Percival	rm keepmeta $1.metadata $2.metadata $3.add $3.remove
191048ffe56aSColin Percival}
191148ffe56aSColin Percival
191248ffe56aSColin Percival# Remove lines from $1 and $2 which are identical;
191348ffe56aSColin Percival# no need to update a file if it isn't changing.
191448ffe56aSColin Percivalfetch_filter_uptodate () {
191548ffe56aSColin Percival	comm -23 $1 $2 > $1.tmp
191648ffe56aSColin Percival	comm -13 $1 $2 > $2.tmp
191748ffe56aSColin Percival
191848ffe56aSColin Percival	mv $1.tmp $1
191948ffe56aSColin Percival	mv $2.tmp $2
192048ffe56aSColin Percival}
192148ffe56aSColin Percival
1922db6b0a61SColin Percival# Fetch any "clean" old versions of files we need for merging changes.
1923db6b0a61SColin Percivalfetch_files_premerge () {
1924db6b0a61SColin Percival	# We only need to do anything if $1 is non-empty.
1925db6b0a61SColin Percival	if [ -s $1 ]; then
1926db6b0a61SColin Percival		# Tell the user what we're doing
1927db6b0a61SColin Percival		echo -n "Fetching files from ${OLDRELNUM} for merging... "
1928db6b0a61SColin Percival
1929db6b0a61SColin Percival		# List of files wanted
1930db6b0a61SColin Percival		fgrep '|f|' < $1 |
1931db6b0a61SColin Percival		    cut -f 3 -d '|' |
1932db6b0a61SColin Percival		    sort -u > files.wanted
1933db6b0a61SColin Percival
1934db6b0a61SColin Percival		# Only fetch the files we don't already have
1935db6b0a61SColin Percival		while read Y; do
1936db6b0a61SColin Percival			if [ ! -f "files/${Y}.gz" ]; then
1937db6b0a61SColin Percival				echo ${Y};
1938db6b0a61SColin Percival			fi
1939db6b0a61SColin Percival		done < files.wanted > filelist
1940db6b0a61SColin Percival
1941db6b0a61SColin Percival		# Actually fetch them
1942db6b0a61SColin Percival		lam -s "${OLDFETCHDIR}/f/" - -s ".gz" < filelist |
1943db6b0a61SColin Percival		    xargs ${XARGST} ${PHTTPGET} ${SERVERNAME}	\
1944db6b0a61SColin Percival		    2>${QUIETREDIR}
1945db6b0a61SColin Percival
1946db6b0a61SColin Percival		# Make sure we got them all, and move them into /files/
1947db6b0a61SColin Percival		while read Y; do
1948db6b0a61SColin Percival			if ! [ -f ${Y}.gz ]; then
1949db6b0a61SColin Percival				echo "failed."
1950db6b0a61SColin Percival				return 1
1951db6b0a61SColin Percival			fi
1952db6b0a61SColin Percival			if [ `gunzip -c < ${Y}.gz |
1953db6b0a61SColin Percival			    ${SHA256} -q` = ${Y} ]; then
1954db6b0a61SColin Percival				mv ${Y}.gz files/${Y}.gz
1955db6b0a61SColin Percival			else
1956db6b0a61SColin Percival				echo "${Y} has incorrect hash."
1957db6b0a61SColin Percival				return 1
1958db6b0a61SColin Percival			fi
1959db6b0a61SColin Percival		done < filelist
1960db6b0a61SColin Percival		echo "done."
1961db6b0a61SColin Percival
1962db6b0a61SColin Percival		# Clean up
1963db6b0a61SColin Percival		rm filelist files.wanted
1964db6b0a61SColin Percival	fi
1965db6b0a61SColin Percival}
1966db6b0a61SColin Percival
196748ffe56aSColin Percival# Prepare to fetch files: Generate a list of the files we need,
196848ffe56aSColin Percival# copy the unmodified files we have into /files/, and generate
196948ffe56aSColin Percival# a list of patches to download.
197048ffe56aSColin Percivalfetch_files_prepare () {
197148ffe56aSColin Percival	# Tell the user why his disk is suddenly making lots of noise
197248ffe56aSColin Percival	echo -n "Preparing to download files... "
197348ffe56aSColin Percival
197448ffe56aSColin Percival	# Reduce indices to ${PATH}|${HASH} pairs
197548ffe56aSColin Percival	for X in $1 $2 $3; do
197648ffe56aSColin Percival		cut -f 1,2,7 -d '|' < ${X} |
197748ffe56aSColin Percival		    fgrep '|f|' |
197848ffe56aSColin Percival		    cut -f 1,3 -d '|' |
197948ffe56aSColin Percival		    sort > ${X}.hashes
198048ffe56aSColin Percival	done
198148ffe56aSColin Percival
198248ffe56aSColin Percival	# List of files wanted
198348ffe56aSColin Percival	cut -f 2 -d '|' < $3.hashes |
19842328d598SColin Percival	    sort -u |
19852328d598SColin Percival	    while read HASH; do
19862328d598SColin Percival		if ! [ -f files/${HASH}.gz ]; then
19872328d598SColin Percival			echo ${HASH}
19882328d598SColin Percival		fi
19892328d598SColin Percival	done > files.wanted
199048ffe56aSColin Percival
199148ffe56aSColin Percival	# Generate a list of unmodified files
199248ffe56aSColin Percival	comm -12 $1.hashes $2.hashes |
199348ffe56aSColin Percival	    sort -k 1,1 -t '|' > unmodified.files
199448ffe56aSColin Percival
199548ffe56aSColin Percival	# Copy all files into /files/.  We only need the unmodified files
199648ffe56aSColin Percival	# for use in patching; but we'll want all of them if the user asks
199748ffe56aSColin Percival	# to rollback the updates later.
1998210b8123SColin Percival	while read LINE; do
1999210b8123SColin Percival		F=`echo "${LINE}" | cut -f 1 -d '|'`
2000210b8123SColin Percival		HASH=`echo "${LINE}" | cut -f 2 -d '|'`
2001210b8123SColin Percival
2002210b8123SColin Percival		# Skip files we already have.
2003210b8123SColin Percival		if [ -f files/${HASH}.gz ]; then
2004210b8123SColin Percival			continue
2005210b8123SColin Percival		fi
2006210b8123SColin Percival
2007210b8123SColin Percival		# Make sure the file hasn't changed.
200848ffe56aSColin Percival		cp "${BASEDIR}/${F}" tmpfile
2009210b8123SColin Percival		if [ `sha256 -q tmpfile` != ${HASH} ]; then
2010210b8123SColin Percival			echo
2011210b8123SColin Percival			echo "File changed while FreeBSD Update running: ${F}"
2012210b8123SColin Percival			return 1
2013210b8123SColin Percival		fi
2014210b8123SColin Percival
2015210b8123SColin Percival		# Place the file into storage.
2016210b8123SColin Percival		gzip -c < tmpfile > files/${HASH}.gz
201748ffe56aSColin Percival		rm tmpfile
2018210b8123SColin Percival	done < $2.hashes
201948ffe56aSColin Percival
202048ffe56aSColin Percival	# Produce a list of patches to download
202148ffe56aSColin Percival	sort -k 1,1 -t '|' $3.hashes |
202248ffe56aSColin Percival	    join -t '|' -o 2.2,1.2 - unmodified.files |
202348ffe56aSColin Percival	    fetch_make_patchlist > patchlist
202448ffe56aSColin Percival
202548ffe56aSColin Percival	# Garbage collect
202648ffe56aSColin Percival	rm unmodified.files $1.hashes $2.hashes $3.hashes
202748ffe56aSColin Percival
202848ffe56aSColin Percival	# We don't need the list of possible old files any more.
202948ffe56aSColin Percival	rm $1
203048ffe56aSColin Percival
203148ffe56aSColin Percival	# We're finished making noise
203248ffe56aSColin Percival	echo "done."
203348ffe56aSColin Percival}
203448ffe56aSColin Percival
203548ffe56aSColin Percival# Fetch files.
203648ffe56aSColin Percivalfetch_files () {
203748ffe56aSColin Percival	# Attempt to fetch patches
203848ffe56aSColin Percival	if [ -s patchlist ]; then
203948ffe56aSColin Percival		echo -n "Fetching `wc -l < patchlist | tr -d ' '` "
204048ffe56aSColin Percival		echo ${NDEBUG} "patches.${DDSTATS}"
204148ffe56aSColin Percival		tr '|' '-' < patchlist |
2042db6b0a61SColin Percival		    lam -s "${PATCHDIR}/" - |
204348ffe56aSColin Percival		    xargs ${XARGST} ${PHTTPGET} ${SERVERNAME}	\
204448ffe56aSColin Percival			2>${STATSREDIR} | fetch_progress
204548ffe56aSColin Percival		echo "done."
204648ffe56aSColin Percival
204748ffe56aSColin Percival		# Attempt to apply patches
204848ffe56aSColin Percival		echo -n "Applying patches... "
204948ffe56aSColin Percival		tr '|' ' ' < patchlist |
205048ffe56aSColin Percival		    while read X Y; do
205148ffe56aSColin Percival			if [ ! -f "${X}-${Y}" ]; then continue; fi
205248ffe56aSColin Percival			gunzip -c < files/${X}.gz > OLD
205348ffe56aSColin Percival
205448ffe56aSColin Percival			bspatch OLD NEW ${X}-${Y}
205548ffe56aSColin Percival
205648ffe56aSColin Percival			if [ `${SHA256} -q NEW` = ${Y} ]; then
205748ffe56aSColin Percival				mv NEW files/${Y}
205848ffe56aSColin Percival				gzip -n files/${Y}
205948ffe56aSColin Percival			fi
206048ffe56aSColin Percival			rm -f diff OLD NEW ${X}-${Y}
206148ffe56aSColin Percival		done 2>${QUIETREDIR}
206248ffe56aSColin Percival		echo "done."
206348ffe56aSColin Percival	fi
206448ffe56aSColin Percival
206548ffe56aSColin Percival	# Download files which couldn't be generate via patching
206648ffe56aSColin Percival	while read Y; do
206748ffe56aSColin Percival		if [ ! -f "files/${Y}.gz" ]; then
206848ffe56aSColin Percival			echo ${Y};
206948ffe56aSColin Percival		fi
207048ffe56aSColin Percival	done < files.wanted > filelist
207148ffe56aSColin Percival
207248ffe56aSColin Percival	if [ -s filelist ]; then
207348ffe56aSColin Percival		echo -n "Fetching `wc -l < filelist | tr -d ' '` "
207448ffe56aSColin Percival		echo ${NDEBUG} "files... "
207548ffe56aSColin Percival		lam -s "${FETCHDIR}/f/" - -s ".gz" < filelist |
207648ffe56aSColin Percival		    xargs ${XARGST} ${PHTTPGET} ${SERVERNAME}	\
2077d6e1e31aSConrad Meyer			2>${STATSREDIR} | fetch_progress
207848ffe56aSColin Percival
207948ffe56aSColin Percival		while read Y; do
208048ffe56aSColin Percival			if ! [ -f ${Y}.gz ]; then
208148ffe56aSColin Percival				echo "failed."
208248ffe56aSColin Percival				return 1
208348ffe56aSColin Percival			fi
208448ffe56aSColin Percival			if [ `gunzip -c < ${Y}.gz |
208548ffe56aSColin Percival			    ${SHA256} -q` = ${Y} ]; then
208648ffe56aSColin Percival				mv ${Y}.gz files/${Y}.gz
208748ffe56aSColin Percival			else
208848ffe56aSColin Percival				echo "${Y} has incorrect hash."
208948ffe56aSColin Percival				return 1
209048ffe56aSColin Percival			fi
209148ffe56aSColin Percival		done < filelist
209248ffe56aSColin Percival		echo "done."
209348ffe56aSColin Percival	fi
209448ffe56aSColin Percival
209548ffe56aSColin Percival	# Clean up
209648ffe56aSColin Percival	rm files.wanted filelist patchlist
209748ffe56aSColin Percival}
209848ffe56aSColin Percival
209948ffe56aSColin Percival# Create and populate install manifest directory; and report what updates
210048ffe56aSColin Percival# are available.
210148ffe56aSColin Percivalfetch_create_manifest () {
210248ffe56aSColin Percival	# If we have an existing install manifest, nuke it.
210348ffe56aSColin Percival	if [ -L "${BDHASH}-install" ]; then
210448ffe56aSColin Percival		rm -r ${BDHASH}-install/
210548ffe56aSColin Percival		rm ${BDHASH}-install
210648ffe56aSColin Percival	fi
210748ffe56aSColin Percival
210848ffe56aSColin Percival	# Report to the user if any updates were avoided due to local changes
210948ffe56aSColin Percival	if [ -s modifiedfiles ]; then
2110fc24ba59SEd Maste		cat - modifiedfiles <<- EOF | ${PAGER}
2111b882e02bSEnji Cooper			The following files are affected by updates. No changes have
2112b882e02bSEnji Cooper			been downloaded, however, because the files have been modified
2113b882e02bSEnji Cooper			locally:
2114fc24ba59SEd Maste		EOF
2115fc24ba59SEd Maste	fi
211648ffe56aSColin Percival	rm modifiedfiles
211748ffe56aSColin Percival
211848ffe56aSColin Percival	# If no files will be updated, tell the user and exit
211948ffe56aSColin Percival	if ! [ -s INDEX-PRESENT ] &&
212048ffe56aSColin Percival	    ! [ -s INDEX-NEW ]; then
212148ffe56aSColin Percival		rm INDEX-PRESENT INDEX-NEW
212248ffe56aSColin Percival		echo
212348ffe56aSColin Percival		echo -n "No updates needed to update system to "
212448ffe56aSColin Percival		echo "${RELNUM}-p${RELPATCHNUM}."
212548ffe56aSColin Percival		return
212648ffe56aSColin Percival	fi
212748ffe56aSColin Percival
212848ffe56aSColin Percival	# Divide files into (a) removed files, (b) added files, and
212948ffe56aSColin Percival	# (c) updated files.
213048ffe56aSColin Percival	cut -f 1 -d '|' < INDEX-PRESENT |
213148ffe56aSColin Percival	    sort > INDEX-PRESENT.flist
213248ffe56aSColin Percival	cut -f 1 -d '|' < INDEX-NEW |
213348ffe56aSColin Percival	    sort > INDEX-NEW.flist
213448ffe56aSColin Percival	comm -23 INDEX-PRESENT.flist INDEX-NEW.flist > files.removed
213548ffe56aSColin Percival	comm -13 INDEX-PRESENT.flist INDEX-NEW.flist > files.added
213648ffe56aSColin Percival	comm -12 INDEX-PRESENT.flist INDEX-NEW.flist > files.updated
213748ffe56aSColin Percival	rm INDEX-PRESENT.flist INDEX-NEW.flist
213848ffe56aSColin Percival
213948ffe56aSColin Percival	# Report removed files, if any
214048ffe56aSColin Percival	if [ -s files.removed ]; then
2141fc24ba59SEd Maste		cat - files.removed <<- EOF | ${PAGER}
2142fc24ba59SEd Maste			The following files will be removed as part of updating to
2143fc24ba59SEd Maste			${RELNUM}-p${RELPATCHNUM}:
2144fc24ba59SEd Maste		EOF
2145fc24ba59SEd Maste	fi
214648ffe56aSColin Percival	rm files.removed
214748ffe56aSColin Percival
214848ffe56aSColin Percival	# Report added files, if any
214948ffe56aSColin Percival	if [ -s files.added ]; then
2150fc24ba59SEd Maste		cat - files.added <<- EOF | ${PAGER}
2151fc24ba59SEd Maste			The following files will be added as part of updating to
2152fc24ba59SEd Maste			${RELNUM}-p${RELPATCHNUM}:
2153fc24ba59SEd Maste		EOF
2154fc24ba59SEd Maste	fi
215548ffe56aSColin Percival	rm files.added
215648ffe56aSColin Percival
215748ffe56aSColin Percival	# Report updated files, if any
215848ffe56aSColin Percival	if [ -s files.updated ]; then
2159fc24ba59SEd Maste		cat - files.updated <<- EOF | ${PAGER}
2160fc24ba59SEd Maste			The following files will be updated as part of updating to
2161fc24ba59SEd Maste			${RELNUM}-p${RELPATCHNUM}:
2162fc24ba59SEd Maste		EOF
2163fc24ba59SEd Maste	fi
216448ffe56aSColin Percival	rm files.updated
216548ffe56aSColin Percival
216648ffe56aSColin Percival	# Create a directory for the install manifest.
216748ffe56aSColin Percival	MDIR=`mktemp -d install.XXXXXX` || return 1
216848ffe56aSColin Percival
216948ffe56aSColin Percival	# Populate it
217048ffe56aSColin Percival	mv INDEX-PRESENT ${MDIR}/INDEX-OLD
217148ffe56aSColin Percival	mv INDEX-NEW ${MDIR}/INDEX-NEW
217248ffe56aSColin Percival
217348ffe56aSColin Percival	# Link it into place
217448ffe56aSColin Percival	ln -s ${MDIR} ${BDHASH}-install
217548ffe56aSColin Percival}
217648ffe56aSColin Percival
217748ffe56aSColin Percival# Warn about any upcoming EoL
217848ffe56aSColin Percivalfetch_warn_eol () {
217948ffe56aSColin Percival	# What's the current time?
218048ffe56aSColin Percival	NOWTIME=`date "+%s"`
218148ffe56aSColin Percival
218248ffe56aSColin Percival	# When did we last warn about the EoL date?
218348ffe56aSColin Percival	if [ -f lasteolwarn ]; then
218448ffe56aSColin Percival		LASTWARN=`cat lasteolwarn`
218548ffe56aSColin Percival	else
218648ffe56aSColin Percival		LASTWARN=`expr ${NOWTIME} - 63072000`
218748ffe56aSColin Percival	fi
218848ffe56aSColin Percival
218948ffe56aSColin Percival	# If the EoL time is past, warn.
219048ffe56aSColin Percival	if [ ${EOLTIME} -lt ${NOWTIME} ]; then
219148ffe56aSColin Percival		echo
219248ffe56aSColin Percival		cat <<-EOF
2193b698a3abSColin Percival		WARNING: `uname -sr` HAS PASSED ITS END-OF-LIFE DATE.
219448ffe56aSColin Percival		Any security issues discovered after `date -r ${EOLTIME}`
219548ffe56aSColin Percival		will not have been corrected.
219648ffe56aSColin Percival		EOF
219748ffe56aSColin Percival		return 1
219848ffe56aSColin Percival	fi
219948ffe56aSColin Percival
220048ffe56aSColin Percival	# Figure out how long it has been since we last warned about the
220148ffe56aSColin Percival	# upcoming EoL, and how much longer we have left.
220248ffe56aSColin Percival	SINCEWARN=`expr ${NOWTIME} - ${LASTWARN}`
220348ffe56aSColin Percival	TIMELEFT=`expr ${EOLTIME} - ${NOWTIME}`
220448ffe56aSColin Percival
220589b14566SColin Percival	# Don't warn if the EoL is more than 3 months away
220689b14566SColin Percival	if [ ${TIMELEFT} -gt 7884000 ]; then
220748ffe56aSColin Percival		return 0
220848ffe56aSColin Percival	fi
220948ffe56aSColin Percival
221048ffe56aSColin Percival	# Don't warn if the time remaining is more than 3 times the time
221148ffe56aSColin Percival	# since the last warning.
221248ffe56aSColin Percival	if [ ${TIMELEFT} -gt `expr ${SINCEWARN} \* 3` ]; then
221348ffe56aSColin Percival		return 0
221448ffe56aSColin Percival	fi
221548ffe56aSColin Percival
221648ffe56aSColin Percival	# Figure out what time units to use.
221748ffe56aSColin Percival	if [ ${TIMELEFT} -lt 604800 ]; then
221848ffe56aSColin Percival		UNIT="day"
221948ffe56aSColin Percival		SIZE=86400
222048ffe56aSColin Percival	elif [ ${TIMELEFT} -lt 2678400 ]; then
222148ffe56aSColin Percival		UNIT="week"
222248ffe56aSColin Percival		SIZE=604800
222348ffe56aSColin Percival	else
222448ffe56aSColin Percival		UNIT="month"
222548ffe56aSColin Percival		SIZE=2678400
222648ffe56aSColin Percival	fi
222748ffe56aSColin Percival
222848ffe56aSColin Percival	# Compute the right number of units
222948ffe56aSColin Percival	NUM=`expr ${TIMELEFT} / ${SIZE}`
223048ffe56aSColin Percival	if [ ${NUM} != 1 ]; then
223148ffe56aSColin Percival		UNIT="${UNIT}s"
223248ffe56aSColin Percival	fi
223348ffe56aSColin Percival
223448ffe56aSColin Percival	# Print the warning
223548ffe56aSColin Percival	echo
223648ffe56aSColin Percival	cat <<-EOF
223748ffe56aSColin Percival		WARNING: `uname -sr` is approaching its End-of-Life date.
223848ffe56aSColin Percival		It is strongly recommended that you upgrade to a newer
223948ffe56aSColin Percival		release within the next ${NUM} ${UNIT}.
224048ffe56aSColin Percival	EOF
224148ffe56aSColin Percival
224248ffe56aSColin Percival	# Update the stored time of last warning
224348ffe56aSColin Percival	echo ${NOWTIME} > lasteolwarn
224448ffe56aSColin Percival}
224548ffe56aSColin Percival
224648ffe56aSColin Percival# Do the actual work involved in "fetch" / "cron".
224748ffe56aSColin Percivalfetch_run () {
224848ffe56aSColin Percival	workdir_init || return 1
224948ffe56aSColin Percival
225048ffe56aSColin Percival	# Prepare the mirror list.
225148ffe56aSColin Percival	fetch_pick_server_init && fetch_pick_server
225248ffe56aSColin Percival
225348ffe56aSColin Percival	# Try to fetch the public key until we run out of servers.
225448ffe56aSColin Percival	while ! fetch_key; do
225548ffe56aSColin Percival		fetch_pick_server || return 1
225648ffe56aSColin Percival	done
225748ffe56aSColin Percival
225848ffe56aSColin Percival	# Try to fetch the metadata index signature ("tag") until we run
225948ffe56aSColin Percival	# out of available servers; and sanity check the downloaded tag.
226048ffe56aSColin Percival	while ! fetch_tag; do
226148ffe56aSColin Percival		fetch_pick_server || return 1
226248ffe56aSColin Percival	done
226348ffe56aSColin Percival	fetch_tagsanity || return 1
226448ffe56aSColin Percival
226548ffe56aSColin Percival	# Fetch the latest INDEX-NEW and INDEX-OLD files.
226648ffe56aSColin Percival	fetch_metadata INDEX-NEW INDEX-OLD || return 1
226748ffe56aSColin Percival
226848ffe56aSColin Percival	# Generate filtered INDEX-NEW and INDEX-OLD files containing only
226948ffe56aSColin Percival	# the lines which (a) belong to components we care about, and (b)
227048ffe56aSColin Percival	# don't correspond to paths we're explicitly ignoring.
227148ffe56aSColin Percival	fetch_filter_metadata INDEX-NEW || return 1
227248ffe56aSColin Percival	fetch_filter_metadata INDEX-OLD || return 1
227348ffe56aSColin Percival
2274db6b0a61SColin Percival	# Translate /boot/${KERNCONF} into ${KERNELDIR}
2275db6b0a61SColin Percival	fetch_filter_kernel_names INDEX-NEW ${KERNCONF}
2276db6b0a61SColin Percival	fetch_filter_kernel_names INDEX-OLD ${KERNCONF}
227748ffe56aSColin Percival
227848ffe56aSColin Percival	# For all paths appearing in INDEX-OLD or INDEX-NEW, inspect the
227948ffe56aSColin Percival	# system and generate an INDEX-PRESENT file.
228048ffe56aSColin Percival	fetch_inspect_system INDEX-OLD INDEX-PRESENT INDEX-NEW || return 1
228148ffe56aSColin Percival
228248ffe56aSColin Percival	# Based on ${UPDATEIFUNMODIFIED}, remove lines from INDEX-* which
228348ffe56aSColin Percival	# correspond to lines in INDEX-PRESENT with hashes not appearing
228448ffe56aSColin Percival	# in INDEX-OLD or INDEX-NEW.  Also remove lines where the entry in
228548ffe56aSColin Percival	# INDEX-PRESENT has type - and there isn't a corresponding entry in
228648ffe56aSColin Percival	# INDEX-OLD with type -.
2287db6b0a61SColin Percival	fetch_filter_unmodified_notpresent	\
2288db6b0a61SColin Percival	    INDEX-OLD INDEX-PRESENT INDEX-NEW /dev/null
228948ffe56aSColin Percival
229048ffe56aSColin Percival	# For each entry in INDEX-PRESENT of type -, remove any corresponding
229148ffe56aSColin Percival	# entry from INDEX-NEW if ${ALLOWADD} != "yes".  Remove all entries
229248ffe56aSColin Percival	# of type - from INDEX-PRESENT.
229348ffe56aSColin Percival	fetch_filter_allowadd INDEX-PRESENT INDEX-NEW
229448ffe56aSColin Percival
229548ffe56aSColin Percival	# If ${ALLOWDELETE} != "yes", then remove any entries from
229648ffe56aSColin Percival	# INDEX-PRESENT which don't correspond to entries in INDEX-NEW.
229748ffe56aSColin Percival	fetch_filter_allowdelete INDEX-PRESENT INDEX-NEW
229848ffe56aSColin Percival
229948ffe56aSColin Percival	# If ${KEEPMODIFIEDMETADATA} == "yes", then for each entry in
230048ffe56aSColin Percival	# INDEX-PRESENT with metadata not matching any entry in INDEX-OLD,
230148ffe56aSColin Percival	# replace the corresponding line of INDEX-NEW with one having the
230248ffe56aSColin Percival	# same metadata as the entry in INDEX-PRESENT.
230348ffe56aSColin Percival	fetch_filter_modified_metadata INDEX-OLD INDEX-PRESENT INDEX-NEW
230448ffe56aSColin Percival
230548ffe56aSColin Percival	# Remove lines from INDEX-PRESENT and INDEX-NEW which are identical;
230648ffe56aSColin Percival	# no need to update a file if it isn't changing.
230748ffe56aSColin Percival	fetch_filter_uptodate INDEX-PRESENT INDEX-NEW
230848ffe56aSColin Percival
230948ffe56aSColin Percival	# Prepare to fetch files: Generate a list of the files we need,
231048ffe56aSColin Percival	# copy the unmodified files we have into /files/, and generate
231148ffe56aSColin Percival	# a list of patches to download.
2312210b8123SColin Percival	fetch_files_prepare INDEX-OLD INDEX-PRESENT INDEX-NEW || return 1
231348ffe56aSColin Percival
231448ffe56aSColin Percival	# Fetch files.
231548ffe56aSColin Percival	fetch_files || return 1
231648ffe56aSColin Percival
231748ffe56aSColin Percival	# Create and populate install manifest directory; and report what
231848ffe56aSColin Percival	# updates are available.
231948ffe56aSColin Percival	fetch_create_manifest || return 1
232048ffe56aSColin Percival
232148ffe56aSColin Percival	# Warn about any upcoming EoL
232248ffe56aSColin Percival	fetch_warn_eol || return 1
232348ffe56aSColin Percival}
232448ffe56aSColin Percival
2325db6b0a61SColin Percival# If StrictComponents is not "yes", generate a new components list
2326db6b0a61SColin Percival# with only the components which appear to be installed.
2327db6b0a61SColin Percivalupgrade_guess_components () {
2328db6b0a61SColin Percival	if [ "${STRICTCOMPONENTS}" = "no" ]; then
2329db6b0a61SColin Percival		# Generate filtered INDEX-ALL with only the components listed
2330db6b0a61SColin Percival		# in COMPONENTS.
2331db6b0a61SColin Percival		fetch_filter_metadata_components $1 || return 1
2332db6b0a61SColin Percival
2333db6b0a61SColin Percival		# Tell the user why his disk is suddenly making lots of noise
2334db6b0a61SColin Percival		echo -n "Inspecting system... "
2335db6b0a61SColin Percival
2336db6b0a61SColin Percival		# Look at the files on disk, and assume that a component is
2337db6b0a61SColin Percival		# supposed to be present if it is more than half-present.
2338db6b0a61SColin Percival		cut -f 1-3 -d '|' < INDEX-ALL |
2339db6b0a61SColin Percival		    tr '|' ' ' |
2340db6b0a61SColin Percival		    while read C S F; do
2341db6b0a61SColin Percival			if [ -e ${BASEDIR}/${F} ]; then
2342db6b0a61SColin Percival				echo "+ ${C}|${S}"
2343db6b0a61SColin Percival			fi
2344db6b0a61SColin Percival			echo "= ${C}|${S}"
2345db6b0a61SColin Percival		    done |
2346db6b0a61SColin Percival		    sort |
2347db6b0a61SColin Percival		    uniq -c |
2348db6b0a61SColin Percival		    sed -E 's,^ +,,' > compfreq
2349db6b0a61SColin Percival		grep ' = ' compfreq |
2350db6b0a61SColin Percival		    cut -f 1,3 -d ' ' |
2351db6b0a61SColin Percival		    sort -k 2,2 -t ' ' > compfreq.total
2352db6b0a61SColin Percival		grep ' + ' compfreq |
2353db6b0a61SColin Percival		    cut -f 1,3 -d ' ' |
2354db6b0a61SColin Percival		    sort -k 2,2 -t ' ' > compfreq.present
2355db6b0a61SColin Percival		join -t ' ' -1 2 -2 2 compfreq.present compfreq.total |
2356db6b0a61SColin Percival		    while read S P T; do
2357ae97aa98SEd Maste			if [ ${T} -ne 0 -a ${P} -gt `expr ${T} / 2` ]; then
2358db6b0a61SColin Percival				echo ${S}
2359db6b0a61SColin Percival			fi
2360db6b0a61SColin Percival		    done > comp.present
2361db6b0a61SColin Percival		cut -f 2 -d ' ' < compfreq.total > comp.total
2362db6b0a61SColin Percival		rm INDEX-ALL compfreq compfreq.total compfreq.present
2363db6b0a61SColin Percival
2364db6b0a61SColin Percival		# We're done making noise.
2365db6b0a61SColin Percival		echo "done."
2366db6b0a61SColin Percival
2367db6b0a61SColin Percival		# Sometimes the kernel isn't installed where INDEX-ALL
2368db6b0a61SColin Percival		# thinks that it should be: In particular, it is often in
2369db6b0a61SColin Percival		# /boot/kernel instead of /boot/GENERIC or /boot/SMP.  To
2370db6b0a61SColin Percival		# deal with this, if "kernel|X" is listed in comp.total
2371db6b0a61SColin Percival		# (i.e., is a component which would be upgraded if it is
2372db6b0a61SColin Percival		# found to be present) we will add it to comp.present.
2373db6b0a61SColin Percival		# If "kernel|<anything>" is in comp.total but "kernel|X" is
2374db6b0a61SColin Percival		# not, we print a warning -- the user is running a kernel
2375db6b0a61SColin Percival		# which isn't part of the release.
2376db6b0a61SColin Percival		KCOMP=`echo ${KERNCONF} | tr 'A-Z' 'a-z'`
2377db6b0a61SColin Percival		grep -E "^kernel\|${KCOMP}\$" comp.total >> comp.present
2378db6b0a61SColin Percival
2379db6b0a61SColin Percival		if grep -qE "^kernel\|" comp.total &&
2380db6b0a61SColin Percival		    ! grep -qE "^kernel\|${KCOMP}\$" comp.total; then
2381db6b0a61SColin Percival			cat <<-EOF
2382db6b0a61SColin Percival
2383db6b0a61SColin PercivalWARNING: This system is running a "${KCOMP}" kernel, which is not a
2384db6b0a61SColin Percivalkernel configuration distributed as part of FreeBSD ${RELNUM}.
2385db6b0a61SColin PercivalThis kernel will not be updated: you MUST update the kernel manually
238659b02bb4SMichael Osipovbefore running '`basename $0` [options] install'.
2387db6b0a61SColin Percival			EOF
2388db6b0a61SColin Percival		fi
2389db6b0a61SColin Percival
2390db6b0a61SColin Percival		# Re-sort the list of installed components and generate
2391db6b0a61SColin Percival		# the list of non-installed components.
2392db6b0a61SColin Percival		sort -u < comp.present > comp.present.tmp
2393db6b0a61SColin Percival		mv comp.present.tmp comp.present
2394db6b0a61SColin Percival		comm -13 comp.present comp.total > comp.absent
2395db6b0a61SColin Percival
2396db6b0a61SColin Percival		# Ask the user to confirm that what we have is correct.  To
2397db6b0a61SColin Percival		# reduce user confusion, translate "X|Y" back to "X/Y" (as
2398db6b0a61SColin Percival		# subcomponents must be listed in the configuration file).
2399db6b0a61SColin Percival		echo
2400db6b0a61SColin Percival		echo -n "The following components of FreeBSD "
2401db6b0a61SColin Percival		echo "seem to be installed:"
2402db6b0a61SColin Percival		tr '|' '/' < comp.present |
2403db6b0a61SColin Percival		    fmt -72
2404db6b0a61SColin Percival		echo
2405db6b0a61SColin Percival		echo -n "The following components of FreeBSD "
2406db6b0a61SColin Percival		echo "do not seem to be installed:"
2407db6b0a61SColin Percival		tr '|' '/' < comp.absent |
2408db6b0a61SColin Percival		    fmt -72
2409db6b0a61SColin Percival		echo
2410db6b0a61SColin Percival		continuep || return 1
2411db6b0a61SColin Percival		echo
2412db6b0a61SColin Percival
2413db6b0a61SColin Percival		# Suck the generated list of components into ${COMPONENTS}.
2414db6b0a61SColin Percival		# Note that comp.present.tmp is used due to issues with
2415db6b0a61SColin Percival		# pipelines and setting variables.
2416db6b0a61SColin Percival		COMPONENTS=""
2417db6b0a61SColin Percival		tr '|' '/' < comp.present > comp.present.tmp
2418db6b0a61SColin Percival		while read C; do
2419db6b0a61SColin Percival			COMPONENTS="${COMPONENTS} ${C}"
2420db6b0a61SColin Percival		done < comp.present.tmp
2421db6b0a61SColin Percival
2422db6b0a61SColin Percival		# Delete temporary files
2423db6b0a61SColin Percival		rm comp.present comp.present.tmp comp.absent comp.total
2424db6b0a61SColin Percival	fi
2425db6b0a61SColin Percival}
2426db6b0a61SColin Percival
2427db6b0a61SColin Percival# If StrictComponents is not "yes", COMPONENTS contains an entry
2428db6b0a61SColin Percival# corresponding to the currently running kernel, and said kernel
2429db6b0a61SColin Percival# does not exist in the new release, add "kernel/generic" to the
2430db6b0a61SColin Percival# list of components.
2431db6b0a61SColin Percivalupgrade_guess_new_kernel () {
2432db6b0a61SColin Percival	if [ "${STRICTCOMPONENTS}" = "no" ]; then
2433db6b0a61SColin Percival		# Grab the unfiltered metadata file.
2434db6b0a61SColin Percival		METAHASH=`look "$1|" tINDEX.present | cut -f 2 -d '|'`
2435db6b0a61SColin Percival		gunzip -c < files/${METAHASH}.gz > $1.all
2436db6b0a61SColin Percival
2437db6b0a61SColin Percival		# If "kernel/${KCOMP}" is in ${COMPONENTS} and that component
2438db6b0a61SColin Percival		# isn't in $1.all, we need to add kernel/generic.
2439db6b0a61SColin Percival		for C in ${COMPONENTS}; do
2440db6b0a61SColin Percival			if [ ${C} = "kernel/${KCOMP}" ] &&
2441db6b0a61SColin Percival			    ! grep -qE "^kernel\|${KCOMP}\|" $1.all; then
2442db6b0a61SColin Percival				COMPONENTS="${COMPONENTS} kernel/generic"
2443db6b0a61SColin Percival				NKERNCONF="GENERIC"
2444db6b0a61SColin Percival				cat <<-EOF
2445db6b0a61SColin Percival
2446db6b0a61SColin PercivalWARNING: This system is running a "${KCOMP}" kernel, which is not a
2447db6b0a61SColin Percivalkernel configuration distributed as part of FreeBSD ${RELNUM}.
2448db6b0a61SColin PercivalAs part of upgrading to FreeBSD ${RELNUM}, this kernel will be
2449db6b0a61SColin Percivalreplaced with a "generic" kernel.
2450db6b0a61SColin Percival				EOF
2451db6b0a61SColin Percival				continuep || return 1
2452db6b0a61SColin Percival			fi
2453db6b0a61SColin Percival		done
2454db6b0a61SColin Percival
2455db6b0a61SColin Percival		# Don't need this any more...
2456db6b0a61SColin Percival		rm $1.all
2457db6b0a61SColin Percival	fi
2458db6b0a61SColin Percival}
2459db6b0a61SColin Percival
2460db6b0a61SColin Percival# Convert INDEX-OLD (last release) and INDEX-ALL (new release) into
2461db6b0a61SColin Percival# INDEX-OLD and INDEX-NEW files (in the sense of normal upgrades).
2462db6b0a61SColin Percivalupgrade_oldall_to_oldnew () {
2463db6b0a61SColin Percival	# For each ${F}|... which appears in INDEX-ALL but does not appear
2464db6b0a61SColin Percival	# in INDEX-OLD, add ${F}|-|||||| to INDEX-OLD.
2465db6b0a61SColin Percival	cut -f 1 -d '|' < $1 |
2466db6b0a61SColin Percival	    sort -u > $1.paths
2467db6b0a61SColin Percival	cut -f 1 -d '|' < $2 |
2468db6b0a61SColin Percival	    sort -u |
2469db6b0a61SColin Percival	    comm -13 $1.paths - |
2470db6b0a61SColin Percival	    lam - -s "|-||||||" |
2471db6b0a61SColin Percival	    sort - $1 > $1.tmp
2472db6b0a61SColin Percival	mv $1.tmp $1
2473db6b0a61SColin Percival
2474db6b0a61SColin Percival	# Remove lines from INDEX-OLD which also appear in INDEX-ALL
2475db6b0a61SColin Percival	comm -23 $1 $2 > $1.tmp
2476db6b0a61SColin Percival	mv $1.tmp $1
2477db6b0a61SColin Percival
2478db6b0a61SColin Percival	# Remove lines from INDEX-ALL which have a file name not appearing
2479db6b0a61SColin Percival	# anywhere in INDEX-OLD (since these must be files which haven't
2480db6b0a61SColin Percival	# changed -- if they were new, there would be an entry of type "-").
2481db6b0a61SColin Percival	cut -f 1 -d '|' < $1 |
2482db6b0a61SColin Percival	    sort -u > $1.paths
2483db6b0a61SColin Percival	sort -k 1,1 -t '|' < $2 |
2484db6b0a61SColin Percival	    join -t '|' - $1.paths |
2485db6b0a61SColin Percival	    sort > $2.tmp
2486db6b0a61SColin Percival	rm $1.paths
2487db6b0a61SColin Percival	mv $2.tmp $2
2488db6b0a61SColin Percival
2489db6b0a61SColin Percival	# Rename INDEX-ALL to INDEX-NEW.
2490db6b0a61SColin Percival	mv $2 $3
2491db6b0a61SColin Percival}
2492db6b0a61SColin Percival
24937449d2f5SColin Percival# Helper for upgrade_merge: Return zero true iff the two files differ only
24946d514f10SDag-Erling Smørgrav# in the contents of their RCS tags.
24957449d2f5SColin Percivalsamef () {
24967449d2f5SColin Percival	X=`sed -E 's/\\$FreeBSD.*\\$/\$FreeBSD\$/' < $1 | ${SHA256}`
24977449d2f5SColin Percival	Y=`sed -E 's/\\$FreeBSD.*\\$/\$FreeBSD\$/' < $2 | ${SHA256}`
24987449d2f5SColin Percival
24997449d2f5SColin Percival	if [ $X = $Y ]; then
25007449d2f5SColin Percival		return 0;
25017449d2f5SColin Percival	else
25027449d2f5SColin Percival		return 1;
25037449d2f5SColin Percival	fi
25047449d2f5SColin Percival}
25057449d2f5SColin Percival
2506db6b0a61SColin Percival# From the list of "old" files in $1, merge changes in $2 with those in $3,
2507db6b0a61SColin Percival# and update $3 to reflect the hashes of merged files.
2508db6b0a61SColin Percivalupgrade_merge () {
2509db6b0a61SColin Percival	# We only need to do anything if $1 is non-empty.
2510db6b0a61SColin Percival	if [ -s $1 ]; then
2511db6b0a61SColin Percival		cut -f 1 -d '|' $1 |
2512db6b0a61SColin Percival		    sort > $1-paths
2513db6b0a61SColin Percival
2514db6b0a61SColin Percival		# Create staging area for merging files
2515db6b0a61SColin Percival		rm -rf merge/
2516db6b0a61SColin Percival		while read F; do
2517db6b0a61SColin Percival			D=`dirname ${F}`
2518db6b0a61SColin Percival			mkdir -p merge/old/${D}
2519db6b0a61SColin Percival			mkdir -p merge/${OLDRELNUM}/${D}
2520db6b0a61SColin Percival			mkdir -p merge/${RELNUM}/${D}
2521db6b0a61SColin Percival			mkdir -p merge/new/${D}
2522db6b0a61SColin Percival		done < $1-paths
2523db6b0a61SColin Percival
2524db6b0a61SColin Percival		# Copy in files
2525db6b0a61SColin Percival		while read F; do
2526db6b0a61SColin Percival			# Currently installed file
2527db6b0a61SColin Percival			V=`look "${F}|" $2 | cut -f 7 -d '|'`
2528db6b0a61SColin Percival			gunzip < files/${V}.gz > merge/old/${F}
2529db6b0a61SColin Percival
2530db6b0a61SColin Percival			# Old release
2531db6b0a61SColin Percival			if look "${F}|" $1 | fgrep -q "|f|"; then
2532db6b0a61SColin Percival				V=`look "${F}|" $1 | cut -f 3 -d '|'`
2533db6b0a61SColin Percival				gunzip < files/${V}.gz		\
2534db6b0a61SColin Percival				    > merge/${OLDRELNUM}/${F}
2535db6b0a61SColin Percival			fi
2536db6b0a61SColin Percival
2537db6b0a61SColin Percival			# New release
2538db6b0a61SColin Percival			if look "${F}|" $3 | cut -f 1,2,7 -d '|' |
2539db6b0a61SColin Percival			    fgrep -q "|f|"; then
2540db6b0a61SColin Percival				V=`look "${F}|" $3 | cut -f 7 -d '|'`
2541db6b0a61SColin Percival				gunzip < files/${V}.gz		\
2542db6b0a61SColin Percival				    > merge/${RELNUM}/${F}
2543db6b0a61SColin Percival			fi
2544db6b0a61SColin Percival		done < $1-paths
2545db6b0a61SColin Percival
2546db6b0a61SColin Percival		# Attempt to automatically merge changes
2547db6b0a61SColin Percival		echo -n "Attempting to automatically merge "
2548db6b0a61SColin Percival		echo -n "changes in files..."
2549db6b0a61SColin Percival		: > failed.merges
2550db6b0a61SColin Percival		while read F; do
2551db6b0a61SColin Percival			# If the file doesn't exist in the new release,
2552db6b0a61SColin Percival			# the result of "merging changes" is having the file
2553db6b0a61SColin Percival			# not exist.
2554db6b0a61SColin Percival			if ! [ -f merge/${RELNUM}/${F} ]; then
2555db6b0a61SColin Percival				continue
2556db6b0a61SColin Percival			fi
2557db6b0a61SColin Percival
2558db6b0a61SColin Percival			# If the file didn't exist in the old release, we're
2559db6b0a61SColin Percival			# going to throw away the existing file and hope that
2560db6b0a61SColin Percival			# the version from the new release is what we want.
2561db6b0a61SColin Percival			if ! [ -f merge/${OLDRELNUM}/${F} ]; then
2562db6b0a61SColin Percival				cp merge/${RELNUM}/${F} merge/new/${F}
2563db6b0a61SColin Percival				continue
2564db6b0a61SColin Percival			fi
2565db6b0a61SColin Percival
2566db6b0a61SColin Percival			# Some files need special treatment.
2567db6b0a61SColin Percival			case ${F} in
2568db6b0a61SColin Percival			/etc/spwd.db | /etc/pwd.db | /etc/login.conf.db)
2569db6b0a61SColin Percival				# Don't merge these -- we're rebuild them
2570db6b0a61SColin Percival				# after updates are installed.
2571db6b0a61SColin Percival				cp merge/old/${F} merge/new/${F}
2572db6b0a61SColin Percival				;;
2573db6b0a61SColin Percival			*)
2574073dd712SBaptiste Daroussin				if ! diff3 -E -m -L "current version"	\
2575db6b0a61SColin Percival				    -L "${OLDRELNUM}" -L "${RELNUM}"	\
2576db6b0a61SColin Percival				    merge/old/${F}			\
2577db6b0a61SColin Percival				    merge/${OLDRELNUM}/${F}		\
2578db6b0a61SColin Percival				    merge/${RELNUM}/${F}		\
2579db6b0a61SColin Percival				    > merge/new/${F} 2>/dev/null; then
2580db6b0a61SColin Percival					echo ${F} >> failed.merges
2581db6b0a61SColin Percival				fi
2582db6b0a61SColin Percival				;;
2583db6b0a61SColin Percival			esac
2584db6b0a61SColin Percival		done < $1-paths
2585db6b0a61SColin Percival		echo " done."
2586db6b0a61SColin Percival
2587db6b0a61SColin Percival		# Ask the user to handle any files which didn't merge.
2588db6b0a61SColin Percival		while read F; do
25897449d2f5SColin Percival			# If the installed file differs from the version in
25906d514f10SDag-Erling Smørgrav			# the old release only due to RCS tag expansion
25917449d2f5SColin Percival			# then just use the version in the new release.
25927449d2f5SColin Percival			if samef merge/old/${F} merge/${OLDRELNUM}/${F}; then
25937449d2f5SColin Percival				cp merge/${RELNUM}/${F} merge/new/${F}
25947449d2f5SColin Percival				continue
25957449d2f5SColin Percival			fi
25967449d2f5SColin Percival
2597db6b0a61SColin Percival			cat <<-EOF
2598db6b0a61SColin Percival
2599db6b0a61SColin PercivalThe following file could not be merged automatically: ${F}
2600db6b0a61SColin PercivalPress Enter to edit this file in ${EDITOR} and resolve the conflicts
2601db6b0a61SColin Percivalmanually...
2602db6b0a61SColin Percival			EOF
2603ceb5f28bSEd Maste			while true; do
26043d442415SEd Maste				read response </dev/tty
26053d442415SEd Maste				if expr "${response}" : '[Aa][Cc][Cc][Ee][Pp][Tt]' > /dev/null; then
26063d442415SEd Maste					echo
26073d442415SEd Maste					break
26083d442415SEd Maste				fi
2609db6b0a61SColin Percival				${EDITOR} `pwd`/merge/new/${F} < /dev/tty
2610ceb5f28bSEd Maste
2611e27ded83SEd Maste				if ! grep -qE '^(<<<<<<<|=======|>>>>>>>)([[:space:]].*)?$' $(pwd)/merge/new/${F} ; then
2612ceb5f28bSEd Maste					break
2613ceb5f28bSEd Maste				fi
2614ceb5f28bSEd Maste				cat <<-EOF
2615ceb5f28bSEd Maste
2616ceb5f28bSEd MasteMerge conflict markers remain in: ${F}
2617ceb5f28bSEd MasteThese must be resolved for the system to be functional.
2618ceb5f28bSEd Maste
26193d442415SEd MastePress Enter to return to editing this file, or type "ACCEPT" to carry on with
26203d442415SEd Mastethese lines remaining in the file.
2621ceb5f28bSEd Maste				EOF
2622ceb5f28bSEd Maste			done
2623db6b0a61SColin Percival		done < failed.merges
2624db6b0a61SColin Percival		rm failed.merges
2625db6b0a61SColin Percival
2626db6b0a61SColin Percival		# Ask the user to confirm that he likes how the result
2627db6b0a61SColin Percival		# of merging files.
2628db6b0a61SColin Percival		while read F; do
26297449d2f5SColin Percival			# Skip files which haven't changed except possibly
26306d514f10SDag-Erling Smørgrav			# in their RCS tags.
26317449d2f5SColin Percival			if [ -f merge/old/${F} ] && [ -f merge/new/${F} ] &&
26327449d2f5SColin Percival			    samef merge/old/${F} merge/new/${F}; then
26337449d2f5SColin Percival				continue
26347449d2f5SColin Percival			fi
26357449d2f5SColin Percival
26367449d2f5SColin Percival			# Skip files where the installed file differs from
26376d514f10SDag-Erling Smørgrav			# the old file only due to RCS tags.
26387449d2f5SColin Percival			if [ -f merge/old/${F} ] &&
26397449d2f5SColin Percival			    [ -f merge/${OLDRELNUM}/${F} ] &&
26407449d2f5SColin Percival			    samef merge/old/${F} merge/${OLDRELNUM}/${F}; then
2641db6b0a61SColin Percival				continue
2642db6b0a61SColin Percival			fi
2643db6b0a61SColin Percival
2644db6b0a61SColin Percival			# Warn about files which are ceasing to exist.
2645db6b0a61SColin Percival			if ! [ -f merge/new/${F} ]; then
2646db6b0a61SColin Percival				cat <<-EOF
2647db6b0a61SColin Percival
2648db6b0a61SColin PercivalThe following file will be removed, as it no longer exists in
2649db6b0a61SColin PercivalFreeBSD ${RELNUM}: ${F}
2650db6b0a61SColin Percival				EOF
2651db6b0a61SColin Percival				continuep < /dev/tty || return 1
2652db6b0a61SColin Percival				continue
2653db6b0a61SColin Percival			fi
2654db6b0a61SColin Percival
2655db6b0a61SColin Percival			# Print changes for the user's approval.
2656db6b0a61SColin Percival			cat <<-EOF
2657db6b0a61SColin Percival
2658db6b0a61SColin PercivalThe following changes, which occurred between FreeBSD ${OLDRELNUM} and
2659db6b0a61SColin PercivalFreeBSD ${RELNUM} have been merged into ${F}:
2660db6b0a61SColin PercivalEOF
2661db6b0a61SColin Percival			diff -U 5 -L "current version" -L "new version"	\
2662db6b0a61SColin Percival			    merge/old/${F} merge/new/${F} || true
2663db6b0a61SColin Percival			continuep < /dev/tty || return 1
2664db6b0a61SColin Percival		done < $1-paths
2665db6b0a61SColin Percival
2666db6b0a61SColin Percival		# Store merged files.
2667db6b0a61SColin Percival		while read F; do
2668c58b62efSColin Percival			if [ -f merge/new/${F} ]; then
2669db6b0a61SColin Percival				V=`${SHA256} -q merge/new/${F}`
2670db6b0a61SColin Percival
2671db6b0a61SColin Percival				gzip -c < merge/new/${F} > files/${V}.gz
2672db6b0a61SColin Percival				echo "${F}|${V}"
2673db6b0a61SColin Percival			fi
2674db6b0a61SColin Percival		done < $1-paths > newhashes
2675db6b0a61SColin Percival
2676db6b0a61SColin Percival		# Pull lines out from $3 which need to be updated to
2677db6b0a61SColin Percival		# reflect merged files.
2678db6b0a61SColin Percival		while read F; do
2679db6b0a61SColin Percival			look "${F}|" $3
2680db6b0a61SColin Percival		done < $1-paths > $3-oldlines
2681db6b0a61SColin Percival
2682db6b0a61SColin Percival		# Update lines to reflect merged files
2683db6b0a61SColin Percival		join -t '|' -o 1.1,1.2,1.3,1.4,1.5,1.6,2.2,1.8		\
2684db6b0a61SColin Percival		    $3-oldlines newhashes > $3-newlines
2685db6b0a61SColin Percival
2686db6b0a61SColin Percival		# Remove old lines from $3 and add new lines.
2687db6b0a61SColin Percival		sort $3-oldlines |
2688db6b0a61SColin Percival		    comm -13 - $3 |
2689db6b0a61SColin Percival		    sort - $3-newlines > $3.tmp
2690db6b0a61SColin Percival		mv $3.tmp $3
2691db6b0a61SColin Percival
2692db6b0a61SColin Percival		# Clean up
2693db6b0a61SColin Percival		rm $1-paths newhashes $3-oldlines $3-newlines
2694db6b0a61SColin Percival		rm -rf merge/
2695db6b0a61SColin Percival	fi
2696db6b0a61SColin Percival
2697db6b0a61SColin Percival	# We're done with merging files.
2698db6b0a61SColin Percival	rm $1
2699db6b0a61SColin Percival}
2700db6b0a61SColin Percival
2701db6b0a61SColin Percival# Do the work involved in fetching upgrades to a new release
2702db6b0a61SColin Percivalupgrade_run () {
2703db6b0a61SColin Percival	workdir_init || return 1
2704db6b0a61SColin Percival
2705db6b0a61SColin Percival	# Prepare the mirror list.
2706db6b0a61SColin Percival	fetch_pick_server_init && fetch_pick_server
2707db6b0a61SColin Percival
2708db6b0a61SColin Percival	# Try to fetch the public key until we run out of servers.
2709db6b0a61SColin Percival	while ! fetch_key; do
2710db6b0a61SColin Percival		fetch_pick_server || return 1
2711db6b0a61SColin Percival	done
2712db6b0a61SColin Percival
2713db6b0a61SColin Percival	# Try to fetch the metadata index signature ("tag") until we run
2714db6b0a61SColin Percival	# out of available servers; and sanity check the downloaded tag.
2715db6b0a61SColin Percival	while ! fetch_tag; do
2716db6b0a61SColin Percival		fetch_pick_server || return 1
2717db6b0a61SColin Percival	done
2718db6b0a61SColin Percival	fetch_tagsanity || return 1
2719db6b0a61SColin Percival
2720db6b0a61SColin Percival	# Fetch the INDEX-OLD and INDEX-ALL.
2721db6b0a61SColin Percival	fetch_metadata INDEX-OLD INDEX-ALL || return 1
2722db6b0a61SColin Percival
2723db6b0a61SColin Percival	# If StrictComponents is not "yes", generate a new components list
2724db6b0a61SColin Percival	# with only the components which appear to be installed.
2725db6b0a61SColin Percival	upgrade_guess_components INDEX-ALL || return 1
2726db6b0a61SColin Percival
2727db6b0a61SColin Percival	# Generate filtered INDEX-OLD and INDEX-ALL files containing only
2728db6b0a61SColin Percival	# the components we want and without anything marked as "Ignore".
2729db6b0a61SColin Percival	fetch_filter_metadata INDEX-OLD || return 1
2730db6b0a61SColin Percival	fetch_filter_metadata INDEX-ALL || return 1
2731db6b0a61SColin Percival
2732db6b0a61SColin Percival	# Merge the INDEX-OLD and INDEX-ALL files into INDEX-OLD.
2733db6b0a61SColin Percival	sort INDEX-OLD INDEX-ALL > INDEX-OLD.tmp
2734db6b0a61SColin Percival	mv INDEX-OLD.tmp INDEX-OLD
2735db6b0a61SColin Percival	rm INDEX-ALL
2736db6b0a61SColin Percival
2737db6b0a61SColin Percival	# Adjust variables for fetching files from the new release.
2738db6b0a61SColin Percival	OLDRELNUM=${RELNUM}
2739db6b0a61SColin Percival	RELNUM=${TARGETRELEASE}
2740db6b0a61SColin Percival	OLDFETCHDIR=${FETCHDIR}
2741db6b0a61SColin Percival	FETCHDIR=${RELNUM}/${ARCH}
2742db6b0a61SColin Percival
2743db6b0a61SColin Percival	# Try to fetch the NEW metadata index signature ("tag") until we run
2744db6b0a61SColin Percival	# out of available servers; and sanity check the downloaded tag.
2745db6b0a61SColin Percival	while ! fetch_tag; do
2746db6b0a61SColin Percival		fetch_pick_server || return 1
2747db6b0a61SColin Percival	done
2748db6b0a61SColin Percival
2749db6b0a61SColin Percival	# Fetch the new INDEX-ALL.
2750db6b0a61SColin Percival	fetch_metadata INDEX-ALL || return 1
2751db6b0a61SColin Percival
2752db6b0a61SColin Percival	# If StrictComponents is not "yes", COMPONENTS contains an entry
2753db6b0a61SColin Percival	# corresponding to the currently running kernel, and said kernel
2754db6b0a61SColin Percival	# does not exist in the new release, add "kernel/generic" to the
2755db6b0a61SColin Percival	# list of components.
2756db6b0a61SColin Percival	upgrade_guess_new_kernel INDEX-ALL || return 1
2757db6b0a61SColin Percival
2758db6b0a61SColin Percival	# Filter INDEX-ALL to contain only the components we want and without
2759db6b0a61SColin Percival	# anything marked as "Ignore".
2760db6b0a61SColin Percival	fetch_filter_metadata INDEX-ALL || return 1
2761db6b0a61SColin Percival
2762db6b0a61SColin Percival	# Convert INDEX-OLD (last release) and INDEX-ALL (new release) into
2763db6b0a61SColin Percival	# INDEX-OLD and INDEX-NEW files (in the sense of normal upgrades).
2764db6b0a61SColin Percival	upgrade_oldall_to_oldnew INDEX-OLD INDEX-ALL INDEX-NEW
2765db6b0a61SColin Percival
2766db6b0a61SColin Percival	# Translate /boot/${KERNCONF} or /boot/${NKERNCONF} into ${KERNELDIR}
2767db6b0a61SColin Percival	fetch_filter_kernel_names INDEX-NEW ${NKERNCONF}
2768db6b0a61SColin Percival	fetch_filter_kernel_names INDEX-OLD ${KERNCONF}
2769db6b0a61SColin Percival
2770db6b0a61SColin Percival	# For all paths appearing in INDEX-OLD or INDEX-NEW, inspect the
2771db6b0a61SColin Percival	# system and generate an INDEX-PRESENT file.
2772db6b0a61SColin Percival	fetch_inspect_system INDEX-OLD INDEX-PRESENT INDEX-NEW || return 1
2773db6b0a61SColin Percival
2774db6b0a61SColin Percival	# Based on ${MERGECHANGES}, generate a file tomerge-old with the
2775db6b0a61SColin Percival	# paths and hashes of old versions of files to merge.
2776c55b7e52SColin Percival	fetch_filter_mergechanges INDEX-OLD INDEX-PRESENT INDEX-NEW tomerge-old
2777db6b0a61SColin Percival
2778db6b0a61SColin Percival	# Based on ${UPDATEIFUNMODIFIED}, remove lines from INDEX-* which
2779db6b0a61SColin Percival	# correspond to lines in INDEX-PRESENT with hashes not appearing
2780db6b0a61SColin Percival	# in INDEX-OLD or INDEX-NEW.  Also remove lines where the entry in
2781db6b0a61SColin Percival	# INDEX-PRESENT has type - and there isn't a corresponding entry in
2782db6b0a61SColin Percival	# INDEX-OLD with type -.
2783db6b0a61SColin Percival	fetch_filter_unmodified_notpresent	\
2784db6b0a61SColin Percival	    INDEX-OLD INDEX-PRESENT INDEX-NEW tomerge-old
2785db6b0a61SColin Percival
2786db6b0a61SColin Percival	# For each entry in INDEX-PRESENT of type -, remove any corresponding
2787db6b0a61SColin Percival	# entry from INDEX-NEW if ${ALLOWADD} != "yes".  Remove all entries
2788db6b0a61SColin Percival	# of type - from INDEX-PRESENT.
2789db6b0a61SColin Percival	fetch_filter_allowadd INDEX-PRESENT INDEX-NEW
2790db6b0a61SColin Percival
2791db6b0a61SColin Percival	# If ${ALLOWDELETE} != "yes", then remove any entries from
2792db6b0a61SColin Percival	# INDEX-PRESENT which don't correspond to entries in INDEX-NEW.
2793db6b0a61SColin Percival	fetch_filter_allowdelete INDEX-PRESENT INDEX-NEW
2794db6b0a61SColin Percival
2795db6b0a61SColin Percival	# If ${KEEPMODIFIEDMETADATA} == "yes", then for each entry in
2796db6b0a61SColin Percival	# INDEX-PRESENT with metadata not matching any entry in INDEX-OLD,
2797db6b0a61SColin Percival	# replace the corresponding line of INDEX-NEW with one having the
2798db6b0a61SColin Percival	# same metadata as the entry in INDEX-PRESENT.
2799db6b0a61SColin Percival	fetch_filter_modified_metadata INDEX-OLD INDEX-PRESENT INDEX-NEW
2800db6b0a61SColin Percival
2801db6b0a61SColin Percival	# Remove lines from INDEX-PRESENT and INDEX-NEW which are identical;
2802db6b0a61SColin Percival	# no need to update a file if it isn't changing.
2803db6b0a61SColin Percival	fetch_filter_uptodate INDEX-PRESENT INDEX-NEW
2804db6b0a61SColin Percival
2805db6b0a61SColin Percival	# Fetch "clean" files from the old release for merging changes.
2806db6b0a61SColin Percival	fetch_files_premerge tomerge-old
2807db6b0a61SColin Percival
2808db6b0a61SColin Percival	# Prepare to fetch files: Generate a list of the files we need,
2809db6b0a61SColin Percival	# copy the unmodified files we have into /files/, and generate
2810db6b0a61SColin Percival	# a list of patches to download.
2811db6b0a61SColin Percival	fetch_files_prepare INDEX-OLD INDEX-PRESENT INDEX-NEW || return 1
2812db6b0a61SColin Percival
2813db6b0a61SColin Percival	# Fetch patches from to-${RELNUM}/${ARCH}/bp/
2814db6b0a61SColin Percival	PATCHDIR=to-${RELNUM}/${ARCH}/bp
2815db6b0a61SColin Percival	fetch_files || return 1
2816db6b0a61SColin Percival
2817db6b0a61SColin Percival	# Merge configuration file changes.
2818db6b0a61SColin Percival	upgrade_merge tomerge-old INDEX-PRESENT INDEX-NEW || return 1
2819db6b0a61SColin Percival
2820db6b0a61SColin Percival	# Create and populate install manifest directory; and report what
2821db6b0a61SColin Percival	# updates are available.
2822db6b0a61SColin Percival	fetch_create_manifest || return 1
2823db6b0a61SColin Percival
2824db6b0a61SColin Percival	# Leave a note behind to tell the "install" command that the kernel
2825db6b0a61SColin Percival	# needs to be installed before the world.
2826db6b0a61SColin Percival	touch ${BDHASH}-install/kernelfirst
282785451f90SColin Percival
282885451f90SColin Percival	# Remind the user that they need to run "freebsd-update install"
282985451f90SColin Percival	# to install the downloaded bits, in case they didn't RTFM.
283059b02bb4SMichael Osipov	echo "To install the downloaded upgrades, run '`basename $0` [options] install'."
2831db6b0a61SColin Percival}
2832db6b0a61SColin Percival
283348ffe56aSColin Percival# Make sure that all the file hashes mentioned in $@ have corresponding
283448ffe56aSColin Percival# gzipped files stored in /files/.
283548ffe56aSColin Percivalinstall_verify () {
283648ffe56aSColin Percival	# Generate a list of hashes
283748ffe56aSColin Percival	cat $@ |
283848ffe56aSColin Percival	    cut -f 2,7 -d '|' |
283948ffe56aSColin Percival	    grep -E '^f' |
284048ffe56aSColin Percival	    cut -f 2 -d '|' |
284148ffe56aSColin Percival	    sort -u > filelist
284248ffe56aSColin Percival
284348ffe56aSColin Percival	# Make sure all the hashes exist
284448ffe56aSColin Percival	while read HASH; do
284548ffe56aSColin Percival		if ! [ -f files/${HASH}.gz ]; then
284648ffe56aSColin Percival			echo -n "Update files missing -- "
284748ffe56aSColin Percival			echo "this should never happen."
284859b02bb4SMichael Osipov			echo "Re-run '`basename $0` [options] fetch'."
284948ffe56aSColin Percival			return 1
285048ffe56aSColin Percival		fi
285148ffe56aSColin Percival	done < filelist
285248ffe56aSColin Percival
285348ffe56aSColin Percival	# Clean up
285448ffe56aSColin Percival	rm filelist
285548ffe56aSColin Percival}
285648ffe56aSColin Percival
285748ffe56aSColin Percival# Remove the system immutable flag from files
285848ffe56aSColin Percivalinstall_unschg () {
285948ffe56aSColin Percival	# Generate file list
286048ffe56aSColin Percival	cat $@ |
286148ffe56aSColin Percival	    cut -f 1 -d '|' > filelist
286248ffe56aSColin Percival
286348ffe56aSColin Percival	# Remove flags
286448ffe56aSColin Percival	while read F; do
2865e829ed67SColin Percival		if ! [ -e ${BASEDIR}/${F} ]; then
286648ffe56aSColin Percival			continue
28675a74378cSXin LI		else
28685a74378cSXin LI			echo ${BASEDIR}/${F}
286948ffe56aSColin Percival		fi
28705a74378cSXin LI	done < filelist | xargs chflags noschg || return 1
287148ffe56aSColin Percival
287248ffe56aSColin Percival	# Clean up
287348ffe56aSColin Percival	rm filelist
287448ffe56aSColin Percival}
287548ffe56aSColin Percival
287623d827efSSimon L. B. Nielsen# Decide which directory name to use for kernel backups.
287723d827efSSimon L. B. Nielsenbackup_kernel_finddir () {
287823d827efSSimon L. B. Nielsen	CNT=0
287923d827efSSimon L. B. Nielsen	while true ; do
288023d827efSSimon L. B. Nielsen		# Pathname does not exist, so it is OK use that name
288123d827efSSimon L. B. Nielsen		# for backup directory.
2882c4a0c62cSThomas Quinot		if [ ! -e $BASEDIR/$BACKUPKERNELDIR ]; then
288323d827efSSimon L. B. Nielsen			return 0
288423d827efSSimon L. B. Nielsen		fi
288523d827efSSimon L. B. Nielsen
288623d827efSSimon L. B. Nielsen		# If directory do exist, we only use if it has our
288723d827efSSimon L. B. Nielsen		# marker file.
2888c4a0c62cSThomas Quinot		if [ -d $BASEDIR/$BACKUPKERNELDIR -a \
2889c4a0c62cSThomas Quinot			-e $BASEDIR/$BACKUPKERNELDIR/.freebsd-update ]; then
289023d827efSSimon L. B. Nielsen			return 0
289123d827efSSimon L. B. Nielsen		fi
289223d827efSSimon L. B. Nielsen
289323d827efSSimon L. B. Nielsen		# We could not use current directory name, so add counter to
289423d827efSSimon L. B. Nielsen		# the end and try again.
289523d827efSSimon L. B. Nielsen		CNT=$((CNT + 1))
289623d827efSSimon L. B. Nielsen		if [ $CNT -gt 9 ]; then
2897c4a0c62cSThomas Quinot			echo "Could not find valid backup dir ($BASEDIR/$BACKUPKERNELDIR)"
289823d827efSSimon L. B. Nielsen			exit 1
289923d827efSSimon L. B. Nielsen		fi
290023d827efSSimon L. B. Nielsen		BACKUPKERNELDIR="`echo $BACKUPKERNELDIR | sed -Ee 's/[0-9]\$//'`"
290123d827efSSimon L. B. Nielsen		BACKUPKERNELDIR="${BACKUPKERNELDIR}${CNT}"
290223d827efSSimon L. B. Nielsen	done
290323d827efSSimon L. B. Nielsen}
290423d827efSSimon L. B. Nielsen
290523d827efSSimon L. B. Nielsen# Backup the current kernel using hardlinks, if not disabled by user.
290623d827efSSimon L. B. Nielsen# Since we delete all files in the directory used for previous backups
290723d827efSSimon L. B. Nielsen# we create a marker file called ".freebsd-update" in the directory so
290823d827efSSimon L. B. Nielsen# we can determine on the next run that the directory was created by
290923d827efSSimon L. B. Nielsen# freebsd-update and we then do not accidentally remove user files in
291023d827efSSimon L. B. Nielsen# the unlikely case that the user has created a directory with a
291123d827efSSimon L. B. Nielsen# conflicting name.
291223d827efSSimon L. B. Nielsenbackup_kernel () {
291323d827efSSimon L. B. Nielsen	# Only make kernel backup is so configured.
291423d827efSSimon L. B. Nielsen	if [ $BACKUPKERNEL != yes ]; then
291523d827efSSimon L. B. Nielsen		return 0
291623d827efSSimon L. B. Nielsen	fi
291723d827efSSimon L. B. Nielsen
291823d827efSSimon L. B. Nielsen	# Decide which directory name to use for kernel backups.
291923d827efSSimon L. B. Nielsen	backup_kernel_finddir
292023d827efSSimon L. B. Nielsen
292123d827efSSimon L. B. Nielsen	# Remove old kernel backup files.  If $BACKUPKERNELDIR was
292223d827efSSimon L. B. Nielsen	# "not ours", backup_kernel_finddir would have exited, so
292323d827efSSimon L. B. Nielsen	# deleting the directory content is as safe as we can make it.
2924c4a0c62cSThomas Quinot	if [ -d $BASEDIR/$BACKUPKERNELDIR ]; then
2925c4a0c62cSThomas Quinot		rm -fr $BASEDIR/$BACKUPKERNELDIR
292623d827efSSimon L. B. Nielsen	fi
292723d827efSSimon L. B. Nielsen
2928ab7d0151SJaakko Heinonen	# Create directories for backup.
2929c4a0c62cSThomas Quinot	mkdir -p $BASEDIR/$BACKUPKERNELDIR
2930c4a0c62cSThomas Quinot	mtree -cdn -p "${BASEDIR}/${KERNELDIR}" | \
2931c4a0c62cSThomas Quinot	    mtree -Ue -p "${BASEDIR}/${BACKUPKERNELDIR}" > /dev/null
293223d827efSSimon L. B. Nielsen
293323d827efSSimon L. B. Nielsen	# Mark the directory as having been created by freebsd-update.
2934c4a0c62cSThomas Quinot	touch $BASEDIR/$BACKUPKERNELDIR/.freebsd-update
293523d827efSSimon L. B. Nielsen	if [ $? -ne 0 ]; then
293623d827efSSimon L. B. Nielsen		echo "Could not create kernel backup directory"
293723d827efSSimon L. B. Nielsen		exit 1
293823d827efSSimon L. B. Nielsen	fi
293923d827efSSimon L. B. Nielsen
294023d827efSSimon L. B. Nielsen	# Disable pathname expansion to be sure *.symbols is not
294123d827efSSimon L. B. Nielsen	# expanded.
294223d827efSSimon L. B. Nielsen	set -f
294323d827efSSimon L. B. Nielsen
294423d827efSSimon L. B. Nielsen	# Use find to ignore symbol files, unless disabled by user.
294523d827efSSimon L. B. Nielsen	if [ $BACKUPKERNELSYMBOLFILES = yes ]; then
294623d827efSSimon L. B. Nielsen		FINDFILTER=""
294723d827efSSimon L. B. Nielsen	else
29487e1ed2c7SEd Maste		FINDFILTER="-a ! -name *.debug -a ! -name *.symbols"
294923d827efSSimon L. B. Nielsen	fi
295023d827efSSimon L. B. Nielsen
295123d827efSSimon L. B. Nielsen	# Backup all the kernel files using hardlinks.
2952c4a0c62cSThomas Quinot	(cd ${BASEDIR}/${KERNELDIR} && find . -type f $FINDFILTER -exec \
2953c4a0c62cSThomas Quinot	    cp -pl '{}' ${BASEDIR}/${BACKUPKERNELDIR}/'{}' \;)
295423d827efSSimon L. B. Nielsen
295591811711Sa-biardi	# Re-enable pathname expansion.
295623d827efSSimon L. B. Nielsen	set +f
295723d827efSSimon L. B. Nielsen}
295823d827efSSimon L. B. Nielsen
2959c0f52443SEd Maste# Check for and remove an existing directory that conflicts with the file or
2960c0f52443SEd Maste# symlink that we are going to install.
2961c0f52443SEd Mastedir_conflict () {
2962c0f52443SEd Maste	if [ -d "$1" ]; then
2963c0f52443SEd Maste		echo "Removing conflicting directory $1"
2964c0f52443SEd Maste		rm -rf -- "$1"
2965c0f52443SEd Maste	fi
2966c0f52443SEd Maste}
2967c0f52443SEd Maste
296848ffe56aSColin Percival# Install new files
296948ffe56aSColin Percivalinstall_from_index () {
297048ffe56aSColin Percival	# First pass: Do everything apart from setting file flags.  We
297148ffe56aSColin Percival	# can't set flags yet, because schg inhibits hard linking.
297248ffe56aSColin Percival	sort -k 1,1 -t '|' $1 |
297348ffe56aSColin Percival	    tr '|' ' ' |
297448ffe56aSColin Percival	    while read FPATH TYPE OWNER GROUP PERM FLAGS HASH LINK; do
297548ffe56aSColin Percival		case ${TYPE} in
297648ffe56aSColin Percival		d)
2977f6d37c9cSEd Maste			# Create a directory.  A file may change to a directory
2978f6d37c9cSEd Maste			# on upgrade (PR273661).  If that happens, remove the
2979f6d37c9cSEd Maste			# file first.
2980f6d37c9cSEd Maste			if [ -e "${BASEDIR}/${FPATH}" ] && \
2981f6d37c9cSEd Maste			    ! [ -d "${BASEDIR}/${FPATH}" ]; then
2982f6d37c9cSEd Maste				rm -f -- "${BASEDIR}/${FPATH}"
2983f6d37c9cSEd Maste			fi
298448ffe56aSColin Percival			install -d -o ${OWNER} -g ${GROUP}		\
298548ffe56aSColin Percival			    -m ${PERM} ${BASEDIR}/${FPATH}
298648ffe56aSColin Percival			;;
298748ffe56aSColin Percival		f)
2988c0f52443SEd Maste			dir_conflict "${BASEDIR}/${FPATH}"
298948ffe56aSColin Percival			if [ -z "${LINK}" ]; then
299048ffe56aSColin Percival				# Create a file, without setting flags.
299148ffe56aSColin Percival				gunzip < files/${HASH}.gz > ${HASH}
299248ffe56aSColin Percival				install -S -o ${OWNER} -g ${GROUP}	\
299348ffe56aSColin Percival				    -m ${PERM} ${HASH} ${BASEDIR}/${FPATH}
299448ffe56aSColin Percival				rm ${HASH}
299548ffe56aSColin Percival			else
299648ffe56aSColin Percival				# Create a hard link.
2997e829ed67SColin Percival				ln -f ${BASEDIR}/${LINK} ${BASEDIR}/${FPATH}
299848ffe56aSColin Percival			fi
299948ffe56aSColin Percival			;;
300048ffe56aSColin Percival		L)
3001c0f52443SEd Maste			dir_conflict "${BASEDIR}/${FPATH}"
300248ffe56aSColin Percival			# Create a symlink
300348ffe56aSColin Percival			ln -sfh ${HASH} ${BASEDIR}/${FPATH}
300448ffe56aSColin Percival			;;
300548ffe56aSColin Percival		esac
300648ffe56aSColin Percival	    done
300748ffe56aSColin Percival
300848ffe56aSColin Percival	# Perform a second pass, adding file flags.
300948ffe56aSColin Percival	tr '|' ' ' < $1 |
301048ffe56aSColin Percival	    while read FPATH TYPE OWNER GROUP PERM FLAGS HASH LINK; do
301148ffe56aSColin Percival		if [ ${TYPE} = "f" ] &&
301248ffe56aSColin Percival		    ! [ ${FLAGS} = "0" ]; then
301348ffe56aSColin Percival			chflags ${FLAGS} ${BASEDIR}/${FPATH}
301448ffe56aSColin Percival		fi
301548ffe56aSColin Percival	    done
301648ffe56aSColin Percival}
301748ffe56aSColin Percival
301848ffe56aSColin Percival# Remove files which we want to delete
301948ffe56aSColin Percivalinstall_delete () {
302048ffe56aSColin Percival	# Generate list of new files
302148ffe56aSColin Percival	cut -f 1 -d '|' < $2 |
302248ffe56aSColin Percival	    sort > newfiles
302348ffe56aSColin Percival
302448ffe56aSColin Percival	# Generate subindex of old files we want to nuke
302548ffe56aSColin Percival	sort -k 1,1 -t '|' $1 |
302648ffe56aSColin Percival	    join -t '|' -v 1 - newfiles |
3027bce02f98SColin Percival	    sort -r -k 1,1 -t '|' |
302848ffe56aSColin Percival	    cut -f 1,2 -d '|' |
302948ffe56aSColin Percival	    tr '|' ' ' > killfiles
303048ffe56aSColin Percival
303148ffe56aSColin Percival	# Remove the offending bits
303248ffe56aSColin Percival	while read FPATH TYPE; do
303348ffe56aSColin Percival		case ${TYPE} in
303448ffe56aSColin Percival		d)
303548ffe56aSColin Percival			rmdir ${BASEDIR}/${FPATH}
303648ffe56aSColin Percival			;;
303748ffe56aSColin Percival		f)
3038c0f52443SEd Maste			if [ -f "${BASEDIR}/${FPATH}" ]; then
3039c0f52443SEd Maste				rm "${BASEDIR}/${FPATH}"
3040c0f52443SEd Maste			fi
304148ffe56aSColin Percival			;;
304248ffe56aSColin Percival		L)
3043c0f52443SEd Maste			if [ -L "${BASEDIR}/${FPATH}" ]; then
3044c0f52443SEd Maste				rm "${BASEDIR}/${FPATH}"
3045c0f52443SEd Maste			fi
304648ffe56aSColin Percival			;;
304748ffe56aSColin Percival		esac
304848ffe56aSColin Percival	done < killfiles
304948ffe56aSColin Percival
305048ffe56aSColin Percival	# Clean up
305148ffe56aSColin Percival	rm newfiles killfiles
305248ffe56aSColin Percival}
305348ffe56aSColin Percival
30542b17527cSKyle Evans# Install new files, delete old files, and update generated files
3055db6b0a61SColin Percivalinstall_files () {
3056db6b0a61SColin Percival	# If we haven't already dealt with the kernel, deal with it.
3057db6b0a61SColin Percival	if ! [ -f $1/kerneldone ]; then
3058db6b0a61SColin Percival		grep -E '^/boot/' $1/INDEX-OLD > INDEX-OLD
3059db6b0a61SColin Percival		grep -E '^/boot/' $1/INDEX-NEW > INDEX-NEW
3060db6b0a61SColin Percival
306123d827efSSimon L. B. Nielsen		# Backup current kernel before installing a new one
306223d827efSSimon L. B. Nielsen		backup_kernel || return 1
306323d827efSSimon L. B. Nielsen
3064db6b0a61SColin Percival		# Install new files
3065db6b0a61SColin Percival		install_from_index INDEX-NEW || return 1
3066db6b0a61SColin Percival
3067db6b0a61SColin Percival		# Remove files which need to be deleted
3068db6b0a61SColin Percival		install_delete INDEX-OLD INDEX-NEW || return 1
3069db6b0a61SColin Percival
3070db6b0a61SColin Percival		# Update linker.hints if necessary
3071db6b0a61SColin Percival		if [ -s INDEX-OLD -o -s INDEX-NEW ]; then
3072c4a0c62cSThomas Quinot			kldxref -R ${BASEDIR}/boot/ 2>/dev/null
307348ffe56aSColin Percival		fi
3074db6b0a61SColin Percival
3075db6b0a61SColin Percival		# We've finished updating the kernel.
3076db6b0a61SColin Percival		touch $1/kerneldone
3077db6b0a61SColin Percival
3078db6b0a61SColin Percival		# Do we need to ask for a reboot now?
3079db6b0a61SColin Percival		if [ -f $1/kernelfirst ] &&
3080db6b0a61SColin Percival		    [ -s INDEX-OLD -o -s INDEX-NEW ]; then
3081db6b0a61SColin Percival			cat <<-EOF
3082db6b0a61SColin Percival
3083db6b0a61SColin PercivalKernel updates have been installed.  Please reboot and run
308459b02bb4SMichael Osipov'`basename $0` [options] install' again to finish installing updates.
3085db6b0a61SColin Percival			EOF
3086db6b0a61SColin Percival			exit 0
3087db6b0a61SColin Percival		fi
3088db6b0a61SColin Percival	fi
3089db6b0a61SColin Percival
3090db6b0a61SColin Percival	# If we haven't already dealt with the world, deal with it.
3091db6b0a61SColin Percival	if ! [ -f $1/worlddone ]; then
3092cd1ab228SColin Percival		# Create any necessary directories first
3093cd1ab228SColin Percival		grep -vE '^/boot/' $1/INDEX-NEW |
3094cd1ab228SColin Percival		    grep -E '^[^|]+\|d\|' > INDEX-NEW
3095cd1ab228SColin Percival		install_from_index INDEX-NEW || return 1
3096cd1ab228SColin Percival
3097722d81b5SBrooks Davis		# Install new runtime linker
3098722d81b5SBrooks Davis		grep -vE '^/boot/' $1/INDEX-NEW |
3099722d81b5SBrooks Davis		    grep -vE '^[^|]+\|d\|' |
3100722d81b5SBrooks Davis		    grep -E '^/libexec/ld-elf[^|]*\.so\.[0-9]+\|' > INDEX-NEW
3101722d81b5SBrooks Davis		install_from_index INDEX-NEW || return 1
3102722d81b5SBrooks Davis
3103db6b0a61SColin Percival		# Install new shared libraries next
3104db6b0a61SColin Percival		grep -vE '^/boot/' $1/INDEX-NEW |
3105cd1ab228SColin Percival		    grep -vE '^[^|]+\|d\|' |
3106722d81b5SBrooks Davis		    grep -vE '^/libexec/ld-elf[^|]*\.so\.[0-9]+\|' |
31079546dbd1SColin Percival		    grep -E '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' > INDEX-NEW
3108db6b0a61SColin Percival		install_from_index INDEX-NEW || return 1
3109db6b0a61SColin Percival
3110db6b0a61SColin Percival		# Deal with everything else
3111db6b0a61SColin Percival		grep -vE '^/boot/' $1/INDEX-OLD |
3112cd1ab228SColin Percival		    grep -vE '^[^|]+\|d\|' |
3113722d81b5SBrooks Davis		    grep -vE '^/libexec/ld-elf[^|]*\.so\.[0-9]+\|' |
31149546dbd1SColin Percival		    grep -vE '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' > INDEX-OLD
3115db6b0a61SColin Percival		grep -vE '^/boot/' $1/INDEX-NEW |
3116cd1ab228SColin Percival		    grep -vE '^[^|]+\|d\|' |
3117722d81b5SBrooks Davis		    grep -vE '^/libexec/ld-elf[^|]*\.so\.[0-9]+\|' |
31189546dbd1SColin Percival		    grep -vE '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' > INDEX-NEW
3119db6b0a61SColin Percival		install_from_index INDEX-NEW || return 1
3120db6b0a61SColin Percival		install_delete INDEX-OLD INDEX-NEW || return 1
3121db6b0a61SColin Percival
31228ee97b19SEd Maste		# Restart host sshd if running (PR263489).  Note that this does
31238ee97b19SEd Maste		# not affect child sshd processes handling existing sessions.
31248ee97b19SEd Maste		if [ "$BASEDIR" = / ] && \
31258ee97b19SEd Maste		    service sshd status >/dev/null 2>/dev/null; then
31266cd1bc53SEd Maste			echo
31276cd1bc53SEd Maste			echo "Restarting sshd after upgrade"
31286cd1bc53SEd Maste			service sshd restart
31296cd1bc53SEd Maste		fi
31306cd1bc53SEd Maste
31312b17527cSKyle Evans		# Rehash certs if we actually have certctl installed.
31322b17527cSKyle Evans		if which certctl>/dev/null; then
31332b17527cSKyle Evans			env DESTDIR=${BASEDIR} certctl rehash
31342b17527cSKyle Evans		fi
31352b17527cSKyle Evans
3136ebebc41eSKyle Evans		# Rebuild generated pwd files and /etc/login.conf.db.
31379b659110SEd Maste		pwd_mkdb -d ${BASEDIR}/etc -p ${BASEDIR}/etc/master.passwd
3138c4a0c62cSThomas Quinot		cap_mkdb ${BASEDIR}/etc/login.conf
3139db6b0a61SColin Percival
31409c812c8dSEd Maste		# Rebuild man page databases, if necessary.
31419c812c8dSEd Maste		for D in /usr/share/man /usr/share/openssl/man; do
31429c812c8dSEd Maste			if [ ! -d ${BASEDIR}/$D ]; then
31439c812c8dSEd Maste				continue
31449c812c8dSEd Maste			fi
3145741223a6SEd Maste			if [ -f ${BASEDIR}/$D/mandoc.db ] && \
3146741223a6SEd Maste			    [ -z "$(find ${BASEDIR}/$D -type f -newer ${BASEDIR}/$D/mandoc.db)" ]; then
31479c812c8dSEd Maste				continue;
31489c812c8dSEd Maste			fi
31499c812c8dSEd Maste			makewhatis ${BASEDIR}/$D
31509c812c8dSEd Maste		done
31519c812c8dSEd Maste
3152db6b0a61SColin Percival		# We've finished installing the world and deleting old files
3153db6b0a61SColin Percival		# which are not shared libraries.
3154db6b0a61SColin Percival		touch $1/worlddone
3155db6b0a61SColin Percival
3156db6b0a61SColin Percival		# Do we need to ask the user to portupgrade now?
3157db6b0a61SColin Percival		grep -vE '^/boot/' $1/INDEX-NEW |
31589546dbd1SColin Percival		    grep -E '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' |
3159db6b0a61SColin Percival		    cut -f 1 -d '|' |
3160db6b0a61SColin Percival		    sort > newfiles
3161db6b0a61SColin Percival		if grep -vE '^/boot/' $1/INDEX-OLD |
31629546dbd1SColin Percival		    grep -E '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' |
3163db6b0a61SColin Percival		    cut -f 1 -d '|' |
3164db6b0a61SColin Percival		    sort |
3165db6b0a61SColin Percival		    join -v 1 - newfiles |
3166db6b0a61SColin Percival		    grep -q .; then
3167db6b0a61SColin Percival			cat <<-EOF
3168db6b0a61SColin Percival
3169db6b0a61SColin PercivalCompleting this upgrade requires removing old shared object files.
3170db6b0a61SColin PercivalPlease rebuild all installed 3rd party software (e.g., programs
317159b02bb4SMichael Osipovinstalled from the ports tree) and then run
317259b02bb4SMichael Osipov'`basename $0` [options] install' again to finish installing updates.
3173db6b0a61SColin Percival			EOF
3174db6b0a61SColin Percival			rm newfiles
3175db6b0a61SColin Percival			exit 0
3176db6b0a61SColin Percival		fi
3177db6b0a61SColin Percival		rm newfiles
3178db6b0a61SColin Percival	fi
3179db6b0a61SColin Percival
3180db6b0a61SColin Percival	# Remove old shared libraries
3181db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-NEW |
3182cd1ab228SColin Percival	    grep -vE '^[^|]+\|d\|' |
31839546dbd1SColin Percival	    grep -E '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' > INDEX-NEW
3184db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-OLD |
3185cd1ab228SColin Percival	    grep -vE '^[^|]+\|d\|' |
31869546dbd1SColin Percival	    grep -E '^[^|]*/lib/[^|]*\.so\.[0-9]+\|' > INDEX-OLD
3187db6b0a61SColin Percival	install_delete INDEX-OLD INDEX-NEW || return 1
3188db6b0a61SColin Percival
3189cd1ab228SColin Percival	# Remove old directories
3190ebc1d19cSColin Percival	grep -vE '^/boot/' $1/INDEX-NEW |
3191ebc1d19cSColin Percival	    grep -E '^[^|]+\|d\|' > INDEX-NEW
3192cd1ab228SColin Percival	grep -vE '^/boot/' $1/INDEX-OLD |
3193cd1ab228SColin Percival	    grep -E '^[^|]+\|d\|' > INDEX-OLD
3194cd1ab228SColin Percival	install_delete INDEX-OLD INDEX-NEW || return 1
3195cd1ab228SColin Percival
3196db6b0a61SColin Percival	# Remove temporary files
3197db6b0a61SColin Percival	rm INDEX-OLD INDEX-NEW
319848ffe56aSColin Percival}
319948ffe56aSColin Percival
320048ffe56aSColin Percival# Rearrange bits to allow the installed updates to be rolled back
320148ffe56aSColin Percivalinstall_setup_rollback () {
3202db6b0a61SColin Percival	# Remove the "reboot after installing kernel", "kernel updated", and
3203db6b0a61SColin Percival	# "finished installing the world" flags if present -- they are
3204db6b0a61SColin Percival	# irrelevant when rolling back updates.
3205db6b0a61SColin Percival	if [ -f ${BDHASH}-install/kernelfirst ]; then
3206db6b0a61SColin Percival		rm ${BDHASH}-install/kernelfirst
3207db6b0a61SColin Percival		rm ${BDHASH}-install/kerneldone
3208db6b0a61SColin Percival	fi
3209db6b0a61SColin Percival	if [ -f ${BDHASH}-install/worlddone ]; then
3210db6b0a61SColin Percival		rm ${BDHASH}-install/worlddone
3211db6b0a61SColin Percival	fi
3212db6b0a61SColin Percival
321348ffe56aSColin Percival	if [ -L ${BDHASH}-rollback ]; then
321448ffe56aSColin Percival		mv ${BDHASH}-rollback ${BDHASH}-install/rollback
321548ffe56aSColin Percival	fi
321648ffe56aSColin Percival
321748ffe56aSColin Percival	mv ${BDHASH}-install ${BDHASH}-rollback
321848ffe56aSColin Percival}
321948ffe56aSColin Percival
322048ffe56aSColin Percival# Actually install updates
322148ffe56aSColin Percivalinstall_run () {
322248ffe56aSColin Percival	echo -n "Installing updates..."
322348ffe56aSColin Percival
322448ffe56aSColin Percival	# Make sure we have all the files we should have
322548ffe56aSColin Percival	install_verify ${BDHASH}-install/INDEX-OLD	\
322648ffe56aSColin Percival	    ${BDHASH}-install/INDEX-NEW || return 1
322748ffe56aSColin Percival
322848ffe56aSColin Percival	# Remove system immutable flag from files
322948ffe56aSColin Percival	install_unschg ${BDHASH}-install/INDEX-OLD	\
323048ffe56aSColin Percival	    ${BDHASH}-install/INDEX-NEW || return 1
323148ffe56aSColin Percival
3232db6b0a61SColin Percival	# Install new files, delete old files, and update linker.hints
3233db6b0a61SColin Percival	install_files ${BDHASH}-install || return 1
323448ffe56aSColin Percival
323548ffe56aSColin Percival	# Rearrange bits to allow the installed updates to be rolled back
323648ffe56aSColin Percival	install_setup_rollback
323748ffe56aSColin Percival
323848ffe56aSColin Percival	echo " done."
323948ffe56aSColin Percival}
324048ffe56aSColin Percival
324148ffe56aSColin Percival# Rearrange bits to allow the previous set of updates to be rolled back next.
324248ffe56aSColin Percivalrollback_setup_rollback () {
324348ffe56aSColin Percival	if [ -L ${BDHASH}-rollback/rollback ]; then
324448ffe56aSColin Percival		mv ${BDHASH}-rollback/rollback rollback-tmp
324548ffe56aSColin Percival		rm -r ${BDHASH}-rollback/
324648ffe56aSColin Percival		rm ${BDHASH}-rollback
324748ffe56aSColin Percival		mv rollback-tmp ${BDHASH}-rollback
324848ffe56aSColin Percival	else
324948ffe56aSColin Percival		rm -r ${BDHASH}-rollback/
325048ffe56aSColin Percival		rm ${BDHASH}-rollback
325148ffe56aSColin Percival	fi
325248ffe56aSColin Percival}
325348ffe56aSColin Percival
3254db6b0a61SColin Percival# Install old files, delete new files, and update linker.hints
3255db6b0a61SColin Percivalrollback_files () {
32566b27e1f2SEd Maste	# Create directories first.  They may be needed by files we will
32576b27e1f2SEd Maste	# install in subsequent steps (PR273950).
32586b27e1f2SEd Maste	awk -F \| '{if ($2 == "d") print }' $1/INDEX-OLD > INDEX-OLD
32596b27e1f2SEd Maste	install_from_index INDEX-OLD || return 1
32606b27e1f2SEd Maste
32611ec4fb3aSColin Percival	# Install old shared library files which don't have the same path as
32621ec4fb3aSColin Percival	# a new shared library file.
32631ec4fb3aSColin Percival	grep -vE '^/boot/' $1/INDEX-NEW |
3264fd0963d1SColin Percival	    grep -E '/lib/.*\.so\.[0-9]+\|' |
32651ec4fb3aSColin Percival	    cut -f 1 -d '|' |
32661ec4fb3aSColin Percival	    sort > INDEX-NEW.libs.flist
3267db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-OLD |
3268fd0963d1SColin Percival	    grep -E '/lib/.*\.so\.[0-9]+\|' |
32691ec4fb3aSColin Percival	    sort -k 1,1 -t '|' - |
32701ec4fb3aSColin Percival	    join -t '|' -v 1 - INDEX-NEW.libs.flist > INDEX-OLD
3271db6b0a61SColin Percival	install_from_index INDEX-OLD || return 1
3272db6b0a61SColin Percival
3273db6b0a61SColin Percival	# Deal with files which are neither kernel nor shared library
3274db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-OLD |
3275fd0963d1SColin Percival	    grep -vE '/lib/.*\.so\.[0-9]+\|' > INDEX-OLD
3276db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-NEW |
3277fd0963d1SColin Percival	    grep -vE '/lib/.*\.so\.[0-9]+\|' > INDEX-NEW
3278db6b0a61SColin Percival	install_from_index INDEX-OLD || return 1
3279db6b0a61SColin Percival	install_delete INDEX-NEW INDEX-OLD || return 1
3280db6b0a61SColin Percival
32811ec4fb3aSColin Percival	# Install any old shared library files which we didn't install above.
32821ec4fb3aSColin Percival	grep -vE '^/boot/' $1/INDEX-OLD |
3283fd0963d1SColin Percival	    grep -E '/lib/.*\.so\.[0-9]+\|' |
32841ec4fb3aSColin Percival	    sort -k 1,1 -t '|' - |
32851ec4fb3aSColin Percival	    join -t '|' - INDEX-NEW.libs.flist > INDEX-OLD
32861ec4fb3aSColin Percival	install_from_index INDEX-OLD || return 1
32871ec4fb3aSColin Percival
3288db6b0a61SColin Percival	# Delete unneeded shared library files
3289db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-OLD |
3290fd0963d1SColin Percival	    grep -E '/lib/.*\.so\.[0-9]+\|' > INDEX-OLD
3291db6b0a61SColin Percival	grep -vE '^/boot/' $1/INDEX-NEW |
3292fd0963d1SColin Percival	    grep -E '/lib/.*\.so\.[0-9]+\|' > INDEX-NEW
3293db6b0a61SColin Percival	install_delete INDEX-NEW INDEX-OLD || return 1
3294db6b0a61SColin Percival
3295db6b0a61SColin Percival	# Deal with kernel files
3296db6b0a61SColin Percival	grep -E '^/boot/' $1/INDEX-OLD > INDEX-OLD
3297db6b0a61SColin Percival	grep -E '^/boot/' $1/INDEX-NEW > INDEX-NEW
3298db6b0a61SColin Percival	install_from_index INDEX-OLD || return 1
3299db6b0a61SColin Percival	install_delete INDEX-NEW INDEX-OLD || return 1
3300db6b0a61SColin Percival	if [ -s INDEX-OLD -o -s INDEX-NEW ]; then
3301db6b0a61SColin Percival		kldxref -R /boot/ 2>/dev/null
3302db6b0a61SColin Percival	fi
3303db6b0a61SColin Percival
3304db6b0a61SColin Percival	# Remove temporary files
33050e0d8d5aSColin Percival	rm INDEX-OLD INDEX-NEW INDEX-NEW.libs.flist
3306db6b0a61SColin Percival}
3307db6b0a61SColin Percival
330848ffe56aSColin Percival# Actually rollback updates
330948ffe56aSColin Percivalrollback_run () {
331048ffe56aSColin Percival	echo -n "Uninstalling updates..."
331148ffe56aSColin Percival
331248ffe56aSColin Percival	# If there are updates waiting to be installed, remove them; we
331348ffe56aSColin Percival	# want the user to re-run 'fetch' after rolling back updates.
331448ffe56aSColin Percival	if [ -L ${BDHASH}-install ]; then
331548ffe56aSColin Percival		rm -r ${BDHASH}-install/
331648ffe56aSColin Percival		rm ${BDHASH}-install
331748ffe56aSColin Percival	fi
331848ffe56aSColin Percival
331948ffe56aSColin Percival	# Make sure we have all the files we should have
332048ffe56aSColin Percival	install_verify ${BDHASH}-rollback/INDEX-NEW	\
332148ffe56aSColin Percival	    ${BDHASH}-rollback/INDEX-OLD || return 1
332248ffe56aSColin Percival
332348ffe56aSColin Percival	# Remove system immutable flag from files
332448ffe56aSColin Percival	install_unschg ${BDHASH}-rollback/INDEX-NEW	\
332548ffe56aSColin Percival	    ${BDHASH}-rollback/INDEX-OLD || return 1
332648ffe56aSColin Percival
3327db6b0a61SColin Percival	# Install old files, delete new files, and update linker.hints
3328db6b0a61SColin Percival	rollback_files ${BDHASH}-rollback || return 1
332948ffe56aSColin Percival
333048ffe56aSColin Percival	# Remove the rollback directory and the symlink pointing to it; and
333148ffe56aSColin Percival	# rearrange bits to allow the previous set of updates to be rolled
333248ffe56aSColin Percival	# back next.
333348ffe56aSColin Percival	rollback_setup_rollback
333448ffe56aSColin Percival
333548ffe56aSColin Percival	echo " done."
333648ffe56aSColin Percival}
333748ffe56aSColin Percival
333808e23beeSColin Percival# Compare INDEX-ALL and INDEX-PRESENT and print warnings about differences.
333908e23beeSColin PercivalIDS_compare () {
3340bb10a826SColin Percival	# Get all the lines which mismatch in something other than file
3341bb10a826SColin Percival	# flags.  We ignore file flags because sysinstall doesn't seem to
3342bb10a826SColin Percival	# set them when it installs FreeBSD; warning about these adds a
3343bb10a826SColin Percival	# very large amount of noise.
3344bb10a826SColin Percival	cut -f 1-5,7-8 -d '|' $1 > $1.noflags
3345bb10a826SColin Percival	sort -k 1,1 -t '|' $1.noflags > $1.sorted
3346bb10a826SColin Percival	cut -f 1-5,7-8 -d '|' $2 |
3347bb10a826SColin Percival	    comm -13 $1.noflags - |
3348bb10a826SColin Percival	    fgrep -v '|-|||||' |
334908e23beeSColin Percival	    sort -k 1,1 -t '|' |
335008e23beeSColin Percival	    join -t '|' $1.sorted - > INDEX-NOTMATCHING
335108e23beeSColin Percival
335208e23beeSColin Percival	# Ignore files which match IDSIGNOREPATHS.
335308e23beeSColin Percival	for X in ${IDSIGNOREPATHS}; do
335408e23beeSColin Percival		grep -E "^${X}" INDEX-NOTMATCHING
335508e23beeSColin Percival	done |
335608e23beeSColin Percival	    sort -u |
335708e23beeSColin Percival	    comm -13 - INDEX-NOTMATCHING > INDEX-NOTMATCHING.tmp
335808e23beeSColin Percival	mv INDEX-NOTMATCHING.tmp INDEX-NOTMATCHING
335908e23beeSColin Percival
336008e23beeSColin Percival	# Go through the lines and print warnings.
3361aa60062eSColin Percival	local IFS='|'
3362aa60062eSColin Percival	while read FPATH TYPE OWNER GROUP PERM HASH LINK P_TYPE P_OWNER P_GROUP P_PERM P_HASH P_LINK; do
336308e23beeSColin Percival		# Warn about different object types.
336408e23beeSColin Percival		if ! [ "${TYPE}" = "${P_TYPE}" ]; then
336508e23beeSColin Percival			echo -n "${FPATH} is a "
336608e23beeSColin Percival			case "${P_TYPE}" in
336708e23beeSColin Percival			f)	echo -n "regular file, "
336808e23beeSColin Percival				;;
336908e23beeSColin Percival			d)	echo -n "directory, "
337008e23beeSColin Percival				;;
337108e23beeSColin Percival			L)	echo -n "symlink, "
337208e23beeSColin Percival				;;
337308e23beeSColin Percival			esac
337408e23beeSColin Percival			echo -n "but should be a "
337508e23beeSColin Percival			case "${TYPE}" in
337608e23beeSColin Percival			f)	echo -n "regular file."
337708e23beeSColin Percival				;;
337808e23beeSColin Percival			d)	echo -n "directory."
337908e23beeSColin Percival				;;
338008e23beeSColin Percival			L)	echo -n "symlink."
338108e23beeSColin Percival				;;
338208e23beeSColin Percival			esac
338308e23beeSColin Percival			echo
338408e23beeSColin Percival
338508e23beeSColin Percival			# Skip other tests, since they don't make sense if
338608e23beeSColin Percival			# we're comparing different object types.
338708e23beeSColin Percival			continue
338808e23beeSColin Percival		fi
338908e23beeSColin Percival
339008e23beeSColin Percival		# Warn about different owners.
339108e23beeSColin Percival		if ! [ "${OWNER}" = "${P_OWNER}" ]; then
339208e23beeSColin Percival			echo -n "${FPATH} is owned by user id ${P_OWNER}, "
339308e23beeSColin Percival			echo "but should be owned by user id ${OWNER}."
339408e23beeSColin Percival		fi
339508e23beeSColin Percival
339608e23beeSColin Percival		# Warn about different groups.
339708e23beeSColin Percival		if ! [ "${GROUP}" = "${P_GROUP}" ]; then
339808e23beeSColin Percival			echo -n "${FPATH} is owned by group id ${P_GROUP}, "
339908e23beeSColin Percival			echo "but should be owned by group id ${GROUP}."
340008e23beeSColin Percival		fi
340108e23beeSColin Percival
340208e23beeSColin Percival		# Warn about different permissions.  We do not warn about
340308e23beeSColin Percival		# different permissions on symlinks, since some archivers
340408e23beeSColin Percival		# don't extract symlink permissions correctly and they are
340508e23beeSColin Percival		# ignored anyway.
340608e23beeSColin Percival		if ! [ "${PERM}" = "${P_PERM}" ] &&
340708e23beeSColin Percival		    ! [ "${TYPE}" = "L" ]; then
340808e23beeSColin Percival			echo -n "${FPATH} has ${P_PERM} permissions, "
340908e23beeSColin Percival			echo "but should have ${PERM} permissions."
341008e23beeSColin Percival		fi
341108e23beeSColin Percival
341208e23beeSColin Percival		# Warn about different file hashes / symlink destinations.
341308e23beeSColin Percival		if ! [ "${HASH}" = "${P_HASH}" ]; then
341408e23beeSColin Percival			if [ "${TYPE}" = "L" ]; then
341508e23beeSColin Percival				echo -n "${FPATH} is a symlink to ${P_HASH}, "
341608e23beeSColin Percival				echo "but should be a symlink to ${HASH}."
341708e23beeSColin Percival			fi
341808e23beeSColin Percival			if [ "${TYPE}" = "f" ]; then
341908e23beeSColin Percival				echo -n "${FPATH} has SHA256 hash ${P_HASH}, "
342008e23beeSColin Percival				echo "but should have SHA256 hash ${HASH}."
342108e23beeSColin Percival			fi
342208e23beeSColin Percival		fi
342308e23beeSColin Percival
342408e23beeSColin Percival		# We don't warn about different hard links, since some
342508e23beeSColin Percival		# some archivers break hard links, and as long as the
342608e23beeSColin Percival		# underlying data is correct they really don't matter.
342708e23beeSColin Percival	done < INDEX-NOTMATCHING
342808e23beeSColin Percival
342908e23beeSColin Percival	# Clean up
3430bb10a826SColin Percival	rm $1 $1.noflags $1.sorted $2 INDEX-NOTMATCHING
343108e23beeSColin Percival}
343208e23beeSColin Percival
343308e23beeSColin Percival# Do the work involved in comparing the system to a "known good" index
343408e23beeSColin PercivalIDS_run () {
343508e23beeSColin Percival	workdir_init || return 1
343608e23beeSColin Percival
343708e23beeSColin Percival	# Prepare the mirror list.
343808e23beeSColin Percival	fetch_pick_server_init && fetch_pick_server
343908e23beeSColin Percival
344008e23beeSColin Percival	# Try to fetch the public key until we run out of servers.
344108e23beeSColin Percival	while ! fetch_key; do
344208e23beeSColin Percival		fetch_pick_server || return 1
344308e23beeSColin Percival	done
344408e23beeSColin Percival
344508e23beeSColin Percival	# Try to fetch the metadata index signature ("tag") until we run
344608e23beeSColin Percival	# out of available servers; and sanity check the downloaded tag.
344708e23beeSColin Percival	while ! fetch_tag; do
344808e23beeSColin Percival		fetch_pick_server || return 1
344908e23beeSColin Percival	done
345008e23beeSColin Percival	fetch_tagsanity || return 1
345108e23beeSColin Percival
345208e23beeSColin Percival	# Fetch INDEX-OLD and INDEX-ALL.
345308e23beeSColin Percival	fetch_metadata INDEX-OLD INDEX-ALL || return 1
345408e23beeSColin Percival
345508e23beeSColin Percival	# Generate filtered INDEX-OLD and INDEX-ALL files containing only
345608e23beeSColin Percival	# the components we want and without anything marked as "Ignore".
345708e23beeSColin Percival	fetch_filter_metadata INDEX-OLD || return 1
345808e23beeSColin Percival	fetch_filter_metadata INDEX-ALL || return 1
345908e23beeSColin Percival
346008e23beeSColin Percival	# Merge the INDEX-OLD and INDEX-ALL files into INDEX-ALL.
346108e23beeSColin Percival	sort INDEX-OLD INDEX-ALL > INDEX-ALL.tmp
346208e23beeSColin Percival	mv INDEX-ALL.tmp INDEX-ALL
346308e23beeSColin Percival	rm INDEX-OLD
346408e23beeSColin Percival
346508e23beeSColin Percival	# Translate /boot/${KERNCONF} to ${KERNELDIR}
346608e23beeSColin Percival	fetch_filter_kernel_names INDEX-ALL ${KERNCONF}
346708e23beeSColin Percival
346808e23beeSColin Percival	# Inspect the system and generate an INDEX-PRESENT file.
346908e23beeSColin Percival	fetch_inspect_system INDEX-ALL INDEX-PRESENT /dev/null || return 1
347008e23beeSColin Percival
347108e23beeSColin Percival	# Compare INDEX-ALL and INDEX-PRESENT and print warnings about any
347208e23beeSColin Percival	# differences.
347308e23beeSColin Percival	IDS_compare INDEX-ALL INDEX-PRESENT
347408e23beeSColin Percival}
347508e23beeSColin Percival
347648ffe56aSColin Percival#### Main functions -- call parameter-handling and core functions
347748ffe56aSColin Percival
347848ffe56aSColin Percival# Using the command line, configuration file, and defaults,
347948ffe56aSColin Percival# set all the parameters which are needed later.
348048ffe56aSColin Percivalget_params () {
348148ffe56aSColin Percival	init_params
348248ffe56aSColin Percival	parse_cmdline $@
348348ffe56aSColin Percival	parse_conffile
348448ffe56aSColin Percival	default_params
348548ffe56aSColin Percival}
348648ffe56aSColin Percival
348748ffe56aSColin Percival# Fetch command.  Make sure that we're being called
348848ffe56aSColin Percival# interactively, then run fetch_check_params and fetch_run
348948ffe56aSColin Percivalcmd_fetch () {
34909a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
34918bf2dcceSAllan Jude	if [ ! -t 0 -a $NOTTYOK -eq 0 ]; then
349248ffe56aSColin Percival		echo -n "`basename $0` fetch should not "
349348ffe56aSColin Percival		echo "be run non-interactively."
349448ffe56aSColin Percival		echo "Run `basename $0` cron instead."
349548ffe56aSColin Percival		exit 1
349648ffe56aSColin Percival	fi
349748ffe56aSColin Percival	fetch_check_params
349848ffe56aSColin Percival	fetch_run || exit 1
349933bd05c3SGuangyuan Yang	ISFETCHED=1
350048ffe56aSColin Percival}
350148ffe56aSColin Percival
350248ffe56aSColin Percival# Cron command.  Make sure the parameters are sensible; wait
350348ffe56aSColin Percival# rand(3600) seconds; then fetch updates.  While fetching updates,
350448ffe56aSColin Percival# send output to a temporary file; only print that file if the
350548ffe56aSColin Percival# fetching failed.
350648ffe56aSColin Percivalcmd_cron () {
350748ffe56aSColin Percival	fetch_check_params
350848ffe56aSColin Percival	sleep `jot -r 1 0 3600`
350948ffe56aSColin Percival
351048ffe56aSColin Percival	TMPFILE=`mktemp /tmp/freebsd-update.XXXXXX` || exit 1
35119a63bbc9SColin Percival	finalize_components_config ${COMPONENTS} >> ${TMPFILE}
351248ffe56aSColin Percival	if ! fetch_run >> ${TMPFILE} ||
351348ffe56aSColin Percival	    ! grep -q "No updates needed" ${TMPFILE} ||
351448ffe56aSColin Percival	    [ ${VERBOSELEVEL} = "debug" ]; then
351548ffe56aSColin Percival		mail -s "`hostname` security updates" ${MAILTO} < ${TMPFILE}
351648ffe56aSColin Percival	fi
351785c3ef77SMichael Osipov	ISFETCHED=1
351848ffe56aSColin Percival
351948ffe56aSColin Percival	rm ${TMPFILE}
352048ffe56aSColin Percival}
352148ffe56aSColin Percival
3522db6b0a61SColin Percival# Fetch files for upgrading to a new release.
3523db6b0a61SColin Percivalcmd_upgrade () {
35249a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
3525db6b0a61SColin Percival	upgrade_check_params
3526*bc0c6c9cSFernando Apesteguía	upgrade_check_kmod_ports
3527db6b0a61SColin Percival	upgrade_run || exit 1
3528db6b0a61SColin Percival}
3529db6b0a61SColin Percival
3530101d33b8SMichael Gmelin# Check if there are fetched updates ready to install.
3531101d33b8SMichael Gmelin# Chdir into the working directory.
35328cfda118SMichael Gmelincmd_updatesready () {
35339a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
3534101d33b8SMichael Gmelin	# Check if working directory exists (if not, no updates pending)
3535101d33b8SMichael Gmelin	if ! [ -e "${WORKDIR}" ]; then
3536101d33b8SMichael Gmelin		echo "No updates are available to install."
3537101d33b8SMichael Gmelin		exit 2
3538101d33b8SMichael Gmelin	fi
3539101d33b8SMichael Gmelin
3540101d33b8SMichael Gmelin	# Change into working directory (fail if no permission/directory etc.)
3541101d33b8SMichael Gmelin	cd ${WORKDIR} || exit 1
3542101d33b8SMichael Gmelin
35438cfda118SMichael Gmelin	# Construct a unique name from ${BASEDIR}
35448cfda118SMichael Gmelin	BDHASH=`echo ${BASEDIR} | sha256 -q`
35458cfda118SMichael Gmelin
35468cfda118SMichael Gmelin	# Check that we have updates ready to install
35478cfda118SMichael Gmelin	if ! [ -L ${BDHASH}-install ]; then
35488cfda118SMichael Gmelin		echo "No updates are available to install."
35498cfda118SMichael Gmelin		exit 2
35508cfda118SMichael Gmelin	fi
35518cfda118SMichael Gmelin
35528cfda118SMichael Gmelin	echo "There are updates available to install."
355359b02bb4SMichael Osipov	echo "Run '`basename $0` [options] install' to proceed."
35548cfda118SMichael Gmelin}
35558cfda118SMichael Gmelin
355648ffe56aSColin Percival# Install downloaded updates.
355748ffe56aSColin Percivalcmd_install () {
35589a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
355948ffe56aSColin Percival	install_check_params
3560f28f1389SDave Fullard	install_create_be
356148ffe56aSColin Percival	install_run || exit 1
356248ffe56aSColin Percival}
356348ffe56aSColin Percival
356448ffe56aSColin Percival# Rollback most recently installed updates.
356548ffe56aSColin Percivalcmd_rollback () {
35669a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
356748ffe56aSColin Percival	rollback_check_params
356848ffe56aSColin Percival	rollback_run || exit 1
356948ffe56aSColin Percival}
357048ffe56aSColin Percival
357108e23beeSColin Percival# Compare system against a "known good" index.
357208e23beeSColin Percivalcmd_IDS () {
35739a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
357408e23beeSColin Percival	IDS_check_params
357508e23beeSColin Percival	IDS_run || exit 1
357608e23beeSColin Percival}
357708e23beeSColin Percival
35788cfda118SMichael Gmelin# Output configuration.
35798cfda118SMichael Gmelincmd_showconfig () {
35809a63bbc9SColin Percival	finalize_components_config ${COMPONENTS}
35818cfda118SMichael Gmelin	for X in ${CONFIGOPTIONS}; do
35828cfda118SMichael Gmelin		echo $X=$(eval echo \$${X})
35838cfda118SMichael Gmelin	done
35848cfda118SMichael Gmelin}
35858cfda118SMichael Gmelin
358648ffe56aSColin Percival#### Entry point
358748ffe56aSColin Percival
358848ffe56aSColin Percival# Make sure we find utilities from the base system
358948ffe56aSColin Percivalexport PATH=/sbin:/bin:/usr/sbin:/usr/bin:${PATH}
359048ffe56aSColin Percival
35919c990fb2SGordon Tetlow# Set a pager if the user doesn't
35929c990fb2SGordon Tetlowif [ -z "$PAGER" ]; then
359347cc9ee1SAlan Somers	PAGER=/usr/bin/less
35949c990fb2SGordon Tetlowfi
35959c990fb2SGordon Tetlow
3596f2890dbdSColin Percival# Set LC_ALL in order to avoid problems with character ranges like [A-Z].
3597f2890dbdSColin Percivalexport LC_ALL=C
3598f2890dbdSColin Percival
3599e093c61bSEd Maste# Clear environment variables that may affect operation of tools that we use.
3600e093c61bSEd Masteunset GREP_OPTIONS
3601e093c61bSEd Maste
360248ffe56aSColin Percivalget_params $@
360348ffe56aSColin Percivalfor COMMAND in ${COMMANDS}; do
360448ffe56aSColin Percival	cmd_${COMMAND}
360548ffe56aSColin Percivaldone
3606