add timesketch deployscript
This commit is contained in:
TKE
127
deploy_timesketch.sh
Executable file
127
deploy_timesketch.sh
Executable file
@@ -0,0 +1,127 @@
|
||||
#!/bin/bash
|
||||
# Copyright 2020 Google Inc. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
set -e
|
||||
|
||||
# Exit early if run as non-root user.
|
||||
if [ "$EUID" -ne 0 ]; then
|
||||
echo "ERROR: This script need to run as root."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Exit early if a timesketch directory already exists.
|
||||
if [ -d "./timesketch" ]; then
|
||||
echo "ERROR: Timesketch directory already exist."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Exit early if docker is not available.
|
||||
if [ "$(systemctl is-active docker)" != "active" ]; then
|
||||
echo "ERROR: Docker is not available."
|
||||
echo "See: https://docs.docker.com/engine/install/ubuntu/"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Exit early if docker-compose is not installed.
|
||||
if ! command -v docker-compose; then
|
||||
echo "ERROR: docker-compose is not installed."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Exit early if there are Timesketch containers already running.
|
||||
if [ ! -z "$(docker ps | grep timesketch)" ]; then
|
||||
echo "ERROR: Timesketch containers already running."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Tweak for Elasticsearch
|
||||
echo "* Setting vm.max_map_count for Elasticsearch"
|
||||
sysctl -q -w vm.max_map_count=262144
|
||||
if [ -z "$(grep vm.max_map_count /etc/sysctl.conf)" ]; then
|
||||
echo "vm.max_map_count=262144" >> /etc/sysctl.conf
|
||||
fi
|
||||
|
||||
# Create dirs
|
||||
mkdir -p timesketch/{data/postgresql,data/elasticsearch,logs,etc,etc/timesketch,etc/timesketch/sigma/rules,upload}
|
||||
|
||||
echo -n "* Setting default config parameters.."
|
||||
POSTGRES_USER="timesketch"
|
||||
POSTGRES_PASSWORD="$(< /dev/urandom tr -dc A-Za-z0-9 | head -c 32 ; echo)"
|
||||
POSTGRES_ADDRESS="postgres"
|
||||
POSTGRES_PORT=5432
|
||||
SECRET_KEY="$(< /dev/urandom tr -dc A-Za-z0-9 | head -c 32 ; echo)"
|
||||
ELASTIC_ADDRESS="elasticsearch"
|
||||
ELASTIC_PORT=9200
|
||||
REDIS_ADDRESS="redis"
|
||||
REDIS_PORT=6379
|
||||
GITHUB_BASE_URL="https://raw.githubusercontent.com/google/timesketch/master"
|
||||
ELASTIC_MEM_USE_GB=$(cat /proc/meminfo | grep MemTotal | awk '{printf "%.0f", ($2 / 1000000 / 2)}')
|
||||
echo "OK"
|
||||
echo "* Setting Elasticsearch memory allocation to ${ELASTIC_MEM_USE_GB}GB"
|
||||
|
||||
# Docker compose and configuration
|
||||
echo -n "* Fetching configuration files.."
|
||||
curl -s $GITHUB_BASE_URL/docker/release/docker-compose.yml > timesketch/docker-compose.yml
|
||||
curl -s $GITHUB_BASE_URL/docker/release/config.env > timesketch/config.env
|
||||
|
||||
# Fetch default Timesketch config files
|
||||
curl -s $GITHUB_BASE_URL/data/timesketch.conf > timesketch/etc/timesketch/timesketch.conf
|
||||
curl -s $GITHUB_BASE_URL/data/tags.yaml > timesketch/etc/timesketch/tags.yaml
|
||||
curl -s $GITHUB_BASE_URL/data/plaso.mappings > timesketch/etc/timesketch/plaso.mappings
|
||||
curl -s $GITHUB_BASE_URL/data/features.yaml > timesketch/etc/timesketch/features.yaml
|
||||
curl -s $GITHUB_BASE_URL/data/sigma_config.yaml > timesketch/etc/timesketch/sigma_config.yaml
|
||||
curl -s $GITHUB_BASE_URL/data/sigma/rules/lnx_susp_zenmap.yml > timesketch/etc/timesketch/sigma/rules/lnx_susp_zenmap.yml
|
||||
curl -s $GITHUB_BASE_URL/contrib/nginx.conf > timesketch/etc/nginx.conf
|
||||
echo "OK"
|
||||
|
||||
# Create a minimal Timesketch config
|
||||
echo -n "* Edit configuration files.."
|
||||
sed -i 's#SECRET_KEY = \x27\x3CKEY_GOES_HERE\x3E\x27#SECRET_KEY = \x27'$SECRET_KEY'\x27#' timesketch/etc/timesketch/timesketch.conf
|
||||
|
||||
# Set up the Elastic connection
|
||||
sed -i 's#^ELASTIC_HOST = \x27127.0.0.1\x27#ELASTIC_HOST = \x27'$ELASTIC_ADDRESS'\x27#' timesketch/etc/timesketch/timesketch.conf
|
||||
sed -i 's#^ELASTIC_PORT = 9200#ELASTIC_PORT = '$ELASTIC_PORT'#' timesketch/etc/timesketch/timesketch.conf
|
||||
|
||||
# Set up the Redis connection
|
||||
sed -i 's#^UPLOAD_ENABLED = False#UPLOAD_ENABLED = True#' timesketch/etc/timesketch/timesketch.conf
|
||||
sed -i 's#^UPLOAD_FOLDER = \x27/tmp\x27#UPLOAD_FOLDER = \x27/usr/share/timesketch/upload\x27#' timesketch/etc/timesketch/timesketch.conf
|
||||
|
||||
sed -i 's#^CELERY_BROKER_URL =.*#CELERY_BROKER_URL = \x27redis://'$REDIS_ADDRESS':'$REDIS_PORT'\x27#' timesketch/etc/timesketch/timesketch.conf
|
||||
sed -i 's#^CELERY_RESULT_BACKEND =.*#CELERY_RESULT_BACKEND = \x27redis://'$REDIS_ADDRESS':'$REDIS_PORT'\x27#' timesketch/etc/timesketch/timesketch.conf
|
||||
|
||||
# Set up the Postgres connection
|
||||
sed -i 's#postgresql://<USERNAME>:<PASSWORD>@localhost#postgresql://'$POSTGRES_USER':'$POSTGRES_PASSWORD'@'$POSTGRES_ADDRESS':'$POSTGRES_PORT'#' timesketch/etc/timesketch/timesketch.conf
|
||||
|
||||
sed -i 's#^POSTGRES_PASSWORD=#POSTGRES_PASSWORD='$POSTGRES_PASSWORD'#' timesketch/config.env
|
||||
sed -i 's#^ELASTIC_MEM_USE_GB=#ELASTIC_MEM_USE_GB='$ELASTIC_MEM_USE_GB'#' timesketch/config.env
|
||||
|
||||
echo "\n* Turning off autostart.."
|
||||
sed -i '/restart: always/d' timesketch/docker-compose.yml
|
||||
|
||||
ln -s ./config.env ./timesketch/.env
|
||||
echo "OK"
|
||||
echo "* Installation done."
|
||||
|
||||
echo
|
||||
echo "Start the system:"
|
||||
echo "1. cd timesketch"
|
||||
echo "2. docker-compose up -d"
|
||||
echo "3. docker-compose exec timesketch-web tsctl add_user --username <USERNAME>"
|
||||
echo
|
||||
echo "WARNING: The server is running without encryption."
|
||||
echo "Follow the instructions to enable SSL to secure the communications:"
|
||||
echo "https://github.com/google/timesketch/blob/master/docs/Installation.md"
|
||||
echo
|
||||
echo
|
||||
Reference in New Issue
Block a user