Exit gracefully when run as non-root

Patch for pull request #33
This commit is contained in:
derv82
2017-05-18 01:58:08 -04:00
parent df7167a945
commit 433d30861d
2 changed files with 2 additions and 3 deletions

View File

@@ -23,8 +23,7 @@ class Wifite(object):
if os.getuid() != 0: if os.getuid() != 0:
Color.pl('{!} {R}error: {O}wifite{R} must be run as {O}root{W}') Color.pl('{!} {R}error: {O}wifite{R} must be run as {O}root{W}')
Color.pl('{!} {O}re-run as: sudo ./Wifite.py{W}') Color.pl('{!} {O}re-run as: sudo ./Wifite.py{W}')
exit(0) Configuration.exit_gracefully(0)
return
Configuration.initialize(load_interface=False) Configuration.initialize(load_interface=False)

View File

@@ -298,7 +298,7 @@ class Configuration(object):
Configuration.delete_temp() Configuration.delete_temp()
Macchanger.reset_if_changed() Macchanger.reset_if_changed()
from Airmon import Airmon 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.stop(Configuration.interface)
Airmon.put_interface_up(Airmon.base_interface) Airmon.put_interface_up(Airmon.base_interface)
Airmon.start_network_manager() Airmon.start_network_manager()