From 8404897fe3cf7fe9dbd66b15552e8a995bbf3b05 Mon Sep 17 00:00:00 2001
From: Wes Lambert <wlambertts@gmail.com>
Date: Wed, 12 Dec 2018 06:05:13 +0000
Subject: [PATCH] Wazuh - Move agent config to init.sls

---
 salt/wazuh/files/agent/wazuh-register-agent |  5 +++--
 salt/wazuh/init.sls                         | 13 ++++++++-----
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/salt/wazuh/files/agent/wazuh-register-agent b/salt/wazuh/files/agent/wazuh-register-agent
index e9f9dbeb5..1854f55ff 100755
--- a/salt/wazuh/files/agent/wazuh-register-agent
+++ b/salt/wazuh/files/agent/wazuh-register-agent
@@ -1,3 +1,4 @@
+{%- set ip = salt['pillar.get']('static:masterip', '') %}
 #!/bin/bash
 
 ###
@@ -23,7 +24,7 @@ PROTOCOL="https"
 USER="foo"
 PASSWORD="bar"
 AGENT_NAME=$(hostname)
-AGENT_IP=""
+AGENT_IP="{{ip}}"
 
 display_help() {
 cat <<HELP_USAGE
@@ -49,7 +50,7 @@ register_agent() {
 
   if [ "$?" != "0" ]; then
     echo -e $API_RESULT | sed -rn 's/.*"message":"(.+)".*/\1/p'
-    exit 1
+    exit 0
   fi
   # Get agent id and agent key 
   AGENT_ID=$(echo $API_RESULT | cut -d':' -f 4 | cut -d ',' -f 1)
diff --git a/salt/wazuh/init.sls b/salt/wazuh/init.sls
index 1d0b9a99e..ac05f1984 100644
--- a/salt/wazuh/init.sls
+++ b/salt/wazuh/init.sls
@@ -1,10 +1,6 @@
 {%- set HOSTNAME = salt['grains.get']('host', '') %}
 
-#vm.max_map_count:
-#  sysctl.present:
-#    - value: 262144
-
-# Add ossec Group
+# Add ossec group
 ossecgroup:
   group.present:
     - name: ossec
@@ -58,6 +54,7 @@ wazuhagentregister:
     - user: 0
     - group: 0
     - mode: 755
+    - template: jinja
 
 so-wazuh:
   docker_container.running:
@@ -72,3 +69,9 @@ so-wazuh:
     - binds:
       - /opt/so/wazuh/:/var/ossec/data/:rw
 
+# Register the agent
+registertheagent:
+  cmd.run:
+    - name: /usr/sbin/wazuh-register-agent
+    - cwd: /
+    #- stateful: True