Detect RustPython magic

Author: rocky

xdis/disasm.py

@@ -334,6 +334,8 @@ def disassemble_file(
             source_size,
             sip_hash,
         ) = load_module(pyc_filename)
+    except (ImportError, NotImplementedError):
+        raise
     except Exception:
         # Hack alert: we're using pyc_filename set as a proxy for whether the filename exists.
         # check_object_path() will succeed if the file exists.

xdis/load.py

@@ -329,6 +329,8 @@ def load_module_from_file_object(
                 pass
             else:
                 co = None
+        except NotImplementedError:
+            raise
         except Exception:
             kind, msg = sys.exc_info()[0:2]
             import traceback

xdis/magics.py

@@ -41,6 +41,7 @@
 
 PYPY3_MAGICS = (48, 64, 112, 160, 192, 240, 244, 256, 336, 384)
 GRAAL3_MAGICS = (21150, 21280)
+RUSTPYTHON_MAGICS = (12641,)
 
 
 def add_magic_from_int(magic_int, version):
@@ -457,6 +458,8 @@ def __by_version(magic_versions: Dict[bytes, str]) -> dict:
 add_magic_from_int(336, "3.9pypy")  # PyPy 3.9.15, PyPy 3.9.17
 add_magic_from_int(384, "3.10pypy")  # PyPy 3.10.12
 
+add_magic_from_int(12641, "3.13.0rust")  # RustPython 3.13.0
+
 # NOTE: This is JVM bytecode not Python bytecode
 add_magic_from_int(21150, "3.8.5Graal")
 

xdis/unmarshal.py

@@ -33,7 +33,7 @@
 
 from xdis.codetype import to_portable
 from xdis.cross_types import LongTypeForPython3, UnicodeForPython3
-from xdis.magics import GRAAL3_MAGICS, PYPY3_MAGICS, magic_int2tuple
+from xdis.magics import GRAAL3_MAGICS, PYPY3_MAGICS, RUSTPYTHON_MAGICS, magic_int2tuple
 from xdis.version_info import PYTHON3, PYTHON_VERSION_TRIPLE
 
 if PYTHON3:
@@ -141,6 +141,8 @@ def __init__(self, fp, magic_int, bytes_for_s, code_objects={}):
 
         In Python 3, a ``bytes`` type is used for strings.
         """
+        if magic_int in RUSTPYTHON_MAGICS:
+            raise NotImplementedError("RustPython not supported yet")
         self.fp = fp
         self.magic_int = magic_int
         self.code_objects = code_objects
@@ -162,8 +164,8 @@ def __init__(self, fp, magic_int, bytes_for_s, code_objects={}):
         self.internStrings = []
         self.internObjects = []
         self.version_tuple = tuple()
-        self.is_graal = False
-        self.is_pypy = False
+        self.is_graal = magic_int in GRAAL3_MAGICS
+        self.is_pypy = magic_int in PYPY3_MAGICS
 
     def load(self):
         """