Update co_filename if needed (#49)

Sometimes co_filename is set to something different than what was written -- it is changed (using `_imp.fix_co_filename`) to match the file from which it was loaded. Here we ensure that this is propagated through hydration.

Author: iritkatriel

Include/internal/pycore_code.h

@@ -261,6 +261,7 @@ struct _PyCodeConstructor {
 PyAPI_FUNC(int) _PyCode_Validate(struct _PyCodeConstructor *);
 PyAPI_FUNC(PyCodeObject *) _PyCode_New(struct _PyCodeConstructor *);
 PyAPI_FUNC(PyCodeObject *) _PyCode_Update(struct _PyCodeConstructor *, PyCodeObject *);
+PyAPI_FUNC(void) _PyCode_UpdateFilenames(PyCodeObject *co, PyObject *oldname, PyObject *newname);
 
 
 /* Private API */

Objects/codeobject.c

@@ -430,6 +430,28 @@ _PyCode_New(struct _PyCodeConstructor *con)
     return co;
 }
 
+void
+_PyCode_UpdateFilenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
+{
+    PyObject *constants, *tmp;
+    Py_ssize_t i, n;
+
+    if (PyUnicode_Compare(co->co_filename, oldname))
+        return;
+
+    Py_INCREF(newname);
+    Py_XSETREF(co->co_filename, newname);
+
+    constants = co->co_consts;
+    n = constants != NULL ? PyTuple_GET_SIZE(constants) : 0;
+    for (i = 0; i < n; i++) {
+        tmp = PyTuple_GET_ITEM(constants, i);
+        if (PyCode_Check(tmp))
+            _PyCode_UpdateFilenames((PyCodeObject *)tmp,
+                oldname, newname);
+    }
+}
+
 PyCodeObject *
 _PyCode_Update(struct _PyCodeConstructor *con, PyCodeObject *code)
 {
@@ -447,12 +469,16 @@ _PyCode_Update(struct _PyCodeConstructor *con, PyCodeObject *code)
         con->columntable = Py_None;
     }
 
-    Py_XDECREF(code->co_filename);
-    Py_XDECREF(code->co_name);
-    Py_XDECREF(code->co_qualname);
+    PyObject *newname = code->co_filename;
+    Py_DECREF(code->co_name);
+    Py_DECREF(code->co_qualname);
 
     init_code(code, con);
 
+    assert(newname);
+    _PyCode_UpdateFilenames(code, code->co_filename, newname);
+    Py_DECREF(newname);
+
     return code;
 }
 

Python/import.c

@@ -823,28 +823,6 @@ PyImport_ExecCodeModuleObject(PyObject *name, PyObject *co, PyObject *pathname,
 }
 
 
-static void
-update_code_filenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
-{
-    PyObject *constants, *tmp;
-    Py_ssize_t i, n;
-
-    if (PyUnicode_Compare(co->co_filename, oldname))
-        return;
-
-    Py_INCREF(newname);
-    Py_XSETREF(co->co_filename, newname);
-
-    constants = co->co_consts;
-    n = constants != NULL ? PyTuple_GET_SIZE(constants) : 0;
-    for (i = 0; i < n; i++) {
-        tmp = PyTuple_GET_ITEM(constants, i);
-        if (PyCode_Check(tmp))
-            update_code_filenames((PyCodeObject *)tmp,
-                                  oldname, newname);
-    }
-}
-
 static void
 update_compiled_module(PyCodeObject *co, PyObject *newname)
 {
@@ -855,7 +833,7 @@ update_compiled_module(PyCodeObject *co, PyObject *newname)
 
     oldname = co->co_filename;
     Py_INCREF(oldname);
-    update_code_filenames(co, oldname, newname);
+    _PyCode_UpdateFilenames(co, oldname, newname);
     Py_DECREF(oldname);
 }