From 9e873d34cbeff5041c09a7d8d31ff40795c9c89b Mon Sep 17 00:00:00 2001
From: Mike Reeves <mike.reeves@securityonionsolutions.com>
Date: Mon, 26 Mar 2018 15:24:23 -0400
Subject: [PATCH] Setup Script - Node LS Heap

---
 so-setup-network.sh | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/so-setup-network.sh b/so-setup-network.sh
index e918af7cc..757084bf7 100644
--- a/so-setup-network.sh
+++ b/so-setup-network.sh
@@ -421,7 +421,11 @@ whiptail_node_es_heap() {
   NODEESHEAP=$(whiptail --title "Security Onion Setup" --inputbox \
   "Enter ES Heap Size" 10 60 $ES_HEAP_SIZE 3>&1 1>&2 2>&3)
 }
-#whiptail_node_ls_heap
+whiptail_node_ls_heap() {
+  ls_heapsize
+  NODELSHEAP=$(whiptail --title "Security Onion Setup" --inputbox \
+  "Enter LogStash Heap Size: Recommended value is pre-populated" 10 60 $LS_HEAP_SIZE 3>&1 1>&2 2>&3)
+}
 #whiptail_node_ls_pipeline_worker
 #whiptail_node_ls_pipline_batchsize
 #whiptail_node_ls_input_threads
@@ -570,7 +574,7 @@ if (whiptail_you_sure) then
     whiptail_node_advanced
     if [ $NODESETUP == 'NODEADVANCED' ]; then
       whiptail_node_es_heap
-      #whiptail_node_ls_heap
+      whiptail_node_ls_heap
       #whiptail_node_ls_pipeline_worker
       #whiptail_node_ls_pipline_batchsize
       #whiptail_node_ls_input_threads