crowdsec/scripts/test_env.sh

119 lines
2.4 KiB
Bash
Raw Normal View History

2020-05-15 09:39:16 +00:00
#!/bin/bash
BASE="./tests"
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"
2020-05-27 14:31:08 +00:00
CONFIG_FILE="$BASE/dev.yaml"
2020-05-15 09:39:16 +00:00
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"
PLUGIN_BACKEND_DIR="$CONFIG_DIR/plugins/backend/"
DB_PLUGIN_FILE="$PLUGIN_BACKEND_DIR/database.yaml"
2020-05-15 09:39:16 +00:00
gen_sqlite_config() {
echo "name: database" >> "$DB_PLUGIN_FILE"
echo "path: ./plugins/backend/database.so" >> "$DB_PLUGIN_FILE"
echo "config:" >> "$DB_PLUGIN_FILE"
echo " type: sqlite" >> "$DB_PLUGIN_FILE"
echo " db_path: ./test.db" >> "$DB_PLUGIN_FILE"
echo " flush: true" >> "$DB_PLUGIN_FILE"
2020-05-15 09:39:16 +00:00
}
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 "$PLUGIN_BACKEND_DIR"
}
copy_files() {
cp "./config/profiles.yaml" "$CONFIG_DIR"
cp "./config/dev.yaml" "$BASE"
cp "./config/simulation.yaml" "$CONFIG_DIR"
2020-05-15 09:39:16 +00:00
cp "./cmd/crowdsec/crowdsec" "$BASE"
cp "./cmd/crowdsec-cli/cscli" "$BASE"
cp -r "./config/patterns" "$CONFIG_DIR"
cp -r "./data/" "$BASE"
cp -r "./plugins/" "$BASE"
}
setup() {
2020-05-27 14:31:08 +00:00
$BASE/cscli -c "$CONFIG_FILE" update
$BASE/cscli -c "$CONFIG_FILE" install collection crowdsecurity/linux
2020-05-15 09:39:16 +00:00
}
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"
2020-05-27 14:31:08 +00:00
CURRENT_PWD=$(pwd)
cd $BASE
2020-05-15 09:39:16 +00:00
setup
2020-05-27 14:31:08 +00:00
cd $CURRENT_PWD
2020-05-15 09:39:16 +00:00
gen_sqlite_config
log_info "Environment is ready in $BASE"
}
usage() {
echo "Usage:"
echo " ./wizard.sh -h Display this help message."
echo " ./env_test.sh -d ./tests Create test environment in './tests' folder"
exit 0
}
main