===================================================================
RCS file: /cvs/openbsd/update_openbsd/update_openbsd,v
retrieving revision 1.11
retrieving revision 1.17
diff -u -r1.11 -r1.17
--- openbsd/update_openbsd/update_openbsd	2008/11/09 07:22:52	1.11
+++ openbsd/update_openbsd/update_openbsd	2009/01/14 17:58:40	1.17
@@ -1,22 +1,6 @@
 #!/bin/sh
-# $RedRiver: update_openbsd,v 1.10 2008/11/05 17:07:58 andrew Exp $
+# $RedRiver: update_openbsd,v 1.16 2009/01/13 21:47:49 andrew Exp $
 
-if [ -e /etc/update_openbsd.conf ]; then
-    . /etc/update_openbsd.conf
-fi
-
-if [ -e ${HOME}/.update_openbsdrc ]; then
-    . ${HOME}/.update_openbsdrc
-fi
-
-MIRROR=${MIRROR:=ftp://ftp.openbsd.org/pub/OpenBSD}
-FTP_CMD=${FTP_CMD:=ftp -V}
-PKG_PATH=${PKG_PATH:=/usr/ports/packages/`machine`/all/:${MIRROR}/`uname -r`/packages/`machine`/}
-
-DESTDIR=${DESTDIR:=/}
-SYSMERGE=${SYSMERGE:=/usr/sbin/sysmerge}
-FORCE_DIR=${FORCE_DIR:=No}
-
 installed_sets() {
     local misc=/usr/share/doc/README
     local man=/usr/share/man/cat1/intro.0
@@ -40,19 +24,36 @@
         fi
     done
 }
-INSTALLED_SETS=${INSTALLED_SETS:=`installed_sets`}
 
-sets_exist () {
+version_in() {
         local _proto=${FTP%%://*}
+        local _file
 
         if [ X"ftp" == X"${_proto}" ]; then
-            echo "ls base*.tgz" | ${FTP_CMD} ${FTP}/ | awk '{ print $9 }'
+            local _list=`echo "ls base*.tgz" | ${FTP_CMD} ${FTP}/`
+            _file=`echo ${_list} | awk '/base[0-9][0-9].tgz/ { print $9 }'`
+
         elif [ X"http" == X"${_proto}" ]; then
-            ${FTP_CMD} -V -o - ${FTP}/ | grep '"base[0-9][0-9].tgz' | sed -e 's/.*\(base..\.tgz\).*/\1/'
+            local _list=`${FTP_CMD} -V -o - ${FTP}/`
+            _file=`echo ${_list} | awk '/[^x]base[0-9][0-9]*\.tgz/ { 
+                    sub("^.*base","base");
+                    sub("\.tgz.*",".tgz"); 
+                    print $0;
+                }'`
+
+        elif [ X"scp" == X"${_proto}" ]; then
+            echo SCP is not yet supported >&2
+            return 2
+
         else 
             echo Unsupported FTP ${FTP} >&2
-            exit 2
+            return 2
+
         fi
+
+        local _v=${_file##*base}
+        _v=${_v%.tgz*}
+        echo $_v
 }
 
 set_version() {
@@ -89,7 +90,7 @@
 
     fi
 
-    if [ X"" == X"${_v}" ]; then
+    if [ X"" != X"${MIRROR}" -a X"" == X"${_v}" ]; then
         if [ X"No" == X"${FORCE_DIR}" ]; then
             _dir=${NEW_VER}
         else
@@ -97,38 +98,31 @@
         fi
         FTP=${MIRROR}/${_dir}/`machine`
 
-        _file=`sets_exist`
+        _v=`version_in`
 
-        if [ X"" == X"${_file}" ]; then
+        if [ X"" == X"${_v}" ]; then
             if [ X"No" != X"$FORCE_DIR" ]; then
                 echo No sets in forced [${FTP}] >&2
-                exit 2
+                return 2
             fi
 
             NEW_VER=$CUR_VER
             _dir=${NEW_VER}
             FTP=${MIRROR}/${_dir}/`machine`
 
-            _file=`sets_exist`
-            if [ X"" == X"${_file}" ]; then
-                echo No sets in [${FTP}] >&2
-                exit 2
-            fi
+            _v=`version_in`
         fi
 
-        _v=${_file##*base}
-        _v=${_v%.tgz*}
-
         if [ X"" == X"${_v}" ]; then
-            echo No version in file [$_file] >&2
-            exit 2 
+            echo No sets in [${FTP}] >&2
+            return 2 
         elif [ X"${_cv}" == X"${_v}" ]; then
             NEW_VER=$CUR_VER
         elif [ X"${_nv}" == X"${_v}" ]; then
             NEW_VER=$NEW_VER
         else
             echo Invalid version [$_v] >&2
-            exit 2
+            return 2
         fi
 
         if [ X"No" == X"$FORCE_DIR" ]; then
@@ -139,7 +133,7 @@
 
     if [ X"" == X"${_v}" ]; then
         echo ERROR: Unable to determine file version! >&2
-        exit 1
+        return 1
     fi
 
     if [ X"" == X"$RELEASEDIR" ]; then
@@ -147,14 +141,16 @@
     fi
 
     FILE_VER=$_v
-    FTP=${MIRROR}/${_dir}/`machine`
+    if [ X"" != X"${MIRROR}" ]; then
+        FTP=${MIRROR}/${_dir}/`machine`
+    fi
 }
 
 get_sets() {
     echo '### GETTING SETS ###'
     if [ X"" == X"$FTP" ]; then
         echo ERROR: No FTP site set! >&2
-        exit 1
+        return 1
     fi
 
     mkdir -p ${RELEASEDIR}
@@ -180,16 +176,53 @@
         echo $FTP_CMD ${FTP}/MD5
         $FTP_CMD ${FTP}/MD5
     fi
+}
 
+check_md5() {
+    echo '### CHECKING MD5 ###'
+    cd $RELEASEDIR
+
+    if [ ! -e MD5 ]; then
+        echo MD5 File does not exist!
+        return 1
+    fi
+
     ls bsd* *.tgz | sed -e 's/\(.*\)/(\1)/' > index 
     grep -f index MD5 | md5 -c
 
     if [ $? -ne 0 ]; then
         echo ERROR: MD5 does not match! >&2
-        exit 1
+        return 1
     fi
 }
 
+check_sets() {
+    echo '### CHECKING SETS ###'
+    cd $RELEASEDIR
+
+    local _v=$FILE_VER
+
+    for _b in `echo /bsd* bsd bsd.mp bsd.rd | sort -u`; do
+	_b=${_b#/}
+        local _n=$_b
+        if [ X"bsd.sp" == X"${_n}" ]; then
+            _n=bsd
+        fi
+        if [ -e /${_b} -a ! -e ./${_n} ]; then
+            echo ${_n} does not exist
+        fi
+    done
+
+    for _s in $INSTALLED_SETS; do
+        if [ ! -e ./${_s}${_v}.tgz ]; then
+            echo ${_s}${_v}.tgz does not exist
+        fi
+    done
+
+    check_md5
+}
+
+
 install_kernel() {
     echo '### INSTALLING KERNEL ###'
     if [ X"$USER" != X"root" -a X"$SUDO" == X"" ]; then
@@ -221,8 +254,9 @@
 
     local _v=$FILE_VER
     if [ -e base${_v}.tgz ]; then
-        echo Extracting new firmware from base${_v}.tgz
-        $SUDO tar -C $DESTDIR -xzphf base${_v}.tgz "*etc/firmware/*"
+        echo Extracting new firmware and ifconfig from base${_v}.tgz
+        $SUDO tar -C $DESTDIR -xzphf base${_v}.tgz \
+		./etc/firmware ./sbin/ifconfig
     fi
 
     $SUDO rm -f /obsd
@@ -358,7 +392,26 @@
     fi
 }
 
+if [ -e /etc/update_openbsd.conf ]; then
+    . /etc/update_openbsd.conf
+fi
+
+if [ -e ${HOME}/.update_openbsdrc ]; then
+    . ${HOME}/.update_openbsdrc
+fi
+
+#MIRROR=${MIRROR:=ftp://ftp.openbsd.org/pub/OpenBSD}
+FTP_CMD=${FTP_CMD:=ftp -V}
+PKG_PATH=${PKG_PATH:=/usr/ports/packages/`machine`/all/:${MIRROR}/`uname -r`/packages/`machine`/}
+
+DESTDIR=${DESTDIR:=/}
+SYSMERGE=${SYSMERGE:=/usr/sbin/sysmerge}
+FORCE_DIR=${FORCE_DIR:=No}
+
+INSTALLED_SETS=${INSTALLED_SETS:=`installed_sets`}
+
 set_version
+local _error=$?
 
 echo
 echo "-= update_openbsd - helper script to update OpenBSD =-"
@@ -376,7 +429,15 @@
 #echo "      FILE_VER: $FILE_VER"
 echo
 
-get_sets
+if [ ${_error} -ne 0 ]; then
+	exit ${_error}
+fi
+
+if [ X"" != X"${FTP}" ]; then
+    get_sets || exit
+fi
+
+check_sets || exit
 
 if [ X"" != X"$SUDO" ]; then
     echo Please enter your sudo password if prompted.