Add line table info to RustPython 3.12

Author: rocky

xdis/codetype/code312rust.py

@@ -47,6 +47,7 @@ def __init__(
         co_argcount: int,
         co_posonlyargcount: int,
         co_kwonlyargcount: int,
+        co_lnotab: dict,
         co_stacksize: int,
         co_flags: int,
         co_code: bytes,
@@ -65,7 +66,7 @@ def __init__(
         self.co_argcount = co_argcount
         self.co_posonlyargcount = co_posonlyargcount
         self.co_kwonlyargcount = co_kwonlyargcount
-        self.co_lnotab={}  # FIXME: compute this from locations.
+        self.co_lnotab = co_lnotab
         self.co_stacksize = co_stacksize
         self.co_flags = co_flags
         self.co_code = co_code

xdis/unmarsh_rust.py

@@ -163,12 +163,15 @@ def t_code_rust(self, save_ref, bytes_for_s: bool = False) -> Code313Rust:
         instr_count = self.read_int32()
         co_code = self.read_slice(instr_count * 2)
 
-        instructions = [int(co_code[i]) for i in range(len(co_code))] # debug
+        # instructions = [int(co_code[i]) for i in range(len(co_code))] # debug
+        co_lnotab = {}
 
         # read locations
         loc_count = self.read_int32()
         locations: List[SourceLocation] = []
-        for _ in range(loc_count):
+        last_line_number = -1
+        for i in range(loc_count):
+            offset = i * 2
             line_number = self.read_int32()
             if line_number == 0:
                 raise MarshalError("invalid source location")
@@ -180,6 +183,9 @@ def t_code_rust(self, save_ref, bytes_for_s: bool = False) -> Code313Rust:
                     column_offset=char_off_zero_indexed + 1  # convert from zero-indexed
                 )
             )
+            if line_number != last_line_number and offset not in co_lnotab:
+                co_lnotab[offset] = line_number
+                last_line_number = line_number
 
         flags = self.read_int16()
 
@@ -269,6 +275,7 @@ def read_names():
                 co_argcount=arg_count,
                 co_posonlyargcount=posonlyarg_count,
                 co_kwonlyargcount=kwonlyarg_count,
+                co_lnotab=co_lnotab,
                 co_stacksize=max_stackdepth,
                 co_flags=flags,
                 co_code=co_code,