From aaa6cf816a1dd950be432417c61082246d44c4f7 Mon Sep 17 00:00:00 2001
From: m0duspwnens <josh.patterson@securityonionsolutions.com>
Date: Mon, 15 Jun 2020 09:42:06 -0400
Subject: [PATCH] set firewall during setup for fleet standalone

---
 setup/so-functions | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/setup/so-functions b/setup/so-functions
index e5bbaa594..72c6395b0 100755
--- a/setup/so-functions
+++ b/setup/so-functions
@@ -1460,6 +1460,9 @@ set_initial_firewall_policy() {
 					ssh -i /root/.ssh/so.key soremote@"$MSRV" sudo $default_salt_dir/pillar/data/addtotab.sh sensorstab "$MINION_ID" "$MAINIP" "$num_cpu_cores" "$random_uid" "$MNIC" "$filesystem_root" "$filesystem_nsm" bond0
 					ssh -i /root/.ssh/so.key soremote@"$MSRV" sudo $default_salt_dir/pillar/data/addtotab.sh nodestab "$MINION_ID" "$MAINIP" "$num_cpu_cores" "$random_uid" "$MNIC" "$filesystem_root" "$filesystem_nsm"
 					;;
+				'FLEET')
+				    ssh -i /root/.ssh/so.key soremote@"$MSRV" sudo $default_salt_dir/salt/common/tools/sbin/so-firewall --apply includehost beats_endpoint "$MAINIP"
+					;;
 			esac
 			;;
 		'PARSINGNODE')