#!/bin/sh -e

DIRNAME=`dirname $0`
cd $DIRNAME
USAGE="$0 [ --update ]"
if [ `id -u` != 0 ]; then
echo 'You must be root to run this script'
exit 1
fi

if [ $# -eq 1 ]; then
	if [ "$1" = "--update" ] ; then
		time=`ls -l --time-style="+%x %X" amazon_ssm_agent.te | awk '{ printf "%s %s", $6, $7 }'`
		rules=`ausearch --start $time -m avc --raw -se amazon_ssm_agent`
		if [ x"$rules" != "x" ] ; then
			echo "Found avc's to update policy with"
			echo -e "$rules" | audit2allow -R
			echo "Do you want these changes added to policy [y/n]?"
			read ANS
			if [ "$ANS" = "y" -o "$ANS" = "Y" ] ; then
				echo "Updating policy"
				echo -e "$rules" | audit2allow -R >> amazon_ssm_agent.te
				# Fall though and rebuild policy
			else
				exit 0
			fi
		else
			echo "No new avcs found"
			exit 0
		fi
	else
		echo -e $USAGE
		exit 1
	fi
elif [ $# -ge 2 ] ; then
	echo -e $USAGE
	exit 1
fi

echo "Building and Loading Policy"
set -x
make -f /usr/share/selinux/devel/Makefile amazon_ssm_agent.pp || exit
/usr/sbin/semodule -i amazon_ssm_agent.pp

# Generate a man page off the installed module
sepolicy manpage -p . -d amazon_ssm_agent_t
# Fixing the file context on /usr/bin/amazon-ssm-agent
/sbin/restorecon -F -R -v /usr/bin/amazon-ssm-agent
# Fixing the file context on /var/lib/amazon/ssm
/sbin/restorecon -F -R -v /var/lib/amazon/ssm
# Fixing the file context on /var/log/amazon/ssm
/sbin/restorecon -F -R -v /var/log/amazon/ssm
# Fixing the file context on /var/lib/amazon/ssm/ipc/termination
/sbin/restorecon -R -v /var/lib/amazon/ssm/ipc/termination
# Fixing the file context on /usr/bin/ssm-session-worker
/sbin/restorecon -R -v /usr/bin/ssm-session-worker
# Fixing the file context on /usr/lib/systemd/system/amazon-ssm-agent.service
/sbin/restorecon -F -R -v /usr/lib/systemd/system/amazon-ssm-agent.service
# # Generate a rpm package for the newly generated policy

pwd=$(pwd)
rpmbuild --define "_sourcedir ${pwd}" --define "_specdir ${pwd}" --define "_builddir ${pwd}" --define "_srcrpmdir ${pwd}" --define "_rpmdir ${pwd}" --define "_buildrootdir ${pwd}/.build"  -ba amazon_ssm_agent_selinux.spec