gh-120170: Exclude __mp_main__ in C version of whichmodule() (#120171)

li-dan · lysnikolaou · web-flow · commit 05a19b5e5689 · 2025-05-23T21:45:45.000+03:00
Co-authored-by: Lysandros Nikolaou &lt;lisandrosnik@gmail.com&gt;
diff --git a/Misc/NEWS.d/next/Library/2024-06-06-17-49-07.gh-issue-120170.DUxhmT.rst b/Misc/NEWS.d/next/Library/2024-06-06-17-49-07.gh-issue-120170.DUxhmT.rst
@@ -0,0 +1,3 @@
+Fix an issue in the :mod:`!_pickle` extension module in which importing
+:mod:`multiprocessing` could change how pickle identifies which module an
+object belongs to, potentially breaking the unpickling of those objects.
diff --git a/Modules/_pickle.c b/Modules/_pickle.c
@@ -1879,6 +1879,10 @@ _checkmodule(PyObject *module_name, PyObject *module,
             _PyUnicode_EqualToASCIIString(module_name, "__main__")) {
         return -1;
     }
+    if (PyUnicode_Check(module_name) &&
+            _PyUnicode_EqualToASCIIString(module_name, "__mp_main__")) {
+        return -1;
+    }
 
     PyObject *candidate = getattribute(module, dotted_path, 0);
     if (candidate == NULL) {

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	+Fix an issue in the :mod:`!_pickle` extension module in which importing
	`2`	+:mod:`multiprocessing` could change how pickle identifies which module an
	`3`	`+object belongs to, potentially breaking the unpickling of those objects.`