From 49f3bbadcebb4eb3436341ca00727a183d100a35 Mon Sep 17 00:00:00 2001
From: Ben Hutchings <benh@debian.org>
Date: Tue, 22 Apr 2025 02:29:40 +0200
Subject: [PATCH] test: Add autopkgtest case for unmkinitramfs

Add a test that exercises unmkinitramfs with:

- 0-2 early cpio archives
- 0-2 main cpio archives
- All supported compressors, including none, for the last main archive

and verifies that the output is as expected.

Signed-off-by: Ben Hutchings <benh@debian.org>
---
 debian/tests/control       |   4 ++
 debian/tests/unmkinitramfs | 127 +++++++++++++++++++++++++++++++++++++
 2 files changed, 131 insertions(+)
 create mode 100755 debian/tests/unmkinitramfs

diff --git a/debian/tests/control b/debian/tests/control
index c69689e..ab81f98 100644
--- a/debian/tests/control
+++ b/debian/tests/control
@@ -2,6 +2,10 @@ Tests: copy-file
 Restrictions: needs-root, breaks-testbed, skip-not-installable, superficial, allow-stderr
 Depends: linux-image-generic, zstd, @
 
+Tests: unmkinitramfs
+Restrictions: superficial
+Depends: bzip2, xz-utils, lzop, lz4, zstd, @
+
 Tests: qemu-klibc
 Architecture: amd64 armhf s390x
 Restrictions: needs-root, breaks-testbed
diff --git a/debian/tests/unmkinitramfs b/debian/tests/unmkinitramfs
new file mode 100755
index 0000000..af042af
--- /dev/null
+++ b/debian/tests/unmkinitramfs
@@ -0,0 +1,127 @@
+#!/bin/sh
+set -eu
+
+make_one_archive() {
+	local type="$1"
+	local i="$2"
+	local dir_name
+	local j
+
+	case "$type" in
+	early)
+		dir_name=kernel
+		;;
+	main)
+		dir_name="dir$i"
+		;;
+	*)
+		echo >&2 "E: Bad archive type $type"
+		exit 1
+		;;
+	esac
+
+	mkdir -p "$AUTOPKGTEST_TMP/input/$type$i/$dir_name"
+	for j in $(seq 0 9); do
+		size=$((j * 987 + i * 654 + 321))
+		dd if=/dev/urandom of="$AUTOPKGTEST_TMP/input/$type$i/$dir_name/file$j" \
+		   bs="$size" count=1 status=none
+	done
+	(
+		cd "$AUTOPKGTEST_TMP/input/$type$i"
+		find . | cpio -H newc -o --quiet > ../"$type$i.cpio"
+	)
+}
+
+# Create input files and archives
+mkdir "$AUTOPKGTEST_TMP/input"
+for in_type in early main; do
+	for i in 0 1 2; do
+		make_one_archive "$in_type" "$i"
+	done
+done
+
+# Dummy compressor; wrapper for cat that ignores its argument (-c)
+no_compressor() {
+	# shellcheck disable=SC2317
+	cat
+}
+
+rc=0
+
+# Test up to 2 early and 2 main cpio archives, and all supported
+# compressors (including none) for the last main archive
+for n_early in 0 1 2; do
+	for n_main in 0 1 2; do
+		# There must be at least 1 archive
+		if [ $((n_early + n_main)) -eq 0 ]; then
+			continue
+		fi
+
+		for compressor in no_compressor gzip bzip2 xz lzop 'lz4 -l' zstd; do
+			# If last archive is early, it can't be compressed
+			if [ $n_main -eq 0 ] && [ "$compressor" != no_compressor ]; then
+				continue
+			fi
+
+			echo "I: Testing $n_early early + $n_main main archive(s) with $compressor"
+
+			# Construct initramfs image
+			: > "$AUTOPKGTEST_TMP/initrd.img"
+			for i in $(seq 0 $((n_early - 1))); do
+				cat "$AUTOPKGTEST_TMP/input/early$i.cpio" \
+				    >> "$AUTOPKGTEST_TMP/initrd.img"
+			done
+			for i in $(seq 0 $((n_main - 2))); do
+				cat "$AUTOPKGTEST_TMP/input/main$i.cpio" \
+				    >> "$AUTOPKGTEST_TMP/initrd.img"
+			done
+			if [ $n_main -ge 1 ]; then
+				$compressor -c < "$AUTOPKGTEST_TMP/input/main$((n_main - 1)).cpio" \
+					    >> "$AUTOPKGTEST_TMP/initrd.img"
+			fi
+
+			# Unpack it
+			rm -rf "$AUTOPKGTEST_TMP/output"
+			unmkinitramfs "$AUTOPKGTEST_TMP/initrd.img" \
+				      "$AUTOPKGTEST_TMP/output" \
+				|| {
+				echo >&2 'E: unmkinitramfs failed'
+				rc=1
+				continue
+			}
+
+			# Construct what we expect output to look like
+			rm -rf "$AUTOPKGTEST_TMP/reference"
+			mkdir "$AUTOPKGTEST_TMP/reference"
+			for i in $(seq 0 $((n_early - 1))); do
+				if [ "$i" -eq 0 ]; then
+					dir_name=early
+				else
+					dir_name=early$((i + 1))
+				fi
+				ln -s ../input/"early$i" \
+				   "$AUTOPKGTEST_TMP/reference/$dir_name"
+			done
+			for i in $(seq 0 $((n_main - 1))); do
+				if [ $n_early = 0 ]; then
+					ln -s ../input/"main$i/dir$i" \
+					   "$AUTOPKGTEST_TMP/reference/dir$i"
+				else
+					mkdir -p "$AUTOPKGTEST_TMP/reference/main"
+					ln -s ../../input/"main$i/dir$i" \
+					   "$AUTOPKGTEST_TMP/reference/main/dir$i"
+				fi
+			done
+
+			# Compare reference and output
+			diff -r "$AUTOPKGTEST_TMP/reference" \
+			     "$AUTOPKGTEST_TMP/output" \
+				|| {
+				echo >&2 'E: Output files do not match input'
+				rc=1
+			}
+		done
+	done
+done
+
+exit $rc