From 6487fdf5e6b85923f4f8c40b0598610b6a13f694 Mon Sep 17 00:00:00 2001
From: Wes Lambert <wlambertts@gmail.com>
Date: Thu, 25 Jun 2020 15:46:37 +0000
Subject: [PATCH] Add Strelka YARA function

---
 setup/so-functions | 4 ++++
 setup/so-setup     | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/setup/so-functions b/setup/so-functions
index 37145e12b..1afd6b90f 100755
--- a/setup/so-functions
+++ b/setup/so-functions
@@ -1622,3 +1622,7 @@ es_heapsize() {
 		export NODE_ES_HEAP_SIZE
 	fi
 }
+
+strelka_yara_update() {
+        so-yara-update
+}
diff --git a/setup/so-setup b/setup/so-setup
index 634389dcd..60bffd618 100755
--- a/setup/so-setup
+++ b/setup/so-setup
@@ -257,6 +257,9 @@ if [[ $is_master ]]; then
 	whiptail_enable_components
 	if [[ $STRELKA == 1 ]]; then
 	    whiptail_strelka_rules
+            if [[ $STRELKARULES == 1 ]]; then
+                strelka_yara_update
+            fi
 	fi
 	collect_webuser_inputs
 	get_redirect