gh-92041: Preserve frame filename semantics

Author: mdeck

Lib/inspect.py

@@ -943,7 +943,22 @@ def getmodule(object, _filename=None):
             if not isinstance(module_name, str):
                 return None
             module = sys.modules.get(module_name)
-            if ismodule(module) and module.__dict__ is object_globals:
+            if not (ismodule(module) and module.__dict__ is object_globals):
+                return None
+            module_file = getattr(module, '__file__', None)
+            if module_file is None:
+                return None
+            try:
+                file = getabsfile(object, _filename)
+            except (TypeError, FileNotFoundError):
+                return None
+            if object.f_code.co_filename == module_file:
+                return module
+            try:
+                module_file = getabsfile(module)
+            except (TypeError, FileNotFoundError):
+                return None
+            if file == module_file or file == os.path.realpath(module_file):
                 return module
             return None
 

Lib/test/test_inspect/test_inspect.py

@@ -871,6 +871,46 @@ def exec_namespace(namespace):
                 finally:
                     del sys.modules[module_name]
 
+    def test_getmodule_registered_exec_frame(self):
+        def exec_module(module, filename):
+            module.inspect = inspect
+            exec(compile(textwrap.dedent("""
+                frame = inspect.currentframe()
+                try:
+                    1 / 0
+                except ZeroDivisionError as error:
+                    traceback = error.__traceback__
+            """), filename, "exec"), module.__dict__)
+
+        module_name = f"{__name__}.registered"
+        module = types.ModuleType(module_name)
+        sys.modules[module_name] = module
+        try:
+            with temp_cwd() as cwd:
+                filename = os.path.join(cwd, "registered.py")
+                with open(filename, "w"):
+                    pass
+                module.__file__ = filename
+
+                exec_module(module, filename)
+                self.assertIs(inspect.getmodule(module.frame), module)
+                self.assertIs(inspect.getmodule(module.traceback), module)
+
+                # Globals identity is insufficient when the code came from a
+                # different origin than the registered module.
+                exec_module(module, filename + ".other")
+                self.assertIsNone(inspect.getmodule(module.frame))
+                self.assertIsNone(inspect.getmodule(module.traceback))
+
+                # Preserve the existing result for fileless modules while
+                # avoiding a scan of sys.modules.
+                del module.__file__
+                exec_module(module, "<fileless>")
+                self.assertIsNone(inspect.getmodule(module.frame))
+                self.assertIsNone(inspect.getmodule(module.traceback))
+        finally:
+            del sys.modules[module_name]
+
     def test_getmodule_file_not_found(self):
         # See bpo-45406
         def _getabsfile(obj, _filename):