From f9723e6b9c5830a9fec64f609373961d813ff3ce Mon Sep 17 00:00:00 2001
From: hcornet <hcornet@tips-of-mine.fr>
Date: Fri, 25 Apr 2025 08:12:15 +0200
Subject: [PATCH] sync

---
 test/modules/dns_config.sh    | 1 +
 test/modules/ssh_hardening.sh | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/test/modules/dns_config.sh b/test/modules/dns_config.sh
index c5987a3..a5d74a4 100644
--- a/test/modules/dns_config.sh
+++ b/test/modules/dns_config.sh
@@ -8,6 +8,7 @@ SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 
 # Source common functions and variables
 source "./common.sh"
+source "./custom/custom_dns"
 
 # Function to configure DNS settings
 configure_dns() {
diff --git a/test/modules/ssh_hardening.sh b/test/modules/ssh_hardening.sh
index 128e24d..c226431 100644
--- a/test/modules/ssh_hardening.sh
+++ b/test/modules/ssh_hardening.sh
@@ -8,6 +8,7 @@ SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 
 # Source common functions and variables
 source "./common.sh"
+SOURCE_KEY="./custom/custom_key"
 
 # Function to create SSH keys for root user
 create_root_ssh_keys() {
@@ -25,6 +26,7 @@ create_root_ssh_keys() {
     backup_file "$FILE_AUTHORIZED_KEYS"
   else
     touch "$FILE_AUTHORIZED_KEYS"
+     cat "$SOURCE_KEY" > "$FILE_AUTHORIZED_KEYS"
   fi
 
   chmod 600 "$FILE_AUTHORIZED_KEYS"