From 2f9b453635215774d2bf6a47db5ba26642b25b8e Mon Sep 17 00:00:00 2001
From: =?utf8?q?Bj=C3=B6rn=20Baumbach?= <bb@sernet.de>
Date: Wed, 2 Nov 2011 20:05:24 +0100
Subject: [PATCH] s3-selftest: introduce registry upgrade test

Signed-off-by: Michael Adam <obnox@samba.org>
---
 source3/script/tests/test_registry_upgrade.sh | 176 ++++++++++++++++++++++++++
 1 file changed, 176 insertions(+)
 create mode 100755 source3/script/tests/test_registry_upgrade.sh

diff --git a/source3/script/tests/test_registry_upgrade.sh b/source3/script/tests/test_registry_upgrade.sh
new file mode 100755
index 00000000000..b860ef4c35b
--- /dev/null
+++ b/source3/script/tests/test_registry_upgrade.sh
@@ -0,0 +1,176 @@
+#!/bin/sh
+#
+# Test for registry upgrades.
+#
+# Copyright (C) 2011 Björn Baumbach <bb@sernet.de>
+
+if [ $# -lt 2 ]; then
+    echo "Usage: test_registry_upgrade.sh NET DBWRAP_TOOL"
+    exit 1
+fi
+
+SCRIPT_DIR=$(dirname $0)
+BASE_DIR="${SCRIPT_DIR}/../../.."
+
+NET="$1"
+DBWRAP_TOOL="$2"
+DATADIR="${BASE_DIR}/testdata/samba3"
+WORKSPACE="${PREFIX}/registry_upgrade"
+CONFIG_FILE="${WORKSPACE}/smb.conf"
+CONFIGURATION="--configfile=${CONFIG_FILE}"
+
+NETCMD="$NET $CONFIGURATION"
+
+incdir="${BASE_DIR}/testprogs/blackbox"
+. $incdir/subunit.sh
+
+failed=0
+
+REGPATH="HKLM\Software\Samba"
+
+LOGDIR_PREFIX="registry_upgrade"
+
+registry_check()
+{
+    local CHECKNO="$1"
+    local CHECKDIFF="$2"
+    local REGVER=""
+    local ALLOWEDERR="INFO: version =|Check database:|overwrite registry format version 0 with 1|no INFO/version found"
+
+    test "x$CHECKNO" = "x0" && {
+	REGVER="--reg-version=1"
+    }
+
+    echo "Registry check $CHECKNO" >> $LOG
+    CHECK="$($NETCMD registry check $REGVER 2>&1)"
+    RC=$?
+    ERRORSTR="$(echo "$CHECK" | grep -vE $ALLOWEDERR )"
+
+    test "x$RC" = "x0" || {
+        echo "upgrade check $CHECKNO failed:" >> $LOG
+        return 1
+    }
+
+    test "x$ERRORSTR" = "x" || {
+        echo "upgrade check $CHECKNO failed:" >> $LOG
+	echo "reason: $CHECK" >> $LOG
+	return 1
+    }
+
+    test "x$CHECKDIFF" = "xcheckdiff" && {
+        $NETCMD registry export 'HKLM' $WORKSPACE/export_${CHECKNO}.reg >> $LOG 2>&1
+        diff $WORKSPACE/export_0.reg $WORKSPACE/export_${CHECKNO}.reg >> $LOG 2>&1
+        test "x$?" = "x0" || {
+            echo "Error: $WORKSPACE/export_0.reg differs from $WORKSPACE/export_${CHECKNO}.reg" >> $LOG
+            return 1
+        }
+    }
+
+    return 0
+}
+
+registry_upgrade()
+{
+    local DIR=$(mktemp -d ${PREFIX}/${LOGDIR_PREFIX}_XXXXXX)
+    local LOG=$DIR/log
+
+    echo registry_upgrade $1 > $LOG
+
+    mkdir $WORKSPACE
+    cp -v $DATADIR/registry.tdb $WORKSPACE/registry.tdb >> $LOG 2>&1
+
+    REGISTRY="${WORKSPACE}/registry.tdb"
+
+    test -e $REGISTRY || {
+        echo "Error: Database file not available" >> $LOG
+        return 1
+    }
+
+    # create config file
+    echo '[global]' > ${CONFIG_FILE}
+    echo "	state directory = ${WORKSPACE}" >> ${CONFIG_FILE}
+
+    # set database INFO/version to 1
+    #$DBWRAP_TOOL $REGISTRY store 'INFO/version' uint32 1
+    #test "x$?" = "x0" || {
+    #    echo "Error: Can not set INFO/version" >> $LOG
+    #    return 1
+    #}
+
+    # check original registry.tdb
+    echo "$REGISTRY" >> $LOG
+    registry_check 0
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # trigger upgrade
+    $NETCMD registry enumerate $REGPATH >> $LOG 2>&1
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # check upgraded database
+    registry_check 1
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # export database for diffs
+    $NETCMD registry export 'HKLM' $WORKSPACE/export_0.reg >> $LOG 2>&1
+
+    # remove version string
+    $DBWRAP_TOOL $REGISTRY delete INFO/version >> $LOG 2>&1
+    test "x$?" = "x0" || {
+        echo "Error: Can not remove INFO/version key from registry" >> $LOG
+        return 1
+    }
+
+    # trigger upgrade on upgraded database
+    $NETCMD registry enumerate $REGPATH >> $LOG 2>&1
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # check upgraded database again
+    registry_check 2 checkdiff
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # set database INFO/version to version 2
+    $DBWRAP_TOOL $REGISTRY store 'INFO/version' uint32 2
+    test "x$?" = "x0" || {
+        echo "Error: Can not set INFO/version" >> $LOG
+        return 1
+    }
+
+    # trigger upgrade
+    $NETCMD registry enumerate $REGPATH >> $LOG 2>&1
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # check upgraded database again
+    registry_check 3 checkdiff
+    test "x$?" = "x0" || {
+        return 1
+    }
+
+    # remove workspace
+    rm -r $DIR
+    rm -r $WORKSPACE
+}
+
+# remove old logs
+for OLDDIR in $(find ${PREFIX} -type d -name "${LOGDIR_PREFIX}_*") ; do
+	echo "removing old directory ${OLDDIR}"
+	rm -rf ${OLDDIR}
+done
+# remove old workspace
+rm -rf $WORKSPACE
+
+testit "registry_upgrade" registry_upgrade || failed=`expr $failed + 1`
+
+testok $0 $failed
+
-- 
2.11.4.GIT