From 2df6ec519d3348a23bff32c7f337bda72d443ba2 Mon Sep 17 00:00:00 2001
From: Radostin Stoyanov <rstoyanov@fedoraproject.org>
Date: Thu, 23 Mar 2023 20:21:04 +0000
Subject: [PATCH] test/other: add test for action-script

This commit is introducing a test for the action-script functionality
of CRIU to verify that pre-dump, post-dump, pre-restore, pre-resume,
post-restore, post-resume hooks are executed during dump/restore.

Signed-off-by: Radostin Stoyanov <rstoyanov@fedoraproject.org>
---
 Makefile                                   |  1 +
 scripts/ci/run-ci-tests.sh                 |  3 ++
 test/others/action-script/.gitignore       |  1 +
 test/others/action-script/Makefile         |  5 ++
 test/others/action-script/action-script.sh |  2 +
 test/others/action-script/run.sh           | 60 ++++++++++++++++++++++
 6 files changed, 72 insertions(+)
 create mode 100644 test/others/action-script/.gitignore
 create mode 100644 test/others/action-script/Makefile
 create mode 100755 test/others/action-script/action-script.sh
 create mode 100755 test/others/action-script/run.sh

diff --git a/Makefile b/Makefile
index 9a297d2d8..e4ddd887f 100644
--- a/Makefile
+++ b/Makefile
@@ -449,6 +449,7 @@ lint:
 	shellcheck -x test/others/libcriu/*.sh
 	shellcheck -x test/others/crit/*.sh test/others/criu-coredump/*.sh
 	shellcheck -x test/others/config-file/*.sh
+	shellcheck -x test/others/action-script/*.sh
 	codespell -S tags
 	# Do not append \n to pr_perror, pr_pwarn or fail
 	! git --no-pager grep -E '^\s*\<(pr_perror|pr_pwarn|fail)\>.*\\n"'
diff --git a/scripts/ci/run-ci-tests.sh b/scripts/ci/run-ci-tests.sh
index 79744c750..47749e7fa 100755
--- a/scripts/ci/run-ci-tests.sh
+++ b/scripts/ci/run-ci-tests.sh
@@ -316,6 +316,9 @@ make -C test/others/ns_ext run
 # config file parser and parameter testing
 make -C test/others/config-file run
 
+# action script testing
+make -C test/others/action-script run
+
 # Skip all further tests when running with GCOV=1
 # The one test which currently cannot handle GCOV testing is compel/test
 # Probably because the GCOV Makefile infrastructure does not exist in compel
diff --git a/test/others/action-script/.gitignore b/test/others/action-script/.gitignore
new file mode 100644
index 000000000..c0b6a2490
--- /dev/null
+++ b/test/others/action-script/.gitignore
@@ -0,0 +1 @@
+img-dir-*
diff --git a/test/others/action-script/Makefile b/test/others/action-script/Makefile
new file mode 100644
index 000000000..f1ce191db
--- /dev/null
+++ b/test/others/action-script/Makefile
@@ -0,0 +1,5 @@
+run:
+	@make -C .. loop
+	./run.sh
+
+.PHONY: run
diff --git a/test/others/action-script/action-script.sh b/test/others/action-script/action-script.sh
new file mode 100755
index 000000000..aba8292c0
--- /dev/null
+++ b/test/others/action-script/action-script.sh
@@ -0,0 +1,2 @@
+#!/bin/bash
+touch action-hook-"$CRTOOLS_SCRIPT_ACTION"
diff --git a/test/others/action-script/run.sh b/test/others/action-script/run.sh
new file mode 100755
index 000000000..a82fccf35
--- /dev/null
+++ b/test/others/action-script/run.sh
@@ -0,0 +1,60 @@
+#!/bin/bash
+
+set -ebm
+
+# shellcheck source=test/others/env.sh
+source ../env.sh || exit 1
+
+SELFDIR="$(dirname "$(readlink -f "$0")")"
+SCRIPT="$SELFDIR/action-script.sh"
+IMGDIR="$SELFDIR/img-dir-$$"
+
+rm -rf "$IMGDIR"
+mkdir "$IMGDIR"
+
+trap "cleanup" QUIT TERM INT HUP EXIT
+
+# shellcheck disable=SC2317
+# https://github.com/koalaman/shellcheck/issues/2660
+function cleanup()
+{
+	if [[ -n "$PID" ]]; then
+		kill -9 "$PID"
+	fi
+}
+
+PID=$(../loop)
+if ! $CRIU dump -v4 -o dump.log -t "$PID" -D "$IMGDIR" --action-script "$SCRIPT"; then
+	echo "Failed to checkpoint process $PID"
+	cat dump.log
+	kill -9 "$PID"
+	exit 1
+fi
+
+if ! $CRIU restore -v4 -o restore.log -D "$IMGDIR" -d --pidfile test.pidfile --action-script "$SCRIPT"; then
+	echo "CRIU restore failed"
+	echo FAIL
+	exit 1
+fi
+
+PID=$(cat "$IMGDIR"/test.pidfile)
+
+found_missing_file=false
+hooks=("pre-dump" "post-dump" "pre-restore" "pre-resume" "post-restore" "post-resume")
+
+for hook in "${hooks[@]}"
+do
+	if [ ! -e "$IMGDIR/action-hook-$hook" ]; then
+		echo "ERROR: action-hook-$hook does not exist"
+		found_missing_file=true
+	fi
+done
+
+if [ "$found_missing_file" = true ]; then
+	exit 1
+fi
+
+echo PASS
+
+rm -rf "$IMGDIR"
+exit 0