ail-framework/reset_AIL.sh

171 lines
3.4 KiB
Bash
Raw Permalink Normal View History

2017-08-22 15:16:06 +00:00
#!/bin/bash
RED="\\033[1;31m"
DEFAULT="\\033[0;39m"
GREEN="\\033[1;32m"
[ -z "$AIL_HOME" ] && echo "Needs the env var AIL_HOME. Run the script from the virtual environment." && exit 1;
function helptext {
echo -e $GREEN"
.o. ooooo ooooo
.888. \`888' \`888'
.8\"888. 888 888
.8' \`888. 888 888
.88ooo8888. 888 888
.8' \`888. 888 888 o
o88o o8888o o o888o o o888ooooood8
Analysis Information Leak framework
"$DEFAULT"
Use this script to reset AIL (DB + stored items):
Usage:
-----
reset_AIL.sh
[--softReset] Keep All users accounts
[-h | --help] Help
"
}
function reset_dir {
2019-07-30 13:21:27 +00:00
# Access dirs and delete
cd $AIL_HOME
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
# Kill all screens
screen -ls | grep Detached | cut -d. -f1 | awk '{print $1}' | xargs kill
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
# Access dirs and delete
cd $AIL_HOME
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
if [ -d indexdir/ ]; then
pushd indexdir/
rm -r *
echo 'cleaned indexdir'
popd
fi
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
if [ -d logs/ ]; then
pushd logs/
rm *
echo 'cleaned logs'
popd
fi
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
if [ -d PASTES/ ]; then
pushd PASTES/
rm -r *
echo 'cleaned PASTES'
popd
fi
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
if [ -d HASHS/ ]; then
pushd HASHS/
rm -r *
echo 'cleaned HASHS'
popd
fi
2017-08-22 15:16:06 +00:00
if [ -d CRAWLED_SCREENSHOT/ ]; then
pushd CRAWLED_SCREENSHOT/
2019-07-30 13:21:27 +00:00
rm -r *
echo 'cleaned CRAWLED_SCREENSHOT'
2019-07-30 13:21:27 +00:00
popd
fi
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
if [ -d temp/ ]; then
pushd temp/
rm -r *
echo 'cleaned temp'
popd
fi
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
if [ -d var/www/submitted/ ]; then
pushd var/www/submitted
rm -r *
echo 'cleaned submitted'
popd
fi
2017-08-22 15:16:06 +00:00
2019-07-30 13:21:27 +00:00
echo -e $GREEN"* AIL has been reset *"$DEFAULT
}
2019-07-30 13:58:11 +00:00
function flush_DB_keep_user {
bash ${AIL_BIN}LAUNCH.sh -lkv &
2019-07-30 14:04:42 +00:00
wait
echo ""
2019-07-30 13:58:11 +00:00
pushd redis/src
./redis-cli -p 6383 -a ail_correls FLUSHDB;
./redis-cli -p 6383 -a ail_crawlers FLUSHDB;
./redis-cli -p 6383 -a ail_dups FLUSHDB;
./redis-cli -p 6383 -a ail_objs FLUSHDB;
./redis-cli -p 6383 -a ail_stats FLUSHDB;
./redis-cli -p 6383 -a ail_tags FLUSHDB;
./redis-cli -p 6383 -a ail_trackers FLUSHDB;
echo "KVROCKS FLUSHED"
2019-07-30 13:58:11 +00:00
popd
2019-07-30 14:04:42 +00:00
bash ${AIL_BIN}LAUNCH.sh -k
2019-07-30 13:58:11 +00:00
}
function validate_reset {
echo -e $RED"WARNING: DELETE AIL DATA"$DEFAULT
# Make sure the reset is intentional
num=$(( ( RANDOM % 100 ) + 1 ))
echo -e $RED"To reset the platform, enter the following number: "$DEFAULT $num
read userInput
if [ $userInput -eq $num ]
then
echo "Resetting AIL..."
else
echo "Wrong number"
exit 1;
fi
}
2019-07-30 14:12:51 +00:00
function soft_reset {
validate_reset;
reset_dir;
flush_DB_keep_user;
}
2019-07-30 13:21:27 +00:00
#If no params,
[[ $@ ]] || {
validate_reset;
2019-07-30 13:21:27 +00:00
num=$(( ( RANDOM % 100 ) + 1 ))
echo -e $RED"If you want to delete the DB , enter the following number: "$DEFAULT $num
2019-07-30 13:21:27 +00:00
read userInput
reset_dir;
2019-07-30 13:21:27 +00:00
if [ $userInput -eq $num ]
then
if [ -d DATA_KVROCKS/ ]; then
pushd DATA_KVROCKS/
2019-07-30 13:21:27 +00:00
rm -r *
echo 'cleaned DATA_KVROCKS'
2019-07-30 13:21:27 +00:00
popd
fi
fi
echo -e $GREEN"* AIL has been reset *"$DEFAULT
exit
}
while [ "$1" != "" ]; do
case $1 in
2019-07-30 14:12:51 +00:00
--softReset ) soft_reset;
2019-07-30 13:58:11 +00:00
;;
-h | --help ) helptext;
exit
;;
2019-07-30 13:21:27 +00:00
* ) exit 1
esac
shift
done