From b3114e72935a07d2ee03b08a174e698dab7db08e Mon Sep 17 00:00:00 2001
From: AlfonsoRBJ <sitoruizbravo@hotmail.com>
Date: Thu, 21 Mar 2019 18:25:56 +0100
Subject: [PATCH] add race condition for wazuh alerts template

---
 kibana/config/entrypoint.sh   | 21 +++++++++++++++++++++
 logstash/config/entrypoint.sh | 19 ++++++++++++++++++-
 2 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/kibana/config/entrypoint.sh b/kibana/config/entrypoint.sh
index 1ba0a44d..fc4baf56 100644
--- a/kibana/config/entrypoint.sh
+++ b/kibana/config/entrypoint.sh
@@ -3,6 +3,10 @@
 
 set -e
 
+##############################################################################
+# Waiting for elasticsearch
+##############################################################################
+
 if [ "x${ELASTICSEARCH_URL}" = "x" ]; then
   el_url="http://elasticsearch:9200"
 else
@@ -19,6 +23,23 @@ sleep 10
 >&2 echo "Elasticsearch is up."
 
 
+##############################################################################
+# Waiting for wazuh alerts template
+##############################################################################
+
+strlen=0
+
+while [[ $strlen -eq 0 ]]
+do
+  template=$(curl $el_url:9200/_cat/templates/wazuh -s)
+  strlen=${#template}
+done
+
+sleep 10
+
+>&2 echo "Wazuh alerts template is load."
+
+
 ./wazuh_app_config.sh
 
 sleep 5
diff --git a/logstash/config/entrypoint.sh b/logstash/config/entrypoint.sh
index 4eef7fa7..32c347ec 100644
--- a/logstash/config/entrypoint.sh
+++ b/logstash/config/entrypoint.sh
@@ -26,11 +26,28 @@ sleep 10
 
 >&2 echo "Elasticsearch is up."
 
+##############################################################################
+# Waiting for wazuh alerts template
+##############################################################################
+
+strlen=0
+
+while [[ $strlen -eq 0 ]]
+do
+  template=$(curl $el_url:9200/_cat/templates/wazuh -s)
+  strlen=${#template}
+done
+
+sleep 10
+
+>&2 echo "Wazuh alerts template is load."
+
 ##############################################################################
 # Customize logstash output ip
 ##############################################################################
+
 if [ "$LOGSTASH_OUTPUT" != "" ]; then
-  echo "Customize Logstash ouput ip."
+  >&2 echo "Customize Logstash ouput ip."
   sed -i "s/elasticsearch:9200/$LOGSTASH_OUTPUT:9200/" /usr/share/logstash/pipeline/01-wazuh.conf
   sed -i "s/elasticsearch:9200/$LOGSTASH_OUTPUT:9200/" /usr/share/logstash/config/logstash.yml 
 fi