diff --git a/defender.py b/defender.py index 8204dd5..06f4ecb 100644 --- a/defender.py +++ b/defender.py @@ -56,7 +56,8 @@ def check_http_post_attack(request): if __name__ == "__main__": - ip = ni.ifaddresses('br-abfe242530cd')[ni.AF_INET][0]['addr'] + network_interface = "br-abfe242530cd" # CHANGE THIS TO THE CONNECTED NETWORK INTERFACE + ip = ni.ifaddresses(network_interface)[ni.AF_INET][0]['addr'] url = "http://{0}:5000".format(ip) HOST = "localhost" @@ -79,7 +80,7 @@ if __name__ == "__main__": check_availability(url) check_http_post_attack(response) check_pragma_attack(response) - pkts = sniff(iface="br-abfe242530cd",prn=lambda x:x.sprintf("{IP:%IP.src% -> %IP.dst%\n}{IP:%IP.ttl%\n}{Raw:%Raw.load%\n}")) + pkts = sniff(iface=network_interface,prn=lambda x:x.sprintf("{IP:%IP.src% -> %IP.dst%\n}{IP:%IP.ttl%\n}{Raw:%Raw.load%\n}"))