84 lines
2.5 KiB
Bash
Executable File
84 lines
2.5 KiB
Bash
Executable File
#!/bin/sh
|
|
|
|
# Exit Codes: 0 = Success, 1 = Not Authorized, 2 = Auth File Not Found, 3 = Token Removal Failed
|
|
|
|
EXIT_CODE=0
|
|
AUTH_FILE="/tmp/quecmanager/auth_success"
|
|
|
|
cleanup() {
|
|
MAX_AGE=$((2 * 3600)) # 2 hours in seconds
|
|
NOW_TIME=$(date +%s)
|
|
TMP_FILE=$(mktemp)
|
|
# AUTH_FILE cleanup process, Remove any token lines older than 2 hours from AUTH_FILE
|
|
if [ -f $AUTH_FILE ]; then
|
|
while read -r line; do
|
|
if [ -n "$(echo "$line" | tr -d '[:space:]')" ]; then
|
|
# Extract the date from the line and convert it to a timestamp
|
|
TOKEN_DATE=$(echo "$line" | awk '{print $1}' | sed 's/T/ /')
|
|
TOKEN_TIME=$(date -d "$TOKEN_DATE" +%s 2>/dev/null)
|
|
# If date is valid and not older than MAX_AGE, keep the line
|
|
if [ -n "$TOKEN_TIME" ] && [ $((NOW_TIME - TOKEN_TIME)) -le $MAX_AGE ]; then
|
|
echo "$line" >> "$TMP_FILE"
|
|
fi
|
|
fi
|
|
done < "$AUTH_FILE"
|
|
mv "$TMP_FILE" "$AUTH_FILE"
|
|
fi
|
|
}
|
|
|
|
removeToken() {
|
|
TOKEN=$1
|
|
# Remove token from file
|
|
if [ -f $AUTH_FILE ] && [ -n "${TOKEN}" ]; then
|
|
sed -i -e "s/.*${TOKEN}.*//g" ${AUTH_FILE} 2>/dev/null
|
|
echo '{"state":"success", "message":"Logged out successfully"}'
|
|
EXIT_CODE=0
|
|
else
|
|
echo '{"state":"failed", "message":"Token Removal Failed"}'
|
|
EXIT_CODE=3
|
|
fi
|
|
}
|
|
process() {
|
|
if [ -n "$1" ]; then
|
|
TOKEN=$1
|
|
else
|
|
TOKEN=$(head -c 16 /dev/urandom | hexdump -v -e '/1 "%02x"')
|
|
touch ${AUTH_FILE}
|
|
echo "$(date +"%Y-%m-%dT%H:%M:%S") ${TOKEN}" >> ${AUTH_FILE}
|
|
echo "" >> ${AUTH_FILE}
|
|
fi
|
|
|
|
if [ ! -f $AUTH_FILE ]; then
|
|
echo '{"state":"failed", "message":"Authentication file not found"}'
|
|
EXIT_CODE=2
|
|
fi
|
|
|
|
if [ $EXIT_CODE -eq 0 ] && ( [ -z "$TOKEN" ] || [ "$TOKEN" = "" ] || [ $(grep "${TOKEN}" "${AUTH_FILE}" | wc -l) -eq 0 ] ); then
|
|
echo "{\"response\": { \"status\": \"error\", \"raw_output\": \"Not Authorized\" }, \"command\": {\"timestamp\": \"$(date +%Y%m%d'T'%H%M%S)\"}, \"error\":\"Not Authorized\"}"
|
|
EXIT_CODE=1
|
|
fi
|
|
|
|
if [ $EXIT_CODE -eq 0 ] && grep -q "$TOKEN" "$AUTH_FILE"; then
|
|
echo "{\"state\":\"success\", \"token\":\"$TOKEN\"}"
|
|
EXIT_CODE=0
|
|
fi
|
|
|
|
}
|
|
case $1 in
|
|
removeToken)
|
|
removeToken $2
|
|
;;
|
|
cleanup)
|
|
cleanup
|
|
;;
|
|
process)
|
|
cleanup
|
|
process $2
|
|
;;
|
|
*)
|
|
cleanup
|
|
process $1
|
|
;;
|
|
esac
|
|
|
|
exit $EXIT_CODE |