1#!/bin/sh 2 3#- 4# Copyright 2004-2006 Colin Percival 5# All rights reserved 6# 7# Redistribution and use in source and binary forms, with or without 8# modification, are permitted providing that the following conditions 9# are met: 10# 1. Redistributions of source code must retain the above copyright 11# notice, this list of conditions and the following disclaimer. 12# 2. Redistributions in binary form must reproduce the above copyright 13# notice, this list of conditions and the following disclaimer in the 14# documentation and/or other materials provided with the distribution. 15# 16# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 18# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY 20# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 24# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING 25# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 26# POSSIBILITY OF SUCH DAMAGE. 27 28# $FreeBSD$ 29 30#### Usage function -- called from command-line handling code. 31 32# Usage instructions. Options not listed: 33# --debug -- don't filter output from utilities 34# --no-stats -- don't show progress statistics while fetching files 35usage () { 36 cat <<EOF 37usage: `basename $0` [options] command ... [path] 38 39Options: 40 -b basedir -- Operate on a system mounted at basedir 41 (default: /) 42 -d workdir -- Store working files in workdir 43 (default: /var/db/freebsd-update/) 44 -f conffile -- Read configuration options from conffile 45 (default: /etc/freebsd-update.conf) 46 -k KEY -- Trust an RSA key with SHA256 hash of KEY 47 -s server -- Server from which to fetch updates 48 (default: update.FreeBSD.org) 49 -t address -- Mail output of cron command, if any, to address 50 (default: root) 51Commands: 52 fetch -- Fetch updates from server 53 cron -- Sleep rand(3600) seconds, fetch updates, and send an 54 email if updates were found 55 install -- Install downloaded updates 56 rollback -- Uninstall most recently installed updates 57EOF 58 exit 0 59} 60 61#### Configuration processing functions 62 63#- 64# Configuration options are set in the following order of priority: 65# 1. Command line options 66# 2. Configuration file options 67# 3. Default options 68# In addition, certain options (e.g., IgnorePaths) can be specified multiple 69# times and (as long as these are all in the same place, e.g., inside the 70# configuration file) they will accumulate. Finally, because the path to the 71# configuration file can be specified at the command line, the entire command 72# line must be processed before we start reading the configuration file. 73# 74# Sound like a mess? It is. Here's how we handle this: 75# 1. Initialize CONFFILE and all the options to "". 76# 2. Process the command line. Throw an error if a non-accumulating option 77# is specified twice. 78# 3. If CONFFILE is "", set CONFFILE to /etc/freebsd-update.conf . 79# 4. For all the configuration options X, set X_saved to X. 80# 5. Initialize all the options to "". 81# 6. Read CONFFILE line by line, parsing options. 82# 7. For each configuration option X, set X to X_saved iff X_saved is not "". 83# 8. Repeat steps 4-7, except setting options to their default values at (6). 84 85CONFIGOPTIONS="KEYPRINT WORKDIR SERVERNAME MAILTO ALLOWADD ALLOWDELETE 86 KEEPMODIFIEDMETADATA COMPONENTS IGNOREPATHS UPDATEIFUNMODIFIED 87 BASEDIR VERBOSELEVEL" 88 89# Set all the configuration options to "". 90nullconfig () { 91 for X in ${CONFIGOPTIONS}; do 92 eval ${X}="" 93 done 94} 95 96# For each configuration option X, set X_saved to X. 97saveconfig () { 98 for X in ${CONFIGOPTIONS}; do 99 eval ${X}_saved=\$${X} 100 done 101} 102 103# For each configuration option X, set X to X_saved if X_saved is not "". 104mergeconfig () { 105 for X in ${CONFIGOPTIONS}; do 106 eval _=\$${X}_saved 107 if ! [ -z "${_}" ]; then 108 eval ${X}=\$${X}_saved 109 fi 110 done 111} 112 113# Set the trusted keyprint. 114config_KeyPrint () { 115 if [ -z ${KEYPRINT} ]; then 116 KEYPRINT=$1 117 else 118 return 1 119 fi 120} 121 122# Set the working directory. 123config_WorkDir () { 124 if [ -z ${WORKDIR} ]; then 125 WORKDIR=$1 126 else 127 return 1 128 fi 129} 130 131# Set the name of the server (pool) from which to fetch updates 132config_ServerName () { 133 if [ -z ${SERVERNAME} ]; then 134 SERVERNAME=$1 135 else 136 return 1 137 fi 138} 139 140# Set the address to which 'cron' output will be mailed. 141config_MailTo () { 142 if [ -z ${MAILTO} ]; then 143 MAILTO=$1 144 else 145 return 1 146 fi 147} 148 149# Set whether FreeBSD Update is allowed to add files (or directories, or 150# symlinks) which did not previously exist. 151config_AllowAdd () { 152 if [ -z ${ALLOWADD} ]; then 153 case $1 in 154 [Yy][Ee][Ss]) 155 ALLOWADD=yes 156 ;; 157 [Nn][Oo]) 158 ALLOWADD=no 159 ;; 160 *) 161 return 1 162 ;; 163 esac 164 else 165 return 1 166 fi 167} 168 169# Set whether FreeBSD Update is allowed to remove files/directories/symlinks. 170config_AllowDelete () { 171 if [ -z ${ALLOWDELETE} ]; then 172 case $1 in 173 [Yy][Ee][Ss]) 174 ALLOWDELETE=yes 175 ;; 176 [Nn][Oo]) 177 ALLOWDELETE=no 178 ;; 179 *) 180 return 1 181 ;; 182 esac 183 else 184 return 1 185 fi 186} 187 188# Set whether FreeBSD Update should keep existing inode ownership, 189# permissions, and flags, in the event that they have been modified locally 190# after the release. 191config_KeepModifiedMetadata () { 192 if [ -z ${KEEPMODIFIEDMETADATA} ]; then 193 case $1 in 194 [Yy][Ee][Ss]) 195 KEEPMODIFIEDMETADATA=yes 196 ;; 197 [Nn][Oo]) 198 KEEPMODIFIEDMETADATA=no 199 ;; 200 *) 201 return 1 202 ;; 203 esac 204 else 205 return 1 206 fi 207} 208 209# Add to the list of components which should be kept updated. 210config_Components () { 211 for C in $@; do 212 COMPONENTS="${COMPONENTS} ${C}" 213 done 214} 215 216# Add to the list of paths under which updates will be ignored. 217config_IgnorePaths () { 218 for C in $@; do 219 IGNOREPATHS="${IGNOREPATHS} ${C}" 220 done 221} 222 223# Add to the list of paths within which updates will be performed only if the 224# file on disk has not been modified locally. 225config_UpdateIfUnmodified () { 226 for C in $@; do 227 UPDATEIFUNMODIFIED="${UPDATEIFUNMODIFIED} ${C}" 228 done 229} 230 231# Work on a FreeBSD installation mounted under $1 232config_BaseDir () { 233 if [ -z ${BASEDIR} ]; then 234 BASEDIR=$1 235 else 236 return 1 237 fi 238} 239 240# Define what happens to output of utilities 241config_VerboseLevel () { 242 if [ -z ${VERBOSELEVEL} ]; then 243 case $1 in 244 [Dd][Ee][Bb][Uu][Gg]) 245 VERBOSELEVEL=debug 246 ;; 247 [Nn][Oo][Ss][Tt][Aa][Tt][Ss]) 248 VERBOSELEVEL=nostats 249 ;; 250 [Ss][Tt][Aa][Tt][Ss]) 251 VERBOSELEVEL=stats 252 ;; 253 *) 254 return 1 255 ;; 256 esac 257 else 258 return 1 259 fi 260} 261 262# Handle one line of configuration 263configline () { 264 if [ $# -eq 0 ]; then 265 return 266 fi 267 268 OPT=$1 269 shift 270 config_${OPT} $@ 271} 272 273#### Parameter handling functions. 274 275# Initialize parameters to null, just in case they're 276# set in the environment. 277init_params () { 278 # Configration settings 279 nullconfig 280 281 # No configuration file set yet 282 CONFFILE="" 283 284 # No commands specified yet 285 COMMANDS="" 286} 287 288# Parse the command line 289parse_cmdline () { 290 while [ $# -gt 0 ]; do 291 case "$1" in 292 # Location of configuration file 293 -f) 294 if [ $# -eq 1 ]; then usage; fi 295 if [ ! -z "${CONFFILE}" ]; then usage; fi 296 shift; CONFFILE="$1" 297 ;; 298 299 # Configuration file equivalents 300 -b) 301 if [ $# -eq 1 ]; then usage; fi; shift 302 config_BaseDir $1 || usage 303 ;; 304 -d) 305 if [ $# -eq 1 ]; then usage; fi; shift 306 config_WorkDir $1 || usage 307 ;; 308 -k) 309 if [ $# -eq 1 ]; then usage; fi; shift 310 config_KeyPrint $1 || usage 311 ;; 312 -s) 313 if [ $# -eq 1 ]; then usage; fi; shift 314 config_ServerName $1 || usage 315 ;; 316 -t) 317 if [ $# -eq 1 ]; then usage; fi; shift 318 config_MailTo $1 || usage 319 ;; 320 -v) 321 if [ $# -eq 1 ]; then usage; fi; shift 322 config_VerboseLevel $1 || usage 323 ;; 324 325 # Aliases for "-v debug" and "-v nostats" 326 --debug) 327 config_VerboseLevel debug || usage 328 ;; 329 --no-stats) 330 config_VerboseLevel nostats || usage 331 ;; 332 333 # Commands 334 cron | fetch | install | rollback) 335 COMMANDS="${COMMANDS} $1" 336 ;; 337 338 # Anything else is an error 339 *) 340 usage 341 ;; 342 esac 343 shift 344 done 345 346 # Make sure we have at least one command 347 if [ -z "${COMMANDS}" ]; then 348 usage 349 fi 350} 351 352# Parse the configuration file 353parse_conffile () { 354 # If a configuration file was specified on the command line, check 355 # that it exists and is readable. 356 if [ ! -z "${CONFFILE}" ] && [ ! -r "${CONFFILE}" ]; then 357 echo -n "File does not exist " 358 echo -n "or is not readable: " 359 echo ${CONFFILE} 360 exit 1 361 fi 362 363 # If a configuration file was not specified on the command line, 364 # use the default configuration file path. If that default does 365 # not exist, give up looking for any configuration. 366 if [ -z "${CONFFILE}" ]; then 367 CONFFILE="/etc/freebsd-update.conf" 368 if [ ! -r "${CONFFILE}" ]; then 369 return 370 fi 371 fi 372 373 # Save the configuration options specified on the command line, and 374 # clear all the options in preparation for reading the config file. 375 saveconfig 376 nullconfig 377 378 # Read the configuration file. Anything after the first '#' is 379 # ignored, and any blank lines are ignored. 380 L=0 381 while read LINE; do 382 L=$(($L + 1)) 383 LINEX=`echo "${LINE}" | cut -f 1 -d '#'` 384 if ! configline ${LINEX}; then 385 echo "Error processing configuration file, line $L:" 386 echo "==> ${LINE}" 387 exit 1 388 fi 389 done < ${CONFFILE} 390 391 # Merge the settings read from the configuration file with those 392 # provided at the command line. 393 mergeconfig 394} 395 396# Provide some default parameters 397default_params () { 398 # Save any parameters already configured, and clear the slate 399 saveconfig 400 nullconfig 401 402 # Default configurations 403 config_WorkDir /var/db/freebsd-update 404 config_MailTo root 405 config_AllowAdd yes 406 config_AllowDelete yes 407 config_KeepModifiedMetadata yes 408 config_BaseDir / 409 config_VerboseLevel stats 410 411 # Merge these defaults into the earlier-configured settings 412 mergeconfig 413} 414 415# Set utility output filtering options, based on ${VERBOSELEVEL} 416fetch_setup_verboselevel () { 417 case ${VERBOSELEVEL} in 418 debug) 419 QUIETREDIR="/dev/stderr" 420 QUIETFLAG=" " 421 STATSREDIR="/dev/stderr" 422 DDSTATS=".." 423 XARGST="-t" 424 NDEBUG=" " 425 ;; 426 nostats) 427 QUIETREDIR="" 428 QUIETFLAG="" 429 STATSREDIR="/dev/null" 430 DDSTATS=".." 431 XARGST="" 432 NDEBUG="" 433 ;; 434 stats) 435 QUIETREDIR="/dev/null" 436 QUIETFLAG="-q" 437 STATSREDIR="/dev/stdout" 438 DDSTATS="" 439 XARGST="" 440 NDEBUG="-n" 441 ;; 442 esac 443} 444 445# Perform sanity checks and set some final parameters 446# in preparation for fetching files. Figure out which 447# set of updates should be downloaded: If the user is 448# running *-p[0-9]+, strip off the last part; if the 449# user is running -SECURITY, call it -RELEASE. Chdir 450# into the working directory. 451fetch_check_params () { 452 export HTTP_USER_AGENT="freebsd-update (${COMMAND}, `uname -r`)" 453 454 _SERVERNAME_z=\ 455"SERVERNAME must be given via command line or configuration file." 456 _KEYPRINT_z="Key must be given via -k option or configuration file." 457 _KEYPRINT_bad="Invalid key fingerprint: " 458 _WORKDIR_bad="Directory does not exist or is not writable: " 459 460 if [ -z "${SERVERNAME}" ]; then 461 echo -n "`basename $0`: " 462 echo "${_SERVERNAME_z}" 463 exit 1 464 fi 465 if [ -z "${KEYPRINT}" ]; then 466 echo -n "`basename $0`: " 467 echo "${_KEYPRINT_z}" 468 exit 1 469 fi 470 if ! echo "${KEYPRINT}" | grep -qE "^[0-9a-f]{64}$"; then 471 echo -n "`basename $0`: " 472 echo -n "${_KEYPRINT_bad}" 473 echo ${KEYPRINT} 474 exit 1 475 fi 476 if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then 477 echo -n "`basename $0`: " 478 echo -n "${_WORKDIR_bad}" 479 echo ${WORKDIR} 480 exit 1 481 fi 482 cd ${WORKDIR} || exit 1 483 484 # Generate release number. The s/SECURITY/RELEASE/ bit exists 485 # to provide an upgrade path for FreeBSD Update 1.x users, since 486 # the kernels provided by FreeBSD Update 1.x are always labelled 487 # as X.Y-SECURITY. 488 RELNUM=`uname -r | 489 sed -E 's,-p[0-9]+,,' | 490 sed -E 's,-SECURITY,-RELEASE,'` 491 ARCH=`uname -m` 492 FETCHDIR=${RELNUM}/${ARCH} 493 494 # Figure out what directory contains the running kernel 495 BOOTFILE=`sysctl -n kern.bootfile` 496 KERNELDIR=${BOOTFILE%/kernel} 497 if ! [ -d ${KERNELDIR} ]; then 498 echo "Cannot identify running kernel" 499 exit 1 500 fi 501 502 # Figure out what kernel configuration is running. We start with 503 # the output of `uname -i`, and then make the following adjustments: 504 # 1. Replace "SMP-GENERIC" with "SMP". Why the SMP kernel config 505 # file says "ident SMP-GENERIC", I don't know... 506 # 2. If the kernel claims to be GENERIC _and_ ${ARCH} is "amd64" 507 # _and_ `sysctl kern.version` contains a line which ends "/SMP", then 508 # we're running an SMP kernel. This mis-identification is a bug 509 # which was fixed in 6.2-STABLE. 510 KERNCONF=`uname -i` 511 if [ ${KERNCONF} = "SMP-GENERIC" ]; then 512 KERNCONF=SMP 513 fi 514 if [ ${KERNCONF} = "GENERIC" ] && [ ${ARCH} = "amd64" ]; then 515 if sysctl kern.version | grep -qE '/SMP$'; then 516 KERNCONF=SMP 517 fi 518 fi 519 520 # Define some paths 521 BSPATCH=/usr/bin/bspatch 522 SHA256=/sbin/sha256 523 PHTTPGET=/usr/libexec/phttpget 524 525 # Set up variables relating to VERBOSELEVEL 526 fetch_setup_verboselevel 527 528 # Construct a unique name from ${BASEDIR} 529 BDHASH=`echo ${BASEDIR} | sha256 -q` 530} 531 532# Perform sanity checks and set some final parameters in 533# preparation for installing updates. 534install_check_params () { 535 # Check that we are root. All sorts of things won't work otherwise. 536 if [ `id -u` != 0 ]; then 537 echo "You must be root to run this." 538 exit 1 539 fi 540 541 # Check that we have a working directory 542 _WORKDIR_bad="Directory does not exist or is not writable: " 543 if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then 544 echo -n "`basename $0`: " 545 echo -n "${_WORKDIR_bad}" 546 echo ${WORKDIR} 547 exit 1 548 fi 549 cd ${WORKDIR} || exit 1 550 551 # Construct a unique name from ${BASEDIR} 552 BDHASH=`echo ${BASEDIR} | sha256 -q` 553 554 # Check that we have updates ready to install 555 if ! [ -L ${BDHASH}-install ]; then 556 echo "No updates are available to install." 557 echo "Run '$0 fetch' first." 558 exit 1 559 fi 560 if ! [ -f ${BDHASH}-install/INDEX-OLD ] || 561 ! [ -f ${BDHASH}-install/INDEX-NEW ]; then 562 echo "Update manifest is corrupt -- this should never happen." 563 echo "Re-run '$0 fetch'." 564 exit 1 565 fi 566} 567 568# Perform sanity checks and set some final parameters in 569# preparation for UNinstalling updates. 570rollback_check_params () { 571 # Check that we are root. All sorts of things won't work otherwise. 572 if [ `id -u` != 0 ]; then 573 echo "You must be root to run this." 574 exit 1 575 fi 576 577 # Check that we have a working directory 578 _WORKDIR_bad="Directory does not exist or is not writable: " 579 if ! [ -d "${WORKDIR}" -a -w "${WORKDIR}" ]; then 580 echo -n "`basename $0`: " 581 echo -n "${_WORKDIR_bad}" 582 echo ${WORKDIR} 583 exit 1 584 fi 585 cd ${WORKDIR} || exit 1 586 587 # Construct a unique name from ${BASEDIR} 588 BDHASH=`echo ${BASEDIR} | sha256 -q` 589 590 # Check that we have updates ready to rollback 591 if ! [ -L ${BDHASH}-rollback ]; then 592 echo "No rollback directory found." 593 exit 1 594 fi 595 if ! [ -f ${BDHASH}-rollback/INDEX-OLD ] || 596 ! [ -f ${BDHASH}-rollback/INDEX-NEW ]; then 597 echo "Update manifest is corrupt -- this should never happen." 598 exit 1 599 fi 600} 601 602#### Core functionality -- the actual work gets done here 603 604# Use an SRV query to pick a server. If the SRV query doesn't provide 605# a useful answer, use the server name specified by the user. 606# Put another way... look up _http._tcp.${SERVERNAME} and pick a server 607# from that; or if no servers are returned, use ${SERVERNAME}. 608# This allows a user to specify "portsnap.freebsd.org" (in which case 609# portsnap will select one of the mirrors) or "portsnap5.tld.freebsd.org" 610# (in which case portsnap will use that particular server, since there 611# won't be an SRV entry for that name). 612# 613# We ignore the Port field, since we are always going to use port 80. 614 615# Fetch the mirror list, but do not pick a mirror yet. Returns 1 if 616# no mirrors are available for any reason. 617fetch_pick_server_init () { 618 : > serverlist_tried 619 620# Check that host(1) exists (i.e., that the system wasn't built with the 621# WITHOUT_BIND set) and don't try to find a mirror if it doesn't exist. 622 if ! which -s host; then 623 : > serverlist_full 624 return 1 625 fi 626 627 echo -n "Looking up ${SERVERNAME} mirrors... " 628 629# Issue the SRV query and pull out the Priority, Weight, and Target fields. 630# BIND 9 prints "$name has SRV record ..." while BIND 8 prints 631# "$name server selection ..."; we allow either format. 632 MLIST="_http._tcp.${SERVERNAME}" 633 host -t srv "${MLIST}" | 634 sed -nE "s/${MLIST} (has SRV record|server selection) //p" | 635 cut -f 1,2,4 -d ' ' | 636 sed -e 's/\.$//' | 637 sort > serverlist_full 638 639# If no records, give up -- we'll just use the server name we were given. 640 if [ `wc -l < serverlist_full` -eq 0 ]; then 641 echo "none found." 642 return 1 643 fi 644 645# Report how many mirrors we found. 646 echo `wc -l < serverlist_full` "mirrors found." 647 648# Generate a random seed for use in picking mirrors. If HTTP_PROXY 649# is set, this will be used to generate the seed; otherwise, the seed 650# will be random. 651 if [ -n "${HTTP_PROXY}${http_proxy}" ]; then 652 RANDVALUE=`sha256 -qs "${HTTP_PROXY}${http_proxy}" | 653 tr -d 'a-f' | 654 cut -c 1-9` 655 else 656 RANDVALUE=`jot -r 1 0 999999999` 657 fi 658} 659 660# Pick a mirror. Returns 1 if we have run out of mirrors to try. 661fetch_pick_server () { 662# Generate a list of not-yet-tried mirrors 663 sort serverlist_tried | 664 comm -23 serverlist_full - > serverlist 665 666# Have we run out of mirrors? 667 if [ `wc -l < serverlist` -eq 0 ]; then 668 echo "No mirrors remaining, giving up." 669 return 1 670 fi 671 672# Find the highest priority level (lowest numeric value). 673 SRV_PRIORITY=`cut -f 1 -d ' ' serverlist | sort -n | head -1` 674 675# Add up the weights of the response lines at that priority level. 676 SRV_WSUM=0; 677 while read X; do 678 case "$X" in 679 ${SRV_PRIORITY}\ *) 680 SRV_W=`echo $X | cut -f 2 -d ' '` 681 SRV_WSUM=$(($SRV_WSUM + $SRV_W)) 682 ;; 683 esac 684 done < serverlist 685 686# If all the weights are 0, pretend that they are all 1 instead. 687 if [ ${SRV_WSUM} -eq 0 ]; then 688 SRV_WSUM=`grep -E "^${SRV_PRIORITY} " serverlist | wc -l` 689 SRV_W_ADD=1 690 else 691 SRV_W_ADD=0 692 fi 693 694# Pick a value between 0 and the sum of the weights - 1 695 SRV_RND=`expr ${RANDVALUE} % ${SRV_WSUM}` 696 697# Read through the list of mirrors and set SERVERNAME. Write the line 698# corresponding to the mirror we selected into serverlist_tried so that 699# we won't try it again. 700 while read X; do 701 case "$X" in 702 ${SRV_PRIORITY}\ *) 703 SRV_W=`echo $X | cut -f 2 -d ' '` 704 SRV_W=$(($SRV_W + $SRV_W_ADD)) 705 if [ $SRV_RND -lt $SRV_W ]; then 706 SERVERNAME=`echo $X | cut -f 3 -d ' '` 707 echo "$X" >> serverlist_tried 708 break 709 else 710 SRV_RND=$(($SRV_RND - $SRV_W)) 711 fi 712 ;; 713 esac 714 done < serverlist 715} 716 717# Take a list of ${oldhash}|${newhash} and output a list of needed patches, 718# i.e., those for which we have ${oldhash} and don't have ${newhash}. 719fetch_make_patchlist () { 720 grep -vE "^([0-9a-f]{64})\|\1$" | 721 tr '|' ' ' | 722 while read X Y; do 723 if [ -f "files/${Y}.gz" ] || 724 [ ! -f "files/${X}.gz" ]; then 725 continue 726 fi 727 echo "${X}|${Y}" 728 done | uniq 729} 730 731# Print user-friendly progress statistics 732fetch_progress () { 733 LNC=0 734 while read x; do 735 LNC=$(($LNC + 1)) 736 if [ $(($LNC % 10)) = 0 ]; then 737 echo -n $LNC 738 elif [ $(($LNC % 2)) = 0 ]; then 739 echo -n . 740 fi 741 done 742 echo -n " " 743} 744 745# Initialize the working directory 746workdir_init () { 747 mkdir -p files 748 touch tINDEX.present 749} 750 751# Check that we have a public key with an appropriate hash, or 752# fetch the key if it doesn't exist. Returns 1 if the key has 753# not yet been fetched. 754fetch_key () { 755 if [ -r pub.ssl ] && [ `${SHA256} -q pub.ssl` = ${KEYPRINT} ]; then 756 return 0 757 fi 758 759 echo -n "Fetching public key from ${SERVERNAME}... " 760 rm -f pub.ssl 761 fetch ${QUIETFLAG} http://${SERVERNAME}/${FETCHDIR}/pub.ssl \ 762 2>${QUIETREDIR} || true 763 if ! [ -r pub.ssl ]; then 764 echo "failed." 765 return 1 766 fi 767 if ! [ `${SHA256} -q pub.ssl` = ${KEYPRINT} ]; then 768 echo "key has incorrect hash." 769 rm -f pub.ssl 770 return 1 771 fi 772 echo "done." 773} 774 775# Fetch metadata signature, aka "tag". 776fetch_tag () { 777 echo ${NDEBUG} "Fetching metadata signature from ${SERVERNAME}... " 778 rm -f latest.ssl 779 fetch ${QUIETFLAG} http://${SERVERNAME}/${FETCHDIR}/latest.ssl \ 780 2>${QUIETREDIR} || true 781 if ! [ -r latest.ssl ]; then 782 echo "failed." 783 return 1 784 fi 785 786 openssl rsautl -pubin -inkey pub.ssl -verify \ 787 < latest.ssl > tag.new 2>${QUIETREDIR} || true 788 rm latest.ssl 789 790 if ! [ `wc -l < tag.new` = 1 ] || 791 ! grep -qE \ 792 "^freebsd-update\|${ARCH}\|${RELNUM}\|[0-9]+\|[0-9a-f]{64}\|[0-9]{10}" \ 793 tag.new; then 794 echo "invalid signature." 795 return 1 796 fi 797 798 echo "done." 799 800 RELPATCHNUM=`cut -f 4 -d '|' < tag.new` 801 TINDEXHASH=`cut -f 5 -d '|' < tag.new` 802 EOLTIME=`cut -f 6 -d '|' < tag.new` 803} 804 805# Sanity-check the patch number in a tag, to make sure that we're not 806# going to "update" backwards and to prevent replay attacks. 807fetch_tagsanity () { 808 # Check that we're not going to move from -pX to -pY with Y < X. 809 RELPX=`uname -r | sed -E 's,.*-,,'` 810 if echo ${RELPX} | grep -qE '^p[0-9]+$'; then 811 RELPX=`echo ${RELPX} | cut -c 2-` 812 else 813 RELPX=0 814 fi 815 if [ "${RELPATCHNUM}" -lt "${RELPX}" ]; then 816 echo 817 echo -n "Files on mirror (${RELNUM}-p${RELPATCHNUM})" 818 echo " appear older than what" 819 echo "we are currently running (`uname -r`)!" 820 echo "Cowardly refusing to proceed any further." 821 return 1 822 fi 823 824 # If "tag" exists and corresponds to ${RELNUM}, make sure that 825 # it contains a patch number <= RELPATCHNUM, in order to protect 826 # against rollback (replay) attacks. 827 if [ -f tag ] && 828 grep -qE \ 829 "^freebsd-update\|${ARCH}\|${RELNUM}\|[0-9]+\|[0-9a-f]{64}\|[0-9]{10}" \ 830 tag; then 831 LASTRELPATCHNUM=`cut -f 4 -d '|' < tag` 832 833 if [ "${RELPATCHNUM}" -lt "${LASTRELPATCHNUM}" ]; then 834 echo 835 echo -n "Files on mirror (${RELNUM}-p${RELPATCHNUM})" 836 echo " are older than the" 837 echo -n "most recently seen updates" 838 echo " (${RELNUM}-p${LASTRELPATCHNUM})." 839 echo "Cowardly refusing to proceed any further." 840 return 1 841 fi 842 fi 843} 844 845# Fetch metadata index file 846fetch_metadata_index () { 847 echo ${NDEBUG} "Fetching metadata index... " 848 rm -f ${TINDEXHASH} 849 fetch ${QUIETFLAG} http://${SERVERNAME}/${FETCHDIR}/t/${TINDEXHASH} 850 2>${QUIETREDIR} 851 if ! [ -f ${TINDEXHASH} ]; then 852 echo "failed." 853 return 1 854 fi 855 if [ `${SHA256} -q ${TINDEXHASH}` != ${TINDEXHASH} ]; then 856 echo "update metadata index corrupt." 857 return 1 858 fi 859 echo "done." 860} 861 862# Print an error message about signed metadata being bogus. 863fetch_metadata_bogus () { 864 echo 865 echo "The update metadata$1 is correctly signed, but" 866 echo "failed an integrity check." 867 echo "Cowardly refusing to proceed any further." 868 return 1 869} 870 871# Construct tINDEX.new by merging the lines named in $1 from ${TINDEXHASH} 872# with the lines not named in $@ from tINDEX.present (if that file exists). 873fetch_metadata_index_merge () { 874 for METAFILE in $@; do 875 if [ `grep -E "^${METAFILE}\|" ${TINDEXHASH} | wc -l` \ 876 -ne 1 ]; then 877 fetch_metadata_bogus " index" 878 return 1 879 fi 880 881 grep -E "${METAFILE}\|" ${TINDEXHASH} 882 done | 883 sort > tINDEX.wanted 884 885 if [ -f tINDEX.present ]; then 886 join -t '|' -v 2 tINDEX.wanted tINDEX.present | 887 sort -m - tINDEX.wanted > tINDEX.new 888 rm tINDEX.wanted 889 else 890 mv tINDEX.wanted tINDEX.new 891 fi 892} 893 894# Sanity check all the lines of tINDEX.new. Even if more metadata lines 895# are added by future versions of the server, this won't cause problems, 896# since the only lines which appear in tINDEX.new are the ones which we 897# specifically grepped out of ${TINDEXHASH}. 898fetch_metadata_index_sanity () { 899 if grep -qvE '^[0-9A-Z.-]+\|[0-9a-f]{64}$' tINDEX.new; then 900 fetch_metadata_bogus " index" 901 return 1 902 fi 903} 904 905# Sanity check the metadata file $1. 906fetch_metadata_sanity () { 907 # Some aliases to save space later: ${P} is a character which can 908 # appear in a path; ${M} is the four numeric metadata fields; and 909 # ${H} is a sha256 hash. 910 P="[-+./:=_[[:alnum:]]" 911 M="[0-9]+\|[0-9]+\|[0-9]+\|[0-9]+" 912 H="[0-9a-f]{64}" 913 914 # Check that the first four fields make sense. 915 if gunzip -c < files/$1.gz | 916 grep -qvE "^[a-z]+\|[0-9a-z]+\|${P}+\|[fdL-]\|"; then 917 fetch_metadata_bogus "" 918 return 1 919 fi 920 921 # Remove the first three fields. 922 gunzip -c < files/$1.gz | 923 cut -f 4- -d '|' > sanitycheck.tmp 924 925 # Sanity check entries with type 'f' 926 if grep -E '^f' sanitycheck.tmp | 927 grep -qvE "^f\|${M}\|${H}\|${P}*\$"; then 928 fetch_metadata_bogus "" 929 return 1 930 fi 931 932 # Sanity check entries with type 'd' 933 if grep -E '^d' sanitycheck.tmp | 934 grep -qvE "^d\|${M}\|\|\$"; then 935 fetch_metadata_bogus "" 936 return 1 937 fi 938 939 # Sanity check entries with type 'L' 940 if grep -E '^L' sanitycheck.tmp | 941 grep -qvE "^L\|${M}\|${P}*\|\$"; then 942 fetch_metadata_bogus "" 943 return 1 944 fi 945 946 # Sanity check entries with type '-' 947 if grep -E '^-' sanitycheck.tmp | 948 grep -qvE "^-\|\|\|\|\|\|"; then 949 fetch_metadata_bogus "" 950 return 1 951 fi 952 953 # Clean up 954 rm sanitycheck.tmp 955} 956 957# Fetch the metadata index and metadata files listed in $@, 958# taking advantage of metadata patches where possible. 959fetch_metadata () { 960 fetch_metadata_index || return 1 961 fetch_metadata_index_merge $@ || return 1 962 fetch_metadata_index_sanity || return 1 963 964 # Generate a list of wanted metadata patches 965 join -t '|' -o 1.2,2.2 tINDEX.present tINDEX.new | 966 fetch_make_patchlist > patchlist 967 968 if [ -s patchlist ]; then 969 # Attempt to fetch metadata patches 970 echo -n "Fetching `wc -l < patchlist | tr -d ' '` " 971 echo ${NDEBUG} "metadata patches.${DDSTATS}" 972 tr '|' '-' < patchlist | 973 lam -s "${FETCHDIR}/tp/" - -s ".gz" | 974 xargs ${XARGST} ${PHTTPGET} ${SERVERNAME} \ 975 2>${STATSREDIR} | fetch_progress 976 echo "done." 977 978 # Attempt to apply metadata patches 979 echo -n "Applying metadata patches... " 980 tr '|' ' ' < patchlist | 981 while read X Y; do 982 if [ ! -f "${X}-${Y}.gz" ]; then continue; fi 983 gunzip -c < ${X}-${Y}.gz > diff 984 gunzip -c < files/${X}.gz > diff-OLD 985 986 # Figure out which lines are being added and removed 987 grep -E '^-' diff | 988 cut -c 2- | 989 while read PREFIX; do 990 look "${PREFIX}" diff-OLD 991 done | 992 sort > diff-rm 993 grep -E '^\+' diff | 994 cut -c 2- > diff-add 995 996 # Generate the new file 997 comm -23 diff-OLD diff-rm | 998 sort - diff-add > diff-NEW 999 1000 if [ `${SHA256} -q diff-NEW` = ${Y} ]; then 1001 mv diff-NEW files/${Y} 1002 gzip -n files/${Y} 1003 else 1004 mv diff-NEW ${Y}.bad 1005 fi 1006 rm -f ${X}-${Y}.gz diff 1007 rm -f diff-OLD diff-NEW diff-add diff-rm 1008 done 2>${QUIETREDIR} 1009 echo "done." 1010 fi 1011 1012 # Update metadata without patches 1013 cut -f 2 -d '|' < tINDEX.new | 1014 while read Y; do 1015 if [ ! -f "files/${Y}.gz" ]; then 1016 echo ${Y}; 1017 fi 1018 done | 1019 sort -u > filelist 1020 1021 if [ -s filelist ]; then 1022 echo -n "Fetching `wc -l < filelist | tr -d ' '` " 1023 echo ${NDEBUG} "metadata files... " 1024 lam -s "${FETCHDIR}/m/" - -s ".gz" < filelist | 1025 xargs ${XARGST} ${PHTTPGET} ${SERVERNAME} \ 1026 2>${QUIETREDIR} 1027 1028 while read Y; do 1029 if ! [ -f ${Y}.gz ]; then 1030 echo "failed." 1031 return 1 1032 fi 1033 if [ `gunzip -c < ${Y}.gz | 1034 ${SHA256} -q` = ${Y} ]; then 1035 mv ${Y}.gz files/${Y}.gz 1036 else 1037 echo "metadata is corrupt." 1038 return 1 1039 fi 1040 done < filelist 1041 echo "done." 1042 fi 1043 1044# Sanity-check the metadata files. 1045 cut -f 2 -d '|' tINDEX.new > filelist 1046 while read X; do 1047 fetch_metadata_sanity ${X} || return 1 1048 done < filelist 1049 1050# Remove files which are no longer needed 1051 cut -f 2 -d '|' tINDEX.present | 1052 sort > oldfiles 1053 cut -f 2 -d '|' tINDEX.new | 1054 sort | 1055 comm -13 - oldfiles | 1056 lam -s "files/" - -s ".gz" | 1057 xargs rm -f 1058 rm patchlist filelist oldfiles 1059 rm ${TINDEXHASH} 1060 1061# We're done! 1062 mv tINDEX.new tINDEX.present 1063 mv tag.new tag 1064 1065 return 0 1066} 1067 1068# Generate a filtered version of the metadata file $1 from the downloaded 1069# file, by fishing out the lines corresponding to components we're trying 1070# to keep updated, and then removing lines corresponding to paths we want 1071# to ignore. 1072fetch_filter_metadata () { 1073 METAHASH=`look "$1|" tINDEX.present | cut -f 2 -d '|'` 1074 gunzip -c < files/${METAHASH}.gz > $1.all 1075 1076 # Fish out the lines belonging to components we care about. 1077 # Canonicalize directory names by removing any trailing / in 1078 # order to avoid listing directories multiple times if they 1079 # belong to multiple components. Turning "/" into "" doesn't 1080 # matter, since we add a leading "/" when we use paths later. 1081 for C in ${COMPONENTS}; do 1082 look "`echo ${C} | tr '/' '|'`|" $1.all 1083 done | 1084 cut -f 3- -d '|' | 1085 sed -e 's,/|d|,|d|,' | 1086 sort -u > $1.tmp 1087 1088 # Figure out which lines to ignore and remove them. 1089 for X in ${IGNOREPATHS}; do 1090 grep -E "^${X}" $1.tmp 1091 done | 1092 sort -u | 1093 comm -13 - $1.tmp > $1 1094 1095 # Remove temporary files. 1096 rm $1.all $1.tmp 1097} 1098 1099# Filter the metadata file $1 by adding lines with "/boot/${KERNCONF}" 1100# replaced by ${KERNELDIR} (which is `sysctl -n kern.bootfile` minus the 1101# trailing "/kernel"); and if "/boot/${KERNCONF}" does not exist, remove 1102# the original lines which start with that. 1103# Put another way: Deal with the fact that the FOO kernel is sometimes 1104# installed in /boot/FOO/ and is sometimes installed elsewhere. 1105fetch_filter_kernel_names () { 1106 1107 grep ^/boot/${KERNCONF} $1 | 1108 sed -e "s,/boot/${KERNCONF},${KERNELDIR},g" | 1109 sort - $1 > $1.tmp 1110 mv $1.tmp $1 1111 1112 if ! [ -d /boot/${KERNCONF} ]; then 1113 grep -v ^/boot/${KERNCONF} $1 > $1.tmp 1114 mv $1.tmp $1 1115 fi 1116} 1117 1118# For all paths appearing in $1 or $3, inspect the system 1119# and generate $2 describing what is currently installed. 1120fetch_inspect_system () { 1121 # No errors yet... 1122 rm -f .err 1123 1124 # Tell the user why his disk is suddenly making lots of noise 1125 echo -n "Inspecting system... " 1126 1127 # Generate list of files to inspect 1128 cat $1 $3 | 1129 cut -f 1 -d '|' | 1130 sort -u > filelist 1131 1132 # Examine each file and output lines of the form 1133 # /path/to/file|type|device-inum|user|group|perm|flags|value 1134 # sorted by device and inode number. 1135 while read F; do 1136 # If the symlink/file/directory does not exist, record this. 1137 if ! [ -e ${BASEDIR}/${F} ]; then 1138 echo "${F}|-||||||" 1139 continue 1140 fi 1141 if ! [ -r ${BASEDIR}/${F} ]; then 1142 echo "Cannot read file: ${BASEDIR}/${F}" \ 1143 >/dev/stderr 1144 touch .err 1145 return 1 1146 fi 1147 1148 # Otherwise, output an index line. 1149 if [ -L ${BASEDIR}/${F} ]; then 1150 echo -n "${F}|L|" 1151 stat -n -f '%d-%i|%u|%g|%Mp%Lp|%Of|' ${BASEDIR}/${F}; 1152 readlink ${BASEDIR}/${F}; 1153 elif [ -f ${BASEDIR}/${F} ]; then 1154 echo -n "${F}|f|" 1155 stat -n -f '%d-%i|%u|%g|%Mp%Lp|%Of|' ${BASEDIR}/${F}; 1156 sha256 -q ${BASEDIR}/${F}; 1157 elif [ -d ${BASEDIR}/${F} ]; then 1158 echo -n "${F}|d|" 1159 stat -f '%d-%i|%u|%g|%Mp%Lp|%Of|' ${BASEDIR}/${F}; 1160 else 1161 echo "Unknown file type: ${BASEDIR}/${F}" \ 1162 >/dev/stderr 1163 touch .err 1164 return 1 1165 fi 1166 done < filelist | 1167 sort -k 3,3 -t '|' > $2.tmp 1168 rm filelist 1169 1170 # Check if an error occured during system inspection 1171 if [ -f .err ]; then 1172 return 1 1173 fi 1174 1175 # Convert to the form 1176 # /path/to/file|type|user|group|perm|flags|value|hlink 1177 # by resolving identical device and inode numbers into hard links. 1178 cut -f 1,3 -d '|' $2.tmp | 1179 sort -k 1,1 -t '|' | 1180 sort -s -u -k 2,2 -t '|' | 1181 join -1 2 -2 3 -t '|' - $2.tmp | 1182 awk -F \| -v OFS=\| \ 1183 '{ 1184 if (($2 == $3) || ($4 == "-")) 1185 print $3,$4,$5,$6,$7,$8,$9,"" 1186 else 1187 print $3,$4,$5,$6,$7,$8,$9,$2 1188 }' | 1189 sort > $2 1190 rm $2.tmp 1191 1192 # We're finished looking around 1193 echo "done." 1194} 1195 1196# For any paths matching ${UPDATEIFUNMODIFIED}, remove lines from $[123] 1197# which correspond to lines in $2 with hashes not matching $1 or $3. For 1198# entries in $2 marked "not present" (aka. type -), remove lines from $[123] 1199# unless there is a corresponding entry in $1. 1200fetch_filter_unmodified_notpresent () { 1201 # Figure out which lines of $1 and $3 correspond to bits which 1202 # should only be updated if they haven't changed, and fish out 1203 # the (path, type, value) tuples. 1204 # NOTE: We don't consider a file to be "modified" if it matches 1205 # the hash from $3. 1206 for X in ${UPDATEIFUNMODIFIED}; do 1207 grep -E "^${X}" $1 1208 grep -E "^${X}" $3 1209 done | 1210 cut -f 1,2,7 -d '|' | 1211 sort > $1-values 1212 1213 # Do the same for $2. 1214 for X in ${UPDATEIFUNMODIFIED}; do 1215 grep -E "^${X}" $2 1216 done | 1217 cut -f 1,2,7 -d '|' | 1218 sort > $2-values 1219 1220 # Any entry in $2-values which is not in $1-values corresponds to 1221 # a path which we need to remove from $1, $2, and $3. 1222 comm -13 $1-values $2-values > mlines 1223 rm $1-values $2-values 1224 1225 # Any lines in $2 which are not in $1 AND are "not present" lines 1226 # also belong in mlines. 1227 comm -13 $1 $2 | 1228 cut -f 1,2,7 -d '|' | 1229 fgrep '|-|' >> mlines 1230 1231 # Remove lines from $1, $2, and $3 1232 for X in $1 $2 $3; do 1233 sort -t '|' -k 1,1 ${X} > ${X}.tmp 1234 cut -f 1 -d '|' < mlines | 1235 sort | 1236 join -v 2 -t '|' - ${X}.tmp | 1237 sort > ${X} 1238 rm ${X}.tmp 1239 done 1240 1241 # Store a list of the modified files, for future reference 1242 fgrep -v '|-|' mlines | 1243 cut -f 1 -d '|' > modifiedfiles 1244 rm mlines 1245} 1246 1247# For each entry in $1 of type -, remove any corresponding 1248# entry from $2 if ${ALLOWADD} != "yes". Remove all entries 1249# of type - from $1. 1250fetch_filter_allowadd () { 1251 cut -f 1,2 -d '|' < $1 | 1252 fgrep '|-' | 1253 cut -f 1 -d '|' > filesnotpresent 1254 1255 if [ ${ALLOWADD} != "yes" ]; then 1256 sort < $2 | 1257 join -v 1 -t '|' - filesnotpresent | 1258 sort > $2.tmp 1259 mv $2.tmp $2 1260 fi 1261 1262 sort < $1 | 1263 join -v 1 -t '|' - filesnotpresent | 1264 sort > $1.tmp 1265 mv $1.tmp $1 1266 rm filesnotpresent 1267} 1268 1269# If ${ALLOWDELETE} != "yes", then remove any entries from $1 1270# which don't correspond to entries in $2. 1271fetch_filter_allowdelete () { 1272 # Produce a lists ${PATH}|${TYPE} 1273 for X in $1 $2; do 1274 cut -f 1-2 -d '|' < ${X} | 1275 sort -u > ${X}.nodes 1276 done 1277 1278 # Figure out which lines need to be removed from $1. 1279 if [ ${ALLOWDELETE} != "yes" ]; then 1280 comm -23 $1.nodes $2.nodes > $1.badnodes 1281 else 1282 : > $1.badnodes 1283 fi 1284 1285 # Remove the relevant lines from $1 1286 while read X; do 1287 look "${X}|" $1 1288 done < $1.badnodes | 1289 comm -13 - $1 > $1.tmp 1290 mv $1.tmp $1 1291 1292 rm $1.badnodes $1.nodes $2.nodes 1293} 1294 1295# If ${KEEPMODIFIEDMETADATA} == "yes", then for each entry in $2 1296# with metadata not matching any entry in $1, replace the corresponding 1297# line of $3 with one having the same metadata as the entry in $2. 1298fetch_filter_modified_metadata () { 1299 # Fish out the metadata from $1 and $2 1300 for X in $1 $2; do 1301 cut -f 1-6 -d '|' < ${X} > ${X}.metadata 1302 done 1303 1304 # Find the metadata we need to keep 1305 if [ ${KEEPMODIFIEDMETADATA} = "yes" ]; then 1306 comm -13 $1.metadata $2.metadata > keepmeta 1307 else 1308 : > keepmeta 1309 fi 1310 1311 # Extract the lines which we need to remove from $3, and 1312 # construct the lines which we need to add to $3. 1313 : > $3.remove 1314 : > $3.add 1315 while read LINE; do 1316 NODE=`echo "${LINE}" | cut -f 1-2 -d '|'` 1317 look "${NODE}|" $3 >> $3.remove 1318 look "${NODE}|" $3 | 1319 cut -f 7- -d '|' | 1320 lam -s "${LINE}|" - >> $3.add 1321 done < keepmeta 1322 1323 # Remove the specified lines and add the new lines. 1324 sort $3.remove | 1325 comm -13 - $3 | 1326 sort -u - $3.add > $3.tmp 1327 mv $3.tmp $3 1328 1329 rm keepmeta $1.metadata $2.metadata $3.add $3.remove 1330} 1331 1332# Remove lines from $1 and $2 which are identical; 1333# no need to update a file if it isn't changing. 1334fetch_filter_uptodate () { 1335 comm -23 $1 $2 > $1.tmp 1336 comm -13 $1 $2 > $2.tmp 1337 1338 mv $1.tmp $1 1339 mv $2.tmp $2 1340} 1341 1342# Prepare to fetch files: Generate a list of the files we need, 1343# copy the unmodified files we have into /files/, and generate 1344# a list of patches to download. 1345fetch_files_prepare () { 1346 # Tell the user why his disk is suddenly making lots of noise 1347 echo -n "Preparing to download files... " 1348 1349 # Reduce indices to ${PATH}|${HASH} pairs 1350 for X in $1 $2 $3; do 1351 cut -f 1,2,7 -d '|' < ${X} | 1352 fgrep '|f|' | 1353 cut -f 1,3 -d '|' | 1354 sort > ${X}.hashes 1355 done 1356 1357 # List of files wanted 1358 cut -f 2 -d '|' < $3.hashes | 1359 sort -u > files.wanted 1360 1361 # Generate a list of unmodified files 1362 comm -12 $1.hashes $2.hashes | 1363 sort -k 1,1 -t '|' > unmodified.files 1364 1365 # Copy all files into /files/. We only need the unmodified files 1366 # for use in patching; but we'll want all of them if the user asks 1367 # to rollback the updates later. 1368 cut -f 1 -d '|' < $2.hashes | 1369 while read F; do 1370 cp "${BASEDIR}/${F}" tmpfile 1371 gzip -c < tmpfile > files/`sha256 -q tmpfile`.gz 1372 rm tmpfile 1373 done 1374 1375 # Produce a list of patches to download 1376 sort -k 1,1 -t '|' $3.hashes | 1377 join -t '|' -o 2.2,1.2 - unmodified.files | 1378 fetch_make_patchlist > patchlist 1379 1380 # Garbage collect 1381 rm unmodified.files $1.hashes $2.hashes $3.hashes 1382 1383 # We don't need the list of possible old files any more. 1384 rm $1 1385 1386 # We're finished making noise 1387 echo "done." 1388} 1389 1390# Fetch files. 1391fetch_files () { 1392 # Attempt to fetch patches 1393 if [ -s patchlist ]; then 1394 echo -n "Fetching `wc -l < patchlist | tr -d ' '` " 1395 echo ${NDEBUG} "patches.${DDSTATS}" 1396 tr '|' '-' < patchlist | 1397 lam -s "${FETCHDIR}/bp/" - | 1398 xargs ${XARGST} ${PHTTPGET} ${SERVERNAME} \ 1399 2>${STATSREDIR} | fetch_progress 1400 echo "done." 1401 1402 # Attempt to apply patches 1403 echo -n "Applying patches... " 1404 tr '|' ' ' < patchlist | 1405 while read X Y; do 1406 if [ ! -f "${X}-${Y}" ]; then continue; fi 1407 gunzip -c < files/${X}.gz > OLD 1408 1409 bspatch OLD NEW ${X}-${Y} 1410 1411 if [ `${SHA256} -q NEW` = ${Y} ]; then 1412 mv NEW files/${Y} 1413 gzip -n files/${Y} 1414 fi 1415 rm -f diff OLD NEW ${X}-${Y} 1416 done 2>${QUIETREDIR} 1417 echo "done." 1418 fi 1419 1420 # Download files which couldn't be generate via patching 1421 while read Y; do 1422 if [ ! -f "files/${Y}.gz" ]; then 1423 echo ${Y}; 1424 fi 1425 done < files.wanted > filelist 1426 1427 if [ -s filelist ]; then 1428 echo -n "Fetching `wc -l < filelist | tr -d ' '` " 1429 echo ${NDEBUG} "files... " 1430 lam -s "${FETCHDIR}/f/" - -s ".gz" < filelist | 1431 xargs ${XARGST} ${PHTTPGET} ${SERVERNAME} \ 1432 2>${QUIETREDIR} 1433 1434 while read Y; do 1435 if ! [ -f ${Y}.gz ]; then 1436 echo "failed." 1437 return 1 1438 fi 1439 if [ `gunzip -c < ${Y}.gz | 1440 ${SHA256} -q` = ${Y} ]; then 1441 mv ${Y}.gz files/${Y}.gz 1442 else 1443 echo "${Y} has incorrect hash." 1444 return 1 1445 fi 1446 done < filelist 1447 echo "done." 1448 fi 1449 1450 # Clean up 1451 rm files.wanted filelist patchlist 1452} 1453 1454# Create and populate install manifest directory; and report what updates 1455# are available. 1456fetch_create_manifest () { 1457 # If we have an existing install manifest, nuke it. 1458 if [ -L "${BDHASH}-install" ]; then 1459 rm -r ${BDHASH}-install/ 1460 rm ${BDHASH}-install 1461 fi 1462 1463 # Report to the user if any updates were avoided due to local changes 1464 if [ -s modifiedfiles ]; then 1465 echo 1466 echo -n "The following files are affected by updates, " 1467 echo "but no changes have" 1468 echo -n "been downloaded because the files have been " 1469 echo "modified locally:" 1470 cat modifiedfiles 1471 fi 1472 rm modifiedfiles 1473 1474 # If no files will be updated, tell the user and exit 1475 if ! [ -s INDEX-PRESENT ] && 1476 ! [ -s INDEX-NEW ]; then 1477 rm INDEX-PRESENT INDEX-NEW 1478 echo 1479 echo -n "No updates needed to update system to " 1480 echo "${RELNUM}-p${RELPATCHNUM}." 1481 return 1482 fi 1483 1484 # Divide files into (a) removed files, (b) added files, and 1485 # (c) updated files. 1486 cut -f 1 -d '|' < INDEX-PRESENT | 1487 sort > INDEX-PRESENT.flist 1488 cut -f 1 -d '|' < INDEX-NEW | 1489 sort > INDEX-NEW.flist 1490 comm -23 INDEX-PRESENT.flist INDEX-NEW.flist > files.removed 1491 comm -13 INDEX-PRESENT.flist INDEX-NEW.flist > files.added 1492 comm -12 INDEX-PRESENT.flist INDEX-NEW.flist > files.updated 1493 rm INDEX-PRESENT.flist INDEX-NEW.flist 1494 1495 # Report removed files, if any 1496 if [ -s files.removed ]; then 1497 echo 1498 echo -n "The following files will be removed " 1499 echo "as part of updating to ${RELNUM}-p${RELPATCHNUM}:" 1500 cat files.removed 1501 fi 1502 rm files.removed 1503 1504 # Report added files, if any 1505 if [ -s files.added ]; then 1506 echo 1507 echo -n "The following files will be added " 1508 echo "as part of updating to ${RELNUM}-p${RELPATCHNUM}:" 1509 cat files.added 1510 fi 1511 rm files.added 1512 1513 # Report updated files, if any 1514 if [ -s files.updated ]; then 1515 echo 1516 echo -n "The following files will be updated " 1517 echo "as part of updating to ${RELNUM}-p${RELPATCHNUM}:" 1518 1519 cat files.updated 1520 fi 1521 rm files.updated 1522 1523 # Create a directory for the install manifest. 1524 MDIR=`mktemp -d install.XXXXXX` || return 1 1525 1526 # Populate it 1527 mv INDEX-PRESENT ${MDIR}/INDEX-OLD 1528 mv INDEX-NEW ${MDIR}/INDEX-NEW 1529 1530 # Link it into place 1531 ln -s ${MDIR} ${BDHASH}-install 1532} 1533 1534# Warn about any upcoming EoL 1535fetch_warn_eol () { 1536 # What's the current time? 1537 NOWTIME=`date "+%s"` 1538 1539 # When did we last warn about the EoL date? 1540 if [ -f lasteolwarn ]; then 1541 LASTWARN=`cat lasteolwarn` 1542 else 1543 LASTWARN=`expr ${NOWTIME} - 63072000` 1544 fi 1545 1546 # If the EoL time is past, warn. 1547 if [ ${EOLTIME} -lt ${NOWTIME} ]; then 1548 echo 1549 cat <<-EOF 1550 WARNING: `uname -sr` HAS PASSED ITS END-OF-LIFE DATE. 1551 Any security issues discovered after `date -r ${EOLTIME}` 1552 will not have been corrected. 1553 EOF 1554 return 1 1555 fi 1556 1557 # Figure out how long it has been since we last warned about the 1558 # upcoming EoL, and how much longer we have left. 1559 SINCEWARN=`expr ${NOWTIME} - ${LASTWARN}` 1560 TIMELEFT=`expr ${EOLTIME} - ${NOWTIME}` 1561 1562 # Don't warn if the EoL is more than 6 months away 1563 if [ ${TIMELEFT} -gt 15768000 ]; then 1564 return 0 1565 fi 1566 1567 # Don't warn if the time remaining is more than 3 times the time 1568 # since the last warning. 1569 if [ ${TIMELEFT} -gt `expr ${SINCEWARN} \* 3` ]; then 1570 return 0 1571 fi 1572 1573 # Figure out what time units to use. 1574 if [ ${TIMELEFT} -lt 604800 ]; then 1575 UNIT="day" 1576 SIZE=86400 1577 elif [ ${TIMELEFT} -lt 2678400 ]; then 1578 UNIT="week" 1579 SIZE=604800 1580 else 1581 UNIT="month" 1582 SIZE=2678400 1583 fi 1584 1585 # Compute the right number of units 1586 NUM=`expr ${TIMELEFT} / ${SIZE}` 1587 if [ ${NUM} != 1 ]; then 1588 UNIT="${UNIT}s" 1589 fi 1590 1591 # Print the warning 1592 echo 1593 cat <<-EOF 1594 WARNING: `uname -sr` is approaching its End-of-Life date. 1595 It is strongly recommended that you upgrade to a newer 1596 release within the next ${NUM} ${UNIT}. 1597 EOF 1598 1599 # Update the stored time of last warning 1600 echo ${NOWTIME} > lasteolwarn 1601} 1602 1603# Do the actual work involved in "fetch" / "cron". 1604fetch_run () { 1605 workdir_init || return 1 1606 1607 # Prepare the mirror list. 1608 fetch_pick_server_init && fetch_pick_server 1609 1610 # Try to fetch the public key until we run out of servers. 1611 while ! fetch_key; do 1612 fetch_pick_server || return 1 1613 done 1614 1615 # Try to fetch the metadata index signature ("tag") until we run 1616 # out of available servers; and sanity check the downloaded tag. 1617 while ! fetch_tag; do 1618 fetch_pick_server || return 1 1619 done 1620 fetch_tagsanity || return 1 1621 1622 # Fetch the latest INDEX-NEW and INDEX-OLD files. 1623 fetch_metadata INDEX-NEW INDEX-OLD || return 1 1624 1625 # Generate filtered INDEX-NEW and INDEX-OLD files containing only 1626 # the lines which (a) belong to components we care about, and (b) 1627 # don't correspond to paths we're explicitly ignoring. 1628 fetch_filter_metadata INDEX-NEW || return 1 1629 fetch_filter_metadata INDEX-OLD || return 1 1630 1631 # Translate /boot/`uname -i` into ${KERNELDIR} 1632 fetch_filter_kernel_names INDEX-NEW 1633 fetch_filter_kernel_names INDEX-OLD 1634 1635 # For all paths appearing in INDEX-OLD or INDEX-NEW, inspect the 1636 # system and generate an INDEX-PRESENT file. 1637 fetch_inspect_system INDEX-OLD INDEX-PRESENT INDEX-NEW || return 1 1638 1639 # Based on ${UPDATEIFUNMODIFIED}, remove lines from INDEX-* which 1640 # correspond to lines in INDEX-PRESENT with hashes not appearing 1641 # in INDEX-OLD or INDEX-NEW. Also remove lines where the entry in 1642 # INDEX-PRESENT has type - and there isn't a corresponding entry in 1643 # INDEX-OLD with type -. 1644 fetch_filter_unmodified_notpresent INDEX-OLD INDEX-PRESENT INDEX-NEW 1645 1646 # For each entry in INDEX-PRESENT of type -, remove any corresponding 1647 # entry from INDEX-NEW if ${ALLOWADD} != "yes". Remove all entries 1648 # of type - from INDEX-PRESENT. 1649 fetch_filter_allowadd INDEX-PRESENT INDEX-NEW 1650 1651 # If ${ALLOWDELETE} != "yes", then remove any entries from 1652 # INDEX-PRESENT which don't correspond to entries in INDEX-NEW. 1653 fetch_filter_allowdelete INDEX-PRESENT INDEX-NEW 1654 1655 # If ${KEEPMODIFIEDMETADATA} == "yes", then for each entry in 1656 # INDEX-PRESENT with metadata not matching any entry in INDEX-OLD, 1657 # replace the corresponding line of INDEX-NEW with one having the 1658 # same metadata as the entry in INDEX-PRESENT. 1659 fetch_filter_modified_metadata INDEX-OLD INDEX-PRESENT INDEX-NEW 1660 1661 # Remove lines from INDEX-PRESENT and INDEX-NEW which are identical; 1662 # no need to update a file if it isn't changing. 1663 fetch_filter_uptodate INDEX-PRESENT INDEX-NEW 1664 1665 # Prepare to fetch files: Generate a list of the files we need, 1666 # copy the unmodified files we have into /files/, and generate 1667 # a list of patches to download. 1668 fetch_files_prepare INDEX-OLD INDEX-PRESENT INDEX-NEW 1669 1670 # Fetch files. 1671 fetch_files || return 1 1672 1673 # Create and populate install manifest directory; and report what 1674 # updates are available. 1675 fetch_create_manifest || return 1 1676 1677 # Warn about any upcoming EoL 1678 fetch_warn_eol || return 1 1679} 1680 1681# Make sure that all the file hashes mentioned in $@ have corresponding 1682# gzipped files stored in /files/. 1683install_verify () { 1684 # Generate a list of hashes 1685 cat $@ | 1686 cut -f 2,7 -d '|' | 1687 grep -E '^f' | 1688 cut -f 2 -d '|' | 1689 sort -u > filelist 1690 1691 # Make sure all the hashes exist 1692 while read HASH; do 1693 if ! [ -f files/${HASH}.gz ]; then 1694 echo -n "Update files missing -- " 1695 echo "this should never happen." 1696 echo "Re-run '$0 fetch'." 1697 return 1 1698 fi 1699 done < filelist 1700 1701 # Clean up 1702 rm filelist 1703} 1704 1705# Remove the system immutable flag from files 1706install_unschg () { 1707 # Generate file list 1708 cat $@ | 1709 cut -f 1 -d '|' > filelist 1710 1711 # Remove flags 1712 while read F; do 1713 if ! [ -e ${F} ]; then 1714 continue 1715 fi 1716 1717 chflags noschg ${F} || return 1 1718 done < filelist 1719 1720 # Clean up 1721 rm filelist 1722} 1723 1724# Install new files 1725install_from_index () { 1726 # First pass: Do everything apart from setting file flags. We 1727 # can't set flags yet, because schg inhibits hard linking. 1728 sort -k 1,1 -t '|' $1 | 1729 tr '|' ' ' | 1730 while read FPATH TYPE OWNER GROUP PERM FLAGS HASH LINK; do 1731 case ${TYPE} in 1732 d) 1733 # Create a directory 1734 install -d -o ${OWNER} -g ${GROUP} \ 1735 -m ${PERM} ${BASEDIR}/${FPATH} 1736 ;; 1737 f) 1738 if [ -z "${LINK}" ]; then 1739 # Create a file, without setting flags. 1740 gunzip < files/${HASH}.gz > ${HASH} 1741 install -S -o ${OWNER} -g ${GROUP} \ 1742 -m ${PERM} ${HASH} ${BASEDIR}/${FPATH} 1743 rm ${HASH} 1744 else 1745 # Create a hard link. 1746 ln -f ${LINK} ${BASEDIR}/${FPATH} 1747 fi 1748 ;; 1749 L) 1750 # Create a symlink 1751 ln -sfh ${HASH} ${BASEDIR}/${FPATH} 1752 ;; 1753 esac 1754 done 1755 1756 # Perform a second pass, adding file flags. 1757 tr '|' ' ' < $1 | 1758 while read FPATH TYPE OWNER GROUP PERM FLAGS HASH LINK; do 1759 if [ ${TYPE} = "f" ] && 1760 ! [ ${FLAGS} = "0" ]; then 1761 chflags ${FLAGS} ${BASEDIR}/${FPATH} 1762 fi 1763 done 1764} 1765 1766# Remove files which we want to delete 1767install_delete () { 1768 # Generate list of new files 1769 cut -f 1 -d '|' < $2 | 1770 sort > newfiles 1771 1772 # Generate subindex of old files we want to nuke 1773 sort -k 1,1 -t '|' $1 | 1774 join -t '|' -v 1 - newfiles | 1775 sort -r -k 1,1 -t '|' | 1776 cut -f 1,2 -d '|' | 1777 tr '|' ' ' > killfiles 1778 1779 # Remove the offending bits 1780 while read FPATH TYPE; do 1781 case ${TYPE} in 1782 d) 1783 rmdir ${BASEDIR}/${FPATH} 1784 ;; 1785 f) 1786 rm ${BASEDIR}/${FPATH} 1787 ;; 1788 L) 1789 rm ${BASEDIR}/${FPATH} 1790 ;; 1791 esac 1792 done < killfiles 1793 1794 # Clean up 1795 rm newfiles killfiles 1796} 1797 1798# Update linker.hints if anything in /boot/ was touched 1799install_kldxref () { 1800 if cat $@ | 1801 grep -qE '^/boot/'; then 1802 kldxref -R /boot/ 1803 fi 1804} 1805 1806# Rearrange bits to allow the installed updates to be rolled back 1807install_setup_rollback () { 1808 if [ -L ${BDHASH}-rollback ]; then 1809 mv ${BDHASH}-rollback ${BDHASH}-install/rollback 1810 fi 1811 1812 mv ${BDHASH}-install ${BDHASH}-rollback 1813} 1814 1815# Actually install updates 1816install_run () { 1817 echo -n "Installing updates..." 1818 1819 # Make sure we have all the files we should have 1820 install_verify ${BDHASH}-install/INDEX-OLD \ 1821 ${BDHASH}-install/INDEX-NEW || return 1 1822 1823 # Remove system immutable flag from files 1824 install_unschg ${BDHASH}-install/INDEX-OLD \ 1825 ${BDHASH}-install/INDEX-NEW || return 1 1826 1827 # Install new files 1828 install_from_index \ 1829 ${BDHASH}-install/INDEX-NEW || return 1 1830 1831 # Remove files which we want to delete 1832 install_delete ${BDHASH}-install/INDEX-OLD \ 1833 ${BDHASH}-install/INDEX-NEW || return 1 1834 1835 # Update linker.hints if anything in /boot/ was touched 1836 install_kldxref ${BDHASH}-install/INDEX-OLD \ 1837 ${BDHASH}-install/INDEX-NEW 1838 1839 # Rearrange bits to allow the installed updates to be rolled back 1840 install_setup_rollback 1841 1842 echo " done." 1843} 1844 1845# Rearrange bits to allow the previous set of updates to be rolled back next. 1846rollback_setup_rollback () { 1847 if [ -L ${BDHASH}-rollback/rollback ]; then 1848 mv ${BDHASH}-rollback/rollback rollback-tmp 1849 rm -r ${BDHASH}-rollback/ 1850 rm ${BDHASH}-rollback 1851 mv rollback-tmp ${BDHASH}-rollback 1852 else 1853 rm -r ${BDHASH}-rollback/ 1854 rm ${BDHASH}-rollback 1855 fi 1856} 1857 1858# Actually rollback updates 1859rollback_run () { 1860 echo -n "Uninstalling updates..." 1861 1862 # If there are updates waiting to be installed, remove them; we 1863 # want the user to re-run 'fetch' after rolling back updates. 1864 if [ -L ${BDHASH}-install ]; then 1865 rm -r ${BDHASH}-install/ 1866 rm ${BDHASH}-install 1867 fi 1868 1869 # Make sure we have all the files we should have 1870 install_verify ${BDHASH}-rollback/INDEX-NEW \ 1871 ${BDHASH}-rollback/INDEX-OLD || return 1 1872 1873 # Remove system immutable flag from files 1874 install_unschg ${BDHASH}-rollback/INDEX-NEW \ 1875 ${BDHASH}-rollback/INDEX-OLD || return 1 1876 1877 # Install new files 1878 install_from_index \ 1879 ${BDHASH}-rollback/INDEX-OLD || return 1 1880 1881 # Remove files which we want to delete 1882 install_delete ${BDHASH}-rollback/INDEX-NEW \ 1883 ${BDHASH}-rollback/INDEX-OLD || return 1 1884 1885 # Update linker.hints if anything in /boot/ was touched 1886 install_kldxref ${BDHASH}-rollback/INDEX-NEW \ 1887 ${BDHASH}-rollback/INDEX-OLD 1888 1889 # Remove the rollback directory and the symlink pointing to it; and 1890 # rearrange bits to allow the previous set of updates to be rolled 1891 # back next. 1892 rollback_setup_rollback 1893 1894 echo " done." 1895} 1896 1897#### Main functions -- call parameter-handling and core functions 1898 1899# Using the command line, configuration file, and defaults, 1900# set all the parameters which are needed later. 1901get_params () { 1902 init_params 1903 parse_cmdline $@ 1904 parse_conffile 1905 default_params 1906} 1907 1908# Fetch command. Make sure that we're being called 1909# interactively, then run fetch_check_params and fetch_run 1910cmd_fetch () { 1911 if [ ! -t 0 ]; then 1912 echo -n "`basename $0` fetch should not " 1913 echo "be run non-interactively." 1914 echo "Run `basename $0` cron instead." 1915 exit 1 1916 fi 1917 fetch_check_params 1918 fetch_run || exit 1 1919} 1920 1921# Cron command. Make sure the parameters are sensible; wait 1922# rand(3600) seconds; then fetch updates. While fetching updates, 1923# send output to a temporary file; only print that file if the 1924# fetching failed. 1925cmd_cron () { 1926 fetch_check_params 1927 sleep `jot -r 1 0 3600` 1928 1929 TMPFILE=`mktemp /tmp/freebsd-update.XXXXXX` || exit 1 1930 if ! fetch_run >> ${TMPFILE} || 1931 ! grep -q "No updates needed" ${TMPFILE} || 1932 [ ${VERBOSELEVEL} = "debug" ]; then 1933 mail -s "`hostname` security updates" ${MAILTO} < ${TMPFILE} 1934 fi 1935 1936 rm ${TMPFILE} 1937} 1938 1939# Install downloaded updates. 1940cmd_install () { 1941 install_check_params 1942 install_run || exit 1 1943} 1944 1945# Rollback most recently installed updates. 1946cmd_rollback () { 1947 rollback_check_params 1948 rollback_run || exit 1 1949} 1950 1951#### Entry point 1952 1953# Make sure we find utilities from the base system 1954export PATH=/sbin:/bin:/usr/sbin:/usr/bin:${PATH} 1955 1956# Set LC_ALL in order to avoid problems with character ranges like [A-Z]. 1957export LC_ALL=C 1958 1959get_params $@ 1960for COMMAND in ${COMMANDS}; do 1961 cmd_${COMMAND} 1962done 1963