diff --git a/Wifite.py b/Wifite.py index 70b2572..c9cf2e9 100755 --- a/Wifite.py +++ b/Wifite.py @@ -23,8 +23,7 @@ class Wifite(object): if os.getuid() != 0: Color.pl('{!} {R}error: {O}wifite{R} must be run as {O}root{W}') Color.pl('{!} {O}re-run as: sudo ./Wifite.py{W}') - exit(0) - return + Configuration.exit_gracefully(0) Configuration.initialize(load_interface=False) diff --git a/py/Configuration.py b/py/Configuration.py index e9a9f0e..f64671d 100644 --- a/py/Configuration.py +++ b/py/Configuration.py @@ -298,7 +298,7 @@ class Configuration(object): Configuration.delete_temp() Macchanger.reset_if_changed() from Airmon import Airmon - if Configuration.interface is not None and Airmon.base_interface is not None: + if hasattr(Configuration, "interface") and Configuration.interface is not None and Airmon.base_interface is not None: Airmon.stop(Configuration.interface) Airmon.put_interface_up(Airmon.base_interface) Airmon.start_network_manager()