test.sh

#!/bin/bash

# Codes d'erreur
# 1 : Argument inconnu
# 2 : Conteneur non créé
# 3 : Conteneur non démarré
# 4 : Keystore non supprimé sur le dashboard
# 5 : Version du module filebeat incorrecte
# 11 : Pas d'idempotence

# FONCTIONS

afficher_separateur()
{
	echo ""
	echo "#####################################################"
	echo "${1}"
	echo "#####################################################"
}

afficher_separateur_test()
{
	echo ""
	echo "-----------------------------------------------------"
	echo "${1}"
	echo "-----------------------------------------------------"
}

# Création d'un conteneur
creation_conteneur()
{
	# Destruction de l'ancien conteneur si existant
	lxc-ls | grep --word-regexp --only-matching "${1}" > /dev/null
	if [ "$?" -eq "0" ]
	then
		echo "Ancien conteneur ${1} trouvé" > ${sortie_flux}
		lxc-stop -n "${1}" > ${sortie_flux} 2>&1
		lxc-destroy -n "${1}" && echo "Ancien conteneur ${1} détruit" > ${sortie_flux}
	fi
	# Création du nouveau conteneur
	lxc-copy -n "${conteneur_base}" -N "${1}" && echo "Conteneur ${1} créé" > ${sortie_flux} || exit 2
	# Démarage du nouveau conteneur
	systemd-run --user --scope -p "Delegate=yes" -- lxc-start -n ${1}  && echo "Conteneur ${1} démarré" > ${sortie_flux} || exit 3
}

# Création du groupe de conteneurs défini dans la variable $conteneurs
creation_conteneurs()
{
	afficher_separateur_test "Création des conteneurs"
	for conteneur in "${conteneurs[@]}"
	do
		creation_conteneur "${conteneur}"
	done
}

# passage de puppet sur les conteneurs
passage_puppet()
{
	for conteneur in "${conteneurs[@]}"
	do
		readarray -d "-" -t tab <<< "${conteneur}"
		manifest="${tab[1]}${suffixe_manifests}.pp"
		afficher_separateur_test "Passage de Puppet sur ${conteneur}"
		lxc-attach -n "${conteneur}" -- puppet apply "/root/Manifests/Wazuh/${manifest}" > ${sortie_flux}
	done
}

# Procédure de test appliquée à chaque situation à tester
procedure_test()
{
	# Passage de Puppet
	passage_puppet

	# Tests
	test_idempotence
	test_keystore_dashboard
	test_version_module_filebeat

	return 0
}

# Vérification de l'idempotence
test_idempotence()
{
	afficher_separateur_test "Vérification de l'idempotence"

	ancien_sortie_flux="${sortie_flux}"
	sortie_flux="/dev/stdout"
	sortie=`passage_puppet`
	sortie_flux="${ancien_sortie_flux}"
	compteur_notice=`echo "${sortie}" | grep --count "Notice:"`
	echo -e "\n'Notice:' comptés : ${compteur_notice}" > ${sortie_flux}
	notice_attendu=`expr ${#conteneurs[@]} \* 2`
	echo "'Notice:' attendus : ${notice_attendu}" > ${sortie_flux}
	if [ "${compteur_notice}" -ne "${notice_attendu}" ]
	then
		echo "Pas d'idempotence" >&2
		echo "${sortie}" >&2
		exit 11
	else
		echo "Idempotence respectée"
	fi

	return 0
}

# Vérification de la présence/absence du keystore sur le dashboard
test_keystore_dashboard()
{
	if [ "${suffixe_manifests}" == "1" ]
	then
		afficher_separateur_test "Vérification de la suppression du keystore sur le dashboard"
		# Vérification à faire uniquement sur un conteneur qui a le Wazuh Dashboard
		for conteneur in "${conteneurs[@]}"
		do
			lxc-attach -n "${conteneur}" -- dpkg --list | grep -o 'wazuh-dashboard' > '/dev/null'
			if [ "$?" -eq "0" ]
			then
				# Test de l'absence du keystore
				lxc-attach -n "${conteneur}" -- test -f '/usr/share/wazuh-dashboard/config/opensearch_dashboards.keystore'
				if [ "$?" -eq "0" ]
				then
					echo "Le keystore n'a pas été supprimé sur ${conteneur}" >&2
					exit 4
				fi
				echo "Le keystore a été supprimé sur ${conteneur}"
			fi
		done
	fi

	return 0
}

# Vérification de la version du module Filebeat
test_version_module_filebeat()
{
	afficher_separateur_test "Vérification de la version du module filebeat"
	# Vérification à faire uniquement sur un conteneur qui a Filebeat
	for conteneur in "${conteneurs[@]}"
	do
		lxc-attach -n "${conteneur}" -- dpkg --list | grep -o 'filebeat' > '/dev/null'
		if [ "$?" -eq "0" ]
		then
			if [ "${suffixe_manifests}" == "2" ]
			then
				# La version du module Filebeat doit être 0.1 dans le cas du manifest 2
				lxc-attach -n "${conteneur}" -- test -d '/opt/filebeat_wazuh_modules/wazuh-filebeat-0.1/wazuh'
			else
				# La version du module Filebeat doit être 0.2 dans les autres cas
				lxc-attach -n "${conteneur}" -- test -d '/opt/filebeat_wazuh_modules/wazuh-filebeat-0.2/wazuh'
			fi
			if [ "$?" -eq "0" ]
			then
				echo "La version du module filebeat est correcte sur ${conteneur}"
			else
				echo "La version du module filebeat est incorrecte sur ${conteneur}" >&2
				exit 5
			fi
		fi
	done
}

# DÉFINITION DES VARIABLES GLOBALES

# Mode silencieux
if [ "$1" == "quiet" ]
then
	sortie_flux="/dev/null"
else
	sortie_flux="/dev/stdout"
fi

# Définition des noms des conteneurs
conteneur_base="base-puppet"
conteneurs_monolithique=("wazuh-agent-1" "wazuh-monolithe-0")
conteneurs_distribuee=("wazuh-agent-1" "wazuh-master-1" "wazuh-indexer-1" "wazuh-dashboard-0")
conteneurs_cluster=("wazuh-agent-1" "wazuh-master-1" "wazuh-worker-1" "wazuh-worker-2" "wazuh-worker-3" "wazuh-indexer-1" "wazuh-indexer-2" "wazuh-indexer-3" "wazuh-dashboard-0")

# MAIN

# Situation 1, création depuis 0, architecture monolithique
afficher_separateur "Situation 1 : Création depuis 0, architecture monolithique"

conteneurs=("${conteneurs_monolithique[@]}")
creation_conteneurs

suffixe_manifests=""
procedure_test && echo -e "\nSituation réussie"

# Situation 2, sur architecture précédente, suppression de keystore sur dashboard
afficher_separateur "Situation 2 : Suppression du keystore sur le dashboard"

suffixe_manifests="1"
procedure_test && echo -e "\nSituation réussie"

# Situation 3, mise à jour du module Filebeat
afficher_separateur "Situation 3 : Mise à jour du module Filebeat"

creation_conteneurs

afficher_separateur_test "Avant mise à jour"
suffixe_manifests="2"
procedure_test
afficher_separateur_test "Mise à jour vers version actuelle"
suffixe_manifests=""
procedure_test && echo -e "\nSituation réussie"

# Arrêt des conteneurs
afficher_separateur "Arrêt des conteneurs"
for conteneur in "${conteneurs[@]}"
do
	lxc-stop -n "${conteneur}" && echo "Conteneur ${conteneur} arrêté" > ${sortie_flux}
done
echo "Conteneurs arrêtés"

exit 0