mirror of
https://github.com/python/cpython.git
synced 2024-11-25 19:03:49 +08:00
55e498058f
Fix PYTHONCOERCECLOCALE=1 environment variable: only coerce the C locale if the LC_CTYPE locale is "C".
1580 lines
44 KiB
C
1580 lines
44 KiB
C
#include "Python.h"
|
|
#include "pycore_fileutils.h"
|
|
#include "pycore_pylifecycle.h"
|
|
#include "pycore_pymem.h"
|
|
#include "pycore_pathconfig.h"
|
|
#include "pycore_pystate.h"
|
|
#include <locale.h>
|
|
#ifdef HAVE_LANGINFO_H
|
|
# include <langinfo.h>
|
|
#endif
|
|
|
|
#include <locale.h> /* setlocale() */
|
|
#ifdef HAVE_LANGINFO_H
|
|
#include <langinfo.h> /* nl_langinfo(CODESET) */
|
|
#endif
|
|
|
|
|
|
#define DECODE_LOCALE_ERR(NAME, LEN) \
|
|
(((LEN) == -2) \
|
|
? _Py_INIT_USER_ERR("cannot decode " NAME) \
|
|
: _Py_INIT_NO_MEMORY())
|
|
|
|
|
|
/* Global configuration variables */
|
|
|
|
/* The filesystem encoding is chosen by config_init_fs_encoding(),
|
|
see also initfsencoding(). */
|
|
const char *Py_FileSystemDefaultEncoding = NULL;
|
|
int Py_HasFileSystemDefaultEncoding = 0;
|
|
const char *Py_FileSystemDefaultEncodeErrors = NULL;
|
|
static int _Py_HasFileSystemDefaultEncodeErrors = 0;
|
|
|
|
/* UTF-8 mode (PEP 540): if equals to 1, use the UTF-8 encoding, and change
|
|
stdin and stdout error handler to "surrogateescape". It is equal to
|
|
-1 by default: unknown, will be set by Py_Main() */
|
|
int Py_UTF8Mode = -1;
|
|
int Py_DebugFlag = 0; /* Needed by parser.c */
|
|
int Py_VerboseFlag = 0; /* Needed by import.c */
|
|
int Py_QuietFlag = 0; /* Needed by sysmodule.c */
|
|
int Py_InteractiveFlag = 0; /* Needed by Py_FdIsInteractive() below */
|
|
int Py_InspectFlag = 0; /* Needed to determine whether to exit at SystemExit */
|
|
int Py_OptimizeFlag = 0; /* Needed by compile.c */
|
|
int Py_NoSiteFlag = 0; /* Suppress 'import site' */
|
|
int Py_BytesWarningFlag = 0; /* Warn on str(bytes) and str(buffer) */
|
|
int Py_FrozenFlag = 0; /* Needed by getpath.c */
|
|
int Py_IgnoreEnvironmentFlag = 0; /* e.g. PYTHONPATH, PYTHONHOME */
|
|
int Py_DontWriteBytecodeFlag = 0; /* Suppress writing bytecode files (*.pyc) */
|
|
int Py_NoUserSiteDirectory = 0; /* for -s and site.py */
|
|
int Py_UnbufferedStdioFlag = 0; /* Unbuffered binary std{in,out,err} */
|
|
int Py_HashRandomizationFlag = 0; /* for -R and PYTHONHASHSEED */
|
|
int Py_IsolatedFlag = 0; /* for -I, isolate from user's env */
|
|
#ifdef MS_WINDOWS
|
|
int Py_LegacyWindowsFSEncodingFlag = 0; /* Uses mbcs instead of utf-8 */
|
|
int Py_LegacyWindowsStdioFlag = 0; /* Uses FileIO instead of WindowsConsoleIO */
|
|
#endif
|
|
|
|
|
|
PyObject *
|
|
_Py_GetGlobalVariablesAsDict(void)
|
|
{
|
|
PyObject *dict, *obj;
|
|
|
|
dict = PyDict_New();
|
|
if (dict == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
#define SET_ITEM(KEY, EXPR) \
|
|
do { \
|
|
obj = (EXPR); \
|
|
if (obj == NULL) { \
|
|
return NULL; \
|
|
} \
|
|
int res = PyDict_SetItemString(dict, (KEY), obj); \
|
|
Py_DECREF(obj); \
|
|
if (res < 0) { \
|
|
goto fail; \
|
|
} \
|
|
} while (0)
|
|
#define SET_ITEM_INT(VAR) \
|
|
SET_ITEM(#VAR, PyLong_FromLong(VAR))
|
|
#define FROM_STRING(STR) \
|
|
((STR != NULL) ? \
|
|
PyUnicode_FromString(STR) \
|
|
: (Py_INCREF(Py_None), Py_None))
|
|
#define SET_ITEM_STR(VAR) \
|
|
SET_ITEM(#VAR, FROM_STRING(VAR))
|
|
|
|
SET_ITEM_STR(Py_FileSystemDefaultEncoding);
|
|
SET_ITEM_INT(Py_HasFileSystemDefaultEncoding);
|
|
SET_ITEM_STR(Py_FileSystemDefaultEncodeErrors);
|
|
SET_ITEM_INT(_Py_HasFileSystemDefaultEncodeErrors);
|
|
|
|
SET_ITEM_INT(Py_UTF8Mode);
|
|
SET_ITEM_INT(Py_DebugFlag);
|
|
SET_ITEM_INT(Py_VerboseFlag);
|
|
SET_ITEM_INT(Py_QuietFlag);
|
|
SET_ITEM_INT(Py_InteractiveFlag);
|
|
SET_ITEM_INT(Py_InspectFlag);
|
|
|
|
SET_ITEM_INT(Py_OptimizeFlag);
|
|
SET_ITEM_INT(Py_NoSiteFlag);
|
|
SET_ITEM_INT(Py_BytesWarningFlag);
|
|
SET_ITEM_INT(Py_FrozenFlag);
|
|
SET_ITEM_INT(Py_IgnoreEnvironmentFlag);
|
|
SET_ITEM_INT(Py_DontWriteBytecodeFlag);
|
|
SET_ITEM_INT(Py_NoUserSiteDirectory);
|
|
SET_ITEM_INT(Py_UnbufferedStdioFlag);
|
|
SET_ITEM_INT(Py_HashRandomizationFlag);
|
|
SET_ITEM_INT(Py_IsolatedFlag);
|
|
|
|
#ifdef MS_WINDOWS
|
|
SET_ITEM_INT(Py_LegacyWindowsFSEncodingFlag);
|
|
SET_ITEM_INT(Py_LegacyWindowsStdioFlag);
|
|
#endif
|
|
|
|
return dict;
|
|
|
|
fail:
|
|
Py_DECREF(dict);
|
|
return NULL;
|
|
|
|
#undef FROM_STRING
|
|
#undef SET_ITEM
|
|
#undef SET_ITEM_INT
|
|
#undef SET_ITEM_STR
|
|
}
|
|
|
|
|
|
void
|
|
_Py_wstrlist_clear(int len, wchar_t **list)
|
|
{
|
|
for (int i=0; i < len; i++) {
|
|
PyMem_RawFree(list[i]);
|
|
}
|
|
PyMem_RawFree(list);
|
|
}
|
|
|
|
|
|
wchar_t**
|
|
_Py_wstrlist_copy(int len, wchar_t **list)
|
|
{
|
|
assert((len > 0 && list != NULL) || len == 0);
|
|
size_t size = len * sizeof(list[0]);
|
|
wchar_t **list_copy = PyMem_RawMalloc(size);
|
|
if (list_copy == NULL) {
|
|
return NULL;
|
|
}
|
|
for (int i=0; i < len; i++) {
|
|
wchar_t* arg = _PyMem_RawWcsdup(list[i]);
|
|
if (arg == NULL) {
|
|
_Py_wstrlist_clear(i, list_copy);
|
|
return NULL;
|
|
}
|
|
list_copy[i] = arg;
|
|
}
|
|
return list_copy;
|
|
}
|
|
|
|
|
|
void
|
|
_Py_ClearFileSystemEncoding(void)
|
|
{
|
|
if (!Py_HasFileSystemDefaultEncoding && Py_FileSystemDefaultEncoding) {
|
|
PyMem_RawFree((char*)Py_FileSystemDefaultEncoding);
|
|
Py_FileSystemDefaultEncoding = NULL;
|
|
}
|
|
if (!_Py_HasFileSystemDefaultEncodeErrors && Py_FileSystemDefaultEncodeErrors) {
|
|
PyMem_RawFree((char*)Py_FileSystemDefaultEncodeErrors);
|
|
Py_FileSystemDefaultEncodeErrors = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
/* Set Py_FileSystemDefaultEncoding and Py_FileSystemDefaultEncodeErrors
|
|
global configuration variables. */
|
|
int
|
|
_Py_SetFileSystemEncoding(const char *encoding, const char *errors)
|
|
{
|
|
char *encoding2 = _PyMem_RawStrdup(encoding);
|
|
if (encoding2 == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
char *errors2 = _PyMem_RawStrdup(errors);
|
|
if (errors2 == NULL) {
|
|
PyMem_RawFree(encoding2);
|
|
return -1;
|
|
}
|
|
|
|
_Py_ClearFileSystemEncoding();
|
|
|
|
Py_FileSystemDefaultEncoding = encoding2;
|
|
Py_HasFileSystemDefaultEncoding = 0;
|
|
|
|
Py_FileSystemDefaultEncodeErrors = errors2;
|
|
_Py_HasFileSystemDefaultEncodeErrors = 0;
|
|
return 0;
|
|
}
|
|
|
|
|
|
/* Helper to allow an embedding application to override the normal
|
|
* mechanism that attempts to figure out an appropriate IO encoding
|
|
*/
|
|
|
|
static char *_Py_StandardStreamEncoding = NULL;
|
|
static char *_Py_StandardStreamErrors = NULL;
|
|
|
|
int
|
|
Py_SetStandardStreamEncoding(const char *encoding, const char *errors)
|
|
{
|
|
if (Py_IsInitialized()) {
|
|
/* This is too late to have any effect */
|
|
return -1;
|
|
}
|
|
|
|
int res = 0;
|
|
|
|
/* Py_SetStandardStreamEncoding() can be called before Py_Initialize(),
|
|
but Py_Initialize() can change the allocator. Use a known allocator
|
|
to be able to release the memory later. */
|
|
PyMemAllocatorEx old_alloc;
|
|
_PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
|
|
|
|
/* Can't call PyErr_NoMemory() on errors, as Python hasn't been
|
|
* initialised yet.
|
|
*
|
|
* However, the raw memory allocators are initialised appropriately
|
|
* as C static variables, so _PyMem_RawStrdup is OK even though
|
|
* Py_Initialize hasn't been called yet.
|
|
*/
|
|
if (encoding) {
|
|
_Py_StandardStreamEncoding = _PyMem_RawStrdup(encoding);
|
|
if (!_Py_StandardStreamEncoding) {
|
|
res = -2;
|
|
goto done;
|
|
}
|
|
}
|
|
if (errors) {
|
|
_Py_StandardStreamErrors = _PyMem_RawStrdup(errors);
|
|
if (!_Py_StandardStreamErrors) {
|
|
if (_Py_StandardStreamEncoding) {
|
|
PyMem_RawFree(_Py_StandardStreamEncoding);
|
|
}
|
|
res = -3;
|
|
goto done;
|
|
}
|
|
}
|
|
#ifdef MS_WINDOWS
|
|
if (_Py_StandardStreamEncoding) {
|
|
/* Overriding the stream encoding implies legacy streams */
|
|
Py_LegacyWindowsStdioFlag = 1;
|
|
}
|
|
#endif
|
|
|
|
done:
|
|
PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
void
|
|
_Py_ClearStandardStreamEncoding(void)
|
|
{
|
|
/* Use the same allocator than Py_SetStandardStreamEncoding() */
|
|
PyMemAllocatorEx old_alloc;
|
|
_PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
|
|
|
|
/* We won't need them anymore. */
|
|
if (_Py_StandardStreamEncoding) {
|
|
PyMem_RawFree(_Py_StandardStreamEncoding);
|
|
_Py_StandardStreamEncoding = NULL;
|
|
}
|
|
if (_Py_StandardStreamErrors) {
|
|
PyMem_RawFree(_Py_StandardStreamErrors);
|
|
_Py_StandardStreamErrors = NULL;
|
|
}
|
|
|
|
PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
|
|
}
|
|
|
|
|
|
/* Free memory allocated in config, but don't clear all attributes */
|
|
void
|
|
_PyCoreConfig_Clear(_PyCoreConfig *config)
|
|
{
|
|
#define CLEAR(ATTR) \
|
|
do { \
|
|
PyMem_RawFree(ATTR); \
|
|
ATTR = NULL; \
|
|
} while (0)
|
|
#define CLEAR_WSTRLIST(LEN, LIST) \
|
|
do { \
|
|
_Py_wstrlist_clear(LEN, LIST); \
|
|
LEN = 0; \
|
|
LIST = NULL; \
|
|
} while (0)
|
|
|
|
CLEAR(config->pycache_prefix);
|
|
CLEAR(config->module_search_path_env);
|
|
CLEAR(config->home);
|
|
CLEAR(config->program_name);
|
|
CLEAR(config->program);
|
|
|
|
CLEAR_WSTRLIST(config->argc, config->argv);
|
|
config->argc = -1;
|
|
|
|
CLEAR_WSTRLIST(config->nwarnoption, config->warnoptions);
|
|
CLEAR_WSTRLIST(config->nxoption, config->xoptions);
|
|
CLEAR_WSTRLIST(config->nmodule_search_path, config->module_search_paths);
|
|
config->nmodule_search_path = -1;
|
|
|
|
CLEAR(config->executable);
|
|
CLEAR(config->prefix);
|
|
CLEAR(config->base_prefix);
|
|
CLEAR(config->exec_prefix);
|
|
#ifdef MS_WINDOWS
|
|
CLEAR(config->dll_path);
|
|
#endif
|
|
CLEAR(config->base_exec_prefix);
|
|
|
|
CLEAR(config->filesystem_encoding);
|
|
CLEAR(config->filesystem_errors);
|
|
CLEAR(config->stdio_encoding);
|
|
CLEAR(config->stdio_errors);
|
|
#undef CLEAR
|
|
#undef CLEAR_WSTRLIST
|
|
}
|
|
|
|
|
|
int
|
|
_PyCoreConfig_Copy(_PyCoreConfig *config, const _PyCoreConfig *config2)
|
|
{
|
|
_PyCoreConfig_Clear(config);
|
|
|
|
#define COPY_ATTR(ATTR) config->ATTR = config2->ATTR
|
|
#define COPY_STR_ATTR(ATTR) \
|
|
do { \
|
|
if (config2->ATTR != NULL) { \
|
|
config->ATTR = _PyMem_RawStrdup(config2->ATTR); \
|
|
if (config->ATTR == NULL) { \
|
|
return -1; \
|
|
} \
|
|
} \
|
|
} while (0)
|
|
#define COPY_WSTR_ATTR(ATTR) \
|
|
do { \
|
|
if (config2->ATTR != NULL) { \
|
|
config->ATTR = _PyMem_RawWcsdup(config2->ATTR); \
|
|
if (config->ATTR == NULL) { \
|
|
return -1; \
|
|
} \
|
|
} \
|
|
} while (0)
|
|
#define COPY_WSTRLIST(LEN, LIST) \
|
|
do { \
|
|
if (config2->LIST != NULL) { \
|
|
config->LIST = _Py_wstrlist_copy(config2->LEN, config2->LIST); \
|
|
if (config->LIST == NULL) { \
|
|
return -1; \
|
|
} \
|
|
} \
|
|
config->LEN = config2->LEN; \
|
|
} while (0)
|
|
|
|
COPY_ATTR(install_signal_handlers);
|
|
COPY_ATTR(use_environment);
|
|
COPY_ATTR(use_hash_seed);
|
|
COPY_ATTR(hash_seed);
|
|
COPY_ATTR(_install_importlib);
|
|
COPY_ATTR(allocator);
|
|
COPY_ATTR(dev_mode);
|
|
COPY_ATTR(faulthandler);
|
|
COPY_ATTR(tracemalloc);
|
|
COPY_ATTR(import_time);
|
|
COPY_ATTR(show_ref_count);
|
|
COPY_ATTR(show_alloc_count);
|
|
COPY_ATTR(dump_refs);
|
|
COPY_ATTR(malloc_stats);
|
|
|
|
COPY_ATTR(coerce_c_locale);
|
|
COPY_ATTR(coerce_c_locale_warn);
|
|
COPY_ATTR(utf8_mode);
|
|
|
|
COPY_WSTR_ATTR(pycache_prefix);
|
|
COPY_WSTR_ATTR(module_search_path_env);
|
|
COPY_WSTR_ATTR(home);
|
|
COPY_WSTR_ATTR(program_name);
|
|
COPY_WSTR_ATTR(program);
|
|
|
|
COPY_WSTRLIST(argc, argv);
|
|
COPY_WSTRLIST(nwarnoption, warnoptions);
|
|
COPY_WSTRLIST(nxoption, xoptions);
|
|
COPY_WSTRLIST(nmodule_search_path, module_search_paths);
|
|
|
|
COPY_WSTR_ATTR(executable);
|
|
COPY_WSTR_ATTR(prefix);
|
|
COPY_WSTR_ATTR(base_prefix);
|
|
COPY_WSTR_ATTR(exec_prefix);
|
|
#ifdef MS_WINDOWS
|
|
COPY_WSTR_ATTR(dll_path);
|
|
#endif
|
|
COPY_WSTR_ATTR(base_exec_prefix);
|
|
|
|
COPY_ATTR(isolated);
|
|
COPY_ATTR(site_import);
|
|
COPY_ATTR(bytes_warning);
|
|
COPY_ATTR(inspect);
|
|
COPY_ATTR(interactive);
|
|
COPY_ATTR(optimization_level);
|
|
COPY_ATTR(parser_debug);
|
|
COPY_ATTR(write_bytecode);
|
|
COPY_ATTR(verbose);
|
|
COPY_ATTR(quiet);
|
|
COPY_ATTR(user_site_directory);
|
|
COPY_ATTR(buffered_stdio);
|
|
COPY_STR_ATTR(filesystem_encoding);
|
|
COPY_STR_ATTR(filesystem_errors);
|
|
COPY_STR_ATTR(stdio_encoding);
|
|
COPY_STR_ATTR(stdio_errors);
|
|
#ifdef MS_WINDOWS
|
|
COPY_ATTR(legacy_windows_fs_encoding);
|
|
COPY_ATTR(legacy_windows_stdio);
|
|
#endif
|
|
COPY_ATTR(_check_hash_pycs_mode);
|
|
COPY_ATTR(_frozen);
|
|
|
|
#undef COPY_ATTR
|
|
#undef COPY_STR_ATTR
|
|
#undef COPY_WSTR_ATTR
|
|
#undef COPY_WSTRLIST
|
|
return 0;
|
|
}
|
|
|
|
|
|
const char*
|
|
_PyCoreConfig_GetEnv(const _PyCoreConfig *config, const char *name)
|
|
{
|
|
assert(config->use_environment >= 0);
|
|
|
|
if (!config->use_environment) {
|
|
return NULL;
|
|
}
|
|
|
|
const char *var = getenv(name);
|
|
if (var && var[0] != '\0') {
|
|
return var;
|
|
}
|
|
else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
|
|
int
|
|
_PyCoreConfig_GetEnvDup(const _PyCoreConfig *config,
|
|
wchar_t **dest,
|
|
wchar_t *wname, char *name)
|
|
{
|
|
assert(config->use_environment >= 0);
|
|
|
|
if (!config->use_environment) {
|
|
*dest = NULL;
|
|
return 0;
|
|
}
|
|
|
|
#ifdef MS_WINDOWS
|
|
const wchar_t *var = _wgetenv(wname);
|
|
if (!var || var[0] == '\0') {
|
|
*dest = NULL;
|
|
return 0;
|
|
}
|
|
|
|
wchar_t *copy = _PyMem_RawWcsdup(var);
|
|
if (copy == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
*dest = copy;
|
|
#else
|
|
const char *var = getenv(name);
|
|
if (!var || var[0] == '\0') {
|
|
*dest = NULL;
|
|
return 0;
|
|
}
|
|
|
|
size_t len;
|
|
wchar_t *wvar = Py_DecodeLocale(var, &len);
|
|
if (!wvar) {
|
|
if (len == (size_t)-2) {
|
|
return -2;
|
|
}
|
|
else {
|
|
return -1;
|
|
}
|
|
}
|
|
*dest = wvar;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
|
|
void
|
|
_PyCoreConfig_GetGlobalConfig(_PyCoreConfig *config)
|
|
{
|
|
#define COPY_FLAG(ATTR, VALUE) \
|
|
if (config->ATTR == -1) { \
|
|
config->ATTR = VALUE; \
|
|
}
|
|
#define COPY_NOT_FLAG(ATTR, VALUE) \
|
|
if (config->ATTR == -1) { \
|
|
config->ATTR = !(VALUE); \
|
|
}
|
|
|
|
COPY_FLAG(utf8_mode, Py_UTF8Mode);
|
|
COPY_FLAG(isolated, Py_IsolatedFlag);
|
|
COPY_FLAG(bytes_warning, Py_BytesWarningFlag);
|
|
COPY_FLAG(inspect, Py_InspectFlag);
|
|
COPY_FLAG(interactive, Py_InteractiveFlag);
|
|
COPY_FLAG(optimization_level, Py_OptimizeFlag);
|
|
COPY_FLAG(parser_debug, Py_DebugFlag);
|
|
COPY_FLAG(verbose, Py_VerboseFlag);
|
|
COPY_FLAG(quiet, Py_QuietFlag);
|
|
#ifdef MS_WINDOWS
|
|
COPY_FLAG(legacy_windows_fs_encoding, Py_LegacyWindowsFSEncodingFlag);
|
|
COPY_FLAG(legacy_windows_stdio, Py_LegacyWindowsStdioFlag);
|
|
#endif
|
|
COPY_FLAG(_frozen, Py_FrozenFlag);
|
|
|
|
COPY_NOT_FLAG(use_environment, Py_IgnoreEnvironmentFlag);
|
|
COPY_NOT_FLAG(buffered_stdio, Py_UnbufferedStdioFlag);
|
|
COPY_NOT_FLAG(site_import, Py_NoSiteFlag);
|
|
COPY_NOT_FLAG(write_bytecode, Py_DontWriteBytecodeFlag);
|
|
COPY_NOT_FLAG(user_site_directory, Py_NoUserSiteDirectory);
|
|
|
|
#undef COPY_FLAG
|
|
#undef COPY_NOT_FLAG
|
|
}
|
|
|
|
|
|
/* Set Py_xxx global configuration variables from 'config' configuration. */
|
|
void
|
|
_PyCoreConfig_SetGlobalConfig(const _PyCoreConfig *config)
|
|
{
|
|
#define COPY_FLAG(ATTR, VAR) \
|
|
if (config->ATTR != -1) { \
|
|
VAR = config->ATTR; \
|
|
}
|
|
#define COPY_NOT_FLAG(ATTR, VAR) \
|
|
if (config->ATTR != -1) { \
|
|
VAR = !config->ATTR; \
|
|
}
|
|
|
|
COPY_FLAG(utf8_mode, Py_UTF8Mode);
|
|
COPY_FLAG(isolated, Py_IsolatedFlag);
|
|
COPY_FLAG(bytes_warning, Py_BytesWarningFlag);
|
|
COPY_FLAG(inspect, Py_InspectFlag);
|
|
COPY_FLAG(interactive, Py_InteractiveFlag);
|
|
COPY_FLAG(optimization_level, Py_OptimizeFlag);
|
|
COPY_FLAG(parser_debug, Py_DebugFlag);
|
|
COPY_FLAG(verbose, Py_VerboseFlag);
|
|
COPY_FLAG(quiet, Py_QuietFlag);
|
|
#ifdef MS_WINDOWS
|
|
COPY_FLAG(legacy_windows_fs_encoding, Py_LegacyWindowsFSEncodingFlag);
|
|
COPY_FLAG(legacy_windows_stdio, Py_LegacyWindowsStdioFlag);
|
|
#endif
|
|
COPY_FLAG(_frozen, Py_FrozenFlag);
|
|
|
|
COPY_NOT_FLAG(use_environment, Py_IgnoreEnvironmentFlag);
|
|
COPY_NOT_FLAG(buffered_stdio, Py_UnbufferedStdioFlag);
|
|
COPY_NOT_FLAG(site_import, Py_NoSiteFlag);
|
|
COPY_NOT_FLAG(write_bytecode, Py_DontWriteBytecodeFlag);
|
|
COPY_NOT_FLAG(user_site_directory, Py_NoUserSiteDirectory);
|
|
|
|
/* Random or non-zero hash seed */
|
|
Py_HashRandomizationFlag = (config->use_hash_seed == 0 ||
|
|
config->hash_seed != 0);
|
|
|
|
#undef COPY_FLAG
|
|
#undef COPY_NOT_FLAG
|
|
}
|
|
|
|
|
|
/* Get the program name: use PYTHONEXECUTABLE and __PYVENV_LAUNCHER__
|
|
environment variables on macOS if available. */
|
|
static _PyInitError
|
|
config_init_program_name(_PyCoreConfig *config)
|
|
{
|
|
assert(config->program_name == NULL);
|
|
|
|
/* If Py_SetProgramName() was called, use its value */
|
|
const wchar_t *program_name = _Py_path_config.program_name;
|
|
if (program_name != NULL) {
|
|
config->program_name = _PyMem_RawWcsdup(program_name);
|
|
if (config->program_name == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
#ifdef __APPLE__
|
|
/* On MacOS X, when the Python interpreter is embedded in an
|
|
application bundle, it gets executed by a bootstrapping script
|
|
that does os.execve() with an argv[0] that's different from the
|
|
actual Python executable. This is needed to keep the Finder happy,
|
|
or rather, to work around Apple's overly strict requirements of
|
|
the process name. However, we still need a usable sys.executable,
|
|
so the actual executable path is passed in an environment variable.
|
|
See Lib/plat-mac/bundlebuiler.py for details about the bootstrap
|
|
script. */
|
|
const char *p = _PyCoreConfig_GetEnv(config, "PYTHONEXECUTABLE");
|
|
if (p != NULL) {
|
|
size_t len;
|
|
wchar_t* program_name = Py_DecodeLocale(p, &len);
|
|
if (program_name == NULL) {
|
|
return DECODE_LOCALE_ERR("PYTHONEXECUTABLE environment "
|
|
"variable", (Py_ssize_t)len);
|
|
}
|
|
config->program_name = program_name;
|
|
return _Py_INIT_OK();
|
|
}
|
|
#ifdef WITH_NEXT_FRAMEWORK
|
|
else {
|
|
const char* pyvenv_launcher = getenv("__PYVENV_LAUNCHER__");
|
|
if (pyvenv_launcher && *pyvenv_launcher) {
|
|
/* Used by Mac/Tools/pythonw.c to forward
|
|
* the argv0 of the stub executable
|
|
*/
|
|
size_t len;
|
|
wchar_t* program_name = Py_DecodeLocale(pyvenv_launcher, &len);
|
|
if (program_name == NULL) {
|
|
return DECODE_LOCALE_ERR("__PYVENV_LAUNCHER__ environment "
|
|
"variable", (Py_ssize_t)len);
|
|
}
|
|
config->program_name = program_name;
|
|
return _Py_INIT_OK();
|
|
}
|
|
}
|
|
#endif /* WITH_NEXT_FRAMEWORK */
|
|
#endif /* __APPLE__ */
|
|
|
|
/* Use argv[0] by default, if available */
|
|
if (config->program != NULL) {
|
|
config->program_name = _PyMem_RawWcsdup(config->program);
|
|
if (config->program_name == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
/* Last fall back: hardcoded string */
|
|
#ifdef MS_WINDOWS
|
|
const wchar_t *default_program_name = L"python";
|
|
#else
|
|
const wchar_t *default_program_name = L"python3";
|
|
#endif
|
|
config->program_name = _PyMem_RawWcsdup(default_program_name);
|
|
if (config->program_name == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
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 const wchar_t*
|
|
config_get_xoption(const _PyCoreConfig *config, wchar_t *name)
|
|
{
|
|
int nxoption = config->nxoption;
|
|
wchar_t **xoptions = config->xoptions;
|
|
for (int i=0; i < nxoption; i++) {
|
|
wchar_t *option = xoptions[i];
|
|
size_t len;
|
|
wchar_t *sep = wcschr(option, L'=');
|
|
if (sep != NULL) {
|
|
len = (sep - option);
|
|
}
|
|
else {
|
|
len = wcslen(option);
|
|
}
|
|
if (wcsncmp(option, name, len) == 0 && name[len] == L'\0') {
|
|
return option;
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_home(_PyCoreConfig *config)
|
|
{
|
|
wchar_t *home;
|
|
|
|
/* If Py_SetPythonHome() was called, use its value */
|
|
home = _Py_path_config.home;
|
|
if (home) {
|
|
config->home = _PyMem_RawWcsdup(home);
|
|
if (config->home == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
int res = _PyCoreConfig_GetEnvDup(config, &home,
|
|
L"PYTHONHOME", "PYTHONHOME");
|
|
if (res < 0) {
|
|
return DECODE_LOCALE_ERR("PYTHONHOME", res);
|
|
}
|
|
config->home = home;
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_hash_seed(_PyCoreConfig *config)
|
|
{
|
|
const char *seed_text = _PyCoreConfig_GetEnv(config, "PYTHONHASHSEED");
|
|
|
|
Py_BUILD_ASSERT(sizeof(_Py_HashSecret_t) == sizeof(_Py_HashSecret.uc));
|
|
/* Convert a text seed to a numeric one */
|
|
if (seed_text && strcmp(seed_text, "random") != 0) {
|
|
const char *endptr = seed_text;
|
|
unsigned long seed;
|
|
errno = 0;
|
|
seed = strtoul(seed_text, (char **)&endptr, 10);
|
|
if (*endptr != '\0'
|
|
|| seed > 4294967295UL
|
|
|| (errno == ERANGE && seed == ULONG_MAX))
|
|
{
|
|
return _Py_INIT_USER_ERR("PYTHONHASHSEED must be \"random\" "
|
|
"or an integer in range [0; 4294967295]");
|
|
}
|
|
/* Use a specific hash */
|
|
config->use_hash_seed = 1;
|
|
config->hash_seed = seed;
|
|
}
|
|
else {
|
|
/* Use a random hash */
|
|
config->use_hash_seed = 0;
|
|
config->hash_seed = 0;
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_utf8_mode(_PyCoreConfig *config)
|
|
{
|
|
const wchar_t *xopt = config_get_xoption(config, L"utf8");
|
|
if (xopt) {
|
|
wchar_t *sep = wcschr(xopt, L'=');
|
|
if (sep) {
|
|
xopt = sep + 1;
|
|
if (wcscmp(xopt, L"1") == 0) {
|
|
config->utf8_mode = 1;
|
|
}
|
|
else if (wcscmp(xopt, L"0") == 0) {
|
|
config->utf8_mode = 0;
|
|
}
|
|
else {
|
|
return _Py_INIT_USER_ERR("invalid -X utf8 option value");
|
|
}
|
|
}
|
|
else {
|
|
config->utf8_mode = 1;
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
const char *opt = _PyCoreConfig_GetEnv(config, "PYTHONUTF8");
|
|
if (opt) {
|
|
if (strcmp(opt, "1") == 0) {
|
|
config->utf8_mode = 1;
|
|
}
|
|
else if (strcmp(opt, "0") == 0) {
|
|
config->utf8_mode = 0;
|
|
}
|
|
else {
|
|
return _Py_INIT_USER_ERR("invalid PYTHONUTF8 environment "
|
|
"variable value");
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static int
|
|
config_str_to_int(const char *str, int *result)
|
|
{
|
|
const char *endptr = str;
|
|
errno = 0;
|
|
long value = strtol(str, (char **)&endptr, 10);
|
|
if (*endptr != '\0' || errno == ERANGE) {
|
|
return -1;
|
|
}
|
|
if (value < INT_MIN || value > INT_MAX) {
|
|
return -1;
|
|
}
|
|
|
|
*result = (int)value;
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int
|
|
config_wstr_to_int(const wchar_t *wstr, int *result)
|
|
{
|
|
const wchar_t *endptr = wstr;
|
|
errno = 0;
|
|
long value = wcstol(wstr, (wchar_t **)&endptr, 10);
|
|
if (*endptr != '\0' || errno == ERANGE) {
|
|
return -1;
|
|
}
|
|
if (value < INT_MIN || value > INT_MAX) {
|
|
return -1;
|
|
}
|
|
|
|
*result = (int)value;
|
|
return 0;
|
|
}
|
|
|
|
|
|
static void
|
|
get_env_flag(_PyCoreConfig *config, int *flag, const char *name)
|
|
{
|
|
const char *var = _PyCoreConfig_GetEnv(config, name);
|
|
if (!var) {
|
|
return;
|
|
}
|
|
int value;
|
|
if (config_str_to_int(var, &value) < 0 || value < 0) {
|
|
/* PYTHONDEBUG=text and PYTHONDEBUG=-2 behave as PYTHONDEBUG=1 */
|
|
value = 1;
|
|
}
|
|
if (*flag < value) {
|
|
*flag = value;
|
|
}
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_read_env_vars(_PyCoreConfig *config)
|
|
{
|
|
/* Get environment variables */
|
|
get_env_flag(config, &config->parser_debug, "PYTHONDEBUG");
|
|
get_env_flag(config, &config->verbose, "PYTHONVERBOSE");
|
|
get_env_flag(config, &config->optimization_level, "PYTHONOPTIMIZE");
|
|
get_env_flag(config, &config->inspect, "PYTHONINSPECT");
|
|
|
|
int dont_write_bytecode = 0;
|
|
get_env_flag(config, &dont_write_bytecode, "PYTHONDONTWRITEBYTECODE");
|
|
if (dont_write_bytecode) {
|
|
config->write_bytecode = 0;
|
|
}
|
|
|
|
int no_user_site_directory = 0;
|
|
get_env_flag(config, &no_user_site_directory, "PYTHONNOUSERSITE");
|
|
if (no_user_site_directory) {
|
|
config->user_site_directory = 0;
|
|
}
|
|
|
|
int unbuffered_stdio = 0;
|
|
get_env_flag(config, &unbuffered_stdio, "PYTHONUNBUFFERED");
|
|
if (unbuffered_stdio) {
|
|
config->buffered_stdio = 0;
|
|
}
|
|
|
|
#ifdef MS_WINDOWS
|
|
get_env_flag(config, &config->legacy_windows_fs_encoding,
|
|
"PYTHONLEGACYWINDOWSFSENCODING");
|
|
get_env_flag(config, &config->legacy_windows_stdio,
|
|
"PYTHONLEGACYWINDOWSSTDIO");
|
|
#endif
|
|
|
|
if (config->allocator == NULL) {
|
|
config->allocator = _PyCoreConfig_GetEnv(config, "PYTHONMALLOC");
|
|
}
|
|
|
|
if (_PyCoreConfig_GetEnv(config, "PYTHONDUMPREFS")) {
|
|
config->dump_refs = 1;
|
|
}
|
|
if (_PyCoreConfig_GetEnv(config, "PYTHONMALLOCSTATS")) {
|
|
config->malloc_stats = 1;
|
|
}
|
|
|
|
const char *env = _PyCoreConfig_GetEnv(config, "PYTHONCOERCECLOCALE");
|
|
if (env) {
|
|
if (strcmp(env, "0") == 0) {
|
|
if (config->coerce_c_locale < 0) {
|
|
config->coerce_c_locale = 0;
|
|
}
|
|
}
|
|
else if (strcmp(env, "warn") == 0) {
|
|
config->coerce_c_locale_warn = 1;
|
|
}
|
|
else {
|
|
if (config->coerce_c_locale < 0) {
|
|
config->coerce_c_locale = 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
wchar_t *path;
|
|
int res = _PyCoreConfig_GetEnvDup(config, &path,
|
|
L"PYTHONPATH", "PYTHONPATH");
|
|
if (res < 0) {
|
|
return DECODE_LOCALE_ERR("PYTHONPATH", res);
|
|
}
|
|
config->module_search_path_env = path;
|
|
|
|
if (config->use_hash_seed < 0) {
|
|
_PyInitError err = config_init_hash_seed(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_tracemalloc(_PyCoreConfig *config)
|
|
{
|
|
int nframe;
|
|
int valid;
|
|
|
|
const char *env = _PyCoreConfig_GetEnv(config, "PYTHONTRACEMALLOC");
|
|
if (env) {
|
|
if (!config_str_to_int(env, &nframe)) {
|
|
valid = (nframe >= 0);
|
|
}
|
|
else {
|
|
valid = 0;
|
|
}
|
|
if (!valid) {
|
|
return _Py_INIT_USER_ERR("PYTHONTRACEMALLOC: invalid number "
|
|
"of frames");
|
|
}
|
|
config->tracemalloc = nframe;
|
|
}
|
|
|
|
const wchar_t *xoption = config_get_xoption(config, L"tracemalloc");
|
|
if (xoption) {
|
|
const wchar_t *sep = wcschr(xoption, L'=');
|
|
if (sep) {
|
|
if (!config_wstr_to_int(sep + 1, &nframe)) {
|
|
valid = (nframe >= 0);
|
|
}
|
|
else {
|
|
valid = 0;
|
|
}
|
|
if (!valid) {
|
|
return _Py_INIT_USER_ERR("-X tracemalloc=NFRAME: "
|
|
"invalid number of frames");
|
|
}
|
|
}
|
|
else {
|
|
/* -X tracemalloc behaves as -X tracemalloc=1 */
|
|
nframe = 1;
|
|
}
|
|
config->tracemalloc = nframe;
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_pycache_prefix(_PyCoreConfig *config)
|
|
{
|
|
assert(config->pycache_prefix == NULL);
|
|
|
|
const wchar_t *xoption = config_get_xoption(config, L"pycache_prefix");
|
|
if (xoption) {
|
|
const wchar_t *sep = wcschr(xoption, L'=');
|
|
if (sep && wcslen(sep) > 1) {
|
|
config->pycache_prefix = _PyMem_RawWcsdup(sep + 1);
|
|
if (config->pycache_prefix == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
else {
|
|
// -X pycache_prefix= can cancel the env var
|
|
config->pycache_prefix = NULL;
|
|
}
|
|
}
|
|
else {
|
|
wchar_t *env;
|
|
int res = _PyCoreConfig_GetEnvDup(config, &env,
|
|
L"PYTHONPYCACHEPREFIX",
|
|
"PYTHONPYCACHEPREFIX");
|
|
if (res < 0) {
|
|
return DECODE_LOCALE_ERR("PYTHONPYCACHEPREFIX", res);
|
|
}
|
|
|
|
if (env) {
|
|
config->pycache_prefix = env;
|
|
}
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_read_complex_options(_PyCoreConfig *config)
|
|
{
|
|
/* More complex options configured by env var and -X option */
|
|
if (config->faulthandler < 0) {
|
|
if (_PyCoreConfig_GetEnv(config, "PYTHONFAULTHANDLER")
|
|
|| config_get_xoption(config, L"faulthandler")) {
|
|
config->faulthandler = 1;
|
|
}
|
|
}
|
|
if (_PyCoreConfig_GetEnv(config, "PYTHONPROFILEIMPORTTIME")
|
|
|| config_get_xoption(config, L"importtime")) {
|
|
config->import_time = 1;
|
|
}
|
|
if (config_get_xoption(config, L"dev" ) ||
|
|
_PyCoreConfig_GetEnv(config, "PYTHONDEVMODE"))
|
|
{
|
|
config->dev_mode = 1;
|
|
}
|
|
|
|
_PyInitError err;
|
|
if (config->tracemalloc < 0) {
|
|
err = config_init_tracemalloc(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
if (config->pycache_prefix == NULL) {
|
|
err = config_init_pycache_prefix(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static void
|
|
config_init_locale(_PyCoreConfig *config)
|
|
{
|
|
/* Test also if coerce_c_locale equals 1: PYTHONCOERCECLOCALE=1 doesn't
|
|
imply that the C locale is always coerced. It is only coerced if
|
|
if the LC_CTYPE locale is "C". */
|
|
if (config->coerce_c_locale != 0) {
|
|
/* The C locale enables the C locale coercion (PEP 538) */
|
|
if (_Py_LegacyLocaleDetected()) {
|
|
config->coerce_c_locale = 1;
|
|
}
|
|
else {
|
|
config->coerce_c_locale = 0;
|
|
}
|
|
}
|
|
|
|
#ifndef MS_WINDOWS
|
|
if (config->utf8_mode < 0) {
|
|
/* The C locale and the POSIX locale enable the UTF-8 Mode (PEP 540) */
|
|
const char *ctype_loc = setlocale(LC_CTYPE, NULL);
|
|
if (ctype_loc != NULL
|
|
&& (strcmp(ctype_loc, "C") == 0
|
|
|| strcmp(ctype_loc, "POSIX") == 0))
|
|
{
|
|
config->utf8_mode = 1;
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
static const char *
|
|
get_stdio_errors(const _PyCoreConfig *config)
|
|
{
|
|
#ifndef MS_WINDOWS
|
|
const char *loc = setlocale(LC_CTYPE, NULL);
|
|
if (loc != NULL) {
|
|
/* surrogateescape is the default in the legacy C and POSIX locales */
|
|
if (strcmp(loc, "C") == 0 || strcmp(loc, "POSIX") == 0) {
|
|
return "surrogateescape";
|
|
}
|
|
|
|
#ifdef PY_COERCE_C_LOCALE
|
|
/* surrogateescape is the default in locale coercion target locales */
|
|
if (_Py_IsLocaleCoercionTarget(loc)) {
|
|
return "surrogateescape";
|
|
}
|
|
#endif
|
|
}
|
|
|
|
return "strict";
|
|
#else
|
|
/* On Windows, always use surrogateescape by default */
|
|
return "surrogateescape";
|
|
#endif
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
get_locale_encoding(char **locale_encoding)
|
|
{
|
|
#ifdef MS_WINDOWS
|
|
char encoding[20];
|
|
PyOS_snprintf(encoding, sizeof(encoding), "cp%d", GetACP());
|
|
#elif defined(__ANDROID__)
|
|
const char *encoding = "UTF-8";
|
|
#else
|
|
const char *encoding = nl_langinfo(CODESET);
|
|
if (!encoding || encoding[0] == '\0') {
|
|
return _Py_INIT_USER_ERR("failed to get the locale encoding: "
|
|
"nl_langinfo(CODESET) failed");
|
|
}
|
|
#endif
|
|
*locale_encoding = _PyMem_RawStrdup(encoding);
|
|
if (*locale_encoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_stdio_encoding(_PyCoreConfig *config)
|
|
{
|
|
/* If Py_SetStandardStreamEncoding() have been called, use these
|
|
parameters. */
|
|
if (config->stdio_encoding == NULL && _Py_StandardStreamEncoding != NULL) {
|
|
config->stdio_encoding = _PyMem_RawStrdup(_Py_StandardStreamEncoding);
|
|
if (config->stdio_encoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
if (config->stdio_errors == NULL && _Py_StandardStreamErrors != NULL) {
|
|
config->stdio_errors = _PyMem_RawStrdup(_Py_StandardStreamErrors);
|
|
if (config->stdio_errors == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
if (config->stdio_encoding != NULL && config->stdio_errors != NULL) {
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
/* PYTHONIOENCODING environment variable */
|
|
const char *opt = _PyCoreConfig_GetEnv(config, "PYTHONIOENCODING");
|
|
if (opt) {
|
|
char *pythonioencoding = _PyMem_RawStrdup(opt);
|
|
if (pythonioencoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
|
|
char *err = strchr(pythonioencoding, ':');
|
|
if (err) {
|
|
*err = '\0';
|
|
err++;
|
|
if (!err[0]) {
|
|
err = NULL;
|
|
}
|
|
}
|
|
|
|
/* Does PYTHONIOENCODING contain an encoding? */
|
|
if (pythonioencoding[0]) {
|
|
if (config->stdio_encoding == NULL) {
|
|
config->stdio_encoding = _PyMem_RawStrdup(pythonioencoding);
|
|
if (config->stdio_encoding == NULL) {
|
|
PyMem_RawFree(pythonioencoding);
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
/* If the encoding is set but not the error handler,
|
|
use "strict" error handler by default.
|
|
PYTHONIOENCODING=latin1 behaves as
|
|
PYTHONIOENCODING=latin1:strict. */
|
|
if (!err) {
|
|
err = "strict";
|
|
}
|
|
}
|
|
|
|
if (config->stdio_errors == NULL && err != NULL) {
|
|
config->stdio_errors = _PyMem_RawStrdup(err);
|
|
if (config->stdio_errors == NULL) {
|
|
PyMem_RawFree(pythonioencoding);
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
PyMem_RawFree(pythonioencoding);
|
|
}
|
|
|
|
/* UTF-8 Mode uses UTF-8/surrogateescape */
|
|
if (config->utf8_mode) {
|
|
if (config->stdio_encoding == NULL) {
|
|
config->stdio_encoding = _PyMem_RawStrdup("utf-8");
|
|
if (config->stdio_encoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
if (config->stdio_errors == NULL) {
|
|
config->stdio_errors = _PyMem_RawStrdup("surrogateescape");
|
|
if (config->stdio_errors == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Choose the default error handler based on the current locale. */
|
|
if (config->stdio_encoding == NULL) {
|
|
_PyInitError err = get_locale_encoding(&config->stdio_encoding);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
if (config->stdio_errors == NULL) {
|
|
const char *errors = get_stdio_errors(config);
|
|
config->stdio_errors = _PyMem_RawStrdup(errors);
|
|
if (config->stdio_errors == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
static _PyInitError
|
|
config_init_fs_encoding(_PyCoreConfig *config)
|
|
{
|
|
#ifdef MS_WINDOWS
|
|
if (config->legacy_windows_fs_encoding) {
|
|
/* Legacy Windows filesystem encoding: mbcs/replace */
|
|
if (config->filesystem_encoding == NULL) {
|
|
config->filesystem_encoding = _PyMem_RawStrdup("mbcs");
|
|
if (config->filesystem_encoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
if (config->filesystem_errors == NULL) {
|
|
config->filesystem_errors = _PyMem_RawStrdup("replace");
|
|
if (config->filesystem_errors == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Windows defaults to utf-8/surrogatepass (PEP 529).
|
|
|
|
Note: UTF-8 Mode takes the same code path and the Legacy Windows FS
|
|
encoding has the priortiy over UTF-8 Mode. */
|
|
if (config->filesystem_encoding == NULL) {
|
|
config->filesystem_encoding = _PyMem_RawStrdup("utf-8");
|
|
if (config->filesystem_encoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
if (config->filesystem_errors == NULL) {
|
|
config->filesystem_errors = _PyMem_RawStrdup("surrogatepass");
|
|
if (config->filesystem_errors == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
#else
|
|
if (config->filesystem_encoding == NULL) {
|
|
if (config->utf8_mode) {
|
|
/* UTF-8 Mode use: utf-8/surrogateescape */
|
|
config->filesystem_encoding = _PyMem_RawStrdup("utf-8");
|
|
/* errors defaults to surrogateescape above */
|
|
}
|
|
else if (_Py_GetForceASCII()) {
|
|
config->filesystem_encoding = _PyMem_RawStrdup("ascii");
|
|
}
|
|
else {
|
|
/* macOS and Android use UTF-8,
|
|
other platforms use the locale encoding. */
|
|
#if defined(__APPLE__) || defined(__ANDROID__)
|
|
config->filesystem_encoding = _PyMem_RawStrdup("utf-8");
|
|
#else
|
|
_PyInitError err = get_locale_encoding(&config->filesystem_encoding);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
if (config->filesystem_encoding == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
|
|
if (config->filesystem_errors == NULL) {
|
|
/* by default, use the "surrogateescape" error handler */
|
|
config->filesystem_errors = _PyMem_RawStrdup("surrogateescape");
|
|
if (config->filesystem_errors == NULL) {
|
|
return _Py_INIT_NO_MEMORY();
|
|
}
|
|
}
|
|
#endif
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
/* Read configuration settings from standard locations
|
|
*
|
|
* This function doesn't make any changes to the interpreter state - it
|
|
* merely populates any missing configuration settings. This allows an
|
|
* embedding application to completely override a config option by
|
|
* setting it before calling this function, or else modify the default
|
|
* setting before passing the fully populated config to Py_EndInitialization.
|
|
*
|
|
* More advanced selective initialization tricks are possible by calling
|
|
* this function multiple times with various preconfigured settings.
|
|
*/
|
|
|
|
_PyInitError
|
|
_PyCoreConfig_Read(_PyCoreConfig *config)
|
|
{
|
|
_PyInitError err;
|
|
|
|
_PyCoreConfig_GetGlobalConfig(config);
|
|
assert(config->use_environment >= 0);
|
|
|
|
if (config->isolated > 0) {
|
|
config->use_environment = 0;
|
|
config->user_site_directory = 0;
|
|
}
|
|
|
|
#ifdef MS_WINDOWS
|
|
if (config->legacy_windows_fs_encoding) {
|
|
config->utf8_mode = 0;
|
|
}
|
|
#endif
|
|
|
|
if (config->use_environment) {
|
|
err = config_read_env_vars(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
/* -X options */
|
|
if (config_get_xoption(config, L"showrefcount")) {
|
|
config->show_ref_count = 1;
|
|
}
|
|
if (config_get_xoption(config, L"showalloccount")) {
|
|
config->show_alloc_count = 1;
|
|
}
|
|
|
|
err = config_read_complex_options(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
|
|
if (config->utf8_mode < 0) {
|
|
err = config_init_utf8_mode(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
if (config->home == NULL) {
|
|
err = config_init_home(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
if (config->program_name == NULL) {
|
|
err = config_init_program_name(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
if (config->executable == NULL) {
|
|
err = config_init_executable(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
if (config->coerce_c_locale != 0 || config->utf8_mode < 0) {
|
|
config_init_locale(config);
|
|
}
|
|
|
|
if (config->_install_importlib) {
|
|
err = _PyCoreConfig_InitPathConfig(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
/* default values */
|
|
if (config->dev_mode) {
|
|
if (config->faulthandler < 0) {
|
|
config->faulthandler = 1;
|
|
}
|
|
if (config->allocator == NULL) {
|
|
config->allocator = "debug";
|
|
}
|
|
}
|
|
if (config->use_hash_seed < 0) {
|
|
config->use_hash_seed = 0;
|
|
config->hash_seed = 0;
|
|
}
|
|
if (config->faulthandler < 0) {
|
|
config->faulthandler = 0;
|
|
}
|
|
if (config->tracemalloc < 0) {
|
|
config->tracemalloc = 0;
|
|
}
|
|
if (config->coerce_c_locale < 0) {
|
|
config->coerce_c_locale = 0;
|
|
}
|
|
if (config->utf8_mode < 0) {
|
|
config->utf8_mode = 0;
|
|
}
|
|
if (config->argc < 0) {
|
|
config->argc = 0;
|
|
}
|
|
|
|
if (config->filesystem_encoding == NULL || config->filesystem_errors == NULL) {
|
|
err = config_init_fs_encoding(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
}
|
|
|
|
err = config_init_stdio_encoding(config);
|
|
if (_Py_INIT_FAILED(err)) {
|
|
return err;
|
|
}
|
|
|
|
assert(config->coerce_c_locale >= 0);
|
|
assert(config->use_environment >= 0);
|
|
assert(config->filesystem_encoding != NULL);
|
|
assert(config->filesystem_errors != NULL);
|
|
assert(config->stdio_encoding != NULL);
|
|
assert(config->stdio_errors != NULL);
|
|
assert(config->_check_hash_pycs_mode != NULL);
|
|
|
|
return _Py_INIT_OK();
|
|
}
|
|
|
|
|
|
PyObject *
|
|
_PyCoreConfig_AsDict(const _PyCoreConfig *config)
|
|
{
|
|
PyObject *dict, *obj;
|
|
|
|
dict = PyDict_New();
|
|
if (dict == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
#define SET_ITEM(KEY, EXPR) \
|
|
do { \
|
|
obj = (EXPR); \
|
|
if (obj == NULL) { \
|
|
return NULL; \
|
|
} \
|
|
int res = PyDict_SetItemString(dict, (KEY), obj); \
|
|
Py_DECREF(obj); \
|
|
if (res < 0) { \
|
|
goto fail; \
|
|
} \
|
|
} while (0)
|
|
#define FROM_STRING(STR) \
|
|
((STR != NULL) ? \
|
|
PyUnicode_FromString(STR) \
|
|
: (Py_INCREF(Py_None), Py_None))
|
|
#define SET_ITEM_INT(ATTR) \
|
|
SET_ITEM(#ATTR, PyLong_FromLong(config->ATTR))
|
|
#define SET_ITEM_UINT(ATTR) \
|
|
SET_ITEM(#ATTR, PyLong_FromUnsignedLong(config->ATTR))
|
|
#define SET_ITEM_STR(ATTR) \
|
|
SET_ITEM(#ATTR, FROM_STRING(config->ATTR))
|
|
#define FROM_WSTRING(STR) \
|
|
((STR != NULL) ? \
|
|
PyUnicode_FromWideChar(STR, -1) \
|
|
: (Py_INCREF(Py_None), Py_None))
|
|
#define SET_ITEM_WSTR(ATTR) \
|
|
SET_ITEM(#ATTR, FROM_WSTRING(config->ATTR))
|
|
#define SET_ITEM_WSTRLIST(NOPTION, OPTIONS) \
|
|
SET_ITEM(#OPTIONS, _Py_wstrlist_as_pylist(config->NOPTION, config->OPTIONS))
|
|
|
|
SET_ITEM_INT(install_signal_handlers);
|
|
SET_ITEM_INT(use_environment);
|
|
SET_ITEM_INT(use_hash_seed);
|
|
SET_ITEM_UINT(hash_seed);
|
|
SET_ITEM_STR(allocator);
|
|
SET_ITEM_INT(dev_mode);
|
|
SET_ITEM_INT(faulthandler);
|
|
SET_ITEM_INT(tracemalloc);
|
|
SET_ITEM_INT(import_time);
|
|
SET_ITEM_INT(show_ref_count);
|
|
SET_ITEM_INT(show_alloc_count);
|
|
SET_ITEM_INT(dump_refs);
|
|
SET_ITEM_INT(malloc_stats);
|
|
SET_ITEM_INT(coerce_c_locale);
|
|
SET_ITEM_INT(coerce_c_locale_warn);
|
|
SET_ITEM_STR(filesystem_encoding);
|
|
SET_ITEM_STR(filesystem_errors);
|
|
SET_ITEM_INT(utf8_mode);
|
|
SET_ITEM_WSTR(pycache_prefix);
|
|
SET_ITEM_WSTR(program_name);
|
|
SET_ITEM_WSTRLIST(argc, argv);
|
|
SET_ITEM_WSTR(program);
|
|
SET_ITEM_WSTRLIST(nxoption, xoptions);
|
|
SET_ITEM_WSTRLIST(nwarnoption, warnoptions);
|
|
SET_ITEM_WSTR(module_search_path_env);
|
|
SET_ITEM_WSTR(home);
|
|
SET_ITEM_WSTRLIST(nmodule_search_path, module_search_paths);
|
|
SET_ITEM_WSTR(executable);
|
|
SET_ITEM_WSTR(prefix);
|
|
SET_ITEM_WSTR(base_prefix);
|
|
SET_ITEM_WSTR(exec_prefix);
|
|
SET_ITEM_WSTR(base_exec_prefix);
|
|
#ifdef MS_WINDOWS
|
|
SET_ITEM_WSTR(dll_path);
|
|
#endif
|
|
SET_ITEM_INT(isolated);
|
|
SET_ITEM_INT(site_import);
|
|
SET_ITEM_INT(bytes_warning);
|
|
SET_ITEM_INT(inspect);
|
|
SET_ITEM_INT(interactive);
|
|
SET_ITEM_INT(optimization_level);
|
|
SET_ITEM_INT(parser_debug);
|
|
SET_ITEM_INT(write_bytecode);
|
|
SET_ITEM_INT(verbose);
|
|
SET_ITEM_INT(quiet);
|
|
SET_ITEM_INT(user_site_directory);
|
|
SET_ITEM_INT(buffered_stdio);
|
|
SET_ITEM_STR(stdio_encoding);
|
|
SET_ITEM_STR(stdio_errors);
|
|
#ifdef MS_WINDOWS
|
|
SET_ITEM_INT(legacy_windows_fs_encoding);
|
|
SET_ITEM_INT(legacy_windows_stdio);
|
|
#endif
|
|
SET_ITEM_INT(_install_importlib);
|
|
SET_ITEM_STR(_check_hash_pycs_mode);
|
|
SET_ITEM_INT(_frozen);
|
|
|
|
return dict;
|
|
|
|
fail:
|
|
Py_DECREF(dict);
|
|
return NULL;
|
|
|
|
#undef FROM_STRING
|
|
#undef FROM_WSTRING
|
|
#undef SET_ITEM
|
|
#undef SET_ITEM_INT
|
|
#undef SET_ITEM_UINT
|
|
#undef SET_ITEM_STR
|
|
#undef SET_ITEM_WSTR
|
|
#undef SET_ITEM_WSTRLIST
|
|
}
|