Oh so shiny

Author: Al-Azif

start.py

@@ -195,39 +195,39 @@ def check_update_pups():
     try:
         with open(os.path.join(CWD, 'updates',
                                'PS4UPDATE_SYSTEM.PUP'), 'rb') as buf:
-            print('>> Checking PS4UPDATE_SYSTEM.PUP\'s checksum')
+            print('>> Checking PS4UPDATE_SYSTEM.PUP\'s checksum', end='\r')
             hasher = hashlib.md5()
             data = buf.read()
             hasher.update(data)
             system_hash = hasher.hexdigest().upper()
             if system_hash != '203C76C97F7BE5B881DD0C77C8EDF385':
                 closer('ERROR: PS4UPDATE_SYSTEM.PUP is not version 4.05')
-            print('>> PS4UPDATE_SYSTEM.PUP checksum matches')
+            print('>> PS4UPDATE_SYSTEM.PUP checksum matches   ')
     except IOError:
         pass
 
     try:
         with open(os.path.join(CWD, 'updates',
                                'PS4UPDATE_RECOVERY.PUP'), 'rb') as buf:
-            print('>> Checking PS4UPDATE_RECOVERY.PUP\'s checksum')
+            print('>> Checking PS4UPDATE_RECOVERY.PUP\'s checksum', end='\r')
             hasher = hashlib.md5()
             data = buf.read()
             hasher.update(data)
             recovery_hash = hasher.hexdigest().upper()
             if recovery_hash != '741CFE2F0DEC1BB4663571DE78AE31CF':
                 closer('ERROR: PS4UPDATE_RECOVERY.PUP is not version 4.05')
-            print('>> PS4UPDATE_RECOVERY.PUP checksum matches')
+            print('>> PS4UPDATE_RECOVERY.PUP checksum matches   ')
     except IOError:
         pass
 
 
 def start_servers():
     """Start DNS and HTTP servers on seperate threads"""
-    print('>> Starting DNS server thread...')
+    print('>> Starting DNS server thread...', end='\r')
     fakedns.main(DNS_LOC, DEBUG)
     print('>> DNS server thread is running...')
 
-    print('>> Starting HTTP server thread...')
+    print('>> Starting HTTP server thread...', end='\r')
     server = ThreadedHTTPServer(('', 80), MyHandler)
     thread = threading.Thread(name='HTTP_Server',
                               target=server.serve_forever,
@@ -266,12 +266,20 @@ def payload_menu(payloads):
     """Displays a menu with all available payloads in it"""
     i = 1
     choice = -1
-    print('{} Payloads {}'.format('-' * 4, '-' * 46))
-    print('0. Don\'t send a payload')
+    print('┌────────────────────────────────────────────────────────┐')
+    print('│  Payloads                                              │')
+    print('├────────────────────────────────────────────────────────┤')
+    print('│  0. Don\'t send a payload                               │')
     for payload in payloads:
-        print('{}. {}'.format(i, payload))
+        payload = '│  {}. {}'.format(i, payload)
+        while len(payload) < 57:
+            payload += ' '
+        payload += '│'
+        if len(payload) > 58:
+            payload = payload[:57] + '│'
+        print(payload)
         i += 1
-    print('-' * 60)
+    print('└────────────────────────────────────────────────────────┘')
     while choice < 0 or choice >= i:
         choice = input('Choose a payload to send: ')
         try:
@@ -317,11 +325,19 @@ def exploit_menu():
     exploits = os.listdir(EXPLOIT_LOC)
     if not exploits:
         closer('ERROR: No exploits found')
-    print('{} Exploits {}'.format('-' * 4, '-' * 46))
+    print('┌────────────────────────────────────────────────────────┐')
+    print('│  Exploits                                              │')
+    print('├────────────────────────────────────────────────────────┤')
     for exploit in exploits:
-        print('{}. {}'.format(i, exploit))
+        exploit = '│  {}. {}'.format(i, exploit)
+        while len(exploit) < 57:
+            exploit += ' '
+        exploit += '│'
+        if len(exploit) > 58:
+            exploit = exploit[:57] + '│'
+        print(exploit)
         i += 1
-    print('-' * 60)
+    print('└────────────────────────────────────────────────────────┘')
     while choice < 1 or choice >= i:
         choice = input('Choose an exploit to host: ')
         try:
@@ -354,7 +370,7 @@ def getch():
 def closer(message):
     """Closing method"""
     print(message)
-    if message != '>> Exiting...':
+    if message != '\r>> Exiting...':
         print('Press any key to exit...', end='')
         sys.stdout.flush()
         if os.name == 'nt':
@@ -366,12 +382,25 @@ def closer(message):
     sys.exit()
 
 
+def menu_header():
+    """Very first thing that prints"""
+    print('┌────────────────────────────────────────────────────────┐')
+    print('│  PS4 Exploit Host                           by Al Azif │')
+    print('└────────────────────────────────────────────────────────┘')
+
+
+def menu_padding(item, term_char):
+    pass
+
+
 def main():
     """The main logic"""
     global DEBUG
     global AUTOSEND
     global EXPLOIT_LOC
 
+    menu_header()
+
     if not check_root():
         closer('ERROR: This must be run by root as it requires port 53 & 80')
 
@@ -408,7 +437,12 @@ def main():
         lan = get_lan()
 
         if write_conf(lan):
-            print('>> Your DNS IP is {}'.format(lan))
+            while len(lan) < 15:
+                lan += ' '
+            print('╔════════════════════════════════════════════════════════╗')
+            print('║               Your DNS IP is {}           ║'.format(lan))
+            print('╚════════════════════════════════════════════════════════╝')
+            # print('>> Your DNS IP is {}'.format(lan))
         else:
             closer('ERROR: Unable to write {}'.format(DNS_LOC))
 
@@ -417,7 +451,7 @@ def main():
         while True:
             pass
     except KeyboardInterrupt:
-        closer('>> Exiting...')
+        closer('\r>> Exiting...')
 
 
 if __name__ == '__main__':