class AirHost(object):
def __init__(self, hostapd_config_path, dnsmasq_config_path):
# Mandatory configuration files to start the access point successfully
if (dnsmasq_config_path == None or hostapd_config_path == None):
raise MissingConfigurationFileException(
"dnsmasq and hostapd configuration files must be specified\n \
either by argument in the command line or in the etf.conf file"
)
self.aplauncher = APLauncher(hostapd_config_path)
self.dnsmasqhandler = DNSMasqHandler(dnsmasq_config_path)
self.running_interface = None
self.plugins = []
def add_plugin(self, airhost_plugin):
self.plugins.append(airhost_plugin)
def start_access_point(self, interface):
print "[+] Killing already started processes and restarting network services"
self.stop_access_point(
False
) # Restarting services helps avoiding some conflicts with dnsmasq
for plugin in self.plugins:
plugin.start()
self.aplauncher.start_access_point(interface)
if not self.dnsmasqhandler.start_dnsmasq():
print "[-] Error starting dnsmasq, aborting AP launch"
return False
self.running_interface = interface
print "[+] Access Point launched successfully"
return True
def stop_access_point(self, free_plugins=True):
print "[+] Stopping dnsmasq and hostapd services"
self.aplauncher.stop_access_point()
self.dnsmasqhandler.stop_dnsmasq()
self.running_interface = None
print "[+] Access Point stopped, restarting network services..."
os.system('service networking restart')
os.system('service NetworkManager restart')
if free_plugins:
for plugin in self.plugins:
plugin.restore()
del self.plugins[:]
def is_running(self):
return self.running_interface != None
def cleanup(self):
self.dnsmasqhandler.cleanup()
self.aplauncher.cleanup()
class AirHost(object):
"""
The AirHost class.
Receives the hostapd and dnsmasq config path
to configure and launch the services.
"""
def __init__(self, hostapd_config_path, dnsmasq_config_path):
"""The AirHost module constructor."""
# Mandatory configuration files to start the access point successfully
if (dnsmasq_config_path is None or hostapd_config_path is None):
raise MissingConfigurationFileException("dnsmasq and hostapd configuration files must be specified\n \
either by argument in the command line or in the etf.conf file")
self.aplauncher = APLauncher(hostapd_config_path)
self.dnsmasqhandler = DNSMasqHandler(dnsmasq_config_path)
self.running_interface = None
self.plugins = []
def add_plugin(self, airhost_plugin):
"""Add an AirHostPlugin to this module instance."""
self.plugins.append(airhost_plugin)
def start_access_point(self, interface, print_credentials):
"""
This method starts an access point on the specified interface.
It assumes that the interface is umanaged.
It also starts the DHCP and DNS servers with dnsmasq
alongside the access point so it is instantly fully functional.
Access Point configurations are read by the APLauncher.
The print_credentials flag is used by the APLauncher
to print out EAP credentials caught by hostapd-wpe.
"""
print "[+] Killing already started processes and restarting network services"
# Restarting services helps avoiding some conflicts with dnsmasq
self.stop_access_point(False)
print "[+] Running airhost plugins pre_start"
for plugin in self.plugins:
plugin.pre_start()
self.aplauncher.print_creds = print_credentials
self.aplauncher.start_access_point(interface)
if not self.dnsmasqhandler.start_dnsmasq():
print "[-] Error starting dnsmasq, aborting AP launch"
return False
print "[+] Running airhost plugins post_start"
for plugin in self.plugins:
plugin.post_start()
self.running_interface = interface
print "[+] Access Point launched successfully"
return True
def stop_access_point(self, free_plugins = True):
"""This method cleanly stops the access point and all its background services."""
print "[+] Stopping dnsmasq and hostapd services"
self.aplauncher.stop_access_point()
self.dnsmasqhandler.stop_dnsmasq()
self.running_interface = None
print "[+] Access Point stopped..."
for plugin in self.plugins:
plugin.stop()
if free_plugins and len(self.plugins) > 0:
for plugin in self.plugins:
plugin.restore()
del self.plugins[:]
print "[+] Cleared Airhost plugins"
def is_running(self):
"""Checks if the access point is active."""
return self.running_interface is not None
def cleanup(self):
"""Cleanup method for the airhost module."""
self.dnsmasqhandler.cleanup()
self.aplauncher.cleanup()
class AirHost(object):
"""
The AirHost class.
Receives the hostapd and dnsmasq config path
to configure and launch the services.
"""
def __init__(self, config):
"""The AirHost module constructor."""
# Mandatory configuration files to start the access point successfully
if (not isfile(config["dnsmasq_conf"]) or not isfile(config["hostapd_conf"])):
raise MissingConfigurationFileException("dnsmasq and hostapd configuration files must be specified\n \
either by argument in the command line or in the etf.conf file")
self.aplauncher = APLauncher(config["hostapd_conf"])
self.dnsmasqhandler = DNSMasqHandler(config["dnsmasq_conf"])
self.running_interface = None
self.plugins = []
def add_plugin(self, airhost_plugin):
"""Add an AirHostPlugin to this module instance."""
self.plugins.append(airhost_plugin)
def start_access_point(self, interface, print_credentials):
"""
This method starts an access point on the specified interface.
It assumes that the interface is umanaged.
It also starts the DHCP and DNS servers with dnsmasq
alongside the access point so it is instantly fully functional.
Access Point configurations are read by the APLauncher.
The print_credentials flag is used by the APLauncher
to print out EAP credentials caught by hostapd-wpe.
"""
SessionManager().log_event(NeutralEvent("Starting AirHost module."))
print "[+] Killing already started processes and restarting network services"
# Restarting services helps avoiding some conflicts with dnsmasq
self.stop_access_point(False)
print "[+] Running airhost plugins pre_start"
for plugin in self.plugins:
plugin.pre_start()
self.aplauncher.print_creds = print_credentials
self.aplauncher.start_access_point(interface)
if not self.dnsmasqhandler.start_dnsmasq():
SessionManager().log_event(UnsuccessfulEvent(
"Error starting dnsmasq. AirHost module start was aborted."), True)
self.stop_access_point()
return False
self.running_interface = interface
print "[+] Running airhost plugins post_start"
for plugin in self.plugins:
plugin.post_start()
print "[+] Access Point launched successfully"
SessionManager().log_event(SuccessfulEvent("AirHost module started successfully."))
return True
def stop_access_point(self, free_plugins = True):
"""This method cleanly stops the access point and all its background services."""
for plugin in self.plugins:
SessionManager().log_event(NeutralEvent("Running pre_stop method for '{}' plugin.".format(plugin)))
plugin.pre_stop()
print "[+] Stopping dnsmasq and hostapd services"
self.aplauncher.stop_access_point()
self.dnsmasqhandler.stop_dnsmasq()
self.running_interface = None
SessionManager().log_event(NeutralEvent("AirHost module stopped."))
print "[+] Access Point stopped..."
for plugin in self.plugins:
SessionManager().log_event(NeutralEvent("Running post_stop method for '{}' plugin.".format(plugin)))
plugin.post_stop()
if free_plugins and len(self.plugins) > 0:
for plugin in self.plugins:
SessionManager().log_event(NeutralEvent(
"Running restore method for '{}' plugin.".format(plugin)))
plugin.restore()
del self.plugins[:]
print "[+] Cleared Airhost plugins"
def is_running(self):
"""Checks if the access point is active."""
return self.running_interface is not None
def cleanup(self):
"""Cleanup method for the airhost module."""
self.dnsmasqhandler.cleanup()
self.aplauncher.cleanup()