bpo-34725: Adds _Py_SetProgramFullPath so embedders may override sys.executable (GH-9861)
diff --git a/Modules/main.c b/Modules/main.c
index f0f7fe5..6dbe6a3 100644
--- a/Modules/main.c
+++ b/Modules/main.c
@@ -1206,6 +1206,25 @@
}
+static _PyInitError
+config_init_executable(_PyCoreConfig *config)
+{
+ assert(config->executable == NULL);
+
+ /* If Py_SetProgramFullPath() was called, use its value */
+ const wchar_t *program_full_path = _Py_path_config.program_full_path;
+ if (program_full_path != NULL) {
+ config->executable = _PyMem_RawWcsdup(program_full_path);
+ if (config->executable == NULL) {
+ return _Py_INIT_NO_MEMORY();
+ }
+ return _Py_INIT_OK();
+ }
+
+ return _Py_INIT_OK();
+}
+
+
static void
pymain_header(_PyMain *pymain)
{
@@ -2350,6 +2369,13 @@
}
}
+ if (config->executable == NULL) {
+ err = config_init_executable(config);
+ if (_Py_INIT_FAILED(err)) {
+ return err;
+ }
+ }
+
if (config->utf8_mode < 0 || config->coerce_c_locale < 0) {
config_init_locale(config);
}