Re-regexed handler

Author: Al-Azif

start.py

@@ -120,25 +120,24 @@ def payload_launcher(self):
     def do_GET(self):
         """Determines how to handle HTTP requests"""
         try:
-            path = self.path.rsplit('/', 1)[-1]
-            if path == 'ps4-updatelist.xml':
+            if re.match('^/update/ps4/list/[a-z]{2}/ps4-updatelist.xml', self.path):
                 self.updatelist()
-            elif path == 'ps4-updatefeature.html':
+            elif re.match('^/update/ps4/list/[a-z]{2}/ps4-updatelist.xml', self.path):
                 self.updatefeature()
-            elif path.endswith('PS4UPDATE.PUP'):
+            elif re.match('^/update/ps4/image/[0-9]{4}_[0-9]{4}/(sys|rec)_[a-f0-9]{32}/PS4UPDATE.PUP', self.path):
                 self.update_pup()
             elif re.match('^/networktest/get_2m', self.path):
                 self.network_test(2097152)
             elif re.match('^/networktest/get_6m', self.path):
                 self.network_test(6291456)
-            elif re.match('/document/[a-zA-Z\-]{2,5}/ps4/', self.path):
+            elif re.match('^/document/[a-zA-Z\-]{2,5}/ps4/', self.path):
                 self.exploit_match()
             else:
                 self.send_error(404)
         except IOError:
             self.send_error(404)
 
-        if path == 'rop.js':
+        if self.path.rsplit('/', 1)[-1] == 'rop.js':
             self.payload_launcher()
 
     def do_POST(self):
@@ -180,8 +179,7 @@ def generate_dns_rules(lan):
     rules = []
 
     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 (get|post).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')