From 7489a761bbd3f5d471c82dafe2c07e6420de6ba7 Mon Sep 17 00:00:00 2001
From: Eric Biggers <ebiggers3@gmail.com>
Date: Thu, 30 Aug 2012 14:25:37 -0500
Subject: [PATCH] test-imagex-ntfs (IN PROGRESS)

---
 Makefile.am            |   2 +-
 tests/test-imagex-ntfs | 126 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 127 insertions(+), 1 deletion(-)
 create mode 100755 tests/test-imagex-ntfs

diff --git a/Makefile.am b/Makefile.am
index e7cf2b55..6ba00106 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -122,6 +122,6 @@ man1_MANS =			\
 
 $(man1_MANS): config.status
 
-dist_check_SCRIPTS = tests/test-imagex
+dist_check_SCRIPTS = tests/test-imagex tests/test-imagex-ntfs
 TESTS = $(dist_check_SCRIPTS)
 
diff --git a/tests/test-imagex-ntfs b/tests/test-imagex-ntfs
new file mode 100755
index 00000000..258a8cab
--- /dev/null
+++ b/tests/test-imagex-ntfs
@@ -0,0 +1,126 @@
+#!/bin/bash
+
+# This script does some sanity testing of the 'imagex' program, specifically
+# checking the NTFS capture and apply features.
+#
+# This test will fail if wimlib was compiled with --without-ntfs-3g.
+
+# Assume an in-tree build.
+set -e
+cd tests
+
+imagex() {
+	echo "imagex $@"
+	../imagex $@ > /dev/null
+}
+
+do_unmount() {
+	if mountpoint $1 &> /dev/null; then
+		if !  fusermount -u $1; then
+			error "Failed to unmount \"$1\""
+		fi
+	fi
+}
+
+do_mkntfs() {
+	if ! mkntfs --force $1 &> /dev/null; then
+		error "Could not create NTFS volume on \"$1\".  Make sure ntfs-3g / ntfsprogs are installed"
+	fi
+}
+
+init() {
+	echo "Creating NTFS volumes and empty directories to use as mountpoints"
+	dd if=/dev/zero of=in.ntfs bs=4096 count=1000 &> /dev/null
+	dd if=/dev/zero of=out.ntfs bs=4096 count=1000 &> /dev/null
+	mkdir in.mnt out.mnt
+	do_mkntfs in.ntfs
+	do_mkntfs out.ntfs
+}
+
+cleanup() {
+	do_unmount in.mnt
+	do_unmount out.mnt
+	rm -rf in.ntfs out.ntfs in.mnt out.mnt in.xattr out.xattr
+}
+#trap cleanup exit
+
+
+error() {
+	echo "****************************************************************"
+	echo "                         Test failure                           "
+	echo $*
+	echo "****************************************************************"
+	exit 1
+}
+
+do_capture() {
+	if ! imagex capture in.ntfs ntfs.wim; then
+		error "Failed to capture NTFS volume into a WIM"
+	fi
+}
+
+do_apply() {
+	do_unmount out.ntfs
+	do_mkntfs out.ntfs
+	if ! imagex apply ntfs.wim 1 out.ntfs; then
+		error "Failed to apply WIM to NTFS volume"
+	fi
+	echo "do_apply"
+}
+
+cmp_xattrs() {
+	infile=$1
+	outfile=$2
+	xattr=$3
+	#echo "Comparing xattr $xattr of $infile and $outfile"
+	if test "$xattr" = "system.ntfs_times"; then
+		headnum=24
+	else
+		headnum=1000000000
+	fi
+	if eval getfattr --only-values -d -n $xattr $infile 2>/dev/null\
+					| head -c $headnum > in.xattr; then
+		if eval getfattr --only-values -d -n $xattr $outfile 2>/dev/null\
+					| head -c $headnum > out.xattr; then
+			if ! cmp in.xattr out.xattr; then
+				error "Extended attribute $xattr of $infile and $outfile differs"
+			fi
+		else
+			error "$infile has extended attribute $xattr, but $outfile doesn't"
+		fi
+	else
+		if eval getfattr --only-values -d -n $xattr $outfile 2>/dev/null\
+					| head -c $headnum > out.xattr; then
+			error "$outfile has extended attribute $xattr, but $infile doesn't"
+		fi
+	fi
+}
+
+do_capture_and_apply() {
+	do_unmount in.mnt
+	do_unmount out.mnt
+	do_capture
+	if ! ntfs-3g -o ro in.ntfs in.mnt || ! ntfs-3g -o ro out.ntfs out.mnt; then
+		error "Could not mount NTFS volume.  Make sure ntfs-3g is installed"
+	fi
+	if ! diff -r in.mnt out.mnt; then
+		error "Recursive diff of original NTFS volume with applied NTFS volume failed"
+	fi
+	for infile in `find in.mnt`; do
+		outfile=out.mnt${infile##in.mnt}
+		echo "Comparing xattrs of $infile and $outfile"
+		cmp_xattrs $infile $outfile system.ntfs_attrib
+		cmp_xattrs $infile $outfile system.ntfs_reparse_data
+		cmp_xattrs $infile $outfile system.ntfs_acl
+		cmp_xattrs $infile $outfile system.ntfs_dos_name
+		cmp_xattrs $infile $outfile system.ntfs_times
+	done
+}
+
+cleanup
+init
+
+echo "Testing capture and apply of empty NTFS volume"
+do_capture_and_apply
+
+
-- 
2.43.0