Files
test2/source/blender/python/intern/bpy_capi_utils.cc
Campbell Barton 6a0f98aeef Fix memory leak in PyC_ExceptionBuffer for Py 3.12
Non-matching calls to PyErr_Fetch/Restore cause a leak in v3.12,
so ensure calls are symmetrical or avoid where possible.

Simplify extraction of the exception buffer.

- Only overwrite the stderr (the stdio isn't used).
- Simplify pyc_exception_buffer_handle_system_exit usage.
- Remove goto's.

Also simplify calling conventions for PyC_ExceptionBuffer functions.
- They must be called when an error has occurred.
- Always return a string, never null since a null return value would
  only happened in rare/unexpected cases which wasn't being checked
  for by some callers, leading to potential crashes.
2023-09-18 22:09:19 +10:00

115 lines
2.8 KiB
C++

/* SPDX-FileCopyrightText: 2023 Blender Authors
*
* SPDX-License-Identifier: GPL-2.0-or-later */
/** \file
* \ingroup pythonintern
*
* This file contains Blender/Python utility functions to help implementing API's.
* This is not related to a particular module.
*/
#include <Python.h>
#include "BLI_dynstr.h"
#include "BLI_listbase.h"
#include "BLI_utildefines.h"
#include "bpy_capi_utils.h"
#include "MEM_guardedalloc.h"
#include "BKE_context.h"
#include "BKE_report.h"
#include "BLT_translation.h"
#include "../generic/py_capi_utils.h"
short BPy_reports_to_error(ReportList *reports, PyObject *exception, const bool clear)
{
char *report_str;
report_str = BKE_reports_string(reports, RPT_ERROR);
if (clear == true) {
BKE_reports_clear(reports);
}
if (report_str) {
PyErr_SetString(exception, report_str);
MEM_freeN(report_str);
}
return (report_str == nullptr) ? 0 : -1;
}
void BPy_reports_write_stdout(const ReportList *reports, const char *header)
{
if (header) {
PySys_WriteStdout("%s\n", header);
}
LISTBASE_FOREACH (const Report *, report, &reports->list) {
PySys_WriteStdout("%s: %s\n", report->typestr, report->message);
}
}
bool BPy_errors_to_report_ex(ReportList *reports,
const char *err_prefix,
const bool use_full,
const bool use_location)
{
if (!PyErr_Occurred()) {
return true;
}
PyObject *err_str_py = use_full ? PyC_ExceptionBuffer() : PyC_ExceptionBuffer_Simple();
/* Strip trailing newlines so the report doesn't show a blank-line in the info space. */
Py_ssize_t err_str_len;
const char *err_str = PyUnicode_AsUTF8AndSize(err_str_py, &err_str_len);
while (err_str_len > 0 && err_str[err_str_len - 1] == '\n') {
err_str_len -= 1;
}
if (err_prefix == nullptr) {
/* Not very helpful, better than nothing. */
err_prefix = "Python";
}
const char *location_filepath = nullptr;
int location_line_number = -1;
/* Give some additional context. */
if (use_location) {
PyC_FileAndNum(&location_filepath, &location_line_number);
}
if (location_filepath) {
BKE_reportf(reports,
RPT_ERROR,
"%s: %.*s\n"
/* Location (when available). */
"Location: %s:%d",
err_prefix,
int(err_str_len),
err_str,
location_filepath,
location_line_number);
}
else {
BKE_reportf(reports, RPT_ERROR, "%s: %.*s", err_prefix, int(err_str_len), err_str);
}
/* Ensure this is _always_ printed to the output so developers don't miss exceptions. */
Py_DECREF(err_str_py);
return true;
}
bool BPy_errors_to_report(ReportList *reports)
{
return BPy_errors_to_report_ex(reports, nullptr, true, true);
}