[BACK]Return to fw_install.sh CVS log [TXT][DIR] Up to [local] / openbsd / fw_update

Diff for /openbsd/fw_update/fw_install.sh between version 1.4 and 1.33

version 1.4, 2021/10/05 01:49:17 version 1.33, 2021/11/19 03:13:11
Line 1 
Line 1 
 #!/bin/ksh  #!/bin/ksh
 set -e  set -e
   
 scan_dmesg() {  # Fake up some things from install.sub that we don't need to actually do
         # no bsort for now  prefetcharea_fs_list() {
         sed -n "$1" /var/run/dmesg.boot          echo "${DESTDIR}/tmp"
 }  }
   
 installed_firmware() {  # tmpdir, do_as, unpriv, and unpriv2 are from install.sub
         for fw in ${PKGDIR}/$1-firmware*; do  
                 [ -e "$fw" ] || continue  # Create a temporary directory based on the supplied directory name prefix.
                 echo ${fw##*/}  tmpdir() {
           local _i=1 _dir
   
           until _dir="${1?}.$_i.$RANDOM" && mkdir -- "$_dir" 2>/dev/null; do
                   ((++_i < 10000)) || return 1
         done          done
           echo "$_dir"
 }  }
   
 set -A _KERNV -- $( scan_dmesg '/^OpenBSD \([1-9][0-9]*\.[0-9]\)\([^ ]*\) .*/ { s//\1 \2/p; q; }' )  # Run a command ($2+) as unprivileged user ($1).
 VNAME=${_KERNV[0]}  # Take extra care that after "cmd" no "user" processes exist.
 OSDIR=$VNAME  #
 if ((${#_KERNV[*]} > 1)) && [ "$_KERNV[1]" = "-current" -o "$_KERNV[1]" = "-beta" ]; then  # Optionally:
         OSDIR=snapshots  #       - create "file" and chown it to "user"
 fi  #       - after "cmd", chown "file" back to root
   #
   # Usage: do_as user [-f file] cmd
   do_as() {
           (( $# >= 2 )) || return
   
 FWURL=http://firmware.openbsd.org/firmware/${OSDIR}          local _file _rc _user=$1
 PKGDIR=${DESTDIR}/var/db/pkg          shift
 PATTERNS="file:${0%/*}/firmware_patterns"  
   
 drivers=$(          if [[ $1 == -f ]]; then
         last=''                  _file=$2
         ftp -D "Detecting" -Vmo- $PATTERNS |                  shift 2
         while read d m; do          fi
                 grep=fgrep  
                 [ "$last" = "$d" ] && continue          if [[ -n $_file ]]; then
                 [ "$m" ] || m="^$d[0-9][0-9]* at "                  >$_file
                 [ "$m" != "${m#^}" ] && grep=grep                  chown "$_user" "$_file"
                 $grep -q "$m" /var/run/dmesg.boot || continue          fi
                 echo $d  
                 last=$d          doas -u "$_user" "$@"
           _rc=$?
   
           while doas -u "$_user" kill -9 -1 2>/dev/null; do
                   echo "Processes still running for user $_user after: $@"
                   sleep 1
         done          done
 )  
   
 if [ -z "$drivers" ]; then          [[ -n $_file ]] && chown root "$_file"
         echo "No devices found which need firmware files to be downloaded." >&2  
         exit 0  
 fi  
   
 tmpdir=${DESTDIR}/tmp/fw_update          return $_rc
 [ -e "$tmpdir" ] && rm -rf "$tmpdir"  }
 mkdir -p "$tmpdir"  
 cd "$tmpdir"  
   
 # TODO: Drop privs during fetch and verify  unpriv() {
 ftp -D Get -Vm "$FWURL/SHA256.sig"          do_as _sndio "$@"
   }
   
 # Probably should bundle the firmware sigfile on the installer,  unpriv2() {
 # although we can just get it from the recently installed system.          do_as _file "$@"
 if [ "$DESTDIR" ]; then  }
         sigfile=$( sed -n '/^untrusted comment: verify with \(.*\)$/ { s//\1/p; q; }' SHA256.sig )  
         if [ ! -e "/etc/signify/$sigfile" ] \  VNAME=${VNAME:-$(sysctl -n kern.osrelease)}
           && [ -e "${DESTDIR}/etc/signify/$sigfile" ]; then  VERSION=${VERSION:"${VNAME%.*}${VNAME#*.}"}
                 cp -a "${DESTDIR}/etc/signify/$sigfile" "/etc/signify/$sigfile"  FWDIR=${FWDIR:-$VNAME}
   MODE=${MODE:-install}
   
   # TODO: We need the firmware for the system we just installed
   #       not the one we booted from.  For example:
   #       * booting from a snapshot bsd.rd that thinks it is the 7.0 release
   #         will install the firmware from the 7.0 directory instead of
   #         from the snapshots dir.
   #       If they're using sysupgrade, then the installer kernel will be correct.
   #       If we're doing this in the installer we can check what they picked
   #       for downloading sets and use that value.
   #       Otherwise, the fw_update after first boot will fix it up for us.
   
   HTTP_FWDIR=$FWDIR
   set -- sed -n "/^OpenBSD $VNAME\([^ ]*\).*$/s//\1/p" /var/run/dmesg.boot
   [[ $1 == -!(stable) ]] && HTTP_FWDIR=snapshots
   
   FWURL=http://firmware.openbsd.org/firmware/${HTTP_FWDIR}
   FWPUB_KEY=${DESTDIR}/etc/signify/openbsd-${VERSION}-fw.pub
   FWPATTERNS="file:${0%/*}/firmware_patterns"
   
   # TODO: support srclocal installation of firmware somehow
   fw_update() {
           local _src=$1 _tmpfs_list _tmpfs _tmpsrc \
                   _t=Get _cfile="/tmp/SHA256" _pkgdir=${DESTDIR}/var/db/pkg \
                   _f _r _remove _i _installed
           local _srclocal=false _unpriv=unpriv
   
           echo "Let's $MODE firmware!"
           local _d _drivers=$(
                   last=''
                   $_unpriv ftp -D "Detecting" -Vmo- $FWPATTERNS |
                   while read _d _m; do
                           grep=grep
                           [ "$last" = "$_d" ] && continue
                           [ "$_m" ] || _m="^$_d[0-9][0-9]* at "
                           [ "$_m" = "${_m#^}" ] && grep=fgrep
                           $grep -q "$_m" /var/run/dmesg.boot || continue
                           echo $_d
                           last=$_d
                   done
           )
   
           if [ -z "$_drivers" ]; then
                   echo "No devices found which need firmware files to be downloaded."
                   return
         fi          fi
 fi  
   
 signify -Ve -x SHA256.sig -m - < SHA256.sig > SHA256          ! _tmpfs_list=$(prefetcharea_fs_list) &&
                   echo "Cannot determine prefetch area" >&2 && return
   
 for d in $drivers; do          for _tmpfs in $_tmpfs_list; do
         firmware=$( sed -n "s/.*(\($d-firmware-.*\.tgz\)).*/\1/p" SHA256 )                  # Try to clean up from previous runs, assuming
         installed=$( installed_firmware $d )                  # the _tmpfs selection yields the same mount
                   # point.
                   for _tmpsrc in $_tmpfs/firmware.+([0-9]).+([0-9]); do
                           [[ -d $_tmpsrc ]] && rm -r $_tmpsrc
                   done
   
         for i in $installed; do                  # Create a download directory for the firmware and
                 if [ "$firmware" = "$i.tgz" ]; then                  # check that the _sndio user can read files from
                         echo "Firmware for $d already installed ($installed)"                  # it. Otherwise cleanup and skip the filesystem.
                         continue 2                  if _tmpsrc=$(tmpdir "$_tmpfs/firmware"); then
                           (
                           >$_tmpsrc/t &&
                           $_unpriv cat $_tmpsrc/t
                           ) >/dev/null 2>&1 && break ||
                                   rm -r $_tmpsrc
                 fi                  fi
         done          done
   
         mkdir $d          [[ ! -d $_tmpsrc ]] &&
                   echo "Cannot create prefetch area" >&2 && return 1
   
         # TODO: Drop privs during fetch and verify          # Cleanup from previous runs.
         ftp -D "Get/Verify" -Vmo- "$FWURL/$firmware" | sha256 -bph "$d/h" > "$firmware"          rm -f $_cfile $_cfile.sig
         fgrep -qx "SHA256 ($firmware) = $(<$d/h)" SHA256  
   
         # TODO: Check hash for files before deleting          _t=Get/Verify
         if [ "$installed" ] && [ -e "${PKGDIR}/$installed/+CONTENTS" ]; then  
                 echo "Uninstalling $installed"  
                 cwd=${PKGDIR}/$installed  
   
                 remove="${cwd}/+CONTENTS ${cwd}"          ! $_unpriv ftp -D "$_t" -Vmo - "$_src/SHA256.sig" >"$_cfile.sig" &&
               echo "Cannot fetch SHA256.sig" >&2 && return 1
   
                 while read c g; do          # Verify signature file with public keys.
                         case $c in          ! $_unpriv -f "$_cfile" \
                         @cwd) cwd=$g              signify -Vep $FWPUB_KEY -x "$_cfile.sig" -m "$_cfile" &&
                           ;;              echo "Signature check of SHA256.sig failed" >&2 && return 1
                         @*) continue  
                           ;;  
                         *)  remove="$cwd/$c $remove"  
                           ;;  
                         esac  
                 done < "${PKGDIR}/$installed/+CONTENTS"  
   
                 for r in $remove ; do          for _d in $_drivers; do
                         if [ -d "$r" ]; then                  _f=$( sed -n "s/.*(\($_d-firmware-.*\.tgz\)).*/\1/p" "$_cfile" )
                                 # Try hard not to actually remove recursively                  _installed=$(
                                 # without rmdir on the install media.                  for fw in "${_pkgdir}/$_d-firmware"*; do
                                 [ "$r/*" = $( echo "$r"/* ) ] && rm -rf "$r"                          [ -e "$fw" ] || continue
                         else                          echo ${fw##*/}
                                 rm -f "$r"                  done
                   )
   
                   for _i in $_installed; do
                           if [ "$_f" = "$_i.tgz" ]; then
                                   echo "$_i already installed"
                                   continue 2
                         fi                          fi
                 done                  done
         fi  
   
         # TODO: Add some details about the install to +CONTENTS like pkg_add                  rm -f /tmp/h /tmp/fail
         # TODO: Or, maybe we save the firmware someplace and make pkg_add reinstall  
         echo "Installing $firmware"                  # Fetch firmware file and create a checksum by piping through
         tar -zxphf "$firmware" -C /etc "firmware/*"                  # sha256. Create a flag file in case ftp failed. Firmware
         mkdir -p ${PKGDIR}/${firmware%.tgz}/                  # from net is written to the prefetch area.
         tar -zxphf "$firmware" -C "${PKGDIR}/${firmware%.tgz}" "+*"                  ( $_unpriv ftp -D "$_t" -Vmo - "$_src/$_f" || >/tmp/fail ) |
         ed -s "${PKGDIR}/${firmware%.tgz}/+CONTENTS" <<EOL                  ( $_srclocal && unpriv2 sha256 -b >/tmp/h ||
                       unpriv2 -f /tmp/h sha256 -bph /tmp/h >"$_tmpsrc/$_f" )
   
                   # Handle failed transfer.
                   if [[ -f /tmp/fail ]]; then
                           rm -f "$_tmpsrc/$_f"
                           echo "Fetching of $_f failed!" >&2
                           continue
                   fi
   
                   # Verify firmware by comparing its checksum with SHA256.
                   if ! fgrep -qx "SHA256 ($_f) = $(</tmp/h)" "$_cfile"; then
                           [[ -d "$_tmpsrc" ]] && rm -rf "$_tmpsrc"
                           echo "Checksum test for $_f failed." >&2
                           continue
                   fi
   
                   # TODO: Check hash for files before deleting
                   if [ "$_installed" ] && [ -e "${_pkgdir}/$_installed/+CONTENTS" ]; then
                           echo "Uninstalling $_installed"
                           cwd=${_pkgdir}/$_installed
   
                           set -A _remove -- "${cwd}/+CONTENTS" "${cwd}"
   
                           while read c g; do
                                   case $c in
                                   @cwd) cwd="${DESTDIR}/$g"
                                     ;;
                                   @*) continue
                                     ;;
                                   *)  set -A _remove -- "$cwd/$c" "${_remove[@]}"
                                     ;;
                                   esac
                           done < "${_pkgdir}/$_installed/+CONTENTS"
   
                           # We specifically rm -f here because not removing files/dirs
                           # is probably not worth failing over.
                           for _r in "${_remove[@]}" ; do
                                   if [ -d "$_r" ]; then
                                           # Try hard not to actually remove recursively
                                           # without rmdir on the install media.
                                           [ "$_r/*" = $( echo "$_r"/* ) ] && rm -rf "$_r"
                                   else
                                           rm -f "$_r"
                                   fi
                           done
                   fi
   
                   # TODO: Should we mark these so real fw_update can -Drepair?
                   ftp -D "Install" -Vmo- "file:$_tmpsrc/$_f" |
                           tar -s ",^\+,${_pkgdir}/${_f%.tgz}/+," \
                           -s ",^firmware,${DESTDIR}/etc/firmware," \
                           -C / -zxphf - "+*" "firmware/*"
   
                   ed -s "${_pkgdir}/${_f%.tgz}/+CONTENTS" <<EOL
 /^@comment pkgpath/ -1a  /^@comment pkgpath/ -1a
 @option manual-installation  @option manual-installation
 @option firmware  @option firmware
Line 123 
Line 236 
 .  .
 w  w
 EOL  EOL
 done          done
   }
   
   fw_update "$FWURL"
Legend:
Removed from v.1.4  
changed lines
  Added in v.1.33
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>