fix for an issue where whitewashing doesn't work (issue #168)

Author: Ekultek

lib/exploitation/exploiter.py

@@ -16,18 +16,22 @@ def whitelist_wash(hosts, whitelist_file):
     """
     remove IPs from hosts list that do not appear in WHITELIST_FILE
     """
-    whitelist_hosts = [x.strip() for x in open(whitelist_file).readlines() if x.strip()]
-    lib.output.info('Found {} entries in whitelist.txt, scrubbing'.format(str(len(whitelist_hosts))))
-    washed_hosts = []
-    # return supplied hosts if whitelist file is empty
-    if len(whitelist_hosts) == 0:
-        return hosts
-    else:
-        for host in hosts:
-            if host.strip() in whitelist_hosts:
-                washed_hosts.append(host)
+    try:
+        whitelist_hosts = [x.strip() for x in open(whitelist_file).readlines() if x.strip()]
+        lib.output.info('Found {} entries in whitelist.txt, scrubbing'.format(str(len(whitelist_hosts))))
+        washed_hosts = []
+        # return supplied hosts if whitelist file is empty
+        if len(whitelist_hosts) == 0:
+            return hosts
+        else:
+            for host in hosts:
+                if host.strip() in whitelist_hosts:
+                    washed_hosts.append(host)
 
-    return washed_hosts
+        return washed_hosts
+    except Exception:
+        lib.output.warning("unable to whitewash host list, does the file exist?")
+        return hosts
 
 
 class AutoSploitExploiter(object):