Merge pull request #304 from Veil-Framework/update_sbrun_patch

Patch provided by @sbrun to help with kali compatibility

CHANGELOG

@@ -1,3 +1,7 @@
+[04.01.2016]
+     Released.: 2.26.1
+     Updated..: Our update.py script has been updated for better compatibility with Kali.  Thanks to @Sbrun for providing the patch!
+
 [03.31.2016]
      Released.: 2.26
      Fixed....: Fixed command injection vulnerability because @botnet_hunter supplied us with a fix that worked against all of his tests.  Thanks to @botnet_hunter for helping to remediate this issue!

config/update.py

@@ -1,6 +1,6 @@
 #!/usr/bin/python
 
-import platform, os, sys
+import platform, os, sys, pwd
 
 """
 
@@ -200,15 +200,11 @@ def generateConfig(options):
 
         # Get the real user if we're being ran under sudo
         wineprefix = ""
-        try:
-            user = os.environ.get("SUDO_USER")
-            if user == 'root':
-                wineprefix = "/root/.config/wine/veil/"
-            else:
-                wineprefix = "/home/" + user + "/.config/wine/veil/"
-        except KeyError:
-            print(" [*] Unable to get user from SUDO_USER environment variable\n")
-            print("     Your WINEPREFIX option might be screwed. You can fix it in /etc/veil/settings.py")
+        user = os.environ.get("SUDO_USER", pwd.getpwuid(os.getuid()).pw_name)
+        if user == 'root':
+            wineprefix = "/root/.config/wine/veil/"
+        else:
+            wineprefix = "/home/" + user + "/.config/wine/veil/"
         options["WINEPREFIX"] = wineprefix
 
         # Veil-Evasion specific options

modules/common/messages.py

@@ -8,7 +8,7 @@
 import helpers
 
 
-version = "2.26"
+version = "2.26.1"
 
 
 # try to find and import the settings.py config file