✨ Add interactive stack trace

pros/cli/terminal.py

@@ -30,6 +30,7 @@ def terminal_cli():
               help='Specify 2 ports for the "share" backend. The default option deterministically selects ports '
                    'based on the serial port name')
 @click.option('--banner/--no-banner', 'request_banner', default=True)
+@click.option('--auto_stack_trace', default=True)
 @click.option('--output', nargs = 1, type=str, is_eager = True, help='Redirect terminal output to a file', default=None)
 
 def terminal(port: str, backend: str, **kwargs):
@@ -83,7 +84,7 @@ def terminal(port: str, backend: str, **kwargs):
         device = devices.RawStreamDevice(ser)
     else:
         device = devices.vex.V5UserDevice(ser)
-    term = Terminal(device, request_banner=kwargs.pop('request_banner', True))
+    term = Terminal(device, request_banner=kwargs.pop('request_banner', True), auto_stack_trace=kwargs.pop('auto_stack_trace', True))
 
     class TerminalOutput(object):
         def __init__(self, file):

pros/serial/terminal/terminal.py

@@ -1,10 +1,12 @@
 import codecs
 import os
 import signal
+import subprocess
 import sys
 import threading
 
 import colorama
+from sqlalchemy import null
 
 from pros.common.utils import logger
 from pros.serial import decode_bytes_to_str
@@ -165,12 +167,13 @@ def cleanup(self):
         'Sorry no implementation for your platform ({})'
         ' available.'.format(sys.platform))
 
-
 class Terminal(object):
     """This class is loosely based off of the pyserial miniterm"""
+    beginStackTrace = False
+    stackTraceFile = null
 
     def __init__(self, port_instance: StreamDevice, transformations=(),
-                 output_raw: bool = False, request_banner: bool = True):
+                 output_raw: bool = False, request_banner: bool = True, auto_stack_trace: bool = True):
         self.device = port_instance
         self.device.subscribe(b'sout')
         self.device.subscribe(b'serr')
@@ -224,7 +227,35 @@ def reader(self):
                 if data[0] == b'sout':
                     text = decode_bytes_to_str(data[1])
                 elif data[0] == b'serr':
-                    text = '{}{}{}'.format(colorama.Fore.RED, decode_bytes_to_str(data[1]), colorama.Style.RESET_ALL)
+                    #print(len(text))
+                    addr = "0x" + decode_bytes_to_str(data[1])[:7]
+                    if self.beginStackTrace and addr.isalnum() and addr[3] != 'x':
+                        def getTrace(s, path):
+                            if not os.path.exists(path):
+                                return ' : ??'
+                            temp = subprocess.Popen(['addr2line', '-faps', '-e', path, s],
+                                stdout=subprocess.PIPE).communicate()[0].decode('utf-8')
+                            if (temp.find('?') != -1):
+                                return ' : ??'
+                            else:
+                                return ' : ' + temp[12: len(temp) - 2]
+                        trace = getTrace(addr, "bin/hot.package.elf")
+                        trace += getTrace(addr, "bin/cold.package.elf")
+                        trace += getTrace(addr, "bin/monolith.elf")
+                        text = '{}{}{}'.format(colorama.Fore.RED, decode_bytes_to_str(data[1]) + trace, colorama.Style.RESET_ALL)
+                        file.write(addr + trace + '\n')
+                    else:
+                        text = '{}{}{}'.format(colorama.Fore.RED, decode_bytes_to_str(data[1]), colorama.Style.RESET_ALL)
+
+                    if "BEGIN STACK TRACE" in text:
+                        file = open("stack_trace.txt", "w")
+                        self.beginStackTrace = True
+
+                    if "END OF TRACE" in text:
+                        file.close()
+                        file = null
+                        self.beginStackTrace = False
+
                 elif data[0] == b'kdbg':
                     text = '{}\n\nKERNEL DEBUG:\t{}{}\n'.format(colorama.Back.GREEN + colorama.Style.BRIGHT,
                                                                 decode_bytes_to_str(data[1]),