diff options
Diffstat (limited to 'Python')
| -rw-r--r-- | Python/initconfig.c | 9 | ||||
| -rw-r--r-- | Python/pylifecycle.c | 20 |
2 files changed, 29 insertions, 0 deletions
diff --git a/Python/initconfig.c b/Python/initconfig.c index d328f22..61cd0e6 100644 --- a/Python/initconfig.c +++ b/Python/initconfig.c @@ -898,6 +898,7 @@ _PyConfig_Copy(PyConfig *config, const PyConfig *config2) COPY_ATTR(no_debug_ranges); COPY_ATTR(show_ref_count); COPY_ATTR(dump_refs); + COPY_ATTR(dump_refs_file); COPY_ATTR(malloc_stats); COPY_WSTR_ATTR(pycache_prefix); @@ -1701,6 +1702,14 @@ config_read_env_vars(PyConfig *config) config->malloc_stats = 1; } + if (config->dump_refs_file == NULL) { + status = CONFIG_GET_ENV_DUP(config, &config->dump_refs_file, + L"PYTHONDUMPREFSFILE", "PYTHONDUMPREFSFILE"); + if (_PyStatus_EXCEPTION(status)) { + return status; + } + } + if (config->pythonpath_env == NULL) { status = CONFIG_GET_ENV_DUP(config, &config->pythonpath_env, L"PYTHONPATH", "PYTHONPATH"); diff --git a/Python/pylifecycle.c b/Python/pylifecycle.c index eeaf20b..f3b6b0a 100644 --- a/Python/pylifecycle.c +++ b/Python/pylifecycle.c @@ -1737,6 +1737,7 @@ Py_FinalizeEx(void) #endif #ifdef Py_TRACE_REFS int dump_refs = tstate->interp->config.dump_refs; + wchar_t *dump_refs_file = tstate->interp->config.dump_refs_file; #endif #ifdef WITH_PYMALLOC int malloc_stats = tstate->interp->config.malloc_stats; @@ -1835,9 +1836,22 @@ Py_FinalizeEx(void) * Alas, a lot of stuff may still be alive now that will be cleaned * up later. */ + + FILE *dump_refs_fp = NULL; + if (dump_refs_file != NULL) { + dump_refs_fp = _Py_wfopen(dump_refs_file, L"w"); + if (dump_refs_fp == NULL) { + fprintf(stderr, "PYTHONDUMPREFSFILE: cannot create file: %ls\n", dump_refs_file); + } + } + if (dump_refs) { _Py_PrintReferences(stderr); } + + if (dump_refs_fp != NULL) { + _Py_PrintReferences(dump_refs_fp); + } #endif /* Py_TRACE_REFS */ finalize_interp_clear(tstate); @@ -1848,9 +1862,15 @@ Py_FinalizeEx(void) * An address can be used to find the repr of the object, printed * above by _Py_PrintReferences. */ + if (dump_refs) { _Py_PrintReferenceAddresses(stderr); } + + if (dump_refs_fp != NULL) { + _Py_PrintReferenceAddresses(dump_refs_fp); + fclose(dump_refs_fp); + } #endif /* Py_TRACE_REFS */ #ifdef WITH_PYMALLOC if (malloc_stats) { |