summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBjörn Baumbach <bb@sernet.de>2011-11-02 20:05:24 +0100
committerMichael Adam <obnox@samba.org>2011-11-04 00:28:08 +0100
commit2f9b453635215774d2bf6a47db5ba26642b25b8e (patch)
tree9d894f5f06e6d9d1b07ea72c49ceab5fcd805c72
parent767c54d8dd9596718579699398392ae234b40aa2 (diff)
downloadsamba-2f9b453635215774d2bf6a47db5ba26642b25b8e.tar.gz
samba-2f9b453635215774d2bf6a47db5ba26642b25b8e.tar.bz2
samba-2f9b453635215774d2bf6a47db5ba26642b25b8e.zip
s3-selftest: introduce registry upgrade test
Signed-off-by: Michael Adam <obnox@samba.org>
-rwxr-xr-xsource3/script/tests/test_registry_upgrade.sh176
1 files changed, 176 insertions, 0 deletions
diff --git a/source3/script/tests/test_registry_upgrade.sh b/source3/script/tests/test_registry_upgrade.sh
new file mode 100755
index 0000000000..b860ef4c35
--- /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
+