From fc886341591d6d34cbdf1d4057f63aada1363ceb Mon Sep 17 00:00:00 2001
From: Mike Reeves <mike.reeves@securityonionsolutions.com>
Date: Mon, 5 Apr 2021 15:01:21 -0400
Subject: [PATCH] Set the Repo for airgap during install

---
 setup/so-functions | 11 +++++++++++
 setup/so-setup     |  2 ++
 2 files changed, 13 insertions(+)

diff --git a/setup/so-functions b/setup/so-functions
index c9f0925ec..47d75f1c0 100755
--- a/setup/so-functions
+++ b/setup/so-functions
@@ -44,6 +44,17 @@ logCmd() {
 }
 ### End Logging Section ###
 
+airgap_repo() {
+	# Remove all the repo files
+	rm -rf /etc/yum.repos.d/*
+	echo "[airgap_repo]" > /etc/yum./repos.d/airgap_repo.repo
+    echo "baseurl=https://$MSRV/repo" >> /etc/yum.repos.d/airgap_repo.repo
+    echo "gpgcheck=1" >> /etc/yum.repos.d/airgap_repo.repo
+    echo "sslverify=0" >> /etc/yum.repos.d/airgap_repo.repo
+    echo "name=Airgap Repo" >> /etc/yum.repos.d/airgap_repo.repo
+    echo "enabled=1" >> /etc/yum.repos.d/airgap_repo.repo
+}
+
 airgap_rules() {
 	# Copy the rules for suricata if using Airgap
 	mkdir -p /nsm/repo/rules
diff --git a/setup/so-setup b/setup/so-setup
index 0aa78aa10..d3ef48ba6 100755
--- a/setup/so-setup
+++ b/setup/so-setup
@@ -648,6 +648,8 @@ set_redirect >> $setup_log 2>&1
 	if [[ ! $is_airgap ]]; then
 	    securityonion_repo >> $setup_log 2>&1
 	    update_packages >> $setup_log 2>&1
+	else
+	    airgap_repo >> $setup_log 2>&1
 	fi
 
 	if [[ $is_sensor || $is_helix || $is_import ]]; then