crypto = include_lib("/lib/crypto.so")

printInfo = function(message)
print("<color=#1C3879>[*] " + message + "</color>")
end function 
printError = function(message)
print("<color=#FF0000>[-] " + message + "</color>")
end function 
printSuccess = function(message)
print("<color=#7DCE13>[+] " + message + "</color>")
end function 

printBanner = function(bool)
banners = ["  _______  ______  _______  __ __   __"," |__   __||  ____||__   __|/_ |\ \ / /","    | |   | |__      | |    | | \ V / ","    | |   |  __|     | |    | |  > <  ","    | |   | |____    | |    | | / . \ ","    |_|   |______|   |_|    |_|/_/ \_\"]
for banner in banners
	print("<color=#607EAA>" + banner + "</color>")
end for
print("<color=#607EAA>			   v1.0.0 - h1dr4</color>")
print(char(10) * 1.5)                                      
end function

if not crypto then
	printError("Crypto not found.")
	return
end if

aireplayWifi = function(net)
acks = net.signal.replace("%","").to_int
reqacks = 300000 / acks + 100
crypto.airmon("start","wlan0")
printInfo("Collecting " + reqacks + " ACKs...")
print("\n")
crypto.aireplay(net.bssid,net.essid,reqacks)
pass = crypto.aircrack(current_path + "/file.cap")
printSuccess("WiFi Cracked! Password: " + pass)
file = get_shell.host_computer.File(current_path + "/file.cap")
file.delete()
crypto.airmon("stop","wlan0")
return pass
end function

netHandler = function(net)
printInfo("Now handling WiFi: " + net.essid)
while true
input = user_input("<color=#607EAA>net > ")
args = input.split(" ")
if input == "attack" then
	aireplayWifi(net)
else if input == "exit" then
	main(true)
end if
end while
end function

main = function(boolean)
if not boolean then
	clear_screen()
	printBanner()
end if
printInfo("Loading WiFi networks...")
networks = get_shell.host_computer.wifi_networks("wlan0")
formatted = []
for network in networks
	bssid = network.split(" ")[0]
	essid = network.split(" ")[2]
	pwr = network.split(" ")[1]
	id = networks.indexOf(network)
	id = id + 1
	formatted.push({"id":id,"bssid":bssid,"essid":essid,"signal":pwr})
	printSuccess("Loaded " + essid + " with ID " + id)
end for
while true
input = user_input("<color=#607EAA>tet1x > ")
args = input.split(" ")
if args[0] == "interact" then
	if not args.len > 1 then
		printError("Please enter an ID.")
		continue
	end if
    id = args[1].to_int - 1
    netHandler(formatted[id])
end if
end while
end function
main()