install_ISIS_frontend.sh

#!/bin/bash

set -x

url="https://github.com/HazCod/ISIS-frontend.git"
user="isis"
dir="/home/$user/ISIS-frontend"
cron="*/1 * * * * /home/isis/ISIS-frontend/check_assignments.py &>/dev/null"

function isPackageInstalled() {
	 return sudo dpkg-query -l | grep $1 | wc -l
}


function installPackage() {
	sudo apt-get -q -y install $1
}
	
function valid_ip() {
    local  ip=$1
    local  stat=1

    if [[ $ip =~ ^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$ ]]; then
        OIFS=$IFS
        IFS='.'
        ip=($ip)
        IFS=$OIFS
        [[ ${ip[0]} -le 255 && ${ip[1]} -le 255 \
            && ${ip[2]} -le 255 && ${ip[3]} -le 255 ]]
        stat=$?
    fi
    return $stat
}

function installDependencies {
	export DEBIAN_FRONTEND=noninteractive
	sudo apt-get update
	installPackage git
	installPackage python
	installPackage python-mysqldb
	installPackage python-git
	installPackage libssl-dev
	installPackage iw
	installPackage python-imaging
	installPackage python-nmap
	installPackage python-netaddr
	installPackage wpasupplicant
	installPackage ettercap-text-only
	installPackage isc-dhcp-server
	installPackage python-netifaces
	installPackage nmap
}

function getFromGit {
	/usr/bin/git clone $url $dir
	sudo chown isis $dir
}

function checkDefArgs() {
	if ! valid_ip $1; then
		echo "Please provide a valid IP."
		exit 1
	fi
	def_ip=$1
	shift
	
	if ! valid_ip $2; then
		echo "Please provide a valid gateway IP.";
		exit 1
	fi
	def_gateway=$2
	shift

	if ! $3 ~= ^[0-255]{4}$; then
		echo "Please provide a valid netmask."
		exit 1
	fi
	def_netmask=$3
	shift

	if ! valid_ip $4; then
		echo "Please provide a valid dns IP.";
		exit 1
	fi
	def_dns=$4
	shift
}

function checkArgs() {
	if ! valid_ip $1; then
		echo "Please provide a valid IP."
		exit 1
	fi
	ip=$1
	
	if ! valid_ip $2; then
		echo "Please provide a valid gateway IP.";
		exit 1
	fi
	gateway=$2

	if [[ $3 =~ ^[0-255]{4}$ ]]; then
		echo "Please provide a valid netmask."
		exit 1
	fi
	netmask=$3

	if ! valid_ip $4; then
		echo "Please provide a valid dns IP.";
		exit 1
	fi
	dns=$4
}

function writeDefaultsIP() {
	cd /etc/network/
	echo "auto eth0" >> $1
	echo "iface eth0 inet static" >> $1
	echo "address $def_ip" >> $1
	echo "netmask $def_netmask" >> $1
	echo "gateway $def_gateway" >> $1
	echo "dns-nameservers $def_dns" >> $1
}

function writeIP() {
	sudo echo "auto lo" >> /etc/network/interfaces
	sudo echo "iface lo inet loopback" >> /etc/network/interfaces
	sudo echo "auto eth0" >> /etc/network/interfaces
	sudo echo "iface eth0 inet static" >> /etc/network/interfaces
	sudo echo "address $ip" >> /etc/network/interfaces
	sudo echo "netmask $netmask" >> /etc/network/interfaces
	sudo echo "gateway $gateway" >> /etc/network/interfaces
	sudo echo "dns-nameservers $dns" >> /etc/network/interfaces
}

function installIP {
	sudo -k
	cd /etc/network/
	mv interfaces interfaces.old
	writeIP
	sleep 5s #Silly bash, working faster then allowed.
	sudo service networking restart
	cd ~

}

function addToSudoers {
	sudo adduser isis sudo
	sudo echo "$user ALL=(ALL) NOPASSWD: ALL" >> /etc/sudoers
}

function chmodFiles {
	sudo chmod -R +x $dir
	#find $dir -type f -exec chmod +x {} \;
	sudo chown -R isis:isis $dir
	#find $dir -type f -exec chown isis {} \;
}

function setCron() {
	crontab -l > mycron
	echo "$cron" >> mycron
	sudo /usr/bin/crontab mycron
	rm mycron
}

function addUser {
	sudo adduser isis
}

function usage {
	echo "install_ISIS_frontend.sh";
	echo "----------------------------";
	echo "Usage: install_ISIS_frontend.sh [-d ip_address ip_gateway ip_netmask ip_dns] [ip_address ip_gateway ip_netmask ip_dns]";
	echo "";
	echo "Options:";
	echo "-d	default		If DHCP or given IP settings don't work, it falls back to the following IP settings.";
}

# SCRIPT BEGIN

if [ "$(id -u)" != "0" ]; then
	echo "Please run this script as root."
	exit 1
fi

# if there are any arguments
if [ $# > 1 ]; then 
	# if we are giving a default value
	if [ $1 = "-d" ]; then
		# we must have at least 5 values (-d, ip, gateway, netmask, dns) OR 9 values.
		if [ ! $# = 5 ] && ![ $# = 9 ]; then
			usage
			exit 1
		
		fi
		# 5 arguments, so dhcp with fallback defaults
		if [ $# = 5]; then
			shift			
			checkDefArgs $1 $2 $3 $4
		else
		# 9 arguments, so given value with fallback defaults.
			shift			
			checkDefArgs $1 $2 $3 $4
			checkArgs $1 $2 $3 $4
		fi	
	# else we need 4 values (ip, gateway, netmask, dns)
	elif [ $# = 4 ]; then
		checkArgs $1 $2 $3 $4
	else
		usage
		exit 1
	fi
fi

function installHostname(){
	sudo echo $1 > "/etc/hostname"
	sudo sed -i -e "s/raspberrypi/$1/" /etc/hosts
	sudo /etc/init.d/hostname.sh start
}

function installAircrack(){
	wget http://download.aircrack-ng.org/aircrack-ng-1.2-beta2.tar.gz
	tar -xzvf aircrack-ng-1.2-beta2.tar.gz
	cd aircrack-ng-1.2-beta2
	make
	sudo make install
	sudo airodump-ng-oui-update
	cd ..
	rm aircrack-ng-1.2-beta2.tar.gz
}

function installSslstrip(){
	wget http://www.thoughtcrime.org/software/sslstrip/sslstrip-0.9.tar.gz
	tar zxvf sslstrip-0.9.tar.gz
	cd sslstrip-0.9
	sudo python ./setup.py install
}

function createServersettings(){
	read -p "give the address of the server" address
	read -p "give the username of the server" server_username

	echo "#!/usr/bin/python">/home/isis/ISIS-frontend/server_settings.py
	echo "server_address=\""$address"\"">>/home/isis/ISIS-frontend/server_settings.py
	echo "server_username=\""$server_username"\"">>/home/isis/ISIS-frontend/server_settings.py
	echo "">>/home/isis/ISIS-frontend/server_settings.py
	
	read -p "enter the name of the database" databaseName
	read -p "enter the username of the database" databaseUser
	read -p "enter the password for the database" databasePassword

	echo "database_name=\""$databaseName"\"">>/home/isis/ISIS-frontend/server_settings.py
	echo "database_user=\""$databaseUser"\"">>/home/isis/ISIS-frontend/server_settings.py
	echo "database_password=\""$databasePassword"\"">>/home/isis/ISIS-frontend/server_settings.py
}

function copy_ssh(){
	python $dir/copy_ssh.py
}

read -p "What hostname/ID should be given to this unit? This must be unique!" host
cd ~
addUser
installIP
installDependencies
getFromGit
installHostname $host
sudo service networking restart
addToSudoers
createServersettings
chmodFiles
setCron
cd ~
installAircrack
installSslstrip
copy_ssh
su isis
# SCRIPT END