|
@@ -0,0 +1,122 @@
|
|
|
+#!/bin/bash
|
|
|
+
|
|
|
+BASE="./tests"
|
|
|
+
|
|
|
+usage() {
|
|
|
+ echo "Usage:"
|
|
|
+ echo " ./wizard.sh -h Display this help message."
|
|
|
+ echo " ./test_env.sh -d ./tests Create test environment in './tests' folder"
|
|
|
+ exit 0
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+while [[ $# -gt 0 ]]
|
|
|
+do
|
|
|
+ key="${1}"
|
|
|
+ case ${key} in
|
|
|
+ -d|--directory)
|
|
|
+ BASE=${2}
|
|
|
+ shift #past argument
|
|
|
+ shift
|
|
|
+ ;;
|
|
|
+ -h|--help)
|
|
|
+ usage
|
|
|
+ exit 0
|
|
|
+ ;;
|
|
|
+ *) # unknown option
|
|
|
+ log_err "Unknown argument ${key}."
|
|
|
+ usage
|
|
|
+ exit 1
|
|
|
+ ;;
|
|
|
+ esac
|
|
|
+done
|
|
|
+
|
|
|
+BASE=$(realpath $BASE)
|
|
|
+
|
|
|
+DATA_DIR="$BASE/data"
|
|
|
+
|
|
|
+LOG_DIR="$BASE/logs/"
|
|
|
+
|
|
|
+CONFIG_DIR="$BASE/config"
|
|
|
+CONFIG_FILE="$BASE/dev.yaml"
|
|
|
+CSCLI_DIR="$CONFIG_DIR/crowdsec-cli"
|
|
|
+PARSER_DIR="$CONFIG_DIR/parsers"
|
|
|
+PARSER_S00="$PARSER_DIR/s00-raw"
|
|
|
+PARSER_S01="$PARSER_DIR/s01-parse"
|
|
|
+PARSER_S02="$PARSER_DIR/s02-enrich"
|
|
|
+SCENARIOS_DIR="$CONFIG_DIR/scenarios"
|
|
|
+POSTOVERFLOWS_DIR="$CONFIG_DIR/postoverflows"
|
|
|
+HUB_DIR="$CONFIG_DIR/hub"
|
|
|
+PLUGINS="http slack splunk"
|
|
|
+PLUGINS_DIR="plugins"
|
|
|
+NOTIF_DIR="notifications"
|
|
|
+
|
|
|
+log_info() {
|
|
|
+ msg=$1
|
|
|
+ date=$(date +%x:%X)
|
|
|
+ echo -e "[$date][INFO] $msg"
|
|
|
+}
|
|
|
+
|
|
|
+create_arbo() {
|
|
|
+ mkdir -p "$BASE"
|
|
|
+ mkdir -p "$DATA_DIR"
|
|
|
+ mkdir -p "$LOG_DIR"
|
|
|
+ mkdir -p "$CONFIG_DIR"
|
|
|
+ mkdir -p "$PARSER_DIR"
|
|
|
+ mkdir -p "$PARSER_S00"
|
|
|
+ mkdir -p "$PARSER_S01"
|
|
|
+ mkdir -p "$PARSER_S02"
|
|
|
+ mkdir -p "$SCENARIOS_DIR"
|
|
|
+ mkdir -p "$POSTOVERFLOWS_DIR"
|
|
|
+ mkdir -p "$CSCLI_DIR"
|
|
|
+ mkdir -p "$HUB_DIR"
|
|
|
+ mkdir -p $CONFIG_DIR/$NOTIF_DIR/$plugin
|
|
|
+ mkdir -p $BASE/$PLUGINS_DIR
|
|
|
+}
|
|
|
+
|
|
|
+copy_files() {
|
|
|
+ cp "./config/profiles.yaml" "$CONFIG_DIR"
|
|
|
+ cp "./config/simulation.yaml" "$CONFIG_DIR"
|
|
|
+ cp "./cmd/crowdsec/crowdsec" "$BASE"
|
|
|
+ cp "./cmd/crowdsec-cli/cscli" "$BASE"
|
|
|
+ cp -r "./config/patterns" "$CONFIG_DIR"
|
|
|
+ cp "./config/acquis.yaml" "$CONFIG_DIR"
|
|
|
+ touch "$CONFIG_DIR"/local_api_credentials.yaml
|
|
|
+ touch "$CONFIG_DIR"/online_api_credentials.yaml
|
|
|
+ envsubst < "./config/dev.yaml" > $BASE/dev.yaml
|
|
|
+ for plugin in $PLUGINS
|
|
|
+ do
|
|
|
+ cp $PLUGINS_DIR/$NOTIF_DIR/$plugin/notification-$plugin $BASE/$PLUGINS_DIR/notification-$plugin
|
|
|
+ cp $PLUGINS_DIR/$NOTIF_DIR/$plugin/$plugin.yaml $CONFIG_DIR/$NOTIF_DIR/$plugin.yaml
|
|
|
+ done
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+setup() {
|
|
|
+ $BASE/cscli -c "$CONFIG_FILE" hub update
|
|
|
+ $BASE/cscli -c "$CONFIG_FILE" collections install crowdsecurity/linux
|
|
|
+}
|
|
|
+
|
|
|
+setup_api() {
|
|
|
+ $BASE/cscli -c "$CONFIG_FILE" machines add test -p testpassword -f $CONFIG_DIR/local_api_credentials.yaml --force
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+main() {
|
|
|
+ log_info "Creating test arboresence in $BASE"
|
|
|
+ create_arbo
|
|
|
+ log_info "Arboresence created"
|
|
|
+ log_info "Copying needed files for tests environment"
|
|
|
+ copy_files
|
|
|
+ log_info "Files copied"
|
|
|
+ log_info "Setting up configurations"
|
|
|
+ CURRENT_PWD=$(pwd)
|
|
|
+ cd $BASE
|
|
|
+ setup_api
|
|
|
+ setup
|
|
|
+ cd $CURRENT_PWD
|
|
|
+ log_info "Environment is ready in $BASE"
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+main
|