firewall/firewall2.sh

#!/bin/bash
MY_IP="47.5.112.50"
ATTACK_THRESHOLD="50"
SERVER_IP='192.168.0.55'
NGINX_ACCESS="/tmp/access.log"
#WIREGUARD=(51820)
WIREGUARD=(57692)
WEB=(80 443)
ADGUARD=(3000 8082 853 )
UPTIME=(4001)
DNS=(53 67 68)
CUPS=(631 5353)
BITCOIN=(8333 8332 8334 4050)
LND=(10009 9735 8080 28334 28333 19998 29000)
SYNCTHING=(22000 8384 21027)
NFS=(2049 111)
JELLYFIN=(8096 7359)
MACHINES=(127.0.0.1)
VIRT_BRIDGE="virbr0"
ADMIN=(22)
#### NFT CONFIG ####
NFT='/usr/sbin/nft'
NFT_TCP="$NFT add rule ip filter input tcp dport"
NFT_UDP="$NFT add rule ip filter input udp dport"
NFT6_UDP="$NFT add rule ip6 filter input udp dport"
NFT6_TCP="$NFT add rule ip6 filter input tcp dport"
NFT_DROP='counter drop'
NFT_ACCEPT='counter accept'
NFT='/usr/sbin/nft'
NFT_CACHE='/tmp/nft.cache'
####
SAVED_BOTS='/opt/firewall/bots.txt'
CRAWLER_DB='/opt/firewall/crawlers.txt'
SAFE_TRAFFIC='/opt/firewall/safe.txt'
PEDO_DB='/opt/firewall/pedo.txt'
PEDO_LOG='/opt/firewall/pedo-log.txt'
fediblock_DB='/opt/firewall/fediblock.txt'
fediblock_TMP='/tmp/fediblock.tmp'
ATTACKER_DB='/opt/firewall/attacker-db.txt'
ATTACKER_LOG='/opt/firewall/attackers.txt'
BOT_ACCOUNT="blockbot@detroitriotcity.com"
CRAWLER_TMP='/tmp/crawlers.txt'
DATE="$(date +%Y:%H: -d "1 hour ago")"
#DATE="$(date +%Y:%H:)";
RULE_SET='/opt/firewall/nft.rules'
COUNTRY=(
	https://www.ipdeny.com/ipblocks/data/countries/il.zone
	https://www.ipdeny.com/ipblocks/data/countries/cn.zone
)

noscl setprivate 0fe0b7d521c1b599b12a3b1e72acc6f08d2011083f25379d0b90a506d044266f
nft list table filter > $NFT_CACHE

blockCountry() {
	for i in "${COUNTRY[@]}"; do
		echo
		echo "Blocking $i"
		DB=( $(curl $i) )
		for j in "${DB[@]}"; do
			$NFT add rule ip filter input ip saddr $j $NFT_DROP
		done
	done

}


wireguard-networking() {
	$NFT add table nat
	$NFT add chain nat postrouting
	$NFT add rule nat postrouting oif wg0 iif enp11s0
	$NFT add rule nat postrouting oif enp11s0 iif wg0
	$NFT add rule nat postrouting masquerade
	$NFT add rule filter forward iifname wg0 oif enp11s0 $NFT_ACCEPT
	$NFT add rule filter forward iifname enp11s0 oif wg0 $NFT_ACCEPT
	$NFT add rule ip filter input ip saddr 192.168.5.0/24 $NFT_ACCEPT
}

attacker-protection() {
	saved-attackers
	watch
	pedo-search
	bot-search
	attacker-search
}

bot-search() {
	DATE="$(date +%d/%b/%Y:%H:%M -d '1 min ago' )"
	CRAWLERS=( $( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | grep -Evi 'Guro|spank|report|rape|block' | grep -Ei -f $CRAWLER_DB | cut -d "-" -f1 | sort -u))

	echo
	echo "Processing Web Crawler list into NFT....."
	echo
	for i in "${CRAWLERS[@]}"; do
		CHECK=$( cat $NFT_CACHE | grep $i)
		if [ "$CHECK"  = "" ];
			then
			       $NFT add rule ip filter input ip saddr $i $NFT_DROP
			else
				echo 
				echo "Skipping Duplicate IP $i"
				echo
		fi	
	done
}

pedo-search() {
	DATE="$(date +%d/%b/%Y:%H:%M -d '1 min ago' )"
	echo
	echo "Processing Pedo Searches into NFT....."
	PEDO_SEARCH=$( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | grep -Ei 'tag|search' | grep -Evi -f $CRAWLER_DB | grep -Ei -f $PEDO_DB | head -1) 
		echo $PEDO_SEARCH
		if [ -z "$PEDO_SEARCH" ];
			then
			       echo "No Pedos Found"
			       else
			       	IP=$(echo $PEDO_SEARCH | cut -d ' ' -f1)
			       $NFT add rule ip filter input ip saddr $IP $NFT_DROP
			       noscl setprivate 0fe0b7d521c1b599b12a3b1e72acc6f08d2011083f25379d0b90a506d044266f
			       noscl publish --profile=33c74427f3b2b73d5e38f3e6c991c122a55d204072356f71da49a0e209fb6940 "$QUERY"

		fi
}

attacker-search() {
	echo
	echo "Processing Attacker Searches into NFT....."
	echo
	DATE="$(date +%d/%b/%Y:%H:%M -d '1 min ago' )"
	ATTACKER_SEARCH=( $( grep $DATE $NGINX_ACCESS | grep -Ei '127.0.0.1|"$DATE"' | grep -vi $MY_IP | grep -Ei -f $ATTACKER_DB | cut -d "-" -f1 | sort -u))
	for i in "${ATTACKER_SEARCH[@]}"; do
		$NFT add rule ip filter input ip saddr $i $NFT_DROP
		QUERY=$(cat $NGINX_ACCESS | grep -i "$DATE" | grep -vi $MY_IP | grep $i | grep -Ei -f $ATTACKER_DB | head -1)
		if [ -z "$QUERY" ]; then
			echo "No Attackers Found"
		else
			echo "Found Attacker!"
			noscl publish --profile=33c74427f3b2b73d5e38f3e6c991c122a55d204072356f71da49a0e209fb6940 "$QUERY"
			echo $i >>$ATTACKER_LOG
		fi
	done

}

basic-security() {
	$NFT add rule filter input icmp type echo-request $NFT_DROP
	$NFS add rule filter input log
	$NFT rule filter input log $NFT_DROP
	$NFT rule filter output $NFT_ACCEPT
	$NFT rule filter forward $NFT_ACCEPT
	$NFT insert rule filter input ct state established $NFT_ACCEPT
	$NFT insert rule filter input iif lo $NFT_ACCEPT

	$NFT -f /usr/share/nftables/ipv6-filter.nft
	$NFT add rule ip6 filter input icmpv6 type nd-neighbor-solicit $NFT_DROP
	$NFT add rule ip6 filter input icmpv6 type nd-router-advert $NFT_DROP
}

virtualization() {
	ip link del virbr0
	killall dnsmasq
	/usr/bin/systemctl restart libvirtd
	/usr/bin/virsh net-start default
	/usr/bin/systemctl restart pleroma
	$NFT insert rule filter input iif $VIRT_BRIDGE $NFT_ACCEPT
}


uptimeKuma() {
	for i in "${UPTIME[@]}"; do
		$NFT add rule ip filter input ip saddr $SERVER_IP tcp dport $i accept
	done
}

admin() {
	for i in "${ADMIN[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
	done
}


wireguard() {
	sysctl -w net.ipv4.conf.all.forwarding=1
	for i in "${WIREGUARD[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
		$NFT6_UDP $i $NFT_ACCEPT
	done
}

web() {
	for i in "${WEB[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
	done
}

dns(){
	for i in "${DNS[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
		$NFT6_UDP $i $NFT_ACCEPT
	done

}

adguard() {
	for i in "${ADGUARD[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
	done
}

cups() {
	for i in "${CUPS[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
		$NFT6_UDP $i $NFT_ACCEPT
	done
}

bitcoin() {
	for i in "${BITCOIN[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
	done
}

lnd() {
	for i in "${LND[@]}"; do
		$NFT add rule ip filter input ip saddr $SERVER_IP tcp dport $i accept
	done
}

syncthingServer() {
	for i in "${SYNCTHING[@]}"; do
		$NFT add rule ip filter input ip saddr $SERVER_IP tcp dport $i accept
	done

}

syncthing() {
	for i in "${SYNCTHING[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
		$NFT6_UDP $i $NFT_ACCEPT
	done
}

jellyfin() {
	for i in "${JELLYFIN[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
		$NFT6_TCP $i $NFT_ACCEPT
		$NFT6_UDP $i $NFT_ACCEPT
	done
}

kde-connect() {
		$NFT_TCP 1714-1764 $NFT_ACCEPT
		$NFT_UDP 1714-1764 $NFT_ACCEPT
}

nfs() {
	for i in "${NFS[@]}"; do
		$NFT_TCP $i $NFT_ACCEPT
		$NFT_UDP $i $NFT_ACCEPT
	done
}

trust() {
	for i in "${MACHINES[@]}"; do
		$NFT add rule filter input ip saddr $i $NFT_ACCEPT
	done
}

start() {

	$NFT flush ruleset

#	if [ -f "$RULE_SET" ]; then
#		echo
		echo "Importing Existing Rule Set"
		#$NFT -f $RULE_SET
#	else
		#echo
		#echo "No existing Rules saved"
		$NFT -f /usr/share/nftables/ipv4-filter.nft
#	fi

	if [[ $HOSTNAME == *"nas"* ]]; then
		attacker-protection
		wireguard
		web
		admin
		adguard
		dns
		cups
		syncthingServer
		syncthing
		#blockCountry
		jellyfin
		wireguard-networking
		uptimeKuma
		docker restart uptime-kuma
		$NFT insert rule filter input iif docker0 $NFT_ACCEPT
		basic-security
	else
		virtualization
		basic-security
	fi

}

status() {
	DATE="$(date +%d/%b/%Y:%H:%M -d '1 min ago' )"
	STATS=$( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | wc -l  )
	GET=$( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | grep GET | wc -l  )
	POST=$( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | grep POST | wc -l  )
	PUT=$( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | grep -i PUT | wc -l  )
	CRAWL=$( grep $DATE $NGINX_ACCESS | grep -vi $MY_IP | grep -Ei -f $CRAWLER_DB | wc -l  )
	echo "=================================================================="
	echo "Attack Threshold: $ATTACK_THRESHOLD"
	echo "Firewall Rules: $($NFT list table filter | wc -l)"
	echo
	echo "Traffic Last Minute: $STATS"
	echo "	GET: $GET"
	echo "	PUT: $PUT"
	echo "	POST: $POST"
	echo "	Crawlers: $CRAWL"
	echo
	echo "Blocked IP's:"
	cat /tmp/tmp-blocked.txt
	echo "=================================================================="
}

stop() {
	forgive	
	$NFT -s list ruleset | tee $RULE_SET
	$NFT flush ruleset
	$NFT -f /usr/share/nftables/ipv4-filter.nft
	$NFT add rule filter input icmp type echo-request $NFT_ACCEPT
	$NFT rule filter input $NFT_ACCEPT
	$NFT rule filter output $NFT_ACCEPT
	$NFT rule filter forward $NFT_ACCEPT
	$NFT insert rule filter input ct state established $NFT_ACCEPT
	$NFT insert rule filter input iif lo $NFT_ACCEPT

	$NFT -f /usr/share/nftables/ipv6-filter.nft
	$NFT add rule ip6 filter input icmpv6 type nd-neighbor-solicit $NFT_ACCEPT
	$NFT add rule ip6 filter input icmpv6 type nd-router-advert $NFT_ACCEPT
}

forgive() {
	IP=( $( grep -vi $MY_IP /tmp/tmp-blocked.txt ) )
	echo $IP
	for i in "${IP[@]}"; do
		echo "Checking $i"
		HANDLE=$(nft -n -a list ruleset | grep $i | grep handle | cut -d '#' -f2 | cut -d ' ' -f3)
		echo "Removing: $i"
		echo $NFT delete rule ip filter input handle $HANDLE
		$NFT delete rule ip filter input handle $HANDLE
	done
	echo > /tmp/tmp-blocked.txt
}

saved-attackers() {
	echo
	IP=( $( cat $ATTACKER_LOG | grep -vi $MY_IP | cut -d ' ' -f1 | sort -u  ) )
	for i in "${IP[@]}"; do
		CHECK=$( cat $NFT_CACHE | grep $i)
		if [ "$CHECK" = "" ];
			then
					echo "Blocking IP: $i"
					logger "Blocking IP: $i"
					$NFT add rule ip filter input ip saddr $i $NFT_DROP
			else
				echo 
				echo "Skipping Duplicate IP $i"
				echo
		fi	
	done
}

watch() {
	DATE="$(date +%d/%b/%Y:%H:%M -d '1 min ago' )"
	echo "Scanning $DATE"
	echo
	IP=( $( grep $DATE $NGINX_ACCESS | grep -Evi -f $SAFE_TRAFFIC | grep -vi $MY_IP | cut -d ' ' -f1 | sort -u  ) )
	for i in "${IP[@]}"; do
		COUNT=$( grep $DATE $NGINX_ACCESS | grep "$i" | wc -l) 
		echo "$i $COUNT"
		CHECK=$( cat $NFT_CACHE | grep $i)
		if [ "$CHECK" = "" ];
			then
				if [[ "$COUNT" -gt "$ATTACK_THRESHOLD" ]]; then
					echo "Blocking IP: $i"
					logger "Blocking IP: $i"
					echo $i >> /tmp/tmp-blocked.txt
					$NFT add rule ip filter input ip saddr $i $NFT_DROP
			 	fi
			else
				echo 
				echo "Skipping Duplicate IP $i"
				echo
		fi	
	done
}

if [ "$1" = "start" ]; then
	start
elif [ "$1" = "virt" ]; then
	virtualization
elif [ "$1" = "bot-search" ]; then
	bot-search
elif [ "$1" = "attacker-protection" ]; then
	attacker-protection
elif [ "$1" = "country" ]; then
	blockCountry
elif [ "$1" = "status" ]; then
	status
elif [ "$1" = "forgive" ]; then
	forgive
elif [ "$1" = "watch" ]; then
	watch
elif [ "$1" = "stop" ]; then
	stop
elif [ "$1" = "saved" ]; then
	saved-bots
else
	echo "Invalid Choice"
fi