From 38bf8c4bc0f6614a6c137a23bf6fbd81045580dd Mon Sep 17 00:00:00 2001
From: "thibault.dewailly" <thibault.dewailly@corp.ovh.com>
Date: Tue, 24 Dec 2024 09:54:51 +0000
Subject: [PATCH] feat: add tftp check

---
 bin/hardening/2.2.18_disable_tftp.sh   | 73 ++++++++++++++++++++++++++
 tests/hardening/2.2.18_disable_tftp.sh | 16 ++++++
 2 files changed, 89 insertions(+)
 create mode 100755 bin/hardening/2.2.18_disable_tftp.sh
 create mode 100644 tests/hardening/2.2.18_disable_tftp.sh

diff --git a/bin/hardening/2.2.18_disable_tftp.sh b/bin/hardening/2.2.18_disable_tftp.sh
new file mode 100755
index 0000000..3a0f0e7
--- /dev/null
+++ b/bin/hardening/2.2.18_disable_tftp.sh
@@ -0,0 +1,73 @@
+#!/bin/bash
+
+# run-shellcheck
+#
+# CIS Debian Hardening
+#
+
+#
+# 2.2.18 Ensure TFTP server is not enabled  (Scored)
+#
+
+set -e # One error, it's over
+set -u # One variable unset, it's over
+
+# shellcheck disable=2034
+HARDENING_LEVEL=3
+# shellcheck disable=2034
+DESCRIPTION="Ensure Trivial File Transfer Protocol server is not enabled."
+# shellcheck disable=2034
+HARDENING_EXCEPTION=tftp
+
+PACKAGES='tftpd tftpd-hpa'
+
+# This function will be called if the script status is on enabled / audit mode
+audit() {
+    for PACKAGE in $PACKAGES; do
+        is_pkg_installed "$PACKAGE"
+        if [ "$FNRET" = 0 ]; then
+            crit "$PACKAGE is installed!"
+        else
+            ok "$PACKAGE is absent"
+        fi
+    done
+}
+
+# This function will be called if the script status is on enabled mode
+apply() {
+    for PACKAGE in $PACKAGES; do
+        is_pkg_installed "$PACKAGE"
+        if [ "$FNRET" = 0 ]; then
+            crit "$PACKAGE is installed, purging it"
+            apt-get purge "$PACKAGE" -y
+            apt-get autoremove -y
+        else
+            ok "$PACKAGE is absent"
+        fi
+    done
+}
+
+# This function will check config parameters required
+check_config() {
+    :
+}
+
+# Source Root Dir Parameter
+if [ -r /etc/default/cis-hardening ]; then
+    # shellcheck source=../../debian/default
+    . /etc/default/cis-hardening
+fi
+if [ -z "$CIS_LIB_DIR" ]; then
+    echo "There is no /etc/default/cis-hardening file nor cis-hardening directory in current environment."
+    echo "Cannot source CIS_LIB_DIR variable, aborting."
+    exit 128
+fi
+
+# Main function, will call the proper functions given the configuration (audit, enabled, disabled)
+if [ -r "${CIS_LIB_DIR}"/main.sh ]; then
+    # shellcheck source=../../lib/main.sh
+    . "${CIS_LIB_DIR}"/main.sh
+else
+    echo "Cannot find main.sh, have you correctly defined your root directory? Current value is $CIS_LIB_DIR in /etc/default/cis-hardening"
+    exit 128
+fi
diff --git a/tests/hardening/2.2.18_disable_tftp.sh b/tests/hardening/2.2.18_disable_tftp.sh
new file mode 100644
index 0000000..4ad9ef8
--- /dev/null
+++ b/tests/hardening/2.2.18_disable_tftp.sh
@@ -0,0 +1,16 @@
+# shellcheck shell=bash
+# run-shellcheck
+test_audit() {
+    describe Running on blank host
+    register_test retvalshouldbe 0
+    dismiss_count_for_test
+    # shellcheck disable=2154
+    run blank "${CIS_CHECKS_DIR}/${script}.sh" --audit-all
+
+    ##################################################################
+    # For this test, we only check that it runs properly on a blank  #
+    # host, and we check root/sudo consistency. But, we don't test   #
+    # the apply function because it can't be automated or it is very #
+    # long to test and not very useful.                              #
+    ##################################################################
+}