gh-67224: Show source lines in tracebacks when using the -c option when running Python

Author: pablogsal

Lib/linecache.py

@@ -182,6 +182,7 @@ def get_lines(name=name, *args, **kwargs):
             return True
     return False
 
+
 def _register_code(code, string, name):
     cache[code] = (
             len(string),

Lib/test/test_cmd_line_script.py

@@ -683,6 +683,16 @@ def test_syntaxerror_null_bytes_in_multiline_string(self):
                         b'SyntaxError: source code cannot contain null bytes'
                     ]
                 )
+    
+    def test_source_lines_are_shown_when_running_source(self):
+        _, _, stderr = assert_python_failure("-c", "1/0")
+        expected_lines = [
+            b'Traceback (most recent call last):',
+            b'  File "<stdin>", line 1, in <module>',
+            b'    1/0',
+            b'    ~^~',
+            b'ZeroDivisionError: division by zero']
+        self.assertEqual(stderr.splitlines(), expected_lines)
 
     def test_consistent_sys_path_for_direct_execution(self):
         # This test case ensures that the following all give the same

Python/pythonrun.c

@@ -1147,8 +1147,15 @@ PyRun_StringFlags(const char *str, int start, PyObject *globals,
     mod = _PyParser_ASTFromString(
             str, &_Py_STR(anon_string), start, flags, arena);
 
-    if (mod != NULL)
-        ret = run_mod(mod, &_Py_STR(anon_string), globals, locals, flags, arena, NULL);
+    PyObject* source = PyUnicode_FromString(str);
+    if (!source) {
+        goto exit;
+    }
+    if (mod != NULL) {
+        ret = run_mod(mod, &_Py_STR(anon_string), globals, locals, flags, arena, source);
+    }
+    Py_DECREF(source);
+exit:
     _PyArena_Free(arena);
     return ret;
 }