mirror of
https://github.com/python/cpython.git
synced 2024-11-30 18:51:15 +01:00
5d39e04290
* Fix _PyMem_SetupAllocators("debug"): always restore allocators to the defaults, rather than only caling _PyMem_SetupDebugHooks(). * Add _PyMem_SetDefaultAllocator() helper to set the "default" allocator. * Add _PyMem_GetAllocatorsName(): get the name of the allocators * main() now uses debug hooks on memory allocators if Py_DEBUG is defined, rather than calling directly malloc() * Document default memory allocators in C API documentation * _Py_InitializeCore() now fails with a fatal user error if PYTHONMALLOC value is an unknown memory allocator, instead of failing with a fatal internal error. * Add new tests on the PYTHONMALLOC environment variable * Add support.with_pymalloc() * Add the _testcapi.WITH_PYMALLOC constant and expose it as support.with_pymalloc(). * sysconfig.get_config_var('WITH_PYMALLOC') doesn't work on Windows, so replace it with support.with_pymalloc(). * pythoninfo: add _testcapi collector for pymem
110 lines
3.1 KiB
C
110 lines
3.1 KiB
C
/* Minimal main program -- everything is loaded from the library */
|
|
|
|
#include "Python.h"
|
|
#include "internal/pystate.h"
|
|
#include <locale.h>
|
|
|
|
#ifdef __FreeBSD__
|
|
#include <fenv.h>
|
|
#endif
|
|
|
|
#ifdef MS_WINDOWS
|
|
int
|
|
wmain(int argc, wchar_t **argv)
|
|
{
|
|
return Py_Main(argc, argv);
|
|
}
|
|
#else
|
|
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
wchar_t **argv_copy;
|
|
/* We need a second copy, as Python might modify the first one. */
|
|
wchar_t **argv_copy2;
|
|
int i, status;
|
|
char *oldloc;
|
|
|
|
_PyInitError err = _PyRuntime_Initialize();
|
|
if (_Py_INIT_FAILED(err)) {
|
|
fprintf(stderr, "Fatal Python error: %s\n", err.msg);
|
|
fflush(stderr);
|
|
exit(1);
|
|
}
|
|
|
|
/* Force default allocator, to be able to release memory above
|
|
with a known allocator. */
|
|
_PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, NULL);
|
|
|
|
argv_copy = (wchar_t **)PyMem_RawMalloc(sizeof(wchar_t*) * (argc+1));
|
|
argv_copy2 = (wchar_t **)PyMem_RawMalloc(sizeof(wchar_t*) * (argc+1));
|
|
if (!argv_copy || !argv_copy2) {
|
|
fprintf(stderr, "out of memory\n");
|
|
return 1;
|
|
}
|
|
|
|
/* 754 requires that FP exceptions run in "no stop" mode by default,
|
|
* and until C vendors implement C99's ways to control FP exceptions,
|
|
* Python requires non-stop mode. Alas, some platforms enable FP
|
|
* exceptions by default. Here we disable them.
|
|
*/
|
|
#ifdef __FreeBSD__
|
|
fedisableexcept(FE_OVERFLOW);
|
|
#endif
|
|
|
|
oldloc = _PyMem_RawStrdup(setlocale(LC_ALL, NULL));
|
|
if (!oldloc) {
|
|
fprintf(stderr, "out of memory\n");
|
|
return 1;
|
|
}
|
|
|
|
/* Reconfigure the locale to the default for this process */
|
|
_Py_SetLocaleFromEnv(LC_ALL);
|
|
|
|
/* The legacy C locale assumes ASCII as the default text encoding, which
|
|
* causes problems not only for the CPython runtime, but also other
|
|
* components like GNU readline.
|
|
*
|
|
* Accordingly, when the CLI detects it, it attempts to coerce it to a
|
|
* more capable UTF-8 based alternative.
|
|
*
|
|
* See the documentation of the PYTHONCOERCECLOCALE setting for more
|
|
* details.
|
|
*/
|
|
if (_Py_LegacyLocaleDetected()) {
|
|
_Py_CoerceLegacyLocale();
|
|
}
|
|
|
|
/* Convert from char to wchar_t based on the locale settings */
|
|
for (i = 0; i < argc; i++) {
|
|
argv_copy[i] = Py_DecodeLocale(argv[i], NULL);
|
|
if (!argv_copy[i]) {
|
|
PyMem_RawFree(oldloc);
|
|
fprintf(stderr, "Fatal Python error: "
|
|
"unable to decode the command line argument #%i\n",
|
|
i + 1);
|
|
return 1;
|
|
}
|
|
argv_copy2[i] = argv_copy[i];
|
|
}
|
|
argv_copy2[argc] = argv_copy[argc] = NULL;
|
|
|
|
setlocale(LC_ALL, oldloc);
|
|
PyMem_RawFree(oldloc);
|
|
|
|
status = Py_Main(argc, argv_copy);
|
|
|
|
/* Py_Main() can change PyMem_RawMalloc() allocator, so restore the default
|
|
to release memory blocks allocated before Py_Main() */
|
|
_PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, NULL);
|
|
|
|
for (i = 0; i < argc; i++) {
|
|
PyMem_RawFree(argv_copy2[i]);
|
|
}
|
|
PyMem_RawFree(argv_copy);
|
|
PyMem_RawFree(argv_copy2);
|
|
return status;
|
|
}
|
|
#endif
|