blob: bc549369393fc798ef35c87955c74cb4d6f8c190 [file] [log] [blame]
Victor Stinner91c99872019-05-14 22:01:51 +02001/* FIXME: PEP 587 makes these functions public */
2#ifndef Py_BUILD_CORE_MODULE
3# define Py_BUILD_CORE_MODULE
4#endif
5
Antoine Pitrou8e605772011-04-25 21:21:07 +02006#include <Python.h>
Victor Stinner91c99872019-05-14 22:01:51 +02007#include "pycore_coreconfig.h" /* FIXME: PEP 587 makes these functions public */
Victor Stinnerbab0db62019-05-18 03:21:27 +02008#include <Python.h>
Victor Stinnerb4d1e1f2017-11-30 22:05:00 +01009#include "pythread.h"
Eric Snowe3774162017-05-22 19:46:40 -070010#include <inttypes.h>
Antoine Pitrou8e605772011-04-25 21:21:07 +020011#include <stdio.h>
Nick Coghlanbc77eff2018-03-25 20:44:30 +100012#include <wchar.h>
Antoine Pitrou8e605772011-04-25 21:21:07 +020013
Nick Coghlan7d270ee2013-10-17 22:35:35 +100014/*********************************************************
15 * Embedded interpreter tests that need a custom exe
16 *
17 * Executed via 'EmbeddingTests' in Lib/test/test_capi.py
18 *********************************************************/
19
20static void _testembed_Py_Initialize(void)
21{
22 /* HACK: the "./" at front avoids a search along the PATH in
23 Modules/getpath.c */
24 Py_SetProgramName(L"./_testembed");
25 Py_Initialize();
26}
27
28
29/*****************************************************
Martin Panter8f265652016-04-19 04:03:41 +000030 * Test repeated initialisation and subinterpreters
Nick Coghlan7d270ee2013-10-17 22:35:35 +100031 *****************************************************/
32
33static void print_subinterp(void)
Antoine Pitrou8e605772011-04-25 21:21:07 +020034{
Eric Snowe3774162017-05-22 19:46:40 -070035 /* Output information about the interpreter in the format
36 expected in Lib/test/test_capi.py (test_subinterps). */
Antoine Pitrou8e605772011-04-25 21:21:07 +020037 PyThreadState *ts = PyThreadState_Get();
Eric Snowe3774162017-05-22 19:46:40 -070038 PyInterpreterState *interp = ts->interp;
39 int64_t id = PyInterpreterState_GetID(interp);
Eric Snowd1c3c132017-05-24 17:19:47 -070040 printf("interp %" PRId64 " <0x%" PRIXPTR ">, thread state <0x%" PRIXPTR ">: ",
Eric Snowe3774162017-05-22 19:46:40 -070041 id, (uintptr_t)interp, (uintptr_t)ts);
Antoine Pitrou8e605772011-04-25 21:21:07 +020042 fflush(stdout);
43 PyRun_SimpleString(
44 "import sys;"
45 "print('id(modules) =', id(sys.modules));"
46 "sys.stdout.flush()"
47 );
48}
49
Steve Dowerea74f0c2017-01-01 20:25:03 -080050static int test_repeated_init_and_subinterpreters(void)
Antoine Pitrou8e605772011-04-25 21:21:07 +020051{
52 PyThreadState *mainstate, *substate;
53 PyGILState_STATE gilstate;
54 int i, j;
55
Ned Deily939231b2016-08-16 00:17:42 -040056 for (i=0; i<15; i++) {
Antoine Pitrou8e605772011-04-25 21:21:07 +020057 printf("--- Pass %d ---\n", i);
Nick Coghlan7d270ee2013-10-17 22:35:35 +100058 _testembed_Py_Initialize();
Antoine Pitrou8e605772011-04-25 21:21:07 +020059 mainstate = PyThreadState_Get();
60
61 PyEval_InitThreads();
62 PyEval_ReleaseThread(mainstate);
63
64 gilstate = PyGILState_Ensure();
65 print_subinterp();
66 PyThreadState_Swap(NULL);
67
68 for (j=0; j<3; j++) {
69 substate = Py_NewInterpreter();
70 print_subinterp();
71 Py_EndInterpreter(substate);
72 }
73
74 PyThreadState_Swap(mainstate);
75 print_subinterp();
76 PyGILState_Release(gilstate);
Antoine Pitrou8e605772011-04-25 21:21:07 +020077
78 PyEval_RestoreThread(mainstate);
79 Py_Finalize();
80 }
Steve Dowerea74f0c2017-01-01 20:25:03 -080081 return 0;
Nick Coghlan7d270ee2013-10-17 22:35:35 +100082}
83
84/*****************************************************
85 * Test forcing a particular IO encoding
86 *****************************************************/
87
88static void check_stdio_details(const char *encoding, const char * errors)
89{
90 /* Output info for the test case to check */
91 if (encoding) {
92 printf("Expected encoding: %s\n", encoding);
93 } else {
94 printf("Expected encoding: default\n");
95 }
96 if (errors) {
97 printf("Expected errors: %s\n", errors);
98 } else {
99 printf("Expected errors: default\n");
100 }
101 fflush(stdout);
102 /* Force the given IO encoding */
103 Py_SetStandardStreamEncoding(encoding, errors);
104 _testembed_Py_Initialize();
105 PyRun_SimpleString(
106 "import sys;"
107 "print('stdin: {0.encoding}:{0.errors}'.format(sys.stdin));"
108 "print('stdout: {0.encoding}:{0.errors}'.format(sys.stdout));"
109 "print('stderr: {0.encoding}:{0.errors}'.format(sys.stderr));"
110 "sys.stdout.flush()"
111 );
112 Py_Finalize();
113}
114
Steve Dowerea74f0c2017-01-01 20:25:03 -0800115static int test_forced_io_encoding(void)
Nick Coghlan7d270ee2013-10-17 22:35:35 +1000116{
117 /* Check various combinations */
118 printf("--- Use defaults ---\n");
119 check_stdio_details(NULL, NULL);
120 printf("--- Set errors only ---\n");
Victor Stinnerb2bef622014-03-18 02:38:12 +0100121 check_stdio_details(NULL, "ignore");
Nick Coghlan7d270ee2013-10-17 22:35:35 +1000122 printf("--- Set encoding only ---\n");
Victor Stinner9e4994d2018-08-28 23:26:33 +0200123 check_stdio_details("iso8859-1", NULL);
Nick Coghlan7d270ee2013-10-17 22:35:35 +1000124 printf("--- Set encoding and errors ---\n");
Victor Stinner9e4994d2018-08-28 23:26:33 +0200125 check_stdio_details("iso8859-1", "replace");
Nick Coghlan7d270ee2013-10-17 22:35:35 +1000126
127 /* Check calling after initialization fails */
128 Py_Initialize();
129
130 if (Py_SetStandardStreamEncoding(NULL, NULL) == 0) {
131 printf("Unexpected success calling Py_SetStandardStreamEncoding");
132 }
133 Py_Finalize();
Steve Dowerea74f0c2017-01-01 20:25:03 -0800134 return 0;
Nick Coghlan7d270ee2013-10-17 22:35:35 +1000135}
136
Victor Stinner9e87e772017-11-24 12:09:24 +0100137/*********************************************************
138 * Test parts of the C-API that work before initialization
139 *********************************************************/
140
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000141/* The pre-initialization tests tend to break by segfaulting, so explicitly
142 * flushed progress messages make the broken API easier to find when they fail.
143 */
144#define _Py_EMBED_PREINIT_CHECK(msg) \
145 do {printf(msg); fflush(stdout);} while (0);
146
Victor Stinner9e87e772017-11-24 12:09:24 +0100147static int test_pre_initialization_api(void)
148{
Victor Stinnera9df6512019-03-05 23:31:54 +0100149 /* the test doesn't support custom memory allocators */
150 putenv("PYTHONMALLOC=");
151
Nick Coghlan42746092017-11-26 14:19:13 +1000152 /* Leading "./" ensures getpath.c can still find the standard library */
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000153 _Py_EMBED_PREINIT_CHECK("Checking Py_DecodeLocale\n");
Nick Coghlan42746092017-11-26 14:19:13 +1000154 wchar_t *program = Py_DecodeLocale("./spam", NULL);
Victor Stinner9e87e772017-11-24 12:09:24 +0100155 if (program == NULL) {
156 fprintf(stderr, "Fatal error: cannot decode program name\n");
157 return 1;
158 }
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000159 _Py_EMBED_PREINIT_CHECK("Checking Py_SetProgramName\n");
Victor Stinner9e87e772017-11-24 12:09:24 +0100160 Py_SetProgramName(program);
161
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000162 _Py_EMBED_PREINIT_CHECK("Initializing interpreter\n");
Victor Stinner9e87e772017-11-24 12:09:24 +0100163 Py_Initialize();
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000164 _Py_EMBED_PREINIT_CHECK("Check sys module contents\n");
165 PyRun_SimpleString("import sys; "
166 "print('sys.executable:', sys.executable)");
167 _Py_EMBED_PREINIT_CHECK("Finalizing interpreter\n");
Victor Stinner9e87e772017-11-24 12:09:24 +0100168 Py_Finalize();
169
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000170 _Py_EMBED_PREINIT_CHECK("Freeing memory allocated by Py_DecodeLocale\n");
Victor Stinner9e87e772017-11-24 12:09:24 +0100171 PyMem_RawFree(program);
172 return 0;
173}
174
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000175
176/* bpo-33042: Ensure embedding apps can predefine sys module options */
177static int test_pre_initialization_sys_options(void)
178{
Nick Coghlan69f5c732018-03-30 15:36:42 +1000179 /* We allocate a couple of the options dynamically, and then delete
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000180 * them before calling Py_Initialize. This ensures the interpreter isn't
181 * relying on the caller to keep the passed in strings alive.
182 */
Nick Coghlan69f5c732018-03-30 15:36:42 +1000183 const wchar_t *static_warnoption = L"once";
184 const wchar_t *static_xoption = L"also_not_an_option=2";
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000185 size_t warnoption_len = wcslen(static_warnoption);
186 size_t xoption_len = wcslen(static_xoption);
Nick Coghlan69f5c732018-03-30 15:36:42 +1000187 wchar_t *dynamic_once_warnoption = \
188 (wchar_t *) calloc(warnoption_len+1, sizeof(wchar_t));
189 wchar_t *dynamic_xoption = \
190 (wchar_t *) calloc(xoption_len+1, sizeof(wchar_t));
Nick Coghlanbc77eff2018-03-25 20:44:30 +1000191 wcsncpy(dynamic_once_warnoption, static_warnoption, warnoption_len+1);
192 wcsncpy(dynamic_xoption, static_xoption, xoption_len+1);
193
194 _Py_EMBED_PREINIT_CHECK("Checking PySys_AddWarnOption\n");
195 PySys_AddWarnOption(L"default");
196 _Py_EMBED_PREINIT_CHECK("Checking PySys_ResetWarnOptions\n");
197 PySys_ResetWarnOptions();
198 _Py_EMBED_PREINIT_CHECK("Checking PySys_AddWarnOption linked list\n");
199 PySys_AddWarnOption(dynamic_once_warnoption);
200 PySys_AddWarnOption(L"module");
201 PySys_AddWarnOption(L"default");
202 _Py_EMBED_PREINIT_CHECK("Checking PySys_AddXOption\n");
203 PySys_AddXOption(L"not_an_option=1");
204 PySys_AddXOption(dynamic_xoption);
205
206 /* Delete the dynamic options early */
207 free(dynamic_once_warnoption);
208 dynamic_once_warnoption = NULL;
209 free(dynamic_xoption);
210 dynamic_xoption = NULL;
211
212 _Py_EMBED_PREINIT_CHECK("Initializing interpreter\n");
213 _testembed_Py_Initialize();
214 _Py_EMBED_PREINIT_CHECK("Check sys module contents\n");
215 PyRun_SimpleString("import sys; "
216 "print('sys.warnoptions:', sys.warnoptions); "
217 "print('sys._xoptions:', sys._xoptions); "
218 "warnings = sys.modules['warnings']; "
219 "latest_filters = [f[0] for f in warnings.filters[:3]]; "
220 "print('warnings.filters[:3]:', latest_filters)");
221 _Py_EMBED_PREINIT_CHECK("Finalizing interpreter\n");
222 Py_Finalize();
223
224 return 0;
225}
226
227
228/* bpo-20891: Avoid race condition when initialising the GIL */
Victor Stinnerb4d1e1f2017-11-30 22:05:00 +0100229static void bpo20891_thread(void *lockp)
230{
231 PyThread_type_lock lock = *((PyThread_type_lock*)lockp);
232
233 PyGILState_STATE state = PyGILState_Ensure();
234 if (!PyGILState_Check()) {
235 fprintf(stderr, "PyGILState_Check failed!");
236 abort();
237 }
238
239 PyGILState_Release(state);
240
241 PyThread_release_lock(lock);
242
243 PyThread_exit_thread();
244}
245
246static int test_bpo20891(void)
247{
Victor Stinnera9df6512019-03-05 23:31:54 +0100248 /* the test doesn't support custom memory allocators */
249 putenv("PYTHONMALLOC=");
250
Victor Stinnerb4d1e1f2017-11-30 22:05:00 +0100251 /* bpo-20891: Calling PyGILState_Ensure in a non-Python thread before
252 calling PyEval_InitThreads() must not crash. PyGILState_Ensure() must
253 call PyEval_InitThreads() for us in this case. */
254 PyThread_type_lock lock = PyThread_allocate_lock();
255 if (!lock) {
256 fprintf(stderr, "PyThread_allocate_lock failed!");
257 return 1;
258 }
259
260 _testembed_Py_Initialize();
261
262 unsigned long thrd = PyThread_start_new_thread(bpo20891_thread, &lock);
263 if (thrd == PYTHREAD_INVALID_THREAD_ID) {
264 fprintf(stderr, "PyThread_start_new_thread failed!");
265 return 1;
266 }
267 PyThread_acquire_lock(lock, WAIT_LOCK);
268
269 Py_BEGIN_ALLOW_THREADS
270 /* wait until the thread exit */
271 PyThread_acquire_lock(lock, WAIT_LOCK);
272 Py_END_ALLOW_THREADS
273
274 PyThread_free_lock(lock);
275
276 return 0;
277}
278
Victor Stinner209abf72018-06-22 19:14:51 +0200279static int test_initialize_twice(void)
280{
281 _testembed_Py_Initialize();
282
283 /* bpo-33932: Calling Py_Initialize() twice should do nothing
284 * (and not crash!). */
285 Py_Initialize();
286
287 Py_Finalize();
288
289 return 0;
290}
291
Victor Stinnerfb47bca2018-07-20 17:34:23 +0200292static int test_initialize_pymain(void)
293{
294 wchar_t *argv[] = {L"PYTHON", L"-c",
Victor Stinnercab5d072019-05-17 19:01:14 +0200295 (L"import sys; "
296 L"print(f'Py_Main() after Py_Initialize: "
297 L"sys.argv={sys.argv}')"),
Victor Stinnerfb47bca2018-07-20 17:34:23 +0200298 L"arg2"};
299 _testembed_Py_Initialize();
300
301 /* bpo-34008: Calling Py_Main() after Py_Initialize() must not crash */
302 Py_Main(Py_ARRAY_LENGTH(argv), argv);
303
304 Py_Finalize();
305
306 return 0;
307}
308
Victor Stinner9e87e772017-11-24 12:09:24 +0100309
Victor Stinner00b137c2018-11-13 19:59:26 +0100310static void
311dump_config(void)
312{
Victor Stinner23bace22019-04-18 11:37:26 +0200313 (void) PyRun_SimpleStringFlags(
314 "import _testinternalcapi, json; "
315 "print(json.dumps(_testinternalcapi.get_configs()))",
316 0);
Victor Stinner00b137c2018-11-13 19:59:26 +0100317}
318
319
Victor Stinner56b29b62018-07-26 18:57:56 +0200320static int test_init_default_config(void)
321{
322 _testembed_Py_Initialize();
323 dump_config();
324 Py_Finalize();
325 return 0;
326}
327
328
329static int test_init_global_config(void)
330{
331 /* FIXME: test Py_IgnoreEnvironmentFlag */
332
333 putenv("PYTHONUTF8=0");
334 Py_UTF8Mode = 1;
335
336 /* Test initialization from global configuration variables (Py_xxx) */
337 Py_SetProgramName(L"./globalvar");
338
339 /* Py_IsolatedFlag is not tested */
340 Py_NoSiteFlag = 1;
341 Py_BytesWarningFlag = 1;
342
343 putenv("PYTHONINSPECT=");
344 Py_InspectFlag = 1;
345
346 putenv("PYTHONOPTIMIZE=0");
347 Py_InteractiveFlag = 1;
348
349 putenv("PYTHONDEBUG=0");
350 Py_OptimizeFlag = 2;
351
352 /* Py_DebugFlag is not tested */
353
354 putenv("PYTHONDONTWRITEBYTECODE=");
355 Py_DontWriteBytecodeFlag = 1;
356
357 putenv("PYTHONVERBOSE=0");
358 Py_VerboseFlag = 1;
359
360 Py_QuietFlag = 1;
361 Py_NoUserSiteDirectory = 1;
362
363 putenv("PYTHONUNBUFFERED=");
364 Py_UnbufferedStdioFlag = 1;
365
Victor Stinner54b43bb2019-05-16 18:30:15 +0200366 Py_FrozenFlag = 1;
367
Victor Stinner56b29b62018-07-26 18:57:56 +0200368 /* FIXME: test Py_LegacyWindowsFSEncodingFlag */
369 /* FIXME: test Py_LegacyWindowsStdioFlag */
370
Victor Stinner56b29b62018-07-26 18:57:56 +0200371 Py_Initialize();
372 dump_config();
373 Py_Finalize();
374 return 0;
375}
376
377
378static int test_init_from_config(void)
379{
Victor Stinner20004952019-03-26 02:31:11 +0100380 _PyInitError err;
381
Victor Stinnercab5d072019-05-17 19:01:14 +0200382 _PyPreConfig preconfig;
383 _PyPreConfig_Init(&preconfig);
Victor Stinner20004952019-03-26 02:31:11 +0100384
385 putenv("PYTHONMALLOC=malloc_debug");
Victor Stinnerb16b4e42019-05-17 15:20:52 +0200386 preconfig.allocator = PYMEM_ALLOCATOR_MALLOC;
Victor Stinner20004952019-03-26 02:31:11 +0100387
388 putenv("PYTHONUTF8=0");
389 Py_UTF8Mode = 0;
390 preconfig.utf8_mode = 1;
391
Victor Stinner5ac27a52019-03-27 13:40:14 +0100392 err = _Py_PreInitialize(&preconfig);
Victor Stinner871ff772019-05-17 23:54:00 +0200393 if (_PyInitError_Failed(err)) {
Victor Stinner20004952019-03-26 02:31:11 +0100394 _Py_ExitInitError(err);
395 }
396
Victor Stinner56b29b62018-07-26 18:57:56 +0200397 /* Test _Py_InitializeFromConfig() */
Victor Stinnercab5d072019-05-17 19:01:14 +0200398 _PyCoreConfig config;
399 _PyCoreConfig_Init(&config);
Victor Stinner56b29b62018-07-26 18:57:56 +0200400 config.install_signal_handlers = 0;
401
402 /* FIXME: test use_environment */
403
404 putenv("PYTHONHASHSEED=42");
405 config.use_hash_seed = 1;
406 config.hash_seed = 123;
407
Victor Stinnerbab0db62019-05-18 03:21:27 +0200408 /* dev_mode=1 is tested in test_init_dev_mode() */
Victor Stinner56b29b62018-07-26 18:57:56 +0200409
410 putenv("PYTHONFAULTHANDLER=");
411 config.faulthandler = 1;
412
413 putenv("PYTHONTRACEMALLOC=0");
414 config.tracemalloc = 2;
415
416 putenv("PYTHONPROFILEIMPORTTIME=0");
417 config.import_time = 1;
418
419 config.show_ref_count = 1;
420 config.show_alloc_count = 1;
421 /* FIXME: test dump_refs: bpo-34223 */
422
423 putenv("PYTHONMALLOCSTATS=0");
424 config.malloc_stats = 1;
425
Victor Stinner56b29b62018-07-26 18:57:56 +0200426 putenv("PYTHONPYCACHEPREFIX=env_pycache_prefix");
427 config.pycache_prefix = L"conf_pycache_prefix";
428
429 Py_SetProgramName(L"./globalvar");
430 config.program_name = L"./conf_program_name";
431
Victor Stinner2f549082019-03-29 15:13:46 +0100432 static wchar_t* argv[] = {
433 L"python3",
Victor Stinner01de89c2018-11-14 17:39:45 +0100434 L"-c",
435 L"pass",
Victor Stinner2f549082019-03-29 15:13:46 +0100436 L"arg2",
Victor Stinner01de89c2018-11-14 17:39:45 +0100437 };
Victor Stinner74f65682019-03-15 15:08:05 +0100438 config.argv.length = Py_ARRAY_LENGTH(argv);
439 config.argv.items = argv;
Victor Stinnercab5d072019-05-17 19:01:14 +0200440 config.parse_argv = 1;
Victor Stinner01de89c2018-11-14 17:39:45 +0100441
Victor Stinner01de89c2018-11-14 17:39:45 +0100442 static wchar_t* xoptions[3] = {
443 L"core_xoption1=3",
444 L"core_xoption2=",
445 L"core_xoption3",
446 };
Victor Stinner74f65682019-03-15 15:08:05 +0100447 config.xoptions.length = Py_ARRAY_LENGTH(xoptions);
448 config.xoptions.items = xoptions;
Victor Stinner01de89c2018-11-14 17:39:45 +0100449
Victor Stinnerf8ba6f52019-03-26 16:58:50 +0100450 static wchar_t* warnoptions[1] = {
Victor Stinner01de89c2018-11-14 17:39:45 +0100451 L"error::ResourceWarning",
452 };
Victor Stinner74f65682019-03-15 15:08:05 +0100453 config.warnoptions.length = Py_ARRAY_LENGTH(warnoptions);
454 config.warnoptions.items = warnoptions;
Victor Stinner01de89c2018-11-14 17:39:45 +0100455
Victor Stinner56b29b62018-07-26 18:57:56 +0200456 /* FIXME: test module_search_path_env */
457 /* FIXME: test home */
458 /* FIXME: test path config: module_search_path .. dll_path */
459
460 putenv("PYTHONVERBOSE=0");
461 Py_VerboseFlag = 0;
462 config.verbose = 1;
463
464 Py_NoSiteFlag = 0;
465 config.site_import = 0;
466
467 Py_BytesWarningFlag = 0;
468 config.bytes_warning = 1;
469
470 putenv("PYTHONINSPECT=");
471 Py_InspectFlag = 0;
472 config.inspect = 1;
473
474 Py_InteractiveFlag = 0;
475 config.interactive = 1;
476
477 putenv("PYTHONOPTIMIZE=0");
478 Py_OptimizeFlag = 1;
479 config.optimization_level = 2;
480
Victor Stinner98512272018-08-01 03:07:00 +0200481 /* FIXME: test parser_debug */
Victor Stinner56b29b62018-07-26 18:57:56 +0200482
483 putenv("PYTHONDONTWRITEBYTECODE=");
484 Py_DontWriteBytecodeFlag = 0;
485 config.write_bytecode = 0;
486
487 Py_QuietFlag = 0;
488 config.quiet = 1;
489
Victor Stinnercab5d072019-05-17 19:01:14 +0200490 config.configure_c_stdio = 1;
Victor Stinner54b43bb2019-05-16 18:30:15 +0200491
Victor Stinner56b29b62018-07-26 18:57:56 +0200492 putenv("PYTHONUNBUFFERED=");
493 Py_UnbufferedStdioFlag = 0;
Victor Stinner98512272018-08-01 03:07:00 +0200494 config.buffered_stdio = 0;
Victor Stinner56b29b62018-07-26 18:57:56 +0200495
Victor Stinnerdfe0dc72018-08-29 11:47:29 +0200496 putenv("PYTHONIOENCODING=cp424");
497 Py_SetStandardStreamEncoding("ascii", "ignore");
Victor Stinner01de89c2018-11-14 17:39:45 +0100498#ifdef MS_WINDOWS
499 /* Py_SetStandardStreamEncoding() sets Py_LegacyWindowsStdioFlag to 1.
500 Force it to 0 through the config. */
501 config.legacy_windows_stdio = 0;
502#endif
Victor Stinner709d23d2019-05-02 14:56:30 -0400503 config.stdio_encoding = L"iso8859-1";
504 config.stdio_errors = L"replace";
Victor Stinnerdfe0dc72018-08-29 11:47:29 +0200505
Victor Stinner56b29b62018-07-26 18:57:56 +0200506 putenv("PYTHONNOUSERSITE=");
507 Py_NoUserSiteDirectory = 0;
508 config.user_site_directory = 0;
509
Victor Stinnercb9fbd32019-05-01 23:51:56 -0400510 config.check_hash_pycs_mode = L"always";
Victor Stinner56b29b62018-07-26 18:57:56 +0200511
Victor Stinner54b43bb2019-05-16 18:30:15 +0200512 Py_FrozenFlag = 0;
513 config.pathconfig_warnings = 0;
514
Victor Stinner5ac27a52019-03-27 13:40:14 +0100515 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200516 if (_PyInitError_Failed(err)) {
Victor Stinnerdfe88472019-03-01 12:14:41 +0100517 _Py_ExitInitError(err);
Victor Stinner56b29b62018-07-26 18:57:56 +0200518 }
519 dump_config();
520 Py_Finalize();
521 return 0;
522}
523
524
Victor Stinnerbab0db62019-05-18 03:21:27 +0200525static int check_init_parse_argv(int parse_argv)
Victor Stinnerae239f62019-05-16 17:02:56 +0200526{
527 _PyInitError err;
528
Victor Stinnercab5d072019-05-17 19:01:14 +0200529 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200530 err = _PyCoreConfig_InitPythonConfig(&config);
531 if (_PyInitError_Failed(err)) {
532 _Py_ExitInitError(err);
533 }
Victor Stinnerae239f62019-05-16 17:02:56 +0200534
535 static wchar_t* argv[] = {
Victor Stinnercab5d072019-05-17 19:01:14 +0200536 L"./argv0",
537 L"-E",
Victor Stinnerae239f62019-05-16 17:02:56 +0200538 L"-c",
Victor Stinnercab5d072019-05-17 19:01:14 +0200539 L"pass",
Victor Stinnerae239f62019-05-16 17:02:56 +0200540 L"arg1",
Victor Stinnercab5d072019-05-17 19:01:14 +0200541 L"-v",
542 L"arg3",
Victor Stinnerae239f62019-05-16 17:02:56 +0200543 };
544
Victor Stinnerae239f62019-05-16 17:02:56 +0200545 config.argv.length = Py_ARRAY_LENGTH(argv);
546 config.argv.items = argv;
Victor Stinnercab5d072019-05-17 19:01:14 +0200547 config.parse_argv = parse_argv;
Victor Stinnerae239f62019-05-16 17:02:56 +0200548
549 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200550 if (_PyInitError_Failed(err)) {
Victor Stinnerae239f62019-05-16 17:02:56 +0200551 _Py_ExitInitError(err);
552 }
553 dump_config();
554 Py_Finalize();
555 return 0;
556}
557
558
Victor Stinnerbab0db62019-05-18 03:21:27 +0200559static int test_init_parse_argv(void)
Victor Stinnercab5d072019-05-17 19:01:14 +0200560{
Victor Stinnerbab0db62019-05-18 03:21:27 +0200561 return check_init_parse_argv(1);
Victor Stinnercab5d072019-05-17 19:01:14 +0200562}
563
564
Victor Stinnerbab0db62019-05-18 03:21:27 +0200565static int test_init_dont_parse_argv(void)
Victor Stinnercab5d072019-05-17 19:01:14 +0200566{
Victor Stinnerbab0db62019-05-18 03:21:27 +0200567 return check_init_parse_argv(0);
Victor Stinnercab5d072019-05-17 19:01:14 +0200568}
569
570
Victor Stinner425717f2019-05-20 16:38:48 +0200571static void set_most_env_vars(void)
Victor Stinner56b29b62018-07-26 18:57:56 +0200572{
573 putenv("PYTHONHASHSEED=42");
Victor Stinner25d13f32019-03-06 12:51:53 +0100574 putenv("PYTHONMALLOC=malloc");
Victor Stinner56b29b62018-07-26 18:57:56 +0200575 putenv("PYTHONTRACEMALLOC=2");
576 putenv("PYTHONPROFILEIMPORTTIME=1");
577 putenv("PYTHONMALLOCSTATS=1");
578 putenv("PYTHONUTF8=1");
579 putenv("PYTHONVERBOSE=1");
580 putenv("PYTHONINSPECT=1");
581 putenv("PYTHONOPTIMIZE=2");
582 putenv("PYTHONDONTWRITEBYTECODE=1");
583 putenv("PYTHONUNBUFFERED=1");
584 putenv("PYTHONPYCACHEPREFIX=env_pycache_prefix");
585 putenv("PYTHONNOUSERSITE=1");
586 putenv("PYTHONFAULTHANDLER=1");
Victor Stinnerdfe0dc72018-08-29 11:47:29 +0200587 putenv("PYTHONIOENCODING=iso8859-1:replace");
Victor Stinner425717f2019-05-20 16:38:48 +0200588}
589
590
591static void set_all_env_vars(void)
592{
593 set_most_env_vars();
594
595 putenv("PYTHONWARNINGS=EnvVar");
596 putenv("PYTHONPATH=/my/path");
Victor Stinner56b29b62018-07-26 18:57:56 +0200597}
598
599
600static int test_init_env(void)
601{
602 /* Test initialization from environment variables */
603 Py_IgnoreEnvironmentFlag = 0;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200604 set_all_env_vars();
Victor Stinner56b29b62018-07-26 18:57:56 +0200605 _testembed_Py_Initialize();
606 dump_config();
607 Py_Finalize();
608 return 0;
609}
610
611
Victor Stinner6d1c4672019-05-20 11:02:00 +0200612static void set_all_env_vars_dev_mode(void)
Victor Stinner25d13f32019-03-06 12:51:53 +0100613{
Victor Stinner25d13f32019-03-06 12:51:53 +0100614 putenv("PYTHONMALLOC=");
615 putenv("PYTHONFAULTHANDLER=");
616 putenv("PYTHONDEVMODE=1");
617}
618
619
Victor Stinnerb35be4b2019-03-05 17:37:44 +0100620static int test_init_env_dev_mode(void)
621{
622 /* Test initialization from environment variables */
623 Py_IgnoreEnvironmentFlag = 0;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200624 set_all_env_vars_dev_mode();
Victor Stinnerb35be4b2019-03-05 17:37:44 +0100625 _testembed_Py_Initialize();
626 dump_config();
627 Py_Finalize();
628 return 0;
629}
630
631
Victor Stinner25d13f32019-03-06 12:51:53 +0100632static int test_init_env_dev_mode_alloc(void)
633{
634 /* Test initialization from environment variables */
635 Py_IgnoreEnvironmentFlag = 0;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200636 set_all_env_vars_dev_mode();
Victor Stinner25d13f32019-03-06 12:51:53 +0100637 putenv("PYTHONMALLOC=malloc");
638 _testembed_Py_Initialize();
639 dump_config();
640 Py_Finalize();
641 return 0;
642}
643
644
Victor Stinnerbab0db62019-05-18 03:21:27 +0200645static int test_init_isolated_flag(void)
Victor Stinner56b29b62018-07-26 18:57:56 +0200646{
Victor Stinner20004952019-03-26 02:31:11 +0100647 _PyInitError err;
648
Victor Stinner56b29b62018-07-26 18:57:56 +0200649 /* Test _PyCoreConfig.isolated=1 */
Victor Stinnercab5d072019-05-17 19:01:14 +0200650 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200651 err = _PyCoreConfig_InitPythonConfig(&config);
652 if (_PyInitError_Failed(err)) {
653 _Py_ExitInitError(err);
654 }
Victor Stinner56b29b62018-07-26 18:57:56 +0200655
Victor Stinnercad1f742019-03-05 02:01:27 +0100656 Py_IsolatedFlag = 0;
Victor Stinner20004952019-03-26 02:31:11 +0100657 config.isolated = 1;
Victor Stinnercad1f742019-03-05 02:01:27 +0100658
Victor Stinner56b29b62018-07-26 18:57:56 +0200659 /* Use path starting with "./" avoids a search along the PATH */
660 config.program_name = L"./_testembed";
661
Victor Stinnerbab0db62019-05-18 03:21:27 +0200662 set_all_env_vars();
Victor Stinner5ac27a52019-03-27 13:40:14 +0100663 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200664 if (_PyInitError_Failed(err)) {
Victor Stinnerdfe88472019-03-01 12:14:41 +0100665 _Py_ExitInitError(err);
Victor Stinner56b29b62018-07-26 18:57:56 +0200666 }
667 dump_config();
668 Py_Finalize();
669 return 0;
670}
671
672
Victor Stinner6da20a42019-03-27 00:26:18 +0100673/* _PyPreConfig.isolated=1, _PyCoreConfig.isolated=0 */
674static int test_preinit_isolated1(void)
675{
676 _PyInitError err;
677
Victor Stinnercab5d072019-05-17 19:01:14 +0200678 _PyPreConfig preconfig;
679 _PyPreConfig_Init(&preconfig);
Victor Stinner6da20a42019-03-27 00:26:18 +0100680 preconfig.isolated = 1;
681
Victor Stinner5ac27a52019-03-27 13:40:14 +0100682 err = _Py_PreInitialize(&preconfig);
Victor Stinner871ff772019-05-17 23:54:00 +0200683 if (_PyInitError_Failed(err)) {
Victor Stinner6da20a42019-03-27 00:26:18 +0100684 _Py_ExitInitError(err);
685 }
686
Victor Stinnercab5d072019-05-17 19:01:14 +0200687 _PyCoreConfig config;
688 _PyCoreConfig_Init(&config);
Victor Stinner6da20a42019-03-27 00:26:18 +0100689 config.program_name = L"./_testembed";
690
Victor Stinnerbab0db62019-05-18 03:21:27 +0200691 set_all_env_vars();
Victor Stinner5ac27a52019-03-27 13:40:14 +0100692 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200693 if (_PyInitError_Failed(err)) {
Victor Stinner6da20a42019-03-27 00:26:18 +0100694 _Py_ExitInitError(err);
695 }
696 dump_config();
697 Py_Finalize();
698 return 0;
699}
700
701
702/* _PyPreConfig.isolated=0, _PyCoreConfig.isolated=1 */
703static int test_preinit_isolated2(void)
704{
705 _PyInitError err;
706
Victor Stinnercab5d072019-05-17 19:01:14 +0200707 _PyPreConfig preconfig;
708 _PyPreConfig_Init(&preconfig);
Victor Stinner6da20a42019-03-27 00:26:18 +0100709 preconfig.isolated = 0;
710
Victor Stinner5ac27a52019-03-27 13:40:14 +0100711 err = _Py_PreInitialize(&preconfig);
Victor Stinner871ff772019-05-17 23:54:00 +0200712 if (_PyInitError_Failed(err)) {
Victor Stinner6da20a42019-03-27 00:26:18 +0100713 _Py_ExitInitError(err);
714 }
715
716 /* Test _PyCoreConfig.isolated=1 */
Victor Stinnercab5d072019-05-17 19:01:14 +0200717 _PyCoreConfig config;
718 _PyCoreConfig_Init(&config);
Victor Stinner6da20a42019-03-27 00:26:18 +0100719
720 Py_IsolatedFlag = 0;
721 config.isolated = 1;
722
723 /* Use path starting with "./" avoids a search along the PATH */
724 config.program_name = L"./_testembed";
725
Victor Stinnerbab0db62019-05-18 03:21:27 +0200726 set_all_env_vars();
Victor Stinner5ac27a52019-03-27 13:40:14 +0100727 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200728 if (_PyInitError_Failed(err)) {
Victor Stinner6da20a42019-03-27 00:26:18 +0100729 _Py_ExitInitError(err);
730 }
731 dump_config();
732 Py_Finalize();
733 return 0;
734}
735
736
Victor Stinner6d1c4672019-05-20 11:02:00 +0200737static int test_preinit_dont_parse_argv(void)
738{
739 _PyInitError err;
740
741 _PyPreConfig preconfig;
742 _PyPreConfig_InitIsolatedConfig(&preconfig);
743
744 preconfig.isolated = 0;
745
746 /* -X dev must be ignored by isolated preconfiguration */
747 wchar_t *argv[] = {L"python3",
748 L"-E",
749 L"-I",
750 L"-X", L"dev",
751 L"-X", L"utf8",
752 L"script.py"};
753 err = _Py_PreInitializeFromWideArgs(&preconfig, Py_ARRAY_LENGTH(argv), argv);
754 if (_PyInitError_Failed(err)) {
755 goto failed;
756 }
757
758 _PyCoreConfig config;
759
760 err = _PyCoreConfig_InitIsolatedConfig(&config);
761 if (_PyInitError_Failed(err)) {
762 goto failed;
763 }
764
765 config.isolated = 0;
766
767 /* Pre-initialize implicitly using argv: make sure that -X dev
768 is used to configure the allocation in preinitialization */
769 err = _PyCoreConfig_SetWideArgv(&config, Py_ARRAY_LENGTH(argv), argv);
770 if (_PyInitError_Failed(err)) {
771 goto failed;
772 }
773
774 err = _PyCoreConfig_SetString(&config, &config.program_name,
775 L"./_testembed");
776 if (_PyInitError_Failed(err)) {
777 goto failed;
778 }
779
780 err = _Py_InitializeFromConfig(&config);
781 if (_PyInitError_Failed(err)) {
782 goto failed;
783 }
784 _PyCoreConfig_Clear(&config);
785
786 dump_config();
787 Py_Finalize();
788 return 0;
789
790failed:
791 _PyCoreConfig_Clear(&config);
792 _Py_ExitInitError(err);
793}
794
795
796static int test_preinit_parse_argv(void)
797{
798 _PyInitError err;
799 _PyCoreConfig config;
800
801 err = _PyCoreConfig_InitPythonConfig(&config);
802 if (_PyInitError_Failed(err)) {
803 goto failed;
804 }
805
806 /* Pre-initialize implicitly using argv: make sure that -X dev
807 is used to configure the allocation in preinitialization */
808 wchar_t *argv[] = {L"python3", L"-X", L"dev", L"script.py"};
809 err = _PyCoreConfig_SetWideArgv(&config, Py_ARRAY_LENGTH(argv), argv);
810 if (_PyInitError_Failed(err)) {
811 goto failed;
812 }
813
814 err = _PyCoreConfig_SetString(&config, &config.program_name,
815 L"./_testembed");
816 if (_PyInitError_Failed(err)) {
817 goto failed;
818 }
819
820 err = _Py_InitializeFromConfig(&config);
821 if (_PyInitError_Failed(err)) {
822 goto failed;
823 }
824 _PyCoreConfig_Clear(&config);
825
826 dump_config();
827 Py_Finalize();
828 return 0;
829
830failed:
831 _PyCoreConfig_Clear(&config);
832 _Py_ExitInitError(err);
833}
834
835
836
837
Victor Stinnerbab0db62019-05-18 03:21:27 +0200838static void set_all_global_config_variables(void)
839{
840 Py_IsolatedFlag = 0;
841 Py_IgnoreEnvironmentFlag = 0;
842 Py_BytesWarningFlag = 2;
843 Py_InspectFlag = 1;
844 Py_InteractiveFlag = 1;
845 Py_OptimizeFlag = 1;
846 Py_DebugFlag = 1;
847 Py_VerboseFlag = 1;
848 Py_QuietFlag = 1;
849 Py_FrozenFlag = 0;
850 Py_UnbufferedStdioFlag = 1;
851 Py_NoSiteFlag = 1;
852 Py_DontWriteBytecodeFlag = 1;
853 Py_NoUserSiteDirectory = 1;
854#ifdef MS_WINDOWS
855 Py_LegacyWindowsStdioFlag = 1;
856#endif
857}
858
859
Victor Stinner6d1c4672019-05-20 11:02:00 +0200860static int check_preinit_isolated_config(int preinit)
Victor Stinner56b29b62018-07-26 18:57:56 +0200861{
Victor Stinnercab5d072019-05-17 19:01:14 +0200862 _PyInitError err;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200863 _PyPreConfig *rt_preconfig;
Victor Stinnercab5d072019-05-17 19:01:14 +0200864
Victor Stinnerbab0db62019-05-18 03:21:27 +0200865 /* environment variables must be ignored */
866 set_all_env_vars();
867
868 /* global configuration variables must be ignored */
869 set_all_global_config_variables();
870
Victor Stinner6d1c4672019-05-20 11:02:00 +0200871 if (preinit) {
872 _PyPreConfig preconfig;
873 _PyPreConfig_InitIsolatedConfig(&preconfig);
Victor Stinnercab5d072019-05-17 19:01:14 +0200874
Victor Stinner6d1c4672019-05-20 11:02:00 +0200875 err = _Py_PreInitialize(&preconfig);
876 if (_PyInitError_Failed(err)) {
877 _Py_ExitInitError(err);
878 }
879
880 rt_preconfig = &_PyRuntime.preconfig;
881 assert(rt_preconfig->isolated == 1);
882 assert(rt_preconfig->use_environment == 0);
Victor Stinnercab5d072019-05-17 19:01:14 +0200883 }
884
Victor Stinnercab5d072019-05-17 19:01:14 +0200885 _PyCoreConfig config;
886 err = _PyCoreConfig_InitIsolatedConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200887 if (_PyInitError_Failed(err)) {
Victor Stinnercab5d072019-05-17 19:01:14 +0200888 _Py_ExitInitError(err);
889 }
890 config.program_name = L"./_testembed";
891
892 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200893 if (_PyInitError_Failed(err)) {
Victor Stinnercab5d072019-05-17 19:01:14 +0200894 _Py_ExitInitError(err);
895 }
Victor Stinner6d1c4672019-05-20 11:02:00 +0200896
897 rt_preconfig = &_PyRuntime.preconfig;
898 assert(rt_preconfig->isolated == 1);
899 assert(rt_preconfig->use_environment == 0);
900
Victor Stinnercab5d072019-05-17 19:01:14 +0200901 dump_config();
902 Py_Finalize();
903 return 0;
904}
905
906
Victor Stinner6d1c4672019-05-20 11:02:00 +0200907static int test_preinit_isolated_config(void)
908{
909 return check_preinit_isolated_config(1);
910}
911
912
913static int test_init_isolated_config(void)
914{
915 return check_preinit_isolated_config(0);
916}
917
918
919static int check_init_python_config(int preinit)
Victor Stinnercab5d072019-05-17 19:01:14 +0200920{
921 _PyInitError err;
922
Victor Stinnerbab0db62019-05-18 03:21:27 +0200923 /* global configuration variables must be ignored */
924 set_all_global_config_variables();
925 Py_IsolatedFlag = 1;
926 Py_IgnoreEnvironmentFlag = 1;
927 Py_FrozenFlag = 1;
928 Py_UnbufferedStdioFlag = 1;
929 Py_NoSiteFlag = 1;
930 Py_DontWriteBytecodeFlag = 1;
931 Py_NoUserSiteDirectory = 1;
932#ifdef MS_WINDOWS
933 Py_LegacyWindowsStdioFlag = 1;
934#endif
935
Victor Stinner6d1c4672019-05-20 11:02:00 +0200936 if (preinit) {
937 _PyPreConfig preconfig;
938 _PyPreConfig_InitPythonConfig(&preconfig);
Victor Stinnercab5d072019-05-17 19:01:14 +0200939
Victor Stinner6d1c4672019-05-20 11:02:00 +0200940 err = _Py_PreInitialize(&preconfig);
941 if (_PyInitError_Failed(err)) {
942 _Py_ExitInitError(err);
943 }
Victor Stinnercab5d072019-05-17 19:01:14 +0200944 }
945
946 _PyCoreConfig config;
947 err = _PyCoreConfig_InitPythonConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200948 if (_PyInitError_Failed(err)) {
Victor Stinnercab5d072019-05-17 19:01:14 +0200949 _Py_ExitInitError(err);
950 }
951 config.program_name = L"./_testembed";
952
953 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200954 if (_PyInitError_Failed(err)) {
Victor Stinnercab5d072019-05-17 19:01:14 +0200955 _Py_ExitInitError(err);
956 }
957 dump_config();
958 Py_Finalize();
959 return 0;
960}
961
962
Victor Stinner6d1c4672019-05-20 11:02:00 +0200963static int test_preinit_python_config(void)
964{
965 return check_init_python_config(1);
966}
967
968
969static int test_init_python_config(void)
970{
971 return check_init_python_config(0);
972}
973
974
Victor Stinnerbab0db62019-05-18 03:21:27 +0200975static int test_init_dont_configure_locale(void)
Victor Stinnerbcfbbd72019-05-17 22:44:16 +0200976{
977 _PyInitError err;
978
Victor Stinnerbab0db62019-05-18 03:21:27 +0200979 _PyPreConfig preconfig;
980 _PyPreConfig_InitPythonConfig(&preconfig);
Victor Stinnerbcfbbd72019-05-17 22:44:16 +0200981 preconfig.configure_locale = 0;
982 preconfig.coerce_c_locale = 1;
983 preconfig.coerce_c_locale_warn = 1;
984
985 err = _Py_PreInitialize(&preconfig);
Victor Stinner871ff772019-05-17 23:54:00 +0200986 if (_PyInitError_Failed(err)) {
Victor Stinnerbcfbbd72019-05-17 22:44:16 +0200987 _Py_ExitInitError(err);
988 }
989
Victor Stinnerbab0db62019-05-18 03:21:27 +0200990 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +0200991 err = _PyCoreConfig_InitPythonConfig(&config);
992 if (_PyInitError_Failed(err)) {
993 _Py_ExitInitError(err);
994 }
Victor Stinnerbcfbbd72019-05-17 22:44:16 +0200995 config.program_name = L"./_testembed";
996 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +0200997 if (_PyInitError_Failed(err)) {
Victor Stinnerbcfbbd72019-05-17 22:44:16 +0200998 _Py_ExitInitError(err);
999 }
1000
1001 dump_config();
1002 Py_Finalize();
1003 return 0;
1004}
1005
1006
Victor Stinnerbab0db62019-05-18 03:21:27 +02001007static int test_init_dev_mode(void)
Victor Stinnercab5d072019-05-17 19:01:14 +02001008{
Victor Stinner6d1c4672019-05-20 11:02:00 +02001009 _PyInitError err;
Victor Stinnercab5d072019-05-17 19:01:14 +02001010 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +02001011 err = _PyCoreConfig_InitPythonConfig(&config);
1012 if (_PyInitError_Failed(err)) {
1013 _Py_ExitInitError(err);
1014 }
Victor Stinner56b29b62018-07-26 18:57:56 +02001015 putenv("PYTHONFAULTHANDLER=");
1016 putenv("PYTHONMALLOC=");
Victor Stinner20004952019-03-26 02:31:11 +01001017 config.dev_mode = 1;
Victor Stinner56b29b62018-07-26 18:57:56 +02001018 config.program_name = L"./_testembed";
Victor Stinner6d1c4672019-05-20 11:02:00 +02001019 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +02001020 if (_PyInitError_Failed(err)) {
Victor Stinnerdfe88472019-03-01 12:14:41 +01001021 _Py_ExitInitError(err);
Victor Stinner56b29b62018-07-26 18:57:56 +02001022 }
1023 dump_config();
1024 Py_Finalize();
1025 return 0;
1026}
1027
1028
Victor Stinner91c99872019-05-14 22:01:51 +02001029static int test_init_read_set(void)
1030{
1031 _PyInitError err;
Victor Stinnercab5d072019-05-17 19:01:14 +02001032 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +02001033 err = _PyCoreConfig_InitPythonConfig(&config);
1034 if (_PyInitError_Failed(err)) {
1035 _Py_ExitInitError(err);
1036 }
Victor Stinner91c99872019-05-14 22:01:51 +02001037
Victor Stinner6d1c4672019-05-20 11:02:00 +02001038 err = _PyCoreConfig_DecodeLocale(&config, &config.program_name,
1039 "./init_read_set");
Victor Stinner871ff772019-05-17 23:54:00 +02001040 if (_PyInitError_Failed(err)) {
Victor Stinner91c99872019-05-14 22:01:51 +02001041 goto fail;
1042 }
1043
1044 err = _PyCoreConfig_Read(&config);
Victor Stinner871ff772019-05-17 23:54:00 +02001045 if (_PyInitError_Failed(err)) {
Victor Stinner91c99872019-05-14 22:01:51 +02001046 goto fail;
1047 }
1048
1049 if (_PyWstrList_Append(&config.module_search_paths,
1050 L"init_read_set_path") < 0) {
Victor Stinner871ff772019-05-17 23:54:00 +02001051 err = _PyInitError_NoMemory();
Victor Stinner91c99872019-05-14 22:01:51 +02001052 goto fail;
1053 }
1054
1055 /* override executable computed by _PyCoreConfig_Read() */
Victor Stinner6d1c4672019-05-20 11:02:00 +02001056 err = _PyCoreConfig_SetString(&config, &config.executable, L"my_executable");
Victor Stinner871ff772019-05-17 23:54:00 +02001057 if (_PyInitError_Failed(err)) {
Victor Stinner91c99872019-05-14 22:01:51 +02001058 goto fail;
1059 }
1060
1061 err = _Py_InitializeFromConfig(&config);
1062 _PyCoreConfig_Clear(&config);
Victor Stinner871ff772019-05-17 23:54:00 +02001063 if (_PyInitError_Failed(err)) {
Victor Stinner91c99872019-05-14 22:01:51 +02001064 goto fail;
1065 }
1066 dump_config();
1067 Py_Finalize();
1068 return 0;
1069
1070fail:
1071 _Py_ExitInitError(err);
1072}
1073
1074
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001075wchar_t *init_main_argv[] = {
1076 L"python3", L"-c",
1077 (L"import _testinternalcapi, json; "
1078 L"print(json.dumps(_testinternalcapi.get_configs()))"),
1079 L"arg2"};
1080
1081
1082static void configure_init_main(_PyCoreConfig *config)
1083{
1084 config->argv.length = Py_ARRAY_LENGTH(init_main_argv);
1085 config->argv.items = init_main_argv;
Victor Stinnercab5d072019-05-17 19:01:14 +02001086 config->parse_argv = 1;
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001087 config->program_name = L"./python3";
1088}
1089
1090
1091static int test_init_run_main(void)
Victor Stinner2f549082019-03-29 15:13:46 +01001092{
Victor Stinner6d1c4672019-05-20 11:02:00 +02001093 _PyInitError err;
Victor Stinnercab5d072019-05-17 19:01:14 +02001094 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +02001095 err = _PyCoreConfig_InitPythonConfig(&config);
1096 if (_PyInitError_Failed(err)) {
1097 _Py_ExitInitError(err);
1098 }
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001099 configure_init_main(&config);
Victor Stinner2f549082019-03-29 15:13:46 +01001100
Victor Stinner6d1c4672019-05-20 11:02:00 +02001101 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +02001102 if (_PyInitError_Failed(err)) {
Victor Stinner2f549082019-03-29 15:13:46 +01001103 _Py_ExitInitError(err);
1104 }
1105
1106 return _Py_RunMain();
1107}
1108
1109
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001110static int test_init_main(void)
1111{
Victor Stinner6d1c4672019-05-20 11:02:00 +02001112 _PyInitError err;
Victor Stinnercab5d072019-05-17 19:01:14 +02001113 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +02001114
1115 err = _PyCoreConfig_InitPythonConfig(&config);
1116 if (_PyInitError_Failed(err)) {
1117 _Py_ExitInitError(err);
1118 }
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001119 configure_init_main(&config);
1120 config._init_main = 0;
1121
Victor Stinner6d1c4672019-05-20 11:02:00 +02001122 err = _Py_InitializeFromConfig(&config);
Victor Stinner871ff772019-05-17 23:54:00 +02001123 if (_PyInitError_Failed(err)) {
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001124 _Py_ExitInitError(err);
1125 }
1126
1127 /* sys.stdout don't exist yet: it is created by _Py_InitializeMain() */
1128 int res = PyRun_SimpleString(
1129 "import sys; "
1130 "print('Run Python code before _Py_InitializeMain', "
1131 "file=sys.stderr)");
1132 if (res < 0) {
1133 exit(1);
1134 }
1135
1136 err = _Py_InitializeMain();
Victor Stinner871ff772019-05-17 23:54:00 +02001137 if (_PyInitError_Failed(err)) {
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001138 _Py_ExitInitError(err);
1139 }
1140
1141 return _Py_RunMain();
1142}
1143
1144
1145static int test_run_main(void)
Victor Stinner5eb8b072019-05-15 02:12:48 +02001146{
Victor Stinner6d1c4672019-05-20 11:02:00 +02001147 _PyInitError err;
Victor Stinnercab5d072019-05-17 19:01:14 +02001148 _PyCoreConfig config;
Victor Stinner6d1c4672019-05-20 11:02:00 +02001149
1150 err = _PyCoreConfig_InitPythonConfig(&config);
1151 if (_PyInitError_Failed(err)) {
1152 goto failed;
1153 }
Victor Stinner5eb8b072019-05-15 02:12:48 +02001154
1155 wchar_t *argv[] = {L"python3", L"-c",
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001156 (L"import sys; "
1157 L"print(f'_Py_RunMain(): sys.argv={sys.argv}')"),
Victor Stinner5eb8b072019-05-15 02:12:48 +02001158 L"arg2"};
Victor Stinner6d1c4672019-05-20 11:02:00 +02001159 err = _PyCoreConfig_SetWideArgv(&config, Py_ARRAY_LENGTH(argv), argv);
Victor Stinner871ff772019-05-17 23:54:00 +02001160 if (_PyInitError_Failed(err)) {
Victor Stinner6d1c4672019-05-20 11:02:00 +02001161 goto failed;
Victor Stinner5eb8b072019-05-15 02:12:48 +02001162 }
1163
Victor Stinner6d1c4672019-05-20 11:02:00 +02001164 err = _PyCoreConfig_SetString(&config, &config.program_name,
1165 L"./python3");
1166 if (_PyInitError_Failed(err)) {
1167 goto failed;
1168 }
1169
1170 err = _Py_InitializeFromConfig(&config);
1171 if (_PyInitError_Failed(err)) {
1172 goto failed;
1173 }
1174 _PyCoreConfig_Clear(&config);
1175
Victor Stinner5eb8b072019-05-15 02:12:48 +02001176 return _Py_RunMain();
Victor Stinner6d1c4672019-05-20 11:02:00 +02001177
1178failed:
1179 _PyCoreConfig_Clear(&config);
1180 _Py_ExitInitError(err);
Victor Stinner5eb8b072019-05-15 02:12:48 +02001181}
1182
1183
Steve Dowerea74f0c2017-01-01 20:25:03 -08001184/* *********************************************************
1185 * List of test cases and the function that implements it.
Serhiy Storchaka13ad3b72017-09-14 09:38:36 +03001186 *
Steve Dowerea74f0c2017-01-01 20:25:03 -08001187 * Names are compared case-sensitively with the first
1188 * argument. If no match is found, or no first argument was
1189 * provided, the names of all test cases are printed and
1190 * the exit code will be -1.
1191 *
1192 * The int returned from test functions is used as the exit
1193 * code, and test_capi treats all non-zero exit codes as a
Serhiy Storchaka13ad3b72017-09-14 09:38:36 +03001194 * failed test.
Steve Dowerea74f0c2017-01-01 20:25:03 -08001195 *********************************************************/
1196struct TestCase
1197{
1198 const char *name;
1199 int (*func)(void);
1200};
1201
1202static struct TestCase TestCases[] = {
1203 { "forced_io_encoding", test_forced_io_encoding },
1204 { "repeated_init_and_subinterpreters", test_repeated_init_and_subinterpreters },
Victor Stinner9e87e772017-11-24 12:09:24 +01001205 { "pre_initialization_api", test_pre_initialization_api },
Nick Coghlanbc77eff2018-03-25 20:44:30 +10001206 { "pre_initialization_sys_options", test_pre_initialization_sys_options },
Victor Stinnerb4d1e1f2017-11-30 22:05:00 +01001207 { "bpo20891", test_bpo20891 },
Victor Stinner209abf72018-06-22 19:14:51 +02001208 { "initialize_twice", test_initialize_twice },
Victor Stinnerfb47bca2018-07-20 17:34:23 +02001209 { "initialize_pymain", test_initialize_pymain },
Victor Stinner56b29b62018-07-26 18:57:56 +02001210 { "init_default_config", test_init_default_config },
1211 { "init_global_config", test_init_global_config },
1212 { "init_from_config", test_init_from_config },
Victor Stinnerbab0db62019-05-18 03:21:27 +02001213 { "init_parse_argv", test_init_parse_argv },
1214 { "init_dont_parse_argv", test_init_dont_parse_argv },
Victor Stinner56b29b62018-07-26 18:57:56 +02001215 { "init_env", test_init_env },
Victor Stinnerb35be4b2019-03-05 17:37:44 +01001216 { "init_env_dev_mode", test_init_env_dev_mode },
Victor Stinner25d13f32019-03-06 12:51:53 +01001217 { "init_env_dev_mode_alloc", test_init_env_dev_mode_alloc },
Victor Stinnerbab0db62019-05-18 03:21:27 +02001218 { "init_dont_configure_locale", test_init_dont_configure_locale },
1219 { "init_dev_mode", test_init_dev_mode },
1220 { "init_isolated_flag", test_init_isolated_flag },
Victor Stinner6d1c4672019-05-20 11:02:00 +02001221 { "preinit_isolated_config", test_preinit_isolated_config },
Victor Stinnerbab0db62019-05-18 03:21:27 +02001222 { "init_isolated_config", test_init_isolated_config },
Victor Stinner6d1c4672019-05-20 11:02:00 +02001223 { "preinit_python_config", test_preinit_python_config },
Victor Stinnerbab0db62019-05-18 03:21:27 +02001224 { "init_python_config", test_init_python_config },
Victor Stinner6da20a42019-03-27 00:26:18 +01001225 { "preinit_isolated1", test_preinit_isolated1 },
1226 { "preinit_isolated2", test_preinit_isolated2 },
Victor Stinner6d1c4672019-05-20 11:02:00 +02001227 { "preinit_parse_argv", test_preinit_parse_argv },
1228 { "preinit_dont_parse_argv", test_preinit_dont_parse_argv },
Victor Stinner91c99872019-05-14 22:01:51 +02001229 { "init_read_set", test_init_read_set },
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001230 { "init_run_main", test_init_run_main },
1231 { "init_main", test_init_main },
Victor Stinner2f549082019-03-29 15:13:46 +01001232 { "run_main", test_run_main },
Steve Dowerea74f0c2017-01-01 20:25:03 -08001233 { NULL, NULL }
1234};
1235
Nick Coghlan7d270ee2013-10-17 22:35:35 +10001236int main(int argc, char *argv[])
1237{
Nick Coghlan7d270ee2013-10-17 22:35:35 +10001238 if (argc > 1) {
Steve Dowerea74f0c2017-01-01 20:25:03 -08001239 for (struct TestCase *tc = TestCases; tc && tc->name; tc++) {
1240 if (strcmp(argv[1], tc->name) == 0)
1241 return (*tc->func)();
1242 }
Nick Coghlan7d270ee2013-10-17 22:35:35 +10001243 }
Steve Dowerea74f0c2017-01-01 20:25:03 -08001244
1245 /* No match found, or no test name provided, so display usage */
1246 printf("Python " PY_VERSION " _testembed executable for embedded interpreter tests\n"
Nick Coghlanbc77eff2018-03-25 20:44:30 +10001247 "Normally executed via 'EmbeddingTests' in Lib/test/test_embed.py\n\n"
Steve Dowerea74f0c2017-01-01 20:25:03 -08001248 "Usage: %s TESTNAME\n\nAll available tests:\n", argv[0]);
1249 for (struct TestCase *tc = TestCases; tc && tc->name; tc++) {
1250 printf(" %s\n", tc->name);
1251 }
1252
Nick Coghlanbc77eff2018-03-25 20:44:30 +10001253 /* Non-zero exit code will cause test_embed.py tests to fail.
Steve Dowerea74f0c2017-01-01 20:25:03 -08001254 This is intentional. */
1255 return -1;
Antoine Pitrou8e605772011-04-25 21:21:07 +02001256}