From 433d30861d6a51208bb36a7172a30577c4ba7011 Mon Sep 17 00:00:00 2001 From: derv82 Date: Thu, 18 May 2017 01:58:08 -0400 Subject: [PATCH] Exit gracefully when run as non-root Patch for pull request #33 --- Wifite.py | 3 +-- py/Configuration.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) 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()