From 7929e2e319f2cd9f5231b4ba0a626f80dc1cfcd1 Mon Sep 17 00:00:00 2001
From: "Erlend E. Aasland" <erlend@python.org>
Date: Mon, 27 Jan 2025 18:16:19 +0100
Subject: [PATCH] gh-129346: Handle allocation errors for SQLite aggregate
context (GH-129347) (cherry picked from commit
379ab856f59423c570333403a7d5d72f3ea82d52)
Co-authored-by: Erlend E. Aasland <erlend@python.org>
---
.../Library/2025-01-27-14-05-19.gh-issue-129346.gZRd3g.rst | 2 ++
Modules/_sqlite/connection.c | 5 +++++
2 files changed, 7 insertions(+)
create mode 100644 Misc/NEWS.d/next/Library/2025-01-27-14-05-19.gh-issue-129346.gZRd3g.rst
diff --git a/Misc/NEWS.d/next/Library/2025-01-27-14-05-19.gh-issue-129346.gZRd3g.rst b/Misc/NEWS.d/next/Library/2025-01-27-14-05-19.gh-issue-129346.gZRd3g.rst
new file mode 100644
index 00000000000000..b5377277f6c51c
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2025-01-27-14-05-19.gh-issue-129346.gZRd3g.rst
@@ -0,0 +1,2 @@
+In :mod:`sqlite3`, handle out-of-memory when creating user-defined SQL
+functions.
diff --git a/Modules/_sqlite/connection.c b/Modules/_sqlite/connection.c
index 12e5c135aafa50..1450037ca95c83 100644
--- a/Modules/_sqlite/connection.c
+++ b/Modules/_sqlite/connection.c
@@ -927,6 +927,11 @@ step_callback(sqlite3_context *context, int argc, sqlite3_value **params)
assert(ctx != NULL);
aggregate_instance = (PyObject**)sqlite3_aggregate_context(context, sizeof(PyObject*));
+ if (aggregate_instance == NULL) {
+ (void)PyErr_NoMemory();
+ set_sqlite_error(context, "unable to allocate SQLite aggregate context");
+ goto error;
+ }
if (*aggregate_instance == NULL) {
*aggregate_instance = PyObject_CallNoArgs(ctx->callable);
if (!*aggregate_instance) {