v1.14.4

[wimlib] / tests / test-imagex

#!/usr/bin/env bash

# This script does some sanity testing of the 'wimlib-imagex' program.  It by no
# means tests every aspect of wimlib comprehensively.

set -e
cd tests
srcdir="${srcdir:-.}/.."
srcdir="$(cd $srcdir; pwd)"
. "$srcdir/tests/test_utils.sh"

TEST_SUBDIR=tmpdir_test-imagex

# Execute the tests in a subdirectory to avoid conflicts with concurrent tests
default_cleanup
mkdir $TEST_SUBDIR
cd $TEST_SUBDIR

# Make test directory
mkdir dir
cp $srcdir/src/*.c dir
mkdir dir/subdir
echo 'hello' > dir/subdir/hello
echo 'hello' > dir/subdir/hello2
ln dir/subdir/hello dir/subdir/hellolink
echo -n > dir/subdir/empty_file
ln -s hello dir/subdir/rel_symlink

mkdir dir2
echo 'testing' > dir2/file
dd if=/dev/zero of=dir2/zeroes bs=4096 count=5

# Capturing and applying WIM with None, LZX, and XPRESS compression

for comp_type in None LZX XPRESS; do
        echo "Testing capture and application of $comp_type-compressed WIM"
        if ! wimcapture dir dir.wim --compress=$comp_type; then
                error "'wimcapture' failed"
        fi
        if ! wimapply dir.wim tmp; then
                error "'wimapply' failed"
        fi
        if ! test `wim_ctype dir.wim` = "$comp_type"; then
                error "'wiminfo' didn't report the compression type correctly"
        fi
        if ! diff -q -r dir tmp; then
                error "Recursive diff of extracted directory with original failed"
        fi
        if ! test `get_link_count tmp/subdir/hello` = 2; then
                error "Incorrect number of hard links in extracted file"
        fi
        if ! test `get_inode_number tmp/subdir/hello` != `get_inode_number tmp/subdir/hello2`; then
                error "Expected different inode numbers in files not hard-linked"
        fi
        if ! test "`get_inode_number tmp/subdir/hello`" = "`get_inode_number tmp/subdir/hellolink`"; then
                error "Expected same inode numbers in hard-linked files"
        fi
        if ! test -L tmp/subdir/rel_symlink; then
                error "Symlink not extracted correctly"
        fi
        if ! test "`readlink tmp/subdir/rel_symlink`" = "hello"; then
                error "Symlink target not correct"
        fi

        rm -rf dir.wim tmp
done

# Test wimverify and the SHA-1 code
WIMLIB_DISABLE_CPU_FEATURES='*' wimcapture dir dir.wim --compress=none
disabled=''
for cpu_feature in '' sha1 bmi2 avx sse4.2 sse4.1 ssse3; do
        [ -n "$disabled" ] && disabled+=','
        disabled+="$cpu_feature"
        if ! WIMLIB_DISABLE_CPU_FEATURES=$disabled wimverify dir.wim; then
                error "wimverify failed (cpu_features_disabled=$disabled)"
        fi
done

# Test wimappend --create
rm -f dir.wim
if wimappend dir dir.wim; then
        error "wimappend to nonexisting file unexpectedly succeeded"
fi
if ! wimappend dir dir.wim --create; then
        error "wimappend --create to nonexisting file failed"
fi
if ! wimappend dir dir.wim --create; then
        error "wimappend --create to existing file failed"
fi
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 2; then
        error "Incorrect WIM image count after wimappend --create"
fi

# Capturing and modifying name, description, and bootable flag

echo "Testing capture of WIM with default name and description"
wimcapture dir dir.wim
if ! test "`wiminfo dir.wim | grep Name | awk '{print $2}'`" = "dir"; then
        error "WIM name not set correctly"
fi
if ! test "`wiminfo dir.wim | grep Description | awk '{print $2}'`" = ""; then
        error "WIM description not set correctly"
fi

echo "Testing capture of WIM with default boot flag"
wimcapture dir dir.wim
if ! test "`wiminfo dir.wim | grep '^Boot Index' | awk '{print $3}'`" = "0"; then
        error "WIM boot flag not set correctly"
fi

echo "Testing changing image bootable flag"
if ! wiminfo dir.wim 1 --boot; then
        error "Failed to change bootable image"
fi
if ! test "`wiminfo dir.wim | grep '^Boot Index' | awk '{print $3}'`" = "1"; then
        error "Bootable image not changed correctly"
fi
echo "Testing changing image bootable flag"
if ! wiminfo dir.wim 0 --boot; then
        error "Failed to reset bootable image"
fi
if ! test "`wiminfo dir.wim | grep '^Boot Index' | awk '{print $3}'`" = "0"; then
        error "Bootable image not reset correctly"
fi
echo "Testing changing image bootable flag to invalid image (this should generate errors)"
if wiminfo dir.wim 2 --boot; then
        error "Succeeded in changing bootable image to invalid number"
fi
if ! test "`wiminfo dir.wim | grep '^Boot Index' | awk '{print $3}'`" = "0"; then
        error "Boot flag was changed even though the change command was supposed to fail"
fi
rm -rf dir.wim tmp

name_desc_test() {
        local name=$1
        local desc=$2
        if ! wimcapture dir dir.wim "$name" "$desc"; then
                error "Failed to capture WIM with specified name and description"
        fi
        if ! test "`wiminfo dir.wim | grep Name | awk '{print $2}'`" = "$name"; then
                error "WIM name not set correctly"
        fi
        if ! test "`wiminfo dir.wim | grep Description | awk '{print $2}'`" = "$desc"; then
                error "WIM description not set correctly"
        fi
}

echo "Testing capture of WIM with name and description"
name_desc_test "myname" "mydesc"

echo "Testing capture of WIM with non-ASCII name and description"
name_desc_test "áéíóú" "¿?"

echo "Testing printing WIM lookup table"
if ! wiminfo --lookup-table dir.wim > /dev/null; then
        error "Failed to print WIM lookup table"
fi
echo "Testing printing WIM header"
if ! wiminfo --header dir.wim > /dev/null; then
        error "Failed to print WIM header"
fi
echo "Testing printing WIM XML info"
if ! wiminfo --xml dir.wim > /dev/null; then
        error "Failed to print WIM XML data"
fi
echo "Testing extracting WIM XML info"
if ! wiminfo --extract-xml=dir.xml dir.wim; then
        error "Failed to extract WIM XML data"
fi
echo "Testing printing WIM metadata"
if ! wimdir --detailed dir.wim > /dev/null; then
        error "Failed to print WIM metadata"
fi
rm -rf dir.wim tmp dir.xml

echo "Testing capture of bootable WIM"
if ! wimcapture dir dir.wim --boot; then
        error "Failed to capture bootable WIM"
fi
if ! test "`wiminfo dir.wim | grep '^Boot Index' | awk '{print $3}'`" = "1"; then
        error "Boot flag on bootable WIM not set correctly"
fi
rm -rf dir.wim tmp

# Integrity table

echo "Testing capture of WIM with integrity table"
if ! wimcapture dir dir.wim --check; then
        error "Failed to capture WIM with integrity table"
fi
if ! wiminfo dir.wim | grep -q Integrity; then
        error "Integrity table on WIM not made"
fi
if ! wimapply --check dir.wim tmp; then
        error "Integrity table on WIM not made correctly"
fi
if ! diff -q -r dir tmp; then
        error "Recursive diff of applied WIM with original directory failed"
fi
rm -rf dir.wim tmp

# Appending and deleting images

echo "Testing appending WIM image"
wimcapture dir dir.wim
if ! wimappend dir2 dir.wim; then
        error "Appending WIM image failed"
fi
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 2; then
        error "WIM image count not correct"
fi

echo "Testing appending WIM image with existing name (this should generate errors)"
if wimappend dir2 dir.wim "dir"; then
        error "Adding duplicate image name didn't fail"
fi
echo "Testing appending WIM image with new name"
if ! wimappend dir2 dir.wim "newname"; then
        error "Appending WIM image failed"
fi
echo "Testing appending WIM image with integrity check"
if ! wimappend dir2 dir.wim "newname2" --check; then
        error "Appending WIM image failed"
fi
if ! wiminfo dir.wim | grep -q Integrity; then
        error "Integrity table not set correctly on image append"
fi
echo "Testing appending WIM image with no integrity check"
if ! wimappend dir2 dir.wim "newname3" --nocheck; then
        error "Appending WIM image failed"
fi
if wiminfo dir.wim | grep -q Integrity; then
        error "WIM integrity table not removed"
fi
# 5 images at this point
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 5; then
        error "WIM does not contain the expected 5 images"
fi
echo "Testing deleting first WIM image"
if ! wimdelete dir.wim 1; then
        error "Failed to delete WIM image"
fi
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 4; then
        error "WIM image not deleted correctly"
fi
echo "Testing deleting last WIM image"
if ! wimdelete dir.wim 4; then
        error "Failed to delete WIM image"
fi
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 3; then
        error "WIM image not deleted correctly"
fi
echo "Testing deleting invalid WIM image (this should generate errors)"
if wimdelete dir.wim 4; then
        error "Expected to fail to delete non-existent WIM image"
fi
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 3; then
        error "Image count changed even though we intentionally failed to delete an image"
fi
echo "Testing deleting all WIM images"
if ! wimdelete dir.wim all; then
        error "Failed to delete all images from WIM"
fi
if ! test "`wiminfo dir.wim | grep 'Image Count' | awk '{print $3}'`" = 0; then
        error "Couldn't delete all WIM images correctly"
fi
echo "Testing appending directory to empty WIM and making it bootable"
if ! wimappend dir dir.wim "myname" "mydesc" --check --boot; then
        error "Couldn't append named, described, bootable image to empty WIM with integrity check"
fi
if ! wiminfo dir.wim | grep -q Integrity; then
        error "Integrity check not found"
fi
if ! test "`wiminfo dir.wim | grep '^Boot Index' | awk '{print $3}'`" = "1"; then
        error "Bootable image not set correctly"
fi
echo "Testing appending non-directory (should generate errors)"
if wimappend dir.wim dir.wim; then
        error "Incorrectly succeeded to append non-directory to WIM"
fi
echo "Testing appending non-existent file (should generate errors)"
if wimappend SOME_NONEXISTENT_FILE dir.wim; then
        error "Incorrectly succeeded to append non-existent file to WIM"
fi
if [ `id -u` != 0 ]; then
        echo "Testing appending directory containing unreadable file (should generate errors)"
        mkdir -p dir3
        echo 1 > dir3/file
        chmod -r dir3/file
        if wimappend dir3 dir.wim; then
                error "Incorrectly succeeded in capturing directory with unreadable file"
        fi
fi
rm -rf dir3 dir.wim

# Applying multiple images, applying with hardlinks/symlinks

echo "Testing application of multiple images"
if ! wimcapture dir dir.wim; then
        error "Failed to prepare test WIM"
fi
if ! wimappend dir dir.wim "myname"; then
        error "Failed to append image to test WIM"
fi
if ! wimapply dir.wim all tmp; then
        error "Applying multiple images failed"
fi
if ! diff -q -r tmp/dir tmp/myname || ! diff -q -r dir tmp/dir; then
        error "Recursive diff of applied WIM with original directory failed"
fi
if test "`get_link_count tmp/dir/write.c`" != 1; then
        error "Incorrect link count on extracted file"
fi
if test "`get_link_count tmp/myname/write.c`" != 1; then
        error "Incorrect link count on extracted file"
fi
if test "`get_inode_number tmp/myname/write.c`" = "`get_inode_number tmp/dir/write.c`"; then
        error "Incorrect inode number"
fi
rm -rf tmp

echo "Testing application of single image containing identical files"
if ! wimapply dir.wim 1 tmp; then
        error "Failed to apply WIM"
fi
if test "`get_link_count tmp/subdir/hello`" != 2; then
        error "Incorrect link count on extracted file"
fi
if test "`get_link_count tmp/subdir/hello2`" != 1; then
        error "Incorrect link count on extracted file"
fi
if test "`get_inode_number tmp/subdir/hello`" = "`get_inode_number tmp/subdir/hello2`"; then
        error "Inode numbers on non-hard-linked files are the same"
fi
if test "`get_inode_number tmp/subdir/hello`" != "`get_inode_number tmp/subdir/hellolink`"; then
        error "Inode numbers on hard-linked files are different"
fi
rm -rf tmp

# wimsplit, wimjoin

echo "Creating random files to test WIM splitting on"
mkdir tmp
for ((i = 0; i < 100; i++)); do
        dd if=/dev/urandom of=tmp/file$i bs=4096 count=10 &> /dev/null
done
for flag in "--compress=none" "--compress=maximum" "--compress=fast"; do
        echo "Using flag $flag"
        if ! wimcapture tmp tmp.wim $flag; then
                error "Failed to capture test WIM"
        fi
        echo "Splitting WIM into 1 MiB chunks"
        if ! wimsplit tmp.wim tmp.swm 1; then
                error "Failed to split WIM"
        fi
        echo "Verifying the split WIMs (some errors expected)"
        if test "`wiminfo tmp.swm | grep 'Part Number' | awk '{print $3}'`" != "1/4"; then
                error "Part number of split WIM not correct"
        fi
        if ! wimdir tmp.swm > /dev/null; then
                error "Failed to list files in split WIM"
        fi
        if ! test -e tmp2.swm; then
                error "Could not find split-WIM part 2"
        fi
        if wimdir tmp2.swm > /dev/null; then
                error "Listed files in part 2 of split WIM (this should have failed)"
        fi

        # Unsupported, should fail
        if wiminfo tmp.swm --boot 1; then
                error "Should not have been able to change boot index of split WIM"
        fi
        echo "Joining the split WIMs and applying the result"
        if ! wimjoin tmp2.wim tmp*.wim; then
                error "Failed to join split WIMs"
        fi
        if ! wimapply tmp2.wim tmp2; then
                error "Failed to apply joined split WIM"
        fi
        if ! wimapply tmp.wim tmp3; then
                error "Failed to apply test WIM"
        fi
        if ! diff -q -r tmp tmp2 || ! diff -q -r tmp tmp3; then
                error "Recursive diff of applied joined split WIM with original directory failed"
        fi
        rm -f *.wim *.swm
        rm -rf tmp2 tmp3
done
rm -rf tmp

# wimexport
echo "Testing export of single image to new WIM"
if ! wimcapture dir dir.wim; then
        error "Failed to capture test WIM"
fi
if ! wimappend dir2 dir.wim; then
        error "Failed to append image to test WIM"
fi
if ! wimexport dir.wim dir new.wim; then
        error "Failed to export single image to new WIM"
fi
if test "`wiminfo new.wim | grep 'Image Count' | awk '{print $3}'`" != 1; then
        error "Exporting single image to new WIM wasn't done correctly"
fi
echo "Testing export of single image to existing WIM"
if ! wimexport dir.wim dir2 new.wim; then
        error "Failed to export single image to existing WIM"
fi
if test "`wiminfo new.wim | grep 'Image Count' | awk '{print $3}'`" != 2; then
        error "Exporting single image to existing WIM wasn't done correctly"
fi
echo "Testing export of single image to existing WIM using wrong compression type"
if wimexport dir.wim dir2 new.wim newname --compress=xpress; then
        error "Successfully exported image using wrong compression type"
fi
rm -f new.wim
echo "Testing export of multiple images to new WIM"
if ! wimexport dir.wim all new.wim; then
        error "Failed to export multiple images to new WIM"
fi
if test "`wiminfo new.wim | grep 'Image Count' | awk '{print $3}'`" != 2; then
        error "Exporting multiple images to new WIM wasn't done correctly"
fi
if ! wimcapture dir2 new.wim newname; then
        error "Failed to capture test WIM"
fi
echo "Testing export of multiple images to existing WIM"
if ! wimexport dir.wim all new.wim; then
        error "Failed to export multiple images to existing WIM"
fi
echo "Testing export of multiple images to existing WIM with --boot"
if ! wimcapture dir2 new.wim newname; then
        error "Failed to capture test WIM"
fi
if ! wiminfo dir.wim --boot 1; then
        error "Failed to set boot index on test WIM"
fi
if ! wimexport dir.wim all new.wim --boot; then
        error "Failed to export multiple images to existing WIM with bootable image"
fi
echo "Testing export of multiple images to existing WIM with --boot, but no bootable image (errors expected)"
if ! wimcapture dir2 new.wim newname; then
        error "Failed to capture test WIM"
fi
if ! wiminfo dir.wim --boot 0; then
        error "Failed to clear boot index on test WIM"
fi
if wimexport dir.wim all new.wim --boot; then
        error "Successfully exported multiple images with --boot but with no bootable images"
fi

# Test exporting an image to another WIM, then applying it.
# We try with 5 different combinations of compression types to make sure we go
# through all paths in the resource-handling code.
for i in 1 2 3 4 5; do
        case $i in
        1)
                cflag1="--compress=none";
                cflag2="--compress=none";
                ;;
        2)
                cflag1="--compress=xpress";
                cflag2="--compress=xpress";
                ;;
        3)
                cflag1="--compress=xpress"
                cflag2="--compress=lzx"
                ;;
        4)
                cflag1="--compress=none"
                cflag2="--compress=xpress"
                ;;
        5)
                cflag1="--compress=xpress"
                cflag2="--compress=none"
                ;;
        esac
        echo "Testing exporting then applying an image (\"$cflag1\" => \"$cflag2\")"
        rm -rf dir.wim new.wim tmp tmp2
        wimcapture dir dir.wim $cflag1
        wimcapture dir2 dir2.wim $cflag2
        wimexport dir.wim dir dir2.wim
        wimapply dir.wim dir tmp
        if ! wimapply dir2.wim dir tmp2; then
                error "Failed to apply image that was exported to a WIM"
        fi
        if ! diff -r tmp tmp2; then
                error "Image that was exported to a WIM was not applied correctly"
        fi
done

echo "**********************************************************"
echo "             Basic wimlib-imagex tests passed             "
echo "**********************************************************"

# Leave test subdirectory and cleanup
cd ..
default_cleanup