mirror of
https://github.com/Security-Onion-Solutions/securityonion.git
synced 2025-12-06 09:12:45 +01:00
84 lines
3.2 KiB
Plaintext
84 lines
3.2 KiB
Plaintext
# Copyright Security Onion Solutions LLC and/or licensed to Security Onion Solutions LLC under one
|
|
# or more contributor license agreements. Licensed under the Elastic License 2.0; you may not use
|
|
# this file except in compliance with the Elastic License 2.0.
|
|
{% from 'vars/globals.map.jinja' import GLOBALS %}
|
|
{% set CUSTOMFQDN = salt['pillar.get']('elasticfleet:config:server:custom_fqdn') %}
|
|
|
|
. /usr/sbin/so-common
|
|
|
|
# Only run on Managers
|
|
if ! is_manager_node; then
|
|
printf "Not a Manager Node... Exiting"
|
|
exit 0
|
|
fi
|
|
|
|
function update_logstash_outputs() {
|
|
# Generate updated JSON payload
|
|
JSON_STRING=$(jq -n --arg UPDATEDLIST $NEW_LIST_JSON '{"name":"grid-logstash","type":"logstash","hosts": $UPDATEDLIST,"is_default":true,"is_default_monitoring":true,"config_yaml":""}')
|
|
|
|
# Update Logstash Outputs
|
|
curl -K /opt/so/conf/elasticsearch/curl.config -L -X PUT "localhost:5601/api/fleet/outputs/so-manager_logstash" -H 'kbn-xsrf: true' -H 'Content-Type: application/json' -d "$JSON_STRING" | jq
|
|
}
|
|
|
|
# Get current list of Logstash Outputs
|
|
RAW_JSON=$(curl -K /opt/so/conf/elasticsearch/curl.config 'http://localhost:5601/api/fleet/outputs/so-manager_logstash')
|
|
|
|
# Check to make sure that the server responded with good data - else, bail from script
|
|
CHECKSUM=$(jq -r '.item.id' <<< "$RAW_JSON")
|
|
if [ "$CHECKSUM" != "so-manager_logstash" ]; then
|
|
printf "Failed to query for current Logstash Outputs..."
|
|
exit 1
|
|
fi
|
|
|
|
# Get the current list of Logstash outputs & hash them
|
|
CURRENT_LIST=$(jq -c -r '.item.hosts' <<< "$RAW_JSON")
|
|
CURRENT_HASH=$(sha1sum <<< "$CURRENT_LIST" | awk '{print $1}')
|
|
|
|
# Create array & add initial elements
|
|
if [ "{{ GLOBALS.hostname }}" = "{{ GLOBALS.url_base }}" ]; then
|
|
NEW_LIST=("{{ GLOBALS.url_base }}:5055")
|
|
else
|
|
NEW_LIST=("{{ GLOBALS.url_base }}:5055" "{{ GLOBALS.hostname }}:5055")
|
|
fi
|
|
|
|
{% if CUSTOMFQDN != "" %}
|
|
# Add Custom Hostname to list
|
|
NEW_LIST+=("{{ CUSTOMFQDN }}:5055")
|
|
{% endif %}
|
|
|
|
# Query for the current Grid Nodes that are running Logstash
|
|
LOGSTASHNODES=$(salt-call --out=json pillar.get logstash:nodes | jq '.local')
|
|
|
|
# Query for Receiver Nodes & add them to the list
|
|
if grep -q "receiver" <<< $LOGSTASHNODES; then
|
|
readarray -t RECEIVERNODES < <(jq -r ' .receiver | keys_unsorted[]' <<< $LOGSTASHNODES)
|
|
for NODE in "${RECEIVERNODES[@]}"
|
|
do
|
|
NEW_LIST+=("$NODE:5055")
|
|
done
|
|
fi
|
|
|
|
# Query for Fleet Nodes & add them to the list
|
|
if grep -q "fleet" <<< $LOGSTASHNODES; then
|
|
readarray -t FLEETNODES < <(jq -r ' .fleet | keys_unsorted[]' <<< $LOGSTASHNODES)
|
|
for NODE in "${FLEETNODES[@]}"
|
|
do
|
|
NEW_LIST+=("$NODE:5055")
|
|
done
|
|
fi
|
|
|
|
# Sort & hash the new list of Logstash Outputs
|
|
NEW_LIST_JSON=$(jq --compact-output --null-input '$ARGS.positional' --args -- "${NEW_LIST[@]}")
|
|
NEW_HASH=$(sha1sum <<< "$NEW_LIST_JSON" | awk '{print $1}')
|
|
|
|
# Compare the current & new list of outputs - if different, update the Logstash outputs
|
|
if [ "$NEW_HASH" = "$CURRENT_HASH" ]; then
|
|
printf "\nHashes match - no update needed.\n"
|
|
printf "Current List: $CURRENT_LIST\nNew List: $NEW_LIST_JSON\n"
|
|
exit 0
|
|
else
|
|
printf "\nHashes don't match - update needed.\n"
|
|
printf "Current List: $CURRENT_LIST\nNew List: $NEW_LIST_JSON\n"
|
|
update_logstash_outputs
|
|
fi
|