Send DNS list as array, update submodule

Author: Al-Azif

FakeDns

@@ -39,7 +39,6 @@
 CWD = os.path.dirname(SCRIPT_LOC)
 EXPLOIT_LOC = os.path.join(CWD, 'exploit')
 PAYLOAD_LOC = os.path.join(CWD, 'payloads')
-DNS_LOC = os.path.join(CWD, 'dns.conf')
 DEBUG = False
 AUTOSEND = None
 
@@ -176,62 +175,27 @@ def get_lan():
     return lan
 
 
-def write_conf(lan):
-    """Writes the configuration file for FakeDns"""
-    exists = os.path.isfile(DNS_LOC)
+def generate_dns_rules(lan):
+    """Creates domain array for FakeDns"""
+    rules = []
 
-    try:
-        with open(DNS_LOC, 'w+') as buf:
-            buf.write('A manuals.playstation.net ' + lan + '\n')
-            buf.write('A post.net.playstation.net ' + lan + '\n')
-            buf.write('A get.net.playstation.net ' + lan + '\n')
-            buf.write('A d[a-z]{2}01.ps4.update.playstation.net ' + lan + '\n')
-            buf.write('A f[a-z]{2}01.ps4.update.playstation.net ' + lan + '\n')
-            buf.write('A h[a-z]{2}01.ps4.update.playstation.net ' + lan + '\n')
-            buf.write('A gs2.ww.prod.dl.playstation.net ' + lan + '\n')
-            buf.write('A [a-z0-9\.\-]*.cddbp.net 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.ea.com 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.llnwd.net 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.playstation.com 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.playstation.net 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.playstation.org 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.ribob01.net 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.sbdnpd.com 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.scea.com 0.0.0.0\n')
-            buf.write('A [a-z0-9\.\-]*.sonyentertainmentnetwork.com 0.0.0.0\n')
-        if not exists:
-            fix_permissions()
-        return True
-    except IOError:
-        return False
-
-
-def fix_permissions():
-    """Make FakeDNS config file the same permissions as start.py
-
-       This should only be run if the config didn't exist before
-       It will not include execution privileges
-    """
-    try:
-        stats = os.stat(SCRIPT_LOC)
+    rules.append('A manuals.playstation.net ' + lan)
+    rules.append('A post.net.playstation.net ' + lan)
+    rules.append('A get.net.playstation.net ' + lan)
+    rules.append('A (d|f|h)[a-z]{2}01.ps4.update.playstation.net ' + lan)
+    rules.append('A gs2.ww.prod.dl.playstation.net ' + lan)
+    rules.append('A [a-z0-9\.\-]*.cddbp.net 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.ea.com 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.llnwd.net 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.playstation.com 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.playstation.net 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.playstation.org 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.ribob01.net 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.sbdnpd.com 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.scea.com 0.0.0.0')
+    rules.append('A [a-z0-9\.\-]*.sonyentertainmentnetwork.com 0.0.0.0')
 
-        os.chown(DNS_LOC, stats.st_uid, stats.st_gid)
-
-        mask = oct(stats.st_mode & 0o777)
-        newmask = ''
-
-        for i in mask:
-            if i != 'o':
-                if int(i) % 2 != 0:
-                    i = str(int(i) - 1)
-            newmask += i
-
-        mask = int(newmask, 8)
-        os.chmod(DNS_LOC, mask)
-    except AttributeError:
-        pass
-    except OSError:
-        print('NON-FATAL ERROR: Unable to change permissions of DNS config')
+    return rules
 
 
 def check_update_pup(type, md5):
@@ -251,13 +215,11 @@ def check_update_pup(type, md5):
         pass
 
 
-def start_servers():
+def start_servers(rules):
     """Start DNS and HTTP servers on seperate threads"""
-    print('>> Starting DNS server thread...', end='\r')
-    fakedns.main(DNS_LOC, DEBUG)
+    fakedns.main(rules, DEBUG)
     print('>> DNS server thread is running...')
 
-    print('>> Starting HTTP server thread...', end='\r')
     server = ThreadedHTTPServer(('', 80), MyHandler)
     thread = threading.Thread(name='HTTP_Server',
                               target=server.serve_forever,
@@ -459,16 +421,15 @@ def main():
 
         lan = get_lan()
 
-        if write_conf(lan):
-            while len(lan) < 15:
-                lan += ' '
-            print('╔════════════════════════════════════════════════════════╗')
-            print('║             Your DNS IP is {}             ║'.format(lan))
-            print('╚════════════════════════════════════════════════════════╝')
-        else:
-            closer('ERROR: Unable to write {}'.format(DNS_LOC))
+        rules = generate_dns_rules(lan)
+
+        while len(lan) < 15:
+            lan += ' '
+        print('╔════════════════════════════════════════════════════════╗')
+        print('║             Your DNS IP is {}             ║'.format(lan))
+        print('╚════════════════════════════════════════════════════════╝')
 
-        start_servers()
+        start_servers(rules)
 
         while True:
             pass