1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
|
#!/bin/bash
set -e
umask 0002
# Allow user to specify custom CMD
if [[ "$1" == "wfwrapper" ]]; then
# Rewrite CMD args to remove the explicit command,
set -- "${@:2}"
else
# Run whatever command the user wanted
exec "$@"
fi
if [[ "x$TRACKALERT" == "x" ]]; then
WFORCE_CMD="/usr/bin/wforce -D"
if [[ "$WFORCE_VERBOSE" -eq "1" ]]; then
WFORCE_CMD="$WFORCE_CMD -v"
fi
# Check if they just supplied their own wforce.conf file
if [[ -f /etc/wforce/wforce.conf ]]; then
echo "Using supplied wforce config file /etc/wforce/wforce.conf"
WFORCE_CONFIG_FILE=/etc/wforce/wforce.conf
fi
if [[ "x$WFORCE_CONFIG_FILE" == "x" ]]; then
if [[ "x$WFORCE_HTTP_PASSWORD" == "x" ]]; then
2>&1 echo "WFORCE_HTTP_PASSWORD environment variable must be set"
else
echo "Note you are using the default config file - to take full advantage of the capabilities of weakforced, you should specify a custom config file with the WFORCE_CONFIG_FILE environment variable or just replace /etc/wforce/wforce.conf with your own config file."
MYKEY=`cat /proc/sys/kernel/random/uuid | sed 's/[-]//g' | head -c 32 | base64`
export WFORCE_KEY="setKey(\"$MYKEY\")"
create_config.sh /etc/wforce/wforce.conf.j2 /etc/wforce/wforce.conf
echo "Starting $WFORCE_CMD"
exec $WFORCE_CMD
fi
else
WFORCE_CMD="$WFORCE_CMD -C $WFORCE_CONFIG_FILE"
exec $WFORCE_CMD
fi
else
TRACKALERT_CMD="/usr/bin/trackalert -D"
if [[ "$WFORCE_VERBOSE" -eq "1" ]]; then
TRACKALERT_CMD="$TRACKALERT_CMD -v"
fi
# Check if they just supplied their own trackalert.conf file
if [[ -f /etc/wforce/trackalert.conf ]]; then
echo "Using supplied trackalert config file /etc/wforce/trackalert.conf"
TRACKALERT_CONFIG_FILE=/etc/wforce/trackalert.conf
fi
if [[ "x$TRACKALERT_CONFIG_FILE" == "x" ]]; then
if [[ "x$TRACKALERT_HTTP_PASSWORD" == "x" ]]; then
2>&1 echo "TRACKALERT_HTTP_PASSWORD environment variable must be set"
else
echo "Note you are using the default config file - to take full advantage of the capabilities of trackalert, you should specify a custom config file with the TRACKALERT_CONFIG_FILE environment variable or just replace /etc/wforce/trackalert.conf with your own config file."
MYKEY=`cat /proc/sys/kernel/random/uuid | sed 's/[-]//g' | head -c 32 | base64`
export TRACKALERT_KEY="setKey(\"$MYKEY\")"
create_config.sh /etc/wforce/trackalert.conf.j2 /etc/wforce/trackalert.conf
echo "Starting $TRACKALERT_CMD"
exec $TRACKALERT_CMD
fi
else
TRACKALERT_CMD="$TRACKALERT_CMD -C $TRACKALERT_CONFIG_FILE"
exec $TRACKALERT_CMD
fi
fi
exit 1
|