From fda9faffe19b08d71dbb45684f9e1a23bccfd656 Mon Sep 17 00:00:00 2001
From: chanakya <chanakyavasantha@gmail.com>
Date: Fri, 15 Dec 2023 19:27:56 +0530
Subject: [PATCH] Pre-Link test Updaated with audit message

---
 tests/test_remove_prelink.sh | 43 ++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 tests/test_remove_prelink.sh

diff --git a/tests/test_remove_prelink.sh b/tests/test_remove_prelink.sh
new file mode 100644
index 0000000..c623534
--- /dev/null
+++ b/tests/test_remove_prelink.sh
@@ -0,0 +1,43 @@
+#!/bin/bash
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
+PYTHON_SCRIPT="$SCRIPT_DIR/../harden/process_hardening/remove_prelink.py"
+echo "Python script path: $PYTHON_SCRIPT"
+
+# Check if the Python script exists
+if [ ! -f "$PYTHON_SCRIPT" ]; then
+    echo "Error: Python script not found."
+    exit 1
+fi
+
+# Run the Python script and capture its output
+echo "Running the Python script..."
+script_output=$(python3 "$PYTHON_SCRIPT")
+
+# Check the exit status of the Python script
+if [ $? -ne 0 ]; then
+    echo "Python script execution failed."
+    exit 1
+fi
+
+# Optionally, print the output for verification
+echo "Python script output:"
+echo "$script_output"
+
+# Execute the output as a Bash script
+# WARNING: Executing scripts directly can be risky, especially with sudo commands.
+# Ensure you thoroughly understand and trust the script before executing.
+echo "Executing the generated Bash script..."
+bash -c "$script_output"
+
+# Execute dpkg-query command and check the output
+echo "Executing dpkg-query..."
+dpkg_output=$(dpkg-query -W -f='${binary:Package}\t${Status}\t${db:Status-Status}\n')
+
+# Check if dpkg-query output contains the specific line
+if echo "$dpkg_output" | grep -q "prelink unknown ok not-installed not-installed"; then
+    echo "pre-link is not installed."
+else
+    echo "pre-link may be installed."
+fi
+
+echo "Script executed successfully."